From 705f53a9b49e7ee3d1ca4bcc7133534ed64590dc Mon Sep 17 00:00:00 2001
From: qianouyang <qianouyang@6f19259b-4bc3-4df7-8a09-765794883524>
Date: Fri, 15 Oct 2010 05:40:41 +0000
Subject: Update IPsec.h file to follow approved ECR which will be collected
 into future UEFI 2.3 Specification after 2.3 errata B and future UEFI
 Specifications after 2.3. The changes mainly include:   1. Add
 EFI_IPSEC2_PROTOCOL   2. Remove IPsec Authentication Algorithm Definition and
 IPsec Encryption Algorithm      Definition.   3. Add EFI_IPSEC_SA_DATA2 data
 structure. And also update IPv4 driver to call EFI_IPSEC2_PROTOCOL.

git-svn-id: https://edk2.svn.sourceforge.net/svnroot/edk2/trunk/edk2@10941 6f19259b-4bc3-4df7-8a09-765794883524
---
 MdeModulePkg/Library/DxeNetLib/NetBuffer.c | 48 ++++++++++++++++++++++++++++++
 1 file changed, 48 insertions(+)

(limited to 'MdeModulePkg/Library')

diff --git a/MdeModulePkg/Library/DxeNetLib/NetBuffer.c b/MdeModulePkg/Library/DxeNetLib/NetBuffer.c
index 43bcae6600..bbbdbc048a 100644
--- a/MdeModulePkg/Library/DxeNetLib/NetBuffer.c
+++ b/MdeModulePkg/Library/DxeNetLib/NetBuffer.c
@@ -1842,3 +1842,51 @@ NetIp6PseudoHeadChecksum (
   return NetblockChecksum ((UINT8 *) &Hdr, sizeof (Hdr));
 }
 
+/**
+  The function frees the net buffer which allocated by the IP protocol. It releases 
+  only the net buffer and doesn't call the external free function. 
+
+  This function should be called after finishing the process of mIpSec->ProcessExt() 
+  for outbound traffic. The (EFI_IPSEC2_PROTOCOL)->ProcessExt() allocates a new 
+  buffer for the ESP, so there needs a function to free the old net buffer.
+
+  @param[in]  Nbuf       The network buffer to be freed.
+
+**/
+VOID
+NetIpSecNetbufFree (
+  NET_BUF   *Nbuf
+  )
+{
+  NET_CHECK_SIGNATURE (Nbuf, NET_BUF_SIGNATURE);
+  ASSERT (Nbuf->RefCnt > 0);
+
+  Nbuf->RefCnt--;
+
+  if (Nbuf->RefCnt == 0) {
+    
+    //
+    // Update Vector only when NBuf is to be released. That is,
+    // all the sharing of Nbuf increse Vector's RefCnt by one
+    //
+    NET_CHECK_SIGNATURE (Nbuf->Vector, NET_VECTOR_SIGNATURE);
+    ASSERT (Nbuf->Vector->RefCnt > 0);
+
+    Nbuf->Vector->RefCnt--;
+
+    if (Nbuf->Vector->RefCnt > 0) {
+      return;
+    }
+
+    //
+    // If NET_VECTOR_OWN_FIRST is set, release the first block since it is 
+    // allocated by us
+    //
+    if ((Nbuf->Vector->Flag & NET_VECTOR_OWN_FIRST) != 0) {
+      FreePool (Nbuf->Vector->Block[0].Bulk);
+    }
+    FreePool (Nbuf->Vector);
+    FreePool (Nbuf); 
+  } 
+}
+
-- 
cgit v1.2.3