enhanced security check.

[mirror_edk2.git] / MdeModulePkg / Bus / Usb / UsbMassStorageDxe / UsbMassImpl.c

diff --git a/MdeModulePkg/Bus/Usb/UsbMassStorageDxe/UsbMassImpl.c b/MdeModulePkg/Bus/Usb/UsbMassStorageDxe/UsbMassImpl.c
index 15fdad573d91f2c151175f2ba098cba47e5d2e67..f80895bda4567afb58c29f65bf1297377383c1d6 100644 (file)
--- a/MdeModulePkg/Bus/Usb/UsbMassStorageDxe/UsbMassImpl.c
+++ b/MdeModulePkg/Bus/Usb/UsbMassStorageDxe/UsbMassImpl.c
@@ -14,14 +14,14 @@ WITHOUT WARRANTIES OR REPRESENTATIONS OF ANY KIND, EITHER EXPRESS OR IMPLIED.
 \r
 #include "UsbMassImpl.h"\r
 \r
+#define USB_MASS_TRANSPORT_COUNT    3\r
 //\r
 // Array of USB transport interfaces. \r
 //\r
-USB_MASS_TRANSPORT *mUsbMassTransport[] = {\r
+USB_MASS_TRANSPORT *mUsbMassTransport[USB_MASS_TRANSPORT_COUNT] = {\r
   &mUsbCbi0Transport,\r
   &mUsbCbi1Transport,\r
   &mUsbBotTransport,\r
-  NULL\r
 };\r
 \r
 EFI_DRIVER_BINDING_PROTOCOL gUSBMassDriverBinding = {\r
@@ -424,7 +424,7 @@ UsbMassInitTransport (
   // If not found, return EFI_UNSUPPORTED.\r
   // If found, execute USB_MASS_TRANSPORT.Init() to initialize the transport context.\r
   //\r
-  for (Index = 0; mUsbMassTransport[Index] != NULL; Index++) {\r
+  for (Index = 0; Index < USB_MASS_TRANSPORT_COUNT; Index++) {\r
     *Transport = mUsbMassTransport[Index];\r
 \r
     if (Interface.InterfaceProtocol == (*Transport)->Protocol) {\r
@@ -780,7 +780,7 @@ USBMassDriverBindingSupported (
   // If not found, return EFI_UNSUPPORTED.\r
   // If found, execute USB_MASS_TRANSPORT.Init() to initialize the transport context.\r
   //\r
-  for (Index = 0; mUsbMassTransport[Index] != NULL; Index++) {\r
+  for (Index = 0; Index < USB_MASS_TRANSPORT_COUNT; Index++) {\r
     Transport = mUsbMassTransport[Index];\r
     if (Interface.InterfaceProtocol == Transport->Protocol) {\r
       Status = Transport->Init (UsbIo, NULL);\r