Demos/Device/ClassDriver/MassStorage/MassStorage.c

   1 /*
   2              LUFA Library
   3      Copyright (C) Dean Camera, 2013.
   4
   5   dean [at] fourwalledcubicle [dot] com
   6            www.lufa-lib.org
   7 */
   8
   9 /*
  10   Copyright 2013  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 disclaims 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 MassStorage demo. This file contains the main tasks of
  34  *  the demo and is responsible for the initial application hardware configuration.
  35  */
  36
  37 #include "MassStorage.h"
  38
  39 /** LUFA Mass Storage Class driver interface configuration and state information. This structure is
  40  *  passed to all Mass Storage 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_MS_Device_t Disk_MS_Interface =
  44         {
  45                 .Config =
  46                         {
  47                                 .InterfaceNumber           = 0,
  48                                 .DataINEndpoint            =
  49                                         {
  50                                                 .Address           = MASS_STORAGE_IN_EPADDR,
  51                                                 .Size              = MASS_STORAGE_IO_EPSIZE,
  52                                                 .Banks             = 1,
  53                                         },
  54                                 .DataOUTEndpoint           =
  55                                         {
  56                                                 .Address           = MASS_STORAGE_OUT_EPADDR,
  57                                                 .Size              = MASS_STORAGE_IO_EPSIZE,
  58                                                 .Banks             = 1,
  59                                         },
  60                                 .TotalLUNs                 = TOTAL_LUNS,
  61                         },
  62         };
  63
  64
  65 /** Main program entry point. This routine contains the overall program flow, including initial
  66  *  setup of all components and the main program loop.
  67  */
  68 int main(void)
  69 {
  70         SetupHardware();
  71
  72         LEDs_SetAllLEDs(LEDMASK_USB_NOTREADY);
  73         GlobalInterruptEnable();
  74
  75         for (;;)
  76         {
  77                 MS_Device_USBTask(&Disk_MS_Interface);
  78                 USB_USBTask();
  79         }
  80 }
  81
  82 /** Configures the board hardware and chip peripherals for the demo's functionality. */
  83 void SetupHardware(void)
  84 {
  85 #if (ARCH == ARCH_AVR8)
  86         /* Disable watchdog if enabled by bootloader/fuses */
  87         MCUSR &= ~(1 << WDRF);
  88         wdt_disable();
  89
  90         /* Disable clock division */
  91         clock_prescale_set(clock_div_1);
  92 #elif (ARCH == ARCH_XMEGA)
  93         /* Start the PLL to multiply the 2MHz RC oscillator to 32MHz and switch the CPU core to run from it */
  94         XMEGACLK_StartPLL(CLOCK_SRC_INT_RC2MHZ, 2000000, F_CPU);
  95         XMEGACLK_SetCPUClockSource(CLOCK_SRC_PLL);
  96
  97         /* Start the 32MHz internal RC oscillator and start the DFLL to increase it to 48MHz using the USB SOF as a reference */
  98         XMEGACLK_StartInternalOscillator(CLOCK_SRC_INT_RC32MHZ);
  99         XMEGACLK_StartDFLL(CLOCK_SRC_INT_RC32MHZ, DFLL_REF_INT_USBSOF, F_USB);
 100
 101         PMIC.CTRL = PMIC_LOLVLEN_bm | PMIC_MEDLVLEN_bm | PMIC_HILVLEN_bm;
 102 #endif
 103
 104         /* Hardware Initialization */
 105         LEDs_Init();
 106         SPI_Init(SPI_SPEED_FCPU_DIV_2 | SPI_ORDER_MSB_FIRST | SPI_SCK_LEAD_FALLING | SPI_SAMPLE_TRAILING | SPI_MODE_MASTER);
 107         Dataflash_Init();
 108         USB_Init();
 109
 110         /* Check if the Dataflash is working, abort if not */
 111         if (!(DataflashManager_CheckDataflashOperation()))
 112         {
 113                 LEDs_SetAllLEDs(LEDMASK_USB_ERROR);
 114                 for(;;);
 115         }
 116
 117         /* Clear Dataflash sector protections, if enabled */
 118         DataflashManager_ResetDataflashProtections();
 119 }
 120
 121 /** Event handler for the library USB Connection event. */
 122 void EVENT_USB_Device_Connect(void)
 123 {
 124         LEDs_SetAllLEDs(LEDMASK_USB_ENUMERATING);
 125 }
 126
 127 /** Event handler for the library USB Disconnection event. */
 128 void EVENT_USB_Device_Disconnect(void)
 129 {
 130         LEDs_SetAllLEDs(LEDMASK_USB_NOTREADY);
 131 }
 132
 133 /** Event handler for the library USB Configuration Changed event. */
 134 void EVENT_USB_Device_ConfigurationChanged(void)
 135 {
 136         bool ConfigSuccess = true;
 137
 138         ConfigSuccess &= MS_Device_ConfigureEndpoints(&Disk_MS_Interface);
 139
 140         LEDs_SetAllLEDs(ConfigSuccess ? LEDMASK_USB_READY : LEDMASK_USB_ERROR);
 141 }
 142
 143 /** Event handler for the library USB Control Request reception event. */
 144 void EVENT_USB_Device_ControlRequest(void)
 145 {
 146         MS_Device_ProcessControlRequest(&Disk_MS_Interface);
 147 }
 148
 149 /** Mass Storage class driver callback function the reception of SCSI commands from the host, which must be processed.
 150  *
 151  *  \param[in] MSInterfaceInfo  Pointer to the Mass Storage class interface configuration structure being referenced
 152  */
 153 bool CALLBACK_MS_Device_SCSICommandReceived(USB_ClassInfo_MS_Device_t* const MSInterfaceInfo)
 154 {
 155         bool CommandSuccess;
 156
 157         LEDs_SetAllLEDs(LEDMASK_USB_BUSY);
 158         CommandSuccess = SCSI_DecodeSCSICommand(MSInterfaceInfo);
 159         LEDs_SetAllLEDs(LEDMASK_USB_READY);
 160
 161         return CommandSuccess;
 162 }
 163