Magisk/native/jni/resetprop/resetprop.cpp

305 lines
8.0 KiB
C++
Raw Normal View History

#include <dlfcn.h>
2017-04-14 00:06:41 +02:00
#include <sys/types.h>
2019-01-20 05:59:37 +01:00
#include <vector>
2020-05-07 15:08:30 +02:00
#include <map>
2017-04-14 00:06:41 +02:00
2020-03-09 09:50:30 +01:00
#include <logging.hpp>
#include <resetprop.hpp>
#include <utils.hpp>
2019-02-10 09:57:51 +01:00
#include <flags.h>
2017-04-14 00:06:41 +02:00
#define _REALLY_INCLUDE_SYS__SYSTEM_PROPERTIES_H_
2020-03-09 09:50:30 +01:00
#include <_system_properties.h>
#include "_resetprop.hpp"
2017-04-27 19:45:57 +02:00
2019-01-20 05:59:37 +01:00
using namespace std;
2018-11-03 04:56:15 +01:00
static bool verbose = false;
2017-04-14 00:06:41 +02:00
static int (*system_property_get)(const char*, char*);
static int (*system_property_set)(const char*, const char*);
[[noreturn]] static void usage(char* arg0) {
fprintf(stderr,
NAME_WITH_VER(resetprop) R"EOF(" - System Props Modification Tool
Usage: %s [flags] [options...]
Options:
-h, --help show this message
(no arguments) print all properties
NAME get property
NAME VALUE set property entry NAME with VALUE
--file FILE load props from FILE
--delete NAME delete property
Flags:
-v print verbose output to stderr
-n set properties without going through init
affects setprop and prop file loading
-p also access props directly from persist storage
affects getprop and delprop
)EOF", arg0);
exit(1);
}
2018-11-03 04:56:15 +01:00
static bool check_legal_property_name(const char *name) {
2017-10-09 20:04:50 +02:00
int namelen = strlen(name);
if (namelen < 1) goto illegal;
if (name[0] == '.') goto illegal;
if (name[namelen - 1] == '.') goto illegal;
/* Only allow alphanumeric, plus '.', '-', '@', ':', or '_' */
/* Don't allow ".." to appear in a property name */
for (size_t i = 0; i < namelen; i++) {
if (name[i] == '.') {
// i=0 is guaranteed to never have a dot. See above.
if (name[i-1] == '.') goto illegal;
continue;
}
if (name[i] == '_' || name[i] == '-' || name[i] == '@' || name[i] == ':') continue;
if (name[i] >= 'a' && name[i] <= 'z') continue;
if (name[i] >= 'A' && name[i] <= 'Z') continue;
if (name[i] >= '0' && name[i] <= '9') continue;
goto illegal;
}
2018-11-03 04:56:15 +01:00
return true;
2017-10-09 20:04:50 +02:00
illegal:
2018-09-27 09:11:10 +02:00
LOGE("Illegal property name: [%s]\n", name);
2018-11-03 04:56:15 +01:00
return false;
2017-02-28 09:52:27 +01:00
}
static void read_prop(const prop_info *pi, void *cb) {
2018-11-03 04:56:15 +01:00
__system_property_read_callback(
2019-07-01 04:09:31 +02:00
pi, [](auto cb, auto name, auto value, auto) {
2020-05-07 15:08:30 +02:00
reinterpret_cast<prop_cb*>(cb)->exec(name, value);
}, cb);
2018-04-28 14:22:42 +02:00
}
2018-11-04 09:38:06 +01:00
static void print_props(bool persist) {
2020-05-07 15:08:30 +02:00
getprops([](auto name, auto value, auto) {
printf("[%s]: [%s]\n", name, value);
}, nullptr, persist);
2018-04-28 14:22:42 +02:00
}
struct sysprop {
virtual int setprop(const char *name, const char *value, bool trigger) { return 1; }
virtual string getprop(const char *name, bool persist) { return ""; }
virtual void getprops(void (*callback)(const char *, const char *, void *),
void *cookie, bool persist) {}
virtual int delprop(const char *name, bool persist) { return 1; }
};
struct stock_sysprop : public sysprop {
int setprop(const char *name, const char *value, bool) override {
return system_property_set(name, value);
}
string getprop(const char *name, bool) override {
2020-05-07 15:08:30 +02:00
char buf[PROP_VALUE_MAX];
buf[0] = '\0';
system_property_get(name, buf);
2020-05-07 15:08:30 +02:00
return buf;
2017-10-09 20:04:50 +02:00
}
};
struct resetprop : public sysprop {
int setprop(const char *name, const char *value, bool trigger) override {
if (!check_legal_property_name(name))
return 1;
int ret;
auto pi = const_cast<prop_info *>(__system_property_find(name));
if (pi != nullptr) {
if (trigger) {
if (strncmp(name, "ro.", 3) == 0)
delprop(name, false);
ret = system_property_set(name, value);
} else {
ret = __system_property_update(pi, value, strlen(value));
}
LOGD("resetprop: update prop [%s]: [%s] by %s\n", name, value,
trigger ? "property_service" : "modifying prop data structure");
} else {
if (trigger) {
ret = system_property_set(name, value);
} else {
ret = __system_property_add(name, strlen(name), value, strlen(value));
}
LOGD("resetprop: create prop [%s]: [%s] by %s\n", name, value,
trigger ? "property_service" : "modifying prop data structure");
}
2017-07-18 05:56:57 +02:00
if (ret)
LOGE("resetprop: setprop error\n");
2018-11-03 04:56:15 +01:00
return ret;
}
2017-10-09 20:04:50 +02:00
string getprop(const char *name, bool persist) override {
if (!check_legal_property_name(name))
return string();
auto pi = __system_property_find(name);
if (pi == nullptr) {
if (persist && strncmp(name, "persist.", 8) == 0) {
auto value = persist_getprop(name);
if (value.empty())
goto not_found;
return value;
}
not_found:
LOGD("resetprop: prop [%s] does not exist\n", name);
return string();
2017-10-09 20:04:50 +02:00
} else {
string val;
auto reader = make_prop_cb(val, [](auto, auto value, auto str){ str = value; });
read_prop(pi, &reader);
LOGD("resetprop: getprop [%s]: [%s]\n", name, val.data());
return val;
2017-10-09 20:04:50 +02:00
}
}
void getprops(void (*callback)(const char *, const char *, void *),
void *cookie, bool persist) override {
prop_list list;
prop_collector collector(list);
__system_property_foreach(read_prop, &collector);
if (persist)
persist_getprops(&collector);
for (auto &[key, val] : list)
callback(key.data(), val.data(), cookie);
if (persist)
persist_cleanup();
}
int delprop(const char *name, bool persist) override {
if (!check_legal_property_name(name))
return 1;
LOGD("resetprop: delete prop [%s]\n", name);
if (persist && strncmp(name, "persist.", 8) == 0)
persist = persist_deleteprop(name);
return __system_property_delete(name) && !(persist && strncmp(name, "persist.", 8) == 0);
}
};
#define DLOAD(name) \
*(void **) &name = dlsym(RTLD_DEFAULT, "__" #name)
static sysprop *get_impl() {
static sysprop *impl = nullptr;
if (impl == nullptr) {
use_pb = access(PERSISTENT_PROPERTY_DIR "/persistent_properties", R_OK) == 0;
#ifdef APPLET_STUB_MAIN
if (__system_properties_init()) {
LOGE("resetprop: __system_properties_init error\n");
exit(1);
}
impl = new resetprop();
system_property_set = &__system_property_set;
#else
DLOAD(system_property_set); /* Always prefer to use setprop from platform */
if (__system_properties_init()) {
LOGW("resetprop: __system_properties_init error\n");
DLOAD(system_property_get);
impl = system_property_get ? new stock_sysprop() : new sysprop();
2017-10-09 20:04:50 +02:00
} else {
impl = new resetprop();
2017-10-09 20:04:50 +02:00
}
#endif
2017-10-09 20:04:50 +02:00
}
return impl;
}
/*********************************
* Implementation of public APIs
*********************************/
2017-10-09 20:04:50 +02:00
string getprop(const char *name, bool persist) {
return get_impl()->getprop(name, persist);
}
2017-10-09 20:04:50 +02:00
void getprops(void (*callback)(const char *, const char *, void *), void *cookie, bool persist) {
get_impl()->getprops(callback, cookie, persist);
}
2017-10-09 20:04:50 +02:00
int setprop(const char *name, const char *value, bool trigger) {
return get_impl()->setprop(name, value, trigger);
2017-10-09 20:04:50 +02:00
}
2020-03-09 10:05:24 +01:00
int delprop(const char *name, bool persist) {
return get_impl()->delprop(name, persist);
2017-02-28 09:52:27 +01:00
}
2019-03-06 02:27:09 +01:00
void load_prop_file(const char *filename, bool trigger) {
auto impl = get_impl();
2019-01-20 21:20:34 +01:00
LOGD("resetprop: Parse prop file [%s]\n", filename);
2019-03-06 02:27:09 +01:00
parse_prop_file(filename, [=](auto key, auto val) -> bool {
impl->setprop(key.data(), val.data(), trigger);
2019-01-20 21:20:34 +01:00
return true;
});
}
int resetprop_main(int argc, char *argv[]) {
2018-11-03 04:56:15 +01:00
log_cb.d = [](auto fmt, auto ap) -> int { return verbose ? vfprintf(stderr, fmt, ap) : 0; };
2018-11-04 09:38:06 +01:00
bool trigger = true, persist = false;
2018-11-07 08:10:38 +01:00
char *argv0 = argv[0];
2019-01-20 05:59:37 +01:00
string prop;
2017-10-09 20:04:50 +02:00
--argc;
++argv;
// Parse flags and -- options
while (argc && argv[0][0] == '-') {
2019-01-20 05:59:37 +01:00
for (int idx = 1; true; ++idx) {
2017-10-09 20:04:50 +02:00
switch (argv[0][idx]) {
case '-':
if (strcmp(argv[0], "--file") == 0 && argc == 2) {
2019-03-06 02:27:09 +01:00
load_prop_file(argv[1], trigger);
return 0;
2017-10-09 20:04:50 +02:00
} else if (strcmp(argv[0], "--delete") == 0 && argc == 2) {
2020-03-09 10:05:24 +01:00
return delprop(argv[1], persist);
2017-10-09 20:04:50 +02:00
} else if (strcmp(argv[0], "--help") == 0) {
2018-11-03 04:56:15 +01:00
usage(argv0);
2017-10-09 20:04:50 +02:00
}
case 'v':
2018-11-03 04:56:15 +01:00
verbose = true;
2017-10-09 20:04:50 +02:00
continue;
case 'p':
2019-01-20 05:59:37 +01:00
persist = true;
2017-10-09 20:04:50 +02:00
continue;
case 'n':
2019-01-20 05:59:37 +01:00
trigger = false;
2017-10-09 20:04:50 +02:00
continue;
case '\0':
break;
case 'h':
default:
2018-11-03 04:56:15 +01:00
usage(argv0);
2017-10-09 20:04:50 +02:00
}
break;
}
--argc;
++argv;
}
switch (argc) {
case 0:
2018-04-28 14:22:42 +02:00
print_props(persist);
2017-10-09 20:04:50 +02:00
return 0;
case 1:
2019-01-20 05:59:37 +01:00
prop = getprop(argv[0], persist);
if (prop.empty()) return 1;
printf("%s\n", prop.data());
2017-10-09 20:04:50 +02:00
return 0;
case 2:
2018-11-04 09:38:06 +01:00
return setprop(argv[0], argv[1], trigger);
2017-10-09 20:04:50 +02:00
default:
usage(argv0);
}
2017-02-28 09:52:27 +01:00
}