assigned-dev.c 19 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801
  1. /*
  2. * Kernel-based Virtual Machine - device assignment support
  3. *
  4. * Copyright (C) 2010 Red Hat, Inc. and/or its affiliates.
  5. *
  6. * This work is licensed under the terms of the GNU GPL, version 2. See
  7. * the COPYING file in the top-level directory.
  8. *
  9. */
  10. #include <linux/kvm_host.h>
  11. #include <linux/kvm.h>
  12. #include <linux/uaccess.h>
  13. #include <linux/vmalloc.h>
  14. #include <linux/errno.h>
  15. #include <linux/spinlock.h>
  16. #include <linux/pci.h>
  17. #include <linux/interrupt.h>
  18. #include <linux/slab.h>
  19. #include "irq.h"
  20. static struct kvm_assigned_dev_kernel *kvm_find_assigned_dev(struct list_head *head,
  21. int assigned_dev_id)
  22. {
  23. struct list_head *ptr;
  24. struct kvm_assigned_dev_kernel *match;
  25. list_for_each(ptr, head) {
  26. match = list_entry(ptr, struct kvm_assigned_dev_kernel, list);
  27. if (match->assigned_dev_id == assigned_dev_id)
  28. return match;
  29. }
  30. return NULL;
  31. }
  32. static int find_index_from_host_irq(struct kvm_assigned_dev_kernel
  33. *assigned_dev, int irq)
  34. {
  35. int i, index;
  36. struct msix_entry *host_msix_entries;
  37. host_msix_entries = assigned_dev->host_msix_entries;
  38. index = -1;
  39. for (i = 0; i < assigned_dev->entries_nr; i++)
  40. if (irq == host_msix_entries[i].vector) {
  41. index = i;
  42. break;
  43. }
  44. if (index < 0) {
  45. printk(KERN_WARNING "Fail to find correlated MSI-X entry!\n");
  46. return 0;
  47. }
  48. return index;
  49. }
  50. static irqreturn_t kvm_assigned_dev_thread(int irq, void *dev_id)
  51. {
  52. struct kvm_assigned_dev_kernel *assigned_dev = dev_id;
  53. if (assigned_dev->irq_requested_type & KVM_DEV_IRQ_HOST_INTX) {
  54. spin_lock(&assigned_dev->intx_lock);
  55. disable_irq_nosync(irq);
  56. assigned_dev->host_irq_disabled = true;
  57. spin_unlock(&assigned_dev->intx_lock);
  58. }
  59. kvm_set_irq(assigned_dev->kvm, assigned_dev->irq_source_id,
  60. assigned_dev->guest_irq, 1);
  61. return IRQ_HANDLED;
  62. }
  63. #ifdef __KVM_HAVE_MSIX
  64. static irqreturn_t kvm_assigned_dev_thread_msix(int irq, void *dev_id)
  65. {
  66. struct kvm_assigned_dev_kernel *assigned_dev = dev_id;
  67. int index = find_index_from_host_irq(assigned_dev, irq);
  68. u32 vector;
  69. if (index >= 0) {
  70. vector = assigned_dev->guest_msix_entries[index].vector;
  71. kvm_set_irq(assigned_dev->kvm, assigned_dev->irq_source_id,
  72. vector, 1);
  73. }
  74. return IRQ_HANDLED;
  75. }
  76. #endif
  77. /* Ack the irq line for an assigned device */
  78. static void kvm_assigned_dev_ack_irq(struct kvm_irq_ack_notifier *kian)
  79. {
  80. struct kvm_assigned_dev_kernel *dev =
  81. container_of(kian, struct kvm_assigned_dev_kernel,
  82. ack_notifier);
  83. kvm_set_irq(dev->kvm, dev->irq_source_id, dev->guest_irq, 0);
  84. /* The guest irq may be shared so this ack may be
  85. * from another device.
  86. */
  87. spin_lock(&dev->intx_lock);
  88. if (dev->host_irq_disabled) {
  89. enable_irq(dev->host_irq);
  90. dev->host_irq_disabled = false;
  91. }
  92. spin_unlock(&dev->intx_lock);
  93. }
  94. static void deassign_guest_irq(struct kvm *kvm,
  95. struct kvm_assigned_dev_kernel *assigned_dev)
  96. {
  97. if (assigned_dev->ack_notifier.gsi != -1)
  98. kvm_unregister_irq_ack_notifier(kvm,
  99. &assigned_dev->ack_notifier);
  100. kvm_set_irq(assigned_dev->kvm, assigned_dev->irq_source_id,
  101. assigned_dev->guest_irq, 0);
  102. if (assigned_dev->irq_source_id != -1)
  103. kvm_free_irq_source_id(kvm, assigned_dev->irq_source_id);
  104. assigned_dev->irq_source_id = -1;
  105. assigned_dev->irq_requested_type &= ~(KVM_DEV_IRQ_GUEST_MASK);
  106. }
  107. /* The function implicit hold kvm->lock mutex due to cancel_work_sync() */
  108. static void deassign_host_irq(struct kvm *kvm,
  109. struct kvm_assigned_dev_kernel *assigned_dev)
  110. {
  111. /*
  112. * We disable irq here to prevent further events.
  113. *
  114. * Notice this maybe result in nested disable if the interrupt type is
  115. * INTx, but it's OK for we are going to free it.
  116. *
  117. * If this function is a part of VM destroy, please ensure that till
  118. * now, the kvm state is still legal for probably we also have to wait
  119. * on a currently running IRQ handler.
  120. */
  121. if (assigned_dev->irq_requested_type & KVM_DEV_IRQ_HOST_MSIX) {
  122. int i;
  123. for (i = 0; i < assigned_dev->entries_nr; i++)
  124. disable_irq(assigned_dev->host_msix_entries[i].vector);
  125. for (i = 0; i < assigned_dev->entries_nr; i++)
  126. free_irq(assigned_dev->host_msix_entries[i].vector,
  127. assigned_dev);
  128. assigned_dev->entries_nr = 0;
  129. kfree(assigned_dev->host_msix_entries);
  130. kfree(assigned_dev->guest_msix_entries);
  131. pci_disable_msix(assigned_dev->dev);
  132. } else {
  133. /* Deal with MSI and INTx */
  134. disable_irq(assigned_dev->host_irq);
  135. free_irq(assigned_dev->host_irq, assigned_dev);
  136. if (assigned_dev->irq_requested_type & KVM_DEV_IRQ_HOST_MSI)
  137. pci_disable_msi(assigned_dev->dev);
  138. }
  139. assigned_dev->irq_requested_type &= ~(KVM_DEV_IRQ_HOST_MASK);
  140. }
  141. static int kvm_deassign_irq(struct kvm *kvm,
  142. struct kvm_assigned_dev_kernel *assigned_dev,
  143. unsigned long irq_requested_type)
  144. {
  145. unsigned long guest_irq_type, host_irq_type;
  146. if (!irqchip_in_kernel(kvm))
  147. return -EINVAL;
  148. /* no irq assignment to deassign */
  149. if (!assigned_dev->irq_requested_type)
  150. return -ENXIO;
  151. host_irq_type = irq_requested_type & KVM_DEV_IRQ_HOST_MASK;
  152. guest_irq_type = irq_requested_type & KVM_DEV_IRQ_GUEST_MASK;
  153. if (host_irq_type)
  154. deassign_host_irq(kvm, assigned_dev);
  155. if (guest_irq_type)
  156. deassign_guest_irq(kvm, assigned_dev);
  157. return 0;
  158. }
  159. static void kvm_free_assigned_irq(struct kvm *kvm,
  160. struct kvm_assigned_dev_kernel *assigned_dev)
  161. {
  162. kvm_deassign_irq(kvm, assigned_dev, assigned_dev->irq_requested_type);
  163. }
  164. static void kvm_free_assigned_device(struct kvm *kvm,
  165. struct kvm_assigned_dev_kernel
  166. *assigned_dev)
  167. {
  168. kvm_free_assigned_irq(kvm, assigned_dev);
  169. pci_reset_function(assigned_dev->dev);
  170. if (pci_load_and_free_saved_state(assigned_dev->dev,
  171. &assigned_dev->pci_saved_state))
  172. printk(KERN_INFO "%s: Couldn't reload %s saved state\n",
  173. __func__, dev_name(&assigned_dev->dev->dev));
  174. else
  175. pci_restore_state(assigned_dev->dev);
  176. pci_release_regions(assigned_dev->dev);
  177. pci_disable_device(assigned_dev->dev);
  178. pci_dev_put(assigned_dev->dev);
  179. list_del(&assigned_dev->list);
  180. kfree(assigned_dev);
  181. }
  182. void kvm_free_all_assigned_devices(struct kvm *kvm)
  183. {
  184. struct list_head *ptr, *ptr2;
  185. struct kvm_assigned_dev_kernel *assigned_dev;
  186. list_for_each_safe(ptr, ptr2, &kvm->arch.assigned_dev_head) {
  187. assigned_dev = list_entry(ptr,
  188. struct kvm_assigned_dev_kernel,
  189. list);
  190. kvm_free_assigned_device(kvm, assigned_dev);
  191. }
  192. }
  193. static int assigned_device_enable_host_intx(struct kvm *kvm,
  194. struct kvm_assigned_dev_kernel *dev)
  195. {
  196. dev->host_irq = dev->dev->irq;
  197. /* Even though this is PCI, we don't want to use shared
  198. * interrupts. Sharing host devices with guest-assigned devices
  199. * on the same interrupt line is not a happy situation: there
  200. * are going to be long delays in accepting, acking, etc.
  201. */
  202. if (request_threaded_irq(dev->host_irq, NULL, kvm_assigned_dev_thread,
  203. IRQF_ONESHOT, dev->irq_name, dev))
  204. return -EIO;
  205. return 0;
  206. }
  207. #ifdef __KVM_HAVE_MSI
  208. static int assigned_device_enable_host_msi(struct kvm *kvm,
  209. struct kvm_assigned_dev_kernel *dev)
  210. {
  211. int r;
  212. if (!dev->dev->msi_enabled) {
  213. r = pci_enable_msi(dev->dev);
  214. if (r)
  215. return r;
  216. }
  217. dev->host_irq = dev->dev->irq;
  218. if (request_threaded_irq(dev->host_irq, NULL, kvm_assigned_dev_thread,
  219. 0, dev->irq_name, dev)) {
  220. pci_disable_msi(dev->dev);
  221. return -EIO;
  222. }
  223. return 0;
  224. }
  225. #endif
  226. #ifdef __KVM_HAVE_MSIX
  227. static int assigned_device_enable_host_msix(struct kvm *kvm,
  228. struct kvm_assigned_dev_kernel *dev)
  229. {
  230. int i, r = -EINVAL;
  231. /* host_msix_entries and guest_msix_entries should have been
  232. * initialized */
  233. if (dev->entries_nr == 0)
  234. return r;
  235. r = pci_enable_msix(dev->dev, dev->host_msix_entries, dev->entries_nr);
  236. if (r)
  237. return r;
  238. for (i = 0; i < dev->entries_nr; i++) {
  239. r = request_threaded_irq(dev->host_msix_entries[i].vector,
  240. NULL, kvm_assigned_dev_thread_msix,
  241. 0, dev->irq_name, dev);
  242. if (r)
  243. goto err;
  244. }
  245. return 0;
  246. err:
  247. for (i -= 1; i >= 0; i--)
  248. free_irq(dev->host_msix_entries[i].vector, dev);
  249. pci_disable_msix(dev->dev);
  250. return r;
  251. }
  252. #endif
  253. static int assigned_device_enable_guest_intx(struct kvm *kvm,
  254. struct kvm_assigned_dev_kernel *dev,
  255. struct kvm_assigned_irq *irq)
  256. {
  257. dev->guest_irq = irq->guest_irq;
  258. dev->ack_notifier.gsi = irq->guest_irq;
  259. return 0;
  260. }
  261. #ifdef __KVM_HAVE_MSI
  262. static int assigned_device_enable_guest_msi(struct kvm *kvm,
  263. struct kvm_assigned_dev_kernel *dev,
  264. struct kvm_assigned_irq *irq)
  265. {
  266. dev->guest_irq = irq->guest_irq;
  267. dev->ack_notifier.gsi = -1;
  268. dev->host_irq_disabled = false;
  269. return 0;
  270. }
  271. #endif
  272. #ifdef __KVM_HAVE_MSIX
  273. static int assigned_device_enable_guest_msix(struct kvm *kvm,
  274. struct kvm_assigned_dev_kernel *dev,
  275. struct kvm_assigned_irq *irq)
  276. {
  277. dev->guest_irq = irq->guest_irq;
  278. dev->ack_notifier.gsi = -1;
  279. dev->host_irq_disabled = false;
  280. return 0;
  281. }
  282. #endif
  283. static int assign_host_irq(struct kvm *kvm,
  284. struct kvm_assigned_dev_kernel *dev,
  285. __u32 host_irq_type)
  286. {
  287. int r = -EEXIST;
  288. if (dev->irq_requested_type & KVM_DEV_IRQ_HOST_MASK)
  289. return r;
  290. snprintf(dev->irq_name, sizeof(dev->irq_name), "kvm:%s",
  291. pci_name(dev->dev));
  292. switch (host_irq_type) {
  293. case KVM_DEV_IRQ_HOST_INTX:
  294. r = assigned_device_enable_host_intx(kvm, dev);
  295. break;
  296. #ifdef __KVM_HAVE_MSI
  297. case KVM_DEV_IRQ_HOST_MSI:
  298. r = assigned_device_enable_host_msi(kvm, dev);
  299. break;
  300. #endif
  301. #ifdef __KVM_HAVE_MSIX
  302. case KVM_DEV_IRQ_HOST_MSIX:
  303. r = assigned_device_enable_host_msix(kvm, dev);
  304. break;
  305. #endif
  306. default:
  307. r = -EINVAL;
  308. }
  309. if (!r)
  310. dev->irq_requested_type |= host_irq_type;
  311. return r;
  312. }
  313. static int assign_guest_irq(struct kvm *kvm,
  314. struct kvm_assigned_dev_kernel *dev,
  315. struct kvm_assigned_irq *irq,
  316. unsigned long guest_irq_type)
  317. {
  318. int id;
  319. int r = -EEXIST;
  320. if (dev->irq_requested_type & KVM_DEV_IRQ_GUEST_MASK)
  321. return r;
  322. id = kvm_request_irq_source_id(kvm);
  323. if (id < 0)
  324. return id;
  325. dev->irq_source_id = id;
  326. switch (guest_irq_type) {
  327. case KVM_DEV_IRQ_GUEST_INTX:
  328. r = assigned_device_enable_guest_intx(kvm, dev, irq);
  329. break;
  330. #ifdef __KVM_HAVE_MSI
  331. case KVM_DEV_IRQ_GUEST_MSI:
  332. r = assigned_device_enable_guest_msi(kvm, dev, irq);
  333. break;
  334. #endif
  335. #ifdef __KVM_HAVE_MSIX
  336. case KVM_DEV_IRQ_GUEST_MSIX:
  337. r = assigned_device_enable_guest_msix(kvm, dev, irq);
  338. break;
  339. #endif
  340. default:
  341. r = -EINVAL;
  342. }
  343. if (!r) {
  344. dev->irq_requested_type |= guest_irq_type;
  345. if (dev->ack_notifier.gsi != -1)
  346. kvm_register_irq_ack_notifier(kvm, &dev->ack_notifier);
  347. } else
  348. kvm_free_irq_source_id(kvm, dev->irq_source_id);
  349. return r;
  350. }
  351. /* TODO Deal with KVM_DEV_IRQ_ASSIGNED_MASK_MSIX */
  352. static int kvm_vm_ioctl_assign_irq(struct kvm *kvm,
  353. struct kvm_assigned_irq *assigned_irq)
  354. {
  355. int r = -EINVAL;
  356. struct kvm_assigned_dev_kernel *match;
  357. unsigned long host_irq_type, guest_irq_type;
  358. if (!irqchip_in_kernel(kvm))
  359. return r;
  360. mutex_lock(&kvm->lock);
  361. r = -ENODEV;
  362. match = kvm_find_assigned_dev(&kvm->arch.assigned_dev_head,
  363. assigned_irq->assigned_dev_id);
  364. if (!match)
  365. goto out;
  366. host_irq_type = (assigned_irq->flags & KVM_DEV_IRQ_HOST_MASK);
  367. guest_irq_type = (assigned_irq->flags & KVM_DEV_IRQ_GUEST_MASK);
  368. r = -EINVAL;
  369. /* can only assign one type at a time */
  370. if (hweight_long(host_irq_type) > 1)
  371. goto out;
  372. if (hweight_long(guest_irq_type) > 1)
  373. goto out;
  374. if (host_irq_type == 0 && guest_irq_type == 0)
  375. goto out;
  376. r = 0;
  377. if (host_irq_type)
  378. r = assign_host_irq(kvm, match, host_irq_type);
  379. if (r)
  380. goto out;
  381. if (guest_irq_type)
  382. r = assign_guest_irq(kvm, match, assigned_irq, guest_irq_type);
  383. out:
  384. mutex_unlock(&kvm->lock);
  385. return r;
  386. }
  387. static int kvm_vm_ioctl_deassign_dev_irq(struct kvm *kvm,
  388. struct kvm_assigned_irq
  389. *assigned_irq)
  390. {
  391. int r = -ENODEV;
  392. struct kvm_assigned_dev_kernel *match;
  393. mutex_lock(&kvm->lock);
  394. match = kvm_find_assigned_dev(&kvm->arch.assigned_dev_head,
  395. assigned_irq->assigned_dev_id);
  396. if (!match)
  397. goto out;
  398. r = kvm_deassign_irq(kvm, match, assigned_irq->flags);
  399. out:
  400. mutex_unlock(&kvm->lock);
  401. return r;
  402. }
  403. static int kvm_vm_ioctl_assign_device(struct kvm *kvm,
  404. struct kvm_assigned_pci_dev *assigned_dev)
  405. {
  406. int r = 0, idx;
  407. struct kvm_assigned_dev_kernel *match;
  408. struct pci_dev *dev;
  409. mutex_lock(&kvm->lock);
  410. idx = srcu_read_lock(&kvm->srcu);
  411. match = kvm_find_assigned_dev(&kvm->arch.assigned_dev_head,
  412. assigned_dev->assigned_dev_id);
  413. if (match) {
  414. /* device already assigned */
  415. r = -EEXIST;
  416. goto out;
  417. }
  418. match = kzalloc(sizeof(struct kvm_assigned_dev_kernel), GFP_KERNEL);
  419. if (match == NULL) {
  420. printk(KERN_INFO "%s: Couldn't allocate memory\n",
  421. __func__);
  422. r = -ENOMEM;
  423. goto out;
  424. }
  425. dev = pci_get_domain_bus_and_slot(assigned_dev->segnr,
  426. assigned_dev->busnr,
  427. assigned_dev->devfn);
  428. if (!dev) {
  429. printk(KERN_INFO "%s: host device not found\n", __func__);
  430. r = -EINVAL;
  431. goto out_free;
  432. }
  433. if (pci_enable_device(dev)) {
  434. printk(KERN_INFO "%s: Could not enable PCI device\n", __func__);
  435. r = -EBUSY;
  436. goto out_put;
  437. }
  438. r = pci_request_regions(dev, "kvm_assigned_device");
  439. if (r) {
  440. printk(KERN_INFO "%s: Could not get access to device regions\n",
  441. __func__);
  442. goto out_disable;
  443. }
  444. pci_reset_function(dev);
  445. pci_save_state(dev);
  446. match->pci_saved_state = pci_store_saved_state(dev);
  447. if (!match->pci_saved_state)
  448. printk(KERN_DEBUG "%s: Couldn't store %s saved state\n",
  449. __func__, dev_name(&dev->dev));
  450. match->assigned_dev_id = assigned_dev->assigned_dev_id;
  451. match->host_segnr = assigned_dev->segnr;
  452. match->host_busnr = assigned_dev->busnr;
  453. match->host_devfn = assigned_dev->devfn;
  454. match->flags = assigned_dev->flags;
  455. match->dev = dev;
  456. spin_lock_init(&match->intx_lock);
  457. match->irq_source_id = -1;
  458. match->kvm = kvm;
  459. match->ack_notifier.irq_acked = kvm_assigned_dev_ack_irq;
  460. list_add(&match->list, &kvm->arch.assigned_dev_head);
  461. if (assigned_dev->flags & KVM_DEV_ASSIGN_ENABLE_IOMMU) {
  462. if (!kvm->arch.iommu_domain) {
  463. r = kvm_iommu_map_guest(kvm);
  464. if (r)
  465. goto out_list_del;
  466. }
  467. r = kvm_assign_device(kvm, match);
  468. if (r)
  469. goto out_list_del;
  470. }
  471. out:
  472. srcu_read_unlock(&kvm->srcu, idx);
  473. mutex_unlock(&kvm->lock);
  474. return r;
  475. out_list_del:
  476. if (pci_load_and_free_saved_state(dev, &match->pci_saved_state))
  477. printk(KERN_INFO "%s: Couldn't reload %s saved state\n",
  478. __func__, dev_name(&dev->dev));
  479. list_del(&match->list);
  480. pci_release_regions(dev);
  481. out_disable:
  482. pci_disable_device(dev);
  483. out_put:
  484. pci_dev_put(dev);
  485. out_free:
  486. kfree(match);
  487. srcu_read_unlock(&kvm->srcu, idx);
  488. mutex_unlock(&kvm->lock);
  489. return r;
  490. }
  491. static int kvm_vm_ioctl_deassign_device(struct kvm *kvm,
  492. struct kvm_assigned_pci_dev *assigned_dev)
  493. {
  494. int r = 0;
  495. struct kvm_assigned_dev_kernel *match;
  496. mutex_lock(&kvm->lock);
  497. match = kvm_find_assigned_dev(&kvm->arch.assigned_dev_head,
  498. assigned_dev->assigned_dev_id);
  499. if (!match) {
  500. printk(KERN_INFO "%s: device hasn't been assigned before, "
  501. "so cannot be deassigned\n", __func__);
  502. r = -EINVAL;
  503. goto out;
  504. }
  505. if (match->flags & KVM_DEV_ASSIGN_ENABLE_IOMMU)
  506. kvm_deassign_device(kvm, match);
  507. kvm_free_assigned_device(kvm, match);
  508. out:
  509. mutex_unlock(&kvm->lock);
  510. return r;
  511. }
  512. #ifdef __KVM_HAVE_MSIX
  513. static int kvm_vm_ioctl_set_msix_nr(struct kvm *kvm,
  514. struct kvm_assigned_msix_nr *entry_nr)
  515. {
  516. int r = 0;
  517. struct kvm_assigned_dev_kernel *adev;
  518. mutex_lock(&kvm->lock);
  519. adev = kvm_find_assigned_dev(&kvm->arch.assigned_dev_head,
  520. entry_nr->assigned_dev_id);
  521. if (!adev) {
  522. r = -EINVAL;
  523. goto msix_nr_out;
  524. }
  525. if (adev->entries_nr == 0) {
  526. adev->entries_nr = entry_nr->entry_nr;
  527. if (adev->entries_nr == 0 ||
  528. adev->entries_nr > KVM_MAX_MSIX_PER_DEV) {
  529. r = -EINVAL;
  530. goto msix_nr_out;
  531. }
  532. adev->host_msix_entries = kzalloc(sizeof(struct msix_entry) *
  533. entry_nr->entry_nr,
  534. GFP_KERNEL);
  535. if (!adev->host_msix_entries) {
  536. r = -ENOMEM;
  537. goto msix_nr_out;
  538. }
  539. adev->guest_msix_entries =
  540. kzalloc(sizeof(struct msix_entry) * entry_nr->entry_nr,
  541. GFP_KERNEL);
  542. if (!adev->guest_msix_entries) {
  543. kfree(adev->host_msix_entries);
  544. r = -ENOMEM;
  545. goto msix_nr_out;
  546. }
  547. } else /* Not allowed set MSI-X number twice */
  548. r = -EINVAL;
  549. msix_nr_out:
  550. mutex_unlock(&kvm->lock);
  551. return r;
  552. }
  553. static int kvm_vm_ioctl_set_msix_entry(struct kvm *kvm,
  554. struct kvm_assigned_msix_entry *entry)
  555. {
  556. int r = 0, i;
  557. struct kvm_assigned_dev_kernel *adev;
  558. mutex_lock(&kvm->lock);
  559. adev = kvm_find_assigned_dev(&kvm->arch.assigned_dev_head,
  560. entry->assigned_dev_id);
  561. if (!adev) {
  562. r = -EINVAL;
  563. goto msix_entry_out;
  564. }
  565. for (i = 0; i < adev->entries_nr; i++)
  566. if (adev->guest_msix_entries[i].vector == 0 ||
  567. adev->guest_msix_entries[i].entry == entry->entry) {
  568. adev->guest_msix_entries[i].entry = entry->entry;
  569. adev->guest_msix_entries[i].vector = entry->gsi;
  570. adev->host_msix_entries[i].entry = entry->entry;
  571. break;
  572. }
  573. if (i == adev->entries_nr) {
  574. r = -ENOSPC;
  575. goto msix_entry_out;
  576. }
  577. msix_entry_out:
  578. mutex_unlock(&kvm->lock);
  579. return r;
  580. }
  581. #endif
  582. long kvm_vm_ioctl_assigned_device(struct kvm *kvm, unsigned ioctl,
  583. unsigned long arg)
  584. {
  585. void __user *argp = (void __user *)arg;
  586. int r;
  587. switch (ioctl) {
  588. case KVM_ASSIGN_PCI_DEVICE: {
  589. struct kvm_assigned_pci_dev assigned_dev;
  590. r = -EFAULT;
  591. if (copy_from_user(&assigned_dev, argp, sizeof assigned_dev))
  592. goto out;
  593. r = kvm_vm_ioctl_assign_device(kvm, &assigned_dev);
  594. if (r)
  595. goto out;
  596. break;
  597. }
  598. case KVM_ASSIGN_IRQ: {
  599. r = -EOPNOTSUPP;
  600. break;
  601. }
  602. case KVM_ASSIGN_DEV_IRQ: {
  603. struct kvm_assigned_irq assigned_irq;
  604. r = -EFAULT;
  605. if (copy_from_user(&assigned_irq, argp, sizeof assigned_irq))
  606. goto out;
  607. r = kvm_vm_ioctl_assign_irq(kvm, &assigned_irq);
  608. if (r)
  609. goto out;
  610. break;
  611. }
  612. case KVM_DEASSIGN_DEV_IRQ: {
  613. struct kvm_assigned_irq assigned_irq;
  614. r = -EFAULT;
  615. if (copy_from_user(&assigned_irq, argp, sizeof assigned_irq))
  616. goto out;
  617. r = kvm_vm_ioctl_deassign_dev_irq(kvm, &assigned_irq);
  618. if (r)
  619. goto out;
  620. break;
  621. }
  622. case KVM_DEASSIGN_PCI_DEVICE: {
  623. struct kvm_assigned_pci_dev assigned_dev;
  624. r = -EFAULT;
  625. if (copy_from_user(&assigned_dev, argp, sizeof assigned_dev))
  626. goto out;
  627. r = kvm_vm_ioctl_deassign_device(kvm, &assigned_dev);
  628. if (r)
  629. goto out;
  630. break;
  631. }
  632. #ifdef KVM_CAP_IRQ_ROUTING
  633. case KVM_SET_GSI_ROUTING: {
  634. struct kvm_irq_routing routing;
  635. struct kvm_irq_routing __user *urouting;
  636. struct kvm_irq_routing_entry *entries;
  637. r = -EFAULT;
  638. if (copy_from_user(&routing, argp, sizeof(routing)))
  639. goto out;
  640. r = -EINVAL;
  641. if (routing.nr >= KVM_MAX_IRQ_ROUTES)
  642. goto out;
  643. if (routing.flags)
  644. goto out;
  645. r = -ENOMEM;
  646. entries = vmalloc(routing.nr * sizeof(*entries));
  647. if (!entries)
  648. goto out;
  649. r = -EFAULT;
  650. urouting = argp;
  651. if (copy_from_user(entries, urouting->entries,
  652. routing.nr * sizeof(*entries)))
  653. goto out_free_irq_routing;
  654. r = kvm_set_irq_routing(kvm, entries, routing.nr,
  655. routing.flags);
  656. out_free_irq_routing:
  657. vfree(entries);
  658. break;
  659. }
  660. #endif /* KVM_CAP_IRQ_ROUTING */
  661. #ifdef __KVM_HAVE_MSIX
  662. case KVM_ASSIGN_SET_MSIX_NR: {
  663. struct kvm_assigned_msix_nr entry_nr;
  664. r = -EFAULT;
  665. if (copy_from_user(&entry_nr, argp, sizeof entry_nr))
  666. goto out;
  667. r = kvm_vm_ioctl_set_msix_nr(kvm, &entry_nr);
  668. if (r)
  669. goto out;
  670. break;
  671. }
  672. case KVM_ASSIGN_SET_MSIX_ENTRY: {
  673. struct kvm_assigned_msix_entry entry;
  674. r = -EFAULT;
  675. if (copy_from_user(&entry, argp, sizeof entry))
  676. goto out;
  677. r = kvm_vm_ioctl_set_msix_entry(kvm, &entry);
  678. if (r)
  679. goto out;
  680. break;
  681. }
  682. #endif
  683. default:
  684. r = -ENOTTY;
  685. break;
  686. }
  687. out:
  688. return r;
  689. }