Subversion Repositories freemyipod

Rev

Go to most recent revision | Details | Last modification | View Log | RSS feed

Rev Author Line No. Line
15 theseven 1
//
2
//
3
//    Copyright 2010 TheSeven
4
//
5
//
427 farthen 6
//    This file is part of emCORE.
15 theseven 7
//
427 farthen 8
//    emCORE is free software: you can redistribute it and/or
15 theseven 9
//    modify it under the terms of the GNU General Public License as
10
//    published by the Free Software Foundation, either version 2 of the
11
//    License, or (at your option) any later version.
12
//
427 farthen 13
//    emCORE is distributed in the hope that it will be useful,
15 theseven 14
//    but WITHOUT ANY WARRANTY; without even the implied warranty of
15
//    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.
16
//    See the GNU General Public License for more details.
17
//
18
//    You should have received a copy of the GNU General Public License along
427 farthen 19
//    with emCORE.  If not, see <http://www.gnu.org/licenses/>.
15 theseven 20
//
21
//
22
 
23
 
24
#include "global.h"
25
#include "panic.h"
26
#include "usb.h"
27
#include "usb_ch9.h"
28
#include "usbdrv.h"
29
#include "thread.h"
30
#include "console.h"
31
#include "util.h"
25 theseven 32
#include "contextswitch.h"
54 theseven 33
#include "power.h"
35 theseven 34
#include "mmu.h"
29 theseven 35
#include "shutdown.h"
92 theseven 36
#include "execimage.h"
85 theseven 37
#ifdef HAVE_I2C
38
#include "i2c.h"
39
#endif
95 theseven 40
#ifdef HAVE_BOOTFLASH
41
#include "bootflash.h"
42
#endif
157 theseven 43
#ifdef HAVE_HWKEYAES
44
#include "hwkeyaes.h"
45
#endif
46
#ifdef HAVE_HMACSHA1
47
#include "hmacsha1.h"
48
#endif
226 theseven 49
#ifdef USB_HAVE_TARGET_SPECIFIC_REQUESTS
50
#include "usbtarget.h"
51
#endif
346 theseven 52
#ifdef HAVE_STORAGE
53
#include "storage.h"
54
#include "disk.h"
55
#include "file.h"
56
#include "dir.h"
347 theseven 57
#include "libc/include/errno.h"
346 theseven 58
#endif
15 theseven 59
 
60
 
61
static uint8_t ctrlresp[2] CACHEALIGN_ATTR;
62
static uint32_t dbgrecvbuf[0x80] CACHEALIGN_ATTR;
63
static uint32_t dbgsendbuf[0x80] CACHEALIGN_ATTR;
64
static uint32_t dbgasyncsendbuf[0x80] CACHEALIGN_ATTR;
65
static char dbgendpoints[4] IBSS_ATTR;
66
 
67
enum dbgaction_t
68
{
69
    DBGACTION_IDLE = 0,
70
    DBGACTION_I2CSEND,
71
    DBGACTION_I2CRECV,
29 theseven 72
    DBGACTION_RESET,
73
    DBGACTION_POWEROFF,
74
    DBGACTION_CWRITE,
75
    DBGACTION_CREAD,
92 theseven 76
    DBGACTION_CFLUSH,
95 theseven 77
    DBGACTION_EXECIMAGE,
127 theseven 78
    DBGACTION_EXECFIRMWARE,
95 theseven 79
    DBGACTION_READBOOTFLASH,
157 theseven 80
    DBGACTION_WRITEBOOTFLASH,
81
    DBGACTION_HWKEYAES,
226 theseven 82
    DBGACTION_HMACSHA1,
346 theseven 83
    DBGACTION_TARGETSPECIFIC,
438 theseven 84
    DBGACTION_STORAGE,
85
    DBGACTION_MALLOC,
86
    DBGACTION_MEMALIGN,
87
    DBGACTION_REALLOC,
88
    DBGACTION_REOWNALLOC,
89
    DBGACTION_FREE
15 theseven 90
};
91
 
429 theseven 92
static struct scheduler_thread dbgthread_handle IBSS_ATTR;
346 theseven 93
static uint32_t dbgstack[0x200] STACK_ATTR;
15 theseven 94
struct wakeup dbgwakeup IBSS_ATTR;
95
static enum dbgaction_t dbgaction IBSS_ATTR;
29 theseven 96
static int dbgi2cbus;
97
static int dbgi2cslave;
98
static int dbgactionaddr;
95 theseven 99
static int dbgactionoffset;
29 theseven 100
static int dbgactionlength;
101
static int dbgactionconsoles;
102
static int dbgactiontype;
25 theseven 103
static char dbgconsendbuf[4096];
104
static char dbgconrecvbuf[1024];
105
static int dbgconsendreadidx IBSS_ATTR;
106
static int dbgconsendwriteidx IBSS_ATTR;
107
static int dbgconrecvreadidx IBSS_ATTR;
108
static int dbgconrecvwriteidx IBSS_ATTR;
109
static struct wakeup dbgconsendwakeup IBSS_ATTR;
110
static struct wakeup dbgconrecvwakeup IBSS_ATTR;
111
static bool dbgconsoleattached IBSS_ATTR;
15 theseven 112
 
25 theseven 113
static const char dbgconoverflowstr[] = "\n\n[overflowed]\n\n";
15 theseven 114
 
438 theseven 115
extern int _poolstart;   // These aren't ints at all, but gcc complains about void types being
116
extern int _poolend;     // used here, and we only need the address, so just make it happy...
25 theseven 117
 
35 theseven 118
 
15 theseven 119
static struct usb_device_descriptor CACHEALIGN_ATTR device_descriptor =
120
{
121
    .bLength            = sizeof(struct usb_device_descriptor),
122
    .bDescriptorType    = USB_DT_DEVICE,
123
    .bcdUSB             = 0x0200,
124
    .bDeviceClass       = USB_CLASS_VENDOR_SPEC,
125
    .bDeviceSubClass    = 0xff,
126
    .bDeviceProtocol    = 0xff,
127
    .bMaxPacketSize0    = 64,
128
    .idVendor           = 0xffff,
129
    .idProduct          = 0xe000,
130
    .bcdDevice          = 0x0001,
131
    .iManufacturer      = 1,
132
    .iProduct           = 2,
133
    .iSerialNumber      = 0,
134
    .bNumConfigurations = 1
135
};
136
 
