Python MtpObjectの例

コード例 #1

ファイル: mtp_object_tests.py プロジェクト: BinyaminSharet/Mtp

 def testSetStorage(self):
     '''
     .. todo:: should this be in storage tests? (TBD)
     '''
     uut = MtpObject(self.default_data, self.get_default_info())
     storage = self.get_default_storage()
     uut.set_storage(storage)
     self.assertEqual(uut.storage, storage)

コード例 #2

ファイル: mtp_object_tests.py プロジェクト: dark-lbp/Mtp

 def testSetStorage(self):
     '''
     .. todo:: should this be in storage tests? (TBD)
     '''
     uut = MtpObject(self.default_data, self.get_default_info())
     storage = self.get_default_storage()
     uut.set_storage(storage)
     self.assertEqual(uut.storage, storage)

コード例 #3

ファイル: mtp_object_tests.py プロジェクト: BinyaminSharet/Mtp

 def testAddObjectSetsStorageInChild(self):
     parent = MtpObject(self.default_data, self.get_default_info())
     uut = MtpObject(self.default_data, self.get_default_info())
     storage = self.get_default_storage()
     parent.set_storage(storage)
     parent.add_object(uut)
     self.assertEqual(uut.storage, storage)
     self.assertEqual(parent.storage, storage)

コード例 #4

ファイル: mtp_object_tests.py プロジェクト: BinyaminSharet/Mtp

 def testGetObjectSelfWithChilds(self):
     others = [
         MtpObject(self.default_data, self.get_default_info()),
         MtpObject(self.default_data, self.get_default_info()),
         MtpObject(self.default_data, self.get_default_info())
     ]
     uut = MtpObject(self.default_data, self.get_default_info())
     for obj in others:
         uut.add_object(obj)
     obj = uut.get_object(uut.get_uid())
     self.assertEqual(uut, obj)

コード例 #5

ファイル: mtp_object_tests.py プロジェクト: BinyaminSharet/Mtp

 def testSetStorageIsRecursive(self):
     '''
     .. todo:: should this be in storage tests? (TBD)
     '''
     parent = MtpObject(self.default_data, self.get_default_info())
     storage = self.get_default_storage()
     uut = MtpObject(self.default_data, self.get_default_info())
     parent.add_object(uut)
     parent.set_storage(storage)
     self.assertEqual(uut.storage, storage)
     self.assertEqual(parent.storage, storage)

コード例 #6

ファイル: mtp_object_tests.py プロジェクト: BinyaminSharet/Mtp

 def testGetObjectChildren(self):
     others = [
         MtpObject(self.default_data, self.get_default_info()),
         MtpObject(self.default_data, self.get_default_info()),
         MtpObject(self.default_data, self.get_default_info())
     ]
     uut = MtpObject(self.default_data, self.get_default_info())
     for obj in others:
         uut.add_object(obj)
     obj = uut.get_object(others[0].get_uid())
     self.assertEqual(obj, others[0])

コード例 #7

ファイル: mtp_object_tests.py プロジェクト: BinyaminSharet/Mtp

 def testGetObjectMultipleLevels(self):
     others = [
         MtpObject(self.default_data, self.get_default_info()),
         MtpObject(self.default_data, self.get_default_info()),
         MtpObject(self.default_data, self.get_default_info())
     ]
     uut = MtpObject(self.default_data, self.get_default_info())
     cur = uut
     for obj in others:
         cur.add_object(obj)
         cur = obj
     obj = uut.get_object(others[-1].get_uid())
     self.assertEqual(obj, others[-1])

コード例 #8

ファイル: mtp_object_tests.py プロジェクト: BinyaminSharet/Mtp

 def testGetObjectsWithMultipleChildes(self):
     others = [
         MtpObject(self.default_data, self.get_default_info()),
         MtpObject(self.default_data, self.get_default_info()),
         MtpObject(self.default_data, self.get_default_info())
     ]
     uut = MtpObject(self.default_data, self.get_default_info())
     for obj in others:
         uut.add_object(obj)
     objs = uut.get_objects()
     self.assertEqual(len(objs), len(set(objs)))
     self.assertEqual(len(objs), len(others))
     self.assertEqual(set(objs), set(others))

