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
)
51 #cmd.append("-l") #<<<
54 cmd
= ["timeout", str(timeout
)] + cmd
59 self
.port
= unused_port()
60 cmd
+= ['--gdb-port', str(self
.port
)]
64 logfile
= open("spike.log", "w")
65 logfile
.write("+ %s\n" % " ".join(cmd
))
66 self
.process
= subprocess
.Popen(cmd
, stdin
=subprocess
.PIPE
, stdout
=logfile
,
76 def wait(self
, *args
, **kwargs
):
77 return self
.process
.wait(*args
, **kwargs
)
79 class Openocd(object):
80 def __init__(self
, cmd
=None, config
=None, debug
=True):
82 cmd
= shlex
.split(cmd
)
86 cmd
+= ["-f", find_file(config
)]
89 logfile
= open("openocd.log", "w")
90 self
.process
= subprocess
.Popen(cmd
, stdin
=subprocess
.PIPE
, stdout
=logfile
,
92 # TODO: Pick a random port
104 path
= os
.path
.expandvars("$RISCV/bin/riscv64-unknown-elf-gdb")
105 self
.child
= pexpect
.spawn(path
)
106 self
.child
.logfile
= file("gdb.log", "w")
108 self
.command("set width 0")
109 self
.command("set height 0")
111 self
.command("set print entry-values no")
114 """Wait for prompt."""
115 self
.child
.expect("\(gdb\)")
117 def command(self
, command
, timeout
=-1):
118 self
.child
.sendline(command
)
119 self
.child
.expect("\n", timeout
=timeout
)
120 self
.child
.expect("\(gdb\)", timeout
=timeout
)
121 return self
.child
.before
.strip()
123 def c(self
, wait
=True):
125 output
= self
.command("c")
126 assert "Continuing" in output
129 self
.child
.sendline("c")
130 self
.child
.expect("Continuing")
133 self
.child
.send("\003");
134 self
.child
.expect("\(gdb\)")
135 return self
.child
.before
.strip()
137 def x(self
, address
, size
='w'):
138 output
= self
.command("x/%s %s" % (size
, address
))
139 value
= int(output
.split(':')[1].strip(), 0)
143 output
= self
.command("p/x %s" % obj
)
144 value
= int(output
.split('=')[-1].strip(), 0)
148 return self
.command("stepi")
151 output
= self
.command("load")
152 assert "failed" not in output
153 assert "Transfer rate" in output
155 def b(self
, location
):
156 output
= self
.command("b %s" % location
)
157 assert "not defined" not in output
158 assert "Breakpoint" in output