kernel: Remove fallback scan for packages.list

Enhance scan support for active users in /data/user_de

Signed-off-by: ShirkNeko <109797057+ShirkNeko@users.noreply.github.com>
This commit is contained in:
ShirkNeko
2025-09-12 16:02:51 +08:00
parent eebde9be16
commit 316cb79f32
2 changed files with 166 additions and 130 deletions

View File

@@ -298,26 +298,19 @@ int ksu_handle_rename(struct dentry *old_dentry, struct dentry *new_dentry)
return 0;
}
// /data/system/packages.list.tmp -> /data/system/packages.list
if (strcmp(new_dentry->d_iname, "packages.list")) {
// It still monitors changes to certain system files to trigger scans, but does not rely on packages.list.
if (strstr(new_dentry->d_iname, "packages") &&
strstr(new_dentry->d_iname, "list")) {
char path[128];
char *buf = dentry_path_raw(new_dentry, path, sizeof(path));
if (!IS_ERR(buf) && strstr(buf, "/system/")) {
pr_info("System package change detected: %s -> %s, triggering user scan\n",
old_dentry->d_iname, new_dentry->d_iname);
track_throne();
}
return 0;
}
char path[128];
char *buf = dentry_path_raw(new_dentry, path, sizeof(path));
if (IS_ERR(buf)) {
pr_err("dentry_path_raw failed.\n");
return 0;
}
if (!strstr(buf, "/system/packages.list")) {
return 0;
}
pr_info("renameat: %s -> %s, new path: %s\n", old_dentry->d_iname,
new_dentry->d_iname, buf);
track_throne();
return 0;
}

View File

