]> Git Repo - u-boot.git/blame - cmd/bcb.c
Convert CONFIG_SDCARD et al to Kconfig
[u-boot.git] / cmd / bcb.c
CommitLineData
db7b7a05
ER
1// SPDX-License-Identifier: GPL-2.0+
2/*
3 * Copyright (C) 2019 Eugeniu Rosca <[email protected]>
4 *
5 * Command to read/modify/write Android BCB fields
6 */
7
8#include <android_bootloader_message.h>
e74670ad 9#include <bcb.h>
db7b7a05
ER
10#include <command.h>
11#include <common.h>
f7ae49fc 12#include <log.h>
e6f6f9e6 13#include <part.h>
bafdf4ca 14#include <malloc.h>
18956cd4 15#include <memalign.h>
db7b7a05
ER
16
17enum bcb_cmd {
18 BCB_CMD_LOAD,
19 BCB_CMD_FIELD_SET,
20 BCB_CMD_FIELD_CLEAR,
21 BCB_CMD_FIELD_TEST,
22 BCB_CMD_FIELD_DUMP,
23 BCB_CMD_STORE,
24};
25
26static int bcb_dev = -1;
27static int bcb_part = -1;
18956cd4 28static struct bootloader_message bcb __aligned(ARCH_DMA_MINALIGN) = { { 0 } };
db7b7a05
ER
29
30static int bcb_cmd_get(char *cmd)
31{
739168cf 32 if (!strcmp(cmd, "load"))
db7b7a05 33 return BCB_CMD_LOAD;
739168cf 34 if (!strcmp(cmd, "set"))
db7b7a05 35 return BCB_CMD_FIELD_SET;
739168cf 36 if (!strcmp(cmd, "clear"))
db7b7a05 37 return BCB_CMD_FIELD_CLEAR;
739168cf 38 if (!strcmp(cmd, "test"))
db7b7a05 39 return BCB_CMD_FIELD_TEST;
739168cf 40 if (!strcmp(cmd, "store"))
db7b7a05 41 return BCB_CMD_STORE;
739168cf 42 if (!strcmp(cmd, "dump"))
db7b7a05
ER
43 return BCB_CMD_FIELD_DUMP;
44 else
45 return -1;
46}
47
48static int bcb_is_misused(int argc, char *const argv[])
49{
50 int cmd = bcb_cmd_get(argv[0]);
51
52 switch (cmd) {
53 case BCB_CMD_LOAD:
db7b7a05
ER
54 case BCB_CMD_FIELD_SET:
55 if (argc != 3)
56 goto err;
57 break;
58 case BCB_CMD_FIELD_TEST:
59 if (argc != 4)
60 goto err;
61 break;
62 case BCB_CMD_FIELD_CLEAR:
63 if (argc != 1 && argc != 2)
64 goto err;
65 break;
66 case BCB_CMD_STORE:
67 if (argc != 1)
68 goto err;
69 break;
70 case BCB_CMD_FIELD_DUMP:
71 if (argc != 2)
72 goto err;
73 break;
74 default:
75 printf("Error: 'bcb %s' not supported\n", argv[0]);
76 return -1;
77 }
78
79 if (cmd != BCB_CMD_LOAD && (bcb_dev < 0 || bcb_part < 0)) {
80 printf("Error: Please, load BCB first!\n");
81 return -1;
82 }
83
84 return 0;
85err:
86 printf("Error: Bad usage of 'bcb %s'\n", argv[0]);
87
88 return -1;
89}
90
37f1811a 91static int bcb_field_get(char *name, char **fieldp, int *sizep)
db7b7a05 92{
739168cf 93 if (!strcmp(name, "command")) {
37f1811a
ER
94 *fieldp = bcb.command;
95 *sizep = sizeof(bcb.command);
739168cf 96 } else if (!strcmp(name, "status")) {
37f1811a
ER
97 *fieldp = bcb.status;
98 *sizep = sizeof(bcb.status);
739168cf 99 } else if (!strcmp(name, "recovery")) {
37f1811a
ER
100 *fieldp = bcb.recovery;
101 *sizep = sizeof(bcb.recovery);
739168cf 102 } else if (!strcmp(name, "stage")) {
37f1811a
ER
103 *fieldp = bcb.stage;
104 *sizep = sizeof(bcb.stage);
739168cf 105 } else if (!strcmp(name, "reserved")) {
37f1811a
ER
106 *fieldp = bcb.reserved;
107 *sizep = sizeof(bcb.reserved);
db7b7a05
ER
108 } else {
109 printf("Error: Unknown bcb field '%s'\n", name);
110 return -1;
111 }
112
113 return 0;
114}
115
24e85bbd 116static int __bcb_load(int devnum, const char *partp)
db7b7a05
ER
117{
118 struct blk_desc *desc;
0528979f 119 struct disk_partition info;
db7b7a05
ER
120 u64 cnt;
121 char *endp;
122 int part, ret;
123
24e85bbd
ER
124 desc = blk_get_devnum_by_type(IF_TYPE_MMC, devnum);
125 if (!desc) {
126 ret = -ENODEV;
37f1811a 127 goto err_read_fail;
24e85bbd 128 }
db7b7a05 129
24e85bbd 130 part = simple_strtoul(partp, &endp, 0);
db7b7a05
ER
131 if (*endp == '\0') {
132 ret = part_get_info(desc, part, &info);
133 if (ret)
37f1811a 134 goto err_read_fail;
db7b7a05 135 } else {
24e85bbd 136 part = part_get_info_by_name(desc, partp, &info);
db7b7a05
ER
137 if (part < 0) {
138 ret = part;
37f1811a 139 goto err_read_fail;
db7b7a05
ER
140 }
141 }
142
143 cnt = DIV_ROUND_UP(sizeof(struct bootloader_message), info.blksz);
144 if (cnt > info.size)
37f1811a 145 goto err_too_small;
db7b7a05
ER
146
147 if (blk_dread(desc, info.start, cnt, &bcb) != cnt) {
148 ret = -EIO;
37f1811a 149 goto err_read_fail;
db7b7a05
ER
150 }
151
152 bcb_dev = desc->devnum;
153 bcb_part = part;
154 debug("%s: Loaded from mmc %d:%d\n", __func__, bcb_dev, bcb_part);
155
156 return CMD_RET_SUCCESS;
37f1811a 157err_read_fail:
24e85bbd 158 printf("Error: mmc %d:%s read failed (%d)\n", devnum, partp, ret);
db7b7a05 159 goto err;
37f1811a 160err_too_small:
24e85bbd 161 printf("Error: mmc %d:%s too small!", devnum, partp);
db7b7a05
ER
162 goto err;
163err:
164 bcb_dev = -1;
165 bcb_part = -1;
166
167 return CMD_RET_FAILURE;
168}
169
24e85bbd
ER
170static int do_bcb_load(struct cmd_tbl *cmdtp, int flag, int argc,
171 char * const argv[])
172{
173 char *endp;
174 int devnum = simple_strtoul(argv[1], &endp, 0);
175
176 if (*endp != '\0') {
177 printf("Error: Device id '%s' not a number\n", argv[1]);
178 return CMD_RET_FAILURE;
179 }
180
181 return __bcb_load(devnum, argv[2]);
182}
183
bafdf4ca 184static int __bcb_set(char *fieldp, const char *valp)
db7b7a05
ER
185{
186 int size, len;
bafdf4ca 187 char *field, *str, *found, *tmp;
db7b7a05 188
02ebe6de 189 if (bcb_field_get(fieldp, &field, &size))
db7b7a05
ER
190 return CMD_RET_FAILURE;
191
02ebe6de 192 len = strlen(valp);
db7b7a05
ER
193 if (len >= size) {
194 printf("Error: sizeof('%s') = %d >= %d = sizeof(bcb.%s)\n",
02ebe6de 195 valp, len, size, fieldp);
db7b7a05
ER
196 return CMD_RET_FAILURE;
197 }
bafdf4ca
ER
198 str = strdup(valp);
199 if (!str) {
200 printf("Error: Out of memory while strdup\n");
201 return CMD_RET_FAILURE;
202 }
db7b7a05 203
bafdf4ca 204 tmp = str;
db7b7a05 205 field[0] = '\0';
bafdf4ca 206 while ((found = strsep(&tmp, ":"))) {
db7b7a05
ER
207 if (field[0] != '\0')
208 strcat(field, "\n");
209 strcat(field, found);
210 }
bafdf4ca 211 free(str);
db7b7a05
ER
212
213 return CMD_RET_SUCCESS;
214}
215
02ebe6de
ER
216static int do_bcb_set(struct cmd_tbl *cmdtp, int flag, int argc,
217 char * const argv[])
218{
219 return __bcb_set(argv[1], argv[2]);
220}
221
09140113
SG
222static int do_bcb_clear(struct cmd_tbl *cmdtp, int flag, int argc,
223 char *const argv[])
db7b7a05
ER
224{
225 int size;
226 char *field;
227
228 if (argc == 1) {
229 memset(&bcb, 0, sizeof(bcb));
230 return CMD_RET_SUCCESS;
231 }
232
233 if (bcb_field_get(argv[1], &field, &size))
234 return CMD_RET_FAILURE;
235
236 memset(field, 0, size);
237
238 return CMD_RET_SUCCESS;
239}
240
09140113
SG
241static int do_bcb_test(struct cmd_tbl *cmdtp, int flag, int argc,
242 char *const argv[])
db7b7a05
ER
243{
244 int size;
245 char *field;
246 char *op = argv[2];
247
248 if (bcb_field_get(argv[1], &field, &size))
249 return CMD_RET_FAILURE;
250
251 if (*op == '=' && *(op + 1) == '\0') {
252 if (!strncmp(argv[3], field, size))
253 return CMD_RET_SUCCESS;
254 else
255 return CMD_RET_FAILURE;
256 } else if (*op == '~' && *(op + 1) == '\0') {
257 if (!strstr(field, argv[3]))
258 return CMD_RET_FAILURE;
259 else
260 return CMD_RET_SUCCESS;
261 } else {
262 printf("Error: Unknown operator '%s'\n", op);
263 }
264
265 return CMD_RET_FAILURE;
266}
267
09140113
SG
268static int do_bcb_dump(struct cmd_tbl *cmdtp, int flag, int argc,
269 char *const argv[])
db7b7a05
ER
270{
271 int size;
272 char *field;
273
274 if (bcb_field_get(argv[1], &field, &size))
275 return CMD_RET_FAILURE;
276
277 print_buffer((ulong)field - (ulong)&bcb, (void *)field, 1, size, 16);
278
279 return CMD_RET_SUCCESS;
280}
281
e2ffb3be 282static int __bcb_store(void)
db7b7a05
ER
283{
284 struct blk_desc *desc;
0528979f 285 struct disk_partition info;
db7b7a05
ER
286 u64 cnt;
287 int ret;
288
289 desc = blk_get_devnum_by_type(IF_TYPE_MMC, bcb_dev);
290 if (!desc) {
291 ret = -ENODEV;
292 goto err;
293 }
294
295 ret = part_get_info(desc, bcb_part, &info);
296 if (ret)
297 goto err;
298
299 cnt = DIV_ROUND_UP(sizeof(struct bootloader_message), info.blksz);
300
301 if (blk_dwrite(desc, info.start, cnt, &bcb) != cnt) {
302 ret = -EIO;
303 goto err;
304 }
305
306 return CMD_RET_SUCCESS;
307err:
308 printf("Error: mmc %d:%d write failed (%d)\n", bcb_dev, bcb_part, ret);
309
310 return CMD_RET_FAILURE;
311}
312
e2ffb3be
ER
313static int do_bcb_store(struct cmd_tbl *cmdtp, int flag, int argc,
314 char * const argv[])
315{
316 return __bcb_store();
317}
318
bafdf4ca 319int bcb_write_reboot_reason(int devnum, char *partp, const char *reasonp)
e74670ad
ER
320{
321 int ret;
322
323 ret = __bcb_load(devnum, partp);
324 if (ret != CMD_RET_SUCCESS)
325 return ret;
326
327 ret = __bcb_set("command", reasonp);
328 if (ret != CMD_RET_SUCCESS)
329 return ret;
330
331 ret = __bcb_store();
332 if (ret != CMD_RET_SUCCESS)
333 return ret;
334
335 return 0;
336}
337
09140113 338static struct cmd_tbl cmd_bcb_sub[] = {
db7b7a05
ER
339 U_BOOT_CMD_MKENT(load, CONFIG_SYS_MAXARGS, 1, do_bcb_load, "", ""),
340 U_BOOT_CMD_MKENT(set, CONFIG_SYS_MAXARGS, 1, do_bcb_set, "", ""),
341 U_BOOT_CMD_MKENT(clear, CONFIG_SYS_MAXARGS, 1, do_bcb_clear, "", ""),
342 U_BOOT_CMD_MKENT(test, CONFIG_SYS_MAXARGS, 1, do_bcb_test, "", ""),
343 U_BOOT_CMD_MKENT(dump, CONFIG_SYS_MAXARGS, 1, do_bcb_dump, "", ""),
344 U_BOOT_CMD_MKENT(store, CONFIG_SYS_MAXARGS, 1, do_bcb_store, "", ""),
345};
346
09140113 347static int do_bcb(struct cmd_tbl *cmdtp, int flag, int argc, char *const argv[])
db7b7a05 348{
09140113 349 struct cmd_tbl *c;
db7b7a05
ER
350
351 if (argc < 2)
352 return CMD_RET_USAGE;
353
354 argc--;
355 argv++;
356
357 c = find_cmd_tbl(argv[0], cmd_bcb_sub, ARRAY_SIZE(cmd_bcb_sub));
358 if (!c)
359 return CMD_RET_USAGE;
360
361 if (bcb_is_misused(argc, argv)) {
37f1811a
ER
362 /*
363 * We try to improve the user experience by reporting the
db7b7a05
ER
364 * root-cause of misusage, so don't return CMD_RET_USAGE,
365 * since the latter prints out the full-blown help text
366 */
367 return CMD_RET_FAILURE;
368 }
369
370 return c->cmd(cmdtp, flag, argc, argv);
371}
372
373U_BOOT_CMD(
374 bcb, CONFIG_SYS_MAXARGS, 1, do_bcb,
375 "Load/set/clear/test/dump/store Android BCB fields",
376 "load <dev> <part> - load BCB from mmc <dev>:<part>\n"
377 "bcb set <field> <val> - set BCB <field> to <val>\n"
378 "bcb clear [<field>] - clear BCB <field> or all fields\n"
379 "bcb test <field> <op> <val> - test BCB <field> against <val>\n"
380 "bcb dump <field> - dump BCB <field>\n"
381 "bcb store - store BCB back to mmc\n"
382 "\n"
383 "Legend:\n"
384 "<dev> - MMC device index containing the BCB partition\n"
385 "<part> - MMC partition index or name containing the BCB\n"
386 "<field> - one of {command,status,recovery,stage,reserved}\n"
387 "<op> - the binary operator used in 'bcb test':\n"
388 " '=' returns true if <val> matches the string stored in <field>\n"
389 " '~' returns true if <val> matches a subset of <field>'s string\n"
390 "<val> - string/text provided as input to bcb {set,test}\n"
391 " NOTE: any ':' character in <val> will be replaced by line feed\n"
392 " during 'bcb set' and used as separator by upper layers\n"
393);
This page took 0.145368 seconds and 4 git commands to generate.