Demos/Host/ClassDriver/MassStorageHost/MassStorageHost.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 /** \file
  32  *
  33  *  Main source file for the MassStorageHost demo. This file contains the main tasks of
  34  *  the demo and is responsible for the initial application hardware configuration.
  35  */
  36
  37 #include "MassStorageHost.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_Host_t FlashDisk_MS_Interface =
  44         {
  45                 .Config =
  46                         {
  47                                 .DataINPipeNumber       = 1,
  48                                 .DataOUTPipeNumber      = 2,
  49                         },
  50         };
  51
  52
  53 /** Main program entry point. This routine configures the hardware required by the application, then
  54  *  enters a loop to run the application tasks in sequence.
  55  */
  56 int main(void)
  57 {
  58         SetupHardware();
  59
  60         puts_P(PSTR(ESC_FG_CYAN "Mass Storage Host Demo running.\r\n" ESC_FG_WHITE));
  61
  62         LEDs_SetAllLEDs(LEDMASK_USB_NOTREADY);
  63
  64         for (;;)
  65         {
  66                 switch (USB_HostState)
  67                 {
  68                         case HOST_STATE_Addressed:
  69                                 LEDs_SetAllLEDs(LEDMASK_USB_ENUMERATING);
  70
  71                                 uint16_t ConfigDescriptorSize;
  72                                 uint8_t  ConfigDescriptorData[512];
  73
  74                                 if (USB_GetDeviceConfigDescriptor(1, &ConfigDescriptorSize, ConfigDescriptorData,
  75                                                                   sizeof(ConfigDescriptorData)) != HOST_GETCONFIG_Successful)
  76                                 {
  77                                         printf("Error Retrieving Configuration Descriptor.\r\n");
  78                                         LEDs_SetAllLEDs(LEDMASK_USB_ERROR);
  79                                         USB_HostState = HOST_STATE_WaitForDeviceRemoval;
  80                                         break;
  81                                 }
  82
  83                                 if (MS_Host_ConfigurePipes(&FlashDisk_MS_Interface,
  84                                                             ConfigDescriptorSize, ConfigDescriptorData) != MS_ENUMERROR_NoError)
  85                                 {
  86                                         printf("Attached Device Not a Valid Mass Storage Device.\r\n");
  87                                         LEDs_SetAllLEDs(LEDMASK_USB_ERROR);
  88                                         USB_HostState = HOST_STATE_WaitForDeviceRemoval;
  89                                         break;
  90                                 }
  91
  92                                 if (USB_Host_SetDeviceConfiguration(1) != HOST_SENDCONTROL_Successful)
  93                                 {
  94                                         printf("Error Setting Device Configuration.\r\n");
  95                                         LEDs_SetAllLEDs(LEDMASK_USB_ERROR);
  96                                         USB_HostState = HOST_STATE_WaitForDeviceRemoval;
  97                                         break;
  98                                 }
  99
 100                                 printf("Mass Storage Device Enumerated.\r\n");
 101                                 USB_HostState = HOST_STATE_Configured;
 102                                 break;
 103                         case HOST_STATE_Configured:
 104                                 LEDs_SetAllLEDs(LEDMASK_USB_BUSY);
 105
 106                                 uint8_t MaxLUNIndex;
 107                                 if (MS_Host_GetMaxLUN(&FlashDisk_MS_Interface, &MaxLUNIndex))
 108                                 {
 109                                         printf("Error retrieving max LUN index.\r\n");
 110                                         LEDs_SetAllLEDs(LEDMASK_USB_ERROR);
 111                                         USB_HostState = HOST_STATE_WaitForDeviceRemoval;
 112                                         break;
 113                                 }
 114
 115                                 printf("Total LUNs: %d - Using first LUN in device.\r\n", (MaxLUNIndex + 1));
 116
 117                                 if (MS_Host_ResetMSInterface(&FlashDisk_MS_Interface))
 118                                 {
 119                                         printf("Error resetting Mass Storage interface.\r\n");
 120                                         LEDs_SetAllLEDs(LEDMASK_USB_ERROR);
 121                                         USB_HostState = HOST_STATE_WaitForDeviceRemoval;
 122                                         break;
 123                                 }
 124
 125                                 SCSI_Request_Sense_Response_t SenseData;
 126                                 if (MS_Host_RequestSense(&FlashDisk_MS_Interface, 0, &SenseData) != 0)
 127                                 {
 128                                         printf("Error retrieving device sense.\r\n");
 129                                         LEDs_SetAllLEDs(LEDMASK_USB_ERROR);
 130                                         USB_HostState = HOST_STATE_WaitForDeviceRemoval;
 131                                         break;
 132                                 }
 133
 134                                 if (MS_Host_PreventAllowMediumRemoval(&FlashDisk_MS_Interface, 0, true))
 135                                 {
 136                                         printf("Error setting Prevent Device Removal bit.\r\n");
 137                                         LEDs_SetAllLEDs(LEDMASK_USB_ERROR);
 138                                         USB_HostState = HOST_STATE_WaitForDeviceRemoval;
 139                                         break;
 140                                 }
 141
 142                                 SCSI_Inquiry_Response_t InquiryData;
 143                                 if (MS_Host_GetInquiryData(&FlashDisk_MS_Interface, 0, &InquiryData))
 144                                 {
 145                                         printf("Error retreiving device Inquiry data.\r\n");
 146                                         LEDs_SetAllLEDs(LEDMASK_USB_ERROR);
 147                                         USB_HostState = HOST_STATE_WaitForDeviceRemoval;
 148                                         break;
 149                                 }
 150
 151                                 printf("Vendor \"%.8s\", Product \"%.16s\"\r\n", InquiryData.VendorID, InquiryData.ProductID);
 152
 153                                 printf("Waiting until ready...\r\n");
 154
 155                                 for (;;)
 156                                 {
 157                                         uint8_t ErrorCode = MS_Host_TestUnitReady(&FlashDisk_MS_Interface, 0);
 158
 159                                         if (!(ErrorCode))
 160                                           break;
 161
 162                                         /* Check if an error other than a logical command error (device busy) received */
 163                                         if (ErrorCode != MS_ERROR_LOGICAL_CMD_FAILED)
 164                                         {
 165                                                 printf("Error waiting for device to be ready.\r\n");
 166                                                 LEDs_SetAllLEDs(LEDMASK_USB_ERROR);
 167                                                 USB_HostState = HOST_STATE_WaitForDeviceRemoval;
 168                                                 break;
 169                                         }
 170                                 }
 171
 172                                 printf("Retrieving Capacity... ");
 173
 174                                 SCSI_Capacity_t DiskCapacity;
 175                                 if (MS_Host_ReadDeviceCapacity(&FlashDisk_MS_Interface, 0, &DiskCapacity))
 176                                 {
 177                                         printf("Error retrieving device capacity.\r\n");
 178                                         LEDs_SetAllLEDs(LEDMASK_USB_ERROR);
 179                                         USB_HostState = HOST_STATE_WaitForDeviceRemoval;
 180                                         break;
 181                                 }
 182
 183                                 printf("%lu blocks of %lu bytes.\r\n", DiskCapacity.Blocks, DiskCapacity.BlockSize);
 184
 185                                 uint8_t BlockBuffer[DiskCapacity.BlockSize];
 186
 187                                 if (MS_Host_ReadDeviceBlocks(&FlashDisk_MS_Interface, 0, 0x00000000, 1, DiskCapacity.BlockSize, BlockBuffer))
 188                                 {
 189                                         printf("Error reading device block.\r\n");
 190                                         LEDs_SetAllLEDs(LEDMASK_USB_ERROR);
 191                                         USB_HostState = HOST_STATE_WaitForDeviceRemoval;
 192                                         break;
 193                                 }
 194
 195                                 printf("\r\nContents of first block:\r\n");
 196
 197                                 for (uint16_t Chunk = 0; Chunk < (DiskCapacity.BlockSize >> 4); Chunk++)
 198                                 {
 199                                         uint8_t* ChunkPtr = &BlockBuffer[Chunk << 4];
 200
 201                                         /* Print out the 16 bytes of the chunk in HEX format */
 202                                         for (uint8_t ByteOffset = 0; ByteOffset < (1 << 4); ByteOffset++)
 203                                         {
 204                                                 char CurrByte = *(ChunkPtr + ByteOffset);
 205                                                 printf_P(PSTR("%.2X "), CurrByte);
 206                                         }
 207
 208                                         printf("    ");
 209
 210                                         /* Print out the 16 bytes of the chunk in ASCII format */
 211                                         for (uint8_t ByteOffset = 0; ByteOffset < (1 << 4); ByteOffset++)
 212                                         {
 213                                                 char CurrByte = *(ChunkPtr + ByteOffset);
 214                                                 putchar(isprint(CurrByte) ? CurrByte : '.');
 215                                         }
 216
 217                                         printf("\r\n");
 218                                 }
 219
 220                                 LEDs_SetAllLEDs(LEDMASK_USB_READY);
 221                                 USB_HostState = HOST_STATE_WaitForDeviceRemoval;
 222                                 break;
 223                 }
 224
 225                 MS_Host_USBTask(&FlashDisk_MS_Interface);
 226                 USB_USBTask();
 227         }
 228 }
 229
 230 /** Configures the board hardware and chip peripherals for the demo's functionality. */
 231 void SetupHardware(void)
 232 {
 233         /* Disable watchdog if enabled by bootloader/fuses */
 234         MCUSR &= ~(1 << WDRF);
 235         wdt_disable();
 236
 237         /* Disable clock division */
 238         clock_prescale_set(clock_div_1);
 239
 240         /* Hardware Initialization */
 241         SerialStream_Init(9600, false);
 242         LEDs_Init();
 243         USB_Init();
 244 }
 245
 246 /** Event handler for the USB_DeviceAttached event. This indicates that a device has been attached to the host, and
 247  *  starts the library USB task to begin the enumeration and USB management process.
 248  */
 249 void EVENT_USB_Host_DeviceAttached(void)
 250 {
 251         puts_P(PSTR("Device Attached.\r\n"));
 252         LEDs_SetAllLEDs(LEDMASK_USB_ENUMERATING);
 253 }
 254
 255 /** Event handler for the USB_DeviceUnattached event. This indicates that a device has been removed from the host, and
 256  *  stops the library USB task management process.
 257  */
 258 void EVENT_USB_Host_DeviceUnattached(void)
 259 {
 260         puts_P(PSTR("\r\nDevice Unattached.\r\n"));
 261         LEDs_SetAllLEDs(LEDMASK_USB_NOTREADY);
 262 }
 263
 264 /** Event handler for the USB_DeviceEnumerationComplete event. This indicates that a device has been successfully
 265  *  enumerated by the host and is now ready to be used by the application.
 266  */
 267 void EVENT_USB_Host_DeviceEnumerationComplete(void)
 268 {
 269         LEDs_SetAllLEDs(LEDMASK_USB_READY);
 270 }
 271
 272 /** Event handler for the USB_HostError event. This indicates that a hardware error occurred while in host mode. */
 273 void EVENT_USB_Host_HostError(const uint8_t ErrorCode)
 274 {
 275         USB_ShutDown();
 276
 277         printf_P(PSTR(ESC_FG_RED "Host Mode Error\r\n"
 278                                  " -- Error Code %d\r\n" ESC_FG_WHITE), ErrorCode);
 279
 280         LEDs_SetAllLEDs(LEDMASK_USB_ERROR);
 281         for(;;);
 282 }
 283
 284 /** Event handler for the USB_DeviceEnumerationFailed event. This indicates that a problem occurred while
 285  *  enumerating an attached USB device.
 286  */
 287 void EVENT_USB_Host_DeviceEnumerationFailed(const uint8_t ErrorCode, const uint8_t SubErrorCode)
 288 {
 289         printf_P(PSTR(ESC_FG_RED "Dev Enum Error\r\n"
 290                                  " -- Error Code %d\r\n"
 291                                  " -- Sub Error Code %d\r\n"
 292                                  " -- In State %d\r\n" ESC_FG_WHITE), ErrorCode, SubErrorCode, USB_HostState);
 293
 294         LEDs_SetAllLEDs(LEDMASK_USB_ERROR);
 295 }