Run Nindent on com32/modules/ifcpu64.c
[people/sha0/syslinux.git] / com32 / modules / ifcpu64.c
1 /* ----------------------------------------------------------------------- *
2  *
3  *   Copyright 2008 H. Peter Anvin - All Rights Reserved
4  *
5  *   This program is free software; you can redistribute it and/or modify
6  *   it under the terms of the GNU General Public License as published by
7  *   the Free Software Foundation, Inc., 51 Franklin St, Fifth Floor,
8  *   Boston MA 02110-1301, USA; either version 2 of the License, or
9  *   (at your option) any later version; incorporated herein by reference.
10  *
11  * ----------------------------------------------------------------------- */
12
13 /*
14  * ifcpu64.c
15  *
16  * Run one command if the CPU has 64-bit support, and another if it doesn't.
17  * Eventually this and other features should get folded into some kind
18  * of scripting engine.
19  *
20  * Usage:
21  *
22  *    label boot_kernel
23  *        kernel ifcpu64.c
24  *        append boot_kernel_64 [-- boot_kernel_32pae] -- boot_kernel_32
25  *    label boot_kernel_32
26  *        kernel vmlinuz_32
27  *        append ...
28  *    label boot_kernel_64
29  *        kernel vmlinuz_64
30  *        append ...
31  */
32
33 #include <alloca.h>
34 #include <stdlib.h>
35 #include <string.h>
36 #include <cpuid.h>
37 #include <syslinux/boot.h>
38
39 static bool __constfunc cpu_has_cpuid(void)
40 {
41     return cpu_has_eflag(X86_EFLAGS_ID);
42 }
43
44 static bool __constfunc cpu_has_level(uint32_t level)
45 {
46     uint32_t group;
47     uint32_t limit;
48
49     if (!cpu_has_cpuid())
50         return false;
51
52     group = level & 0xffff0000;
53     limit = cpuid_eax(group);
54
55     if ((limit & 0xffff0000) != group)
56         return false;
57
58     if (level > limit)
59         return false;
60
61     return true;
62 }
63
64 /* This only supports feature groups 0 and 1, corresponding to the
65    Intel and AMD EDX bit vectors.  We can add more later if need be. */
66 static bool __constfunc cpu_has_feature(int x)
67 {
68     uint32_t level = ((x & 1) << 31) | 1;
69
70     return cpu_has_level(level) && ((cpuid_edx(level) >> (x & 31) & 1));
71 }
72
73 /* XXX: this really should be librarized */
74 static void boot_args(char **args)
75 {
76     int len = 0;
77     char **pp;
78     const char *p;
79     char c, *q, *str;
80
81     for (pp = args; *pp; pp++)
82         len += strlen(*pp);
83
84     q = str = alloca(len + 1);
85     for (pp = args; *pp; pp++) {
86         p = *pp;
87         while ((c = *p++))
88             *q++ = c;
89     }
90     *q = '\0';
91
92     if (!str[0])
93         syslinux_run_default();
94     else
95         syslinux_run_command(str);
96 }
97
98 int main(int argc, char *argv[])
99 {
100     char **args[3];
101     int i;
102     int n;
103
104     args[0] = &argv[1];
105     n = 1;
106     for (i = 1; i < argc; i++) {
107         if (!strcmp(argv[i], "--")) {
108             argv[i] = NULL;
109             args[n++] = &argv[i + 1];
110         }
111         if (n >= 3)
112             break;
113     }
114     while (n < 3) {
115         args[n] = args[n - 1];
116         n++;
117     }
118
119     boot_args(cpu_has_feature(X86_FEATURE_LM) ? args[0] :
120               cpu_has_feature(X86_FEATURE_PAE) ? args[1] : args[2]);
121     return -1;
122 }