1 files changed, 87 insertions, 0 deletions
diff --git a/sources/apps/QUAC-TRNG/scripts/sha_bitstreams.py b/sources/apps/QUAC-TRNG/scripts/sha_bitstreams.py
new file mode 100644
index 0000000..fd4dbfb
--- /dev/null
+++ b/sources/apps/QUAC-TRNG/scripts/sha_bitstreams.py
@@ -0,0 +1,87 @@
+import os
+import argparse
+import subprocess
+import time
+
+parser = argparse.ArgumentParser(description = "Collects sha bitstreams for a data pattern")
+
+parser.add_argument('dimm', help="DIMM label under test")
+parser.add_argument('pattern', help="data pattern to use")
+parser.add_argument('temperature', help="DIMM temperature in centigrades")
+
+args = parser.parse_args()
+
+bitstreams_available = False
+
+if not os.path.isdir("data/" + args.dimm + "/sha"):
+  os.mkdir("data/" + args.dimm + "/sha")
+
+if os.path.isdir("data/" + args.dimm + "/sha/" + args.temperature + "C"):
+  bitstreams_available = True
+
+if not os.path.isdir("data/" + args.dimm + "/sha/" + args.temperature + "C"):
+  os.mkdir("data/" + args.dimm + "/sha/" + args.temperature + "C")
+
+infile = os.path.normpath("data/" + args.dimm + "/ent/" + args.temperature + "C/top.txt")
+outdir = os.path.normpath("data/" + args.dimm + "/sha/" + args.temperature + "C")
+
+if not os.path.isdir("data/" + args.dimm + "/sha"):
+  os.mkdir("data/" + args.dimm + "/sha_nist_results")
+
+if not os.path.isdir("data/" + args.dimm + "/sha/" + args.temperature + "C"):
+  os.mkdir("data/" + args.dimm + "/sha_nist_results/" + args.temperature + "C")
+
+sha_output = os.path.normpath("data/" + args.dimm + "/sha_nist_results/" + args.temperature + "C")
+
+if not bitstreams_available:
+  sha_path = "bin/SHA"
+  print("sudo " + sha_path + " " + outdir + " " + infile)
+  os.system("sudo " + sha_path + " " + outdir + " " + infile)
+else:
+  print("I think we have the bitstreams available, I won't run softmc code again")
+
+i=0
+bitstream = []
+
+for j in range(0,10):
+  bitstream.append([])
+
+for fname in os.listdir(outdir):
+  bitstream[i].append(fname)
+  i+=1
+  i = i % 10
+
+print(bitstream)
+
+for i in range(len(bitstream[9])):
+  print("Finished " + str(i) + " out of " + str(len(bitstream[9])))
+  processes = []
+  for thr in range(0,10):
+    outpath = os.path.normpath(sha_output+'/thr'+str(thr))
+    if not os.path.isdir(outpath):
+      os.mkdir(outpath)
+    command = 'tools/sts -O -i 1024 -I 1 -P 11=0.001 -S '+ str(1024*1024) +' -w '+ outpath +' -F r ' + outdir+'/'+bitstream[thr][i] + " 2>/dev/null"
+    process = subprocess.Popen(command, shell=True)
+    processes.append(process)
+    print(command)
+
+  # Collect statuses
+  output = [p.wait() for p in processes]
+
+  for thr in range(0,10):
+    outpath = os.path.normpath(sha_output+'/thr'+str(thr))
+    res_txt = open(outpath + '/finalAnalysisReport.txt', "r")
+    lines = res_txt.readlines()
+    n_success = 0
+    for j in range(7, len(lines)):
+      line = lines[j].strip()
+      if not ("1" in line or "0" in line):
+        break
+      split = line.split()
+      if "/" in split[11]:
+        n_success += int(split[11].split("/")[0])
+    res_txt.close()
+    print(bitstream[thr][i] + " succeeded in " + str(n_success) + "tests + bits: " + str(1024*1024))
+    if n_success > 186:
+      os.system("mkdir "+sha_output+"/" + bitstream[thr][i])
+      os.system("cp "+sha_output+'/thr'+str(thr)+"/finalAnalysisReport.txt "+sha_output+"/" + bitstream[thr][i])