u-boot/common/cmd_test.c
Stephen Warren e5e897c01b cmd_test: implement -e test for file existence
This is much like a regular shell's -e operator, except that it takes
multiple arguments to specify the device type and  device/partition ID
in addition to the usual filename:

if test -e mmc 0:1 /boot/boot.scr; then echo yes; else echo no; fi

Signed-off-by: Stephen Warren <swarren@nvidia.com>
2014-02-19 09:47:34 -05:00

221 lines
4.3 KiB
C

/*
* Copyright 2000-2009
* Wolfgang Denk, DENX Software Engineering, wd@denx.de.
*
* SPDX-License-Identifier: GPL-2.0+
*/
/*
* Define _STDBOOL_H here to avoid macro expansion of true and false.
* If the future code requires macro true or false, remove this define
* and undef true and false before U_BOOT_CMD. This define and comment
* shall be removed if change to U_BOOT_CMD is made to take string
* instead of stringifying it.
*/
#define _STDBOOL_H
#include <common.h>
#include <command.h>
#include <fs.h>
#define OP_INVALID 0
#define OP_NOT 1
#define OP_OR 2
#define OP_AND 3
#define OP_STR_EMPTY 4
#define OP_STR_NEMPTY 5
#define OP_STR_EQ 6
#define OP_STR_NEQ 7
#define OP_STR_LT 8
#define OP_STR_GT 9
#define OP_INT_EQ 10
#define OP_INT_NEQ 11
#define OP_INT_LT 12
#define OP_INT_LE 13
#define OP_INT_GT 14
#define OP_INT_GE 15
#define OP_FILE_EXISTS 16
const struct {
int arg;
const char *str;
int op;
int adv;
} op_adv[] = {
{1, "=", OP_STR_EQ, 3},
{1, "!=", OP_STR_NEQ, 3},
{1, "<", OP_STR_LT, 3},
{1, ">", OP_STR_GT, 3},
{1, "-eq", OP_INT_EQ, 3},
{1, "-ne", OP_INT_NEQ, 3},
{1, "-lt", OP_INT_LT, 3},
{1, "-le", OP_INT_LE, 3},
{1, "-gt", OP_INT_GT, 3},
{1, "-ge", OP_INT_GE, 3},
{0, "!", OP_NOT, 1},
{0, "-o", OP_OR, 1},
{0, "-a", OP_AND, 1},
{0, "-z", OP_STR_EMPTY, 2},
{0, "-n", OP_STR_NEMPTY, 2},
{0, "-e", OP_FILE_EXISTS, 4},
};
static int do_test(cmd_tbl_t *cmdtp, int flag, int argc, char * const argv[])
{
char * const *ap;
int i, op, left, adv, expr, last_expr, last_unop, last_binop;
/* args? */
if (argc < 3)
return 1;
#ifdef DEBUG
{
debug("test(%d):", argc);
left = 1;
while (argv[left])
debug(" '%s'", argv[left++]);
}
#endif
left = argc - 1;
ap = argv + 1;
expr = 0;
last_unop = OP_INVALID;
last_binop = OP_INVALID;
last_expr = -1;
while (left > 0) {
for (i = 0; i < ARRAY_SIZE(op_adv); i++) {
if (left <= op_adv[i].arg)
continue;
if (!strcmp(ap[op_adv[i].arg], op_adv[i].str)) {
op = op_adv[i].op;
adv = op_adv[i].adv;
break;
}
}
if (i == ARRAY_SIZE(op_adv)) {
expr = 1;
break;
}
if (left < adv) {
expr = 1;
break;
}
switch (op) {
case OP_STR_EMPTY:
expr = strlen(ap[1]) == 0 ? 1 : 0;
break;
case OP_STR_NEMPTY:
expr = strlen(ap[1]) == 0 ? 0 : 1;
break;
case OP_STR_EQ:
expr = strcmp(ap[0], ap[2]) == 0;
break;
case OP_STR_NEQ:
expr = strcmp(ap[0], ap[2]) != 0;
break;
case OP_STR_LT:
expr = strcmp(ap[0], ap[2]) < 0;
break;
case OP_STR_GT:
expr = strcmp(ap[0], ap[2]) > 0;
break;
case OP_INT_EQ:
expr = simple_strtol(ap[0], NULL, 10) ==
simple_strtol(ap[2], NULL, 10);
break;
case OP_INT_NEQ:
expr = simple_strtol(ap[0], NULL, 10) !=
simple_strtol(ap[2], NULL, 10);
break;
case OP_INT_LT:
expr = simple_strtol(ap[0], NULL, 10) <
simple_strtol(ap[2], NULL, 10);
break;
case OP_INT_LE:
expr = simple_strtol(ap[0], NULL, 10) <=
simple_strtol(ap[2], NULL, 10);
break;
case OP_INT_GT:
expr = simple_strtol(ap[0], NULL, 10) >
simple_strtol(ap[2], NULL, 10);
break;
case OP_INT_GE:
expr = simple_strtol(ap[0], NULL, 10) >=
simple_strtol(ap[2], NULL, 10);
break;
case OP_FILE_EXISTS:
expr = file_exists(ap[1], ap[2], ap[3], FS_TYPE_ANY);
break;
}
switch (op) {
case OP_OR:
last_expr = expr;
last_binop = OP_OR;
break;
case OP_AND:
last_expr = expr;
last_binop = OP_AND;
break;
case OP_NOT:
if (last_unop == OP_NOT)
last_unop = OP_INVALID;
else
last_unop = OP_NOT;
break;
default:
if (last_unop == OP_NOT) {
expr = !expr;
last_unop = OP_INVALID;
}
if (last_binop == OP_OR)
expr = last_expr || expr;
else if (last_binop == OP_AND)
expr = last_expr && expr;
last_binop = OP_INVALID;
break;
}
ap += adv; left -= adv;
}
expr = !expr;
debug (": returns %d\n", expr);
return expr;
}
U_BOOT_CMD(
test, CONFIG_SYS_MAXARGS, 1, do_test,
"minimal test like /bin/sh",
"[args..]"
);
static int do_false(cmd_tbl_t *cmdtp, int flag, int argc, char * const argv[])
{
return 1;
}
U_BOOT_CMD(
false, CONFIG_SYS_MAXARGS, 1, do_false,
"do nothing, unsuccessfully",
NULL
);
static int do_true(cmd_tbl_t *cmdtp, int flag, int argc, char * const argv[])
{
return 0;
}
U_BOOT_CMD(
true, CONFIG_SYS_MAXARGS, 1, do_true,
"do nothing, successfully",
NULL
);