#!/usr/bin/python import os, sys, mmap from regression import Regression loader = os.environ['PAL_LOADER'] try: sgx = os.environ['SGX_RUN'] except KeyError: sgx = False def manifest_file(file): if 'SGX_RUN' in os.environ and os.environ['SGX_RUN'] == '1': return file + '.manifest.sgx' else: return file + '.manifest' # Running Bootstrap regression = Regression(loader, "Bootstrap") regression.add_check(name="Basic Bootstrapping", check=lambda res: "User Program Started" in res[0].log) regression.add_check(name="Control Block: Executable Name", check=lambda res: "Loaded Executable: file:Bootstrap" in res[0].log) regression.add_check(name="One Argument Given", check=lambda res: "# of Arguments: 1" in res[0].log and \ "argv[0] = file:Bootstrap" in res[0].log) regression.add_check(name="Five Arguments Given", args = ['a', 'b', 'c', 'd'], check=lambda res: "# of Arguments: 5" in res[0].log and \ "argv[1] = a" in res[0].log and "argv[2] = b" in res[0].log and \ "argv[3] = c" in res[0].log and "argv[4] = d" in res[0].log) regression.add_check(name="Control Block: Debug Stream (Inline)", check=lambda res: "Written to Debug Stream" in res[0].out) regression.add_check(name="Control Block: Page Size", check=lambda res: ("Page Size: %d" % (mmap.PAGESIZE)) in res[0].log) regression.add_check(name="Control Block: Allocation Alignment", check=lambda res: ("Allocation Alignment: %d" % (mmap.ALLOCATIONGRANULARITY)) in res[0].log) regression.add_check(name="Control Block: Executable Range", check=lambda res: "Executable Range OK" in res[0].log) def check_cpu_info(res): cpu_num = cpu_model = cpu_family = cpu_stepping = 0 cpu_vendor = cpu_brand = cpu_flags = None f = open("/proc/cpuinfo", "r") for line in f: line = line.strip() pos = line.find(":") if pos == -1: continue key = line[:pos].strip() val = line[pos+1:].strip() if key == "processor": cpu_num += 1 if key == "vendor_id": cpu_vendor = val if key == "cpu family": cpu_family = int(val) if key == "model": cpu_model = int(val) if key == "model name": cpu_brand = val if key == "stepping": cpu_stepping = int(val) if key == "flags": cpu_flags = [] for flag in val.split(" "): if flag in ["fpu", "vme", "de", "pse", "tsc", "msr", "pae", "mce", "cx8", "apic", "sep", "mtrr", "pge", "mca", "cmov", "pat", "pse36", "pn", "clflush", "dts", "acpi", "mmx", "fxsr", "sse", "sse2", "ss", "ht", "tm", "ia64", "pbe"]: cpu_flags.append(flag) cpu_flags = " ".join(cpu_flags) return ("CPU num: %d" % cpu_num) in res[0].log and \ ("CPU vendor: %s" % cpu_vendor) in res[0].log and \ ("CPU brand: %s" % cpu_brand) in res[0].log and \ ("CPU family: %d" % cpu_family) in res[0].log and \ ("CPU model: %d" % cpu_model) in res[0].log and \ ("CPU stepping: %d" % cpu_stepping) in res[0].log and \ ("CPU flags: %s" % cpu_flags) in res[0].log regression.add_check(name="Control Block: CPU Info", check=check_cpu_info) rv = regression.run_checks() if rv: sys.exit(rv) # Running ..Bootstrap regression = Regression(loader, "..Bootstrap") regression.add_check(name="Dotdot handled properly", check=lambda res: "User Program Started" in res[0].log) rv = regression.run_checks() if rv: sys.exit(rv) # Running Bootstrap2 regression = Regression(loader, "Bootstrap2") regression.add_check(name="Control Block: Manifest as Executable Name", check=lambda res: "Loaded Manifest: file:" + manifest_file("Bootstrap2") in res[0].log) rv = regression.run_checks() if rv: sys.exit(rv) # Running Bootstrap3 regression = Regression(loader, "Bootstrap3") regression.add_check(name="Preload Libraries", check=lambda res: "Binary 1 Preloaded" in res[0].log and "Binary 2 Preloaded" in res[0].log) regression.add_check(name="Preload Libraries Linking", check=lambda res: "Preloaded Function 1 Called" in res[0].log and "Preloaded Function 2 Called" in res[0].log) rv = regression.run_checks() if rv: sys.exit(rv) # Running Bootstrap4 regression = Regression(loader, manifest_file("Bootstrap4")) regression.add_check(name="Control Block: Manifest as Argument", check=lambda res: any([line.startswith("Loaded Manifest: file:" + manifest_file("Bootstrap4")) for line in res[0].log])) regression.add_check(name="Control Block: Executable as in Manifest", check=lambda res: "Loaded Executable: file:Bootstrap" in res[0].log) rv = regression.run_checks() if rv: sys.exit(rv) # Running Bootstrap4.manifest regression = Regression(executable = "./" + manifest_file("Bootstrap4")) regression.add_check(name="Control Block: Manifest as Argument (Load by Shebang)", check=lambda res: "Loaded Manifest: file:" + manifest_file("Bootstrap4") in res[0].log) regression.add_check(name="Control Block: Executable as in Manifest (Load by Shebang)", check=lambda res: "Loaded Executable: file:Bootstrap" in res[0].log) regression.add_check(name="Arguments: loader.execname in Manifest", check=lambda res: "argv[0] = Bootstrap" in res[0].log) rv = regression.run_checks() if rv: sys.exit(rv) # Running Bootstrap5.manifest regression = Regression(loader, manifest_file("Bootstrap5")) regression.add_check(name="Bootstrap without Executable but Preload Libraries", check=lambda res: "Binary 1 Preloaded" in res[0].log and "Binary 2 Preloaded" in res[0].log) rv = regression.run_checks() if rv: sys.exit(rv) # Running Bootstrap6.manifest - SGX-specific test if sgx: regression = Regression(loader, manifest_file("Bootstrap6"), timeout = 100000) regression.add_check(name="8GB Enclave Creation (SGX Only)", check=lambda res: "Loaded Manifest: file:Bootstrap6.manifest.sgx" in res[0].log and "Executable Range OK" in res[0].log) rv = regression.run_checks() if rv: sys.exit(rv) # Running Bootstrap7.manifest regression = Regression(loader, manifest_file("Bootstrap7")) regression.add_check(name="Load Large Number of Items in Manifest", check=lambda res: "key1000=na" in res[0].log and "key1=na" in res[0].log) rv = regression.run_checks() if rv: sys.exit(rv)