41ef169096ead466fb67de0efcc7393cc1f9c056
1 from pygdbmi
.gdbcontroller
import GdbController
4 launch_args_be
= ['qemu-system-ppc64',
5 '-machine', 'powernv9',
9 launch_args_le
= ['qemu-system-ppc64le',
10 '-machine', 'powernv9',
14 def swap_order(x
, nbytes
):
15 x
= x
.to_bytes(nbytes
, byteorder
='little')
16 x
= int.from_bytes(x
, byteorder
='big', signed
=False)
21 def __init__(self
, kernel
, bigendian
):
23 args
= launch_args_be
+ ['-kernel', kernel
]
25 args
= launch_args_le
+ ['-kernel', kernel
]
26 self
.qemu_popen
= subprocess
.Popen(args
,
27 stdout
=subprocess
.PIPE
,
28 stdin
=subprocess
.PIPE
)
29 self
.gdb
= GdbController(gdb_path
='powerpc64-linux-gnu-gdb')
30 self
.bigendian
= bigendian
35 def __exit__(self
, type, value
, traceback
):
39 return self
.gdb
.write('-target-select remote localhost:1234')
41 def set_endian(self
, bigendian
):
43 cmd
= '-gdb-set endian big'
45 cmd
= '-gdb-set endian little'
46 return self
.gdb
.write(cmd
)
48 def break_address(self
, addr
):
49 cmd
= '-break-insert *0x{:x}'.format(addr
)
50 return self
.gdb
.write(cmd
)
52 def delete_breakpoint(self
, breakpoint
=None):
55 breakstring
= f
' {breakpoint}'
56 return self
.gdb
.write('-break-delete' + breakstring
)
58 def set_byte(self
, addr
, v
):
59 print ("qemu set byte", hex(addr
), hex(v
))
60 faddr
= '&{int}0x%x' % addr
61 res
= self
.gdb
.write('-data-write-memory-bytes %s "%02x"' % (faddr
, v
))
62 print ("confirm", self
.get_mem(addr
, 1))
64 def get_mem(self
, addr
, nbytes
):
65 res
= self
.gdb
.write("-data-read-memory %d u 1 1 %d" % (addr
, 8*nbytes
))
66 #print ("get_mem", res)
68 if(x
["type"]=="result"):
69 l
= list(map(int, x
['payload']['memory'][0]['data']))
71 for j
in range(0, len(l
), 8):
73 for i
, v
in enumerate(l
[j
:j
+8]):
79 def get_registers(self
):
80 return self
.gdb
.write('-data-list-register-values x')
82 def _get_register(self
, fmt
):
83 res
= self
.gdb
.write('-data-list-register-values '+fmt
,
84 timeout_sec
=1.0) # increase this timeout if needed
86 if(x
["type"]=="result"):
87 assert 'register-values' in x
['payload']
88 res
= int(x
['payload']['register-values'][0]['value'], 0)
90 #return swap_order(res, 8)
93 # TODO: use -data-list-register-names instead of hardcoding the values
94 def get_pc(self
): return self
._get
_register
('x 64')
95 def get_msr(self
): return self
._get
_register
('x 65')
96 def get_cr(self
): return self
._get
_register
('x 66')
97 def get_lr(self
): return self
._get
_register
('x 67')
98 def get_ctr(self
): return self
._get
_register
('x 68') # probably
99 def get_xer(self
): return self
._get
_register
('x 69')
100 def get_fpscr(self
): return self
._get
_register
('x 70')
101 def get_mq(self
): return self
._get
_register
('x 71')
102 def get_register(self
, num
):
103 return self
._get
_register
('x {}'.format(num
))
106 return self
.gdb
.write('-exec-next-instruction')
108 def gdb_continue(self
):
109 return self
.gdb
.write('-exec-continue')
111 def gdb_eval(self
, expr
):
112 return self
.gdb
.write(f
'-data-evaluate-expression {expr}')
116 self
.qemu_popen
.kill()
117 outs
, errs
= self
.qemu_popen
.communicate()
118 self
.qemu_popen
.stdout
.close()
119 self
.qemu_popen
.stdin
.close()
122 def run_program(program
, initial_mem
=None, extra_break_addr
=None,
124 q
= QemuController(program
.binfile
.name
, bigendian
)
126 q
.set_endian(True) # easier to set variables this way
128 # Run to the start of the program
130 for addr
, (v
, wid
) in initial_mem
.items():
132 q
.set_byte(addr
+i
, (v
>>i
*8) & 0xff)
134 # set breakpoint at start
135 q
.break_address(0x20000000)
137 # set the MSR bit 63, to set bigendian/littleendian mode
139 print ("msr", bigendian
, hex(msr
))
142 msr
= msr
& ((1<<64)-1)
145 q
.gdb_eval('$msr=%d' % msr
)
146 print ("msr set to", hex(msr
))
147 # set the CR to 0, matching the simulator
149 # delete the previous breakpoint so loops don't screw things up
150 q
.delete_breakpoint()
152 q
.break_address(0x20000000 + program
.size())
154 q
.break_address(0x700)
155 # or to alternative (absolute) address)
157 q
.break_address(extra_break_addr
)
159 q
.set_endian(bigendian
)
164 if __name__
== '__main__':
165 q
= QemuController("simulator/qemu_test/kernel.bin", bigendian
=True)
167 q
.break_address(0x20000000)
169 print(q
.get_register(1))
171 print(q
.get_register(1))