pxe_preboot.c

Go to the documentation of this file.

/** @file
 *
 * PXE Preboot API
 *
 */

/* PXE API interface for Etherboot.
 *
 * Copyright (C) 2004 Michael Brown <mbrown@fensystems.co.uk>.
 *
 * This program is free software; you can redistribute it and/or
 * modify it under the terms of the GNU General Public License as
 * published by the Free Software Foundation; either version 2 of the
 * License, or any later version.
 *
 * This program is distributed in the hope that it will be useful, but
 * WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
 * General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program; if not, write to the Free Software
 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA
 * 02110-1301, USA.
 *
 * You can also choose to distribute this program under the terms of
 * the Unmodified Binary Distribution Licence (as given in the file
 * COPYING.UBDL), provided that you have satisfied its requirements.
 */

FILE_LICENCE ( GPL2_OR_LATER_OR_UBDL );

#include <stdint.h>
#include <string.h>
#include <stdlib.h>
#include <ipxe/uaccess.h>
#include <ipxe/dhcp.h>
#include <ipxe/fakedhcp.h>
#include <ipxe/device.h>
#include <ipxe/netdevice.h>
#include <ipxe/isapnp.h>
#include <ipxe/init.h>
#include <ipxe/if_ether.h>
#include <basemem_packet.h>
#include <biosint.h>
#include <rmsetjmp.h>
#include "pxe.h"
#include "pxe_call.h"

/* Avoid dragging in isapnp.o unnecessarily */
uint16_t isapnp_read_port;

/** Zero-based versions of PXENV_GET_CACHED_INFO::PacketType */
enum pxe_cached_info_indices {
        CACHED_INFO_DHCPDISCOVER = ( PXENV_PACKET_TYPE_DHCP_DISCOVER - 1 ),
        CACHED_INFO_DHCPACK = ( PXENV_PACKET_TYPE_DHCP_ACK - 1 ),
        CACHED_INFO_BINL = ( PXENV_PACKET_TYPE_CACHED_REPLY - 1 ),
        NUM_CACHED_INFOS
};

/** A cached DHCP packet */
union pxe_cached_info {
        struct dhcphdr dhcphdr;
        /* This buffer must be *exactly* the size of a BOOTPLAYER_t
         * structure, otherwise WinPE will die horribly.  It takes the
         * size of *our* buffer and feeds it in to us as the size of
         * one of *its* buffers.  If our buffer is larger than it
         * expects, we therefore end up overwriting part of its data
         * segment, since it tells us to do so.  (D'oh!)
         *
         * Note that a BOOTPLAYER_t is not necessarily large enough to
         * hold a DHCP packet; this is a flaw in the PXE spec.
         */
        BOOTPLAYER_t packet;
} __attribute__ (( packed ));

/** A PXE DHCP packet creator */
struct pxe_dhcp_packet_creator {
        /** Create DHCP packet
         *
         * @v netdev            Network device
         * @v data              Buffer for DHCP packet
         * @v max_len           Size of DHCP packet buffer
         * @ret rc              Return status code
         */
        int ( * create ) ( struct net_device *netdev, void *data,
                           size_t max_len );
};

/** PXE DHCP packet creators */
static struct pxe_dhcp_packet_creator pxe_dhcp_packet_creators[] = {
        [CACHED_INFO_DHCPDISCOVER] = { create_fakedhcpdiscover },
        [CACHED_INFO_DHCPACK] = { create_fakedhcpack },
        [CACHED_INFO_BINL] = { create_fakepxebsack },
};

/**
 * Name PXENV_GET_CACHED_INFO packet type
 *
 * @v packet_type       Packet type
 * @ret name            Name of packet type
 */
