9 # Note that gdb comes with its own testsuite. I was unable to figure out how to
10 # run that testsuite against the spike simulator.
13 for directory
in (os
.getcwd(), os
.path
.dirname(testlib
.__file
__)):
14 fullpath
= os
.path
.join(directory
, path
)
15 if os
.path
.exists(fullpath
):
19 def compile(args
, xlen
=32):
20 """Compile a single .c file into a binary."""
21 dst
= os
.path
.splitext(args
[0])[0]
22 cc
= os
.path
.expandvars("$RISCV/bin/riscv%d-unknown-elf-gcc" % xlen
)
23 cmd
= [cc
, "-g", "-o", dst
]
25 found
= find_file(arg
)
31 result
= os
.system(cmd
)
32 assert result
== 0, "%r failed" % cmd
36 # http://stackoverflow.com/questions/2838244/get-open-tcp-port-in-python/2838309#2838309
38 s
= socket
.socket(socket
.AF_INET
, socket
.SOCK_STREAM
)
40 port
= s
.getsockname()[1]
45 def __init__(self
, cmd
, binary
=None, halted
=False, with_gdb
=True, timeout
=None):
46 """Launch spike. Return tuple of its process and the port it's running on."""
48 cmd
= shlex
.split(cmd
)
53 cmd
= ["timeout", str(timeout
)] + cmd
58 self
.port
= unused_port()
59 cmd
+= ['--gdb-port', str(self
.port
)]
63 logfile
= open("spike.log", "w")
64 self
.process
= subprocess
.Popen(cmd
, stdin
=subprocess
.PIPE
, stdout
=logfile
,
74 def wait(self
, *args
, **kwargs
):
75 return self
.process
.wait(*args
, **kwargs
)
77 class Openocd(object):
78 def __init__(self
, cmd
=None, config
=None, debug
=True):
80 cmd
= shlex
.split(cmd
)
84 cmd
+= ["-f", find_file(config
)]
87 logfile
= open("openocd.log", "w")
88 self
.process
= subprocess
.Popen(cmd
, stdin
=subprocess
.PIPE
, stdout
=logfile
,
90 # TODO: Pick a random port
102 path
= os
.path
.expandvars("$RISCV/bin/riscv64-unknown-elf-gdb")
103 self
.child
= pexpect
.spawn(path
)
104 self
.child
.logfile
= file("gdb.log", "w")
106 self
.command("set width 0")
107 self
.command("set height 0")
109 self
.command("set print entry-values no")
112 """Wait for prompt."""
113 self
.child
.expect("\(gdb\)")
115 def command(self
, command
, timeout
=-1):
116 self
.child
.sendline(command
)
117 self
.child
.expect("\n", timeout
=timeout
)
118 self
.child
.expect("\(gdb\)", timeout
=timeout
)
119 return self
.child
.before
.strip()
121 def c(self
, wait
=True):
123 output
= self
.command("c")
124 assert "Continuing" in output
127 self
.child
.sendline("c")
128 self
.child
.expect("Continuing")
131 self
.child
.send("\003");
132 self
.child
.expect("\(gdb\)")
133 return self
.child
.before
.strip()
135 def x(self
, address
, size
='w'):
136 output
= self
.command("x/%s %s" % (size
, address
))
137 value
= int(output
.split(':')[1].strip(), 0)
141 output
= self
.command("p/x %s" % obj
)
142 value
= int(output
.split('=')[-1].strip(), 0)
146 return self
.command("stepi")
148 def load(self
, binary
):
149 return self
.command("load %s" % binary
)
151 def b(self
, location
):
152 output
= self
.command("b %s" % location
)
153 assert "not defined" not in output
154 assert "Breakpoint" in output