From 3916c7f46236e4d477b8ba680ed7cc456ca5c1f9 Mon Sep 17 00:00:00 2001
From: Madura <madura.x86@gmail.com>
Date: Sat, 20 May 2017 02:10:35 +0530
Subject: added floader files

---
 src/floader/README.md |   9 +++
 src/floader/loader.c  | 161 ++++++++++++++++++++++++++++++++++++++++++++++++++
 src/floader/makefile  |   7 +++
 3 files changed, 177 insertions(+)
 create mode 100644 src/floader/README.md
 create mode 100644 src/floader/loader.c
 create mode 100644 src/floader/makefile

diff --git a/src/floader/README.md b/src/floader/README.md
new file mode 100644
index 000000000..d437763a7
--- /dev/null
+++ b/src/floader/README.md
@@ -0,0 +1,9 @@
+READ ME
+-------
+
+* Run 'make'
+* Add comma separated process names to ~/.loader.conf
+* export LD_PRELOAD=<path>./loader.so (ideally to .bashrc)
+* Run any application within shell
+
+
diff --git a/src/floader/loader.c b/src/floader/loader.c
new file mode 100644
index 000000000..0970794e9
--- /dev/null
+++ b/src/floader/loader.c
@@ -0,0 +1,161 @@
+/*
+ * Copyright (C) 2017 Madura A. (madura.x86@gmail.com)
+ * 
+ */
+#include <sys/types.h>
+#include <sys/stat.h>
+#include <sys/mman.h>
+#include <fcntl.h>
+#include <unistd.h>
+    
+#include <string.h>
+#include <stdio.h>
+#include <stdlib.h>
+#include <ctype.h>
+
+#define MAX_MATCHES 32
+#define MAX_ARGS 1024
+#define MAX_ARGS_LEN 4096
+static void loader_main() __attribute__((constructor));
+
+char cmdline[MAX_ARGS_LEN];
+char *args[MAX_ARGS];
+char loader[] = "firejail";
+char confFile[256];
+char *names[MAX_MATCHES];
+
+#ifdef DEBUG
+#define DBG printf
+#else
+#define DBG
+#endif
+void remove_trailing_spaces(char *str)
+{
+    while (!isspace(*str))
+    {
+        str++;
+    }
+    
+    while (*str != '\0')
+    {
+        *str = '\0';
+        str++;
+    }
+}
+
+void read_cmdline()
+{
+    int fd = open("/proc/self/cmdline", O_RDONLY);
+    ssize_t ret = 0, total = 0;
+    char* wcmdbuf = cmdline;
+    while ((ret = read(fd, wcmdbuf, 1)) != 0)
+    {
+        wcmdbuf++;
+        total += ret;
+        if (total > MAX_ARGS_LEN)
+        {
+            printf("Not enough memory\n");
+            close(fd);
+            return ;
+        }
+    }
+    close(fd);
+}
+
+void make_args()
+{
+    int cI = 0, argI=0;
+    char* argstart = &cmdline[0];
+    for (;cI<MAX_ARGS_LEN;cI++)
+    {
+        if (cmdline[cI] == '\0')
+        {
+            args[argI]= argstart; 
+            argstart = &cmdline[cI+1];
+            argI++;
+            if (*argstart  == '\0')
+            {
+                break;
+            }
+        }
+    }
+    args[argI] = argstart;
+    argI++;
+    args[argI] = NULL;
+}
+
+void loader_main()
+{
+    snprintf(confFile, 255, "%s/.loader.conf", getenv("HOME"));
+
+    struct stat confFileStat;
+    
+    stat(confFile, &confFileStat);
+    
+    int confFd = open(confFile, O_RDONLY);
+    
+    if (confFd == -1)
+    {
+        close(confFd);
+        return;
+    }
+    char* conf = (char*) malloc(confFileStat.st_size);
+    if (conf == NULL)
+    {
+        close(confFd);
+        return;
+    }
+    ssize_t ret  = read(confFd, conf, confFileStat.st_size);
+    if (ret == -1)
+    {
+        close(confFd);
+        return;
+    }
+    
+    close(confFd);
+    size_t fI = 0;
+    int matchId = 0;
+    names[matchId] = conf;
+    matchId++;
+    for (;fI < confFileStat.st_size-1;fI++)
+    {
+        if (conf[fI] == ',')
+        {
+            names[matchId] = &conf[fI+1];
+            conf[fI] = '\0';
+            
+            matchId++;
+        }
+    }
+       
+    remove_trailing_spaces(names[matchId-1]);
+   
+    read_cmdline();
+    
+    make_args();
+    
+#ifdef DEBUG
+    int xarg=0;
+    while (args[xarg] != NULL)
+    {
+        DBG(".%s\n", args[xarg]);
+        xarg++;
+    }
+#endif
+
+    int x;
+       
+    for (x = 0;x<matchId;x++)
+    {
+        DBG("%s\n",names[x]);
+        if (strstr(args[0], names[x]) != NULL)
+        {
+            DBG("highjack!\n");
+            
+            free(conf);
+            
+            execvp(loader, args );
+        }
+    }
+    
+}
diff --git a/src/floader/makefile b/src/floader/makefile
new file mode 100644
index 000000000..0de6a3138
--- /dev/null
+++ b/src/floader/makefile
@@ -0,0 +1,7 @@
+all:
+	gcc -ggdb -shared -fPIC loader.c -o loader.so
+
+debug:
+	gcc -ggdb -shared -DDEBUG -fPIC loader.c -o loader.so
+
+
-- 
cgit v1.2.3-54-g00ecf