cmd/ab_select.c

   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
  12 static 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
  54 U_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 );