blob: 64ec4fde493f4cfaa12d610beaf1ebdba644242e [file] [log] [blame]
Simon Glass017656e2022-04-24 23:31:07 -06001// SPDX-License-Identifier: GPL-2.0+
2/*
3 * Copyright 2021 Google LLC
4 * Written by Simon Glass <sjg@chromium.org>
5 */
6
7#define LOG_CATEGORY UCLASS_BOOTSTD
8
Simon Glass017656e2022-04-24 23:31:07 -06009#include <dm.h>
10#include <bootdev.h>
11#include <bootflow.h>
Simon Glass03fcbf92022-04-24 23:31:09 -060012#include <bootmeth.h>
Simon Glass017656e2022-04-24 23:31:07 -060013#include <bootstd.h>
Simon Glass017656e2022-04-24 23:31:07 -060014#include <fs.h>
15#include <log.h>
16#include <malloc.h>
17#include <part.h>
18#include <sort.h>
19#include <dm/device-internal.h>
20#include <dm/lists.h>
21#include <dm/uclass-internal.h>
22
23enum {
24 /*
25 * Set some sort of limit on the number of partitions a bootdev can
26 * have. Note that for disks this limits the partitions numbers that
27 * are scanned to 1..MAX_BOOTFLOWS_PER_BOOTDEV
28 */
29 MAX_PART_PER_BOOTDEV = 30,
30
31 /* Maximum supported length of the "boot_targets" env string */
32 BOOT_TARGETS_MAX_LEN = 100,
33};
34
35int bootdev_add_bootflow(struct bootflow *bflow)
36{
Simon Glass017656e2022-04-24 23:31:07 -060037 struct bootstd_priv *std;
38 struct bootflow *new;
39 int ret;
40
Simon Glass017656e2022-04-24 23:31:07 -060041 ret = bootstd_get_priv(&std);
42 if (ret)
43 return ret;
44
45 new = malloc(sizeof(*bflow));
46 if (!new)
47 return log_msg_ret("bflow", -ENOMEM);
48 memcpy(new, bflow, sizeof(*bflow));
49
50 list_add_tail(&new->glob_node, &std->glob_head);
Simon Glassdee4d642022-07-30 15:52:23 -060051 if (bflow->dev) {
52 struct bootdev_uc_plat *ucp = dev_get_uclass_plat(bflow->dev);
53
54 list_add_tail(&new->bm_node, &ucp->bootflow_head);
55 }
Simon Glass017656e2022-04-24 23:31:07 -060056
57 return 0;
58}
59
60int bootdev_first_bootflow(struct udevice *dev, struct bootflow **bflowp)
61{
62 struct bootdev_uc_plat *ucp = dev_get_uclass_plat(dev);
63
64 if (list_empty(&ucp->bootflow_head))
65 return -ENOENT;
66
67 *bflowp = list_first_entry(&ucp->bootflow_head, struct bootflow,
68 bm_node);
69
70 return 0;
71}
72
73int bootdev_next_bootflow(struct bootflow **bflowp)
74{
75 struct bootflow *bflow = *bflowp;
76 struct bootdev_uc_plat *ucp = dev_get_uclass_plat(bflow->dev);
77
78 *bflowp = NULL;
79
80 if (list_is_last(&bflow->bm_node, &ucp->bootflow_head))
81 return -ENOENT;
82
83 *bflowp = list_entry(bflow->bm_node.next, struct bootflow, bm_node);
84
85 return 0;
86}
87
88int bootdev_bind(struct udevice *parent, const char *drv_name, const char *name,
89 struct udevice **devp)
90{
91 struct udevice *dev;
92 char dev_name[30];
93 char *str;
94 int ret;
95
96 snprintf(dev_name, sizeof(dev_name), "%s.%s", parent->name, name);
97 str = strdup(dev_name);
98 if (!str)
99 return -ENOMEM;
100 ret = device_bind_driver(parent, drv_name, str, &dev);
101 if (ret)
102 return ret;
103 device_set_name_alloced(dev);
104 *devp = dev;
105
106 return 0;
107}
108
109int bootdev_find_in_blk(struct udevice *dev, struct udevice *blk,
110 struct bootflow_iter *iter, struct bootflow *bflow)
111{
Simon Glass0fca7e82023-08-24 13:55:43 -0600112 struct bootmeth_uc_plat *plat = dev_get_uclass_plat(bflow->method);
113 bool allow_any_part = plat->flags & BOOTMETHF_ANY_PART;
Simon Glass017656e2022-04-24 23:31:07 -0600114 struct blk_desc *desc = dev_get_uclass_plat(blk);
115 struct disk_partition info;
116 char partstr[20];
117 char name[60];
118 int ret;
119
120 /* Sanity check */
121 if (iter->part >= MAX_PART_PER_BOOTDEV)
122 return log_msg_ret("max", -ESHUTDOWN);
123
124 bflow->blk = blk;
125 if (iter->part)
126 snprintf(partstr, sizeof(partstr), "part_%x", iter->part);
127 else
128 strcpy(partstr, "whole");
129 snprintf(name, sizeof(name), "%s.%s", dev->name, partstr);
130 bflow->name = strdup(name);
131 if (!bflow->name)
132 return log_msg_ret("name", -ENOMEM);
133
134 bflow->part = iter->part;
135
Simon Glass03fcbf92022-04-24 23:31:09 -0600136 ret = bootmeth_check(bflow->method, iter);
137 if (ret)
138 return log_msg_ret("check", ret);
139
Simon Glass017656e2022-04-24 23:31:07 -0600140 /*
141 * partition numbers start at 0 so this cannot succeed, but it can tell
142 * us whether there is valid media there
143 */
144 ret = part_get_info(desc, iter->part, &info);
Simon Glass0fca7e82023-08-24 13:55:43 -0600145 log_debug("part_get_info() returned %d\n", ret);
Simon Glass017656e2022-04-24 23:31:07 -0600146 if (!iter->part && ret == -ENOENT)
147 ret = 0;
148
149 /*
150 * This error indicates the media is not present. Otherwise we just
151 * blindly scan the next partition. We could be more intelligent here
152 * and check which partition numbers actually exist.
153 */
154 if (ret == -EOPNOTSUPP)
155 ret = -ESHUTDOWN;
156 else
157 bflow->state = BOOTFLOWST_MEDIA;
Simon Glass0fca7e82023-08-24 13:55:43 -0600158 if (ret && !allow_any_part) {
Simon Glassa8e7d512023-04-28 13:18:09 -0600159 /* allow partition 1 to be missing */
160 if (iter->part == 1) {
161 iter->max_part = 3;
162 ret = -ENOENT;
163 }
164
Simon Glass017656e2022-04-24 23:31:07 -0600165 return log_msg_ret("part", ret);
Simon Glassa8e7d512023-04-28 13:18:09 -0600166 }
Simon Glass017656e2022-04-24 23:31:07 -0600167
168 /*
169 * Currently we don't get the number of partitions, so just
170 * assume a large number
171 */
172 iter->max_part = MAX_PART_PER_BOOTDEV;
173
Nam Caocb0d3fb2024-02-21 13:41:44 +0100174 if (iter->flags & BOOTFLOWIF_SINGLE_PARTITION) {
175 /* a particular partition was specified, scan it without checking */
176 } else if (!iter->part) {
177 /* This is the whole disk, check if we have bootable partitions */
Simon Glass64cbc5c2023-01-17 10:47:42 -0700178 iter->first_bootable = part_get_bootable(desc);
179 log_debug("checking bootable=%d\n", iter->first_bootable);
Simon Glass0fca7e82023-08-24 13:55:43 -0600180 } else if (allow_any_part) {
181 /*
182 * allow any partition to be scanned, by skipping any checks
183 * for filesystems or partition contents on this disk
184 */
Simon Glass64cbc5c2023-01-17 10:47:42 -0700185
186 /* if there are bootable partitions, scan only those */
Simon Glass0fca7e82023-08-24 13:55:43 -0600187 } else if (iter->first_bootable >= 0 &&
188 (iter->first_bootable ? !info.bootable : iter->part != 1)) {
Simon Glass64cbc5c2023-01-17 10:47:42 -0700189 return log_msg_ret("boot", -EINVAL);
190 } else {
Simon Glass017656e2022-04-24 23:31:07 -0600191 ret = fs_set_blk_dev_with_part(desc, bflow->part);
192 bflow->state = BOOTFLOWST_PART;
Simon Glassa40f7822023-04-24 13:49:49 +1200193 if (ret)
194 return log_msg_ret("fs", ret);
Simon Glass017656e2022-04-24 23:31:07 -0600195
Simon Glass017656e2022-04-24 23:31:07 -0600196 log_debug("%s: Found partition %x type %x fstype %d\n",
Simon Glassed0ec662023-08-24 13:55:33 -0600197 blk->name, bflow->part,
198 IS_ENABLED(CONFIG_DOS_PARTITION) ?
199 disk_partition_sys_ind(&info) : 0,
Simon Glass017656e2022-04-24 23:31:07 -0600200 ret ? -1 : fs_get_type());
Simon Glassa40f7822023-04-24 13:49:49 +1200201 bflow->blk = blk;
Simon Glass017656e2022-04-24 23:31:07 -0600202 bflow->state = BOOTFLOWST_FS;
203 }
204
Simon Glass0fca7e82023-08-24 13:55:43 -0600205 log_debug("method %s\n", bflow->method->name);
Simon Glass03fcbf92022-04-24 23:31:09 -0600206 ret = bootmeth_read_bootflow(bflow->method, bflow);
207 if (ret)
208 return log_msg_ret("method", ret);
209
Simon Glass017656e2022-04-24 23:31:07 -0600210 return 0;
211}
212
213void bootdev_list(bool probe)
214{
215 struct udevice *dev;
216 int ret;
217 int i;
218
219 printf("Seq Probed Status Uclass Name\n");
220 printf("--- ------ ------ -------- ------------------\n");
221 if (probe)
Michal Suchanek675e9082022-10-12 21:57:53 +0200222 ret = uclass_first_device_check(UCLASS_BOOTDEV, &dev);
Simon Glass017656e2022-04-24 23:31:07 -0600223 else
224 ret = uclass_find_first_device(UCLASS_BOOTDEV, &dev);
225 for (i = 0; dev; i++) {
226 printf("%3x [ %c ] %6s %-9.9s %s\n", dev_seq(dev),
227 device_active(dev) ? '+' : ' ',
Heinrich Schuchardt91592d62023-07-30 16:29:25 +0200228 ret ? simple_itoa(-ret) : "OK",
Simon Glass017656e2022-04-24 23:31:07 -0600229 dev_get_uclass_name(dev_get_parent(dev)), dev->name);
230 if (probe)
Michal Suchanek675e9082022-10-12 21:57:53 +0200231 ret = uclass_next_device_check(&dev);
Simon Glass017656e2022-04-24 23:31:07 -0600232 else
233 ret = uclass_find_next_device(&dev);
234 }
235 printf("--- ------ ------ -------- ------------------\n");
236 printf("(%d bootdev%s)\n", i, i != 1 ? "s" : "");
237}
238
239int bootdev_setup_for_dev(struct udevice *parent, const char *drv_name)
240{
241 struct udevice *bdev;
242 int ret;
243
244 ret = device_find_first_child_by_uclass(parent, UCLASS_BOOTDEV,
245 &bdev);
246 if (ret) {
247 if (ret != -ENODEV) {
248 log_debug("Cannot access bootdev device\n");
249 return ret;
250 }
251
252 ret = bootdev_bind(parent, drv_name, "bootdev", &bdev);
253 if (ret) {
254 log_debug("Cannot create bootdev device\n");
255 return ret;
256 }
257 }
258
259 return 0;
260}
261
Simon Glassb1dc36a2023-01-17 10:47:25 -0700262static int bootdev_get_suffix_start(struct udevice *dev, const char *suffix)
263{
264 int len, slen;
265
266 len = strlen(dev->name);
267 slen = strlen(suffix);
268 if (len > slen && !strcmp(suffix, dev->name + len - slen))
269 return len - slen;
270
271 return len;
272}
273
Simon Glassb1d581d2023-07-30 11:15:14 -0600274int bootdev_setup_for_sibling_blk(struct udevice *blk, const char *drv_name)
Simon Glass017656e2022-04-24 23:31:07 -0600275{
276 struct udevice *parent, *dev;
277 char dev_name[50];
Simon Glassb1dc36a2023-01-17 10:47:25 -0700278 int ret, len;
Simon Glass017656e2022-04-24 23:31:07 -0600279
Simon Glassb1dc36a2023-01-17 10:47:25 -0700280 len = bootdev_get_suffix_start(blk, ".blk");
281 snprintf(dev_name, sizeof(dev_name), "%.*s.%s", len, blk->name,
282 "bootdev");
Simon Glass017656e2022-04-24 23:31:07 -0600283
284 parent = dev_get_parent(blk);
285 ret = device_find_child_by_name(parent, dev_name, &dev);
286 if (ret) {
287 char *str;
288
289 if (ret != -ENODEV) {
290 log_debug("Cannot access bootdev device\n");
291 return ret;
292 }
293 str = strdup(dev_name);
294 if (!str)
295 return -ENOMEM;
296
297 ret = device_bind_driver(parent, drv_name, str, &dev);
298 if (ret) {
299 log_debug("Cannot create bootdev device\n");
300 return ret;
301 }
302 device_set_name_alloced(dev);
303 }
304
305 return 0;
306}
307
308int bootdev_get_sibling_blk(struct udevice *dev, struct udevice **blkp)
309{
310 struct udevice *parent = dev_get_parent(dev);
311 struct udevice *blk;
312 int ret, len;
Simon Glass017656e2022-04-24 23:31:07 -0600313
314 if (device_get_uclass_id(dev) != UCLASS_BOOTDEV)
315 return -EINVAL;
316
Simon Glassb1d581d2023-07-30 11:15:14 -0600317 /*
318 * This should always work if bootdev_setup_for_sibling_blk() was used
319 */
Simon Glassb1dc36a2023-01-17 10:47:25 -0700320 len = bootdev_get_suffix_start(dev, ".bootdev");
Simon Glass017656e2022-04-24 23:31:07 -0600321 ret = device_find_child_by_namelen(parent, dev->name, len, &blk);
Simon Glassb1dc36a2023-01-17 10:47:25 -0700322 if (ret) {
323 char dev_name[50];
324
325 snprintf(dev_name, sizeof(dev_name), "%.*s.blk", len,
326 dev->name);
327 ret = device_find_child_by_name(parent, dev_name, &blk);
328 if (ret)
329 return log_msg_ret("find", ret);
330 }
Simon Glassb73a8292023-01-28 15:00:19 -0700331 ret = device_probe(blk);
332 if (ret)
333 return log_msg_ret("act", ret);
Simon Glass017656e2022-04-24 23:31:07 -0600334 *blkp = blk;
335
336 return 0;
337}
338
339static int bootdev_get_from_blk(struct udevice *blk, struct udevice **bootdevp)
340{
341 struct udevice *parent = dev_get_parent(blk);
342 struct udevice *bootdev;
343 char dev_name[50];
Simon Glassb1dc36a2023-01-17 10:47:25 -0700344 int ret, len;
Simon Glass017656e2022-04-24 23:31:07 -0600345
346 if (device_get_uclass_id(blk) != UCLASS_BLK)
347 return -EINVAL;
348
Simon Glassb1d581d2023-07-30 11:15:14 -0600349 /* This should always work if bootdev_setup_for_sibling_blk() was used */
Simon Glassb1dc36a2023-01-17 10:47:25 -0700350 len = bootdev_get_suffix_start(blk, ".blk");
351 snprintf(dev_name, sizeof(dev_name), "%.*s.%s", len, blk->name,
352 "bootdev");
Simon Glass017656e2022-04-24 23:31:07 -0600353 ret = device_find_child_by_name(parent, dev_name, &bootdev);
354 if (ret)
355 return log_msg_ret("find", ret);
356 *bootdevp = bootdev;
357
358 return 0;
359}
360
361int bootdev_unbind_dev(struct udevice *parent)
362{
363 struct udevice *dev;
364 int ret;
365
366 ret = device_find_first_child_by_uclass(parent, UCLASS_BOOTDEV, &dev);
367 if (!ret) {
368 ret = device_remove(dev, DM_REMOVE_NORMAL);
369 if (ret)
370 return log_msg_ret("rem", ret);
371 ret = device_unbind(dev);
372 if (ret)
373 return log_msg_ret("unb", ret);
374 }
375
376 return 0;
377}
378
379/**
Simon Glass73708252023-01-17 10:48:00 -0700380 * label_to_uclass() - Convert a label to a uclass and sequence number
381 *
382 * @label: Label to look up (e.g. "mmc1" or "mmc0")
383 * @seqp: Returns the sequence number, or -1 if none
Simon Glasse22fe922023-01-17 10:48:05 -0700384 * @method_flagsp: If non-NULL, returns any flags implied by the label
385 * (enum bootflow_meth_flags_t), 0 if none
Simon Glass4108c6b2023-04-24 13:49:47 +1200386 * Returns: sequence number on success, -EPFNOSUPPORT is the uclass is not
387 * known, other -ve error code on other error
Simon Glass73708252023-01-17 10:48:00 -0700388 */
Simon Glasse22fe922023-01-17 10:48:05 -0700389static int label_to_uclass(const char *label, int *seqp, int *method_flagsp)
Simon Glass73708252023-01-17 10:48:00 -0700390{
Simon Glasse22fe922023-01-17 10:48:05 -0700391 int seq, len, method_flags;
Simon Glass73708252023-01-17 10:48:00 -0700392 enum uclass_id id;
393 const char *end;
Simon Glass73708252023-01-17 10:48:00 -0700394
Simon Glasse22fe922023-01-17 10:48:05 -0700395 method_flags = 0;
Simon Glass73708252023-01-17 10:48:00 -0700396 seq = trailing_strtoln_end(label, NULL, &end);
397 len = end - label;
398 if (!len)
399 return -EINVAL;
400 id = uclass_get_by_namelen(label, len);
401 log_debug("find %s: seq=%d, id=%d/%s\n", label, seq, id,
402 uclass_get_name(id));
403 if (id == UCLASS_INVALID) {
Simon Glass215be682023-01-17 10:48:03 -0700404 /* try some special cases */
405 if (IS_ENABLED(CONFIG_BOOTDEV_SPI_FLASH) &&
406 !strncmp("spi", label, len)) {
407 id = UCLASS_SPI_FLASH;
Simon Glasse22fe922023-01-17 10:48:05 -0700408 } else if (IS_ENABLED(CONFIG_BOOTDEV_ETH) &&
409 !strncmp("pxe", label, len)) {
410 id = UCLASS_ETH;
411 method_flags |= BOOTFLOW_METHF_PXE_ONLY;
412 } else if (IS_ENABLED(CONFIG_BOOTDEV_ETH) &&
413 !strncmp("dhcp", label, len)) {
414 id = UCLASS_ETH;
415 method_flags |= BOOTFLOW_METHF_DHCP_ONLY;
Simon Glass215be682023-01-17 10:48:03 -0700416 } else {
Simon Glass4108c6b2023-04-24 13:49:47 +1200417 return -EPFNOSUPPORT;
Simon Glass215be682023-01-17 10:48:03 -0700418 }
Simon Glass73708252023-01-17 10:48:00 -0700419 }
420 if (id == UCLASS_USB)
421 id = UCLASS_MASS_STORAGE;
422 *seqp = seq;
Simon Glasse22fe922023-01-17 10:48:05 -0700423 if (method_flagsp)
424 *method_flagsp = method_flags;
Simon Glass73708252023-01-17 10:48:00 -0700425
426 return id;
427}
428
Simon Glasse22fe922023-01-17 10:48:05 -0700429int bootdev_find_by_label(const char *label, struct udevice **devp,
430 int *method_flagsp)
Simon Glass017656e2022-04-24 23:31:07 -0600431{
Simon Glasse22fe922023-01-17 10:48:05 -0700432 int seq, ret, method_flags = 0;
Simon Glass017656e2022-04-24 23:31:07 -0600433 struct udevice *media;
434 struct uclass *uc;
435 enum uclass_id id;
Simon Glass017656e2022-04-24 23:31:07 -0600436
Simon Glass1d809642024-09-01 16:27:28 -0600437 if (!CONFIG_IS_ENABLED(BLK))
438 return -ENOSYS;
439
Simon Glasse22fe922023-01-17 10:48:05 -0700440 ret = label_to_uclass(label, &seq, &method_flags);
Simon Glass73708252023-01-17 10:48:00 -0700441 if (ret < 0)
442 return log_msg_ret("uc", ret);
443 id = ret;
Simon Glass017656e2022-04-24 23:31:07 -0600444
445 /* Iterate through devices in the media uclass (e.g. UCLASS_MMC) */
446 uclass_id_foreach_dev(id, media, uc) {
447 struct udevice *bdev, *blk;
448 int ret;
449
450 /* if there is no seq, match anything */
451 if (seq != -1 && dev_seq(media) != seq) {
452 log_debug("- skip, media seq=%d\n", dev_seq(media));
453 continue;
454 }
455
456 ret = device_find_first_child_by_uclass(media, UCLASS_BOOTDEV,
457 &bdev);
458 if (ret) {
459 log_debug("- looking via blk, seq=%d, id=%d\n", seq,
460 id);
461 ret = blk_find_device(id, seq, &blk);
462 if (!ret) {
463 log_debug("- get from blk %s\n", blk->name);
464 ret = bootdev_get_from_blk(blk, &bdev);
465 }
466 }
467 if (!ret) {
468 log_debug("- found %s\n", bdev->name);
469 *devp = bdev;
Simon Glassde567b12023-01-17 10:48:09 -0700470
471 /*
472 * if no sequence number was provided, we must scan all
473 * bootdevs for this media uclass
474 */
Simon Glass60bcbf02023-09-23 14:50:15 -0600475 if (seq == -1)
Simon Glassde567b12023-01-17 10:48:09 -0700476 method_flags |= BOOTFLOW_METHF_SINGLE_UCLASS;
Simon Glasse22fe922023-01-17 10:48:05 -0700477 if (method_flagsp)
478 *method_flagsp = method_flags;
Simon Glass60bcbf02023-09-23 14:50:15 -0600479 log_debug("method flags %x\n", method_flags);
Simon Glass017656e2022-04-24 23:31:07 -0600480 return 0;
481 }
482 log_debug("- no device in %s\n", media->name);
483 }
Simon Glass017656e2022-04-24 23:31:07 -0600484
485 return -ENOENT;
486}
487
Simon Glasse22fe922023-01-17 10:48:05 -0700488int bootdev_find_by_any(const char *name, struct udevice **devp,
489 int *method_flagsp)
Simon Glass017656e2022-04-24 23:31:07 -0600490{
491 struct udevice *dev;
Simon Glasse22fe922023-01-17 10:48:05 -0700492 int method_flags = 0;
Simon Glassde567b12023-01-17 10:48:09 -0700493 int ret = -ENODEV, seq;
Simon Glass017656e2022-04-24 23:31:07 -0600494 char *endp;
495
496 seq = simple_strtol(name, &endp, 16);
497
498 /* Select by name, label or number */
499 if (*endp) {
500 ret = uclass_get_device_by_name(UCLASS_BOOTDEV, name, &dev);
501 if (ret == -ENODEV) {
Simon Glasse22fe922023-01-17 10:48:05 -0700502 ret = bootdev_find_by_label(name, &dev, &method_flags);
Simon Glass017656e2022-04-24 23:31:07 -0600503 if (ret) {
504 printf("Cannot find bootdev '%s' (err=%d)\n",
505 name, ret);
Simon Glasse22fe922023-01-17 10:48:05 -0700506 return log_msg_ret("lab", ret);
Simon Glass017656e2022-04-24 23:31:07 -0600507 }
508 ret = device_probe(dev);
509 }
510 if (ret) {
511 printf("Cannot probe bootdev '%s' (err=%d)\n", name,
512 ret);
Simon Glasse22fe922023-01-17 10:48:05 -0700513 return log_msg_ret("pro", ret);
Simon Glass017656e2022-04-24 23:31:07 -0600514 }
Simon Glassde567b12023-01-17 10:48:09 -0700515 } else if (IS_ENABLED(CONFIG_BOOTSTD_FULL)) {
Simon Glass017656e2022-04-24 23:31:07 -0600516 ret = uclass_get_device_by_seq(UCLASS_BOOTDEV, seq, &dev);
Simon Glassde567b12023-01-17 10:48:09 -0700517 method_flags |= BOOTFLOW_METHF_SINGLE_DEV;
Simon Glass017656e2022-04-24 23:31:07 -0600518 }
519 if (ret) {
520 printf("Cannot find '%s' (err=%d)\n", name, ret);
521 return ret;
522 }
523
524 *devp = dev;
Simon Glasse22fe922023-01-17 10:48:05 -0700525 if (method_flagsp)
526 *method_flagsp = method_flags;
Simon Glass017656e2022-04-24 23:31:07 -0600527
528 return 0;
529}
530
Simon Glassde567b12023-01-17 10:48:09 -0700531int bootdev_hunt_and_find_by_label(const char *label, struct udevice **devp,
532 int *method_flagsp)
533{
534 int ret;
535
536 ret = bootdev_hunt(label, false);
537 if (ret)
538 return log_msg_ret("scn", ret);
539 ret = bootdev_find_by_label(label, devp, method_flagsp);
540 if (ret)
541 return log_msg_ret("fnd", ret);
542
543 return 0;
544}
545
Simon Glassd6e39d12023-01-17 10:47:26 -0700546static int default_get_bootflow(struct udevice *dev, struct bootflow_iter *iter,
547 struct bootflow *bflow)
548{
549 struct udevice *blk;
550 int ret;
551
552 ret = bootdev_get_sibling_blk(dev, &blk);
Simon Glassbf2d02a2023-07-30 11:15:16 -0600553 log_debug("sibling_blk ret=%d, blk=%s\n", ret,
554 ret ? "(none)" : blk->name);
Simon Glassd6e39d12023-01-17 10:47:26 -0700555 /*
556 * If there is no media, indicate that no more partitions should be
557 * checked
558 */
559 if (ret == -EOPNOTSUPP)
560 ret = -ESHUTDOWN;
561 if (ret)
562 return log_msg_ret("blk", ret);
563 assert(blk);
564 ret = bootdev_find_in_blk(dev, blk, iter, bflow);
565 if (ret)
566 return log_msg_ret("find", ret);
567
568 return 0;
569}
570
Simon Glass017656e2022-04-24 23:31:07 -0600571int bootdev_get_bootflow(struct udevice *dev, struct bootflow_iter *iter,
572 struct bootflow *bflow)
573{
574 const struct bootdev_ops *ops = bootdev_get_ops(dev);
575
Simon Glass0fca7e82023-08-24 13:55:43 -0600576 log_debug("->get_bootflow %s,%x=%p\n", dev->name, iter->part,
577 ops->get_bootflow);
Simon Glass00501fc2022-10-20 18:22:51 -0600578 bootflow_init(bflow, dev, iter->method);
Simon Glassd6e39d12023-01-17 10:47:26 -0700579 if (!ops->get_bootflow)
580 return default_get_bootflow(dev, iter, bflow);
Simon Glass017656e2022-04-24 23:31:07 -0600581
582 return ops->get_bootflow(dev, iter, bflow);
583}
584
585void bootdev_clear_bootflows(struct udevice *dev)
586{
587 struct bootdev_uc_plat *ucp = dev_get_uclass_plat(dev);
588
589 while (!list_empty(&ucp->bootflow_head)) {
590 struct bootflow *bflow;
591
592 bflow = list_first_entry(&ucp->bootflow_head, struct bootflow,
593 bm_node);
Simon Glass03fcbf92022-04-24 23:31:09 -0600594 bootflow_remove(bflow);
Simon Glass017656e2022-04-24 23:31:07 -0600595 }
596}
597
Simon Glassac06b26a2023-01-17 10:48:10 -0700598int bootdev_next_label(struct bootflow_iter *iter, struct udevice **devp,
599 int *method_flagsp)
600{
601 struct udevice *dev;
602
603 log_debug("next\n");
604 for (dev = NULL; !dev && iter->labels[++iter->cur_label];) {
Simon Glass4108c6b2023-04-24 13:49:47 +1200605 const char *label = iter->labels[iter->cur_label];
606 int ret;
607
608 log_debug("Scanning: %s\n", label);
609 ret = bootdev_hunt_and_find_by_label(label, &dev,
610 method_flagsp);
611 if (iter->flags & BOOTFLOWIF_SHOW) {
612 if (ret == -EPFNOSUPPORT) {
613 log_warning("Unknown uclass '%s' in label\n",
614 label);
615 } else if (ret == -ENOENT) {
616 /*
617 * looking for, e.g. 'scsi0' should find
618 * something if SCSI is present
619 */
620 if (!trailing_strtol(label)) {
621 log_warning("No bootdevs for '%s'\n",
622 label);
623 }
624 }
625 }
626
Simon Glassac06b26a2023-01-17 10:48:10 -0700627 }
628
629 if (!dev)
630 return log_msg_ret("fin", -ENODEV);
631 *devp = dev;
632
633 return 0;
634}
635
Simon Glass660a9952023-01-17 10:48:11 -0700636int bootdev_next_prio(struct bootflow_iter *iter, struct udevice **devp)
637{
Simon Glass15f128a2024-08-15 14:30:21 -0600638 struct udevice *dev = *devp;
Simon Glass660a9952023-01-17 10:48:11 -0700639 bool found;
640 int ret;
641
642 /* find the next device with this priority */
643 *devp = NULL;
644 log_debug("next prio %d: dev=%p/%s\n", iter->cur_prio, dev,
645 dev ? dev->name : "none");
Simon Glass5aaa5622024-08-15 14:30:22 -0600646 found = false;
Simon Glass660a9952023-01-17 10:48:11 -0700647 do {
648 /*
649 * Don't probe devices here since they may not be of the
650 * required priority
651 */
652 if (!dev)
653 uclass_find_first_device(UCLASS_BOOTDEV, &dev);
654 else
655 uclass_find_next_device(&dev);
656 found = false;
657
658 /* scan for the next device with the correct priority */
659 while (dev) {
660 struct bootdev_uc_plat *plat;
661
662 plat = dev_get_uclass_plat(dev);
663 log_debug("- %s: %d, want %d\n", dev->name, plat->prio,
664 iter->cur_prio);
665 if (plat->prio == iter->cur_prio)
666 break;
667 uclass_find_next_device(&dev);
668 }
669
670 /* none found for this priority, so move to the next */
671 if (!dev) {
672 log_debug("None found at prio %d, moving to %d\n",
673 iter->cur_prio, iter->cur_prio + 1);
674 if (++iter->cur_prio == BOOTDEVP_COUNT)
675 return log_msg_ret("fin", -ENODEV);
676
Simon Glass99e68182023-02-22 12:17:03 -0700677 if (iter->flags & BOOTFLOWIF_HUNT) {
Simon Glass660a9952023-01-17 10:48:11 -0700678 /* hunt to find new bootdevs */
679 ret = bootdev_hunt_prio(iter->cur_prio,
680 iter->flags &
Simon Glass99e68182023-02-22 12:17:03 -0700681 BOOTFLOWIF_SHOW);
Simon Glassbf2d02a2023-07-30 11:15:16 -0600682 log_debug("- bootdev_hunt_prio() ret %d\n",
683 ret);
Simon Glass660a9952023-01-17 10:48:11 -0700684 if (ret)
685 return log_msg_ret("hun", ret);
686 }
687 } else {
688 ret = device_probe(dev);
Simon Glass5aaa5622024-08-15 14:30:22 -0600689 if (ret)
Simon Glass15f128a2024-08-15 14:30:21 -0600690 log_debug("Device '%s' failed to probe\n",
Simon Glass660a9952023-01-17 10:48:11 -0700691 dev->name);
Simon Glass5aaa5622024-08-15 14:30:22 -0600692 else
693 found = true;
Simon Glass660a9952023-01-17 10:48:11 -0700694 }
Simon Glass5aaa5622024-08-15 14:30:22 -0600695 } while (!found);
Simon Glass660a9952023-01-17 10:48:11 -0700696
697 *devp = dev;
698
699 return 0;
700}
701
Simon Glassba3d5372023-01-17 10:48:15 -0700702int bootdev_setup_iter(struct bootflow_iter *iter, const char *label,
703 struct udevice **devp, int *method_flagsp)
Simon Glass017656e2022-04-24 23:31:07 -0600704{
Simon Glassba3d5372023-01-17 10:48:15 -0700705 struct udevice *bootstd, *dev = NULL;
Simon Glass99e68182023-02-22 12:17:03 -0700706 bool show = iter->flags & BOOTFLOWIF_SHOW;
Simon Glass484e4072023-01-17 10:48:14 -0700707 int method_flags;
Nam Caocb0d3fb2024-02-21 13:41:44 +0100708 char buf[32];
Simon Glass017656e2022-04-24 23:31:07 -0600709 int ret;
710
Nam Caocb0d3fb2024-02-21 13:41:44 +0100711 if (label) {
712 const char *end = strchr(label, ':');
713
714 if (end) {
715 size_t len = (size_t)(end - label);
716 const char *part = end + 1;
717
718 if (len + 1 > sizeof(buf)) {
719 log_err("label \"%s\" is way too long\n", label);
720 return -EINVAL;
721 }
722
723 memcpy(buf, label, len);
724 buf[len] = '\0';
725 label = buf;
726
727 unsigned long tmp;
728
729 if (strict_strtoul(part, 0, &tmp)) {
730 log_err("Invalid partition number: %s\n", part);
731 return -EINVAL;
732 }
733
734 iter->flags |= BOOTFLOWIF_SINGLE_PARTITION;
735 iter->part = tmp;
736 }
737 }
738
Simon Glass017656e2022-04-24 23:31:07 -0600739 ret = uclass_first_device_err(UCLASS_BOOTSTD, &bootstd);
740 if (ret) {
741 log_err("Missing bootstd device\n");
742 return log_msg_ret("std", ret);
743 }
744
Simon Glass7e1f6a42023-01-17 10:48:08 -0700745 /* hunt for any pre-scan devices */
Simon Glass99e68182023-02-22 12:17:03 -0700746 if (iter->flags & BOOTFLOWIF_HUNT) {
Simon Glass7e1f6a42023-01-17 10:48:08 -0700747 ret = bootdev_hunt_prio(BOOTDEVP_1_PRE_SCAN, show);
Simon Glassbf2d02a2023-07-30 11:15:16 -0600748 log_debug("- bootdev_hunt_prio() ret %d\n", ret);
Simon Glass7e1f6a42023-01-17 10:48:08 -0700749 if (ret)
750 return log_msg_ret("pre", ret);
751 }
752
Simon Glass017656e2022-04-24 23:31:07 -0600753 /* Handle scanning a single device */
Simon Glassba3d5372023-01-17 10:48:15 -0700754 if (IS_ENABLED(CONFIG_BOOTSTD_FULL) && label) {
Simon Glass99e68182023-02-22 12:17:03 -0700755 if (iter->flags & BOOTFLOWIF_HUNT) {
Simon Glassba3d5372023-01-17 10:48:15 -0700756 ret = bootdev_hunt(label, show);
757 if (ret)
758 return log_msg_ret("hun", ret);
759 }
760 ret = bootdev_find_by_any(label, &dev, &method_flags);
761 if (ret)
762 return log_msg_ret("lab", ret);
763
764 log_debug("method_flags: %x\n", method_flags);
765 if (method_flags & BOOTFLOW_METHF_SINGLE_UCLASS)
Simon Glass99e68182023-02-22 12:17:03 -0700766 iter->flags |= BOOTFLOWIF_SINGLE_UCLASS;
Simon Glassba3d5372023-01-17 10:48:15 -0700767 else if (method_flags & BOOTFLOW_METHF_SINGLE_DEV)
Simon Glass99e68182023-02-22 12:17:03 -0700768 iter->flags |= BOOTFLOWIF_SINGLE_DEV;
Simon Glassba3d5372023-01-17 10:48:15 -0700769 else
Simon Glass99e68182023-02-22 12:17:03 -0700770 iter->flags |= BOOTFLOWIF_SINGLE_MEDIA;
Simon Glassba3d5372023-01-17 10:48:15 -0700771 log_debug("Selected label: %s, flags %x\n", label, iter->flags);
Simon Glass484e4072023-01-17 10:48:14 -0700772 } else {
773 bool ok;
Simon Glass017656e2022-04-24 23:31:07 -0600774
Simon Glass484e4072023-01-17 10:48:14 -0700775 /* This either returns a non-empty list or NULL */
776 iter->labels = bootstd_get_bootdev_order(bootstd, &ok);
777 if (!ok)
778 return log_msg_ret("ord", -ENOMEM);
779 log_debug("setup labels %p\n", iter->labels);
780 if (iter->labels) {
781 iter->cur_label = -1;
782 ret = bootdev_next_label(iter, &dev, &method_flags);
783 } else {
784 ret = bootdev_next_prio(iter, &dev);
785 method_flags = 0;
786 }
787 if (!dev)
788 return log_msg_ret("fin", -ENOENT);
789 log_debug("Selected bootdev: %s\n", dev->name);
Simon Glass017656e2022-04-24 23:31:07 -0600790 }
791
Simon Glass017656e2022-04-24 23:31:07 -0600792 ret = device_probe(dev);
793 if (ret)
794 return log_msg_ret("probe", ret);
Simon Glass484e4072023-01-17 10:48:14 -0700795 if (method_flagsp)
796 *method_flagsp = method_flags;
Simon Glass017656e2022-04-24 23:31:07 -0600797 *devp = dev;
798
799 return 0;
800}
801
Simon Glass1248d2b2023-01-17 10:47:34 -0700802static int bootdev_hunt_drv(struct bootdev_hunter *info, uint seq, bool show)
803{
804 const char *name = uclass_get_name(info->uclass);
805 struct bootstd_priv *std;
806 int ret;
807
808 ret = bootstd_get_priv(&std);
809 if (ret)
810 return log_msg_ret("std", ret);
811
812 if (!(std->hunters_used & BIT(seq))) {
813 if (show)
814 printf("Hunting with: %s\n",
815 uclass_get_name(info->uclass));
816 log_debug("Hunting with: %s\n", name);
817 if (info->hunt) {
818 ret = info->hunt(info, show);
Simon Glassbf2d02a2023-07-30 11:15:16 -0600819 log_debug(" - hunt result %d\n", ret);
Tony Dinh81356b42023-11-02 11:51:15 -0700820 if (ret && ret != -ENOENT)
Simon Glass1248d2b2023-01-17 10:47:34 -0700821 return ret;
822 }
823 std->hunters_used |= BIT(seq);
824 }
825
826 return 0;
827}
828
829int bootdev_hunt(const char *spec, bool show)
830{
831 struct bootdev_hunter *start;
832 const char *end;
833 int n_ent, i;
834 int result;
835 size_t len;
836
837 start = ll_entry_start(struct bootdev_hunter, bootdev_hunter);
838 n_ent = ll_entry_count(struct bootdev_hunter, bootdev_hunter);
839 result = 0;
840
841 len = SIZE_MAX;
842 if (spec) {
843 trailing_strtoln_end(spec, NULL, &end);
844 len = end - spec;
845 }
846
847 for (i = 0; i < n_ent; i++) {
848 struct bootdev_hunter *info = start + i;
849 const char *name = uclass_get_name(info->uclass);
850 int ret;
851
852 log_debug("looking at %.*s for %s\n",
853 (int)max(strlen(name), len), spec, name);
Simon Glassac06b26a2023-01-17 10:48:10 -0700854 if (spec && strncmp(spec, name, max(strlen(name), len))) {
855 if (info->uclass != UCLASS_ETH ||
856 (strcmp("dhcp", spec) && strcmp("pxe", spec)))
857 continue;
858 }
Simon Glass1248d2b2023-01-17 10:47:34 -0700859 ret = bootdev_hunt_drv(info, i, show);
860 if (ret)
861 result = ret;
862 }
863
864 return result;
865}
866
Simon Glass62b03cc2023-09-20 07:29:49 -0600867int bootdev_unhunt(enum uclass_id id)
868{
869 struct bootdev_hunter *start;
870 int n_ent, i;
871
872 start = ll_entry_start(struct bootdev_hunter, bootdev_hunter);
873 n_ent = ll_entry_count(struct bootdev_hunter, bootdev_hunter);
874 for (i = 0; i < n_ent; i++) {
875 struct bootdev_hunter *info = start + i;
876
877 if (info->uclass == id) {
878 struct bootstd_priv *std;
879 int ret;
880
881 ret = bootstd_get_priv(&std);
882 if (ret)
883 return log_msg_ret("std", ret);
884 if (!(std->hunters_used & BIT(i)))
885 return -EALREADY;
886 std->hunters_used &= ~BIT(i);
887 return 0;
888 }
889 }
890
891 return -ENOENT;
892}
893
Simon Glass3e9f6be2023-01-17 10:48:07 -0700894int bootdev_hunt_prio(enum bootdev_prio_t prio, bool show)
895{
896 struct bootdev_hunter *start;
897 int n_ent, i;
898 int result;
899
900 start = ll_entry_start(struct bootdev_hunter, bootdev_hunter);
901 n_ent = ll_entry_count(struct bootdev_hunter, bootdev_hunter);
902 result = 0;
903
904 log_debug("Hunting for priority %d\n", prio);
905 for (i = 0; i < n_ent; i++) {
906 struct bootdev_hunter *info = start + i;
907 int ret;
908
909 if (prio != info->prio)
910 continue;
911 ret = bootdev_hunt_drv(info, i, show);
Simon Glassbf2d02a2023-07-30 11:15:16 -0600912 log_debug("bootdev_hunt_drv() return %d\n", ret);
Simon Glass3e9f6be2023-01-17 10:48:07 -0700913 if (ret && ret != -ENOENT)
914 result = ret;
915 }
Simon Glassbf2d02a2023-07-30 11:15:16 -0600916 log_debug("exit %d\n", result);
Simon Glass3e9f6be2023-01-17 10:48:07 -0700917
918 return result;
919}
920
Simon Glass5acb97a2023-01-17 10:47:33 -0700921void bootdev_list_hunters(struct bootstd_priv *std)
922{
923 struct bootdev_hunter *orig, *start;
924 int n_ent, i;
925
926 orig = ll_entry_start(struct bootdev_hunter, bootdev_hunter);
927 n_ent = ll_entry_count(struct bootdev_hunter, bootdev_hunter);
928
929 /*
930 * workaround for strange bug in clang-12 which sees all the below data
931 * as zeroes. Any access of start seems to fix it, such as
932 *
933 * printf("%p", start);
934 *
935 * Use memcpy() to force the correct behaviour.
936 */
937 memcpy(&start, &orig, sizeof(orig));
938 printf("%4s %4s %-15s %s\n", "Prio", "Used", "Uclass", "Hunter");
939 printf("%4s %4s %-15s %s\n", "----", "----", "---------------", "---------------");
940 for (i = 0; i < n_ent; i++) {
941 struct bootdev_hunter *info = start + i;
942
943 printf("%4d %4s %-15s %s\n", info->prio,
944 std->hunters_used & BIT(i) ? "*" : "",
945 uclass_get_name(info->uclass),
946 info->drv ? info->drv->name : "(none)");
947 }
948
949 printf("(total hunters: %d)\n", n_ent);
950}
951
Simon Glass017656e2022-04-24 23:31:07 -0600952static int bootdev_post_bind(struct udevice *dev)
953{
954 struct bootdev_uc_plat *ucp = dev_get_uclass_plat(dev);
955
956 INIT_LIST_HEAD(&ucp->bootflow_head);
957
958 return 0;
959}
960
961static int bootdev_pre_unbind(struct udevice *dev)
962{
963 bootdev_clear_bootflows(dev);
964
965 return 0;
966}
967
968UCLASS_DRIVER(bootdev) = {
969 .id = UCLASS_BOOTDEV,
970 .name = "bootdev",
971 .flags = DM_UC_FLAG_SEQ_ALIAS,
972 .per_device_plat_auto = sizeof(struct bootdev_uc_plat),
973 .post_bind = bootdev_post_bind,
974 .pre_unbind = bootdev_pre_unbind,
975};