Magisk/core/jni/magiskboot/main.c

165 lines
5.7 KiB
C
Raw Normal View History

2017-09-14 17:11:56 +02:00
#include <stdlib.h>
#include <stdio.h>
#include <string.h>
#include <unistd.h>
#include <sys/mman.h>
2017-02-27 22:37:47 +01:00
#include "magiskboot.h"
#include "utils.h"
2017-09-14 17:11:56 +02:00
#include "sha1.h"
/********************
Patch Boot Image
*********************/
2017-02-24 20:29:12 +01:00
static void usage(char *arg0) {
2017-04-27 21:15:48 +02:00
fprintf(stderr,
2017-09-14 17:11:56 +02:00
"Usage: %s <action> [args...]\n"
"\n"
"Supported actions:\n"
2017-12-06 05:51:16 +01:00
" --parse <bootimg>\n"
2017-12-06 18:30:48 +01:00
" Parse <bootimg> only, do not unpack. Return values: \n"
2017-12-06 05:51:16 +01:00
" 0:OK 1:error 2:insufficient boot partition size\n"
" 3:chromeos 4:ELF32 5:ELF64\n"
"\n"
2017-09-14 17:11:56 +02:00
" --unpack <bootimg>\n"
2017-11-10 13:25:41 +01:00
" Unpack <bootimg> to kernel, ramdisk.cpio, (second), (dtb), (extra) into\n"
2017-12-06 18:30:48 +01:00
" the current directory. Return value is the same as --parse\n"
2017-04-27 21:15:48 +02:00
"\n"
2017-09-14 17:11:56 +02:00
" --repack <origbootimg> [outbootimg]\n"
2017-04-27 21:15:48 +02:00
" Repack kernel, ramdisk.cpio[.ext], second, dtb... from current directory\n"
" to [outbootimg], or new-boot.img if not specified.\n"
2017-12-06 05:51:16 +01:00
" It will compress ramdisk.cpio with the same method used in <origbootimg>,\n"
" or attempt to find ramdisk.cpio.[ext], and repack directly with the\n"
" compressed ramdisk file\n"
2017-04-27 21:15:48 +02:00
"\n"
2017-09-14 17:11:56 +02:00
" --hexpatch <file> <hexpattern1> <hexpattern2>\n"
2017-04-27 21:15:48 +02:00
" Search <hexpattern1> in <file>, and replace with <hexpattern2>\n"
"\n"
2017-09-14 17:11:56 +02:00
" --cpio-<cmd> <incpio> [flags...] [args...]\n"
" Do cpio related cmds to <incpio> (modifications are done directly)\n"
" Supported commands:\n"
" -rm [-r] <entry>\n"
2017-12-03 20:22:26 +01:00
" Remove entry from <incpio>, flag [-r] to remove recursively\n"
2017-09-14 17:11:56 +02:00
" -mkdir <mode> <entry>\n"
" Create directory as an <entry>\n"
2017-12-03 20:22:26 +01:00
" -ln <target> <entry>\n"
" Create symlink <entry> to point to <target>\n"
2017-09-14 17:11:56 +02:00
" -mv <from-entry> <to-entry>\n"
" Move <from-entry> to <to-entry>\n"
2017-12-06 05:51:16 +01:00
" -add <mode> <entry> <infile>\n"
" Add <infile> as an <entry>; replaces <entry> if already exists\n"
2017-12-04 20:32:37 +01:00
" -extract [<entry> <outfile>]\n"
" Extract <entry> to <outfile>, or extract all to current directory\n"
2017-11-09 20:45:06 +01:00
" -test\n"
" Return value: 0/stock 1/Magisk 2/other (phh, SuperSU, Xposed)\n"
2017-09-14 17:11:56 +02:00
" -patch <KEEPVERITY> <KEEPFORCEENCRYPT>\n"
2017-12-06 18:30:48 +01:00
" Ramdisk patches. KEEP**** are boolean values\n"
" This command is no longer used in Magisk installations\n"
" -backup <origcpio> <HIGHCOMP> <KEEPVERITY> <KEEPFORCEENCRYPT> [SHA1]\n"
2017-09-14 17:11:56 +02:00
" Create ramdisk backups into <incpio> from <origcpio>\n"
2017-12-06 18:30:48 +01:00
" HIGHCOMP, KEEP**** are boolean values\n"
" SHA1 of stock boot image is optional\n"
2017-09-14 17:11:56 +02:00
" -restore\n"
" Restore ramdisk from ramdisk backup within <incpio>\n"
" -stocksha1\n"
" Get stock boot SHA1 recorded within <incpio>\n"
2017-04-27 21:15:48 +02:00
"\n"
2017-11-09 20:45:06 +01:00
" --dtb-<cmd> <dtb>\n"
" Do dtb related cmds to <dtb> (modifications are done directly)\n"
" Supported commands:\n"
" -dump\n"
" Dump all contents from dtb for debugging\n"
2017-12-06 05:51:16 +01:00
" -test\n"
" Check if fstab has verity/avb flags\n"
" Return value: 0/no flags 1/flag exists"
2017-11-09 20:45:06 +01:00
" -patch\n"
" Search for fstab and remove verity/avb\n"
2017-09-14 20:52:27 +02:00
"\n"
2017-09-14 17:11:56 +02:00
" --compress[=method] <infile> [outfile]\n"
" Compress <infile> with [method] (default: gzip), optionally to [outfile]\n"
2017-11-10 13:25:41 +01:00
" <infile>/[outfile] can be '-' to be STDIN/STDOUT\n"
2017-09-14 17:11:56 +02:00
" Supported methods: "
, arg0);
2017-04-27 21:15:48 +02:00
for (int i = 0; SUP_LIST[i]; ++i)
fprintf(stderr, "%s ", SUP_LIST[i]);
fprintf(stderr,
2017-12-06 05:51:16 +01:00
"\n\n"
2017-09-14 17:11:56 +02:00
" --decompress <infile> [outfile]\n"
2017-11-10 13:25:41 +01:00
" Detect method and decompress <infile>, optionally to [outfile]\n"
" <infile>/[outfile] can be '-' to be STDIN/STDOUT\n"
" Supported methods: ");
2017-04-27 21:15:48 +02:00
for (int i = 0; SUP_LIST[i]; ++i)
fprintf(stderr, "%s ", SUP_LIST[i]);
fprintf(stderr,
2017-12-06 05:51:16 +01:00
"\n\n"
2017-09-14 17:11:56 +02:00
" --sha1 <file>\n"
2017-04-27 21:15:48 +02:00
" Print the SHA1 checksum for <file>\n"
"\n"
2017-09-14 17:11:56 +02:00
" --cleanup\n"
2017-04-27 21:15:48 +02:00
" Cleanup the current working directory\n"
2017-09-14 17:11:56 +02:00
"\n");
2017-03-07 17:54:23 +01:00
2017-02-24 20:29:12 +01:00
exit(1);
}
2017-02-24 07:58:44 +01:00
int main(int argc, char *argv[]) {
2017-07-18 05:53:28 +02:00
fprintf(stderr, "MagiskBoot v" xstr(MAGISK_VERSION) "(" xstr(MAGISK_VER_CODE) ") (by topjohnwu) - Boot Image Modification Tool\n\n");
2017-03-07 17:54:23 +01:00
2017-12-03 20:14:04 +01:00
umask(0);
2017-03-07 17:54:23 +01:00
if (argc > 1 && strcmp(argv[1], "--cleanup") == 0) {
2017-09-14 17:11:56 +02:00
fprintf(stderr, "Cleaning up...\n\n");
char name[PATH_MAX];
unlink(KERNEL_FILE);
unlink(RAMDISK_FILE);
unlink(RAMDISK_FILE ".raw");
unlink(SECOND_FILE);
unlink(DTB_FILE);
2017-10-07 16:08:10 +02:00
unlink(EXTRA_FILE);
2017-09-14 17:11:56 +02:00
for (int i = 0; SUP_EXT_LIST[i]; ++i) {
sprintf(name, "%s.%s", RAMDISK_FILE, SUP_EXT_LIST[i]);
unlink(name);
}
2017-03-09 21:08:17 +01:00
} else if (argc > 2 && strcmp(argv[1], "--sha1") == 0) {
char sha1[21], *buf;
size_t size;
2017-09-12 09:27:28 +02:00
mmap_ro(argv[2], (void **) &buf, &size);
2017-03-09 21:08:17 +01:00
SHA1(sha1, buf, size);
for (int i = 0; i < 20; ++i)
2017-07-24 20:02:19 +02:00
printf("%02x", sha1[i]);
printf("\n");
2017-03-09 21:08:17 +01:00
munmap(buf, size);
2017-12-06 05:51:16 +01:00
} else if (argc > 2 && strcmp(argv[1], "--parse") == 0) {
boot_img boot;
exit(parse_img(argv[2], &boot));
2017-03-07 17:54:23 +01:00
} else if (argc > 2 && strcmp(argv[1], "--unpack") == 0) {
unpack(argv[2]);
} else if (argc > 2 && strcmp(argv[1], "--repack") == 0) {
repack(argv[2], argc > 3 ? argv[3] : NEW_BOOT);
} else if (argc > 2 && strcmp(argv[1], "--decompress") == 0) {
decomp_file(argv[2], argc > 3 ? argv[3] : NULL);
2017-03-07 17:54:23 +01:00
} else if (argc > 2 && strncmp(argv[1], "--compress", 10) == 0) {
char *method;
method = strchr(argv[1], '=');
if (method == NULL) method = "gzip";
else method++;
comp_file(method, argv[2], argc > 3 ? argv[3] : NULL);
2017-03-07 17:54:23 +01:00
} else if (argc > 4 && strcmp(argv[1], "--hexpatch") == 0) {
hexpatch(argv[2], argv[3], argv[4]);
2017-03-09 21:08:17 +01:00
} else if (argc > 2 && strncmp(argv[1], "--cpio", 6) == 0) {
2017-11-10 18:30:33 +01:00
char *cmd = argv[1] + 6;
if (*cmd == '\0') usage(argv[0]);
else ++cmd;
if (cpio_commands(cmd, argc - 2, argv + 2)) usage(argv[0]);
} else if (argc > 2 && strncmp(argv[1], "--dtb", 5) == 0) {
char *cmd = argv[1] + 5;
if (*cmd == '\0') usage(argv[0]);
else ++cmd;
if (dtb_commands(cmd, argc - 2, argv + 2)) usage(argv[0]);
2017-03-07 17:54:23 +01:00
} else {
usage(argv[0]);
}
2017-02-24 20:29:12 +01:00
2017-03-07 17:54:23 +01:00
return 0;
2017-02-24 07:58:44 +01:00
}