pxemenu.c File Reference

PXE Boot Menus. More...

#include <stdint.h>
#include <stdlib.h>
#include <stdio.h>
#include <string.h>
#include <errno.h>
#include <ctype.h>
#include <byteswap.h>
#include <curses.h>
#include <ipxe/console.h>
#include <ipxe/dhcp.h>
#include <ipxe/keys.h>
#include <ipxe/timer.h>
#include <ipxe/uri.h>
#include <ipxe/ansicol.h>
#include <usr/dhcpmgmt.h>
#include <usr/autoboot.h>

Go to the source code of this file.

Data Structures
struct	pxe_menu_item
	A PXE boot menu item. More...
struct	pxe_menu
	A PXE boot menu. More...

Functions
	FILE_LICENCE (GPL2_OR_LATER_OR_UBDL)
static int	pxe_menu_parse (struct pxe_menu **menu)
	Parse and allocate PXE boot menu. More...
static void	pxe_menu_draw_item (struct pxe_menu *menu, unsigned int index, int selected)
	Draw PXE boot menu item. More...
static int	pxe_menu_select (struct pxe_menu *menu)
	Make selection from PXE boot menu. More...
static int	pxe_menu_prompt_and_select (struct pxe_menu *menu)
	Prompt for (and make selection from) PXE boot menu. More...
int	pxe_menu_boot (struct net_device *netdev)
	Boot using PXE boot menu. More...

Detailed Description

PXE Boot Menus.

Definition in file pxemenu.c.

Function Documentation

FILE_LICENCE()

FILE_LICENCE

(

GPL2_OR_LATER_OR_UBDL

)

pxe_menu_parse()

static int pxe_menu_parse ( struct pxe_menu **  menu )

static

Parse and allocate PXE boot menu.

Parameters

menu	PXE boot menu to fill in

Return values

rc	Return status code

It is the callers responsibility to eventually free the allocated boot menu.

Definition at line 88 of file pxemenu.c.

                                                     {
        struct setting pxe_boot_menu_prompt_setting =
                { .tag = DHCP_PXE_BOOT_MENU_PROMPT };
        struct setting pxe_boot_menu_setting =
                { .tag = DHCP_PXE_BOOT_MENU };
        uint8_t raw_menu[256];
        int raw_prompt_len;
        int raw_menu_len;
        struct dhcp_pxe_boot_menu *raw_menu_item;
        struct dhcp_pxe_boot_menu_prompt *raw_menu_prompt;
        void *raw_menu_end;
        unsigned int num_menu_items;
        unsigned int i;
        int rc;

        /* Fetch raw menu */
        memset ( raw_menu, 0, sizeof ( raw_menu ) );
        if ( ( raw_menu_len = fetch_raw_setting ( NULL, &pxe_boot_menu_setting,
                                                  raw_menu,
                                                  sizeof ( raw_menu ) ) ) < 0 ){
                rc = raw_menu_len;
                DBG ( "Could not retrieve raw PXE boot menu: %s\n",
                      strerror ( rc ) );
                return rc;
        }
        if ( raw_menu_len >= ( int ) sizeof ( raw_menu ) ) {
                DBG ( "Raw PXE boot menu too large for buffer\n" );
                return -ENOSPC;
        }
        raw_menu_end = ( raw_menu + raw_menu_len );

        /* Fetch raw prompt length */
        raw_prompt_len =
                fetch_raw_setting ( NULL, &pxe_boot_menu_prompt_setting,
                                    NULL, 0 );
        if ( raw_prompt_len < 0 )
                raw_prompt_len = 0;

        /* Count menu items */
        num_menu_items = 0;
        raw_menu_item = ( ( void * ) raw_menu );
        while ( 1 ) {
                if ( ( ( ( void * ) raw_menu_item ) +
                       sizeof ( *raw_menu_item ) ) > raw_menu_end )
                        break;
                if ( ( ( ( void * ) raw_menu_item ) +
                       sizeof ( *raw_menu_item ) +
                       raw_menu_item->desc_len ) > raw_menu_end )
                        break;
                num_menu_items++;
                raw_menu_item = ( ( ( void * ) raw_menu_item ) +
                                  sizeof ( *raw_menu_item ) +
                                  raw_menu_item->desc_len );
        }

        /* Allocate space for parsed menu */
        *menu = zalloc ( sizeof ( **menu ) +
                         ( num_menu_items * sizeof ( (*menu)->items[0] ) ) +
                         raw_menu_len + 1 /* NUL */ +
                         raw_prompt_len + 1 /* NUL */ );
        if ( ! *menu ) {
                DBG ( "Could not allocate PXE boot menu\n" );
                return -ENOMEM;
        }

        /* Fill in parsed menu */
        (*menu)->num_items = num_menu_items;
        raw_menu_item = ( ( ( void * ) (*menu) ) + sizeof ( **menu ) +
                          ( num_menu_items * sizeof ( (*menu)->items[0] ) ) );
        memcpy ( raw_menu_item, raw_menu, raw_menu_len );
        for ( i = 0 ; i < num_menu_items ; i++ ) {
                (*menu)->items[i].type = le16_to_cpu ( raw_menu_item->type );
                (*menu)->items[i].desc = raw_menu_item->desc;
                /* Set type to 0; this ensures that the description
                 * for the previous menu item is NUL-terminated.
                 * (Final item is NUL-terminated anyway.)
                 */
                raw_menu_item->type = 0;
                raw_menu_item = ( ( ( void * ) raw_menu_item ) +
                                  sizeof ( *raw_menu_item ) +
                                  raw_menu_item->desc_len );
        }
        if ( raw_prompt_len ) {
                raw_menu_prompt = ( ( ( void * ) raw_menu_item ) +
                                    1 /* NUL */ );
                fetch_raw_setting ( NULL, &pxe_boot_menu_prompt_setting,
                                    raw_menu_prompt, raw_prompt_len );
                (*menu)->timeout =
                        ( ( raw_menu_prompt->timeout == 0xff ) ?
                          -1 : raw_menu_prompt->timeout );
                (*menu)->prompt = raw_menu_prompt->prompt;
        } else {
                (*menu)->timeout = -1;
        }

        return 0;
}

