Demos/Device/ClassDriver/VirtualSerialMassStorage/VirtualSerialMassStorage.c

   1 /*
   2              LUFA Library
   3      Copyright (C) Dean Camera, 2012.
   4
   5   dean [at] fourwalledcubicle [dot] com
   6            www.lufa-lib.org
   7 */
   8
   9 /*
  10   Copyright 2012  Dean Camera (dean [at] fourwalledcubicle [dot] com)
  11
  12   Permission to use, copy, modify, distribute, and sell this
  13   software and its documentation for any purpose is hereby granted
  14   without fee, provided that the above copyright notice appear in
  15   all 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 /** \file
  32  *
  33  *  Main source file for the VirtualSerialMassStorage demo. This file contains the main tasks of
  34  *  the demo and is responsible for the initial application hardware configuration.
  35  */
  36
  37 #include "VirtualSerialMassStorage.h"
  38
  39 /** LUFA CDC Class driver interface configuration and state information. This structure is
  40  *  passed to all CDC Class driver functions, so that multiple instances of the same class
  41  *  within a device can be differentiated from one another.
  42  */
  43 USB_ClassInfo_CDC_Device_t VirtualSerial_CDC_Interface =
  44         {
  45                 .Config =
  46                         {
  47                                 .ControlInterfaceNumber         = 0,
  48
  49                                 .DataINEndpointNumber           = CDC_TX_EPNUM,
  50                                 .DataINEndpointSize             = CDC_TXRX_EPSIZE,
  51                                 .DataINEndpointDoubleBank       = false,
  52
  53                                 .DataOUTEndpointNumber          = CDC_RX_EPNUM,
  54                                 .DataOUTEndpointSize            = CDC_TXRX_EPSIZE,
  55                                 .DataOUTEndpointDoubleBank      = false,
  56
  57                                 .NotificationEndpointNumber     = CDC_NOTIFICATION_EPNUM,
  58                                 .NotificationEndpointSize       = CDC_NOTIFICATION_EPSIZE,
  59                                 .NotificationEndpointDoubleBank = false,
  60                         },
  61         };
  62
  63 /** LUFA Mass Storage Class driver interface configuration and state information. This structure is
  64  *  passed to all Mass Storage Class driver functions, so that multiple instances of the same class
  65  *  within a device can be differentiated from one another.
  66  */
  67 USB_ClassInfo_MS_Device_t Disk_MS_Interface =
  68         {
  69                 .Config =
  70                         {
  71                                 .InterfaceNumber                = 2,
  72
  73                                 .DataINEndpointNumber           = MASS_STORAGE_IN_EPNUM,
  74                                 .DataINEndpointSize             = MASS_STORAGE_IO_EPSIZE,
  75                                 .DataINEndpointDoubleBank       = false,
  76
  77                                 .DataOUTEndpointNumber          = MASS_STORAGE_OUT_EPNUM,
  78                                 .DataOUTEndpointSize            = MASS_STORAGE_IO_EPSIZE,
  79                                 .DataOUTEndpointDoubleBank      = false,
  80
  81                                 .TotalLUNs                      = TOTAL_LUNS,
  82                         },
  83         };
  84
  85 /** Standard file stream for the CDC interface when set up, so that the virtual CDC COM port can be
  86  *  used like any regular character stream in the C APIs
  87  */
  88 static FILE USBSerialStream;
  89
  90
  91 /** Main program entry point. This routine contains the overall program flow, including initial
  92  *  setup of all components and the main program loop.
  93  */
  94 int main(void)
  95 {
  96         SetupHardware();
  97
  98         /* Create a regular character stream for the interface so that it can be used with the stdio.h functions */
  99         CDC_Device_CreateStream(&VirtualSerial_CDC_Interface, &USBSerialStream);
 100
 101         LEDs_SetAllLEDs(LEDMASK_USB_NOTREADY);
 102         sei();
 103
 104         for (;;)
 105         {
 106                 CheckJoystickMovement();
 107
 108                 /* Must throw away unused bytes from the host, or it will lock up while waiting for the device */
 109                 CDC_Device_ReceiveByte(&VirtualSerial_CDC_Interface);
 110
 111                 CDC_Device_USBTask(&VirtualSerial_CDC_Interface);
 112                 MS_Device_USBTask(&Disk_MS_Interface);
 113                 USB_USBTask();
 114         }
 115 }
 116
 117 /** Configures the board hardware and chip peripherals for the demo's functionality. */
 118 void SetupHardware(void)
 119 {
 120         /* Disable watchdog if enabled by bootloader/fuses */
 121         MCUSR &= ~(1 << WDRF);
 122         wdt_disable();
 123
 124         /* Disable clock division */
 125         clock_prescale_set(clock_div_1);
 126
 127         /* Hardware Initialization */
 128         LEDs_Init();
 129         Joystick_Init();
 130         SPI_Init(SPI_SPEED_FCPU_DIV_2 | SPI_ORDER_MSB_FIRST | SPI_SCK_LEAD_FALLING | SPI_SAMPLE_TRAILING | SPI_MODE_MASTER);
 131         Dataflash_Init();
 132         USB_Init();
 133
 134         /* Check if the Dataflash is working, abort if not */
 135         if (!(DataflashManager_CheckDataflashOperation()))
 136         {
 137                 LEDs_SetAllLEDs(LEDMASK_USB_ERROR);
 138                 for(;;);
 139         }
 140
 141         /* Clear Dataflash sector protections, if enabled */
 142         DataflashManager_ResetDataflashProtections();
 143 }
 144
 145 /** Checks for changes in the position of the board joystick, sending strings to the host upon each change. */
 146 void CheckJoystickMovement(void)
 147 {
 148         uint8_t     JoyStatus_LCL = Joystick_GetStatus();
 149         char*       ReportString  = NULL;
 150         static bool ActionSent    = false;
 151
 152         if (JoyStatus_LCL & JOY_UP)
 153           ReportString = "Joystick Up\r\n";
 154         else if (JoyStatus_LCL & JOY_DOWN)
 155           ReportString = "Joystick Down\r\n";
 156         else if (JoyStatus_LCL & JOY_LEFT)
 157           ReportString = "Joystick Left\r\n";
 158         else if (JoyStatus_LCL & JOY_RIGHT)
 159           ReportString = "Joystick Right\r\n";
 160         else if (JoyStatus_LCL & JOY_PRESS)
 161           ReportString = "Joystick Pressed\r\n";
 162         else
 163           ActionSent = false;
 164
 165         if ((ReportString != NULL) && (ActionSent == false))
 166         {
 167                 ActionSent = true;
 168
 169                 /* Write the string to the virtual COM port via the created character stream */
 170                 fputs(ReportString, &USBSerialStream);
 171
 172                 /* Alternatively, without the stream: */
 173                 // CDC_Device_SendString(&VirtualSerial_CDC_Interface, ReportString);
 174         }
 175 }
 176
 177 /** Event handler for the library USB Connection event. */
 178 void EVENT_USB_Device_Connect(void)
 179 {
 180         LEDs_SetAllLEDs(LEDMASK_USB_ENUMERATING);
 181 }
 182
 183 /** Event handler for the library USB Disconnection event. */
 184 void EVENT_USB_Device_Disconnect(void)
 185 {
 186         LEDs_SetAllLEDs(LEDMASK_USB_NOTREADY);
 187 }
 188
 189 /** Event handler for the library USB Configuration Changed event. */
 190 void EVENT_USB_Device_ConfigurationChanged(void)
 191 {
 192         bool ConfigSuccess = true;
 193
 194         ConfigSuccess &= CDC_Device_ConfigureEndpoints(&VirtualSerial_CDC_Interface);
 195         ConfigSuccess &= MS_Device_ConfigureEndpoints(&Disk_MS_Interface);
 196
 197         LEDs_SetAllLEDs(ConfigSuccess ? LEDMASK_USB_READY : LEDMASK_USB_ERROR);
 198 }
 199
 200 /** Event handler for the library USB Control Request reception event. */
 201 void EVENT_USB_Device_ControlRequest(void)
 202 {
 203         CDC_Device_ProcessControlRequest(&VirtualSerial_CDC_Interface);
 204         MS_Device_ProcessControlRequest(&Disk_MS_Interface);
 205 }
 206
 207 /** Mass Storage class driver callback function the reception of SCSI commands from the host, which must be processed.
 208  *
 209  *  \param[in] MSInterfaceInfo  Pointer to the Mass Storage class interface configuration structure being referenced
 210  */
 211 bool CALLBACK_MS_Device_SCSICommandReceived(USB_ClassInfo_MS_Device_t* const MSInterfaceInfo)
 212 {
 213         bool CommandSuccess;
 214
 215         LEDs_SetAllLEDs(LEDMASK_USB_BUSY);
 216         CommandSuccess = SCSI_DecodeSCSICommand(MSInterfaceInfo);
 217         LEDs_SetAllLEDs(LEDMASK_USB_READY);
 218
 219         return CommandSuccess;
 220 }
 221