Before decrypting the packet, Ipsec will check if the inbound protected packet is...

[mirror_edk2.git] / NetworkPkg / IpSecDxe / IpSecDriver.c

diff --git a/NetworkPkg/IpSecDxe/IpSecDriver.c b/NetworkPkg/IpSecDxe/IpSecDriver.c
index 7e0d1fa4ed5e9ffd1b24c1fbbddadc26a0b5f947..bd12aa2e07a29b695af34002a76b2eca1776f77a 100644 (file)
--- a/NetworkPkg/IpSecDxe/IpSecDriver.c
+++ b/NetworkPkg/IpSecDxe/IpSecDriver.c
@@ -94,7 +94,7 @@ IpSecDriverBindingStart (
   IN EFI_DEVICE_PATH_PROTOCOL     *RemainingDevicePath OPTIONAL\r
   )\r
 {\r
-  EFI_IPSEC_PROTOCOL  *IpSec;\r
+  EFI_IPSEC2_PROTOCOL *IpSec;\r
   EFI_STATUS          Status;\r
   EFI_STATUS          Udp4Status;\r
   EFI_STATUS          Udp6Status;\r
@@ -103,7 +103,7 @@ IpSecDriverBindingStart (
   //\r
   // Ipsec protocol should be installed when load image.\r
   //\r
-  Status = gBS->LocateProtocol (&gEfiIpSecProtocolGuid, NULL, (VOID **) &IpSec);\r
+  Status = gBS->LocateProtocol (&gEfiIpSec2ProtocolGuid, NULL, (VOID **) &IpSec);\r
 \r
   if (EFI_ERROR (Status)) {\r
     return Status;\r
@@ -171,7 +171,7 @@ IpSecDriverBindingStop (
   IN EFI_HANDLE                   *ChildHandleBuffer\r
   )\r
 {\r
-  EFI_IPSEC_PROTOCOL  *IpSec;\r
+  EFI_IPSEC2_PROTOCOL *IpSec;\r
   EFI_STATUS          Status;\r
   IPSEC_PRIVATE_DATA  *Private;\r
   IKE_UDP_SERVICE     *UdpSrv;\r
@@ -181,7 +181,7 @@ IpSecDriverBindingStop (
   //\r
   // Locate ipsec protocol to get private data.\r
   //\r
-  Status = gBS->LocateProtocol (&gEfiIpSecProtocolGuid, NULL, (VOID **) &IpSec);\r
+  Status = gBS->LocateProtocol (&gEfiIpSec2ProtocolGuid, NULL, (VOID **) &IpSec);\r
 \r
   if (EFI_ERROR (Status)) {\r
     return Status;\r
@@ -301,12 +301,12 @@ IpSecDriverEntryPoint (
 {\r
   EFI_STATUS          Status;\r
   IPSEC_PRIVATE_DATA  *Private;\r
-  EFI_IPSEC_PROTOCOL  *IpSec;\r
+  EFI_IPSEC2_PROTOCOL *IpSec;\r
 \r
   //\r
   // Check whether ipsec protocol has already been installed.\r
   //\r
-  Status = gBS->LocateProtocol (&gEfiIpSecProtocolGuid, NULL, (VOID **) &IpSec);\r
+  Status = gBS->LocateProtocol (&gEfiIpSec2ProtocolGuid, NULL, (VOID **) &IpSec);\r
 \r
   if (!EFI_ERROR (Status)) {\r
     DEBUG ((DEBUG_WARN, "_ModuleEntryPoint: IpSec has been already loaded\n"));\r
@@ -345,7 +345,7 @@ IpSecDriverEntryPoint (
 \r
   Private->Signature    = IPSEC_PRIVATE_DATA_SIGNATURE;\r
   Private->ImageHandle  = ImageHandle;\r
-  CopyMem (&Private->IpSec, &mIpSecInstance, sizeof (EFI_IPSEC_PROTOCOL));\r
+  CopyMem (&Private->IpSec, &mIpSecInstance, sizeof (EFI_IPSEC2_PROTOCOL));\r
   \r
   //\r
   // Initilize Private's members. Thess members is used for IKE.\r
@@ -371,7 +371,7 @@ IpSecDriverEntryPoint (
   //\r
   Status = gBS->InstallMultipleProtocolInterfaces (\r
                   &Private->Handle,\r
-                  &gEfiIpSecProtocolGuid,\r
+                  &gEfiIpSec2ProtocolGuid,\r
                   &Private->IpSec,\r
                   NULL\r
                   );\r
@@ -396,7 +396,7 @@ IpSecDriverEntryPoint (
 ON_UNINSTALL_IPSEC:\r
   gBS->UninstallProtocolInterface (\r
          Private->Handle,\r
-         &gEfiIpSecProtocolGuid,\r
+         &gEfiIpSec2ProtocolGuid,\r
          &Private->IpSec\r
          );\r
 ON_UNINSTALL_CONFIG:\r