blob: 4f258b540fc705051f08d29ed141bbda97dc3bcc [file] [log] [blame]
Nico Huber454f6132012-12-10 13:34:10 +00001/*
2 * This file is part of the flashrom project.
3 *
4 * Copyright (C) 2012, 2016 secunet Security Networks AG
5 * (Written by Nico Huber <nico.huber@secunet.com> for secunet)
6 *
Nico Huberd4926fe2026-03-08 21:37:52 +01007 * Copyright (C) 2026 Nico Huber <nico.h@gmx.de>
8 *
Nico Huber454f6132012-12-10 13:34:10 +00009 * This program is free software; you can redistribute it and/or modify
10 * it under the terms of the GNU General Public License as published by
11 * the Free Software Foundation; either version 2 of the License, or
12 * (at your option) any later version.
13 *
14 * This program is distributed in the hope that it will be useful,
15 * but WITHOUT ANY WARRANTY; without even the implied warranty of
16 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
17 * GNU General Public License for more details.
Nico Huber454f6132012-12-10 13:34:10 +000018 */
19/**
20 * @mainpage
21 *
22 * Have a look at the Modules section for a function reference.
23 */
24
Nico Huber70461a92019-06-15 14:56:19 +020025#include <errno.h>
Nico Huber454f6132012-12-10 13:34:10 +000026#include <stdlib.h>
27#include <string.h>
28#include <stdarg.h>
29
Nico Huberd4926fe2026-03-08 21:37:52 +010030#define flashprog_chip flashchip /* For now, we use direct pointers */
31#include "libflashprog.h" /* to the internal struct flashchip. */
32
Nico Huber454f6132012-12-10 13:34:10 +000033#include "flash.h"
Arthur Heymansc82900b2018-01-10 12:48:16 +010034#include "fmap.h"
Nico Huber454f6132012-12-10 13:34:10 +000035#include "programmer.h"
36#include "layout.h"
Nico Huberd4926fe2026-03-08 21:37:52 +010037#include "hwaccess_physmap.h"
Nico Huber305f4172013-06-14 11:55:26 +020038#include "ich_descriptors.h"
Nikolai Artemievda1c8342021-10-21 00:58:12 +110039#include "writeprotect.h"
Nico Huber454f6132012-12-10 13:34:10 +000040
41/**
Nico Huberc3b02dc2023-08-12 01:13:45 +020042 * @defgroup flashprog-general General
Nico Huber454f6132012-12-10 13:34:10 +000043 * @{
44 */
45
46/** Pointer to log callback function. */
Nico Huberc3b02dc2023-08-12 01:13:45 +020047static flashprog_log_callback *global_log_callback = NULL;
Nico Huber454f6132012-12-10 13:34:10 +000048
49/**
Nico Huberc3b02dc2023-08-12 01:13:45 +020050 * @brief Initialize libflashprog.
Nico Huber454f6132012-12-10 13:34:10 +000051 *
52 * @param perform_selfcheck If not zero, perform a self check.
53 * @return 0 on success
54 */
Nico Huberc3b02dc2023-08-12 01:13:45 +020055int flashprog_init(const int perform_selfcheck)
Nico Huber454f6132012-12-10 13:34:10 +000056{
57 if (perform_selfcheck && selfcheck())
58 return 1;
59 myusec_calibrate_delay();
60 return 0;
61}
62
63/**
Nico Huberc3b02dc2023-08-12 01:13:45 +020064 * @brief Shut down libflashprog.
Nico Huber454f6132012-12-10 13:34:10 +000065 * @return 0 on success
66 */
Nico Huberc3b02dc2023-08-12 01:13:45 +020067int flashprog_shutdown(void)
Nico Huber454f6132012-12-10 13:34:10 +000068{
69 return 0; /* TODO: nothing to do? */
70}
71
Nico Huberc3b02dc2023-08-12 01:13:45 +020072/* TODO: flashprog_set_loglevel()? do we need it?
Angel Pons0be623c2021-04-17 17:08:44 +020073 For now, let the user decide in their callback. */
Nico Huber454f6132012-12-10 13:34:10 +000074
75/**
76 * @brief Set the log callback function.
77 *
Nico Huberc3b02dc2023-08-12 01:13:45 +020078 * Set a callback function which will be invoked whenever libflashprog wants
Nico Huber454f6132012-12-10 13:34:10 +000079 * to output messages. This allows frontends to do whatever they see fit with
80 * such messages, e.g. write them to syslog, or to file, or print them in a
81 * GUI window, etc.
82 *
83 * @param log_callback Pointer to the new log callback function.
84 */
Nico Huberc3b02dc2023-08-12 01:13:45 +020085void flashprog_set_log_callback(flashprog_log_callback *const log_callback)
Nico Huber454f6132012-12-10 13:34:10 +000086{
87 global_log_callback = log_callback;
88}
89/** @private */
Nico Huberc3b02dc2023-08-12 01:13:45 +020090int print(const enum flashprog_log_level level, const char *const fmt, ...)
Nico Huber454f6132012-12-10 13:34:10 +000091{
92 if (global_log_callback) {
93 int ret;
94 va_list args;
95 va_start(args, fmt);
Nico Huberd152fb92017-06-19 12:57:10 +020096 ret = global_log_callback(level, fmt, args);
Nico Huber454f6132012-12-10 13:34:10 +000097 va_end(args);
98 return ret;
99 }
100 return 0;
101}
102
Nico Huberc3b02dc2023-08-12 01:13:45 +0200103/** @} */ /* end flashprog-general */
Nico Huber454f6132012-12-10 13:34:10 +0000104
105
106
107/**
Nico Huberc3b02dc2023-08-12 01:13:45 +0200108 * @defgroup flashprog-prog Programmers
Nico Huber454f6132012-12-10 13:34:10 +0000109 * @{
110 */
111
112/**
113 * @brief Initialize the specified programmer.
114 *
115 * Currently, only one programmer may be initialized at a time.
116 *
Nico Huberc3b02dc2023-08-12 01:13:45 +0200117 * @param[out] flashprog Points to a pointer of type struct flashprog_programmer
Nico Huber454f6132012-12-10 13:34:10 +0000118 * that will be set if programmer initialization succeeds.
119 * *flashprog has to be shutdown by the caller with @ref
Nico Huberc3b02dc2023-08-12 01:13:45 +0200120 * flashprog_programmer_shutdown.
Nico Huber454f6132012-12-10 13:34:10 +0000121 * @param[in] prog_name Name of the programmer to initialize.
122 * @param[in] prog_param Pointer to programmer specific parameters.
123 * @return 0 on success
124 */
Nico Huberc3b02dc2023-08-12 01:13:45 +0200125int flashprog_programmer_init(struct flashprog_programmer **const flashprog,
Nico Huber454f6132012-12-10 13:34:10 +0000126 const char *const prog_name, const char *const prog_param)
127{
128 unsigned prog;
129
Thomas Heijligend45cb592021-05-19 14:12:18 +0200130 for (prog = 0; prog < programmer_table_size; prog++) {
Thomas Heijligen633d6db2021-03-31 19:09:44 +0200131 if (strcmp(prog_name, programmer_table[prog]->name) == 0)
Nico Huber454f6132012-12-10 13:34:10 +0000132 break;
133 }
Thomas Heijligend45cb592021-05-19 14:12:18 +0200134 if (prog >= programmer_table_size) {
Nico Huber454f6132012-12-10 13:34:10 +0000135 msg_ginfo("Error: Unknown programmer \"%s\". Valid choices are:\n", prog_name);
136 list_programmers_linebreak(0, 80, 0);
Nico Huberdafd51e2023-02-10 23:58:19 +0100137 msg_ginfo(".\n");
Nico Huber454f6132012-12-10 13:34:10 +0000138 return 1;
139 }
Nico Huber2b66ad92023-01-11 20:15:15 +0100140
141 *flashprog = malloc(sizeof(**flashprog));
142 if (!*flashprog) {
143 msg_gerr("Out of memory!\n");
144 return 1;
145 }
146
147 (*flashprog)->driver = programmer_table[prog];
148 if (prog_param) {
149 (*flashprog)->param = strdup(prog_param);
150 if (!(*flashprog)->param) {
151 msg_gerr("Out of memory!\n");
152 goto _free_err;
153 }
154 } else {
155 (*flashprog)->param = NULL;
156 }
157
158 if (programmer_init(*flashprog))
159 goto _free_err;
160
161 return 0;
162
163_free_err:
Nico Huber74275692024-08-16 13:49:10 +0200164 programmer_shutdown(*flashprog);
Nico Huber2b66ad92023-01-11 20:15:15 +0100165 free((*flashprog)->param);
166 free(*flashprog);
167 return 1;
Nico Huber454f6132012-12-10 13:34:10 +0000168}
169
170/**
171 * @brief Shut down the initialized programmer.
172 *
173 * @param flashprog The programmer to shut down.
174 * @return 0 on success
175 */
Nico Huberc3b02dc2023-08-12 01:13:45 +0200176int flashprog_programmer_shutdown(struct flashprog_programmer *const flashprog)
Nico Huber454f6132012-12-10 13:34:10 +0000177{
Nico Huber2b66ad92023-01-11 20:15:15 +0100178 if (programmer_shutdown(flashprog))
179 return 1;
180 free(flashprog);
181 return 0;
Nico Huber454f6132012-12-10 13:34:10 +0000182}
183
Nico Huberc3b02dc2023-08-12 01:13:45 +0200184/* TODO: flashprog_programmer_capabilities()? */
Nico Huber454f6132012-12-10 13:34:10 +0000185
Nico Huberc3b02dc2023-08-12 01:13:45 +0200186/** @} */ /* end flashprog-prog */
Nico Huber454f6132012-12-10 13:34:10 +0000187
188
189
190/**
Nico Huberc3b02dc2023-08-12 01:13:45 +0200191 * @defgroup flashprog-flash Flash chips
Nico Huber454f6132012-12-10 13:34:10 +0000192 * @{
193 */
194
Nico Huberda16bd42026-03-08 22:00:28 +0100195static int flashprog_flash_probe_any(struct flashprog_flashctx **flashctx,
196 const struct flashprog_programmer *flashprog)
197{
198 struct flashprog_chips *chip_matches;
199 const struct flashprog_chip *chip;
200 bool first = true;
201 int ret;
202
203 if (flashprog_chips_probe(&chip_matches, flashprog))
204 return 1;
205
206 switch (flashprog_chips_count(chip_matches)) {
207 case 0:
208 ret = 2;
209 goto release_matches;
210 case 1:
211 chip = flashprog_chip_first(chip_matches);
212 break;
213 default:
214 msg_cinfo("Multiple flash chip definitions match the detected chip(s): ");
215 for (chip = flashprog_chip_first(chip_matches); chip;
216 chip = flashprog_chip_next(chip), first = false)
217 msg_cinfo("%s\"%s\"", first ? "" : ", ", flashprog_chip_name(chip));
218 msg_cinfo("\n");
219 ret = 3;
220 goto release_matches;
221 }
222
223 ret = flashprog_flash_probe_chip(flashctx, flashprog, chip);
224
225release_matches:
226 flashprog_chips_release(chip_matches);
227 return ret;
228}
229
Nico Huber47412a92026-03-10 23:48:25 +0100230/** @private */
231const struct flashchip *flashprog_chip_by_name(const char *chip_name)
Nico Huberda16bd42026-03-08 22:00:28 +0100232{
233 const struct flashchip *chip;
234 for (chip = flashchips; chip->name; ++chip) {
235 if (!strcmp(chip->name, chip_name))
236 return chip;
237 }
238 msg_cerr("Error: Unknown chip '%s' specified.\n", chip_name);
239 return NULL;
240}
241
Nico Huber454f6132012-12-10 13:34:10 +0000242/**
243 * @brief Probe for a flash chip.
244 *
245 * Probes for a flash chip and returns a flash context, that can be used
Nico Huberc3b02dc2023-08-12 01:13:45 +0200246 * later with flash chip and @ref flashprog-ops "image operations", if
Nico Huber454f6132012-12-10 13:34:10 +0000247 * exactly one matching chip is found.
248 *
Nico Huberc3b02dc2023-08-12 01:13:45 +0200249 * @param[out] flashctx Points to a pointer of type struct flashprog_flashctx
Nico Huber454f6132012-12-10 13:34:10 +0000250 * that will be set if exactly one chip is found. *flashctx
Nico Huberc3b02dc2023-08-12 01:13:45 +0200251 * has to be freed by the caller with @ref flashprog_flash_release.
Nico Huber454f6132012-12-10 13:34:10 +0000252 * @param[in] flashprog The flash programmer used to access the chip.
253 * @param[in] chip_name Name of a chip to probe for, or NULL to probe for
254 * all known chips.
255 * @return 0 on success,
Nico Huberda16bd42026-03-08 22:00:28 +0100256 * 5 if an unknown chip name was provided,
257 * 4 if the chip was detected, but preparation failed,
Nico Huber454f6132012-12-10 13:34:10 +0000258 * 3 if multiple chips were found,
259 * 2 if no chip was found,
260 * or 1 on any other error.
261 */
Nico Huberc3b02dc2023-08-12 01:13:45 +0200262int flashprog_flash_probe(struct flashprog_flashctx **const flashctx,
263 const struct flashprog_programmer *const flashprog,
Nico Huber454f6132012-12-10 13:34:10 +0000264 const char *const chip_name)
265{
Nico Huberda16bd42026-03-08 22:00:28 +0100266 if (!chip_name)
267 return flashprog_flash_probe_any(flashctx, flashprog);
Nico Huber454f6132012-12-10 13:34:10 +0000268
Nico Huberda16bd42026-03-08 22:00:28 +0100269 const struct flashchip *const chip = flashprog_chip_by_name(chip_name);
270 if (!chip)
271 return 5;
Nico Huber454f6132012-12-10 13:34:10 +0000272
Nico Huberda16bd42026-03-08 22:00:28 +0100273 return flashprog_flash_probe_chip(flashctx, flashprog, chip);
Nico Huber454f6132012-12-10 13:34:10 +0000274}
275
Nico Huberd4926fe2026-03-08 21:37:52 +0100276/** @private */
277int flashprog_flash_prepare_context(struct flashprog_flashctx **flashctx,
278 const struct flashprog_programmer *flashprog,
279 const struct master_common *bus,
280 const struct flashchip *chip)
281{
282 if (bus->adapt_voltage) {
283 if (bus->adapt_voltage(bus, chip->voltage.min, chip->voltage.max))
284 return 4;
285 }
286
287 struct flashprog_flashctx *const flash = calloc(1, sizeof(*flash));
288 if (!flash) {
289 msg_gerr("Out of memory!\n");
290 return 1;
291 }
292
293 flash->chip = calloc(1, sizeof(*flash->chip));
294 if (!flash->chip) {
295 msg_gerr("Out of memory!\n");
296 free(flash);
297 return 1;
298 }
299
300 *flash->chip = *chip;
301 flash->mst.common = bus; /* `mst` is a union, so we need only one pointer */
302
Nico Huber517d9372026-03-14 21:48:36 +0100303 if (chip->prepare_access && chip->prepare_access(flash)) {
304 free(flash);
305 return 4;
306 }
Nico Huberd4926fe2026-03-08 21:37:52 +0100307
308 /* Fill default layout covering the whole chip. */
309 if (flashprog_layout_new(&flash->default_layout) ||
310 flashprog_layout_add_region(flash->default_layout,
Nico Huber5421b1b2026-03-14 20:18:56 +0100311 0, flashprog_flash_getsize(flash) - 1, "complete flash") ||
Nico Huber517d9372026-03-14 21:48:36 +0100312 flashprog_layout_include_region(flash->default_layout, "complete flash")) {
313 flashprog_flash_release(flash);
314 return 1;
315 }
Nico Huberd4926fe2026-03-08 21:37:52 +0100316
317 char *const tmp = flashbuses_to_text(flash->chip->bustype);
318 msg_cinfo("Using %s flash chip \"%s\" (%d kB, %s) ",
319 flash->chip->vendor, flash->chip->name, flash->chip->total_size, tmp);
320 free(tmp);
321 if (strcmp(flashprog->driver->name, "internal") == 0 && flash->physical_memory != 0)
322 msg_cinfo("mapped at physical address 0x%0*" PRIxPTR ".\n",
323 PRIxPTR_WIDTH, flash->physical_memory);
324 else
325 msg_cinfo("on %s.\n", flashprog->driver->name);
326
327 if (flash->chip->printlock)
328 flash->chip->printlock(flash);
329
Nico Huberd4926fe2026-03-08 21:37:52 +0100330 *flashctx = flash;
331 return 0;
Nico Huberd4926fe2026-03-08 21:37:52 +0100332}
333
334/**
335 * @brief Probe for a specific flash chip.
336 *
337 * Probes for a flash chip and returns a flash context, that can be used
338 * later with flash chip and @ref flashprog-ops "image operations".
339 *
340 * @param[out] flashctx Points to a pointer of type struct flashprog_flashctx
341 * that will be set. *flashctx has to be freed by the
342 * caller with @ref flashprog_flash_release.
343 * @param[in] flashprog The flash programmer used to access the chip.
344 * @param[in] chip A reference to a chip structure that was previously
345 * fetched by @ref flashprog-chips "enumeration". Its
346 * contents will be copied, so the underlying chip
347 * enumeration can be released directly after the call.
348 * @return 0 on success,
349 * 4 if the chip was detected, but preparation failed,
350 * 3 if multiple chips were found,
351 * 2 if no chip was detected,
352 * or 1 on any other error.
353 */
354int flashprog_flash_probe_chip(struct flashprog_flashctx **flashctx,
355 const struct flashprog_programmer *flashprog,
356 const struct flashprog_chip *chip)
357{
358 const struct master_common *const bus = flashprog_chip_probe(flashprog, chip);
359 if (!bus)
360 return 2;
361
362 return flashprog_flash_prepare_context(flashctx, flashprog, bus, chip);
363}
364
Nico Huber454f6132012-12-10 13:34:10 +0000365/**
366 * @brief Returns the size of the specified flash chip in bytes.
367 *
368 * @param flashctx The queried flash context.
369 * @return Size of flash chip in bytes.
370 */
Nico Huberc3b02dc2023-08-12 01:13:45 +0200371size_t flashprog_flash_getsize(const struct flashprog_flashctx *const flashctx)
Nico Huber454f6132012-12-10 13:34:10 +0000372{
373 return flashctx->chip->total_size * 1024;
374}
375
376/**
377 * @brief Free a flash context.
378 *
379 * @param flashctx Flash context to free.
380 */
Nico Huberc3b02dc2023-08-12 01:13:45 +0200381void flashprog_flash_release(struct flashprog_flashctx *const flashctx)
Nico Huber454f6132012-12-10 13:34:10 +0000382{
Nico Huber6fb2f622022-02-24 18:17:40 +0100383 if (!flashctx)
384 return;
385
Nico Huber517d9372026-03-14 21:48:36 +0100386 if (flashctx->chip->finish_access)
387 flashctx->chip->finish_access(flashctx);
Nico Huberc3b02dc2023-08-12 01:13:45 +0200388 flashprog_layout_release(flashctx->default_layout);
Nico Huber38450ce2019-06-16 20:07:28 +0200389 free(flashctx->chip);
Nico Huber454f6132012-12-10 13:34:10 +0000390 free(flashctx);
391}
392
393/**
Richard Hughes842d6782021-01-15 09:48:12 +0000394 * @brief Set the progress callback function.
395 *
396 * Set a callback function which will be invoked whenever libflashprog wants
397 * to indicate the progress has changed. This allows frontends to do whatever
398 * they see fit with such values, e.g. update a progress bar in a GUI tool.
399 *
400 * @param flashctx Current flash context.
401 * @param progress_callback Pointer to the new progress callback function.
402 * @param user_data Pointer to any data the API user wants to have passed to the callback.
403 */
404void flashprog_set_progress_callback(struct flashprog_flashctx *const flashctx,
405 flashprog_progress_callback *const progress_callback,
406 void *const user_data)
407{
408 flashctx->progress.callback = progress_callback;
409 flashctx->progress.user_data = user_data;
410}
411
412/**
Nico Huber454f6132012-12-10 13:34:10 +0000413 * @brief Set a flag in the given flash context.
414 *
415 * @param flashctx Flash context to alter.
416 * @param flag Flag that is to be set / cleared.
417 * @param value Value to set.
418 */
Nico Huberc3b02dc2023-08-12 01:13:45 +0200419void flashprog_flag_set(struct flashprog_flashctx *const flashctx,
420 const enum flashprog_flag flag, const bool value)
Nico Huber454f6132012-12-10 13:34:10 +0000421{
422 switch (flag) {
Nico Huberc3b02dc2023-08-12 01:13:45 +0200423 case FLASHPROG_FLAG_FORCE: flashctx->flags.force = value; break;
424 case FLASHPROG_FLAG_FORCE_BOARDMISMATCH: flashctx->flags.force_boardmismatch = value; break;
425 case FLASHPROG_FLAG_VERIFY_AFTER_WRITE: flashctx->flags.verify_after_write = value; break;
426 case FLASHPROG_FLAG_VERIFY_WHOLE_CHIP: flashctx->flags.verify_whole_chip = value; break;
Nico Huber55e78842024-07-21 00:46:19 +0200427 case FLASHPROG_FLAG_NON_VOLATILE_WRSR: flashctx->flags.non_volatile_wrsr = value; break;
Nico Huber454f6132012-12-10 13:34:10 +0000428 }
429}
430
431/**
432 * @brief Return the current value of a flag in the given flash context.
433 *
434 * @param flashctx Flash context to read from.
435 * @param flag Flag to be read.
436 * @return Current value of the flag.
437 */
Nico Huberc3b02dc2023-08-12 01:13:45 +0200438bool flashprog_flag_get(const struct flashprog_flashctx *const flashctx, const enum flashprog_flag flag)
Nico Huber454f6132012-12-10 13:34:10 +0000439{
440 switch (flag) {
Nico Huberc3b02dc2023-08-12 01:13:45 +0200441 case FLASHPROG_FLAG_FORCE: return flashctx->flags.force;
442 case FLASHPROG_FLAG_FORCE_BOARDMISMATCH: return flashctx->flags.force_boardmismatch;
443 case FLASHPROG_FLAG_VERIFY_AFTER_WRITE: return flashctx->flags.verify_after_write;
444 case FLASHPROG_FLAG_VERIFY_WHOLE_CHIP: return flashctx->flags.verify_whole_chip;
Nico Huber55e78842024-07-21 00:46:19 +0200445 case FLASHPROG_FLAG_NON_VOLATILE_WRSR: return flashctx->flags.non_volatile_wrsr;
Nico Huberc3b02dc2023-08-12 01:13:45 +0200446 default: return false;
Nico Huber454f6132012-12-10 13:34:10 +0000447 }
448}
449
Nico Huberc3b02dc2023-08-12 01:13:45 +0200450/** @} */ /* end flashprog-flash */
Nico Huber454f6132012-12-10 13:34:10 +0000451
452
453
454/**
Nico Huberc3b02dc2023-08-12 01:13:45 +0200455 * @defgroup flashprog-layout Layout handling
Nico Huber454f6132012-12-10 13:34:10 +0000456 * @{
457 */
458
Nico Huberdb90cf72023-01-24 23:35:52 +0100459#ifdef __FLASHPROG_LITTLE_ENDIAN__
460static int layout_cmp(const struct romentry *const a, const struct romentry *const b)
461{
462 int ret;
463
464 ret = (int)a->start - (int)b->start;
465 if (ret)
466 return ret;
467
468 ret = (int)a->end - (int)b->end;
469 if (ret)
470 return ret;
471
472 return strcmp(a->name, b->name);
473}
474#endif
475
Nico Huber454f6132012-12-10 13:34:10 +0000476/**
Nico Huber305f4172013-06-14 11:55:26 +0200477 * @brief Read a layout from the Intel ICH descriptor in the flash.
478 *
479 * Optionally verify that the layout matches the one in the given
480 * descriptor dump.
481 *
Nico Huberc3b02dc2023-08-12 01:13:45 +0200482 * @param[out] layout Points to a struct flashprog_layout pointer that
Nico Huber305f4172013-06-14 11:55:26 +0200483 * gets set if the descriptor is read and parsed
484 * successfully.
485 * @param[in] flashctx Flash context to read the descriptor from flash.
486 * @param[in] dump The descriptor dump to compare to or NULL.
487 * @param[in] len The length of the descriptor dump.
488 *
489 * @return 0 on success,
490 * 6 if descriptor parsing isn't implemented for the host,
491 * 5 if the descriptors don't match,
492 * 4 if the descriptor dump couldn't be parsed,
493 * 3 if the descriptor on flash couldn't be parsed,
494 * 2 if the descriptor on flash couldn't be read,
495 * 1 on any other error.
496 */
Nico Huberc3b02dc2023-08-12 01:13:45 +0200497int flashprog_layout_read_from_ifd(struct flashprog_layout **const layout, struct flashctx *const flashctx,
Nico Huber305f4172013-06-14 11:55:26 +0200498 const void *const dump, const size_t len)
499{
Nico Huberc3b02dc2023-08-12 01:13:45 +0200500#ifndef __FLASHPROG_LITTLE_ENDIAN__
Nico Huber305f4172013-06-14 11:55:26 +0200501 return 6;
502#else
Nico Huberc3b02dc2023-08-12 01:13:45 +0200503 struct flashprog_layout *dump_layout = NULL, *chip_layout = NULL;
Nico Huber305f4172013-06-14 11:55:26 +0200504 int ret = 1;
505
506 void *const desc = malloc(0x1000);
Nico Huber305f4172013-06-14 11:55:26 +0200507 if (prepare_flash_access(flashctx, true, false, false, false))
508 goto _free_ret;
509
510 msg_cinfo("Reading ich descriptor... ");
Richard Hughes842d6782021-01-15 09:48:12 +0000511 if (flashprog_read_range(flashctx, desc, 0, 0x1000)) {
Nico Huber305f4172013-06-14 11:55:26 +0200512 msg_cerr("Read operation failed!\n");
513 msg_cinfo("FAILED.\n");
514 ret = 2;
515 goto _finalize_ret;
516 }
517 msg_cinfo("done.\n");
518
Nico Huber5bd990c2019-06-16 19:46:46 +0200519 if (layout_from_ich_descriptors(&chip_layout, desc, 0x1000)) {
Patrick Rudolph911b8d82019-06-06 11:23:55 +0200520 msg_cerr("Couldn't parse the descriptor!\n");
Nico Huber305f4172013-06-14 11:55:26 +0200521 ret = 3;
522 goto _finalize_ret;
523 }
524
525 if (dump) {
526 if (layout_from_ich_descriptors(&dump_layout, dump, len)) {
Patrick Rudolph911b8d82019-06-06 11:23:55 +0200527 msg_cerr("Couldn't parse the descriptor!\n");
Nico Huber305f4172013-06-14 11:55:26 +0200528 ret = 4;
529 goto _finalize_ret;
530 }
531
Nico Huber5bd990c2019-06-16 19:46:46 +0200532 const struct romentry *chip_entry = layout_next(chip_layout, NULL);
533 const struct romentry *dump_entry = layout_next(dump_layout, NULL);
Nico Huberdb90cf72023-01-24 23:35:52 +0100534 while (chip_entry && dump_entry && !layout_cmp(chip_entry, dump_entry)) {
Nico Huber5bd990c2019-06-16 19:46:46 +0200535 chip_entry = layout_next(chip_layout, chip_entry);
536 dump_entry = layout_next(dump_layout, dump_entry);
Nico Huber354766b2019-06-16 19:28:35 +0200537 }
Nico Huberc3b02dc2023-08-12 01:13:45 +0200538 flashprog_layout_release(dump_layout);
Nico Huber354766b2019-06-16 19:28:35 +0200539 if (chip_entry || dump_entry) {
Patrick Rudolph911b8d82019-06-06 11:23:55 +0200540 msg_cerr("Descriptors don't match!\n");
Nico Huber305f4172013-06-14 11:55:26 +0200541 ret = 5;
542 goto _finalize_ret;
543 }
544 }
545
Nico Huberc3b02dc2023-08-12 01:13:45 +0200546 *layout = (struct flashprog_layout *)chip_layout;
Nico Huber305f4172013-06-14 11:55:26 +0200547 ret = 0;
548
549_finalize_ret:
550 finalize_flash_access(flashctx);
551_free_ret:
552 if (ret)
Nico Huberc3b02dc2023-08-12 01:13:45 +0200553 flashprog_layout_release(chip_layout);
Nico Huber305f4172013-06-14 11:55:26 +0200554 free(desc);
555 return ret;
556#endif
557}
558
Nico Huberc3b02dc2023-08-12 01:13:45 +0200559#ifdef __FLASHPROG_LITTLE_ENDIAN__
560static int flashprog_layout_parse_fmap(struct flashprog_layout **layout,
Arthur Heymansc82900b2018-01-10 12:48:16 +0100561 struct flashctx *const flashctx, const struct fmap *const fmap)
562{
563 int i;
Nico Huber92e0b622019-06-15 15:55:11 +0200564 char name[FMAP_STRLEN + 1];
565 const struct fmap_area *area;
Nico Huberc3b02dc2023-08-12 01:13:45 +0200566 struct flashprog_layout *l;
Arthur Heymansc82900b2018-01-10 12:48:16 +0100567
Nico Huberc3b02dc2023-08-12 01:13:45 +0200568 if (!fmap || flashprog_layout_new(&l))
Arthur Heymansc82900b2018-01-10 12:48:16 +0100569 return 1;
570
Nico Huber92e0b622019-06-15 15:55:11 +0200571 for (i = 0, area = fmap->areas; i < fmap->nareas; i++, area++) {
572 snprintf(name, sizeof(name), "%s", area->name);
Nico Huberc3b02dc2023-08-12 01:13:45 +0200573 if (flashprog_layout_add_region(l, area->offset, area->offset + area->size - 1, name)) {
574 flashprog_layout_release(l);
Nico Huber70461a92019-06-15 14:56:19 +0200575 return 1;
Nico Huberefe96a92021-05-14 00:39:24 +0200576 }
Arthur Heymansc82900b2018-01-10 12:48:16 +0100577 }
578
579 *layout = l;
580 return 0;
581}
Nico Huberc3b02dc2023-08-12 01:13:45 +0200582#endif /* __FLASHPROG_LITTLE_ENDIAN__ */
Arthur Heymansc82900b2018-01-10 12:48:16 +0100583
584/**
585 * @brief Read a layout by searching the flash chip for fmap.
586 *
Nico Huberc3b02dc2023-08-12 01:13:45 +0200587 * @param[out] layout Points to a struct flashprog_layout pointer that
Arthur Heymansc82900b2018-01-10 12:48:16 +0100588 * gets set if the fmap is read and parsed successfully.
589 * @param[in] flashctx Flash context
590 * @param[in] offset Offset to begin searching for fmap.
591 * @param[in] offset Length of address space to search.
592 *
593 * @return 0 on success,
594 * 3 if fmap parsing isn't implemented for the host,
595 * 2 if the fmap couldn't be read,
596 * 1 on any other error.
597 */
Nico Huberc3b02dc2023-08-12 01:13:45 +0200598int flashprog_layout_read_fmap_from_rom(struct flashprog_layout **const layout,
Julius Werner8f0db7d2022-02-14 17:07:39 -0800599 struct flashctx *const flashctx, size_t offset, size_t len)
Arthur Heymansc82900b2018-01-10 12:48:16 +0100600{
Nico Huberc3b02dc2023-08-12 01:13:45 +0200601#ifndef __FLASHPROG_LITTLE_ENDIAN__
Arthur Heymansc82900b2018-01-10 12:48:16 +0100602 return 3;
603#else
604 struct fmap *fmap = NULL;
605 int ret = 0;
606
607 msg_gdbg("Attempting to read fmap from ROM content.\n");
608 if (fmap_read_from_rom(&fmap, flashctx, offset, len)) {
609 msg_gerr("Failed to read fmap from ROM.\n");
610 return 1;
611 }
612
613 msg_gdbg("Adding fmap layout to global layout.\n");
Nico Huberc3b02dc2023-08-12 01:13:45 +0200614 if (flashprog_layout_parse_fmap(layout, flashctx, fmap)) {
Arthur Heymansc82900b2018-01-10 12:48:16 +0100615 msg_gerr("Failed to add fmap regions to layout.\n");
616 ret = 1;
617 }
618
619 free(fmap);
620 return ret;
621#endif
622}
623
624/**
625 * @brief Read a layout by searching buffer for fmap.
626 *
Nico Huberc3b02dc2023-08-12 01:13:45 +0200627 * @param[out] layout Points to a struct flashprog_layout pointer that
Arthur Heymansc82900b2018-01-10 12:48:16 +0100628 * gets set if the fmap is read and parsed successfully.
629 * @param[in] flashctx Flash context
630 * @param[in] buffer Buffer to search in
631 * @param[in] size Size of buffer to search
632 *
633 * @return 0 on success,
634 * 3 if fmap parsing isn't implemented for the host,
635 * 2 if the fmap couldn't be read,
636 * 1 on any other error.
637 */
Nico Huberc3b02dc2023-08-12 01:13:45 +0200638int flashprog_layout_read_fmap_from_buffer(struct flashprog_layout **const layout,
Arthur Heymansc82900b2018-01-10 12:48:16 +0100639 struct flashctx *const flashctx, const uint8_t *const buf, size_t size)
640{
Nico Huberc3b02dc2023-08-12 01:13:45 +0200641#ifndef __FLASHPROG_LITTLE_ENDIAN__
Arthur Heymansc82900b2018-01-10 12:48:16 +0100642 return 3;
643#else
644 struct fmap *fmap = NULL;
645 int ret = 1;
646
647 if (!buf || !size)
648 goto _ret;
649
650 msg_gdbg("Attempting to read fmap from buffer.\n");
651 if (fmap_read_from_buffer(&fmap, buf, size)) {
652 msg_gerr("Failed to read fmap from buffer.\n");
653 goto _ret;
654 }
655
656 msg_gdbg("Adding fmap layout to global layout.\n");
Nico Huberc3b02dc2023-08-12 01:13:45 +0200657 if (flashprog_layout_parse_fmap(layout, flashctx, fmap)) {
Arthur Heymansc82900b2018-01-10 12:48:16 +0100658 msg_gerr("Failed to add fmap regions to layout.\n");
659 goto _free_ret;
660 }
661
662 ret = 0;
663_free_ret:
664 free(fmap);
665_ret:
666 return ret;
667#endif
668}
669
Nico Huber305f4172013-06-14 11:55:26 +0200670/**
Nico Huber454f6132012-12-10 13:34:10 +0000671 * @brief Set the active layout for a flash context.
672 *
673 * Note: This just sets a pointer. The caller must not release the layout
674 * as long as he uses it through the given flash context.
675 *
676 * @param flashctx Flash context whose layout will be set.
677 * @param layout Layout to bet set.
678 */
Nico Huberc3b02dc2023-08-12 01:13:45 +0200679void flashprog_layout_set(struct flashprog_flashctx *const flashctx, const struct flashprog_layout *const layout)
Nico Huber454f6132012-12-10 13:34:10 +0000680{
681 flashctx->layout = layout;
682}
683
Nico Huberc3b02dc2023-08-12 01:13:45 +0200684/** @} */ /* end flashprog-layout */
Nikolai Artemievda1c8342021-10-21 00:58:12 +1100685
686
687/**
Nico Huberfd0fa752026-06-17 21:10:59 +0200688 * @defgroup flashprog-wp Write Protection
Nikolai Artemievda1c8342021-10-21 00:58:12 +1100689 * @{
690 */
691
692/**
693 * @brief Create a new empty WP configuration.
694 *
Nico Huberc3b02dc2023-08-12 01:13:45 +0200695 * @param[out] cfg Points to a pointer of type struct flashprog_wp_cfg that will
Nikolai Artemievda1c8342021-10-21 00:58:12 +1100696 * be set if creation succeeds. *cfg has to be freed by the
Nico Huberc3b02dc2023-08-12 01:13:45 +0200697 * caller with @ref flashprog_wp_cfg_release.
Nikolai Artemievda1c8342021-10-21 00:58:12 +1100698 * @return 0 on success
699 * >0 on failure
700 */
Nico Huberc3b02dc2023-08-12 01:13:45 +0200701enum flashprog_wp_result flashprog_wp_cfg_new(struct flashprog_wp_cfg **cfg)
Nikolai Artemievda1c8342021-10-21 00:58:12 +1100702{
703 *cfg = calloc(1, sizeof(**cfg));
Nico Huberc3b02dc2023-08-12 01:13:45 +0200704 return *cfg ? 0 : FLASHPROG_WP_ERR_OTHER;
Nikolai Artemievda1c8342021-10-21 00:58:12 +1100705}
706
707/**
708 * @brief Free a WP configuration.
709 *
Nico Huberc3b02dc2023-08-12 01:13:45 +0200710 * @param[out] cfg Pointer to the flashprog_wp_cfg to free.
Nikolai Artemievda1c8342021-10-21 00:58:12 +1100711 */
Nico Huberc3b02dc2023-08-12 01:13:45 +0200712void flashprog_wp_cfg_release(struct flashprog_wp_cfg *cfg)
Nikolai Artemievda1c8342021-10-21 00:58:12 +1100713{
714 free(cfg);
715}
716
717/**
718 * @brief Set the protection mode for a WP configuration.
719 *
720 * @param[in] mode The protection mode to set.
Nico Huberc3b02dc2023-08-12 01:13:45 +0200721 * @param[out] cfg Pointer to the flashprog_wp_cfg structure to modify.
Nikolai Artemievda1c8342021-10-21 00:58:12 +1100722 */
Nico Huberc3b02dc2023-08-12 01:13:45 +0200723void flashprog_wp_set_mode(struct flashprog_wp_cfg *cfg, enum flashprog_wp_mode mode)
Nikolai Artemievda1c8342021-10-21 00:58:12 +1100724{
725 cfg->mode = mode;
726}
727
728/**
729 * @brief Get the protection mode from a WP configuration.
730 *
731 * @param[in] cfg The WP configuration to get the protection mode from.
732 * @return The configuration's protection mode.
733 */
Nico Huberc3b02dc2023-08-12 01:13:45 +0200734enum flashprog_wp_mode flashprog_wp_get_mode(const struct flashprog_wp_cfg *cfg)
Nikolai Artemievda1c8342021-10-21 00:58:12 +1100735{
736 return cfg->mode;
737}
738
739/**
740 * @brief Set the protection range for a WP configuration.
741 *
Nico Huberc3b02dc2023-08-12 01:13:45 +0200742 * @param[out] cfg Pointer to the flashprog_wp_cfg structure to modify.
Nikolai Artemievda1c8342021-10-21 00:58:12 +1100743 * @param[in] start The range's start address.
744 * @param[in] len The range's length.
745 */
Nico Huberc3b02dc2023-08-12 01:13:45 +0200746void flashprog_wp_set_range(struct flashprog_wp_cfg *cfg, size_t start, size_t len)
Nikolai Artemievda1c8342021-10-21 00:58:12 +1100747{
748 cfg->range.start = start;
749 cfg->range.len = len;
750}
751
752/**
753 * @brief Get the protection range from a WP configuration.
754 *
755 * @param[out] start Points to a size_t to write the range start to.
756 * @param[out] len Points to a size_t to write the range length to.
757 * @param[in] cfg The WP configuration to get the range from.
758 */
Nico Huberc3b02dc2023-08-12 01:13:45 +0200759void flashprog_wp_get_range(size_t *start, size_t *len, const struct flashprog_wp_cfg *cfg)
Nikolai Artemievda1c8342021-10-21 00:58:12 +1100760{
761 *start = cfg->range.start;
762 *len = cfg->range.len;
763}
764
765/**
766 * @brief Write a WP configuration to a flash chip.
767 *
768 * @param[in] flash The flash context used to access the chip.
769 * @param[in] cfg The WP configuration to write to the chip.
770 * @return 0 on success
771 * >0 on failure
772 */
Nico Huberc3b02dc2023-08-12 01:13:45 +0200773enum flashprog_wp_result flashprog_wp_write_cfg(struct flashctx *flash, const struct flashprog_wp_cfg *cfg)
Nikolai Artemievda1c8342021-10-21 00:58:12 +1100774{
Nico Huberaabb3e02023-01-13 00:22:30 +0100775 if (!flash->chip->wp_write_cfg)
776 return FLASHPROG_WP_ERR_CHIP_UNSUPPORTED;
Nikolai Artemievda1c8342021-10-21 00:58:12 +1100777
Nico Huberaabb3e02023-01-13 00:22:30 +0100778 return flash->chip->wp_write_cfg(flash, cfg);
Nikolai Artemievda1c8342021-10-21 00:58:12 +1100779}
780
781/**
782 * @brief Read the current WP configuration from a flash chip.
783 *
Nico Huberc3b02dc2023-08-12 01:13:45 +0200784 * @param[out] cfg Pointer to a struct flashprog_wp_cfg to store the chip's
Nikolai Artemievda1c8342021-10-21 00:58:12 +1100785 * configuration in.
786 * @param[in] flash The flash context used to access the chip.
787 * @return 0 on success
788 * >0 on failure
789 */
Nico Huberc3b02dc2023-08-12 01:13:45 +0200790enum flashprog_wp_result flashprog_wp_read_cfg(struct flashprog_wp_cfg *cfg, struct flashctx *flash)
Nikolai Artemievda1c8342021-10-21 00:58:12 +1100791{
Nico Huberaabb3e02023-01-13 00:22:30 +0100792 if (!flash->chip->wp_read_cfg)
793 return FLASHPROG_WP_ERR_CHIP_UNSUPPORTED;
Nikolai Artemievda1c8342021-10-21 00:58:12 +1100794
Nico Huberaabb3e02023-01-13 00:22:30 +0100795 return flash->chip->wp_read_cfg(cfg, flash);
Nikolai Artemievda1c8342021-10-21 00:58:12 +1100796}
797
Nikolai Artemiev077c0d12021-10-21 01:50:15 +1100798/**
799 * @brief Get a list of protection ranges supported by the flash chip.
800 *
Nico Huberc3b02dc2023-08-12 01:13:45 +0200801 * @param[out] ranges Points to a pointer of type struct flashprog_wp_ranges
Nikolai Artemiev077c0d12021-10-21 01:50:15 +1100802 * that will be set if available ranges are found. Finding
803 * available ranges may not always be possible, even if the
804 * chip's protection range can be read or modified. *ranges
Nico Huberc3b02dc2023-08-12 01:13:45 +0200805 * must be freed using @ref flashprog_wp_ranges_free.
Nikolai Artemiev077c0d12021-10-21 01:50:15 +1100806 * @param[in] flash The flash context used to access the chip.
807 * @return 0 on success
808 * >0 on failure
809 */
Nico Huberc3b02dc2023-08-12 01:13:45 +0200810enum flashprog_wp_result flashprog_wp_get_available_ranges(struct flashprog_wp_ranges **list, struct flashprog_flashctx *flash)
Nikolai Artemiev077c0d12021-10-21 01:50:15 +1100811{
Nico Huberaabb3e02023-01-13 00:22:30 +0100812 if (!flash->chip->wp_get_ranges)
813 return FLASHPROG_WP_ERR_CHIP_UNSUPPORTED;
Nikolai Artemiev077c0d12021-10-21 01:50:15 +1100814
Nico Huberaabb3e02023-01-13 00:22:30 +0100815 return flash->chip->wp_get_ranges(list, flash);
Nikolai Artemiev077c0d12021-10-21 01:50:15 +1100816}
817
818/**
819 * @brief Get a number of protection ranges in a range list.
820 *
821 * @param[in] ranges The range list to get the count from.
822 * @return Number of ranges in the list.
823 */
Nico Huberc3b02dc2023-08-12 01:13:45 +0200824size_t flashprog_wp_ranges_get_count(const struct flashprog_wp_ranges *list)
Nikolai Artemiev077c0d12021-10-21 01:50:15 +1100825{
826 return list->count;
827}
828
829/**
830 * @brief Get a protection range from a range list.
831 *
832 * @param[out] start Points to a size_t to write the range's start to.
833 * @param[out] len Points to a size_t to write the range's length to.
834 * @param[in] ranges The range list to get the range from.
835 * @param[in] index Index of the range to get.
836 * @return 0 on success
837 * >0 on failure
838 */
Nico Huberc3b02dc2023-08-12 01:13:45 +0200839enum flashprog_wp_result flashprog_wp_ranges_get_range(size_t *start, size_t *len, const struct flashprog_wp_ranges *list, unsigned int index)
Nikolai Artemiev077c0d12021-10-21 01:50:15 +1100840{
841 if (index >= list->count)
Nico Huberc3b02dc2023-08-12 01:13:45 +0200842 return FLASHPROG_WP_ERR_OTHER;
Nikolai Artemiev077c0d12021-10-21 01:50:15 +1100843
844 *start = list->ranges[index].start;
845 *len = list->ranges[index].len;
846
847 return 0;
848}
849
850/**
851 * @brief Free a WP range list.
852 *
Nico Huberc3b02dc2023-08-12 01:13:45 +0200853 * @param[out] cfg Pointer to the flashprog_wp_ranges to free.
Nikolai Artemiev077c0d12021-10-21 01:50:15 +1100854 */
Nico Huberc3b02dc2023-08-12 01:13:45 +0200855void flashprog_wp_ranges_release(struct flashprog_wp_ranges *list)
Nikolai Artemiev077c0d12021-10-21 01:50:15 +1100856{
857 if (!list)
858 return;
859
860 free(list->ranges);
861 free(list);
862}
863
864
Nico Huberc3b02dc2023-08-12 01:13:45 +0200865/** @} */ /* end flashprog-wp */