Gitiles
Code Review Sign In
git01.mediatek.com / filogic / uboot / 4db52a5e7609cf9bb456ea081158c360dbc3cc56 / . / boot / bootmeth_script.c
blob: d1c3f9400372f88bb708b2ed7dda72ce89ae5040 [file] [log] [blame]
Simon Glass7e03e742022-04-24 23:31:22 -0600[diff] [blame]1// SPDX-License-Identifier: GPL-2.0+
2/*
3 * Bootmethod for booting via a U-Boot script
4 *
5 * Copyright 2021 Google LLC
6 * Written by Simon Glass <sjg@chromium.org>
7 */
8
9#define LOG_CATEGORY UCLASS_BOOTSTD
10
11#include <common.h>
12#include <blk.h>
13#include <bootflow.h>
14#include <bootmeth.h>
15#include <bootstd.h>
16#include <dm.h>
17#include <env.h>
18#include <fs.h>
19#include <image.h>
20#include <malloc.h>
21#include <mapmem.h>
22
23#define SCRIPT_FNAME1 "boot.scr.uimg"
24#define SCRIPT_FNAME2 "boot.scr"
25
26static int script_check(struct udevice *dev, struct bootflow_iter *iter)
27{
28 int ret;
29
30 /* This only works on block devices */
31 ret = bootflow_iter_uses_blk_dev(iter);
32 if (ret)
33 return log_msg_ret("blk", ret);
34
35 return 0;
36}
37
38static int script_read_bootflow(struct udevice *dev, struct bootflow *bflow)
39{
40 struct blk_desc *desc = NULL;
41 const char *const *prefixes;
42 struct udevice *bootstd;
43 const char *prefix;
44 int ret, i;
45
46 ret = uclass_first_device_err(UCLASS_BOOTSTD, &bootstd);
47 if (ret)
48 return log_msg_ret("std", ret);
49
50 /* We require a partition table */
51 if (!bflow->part)
52 return -ENOENT;
53
54 if (bflow->blk)
55 desc = dev_get_uclass_plat(bflow->blk);
56
57 prefixes = bootstd_get_prefixes(bootstd);
58 i = 0;
59 do {
60 prefix = prefixes ? prefixes[i] : NULL;
61
62 ret = bootmeth_try_file(bflow, desc, prefix, SCRIPT_FNAME1);
63 if (ret)
64 ret = bootmeth_try_file(bflow, desc, prefix,
65 SCRIPT_FNAME2);
66 } while (ret && prefixes && prefixes[++i]);
67 if (ret)
68 return log_msg_ret("try", ret);
69
70 bflow->subdir = strdup(prefix ? prefix : "");
71 if (!bflow->subdir)
72 return log_msg_ret("prefix", -ENOMEM);
73
74 ret = bootmeth_alloc_file(bflow, 0x10000, 1);
75 if (ret)
76 return log_msg_ret("read", ret);
77
78 return 0;
79}
80
81static int script_boot(struct udevice *dev, struct bootflow *bflow)
82{
83 struct blk_desc *desc = dev_get_uclass_plat(bflow->blk);
84 ulong addr;
85 int ret;
86
87 ret = env_set("devtype", blk_get_devtype(bflow->blk));
88 if (!ret)
89 ret = env_set_hex("devnum", desc->devnum);
90 if (!ret)
91 ret = env_set("prefix", bflow->subdir);
92 if (!ret && IS_ENABLED(CONFIG_ARCH_SUNXI) &&
93 !strcmp("mmc", blk_get_devtype(bflow->blk)))
94 ret = env_set_hex("mmc_bootdev", desc->devnum);
95 if (ret)
96 return log_msg_ret("env", ret);
97
98 log_debug("devtype: %s\n", env_get("devtype"));
99 log_debug("devnum: %s\n", env_get("devnum"));
100 log_debug("prefix: %s\n", env_get("prefix"));
101 log_debug("mmc_bootdev: %s\n", env_get("mmc_bootdev"));
102
103 addr = map_to_sysmem(bflow->buf);
104 ret = image_source_script(addr, NULL);
105 if (ret)
106 return log_msg_ret("boot", ret);
107
108 return 0;
109}
110
111static int script_bootmeth_bind(struct udevice *dev)
112{
113 struct bootmeth_uc_plat *plat = dev_get_uclass_plat(dev);
114
115 plat->desc = IS_ENABLED(CONFIG_BOOTSTD_FULL) ?
116 "Script boot from a block device" : "script";
117
118 return 0;
119}
120
121static struct bootmeth_ops script_bootmeth_ops = {
122 .check = script_check,
123 .read_bootflow = script_read_bootflow,
124 .read_file = bootmeth_common_read_file,
125 .boot = script_boot,
126};
127
128static const struct udevice_id script_bootmeth_ids[] = {
129 { .compatible = "u-boot,script" },
130 { }
131};
132
133U_BOOT_DRIVER(bootmeth_script) = {
134 .name = "bootmeth_script",
135 .id = UCLASS_BOOTMETH,
136 .of_match = script_bootmeth_ids,
137 .ops = &script_bootmeth_ops,
138 .bind = script_bootmeth_bind,
139};