diff options
Diffstat (limited to 'roms/u-boot/net/bootp.c')
| -rw-r--r-- | roms/u-boot/net/bootp.c | 1137 |
1 files changed, 1137 insertions, 0 deletions
diff --git a/roms/u-boot/net/bootp.c b/roms/u-boot/net/bootp.c new file mode 100644 index 000000000..163af41e9 --- /dev/null +++ b/roms/u-boot/net/bootp.c @@ -0,0 +1,1137 @@ +/* + * Based on LiMon - BOOTP. + * + * Copyright 1994, 1995, 2000 Neil Russell. + * (See License) + * Copyright 2000 Roland Borde + * Copyright 2000 Paolo Scaffardi + * Copyright 2000-2004 Wolfgang Denk, wd@denx.de + */ + +#include <common.h> +#include <bootstage.h> +#include <command.h> +#include <env.h> +#include <efi_loader.h> +#include <log.h> +#include <net.h> +#include <rand.h> +#include <uuid.h> +#include <linux/delay.h> +#include <net/tftp.h> +#include "bootp.h" +#ifdef CONFIG_LED_STATUS +#include <status_led.h> +#endif +#ifdef CONFIG_BOOTP_RANDOM_DELAY +#include "net_rand.h" +#endif + +#define BOOTP_VENDOR_MAGIC 0x63825363 /* RFC1048 Magic Cookie */ + +/* + * The timeout for the initial BOOTP/DHCP request used to be described by a + * counter of fixed-length timeout periods. TIMEOUT_COUNT represents + * that counter + * + * Now that the timeout periods are variable (exponential backoff and retry) + * we convert the timeout count to the absolute time it would have take to + * execute that many retries, and keep sending retry packets until that time + * is reached. + */ +#ifndef CONFIG_NET_RETRY_COUNT +# define TIMEOUT_COUNT 5 /* # of timeouts before giving up */ +#else +# define TIMEOUT_COUNT (CONFIG_NET_RETRY_COUNT) +#endif +#define TIMEOUT_MS ((3 + (TIMEOUT_COUNT * 5)) * 1000) + +#define PORT_BOOTPS 67 /* BOOTP server UDP port */ +#define PORT_BOOTPC 68 /* BOOTP client UDP port */ + +#ifndef CONFIG_DHCP_MIN_EXT_LEN /* minimal length of extension list */ +#define CONFIG_DHCP_MIN_EXT_LEN 64 +#endif + +#ifndef CONFIG_BOOTP_ID_CACHE_SIZE +#define CONFIG_BOOTP_ID_CACHE_SIZE 4 +#endif + +u32 bootp_ids[CONFIG_BOOTP_ID_CACHE_SIZE]; +unsigned int bootp_num_ids; +int bootp_try; +ulong bootp_start; +ulong bootp_timeout; +char net_nis_domain[32] = {0,}; /* Our NIS domain */ +char net_hostname[32] = {0,}; /* Our hostname */ +char net_root_path[64] = {0,}; /* Our bootpath */ + +static ulong time_taken_max; + +#if defined(CONFIG_CMD_DHCP) +static dhcp_state_t dhcp_state = INIT; +static u32 dhcp_leasetime; +static struct in_addr dhcp_server_ip; +static u8 dhcp_option_overload; +#define OVERLOAD_FILE 1 +#define OVERLOAD_SNAME 2 +static void dhcp_handler(uchar *pkt, unsigned dest, struct in_addr sip, + unsigned src, unsigned len); + +/* For Debug */ +#if 0 +static char *dhcpmsg2str(int type) +{ + switch (type) { + case 1: return "DHCPDISCOVER"; break; + case 2: return "DHCPOFFER"; break; + case 3: return "DHCPREQUEST"; break; + case 4: return "DHCPDECLINE"; break; + case 5: return "DHCPACK"; break; + case 6: return "DHCPNACK"; break; + case 7: return "DHCPRELEASE"; break; + default: return "UNKNOWN/INVALID MSG TYPE"; break; + } +} +#endif +#endif + +static void bootp_add_id(ulong id) +{ + if (bootp_num_ids >= ARRAY_SIZE(bootp_ids)) { + size_t size = sizeof(bootp_ids) - sizeof(id); + + memmove(bootp_ids, &bootp_ids[1], size); + bootp_ids[bootp_num_ids - 1] = id; + } else { + bootp_ids[bootp_num_ids] = id; + bootp_num_ids++; + } +} + +static bool bootp_match_id(ulong id) +{ + unsigned int i; + + for (i = 0; i < bootp_num_ids; i++) + if (bootp_ids[i] == id) + return true; + + return false; +} + +static int check_reply_packet(uchar *pkt, unsigned dest, unsigned src, + unsigned len) +{ + struct bootp_hdr *bp = (struct bootp_hdr *)pkt; + int retval = 0; + + if (dest != PORT_BOOTPC || src != PORT_BOOTPS) + retval = -1; + else if (len < sizeof(struct bootp_hdr) - OPT_FIELD_SIZE) + retval = -2; + else if (bp->bp_op != OP_BOOTREPLY) + retval = -3; + else if (bp->bp_htype != HWT_ETHER) + retval = -4; + else if (bp->bp_hlen != HWL_ETHER) + retval = -5; + else if (!bootp_match_id(net_read_u32(&bp->bp_id))) + retval = -6; + else if (memcmp(bp->bp_chaddr, net_ethaddr, HWL_ETHER) != 0) + retval = -7; + + debug("Filtering pkt = %d\n", retval); + + return retval; +} + +static void store_bootp_params(struct bootp_hdr *bp) +{ +#if !defined(CONFIG_BOOTP_SERVERIP) + struct in_addr tmp_ip; + bool overwrite_serverip = true; + +#if defined(CONFIG_BOOTP_PREFER_SERVERIP) + overwrite_serverip = false; +#endif + + net_copy_ip(&tmp_ip, &bp->bp_siaddr); + if (tmp_ip.s_addr != 0 && (overwrite_serverip || !net_server_ip.s_addr)) + net_copy_ip(&net_server_ip, &bp->bp_siaddr); + memcpy(net_server_ethaddr, + ((struct ethernet_hdr *)net_rx_packet)->et_src, 6); + if ( +#if defined(CONFIG_CMD_DHCP) + !(dhcp_option_overload & OVERLOAD_FILE) && +#endif + (strlen(bp->bp_file) > 0) && + !net_boot_file_name_explicit) { + copy_filename(net_boot_file_name, bp->bp_file, + sizeof(net_boot_file_name)); + } + + debug("net_boot_file_name: %s\n", net_boot_file_name); + + /* Propagate to environment: + * don't delete exising entry when BOOTP / DHCP reply does + * not contain a new value + */ + if (*net_boot_file_name) + env_set("bootfile", net_boot_file_name); +#endif +} + +/* + * Copy parameters of interest from BOOTP_REPLY/DHCP_OFFER packet + */ +static void store_net_params(struct bootp_hdr *bp) +{ +#if !defined(CONFIG_SERVERIP_FROM_PROXYDHCP) + store_bootp_params(bp); +#endif + net_copy_ip(&net_ip, &bp->bp_yiaddr); +} + +static int truncate_sz(const char *name, int maxlen, int curlen) +{ + if (curlen >= maxlen) { + printf("*** WARNING: %s is too long (%d - max: %d)" + " - truncated\n", name, curlen, maxlen); + curlen = maxlen - 1; + } + return curlen; +} + +#if !defined(CONFIG_CMD_DHCP) + +static void bootp_process_vendor_field(u8 *ext) +{ + int size = *(ext + 1); + + debug("[BOOTP] Processing extension %d... (%d bytes)\n", *ext, + *(ext + 1)); + + net_boot_file_expected_size_in_blocks = 0; + + switch (*ext) { + /* Fixed length fields */ + case 1: /* Subnet mask */ + if (net_netmask.s_addr == 0) + net_copy_ip(&net_netmask, (struct in_addr *)(ext + 2)); + break; + case 2: /* Time offset - Not yet supported */ + break; + /* Variable length fields */ + case 3: /* Gateways list */ + if (net_gateway.s_addr == 0) + net_copy_ip(&net_gateway, (struct in_addr *)(ext + 2)); + break; + case 4: /* Time server - Not yet supported */ + break; + case 5: /* IEN-116 name server - Not yet supported */ + break; + case 6: + if (net_dns_server.s_addr == 0) + net_copy_ip(&net_dns_server, + (struct in_addr *)(ext + 2)); +#if defined(CONFIG_BOOTP_DNS2) + if ((net_dns_server2.s_addr == 0) && (size > 4)) + net_copy_ip(&net_dns_server2, + (struct in_addr *)(ext + 2 + 4)); +#endif + break; + case 7: /* Log server - Not yet supported */ + break; + case 8: /* Cookie/Quote server - Not yet supported */ + break; + case 9: /* LPR server - Not yet supported */ + break; + case 10: /* Impress server - Not yet supported */ + break; + case 11: /* RPL server - Not yet supported */ + break; + case 12: /* Host name */ + if (net_hostname[0] == 0) { + size = truncate_sz("Host Name", + sizeof(net_hostname), size); + memcpy(&net_hostname, ext + 2, size); + net_hostname[size] = 0; + } + break; + case 13: /* Boot file size */ + if (size == 2) + net_boot_file_expected_size_in_blocks = + ntohs(*(ushort *)(ext + 2)); + else if (size == 4) + net_boot_file_expected_size_in_blocks = + ntohl(*(ulong *)(ext + 2)); + break; + case 14: /* Merit dump file - Not yet supported */ + break; + case 15: /* Domain name - Not yet supported */ + break; + case 16: /* Swap server - Not yet supported */ + break; + case 17: /* Root path */ + if (net_root_path[0] == 0) { + size = truncate_sz("Root Path", + sizeof(net_root_path), size); + memcpy(&net_root_path, ext + 2, size); + net_root_path[size] = 0; + } + break; + case 18: /* Extension path - Not yet supported */ + /* + * This can be used to send the information of the + * vendor area in another file that the client can + * access via TFTP. + */ + break; + /* IP host layer fields */ + case 40: /* NIS Domain name */ + if (net_nis_domain[0] == 0) { + size = truncate_sz("NIS Domain Name", + sizeof(net_nis_domain), size); + memcpy(&net_nis_domain, ext + 2, size); + net_nis_domain[size] = 0; + } + break; +#if defined(CONFIG_CMD_SNTP) && defined(CONFIG_BOOTP_NTPSERVER) + case 42: /* NTP server IP */ + net_copy_ip(&net_ntp_server, (struct in_addr *)(ext + 2)); + break; +#endif + /* Application layer fields */ + case 43: /* Vendor specific info - Not yet supported */ + /* + * Binary information to exchange specific + * product information. + */ + break; + /* Reserved (custom) fields (128..254) */ + } +} + +static void bootp_process_vendor(u8 *ext, int size) +{ + u8 *end = ext + size; + + debug("[BOOTP] Checking extension (%d bytes)...\n", size); + + while ((ext < end) && (*ext != 0xff)) { + if (*ext == 0) { + ext++; + } else { + u8 *opt = ext; + + ext += ext[1] + 2; + if (ext <= end) + bootp_process_vendor_field(opt); + } + } + + debug("[BOOTP] Received fields:\n"); + if (net_netmask.s_addr) + debug("net_netmask : %pI4\n", &net_netmask); + + if (net_gateway.s_addr) + debug("net_gateway : %pI4", &net_gateway); + + if (net_boot_file_expected_size_in_blocks) + debug("net_boot_file_expected_size_in_blocks : %d\n", + net_boot_file_expected_size_in_blocks); + + if (net_hostname[0]) + debug("net_hostname : %s\n", net_hostname); + + if (net_root_path[0]) + debug("net_root_path : %s\n", net_root_path); + + if (net_nis_domain[0]) + debug("net_nis_domain : %s\n", net_nis_domain); + +#if defined(CONFIG_CMD_SNTP) && defined(CONFIG_BOOTP_NTPSERVER) + if (net_ntp_server.s_addr) + debug("net_ntp_server : %pI4\n", &net_ntp_server); +#endif +} + +/* + * Handle a BOOTP received packet. + */ +static void bootp_handler(uchar *pkt, unsigned dest, struct in_addr sip, + unsigned src, unsigned len) +{ + struct bootp_hdr *bp; + + debug("got BOOTP packet (src=%d, dst=%d, len=%d want_len=%zu)\n", + src, dest, len, sizeof(struct bootp_hdr)); + + bp = (struct bootp_hdr *)pkt; + + /* Filter out pkts we don't want */ + if (check_reply_packet(pkt, dest, src, len)) + return; + + /* + * Got a good BOOTP reply. Copy the data into our variables. + */ +#if defined(CONFIG_LED_STATUS) && defined(CONFIG_LED_STATUS_BOOT_ENABLE) + status_led_set(CONFIG_LED_STATUS_BOOT, CONFIG_LED_STATUS_OFF); +#endif + + store_net_params(bp); /* Store net parameters from reply */ + + /* Retrieve extended information (we must parse the vendor area) */ + if (net_read_u32((u32 *)&bp->bp_vend[0]) == htonl(BOOTP_VENDOR_MAGIC)) + bootp_process_vendor((uchar *)&bp->bp_vend[4], len); + + net_set_timeout_handler(0, (thand_f *)0); + bootstage_mark_name(BOOTSTAGE_ID_BOOTP_STOP, "bootp_stop"); + + debug("Got good BOOTP\n"); + + net_auto_load(); +} +#endif + +/* + * Timeout on BOOTP/DHCP request. + */ +static void bootp_timeout_handler(void) +{ + ulong time_taken = get_timer(bootp_start); + + if (time_taken >= time_taken_max) { +#ifdef CONFIG_BOOTP_MAY_FAIL + char *ethrotate; + + ethrotate = env_get("ethrotate"); + if ((ethrotate && strcmp(ethrotate, "no") == 0) || + net_restart_wrap) { + puts("\nRetry time exceeded\n"); + net_set_state(NETLOOP_FAIL); + } else +#endif + { + puts("\nRetry time exceeded; starting again\n"); + net_start_again(); + } + } else { + bootp_timeout *= 2; + if (bootp_timeout > 2000) + bootp_timeout = 2000; + net_set_timeout_handler(bootp_timeout, bootp_timeout_handler); + bootp_request(); + } +} + +#define put_vci(e, str) \ + do { \ + size_t vci_strlen = strlen(str); \ + *e++ = 60; /* Vendor Class Identifier */ \ + *e++ = vci_strlen; \ + memcpy(e, str, vci_strlen); \ + e += vci_strlen; \ + } while (0) + +static u8 *add_vci(u8 *e) +{ + char *vci = NULL; + char *env_vci = env_get("bootp_vci"); + +#if defined(CONFIG_SPL_BUILD) && defined(CONFIG_SPL_NET_VCI_STRING) + vci = CONFIG_SPL_NET_VCI_STRING; +#elif defined(CONFIG_BOOTP_VCI_STRING) + vci = CONFIG_BOOTP_VCI_STRING; +#endif + + if (env_vci) + vci = env_vci; + + if (vci) + put_vci(e, vci); + + return e; +} + +/* + * Initialize BOOTP extension fields in the request. + */ +#if defined(CONFIG_CMD_DHCP) +static int dhcp_extended(u8 *e, int message_type, struct in_addr server_ip, + struct in_addr requested_ip) +{ + u8 *start = e; + u8 *cnt; +#ifdef CONFIG_LIB_UUID + char *uuid; +#endif + int clientarch = -1; + +#if defined(CONFIG_BOOTP_VENDOREX) + u8 *x; +#endif +#if defined(CONFIG_BOOTP_SEND_HOSTNAME) + char *hostname; +#endif + + *e++ = 99; /* RFC1048 Magic Cookie */ + *e++ = 130; + *e++ = 83; + *e++ = 99; + + *e++ = 53; /* DHCP Message Type */ + *e++ = 1; + *e++ = message_type; + + *e++ = 57; /* Maximum DHCP Message Size */ + *e++ = 2; + *e++ = (576 - 312 + OPT_FIELD_SIZE) >> 8; + *e++ = (576 - 312 + OPT_FIELD_SIZE) & 0xff; + + if (server_ip.s_addr) { + int tmp = ntohl(server_ip.s_addr); + + *e++ = 54; /* ServerID */ + *e++ = 4; + *e++ = tmp >> 24; + *e++ = tmp >> 16; + *e++ = tmp >> 8; + *e++ = tmp & 0xff; + } + + if (requested_ip.s_addr) { + int tmp = ntohl(requested_ip.s_addr); + + *e++ = 50; /* Requested IP */ + *e++ = 4; + *e++ = tmp >> 24; + *e++ = tmp >> 16; + *e++ = tmp >> 8; + *e++ = tmp & 0xff; + } +#if defined(CONFIG_BOOTP_SEND_HOSTNAME) + hostname = env_get("hostname"); + if (hostname) { + int hostnamelen = strlen(hostname); + + *e++ = 12; /* Hostname */ + *e++ = hostnamelen; + memcpy(e, hostname, hostnamelen); + e += hostnamelen; + } +#endif + +#ifdef CONFIG_BOOTP_PXE_CLIENTARCH + clientarch = CONFIG_BOOTP_PXE_CLIENTARCH; +#endif + + if (env_get("bootp_arch")) + clientarch = env_get_ulong("bootp_arch", 16, clientarch); + + if (clientarch > 0) { + *e++ = 93; /* Client System Architecture */ + *e++ = 2; + *e++ = (clientarch >> 8) & 0xff; + *e++ = clientarch & 0xff; + } + + *e++ = 94; /* Client Network Interface Identifier */ + *e++ = 3; + *e++ = 1; /* type field for UNDI */ + *e++ = 0; /* major revision */ + *e++ = 0; /* minor revision */ + +#ifdef CONFIG_LIB_UUID + uuid = env_get("pxeuuid"); + + if (uuid) { + if (uuid_str_valid(uuid)) { + *e++ = 97; /* Client Machine Identifier */ + *e++ = 17; + *e++ = 0; /* type 0 - UUID */ + + uuid_str_to_bin(uuid, e, UUID_STR_FORMAT_STD); + e += 16; + } else { + printf("Invalid pxeuuid: %s\n", uuid); + } + } +#endif + + e = add_vci(e); + +#if defined(CONFIG_BOOTP_VENDOREX) + x = dhcp_vendorex_prep(e); + if (x) + return x - start; +#endif + + *e++ = 55; /* Parameter Request List */ + cnt = e++; /* Pointer to count of requested items */ + *cnt = 0; +#if defined(CONFIG_BOOTP_SUBNETMASK) + *e++ = 1; /* Subnet Mask */ + *cnt += 1; +#endif +#if defined(CONFIG_BOOTP_TIMEOFFSET) + *e++ = 2; + *cnt += 1; +#endif +#if defined(CONFIG_BOOTP_GATEWAY) + *e++ = 3; /* Router Option */ + *cnt += 1; +#endif +#if defined(CONFIG_BOOTP_DNS) + *e++ = 6; /* DNS Server(s) */ + *cnt += 1; +#endif +#if defined(CONFIG_BOOTP_HOSTNAME) + *e++ = 12; /* Hostname */ + *cnt += 1; +#endif +#if defined(CONFIG_BOOTP_BOOTFILESIZE) + *e++ = 13; /* Boot File Size */ + *cnt += 1; +#endif +#if defined(CONFIG_BOOTP_BOOTPATH) + *e++ = 17; /* Boot path */ + *cnt += 1; +#endif +#if defined(CONFIG_BOOTP_NISDOMAIN) + *e++ = 40; /* NIS Domain name request */ + *cnt += 1; +#endif +#if defined(CONFIG_BOOTP_NTPSERVER) + *e++ = 42; + *cnt += 1; +#endif + /* no options, so back up to avoid sending an empty request list */ + if (*cnt == 0) + e -= 2; + + *e++ = 255; /* End of the list */ + + /* Pad to minimal length */ +#ifdef CONFIG_DHCP_MIN_EXT_LEN + while ((e - start) < CONFIG_DHCP_MIN_EXT_LEN) + *e++ = 0; +#endif + + return e - start; +} + +#else +/* + * Warning: no field size check - change CONFIG_BOOTP_* at your own risk! + */ +static int bootp_extended(u8 *e) +{ + u8 *start = e; + + *e++ = 99; /* RFC1048 Magic Cookie */ + *e++ = 130; + *e++ = 83; + *e++ = 99; + +#if defined(CONFIG_CMD_DHCP) + *e++ = 53; /* DHCP Message Type */ + *e++ = 1; + *e++ = DHCP_DISCOVER; + + *e++ = 57; /* Maximum DHCP Message Size */ + *e++ = 2; + *e++ = (576 - 312 + OPT_FIELD_SIZE) >> 16; + *e++ = (576 - 312 + OPT_FIELD_SIZE) & 0xff; +#endif + + add_vci(e); + +#if defined(CONFIG_BOOTP_SUBNETMASK) + *e++ = 1; /* Subnet mask request */ + *e++ = 4; + e += 4; +#endif + +#if defined(CONFIG_BOOTP_GATEWAY) + *e++ = 3; /* Default gateway request */ + *e++ = 4; + e += 4; +#endif + +#if defined(CONFIG_BOOTP_DNS) + *e++ = 6; /* Domain Name Server */ + *e++ = 4; + e += 4; +#endif + +#if defined(CONFIG_BOOTP_HOSTNAME) + *e++ = 12; /* Host name request */ + *e++ = 32; + e += 32; +#endif + +#if defined(CONFIG_BOOTP_BOOTFILESIZE) + *e++ = 13; /* Boot file size */ + *e++ = 2; + e += 2; +#endif + +#if defined(CONFIG_BOOTP_BOOTPATH) + *e++ = 17; /* Boot path */ + *e++ = 32; + e += 32; +#endif + +#if defined(CONFIG_BOOTP_NISDOMAIN) + *e++ = 40; /* NIS Domain name request */ + *e++ = 32; + e += 32; +#endif +#if defined(CONFIG_BOOTP_NTPSERVER) + *e++ = 42; + *e++ = 4; + e += 4; +#endif + + *e++ = 255; /* End of the list */ + + /* + * If nothing in list, remove it altogether. Some DHCP servers get + * upset by this minor faux pas and do not respond at all. + */ + if (e == start + 3) { + printf("*** Warning: no DHCP options requested\n"); + e -= 3; + } + + return e - start; +} +#endif + +void bootp_reset(void) +{ + bootp_num_ids = 0; + bootp_try = 0; + bootp_start = get_timer(0); + bootp_timeout = 250; +} + +void bootp_request(void) +{ + uchar *pkt, *iphdr; + struct bootp_hdr *bp; + int extlen, pktlen, iplen; + int eth_hdr_size; +#ifdef CONFIG_BOOTP_RANDOM_DELAY + ulong rand_ms; +#endif + u32 bootp_id; + struct in_addr zero_ip; + struct in_addr bcast_ip; + char *ep; /* Environment pointer */ + + bootstage_mark_name(BOOTSTAGE_ID_BOOTP_START, "bootp_start"); +#if defined(CONFIG_CMD_DHCP) + dhcp_state = INIT; +#endif + + ep = env_get("bootpretryperiod"); + if (ep != NULL) + time_taken_max = simple_strtoul(ep, NULL, 10); + else + time_taken_max = TIMEOUT_MS; + +#ifdef CONFIG_BOOTP_RANDOM_DELAY /* Random BOOTP delay */ + if (bootp_try == 0) + srand_mac(); + + if (bootp_try <= 2) /* Start with max 1024 * 1ms */ + rand_ms = rand() >> (22 - bootp_try); + else /* After 3rd BOOTP request max 8192 * 1ms */ + rand_ms = rand() >> 19; + + printf("Random delay: %ld ms...\n", rand_ms); + mdelay(rand_ms); + +#endif /* CONFIG_BOOTP_RANDOM_DELAY */ + + printf("BOOTP broadcast %d\n", ++bootp_try); + pkt = net_tx_packet; + memset((void *)pkt, 0, PKTSIZE); + + eth_hdr_size = net_set_ether(pkt, net_bcast_ethaddr, PROT_IP); + pkt += eth_hdr_size; + + /* + * Next line results in incorrect packet size being transmitted, + * resulting in errors in some DHCP servers, reporting missing bytes. + * Size must be set in packet header after extension length has been + * determined. + * C. Hallinan, DS4.COM, Inc. + */ + /* net_set_udp_header(pkt, 0xFFFFFFFFL, PORT_BOOTPS, PORT_BOOTPC, + sizeof (struct bootp_hdr)); */ + iphdr = pkt; /* We need this later for net_set_udp_header() */ + pkt += IP_UDP_HDR_SIZE; + + bp = (struct bootp_hdr *)pkt; + bp->bp_op = OP_BOOTREQUEST; + bp->bp_htype = HWT_ETHER; + bp->bp_hlen = HWL_ETHER; + bp->bp_hops = 0; + /* + * according to RFC1542, should be 0 on first request, secs since + * first request otherwise + */ + bp->bp_secs = htons(get_timer(bootp_start) / 1000); + zero_ip.s_addr = 0; + net_write_ip(&bp->bp_ciaddr, zero_ip); + net_write_ip(&bp->bp_yiaddr, zero_ip); + net_write_ip(&bp->bp_siaddr, zero_ip); + net_write_ip(&bp->bp_giaddr, zero_ip); + memcpy(bp->bp_chaddr, net_ethaddr, 6); + copy_filename(bp->bp_file, net_boot_file_name, sizeof(bp->bp_file)); + + /* Request additional information from the BOOTP/DHCP server */ +#if defined(CONFIG_CMD_DHCP) + extlen = dhcp_extended((u8 *)bp->bp_vend, DHCP_DISCOVER, zero_ip, + zero_ip); +#else + extlen = bootp_extended((u8 *)bp->bp_vend); +#endif + + /* + * Bootp ID is the lower 4 bytes of our ethernet address + * plus the current time in ms. + */ + bootp_id = ((u32)net_ethaddr[2] << 24) + | ((u32)net_ethaddr[3] << 16) + | ((u32)net_ethaddr[4] << 8) + | (u32)net_ethaddr[5]; + bootp_id += get_timer(0); + bootp_id = htonl(bootp_id); + bootp_add_id(bootp_id); + net_copy_u32(&bp->bp_id, &bootp_id); + + /* + * Calculate proper packet lengths taking into account the + * variable size of the options field + */ + iplen = BOOTP_HDR_SIZE - OPT_FIELD_SIZE + extlen; + pktlen = eth_hdr_size + IP_UDP_HDR_SIZE + iplen; + bcast_ip.s_addr = 0xFFFFFFFFL; + net_set_udp_header(iphdr, bcast_ip, PORT_BOOTPS, PORT_BOOTPC, iplen); + net_set_timeout_handler(bootp_timeout, bootp_timeout_handler); + +#if defined(CONFIG_CMD_DHCP) + dhcp_state = SELECTING; + net_set_udp_handler(dhcp_handler); +#else + net_set_udp_handler(bootp_handler); +#endif + net_send_packet(net_tx_packet, pktlen); +} + +#if defined(CONFIG_CMD_DHCP) +static void dhcp_process_options(uchar *popt, uchar *end) +{ + int oplen, size; +#if defined(CONFIG_CMD_SNTP) && defined(CONFIG_BOOTP_TIMEOFFSET) + int *to_ptr; +#endif + + while (popt < end && *popt != 0xff) { + oplen = *(popt + 1); + switch (*popt) { + case 0: + oplen = -1; /* Pad omits len byte */ + break; + case 1: + net_copy_ip(&net_netmask, (popt + 2)); + break; +#if defined(CONFIG_CMD_SNTP) && defined(CONFIG_BOOTP_TIMEOFFSET) + case 2: /* Time offset */ + to_ptr = &net_ntp_time_offset; + net_copy_u32((u32 *)to_ptr, (u32 *)(popt + 2)); + net_ntp_time_offset = ntohl(net_ntp_time_offset); + break; +#endif + case 3: + net_copy_ip(&net_gateway, (popt + 2)); + break; + case 6: + net_copy_ip(&net_dns_server, (popt + 2)); +#if defined(CONFIG_BOOTP_DNS2) + if (*(popt + 1) > 4) + net_copy_ip(&net_dns_server2, (popt + 2 + 4)); +#endif + break; + case 12: + size = truncate_sz("Host Name", + sizeof(net_hostname), oplen); + memcpy(&net_hostname, popt + 2, size); + net_hostname[size] = 0; + break; + case 15: /* Ignore Domain Name Option */ + break; + case 17: + size = truncate_sz("Root Path", + sizeof(net_root_path), oplen); + memcpy(&net_root_path, popt + 2, size); + net_root_path[size] = 0; + break; + case 28: /* Ignore Broadcast Address Option */ + break; +#if defined(CONFIG_CMD_SNTP) && defined(CONFIG_BOOTP_NTPSERVER) + case 42: /* NTP server IP */ + net_copy_ip(&net_ntp_server, (popt + 2)); + break; +#endif + case 51: + net_copy_u32(&dhcp_leasetime, (u32 *)(popt + 2)); + break; + case 52: + dhcp_option_overload = popt[2]; + break; + case 53: /* Ignore Message Type Option */ + break; + case 54: + net_copy_ip(&dhcp_server_ip, (popt + 2)); + break; + case 58: /* Ignore Renewal Time Option */ + break; + case 59: /* Ignore Rebinding Time Option */ + break; + case 66: /* Ignore TFTP server name */ + break; + case 67: /* Bootfile option */ + if (!net_boot_file_name_explicit) { + size = truncate_sz("Bootfile", + sizeof(net_boot_file_name), + oplen); + memcpy(&net_boot_file_name, popt + 2, size); + net_boot_file_name[size] = 0; + } + break; + default: +#if defined(CONFIG_BOOTP_VENDOREX) + if (dhcp_vendorex_proc(popt)) + break; +#endif + printf("*** Unhandled DHCP Option in OFFER/ACK:" + " %d\n", *popt); + break; + } + popt += oplen + 2; /* Process next option */ + } +} + +static void dhcp_packet_process_options(struct bootp_hdr *bp) +{ + uchar *popt = (uchar *)&bp->bp_vend[4]; + uchar *end = popt + BOOTP_HDR_SIZE; + + if (net_read_u32((u32 *)&bp->bp_vend[0]) != htonl(BOOTP_VENDOR_MAGIC)) + return; + + dhcp_option_overload = 0; + + /* + * The 'options' field MUST be interpreted first, 'file' next, + * 'sname' last. + */ + dhcp_process_options(popt, end); + + if (dhcp_option_overload & OVERLOAD_FILE) { + popt = (uchar *)bp->bp_file; + end = popt + sizeof(bp->bp_file); + dhcp_process_options(popt, end); + } + + if (dhcp_option_overload & OVERLOAD_SNAME) { + popt = (uchar *)bp->bp_sname; + end = popt + sizeof(bp->bp_sname); + dhcp_process_options(popt, end); + } +} + +static int dhcp_message_type(unsigned char *popt) +{ + if (net_read_u32((u32 *)popt) != htonl(BOOTP_VENDOR_MAGIC)) + return -1; + + popt += 4; + while (*popt != 0xff) { + if (*popt == 53) /* DHCP Message Type */ + return *(popt + 2); + if (*popt == 0) { + /* Pad */ + popt += 1; + } else { + /* Scan through all options */ + popt += *(popt + 1) + 2; + } + } + return -1; +} + +static void dhcp_send_request_packet(struct bootp_hdr *bp_offer) +{ + uchar *pkt, *iphdr; + struct bootp_hdr *bp; + int pktlen, iplen, extlen; + int eth_hdr_size; + struct in_addr offered_ip; + struct in_addr zero_ip; + struct in_addr bcast_ip; + + debug("dhcp_send_request_packet: Sending DHCPREQUEST\n"); + pkt = net_tx_packet; + memset((void *)pkt, 0, PKTSIZE); + + eth_hdr_size = net_set_ether(pkt, net_bcast_ethaddr, PROT_IP); + pkt += eth_hdr_size; + + iphdr = pkt; /* We'll need this later to set proper pkt size */ + pkt += IP_UDP_HDR_SIZE; + + bp = (struct bootp_hdr *)pkt; + bp->bp_op = OP_BOOTREQUEST; + bp->bp_htype = HWT_ETHER; + bp->bp_hlen = HWL_ETHER; + bp->bp_hops = 0; + bp->bp_secs = htons(get_timer(bootp_start) / 1000); + /* Do not set the client IP, your IP, or server IP yet, since it + * hasn't been ACK'ed by the server yet */ + + /* + * RFC3046 requires Relay Agents to discard packets with + * nonzero and offered giaddr + */ + zero_ip.s_addr = 0; + net_write_ip(&bp->bp_giaddr, zero_ip); + + memcpy(bp->bp_chaddr, net_ethaddr, 6); + copy_filename(bp->bp_file, net_boot_file_name, sizeof(bp->bp_file)); + + /* + * ID is the id of the OFFER packet + */ + + net_copy_u32(&bp->bp_id, &bp_offer->bp_id); + + /* + * Copy options from OFFER packet if present + */ + + /* Copy offered IP into the parameters request list */ + net_copy_ip(&offered_ip, &bp_offer->bp_yiaddr); + extlen = dhcp_extended((u8 *)bp->bp_vend, DHCP_REQUEST, + dhcp_server_ip, offered_ip); + + iplen = BOOTP_HDR_SIZE - OPT_FIELD_SIZE + extlen; + pktlen = eth_hdr_size + IP_UDP_HDR_SIZE + iplen; + bcast_ip.s_addr = 0xFFFFFFFFL; + net_set_udp_header(iphdr, bcast_ip, PORT_BOOTPS, PORT_BOOTPC, iplen); + +#ifdef CONFIG_BOOTP_DHCP_REQUEST_DELAY + udelay(CONFIG_BOOTP_DHCP_REQUEST_DELAY); +#endif /* CONFIG_BOOTP_DHCP_REQUEST_DELAY */ + debug("Transmitting DHCPREQUEST packet: len = %d\n", pktlen); + net_send_packet(net_tx_packet, pktlen); +} + +/* + * Handle DHCP received packets. + */ +static void dhcp_handler(uchar *pkt, unsigned dest, struct in_addr sip, + unsigned src, unsigned len) +{ + struct bootp_hdr *bp = (struct bootp_hdr *)pkt; + + debug("DHCPHandler: got packet: (src=%d, dst=%d, len=%d) state: %d\n", + src, dest, len, dhcp_state); + + /* Filter out pkts we don't want */ + if (check_reply_packet(pkt, dest, src, len)) + return; + + debug("DHCPHandler: got DHCP packet: (src=%d, dst=%d, len=%d) state: " + "%d\n", src, dest, len, dhcp_state); + + if (net_read_ip(&bp->bp_yiaddr).s_addr == 0) { +#if defined(CONFIG_SERVERIP_FROM_PROXYDHCP) + store_bootp_params(bp); +#endif + return; + } + + switch (dhcp_state) { + case SELECTING: + /* + * Wait an appropriate time for any potential DHCPOFFER packets + * to arrive. Then select one, and generate DHCPREQUEST + * response. If filename is in format we recognize, assume it + * is a valid OFFER from a server we want. + */ + debug("DHCP: state=SELECTING bp_file: \"%s\"\n", bp->bp_file); +#ifdef CONFIG_SYS_BOOTFILE_PREFIX + if (strncmp(bp->bp_file, + CONFIG_SYS_BOOTFILE_PREFIX, + strlen(CONFIG_SYS_BOOTFILE_PREFIX)) == 0) { +#endif /* CONFIG_SYS_BOOTFILE_PREFIX */ + dhcp_packet_process_options(bp); + efi_net_set_dhcp_ack(pkt, len); + +#if defined(CONFIG_SERVERIP_FROM_PROXYDHCP) + if (!net_server_ip.s_addr) + udelay(CONFIG_SERVERIP_FROM_PROXYDHCP_DELAY_MS * + 1000); +#endif /* CONFIG_SERVERIP_FROM_PROXYDHCP */ + + debug("TRANSITIONING TO REQUESTING STATE\n"); + dhcp_state = REQUESTING; + + net_set_timeout_handler(5000, bootp_timeout_handler); + dhcp_send_request_packet(bp); +#ifdef CONFIG_SYS_BOOTFILE_PREFIX + } +#endif /* CONFIG_SYS_BOOTFILE_PREFIX */ + + return; + break; + case REQUESTING: + debug("DHCP State: REQUESTING\n"); + + if (dhcp_message_type((u8 *)bp->bp_vend) == DHCP_ACK) { + dhcp_packet_process_options(bp); + /* Store net params from reply */ + store_net_params(bp); + dhcp_state = BOUND; + printf("DHCP client bound to address %pI4 (%lu ms)\n", + &net_ip, get_timer(bootp_start)); + net_set_timeout_handler(0, (thand_f *)0); + bootstage_mark_name(BOOTSTAGE_ID_BOOTP_STOP, + "bootp_stop"); + + net_auto_load(); + return; + } + break; + case BOUND: + /* DHCP client bound to address */ + break; + default: + puts("DHCP: INVALID STATE\n"); + break; + } +} + +void dhcp_request(void) +{ + bootp_request(); +} +#endif /* CONFIG_CMD_DHCP */ |