3 * Tait Electronics Limited, Christchurch, New Zealand
5 * SPDX-License-Identifier: GPL-2.0+
9 * This file provides a shell like 'test' function to return
10 * true/false from an integer or string compare of two memory
11 * locations or a location and a scalar/literal.
12 * A few parts were lifted from bash 'test' command
30 char *op; /* operator string */
31 int opcode; /* internal representation of opcode */
34 typedef struct op_tbl_s op_tbl_t;
36 static const op_tbl_t op_table [] = {
52 static long evalexp(char *s, int w)
58 /* if the parameter starts with a * then assume is a pointer to the value we want */
60 addr = simple_strtoul(&s[1], NULL, 16);
61 buf = map_physmem(addr, w, MAP_WRBACK);
63 puts("Failed to map physical memory\n");
68 l = (long)(*(u8 *)buf);
71 l = (long)(*(u16 *)buf);
74 l = (long)(*(u32 *)buf);
77 unmap_physmem(buf, w);
80 l = simple_strtoul(s, NULL, 16);
83 /* avoid overflow on mask calculus */
84 return (w >= sizeof(long)) ? l : (l & ((1UL << (w * 8)) - 1));
87 static char * evalstr(char *s)
89 /* if the parameter starts with a * then assume a string pointer else its a literal */
91 return (char *)simple_strtoul(&s[1], NULL, 16);
92 } else if (s[0] == '$') {
104 return env_get((const char *)&s[2]);
110 static int stringcomp(char *s, char *t, int op)
120 case EQ: return (p == 0);
121 case NE: return (p != 0);
122 case LT: return (p < 0);
123 case GT: return (p > 0);
124 case LE: return (p <= 0);
125 case GE: return (p >= 0);
130 static int arithcomp (char *s, char *t, int op, int w)
138 case EQ: return (l == r);
139 case NE: return (l != r);
140 case LT: return (l < r);
141 case GT: return (l > r);
142 case LE: return (l <= r);
143 case GE: return (l >= r);
148 static int binary_test(char *op, char *arg1, char *arg2, int w)
151 const op_tbl_t *optp;
155 for (optp = (op_tbl_t *)&op_table, i = 0;
156 i < ARRAY_SIZE(op_table);
159 if ((strncmp (op, optp->op, len) == 0) && (len == strlen (optp->op))) {
161 return (stringcomp(arg1, arg2, optp->opcode));
163 return (arithcomp (arg1, arg2, optp->opcode, w));
168 printf("Unknown operator '%s'\n", op);
169 return 0; /* op code not found */
172 /* command line interface to the shell test */
173 static int do_itest(cmd_tbl_t *cmdtp, int flag, int argc, char * const argv[])
177 /* Validate arguments */
179 return CMD_RET_USAGE;
181 /* Check for a data width specification.
182 * Defaults to long (4) if no specification.
183 * Uses -2 as 'width' for .s (string) so as not to upset existing code
185 switch (w = cmd_get_data_size(argv[0], 4)) {
189 value = binary_test (argv[2], argv[1], argv[3], w);
192 value = binary_test (argv[2], argv[1], argv[3], 0);
196 puts("Invalid data width specifier\n");
205 itest, 4, 0, do_itest,
206 "return true/false on integer compare",
207 "[.b, .w, .l, .s] [*]value1 <op> [*]value2"