tyler.durden
/
audk
mirror of https://github.com/acidanthera/audk.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

MdeModulePkg/IpIoLib: Check the input parameters before use them. 

This patch updates the DxeIpIoLib to check the input parameters before using.

Contributed-under: TianoCore Contribution Agreement 1.0
Signed-off-by: Fu Siyuan <siyuan.fu@intel.com>
Reviewed-by: Jiaxin Wu <jiaxin.wu@intel.com>
This commit is contained in:
Fu Siyuan 2017-12-13 16:06:42 +08:00
parent 6dbfed92f8
commit 6ccfeec24c
2 changed files with 68 additions and 25 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
MdeModulePkg/Include/Library/IpIoLib.h
View File

@ -360,6 +360,7 @@ IpIoDestroy (
@param[in, out] IpIo The pointer to the IP_IO instance that needs to stop.
@retval EFI_SUCCESS The IP_IO instance stopped successfully.
@retval EFI_INVALID_PARAMETER Invalid input parameter.
@retval Others Anrror condition occurred.
**/
@ -383,10 +384,11 @@ IpIoStop (
@retval EFI_SUCCESS The IP_IO instance opened with OpenData
successfully.
@retval EFI_ACCESS_DENIED The IP_IO instance is configured; avoid
reopening it.
@retval EFI_ACCESS_DENIED The IP_IO instance is configured, avoid to
reopen it.
@retval EFI_UNSUPPORTED IPv4 RawData mode is no supported.
@retval Others An error condition occurred.
@retval EFI_INVALID_PARAMETER Invalid input parameter.
@retval Others Error condition occurred.
**/
EFI_STATUS
@ -522,7 +524,7 @@ IpIoRemoveIp (
@param[in] Src The local IP address.
@return The pointer to the IP protocol can be used for sending purpose and its local
address is the same with Src.
address is the same with Src. NULL if failed.
**/
IP_IO_IP_INFO *

53
MdeModulePkg/Library/DxeIpIoLib/DxeIpIoLib.c
View File

@ -284,14 +284,21 @@ IpIoIcmpv4Handler (
UINT8 Code;
UINT32 TrimBytes;
ASSERT (IpIo != NULL);
ASSERT (Pkt != NULL);
ASSERT (Session != NULL);
ASSERT (IpIo->IpVersion == IP_VERSION_4);
IcmpHdr = NET_PROTO_HDR (Pkt, IP4_ICMP_ERROR_HEAD);
IpHdr = (EFI_IP4_HEADER *) (&IcmpHdr->IpHead);
//
// Check the ICMP packet length.
//
if (Pkt->TotalSize < sizeof (IP4_ICMP_ERROR_HEAD)) {
return EFI_ABORTED;
}
IcmpHdr = NET_PROTO_HDR (Pkt, IP4_ICMP_ERROR_HEAD);
IpHdr = (EFI_IP4_HEADER *) (&IcmpHdr->IpHead);
if (Pkt->TotalSize < ICMP_ERRLEN (IpHdr)) {
return EFI_ABORTED;
@ -421,6 +428,9 @@ IpIoIcmpv6Handler (
UINT32 TrimBytes;
BOOLEAN Flag;
ASSERT (IpIo != NULL);
ASSERT (Pkt != NULL);
ASSERT (Session != NULL);
ASSERT (IpIo->IpVersion == IP_VERSION_6);
//
@ -1043,6 +1053,7 @@ IpIoListenHandlerDpc (
}
if (IpIo->IpVersion == IP_VERSION_4) {
ASSERT (RxData->Ip4RxData.Header != NULL);
if (IP4_IS_LOCAL_BROADCAST (EFI_IP4 (RxData->Ip4RxData.Header->SourceAddress))) {
//
// The source address is a broadcast address, discard it.
@ -1066,6 +1077,11 @@ IpIoListenHandlerDpc (
goto CleanUp;
}
//
// The fragment should always be valid for non-zero length packet.
//
ASSERT (RxData->Ip4RxData.FragmentCount != 0);
//
// Create a netbuffer representing IPv4 packet
//
@ -1090,7 +1106,7 @@ IpIoListenHandlerDpc (
Session.IpHdrLen = RxData->Ip4RxData.HeaderLength;
Session.IpVersion = IP_VERSION_4;
} else {
ASSERT (RxData->Ip6RxData.Header != NULL);
if (!NetIp6IsValidUnicast(&RxData->Ip6RxData.Header->SourceAddress)) {
goto CleanUp;
}
@ -1102,6 +1118,11 @@ IpIoListenHandlerDpc (
goto CleanUp;
}
//
// The fragment should always be valid for non-zero length packet.
//
ASSERT (RxData->Ip6RxData.FragmentCount != 0);
//
// Create a netbuffer representing IPv6 packet
//
@ -1292,6 +1313,7 @@ ReleaseIpIo:
@retval EFI_ACCESS_DENIED The IP_IO instance is configured, avoid to
reopen it.
@retval EFI_UNSUPPORTED IPv4 RawData mode is no supported.
@retval EFI_INVALID_PARAMETER Invalid input parameter.
@retval Others Error condition occurred.
**/
@ -1305,6 +1327,10 @@ IpIoOpen (
EFI_STATUS Status;
UINT8 IpVersion;
if (IpIo == NULL || OpenData == NULL) {
return EFI_INVALID_PARAMETER;
}
if (IpIo->IsConfigured) {
return EFI_ACCESS_DENIED;
}
@ -1417,6 +1443,7 @@ ErrorExit:
@param[in, out] IpIo Pointer to the IP_IO instance that needs to stop.
@retval EFI_SUCCESS The IP_IO instance stopped successfully.
@retval EFI_INVALID_PARAMETER Invalid input parameter.
@retval Others Error condition occurred.
**/
@ -1430,6 +1457,10 @@ IpIoStop (
IP_IO_IP_INFO *IpInfo;
UINT8 IpVersion;
if (IpIo == NULL) {
return EFI_INVALID_PARAMETER;
}
if (!IpIo->IsConfigured) {
return EFI_SUCCESS;
}
@ -1958,6 +1989,10 @@ IpIoRemoveIp (
UINT8 IpVersion;
if (IpIo == NULL || IpInfo == NULL) {
return;
}
ASSERT (IpInfo->RefCnt > 0);
NET_PUT_REF (IpInfo);
@ -2021,7 +2056,7 @@ IpIoRemoveIp (
@param[in] Src The local IP address.
@return Pointer to the IP protocol can be used for sending purpose and its local
address is the same with Src.
address is the same with Src. NULL if failed.
**/
IP_IO_IP_INFO *
@ -2037,7 +2072,13 @@ IpIoFindSender (
LIST_ENTRY *IpInfoEntry;
IP_IO_IP_INFO *IpInfo;
ASSERT ((IpVersion == IP_VERSION_4) || (IpVersion == IP_VERSION_6));
if (IpIo == NULL || Src == NULL) {
return NULL;
}
if ((IpVersion != IP_VERSION_4) && (IpVersion != IP_VERSION_6)) {
return NULL;
}
NET_LIST_FOR_EACH (IpIoEntry, &mActiveIpIoList) {
IpIoPtr = NET_LIST_USER_STRUCT (IpIoEntry, IP_IO, Entry);