From 76d05dc695b06c4e987bb8078f78032441e1430c Mon Sep 17 00:00:00 2001 From: Greg Hartman Date: Wed, 23 Nov 2016 15:51:27 -0800 Subject: Clean checkin of the 6.03 code BUG: 32495852 Change-Id: I5038a3bb41e217380c1188463daec07b1e9b6b48 --- efi/pxe.c | 175 ++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 175 insertions(+) create mode 100644 efi/pxe.c (limited to 'efi/pxe.c') diff --git a/efi/pxe.c b/efi/pxe.c new file mode 100644 index 0000000..0c9fb06 --- /dev/null +++ b/efi/pxe.c @@ -0,0 +1,175 @@ +/* + * Copyright 2013-2014 Intel Corporation - All Rights Reserved + */ + +#include +#include +#include "efi.h" +#include "net.h" +#include "fs/pxe/pxe.h" + +const struct url_scheme url_schemes[] = { + { "tftp", tftp_open, 0 }, + { "http", http_open, O_DIRECTORY }, + { "ftp", ftp_open, O_DIRECTORY }, + { NULL, NULL, 0 }, +}; + +/** + * Network stack-specific initialization + */ +void net_core_init(void) +{ + http_bake_cookies(); +} + +void pxe_init_isr(void) {} +void gpxe_init(void) {} +void pxe_idle_init(void) {} + +int reset_pxe(void) +{ + return 0; +} + +#define DNS_MAX_SERVERS 4 /* Max no of DNS servers */ +uint32_t dns_server[DNS_MAX_SERVERS] = {0, }; + +__export uint32_t dns_resolv(const char *name) +{ + /* + * Return failure on an empty input... this can happen during + * some types of URL parsing, and this is the easiest place to + * check for it. + */ + if (!name || !*name) + return 0; + + return 0; +} + +int pxe_init(bool quiet) +{ + EFI_HANDLE *handles; + EFI_STATUS status; + UINTN nr_handles; + + status = LibLocateHandle(ByProtocol, &PxeBaseCodeProtocol, + NULL, &nr_handles, &handles); + if (status != EFI_SUCCESS) { + if (!quiet) + Print(L"No PXE Base Code Protocol\n"); + return -1; + } + + return 0; +} + +#define EDHCP_BUF_LEN 8192 + +struct embedded_dhcp_options { + uint32_t magic[4]; + uint32_t bdhcp_len; + uint32_t adhcp_len; + uint32_t buffer_size; + uint32_t reserved; + uint8_t dhcp_data[EDHCP_BUF_LEN]; +} __attribute__((aligned(16))); + +struct embedded_dhcp_options embedded_dhcp_options = +{ + .magic[0] = 0x2a171ead, + .magic[1] = 0x0600e65e, + .magic[2] = 0x4025a4e4, + .magic[3] = 0x42388fc8, + .bdhcp_len = 0, + .adhcp_len = 0, + .buffer_size = EDHCP_BUF_LEN, +}; + +void net_parse_dhcp(void) +{ + EFI_PXE_BASE_CODE_MODE *mode; + EFI_PXE_BASE_CODE *bc; + unsigned int pkt_len = sizeof(EFI_PXE_BASE_CODE_PACKET); + EFI_STATUS status; + EFI_HANDLE *handles = NULL; + UINTN nr_handles = 0; + uint8_t hardlen; + uint32_t ip; + char dst[256]; + + status = LibLocateHandle(ByProtocol, &PxeBaseCodeProtocol, + NULL, &nr_handles, &handles); + if (status != EFI_SUCCESS) + return; + + /* Probably want to use IPv4 protocol to decide which handle to use */ + status = uefi_call_wrapper(BS->HandleProtocol, 3, handles[0], + &PxeBaseCodeProtocol, (void **)&bc); + if (status != EFI_SUCCESS) { + Print(L"Failed to lookup PxeBaseCodeProtocol\n"); + } + + mode = bc->Mode; + + /* + * Parse any "before" hardcoded options + */ + parse_dhcp_options(embedded_dhcp_options.dhcp_data, + embedded_dhcp_options.bdhcp_len, 0); + + /* + * Get the DHCP client identifiers (query info 1) + */ + Print(L"Getting cached packet "); + parse_dhcp(&mode->DhcpDiscover.Dhcpv4, pkt_len); + /* + * We don't use flags from the request packet, so + * this is a good time to initialize DHCPMagic... + * Initialize it to 1 meaning we will accept options found; + * in earlier versions of PXELINUX bit 0 was used to indicate + * we have found option 208 with the appropriate magic number; + * we no longer require that, but MAY want to re-introduce + * it in the future for vendor encapsulated options. + */ + *(char *)&DHCPMagic = 1; + + /* + * Get the BOOTP/DHCP packet that brought us file (and an IP + * address). This lives in the DHCPACK packet (query info 2) + */ + parse_dhcp(&mode->DhcpAck.Dhcpv4, pkt_len); + /* + * Save away MAC address (assume this is in query info 2. If this + * turns out to be problematic it might be better getting it from + * the query info 1 packet + */ + hardlen = mode->DhcpAck.Dhcpv4.BootpHwAddrLen; + MAC_len = hardlen > 16 ? 0 : hardlen; + MAC_type = mode->DhcpAck.Dhcpv4.BootpHwType; + memcpy(MAC, mode->DhcpAck.Dhcpv4.BootpHwAddr, MAC_len); + + /* + * Get the boot file and other info. This lives in the CACHED_REPLY + * packet (query info 3) + */ + parse_dhcp(&mode->PxeReply.Dhcpv4, pkt_len); + Print(L"\n"); + + /* + * Parse any "after" hardcoded options + */ + parse_dhcp_options(embedded_dhcp_options.dhcp_data + + embedded_dhcp_options.bdhcp_len, + embedded_dhcp_options.adhcp_len, 0); + + ip = IPInfo.myip; + sprintf(dst, "%u.%u.%u.%u", + ((const uint8_t *)&ip)[0], + ((const uint8_t *)&ip)[1], + ((const uint8_t *)&ip)[2], + ((const uint8_t *)&ip)[3]); + + Print(L"My IP is %a\n", dst); +} -- cgit v1.2.3