forked from mfulz_github/qmk_firmware
		
	
		
			
				
	
	
		
			295 lines
		
	
	
		
			11 KiB
		
	
	
	
		
			C
		
	
	
	
	
	
			
		
		
	
	
			295 lines
		
	
	
		
			11 KiB
		
	
	
	
		
			C
		
	
	
	
	
	
| /*
 | |
|              LUFA Library
 | |
|      Copyright (C) Dean Camera, 2017.
 | |
| 
 | |
|   dean [at] fourwalledcubicle [dot] com
 | |
|            www.lufa-lib.org
 | |
| */
 | |
| 
 | |
| /*
 | |
|   Copyright 2017  Dean Camera (dean [at] fourwalledcubicle [dot] com)
 | |
| 
 | |
|   Permission to use, copy, modify, distribute, and sell this
 | |
|   software and its documentation for any purpose is hereby granted
 | |
|   without fee, provided that the above copyright notice appear in
 | |
|   all copies and that both that the copyright notice and this
 | |
|   permission notice and warranty disclaimer appear in supporting
 | |
|   documentation, and that the name of the author not be used in
 | |
|   advertising or publicity pertaining to distribution of the
 | |
|   software without specific, written prior permission.
 | |
| 
 | |
|   The author disclaims all warranties with regard to this
 | |
|   software, including all implied warranties of merchantability
 | |
|   and fitness.  In no event shall the author be liable for any
 | |
|   special, indirect or consequential damages or any damages
 | |
|   whatsoever resulting from loss of use, data or profits, whether
 | |
|   in an action of contract, negligence or other tortious action,
 | |
|   arising out of or in connection with the use or performance of
 | |
|   this software.
 | |
| */
 | |
| 
 | |
| /** \file
 | |
|  *
 | |
|  *  SCSI command processing routines, for SCSI commands issued by the host. Mass Storage
 | |
|  *  devices use a thin "Bulk-Only Transport" protocol for issuing commands and status information,
 | |
|  *  which wrap around standard SCSI device commands for controlling the actual storage medium.
 | |
|  */
 | |
| 
 | |
| #define  INCLUDE_FROM_SCSI_C
 | |
| #include "SCSI.h"
 | |
| 
 | |
| /** Structure to hold the SCSI response data to a SCSI INQUIRY command. This gives information about the device's
 | |
|  *  features and capabilities.
 | |
|  */
 | |
| static const SCSI_Inquiry_Response_t InquiryData =
 | |
| 	{
 | |
| 		.DeviceType          = DEVICE_TYPE_BLOCK,
 | |
| 		.PeripheralQualifier = 0,
 | |
| 
 | |
| 		.Removable           = true,
 | |
| 
 | |
| 		.Version             = 0,
 | |
| 
 | |
| 		.ResponseDataFormat  = 2,
 | |
| 		.NormACA             = false,
 | |
| 		.TrmTsk              = false,
 | |
| 		.AERC                = false,
 | |
| 
 | |
| 		.AdditionalLength    = 0x1F,
 | |
| 
 | |
| 		.SoftReset           = false,
 | |
| 		.CmdQue              = false,
 | |
| 		.Linked              = false,
 | |
| 		.Sync                = false,
 | |
| 		.WideBus16Bit        = false,
 | |
| 		.WideBus32Bit        = false,
 | |
| 		.RelAddr             = false,
 | |
| 
 | |
| 		.VendorID            = "LUFA",
 | |
| 		.ProductID           = "Bootloader",
 | |
| 		.RevisionID          = {'0','.','0','0'},
 | |
| 	};
 | |
| 
 | |
| /** Structure to hold the sense data for the last issued SCSI command, which is returned to the host after a SCSI REQUEST SENSE
 | |
|  *  command is issued. This gives information on exactly why the last command failed to complete.
 | |
|  */
 | |
| static SCSI_Request_Sense_Response_t SenseData =
 | |
| 	{
 | |
| 		.ResponseCode        = 0x70,
 | |
| 		.AdditionalLength    = 0x0A,
 | |
| 	};
 | |
| 
 | |
| 
 | |
| /** Main routine to process the SCSI command located in the Command Block Wrapper read from the host. This dispatches
 | |
|  *  to the appropriate SCSI command handling routine if the issued command is supported by the device, else it returns
 | |
|  *  a command failure due to a ILLEGAL REQUEST.
 | |
|  *
 | |
|  *  \param[in] MSInterfaceInfo  Pointer to the Mass Storage class interface structure that the command is associated with
 | |
|  *
 | |
|  *  \return Boolean \c true if the command completed successfully, \c false otherwise
 | |
|  */
 | |