References DBG, dhcp_pxe_boot_menu::desc, dhcp_pxe_boot_menu::desc_len, DHCP_PXE_BOOT_MENU, DHCP_PXE_BOOT_MENU_PROMPT, ENOMEM, ENOSPC, fetch_raw_setting(), le16_to_cpu, memcpy(), memset(), NULL, dhcp_pxe_boot_menu_prompt::prompt, rc, strerror(), setting::tag, dhcp_pxe_boot_menu_prompt::timeout, dhcp_pxe_boot_menu::type, and zalloc().

Referenced by pxe_menu_boot().

pxe_menu_draw_item()

static void pxe_menu_draw_item ( struct pxe_menu *  menu, unsigned int  index, int  selected  )

static

Draw PXE boot menu item.

Parameters

menu	PXE boot menu
index	Index of item to draw
selected	Item is selected

Definition at line 193 of file pxemenu.c.

                                                                    {
        char buf[COLS+1];
        size_t len;
        unsigned int row;

        /* Prepare space-padded row content */
        len = snprintf ( buf, sizeof ( buf ), " %c. %s",
                         ( 'A' + index ), menu->items[index].desc );
        while ( len < ( sizeof ( buf ) - 1 ) )
                buf[len++] = ' ';
        buf[ sizeof ( buf ) - 1 ] = '\0';

        /* Draw row */
        row = ( LINES - menu->num_items + index - 1 );
        color_set ( ( selected ? CPAIR_PXE : CPAIR_DEFAULT ), NULL );
        mvprintw ( row, 0, "%s", buf );
        move ( row, 1 );
}

References color_set, COLS, CPAIR_DEFAULT, CPAIR_PXE, pxe_menu_item::desc, index, pxe_menu::items, len, LINES, move(), mvprintw, NULL, pxe_menu::num_items, and snprintf().

Referenced by pxe_menu_select().

pxe_menu_select()

static int pxe_menu_select ( struct pxe_menu *  menu )

static

Make selection from PXE boot menu.

Parameters

menu	PXE boot menu

Return values

rc	Return status code

Definition at line 219 of file pxemenu.c.

                                                     {
        int key;
        unsigned int key_selection;
        unsigned int i;
        int rc = 0;

        /* Initialise UI */
        initscr();
        start_color();
        color_set ( CPAIR_DEFAULT, NULL );

        /* Draw initial menu */
        for ( i = 0 ; i < menu->num_items ; i++ )
                printf ( "\n" );
        for ( i = 0 ; i < menu->num_items ; i++ )
                pxe_menu_draw_item ( menu, ( menu->num_items - i - 1 ), 0 );

        while ( 1 ) {

                /* Highlight currently selected item */
                pxe_menu_draw_item ( menu, menu->selection, 1 );

                /* Wait for keyboard input */
                key = getkey ( 0 );

                /* Unhighlight currently selected item */
                pxe_menu_draw_item ( menu, menu->selection, 0 );

                /* Act upon key */
                if ( ( key == CR ) || ( key == LF ) ) {
                        pxe_menu_draw_item ( menu, menu->selection, 1 );
                        break;
                } else if ( ( key == CTRL_C ) || ( key == ESC ) ) {
                        rc = -ECANCELED;
                        break;
                } else if ( key == KEY_UP ) {
                        if ( menu->selection > 0 )
                                menu->selection--;
                } else if ( key == KEY_DOWN ) {
                        if ( menu->selection < ( menu->num_items - 1 ) )
                                menu->selection++;
                } else if ( ( key < KEY_MIN ) &&
                            ( ( key_selection = ( toupper ( key ) - 'A' ) )
                              < menu->num_items ) ) {
                        menu->selection = key_selection;
                        pxe_menu_draw_item ( menu, menu->selection, 1 );
                        break;
                }
        }

        /* Shut down UI */
        endwin();

        return rc;
}

