119 lines
3.5 KiB
C
119 lines
3.5 KiB
C
#include <stdio.h>
|
|
#include <unistd.h>
|
|
#include <stdlib.h>
|
|
#include <sys/prctl.h>
|
|
#include <string.h>
|
|
#include <errno.h>
|
|
|
|
#define KERNEL_SU_OPTION 0xDEADBEEF
|
|
#define KSU_OPTIONS 0xdeadbeef
|
|
|
|
// KPM控制代码
|
|
#define CMD_KPM_CONTROL 28
|
|
#define CMD_KPM_CONTROL_MAX 7
|
|
|
|
// 控制代码
|
|
// prctl(xxx, 28, "PATH", "ARGS")
|
|
// success return 0, error return -N
|
|
#define SUKISU_KPM_LOAD 28
|
|
|
|
// prctl(xxx, 29, "NAME")
|
|
// success return 0, error return -N
|
|
#define SUKISU_KPM_UNLOAD 29
|
|
|
|
// num = prctl(xxx, 30)
|
|
// error return -N
|
|
// success return +num or 0
|
|
#define SUKISU_KPM_NUM 30
|
|
|
|
// prctl(xxx, 31, Buffer, BufferSize)
|
|
// success return +out, error return -N
|
|
#define SUKISU_KPM_LIST 31
|
|
|
|
// prctl(xxx, 32, "NAME", Buffer[256])
|
|
// success return +out, error return -N
|
|
#define SUKISU_KPM_INFO 32
|
|
|
|
// prctl(xxx, 33, "NAME", "ARGS")
|
|
// success return KPM's result value
|
|
// error return -N
|
|
#define SUKISU_KPM_CONTROL 33
|
|
|
|
// prctl(xxx, 34, buffer, bufferSize)
|
|
// success return KPM's result value
|
|
// error return -N
|
|
#define SUKISU_KPM_VERSION 34
|
|
|
|
#define CONTROL_CODE(n) (n)
|
|
|
|
void print_usage(const char *prog) {
|
|
printf("Usage: %s <command> [args]\n", prog);
|
|
printf("Commands:\n");
|
|
printf(" load <path> <args> Load a KPM module\n");
|
|
printf(" unload <name> Unload a KPM module\n");
|
|
printf(" num Get number of loaded modules\n");
|
|
printf(" list List loaded KPM modules\n");
|
|
printf(" info <name> Get info of a KPM module\n");
|
|
printf(" control <name> <args> Send control command to a KPM module\n");
|
|
printf(" version Print KPM Loader version\n");
|
|
}
|
|
|
|
int main(int argc, char *argv[]) {
|
|
if (argc < 2) {
|
|
print_usage(argv[0]);
|
|
return 1;
|
|
}
|
|
|
|
int ret = -1;
|
|
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);
|
|
if(out > 0) {
|
|
printf("Success");
|
|
}
|
|
} else if (strcmp(argv[1], "unload") == 0 && argc >= 3) {
|
|
// 卸载 KPM 模块
|
|
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), NULL, NULL, &out);
|
|
printf("%d", out);
|
|
return 0;
|
|
} else if (strcmp(argv[1], "list") == 0) {
|
|
// 获取模块列表
|
|
char buffer[1024] = {0};
|
|
ret = prctl(KSU_OPTIONS, CONTROL_CODE(SUKISU_KPM_LIST), buffer, sizeof(buffer), &out);
|
|
if (out >= 0) {
|
|
printf("%s", buffer);
|
|
}
|
|
} else if (strcmp(argv[1], "info") == 0 && argc >= 3) {
|
|
// 获取指定模块信息
|
|
char buffer[256] = {0};
|
|
ret = prctl(KSU_OPTIONS, CONTROL_CODE(SUKISU_KPM_INFO), argv[2], buffer, &out);
|
|
if (out >= 0) {
|
|
printf("%s\n", buffer);
|
|
}
|
|
} else if (strcmp(argv[1], "control") == 0 && argc >= 4) {
|
|
// 控制 KPM 模块
|
|
ret = prctl(KSU_OPTIONS, CONTROL_CODE(SUKISU_KPM_CONTROL), argv[2], argv[3], &out);
|
|
} else if (strcmp(argv[1], "version") == 0) {
|
|
char buffer[1024] = {0};
|
|
ret = prctl(KSU_OPTIONS, CONTROL_CODE(SUKISU_KPM_VERSION), buffer, sizeof(buffer), &out);
|
|
if (out >= 0) {
|
|
printf("%s", buffer);
|
|
}
|
|
} else {
|
|
print_usage(argv[0]);
|
|
return 1;
|
|
}
|
|
|
|
if (out < 0) {
|
|
printf("Error: %s\n", strerror(-out));
|
|
return -1;
|
|
}
|
|
|
|
return 0;
|
|
}
|