| /* |
| * This file is part of the coreboot project. |
| * |
| * Copyright (C) 2003-2004 Eric Biederman |
| * Copyright (C) 2005-2010 coresystems GmbH |
| * Copyright (C) 2014 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. |
| */ |
| |
| #include <console/console.h> |
| #include <bootmem.h> |
| #include <cbmem.h> |
| #include <device/resource.h> |
| #include <stdlib.h> |
| #include <commonlib/region.h> |
| #include <fit.h> |
| #include <program_loading.h> |
| #include <timestamp.h> |
| #include <cbfs.h> |
| #include <string.h> |
| #include <commonlib/compression.h> |
| #include <lib.h> |
| #include <fit_payload.h> |
| |
| /* Pack the device_tree and place it at given position. */ |
| static void pack_fdt(struct region *fdt, struct device_tree *dt) |
| { |
| printk(BIOS_INFO, "FIT: Flattening FDT to %p\n", |
| (void *)fdt->offset); |
| |
| dt_flatten(dt, (void *)fdt->offset); |
| prog_segment_loaded(fdt->offset, fdt->size, 0); |
| } |
| |
| /** |
| * Extract a node to given regions. |
| * Returns true on error, false on success. |
| */ |
| static bool extract(struct region *region, struct fit_image_node *node) |
| { |
| void *dst = (void *)region->offset; |
| const char *comp_name; |
| size_t true_size = 0; |
| |
| switch (node->compression) { |
| case CBFS_COMPRESS_NONE: |
| comp_name = "Relocating uncompressed"; |
| break; |
| case CBFS_COMPRESS_LZMA: |
| comp_name = "Decompressing LZMA"; |
| break; |
| case CBFS_COMPRESS_LZ4: |
| comp_name = "Decompressing LZ4"; |
| break; |
| default: |
| printk(BIOS_ERR, "ERROR: Unsupported compression\n"); |
| return true; |
| } |
| |
| printk(BIOS_INFO, "FIT: %s %s to %p\n", comp_name, node->name, dst); |
| |
| switch (node->compression) { |
| case CBFS_COMPRESS_NONE: |
| memcpy(dst, node->data, node->size); |
| true_size = node->size; |
| break; |
| case CBFS_COMPRESS_LZMA: |
| timestamp_add_now(TS_START_ULZMA); |
| true_size = ulzman(node->data, node->size, dst, region->size); |
| timestamp_add_now(TS_END_ULZMA); |
| break; |
| case CBFS_COMPRESS_LZ4: |
| timestamp_add_now(TS_START_ULZ4F); |
| true_size = ulz4fn(node->data, node->size, dst, region->size); |
| timestamp_add_now(TS_END_ULZ4F); |
| break; |
| default: |
| return true; |
| } |
| |
| if (!true_size) { |
| printk(BIOS_ERR, "ERROR: %s node failed!\n", comp_name); |
| return true; |
| } |
| |
| prog_segment_loaded(region->offset, true_size, 0); |
| |
| return false; |
| } |
| |
| /* |
| * Parse the uImage FIT, choose a configuration and extract images. |
| */ |
| void fit_payload(struct prog *payload) |
| { |
| struct device_tree *dt = NULL; |
| struct region kernel = {0}, fdt = {0}, initrd = {0}; |
| void *data; |
| |
| data = rdev_mmap_full(prog_rdev(payload)); |
| |
| if (data == NULL) |
| return; |
| |
| printk(BIOS_INFO, "FIT: Examine payload %s\n", payload->name); |
| |
| struct fit_config_node *config = fit_load(data); |
| |
| if (!config || !config->kernel_node) { |
| printk(BIOS_ERR, "ERROR: Could not load FIT\n"); |
| rdev_munmap(prog_rdev(payload), data); |
| return; |
| } |
| |
| if (config->fdt_node) { |
| dt = fdt_unflatten(config->fdt_node->data); |
| if (!dt) { |
| printk(BIOS_ERR, |
| "ERROR: Failed to unflatten the FDT.\n"); |
| rdev_munmap(prog_rdev(payload), data); |
| return; |
| } |
| |
| dt_apply_fixups(dt); |
| |
| /* Update device_tree */ |
| #if defined(CONFIG_LINUX_COMMAND_LINE) |
| fit_update_chosen(dt, (char *)CONFIG_LINUX_COMMAND_LINE); |
| #endif |
| fit_update_memory(dt); |
| } |
| |
| /* Collect infos for fit_payload_arch */ |
| kernel.size = config->kernel_node->size; |
| fdt.size = dt ? dt_flat_size(dt) : 0; |
| initrd.size = config->ramdisk_node ? config->ramdisk_node->size : 0; |
| |
| /* Invoke arch specific payload placement and fixups */ |
| if (!fit_payload_arch(payload, config, &kernel, &fdt, &initrd)) { |
| printk(BIOS_ERR, "ERROR: Failed to find free memory region\n"); |
| bootmem_dump_ranges(); |
| rdev_munmap(prog_rdev(payload), data); |
| return; |
| } |
| |
| /* Load the images to given position */ |
| if (config->fdt_node) { |
| /* Update device_tree */ |
| if (config->ramdisk_node) |
| fit_add_ramdisk(dt, (void *)initrd.offset, initrd.size); |
| |
| pack_fdt(&fdt, dt); |
| } |
| |
| if (config->ramdisk_node && |
| extract(&initrd, config->ramdisk_node)) { |
| printk(BIOS_ERR, "ERROR: Failed to extract initrd\n"); |
| rdev_munmap(prog_rdev(payload), data); |
| return; |
| } |
| |
| timestamp_add_now(TS_KERNEL_DECOMPRESSION); |
| |
| if (extract(&kernel, config->kernel_node)) { |
| printk(BIOS_ERR, "ERROR: Failed to extract kernel\n"); |
| rdev_munmap(prog_rdev(payload), data); |
| return; |
| } |
| |
| timestamp_add_now(TS_START_KERNEL); |
| |
| rdev_munmap(prog_rdev(payload), data); |
| } |