1
2
3
4
5
6
7#include <linux/device.h>
8#include <linux/iommu.h>
9#include <linux/module.h>
10#include <linux/mutex.h>
11#include <linux/slab.h>
12#include <linux/types.h>
13#include <linux/vfio.h>
14#include <linux/fsl/mc.h>
15#include <linux/delay.h>
16#include <linux/io-64-nonatomic-hi-lo.h>
17
18#include "vfio_fsl_mc_private.h"
19
20static struct fsl_mc_driver vfio_fsl_mc_driver;
21
22static int vfio_fsl_mc_open_device(struct vfio_device *core_vdev)
23{
24 struct vfio_fsl_mc_device *vdev =
25 container_of(core_vdev, struct vfio_fsl_mc_device, vdev);
26 struct fsl_mc_device *mc_dev = vdev->mc_dev;
27 int count = mc_dev->obj_desc.region_count;
28 int i;
29
30 vdev->regions = kcalloc(count, sizeof(struct vfio_fsl_mc_region),
31 GFP_KERNEL);
32 if (!vdev->regions)
33 return -ENOMEM;
34
35 for (i = 0; i < count; i++) {
36 struct resource *res = &mc_dev->regions[i];
37 int no_mmap = is_fsl_mc_bus_dprc(mc_dev);
38
39 vdev->regions[i].addr = res->start;
40 vdev->regions[i].size = resource_size(res);
41 vdev->regions[i].type = mc_dev->regions[i].flags & IORESOURCE_BITS;
42
43
44
45
46 if (!no_mmap && !(vdev->regions[i].addr & ~PAGE_MASK) &&
47 !(vdev->regions[i].size & ~PAGE_MASK))
48 vdev->regions[i].flags |=
49 VFIO_REGION_INFO_FLAG_MMAP;
50 vdev->regions[i].flags |= VFIO_REGION_INFO_FLAG_READ;
51 if (!(mc_dev->regions[i].flags & IORESOURCE_READONLY))
52 vdev->regions[i].flags |= VFIO_REGION_INFO_FLAG_WRITE;
53 }
54
55 return 0;
56}
57
58static void vfio_fsl_mc_regions_cleanup(struct vfio_fsl_mc_device *vdev)
59{
60 struct fsl_mc_device *mc_dev = vdev->mc_dev;
61 int i;
62
63 for (i = 0; i < mc_dev->obj_desc.region_count; i++)
64 iounmap(vdev->regions[i].ioaddr);
65 kfree(vdev->regions);
66}
67
68static int vfio_fsl_mc_reset_device(struct vfio_fsl_mc_device *vdev)
69{
70 struct fsl_mc_device *mc_dev = vdev->mc_dev;
71 int ret = 0;
72
73 if (is_fsl_mc_bus_dprc(vdev->mc_dev)) {
74 return dprc_reset_container(mc_dev->mc_io, 0,
75 mc_dev->mc_handle,
76 mc_dev->obj_desc.id,
77 DPRC_RESET_OPTION_NON_RECURSIVE);
78 } else {
79 u16 token;
80
81 ret = fsl_mc_obj_open(mc_dev->mc_io, 0, mc_dev->obj_desc.id,
82 mc_dev->obj_desc.type,
83 &token);
84 if (ret)
85 goto out;
86 ret = fsl_mc_obj_reset(mc_dev->mc_io, 0, token);
87 if (ret) {
88 fsl_mc_obj_close(mc_dev->mc_io, 0, token);
89 goto out;
90 }
91 ret = fsl_mc_obj_close(mc_dev->mc_io, 0, token);
92 }
93out:
94 return ret;
95}
96
97static void vfio_fsl_mc_close_device(struct vfio_device *core_vdev)
98{
99 struct vfio_fsl_mc_device *vdev =
100 container_of(core_vdev, struct vfio_fsl_mc_device, vdev);
101 struct fsl_mc_device *mc_dev = vdev->mc_dev;
102 struct device *cont_dev = fsl_mc_cont_dev(&mc_dev->dev);
103 struct fsl_mc_device *mc_cont = to_fsl_mc_device(cont_dev);
104 int ret;
105
106 vfio_fsl_mc_regions_cleanup(vdev);
107
108
109 ret = vfio_fsl_mc_reset_device(vdev);
110
111 if (WARN_ON(ret))
112 dev_warn(&mc_cont->dev,
113 "VFIO_FLS_MC: reset device has failed (%d)\n", ret);
114
115 vfio_fsl_mc_irqs_cleanup(vdev);
116
117 fsl_mc_cleanup_irq_pool(mc_cont);
118}
119
120static long vfio_fsl_mc_ioctl(struct vfio_device *core_vdev,
121 unsigned int cmd, unsigned long arg)
122{
123 unsigned long minsz;
124 struct vfio_fsl_mc_device *vdev =
125 container_of(core_vdev, struct vfio_fsl_mc_device, vdev);
126 struct fsl_mc_device *mc_dev = vdev->mc_dev;
127
128 switch (cmd) {
129 case VFIO_DEVICE_GET_INFO:
130 {
131 struct vfio_device_info info;
132
133 minsz = offsetofend(struct vfio_device_info, num_irqs);
134
135 if (copy_from_user(&info, (void __user *)arg, minsz))
136 return -EFAULT;
137
138 if (info.argsz < minsz)
139 return -EINVAL;
140
141 info.flags = VFIO_DEVICE_FLAGS_FSL_MC;
142
143 if (is_fsl_mc_bus_dprc(mc_dev))
144 info.flags |= VFIO_DEVICE_FLAGS_RESET;
145
146 info.num_regions = mc_dev->obj_desc.region_count;
147 info.num_irqs = mc_dev->obj_desc.irq_count;
148
149 return copy_to_user((void __user *)arg, &info, minsz) ?
150 -EFAULT : 0;
151 }
152 case VFIO_DEVICE_GET_REGION_INFO:
153 {
154 struct vfio_region_info info;
155
156 minsz = offsetofend(struct vfio_region_info, offset);
157
158 if (copy_from_user(&info, (void __user *)arg, minsz))
159 return -EFAULT;
160
161 if (info.argsz < minsz)
162 return -EINVAL;
163
164 if (info.index >= mc_dev->obj_desc.region_count)
165 return -EINVAL;
166
167
168 info.offset = VFIO_FSL_MC_INDEX_TO_OFFSET(info.index);
169 info.size = vdev->regions[info.index].size;
170 info.flags = vdev->regions[info.index].flags;
171
172 if (copy_to_user((void __user *)arg, &info, minsz))
173 return -EFAULT;
174 return 0;
175 }
176 case VFIO_DEVICE_GET_IRQ_INFO:
177 {
178 struct vfio_irq_info info;
179
180 minsz = offsetofend(struct vfio_irq_info, count);
181 if (copy_from_user(&info, (void __user *)arg, minsz))
182 return -EFAULT;
183
184 if (info.argsz < minsz)
185 return -EINVAL;
186
187 if (info.index >= mc_dev->obj_desc.irq_count)
188 return -EINVAL;
189
190 info.flags = VFIO_IRQ_INFO_EVENTFD;
191 info.count = 1;
192
193 if (copy_to_user((void __user *)arg, &info, minsz))
194 return -EFAULT;
195 return 0;
196 }
197 case VFIO_DEVICE_SET_IRQS:
198 {
199 struct vfio_irq_set hdr;
200 u8 *data = NULL;
201 int ret = 0;
202 size_t data_size = 0;
203
204 minsz = offsetofend(struct vfio_irq_set, count);
205
206 if (copy_from_user(&hdr, (void __user *)arg, minsz))
207 return -EFAULT;
208
209 ret = vfio_set_irqs_validate_and_prepare(&hdr, mc_dev->obj_desc.irq_count,
210 mc_dev->obj_desc.irq_count, &data_size);
211 if (ret)
212 return ret;
213
214 if (data_size) {
215 data = memdup_user((void __user *)(arg + minsz),
216 data_size);
217 if (IS_ERR(data))
218 return PTR_ERR(data);
219 }
220
221 mutex_lock(&vdev->igate);
222 ret = vfio_fsl_mc_set_irqs_ioctl(vdev, hdr.flags,
223 hdr.index, hdr.start,
224 hdr.count, data);
225 mutex_unlock(&vdev->igate);
226 kfree(data);
227
228 return ret;
229 }
230 case VFIO_DEVICE_RESET:
231 {
232 return vfio_fsl_mc_reset_device(vdev);
233
234 }
235 default:
236 return -ENOTTY;
237 }
238}
239
240static ssize_t vfio_fsl_mc_read(struct vfio_device *core_vdev, char __user *buf,
241 size_t count, loff_t *ppos)
242{
243 struct vfio_fsl_mc_device *vdev =
244 container_of(core_vdev, struct vfio_fsl_mc_device, vdev);
245 unsigned int index = VFIO_FSL_MC_OFFSET_TO_INDEX(*ppos);
246 loff_t off = *ppos & VFIO_FSL_MC_OFFSET_MASK;
247 struct fsl_mc_device *mc_dev = vdev->mc_dev;
248 struct vfio_fsl_mc_region *region;
249 u64 data[8];
250 int i;
251
252 if (index >= mc_dev->obj_desc.region_count)
253 return -EINVAL;
254
255 region = &vdev->regions[index];
256
257 if (!(region->flags & VFIO_REGION_INFO_FLAG_READ))
258 return -EINVAL;
259
260 if (!region->ioaddr) {
261 region->ioaddr = ioremap(region->addr, region->size);
262 if (!region->ioaddr)
263 return -ENOMEM;
264 }
265
266 if (count != 64 || off != 0)
267 return -EINVAL;
268
269 for (i = 7; i >= 0; i--)
270 data[i] = readq(region->ioaddr + i * sizeof(uint64_t));
271
272 if (copy_to_user(buf, data, 64))
273 return -EFAULT;
274
275 return count;
276}
277
278#define MC_CMD_COMPLETION_TIMEOUT_MS 5000
279#define MC_CMD_COMPLETION_POLLING_MAX_SLEEP_USECS 500
280
281static int vfio_fsl_mc_send_command(void __iomem *ioaddr, uint64_t *cmd_data)
282{
283 int i;
284 enum mc_cmd_status status;
285 unsigned long timeout_usecs = MC_CMD_COMPLETION_TIMEOUT_MS * 1000;
286
287
288 for (i = 7; i >= 1; i--)
289 writeq_relaxed(cmd_data[i], ioaddr + i * sizeof(uint64_t));
290
291
292 writeq(cmd_data[0], ioaddr);
293
294
295
296
297
298 for (;;) {
299 u64 header;
300 struct mc_cmd_header *resp_hdr;
301
302 header = cpu_to_le64(readq_relaxed(ioaddr));
303
304 resp_hdr = (struct mc_cmd_header *)&header;
305 status = (enum mc_cmd_status)resp_hdr->status;
306 if (status != MC_CMD_STATUS_READY)
307 break;
308
309 udelay(MC_CMD_COMPLETION_POLLING_MAX_SLEEP_USECS);
310 timeout_usecs -= MC_CMD_COMPLETION_POLLING_MAX_SLEEP_USECS;
311 if (timeout_usecs == 0)
312 return -ETIMEDOUT;
313 }
314
315 return 0;
316}
317
318static ssize_t vfio_fsl_mc_write(struct vfio_device *core_vdev,
319 const char __user *buf, size_t count,
320 loff_t *ppos)
321{
322 struct vfio_fsl_mc_device *vdev =
323 container_of(core_vdev, struct vfio_fsl_mc_device, vdev);
324 unsigned int index = VFIO_FSL_MC_OFFSET_TO_INDEX(*ppos);
325 loff_t off = *ppos & VFIO_FSL_MC_OFFSET_MASK;
326 struct fsl_mc_device *mc_dev = vdev->mc_dev;
327 struct vfio_fsl_mc_region *region;
328 u64 data[8];
329 int ret;
330
331 if (index >= mc_dev->obj_desc.region_count)
332 return -EINVAL;
333
334 region = &vdev->regions[index];
335
336 if (!(region->flags & VFIO_REGION_INFO_FLAG_WRITE))
337 return -EINVAL;
338
339 if (!region->ioaddr) {
340 region->ioaddr = ioremap(region->addr, region->size);
341 if (!region->ioaddr)
342 return -ENOMEM;
343 }
344
345 if (count != 64 || off != 0)
346 return -EINVAL;
347
348 if (copy_from_user(&data, buf, 64))
349 return -EFAULT;
350
351 ret = vfio_fsl_mc_send_command(region->ioaddr, data);
352 if (ret)
353 return ret;
354
355 return count;
356
357}
358
359static int vfio_fsl_mc_mmap_mmio(struct vfio_fsl_mc_region region,
360 struct vm_area_struct *vma)
361{
362 u64 size = vma->vm_end - vma->vm_start;
363 u64 pgoff, base;
364 u8 region_cacheable;
365
366 pgoff = vma->vm_pgoff &
367 ((1U << (VFIO_FSL_MC_OFFSET_SHIFT - PAGE_SHIFT)) - 1);
368 base = pgoff << PAGE_SHIFT;
369
370 if (region.size < PAGE_SIZE || base + size > region.size)
371 return -EINVAL;
372
373 region_cacheable = (region.type & FSL_MC_REGION_CACHEABLE) &&
374 (region.type & FSL_MC_REGION_SHAREABLE);
375 if (!region_cacheable)
376 vma->vm_page_prot = pgprot_noncached(vma->vm_page_prot);
377
378 vma->vm_pgoff = (region.addr >> PAGE_SHIFT) + pgoff;
379
380 return remap_pfn_range(vma, vma->vm_start, vma->vm_pgoff,
381 size, vma->vm_page_prot);
382}
383
384static int vfio_fsl_mc_mmap(struct vfio_device *core_vdev,
385 struct vm_area_struct *vma)
386{
387 struct vfio_fsl_mc_device *vdev =
388 container_of(core_vdev, struct vfio_fsl_mc_device, vdev);
389 struct fsl_mc_device *mc_dev = vdev->mc_dev;
390 unsigned int index;
391
392 index = vma->vm_pgoff >> (VFIO_FSL_MC_OFFSET_SHIFT - PAGE_SHIFT);
393
394 if (vma->vm_end < vma->vm_start)
395 return -EINVAL;
396 if (vma->vm_start & ~PAGE_MASK)
397 return -EINVAL;
398 if (vma->vm_end & ~PAGE_MASK)
399 return -EINVAL;
400 if (!(vma->vm_flags & VM_SHARED))
401 return -EINVAL;
402 if (index >= mc_dev->obj_desc.region_count)
403 return -EINVAL;
404
405 if (!(vdev->regions[index].flags & VFIO_REGION_INFO_FLAG_MMAP))
406 return -EINVAL;
407
408 if (!(vdev->regions[index].flags & VFIO_REGION_INFO_FLAG_READ)
409 && (vma->vm_flags & VM_READ))
410 return -EINVAL;
411
412 if (!(vdev->regions[index].flags & VFIO_REGION_INFO_FLAG_WRITE)
413 && (vma->vm_flags & VM_WRITE))
414 return -EINVAL;
415
416 vma->vm_private_data = mc_dev;
417
418 return vfio_fsl_mc_mmap_mmio(vdev->regions[index], vma);
419}
420
421static const struct vfio_device_ops vfio_fsl_mc_ops = {
422 .name = "vfio-fsl-mc",
423 .open_device = vfio_fsl_mc_open_device,
424 .close_device = vfio_fsl_mc_close_device,
425 .ioctl = vfio_fsl_mc_ioctl,
426 .read = vfio_fsl_mc_read,
427 .write = vfio_fsl_mc_write,
428 .mmap = vfio_fsl_mc_mmap,
429};
430
431static int vfio_fsl_mc_bus_notifier(struct notifier_block *nb,
432 unsigned long action, void *data)
433{
434 struct vfio_fsl_mc_device *vdev = container_of(nb,
435 struct vfio_fsl_mc_device, nb);
436 struct device *dev = data;
437 struct fsl_mc_device *mc_dev = to_fsl_mc_device(dev);
438 struct fsl_mc_device *mc_cont = to_fsl_mc_device(mc_dev->dev.parent);
439
440 if (action == BUS_NOTIFY_ADD_DEVICE &&
441 vdev->mc_dev == mc_cont) {
442 mc_dev->driver_override = kasprintf(GFP_KERNEL, "%s",
443 vfio_fsl_mc_ops.name);
444 if (!mc_dev->driver_override)
445 dev_warn(dev, "VFIO_FSL_MC: Setting driver override for device in dprc %s failed\n",
446 dev_name(&mc_cont->dev));
447 else
448 dev_info(dev, "VFIO_FSL_MC: Setting driver override for device in dprc %s\n",
449 dev_name(&mc_cont->dev));
450 } else if (action == BUS_NOTIFY_BOUND_DRIVER &&
451 vdev->mc_dev == mc_cont) {
452 struct fsl_mc_driver *mc_drv = to_fsl_mc_driver(dev->driver);
453
454 if (mc_drv && mc_drv != &vfio_fsl_mc_driver)
455 dev_warn(dev, "VFIO_FSL_MC: Object %s bound to driver %s while DPRC bound to vfio-fsl-mc\n",
456 dev_name(dev), mc_drv->driver.name);
457 }
458
459 return 0;
460}
461
462static int vfio_fsl_mc_init_device(struct vfio_fsl_mc_device *vdev)
463{
464 struct fsl_mc_device *mc_dev = vdev->mc_dev;
465 int ret;
466
467
468 if (!is_fsl_mc_bus_dprc(mc_dev)) {
469 struct fsl_mc_device *mc_cont = to_fsl_mc_device(mc_dev->dev.parent);
470
471 mc_dev->mc_io = mc_cont->mc_io;
472 return 0;
473 }
474
475 vdev->nb.notifier_call = vfio_fsl_mc_bus_notifier;
476 ret = bus_register_notifier(&fsl_mc_bus_type, &vdev->nb);
477 if (ret)
478 return ret;
479
480
481 ret = dprc_setup(mc_dev);
482 if (ret) {
483 dev_err(&mc_dev->dev, "VFIO_FSL_MC: Failed to setup DPRC (%d)\n", ret);
484 goto out_nc_unreg;
485 }
486 return 0;
487
488out_nc_unreg:
489 bus_unregister_notifier(&fsl_mc_bus_type, &vdev->nb);
490 return ret;
491}
492
493static int vfio_fsl_mc_scan_container(struct fsl_mc_device *mc_dev)
494{
495 int ret;
496
497
498 if (!is_fsl_mc_bus_dprc(mc_dev))
499 return 0;
500 ret = dprc_scan_container(mc_dev, false);
501 if (ret) {
502 dev_err(&mc_dev->dev,
503 "VFIO_FSL_MC: Container scanning failed (%d)\n", ret);
504 dprc_remove_devices(mc_dev, NULL, 0);
505 return ret;
506 }
507 return 0;
508}
509
510static void vfio_fsl_uninit_device(struct vfio_fsl_mc_device *vdev)
511{
512 struct fsl_mc_device *mc_dev = vdev->mc_dev;
513
514 if (!is_fsl_mc_bus_dprc(mc_dev))
515 return;
516
517 dprc_cleanup(mc_dev);
518 bus_unregister_notifier(&fsl_mc_bus_type, &vdev->nb);
519}
520
521static int vfio_fsl_mc_probe(struct fsl_mc_device *mc_dev)
522{
523 struct vfio_fsl_mc_device *vdev;
524 struct device *dev = &mc_dev->dev;
525 int ret;
526
527 vdev = kzalloc(sizeof(*vdev), GFP_KERNEL);
528 if (!vdev)
529 return -ENOMEM;
530
531 vfio_init_group_dev(&vdev->vdev, dev, &vfio_fsl_mc_ops);
532 vdev->mc_dev = mc_dev;
533 mutex_init(&vdev->igate);
534
535 if (is_fsl_mc_bus_dprc(mc_dev))
536 ret = vfio_assign_device_set(&vdev->vdev, &mc_dev->dev);
537 else
538 ret = vfio_assign_device_set(&vdev->vdev, mc_dev->dev.parent);
539 if (ret)
540 goto out_uninit;
541
542 ret = vfio_fsl_mc_init_device(vdev);
543 if (ret)
544 goto out_uninit;
545
546 ret = vfio_register_group_dev(&vdev->vdev);
547 if (ret) {
548 dev_err(dev, "VFIO_FSL_MC: Failed to add to vfio group\n");
549 goto out_device;
550 }
551
552 ret = vfio_fsl_mc_scan_container(mc_dev);
553 if (ret)
554 goto out_group_dev;
555 dev_set_drvdata(dev, vdev);
556 return 0;
557
558out_group_dev:
559 vfio_unregister_group_dev(&vdev->vdev);
560out_device:
561 vfio_fsl_uninit_device(vdev);
562out_uninit:
563 vfio_uninit_group_dev(&vdev->vdev);
564 kfree(vdev);
565 return ret;
566}
567
568static int vfio_fsl_mc_remove(struct fsl_mc_device *mc_dev)
569{
570 struct device *dev = &mc_dev->dev;
571 struct vfio_fsl_mc_device *vdev = dev_get_drvdata(dev);
572
573 vfio_unregister_group_dev(&vdev->vdev);
574 mutex_destroy(&vdev->igate);
575
576 dprc_remove_devices(mc_dev, NULL, 0);
577 vfio_fsl_uninit_device(vdev);
578
579 vfio_uninit_group_dev(&vdev->vdev);
580 kfree(vdev);
581 return 0;
582}
583
584static struct fsl_mc_driver vfio_fsl_mc_driver = {
585 .probe = vfio_fsl_mc_probe,
586 .remove = vfio_fsl_mc_remove,
587 .driver = {
588 .name = "vfio-fsl-mc",
589 .owner = THIS_MODULE,
590 },
591};
592
593static int __init vfio_fsl_mc_driver_init(void)
594{
595 return fsl_mc_driver_register(&vfio_fsl_mc_driver);
596}
597
598static void __exit vfio_fsl_mc_driver_exit(void)
599{
600 fsl_mc_driver_unregister(&vfio_fsl_mc_driver);
601}
602
603module_init(vfio_fsl_mc_driver_init);
604module_exit(vfio_fsl_mc_driver_exit);
605
606MODULE_LICENSE("Dual BSD/GPL");
607MODULE_DESCRIPTION("VFIO for FSL-MC devices - User Level meta-driver");
608
