DFU: increase detach timeout to 2 seconds

[clinton/Smoothieware.git] / src / libs / USBDevice / USBSerial / USBSerial.cpp

/* Copyright (c) 2010-2011 mbed.org, MIT License\r
*\r
* Permission is hereby granted, free of charge, to any person obtaining a copy of this software\r
* and associated documentation files (the "Software"), to deal in the Software without\r
* restriction, including without limitation the rights to use, copy, modify, merge, publish,\r
* distribute, sublicense, and/or sell copies of the Software, and to permit persons to whom the\r
* Software is furnished to do so, subject to the following conditions:\r
*\r
* The above copyright notice and this permission notice shall be included in all copies or\r
* substantial portions of the Software.\r
*\r
* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR IMPLIED, INCLUDING\r
* BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND\r
* NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM,\r
* DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\r
* OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.\r
*/\r
\r
#include <cstdint>\r
#include <cstdio>\r
\r
#include "USBSerial.h"\r
\r
#include "libs/Kernel.h"\r
#include "libs/SerialMessage.h"\r
\r
// extern void setled(int, bool);\r
#define setled(a, b) do {} while (0)\r
\r
// #define iprintf(...)\r
\r
USBSerial::USBSerial(USB *u): USBCDC(u), rxbuf(128), txbuf(128)\r
{\r
    usb = u;\r
    nl_in_rx = 0;\r
}\r
\r
int USBSerial::_putc(int c)\r
{\r
//     send((uint8_t *)&c, 1);\r
    if (txbuf.free())\r
        txbuf.queue(c);\r
\r
    usb->endpointSetInterrupt(CDC_BulkIn.bEndpointAddress, true);\r
//     usb->endpointTriggerInterrupt(CDC_BulkIn.bEndpointAddress);\r
    return 1;\r
}\r
\r
int USBSerial::_getc()\r
{\r
    uint8_t c = 0;\r
    setled(4, 1); while (rxbuf.isEmpty()); setled(4, 0);\r
    rxbuf.dequeue(&c);\r
    if (rxbuf.free() == MAX_PACKET_SIZE_EPBULK)\r
    {\r
        usb->endpointSetInterrupt(CDC_BulkOut.bEndpointAddress, true);\r
        iprintf("rxbuf has room for another packet, interrupt enabled\n");\r
//         usb->endpointTriggerInterrupt(CDC_BulkOut.bEndpointAddress);\r
    }\r
    if (nl_in_rx > 0)\r
        if (c == '\n')\r
            nl_in_rx--;\r
\r
    return c;\r
}\r
\r
uint16_t USBSerial::writeBlock(uint8_t * buf, uint16_t size)\r
{\r
    if (size > txbuf.free())\r
    {\r
        size = txbuf.free();\r
    }\r
    if (size > 0)\r
    {\r
        for (uint8_t i = 0; i < size; i++)\r
        {\r
            txbuf.queue(buf[i]);\r
        }\r
        usb->endpointSetInterrupt(CDC_BulkIn.bEndpointAddress, true);\r
    }\r
    return size;\r
}\r
\r
bool USBSerial::USBEvent_EPIn(uint8_t bEP, uint8_t bEPStatus)\r
{\r
    /*\r
     * Called in ISR context\r
     */\r
\r
//     static bool needToSendNull = false;\r
\r
    bool r = true;\r
\r
    if (bEP != CDC_BulkIn.bEndpointAddress)\r
        return false;\r
\r
    iprintf("USBSerial:EpIn: 0x%02X\n", bEPStatus);\r
\r
    uint8_t b[MAX_PACKET_SIZE_EPBULK];\r
\r
    int l = txbuf.available();\r
    iprintf("%d bytes queued\n", l);\r
    if (l > 0)\r
    {\r
        if (l > MAX_PACKET_SIZE_EPBULK)\r
            l = MAX_PACKET_SIZE_EPBULK;\r
        iprintf("Sending %d bytes:\n\t", l);\r
        int i;\r
        for (i = 0; i < l; i++) {\r
            txbuf.dequeue(&b[i]);\r
            if (b[i] >= 32 && b[i] < 128)\r
                iprintf("%c", b[i]);\r
            else {\r
                iprintf("\\x%02X", b[i]);\r
            }\r
        }\r
        iprintf("\nSending...\n");\r
        send(b, l);\r
        iprintf("Sent\n");\r
//         if (l == 64 && txbuf.available() == 0)\r
//             needToSendNull = true;\r
        if (txbuf.available() == 0)\r
            r = false;\r
    }\r
    else\r
    {\r
//         if (needToSendNull)\r
//         {\r
//             send(NULL, 0);\r
//             needToSendNull = false;\r
//         }\r
//         else\r
//         {\r
            r = false;\r
//         }\r
//         usb->endpointSetInterrupt(bEP, false);\r
    }\r
    iprintf("USBSerial:EpIn Complete\n");\r
    return r;\r
}\r
\r
bool USBSerial::USBEvent_EPOut(uint8_t bEP, uint8_t bEPStatus)\r
{\r
    /*\r
     * Called in ISR context\r
     */\r
\r
    bool r = true;\r
\r
    iprintf("USBSerial:EpOut\n");\r
    if (bEP != CDC_BulkOut.bEndpointAddress)\r
        return false;\r
\r
    if (rxbuf.free() < MAX_PACKET_SIZE_EPBULK)\r
    {\r
//         usb->endpointSetInterrupt(bEP, false);\r
        return false;\r
    }\r
\r
    uint8_t c[MAX_PACKET_SIZE_EPBULK];\r
    uint32_t size = 64;\r
\r
    //we read the packet received and put it on the circular buffer\r
    readEP(c, &size);\r
    iprintf("Read %ld bytes:\n\t", size);\r
    for (uint8_t i = 0; i < size; i++) {\r
        rxbuf.queue(c[i]);\r
        if (c[i] >= 32 && c[i] < 128)\r
        {\r
            iprintf("%c", c[i]);\r
        }\r
        else\r
        {\r
            iprintf("\\x%02X", c[i]);\r
        }\r
        if (c[i] == '\n')\r
            nl_in_rx++;\r
    }\r
    iprintf("\nQueued, %d empty\n", rxbuf.free());\r
\r
    if (rxbuf.free() < MAX_PACKET_SIZE_EPBULK)\r
    {\r
//         usb->endpointSetInterrupt(bEP, false);\r
        r = false;\r
    }\r
\r
    usb->readStart(CDC_BulkOut.bEndpointAddress, MAX_PACKET_SIZE_EPBULK);\r
    iprintf("USBSerial:EpOut Complete\n");\r
    return r;\r
}\r
\r
/*\r
bool USBSerial::EpCallback(uint8_t bEP, uint8_t bEPStatus) {\r
    if (bEP == CDC_BulkOut.bEndpointAddress) {\r
        uint8_t c[65];\r
        uint32_t size = 0;\r
\r
        //we read the packet received and put it on the circular buffer\r
        readEP(c, &size);\r
        for (uint8_t i = 0; i < size; i++) {\r
            buf.queue(c[i]);\r
        }\r
\r
        //call a potential handler\r
        rx.call();\r
\r
        // We reactivate the endpoint to receive next characters\r
        usb->readStart(CDC_BulkOut.bEndpointAddress, MAX_PACKET_SIZE_EPBULK);\r
        return true;\r
    }\r
    return false;\r
}*/\r
\r
uint8_t USBSerial::available()\r
{\r
    return rxbuf.available();\r
}\r
\r
void USBSerial::on_module_loaded()\r
{\r
    this->register_for_event(ON_MAIN_LOOP);\r
//     this->kernel->streams->append_stream(this);\r
}\r
\r
void USBSerial::on_main_loop(void *argument)\r
{\r
//     this->kernel->streams->printf("!");\r
    if (nl_in_rx)\r
    {\r
        string received;\r
        while (available())\r
        {\r
            char c = _getc();\r
            if( c == '\n' )\r
            {\r
                struct SerialMessage message;\r
                message.message = received;\r
                message.stream = this;\r
                this->kernel->call_event(ON_CONSOLE_LINE_RECEIVED, &message );\r
                return;\r
            }\r
            else\r
            {\r
                received += c;\r
            }\r
        }\r
    }\r
}\r
\r
void USBSerial::on_attach()\r
{\r
    this->kernel->streams->append_stream(this);\r
    writeBlock((uint8_t *) "Smoothie\nok\n", 12);\r
}\r
\r
void USBSerial::on_detach()\r
{\r
    this->kernel->streams->remove_stream(this);\r
    txbuf.flush();\r
    rxbuf.flush();\r
    nl_in_rx = 0;\r
}\r