1 files changed, 87 insertions, 0 deletions
diff --git a/samloader/main.py b/samloader/main.py
new file mode 100644
index 0000000..e8fca4f
--- /dev/null
+++ b/samloader/main.py
@@ -0,0 +1,87 @@
+# SPDX-License-Identifier: GPL-3.0+
+# Copyright (C) 2020 Nayil Mukhametshin
+
+import click
+import os
+import xml.etree.ElementTree as ET
+from clint.textui import progress
+
+from . import request
+from . import crypt
+from . import fusclient
+from . import versionfetch
+
+def getbinaryfile(client, fw, region, model):
+    req = request.binaryinform(fw, region, model, client.nonce)
+    resp = client.makereq("NF_DownloadBinaryInform.do", req)
+    root = ET.fromstring(resp)
+    filename = root.find("./FUSBody/Put/BINARY_NAME/Data").text
+    path = root.find("./FUSBody/Put/MODEL_PATH/Data").text
+    return path, filename
+
+def initdownload(client, filename):
+    req = request.binaryinit(filename, client.nonce)
+    resp = client.makereq("NF_DownloadBinaryInitForMass.do", req)
+
+@click.group()
+def cli():
+    pass
+
+@cli.command(help="Check the update server for the latest available firmware.")
+@click.argument("model")
+@click.argument("region")
+def checkupdate(model, region):
+    fw = versionfetch.getlatestver(region, model)
+    print(fw)
+
+@cli.command(help="Download the specified firmware version.")
+@click.argument("version")
+@click.argument("model")
+@click.argument("region")
+@click.argument("outfile")
+def download(version, model, region, outfile):
+    client = fusclient.FUSClient()
+    path, filename = getbinaryfile(client, version, region, model)
+    print("Downloading file {} ...".format(path+filename))
+    initdownload(client, filename)
+    r = client.downloadfile(path+filename)
+    length = int(r.headers["Content-Length"])
+    with open(outfile, "wb") as f:
+        for chunk in progress.bar(r.iter_content(chunk_size=0x10000), expected_size=(length/0x10000)+1):
+            if chunk:
+                f.write(chunk)
+                f.flush()
+    print("Done!")
+
+@cli.command(help="Decrypt enc4 files.")
+@click.argument("version")
+@click.argument("model")
+@click.argument("region")
+@click.argument("infile")
+@click.argument("outfile")
+def decrypt4(version, model, region, infile, outfile):
+    key = crypt.getv4key(version, model, region)
+    print("Decrypting with key {}...".format(key.hex()))
+    length = os.stat(infile).st_size
+    with open(infile, "rb") as inf:
+        with open(outfile, "wb") as outf:
+            crypt.decrypt_progress(inf, outf, key, length)
+    print("Done!")
+
+@cli.command(help="Decrypt enc2 files.")
+@click.argument("version")
+@click.argument("model")
+@click.argument("region")
+@click.argument("infile")
+@click.argument("outfile")
+def decrypt2(version, model, region, infile, outfile):
+    key = crypt.getv2key(version, model, region)
+    print("Decrypting with key {}...".format(key.hex()))
+    length = os.stat(infile).st_size
+    with open(infile, "rb") as inf:
+        with open(outfile, "wb") as outf:
+            crypt.decrypt_progress(inf, outf, key, length)
+    print("Done!")
+
+if __name__ == "__main__":
+    cli()