[J-u-boot.git] / cmd / test.c

// SPDX-License-Identifier: GPL-2.0+
/*
 * Copyright 2000-2009
 * Wolfgang Denk, DENX Software Engineering, [email protected].
 */

#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, 0) ==
					simple_strtol(ap[2], NULL, 0);
			break;
		case OP_INT_NEQ:
			expr = simple_strtol(ap[0], NULL, 0) !=
					simple_strtol(ap[2], NULL, 0);
			break;
		case OP_INT_LT:
			expr = simple_strtol(ap[0], NULL, 0) <
					simple_strtol(ap[2], NULL, 0);
			break;
		case OP_INT_LE:
			expr = simple_strtol(ap[0], NULL, 0) <=
					simple_strtol(ap[2], NULL, 0);
			break;
		case OP_INT_GT:
			expr = simple_strtol(ap[0], NULL, 0) >
					simple_strtol(ap[2], NULL, 0);
			break;
		case OP_INT_GE:
			expr = simple_strtol(ap[0], NULL, 0) >=
					simple_strtol(ap[2], NULL, 0);
			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;
}

#undef true
#undef false

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
);
Commit	Line	Data
83d290c5	1	// SPDX-License-Identifier: GPL-2.0+
6b8f5ad1 PT	2	/*
	3	* Copyright 2000-2009
	4	* Wolfgang Denk, DENX Software Engineering, [email protected].
6b8f5ad1 PT	5	*/
	6
	7	#include <common.h>
	8	#include <command.h>
e5e897c0	9	#include <fs.h>
6b8f5ad1	10
490ba833	11	#define OP_INVALID 0
d9b651ce	12	#define OP_NOT 1
490ba833 SW	13	#define OP_OR 2
	14	#define OP_AND 3
	15	#define OP_STR_EMPTY 4
	16	#define OP_STR_NEMPTY 5
	17	#define OP_STR_EQ 6
	18	#define OP_STR_NEQ 7
	19	#define OP_STR_LT 8
	20	#define OP_STR_GT 9
	21	#define OP_INT_EQ 10
	22	#define OP_INT_NEQ 11
	23	#define OP_INT_LT 12
	24	#define OP_INT_LE 13
	25	#define OP_INT_GT 14
	26	#define OP_INT_GE 15