137
static struct usb_config_bundle
138
{
139
    struct usb_config_descriptor config_descriptor;
140
    struct usb_interface_descriptor interface_descriptor;
141
    struct usb_endpoint_descriptor endpoint1_descriptor;
142
    struct usb_endpoint_descriptor endpoint2_descriptor;
143
    struct usb_endpoint_descriptor endpoint3_descriptor;
144
    struct usb_endpoint_descriptor endpoint4_descriptor;
145
} __attribute__((packed)) CACHEALIGN_ATTR config_bundle = 
146
{
147
    .config_descriptor =
148
    {
149
        .bLength             = sizeof(struct usb_config_descriptor),
150
        .bDescriptorType     = USB_DT_CONFIG,
151
        .wTotalLength        = sizeof(struct usb_config_descriptor)
152
                             + sizeof(struct usb_interface_descriptor)
153
                             + sizeof(struct usb_endpoint_descriptor) * 4,
154
        .bNumInterfaces      = 1,
155
        .bConfigurationValue = 1,
156
        .iConfiguration      = 0,
157
        .bmAttributes        = USB_CONFIG_ATT_ONE,
158
        .bMaxPower           = 250
159
    },
160
    .interface_descriptor =
161
    {
162
        .bLength             = sizeof(struct usb_interface_descriptor),
163
        .bDescriptorType     = USB_DT_INTERFACE,
164
        .bInterfaceNumber    = 0,
165
        .bAlternateSetting   = 0,
166
        .bNumEndpoints       = 4,
167
        .bInterfaceClass     = USB_CLASS_VENDOR_SPEC,
168
        .bInterfaceSubClass  = 0xff,
169
        .bInterfaceProtocol  = 0xff,
170
        .iInterface          = 0
171
    },
172
    .endpoint1_descriptor =
173
    {
174
        .bLength             = sizeof(struct usb_endpoint_descriptor),
175
        .bDescriptorType     = USB_DT_ENDPOINT,
176
        .bEndpointAddress    = 0,
177
        .bmAttributes        = USB_ENDPOINT_XFER_BULK,
178
        .wMaxPacketSize      = 0,
179
        .bInterval           = 1
180
    },
181
    .endpoint2_descriptor =
182
    {
183
        .bLength             = sizeof(struct usb_endpoint_descriptor),
184
        .bDescriptorType     = USB_DT_ENDPOINT,
185
        .bEndpointAddress    = 0,
186
        .bmAttributes        = USB_ENDPOINT_XFER_BULK,
187
        .wMaxPacketSize      = 0,
188
        .bInterval           = 1
189
    },
190
    .endpoint3_descriptor =
191
    {
192
        .bLength             = sizeof(struct usb_endpoint_descriptor),
193
        .bDescriptorType     = USB_DT_ENDPOINT,
194
        .bEndpointAddress    = 0,
195
        .bmAttributes        = USB_ENDPOINT_XFER_BULK,
196
        .wMaxPacketSize      = 0,
197
        .bInterval           = 1
198
    },
199
    .endpoint4_descriptor =
200
    {
201
        .bLength             = sizeof(struct usb_endpoint_descriptor),
202
        .bDescriptorType     = USB_DT_ENDPOINT,
203
        .bEndpointAddress    = 0,
204
        .bmAttributes        = USB_ENDPOINT_XFER_BULK,
205
        .wMaxPacketSize      = 0,
206
        .bInterval           = 1
207
    }
208
};
209
 
210
static struct usb_string_descriptor CACHEALIGN_ATTR string_devicename =
211
{
212
    32,
213
    USB_DT_STRING,
214
    {'e', 'm', 'B', 'I', 'O', 'S', ' ', 'D', 'e', 'b', 'u', 'g', 'g', 'e', 'r'}
215
};
216
 
217
static const struct usb_string_descriptor CACHEALIGN_ATTR lang_descriptor =
218
{
219
    4,
220
    USB_DT_STRING,
221
    {0x0409}
222
};
223
 
224
 
225
void usb_setup_dbg_listener()
226
{
227
    usb_drv_recv(dbgendpoints[0], dbgrecvbuf, usb_drv_port_speed() ? 512 : 64);
228
}
229
 
230
void usb_handle_control_request(struct usb_ctrlrequest* req)
231
{
232
    const void* addr;
233
    int size = -1;
234
    switch (req->bRequest)
235
    {
236
    case USB_REQ_GET_STATUS:
237
        if (req->bRequestType == USB_DIR_IN) ctrlresp[0] = 1;
238
        else ctrlresp[0] = 0;
239
        ctrlresp[1] = 0;
240
        addr = ctrlresp;
241
        size = 2;
242
        break;
243
    case USB_REQ_CLEAR_FEATURE:
244
        if (req->bRequestType == USB_RECIP_ENDPOINT && req->wValue == USB_ENDPOINT_HALT)
245
            usb_drv_stall(req->wIndex & 0xf, false, req->wIndex >> 7);
246
        size = 0;
247
        break;
248
    case USB_REQ_SET_FEATURE:
249
        size = 0;
250
        break;
251
    case USB_REQ_SET_ADDRESS:
252
        size = 0;
253
        usb_drv_cancel_all_transfers();
254
        usb_drv_set_address(req->wValue);
255
        usb_setup_dbg_listener();
256
        break;
257
    case USB_REQ_GET_DESCRIPTOR:
258
        switch (req->wValue >> 8)
259
        {
260
        case USB_DT_DEVICE:
261
            addr = &device_descriptor;
262
            size = sizeof(device_descriptor);
263
            break;
264
        case USB_DT_OTHER_SPEED_CONFIG:
265
        case USB_DT_CONFIG:
266
            if ((req->wValue >> 8) == USB_DT_CONFIG)
267
            {
268
                int maxpacket = usb_drv_port_speed() ? 512 : 64;
269
                config_bundle.endpoint1_descriptor.wMaxPacketSize = maxpacket;
270
                config_bundle.endpoint2_descriptor.wMaxPacketSize = maxpacket;
271
                config_bundle.endpoint3_descriptor.wMaxPacketSize = maxpacket;
272
                config_bundle.endpoint4_descriptor.wMaxPacketSize = maxpacket;
273
                config_bundle.config_descriptor.bDescriptorType = USB_DT_CONFIG;
274
            }
275
            else
276
            {
277
                int maxpacket = usb_drv_port_speed() ? 64 : 512;
278
                config_bundle.endpoint1_descriptor.wMaxPacketSize = maxpacket;
279
                config_bundle.endpoint2_descriptor.wMaxPacketSize = maxpacket;
280
                config_bundle.endpoint3_descriptor.wMaxPacketSize = maxpacket;
281
                config_bundle.endpoint4_descriptor.wMaxPacketSize = maxpacket;
282
                config_bundle.config_descriptor.bDescriptorType = USB_DT_OTHER_SPEED_CONFIG;
283
            }
284
            addr = &config_bundle;
285
            size = sizeof(config_bundle);
286
            break;
287
        case USB_DT_STRING:
288
            switch (req->wValue & 0xff)
289
            {
290
            case 0:
291
                addr = &lang_descriptor;
234 theseven 292
                size = lang_descriptor.bLength;
15 theseven 293
                break;
294
            case 1:
295
                string_devicename.bLength = 14;
296
                addr = &string_devicename;
234 theseven 297
                size = string_devicename.bLength;
298
                break;
15 theseven 299
            case 2:
235 theseven 300
                string_devicename.bLength = 32;
15 theseven 301
                addr = &string_devicename;
234 theseven 302
                size = string_devicename.bLength;
15 theseven 303
                break;
304
            }
305
            break;
306
        }
307
        break;
308
    case USB_REQ_GET_CONFIGURATION:
309
        ctrlresp[0] = 1;
310
        addr = ctrlresp;
311
        size = 1;
312
        break;
313
    case USB_REQ_SET_CONFIGURATION:
314
        usb_drv_cancel_all_transfers();
315
        usb_setup_dbg_listener();
316
        size = 0;
317
        break;
318
    }
319
    if (!size) usb_drv_send_nonblocking(0, NULL, 0);
320
    else if (size == -1)
321
    {
322
        usb_drv_stall(0, true, true);
323
        usb_drv_stall(0, true, false);
324
    }
325
    else
326
    {
327
        usb_drv_recv(0, NULL, 0);
328
        usb_drv_send_nonblocking(0, addr, size > req->wLength ? req->wLength : size);
329
    }
330
}
331
 
