graphene/Pal/regression/05_Process.py

#!/usr/bin/env python2
## This test is specifically for the reference monitor code, not process creation in general.
## It is not well-tested right now, but keep the tests around for future work.

import os, sys, mmap
from regression import Regression

loader = os.environ['PAL_SEC']

if not os.path.exists(loader):
    print "Reference monitor mode is not available on this platform"
    exit(0)

regression = Regression(loader, "Process")

def check_times(target, lines, times):
    count = 0
    for line in lines:
        if target == line:
            count += 1
    return count == times

regression.add_check(name="Process Creation",
    check=lambda res: check_times("Child Process Created", res[0].log, 3))

regression.add_check(name="Process Creation Arguments",
    check=lambda res: check_times("argv[0] = Process", res[0].log, 3) and
                      check_times("argv[1] = Child",   res[0].log, 3))

regression.add_check(name="Process Channel Transmission",
    check=lambda res: check_times("Process Write 1 OK",            res[0].log, 3) and
                      check_times("Process Read 1: Hello World 1", res[0].log, 3) and
                      check_times("Process Write 2 OK",            res[0].log, 3) and
                      check_times("Process Read 2: Hello World 2", res[0].log, 3))

regression.add_check(name="Multi-Process Broadcast Channel Transmission",
    check=lambda res: check_times("Broadcast Write OK",            res[0].log, 1) and
                      check_times("Broadcast Read: Hello World 1", res[0].log, 3))

rv = regression.run_checks()
## dp : For now, let these tests fail.  We should fix this.
#if rv: sys.exit(rv)

regression = Regression(loader, "Process2")

regression.add_check(name="Process Creation without Executable",
    check=lambda res: check_times("Binary 1 Preloaded", res[0].log, 2) and
                      check_times("Binary 2 Preloaded", res[0].log, 2))

rv = regression.run_checks()
#if rv: sys.exit(rv)