Reduce wait interval and various bug fixes

This commit is contained in:
topjohnwu 2016-12-28 15:28:42 +08:00
parent b6412afe96
commit dc29018ec0

View File

@ -16,11 +16,11 @@
#define LOGFILE "/cache/magisk.log" #define LOGFILE "/cache/magisk.log"
#define HIDELIST "/magisk/.core/magiskhide/hidelist" #define HIDELIST "/magisk/.core/magiskhide/hidelist"
#define DUMMYPATH "/dev/magisk/dummy"
FILE *logfile; FILE *logfile;
int i, list_size, pipefd[2], zygote_num = 0; int i, list_size, pipefd[2];
char **hide_list = NULL; char **hide_list = NULL, buffer[512];
char zygote_ns[2][32];
pthread_mutex_t mutex; pthread_mutex_t mutex;
char **file_to_str_arr(FILE *fp, int *size) { char **file_to_str_arr(FILE *fp, int *size) {
@ -64,35 +64,49 @@ void lazy_unmount(const char* mountpoint) {
} }
int hideMagisk() { int hideMagisk() {
int pid;
char path[256], cache_block[256], namespace[32];
cache_block[0] = 0;
close(pipefd[1]); close(pipefd[1]);
int pid, zygote_num = 0;
char cache_block[256], zygote_ns[2][32];
cache_block[0] = '\0';
// Get the mount namespace of zygote
FILE *p = popen("/data/busybox/ps | grep zygote | grep -v grep", "r");
while(fgets(buffer, sizeof(buffer), p)) {
if (zygote_num == 2) break;
sscanf(buffer, "%d", &pid);
read_namespace(pid, zygote_ns[zygote_num], 32);
++zygote_num;
}
pclose(p);
for (i = 0; i < zygote_num; ++i)
fprintf(logfile, "Zygote(%d) ns=%s ", i, zygote_ns[i]);
fprintf(logfile, "\n");
while(1) { while(1) {
read(pipefd[0], &pid, sizeof(pid)); read(pipefd[0], &pid, sizeof(pid));
if(pid == -1) break; if(pid == -1) break;
int badns; int badns;
do { while(1) {
badns = 0; badns = 0;
read_namespace(pid, namespace, 32); read_namespace(pid, buffer, 32);
for (i = 0; i < zygote_num; ++i) { for (i = 0; i < zygote_num; ++i) {
if (strcmp(namespace, zygote_ns[i]) == 0) { if (strcmp(buffer, zygote_ns[i]) == 0) {
usleep(50000); usleep(1000);
badns = 1; badns = 1;
break; break;
} }
} }
if (badns) continue; if (!badns) break;
break; }
} while(1);
fprintf(logfile, "ns=%s\n", namespace); fprintf(logfile, "ns=%s]\n", buffer);
snprintf(path, 256, "/proc/%d/ns/mnt", pid); snprintf(buffer, sizeof(buffer), "/proc/%d/ns/mnt", pid);
int fd = open(path, O_RDONLY); int fd = open(buffer, O_RDONLY);
if(fd == -1) continue; // Maybe process died.. if(fd == -1) continue; // Maybe process died..
if(setns(fd, 0) == -1) { if(setns(fd, 0) == -1) {
fprintf(logfile, "MagiskHide: Unable to change namespace for pid=%d\n", pid); fprintf(logfile, "MagiskHide: Unable to change namespace for pid=%d\n", pid);
@ -100,15 +114,15 @@ int hideMagisk() {
} }
close(fd); close(fd);
snprintf(path, 256, "/proc/%d/mounts", pid); snprintf(buffer, sizeof(buffer), "/proc/%d/mounts", pid);
FILE *mount_fp = fopen(path, "r"); FILE *mount_fp = fopen(buffer, "r");
if (mount_fp == NULL) { if (mount_fp == NULL) {
fprintf(logfile, "MagiskHide: Error opening mount list!\n"); fprintf(logfile, "MagiskHide: Error opening mount list!\n");
continue; continue;
} }
int mount_size; int mount_size;
char **mount_list = file_to_str_arr(mount_fp, &mount_size), mountpoint[256], cache_block[256]; char **mount_list = file_to_str_arr(mount_fp, &mount_size);
// Find the cache block name if not found yet // Find the cache block name if not found yet
if (strlen(cache_block) == 0) { if (strlen(cache_block) == 0) {
@ -124,12 +138,11 @@ int hideMagisk() {
for(i = mount_size - 1; i >= 0; --i) { for(i = mount_size - 1; i >= 0; --i) {
if (strstr(mount_list[i], "tmpfs /system/") || strstr(mount_list[i], "tmpfs /vendor/") if (strstr(mount_list[i], "tmpfs /system/") || strstr(mount_list[i], "tmpfs /vendor/")
|| (strstr(mount_list[i], cache_block) && strstr(mount_list[i], "/system")) ) { || (strstr(mount_list[i], cache_block) && strstr(mount_list[i], "/system")) ) {
sscanf(mount_list[i], "%*s %256s", mountpoint); sscanf(mount_list[i], "%*s %512s", buffer);
lazy_unmount(mountpoint); lazy_unmount(buffer);
} }
free(mount_list[i]); free(mount_list[i]);
} }
// Free memory
free(mount_list); free(mount_list);
// Re-read mount infos // Re-read mount infos
@ -139,13 +152,12 @@ int hideMagisk() {
// Unmount loop mounts // Unmount loop mounts
for(i = mount_size - 1; i >= 0; --i) { for(i = mount_size - 1; i >= 0; --i) {
if (strstr(mount_list[i], "/dev/block/loop")) { if (strstr(mount_list[i], "/dev/block/loop") && !strstr(mount_list[i], DUMMYPATH)) {
sscanf(mount_list[i], "%*s %256s", mountpoint); sscanf(mount_list[i], "%*s %512s", buffer);
lazy_unmount(mountpoint); lazy_unmount(buffer);
} }
free(mount_list[i]); free(mount_list[i]);
} }
// Free memory
free(mount_list); free(mount_list);
} }
@ -154,7 +166,7 @@ int hideMagisk() {
} }
void update_list(const char *listpath) { void update_list(const char *listpath) {
FILE *hide_fp = fopen((char*) listpath, "r"); FILE *hide_fp = fopen(listpath, "r");
if (hide_fp == NULL) { if (hide_fp == NULL) {
fprintf(logfile, "MagiskHide: Error opening hide list\n"); fprintf(logfile, "MagiskHide: Error opening hide list\n");
exit(1); exit(1);
@ -187,17 +199,17 @@ void *monitor_list(void *path) {
signal(SIGQUIT, quit_pthread); signal(SIGQUIT, quit_pthread);
int inotifyFd = -1; int inotifyFd = -1;
char buffer[512]; char str[512];
while(1) { while(1) {
if (inotifyFd == -1 || read(inotifyFd, buffer, 512) == -1) { if (inotifyFd == -1 || read(inotifyFd, str, sizeof(str)) == -1) {
close(inotifyFd); close(inotifyFd);
inotifyFd = inotify_init(); inotifyFd = inotify_init();
if (inotifyFd == -1) { if (inotifyFd == -1) {
fprintf(logfile, "MagiskHide: Unable to watch %s\n", listpath); fprintf(logfile, "MagiskHide: Unable to watch %s\n", listpath);
exit(1); exit(1);
} }
if (inotify_add_watch(inotifyFd, (char*) listpath, IN_MODIFY) == -1) { if (inotify_add_watch(inotifyFd, listpath, IN_MODIFY) == -1) {
fprintf(logfile, "MagiskHide: Unable to watch %s\n", listpath); fprintf(logfile, "MagiskHide: Unable to watch %s\n", listpath);
exit(1); exit(1);
} }
@ -230,22 +242,6 @@ int main(int argc, char **argv, char **envp) {
run_as_daemon(); run_as_daemon();
// Get the mount namespace of zygote for checking
char buffer[512];
int pid;
FILE *p = popen("/data/busybox/ps | grep zygote | grep -v grep", "r");
while(fgets(buffer, sizeof(buffer), p)) {
if (zygote_num == 2) break;
sscanf(buffer, "%d", &pid);
read_namespace(pid, zygote_ns[zygote_num], 32);
++zygote_num;
}
pclose(p);
for (i = 0; i < zygote_num; ++i)
fprintf(logfile, "Zygote(%d) ns=%s ", i, zygote_ns[i]);
fprintf(logfile, "\n");
// Fork a child to handle namespace switches and unmounts // Fork a child to handle namespace switches and unmounts
pipe(pipefd); pipe(pipefd);
switch(fork()) { switch(fork()) {
@ -263,8 +259,8 @@ int main(int argc, char **argv, char **envp) {
pthread_mutex_init(&mutex, NULL); pthread_mutex_init(&mutex, NULL);
pthread_create(&list_monitor, NULL, monitor_list, HIDELIST); pthread_create(&list_monitor, NULL, monitor_list, HIDELIST);
p = popen("while true; do logcat -b events -v raw -s am_proc_start; sleep 1; done", "r"); // Monitor am_proc_start in main thread
FILE *p = popen("while true; do logcat -b events -v raw -s am_proc_start; sleep 1; done", "r");
while(!feof(p)) { while(!feof(p)) {
//Format of am_proc_start is (as of Android 5.1 and 6.0) //Format of am_proc_start is (as of Android 5.1 and 6.0)
//UserID, pid, unix uid, processName, hostingType, hostingName //UserID, pid, unix uid, processName, hostingType, hostingName
@ -289,7 +285,7 @@ int main(int argc, char **argv, char **envp) {
for (i = 0; i < list_size; ++i) { for (i = 0; i < list_size; ++i) {
if(strstr(processName, hide_list[i])) { if(strstr(processName, hide_list[i])) {
fprintf(logfile, "MagiskHide: Disabling for process=%s, PID=%d, ", processName, pid, uid); fprintf(logfile, "MagiskHide: %s[PID=%d ", processName, pid);
write(pipefd[1], &pid, sizeof(pid)); write(pipefd[1], &pid, sizeof(pid));
} }
} }