@@ -18,9 +18,11 @@
uid_t ksu_manager_uid = KSU_INVALID_UID;
#define SYSTEM_PACKAGES_LIST_PATH "/data/system/packages.list.tmp"
#define USER_DATA_PATH "/data/user_de/0"
#define USER_DATA_PATH_LEN 256
#define USER_DATA_BASE_PATH "/data/user_de"
#define USER_DATA_LEGACY_BASE_PATH "/data/user"
#define USER_DATA_PATH_LEN 256 // 256 is enough for /data/user_de/{userid}/<package>
#define MAX_ANDROID_USER_ID 999
#define MIN_ANDROID_USER_ID 0
struct uid_data {
struct list_head list;
@@ -28,6 +30,13 @@ struct uid_data {
char package[KSU_MAX_PACKAGE_NAME];
};
struct user_scan_context {
struct list_head *uid_list;
uid_t user_id;
size_t packages_found;
size_t errors_count;
};
static int get_pkg_from_apk_path(char *pkg, const char *path)
{
int len = strlen(path);
@@ -150,10 +159,53 @@ struct uid_scan_stats {
struct user_data_context {
struct dir_context ctx;
struct list_head *uid_list;
struct uid_scan_stats *stats;
struct user_scan_context *scan_ctx;
};
// Retrieve a list of all active Android user IDs in the system
static int get_active_user_ids(uid_t *user_ids, size_t max_users, size_t *found_users)
{
struct file *dir_file;
struct dir_context ctx __maybe_unused = {0};
int ret = 0;
size_t count = 0;
*found_users = 0;
dir_file = ksu_filp_open_compat(USER_DATA_BASE_PATH, O_RDONLY, 0);
if (IS_ERR(dir_file)) {
pr_warn("Failed to open %s: %ld, trying legacy path\n",
USER_DATA_BASE_PATH, PTR_ERR(dir_file));
// Try the legacy path
dir_file = ksu_filp_open_compat(USER_DATA_LEGACY_BASE_PATH, O_RDONLY, 0);
if (IS_ERR(dir_file)) {
pr_err("Failed to open both user data paths: %ld\n", PTR_ERR(dir_file));
return PTR_ERR(dir_file);
}
}
struct path path;
for (uid_t user_id = MIN_ANDROID_USER_ID; user_id <= MAX_ANDROID_USER_ID && count < max_users; user_id++) {
char user_path[USER_DATA_PATH_LEN];
snprintf(user_path, sizeof(user_path), "%s/%u", USER_DATA_BASE_PATH, user_id);
if (!kern_path(user_path, 0, &path)) {
user_ids[count++] = user_id;
path_put(&path);
}
}
filp_close(dir_file, NULL);
*found_users = count;
if (count > 0) {
pr_info("UserDE UID: Found %zu active users\n", count);
}
return ret;
}
FILLDIR_RETURN_TYPE user_data_actor(struct dir_context *ctx, const char *name,
int namelen, loff_t off, u64 ino,
unsigned int d_type)
@@ -161,10 +213,12 @@ FILLDIR_RETURN_TYPE user_data_actor(struct dir_context *ctx, const char *name,
struct user_data_context *my_ctx =
container_of(ctx, struct user_data_context, ctx);
if (!my_ctx || !my_ctx->uid_list) {
if (!my_ctx || !my_ctx->scan_ctx || !my_ctx->scan_ctx->uid_list) {
return FILLDIR_ACTOR_STOP;
}
struct user_scan_context *scan_ctx = my_ctx->scan_ctx;
if (!strncmp(name, "..", namelen) || !strncmp(name, ".", namelen))
return FILLDIR_ACTOR_CONTINUE;
@@ -172,18 +226,16 @@ FILLDIR_RETURN_TYPE user_data_actor(struct dir_context *ctx, const char *name,
return FILLDIR_ACTOR_CONTINUE;
if (namelen >= KSU_MAX_PACKAGE_NAME) {
pr_warn("Package name too long: %.*s\n", namelen, name);
if (my_ctx->stats)
my_ctx->stats->errors_encountered++;
pr_warn("Package name too long: %.*s (user %u)\n", namelen, name, scan_ctx->user_id);
scan_ctx->errors_count++;
return FILLDIR_ACTOR_CONTINUE;
}
char package_path[USER_DATA_PATH_LEN];
if (snprintf(package_path, sizeof(package_path), "%s/%.*s",
USER_DATA_PATH, namelen, name) >= sizeof(package_path)) {
pr_err("Path too long for package: %.*s\n", namelen, name);
if (my_ctx->stats)
my_ctx->stats->errors_encountered++;
if (snprintf(package_path, sizeof(package_path), "%s/%u/%.*s",
USER_DATA_BASE_PATH, scan_ctx->user_id, namelen, name) >= sizeof(package_path)) {
pr_err("Path too long for package: %.*s (user %u)\n", namelen, name, scan_ctx->user_id);
scan_ctx->errors_count++;
return FILLDIR_ACTOR_CONTINUE;
}
@@ -191,8 +243,7 @@ FILLDIR_RETURN_TYPE user_data_actor(struct dir_context *ctx, const char *name,
int err = kern_path(package_path, LOOKUP_FOLLOW, &path);
if (err) {
pr_debug("Package path lookup failed: %s (err: %d)\n", package_path, err);
if (my_ctx->stats)
my_ctx->stats->errors_encountered++;
scan_ctx->errors_count++;
return FILLDIR_ACTOR_CONTINUE;
}
@@ -202,24 +253,21 @@ FILLDIR_RETURN_TYPE user_data_actor(struct dir_context *ctx, const char *name,
if (err) {
pr_debug("Failed to get attributes for: %s (err: %d)\n", package_path, err);
if (my_ctx->stats)
my_ctx->stats->errors_encountered++;
scan_ctx->errors_count++;
return FILLDIR_ACTOR_CONTINUE;
}
uid_t uid = from_kuid(&init_user_ns, stat.uid);
if (uid == (uid_t)-1) {
pr_warn("Invalid UID for package: %.*s\n", namelen, name);
if (my_ctx->stats)
my_ctx->stats->errors_encountered++;
pr_warn("Invalid UID for package: %.*s (user %u)\n", namelen, name, scan_ctx->user_id);
scan_ctx->errors_count++;
return FILLDIR_ACTOR_CONTINUE;
}
struct uid_data *data = kzalloc(sizeof(struct uid_data), GFP_ATOMIC);
if (!data) {
pr_err("Failed to allocate memory for package: %.*s\n", namelen, name);
if (my_ctx->stats)
my_ctx->stats->errors_encountered++;
pr_err("Failed to allocate memory for package: %.*s (user %u)\n", namelen, name, scan_ctx->user_id);
scan_ctx->errors_count++;
return FILLDIR_ACTOR_CONTINUE;
}
@@ -228,50 +276,105 @@ FILLDIR_RETURN_TYPE user_data_actor(struct dir_context *ctx, const char *name,
strncpy(data->package, name, copy_len);
data->package[copy_len] = '\0';
list_add_tail(&data->list, my_ctx->uid_list);
list_add_tail(&data->list, scan_ctx->uid_list);
scan_ctx->packages_found++;
if (my_ctx->stats)
my_ctx->stats->total_found++;
pr_info("UserDE UID: Found package: %s, uid: %u\n", data->package, data->uid);
pr_info("UserDE UID: Found package: %s, uid: %u (user %u)\n",
data->package, data->uid, scan_ctx->user_id);
return FILLDIR_ACTOR_CONTINUE;
}
int scan_user_data_for_uids(struct list_head *uid_list)
static int scan_user_data_for_user(uid_t user_id, struct list_head *uid_list,
size_t *packages_found, size_t *errors_count)
{
struct file *dir_file;
struct uid_scan_stats stats = {0};
char user_path[USER_DATA_PATH_LEN];
int ret = 0;
*packages_found = 0;
*errors_count = 0;
snprintf(user_path, sizeof(user_path), "%s/%u", USER_DATA_BASE_PATH, user_id);
dir_file = ksu_filp_open_compat(user_path, O_RDONLY, 0);
if (IS_ERR(dir_file)) {
pr_debug("Failed to open user data path: %s (%ld)\n", user_path, PTR_ERR(dir_file));
return PTR_ERR(dir_file);
}
struct user_scan_context scan_ctx = {
.uid_list = uid_list,
.user_id = user_id,
.packages_found = 0,
.errors_count = 0
};
struct user_data_context ctx = {
.ctx.actor = user_data_actor,
.scan_ctx = &scan_ctx
};
ret = iterate_dir(dir_file, &ctx.ctx);
filp_close(dir_file, NULL);
*packages_found = scan_ctx.packages_found;
*errors_count = scan_ctx.errors_count;
if (scan_ctx.packages_found > 0 && scan_ctx.errors_count > 0) {
pr_info("UserDE UID: Scanned user %u, found %zu packages with %zu errors\n",
user_id, scan_ctx.packages_found, scan_ctx.errors_count);
}
return ret;
}
int scan_user_data_for_uids(struct list_head *uid_list)
{
uid_t user_ids[32]; // Supports up to 32 users
size_t active_users = 0;
size_t total_packages = 0;
size_t total_errors = 0;
int ret = 0;
if (!uid_list) {
return -EINVAL;
}
dir_file = ksu_filp_open_compat(USER_DATA_PATH, O_RDONLY, 0);
if (IS_ERR(dir_file)) {
pr_err("UserDE UID: Failed to open %s: %ld\n", USER_DATA_PATH, PTR_ERR(dir_file));
return PTR_ERR(dir_file);
// Retrieve all active user IDs
ret = get_active_user_ids(user_ids, ARRAY_SIZE(user_ids), &active_users);
if (ret < 0 || active_users == 0) {
pr_err("Failed to get active user IDs or no users found: %d\n", ret);
return -ENOENT;
}
struct user_data_context ctx = {
.ctx.actor = user_data_actor,
.uid_list = uid_list,
.stats = &stats
};
ret = iterate_dir(dir_file, &ctx.ctx);
filp_close(dir_file, NULL);
if (stats.errors_encountered > 0) {
pr_warn("Encountered %zu errors while scanning user data directory\n",
stats.errors_encountered);
// Scan each user's data directory
for (size_t i = 0; i < active_users; i++) {
uid_t user_id = user_ids[i];
size_t packages_found = 0;
size_t errors_count = 0;
ret = scan_user_data_for_user(user_id, uid_list, &packages_found, &errors_count);
if (ret < 0) {
pr_warn("Failed to scan user %u data directory: %d\n", user_id, ret);
total_errors++;
continue;
}
total_packages += packages_found;
total_errors += errors_count;
}
pr_info("UserDE UID: Scanned user data directory, found %zu packages with %zu errors\n",
stats.total_found, stats.errors_encountered);
if (total_errors > 0) {
pr_warn("UserDE UID: Encountered %zu errors while scanning user data directories\n",
total_errors);
}
return ret;
pr_info("UserDE UID: Scan completed - %zu users, %zu packages found\n",
active_users, total_packages);
return total_packages > 0 ? 0 : -ENOENT;
}
FILLDIR_RETURN_TYPE my_actor(struct dir_context *ctx, const char *name,
@@ -469,72 +572,12 @@ void track_throne(void)
{
struct list_head uid_list;
INIT_LIST_HEAD(&uid_list);
pr_info("Starting UID scan from user data directory\n");
// scan user data for uids
int ret = scan_user_data_for_uids(&uid_list);
if (ret < 0) {
pr_warn("Failed to scan user data directory (%d), falling back to packages.list\n", ret);
// fallback to packages.list method
struct file *fp = ksu_filp_open_compat(SYSTEM_PACKAGES_LIST_PATH, O_RDONLY, 0);
if (IS_ERR(fp)) {
pr_err("Both user data scan and packages.list failed: %ld\n", PTR_ERR(fp));
goto out;
}
char chr = 0;
loff_t pos = 0;
loff_t line_start = 0;
char buf[KSU_MAX_PACKAGE_NAME];
size_t fallback_count = 0;
for (;;) {
ssize_t count =
ksu_kernel_read_compat(fp, &chr, sizeof(chr), &pos);
if (count != sizeof(chr))
break;
if (chr != '\n')
continue;
count = ksu_kernel_read_compat(fp, buf, sizeof(buf),
&line_start);
struct uid_data *data =
kzalloc(sizeof(struct uid_data), GFP_ATOMIC);
if (!data) {
filp_close(fp, 0);
goto out;
}
char *tmp = buf;
const char *delim = " ";
char *package = strsep(&tmp, delim);
char *uid = strsep(&tmp, delim);
if (!uid || !package) {
pr_err("update_uid: package or uid is NULL!\n");
kfree(data);
break;
}
u32 res;
if (kstrtou32(uid, 10, &res)) {
pr_err("update_uid: uid parse err\n");
kfree(data);
break;
}
data->uid = res;
strncpy(data->package, package, KSU_MAX_PACKAGE_NAME);
list_add_tail(&data->list, &uid_list);
fallback_count++;
// reset line start
line_start = pos;
}
filp_close(fp, 0);
pr_info("Loaded %zu packages from packages.list fallback\n", fallback_count);
} else {
pr_info("UserDE UID: Successfully loaded %zu packages from user data directory\n", list_count_nodes(&uid_list));
pr_err("UserDE UID scan user data failed: %d.\n", ret);
goto out;
}
// now update uid list