e5e897c0	27	#define OP_FILE_EXISTS 16
490ba833 SW	28
	29	const struct {
	30	int arg;
	31	const char *str;
	32	int op;
	33	int adv;
	34	} op_adv[] = {
490ba833 SW	35	{1, "=", OP_STR_EQ, 3},
	36	{1, "!=", OP_STR_NEQ, 3},
	37	{1, "<", OP_STR_LT, 3},
	38	{1, ">", OP_STR_GT, 3},
	39	{1, "-eq", OP_INT_EQ, 3},
	40	{1, "-ne", OP_INT_NEQ, 3},
	41	{1, "-lt", OP_INT_LT, 3},
	42	{1, "-le", OP_INT_LE, 3},
	43	{1, "-gt", OP_INT_GT, 3},
	44	{1, "-ge", OP_INT_GE, 3},
d9b651ce	45	{0, "!", OP_NOT, 1},
4c80f29e SW	46	{0, "-o", OP_OR, 1},
	47	{0, "-a", OP_AND, 1},
	48	{0, "-z", OP_STR_EMPTY, 2},
	49	{0, "-n", OP_STR_NEMPTY, 2},
e5e897c0	50	{0, "-e", OP_FILE_EXISTS, 4},
490ba833 SW	51	};
490ba833 SW	52
088f1b19	53	static int do_test(cmd_tbl_t cmdtp, int flag, int argc, char const argv[])
6b8f5ad1	54	{
54841ab5	55	char * const *ap;
d9b651ce	56	int i, op, left, adv, expr, last_expr, last_unop, last_binop;
6b8f5ad1 PT	57
	58	/* args? */
	59	if (argc < 3)
	60	return 1;
	61
b0fe6abd	62	#ifdef DEBUG
6b8f5ad1	63	{
b0fe6abd	64	debug("test(%d):", argc);
6b8f5ad1 PT	65	left = 1;
6b8f5ad1 PT	66	while (argv[left])
b0fe6abd	67	debug(" '%s'", argv[left++]);
6b8f5ad1 PT	68	}
	69	#endif
	70
d9b651ce SW	71	left = argc - 1;
d9b651ce SW	72	ap = argv + 1;
2453de99	73	expr = 0;
d9b651ce SW	74	last_unop = OP_INVALID;
d9b651ce SW	75	last_binop = OP_INVALID;
6b8f5ad1 PT	76	last_expr = -1;
6b8f5ad1 PT	77	while (left > 0) {
490ba833 SW	78	for (i = 0; i < ARRAY_SIZE(op_adv); i++) {
	79	if (left <= op_adv[i].arg)
	80	continue;
	81	if (!strcmp(ap[op_adv[i].arg], op_adv[i].str)) {
	82	op = op_adv[i].op;
	83	adv = op_adv[i].adv;
	84	break;
	85	}
	86	}
	87	if (i == ARRAY_SIZE(op_adv)) {
6b8f5ad1 PT	88	expr = 1;
	89	break;
	90	}
490ba833 SW	91	if (left < adv) {
	92	expr = 1;
	93	break;
6b8f5ad1 PT	94	}
6b8f5ad1 PT	95
490ba833 SW	96	switch (op) {
	97	case OP_STR_EMPTY:
	98	expr = strlen(ap[1]) == 0 ? 1 : 0;
	99	break;
	100	case OP_STR_NEMPTY:
	101	expr = strlen(ap[1]) == 0 ? 0 : 1;
	102	break;
	103	case OP_STR_EQ:
	104	expr = strcmp(ap[0], ap[2]) == 0;
	105	break;
	106	case OP_STR_NEQ:
	107	expr = strcmp(ap[0], ap[2]) != 0;
	108	break;
	109	case OP_STR_LT:
	110	expr = strcmp(ap[0], ap[2]) < 0;
	111	break;
	112	case OP_STR_GT:
	113	expr = strcmp(ap[0], ap[2]) > 0;
	114	break;
	115	case OP_INT_EQ:
100e75bb KR	116	expr = simple_strtol(ap[0], NULL, 0) ==
100e75bb KR	117	simple_strtol(ap[2], NULL, 0);
490ba833 SW	118	break;
490ba833 SW	119	case OP_INT_NEQ:
100e75bb KR	120	expr = simple_strtol(ap[0], NULL, 0) !=
100e75bb KR	121	simple_strtol(ap[2], NULL, 0);
490ba833 SW	122	break;
490ba833 SW	123	case OP_INT_LT:
100e75bb KR	124	expr = simple_strtol(ap[0], NULL, 0) <
100e75bb KR	125	simple_strtol(ap[2], NULL, 0);
490ba833 SW	126	break;
490ba833 SW	127	case OP_INT_LE:
100e75bb KR	128	expr = simple_strtol(ap[0], NULL, 0) <=
100e75bb KR	129	simple_strtol(ap[2], NULL, 0);
490ba833 SW	130	break;
490ba833 SW	131	case OP_INT_GT:
100e75bb KR	132	expr = simple_strtol(ap[0], NULL, 0) >
100e75bb KR	133	simple_strtol(ap[2], NULL, 0);
490ba833 SW	134	break;
490ba833 SW	135	case OP_INT_GE:
100e75bb KR	136	expr = simple_strtol(ap[0], NULL, 0) >=
100e75bb KR	137	simple_strtol(ap[2], NULL, 0);
490ba833	138	break;
e5e897c0 SW	139	case OP_FILE_EXISTS:
	140	expr = file_exists(ap[1], ap[2], ap[3], FS_TYPE_ANY);
	141	break;
6b8f5ad1 PT	142	}
6b8f5ad1 PT	143
490ba833 SW	144	switch (op) {
	145	case OP_OR:
	146	last_expr = expr;
d9b651ce	147	last_binop = OP_OR;
490ba833 SW	148	break;
	149	case OP_AND:
	150	last_expr = expr;
d9b651ce SW	151	last_binop = OP_AND;
	152	break;
	153	case OP_NOT:
	154	if (last_unop == OP_NOT)
	155	last_unop = OP_INVALID;
	156	else
	157	last_unop = OP_NOT;
490ba833 SW	158	break;
490ba833 SW	159	default:
d9b651ce SW	160	if (last_unop == OP_NOT) {
	161	expr = !expr;
	162	last_unop = OP_INVALID;
	163	}
	164
	165	if (last_binop == OP_OR)
6b8f5ad1	166	expr = last_expr \|\| expr;
d9b651ce	167	else if (last_binop == OP_AND)
6b8f5ad1	168	expr = last_expr && expr;
d9b651ce SW	169	last_binop = OP_INVALID;
d9b651ce SW	170
490ba833	171	break;
6b8f5ad1 PT	172	}
	173
	174	ap += adv; left -= adv;
	175	}
	176
6b8f5ad1 PT	177	expr = !expr;
	178
	179	debug (": returns %d\n", expr);
	180
	181	return expr;
	182	}
	183
09c32807 HS	184	#undef true
	185	#undef false
	186
6b8f5ad1 PT	187	U_BOOT_CMD(
	188	test, CONFIG_SYS_MAXARGS, 1, do_test,
	189	"minimal test like /bin/sh",
	190	"[args..]"
	191	);
396fd173	192
088f1b19	193	static int do_false(cmd_tbl_t cmdtp, int flag, int argc, char const argv[])
396fd173 PT	194	{
	195	return 1;
	196	}
	197
	198	U_BOOT_CMD(
	199	false, CONFIG_SYS_MAXARGS, 1, do_false,
	200	"do nothing, unsuccessfully",
	201	NULL
	202	);
	203
088f1b19	204	static int do_true(cmd_tbl_t cmdtp, int flag, int argc, char const argv[])
396fd173 PT	205	{
	206	return 0;
	207	}
	208
	209	U_BOOT_CMD(
	210	true, CONFIG_SYS_MAXARGS, 1, do_true,
	211	"do nothing, successfully",
	212	NULL
	213	);