コード例 #9

ファイル: mtp_object_tests.py プロジェクト: dark-lbp/Mtp

 def testGetObjectSelfWithChilds(self):
     others = [
         MtpObject(self.default_data, self.get_default_info()),
         MtpObject(self.default_data, self.get_default_info()),
         MtpObject(self.default_data, self.get_default_info())
     ]
     uut = MtpObject(self.default_data, self.get_default_info())
     for obj in others:
         uut.add_object(obj)
     obj = uut.get_object(uut.get_uid())
     self.assertEqual(uut, obj)

コード例 #10

ファイル: mtp_object_tests.py プロジェクト: BinyaminSharet/Mtp

 def testGetObjectsMultipleLevels(self):
     others = [
         MtpObject(self.default_data, self.get_default_info()),
         MtpObject(self.default_data, self.get_default_info()),
         MtpObject(self.default_data, self.get_default_info())
     ]
     uut = MtpObject(self.default_data, self.get_default_info())
     cur = uut
     for obj in others:
         cur.add_object(obj)
         cur = obj
     objs = uut.get_objects()
     self.assertEqual(len(objs), len(set(objs)))
     self.assertEqual(len(objs), len(others))
     self.assertEqual(set(objs), set(others))

コード例 #11

 def setUp(self):
     super(MtpDeviceTests, self).setUp()
     self.object = MtpObject.from_fs_recursive('.')
     self.storage_info = MtpStorageInfo(
         st_type=0,
         fs_type=0,
         access=0,
         max_cap=150000,
         free_bytes=0,
         free_objs=0,
         desc='MyStorage',
         vol_id='Python MTP Device Stack',
     )
     self.storage = MtpStorage(self.storage_info)
     self.storage.add_object(self.object)
     self.dev_info = MtpDeviceInfo(
         std_version=0x0102,
         mtp_vendor_ext_id=0x03040506,
         mtp_version=0x0708,
         mtp_extensions='Some Extension',
         functional_mode=0x0000,
         capture_formats=[],
         playback_formats=[],
         manufacturer='BinyaminSharet',
         model='Role',
         device_version='1.2.3',
         serial_number='0123456789abcdef',
     )
     self.dev = MtpDevice(self.dev_info)
     self.dev.add_storage(self.storage)
     self.tid = 1

コード例 #12

ファイル: mtp_api_tests.py プロジェクト: BinyaminSharet/Mtp

 def setUp(self):
     super(MtpApiTest, self).setUp()
     self.object = MtpObject.from_fs_recursive('.')
     self.storage_info = MtpStorageInfo(
         st_type=0,
         fs_type=0,
         access=0,
         max_cap=150000,
         free_bytes=0,
         free_objs=0,
         desc='MyStorage',
         vol_id='Python MTP Device Stack',
     )
     self.storage = MtpStorage(self.storage_info)
     self.storage.add_object(self.object)
     self.dev_info = MtpDeviceInfo(
         std_version=0x0102,
         mtp_vendor_ext_id=0x03040506,
         mtp_version=0x0708,
         mtp_extensions='Some Extension',
         functional_mode=0x0000,
         capture_formats=[],
         playback_formats=[],
         manufacturer='BinyaminSharet',
         model='Role',
         device_version='1.2.3',
         serial_number='0123456789abcdef',
     )
     properties = [
         MtpDeviceProperty(MtpDevicePropertyCode.BatteryLevel, 0, UInt8(50), UInt8(0))
     ]
     self.dev = MtpDevice(self.dev_info, properties)
     self.dev.add_storage(self.storage)
     self.api = MtpApi(self.dev)

コード例 #13

ファイル: mtp_object_tests.py プロジェクト: dark-lbp/Mtp

 def testGetObjectChildren(self):
     others = [
         MtpObject(self.default_data, self.get_default_info()),
         MtpObject(self.default_data, self.get_default_info()),
         MtpObject(self.default_data, self.get_default_info())
     ]
     uut = MtpObject(self.default_data, self.get_default_info())
     for obj in others:
         uut.add_object(obj)
     obj = uut.get_object(others[0].get_uid())
     self.assertEqual(obj, others[0])

