Merge remote-tracking branch 'upstream/master'

[proxmark3-svn] / common / usb_cdc.c

diff --git a/common/usb_cdc.c b/common/usb_cdc.c
index 049cb8cf03de52eec8eb95c77bd93cbc47b4b6c6..9d621777fe805dc0a290cf87c442874417ea0d9c 100644 (file)
--- a/common/usb_cdc.c
+++ b/common/usb_cdc.c
@@ -46,7 +46,8 @@ const char devDescriptor[] = {
        /* Device descriptor */\r
        0x12,      // bLength\r
        0x01,      // bDescriptorType\r
        /* Device descriptor */\r
        0x12,      // bLength\r
        0x01,      // bDescriptorType\r

-       0x10,0x01, // Complies with USB Spec. Release (0110h = release 1.10)\r
+       //0x10,0x01, // Complies with USB Spec. Release (0110h = release 1.10)\r
+       0x00,0x02, // Complies with USB Spec. Release (0110h = release 2.00)\r

        0x02,      // bDeviceClass:    CDC class code\r
        0x00,      // bDeviceSubclass: CDC class sub code\r
        0x00,      // bDeviceProtocol: CDC Device protocol\r
        0x02,      // bDeviceClass:    CDC class code\r
        0x00,      // bDeviceSubclass: CDC class sub code\r
        0x00,      // bDeviceProtocol: CDC Device protocol\r


@@ -56,7 +57,7 @@ const char devDescriptor[] = {
        0x01,0x00, // Device release number (0001)\r
        0x01,      // iManufacturer    // 0x01\r
        0x00,      // iProduct\r
        0x01,0x00, // Device release number (0001)\r
        0x01,      // iManufacturer    // 0x01\r
        0x00,      // iProduct\r

-       0x00,      // SerialNumber\r
+       0xFD,      // SerialNumber\r

        0x01       // bNumConfigs\r
 };\r
 \r
        0x01       // bNumConfigs\r
 };\r
 \r


