mobilebackup.pxi 5.0 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106
  1. cdef extern from "libimobiledevice/mobilebackup.h":
  2. cdef struct mobilebackup_client_private:
  3. pass
  4. ctypedef mobilebackup_client_private *mobilebackup_client_t
  5. ctypedef enum mobilebackup_error_t:
  6. MOBILEBACKUP_E_SUCCESS = 0
  7. MOBILEBACKUP_E_INVALID_ARG = -1
  8. MOBILEBACKUP_E_PLIST_ERROR = -2
  9. MOBILEBACKUP_E_MUX_ERROR = -3
  10. MOBILEBACKUP_E_BAD_VERSION = -4
  11. MOBILEBACKUP_E_REPLY_NOT_OK = -5
  12. MOBILEBACKUP_E_UNKNOWN_ERROR = -256
  13. ctypedef enum mobilebackup_flags_t:
  14. MB_RESTORE_NOTIFY_SPRINGBOARD = (1 << 0)
  15. MB_RESTORE_PRESERVE_SETTINGS = (1 << 1)
  16. MB_RESTORE_PRESERVE_CAMERA_ROLL = (1 << 2)
  17. mobilebackup_error_t mobilebackup_client_new(idevice_t device, lockdownd_service_descriptor_t descriptor, mobilebackup_client_t * client)
  18. mobilebackup_error_t mobilebackup_client_free(mobilebackup_client_t client)
  19. mobilebackup_error_t mobilebackup_receive(mobilebackup_client_t client, plist.plist_t *plist)
  20. mobilebackup_error_t mobilebackup_send(mobilebackup_client_t client, plist.plist_t plist)
  21. mobilebackup_error_t mobilebackup_request_backup(mobilebackup_client_t client, plist.plist_t backup_manifest, char *base_path, char *proto_version)
  22. mobilebackup_error_t mobilebackup_send_backup_file_received(mobilebackup_client_t client)
  23. mobilebackup_error_t mobilebackup_request_restore(mobilebackup_client_t client, plist.plist_t backup_manifest, mobilebackup_flags_t flags, char *proto_version)
  24. mobilebackup_error_t mobilebackup_receive_restore_file_received(mobilebackup_client_t client, plist.plist_t *result)
  25. mobilebackup_error_t mobilebackup_receive_restore_application_received(mobilebackup_client_t client, plist.plist_t *result)
  26. mobilebackup_error_t mobilebackup_send_restore_complete(mobilebackup_client_t client)
  27. mobilebackup_error_t mobilebackup_send_error(mobilebackup_client_t client, char *reason)
  28. cdef class MobileBackupError(BaseError):
  29. def __init__(self, *args, **kwargs):
  30. self._lookup_table = {
  31. MOBILEBACKUP_E_SUCCESS: "Success",
  32. MOBILEBACKUP_E_INVALID_ARG: "Invalid argument",
  33. MOBILEBACKUP_E_PLIST_ERROR: "Property list error",
  34. MOBILEBACKUP_E_MUX_ERROR: "MUX error",
  35. MOBILEBACKUP_E_BAD_VERSION: "Bad version",
  36. MOBILEBACKUP_E_REPLY_NOT_OK: "Reply not OK",
  37. MOBILEBACKUP_E_UNKNOWN_ERROR: "Unknown error"
  38. }
  39. BaseError.__init__(self, *args, **kwargs)
  40. cdef class MobileBackupClient(PropertyListService):
  41. __service_name__ = "com.apple.mobilebackup"
  42. cdef mobilebackup_client_t _c_client
  43. def __cinit__(self, iDevice device not None, LockdownServiceDescriptor descriptor, *args, **kwargs):
  44. self.handle_error(mobilebackup_client_new(device._c_dev, descriptor._c_service_descriptor, &self._c_client))
  45. def __dealloc__(self):
  46. cdef mobilebackup_error_t err
  47. if self._c_client is not NULL:
  48. err = mobilebackup_client_free(self._c_client)
  49. self.handle_error(err)
  50. cdef inline BaseError _error(self, int16_t ret):
  51. return MobileBackupError(ret)
  52. cdef inline int16_t _send(self, plist.plist_t node):
  53. return mobilebackup_send(self._c_client, node)
  54. cdef inline int16_t _receive(self, plist.plist_t* node):
  55. return mobilebackup_receive(self._c_client, node)
  56. cdef request_backup(self, plist.Node backup_manifest, bytes base_path, bytes proto_version):
  57. self.handle_error(mobilebackup_request_backup(self._c_client, backup_manifest._c_node, base_path, proto_version))
  58. cdef send_backup_file_received(self):
  59. self.handle_error(mobilebackup_send_backup_file_received(self._c_client))
  60. cdef request_restore(self, plist.Node backup_manifest, int flags, proto_version):
  61. self.handle_error(mobilebackup_request_restore(self._c_client, backup_manifest._c_node, <mobilebackup_flags_t>flags, proto_version))
  62. cpdef plist.Node receive_restore_file_received(self):
  63. cdef:
  64. plist.plist_t c_node = NULL
  65. mobilebackup_error_t err
  66. err = mobilebackup_receive_restore_file_received(self._c_client, &c_node)
  67. try:
  68. self.handle_error(err)
  69. return plist.plist_t_to_node(c_node)
  70. except BaseError, e:
  71. if c_node != NULL:
  72. plist.plist_free(c_node)
  73. raise
  74. cpdef plist.Node receive_restore_application_received(self):
  75. cdef:
  76. plist.plist_t c_node = NULL
  77. mobilebackup_error_t err
  78. err = mobilebackup_receive_restore_application_received(self._c_client, &c_node)
  79. try:
  80. self.handle_error(err)
  81. return plist.plist_t_to_node(c_node)
  82. except BaseError, e:
  83. if c_node != NULL:
  84. plist.plist_free(c_node)
  85. raise
  86. cdef send_restore_complete(self):
  87. self.handle_error(mobilebackup_send_restore_complete(self._c_client))
  88. cdef send_error(self, bytes reason):
  89. self.handle_error(mobilebackup_send_error(self._c_client, reason))