pymobiledevice3 4.14.6__py3-none-any.whl → 7.0.6__py3-none-any.whl

pymobiledevice3/irecv.py

@@ -1,4 +1,5 @@
 import binascii
+import contextlib
 import logging
 import math
 import struct
@@ -10,7 +11,7 @@ from tqdm import trange
 from usb.core import Device, USBError, find
 from usb.util import get_string
 
-from pymobiledevice3.exceptions import IRecvNoDeviceConnectedError, PyMobileDevice3Exception
+from pymobiledevice3.exceptions import IRecvError, IRecvNoDeviceConnectedError, PyMobileDevice3Exception
 from pymobiledevice3.irecv_devices import IRECV_DEVICES, IRecvDevice
 
 USB_TIMEOUT = 10000
@@ -28,10 +29,7 @@ class Mode(Enum):
 
     @classmethod
     def has_value(cls, value):
-        for m in cls:
-            if value == m.value:
-                return True
-        return False
+        return any(value == m.value for m in cls)
 
     @classmethod
     def get_mode_from_value(cls, value):
@@ -61,7 +59,7 @@ logger = logging.getLogger(__name__)
 
 
 class IRecv:
-    def __init__(self, ecid=None, timeout=0xffffffff, is_recovery=None):
+    def __init__(self, ecid=None, timeout=0xFFFFFFFF, is_recovery=None):
         self.mode: Optional[Mode] = None
         self._device_info = {}
         self._device: Optional[Device] = None
@@ -69,27 +67,27 @@ class IRecv:
 
     @property
     def ecid(self):
-        return int(self._device_info['ECID'], 16)
+        return int(self._device_info["ECID"], 16)
 
     @property
     def ibfl(self):
-        return int(self._device_info['IBFL'], 16)
+        return int(self._device_info["IBFL"], 16)
 
     @property
     def chip_id(self):
-        return int(self._device_info['CPID'], 16)
+        return int(self._device_info["CPID"], 16)
 
     @property
     def board_id(self):
-        return int(self._device_info['BDID'], 16)
+        return int(self._device_info["BDID"], 16)
 
     @property
     def serial_number(self) -> int:
-        return int(self._device_info['SRNM'], 16)
+        return int(self._device_info["SRNM"], 16)
 
     @property
     def iboot_version(self) -> str:
-        return self._device_info['SRTG']
+        return self._device_info["SRTG"]
 
     @property
     def is_image4_supported(self):
@@ -100,7 +98,7 @@ class IRecv:
         for device in IRECV_DEVICES:
             if device.board_id == self.board_id and device.chip_id == self.chip_id:
                 return device
-        raise KeyError(f'failed to find device of: board_id: {self.board_id} chip_id: {self.chip_id}')
+        raise KeyError(f"failed to find device of: board_id: {self.board_id} chip_id: {self.chip_id}")
 
     @property
     def product_type(self):
@@ -116,15 +114,15 @@ class IRecv:
 
     @property
     def status(self):
-        return self.ctrl_transfer(0xa1, 3, data_or_wLength=b'\x00' * 6)[4]
+        return self.ctrl_transfer(0xA1, 3, data_or_wLength=b"\x00" * 6)[4]
 
     def set_interface_altsetting(self, interface=None, alternate_setting=None):
-        logger.debug(f'set_interface_altsetting: {interface} {alternate_setting}')
+        logger.debug(f"set_interface_altsetting: {interface} {alternate_setting}")
         if interface == 1:
             self._device.set_interface_altsetting(interface=interface, alternate_setting=alternate_setting)
 
     def set_configuration(self, configuration=None):
-        logger.debug(f'set_configuration: {configuration}')
+        logger.debug(f"set_configuration: {configuration}")
         try:
             if self._device.get_active_configuration().bConfigurationValue == configuration:
                 return
@@ -143,44 +141,44 @@ class IRecv:
         if self.mode.is_recovery:
             self.ctrl_transfer(0x41, 0)
         else:
