Fixup the case study script

New features:
- Checks arguments
- Checks that we're root first
- Checks if task launches succeed
- Checks that a case study is not already running

Fixes:
- Removes invalid uses of sudo
- Fixes use of ',' rather than ';' in resctrl specification
- Fixes PID determination
- Always launches Level-C tasks as asynchronous threads

Additionally converts the whole file to only use spaces and to be
directly executable.

1 files changed, 185 insertions, 89 deletions

diff --git a/run_case_study.py b/run_case_study.py
index bcbaca6..eff06fe 100644..100755
--- a/run_case_study.py
+++ b/run_case_study.py
@@ -1,3 +1,5 @@
+#!/usr/bin/python3
+# Make sure to run this as root!!!!!
 import os
 import sys
 import re
@@ -5,35 +7,78 @@ import csv
 import time
 import subprocess
 
+A_SUITE = 1
+A_BIN = 2
+A_CRIT = 3
+A_PERIOD = 5
+
+P_CORE = 0
+P_CRIT = 1
+P_T1_ID = 2
+P_T2_ID = 5
+
+T_ID = 0
+T_BIN = 1
+T_PERIOD = 2
+T_CCX = 3
+
+CRIT_LEVEL_C = 2
+
 all_pids=[]
 def run(command):
     print(command)
     os.system(command)
+
 def addpid(pid):
     with open("./pids.txt", "a") as f:
         f.write(str(pid) + "\n")
+def ID2PID(lookupID):
+    res = None
+    # Find benchmark PID (avoid getting shell or numactl PID)
+    try:
+        res = subprocess.check_output("pgrep -af " + lookupID + " | grep -v numactl | cut -d ' ' -f 1", shell=True)
+    except:
+        res = None
+    if res:
+        # Ignore the newline
+        return res.decode("utf-8").strip()
+    else:
+        return None
 
-
-def main():
+def main(pathName):
     with open("./pids.txt", "w") as f:
         f.write("")
 
-    pathName = sys.argv[1]
     tacle_pairs_path = "./all_pairs"
     tacle_baseline_path = "./baseline"
     dis_path = "./dis"
+    # List of lists
+    # Each list represents the parameters for a task as:
+    # [id, suite, binary name, criticality level (0-base), pet us (unused), period ms, wss (unused)]
     all_tasks = []
+    # List of lists
+    # Each list represents the parameters for a pairing as:
+    # [core, criticality level (0-base), task 1 ID, task 1 name (unused), task 1 period, task 2 ID, task 2 name, task 2 period]
+    # if task 1 ID = task 2 ID, it's not actually a pair at all
     levelAB = []
     levelC = []
     levelC_s = []
-    dis_inputs = dict()
-    dis_inputs["field"] = "./gen_input.py field inputs/Field/in0 2097152"
-    dis_inputs["matrix"] = "gen_input.py matrix inputs/Matrix/in0 2097152"
-    dis_inputs["neighborhood"] = "./gen_input.py neighborhood inputs/Neighborhood/in0 2097152"
-    dis_inputs["pointer"] = "./gen_input.py pointer inputs/Pointer/in0 2097152"
-    dis_inputs["transitive"] = "./gen_input.py transitive inputs/Transitive/in0 2097152"
-    dis_inputs["update"] = "./gen_input.py update inputs/Update/in0 2097152"
+    input_cmd = {}
+    # Load input command for each binary
+    with open("baseline/tacleNames.txt") as f:
+        for benchName in f:
+            # TACLeBench doesn't need inputs
+            input_cmd[benchName.strip()] = "echo ' '"
+    with open("SD-VBS/sd-vbsNames.txt") as f:
+        for line in f:
+            name, cmd = line.split(maxsplit=1)
+            input_cmd[name.replace("./", "")] = cmd.replace("./", "./SD-VBS/").strip()
+    with open("dis/dis2MbInNames.txt") as f:
+        for line in f:
+            name, cmd = line.split(maxsplit=1)
+            input_cmd[name.replace("./", "")] = cmd.replace("./", "./dis/").strip()
 
+    # Load task specifications
     with open(pathName+"/all_tasks.csv", "r") as csv_file:
         csv_reader = csv.reader(csv_file, delimiter = ',')
         line = 0
@@ -56,6 +101,7 @@ def main():
         for row in csv_reader:
             if(len(row) < 3):
                 continue
+            # Start of a cluster - read definition
             if(row[0] == "threaded" or row[0] == "solo"):
                 threaded = row[0] == "threaded"
                 if(row[1] == "4"):
@@ -63,6 +109,7 @@ def main():
                 else:
                     ccx = "0"
                 continue
+            # Start of a thread speficication
             if(row[0] != "task id" and row[0] != ""):
                 row.append(ccx)
                 if(threaded):
