mirror of
https://github.com/cea-sec/miasm
synced 2026-06-21 13:48:18 +00:00
ajax
473f60bbd7
- cosmetics: display functions - monothread: callbacks for basic test run view - screendisplay: callbacks for enhanced test run view - test: describe a Test case - testset: describe a Test set and implement methods to run it
116 lines
3.5 KiB
Python
116 lines
3.5 KiB
Python
import time
|
|
import signal
|
|
from cosmetics import getTerminalSize, colors
|
|
|
|
|
|
global_state = {"termSize": getTerminalSize(),
|
|
"message": "",
|
|
"pstate": []}
|
|
|
|
|
|
def print_conf(conf, value):
|
|
"Print a configuration line"
|
|
return colors["green"] + conf + ": " + colors["end"] + str(value)
|
|
|
|
|
|
def clr_screen():
|
|
"Update the screen to display some information"
|
|
|
|
# Header
|
|
to_print = []
|
|
to_print.append(" " * (global_state["termSize"][0] / 2 - 10) + colors[
|
|
"blue"] + "Miasm2 Regression tests" + colors["end"])
|
|
to_print.append("")
|
|
to_print.append("=" * global_state["termSize"][0])
|
|
to_print.append("")
|
|
to_print.append(print_conf("Current mode", "Multiprocessing"))
|
|
to_print.append(print_conf("Nb CPU detected", global_state["cpu_c"]))
|
|
to_print.append("")
|
|
to_print.append("=" * global_state["termSize"][0])
|
|
to_print.append("")
|
|
test_done = 0
|
|
test_failed = 0
|
|
message = global_state["message"] + "\n"
|
|
for v in global_state["pstate"]:
|
|
if v["status"] != "running":
|
|
test_done += 1
|
|
if v["status"] != 0:
|
|
test_failed += 1
|
|
cmd_line = " ".join(v["test"].command_line)
|
|
message += colors["red"] + "FAIL:" + colors["end"] + cmd_line
|
|
message += "\n" + v["message"] + "\n"
|
|
|
|
to_print.append(print_conf("Success rate", "%d/%d" %
|
|
(test_done - test_failed, test_done)))
|
|
printed_time = time.strftime(
|
|
"%M:%S", time.gmtime(time.time() - global_state["init_time"]))
|
|
to_print.append(print_conf("Cumulated time", printed_time))
|
|
to_print.append("")
|
|
to_print.append("=" * global_state["termSize"][0])
|
|
|
|
cur = "\n".join(to_print)
|
|
cur += "\n"
|
|
|
|
# Message
|
|
cur += message
|
|
print cur
|
|
already_printed = cur.count("\n")
|
|
|
|
# Current state
|
|
current_job = []
|
|
for process in global_state["pstate"]:
|
|
if process["status"] == "running":
|
|
current_job.append(process)
|
|
print "\n" * (global_state["termSize"][1] - already_printed - 3 - len(current_job))
|
|
|
|
for job in current_job:
|
|
command_line = " ".join(job["test"].command_line)
|
|
base_dir = job["test"].base_dir.upper()
|
|
s = "[" + colors["lightcyan"] + command_line + colors["end"]
|
|
s_end = base_dir
|
|
cur_time = time.strftime(
|
|
"%M:%Ss", time.gmtime(time.time() - job["init_time"]))
|
|
l = len(command_line) + len(s_end) + 4 + len(str(cur_time)) + 2
|
|
s_end += " " + colors["blue"] + cur_time + colors["end"] + "]"
|
|
print "%s%s%s" % (s, " " * (global_state["termSize"][0] - l), s_end)
|
|
|
|
|
|
def on_signal(sig1, sig2):
|
|
"Update view every second"
|
|
clr_screen()
|
|
signal.alarm(1)
|
|
|
|
|
|
def init(cpu_c):
|
|
"""Initialize global state
|
|
@cpu_c: number of cpu (for conf displaying)
|
|
"""
|
|
# Init global_state
|
|
global_state["cpu_c"] = cpu_c
|
|
global_state["init_time"] = time.time()
|
|
|
|
# Launch view updater
|
|
signal.signal(signal.SIGALRM, on_signal)
|
|
signal.alarm(1)
|
|
|
|
|
|
def task_done(test, error):
|
|
"Report a test done"
|
|
for task in global_state["pstate"]:
|
|
if task["test"] == test:
|
|
if error is not None:
|
|
task["status"] = -1
|
|
task["message"] = error
|
|
else:
|
|
task["status"] = 0
|
|
break
|
|
clr_screen()
|
|
|
|
|
|
def task_new(test):
|
|
"Report a new test"
|
|
global_state["pstate"].append({"status": "running",
|
|
"test": test,
|
|
"init_time": time.time()})
|
|
clr_screen()
|