| /* |
| * common utility functions for cbfstool |
| * |
| * Copyright (C) 2009 coresystems GmbH |
| * written by Patrick Georgi <patrick.georgi@coresystems.de> |
| * Copyright (C) 2012 Google, Inc. |
| * |
| * This program is free software; you can redistribute it and/or modify |
| * it under the terms of the GNU General Public License as published by |
| * the Free Software Foundation; version 2 of the License. |
| * |
| * This program is distributed in the hope that it will be useful, |
| * but WITHOUT ANY WARRANTY; without even the implied warranty of |
| * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the |
| * GNU General Public License for more details. |
| * |
| * You should have received a copy of the GNU General Public License |
| * along with this program; if not, write to the Free Software |
| * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA, 02110-1301 USA |
| */ |
| |
| #include <stdio.h> |
| #include <stdlib.h> |
| #include <string.h> |
| #include <libgen.h> |
| #include "common.h" |
| #include "cbfs.h" |
| #include "elf.h" |
| |
| size_t getfilesize(const char *filename) |
| { |
| size_t size; |
| FILE *file = fopen(filename, "rb"); |
| if (file == NULL) |
| return -1; |
| |
| fseek(file, 0, SEEK_END); |
| size = ftell(file); |
| fclose(file); |
| return size; |
| } |
| |
| void *loadfile(const char *filename, uint32_t * romsize_p, void *content, |
| int place) |
| { |
| FILE *file = fopen(filename, "rb"); |
| if (file == NULL) |
| return NULL; |
| |
| fseek(file, 0, SEEK_END); |
| *romsize_p = ftell(file); |
| fseek(file, 0, SEEK_SET); |
| if (!content) { |
| content = malloc(*romsize_p); |
| if (!content) { |
| ERROR("Could not get %d bytes for file %s\n", |
| *romsize_p, filename); |
| exit(1); |
| } |
| } else if (place == SEEK_END) |
| content -= *romsize_p; |
| |
| if (!fread(content, *romsize_p, 1, file)) { |
| ERROR("Failed to read %s\n", filename); |
| return NULL; |
| } |
| fclose(file); |
| return content; |
| } |
| |
| static struct cbfs_header *master_header; |
| static uint32_t phys_start, phys_end, align; |
| uint32_t romsize; |
| void *offset; |
| uint32_t arch = CBFS_ARCHITECTURE_UNKNOWN; |
| |
| static struct { |
| uint32_t arch; |
| const char *name; |
| } arch_names[] = { |
| { CBFS_ARCHITECTURE_ARMV7, "armv7" }, |
| { CBFS_ARCHITECTURE_X86, "x86" }, |
| { CBFS_ARCHITECTURE_UNKNOWN, "unknown" } |
| }; |
| |
| uint32_t string_to_arch(const char *arch_string) |
| { |
| int i; |
| uint32_t ret = CBFS_ARCHITECTURE_UNKNOWN; |
| |
| for (i = 0; i < ARRAY_SIZE(arch_names); i++) { |
| if (!strcasecmp(arch_string, arch_names[i].name)) { |
| ret = arch_names[i].arch; |
| break; |
| } |
| } |
| |
| return ret; |
| } |
| |
| const char *arch_to_string(uint32_t a) |
| { |
| int i; |
| const char *ret = NULL; |
| |
| for (i = 0; i < ARRAY_SIZE(arch_names); i++) { |
| if (a == arch_names[i].arch) { |
| ret = arch_names[i].name; |
| break; |
| } |
| } |
| |
| return ret; |
| |
| } |
| |
| int find_master_header(void *romarea, size_t size) |
| { |
| size_t offset; |
| |
| if (master_header) |
| return 0; |
| |
| for (offset = 0; offset < size - sizeof(struct cbfs_header); offset++) { |
| struct cbfs_header *tmp = romarea + offset; |
| |
| if (tmp->magic == ntohl(CBFS_HEADER_MAGIC)) { |
| master_header = tmp; |
| break; |
| } |
| } |
| |
| return master_header ? 0 : 1; |
| } |
| |
| void recalculate_rom_geometry(void *romarea) |
| { |
| if (find_master_header(romarea, romsize)) { |
| ERROR("Cannot find master header\n"); |
| exit(1); |
| } |
| |
| /* Update old headers */ |
| if (master_header->version == CBFS_HEADER_VERSION1 && |
| ntohl(master_header->architecture) == CBFS_ARCHITECTURE_UNKNOWN) { |
| DEBUG("Updating CBFS master header to version 2\n"); |
| master_header->architecture = htonl(CBFS_ARCHITECTURE_X86); |
| } |
| |
| arch = ntohl(master_header->architecture); |
| |
| switch (arch) { |
| case CBFS_ARCHITECTURE_ARMV7: |
| offset = romarea; |
| phys_start = (0 + ntohl(master_header->offset)) & 0xffffffff; |
| phys_end = romsize & 0xffffffff; |
| break; |
| case CBFS_ARCHITECTURE_X86: |
| offset = romarea + romsize - 0x100000000ULL; |
| phys_start = (0 - romsize + ntohl(master_header->offset)) & |
| 0xffffffff; |
| phys_end = (0 - ntohl(master_header->bootblocksize) - |
| sizeof(struct cbfs_header)) & 0xffffffff; |
| break; |
| default: |
| ERROR("Unknown architecture\n"); |
| exit(1); |
| } |
| |
| align = ntohl(master_header->align); |
| } |
| |
| void *loadrom(const char *filename) |
| { |
| void *romarea = loadfile(filename, &romsize, 0, SEEK_SET); |
| if (romarea == NULL) |
| return NULL; |
| recalculate_rom_geometry(romarea); |
| return romarea; |
| } |
| |
| int writerom(const char *filename, void *start, uint32_t size) |
| { |
| FILE *file = fopen(filename, "wb"); |
| if (!file) { |
| ERROR("Could not open '%s' for writing: ", filename); |
| perror(""); |
| return 1; |
| } |
| |
| if (fwrite(start, size, 1, file) != 1) { |
| ERROR("Could not write to '%s': ", filename); |
| perror(""); |
| return 1; |
| } |
| |
| fclose(file); |
| return 0; |
| } |
| |
| int cbfs_file_header(unsigned long physaddr) |
| { |
| /* maybe improve this test */ |
| return (strncmp(phys_to_virt(physaddr), "LARCHIVE", 8) == 0); |
| } |
| |
| struct cbfs_file *cbfs_create_empty_file(uint32_t physaddr, uint32_t size) |
| { |
| struct cbfs_file *nextfile = (struct cbfs_file *)phys_to_virt(physaddr); |
| strncpy((char *)(nextfile->magic), "LARCHIVE", 8); |
| nextfile->len = htonl(size); |
| nextfile->type = htonl(0xffffffff); |
| nextfile->checksum = 0; // FIXME? |
| nextfile->offset = htonl(sizeof(struct cbfs_file) + 16); |
| memset(((void *)nextfile) + sizeof(struct cbfs_file), 0, 16); |
| return nextfile; |
| } |
| |
| int iself(unsigned char *input) |
| { |
| Elf32_Ehdr *ehdr = (Elf32_Ehdr *) input; |
| return !memcmp(ehdr->e_ident, ELFMAG, 4); |
| } |
| |
| static struct filetypes_t { |
| uint32_t type; |
| const char *name; |
| } filetypes[] = { |
| {CBFS_COMPONENT_STAGE, "stage"}, |
| {CBFS_COMPONENT_PAYLOAD, "payload"}, |
| {CBFS_COMPONENT_OPTIONROM, "optionrom"}, |
| {CBFS_COMPONENT_BOOTSPLASH, "bootsplash"}, |
| {CBFS_COMPONENT_RAW, "raw"}, |
| {CBFS_COMPONENT_VSA, "vsa"}, |
| {CBFS_COMPONENT_MBI, "mbi"}, |
| {CBFS_COMPONENT_MICROCODE, "microcode"}, |
| {CBFS_COMPONENT_CMOS_DEFAULT, "cmos default"}, |
| {CBFS_COMPONENT_CMOS_LAYOUT, "cmos layout"}, |
| {CBFS_COMPONENT_DELETED, "deleted"}, |
| {CBFS_COMPONENT_NULL, "null"} |
| }; |
| |
| void print_supported_filetypes(void) |
| { |
| int i, number = ARRAY_SIZE(filetypes); |
| |
| for (i=0; i<number; i++) { |
| LOG(" %s%c", filetypes[i].name, (i==(number-1))?'\n':','); |
| if ((i%8) == 7) |
| LOG("\n"); |
| } |
| } |
| |
| const char *strfiletype(uint32_t number) |
| { |
| size_t i; |
| for (i = 0; i < (sizeof(filetypes) / sizeof(struct filetypes_t)); i++) |
| if (filetypes[i].type == number) |
| return filetypes[i].name; |
| return "unknown"; |
| } |
| |
| uint64_t intfiletype(const char *name) |
| { |
| size_t i; |
| for (i = 0; i < (sizeof(filetypes) / sizeof(struct filetypes_t)); i++) |
| if (strcmp(filetypes[i].name, name) == 0) |
| return filetypes[i].type; |
| return -1; |
| } |
| |
| void print_cbfs_directory(const char *filename) |
| { |
| char *name = strdup(filename); |
| printf |
| ("%s: %d kB, bootblocksize %d, romsize %d, offset 0x%x\n" |
| "alignment: %d bytes, architecture: %s\n\n", |
| basename(name), romsize / 1024, ntohl(master_header->bootblocksize), |
| romsize, ntohl(master_header->offset), align, arch_to_string(arch)); |
| free(name); |
| printf("%-30s %-10s %-12s Size\n", "Name", "Offset", "Type"); |
| uint32_t current = phys_start; |
| while (current < phys_end) { |
| if (!cbfs_file_header(current)) { |
| current += align; |
| continue; |
| } |
| struct cbfs_file *thisfile = |
| (struct cbfs_file *)phys_to_virt(current); |
| uint32_t length = ntohl(thisfile->len); |
| char *fname = (char *)(phys_to_virt(current) + sizeof(struct cbfs_file)); |
| if (strlen(fname) == 0) |
| fname = "(empty)"; |
| |
| printf("%-30s 0x%-8x %-12s %d\n", fname, |
| current - phys_start + ntohl(master_header->offset), |
| strfiletype(ntohl(thisfile->type)), length); |
| |
| /* note the components of the subheader are in host order ... */ |
| switch (ntohl(thisfile->type)) { |
| case CBFS_COMPONENT_STAGE: |
| { |
| struct cbfs_stage *stage = CBFS_SUBHEADER(thisfile); |
| INFO(" %s compression, entry: 0x%llx, load: 0x%llx, length: %d/%d\n", |
| stage->compression == CBFS_COMPRESS_LZMA ? "LZMA" : "no", |
| (unsigned long long)stage->entry, |
| (unsigned long long)stage->load, |
| stage->len, |
| stage->memlen); |
| break; |
| } |
| case CBFS_COMPONENT_PAYLOAD: |
| { |
| struct cbfs_payload_segment *payload = CBFS_SUBHEADER(thisfile); |
| while(payload) { |
| switch(payload->type) { |
| case PAYLOAD_SEGMENT_CODE: |
| case PAYLOAD_SEGMENT_DATA: |
| INFO(" %s (%s compression, offset: 0x%x, load: 0x%llx, length: %d/%d)\n", |
| payload->type == PAYLOAD_SEGMENT_CODE ? "code " : "data" , |
| payload->compression == CBFS_COMPRESS_LZMA ? "LZMA" : "no", |
| ntohl(payload->offset), |
| (unsigned long long)ntohll(payload->load_addr), |
| ntohl(payload->len), ntohl(payload->mem_len)); |
| break; |
| case PAYLOAD_SEGMENT_ENTRY: |
| INFO(" entry (0x%llx)\n", (unsigned long long)ntohll(payload->load_addr)); |
| break; |
| case PAYLOAD_SEGMENT_BSS: |
| INFO(" BSS (address 0x%016llx, length 0x%x)\n", (unsigned long long)ntohll(payload->load_addr), ntohl(payload->len)); |
| break; |
| case PAYLOAD_SEGMENT_PARAMS: |
| INFO(" parameters\n"); |
| break; |
| default: |
| INFO(" %x (%s compression, offset: 0x%x, load: 0x%llx, length: %d/%d\n", |
| payload->type, |
| payload->compression == CBFS_COMPRESS_LZMA ? "LZMA" : "no", |
| ntohl(payload->offset), |
| (unsigned long long)ntohll(payload->load_addr), |
| ntohl(payload->len), |
| ntohl(payload->mem_len)); |
| break; |
| } |
| |
| if(payload->type == PAYLOAD_SEGMENT_ENTRY) |
| payload=NULL; |
| else |
| payload++; |
| } |
| break; |
| } |
| default: |
| break; |
| } |
| current = |
| ALIGN(current + ntohl(thisfile->len) + |
| ntohl(thisfile->offset), align); |
| } |
| } |
| |
| int extract_file_from_cbfs(const char *filename, const char *payloadname, const char *outpath) |
| { |
| FILE *outfile = NULL; |
| uint32_t current = phys_start; |
| while (current < phys_end) { |
| if (!cbfs_file_header(current)) { |
| current += align; |
| continue; |
| } |
| |
| // Locate the file start struct |
| struct cbfs_file *thisfile = |
| (struct cbfs_file *)phys_to_virt(current); |
| // And its length |
| uint32_t length = ntohl(thisfile->len); |
| // Locate the file name |
| char *fname = (char *)(phys_to_virt(current) + sizeof(struct cbfs_file)); |
| |
| // It's not the file we are looking for.. |
| if (strcmp(fname, payloadname) != 0) |
| { |
| current = |
| ALIGN(current + ntohl(thisfile->len) + |
| ntohl(thisfile->offset), align); |
| continue; |
| } |
| |
| // Else, it's our file. |
| LOG("Found file %.30s at 0x%x, type %.12s, size %d\n", fname, |
| current - phys_start, strfiletype(ntohl(thisfile->type)), |
| length); |
| |
| // If we are not dumping to stdout, open the out file. |
| outfile = fopen(outpath, "wb"); |
| if (!outfile) |
| { |
| ERROR("Could not open the file %s for writing.\n", outpath); |
| return 1; |
| } |
| |
| if (ntohl(thisfile->type) != CBFS_COMPONENT_RAW) |
| { |
| WARN("Only 'raw' files are safe to extract.\n"); |
| } |
| |
| fwrite(((char *)thisfile) |
| + ntohl(thisfile->offset), length, 1, outfile); |
| |
| fclose(outfile); |
| LOG("Successfully dumped the file.\n"); |
| |
| // We'll only dump one file. |
| return 0; |
| } |
| ERROR("File %s not found.\n", payloadname); |
| return 1; |
| } |
| |
| |
| int add_file_to_cbfs(void *content, uint32_t contentsize, uint32_t location) |
| { |
| uint32_t current = phys_start; |
| while (current < phys_end) { |
| if (!cbfs_file_header(current)) { |
| current += align; |
| continue; |
| } |
| struct cbfs_file *thisfile = |
| (struct cbfs_file *)phys_to_virt(current); |
| uint32_t length = ntohl(thisfile->len); |
| |
| DEBUG("at %x, %x bytes\n", current, length); |
| /* Is this a free chunk? */ |
| if ((thisfile->type == CBFS_COMPONENT_DELETED) |
| || (thisfile->type == CBFS_COMPONENT_NULL)) { |
| DEBUG("null||deleted at %x, %x bytes\n", current, |
| length); |
| /* if this is the right size, and if specified, the right location, use it */ |
| if ((contentsize <= length) |
| && ((location == 0) || (current == location))) { |
| if (contentsize < length) { |
| DEBUG("this chunk is %x bytes, we need %x. create a new chunk at %x with %x bytes\n", |
| length, contentsize, |
| ALIGN(current + contentsize, |
| align), |
| length - contentsize); |
| uint32_t start = |
| ALIGN(current + contentsize, align); |
| uint32_t size = |
| current + ntohl(thisfile->offset) |
| + length - start - 16 - |
| sizeof(struct cbfs_file); |
| cbfs_create_empty_file(start, size); |
| } |
| DEBUG("copying data\n"); |
| memcpy(phys_to_virt(current), content, |
| contentsize); |
| return 0; |
| } |
| if (location != 0) { |
| /* CBFS has the constraint that the chain always moves up in memory. so once |
| we're past the place we seek, we don't need to look any further */ |
| if (current > location) { |
| ERROR("The requested space is not available\n"); |
| return 1; |
| } |
| |
| /* Is the requested location inside the current chunk? */ |
| if ((current < location) |
| && ((location + contentsize) <= |
| (current + length))) { |
| /* Split it up. In the next iteration the code will be at the right place. */ |
| DEBUG("split up. new length: %x\n", |
| location - current - |
| ntohl(thisfile->offset)); |
| thisfile->len = |
| htonl(location - current - |
| ntohl(thisfile->offset)); |
| cbfs_create_empty_file(location, |
| length - |
| (location - |
| current)); |
| } |
| } |
| } |
| current = |
| ALIGN(current + ntohl(thisfile->len) + |
| ntohl(thisfile->offset), align); |
| } |
| ERROR("Could not add the file to CBFS, it's probably too big.\n"); |
| ERROR("File size: %d bytes (%d KB).\n", contentsize, contentsize/1024); |
| 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; |
| } |
| ERROR("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, |
| uint32_t type, uint32_t * location) |
| { |
| uint32_t filename_len = ALIGN(strlen(filename) + 1, 16); |
| uint32_t headersize = sizeof(struct cbfs_file) + filename_len; |
| if ((location != 0) && (*location != 0)) { |
| uint32_t offset = *location % align; |
| /* If offset >= (headersize % align), we can stuff the header into the offset. |
| Otherwise the header has to be aligned itself, and put before the offset data */ |
| if (offset >= (headersize % align)) { |
| offset -= (headersize % align); |
| } else { |
| offset += align - (headersize % align); |
| } |
| headersize += offset; |
| *location -= headersize; |
| } |
| void *newdata = malloc(*datasize + headersize); |
| if (!newdata) { |
| ERROR("Could not get %d bytes for CBFS file.\n", *datasize + |
| headersize); |
| exit(1); |
| } |
| memset(newdata, 0xff, *datasize + headersize); |
| struct cbfs_file *nextfile = (struct cbfs_file *)newdata; |
| strncpy((char *)(nextfile->magic), "LARCHIVE", 8); |
| nextfile->len = htonl(*datasize); |
| nextfile->type = htonl(type); |
| nextfile->checksum = 0; // FIXME? |
| nextfile->offset = htonl(headersize); |
| strcpy(newdata + sizeof(struct cbfs_file), filename); |
| memcpy(newdata + headersize, data, *datasize); |
| *datasize += headersize; |
| return newdata; |
| } |
| |
| int create_cbfs_image(const char *romfile, uint32_t _romsize, |
| const char *bootblock, uint32_t align, uint32_t offs) |
| { |
| uint32_t bootblocksize = 0; |
| struct cbfs_header *master_header; |
| unsigned char *romarea, *bootblk; |
| |
| romsize = _romsize; |
| romarea = malloc(romsize); |
| if (!romarea) { |
| ERROR("Could not get %d bytes of memory" |
| " for CBFS image.\n", romsize); |
| exit(1); |
| } |
| memset(romarea, 0xff, romsize); |
| |
| if (align == 0) |
| align = 64; |
| |
| bootblk = loadfile(bootblock, &bootblocksize, |
| romarea + romsize, SEEK_END); |
| if (!bootblk) { |
| ERROR("Could not load bootblock %s.\n", |
| bootblock); |
| free(romarea); |
| return 1; |
| } |
| |
| // TODO(hungte) Replace magic numbers by named constants. |
| switch (arch) { |
| case CBFS_ARCHITECTURE_ARMV7: |
| /* Set up physical/virtual mapping */ |
| offset = romarea; |
| |
| /* |
| * The initial jump instruction and bootblock will be placed |
| * before and after the master header, respectively. The |
| * bootblock image must contain a blank, aligned region large |
| * enough for the master header to fit. |
| * |
| * An anchor string must be left such that when cbfstool is run |
| * we can find it and insert the master header at the next |
| * aligned boundary. |
| */ |
| loadfile(bootblock, &bootblocksize, romarea + offs, SEEK_SET); |
| |
| unsigned char *p = romarea + offs; |
| while (1) { |
| /* FIXME: assumes little endian... */ |
| if (*(uint32_t *)p == 0xdeadbeef) |
| break; |
| if (p >= (romarea + _romsize)) { |
| ERROR("Could not determine CBFS " |
| "header location.\n"); |
| return 1; |
| } |
| p += (sizeof(unsigned int)); |
| } |
| unsigned int u = ALIGN((unsigned int)(p - romarea), align); |
| master_header = (struct cbfs_header *)(romarea + u); |
| |
| master_header->magic = ntohl(CBFS_HEADER_MAGIC); |
| master_header->version = ntohl(CBFS_HEADER_VERSION); |
| master_header->romsize = htonl(romsize); |
| master_header->bootblocksize = htonl(bootblocksize); |
| master_header->align = htonl(align); |
| master_header->offset = htonl( |
| ALIGN((0x40 + bootblocksize), align)); |
| master_header->architecture = htonl(CBFS_ARCHITECTURE_ARMV7); |
| |
| ((uint32_t *) phys_to_virt(0x4 + offs))[0] = |
| virt_to_phys(master_header); |
| |
| recalculate_rom_geometry(romarea); |
| |
| cbfs_create_empty_file( |
| offs + ALIGN((0x40 + bootblocksize), align), |
| romsize - offs - sizeof(struct cbfs_file) - |
| ALIGN((bootblocksize + 0x40), align)); |
| break; |
| |
| case CBFS_ARCHITECTURE_X86: |
| // Set up physical/virtual mapping |
| offset = romarea + romsize - 0x100000000ULL; |
| |
| loadfile(bootblock, &bootblocksize, romarea + romsize, |
| SEEK_END); |
| master_header = (struct cbfs_header *)(romarea + romsize - |
| bootblocksize - sizeof(struct cbfs_header)); |
| |
| master_header->magic = ntohl(CBFS_HEADER_MAGIC); |
| master_header->version = ntohl(CBFS_HEADER_VERSION); |
| master_header->romsize = htonl(romsize); |
| master_header->bootblocksize = htonl(bootblocksize); |
| master_header->align = htonl(align); |
| master_header->offset = htonl(offs); |
| master_header->architecture = htonl(CBFS_ARCHITECTURE_X86); |
| |
| ((uint32_t *) phys_to_virt(CBFS_HEADPTR_ADDR_X86))[0] = |
| virt_to_phys(master_header); |
| |
| recalculate_rom_geometry(romarea); |
| |
| cbfs_create_empty_file((0 - romsize + offs) & 0xffffffff, |
| romsize - offs - bootblocksize - |
| sizeof(struct cbfs_header) - |
| sizeof(struct cbfs_file) - 16); |
| break; |
| |
| default: |
| // Should not happen. |
| ERROR("You found a bug in cbfstool.\n"); |
| exit(1); |
| } |
| |
| writerom(romfile, romarea, romsize); |
| free(romarea); |
| return 0; |
| } |
| |
| static int in_segment(int addr, int size, int gran) |
| { |
| return ((addr & ~(gran - 1)) == ((addr + size) & ~(gran - 1))); |
| } |
| |
| uint32_t cbfs_find_location(const char *romfile, uint32_t filesize, |
| const char *filename, uint32_t alignment) |
| { |
| void *rom; |
| size_t filename_size, headersize, totalsize; |
| int ret = 0; |
| uint32_t current; |
| |
| rom = loadrom(romfile); |
| if (rom == NULL) { |
| ERROR("Could not load ROM image '%s'.\n", romfile); |
| return 0; |
| } |
| |
| filename_size = strlen(filename); |
| headersize = sizeof(struct cbfs_file) + ALIGN(filename_size + 1, 16) + |
| sizeof(struct cbfs_stage); |
| totalsize = headersize + filesize; |
| |
| current = phys_start; |
| while (current < phys_end) { |
| uint32_t top; |
| struct cbfs_file *thisfile; |
| |
| if (!cbfs_file_header(current)) { |
| current += align; |
| continue; |
| } |
| |
| thisfile = (struct cbfs_file *)phys_to_virt(current); |
| |
| top = current + ntohl(thisfile->len) + ntohl(thisfile->offset); |
| |
| if (((ntohl(thisfile->type) == 0x0) |
| || (ntohl(thisfile->type) == 0xffffffff)) |
| && (ntohl(thisfile->len) + ntohl(thisfile->offset) >= |
| totalsize)) { |
| if (in_segment |
| (current + headersize, filesize, alignment)) { |
| ret = current + headersize; |
| break; |
| } |
| if ((ALIGN(current, alignment) + filesize < top) |
| && (ALIGN(current, alignment) - headersize > |
| current) |
| && in_segment(ALIGN(current, alignment), filesize, |
| alignment)) { |
| ret = ALIGN(current, alignment); |
| break; |
| } |
| if ((ALIGN(current, alignment) + alignment + filesize < |
| top) |
| && (ALIGN(current, alignment) + alignment - |
| headersize > current) |
| && in_segment(ALIGN(current, alignment) + alignment, |
| filesize, alignment)) { |
| ret = ALIGN(current, alignment) + alignment; |
| break; |
| } |
| } |
| current = |
| ALIGN(current + ntohl(thisfile->len) + |
| ntohl(thisfile->offset), align); |
| } |
| |
| free(rom); |
| return ret; |
| } |