source: bootcd/isolinux/syslinux-6.03/com32/modules/ifcpu.c @ dd1be7c

/* ----------------------------------------------------------------------- *
 *
 *   Copyright 2009 Erwan Velu - All Rights Reserved
 *
 *   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, Inc., 51 Franklin St, Fifth Floor,
 *   Boston MA 02110-1301, USA; either version 2 of the License, or
 *   (at your option) any later version; incorporated herein by reference.
 *
 * ----------------------------------------------------------------------- */

/*
 * ifcpu.c
 *
 */

#include <alloca.h>
#include <stdlib.h>
#include <stdio.h>
#include <string.h>
#include <cpuid.h>
#include <unistd.h>
#include <syslinux/boot.h>
#include <com32.h>
#include <consoles.h>

static inline void error(const char *msg)
{
    fputs(msg, stderr);
}

static void usage(void) 
{
 error("Run one command if system match some CPU features, another if it doesn't. \n"
 "Usage: \n"
 "   label ifcpu \n"
 "       com32 ifcpu.c32 \n"
 "       append <option> <cpu_features> -- boot_entry_1 -- boot_entry_2 \n"
 "   label boot_entry_1 \n"
 "        kernel vmlinuz_entry1 \n"
 "        append ... \n"
 "   label boot_entry_2 \n"
 "       kernel vmlinuz_entry2 \n"
 "       append ... \n"
 "\n"
 "options could be :\n"
 "   debug     : display some debugging messages \n"
 "   dry-run   : just do the detection, don't boot \n"
 "\n"
 "cpu_features could be:\n"
 "   64         : Processor is x86_64 compatible (lm cpu flag)\n"
 "   hvm        : Processor features hardware virtualization (hvm or svm cpu flag)\n"
 "   multicore  : Processor must be multi-core \n"
 "   smp        : System must be multi-processor \n"
 "   pae        : Processor features Physical Address Extension (PAE)\n"
 "   hypervisor : Processor is running under an hypervisor\n"
 "\n"
 "if you want to match many cpu features, just separate them with a single space.\n");
}

/* XXX: this really should be librarized */
static void boot_args(char **args)
{
    int len = 0, a = 0;
    char **pp;
    const char *p;
    char c, *q, *str;

    for (pp = args; *pp; pp++)
        len += strlen(*pp) + 1;

    q = str = alloca(len);
    for (pp = args; *pp; pp++) {
        p = *pp;
        while ((c = *p++))
            *q++ = c;
        *q++ = ' ';
        a = 1;
    }
    q -= a;
    *q = '\0';

    if (!str[0])
        syslinux_run_default();
    else
        syslinux_run_command(str);
}

#define show_bool(mybool) mybool ? "found":"not found"

int main(int argc, char *argv[])
{
    char **args[3];
    int i=0;
    int n=0;
    bool hardware_matches = true;
    bool multicore = false;
    bool dryrun = false;
    bool debug = false;

    s_cpu cpu;
    console_ansi_raw();
    detect_cpu(&cpu);

    /* If no argument got passed, let's show the usage */
    if (argc == 1) {
            usage();
            return -1;
    }

    for (i = 1; i < argc; i++) {
        if (!strcmp(argv[i], "--")) {
            argv[i] = NULL;
            args[n++] = &argv[i + 1];
        } else if (!strcmp(argv[i], "64")) {
            if (debug)
                printf(" 64bit      : %s on this system\n",
                       show_bool(cpu.flags.lm));
            hardware_matches = cpu.flags.lm && hardware_matches;
        } else if (!strcmp(argv[i], "pae")) {
            if (debug)
                printf(" pae        : %s on this system\n",
                       show_bool(cpu.flags.pae));
            hardware_matches = cpu.flags.pae && hardware_matches;
        } else if (!strcmp(argv[i], "hvm")) {
            if (debug)
                printf(" hvm        : %s on this system\n",
                       show_bool((cpu.flags.vmx || cpu.flags.svm)));
            hardware_matches = (cpu.flags.vmx || cpu.flags.svm)
                && hardware_matches;
        } else if (!strcmp(argv[i], "multicore")) {
            if (debug)
                printf(" multicore  : %d cores on this system\n", cpu.num_cores);
            if (cpu.num_cores > 1)
                multicore = true;
            hardware_matches = multicore && hardware_matches;
        } else if (!strcmp(argv[i], "smp")) {
            if (debug)
                printf(" smp        : %s on this system\n", show_bool(cpu.flags.smp));
            hardware_matches = cpu.flags.smp && hardware_matches;
        } else if (!strcmp(argv[i], "hypervisor")) {
            if (debug)
                printf(" hypervisor : %s on this system\n", show_bool(cpu.flags.hypervisor));
            hardware_matches = cpu.flags.hypervisor && hardware_matches;
        } else if (!strcmp(argv[i], "dry-run")) {
            dryrun = true;
        } else if (!strcmp(argv[i], "debug")) {
            debug = true;
        }
        if (n >= 2)
            break;
    }
    while (n < 2) {
        args[n] = args[n - 1];
        n++;
    }
    if (debug) {
        printf("\nBooting labels are : '%s' or '%s'\n", *args[0], *args[1]);
        printf("Hardware requirements%smatch this system, let's booting '%s'\n",
               hardware_matches ? " " : " doesn't ",
               hardware_matches ? *args[0] : *args[1]);
        printf("Sleeping 5sec before booting\n");
        if (!dryrun)
            sleep(5);
    }

    if (!dryrun)
        boot_args(hardware_matches ? args[0] : args[1]);
    else
        printf("Dry-run mode, let's exiting\n");

    return -1;
}