class MobileBackup2(MobileBackup):
service = None
def __init__(self, lockdown = None,backupPath = None, password=""):
if lockdown:
self.lockdown = lockdown
else:
self.lockdown = LockdownClient()
ProductVersion = self.lockdown.getValue("", "ProductVersion")
if ProductVersion and int(ProductVersion[:ProductVersion.find('.')]) < 5:
raise DeviceVersionNotSupported
self.udid = lockdown.getValue("", "UniqueDeviceID")
self.willEncrypt = lockdown.getValue("com.apple.mobile.backup", "WillEncrypt")
self.escrowBag = lockdown.getValue('', 'EscrowBag')
self.service = self.lockdown.startServiceWithEscrowBag("com.apple.mobilebackup2", self.escrowBag)
if not self.service:
raise Exception("MobileBackup2 init error : Could not start com.apple.mobilebackup2")
if backupPath:
self.backupPath = backupPath
else:
self.backupPath = "backups"
if not os.path.isdir(self.backupPath):
os.makedirs(self.backupPath,0o0755)
print "Starting new com.apple.mobilebackup2 service with working dir: %s" % self.backupPath
self.password = password
DLMessageVersionExchange = self.service.recvPlist()
version_major = DLMessageVersionExchange[1]
self.service.sendPlist(["DLMessageVersionExchange", "DLVersionsOk", version_major])
DLMessageDeviceReady = self.service.recvPlist()
if DLMessageDeviceReady and DLMessageDeviceReady[0] == "DLMessageDeviceReady":
self.version_exchange()
else:
raise Exception("MobileBackup2 init error %s" % DLMessageDeviceReady)
def __del__(self):
if self.service:
self.service.sendPlist(["DLMessageDisconnect", "___EmptyParameterString___"])
def internal_mobilebackup2_send_message(self, name, data):
data["MessageName"] = name
self.device_link_service_send_process_message(data)
def internal_mobilebackup2_receive_message(self, name=None):
res = self.device_link_service_receive_process_message()
if res:
if name and res["MessageName"] != name:
print "MessageName does not match %s %s" % (name, str(res))
return res
def version_exchange(self):
self.internal_mobilebackup2_send_message("Hello", {"SupportedProtocolVersions": [2.0,2.1]})
return self.internal_mobilebackup2_receive_message("Response")
def mobilebackup2_send_request(self, request, target, source, options={}):
d = {"TargetIdentifier": target,
"SourceIdentifier": source,
"Options": options}
self.internal_mobilebackup2_send_message(request, d)
def mobilebackup2_receive_message(self):
return self.service.recvPlist()
def mobilebackup2_send_status_response(self, status_code, status1="___EmptyParameterString___", status2={}):
a = ["DLMessageStatusResponse", status_code, status1, status2]
self.service.sendPlist(a)
def mb2_handle_free_disk_space(self,msg):
s = os.statvfs(self.backupPath)
freeSpace = s.f_bsize * s.f_bavail
a = ["DLMessageStatusResponse", 0, freeSpace]
self.service.sendPlist(a)
def mb2_multi_status_add_file_error(self, errplist, path, error_code, error_message):
errplist[path] = {"DLFileErrorCode": error_code, "DLFileErrorString": error_message}
def mb2_handle_copy_item(self, msg):
src = self.check_filename(msg[1])
dst = self.check_filename(msg[2])
if os.path.isfile(src):
data = self.read_file(src)
self.write_file(dst, data)
else:
os.makedirs(dst)
self.mobilebackup2_send_status_response(0)
def mb2_handle_send_file(self, filename, errplist):
self.service.send_raw(filename)
if not filename.startswith(self.udid):
filename = self.udid + "/" + filename
data = self.read_file(self.check_filename(filename))
if data != None:
print "Sending %s to device" % filename
self.service.send_raw(chr(CODE_FILE_DATA) + data)
self.service.send_raw(chr(CODE_SUCCESS))
else:
print "File %s requested from device not found" % filename
self.service.send_raw(chr(CODE_ERROR_LOCAL))
self.mb2_multi_status_add_file_error(errplist, filename, ERROR_ENOENT, "Could not find the droid you were looking for ;)")
def mb2_handle_send_files(self, msg):
errplist = {}
for f in msg[1]:
self.mb2_handle_send_file(f, errplist)
self.service.send("\x00\x00\x00\x00")
if len(errplist):
self.mobilebackup2_send_status_response(-13, "Multi status", errplist)
else:
self.mobilebackup2_send_status_response(0)
def mb2_handle_list_directory(self, msg):
path = msg[1]
dirlist = {}
self.mobilebackup2_send_status_response(0, status2=dirlist);
def mb2_handle_make_directory(self, msg):
dirname = self.check_filename(msg[1])
print "Creating directory %s" % dirname
if not os.path.isdir(dirname):
os.makedirs(dirname)
self.mobilebackup2_send_status_response(0, "")
def mb2_handle_receive_files(self, msg):
done = 0
while not done:
device_filename = self.service.recv_raw()
if device_filename == "":
break
backup_filename = self.service.recv_raw()
filedata = ""
while True:
stuff = self.service.recv_raw()
if ord(stuff[0]) == CODE_FILE_DATA:
filedata += stuff[1:]
elif ord(stuff[0]) == CODE_SUCCESS:
self.write_file(self.check_filename(backup_filename), filedata)
break
else:
print "Unknown code", ord(stuff[0])
break
self.mobilebackup2_send_status_response(0)
def mb2_handle_move_files(self, msg):
for k,v in msg[1].items():
print "Renaming %s to %s" % (self.check_filename(k),self.check_filename(v))
os.rename(self.check_filename(k),self.check_filename(v))
self.mobilebackup2_send_status_response(0)
def mb2_handle_remove_files(self, msg):
for filename in msg[1]:
print "Removing ", self.check_filename(filename)
try:
filename = self.check_filename(filename)
if os.path.isfile(filename):
os.unlink(filename)
except Exception, e:
print e
self.mobilebackup2_send_status_response(0)
