]>
Commit | Line | Data |
---|---|---|
a72007d9 SG |
1 | /* |
2 | * Copyright (c) 2012, The Chromium Authors | |
3 | * | |
1a459660 | 4 | * SPDX-License-Identifier: GPL-2.0+ |
a72007d9 SG |
5 | */ |
6 | ||
7 | #define DEBUG | |
8 | ||
9 | #include <common.h> | |
cfd13e8d SW |
10 | #ifdef CONFIG_SANDBOX |
11 | #include <os.h> | |
12 | #endif | |
a72007d9 SG |
13 | |
14 | static const char test_cmd[] = "setenv list 1\n setenv list ${list}2; " | |
15 | "setenv list ${list}3\0" | |
16 | "setenv list ${list}4"; | |
17 | ||
18 | static int do_ut_cmd(cmd_tbl_t *cmdtp, int flag, int argc, char * const argv[]) | |
19 | { | |
20 | printf("%s: Testing commands\n", __func__); | |
67486728 | 21 | run_command("env default -f -a", 0); |
a72007d9 SG |
22 | |
23 | /* run a single command */ | |
24 | run_command("setenv single 1", 0); | |
25 | assert(!strcmp("1", getenv("single"))); | |
26 | ||
27 | /* make sure that compound statements work */ | |
28 | #ifdef CONFIG_SYS_HUSH_PARSER | |
29 | run_command("if test -n ${single} ; then setenv check 1; fi", 0); | |
30 | assert(!strcmp("1", getenv("check"))); | |
31 | run_command("setenv check", 0); | |
32 | #endif | |
33 | ||
34 | /* commands separated by ; */ | |
35 | run_command_list("setenv list 1; setenv list ${list}1", -1, 0); | |
36 | assert(!strcmp("11", getenv("list"))); | |
37 | ||
38 | /* commands separated by \n */ | |
39 | run_command_list("setenv list 1\n setenv list ${list}1", -1, 0); | |
40 | assert(!strcmp("11", getenv("list"))); | |
41 | ||
42 | /* command followed by \n and nothing else */ | |
43 | run_command_list("setenv list 1${list}\n", -1, 0); | |
44 | assert(!strcmp("111", getenv("list"))); | |
45 | ||
46 | /* three commands in a row */ | |
47 | run_command_list("setenv list 1\n setenv list ${list}2; " | |
48 | "setenv list ${list}3", -1, 0); | |
49 | assert(!strcmp("123", getenv("list"))); | |
50 | ||
51 | /* a command string with \0 in it. Stuff after \0 should be ignored */ | |
52 | run_command("setenv list", 0); | |
53 | run_command_list(test_cmd, sizeof(test_cmd), 0); | |
54 | assert(!strcmp("123", getenv("list"))); | |
55 | ||
56 | /* | |
57 | * a command list where we limit execution to only the first command | |
58 | * using the length parameter. | |
59 | */ | |
60 | run_command_list("setenv list 1\n setenv list ${list}2; " | |
61 | "setenv list ${list}3", strlen("setenv list 1"), 0); | |
62 | assert(!strcmp("1", getenv("list"))); | |
63 | ||
93ce7561 SG |
64 | assert(run_command("false", 0) == 1); |
65 | assert(run_command("echo", 0) == 0); | |
66 | assert(run_command_list("false", -1, 0) == 1); | |
67 | assert(run_command_list("echo", -1, 0) == 0); | |
68 | ||
87b6398b SG |
69 | run_command("setenv foo 'setenv monty 1; setenv python 2'", 0); |
70 | run_command("run foo", 0); | |
71 | assert(getenv("monty") != NULL); | |
72 | assert(!strcmp("1", getenv("monty"))); | |
73 | assert(getenv("python") != NULL); | |
74 | assert(!strcmp("2", getenv("python"))); | |
75 | ||
f2afe701 | 76 | #ifdef CONFIG_SYS_HUSH_PARSER |
87b6398b SG |
77 | run_command("setenv foo 'setenv black 1\nsetenv adder 2'", 0); |
78 | run_command("run foo", 0); | |
79 | assert(getenv("black") != NULL); | |
80 | assert(!strcmp("1", getenv("black"))); | |
81 | assert(getenv("adder") != NULL); | |
82 | assert(!strcmp("2", getenv("adder"))); | |
83 | ||
f2afe701 SW |
84 | /* Test the 'test' command */ |
85 | ||
86 | #define HUSH_TEST(name, expr, expected_result) \ | |
87 | run_command("if test " expr " ; then " \ | |
88 | "setenv " #name "_" #expected_result " y; else " \ | |
89 | "setenv " #name "_" #expected_result " n; fi", 0); \ | |
eebd1b58 SW |
90 | assert(!strcmp(#expected_result, getenv(#name "_" #expected_result))); \ |
91 | setenv(#name "_" #expected_result, NULL); | |
f2afe701 SW |
92 | |
93 | /* Basic operators */ | |
94 | HUSH_TEST(streq, "aaa = aaa", y); | |
95 | HUSH_TEST(streq, "aaa = bbb", n); | |
96 | ||
97 | HUSH_TEST(strneq, "aaa != bbb", y); | |
98 | HUSH_TEST(strneq, "aaa != aaa", n); | |
99 | ||
100 | HUSH_TEST(strlt, "aaa < bbb", y); | |
101 | HUSH_TEST(strlt, "bbb < aaa", n); | |
102 | ||
103 | HUSH_TEST(strgt, "bbb > aaa", y); | |
104 | HUSH_TEST(strgt, "aaa > bbb", n); | |
105 | ||
106 | HUSH_TEST(eq, "123 -eq 123", y); | |
107 | HUSH_TEST(eq, "123 -eq 456", n); | |
108 | ||
109 | HUSH_TEST(ne, "123 -ne 456", y); | |
110 | HUSH_TEST(ne, "123 -ne 123", n); | |
111 | ||
112 | HUSH_TEST(lt, "123 -lt 456", y); | |
113 | HUSH_TEST(lt_eq, "123 -lt 123", n); | |
114 | HUSH_TEST(lt, "456 -lt 123", n); | |
115 | ||
116 | HUSH_TEST(le, "123 -le 456", y); | |
117 | HUSH_TEST(le_eq, "123 -le 123", y); | |
118 | HUSH_TEST(le, "456 -le 123", n); | |
119 | ||
120 | HUSH_TEST(gt, "456 -gt 123", y); | |
121 | HUSH_TEST(gt_eq, "123 -gt 123", n); | |
122 | HUSH_TEST(gt, "123 -gt 456", n); | |
123 | ||
124 | HUSH_TEST(ge, "456 -ge 123", y); | |
125 | HUSH_TEST(ge_eq, "123 -ge 123", y); | |
126 | HUSH_TEST(ge, "123 -ge 456", n); | |
127 | ||
128 | HUSH_TEST(z, "-z \"\"", y); | |
129 | HUSH_TEST(z, "-z \"aaa\"", n); | |
130 | ||
131 | HUSH_TEST(n, "-n \"aaa\"", y); | |
132 | HUSH_TEST(n, "-n \"\"", n); | |
133 | ||
134 | /* Inversion of simple tests */ | |
135 | HUSH_TEST(streq_inv, "! aaa = aaa", n); | |
136 | HUSH_TEST(streq_inv, "! aaa = bbb", y); | |
137 | ||
138 | HUSH_TEST(streq_inv_inv, "! ! aaa = aaa", y); | |
139 | HUSH_TEST(streq_inv_inv, "! ! aaa = bbb", n); | |
140 | ||
141 | /* Binary operators */ | |
142 | HUSH_TEST(or_0_0, "aaa != aaa -o bbb != bbb", n); | |
143 | HUSH_TEST(or_0_1, "aaa != aaa -o bbb = bbb", y); | |
144 | HUSH_TEST(or_1_0, "aaa = aaa -o bbb != bbb", y); | |
145 | HUSH_TEST(or_1_1, "aaa = aaa -o bbb = bbb", y); | |
146 | ||
147 | HUSH_TEST(and_0_0, "aaa != aaa -a bbb != bbb", n); | |
148 | HUSH_TEST(and_0_1, "aaa != aaa -a bbb = bbb", n); | |
149 | HUSH_TEST(and_1_0, "aaa = aaa -a bbb != bbb", n); | |
150 | HUSH_TEST(and_1_1, "aaa = aaa -a bbb = bbb", y); | |
151 | ||
152 | /* Inversion within binary operators */ | |
153 | HUSH_TEST(or_0_0_inv, "! aaa != aaa -o ! bbb != bbb", y); | |
154 | HUSH_TEST(or_0_1_inv, "! aaa != aaa -o ! bbb = bbb", y); | |
155 | HUSH_TEST(or_1_0_inv, "! aaa = aaa -o ! bbb != bbb", y); | |
156 | HUSH_TEST(or_1_1_inv, "! aaa = aaa -o ! bbb = bbb", n); | |
157 | ||
158 | HUSH_TEST(or_0_0_inv_inv, "! ! aaa != aaa -o ! ! bbb != bbb", n); | |
159 | HUSH_TEST(or_0_1_inv_inv, "! ! aaa != aaa -o ! ! bbb = bbb", y); | |
160 | HUSH_TEST(or_1_0_inv_inv, "! ! aaa = aaa -o ! ! bbb != bbb", y); | |
161 | HUSH_TEST(or_1_1_inv_inv, "! ! aaa = aaa -o ! ! bbb = bbb", y); | |
162 | ||
fe9ca3d3 SW |
163 | setenv("ut_var_nonexistent", NULL); |
164 | setenv("ut_var_exists", "1"); | |
165 | HUSH_TEST(z_varexp_quoted, "-z \"$ut_var_nonexistent\"", y); | |
166 | HUSH_TEST(z_varexp_quoted, "-z \"$ut_var_exists\"", n); | |
167 | setenv("ut_var_exists", NULL); | |
168 | ||
169 | run_command("setenv ut_var_space \" \"", 0); | |
170 | assert(!strcmp(getenv("ut_var_space"), " ")); | |
171 | run_command("setenv ut_var_test $ut_var_space", 0); | |
172 | assert(!getenv("ut_var_test")); | |
173 | run_command("setenv ut_var_test \"$ut_var_space\"", 0); | |
174 | assert(!strcmp(getenv("ut_var_test"), " ")); | |
175 | run_command("setenv ut_var_test \" 1${ut_var_space}${ut_var_space} 2 \"", 0); | |
176 | assert(!strcmp(getenv("ut_var_test"), " 1 2 ")); | |
177 | setenv("ut_var_space", NULL); | |
178 | setenv("ut_var_test", NULL); | |
179 | ||
f2afe701 | 180 | #ifdef CONFIG_SANDBOX |
cfd13e8d | 181 | /* File existence */ |
4d907025 SW |
182 | HUSH_TEST(e, "-e hostfs - creating_this_file_breaks_uboot_unit_test", n); |
183 | run_command("sb save hostfs - creating_this_file_breaks_uboot_unit_test 0 1", 0); | |
184 | HUSH_TEST(e, "-e hostfs - creating_this_file_breaks_uboot_unit_test", y); | |
cfd13e8d SW |
185 | /* Perhaps this could be replaced by an "rm" shell command one day */ |
186 | assert(!os_unlink("creating_this_file_breaks_uboot_unit_test")); | |
4d907025 | 187 | HUSH_TEST(e, "-e hostfs - creating_this_file_breaks_uboot_unit_test", n); |
f2afe701 SW |
188 | #endif |
189 | #endif | |
190 | ||
484408fb RV |
191 | assert(run_command("", 0) == 0); |
192 | assert(run_command(" ", 0) == 0); | |
193 | ||
2302b3ab RV |
194 | assert(run_command("'", 0) == 1); |
195 | ||
a72007d9 SG |
196 | printf("%s: Everything went swimmingly\n", __func__); |
197 | return 0; | |
198 | } | |
199 | ||
200 | U_BOOT_CMD( | |
201 | ut_cmd, 5, 1, do_ut_cmd, | |
202 | "Very basic test of command parsers", | |
203 | "" | |
204 | ); |