1 /*
2 * Copyright (c) 2018 Nordic Semiconductor ASA
3 * Copyright (c) 2015 Runtime Inc
4 *
5 * SPDX-License-Identifier: Apache-2.0
6 */
7
8 #include <zephyr/kernel.h>
9 #include <zephyr/devicetree.h>
10 #include <zephyr/drivers/flash.h>
11
12 #include "target.h"
13
14 #include <flash_map_backend/flash_map_backend.h>
15 #include <sysflash/sysflash.h>
16
17 #include "bootutil/bootutil_log.h"
18
19 BOOT_LOG_MODULE_DECLARE(mcuboot);
20
21 #if (!defined(CONFIG_XTENSA) && DT_HAS_CHOSEN(zephyr_flash_controller))
22 #define FLASH_DEVICE_ID SOC_FLASH_0_ID
23 #define FLASH_DEVICE_BASE CONFIG_FLASH_BASE_ADDRESS
24 #define FLASH_DEVICE_NODE DT_CHOSEN(zephyr_flash_controller)
25
26 #elif (defined(CONFIG_XTENSA) && DT_NODE_EXISTS(DT_INST(0, jedec_spi_nor)))
27 #define FLASH_DEVICE_ID SPI_FLASH_0_ID
28 #define FLASH_DEVICE_BASE 0
29 #define FLASH_DEVICE_NODE DT_INST(0, jedec_spi_nor)
30
31 #elif defined(CONFIG_SOC_FAMILY_ESPRESSIF_ESP32)
32
33 #define FLASH_DEVICE_ID SPI_FLASH_0_ID
34 #define FLASH_DEVICE_BASE 0
35 #define FLASH_DEVICE_NODE DT_CHOSEN(zephyr_flash_controller)
36
37 #else
38 #error "FLASH_DEVICE_ID could not be determined"
39 #endif
40
41 static const struct device *flash_dev = DEVICE_DT_GET(FLASH_DEVICE_NODE);
42
flash_device_base(uint8_t fd_id,uintptr_t * ret)43 int flash_device_base(uint8_t fd_id, uintptr_t *ret)
44 {
45 if (fd_id != FLASH_DEVICE_ID) {
46 BOOT_LOG_ERR("invalid flash ID %d; expected %d",
47 fd_id, FLASH_DEVICE_ID);
48 return -EINVAL;
49 }
50 *ret = FLASH_DEVICE_BASE;
51 return 0;
52 }
53
54 /*
55 * This depends on the mappings defined in sysflash.h.
56 * MCUBoot uses continuous numbering for the primary slot, the secondary slot,
57 * and the scratch while zephyr might number it differently.
58 */
flash_area_id_from_multi_image_slot(int image_index,int slot)59 int flash_area_id_from_multi_image_slot(int image_index, int slot)
60 {
61 switch (slot) {
62 case 0: return FLASH_AREA_IMAGE_PRIMARY(image_index);
63 #if !defined(CONFIG_SINGLE_APPLICATION_SLOT)
64 case 1: return FLASH_AREA_IMAGE_SECONDARY(image_index);
65 #endif
66 #if defined(CONFIG_BOOT_SWAP_USING_SCRATCH)
67 case 2: return FLASH_AREA_IMAGE_SCRATCH;
68 #endif
69 }
70
71 return -EINVAL; /* flash_area_open will fail on that */
72 }
73
flash_area_id_from_image_slot(int slot)74 int flash_area_id_from_image_slot(int slot)
75 {
76 return flash_area_id_from_multi_image_slot(0, slot);
77 }
78
flash_area_id_to_multi_image_slot(int image_index,int area_id)79 int flash_area_id_to_multi_image_slot(int image_index, int area_id)
80 {
81 if (area_id == FLASH_AREA_IMAGE_PRIMARY(image_index)) {
82 return 0;
83 }
84 #if !defined(CONFIG_SINGLE_APPLICATION_SLOT)
85 if (area_id == FLASH_AREA_IMAGE_SECONDARY(image_index)) {
86 return 1;
87 }
88 #endif
89
90 BOOT_LOG_ERR("invalid flash area ID");
91 return -1;
92 }
93
94 #if defined(CONFIG_MCUBOOT_SERIAL_DIRECT_IMAGE_UPLOAD)
flash_area_id_from_direct_image(int image_id)95 int flash_area_id_from_direct_image(int image_id)
96 {
97 switch (image_id) {
98 case 0:
99 case 1:
100 return FIXED_PARTITION_ID(slot0_partition);
101 #if FIXED_PARTITION_EXISTS(slot1_partition)
102 case 2:
103 return FIXED_PARTITION_ID(slot1_partition);
104 #endif
105 #if FIXED_PARTITION_EXISTS(slot2_partition)
106 case 3:
107 return FIXED_PARTITION_ID(slot2_partition);
108 #endif
109 #if FIXED_PARTITION_EXISTS(slot3_partition)
110 case 4:
111 return FIXED_PARTITION_ID(slot3_partition);
112 #endif
113 #if FIXED_PARTITION_EXISTS(slot4_partition)
114 case 5:
115 return FIXED_PARTITION_ID(slot4_partition);
116 #endif
117 #if FIXED_PARTITION_EXISTS(slot5_partition)
118 case 6:
119 return FIXED_PARTITION_ID(slot5_partition);
120 #endif
121 }
122 return -EINVAL;
123 }
124 #endif
125
flash_area_sector_from_off(off_t off,struct flash_sector * sector)126 int flash_area_sector_from_off(off_t off, struct flash_sector *sector)
127 {
128 int rc;
129 struct flash_pages_info page;
130
131 rc = flash_get_page_info_by_offs(flash_dev, off, &page);
132 if (rc) {
133 return rc;
134 }
135
136 sector->fs_off = page.start_offset;
137 sector->fs_size = page.size;
138
139 return rc;
140 }
141
flash_area_get_device_id(const struct flash_area * fa)142 uint8_t flash_area_get_device_id(const struct flash_area *fa)
143 {
144 #if defined(CONFIG_ARM)
145 return fa->fa_id;
146 #else
147 (void)fa;
148 return FLASH_DEVICE_ID;
149 #endif
150 }
151
152 #define ERASED_VAL 0xff
flash_area_erased_val(const struct flash_area * fap)153 __weak uint8_t flash_area_erased_val(const struct flash_area *fap)
154 {
155 (void)fap;
156 return ERASED_VAL;
157 }
158
flash_area_get_sector(const struct flash_area * fap,off_t off,struct flash_sector * fsp)159 int flash_area_get_sector(const struct flash_area *fap, off_t off,
160 struct flash_sector *fsp)
161 {
162 struct flash_pages_info fpi;
163 int rc;
164
165 if (off >= fap->fa_size) {
166 return -ERANGE;
167 }
168
169 rc = flash_get_page_info_by_offs(fap->fa_dev, fap->fa_off + off,
170 &fpi);
171
172 if (rc == 0) {
173 fsp->fs_off = fpi.start_offset - fap->fa_off;
174 fsp->fs_size = fpi.size;
175 }
176
177 return rc;
178 }
179