aboutsummaryrefslogtreecommitdiffstats
path: root/src/post_menu.c
blob: caba7da134ac5aa12dc9ec11f602626165779375 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
// Menu presented during final phase of "post".
//
// Copyright (C) 2008  Kevin O'Connor <kevin@koconnor.net>
// Copyright (C) 2002  MandrakeSoft S.A.
//
// This file may be distributed under the terms of the GNU GPLv3 license.

#include "biosvar.h" // GET_EBDA
#include "util.h" // usleep
#include "bregs.h" // struct bregs

static u8
check_for_keystroke()
{
    struct bregs br;
    memset(&br, 0, sizeof(br));
    br.ah = 1;
    call16_int(0x16, &br);
    return !(br.flags & F_ZF);
}

static u8
get_keystroke()
{
    struct bregs br;
    memset(&br, 0, sizeof(br));
    call16_int(0x16, &br);
    return br.ah;
}

static void
udelay_and_check_for_keystroke(u32 usec, int count)
{
    int i;
    for (i = 1; i <= count; i++) {
        usleep(usec);
        if (check_for_keystroke())
            break;
    }
}

void
interactive_bootmenu()
{
    if (! CONFIG_BOOTMENU)
        return;

    while (check_for_keystroke())
        get_keystroke();

    printf("Press F12 for boot menu.\n\n");

    udelay_and_check_for_keystroke(500000, 5);
    if (! check_for_keystroke())
        return;
    u8 scan_code = get_keystroke();
    if (scan_code != 0x58)
        /* not F12 */
        return;

    while (check_for_keystroke())
        get_keystroke();

    printf("Select boot device:\n\n");

    int count = GET_EBDA(ipl.count);
    int i;
    for (i = 0; i < count; i++) {
        printf("%d. ", i+1);
        printf_bootdev(i);
        printf("\n");
    }

    for (;;) {
        scan_code = get_keystroke();
        if (scan_code == 0x01 || scan_code == 0x58)
            /* ESC or F12 */
            break;
        if (scan_code <= count + 1) {
            // Add user choice to the boot order.
            u16 choice = scan_code - 1;
            u32 bootorder = GET_EBDA(ipl.bootorder);
            SET_EBDA(ipl.bootorder, (bootorder << 4) | choice);
            break;
        }
    }
    printf("\n");
}