完善super_access

kernel/kpm/Makefile

@@ -1,2 +1,8 @@
 obj-y += kpm.o
-obj-y += compact.o
+obj-y += compact.o
+obj-y += super_access.c
+
+ccflags-y += -Wno-implicit-function-declaration -Wno-strict-prototypes -Wno-int-conversion -Wno-gcc-compat
+ccflags-y += -Wno-declaration-after-statement -Wno-unused-function
+
+# End

kernel/kpm/compact.c

@@ -27,6 +27,7 @@
 #include "kpm.h"
 #include "compact.h"
 #include "../allowlist.h"
+#include "../manager.h"
 
 unsigned long sukisu_compact_find_symbol(const char* name);
 
@@ -49,6 +50,16 @@ int sukisu_is_uid_should_umount(uid_t uid) {
     return ksu_uid_should_umount(uid) ? 1 : 0;
 }
 
+static
+int sukisu_is_current_uid_manager() {
+    return is_manager();
+}
+
+static
+uid_t sukisu_get_manager_uid() {
+    return ksu_manager_uid;
+}
+
 // ======================================================================
 
 struct CompactAddressSymbol {
@@ -62,7 +73,9 @@ static struct CompactAddressSymbol address_symbol [] = {
     { "is_run_in_sukisu_ultra", (void*)1 },
     { "is_su_allow_uid", &sukisu_is_su_allow_uid },
     { "get_ap_mod_exclude", &sukisu_get_ap_mod_exclude },
-    { "is_uid_should_umount", &sukisu_is_uid_should_umount }
+    { "is_uid_should_umount", &sukisu_is_uid_should_umount },
+    { "is_current_uid_manager", &sukisu_is_current_uid_manager },
+    { "get_manager_uid", &sukisu_get_manager_uid }
 };
 
 unsigned long sukisu_compact_find_symbol(const char* name) {

kernel/kpm/super_access.c

@@ -26,7 +26,8 @@
 #include <linux/slab.h>
 #include "kpm.h"
 #include "compact.h"
-#include <stddef.h> // 需要包含 offsetof 宏
+#include <linux/types.h>
+#include <linux/stddef.h>
 
 // 结构体成员元数据
 struct DynamicStructMember {
@@ -156,7 +157,7 @@ struct DynamicStructInfo* dynamic_struct_infos[] = {
     STRUCT_INFO(vfsmount),
     STRUCT_INFO(mnt_namespace),
     #ifdef CONFIG_KPROBES
-        STRUCT_INFO(kprobe)
+        STRUCT_INFO(kprobe),
     #endif
     STRUCT_INFO(vm_area_struct),
     STRUCT_INFO(vm_operations_struct),

kernel/kpm/super_access.h

@@ -0,0 +1,39 @@
+#ifndef __SUKISU_SUPER_ACCESS_H
+#define __SUKISU_SUPER_ACCESS_H
+
+#include <linux/types.h>
+#include <linux/stddef.h>
+#include "kpm.h"
+#include "compact.h"
+
+// return 0 if successful
+// return -1 if struct not defined
+int sukisu_super_find_struct(
+    const char* struct_name,
+    size_t* out_size,
+    int* out_members
+);
+
+// Dynamic access struct
+// return 0 if successful
+// return -1 if struct not defined
+// return -2 if member not defined
+int sukisu_super_access (
+    const char* struct_name,
+    const char* member_name,
+    size_t* out_offset,
+    size_t* out_size
+);
+
+// Dynamic container_of
+// return 0 if success
+// return -1 if current struct not defined
+// return -2 if target member not defined
+int sukisu_super_container_of(
+    const char* struct_name,
+    const char* member_name,
+    void* ptr,
+    void** out_ptr
+);
+
+#endif