| bool SCSI_DecodeSCSICommand(USB_ClassInfo_MS_Device_t* const MSInterfaceInfo)
 | |
| {
 | |
| 	bool CommandSuccess = false;
 | |
| 
 | |
| 	/* Run the appropriate SCSI command hander function based on the passed command */
 | |
| 	switch (MSInterfaceInfo->State.CommandBlock.SCSICommandData[0])
 | |
| 	{
 | |
| 		case SCSI_CMD_INQUIRY:
 | |
| 			CommandSuccess = SCSI_Command_Inquiry(MSInterfaceInfo);
 | |
| 			break;
 | |
| 		case SCSI_CMD_REQUEST_SENSE:
 | |
| 			CommandSuccess = SCSI_Command_Request_Sense(MSInterfaceInfo);
 | |
| 			break;
 | |
| 		case SCSI_CMD_READ_CAPACITY_10:
 | |
| 			CommandSuccess = SCSI_Command_Read_Capacity_10(MSInterfaceInfo);
 | |
| 			break;
 | |
| 		case SCSI_CMD_WRITE_10:
 | |
| 			CommandSuccess = SCSI_Command_ReadWrite_10(MSInterfaceInfo, DATA_WRITE);
 | |
| 			break;
 | |
| 		case SCSI_CMD_READ_10:
 | |
| 			CommandSuccess = SCSI_Command_ReadWrite_10(MSInterfaceInfo, DATA_READ);
 | |
| 			break;
 | |
| 		case SCSI_CMD_MODE_SENSE_6:
 | |
| 			CommandSuccess = SCSI_Command_ModeSense_6(MSInterfaceInfo);
 | |
| 			break;
 | |
| 		case SCSI_CMD_START_STOP_UNIT:
 | |
| #if !defined(NO_APP_START_ON_EJECT)
 | |
| 			/* If the user ejected the volume, signal bootloader exit at next opportunity. */
 | |
| 			RunBootloader = ((MSInterfaceInfo->State.CommandBlock.SCSICommandData[4] & 0x03) != 0x02);
 | |
| #endif
 | |
| 		case SCSI_CMD_SEND_DIAGNOSTIC:
 | |
| 		case SCSI_CMD_TEST_UNIT_READY:
 | |
| 		case SCSI_CMD_PREVENT_ALLOW_MEDIUM_REMOVAL:
 | |
| 		case SCSI_CMD_VERIFY_10:
 | |
| 			/* These commands should just succeed, no handling required */
 | |
| 			CommandSuccess = true;
 | |
| 			MSInterfaceInfo->State.CommandBlock.DataTransferLength = 0;
 | |
| 			break;
 | |
| 		default:
 | |
| 			/* Update the SENSE key to reflect the invalid command */
 | |
| 			SCSI_SET_SENSE(SCSI_SENSE_KEY_ILLEGAL_REQUEST,
 | |
| 		                   SCSI_ASENSE_INVALID_COMMAND,
 | |
| 		                   SCSI_ASENSEQ_NO_QUALIFIER);
 | |
| 			break;
 | |
| 	}
 | |
| 
 | |
| 	/* Check if command was successfully processed */
 | |
| 	if (CommandSuccess)
 | |
| 	{
 | |
| 		SCSI_SET_SENSE(SCSI_SENSE_KEY_GOOD,
 | |
| 		               SCSI_ASENSE_NO_ADDITIONAL_INFORMATION,
 | |
| 		               SCSI_ASENSEQ_NO_QUALIFIER);
 | |
| 
 | |
| 		return true;
 | |
| 	}
 | |
| 
 | |
| 	return false;
 | |
| }
 | |
| 
 | |
| /** Command processing for an issued SCSI INQUIRY command. This command returns information about the device's features
 | |
|  *  and capabilities to the host.
 | |
|  *
 | |
|  *  \param[in] MSInterfaceInfo  Pointer to the Mass Storage class interface structure that the command is associated with
 | |
|  *
 | |
|  *  \return Boolean \c true if the command completed successfully, \c false otherwise.
 | |
|  */
 | |