References color_set, CPAIR_DEFAULT, CR, CTRL_C, ECANCELED, endwin(), ESC, getkey(), initscr(), key, KEY_DOWN, KEY_MIN, KEY_UP, LF, NULL, pxe_menu::num_items, printf(), pxe_menu_draw_item(), rc, pxe_menu::selection, start_color, and toupper().

Referenced by pxe_menu_prompt_and_select().

pxe_menu_prompt_and_select()

static int pxe_menu_prompt_and_select ( struct pxe_menu *  menu )

static

Prompt for (and make selection from) PXE boot menu.

Parameters

menu	PXE boot menu

Return values

rc	Return status code

Definition at line 281 of file pxemenu.c.

                                                                {
        unsigned long start = currticks();
        unsigned long now;
        unsigned long elapsed;
        size_t len = 0;
        int key;
        int rc = 0;

        /* Display menu immediately, if specified to do so */
        if ( menu->timeout < 0 ) {
                if ( menu->prompt )
                        printf ( "%s\n", menu->prompt );
                return pxe_menu_select ( menu );
        }

        /* Display prompt, if specified */
        if ( menu->prompt )
                printf ( "%s", menu->prompt );

        /* Wait for timeout, if specified */
        while ( menu->timeout > 0 ) {
                if ( ! len )
                        len = printf ( " (%d)", menu->timeout );
                if ( iskey() ) {
                        key = getkey ( 0 );
                        if ( key == KEY_F8 ) {
                                /* Display menu */
                                printf ( "\n" );
                                return pxe_menu_select ( menu );
                        } else if ( ( key == CTRL_C ) || ( key == ESC ) ) {
                                /* Abort */
                                rc = -ECANCELED;
                                break;
                        } else {
                                /* Stop waiting */
                                break;
                        }
                }
                now = currticks();
                elapsed = ( now - start );
                if ( elapsed >= TICKS_PER_SEC ) {
                        menu->timeout -= 1;
                        do {
                                printf ( "\b \b" );
                        } while ( --len );
                        start = now;
                }
        }

        /* Return with default option selected */
        printf ( "\n" );
        return rc;
}

References CTRL_C, currticks(), ECANCELED, ESC, getkey(), iskey(), key, KEY_F8, len, printf(), pxe_menu::prompt, pxe_menu_select(), rc, start, TICKS_PER_SEC, and pxe_menu::timeout.

Referenced by pxe_menu_boot().

pxe_menu_boot()

int pxe_menu_boot

(

struct net_device *

netdev

)

Boot using PXE boot menu.

Return values

rc	Return status code

Note that a success return status indicates that a PXE boot menu item has been selected, and that the DHCP session should perform a boot server request/ack.

Definition at line 344 of file pxemenu.c.

                                                {
        struct pxe_menu *menu;
        unsigned int pxe_type;
        struct settings *pxebs_settings;
        struct uri *uri;
        int rc;

        /* Parse and allocate boot menu */
        if ( ( rc = pxe_menu_parse ( &menu ) ) != 0 )
                return rc;

        /* Make selection from boot menu */
        if ( ( rc = pxe_menu_prompt_and_select ( menu ) ) != 0 ) {
                free ( menu );
                return rc;
        }
        pxe_type = menu->items[menu->selection].type;

        /* Free boot menu */
        free ( menu );

        /* Return immediately if local boot selected */
        if ( ! pxe_type )
                return 0;

        /* Attempt PXE Boot Server Discovery */
        if ( ( rc = pxebs ( netdev, pxe_type ) ) != 0 )
                return rc;

        /* Fetch next server and filename */
        pxebs_settings = find_settings ( PXEBS_SETTINGS_NAME );
        assert ( pxebs_settings );
        uri = fetch_next_server_and_filename ( pxebs_settings );
        if ( ! uri )
                return -ENOMEM;

        /* Attempt boot */
        rc = uriboot ( uri, NULL, 0, 0, NULL, URIBOOT_NO_SAN );
        uri_put ( uri );
        return rc;
}

References assert(), ENOMEM, fetch_next_server_and_filename(), find_settings(), free, pxe_menu::items, netdev, NULL, pxe_menu_parse(), pxe_menu_prompt_and_select(), pxebs(), PXEBS_SETTINGS_NAME, rc, pxe_menu::selection, pxe_menu_item::type, uri_put(), uriboot(), and URIBOOT_NO_SAN.