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
  39 static char *g_vtoytool_name = NULL;
  40 static cmd_def g_cmd_list[] =
  41 {
  42     { "vtoydump",    vtoydump_main    },
  43     { "vtoydm",      vtoydm_main      },
  44     { "loader",      vtoyloader_main  },
  45     { "--install",   vtoytool_install },
  46 };
  47
  48
  49 int vtoytool_install(int argc, char **argv)
  50 {
  51     int i;
  52     char toolpath[128];
  53     char filepath[128];
  54
  55     for (i = 0; i < sizeof(g_cmd_list) / sizeof(g_cmd_list[0]); i++)
  56     {
  57         if (g_cmd_list[i].cmd[0] != '-')
  58         {
  59             snprintf(toolpath, sizeof(toolpath), "/ventoy/tool/%s", g_vtoytool_name);
  60             snprintf(filepath, sizeof(filepath), "/ventoy/tool/%s", g_cmd_list[i].cmd);
  61             link(toolpath, filepath);
  62         }
  63     }
  64
  65     return 0;
  66 }
  67
  68 int main(int argc, char **argv)
  69 {
  70     int i;
  71
  72     if ((g_vtoytool_name = strstr(argv[0], "vtoytool")) != NULL)
  73     {
  74         argc--;
  75         argv++;
  76     }
  77
  78     if (argc == 0)
  79     {
  80         fprintf(stderr, "Invalid param number\n");
  81         return 1;
  82     }
  83
  84     for (i = 0; i < sizeof(g_cmd_list) / sizeof(g_cmd_list[0]); i++)
  85     {
  86         if (strstr(argv[0], g_cmd_list[i].cmd))
  87         {
  88             return g_cmd_list[i].func(argc, argv);
  89         }
  90     }
  91
  92     fprintf(stderr, "Invalid cmd %s\n", argv[0]);
  93     return 1;
  94 }
  95