29 theseven 332
bool set_dbgaction(enum dbgaction_t action, int addsize)
15 theseven 333
{
334
    if (dbgaction != DBGACTION_IDLE)
335
    {
336
        dbgsendbuf[0] = 3;
29 theseven 337
        usb_drv_send_nonblocking(dbgendpoints[1], dbgsendbuf, 16 + addsize);
15 theseven 338
        return true;
339
    }
340
    dbgaction = action;
341
    wakeup_signal(&dbgwakeup);
342
    return false;
343
}
344
 
345
void reset() __attribute__((noreturn));
346
 
347
void usb_handle_transfer_complete(int endpoint, int dir, int status, int length)
348
{
349
    void* addr = dbgsendbuf;
350
    int size = 0;
351
    if (endpoint == dbgendpoints[0])
352
    {
226 theseven 353
#ifdef USB_HAVE_TARGET_SPECIFIC_REQUESTS
354
        if (dbgrecvbuf[0] >= 0xffff0000)
355
        {
356
            if (!set_dbgaction(DBGACTION_TARGETSPECIFIC, 0))
357
                memcpy(dbgasyncsendbuf, dbgrecvbuf, sizeof(dbgasyncsendbuf));
358
            usb_setup_dbg_listener();
359
            return;
360
        }
361
#endif
15 theseven 362
        switch (dbgrecvbuf[0])
363
        {
28 theseven 364
        case 1:  // GET INFO
15 theseven 365
            dbgsendbuf[0] = 1;
366
            size = 16;
28 theseven 367
            switch (dbgrecvbuf[1])
368
            {
369
            case 0:  // GET VERSION INFO
85 theseven 370
                dbgsendbuf[1] = VERSION_SVN_INT;
371
                dbgsendbuf[2] = VERSION_MAJOR | (VERSION_MINOR << 8)
429 theseven 372
                              | (VERSION_PATCH << 16) | (2 << 24);
85 theseven 373
                dbgsendbuf[3] = PLATFORM_ID;
28 theseven 374
                break;
375
            case 1:  // GET PACKET SIZE INFO
376
                dbgsendbuf[1] = 0x02000200;
377
                dbgsendbuf[2] = usb_drv_get_max_out_size();
378
                dbgsendbuf[3] = usb_drv_get_max_in_size();
379
                break;
35 theseven 380
            case 2:  // GET USER MEMORY INFO
438 theseven 381
                dbgsendbuf[1] = (uint32_t)&_poolstart;
382
                dbgsendbuf[2] = (uint32_t)&_poolend;
35 theseven 383
                break;
28 theseven 384
            default:
385
                dbgsendbuf[0] = 2;
386
            }
15 theseven 387
            break;
388
        case 2:  // RESET
29 theseven 389
            if (dbgrecvbuf[1])
390
            {
391
                if (set_dbgaction(DBGACTION_RESET, 0)) break;
392
                dbgsendbuf[0] = 1;
393
                size = 16;
394
            }
395
            else reset();
15 theseven 396
            break;
397
        case 3:  // POWER OFF
29 theseven 398
            if (set_dbgaction(DBGACTION_POWEROFF, 0)) break;
399
            dbgactiontype = dbgrecvbuf[1];
400
            dbgsendbuf[0] = 1;
401
            size = 16;
15 theseven 402
            break;
403
        case 4:  // READ MEMORY
404
            dbgsendbuf[0] = 1;
405
            memcpy(&dbgsendbuf[4], (const void*)dbgrecvbuf[1], dbgrecvbuf[2]);
406
            size = dbgrecvbuf[2] + 16;
407
            break;
408
        case 5:  // WRITE MEMORY
409
            dbgsendbuf[0] = 1;
410
            memcpy((void*)dbgrecvbuf[1], &dbgrecvbuf[4], dbgrecvbuf[2]);
411
            size = 16;
412
            break;
413
        case 6:  // READ DMA
414
            dbgsendbuf[0] = 1;
415
            usb_drv_send_nonblocking(dbgendpoints[1], dbgsendbuf, 16);
416
            usb_drv_send_nonblocking(dbgendpoints[3], (const void*)dbgrecvbuf[1], dbgrecvbuf[2]);
417
            break;
418
        case 7:  // WRITE DMA
419
            dbgsendbuf[0] = 1;
420
            size = 16;
421
            usb_drv_recv(dbgendpoints[2], (void*)dbgrecvbuf[1], dbgrecvbuf[2]);
422
            break;
85 theseven 423
#ifdef HAVE_I2C
15 theseven 424
        case 8:  // READ I2C
29 theseven 425
            if (set_dbgaction(DBGACTION_I2CRECV, dbgrecvbuf[1] >> 24)) break;
15 theseven 426
            dbgi2cbus = dbgrecvbuf[1] & 0xff;
427
            dbgi2cslave = (dbgrecvbuf[1] >> 8) & 0xff;
29 theseven 428
            dbgactionaddr = (dbgrecvbuf[1] >> 16) & 0xff;
429
            dbgactionlength = dbgrecvbuf[1] >> 24;
175 theseven 430
            if (!dbgactionlength) dbgactionlength = 256;
15 theseven 431
            break;
432
        case 9:  // WRITE I2C
29 theseven 433
            if (set_dbgaction(DBGACTION_I2CSEND, 0)) break;
15 theseven 434
            dbgi2cbus = dbgrecvbuf[1] & 0xff;
435
            dbgi2cslave = (dbgrecvbuf[1] >> 8) & 0xff;
29 theseven 436
            dbgactionaddr = (dbgrecvbuf[1] >> 16) & 0xff;
437
            dbgactionlength = dbgrecvbuf[1] >> 24;
175 theseven 438
            if (!dbgactionlength) dbgactionlength = 256;
212 theseven 439
            memcpy(dbgasyncsendbuf, &dbgrecvbuf[4], dbgactionlength);
15 theseven 440
            break;
85 theseven 441
#endif
25 theseven 442
        case 10:  // READ CONSOLE
443
            dbgconsoleattached = true;
444
            int bytes = dbgconsendwriteidx - dbgconsendreadidx;
336 theseven 445
            int used = 0;
25 theseven 446
            if (bytes)
447
            {
448
                if (bytes < 0) bytes += sizeof(dbgconsendbuf);
336 theseven 449
                used = bytes;
25 theseven 450
                if (bytes > dbgrecvbuf[1]) bytes = dbgrecvbuf[1];
451
                int readbytes = bytes;
452
                char* outptr = (char*)&dbgsendbuf[4];
453
                if (dbgconsendreadidx + bytes >= sizeof(dbgconsendbuf))
454
                {
455
                    readbytes = sizeof(dbgconsendbuf) - dbgconsendreadidx;
456
                    memcpy(outptr, &dbgconsendbuf[dbgconsendreadidx], readbytes);
457
                    dbgconsendreadidx = 0;
458
                    outptr = &outptr[readbytes];
459
                    readbytes = bytes - readbytes;
460
                }
461
                if (readbytes) memcpy(outptr, &dbgconsendbuf[dbgconsendreadidx], readbytes);
462
                dbgconsendreadidx += readbytes;
26 theseven 463
                wakeup_signal(&dbgconsendwakeup);
25 theseven 464
            }
465
            dbgsendbuf[0] = 1;
466
            dbgsendbuf[1] = bytes;
467
            dbgsendbuf[2] = sizeof(dbgconsendbuf);
336 theseven 468
            dbgsendbuf[3] = used - bytes;
25 theseven 469
            size = 16 + dbgrecvbuf[1];
470
            break;
26 theseven 471
        case 11:  // WRITE CONSOLE
472
            bytes = dbgconrecvreadidx - dbgconrecvwriteidx - 1;
473
            if (bytes < 0) bytes += sizeof(dbgconrecvbuf);
474
            if (bytes)
475
            {
476
                if (bytes > dbgrecvbuf[1]) bytes = dbgrecvbuf[1];
477
                int writebytes = bytes;
478
                char* readptr = (char*)&dbgrecvbuf[4];
479
                if (dbgconrecvwriteidx + bytes >= sizeof(dbgconrecvbuf))
480
                {
481
                    writebytes = sizeof(dbgconrecvbuf) - dbgconrecvwriteidx;
482
                    memcpy(&dbgconrecvbuf[dbgconrecvwriteidx], readptr, writebytes);
483
                    dbgconrecvwriteidx = 0;
484
                    readptr = &readptr[writebytes];
485
                    writebytes = bytes - writebytes;
486
                }
487
                if (writebytes) memcpy(&dbgconrecvbuf[dbgconrecvwriteidx], readptr, writebytes);
488
                dbgconrecvwriteidx += writebytes;
489
                wakeup_signal(&dbgconrecvwakeup);
490
            }
491
            dbgsendbuf[0] = 1;
492
            dbgsendbuf[1] = bytes;
493
            dbgsendbuf[2] = sizeof(dbgconrecvbuf);
494
            dbgsendbuf[3] = dbgconrecvreadidx - dbgconrecvwriteidx - 1;
495
            size = 16;
496
            break;
29 theseven 497
        case 12:  // CWRITE
498
            if (set_dbgaction(DBGACTION_CWRITE, 0)) break;
499
            dbgactionconsoles = dbgrecvbuf[1];
500
            dbgactionlength = dbgrecvbuf[2];
501
            memcpy(dbgasyncsendbuf, &dbgrecvbuf[4], dbgactionlength);
502
            break;
503
        case 13:  // CREAD
504
            if (set_dbgaction(DBGACTION_CREAD, dbgrecvbuf[2])) break;
505
            dbgactionconsoles = dbgrecvbuf[1];
506
            dbgactionlength = dbgrecvbuf[2];
507
            break;
508
        case 14:  // CFLUSH
509
            if (set_dbgaction(DBGACTION_CFLUSH, 0)) break;
510
            dbgactionconsoles = dbgrecvbuf[1];
511
            break;
31 theseven 512
        case 15:  // GET PROCESS INFO
513
            dbgsendbuf[0] = 1;
514
            dbgsendbuf[1] = SCHEDULER_THREAD_INFO_VERSION;
429 theseven 515
            dbgsendbuf[2] = (uint32_t)head_thread;
516
            size = 16;
31 theseven 517
            break;
34 theseven 518
        case 16:  // FREEZE SCHEDULER
54 theseven 519
            dbgsendbuf[1] = scheduler_freeze(dbgrecvbuf[1]);
429 theseven 520
            scheduler_switch(NULL);
34 theseven 521
            dbgsendbuf[0] = 1;
522
            size = 16;
523
            break;
35 theseven 524
        case 17:  // SUSPEND THREAD
54 theseven 525
            if (dbgrecvbuf[1])
526
            {
429 theseven 527
                if (thread_suspend((struct scheduler_thread*)(dbgrecvbuf[2])) == ALREADY_SUSPENDED)
528
                    dbgsendbuf[1] = 1;
54 theseven 529
                else dbgsendbuf[1] = 0;
530
            }
531
            else
532
            {
429 theseven 533
                if (thread_resume((struct scheduler_thread*)(dbgrecvbuf[2])) == ALREADY_RESUMED)
534
                    dbgsendbuf[1] = 0;
54 theseven 535
                else dbgsendbuf[1] = 1;
536
            }
35 theseven 537
            dbgsendbuf[0] = 1;
538
            size = 16;
539
            break;
540
        case 18:  // KILL THREAD
429 theseven 541
            thread_terminate((struct scheduler_thread*)(dbgrecvbuf[1]));
35 theseven 542
            dbgsendbuf[0] = 1;
543
            size = 16;
544
            break;
438 theseven 545
        case 19:  // CREATE THREAD
35 theseven 546
            dbgsendbuf[0] = 1;
429 theseven 547
            dbgsendbuf[1] = (uint32_t)thread_create(NULL, (const char*)(dbgsendbuf[1]),
548
                                                    (const void*)(dbgsendbuf[2]),
549
                                                    (char*)(dbgsendbuf[3]),
550
                                                    dbgsendbuf[4], (enum thread_type)dbgsendbuf[5],
551
                                                    dbgsendbuf[6], dbgsendbuf[7]);
35 theseven 552
            size = 16;
553
            break;
554
        case 20:  // FLUSH CACHE
555
            clean_dcache();
556
            invalidate_icache();
557
            dbgsendbuf[0] = 1;
558
            size = 16;
559
            break;
92 theseven 560
        case 21:  // EXECIMAGE
561
            if (set_dbgaction(DBGACTION_EXECIMAGE, 0)) break;
562
            dbgactionaddr = dbgrecvbuf[1];
563
            break;
95 theseven 564
#ifdef HAVE_BOOTFLASH
565
        case 22:  // READ BOOT FLASH
566
            if (set_dbgaction(DBGACTION_READBOOTFLASH, 0)) break;
567
            dbgactionaddr = dbgrecvbuf[1];
568
            dbgactionoffset = dbgrecvbuf[2];
569
            dbgactionlength = dbgrecvbuf[3];
570
            break;
571
        case 23:  // WRITE BOOT FLASH
572
            if (set_dbgaction(DBGACTION_WRITEBOOTFLASH, 0)) break;
573
            dbgactionaddr = dbgrecvbuf[1];
574
            dbgactionoffset = dbgrecvbuf[2];
575
            dbgactionlength = dbgrecvbuf[3];
576
            break;
577
#endif
127 theseven 578
        case 24:  // EXECFIRMWARE
579
            if (set_dbgaction(DBGACTION_EXECFIRMWARE, 0)) break;
580
            dbgactionaddr = dbgrecvbuf[1];
438 theseven 581
            dbgactionoffset = dbgrecvbuf[2];
582
            dbgactionlength = dbgrecvbuf[3];
127 theseven 583
            break;
157 theseven 584
#ifdef HAVE_HWKEYAES
585
        case 25:  // HWKEYAES
586
            if (set_dbgaction(DBGACTION_HWKEYAES, 0)) break;
587
            dbgactiontype = ((uint8_t*)dbgrecvbuf)[4];
588
            dbgactionoffset = ((uint16_t*)dbgrecvbuf)[3];
589
            dbgactionaddr = dbgrecvbuf[2];
590
            dbgactionlength = dbgrecvbuf[3];
179 theseven 591
            break;
157 theseven 592
#endif
593
#ifdef HAVE_HMACSHA1
594
        case 26:  // HMACSHA1
595
            if (set_dbgaction(DBGACTION_HMACSHA1, 0)) break;
596
            dbgactionaddr = dbgrecvbuf[1];
597
            dbgactionlength = dbgrecvbuf[2];
598
            dbgactionoffset = dbgrecvbuf[3];
179 theseven 599
            break;
157 theseven 600
#endif
346 theseven 601
#ifdef HAVE_STORAGE
602
        case 27:  // STORAGE_GET_INFO
603
        case 28:  // STORAGE_READ_SECTORS_MD
604
        case 29:  // STORAGE_WRITE_SECTORS_MD
605
        case 30:  // FILE_OPEN
606
        case 31:  // FILESIZE
607
        case 32:  // READ
608
        case 33:  // WRITE
609
        case 34:  // LSEEK
610
        case 35:  // FTRUNCATE
611
        case 36:  // FSYNC
612
        case 37:  // CLOSE
613
        case 38:  // CLOSE_MONITOR_FILES
614
        case 39:  // RELEASE_FILES
615
        case 40:  // REMOVE
616
        case 41:  // RENAME
617
        case 42:  // OPENDIR
618
        case 43:  // READDIR
619
        case 44:  // CLOSEDIR
620
        case 45:  // CLOSE_MONITOR_DIRS
621
        case 46:  // RELEASE_DIRS
622
        case 47:  // MKDIR
623
        case 48:  // RMDIR
624
        case 49:  // ERRNO
625
#ifdef HAVE_HOTSWAP
626
        case 50:  // DISK_MOUNT
627
        case 51:  // DISK_UNMOUNT
628
#endif
629
            if (!set_dbgaction(DBGACTION_STORAGE, 0))
630
                memcpy(dbgasyncsendbuf, dbgrecvbuf, sizeof(dbgasyncsendbuf));
631
            break;
632
#endif
438 theseven 633
        case 52:  // MALLOC
634
            if (set_dbgaction(DBGACTION_MALLOC, 0)) break;
635
            dbgactionlength = dbgrecvbuf[1];
636
            break;
637
        case 53:  // MEMALIGN
638
            if (set_dbgaction(DBGACTION_MEMALIGN, 0)) break;
639
            dbgactionoffset = dbgrecvbuf[1];
640
            dbgactionlength = dbgrecvbuf[2];
641
            break;
642
        case 54:  // REALLOC
643
            if (set_dbgaction(DBGACTION_REALLOC, 0)) break;
644
            dbgactionaddr = dbgrecvbuf[1];
645
            dbgactionlength = dbgrecvbuf[2];
646
            break;
647
        case 55:  // REOWNALLOC
648
            if (set_dbgaction(DBGACTION_REOWNALLOC, 0)) break;
649
            dbgactionaddr = dbgrecvbuf[1];
650
            dbgactionoffset = dbgrecvbuf[2];
651
            break;
652
        case 56:  // FREE
653
            if (set_dbgaction(DBGACTION_FREE, 0)) break;
654
            dbgactionaddr = dbgrecvbuf[1];
655
            break;
15 theseven 656
        default:
657
            dbgsendbuf[0] = 2;
658
            size = 16;
659
        }
660
        usb_setup_dbg_listener();
661
        if (size) usb_drv_send_nonblocking(dbgendpoints[1], addr, size);
662
    }
663
}
664
 
