1bace3d00SSimon Glass /* 26fb62078SSimon Glass * Copyright (c) 2011-2012 The Chromium OS Authors. 31a459660SWolfgang Denk * SPDX-License-Identifier: GPL-2.0+ 4bace3d00SSimon Glass */ 5bace3d00SSimon Glass 6bace3d00SSimon Glass #include <common.h> 7*5c2859cdSSimon Glass #include <os.h> 870db4212SSimon Glass #include <asm/getopt.h> 970db4212SSimon Glass #include <asm/sections.h> 106fb62078SSimon Glass #include <asm/state.h> 11bace3d00SSimon Glass 12808434cdSSimon Glass DECLARE_GLOBAL_DATA_PTR; 13808434cdSSimon Glass 1470db4212SSimon Glass int sandbox_early_getopt_check(void) 1570db4212SSimon Glass { 1670db4212SSimon Glass struct sandbox_state *state = state_get_current(); 177b3efc66SSimon Glass struct sandbox_cmdline_option **sb_opt = __u_boot_sandbox_option_start; 1870db4212SSimon Glass size_t num_options = __u_boot_sandbox_option_count(); 1970db4212SSimon Glass size_t i; 2070db4212SSimon Glass int max_arg_len, max_noarg_len; 2170db4212SSimon Glass 2270db4212SSimon Glass /* parse_err will be a string of the faulting option */ 2370db4212SSimon Glass if (!state->parse_err) 2470db4212SSimon Glass return 0; 2570db4212SSimon Glass 2670db4212SSimon Glass if (strcmp(state->parse_err, "help")) { 2770db4212SSimon Glass printf("u-boot: error: failed while parsing option: %s\n" 2870db4212SSimon Glass "\ttry running with --help for more information.\n", 2970db4212SSimon Glass state->parse_err); 3070db4212SSimon Glass os_exit(1); 3170db4212SSimon Glass } 3270db4212SSimon Glass 3370db4212SSimon Glass printf( 3470db4212SSimon Glass "u-boot, a command line test interface to U-Boot\n\n" 3570db4212SSimon Glass "Usage: u-boot [options]\n" 3670db4212SSimon Glass "Options:\n"); 3770db4212SSimon Glass 3870db4212SSimon Glass max_arg_len = 0; 3970db4212SSimon Glass for (i = 0; i < num_options; ++i) 4070db4212SSimon Glass max_arg_len = max(strlen(sb_opt[i]->flag), max_arg_len); 4170db4212SSimon Glass max_noarg_len = max_arg_len + 7; 4270db4212SSimon Glass 4370db4212SSimon Glass for (i = 0; i < num_options; ++i) { 447b3efc66SSimon Glass struct sandbox_cmdline_option *opt = sb_opt[i]; 4570db4212SSimon Glass 4670db4212SSimon Glass /* first output the short flag if it has one */ 4770db4212SSimon Glass if (opt->flag_short >= 0x100) 4870db4212SSimon Glass printf(" "); 4970db4212SSimon Glass else 5070db4212SSimon Glass printf(" -%c, ", opt->flag_short); 5170db4212SSimon Glass 5270db4212SSimon Glass /* then the long flag */ 5370db4212SSimon Glass if (opt->has_arg) 5470db4212SSimon Glass printf("--%-*s <arg> ", max_arg_len, opt->flag); 556ebcab8dSSimon Glass else 566ebcab8dSSimon Glass printf("--%-*s", max_noarg_len, opt->flag); 5770db4212SSimon Glass 5870db4212SSimon Glass /* finally the help text */ 5970db4212SSimon Glass printf(" %s\n", opt->help); 6070db4212SSimon Glass } 6170db4212SSimon Glass 6270db4212SSimon Glass os_exit(0); 6370db4212SSimon Glass } 6470db4212SSimon Glass 657b3efc66SSimon Glass static int sandbox_cmdline_cb_help(struct sandbox_state *state, const char *arg) 6670db4212SSimon Glass { 6770db4212SSimon Glass /* just flag to sandbox_early_getopt_check to show usage */ 6870db4212SSimon Glass return 1; 6970db4212SSimon Glass } 707b3efc66SSimon Glass SANDBOX_CMDLINE_OPT_SHORT(help, 'h', 0, "Display help"); 7170db4212SSimon Glass 72ab4e07ebSSimon Glass int sandbox_main_loop_init(void) 73ab4e07ebSSimon Glass { 7470db4212SSimon Glass struct sandbox_state *state = state_get_current(); 7570db4212SSimon Glass 7670db4212SSimon Glass /* Execute command if required */ 7770db4212SSimon Glass if (state->cmd) { 7839042d82SSimon Glass run_command_list(state->cmd, -1, 0); 79c5a62d4aSSimon Glass if (!state->interactive) 8070db4212SSimon Glass os_exit(state->exit_type); 8170db4212SSimon Glass } 8270db4212SSimon Glass 83ab4e07ebSSimon Glass return 0; 84ab4e07ebSSimon Glass } 85ab4e07ebSSimon Glass 867b3efc66SSimon Glass static int sandbox_cmdline_cb_command(struct sandbox_state *state, 877b3efc66SSimon Glass const char *arg) 8870db4212SSimon Glass { 8970db4212SSimon Glass state->cmd = arg; 9070db4212SSimon Glass return 0; 9170db4212SSimon Glass } 927b3efc66SSimon Glass SANDBOX_CMDLINE_OPT_SHORT(command, 'c', 1, "Execute U-Boot command"); 9370db4212SSimon Glass 947b3efc66SSimon Glass static int sandbox_cmdline_cb_fdt(struct sandbox_state *state, const char *arg) 95f828bf25SSimon Glass { 96f828bf25SSimon Glass state->fdt_fname = arg; 97f828bf25SSimon Glass return 0; 98f828bf25SSimon Glass } 997b3efc66SSimon Glass SANDBOX_CMDLINE_OPT_SHORT(fdt, 'd', 1, "Specify U-Boot's control FDT"); 100f828bf25SSimon Glass 101c5a62d4aSSimon Glass static int sandbox_cmdline_cb_interactive(struct sandbox_state *state, 102c5a62d4aSSimon Glass const char *arg) 103c5a62d4aSSimon Glass { 104c5a62d4aSSimon Glass state->interactive = true; 105c5a62d4aSSimon Glass return 0; 106c5a62d4aSSimon Glass } 107c5a62d4aSSimon Glass 108c5a62d4aSSimon Glass SANDBOX_CMDLINE_OPT_SHORT(interactive, 'i', 0, "Enter interactive mode"); 109c5a62d4aSSimon Glass 110*5c2859cdSSimon Glass static int sandbox_cmdline_cb_memory(struct sandbox_state *state, 111*5c2859cdSSimon Glass const char *arg) 112*5c2859cdSSimon Glass { 113*5c2859cdSSimon Glass int err; 114*5c2859cdSSimon Glass 115*5c2859cdSSimon Glass /* For now assume we always want to write it */ 116*5c2859cdSSimon Glass state->write_ram_buf = true; 117*5c2859cdSSimon Glass state->ram_buf_fname = arg; 118*5c2859cdSSimon Glass 119*5c2859cdSSimon Glass if (os_read_ram_buf(arg)) { 120*5c2859cdSSimon Glass printf("Failed to read RAM buffer\n"); 121*5c2859cdSSimon Glass return err; 122*5c2859cdSSimon Glass } 123*5c2859cdSSimon Glass 124*5c2859cdSSimon Glass return 0; 125*5c2859cdSSimon Glass } 126*5c2859cdSSimon Glass SANDBOX_CMDLINE_OPT_SHORT(memory, 'm', 1, 127*5c2859cdSSimon Glass "Read/write ram_buf memory contents from file"); 128*5c2859cdSSimon Glass 129bace3d00SSimon Glass int main(int argc, char *argv[]) 130bace3d00SSimon Glass { 13170db4212SSimon Glass struct sandbox_state *state; 1326fb62078SSimon Glass int err; 1336fb62078SSimon Glass 1346fb62078SSimon Glass err = state_init(); 1356fb62078SSimon Glass if (err) 1366fb62078SSimon Glass return err; 1376fb62078SSimon Glass 13870db4212SSimon Glass state = state_get_current(); 13970db4212SSimon Glass if (os_parse_args(state, argc, argv)) 14070db4212SSimon Glass return 1; 14170db4212SSimon Glass 142808434cdSSimon Glass /* Do pre- and post-relocation init */ 143bace3d00SSimon Glass board_init_f(0); 1448ec21bbeSAllen Martin 145808434cdSSimon Glass board_init_r(gd->new_gd, 0); 146808434cdSSimon Glass 147808434cdSSimon Glass /* NOTREACHED - board_init_r() does not return */ 1488ec21bbeSAllen Martin return 0; 149bace3d00SSimon Glass } 150