Fixed Endpoint_Write_Control_* functions writing more data than expected by the host...
[pub/USBasp.git] / Demos / Host / KeyboardHost / KeyboardHost.c
index 6249e52..f17f950 100644 (file)
  \r
 #include "KeyboardHost.h"\r
 \r
-/* Project Tags, for reading out using the ButtLoad project */\r
-BUTTLOADTAG(ProjName,    "LUFA KBD Host App");\r
-BUTTLOADTAG(BuildTime,   __TIME__);\r
-BUTTLOADTAG(BuildDate,   __DATE__);\r
-BUTTLOADTAG(LUFAVersion, "LUFA V" LUFA_VERSION_STRING);\r
-\r
 /* Scheduler Task List */\r
 TASK_LIST\r
 {\r
-       { Task: USB_USBTask          , TaskStatus: TASK_STOP },\r
-       { Task: USB_Keyboard_Host    , TaskStatus: TASK_STOP },\r
+       { .Task = USB_USBTask          , .TaskStatus = TASK_STOP },\r
+       { .Task = USB_Keyboard_Host    , .TaskStatus = TASK_STOP },\r
 };\r
 \r
 \r
@@ -75,7 +69,7 @@ int main(void)
        /* Initialize USB Subsystem */\r
        USB_Init();\r
 \r
-       /* Startup message */\r
+       /* Start-up message */\r
        puts_P(PSTR(ESC_RESET ESC_BG_WHITE ESC_INVERSE_ON ESC_ERASE_DISPLAY\r
               "Keyboard Host Demo running.\r\n" ESC_INVERSE_OFF));\r
                   \r
@@ -132,7 +126,7 @@ EVENT_HANDLER(USB_HostError)
        for(;;);\r
 }\r
 \r
-/** Event handler for the USB_DeviceEnumerationFailed event. This indicates that a problem occured while\r
+/** Event handler for the USB_DeviceEnumerationFailed event. This indicates that a problem occurred while\r
  *  enumerating an attached USB device.\r
  */\r
 EVENT_HANDLER(USB_DeviceEnumerationFailed)\r
