mirror of
https://github.com/u-boot/u-boot.git
synced 2024-12-11 21:53:29 +08:00
c9bcb6f13d
The mask value used in itest overflows and therefore it can return an incorrect result for something like 'itest 0 == 1'. Fix it. Signed-off-by: Simon Glass <sjg@chromium.org>
191 lines
3.6 KiB
C
191 lines
3.6 KiB
C
/*
|
|
* (C) Copyright 2003
|
|
* Tait Electronics Limited, Christchurch, New Zealand
|
|
*
|
|
* SPDX-License-Identifier: GPL-2.0+
|
|
*/
|
|
|
|
/*
|
|
* This file provides a shell like 'test' function to return
|
|
* true/false from an integer or string compare of two memory
|
|
* locations or a location and a scalar/literal.
|
|
* A few parts were lifted from bash 'test' command
|
|
*/
|
|
|
|
#include <common.h>
|
|
#include <config.h>
|
|
#include <command.h>
|
|
|
|
#define EQ 0
|
|
#define NE 1
|
|
#define LT 2
|
|
#define GT 3
|
|
#define LE 4
|
|
#define GE 5
|
|
|
|
struct op_tbl_s {
|
|
char *op; /* operator string */
|
|
int opcode; /* internal representation of opcode */
|
|
};
|
|
|
|
typedef struct op_tbl_s op_tbl_t;
|
|
|
|
static const op_tbl_t op_table [] = {
|
|
{ "-lt", LT },
|
|
{ "<" , LT },
|
|
{ "-gt", GT },
|
|
{ ">" , GT },
|
|
{ "-eq", EQ },
|
|
{ "==" , EQ },
|
|
{ "-ne", NE },
|
|
{ "!=" , NE },
|
|
{ "<>" , NE },
|
|
{ "-ge", GE },
|
|
{ ">=" , GE },
|
|
{ "-le", LE },
|
|
{ "<=" , LE },
|
|
};
|
|
|
|
static long evalexp(char *s, int w)
|
|
{
|
|
long l = 0;
|
|
long *p;
|
|
|
|
/* if the parameter starts with a * then assume is a pointer to the value we want */
|
|
if (s[0] == '*') {
|
|
p = (long *)simple_strtoul(&s[1], NULL, 16);
|
|
switch (w) {
|
|
case 1: return((long)(*(unsigned char *)p));
|
|
case 2: return((long)(*(unsigned short *)p));
|
|
case 4: return(*p);
|
|
}
|
|
} else {
|
|
l = simple_strtoul(s, NULL, 16);
|
|
}
|
|
|
|
return l & ((1UL << (w * 8)) - 1);
|
|
}
|
|
|
|
static char * evalstr(char *s)
|
|
{
|
|
/* if the parameter starts with a * then assume a string pointer else its a literal */
|
|
if (s[0] == '*') {
|
|
return (char *)simple_strtoul(&s[1], NULL, 16);
|
|
} else if (s[0] == '$') {
|
|
int i = 2;
|
|
|
|
if (s[1] != '{')
|
|
return NULL;
|
|
|
|
while (s[i] != '}') {
|
|
if (s[i] == 0)
|
|
return NULL;
|
|
i++;
|
|
}
|
|
s[i] = 0;
|
|
return getenv((const char *)&s[2]);
|
|
} else {
|
|
return s;
|
|
}
|
|
}
|
|
|
|
static int stringcomp(char *s, char *t, int op)
|
|
{
|
|
int p;
|
|
char *l, *r;
|
|
|
|
l = evalstr(s);
|
|
r = evalstr(t);
|
|
|
|
p = strcmp(l, r);
|
|
switch (op) {
|
|
case EQ: return (p == 0);
|
|
case NE: return (p != 0);
|
|
case LT: return (p < 0);
|
|
case GT: return (p > 0);
|
|
case LE: return (p <= 0);
|
|
case GE: return (p >= 0);
|
|
}
|
|
return (0);
|
|
}
|
|
|
|
static int arithcomp (char *s, char *t, int op, int w)
|
|
{
|
|
long l, r;
|
|
|
|
l = evalexp (s, w);
|
|
r = evalexp (t, w);
|
|
|
|
switch (op) {
|
|
case EQ: return (l == r);
|
|
case NE: return (l != r);
|
|
case LT: return (l < r);
|
|
case GT: return (l > r);
|
|
case LE: return (l <= r);
|
|
case GE: return (l >= r);
|
|
}
|
|
return (0);
|
|
}
|
|
|
|
static int binary_test(char *op, char *arg1, char *arg2, int w)
|
|
{
|
|
int len, i;
|
|
const op_tbl_t *optp;
|
|
|
|
len = strlen(op);
|
|
|
|
for (optp = (op_tbl_t *)&op_table, i = 0;
|
|
i < ARRAY_SIZE(op_table);
|
|
optp++, i++) {
|
|
|
|
if ((strncmp (op, optp->op, len) == 0) && (len == strlen (optp->op))) {
|
|
if (w == 0) {
|
|
return (stringcomp(arg1, arg2, optp->opcode));
|
|
} else {
|
|
return (arithcomp (arg1, arg2, optp->opcode, w));
|
|
}
|
|
}
|
|
}
|
|
|
|
printf("Unknown operator '%s'\n", op);
|
|
return 0; /* op code not found */
|
|
}
|
|
|
|
/* command line interface to the shell test */
|
|
static int do_itest(cmd_tbl_t *cmdtp, int flag, int argc, char * const argv[])
|
|
{
|
|
int value, w;
|
|
|
|
/* Validate arguments */
|
|
if ((argc != 4))
|
|
return CMD_RET_USAGE;
|
|
|
|
/* Check for a data width specification.
|
|
* Defaults to long (4) if no specification.
|
|
* Uses -2 as 'width' for .s (string) so as not to upset existing code
|
|
*/
|
|
switch (w = cmd_get_data_size(argv[0], 4)) {
|
|
case 1:
|
|
case 2:
|
|
case 4:
|
|
value = binary_test (argv[2], argv[1], argv[3], w);
|
|
break;
|
|
case -2:
|
|
value = binary_test (argv[2], argv[1], argv[3], 0);
|
|
break;
|
|
case -1:
|
|
default:
|
|
puts("Invalid data width specifier\n");
|
|
value = 0;
|
|
break;
|
|
}
|
|
|
|
return !value;
|
|
}
|
|
|
|
U_BOOT_CMD(
|
|
itest, 4, 0, do_itest,
|
|
"return true/false on integer compare",
|
|
"[.b, .w, .l, .s] [*]value1 <op> [*]value2"
|
|
);
|