Fix multi-page flash writing.
[fw/stlink] / src / gdb-server.c
1 /* -*- tab-width:8 -*- */
2
3 /*
4  Copyright (C)  2011 Peter Zotov <whitequark@whitequark.org>
5  Use of this source code is governed by a BSD-style
6  license that can be found in the LICENSE file.
7 */
8
9 #include <stdio.h>
10 #include <string.h>
11 #include <stdlib.h>
12 #include <unistd.h>
13 #include <sys/types.h>
14 #include <sys/socket.h>
15 #include <netinet/in.h>
16 #include <arpa/inet.h>
17 #include "gdb-remote.h"
18 #include "stlink-hw.h"
19
20 static const char hex[] = "0123456789abcdef";
21
22 // configured for STM32F100RB
23 static const char* const c_memory_map =
24   "<?xml version=\"1.0\"?>"
25   "<!DOCTYPE memory-map PUBLIC \"+//IDN gnu.org//DTD GDB Memory Map V1.0//EN\""
26   "     \"http://sourceware.org/gdb/gdb-memory-map.dtd\">"
27   "<memory-map>"
28   "  <memory type=\"rom\" start=\"0x00000000\" length=\"0x20000\"/>"    // code = sram or flash
29   "  <memory type=\"ram\" start=\"0x20000000\" length=\"0x2000\"/>"     // sram 8k
30   "  <memory type=\"flash\" start=\"0x08000000\" length=\"0x20000\">"   // flash 128k
31   "    <property name=\"blocksize\">0x400</property>"                   // 1k pages
32   "  </memory>"
33   "  <memory type=\"ram\" start=\"0x40000000\" length=\"0x1fffffff\"/>" // peripheral regs
34   "  <memory type=\"ram\" start=\"0xe0000000\" length=\"0x1fffffff\"/>" // cortex regs
35   "</memory-map>";
36
37 int serve(struct stlink* sl, int port);
38
39 int main(int argc, char** argv) {
40         if(argc != 3) {
41                 fprintf(stderr, "Usage: %s <port> /dev/sgX\n", argv[0]);
42                 return 1;
43         }
44
45         struct stlink *sl = stlink_quirk_open(argv[2], 0);
46         if (sl == NULL)
47                 return 1;
48
49         if(stlink_current_mode(sl) != STLINK_DEV_DEBUG_MODE)
50                 stlink_enter_swd_mode(sl);
51
52         stlink_core_id(sl);
53         printf("Debugging ARM core %08x.\n", sl->core_id);
54
55         int port = atoi(argv[1]);
56
57         while(serve(sl, port) == 0);
58
59         stlink_close(sl);
60
61         return 0;
62 }
63
64 #define CODE_BREAK_NUM  6
65
66 #define CODE_BREAK_LOW  0x01
67 #define CODE_BREAK_HIGH 0x02
68
69 struct code_hw_breakpoint {
70         stm32_addr_t addr;
71         int          type;
72 };
73
74 struct code_hw_breakpoint code_breaks[CODE_BREAK_NUM];
75
76 static void init_code_breakpoints(struct stlink* sl) {
77         memset(sl->q_buf, 0, 4);
78         sl->q_buf[0] = 0x03; // KEY | ENABLE
79         stlink_write_mem32(sl, 0xe0002000, 4);
80
81         memset(sl->q_buf, 0, 4);
82         for(int i = 0; i < CODE_BREAK_NUM; i++) {
83                 code_breaks[i].type = 0;
84                 stlink_write_mem32(sl, 0xe0002008 + i * 4, 4);
85         }
86 }
87
88 static int update_code_breakpoint(struct stlink* sl, stm32_addr_t addr, int set) {
89         stm32_addr_t fpb_addr = addr & ~0x3;
90         int type = addr & 0x2 ? CODE_BREAK_HIGH : CODE_BREAK_LOW;
91
92         if(addr & 1) {
93                 fprintf(stderr, "update_code_breakpoint: unaligned address %08x\n", addr);
94                 return -1;
95         }
96
97         int id = -1;
98         for(int i = 0; i < CODE_BREAK_NUM; i++) {
99                 if(fpb_addr == code_breaks[i].addr ||
100                         (set && code_breaks[i].type == 0)) {
101                         id = i;
102                         break;
103                 }
104         }
105
106         if(id == -1) {
107                 if(set) return -1; // Free slot not found
108                 else    return 0;  // Breakpoint is already removed
109         }
110
111         struct code_hw_breakpoint* brk = &code_breaks[id];
112
113         brk->addr = fpb_addr;
114
115         if(set) brk->type |= type;
116         else    brk->type &= ~type;
117
118         memset(sl->q_buf, 0, 4);
119
120         if(brk->type == 0) {
121                 #ifdef DEBUG
122                 printf("clearing hw break %d\n", id);
123                 #endif
124
125                 stlink_write_mem32(sl, 0xe0002008 + id * 4, 4);
126         } else {
127                 sl->q_buf[0] = ( brk->addr        & 0xff) | 1;
128                 sl->q_buf[1] = ((brk->addr >> 8)  & 0xff);
129                 sl->q_buf[2] = ((brk->addr >> 16) & 0xff);
130                 sl->q_buf[3] = ((brk->addr >> 24) & 0xff) | (brk->type << 6);
131
132                 #ifdef DEBUG
133                 printf("setting hw break %d at %08x (%d)\n",
134                         id, brk->addr, brk->type);
135                 printf("reg %02x %02x %02x %02x\n",
136                         sl->q_buf[3], sl->q_buf[2], sl->q_buf[1], sl->q_buf[0]);
137                 #endif
138
139                 stlink_write_mem32(sl, 0xe0002008 + id * 4, 4);
140         }
141
142         return 0;
143 }
144
145 #define FLASH_BASE 0x08000000
146 #define FLASH_PAGE 0x400
147 #define FLASH_PAGE_MASK (~((1 << 10) - 1))
148 #define FLASH_SIZE (FLASH_PAGE * 128)
149
150 struct flash_block {
151         stm32_addr_t addr;
152         unsigned     length;
153         uint8_t*     data;
154
155         struct flash_block* next;
156 };
157
158 static struct flash_block* flash_root;
159
160 static int flash_add_block(stm32_addr_t addr, unsigned length) {
161         if(addr < FLASH_BASE || addr + length > FLASH_BASE + FLASH_SIZE) {
162                 fprintf(stderr, "flash_add_block: incorrect bounds\n");
163                 return -1;
164         }
165
166         if(addr % FLASH_PAGE != 0 || length % FLASH_PAGE != 0) {
167                 fprintf(stderr, "flash_add_block: unaligned block\n");
168                 return -1;
169         }
170
171         struct flash_block* new = malloc(sizeof(struct flash_block));
172         new->next = flash_root;
173
174         new->addr   = addr;
175         new->length = length;
176         new->data   = calloc(length, 1);
177
178         flash_root = new;
179
180         return 0;
181 }
182
183 static int flash_populate(stm32_addr_t addr, uint8_t* data, unsigned length) {
184         int fit_blocks = 0, fit_length = 0;
185
186         for(struct flash_block* fb = flash_root; fb; fb = fb->next) {
187                 /* Block: ------X------Y--------
188                  * Data:            a-----b
189                  *                a--b
190                  *            a-----------b
191                  * Block intersects with data, if:
192                  *  a < Y && b > x
193                  */
194
195                 unsigned X = fb->addr, Y = fb->addr + fb->length;
196                 unsigned a = addr, b = addr + length;
197                 if(a < Y && b > X) {
198                         // from start of the block
199                         unsigned start = (a > X ? a : X) - X;
200                         unsigned end   = (b > Y ? Y : b) - X;
201
202                         memcpy(fb->data + start, data, end - start);
203
204                         fit_blocks++;
205                         fit_length += end - start;
206                 }
207         }
208
209         if(fit_blocks == 0) {
210                 fprintf(stderr, "Unfit data block %08x -> %04x\n", addr, length);
211                 return -1;
212         }
213
214         if(fit_length != length) {
215                 fprintf(stderr, "warning: data block %08x -> %04x truncated to %04x\n",
216                         addr, length, fit_length);
217                 fprintf(stderr, "(this is not an error, just a GDB glitch)\n");
218         }
219
220         return 0;
221 }
222
223 static int flash_go(struct stlink* sl) {
224         int error = -1;
225
226         // Some kinds of clock settings do not allow writing to flash.
227         stlink_reset(sl);
228
229         for(struct flash_block* fb = flash_root; fb; fb = fb->next) {
230                 #ifdef DEBUG
231                 printf("flash_do: block %08x -> %04x\n", fb->addr, fb->length);
232                 #endif
233
234                 unsigned length = fb->length;
235                 for(stm32_addr_t page = fb->addr; page < fb->addr + fb->length; page += 0x400) {
236                         #ifdef DEBUG
237                         printf("flash_do: page %08x\n", page);
238                         #endif
239
240                         stlink_erase_flash_page(sl, page);
241
242                         if(stlink_write_flash(sl, page, fb->data + (page - fb->addr),
243                                         length > 0x400 ? 0x400 : length) < 0)
244                                 goto error;
245                 }
246
247         }
248
249         stlink_reset(sl);
250
251         error = 0;
252
253 error:
254         for(struct flash_block* fb = flash_root, *next; fb; fb = next) {
255                 next = fb->next;
256                 free(fb->data);
257                 free(fb);
258         }
259
260         flash_root = NULL;
261
262         return error;
263 }
264
265 int serve(struct stlink* sl, int port) {
266         int sock = socket(AF_INET, SOCK_STREAM, 0);
267         if(sock < 0) {
268                 perror("socket");
269                 return 1;
270         }
271
272         unsigned int val = 1;
273         setsockopt(sock, SOL_SOCKET, SO_REUSEADDR, &val, sizeof(val));
274
275         struct sockaddr_in serv_addr = {0};
276         serv_addr.sin_family = AF_INET;
277         serv_addr.sin_addr.s_addr = inet_addr("127.0.0.1");
278         serv_addr.sin_port = htons(port);
279
280         if(bind(sock, (struct sockaddr *) &serv_addr, sizeof(serv_addr)) < 0) {
281                 perror("bind");
282                 return 1;
283         }
284
285         if(listen(sock, 5) < 0) {
286                 perror("listen");
287                 return 1;
288         }
289
290         stlink_force_debug(sl);
291         stlink_reset(sl);
292         init_code_breakpoints(sl);
293
294         printf("Listening at *:%d...\n", port);
295
296         int client = accept(sock, NULL, NULL);
297         if(client < 0) {
298                 perror("accept");
299                 return 1;
300         }
301
302         close(sock);
303
304         printf("GDB connected.\n");
305
306         while(1) {
307                 char* packet;
308
309                 int status = gdb_recv_packet(client, &packet);
310                 if(status < 0) {
311                         fprintf(stderr, "cannot recv: %d\n", status);
312                         return 1;
313                 }
314
315                 #ifdef DEBUG
316                 printf("recv: %s\n", packet);
317                 #endif
318
319                 char* reply = NULL;
320
321                 switch(packet[0]) {
322                 case 'q': {
323                         if(packet[1] == 'P' || packet[1] == 'C' || packet[1] == 'L') {
324                                 reply = strdup("");
325                                 break;
326                         }
327
328                         char *separator = strstr(packet, ":"), *params = "";
329                         if(separator == NULL) {
330                                 separator = packet + strlen(packet);
331                         } else {
332                                 params = separator + 1;
333                         }
334
335                         unsigned queryNameLength = (separator - &packet[1]);
336                         char* queryName = calloc(queryNameLength + 1, 1);
337                         strncpy(queryName, &packet[1], queryNameLength);
338
339                         #ifdef DEBUG
340                         printf("query: %s;%s\n", queryName, params);
341                         #endif
342
343                         if(!strcmp(queryName, "Supported")) {
344                                 reply = strdup("PacketSize=3fff;qXfer:memory-map:read+");
345                         } else if(!strcmp(queryName, "Xfer")) {
346                                 char *type, *op, *annex, *s_addr, *s_length;
347                                 char *tok = params;
348
349                                 type     = strsep(&tok, ":");
350                                 op       = strsep(&tok, ":");
351                                 annex    = strsep(&tok, ":");
352                                 s_addr   = strsep(&tok, ",");
353                                 s_length = tok;
354
355                                 unsigned addr = strtoul(s_addr, NULL, 16),
356                                        length = strtoul(s_length, NULL, 16);
357
358                                 #ifdef DEBUG
359                                 printf("Xfer: type:%s;op:%s;annex:%s;addr:%d;length:%d\n",
360                                         type, op, annex, addr, length);
361                                 #endif
362
363                                 const char* data = NULL;
364
365                                 if(!strcmp(type, "memory-map") && !strcmp(op, "read"))
366                                         data = c_memory_map;
367
368                                 if(data) {
369                                         unsigned data_length = strlen(data);
370                                         if(addr + length > data_length)
371                                                 length = data_length - addr;
372
373                                         if(length == 0) {
374                                                 reply = strdup("l");
375                                         } else {
376                                                 reply = calloc(length + 2, 1);
377                                                 reply[0] = 'm';
378                                                 strncpy(&reply[1], data, length);
379                                         }
380                                 }
381                         }
382
383                         if(reply == NULL)
384                                 reply = strdup("");
385
386                         free(queryName);
387
388                         break;
389                 }
390
391                 case 'v': {
392                         char *separator = strstr(packet, ":"), *params = "";
393                         if(separator == NULL) {
394                                 separator = packet + strlen(packet);
395                         } else {
396                                 params = separator + 1;
397                         }
398
399                         unsigned cmdNameLength = (separator - &packet[1]);
400                         char* cmdName = calloc(cmdNameLength + 1, 1);
401                         strncpy(cmdName, &packet[1], cmdNameLength);
402
403                         if(!strcmp(cmdName, "FlashErase")) {
404                                 char *s_addr, *s_length;
405                                 char *tok = params;
406
407                                 s_addr   = strsep(&tok, ",");
408                                 s_length = tok;
409
410                                 unsigned addr = strtoul(s_addr, NULL, 16),
411                                        length = strtoul(s_length, NULL, 16);
412
413                                 #ifdef DEBUG
414                                 printf("FlashErase: addr:%08x,len:%04x\n",
415                                         addr, length);
416                                 #endif
417
418                                 if(flash_add_block(addr, length) < 0) {
419                                         reply = strdup("E00");
420                                 } else {
421                                         reply = strdup("OK");
422                                 }
423                         } else if(!strcmp(cmdName, "FlashWrite")) {
424                                 char *s_addr, *data;
425                                 char *tok = params;
426
427                                 s_addr = strsep(&tok, ":");
428                                 data   = tok;
429
430                                 unsigned addr = strtoul(s_addr, NULL, 16);
431                                 unsigned data_length = status - (data - packet);
432
433                                 // Length of decoded data cannot be more than
434                                 // encoded, as escapes are removed.
435                                 // Additional byte is reserved for alignment fix.
436                                 uint8_t *decoded = calloc(data_length + 1, 1);
437                                 unsigned dec_index = 0;
438                                 for(int i = 0; i < data_length; i++) {
439                                         if(data[i] == 0x7d) {
440                                                 i++;
441                                                 decoded[dec_index++] = data[i] ^ 0x20;
442                                         } else {
443                                                 decoded[dec_index++] = data[i];
444                                         }
445                                 }
446
447                                 // Fix alignment
448                                 if(dec_index % 2 != 0)
449                                         dec_index++;
450
451                                 #ifdef DEBUG
452                                 printf("binary packet %d -> %d\n", data_length, dec_index);
453                                 #endif
454
455                                 if(flash_populate(addr, decoded, dec_index) < 0) {
456                                         reply = strdup("E00");
457                                 } else {
458                                         reply = strdup("OK");
459                                 }
460                         } else if(!strcmp(cmdName, "FlashDone")) {
461                                 if(flash_go(sl) < 0) {
462                                         reply = strdup("E00");
463                                 } else {
464                                         reply = strdup("OK");
465                                 }
466                         }
467
468                         if(reply == NULL)
469                                 reply = strdup("");
470
471                         free(cmdName);
472
473                         break;
474                 }
475
476                 case 'c':
477                         stlink_run(sl);
478
479                         printf("Core running, waiting for interrupt (either in chip or GDB).\n");
480
481                         while(1) {
482                                 int status = gdb_check_for_interrupt(client);
483                                 if(status < 0) {
484                                         fprintf(stderr, "cannot check for int: %d\n", status);
485                                         return 1;
486                                 }
487
488                                 if(status == 1) {
489                                         stlink_force_debug(sl);
490                                         break;
491                                 }
492
493                                 stlink_status(sl);
494                                 if(sl->core_stat == STLINK_CORE_HALTED) {
495                                         break;
496                                 }
497
498                                 usleep(100000);
499                         }
500
501                         reply = strdup("S05"); // TRAP
502                         break;
503
504                 case 's':
505                         stlink_step(sl);
506
507                         reply = strdup("S05"); // TRAP
508                         break;
509
510                 case '?':
511                         reply = strdup("S05"); // TRAP
512                         break;
513
514                 case 'g':
515                         stlink_read_all_regs(sl);
516
517                         reply = calloc(8 * 16 + 1, 1);
518                         for(int i = 0; i < 16; i++)
519                                 sprintf(&reply[i * 8], "%08x", htonl(sl->reg.r[i]));
520
521                         break;
522
523                 case 'p': {
524                         unsigned id = strtoul(&packet[1], NULL, 16), reg = 0xDEADDEAD;
525
526                         if(id < 16) {
527                                 stlink_read_reg(sl, id);
528                                 reg = htonl(sl->reg.r[id]);
529                         } else if(id == 0x19) {
530                                 stlink_read_reg(sl, 16);
531                                 reg = htonl(sl->reg.xpsr);
532                         } else {
533                                 reply = strdup("E00");
534                         }
535
536                         reply = calloc(8 + 1, 1);
537                         sprintf(reply, "%08x", reg);
538
539                         break;
540                 }
541
542                 case 'P': {
543                         char* s_reg = &packet[1];
544                         char* s_value = strstr(&packet[1], "=") + 1;
545
546                         unsigned reg   = strtoul(s_reg,   NULL, 16);
547                         unsigned value = strtoul(s_value, NULL, 16);
548
549                         if(reg < 16) {
550                                 stlink_write_reg(sl, ntohl(value), reg);
551                         } else if(reg == 0x19) {
552                                 stlink_write_reg(sl, ntohl(value), 16);
553                         } else {
554                                 reply = strdup("E00");
555                         }
556
557                         if(!reply) {
558                                 reply = strdup("OK");
559                         }
560
561                         break;
562                 }
563
564                 case 'G':
565                         for(int i = 0; i < 16; i++) {
566                                 char str[9] = {0};
567                                 strncpy(str, &packet[1 + i * 8], 8);
568                                 uint32_t reg = strtoul(str, NULL, 16);
569                                 stlink_write_reg(sl, ntohl(reg), i);
570                         }
571
572                         reply = strdup("OK");
573                         break;
574
575                 case 'm': {
576                         char* s_start = &packet[1];
577                         char* s_count = strstr(&packet[1], ",") + 1;
578
579                         stm32_addr_t start = strtoul(s_start, NULL, 16);
580                         unsigned     count = strtoul(s_count, NULL, 16);
581
582                         unsigned adj_start = start % 4;
583
584                         stlink_read_mem32(sl, start - adj_start, (count % 4 == 0) ?
585                                                 count : count + 4 - (count % 4));
586
587                         reply = calloc(count * 2 + 1, 1);
588                         for(int i = 0; i < count; i++) {
589                                 reply[i * 2 + 0] = hex[sl->q_buf[i + adj_start] >> 4];
590                                 reply[i * 2 + 1] = hex[sl->q_buf[i + adj_start] & 0xf];
591                         }
592
593                         break;
594                 }
595
596                 case 'M': {
597                         char* s_start = &packet[1];
598                         char* s_count = strstr(&packet[1], ",") + 1;
599                         char* hexdata = strstr(packet, ":") + 1;
600
601                         stm32_addr_t start = strtoul(s_start, NULL, 16);
602                         unsigned     count = strtoul(s_count, NULL, 16);
603
604                         for(int i = 0; i < count; i ++) {
605                                 char hex[3] = { hexdata[i*2], hexdata[i*2+1], 0 };
606                                 uint8_t byte = strtoul(hex, NULL, 16);
607                                 sl->q_buf[i] = byte;
608                         }
609
610                         if((count % 4) == 0 && (start % 4) == 0) {
611                                 stlink_write_mem32(sl, start, count);
612                         } else {
613                                 stlink_write_mem8(sl, start, count);
614                         }
615
616                         reply = strdup("OK");
617
618                         break;
619                 }
620
621                 case 'Z': {
622                         if(packet[1] == '1') {
623                                 stm32_addr_t addr = strtoul(&packet[3], NULL, 16);
624                                 if(update_code_breakpoint(sl, addr, 1) < 0) {
625                                         reply = strdup("E00");
626                                 } else {
627                                         reply = strdup("OK");
628                                 }
629                         } else {
630                                 reply = strdup("");
631                         }
632
633                         break;
634                 }
635
636                 case 'z': {
637                         if(packet[1] == '1') {
638                                 stm32_addr_t addr = strtoul(&packet[3], NULL, 16);
639                                 update_code_breakpoint(sl, addr, 0);
640
641                                 reply = strdup("OK");
642                         } else {
643                                 reply = strdup("");
644                         }
645
646                         break;
647                 }
648
649                 case 'k': {
650                         // After this function will be entered afterwards, the
651                         // chip will be reset anyway. So this is a no-op.
652
653                         close(client);
654                         return 0;
655                 }
656
657                 default:
658                         reply = strdup("");
659                 }
660
661                 if(reply) {
662                         #ifdef DEBUG
663                         printf("send: %s\n", reply);
664                         #endif
665
666                         int result = gdb_send_packet(client, reply);
667                         if(result != 0) {
668                                 fprintf(stderr, "cannot send: %d\n", result);
669                                 return 1;
670                         }
671
672                         free(reply);
673                 }
674
675                 free(packet);
676         }
677
678         return 0;
679 }