| static bool SCSI_Command_Inquiry(USB_ClassInfo_MS_Device_t* const MSInterfaceInfo)
 | |
| {
 | |
| 	uint16_t AllocationLength  = SwapEndian_16(*(uint16_t*)&MSInterfaceInfo->State.CommandBlock.SCSICommandData[3]);
 | |
| 	uint16_t BytesTransferred  = MIN(AllocationLength, sizeof(InquiryData));
 | |
| 
 | |
| 	/* Only the standard INQUIRY data is supported, check if any optional INQUIRY bits set */
 | |
| 	if ((MSInterfaceInfo->State.CommandBlock.SCSICommandData[1] & ((1 << 0) | (1 << 1))) ||
 | |
| 	     MSInterfaceInfo->State.CommandBlock.SCSICommandData[2])
 | |
| 	{
 | |
| 		/* Optional but unsupported bits set - update the SENSE key and fail the request */
 | |
| 		SCSI_SET_SENSE(SCSI_SENSE_KEY_ILLEGAL_REQUEST,
 | |
| 		               SCSI_ASENSE_INVALID_FIELD_IN_CDB,
 | |
| 		               SCSI_ASENSEQ_NO_QUALIFIER);
 | |
| 
 | |
| 		return false;
 | |
| 	}
 | |
| 
 | |
| 	Endpoint_Write_Stream_LE(&InquiryData, BytesTransferred, NULL);
 | |
| 
 | |
| 	/* Pad out remaining bytes with 0x00 */
 | |
| 	Endpoint_Null_Stream((AllocationLength - BytesTransferred), NULL);
 | |
| 
 | |
| 	/* Finalize the stream transfer to send the last packet */
 | |
| 	Endpoint_ClearIN();
 | |
| 
 | |
| 	/* Succeed the command and update the bytes transferred counter */
 | |
| 	MSInterfaceInfo->State.CommandBlock.DataTransferLength -= BytesTransferred;
 | |
| 
 | |
| 	return true;
 | |
| }
 | |
| 
 | |
| /** Command processing for an issued SCSI REQUEST SENSE command. This command returns information about the last issued command,
 | |
|  *  including the error code and additional error information so that the host can determine why a command failed to complete.
 | |
|  *
 | |
|  *  \param[in] MSInterfaceInfo  Pointer to the Mass Storage class interface structure that the command is associated with
 | |
|  *
 | |
|  *  \return Boolean \c true if the command completed successfully, \c false otherwise.
 | |
|  */
 | |
| static bool SCSI_Command_Request_Sense(USB_ClassInfo_MS_Device_t* const MSInterfaceInfo)
 | |
| {
 | |
| 	uint8_t AllocationLength = MSInterfaceInfo->State.CommandBlock.SCSICommandData[4];
 | |
| 	uint8_t BytesTransferred = MIN(AllocationLength, sizeof(SenseData));
 | |
| 
 | |
| 	Endpoint_Write_Stream_LE(&SenseData, BytesTransferred, NULL);
 | |
| 	Endpoint_Null_Stream((AllocationLength - BytesTransferred), NULL);
 | |
| 	Endpoint_ClearIN();
 | |
| 
 | |
| 	/* Succeed the command and update the bytes transferred counter */
 | |
| 	MSInterfaceInfo->State.CommandBlock.DataTransferLength -= BytesTransferred;
 | |
| 
 | |
| 	return true;
 | |
| }
 | |
| 
 | |
| /** Command processing for an issued SCSI READ CAPACITY (10) command. This command returns information about the device's capacity
 | |
|  *  on the selected Logical Unit (drive), as a number of OS-sized blocks.
 | |
|  *
 | |
|  *  \param[in] MSInterfaceInfo  Pointer to the Mass Storage class interface structure that the command is associated with
 | |
|  *
 | |
|  *  \return Boolean \c true if the command completed successfully, \c false otherwise.
 | |
|  */
 | |
| static bool SCSI_Command_Read_Capacity_10(USB_ClassInfo_MS_Device_t* const MSInterfaceInfo)
 | |
| {
 | |
| 	Endpoint_Write_32_BE(LUN_MEDIA_BLOCKS - 1);
 | |
| 	Endpoint_Write_32_BE(SECTOR_SIZE_BYTES);
 | |
| 	Endpoint_ClearIN();
 | |
| 
 | |
| 	/* Succeed the command and update the bytes transferred counter */
 | |
| 	MSInterfaceInfo->State.CommandBlock.DataTransferLength -= 8;
 | |
| 
 | |
| 	return true;
 | |
| }
 | |
