cdc_enumerate.c

关于usb的一段程序

              CLASS:PRINTER ;\
              DESCRIPTION:UC24 chipset ;";


#endif



/* USB standard request code */
#define STD_GET_STATUS_ZERO           0x0080
#define STD_GET_STATUS_INTERFACE      0x0081
#define STD_GET_STATUS_ENDPOINT       0x0082

#define STD_CLEAR_FEATURE_ZERO        0x0100
#define STD_CLEAR_FEATURE_INTERFACE   0x0101
#define STD_CLEAR_FEATURE_ENDPOINT    0x0102

#define STD_SET_FEATURE_ZERO          0x0300
#define STD_SET_FEATURE_INTERFACE     0x0301
#define STD_SET_FEATURE_ENDPOINT      0x0302

#define STD_SET_ADDRESS               0x0500
#define STD_GET_DESCRIPTOR            0x0680
#define STD_SET_DESCRIPTOR            0x0700
#define STD_GET_CONFIGURATION         0x0880
#define STD_SET_CONFIGURATION         0x0900
#define STD_GET_INTERFACE             0x0A81
#define STD_SET_INTERFACE             0x0B01
#define STD_SYNCH_FRAME               0x0C82

/* CDC Class Specific Request Code */
#define GET_LINE_CODING               0x21A1
#define SET_LINE_CODING               0x2021
#define SET_CONTROL_LINE_STATE        0x2221


/*Printer Class Specific Request Code*/
#define GET_DEVICE_ID                 0x00A1
#define GET_PORT_STATUS               0x01A1
#define SOFT_RESET                    0x0221


typedef struct {
    unsigned int dwDTERRate;
    char bCharFormat;
    char bParityType;
    char bDataBits;
} AT91S_CDC_LINE_CODING, *AT91PS_CDC_LINE_CODING;

AT91S_CDC_LINE_CODING line = {
    115200, // baudrate
    0,      // 1 Stop Bit
    0,      // None Parity
    8};     // 8 Data bits

uint currentReceiveBank = AT91C_UDP_RX_DATA_BK0;


static uchar AT91F_UDP_IsConfigured(AT91PS_CDC pCdc);
static uint AT91F_UDP_Read(AT91PS_CDC pCdc, char *pData, uint length);
static uint AT91F_UDP_Write(AT91PS_CDC pCdc, const char *pData, uint length);
static void AT91F_CDC_Enumerate(AT91PS_CDC pCdc);


//*----------------------------------------------------------------------------
//* \fn    AT91F_CDC_Open
//* \brief
//*----------------------------------------------------------------------------
AT91PS_CDC AT91F_CDC_Open(AT91PS_CDC pCdc, AT91PS_UDP pUdp)
{
    pCdc->pUdp = pUdp;
    pCdc->currentConfiguration = 0;
    pCdc->currentConnection    = 0;
    pCdc->currentRcvBank       = AT91C_UDP_RX_DATA_BK0;
    pCdc->IsConfigured = AT91F_UDP_IsConfigured;
    pCdc->Write        = AT91F_UDP_Write;
    pCdc->Read         = AT91F_UDP_Read;
    return pCdc;
}

//*----------------------------------------------------------------------------
//* \fn    AT91F_UDP_IsConfigured
//* \brief Test if the device is configured and handle enumeration
//*----------------------------------------------------------------------------
static uchar AT91F_UDP_IsConfigured(AT91PS_CDC pCdc)
{
    AT91PS_UDP pUDP = pCdc->pUdp;
    AT91_REG isr = pUDP->UDP_ISR;

    if (isr & AT91C_UDP_ENDBUSRES) {
        pUDP->UDP_ICR = AT91C_UDP_ENDBUSRES;
        // reset all endpoints
        pUDP->UDP_RSTEP  = (unsigned int)-1;
        pUDP->UDP_RSTEP  = 0;
        // Enable the function
        pUDP->UDP_FADDR = AT91C_UDP_FEN;
        // Configure endpoint 0
        pUDP->UDP_CSR[0] = (AT91C_UDP_EPEDS | AT91C_UDP_EPTYPE_CTRL);
    }
    else if (isr & AT91C_UDP_EPINT0) {
        pUDP->UDP_ICR = AT91C_UDP_EPINT0;
        AT91F_CDC_Enumerate(pCdc);
    }
    return pCdc->currentConfiguration;
}

