Refine USB Mass Storage Driver.

git-svn-id: https://edk2.svn.sourceforge.net/svnroot/edk2/trunk/edk2@7247 6f19259b-4bc3-4df7-8a09-765794883524

1 files changed, 3 insertions, 5 deletions

diff --git a/MdeModulePkg/Bus/Usb/UsbMassStorageDxe/UsbMassBot.c b/MdeModulePkg/Bus/Usb/UsbMassStorageDxe/UsbMassBot.c
index 178aa80d3e..c6c8f8d11b 100644
--- a/MdeModulePkg/Bus/Usb/UsbMassStorageDxe/UsbMassBot.c
+++ b/MdeModulePkg/Bus/Usb/UsbMassStorageDxe/UsbMassBot.c
@@ -13,7 +13,7 @@ WITHOUT WARRANTIES OR REPRESENTATIONS OF ANY KIND, EITHER EXPRESS OR IMPLIED.
 

 **/

 

-#include "UsbMass.h"

+#include "UsbMassBoot.h"

 #include "UsbMassBot.h"

 

 //

@@ -58,9 +58,7 @@ UsbBotInit (
   //

   // Allocate the BOT context for USB_BOT_PROTOCOL and two endpoint descriptors.

   //

-  UsbBot = AllocateZeroPool (

-             sizeof (USB_BOT_PROTOCOL) + 2 * sizeof (EFI_USB_ENDPOINT_DESCRIPTOR)

-             );

+  UsbBot = AllocateZeroPool (sizeof (USB_BOT_PROTOCOL) + 2 * sizeof (EFI_USB_ENDPOINT_DESCRIPTOR));

   ASSERT (UsbBot != NULL);

 

   UsbBot->UsbIo = UsbIo;

@@ -103,7 +101,7 @@ UsbBotInit (
        (UsbBot->BulkOutEndpoint == NULL)) {

 

       UsbBot->BulkOutEndpoint   = (EFI_USB_ENDPOINT_DESCRIPTOR *) (UsbBot + 1) + 1;

-      CopyMem(UsbBot->BulkOutEndpoint, &EndPoint, sizeof(EndPoint));

+      CopyMem (UsbBot->BulkOutEndpoint, &EndPoint, sizeof(EndPoint));

     }

   }