-            response = self.ctrl_transfer(0xa1, 5, data_or_wLength=1)
+            response = self.ctrl_transfer(0xA1, 5, data_or_wLength=1)
             state = response[0]
-            logger.debug(f'irecv state: {state}')
+            logger.debug(f"irecv state: {state}")
             if state == 2:
                 # DFU IDLE
                 pass
             elif state == 10:
                 self.ctrl_transfer(0x21, 4)
-                raise PyMobileDevice3Exception('DFU ERROR, issuing CLRSTATUS')
+                raise PyMobileDevice3Exception("DFU ERROR, issuing CLRSTATUS")
             else:
                 self.ctrl_transfer(0x21, 6)
-                raise PyMobileDevice3Exception(f'Unexpected state {state}, issuing ABORT')
+                raise PyMobileDevice3Exception(f"Unexpected state {state}, issuing ABORT")
 
         crc = -1
 
         num_packets = math.ceil(len(buf) / packet_size)
 
-        for offset in trange(0, len(buf), packet_size, dynamic_ncols=True):
+        for _offset in trange(0, len(buf), packet_size, dynamic_ncols=True):
             # Use bulk transfer for recovery mode and control transfer for DFU and WTF mode
-            chunk = buf[offset:offset + packet_size]
-            packet_index = offset // packet_size
+            chunk = buf[_offset : _offset + packet_size]
+            packet_index = _offset // packet_size
 
             if self.mode.is_recovery:
                 n = self._device.write(0x04, chunk, timeout=USB_TIMEOUT)
                 if n != len(chunk):
-                    raise IOError('failed to upload data')
+                    raise OSError("failed to upload data")
             else:
-                if offset + packet_size >= len(buf):
+                if _offset + packet_size >= len(buf):
                     # last packet
 
                     # calculate crc of all sent data
                     crc = binascii.crc32(buf, crc)
 
                     # add crc of dfu_xbuf (salted value)
-                    dfu_xbuf = bytearray([0xff, 0xff, 0xff, 0xff, 0xac, 0x05, 0x00, 0x01, 0x55, 0x46, 0x44, 0x10])
+                    dfu_xbuf = bytearray([0xFF, 0xFF, 0xFF, 0xFF, 0xAC, 0x05, 0x00, 0x01, 0x55, 0x46, 0x44, 0x10])
                     crc = binascii.crc32(dfu_xbuf, crc)
 
-                    crc_chunk = dfu_xbuf + struct.pack('<I', crc)
+                    crc_chunk = dfu_xbuf + struct.pack("<I", crc)
 
                     if len(chunk) + 16 > packet_size:
                         # crc exceeds the max allowed packet size
@@ -192,13 +190,13 @@ class IRecv:
                     self.ctrl_transfer(0x21, 1, wValue=packet_index, wIndex=0, data_or_wLength=chunk)
 
         if not self.mode.is_recovery:
-            logger.debug('waiting for status == 5')
+            logger.debug("waiting for status == 5")
             while self.status != 5:
                 time.sleep(1)
 
             self.ctrl_transfer(0x21, 1, wValue=num_packets, wIndex=0)
 
-            for offset in range(2):
+            for _offset in range(2):
                 # i know it's not used but idevicerestore does that also
                 _ = self.status
 
@@ -206,8 +204,8 @@ class IRecv:
 
     def reset(self):
         try:
-            logger.debug('resetting usb device')
-            logger.info('If the restore hangs here, disconnect & reconnect the device')
+            logger.debug("resetting usb device")
+            logger.info("If the restore hangs here, disconnect & reconnect the device")
             self._device.reset()
         except USBError:
             pass
@@ -215,36 +213,34 @@ class IRecv:
         self._reinit(ecid=self.ecid)
 
     def send_command(self, cmd: str, timeout=USB_TIMEOUT, b_request=0):
-        self._device.ctrl_transfer(0x40, b_request, 0, 0, cmd.encode() + b'\0', timeout=timeout)
+        self._device.ctrl_transfer(0x40, b_request, 0, 0, cmd.encode() + b"\0", timeout=timeout)
 
     def getenv(self, name):
         try:
