/* daemon.c - Magisk Daemon
 *
 * Start the daemon and wait for requests
 * Connect the daemon and send requests through sockets
 */

#include <stdlib.h>
#include <unistd.h>
#include <fcntl.h>
#include <string.h>
#include <pthread.h>
#include <signal.h>
#include <sys/un.h>
#include <sys/types.h>
#include <sys/mount.h>
#include <selinux/selinux.h>

#include "magisk.h"
#include "utils.h"
#include "daemon.h"
#include "resetprop.h"

int is_daemon_init = 0, seperate_vendor = 0;

static void *request_handler(void *args) {
	int client = *((int *) args);
	free(args);
	int req = read_int(client);

	struct ucred credential;
	get_client_cred(client, &credential);

	switch (req) {
	case LAUNCH_MAGISKHIDE:
	case STOP_MAGISKHIDE:
	case ADD_HIDELIST:
	case RM_HIDELIST:
	case LS_HIDELIST:
	case POST_FS:
	case POST_FS_DATA:
	case LATE_START:
		if (credential.uid != 0) {
			write_int(client, ROOT_REQUIRED);
			close(client);
			return NULL;
		}
	default:
		break;
	}

	switch (req) {
	case LAUNCH_MAGISKHIDE:
		launch_magiskhide(client);
		break;
	case STOP_MAGISKHIDE:
		stop_magiskhide(client);
		break;
	case ADD_HIDELIST:
		add_hide_list(client);
		break;
	case RM_HIDELIST:
		rm_hide_list(client);
		break;
	case LS_HIDELIST:
		ls_hide_list(client);
		break;
	case SUPERUSER:
		su_daemon_receiver(client, &credential);
		break;
	case CHECK_VERSION:
		write_string(client, MAGISK_VER_STR);
		close(client);
		break;
	case CHECK_VERSION_CODE:
		write_int(client, MAGISK_VER_CODE);
		close(client);
		break;
	case POST_FS:
		post_fs(client);
		break;
	case POST_FS_DATA:
		post_fs_data(client);
		break;
	case LATE_START:
		late_start(client);
		break;
	default:
		break;
	}
	return NULL;
}


static void *start_magisk_hide(void *args) {
	launch_magiskhide(-1);
	return NULL;
}

void auto_start_magiskhide() {
	char *hide_prop = getprop2(MAGISKHIDE_PROP, 1);
	if (hide_prop == NULL || strcmp(hide_prop, "0") != 0) {
		pthread_t thread;
		xpthread_create(&thread, NULL, start_magisk_hide, NULL);
		pthread_detach(thread);
	}
	free(hide_prop);
}

