Angel Pons | 118a9c7 | 2020-04-02 23:48:34 +0200 | [diff] [blame] | 1 | /* SPDX-License-Identifier: GPL-2.0-only */ |
Peter Stuge | 483b7bb | 2009-04-14 07:40:01 +0000 | [diff] [blame] | 2 | |
Aaron Durbin | 899d13d | 2015-05-15 23:39:23 -0500 | [diff] [blame] | 3 | #include <assert.h> |
Aaron Durbin | 899d13d | 2015-05-15 23:39:23 -0500 | [diff] [blame] | 4 | #include <boot_device.h> |
| 5 | #include <cbfs.h> |
Julius Werner | 1e37c9c | 2019-12-11 17:09:39 -0800 | [diff] [blame] | 6 | #include <cbmem.h> |
Julius Werner | 57d4bc6 | 2021-11-15 10:13:37 -0800 | [diff] [blame] | 7 | #include <commonlib/bsd/cbfs_private.h> |
Julius Werner | 98eeb96 | 2019-12-11 15:47:42 -0800 | [diff] [blame] | 8 | #include <commonlib/bsd/compression.h> |
Aaron Durbin | 0b418bd | 2020-10-08 14:56:03 -0600 | [diff] [blame] | 9 | #include <commonlib/endian.h> |
Bill XIE | c79e96b | 2019-08-22 20:28:36 +0800 | [diff] [blame] | 10 | #include <console/console.h> |
Bill XIE | c79e96b | 2019-08-22 20:28:36 +0800 | [diff] [blame] | 11 | #include <fmap.h> |
Aaron Durbin | 899d13d | 2015-05-15 23:39:23 -0500 | [diff] [blame] | 12 | #include <lib.h> |
Raul E Rangel | 4cfb862 | 2021-11-01 13:40:14 -0600 | [diff] [blame] | 13 | #include <list.h> |
Julius Werner | fdabf3f | 2020-05-06 17:06:35 -0700 | [diff] [blame] | 14 | #include <metadata_hash.h> |
Bill XIE | c79e96b | 2019-08-22 20:28:36 +0800 | [diff] [blame] | 15 | #include <security/tpm/tspi/crtm.h> |
| 16 | #include <security/vboot/vboot_common.h> |
| 17 | #include <stdlib.h> |
| 18 | #include <string.h> |
Aaron Durbin | 899d13d | 2015-05-15 23:39:23 -0500 | [diff] [blame] | 19 | #include <symbols.h> |
Raul E Rangel | 4cfb862 | 2021-11-01 13:40:14 -0600 | [diff] [blame] | 20 | #include <thread.h> |
Julius Werner | 09f2921 | 2015-09-29 13:51:35 -0700 | [diff] [blame] | 21 | #include <timestamp.h> |
Patrick Georgi | 58a150a | 2016-05-02 17:22:29 +0800 | [diff] [blame] | 22 | |
Raul E Rangel | 3d7b984 | 2021-10-08 13:10:38 -0600 | [diff] [blame] | 23 | #if ENV_STAGE_HAS_DATA_SECTION |
Raul E Rangel | 5ac82dc | 2021-07-23 16:43:18 -0600 | [diff] [blame] | 24 | struct mem_pool cbfs_cache = |
Raul E Rangel | 6938f35 | 2021-07-23 16:43:18 -0600 | [diff] [blame] | 25 | MEM_POOL_INIT(_cbfs_cache, REGION_SIZE(cbfs_cache), CONFIG_CBFS_CACHE_ALIGN); |
Raul E Rangel | 3d7b984 | 2021-10-08 13:10:38 -0600 | [diff] [blame] | 26 | #else |
Raul E Rangel | 5ac82dc | 2021-07-23 16:43:18 -0600 | [diff] [blame] | 27 | struct mem_pool cbfs_cache = MEM_POOL_INIT(NULL, 0, 0); |
Raul E Rangel | 3d7b984 | 2021-10-08 13:10:38 -0600 | [diff] [blame] | 28 | #endif |
Julius Werner | 9b1f3cc | 2020-12-30 17:30:12 -0800 | [diff] [blame] | 29 | |
| 30 | static void switch_to_postram_cache(int unused) |
| 31 | { |
| 32 | if (_preram_cbfs_cache != _postram_cbfs_cache) |
Raul E Rangel | 5ac82dc | 2021-07-23 16:43:18 -0600 | [diff] [blame] | 33 | mem_pool_init(&cbfs_cache, _postram_cbfs_cache, REGION_SIZE(postram_cbfs_cache), |
Raul E Rangel | 6938f35 | 2021-07-23 16:43:18 -0600 | [diff] [blame] | 34 | CONFIG_CBFS_CACHE_ALIGN); |
Julius Werner | 9b1f3cc | 2020-12-30 17:30:12 -0800 | [diff] [blame] | 35 | } |
| 36 | ROMSTAGE_CBMEM_INIT_HOOK(switch_to_postram_cache); |
Julius Werner | 9b1f3cc | 2020-12-30 17:30:12 -0800 | [diff] [blame] | 37 | |
Julius Werner | 57d4bc6 | 2021-11-15 10:13:37 -0800 | [diff] [blame] | 38 | cb_err_t _cbfs_boot_lookup(const char *name, bool force_ro, |
| 39 | union cbfs_mdata *mdata, struct region_device *rdev) |
Julius Werner | 1e37c9c | 2019-12-11 17:09:39 -0800 | [diff] [blame] | 40 | { |
Julius Werner | 0d9072b | 2020-03-05 12:51:08 -0800 | [diff] [blame] | 41 | const struct cbfs_boot_device *cbd = cbfs_get_boot_device(force_ro); |
| 42 | if (!cbd) |
| 43 | return CB_ERR; |
| 44 | |
| 45 | size_t data_offset; |
Julius Werner | 1e37c9c | 2019-12-11 17:09:39 -0800 | [diff] [blame] | 46 | cb_err_t err = CB_CBFS_CACHE_FULL; |
Julius Werner | 34cf073 | 2020-12-08 14:21:43 -0800 | [diff] [blame] | 47 | if (!CONFIG(NO_CBFS_MCACHE) && !ENV_SMM && cbd->mcache_size) |
Julius Werner | 1e37c9c | 2019-12-11 17:09:39 -0800 | [diff] [blame] | 48 | err = cbfs_mcache_lookup(cbd->mcache, cbd->mcache_size, |
Julius Werner | 723e3b1 | 2020-12-29 17:33:30 -0800 | [diff] [blame] | 49 | name, mdata, &data_offset); |
Julius Werner | fdabf3f | 2020-05-06 17:06:35 -0700 | [diff] [blame] | 50 | if (err == CB_CBFS_CACHE_FULL) { |
| 51 | struct vb2_hash *metadata_hash = NULL; |
| 52 | if (CONFIG(TOCTOU_SAFETY)) { |
| 53 | if (ENV_SMM) /* Cannot provide TOCTOU safety for SMM */ |
| 54 | dead_code(); |
Julius Werner | 34cf073 | 2020-12-08 14:21:43 -0800 | [diff] [blame] | 55 | if (!cbd->mcache_size) |
| 56 | die("Cannot access CBFS TOCTOU-safely in " ENV_STRING " before CBMEM init!\n"); |
Julius Werner | 723e3b1 | 2020-12-29 17:33:30 -0800 | [diff] [blame] | 57 | /* We can only reach this for the RW CBFS -- an mcache overflow in the |
| 58 | RO CBFS would have been caught when building the mcache in cbfs_get |
| 59 | boot_device(). (Note that TOCTOU_SAFETY implies !NO_CBFS_MCACHE.) */ |
Julius Werner | fdabf3f | 2020-05-06 17:06:35 -0700 | [diff] [blame] | 60 | assert(cbd == vboot_get_cbfs_boot_device()); |
| 61 | /* TODO: set metadata_hash to RW metadata hash here. */ |
| 62 | } |
Julius Werner | 723e3b1 | 2020-12-29 17:33:30 -0800 | [diff] [blame] | 63 | err = cbfs_lookup(&cbd->rdev, name, mdata, &data_offset, metadata_hash); |
Julius Werner | fdabf3f | 2020-05-06 17:06:35 -0700 | [diff] [blame] | 64 | } |
Julius Werner | 0d9072b | 2020-03-05 12:51:08 -0800 | [diff] [blame] | 65 | |
Julius Werner | 723e3b1 | 2020-12-29 17:33:30 -0800 | [diff] [blame] | 66 | if (CONFIG(VBOOT_ENABLE_CBFS_FALLBACK) && !force_ro && err == CB_CBFS_NOT_FOUND) { |
| 67 | printk(BIOS_INFO, "CBFS: Fall back to RO region for %s\n", name); |
Julius Werner | 57d4bc6 | 2021-11-15 10:13:37 -0800 | [diff] [blame] | 68 | return _cbfs_boot_lookup(name, true, mdata, rdev); |
Julius Werner | 0d9072b | 2020-03-05 12:51:08 -0800 | [diff] [blame] | 69 | } |
Julius Werner | 0247fcf | 2020-12-03 12:41:00 -0800 | [diff] [blame] | 70 | if (err) { |
| 71 | if (err == CB_CBFS_NOT_FOUND) |
| 72 | printk(BIOS_WARNING, "CBFS: '%s' not found.\n", name); |
| 73 | else if (err == CB_CBFS_HASH_MISMATCH) |
| 74 | printk(BIOS_ERR, "CBFS ERROR: metadata hash mismatch!\n"); |
| 75 | else |
Julius Werner | 723e3b1 | 2020-12-29 17:33:30 -0800 | [diff] [blame] | 76 | printk(BIOS_ERR, "CBFS ERROR: error %d when looking up '%s'\n", |
Julius Werner | 0247fcf | 2020-12-03 12:41:00 -0800 | [diff] [blame] | 77 | err, name); |
Julius Werner | 0d9072b | 2020-03-05 12:51:08 -0800 | [diff] [blame] | 78 | return err; |
Julius Werner | 0247fcf | 2020-12-03 12:41:00 -0800 | [diff] [blame] | 79 | } |
Julius Werner | 0d9072b | 2020-03-05 12:51:08 -0800 | [diff] [blame] | 80 | |
| 81 | if (rdev_chain(rdev, &cbd->rdev, data_offset, be32toh(mdata->h.len))) |
| 82 | return CB_ERR; |
| 83 | |
Arthur Heymans | 27545df | 2021-02-26 14:22:47 +0100 | [diff] [blame] | 84 | if (tspi_measure_cbfs_hook(rdev, name, be32toh(mdata->h.type))) { |
| 85 | printk(BIOS_ERR, "CBFS ERROR: error when measuring '%s'\n", name); |
| 86 | } |
Julius Werner | 0d9072b | 2020-03-05 12:51:08 -0800 | [diff] [blame] | 87 | |
| 88 | return CB_SUCCESS; |
Julius Werner | 1e37c9c | 2019-12-11 17:09:39 -0800 | [diff] [blame] | 89 | } |
| 90 | |
Aaron Durbin | 37a5d15 | 2015-09-17 16:09:30 -0500 | [diff] [blame] | 91 | int cbfs_boot_locate(struct cbfsf *fh, const char *name, uint32_t *type) |
Aaron Durbin | 899d13d | 2015-05-15 23:39:23 -0500 | [diff] [blame] | 92 | { |
Julius Werner | 57d4bc6 | 2021-11-15 10:13:37 -0800 | [diff] [blame] | 93 | if (_cbfs_boot_lookup(name, false, &fh->mdata, &fh->data)) |
Julius Werner | 1cd013b | 2019-12-11 16:50:02 -0800 | [diff] [blame] | 94 | return -1; |
| 95 | |
| 96 | size_t msize = be32toh(fh->mdata.h.offset); |
Julius Werner | c893197 | 2021-04-16 16:48:32 -0700 | [diff] [blame] | 97 | if (rdev_chain_mem(&fh->metadata, &fh->mdata, msize)) |
Julius Werner | 1cd013b | 2019-12-11 16:50:02 -0800 | [diff] [blame] | 98 | return -1; |
Julius Werner | 0d9072b | 2020-03-05 12:51:08 -0800 | [diff] [blame] | 99 | |
Julius Werner | 1cd013b | 2019-12-11 16:50:02 -0800 | [diff] [blame] | 100 | if (type) { |
| 101 | if (!*type) |
| 102 | *type = be32toh(fh->mdata.h.type); |
| 103 | else if (*type != be32toh(fh->mdata.h.type)) |
| 104 | return -1; |
Wim Vervoorn | 114e2e8 | 2019-11-05 14:09:16 +0100 | [diff] [blame] | 105 | } |
| 106 | |
Julius Werner | 1cd013b | 2019-12-11 16:50:02 -0800 | [diff] [blame] | 107 | return 0; |
Aaron Durbin | 899d13d | 2015-05-15 23:39:23 -0500 | [diff] [blame] | 108 | } |
| 109 | |
Julius Werner | 9b1f3cc | 2020-12-30 17:30:12 -0800 | [diff] [blame] | 110 | void cbfs_unmap(void *mapping) |
Julius Werner | 834b3ec | 2020-03-04 16:52:08 -0800 | [diff] [blame] | 111 | { |
Julius Werner | 9b1f3cc | 2020-12-30 17:30:12 -0800 | [diff] [blame] | 112 | /* |
| 113 | * This is save to call with mappings that weren't allocated in the cache (e.g. x86 |
| 114 | * direct mappings) -- mem_pool_free() just does nothing for addresses it doesn't |
| 115 | * recognize. This hardcodes the assumption that if platforms implement an rdev_mmap() |
| 116 | * that requires a free() for the boot_device, they need to implement it via the |
| 117 | * cbfs_cache mem_pool. |
| 118 | */ |
Raul E Rangel | 3d7b984 | 2021-10-08 13:10:38 -0600 | [diff] [blame] | 119 | mem_pool_free(&cbfs_cache, mapping); |
Julius Werner | 834b3ec | 2020-03-04 16:52:08 -0800 | [diff] [blame] | 120 | } |
| 121 | |
Pratik Prajapati | 2a7708a | 2016-11-30 17:29:10 -0800 | [diff] [blame] | 122 | int cbfs_locate_file_in_region(struct cbfsf *fh, const char *region_name, |
Philipp Deppenwiese | 66f9a09 | 2018-11-08 10:59:40 +0100 | [diff] [blame] | 123 | const char *name, uint32_t *type) |
Pratik Prajapati | 2a7708a | 2016-11-30 17:29:10 -0800 | [diff] [blame] | 124 | { |
| 125 | struct region_device rdev; |
Bill XIE | bad08c2 | 2020-02-13 11:11:35 +0800 | [diff] [blame] | 126 | int ret = 0; |
Pratik Prajapati | 2a7708a | 2016-11-30 17:29:10 -0800 | [diff] [blame] | 127 | if (fmap_locate_area_as_rdev(region_name, &rdev)) { |
Julius Werner | 723e3b1 | 2020-12-29 17:33:30 -0800 | [diff] [blame] | 128 | LOG("%s region not found while looking for %s\n", region_name, name); |
Pratik Prajapati | 2a7708a | 2016-11-30 17:29:10 -0800 | [diff] [blame] | 129 | return -1; |
| 130 | } |
| 131 | |
Julius Werner | 0d9072b | 2020-03-05 12:51:08 -0800 | [diff] [blame] | 132 | uint32_t dummy_type = 0; |
| 133 | if (!type) |
| 134 | type = &dummy_type; |
| 135 | |
Bill XIE | bad08c2 | 2020-02-13 11:11:35 +0800 | [diff] [blame] | 136 | ret = cbfs_locate(fh, &rdev, name, type); |
| 137 | if (!ret) |
Julius Werner | 0d9072b | 2020-03-05 12:51:08 -0800 | [diff] [blame] | 138 | if (tspi_measure_cbfs_hook(&rdev, name, *type)) |
Arthur Heymans | 27545df | 2021-02-26 14:22:47 +0100 | [diff] [blame] | 139 | LOG("error measuring %s in region %s\n", name, region_name); |
Bill XIE | bad08c2 | 2020-02-13 11:11:35 +0800 | [diff] [blame] | 140 | return ret; |
Pratik Prajapati | 2a7708a | 2016-11-30 17:29:10 -0800 | [diff] [blame] | 141 | } |
| 142 | |
Aaron Durbin | a85febc | 2020-05-15 15:09:10 -0600 | [diff] [blame] | 143 | static inline bool fsps_env(void) |
| 144 | { |
| 145 | /* FSP-S is assumed to be loaded in ramstage. */ |
| 146 | if (ENV_RAMSTAGE) |
| 147 | return true; |
| 148 | return false; |
| 149 | } |
| 150 | |
Aaron Durbin | ecbfa99 | 2020-05-15 17:01:58 -0600 | [diff] [blame] | 151 | static inline bool fspm_env(void) |
| 152 | { |
| 153 | /* FSP-M is assumed to be loaded in romstage. */ |
| 154 | if (ENV_ROMSTAGE) |
| 155 | return true; |
| 156 | return false; |
| 157 | } |
| 158 | |
Aaron Durbin | a121f95 | 2020-05-26 15:48:10 -0600 | [diff] [blame] | 159 | static inline bool cbfs_lz4_enabled(void) |
| 160 | { |
Aaron Durbin | a85febc | 2020-05-15 15:09:10 -0600 | [diff] [blame] | 161 | if (fsps_env() && CONFIG(FSP_COMPRESS_FSP_S_LZ4)) |
| 162 | return true; |
Aaron Durbin | ecbfa99 | 2020-05-15 17:01:58 -0600 | [diff] [blame] | 163 | if (fspm_env() && CONFIG(FSP_COMPRESS_FSP_M_LZ4)) |
| 164 | return true; |
Aaron Durbin | a85febc | 2020-05-15 15:09:10 -0600 | [diff] [blame] | 165 | |
Aaron Durbin | a121f95 | 2020-05-26 15:48:10 -0600 | [diff] [blame] | 166 | if ((ENV_BOOTBLOCK || ENV_SEPARATE_VERSTAGE) && !CONFIG(COMPRESS_PRERAM_STAGES)) |
| 167 | return false; |
| 168 | |
Julius Werner | 9b1f3cc | 2020-12-30 17:30:12 -0800 | [diff] [blame] | 169 | if (ENV_SMM) |
| 170 | return false; |
| 171 | |
Aaron Durbin | a121f95 | 2020-05-26 15:48:10 -0600 | [diff] [blame] | 172 | return true; |
| 173 | } |
| 174 | |
| 175 | static inline bool cbfs_lzma_enabled(void) |
| 176 | { |
Aaron Durbin | a85febc | 2020-05-15 15:09:10 -0600 | [diff] [blame] | 177 | if (fsps_env() && CONFIG(FSP_COMPRESS_FSP_S_LZMA)) |
| 178 | return true; |
Aaron Durbin | ecbfa99 | 2020-05-15 17:01:58 -0600 | [diff] [blame] | 179 | if (fspm_env() && CONFIG(FSP_COMPRESS_FSP_M_LZMA)) |
| 180 | return true; |
Aaron Durbin | a121f95 | 2020-05-26 15:48:10 -0600 | [diff] [blame] | 181 | /* We assume here romstage and postcar are never compressed. */ |
| 182 | if (ENV_BOOTBLOCK || ENV_SEPARATE_VERSTAGE) |
| 183 | return false; |
| 184 | if (ENV_ROMSTAGE && CONFIG(POSTCAR_STAGE)) |
| 185 | return false; |
Julius Werner | 723e3b1 | 2020-12-29 17:33:30 -0800 | [diff] [blame] | 186 | if ((ENV_ROMSTAGE || ENV_POSTCAR) && !CONFIG(COMPRESS_RAMSTAGE)) |
Aaron Durbin | a121f95 | 2020-05-26 15:48:10 -0600 | [diff] [blame] | 187 | return false; |
Julius Werner | 9b1f3cc | 2020-12-30 17:30:12 -0800 | [diff] [blame] | 188 | if (ENV_SMM) |
| 189 | return false; |
Aaron Durbin | a121f95 | 2020-05-26 15:48:10 -0600 | [diff] [blame] | 190 | return true; |
| 191 | } |
| 192 | |
Julius Werner | 05714cc | 2021-04-15 23:25:44 -0700 | [diff] [blame] | 193 | static bool cbfs_file_hash_mismatch(const void *buffer, size_t size, |
| 194 | const union cbfs_mdata *mdata, bool skip_verification) |
Julius Werner | fccf122 | 2021-04-02 15:58:05 -0700 | [diff] [blame] | 195 | { |
| 196 | /* Avoid linking hash functions when verification is disabled. */ |
Julius Werner | 05714cc | 2021-04-15 23:25:44 -0700 | [diff] [blame] | 197 | if (!CONFIG(CBFS_VERIFICATION) || skip_verification) |
Julius Werner | fccf122 | 2021-04-02 15:58:05 -0700 | [diff] [blame] | 198 | return false; |
| 199 | |
Julius Werner | 05714cc | 2021-04-15 23:25:44 -0700 | [diff] [blame] | 200 | const struct vb2_hash *hash = cbfs_file_hash(mdata); |
| 201 | if (!hash) { |
| 202 | ERROR("'%s' does not have a file hash!\n", mdata->h.filename); |
| 203 | return true; |
| 204 | } |
Julius Werner | fccf122 | 2021-04-02 15:58:05 -0700 | [diff] [blame] | 205 | |
Julius Werner | 05714cc | 2021-04-15 23:25:44 -0700 | [diff] [blame] | 206 | if (vb2_hash_verify(buffer, size, hash) != VB2_SUCCESS) { |
| 207 | ERROR("'%s' file hash mismatch!\n", mdata->h.filename); |
| 208 | return true; |
| 209 | } |
| 210 | |
| 211 | return false; |
Julius Werner | fccf122 | 2021-04-02 15:58:05 -0700 | [diff] [blame] | 212 | } |
| 213 | |
| 214 | static size_t cbfs_load_and_decompress(const struct region_device *rdev, void *buffer, |
| 215 | size_t buffer_size, uint32_t compression, |
Julius Werner | 05714cc | 2021-04-15 23:25:44 -0700 | [diff] [blame] | 216 | const union cbfs_mdata *mdata, bool skip_verification) |
Aaron Durbin | 899d13d | 2015-05-15 23:39:23 -0500 | [diff] [blame] | 217 | { |
Julius Werner | eca99af | 2021-03-10 18:49:37 -0800 | [diff] [blame] | 218 | size_t in_size = region_device_sz(rdev); |
Julius Werner | fccf122 | 2021-04-02 15:58:05 -0700 | [diff] [blame] | 219 | size_t out_size = 0; |
Aaron Durbin | 84f394e | 2020-05-26 16:16:42 -0600 | [diff] [blame] | 220 | void *map; |
Julius Werner | 09f2921 | 2015-09-29 13:51:35 -0700 | [diff] [blame] | 221 | |
Julius Werner | 05714cc | 2021-04-15 23:25:44 -0700 | [diff] [blame] | 222 | DEBUG("Decompressing %zu bytes from '%s' to %p with algo %d\n", |
| 223 | in_size, mdata->h.filename, buffer, compression); |
Julius Werner | 7778cf2 | 2021-01-05 18:54:19 -0800 | [diff] [blame] | 224 | |
Julius Werner | 09f2921 | 2015-09-29 13:51:35 -0700 | [diff] [blame] | 225 | switch (compression) { |
| 226 | case CBFS_COMPRESS_NONE: |
Julius Werner | f975e55 | 2016-08-19 15:43:06 -0700 | [diff] [blame] | 227 | if (buffer_size < in_size) |
| 228 | return 0; |
Julius Werner | eca99af | 2021-03-10 18:49:37 -0800 | [diff] [blame] | 229 | if (rdev_readat(rdev, buffer, 0, in_size) != in_size) |
Julius Werner | 09f2921 | 2015-09-29 13:51:35 -0700 | [diff] [blame] | 230 | return 0; |
Julius Werner | 05714cc | 2021-04-15 23:25:44 -0700 | [diff] [blame] | 231 | if (cbfs_file_hash_mismatch(buffer, in_size, mdata, skip_verification)) |
Julius Werner | fccf122 | 2021-04-02 15:58:05 -0700 | [diff] [blame] | 232 | return 0; |
Julius Werner | 09f2921 | 2015-09-29 13:51:35 -0700 | [diff] [blame] | 233 | return in_size; |
| 234 | |
| 235 | case CBFS_COMPRESS_LZ4: |
Aaron Durbin | a121f95 | 2020-05-26 15:48:10 -0600 | [diff] [blame] | 236 | if (!cbfs_lz4_enabled()) |
Julius Werner | 09f2921 | 2015-09-29 13:51:35 -0700 | [diff] [blame] | 237 | return 0; |
| 238 | |
Julius Werner | eca99af | 2021-03-10 18:49:37 -0800 | [diff] [blame] | 239 | /* cbfs_prog_stage_load() takes care of in-place LZ4 decompression by |
Julius Werner | 723e3b1 | 2020-12-29 17:33:30 -0800 | [diff] [blame] | 240 | setting up the rdev to be in memory. */ |
Julius Werner | eca99af | 2021-03-10 18:49:37 -0800 | [diff] [blame] | 241 | map = rdev_mmap_full(rdev); |
Aaron Durbin | 84f394e | 2020-05-26 16:16:42 -0600 | [diff] [blame] | 242 | if (map == NULL) |
Julius Werner | 09f2921 | 2015-09-29 13:51:35 -0700 | [diff] [blame] | 243 | return 0; |
| 244 | |
Julius Werner | 05714cc | 2021-04-15 23:25:44 -0700 | [diff] [blame] | 245 | if (!cbfs_file_hash_mismatch(map, in_size, mdata, skip_verification)) { |
Julius Werner | fccf122 | 2021-04-02 15:58:05 -0700 | [diff] [blame] | 246 | timestamp_add_now(TS_START_ULZ4F); |
| 247 | out_size = ulz4fn(map, in_size, buffer, buffer_size); |
| 248 | timestamp_add_now(TS_END_ULZ4F); |
| 249 | } |
Aaron Durbin | 84f394e | 2020-05-26 16:16:42 -0600 | [diff] [blame] | 250 | |
| 251 | rdev_munmap(rdev, map); |
| 252 | |
Julius Werner | 09f2921 | 2015-09-29 13:51:35 -0700 | [diff] [blame] | 253 | return out_size; |
| 254 | |
| 255 | case CBFS_COMPRESS_LZMA: |
Aaron Durbin | a121f95 | 2020-05-26 15:48:10 -0600 | [diff] [blame] | 256 | if (!cbfs_lzma_enabled()) |
Julius Werner | 09f2921 | 2015-09-29 13:51:35 -0700 | [diff] [blame] | 257 | return 0; |
Julius Werner | eca99af | 2021-03-10 18:49:37 -0800 | [diff] [blame] | 258 | map = rdev_mmap_full(rdev); |
Julius Werner | 09f2921 | 2015-09-29 13:51:35 -0700 | [diff] [blame] | 259 | if (map == NULL) |
| 260 | return 0; |
| 261 | |
Julius Werner | 05714cc | 2021-04-15 23:25:44 -0700 | [diff] [blame] | 262 | if (!cbfs_file_hash_mismatch(map, in_size, mdata, skip_verification)) { |
Julius Werner | fccf122 | 2021-04-02 15:58:05 -0700 | [diff] [blame] | 263 | /* Note: timestamp not useful for memory-mapped media (x86) */ |
| 264 | timestamp_add_now(TS_START_ULZMA); |
| 265 | out_size = ulzman(map, in_size, buffer, buffer_size); |
| 266 | timestamp_add_now(TS_END_ULZMA); |
| 267 | } |
Julius Werner | 09f2921 | 2015-09-29 13:51:35 -0700 | [diff] [blame] | 268 | |
| 269 | rdev_munmap(rdev, map); |
| 270 | |
| 271 | return out_size; |
| 272 | |
| 273 | default: |
Aaron Durbin | 899d13d | 2015-05-15 23:39:23 -0500 | [diff] [blame] | 274 | return 0; |
Julius Werner | 09f2921 | 2015-09-29 13:51:35 -0700 | [diff] [blame] | 275 | } |
Aaron Durbin | 899d13d | 2015-05-15 23:39:23 -0500 | [diff] [blame] | 276 | } |
| 277 | |
Raul E Rangel | 4cfb862 | 2021-11-01 13:40:14 -0600 | [diff] [blame] | 278 | struct cbfs_preload_context { |
| 279 | struct region_device rdev; |
| 280 | struct thread_handle handle; |
| 281 | struct list_node list_node; |
| 282 | void *buffer; |
| 283 | char name[]; |
| 284 | }; |
| 285 | |
| 286 | static struct list_node cbfs_preload_context_list; |
| 287 | |
| 288 | static struct cbfs_preload_context *alloc_cbfs_preload_context(size_t additional) |
| 289 | { |
| 290 | struct cbfs_preload_context *context; |
| 291 | size_t size = sizeof(*context) + additional; |
| 292 | |
| 293 | context = mem_pool_alloc(&cbfs_cache, size); |
| 294 | |
| 295 | if (!context) |
| 296 | return NULL; |
| 297 | |
| 298 | memset(context, 0, size); |
| 299 | |
| 300 | return context; |
| 301 | } |
| 302 | |
| 303 | static void append_cbfs_preload_context(struct cbfs_preload_context *context) |
| 304 | { |
| 305 | list_append(&context->list_node, &cbfs_preload_context_list); |
| 306 | } |
| 307 | |
| 308 | static void free_cbfs_preload_context(struct cbfs_preload_context *context) |
| 309 | { |
| 310 | list_remove(&context->list_node); |
| 311 | |
| 312 | mem_pool_free(&cbfs_cache, context); |
| 313 | } |
| 314 | |
| 315 | static enum cb_err cbfs_preload_thread_entry(void *arg) |
| 316 | { |
| 317 | struct cbfs_preload_context *context = arg; |
| 318 | |
Julius Werner | f364eb9 | 2021-12-01 14:48:59 -0800 | [diff] [blame^] | 319 | if (rdev_read_full(&context->rdev, context->buffer) < 0) { |
Raul E Rangel | 4cfb862 | 2021-11-01 13:40:14 -0600 | [diff] [blame] | 320 | ERROR("%s(name='%s') readat failed\n", __func__, context->name); |
| 321 | return CB_ERR; |
| 322 | } |
| 323 | |
| 324 | return CB_SUCCESS; |
| 325 | } |
| 326 | |
| 327 | void cbfs_preload(const char *name) |
| 328 | { |
| 329 | struct region_device rdev; |
| 330 | union cbfs_mdata mdata; |
| 331 | struct cbfs_preload_context *context; |
| 332 | bool force_ro = false; |
| 333 | size_t size; |
| 334 | |
| 335 | if (!CONFIG(CBFS_PRELOAD)) |
| 336 | dead_code(); |
| 337 | |
| 338 | DEBUG("%s(name='%s')\n", __func__, name); |
| 339 | |
Julius Werner | 57d4bc6 | 2021-11-15 10:13:37 -0800 | [diff] [blame] | 340 | if (_cbfs_boot_lookup(name, force_ro, &mdata, &rdev)) |
Raul E Rangel | 4cfb862 | 2021-11-01 13:40:14 -0600 | [diff] [blame] | 341 | return; |
| 342 | |
| 343 | size = region_device_sz(&rdev); |
| 344 | |
| 345 | context = alloc_cbfs_preload_context(strlen(name) + 1); |
| 346 | if (!context) { |
| 347 | ERROR("%s(name='%s') failed to allocate preload context\n", __func__, name); |
| 348 | return; |
| 349 | } |
| 350 | |
| 351 | context->buffer = mem_pool_alloc(&cbfs_cache, size); |
| 352 | if (context->buffer == NULL) { |
| 353 | ERROR("%s(name='%s') failed to allocate %zu bytes for preload buffer\n", |
| 354 | __func__, name, size); |
| 355 | goto out; |
| 356 | } |
| 357 | |
| 358 | context->rdev = rdev; |
| 359 | strcpy(context->name, name); |
| 360 | |
| 361 | append_cbfs_preload_context(context); |
| 362 | |
| 363 | if (thread_run(&context->handle, cbfs_preload_thread_entry, context) == 0) |
| 364 | return; |
| 365 | |
| 366 | ERROR("%s(name='%s') failed to start preload thread\n", __func__, name); |
| 367 | mem_pool_free(&cbfs_cache, context->buffer); |
| 368 | |
| 369 | out: |
| 370 | free_cbfs_preload_context(context); |
| 371 | } |
| 372 | |
| 373 | static struct cbfs_preload_context *find_cbfs_preload_context(const char *name) |
| 374 | { |
| 375 | struct cbfs_preload_context *context; |
| 376 | |
| 377 | list_for_each(context, cbfs_preload_context_list, list_node) { |
| 378 | if (strcmp(context->name, name) == 0) |
| 379 | return context; |
| 380 | } |
| 381 | |
| 382 | return NULL; |
| 383 | } |
| 384 | |
| 385 | static enum cb_err get_preload_rdev(struct region_device *rdev, const char *name) |
| 386 | { |
| 387 | enum cb_err err; |
| 388 | struct cbfs_preload_context *context; |
| 389 | |
| 390 | if (!CONFIG(CBFS_PRELOAD) || (!ENV_RAMSTAGE && !ENV_ROMSTAGE)) |
| 391 | return CB_ERR_ARG; |
| 392 | |
| 393 | context = find_cbfs_preload_context(name); |
| 394 | if (!context) |
| 395 | return CB_ERR_ARG; |
| 396 | |
| 397 | err = thread_join(&context->handle); |
| 398 | if (err != CB_SUCCESS) { |
| 399 | ERROR("%s(name='%s') Preload thread failed: %u\n", __func__, name, err); |
| 400 | |
| 401 | goto out; |
| 402 | } |
| 403 | |
| 404 | if (rdev_chain_mem(rdev, context->buffer, region_device_sz(&context->rdev)) != 0) { |
| 405 | ERROR("%s(name='%s') chaining failed\n", __func__, name); |
| 406 | |
| 407 | err = CB_ERR; |
| 408 | goto out; |
| 409 | } |
| 410 | |
| 411 | err = CB_SUCCESS; |
| 412 | |
| 413 | DEBUG("%s(name='%s') preload successful\n", __func__, name); |
| 414 | |
| 415 | out: |
| 416 | free_cbfs_preload_context(context); |
| 417 | |
| 418 | return err; |
| 419 | } |
| 420 | |
Julius Werner | 05714cc | 2021-04-15 23:25:44 -0700 | [diff] [blame] | 421 | static void *do_alloc(union cbfs_mdata *mdata, struct region_device *rdev, |
| 422 | cbfs_allocator_t allocator, void *arg, size_t *size_out, |
| 423 | bool skip_verification) |
| 424 | { |
| 425 | size_t size = region_device_sz(rdev); |
| 426 | void *loc = NULL; |
| 427 | |
| 428 | uint32_t compression = CBFS_COMPRESS_NONE; |
| 429 | const struct cbfs_file_attr_compression *cattr = cbfs_find_attr(mdata, |
| 430 | CBFS_FILE_ATTR_TAG_COMPRESSION, sizeof(*cattr)); |
| 431 | if (cattr) { |
| 432 | compression = be32toh(cattr->compression); |
| 433 | size = be32toh(cattr->decompressed_size); |
| 434 | } |
| 435 | |
| 436 | if (size_out) |
| 437 | *size_out = size; |
| 438 | |
| 439 | /* allocator == NULL means do a cbfs_map() */ |
| 440 | if (allocator) { |
| 441 | loc = allocator(arg, size, mdata); |
| 442 | } else if (compression == CBFS_COMPRESS_NONE) { |
| 443 | void *mapping = rdev_mmap_full(rdev); |
| 444 | if (!mapping) |
| 445 | return NULL; |
| 446 | if (cbfs_file_hash_mismatch(mapping, size, mdata, skip_verification)) { |
| 447 | rdev_munmap(rdev, mapping); |
| 448 | return NULL; |
| 449 | } |
| 450 | return mapping; |
| 451 | } else if (!cbfs_cache.size) { |
| 452 | /* In order to use the cbfs_cache you need to add a CBFS_CACHE to your |
| 453 | * memlayout. For stages that don't have .data sections (x86 pre-RAM), |
| 454 | * it is not possible to add a CBFS_CACHE. */ |
| 455 | ERROR("Cannot map compressed file %s without cbfs_cache\n", mdata->h.filename); |
| 456 | return NULL; |
| 457 | } else { |
| 458 | loc = mem_pool_alloc(&cbfs_cache, size); |
| 459 | } |
| 460 | |
| 461 | if (!loc) { |
| 462 | ERROR("'%s' allocation failure\n", mdata->h.filename); |
| 463 | return NULL; |
| 464 | } |
| 465 | |
| 466 | size = cbfs_load_and_decompress(rdev, loc, size, compression, mdata, skip_verification); |
| 467 | if (!size) |
| 468 | return NULL; |
| 469 | |
| 470 | return loc; |
| 471 | } |
| 472 | |
Julius Werner | 7778cf2 | 2021-01-05 18:54:19 -0800 | [diff] [blame] | 473 | void *_cbfs_alloc(const char *name, cbfs_allocator_t allocator, void *arg, |
| 474 | size_t *size_out, bool force_ro, enum cbfs_type *type) |
Julius Werner | f975e55 | 2016-08-19 15:43:06 -0700 | [diff] [blame] | 475 | { |
Julius Werner | d17ce41 | 2020-10-01 18:25:49 -0700 | [diff] [blame] | 476 | struct region_device rdev; |
Raul E Rangel | 4cfb862 | 2021-11-01 13:40:14 -0600 | [diff] [blame] | 477 | bool preload_successful = false; |
Julius Werner | d17ce41 | 2020-10-01 18:25:49 -0700 | [diff] [blame] | 478 | union cbfs_mdata mdata; |
Julius Werner | 7778cf2 | 2021-01-05 18:54:19 -0800 | [diff] [blame] | 479 | |
| 480 | DEBUG("%s(name='%s', alloc=%p(%p), force_ro=%s, type=%d)\n", __func__, name, allocator, |
| 481 | arg, force_ro ? "true" : "false", type ? *type : -1); |
Julius Werner | f975e55 | 2016-08-19 15:43:06 -0700 | [diff] [blame] | 482 | |
Julius Werner | 57d4bc6 | 2021-11-15 10:13:37 -0800 | [diff] [blame] | 483 | if (_cbfs_boot_lookup(name, force_ro, &mdata, &rdev)) |
Julius Werner | 7778cf2 | 2021-01-05 18:54:19 -0800 | [diff] [blame] | 484 | return NULL; |
Julius Werner | f975e55 | 2016-08-19 15:43:06 -0700 | [diff] [blame] | 485 | |
Julius Werner | 7778cf2 | 2021-01-05 18:54:19 -0800 | [diff] [blame] | 486 | if (type) { |
| 487 | const enum cbfs_type real_type = be32toh(mdata.h.type); |
| 488 | if (*type == CBFS_TYPE_QUERY) |
| 489 | *type = real_type; |
| 490 | else if (*type != real_type) { |
| 491 | ERROR("'%s' type mismatch (is %u, expected %u)\n", |
| 492 | mdata.h.filename, real_type, *type); |
| 493 | return NULL; |
| 494 | } |
Julius Werner | d17ce41 | 2020-10-01 18:25:49 -0700 | [diff] [blame] | 495 | } |
Julius Werner | f975e55 | 2016-08-19 15:43:06 -0700 | [diff] [blame] | 496 | |
Raul E Rangel | 4cfb862 | 2021-11-01 13:40:14 -0600 | [diff] [blame] | 497 | /* Update the rdev with the preload content */ |
| 498 | if (!force_ro && get_preload_rdev(&rdev, name) == CB_SUCCESS) |
| 499 | preload_successful = true; |
| 500 | |
Julius Werner | 05714cc | 2021-04-15 23:25:44 -0700 | [diff] [blame] | 501 | void *ret = do_alloc(&mdata, &rdev, allocator, arg, size_out, false); |
Raul E Rangel | 3f41d32 | 2021-07-23 15:23:12 -0600 | [diff] [blame] | 502 | |
Julius Werner | 05714cc | 2021-04-15 23:25:44 -0700 | [diff] [blame] | 503 | /* When using cbfs_preload we need to free the preload buffer after populating the |
| 504 | * destination buffer. We know we must have a mem_rdev here, so extra mmap is fine. */ |
Raul E Rangel | 4cfb862 | 2021-11-01 13:40:14 -0600 | [diff] [blame] | 505 | if (preload_successful) |
| 506 | cbfs_unmap(rdev_mmap_full(&rdev)); |
Julius Werner | 7778cf2 | 2021-01-05 18:54:19 -0800 | [diff] [blame] | 507 | |
Julius Werner | 05714cc | 2021-04-15 23:25:44 -0700 | [diff] [blame] | 508 | return ret; |
| 509 | } |
| 510 | |
| 511 | void *_cbfs_unverified_area_alloc(const char *area, const char *name, |
| 512 | cbfs_allocator_t allocator, void *arg, size_t *size_out) |
| 513 | { |
| 514 | struct region_device area_rdev, file_rdev; |
| 515 | union cbfs_mdata mdata; |
| 516 | size_t data_offset; |
| 517 | |
| 518 | DEBUG("%s(area='%s', name='%s', alloc=%p(%p))\n", __func__, area, name, allocator, arg); |
| 519 | |
| 520 | if (fmap_locate_area_as_rdev(area, &area_rdev)) |
| 521 | return NULL; |
| 522 | |
| 523 | if (cbfs_lookup(&area_rdev, name, &mdata, &data_offset, NULL)) { |
| 524 | ERROR("'%s' not found in '%s'\n", name, area); |
| 525 | return NULL; |
| 526 | } |
| 527 | |
| 528 | if (rdev_chain(&file_rdev, &area_rdev, data_offset, be32toh(mdata.h.len))) |
| 529 | return NULL; |
| 530 | |
| 531 | if (tspi_measure_cbfs_hook(&file_rdev, name, be32toh(mdata.h.type))) |
| 532 | ERROR("error measuring '%s' in '%s'\n", name, area); |
| 533 | |
| 534 | return do_alloc(&mdata, &file_rdev, allocator, arg, size_out, true); |
Julius Werner | 7778cf2 | 2021-01-05 18:54:19 -0800 | [diff] [blame] | 535 | } |
| 536 | |
Julius Werner | 4676ec5 | 2021-03-10 16:52:14 -0800 | [diff] [blame] | 537 | void *_cbfs_default_allocator(void *arg, size_t size, const union cbfs_mdata *unused) |
Julius Werner | 7778cf2 | 2021-01-05 18:54:19 -0800 | [diff] [blame] | 538 | { |
| 539 | struct _cbfs_default_allocator_arg *darg = arg; |
| 540 | if (size > darg->buf_size) |
| 541 | return NULL; |
| 542 | return darg->buf; |
| 543 | } |
| 544 | |
Julius Werner | 4676ec5 | 2021-03-10 16:52:14 -0800 | [diff] [blame] | 545 | void *_cbfs_cbmem_allocator(void *arg, size_t size, const union cbfs_mdata *unused) |
Julius Werner | 7778cf2 | 2021-01-05 18:54:19 -0800 | [diff] [blame] | 546 | { |
| 547 | return cbmem_add((uintptr_t)arg, size); |
Julius Werner | f975e55 | 2016-08-19 15:43:06 -0700 | [diff] [blame] | 548 | } |
| 549 | |
Julius Werner | 1de8708 | 2020-12-23 17:38:11 -0800 | [diff] [blame] | 550 | cb_err_t cbfs_prog_stage_load(struct prog *pstage) |
Aaron Durbin | 899d13d | 2015-05-15 23:39:23 -0500 | [diff] [blame] | 551 | { |
Julius Werner | 1de8708 | 2020-12-23 17:38:11 -0800 | [diff] [blame] | 552 | union cbfs_mdata mdata; |
| 553 | struct region_device rdev; |
Julius Werner | 1de8708 | 2020-12-23 17:38:11 -0800 | [diff] [blame] | 554 | cb_err_t err; |
Hung-Te Lin | 6fe0cab | 2013-01-22 18:57:56 +0800 | [diff] [blame] | 555 | |
Julius Werner | 1de8708 | 2020-12-23 17:38:11 -0800 | [diff] [blame] | 556 | prog_locate_hook(pstage); |
Aaron Durbin | 899d13d | 2015-05-15 23:39:23 -0500 | [diff] [blame] | 557 | |
Julius Werner | 57d4bc6 | 2021-11-15 10:13:37 -0800 | [diff] [blame] | 558 | if ((err = _cbfs_boot_lookup(prog_name(pstage), false, &mdata, &rdev))) |
Julius Werner | 1de8708 | 2020-12-23 17:38:11 -0800 | [diff] [blame] | 559 | return err; |
| 560 | |
| 561 | assert(be32toh(mdata.h.type) == CBFS_TYPE_STAGE); |
| 562 | pstage->cbfs_type = CBFS_TYPE_STAGE; |
| 563 | |
Julius Werner | 81dc20e | 2020-10-15 17:37:57 -0700 | [diff] [blame] | 564 | enum cbfs_compression compression = CBFS_COMPRESS_NONE; |
| 565 | const struct cbfs_file_attr_compression *cattr = cbfs_find_attr(&mdata, |
| 566 | CBFS_FILE_ATTR_TAG_COMPRESSION, sizeof(*cattr)); |
| 567 | if (cattr) |
| 568 | compression = be32toh(cattr->compression); |
Julius Werner | 1de8708 | 2020-12-23 17:38:11 -0800 | [diff] [blame] | 569 | |
Julius Werner | 81dc20e | 2020-10-15 17:37:57 -0700 | [diff] [blame] | 570 | const struct cbfs_file_attr_stageheader *sattr = cbfs_find_attr(&mdata, |
| 571 | CBFS_FILE_ATTR_TAG_STAGEHEADER, sizeof(*sattr)); |
| 572 | if (!sattr) |
| 573 | return CB_ERR; |
| 574 | prog_set_area(pstage, (void *)(uintptr_t)be64toh(sattr->loadaddr), |
| 575 | be32toh(sattr->memlen)); |
| 576 | prog_set_entry(pstage, prog_start(pstage) + |
| 577 | be32toh(sattr->entry_offset), NULL); |
Aaron Durbin | 899d13d | 2015-05-15 23:39:23 -0500 | [diff] [blame] | 578 | |
Aaron Durbin | ed253c8 | 2015-10-07 17:22:42 -0500 | [diff] [blame] | 579 | /* Hacky way to not load programs over read only media. The stages |
| 580 | * that would hit this path initialize themselves. */ |
Julius Werner | 21a4053 | 2020-04-21 16:03:53 -0700 | [diff] [blame] | 581 | if ((ENV_BOOTBLOCK || ENV_SEPARATE_VERSTAGE) && |
| 582 | !CONFIG(NO_XIP_EARLY_STAGES) && CONFIG(BOOT_DEVICE_MEMORY_MAPPED)) { |
Julius Werner | 81dc20e | 2020-10-15 17:37:57 -0700 | [diff] [blame] | 583 | void *mapping = rdev_mmap_full(&rdev); |
Julius Werner | 1de8708 | 2020-12-23 17:38:11 -0800 | [diff] [blame] | 584 | rdev_munmap(&rdev, mapping); |
Julius Werner | 05714cc | 2021-04-15 23:25:44 -0700 | [diff] [blame] | 585 | if (cbfs_file_hash_mismatch(mapping, region_device_sz(&rdev), &mdata, false)) |
Julius Werner | fccf122 | 2021-04-02 15:58:05 -0700 | [diff] [blame] | 586 | return CB_CBFS_HASH_MISMATCH; |
Julius Werner | 81dc20e | 2020-10-15 17:37:57 -0700 | [diff] [blame] | 587 | if (mapping == prog_start(pstage)) |
| 588 | return CB_SUCCESS; |
Aaron Durbin | ed253c8 | 2015-10-07 17:22:42 -0500 | [diff] [blame] | 589 | } |
| 590 | |
Julius Werner | eca99af | 2021-03-10 18:49:37 -0800 | [diff] [blame] | 591 | /* LZ4 stages can be decompressed in-place to save mapping scratch space. Load the |
| 592 | compressed data to the end of the buffer and point &rdev to that memory location. */ |
| 593 | if (cbfs_lz4_enabled() && compression == CBFS_COMPRESS_LZ4) { |
| 594 | size_t in_size = region_device_sz(&rdev); |
| 595 | void *compr_start = prog_start(pstage) + prog_size(pstage) - in_size; |
| 596 | if (rdev_readat(&rdev, compr_start, 0, in_size) != in_size) |
| 597 | return CB_ERR; |
Julius Werner | c893197 | 2021-04-16 16:48:32 -0700 | [diff] [blame] | 598 | rdev_chain_mem(&rdev, compr_start, in_size); |
Julius Werner | eca99af | 2021-03-10 18:49:37 -0800 | [diff] [blame] | 599 | } |
| 600 | |
| 601 | size_t fsize = cbfs_load_and_decompress(&rdev, prog_start(pstage), prog_size(pstage), |
Julius Werner | 05714cc | 2021-04-15 23:25:44 -0700 | [diff] [blame] | 602 | compression, &mdata, false); |
Julius Werner | 09f2921 | 2015-09-29 13:51:35 -0700 | [diff] [blame] | 603 | if (!fsize) |
Julius Werner | 1de8708 | 2020-12-23 17:38:11 -0800 | [diff] [blame] | 604 | return CB_ERR; |
Aaron Durbin | 899d13d | 2015-05-15 23:39:23 -0500 | [diff] [blame] | 605 | |
| 606 | /* Clear area not covered by file. */ |
Julius Werner | 81dc20e | 2020-10-15 17:37:57 -0700 | [diff] [blame] | 607 | memset(prog_start(pstage) + fsize, 0, prog_size(pstage) - fsize); |
Aaron Durbin | 899d13d | 2015-05-15 23:39:23 -0500 | [diff] [blame] | 608 | |
Julius Werner | 81dc20e | 2020-10-15 17:37:57 -0700 | [diff] [blame] | 609 | prog_segment_loaded((uintptr_t)prog_start(pstage), prog_size(pstage), |
| 610 | SEG_FINAL); |
Aaron Durbin | 899d13d | 2015-05-15 23:39:23 -0500 | [diff] [blame] | 611 | |
Julius Werner | 1de8708 | 2020-12-23 17:38:11 -0800 | [diff] [blame] | 612 | return CB_SUCCESS; |
Hung-Te Lin | 6fe0cab | 2013-01-22 18:57:56 +0800 | [diff] [blame] | 613 | } |
Aaron Durbin | 6d720f3 | 2015-12-08 17:00:23 -0600 | [diff] [blame] | 614 | |
Julius Werner | 1e37c9c | 2019-12-11 17:09:39 -0800 | [diff] [blame] | 615 | void cbfs_boot_device_find_mcache(struct cbfs_boot_device *cbd, uint32_t id) |
Aaron Durbin | 6d720f3 | 2015-12-08 17:00:23 -0600 | [diff] [blame] | 616 | { |
Julius Werner | 1e37c9c | 2019-12-11 17:09:39 -0800 | [diff] [blame] | 617 | if (CONFIG(NO_CBFS_MCACHE) || ENV_SMM) |
| 618 | return; |
| 619 | |
Julius Werner | 34cf073 | 2020-12-08 14:21:43 -0800 | [diff] [blame] | 620 | if (cbd->mcache_size) |
| 621 | return; |
| 622 | |
Julius Werner | 1e37c9c | 2019-12-11 17:09:39 -0800 | [diff] [blame] | 623 | const struct cbmem_entry *entry; |
| 624 | if (cbmem_possibly_online() && |
| 625 | (entry = cbmem_entry_find(id))) { |
| 626 | cbd->mcache = cbmem_entry_start(entry); |
| 627 | cbd->mcache_size = cbmem_entry_size(entry); |
| 628 | } else if (ENV_ROMSTAGE_OR_BEFORE) { |
| 629 | u8 *boundary = _ecbfs_mcache - REGION_SIZE(cbfs_mcache) * |
| 630 | CONFIG_CBFS_MCACHE_RW_PERCENTAGE / 100; |
Julius Werner | 723e3b1 | 2020-12-29 17:33:30 -0800 | [diff] [blame] | 631 | boundary = (u8 *)ALIGN_DOWN((uintptr_t)boundary, CBFS_MCACHE_ALIGNMENT); |
Julius Werner | 1e37c9c | 2019-12-11 17:09:39 -0800 | [diff] [blame] | 632 | if (id == CBMEM_ID_CBFS_RO_MCACHE) { |
| 633 | cbd->mcache = _cbfs_mcache; |
| 634 | cbd->mcache_size = boundary - _cbfs_mcache; |
| 635 | } else if (id == CBMEM_ID_CBFS_RW_MCACHE) { |
| 636 | cbd->mcache = boundary; |
| 637 | cbd->mcache_size = _ecbfs_mcache - boundary; |
| 638 | } |
| 639 | } |
Aaron Durbin | 6d720f3 | 2015-12-08 17:00:23 -0600 | [diff] [blame] | 640 | } |
Julius Werner | 1e37c9c | 2019-12-11 17:09:39 -0800 | [diff] [blame] | 641 | |
Julius Werner | fdabf3f | 2020-05-06 17:06:35 -0700 | [diff] [blame] | 642 | cb_err_t cbfs_init_boot_device(const struct cbfs_boot_device *cbd, |
Julius Werner | 723e3b1 | 2020-12-29 17:33:30 -0800 | [diff] [blame] | 643 | struct vb2_hash *mdata_hash) |
Julius Werner | fdabf3f | 2020-05-06 17:06:35 -0700 | [diff] [blame] | 644 | { |
| 645 | /* If we have an mcache, mcache_build() will also check mdata hash. */ |
| 646 | if (!CONFIG(NO_CBFS_MCACHE) && !ENV_SMM && cbd->mcache_size > 0) |
Julius Werner | 723e3b1 | 2020-12-29 17:33:30 -0800 | [diff] [blame] | 647 | return cbfs_mcache_build(&cbd->rdev, cbd->mcache, cbd->mcache_size, mdata_hash); |
Julius Werner | fdabf3f | 2020-05-06 17:06:35 -0700 | [diff] [blame] | 648 | |
| 649 | /* No mcache and no verification means we have nothing special to do. */ |
Julius Werner | 723e3b1 | 2020-12-29 17:33:30 -0800 | [diff] [blame] | 650 | if (!CONFIG(CBFS_VERIFICATION) || !mdata_hash) |
Julius Werner | fdabf3f | 2020-05-06 17:06:35 -0700 | [diff] [blame] | 651 | return CB_SUCCESS; |
| 652 | |
Julius Werner | 723e3b1 | 2020-12-29 17:33:30 -0800 | [diff] [blame] | 653 | /* Verification only: use cbfs_walk() without a walker() function to just run through |
| 654 | the CBFS once, will return NOT_FOUND by default. */ |
| 655 | cb_err_t err = cbfs_walk(&cbd->rdev, NULL, NULL, mdata_hash, 0); |
Julius Werner | fdabf3f | 2020-05-06 17:06:35 -0700 | [diff] [blame] | 656 | if (err == CB_CBFS_NOT_FOUND) |
| 657 | err = CB_SUCCESS; |
| 658 | return err; |
| 659 | } |
| 660 | |
Julius Werner | 1e37c9c | 2019-12-11 17:09:39 -0800 | [diff] [blame] | 661 | const struct cbfs_boot_device *cbfs_get_boot_device(bool force_ro) |
| 662 | { |
| 663 | static struct cbfs_boot_device ro; |
| 664 | |
Julius Werner | 723e3b1 | 2020-12-29 17:33:30 -0800 | [diff] [blame] | 665 | /* Ensure we always init RO mcache, even if the first file is from the RW CBFS. |
Julius Werner | 1e37c9c | 2019-12-11 17:09:39 -0800 | [diff] [blame] | 666 | Otherwise it may not be available when needed in later stages. */ |
| 667 | if (ENV_INITIAL_STAGE && !force_ro && !region_device_sz(&ro.rdev)) |
| 668 | cbfs_get_boot_device(true); |
| 669 | |
| 670 | if (!force_ro) { |
| 671 | const struct cbfs_boot_device *rw = vboot_get_cbfs_boot_device(); |
Julius Werner | 723e3b1 | 2020-12-29 17:33:30 -0800 | [diff] [blame] | 672 | /* This will return NULL if vboot isn't enabled, didn't run yet or decided to |
| 673 | boot into recovery mode. */ |
Julius Werner | 1e37c9c | 2019-12-11 17:09:39 -0800 | [diff] [blame] | 674 | if (rw) |
| 675 | return rw; |
| 676 | } |
| 677 | |
Julius Werner | 723e3b1 | 2020-12-29 17:33:30 -0800 | [diff] [blame] | 678 | /* In rare cases post-RAM stages may run this before cbmem_initialize(), so we can't |
| 679 | lock in the result of find_mcache() on the first try and should keep trying every |
| 680 | time until an mcache is found. */ |
Julius Werner | 34cf073 | 2020-12-08 14:21:43 -0800 | [diff] [blame] | 681 | cbfs_boot_device_find_mcache(&ro, CBMEM_ID_CBFS_RO_MCACHE); |
| 682 | |
Julius Werner | 1e37c9c | 2019-12-11 17:09:39 -0800 | [diff] [blame] | 683 | if (region_device_sz(&ro.rdev)) |
| 684 | return &ro; |
| 685 | |
| 686 | if (fmap_locate_area_as_rdev("COREBOOT", &ro.rdev)) |
Julius Werner | fdabf3f | 2020-05-06 17:06:35 -0700 | [diff] [blame] | 687 | die("Cannot locate primary CBFS"); |
Julius Werner | 1e37c9c | 2019-12-11 17:09:39 -0800 | [diff] [blame] | 688 | |
Julius Werner | fdabf3f | 2020-05-06 17:06:35 -0700 | [diff] [blame] | 689 | if (ENV_INITIAL_STAGE) { |
| 690 | cb_err_t err = cbfs_init_boot_device(&ro, metadata_hash_get()); |
| 691 | if (err == CB_CBFS_HASH_MISMATCH) |
| 692 | die("RO CBFS metadata hash verification failure"); |
| 693 | else if (CONFIG(TOCTOU_SAFETY) && err == CB_CBFS_CACHE_FULL) |
| 694 | die("RO mcache overflow breaks TOCTOU safety!\n"); |
| 695 | else if (err && err != CB_CBFS_CACHE_FULL) |
| 696 | die("RO CBFS initialization error: %d", err); |
Julius Werner | 1e37c9c | 2019-12-11 17:09:39 -0800 | [diff] [blame] | 697 | } |
| 698 | |
| 699 | return &ro; |
| 700 | } |
| 701 | |
| 702 | #if !CONFIG(NO_CBFS_MCACHE) |
| 703 | static void mcache_to_cbmem(const struct cbfs_boot_device *cbd, u32 cbmem_id) |
| 704 | { |
| 705 | if (!cbd) |
| 706 | return; |
| 707 | |
| 708 | size_t real_size = cbfs_mcache_real_size(cbd->mcache, cbd->mcache_size); |
| 709 | void *cbmem_mcache = cbmem_add(cbmem_id, real_size); |
| 710 | if (!cbmem_mcache) { |
| 711 | printk(BIOS_ERR, "ERROR: Cannot allocate CBMEM mcache %#x (%#zx bytes)!\n", |
| 712 | cbmem_id, real_size); |
| 713 | return; |
| 714 | } |
| 715 | memcpy(cbmem_mcache, cbd->mcache, real_size); |
| 716 | } |
| 717 | |
| 718 | static void cbfs_mcache_migrate(int unused) |
| 719 | { |
| 720 | mcache_to_cbmem(vboot_get_cbfs_boot_device(), CBMEM_ID_CBFS_RW_MCACHE); |
| 721 | mcache_to_cbmem(cbfs_get_boot_device(true), CBMEM_ID_CBFS_RO_MCACHE); |
| 722 | } |
| 723 | ROMSTAGE_CBMEM_INIT_HOOK(cbfs_mcache_migrate) |
| 724 | #endif |