1 // SPDX-License-Identifier: GPL-2.0+ 2 /* 3 * Copyright (c) 2012, NVIDIA CORPORATION. All rights reserved. 4 * 5 * made from cmd_ext2, which was: 6 * 7 * (C) Copyright 2004 8 * esd gmbh <www.esd-electronics.com> 9 * Reinhard Arlt <reinhard.arlt@esd-electronics.com> 10 * 11 * made from cmd_reiserfs by 12 * 13 * (C) Copyright 2003 - 2004 14 * Sysgo Real-Time Solutions, AG <www.elinos.com> 15 * Pavel Bartusek <pba@sysgo.com> 16 */ 17 18 #include <common.h> 19 #include <config.h> 20 #include <command.h> 21 #include <part.h> 22 #include <vsprintf.h> 23 24 enum cmd_part_info { 25 CMD_PART_INFO_START = 0, 26 CMD_PART_INFO_SIZE, 27 }; 28 29 static int do_part_uuid(int argc, char * const argv[]) 30 { 31 int part; 32 struct blk_desc *dev_desc; 33 disk_partition_t info; 34 35 if (argc < 2) 36 return CMD_RET_USAGE; 37 if (argc > 3) 38 return CMD_RET_USAGE; 39 40 part = blk_get_device_part_str(argv[0], argv[1], &dev_desc, &info, 0); 41 if (part < 0) 42 return 1; 43 44 if (argc > 2) 45 env_set(argv[2], info.uuid); 46 else 47 printf("%s\n", info.uuid); 48 49 return 0; 50 } 51 52 static int do_part_list(int argc, char * const argv[]) 53 { 54 int ret; 55 struct blk_desc *desc; 56 char *var = NULL; 57 bool bootable = false; 58 int i; 59 60 if (argc < 2) 61 return CMD_RET_USAGE; 62 63 if (argc > 2) { 64 for (i = 2; i < argc ; i++) { 65 if (argv[i][0] == '-') { 66 if (!strcmp(argv[i], "-bootable")) { 67 bootable = true; 68 } else { 69 printf("Unknown option %s\n", argv[i]); 70 return CMD_RET_USAGE; 71 } 72 } else { 73 var = argv[i]; 74 break; 75 } 76 } 77 78 /* Loops should have been exited at the last argument, which 79 * as it contained the variable */ 80 if (argc != i + 1) 81 return CMD_RET_USAGE; 82 } 83 84 ret = blk_get_device_by_str(argv[0], argv[1], &desc); 85 if (ret < 0) 86 return 1; 87 88 if (var != NULL) { 89 int p; 90 char str[512] = { '\0', }; 91 disk_partition_t info; 92 93 for (p = 1; p < MAX_SEARCH_PARTITIONS; p++) { 94 char t[5]; 95 int r = part_get_info(desc, p, &info); 96 97 if (r != 0) 98 continue; 99 100 if (bootable && !info.bootable) 101 continue; 102 103 sprintf(t, "%s%x", str[0] ? " " : "", p); 104 strcat(str, t); 105 } 106 env_set(var, str); 107 return 0; 108 } 109 110 part_print(desc); 111 112 return 0; 113 } 114 115 static int do_part_info(int argc, char * const argv[], enum cmd_part_info param) 116 { 117 struct blk_desc *desc; 118 disk_partition_t info; 119 char buf[512] = { 0 }; 120 char *endp; 121 int part; 122 int err; 123 int ret; 124 125 if (argc < 3) 126 return CMD_RET_USAGE; 127 if (argc > 4) 128 return CMD_RET_USAGE; 129 130 ret = blk_get_device_by_str(argv[0], argv[1], &desc); 131 if (ret < 0) 132 return 1; 133 134 part = simple_strtoul(argv[2], &endp, 0); 135 if (*endp == '\0') { 136 err = part_get_info(desc, part, &info); 137 if (err) 138 return 1; 139 } else { 140 part = part_get_info_by_name(desc, argv[2], &info); 141 if (part == -1) 142 return 1; 143 } 144 145 switch (param) { 146 case CMD_PART_INFO_START: 147 snprintf(buf, sizeof(buf), LBAF, info.start); 148 break; 149 case CMD_PART_INFO_SIZE: 150 snprintf(buf, sizeof(buf), LBAF, info.size); 151 break; 152 default: 153 printf("** Unknown cmd_part_info value: %d\n", param); 154 return 1; 155 } 156 157 if (argc > 3) 158 env_set(argv[3], buf); 159 else 160 printf("%s\n", buf); 161 162 return 0; 163 } 164 165 static int do_part_start(int argc, char * const argv[]) 166 { 167 return do_part_info(argc, argv, CMD_PART_INFO_START); 168 } 169 170 static int do_part_size(int argc, char * const argv[]) 171 { 172 return do_part_info(argc, argv, CMD_PART_INFO_SIZE); 173 } 174 175 static int do_part(cmd_tbl_t *cmdtp, int flag, int argc, char * const argv[]) 176 { 177 if (argc < 2) 178 return CMD_RET_USAGE; 179 180 if (!strcmp(argv[1], "uuid")) 181 return do_part_uuid(argc - 2, argv + 2); 182 else if (!strcmp(argv[1], "list")) 183 return do_part_list(argc - 2, argv + 2); 184 else if (!strcmp(argv[1], "start")) 185 return do_part_start(argc - 2, argv + 2); 186 else if (!strcmp(argv[1], "size")) 187 return do_part_size(argc - 2, argv + 2); 188 189 return CMD_RET_USAGE; 190 } 191 192 U_BOOT_CMD( 193 part, CONFIG_SYS_MAXARGS, 1, do_part, 194 "disk partition related commands", 195 "uuid <interface> <dev>:<part>\n" 196 " - print partition UUID\n" 197 "part uuid <interface> <dev>:<part> <varname>\n" 198 " - set environment variable to partition UUID\n" 199 "part list <interface> <dev>\n" 200 " - print a device's partition table\n" 201 "part list <interface> <dev> [flags] <varname>\n" 202 " - set environment variable to the list of partitions\n" 203 " flags can be -bootable (list only bootable partitions)\n" 204 "part start <interface> <dev> <part> <varname>\n" 205 " - set environment variable to the start of the partition (in blocks)\n" 206 " part can be either partition number or partition name\n" 207 "part size <interface> <dev> <part> <varname>\n" 208 " - set environment variable to the size of the partition (in blocks)\n" 209 " part can be either partition number or partition name" 210 ); 211