-            self.send_command(f'getenv {name}')
+            self.send_command(f"getenv {name}")
         except USBError:
             return None
-        return bytearray(self._device.ctrl_transfer(0xc0, 0, 0, 0, 255))
+        return bytearray(self._device.ctrl_transfer(0xC0, 0, 0, 0, 255))
 
     def set_autoboot(self, enable: bool):
-        self.send_command(f'setenv auto-boot {str(enable).lower()}')
-        self.send_command('saveenv')
+        self.send_command(f"setenv auto-boot {str(enable).lower()}")
+        self.send_command("saveenv")
 
     def reboot(self):
-        try:
-            self.send_command('reboot')
-        except USBError:
-            pass
+        with contextlib.suppress(USBError):
+            self.send_command("reboot")
 
-    def _reinit(self, ecid=None, timeout=0xffffffff, is_recovery=None):
+    def _reinit(self, ecid=None, timeout=0xFFFFFFFF, is_recovery=None):
         self._device = None
         self._device_info = {}
         self.mode = None
         self._find(ecid=ecid, timeout=timeout, is_recovery=is_recovery)
         if self._device is None:
-            raise IRecvNoDeviceConnectedError('Failed to find a connected iDevice')
+            raise IRecvNoDeviceConnectedError("Failed to find a connected iDevice")
         self._populate_device_info()
 
-        self.ap_nonce = self._copy_nonce_with_tag('NONC')
-        self.sep_nonce = self._copy_nonce_with_tag('SNON')
+        self.ap_nonce = self._copy_nonce_with_tag("NONC")
+        self.sep_nonce = self._copy_nonce_with_tag("SNON")
 
         self.set_configuration(1)
 
@@ -256,9 +252,12 @@ class IRecv:
             self.set_interface_altsetting(0, 0)
 
     def _copy_nonce_with_tag(self, tag):
-        return binascii.unhexlify(get_string(self._device, 1).split(f'{tag}:')[1].split(' ')[0])
+        if tag in get_string(self._device, 1):
+            return binascii.unhexlify(get_string(self._device, 1).split(f"{tag}:")[1].split(" ")[0])
+        else:
+            return None
 
-    def _find(self, ecid=None, timeout=0xffffffff, is_recovery=None):
+    def _find(self, ecid=None, timeout=0xFFFFFFFF, is_recovery=None):
         start = time.time()
         end = start + timeout
         while (self._device is None) and (time.time() < end):
@@ -266,7 +265,7 @@ class IRecv:
                 try:
                     if device.manufacturer is None:
                         continue
-                    if not device.manufacturer.startswith('Apple'):
+                    if not device.manufacturer.startswith("Apple"):
                         continue
 
                     mode = Mode.get_mode_from_value(device.idProduct)
@@ -278,13 +277,13 @@ class IRecv:
                         continue
 
                     if self._device is not None:
-                        raise Exception('More then one connected device was found connected in recovery mode')
+                        raise IRecvError("More then one connected device was found connected in recovery mode")
                     self._device = device
                     self.mode = mode
                     self._populate_device_info()
 
                     if ecid is not None:
-                        found_ecid = int(self._device_info['ECID'], 16)
+                        found_ecid = int(self._device_info["ECID"], 16)
                         if found_ecid != ecid:
                             # wrong device - move on
                             self._device = None
@@ -293,9 +292,9 @@ class IRecv:
                     continue
 
     def _populate_device_info(self):
-        for component in self._device.serial_number.split(' '):
-            k, v = component.split(':')
-            if k in ('SRNM', 'SRTG') and '[' in v:
+        for component in self._device.serial_number.split(" "):
+            k, v = component.split(":")
+            if k in ("SRNM", "SRTG") and "[" in v:
                 # trim the `[]`
                 v = v[1:-1]
             self._device_info[k] = v
@@ -314,5 +313,5 @@ def main():
     print(IRecv())
 
 
-if __name__ == '__main__':
+if __name__ == "__main__":
     main()