From d0db78432fc02bacbd57cc9f15eb05b4e56981cb Mon Sep 17 00:00:00 2001 From: Dean Camera Date: Tue, 21 Jul 2009 04:46:48 +0000 Subject: Add extra tests to the MassStorage device demo and class driver for validating command blocks from the host. --- Demos/Device/LowLevel/MassStorage/MassStorage.c | 2 ++ 1 file changed, 2 insertions(+) (limited to 'Demos') diff --git a/Demos/Device/LowLevel/MassStorage/MassStorage.c b/Demos/Device/LowLevel/MassStorage/MassStorage.c index ebb8e5c48..2c80fc595 100644 --- a/Demos/Device/LowLevel/MassStorage/MassStorage.c +++ b/Demos/Device/LowLevel/MassStorage/MassStorage.c @@ -256,6 +256,8 @@ static bool ReadInCommandBlock(void) /* Verify the command block - abort if invalid */ if ((CommandBlock.Signature != CBW_SIGNATURE) || (CommandBlock.LUN >= TOTAL_LUNS) || + (CommandBlock.Flags & 0x1F) || + (CommandBlock.SCSICommandLength == 0) || (CommandBlock.SCSICommandLength > MAX_SCSI_COMMAND_LENGTH)) { /* Stall both data pipes until reset by host */ -- cgit v1.2.3