Split inject code to a separate file to be more easily integrated in external projects

1 files changed, 50 insertions, 0 deletions

diff --git a/main.c b/main.c
new file mode 100644
index 0000000..66affcd
--- /dev/null
+++ b/main.c
@@ -0,0 +1,50 @@
+/*
+ *  inject.m
+ *  
+ *  Created by Sam Bingner on 9/27/2018
+ *  Copyright 2018 Sam Bingner. All Rights Reserved.
+ *
+ */
+
+#include <CoreFoundation/CoreFoundation.h>
+#include <mach/mach.h>
+#include <dlfcn.h>
+#include "patchfinder64.h"
+#include "CSCommon.h"
+#include "kern_funcs.h"
+#include "inject.h"
+
+
+mach_port_t try_restore_port() {
+    mach_port_t port = MACH_PORT_NULL;
+    kern_return_t err;
+
+    err = host_get_special_port(mach_host_self(), 0, 4, &port);
+    if (err == KERN_SUCCESS && port != MACH_PORT_NULL) {
+        fprintf(stderr, "got persisted port!\n");
+        // make sure rk64 etc use this port
+        return port;
+    }
+    fprintf(stderr, "unable to retrieve persisted port\n");
+    return MACH_PORT_NULL;
+}
+
+int main(int argc, char* argv[]) {
+    if (argc < 2) {
+        fprintf(stderr,"Usage: inject /full/path/to/executable\n");
+        fprintf(stderr,"Inject executables to trust cache\n");
+        return -1;
+    }
+    mach_port_t tfp0 = try_restore_port();
+    if (tfp0 == MACH_PORT_NULL)
+        return -2;
+    set_tfp0(tfp0);
+    uint64_t kernel_base = get_kernel_base(tfp0);
+    init_kernel(kernel_base, NULL);
+    uint64_t trust_chain = find_trustcache();
+    term_kernel();
+    printf("Injecting to trust cache...\n");
+    int ninjected = injectTrustCache(argc, argv, trust_chain);
+    printf("Successfully injected [%d/%d] to trust cache.\n", ninjected, argc - 1);
+    return argc - ninjected - 1;
+}