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