com32/modules/gpxecmd.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  * gpxecmd.c
  15  *
  16  * Invoke an arbitrary gPXE command, if available.
  17  */
  18
  19 #include <alloca.h>
  20 #include <inttypes.h>
  21 #include <stdio.h>
  22 #include <console.h>
  23 #include <com32.h>
  24 #include <string.h>
  25
  26 #include <sys/gpxe.h>
  27 #include <syslinux/pxe_api.h>
  28
  29 struct segoff16 {
  30     uint16_t offs, seg;
  31 };
  32
  33 struct s_PXENV_FILE_EXEC {
  34     uint16_t Status;
  35     struct segoff16 Command;
  36 };
  37
  38 static void gpxecmd(const char **args)
  39 {
  40     char *q;
  41     struct s_PXENV_FILE_EXEC *fx;
  42
  43     fx = lmalloc(sizeof *fx);
  44     if (!fx)
  45         return;
  46
  47     q = (char *)(fx + 1);
  48
  49     fx->Status = 1;
  50     fx->Command.offs = OFFS(q);
  51     fx->Command.seg = SEG(q);
  52
  53     while (*args) {
  54         q = stpcpy(q, *args);
  55         *q++ = ' ';
  56         args++;
  57     }
  58     *--q = '\0';
  59
  60     pxe_call(PXENV_FILE_EXEC, fx);
  61
  62     /* This should not return... */
  63 }
  64
  65 int main(int argc, const char *argv[])
  66 {
  67     if (argc < 2) {
  68         printf("Usage: gpxecmd command...\n");
  69         return 1;
  70     }
  71
  72     if (!is_gpxe()) {
  73         printf("gpxecmd: gPXE API not detected\n");
  74         return 1;
  75     }
  76
  77     gpxecmd(argv + 1);
  78
  79     return 0;
  80 }