1 /* 2 * Copyright (c) 2011-2012 The Chromium OS Authors. 3 * SPDX-License-Identifier: GPL-2.0+ 4 */ 5 6 #include <common.h> 7 #include <os.h> 8 #include <cli.h> 9 #include <asm/getopt.h> 10 #include <asm/io.h> 11 #include <asm/sections.h> 12 #include <asm/state.h> 13 14 DECLARE_GLOBAL_DATA_PTR; 15 16 int sandbox_early_getopt_check(void) 17 { 18 struct sandbox_state *state = state_get_current(); 19 struct sandbox_cmdline_option **sb_opt = __u_boot_sandbox_option_start; 20 size_t num_options = __u_boot_sandbox_option_count(); 21 size_t i; 22 int max_arg_len, max_noarg_len; 23 24 /* parse_err will be a string of the faulting option */ 25 if (!state->parse_err) 26 return 0; 27 28 if (strcmp(state->parse_err, "help")) { 29 printf("u-boot: error: failed while parsing option: %s\n" 30 "\ttry running with --help for more information.\n", 31 state->parse_err); 32 os_exit(1); 33 } 34 35 printf( 36 "u-boot, a command line test interface to U-Boot\n\n" 37 "Usage: u-boot [options]\n" 38 "Options:\n"); 39 40 max_arg_len = 0; 41 for (i = 0; i < num_options; ++i) 42 max_arg_len = max((int)strlen(sb_opt[i]->flag), max_arg_len); 43 max_noarg_len = max_arg_len + 7; 44 45 for (i = 0; i < num_options; ++i) { 46 struct sandbox_cmdline_option *opt = sb_opt[i]; 47 48 /* first output the short flag if it has one */ 49 if (opt->flag_short >= 0x100) 50 printf(" "); 51 else 52 printf(" -%c, ", opt->flag_short); 53 54 /* then the long flag */ 55 if (opt->has_arg) 56 printf("--%-*s <arg> ", max_arg_len, opt->flag); 57 else 58 printf("--%-*s", max_noarg_len, opt->flag); 59 60 /* finally the help text */ 61 printf(" %s\n", opt->help); 62 } 63 64 os_exit(0); 65 } 66 67 static int sandbox_cmdline_cb_help(struct sandbox_state *state, const char *arg) 68 { 69 /* just flag to sandbox_early_getopt_check to show usage */ 70 return 1; 71 } 72 SANDBOX_CMDLINE_OPT_SHORT(help, 'h', 0, "Display help"); 73 74 int sandbox_main_loop_init(void) 75 { 76 struct sandbox_state *state = state_get_current(); 77 78 /* Execute command if required */ 79 if (state->cmd) { 80 cli_init(); 81 82 run_command_list(state->cmd, -1, 0); 83 if (!state->interactive) 84 os_exit(state->exit_type); 85 } 86 87 return 0; 88 } 89 90 static int sandbox_cmdline_cb_command(struct sandbox_state *state, 91 const char *arg) 92 { 93 state->cmd = arg; 94 return 0; 95 } 96 SANDBOX_CMDLINE_OPT_SHORT(command, 'c', 1, "Execute U-Boot command"); 97 98 static int sandbox_cmdline_cb_fdt(struct sandbox_state *state, const char *arg) 99 { 100 state->fdt_fname = arg; 101 return 0; 102 } 103 SANDBOX_CMDLINE_OPT_SHORT(fdt, 'd', 1, "Specify U-Boot's control FDT"); 104 105 static int sandbox_cmdline_cb_interactive(struct sandbox_state *state, 106 const char *arg) 107 { 108 state->interactive = true; 109 return 0; 110 } 111 112 SANDBOX_CMDLINE_OPT_SHORT(interactive, 'i', 0, "Enter interactive mode"); 113 114 static int sandbox_cmdline_cb_jump(struct sandbox_state *state, 115 const char *arg) 116 { 117 /* Remember to delete this U-Boot image later */ 118 state->jumped_fname = arg; 119 120 return 0; 121 } 122 SANDBOX_CMDLINE_OPT_SHORT(jump, 'j', 1, "Jumped from previous U-Boot"); 123 124 static int sandbox_cmdline_cb_memory(struct sandbox_state *state, 125 const char *arg) 126 { 127 int err; 128 129 /* For now assume we always want to write it */ 130 state->write_ram_buf = true; 131 state->ram_buf_fname = arg; 132 133 err = os_read_ram_buf(arg); 134 if (err) { 135 printf("Failed to read RAM buffer\n"); 136 return err; 137 } 138 139 return 0; 140 } 141 SANDBOX_CMDLINE_OPT_SHORT(memory, 'm', 1, 142 "Read/write ram_buf memory contents from file"); 143 144 static int sandbox_cmdline_cb_rm_memory(struct sandbox_state *state, 145 const char *arg) 146 { 147 state->ram_buf_rm = true; 148 149 return 0; 150 } 151 SANDBOX_CMDLINE_OPT(rm_memory, 0, "Remove memory file after reading"); 152 153 static int sandbox_cmdline_cb_state(struct sandbox_state *state, 154 const char *arg) 155 { 156 state->state_fname = arg; 157 return 0; 158 } 159 SANDBOX_CMDLINE_OPT_SHORT(state, 's', 1, "Specify the sandbox state FDT"); 160 161 static int sandbox_cmdline_cb_read(struct sandbox_state *state, 162 const char *arg) 163 { 164 state->read_state = true; 165 return 0; 166 } 167 SANDBOX_CMDLINE_OPT_SHORT(read, 'r', 0, "Read the state FDT on startup"); 168 169 static int sandbox_cmdline_cb_write(struct sandbox_state *state, 170 const char *arg) 171 { 172 state->write_state = true; 173 return 0; 174 } 175 SANDBOX_CMDLINE_OPT_SHORT(write, 'w', 0, "Write state FDT on exit"); 176 177 static int sandbox_cmdline_cb_ignore_missing(struct sandbox_state *state, 178 const char *arg) 179 { 180 state->ignore_missing_state_on_read = true; 181 return 0; 182 } 183 SANDBOX_CMDLINE_OPT_SHORT(ignore_missing, 'n', 0, 184 "Ignore missing state on read"); 185 186 static int sandbox_cmdline_cb_show_lcd(struct sandbox_state *state, 187 const char *arg) 188 { 189 state->show_lcd = true; 190 return 0; 191 } 192 SANDBOX_CMDLINE_OPT_SHORT(show_lcd, 'l', 0, 193 "Show the sandbox LCD display"); 194 195 static const char *term_args[STATE_TERM_COUNT] = { 196 "raw-with-sigs", 197 "raw", 198 "cooked", 199 }; 200 201 static int sandbox_cmdline_cb_terminal(struct sandbox_state *state, 202 const char *arg) 203 { 204 int i; 205 206 for (i = 0; i < STATE_TERM_COUNT; i++) { 207 if (!strcmp(arg, term_args[i])) { 208 state->term_raw = i; 209 return 0; 210 } 211 } 212 213 printf("Unknown terminal setting '%s' (", arg); 214 for (i = 0; i < STATE_TERM_COUNT; i++) 215 printf("%s%s", i ? ", " : "", term_args[i]); 216 puts(")\n"); 217 218 return 1; 219 } 220 SANDBOX_CMDLINE_OPT_SHORT(terminal, 't', 1, 221 "Set terminal to raw/cooked mode"); 222 223 int main(int argc, char *argv[]) 224 { 225 struct sandbox_state *state; 226 gd_t data; 227 int ret; 228 229 ret = state_init(); 230 if (ret) 231 goto err; 232 233 state = state_get_current(); 234 if (os_parse_args(state, argc, argv)) 235 return 1; 236 237 ret = sandbox_read_state(state, state->state_fname); 238 if (ret) 239 goto err; 240 241 /* Remove old memory file if required */ 242 if (state->ram_buf_rm && state->ram_buf_fname) 243 os_unlink(state->ram_buf_fname); 244 245 memset(&data, '\0', sizeof(data)); 246 gd = &data; 247 #ifdef CONFIG_SYS_MALLOC_F_LEN 248 gd->malloc_base = CONFIG_MALLOC_F_ADDR; 249 #endif 250 251 /* Do pre- and post-relocation init */ 252 board_init_f(0); 253 254 board_init_r(gd->new_gd, 0); 255 256 /* NOTREACHED - board_init_r() does not return */ 257 return 0; 258 259 err: 260 printf("Error %d\n", ret); 261 return 1; 262 } 263