665
void usb_handle_bus_reset(void)
666
{
667
    dbgendpoints[0] = usb_drv_request_endpoint(USB_ENDPOINT_XFER_BULK, USB_DIR_OUT);
668
    dbgendpoints[1] = usb_drv_request_endpoint(USB_ENDPOINT_XFER_BULK, USB_DIR_IN);
669
    dbgendpoints[2] = usb_drv_request_endpoint(USB_ENDPOINT_XFER_BULK, USB_DIR_OUT);
670
    dbgendpoints[3] = usb_drv_request_endpoint(USB_ENDPOINT_XFER_BULK, USB_DIR_IN);
671
    config_bundle.endpoint1_descriptor.bEndpointAddress = dbgendpoints[0];
672
    config_bundle.endpoint2_descriptor.bEndpointAddress = dbgendpoints[1];
673
    config_bundle.endpoint3_descriptor.bEndpointAddress = dbgendpoints[2];
674
    config_bundle.endpoint4_descriptor.bEndpointAddress = dbgendpoints[3];
675
    usb_setup_dbg_listener();
676
}
677
 
678
void dbgthread(void)
679
{
429 theseven 680
    struct scheduler_thread* t;
15 theseven 681
    while (1)
682
    {
683
        wakeup_wait(&dbgwakeup, TIMEOUT_BLOCK);
429 theseven 684
        for (t = head_thread; t; t = t->thread_next)
685
            if (t->state == THREAD_DEFUNCT)
15 theseven 686
            {
429 theseven 687
                if (t->block_type == THREAD_DEFUNCT_STKOV)
35 theseven 688
                {
429 theseven 689
                    if (t->name) cprintf(1, "\n*PANIC*\nStack overflow! (%s)\n", t->name);
690
                    else cprintf(1, "\n*PANIC*\nStack overflow! (%08X)\n", t);
35 theseven 691
                }
429 theseven 692
                t->state = THREAD_DEFUNCT_ACK;
15 theseven 693
            }
694
        if (dbgaction != DBGACTION_IDLE)
695
        {
696
            switch (dbgaction)
697
            {
85 theseven 698
#ifdef HAVE_I2C
15 theseven 699
            case DBGACTION_I2CSEND:
29 theseven 700
                i2c_send(dbgi2cbus, dbgi2cslave, dbgactionaddr,
701
                         (uint8_t*)dbgasyncsendbuf, dbgactionlength);
15 theseven 702
                dbgasyncsendbuf[0] = 1;
703
                usb_drv_send_nonblocking(dbgendpoints[1], dbgasyncsendbuf, 16);
704
                break;
705
            case DBGACTION_I2CRECV:
29 theseven 706
                i2c_recv(dbgi2cbus, dbgi2cslave, dbgactionaddr,
707
                         (uint8_t*)(&dbgasyncsendbuf[4]), dbgactionlength);
15 theseven 708
                dbgasyncsendbuf[0] = 1;
29 theseven 709
                usb_drv_send_nonblocking(dbgendpoints[1], dbgasyncsendbuf, 16 + dbgactionlength);
15 theseven 710
                break;
85 theseven 711
#endif
15 theseven 712
            case DBGACTION_POWEROFF:
127 theseven 713
                if (dbgactiontype) shutdown(true);
54 theseven 714
                power_off();
15 theseven 715
                break;
29 theseven 716
            case DBGACTION_RESET:
439 theseven 717
                shutdown(false);
29 theseven 718
                reset();
719
                break;
720
            case DBGACTION_CWRITE:
721
                cwrite(dbgactionconsoles, (const char*)dbgasyncsendbuf, dbgactionlength);
722
                dbgasyncsendbuf[0] = 1;
723
                usb_drv_send_nonblocking(dbgendpoints[1], dbgasyncsendbuf, 16);
724
                break;
725
            case DBGACTION_CREAD:
726
                dbgasyncsendbuf[0] = 1;
30 theseven 727
                dbgasyncsendbuf[1] = cread(dbgactionconsoles, (char*)&dbgasyncsendbuf[4],
728
                                           dbgactionlength, 0);
226 theseven 729
                usb_drv_send_nonblocking(dbgendpoints[1], dbgasyncsendbuf, 16 + dbgactionlength);
29 theseven 730
                break;
731
            case DBGACTION_CFLUSH:
732
                cflush(dbgactionconsoles);
733
                dbgasyncsendbuf[0] = 1;
734
                usb_drv_send_nonblocking(dbgendpoints[1], dbgasyncsendbuf, 16);
735
                break;
92 theseven 736
            case DBGACTION_EXECIMAGE:
737
                dbgasyncsendbuf[0] = 1;
436 theseven 738
                dbgasyncsendbuf[1] = execimage((void*)dbgactionaddr, false);
92 theseven 739
                usb_drv_send_nonblocking(dbgendpoints[1], dbgasyncsendbuf, 16);
740
                break;
127 theseven 741
            case DBGACTION_EXECFIRMWARE:
742
                dbgasyncsendbuf[0] = 1;
743
                usb_drv_send_nonblocking(dbgendpoints[1], dbgasyncsendbuf, 16);
439 theseven 744
                shutdown(false);
438 theseven 745
                execfirmware((void*)dbgactionaddr, (void*)dbgactionoffset,
746
                             (size_t)dbgactionlength);
95 theseven 747
#ifdef HAVE_BOOTFLASH
748
            case DBGACTION_READBOOTFLASH:
749
                bootflash_readraw((void*)dbgactionaddr, dbgactionoffset, dbgactionlength);
750
                dbgasyncsendbuf[0] = 1;
751
                usb_drv_send_nonblocking(dbgendpoints[1], dbgasyncsendbuf, 16);
752
                break;
753
            case DBGACTION_WRITEBOOTFLASH:
754
                bootflash_writeraw((void*)dbgactionaddr, dbgactionoffset, dbgactionlength);
755
                dbgasyncsendbuf[0] = 1;
756
                usb_drv_send_nonblocking(dbgendpoints[1], dbgasyncsendbuf, 16);
757
                break;
758
#endif
157 theseven 759
#ifdef HAVE_HWKEYAES
760
            case DBGACTION_HWKEYAES:
761
                hwkeyaes((enum hwkeyaes_direction) dbgactiontype, dbgactionoffset,
762
                         (void*)dbgactionaddr, dbgactionlength);
763
                dbgasyncsendbuf[0] = 1;
764
                usb_drv_send_nonblocking(dbgendpoints[1], dbgasyncsendbuf, 16);
765
                break;
766
#endif
767
#ifdef HAVE_HMACSHA1
768
            case DBGACTION_HMACSHA1:
769
                hmacsha1((void*)dbgactionaddr, dbgactionlength, (void*)dbgactionoffset);
770
                dbgasyncsendbuf[0] = 1;
771
                usb_drv_send_nonblocking(dbgendpoints[1], dbgasyncsendbuf, 16);
772
                break;
773
#endif
226 theseven 774
#ifdef USB_HAVE_TARGET_SPECIFIC_REQUESTS
775
            case DBGACTION_TARGETSPECIFIC:
776
            {
777
                int size = usb_target_handle_request(dbgasyncsendbuf, sizeof(dbgasyncsendbuf));
778
                if (size) usb_drv_send_nonblocking(dbgendpoints[1], dbgasyncsendbuf, size);
779
                break;
15 theseven 780
            }
226 theseven 781
#endif
346 theseven 782
#ifdef HAVE_STORAGE
783
            case DBGACTION_STORAGE:
784
                switch(dbgasyncsendbuf[0])
785
                {
786
                case 27:  // STORAGE_GET_INFO
787
                    dbgasyncsendbuf[0] = 1;
788
                    storage_get_info(dbgasyncsendbuf[1],
789
                                     (struct storage_info*)&dbgasyncsendbuf[4]);
790
                    dbgasyncsendbuf[1] = 1;
791
                    usb_drv_send_nonblocking(dbgendpoints[1], dbgasyncsendbuf,
792
                                             16 + sizeof(struct storage_info));
793
                    break;
794
                case 28:  // STORAGE_READ_SECTORS_MD
795
                {
796
                    dbgasyncsendbuf[0] = 1;
797
                    int rc = storage_read_sectors_md(dbgasyncsendbuf[1],
798
                                                     dbgasyncsendbuf[2]
799
                                                   | (((uint64_t)(dbgasyncsendbuf[3]) << 32)),
800
                                                     dbgasyncsendbuf[4],
801
                                                     (void*)(dbgasyncsendbuf[5]));
802
                    dbgasyncsendbuf[1] = (uint32_t)rc;
803
                    usb_drv_send_nonblocking(dbgendpoints[1], dbgasyncsendbuf, 16);
804
                    break;
805
                }
806
                case 29:  // STORAGE_WRITE_SECTORS_MD
807
                {
808
                    dbgasyncsendbuf[0] = 1;
809
                    int rc = storage_write_sectors_md(dbgasyncsendbuf[1],
810
                                                      dbgasyncsendbuf[2]
811
                                                    | (((uint64_t)(dbgasyncsendbuf[3]) << 32)),
812
                                                      dbgasyncsendbuf[4],
813
                                                      (void*)(dbgasyncsendbuf[5]));
814
                    dbgasyncsendbuf[1] = (uint32_t)rc;
815
                    usb_drv_send_nonblocking(dbgendpoints[1], dbgasyncsendbuf, 16);
816
                    break;
817
                }
818
                case 30:  // FILE_OPEN
819
                {
820
                    dbgasyncsendbuf[0] = 1;
821
                    int fd = file_open((char*)(&dbgasyncsendbuf[4]), (int)(dbgasyncsendbuf[1]));
822
                    dbgasyncsendbuf[1] = (uint32_t)fd;
823
                    usb_drv_send_nonblocking(dbgendpoints[1], dbgasyncsendbuf, 16);
824
                    break;
825
                }
826
                case 31:  // FILESIZE
827
                    dbgasyncsendbuf[0] = 1;
828
                    dbgasyncsendbuf[1] = (uint32_t)filesize((int)(dbgasyncsendbuf[1]));
829
                    usb_drv_send_nonblocking(dbgendpoints[1], dbgasyncsendbuf, 16);
830
                    break;
831
                case 32:  // READ
832
                    dbgasyncsendbuf[0] = 1;
833
                    dbgasyncsendbuf[1] = (uint32_t)read((int)(dbgasyncsendbuf[1]),
834
                                                        (void*)(dbgasyncsendbuf[2]),
835
                                                        (size_t)(dbgasyncsendbuf[3]));
836
                    usb_drv_send_nonblocking(dbgendpoints[1], dbgasyncsendbuf, 16);
837
                    break;
838
                case 33:  // WRITE
839
                    dbgasyncsendbuf[0] = 1;
840
                    dbgasyncsendbuf[1] = (uint32_t)write((int)(dbgasyncsendbuf[1]),
841
                                                         (void*)(dbgasyncsendbuf[2]),
842
                                                         (size_t)(dbgasyncsendbuf[3]));
843
                    usb_drv_send_nonblocking(dbgendpoints[1], dbgasyncsendbuf, 16);
844
                    break;
845
                case 34:  // LSEEK
846
                    dbgasyncsendbuf[0] = 1;
847
                    dbgasyncsendbuf[1] = (uint32_t)lseek((int)(dbgasyncsendbuf[1]),
848
                                                         (off_t)(dbgasyncsendbuf[2]),
849
                                                         (int)(dbgasyncsendbuf[3]));
850
                    usb_drv_send_nonblocking(dbgendpoints[1], dbgasyncsendbuf, 16);
851
                    break;
852
                case 35:  // FTRUNCATE
853
                    dbgasyncsendbuf[0] = 1;
854
                    dbgasyncsendbuf[1] = (uint32_t)ftruncate((int)(dbgasyncsendbuf[1]),
855
                                                             (off_t)(dbgasyncsendbuf[2]));
856
                    usb_drv_send_nonblocking(dbgendpoints[1], dbgasyncsendbuf, 16);
857
                    break;
858
                case 36:  // FSYNC
859
                    dbgasyncsendbuf[0] = 1;
860
                    dbgasyncsendbuf[1] = (uint32_t)fsync((int)(dbgasyncsendbuf[1]));
861
                    usb_drv_send_nonblocking(dbgendpoints[1], dbgasyncsendbuf, 16);
862
                    break;
863
                case 37:  // CLOSE
864
                    dbgasyncsendbuf[0] = 1;
865
                    dbgasyncsendbuf[1] = (uint32_t)close((int)(dbgasyncsendbuf[1]));
866
                    usb_drv_send_nonblocking(dbgendpoints[1], dbgasyncsendbuf, 16);
867
                    break;
868
                case 38:  // CLOSE_MONITOR_FILES
869
                    dbgasyncsendbuf[0] = 1;
870
                    dbgasyncsendbuf[1] = (uint32_t)close_all_of_process(current_thread);
871
                    usb_drv_send_nonblocking(dbgendpoints[1], dbgasyncsendbuf, 16);
872
                    break;
873
                case 39:  // RELEASE_FILES
874
                    dbgasyncsendbuf[0] = 1;
875
                    dbgasyncsendbuf[1] = (uint32_t)release_files((int)(dbgasyncsendbuf[1]));
876
                    usb_drv_send_nonblocking(dbgendpoints[1], dbgasyncsendbuf, 16);
877
                    break;
878
                case 40:  // REMOVE
879
                    dbgasyncsendbuf[0] = 1;
880
                    dbgasyncsendbuf[1] = (uint32_t)remove((char*)(&dbgasyncsendbuf[4]));
881
                    usb_drv_send_nonblocking(dbgendpoints[1], dbgasyncsendbuf, 16);
882
                    break;
883
                case 41:  // RENAME
884
                    dbgasyncsendbuf[0] = 1;
885
                    dbgasyncsendbuf[1] = (uint32_t)rename((char*)(&dbgasyncsendbuf[4]),
886
                                                          (char*)(&dbgasyncsendbuf[66]));
887
                    usb_drv_send_nonblocking(dbgendpoints[1], dbgasyncsendbuf, 16);
888
                    break;
889
                case 42:  // OPENDIR
890
                    dbgasyncsendbuf[0] = 1;
891
                    dbgasyncsendbuf[1] = (uint32_t)opendir((char*)(&dbgasyncsendbuf[4]));
892
                    usb_drv_send_nonblocking(dbgendpoints[1], dbgasyncsendbuf, 16);
893
                    break;
894
                case 43:  // READDIR
895
                    dbgasyncsendbuf[0] = 1;
896
                    dbgasyncsendbuf[3] = (uint32_t)readdir((DIR*)(dbgasyncsendbuf[1]));
897
                    dbgasyncsendbuf[1] = 1;
898
                    dbgasyncsendbuf[2] = MAX_PATH;
899
                    usb_drv_send_nonblocking(dbgendpoints[1], dbgasyncsendbuf, 16);
900
                    break;
901
                case 44:  // CLOSEDIR
902
                    dbgasyncsendbuf[0] = 1;
903
                    dbgasyncsendbuf[1] = (uint32_t)closedir((DIR*)(dbgasyncsendbuf[1]));
904
                    usb_drv_send_nonblocking(dbgendpoints[1], dbgasyncsendbuf, 16);
905
                    break;
906
                case 45:  // CLOSE_MONITOR_DIRS
907
                    dbgasyncsendbuf[0] = 1;
908
                    dbgasyncsendbuf[1] = (uint32_t)closedir_all_of_process(current_thread);
909
                    usb_drv_send_nonblocking(dbgendpoints[1], dbgasyncsendbuf, 16);
910
                    break;
911
                case 46:  // RELEASE_DIRS
912
                    dbgasyncsendbuf[0] = 1;
913
                    dbgasyncsendbuf[1] = (uint32_t)release_dirs((int)(dbgasyncsendbuf[1]));
914
                    usb_drv_send_nonblocking(dbgendpoints[1], dbgasyncsendbuf, 16);
915
                    break;
916
                case 47:  // MKDIR
917
                    dbgasyncsendbuf[0] = 1;
918
                    dbgasyncsendbuf[1] = (uint32_t)mkdir((char*)(&dbgasyncsendbuf[4]));
919
                    usb_drv_send_nonblocking(dbgendpoints[1], dbgasyncsendbuf, 16);
920
                    break;
921
                case 48:  // RMDIR
922
                    dbgasyncsendbuf[0] = 1;
923
                    dbgasyncsendbuf[1] = (uint32_t)rmdir((char*)(&dbgasyncsendbuf[4]));
924
                    usb_drv_send_nonblocking(dbgendpoints[1], dbgasyncsendbuf, 16);
925
                    break;
926
                case 49:  // ERRNO
927
                    dbgasyncsendbuf[0] = 1;
928
                    dbgasyncsendbuf[1] = (uint32_t)errno;
929
                    usb_drv_send_nonblocking(dbgendpoints[1], dbgasyncsendbuf, 16);
930
                    break;
931
#ifdef HAVE_HOTSWAP
932
                case 50:  // DISK_MOUNT
933
                    dbgasyncsendbuf[0] = 1;
934
                    dbgasyncsendbuf[1] = (uint32_t)disk_mount((int)(dbgasyncsendbuf[1]));
935
                    usb_drv_send_nonblocking(dbgendpoints[1], dbgasyncsendbuf, 16);
936
                    break;
937
                case 51:  // DISK_UNMOUNT
938
                    dbgasyncsendbuf[0] = 1;
939
                    dbgasyncsendbuf[1] = (uint32_t)disk_unmount((int)(dbgasyncsendbuf[1]));
940
                    usb_drv_send_nonblocking(dbgendpoints[1], dbgasyncsendbuf, 16);
941
                    break;
942
#endif
943
                }
944
                break;
945
#endif
438 theseven 946
            case DBGACTION_MALLOC:
947
                dbgasyncsendbuf[0] = 1;
948
                dbgasyncsendbuf[1] = (uint32_t)malloc((size_t)dbgactionlength);
949
                usb_drv_send_nonblocking(dbgendpoints[1], dbgasyncsendbuf, 16);
950
                break;
951
            case DBGACTION_MEMALIGN:
952
                dbgasyncsendbuf[0] = 1;
953
                dbgasyncsendbuf[1] = (uint32_t)memalign((size_t)dbgactionoffset,
954
                                                        (size_t)dbgactionlength);
955
                usb_drv_send_nonblocking(dbgendpoints[1], dbgasyncsendbuf, 16);
956
                break;
957
            case DBGACTION_REALLOC:
958
                dbgasyncsendbuf[0] = 1;
959
                dbgasyncsendbuf[1] = (uint32_t)realloc((void*)dbgactionaddr,
960
                                                       (size_t)dbgactionlength);
961
                usb_drv_send_nonblocking(dbgendpoints[1], dbgasyncsendbuf, 16);
962
                break;
963
            case DBGACTION_REOWNALLOC:
964
                dbgasyncsendbuf[0] = 1;
965
                reownalloc((void*)dbgactionaddr, (void*)dbgactionoffset);
966
                usb_drv_send_nonblocking(dbgendpoints[1], dbgasyncsendbuf, 16);
967
                break;
968
            case DBGACTION_FREE:
969
                dbgasyncsendbuf[0] = 1;
970
                free((void*)dbgactionaddr);
971
                usb_drv_send_nonblocking(dbgendpoints[1], dbgasyncsendbuf, 16);
972
                break;
226 theseven 973
            }
15 theseven 974
            dbgaction = DBGACTION_IDLE;
975
        }
976
    }
977
}
978
 
