part.c 5.0 KB

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