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)
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)
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)
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)
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)
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])
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])
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))
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)
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))
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
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)
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])
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)
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))
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)
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])
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))
def testFormatMatchsNone(self): uut = MtpObject(self.default_data, self.get_default_info()) self.assertTrue(uut.format_matches(None))
def testGetObjectDoesntExist(self): uut = MtpObject(self.default_data, self.get_default_info()) obj = uut.get_object(uut.get_uid() + 1) self.assertIsNone(obj)
def testGetInfoReturnsPackedInfo(self): info = self.get_default_info() uut = MtpObject(self.default_data, info) self.assertEqual(uut.get_info(), info.pack())
def testGetObjectsEmpty(self): uut = MtpObject(self.default_data, self.get_default_info()) self.assertEqual(uut.get_objects(), [])
def testObjectInfo(self): info = self.get_default_info() uut = MtpObject(self.default_data, info) self.assertEqual(uut.info, info)
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)
def testGetObjectDoesntExist(self): uut = MtpObject(self.default_data, self.get_default_info()) obj = uut.get_object(uut.get_uid() + 1) self.assertIsNone(obj)
def testFormatMatchsWildcard(self): uut = MtpObject(self.default_data, self.get_default_info()) self.assertTrue(uut.format_matches(0xffffffff))
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)
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')
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)
def testFormatMatchsWildcard(self): uut = MtpObject(self.default_data, self.get_default_info()) self.assertTrue(uut.format_matches(0xffffffff))
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)
def testFormatMatchsNone(self): uut = MtpObject(self.default_data, self.get_default_info()) self.assertTrue(uut.format_matches(None))
def testObjectData(self): uut = MtpObject(self.default_data, self.get_default_info()) self.assertEqual(uut.data, self.default_data)
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)
def testGetInfoReturnsPackedInfo(self): info = self.get_default_info() uut = MtpObject(self.default_data, info) self.assertEqual(uut.get_info(), info.pack())
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)
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')
def testGetObjectsEmpty(self): uut = MtpObject(self.default_data, self.get_default_info()) self.assertEqual(uut.get_objects(), [])
def testGetObjectSelfAlone(self): uut = MtpObject(self.default_data, self.get_default_info()) obj = uut.get_object(uut.get_uid()) self.assertEqual(uut, obj)
def testGetObjectSelfAlone(self): uut = MtpObject(self.default_data, self.get_default_info()) obj = uut.get_object(uut.get_uid()) self.assertEqual(uut, obj)