@@ -183,47 +177,69 @@ void ReadNextReport(void)
 {\r
        USB_KeyboardReport_Data_t KeyboardReport;\r
                \r
-       /* Select the keyboard report data in pipe */\r
+       /* Select keyboard data pipe */\r
        Pipe_SelectPipe(KEYBOARD_DATAPIPE);     \r
 \r
-       /* Ensure pipe contains data and is ready to be read before continuing */\r
-       if (!(Pipe_ReadWriteAllowed()))\r
-         return;\r
-\r
-       /* Read in keyboard report data */\r
-       Pipe_Read_Stream_LE(&KeyboardReport, sizeof(KeyboardReport));\r
-                                       \r
-       /* Clear the IN endpoint, ready for next data packet */\r
-       Pipe_ClearCurrentBank();\r
+       #if !defined(INTERRUPT_DATA_PIPE)\r
+       /* Unfreeze keyboard data pipe */\r
+       Pipe_Unfreeze();\r
+       #endif\r
 \r
-       /* Indicate if the modifier byte is non-zero (special key such as shift is being pressed) */\r
-       LEDs_ChangeLEDs(LEDS_LED1, (KeyboardReport.Modifier) ? LEDS_LED1 : 0);\r
+       /* Check to see if a packet has been received */\r
+       if (!(Pipe_IsINReceived()))\r
+       {\r
+               #if !defined(INTERRUPT_DATA_PIPE)\r
+               /* Refreeze HID data IN pipe */\r
+               Pipe_Freeze();\r
+               #endif\r
+                       \r
+               return;\r
+       }\r
        \r
-       /* Check if a key has been pressed */\r
-       if (KeyboardReport.KeyCode)\r
+       /* Ensure pipe contains data before trying to read from it */\r
+       if (Pipe_IsReadWriteAllowed())\r
        {\r
-               /* Toggle status LED to indicate keypress */\r
-               if (LEDs_GetLEDs() & LEDS_LED2)\r
-                 LEDs_TurnOffLEDs(LEDS_LED2);\r
-               else\r
-                 LEDs_TurnOnLEDs(LEDS_LED2);\r
-                         \r
-               char PressedKey = 0;\r
-\r
-               /* Retrieve pressed key character if alphanumeric */\r
-               if ((KeyboardReport.KeyCode >= 0x04) && (KeyboardReport.KeyCode <= 0x1D))\r
-                 PressedKey = (KeyboardReport.KeyCode - 0x04) + 'A';\r
-               else if ((KeyboardReport.KeyCode >= 0x1E) && (KeyboardReport.KeyCode <= 0x27))\r
-                 PressedKey = (KeyboardReport.KeyCode - 0x1E) + '0';\r
-               else if (KeyboardReport.KeyCode == 0x2C)\r
-                 PressedKey = ' ';                                             \r
-               else if (KeyboardReport.KeyCode == 0x28)\r
-                 PressedKey = '\n';\r
-                        \r
-               /* Print the pressed key character out through the serial port if valid */\r
-               if (PressedKey)\r
-                 putchar(PressedKey);\r
+               /* Read in keyboard report data */\r
+               Pipe_Read_Stream_LE(&KeyboardReport, sizeof(KeyboardReport));\r
+\r
+               /* Indicate if the modifier byte is non-zero (special key such as shift is being pressed) */\r
+               LEDs_ChangeLEDs(LEDS_LED1, (KeyboardReport.Modifier) ? LEDS_LED1 : 0);\r
+               \r
+               /* Check if a key has been pressed */\r
+               if (KeyboardReport.KeyCode)\r
+               {\r
+                       /* Toggle status LED to indicate keypress */\r
+                       if (LEDs_GetLEDs() & LEDS_LED2)\r
+                         LEDs_TurnOffLEDs(LEDS_LED2);\r
+                       else\r
+                         LEDs_TurnOnLEDs(LEDS_LED2);\r
+                                 \r
+                       char PressedKey = 0;\r
+\r
+                       /* Retrieve pressed key character if alphanumeric */\r
+                       if ((KeyboardReport.KeyCode >= 0x04) && (KeyboardReport.KeyCode <= 0x1D))\r
+                         PressedKey = (KeyboardReport.KeyCode - 0x04) + 'A';\r
+                       else if ((KeyboardReport.KeyCode >= 0x1E) && (KeyboardReport.KeyCode <= 0x27))\r
+                         PressedKey = (KeyboardReport.KeyCode - 0x1E) + '0';\r
+                       else if (KeyboardReport.KeyCode == 0x2C)\r
+                         PressedKey = ' ';                                             \r
+                       else if (KeyboardReport.KeyCode == 0x28)\r
+                         PressedKey = '\n';\r
+                                \r
+                       /* Print the pressed key character out through the serial port if valid */\r
+                       if (PressedKey)\r
+                         putchar(PressedKey);\r
+               }\r
        }\r
+       \r
+                                               \r
+       /* Clear the IN endpoint, ready for next data packet */\r
+       Pipe_ClearIN();\r
+\r
+       #if !defined(INTERRUPT_DATA_PIPE)\r
+       /* Refreeze keyboard data pipe */\r
+       Pipe_Freeze();\r
+       #endif\r
 }\r
 \r
 /** Task to set the configuration of the attached device after it has been enumerated, and to read and process\r
@@ -237,16 +253,19 @@ TASK(USB_Keyboard_Host)
        {\r
                case HOST_STATE_Addressed:\r
                        /* Standard request to set the device configuration to configuration 1 */\r
-                       USB_HostRequest = (USB_Host_Request_Header_t)\r
+                       USB_ControlRequest = (USB_Request_Header_t)\r
                                {\r
-                                       bmRequestType: (REQDIR_HOSTTODEVICE | REQTYPE_STANDARD | REQREC_DEVICE),\r
-                                       bRequest:      REQ_SetConfiguration,\r
-                                       wValue:        1,\r
-                                       wIndex:        0,\r
-                                       wLength:       0,\r
+                                       .bmRequestType = (REQDIR_HOSTTODEVICE | REQTYPE_STANDARD | REQREC_DEVICE),\r
+                                       .bRequest      = REQ_SetConfiguration,\r
+                                       .wValue        = 1,\r
+                                       .wIndex        = 0,\r
+                                       .wLength       = 0,\r
                                };\r
 \r
