Magisk/native/jni/core/daemon.cpp

199 lines
4.3 KiB
C++
Raw Normal View History

2017-04-08 01:37:43 +02:00
/* 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>
2017-04-08 01:37:43 +02:00
#include <sys/un.h>
#include <sys/types.h>
2017-11-27 21:43:46 +01:00
#include <sys/mount.h>
2017-04-08 01:37:43 +02:00
2019-02-10 09:57:51 +01:00
#include <magisk.h>
#include <utils.h>
#include <daemon.h>
#include <selinux.h>
#include <db.h>
#include <resetprop.h>
#include <flags.h>
2019-01-20 23:52:19 +01:00
int SDK_INT = -1;
bool RECOVERY_MODE = false;
static struct stat SERVER_STAT;
2019-01-20 23:52:19 +01:00
static void verify_client(int client, pid_t pid) {
// Verify caller is the same as server
char path[32];
sprintf(path, "/proc/%d/exe", pid);
struct stat st{};
stat(path, &st);
if (st.st_dev != SERVER_STAT.st_dev || st.st_ino != SERVER_STAT.st_ino) {
close(client);
pthread_exit(nullptr);
}
2018-10-13 03:46:09 +02:00
}
2017-04-21 18:54:08 +02:00
static void *request_handler(void *args) {
int client = *((int *) args);
2018-11-04 09:38:06 +01:00
delete (int *) args;
2017-05-05 10:13:26 +02:00
struct ucred credential;
get_client_cred(client, &credential);
if (credential.uid != 0)
verify_client(client, credential.pid);
2017-05-05 10:13:26 +02:00
int req = read_int(client);
2017-05-05 10:13:26 +02:00
switch (req) {
case MAGISKHIDE:
2017-05-05 10:13:26 +02:00
case POST_FS_DATA:
case LATE_START:
2018-08-09 08:52:44 +02:00
case BOOT_COMPLETE:
case SQLITE_CMD:
case BROADCAST_ACK:
if (credential.uid != 0) {
2017-05-05 10:13:26 +02:00
write_int(client, ROOT_REQUIRED);
close(client);
2019-01-20 23:52:19 +01:00
return nullptr;
2017-05-05 10:13:26 +02:00
}
default:
break;
}
switch (req) {
case MAGISKHIDE:
magiskhide_handler(client);
break;
case SUPERUSER:
2018-10-04 10:59:51 +02:00
su_daemon_handler(client, &credential);
2017-04-14 21:23:09 +02:00
break;
case CHECK_VERSION:
2019-02-12 11:17:02 +01:00
write_string(client, MAGISK_VERSION ":MAGISK");
2017-04-14 21:23:09 +02:00
close(client);
break;
case CHECK_VERSION_CODE:
2017-04-18 15:31:12 +02:00
write_int(client, MAGISK_VER_CODE);
2017-04-14 21:23:09 +02:00
close(client);
break;
2017-04-15 13:02:07 +02:00
case POST_FS_DATA:
post_fs_data(client);
2017-04-15 13:02:07 +02:00
break;
case LATE_START:
late_start(client);
2017-04-15 13:02:07 +02:00
break;
2018-08-09 08:52:44 +02:00
case BOOT_COMPLETE:
boot_complete(client);
break;
case SQLITE_CMD:
exec_sql(client);
2019-03-04 22:45:18 +01:00
break;
case BROADCAST_ACK:
LOGD("* Use broadcasts for su logging and notify\n");
CONNECT_BROADCAST = true;
close(client);
2017-05-05 10:13:26 +02:00
default:
2018-07-02 16:11:28 +02:00
close(client);
2017-05-07 21:11:14 +02:00
break;
}
2018-11-07 08:10:38 +01:00
return nullptr;
2017-04-08 01:37:43 +02:00
}
2018-10-12 06:50:47 +02:00
static void main_daemon() {
2018-09-27 09:11:10 +02:00
android_logging();
2017-11-22 09:12:08 +01:00
setsid();
2018-11-04 09:38:06 +01:00
setcon("u:r:" SEPOL_PROC_DOMAIN ":s0");
restore_rootcon();
int fd = xopen("/dev/null", O_RDWR | O_CLOEXEC);
xdup2(fd, STDOUT_FILENO);
xdup2(fd, STDERR_FILENO);
close(fd);
2018-07-02 16:11:28 +02:00
fd = xopen("/dev/zero", O_RDWR | O_CLOEXEC);
xdup2(fd, STDIN_FILENO);
close(fd);
2017-04-08 01:37:43 +02:00
2019-02-12 11:17:02 +01:00
LOGI(SHOW_VER(Magisk) " daemon started\n");
// Get server stat
stat("/proc/self/exe", &SERVER_STAT);
2019-01-20 23:52:19 +01:00
// Get API level
parse_prop_file("/system/build.prop", [](auto key, auto val) -> bool {
2019-03-06 02:27:09 +01:00
if (key == "ro.build.version.sdk") {
LOGI("* Device API level: %s\n", val.data());
2019-03-08 02:31:35 +01:00
SDK_INT = parse_int(val);
2019-01-20 23:52:19 +01:00
return false;
}
return true;
});
// Load config status
parse_prop_file(MAGISKTMP "/config", [](auto key, auto val) -> bool {
if (key == "RECOVERYMODE" && val == "true")
RECOVERY_MODE = true;
return true;
});
struct sockaddr_un sun;
2018-10-12 06:50:47 +02:00
socklen_t len = setup_sockaddr(&sun, MAIN_SOCKET);
fd = xsocket(AF_LOCAL, SOCK_STREAM | SOCK_CLOEXEC, 0);
if (xbind(fd, (struct sockaddr*) &sun, len))
2017-11-27 20:42:48 +01:00
exit(1);
xlisten(fd, 10);
2017-04-08 01:37:43 +02:00
// Change process name
2019-02-15 10:31:39 +01:00
set_nice_name("magiskd");
2019-02-10 10:18:50 +01:00
// Block all signals
sigset_t block_set;
2019-02-10 10:18:50 +01:00
sigfillset(&block_set);
2019-01-20 23:52:19 +01:00
pthread_sigmask(SIG_SETMASK, &block_set, nullptr);
2017-05-07 21:11:14 +02:00
// Loop forever to listen for requests
for (;;) {
2018-11-04 09:38:06 +01:00
int *client = new int;
2019-01-20 23:52:19 +01:00
*client = xaccept4(fd, nullptr, nullptr, SOCK_CLOEXEC);
new_daemon_thread(request_handler, client);
2017-04-08 01:37:43 +02:00
}
}
2018-11-13 08:07:02 +01:00
int switch_mnt_ns(int pid) {
char mnt[32];
snprintf(mnt, sizeof(mnt), "/proc/%d/ns/mnt", pid);
if (access(mnt, R_OK) == -1) return 1; // Maybe process died..
2018-11-13 08:07:02 +01:00
int fd, ret;
fd = xopen(mnt, O_RDONLY);
if (fd < 0) return 1;
// Switch to its namespace
ret = xsetns(fd, 0);
close(fd);
return ret;
}
2019-03-04 22:45:18 +01:00
int connect_daemon(bool create) {
2017-04-08 01:37:43 +02:00
struct sockaddr_un sun;
2018-10-12 06:50:47 +02:00
socklen_t len = setup_sockaddr(&sun, MAIN_SOCKET);
int fd = xsocket(AF_LOCAL, SOCK_STREAM | SOCK_CLOEXEC, 0);
if (connect(fd, (struct sockaddr*) &sun, len)) {
2019-03-04 22:45:18 +01:00
if (!create || getuid() != UID_ROOT || getgid() != UID_ROOT) {
LOGE("No daemon is currently running!\n");
2017-10-13 18:08:12 +02:00
exit(1);
}
2018-10-12 06:50:47 +02:00
LOGD("client: launching new main daemon process\n");
2018-04-29 06:17:28 +02:00
if (fork_dont_care() == 0) {
2018-10-12 06:50:47 +02:00
close(fd);
main_daemon();
}
2018-10-12 06:50:47 +02:00
while (connect(fd, (struct sockaddr*) &sun, len))
2017-11-27 20:42:48 +01:00
usleep(10000);
2017-04-08 01:37:43 +02:00
}
return fd;
}