summaryrefslogtreecommitdiff
path: root/arch/arm/mach-imx/cmd_qb.c
blob: a6b654d342fd83e44810df1956c1f2c669533a60 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
// SPDX-License-Identifier: GPL-2.0+
/*
 * Copyright 2024-2026 NXP
 */
#include <command.h>
#include <spl.h>
#include <stdlib.h>

#include <asm/mach-imx/boot_mode.h>
#include <asm/mach-imx/sys_proto.h>
#include <asm/mach-imx/qb.h>

static void parse_qb_args(int argc, char * const argv[],
			  const char **ifname, const char **dev)
{
	/* qb save/erase -> use boot device */
	if (argc < 2) {
		*ifname = "auto";
		return;
	}

	*ifname = argv[1];

	if (argc == 3)
		*dev = argv[2];
}

static int do_qb(struct cmd_tbl *cmdtp, int flag, int argc,
		 char * const argv[], bool save)
{
	const char *ifname, *dev;

	parse_qb_args(argc, argv, &ifname, &dev);

	if (imx_qb(ifname, dev, save))
		return CMD_RET_FAILURE;

	return CMD_RET_SUCCESS;
}

static int do_qb_check(struct cmd_tbl *cmdtp, int flag,
		       int argc, char * const argv[])
{
	return imx_qb_check() ? CMD_RET_SUCCESS : CMD_RET_FAILURE;
}

static int do_qb_save(struct cmd_tbl *cmdtp, int flag,
		      int argc, char * const argv[])
{
	return do_qb(cmdtp, flag, argc, argv, true);
}

static int do_qb_erase(struct cmd_tbl *cmdtp, int flag,
		       int argc, char * const argv[])
{
	return do_qb(cmdtp, flag, argc, argv, false);
}

static struct cmd_tbl cmd_qb[] = {
	U_BOOT_CMD_MKENT(check, 1, 1, do_qb_check, "", ""),
	U_BOOT_CMD_MKENT(save,  3, 1, do_qb_save,  "", ""),
	U_BOOT_CMD_MKENT(erase, 3, 1, do_qb_erase, "", ""),
};

static int do_qbops(struct cmd_tbl *cmdtp, int flag, int argc,
		    char *const argv[])
{
	struct cmd_tbl *cp;

	cp = find_cmd_tbl(argv[1], cmd_qb, ARRAY_SIZE(cmd_qb));

	/* Drop the qb command */
	argc--;
	argv++;

	if (!cp) {
		printf("qb: %s: command not found\n", argv[0] ? argv[0] : " ");
		return CMD_RET_USAGE;
	}

	if (argc > cp->maxargs) {
		printf("qb %s: too many arguments: %d > %d\n", cp->name,
		       argc - 1, cp->maxargs - 1);
		return CMD_RET_USAGE;
	}

	if (flag == CMD_FLAG_REPEAT && !cmd_is_repeatable(cp)) {
		printf("qb %s: repeat flag set but command is not repeatable\n",
		       cp->name);
		return CMD_RET_SUCCESS;
	}

	return cp->cmd(cmdtp, flag, argc, argv);
}

U_BOOT_CMD(
	qb, 4, 1, do_qbops,
	"DDR Quick Boot sub system",
	"check - check if quick boot data is stored in mem by training flow\n"
	"qb save [interface] [dev]  - save quick boot data in NVM => trigger quick boot flow\n"
	"qb erase [interface] [dev] - erase quick boot data from NVM => trigger training flow\n"
);