blob: 91ae5c2704c882be8d4457d67d5f8488b19d2f1c [file] [log] [blame]
wdenk2d1a5372004-02-23 19:30:57 +00001/*
2 * (C) Copyright 2003
3 * Tait Electronics Limited, Christchurch, New Zealand
4 *
Wolfgang Denk3765b3e2013-10-07 13:07:26 +02005 * SPDX-License-Identifier: GPL-2.0+
wdenk2d1a5372004-02-23 19:30:57 +00006 */
7
8/*
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
13 */
14
15#include <common.h>
16#include <config.h>
17#include <command.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] == '*') {
Stephen Warren78612042015-10-03 13:56:46 -060060 addr = simple_strtoul(&s[1], NULL, 16);
61 buf = map_physmem(addr, w, MAP_WRBACK);
62 if (!buf) {
63 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:
68 l = (long)(*(unsigned char *)buf);
69 break;
70 case 2:
71 l = (long)(*(unsigned short *)buf);
72 break;
73 case 4:
74 l = (long)(*(unsigned long *)buf);
75 break;
Stephen Warren78612042015-10-03 13:56:46 -060076 }
77 unmap_physmem(buf, w);
78 return l;
wdenk2d1a5372004-02-23 19:30:57 +000079 } else {
80 l = simple_strtoul(s, NULL, 16);
81 }
82
Simon Glassc9bcb6f2014-05-30 14:41:49 -060083 return l & ((1UL << (w * 8)) - 1);
wdenk2d1a5372004-02-23 19:30:57 +000084}
85
86static char * evalstr(char *s)
87{
88 /* if the parameter starts with a * then assume a string pointer else its a literal */
89 if (s[0] == '*') {
90 return (char *)simple_strtoul(&s[1], NULL, 16);
Heiko Schocher06109f42014-01-25 07:27:12 +010091 } else if (s[0] == '$') {
92 int i = 2;
93
94 if (s[1] != '{')
95 return NULL;
96
97 while (s[i] != '}') {
98 if (s[i] == 0)
99 return NULL;
100 i++;
101 }
102 s[i] = 0;
103 return getenv((const char *)&s[2]);
wdenk2d1a5372004-02-23 19:30:57 +0000104 } else {
105 return s;
106 }
107}
108
109static int stringcomp(char *s, char *t, int op)
110{
Wolfgang Denkcc22b792011-02-08 16:56:05 +0100111 int p;
wdenk2d1a5372004-02-23 19:30:57 +0000112 char *l, *r;
113
114 l = evalstr(s);
115 r = evalstr(t);
116
Wolfgang Denkcc22b792011-02-08 16:56:05 +0100117 p = strcmp(l, r);
wdenk2d1a5372004-02-23 19:30:57 +0000118 switch (op) {
119 case EQ: return (p == 0);
120 case NE: return (p != 0);
121 case LT: return (p < 0);
122 case GT: return (p > 0);
123 case LE: return (p <= 0);
124 case GE: return (p >= 0);
125 }
126 return (0);
127}
128
129static int arithcomp (char *s, char *t, int op, int w)
130{
131 long l, r;
132
133 l = evalexp (s, w);
134 r = evalexp (t, w);
135
136 switch (op) {
137 case EQ: return (l == r);
138 case NE: return (l != r);
139 case LT: return (l < r);
140 case GT: return (l > r);
141 case LE: return (l <= r);
142 case GE: return (l >= r);
143 }
144 return (0);
145}
146
Kim Phillips088f1b12012-10-29 13:34:31 +0000147static int binary_test(char *op, char *arg1, char *arg2, int w)
wdenk2d1a5372004-02-23 19:30:57 +0000148{
149 int len, i;
Mike Frysingerfc9903f2010-10-20 07:17:33 -0400150 const op_tbl_t *optp;
wdenk2d1a5372004-02-23 19:30:57 +0000151
152 len = strlen(op);
153
154 for (optp = (op_tbl_t *)&op_table, i = 0;
Mike Frysingerfc9903f2010-10-20 07:17:33 -0400155 i < ARRAY_SIZE(op_table);
wdenk2d1a5372004-02-23 19:30:57 +0000156 optp++, i++) {
157
158 if ((strncmp (op, optp->op, len) == 0) && (len == strlen (optp->op))) {
159 if (w == 0) {
160 return (stringcomp(arg1, arg2, optp->opcode));
161 } else {
162 return (arithcomp (arg1, arg2, optp->opcode, w));
163 }
164 }
165 }
166
167 printf("Unknown operator '%s'\n", op);
168 return 0; /* op code not found */
169}
170
171/* command line interface to the shell test */
Kim Phillips088f1b12012-10-29 13:34:31 +0000172static int do_itest(cmd_tbl_t *cmdtp, int flag, int argc, char * const argv[])
wdenk2d1a5372004-02-23 19:30:57 +0000173{
wdenkcd0a9de2004-02-23 20:48:38 +0000174 int value, w;
wdenk2d1a5372004-02-23 19:30:57 +0000175
wdenkcd0a9de2004-02-23 20:48:38 +0000176 /* Validate arguments */
Wolfgang Denk47e26b12010-07-17 01:06:04 +0200177 if ((argc != 4))
Simon Glass4c12eeb2011-12-10 08:44:01 +0000178 return CMD_RET_USAGE;
wdenk2d1a5372004-02-23 19:30:57 +0000179
180 /* Check for a data width specification.
181 * Defaults to long (4) if no specification.
182 * Uses -2 as 'width' for .s (string) so as not to upset existing code
183 */
184 switch (w = cmd_get_data_size(argv[0], 4)) {
185 case 1:
186 case 2:
187 case 4:
188 value = binary_test (argv[2], argv[1], argv[3], w);
189 break;
190 case -2:
191 value = binary_test (argv[2], argv[1], argv[3], 0);
192 break;
193 case -1:
194 default:
195 puts("Invalid data width specifier\n");
196 value = 0;
197 break;
198 }
199
200 return !value;
201}
202
203U_BOOT_CMD(
204 itest, 4, 0, do_itest,
Peter Tyser2fb26042009-01-27 18:03:12 -0600205 "return true/false on integer compare",
Wolfgang Denka89c33d2009-05-24 17:06:54 +0200206 "[.b, .w, .l, .s] [*]value1 <op> [*]value2"
wdenk2d1a5372004-02-23 19:30:57 +0000207);