Demos/Device/LowLevel/USBtoSerial/USBtoSerial.c

   1 /*
   2              LUFA Library
   3      Copyright (C) Dean Camera, 2009.
   4
   5   dean [at] fourwalledcubicle [dot] com
   6       www.fourwalledcubicle.com
   7 */
   8
   9 /*
  10   Copyright 2009  Dean Camera (dean [at] fourwalledcubicle [dot] com)
  11
  12   Permission to use, copy, modify, and distribute this software
  13   and its documentation for any purpose and without fee is hereby
  14   granted, provided that the above copyright notice appear in all
  15   copies and that both that the copyright notice and this
  16   permission notice and warranty disclaimer appear in supporting
  17   documentation, and that the name of the author not be used in
  18   advertising or publicity pertaining to distribution of the
  19   software without specific, written prior permission.
  20
  21   The author disclaim all warranties with regard to this
  22   software, including all implied warranties of merchantability
  23   and fitness.  In no event shall the author be liable for any
  24   special, indirect or consequential damages or any damages
  25   whatsoever resulting from loss of use, data or profits, whether
  26   in an action of contract, negligence or other tortious action,
  27   arising out of or in connection with the use or performance of
  28   this software.
  29 */
  30
  31 #include "USBtoSerial.h"
  32
  33 /* Globals: */
  34 /** Contains the current baud rate and other settings of the virtual serial port.
  35  *
  36  *  These values are set by the host via a class-specific request, and the physical USART should be reconfigured to match the
  37  *  new settings each time they are changed by the host.
  38  */
  39 CDC_Line_Coding_t LineCoding = { .BaudRateBPS = 9600,
  40                                  .CharFormat  = OneStopBit,
  41                                  .ParityType  = Parity_None,
  42                                  .DataBits    = 8            };
  43
  44 /** Ring (circular) buffer to hold the RX data - data from the host to the attached device on the serial port. */
  45 RingBuff_t Rx_Buffer;
  46
  47 /** Ring (circular) buffer to hold the TX data - data from the attached device on the serial port to the host. */
  48 RingBuff_t Tx_Buffer;
  49
  50 /** Flag to indicate if the USART is currently transmitting data from the Rx_Buffer circular buffer. */
  51 volatile bool Transmitting = false;
  52
  53 /** Main program entry point. This routine configures the hardware required by the application, then
  54  *  starts the scheduler to run the application tasks.
  55  */
  56 int main(void)
  57 {
  58         SetupHardware();
  59
  60         /* Ring buffer Initialization */
  61         Buffer_Initialize(&Rx_Buffer);
  62         Buffer_Initialize(&Tx_Buffer);
  63
  64         LEDs_SetAllLEDs(LEDMASK_USB_NOTREADY);
  65
  66         for (;;)
  67         {
  68                 CDC_Task();
  69                 USB_USBTask();
  70         }
  71 }
  72
  73 /** Configures the board hardware and chip peripherals for the demo's functionality. */
  74 void SetupHardware(void)
  75 {
  76         /* Disable watchdog if enabled by bootloader/fuses */
  77         MCUSR &= ~(1 << WDRF);
  78         wdt_disable();
  79
  80         /* Disable clock division */
  81         clock_prescale_set(clock_div_1);
  82
  83         /* Hardware Initialization */
  84         LEDs_Init();
  85         ReconfigureUSART();
  86         USB_Init();
  87 }
  88
  89 /** Event handler for the USB_Connect event. This indicates that the device is enumerating via the status LEDs and
  90  *  starts the library USB task to begin the enumeration and USB management process.
  91  */
  92 void EVENT_USB_Connect(void)
  93 {
  94         /* Indicate USB enumerating */
  95         LEDs_SetAllLEDs(LEDMASK_USB_ENUMERATING);
  96 }
  97
  98 /** Event handler for the USB_Disconnect event. This indicates that the device is no longer connected to a host via
  99  *  the status LEDs and stops the USB management and CDC management tasks.
 100  */
 101 void EVENT_USB_Disconnect(void)
 102 {
 103         /* Reset Tx and Rx buffers, device disconnected */
 104         Buffer_Initialize(&Rx_Buffer);
 105         Buffer_Initialize(&Tx_Buffer);
 106
 107         /* Indicate USB not ready */
 108         LEDs_SetAllLEDs(LEDMASK_USB_NOTREADY);
 109 }
 110
 111 /** Event handler for the USB_ConfigurationChanged event. This is fired when the host set the current configuration
 112  *  of the USB device after enumeration - the device endpoints are configured and the CDC management task started.
 113  */
 114 void EVENT_USB_ConfigurationChanged(void)
 115 {
 116         /* Indicate USB connected and ready */
 117         LEDs_SetAllLEDs(LEDMASK_USB_READY);
 118
 119         /* Setup CDC Notification, Rx and Tx Endpoints */
 120         if (!(Endpoint_ConfigureEndpoint(CDC_NOTIFICATION_EPNUM, EP_TYPE_INTERRUPT,
 121                                              ENDPOINT_DIR_IN, CDC_NOTIFICATION_EPSIZE,
 122                                          ENDPOINT_BANK_SINGLE)))
 123         {
 124                 LEDs_SetAllLEDs(LEDMASK_USB_ERROR);
 125         }
 126
 127         if (!(Endpoint_ConfigureEndpoint(CDC_TX_EPNUM, EP_TYPE_BULK,
 128                                              ENDPOINT_DIR_IN, CDC_TXRX_EPSIZE,
 129                                          ENDPOINT_BANK_SINGLE)))
 130         {
 131                 LEDs_SetAllLEDs(LEDMASK_USB_ERROR);
 132         }
 133
 134         if (!(Endpoint_ConfigureEndpoint(CDC_RX_EPNUM, EP_TYPE_BULK,
 135                                              ENDPOINT_DIR_OUT, CDC_TXRX_EPSIZE,
 136                                          ENDPOINT_BANK_SINGLE)))
 137         {
 138                 LEDs_SetAllLEDs(LEDMASK_USB_ERROR);
 139         }
 140 }
 141
 142 /** Event handler for the USB_UnhandledControlPacket event. This is used to catch standard and class specific
 143  *  control requests that are not handled internally by the USB library (including the CDC control commands,
 144  *  which are all issued via the control endpoint), so that they can be handled appropriately for the application.
 145  */
 146 void EVENT_USB_UnhandledControlPacket(void)
 147 {
 148         uint8_t* LineCodingData = (uint8_t*)&LineCoding;
 149
 150         /* Process CDC specific control requests */
 151         switch (USB_ControlRequest.bRequest)
 152         {
 153                 case REQ_GetLineEncoding:
 154                         if (USB_ControlRequest.bmRequestType == (REQDIR_DEVICETOHOST | REQTYPE_CLASS | REQREC_INTERFACE))
 155                         {
 156                                 /* Acknowledge the SETUP packet, ready for data transfer */
 157                                 Endpoint_ClearSETUP();
 158
 159                                 /* Write the line coding data to the control endpoint */
 160                                 Endpoint_Write_Control_Stream_LE(LineCodingData, sizeof(LineCoding));
 161
 162                                 /* Finalize the stream transfer to send the last packet or clear the host abort */
 163                                 Endpoint_ClearOUT();
 164                         }
 165
 166                         break;
 167                 case REQ_SetLineEncoding:
 168                         if (USB_ControlRequest.bmRequestType == (REQDIR_HOSTTODEVICE | REQTYPE_CLASS | REQREC_INTERFACE))
 169                         {
 170                                 /* Acknowledge the SETUP packet, ready for data transfer */
 171                                 Endpoint_ClearSETUP();
 172
 173                                 /* Read the line coding data in from the host into the global struct */
 174                                 Endpoint_Read_Control_Stream_LE(LineCodingData, sizeof(LineCoding));
 175
 176                                 /* Finalize the stream transfer to clear the last packet from the host */
 177                                 Endpoint_ClearIN();
 178
 179                                 /* Reconfigure the USART with the new settings */
 180                                 ReconfigureUSART();
 181                         }
 182
 183                         break;
 184                 case REQ_SetControlLineState:
 185                         if (USB_ControlRequest.bmRequestType == (REQDIR_HOSTTODEVICE | REQTYPE_CLASS | REQREC_INTERFACE))
 186                         {
 187                                 /* Acknowledge the SETUP packet, ready for data transfer */
 188                                 Endpoint_ClearSETUP();
 189
 190                                 /* NOTE: Here you can read in the line state mask from the host, to get the current state of the output handshake
 191                                          lines. The mask is read in from the wValue parameter in USB_ControlRequest, and can be masked against the
 192                                                  CONTROL_LINE_OUT_* masks to determine the RTS and DTR line states using the following code:
 193                                 */
 194
 195                                 Endpoint_ClearStatusStage();
 196                         }
 197
 198                         break;
 199         }
 200 }
 201
 202 /** Task to manage CDC data transmission and reception to and from the host, from and to the physical USART. */
 203 void CDC_Task(void)
 204 {
 205         /* Device must be connected and configured for the task to run */
 206         if (USB_DeviceState != DEVICE_STATE_Configured)
 207           return;
 208
 209 #if 0
 210         /* NOTE: Here you can use the notification endpoint to send back line state changes to the host, for the special RS-232
 211                          handshake signal lines (and some error states), via the CONTROL_LINE_IN_* masks and the following code:
 212         */
 213
 214         USB_Notification_Header_t Notification = (USB_Notification_Header_t)
 215                 {
 216                         .NotificationType = (REQDIR_DEVICETOHOST | REQTYPE_CLASS | REQREC_INTERFACE),
 217                         .Notification     = NOTIF_SerialState,
 218                         .wValue           = 0,
 219                         .wIndex           = 0,
 220                         .wLength          = sizeof(uint16_t),
 221                 };
 222
 223         uint16_t LineStateMask;
 224
 225         // Set LineStateMask here to a mask of CONTROL_LINE_IN_* masks to set the input handshake line states to send to the host
 226
 227         Endpoint_SelectEndpoint(CDC_NOTIFICATION_EPNUM);
 228         Endpoint_Write_Stream_LE(&Notification, sizeof(Notification));
 229         Endpoint_Write_Stream_LE(&LineStateMask, sizeof(LineStateMask));
 230         Endpoint_ClearIN();
 231 #endif
 232
 233         /* Select the Serial Rx Endpoint */
 234         Endpoint_SelectEndpoint(CDC_RX_EPNUM);
 235
 236         /* Check to see if a packet has been received from the host */
 237         if (Endpoint_IsOUTReceived())
 238         {
 239                 /* Read the bytes in from the endpoint into the buffer while space is available */
 240                 while (Endpoint_BytesInEndpoint() && (Rx_Buffer.Elements != BUFF_STATICSIZE))
 241                 {
 242                         /* Store each character from the endpoint */
 243                         Buffer_StoreElement(&Rx_Buffer, Endpoint_Read_Byte());
 244                 }
 245
 246                 /* Check to see if all bytes in the current packet have been read */
 247                 if (!(Endpoint_BytesInEndpoint()))
 248                 {
 249                         /* Clear the endpoint buffer */
 250                         Endpoint_ClearOUT();
 251                 }
 252         }
 253
 254         /* Check if Rx buffer contains data - if so, send it */
 255         if (Rx_Buffer.Elements)
 256           Serial_TxByte(Buffer_GetElement(&Rx_Buffer));
 257
 258         /* Select the Serial Tx Endpoint */
 259         Endpoint_SelectEndpoint(CDC_TX_EPNUM);
 260
 261         /* Check if the Tx buffer contains anything to be sent to the host */
 262         if (Tx_Buffer.Elements)
 263         {
 264                 /* Wait until Serial Tx Endpoint Ready for Read/Write */
 265                 while (!(Endpoint_IsReadWriteAllowed()))
 266                 {
 267                         if (USB_DeviceState == DEVICE_STATE_Unattached)
 268                           return;
 269                 }
 270
 271                 /* Write the bytes from the buffer to the endpoint while space is available */
 272                 while (Tx_Buffer.Elements && Endpoint_IsReadWriteAllowed())
 273                 {
 274                         /* Write each byte retreived from the buffer to the endpoint */
 275                         Endpoint_Write_Byte(Buffer_GetElement(&Tx_Buffer));
 276                 }
 277
 278                 /* Remember if the packet to send completely fills the endpoint */
 279                 bool IsFull = (Endpoint_BytesInEndpoint() == CDC_TXRX_EPSIZE);
 280
 281                 /* Send the data */
 282                 Endpoint_ClearIN();
 283
 284                 /* If no more data to send and the last packet filled the endpoint, send an empty packet to release
 285                  * the buffer on the receiver (otherwise all data will be cached until a non-full packet is received) */
 286                 if (IsFull && !(Tx_Buffer.Elements))
 287                 {
 288                         /* Wait until Serial Tx Endpoint Ready for Read/Write */
 289                         while (!(Endpoint_IsReadWriteAllowed()))
 290                         {
 291                                 if (USB_DeviceState == DEVICE_STATE_Unattached)
 292                                   return;
 293                         }
 294
 295                         /* Send an empty packet to terminate the transfer */
 296                         Endpoint_ClearIN();
 297                 }
 298         }
 299 }
 300
 301 /** ISR to handle the USART receive complete interrupt, fired each time the USART has received a character. This stores the received
 302  *  character into the Tx_Buffer circular buffer for later transmission to the host.
 303  */
 304 ISR(USART1_RX_vect, ISR_BLOCK)
 305 {
 306         /* Only store received characters if the USB interface is connected */
 307         if (USB_DeviceState != DEVICE_STATE_Configured)
 308           Buffer_StoreElement(&Tx_Buffer, UDR1);
 309 }
 310
 311 /** Reconfigures the USART to match the current serial port settings issued by the host as closely as possible. */
 312 void ReconfigureUSART(void)
 313 {
 314         uint8_t ConfigMask = 0;
 315
 316         /* Determine parity - non odd/even parity mode defaults to no parity */
 317         if (LineCoding.ParityType == Parity_Odd)
 318           ConfigMask = ((1 << UPM11) | (1 << UPM10));
 319         else if (LineCoding.ParityType == Parity_Even)
 320           ConfigMask = (1 << UPM11);
 321
 322         /* Determine stop bits - 1.5 stop bits is set as 1 stop bit due to hardware limitations */
 323         if (LineCoding.CharFormat == TwoStopBits)
 324           ConfigMask |= (1 << USBS1);
 325
 326         /* Determine data size - 5, 6, 7, or 8 bits are supported */
 327         if (LineCoding.DataBits == 6)
 328           ConfigMask |= (1 << UCSZ10);
 329         else if (LineCoding.DataBits == 7)
 330           ConfigMask |= (1 << UCSZ11);
 331         else if (LineCoding.DataBits == 8)
 332           ConfigMask |= ((1 << UCSZ11) | (1 << UCSZ10));
 333
 334         /* Enable double speed, gives better error percentages at 8MHz */
 335         UCSR1A = (1 << U2X1);
 336
 337         /* Enable transmit and receive modules and interrupts */
 338         UCSR1B = ((1 << RXCIE1) | (1 << TXEN1) | (1 << RXEN1));
 339
 340         /* Set the USART mode to the mask generated by the Line Coding options */
 341         UCSR1C = ConfigMask;
 342
 343         /* Set the USART baud rate register to the desired baud rate value */
 344         UBRR1  = SERIAL_2X_UBBRVAL((uint16_t)LineCoding.BaudRateBPS);
 345 }