initial commit

VtoyTool/vtoyloader.c

@@ -0,0 +1,172 @@
+/******************************************************************************
+ * vtoyloader.c  ---- ventoy loader (wapper for binary loader)
+ *
+ * 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 <errno.h>
+#include <unistd.h>
+#include <fcntl.h>
+#include <sys/types.h>
+#include <sys/mman.h>
+#include <sys/ioctl.h>
+#include <sys/stat.h>
+#include <sys/types.h>
+
+#define MAX_EXT_PARAM       256
+#define CMDLINE_BUF_LEN     (1024 * 1024 * 1)
+#define EXEC_PATH_FILE      "/ventoy/loader_exec_file"
+#define CMDLINE_FILE        "/ventoy/loader_exec_cmdline"
+#define HOOK_CMD_FILE       "/ventoy/loader_hook_cmd"
+#define DEBUG_FLAG_FILE     "/ventoy/loader_debug"
+
+static int verbose = 0;
+#define debug(fmt, ...) if(verbose) printf(fmt, ##__VA_ARGS__)
+
+static char g_exec_file[512];
+static char g_hook_cmd[512];
+
+static int vtoy_read_file_to_buf(const char *file, void *buf, int buflen)
+{
+    FILE *fp;
+
+    fp = fopen(file, "r");
+    if (!fp)
+    {
+        fprintf(stderr, "Failed to open file %s err:%d\n", file, errno);
+        return 1;
+    }
+    fread(buf, 1, buflen, fp);
+    fclose(fp);
+
+    return 0;
+}
+
+int vtoyloader_main(int argc, char **argv)
+{
+    int i;
+    int len;
+    int rc;
+    char *pos;
+    char *cmdline;
+    char **cmdlist;
+
+    if (access(DEBUG_FLAG_FILE, F_OK) >= 0)
+    {
+        verbose = 1;
+    }
+
+    debug("ventoy loader ...\n");
+    
+    rc = vtoy_read_file_to_buf(EXEC_PATH_FILE, g_exec_file, sizeof(g_exec_file) - 1);
+    if (rc)
+    {
+        return rc;
+    }
+
+    if (access(g_exec_file, F_OK) < 0)
+    {
+        fprintf(stderr, "File %s not exist\n", g_exec_file);
+        return 1;
+    }
+
+    if (access(HOOK_CMD_FILE, F_OK) >= 0)
+    {
+        rc = vtoy_read_file_to_buf(HOOK_CMD_FILE,  g_hook_cmd,  sizeof(g_hook_cmd) - 1);
+        debug("g_hook_cmd=<%s>\n", g_hook_cmd);
+
+        // call hook script
+        rc = system(g_hook_cmd);
+        debug("system return code =<%d>  errno=<%d>\n", rc, errno);
+    }
+
+    cmdline = (char *)malloc(CMDLINE_BUF_LEN);
+    if (!cmdline)
+    {
+        fprintf(stderr, "Failed to alloc memory err:%d\n", errno);
+        return 1;
+    }
+    memset(cmdline, 0, CMDLINE_BUF_LEN);
+
+    if (access(CMDLINE_FILE, F_OK) >= 0)
+    {
+        rc = vtoy_read_file_to_buf(CMDLINE_FILE, cmdline, CMDLINE_BUF_LEN - 1);
+        if (rc)
+        {
+            return rc;
+        }
+    }
+
+    len = (int)((argc + MAX_EXT_PARAM) * sizeof(char *));
+    cmdlist = (char **)malloc(len);
+    if (!cmdlist)
+    {
+        free(cmdline);
+        fprintf(stderr, "Failed to alloc memory err:%d\n", errno);
+        return 1;
+    }
+    memset(cmdlist, 0, len);
+
+    for (i = 0; i < argc; i++)
+    {
+        cmdlist[i] = argv[i];
+    }
+    
+    cmdlist[0] = g_exec_file;
+    debug("g_exec_file=<%s>\n", g_exec_file);
+
+    pos = cmdline;
+    while ((*pos) && i < MAX_EXT_PARAM)
+    {
+        cmdlist[i++] = pos;
+
+        while (*pos)
+        {
+            if (*pos == '\r')
+            {
+                *pos = ' ';
+            }
+            else if (*pos == '\n')
+            {
+                *pos = 0;
+                pos++;
+                break;
+            }
+        
+            pos++;
+        }
+    }
+
+    debug("execv [%s]...\n", cmdlist[0]);
+
+    execv(cmdlist[0], cmdlist);
+    
+    return 0;
+}
+
+
+// wrapper main
+#ifndef BUILD_VTOY_TOOL
+int main(int argc, char **argv)
+{
+    return vtoyloader_main(argc, argv);
+}
+#endif
+