//*----------------------------------------------------------------------------
//* \fn    AT91F_UDP_Read
//* \brief Read available data from Endpoint OUT
//*----------------------------------------------------------------------------
static uint AT91F_UDP_Read(AT91PS_CDC pCdc, char *pData, uint length)
{
    AT91PS_UDP pUdp = pCdc->pUdp;
    uint packetSize, nbBytesRcv = 0, currentReceiveBank = pCdc->currentRcvBank;

    while (length) {
        if ( !AT91F_UDP_IsConfigured(pCdc) )
            break;
        if ( pUdp->UDP_CSR[AT91C_EP_OUT] & currentReceiveBank ) {
            packetSize = MIN(pUdp->UDP_CSR[AT91C_EP_OUT] >> 16, length);
            length -= packetSize;
            if (packetSize < AT91C_EP_OUT_SIZE)
                length = 0;
            while(packetSize--)
                pData[nbBytesRcv++] = pUdp->UDP_FDR[AT91C_EP_OUT];
            pUdp->UDP_CSR[AT91C_EP_OUT] &= ~(currentReceiveBank);
            if (currentReceiveBank == AT91C_UDP_RX_DATA_BK0)
                currentReceiveBank = AT91C_UDP_RX_DATA_BK1;
            else
                currentReceiveBank = AT91C_UDP_RX_DATA_BK0;

        }
    }
    pCdc->currentRcvBank = currentReceiveBank;
    return nbBytesRcv;

}

//*----------------------------------------------------------------------------
//* \fn    AT91F_CDC_Write
//* \brief Send through endpoint 2
//*----------------------------------------------------------------------------
static uint AT91F_UDP_Write(AT91PS_CDC pCdc, const char *pData, uint length)
{
    AT91PS_UDP pUdp = pCdc->pUdp;
    uint cpt = 0;

    // Send the first packet
    cpt = MIN(length, AT91C_EP_IN_SIZE);
    length -= cpt;
    while (cpt--) pUdp->UDP_FDR[AT91C_EP_IN] = *pData++;
    pUdp->UDP_CSR[AT91C_EP_IN] |= AT91C_UDP_TXPKTRDY;

    while (length) {
        // Fill the second bank
        cpt = MIN(length, AT91C_EP_IN_SIZE);
        length -= cpt;
        while (cpt--) pUdp->UDP_FDR[AT91C_EP_IN] = *pData++;
        // Wait for the the first bank to be sent
        while ( !(pUdp->UDP_CSR[AT91C_EP_IN] & AT91C_UDP_TXCOMP) )
            if ( !AT91F_UDP_IsConfigured(pCdc) ) return length;
        pUdp->UDP_CSR[AT91C_EP_IN] &= ~(AT91C_UDP_TXCOMP);
        while (pUdp->UDP_CSR[AT91C_EP_IN] & AT91C_UDP_TXCOMP);
        pUdp->UDP_CSR[AT91C_EP_IN] |= AT91C_UDP_TXPKTRDY;
    }
    // Wait for the end of transfer
    while ( !(pUdp->UDP_CSR[AT91C_EP_IN] & AT91C_UDP_TXCOMP) )
        if ( !AT91F_UDP_IsConfigured(pCdc) ) return length;
    pUdp->UDP_CSR[AT91C_EP_IN] &= ~(AT91C_UDP_TXCOMP);
    while (pUdp->UDP_CSR[AT91C_EP_IN] & AT91C_UDP_TXCOMP);

    return length;
}

//*----------------------------------------------------------------------------
//* \fn    AT91F_USB_SendData
//* \brief Send Data through the control endpoint
//*----------------------------------------------------------------------------
unsigned int csrTab[100];
unsigned char csrIdx = 0;

static void AT91F_USB_SendData(AT91PS_UDP pUdp, const char *pData, uint length)
{
    uint cpt = 0;
    AT91_REG csr;

    do {
        cpt = MIN(length, 8);
        length -= cpt;

        while (cpt--)
            pUdp->UDP_FDR[0] = *pData++;

        if (pUdp->UDP_CSR[0] & AT91C_UDP_TXCOMP) {
            pUdp->UDP_CSR[0] &= ~(AT91C_UDP_TXCOMP);
            while (pUdp->UDP_CSR[0] & AT91C_UDP_TXCOMP);
        }

        pUdp->UDP_CSR[0] |= AT91C_UDP_TXPKTRDY;
        do {
            csr = pUdp->UDP_CSR[0];

            // Data IN stage has been stopped by a status OUT
            if (csr & AT91C_UDP_RX_DATA_BK0) {
                pUdp->UDP_CSR[0] &= ~(AT91C_UDP_RX_DATA_BK0);
                return;
            }
        } while ( !(csr & AT91C_UDP_TXCOMP) );

    } while (length);

    if (pUdp->UDP_CSR[0] & AT91C_UDP_TXCOMP) {
        pUdp->UDP_CSR[0] &= ~(AT91C_UDP_TXCOMP);
        while (pUdp->UDP_CSR[0] & AT91C_UDP_TXCOMP);
    }
}

