*/\r
\r
#include "usb_cdc.h"\r
+#include "at91sam7s512.h"\r
#include "config_gpio.h"\r
\r
-#define MIN(a, b) (((a) < (b)) ? (a) : (b))\r
-#define MAX(a, b) (((a) > (b)) ? (a) : (b))\r
#define AT91C_EP_IN_SIZE 0x40\r
#define AT91C_EP_OUT 1\r
#define AT91C_EP_OUT_SIZE 0x40\r
#define AT91C_EP_IN 2\r
\r
-const char devDescriptor[] = {\r
+static const char devDescriptor[] = {\r
/* Device descriptor */\r
0x12, // bLength\r
0x01, // bDescriptorType\r
- 0x10,0x01, // Complies with USB Spec. Release (0110h = release 1.10)\r
+ 0x00,0x02, // Complies with USB Spec. Release (0200h = release 2.0)\r
0x02, // bDeviceClass: CDC class code\r
0x00, // bDeviceSubclass: CDC class sub code\r
0x00, // bDeviceProtocol: CDC Device protocol\r
0x08, // bMaxPacketSize0\r
- 0x2d,0x2d, // Vendor ID (--)\r
- 0x4d,0x50, // Product ID (PM), transmitted in reverse\r
+ 0xc4,0x9a, // Vendor ID (0x9ac4 = J. Westhues)\r
+ 0x8f,0x4b, // Product ID (0x4b8f = Proxmark-3 RFID Instrument)\r
0x01,0x00, // Device release number (0001)\r
- 0x01, // iManufacturer // 0x01\r
- 0x00, // iProduct\r
- 0x00, // SerialNumber\r
+ 0x01, // iManufacturer\r
+ 0x02, // iProduct\r
+ 0x00, // iSerialNumber\r
0x01 // bNumConfigs\r
};\r
\r
-const char cfgDescriptor[] = {\r
+static const char cfgDescriptor[] = {\r
/* ============== CONFIGURATION 1 =========== */\r
/* Configuration 1 descriptor */\r
0x09, // CbLength\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
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
0x00 // bInterval\r
};\r
\r
-const char strDescriptor[] = {\r
- 26, // Length\r
+static const char StrDescLanguageCodes[] = {\r
+ 4, // Length\r
+ 0x03, // Type is string\r
+ 0x09, 0x04 // supported language Code 0 = 0x0409 (English)\r
+};\r
+ \r
+static const char StrDescManufacturer[] = {\r
+ 26, // Length\r
0x03, // Type is string\r
'p', 0x00,\r
'r', 0x00,\r
'.', 0x00,\r
'o', 0x00,\r
'r', 0x00,\r
- 'g', 0x00,\r
+ 'g', 0x00\r
};\r
\r
+static const char StrDescProduct[] = {\r
+ 8, // Length\r
+ 0x03, // Type is string\r
+ 'P', 0x00,\r
+ 'M', 0x00,\r
+ '3', 0x00\r
+};\r
+ \r
+static const char* const pStrings[] =\r
+{\r
+ StrDescLanguageCodes,\r
+ StrDescManufacturer,\r
+ StrDescProduct\r
+};\r
+\r
+const char* getStringDescriptor(uint8_t idx)\r
+{\r
+ if(idx >= (sizeof(pStrings) / sizeof(pStrings[0]))) {\r
+ return(NULL);\r
+ } else {\r
+ return(pStrings[idx]);\r
+ }\r
+}\r
\r
/* USB standard request code */\r
#define STD_GET_STATUS_ZERO 0x0080\r
void usb_disable() {\r
// Disconnect the USB device\r
AT91C_BASE_PIOA->PIO_ODR = GPIO_USB_PU;\r
-// SpinDelay(100);\r
\r
// Clear all lingering interrupts\r
if(pUdp->UDP_ISR & AT91C_UDP_ENDBUSRES) {\r
\r
// Wait for a short while\r
for (volatile size_t i=0; i<0x100000; i++);\r
-// SpinDelay(100);\r
\r
// Reconnect USB reconnect\r
AT91C_BASE_PIOA->PIO_SODR = GPIO_USB_PU;\r
return (pUdp->UDP_CSR[AT91C_EP_OUT] & btReceiveBank);\r
}\r
\r
+/**\r
+ In github PR #129, some users appears to get a false positive from\r
+ usb_poll, which returns true, but the usb_read operation\r
+ still returns 0.\r
+ This check is basically the same as above, but also checks\r
+ that the length available to read is non-zero, thus hopefully fixes the\r
+ bug.\r
+**/\r
+bool usb_poll_validate_length()\r
+{\r
+\r
+ if (!usb_check()) return false;\r
+ if (!(pUdp->UDP_CSR[AT91C_EP_OUT] & btReceiveBank)) return false;\r
+ return (pUdp->UDP_CSR[AT91C_EP_OUT] >> 16) > 0;\r
+}\r
+\r
//*----------------------------------------------------------------------------\r
//* \fn usb_read\r
//* \brief Read available data from Endpoint OUT\r
//*----------------------------------------------------------------------------\r
uint32_t usb_read(byte_t* data, size_t len) {\r
- byte_t bank = btReceiveBank;\r
+ byte_t bank = btReceiveBank;\r
uint32_t packetSize, nbBytesRcv = 0;\r
- uint32_t time_out = 0;\r
+ uint32_t time_out = 0;\r
\r
- while (len)\r
- {\r
+ while (len) {\r
if (!usb_check()) break;\r
\r
if ( pUdp->UDP_CSR[AT91C_EP_OUT] & bank ) {\r
packetSize = MIN(pUdp->UDP_CSR[AT91C_EP_OUT] >> 16, len);\r
- len -= packetSize;\r
+ len -= packetSize;\r
while(packetSize--)\r
data[nbBytesRcv++] = pUdp->UDP_FDR[AT91C_EP_OUT];\r
pUdp->UDP_CSR[AT91C_EP_OUT] &= ~(bank);\r
- if (bank == AT91C_UDP_RX_DATA_BK0)\r
- {\r
+ if (bank == AT91C_UDP_RX_DATA_BK0) {\r
bank = AT91C_UDP_RX_DATA_BK1;\r
- } else {\r
+ } else {\r
bank = AT91C_UDP_RX_DATA_BK0;\r
- }\r
+ }\r
}\r
- if (time_out++ == 0x1fff) break;\r
+ if (time_out++ == 0x1fff) break;\r
}\r
\r
btReceiveBank = 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
//* \fn AT91F_USB_SendData\r
//* \brief Send Data through the control endpoint\r
//*----------------------------------------------------------------------------\r
-unsigned int csrTab[100];\r
+unsigned int csrTab[100] = {0x00};\r
unsigned char csrIdx = 0;\r
\r
static void AT91F_USB_SendData(AT91PS_UDP pUdp, const char *pData, uint32_t length) {\r
AT91F_USB_SendData(pUdp, devDescriptor, MIN(sizeof(devDescriptor), wLength));\r
else if (wValue == 0x200) // Return Configuration Descriptor\r
AT91F_USB_SendData(pUdp, cfgDescriptor, MIN(sizeof(cfgDescriptor), wLength));\r
- else if ((wValue & 0x300) == 0x300) // Return String Descriptor\r
- AT91F_USB_SendData(pUdp, strDescriptor, MIN(sizeof(strDescriptor), wLength));\r
+ else if ((wValue & 0xF00) == 0x300) { // Return String Descriptor\r
+ const char *strDescriptor = getStringDescriptor(wValue & 0xff);\r
+ if (strDescriptor != NULL) {\r
+ AT91F_USB_SendData(pUdp, strDescriptor, MIN(strDescriptor[0], wLength));\r
+ } else {\r
+ AT91F_USB_SendStall(pUdp);\r
+ }\r
+ }\r
else\r
AT91F_USB_SendStall(pUdp);\r
break;\r
projects / proxmark3-svn / blobdiff