From cf04ba53df0e4a80c7797b4157ffe25eca02e0a2 Mon Sep 17 00:00:00 2001
From: Markus Birth <mbirth@gmail.com>
Date: Thu, 11 Jan 2018 23:20:33 +0100
Subject: [PATCH] Tool to test request_download without going through
 check.php.

---
 tcldown.py | 82 ++++++++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 82 insertions(+)
 create mode 100644 tcldown.py

diff --git a/tcldown.py b/tcldown.py
new file mode 100644
index 0000000..9c3790a
--- /dev/null
+++ b/tcldown.py
@@ -0,0 +1,82 @@
+#!/usr/bin/env python3
+# -*- coding: utf-8 -*-
+
+# pylint: disable=C0111,C0326,C0103
+
+import os
+import random
+import sys
+import tcllib
+
+fc = tcllib.FotaCheck()
+fc.serid = "3531510"
+#fc.osvs  = "7.1.1"
+
+dp = tcllib.DefaultParser(__file__)
+dp.description = """
+    Downloads the given firmware file.
+    """
+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("fwid", nargs=1, help="Numeric firmware file id, e.g. 268932")
+dp.add_argument("-o", "--ota", metavar="FROMVERSION", help="download OTA from specified version (switches mode to OTA)", type=str)
+dp.add_argument("-i", "--imei", help="use specified IMEI instead of default", type=str)
+dp.add_argument("-t", "--type", help="download device (default to desktop)", default="desktop", type=str, choices=["desktop", "mobile"])
+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)")
+args = dp.parse_args(sys.argv[1:])
+
+def sel_mode(defaultmode, rawval):
+    if rawval:
+        enum = tcllib.default_enum("MODE", {"RAW": rawval})
+        return enum.RAW
+    return defaultmode
+
+def sel_cltp(txtmode, rawval):
+    if rawval:
+        enum = tcllib.default_enum("CLTP", {"RAW": rawval})
+        return enum.RAW
+    if txtmode == "mobile":
+        return fc.CLTP.MOBILE
+    return fc.CLTP.DESKTOP
+
+if args.imei:
+    print("Use specified IMEI: {}".format(args.imei))
+    fc.serid = args.imei
+
+fc.curef = args.prd[0]
+if args.ota:
+    fc.fv   = args.ota[0]
+    fc.mode = sel_mode(fc.MODE.OTA, args.rawmode)
+else:
+    fc.fv   = args.targetversion[0]
+    fc.mode = sel_mode(fc.MODE.FULL, args.rawmode)
+fc.cltp = sel_cltp(args.type, args.rawcltp)
+
+print("Mode: {}".format(fc.mode.value))
+print("CLTP: {}".format(fc.cltp.value))
+
+fv = fc.fv
+tv = args.targetversion[0]
+fw_id = args.fwid[0]
+req_xml = fc.do_request(fc.curef, fv, tv, fw_id)
+print(fc.pretty_xml(req_xml))
+fileid, fileurl, slaves, encslaves = fc.parse_request(req_xml)
+
+for s in slaves:
+    print("http://{}{}".format(s, fileurl))
+
+if fc.mode == fc.MODE.FULL:
+    header = fc.do_encrypt_header(random.choice(encslaves), fileurl)
+    headname = "header_{}.bin".format(tv)
+    headdir = "headers"
+    if not os.path.exists(headdir):
+        os.makedirs(headdir)
+    if len(header) == 4194320:
+        print("Header length check passed. Writing to {}.".format(headname))
+        with open(os.path.join(headdir, headname), "wb") as f:
+            f.write(header)
+    else:
+        print("Header length invalid ({}).".format(len(header)))
+
+tcllib.FotaCheck.write_info_if_dumps_found()