part.c 5.6 KB

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