//*----------------------------------------------------------------------------
//* \fn    AT91F_USB_SendZlp
//* \brief Send zero length packet through the control endpoint
//*----------------------------------------------------------------------------
void AT91F_USB_SendZlp(AT91PS_UDP pUdp)
{
    pUdp->UDP_CSR[0] |= AT91C_UDP_TXPKTRDY;
    while ( !(pUdp->UDP_CSR[0] & AT91C_UDP_TXCOMP) );
    pUdp->UDP_CSR[0] &= ~(AT91C_UDP_TXCOMP);
    while (pUdp->UDP_CSR[0] & AT91C_UDP_TXCOMP);
}

//*----------------------------------------------------------------------------
//* \fn    AT91F_USB_SendStall
//* \brief Stall the control endpoint
//*----------------------------------------------------------------------------
void AT91F_USB_SendStall(AT91PS_UDP pUdp)
{
    pUdp->UDP_CSR[0] |= AT91C_UDP_FORCESTALL;
    while ( !(pUdp->UDP_CSR[0] & AT91C_UDP_ISOERROR) );
    pUdp->UDP_CSR[0] &= ~(AT91C_UDP_FORCESTALL | AT91C_UDP_ISOERROR);
    while (pUdp->UDP_CSR[0] & (AT91C_UDP_FORCESTALL | AT91C_UDP_ISOERROR));
}

