arch/all-pc/boot/grub2-aros/util/grub-mknetdir.c

   1 /*
   2   * Copyright (C) 1999,2000,2001,2002,2003,2004,2005,2006,2007,2008,2009,2010,2013  Free Software Foundation, Inc.
   3   *
   4   * GRUB is free software: you can redistribute it and/or modify
   5   * it under the terms of the GNU General Public License as published by
   6   * the Free Software Foundation, either version 3 of the License, or
   7   * (at your option) any later version.
   8   *
   9   * GRUB is distributed in the hope that it will be useful,
  10   * but WITHOUT ANY WARRANTY; without even the implied warranty of
  11   * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  12   * GNU General Public License for more details.
  13   *
  14   * You should have received a copy of the GNU General Public License
  15   * along with GRUB.  If not, see <http://www.gnu.org/licenses/>.
  16 */
  17
  18 #include <config.h>
  19
  20 #include <grub/util/install.h>
  21 #include <grub/emu/config.h>
  22 #include <grub/util/misc.h>
  23
  24 #include <string.h>
  25 #include <errno.h>
  26
  27 #pragma GCC diagnostic ignored "-Wmissing-prototypes"
  28 #pragma GCC diagnostic ignored "-Wmissing-declarations"
  29 #include <argp.h>
  30 #pragma GCC diagnostic error "-Wmissing-prototypes"
  31 #pragma GCC diagnostic error "-Wmissing-declarations"
  32
  33 static char *rootdir = NULL, *subdir = NULL;
  34 static char *debug_image = NULL;
  35
  36 enum
  37   {
  38     OPTION_NET_DIRECTORY = 0x301,
  39     OPTION_SUBDIR,
  40     OPTION_DEBUG,
  41     OPTION_DEBUG_IMAGE
  42   };
  43
  44 static struct argp_option options[] = {
  45   GRUB_INSTALL_OPTIONS,
  46   {"net-directory", OPTION_NET_DIRECTORY, N_("DIR"),
  47    0, N_("root directory of TFTP server"), 2},
  48   {"subdir", OPTION_SUBDIR, N_("DIR"),
  49    0, N_("relative subdirectory on network server"), 2},
  50   {"debug", OPTION_DEBUG, 0, OPTION_HIDDEN, 0, 2},
  51   {"debug-image", OPTION_DEBUG_IMAGE, N_("STRING"), OPTION_HIDDEN, 0, 2},
  52   {0, 0, 0, 0, 0, 0}
  53 };
  54
  55 static error_t
  56 argp_parser (int key, char *arg, struct argp_state *state)
  57 {
  58   if (grub_install_parse (key, arg))
  59     return 0;
  60   switch (key)
  61     {
  62     case OPTION_NET_DIRECTORY:
  63       free (rootdir);
  64       rootdir = xstrdup (arg);
  65       return 0;
  66     case OPTION_SUBDIR:
  67       free (subdir);
  68       subdir = xstrdup (arg);
  69       return 0;
  70       /* This is an undocumented feature...  */
  71     case OPTION_DEBUG:
  72       verbosity++;
  73       return 0;
  74     case OPTION_DEBUG_IMAGE:
  75       free (debug_image);
  76       debug_image = xstrdup (arg);
  77       return 0;
  78
  79     case ARGP_KEY_ARG:
  80     default:
  81       return ARGP_ERR_UNKNOWN;
  82     }
  83 }
  84
  85
  86 struct argp argp = {
  87   options, argp_parser, NULL,
  88   "\v"N_("Prepares GRUB network boot images at net_directory/subdir "
  89          "assuming net_directory being TFTP root."),
  90   NULL, grub_install_help_filter, NULL
  91 };
  92
  93 static char *base;
  94
  95 static const struct
  96 {
  97   const char *mkimage_target;
  98   const char *netmodule;
  99   const char *ext;
 100 } targets[GRUB_INSTALL_PLATFORM_MAX] =
 101   {
 102     [GRUB_INSTALL_PLATFORM_I386_PC] = { "i386-pc-pxe", "pxe", ".0" },
 103     [GRUB_INSTALL_PLATFORM_SPARC64_IEEE1275] = { "sparc64-ieee1275-aout", "ofnet", ".img" },
 104     [GRUB_INSTALL_PLATFORM_I386_IEEE1275] = { "i386-ieee1275", "ofnet", ".elf" },
 105     [GRUB_INSTALL_PLATFORM_POWERPC_IEEE1275] = { "powerpc-ieee1275", "ofnet", ".elf" },
 106     [GRUB_INSTALL_PLATFORM_I386_EFI] = { "i386-efi", "efinet", ".efi" },
 107     [GRUB_INSTALL_PLATFORM_X86_64_EFI] = { "x86_64-efi", "efinet", ".efi" },
 108     [GRUB_INSTALL_PLATFORM_IA64_EFI] = { "ia64-efi", "efinet", ".efi" },
 109     [GRUB_INSTALL_PLATFORM_ARM_EFI] = { "arm-efi", "efinet", ".efi" }
 110   };
 111
 112 static void
 113 process_input_dir (const char *input_dir, enum grub_install_plat platform)
 114 {
 115   char *platsub = grub_install_get_platform_name (platform);
 116   char *grubdir = grub_util_path_concat (3, rootdir, subdir, platsub);
 117   char *load_cfg = grub_util_path_concat (2, grubdir, "load.cfg");
 118   char *prefix;
 119   char *output;
 120   char *grub_cfg;
 121   FILE *cfg;
 122
 123   grub_install_copy_files (input_dir, base, platform);
 124   grub_util_unlink (load_cfg);
 125
 126   if (debug_image)
 127     {
 128       FILE *f = grub_util_fopen (load_cfg, "wb");
 129       if (!f)
 130         grub_util_error (_("cannot open `%s': %s"), load_cfg,
 131                          strerror (errno));
 132       fprintf (f, "set debug='%s'\n", debug_image);
 133       fclose (f);
 134     }
 135   else
 136     {
 137       free (load_cfg);
 138       load_cfg = 0;
 139     }
 140
 141   prefix = xasprintf ("/%s", subdir);
 142   if (!targets[platform].mkimage_target)
 143     grub_util_error (_("unsupported platform %s\n"), platsub);
 144
 145   grub_cfg = grub_util_path_concat (2, grubdir, "grub.cfg");
 146   cfg = grub_util_fopen (grub_cfg, "wb");
 147   if (!cfg)
 148     grub_util_error (_("cannot open `%s': %s"), grub_cfg,
 149                      strerror (errno));
 150   fprintf (cfg, "source %s/grub.cfg", subdir);
 151   fclose (cfg);
 152
 153   grub_install_push_module (targets[platform].netmodule);
 154
 155   output = grub_util_path_concat_ext (2, grubdir, "core", targets[platform].ext);
 156   grub_install_make_image_wrap (input_dir, prefix, output,
 157                                 0, load_cfg,
 158                                 targets[platform].mkimage_target, 0);
 159   grub_install_pop_module ();
 160
 161   /* TRANSLATORS: First %s is replaced by platform name. Second one by filename.  */
 162   printf (_("Netboot directory for %s created. Configure your DHCP server to point to %s\n"),
 163           platsub, output);
 164
 165   free (platsub);
 166   free (output);
 167   free (prefix);
 168   free (grub_cfg);
 169   free (grubdir);
 170 }
 171
 172
 173 int
 174 main (int argc, char *argv[])
 175 {
 176   const char *pkglibdir;
 177
 178   grub_util_host_init (&argc, &argv);
 179   grub_util_disable_fd_syncs ();
 180   rootdir = xstrdup ("/srv/tftp");
 181   pkglibdir = grub_util_get_pkglibdir ();
 182
 183   subdir = grub_util_path_concat (2, GRUB_BOOT_DIR_NAME, GRUB_DIR_NAME);
 184
 185   argp_parse (&argp, argc, argv, 0, 0, 0);
 186
 187   base = grub_util_path_concat (2, rootdir, subdir);
 188   /* Create the GRUB directory if it is not present.  */
 189
 190   grub_install_mkdir_p (base);
 191
 192   grub_install_push_module ("tftp");
 193
 194   if (!grub_install_source_directory)
 195     {
 196       enum grub_install_plat plat;
 197
 198       for (plat = 0; plat < GRUB_INSTALL_PLATFORM_MAX; plat++)
 199         if (targets[plat].mkimage_target)
 200           {
 201             char *platdir = grub_util_path_concat (2, pkglibdir,
 202                                                    grub_install_get_platform_name (plat));
 203
 204             grub_util_info ("Looking for `%s'", platdir);
 205
 206             if (!grub_util_is_directory (platdir))
 207               {
 208                 free (platdir);
 209                 continue;
 210               }
 211             process_input_dir (platdir, plat);
 212           }
 213     }
 214   else
 215     {
 216       enum grub_install_plat plat;
 217       plat = grub_install_get_target (grub_install_source_directory);
 218       process_input_dir (grub_install_source_directory, plat);
 219     }
 220   return 0;
 221 }