84 lines
2.2 KiB
C
84 lines
2.2 KiB
C
#include <libfdt.h>
|
|
#include <unistd.h>
|
|
#include <sys/mman.h>
|
|
|
|
#include "magiskboot.h"
|
|
#include "utils.h"
|
|
|
|
static void print_subnode(const void *fdt, int parent, int depth) {
|
|
int node;
|
|
fdt_for_each_subnode(node, fdt, parent) {
|
|
for (int i = 0; i < depth; ++i) printf(" ");
|
|
printf("%d: %s\n", node, fdt_get_name(fdt, node, NULL));
|
|
print_subnode(fdt, node, depth + 1);
|
|
}
|
|
}
|
|
|
|
static int find_fstab(const void *fdt, int parent) {
|
|
int node, fstab;
|
|
fdt_for_each_subnode(node, fdt, parent) {
|
|
if (strcmp(fdt_get_name(fdt, node, NULL), "fstab") == 0)
|
|
return node;
|
|
fstab = find_fstab(fdt, node);
|
|
if (fstab != -1)
|
|
return fstab;
|
|
}
|
|
return -1;
|
|
}
|
|
|
|
void dtb_print(const char *file) {
|
|
size_t size ;
|
|
void *dtb, *fdt;
|
|
fprintf(stderr, "Loading dtbs from [%s]\n", file);
|
|
mmap_ro(file, &dtb, &size);
|
|
// Loop through all the dtbs
|
|
int dtb_num = 0;
|
|
for (int i = 0; i < size; ++i) {
|
|
if (memcmp(dtb + i, DTB_MAGIC, 4) == 0) {
|
|
fdt = dtb + i;
|
|
fprintf(stderr, "\nPrinting dtb.%04d\n\n", dtb_num++);
|
|
print_subnode(fdt, 0, 0);
|
|
}
|
|
}
|
|
fprintf(stderr, "\n");
|
|
munmap(dtb, size);
|
|
exit(0);
|
|
}
|
|
|
|
void dtb_patch(const char *file) {
|
|
size_t size ;
|
|
void *dtb, *fdt;
|
|
fprintf(stderr, "Loading dtbs from [%s]\n\n", file);
|
|
mmap_rw(file, &dtb, &size);
|
|
// Loop through all the dtbs
|
|
int dtb_num = 0, patched = 0;
|
|
for (int i = 0; i < size; ++i) {
|
|
if (memcmp(dtb + i, DTB_MAGIC, 4) == 0) {
|
|
fdt = dtb + i;
|
|
int fstab = find_fstab(fdt, 0);
|
|
if (fstab > 0) {
|
|
fprintf(stderr, "Found fstab in dtb.%04d\n", dtb_num++);
|
|
int block;
|
|
fdt_for_each_subnode(block, fdt, fstab) {
|
|
fprintf(stderr, "Found block [%s] in fstab\n", fdt_get_name(fdt, block, NULL));
|
|
int skip, value_size;
|
|
char *value = (char *) fdt_getprop(fdt, block, "fsmgr_flags", &value_size);
|
|
for (int i = 0; i < value_size; ++i) {
|
|
if ((skip = check_verity_pattern(value + i)) > 0) {
|
|
fprintf(stderr, "Remove pattern [%.*s] in [fsmgr_flags]\n", skip, value + i);
|
|
memcpy(value + i, value + i + skip, value_size - i - skip);
|
|
memset(value + value_size - skip, '\0', skip);
|
|
patched = 1;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
fprintf(stderr, "\n");
|
|
munmap(dtb, size);
|
|
exit(!patched);
|
|
}
|
|
|
|
|