avr32: use dlmalloc for DMA buffers
[oweals/u-boot.git] / common / cmd_itest.c
1 /*
2  * (C) Copyright 2003
3  * Tait Electronics Limited, Christchurch, New Zealand
4  *
5  * SPDX-License-Identifier:     GPL-2.0+
6  */
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>
18
19 #define EQ      0
20 #define NE      1
21 #define LT      2
22 #define GT      3
23 #define LE      4
24 #define GE      5
25
26 struct op_tbl_s {
27         char    *op;            /* operator string */
28         int     opcode;         /* internal representation of opcode */
29 };
30
31 typedef struct op_tbl_s op_tbl_t;
32
33 static const op_tbl_t op_table [] = {
34         { "-lt", LT },
35         { "<"  , LT },
36         { "-gt", GT },
37         { ">"  , GT },
38         { "-eq", EQ },
39         { "==" , EQ },
40         { "-ne", NE },
41         { "!=" , NE },
42         { "<>" , NE },
43         { "-ge", GE },
44         { ">=" , GE },
45         { "-le", LE },
46         { "<=" , LE },
47 };
48
49 static long evalexp(char *s, int w)
50 {
51         long l = 0;
52         long *p;
53
54         /* if the parameter starts with a * then assume is a pointer to the value we want */
55         if (s[0] == '*') {
56                 p = (long *)simple_strtoul(&s[1], NULL, 16);
57                 switch (w) {
58                 case 1: return((long)(*(unsigned char *)p));
59                 case 2: return((long)(*(unsigned short *)p));
60                 case 4: return(*p);
61                 }
62         } else {
63                 l = simple_strtoul(s, NULL, 16);
64         }
65
66         return l & ((1UL << (w * 8)) - 1);
67 }
68
69 static char * evalstr(char *s)
70 {
71         /* if the parameter starts with a * then assume a string pointer else its a literal */
72         if (s[0] == '*') {
73                 return (char *)simple_strtoul(&s[1], NULL, 16);
74         } else if (s[0] == '$') {
75                 int i = 2;
76
77                 if (s[1] != '{')
78                         return NULL;
79
80                 while (s[i] != '}') {
81                         if (s[i] == 0)
82                                 return NULL;
83                         i++;
84                 }
85                 s[i] = 0;
86                 return  getenv((const char *)&s[2]);
87         } else {
88                 return s;
89         }
90 }
91
92 static int stringcomp(char *s, char *t, int op)
93 {
94         int p;
95         char *l, *r;
96
97         l = evalstr(s);
98         r = evalstr(t);
99
100         p = strcmp(l, r);
101         switch (op) {
102         case EQ: return (p == 0);
103         case NE: return (p != 0);
104         case LT: return (p < 0);
105         case GT: return (p > 0);
106         case LE: return (p <= 0);
107         case GE: return (p >= 0);
108         }
109         return (0);
110 }
111
112 static int arithcomp (char *s, char *t, int op, int w)
113 {
114         long l, r;
115
116         l = evalexp (s, w);
117         r = evalexp (t, w);
118
119         switch (op) {
120         case EQ: return (l == r);
121         case NE: return (l != r);
122         case LT: return (l < r);
123         case GT: return (l > r);
124         case LE: return (l <= r);
125         case GE: return (l >= r);
126         }
127         return (0);
128 }
129
130 static int binary_test(char *op, char *arg1, char *arg2, int w)
131 {
132         int len, i;
133         const op_tbl_t *optp;
134
135         len = strlen(op);
136
137         for (optp = (op_tbl_t *)&op_table, i = 0;
138              i < ARRAY_SIZE(op_table);
139              optp++, i++) {
140
141                 if ((strncmp (op, optp->op, len) == 0) && (len == strlen (optp->op))) {
142                         if (w == 0) {
143                                 return (stringcomp(arg1, arg2, optp->opcode));
144                         } else {
145                                 return (arithcomp (arg1, arg2, optp->opcode, w));
146                         }
147                 }
148         }
149
150         printf("Unknown operator '%s'\n", op);
151         return 0;       /* op code not found */
152 }
153
154 /* command line interface to the shell test */
155 static int do_itest(cmd_tbl_t *cmdtp, int flag, int argc, char * const argv[])
156 {
157         int     value, w;
158
159         /* Validate arguments */
160         if ((argc != 4))
161                 return CMD_RET_USAGE;
162
163         /* Check for a data width specification.
164          * Defaults to long (4) if no specification.
165          * Uses -2 as 'width' for .s (string) so as not to upset existing code
166          */
167         switch (w = cmd_get_data_size(argv[0], 4)) {
168         case 1:
169         case 2:
170         case 4:
171                 value = binary_test (argv[2], argv[1], argv[3], w);
172                 break;
173         case -2:
174                 value = binary_test (argv[2], argv[1], argv[3], 0);
175                 break;
176         case -1:
177         default:
178                 puts("Invalid data width specifier\n");
179                 value = 0;
180                 break;
181         }
182
183         return !value;
184 }
185
186 U_BOOT_CMD(
187         itest, 4, 0, do_itest,
188         "return true/false on integer compare",
189         "[.b, .w, .l, .s] [*]value1 <op> [*]value2"
190 );