assigned-dev.c 19 KB

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