virtio_mmio.c 10.0 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414
  1. // SPDX-License-Identifier: GPL-2.0+
  2. /*
  3. * Copyright (C) 2018, Tuomas Tynkkynen <tuomas.tynkkynen@iki.fi>
  4. * Copyright (C) 2018, Bin Meng <bmeng.cn@gmail.com>
  5. *
  6. * VirtIO memory-maped I/O transport driver
  7. * Ported from Linux drivers/virtio/virtio_mmio.c
  8. */
  9. #include <common.h>
  10. #include <dm.h>
  11. #include <virtio_types.h>
  12. #include <virtio.h>
  13. #include <virtio_ring.h>
  14. #include <linux/compat.h>
  15. #include <linux/err.h>
  16. #include <linux/io.h>
  17. #include "virtio_mmio.h"
  18. static int virtio_mmio_get_config(struct udevice *udev, unsigned int offset,
  19. void *buf, unsigned int len)
  20. {
  21. struct virtio_mmio_priv *priv = dev_get_priv(udev);
  22. void __iomem *base = priv->base + VIRTIO_MMIO_CONFIG;
  23. u8 b;
  24. __le16 w;
  25. __le32 l;
  26. if (priv->version == 1) {
  27. u8 *ptr = buf;
  28. int i;
  29. for (i = 0; i < len; i++)
  30. ptr[i] = readb(base + offset + i);
  31. return 0;
  32. }
  33. switch (len) {
  34. case 1:
  35. b = readb(base + offset);
  36. memcpy(buf, &b, sizeof(b));
  37. break;
  38. case 2:
  39. w = cpu_to_le16(readw(base + offset));
  40. memcpy(buf, &w, sizeof(w));
  41. break;
  42. case 4:
  43. l = cpu_to_le32(readl(base + offset));
  44. memcpy(buf, &l, sizeof(l));
  45. break;
  46. case 8:
  47. l = cpu_to_le32(readl(base + offset));
  48. memcpy(buf, &l, sizeof(l));
  49. l = cpu_to_le32(readl(base + offset + sizeof(l)));
  50. memcpy(buf + sizeof(l), &l, sizeof(l));
  51. break;
  52. default:
  53. WARN_ON(true);
  54. }
  55. return 0;
  56. }
  57. static int virtio_mmio_set_config(struct udevice *udev, unsigned int offset,
  58. const void *buf, unsigned int len)
  59. {
  60. struct virtio_mmio_priv *priv = dev_get_priv(udev);
  61. void __iomem *base = priv->base + VIRTIO_MMIO_CONFIG;
  62. u8 b;
  63. __le16 w;
  64. __le32 l;
  65. if (priv->version == 1) {
  66. const u8 *ptr = buf;
  67. int i;
  68. for (i = 0; i < len; i++)
  69. writeb(ptr[i], base + offset + i);
  70. return 0;
  71. }
  72. switch (len) {
  73. case 1:
  74. memcpy(&b, buf, sizeof(b));
  75. writeb(b, base + offset);
  76. break;
  77. case 2:
  78. memcpy(&w, buf, sizeof(w));
  79. writew(le16_to_cpu(w), base + offset);
  80. break;
  81. case 4:
  82. memcpy(&l, buf, sizeof(l));
  83. writel(le32_to_cpu(l), base + offset);
  84. break;
  85. case 8:
  86. memcpy(&l, buf, sizeof(l));
  87. writel(le32_to_cpu(l), base + offset);
  88. memcpy(&l, buf + sizeof(l), sizeof(l));
  89. writel(le32_to_cpu(l), base + offset + sizeof(l));
  90. break;
  91. default:
  92. WARN_ON(true);
  93. }
  94. return 0;
  95. }
  96. static int virtio_mmio_generation(struct udevice *udev, u32 *counter)
  97. {
  98. struct virtio_mmio_priv *priv = dev_get_priv(udev);
  99. if (priv->version == 1)
  100. *counter = 0;
  101. else
  102. *counter = readl(priv->base + VIRTIO_MMIO_CONFIG_GENERATION);
  103. return 0;
  104. }
  105. static int virtio_mmio_get_status(struct udevice *udev, u8 *status)
  106. {
  107. struct virtio_mmio_priv *priv = dev_get_priv(udev);
  108. *status = readl(priv->base + VIRTIO_MMIO_STATUS) & 0xff;
  109. return 0;
  110. }
  111. static int virtio_mmio_set_status(struct udevice *udev, u8 status)
  112. {
  113. struct virtio_mmio_priv *priv = dev_get_priv(udev);
  114. /* We should never be setting status to 0 */
  115. WARN_ON(status == 0);
  116. writel(status, priv->base + VIRTIO_MMIO_STATUS);
  117. return 0;
  118. }
  119. static int virtio_mmio_reset(struct udevice *udev)
  120. {
  121. struct virtio_mmio_priv *priv = dev_get_priv(udev);
  122. /* 0 status means a reset */
  123. writel(0, priv->base + VIRTIO_MMIO_STATUS);
  124. return 0;
  125. }
  126. static int virtio_mmio_get_features(struct udevice *udev, u64 *features)
  127. {
  128. struct virtio_mmio_priv *priv = dev_get_priv(udev);
  129. writel(1, priv->base + VIRTIO_MMIO_DEVICE_FEATURES_SEL);
  130. *features = readl(priv->base + VIRTIO_MMIO_DEVICE_FEATURES);
  131. *features <<= 32;
  132. writel(0, priv->base + VIRTIO_MMIO_DEVICE_FEATURES_SEL);
  133. *features |= readl(priv->base + VIRTIO_MMIO_DEVICE_FEATURES);
  134. return 0;
  135. }
  136. static int virtio_mmio_set_features(struct udevice *udev)
  137. {
  138. struct virtio_mmio_priv *priv = dev_get_priv(udev);
  139. struct virtio_dev_priv *uc_priv = dev_get_uclass_priv(udev);
  140. /* Make sure there is are no mixed devices */
  141. if (priv->version == 2 && uc_priv->legacy) {
  142. debug("New virtio-mmio devices (version 2) must provide VIRTIO_F_VERSION_1 feature!\n");
  143. return -EINVAL;
  144. }
  145. writel(1, priv->base + VIRTIO_MMIO_DRIVER_FEATURES_SEL);
  146. writel((u32)(uc_priv->features >> 32),
  147. priv->base + VIRTIO_MMIO_DRIVER_FEATURES);
  148. writel(0, priv->base + VIRTIO_MMIO_DRIVER_FEATURES_SEL);
  149. writel((u32)uc_priv->features,
  150. priv->base + VIRTIO_MMIO_DRIVER_FEATURES);
  151. return 0;
  152. }
  153. static struct virtqueue *virtio_mmio_setup_vq(struct udevice *udev,
  154. unsigned int index)
  155. {
  156. struct virtio_mmio_priv *priv = dev_get_priv(udev);
  157. struct virtqueue *vq;
  158. unsigned int num;
  159. int err;
  160. /* Select the queue we're interested in */
  161. writel(index, priv->base + VIRTIO_MMIO_QUEUE_SEL);
  162. /* Queue shouldn't already be set up */
  163. if (readl(priv->base + (priv->version == 1 ?
  164. VIRTIO_MMIO_QUEUE_PFN : VIRTIO_MMIO_QUEUE_READY))) {
  165. err = -ENOENT;
  166. goto error_available;
  167. }
  168. num = readl(priv->base + VIRTIO_MMIO_QUEUE_NUM_MAX);
  169. if (num == 0) {
  170. err = -ENOENT;
  171. goto error_new_virtqueue;
  172. }
  173. /* Create the vring */
  174. vq = vring_create_virtqueue(index, num, VIRTIO_MMIO_VRING_ALIGN, udev);
  175. if (!vq) {
  176. err = -ENOMEM;
  177. goto error_new_virtqueue;
  178. }
  179. /* Activate the queue */
  180. writel(virtqueue_get_vring_size(vq),
  181. priv->base + VIRTIO_MMIO_QUEUE_NUM);
  182. if (priv->version == 1) {
  183. u64 q_pfn = virtqueue_get_desc_addr(vq) >> PAGE_SHIFT;
  184. /*
  185. * virtio-mmio v1 uses a 32bit QUEUE PFN. If we have something
  186. * that doesn't fit in 32bit, fail the setup rather than
  187. * pretending to be successful.
  188. */
  189. if (q_pfn >> 32) {
  190. debug("platform bug: legacy virtio-mmio must not be used with RAM above 0x%llxGB\n",
  191. 0x1ULL << (32 + PAGE_SHIFT - 30));
  192. err = -E2BIG;
  193. goto error_bad_pfn;
  194. }
  195. writel(PAGE_SIZE, priv->base + VIRTIO_MMIO_QUEUE_ALIGN);
  196. writel(q_pfn, priv->base + VIRTIO_MMIO_QUEUE_PFN);
  197. } else {
  198. u64 addr;
  199. addr = virtqueue_get_desc_addr(vq);
  200. writel((u32)addr, priv->base + VIRTIO_MMIO_QUEUE_DESC_LOW);
  201. writel((u32)(addr >> 32),
  202. priv->base + VIRTIO_MMIO_QUEUE_DESC_HIGH);
  203. addr = virtqueue_get_avail_addr(vq);
  204. writel((u32)addr, priv->base + VIRTIO_MMIO_QUEUE_AVAIL_LOW);
  205. writel((u32)(addr >> 32),
  206. priv->base + VIRTIO_MMIO_QUEUE_AVAIL_HIGH);
  207. addr = virtqueue_get_used_addr(vq);
  208. writel((u32)addr, priv->base + VIRTIO_MMIO_QUEUE_USED_LOW);
  209. writel((u32)(addr >> 32),
  210. priv->base + VIRTIO_MMIO_QUEUE_USED_HIGH);
  211. writel(1, priv->base + VIRTIO_MMIO_QUEUE_READY);
  212. }
  213. return vq;
  214. error_bad_pfn:
  215. vring_del_virtqueue(vq);
  216. error_new_virtqueue:
  217. if (priv->version == 1) {
  218. writel(0, priv->base + VIRTIO_MMIO_QUEUE_PFN);
  219. } else {
  220. writel(0, priv->base + VIRTIO_MMIO_QUEUE_READY);
  221. WARN_ON(readl(priv->base + VIRTIO_MMIO_QUEUE_READY));
  222. }
  223. error_available:
  224. return ERR_PTR(err);
  225. }
  226. static void virtio_mmio_del_vq(struct virtqueue *vq)
  227. {
  228. struct virtio_mmio_priv *priv = dev_get_priv(vq->vdev);
  229. unsigned int index = vq->index;
  230. /* Select and deactivate the queue */
  231. writel(index, priv->base + VIRTIO_MMIO_QUEUE_SEL);
  232. if (priv->version == 1) {
  233. writel(0, priv->base + VIRTIO_MMIO_QUEUE_PFN);
  234. } else {
  235. writel(0, priv->base + VIRTIO_MMIO_QUEUE_READY);
  236. WARN_ON(readl(priv->base + VIRTIO_MMIO_QUEUE_READY));
  237. }
  238. vring_del_virtqueue(vq);
  239. }
  240. static int virtio_mmio_del_vqs(struct udevice *udev)
  241. {
  242. struct virtio_dev_priv *uc_priv = dev_get_uclass_priv(udev);
  243. struct virtqueue *vq, *n;
  244. list_for_each_entry_safe(vq, n, &uc_priv->vqs, list)
  245. virtio_mmio_del_vq(vq);
  246. return 0;
  247. }
  248. static int virtio_mmio_find_vqs(struct udevice *udev, unsigned int nvqs,
  249. struct virtqueue *vqs[])
  250. {
  251. int i;
  252. for (i = 0; i < nvqs; ++i) {
  253. vqs[i] = virtio_mmio_setup_vq(udev, i);
  254. if (IS_ERR(vqs[i])) {
  255. virtio_mmio_del_vqs(udev);
  256. return PTR_ERR(vqs[i]);
  257. }
  258. }
  259. return 0;
  260. }
  261. static int virtio_mmio_notify(struct udevice *udev, struct virtqueue *vq)
  262. {
  263. struct virtio_mmio_priv *priv = dev_get_priv(udev);
  264. /*
  265. * We write the queue's selector into the notification register
  266. * to signal the other end
  267. */
  268. writel(vq->index, priv->base + VIRTIO_MMIO_QUEUE_NOTIFY);
  269. return 0;
  270. }
  271. static int virtio_mmio_ofdata_to_platdata(struct udevice *udev)
  272. {
  273. struct virtio_mmio_priv *priv = dev_get_priv(udev);
  274. priv->base = (void __iomem *)(ulong)dev_read_addr(udev);
  275. if (priv->base == (void __iomem *)FDT_ADDR_T_NONE)
  276. return -EINVAL;
  277. return 0;
  278. }
  279. static int virtio_mmio_probe(struct udevice *udev)
  280. {
  281. struct virtio_mmio_priv *priv = dev_get_priv(udev);
  282. struct virtio_dev_priv *uc_priv = dev_get_uclass_priv(udev);
  283. u32 magic;
  284. /* Check magic value */
  285. magic = readl(priv->base + VIRTIO_MMIO_MAGIC_VALUE);
  286. if (magic != ('v' | 'i' << 8 | 'r' << 16 | 't' << 24)) {
  287. debug("(%s): wrong magic value 0x%08x!\n", udev->name, magic);
  288. return 0;
  289. }
  290. /* Check device version */
  291. priv->version = readl(priv->base + VIRTIO_MMIO_VERSION);
  292. if (priv->version < 1 || priv->version > 2) {
  293. debug("(%s): version %d not supported!\n",
  294. udev->name, priv->version);
  295. return 0;
  296. }
  297. /* Check device ID */
  298. uc_priv->device = readl(priv->base + VIRTIO_MMIO_DEVICE_ID);
  299. if (uc_priv->device == 0) {
  300. /*
  301. * virtio-mmio device with an ID 0 is a (dummy) placeholder
  302. * with no function. End probing now with no error reported.
  303. */
  304. return 0;
  305. }
  306. uc_priv->vendor = readl(priv->base + VIRTIO_MMIO_VENDOR_ID);
  307. if (priv->version == 1)
  308. writel(PAGE_SIZE, priv->base + VIRTIO_MMIO_GUEST_PAGE_SIZE);
  309. debug("(%s): device (%d) vendor (%08x) version (%d)\n", udev->name,
  310. uc_priv->device, uc_priv->vendor, priv->version);
  311. return 0;
  312. }
  313. static const struct dm_virtio_ops virtio_mmio_ops = {
  314. .get_config = virtio_mmio_get_config,
  315. .set_config = virtio_mmio_set_config,
  316. .generation = virtio_mmio_generation,
  317. .get_status = virtio_mmio_get_status,
  318. .set_status = virtio_mmio_set_status,
  319. .reset = virtio_mmio_reset,
  320. .get_features = virtio_mmio_get_features,
  321. .set_features = virtio_mmio_set_features,
  322. .find_vqs = virtio_mmio_find_vqs,
  323. .del_vqs = virtio_mmio_del_vqs,
  324. .notify = virtio_mmio_notify,
  325. };
  326. static const struct udevice_id virtio_mmio_ids[] = {
  327. { .compatible = "virtio,mmio" },
  328. { }
  329. };
  330. U_BOOT_DRIVER(virtio_mmio) = {
  331. .name = "virtio-mmio",
  332. .id = UCLASS_VIRTIO,
  333. .of_match = virtio_mmio_ids,
  334. .ops = &virtio_mmio_ops,
  335. .probe = virtio_mmio_probe,
  336. .ofdata_to_platdata = virtio_mmio_ofdata_to_platdata,
  337. .priv_auto_alloc_size = sizeof(struct virtio_mmio_priv),
  338. };