From c2395d48f2bf95b39490c0762789e96f29946582 Mon Sep 17 00:00:00 2001
From: ShirkNeko <109797057+ShirkNeko@users.noreply.github.com>
Date: Sat, 29 Mar 2025 16:57:14 +0800
Subject: [PATCH] Fix KPM control code logic, adjust prctl call parameters to
 handle return values correctly

---
 userspace/kpmmgr/jni/kpmmgr.c | 16 +++++++++-------
 1 file changed, 9 insertions(+), 7 deletions(-)

diff --git a/userspace/kpmmgr/jni/kpmmgr.c b/userspace/kpmmgr/jni/kpmmgr.c
index befc97ce..8e5f4dd8 100644
--- a/userspace/kpmmgr/jni/kpmmgr.c
+++ b/userspace/kpmmgr/jni/kpmmgr.c
@@ -20,7 +20,7 @@
 #define SUKISU_KPM_CONTROL 6
 #define SUKISU_KPM_PRINT 7
 
-#define CONTROL_CODE(n) (CMD_KPM_CONTROL + n)
+#define CONTROL_CODE(n) (CMD_KPM_CONTROL + n - 1)
 
 void print_usage(const char *prog) {
     printf("Usage: %s <command> [args]\n", prog);
@@ -41,17 +41,19 @@ int main(int argc, char *argv[]) {
     }
 
     int ret = -1;
-    int out = 0;  // 存储返回值
+    int out = -1;  // 存储返回值
 
     if (strcmp(argv[1], "load") == 0 && argc >= 3) {
         // 加载 KPM 模块
         ret = prctl(KSU_OPTIONS, CONTROL_CODE(SUKISU_KPM_LOAD), argv[2], (argc > 3 ? argv[3] : NULL), &out);
     } else if (strcmp(argv[1], "unload") == 0 && argc >= 3) {
         // 卸载 KPM 模块
-        ret = prctl(KSU_OPTIONS, CONTROL_CODE(SUKISU_KPM_UNLOAD), argv[2], &out);
+        ret = prctl(KSU_OPTIONS, CONTROL_CODE(SUKISU_KPM_UNLOAD), argv[2], NULL, &out);
     } else if (strcmp(argv[1], "num") == 0) {
         // 获取加载的 KPM 数量
-        ret = prctl(KSU_OPTIONS, CONTROL_CODE(SUKISU_KPM_NUM), &out);
+        ret = prctl(KSU_OPTIONS, CONTROL_CODE(SUKISU_KPM_NUM), NULL, NULL, &out);
+        printf("%d", out);
+        return 0;
     } else if (strcmp(argv[1], "list") == 0) {
         // 获取模块列表
         char buffer[1024] = {0};
@@ -71,14 +73,14 @@ int main(int argc, char *argv[]) {
         ret = prctl(KSU_OPTIONS, CONTROL_CODE(SUKISU_KPM_CONTROL), argv[2], argv[3], &out);
     } else if (strcmp(argv[1], "print") == 0) {
         // 在 stdout 输出 KPM 列表
-        ret = prctl(KSU_OPTIONS, CONTROL_CODE(SUKISU_KPM_PRINT), &out);
+        ret = prctl(KSU_OPTIONS, CONTROL_CODE(SUKISU_KPM_PRINT), NULL, NULL, &out);
     } else {
         print_usage(argv[0]);
         return 1;
     }
 
-    if (ret < 0) {
-        printf("Error: %s\n", strerror(errno));
+    if (out < 0) {
+        printf("Error: %s\n", strerror(-out));
         return -1;
     }