| /* |
| * This file is part of the coreboot project. |
| * |
| * Copyright (C) 2014 The ChromiumOS Authors. All rights reserved. |
| * |
| * 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 <assert.h> |
| #include <cbfs.h> |
| #include <cbmem.h> |
| #include <console/console.h> |
| #include <reset.h> |
| #include <string.h> |
| #include <vb2_api.h> |
| #include <vboot/misc.h> |
| #include <vboot/symbols.h> |
| #include <vboot/vboot_common.h> |
| |
| struct selected_region { |
| uint32_t offset; |
| uint32_t size; |
| }; |
| |
| /* |
| * this is placed at the start of the vboot work buffer. selected_region is used |
| * for the verstage to return the location of the selected slot. buffer is used |
| * by the vboot2 core. Keep the struct CPU architecture agnostic as it crosses |
| * stage boundaries. |
| */ |
| struct vb2_working_data { |
| struct selected_region selected_region; |
| /* offset of the buffer from the start of this struct */ |
| uint32_t buffer_offset; |
| uint32_t buffer_size; |
| }; |
| |
| static const size_t vb_work_buf_size = 16 * KiB; |
| |
| static struct vb2_working_data * const vboot_get_working_data(void) |
| { |
| if (IS_ENABLED(CONFIG_VBOOT_DYNAMIC_WORK_BUFFER)) |
| /* cbmem_add() does a cbmem_find() first. */ |
| return cbmem_add(CBMEM_ID_VBOOT_WORKBUF, vb_work_buf_size); |
| else |
| return (struct vb2_working_data *)_vboot2_work; |
| } |
| |
| static size_t vb2_working_data_size(void) |
| { |
| if (IS_ENABLED(CONFIG_VBOOT_DYNAMIC_WORK_BUFFER)) |
| return vb_work_buf_size; |
| else |
| return _vboot2_work_size; |
| } |
| |
| static struct selected_region *vb2_selected_region(void) |
| { |
| struct selected_region *sel_reg = NULL; |
| |
| /* Ramstage and postcar always uses cbmem as a source of truth. */ |
| if (ENV_RAMSTAGE || ENV_POSTCAR) |
| sel_reg = cbmem_find(CBMEM_ID_VBOOT_SEL_REG); |
| else if (ENV_ROMSTAGE) { |
| /* Try cbmem first. Fall back on working data if not found. */ |
| sel_reg = cbmem_find(CBMEM_ID_VBOOT_SEL_REG); |
| |
| if (sel_reg == NULL) { |
| struct vb2_working_data *wd = vboot_get_working_data(); |
| sel_reg = &wd->selected_region; |
| } |
| } else { |
| /* Stages such as bootblock and verstage use working data. */ |
| struct vb2_working_data *wd = vboot_get_working_data(); |
| sel_reg = &wd->selected_region; |
| } |
| |
| return sel_reg; |
| } |
| |
| void vb2_init_work_context(struct vb2_context *ctx) |
| { |
| struct vb2_working_data *wd; |
| size_t work_size; |
| |
| /* First initialize the working data region. */ |
| work_size = vb2_working_data_size(); |
| wd = vboot_get_working_data(); |
| memset(wd, 0, work_size); |
| |
| /* |
| * vboot prefers 16-byte alignment. This takes away 16 bytes |
| * from the VBOOT2_WORK region, but the vboot devs said that's okay. |
| */ |
| wd->buffer_offset = ALIGN_UP(sizeof(*wd), 16); |
| wd->buffer_size = work_size - wd->buffer_offset; |
| |
| /* Initialize the vb2_context. */ |
| memset(ctx, 0, sizeof(*ctx)); |
| ctx->workbuf = (void *)vb2_get_shared_data(); |
| ctx->workbuf_size = wd->buffer_size; |
| |
| } |
| |
| struct vb2_shared_data *vb2_get_shared_data(void) |
| { |
| struct vb2_working_data *wd = vboot_get_working_data(); |
| return (void *)((uintptr_t)wd + wd->buffer_offset); |
| } |
| |
| int vb2_get_selected_region(struct region *region) |
| { |
| const struct selected_region *reg = vb2_selected_region(); |
| |
| if (reg == NULL) |
| return -1; |
| |
| if (reg->offset == 0 && reg->size == 0) |
| return -1; |
| |
| region->offset = reg->offset; |
| region->size = reg->size; |
| |
| return 0; |
| } |
| |
| void vb2_set_selected_region(const struct region *region) |
| { |
| struct selected_region *reg = vb2_selected_region(); |
| |
| assert(reg != NULL); |
| |
| reg->offset = region_offset(region); |
| reg->size = region_sz(region); |
| } |
| |
| int vb2_is_slot_selected(void) |
| { |
| const struct selected_region *reg = vb2_selected_region(); |
| |
| assert(reg != NULL); |
| |
| return reg->size > 0; |
| } |
| |
| void vb2_store_selected_region(void) |
| { |
| const struct vb2_working_data *wd; |
| struct selected_region *sel_reg; |
| |
| /* Always use the working data in this path since it's the object |
| * which has the result.. */ |
| wd = vboot_get_working_data(); |
| |
| sel_reg = cbmem_add(CBMEM_ID_VBOOT_SEL_REG, sizeof(*sel_reg)); |
| |
| assert(sel_reg != NULL); |
| |
| sel_reg->offset = wd->selected_region.offset; |
| sel_reg->size = wd->selected_region.size; |
| } |
| |
| /* |
| * For platforms that employ VBOOT_DYNAMIC_WORK_BUFFER, the vboot |
| * verification doesn't happen until after cbmem is brought online. |
| * Therefore, the selected region contents would not be initialized |
| * so don't automatically add results when cbmem comes online. |
| */ |
| #if !IS_ENABLED(CONFIG_VBOOT_DYNAMIC_WORK_BUFFER) |
| static void vb2_store_selected_region_cbmem(int unused) |
| { |
| vb2_store_selected_region(); |
| } |
| ROMSTAGE_CBMEM_INIT_HOOK(vb2_store_selected_region_cbmem) |
| #endif |