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

NetworkPkg: Replace the internal function with exposed one 

This patch is used to replace the internal function with
the exposed one defined in NetLib.h.

Cc: Hegde Nagaraj P <nagaraj-p.hegde@hpe.com>
Cc: El-Haj-Mahmoud Samer <samer.el-haj-mahmoud@hpe.com>
Cc: Ye Ting <ting.ye@intel.com>
Cc: Fu Siyuan <siyuan.fu@intel.com>
Contributed-under: TianoCore Contribution Agreement 1.0
Signed-off-by: Jiaxin Wu <jiaxin.wu@intel.com>
Reviewed-by: Hegde Nagaraj P <nagaraj-p.hegde@hpe.com>
Reviewed-by: Fu Siyuan <siyuan.fu@intel.com>
This commit is contained in:
Jiaxin Wu 2016-01-22 01:40:56 +08:00
parent dba6e9a935
commit 3700da7384
3 changed files with 8 additions and 78 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

63
NetworkPkg/DnsDxe/DnsImpl.c
View File

@ -1037,61 +1037,6 @@ AddDns6ServerIp (
return EFI_SUCCESS;
}
/**
Fill QName for IP querying. QName is a domain name represented as
a sequence of labels, where each label consists of a length octet
followed by that number of octets. The domain name terminates with
the zero length octet for the null label of the root. Caller should
take responsibility to the buffer in QName.
@param HostName Queried HostName
@retval NULL Failed to fill QName.
@return QName filled successfully.
**/
CHAR8 *
EFIAPI
DnsFillinQNameForQueryIp (
IN CHAR16 *HostName
)
{
CHAR8 *QueryName;
CHAR8 *Header;
CHAR8 *Tail;
UINTN Len;
UINTN Index;
QueryName = NULL;
Header = NULL;
Tail = NULL;
QueryName = AllocateZeroPool (DNS_DEFAULT_BLKSIZE);
if (QueryName == NULL) {
return NULL;
}
Header = QueryName;
Tail = Header + 1;
Len = 0;
for (Index = 0; HostName[Index] != 0; Index++) {
*Tail = (CHAR8) HostName[Index];
if (*Tail == '.') {
*Header = (CHAR8) Len;
Header = Tail;
Tail ++;
Len = 0;
} else {
Tail++;
Len++;
}
}
*Header = (CHAR8) Len;
*Tail = 0;
return QueryName;
}
/**
Find out whether the response is valid or invalid.
@ -1806,8 +1751,12 @@ ConstructDNSQuery (
NET_FRAGMENT Frag;
DNS_HEADER *DnsHeader;
DNS_QUERY_SECTION *DnsQuery;
Frag.Bulk = AllocatePool (DNS_DEFAULT_BLKSIZE * sizeof (UINT8));
//
// Messages carried by UDP are restricted to 512 bytes (not counting the IP
// or UDP headers).
//
Frag.Bulk = AllocatePool (DNS_MAX_MESSAGE_SIZE * sizeof (UINT8));
if (Frag.Bulk == NULL) {
return EFI_OUT_OF_RESOURCES;
}

19
NetworkPkg/DnsDxe/DnsImpl.h
View File

@ -87,7 +87,6 @@ extern EFI_DNS6_PROTOCOL mDns6Protocol;
#define DNS_DEFAULT_TIMEOUT 2
#define DNS_DEFAULT_RETRY 3
#define DNS_DEFAULT_BLKSIZE 512
#define DNS_TIME_TO_GETMAP 5
@ -556,24 +555,6 @@ AddDns6ServerIp (
IN EFI_IPv6_ADDRESS ServerIp
);
/**
Fill QName for IP querying. QName is a domain name represented as
a sequence of labels, where each label consists of a length octet
followed by that number of octets. The domain name terminates with
the zero length octet for the null label of the root.
@param HostName Queried HostName
@retval NULL Failed to fill QName.
@return QName filled successfully.
**/
CHAR8 *
EFIAPI
DnsFillinQNameForQueryIp (
IN CHAR16 *HostName
);
/**
Find out whether the response is valid or invalid.

4
NetworkPkg/DnsDxe/DnsProtocol.c
View File

@ -454,7 +454,7 @@ Dns4HostNameToIp (
//
// Construct QName.
//
QueryName = DnsFillinQNameForQueryIp (TokenEntry->QueryHostName);
QueryName = NetLibCreateDnsQName (TokenEntry->QueryHostName);
if (QueryName == NULL) {
Status = EFI_OUT_OF_RESOURCES;
goto ON_EXIT;
@ -1264,7 +1264,7 @@ Dns6HostNameToIp (
//
// Construct QName.
//
QueryName = DnsFillinQNameForQueryIp (TokenEntry->QueryHostName);
QueryName = NetLibCreateDnsQName (TokenEntry->QueryHostName);
if (QueryName == NULL) {
Status = EFI_OUT_OF_RESOURCES;
goto ON_EXIT;