Magisk/native/jni/core/scripting.cpp

95 lines
2.2 KiB
C++
Raw Normal View History

2019-02-16 02:45:05 +01:00
#include <unistd.h>
#include <cstdio>
#include <cstdlib>
#include <string>
#include <vector>
2020-03-09 09:50:30 +01:00
#include <magisk.hpp>
#include <utils.hpp>
#include <selinux.hpp>
2019-02-16 02:45:05 +01:00
using namespace std;
#define BBEXEC_CMD DATABIN "/busybox", "sh"
static void set_standalone() {
setenv("ASH_STANDALONE", "1", 1);
};
2019-02-16 02:45:05 +01:00
void exec_script(const char *script) {
exec_t exec {
.pre_exec = set_standalone,
.fork = fork_no_zombie
};
2020-03-11 08:11:15 +01:00
exec_command_sync(exec, BBEXEC_CMD, script);
}
2020-05-18 14:36:02 +02:00
void exec_common_scripts(const char *stage) {
LOGI("* Running %s.d scripts\n", stage);
2019-02-16 02:45:05 +01:00
char path[4096];
2019-12-21 11:29:38 +01:00
char *name = path + sprintf(path, SECURE_DIR "/%s.d", stage);
2019-12-13 06:37:06 +01:00
auto dir = xopen_dir(path);
if (!dir)
2019-02-16 02:45:05 +01:00
return;
2019-12-21 11:29:38 +01:00
int dfd = dirfd(dir.get());
2019-12-13 06:37:06 +01:00
bool pfs = stage == "post-fs-data"sv;
2019-12-21 11:29:38 +01:00
*(name++) = '/';
2019-12-13 06:37:06 +01:00
for (dirent *entry; (entry = xreaddir(dir.get()));) {
2019-02-16 02:45:05 +01:00
if (entry->d_type == DT_REG) {
2019-12-21 11:29:38 +01:00
if (faccessat(dfd, entry->d_name, X_OK, 0) != 0)
2019-02-16 02:45:05 +01:00
continue;
LOGI("%s.d: exec [%s]\n", stage, entry->d_name);
2019-12-21 11:29:38 +01:00
strcpy(name, entry->d_name);
2019-02-16 02:45:05 +01:00
exec_t exec {
.pre_exec = set_standalone,
2019-03-23 08:36:35 +01:00
.fork = pfs ? fork_no_zombie : fork_dont_care
2019-02-16 02:45:05 +01:00
};
if (pfs)
2020-03-11 08:11:15 +01:00
exec_command_sync(exec, BBEXEC_CMD, path);
2019-02-16 02:45:05 +01:00
else
2020-03-11 08:11:15 +01:00
exec_command(exec, BBEXEC_CMD, path);
2019-02-16 02:45:05 +01:00
}
}
}
2020-05-18 14:36:02 +02:00
void exec_module_scripts(const char *stage, const vector<string> &module_list) {
LOGI("* Running module %s scripts\n", stage);
2019-02-16 02:45:05 +01:00
char path[4096];
2019-12-13 06:37:06 +01:00
bool pfs = stage == "post-fs-data"sv;
2019-02-16 02:45:05 +01:00
for (auto &m : module_list) {
const char* module = m.c_str();
sprintf(path, MODULEROOT "/%s/%s.sh", module, stage);
if (access(path, F_OK) == -1)
continue;
LOGI("%s: exec [%s.sh]\n", module, stage);
exec_t exec {
.pre_exec = set_standalone,
2019-03-23 08:36:35 +01:00
.fork = pfs ? fork_no_zombie : fork_dont_care
2019-02-16 02:45:05 +01:00
};
if (pfs)
2020-03-11 08:11:15 +01:00
exec_command_sync(exec, BBEXEC_CMD, path);
2019-02-16 02:45:05 +01:00
else
2020-03-11 08:11:15 +01:00
exec_command(exec, BBEXEC_CMD, path);
2019-02-16 02:45:05 +01:00
}
}
2019-10-09 23:38:45 +02:00
constexpr char install_script[] = R"EOF(
APK=%s
log -t Magisk "apk_install: $APK"
log -t Magisk "apk_install: `pm install -r $APK 2>&1`"
rm -f $APK
)EOF";
2019-02-16 02:45:05 +01:00
void install_apk(const char *apk) {
setfilecon(apk, "u:object_r:" SEPOL_FILE_DOMAIN ":s0");
exec_t exec {
.pre_exec = set_standalone,
.fork = fork_no_zombie
};
2019-02-16 02:45:05 +01:00
char cmds[sizeof(install_script) + 4096];
sprintf(cmds, install_script, apk);
2020-03-11 08:11:15 +01:00
exec_command_sync(exec, BBEXEC_CMD, "-c", cmds);
}