@@ -71,9 +118,9 @@ def main():
                     row.append("s")
                 levelC.append(row)
             line += 1
+    ### Allocate Cache Ways ###
     run('mount -t resctrl resctrl /sys/fs/resctrl')
-    run('sudo echo "L3:0=0000,1=0000,2=0000,3=0000" > sudo /sys/fs/resctrl/schemata')
-
+#    run('echo "L3:0=0000,1=0000,2=0000,3=0000" > /sys/fs/resctrl/schemata')
     with open(pathName+"/l3alloc.csv", "r") as csv_file:
         csv_reader = csv.reader(csv_file, delimiter = ',')
         line = 0
@@ -86,109 +133,158 @@ def main():
                 # Level C allocation
                 if(core == 1):
                     run("mkdir -p /sys/fs/resctrl/level-c-ccx-0")
-                    run('sudo echo "L3:0=' + ("%0.4x" % (int("1"*way_C, 2))) + ',1=0000,2=0000,3=0000" > sudo /sys/fs/resctrl/level-c-ccx-0/schemata')
+                    run('echo "L3:0=' + ("%0.4x" % (int("1"*way_C, 2))) + ';1=0000;2=0000;3=0000" > /sys/fs/resctrl/level-c-ccx-0/schemata')
 
                 if(core == 4):
                     run("mkdir -p /sys/fs/resctrl/level-c-ccx-1")
-                    run('sudo echo "L3:0=0000, 1=' + ("%0.4x" % (int("1"*way_C, 2))) + ',2=0000,3=0000" > sudo /sys/fs/resctrl/level-c-ccx-1/schemata')
+                    run('echo "L3:0=0000;1=' + ("%0.4x" % (int("1"*way_C, 2))) + ';2=0000;3=0000" > /sys/fs/resctrl/level-c-ccx-1/schemata')
                 run('mkdir -p /sys/fs/resctrl/level-ab-core-' + str(core))
                 if(core < 4):
-                    run('sudo echo "L3:0=' + ("%0.4x" % (int("1"*way_AB + "0"*(16-way_AB), 2))) + ',1=0000,2=0000,3=0000" > sudo /sys/fs/resctrl/level-ab-core-' + str(core) + '/schemata')
+                    run('echo "L3:0=' + ("%0.4x" % (int("1"*way_AB + "0"*(16-way_AB), 2))) + ';1=0000;2=0000;3=0000" > /sys/fs/resctrl/level-ab-core-' + str(core) + '/schemata')
                 else:
-                    run('sudo echo "L3:0=0000,1=' + ("%0.4x" % (int("1"*way_AB + "0"*(16-way_AB), 2))) + ',2=0000,3=0000" > sudo /sys/fs/resctrl/level-ab-core-' + str(core) + '/schemata')
+                    run('echo "L3:0=0000;1=' + ("%0.4x" % (int("1"*way_AB + "0"*(16-way_AB), 2))) + ';2=0000;3=0000" > /sys/fs/resctrl/level-ab-core-' + str(core) + '/schemata')
             line += 1
-    run("sudo rm -rf /dev/shm/*")
+    ### Cleanup Old State ###
+    run("rm -rf /dev/shm/*")
     pairID = 0
     i = 0
-    for task in levelAB:
-        task1 = int(task[2])
-        task2 = int(task[5])
-        name1 = all_tasks[task1][2]
-        name2 = all_tasks[task2][2]
-        if(name1 == "rijndael_e"):
-            name1 = "rijndael_enc"
-
-        if(name2 == "rijndael_e"):
-            name2 = "rijndael_enc"
-        period1 = all_tasks[task1][5]
-        period2 = all_tasks[task2][5]
-        suite1 = all_tasks[task1][1]
-        suite2 = all_tasks[task2][1]
-        core = task[0]
-        criticality = task[1]
+    ### Dispatch all the Level-A and -B paired and unpaired tasks ###
+    for pairing in levelAB:
+        if i > 2:
+            break
+        i += 1
+
+        task1 = int(pairing[P_T1_ID])
+        task2 = int(pairing[P_T2_ID])
+        # Lookup periods and binary location
+        name1 = all_tasks[task1][A_BIN]
+        name2 = all_tasks[task2][A_BIN]
+        period1 = all_tasks[task1][A_PERIOD]
+        period2 = all_tasks[task2][A_PERIOD]
+        suite1 = all_tasks[task1][A_SUITE]
+        suite2 = all_tasks[task2][A_SUITE]
+        core = pairing[P_CORE]
+        criticality = pairing[P_CRIT]
         binary = ""
         binary1 = ""
         binary2 = ""
         arg = ""
-        if(task1 == task2):
+        # If the IDs match, this isn't a pair at all