-                       /* Send the request, display error and wait for device detatch if request fails */\r
+                       /* Select the control pipe for the request transfer */\r
+                       Pipe_SelectPipe(PIPE_CONTROLPIPE);\r
+\r
+                       /* Send the request, display error and wait for device detach if request fails */\r
                        if ((ErrorCode = USB_Host_SendControlRequest(NULL)) != HOST_SENDCONTROL_Successful)\r
                        {\r
                                puts_P(PSTR("Control Error (Set Configuration).\r\n"));\r
@@ -284,16 +303,19 @@ TASK(USB_Keyboard_Host)
                        }\r
                \r
                        /* HID class request to set the keyboard protocol to the Boot Protocol */\r
-                       USB_HostRequest = (USB_Host_Request_Header_t)\r
+                       USB_ControlRequest = (USB_Request_Header_t)\r
                                {\r
-                                       bmRequestType: (REQDIR_HOSTTODEVICE | REQTYPE_CLASS | REQREC_INTERFACE),\r
-                                       bRequest:      REQ_SetProtocol,\r
-                                       wValue:        0,\r
-                                       wIndex:        0,\r
-                                       wLength:       0,\r
+                                       .bmRequestType = (REQDIR_HOSTTODEVICE | REQTYPE_CLASS | REQREC_INTERFACE),\r
+                                       .bRequest      = REQ_SetProtocol,\r
+                                       .wValue        = 0,\r
+                                       .wIndex        = 0,\r
+                                       .wLength       = 0,\r
                                };\r
 \r
-                       /* Send the request, display error and wait for device detatch if request fails */\r
+                       /* Select the control pipe for the request transfer */\r
+                       Pipe_SelectPipe(PIPE_CONTROLPIPE);\r
+\r
+                       /* Send the request, display error and wait for device detach if request fails */\r
                        if ((ErrorCode = USB_Host_SendControlRequest(NULL)) != HOST_SENDCONTROL_Successful)\r
                        {\r
                                puts_P(PSTR("Control Error (Set Protocol).\r\n"));\r
@@ -307,22 +329,21 @@ TASK(USB_Keyboard_Host)
                                break;\r
                        }\r
 \r
+                       #if defined(INTERRUPT_DATA_PIPE)                        \r
+                       /* Select and unfreeze keyboard data pipe */\r
+                       Pipe_SelectPipe(KEYBOARD_DATAPIPE);     \r
+                       Pipe_Unfreeze();\r
+                       #endif\r
+\r
                        puts_P(PSTR("Keyboard Enumerated.\r\n"));\r
-                               \r
+\r
                        USB_HostState = HOST_STATE_Ready;\r
                        break;\r
                #if !defined(INTERRUPT_DATA_PIPE)\r
                case HOST_STATE_Ready:\r
-                       /* Select and unfreeze keyboard data pipe */\r
-                       Pipe_SelectPipe(KEYBOARD_DATAPIPE);     \r
-                       Pipe_Unfreeze();\r
-\r
                        /* If a report has been received, read and process it */\r
-                       if (Pipe_ReadWriteAllowed())\r
-                         ReadNextReport();\r
+                       ReadNextReport();\r
 \r
-                       /* Freeze keyboard data pipe */\r
-                       Pipe_Freeze();\r
                        break;\r
                #endif\r
        }\r
@@ -334,6 +355,9 @@ TASK(USB_Keyboard_Host)
  */\r
 ISR(ENDPOINT_PIPE_vect, ISR_BLOCK)\r
 {\r
+       /* Save previously selected pipe before selecting a new pipe */\r
+       uint8_t PrevSelectedPipe = Pipe_GetCurrentPipe();\r
+\r
        /* Check to see if the keyboard data pipe has caused the interrupt */\r
        if (Pipe_HasPipeInterrupted(KEYBOARD_DATAPIPE))\r
        {\r
@@ -350,5 +374,8 @@ ISR(ENDPOINT_PIPE_vect, ISR_BLOCK)
                        /* Read and process the next report from the device */\r
                        ReadNextReport();\r
        }\r
+       \r
+       /* Restore previously selected pipe */\r
+       Pipe_SelectPipe(PrevSelectedPipe);\r
 }\r
 #endif\r