Grates: Add launch scripts

Author: stupendoussuperpowers

.gitignore

@@ -9,3 +9,5 @@ clang+llvm-*
 #Ignore Build artifacts
 src/glibc/build/
 src/tmp/
+
+.DS_Store

example_grate/README.md

@@ -1,20 +1,35 @@
-This is an example of a grate written with the new APIs. Runs the in-memory file system grate with an example program.
+This folder contains example grate implementations, as well as an example runner for grate-cage configurations.
 
 ### Compile
 
-`./tools/compile.sh`
+`./tools/compile.sh <grate source files>`
 
 ### Run
 
-- Compile `runopen.c` into `LIND_ROOT`
-- `./tools/run.sh`
+- Compile all cages present in the `cage/` folder, and place them into `LIND_ROOT` using the `scripts/lind_compile.sh` command.
+- Compile all grates present in the `grates/` folder, and place them into `LIND_ROOT`.
+- Run an example grate configuration using `./tools/run.sh`. This will run the following configuration:
+
+```
+geteuid_grate:
+    getegid_grate.wasm:
+        bash -c "etest & imfs_grate runopen & getgid_grate gidtest" 
+```
 
 ### File Structure 
 
 ```
-imfs_grate.c // Grate source code with wrappers for individual syscalls.
-imfs.* // IMFS source code.
-runopen.c // Sample cage for testing
+grates/
+    imfs_grate.c // Grate source code with wrappers for individual syscalls.
+    imfs.* // IMFS source code.
+    ... // Other Grate examples 
+
+cages/
+    mash.c // Barebones, minimal bash source code. 
+    etest.c 
+    runopen.c 
+    ... // Example cage source code.
+
 syscalls // List of syscalls along with the arguments required.
 
 tools/

example_grate/cages/etest.c

@@ -0,0 +1,11 @@
+#include <stdio.h>
+#include <unistd.h>
+
+int main(void) {
+    int cageid = getpid();
+    printf("-------ETEST %d ------\n", cageid);
+    printf("getegid() = %d\n", (int)getegid());
+    printf("geteuid() = %d\n", (int)geteuid());
+    printf("----(EXIT) ETEST----\n");
+    return 0;
+}

example_grate/cages/gidtest.c

@@ -0,0 +1,10 @@
+#include <stdio.h>
+#include <unistd.h>
+
+int main(void) {
+    int cageid = getpid();
+    printf("--- GIDTEST %d ---\n", cageid);
+    printf("getgid() = %d\n", (int)getgid());
+    printf("--EXIT GIDTEST-\n");
+    return 0;
+}

example_grate/cages/mash.c

@@ -0,0 +1,100 @@
+#include <stdio.h>
+#include <stdlib.h>
+#include <string.h>
+#include <unistd.h>
+#include <ctype.h>
+#include <sys/wait.h>
+
+#include <fcntl.h>
+
+#include <dirent.h>
+
+static char *trim(char *s) {
+    while (isspace(*s)) s++;
+    if (*s == 0) return s;
+    char *end = s + strlen(s) - 1;
+    while (end > s && isspace(*end)) end--;
+    end[1] = '\0';
+    return s;
+}
+
+// split a string like "cmd arg1 arg2" into argv[]
+static char **make_argv(char *cmd) {
+    int cap = 8, n = 0;
+    char **argv = malloc(sizeof(char*) * cap);
+    char *tok = strtok(cmd, " \t");
+    while (tok) {
+        if (n + 1 >= cap) {
+            cap *= 2;
+            argv = realloc(argv, sizeof(char*) * cap);
+        }
+        argv[n++] = tok;
+        tok = strtok(NULL, " \t");
+    }
+    argv[n] = NULL;
+    return argv;
+}
+
+void list_dir(const char *path) {
+    DIR *d = opendir(path);
+    struct dirent *entry;
+    printf("Contents of %s:\n", path);
+    while ((entry = readdir(d)) != NULL) {
+        printf("  %s\n", entry->d_name);
+    }
+    closedir(d);
+    printf("\n");
+}
+
+int main(int argc, char **argv) {
+    if (argc < 3 || strcmp(argv[1], "-c") != 0) {
+        fprintf(stderr, "usage: %s -c \"command\"\n", argv[0]);
+        return 1;
+    }
+
+    int fd = open("etest.wasm", O_WRONLY, 0);
+    printf("FD etest: %d\n", fd);
+    close(fd);
+
+    int nchildren=0;
+
+    char *cmd = strdup(argv[2]);  // full string
+    char *saveptr;
+    char *segment = strtok_r(cmd, "&", &saveptr);
+
+    while (segment) {
+        char *t = trim(segment);
+        if (*t) {
+            pid_t pid = fork();
+            if (pid == 0) {
+                // child: exec command
+                char *copy = strdup(t);
+                char **cmd_argv = make_argv(copy);
+		
+		execv(cmd_argv[0], cmd_argv);
+		char errbuf[256];
+		snprintf(errbuf, sizeof(errbuf), "execv(%s)", cmd_argv[0]);
+		perror(errbuf);
+                exit(1);
+            } else if (pid < 0) {
+                perror("fork");
+            }else {
+	    	nchildren++;
+	    }
+            // parent: do not wait (background)
+        }
+        segment = strtok_r(NULL, "&", &saveptr);
+    }
+
+    // behave like bash when all jobs are background:
+    // exit immediately, but reap zombies
+    for (int i = 0; i < nchildren; i++) {
+    	int status;
+    	pid_t w = wait(&status);
+    	if (w > 0) {
+        	printf("[mash] terminated pid %d, status %d\n", w, status);
+    	}
+    }
+    
+    return 0;
+}

example_grate/cages/runopen.c

@@ -0,0 +1,17 @@
+#include <unistd.h>
+#include <sys/stat.h>
+#include <sys/utsname.h>
+#include <fcntl.h>
+#include <stdio.h>
+#include <error.h>
+#include <stdlib.h>
+
+int main() {
+	int cageid = getpid();
+        printf("\n---- RUNOPEN %d ----\n", cageid);
+        int fd = open("firstfile", O_CREAT | O_WRONLY, 0755);
+	printf("FD: %d\n", fd);
+	close(fd);
+        printf("\n--EXIT RUNOPEN---\n");
+	return 0;
+}

example_grate/cages/tester.c

@@ -0,0 +1,16 @@
+#include <unistd.h>
+#include <sys/stat.h>
+#include <sys/utsname.h>
+#include <fcntl.h>
+#include <stdio.h>
+#include <error.h>
+#include <stdlib.h>
+
+int main() {
+	printf("\n[PROG]: Top\n");
+        int fd = open("etest.wasm", O_WRONLY, 0755);
+        printf("\n[PROG]: FD: %d\n", fd);
+
+	close(fd);
+	return 0;
+}

example_grate/gratecfg.yaml

@@ -0,0 +1,24 @@
+#include <stdio.h>
+#include <sys/types.h>
+#include <sys/stat.h>
+#include <string.h>
+
+#include <lind_syscall_num.h>
+#include "magic.h"
+
+
+int grate_syscalls[] = {GETEGID_SYSCALL};
+int grate_syscalls_len = 1;
+
+void grate_init() {
+	printf("EGID init'd\n");
+}
+
+void grate_destroy() {
+	printf("EGID exiting.\n");
+}
+
+int getegid_syscall(int cageid) {
+	return 556;
+}
+

example_grate/grates/getegid_grate.c

@@ -0,0 +1,22 @@
+#include <stdio.h>
+#include <sys/types.h>
+#include <sys/stat.h>
+#include <string.h>
+
+#include <lind_syscall_num.h>
+#include "magic.h"
+
+int grate_syscalls[] = {GETEUID_SYSCALL};
+int grate_syscalls_len = 1;
+
+void grate_init() {
+	printf("EUID initing...\n");
+}
+
+void grate_destroy() {
+	printf("EUID exiting.\n");
+}
+
+int geteuid_syscall(int cageid) {
+	return 546;
+}