]> git.proxmox.com Git - mirror_edk2.git/blobdiff - MdeModulePkg/Universal/Network/Ip4Dxe/Ip4Driver.c
MdeModulePkg: Remove trailing white space
[mirror_edk2.git] / MdeModulePkg / Universal / Network / Ip4Dxe / Ip4Driver.c
index 642e453e020eace3db9b8cfb5ec5e633382dff5d..0f2558141459a59b1ad1fc7e617076fee95c0467 100644 (file)
@@ -1,7 +1,7 @@
 /** @file\r
   The driver binding and service binding protocol for IP4 driver.\r
 \r
-Copyright (c) 2005 - 2016, Intel Corporation. All rights reserved.<BR>\r
+Copyright (c) 2005 - 2018, Intel Corporation. All rights reserved.<BR>\r
 (C) Copyright 2015 Hewlett-Packard Development Company, L.P.<BR>\r
 \r
 This program and the accompanying materials\r
@@ -41,12 +41,20 @@ IpSec2InstalledCallback (
   IN VOID       *Context\r
   )\r
 {\r
+  EFI_STATUS    Status;\r
   //\r
-  // Close the event so it does not get called again.\r
+  // Test if protocol was even found.\r
+  // Notification function will be called at least once.\r
   //\r
-  gBS->CloseEvent (Event);\r
+  Status = gBS->LocateProtocol (&gEfiIpSec2ProtocolGuid, NULL, (VOID **)&mIpSec);\r
+  if (Status == EFI_SUCCESS && mIpSec != NULL) {\r
+    //\r
+    // Close the event so it does not get called again.\r
+    //\r
+    gBS->CloseEvent (Event);\r
 \r
-  mIpSec2Installed = TRUE;\r
+    mIpSec2Installed = TRUE;\r
+  }\r
 }\r
 \r
 /**\r
@@ -245,11 +253,12 @@ Ip4CreateService (
   ZeroMem (&IpSb->SnpMode, sizeof (EFI_SIMPLE_NETWORK_MODE));\r
 \r
   IpSb->Timer = NULL;\r
+  IpSb->ReconfigCheckTimer = NULL;\r
 \r
   IpSb->ReconfigEvent = NULL;\r
 \r
   IpSb->Reconfig = FALSE;\r
-  \r
+\r
   IpSb->MediaPresent = TRUE;\r
 \r
   //\r
@@ -276,6 +285,18 @@ Ip4CreateService (
     goto ON_ERROR;\r
   }\r
 \r
+  Status = gBS->CreateEvent (\r
+                  EVT_NOTIFY_SIGNAL | EVT_TIMER,\r
+                  TPL_CALLBACK,\r
+                  Ip4TimerReconfigChecking,\r
+                  IpSb,\r
+                  &IpSb->ReconfigCheckTimer\r
+                  );\r
+\r
+  if (EFI_ERROR (Status)) {\r
+    goto ON_ERROR;\r
+  }\r
+\r
   Status = gBS->CreateEvent (\r
                   EVT_NOTIFY_SIGNAL,\r
                   TPL_NOTIFY,\r
@@ -331,7 +352,7 @@ Ip4CreateService (
 \r
   IpSb->MacString = NULL;\r
   Status = NetLibGetMacString (IpSb->Controller, IpSb->Image, &IpSb->MacString);\r
-  \r
+\r
   if (EFI_ERROR (Status)) {\r
     goto ON_ERROR;\r
   }\r
@@ -346,7 +367,7 @@ Ip4CreateService (
   InsertHeadList (&IpSb->Interfaces, &IpSb->DefaultInterface->Link);\r
 \r
   ZeroMem (&IpSb->Ip4Config2Instance, sizeof (IP4_CONFIG2_INSTANCE));\r
-  \r
+\r
   Status = Ip4Config2InitInstance (&IpSb->Ip4Config2Instance);\r
 \r
   if (EFI_ERROR (Status)) {\r
@@ -402,6 +423,13 @@ Ip4CleanService (
     IpSb->Timer = NULL;\r
   }\r
 \r
+  if (IpSb->ReconfigCheckTimer != NULL) {\r
+    gBS->SetTimer (IpSb->ReconfigCheckTimer, TimerCancel, 0);\r
+    gBS->CloseEvent (IpSb->ReconfigCheckTimer);\r
+\r
+    IpSb->ReconfigCheckTimer = NULL;\r
+  }\r
+\r
   if (IpSb->DefaultInterface != NULL) {\r
     Status = Ip4FreeInterface (IpSb->DefaultInterface, NULL);\r
 \r
@@ -460,7 +488,7 @@ Ip4CleanService (
 \r
 /**\r
   Callback function which provided by user to remove one node in NetDestroyLinkList process.\r
-  \r
+\r
   @param[in]    Entry           The entry to be removed.\r
   @param[in]    Context         Pointer to the callback context corresponds to the Context in NetDestroyLinkList.\r
 \r
@@ -521,7 +549,7 @@ Ip4DriverBindingStart (
   IN EFI_HANDLE                   ControllerHandle,\r
   IN EFI_DEVICE_PATH_PROTOCOL     *RemainingDevicePath OPTIONAL\r
   )\r
-{ \r
+{\r
   EFI_STATUS                    Status;\r
   IP4_SERVICE                   *IpSb;\r
   EFI_IP4_CONFIG2_PROTOCOL      *Ip4Cfg2;\r
@@ -547,13 +575,13 @@ Ip4DriverBindingStart (
   if (Status == EFI_SUCCESS) {\r
     return EFI_ALREADY_STARTED;\r
   }\r
-  \r
+\r
   Status = Ip4CreateService (ControllerHandle, This->DriverBindingHandle, &IpSb);\r
-  \r
+\r
   if (EFI_ERROR (Status)) {\r
     return Status;\r
   }\r
-  \r
+\r
   ASSERT (IpSb != NULL);\r
 \r
   Ip4Cfg2  = &IpSb->Ip4Config2Instance.Ip4Config2;\r
@@ -575,16 +603,16 @@ Ip4DriverBindingStart (
   }\r
 \r
   //\r
-  // Read the config data from NV variable again. \r
+  // Read the config data from NV variable again.\r
   // The default data can be changed by other drivers.\r
   //\r
   Status = Ip4Config2ReadConfigData (IpSb->MacString, &IpSb->Ip4Config2Instance);\r
   if (EFI_ERROR (Status)) {\r
     goto UNINSTALL_PROTOCOL;\r
   }\r
-  \r
+\r
   //\r
-  // Consume the installed EFI_IP4_CONFIG2_PROTOCOL to set the default data items. \r
+  // Consume the installed EFI_IP4_CONFIG2_PROTOCOL to set the default data items.\r
   //\r
   for (Index = Ip4Config2DataTypePolicy; Index < Ip4Config2DataTypeMaximum; Index++) {\r
     DataItem = &IpSb->Ip4Config2Instance.DataItem[Index];\r
@@ -598,13 +626,13 @@ Ip4DriverBindingStart (
       if (EFI_ERROR(Status)) {\r
         goto UNINSTALL_PROTOCOL;\r
       }\r
-      \r
+\r
       if (Index == Ip4Config2DataTypePolicy && (*(DataItem->Data.Policy) == Ip4Config2PolicyDhcp)) {\r
         break;\r
-      } \r
+      }\r
     }\r
   }\r
\r
+\r
   //\r
   // Ready to go: start the receiving and timer.\r
   // Ip4Config2SetPolicy maybe call Ip4ReceiveFrame() to set the default interface's RecvRequest first after\r
@@ -622,6 +650,12 @@ Ip4DriverBindingStart (
     goto UNINSTALL_PROTOCOL;\r
   }\r
 \r
+  Status = gBS->SetTimer (IpSb->ReconfigCheckTimer, TimerPeriodic, 500 * TICKS_PER_MS);\r
+\r
+  if (EFI_ERROR (Status)) {\r
+    goto UNINSTALL_PROTOCOL;\r
+  }\r
+\r
   //\r
   // Initialize the IP4 ID\r
   //\r
@@ -671,9 +705,9 @@ Ip4DriverBindingStop (
   )\r
 {\r
   EFI_SERVICE_BINDING_PROTOCOL             *ServiceBinding;\r
-  IP4_SERVICE                              *IpSb; \r
+  IP4_SERVICE                              *IpSb;\r
   EFI_HANDLE                               NicHandle;\r
-  EFI_STATUS                               Status; \r
+  EFI_STATUS                               Status;\r
   INTN                                     State;\r
   LIST_ENTRY                               *List;\r
   IP4_DESTROY_CHILD_IN_HANDLE_BUF_CONTEXT  Context;\r
@@ -683,20 +717,20 @@ Ip4DriverBindingStop (
   BOOLEAN                                  IsDhcp4;\r
 \r
   IsDhcp4   = FALSE;\r
-   \r
+\r
   NicHandle = NetLibGetNicHandle (ControllerHandle, &gEfiManagedNetworkProtocolGuid);\r
   if (NicHandle == NULL) {\r
     NicHandle = NetLibGetNicHandle (ControllerHandle, &gEfiArpProtocolGuid);\r
     if (NicHandle == NULL) {\r
       NicHandle = NetLibGetNicHandle (ControllerHandle, &gEfiDhcp4ProtocolGuid);\r
       if (NicHandle != NULL) {\r
-        IsDhcp4 = TRUE;  \r
+        IsDhcp4 = TRUE;\r
       } else {\r
         return EFI_SUCCESS;\r
       }\r
     }\r
   }\r
-   \r
+\r
   Status = gBS->OpenProtocol (\r
                   NicHandle,\r
                   &gEfiIp4ServiceBindingProtocolGuid,\r
@@ -708,7 +742,7 @@ Ip4DriverBindingStop (
   if (EFI_ERROR (Status)) {\r
     return EFI_DEVICE_ERROR;\r
   }\r
-  \r
+\r
   IpSb = IP4_SERVICE_FROM_PROTOCOL (ServiceBinding);\r
 \r
   if (IsDhcp4) {\r
@@ -727,7 +761,7 @@ Ip4DriverBindingStop (
                NULL\r
                );\r
   } else if (IpSb->DefaultInterface->ArpHandle == ControllerHandle) {\r
-  \r
+\r
     //\r
     // The ARP protocol for the default interface is being uninstalled and all\r
     // its IP child handles should have been destroyed before. So, release the\r
@@ -736,7 +770,7 @@ Ip4DriverBindingStop (
     Ip4CancelReceive (IpSb->DefaultInterface);\r
     Ip4FreeInterface (IpSb->DefaultInterface, NULL);\r
     Ip4FreeRouteTable (IpSb->DefaultRouteTable);\r
-    \r
+\r
     IpIf = Ip4CreateInterface (IpSb->Mnp, IpSb->Controller, IpSb->Image);\r
     if (IpIf == NULL) {\r
       goto ON_ERROR;\r
@@ -746,7 +780,7 @@ Ip4DriverBindingStop (
       Ip4FreeInterface (IpIf, NULL);\r
       goto ON_ERROR;;\r
     }\r
-    \r
+\r
     IpSb->DefaultInterface  = IpIf;\r
     InsertHeadList (&IpSb->Interfaces, &IpIf->Link);\r
     IpSb->DefaultRouteTable = RouteTable;\r
@@ -773,7 +807,7 @@ Ip4DriverBindingStop (
            &IpSb->Ip4Config2Instance.Ip4Config2,\r
            NULL\r
            );\r
-    \r
+\r
     if (gIp4ControllerNameTable != NULL) {\r
       FreeUnicodeStringTable (gIp4ControllerNameTable);\r
       gIp4ControllerNameTable = NULL;\r
@@ -922,7 +956,6 @@ Ip4ServiceBindingDestroyChild (
   IP4_PROTOCOL              *IpInstance;\r
   EFI_IP4_PROTOCOL          *Ip4;\r
   EFI_TPL                   OldTpl;\r
-  INTN                      State;\r
 \r
   if ((This == NULL) || (ChildHandle == NULL)) {\r
     return EFI_INVALID_PARAMETER;\r
@@ -960,13 +993,12 @@ Ip4ServiceBindingDestroyChild (
   // when UDP driver is being stopped, it will destroy all\r
   // the IP child it opens.\r
   //\r
-  if (IpInstance->State == IP4_STATE_DESTROY) {\r
+  if (IpInstance->InDestroy) {\r
     gBS->RestoreTPL (OldTpl);\r
     return EFI_SUCCESS;\r
   }\r
 \r
-  State             = IpInstance->State;\r
-  IpInstance->State = IP4_STATE_DESTROY;\r
+  IpInstance->InDestroy = TRUE;\r
 \r
   //\r
   // Close the Managed Network protocol.\r
@@ -1009,6 +1041,7 @@ Ip4ServiceBindingDestroyChild (
                   );\r
   OldTpl = gBS->RaiseTPL (TPL_CALLBACK);\r
   if (EFI_ERROR (Status)) {\r
+    IpInstance->InDestroy = FALSE;\r
     goto ON_ERROR;\r
   }\r
 \r
@@ -1033,7 +1066,6 @@ Ip4ServiceBindingDestroyChild (
   return EFI_SUCCESS;\r
 \r
 ON_ERROR:\r
-  IpInstance->State = State;\r
   gBS->RestoreTPL (OldTpl);\r
 \r
   return Status;\r