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
):
20 """Compile a single .c file into a binary."""
21 dst
= os
.path
.splitext(args
[0])[0]
22 cc
= os
.path
.expandvars("$RISCV/bin/riscv64-unknown-elf-gcc")
23 cmd
= [cc
, "-g", "-O", "-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
, binary
, halted
=False, with_gdb
=True, timeout
=None):
46 """Launch spike. Return tuple of its process and the port it's running on."""
49 cmd
+= ["timeout", str(timeout
)]
51 cmd
+= [find_file("spike")]
55 self
.port
= unused_port()
56 cmd
+= ['--gdb-port', str(self
.port
)]
60 logfile
= open("spike.log", "w")
61 self
.process
= subprocess
.Popen(cmd
, stdin
=subprocess
.PIPE
, stdout
=logfile
,
71 def wait(self
, *args
, **kwargs
):
72 return self
.process
.wait(*args
, **kwargs
)
74 class Openocd(object):
75 def __init__(self
, cmd
=None, config
=None, debug
=True):
77 cmd
= shlex
.split(cmd
)
81 cmd
+= ["-f", find_file(config
)]
84 logfile
= open("openocd.log", "w")
85 self
.process
= subprocess
.Popen(cmd
, stdin
=subprocess
.PIPE
, stdout
=logfile
,
87 # TODO: Pick a random port
99 path
= os
.path
.expandvars("$RISCV/bin/riscv64-unknown-elf-gdb")
100 self
.child
= pexpect
.spawn(path
)
101 self
.child
.logfile
= file("gdb.log", "w")
103 self
.command("set width 0")
104 self
.command("set height 0")
106 self
.command("set print entry-values no")
109 """Wait for prompt."""
110 self
.child
.expect("\(gdb\)")
112 def command(self
, command
, timeout
=-1):
113 self
.child
.sendline(command
)
114 self
.child
.expect("\n", timeout
=timeout
)
115 self
.child
.expect("\(gdb\)", timeout
=timeout
)
116 return self
.child
.before
.strip()
118 def c(self
, wait
=True):
120 return self
.command("c")
122 self
.child
.sendline("c")
123 self
.child
.expect("Continuing")
126 self
.child
.send("\003");
127 self
.child
.expect("\(gdb\)")
129 def x(self
, address
, size
='w'):
130 output
= self
.command("x/%s %s" % (size
, address
))
131 value
= int(output
.split(':')[1].strip(), 0)
135 output
= self
.command("p %s" % obj
)
136 value
= int(output
.split('=')[-1].strip())
140 return self
.command("stepi")