blob: 74414cbdc4c67e8dba32b533b1c563d5e182e542 [file] [log] [blame]
Tom Rini83d290c2018-05-06 17:58:06 -04001// SPDX-License-Identifier: GPL-2.0+
wdenk2d1a5372004-02-23 19:30:57 +00002/*
3 * (C) Copyright 2003
4 * Tait Electronics Limited, Christchurch, New Zealand
wdenk2d1a5372004-02-23 19:30:57 +00005 */
6
7/*
8 * This file provides a shell like 'test' function to return
9 * true/false from an integer or string compare of two memory
10 * locations or a location and a scalar/literal.
11 * A few parts were lifted from bash 'test' command
12 */
13
14#include <common.h>
15#include <config.h>
16#include <command.h>
Simon Glass7b51b572019-08-01 09:46:52 -060017#include <env.h>
Stephen Warren78612042015-10-03 13:56:46 -060018#include <mapmem.h>
19
20#include <asm/io.h>
wdenk2d1a5372004-02-23 19:30:57 +000021
22#define EQ 0
23#define NE 1
24#define LT 2
25#define GT 3
26#define LE 4
27#define GE 5
28
29struct op_tbl_s {
30 char *op; /* operator string */
31 int opcode; /* internal representation of opcode */
32};
33
34typedef struct op_tbl_s op_tbl_t;
35
Mike Frysingerfc9903f2010-10-20 07:17:33 -040036static const op_tbl_t op_table [] = {
wdenk2d1a5372004-02-23 19:30:57 +000037 { "-lt", LT },
38 { "<" , LT },
39 { "-gt", GT },
40 { ">" , GT },
41 { "-eq", EQ },
42 { "==" , EQ },
43 { "-ne", NE },
44 { "!=" , NE },
45 { "<>" , NE },
46 { "-ge", GE },
47 { ">=" , GE },
48 { "-le", LE },
49 { "<=" , LE },
50};
51
wdenk2d1a5372004-02-23 19:30:57 +000052static long evalexp(char *s, int w)
53{
Frans Meulenbroeksf3651762010-02-22 22:49:06 +010054 long l = 0;
Stephen Warren78612042015-10-03 13:56:46 -060055 unsigned long addr;
56 void *buf;
wdenk2d1a5372004-02-23 19:30:57 +000057
58 /* if the parameter starts with a * then assume is a pointer to the value we want */
59 if (s[0] == '*') {
Simon Glass7e5f4602021-07-24 09:03:29 -060060 addr = hextoul(&s[1], NULL);
Stephen Warren78612042015-10-03 13:56:46 -060061 buf = map_physmem(addr, w, MAP_WRBACK);
Stephen Warren986fe372016-01-29 16:10:04 -070062 if (!buf && addr) {
Stephen Warren78612042015-10-03 13:56:46 -060063 puts("Failed to map physical memory\n");
64 return 0;
Frans Meulenbroeksf3651762010-02-22 22:49:06 +010065 }
Stephen Warren78612042015-10-03 13:56:46 -060066 switch (w) {
Stephen Warren736d1742015-11-17 10:29:08 -070067 case 1:
Kunihiko Hayashidafd6482016-05-04 14:20:04 +090068 l = (long)(*(u8 *)buf);
Stephen Warren736d1742015-11-17 10:29:08 -070069 break;
70 case 2:
Kunihiko Hayashidafd6482016-05-04 14:20:04 +090071 l = (long)(*(u16 *)buf);
Stephen Warren736d1742015-11-17 10:29:08 -070072 break;
73 case 4:
Kunihiko Hayashidafd6482016-05-04 14:20:04 +090074 l = (long)(*(u32 *)buf);
Stephen Warren736d1742015-11-17 10:29:08 -070075 break;
Marek Szyprowski7d2dc6a2019-10-02 12:24:46 +020076#ifdef CONFIG_PHYS_64BIT
77 case 8:
78 l = (long)(*(unsigned long *)buf);
79 break;
80#endif
Stephen Warren78612042015-10-03 13:56:46 -060081 }
82 unmap_physmem(buf, w);
83 return l;
wdenk2d1a5372004-02-23 19:30:57 +000084 } else {
Simon Glass7e5f4602021-07-24 09:03:29 -060085 l = hextoul(s, NULL);
wdenk2d1a5372004-02-23 19:30:57 +000086 }
87
Sebastien Colleur2c79fd42017-02-10 15:59:15 +030088 /* avoid overflow on mask calculus */
89 return (w >= sizeof(long)) ? l : (l & ((1UL << (w * 8)) - 1));
wdenk2d1a5372004-02-23 19:30:57 +000090}
91
92static char * evalstr(char *s)
93{
94 /* if the parameter starts with a * then assume a string pointer else its a literal */
95 if (s[0] == '*') {
Simon Glass7e5f4602021-07-24 09:03:29 -060096 return (char *)hextoul(&s[1], NULL);
Heiko Schocher06109f42014-01-25 07:27:12 +010097 } else if (s[0] == '$') {
98 int i = 2;
99
100 if (s[1] != '{')
101 return NULL;
102
103 while (s[i] != '}') {
104 if (s[i] == 0)
105 return NULL;
106 i++;
107 }
108 s[i] = 0;
Simon Glass00caae62017-08-03 12:22:12 -0600109 return env_get((const char *)&s[2]);
wdenk2d1a5372004-02-23 19:30:57 +0000110 } else {
111 return s;
112 }
113}
114
115static int stringcomp(char *s, char *t, int op)
116{
Wolfgang Denkcc22b792011-02-08 16:56:05 +0100117 int p;
wdenk2d1a5372004-02-23 19:30:57 +0000118 char *l, *r;
119
120 l = evalstr(s);
121 r = evalstr(t);
122
Wolfgang Denkcc22b792011-02-08 16:56:05 +0100123 p = strcmp(l, r);
wdenk2d1a5372004-02-23 19:30:57 +0000124 switch (op) {
125 case EQ: return (p == 0);
126 case NE: return (p != 0);
127 case LT: return (p < 0);
128 case GT: return (p > 0);
129 case LE: return (p <= 0);
130 case GE: return (p >= 0);
131 }
132 return (0);
133}
134
135static int arithcomp (char *s, char *t, int op, int w)
136{
137 long l, r;
138
139 l = evalexp (s, w);
140 r = evalexp (t, w);
141
142 switch (op) {
143 case EQ: return (l == r);
144 case NE: return (l != r);
145 case LT: return (l < r);
146 case GT: return (l > r);
147 case LE: return (l <= r);
148 case GE: return (l >= r);
149 }
150 return (0);
151}
152
Kim Phillips088f1b12012-10-29 13:34:31 +0000153static int binary_test(char *op, char *arg1, char *arg2, int w)
wdenk2d1a5372004-02-23 19:30:57 +0000154{
155 int len, i;
Mike Frysingerfc9903f2010-10-20 07:17:33 -0400156 const op_tbl_t *optp;
wdenk2d1a5372004-02-23 19:30:57 +0000157
158 len = strlen(op);
159
160 for (optp = (op_tbl_t *)&op_table, i = 0;
Mike Frysingerfc9903f2010-10-20 07:17:33 -0400161 i < ARRAY_SIZE(op_table);
wdenk2d1a5372004-02-23 19:30:57 +0000162 optp++, i++) {
163
164 if ((strncmp (op, optp->op, len) == 0) && (len == strlen (optp->op))) {
165 if (w == 0) {
166 return (stringcomp(arg1, arg2, optp->opcode));
167 } else {
168 return (arithcomp (arg1, arg2, optp->opcode, w));
169 }
170 }
171 }
172
173 printf("Unknown operator '%s'\n", op);
174 return 0; /* op code not found */
175}
176
177/* command line interface to the shell test */
Simon Glass09140112020-05-10 11:40:03 -0600178static int do_itest(struct cmd_tbl *cmdtp, int flag, int argc,
179 char *const argv[])
wdenk2d1a5372004-02-23 19:30:57 +0000180{
wdenkcd0a9de2004-02-23 20:48:38 +0000181 int value, w;
wdenk2d1a5372004-02-23 19:30:57 +0000182
wdenkcd0a9de2004-02-23 20:48:38 +0000183 /* Validate arguments */
Wolfgang Denk47e26b12010-07-17 01:06:04 +0200184 if ((argc != 4))
Simon Glass4c12eeb2011-12-10 08:44:01 +0000185 return CMD_RET_USAGE;
wdenk2d1a5372004-02-23 19:30:57 +0000186
187 /* Check for a data width specification.
188 * Defaults to long (4) if no specification.
189 * Uses -2 as 'width' for .s (string) so as not to upset existing code
190 */
191 switch (w = cmd_get_data_size(argv[0], 4)) {
192 case 1:
193 case 2:
194 case 4:
Marek Szyprowski7d2dc6a2019-10-02 12:24:46 +0200195#ifdef CONFIG_PHYS_64BIT
196 case 8:
197#endif
wdenk2d1a5372004-02-23 19:30:57 +0000198 value = binary_test (argv[2], argv[1], argv[3], w);
199 break;
Simon Glass7526dee2020-11-01 14:15:36 -0700200 case CMD_DATA_SIZE_STR:
wdenk2d1a5372004-02-23 19:30:57 +0000201 value = binary_test (argv[2], argv[1], argv[3], 0);
202 break;
Simon Glass7526dee2020-11-01 14:15:36 -0700203 case CMD_DATA_SIZE_ERR:
wdenk2d1a5372004-02-23 19:30:57 +0000204 default:
205 puts("Invalid data width specifier\n");
206 value = 0;
207 break;
208 }
209
210 return !value;
211}
212
213U_BOOT_CMD(
214 itest, 4, 0, do_itest,
Peter Tyser2fb26042009-01-27 18:03:12 -0600215 "return true/false on integer compare",
Marek Szyprowski7d2dc6a2019-10-02 12:24:46 +0200216#ifdef CONFIG_PHYS_64BIT
217 "[.b, .w, .l, .q, .s] [*]value1 <op> [*]value2"
218#else
Wolfgang Denka89c33d2009-05-24 17:06:54 +0200219 "[.b, .w, .l, .s] [*]value1 <op> [*]value2"
Marek Szyprowski7d2dc6a2019-10-02 12:24:46 +0200220#endif
wdenk2d1a5372004-02-23 19:30:57 +0000221);