Restructure project files

2025-09-06 06:36:58 +00:00 · 2023-11-08 01:46:02 -08:00
parent ecb31eed40
commit 65c18f9c09
44 changed files with 85 additions and 113 deletions
@@ -0,0 +1,126 @@
+#pragma once
+
+#include <pthread.h>
+#include <poll.h>
+#include <string>
+#include <limits>
+#include <atomic>
+#include <functional>
+
+#include "socket.hpp"
+#include "../core-rs.hpp"
+
+#define AID_ROOT   0
+#define AID_SHELL  2000
+#define AID_APP_START 10000
+#define AID_APP_END 19999
+#define AID_USER_OFFSET 100000
+
+#define to_app_id(uid)  (uid % AID_USER_OFFSET)
+#define to_user_id(uid) (uid / AID_USER_OFFSET)
+
+// Daemon command codes
+namespace MainRequest {
+enum : int {
+    START_DAEMON,
+    CHECK_VERSION,
+    CHECK_VERSION_CODE,
+    STOP_DAEMON,
+
+    _SYNC_BARRIER_,
+
+    SUPERUSER,
+    ZYGOTE_RESTART,
+    DENYLIST,
+    SQLITE_CMD,
+    REMOVE_MODULES,
+    ZYGISK,
+
+    _STAGE_BARRIER_,
+
+    POST_FS_DATA,
+    LATE_START,
+    BOOT_COMPLETE,
+
+    END,
+};
+}
+
+// Return codes for daemon
+namespace MainResponse {
+enum : int {
+    ERROR = -1,
+    OK = 0,
+    ROOT_REQUIRED,
+    ACCESS_DENIED,
+    END
+};
+}
+
+struct module_info {
+    std::string name;
+    int z32 = -1;
+#if defined(__LP64__)
+    int z64 = -1;
+#endif
+};
+
+extern bool RECOVERY_MODE;
+extern bool zygisk_enabled;
+extern std::vector<module_info> *module_list;
+
+void reset_zygisk(bool restore);
+extern "C" const char *get_magisk_tmp();
+int connect_daemon(int req, bool create = false);
+std::string find_preinit_device();
+void unlock_blocks();
+void reboot();
+
+// Poll control
+using poll_callback = void(*)(pollfd*);
+void register_poll(const pollfd *pfd, poll_callback callback);
+void unregister_poll(int fd, bool auto_close);
+void clear_poll();
+
+// Thread pool
+void exec_task(std::function<void()> &&task);
+
+// Daemon handlers
+void boot_stage_handler(int client, int code);
+void denylist_handler(int client, const sock_cred *cred);
+void su_daemon_handler(int client, const sock_cred *cred);
+void zygisk_handler(int client, const sock_cred *cred);
+
+// Package
+extern std::atomic<ino_t> pkg_xml_ino;
+void preserve_stub_apk();
+void check_pkg_refresh();
+std::vector<bool> get_app_no_list();
+// Call check_pkg_refresh() before calling get_manager(...)
+// to make sure the package state is invalidated!
+int get_manager(int user_id = 0, std::string *pkg = nullptr, bool install = false);
+void prune_su_access();
+
+// Module stuffs
+void handle_modules();
+void load_modules();
+void disable_modules();
+void remove_modules();
+void exec_module_scripts(const char *stage);
+
+// Scripting
+void exec_script(const char *script);
+void exec_common_scripts(const char *stage);
+void exec_module_scripts(const char *stage, const std::vector<std::string_view> &modules);
+void install_apk(const char *apk);
+void uninstall_pkg(const char *pkg);
+void clear_pkg(const char *pkg, int user_id);
+[[noreturn]] void install_module(const char *file);
+
+// Denylist
+extern std::atomic_flag skip_pkg_rescan;
+extern std::atomic<bool> denylist_enforced;
+int denylist_cli(int argc, char **argv);
+void initialize_denylist();
+bool is_deny_target(int uid, std::string_view process);
+void revert_unmount();