+/**\r
+ Add Count of TX buffers to MnpDeviceData->AllTxBufList and MnpDeviceData->FreeTxBufList.\r
+ The length of the buffer is specified by MnpDeviceData->BufferLength.\r
+\r
+ @param[in, out] MnpDeviceData Pointer to the MNP_DEVICE_DATA.\r
+ @param[in] Count Number of TX buffers to add.\r
+\r
+ @retval EFI_SUCCESS The specified amount of TX buffers are allocated.\r
+ @retval EFI_OUT_OF_RESOURCES Failed to allocate a TX buffer.\r
+\r
+**/\r
+EFI_STATUS\r
+MnpAddFreeTxBuf (\r
+ IN OUT MNP_DEVICE_DATA *MnpDeviceData,\r
+ IN UINTN Count\r
+ )\r
+{\r
+ EFI_STATUS Status;\r
+ UINT32 Index;\r
+ MNP_TX_BUF_WRAP *TxBufWrap;\r
+\r
+ NET_CHECK_SIGNATURE (MnpDeviceData, MNP_DEVICE_DATA_SIGNATURE);\r
+ ASSERT ((Count > 0) && (MnpDeviceData->BufferLength > 0));\r
+\r
+ Status = EFI_SUCCESS;\r
+ for (Index = 0; Index < Count; Index++) {\r
+ TxBufWrap = (MNP_TX_BUF_WRAP*) AllocatePool (sizeof (MNP_TX_BUF_WRAP) + MnpDeviceData->BufferLength - 1);\r
+ if (TxBufWrap == NULL) {\r
+ DEBUG ((EFI_D_ERROR, "MnpAddFreeTxBuf: TxBuf Alloc failed.\n"));\r
+\r
+ Status = EFI_OUT_OF_RESOURCES;\r
+ break;\r
+ }\r
+ DEBUG ((EFI_D_INFO, "MnpAddFreeTxBuf: Add TxBufWrap %p, TxBuf %p\n", TxBufWrap, TxBufWrap->TxBuf));\r
+ TxBufWrap->Signature = MNP_TX_BUF_WRAP_SIGNATURE;\r
+ TxBufWrap->InUse = FALSE;\r
+ InsertTailList (&MnpDeviceData->FreeTxBufList, &TxBufWrap->WrapEntry);\r
+ InsertTailList (&MnpDeviceData->AllTxBufList, &TxBufWrap->AllEntry);\r
+ }\r
+\r
+ MnpDeviceData->TxBufCount += Index;\r
+ return Status;\r
+}\r
+\r
+/**\r
+ Allocate a free TX buffer from MnpDeviceData->FreeTxBufList. If there is none\r
+ in the queue, first try to recycle some from SNP, then try to allocate some and add \r
+ them into the queue, then fetch the NET_BUF from the updated FreeTxBufList.\r
+\r
+ @param[in, out] MnpDeviceData Pointer to the MNP_DEVICE_DATA.\r
+\r
+ @return Pointer to the allocated free NET_BUF structure, if NULL the\r
+ operation is failed.\r
+\r
+**/\r
+UINT8 *\r
+MnpAllocTxBuf (\r
+ IN OUT MNP_DEVICE_DATA *MnpDeviceData\r
+ )\r
+{\r
+ EFI_TPL OldTpl;\r
+ UINT8 *TxBuf;\r
+ EFI_STATUS Status;\r
+ LIST_ENTRY *Entry;\r
+ MNP_TX_BUF_WRAP *TxBufWrap;\r
+ \r
+ NET_CHECK_SIGNATURE (MnpDeviceData, MNP_DEVICE_DATA_SIGNATURE);\r
+\r
+ OldTpl = gBS->RaiseTPL (TPL_CALLBACK);\r
+\r
+ if (IsListEmpty (&MnpDeviceData->FreeTxBufList)) {\r
+ //\r
+ // First try to recycle some TX buffer from SNP\r
+ //\r
+ Status = MnpRecycleTxBuf (MnpDeviceData);\r
+ if (EFI_ERROR (Status)) {\r
+ TxBuf = NULL;\r
+ goto ON_EXIT;\r
+ }\r
+\r
+ //\r
+ // If still no free TX buffer, allocate more.\r
+ //\r
+ if (IsListEmpty (&MnpDeviceData->FreeTxBufList)) {\r
+ if ((MnpDeviceData->TxBufCount + MNP_TX_BUFFER_INCREASEMENT) > MNP_MAX_TX_BUFFER_NUM) {\r
+ DEBUG (\r
+ (EFI_D_ERROR,\r
+ "MnpAllocTxBuf: The maximum TxBuf size is reached for MNP driver instance %p.\n",\r
+ MnpDeviceData)\r
+ );\r
+\r
+ TxBuf = NULL;\r
+ goto ON_EXIT;\r
+ }\r
+\r
+ Status = MnpAddFreeTxBuf (MnpDeviceData, MNP_TX_BUFFER_INCREASEMENT);\r
+ if (IsListEmpty (&MnpDeviceData->FreeTxBufList)) {\r
+ DEBUG (\r
+ (EFI_D_ERROR,\r
+ "MnpAllocNbuf: Failed to add TxBuf into the FreeTxBufList, %r.\n",\r
+ Status)\r
+ );\r
+\r
+ TxBuf = NULL;\r
+ goto ON_EXIT;\r
+ }\r
+ }\r
+ }\r
+\r
+ ASSERT (!IsListEmpty (&MnpDeviceData->FreeTxBufList));\r
+ Entry = MnpDeviceData->FreeTxBufList.ForwardLink;\r
+ RemoveEntryList (MnpDeviceData->FreeTxBufList.ForwardLink);\r
+ TxBufWrap = NET_LIST_USER_STRUCT_S (Entry, MNP_TX_BUF_WRAP, WrapEntry, MNP_TX_BUF_WRAP_SIGNATURE);\r
+ TxBufWrap->InUse = TRUE;\r
+ TxBuf = TxBufWrap->TxBuf;\r
+\r
+ON_EXIT:\r
+ gBS->RestoreTPL (OldTpl);\r
+\r
+ return TxBuf;\r
+}\r
+\r
+/**\r
+ Try to reclaim the TX buffer into the buffer pool.\r
+\r
+ @param[in, out] MnpDeviceData Pointer to the mnp device context data.\r
+ @param[in, out] TxBuf Pointer to the TX buffer to free.\r
+\r
+**/\r
+VOID\r
+MnpFreeTxBuf (\r
+ IN OUT MNP_DEVICE_DATA *MnpDeviceData,\r
+ IN OUT UINT8 *TxBuf\r
+ )\r
+{\r
+ MNP_TX_BUF_WRAP *TxBufWrap;\r
+ EFI_TPL OldTpl;\r
+\r
+ NET_CHECK_SIGNATURE (MnpDeviceData, MNP_DEVICE_DATA_SIGNATURE);\r
+\r
+ if (TxBuf == NULL) {\r
+ return;\r
+ }\r
+\r
+ TxBufWrap = NET_LIST_USER_STRUCT (TxBuf, MNP_TX_BUF_WRAP, TxBuf);\r
+ if (TxBufWrap->Signature != MNP_TX_BUF_WRAP_SIGNATURE) {\r
+ DEBUG (\r
+ (EFI_D_ERROR,\r
+ "MnpFreeTxBuf: Signature check failed in MnpFreeTxBuf.\n")\r
+ );\r
+ return;\r
+ }\r
+\r
+ if (!TxBufWrap->InUse) {\r
+ DEBUG (\r
+ (EFI_D_WARN,\r
+ "MnpFreeTxBuf: Duplicated recycle report from SNP.\n")\r
+ );\r
+ return;\r
+ }\r
+ \r
+ OldTpl = gBS->RaiseTPL (TPL_CALLBACK);\r
+ InsertTailList (&MnpDeviceData->FreeTxBufList, &TxBufWrap->WrapEntry);\r
+ TxBufWrap->InUse = FALSE;\r
+ gBS->RestoreTPL (OldTpl);\r
+}\r
+\r
+/**\r
+ Try to recycle all the transmitted buffer address from SNP.\r
+\r
+ @param[in, out] MnpDeviceData Pointer to the mnp device context data.\r
+\r
+ @retval EFI_SUCCESS Successed to recyclethe transmitted buffer address.\r
+ @retval Others Failed to recyclethe transmitted buffer address.\r
+\r
+**/\r
+EFI_STATUS\r
+MnpRecycleTxBuf (\r
+ IN OUT MNP_DEVICE_DATA *MnpDeviceData\r
+ )\r
+{\r
+ UINT8 *TxBuf;\r
+ EFI_SIMPLE_NETWORK_PROTOCOL *Snp;\r
+ EFI_STATUS Status;\r
+\r
+ Snp = MnpDeviceData->Snp;\r
+ ASSERT (Snp != NULL);\r
+\r
+ do {\r
+ TxBuf = NULL;\r
+ Status = Snp->GetStatus (Snp, NULL, (VOID **) &TxBuf);\r
+ if (EFI_ERROR (Status)) {\r
+ return Status;\r
+ }\r
+\r
+ if (TxBuf != NULL) {\r
+ MnpFreeTxBuf (MnpDeviceData, TxBuf);\r
+ }\r
+ } while (TxBuf != NULL);\r
+\r
+ return EFI_SUCCESS;\r
+}\r