979
void usb_init(void)
980
{
981
    dbgaction = DBGACTION_IDLE;
982
    wakeup_init(&dbgwakeup);
25 theseven 983
    dbgconsendreadidx = 0;
984
    dbgconsendwriteidx = 0;
985
    dbgconrecvreadidx = 0;
986
    dbgconrecvwriteidx = 0;
987
    wakeup_init(&dbgconsendwakeup);
988
    wakeup_init(&dbgconrecvwakeup);
85 theseven 989
    dbgconsoleattached = false;
429 theseven 990
    thread_create(&dbgthread_handle, "monitor worker", dbgthread, dbgstack,
991
                  sizeof(dbgstack), CORE_THREAD, 255, true);
15 theseven 992
    usb_drv_init();
993
}
25 theseven 994
 
995
int dbgconsole_getfree() ICODE_ATTR;
996
int dbgconsole_getfree()
997
{
998
    int free = dbgconsendreadidx - dbgconsendwriteidx - 1;
999
    if (free < 0) free += sizeof(dbgconsendbuf);
1000
    return free;
1001
}
1002
 
1003
int dbgconsole_makespace(int length) ICODE_ATTR;
1004
int dbgconsole_makespace(int length)
1005
{
1006
    int free = dbgconsole_getfree();
1007
    while (!free && dbgconsoleattached)
1008
    {
336 theseven 1009
        dbgconsoleattached = false;
1010
        wakeup_wait(&dbgconsendwakeup, 2000000);
25 theseven 1011
        free = dbgconsole_getfree();
1012
    }
1013
    if (free) return free > length ? length : free;
1014
    if (length > sizeof(dbgconsendbuf) - 17) length = sizeof(dbgconsendbuf) - 17;
1015
    uint32_t mode = enter_critical_section();
1016
    dbgconsendreadidx += length;
1017
    if (dbgconsendreadidx >= sizeof(dbgconsendbuf))
1018
        dbgconsendreadidx -= sizeof(dbgconsendbuf);
1019
    int offset = 0;
1020
    int idx = dbgconsendreadidx;
1021
    if (idx + 16 >= sizeof(dbgconsendbuf))
1022
    {
1023
        offset = sizeof(dbgconsendbuf) - dbgconsendreadidx;
1024
        memcpy(&dbgconsendbuf[dbgconsendreadidx], dbgconoverflowstr, offset);
1025
        idx = 0;
1026
    }
1027
    if (offset != 16) memcpy(&dbgconsendbuf[idx], &dbgconoverflowstr[offset], 16 - offset);
1028
    leave_critical_section(mode);
1029
    return length;
1030
}
1031
 
