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,
47 """Launch spike. Return tuple of its process and the port it's running on."""
49 cmd
= shlex
.split(cmd
)
53 cmd
+= ["--isa", "RV32"]
56 cmd
= ["timeout", str(timeout
)] + cmd
61 self
.port
= unused_port()
62 cmd
+= ['--gdb-port', str(self
.port
)]
66 logfile
= open("spike.log", "w")
67 logfile
.write("+ %s\n" % " ".join(cmd
))
68 self
.process
= subprocess
.Popen(cmd
, stdin
=subprocess
.PIPE
, stdout
=logfile
,
78 def wait(self
, *args
, **kwargs
):
79 return self
.process
.wait(*args
, **kwargs
)
81 class Openocd(object):
82 def __init__(self
, cmd
=None, config
=None, debug
=True):
84 cmd
= shlex
.split(cmd
)
88 cmd
+= ["-f", find_file(config
)]
91 logfile
= open("openocd.log", "w")
92 logfile
.write("+ %s\n" % " ".join(cmd
))
93 self
.process
= subprocess
.Popen(cmd
, stdin
=subprocess
.PIPE
, stdout
=logfile
,
95 # TODO: Pick a random port
107 path
=os
.path
.expandvars("$RISCV/bin/riscv64-unknown-elf-gdb")):
108 self
.child
= pexpect
.spawn(path
)
109 self
.child
.logfile
= file("gdb.log", "w")
110 self
.child
.logfile
.write("+ %s\n" % path
)
112 self
.command("set confirm off")
113 self
.command("set width 0")
114 self
.command("set height 0")
116 self
.command("set print entry-values no")
119 """Wait for prompt."""
120 self
.child
.expect("\(gdb\)")
122 def command(self
, command
, timeout
=-1):
123 self
.child
.sendline(command
)
124 self
.child
.expect("\n", timeout
=timeout
)
125 self
.child
.expect("\(gdb\)", timeout
=timeout
)
126 return self
.child
.before
.strip()
128 def c(self
, wait
=True):
130 output
= self
.command("c")
131 assert "Continuing" in output
134 self
.child
.sendline("c")
135 self
.child
.expect("Continuing")
138 self
.child
.send("\003");
139 self
.child
.expect("\(gdb\)")
140 return self
.child
.before
.strip()
142 def x(self
, address
, size
='w'):
143 output
= self
.command("x/%s %s" % (size
, address
))
144 value
= int(output
.split(':')[1].strip(), 0)
148 output
= self
.command("p/x %s" % obj
)
149 value
= int(output
.split('=')[-1].strip(), 0)
153 output
= self
.command("stepi")
154 assert "Cannot" not in output
158 output
= self
.command("load", timeout
=60)
159 assert "failed" not in output
160 assert "Transfer rate" in output
162 def b(self
, location
):
163 output
= self
.command("b %s" % location
)
164 assert "not defined" not in output
165 assert "Breakpoint" in output
168 def hbreak(self
, location
):
169 output
= self
.command("hbreak %s" % location
)
170 assert "not defined" not in output
171 assert "Hardware assisted breakpoint" in output