1*83d290c5STom Rini // SPDX-License-Identifier: GPL-2.0+
22e192b24SSimon Glass /*
32e192b24SSimon Glass * Copyright 2000-2009
42e192b24SSimon Glass * Wolfgang Denk, DENX Software Engineering, wd@denx.de.
52e192b24SSimon Glass */
62e192b24SSimon Glass
72e192b24SSimon Glass #include <common.h>
82e192b24SSimon Glass #include <command.h>
92e192b24SSimon Glass #include <fs.h>
102e192b24SSimon Glass
112e192b24SSimon Glass #define OP_INVALID 0
122e192b24SSimon Glass #define OP_NOT 1
132e192b24SSimon Glass #define OP_OR 2
142e192b24SSimon Glass #define OP_AND 3
152e192b24SSimon Glass #define OP_STR_EMPTY 4
162e192b24SSimon Glass #define OP_STR_NEMPTY 5
172e192b24SSimon Glass #define OP_STR_EQ 6
182e192b24SSimon Glass #define OP_STR_NEQ 7
192e192b24SSimon Glass #define OP_STR_LT 8
202e192b24SSimon Glass #define OP_STR_GT 9
212e192b24SSimon Glass #define OP_INT_EQ 10
222e192b24SSimon Glass #define OP_INT_NEQ 11
232e192b24SSimon Glass #define OP_INT_LT 12
242e192b24SSimon Glass #define OP_INT_LE 13
252e192b24SSimon Glass #define OP_INT_GT 14
262e192b24SSimon Glass #define OP_INT_GE 15
272e192b24SSimon Glass #define OP_FILE_EXISTS 16
282e192b24SSimon Glass
292e192b24SSimon Glass const struct {
302e192b24SSimon Glass int arg;
312e192b24SSimon Glass const char *str;
322e192b24SSimon Glass int op;
332e192b24SSimon Glass int adv;
342e192b24SSimon Glass } op_adv[] = {
352e192b24SSimon Glass {1, "=", OP_STR_EQ, 3},
362e192b24SSimon Glass {1, "!=", OP_STR_NEQ, 3},
372e192b24SSimon Glass {1, "<", OP_STR_LT, 3},
382e192b24SSimon Glass {1, ">", OP_STR_GT, 3},
392e192b24SSimon Glass {1, "-eq", OP_INT_EQ, 3},
402e192b24SSimon Glass {1, "-ne", OP_INT_NEQ, 3},
412e192b24SSimon Glass {1, "-lt", OP_INT_LT, 3},
422e192b24SSimon Glass {1, "-le", OP_INT_LE, 3},
432e192b24SSimon Glass {1, "-gt", OP_INT_GT, 3},
442e192b24SSimon Glass {1, "-ge", OP_INT_GE, 3},
452e192b24SSimon Glass {0, "!", OP_NOT, 1},
462e192b24SSimon Glass {0, "-o", OP_OR, 1},
472e192b24SSimon Glass {0, "-a", OP_AND, 1},
482e192b24SSimon Glass {0, "-z", OP_STR_EMPTY, 2},
492e192b24SSimon Glass {0, "-n", OP_STR_NEMPTY, 2},
502e192b24SSimon Glass {0, "-e", OP_FILE_EXISTS, 4},
512e192b24SSimon Glass };
522e192b24SSimon Glass
do_test(cmd_tbl_t * cmdtp,int flag,int argc,char * const argv[])532e192b24SSimon Glass static int do_test(cmd_tbl_t *cmdtp, int flag, int argc, char * const argv[])
542e192b24SSimon Glass {
552e192b24SSimon Glass char * const *ap;
562e192b24SSimon Glass int i, op, left, adv, expr, last_expr, last_unop, last_binop;
572e192b24SSimon Glass
582e192b24SSimon Glass /* args? */
592e192b24SSimon Glass if (argc < 3)
602e192b24SSimon Glass return 1;
612e192b24SSimon Glass
622e192b24SSimon Glass #ifdef DEBUG
632e192b24SSimon Glass {
642e192b24SSimon Glass debug("test(%d):", argc);
652e192b24SSimon Glass left = 1;
662e192b24SSimon Glass while (argv[left])
672e192b24SSimon Glass debug(" '%s'", argv[left++]);
682e192b24SSimon Glass }
692e192b24SSimon Glass #endif
702e192b24SSimon Glass
712e192b24SSimon Glass left = argc - 1;
722e192b24SSimon Glass ap = argv + 1;
732e192b24SSimon Glass expr = 0;
742e192b24SSimon Glass last_unop = OP_INVALID;
752e192b24SSimon Glass last_binop = OP_INVALID;
762e192b24SSimon Glass last_expr = -1;
772e192b24SSimon Glass while (left > 0) {
782e192b24SSimon Glass for (i = 0; i < ARRAY_SIZE(op_adv); i++) {
792e192b24SSimon Glass if (left <= op_adv[i].arg)
802e192b24SSimon Glass continue;
812e192b24SSimon Glass if (!strcmp(ap[op_adv[i].arg], op_adv[i].str)) {
822e192b24SSimon Glass op = op_adv[i].op;
832e192b24SSimon Glass adv = op_adv[i].adv;
842e192b24SSimon Glass break;
852e192b24SSimon Glass }
862e192b24SSimon Glass }
872e192b24SSimon Glass if (i == ARRAY_SIZE(op_adv)) {
882e192b24SSimon Glass expr = 1;
892e192b24SSimon Glass break;
902e192b24SSimon Glass }
912e192b24SSimon Glass if (left < adv) {
922e192b24SSimon Glass expr = 1;
932e192b24SSimon Glass break;
942e192b24SSimon Glass }
952e192b24SSimon Glass
962e192b24SSimon Glass switch (op) {
972e192b24SSimon Glass case OP_STR_EMPTY:
982e192b24SSimon Glass expr = strlen(ap[1]) == 0 ? 1 : 0;
992e192b24SSimon Glass break;
1002e192b24SSimon Glass case OP_STR_NEMPTY:
1012e192b24SSimon Glass expr = strlen(ap[1]) == 0 ? 0 : 1;
1022e192b24SSimon Glass break;
1032e192b24SSimon Glass case OP_STR_EQ:
1042e192b24SSimon Glass expr = strcmp(ap[0], ap[2]) == 0;
1052e192b24SSimon Glass break;
1062e192b24SSimon Glass case OP_STR_NEQ:
1072e192b24SSimon Glass expr = strcmp(ap[0], ap[2]) != 0;
1082e192b24SSimon Glass break;
1092e192b24SSimon Glass case OP_STR_LT:
1102e192b24SSimon Glass expr = strcmp(ap[0], ap[2]) < 0;
1112e192b24SSimon Glass break;
1122e192b24SSimon Glass case OP_STR_GT:
1132e192b24SSimon Glass expr = strcmp(ap[0], ap[2]) > 0;
1142e192b24SSimon Glass break;
1152e192b24SSimon Glass case OP_INT_EQ:
1162e192b24SSimon Glass expr = simple_strtol(ap[0], NULL, 10) ==
1172e192b24SSimon Glass simple_strtol(ap[2], NULL, 10);
1182e192b24SSimon Glass break;
1192e192b24SSimon Glass case OP_INT_NEQ:
1202e192b24SSimon Glass expr = simple_strtol(ap[0], NULL, 10) !=
1212e192b24SSimon Glass simple_strtol(ap[2], NULL, 10);
1222e192b24SSimon Glass break;
1232e192b24SSimon Glass case OP_INT_LT:
1242e192b24SSimon Glass expr = simple_strtol(ap[0], NULL, 10) <
1252e192b24SSimon Glass simple_strtol(ap[2], NULL, 10);
1262e192b24SSimon Glass break;
1272e192b24SSimon Glass case OP_INT_LE:
1282e192b24SSimon Glass expr = simple_strtol(ap[0], NULL, 10) <=
1292e192b24SSimon Glass simple_strtol(ap[2], NULL, 10);
1302e192b24SSimon Glass break;
1312e192b24SSimon Glass case OP_INT_GT:
1322e192b24SSimon Glass expr = simple_strtol(ap[0], NULL, 10) >
1332e192b24SSimon Glass simple_strtol(ap[2], NULL, 10);
1342e192b24SSimon Glass break;
1352e192b24SSimon Glass case OP_INT_GE:
1362e192b24SSimon Glass expr = simple_strtol(ap[0], NULL, 10) >=
1372e192b24SSimon Glass simple_strtol(ap[2], NULL, 10);
1382e192b24SSimon Glass break;
1392e192b24SSimon Glass case OP_FILE_EXISTS:
1402e192b24SSimon Glass expr = file_exists(ap[1], ap[2], ap[3], FS_TYPE_ANY);
1412e192b24SSimon Glass break;
1422e192b24SSimon Glass }
1432e192b24SSimon Glass
1442e192b24SSimon Glass switch (op) {
1452e192b24SSimon Glass case OP_OR:
1462e192b24SSimon Glass last_expr = expr;
1472e192b24SSimon Glass last_binop = OP_OR;
1482e192b24SSimon Glass break;
1492e192b24SSimon Glass case OP_AND:
1502e192b24SSimon Glass last_expr = expr;
1512e192b24SSimon Glass last_binop = OP_AND;
1522e192b24SSimon Glass break;
1532e192b24SSimon Glass case OP_NOT:
1542e192b24SSimon Glass if (last_unop == OP_NOT)
1552e192b24SSimon Glass last_unop = OP_INVALID;
1562e192b24SSimon Glass else
1572e192b24SSimon Glass last_unop = OP_NOT;
1582e192b24SSimon Glass break;
1592e192b24SSimon Glass default:
1602e192b24SSimon Glass if (last_unop == OP_NOT) {
1612e192b24SSimon Glass expr = !expr;
1622e192b24SSimon Glass last_unop = OP_INVALID;
1632e192b24SSimon Glass }
1642e192b24SSimon Glass
1652e192b24SSimon Glass if (last_binop == OP_OR)
1662e192b24SSimon Glass expr = last_expr || expr;
1672e192b24SSimon Glass else if (last_binop == OP_AND)
1682e192b24SSimon Glass expr = last_expr && expr;
1692e192b24SSimon Glass last_binop = OP_INVALID;
1702e192b24SSimon Glass
1712e192b24SSimon Glass break;
1722e192b24SSimon Glass }
1732e192b24SSimon Glass
1742e192b24SSimon Glass ap += adv; left -= adv;
1752e192b24SSimon Glass }
1762e192b24SSimon Glass
1772e192b24SSimon Glass expr = !expr;
1782e192b24SSimon Glass
1792e192b24SSimon Glass debug (": returns %d\n", expr);
1802e192b24SSimon Glass
1812e192b24SSimon Glass return expr;
1822e192b24SSimon Glass }
1832e192b24SSimon Glass
1842e192b24SSimon Glass #undef true
1852e192b24SSimon Glass #undef false
1862e192b24SSimon Glass
1872e192b24SSimon Glass U_BOOT_CMD(
1882e192b24SSimon Glass test, CONFIG_SYS_MAXARGS, 1, do_test,
1892e192b24SSimon Glass "minimal test like /bin/sh",
1902e192b24SSimon Glass "[args..]"
1912e192b24SSimon Glass );
1922e192b24SSimon Glass
do_false(cmd_tbl_t * cmdtp,int flag,int argc,char * const argv[])1932e192b24SSimon Glass static int do_false(cmd_tbl_t *cmdtp, int flag, int argc, char * const argv[])
1942e192b24SSimon Glass {
1952e192b24SSimon Glass return 1;
1962e192b24SSimon Glass }
1972e192b24SSimon Glass
1982e192b24SSimon Glass U_BOOT_CMD(
1992e192b24SSimon Glass false, CONFIG_SYS_MAXARGS, 1, do_false,
2002e192b24SSimon Glass "do nothing, unsuccessfully",
2012e192b24SSimon Glass NULL
2022e192b24SSimon Glass );
2032e192b24SSimon Glass
do_true(cmd_tbl_t * cmdtp,int flag,int argc,char * const argv[])2042e192b24SSimon Glass static int do_true(cmd_tbl_t *cmdtp, int flag, int argc, char * const argv[])
2052e192b24SSimon Glass {
2062e192b24SSimon Glass return 0;
2072e192b24SSimon Glass }
2082e192b24SSimon Glass
2092e192b24SSimon Glass U_BOOT_CMD(
2102e192b24SSimon Glass true, CONFIG_SYS_MAXARGS, 1, do_true,
2112e192b24SSimon Glass "do nothing, successfully",
2122e192b24SSimon Glass NULL
2132e192b24SSimon Glass );
214