Revision: 18961
          http://sourceforge.net/p/edk2/code/18961
Author:   luobozhang
Date:     2015-11-26 08:17:46 +0000 (Thu, 26 Nov 2015)
Log Message:
-----------
NetworkPkg:Fix NULL pointer dereference issues.

Revise some errors that some Null pointers may be dereferenced.

Contributed-under: TianoCore Contribution Agreement 1.0
Signed-off-by: Zhang Lubo <[email protected]>
Reviewed-by: Fu Siyuan <[email protected]>
Reviewed-by: Ye Ting <[email protected]>
Reviewed-by: Wu Jiaxin <[email protected]>

Modified Paths:
--------------
    trunk/edk2/NetworkPkg/HttpDxe/HttpImpl.c
    trunk/edk2/NetworkPkg/HttpDxe/HttpImpl.h
    trunk/edk2/NetworkPkg/HttpDxe/HttpProto.c

Modified: trunk/edk2/NetworkPkg/HttpDxe/HttpImpl.c
===================================================================
--- trunk/edk2/NetworkPkg/HttpDxe/HttpImpl.c    2015-11-26 07:01:08 UTC (rev 
18960)
+++ trunk/edk2/NetworkPkg/HttpDxe/HttpImpl.c    2015-11-26 08:17:46 UTC (rev 
18961)
@@ -40,6 +40,7 @@
                                   This is NULL.
                                   HttpConfigData is NULL.
                                   HttpConfigData->AccessPoint is NULL.
+  @retval EFI_OUT_OF_RESOURCES    Could not allocate enough system resources.
   @retval EFI_NOT_STARTED         The HTTP instance is not configured.
 
 **/