コード例 #14

ファイル: mtp_object_tests.py プロジェクト: dark-lbp/Mtp

 def testAddObjectSetsStorageInChild(self):
     parent = MtpObject(self.default_data, self.get_default_info())
     uut = MtpObject(self.default_data, self.get_default_info())
     storage = self.get_default_storage()
     parent.set_storage(storage)
     parent.add_object(uut)
     self.assertEqual(uut.storage, storage)
     self.assertEqual(parent.storage, storage)

コード例 #15

ファイル: mtp_object_tests.py プロジェクト: dark-lbp/Mtp

 def testGetObjectsWithMultipleChildes(self):
     others = [
         MtpObject(self.default_data, self.get_default_info()),
         MtpObject(self.default_data, self.get_default_info()),
         MtpObject(self.default_data, self.get_default_info())
     ]
     uut = MtpObject(self.default_data, self.get_default_info())
     for obj in others:
         uut.add_object(obj)
     objs = uut.get_objects()
     self.assertEqual(len(objs), len(set(objs)))
     self.assertEqual(len(objs), len(others))
     self.assertEqual(set(objs), set(others))

コード例 #16

ファイル: mtp_object_tests.py プロジェクト: dark-lbp/Mtp

 def testSetStorageIsRecursive(self):
     '''
     .. todo:: should this be in storage tests? (TBD)
     '''
     parent = MtpObject(self.default_data, self.get_default_info())
     storage = self.get_default_storage()
     uut = MtpObject(self.default_data, self.get_default_info())
     parent.add_object(uut)
     parent.set_storage(storage)
     self.assertEqual(uut.storage, storage)
     self.assertEqual(parent.storage, storage)

コード例 #17

ファイル: mtp_object_tests.py プロジェクト: dark-lbp/Mtp

 def testGetObjectMultipleLevels(self):
     others = [
         MtpObject(self.default_data, self.get_default_info()),
         MtpObject(self.default_data, self.get_default_info()),
         MtpObject(self.default_data, self.get_default_info())
     ]
     uut = MtpObject(self.default_data, self.get_default_info())
     cur = uut
     for obj in others:
         cur.add_object(obj)
         cur = obj
     obj = uut.get_object(others[-1].get_uid())
     self.assertEqual(obj, others[-1])

コード例 #18

ファイル: mtp_object_tests.py プロジェクト: dark-lbp/Mtp

 def testGetObjectsMultipleLevels(self):
     others = [
         MtpObject(self.default_data, self.get_default_info()),
         MtpObject(self.default_data, self.get_default_info()),
         MtpObject(self.default_data, self.get_default_info())
     ]
     uut = MtpObject(self.default_data, self.get_default_info())
     cur = uut
     for obj in others:
         cur.add_object(obj)
         cur = obj
     objs = uut.get_objects()
     self.assertEqual(len(objs), len(set(objs)))
     self.assertEqual(len(objs), len(others))
     self.assertEqual(set(objs), set(others))

コード例 #19

ファイル: mtp_object_tests.py プロジェクト: dark-lbp/Mtp

 def testFormatMatchsNone(self):
     uut = MtpObject(self.default_data, self.get_default_info())
     self.assertTrue(uut.format_matches(None))

コード例 #20

ファイル: mtp_object_tests.py プロジェクト: BinyaminSharet/Mtp

 def testGetObjectDoesntExist(self):
     uut = MtpObject(self.default_data, self.get_default_info())
     obj = uut.get_object(uut.get_uid() + 1)
     self.assertIsNone(obj)

コード例 #21

ファイル: mtp_object_tests.py プロジェクト: BinyaminSharet/Mtp

 def testGetInfoReturnsPackedInfo(self):
     info = self.get_default_info()
     uut = MtpObject(self.default_data, info)
     self.assertEqual(uut.get_info(), info.pack())

コード例 #22

ファイル: mtp_object_tests.py プロジェクト: dark-lbp/Mtp

 def testGetObjectsEmpty(self):
     uut = MtpObject(self.default_data, self.get_default_info())
     self.assertEqual(uut.get_objects(), [])

