summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--util/cbfstool/cbfstool.c30
-rw-r--r--util/cbfstool/common.c56
2 files changed, 86 insertions, 0 deletions
diff --git a/util/cbfstool/cbfstool.c b/util/cbfstool/cbfstool.c
index 939221e66d..9dbdc1c11a 100644
--- a/util/cbfstool/cbfstool.c
+++ b/util/cbfstool/cbfstool.c
@@ -29,6 +29,7 @@ typedef enum {
CMD_ADD,
CMD_ADD_PAYLOAD,
CMD_ADD_STAGE,
+ CMD_REMOVE,
CMD_CREATE,
CMD_LOCATE,
CMD_PRINT,
@@ -195,6 +196,33 @@ static int cbfs_add_stage(int argc, char **argv)
return 0;
}
+static int cbfs_remove(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 < 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];
@@ -275,6 +303,7 @@ 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_LOCATE, "locate", cbfs_locate},
{CMD_PRINT, "print", cbfs_print},
@@ -292,6 +321,7 @@ static void usage(void)
" 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"
diff --git a/util/cbfstool/common.c b/util/cbfstool/common.c
index df744217d4..49da7aea6c 100644
--- a/util/cbfstool/common.c
+++ b/util/cbfstool/common.c
@@ -350,6 +350,62 @@ int add_file_to_cbfs(void *content, uint32_t contentsize, uint32_t location)
return 1;
}
+
+static struct cbfs_file *merge_adjacent_files(struct cbfs_file *first,
+ struct cbfs_file *second)
+{
+ uint32_t new_length =
+ ntohl(first->len) + ntohl(second->len) + ntohl(second->offset);
+ first->len = htonl(new_length);
+ first->checksum = 0; // FIXME?
+ return first;
+}
+
+static struct cbfs_file *next_file(struct cbfs_file *prev)
+{
+ uint32_t pos = (prev == NULL) ? phys_start :
+ ALIGN(virt_to_phys(prev) + ntohl(prev->len) + ntohl(prev->offset),
+ align);
+
+ for (; pos < phys_end; pos += align) {
+ if (cbfs_file_header(pos))
+ return (struct cbfs_file *)phys_to_virt(pos);
+ }
+ return NULL;
+}
+
+
+int remove_file_from_cbfs(const char *filename)
+{
+ struct cbfs_file *prev = NULL;
+ struct cbfs_file *cur = next_file(prev);
+ struct cbfs_file *next = next_file(cur);
+ for (; cur; prev = cur, cur = next, next = next_file(next)) {
+
+ /* Check if this is the file to remove. */
+ char *name = (char *)cur + sizeof(*cur);
+ if (strcmp(name, filename))
+ continue;
+
+ /* Mark the file as free space and erase its name. */
+ cur->type = CBFS_COMPONENT_NULL;
+ name[0] = '\0';
+
+ /* Merge it with the previous file if possible. */
+ if (prev && prev->type == CBFS_COMPONENT_NULL)
+ cur = merge_adjacent_files(prev, cur);
+
+ /* Merge it with the next file if possible. */
+ if (next && next->type == CBFS_COMPONENT_NULL)
+ merge_adjacent_files(cur, next);
+
+ return 0;
+ }
+ printf("CBFS file %s not found.\n", filename);
+ return 1;
+}
+
+
/* returns new data block with cbfs_file header, suitable to dump into the ROM. location returns
the new location that points to the cbfs_file header */
void *create_cbfs_file(const char *filename, void *data, uint32_t * datasize,