1 /*
2  * Copyright (C) 2017 The Android Open Source Project
3  *
4  * SPDX-License-Identifier: BSD-2-Clause
5  */
6 
7 #include <android_cmds.h>
8 #include <android_ab.h>
9 #include <common.h>
10 #include <command.h>
11 
do_android_ab_select(cmd_tbl_t * cmdtp,int flag,int argc,char * const argv[])12 static int do_android_ab_select(cmd_tbl_t *cmdtp, int flag, int argc,
13 				char * const argv[])
14 {
15 	int ret;
16 	struct blk_desc *dev_desc;
17 	disk_partition_t part_info;
18 	char slot[2];
19 
20 	if (argc != 4)
21 		return CMD_RET_USAGE;
22 
23 	/* Lookup the "misc" partition from argv[2] and argv[3] */
24 	if (part_get_info_by_dev_and_name_or_num(argv[2], argv[3],
25 						 &dev_desc, &part_info) < 0) {
26 		return CMD_RET_FAILURE;
27 	}
28 
29 	ret = android_ab_select(dev_desc, &part_info);
30 	if (ret < 0) {
31 		printf("Android boot failed, error %d.\n", ret);
32 		return CMD_RET_FAILURE;
33 	}
34 
35 	/* Android standard slot names are 'a', 'b', ... */
36 	slot[0] = ANDROID_BOOT_SLOT_NAME(ret);
37 	slot[1] = '\0';
38 	env_set(argv[1], slot);
39 	printf("ANDROID: Booting slot: %s\n", slot);
40 	return CMD_RET_SUCCESS;
41 }
42 
43 U_BOOT_CMD(
44 	android_ab_select, 4, 0, do_android_ab_select,
45 	"Select the slot used to boot from and register the boot attempt.",
46 	"<slot_var_name> <interface> <dev[:part|;part_name]>\n"
47 	"    - Load the slot metadata from the partition 'part' on\n"
48 	"      device type 'interface' instance 'dev' and store the active\n"
49 	"      slot in the 'slot_var_name' variable. This also updates the\n"
50 	"      Android slot metadata with a boot attempt, which can cause\n"
51 	"      successive calls to this function to return a different result\n"
52 	"      if the returned slot runs out of boot attempts.\n"
53 	"    - If 'part_name' is passed, preceded with a ; instead of :, the\n"
54 	"      partition name whose label is 'part_name' will be looked up in\n"
55 	"      the partition table. This is commonly the \"misc\" partition.\n"
56 );
57