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