static inline __attribute__ (( always_inline )) const char *
pxenv_get_cached_info_name ( int packet_type ) {
        switch ( packet_type ) {
        case PXENV_PACKET_TYPE_DHCP_DISCOVER:
                return "DHCPDISCOVER";
        case PXENV_PACKET_TYPE_DHCP_ACK:
                return "DHCPACK";
        case PXENV_PACKET_TYPE_CACHED_REPLY:
                return "BINL";
        default:
                return "<INVALID>";
        }
}

/* The case in which the caller doesn't supply a buffer is really
 * awkward to support given that we have multiple sources of options,
 * and that we don't actually store the DHCP packets.  (We may not
 * even have performed DHCP; we may have obtained all configuration
 * from non-volatile stored options or from the command line.)
 *
 * Some NBPs rely on the buffers we provide being persistent, so we
 * can't just use the temporary packet buffer.  4.5kB of base memory
 * always wasted just because some clients are too lazy to provide
 * their own buffers...
 */
static union pxe_cached_info __bss16_array ( cached_info, [NUM_CACHED_INFOS] );
#define cached_info __use_data16 ( cached_info )

/**
 * Construct cached DHCP packets
 *
 */
void pxe_fake_cached_info ( void ) {
        struct pxe_dhcp_packet_creator *creator;
        union pxe_cached_info *info;
        unsigned int i;
        int rc;

        /* Sanity check */
        assert ( pxe_netdev != NULL );

        /* Erase any stale packets */
        memset ( cached_info, 0, sizeof ( cached_info ) );

        /* Construct all DHCP packets */
        for ( i = 0 ; i < ( sizeof ( pxe_dhcp_packet_creators ) /
                            sizeof ( pxe_dhcp_packet_creators[0] ) ) ; i++ ) {

                /* Construct DHCP packet */
                creator = &pxe_dhcp_packet_creators[i];
                info = &cached_info[i];
                if ( ( rc = creator->create ( pxe_netdev, info,
                                              sizeof ( *info ) ) ) != 0 ) {
                        DBGC ( &pxe_netdev, " failed to build packet: %s\n",
                               strerror ( rc ) );
                        /* Continue constructing remaining packets */
                }
        }
}

/**
 * UNLOAD BASE CODE STACK
 *
 * @v None                              -
 * @ret ...
 *
 */
static PXENV_EXIT_t
pxenv_unload_stack ( struct s_PXENV_UNLOAD_STACK *unload_stack ) {
        DBGC ( &pxe_netdev, "PXENV_UNLOAD_STACK\n" );

        unload_stack->Status = PXENV_STATUS_SUCCESS;
        return PXENV_EXIT_SUCCESS;
}

/* PXENV_GET_CACHED_INFO
 *
 * Status: working
 */