1032
void dbgconsole_putc(char string)
1033
{
1034
    dbgconsole_makespace(1);
1035
    dbgconsendbuf[dbgconsendwriteidx++] = string;
1036
    if (dbgconsendwriteidx >= sizeof(dbgconsendbuf))
1037
        dbgconsendwriteidx -= sizeof(dbgconsendbuf);
1038
}
1039
 
1040
void dbgconsole_write(const char* string, size_t length)
1041
{
1042
    while (length)
1043
    {
1044
        int space = dbgconsole_makespace(length);
1045
        if (dbgconsendwriteidx + space >= sizeof(dbgconsendbuf))
1046
        {
1047
            int bytes = sizeof(dbgconsendbuf) - dbgconsendwriteidx;
1048
            memcpy(&dbgconsendbuf[dbgconsendwriteidx], string, bytes);
1049
            dbgconsendwriteidx = 0;
1050
            string = &string[bytes];
1051
            space -= bytes;
1052
            length -= bytes;
1053
        }
1054
        if (space) memcpy(&dbgconsendbuf[dbgconsendwriteidx], string, space);
1055
        dbgconsendwriteidx += space;
1056
        string = &string[space];
1057
        length -= space;
1058
    }
1059
}
1060
 
1061
void dbgconsole_puts(const char* string)
1062
{
1063
    dbgconsole_write(string, strlen(string));
1064
}
26 theseven 1065
 
