def test_set_binary(): binary = get_binary_from_file('{}/get_files_test/testfile1'.format( get_test_data_dir())) md5 = 'e802ca22f6cd2d9357cf3da1d191879e' firmware = Firmware() firmware.set_binary(binary) assert firmware.md5 == md5
def convert_analysis_task_to_fw_obj(analysis_task): fw = Firmware( scheduled_analysis=analysis_task['requested_analysis_systems']) if 'binary' in analysis_task.keys(): fw.set_binary(analysis_task['binary']) fw.file_name = analysis_task['file_name'] else: if 'file_name' in analysis_task.keys(): fw.file_name = analysis_task['file_name'] fw.overwrite_uid(analysis_task['uid']) fw.set_device_name(analysis_task['device_name']) fw.set_firmware_version(analysis_task['firmware_version']) fw.set_device_class(analysis_task['device_class']) fw.set_vendor(analysis_task['vendor']) fw.set_release_date(analysis_task['release_date']) return fw
def convert_analysis_task_to_fw_obj(analysis_task): fw = Firmware(scheduled_analysis=analysis_task['requested_analysis_systems']) if 'binary' in analysis_task.keys(): fw.set_binary(analysis_task['binary']) fw.file_name = analysis_task['file_name'] else: if 'file_name' in analysis_task.keys(): fw.file_name = analysis_task['file_name'] fw.uid = analysis_task['uid'] fw.set_device_name(analysis_task['device_name']) fw.set_part_name(analysis_task['device_part']) fw.set_firmware_version(analysis_task['version']) fw.set_device_class(analysis_task['device_class']) fw.set_vendor(analysis_task['vendor']) fw.set_release_date(analysis_task['release_date']) for tag in _get_tag_list(analysis_task['tags']): fw.set_tag(tag) return fw
def convert_analysis_task_to_fw_obj(analysis_task: dict) -> Firmware: ''' Convert an analysis task to a firmware object. :param analysis_task: The analysis task data. :return: A new `Firmware` object based on the analysis task data. ''' fw = Firmware(scheduled_analysis=analysis_task['requested_analysis_systems']) if 'binary' in analysis_task.keys(): fw.set_binary(analysis_task['binary']) fw.file_name = analysis_task['file_name'] else: if 'file_name' in analysis_task.keys(): fw.file_name = analysis_task['file_name'] fw.uid = analysis_task['uid'] fw.set_device_name(analysis_task['device_name']) fw.set_part_name(analysis_task['device_part']) fw.set_firmware_version(analysis_task['version']) fw.set_device_class(analysis_task['device_class']) fw.set_vendor(analysis_task['vendor']) fw.set_release_date(analysis_task['release_date']) for tag in _get_tag_list(analysis_task['tags']): fw.set_tag(tag) return fw
def test_set_binary(self): binary = get_binary_from_file('{}/get_files_test/testfile1'.format(get_test_data_dir())) md5 = 'e802ca22f6cd2d9357cf3da1d191879e' firmware = Firmware() firmware.set_binary(binary) self.assertEqual(firmware.md5, md5, 'correct md5 sum')