@@ -71,7 +72,7 @@ const char cfgDescriptor[] = {
        0x01,   // CbConfigurationValue\r
        0x00,   // CiConfiguration\r
        0xC0,   // CbmAttributes 0xA0\r
        0x01,   // CbConfigurationValue\r
        0x00,   // CiConfiguration\r
        0xC0,   // CbmAttributes 0xA0\r

-       0x00,   // CMaxPower\r
+       0xFA,   // CMaxPower\r

 \r
        /* Communication Class Interface Descriptor Requirement */\r
        0x09, // bLength\r
 \r
        /* Communication Class Interface Descriptor Requirement */\r
        0x09, // bLength\r


@@ -81,7 +82,7 @@ const char cfgDescriptor[] = {
        0x01, // bNumEndpoints\r
        0x02, // bInterfaceClass\r
        0x02, // bInterfaceSubclass\r
        0x01, // bNumEndpoints\r
        0x02, // bInterfaceClass\r
        0x02, // bInterfaceSubclass\r

-       0x00, // bInterfaceProtocol\r
+       0x01, // bInterfaceProtocol\r

        0x00, // iInterface\r
 \r
        /* Header Functional Descriptor */\r
        0x00, // iInterface\r
 \r
        /* Header Functional Descriptor */\r


@@ -95,7 +96,7 @@ const char cfgDescriptor[] = {
        0x04, // bFunctionLength\r
        0x24, // bDescriptor Type: CS_INTERFACE\r
        0x02, // bDescriptor Subtype: ACM Func Desc\r
        0x04, // bFunctionLength\r
        0x24, // bDescriptor Type: CS_INTERFACE\r
        0x02, // bDescriptor Subtype: ACM Func Desc\r

-       0x00, // bmCapabilities\r
+       0x02, // bmCapabilities\r

 \r
        /* Union Functional Descriptor */\r
        0x05, // bFunctionLength\r
 \r
        /* Union Functional Descriptor */\r
        0x05, // bFunctionLength\r


@@ -152,7 +153,7 @@ const char cfgDescriptor[] = {
 };\r
 \r
 const char strDescriptor[] = {\r
 };\r
 \r
 const char strDescriptor[] = {\r

-  26,                          // Length\r
+  26,                  // Length\r

   0x03,                        // Type is string\r
   'p', 0x00,\r
   'r', 0x00,\r
   0x03,                        // Type is string\r
   'p', 0x00,\r
   'r', 0x00,\r


@@ -168,7 +169,32 @@ const char strDescriptor[] = {
   'g', 0x00,\r
 };\r
 \r
   'g', 0x00,\r
 };\r
 \r

+// Bitmap for all status bits in CSR.\r
+#define REG_NO_EFFECT_1_ALL      AT91C_UDP_RX_DATA_BK0 | AT91C_UDP_RX_DATA_BK1 \\r
+                                |AT91C_UDP_STALLSENT   | AT91C_UDP_RXSETUP \\r
+                                |AT91C_UDP_TXCOMP\r
+\r
+// Clear flags in the UDP_CSR register and waits for synchronization\r
+#define UDP_CLEAR_EP_FLAGS(endpoint, flags) { \\r
+       volatile unsigned int reg; \\r
+       reg = pUdp->UDP_CSR[(endpoint)]; \\r
+       reg |= REG_NO_EFFECT_1_ALL; \\r
+       reg &= ~(flags); \\r
+       pUdp->UDP_CSR[(endpoint)] = reg; \\r
+       while ( (pUdp->UDP_CSR[(endpoint)] & (flags)) == (flags)); \\r
+} \\r
+\r
+// reset flags in the UDP_CSR register and waits for synchronization\r
+#define UDP_SET_EP_FLAGS(endpoint, flags) { \\r
+       volatile unsigned int reg; \\r
+       reg = pUdp->UDP_CSR[(endpoint)]; \\r
+       reg |= REG_NO_EFFECT_1_ALL; \\r
+       reg |= (flags); \\r
+       pUdp->UDP_CSR[(endpoint)] = reg; \\r
+       while ( ( pUdp->UDP_CSR[(endpoint)] & (flags)) != (flags)); \\r
+} \\r

 \r
 \r

+       \r

 /* USB standard request code */\r
 #define STD_GET_STATUS_ZERO           0x0080\r
 #define STD_GET_STATUS_INTERFACE      0x0081\r
 /* USB standard request code */\r
 #define STD_GET_STATUS_ZERO           0x0080\r
 #define STD_GET_STATUS_INTERFACE      0x0081\r


@@ -221,13 +247,13 @@ byte_t btReceiveBank   = AT91C_UDP_RX_DATA_BK0;
 //* \brief This function deactivates the USB device\r
 //*----------------------------------------------------------------------------\r
 void usb_disable() {\r
 //* \brief This function deactivates the USB device\r
 //*----------------------------------------------------------------------------\r
 void usb_disable() {\r

-  // Disconnect the USB device\r
-  AT91C_BASE_PIOA->PIO_ODR = GPIO_USB_PU;\r
-  \r
-  // Clear all lingering interrupts\r
-  if(pUdp->UDP_ISR & AT91C_UDP_ENDBUSRES) {\r
-    pUdp->UDP_ICR = AT91C_UDP_ENDBUSRES;\r
-  }\r
+       // Disconnect the USB device\r
+       AT91C_BASE_PIOA->PIO_ODR = GPIO_USB_PU;\r
+\r
+       // Clear all lingering interrupts\r
+       if(pUdp->UDP_ISR & AT91C_UDP_ENDBUSRES) {\r
+               pUdp->UDP_ICR = AT91C_UDP_ENDBUSRES;\r
+       }\r

 }\r
 \r
 //*----------------------------------------------------------------------------\r
 }\r
 \r
 //*----------------------------------------------------------------------------\r


@@ -290,8 +316,8 @@ bool usb_check() {
 \r
 bool usb_poll()\r
 {\r
 \r
 bool usb_poll()\r
 {\r

-  if (!usb_check()) return false;\r
-  return (pUdp->UDP_CSR[AT91C_EP_OUT] & btReceiveBank);\r
+       if (!usb_check()) return false;\r
+       return (pUdp->UDP_CSR[AT91C_EP_OUT] & btReceiveBank);\r

 }\r
 \r
 /**\r
 }\r
 \r
 /**\r


@@ -326,9 +352,11 @@ uint32_t usb_read(byte_t* data, size_t len) {
                        len -= packetSize;\r
                        while(packetSize--)\r
                                data[nbBytesRcv++] = pUdp->UDP_FDR[AT91C_EP_OUT];\r
                        len -= packetSize;\r
                        while(packetSize--)\r
                                data[nbBytesRcv++] = pUdp->UDP_FDR[AT91C_EP_OUT];\r

+\r

                        \r
                        \r

-                       pUdp->UDP_CSR[AT91C_EP_OUT] &= ~(bank);\r
-                       \r
+                       UDP_CLEAR_EP_FLAGS(AT91C_EP_OUT, bank)                  \r
+                       //pUdp->UDP_CSR[AT91C_EP_OUT] &= ~(bank);\r
+\r

                        if (bank == AT91C_UDP_RX_DATA_BK0)\r
                                bank = AT91C_UDP_RX_DATA_BK1;\r
                        else\r
                        if (bank == AT91C_UDP_RX_DATA_BK0)\r
                                bank = AT91C_UDP_RX_DATA_BK1;\r
                        else\r


@@ -356,31 +384,36 @@ uint32_t usb_write(const byte_t* data, const size_t len) {
        cpt = MIN(length, AT91C_EP_IN_SIZE-1);\r
        length -= cpt;\r
        while (cpt--) pUdp->UDP_FDR[AT91C_EP_IN] = *data++;\r
        cpt = MIN(length, AT91C_EP_IN_SIZE-1);\r
        length -= cpt;\r
        while (cpt--) pUdp->UDP_FDR[AT91C_EP_IN] = *data++;\r

-       pUdp->UDP_CSR[AT91C_EP_IN] |= AT91C_UDP_TXPKTRDY;\r
+\r
+       UDP_SET_EP_FLAGS(AT91C_EP_IN, AT91C_UDP_TXPKTRDY)\r
+       //pUdp->UDP_CSR[AT91C_EP_IN] |= AT91C_UDP_TXPKTRDY;\r

 \r
        while (length) {\r
                // Fill the second bank\r
                cpt = MIN(length, AT91C_EP_IN_SIZE-1);\r
                length -= cpt;\r
                while (cpt--) pUdp->UDP_FDR[AT91C_EP_IN] = *data++;\r
 \r
        while (length) {\r
                // Fill the second bank\r
                cpt = MIN(length, AT91C_EP_IN_SIZE-1);\r
                length -= cpt;\r
                while (cpt--) pUdp->UDP_FDR[AT91C_EP_IN] = *data++;\r

-               // Wait for the the first bank to be sent\r
+               // Wait for the first bank to be sent\r

                while (!(pUdp->UDP_CSR[AT91C_EP_IN] & AT91C_UDP_TXCOMP)) {\r
                        if (!usb_check()) return length;\r
                }\r
                while (!(pUdp->UDP_CSR[AT91C_EP_IN] & AT91C_UDP_TXCOMP)) {\r
                        if (!usb_check()) return length;\r
                }\r

-               pUdp->UDP_CSR[AT91C_EP_IN] &= ~(AT91C_UDP_TXCOMP);\r

                \r
                \r

-               while (pUdp->UDP_CSR[AT91C_EP_IN] & AT91C_UDP_TXCOMP);\r
+               UDP_CLEAR_EP_FLAGS(AT91C_EP_IN, AT91C_UDP_TXCOMP)\r
+               //pUdp->UDP_CSR[AT91C_EP_IN] &= ~(AT91C_UDP_TXCOMP);            \r
+               //while (pUdp->UDP_CSR[AT91C_EP_IN] & AT91C_UDP_TXCOMP);\r

                \r
                \r

-               pUdp->UDP_CSR[AT91C_EP_IN] |= AT91C_UDP_TXPKTRDY;\r
+               UDP_SET_EP_FLAGS(AT91C_EP_IN, AT91C_UDP_TXPKTRDY)\r
+               //pUdp->UDP_CSR[AT91C_EP_IN] |= AT91C_UDP_TXPKTRDY;\r

        }\r
   \r
        // Wait for the end of transfer\r
        while (!(pUdp->UDP_CSR[AT91C_EP_IN] & AT91C_UDP_TXCOMP)) {\r
                if (!usb_check()) return length;\r
        }\r
        }\r
   \r
        // Wait for the end of transfer\r
        while (!(pUdp->UDP_CSR[AT91C_EP_IN] & AT91C_UDP_TXCOMP)) {\r
                if (!usb_check()) return length;\r
        }\r

-  \r
-       pUdp->UDP_CSR[AT91C_EP_IN] &= ~(AT91C_UDP_TXCOMP);\r
-       while (pUdp->UDP_CSR[AT91C_EP_IN] & AT91C_UDP_TXCOMP);\r
+\r
+       UDP_CLEAR_EP_FLAGS(AT91C_EP_IN, AT91C_UDP_TXCOMP)\r
+       //pUdp->UDP_CSR[AT91C_EP_IN] &= ~(AT91C_UDP_TXCOMP);\r
+       //while (pUdp->UDP_CSR[AT91C_EP_IN] & AT91C_UDP_TXCOMP);\r

 \r
        return length;\r
 }\r
 \r
        return length;\r
 }\r


@@ -404,17 +437,22 @@ static void AT91F_USB_SendData(AT91PS_UDP pUdp, const char *pData, uint32_t leng
                        pUdp->UDP_FDR[0] = *pData++;\r
 \r
                if (pUdp->UDP_CSR[0] & AT91C_UDP_TXCOMP) {\r
                        pUdp->UDP_FDR[0] = *pData++;\r
 \r
                if (pUdp->UDP_CSR[0] & AT91C_UDP_TXCOMP) {\r

-                       pUdp->UDP_CSR[0] &= ~(AT91C_UDP_TXCOMP);\r
-                       while (pUdp->UDP_CSR[0] & AT91C_UDP_TXCOMP);\r
+                       \r
+                       UDP_CLEAR_EP_FLAGS(0, AT91C_UDP_TXCOMP)\r
+                       //pUdp->UDP_CSR[0] &= ~(AT91C_UDP_TXCOMP);\r
+                       //while (pUdp->UDP_CSR[0] & AT91C_UDP_TXCOMP);\r

                }\r
 \r
                }\r
 \r

-               pUdp->UDP_CSR[0] |= AT91C_UDP_TXPKTRDY;\r
+               UDP_SET_EP_FLAGS(0, AT91C_UDP_TXPKTRDY)\r
+               // pUdp->UDP_CSR[0] |= AT91C_UDP_TXPKTRDY;\r

                do {\r
                        csr = pUdp->UDP_CSR[0];\r
 \r
                        // Data IN stage has been stopped by a status OUT\r
                        if (csr & AT91C_UDP_RX_DATA_BK0) {\r
                do {\r
                        csr = pUdp->UDP_CSR[0];\r
 \r
                        // Data IN stage has been stopped by a status OUT\r
                        if (csr & AT91C_UDP_RX_DATA_BK0) {\r

-                               pUdp->UDP_CSR[0] &= ~(AT91C_UDP_RX_DATA_BK0);\r
+                               \r
+                               UDP_CLEAR_EP_FLAGS(0, AT91C_UDP_RX_DATA_BK0)\r
+                               //pUdp->UDP_CSR[0] &= ~(AT91C_UDP_RX_DATA_BK0);\r

                                return;\r
                        }\r
                } while ( !(csr & AT91C_UDP_TXCOMP) );\r
                                return;\r
                        }\r
                } while ( !(csr & AT91C_UDP_TXCOMP) );\r


@@ -422,8 +460,10 @@ static void AT91F_USB_SendData(AT91PS_UDP pUdp, const char *pData, uint32_t leng
        } while (length);\r
 \r
        if (pUdp->UDP_CSR[0] & AT91C_UDP_TXCOMP) {\r
        } while (length);\r
 \r
        if (pUdp->UDP_CSR[0] & AT91C_UDP_TXCOMP) {\r

-               pUdp->UDP_CSR[0] &= ~(AT91C_UDP_TXCOMP);\r
-               while (pUdp->UDP_CSR[0] & AT91C_UDP_TXCOMP);\r
+               \r
+               UDP_CLEAR_EP_FLAGS(0, AT91C_UDP_TXCOMP)\r
+               //pUdp->UDP_CSR[0] &= ~(AT91C_UDP_TXCOMP);\r
+               //while (pUdp->UDP_CSR[0] & AT91C_UDP_TXCOMP);\r

        }\r
 }\r
 \r
        }\r
 }\r
 \r


@@ -432,10 +472,15 @@ static void AT91F_USB_SendData(AT91PS_UDP pUdp, const char *pData, uint32_t leng
 //* \brief Send zero length packet through the control endpoint\r
 //*----------------------------------------------------------------------------\r
 void AT91F_USB_SendZlp(AT91PS_UDP pUdp) {\r
 //* \brief Send zero length packet through the control endpoint\r
 //*----------------------------------------------------------------------------\r
 void AT91F_USB_SendZlp(AT91PS_UDP pUdp) {\r

-       pUdp->UDP_CSR[0] |= AT91C_UDP_TXPKTRDY;\r
+\r
+       UDP_SET_EP_FLAGS(0, AT91C_UDP_TXPKTRDY)\r
+       //pUdp->UDP_CSR[0] |= AT91C_UDP_TXPKTRDY;\r
+       \r

        while ( !(pUdp->UDP_CSR[0] & AT91C_UDP_TXCOMP) );\r
        while ( !(pUdp->UDP_CSR[0] & AT91C_UDP_TXCOMP) );\r

-       pUdp->UDP_CSR[0] &= ~(AT91C_UDP_TXCOMP);\r
-       while (pUdp->UDP_CSR[0] & AT91C_UDP_TXCOMP);\r
+       \r
+       UDP_CLEAR_EP_FLAGS(0, AT91C_UDP_TXCOMP)\r
+       //pUdp->UDP_CSR[0] &= ~(AT91C_UDP_TXCOMP);\r
+       //while (pUdp->UDP_CSR[0] & AT91C_UDP_TXCOMP);\r

 }\r
 \r
 //*----------------------------------------------------------------------------\r
 }\r
 \r
 //*----------------------------------------------------------------------------\r


@@ -445,6 +490,7 @@ void AT91F_USB_SendZlp(AT91PS_UDP pUdp) {
 void AT91F_USB_SendStall(AT91PS_UDP pUdp) {\r
        pUdp->UDP_CSR[0] |= AT91C_UDP_FORCESTALL;\r
        while ( !(pUdp->UDP_CSR[0] & AT91C_UDP_ISOERROR) );\r
 void AT91F_USB_SendStall(AT91PS_UDP pUdp) {\r
        pUdp->UDP_CSR[0] |= AT91C_UDP_FORCESTALL;\r
        while ( !(pUdp->UDP_CSR[0] & AT91C_UDP_ISOERROR) );\r

+       \r

        pUdp->UDP_CSR[0] &= ~(AT91C_UDP_FORCESTALL | AT91C_UDP_ISOERROR);\r
        while (pUdp->UDP_CSR[0] & (AT91C_UDP_FORCESTALL | AT91C_UDP_ISOERROR));\r
 }\r
        pUdp->UDP_CSR[0] &= ~(AT91C_UDP_FORCESTALL | AT91C_UDP_ISOERROR);\r
        while (pUdp->UDP_CSR[0] & (AT91C_UDP_FORCESTALL | AT91C_UDP_ISOERROR));\r
 }\r


@@ -470,11 +516,15 @@ void AT91F_CDC_Enumerate() {
        wLength      |= (pUdp->UDP_FDR[0] << 8);\r
 \r
        if (bmRequestType & 0x80) {\r
        wLength      |= (pUdp->UDP_FDR[0] << 8);\r
 \r
        if (bmRequestType & 0x80) {\r

-               pUdp->UDP_CSR[0] |= AT91C_UDP_DIR;\r
-               while ( !(pUdp->UDP_CSR[0] & AT91C_UDP_DIR) );\r
+\r
+               UDP_SET_EP_FLAGS(0, AT91C_UDP_DIR)\r
+               //pUdp->UDP_CSR[0] |= AT91C_UDP_DIR;\r
+               //while ( !(pUdp->UDP_CSR[0] & AT91C_UDP_DIR) );\r

        }\r
        }\r

-       pUdp->UDP_CSR[0] &= ~AT91C_UDP_RXSETUP;\r
-       while ( (pUdp->UDP_CSR[0]  & AT91C_UDP_RXSETUP)  );\r
+       \r
+       UDP_CLEAR_EP_FLAGS(0, AT91C_UDP_RXSETUP)\r
+       //pUdp->UDP_CSR[0] &= ~AT91C_UDP_RXSETUP;\r
+       //while ( (pUdp->UDP_CSR[0]  & AT91C_UDP_RXSETUP)  );\r

 \r
        // Handle supported standard device request Cf Table 9-3 in USB specification Rev 1.1\r
        switch ((bRequest << 8) | bmRequestType) {\r
 \r
        // Handle supported standard device request Cf Table 9-3 in USB specification Rev 1.1\r
        switch ((bRequest << 8) | bmRequestType) {\r


@@ -565,7 +615,8 @@ void AT91F_CDC_Enumerate() {
        // handle CDC class requests\r
        case SET_LINE_CODING:\r
                while ( !(pUdp->UDP_CSR[0] & AT91C_UDP_RX_DATA_BK0) );\r
        // handle CDC class requests\r
        case SET_LINE_CODING:\r
                while ( !(pUdp->UDP_CSR[0] & AT91C_UDP_RX_DATA_BK0) );\r

-               pUdp->UDP_CSR[0] &= ~(AT91C_UDP_RX_DATA_BK0);\r
+               UDP_CLEAR_EP_FLAGS(0, AT91C_UDP_RX_DATA_BK0)\r
+               //pUdp->UDP_CSR[0] &= ~(AT91C_UDP_RX_DATA_BK0);\r

                AT91F_USB_SendZlp(pUdp);\r
                break;\r
        case GET_LINE_CODING:\r
                AT91F_USB_SendZlp(pUdp);\r
                break;\r
        case GET_LINE_CODING:\r