1066
int dbgconsole_getavailable() ICODE_ATTR;
1067
int dbgconsole_getavailable()
1068
{
1069
    int available = dbgconrecvwriteidx - dbgconrecvreadidx;
1070
    if (available < 0) available += sizeof(dbgconrecvbuf);
1071
    return available;
1072
}
1073
 
1074
int dbgconsole_getc(int timeout)
1075
{
1076
    if (!dbgconsole_getavailable())
1077
    {
1078
        wakeup_wait(&dbgconrecvwakeup, TIMEOUT_NONE);
1079
        if (!dbgconsole_getavailable())
1080
        {
1081
            wakeup_wait(&dbgconrecvwakeup, timeout);
1082
            if (!dbgconsole_getavailable()) return -1;
1083
        }
1084
    }
1085
    int byte = dbgconrecvbuf[dbgconrecvreadidx++];
1086
    if (dbgconrecvreadidx >= sizeof(dbgconrecvbuf))
1087
        dbgconrecvreadidx -= sizeof(dbgconrecvbuf);
1088
    return byte;
1089
}
1090
 
1091
int dbgconsole_read(char* buffer, size_t length, int timeout)
1092
{
1093
    if (!length) return 0;
1094
    int available = dbgconsole_getavailable();
1095
    if (!available)
1096
    {
1097
        wakeup_wait(&dbgconrecvwakeup, TIMEOUT_NONE);
1098
        int available = dbgconsole_getavailable();
1099
        if (!available)
1100
        {
1101
            wakeup_wait(&dbgconrecvwakeup, timeout);
1102
            int available = dbgconsole_getavailable();
1103
            if (!available) return 0;
1104
        }
1105
    }
1106
    if (available > length) available = length;
1107
    int left = available;
1108
    if (dbgconrecvreadidx + available >= sizeof(dbgconrecvbuf))
1109
    {
1110
        int bytes = sizeof(dbgconrecvbuf) - dbgconrecvreadidx;
1111
        memcpy(buffer, &dbgconrecvbuf[dbgconrecvreadidx], bytes);
1112
        dbgconrecvreadidx = 0;
1113
        buffer = &buffer[bytes];
1114
        left -= bytes;
1115
    }
1116
    if (left) memcpy(buffer, &dbgconrecvbuf[dbgconrecvreadidx], left);
1117
    dbgconrecvreadidx += left;
1118
    return available;
1119
}
240 theseven 1120
 
1121
void usb_exit(void)
1122
{
1123
    usb_drv_exit();
1124
}