save: fix to dpa TRACE_LENGTH

Author: lolololol

dpa.py

@@ -12,7 +12,7 @@
 import sparkgap.attack
 
 TRACE_OFFSET = 0
-TRACE_LENGTH = 17000
+TRACE_LENGTH = None
 TRACE_MAX = 0
 
 CONFIG_PLOT = True
@@ -116,12 +116,17 @@ def usage():
     elif opt == "--txt":
       CONFIG_PLOT = False
   print("TRACE_OFFSET = %d" % TRACE_OFFSET)
-  print("TRACE_LENGTH = %d" % TRACE_LENGTH)
+  if TRACE_LENGTH is None:
+    print("Delayed loading TRACE_LENGTH")
+  else:
+    print("TRACE_LENGTH = %d" % TRACE_LENGTH)
   if fn is None:
     print("You must specify a file with -f")
     sys.exit(0)
   print("Stage 1: Loading plaintexts...")
   tm = sparkgap.filemanager.TraceManager(fn)
+  TRACE_LENGTH = len(tm.traces[0]) - TRACE_OFFSET
+  print("TRACE_LENGTH = %d" % TRACE_LENGTH)
   print("Deriving key... wish me luck!")
   if CONFIG_PLOT:
     resultViz = sparkgap.resultviz.VisualizerApp()

emu/des_loader.py

@@ -38,9 +38,12 @@
 DATA_ADDR = 0x18cc0
 OUT_ADDR = 0x18cb0
 
-DODES_END = 0x8974
+# DODES_END = 0x8974   # WHOLE
+DODES_END = 0x8668   # FIRSTROUND
 
-rand_key = np.array([random.randint(0,0xFF) for i in range(0,8)],dtype=np.uint8)
+
+# rand_key = np.array([random.randint(0,0xFF) for i in range(0,8)],dtype=np.uint8)
+rand_key = np.array([0x11,0x22,0x33,0x44,0x55,0x66,0x77,0x88])
 key_str = " ".join(["%02x" % x for x in rand_key])
 
 if CONFIG_REKEY is False: