VtoyTool/vtoytool.c

   1 /******************************************************************************
   2  * vtoytool.c  ---- ventoy os tool
   3  *
   4  * Copyright (c) 2020, longpanda <admin@ventoy.net>
   5  *
   6  * This program is free software; you can redistribute it and/or
   7  * modify it under the terms of the GNU General Public License as
   8  * published by the Free Software Foundation; either version 3 of the
   9  * License, or (at your option) any later version.
  10  *
  11  * This program is distributed in the hope that it will be useful, but
  12  * WITHOUT ANY WARRANTY; without even the implied warranty of
  13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  14  * General Public License for more details.
  15  *
  16  * You should have received a copy of the GNU General Public License
  17  * along with this program; if not, see <http://www.gnu.org/licenses/>.
  18  *
  19  */
  20
  21 #include <stdio.h>
  22 #include <stdlib.h>
  23 #include <string.h>
  24 #include <unistd.h>
  25
  26 typedef int (*main_func)(int argc, char **argv);
  27
  28 typedef struct cmd_def
  29 {
  30     const char *cmd;
  31     main_func func;
  32 }cmd_def;
  33
  34 int vtoydump_main(int argc, char **argv);
  35 int vtoydm_main(int argc, char **argv);
  36 int vtoytool_install(int argc, char **argv);
  37 int vtoyloader_main(int argc, char **argv);
  38 int vtoyvine_main(int argc, char **argv);
  39 int vtoyksym_main(int argc, char **argv);
  40 int vtoykmod_main(int argc, char **argv);
  41 int vtoyexpand_main(int argc, char **argv);
  42
  43 static char *g_vtoytool_name = NULL;
  44 static cmd_def g_cmd_list[] =
  45 {
  46     { "vine_patch_loader",  vtoyvine_main  },
  47     { "vtoydump",    vtoydump_main    },
  48     { "vtoydm",      vtoydm_main      },
  49     { "loader",      vtoyloader_main  },
  50     { "hald",        vtoyloader_main  },
  51     { "vtoyksym",    vtoyksym_main  },
  52     { "vtoykmod",    vtoykmod_main  },
  53     { "vtoyexpand",  vtoyexpand_main  },
  54     { "--install",   vtoytool_install },
  55 };
  56
  57
  58 int vtoytool_install(int argc, char **argv)
  59 {
  60     int i;
  61     char toolpath[128];
  62     char filepath[128];
  63
  64     for (i = 0; i < sizeof(g_cmd_list) / sizeof(g_cmd_list[0]); i++)
  65     {
  66         if (g_cmd_list[i].cmd[0] != '-')
  67         {
  68             snprintf(toolpath, sizeof(toolpath), "/ventoy/tool/%s", g_vtoytool_name);
  69             snprintf(filepath, sizeof(filepath), "/ventoy/tool/%s", g_cmd_list[i].cmd);
  70             link(toolpath, filepath);
  71         }
  72     }
  73
  74     return 0;
  75 }
  76
  77 int main(int argc, char **argv)
  78 {
  79     int i;
  80
  81     if ((g_vtoytool_name = strstr(argv[0], "vtoytool")) != NULL)
  82     {
  83         argc--;
  84         argv++;
  85     }
  86
  87     if (argc == 0)
  88     {
  89         fprintf(stderr, "Invalid param number\n");
  90         return 1;
  91     }
  92
  93     for (i = 0; i < sizeof(g_cmd_list) / sizeof(g_cmd_list[0]); i++)
  94     {
  95         if (strstr(argv[0], g_cmd_list[i].cmd))
  96         {
  97             return g_cmd_list[i].func(argc, argv);
  98         }
  99     }
 100
 101     fprintf(stderr, "Invalid cmd %s\n", argv[0]);
 102     return 1;
 103 }
 104