+static int cbfs_create(int argc, char **argv)
+{
+ char *romname = argv[1];
+ char *cmd = argv[2];
+ if (argc < 5) {
+ printf("not enough arguments to 'create'.\n");
+ return 1;
+ }
+
+ char* suffix;
+ uint32_t size = strtoul(argv[3], &suffix, 0);
+ if (tolower(suffix[0])=='k') {
+ size *= 1024;
+ }
+ if (tolower(suffix[0])=='m') {
+ size *= 1024 * 1024;
+ }
+ char *bootblock = argv[4];
+ uint32_t align = 0;
+
+ if (argc > 5)
+ align = strtoul(argv[5], NULL, 0);
+
+ return create_cbfs_image(romname, size, bootblock, align);
+}
+
+static int cbfs_locate(int argc, char **argv)
+{
+ char *romname = argv[1];
+ if (argc < 6) {
+ printf("not enough arguments to 'locate'.\n");
+ return 1;
+ }
+
+ const char *file = argv[3];
+ uint32_t filesize = getfilesize(file);
+ const char *filename = argv[4];
+ int align = strtoul(argv[5], NULL, 0);
+
+ printf("%x\n", cbfs_find_location(romname, filesize, filename, align));
+ return 0;
+}
+
+static int cbfs_print(int argc, char **argv)
+{
+ char *romname = argv[1];
+ char *cmd = argv[2];
+ void *rom = loadrom(romname);
+
+ if (rom == NULL) {
+ printf("Could not load ROM image '%s'.\n", romname);
+ return 1;
+ }
+
+ print_cbfs_directory(romname);
+ return 0;
+}
+
+static int cbfs_extract(int argc, char **argv)
+{
+ char *romname = argv[1];
+ char *cmd = argv[2];
+ void *rom = loadrom(romname);
+
+ if (rom == NULL) {
+ printf("Could not load ROM image '%s'.\n", romname);
+ return 1;
+ }
+
+ if (argc != 5)
+ {
+ printf("Error: you must specify a CBFS name and a file to dump it in.\n");
+ return 1;
+ }
+
+ return extract_file_from_cbfs(romname, argv[3], argv[4]);
+}
+
+struct command commands[] = {
+ {CMD_ADD, "add", cbfs_add},
+ {CMD_ADD_PAYLOAD, "add-payload", cbfs_add_payload},
+ {CMD_ADD_STAGE, "add-stage", cbfs_add_stage},
+ {CMD_CREATE, "create", cbfs_create},
+ {CMD_LOCATE, "locate", cbfs_locate},
+ {CMD_PRINT, "print", cbfs_print},
+ {CMD_EXTRACT, "extract", cbfs_extract},
+};
+
+void usage(void)
+{
+ printf
+ ("cbfstool: Management utility for CBFS formatted ROM images\n\n"
+ "USAGE:\n" " cbfstool [-h]\n"
+ " cbfstool FILE COMMAND [PARAMETERS]...\n\n" "OPTIONs:\n"
+ " -h Display this help message\n\n"
+ "COMMANDs:\n"
+ " add FILE NAME TYPE [base address] Add a component\n"
+ " add-payload FILE NAME [COMP] [base] Add a payload to the ROM\n"
+ " add-stage FILE NAME [COMP] [base] Add a stage to the ROM\n"
+ " create SIZE BOOTBLOCK [ALIGN] Create a ROM file\n"
+ " locate FILE NAME ALIGN Find a place for a file of that size\n"
+ " print Show the contents of the ROM\n"
+ " extract NAME FILE Extracts a raw payload from ROM\n"
+ "\n"
+ "TYPEs:\n"
+ );
+ print_supported_filetypes();
+}
+
+int main(int argc, char **argv)
+{
+ int i;
+
+ if (argc < 3) {
+ usage();
+ return 1;
+ }
+
+ char *cmd = argv[2];
+
+ for (i = 0; i < ARRAY_SIZE(commands); i++) {
+ if (strcmp(cmd, commands[i].name) != 0)
+ continue;
+ return commands[i].function(argc, argv);
+ }