diff options
Diffstat (limited to 'util/cbfstool/fit.c')
-rw-r--r-- | util/cbfstool/fit.c | 54 |
1 files changed, 29 insertions, 25 deletions
diff --git a/util/cbfstool/fit.c b/util/cbfstool/fit.c index 0a7bea5725..6039714456 100644 --- a/util/cbfstool/fit.c +++ b/util/cbfstool/fit.c @@ -18,13 +18,10 @@ */ #include <stdint.h> +#include <stdio.h> #include <stdlib.h> #include <string.h> -#include <stdio.h> -#include "common.h" -#include "cbfs.h" -#include "cbfs_image.h" #include "fit.h" /* FIXME: This code assumes it is being executed on a little endian machine. */ @@ -69,9 +66,9 @@ struct microcode_entry { int size; }; -static inline void *rom_buffer_pointer(struct cbfs_image *image, int offset) +static inline void *rom_buffer_pointer(struct buffer *buffer, int offset) { - return &image->buffer.data[offset]; + return &buffer->data[offset]; } static inline int fit_entry_size_bytes(struct fit_entry *entry) @@ -104,9 +101,10 @@ static inline int fit_entry_type(struct fit_entry *entry) * in the host address space at [4G - romsize -> 4G). It also assume all * pointers have values within this address range. */ -static inline int ptr_to_offset(uint32_t theromsize, uint32_t host_ptr) +static inline int ptr_to_offset(fit_offset_converter_t helper, + const struct buffer *region, uint32_t host_ptr) { - return (int)(theromsize + host_ptr); + return helper(region, -host_ptr); } /* @@ -114,25 +112,28 @@ static inline int ptr_to_offset(uint32_t theromsize, uint32_t host_ptr) * in the host address space at [4G - romsize -> 4G). It also assume all * pointers have values within this address range. */ -static inline uint32_t offset_to_ptr(uint32_t theromsize, int offset) +static inline uint32_t offset_to_ptr(fit_offset_converter_t helper, + const struct buffer *region, int offset) { - return -(theromsize - (uint32_t )offset); + return -helper(region, offset); } -static struct fit_table *locate_fit_table(struct cbfs_image *image) +static struct fit_table *locate_fit_table(fit_offset_converter_t offset_helper, + struct buffer *buffer) { struct fit_table *table; uint32_t *fit_pointer; - fit_pointer = rom_buffer_pointer(image, - ptr_to_offset(image->buffer.size, FIT_POINTER_LOCATION)); + fit_pointer = rom_buffer_pointer(buffer, + ptr_to_offset(offset_helper, buffer, + FIT_POINTER_LOCATION)); /* Ensure pointer is below 4GiB and within 16MiB of 4GiB */ if (fit_pointer[1] != 0 || fit_pointer[0] < FIT_TABLE_LOWEST_ADDRESS) return NULL; - table = rom_buffer_pointer(image, - ptr_to_offset(image->buffer.size, *fit_pointer)); + table = rom_buffer_pointer(buffer, + ptr_to_offset(offset_helper, buffer, *fit_pointer)); /* Check that the address field has the proper signature. */ if (strncmp((const char *)&table->header.address, FIT_HEADER_ADDRESS, @@ -168,9 +169,10 @@ static void update_fit_checksum(struct fit_table *fit) fit->header.checksum = -result; } -static void add_microcodde_entries(struct cbfs_image *image, - struct fit_table *fit, - struct microcode_entry *mcus, int num_mcus) +static void add_microcodde_entries(struct fit_table *fit, + const struct cbfs_image *image, + int num_mcus, struct microcode_entry *mcus, + fit_offset_converter_t offset_helper) { int i; @@ -178,7 +180,8 @@ static void add_microcodde_entries(struct cbfs_image *image, struct fit_entry *entry = &fit->entries[i]; struct microcode_entry *mcu = &mcus[i]; - entry->address = offset_to_ptr(image->buffer.size, mcu->offset); + entry->address = offset_to_ptr(offset_helper, &image->buffer, + mcu->offset); fit_entry_update_size(entry, mcu->size); entry->version = FIT_MICROCODE_VERSION; entry->type_checksum_valid = FIT_TYPE_MICROCODE; @@ -213,9 +216,9 @@ static int parse_microcode_blob(struct cbfs_image *image, num_mcus = 0; while (file_length > sizeof(struct microcode_header)) { - struct microcode_header *mcu_header; + const struct microcode_header *mcu_header; - mcu_header = rom_buffer_pointer(image, current_offset); + mcu_header = rom_buffer_pointer(&image->buffer, current_offset); /* Quickly sanity check a prospective microcode update. */ if (mcu_header->total_size < sizeof(*mcu_header)) @@ -243,8 +246,9 @@ static int parse_microcode_blob(struct cbfs_image *image, return 0; } -int fit_update_table(struct cbfs_image *image, int empty_entries, - const char *microcode_blob_name) +int fit_update_table(struct buffer *bootblock, struct cbfs_image *image, + const char *microcode_blob_name, int empty_entries, + fit_offset_converter_t offset_fn) { struct fit_table *fit; struct cbfs_file *mcode_file; @@ -252,7 +256,7 @@ int fit_update_table(struct cbfs_image *image, int empty_entries, int ret = 0; // struct rom_image image = { .rom = rom, .size = romsize, }; - fit = locate_fit_table(image); + fit = locate_fit_table(offset_fn, bootblock); if (!fit) { ERROR("FIT not found.\n"); @@ -279,7 +283,7 @@ int fit_update_table(struct cbfs_image *image, int empty_entries, goto out; } - add_microcodde_entries(image, fit, mcus, empty_entries); + add_microcodde_entries(fit, image, empty_entries, mcus, offset_fn); update_fit_checksum(fit); out: |