| 
 | |
| /** Command processing for an issued SCSI READ (10) or WRITE (10) command. This command reads in the block start address
 | |
|  *  and total number of blocks to process, then calls the appropriate low-level Dataflash routine to handle the actual
 | |
|  *  reading and writing of the data.
 | |
|  *
 | |
|  *  \param[in] MSInterfaceInfo  Pointer to the Mass Storage class interface structure that the command is associated with
 | |
|  *  \param[in] IsDataRead  Indicates if the command is a READ (10) command or WRITE (10) command (DATA_READ or DATA_WRITE)
 | |
|  *
 | |
|  *  \return Boolean \c true if the command completed successfully, \c false otherwise.
 | |
|  */
 | |
| static bool SCSI_Command_ReadWrite_10(USB_ClassInfo_MS_Device_t* const MSInterfaceInfo,
 | |
|                                       const bool IsDataRead)
 | |
| {
 | |
| 	uint16_t BlockAddress;
 | |
| 	uint16_t TotalBlocks;
 | |
| 
 | |
| 	/* Load in the 32-bit block address (SCSI uses big-endian, so have to reverse the byte order) */
 | |
| 	BlockAddress = SwapEndian_32(*(uint32_t*)&MSInterfaceInfo->State.CommandBlock.SCSICommandData[2]);
 | |
| 
 | |
| 	/* Load in the 16-bit total blocks (SCSI uses big-endian, so have to reverse the byte order) */
 | |
| 	TotalBlocks  = SwapEndian_16(*(uint16_t*)&MSInterfaceInfo->State.CommandBlock.SCSICommandData[7]);
 | |
| 
 | |
| 	/* Check if the block address is outside the maximum allowable value for the LUN */
 | |
| 	if (BlockAddress >= LUN_MEDIA_BLOCKS)
 | |
| 	{
 | |
| 		/* Block address is invalid, update SENSE key and return command fail */
 | |
| 		SCSI_SET_SENSE(SCSI_SENSE_KEY_ILLEGAL_REQUEST,
 | |
| 		               SCSI_ASENSE_LOGICAL_BLOCK_ADDRESS_OUT_OF_RANGE,
 | |
| 		               SCSI_ASENSEQ_NO_QUALIFIER);
 | |
| 
 | |
| 		return false;
 | |
| 	}
 | |
| 
 | |
| 	/* Determine if the packet is a READ (10) or WRITE (10) command, call appropriate function */
 | |
| 	for (uint16_t i = 0; i < TotalBlocks; i++)
 | |
| 	{
 | |
| 		if (IsDataRead == DATA_READ)
 | |
| 		  VirtualFAT_ReadBlock(BlockAddress + i);
 | |
| 		else
 | |
| 		  VirtualFAT_WriteBlock(BlockAddress + i);
 | |
| 	}
 | |
| 
 | |
| 	/* Update the bytes transferred counter and succeed the command */
 | |
| 	MSInterfaceInfo->State.CommandBlock.DataTransferLength -= ((uint32_t)TotalBlocks * SECTOR_SIZE_BYTES);
 | |
| 
 | |
| 	return true;
 | |
| }
 | |
| 
 | |
| /** Command processing for an issued SCSI MODE SENSE (6) command. This command returns various informational pages about
 | |
|  *  the SCSI device, as well as the device's Write Protect status.
 | |
|  *
 | |
|  *  \param[in] MSInterfaceInfo  Pointer to the Mass Storage class interface structure that the command is associated with
 | |
|  *
 | |
|  *  \return Boolean \c true if the command completed successfully, \c false otherwise.
 | |
|  */
 | |
| static bool SCSI_Command_ModeSense_6(USB_ClassInfo_MS_Device_t* const MSInterfaceInfo)
 | |
| {
 | |
| 	/* Send an empty header response indicating Write Protect flag is off */
 | |
| 	Endpoint_Write_32_LE(0);
 | |
| 	Endpoint_ClearIN();
 | |
| 
 | |
| 	/* Update the bytes transferred counter and succeed the command */
 | |
| 	MSInterfaceInfo->State.CommandBlock.DataTransferLength -= 4;
 | |
| 
 | |
| 	return true;
 | |
| }
 | |
| 
 | 
