mbirth/tcl_ota_check
1
0
Fork 0
mirror of https://github.com/mbirth/tcl_ota_check.git synced 2024-11-10 06:16:46 +00:00
Code Issues Releases Wiki Activity

Converted tcldown.py to new style.

Browse Source
This commit is contained in:
Markus Birth 2018-02-11 01:40:52 +01:00
parent 7c1a178d3e
commit e6e0e83039
Signed by: mbirth
GPG Key ID: A9928D7A098C3A9A
2 changed files with 38 additions and 29 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
tclcheck.py
View File

@ -14,7 +14,6 @@ from tcllib.devices import Device
from tcllib.requests import RequestRunner, CheckRequest, DownloadRequest, \ from tcllib.requests import RequestRunner, CheckRequest, DownloadRequest, \
ChecksumRequest, EncryptHeaderRequest, ServerSelector, \ ChecksumRequest, EncryptHeaderRequest, ServerSelector, \
write_info_if_dumps_found write_info_if_dumps_found
from tcllib.xmltools import pretty_xml
dpdesc = """ dpdesc = """

52
tcldown.py
View File

@ -6,22 +6,19 @@
"""Download a given firmware file.""" """Download a given firmware file."""
import os import os
import random
import sys import sys
import tcllib from tcllib import argparser
import tcllib.argparser
from tcllib.devices import DesktopDevice from tcllib.devices import DesktopDevice
from tcllib.xmltools import pretty_xml from tcllib.requests import RequestRunner, CheckRequest, DownloadRequest, \
ChecksumRequest, EncryptHeaderRequest, ServerSelector, \
write_info_if_dumps_found
fc = tcllib.FotaCheck()
dev = DesktopDevice()
dpdesc = """ dpdesc = """
Downloads the given firmware file. Downloads the given firmware file.
""" """
dp = tcllib.argparser.DefaultParser(__file__, dpdesc) dp = argparser.DefaultParser(__file__, dpdesc)
dp.add_argument("prd", nargs=1, help="CU Reference #, e.g. PRD-63117-011") dp.add_argument("prd", nargs=1, help="CU Reference #, e.g. PRD-63117-011")
dp.add_argument("targetversion", nargs=1, help="Firmware version to download, e.g. AAN990") dp.add_argument("targetversion", nargs=1, help="Firmware version to download, e.g. AAN990")
dp.add_argument("fwid", nargs=1, help="Numeric firmware file id, e.g. 268932") dp.add_argument("fwid", nargs=1, help="Numeric firmware file id, e.g. 268932")
@ -32,6 +29,7 @@ dp.add_argument("--rawmode", help="override --mode with raw value (2=OTA, 4=FULL
dp.add_argument("--rawcltp", help="override --type with raw value (10=MOBILE, 2010=DESKTOP)", metavar="CLTP") dp.add_argument("--rawcltp", help="override --type with raw value (10=MOBILE, 2010=DESKTOP)", metavar="CLTP")
args = dp.parse_args(sys.argv[1:]) args = dp.parse_args(sys.argv[1:])
dev = DesktopDevice()
def sel_mode(defaultmode, rawval): def sel_mode(defaultmode, rawval):
"""Handle custom mode.""" """Handle custom mode."""
@ -65,30 +63,42 @@ dev.cltp = sel_cltp(args.type, args.rawcltp)
print("Mode: {}".format(dev.mode)) print("Mode: {}".format(dev.mode))
print("CLTP: {}".format(dev.cltp)) print("CLTP: {}".format(dev.cltp))
fv = dev.fwver runner = RequestRunner(ServerSelector())
runner.max_tries = 20
tv = args.targetversion[0] tv = args.targetversion[0]
fw_id = args.fwid[0] fw_id = args.fwid[0]
req_xml = fc.do_request(dev.curef, fv, tv, fw_id) dlr = DownloadRequest(dev, tv, fw_id)
print(pretty_xml(req_xml)) runner.run(dlr)
fileid, fileurl, slaves, encslaves, s3_fileurl, s3_slaves = fc.parse_request(req_xml) if not dlr.success:
print("ERROR: {}".format(dlr.error))
sys.exit(3)
for s in slaves: dlrres = dlr.get_result()
print("http://{}{}".format(s, fileurl)) print(dlrres.pretty_xml())
for s in s3_slaves: for s in dlrres.slaves:
print("http://{}{}".format(s, s3_fileurl)) print("http://{}{}".format(s, dlrres.fileurl))
for s in dlrres.s3_slaves:
print("http://{}{}".format(s, dlrres.s3_fileurl))
if dev.mode == dev.MODE_STATES["FULL"]: if dev.mode == dev.MODE_STATES["FULL"]:
header = fc.do_encrypt_header(random.choice(encslaves), fileurl) encrun = RequestRunner(ServerSelector(dlrres.encslaves), https=False)
encrun.max_tries = 20
hdr = EncryptHeaderRequest(dlrres.fileurl)
encrun.run(hdr)
if hdr.success:
hdrres = hdr.get_result()
headname = "header_{}.bin".format(tv) headname = "header_{}.bin".format(tv)
headdir = "headers" headdir = "headers"
if not os.path.exists(headdir): if not os.path.exists(headdir):
os.makedirs(headdir) os.makedirs(headdir)
if len(header) == 4194320: if len(hdrres.rawdata) == 4194320:
print("Header length check passed. Writing to {}.".format(headname)) print("Header length check passed. Writing to {}.".format(headname))
with open(os.path.join(headdir, headname), "wb") as f: with open(os.path.join(headdir, headname), "wb") as f:
f.write(header) f.write(hdrres.rawdata)
else: else:
print("Header length invalid ({}).".format(len(header))) print("Header length invalid ({}).".format(len(hdrres.rawdata)))
tcllib.FotaCheck.write_info_if_dumps_found() write_info_if_dumps_found()