blob: d5df758a1f5442ce5e605697900e7a9719f973b2 [file] [log] [blame]
wdenk2d1a5372004-02-23 19:30:57 +00001/*
2 * (C) Copyright 2003
3 * Tait Electronics Limited, Christchurch, New Zealand
4 *
5 * See file CREDITS for list of people who contributed to this
6 * project.
7 *
8 * This program is free software; you can redistribute it and/or
9 * modify it under the terms of the GNU General Public License as
10 * published by the Free Software Foundation; either version 2 of
11 * the License, or (at your option) any later version.
12 *
13 * This program is distributed in the hope that it will be useful,
14 * but WITHOUT ANY WARRANTY; without even the implied warranty of
wdenkcd0a9de2004-02-23 20:48:38 +000015 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
wdenk2d1a5372004-02-23 19:30:57 +000016 * GNU General Public License for more details.
17 *
18 * You should have received a copy of the GNU General Public License
19 * along with this program; if not, write to the Free Software
20 * Foundation, Inc., 59 Temple Place, Suite 330, Boston,
21 * MA 02111-1307 USA
22 */
23
24/*
25 * This file provides a shell like 'test' function to return
26 * true/false from an integer or string compare of two memory
27 * locations or a location and a scalar/literal.
28 * A few parts were lifted from bash 'test' command
29 */
30
31#include <common.h>
32#include <config.h>
33#include <command.h>
34
35#define EQ 0
36#define NE 1
37#define LT 2
38#define GT 3
39#define LE 4
40#define GE 5
41
42struct op_tbl_s {
43 char *op; /* operator string */
44 int opcode; /* internal representation of opcode */
45};
46
47typedef struct op_tbl_s op_tbl_t;
48
Mike Frysingerfc9903f2010-10-20 07:17:33 -040049static const op_tbl_t op_table [] = {
wdenk2d1a5372004-02-23 19:30:57 +000050 { "-lt", LT },
51 { "<" , LT },
52 { "-gt", GT },
53 { ">" , GT },
54 { "-eq", EQ },
55 { "==" , EQ },
56 { "-ne", NE },
57 { "!=" , NE },
58 { "<>" , NE },
59 { "-ge", GE },
60 { ">=" , GE },
61 { "-le", LE },
62 { "<=" , LE },
63};
64
wdenk2d1a5372004-02-23 19:30:57 +000065static long evalexp(char *s, int w)
66{
Frans Meulenbroeksf3651762010-02-22 22:49:06 +010067 long l = 0;
68 long *p;
wdenk2d1a5372004-02-23 19:30:57 +000069
70 /* if the parameter starts with a * then assume is a pointer to the value we want */
71 if (s[0] == '*') {
72 p = (long *)simple_strtoul(&s[1], NULL, 16);
Frans Meulenbroeksf3651762010-02-22 22:49:06 +010073 switch (w) {
74 case 1: return((long)(*(unsigned char *)p));
75 case 2: return((long)(*(unsigned short *)p));
76 case 4: return(*p);
77 }
wdenk2d1a5372004-02-23 19:30:57 +000078 } else {
79 l = simple_strtoul(s, NULL, 16);
80 }
81
82 return (l & ((1 << (w * 8)) - 1));
83}
84
85static char * evalstr(char *s)
86{
87 /* if the parameter starts with a * then assume a string pointer else its a literal */
88 if (s[0] == '*') {
89 return (char *)simple_strtoul(&s[1], NULL, 16);
90 } else {
91 return s;
92 }
93}
94
95static int stringcomp(char *s, char *t, int op)
96{
Wolfgang Denkcc22b792011-02-08 16:56:05 +010097 int p;
wdenk2d1a5372004-02-23 19:30:57 +000098 char *l, *r;
99
100 l = evalstr(s);
101 r = evalstr(t);
102
Wolfgang Denkcc22b792011-02-08 16:56:05 +0100103 p = strcmp(l, r);
wdenk2d1a5372004-02-23 19:30:57 +0000104 switch (op) {
105 case EQ: return (p == 0);
106 case NE: return (p != 0);
107 case LT: return (p < 0);
108 case GT: return (p > 0);
109 case LE: return (p <= 0);
110 case GE: return (p >= 0);
111 }
112 return (0);
113}
114
115static int arithcomp (char *s, char *t, int op, int w)
116{
117 long l, r;
118
119 l = evalexp (s, w);
120 r = evalexp (t, w);
121
122 switch (op) {
123 case EQ: return (l == r);
124 case NE: return (l != r);
125 case LT: return (l < r);
126 case GT: return (l > r);
127 case LE: return (l <= r);
128 case GE: return (l >= r);
129 }
130 return (0);
131}
132
133int binary_test (char *op, char *arg1, char *arg2, int w)
134{
135 int len, i;
Mike Frysingerfc9903f2010-10-20 07:17:33 -0400136 const op_tbl_t *optp;
wdenk2d1a5372004-02-23 19:30:57 +0000137
138 len = strlen(op);
139
140 for (optp = (op_tbl_t *)&op_table, i = 0;
Mike Frysingerfc9903f2010-10-20 07:17:33 -0400141 i < ARRAY_SIZE(op_table);
wdenk2d1a5372004-02-23 19:30:57 +0000142 optp++, i++) {
143
144 if ((strncmp (op, optp->op, len) == 0) && (len == strlen (optp->op))) {
145 if (w == 0) {
146 return (stringcomp(arg1, arg2, optp->opcode));
147 } else {
148 return (arithcomp (arg1, arg2, optp->opcode, w));
149 }
150 }
151 }
152
153 printf("Unknown operator '%s'\n", op);
154 return 0; /* op code not found */
155}
156
157/* command line interface to the shell test */
Wolfgang Denk54841ab2010-06-28 22:00:46 +0200158int do_itest ( cmd_tbl_t *cmdtp, int flag, int argc, char * const argv[] )
wdenk2d1a5372004-02-23 19:30:57 +0000159{
wdenkcd0a9de2004-02-23 20:48:38 +0000160 int value, w;
wdenk2d1a5372004-02-23 19:30:57 +0000161
wdenkcd0a9de2004-02-23 20:48:38 +0000162 /* Validate arguments */
Wolfgang Denk47e26b12010-07-17 01:06:04 +0200163 if ((argc != 4))
Simon Glass4c12eeb2011-12-10 08:44:01 +0000164 return CMD_RET_USAGE;
wdenk2d1a5372004-02-23 19:30:57 +0000165
166 /* Check for a data width specification.
167 * Defaults to long (4) if no specification.
168 * Uses -2 as 'width' for .s (string) so as not to upset existing code
169 */
170 switch (w = cmd_get_data_size(argv[0], 4)) {
171 case 1:
172 case 2:
173 case 4:
174 value = binary_test (argv[2], argv[1], argv[3], w);
175 break;
176 case -2:
177 value = binary_test (argv[2], argv[1], argv[3], 0);
178 break;
179 case -1:
180 default:
181 puts("Invalid data width specifier\n");
182 value = 0;
183 break;
184 }
185
186 return !value;
187}
188
189U_BOOT_CMD(
190 itest, 4, 0, do_itest,
Peter Tyser2fb26042009-01-27 18:03:12 -0600191 "return true/false on integer compare",
Wolfgang Denka89c33d2009-05-24 17:06:54 +0200192 "[.b, .w, .l, .s] [*]value1 <op> [*]value2"
wdenk2d1a5372004-02-23 19:30:57 +0000193);