summaryrefslogtreecommitdiff
path: root/sim/gdbstub.py
blob: 914f7179589ac9b297737eb777f661c78e7c473e (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
import sys, socket

cycles = None
enable_tty = True
start_halted = True
sock, client = None, None

def init():
    global sock

    sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
    sock.bind(('127.0.0.1', 1234))
    sock.listen()

def halt():
    return yield_to_gdb()

def fatal():
    global stop_reason
    stop_reason = 'fatal'
    yield_to_gdb()

buffer = b''
stop_reason = None

def yield_to_gdb():
    global client, buffer, stop_reason

    if not client:
        print('Listening for gdb on', sock.getsockname(), file=sys.stderr)

        client, peer = sock.accept()
        print('Accepted connection from', peer, file=sys.stderr)

        register_interrupt(client)
        stop_reason = 'reset'

    dead = stop_reason == 'fatal'
    sendStop = True

    if stop_reason == 'break':
        stop_reply = b'S05'
    elif stop_reason == 'reset':
        stop_reply = b'S05'
        stop_reason = 'break'
        sendStop = False
    elif stop_reason == 'fatal':
        stop_reply = b'S0a'

    if sendStop:
        reply(stop_reply)

    while True:
        data = client.recv(4096)
        if not data:
            break

        buffer = buffer + data if buffer else data

        try:
            start = buffer.index(b'$')
            marker = buffer.index(b'#', start + 1)
        except ValueError:
            continue

        if marker + 2 >= len(buffer):
            continue

        data = buffer[start + 1:marker]
        cksum = int(buffer[marker + 1:marker + 3], 16)

        if cksum != (sum(data) & 0xff):
            raise Exception(f'bad packet checksum: {buffer[start:marker + 3]}')

        buffer = buffer[marker + 3:]

        client.send(b'+')

        if data == b'?':
            out = stop_reply
        elif data == b'c' and not dead:
            return 'continue'
        elif data == b'D':
            out = b'OK'
        elif data == b'g':
            out = hexout(read_reg(gdb_reg(r)) for r in range(16))
        elif data[0] == b'm'[0]:
            addr, length = (int(x, 16) for x in data[1:].split(b','))
            out = hexout(read_mem(addr, length))
        elif data[0] == b'M'[0]:
            addrlen, data = data[1:].split(b':')
            addr, length = (int(x, 16) for x in addrlen.split(b','))

            data = bytes.fromhex(str(data, 'ascii'))
            assert len(data) == length

            write_mem(addr, data)
            out = b'OK'
        elif data[0] == b'p'[0]:
            reg = gdb_reg(int(data[1:], 16))
            out = hexout(read_reg(reg) if reg is not None else None)
        elif data == b's' and not dead:
            return 'step'
        else:
            out = b''

        reply(out)

def reply(out):
    client.send(b'$' + out + b'#' + hexout(sum(out) & 0xff, 1))

def gdb_reg(n):
    if 0 <= n < 8:
        return (r0, r1, r2, r3, r4, r5, r6, r7)[n]

    if n == 15:
        return pc

    if n == 0x19:
        return cpsr

    mode = read_reg(cpsr) & 0b11111
    if 8 <= n < 13:
        if mode == 0b10001:
            regs = (r8_fiq, r9_fiq, r10_fiq, r11_fiq, r12_fiq)
        else:
            regs = (r8_usr, r9_usr, r10_usr, r11_usr, r12_usr)

        return regs[n - 8]

    if 13 <= n < 15:
        if mode == 0b10011:
            regs = (r13_svc, r14_svc)
        elif mode == 0b10111:
            regs = (r13_abt, r14_abt)
        elif mode == 0b11011:
            regs = (r13_und, r14_und)
        elif mode == 0b10010:
            regs = (r13_irq, r14_irq)
        elif mode == 0b10001:
            regs = (r13_fiq, r14_fiq)
        else:
            regs = (r13_usr, r14_usr)

        return regs[n - 13]

    print('bad gdb regnum:', n, file=sys.stderr)
    return None

def hexout(data, size=4):
    if data is None:
        return b''
    elif type(data) is bytes:
        return data.hex().encode('ascii')
    elif type(data) is int:
        return data.to_bytes(size, 'little').hex().encode('ascii')

    return b''.join(hexout(d) for d in data)