WebSVN – freemyipod – Diff – /emcore/trunk/tools/libipodcrypto.py

     emcore.aesencrypt(addr + 0x40, 0x10, 1)
     data = emcore.read(addr, 0x50) + data + cert.ljust((len(cert) + 0xf) & ~0xf, b"\0")
     emcore.free(addr)
     return data
+def fileoperation(infilepath, outfilepath, function):
+    with open(infilepath, "rb") as infile:
+        infiledata = infile.read()
+    try:
+        outfiledata = function(infiledata)
+    except:
+        os.remove(outfilepath)
+        raise
+    with open(outfilepath, "wb") as outfile:
+        outfile.write(outfiledata)
 def s5l8701cryptdfufile(infile, outfile):
-    infile = open(infile, "rb")
-    outfile = open(outfile, "wb")
-    outfile.write(s5l8701cryptdfu(infile.read()))
+    fileoperation(infile, outfile, s5l8701cryptdfu)
-    infile.close()
-    outfile.close()
 def s5l8701decryptdfufile(infile, outfile):
-    infile = open(infile, "rb")
-    outfile = open(outfile, "wb")
-    outfile.write(s5l8701decryptdfu(infile.read()))
+    fileoperation(infile, outfile, s5l8701decryptdfu)
-    infile.close()
-    outfile.close()
 def s5l8701cryptfirmwarefile(infile, outfile):
-    infile = open(infile, "rb")
-    outfile = open(outfile, "wb")
-    outfile.write(s5l8701cryptfirmware(infile.read()))
+    fileoperation(infile, outfile, s5l8701cryptfirmware)
-    infile.close()
-    outfile.close()
 def s5l8701decryptfirmwarefile(infile, outfile):
-    infile = open(infile, "rb")
-    outfile = open(outfile, "wb")
-    outfile.write(s5l8701decryptfirmware(infile.read()))
+    fileoperation(infile, outfile, s5l8701decryptfirmware)
-    infile.close()
-    outfile.close()
 def s5l8702cryptnorfile(infile, outfile):
-    infile = open(infile, "rb")
-    outfile = open(outfile, "wb")
-    outfile.write(s5l8702cryptnor(infile.read()))
+    fileoperation(infile, outfile, s5l8702cryptnor)
-    infile.close()
-    outfile.close()
 def s5l8702decryptnorfile(infile, outfile):
-    infile = open(infile, "rb")
-    outfile = open(outfile, "wb")
-    outfile.write(s5l8702decryptnor(infile.read()))
+    fileoperation(infile, outfile, s5l8702decryptnor)
-    infile.close()
-    outfile.close()
 def s5l8702genpwnagefile(infile, outfile):
-    infile = open(infile, "rb")
-    outfile = open(outfile, "wb")
-    outfile.write(s5l8702genpwnage(infile.read()))
+    fileoperation(infile, outfile, s5l8702genpwnage)
-    infile.close()
-    outfile.close()
 def s5l8702genpwnagefile800(infile, outfile):
-    infile = open(infile, "rb")
-    outfile = open(outfile, "wb")
-    outfile.write(s5l8702genpwnage800(infile.read()))
+    fileoperation(infile, outfile, s5l8702genpwnage800)
-    infile.close()
-    outfile.close()
 def s5l8720genpwnagefile(infile, outfile):
-    infile = open(infile, "rb")
-    outfile = open(outfile, "wb")
-    outfile.write(s5l8720genpwnage(infile.read()))
+    fileoperation(infile, outfile, s5l8720genpwnage)
-    infile.close()
-    outfile.close()

Subversion Repositories freemyipod

(root)/emcore/trunk/tools/libipodcrypto.py @ 860 – Rev 794 → 860