//*----------------------------------------------------------------------------
//* \fn    AT91F_CDC_Enumerate
//* \brief This function is a callback invoked when a SETUP packet is received
//*----------------------------------------------------------------------------
static void AT91F_CDC_Enumerate(AT91PS_CDC pCdc)
{
    AT91PS_UDP pUDP = pCdc->pUdp;
    uchar bmRequestType, bRequest;
    ushort wValue, wIndex, wLength, wStatus;


    if ( !(pUDP->UDP_CSR[0] & AT91C_UDP_RXSETUP) )
        return;

    bmRequestType = pUDP->UDP_FDR[0];
    bRequest      = pUDP->UDP_FDR[0];
    wValue        = (pUDP->UDP_FDR[0] & 0xFF);
    wValue       |= (pUDP->UDP_FDR[0] << 8);
    wIndex        = (pUDP->UDP_FDR[0] & 0xFF);
    wIndex       |= (pUDP->UDP_FDR[0] << 8);
    wLength       = (pUDP->UDP_FDR[0] & 0xFF);
    wLength      |= (pUDP->UDP_FDR[0] << 8);

    if (bmRequestType & 0x80) {
        pUDP->UDP_CSR[0] |= AT91C_UDP_DIR;
        while ( !(pUDP->UDP_CSR[0] & AT91C_UDP_DIR) );
    }
    pUDP->UDP_CSR[0] &= ~AT91C_UDP_RXSETUP;
    while ( (pUDP->UDP_CSR[0]  & AT91C_UDP_RXSETUP)  );

    // Handle supported standard device request Cf Table 9-3 in USB specification Rev 1.1
    switch ((bRequest << 8) | bmRequestType) {
    case STD_GET_DESCRIPTOR:
        if (wValue == 0x100)       // Return Device Descriptor
            AT91F_USB_SendData(pUDP, devDescriptor, MIN(sizeof(devDescriptor), wLength));
        else if (wValue == 0x200)  // Return Configuration Descriptor
            AT91F_USB_SendData(pUDP, cfgDescriptor, MIN(sizeof(cfgDescriptor), wLength));
        else if (wValue == 0x300)  // Return String Descriptor
            AT91F_USB_SendData(pUDP, usb_string0, MIN(sizeof(usb_string0), wLength));
        else if (wValue == 0x301)  // Return String Descriptor
            AT91F_USB_SendData(pUDP, usb_string1, MIN(sizeof(usb_string1), wLength));
        else if (wValue == 0x302)  // Return String Descriptor
            AT91F_USB_SendData(pUDP, usb_string2, MIN(sizeof(usb_string2), wLength));
        else
            AT91F_USB_SendStall(pUDP);
        break;
    case STD_SET_ADDRESS:
        AT91F_USB_SendZlp(pUDP);
        pUDP->UDP_FADDR = (AT91C_UDP_FEN | wValue);
        pUDP->UDP_GLBSTATE  = (wValue) ? AT91C_UDP_FADDEN : 0;
        break;
    case STD_SET_CONFIGURATION:
        pCdc->currentConfiguration = wValue;
        AT91F_USB_SendZlp(pUDP);
        pUDP->UDP_GLBSTATE  = (wValue) ? AT91C_UDP_CONFG : AT91C_UDP_FADDEN;
        pUDP->UDP_CSR[1] = (wValue) ? (AT91C_UDP_EPEDS | AT91C_UDP_EPTYPE_BULK_OUT) : 0;
        pUDP->UDP_CSR[2] = (wValue) ? (AT91C_UDP_EPEDS | AT91C_UDP_EPTYPE_BULK_IN)  : 0;
        pUDP->UDP_CSR[3] = (wValue) ? (AT91C_UDP_EPEDS | AT91C_UDP_EPTYPE_INT_IN)   : 0;
        break;
    case STD_GET_CONFIGURATION:
        AT91F_USB_SendData(pUDP, (char *) &(pCdc->currentConfiguration), sizeof(pCdc->currentConfiguration));
        break;
    case STD_GET_STATUS_ZERO:
        wStatus = 0;
        AT91F_USB_SendData(pUDP, (char *) &wStatus, sizeof(wStatus));
        break;
    case STD_GET_STATUS_INTERFACE:
        wStatus = 0;
        AT91F_USB_SendData(pUDP, (char *) &wStatus, sizeof(wStatus));
        break;
    case STD_GET_STATUS_ENDPOINT:
        wStatus = 0;
        wIndex &= 0x0F;
        if ((pUDP->UDP_GLBSTATE & AT91C_UDP_CONFG) && (wIndex <= 3)) {
            wStatus = (pUDP->UDP_CSR[wIndex] & AT91C_UDP_EPEDS) ? 0 : 1;
            AT91F_USB_SendData(pUDP, (char *) &wStatus, sizeof(wStatus));
        }
        else if ((pUDP->UDP_GLBSTATE & AT91C_UDP_FADDEN) && (wIndex == 0)) {
            wStatus = (pUDP->UDP_CSR[wIndex] & AT91C_UDP_EPEDS) ? 0 : 1;
            AT91F_USB_SendData(pUDP, (char *) &wStatus, sizeof(wStatus));
        }
        else
            AT91F_USB_SendStall(pUDP);
        break;
    case STD_SET_FEATURE_ZERO:
        AT91F_USB_SendStall(pUDP);
        break;
    case STD_SET_FEATURE_INTERFACE:
        AT91F_USB_SendZlp(pUDP);
        break;
    case STD_SET_FEATURE_ENDPOINT:
        wIndex &= 0x0F;
        if ((wValue == 0) && wIndex && (wIndex <= 3)) {
            pUDP->UDP_CSR[wIndex] = 0;
            AT91F_USB_SendZlp(pUDP);
        }
        else
            AT91F_USB_SendStall(pUDP);
        break;
    case STD_CLEAR_FEATURE_ZERO:
        AT91F_USB_SendStall(pUDP);
        break;
    case STD_CLEAR_FEATURE_INTERFACE:
        AT91F_USB_SendZlp(pUDP);
        break;
    case STD_CLEAR_FEATURE_ENDPOINT:
        wIndex &= 0x0F;
        if ((wValue == 0) && wIndex && (wIndex <= 3)) {
            if (wIndex == 1)
                pUDP->UDP_CSR[1] = (AT91C_UDP_EPEDS | AT91C_UDP_EPTYPE_BULK_OUT);
            else if (wIndex == 2)
                pUDP->UDP_CSR[2] = (AT91C_UDP_EPEDS | AT91C_UDP_EPTYPE_BULK_IN);
            else if (wIndex == 3)
                pUDP->UDP_CSR[3] = (AT91C_UDP_EPEDS | AT91C_UDP_EPTYPE_ISO_IN);
            AT91F_USB_SendZlp(pUDP);
        }
        else
            AT91F_USB_SendStall(pUDP);
        break;

    // handle CDC class requests
    case SET_LINE_CODING:
        while ( !(pUDP->UDP_CSR[0] & AT91C_UDP_RX_DATA_BK0) );
        pUDP->UDP_CSR[0] &= ~(AT91C_UDP_RX_DATA_BK0);
        AT91F_USB_SendZlp(pUDP);
        break;
    case GET_LINE_CODING:
        AT91F_USB_SendData(pUDP, (char *) &line, MIN(sizeof(line), wLength));
        break;
    case SET_CONTROL_LINE_STATE:
        pCdc->currentConnection = wValue;
        AT91F_USB_SendZlp(pUDP);
        break;

    // handle Printer class requests
    case GET_DEVICE_ID:
        AT91F_USB_SendData(pUDP, usb_device_id_descriptor, MIN(sizeof(usb_device_id_descriptor), wLength));
        break;
    case GET_PORT_STATUS:
        AT91F_USB_SendData(pUDP, &PrtPortStatus, 1);
        break;
    case SOFT_RESET:
        break;
    default:
        AT91F_USB_SendStall(pUDP);
        break;
    }
}