static PXENV_EXIT_t
pxenv_get_cached_info ( struct s_PXENV_GET_CACHED_INFO *get_cached_info ) {
        union pxe_cached_info *info;
        unsigned int idx;
        size_t len;
        userptr_t buffer;

        DBGC ( &pxe_netdev, "PXENV_GET_CACHED_INFO %s to %04x:%04x+%x",
               pxenv_get_cached_info_name ( get_cached_info->PacketType ),
               get_cached_info->Buffer.segment,
               get_cached_info->Buffer.offset, get_cached_info->BufferSize );

        /* Sanity check */
        idx = ( get_cached_info->PacketType - 1 );
        if ( idx >= NUM_CACHED_INFOS ) {
                DBGC ( &pxe_netdev, " bad PacketType %d\n",
                       get_cached_info->PacketType );
                get_cached_info->Status = PXENV_STATUS_UNSUPPORTED;
                return PXENV_EXIT_FAILURE;
        }
        info = &cached_info[idx];

        /* Copy packet (if applicable) */
        len = get_cached_info->BufferSize;
        if ( len == 0 ) {
                /* Point client at our cached buffer.
                 *
                 * To add to the fun, Intel decided at some point in
                 * the evolution of the PXE specification to add the
                 * BufferLimit field, which we are meant to fill in
                 * with the length of our packet buffer, so that the
                 * caller can safely modify the boot server reply
                 * packet stored therein.  However, this field was not
                 * present in earlier versions of the PXE spec, and
                 * there is at least one PXE NBP (Altiris) which
                 * allocates only exactly enough space for this
                 * earlier, shorter version of the structure.  If we
                 * actually fill in the BufferLimit field, we
                 * therefore risk trashing random areas of the
                 * caller's memory.  If we *don't* fill it in, then
                 * the caller is at liberty to assume that whatever
                 * random value happened to be in that location
                 * represents the length of the buffer we've just
                 * passed back to it.
                 *
                 * Since older PXE stacks won't fill this field in
                 * anyway, it's probably safe to assume that no
                 * callers actually rely on it, so we choose to not
                 * fill it in.
                 */
                get_cached_info->Buffer.segment = rm_ds;
                get_cached_info->Buffer.offset = __from_data16 ( info );
                get_cached_info->BufferSize = sizeof ( *info );
                DBGC ( &pxe_netdev, " using %04x:%04x+%04x['%x']",
                       get_cached_info->Buffer.segment,
                       get_cached_info->Buffer.offset,
                       get_cached_info->BufferSize,
                       get_cached_info->BufferLimit );
        } else {
                /* Copy packet to client buffer */
                if ( len > sizeof ( *info ) )
                        len = sizeof ( *info );
                if ( len < sizeof ( *info ) )
                        DBGC ( &pxe_netdev, " buffer may be too short" );
                buffer = real_to_user ( get_cached_info->Buffer.segment,
                                        get_cached_info->Buffer.offset );
                copy_to_user ( buffer, 0, info, len );
                get_cached_info->BufferSize = len;
        }

        DBGC ( &pxe_netdev, "\n" );
        get_cached_info->Status = PXENV_STATUS_SUCCESS;
        return PXENV_EXIT_SUCCESS;
}

/* PXENV_RESTART_TFTP
 *
 * Status: working
 */
static PXENV_EXIT_t
pxenv_restart_tftp ( struct s_PXENV_TFTP_READ_FILE *restart_tftp ) {
        PXENV_EXIT_t tftp_exit;

        DBGC ( &pxe_netdev, "PXENV_RESTART_TFTP\n" );

        /* Words cannot describe the complete mismatch between the PXE
         * specification and any possible version of reality...
         */
        restart_tftp->Buffer = PXE_LOAD_PHYS; /* Fixed by spec, apparently */
        restart_tftp->BufferSize = ( 0xa0000 - PXE_LOAD_PHYS ); /* Near enough */
        tftp_exit = pxenv_tftp_read_file ( restart_tftp );
        if ( tftp_exit != PXENV_EXIT_SUCCESS )
                return tftp_exit;

        /* Restart NBP */
        rmlongjmp ( pxe_restart_nbp, PXENV_RESTART_TFTP );
}

/* PXENV_START_UNDI
 *
 * Status: working
 */
static PXENV_EXIT_t pxenv_start_undi ( struct s_PXENV_START_UNDI *start_undi ) {
        unsigned int bus_type;
        unsigned int location;
        struct net_device *netdev;

        DBGC ( &pxe_netdev, "PXENV_START_UNDI %04x:%04x:%04x\n",
               start_undi->AX, start_undi->BX, start_undi->DX );

        /* Determine bus type and location.  Use a heuristic to decide
         * whether we are PCI or ISAPnP
         */
        if ( ( start_undi->DX >= ISAPNP_READ_PORT_MIN ) &&
             ( start_undi->DX <= ISAPNP_READ_PORT_MAX ) &&
             ( start_undi->BX >= ISAPNP_CSN_MIN ) &&
             ( start_undi->BX <= ISAPNP_CSN_MAX ) ) {
                bus_type = BUS_TYPE_ISAPNP;
                location = start_undi->BX;
                /* Record ISAPnP read port for use by isapnp.c */
                isapnp_read_port = start_undi->DX;
        } else {
                bus_type = BUS_TYPE_PCI;
                location = start_undi->AX;
        }

        /* Probe for devices, etc. */
        startup();

        /* Look for a matching net device */
        netdev = find_netdev_by_location ( bus_type, location );
        if ( ! netdev ) {
                DBGC ( &pxe_netdev, "PXENV_START_UNDI could not find matching "
                       "net device\n" );
                start_undi->Status = PXENV_STATUS_UNDI_CANNOT_INITIALIZE_NIC;
                return PXENV_EXIT_FAILURE;
        }
        DBGC ( &pxe_netdev, "PXENV_START_UNDI found net device %s\n",
               netdev->name );

        /* Activate PXE */
        pxe_activate ( netdev );

        start_undi->Status = PXENV_STATUS_SUCCESS;
        return PXENV_EXIT_SUCCESS;
}

