1
0
mirror of https://github.com/mbirth/tcl_ota_check.git synced 2024-09-19 22:33:25 +01:00

save downloaded encrypted headers in subfolder

This commit is contained in:
thurask 2017-11-03 01:00:49 -04:00
parent 26745f024d
commit f60b44ddea
No known key found for this signature in database
GPG Key ID: A6CCCDEA29795048
2 changed files with 8 additions and 2 deletions

1
.gitignore vendored
View File

@ -1,4 +1,5 @@
/__pycache__/
/logs/
/headers/
/.nvimlog
/header*.bin

View File

@ -3,6 +3,7 @@
# pylint: disable=C0111,C0326,C0103
import os
import random
import sys
import tcllib
@ -44,9 +45,13 @@ for s in slaves:
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 header_{}.bin.".format(tv))
with open("header_{}.bin".format(tv), "wb") as f:
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)))