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