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