コード例 #23

ファイル: mtp_object_tests.py プロジェクト: dark-lbp/Mtp

 def testObjectInfo(self):
     info = self.get_default_info()
     uut = MtpObject(self.default_data, info)
     self.assertEqual(uut.info, info)

コード例 #24

ファイル: mtp_object_tests.py プロジェクト: dark-lbp/Mtp

 def testGetPropertyExceptionIfNotExist(self):
     uut = MtpObject(self.default_data, self.get_default_info())
     with self.assertRaises(MtpProtocolException) as cm:
         uut.get_property(MtpObjectPropertyCode.StorageID - 1)
     self.assertEqual(cm.exception.response,
                      ResponseCodes.OBJECT_PROP_NOT_SUPPORTED)

コード例 #25

ファイル: mtp_object_tests.py プロジェクト: dark-lbp/Mtp

 def testGetObjectDoesntExist(self):
     uut = MtpObject(self.default_data, self.get_default_info())
     obj = uut.get_object(uut.get_uid() + 1)
     self.assertIsNone(obj)

コード例 #26

ファイル: mtp_object_tests.py プロジェクト: BinyaminSharet/Mtp

 def testFormatMatchsWildcard(self):
     uut = MtpObject(self.default_data, self.get_default_info())
     self.assertTrue(uut.format_matches(0xffffffff))

コード例 #27

ファイル: mtp_object_tests.py プロジェクト: BinyaminSharet/Mtp

 def testGetPropertyDefaults(self):
     uut = MtpObject(self.default_data, self.get_default_info())
     prop = uut.get_property(MtpObjectPropertyCode.ObjectFormat)
     self.assertIsNotNone(prop)
     self.assertEqual(prop.value.value, self.default_format)

コード例 #28

ファイル: mtp.py プロジェクト: Manouchehri/umap2

    def __init__(self, app, phy):
        if not mtpdeviceloaded:
            raise Exception('You cannot use USBMtp until you install pymtpdevice')
        # TODO: un-hardcode string index (last arg before 'verbose')
        super(USBMtpInterface, self).__init__(
            app=app,
            phy=phy,
            interface_number=0,
            interface_alternate=0,
            interface_class=USBClass.VendorSpecific,
            interface_subclass=0xff,
            interface_protocol=0,
            interface_string_index=0,
            endpoints=[
                USBEndpoint(
                    app=app,
                    phy=phy,
                    number=1,
                    direction=USBEndpoint.direction_out,
                    transfer_type=USBEndpoint.transfer_type_bulk,
                    sync_type=USBEndpoint.sync_type_none,
                    usage_type=USBEndpoint.usage_type_data,
                    max_packet_size=64,
                    interval=0,
                    handler=self.handle_ep1_data_available
                ),
                USBEndpoint(
                    app=app,
                    phy=phy,
                    number=2,
                    direction=USBEndpoint.direction_in,
                    transfer_type=USBEndpoint.transfer_type_bulk,
                    sync_type=USBEndpoint.sync_type_none,
                    usage_type=USBEndpoint.usage_type_data,
                    max_packet_size=64,
                    interval=0,
                    handler=None
                ),
                USBEndpoint(
                    app=app,
                    phy=phy,
                    number=3,
                    direction=USBEndpoint.direction_in,
                    transfer_type=USBEndpoint.transfer_type_interrupt,
                    sync_type=USBEndpoint.sync_type_none,
                    usage_type=USBEndpoint.usage_type_data,
                    max_packet_size=64,
                    interval=32,
                    handler=None
                ),
            ],
        )
        self.object = MtpObject.from_fs_recursive('mtp_fs')
        # self.object = MtpObject.from_fs_recursive('mtp_fs/eits.mp3')
        self.storage_info = MtpStorageInfo(
            st_type=1,
            fs_type=2,
            access=0,
            max_cap=150000,
            free_bytes=0,
            free_objs=0,
            desc='MyStorage',
            vol_id='Python MTP Device Stack',
        )
        self.storage = MtpStorage(self.storage_info)
        self.storage.add_object(self.object)
        self.dev_info = MtpDeviceInfo(
            std_version=0x0064,
            mtp_vendor_ext_id=0x00000006,
            mtp_version=0x0064,
            mtp_extensions='microsoft.com: 1.0;',
            functional_mode=0x0000,
            capture_formats=[],
            playback_formats=[],
            manufacturer='BinyaminSharet',
            model='Role',
            device_version='1.2',
            serial_number='3031323334353637',
        )
        properties = [
            MtpDeviceProperty(MtpDevicePropertyCode.MTP_DeviceFriendlyName, 0, MStr('UmapMtpDevice'), MStr('')),
            MtpDeviceProperty(MtpDevicePropertyCode.BatteryLevel, 0, UInt8(100), UInt8(0))
        ]
        self.dev = MtpDevice(self.dev_info, properties, self.logger)
        self.dev.add_storage(self.storage)
        self.dev.set_fuzzer(app.fuzzer)
        self.api = MtpApi(self.dev)

        # OS String descriptor
        # self.add_string_with_id(50, 'MTP'.encode('utf-16') + b'\x00\x00')
        self.add_string_with_id(0xee, 'MSFT100'.encode('utf-16') + b'\x00\x00')

