1*552a848eSStefano Babic /*
2*552a848eSStefano Babic * Copyright (C) 2012 Boundary Devices Inc.
3*552a848eSStefano Babic *
4*552a848eSStefano Babic * SPDX-License-Identifier: GPL-2.0+
5*552a848eSStefano Babic */
6*552a848eSStefano Babic #include <common.h>
7*552a848eSStefano Babic #include <linux/errno.h>
8*552a848eSStefano Babic #include <asm/io.h>
9*552a848eSStefano Babic #include <asm/mach-imx/boot_mode.h>
10*552a848eSStefano Babic #include <malloc.h>
11*552a848eSStefano Babic #include <command.h>
12*552a848eSStefano Babic
13*552a848eSStefano Babic static const struct boot_mode *modes[2];
14*552a848eSStefano Babic
search_modes(char * arg)15*552a848eSStefano Babic static const struct boot_mode *search_modes(char *arg)
16*552a848eSStefano Babic {
17*552a848eSStefano Babic int i;
18*552a848eSStefano Babic
19*552a848eSStefano Babic for (i = 0; i < ARRAY_SIZE(modes); i++) {
20*552a848eSStefano Babic const struct boot_mode *p = modes[i];
21*552a848eSStefano Babic if (p) {
22*552a848eSStefano Babic while (p->name) {
23*552a848eSStefano Babic if (!strcmp(p->name, arg))
24*552a848eSStefano Babic return p;
25*552a848eSStefano Babic p++;
26*552a848eSStefano Babic }
27*552a848eSStefano Babic }
28*552a848eSStefano Babic }
29*552a848eSStefano Babic return NULL;
30*552a848eSStefano Babic }
31*552a848eSStefano Babic
create_usage(char * dest)32*552a848eSStefano Babic static int create_usage(char *dest)
33*552a848eSStefano Babic {
34*552a848eSStefano Babic int i;
35*552a848eSStefano Babic int size = 0;
36*552a848eSStefano Babic
37*552a848eSStefano Babic for (i = 0; i < ARRAY_SIZE(modes); i++) {
38*552a848eSStefano Babic const struct boot_mode *p = modes[i];
39*552a848eSStefano Babic if (p) {
40*552a848eSStefano Babic while (p->name) {
41*552a848eSStefano Babic int len = strlen(p->name);
42*552a848eSStefano Babic if (dest) {
43*552a848eSStefano Babic memcpy(dest, p->name, len);
44*552a848eSStefano Babic dest += len;
45*552a848eSStefano Babic *dest++ = '|';
46*552a848eSStefano Babic }
47*552a848eSStefano Babic size += len + 1;
48*552a848eSStefano Babic p++;
49*552a848eSStefano Babic }
50*552a848eSStefano Babic }
51*552a848eSStefano Babic }
52*552a848eSStefano Babic if (dest)
53*552a848eSStefano Babic memcpy(dest - 1, " [noreset]", 11); /* include trailing 0 */
54*552a848eSStefano Babic size += 10;
55*552a848eSStefano Babic return size;
56*552a848eSStefano Babic }
57*552a848eSStefano Babic
do_boot_mode(cmd_tbl_t * cmdtp,int flag,int argc,char * const argv[])58*552a848eSStefano Babic static int do_boot_mode(cmd_tbl_t *cmdtp, int flag, int argc,
59*552a848eSStefano Babic char * const argv[])
60*552a848eSStefano Babic {
61*552a848eSStefano Babic const struct boot_mode *p;
62*552a848eSStefano Babic int reset_requested = 1;
63*552a848eSStefano Babic
64*552a848eSStefano Babic if (argc < 2)
65*552a848eSStefano Babic return CMD_RET_USAGE;
66*552a848eSStefano Babic p = search_modes(argv[1]);
67*552a848eSStefano Babic if (!p)
68*552a848eSStefano Babic return CMD_RET_USAGE;
69*552a848eSStefano Babic if (argc == 3) {
70*552a848eSStefano Babic if (strcmp(argv[2], "noreset"))
71*552a848eSStefano Babic return CMD_RET_USAGE;
72*552a848eSStefano Babic reset_requested = 0;
73*552a848eSStefano Babic }
74*552a848eSStefano Babic
75*552a848eSStefano Babic boot_mode_apply(p->cfg_val);
76*552a848eSStefano Babic if (reset_requested && p->cfg_val)
77*552a848eSStefano Babic do_reset(NULL, 0, 0, NULL);
78*552a848eSStefano Babic return 0;
79*552a848eSStefano Babic }
80*552a848eSStefano Babic
81*552a848eSStefano Babic U_BOOT_CMD(
82*552a848eSStefano Babic bmode, 3, 0, do_boot_mode,
83*552a848eSStefano Babic NULL,
84*552a848eSStefano Babic "");
85*552a848eSStefano Babic
add_board_boot_modes(const struct boot_mode * p)86*552a848eSStefano Babic void add_board_boot_modes(const struct boot_mode *p)
87*552a848eSStefano Babic {
88*552a848eSStefano Babic int size;
89*552a848eSStefano Babic char *dest;
90*552a848eSStefano Babic
91*552a848eSStefano Babic cmd_tbl_t *entry = ll_entry_get(cmd_tbl_t, bmode, cmd);
92*552a848eSStefano Babic
93*552a848eSStefano Babic if (entry->usage) {
94*552a848eSStefano Babic free(entry->usage);
95*552a848eSStefano Babic entry->usage = NULL;
96*552a848eSStefano Babic }
97*552a848eSStefano Babic
98*552a848eSStefano Babic modes[0] = p;
99*552a848eSStefano Babic modes[1] = soc_boot_modes;
100*552a848eSStefano Babic size = create_usage(NULL);
101*552a848eSStefano Babic dest = malloc(size);
102*552a848eSStefano Babic if (dest) {
103*552a848eSStefano Babic create_usage(dest);
104*552a848eSStefano Babic entry->usage = dest;
105*552a848eSStefano Babic }
106*552a848eSStefano Babic }
107