blob: dd49d0e00aa996200b90d6a280754542c4835fc9 [file] [log] [blame]
Tom Rini83d290c2018-05-06 17:58:06 -04001// SPDX-License-Identifier: GPL-2.0+
Peter Tyser6b8f5ad2009-10-16 17:36:25 -05002/*
3 * Copyright 2000-2009
4 * Wolfgang Denk, DENX Software Engineering, wd@denx.de.
Peter Tyser6b8f5ad2009-10-16 17:36:25 -05005 */
6
7#include <common.h>
8#include <command.h>
Stephen Warrene5e897c2014-02-03 13:21:06 -07009#include <fs.h>
Peter Tyser6b8f5ad2009-10-16 17:36:25 -050010
Stephen Warren490ba832014-02-03 13:21:02 -070011#define OP_INVALID 0
Stephen Warrend9b651c2014-02-03 13:21:04 -070012#define OP_NOT 1
Stephen Warren490ba832014-02-03 13:21:02 -070013#define OP_OR 2
14#define OP_AND 3
15#define OP_STR_EMPTY 4
16#define OP_STR_NEMPTY 5
17#define OP_STR_EQ 6
18#define OP_STR_NEQ 7
19#define OP_STR_LT 8
20#define OP_STR_GT 9
21#define OP_INT_EQ 10
22#define OP_INT_NEQ 11
23#define OP_INT_LT 12
24#define OP_INT_LE 13
25#define OP_INT_GT 14
26#define OP_INT_GE 15
Stephen Warrene5e897c2014-02-03 13:21:06 -070027#define OP_FILE_EXISTS 16
Stephen Warren490ba832014-02-03 13:21:02 -070028
29const struct {
30 int arg;
31 const char *str;
32 int op;
33 int adv;
34} op_adv[] = {
Stephen Warren490ba832014-02-03 13:21:02 -070035 {1, "=", OP_STR_EQ, 3},
36 {1, "!=", OP_STR_NEQ, 3},
37 {1, "<", OP_STR_LT, 3},
38 {1, ">", OP_STR_GT, 3},
39 {1, "-eq", OP_INT_EQ, 3},
40 {1, "-ne", OP_INT_NEQ, 3},
41 {1, "-lt", OP_INT_LT, 3},
42 {1, "-le", OP_INT_LE, 3},
43 {1, "-gt", OP_INT_GT, 3},
44 {1, "-ge", OP_INT_GE, 3},
Stephen Warrend9b651c2014-02-03 13:21:04 -070045 {0, "!", OP_NOT, 1},
Stephen Warren4c80f292014-02-03 13:21:03 -070046 {0, "-o", OP_OR, 1},
47 {0, "-a", OP_AND, 1},
48 {0, "-z", OP_STR_EMPTY, 2},
49 {0, "-n", OP_STR_NEMPTY, 2},
Stephen Warrene5e897c2014-02-03 13:21:06 -070050 {0, "-e", OP_FILE_EXISTS, 4},
Stephen Warren490ba832014-02-03 13:21:02 -070051};
52
Simon Glass09140112020-05-10 11:40:03 -060053static int do_test(struct cmd_tbl *cmdtp, int flag, int argc,
54 char *const argv[])
Peter Tyser6b8f5ad2009-10-16 17:36:25 -050055{
Wolfgang Denk54841ab2010-06-28 22:00:46 +020056 char * const *ap;
Stephen Warrend9b651c2014-02-03 13:21:04 -070057 int i, op, left, adv, expr, last_expr, last_unop, last_binop;
Peter Tyser6b8f5ad2009-10-16 17:36:25 -050058
59 /* args? */
60 if (argc < 3)
61 return 1;
62
Joe Hershbergerb0fe6ab2012-08-17 10:56:12 +000063#ifdef DEBUG
Peter Tyser6b8f5ad2009-10-16 17:36:25 -050064 {
Joe Hershbergerb0fe6ab2012-08-17 10:56:12 +000065 debug("test(%d):", argc);
Peter Tyser6b8f5ad2009-10-16 17:36:25 -050066 left = 1;
67 while (argv[left])
Joe Hershbergerb0fe6ab2012-08-17 10:56:12 +000068 debug(" '%s'", argv[left++]);
Peter Tyser6b8f5ad2009-10-16 17:36:25 -050069 }
70#endif
71
Stephen Warrend9b651c2014-02-03 13:21:04 -070072 left = argc - 1;
73 ap = argv + 1;
Stephen Warren2453de92014-02-03 13:21:05 -070074 expr = 0;
Stephen Warrend9b651c2014-02-03 13:21:04 -070075 last_unop = OP_INVALID;
76 last_binop = OP_INVALID;
Peter Tyser6b8f5ad2009-10-16 17:36:25 -050077 last_expr = -1;
78 while (left > 0) {
Stephen Warren490ba832014-02-03 13:21:02 -070079 for (i = 0; i < ARRAY_SIZE(op_adv); i++) {
80 if (left <= op_adv[i].arg)
81 continue;
82 if (!strcmp(ap[op_adv[i].arg], op_adv[i].str)) {
83 op = op_adv[i].op;
84 adv = op_adv[i].adv;
85 break;
86 }
87 }
88 if (i == ARRAY_SIZE(op_adv)) {
89 expr = 1;
90 break;
91 }
Peter Tyser6b8f5ad2009-10-16 17:36:25 -050092 if (left < adv) {
93 expr = 1;
94 break;
95 }
96
Stephen Warren490ba832014-02-03 13:21:02 -070097 switch (op) {
98 case OP_STR_EMPTY:
99 expr = strlen(ap[1]) == 0 ? 1 : 0;
100 break;
101 case OP_STR_NEMPTY:
102 expr = strlen(ap[1]) == 0 ? 0 : 1;
103 break;
104 case OP_STR_EQ:
105 expr = strcmp(ap[0], ap[2]) == 0;
106 break;
107 case OP_STR_NEQ:
108 expr = strcmp(ap[0], ap[2]) != 0;
109 break;
110 case OP_STR_LT:
111 expr = strcmp(ap[0], ap[2]) < 0;
112 break;
113 case OP_STR_GT:
114 expr = strcmp(ap[0], ap[2]) > 0;
115 break;
116 case OP_INT_EQ:
T Karthik Reddy100e75b2019-09-11 15:39:53 +0200117 expr = simple_strtol(ap[0], NULL, 0) ==
118 simple_strtol(ap[2], NULL, 0);
Stephen Warren490ba832014-02-03 13:21:02 -0700119 break;
120 case OP_INT_NEQ:
T Karthik Reddy100e75b2019-09-11 15:39:53 +0200121 expr = simple_strtol(ap[0], NULL, 0) !=
122 simple_strtol(ap[2], NULL, 0);
Stephen Warren490ba832014-02-03 13:21:02 -0700123 break;
124 case OP_INT_LT:
T Karthik Reddy100e75b2019-09-11 15:39:53 +0200125 expr = simple_strtol(ap[0], NULL, 0) <
126 simple_strtol(ap[2], NULL, 0);
Stephen Warren490ba832014-02-03 13:21:02 -0700127 break;
128 case OP_INT_LE:
T Karthik Reddy100e75b2019-09-11 15:39:53 +0200129 expr = simple_strtol(ap[0], NULL, 0) <=
130 simple_strtol(ap[2], NULL, 0);
Stephen Warren490ba832014-02-03 13:21:02 -0700131 break;
132 case OP_INT_GT:
T Karthik Reddy100e75b2019-09-11 15:39:53 +0200133 expr = simple_strtol(ap[0], NULL, 0) >
134 simple_strtol(ap[2], NULL, 0);
Stephen Warren490ba832014-02-03 13:21:02 -0700135 break;
136 case OP_INT_GE:
T Karthik Reddy100e75b2019-09-11 15:39:53 +0200137 expr = simple_strtol(ap[0], NULL, 0) >=
138 simple_strtol(ap[2], NULL, 0);
Stephen Warren490ba832014-02-03 13:21:02 -0700139 break;
Stephen Warrene5e897c2014-02-03 13:21:06 -0700140 case OP_FILE_EXISTS:
141 expr = file_exists(ap[1], ap[2], ap[3], FS_TYPE_ANY);
142 break;
Peter Tyser6b8f5ad2009-10-16 17:36:25 -0500143 }
144
Stephen Warren490ba832014-02-03 13:21:02 -0700145 switch (op) {
146 case OP_OR:
147 last_expr = expr;
Stephen Warrend9b651c2014-02-03 13:21:04 -0700148 last_binop = OP_OR;
Stephen Warren490ba832014-02-03 13:21:02 -0700149 break;
150 case OP_AND:
151 last_expr = expr;
Stephen Warrend9b651c2014-02-03 13:21:04 -0700152 last_binop = OP_AND;
153 break;
154 case OP_NOT:
155 if (last_unop == OP_NOT)
156 last_unop = OP_INVALID;
157 else
158 last_unop = OP_NOT;
Stephen Warren490ba832014-02-03 13:21:02 -0700159 break;
160 default:
Stephen Warrend9b651c2014-02-03 13:21:04 -0700161 if (last_unop == OP_NOT) {
162 expr = !expr;
163 last_unop = OP_INVALID;
164 }
165
166 if (last_binop == OP_OR)
Peter Tyser6b8f5ad2009-10-16 17:36:25 -0500167 expr = last_expr || expr;
Stephen Warrend9b651c2014-02-03 13:21:04 -0700168 else if (last_binop == OP_AND)
Peter Tyser6b8f5ad2009-10-16 17:36:25 -0500169 expr = last_expr && expr;
Stephen Warrend9b651c2014-02-03 13:21:04 -0700170 last_binop = OP_INVALID;
171
Stephen Warren490ba832014-02-03 13:21:02 -0700172 break;
Peter Tyser6b8f5ad2009-10-16 17:36:25 -0500173 }
174
175 ap += adv; left -= adv;
176 }
177
Peter Tyser6b8f5ad2009-10-16 17:36:25 -0500178 expr = !expr;
179
180 debug (": returns %d\n", expr);
181
182 return expr;
183}
184
Heiko Schocher09c32802015-04-27 07:42:05 +0200185#undef true
186#undef false
187
Peter Tyser6b8f5ad2009-10-16 17:36:25 -0500188U_BOOT_CMD(
189 test, CONFIG_SYS_MAXARGS, 1, do_test,
190 "minimal test like /bin/sh",
191 "[args..]"
192);
Peter Tyser396fd172009-10-16 17:36:27 -0500193
Simon Glass09140112020-05-10 11:40:03 -0600194static int do_false(struct cmd_tbl *cmdtp, int flag, int argc,
195 char *const argv[])
Peter Tyser396fd172009-10-16 17:36:27 -0500196{
197 return 1;
198}
199
200U_BOOT_CMD(
201 false, CONFIG_SYS_MAXARGS, 1, do_false,
202 "do nothing, unsuccessfully",
203 NULL
204);
205
Simon Glass09140112020-05-10 11:40:03 -0600206static int do_true(struct cmd_tbl *cmdtp, int flag, int argc,
207 char *const argv[])
Peter Tyser396fd172009-10-16 17:36:27 -0500208{
209 return 0;
210}
211
212U_BOOT_CMD(
213 true, CONFIG_SYS_MAXARGS, 1, do_true,
214 "do nothing, successfully",
215 NULL
216);