Add injector

loader/src/include/daemon.h

@@ -0,0 +1,69 @@
+#pragma once
+
+#include <string_view>
+#include <string>
+#include <vector>
+
+#if defined(__LP64__)
+# define LP_SELECT(lp32, lp64) lp64
+#else
+# define LP_SELECT(lp32, lp64) lp32
+#endif
+
+constexpr std::string_view kZygiskSocket = LP_SELECT("zygisk32", "zygisk64") "placeholder123456";
+
+class UniqueFd {
+    using Fd = int;
+public:
+    UniqueFd() = default;
+
+    UniqueFd(Fd fd) : fd_(fd) {}
+
+    ~UniqueFd() { close(fd_); }
+
+    // Disallow copy
+    UniqueFd(const UniqueFd&) = delete;
+
+    UniqueFd& operator=(const UniqueFd&) = delete;
+
+    // Allow move
+    UniqueFd(UniqueFd&& other) { std::swap(fd_, other.fd_); }
+
+    UniqueFd& operator=(UniqueFd&& other) {
+        std::swap(fd_, other.fd_);
+        return *this;
+    }
+
+    // Implict cast to Fd
+    operator const Fd&() const { return fd_; }
+
+private:
+    Fd fd_ = -1;
+};
+
+namespace zygiskd {
+
+    struct Module {
+        std::string name;
+        void* handle;
+
+        inline explicit Module(std::string name, void* handle)
+                : name(name), handle(handle) {}
+    };
+
+    enum class SocketAction {
+        HeartBeat,
+        ReadNativeBridge,
+        ReadInjector,
+        ReadModules,
+        RequestCompanionSocket,
+    };
+
+    bool PingHeartbeat();
+
+    std::string ReadNativeBridge();
+
+    UniqueFd ReadInjector();
+
+    std::vector<Module> ReadModules();
+}

loader/src/include/dl.h

@@ -0,0 +1,7 @@
+#pragma once
+
+#include <dlfcn.h>
+
+void *DlopenExt(const char *path, int flags);
+
+void *DlopenMem(int memfd, int flags);

loader/src/include/logging.h

@@ -0,0 +1,30 @@
+#pragma once
+
+#include <android/log.h>
+#include <errno.h>
+#include <string.h>
+
+#ifndef LOG_TAG
+#define LOG_TAG    "zygisksu"
+#endif
+
+#ifdef LOG_DISABLED
+#define LOGD(...)
+#define LOGV(...)
+#define LOGI(...)
+#define LOGW(...)
+#define LOGE(...)
+#else
+#ifndef NDEBUG
+#define LOGD(...)  __android_log_print(ANDROID_LOG_DEBUG, LOG_TAG, __VA_ARGS__)
+#define LOGV(...)  __android_log_print(ANDROID_LOG_VERBOSE, LOG_TAG, __VA_ARGS__)
+#else
+#define LOGD(...)
+#define LOGV(...)
+#endif
+#define LOGI(...)  __android_log_print(ANDROID_LOG_INFO, LOG_TAG, __VA_ARGS__)
+#define LOGW(...)  __android_log_print(ANDROID_LOG_WARN, LOG_TAG, __VA_ARGS__)
+#define LOGE(...)  __android_log_print(ANDROID_LOG_ERROR, LOG_TAG, __VA_ARGS__)
+#define LOGF(...)  __android_log_print(ANDROID_LOG_FATAL, LOG_TAG, __VA_ARGS__)
+#define PLOGE(fmt, args...) LOGE(fmt " failed with %d: %s", ##args, errno, strerror(errno))
+#endif

loader/src/include/native_bridge_callbacks.h

@@ -0,0 +1,33 @@
+#pragma once
+
+#include <android/api-level.h>
+#include <cstdint>
+
+template<unsigned>
+struct NativeBridgeCallbacks;
+
+template<>
+struct NativeBridgeCallbacks<__ANDROID_API_Q__> {
+    [[maybe_unused]] uint32_t version;
+    [[maybe_unused]] void *initialize;
+    [[maybe_unused]] void *loadLibrary;
+    [[maybe_unused]] void *getTrampoline;
+    [[maybe_unused]] void *isSupported;
+    [[maybe_unused]] void *getAppEnv;
+    [[maybe_unused]] void *isCompatibleWith;
+    [[maybe_unused]] void *getSignalHandler;
+    [[maybe_unused]] void *unloadLibrary;
+    [[maybe_unused]] void *getError;
+    [[maybe_unused]] void *isPathSupported;
+    [[maybe_unused]] void *initAnonymousNamespace;
+    [[maybe_unused]] void *createNamespace;
+    [[maybe_unused]] void *linkNamespaces;
+    [[maybe_unused]] void *loadLibraryExt;
+    [[maybe_unused]] void *getVendorNamespace;
+    [[maybe_unused]] void *getExportedNamespace;
+};
+
+template<>
+struct NativeBridgeCallbacks<__ANDROID_API_R__> : NativeBridgeCallbacks<__ANDROID_API_Q__> {
+    [[maybe_unused]] void *preZygoteFork;
+};

loader/src/include/socket_utils.h

@@ -0,0 +1,20 @@
+#pragma once
+
+#include <string>
+
+#include "logging.h"
+
+namespace socket_utils {
+
+    ssize_t xread(int fd, void *buf, size_t count);
+
+    ssize_t xwrite(int fd, const void *buf, size_t count);
+
+    size_t read_usize(int fd);
+
+    std::string read_string(int fd);
+
+    bool write_u8(int fd, uint8_t val);
+
+    int recv_fd(int fd);
+}