initial commit

VtoyTool/vtoytool.c

@@ -0,0 +1,95 @@
+/******************************************************************************
+ * vtoytool.c  ---- ventoy os tool
+ *
+ * Copyright (c) 2020, longpanda <admin@ventoy.net>
+ *
+ * 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; either version 3 of the
+ * License, or (at your option) any later version.
+ * 
+ * This program is distributed in the hope that it will be useful, but
+ * WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
+ * General Public License for more details.
+ * 
+ * You should have received a copy of the GNU General Public License
+ * along with this program; if not, see <http://www.gnu.org/licenses/>.
+ *
+ */
+
+#include <stdio.h>
+#include <stdlib.h>
+#include <string.h>
+#include <unistd.h>
+
+typedef int (*main_func)(int argc, char **argv);
+
+typedef struct cmd_def
+{
+    const char *cmd;
+    main_func func;
+}cmd_def;
+
+int vtoydump_main(int argc, char **argv);
+int vtoydm_main(int argc, char **argv);
+int vtoytool_install(int argc, char **argv);
+int vtoyloader_main(int argc, char **argv);
+
+static char *g_vtoytool_name = NULL;
+static cmd_def g_cmd_list[] = 
+{
+    { "vtoydump",    vtoydump_main    },
+    { "vtoydm",      vtoydm_main      },
+    { "loader",      vtoyloader_main  },
+    { "--install",   vtoytool_install },
+};
+
+
+int vtoytool_install(int argc, char **argv)
+{
+    int i;
+    char toolpath[128];
+    char filepath[128];
+    
+    for (i = 0; i < sizeof(g_cmd_list) / sizeof(g_cmd_list[0]); i++)
+    {
+        if (g_cmd_list[i].cmd[0] != '-')
+        {
+            snprintf(toolpath, sizeof(toolpath), "/ventoy/tool/%s", g_vtoytool_name);
+            snprintf(filepath, sizeof(filepath), "/ventoy/tool/%s", g_cmd_list[i].cmd);
+            link(toolpath, filepath);
+        }
+    }
+    
+    return 0;
+}
+
+int main(int argc, char **argv)
+{
+    int i;
+    
+    if ((g_vtoytool_name = strstr(argv[0], "vtoytool")) != NULL)
+    {
+        argc--;
+        argv++;
+    }
+
+    if (argc == 0)
+    {
+        fprintf(stderr, "Invalid param number\n");
+        return 1;
+    }
+
+    for (i = 0; i < sizeof(g_cmd_list) / sizeof(g_cmd_list[0]); i++)
+    {
+        if (strstr(argv[0], g_cmd_list[i].cmd))
+        {
+            return g_cmd_list[i].func(argc, argv);
+        }
+    }
+
+    fprintf(stderr, "Invalid cmd %s\n", argv[0]);
+    return 1;
+}
+