telomere writefile fixes

Author: lolololol

experiments/ml/telomere-masked2.py

@@ -12,10 +12,11 @@
 import sparkgap.filemanager
 import numpy as np
 import gc
+import binascii
 
 TRACE = []
 TRACECOUNT = 1500
-WRITEFILE = "emu_normal.hdf"
+WRITEFILE = "emu_masked.hdf"
 
 random.seed()
 
@@ -32,6 +33,10 @@
       WRITEFILE = arg
 
 rand_key = np.array([random.randint(0,0xFF) for i in range(0,16)],dtype=np.uint8)
+rand_mask = np.array([random.randint(0,0xFF) for i in range(0,16)],dtype=np.uint8)
+if rekey is False and WRITEFILE == "emu_masked.hdf":
+  WRITEFILE = "%s_masked_%s.hdf" % (binascii.hexlify(rand_key).decode("utf-8"),binascii.hexlify(rand_mask).decode("utf-8"))
+  print("Resetting writefile to %s" % WRITEFILE)
 cs = None
 for i in range(0,TRACECOUNT):
   emu = rainbow_arm(sca_mode=True)
@@ -41,6 +46,10 @@
   emu[0x20000000 + 4] = bytes(rand_text[4:8])
   emu[0x20000000 + 8] = bytes(rand_text[8:12])
   emu[0x20000000 + 12] = bytes(rand_text[12:16])
+  emu[0x20000040] = bytes(rand_mask[0:4])
+  emu[0x20000040 + 4] = bytes(rand_mask[4:8])
+  emu[0x20000040 + 8] = bytes(rand_mask[8:12])
+  emu[0x20000040 + 12] = bytes(rand_mask[12:16])
   if rekey:
     (pt,keybyte) = cherrypicker.getRandomPair()
     rand_key = np.array([random.randint(0,0xFF) for i in range(0,16)],dtype=np.uint8)
@@ -52,9 +61,7 @@
   emu[0x20000010 + 4] = bytes(rand_key[4:8])
   emu[0x20000010 + 8] = bytes(rand_key[8:12])
   emu[0x20000010 + 12] = bytes(rand_key[12:16])
-  print("stop1")
   emu.start(emu.functions["doAES"] | 1,0x08000262)
-  print("stop2")
   new_trace = np.fromiter(map(hw,emu.sca_values_trace),dtype=np.float32)
   if cs is None: 
     print("Creating CaptureSet")

experiments/ml/telomere.py

@@ -17,7 +17,6 @@
 TRACECOUNT = 1500
 WRITEFILE = "emu_normal.hdf"
 
-print("telomere")
 random.seed()
 
 rekey = False
@@ -32,7 +31,12 @@
     elif opt in ("-w","--writefile"):
       WRITEFILE = arg
 
+import binascii
+
 rand_key = np.array([random.randint(0,0xFF) for i in range(0,16)],dtype=np.uint8)
+if rekey is False and WRITEFILE == "emu_normal.hdf":
+  WRITEFILE = "normal_%s.hdf" % binascii.hexlify(rand_key).decode("utf-8")
+  print("Configuring writefile to %s" % WRITEFILE)
 cs = None
 for i in range(0,TRACECOUNT):
   emu = rainbow_arm(sca_mode=True)

nddla.py

@@ -19,11 +19,11 @@
 
 BYTENUM_MIN = 0
 BYTENUM_MAX = 1
-# KEYBYTE_MIN = 0xb0
-# KEYBYTE_MAX = 0xc0
+KEYBYTE_MIN = 0x63
+KEYBYTE_MAX = 0x69
 
-KEYBYTE_MIN = 0xea
-KEYBYTE_MAX = 0xef
+# KEYBYTE_MIN = 0xea
+# KEYBYTE_MAX = 0xef
 
 if __name__ == "__main__":
   opts, args = getopt.getopt(sys.argv[1:],"f:o:n:a:",["file=","offset=","numsamples=","attack="])