@@ -71,6 +72,9 @@
 
   if (HttpInstance->LocalAddressIsIPv6) {
     Http6AccessPoint = AllocateZeroPool (sizeof (EFI_HTTPv6_ACCESS_POINT));
+    if (Http6AccessPoint == NULL) {
+      return EFI_OUT_OF_RESOURCES;
+    }
     CopyMem (
       Http6AccessPoint,
       &HttpInstance->Ipv6Node,
@@ -79,6 +83,9 @@
     HttpConfigData->AccessPoint.IPv6Node = Http6AccessPoint;
   } else {
     Http4AccessPoint = AllocateZeroPool (sizeof (EFI_HTTPv4_ACCESS_POINT));
+    if (Http4AccessPoint == NULL) {
+      return EFI_OUT_OF_RESOURCES;
+    }
     CopyMem (
       Http4AccessPoint,
       &HttpInstance->IPv4Node,
@@ -885,6 +892,8 @@
       goto Error;
     }
 
+    ASSERT (HttpHeaders != NULL);
+
     //
     // Cache the part of body.
     //
@@ -1287,14 +1296,19 @@
   HttpInstance = HTTP_INSTANCE_FROM_PROTOCOL (This);
   ASSERT (HttpInstance != NULL);
 
-  if (HttpInstance->State != HTTP_STATE_TCP_CONNECTED || (HttpInstance->Tcp4 
== NULL && 
-                                                          HttpInstance->Tcp6 
== NULL)) {
+  if (HttpInstance->State != HTTP_STATE_TCP_CONNECTED) {
     return EFI_NOT_STARTED;
   }
   
   if (HttpInstance->LocalAddressIsIPv6) {
+    if (HttpInstance->Tcp6 == NULL) {
+      return EFI_NOT_STARTED;
+    }
     Status = HttpInstance->Tcp6->Poll (HttpInstance->Tcp6);
   } else {
+    if (HttpInstance->Tcp4 == NULL) {
+      return EFI_NOT_STARTED;
+    }
     Status = HttpInstance->Tcp4->Poll (HttpInstance->Tcp4);
   }
   

Modified: trunk/edk2/NetworkPkg/HttpDxe/HttpImpl.h
===================================================================
--- trunk/edk2/NetworkPkg/HttpDxe/HttpImpl.h    2015-11-26 07:01:08 UTC (rev 
18960)
+++ trunk/edk2/NetworkPkg/HttpDxe/HttpImpl.h    2015-11-26 08:17:46 UTC (rev 
18961)
@@ -44,6 +44,7 @@
                                   This is NULL.
                                   HttpConfigData is NULL.
                                   HttpConfigData->AccessPoint is NULL.
+  @retval EFI_OUT_OF_RESOURCES    Could not allocate enough system resources.
   @retval EFI_NOT_STARTED         The HTTP instance is not configured.
 
 **/

Modified: trunk/edk2/NetworkPkg/HttpDxe/HttpProto.c
===================================================================
--- trunk/edk2/NetworkPkg/HttpDxe/HttpProto.c   2015-11-26 07:01:08 UTC (rev 
18960)
+++ trunk/edk2/NetworkPkg/HttpDxe/HttpProto.c   2015-11-26 08:17:46 UTC (rev 
18961)
@@ -565,15 +565,14 @@
   EFI_TCP4_IO_TOKEN        *Rx4Token;
   EFI_TCP6_IO_TOKEN        *Rx6Token;
 
+  ASSERT (Wrap != NULL);
   HttpInstance   = Wrap->HttpInstance;
   Rx4Token       = NULL;
   Rx6Token       = NULL;
   
   if (HttpInstance->LocalAddressIsIPv6) {
-    if (Wrap != NULL) {
-      if (Wrap->TcpWrap.Rx6Token.CompletionToken.Event != NULL) {
-        gBS->CloseEvent (Wrap->TcpWrap.Rx6Token.CompletionToken.Event);
-      } 
+    if (Wrap->TcpWrap.Rx6Token.CompletionToken.Event != NULL) {
+      gBS->CloseEvent (Wrap->TcpWrap.Rx6Token.CompletionToken.Event);
     }
 
     if (HttpInstance->Rx6Token.CompletionToken.Event != NULL) {
@@ -581,10 +580,8 @@
       HttpInstance->Rx6Token.CompletionToken.Event = NULL;
     }
   } else {
-    if (Wrap != NULL) {
-      if (Wrap->TcpWrap.Rx4Token.CompletionToken.Event != NULL) {
-        gBS->CloseEvent (Wrap->TcpWrap.Rx4Token.CompletionToken.Event);
-      }
+    if (Wrap->TcpWrap.Rx4Token.CompletionToken.Event != NULL) {
+      gBS->CloseEvent (Wrap->TcpWrap.Rx4Token.CompletionToken.Event);
     }
     
     if (HttpInstance->Rx4Token.CompletionToken.Event != NULL) {
@@ -1891,23 +1888,22 @@
   EFI_TCP4_IO_TOKEN        *Rx4Token;
   EFI_TCP6_IO_TOKEN        *Rx6Token;
 
+  ASSERT (Wrap != NULL);
   HttpInstance   = Wrap->HttpInstance;
   Rx4Token       = NULL;
   Rx6Token       = NULL;
   
   if (HttpInstance->LocalAddressIsIPv6) {
-    if (Wrap != NULL) {
-      if (Wrap->TcpWrap.Rx6Token.CompletionToken.Event != NULL) {
-        gBS->CloseEvent (Wrap->TcpWrap.Rx6Token.CompletionToken.Event);
-      }
+    if (Wrap->TcpWrap.Rx6Token.CompletionToken.Event != NULL) {
+      gBS->CloseEvent (Wrap->TcpWrap.Rx6Token.CompletionToken.Event);
+    }
 
-      Rx6Token = &Wrap->TcpWrap.Rx6Token;
-      if (Rx6Token->Packet.RxData->FragmentTable[0].FragmentBuffer != NULL) {
-        FreePool (Rx6Token->Packet.RxData->FragmentTable[0].FragmentBuffer);
-        Rx6Token->Packet.RxData->FragmentTable[0].FragmentBuffer = NULL;
-      }
-      FreePool (Wrap); 
+    Rx6Token = &Wrap->TcpWrap.Rx6Token;
+    if (Rx6Token->Packet.RxData->FragmentTable[0].FragmentBuffer != NULL) {
+      FreePool (Rx6Token->Packet.RxData->FragmentTable[0].FragmentBuffer);
+      Rx6Token->Packet.RxData->FragmentTable[0].FragmentBuffer = NULL;
     }
+    FreePool (Wrap);
 
     if (HttpInstance->Rx6Token.CompletionToken.Event != NULL) {
       gBS->CloseEvent (HttpInstance->Rx6Token.CompletionToken.Event);
@@ -1921,18 +1917,16 @@
     }
     
   } else {
-    if (Wrap != NULL) {
-      if (Wrap->TcpWrap.Rx4Token.CompletionToken.Event != NULL) {
-        gBS->CloseEvent (Wrap->TcpWrap.Rx4Token.CompletionToken.Event);
-      }
-      Rx4Token = &Wrap->TcpWrap.Rx4Token;
-      if (Rx4Token->Packet.RxData->FragmentTable[0].FragmentBuffer != NULL) {
-        FreePool (Rx4Token->Packet.RxData->FragmentTable[0].FragmentBuffer);
-        Rx4Token->Packet.RxData->FragmentTable[0].FragmentBuffer = NULL;
-      }
-      FreePool (Wrap);
+    if (Wrap->TcpWrap.Rx4Token.CompletionToken.Event != NULL) {
+      gBS->CloseEvent (Wrap->TcpWrap.Rx4Token.CompletionToken.Event);
     }
-    
+    Rx4Token = &Wrap->TcpWrap.Rx4Token;
+    if (Rx4Token->Packet.RxData->FragmentTable[0].FragmentBuffer != NULL) {
+      FreePool (Rx4Token->Packet.RxData->FragmentTable[0].FragmentBuffer);
+      Rx4Token->Packet.RxData->FragmentTable[0].FragmentBuffer = NULL;
+    }
+    FreePool (Wrap);
+
     if (HttpInstance->Rx4Token.CompletionToken.Event != NULL) {
       gBS->CloseEvent (HttpInstance->Rx4Token.CompletionToken.Event);
       HttpInstance->Rx4Token.CompletionToken.Event = NULL;


------------------------------------------------------------------------------
Go from Idea to Many App Stores Faster with Intel(R) XDK
Give your users amazing mobile app experiences with Intel(R) XDK.
Use one codebase in this all-in-one HTML5 development environment.
Design, debug & build mobile apps & 2D/3D high-impact games for multiple OSs.
http://pubads.g.doubleclick.net/gampad/clk?id=254741551&iu=/4140
_______________________________________________
edk2-commits mailing list
[email protected]
https://lists.sourceforge.net/lists/listinfo/edk2-commits

Reply via email to