heartbeat.pxi 2.4 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556
  1. cdef extern from "libimobiledevice/heartbeat.h":
  2. cdef struct heartbeat_client_private:
  3. pass
  4. ctypedef heartbeat_client_private *heartbeat_client_t
  5. ctypedef enum heartbeat_error_t:
  6. HEARTBEAT_E_SUCCESS = 0
  7. HEARTBEAT_E_INVALID_ARG = -1
  8. HEARTBEAT_E_PLIST_ERROR = -2
  9. HEARTBEAT_E_MUX_ERROR = -3
  10. HEARTBEAT_E_SSL_ERROR = -4
  11. HEARTBEAT_E_UNKNOWN_ERROR = -256
  12. heartbeat_error_t heartbeat_client_new(idevice_t device, lockdownd_service_descriptor_t descriptor, heartbeat_client_t * client)
  13. heartbeat_error_t heartbeat_client_free(heartbeat_client_t client)
  14. heartbeat_error_t heartbeat_send(heartbeat_client_t client, plist.plist_t plist)
  15. heartbeat_error_t heartbeat_receive(heartbeat_client_t client, plist.plist_t * plist)
  16. heartbeat_error_t heartbeat_receive_with_timeout(heartbeat_client_t client, plist.plist_t * plist, uint32_t timeout_ms)
  17. cdef class HeartbeatError(BaseError):
  18. def __init__(self, *args, **kwargs):
  19. self._lookup_table = {
  20. HEARTBEAT_E_SUCCESS: "Success",
  21. HEARTBEAT_E_INVALID_ARG: "Invalid argument",
  22. HEARTBEAT_E_PLIST_ERROR: "Property list error",
  23. HEARTBEAT_E_MUX_ERROR: "MUX error",
  24. HEARTBEAT_E_SSL_ERROR: "SSL Error",
  25. HEARTBEAT_E_UNKNOWN_ERROR: "Unknown error"
  26. }
  27. BaseError.__init__(self, *args, **kwargs)
  28. cdef class HeartbeatClient(PropertyListService):
  29. __service_name__ = "com.apple.heartbeat"
  30. cdef heartbeat_client_t _c_client
  31. def __cinit__(self, iDevice device not None, LockdownServiceDescriptor descriptor, *args, **kwargs):
  32. self.handle_error(heartbeat_client_new(device._c_dev, descriptor._c_service_descriptor, &self._c_client))
  33. def __dealloc__(self):
  34. cdef heartbeat_error_t err
  35. if self._c_client is not NULL:
  36. err = heartbeat_client_free(self._c_client)
  37. self.handle_error(err)
  38. cdef inline int16_t _send(self, plist.plist_t node):
  39. return heartbeat_send(self._c_client, node)
  40. cdef inline int16_t _receive(self, plist.plist_t* node):
  41. return heartbeat_receive(self._c_client, node)
  42. cdef inline int16_t _receive_with_timeout(self, plist.plist_t* node, int timeout_ms):
  43. return heartbeat_receive_with_timeout(self._c_client, node, timeout_ms)
  44. cdef inline BaseError _error(self, int16_t ret):
  45. return HeartbeatError(ret)