Magisk/jni/main.c

146 lines
4.0 KiB
C
Raw Normal View History

2017-04-14 21:23:09 +02:00
/* main.c - The multicall entry point
2017-04-04 21:44:13 +02:00
*/
#include <stdlib.h>
2017-04-15 12:10:54 +02:00
#include <stdio.h>
2017-04-04 21:44:13 +02:00
#include "utils.h"
#include "magisk.h"
2017-04-08 01:37:43 +02:00
#include "daemon.h"
2017-04-04 21:44:13 +02:00
2017-04-06 00:12:29 +02:00
char *argv0;
2017-04-04 21:44:13 +02:00
2017-04-15 12:10:54 +02:00
char *applet[] =
2017-04-22 00:34:14 +02:00
{ "su", "resetprop", "magiskpolicy", "supolicy", "sepolicy-inject", "magiskhide", NULL };
2017-04-15 12:10:54 +02:00
int (*applet_main[]) (int, char *[]) =
2017-04-22 00:34:14 +02:00
{ su_client_main, resetprop_main, magiskpolicy_main, magiskpolicy_main, magiskpolicy_main, magiskhide_main, NULL };
2017-04-15 12:10:54 +02:00
// Global error hander function
// Should be changed each thread/process
__thread void (*err_handler)(void);
2017-04-15 12:10:54 +02:00
static void usage() {
fprintf(stderr,
2017-04-21 17:57:21 +02:00
"Magisk v" xstr(MAGISK_VERSION) " multi-call binary\n"
2017-04-15 12:10:54 +02:00
"\n"
"Usage: %s [applet [arguments]...]\n"
" or: %s --install [SOURCE] <DIR> \n"
" or: %s --list\n"
2017-06-02 21:58:26 +02:00
" or: %s --createimg <PATH> <SIZE>\n"
" create ext4 image, SIZE is interpreted in MB\n"
" or: %s --imgsize <PATH>\n"
" or: %s --resizeimg <PATH> <SIZE>\n"
" SIZE is interpreted in MB\n"
" or: %s --[boot stage]\n"
" start boot stage service\n"
2017-04-15 12:10:54 +02:00
" or: %s [options]\n"
" or: applet [arguments]...\n"
"\n"
"Supported boot stages:\n"
" post-fs, post-fs-data, service\n"
"\n"
"Options:\n"
2017-04-30 19:58:52 +02:00
" -c print client version\n"
" -v print daemon version\n"
2017-04-15 12:10:54 +02:00
" -V print daemon version code\n"
"\n"
"Supported applets:\n"
2017-06-02 21:58:26 +02:00
, argv0, argv0, argv0, argv0, argv0, argv0, argv0, argv0);
2017-04-15 12:10:54 +02:00
for (int i = 0; applet[i]; ++i) {
fprintf(stderr, i ? ", %s" : " %s", applet[i]);
}
fprintf(stderr, "\n");
exit(1);
}
int main(int argc, char *argv[]) {
2017-04-06 00:12:29 +02:00
argv0 = argv[0];
// Exit the whole app if error occurs by default
err_handler = exit_proc;
2017-04-06 00:12:29 +02:00
char * arg = strrchr(argv[0], '/');
if (arg) ++arg;
else arg = argv[0];
2017-04-06 00:12:29 +02:00
if (strcmp(arg, "magisk") == 0) {
2017-04-15 12:10:54 +02:00
if (argc < 2) usage();
2017-04-30 19:58:52 +02:00
if (strcmp(argv[1], "-c") == 0) {
printf("%s\n", MAGISK_VER_STR);
return 0;
} else if (strcmp(argv[1], "-v") == 0) {
2017-04-14 21:23:09 +02:00
int fd = connect_daemon();
write_int(fd, CHECK_VERSION);
char *v = read_string(fd);
2017-04-30 19:58:52 +02:00
printf("%s\n", v);
2017-04-14 21:23:09 +02:00
free(v);
return 0;
} else if (strcmp(argv[1], "-V") == 0) {
int fd = connect_daemon();
write_int(fd, CHECK_VERSION_CODE);
printf("%d\n", read_int(fd));
return 0;
2017-04-15 12:10:54 +02:00
} else if (strcmp(argv[1], "--install") == 0) {
if (argc < 3) usage();
if (argc == 3) return create_links(NULL, argv[2]);
else return create_links(argv[2], argv[3]);
} else if (strcmp(argv[1], "--list") == 0) {
for (int i = 0; applet[i]; ++i)
printf("%s\n", applet[i]);
return 0;
2017-06-02 21:58:26 +02:00
} else if (strcmp(argv[1], "--createimg") == 0) {
if (argc < 4) usage();
int size;
sscanf(argv[3], "%d", &size);
return create_img(argv[2], size);
} else if (strcmp(argv[1], "--imgsize") == 0) {
if (argc < 3) usage();
int used, total;
if (get_img_size(argv[2], &used, &total)) {
fprintf(stderr, "Cannot check %s size\n", argv[2]);
return 1;
}
printf("Used: %dM\tTotal: %dM\n", used, total);
return 0;
} else if (strcmp(argv[1], "--resizeimg") == 0) {
if (argc < 4) usage();
int used, total, size;
sscanf(argv[3], "%d", &size);
if (get_img_size(argv[2], &used, &total)) {
fprintf(stderr, "Cannot check %s size\n", argv[2]);
return 1;
}
if (size <= used) {
fprintf(stderr, "Cannot resize smaller than %dM\n", used);
return 1;
}
return resize_img(argv[2], size);
2017-04-14 21:23:09 +02:00
} else if (strcmp(argv[1], "--post-fs") == 0) {
int fd = connect_daemon();
write_int(fd, POST_FS);
return read_int(fd);
2017-04-06 00:12:29 +02:00
} else if (strcmp(argv[1], "--post-fs-data") == 0) {
int fd = connect_daemon();
write_int(fd, POST_FS_DATA);
return read_int(fd);
2017-04-14 21:23:09 +02:00
} else if (strcmp(argv[1], "--service") == 0) {
int fd = connect_daemon();
write_int(fd, LATE_START);
return read_int(fd);
} else {
// It's calling applets
--argc;
++argv;
2017-04-06 00:12:29 +02:00
arg = argv[0];
}
}
// Applets
2017-04-15 12:10:54 +02:00
for (int i = 0; applet[i]; ++i) {
if (strcmp(arg, applet[i]) == 0)
return (*applet_main[i])(argc, argv);
}
2017-04-15 12:10:54 +02:00
fprintf(stderr, "%s: applet not found\n", arg);
2017-04-06 00:12:29 +02:00
return 1;
2017-04-14 21:23:09 +02:00
}