disk-uclass.c 5.2 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247
  1. // SPDX-License-Identifier: GPL-2.0+
  2. /*
  3. * Software partition device (UCLASS_PARTITION)
  4. *
  5. * Copyright (c) 2021 Linaro Limited
  6. * Author: AKASHI Takahiro
  7. */
  8. #define LOG_CATEGORY UCLASS_PARTITION
  9. #include <common.h>
  10. #include <blk.h>
  11. #include <dm.h>
  12. #include <log.h>
  13. #include <part.h>
  14. #include <vsprintf.h>
  15. #include <dm/device-internal.h>
  16. #include <dm/lists.h>
  17. int part_create_block_devices(struct udevice *blk_dev)
  18. {
  19. int part, count;
  20. struct blk_desc *desc = dev_get_uclass_plat(blk_dev);
  21. struct disk_partition info;
  22. struct disk_part *part_data;
  23. char devname[32];
  24. struct udevice *dev;
  25. int ret;
  26. if (!CONFIG_IS_ENABLED(PARTITIONS) || !blk_enabled())
  27. return 0;
  28. if (device_get_uclass_id(blk_dev) != UCLASS_BLK)
  29. return 0;
  30. /* Add devices for each partition */
  31. for (count = 0, part = 1; part <= MAX_SEARCH_PARTITIONS; part++) {
  32. if (part_get_info(desc, part, &info))
  33. continue;
  34. snprintf(devname, sizeof(devname), "%s:%d", blk_dev->name,
  35. part);
  36. ret = device_bind_driver(blk_dev, "blk_partition",
  37. strdup(devname), &dev);
  38. if (ret)
  39. return ret;
  40. part_data = dev_get_uclass_plat(dev);
  41. part_data->partnum = part;
  42. part_data->gpt_part_info = info;
  43. count++;
  44. ret = device_probe(dev);
  45. if (ret) {
  46. debug("Can't probe\n");
  47. count--;
  48. device_unbind(dev);
  49. continue;
  50. }
  51. }
  52. debug("%s: %d partitions found in %s\n", __func__, count,
  53. blk_dev->name);
  54. return 0;
  55. }
  56. static ulong part_blk_read(struct udevice *dev, lbaint_t start,
  57. lbaint_t blkcnt, void *buffer)
  58. {
  59. struct udevice *parent;
  60. struct disk_part *part;
  61. const struct blk_ops *ops;
  62. parent = dev_get_parent(dev);
  63. ops = blk_get_ops(parent);
  64. if (!ops->read)
  65. return -ENOSYS;
  66. part = dev_get_uclass_plat(dev);
  67. if (start >= part->gpt_part_info.size)
  68. return 0;
  69. if ((start + blkcnt) > part->gpt_part_info.size)
  70. blkcnt = part->gpt_part_info.size - start;
  71. start += part->gpt_part_info.start;
  72. return ops->read(parent, start, blkcnt, buffer);
  73. }
  74. static ulong part_blk_write(struct udevice *dev, lbaint_t start,
  75. lbaint_t blkcnt, const void *buffer)
  76. {
  77. struct udevice *parent;
  78. struct disk_part *part;
  79. const struct blk_ops *ops;
  80. parent = dev_get_parent(dev);
  81. ops = blk_get_ops(parent);
  82. if (!ops->write)
  83. return -ENOSYS;
  84. part = dev_get_uclass_plat(dev);
  85. if (start >= part->gpt_part_info.size)
  86. return 0;
  87. if ((start + blkcnt) > part->gpt_part_info.size)
  88. blkcnt = part->gpt_part_info.size - start;
  89. start += part->gpt_part_info.start;
  90. return ops->write(parent, start, blkcnt, buffer);
  91. }
  92. static ulong part_blk_erase(struct udevice *dev, lbaint_t start,
  93. lbaint_t blkcnt)
  94. {
  95. struct udevice *parent;
  96. struct disk_part *part;
  97. const struct blk_ops *ops;
  98. parent = dev_get_parent(dev);
  99. ops = blk_get_ops(parent);
  100. if (!ops->erase)
  101. return -ENOSYS;
  102. part = dev_get_uclass_plat(dev);
  103. if (start >= part->gpt_part_info.size)
  104. return 0;
  105. if ((start + blkcnt) > part->gpt_part_info.size)
  106. blkcnt = part->gpt_part_info.size - start;
  107. start += part->gpt_part_info.start;
  108. return ops->erase(parent, start, blkcnt);
  109. }
  110. static const struct blk_ops blk_part_ops = {
  111. .read = part_blk_read,
  112. .write = part_blk_write,
  113. .erase = part_blk_erase,
  114. };
  115. U_BOOT_DRIVER(blk_partition) = {
  116. .name = "blk_partition",
  117. .id = UCLASS_PARTITION,
  118. .ops = &blk_part_ops,
  119. };
  120. /*
  121. * BLOCK IO APIs
  122. */
  123. static struct blk_desc *dev_get_blk(struct udevice *dev)
  124. {
  125. struct blk_desc *desc;
  126. switch (device_get_uclass_id(dev)) {
  127. /*
  128. * We won't support UCLASS_BLK with dev_* interfaces.
  129. */
  130. case UCLASS_PARTITION:
  131. desc = dev_get_uclass_plat(dev_get_parent(dev));
  132. break;
  133. default:
  134. desc = NULL;
  135. break;
  136. }
  137. return desc;
  138. }
  139. unsigned long disk_blk_read(struct udevice *dev, lbaint_t start,
  140. lbaint_t blkcnt, void *buffer)
  141. {
  142. struct blk_desc *desc;
  143. const struct blk_ops *ops;
  144. struct disk_part *part;
  145. lbaint_t start_in_disk;
  146. ulong blks_read;
  147. desc = dev_get_blk(dev);
  148. if (!desc)
  149. return -ENOSYS;
  150. ops = blk_get_ops(dev);
  151. if (!ops->read)
  152. return -ENOSYS;
  153. start_in_disk = start;
  154. if (device_get_uclass_id(dev) == UCLASS_PARTITION) {
  155. part = dev_get_uclass_plat(dev);
  156. start_in_disk += part->gpt_part_info.start;
  157. }
  158. if (blkcache_read(desc->uclass_id, desc->devnum, start_in_disk, blkcnt,
  159. desc->blksz, buffer))
  160. return blkcnt;
  161. blks_read = ops->read(dev, start, blkcnt, buffer);
  162. if (blks_read == blkcnt)
  163. blkcache_fill(desc->uclass_id, desc->devnum, start_in_disk,
  164. blkcnt, desc->blksz, buffer);
  165. return blks_read;
  166. }
  167. unsigned long disk_blk_write(struct udevice *dev, lbaint_t start,
  168. lbaint_t blkcnt, const void *buffer)
  169. {
  170. struct blk_desc *desc;
  171. const struct blk_ops *ops;
  172. desc = dev_get_blk(dev);
  173. if (!desc)
  174. return -ENOSYS;
  175. ops = blk_get_ops(dev);
  176. if (!ops->write)
  177. return -ENOSYS;
  178. blkcache_invalidate(desc->uclass_id, desc->devnum);
  179. return ops->write(dev, start, blkcnt, buffer);
  180. }
  181. unsigned long disk_blk_erase(struct udevice *dev, lbaint_t start,
  182. lbaint_t blkcnt)
  183. {
  184. struct blk_desc *desc;
  185. const struct blk_ops *ops;
  186. desc = dev_get_blk(dev);
  187. if (!desc)
  188. return -ENOSYS;
  189. ops = blk_get_ops(dev);
  190. if (!ops->erase)
  191. return -ENOSYS;
  192. blkcache_invalidate(desc->uclass_id, desc->devnum);
  193. return ops->erase(dev, start, blkcnt);
  194. }
  195. UCLASS_DRIVER(partition) = {
  196. .id = UCLASS_PARTITION,
  197. .per_device_plat_auto = sizeof(struct disk_part),
  198. .name = "partition",
  199. };