part.c 4.4 KB

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