コード例 #29

ファイル: mtp_object_tests.py プロジェクト: BinyaminSharet/Mtp

 def testGetPropertyExceptionIfNotExist(self):
     uut = MtpObject(self.default_data, self.get_default_info())
     with self.assertRaises(MtpProtocolException) as cm:
         uut.get_property(MtpObjectPropertyCode.StorageID - 1)
     self.assertEqual(cm.exception.response, ResponseCodes.OBJECT_PROP_NOT_SUPPORTED)

コード例 #30

ファイル: mtp_object_tests.py プロジェクト: dark-lbp/Mtp

 def testFormatMatchsWildcard(self):
     uut = MtpObject(self.default_data, self.get_default_info())
     self.assertTrue(uut.format_matches(0xffffffff))

コード例 #31

ファイル: mtp_object_tests.py プロジェクト: dark-lbp/Mtp

 def testGetPropertyDefaults(self):
     uut = MtpObject(self.default_data, self.get_default_info())
     prop = uut.get_property(MtpObjectPropertyCode.ObjectFormat)
     self.assertIsNotNone(prop)
     self.assertEqual(prop.value.value, self.default_format)

コード例 #32

ファイル: mtp_object_tests.py プロジェクト: BinyaminSharet/Mtp

 def testFormatMatchsNone(self):
     uut = MtpObject(self.default_data, self.get_default_info())
     self.assertTrue(uut.format_matches(None))

コード例 #33

ファイル: mtp_object_tests.py プロジェクト: dark-lbp/Mtp

 def testObjectData(self):
     uut = MtpObject(self.default_data, self.get_default_info())
     self.assertEqual(uut.data, self.default_data)

コード例 #34

ファイル: mtp_object_tests.py プロジェクト: dark-lbp/Mtp

 def testAddObjectSetsParent(self):
     parent = MtpObject(self.default_data, self.get_default_info())
     uut = MtpObject(self.default_data, self.get_default_info())
     parent.add_object(uut)
     self.assertEqual(uut.parent, parent)

コード例 #35

ファイル: mtp_object_tests.py プロジェクト: dark-lbp/Mtp

 def testGetInfoReturnsPackedInfo(self):
     info = self.get_default_info()
     uut = MtpObject(self.default_data, info)
     self.assertEqual(uut.get_info(), info.pack())

コード例 #36

ファイル: mtp_object_tests.py プロジェクト: BinyaminSharet/Mtp

 def testAddObjectSetsParent(self):
     parent = MtpObject(self.default_data, self.get_default_info())
     uut = MtpObject(self.default_data, self.get_default_info())
     parent.add_object(uut)
     self.assertEqual(uut.parent, parent)

コード例 #37

