mirror of
https://github.com/topjohnwu/Magisk.git
synced 2024-12-25 13:27:41 +00:00
Update ramdisk restore implementation
This commit is contained in:
parent
89ff3c6572
commit
3a39dd4049
@ -53,7 +53,9 @@ void cpio::dump(const char *file) {
|
|||||||
dump(xfopen(file, "we"));
|
dump(xfopen(file, "we"));
|
||||||
}
|
}
|
||||||
|
|
||||||
void cpio::rm(entry_map::iterator &it) {
|
void cpio::rm(entry_map::iterator it) {
|
||||||
|
if (it == entries.end())
|
||||||
|
return;
|
||||||
fprintf(stderr, "Remove [%s]\n", it->first.data());
|
fprintf(stderr, "Remove [%s]\n", it->first.data());
|
||||||
entries.erase(it);
|
entries.erase(it);
|
||||||
}
|
}
|
||||||
@ -188,7 +190,7 @@ void cpio::ln(const char *target, const char *name) {
|
|||||||
fprintf(stderr, "Create symlink [%s] -> [%s]\n", name, target);
|
fprintf(stderr, "Create symlink [%s] -> [%s]\n", name, target);
|
||||||
}
|
}
|
||||||
|
|
||||||
void cpio::mv(entry_map::iterator &it, const char *name) {
|
void cpio::mv(entry_map::iterator it, const char *name) {
|
||||||
fprintf(stderr, "Move [%s] -> [%s]\n", it->first.data(), name);
|
fprintf(stderr, "Move [%s] -> [%s]\n", it->first.data(), name);
|
||||||
auto e = it->second.release();
|
auto e = it->second.release();
|
||||||
entries.erase(it);
|
entries.erase(it);
|
||||||
|
@ -45,8 +45,8 @@ protected:
|
|||||||
entry_map entries;
|
entry_map entries;
|
||||||
|
|
||||||
static void extract_entry(const entry_map::value_type &e, const char *file);
|
static void extract_entry(const entry_map::value_type &e, const char *file);
|
||||||
void rm(entry_map::iterator &it);
|
void rm(entry_map::iterator it);
|
||||||
void mv(entry_map::iterator &it, const char *name);
|
void mv(entry_map::iterator it, const char *name);
|
||||||
|
|
||||||
private:
|
private:
|
||||||
void dump(FILE *out);
|
void dump(FILE *out);
|
||||||
|
@ -82,11 +82,10 @@ int magisk_cpio::test() {
|
|||||||
}
|
}
|
||||||
|
|
||||||
#define for_each_line(line, buf, size) \
|
#define for_each_line(line, buf, size) \
|
||||||
for (line = (char *) buf; line < (char *) buf + size && line[0]; line = strchr(line + 1, '\n') + 1)
|
for (char *line = (char *) buf; line < (char *) buf + size && line[0]; line = strchr(line + 1, '\n') + 1)
|
||||||
|
|
||||||
char *magisk_cpio::sha1() {
|
char *magisk_cpio::sha1() {
|
||||||
char sha1[41];
|
char sha1[41];
|
||||||
char *line;
|
|
||||||
for (auto &e : entries) {
|
for (auto &e : entries) {
|
||||||
if (e.first == "init.magisk.rc" || e.first == "overlay/init.magisk.rc") {
|
if (e.first == "init.magisk.rc" || e.first == "overlay/init.magisk.rc") {
|
||||||
for_each_line(line, e.second->data, e.second->filesize) {
|
for_each_line(line, e.second->data, e.second->filesize) {
|
||||||
@ -112,40 +111,48 @@ char *magisk_cpio::sha1() {
|
|||||||
}
|
}
|
||||||
|
|
||||||
#define for_each_str(str, buf, size) \
|
#define for_each_str(str, buf, size) \
|
||||||
for (str = (char *) buf; str < (char *) buf + size; str = str += strlen(str) + 1)
|
for (char *str = (char *) buf; str < (char *) buf + size; str += strlen(str) + 1)
|
||||||
|
|
||||||
void magisk_cpio::restore() {
|
void magisk_cpio::restore() {
|
||||||
// If the backup init is missing, this means that the boot ramdisk
|
// Collect files
|
||||||
// was created from scratch by us. Nothing needs to be restored.
|
auto bk = entries.end();
|
||||||
if (exists(".backup") && !exists(".backup/init")) {
|
auto rl = entries.end();
|
||||||
|
auto mg = entries.end();
|
||||||
|
vector<entry_map::iterator> backups;
|
||||||
|
for (auto it = entries.begin(); it != entries.end(); ++it) {
|
||||||
|
if (it->first == ".backup") {
|
||||||
|
bk = it;
|
||||||
|
} else if (it->first == ".backup/.rmlist") {
|
||||||
|
rl = it;
|
||||||
|
} else if (it->first == ".backup/.magisk") {
|
||||||
|
mg = it;
|
||||||
|
} else if (str_starts(it->first, ".backup/")) {
|
||||||
|
backups.emplace_back(it);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// If the .backup folder is effectively empty, this means that the boot ramdisk was
|
||||||
|
// created from scratch by an old broken magiskboot. This is just a hacky workaround.
|
||||||
|
if (bk != entries.end() && mg != entries.end() && rl == entries.end() && backups.empty()) {
|
||||||
fprintf(stderr, "Remove all in ramdisk\n");
|
fprintf(stderr, "Remove all in ramdisk\n");
|
||||||
entries.clear();
|
entries.clear();
|
||||||
return;
|
return;
|
||||||
}
|
}
|
||||||
|
|
||||||
if (auto it = entries.find(".backup/.rmlist"); it != entries.end()) {
|
// Remove files
|
||||||
char *file;
|
rm(bk);
|
||||||
for_each_str(file, it->second->data, it->second->filesize) {
|
rm(mg);
|
||||||
|
if (rl != entries.end()) {
|
||||||
|
for_each_str(file, rl->second->data, rl->second->filesize) {
|
||||||
rm(file);
|
rm(file);
|
||||||
}
|
}
|
||||||
rm(it);
|
rm(rl);
|
||||||
}
|
}
|
||||||
|
|
||||||
for (auto it = entries.begin(); it != entries.end();) {
|
// Restore files
|
||||||
auto cur = it++;
|
for (auto it : backups) {
|
||||||
if (str_starts(cur->first, ".backup")) {
|
const char *name = &it->first[8];
|
||||||
if (cur->first.length() == 7 || &cur->first[8] == ".magisk"sv) {
|
mv(it, name);
|
||||||
rm(cur);
|
|
||||||
} else {
|
|
||||||
mv(cur, &cur->first[8]);
|
|
||||||
}
|
|
||||||
} else if (str_starts(cur->first, "magisk") ||
|
|
||||||
cur->first == "overlay/init.magisk.rc" ||
|
|
||||||
cur->first == "sbin/magic_mask.sh" ||
|
|
||||||
cur->first == "init.magisk.rc") {
|
|
||||||
// Some known stuff we can remove
|
|
||||||
rm(cur);
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
Loading…
x
Reference in New Issue
Block a user