+++ /dev/null
-/** @file\r
- Support for PxeBc dhcp functions.\r
-\r
-Copyright (c) 2013, Red Hat, Inc.\r
-Copyright (c) 2007 - 2018, Intel Corporation. All rights reserved.<BR>\r
-This program and the accompanying materials\r
-are licensed and made available under the terms and conditions of the BSD License\r
-which accompanies this distribution. The full text of the license may be found at\r
-http://opensource.org/licenses/bsd-license.php\r
-\r
-THE PROGRAM IS DISTRIBUTED UNDER THE BSD LICENSE ON AN "AS IS" BASIS,\r
-WITHOUT WARRANTIES OR REPRESENTATIONS OF ANY KIND, EITHER EXPRESS OR IMPLIED.\r
-\r
-**/\r
-\r
-\r
-#include "PxeBcImpl.h"\r
-\r
-//\r
-// This is a map from the interested DHCP4 option tags' index to the tag value.\r
-//\r
-UINT8 mInterestedDhcp4Tags[PXEBC_DHCP4_TAG_INDEX_MAX] = {\r
- DHCP4_TAG_BOOTFILE_LEN,\r
- DHCP4_TAG_VENDOR,\r
- DHCP4_TAG_OVERLOAD,\r
- DHCP4_TAG_MSG_TYPE,\r
- DHCP4_TAG_SERVER_ID,\r
- DHCP4_TAG_VENDOR_CLASS_ID,\r
- DHCP4_TAG_BOOTFILE\r
-};\r
-\r
-\r
-/**\r
- This function initialize the DHCP4 message instance.\r
-\r
- This function will pad each item of dhcp4 message packet.\r
-\r
- @param Seed Pointer to the message instance of the DHCP4 packet.\r
- @param Udp4 Pointer to the EFI_UDP4_PROTOCOL instance.\r
-\r
-**/\r
-VOID\r
-PxeBcInitSeedPacket (\r
- IN EFI_DHCP4_PACKET *Seed,\r
- IN EFI_UDP4_PROTOCOL *Udp4\r
- )\r
-{\r
- EFI_SIMPLE_NETWORK_MODE Mode;\r
- EFI_DHCP4_HEADER *Header;\r
-\r
- Udp4->GetModeData (Udp4, NULL, NULL, NULL, &Mode);\r
-\r
- Seed->Size = sizeof (EFI_DHCP4_PACKET);\r
- Seed->Length = sizeof (Seed->Dhcp4);\r
-\r
- Header = &Seed->Dhcp4.Header;\r
-\r
- ZeroMem (Header, sizeof (EFI_DHCP4_HEADER));\r
- Header->OpCode = PXEBC_DHCP4_OPCODE_REQUEST;\r
- Header->HwType = Mode.IfType;\r
- Header->HwAddrLen = (UINT8) Mode.HwAddressSize;\r
- CopyMem (Header->ClientHwAddr, &Mode.CurrentAddress, Header->HwAddrLen);\r
-\r
- Seed->Dhcp4.Magik = PXEBC_DHCP4_MAGIC;\r
- Seed->Dhcp4.Option[0] = DHCP4_TAG_EOP;\r
-}\r
-\r
-\r
-/**\r
- Copy the DCHP4 packet from srouce to destination.\r
-\r
- @param[in] Dst Pointer to the cache buffer for DHCPv4 packet.\r
- @param[in] Src Pointer to the DHCPv4 packet to be cached.\r
-\r
- @retval EFI_SUCCESS Packet is copied.\r
- @retval EFI_BUFFER_TOO_SMALL Cache buffer is not big enough to hold the packet.\r
-\r
-**/\r
-EFI_STATUS\r
-PxeBcCopyEfiDhcp4Packet (\r
- IN EFI_DHCP4_PACKET *Dst,\r
- IN EFI_DHCP4_PACKET *Src\r
- )\r
-{\r
- if (Dst->Size < Src->Length) {\r
- return EFI_BUFFER_TOO_SMALL;\r
- }\r
-\r
- CopyMem (&Dst->Dhcp4, &Src->Dhcp4, Src->Length);\r
- Dst->Length = Src->Length;\r
- return EFI_SUCCESS;\r
-}\r
-\r
-\r
-/**\r
- Copy the dhcp4 packet to the PxeBc private data and parse the dhcp4 packet.\r
-\r
- @param Private Pointer to PxeBc private data.\r
- @param OfferIndex Index of cached packets as complements of pxe mode data,\r
- the index is maximum offer number.\r
-\r
- @retval EFI_SUCCESS Cache and parse the packet successfully.\r
- @retval EFI_BUFFER_TOO_SMALL Cache buffer is not big enough to hold the packet.\r
-\r
-**/\r
-EFI_STATUS\r
-PxeBcCopyProxyOffer (\r
- IN PXEBC_PRIVATE_DATA *Private,\r
- IN UINT32 OfferIndex\r
- )\r
-{\r
- EFI_PXE_BASE_CODE_MODE *Mode;\r
- EFI_DHCP4_PACKET *Offer;\r
- EFI_STATUS Status;\r
-\r
- ASSERT (OfferIndex < Private->NumOffers);\r
- ASSERT (OfferIndex < PXEBC_MAX_OFFER_NUM);\r
-\r
- Mode = Private->PxeBc.Mode;\r
- Offer = &Private->Dhcp4Offers[OfferIndex].Packet.Offer;\r
-\r
- Status = PxeBcCopyEfiDhcp4Packet (&Private->ProxyOffer.Packet.Offer, Offer);\r
- if (EFI_ERROR(Status)) {\r
- return Status;\r
- }\r
- CopyMem (&Mode->ProxyOffer, &Offer->Dhcp4, Offer->Length);\r
- Mode->ProxyOfferReceived = TRUE;\r
-\r
- PxeBcParseCachedDhcpPacket (&Private->ProxyOffer);\r
- return EFI_SUCCESS;\r
-}\r
-\r
-\r
-/**\r
- Parse the cached dhcp packet.\r
-\r
- @param CachedPacket Pointer to cached dhcp packet.\r
-\r
- @retval TRUE Succeed to parse and validation.\r
- @retval FALSE Fail to parse or validation.\r
-\r
-**/\r
-BOOLEAN\r
-PxeBcParseCachedDhcpPacket (\r
- IN PXEBC_CACHED_DHCP4_PACKET *CachedPacket\r
- )\r
-{\r
- EFI_DHCP4_PACKET *Offer;\r
- EFI_DHCP4_PACKET_OPTION **Options;\r
- EFI_DHCP4_PACKET_OPTION *Option;\r
- UINT8 OfferType;\r
- UINTN Index;\r
- UINT8 *Ptr8;\r
-\r
- CachedPacket->IsPxeOffer = FALSE;\r
- ZeroMem (CachedPacket->Dhcp4Option, sizeof (CachedPacket->Dhcp4Option));\r
- ZeroMem (&CachedPacket->PxeVendorOption, sizeof (CachedPacket->PxeVendorOption));\r
-\r
- Offer = &CachedPacket->Packet.Offer;\r
- Options = CachedPacket->Dhcp4Option;\r
-\r
- //\r
- // Parse interested dhcp options and store their pointers in CachedPacket->Dhcp4Option.\r
- // First, try to parse DHCPv4 options from the DHCP optional parameters field.\r
- //\r
- for (Index = 0; Index < PXEBC_DHCP4_TAG_INDEX_MAX; Index++) {\r
- Options[Index] = PxeBcParseExtendOptions (\r
- Offer->Dhcp4.Option,\r
- GET_OPTION_BUFFER_LEN (Offer),\r
- mInterestedDhcp4Tags[Index]\r
- );\r
- }\r
- //\r
- // Second, Check if bootfilename and serverhostname is overloaded to carry DHCP options refers to rfc-2132.\r
- // If yes, try to parse options from the BootFileName field, then ServerName field.\r
- //\r
- Option = Options[PXEBC_DHCP4_TAG_INDEX_OVERLOAD];\r
- if (Option != NULL) {\r
- if ((Option->Data[0] & PXEBC_DHCP4_OVERLOAD_FILE) != 0) {\r
- for (Index = 0; Index < PXEBC_DHCP4_TAG_INDEX_MAX; Index++) {\r
- if (Options[Index] == NULL) {\r
- Options[Index] = PxeBcParseExtendOptions (\r
- (UINT8 *) Offer->Dhcp4.Header.BootFileName,\r
- sizeof (Offer->Dhcp4.Header.BootFileName),\r
- mInterestedDhcp4Tags[Index]\r
- );\r
- }\r
- }\r
- }\r
- if ((Option->Data[0] & PXEBC_DHCP4_OVERLOAD_SERVER_NAME) != 0) {\r
- for (Index = 0; Index < PXEBC_DHCP4_TAG_INDEX_MAX; Index++) {\r
- if (Options[Index] == NULL) {\r
- Options[Index] = PxeBcParseExtendOptions (\r
- (UINT8 *) Offer->Dhcp4.Header.ServerName,\r
- sizeof (Offer->Dhcp4.Header.ServerName),\r
- mInterestedDhcp4Tags[Index]\r
- );\r
- }\r
- }\r
- }\r
- }\r
-\r
- //\r
- // Check whether is an offer with PXEClient or not.\r
- //\r
- Option = Options[PXEBC_DHCP4_TAG_INDEX_CLASS_ID];\r
- if ((Option != NULL) && (Option->Length >= 9) &&\r
- (CompareMem (Option->Data, DEFAULT_CLASS_ID_DATA, 9) == 0)) {\r
-\r
- CachedPacket->IsPxeOffer = TRUE;\r
- }\r
-\r
- //\r
- // Parse pxe vendor options and store their content/pointers in CachedPacket->PxeVendorOption.\r
- //\r
- Option = Options[PXEBC_DHCP4_TAG_INDEX_VENDOR];\r
- if (CachedPacket->IsPxeOffer && (Option != NULL)) {\r
-\r
- if (!PxeBcParseVendorOptions (Option, &CachedPacket->PxeVendorOption)) {\r
- return FALSE;\r
- }\r
- }\r
-\r
-\r
- //\r
- // Parse PXE boot file name:\r
- // According to PXE spec, boot file name should be read from DHCP option 67 (bootfile name) if present.\r
- // Otherwise, read from boot file field in DHCP header.\r
- //\r
- if (Options[PXEBC_DHCP4_TAG_INDEX_BOOTFILE] != NULL) {\r
- //\r
- // RFC 2132, Section 9.5 does not strictly state Bootfile name (option 67) is null\r
- // terminated string. So force to append null terminated character at the end of string.\r
- //\r
- Ptr8 = (UINT8*)&Options[PXEBC_DHCP4_TAG_INDEX_BOOTFILE]->Data[0];\r
- Ptr8 += Options[PXEBC_DHCP4_TAG_INDEX_BOOTFILE]->Length;\r
- if (*(Ptr8 - 1) != '\0') {\r
- *Ptr8 = '\0';\r
- }\r
- } else if (Offer->Dhcp4.Header.BootFileName[0] != 0) {\r
- //\r
- // If the bootfile is not present and bootfilename is present in dhcp packet, just parse it.\r
- // And do not count dhcp option header, or else will destroy the serverhostname.\r
- //\r
- // Make sure "BootFileName" is not overloaded.\r
- //\r
- if (Options[PXEBC_DHCP4_TAG_INDEX_OVERLOAD] == NULL ||\r
- (Options[PXEBC_DHCP4_TAG_INDEX_OVERLOAD]->Data[0] & PXEBC_DHCP4_OVERLOAD_FILE) == 0) {\r
- Options[PXEBC_DHCP4_TAG_INDEX_BOOTFILE] = (EFI_DHCP4_PACKET_OPTION *) (&Offer->Dhcp4.Header.BootFileName[0] -\r
- OFFSET_OF (EFI_DHCP4_PACKET_OPTION, Data[0]));\r
- }\r
- }\r
-\r
- //\r
- // Determine offer type of the dhcp packet.\r
- //\r
- Option = Options[PXEBC_DHCP4_TAG_INDEX_MSG_TYPE];\r
- if ((Option == NULL) || (Option->Data[0] == 0)) {\r
- //\r
- // It's a bootp offer\r
- //\r
- Option = CachedPacket->Dhcp4Option[PXEBC_DHCP4_TAG_INDEX_BOOTFILE];\r
- if (Option == NULL) {\r
- //\r
- // bootp offer without bootfilename, discard it.\r
- //\r
- return FALSE;\r
- }\r
-\r
- OfferType = DHCP4_PACKET_TYPE_BOOTP;\r
-\r
- } else {\r
-\r
- if (IS_VALID_DISCOVER_VENDOR_OPTION (CachedPacket->PxeVendorOption.BitMap)) {\r
- //\r
- // It's a pxe10 offer with PXEClient and discover vendor option.\r
- //\r
- OfferType = DHCP4_PACKET_TYPE_PXE10;\r
- } else if (IS_VALID_MTFTP_VENDOR_OPTION (CachedPacket->PxeVendorOption.BitMap)) {\r
- //\r
- // It's a wfm11a offer with PXEClient and mtftp vendor option, and\r
- // return false since mtftp not supported currently.\r
- //\r
- return FALSE;\r
- } else {\r
- //\r
- // If the binl offer with only PXEClient.\r
- //\r
- OfferType = (UINT8) ((CachedPacket->IsPxeOffer) ? DHCP4_PACKET_TYPE_BINL : DHCP4_PACKET_TYPE_DHCP_ONLY);\r
- }\r
- }\r
-\r
- CachedPacket->OfferType = OfferType;\r
-\r
- return TRUE;\r
-}\r
-\r
-\r
-/**\r
- Offer dhcp service with a BINL dhcp offer.\r
-\r
- @param Private Pointer to PxeBc private data.\r
- @param Index Index of cached packets as complements of pxe mode data,\r
- the index is maximum offer number.\r
-\r
- @retval TRUE Offer the service successfully under priority BINL.\r
- @retval FALSE Boot Service failed, parse cached dhcp packet failed or this\r
- BINL ack cannot find options set or bootfile name specified.\r
-\r
-**/\r
-BOOLEAN\r
-PxeBcTryBinl (\r
- IN PXEBC_PRIVATE_DATA *Private,\r
- IN UINT32 Index\r
- )\r
-{\r
- EFI_DHCP4_PACKET *Offer;\r
- EFI_IP_ADDRESS ServerIp;\r
- EFI_STATUS Status;\r
- PXEBC_CACHED_DHCP4_PACKET *CachedPacket;\r
- EFI_DHCP4_PACKET *Reply;\r
-\r
- ASSERT (Index < PXEBC_MAX_OFFER_NUM);\r
- ASSERT (Private->Dhcp4Offers[Index].OfferType == DHCP4_PACKET_TYPE_BINL);\r
-\r
- Offer = &Private->Dhcp4Offers[Index].Packet.Offer;\r
-\r
- //\r
- // Use siaddr(next server) in DHCPOFFER packet header, if zero, use option 54(server identifier)\r
- // in DHCPOFFER packet.\r
- // (It does not comply with PXE Spec, Ver2.1)\r
- //\r
- if (EFI_IP4_EQUAL (&Offer->Dhcp4.Header.ServerAddr.Addr, &mZeroIp4Addr)) {\r
- CopyMem (\r
- &ServerIp.Addr[0],\r
- Private->Dhcp4Offers[Index].Dhcp4Option[PXEBC_DHCP4_TAG_INDEX_SERVER_ID]->Data,\r
- sizeof (EFI_IPv4_ADDRESS)\r
- );\r
- } else {\r
- CopyMem (\r
- &ServerIp.Addr[0],\r
- &Offer->Dhcp4.Header.ServerAddr,\r
- sizeof (EFI_IPv4_ADDRESS)\r
- );\r
- }\r
- if (ServerIp.Addr[0] == 0) {\r
- return FALSE;\r
- }\r
-\r
- CachedPacket = &Private->ProxyOffer;\r
- Reply = &CachedPacket->Packet.Offer;\r
-\r
- Status = PxeBcDiscvBootService (\r
- Private,\r
- 0,\r
- NULL,\r
- FALSE,\r
- &ServerIp,\r
- 0,\r
- NULL,\r
- FALSE,\r
- Reply\r
- );\r
- if (EFI_ERROR (Status)) {\r
- return FALSE;\r
- }\r
-\r
- if (!PxeBcParseCachedDhcpPacket (CachedPacket)) {\r
- return FALSE;\r
- }\r
-\r
- if ((CachedPacket->OfferType != DHCP4_PACKET_TYPE_PXE10) &&\r
- (CachedPacket->Dhcp4Option[PXEBC_DHCP4_TAG_INDEX_BOOTFILE] == NULL)) {\r
- //\r
- // This BINL ack doesn't have discovery options set or bootfile name\r
- // specified.\r
- //\r
- return FALSE;\r
- }\r
-\r
- Private->PxeBc.Mode->ProxyOfferReceived = TRUE;\r
- CopyMem (&Private->PxeBc.Mode->ProxyOffer, &Reply->Dhcp4, Reply->Length);\r
-\r
- return TRUE;\r
-}\r
-\r
-\r
-/**\r
- Offer dhcp service for each proxy with a BINL dhcp offer.\r
-\r
- @param Private Pointer to PxeBc private data\r
- @param OfferIndex Pointer to the index of cached packets as complements of\r
- pxe mode data, the index is maximum offer number.\r
-\r
- @return If there is no service needed offer return FALSE, otherwise TRUE.\r
-\r
-**/\r
-BOOLEAN\r
-PxeBcTryBinlProxy (\r
- IN PXEBC_PRIVATE_DATA *Private,\r
- OUT UINT32 *OfferIndex\r
- )\r
-{\r
- UINT32 Index;\r
-\r
- for (Index = 0; Index < Private->ProxyIndex[DHCP4_PACKET_TYPE_BINL]; Index++) {\r
-\r
- *OfferIndex = Private->BinlIndex[Index];\r
- //\r
- // Try this BINL proxy offer\r
- //\r
- if (PxeBcTryBinl (Private, *OfferIndex)) {\r
- return TRUE;\r
- }\r
- }\r
-\r
- return FALSE;\r
-}\r
-\r
-\r
-/**\r
- This function is to check the selected proxy offer (include BINL dhcp offer and\r
- DHCP_ONLY offer ) and set the flag and copy the DHCP packets to the Pxe base code\r
- mode structure.\r
-\r
- @param Private Pointer to PxeBc private data.\r
-\r
- @retval EFI_SUCCESS Operational successful.\r
- @retval EFI_NO_RESPONSE Offer dhcp service failed.\r
- @retval EFI_BUFFER_TOO_SMALL Failed to copy the packet to Pxe base code mode.\r
-\r
-**/\r
-EFI_STATUS\r
-PxeBcCheckSelectedOffer (\r
- IN PXEBC_PRIVATE_DATA *Private\r
- )\r
-{\r
- PXEBC_CACHED_DHCP4_PACKET *SelectedOffer;\r
- EFI_DHCP4_PACKET_OPTION **Options;\r
- UINT32 Index;\r
- EFI_DHCP4_PACKET *Offer;\r
- UINT32 ProxyOfferIndex;\r
- EFI_STATUS Status;\r
- EFI_PXE_BASE_CODE_MODE *Mode;\r
- EFI_DHCP4_PACKET *Ack;\r
-\r
- ASSERT (Private->SelectedOffer != 0);\r
-\r
- Status = EFI_SUCCESS;\r
- SelectedOffer = &Private->Dhcp4Offers[Private->SelectedOffer - 1];\r
- Options = SelectedOffer->Dhcp4Option;\r
-\r
- if (SelectedOffer->OfferType == DHCP4_PACKET_TYPE_BINL) {\r
- //\r
- // The addresses are acquired from a BINL dhcp offer, try BINL to get\r
- // the bootfile name\r
- //\r
- if (!PxeBcTryBinl (Private, Private->SelectedOffer - 1)) {\r
- Status = EFI_NO_RESPONSE;\r
- }\r
- } else if (SelectedOffer->OfferType == DHCP4_PACKET_TYPE_DHCP_ONLY) {\r
- //\r
- // The selected offer to finish the D.O.R.A. is a DHCP only offer, we need\r
- // try proxy offers if there are some, othewise the bootfile name must be\r
- // set in this DHCP only offer.\r
- //\r
- if (Private->GotProxyOffer) {\r
- //\r
- // Get rid of the compiler warning.\r
- //\r
- ProxyOfferIndex = 0;\r
- if (Private->SortOffers) {\r
- //\r
- // The offers are sorted before selecting, the proxy offer type must be\r
- // already determined.\r
- //\r
- ASSERT (Private->ProxyIndex[Private->ProxyOfferType] > 0);\r
-\r
- if (Private->ProxyOfferType == DHCP4_PACKET_TYPE_BINL) {\r
- //\r
- // We buffer all received BINL proxy offers, try them all one by one\r
- //\r
- if (!PxeBcTryBinlProxy (Private, &ProxyOfferIndex)) {\r
- Status = EFI_NO_RESPONSE;\r
- }\r
- } else {\r
- //\r
- // For other types, only one proxy offer is buffered.\r
- //\r
- ProxyOfferIndex = Private->ProxyIndex[Private->ProxyOfferType] - 1;\r
- }\r
- } else {\r
- //\r
- // The proxy offer type is not determined, choose proxy offer in the\r
- // received order.\r
- //\r
- Status = EFI_NO_RESPONSE;\r
-\r
- ASSERT (Private->NumOffers < PXEBC_MAX_OFFER_NUM);\r
- for (Index = 0; Index < Private->NumOffers; Index++) {\r
-\r
- Offer = &Private->Dhcp4Offers[Index].Packet.Offer;\r
- if (!IS_PROXY_DHCP_OFFER (Offer)) {\r
- //\r
- // Skip non proxy dhcp offers.\r
- //\r
- continue;\r
- }\r
-\r
- if (Private->Dhcp4Offers[Index].OfferType == DHCP4_PACKET_TYPE_BINL) {\r
- //\r
- // Try BINL\r
- //\r
- if (!PxeBcTryBinl (Private, Index)) {\r
- //\r
- // Failed, skip to the next offer\r
- //\r
- continue;\r
- }\r
- }\r
-\r
- Private->ProxyOfferType = Private->Dhcp4Offers[Index].OfferType;\r
- ProxyOfferIndex = Index;\r
- Status = EFI_SUCCESS;\r
- break;\r
- }\r
- }\r
-\r
- if (!EFI_ERROR (Status) && (Private->ProxyOfferType != DHCP4_PACKET_TYPE_BINL)) {\r
- //\r
- // Copy the proxy offer to Mode and set the flag\r
- //\r
- Status = PxeBcCopyProxyOffer (Private, ProxyOfferIndex);\r
- }\r
- } else {\r
- //\r
- // No proxy offer is received, the bootfile name MUST be set.\r
- //\r
- ASSERT (Options[PXEBC_DHCP4_TAG_INDEX_BOOTFILE] != NULL);\r
- }\r
- }\r
-\r
- if (!EFI_ERROR (Status)) {\r
- //\r
- // Everything is OK, set the flag and copy the DHCP packets.\r
- //\r
- Mode = Private->PxeBc.Mode;\r
- Offer = &SelectedOffer->Packet.Offer;\r
-\r
- //\r
- // The discover packet is already copied, just set flag here.\r
- //\r
- Mode->DhcpDiscoverValid = TRUE;\r
-\r
- Ack = &Private->Dhcp4Ack.Packet.Ack;\r
- if (SelectedOffer->OfferType == DHCP4_PACKET_TYPE_BOOTP) {\r
- //\r
- // Other type of ACK is already cached. Bootp is special that we should\r
- // use the bootp reply as the ACK and put it into the DHCP_ONLY buffer.\r
- //\r
- Status = PxeBcCopyEfiDhcp4Packet (&Private->Dhcp4Ack.Packet.Ack, Offer);\r
- }\r
-\r
- PxeBcParseCachedDhcpPacket (&Private->Dhcp4Ack);\r
-\r
- Mode->DhcpAckReceived = TRUE;\r
-\r
- //\r
- // Copy the dhcp ack.\r
- //\r
- CopyMem (&Mode->DhcpAck, &Ack->Dhcp4, Ack->Length);\r
- }\r
-\r
- return Status;\r
-}\r
-\r
-\r
-/**\r
- Cache the Dhcp4 packet offer, Parse and validate each option of the packet.\r
-\r
- @param Private Pointer to PxeBc private data.\r
- @param RcvdOffer Pointer to the received Dhcp proxy offer packet.\r
-\r
- @retval EFI_SUCCESS Cache and parse the packet successfully.\r
- @retval Others Operation failed.\r
-\r
-**/\r
-EFI_STATUS\r
-PxeBcCacheDhcpOffer (\r
- IN PXEBC_PRIVATE_DATA *Private,\r
- IN EFI_DHCP4_PACKET *RcvdOffer\r
- )\r
-{\r
- PXEBC_CACHED_DHCP4_PACKET *CachedOffer;\r
- EFI_DHCP4_PACKET *Offer;\r
- UINT8 OfferType;\r
- EFI_STATUS Status;\r
-\r
- CachedOffer = &Private->Dhcp4Offers[Private->NumOffers];\r
- Offer = &CachedOffer->Packet.Offer;\r
-\r
- //\r
- // Cache the orignal dhcp packet\r
- //\r
- Status = PxeBcCopyEfiDhcp4Packet (Offer, RcvdOffer);\r
- if (EFI_ERROR(Status)) {\r
- return Status;\r
- }\r
-\r
- //\r
- // Parse and validate the options (including dhcp option and vendor option)\r
- //\r
- if (!PxeBcParseCachedDhcpPacket (CachedOffer)) {\r
- return EFI_ABORTED;\r
- }\r
-\r
- OfferType = CachedOffer->OfferType;\r
- if (OfferType >= DHCP4_PACKET_TYPE_MAX) {\r
- return EFI_ABORTED;\r
- }\r
-\r
- if (OfferType == DHCP4_PACKET_TYPE_BOOTP) {\r
-\r
- if (Private->BootpIndex != 0) {\r
- //\r
- // Only cache the first bootp offer, discard others.\r
- //\r
- return EFI_ABORTED;\r
- } else {\r
- //\r
- // Take as a dhcp only offer, but record index specifically.\r
- //\r
- Private->BootpIndex = Private->NumOffers + 1;\r
- }\r
- } else {\r
-\r
- if (IS_PROXY_DHCP_OFFER (Offer)) {\r
- //\r
- // It's a proxy dhcp offer with no your address, including pxe10, wfm11a or binl offer.\r
- //\r
- Private->GotProxyOffer = TRUE;\r
-\r
- if (OfferType == DHCP4_PACKET_TYPE_BINL) {\r
- //\r
- // Cache all binl offers.\r
- //\r
- Private->BinlIndex[Private->ProxyIndex[DHCP4_PACKET_TYPE_BINL]] = Private->NumOffers;\r
- Private->ProxyIndex[DHCP4_PACKET_TYPE_BINL]++;\r
- } else if (Private->ProxyIndex[OfferType] != 0) {\r
- //\r
- // Only cache the first pxe10/wfm11a offers each, discard the others.\r
- //\r
- return EFI_ABORTED;\r
- } else {\r
- //\r
- // Record index of the proxy dhcp offer with type other than binl.\r
- //\r
- Private->ProxyIndex[OfferType] = Private->NumOffers + 1;\r
- }\r
- } else {\r
- //\r
- // It's a dhcp offer with your address.\r
- //\r
- ASSERT (Private->ServerCount[OfferType] < PXEBC_MAX_OFFER_NUM);\r
- Private->OfferIndex[OfferType][Private->ServerCount[OfferType]] = Private->NumOffers;\r
- Private->ServerCount[OfferType]++;\r
- }\r
- }\r
-\r
- //\r
- // Count the accepted offers.\r
- //\r
- Private->NumOffers++;\r
-\r
- return EFI_SUCCESS;\r
-}\r
-\r
-/**\r
- Switch the Ip4 policy to static.\r
-\r
- @param[in] Private The pointer to PXEBC_PRIVATE_DATA.\r
-\r
- @retval EFI_SUCCESS The policy is already configured to static.\r
- @retval Others Other error as indicated..\r
-\r
-**/\r
-EFI_STATUS\r
-PxeBcSetIp4Policy (\r
- IN PXEBC_PRIVATE_DATA *Private\r
- )\r
-{\r
- EFI_STATUS Status;\r
- EFI_IP4_CONFIG2_PROTOCOL *Ip4Config2;\r
- EFI_IP4_CONFIG2_POLICY Policy;\r
- UINTN DataSize;\r
-\r
- Ip4Config2 = Private->Ip4Config2;\r
- DataSize = sizeof (EFI_IP4_CONFIG2_POLICY);\r
- Status = Ip4Config2->GetData (\r
- Ip4Config2,\r
- Ip4Config2DataTypePolicy,\r
- &DataSize,\r
- &Policy\r
- );\r
- if (EFI_ERROR (Status)) {\r
- return Status;\r
- }\r
-\r
- if (Policy != Ip4Config2PolicyStatic) {\r
- Policy = Ip4Config2PolicyStatic;\r
- Status= Ip4Config2->SetData (\r
- Ip4Config2,\r
- Ip4Config2DataTypePolicy,\r
- sizeof (EFI_IP4_CONFIG2_POLICY),\r
- &Policy\r
- );\r
- if (EFI_ERROR (Status)) {\r
- return Status;\r
- }\r
- }\r
-\r
- return EFI_SUCCESS;\r
-}\r
-\r
-\r
-/**\r
- Select the specified proxy offer, such as BINL, DHCP_ONLY and so on.\r
- If the proxy does not exist, try offers with bootfile.\r
-\r
- @param Private Pointer to PxeBc private data.\r
-\r
-**/\r
-VOID\r
-PxeBcSelectOffer (\r
- IN PXEBC_PRIVATE_DATA *Private\r
- )\r
-{\r
- UINT32 Index;\r
- UINT32 OfferIndex;\r
- EFI_DHCP4_PACKET *Offer;\r
-\r
- Private->SelectedOffer = 0;\r
-\r
- if (Private->SortOffers) {\r
- //\r
- // Select offer according to the priority\r
- //\r
- if (Private->ServerCount[DHCP4_PACKET_TYPE_PXE10] > 0) {\r
- //\r
- // DHCP with PXE10\r
- //\r
- Private->SelectedOffer = Private->OfferIndex[DHCP4_PACKET_TYPE_PXE10][0] + 1;\r
-\r
- } else if (Private->ServerCount[DHCP4_PACKET_TYPE_WFM11A] > 0) {\r
- //\r
- // DHCP with WfM\r
- //\r
- Private->SelectedOffer = Private->OfferIndex[DHCP4_PACKET_TYPE_WFM11A][0] + 1;\r
-\r
- } else if ((Private->ProxyIndex[DHCP4_PACKET_TYPE_PXE10] > 0) &&\r
- (Private->ServerCount[DHCP4_PACKET_TYPE_DHCP_ONLY] > 0)\r
- ) {\r
- //\r
- // DHCP only and proxy DHCP with PXE10\r
- //\r
- Private->SelectedOffer = Private->OfferIndex[DHCP4_PACKET_TYPE_DHCP_ONLY][0] + 1;\r
- Private->ProxyOfferType = DHCP4_PACKET_TYPE_PXE10;\r
-\r
- } else if ((Private->ProxyIndex[DHCP4_PACKET_TYPE_WFM11A] > 0) &&\r
- (Private->ServerCount[DHCP4_PACKET_TYPE_DHCP_ONLY] > 0)\r
- ) {\r
- //\r
- // DHCP only and proxy DHCP with WfM\r
- //\r
- Private->SelectedOffer = Private->OfferIndex[DHCP4_PACKET_TYPE_DHCP_ONLY][0] + 1;\r
- Private->ProxyOfferType = DHCP4_PACKET_TYPE_WFM11A;\r
-\r
- } else if (Private->ServerCount[DHCP4_PACKET_TYPE_BINL] > 0) {\r
- //\r
- // DHCP with BINL\r
- //\r
- Private->SelectedOffer = Private->OfferIndex[DHCP4_PACKET_TYPE_BINL][0] + 1;\r
-\r
- } else if ((Private->ProxyIndex[DHCP4_PACKET_TYPE_BINL] > 0) &&\r
- (Private->ServerCount[DHCP4_PACKET_TYPE_DHCP_ONLY] > 0)\r
- ) {\r
- //\r
- // DHCP only and proxy DHCP with BINL\r
- //\r
- Private->SelectedOffer = Private->OfferIndex[DHCP4_PACKET_TYPE_DHCP_ONLY][0] + 1;\r
- Private->ProxyOfferType = DHCP4_PACKET_TYPE_BINL;\r
-\r
- } else {\r
- //\r
- // Try offers with bootfile\r
- //\r
- for (Index = 0; Index < Private->ServerCount[DHCP4_PACKET_TYPE_DHCP_ONLY]; Index++) {\r
- //\r
- // Select the first DHCP only offer with bootfile\r
- //\r
- OfferIndex = Private->OfferIndex[DHCP4_PACKET_TYPE_DHCP_ONLY][Index];\r
- if (Private->Dhcp4Offers[OfferIndex].Dhcp4Option[PXEBC_DHCP4_TAG_INDEX_BOOTFILE] != NULL) {\r
- Private->SelectedOffer = OfferIndex + 1;\r
- break;\r
- }\r
- }\r
-\r
- if (Private->SelectedOffer == 0) {\r
- //\r
- // Select the Bootp reply with bootfile if any\r
- //\r
- Private->SelectedOffer = Private->BootpIndex;\r
- }\r
- }\r
- } else {\r
- //\r
- // Try the offers in the received order.\r
- //\r
- for (Index = 0; Index < Private->NumOffers; Index++) {\r
-\r
- Offer = &Private->Dhcp4Offers[Index].Packet.Offer;\r
-\r
- if (IS_PROXY_DHCP_OFFER (Offer)) {\r
- //\r
- // Skip proxy offers\r
- //\r
- continue;\r
- }\r
-\r
- if ((Private->Dhcp4Offers[Index].OfferType == DHCP4_PACKET_TYPE_DHCP_ONLY) &&\r
- ((!Private->GotProxyOffer) && (Private->Dhcp4Offers[Index].Dhcp4Option[PXEBC_DHCP4_TAG_INDEX_BOOTFILE] == NULL))) {\r
- //\r
- // DHCP only offer but no proxy offer received and no bootfile option in this offer\r
- //\r
- continue;\r
- }\r
-\r
- Private->SelectedOffer = Index + 1;\r
- break;\r
- }\r
- }\r
-}\r
-\r
-\r
-/**\r
- Callback routine.\r
-\r
- EFI_DHCP4_CALLBACK is provided by the consumer of the EFI DHCPv4 Protocol driver\r
- to intercept events that occurred in the configuration process. This structure\r
- provides advanced control of each state transition of the DHCP process. The\r
- returned status code determines the behavior of the EFI DHCPv4 Protocol driver.\r
- There are three possible returned values, which are described in the following\r
- table.\r
-\r
- @param This Pointer to the EFI DHCPv4 Protocol instance that is used to\r
- configure this callback function.\r
- @param Context Pointer to the context that is initialized by\r
- EFI_DHCP4_PROTOCOL.Configure().\r
- @param CurrentState The current operational state of the EFI DHCPv4 Protocol\r
- driver.\r
- @param Dhcp4Event The event that occurs in the current state, which usually means a\r
- state transition.\r
- @param Packet The DHCP packet that is going to be sent or already received.\r
- @param NewPacket The packet that is used to replace the above Packet.\r
-\r
- @retval EFI_SUCCESS Tells the EFI DHCPv4 Protocol driver to continue the DHCP process.\r
- @retval EFI_NOT_READY Only used in the Dhcp4Selecting state. The EFI DHCPv4 Protocol\r
- driver will continue to wait for more DHCPOFFER packets until the retry\r
- timeout expires.\r
- @retval EFI_ABORTED Tells the EFI DHCPv4 Protocol driver to abort the current process and\r
- return to the Dhcp4Init or Dhcp4InitReboot state.\r
-\r
-**/\r
-EFI_STATUS\r
-EFIAPI\r
-PxeBcDhcpCallBack (\r
- IN EFI_DHCP4_PROTOCOL * This,\r
- IN VOID *Context,\r
- IN EFI_DHCP4_STATE CurrentState,\r
- IN EFI_DHCP4_EVENT Dhcp4Event,\r
- IN EFI_DHCP4_PACKET * Packet OPTIONAL,\r
- OUT EFI_DHCP4_PACKET **NewPacket OPTIONAL\r
- )\r
-{\r
- PXEBC_PRIVATE_DATA *Private;\r
- EFI_PXE_BASE_CODE_MODE *Mode;\r
- EFI_PXE_BASE_CODE_CALLBACK_PROTOCOL *Callback;\r
- EFI_DHCP4_PACKET_OPTION *MaxMsgSize;\r
- UINT16 Value;\r
- EFI_STATUS Status;\r
- BOOLEAN Received;\r
- EFI_DHCP4_HEADER *DhcpHeader;\r
-\r
- if ((Dhcp4Event != Dhcp4RcvdOffer) &&\r
- (Dhcp4Event != Dhcp4SelectOffer) &&\r
- (Dhcp4Event != Dhcp4SendDiscover) &&\r
- (Dhcp4Event != Dhcp4RcvdAck) &&\r
- (Dhcp4Event != Dhcp4SendRequest)) {\r
- return EFI_SUCCESS;\r
- }\r
-\r
- Private = (PXEBC_PRIVATE_DATA *) Context;\r
- Mode = Private->PxeBc.Mode;\r
- Callback = Private->PxeBcCallback;\r
-\r
- //\r
- // Override the Maximum DHCP Message Size.\r
- //\r
- MaxMsgSize = PxeBcParseExtendOptions (\r
- Packet->Dhcp4.Option,\r
- GET_OPTION_BUFFER_LEN (Packet),\r
- DHCP4_TAG_MAXMSG\r
- );\r
- if (MaxMsgSize != NULL) {\r
- Value = HTONS (PXEBC_DHCP4_MAX_PACKET_SIZE);\r
- CopyMem (MaxMsgSize->Data, &Value, sizeof (Value));\r
- }\r
-\r
- if ((Dhcp4Event != Dhcp4SelectOffer) && (Callback != NULL)) {\r
- Received = (BOOLEAN) ((Dhcp4Event == Dhcp4RcvdOffer) || (Dhcp4Event == Dhcp4RcvdAck));\r
- Status = Callback->Callback (\r
- Callback,\r
- Private->Function,\r
- Received,\r
- Packet->Length,\r
- (EFI_PXE_BASE_CODE_PACKET *) &Packet->Dhcp4\r
- );\r
- if (Status != EFI_PXE_BASE_CODE_CALLBACK_STATUS_CONTINUE) {\r
- return EFI_ABORTED;\r
- }\r
- }\r
-\r
- Status = EFI_SUCCESS;\r
-\r
- switch (Dhcp4Event) {\r
-\r
- case Dhcp4SendDiscover:\r
- case Dhcp4SendRequest:\r
- if (Packet->Length > PXEBC_DHCP4_MAX_PACKET_SIZE) {\r
- //\r
- // If the to be sent packet exceeds the maximum length, abort the DHCP process.\r
- //\r
- Status = EFI_ABORTED;\r
- break;\r
- }\r
-\r
- if (Mode->SendGUID) {\r
- //\r
- // send the system GUID instead of the MAC address as the hardware address\r
- // in the DHCP packet header.\r
- //\r
- DhcpHeader = &Packet->Dhcp4.Header;\r
-\r
- if (EFI_ERROR (NetLibGetSystemGuid ((EFI_GUID *) DhcpHeader->ClientHwAddr))) {\r
- //\r
- // GUID not yet set - send all 0xff's to show programable (via SetVariable)\r
- // SetMem(DHCPV4_OPTIONS_BUFFER.DhcpPlatformId.Guid, sizeof(EFI_GUID), 0xff);\r
- // GUID not yet set - send all 0's to show not programable\r
- //\r
- DEBUG ((EFI_D_WARN, "PXE: Failed to read system GUID from the smbios table!\n"));\r
- ZeroMem (DhcpHeader->ClientHwAddr, sizeof (EFI_GUID));\r
- }\r
-\r
- DhcpHeader->HwAddrLen = (UINT8) sizeof (EFI_GUID);\r
- }\r
-\r
- if (Dhcp4Event == Dhcp4SendDiscover) {\r
- //\r
- // Cache the dhcp discover packet, of which some information will be used later.\r
- //\r
- CopyMem (Mode->DhcpDiscover.Raw, &Packet->Dhcp4, Packet->Length);\r
- }\r
-\r
- break;\r
-\r
- case Dhcp4RcvdOffer:\r
- Status = EFI_NOT_READY;\r
- if (Packet->Length > PXEBC_DHCP4_MAX_PACKET_SIZE) {\r
- //\r
- // Ignore the incoming Offers which exceed the maximum length.\r
- //\r
- break;\r
- }\r
-\r
- if (Private->NumOffers < PXEBC_MAX_OFFER_NUM) {\r
- //\r
- // Cache the dhcp offers in Private->Dhcp4Offers[]\r
- // If error happens, just ignore this packet and continue to wait more offer.\r
- //\r
- PxeBcCacheDhcpOffer (Private, Packet);\r
- }\r
-\r
- break;\r
-\r
- case Dhcp4SelectOffer:\r
- //\r
- // Select an offer, if succeeded, Private->SelectedOffer points to\r
- // the index of the selected one.\r
- //\r
- PxeBcSelectOffer (Private);\r
-\r
- if (Private->SelectedOffer == 0) {\r
- Status = EFI_ABORTED;\r
- } else {\r
- *NewPacket = &Private->Dhcp4Offers[Private->SelectedOffer - 1].Packet.Offer;\r
- }\r
-\r
- break;\r
-\r
- case Dhcp4RcvdAck:\r
- //\r
- // Cache Ack\r
- //\r
- ASSERT (Private->SelectedOffer != 0);\r
-\r
- Status = PxeBcCopyEfiDhcp4Packet (&Private->Dhcp4Ack.Packet.Ack, Packet);\r
- if (EFI_ERROR (Status)) {\r
- return EFI_ABORTED;\r
- }\r
- break;\r
-\r
- default:\r
- break;\r
- }\r
-\r
- return Status;\r
-}\r
-\r
-\r
-/**\r
- Initialize the DHCP options and build the option list.\r
-\r
- @param Private Pointer to PxeBc private data.\r
- @param OptList Pointer to a DHCP option list.\r
-\r
- @param IsDhcpDiscover Discover dhcp option or not.\r
-\r
- @return The index item number of the option list.\r
-\r
-**/\r
-UINT32\r
-PxeBcBuildDhcpOptions (\r
- IN PXEBC_PRIVATE_DATA *Private,\r
- IN EFI_DHCP4_PACKET_OPTION **OptList,\r
- IN BOOLEAN IsDhcpDiscover\r
- )\r
-{\r
- UINT32 Index;\r
- PXEBC_DHCP4_OPTION_ENTRY OptEnt;\r
- UINT16 Value;\r
-\r
- Index = 0;\r
- OptList[0] = (EFI_DHCP4_PACKET_OPTION *) Private->OptionBuffer;\r
-\r
- if (!IsDhcpDiscover) {\r
- //\r
- // Append message type.\r
- //\r
- OptList[Index]->OpCode = DHCP4_TAG_MSG_TYPE;\r
- OptList[Index]->Length = 1;\r
- OptEnt.Mesg = (PXEBC_DHCP4_OPTION_MESG *) OptList[Index]->Data;\r
- OptEnt.Mesg->Type = PXEBC_DHCP4_MSG_TYPE_REQUEST;\r
- Index++;\r
- OptList[Index] = GET_NEXT_DHCP_OPTION (OptList[Index - 1]);\r
-\r
- //\r
- // Append max message size.\r
- //\r
- OptList[Index]->OpCode = DHCP4_TAG_MAXMSG;\r
- OptList[Index]->Length = (UINT8) sizeof (PXEBC_DHCP4_OPTION_MAX_MESG_SIZE);\r
- OptEnt.MaxMesgSize = (PXEBC_DHCP4_OPTION_MAX_MESG_SIZE *) OptList[Index]->Data;\r
- Value = NTOHS (PXEBC_DHCP4_MAX_PACKET_SIZE);\r
- CopyMem (&OptEnt.MaxMesgSize->Size, &Value, sizeof (UINT16));\r
- Index++;\r
- OptList[Index] = GET_NEXT_DHCP_OPTION (OptList[Index - 1]);\r
- }\r
- //\r
- // Parameter request list option.\r
- //\r
- OptList[Index]->OpCode = DHCP4_TAG_PARA_LIST;\r
- OptList[Index]->Length = 35;\r
- OptEnt.Para = (PXEBC_DHCP4_OPTION_PARA *) OptList[Index]->Data;\r
- OptEnt.Para->ParaList[0] = DHCP4_TAG_NETMASK;\r
- OptEnt.Para->ParaList[1] = DHCP4_TAG_TIME_OFFSET;\r
- OptEnt.Para->ParaList[2] = DHCP4_TAG_ROUTER;\r
- OptEnt.Para->ParaList[3] = DHCP4_TAG_TIME_SERVER;\r
- OptEnt.Para->ParaList[4] = DHCP4_TAG_NAME_SERVER;\r
- OptEnt.Para->ParaList[5] = DHCP4_TAG_DNS_SERVER;\r
- OptEnt.Para->ParaList[6] = DHCP4_TAG_HOSTNAME;\r
- OptEnt.Para->ParaList[7] = DHCP4_TAG_BOOTFILE_LEN;\r
- OptEnt.Para->ParaList[8] = DHCP4_TAG_DOMAINNAME;\r
- OptEnt.Para->ParaList[9] = DHCP4_TAG_ROOTPATH;\r
- OptEnt.Para->ParaList[10] = DHCP4_TAG_EXTEND_PATH;\r
- OptEnt.Para->ParaList[11] = DHCP4_TAG_EMTU;\r
- OptEnt.Para->ParaList[12] = DHCP4_TAG_TTL;\r
- OptEnt.Para->ParaList[13] = DHCP4_TAG_BROADCAST;\r
- OptEnt.Para->ParaList[14] = DHCP4_TAG_NIS_DOMAIN;\r
- OptEnt.Para->ParaList[15] = DHCP4_TAG_NIS_SERVER;\r
- OptEnt.Para->ParaList[16] = DHCP4_TAG_NTP_SERVER;\r
- OptEnt.Para->ParaList[17] = DHCP4_TAG_VENDOR;\r
- OptEnt.Para->ParaList[18] = DHCP4_TAG_REQUEST_IP;\r
- OptEnt.Para->ParaList[19] = DHCP4_TAG_LEASE;\r
- OptEnt.Para->ParaList[20] = DHCP4_TAG_SERVER_ID;\r
- OptEnt.Para->ParaList[21] = DHCP4_TAG_T1;\r
- OptEnt.Para->ParaList[22] = DHCP4_TAG_T2;\r
- OptEnt.Para->ParaList[23] = DHCP4_TAG_VENDOR_CLASS_ID;\r
- OptEnt.Para->ParaList[24] = DHCP4_TAG_TFTP;\r
- OptEnt.Para->ParaList[25] = DHCP4_TAG_BOOTFILE;\r
- OptEnt.Para->ParaList[26] = DHCP4_TAG_UUID;\r
- OptEnt.Para->ParaList[27] = 0x80;\r
- OptEnt.Para->ParaList[28] = 0x81;\r
- OptEnt.Para->ParaList[29] = 0x82;\r
- OptEnt.Para->ParaList[30] = 0x83;\r
- OptEnt.Para->ParaList[31] = 0x84;\r
- OptEnt.Para->ParaList[32] = 0x85;\r
- OptEnt.Para->ParaList[33] = 0x86;\r
- OptEnt.Para->ParaList[34] = 0x87;\r
- Index++;\r
- OptList[Index] = GET_NEXT_DHCP_OPTION (OptList[Index - 1]);\r
-\r
- //\r
- // Append UUID/Guid-based client identifier option\r
- //\r
- OptList[Index]->OpCode = DHCP4_TAG_UUID;\r
- OptList[Index]->Length = (UINT8) sizeof (PXEBC_DHCP4_OPTION_UUID);\r
- OptEnt.Uuid = (PXEBC_DHCP4_OPTION_UUID *) OptList[Index]->Data;\r
- OptEnt.Uuid->Type = 0;\r
- Index++;\r
- OptList[Index] = GET_NEXT_DHCP_OPTION (OptList[Index - 1]);\r
-\r
- if (EFI_ERROR (NetLibGetSystemGuid ((EFI_GUID *) OptEnt.Uuid->Guid))) {\r
- //\r
- // GUID not yet set - send all 0xff's to show programable (via SetVariable)\r
- // SetMem(DHCPV4_OPTIONS_BUFFER.DhcpPlatformId.Guid, sizeof(EFI_GUID), 0xff);\r
- // GUID not yet set - send all 0's to show not programable\r
- //\r
- DEBUG ((EFI_D_WARN, "PXE: Failed to read system GUID from the smbios table!\n"));\r
- ZeroMem (OptEnt.Uuid->Guid, sizeof (EFI_GUID));\r
- }\r
-\r
- //\r
- // Append client network device interface option\r
- //\r
- OptList[Index]->OpCode = DHCP4_TAG_UNDI;\r
- OptList[Index]->Length = (UINT8) sizeof (PXEBC_DHCP4_OPTION_UNDI);\r
- OptEnt.Undi = (PXEBC_DHCP4_OPTION_UNDI *) OptList[Index]->Data;\r
- if (Private->Nii != NULL) {\r
- OptEnt.Undi->Type = Private->Nii->Type;\r
- OptEnt.Undi->MajorVer = Private->Nii->MajorVer;\r
- OptEnt.Undi->MinorVer = Private->Nii->MinorVer;\r
- } else {\r
- OptEnt.Undi->Type = DEFAULT_UNDI_TYPE;\r
- OptEnt.Undi->MajorVer = DEFAULT_UNDI_MAJOR;\r
- OptEnt.Undi->MinorVer = DEFAULT_UNDI_MINOR;\r
- }\r
-\r
- Index++;\r
- OptList[Index] = GET_NEXT_DHCP_OPTION (OptList[Index - 1]);\r
-\r
- //\r
- // Append client system architecture option\r
- //\r
- OptList[Index]->OpCode = DHCP4_TAG_ARCH;\r
- OptList[Index]->Length = (UINT8) sizeof (PXEBC_DHCP4_OPTION_ARCH);\r
- OptEnt.Arch = (PXEBC_DHCP4_OPTION_ARCH *) OptList[Index]->Data;\r
- Value = HTONS (EFI_PXE_CLIENT_SYSTEM_ARCHITECTURE);\r
- CopyMem (&OptEnt.Arch->Type, &Value, sizeof (UINT16));\r
- Index++;\r
- OptList[Index] = GET_NEXT_DHCP_OPTION (OptList[Index - 1]);\r
-\r
- //\r
- // Append client system architecture option\r
- //\r
- OptList[Index]->OpCode = DHCP4_TAG_VENDOR_CLASS_ID;\r
- OptList[Index]->Length = (UINT8) sizeof (PXEBC_DHCP4_OPTION_CLID);\r
- OptEnt.Clid = (PXEBC_DHCP4_OPTION_CLID *) OptList[Index]->Data;\r
- CopyMem (OptEnt.Clid, DEFAULT_CLASS_ID_DATA, sizeof (PXEBC_DHCP4_OPTION_CLID));\r
- CvtNum (EFI_PXE_CLIENT_SYSTEM_ARCHITECTURE, OptEnt.Clid->ArchitectureType, sizeof (OptEnt.Clid->ArchitectureType));\r
-\r
- if (Private->Nii != NULL) {\r
- //\r
- // If NII protocol exists, update DHCP option data\r
- //\r
- CopyMem (OptEnt.Clid->InterfaceName, Private->Nii->StringId, sizeof (OptEnt.Clid->InterfaceName));\r
- CvtNum (Private->Nii->MajorVer, OptEnt.Clid->UndiMajor, sizeof (OptEnt.Clid->UndiMajor));\r
- CvtNum (Private->Nii->MinorVer, OptEnt.Clid->UndiMinor, sizeof (OptEnt.Clid->UndiMinor));\r
- }\r
-\r
- Index++;\r
-\r
- return Index;\r
-}\r
-\r
-\r
-/**\r
- Discover the boot of service and initialize the vendor option if exists.\r
-\r
- @param Private Pointer to PxeBc private data.\r
- @param Type PxeBc option boot item type\r
- @param Layer PxeBc option boot item layer\r
- @param UseBis Use BIS or not\r
- @param DestIp Ip address for server\r
- @param IpCount The total count of the server ip address\r
- @param SrvList Server list\r
- @param IsDiscv Discover the vendor or not\r
- @param Reply The dhcp4 packet of Pxe reply\r
-\r
- @retval EFI_SUCCESS Operation succeeds.\r
- @retval EFI_OUT_OF_RESOURCES Allocate memory pool failed.\r
- @retval EFI_NOT_FOUND There is no vendor option exists.\r
- @retval EFI_TIMEOUT Send Pxe Discover time out.\r
-\r
-**/\r
-EFI_STATUS\r
-PxeBcDiscvBootService (\r
- IN PXEBC_PRIVATE_DATA * Private,\r
- IN UINT16 Type,\r
- IN UINT16 *Layer,\r
- IN BOOLEAN UseBis,\r
- IN EFI_IP_ADDRESS * DestIp,\r
- IN UINT16 IpCount,\r
- IN EFI_PXE_BASE_CODE_SRVLIST * SrvList,\r
- IN BOOLEAN IsDiscv,\r
- OUT EFI_DHCP4_PACKET * Reply OPTIONAL\r
- )\r
-{\r
- EFI_PXE_BASE_CODE_UDP_PORT Sport;\r
- EFI_PXE_BASE_CODE_MODE *Mode;\r
- EFI_DHCP4_PROTOCOL *Dhcp4;\r
- EFI_DHCP4_TRANSMIT_RECEIVE_TOKEN Token;\r
- BOOLEAN IsBCast;\r
- EFI_STATUS Status;\r
- UINT16 RepIndex;\r
- UINT16 SrvIndex;\r
- UINT16 TryIndex;\r
- EFI_DHCP4_LISTEN_POINT ListenPoint;\r
- EFI_DHCP4_PACKET *Response;\r
- EFI_DHCP4_PACKET_OPTION *OptList[PXEBC_DHCP4_MAX_OPTION_NUM];\r
- UINT32 OptCount;\r
- EFI_DHCP4_PACKET_OPTION *PxeOpt;\r
- PXEBC_OPTION_BOOT_ITEM *PxeBootItem;\r
- UINT8 VendorOptLen;\r
- EFI_DHCP4_HEADER *DhcpHeader;\r
- UINT32 Xid;\r
-\r
- Mode = Private->PxeBc.Mode;\r
- Dhcp4 = Private->Dhcp4;\r
- Status = EFI_SUCCESS;\r
-\r
- ZeroMem (&Token, sizeof (EFI_DHCP4_TRANSMIT_RECEIVE_TOKEN));\r
-\r
- if (DestIp == NULL) {\r
- Sport = PXEBC_DHCP4_S_PORT;\r
- IsBCast = TRUE;\r
- } else {\r
- Sport = PXEBC_BS_DISCOVER_PORT;\r
- IsBCast = FALSE;\r
- }\r
-\r
- if (!UseBis && Layer != NULL) {\r
- *Layer &= EFI_PXE_BASE_CODE_BOOT_LAYER_MASK;\r
- }\r
-\r
- OptCount = PxeBcBuildDhcpOptions (Private, OptList, FALSE);\r
-\r
- if (IsDiscv) {\r
- ASSERT (Layer != NULL);\r
- //\r
- // Add vendor option of PXE_BOOT_ITEM\r
- //\r
- VendorOptLen = (UINT8) ((sizeof (EFI_DHCP4_PACKET_OPTION) - 1) * 2 + sizeof (PXEBC_OPTION_BOOT_ITEM) + 1);\r
- OptList[OptCount] = AllocatePool (VendorOptLen);\r
- if (OptList[OptCount] == NULL) {\r
- return EFI_OUT_OF_RESOURCES;\r
- }\r
-\r
- OptList[OptCount]->OpCode = DHCP4_TAG_VENDOR;\r
- OptList[OptCount]->Length = (UINT8) (VendorOptLen - 2);\r
- PxeOpt = (EFI_DHCP4_PACKET_OPTION *) OptList[OptCount]->Data;\r
- PxeOpt->OpCode = PXEBC_VENDOR_TAG_BOOT_ITEM;\r
- PxeOpt->Length = (UINT8) sizeof (PXEBC_OPTION_BOOT_ITEM);\r
- PxeBootItem = (PXEBC_OPTION_BOOT_ITEM *) PxeOpt->Data;\r
- PxeBootItem->Type = HTONS (Type);\r
- PxeBootItem->Layer = HTONS (*Layer);\r
- PxeOpt->Data[PxeOpt->Length] = DHCP4_TAG_EOP;\r
-\r
- OptCount++;\r
- }\r
-\r
- Status = Dhcp4->Build (Dhcp4, &Private->SeedPacket, 0, NULL, OptCount, OptList, &Token.Packet);\r
-\r
- if (IsDiscv) {\r
- FreePool (OptList[OptCount - 1]);\r
- }\r
-\r
- if (EFI_ERROR (Status)) {\r
- return Status;\r
- }\r
-\r
- DhcpHeader = &Token.Packet->Dhcp4.Header;\r
- if (Mode->SendGUID) {\r
- if (EFI_ERROR (NetLibGetSystemGuid ((EFI_GUID *) DhcpHeader->ClientHwAddr))) {\r
- //\r
- // GUID not yet set - send all 0's to show not programable\r
- //\r
- DEBUG ((EFI_D_WARN, "PXE: Failed to read system GUID from the smbios table!\n"));\r
- ZeroMem (DhcpHeader->ClientHwAddr, sizeof (EFI_GUID));\r
- }\r
-\r
- DhcpHeader->HwAddrLen = (UINT8) sizeof (EFI_GUID);\r
- }\r
-\r
- Xid = NET_RANDOM (NetRandomInitSeed ());\r
- Token.Packet->Dhcp4.Header.Xid = HTONL(Xid);\r
- Token.Packet->Dhcp4.Header.Reserved = HTONS((UINT16) ((IsBCast) ? 0x8000 : 0));\r
- CopyMem (&Token.Packet->Dhcp4.Header.ClientAddr, &Private->StationIp, sizeof (EFI_IPv4_ADDRESS));\r
-\r
- Token.RemotePort = Sport;\r
-\r
- if (IsBCast) {\r
- SetMem (&Token.RemoteAddress, sizeof (EFI_IPv4_ADDRESS), 0xff);\r
- } else {\r
- CopyMem (&Token.RemoteAddress, DestIp, sizeof (EFI_IPv4_ADDRESS));\r
- }\r
-\r
- CopyMem (&Token.GatewayAddress, &Private->GatewayIp, sizeof (EFI_IPv4_ADDRESS));\r
-\r
- if (!IsBCast) {\r
- Token.ListenPointCount = 1;\r
- Token.ListenPoints = &ListenPoint;\r
- Token.ListenPoints[0].ListenPort = PXEBC_BS_DISCOVER_PORT;\r
- CopyMem (&Token.ListenPoints[0].ListenAddress, &Private->StationIp, sizeof(EFI_IPv4_ADDRESS));\r
- CopyMem (&Token.ListenPoints[0].SubnetMask, &Private->SubnetMask, sizeof(EFI_IPv4_ADDRESS));\r
- }\r
- //\r
- // Send Pxe Discover\r
- //\r
- for (TryIndex = 1; TryIndex <= PXEBC_BOOT_REQUEST_RETRIES; TryIndex++) {\r
-\r
- Token.TimeoutValue = (UINT16) (PXEBC_BOOT_REQUEST_TIMEOUT * TryIndex);\r
- Token.Packet->Dhcp4.Header.Seconds = (UINT16) (PXEBC_BOOT_REQUEST_TIMEOUT * (TryIndex - 1));\r
-\r
- Status = Dhcp4->TransmitReceive (Dhcp4, &Token);\r
-\r
- if (Token.Status != EFI_TIMEOUT) {\r
- break;\r
- }\r
- }\r
-\r
- if (TryIndex > PXEBC_BOOT_REQUEST_RETRIES) {\r
- //\r
- // No server response our PXE request\r
- //\r
- Status = EFI_TIMEOUT;\r
- }\r
-\r
- if (!EFI_ERROR (Status)) {\r
- //\r
- // Find Pxe Reply\r
- //\r
- RepIndex = 0;\r
- SrvIndex = 0;\r
- Response = Token.ResponseList;\r
-\r
- while (RepIndex < Token.ResponseCount) {\r
- if (Response->Length > PXEBC_DHCP4_MAX_PACKET_SIZE) {\r
- SrvIndex = 0;\r
- RepIndex++;\r
- Response = (EFI_DHCP4_PACKET *) ((UINT8 *) Response + Response->Size);\r
- continue;\r
- }\r
-\r
- while (SrvIndex < IpCount) {\r
-\r
- if (SrvList[SrvIndex].AcceptAnyResponse) {\r
- break;\r
- }\r
-\r
- if ((SrvList[SrvIndex].Type == Type) && EFI_IP4_EQUAL (&(Response->Dhcp4.Header.ServerAddr), &(Private->ServerIp))) {\r
- break;\r
- }\r
-\r
- SrvIndex++;\r
- }\r
-\r
- if ((IpCount != SrvIndex) || (IpCount == 0)) {\r
- break;\r
- }\r
-\r
- SrvIndex = 0;\r
- RepIndex++;\r
- Response = (EFI_DHCP4_PACKET *) ((UINT8 *) Response + Response->Size);\r
- }\r
-\r
- if (RepIndex < Token.ResponseCount) {\r
-\r
- if (Reply != NULL) {\r
- Status = PxeBcCopyEfiDhcp4Packet (Reply, Response);\r
- if (EFI_ERROR(Status)) {\r
- goto ON_EXIT;\r
- }\r
- }\r
-\r
- if (IsDiscv) {\r
- CopyMem (&(Mode->PxeDiscover), &(Token.Packet->Dhcp4), Token.Packet->Length);\r
- Mode->PxeDiscoverValid = TRUE;\r
-\r
- CopyMem (Mode->PxeReply.Raw, &Response->Dhcp4, Response->Length);\r
- Mode->PxeReplyReceived = TRUE;\r
- }\r
- } else {\r
- Status = EFI_NOT_FOUND;\r
- }\r
- }\r
-\r
-ON_EXIT:\r
- //\r
- // free the responselist\r
- //\r
- if (Token.ResponseList != NULL) {\r
- FreePool (Token.ResponseList);\r
- }\r
- //\r
- // Free the dhcp packet\r
- //\r
- if (Token.Packet != NULL) {\r
- FreePool (Token.Packet);\r
- }\r
-\r
- return Status;\r
-}\r
-\r
-\r
-/**\r
- Parse interested dhcp options.\r
-\r
- @param Buffer Pointer to the dhcp options packet.\r
- @param Length The length of the dhcp options.\r
- @param OptTag The option OpCode.\r
-\r
- @return NULL if the buffer length is 0 and OpCode is not\r
- DHCP4_TAG_EOP, or the pointer to the buffer.\r
-\r
-**/\r
-EFI_DHCP4_PACKET_OPTION *\r
-PxeBcParseExtendOptions (\r
- IN UINT8 *Buffer,\r
- IN UINT32 Length,\r
- IN UINT8 OptTag\r
- )\r
-{\r
- EFI_DHCP4_PACKET_OPTION *Option;\r
- UINT32 Offset;\r
-\r
- Option = (EFI_DHCP4_PACKET_OPTION *) Buffer;\r
- Offset = 0;\r
-\r
- while (Offset < Length && Option->OpCode != DHCP4_TAG_EOP) {\r
-\r
- if (Option->OpCode == OptTag) {\r
-\r
- return Option;\r
- }\r
-\r
- if (Option->OpCode == DHCP4_TAG_PAD) {\r
- Offset++;\r
- } else {\r
- Offset += Option->Length + 2;\r
- }\r
-\r
- Option = (EFI_DHCP4_PACKET_OPTION *) (Buffer + Offset);\r
- }\r
-\r
- return NULL;\r
-}\r
-\r
-\r
-/**\r
- This function is to parse and check vendor options.\r
-\r
- @param Dhcp4Option Pointer to dhcp options\r
- @param VendorOption Pointer to vendor options\r
-\r
- @return TRUE if valid for vendor options, or FALSE.\r
-\r
-**/\r
-BOOLEAN\r
-PxeBcParseVendorOptions (\r
- IN EFI_DHCP4_PACKET_OPTION *Dhcp4Option,\r
- IN PXEBC_VENDOR_OPTION *VendorOption\r
- )\r
-{\r
- UINT32 *BitMap;\r
- UINT8 VendorOptionLen;\r
- EFI_DHCP4_PACKET_OPTION *PxeOption;\r
- UINT8 Offset;\r
-\r
- BitMap = VendorOption->BitMap;\r
- VendorOptionLen = Dhcp4Option->Length;\r
- PxeOption = (EFI_DHCP4_PACKET_OPTION *) &Dhcp4Option->Data[0];\r
- Offset = 0;\r
-\r
- while ((Offset < VendorOptionLen) && (PxeOption->OpCode != DHCP4_TAG_EOP)) {\r
- //\r
- // Parse every Vendor Option and set its BitMap\r
- //\r
- switch (PxeOption->OpCode) {\r
-\r
- case PXEBC_VENDOR_TAG_MTFTP_IP:\r
-\r
- CopyMem (&VendorOption->MtftpIp, PxeOption->Data, sizeof (EFI_IPv4_ADDRESS));\r
- break;\r
-\r
- case PXEBC_VENDOR_TAG_MTFTP_CPORT:\r
-\r
- CopyMem (&VendorOption->MtftpCPort, PxeOption->Data, sizeof (VendorOption->MtftpCPort));\r
- break;\r
-\r
- case PXEBC_VENDOR_TAG_MTFTP_SPORT:\r
-\r
- CopyMem (&VendorOption->MtftpSPort, PxeOption->Data, sizeof (VendorOption->MtftpSPort));\r
- break;\r
-\r
- case PXEBC_VENDOR_TAG_MTFTP_TIMEOUT:\r
-\r
- VendorOption->MtftpTimeout = *PxeOption->Data;\r
- break;\r
-\r
- case PXEBC_VENDOR_TAG_MTFTP_DELAY:\r
-\r
- VendorOption->MtftpDelay = *PxeOption->Data;\r
- break;\r
-\r
- case PXEBC_VENDOR_TAG_DISCOVER_CTRL:\r
-\r
- VendorOption->DiscoverCtrl = *PxeOption->Data;\r
- break;\r
-\r
- case PXEBC_VENDOR_TAG_DISCOVER_MCAST:\r
-\r
- CopyMem (&VendorOption->DiscoverMcastIp, PxeOption->Data, sizeof (EFI_IPv4_ADDRESS));\r
- break;\r
-\r
- case PXEBC_VENDOR_TAG_BOOT_SERVERS:\r
-\r
- VendorOption->BootSvrLen = PxeOption->Length;\r
- VendorOption->BootSvr = (PXEBC_BOOT_SVR_ENTRY *) PxeOption->Data;\r
- break;\r
-\r
- case PXEBC_VENDOR_TAG_BOOT_MENU:\r
-\r
- VendorOption->BootMenuLen = PxeOption->Length;\r
- VendorOption->BootMenu = (PXEBC_BOOT_MENU_ENTRY *) PxeOption->Data;\r
- break;\r
-\r
- case PXEBC_VENDOR_TAG_MENU_PROMPT:\r
-\r
- VendorOption->MenuPromptLen = PxeOption->Length;\r
- VendorOption->MenuPrompt = (PXEBC_MENU_PROMPT *) PxeOption->Data;\r
- break;\r
-\r
- case PXEBC_VENDOR_TAG_MCAST_ALLOC:\r
-\r
- CopyMem (&VendorOption->McastIpBase, PxeOption->Data, sizeof (EFI_IPv4_ADDRESS));\r
- CopyMem (&VendorOption->McastIpBlock, PxeOption->Data + 4, sizeof (VendorOption->McastIpBlock));\r
- CopyMem (&VendorOption->McastIpRange, PxeOption->Data + 6, sizeof (VendorOption->McastIpRange));\r
- break;\r
-\r
- case PXEBC_VENDOR_TAG_CREDENTIAL_TYPES:\r
-\r
- VendorOption->CredTypeLen = PxeOption->Length;\r
- VendorOption->CredType = (UINT32 *) PxeOption->Data;\r
- break;\r
-\r
- case PXEBC_VENDOR_TAG_BOOT_ITEM:\r
-\r
- CopyMem (&VendorOption->BootSrvType, PxeOption->Data, sizeof (VendorOption->BootSrvType));\r
- CopyMem (&VendorOption->BootSrvLayer, PxeOption->Data + 2, sizeof (VendorOption->BootSrvLayer));\r
- break;\r
- }\r
-\r
- SET_VENDOR_OPTION_BIT_MAP (BitMap, PxeOption->OpCode);\r
-\r
- if (PxeOption->OpCode == DHCP4_TAG_PAD) {\r
- Offset++;\r
- } else {\r
- Offset = (UINT8) (Offset + PxeOption->Length + 2);\r
- }\r
-\r
- PxeOption = (EFI_DHCP4_PACKET_OPTION *) (Dhcp4Option->Data + Offset);\r
- }\r
-\r
- //\r
- // FixMe, return falas if invalid of any vendor option\r
- //\r
-\r
- return TRUE;\r
-}\r
-\r
-\r
-/**\r
- This function display boot item detail.\r
-\r
- If the length of the boot item string over 70 Char, just display 70 Char.\r
-\r
- @param Str Pointer to a string (boot item string).\r
- @param Len The length of string.\r
-\r
-**/\r
-VOID\r
-PxeBcDisplayBootItem (\r
- IN UINT8 *Str,\r
- IN UINT8 Len\r
- )\r
-{\r
- UINT8 Tmp;\r
-\r
- Len = (UINT8) MIN (70, Len);\r
- Tmp = Str[Len];\r
- Str[Len] = 0;\r
- AsciiPrint ("%a \n", Str);\r
- Str[Len] = Tmp;\r
-}\r
-\r
-\r
-/**\r
- Choose the boot prompt.\r
-\r
- @param Private Pointer to PxeBc private data.\r
-\r
- @retval EFI_SUCCESS Select boot prompt done.\r
- @retval EFI_TIMEOUT Select boot prompt time out.\r
- @retval EFI_NOT_FOUND The proxy offer is not Pxe10.\r
- @retval EFI_ABORTED User cancel the operation.\r
- @retval EFI_NOT_READY Read the input key from the keybroad has not finish.\r
-\r
-**/\r
-EFI_STATUS\r
-PxeBcSelectBootPrompt (\r
- IN PXEBC_PRIVATE_DATA *Private\r
- )\r
-{\r
- PXEBC_CACHED_DHCP4_PACKET *Packet;\r
- PXEBC_VENDOR_OPTION *VendorOpt;\r
- EFI_EVENT TimeoutEvent;\r
- EFI_EVENT DescendEvent;\r
- EFI_INPUT_KEY InputKey;\r
- EFI_STATUS Status;\r
- UINT8 Timeout;\r
- UINT8 *Prompt;\r
- UINT8 PromptLen;\r
- INT32 SecCol;\r
- INT32 SecRow;\r
-\r
- TimeoutEvent = NULL;\r
- DescendEvent = NULL;\r
-\r
- if (Private->PxeBc.Mode->ProxyOfferReceived) {\r
-\r
- Packet = &Private->ProxyOffer;\r
- } else {\r
-\r
- Packet = &Private->Dhcp4Ack;\r
- }\r
-\r
- if (Packet->OfferType != DHCP4_PACKET_TYPE_PXE10) {\r
- return EFI_NOT_FOUND;\r
- }\r
-\r
- VendorOpt = &Packet->PxeVendorOption;\r
- //\r
- // According to the PXE specification 2.1, Table 2-1 PXE DHCP Options (Full\r
- // List), we must not consider a boot prompt or boot menu if all of the\r
- // following hold:\r
- // - the PXE_DISCOVERY_CONTROL PXE tag is present inside the Vendor Options\r
- // (=43) DHCP tag, and\r
- // - the PXE_DISCOVERY_CONTROL PXE tag has bit 3 set, and\r
- // - a boot file name has been presented with DHCP option 67.\r
- //\r
- if (IS_DISABLE_PROMPT_MENU (VendorOpt->DiscoverCtrl) &&\r
- Packet->Dhcp4Option[PXEBC_DHCP4_TAG_INDEX_BOOTFILE] != NULL) {\r
- return EFI_ABORTED;\r
- }\r
-\r
- if (!IS_VALID_BOOT_PROMPT (VendorOpt->BitMap)) {\r
- return EFI_SUCCESS;\r
- }\r
-\r
- Timeout = VendorOpt->MenuPrompt->Timeout;\r
- Prompt = VendorOpt->MenuPrompt->Prompt;\r
- PromptLen = (UINT8) (VendorOpt->MenuPromptLen - 1);\r
-\r
- if (Timeout == 0) {\r
- return EFI_SUCCESS;\r
- }\r
-\r
- if (Timeout == 255) {\r
- return EFI_TIMEOUT;\r
- }\r
-\r
- Status = gBS->CreateEvent (\r
- EVT_TIMER,\r
- TPL_CALLBACK,\r
- NULL,\r
- NULL,\r
- &TimeoutEvent\r
- );\r
-\r
- if (EFI_ERROR (Status)) {\r
- return Status;\r
- }\r
-\r
- Status = gBS->SetTimer (\r
- TimeoutEvent,\r
- TimerRelative,\r
- MultU64x32 (Timeout, TICKS_PER_SECOND)\r
- );\r
-\r
- if (EFI_ERROR (Status)) {\r
- goto ON_EXIT;\r
- }\r
-\r
- Status = gBS->CreateEvent (\r
- EVT_TIMER,\r
- TPL_CALLBACK,\r
- NULL,\r
- NULL,\r
- &DescendEvent\r
- );\r
-\r
- if (EFI_ERROR (Status)) {\r
- goto ON_EXIT;\r
- }\r
-\r
- Status = gBS->SetTimer (\r
- DescendEvent,\r
- TimerPeriodic,\r
- TICKS_PER_SECOND\r
- );\r
-\r
- if (EFI_ERROR (Status)) {\r
- goto ON_EXIT;\r
- }\r
-\r
- SecCol = gST->ConOut->Mode->CursorColumn;\r
- SecRow = gST->ConOut->Mode->CursorRow;\r
-\r
- PxeBcDisplayBootItem (Prompt, PromptLen);\r
-\r
- gST->ConOut->SetCursorPosition (gST->ConOut, SecCol + PromptLen, SecRow);\r
- AsciiPrint ("(%d) ", Timeout--);\r
-\r
- while (EFI_ERROR (gBS->CheckEvent (TimeoutEvent))) {\r
-\r
- if (!EFI_ERROR (gBS->CheckEvent (DescendEvent))) {\r
- gST->ConOut->SetCursorPosition (gST->ConOut, SecCol + PromptLen, SecRow);\r
- AsciiPrint ("(%d) ", Timeout--);\r
- }\r
-\r
- if (gST->ConIn->ReadKeyStroke (gST->ConIn, &InputKey) == EFI_NOT_READY) {\r
-\r
- gBS->Stall (10 * TICKS_PER_MS);\r
- continue;\r
- }\r
-\r
- if (InputKey.ScanCode == 0) {\r
-\r
- switch (InputKey.UnicodeChar) {\r
- case CTRL ('c'):\r
- Status = EFI_ABORTED;\r
- break;\r
-\r
- case CTRL ('m'):\r
- case 'm':\r
- case 'M':\r
- Status = EFI_TIMEOUT;\r
- break;\r
-\r
- default:\r
- continue;\r
- }\r
- } else {\r
-\r
- switch (InputKey.ScanCode) {\r
- case SCAN_F8:\r
- Status = EFI_TIMEOUT;\r
- break;\r
-\r
- case SCAN_ESC:\r
- Status = EFI_ABORTED;\r
- break;\r
-\r
- default:\r
- continue;\r
- }\r
- }\r
-\r
- break;\r
- }\r
-\r
- gST->ConOut->SetCursorPosition (gST->ConOut, 0 , SecRow + 1);\r
-\r
-ON_EXIT:\r
-\r
- if (DescendEvent != NULL) {\r
- gBS->CloseEvent (DescendEvent);\r
- }\r
-\r
- if (TimeoutEvent != NULL) {\r
- gBS->CloseEvent (TimeoutEvent);\r
- }\r
-\r
- return Status;\r
-}\r
-\r
-\r
-/**\r
- Select the boot menu.\r
-\r
- @param Private Pointer to PxeBc private data.\r
- @param Type The type of the menu.\r
- @param UseDefaultItem Use default item or not.\r
-\r
- @retval EFI_ABORTED User cancel operation.\r
- @retval EFI_SUCCESS Select the boot menu success.\r
- @retval EFI_NOT_READY Read the input key from the keybroad has not finish.\r
-\r
-**/\r
-EFI_STATUS\r
-PxeBcSelectBootMenu (\r
- IN PXEBC_PRIVATE_DATA *Private,\r
- OUT UINT16 *Type,\r
- IN BOOLEAN UseDefaultItem\r
- )\r
-{\r
- PXEBC_CACHED_DHCP4_PACKET *Packet;\r
- PXEBC_VENDOR_OPTION *VendorOpt;\r
- EFI_INPUT_KEY InputKey;\r
- UINT8 MenuSize;\r
- UINT8 MenuNum;\r
- INT32 TopRow;\r
- UINT16 Select;\r
- UINT16 LastSelect;\r
- UINT8 Index;\r
- BOOLEAN Finish;\r
- CHAR8 Blank[70];\r
- PXEBC_BOOT_MENU_ENTRY *MenuItem;\r
- PXEBC_BOOT_MENU_ENTRY *MenuArray[PXEBC_MAX_MENU_NUM];\r
-\r
- Finish = FALSE;\r
- Select = 0;\r
- Index = 0;\r
- *Type = 0;\r
-\r
- if (Private->PxeBc.Mode->ProxyOfferReceived) {\r
-\r
- Packet = &Private->ProxyOffer;\r
- } else {\r
-\r
- Packet = &Private->Dhcp4Ack;\r
- }\r
-\r
- ASSERT (Packet->OfferType == DHCP4_PACKET_TYPE_PXE10);\r
-\r
- VendorOpt = &Packet->PxeVendorOption;\r
-\r
- if (!IS_VALID_BOOT_MENU (VendorOpt->BitMap)) {\r
- return EFI_SUCCESS;\r
- }\r
-\r
- SetMem (Blank, sizeof(Blank), ' ');\r
-\r
- MenuSize = VendorOpt->BootMenuLen;\r
- MenuItem = VendorOpt->BootMenu;\r
-\r
- if (MenuSize == 0) {\r
- return EFI_NOT_READY;\r
- }\r
-\r
- while (MenuSize > 0) {\r
- MenuArray[Index++] = MenuItem;\r
- MenuSize = (UINT8) (MenuSize - (MenuItem->DescLen + 3));\r
- MenuItem = (PXEBC_BOOT_MENU_ENTRY *) ((UINT8 *) MenuItem + MenuItem->DescLen + 3);\r
- if (Index >= PXEBC_MAX_MENU_NUM) {\r
- break;\r
- }\r
- }\r
-\r
- if (UseDefaultItem) {\r
- *Type = MenuArray[0]->Type;\r
- *Type = NTOHS (*Type);\r
- return EFI_SUCCESS;\r
- }\r
-\r
- MenuNum = Index;\r
-\r
- for (Index = 0; Index < MenuNum; Index++) {\r
- PxeBcDisplayBootItem (MenuArray[Index]->DescStr, MenuArray[Index]->DescLen);\r
- }\r
-\r
- TopRow = gST->ConOut->Mode->CursorRow - MenuNum;\r
-\r
- do {\r
- ASSERT (Select < PXEBC_MAX_MENU_NUM);\r
- //\r
- // highlight selected row\r
- //\r
- gST->ConOut->SetAttribute (gST->ConOut, EFI_TEXT_ATTR (EFI_BLACK, EFI_LIGHTGRAY));\r
- gST->ConOut->SetCursorPosition (gST->ConOut, 0, TopRow + Select);\r
- Blank[MenuArray[Select]->DescLen] = 0;\r
- AsciiPrint ("%a\r", Blank);\r
- PxeBcDisplayBootItem (MenuArray[Select]->DescStr, MenuArray[Select]->DescLen);\r
- gST->ConOut->SetCursorPosition (gST->ConOut, 0, TopRow + MenuNum);\r
- LastSelect = Select;\r
-\r
- while (gST->ConIn->ReadKeyStroke (gST->ConIn, &InputKey) == EFI_NOT_READY) {\r
- gBS->Stall (10 * TICKS_PER_MS);\r
- }\r
-\r
- if (InputKey.ScanCode == 0) {\r
- switch (InputKey.UnicodeChar) {\r
- case CTRL ('c'):\r
- InputKey.ScanCode = SCAN_ESC;\r
- break;\r
-\r
- case CTRL ('j'): /* linefeed */\r
- case CTRL ('m'): /* return */\r
- Finish = TRUE;\r
- break;\r
-\r
- case CTRL ('i'): /* tab */\r
- case ' ':\r
- case 'd':\r
- case 'D':\r
- InputKey.ScanCode = SCAN_DOWN;\r
- break;\r
-\r
- case CTRL ('h'): /* backspace */\r
- case 'u':\r
- case 'U':\r
- InputKey.ScanCode = SCAN_UP;\r
- break;\r
-\r
- default:\r
- InputKey.ScanCode = 0;\r
- }\r
- }\r
-\r
- switch (InputKey.ScanCode) {\r
- case SCAN_LEFT:\r
- case SCAN_UP:\r
- if (Select > 0) {\r
- --Select;\r
- }\r
-\r
- break;\r
-\r
- case SCAN_DOWN:\r
- case SCAN_RIGHT:\r
- if (++Select == MenuNum) {\r
- --Select;\r
- }\r
-\r
- break;\r
-\r
- case SCAN_PAGE_UP:\r
- case SCAN_HOME:\r
- Select = 0;\r
- break;\r
-\r
- case SCAN_PAGE_DOWN:\r
- case SCAN_END:\r
- Select = (UINT16) (MenuNum - 1);\r
- break;\r
-\r
- case SCAN_ESC:\r
- return EFI_ABORTED;\r
- }\r
-\r
- /* unhighlight last selected row */\r
- gST->ConOut->SetAttribute (gST->ConOut, EFI_TEXT_ATTR (EFI_LIGHTGRAY, EFI_BLACK));\r
- gST->ConOut->SetCursorPosition (gST->ConOut, 0, TopRow + LastSelect);\r
- Blank[MenuArray[LastSelect]->DescLen] = 0;\r
- AsciiPrint ("%a\r", Blank);\r
- PxeBcDisplayBootItem (MenuArray[LastSelect]->DescStr, MenuArray[LastSelect]->DescLen);\r
- gST->ConOut->SetCursorPosition (gST->ConOut, 0, TopRow + MenuNum);\r
- } while (!Finish);\r
-\r
- ASSERT (Select < PXEBC_MAX_MENU_NUM);\r
-\r
- //\r
- // Swap the byte order\r
- //\r
- CopyMem (Type, &MenuArray[Select]->Type, sizeof (UINT16));\r
- *Type = NTOHS (*Type);\r
-\r
- return EFI_SUCCESS;\r
-}\r
-\r
projects / mirror_edk2.git / blobdiff