void daemon_init() {
	is_daemon_init = 1;

	 // Magisk binaries
	char *bin_path = NULL;
	if (access("/cache/data_bin", F_OK) == 0)
		bin_path = "/cache/data_bin";
	else if (access("/data/data/com.topjohnwu.magisk/install", F_OK) == 0)
		bin_path = "/data/data/com.topjohnwu.magisk/install";
	else if (access("/data/user_de/0/com.topjohnwu.magisk/install", F_OK) == 0)
		bin_path = "/data/user_de/0/com.topjohnwu.magisk/install";
	if (bin_path) {
		rm_rf(DATABIN);
		cp_afc(bin_path, DATABIN);
		rm_rf(bin_path);
	}

	// Migration
	rm_rf("/data/magisk");
	unlink("/data/magisk.img");
	unlink("/data/magisk_debug.log");
	chmod("/data/adb", 0700);

	LOGI("* Creating /sbin overlay");
	DIR *dir;
	struct dirent *entry;
	int root, sbin;
	char buf[PATH_MAX], buf2[PATH_MAX];
	char magisk_name[10], init_name[10];

	// Setup links under /sbin
	xmount(NULL, "/", NULL, MS_REMOUNT, NULL);
	xmkdir("/root", 0755);
	chmod("/root", 0755);
	root = xopen("/root", O_RDONLY | O_CLOEXEC);
	sbin = xopen("/sbin", O_RDONLY | O_CLOEXEC);
	link_dir(sbin, root);
	unlink("/sbin/magisk");
	close(sbin);

	xmount("tmpfs", "/sbin", "tmpfs", 0, NULL);
	chmod("/sbin", 0755);
	setfilecon("/sbin", "u:object_r:rootfs:s0");
	dir = xfdopendir(root);
	while((entry = xreaddir(dir))) {
		if (strcmp(entry->d_name, ".") == 0 || strcmp(entry->d_name, "..") == 0) continue;
		snprintf(buf, PATH_MAX, "/root/%s", entry->d_name);
		snprintf(buf2, PATH_MAX, "/sbin/%s", entry->d_name);
		xsymlink(buf, buf2);
	}

	gen_rand_str(magisk_name, sizeof(magisk_name));
	snprintf(buf, PATH_MAX, "/root/%s", magisk_name);
	unlink("/sbin/magisk");
	rename("/root/magisk", buf);
	xsymlink(buf, "/sbin/magisk");
	for (int i = 0; applet[i]; ++i) {
		snprintf(buf2, PATH_MAX, "/sbin/%s", applet[i]);
		xsymlink(buf, buf2);
	}

	gen_rand_str(init_name, sizeof(init_name));
	snprintf(buf, PATH_MAX, "/root/%s", init_name);
	unlink("/sbin/magiskinit");
	rename("/root/magiskinit", buf);
	for (int i = 0; init_applet[i]; ++i) {
		snprintf(buf2, PATH_MAX, "/sbin/%s", init_applet[i]);
		xsymlink(buf, buf2);
	}

	close(root);

	// Backward compatibility
	xsymlink(DATABIN, "/data/magisk");
	xsymlink(MAINIMG, "/data/magisk.img");
	xsymlink(MOUNTPOINT, "/magisk");

	xmount(NULL, "/", NULL, MS_REMOUNT | MS_RDONLY, NULL);

	LOGI("* Mounting mirrors");
	struct vector mounts;
	vec_init(&mounts);
	file_to_vector("/proc/mounts", &mounts);
	char *line;
	int skip_initramfs = 0;
	// Check whether skip_initramfs device
	vec_for_each(&mounts, line) {
		if (strstr(line, " /system_root ")) {
			xmkdirs(MIRRDIR "/system", 0755);
			bind_mount("/system_root/system", MIRRDIR "/system");
			skip_initramfs = 1;
			break;
		}
	}
	vec_for_each(&mounts, line) {
		if (!skip_initramfs && strstr(line, " /system ")) {
			sscanf(line, "%s", buf);
			xmkdirs(MIRRDIR "/system", 0755);
			xmount(buf, MIRRDIR "/system", "ext4", MS_RDONLY, NULL);
			#ifdef MAGISK_DEBUG
				LOGI("mount: %s <- %s\n", MIRRDIR "/system", buf);
			#else
				LOGI("mount: %s\n", MIRRDIR "/system");
			#endif
		} else if (strstr(line, " /vendor ")) {
			seperate_vendor = 1;
			sscanf(line, "%s", buf);
			xmkdirs(MIRRDIR "/vendor", 0755);
			xmount(buf, MIRRDIR "/vendor", "ext4", MS_RDONLY, NULL);
			#ifdef MAGISK_DEBUG
				LOGI("mount: %s <- %s\n", MIRRDIR "/vendor", buf);
			#else
				LOGI("mount: %s\n", MIRRDIR "/vendor");
			#endif
		}
		free(line);
	}
	vec_destroy(&mounts);
	if (!seperate_vendor) {
		xsymlink(MIRRDIR "/system/vendor", MIRRDIR "/vendor");
		#ifdef MAGISK_DEBUG
			LOGI("link: %s <- %s\n", MIRRDIR "/vendor", MIRRDIR "/system/vendor");
		#else
			LOGI("link: %s\n", MIRRDIR "/vendor");
		#endif
	}
	xmkdirs(MIRRDIR "/bin", 0755);
	bind_mount(DATABIN, MIRRDIR "/bin");

	LOGI("* Setting up internal busybox");
	xmkdirs(BBPATH, 0755);
	exec_command_sync(MIRRDIR "/bin/busybox", "--install", "-s", BBPATH, NULL);
	xsymlink(MIRRDIR "/bin/busybox", BBPATH "/busybox");
}

void start_daemon() {
	setsid();
	setcon("u:r:su:s0");
	umask(0);
	int fd = xopen("/dev/null", O_RDWR | O_CLOEXEC);
	xdup2(fd, STDIN_FILENO);
	xdup2(fd, STDOUT_FILENO);
	xdup2(fd, STDERR_FILENO);
	close(fd);

	// Block user signals
	sigset_t block_set;
	sigemptyset(&block_set);
	sigaddset(&block_set, SIGUSR1);
	sigaddset(&block_set, SIGUSR2);
	pthread_sigmask(SIG_SETMASK, &block_set, NULL);

	struct sockaddr_un sun;
	fd = setup_socket(&sun);

	if (xbind(fd, (struct sockaddr*) &sun, sizeof(sun)))
		exit(1);
	xlisten(fd, 10);

	// Start the log monitor
	monitor_logs();

	if ((is_daemon_init = (access(MAGISKTMP, F_OK) == 0))) {
		// Restart stuffs if the daemon is restarted
		exec_command_sync("logcat", "-b", "all", "-c", NULL);
		auto_start_magiskhide();
		start_debug_log();
	} else if (check_data()) {
		daemon_init();
	}

	LOGI("Magisk v" xstr(MAGISK_VERSION) "(" xstr(MAGISK_VER_CODE) ") daemon started\n");

	// Change process name
	strcpy(argv0, "magisk_daemon");

	// Unlock all blocks for rw
	unlock_blocks();

	// Loop forever to listen for requests
	while(1) {
		int *client = xmalloc(sizeof(int));
		*client = xaccept4(fd, NULL, NULL, SOCK_CLOEXEC);
		pthread_t thread;
		xpthread_create(&thread, NULL, request_handler, client);
		// Detach the thread, we will never join it
		pthread_detach(thread);
	}
}

/* Connect the daemon, and return a socketfd */
int connect_daemon() {
	struct sockaddr_un sun;
	int fd = setup_socket(&sun);
	if (connect(fd, (struct sockaddr*) &sun, sizeof(sun))) {
		// If we cannot access the daemon, we start a daemon in the child process if possible

		if (getuid() != UID_ROOT || getgid() != UID_ROOT) {
			fprintf(stderr, "No daemon is currently running!\n");
			exit(1);
		}

		if (xfork() == 0) {
			LOGD("client: connect fail, try launching new daemon process\n");
			close(fd);
			start_daemon();
		}

		while (connect(fd, (struct sockaddr*) &sun, sizeof(sun)))
			usleep(10000);
	}
	return fd;
}