/* PXENV_STOP_UNDI
 *
 * Status: working
 */
static PXENV_EXIT_t pxenv_stop_undi ( struct s_PXENV_STOP_UNDI *stop_undi ) {
        DBGC ( &pxe_netdev, "PXENV_STOP_UNDI\n" );

        /* Deactivate PXE */
        pxe_deactivate();

        /* Prepare for unload */
        shutdown_boot();

        /* Check to see if we still have any hooked interrupts */
        if ( hooked_bios_interrupts != 0 ) {
                DBGC ( &pxe_netdev, "PXENV_STOP_UNDI failed: %d interrupts "
                       "still hooked\n", hooked_bios_interrupts );
                stop_undi->Status = PXENV_STATUS_KEEP_UNDI;
                return PXENV_EXIT_FAILURE;
        }

        stop_undi->Status = PXENV_STATUS_SUCCESS;
        return PXENV_EXIT_SUCCESS;
}

/* PXENV_START_BASE
 *
 * Status: won't implement (requires major structural changes)
 */
static PXENV_EXIT_t pxenv_start_base ( struct s_PXENV_START_BASE *start_base ) {
        DBGC ( &pxe_netdev, "PXENV_START_BASE\n" );

        start_base->Status = PXENV_STATUS_UNSUPPORTED;
        return PXENV_EXIT_FAILURE;
}

/* PXENV_STOP_BASE
 *
 * Status: working
 */
static PXENV_EXIT_t pxenv_stop_base ( struct s_PXENV_STOP_BASE *stop_base ) {
        DBGC ( &pxe_netdev, "PXENV_STOP_BASE\n" );

        /* The only time we will be called is when the NBP is trying
         * to shut down the PXE stack.  There's nothing we need to do
         * in this call.
         */

        stop_base->Status = PXENV_STATUS_SUCCESS;
        return PXENV_EXIT_SUCCESS;
}

/** PXE preboot API */
struct pxe_api_call pxe_preboot_api[] __pxe_api_call = {
        PXE_API_CALL ( PXENV_UNLOAD_STACK, pxenv_unload_stack,
                       struct s_PXENV_UNLOAD_STACK ),
        PXE_API_CALL ( PXENV_GET_CACHED_INFO, pxenv_get_cached_info,
                       struct s_PXENV_GET_CACHED_INFO ),
        PXE_API_CALL ( PXENV_RESTART_TFTP, pxenv_restart_tftp,
                       struct s_PXENV_TFTP_READ_FILE ),
        PXE_API_CALL ( PXENV_START_UNDI, pxenv_start_undi,
                       struct s_PXENV_START_UNDI ),
        PXE_API_CALL ( PXENV_STOP_UNDI, pxenv_stop_undi,
                       struct s_PXENV_STOP_UNDI ),
        PXE_API_CALL ( PXENV_START_BASE, pxenv_start_base,
                       struct s_PXENV_START_BASE ),
        PXE_API_CALL ( PXENV_STOP_BASE, pxenv_stop_base,
                       struct s_PXENV_STOP_BASE ),
};