restore.pxi 4.9 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135
  1. cdef extern from "libimobiledevice/restore.h":
  2. cdef struct restored_client_private:
  3. pass
  4. ctypedef restored_client_private *restored_client_t
  5. ctypedef enum restored_error_t:
  6. RESTORE_E_SUCCESS = 0
  7. RESTORE_E_INVALID_ARG = -1
  8. RESTORE_E_INVALID_CONF = -2
  9. RESTORE_E_PLIST_ERROR = -3
  10. RESTORE_E_DICT_ERROR = -4
  11. RESTORE_E_NOT_ENOUGH_DATA = -5
  12. RESTORE_E_MUX_ERROR = -6
  13. RESTORE_E_START_RESTORE_FAILED = -7
  14. RESTORE_E_UNKNOWN_ERROR = -256
  15. restored_error_t restored_client_new(idevice_t device, restored_client_t *client, char *label)
  16. restored_error_t restored_client_free(restored_client_t client)
  17. restored_error_t restored_query_type(restored_client_t client, char **tp, uint64_t *version)
  18. restored_error_t restored_query_value(restored_client_t client, char *key, plist.plist_t *value)
  19. restored_error_t restored_get_value(restored_client_t client, char *key, plist.plist_t *value)
  20. restored_error_t restored_send(restored_client_t client, plist.plist_t plist)
  21. restored_error_t restored_receive(restored_client_t client, plist.plist_t *plist)
  22. restored_error_t restored_goodbye(restored_client_t client)
  23. restored_error_t restored_start_restore(restored_client_t client, plist.plist_t options, uint64_t version)
  24. restored_error_t restored_reboot(restored_client_t client)
  25. void restored_client_set_label(restored_client_t client, char *label)
  26. cdef class RestoreError(BaseError):
  27. def __init__(self, *args, **kwargs):
  28. self._lookup_table = {
  29. RESTORE_E_SUCCESS: "Success",
  30. RESTORE_E_INVALID_ARG: "Invalid argument",
  31. RESTORE_E_INVALID_CONF: "Invalid configuration",
  32. RESTORE_E_PLIST_ERROR: "Property list error",
  33. RESTORE_E_DICT_ERROR: "Dict error",
  34. RESTORE_E_NOT_ENOUGH_DATA: "Not enough data",
  35. RESTORE_E_MUX_ERROR: "MUX Error",
  36. RESTORE_E_START_RESTORE_FAILED: "Starting restore failed",
  37. RESTORE_E_UNKNOWN_ERROR: "Unknown error"
  38. }
  39. BaseError.__init__(self, *args, **kwargs)
  40. cdef class RestoreClient(PropertyListService):
  41. cdef restored_client_t _c_client
  42. def __cinit__(self, iDevice device not None, bytes label=b'', *args, **kwargs):
  43. cdef:
  44. restored_error_t err
  45. char* c_label = NULL
  46. if label:
  47. c_label = label
  48. err = restored_client_new(device._c_dev, &self._c_client, c_label)
  49. self.handle_error(err)
  50. self.device = device
  51. def __dealloc__(self):
  52. cdef restored_error_t err
  53. if self._c_client is not NULL:
  54. err = restored_client_free(self._c_client)
  55. self.handle_error(err)
  56. cdef inline BaseError _error(self, int16_t ret):
  57. return RestoreError(ret)
  58. cdef inline int16_t _send(self, plist.plist_t node):
  59. return restored_send(self._c_client, node)
  60. cdef inline int16_t _receive(self, plist.plist_t* node):
  61. return restored_receive(self._c_client, node)
  62. cpdef tuple query_type(self):
  63. cdef:
  64. restored_error_t err
  65. char* c_type = NULL
  66. uint64_t c_version = 0
  67. tuple result
  68. err = restored_query_type(self._c_client, &c_type, &c_version)
  69. try:
  70. self.handle_error(err)
  71. result = (c_type, c_version)
  72. return result
  73. except BaseError, e:
  74. raise
  75. finally:
  76. if c_type != NULL:
  77. free(c_type)
  78. cpdef plist.Node query_value(self, bytes key=None):
  79. cdef:
  80. restored_error_t err
  81. plist.plist_t c_node = NULL
  82. char* c_key = NULL
  83. if key is not None:
  84. c_key = key
  85. err = restored_query_value(self._c_client, c_key, &c_node)
  86. try:
  87. self.handle_error(err)
  88. return plist.plist_t_to_node(c_node)
  89. except BaseError, e:
  90. if c_node != NULL:
  91. plist.plist_free(c_node)
  92. raise
  93. cpdef plist.Node get_value(self, bytes key=None):
  94. cdef:
  95. restored_error_t err
  96. plist.plist_t c_node = NULL
  97. char* c_key = NULL
  98. if key is not None:
  99. c_key = key
  100. err = restored_get_value(self._c_client, c_key, &c_node)
  101. try:
  102. self.handle_error(err)
  103. return plist.plist_t_to_node(c_node)
  104. except BaseError, e:
  105. if c_node != NULL:
  106. plist.plist_free(c_node)
  107. raise
  108. cpdef goodbye(self):
  109. self.handle_error(restored_goodbye(self._c_client))
  110. cpdef start_restore(self, plist.Node options, uint64_t version):
  111. self.handle_error(restored_start_restore(self._c_client, options._c_node, version))
  112. cpdef reboot(self):
  113. self.handle_error(restored_reboot(self._c_client))
  114. cpdef set_label(self, bytes label):
  115. restored_client_set_label(self._c_client, label)