Add a "remove" command to cbfstool
[coreboot.git] / util / cbfstool / cbfstool.c
index 7fa7a8510aa9be9cbc07d518a8185a96e98cde38..9dbdc1c11aeb82be6eeb9b8ed98ce2a351be344b 100644 (file)
@@ -19,7 +19,9 @@
  */
 
 #include <stdio.h>
-#include <stdint.h>
+#include <stdlib.h>
+#include <string.h>
+#include <ctype.h>
 #include "common.h"
 #include "cbfs.h"
 
@@ -27,8 +29,11 @@ typedef enum {
        CMD_ADD,
        CMD_ADD_PAYLOAD,
        CMD_ADD_STAGE,
+       CMD_REMOVE,
        CMD_CREATE,
-       CMD_PRINT
+       CMD_LOCATE,
+       CMD_PRINT,
+       CMD_EXTRACT,
 } cmd_t;
 
 struct command {
@@ -78,10 +83,13 @@ static int cbfs_add(int argc, char **argv)
        if (argc > 6) {
                base = strtoul(argv[6], NULL, 0);
        }
-       cbfsfile =
-           create_cbfs_file(cbfsname, filedata, &filesize, type, &base);
-       add_file_to_cbfs(cbfsfile, filesize, base);
-       writerom(romname, rom, romsize);
+       cbfsfile = create_cbfs_file(cbfsname, filedata, &filesize, type, &base);
+       if (add_file_to_cbfs(cbfsfile, filesize, base)) {
+               printf("Adding file '%s' failed.\n", filename);
+               return 1;
+       }
+       if (writerom(romname, rom, romsize))
+               return 1;
        return 0;
 }
 
@@ -127,8 +135,12 @@ static int cbfs_add_payload(int argc, char **argv)
        cbfsfile =
            create_cbfs_file(cbfsname, payload, &filesize,
                             CBFS_COMPONENT_PAYLOAD, &base);
-       add_file_to_cbfs(cbfsfile, filesize, base);
-       writerom(romname, rom, romsize);
+       if (add_file_to_cbfs(cbfsfile, filesize, base)) {
+               printf("Adding payload '%s' failed.\n", filename);
+               return 1;
+       }
+       if (writerom(romname, rom, romsize))
+               return 1;
        return 0;
 }
 
@@ -175,35 +187,88 @@ static int cbfs_add_stage(int argc, char **argv)
            create_cbfs_file(cbfsname, stage, &filesize,
                             CBFS_COMPONENT_STAGE, &base);
 
-       add_file_to_cbfs(cbfsfile, filesize, base);
-       writerom(romname, rom, romsize);
+       if (add_file_to_cbfs(cbfsfile, filesize, base)) {
+               printf("Adding stage '%s' failed.\n", filename);
+               return 1;
+       }
+       if (writerom(romname, rom, romsize))
+               return 1;
        return 0;
 }
 
-static int cbfs_create(int argc, char **argv)
+static int cbfs_remove(int argc, char **argv)
 {
        char *romname = argv[1];
        char *cmd = argv[2];
-       if (argc < 6) {
+       void *rom = loadrom(romname);
+
+       if (rom == NULL) {
+               printf("Could not load ROM image '%s'.\n", romname);
+               return 1;
+       }
+
+       if (argc < 4) {
+               printf("not enough arguments to '%s'.\n", cmd);
+               return 1;
+       }
+
+       char *cbfsname = argv[3];
+
+       if (remove_file_from_cbfs(cbfsname)) {
+               printf("Removing file '%s' failed.\n", cbfsname);
+               return 1;
+       }
+       if (writerom(romname, rom, romsize))
+               return 1;
+       return 0;
+}
+
+static int cbfs_create(int argc, char **argv)
+{
+       char *romname = argv[1];
+       if (argc < 5) {
                printf("not enough arguments to 'create'.\n");
                return 1;
        }
 
-       uint32_t size = strtoul(argv[3], NULL, 0);
-       /* ignore bootblock size. we use whatever we get and won't allocate any larger */
-       char *bootblock = argv[5];
+       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 > 6)
-               align = strtoul(argv[6], NULL, 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);
+       uint32_t location = cbfs_find_location(romname, filesize, filename, align);
+
+       printf("%x\n", location);
+       return location == 0 ? 1 : 0;
+}
+
 static int cbfs_print(int argc, char **argv)
 {
        char *romname = argv[1];
-       char *cmd = argv[2];
        void *rom = loadrom(romname);
 
        if (rom == NULL) {
@@ -215,27 +280,70 @@ static int cbfs_print(int argc, char **argv)
        return 0;
 }
 
-struct command commands[] = {
+static int cbfs_extract(int argc, char **argv)
+{
+       char *romname = argv[1];
+       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]);
+}
+
+static const 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_REMOVE, "remove", cbfs_remove},
        {CMD_CREATE, "create", cbfs_create},
-       {CMD_PRINT, "print", cbfs_print}
+       {CMD_LOCATE, "locate", cbfs_locate},
+       {CMD_PRINT, "print", cbfs_print},
+       {CMD_EXTRACT, "extract", cbfs_extract},
 };
 
-void usage(void)
+static void usage(void)
 {
        printf
-           ("cbfstool: Management utility for CBFS formatted ROM images\n"
-            "USAGE:\n" "cbfstool [-h]\n"
-            "cbfstool FILE COMMAND [PARAMETERS]...\n\n" "OPTIONs:\n"
-            " -h               Display this help message\n\n"
+           ("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 BSIZE BOOTBLOCK [ALIGN]  Create a ROM file\n"
-            "print                                Show the contents of the ROM\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"
+            " remove FILE NAME                     Remove a component\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();
+}
+
+/* Small, OS/libc independent runtime check
+ * for endianess
+ */
+int host_bigendian = 0;
+
+static void which_endian(void)
+{
+       char test[4] = "1234";
+       uint32_t inttest = *(uint32_t *) test;
+       if (inttest == 0x31323334) {
+               host_bigendian = 1;
+       }
 }
 
 int main(int argc, char **argv)
@@ -247,6 +355,8 @@ int main(int argc, char **argv)
                return 1;
        }
 
+       which_endian();
+
        char *cmd = argv[2];
 
        for (i = 0; i < ARRAY_SIZE(commands); i++) {