kopia lustrzana https://github.com/bristol-seds/pico-tracker
165 wiersze
5.2 KiB
Python
165 wiersze
5.2 KiB
Python
#!/usr/bin/env python
|
|
|
|
# ------------------------------------------------------------------------------
|
|
# Verification Framework
|
|
# ------------------------------------------------------------------------------
|
|
# This script is inteneded to be run from inside gdb, with gdb already
|
|
# attached to the target - probably this is done in .gdbinit.
|
|
#
|
|
# You should be running gdb like this:
|
|
#
|
|
# arm-none-eabi-gdb -q -x tools/verification/tc/my_tc.py
|
|
#
|
|
#
|
|
# The useful output from this script appears on stderror, sorry about
|
|
# that. Hence you should supress stdout if you don't want loads of
|
|
# info from gdb.
|
|
#
|
|
# Something like this should work:
|
|
# > /dev/null arm-none-eabi-gdb -q -x tools/verfication/tc/my_tc.py
|
|
#
|
|
# ------------------------------------------------------------------------------
|
|
|
|
from __future__ import print_function
|
|
|
|
import gdb
|
|
import re
|
|
import sys
|
|
from time import *
|
|
from colorama import *
|
|
|
|
LINE_LENGTH = 80
|
|
|
|
##### Public Print utilities
|
|
|
|
def printf(string):
|
|
"""All writes go to stderr"""
|
|
print (string, file=sys.stderr)
|
|
|
|
def print_info(string):
|
|
"""Prints an info line"""
|
|
printf("")
|
|
printf(Fore.CYAN + "INFO " + Fore.RESET + string)
|
|
|
|
##### Tester
|
|
|
|
class samd20_test:
|
|
|
|
def print_centre(self, string):
|
|
"""Prints something in the centre of the line"""
|
|
count = (LINE_LENGTH - len(string)) / 2
|
|
printf ((" " * count) + string)
|
|
|
|
def print_header_line(self):
|
|
"""Prints a yellow line. Yo"""
|
|
printf (Fore.YELLOW + ("*" * LINE_LENGTH) + Fore.RESET)
|
|
|
|
def print_header(self, string):
|
|
"""Prints a pretty header"""
|
|
printf ("")
|
|
self.print_header_line()
|
|
self.print_centre(string)
|
|
self.print_header_line()
|
|
|
|
def print_pass(self, tc_name, time):
|
|
"""Nice green pass notice"""
|
|
offset = (LINE_LENGTH / 2) - len(tc_name)
|
|
|
|
printf("")
|
|
printf(Fore.GREEN + " " + tc_name + " - PASS" \
|
|
+ (" " * offset) + str(time) + Fore.RESET)
|
|
|
|
def print_fail(self, tc_name, time):
|
|
"""Evil red pass notice"""
|
|
|
|
printf("")
|
|
printf(Fore.RED + " " + tc_name + "- FAIL" \
|
|
+ str(time) + Fore.RESET)
|
|
|
|
def print_summary(self, results):
|
|
passes = 0
|
|
for result in results:
|
|
(passed) = result
|
|
if passed:
|
|
passes += 1
|
|
|
|
|
|
self.print_header("SUMMARY - %d%% PASS"%(100*passes/len(results)))
|
|
|
|
for result in results:
|
|
(passed, name, time) = result
|
|
if passed:
|
|
self.print_pass(name, time)
|
|
else:
|
|
self.print_fail(name, time)
|
|
|
|
self.print_header("")
|
|
#### GDB
|
|
|
|
def __init__(self):
|
|
self.inferior = gdb.selected_inferior()
|
|
|
|
# Load everything into gdb and run
|
|
gdb.execute("load")
|
|
gdb.execute("b main")
|
|
gdb.execute("run")
|
|
|
|
# Stopped at the top of main. Go to tc_main
|
|
gdb.execute("del 1")
|
|
gdb.execute("b tc_main")
|
|
gdb.execute("set $pc=tc_main")
|
|
gdb.execute("c")
|
|
|
|
def __del__(self):
|
|
print_info("quit")
|
|
gdb.execute("quit")
|
|
|
|
|
|
def hw_run_tc(self, tc_name, parameters):
|
|
"""Runs a test case on hardware"""
|
|
|
|
# Write the parameters
|
|
self.write_varible(tc_name+"_params", parameters)
|
|
|
|
# Presuming there"s a breakpoint at the top of tc_main
|
|
#gdb.execute("set $lr=tc_main")
|
|
gdb.execute("set tc_ptr="+tc_name+"+1")
|
|
gdb.execute("c")
|
|
|
|
# Test case done. Return results
|
|
return self.read_variable(tc_name+"_results")
|
|
|
|
#### Read / Write
|
|
|
|
def read_variable(self, name):
|
|
gdb.execute("p " + name, to_string=True)
|
|
return gdb.history(0)
|
|
|
|
def write_varible(self, name, value):
|
|
pvar = self.read_variable(name)
|
|
self.inferior.write_memory(pvar.address, value)
|
|
|
|
#### Test Case
|
|
|
|
def run_test_case(self, test_case):
|
|
tc_name = test_case.__class__.__name__
|
|
self.print_header(tc_name)
|
|
fail = False
|
|
|
|
if test_case.iterations:
|
|
for i in range(test_case.iterations):
|
|
params = test_case.get_test()
|
|
result = self.hw_run_tc(tc_name, params)
|
|
|
|
if not test_case.is_correct(params, result):
|
|
fail = True
|
|
break
|
|
|
|
if not fail:
|
|
self.print_pass(tc_name, 0)
|
|
else:
|
|
self.print_fail(tc_name, 0)
|
|
|
|
# Return data tuple
|
|
return (not fail, tc_name, 0)
|