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 #include <mapmem.h>
  19
  20 #include <asm/io.h>
  21
  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
  29 struct op_tbl_s {
  30         char    *op;            /* operator string */
  31         int     opcode;         /* internal representation of opcode */
  32 };
  33
  34 typedef struct op_tbl_s op_tbl_t;
  35
  36 static const op_tbl_t op_table [] = {
  37         { "-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
  52 static long evalexp(char *s, int w)
  53 {
  54         long l = 0;
  55         unsigned long addr;
  56         void *buf;
  57
  58         /* if the parameter starts with a * then assume is a pointer to the value we want */
  59         if (s[0] == '*') {
  60                 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;
  65                 }
  66                 switch (w) {
  67                 case 1: l = (long)(*(unsigned char *)buf);
  68                 case 2: l = (long)(*(unsigned short *)buf);
  69                 case 4: l = (long)(*(unsigned long *)buf);
  70                 }
  71                 unmap_physmem(buf, w);
  72                 return l;
  73         } else {
  74                 l = simple_strtoul(s, NULL, 16);
  75         }
  76
  77         return l & ((1UL << (w * 8)) - 1);
  78 }
  79
  80 static char * evalstr(char *s)
  81 {
  82         /* if the parameter starts with a * then assume a string pointer else its a literal */
  83         if (s[0] == '*') {
  84                 return (char *)simple_strtoul(&s[1], NULL, 16);
  85         } else if (s[0] == '$') {
  86                 int i = 2;
  87
  88                 if (s[1] != '{')
  89                         return NULL;
  90
  91                 while (s[i] != '}') {
  92                         if (s[i] == 0)
  93                                 return NULL;
  94                         i++;
  95                 }
  96                 s[i] = 0;
  97                 return  getenv((const char *)&s[2]);
  98         } else {
  99                 return s;
 100         }
 101 }
 102
 103 static int stringcomp(char *s, char *t, int op)
 104 {
 105         int p;
 106         char *l, *r;
 107
 108         l = evalstr(s);
 109         r = evalstr(t);
 110
 111         p = strcmp(l, r);
 112         switch (op) {
 113         case EQ: return (p == 0);
 114         case NE: return (p != 0);
 115         case LT: return (p < 0);
 116         case GT: return (p > 0);
 117         case LE: return (p <= 0);
 118         case GE: return (p >= 0);
 119         }
 120         return (0);
 121 }
 122
 123 static int arithcomp (char *s, char *t, int op, int w)
 124 {
 125         long l, r;
 126
 127         l = evalexp (s, w);
 128         r = evalexp (t, w);
 129
 130         switch (op) {
 131         case EQ: return (l == r);
 132         case NE: return (l != r);
 133         case LT: return (l < r);
 134         case GT: return (l > r);
 135         case LE: return (l <= r);
 136         case GE: return (l >= r);
 137         }
 138         return (0);
 139 }
 140
 141 static int binary_test(char *op, char *arg1, char *arg2, int w)
 142 {
 143         int len, i;
 144         const op_tbl_t *optp;
 145
 146         len = strlen(op);
 147
 148         for (optp = (op_tbl_t *)&op_table, i = 0;
 149              i < ARRAY_SIZE(op_table);
 150              optp++, i++) {
 151
 152                 if ((strncmp (op, optp->op, len) == 0) && (len == strlen (optp->op))) {
 153                         if (w == 0) {
 154                                 return (stringcomp(arg1, arg2, optp->opcode));
 155                         } else {
 156                                 return (arithcomp (arg1, arg2, optp->opcode, w));
 157                         }
 158                 }
 159         }
 160
 161         printf("Unknown operator '%s'\n", op);
 162         return 0;       /* op code not found */
 163 }
 164
 165 /* command line interface to the shell test */
 166 static int do_itest(cmd_tbl_t *cmdtp, int flag, int argc, char * const argv[])
 167 {
 168         int     value, w;
 169
 170         /* Validate arguments */
 171         if ((argc != 4))
 172                 return CMD_RET_USAGE;
 173
 174         /* Check for a data width specification.
 175          * Defaults to long (4) if no specification.
 176          * Uses -2 as 'width' for .s (string) so as not to upset existing code
 177          */
 178         switch (w = cmd_get_data_size(argv[0], 4)) {
 179         case 1:
 180         case 2:
 181         case 4:
 182                 value = binary_test (argv[2], argv[1], argv[3], w);
 183                 break;
 184         case -2:
 185                 value = binary_test (argv[2], argv[1], argv[3], 0);
 186                 break;
 187         case -1:
 188         default:
 189                 puts("Invalid data width specifier\n");
 190                 value = 0;
 191                 break;
 192         }
 193
 194         return !value;
 195 }
 196
 197 U_BOOT_CMD(
 198         itest, 4, 0, do_itest,
 199         "return true/false on integer compare",
 200         "[.b, .w, .l, .s] [*]value1 <op> [*]value2"
 201 );