| /** @file | |
| Implementation of EFI_DNS4_PROTOCOL and EFI_DNS6_PROTOCOL interfaces. | |
| Copyright (c) 2015, Intel Corporation. All rights reserved.<BR> | |
| This program and the accompanying materials | |
| are licensed and made available under the terms and conditions of the BSD License | |
| which accompanies this distribution. The full text of the license may be found at | |
| http://opensource.org/licenses/bsd-license.php | |
| THE PROGRAM IS DISTRIBUTED UNDER THE BSD LICENSE ON AN "AS IS" BASIS, | |
| WITHOUT WARRANTIES OR REPRESENTATIONS OF ANY KIND, EITHER EXPRESS OR IMPLIED. | |
| **/ | |
| #include "DnsImpl.h" | |
| EFI_DNS4_PROTOCOL mDns4Protocol = { | |
| Dns4GetModeData, | |
| Dns4Configure, | |
| Dns4HostNameToIp, | |
| Dns4IpToHostName, | |
| Dns4GeneralLookUp, | |
| Dns4UpdateDnsCache, | |
| Dns4Poll, | |
| Dns4Cancel | |
| }; | |
| EFI_DNS6_PROTOCOL mDns6Protocol = { | |
| Dns6GetModeData, | |
| Dns6Configure, | |
| Dns6HostNameToIp, | |
| Dns6IpToHostName, | |
| Dns6GeneralLookUp, | |
| Dns6UpdateDnsCache, | |
| Dns6Poll, | |
| Dns6Cancel | |
| }; | |
| /** | |
| This function is used to retrieve DNS mode data for this DNS instance. | |
| @param[in] This Pointer to EFI_DNS4_PROTOCOL instance. | |
| @param[out] DnsModeData Pointer to the caller-allocated storage for the EFI_DNS4_MODE_DATA structure. | |
| @retval EFI_SUCCESS The operation completed successfully. | |
| @retval EFI_NOT_STARTED When DnsConfigData is queried, no configuration data is | |
| available because this instance has not been configured. | |
| @retval EFI_OUT_OF_RESOURCES Failed to allocate needed resources. | |
| @retval EFI_INVALID_PARAMETER This is NULL or DnsModeData is NULL. | |
| **/ | |
| EFI_STATUS | |
| EFIAPI | |
| Dns4GetModeData ( | |
| IN EFI_DNS4_PROTOCOL *This, | |
| OUT EFI_DNS4_MODE_DATA *DnsModeData | |
| ) | |
| { | |
| DNS_INSTANCE *Instance; | |
| EFI_TPL OldTpl; | |
| UINTN Index; | |
| LIST_ENTRY *Entry; | |
| LIST_ENTRY *Next; | |
| DNS4_SERVER_IP *ServerItem; | |
| EFI_IPv4_ADDRESS *ServerList; | |
| DNS4_CACHE *CacheItem; | |
| EFI_DNS4_CACHE_ENTRY *CacheList; | |
| EFI_STATUS Status; | |
| ServerItem = NULL; | |
| ServerList = NULL; | |
| CacheItem = NULL; | |
| CacheList = NULL; | |
| Status = EFI_SUCCESS; | |
| if ((This == NULL) || (DnsModeData == NULL)) { | |
| return EFI_INVALID_PARAMETER; | |
| } | |
| OldTpl = gBS->RaiseTPL (TPL_CALLBACK); | |
| Instance = DNS_INSTANCE_FROM_THIS_PROTOCOL4 (This); | |
| if (Instance->State == DNS_STATE_UNCONFIGED) { | |
| gBS->RestoreTPL (OldTpl); | |
| return EFI_NOT_STARTED; | |
| } | |
| ZeroMem (DnsModeData, sizeof (EFI_DNS4_MODE_DATA)); | |
| // | |
| // Get the current configuration data of this instance. | |
| // | |
| Status = Dns4CopyConfigure (&DnsModeData->DnsConfigData, &Instance->Dns4CfgData); | |
| if (EFI_ERROR (Status)) { | |
| gBS->RestoreTPL (OldTpl); | |
| return Status; | |
| } | |
| // | |
| // Get the DnsServerCount and DnsServerList | |
| // | |
| Index = 0; | |
| NET_LIST_FOR_EACH_SAFE (Entry, Next, &mDriverData->Dns4ServerList) { | |
| Index++; | |
| } | |
| DnsModeData->DnsServerCount = (UINT32) Index; | |
| ServerList = AllocatePool (sizeof (EFI_IPv4_ADDRESS) * DnsModeData->DnsServerCount); | |
| ASSERT (ServerList != NULL); | |
| Index = 0; | |
| NET_LIST_FOR_EACH_SAFE (Entry, Next, &mDriverData->Dns4ServerList) { | |
| ServerItem = NET_LIST_USER_STRUCT (Entry, DNS4_SERVER_IP, AllServerLink); | |
| CopyMem (ServerList + Index, &ServerItem->Dns4ServerIp, sizeof (EFI_IPv4_ADDRESS)); | |
| Index++; | |
| } | |
| DnsModeData->DnsServerList = ServerList; | |
| // | |
| // Get the DnsCacheCount and DnsCacheList | |
| // | |
| Index =0; | |
| NET_LIST_FOR_EACH_SAFE (Entry, Next, &mDriverData->Dns4CacheList) { | |
| Index++; | |
| } | |
| DnsModeData->DnsCacheCount = (UINT32) Index; | |
| CacheList = AllocatePool (sizeof (EFI_DNS4_CACHE_ENTRY) * DnsModeData->DnsCacheCount); | |
| ASSERT (CacheList != NULL); | |
| Index =0; | |
| NET_LIST_FOR_EACH_SAFE (Entry, Next, &mDriverData->Dns4CacheList) { | |
| CacheItem = NET_LIST_USER_STRUCT (Entry, DNS4_CACHE, AllCacheLink); | |
| CopyMem (CacheList + Index, &CacheItem->DnsCache, sizeof (EFI_DNS4_CACHE_ENTRY)); | |
| Index++; | |
| } | |
| DnsModeData->DnsCacheList = CacheList; | |
| gBS->RestoreTPL (OldTpl); | |
| return EFI_SUCCESS; | |
| } | |
| /** | |
| This function is used to configure DNS configuration data for this DNS instance. | |
| @param[in] This Pointer to EFI_DNS4_PROTOCOL instance. | |
| @param[in] DnsConfigData Pointer to caller-allocated buffer containing EFI_DNS4_CONFIG_DATA structure. | |
| If NULL, the driver will reinitialize the protocol instance to the unconfigured state. | |
| @retval EFI_SUCCESS The operation completed successfully. | |
| @retval EFI_UNSUPPORTED The designated protocol is not supported. | |
| @retval EFI_OUT_OF_RESOURCES Failed to allocate needed resources. | |
| @retval EFI_INVALID_PARAMETER This is NULL. | |
| The StationIp address provided in DnsConfigData is not a valid unicast. | |
| DnsServerList is NULL while DnsServerListCount is not equal to Zero. | |
| DnsServerListCount is Zero while DnsServerListCount is not equal to NULL. | |
| @retval EFI_DEVICE_ERROR An unexpected system or network error occurred. The EFI DNSv4 Protocol instance is not configured. | |
| **/ | |
| EFI_STATUS | |
| EFIAPI | |
| Dns4Configure ( | |
| IN EFI_DNS4_PROTOCOL *This, | |
| IN EFI_DNS4_CONFIG_DATA *DnsConfigData | |
| ) | |
| { | |
| EFI_STATUS Status; | |
| DNS_INSTANCE *Instance; | |
| EFI_TPL OldTpl; | |
| IP4_ADDR Ip; | |
| IP4_ADDR Netmask; | |
| UINT32 ServerListCount; | |
| EFI_IPv4_ADDRESS *ServerList; | |
| Status = EFI_SUCCESS; | |
| ServerList = NULL; | |
| if (This == NULL || | |
| (DnsConfigData != NULL && ((DnsConfigData->DnsServerListCount != 0 && DnsConfigData->DnsServerList == NULL) || | |
| (DnsConfigData->DnsServerListCount == 0 && DnsConfigData->DnsServerList != NULL)))) { | |
| return EFI_INVALID_PARAMETER; | |
| } | |
| if (DnsConfigData != NULL && DnsConfigData->Protocol != DNS_PROTOCOL_UDP) { | |
| return EFI_UNSUPPORTED; | |
| } | |
| OldTpl = gBS->RaiseTPL (TPL_CALLBACK); | |
| Instance = DNS_INSTANCE_FROM_THIS_PROTOCOL4 (This); | |
| if (DnsConfigData == NULL) { | |
| ZeroMem (&Instance->SessionDnsServer, sizeof (EFI_IP_ADDRESS)); | |
| // | |
| // Reset the Instance if ConfigData is NULL | |
| // | |
| if (!NetMapIsEmpty(&Instance->Dns4TxTokens)) { | |
| Dns4InstanceCancelToken(Instance, NULL); | |
| } | |
| Instance->MaxRetry = 0; | |
| if (Instance->UdpIo != NULL){ | |
| UdpIoCleanIo (Instance->UdpIo); | |
| } | |
| if (Instance->Dns4CfgData.DnsServerList != NULL) { | |
| FreePool (Instance->Dns4CfgData.DnsServerList); | |
| } | |
| ZeroMem (&Instance->Dns4CfgData, sizeof (EFI_DNS4_CONFIG_DATA)); | |
| Instance->State = DNS_STATE_UNCONFIGED; | |
| } else { | |
| // | |
| // Configure the parameters for new operation. | |
| // | |
| CopyMem (&Ip, &DnsConfigData->StationIp, sizeof (IP4_ADDR)); | |
| CopyMem (&Netmask, &DnsConfigData->SubnetMask, sizeof (IP4_ADDR)); | |
| Ip = NTOHL (Ip); | |
| Netmask = NTOHL (Netmask); | |
| if (!DnsConfigData->UseDefaultSetting && | |
| ((!IP4_IS_VALID_NETMASK (Netmask) || !NetIp4IsUnicast (Ip, Netmask)))) { | |
| Status = EFI_INVALID_PARAMETER; | |
| goto ON_EXIT; | |
| } | |
| Status = Dns4CopyConfigure (&Instance->Dns4CfgData, DnsConfigData); | |
| if (EFI_ERROR (Status)) { | |
| goto ON_EXIT; | |
| } | |
| if (DnsConfigData->DnsServerListCount == 0 || DnsConfigData->DnsServerList == NULL) { | |
| gBS->RestoreTPL (OldTpl); | |
| // | |
| // The DNS instance will retrieve DNS server from DHCP Server | |
| // | |
| Status = GetDns4ServerFromDhcp4 ( | |
| Instance, | |
| &ServerListCount, | |
| &ServerList | |
| ); | |
| if (EFI_ERROR (Status)) { | |
| return Status; | |
| } | |
| ASSERT(ServerList != NULL); | |
| OldTpl = gBS->RaiseTPL (TPL_CALLBACK); | |
| CopyMem (&Instance->SessionDnsServer.v4, &ServerList[0], sizeof (EFI_IPv4_ADDRESS)); | |
| } else { | |
| CopyMem (&Instance->SessionDnsServer.v4, &DnsConfigData->DnsServerList[0], sizeof (EFI_IPv4_ADDRESS)); | |
| } | |
| // | |
| // Config UDP | |
| // | |
| Status = Dns4ConfigUdp (Instance, Instance->UdpIo); | |
| if (EFI_ERROR (Status)) { | |
| if (Instance->Dns4CfgData.DnsServerList != NULL) { | |
| FreePool (Instance->Dns4CfgData.DnsServerList); | |
| } | |
| goto ON_EXIT; | |
| } | |
| // | |
| // Add configured DNS server used by this instance to ServerList. | |
| // | |
| Status = AddDns4ServerIp (&mDriverData->Dns4ServerList, Instance->SessionDnsServer.v4); | |
| if (EFI_ERROR (Status)) { | |
| if (Instance->Dns4CfgData.DnsServerList != NULL) { | |
| FreePool (Instance->Dns4CfgData.DnsServerList); | |
| } | |
| goto ON_EXIT; | |
| } | |
| Instance->State = DNS_STATE_CONFIGED; | |
| } | |
| ON_EXIT: | |
| gBS->RestoreTPL (OldTpl); | |
| return Status; | |
| } | |
| /** | |
| The function is used to translate the host name to host IP address. | |
| A type A query is used to get the one or more IP addresses for this host. | |
| @param[in] This Pointer to EFI_DNS4_PROTOCOL instance. | |
| @param[in] HostName Pointer to caller-supplied buffer containing Host name to be translated. | |
| This buffer contains 16 bit characters but these are translated to ASCII for use with | |
| DNSv4 server and there is no requirement for driver to support non-ASCII Unicode characters. | |
| @param[in] Token Pointer to the caller-allocated completion token to return at the completion of the process to translate host name to host address. | |
| @retval EFI_SUCCESS The operation completed successfully. | |
| @retval EFI_OUT_OF_RESOURCES Failed to allocate needed resources. | |
| @retval EFI_INVALID_PARAMETER This is NULL. | |
| Token is NULL. | |
| Token.Event is.NULL | |
| HostName is NULL | |
| @retval EFI_NO_MAPPING There's no source address is available for use. | |
| @retval EFI_NOT_STARTED This instance has not been started. | |
| **/ | |
| EFI_STATUS | |
| EFIAPI | |
| Dns4HostNameToIp ( | |
| IN EFI_DNS4_PROTOCOL *This, | |
| IN CHAR16 *HostName, | |
| IN EFI_DNS4_COMPLETION_TOKEN *Token | |
| ) | |
| { | |
| EFI_STATUS Status; | |
| DNS_INSTANCE *Instance; | |
| EFI_DNS4_CONFIG_DATA *ConfigData; | |
| UINTN Index; | |
| DNS4_CACHE *Item; | |
| LIST_ENTRY *Entry; | |
| LIST_ENTRY *Next; | |
| CHAR8 *QueryName; | |
| DNS4_TOKEN_ENTRY *TokenEntry; | |
| NET_BUF *Packet; | |
| EFI_TPL OldTpl; | |
| Status = EFI_SUCCESS; | |
| Item = NULL; | |
| QueryName = NULL; | |
| TokenEntry = NULL; | |
| Packet = NULL; | |
| // | |
| // Validate the parameters | |
| // | |
| if ((This == NULL) || (HostName == NULL) || Token == NULL) { | |
| return EFI_INVALID_PARAMETER; | |
| } | |
| OldTpl = gBS->RaiseTPL (TPL_CALLBACK); | |
| Instance = DNS_INSTANCE_FROM_THIS_PROTOCOL4 (This); | |
| ConfigData = &(Instance->Dns4CfgData); | |
| Instance->MaxRetry = ConfigData->RetryCount; | |
| Token->Status = EFI_NOT_READY; | |
| Token->RetryCount = 0; | |
| Token->RetryInterval = ConfigData->RetryInterval; | |
| if (Instance->State != DNS_STATE_CONFIGED) { | |
| Status = EFI_NOT_STARTED; | |
| goto ON_EXIT; | |
| } | |
| // | |
| // Check the MaxRetry and RetryInterval values. | |
| // | |
| if (Instance->MaxRetry == 0) { | |
| Instance->MaxRetry = DNS_DEFAULT_RETRY; | |
| } | |
| if (Token->RetryInterval < DNS_DEFAULT_TIMEOUT) { | |
| Token->RetryInterval = DNS_DEFAULT_TIMEOUT; | |
| } | |
| // | |
| // Check cache | |
| // | |
| if (ConfigData->EnableDnsCache) { | |
| Index = 0; | |
| NET_LIST_FOR_EACH_SAFE (Entry, Next, &mDriverData->Dns4CacheList) { | |
| Item = NET_LIST_USER_STRUCT (Entry, DNS4_CACHE, AllCacheLink); | |
| if (StrCmp (HostName, Item->DnsCache.HostName) == 0) { | |
| Index++; | |
| } | |
| } | |
| if (Index != 0) { | |
| Token->RspData.H2AData = AllocatePool (sizeof (DNS_HOST_TO_ADDR_DATA)); | |
| if (Token->RspData.H2AData == NULL) { | |
| Status = EFI_OUT_OF_RESOURCES; | |
| goto ON_EXIT; | |
| } | |
| Token->RspData.H2AData->IpCount = (UINT32)Index; | |
| Token->RspData.H2AData->IpList = AllocatePool (sizeof (EFI_IPv4_ADDRESS) * Index); | |
| if (Token->RspData.H2AData->IpList == NULL) { | |
| if (Token->RspData.H2AData != NULL) { | |
| FreePool (Token->RspData.H2AData); | |
| } | |
| Status = EFI_OUT_OF_RESOURCES; | |
| goto ON_EXIT; | |
| } | |
| Index = 0; | |
| NET_LIST_FOR_EACH_SAFE (Entry, Next, &mDriverData->Dns4CacheList) { | |
| Item = NET_LIST_USER_STRUCT (Entry, DNS4_CACHE, AllCacheLink); | |
| if ((UINT32)Index < Token->RspData.H2AData->IpCount && StrCmp (HostName, Item->DnsCache.HostName) == 0) { | |
| CopyMem ((Token->RspData.H2AData->IpList) + Index, Item->DnsCache.IpAddress, sizeof (EFI_IPv4_ADDRESS)); | |
| Index++; | |
| } | |
| } | |
| Token->Status = EFI_SUCCESS; | |
| if (Token->Event != NULL) { | |
| gBS->SignalEvent (Token->Event); | |
| DispatchDpc (); | |
| } | |
| Status = Token->Status; | |
| goto ON_EXIT; | |
| } | |
| } | |
| // | |
| // Construct DNS TokenEntry. | |
| // | |
| TokenEntry = AllocateZeroPool (sizeof(DNS4_TOKEN_ENTRY)); | |
| if (TokenEntry == NULL) { | |
| Status = EFI_OUT_OF_RESOURCES; | |
| goto ON_EXIT; | |
| } | |
| TokenEntry->PacketToLive = Token->RetryInterval; | |
| TokenEntry->QueryHostName = HostName; | |
| TokenEntry->Token = Token; | |
| // | |
| // Construct QName. | |
| // | |
| QueryName = DnsFillinQNameForQueryIp (TokenEntry->QueryHostName); | |
| if (QueryName == NULL) { | |
| Status = EFI_OUT_OF_RESOURCES; | |
| goto ON_EXIT; | |
| } | |
| // | |
| // Construct DNS Query Packet. | |
| // | |
| Status = ConstructDNSQuery (Instance, QueryName, DNS_TYPE_A, DNS_CLASS_INET, &Packet); | |
| if (EFI_ERROR (Status)) { | |
| if (TokenEntry != NULL) { | |
| FreePool (TokenEntry); | |
| } | |
| goto ON_EXIT; | |
| } | |
| ASSERT (Packet != NULL); | |
| // | |
| // Save the token into the Dns4TxTokens map. | |
| // | |
| Status = NetMapInsertTail (&Instance->Dns4TxTokens, TokenEntry, Packet); | |
| if (EFI_ERROR (Status)) { | |
| if (TokenEntry != NULL) { | |
| FreePool (TokenEntry); | |
| } | |
| NetbufFree (Packet); | |
| goto ON_EXIT; | |
| } | |
| // | |
| // Dns Query Ip | |
| // | |
| Status = DoDnsQuery (Instance, Packet); | |
| if (EFI_ERROR (Status)) { | |
| if (TokenEntry != NULL) { | |
| FreePool (TokenEntry); | |
| } | |
| NetbufFree (Packet); | |
| } | |
| ON_EXIT: | |
| if (QueryName != NULL) { | |
| FreePool (QueryName); | |
| } | |
| gBS->RestoreTPL (OldTpl); | |
| return Status; | |
| } | |
| /** | |
| The function is used to translate the host address to host name. | |
| A type PTR query is used to get the primary name of the host. | |
| @param[in] This Pointer to EFI_DNS4_PROTOCOL instance. | |
| @param[in] IpAddress IP address. | |
| @param[in] Token Pointer to the caller-allocated completion used token to translate host address to host name. | |
| @retval EFI_SUCCESS The operation completed successfully. | |
| @retval EFI_OUT_OF_RESOURCES Failed to allocate needed resources. | |
| @retval EFI_INVALID_PARAMETER This is NULL. | |
| Token is NULL. | |
| Token.Event is NULL. | |
| IpAddress is not valid IP address. | |
| @retval EFI_NO_MAPPING There's no source address is available for use. | |
| @retval EFI_NOT_STARTED This instance has not been started. | |
| @retval EFI_UNSUPPORTED This function is not supported. | |
| **/ | |
| EFI_STATUS | |
| EFIAPI | |
| Dns4IpToHostName ( | |
| IN EFI_DNS4_PROTOCOL *This, | |
| IN EFI_IPv4_ADDRESS IpAddress, | |
| IN EFI_DNS4_COMPLETION_TOKEN *Token | |
| ) | |
| { | |
| return EFI_UNSUPPORTED; | |
| } | |
| /** | |
| This function retrieves arbitrary information from the DNS. | |
| The caller supplies a QNAME, QTYPE, and QCLASS, and all of the matching RRs are returned. | |
| All RR content (e.g., Ttl) was returned. | |
| The caller need parse the returned RR to get required information. This function is optional. | |
| @param[in] This Pointer to EFI_DNS4_PROTOCOL instance. | |
| @param[in] QName Pointer to Query Name. | |
| @param[in] QType Query Type. | |
| @param[in] QClass Query Name. | |
| @param[in] Token Point to the caller-allocated completion token to retrieve arbitrary information. | |
| @retval EFI_SUCCESS The operation completed successfully. | |
| @retval EFI_OUT_OF_RESOURCES Failed to allocate needed resources. | |
| @retval EFI_INVALID_PARAMETER This is NULL. | |
| Token is NULL. | |
| Token.Event is NULL. | |
| QName is NULL. | |
| @retval EFI_NO_MAPPING There's no source address is available for use. | |
| @retval EFI_ALREADY_STARTED This Token is being used in another DNS session. | |
| @retval EFI_UNSUPPORTED This function is not supported. Or the requested QType is not supported | |
| **/ | |
| EFI_STATUS | |
| EFIAPI | |
| Dns4GeneralLookUp ( | |
| IN EFI_DNS4_PROTOCOL *This, | |
| IN CHAR8 *QName, | |
| IN UINT16 QType, | |
| IN UINT16 QClass, | |
| IN EFI_DNS4_COMPLETION_TOKEN *Token | |
| ) | |
| { | |
| EFI_STATUS Status; | |
| DNS_INSTANCE *Instance; | |
| EFI_DNS4_CONFIG_DATA *ConfigData; | |
| DNS4_TOKEN_ENTRY *TokenEntry; | |
| NET_BUF *Packet; | |
| EFI_TPL OldTpl; | |
| Status = EFI_SUCCESS; | |
| TokenEntry = NULL; | |
| Packet = NULL; | |
| // | |
| // Validate the parameters | |
| // | |
| if ((This == NULL) || (QName == NULL) || Token == NULL) { | |
| return EFI_INVALID_PARAMETER; | |
| } | |
| OldTpl = gBS->RaiseTPL (TPL_CALLBACK); | |
| Instance = DNS_INSTANCE_FROM_THIS_PROTOCOL4 (This); | |
| ConfigData = &(Instance->Dns4CfgData); | |
| Instance->MaxRetry = ConfigData->RetryCount; | |
| Token->Status = EFI_NOT_READY; | |
| Token->RetryCount = 0; | |
| Token->RetryInterval = ConfigData->RetryInterval; | |
| if (Instance->State != DNS_STATE_CONFIGED) { | |
| Status = EFI_NOT_STARTED; | |
| goto ON_EXIT; | |
| } | |
| // | |
| // Check the MaxRetry and RetryInterval values. | |
| // | |
| if (Instance->MaxRetry == 0) { | |
| Instance->MaxRetry = DNS_DEFAULT_RETRY; | |
| } | |
| if (Token->RetryInterval < DNS_DEFAULT_TIMEOUT) { | |
| Token->RetryInterval = DNS_DEFAULT_TIMEOUT; | |
| } | |
| // | |
| // Construct DNS TokenEntry. | |
| // | |
| TokenEntry = AllocateZeroPool (sizeof(DNS4_TOKEN_ENTRY)); | |
| if (TokenEntry == NULL) { | |
| Status = EFI_OUT_OF_RESOURCES; | |
| goto ON_EXIT; | |
| } | |
| TokenEntry->PacketToLive = Token->RetryInterval; | |
| TokenEntry->GeneralLookUp = TRUE; | |
| TokenEntry->Token = Token; | |
| // | |
| // Construct DNS Query Packet. | |
| // | |
| Status = ConstructDNSQuery (Instance, QName, QType, QClass, &Packet); | |
| if (EFI_ERROR (Status)) { | |
| if (TokenEntry != NULL) { | |
| FreePool (TokenEntry); | |
| } | |
| goto ON_EXIT; | |
| } | |
| ASSERT (Packet != NULL); | |
| // | |
| // Save the token into the Dns4TxTokens map. | |
| // | |
| Status = NetMapInsertTail (&Instance->Dns4TxTokens, TokenEntry, Packet); | |
| if (EFI_ERROR (Status)) { | |
| if (TokenEntry != NULL) { | |
| FreePool (TokenEntry); | |
| } | |
| NetbufFree (Packet); | |
| goto ON_EXIT; | |
| } | |
| // | |
| // Dns Query Ip | |
| // | |
| Status = DoDnsQuery (Instance, Packet); | |
| if (EFI_ERROR (Status)) { | |
| if (TokenEntry != NULL) { | |
| FreePool (TokenEntry); | |
| } | |
| NetbufFree (Packet); | |
| } | |
| ON_EXIT: | |
| gBS->RestoreTPL (OldTpl); | |
| return Status; | |
| } | |
| /** | |
| This function is used to add/delete/modify DNS cache entry. | |
| DNS cache can be normally dynamically updated after the DNS resolve succeeds. | |
| This function provided capability to manually add/delete/modify the DNS cache. | |
| @param[in] This Pointer to EFI_DNS4_PROTOCOL instance. | |
| @param[in] DeleteFlag If FALSE, this function is to add one entry to the DNS Cache. | |
| If TRUE, this function will delete matching DNS Cache entry. | |
| @param[in] Override If TRUE, the matching DNS cache entry will be overwritten with the supplied parameter. | |
| If FALSE, EFI_ACCESS_DENIED will be returned if the entry to be added is already exists. | |
| @param[in] DnsCacheEntry Pointer to DNS Cache entry. | |
| @retval EFI_SUCCESS The operation completed successfully. | |
| @retval EFI_INVALID_PARAMETER This is NULL. | |
| DnsCacheEntry.HostName is NULL. | |
| DnsCacheEntry.IpAddress is NULL. | |
| DnsCacheEntry.Timeout is zero. | |
| @retval EFI_ACCESS_DENIED The DNS cache entry already exists and Override is not TRUE. | |
| **/ | |
| EFI_STATUS | |
| EFIAPI | |
| Dns4UpdateDnsCache ( | |
| IN EFI_DNS4_PROTOCOL *This, | |
| IN BOOLEAN DeleteFlag, | |
| IN BOOLEAN Override, | |
| IN EFI_DNS4_CACHE_ENTRY DnsCacheEntry | |
| ) | |
| { | |
| EFI_STATUS Status; | |
| EFI_TPL OldTpl; | |
| Status = EFI_SUCCESS; | |
| if (DnsCacheEntry.HostName == NULL || DnsCacheEntry.IpAddress == NULL || DnsCacheEntry.Timeout == 0) { | |
| return EFI_INVALID_PARAMETER; | |
| } | |
| OldTpl = gBS->RaiseTPL (TPL_CALLBACK); | |
| // | |
| // Update Dns4Cache here. | |
| // | |
| Status = UpdateDns4Cache (&mDriverData->Dns4CacheList, DeleteFlag, Override, DnsCacheEntry); | |
| gBS->RestoreTPL (OldTpl); | |
| return Status; | |
| } | |
| /** | |
| This function can be used by network drivers and applications to increase the rate that data packets are moved between | |
| the communications device and the transmit and receive queues. In some systems, the periodic timer event in the managed | |
| network driver may not poll the underlying communications device fast enough to transmit and/or receive all data packets | |
| without missing incoming packets or dropping outgoing packets. | |
| @param[in] This Pointer to EFI_DNS4_PROTOCOL instance. | |
| @retval EFI_SUCCESS Incoming or outgoing data was processed. | |
| @retval EFI_INVALID_PARAMETER This is NULL. | |
| @retval EFI_NOT_STARTED This EFI DNS Protocol instance has not been started. | |
| @retval EFI_DEVICE_ERROR An unexpected system or network error occurred. | |
| @retval EFI_TIMEOUT Data was dropped out of the transmit and/or receive queue. | |
| Consider increasing the polling rate. | |
| **/ | |
| EFI_STATUS | |
| EFIAPI | |
| Dns4Poll ( | |
| IN EFI_DNS4_PROTOCOL *This | |
| ) | |
| { | |
| DNS_INSTANCE *Instance; | |
| EFI_UDP4_PROTOCOL *Udp; | |
| if (This == NULL) { | |
| return EFI_INVALID_PARAMETER; | |
| } | |
| Instance = DNS_INSTANCE_FROM_THIS_PROTOCOL4 (This); | |
| if (Instance->State == DNS_STATE_UNCONFIGED) { | |
| return EFI_NOT_STARTED; | |
| } else if (Instance->State == DNS_STATE_DESTROY) { | |
| return EFI_DEVICE_ERROR; | |
| } | |
| Udp = Instance->UdpIo->Protocol.Udp4; | |
| return Udp->Poll (Udp); | |
| } | |
| /** | |
| This function is used to abort a pending resolution request. | |
| After calling this function, Token.Status will be set to EFI_ABORTED and then Token. | |
| @param[in] This Pointer to EFI_DNS4_PROTOCOL instance. | |
| @param[in] Token Pointer to a token that has been issued by EFI_DNS4_PROTOCOL.HostNameToIp(), | |
| EFI_DNS4_PROTOCOL.IpToHostName() or EFI_DNS4_PROTOCOL.GeneralLookup(). | |
| If NULL, all pending tokens are aborted. | |
| @retval EFI_SUCCESS Incoming or outgoing data was processed. | |
| @retval EFI_INVALID_PARAMETER This is NULL. | |
| @retval EFI_NOT_STARTED This EFI DNS Protocol instance has not been started. | |
| @retval EFI_NOT_FOUND When Token is not NULL, and the asynchronous DNS operation was not found in the transmit queue. | |
| It was either completed or was not issued by HostNameToIp(), IpToHostName() or GeneralLookup(). | |
| **/ | |
| EFI_STATUS | |
| EFIAPI | |
| Dns4Cancel ( | |
| IN EFI_DNS4_PROTOCOL *This, | |
| IN EFI_DNS4_COMPLETION_TOKEN *Token | |
| ) | |
| { | |
| EFI_STATUS Status; | |
| DNS_INSTANCE *Instance; | |
| EFI_TPL OldTpl; | |
| if (This == NULL) { | |
| return EFI_INVALID_PARAMETER; | |
| } | |
| Instance = DNS_INSTANCE_FROM_THIS_PROTOCOL4 (This); | |
| if (Instance->State == DNS_STATE_UNCONFIGED) { | |
| return EFI_NOT_STARTED; | |
| } | |
| OldTpl = gBS->RaiseTPL (TPL_CALLBACK); | |
| // | |
| // Cancle the tokens specified by Token for this instance. | |
| // | |
| Status = Dns4InstanceCancelToken (Instance, Token); | |
| // | |
| // Dispatch the DPC queued by the NotifyFunction of the canceled token's events. | |
| // | |
| DispatchDpc (); | |
| gBS->RestoreTPL (OldTpl); | |
| return Status; | |
| } | |
| /** | |
| This function is used to retrieve DNS mode data for this DNS instance. | |
| @param[in] This Pointer to EFI_DNS6_PROTOCOL instance. | |
| @param[out] DnsModeData Pointer to the caller-allocated storage for the EFI_DNS6_MODE_DATA structure. | |
| @retval EFI_SUCCESS The operation completed successfully. | |
| @retval EFI_NOT_STARTED When DnsConfigData is queried, no configuration data is | |
| available because this instance has not been configured. | |
| @retval EFI_OUT_OF_RESOURCES Failed to allocate needed resources. | |
| @retval EFI_INVALID_PARAMETER This is NULL or DnsModeData is NULL. | |
| **/ | |
| EFI_STATUS | |
| EFIAPI | |
| Dns6GetModeData ( | |
| IN EFI_DNS6_PROTOCOL *This, | |
| OUT EFI_DNS6_MODE_DATA *DnsModeData | |
| ) | |
| { | |
| DNS_INSTANCE *Instance; | |
| EFI_TPL OldTpl; | |
| UINTN Index; | |
| LIST_ENTRY *Entry; | |
| LIST_ENTRY *Next; | |
| DNS6_SERVER_IP *ServerItem; | |
| EFI_IPv6_ADDRESS *ServerList; | |
| DNS6_CACHE *CacheItem; | |
| EFI_DNS6_CACHE_ENTRY *CacheList; | |
| EFI_STATUS Status; | |
| ServerItem = NULL; | |
| ServerList = NULL; | |
| CacheItem = NULL; | |
| CacheList = NULL; | |
| Status = EFI_SUCCESS; | |
| if ((This == NULL) || (DnsModeData == NULL)) { | |
| return EFI_INVALID_PARAMETER; | |
| } | |
| OldTpl = gBS->RaiseTPL (TPL_CALLBACK); | |
| Instance = DNS_INSTANCE_FROM_THIS_PROTOCOL6 (This); | |
| if (Instance->State == DNS_STATE_UNCONFIGED) { | |
| gBS->RestoreTPL (OldTpl); | |
| return EFI_NOT_STARTED; | |
| } | |
| ZeroMem (DnsModeData, sizeof (EFI_DNS6_MODE_DATA)); | |
| // | |
| // Get the current configuration data of this instance. | |
| // | |
| Status = Dns6CopyConfigure(&DnsModeData->DnsConfigData, &Instance->Dns6CfgData); | |
| if (EFI_ERROR (Status)) { | |
| gBS->RestoreTPL (OldTpl); | |
| return Status; | |
| } | |
| // | |
| // Get the DnsServerCount and DnsServerList | |
| // | |
| Index = 0; | |
| NET_LIST_FOR_EACH_SAFE (Entry, Next, &mDriverData->Dns6ServerList) { | |
| Index++; | |
| } | |
| DnsModeData->DnsServerCount = (UINT32) Index; | |
| ServerList = AllocatePool (sizeof(EFI_IPv6_ADDRESS) * DnsModeData->DnsServerCount); | |
| ASSERT (ServerList != NULL); | |
| Index = 0; | |
| NET_LIST_FOR_EACH_SAFE (Entry, Next, &mDriverData->Dns6ServerList) { | |
| ServerItem = NET_LIST_USER_STRUCT (Entry, DNS6_SERVER_IP, AllServerLink); | |
| CopyMem (ServerList + Index, &ServerItem->Dns6ServerIp, sizeof (EFI_IPv6_ADDRESS)); | |
| Index++; | |
| } | |
| DnsModeData->DnsServerList = ServerList; | |
| // | |
| // Get the DnsCacheCount and DnsCacheList | |
| // | |
| Index =0; | |
| NET_LIST_FOR_EACH_SAFE (Entry, Next, &mDriverData->Dns6CacheList) { | |
| Index++; | |
| } | |
| DnsModeData->DnsCacheCount = (UINT32) Index; | |
| CacheList = AllocatePool (sizeof(EFI_DNS6_CACHE_ENTRY) * DnsModeData->DnsCacheCount); | |
| ASSERT (CacheList != NULL); | |
| Index =0; | |
| NET_LIST_FOR_EACH_SAFE (Entry, Next, &mDriverData->Dns6CacheList) { | |
| CacheItem = NET_LIST_USER_STRUCT (Entry, DNS6_CACHE, AllCacheLink); | |
| CopyMem (CacheList + Index, &CacheItem->DnsCache, sizeof (EFI_DNS6_CACHE_ENTRY)); | |
| Index++; | |
| } | |
| DnsModeData->DnsCacheList = CacheList; | |
| gBS->RestoreTPL (OldTpl); | |
| return EFI_SUCCESS; | |
| } | |
| /** | |
| The function is used to set and change the configuration data for this EFI DNSv6 Protocol driver instance. | |
| Reset the DNS instance if DnsConfigData is NULL. | |
| @param[in] This Pointer to EFI_DNS6_PROTOCOL instance. | |
| @param[in] DnsConfigData Pointer to the configuration data structure. | |
| All associated storage to be allocated and released by caller. | |
| @retval EFI_SUCCESS The operation completed successfully. | |
| @retval EFI_UNSUPPORTED The designated protocol is not supported. | |
| @retval EFI_OUT_OF_RESOURCES Failed to allocate needed resources. | |
| @retval EFI_INVALID_PARAMETER This is NULL. | |
| The StationIp address provided in DnsConfigData is not a valid unicast. | |
| DnsServerList is NULL while DnsServerListCount is not equal to Zero. | |
| DnsServerListCount is Zero while DnsServerList is not equal to NULL. | |
| @retval EFI_DEVICE_ERROR An unexpected system or network error occurred. The EFI DNSv6 Protocol instance is not configured. | |
| **/ | |
| EFI_STATUS | |
| EFIAPI | |
| Dns6Configure ( | |
| IN EFI_DNS6_PROTOCOL *This, | |
| IN EFI_DNS6_CONFIG_DATA *DnsConfigData | |
| ) | |
| { | |
| EFI_STATUS Status; | |
| DNS_INSTANCE *Instance; | |
| EFI_TPL OldTpl; | |
| UINT32 ServerListCount; | |
| EFI_IPv6_ADDRESS *ServerList; | |
| Status = EFI_SUCCESS; | |
| ServerList = NULL; | |
| if (This == NULL || | |
| (DnsConfigData != NULL && ((DnsConfigData->DnsServerCount != 0 && DnsConfigData->DnsServerList == NULL) || | |
| (DnsConfigData->DnsServerCount == 0 && DnsConfigData->DnsServerList != NULL)))) { | |
| return EFI_INVALID_PARAMETER; | |
| } | |
| if (DnsConfigData != NULL && DnsConfigData->Protocol != DNS_PROTOCOL_UDP) { | |
| return EFI_UNSUPPORTED; | |
| } | |
| OldTpl = gBS->RaiseTPL (TPL_CALLBACK); | |
| Instance = DNS_INSTANCE_FROM_THIS_PROTOCOL6 (This); | |
| if (DnsConfigData == NULL) { | |
| ZeroMem (&Instance->SessionDnsServer, sizeof (EFI_IP_ADDRESS)); | |
| // | |
| // Reset the Instance if ConfigData is NULL | |
| // | |
| if (!NetMapIsEmpty(&Instance->Dns6TxTokens)) { | |
| Dns6InstanceCancelToken(Instance, NULL); | |
| } | |
| Instance->MaxRetry = 0; | |
| if (Instance->UdpIo != NULL){ | |
| UdpIoCleanIo (Instance->UdpIo); | |
| } | |
| if (Instance->Dns6CfgData.DnsServerList != NULL) { | |
| FreePool (Instance->Dns6CfgData.DnsServerList); | |
| } | |
| ZeroMem (&Instance->Dns6CfgData, sizeof (EFI_DNS6_CONFIG_DATA)); | |
| Instance->State = DNS_STATE_UNCONFIGED; | |
| } else { | |
| // | |
| // Configure the parameters for new operation. | |
| // | |
| if (!NetIp6IsUnspecifiedAddr (&DnsConfigData->StationIp) && !NetIp6IsValidUnicast (&DnsConfigData->StationIp)) { | |
| Status = EFI_INVALID_PARAMETER; | |
| goto ON_EXIT; | |
| } | |
| Status = Dns6CopyConfigure (&Instance->Dns6CfgData, DnsConfigData); | |
| if (EFI_ERROR (Status)) { | |
| goto ON_EXIT; | |
| } | |
| if (DnsConfigData->DnsServerCount == 0 || DnsConfigData->DnsServerList == NULL) { | |
| gBS->RestoreTPL (OldTpl); | |
| // | |
| //The DNS instance will retrieve DNS server from DHCP Server. | |
| // | |
| Status = GetDns6ServerFromDhcp6 ( | |
| Instance->Service->ImageHandle, | |
| Instance->Service->ControllerHandle, | |
| &ServerListCount, | |
| &ServerList | |
| ); | |
| if (EFI_ERROR (Status)) { | |
| goto ON_EXIT; | |
| } | |
| ASSERT(ServerList != NULL); | |
| OldTpl = gBS->RaiseTPL (TPL_CALLBACK); | |
| CopyMem (&Instance->SessionDnsServer.v6, &ServerList[0], sizeof (EFI_IPv6_ADDRESS)); | |
| } else { | |
| CopyMem (&Instance->SessionDnsServer.v6, &DnsConfigData->DnsServerList[0], sizeof (EFI_IPv6_ADDRESS)); | |
| } | |
| // | |
| // Config UDP | |
| // | |
| Status = Dns6ConfigUdp (Instance, Instance->UdpIo); | |
| if (EFI_ERROR (Status)) { | |
| if (Instance->Dns6CfgData.DnsServerList != NULL) { | |
| FreePool (Instance->Dns6CfgData.DnsServerList); | |
| } | |
| goto ON_EXIT; | |
| } | |
| // | |
| // Add configured DNS server used by this instance to ServerList. | |
| // | |
| Status = AddDns6ServerIp (&mDriverData->Dns6ServerList, Instance->SessionDnsServer.v6); | |
| if (EFI_ERROR (Status)) { | |
| if (Instance->Dns6CfgData.DnsServerList != NULL) { | |
| FreePool (Instance->Dns6CfgData.DnsServerList); | |
| } | |
| goto ON_EXIT; | |
| } | |
| Instance->State = DNS_STATE_CONFIGED; | |
| } | |
| ON_EXIT: | |
| gBS->RestoreTPL (OldTpl); | |
| return Status; | |
| } | |
| /** | |
| The function is used to translate the host name to host IP address. | |
| A type AAAA query is used to get the one or more IPv6 addresses for this host. | |
| @param[in] This Pointer to EFI_DNS6_PROTOCOL instance. | |
| @param[in] HostName Pointer to caller-supplied buffer containing Host name to be translated. | |
| This buffer contains 16 bit characters but these are translated to ASCII for use with | |
| DNSv4 server and there is no requirement for driver to support non-ASCII Unicode characters. | |
| @param[in] Token Pointer to the caller-allocated completion token to return at the completion of the process to translate host name to host address. | |
| @retval EFI_SUCCESS The operation completed successfully. | |
| @retval EFI_OUT_OF_RESOURCES Failed to allocate needed resources. | |
| @retval EFI_INVALID_PARAMETER This is NULL. | |
| Token is NULL. | |
| Token.Event is.NULL | |
| HostName is NULL | |
| @retval EFI_NO_MAPPING There's no source address is available for use. | |
| @retval EFI_NOT_STARTED This instance has not been started. | |
| **/ | |
| EFI_STATUS | |
| EFIAPI | |
| Dns6HostNameToIp ( | |
| IN EFI_DNS6_PROTOCOL *This, | |
| IN CHAR16 *HostName, | |
| IN EFI_DNS6_COMPLETION_TOKEN *Token | |
| ) | |
| { | |
| EFI_STATUS Status; | |
| DNS_INSTANCE *Instance; | |
| EFI_DNS6_CONFIG_DATA *ConfigData; | |
| UINTN Index; | |
| DNS6_CACHE *Item; | |
| LIST_ENTRY *Entry; | |
| LIST_ENTRY *Next; | |
| CHAR8 *QueryName; | |
| DNS6_TOKEN_ENTRY *TokenEntry; | |
| NET_BUF *Packet; | |
| EFI_TPL OldTpl; | |
| Status = EFI_SUCCESS; | |
| Item = NULL; | |
| QueryName = NULL; | |
| TokenEntry = NULL; | |
| Packet = NULL; | |
| // | |
| // Validate the parameters | |
| // | |
| if ((This == NULL) || (HostName == NULL) || Token == NULL) { | |
| return EFI_INVALID_PARAMETER; | |
| } | |
| OldTpl = gBS->RaiseTPL (TPL_CALLBACK); | |
| Instance = DNS_INSTANCE_FROM_THIS_PROTOCOL6 (This); | |
| ConfigData = &(Instance->Dns6CfgData); | |
| Instance->MaxRetry = ConfigData->RetryCount; | |
| Token->Status = EFI_NOT_READY; | |
| Token->RetryCount = 0; | |
| Token->RetryInterval = ConfigData->RetryInterval; | |
| if (Instance->State != DNS_STATE_CONFIGED) { | |
| Status = EFI_NOT_STARTED; | |
| goto ON_EXIT; | |
| } | |
| // | |
| // Check the MaxRetry and RetryInterval values. | |
| // | |
| if (Instance->MaxRetry == 0) { | |
| Instance->MaxRetry = DNS_DEFAULT_RETRY; | |
| } | |
| if (Token->RetryInterval < DNS_DEFAULT_TIMEOUT) { | |
| Token->RetryInterval = DNS_DEFAULT_TIMEOUT; | |
| } | |
| // | |
| // Check cache | |
| // | |
| if (ConfigData->EnableDnsCache) { | |
| Index = 0; | |
| NET_LIST_FOR_EACH_SAFE (Entry, Next, &mDriverData->Dns6CacheList) { | |
| Item = NET_LIST_USER_STRUCT (Entry, DNS6_CACHE, AllCacheLink); | |
| if (StrCmp (HostName, Item->DnsCache.HostName) == 0) { | |
| Index++; | |
| } | |
| } | |
| if (Index != 0) { | |
| Token->RspData.H2AData = AllocatePool (sizeof (DNS6_HOST_TO_ADDR_DATA)); | |
| if (Token->RspData.H2AData == NULL) { | |
| Status = EFI_OUT_OF_RESOURCES; | |
| goto ON_EXIT; | |
| } | |
| Token->RspData.H2AData->IpCount = (UINT32)Index; | |
| Token->RspData.H2AData->IpList = AllocatePool (sizeof (EFI_IPv6_ADDRESS) * Index); | |
| if (Token->RspData.H2AData->IpList == NULL) { | |
| if (Token->RspData.H2AData != NULL) { | |
| FreePool (Token->RspData.H2AData); | |
| } | |
| Status = EFI_OUT_OF_RESOURCES; | |
| goto ON_EXIT; | |
| } | |
| Index = 0; | |
| NET_LIST_FOR_EACH_SAFE (Entry, Next, &mDriverData->Dns6CacheList) { | |
| Item = NET_LIST_USER_STRUCT (Entry, DNS6_CACHE, AllCacheLink); | |
| if ((UINT32)Index < Token->RspData.H2AData->IpCount && StrCmp (HostName, Item->DnsCache.HostName) == 0) { | |
| CopyMem ((Token->RspData.H2AData->IpList) + Index, Item->DnsCache.IpAddress, sizeof (EFI_IPv6_ADDRESS)); | |
| Index++; | |
| } | |
| } | |
| Token->Status = EFI_SUCCESS; | |
| if (Token->Event != NULL) { | |
| gBS->SignalEvent (Token->Event); | |
| DispatchDpc (); | |
| } | |
| Status = Token->Status; | |
| goto ON_EXIT; | |
| } | |
| } | |
| // | |
| // Construct DNS TokenEntry. | |
| // | |
| TokenEntry = AllocateZeroPool (sizeof (DNS6_TOKEN_ENTRY)); | |
| if (TokenEntry == NULL) { | |
| Status = EFI_OUT_OF_RESOURCES; | |
| goto ON_EXIT; | |
| } | |
| TokenEntry->PacketToLive = Token->RetryInterval; | |
| TokenEntry->QueryHostName = HostName; | |
| TokenEntry->Token = Token; | |
| // | |
| // Construct QName. | |
| // | |
| QueryName = DnsFillinQNameForQueryIp (TokenEntry->QueryHostName); | |
| if (QueryName == NULL) { | |
| Status = EFI_OUT_OF_RESOURCES; | |
| goto ON_EXIT; | |
| } | |
| // | |
| // Construct DNS Query Packet. | |
| // | |
| Status = ConstructDNSQuery (Instance, QueryName, DNS_TYPE_AAAA, DNS_CLASS_INET, &Packet); | |
| if (EFI_ERROR (Status)) { | |
| if (TokenEntry != NULL) { | |
| FreePool (TokenEntry); | |
| } | |
| goto ON_EXIT; | |
| } | |
| ASSERT (Packet != NULL); | |
| // | |
| // Save the token into the Dns6TxTokens map. | |
| // | |
| Status = NetMapInsertTail (&Instance->Dns6TxTokens, TokenEntry, Packet); | |
| if (EFI_ERROR (Status)) { | |
| if (TokenEntry != NULL) { | |
| FreePool (TokenEntry); | |
| } | |
| NetbufFree (Packet); | |
| goto ON_EXIT; | |
| } | |
| // | |
| // Dns Query Ip | |
| // | |
| Status = DoDnsQuery (Instance, Packet); | |
| if (EFI_ERROR (Status)) { | |
| if (TokenEntry != NULL) { | |
| FreePool (TokenEntry); | |
| } | |
| NetbufFree (Packet); | |
| } | |
| ON_EXIT: | |
| if (QueryName != NULL) { | |
| FreePool (QueryName); | |
| } | |
| gBS->RestoreTPL (OldTpl); | |
| return Status; | |
| } | |
| /** | |
| The function is used to translate the host address to host name. | |
| A type PTR query is used to get the primary name of the host. | |
| @param[in] This Pointer to EFI_DNS6_PROTOCOL instance. | |
| @param[in] IpAddress IP address. | |
| @param[in] Token Pointer to the caller-allocated completion used token to translate host address to host name. | |
| @retval EFI_SUCCESS The operation completed successfully. | |
| @retval EFI_OUT_OF_RESOURCES Failed to allocate needed resources. | |
| @retval EFI_INVALID_PARAMETER This is NULL. | |
| Token is NULL. | |
| Token.Event is NULL. | |
| IpAddress is not valid IP address. | |
| @retval EFI_NO_MAPPING There's no source address is available for use. | |
| @retval EFI_NOT_STARTED This instance has not been started. | |
| @retval EFI_UNSUPPORTED This function is not supported. | |
| **/ | |
| EFI_STATUS | |
| EFIAPI | |
| Dns6IpToHostName ( | |
| IN EFI_DNS6_PROTOCOL *This, | |
| IN EFI_IPv6_ADDRESS IpAddress, | |
| IN EFI_DNS6_COMPLETION_TOKEN *Token | |
| ) | |
| { | |
| return EFI_UNSUPPORTED; | |
| } | |
| /** | |
| This function retrieves arbitrary information from the DNS. | |
| The caller supplies a QNAME, QTYPE, and QCLASS, and all of the matching RRs are returned. | |
| All RR content (e.g., Ttl) was returned. | |
| The caller need parse the returned RR to get required information. This function is optional. | |
| @param[in] This Pointer to EFI_DNS6_PROTOCOL instance. | |
| @param[in] QName Pointer to Query Name. | |
| @param[in] QType Query Type. | |
| @param[in] QClass Query Name. | |
| @param[in] Token Point to the caller-allocated completion token to retrieve arbitrary information. | |
| @retval EFI_SUCCESS The operation completed successfully. | |
| @retval EFI_OUT_OF_RESOURCES Failed to allocate needed resources. | |
| @retval EFI_INVALID_PARAMETER This is NULL. | |
| Token is NULL. | |
| Token.Event is NULL. | |
| QName is NULL. | |
| @retval EFI_NO_MAPPING There's no source address is available for use. | |
| @retval EFI_NOT_STARTED This instance has not been started. | |
| @retval EFI_UNSUPPORTED This function is not supported. Or the requested QType is not supported | |
| **/ | |
| EFI_STATUS | |
| EFIAPI | |
| Dns6GeneralLookUp ( | |
| IN EFI_DNS6_PROTOCOL *This, | |
| IN CHAR8 *QName, | |
| IN UINT16 QType, | |
| IN UINT16 QClass, | |
| IN EFI_DNS6_COMPLETION_TOKEN *Token | |
| ) | |
| { | |
| EFI_STATUS Status; | |
| DNS_INSTANCE *Instance; | |
| EFI_DNS6_CONFIG_DATA *ConfigData; | |
| DNS6_TOKEN_ENTRY *TokenEntry; | |
| NET_BUF *Packet; | |
| EFI_TPL OldTpl; | |
| Status = EFI_SUCCESS; | |
| TokenEntry = NULL; | |
| Packet = NULL; | |
| // | |
| // Validate the parameters | |
| // | |
| if ((This == NULL) || (QName == NULL) || Token == NULL) { | |
| return EFI_INVALID_PARAMETER; | |
| } | |
| OldTpl = gBS->RaiseTPL (TPL_CALLBACK); | |
| Instance = DNS_INSTANCE_FROM_THIS_PROTOCOL6 (This); | |
| ConfigData = &(Instance->Dns6CfgData); | |
| Instance->MaxRetry = ConfigData->RetryCount; | |
| Token->Status = EFI_NOT_READY; | |
| Token->RetryCount = 0; | |
| Token->RetryInterval = ConfigData->RetryInterval; | |
| if (Instance->State != DNS_STATE_CONFIGED) { | |
| Status = EFI_NOT_STARTED; | |
| goto ON_EXIT; | |
| } | |
| // | |
| // Check the MaxRetry and RetryInterval values. | |
| // | |
| if (Instance->MaxRetry == 0) { | |
| Instance->MaxRetry = DNS_DEFAULT_RETRY; | |
| } | |
| if (Token->RetryInterval < DNS_DEFAULT_TIMEOUT) { | |
| Token->RetryInterval = DNS_DEFAULT_TIMEOUT; | |
| } | |
| // | |
| // Construct DNS TokenEntry. | |
| // | |
| TokenEntry = AllocateZeroPool (sizeof(DNS6_TOKEN_ENTRY)); | |
| if (TokenEntry == NULL) { | |
| Status = EFI_OUT_OF_RESOURCES; | |
| goto ON_EXIT; | |
| } | |
| TokenEntry->PacketToLive = Token->RetryInterval; | |
| TokenEntry->GeneralLookUp = TRUE; | |
| TokenEntry->Token = Token; | |
| // | |
| // Construct DNS Query Packet. | |
| // | |
| Status = ConstructDNSQuery (Instance, QName, QType, QClass, &Packet); | |
| if (EFI_ERROR (Status)) { | |
| if (TokenEntry != NULL) { | |
| FreePool (TokenEntry); | |
| } | |
| goto ON_EXIT; | |
| } | |
| ASSERT (Packet != NULL); | |
| // | |
| // Save the token into the Dns6TxTokens map. | |
| // | |
| Status = NetMapInsertTail (&Instance->Dns6TxTokens, TokenEntry, Packet); | |
| if (EFI_ERROR (Status)) { | |
| if (TokenEntry != NULL) { | |
| FreePool (TokenEntry); | |
| } | |
| NetbufFree (Packet); | |
| goto ON_EXIT; | |
| } | |
| // | |
| // Dns Query Ip | |
| // | |
| Status = DoDnsQuery (Instance, Packet); | |
| if (EFI_ERROR (Status)) { | |
| if (TokenEntry != NULL) { | |
| FreePool (TokenEntry); | |
| } | |
| NetbufFree (Packet); | |
| } | |
| ON_EXIT: | |
| gBS->RestoreTPL (OldTpl); | |
| return Status; | |
| } | |
| /** | |
| This function is used to add/delete/modify DNS cache entry. | |
| DNS cache can be normally dynamically updated after the DNS resolve succeeds. | |
| This function provided capability to manually add/delete/modify the DNS cache. | |
| @param[in] This Pointer to EFI_DNS6_PROTOCOL instance. | |
| @param[in] DeleteFlag If FALSE, this function is to add one entry to the DNS Cache. | |
| If TRUE, this function will delete matching DNS Cache entry. | |
| @param[in] Override If TRUE, the matching DNS cache entry will be overwritten with the supplied parameter. | |
| If FALSE, EFI_ACCESS_DENIED will be returned if the entry to be added is already exists. | |
| @param[in] DnsCacheEntry Pointer to DNS Cache entry. | |
| @retval EFI_SUCCESS The operation completed successfully. | |
| @retval EFI_INVALID_PARAMETER This is NULL. | |
| DnsCacheEntry.HostName is NULL. | |
| DnsCacheEntry.IpAddress is NULL. | |
| DnsCacheEntry.Timeout is zero. | |
| @retval EFI_ACCESS_DENIED The DNS cache entry already exists and Override is not TRUE. | |
| **/ | |
| EFI_STATUS | |
| EFIAPI | |
| Dns6UpdateDnsCache ( | |
| IN EFI_DNS6_PROTOCOL *This, | |
| IN BOOLEAN DeleteFlag, | |
| IN BOOLEAN Override, | |
| IN EFI_DNS6_CACHE_ENTRY DnsCacheEntry | |
| ) | |
| { | |
| EFI_STATUS Status; | |
| EFI_TPL OldTpl; | |
| Status = EFI_SUCCESS; | |
| if (DnsCacheEntry.HostName == NULL || DnsCacheEntry.IpAddress == NULL || DnsCacheEntry.Timeout == 0) { | |
| return EFI_INVALID_PARAMETER; | |
| } | |
| OldTpl = gBS->RaiseTPL (TPL_CALLBACK); | |
| // | |
| // Update Dns6Cache here. | |
| // | |
| Status = UpdateDns6Cache (&mDriverData->Dns6CacheList, DeleteFlag, Override, DnsCacheEntry); | |
| gBS->RestoreTPL (OldTpl); | |
| return Status; | |
| } | |
| /** | |
| This function can be used by network drivers and applications to increase the rate that data packets are moved between | |
| the communications device and the transmit and receive queues. In some systems, the periodic timer event in the managed | |
| network driver may not poll the underlying communications device fast enough to transmit and/or receive all data packets | |
| without missing incoming packets or dropping outgoing packets. | |
| @param[in] This Pointer to EFI_DNS6_PROTOCOL instance. | |
| @retval EFI_SUCCESS Incoming or outgoing data was processed. | |
| @retval EFI_INVALID_PARAMETER This is NULL. | |
| @retval EFI_NOT_STARTED This EFI DNS Protocol instance has not been started. | |
| @retval EFI_DEVICE_ERROR An unexpected system or network error occurred. | |
| @retval EFI_TIMEOUT Data was dropped out of the transmit and/or receive queue. | |
| Consider increasing the polling rate. | |
| **/ | |
| EFI_STATUS | |
| EFIAPI | |
| Dns6Poll ( | |
| IN EFI_DNS6_PROTOCOL *This | |
| ) | |
| { | |
| DNS_INSTANCE *Instance; | |
| EFI_UDP6_PROTOCOL *Udp; | |
| if (This == NULL) { | |
| return EFI_INVALID_PARAMETER; | |
| } | |
| Instance = DNS_INSTANCE_FROM_THIS_PROTOCOL6 (This); | |
| if (Instance->State == DNS_STATE_UNCONFIGED) { | |
| return EFI_NOT_STARTED; | |
| } else if (Instance->State == DNS_STATE_DESTROY) { | |
| return EFI_DEVICE_ERROR; | |
| } | |
| Udp = Instance->UdpIo->Protocol.Udp6; | |
| return Udp->Poll (Udp); | |
| } | |
| /** | |
| This function is used to abort a pending resolution request. | |
| After calling this function, Token.Status will be set to EFI_ABORTED and then Token. | |
| @param[in] This Pointer to EFI_DNS6_PROTOCOL instance. | |
| @param[in] Token Pointer to a token that has been issued by EFI_DNS6_PROTOCOL.HostNameToIp(), | |
| EFI_DNS6_PROTOCOL.IpToHostName() or EFI_DNS6_PROTOCOL.GeneralLookup(). | |
| If NULL, all pending tokens are aborted. | |
| @retval EFI_SUCCESS Incoming or outgoing data was processed. | |
| @retval EFI_INVALID_PARAMETER This is NULL. | |
| @retval EFI_NOT_STARTED This EFI DNS Protocol instance has not been started. | |
| @retval EFI_NOT_FOUND When Token is not NULL, and the asynchronous DNS operation was not found in the transmit queue. | |
| It was either completed or was not issued by HostNameToIp(), IpToHostName() or GeneralLookup(). | |
| **/ | |
| EFI_STATUS | |
| EFIAPI | |
| Dns6Cancel ( | |
| IN EFI_DNS6_PROTOCOL *This, | |
| IN EFI_DNS6_COMPLETION_TOKEN *Token | |
| ) | |
| { | |
| EFI_STATUS Status; | |
| DNS_INSTANCE *Instance; | |
| EFI_TPL OldTpl; | |
| if (This == NULL) { | |
| return EFI_INVALID_PARAMETER; | |
| } | |
| Instance = DNS_INSTANCE_FROM_THIS_PROTOCOL6 (This); | |
| if (Instance->State == DNS_STATE_UNCONFIGED) { | |
| return EFI_NOT_STARTED; | |
| } | |
| OldTpl = gBS->RaiseTPL (TPL_CALLBACK); | |
| // | |
| // Cancle the tokens specified by Token for this instance. | |
| // | |
| Status = Dns6InstanceCancelToken (Instance, Token); | |
| // | |
| // Dispatch the DPC queued by the NotifyFunction of the canceled token's events. | |
| // | |
| DispatchDpc (); | |
| gBS->RestoreTPL (OldTpl); | |
| return Status; | |
| } | |