// SPDX-License-Identifier: GPL-2.0+ /* * Simple network protocol * PXE base code protocol * * Copyright (c) 2016 Alexander Graf * * The simple network protocol has the following statuses and services * to move between them: * * Start(): EfiSimpleNetworkStopped -> EfiSimpleNetworkStarted * Initialize(): EfiSimpleNetworkStarted -> EfiSimpleNetworkInitialized * Shutdown(): EfiSimpleNetworkInitialized -> EfiSimpleNetworkStarted * Stop(): EfiSimpleNetworkStarted -> EfiSimpleNetworkStopped * Reset(): EfiSimpleNetworkInitialized -> EfiSimpleNetworkInitialized */ #include #include #include static const efi_guid_t efi_net_guid = EFI_SIMPLE_NETWORK_PROTOCOL_GUID; static const efi_guid_t efi_pxe_base_code_protocol_guid = EFI_PXE_BASE_CODE_PROTOCOL_GUID; static struct efi_pxe_packet *dhcp_ack; static void *new_tx_packet; static void *transmit_buffer; static uchar **receive_buffer; static size_t *receive_lengths; static int rx_packet_idx; static int rx_packet_num; static struct efi_net_obj *netobj; /* * The notification function of this event is called in every timer cycle * to check if a new network packet has been received. */ static struct efi_event *network_timer_event; /* * This event is signaled when a packet has been received. */ static struct efi_event *wait_for_packet; /** * struct efi_net_obj - EFI object representing a network interface * * @header: EFI object header * @net: simple network protocol interface * @net_mode: status of the network interface * @pxe: PXE base code protocol interface * @pxe_mode: status of the PXE base code protocol */ struct efi_net_obj { struct efi_object header; struct efi_simple_network net; struct efi_simple_network_mode net_mode; struct efi_pxe_base_code_protocol pxe; struct efi_pxe_mode pxe_mode; }; /* * efi_net_start() - start the network interface * * This function implements the Start service of the * EFI_SIMPLE_NETWORK_PROTOCOL. See the Unified Extensible Firmware Interface * (UEFI) specification for details. * * @this: pointer to the protocol instance * Return: status code */ static efi_status_t EFIAPI efi_net_start(struct efi_simple_network *this) { efi_status_t ret = EFI_SUCCESS; EFI_ENTRY("%p", this); /* Check parameters */ if (!this) { ret = EFI_INVALID_PARAMETER; goto out; } if (this->mode->state != EFI_NETWORK_STOPPED) { ret = EFI_ALREADY_STARTED; } else { this->int_status = 0; wait_for_packet->is_signaled = false; this->mode->state = EFI_NETWORK_STARTED; } out: return EFI_EXIT(ret); } /* * efi_net_stop() - stop the network interface * * This function implements the Stop service of the * EFI_SIMPLE_NETWORK_PROTOCOL. See the Unified Extensible Firmware Interface * (UEFI) specification for details. * * @this: pointer to the protocol instance * Return: status code */ static efi_status_t EFIAPI efi_net_stop(struct efi_simple_network *this) { efi_status_t ret = EFI_SUCCESS; EFI_ENTRY("%p", this); /* Check parameters */ if (!this) { ret = EFI_INVALID_PARAMETER; goto out; } if (this->mode->state == EFI_NETWORK_STOPPED) { ret = EFI_NOT_STARTED; } else { /* Disable hardware and put it into the reset state */ eth_halt(); /* Clear cache of packets */ rx_packet_num = 0; this->mode->state = EFI_NETWORK_STOPPED; } out: return EFI_EXIT(ret); } /* * efi_net_initialize() - initialize the network interface * * This function implements the Initialize service of the * EFI_SIMPLE_NETWORK_PROTOCOL. See the Unified Extensible Firmware Interface * (UEFI) specification for details. * * @this: pointer to the protocol instance * @extra_rx: extra receive buffer to be allocated * @extra_tx: extra transmit buffer to be allocated * Return: status code */ static efi_status_t EFIAPI efi_net_initialize(struct efi_simple_network *this, ulong extra_rx, ulong extra_tx) { int ret; efi_status_t r = EFI_SUCCESS; EFI_ENTRY("%p, %lx, %lx", this, extra_rx, extra_tx); /* Check parameters */ if (!this) { r = EFI_INVALID_PARAMETER; goto out; } switch (this->mode->state) { case EFI_NETWORK_INITIALIZED: case EFI_NETWORK_STARTED: break; default: r = EFI_NOT_STARTED; goto out; } /* Setup packet buffers */ net_init(); /* Disable hardware and put it into the reset state */ eth_halt(); /* Clear cache of packets */ rx_packet_num = 0; /* Set current device according to environment variables */ eth_set_current(); /* Get hardware ready for send and receive operations */ ret = eth_init(); if (ret < 0) { eth_halt(); this->mode->state = EFI_NETWORK_STOPPED; r = EFI_DEVICE_ERROR; goto out; } else { this->int_status = 0; wait_for_packet->is_signaled = false; this->mode->state = EFI_NETWORK_INITIALIZED; } out: return EFI_EXIT(r); } /* * efi_net_reset() - reinitialize the network interface * * This function implements the Reset service of the * EFI_SIMPLE_NETWORK_PROTOCOL. See the Unified Extensible Firmware Interface * (UEFI) specification for details. * * @this: pointer to the protocol instance * @extended_verification: execute exhaustive verification * Return: status code */ static efi_status_t EFIAPI efi_net_reset(struct efi_simple_network *this, int extended_verification) { efi_status_t ret; EFI_ENTRY("%p, %x", this, extended_verification); /* Check parameters */ if (!this) { ret = EFI_INVALID_PARAMETER; goto out; } switch (this->mode->state) { case EFI_NETWORK_INITIALIZED: break; case EFI_NETWORK_STOPPED: ret = EFI_NOT_STARTED; goto out; default: ret = EFI_DEVICE_ERROR; goto out; } this->mode->state = EFI_NETWORK_STARTED; ret = EFI_CALL(efi_net_initialize(this, 0, 0)); out: return EFI_EXIT(ret); } /* * efi_net_shutdown() - shut down the network interface * * This function implements the Shutdown service of the * EFI_SIMPLE_NETWORK_PROTOCOL. See the Unified Extensible Firmware Interface * (UEFI) specification for details. * * @this: pointer to the protocol instance * Return: status code */ static efi_status_t EFIAPI efi_net_shutdown(struct efi_simple_network *this) { efi_status_t ret = EFI_SUCCESS; EFI_ENTRY("%p", this); /* Check parameters */ if (!this) { ret = EFI_INVALID_PARAMETER; goto out; } switch (this->mode->state) { case EFI_NETWORK_INITIALIZED: break; case EFI_NETWORK_STOPPED: ret = EFI_NOT_STARTED; goto out; default: ret = EFI_DEVICE_ERROR; goto out; } eth_halt(); this->int_status = 0; wait_for_packet->is_signaled = false; this->mode->state = EFI_NETWORK_STARTED; out: return EFI_EXIT(ret); } /* * efi_net_receive_filters() - mange multicast receive filters * * This function implements the ReceiveFilters service of the * EFI_SIMPLE_NETWORK_PROTOCOL. See the Unified Extensible Firmware Interface * (UEFI) specification for details. * * @this: pointer to the protocol instance * @enable: bit mask of receive filters to enable * @disable: bit mask of receive filters to disable * @reset_mcast_filter: true resets contents of the filters * @mcast_filter_count: number of hardware MAC addresses in the new filters list * @mcast_filter: list of new filters * Return: status code */ static efi_status_t EFIAPI efi_net_receive_filters (struct efi_simple_network *this, u32 enable, u32 disable, int reset_mcast_filter, ulong mcast_filter_count, struct efi_mac_address *mcast_filter) { EFI_ENTRY("%p, %x, %x, %x, %lx, %p", this, enable, disable, reset_mcast_filter, mcast_filter_count, mcast_filter); return EFI_EXIT(EFI_UNSUPPORTED); } /* * efi_net_station_address() - set the hardware MAC address * * This function implements the StationAddress service of the * EFI_SIMPLE_NETWORK_PROTOCOL. See the Unified Extensible Firmware Interface * (UEFI) specification for details. * * @this: pointer to the protocol instance * @reset: if true reset the address to default * @new_mac: new MAC address * Return: status code */ static efi_status_t EFIAPI efi_net_station_address (struct efi_simple_network *this, int reset, struct efi_mac_address *new_mac) { EFI_ENTRY("%p, %x, %p", this, reset, new_mac); return EFI_EXIT(EFI_UNSUPPORTED); } /* * efi_net_statistics() - reset or collect statistics of the network interface * * This function implements the Statistics service of the * EFI_SIMPLE_NETWORK_PROTOCOL. See the Unified Extensible Firmware Interface * (UEFI) specification for details. * * @this: pointer to the protocol instance * @reset: if true, the statistics are reset * @stat_size: size of the statistics table * @stat_table: table to receive the statistics * Return: status code */ static efi_status_t EFIAPI efi_net_statistics(struct efi_simple_network *this, int reset, ulong *stat_size, void *stat_table) { EFI_ENTRY("%p, %x, %p, %p", this, reset, stat_size, stat_table); return EFI_EXIT(EFI_UNSUPPORTED); } /* * efi_net_mcastiptomac() - translate multicast IP address to MAC address * * This function implements the MCastIPtoMAC service of the * EFI_SIMPLE_NETWORK_PROTOCOL. See the Unified Extensible Firmware Interface * (UEFI) specification for details. * * @this: pointer to the protocol instance * @ipv6: true if the IP address is an IPv6 address * @ip: IP address * @mac: MAC address * Return: status code */ static efi_status_t EFIAPI efi_net_mcastiptomac(struct efi_simple_network *this, int ipv6, struct efi_ip_address *ip, struct efi_mac_address *mac) { efi_status_t ret = EFI_SUCCESS; EFI_ENTRY("%p, %x, %p, %p", this, ipv6, ip, mac); if (!this || !ip || !mac) { ret = EFI_INVALID_PARAMETER; goto out; } if (ipv6) { ret = EFI_UNSUPPORTED; goto out; } /* Multi-cast addresses are in the range 224.0.0.0 - 239.255.255.255 */ if ((ip->ip_addr[0] & 0xf0) != 0xe0) { ret = EFI_INVALID_PARAMETER; goto out; }; switch (this->mode->state) { case EFI_NETWORK_INITIALIZED: case EFI_NETWORK_STARTED: break; default: ret = EFI_NOT_STARTED; goto out; } memset(mac, 0, sizeof(struct efi_mac_address)); /* * Copy lower 23 bits of IPv4 multi-cast address * RFC 1112, RFC 7042 2.1.1. */ mac->mac_addr[0] = 0x01; mac->mac_addr[1] = 0x00; mac->mac_addr[2] = 0x5E; mac->mac_addr[3] = ip->ip_addr[1] & 0x7F; mac->mac_addr[4] = ip->ip_addr[2]; mac->mac_addr[5] = ip->ip_addr[3]; out: return EFI_EXIT(ret); } /** * efi_net_nvdata() - read or write NVRAM * * This function implements the GetStatus service of the Simple Network * Protocol. See the UEFI spec for details. * * @this: the instance of the Simple Network Protocol * @read_write: true for read, false for write * @offset: offset in NVRAM * @buffer_size: size of buffer * @buffer: buffer * Return: status code */ static efi_status_t EFIAPI efi_net_nvdata(struct efi_simple_network *this, int read_write, ulong offset, ulong buffer_size, char *buffer) { EFI_ENTRY("%p, %x, %lx, %lx, %p", this, read_write, offset, buffer_size, buffer); return EFI_EXIT(EFI_UNSUPPORTED); } /** * efi_net_get_status() - get interrupt status * * This function implements the GetStatus service of the Simple Network * Protocol. See the UEFI spec for details. * * @this: the instance of the Simple Network Protocol * @int_status: interface status * @txbuf: transmission buffer */ static efi_status_t EFIAPI efi_net_get_status(struct efi_simple_network *this, u32 *int_status, void **txbuf) { efi_status_t ret = EFI_SUCCESS; EFI_ENTRY("%p, %p, %p", this, int_status, txbuf); efi_timer_check(); /* Check parameters */ if (!this) { ret = EFI_INVALID_PARAMETER; goto out; } switch (this->mode->state) { case EFI_NETWORK_STOPPED: ret = EFI_NOT_STARTED; goto out; case EFI_NETWORK_STARTED: ret = EFI_DEVICE_ERROR; goto out; default: break; } if (int_status) { *int_status = this->int_status; this->int_status = 0; } if (txbuf) *txbuf = new_tx_packet; new_tx_packet = NULL; out: return EFI_EXIT(ret); } /** * efi_net_transmit() - transmit a packet * * This function implements the Transmit service of the Simple Network Protocol. * See the UEFI spec for details. * * @this: the instance of the Simple Network Protocol * @header_size: size of the media header * @buffer_size: size of the buffer to receive the packet * @buffer: buffer to receive the packet * @src_addr: source hardware MAC address * @dest_addr: destination hardware MAC address * @protocol: type of header to build * Return: status code */ static efi_status_t EFIAPI efi_net_transmit (struct efi_simple_network *this, size_t header_size, size_t buffer_size, void *buffer, struct efi_mac_address *src_addr, struct efi_mac_address *dest_addr, u16 *protocol) { efi_status_t ret = EFI_SUCCESS; EFI_ENTRY("%p, %lu, %lu, %p, %p, %p, %p", this, (unsigned long)header_size, (unsigned long)buffer_size, buffer, src_addr, dest_addr, protocol); efi_timer_check(); /* Check parameters */ if (!this || !buffer) { ret = EFI_INVALID_PARAMETER; goto out; } /* We do not support jumbo packets */ if (buffer_size > PKTSIZE_ALIGN) { ret = EFI_INVALID_PARAMETER; goto out; } /* At least the IP header has to fit into the buffer */ if (buffer_size < this->mode->media_header_size) { ret = EFI_BUFFER_TOO_SMALL; goto out; } /* * TODO: * Support VLANs. Use net_set_ether() for copying the header. Use a * U_BOOT_ENV_CALLBACK to update the media header size. */ if (header_size) { struct ethernet_hdr *header = buffer; if (!dest_addr || !protocol || header_size != this->mode->media_header_size) { ret = EFI_INVALID_PARAMETER; goto out; } if (!src_addr) src_addr = &this->mode->current_address; memcpy(header->et_dest, dest_addr, ARP_HLEN); memcpy(header->et_src, src_addr, ARP_HLEN); header->et_protlen = htons(*protocol); } switch (this->mode->state) { case EFI_NETWORK_STOPPED: ret = EFI_NOT_STARTED; goto out; case EFI_NETWORK_STARTED: ret = EFI_DEVICE_ERROR; goto out; default: break; } /* Ethernet packets always fit, just bounce */ memcpy(transmit_buffer, buffer, buffer_size); net_send_packet(transmit_buffer, buffer_size); new_tx_packet = buffer; this->int_status |= EFI_SIMPLE_NETWORK_TRANSMIT_INTERRUPT; out: return EFI_EXIT(ret); } /** * efi_net_receive() - receive a packet from a network interface * * This function implements the Receive service of the Simple Network Protocol. * See the UEFI spec for details. * * @this: the instance of the Simple Network Protocol * @header_size: size of the media header * @buffer_size: size of the buffer to receive the packet * @buffer: buffer to receive the packet * @src_addr: source MAC address * @dest_addr: destination MAC address * @protocol: protocol * Return: status code */ static efi_status_t EFIAPI efi_net_receive (struct efi_simple_network *this, size_t *header_size, size_t *buffer_size, void *buffer, struct efi_mac_address *src_addr, struct efi_mac_address *dest_addr, u16 *protocol) { efi_status_t ret = EFI_SUCCESS; struct ethernet_hdr *eth_hdr; size_t hdr_size = sizeof(struct ethernet_hdr); u16 protlen; EFI_ENTRY("%p, %p, %p, %p, %p, %p, %p", this, header_size, buffer_size, buffer, src_addr, dest_addr, protocol); /* Execute events */ efi_timer_check(); /* Check parameters */ if (!this || !buffer || !buffer_size) { ret = EFI_INVALID_PARAMETER; goto out; } switch (this->mode->state) { case EFI_NETWORK_STOPPED: ret = EFI_NOT_STARTED; goto out; case EFI_NETWORK_STARTED: ret = EFI_DEVICE_ERROR; goto out; default: break; } if (!rx_packet_num) { ret = EFI_NOT_READY; goto out; } /* Fill export parameters */ eth_hdr = (struct ethernet_hdr *)receive_buffer[rx_packet_idx]; protlen = ntohs(eth_hdr->et_protlen); if (protlen == 0x8100) { hdr_size += 4; protlen = ntohs(*(u16 *)&receive_buffer[rx_packet_idx][hdr_size - 2]); } if (header_size) *header_size = hdr_size; if (dest_addr) memcpy(dest_addr, eth_hdr->et_dest, ARP_HLEN); if (src_addr) memcpy(src_addr, eth_hdr->et_src, ARP_HLEN); if (protocol) *protocol = protlen; if (*buffer_size < receive_lengths[rx_packet_idx]) { /* Packet doesn't fit, try again with bigger buffer */ *buffer_size = receive_lengths[rx_packet_idx]; ret = EFI_BUFFER_TOO_SMALL; goto out; } /* Copy packet */ memcpy(buffer, receive_buffer[rx_packet_idx], receive_lengths[rx_packet_idx]); *buffer_size = receive_lengths[rx_packet_idx]; rx_packet_idx = (rx_packet_idx + 1) % ETH_PACKETS_BATCH_RECV; rx_packet_num--; if (rx_packet_num) wait_for_packet->is_signaled = true; else this->int_status &= ~EFI_SIMPLE_NETWORK_RECEIVE_INTERRUPT; out: return EFI_EXIT(ret); } /** * efi_net_set_dhcp_ack() - take note of a selected DHCP IP address * * This function is called by dhcp_handler(). * * @pkt: packet received by dhcp_handler() * @len: length of the packet received */ void efi_net_set_dhcp_ack(void *pkt, int len) { int maxsize = sizeof(*dhcp_ack); if (!dhcp_ack) { dhcp_ack = malloc(maxsize); if (!dhcp_ack) return; } memset(dhcp_ack, 0, maxsize); memcpy(dhcp_ack, pkt, min(len, maxsize)); if (netobj) netobj->pxe_mode.dhcp_ack = *dhcp_ack; } /** * efi_net_push() - callback for received network packet * * This function is called when a network packet is received by eth_rx(). * * @pkt: network packet * @len: length */ static void efi_net_push(void *pkt, int len) { int rx_packet_next; /* Check that we at least received an Ethernet header */ if (len < sizeof(struct ethernet_hdr)) return; /* Check that the buffer won't overflow */ if (len > PKTSIZE_ALIGN) return; /* Can't store more than pre-alloced buffer */ if (rx_packet_num >= ETH_PACKETS_BATCH_RECV) return; rx_packet_next = (rx_packet_idx + rx_packet_num) % ETH_PACKETS_BATCH_RECV; memcpy(receive_buffer[rx_packet_next], pkt, len); receive_lengths[rx_packet_next] = len; rx_packet_num++; } /** * efi_network_timer_notify() - check if a new network packet has been received * * This notification function is called in every timer cycle. * * @event: the event for which this notification function is registered * @context: event context - not used in this function */ static void EFIAPI efi_network_timer_notify(struct efi_event *event, void *context) { struct efi_simple_network *this = (struct efi_simple_network *)context; EFI_ENTRY("%p, %p", event, context); /* * Some network drivers do not support calling eth_rx() before * initialization. */ if (!this || this->mode->state != EFI_NETWORK_INITIALIZED) goto out; if (!rx_packet_num) { push_packet = efi_net_push; eth_rx(); push_packet = NULL; if (rx_packet_num) { this->int_status |= EFI_SIMPLE_NETWORK_RECEIVE_INTERRUPT; wait_for_packet->is_signaled = true; } } out: EFI_EXIT(EFI_SUCCESS); } static efi_status_t EFIAPI efi_pxe_base_code_start( struct efi_pxe_base_code_protocol *this, u8 use_ipv6) { return EFI_UNSUPPORTED; } static efi_status_t EFIAPI efi_pxe_base_code_stop( struct efi_pxe_base_code_protocol *this) { return EFI_UNSUPPORTED; } static efi_status_t EFIAPI efi_pxe_base_code_dhcp( struct efi_pxe_base_code_protocol *this, u8 sort_offers) { return EFI_UNSUPPORTED; } static efi_status_t EFIAPI efi_pxe_base_code_discover( struct efi_pxe_base_code_protocol *this, u16 type, u16 *layer, u8 bis, struct efi_pxe_base_code_discover_info *info) { return EFI_UNSUPPORTED; } static efi_status_t EFIAPI efi_pxe_base_code_mtftp( struct efi_pxe_base_code_protocol *this, u32 operation, void *buffer_ptr, u8 overwrite, efi_uintn_t *buffer_size, struct efi_ip_address server_ip, char *filename, struct efi_pxe_base_code_mtftp_info *info, u8 dont_use_buffer) { return EFI_UNSUPPORTED; } static efi_status_t EFIAPI efi_pxe_base_code_udp_write( struct efi_pxe_base_code_protocol *this, u16 op_flags, struct efi_ip_address *dest_ip, u16 *dest_port, struct efi_ip_address *gateway_ip, struct efi_ip_address *src_ip, u16 *src_port, efi_uintn_t *header_size, void *header_ptr, efi_uintn_t *buffer_size, void *buffer_ptr) { return EFI_UNSUPPORTED; } static efi_status_t EFIAPI efi_pxe_base_code_udp_read( struct efi_pxe_base_code_protocol *this, u16 op_flags, struct efi_ip_address *dest_ip, u16 *dest_port, struct efi_ip_address *src_ip, u16 *src_port, efi_uintn_t *header_size, void *header_ptr, efi_uintn_t *buffer_size, void *buffer_ptr) { return EFI_UNSUPPORTED; } static efi_status_t EFIAPI efi_pxe_base_code_set_ip_filter( struct efi_pxe_base_code_protocol *this, struct efi_pxe_base_code_filter *new_filter) { return EFI_UNSUPPORTED; } static efi_status_t EFIAPI efi_pxe_base_code_arp( struct efi_pxe_base_code_protocol *this, struct efi_ip_address *ip_addr, struct efi_mac_address *mac_addr) { return EFI_UNSUPPORTED; } static efi_status_t EFIAPI efi_pxe_base_code_set_parameters( struct efi_pxe_base_code_protocol *this, u8 *new_auto_arp, u8 *new_send_guid, u8 *new_ttl, u8 *new_tos, u8 *new_make_callback) { return EFI_UNSUPPORTED; } static efi_status_t EFIAPI efi_pxe_base_code_set_station_ip( struct efi_pxe_base_code_protocol *this, struct efi_ip_address *new_station_ip, struct efi_ip_address *new_subnet_mask) { return EFI_UNSUPPORTED; } static efi_status_t EFIAPI efi_pxe_base_code_set_packets( struct efi_pxe_base_code_protocol *this, u8 *new_dhcp_discover_valid, u8 *new_dhcp_ack_received, u8 *new_proxy_offer_received, u8 *new_pxe_discover_valid, u8 *new_pxe_reply_received, u8 *new_pxe_bis_reply_received, EFI_PXE_BASE_CODE_PACKET *new_dchp_discover, EFI_PXE_BASE_CODE_PACKET *new_dhcp_acc, EFI_PXE_BASE_CODE_PACKET *new_proxy_offer, EFI_PXE_BASE_CODE_PACKET *new_pxe_discover, EFI_PXE_BASE_CODE_PACKET *new_pxe_reply, EFI_PXE_BASE_CODE_PACKET *new_pxe_bis_reply) { return EFI_UNSUPPORTED; } /** * efi_net_register() - register the simple network protocol * * This gets called from do_bootefi_exec(). */ efi_status_t efi_net_register(void) { efi_status_t r; int i; if (!eth_get_dev()) { /* No network device active, don't expose any */ return EFI_SUCCESS; } /* We only expose the "active" network device, so one is enough */ netobj = calloc(1, sizeof(*netobj)); if (!netobj) goto out_of_resources; /* Allocate an aligned transmit buffer */ transmit_buffer = calloc(1, PKTSIZE_ALIGN + PKTALIGN); if (!transmit_buffer) goto out_of_resources; transmit_buffer = (void *)ALIGN((uintptr_t)transmit_buffer, PKTALIGN); /* Allocate a number of receive buffers */ receive_buffer = calloc(ETH_PACKETS_BATCH_RECV, sizeof(*receive_buffer)); if (!receive_buffer) goto out_of_resources; for (i = 0; i < ETH_PACKETS_BATCH_RECV; i++) { receive_buffer[i] = malloc(PKTSIZE_ALIGN); if (!receive_buffer[i]) goto out_of_resources; } receive_lengths = calloc(ETH_PACKETS_BATCH_RECV, sizeof(*receive_lengths)); if (!receive_lengths) goto out_of_resources; /* Hook net up to the device list */ efi_add_handle(&netobj->header); /* Fill in object data */ r = efi_add_protocol(&netobj->header, &efi_net_guid, &netobj->net); if (r != EFI_SUCCESS) goto failure_to_add_protocol; r = efi_add_protocol(&netobj->header, &efi_guid_device_path, efi_dp_from_eth()); if (r != EFI_SUCCESS) goto failure_to_add_protocol; r = efi_add_protocol(&netobj->header, &efi_pxe_base_code_protocol_guid, &netobj->pxe); if (r != EFI_SUCCESS) goto failure_to_add_protocol; netobj->net.revision = EFI_SIMPLE_NETWORK_PROTOCOL_REVISION; netobj->net.start = efi_net_start; netobj->net.stop = efi_net_stop; netobj->net.initialize = efi_net_initialize; netobj->net.reset = efi_net_reset; netobj->net.shutdown = efi_net_shutdown; netobj->net.receive_filters = efi_net_receive_filters; netobj->net.station_address = efi_net_station_address; netobj->net.statistics = efi_net_statistics; netobj->net.mcastiptomac = efi_net_mcastiptomac; netobj->net.nvdata = efi_net_nvdata; netobj->net.get_status = efi_net_get_status; netobj->net.transmit = efi_net_transmit; netobj->net.receive = efi_net_receive; netobj->net.mode = &netobj->net_mode; netobj->net_mode.state = EFI_NETWORK_STOPPED; memcpy(netobj->net_mode.current_address.mac_addr, eth_get_ethaddr(), 6); netobj->net_mode.hwaddr_size = ARP_HLEN; netobj->net_mode.media_header_size = ETHER_HDR_SIZE; netobj->net_mode.max_packet_size = PKTSIZE; netobj->net_mode.if_type = ARP_ETHER; netobj->pxe.revision = EFI_PXE_BASE_CODE_PROTOCOL_REVISION; netobj->pxe.start = efi_pxe_base_code_start; netobj->pxe.stop = efi_pxe_base_code_stop; netobj->pxe.dhcp = efi_pxe_base_code_dhcp; netobj->pxe.discover = efi_pxe_base_code_discover; netobj->pxe.mtftp = efi_pxe_base_code_mtftp; netobj->pxe.udp_write = efi_pxe_base_code_udp_write; netobj->pxe.udp_read = efi_pxe_base_code_udp_read; netobj->pxe.set_ip_filter = efi_pxe_base_code_set_ip_filter; netobj->pxe.arp = efi_pxe_base_code_arp; netobj->pxe.set_parameters = efi_pxe_base_code_set_parameters; netobj->pxe.set_station_ip = efi_pxe_base_code_set_station_ip; netobj->pxe.set_packets = efi_pxe_base_code_set_packets; netobj->pxe.mode = &netobj->pxe_mode; if (dhcp_ack) netobj->pxe_mode.dhcp_ack = *dhcp_ack; /* * Create WaitForPacket event. */ r = efi_create_event(EVT_NOTIFY_WAIT, TPL_CALLBACK, efi_network_timer_notify, NULL, NULL, &wait_for_packet); if (r != EFI_SUCCESS) { printf("ERROR: Failed to register network event\n"); return r; } netobj->net.wait_for_packet = wait_for_packet; /* * Create a timer event. * * The notification function is used to check if a new network packet * has been received. * * iPXE is running at TPL_CALLBACK most of the time. Use a higher TPL. */ r = efi_create_event(EVT_TIMER | EVT_NOTIFY_SIGNAL, TPL_NOTIFY, efi_network_timer_notify, &netobj->net, NULL, &network_timer_event); if (r != EFI_SUCCESS) { printf("ERROR: Failed to register network event\n"); return r; } /* Network is time critical, create event in every timer cycle */ r = efi_set_timer(network_timer_event, EFI_TIMER_PERIODIC, 0); if (r != EFI_SUCCESS) { printf("ERROR: Failed to set network timer\n"); return r; } return EFI_SUCCESS; failure_to_add_protocol: printf("ERROR: Failure to add protocol\n"); return r; out_of_resources: free(netobj); netobj = NULL; free(transmit_buffer); if (receive_buffer) for (i = 0; i < ETH_PACKETS_BATCH_RECV; i++) free(receive_buffer[i]); free(receive_buffer); free(receive_lengths); printf("ERROR: Out of memory\n"); return EFI_OUT_OF_RESOURCES; }