ファイル: mtp.py プロジェクト: bkerler/sahara_emulator

    def __init__(self, app, phy):
        if not mtpdeviceloaded:
            raise Exception('You cannot use USBMtp until you install pymtpdevice')
        # TODO: un-hardcode string index (last arg before 'verbose')
        super(USBMtpInterface, self).__init__(
            app=app,
            phy=phy,
            interface_number=0,
            interface_alternate=0,
            interface_class=USBClass.VendorSpecific,
            interface_subclass=0xff,
            interface_protocol=0,
            interface_string_index=0,
            endpoints=[
                USBEndpoint(
                    app=app,
                    phy=phy,
                    number=1,
                    direction=USBEndpoint.direction_out,
                    transfer_type=USBEndpoint.transfer_type_bulk,
                    sync_type=USBEndpoint.sync_type_none,
                    usage_type=USBEndpoint.usage_type_data,
                    max_packet_size=64,
                    interval=0,
                    handler=self.handle_ep1_data_available
                ),
                USBEndpoint(
                    app=app,
                    phy=phy,
                    number=2,
                    direction=USBEndpoint.direction_in,
                    transfer_type=USBEndpoint.transfer_type_bulk,
                    sync_type=USBEndpoint.sync_type_none,
                    usage_type=USBEndpoint.usage_type_data,
                    max_packet_size=64,
                    interval=0,
                    handler=None
                ),
                USBEndpoint(
                    app=app,
                    phy=phy,
                    number=3,
                    direction=USBEndpoint.direction_in,
                    transfer_type=USBEndpoint.transfer_type_interrupt,
                    sync_type=USBEndpoint.sync_type_none,
                    usage_type=USBEndpoint.usage_type_data,
                    max_packet_size=64,
                    interval=32,
                    handler=None
                ),
            ],
        )
        self.object = MtpObject.from_fs_recursive('mtp_fs')
        # self.object = MtpObject.from_fs_recursive('mtp_fs/eits.mp3')
        self.storage_info = MtpStorageInfo(
            st_type=1,
            fs_type=2,
            access=0,
            max_cap=150000,
            free_bytes=0,
            free_objs=0,
            desc='MyStorage',
            vol_id='Python MTP Device Stack',
        )
        self.storage = MtpStorage(self.storage_info)
        self.storage.add_object(self.object)
        self.dev_info = MtpDeviceInfo(
            std_version=0x0064,
            mtp_vendor_ext_id=0x00000006,
            mtp_version=0x0064,
            mtp_extensions='microsoft.com: 1.0;',
            functional_mode=0x0000,
            capture_formats=[],
            playback_formats=[],
            manufacturer='UMAP2',
            model='Role',
            device_version='1.2',
            serial_number='3031323334353637',
        )
        properties = [
            MtpDeviceProperty(MtpDevicePropertyCode.MTP_DeviceFriendlyName, 0, MStr('UmapMtpDevice'), MStr('')),
            MtpDeviceProperty(MtpDevicePropertyCode.BatteryLevel, 0, UInt8(100), UInt8(0))
        ]
        self.dev = MtpDevice(self.dev_info, properties, self.logger)
        self.dev.add_storage(self.storage)
        self.dev.set_fuzzer(app.fuzzer)
        self.api = MtpApi(self.dev)

        # OS String descriptor
        # self.add_string_with_id(50, 'MTP'.encode('utf-16') + b'\x00\x00')
        self.add_string_with_id(0xee, 'MSFT100'.encode('utf-16') + b'\x00\x00')

コード例 #38

ファイル: mtp_object_tests.py プロジェクト: BinyaminSharet/Mtp

 def testGetObjectsEmpty(self):
     uut = MtpObject(self.default_data, self.get_default_info())
     self.assertEqual(uut.get_objects(), [])

コード例 #39

ファイル: mtp_object_tests.py プロジェクト: dark-lbp/Mtp

 def testGetObjectSelfAlone(self):
     uut = MtpObject(self.default_data, self.get_default_info())
     obj = uut.get_object(uut.get_uid())
     self.assertEqual(uut, obj)

コード例 #40

ファイル: mtp_object_tests.py プロジェクト: BinyaminSharet/Mtp

 def testGetObjectSelfAlone(self):
     uut = MtpObject(self.default_data, self.get_default_info())
     obj = uut.get_object(uut.get_uid())
     self.assertEqual(uut, obj)