danvics/python-validity
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

upload_fwext: Get firmware dynamically dependent on chip model

Browse source
This commit is contained in:
Marco Trevisan (Treviño) 2020-06-09 19:57:44 +02:00
parent 4b5dd81ec2
commit 64544fab70
1 changed files with 18 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

20
proto97/upload_fwext.py
View file

@ -1,6 +1,7 @@
from binascii import hexlify, unhexlify from binascii import hexlify, unhexlify
from time import ctime from time import ctime
from os.path import basename
from .tls import tls from .tls import tls
from .usb import usb from .usb import usb
@ -8,7 +9,15 @@ from .sensor import reboot, write_hw_reg32, read_hw_reg32, identify_sensor
from .flash import flush_changes, read_flash, erase_flash, write_flash_all, write_fw_signature, get_fw_info from .flash import flush_changes, read_flash, erase_flash, write_flash_all, write_fw_signature, get_fw_info
from .util import assert_status from .util import assert_status
def upload_fwext():
def default_fwext_name():
if usb.usb_dev().idVendor == 0x138a:
if usb.usb_dev().idProduct == 0x0090:
return '6_07f_Lenovo.xpfwext'
elif usb.usb_dev().idProduct == 0x0097:
return '6_07f_lenovo_mis.xpfwext'
def upload_fwext(fw_path=None):
# no idea what this is: # no idea what this is:
write_hw_reg32(0x8000205c, 7) write_hw_reg32(0x8000205c, 7)
if read_hw_reg32(0x80002080) != 2: if read_hw_reg32(0x80002080) != 2:
@ -19,7 +28,14 @@ def upload_fwext():
# ^ TODO -- what is the real reason to detect HW at this stage? # ^ TODO -- what is the real reason to detect HW at this stage?
# just a guess: perhaps it is used to construct fwext filename # just a guess: perhaps it is used to construct fwext filename
with open('6_07f_lenovo_mis.xpfwext', 'rb') as f: default_name = default_fwext_name()
if not fw_path:
fw_path = default_name
elif basename(fw_path) != default_name:
print('WARNING: Your device fw is supposed to be called {}'.format(
default_name))
with open(fw_path, 'rb') as f:
fwext=f.read() fwext=f.read()
fwext=fwext[fwext.index(b'\x1a')+1:] fwext=fwext[fwext.index(b'\x1a')+1:]