Linux Audio

Check our new training course

Loading...
v3.5.6
  1/*
  2 * drivers/pci/iov.c
  3 *
  4 * Copyright (C) 2009 Intel Corporation, Yu Zhao <yu.zhao@intel.com>
  5 *
  6 * PCI Express I/O Virtualization (IOV) support.
  7 *   Single Root IOV 1.0
  8 *   Address Translation Service 1.0
  9 */
 10
 11#include <linux/pci.h>
 12#include <linux/slab.h>
 13#include <linux/mutex.h>
 14#include <linux/export.h>
 15#include <linux/string.h>
 16#include <linux/delay.h>
 17#include <linux/pci-ats.h>
 18#include "pci.h"
 19
 20#define VIRTFN_ID_LEN	16
 21
 22static inline u8 virtfn_bus(struct pci_dev *dev, int id)
 23{
 24	return dev->bus->number + ((dev->devfn + dev->sriov->offset +
 25				    dev->sriov->stride * id) >> 8);
 26}
 27
 28static inline u8 virtfn_devfn(struct pci_dev *dev, int id)
 29{
 30	return (dev->devfn + dev->sriov->offset +
 31		dev->sriov->stride * id) & 0xff;
 32}
 33
 34static struct pci_bus *virtfn_add_bus(struct pci_bus *bus, int busnr)
 35{
 36	int rc;
 37	struct pci_bus *child;
 38
 39	if (bus->number == busnr)
 40		return bus;
 41
 42	child = pci_find_bus(pci_domain_nr(bus), busnr);
 43	if (child)
 44		return child;
 45
 46	child = pci_add_new_bus(bus, NULL, busnr);
 47	if (!child)
 48		return NULL;
 49
 50	child->subordinate = busnr;
 51	child->dev.parent = bus->bridge;
 52	rc = pci_bus_add_child(child);
 53	if (rc) {
 54		pci_remove_bus(child);
 55		return NULL;
 56	}
 57
 58	return child;
 59}
 60
 61static void virtfn_remove_bus(struct pci_bus *bus, int busnr)
 62{
 63	struct pci_bus *child;
 64
 65	if (bus->number == busnr)
 66		return;
 67
 68	child = pci_find_bus(pci_domain_nr(bus), busnr);
 69	BUG_ON(!child);
 70
 71	if (list_empty(&child->devices))
 72		pci_remove_bus(child);
 73}
 74
 75static int virtfn_add(struct pci_dev *dev, int id, int reset)
 76{
 77	int i;
 78	int rc;
 79	u64 size;
 80	char buf[VIRTFN_ID_LEN];
 81	struct pci_dev *virtfn;
 82	struct resource *res;
 83	struct pci_sriov *iov = dev->sriov;
 
 
 
 
 
 
 84
 85	virtfn = alloc_pci_dev();
 86	if (!virtfn)
 87		return -ENOMEM;
 88
 89	mutex_lock(&iov->dev->sriov->lock);
 90	virtfn->bus = virtfn_add_bus(dev->bus, virtfn_bus(dev, id));
 91	if (!virtfn->bus) {
 92		kfree(virtfn);
 93		mutex_unlock(&iov->dev->sriov->lock);
 94		return -ENOMEM;
 95	}
 96	virtfn->devfn = virtfn_devfn(dev, id);
 97	virtfn->vendor = dev->vendor;
 98	pci_read_config_word(dev, iov->pos + PCI_SRIOV_VF_DID, &virtfn->device);
 99	pci_setup_device(virtfn);
100	virtfn->dev.parent = dev->dev.parent;
 
 
 
101
102	for (i = 0; i < PCI_SRIOV_NUM_BARS; i++) {
103		res = dev->resource + PCI_IOV_RESOURCES + i;
104		if (!res->parent)
105			continue;
106		virtfn->resource[i].name = pci_name(virtfn);
107		virtfn->resource[i].flags = res->flags;
108		size = resource_size(res);
109		do_div(size, iov->total);
110		virtfn->resource[i].start = res->start + size * id;
111		virtfn->resource[i].end = virtfn->resource[i].start + size - 1;
112		rc = request_resource(res, &virtfn->resource[i]);
113		BUG_ON(rc);
114	}
115
116	if (reset)
117		__pci_reset_function(virtfn);
118
119	pci_device_add(virtfn, virtfn->bus);
120	mutex_unlock(&iov->dev->sriov->lock);
121
122	virtfn->physfn = pci_dev_get(dev);
123	virtfn->is_virtfn = 1;
124
125	rc = pci_bus_add_device(virtfn);
126	if (rc)
127		goto failed1;
128	sprintf(buf, "virtfn%u", id);
129	rc = sysfs_create_link(&dev->dev.kobj, &virtfn->dev.kobj, buf);
130	if (rc)
131		goto failed1;
132	rc = sysfs_create_link(&virtfn->dev.kobj, &dev->dev.kobj, "physfn");
133	if (rc)
134		goto failed2;
135
136	kobject_uevent(&virtfn->dev.kobj, KOBJ_CHANGE);
137
138	return 0;
139
140failed2:
141	sysfs_remove_link(&dev->dev.kobj, buf);
142failed1:
143	pci_dev_put(dev);
144	mutex_lock(&iov->dev->sriov->lock);
145	pci_stop_and_remove_bus_device(virtfn);
146	virtfn_remove_bus(dev->bus, virtfn_bus(dev, id));
 
 
147	mutex_unlock(&iov->dev->sriov->lock);
148
149	return rc;
150}
151
152static void virtfn_remove(struct pci_dev *dev, int id, int reset)
153{
154	char buf[VIRTFN_ID_LEN];
155	struct pci_bus *bus;
156	struct pci_dev *virtfn;
157	struct pci_sriov *iov = dev->sriov;
158
159	bus = pci_find_bus(pci_domain_nr(dev->bus), virtfn_bus(dev, id));
160	if (!bus)
161		return;
162
163	virtfn = pci_get_slot(bus, virtfn_devfn(dev, id));
164	if (!virtfn)
165		return;
166
167	pci_dev_put(virtfn);
168
169	if (reset) {
170		device_release_driver(&virtfn->dev);
171		__pci_reset_function(virtfn);
172	}
173
174	sprintf(buf, "virtfn%u", id);
175	sysfs_remove_link(&dev->dev.kobj, buf);
176	/*
177	 * pci_stop_dev() could have been called for this virtfn already,
178	 * so the directory for the virtfn may have been removed before.
179	 * Double check to avoid spurious sysfs warnings.
180	 */
181	if (virtfn->dev.kobj.sd)
182		sysfs_remove_link(&virtfn->dev.kobj, "physfn");
183
184	mutex_lock(&iov->dev->sriov->lock);
185	pci_stop_and_remove_bus_device(virtfn);
186	virtfn_remove_bus(dev->bus, virtfn_bus(dev, id));
187	mutex_unlock(&iov->dev->sriov->lock);
188
 
 
189	pci_dev_put(dev);
190}
191
192static int sriov_migration(struct pci_dev *dev)
193{
194	u16 status;
195	struct pci_sriov *iov = dev->sriov;
196
197	if (!iov->nr_virtfn)
198		return 0;
199
200	if (!(iov->cap & PCI_SRIOV_CAP_VFM))
201		return 0;
202
203	pci_read_config_word(dev, iov->pos + PCI_SRIOV_STATUS, &status);
204	if (!(status & PCI_SRIOV_STATUS_VFM))
205		return 0;
206
207	schedule_work(&iov->mtask);
208
209	return 1;
210}
211
212static void sriov_migration_task(struct work_struct *work)
213{
214	int i;
215	u8 state;
216	u16 status;
217	struct pci_sriov *iov = container_of(work, struct pci_sriov, mtask);
218
219	for (i = iov->initial; i < iov->nr_virtfn; i++) {
220		state = readb(iov->mstate + i);
221		if (state == PCI_SRIOV_VFM_MI) {
222			writeb(PCI_SRIOV_VFM_AV, iov->mstate + i);
223			state = readb(iov->mstate + i);
224			if (state == PCI_SRIOV_VFM_AV)
225				virtfn_add(iov->self, i, 1);
226		} else if (state == PCI_SRIOV_VFM_MO) {
227			virtfn_remove(iov->self, i, 1);
228			writeb(PCI_SRIOV_VFM_UA, iov->mstate + i);
229			state = readb(iov->mstate + i);
230			if (state == PCI_SRIOV_VFM_AV)
231				virtfn_add(iov->self, i, 0);
232		}
233	}
234
235	pci_read_config_word(iov->self, iov->pos + PCI_SRIOV_STATUS, &status);
236	status &= ~PCI_SRIOV_STATUS_VFM;
237	pci_write_config_word(iov->self, iov->pos + PCI_SRIOV_STATUS, status);
238}
239
240static int sriov_enable_migration(struct pci_dev *dev, int nr_virtfn)
241{
242	int bir;
243	u32 table;
244	resource_size_t pa;
245	struct pci_sriov *iov = dev->sriov;
246
247	if (nr_virtfn <= iov->initial)
248		return 0;
249
250	pci_read_config_dword(dev, iov->pos + PCI_SRIOV_VFM, &table);
251	bir = PCI_SRIOV_VFM_BIR(table);
252	if (bir > PCI_STD_RESOURCE_END)
253		return -EIO;
254
255	table = PCI_SRIOV_VFM_OFFSET(table);
256	if (table + nr_virtfn > pci_resource_len(dev, bir))
257		return -EIO;
258
259	pa = pci_resource_start(dev, bir) + table;
260	iov->mstate = ioremap(pa, nr_virtfn);
261	if (!iov->mstate)
262		return -ENOMEM;
263
264	INIT_WORK(&iov->mtask, sriov_migration_task);
265
266	iov->ctrl |= PCI_SRIOV_CTRL_VFM | PCI_SRIOV_CTRL_INTR;
267	pci_write_config_word(dev, iov->pos + PCI_SRIOV_CTRL, iov->ctrl);
268
269	return 0;
270}
271
272static void sriov_disable_migration(struct pci_dev *dev)
273{
274	struct pci_sriov *iov = dev->sriov;
275
276	iov->ctrl &= ~(PCI_SRIOV_CTRL_VFM | PCI_SRIOV_CTRL_INTR);
277	pci_write_config_word(dev, iov->pos + PCI_SRIOV_CTRL, iov->ctrl);
278
279	cancel_work_sync(&iov->mtask);
280	iounmap(iov->mstate);
281}
282
283static int sriov_enable(struct pci_dev *dev, int nr_virtfn)
284{
285	int rc;
286	int i, j;
287	int nres;
288	u16 offset, stride, initial;
289	struct resource *res;
290	struct pci_dev *pdev;
291	struct pci_sriov *iov = dev->sriov;
292	int bars = 0;
293
294	if (!nr_virtfn)
295		return 0;
296
297	if (iov->nr_virtfn)
298		return -EINVAL;
299
300	pci_read_config_word(dev, iov->pos + PCI_SRIOV_INITIAL_VF, &initial);
301	if (initial > iov->total ||
302	    (!(iov->cap & PCI_SRIOV_CAP_VFM) && (initial != iov->total)))
303		return -EIO;
304
305	if (nr_virtfn < 0 || nr_virtfn > iov->total ||
306	    (!(iov->cap & PCI_SRIOV_CAP_VFM) && (nr_virtfn > initial)))
307		return -EINVAL;
308
309	pci_write_config_word(dev, iov->pos + PCI_SRIOV_NUM_VF, nr_virtfn);
310	pci_read_config_word(dev, iov->pos + PCI_SRIOV_VF_OFFSET, &offset);
311	pci_read_config_word(dev, iov->pos + PCI_SRIOV_VF_STRIDE, &stride);
312	if (!offset || (nr_virtfn > 1 && !stride))
313		return -EIO;
314
315	nres = 0;
316	for (i = 0; i < PCI_SRIOV_NUM_BARS; i++) {
317		bars |= (1 << (i + PCI_IOV_RESOURCES));
318		res = dev->resource + PCI_IOV_RESOURCES + i;
319		if (res->parent)
320			nres++;
321	}
322	if (nres != iov->nres) {
323		dev_err(&dev->dev, "not enough MMIO resources for SR-IOV\n");
324		return -ENOMEM;
325	}
326
327	iov->offset = offset;
328	iov->stride = stride;
329
330	if (virtfn_bus(dev, nr_virtfn - 1) > dev->bus->subordinate) {
331		dev_err(&dev->dev, "SR-IOV: bus number out of range\n");
332		return -ENOMEM;
333	}
334
335	if (pci_enable_resources(dev, bars)) {
336		dev_err(&dev->dev, "SR-IOV: IOV BARS not allocated\n");
337		return -ENOMEM;
338	}
339
340	if (iov->link != dev->devfn) {
341		pdev = pci_get_slot(dev->bus, iov->link);
342		if (!pdev)
343			return -ENODEV;
344
345		pci_dev_put(pdev);
346
347		if (!pdev->is_physfn)
348			return -ENODEV;
349
350		rc = sysfs_create_link(&dev->dev.kobj,
351					&pdev->dev.kobj, "dep_link");
 
352		if (rc)
353			return rc;
354	}
355
 
356	iov->ctrl |= PCI_SRIOV_CTRL_VFE | PCI_SRIOV_CTRL_MSE;
357	pci_cfg_access_lock(dev);
358	pci_write_config_word(dev, iov->pos + PCI_SRIOV_CTRL, iov->ctrl);
359	msleep(100);
360	pci_cfg_access_unlock(dev);
361
362	iov->initial = initial;
363	if (nr_virtfn < initial)
364		initial = nr_virtfn;
365
366	for (i = 0; i < initial; i++) {
367		rc = virtfn_add(dev, i, 0);
368		if (rc)
369			goto failed;
370	}
371
372	if (iov->cap & PCI_SRIOV_CAP_VFM) {
373		rc = sriov_enable_migration(dev, nr_virtfn);
374		if (rc)
375			goto failed;
376	}
377
378	kobject_uevent(&dev->dev.kobj, KOBJ_CHANGE);
379	iov->nr_virtfn = nr_virtfn;
380
381	return 0;
382
383failed:
384	for (j = 0; j < i; j++)
385		virtfn_remove(dev, j, 0);
386
387	iov->ctrl &= ~(PCI_SRIOV_CTRL_VFE | PCI_SRIOV_CTRL_MSE);
388	pci_cfg_access_lock(dev);
389	pci_write_config_word(dev, iov->pos + PCI_SRIOV_CTRL, iov->ctrl);
 
390	ssleep(1);
391	pci_cfg_access_unlock(dev);
392
393	if (iov->link != dev->devfn)
394		sysfs_remove_link(&dev->dev.kobj, "dep_link");
395
396	return rc;
397}
398
399static void sriov_disable(struct pci_dev *dev)
400{
401	int i;
402	struct pci_sriov *iov = dev->sriov;
403
404	if (!iov->nr_virtfn)
405		return;
406
407	if (iov->cap & PCI_SRIOV_CAP_VFM)
408		sriov_disable_migration(dev);
409
410	for (i = 0; i < iov->nr_virtfn; i++)
411		virtfn_remove(dev, i, 0);
412
413	iov->ctrl &= ~(PCI_SRIOV_CTRL_VFE | PCI_SRIOV_CTRL_MSE);
414	pci_cfg_access_lock(dev);
415	pci_write_config_word(dev, iov->pos + PCI_SRIOV_CTRL, iov->ctrl);
416	ssleep(1);
417	pci_cfg_access_unlock(dev);
418
419	if (iov->link != dev->devfn)
420		sysfs_remove_link(&dev->dev.kobj, "dep_link");
421
422	iov->nr_virtfn = 0;
 
423}
424
425static int sriov_init(struct pci_dev *dev, int pos)
426{
427	int i;
428	int rc;
429	int nres;
430	u32 pgsz;
431	u16 ctrl, total, offset, stride;
432	struct pci_sriov *iov;
433	struct resource *res;
434	struct pci_dev *pdev;
435
436	if (dev->pcie_type != PCI_EXP_TYPE_RC_END &&
437	    dev->pcie_type != PCI_EXP_TYPE_ENDPOINT)
438		return -ENODEV;
439
440	pci_read_config_word(dev, pos + PCI_SRIOV_CTRL, &ctrl);
441	if (ctrl & PCI_SRIOV_CTRL_VFE) {
442		pci_write_config_word(dev, pos + PCI_SRIOV_CTRL, 0);
443		ssleep(1);
444	}
445
446	pci_read_config_word(dev, pos + PCI_SRIOV_TOTAL_VF, &total);
447	if (!total)
448		return 0;
449
450	ctrl = 0;
451	list_for_each_entry(pdev, &dev->bus->devices, bus_list)
452		if (pdev->is_physfn)
453			goto found;
454
455	pdev = NULL;
456	if (pci_ari_enabled(dev->bus))
457		ctrl |= PCI_SRIOV_CTRL_ARI;
458
459found:
460	pci_write_config_word(dev, pos + PCI_SRIOV_CTRL, ctrl);
 
461	pci_read_config_word(dev, pos + PCI_SRIOV_VF_OFFSET, &offset);
462	pci_read_config_word(dev, pos + PCI_SRIOV_VF_STRIDE, &stride);
463	if (!offset || (total > 1 && !stride))
464		return -EIO;
465
466	pci_read_config_dword(dev, pos + PCI_SRIOV_SUP_PGSIZE, &pgsz);
467	i = PAGE_SHIFT > 12 ? PAGE_SHIFT - 12 : 0;
468	pgsz &= ~((1 << i) - 1);
469	if (!pgsz)
470		return -EIO;
471
472	pgsz &= ~(pgsz - 1);
473	pci_write_config_dword(dev, pos + PCI_SRIOV_SYS_PGSIZE, pgsz);
474
475	nres = 0;
476	for (i = 0; i < PCI_SRIOV_NUM_BARS; i++) {
477		res = dev->resource + PCI_IOV_RESOURCES + i;
478		i += __pci_read_base(dev, pci_bar_unknown, res,
479				     pos + PCI_SRIOV_BAR + i * 4);
480		if (!res->flags)
481			continue;
482		if (resource_size(res) & (PAGE_SIZE - 1)) {
483			rc = -EIO;
484			goto failed;
485		}
486		res->end = res->start + resource_size(res) * total - 1;
487		nres++;
488	}
489
490	iov = kzalloc(sizeof(*iov), GFP_KERNEL);
491	if (!iov) {
492		rc = -ENOMEM;
493		goto failed;
494	}
495
496	iov->pos = pos;
497	iov->nres = nres;
498	iov->ctrl = ctrl;
499	iov->total = total;
500	iov->offset = offset;
501	iov->stride = stride;
502	iov->pgsz = pgsz;
503	iov->self = dev;
504	pci_read_config_dword(dev, pos + PCI_SRIOV_CAP, &iov->cap);
505	pci_read_config_byte(dev, pos + PCI_SRIOV_FUNC_LINK, &iov->link);
506	if (dev->pcie_type == PCI_EXP_TYPE_RC_END)
507		iov->link = PCI_DEVFN(PCI_SLOT(dev->devfn), iov->link);
508
509	if (pdev)
510		iov->dev = pci_dev_get(pdev);
511	else
512		iov->dev = dev;
513
514	mutex_init(&iov->lock);
515
516	dev->sriov = iov;
517	dev->is_physfn = 1;
518
519	return 0;
520
521failed:
522	for (i = 0; i < PCI_SRIOV_NUM_BARS; i++) {
523		res = dev->resource + PCI_IOV_RESOURCES + i;
524		res->flags = 0;
525	}
526
527	return rc;
528}
529
530static void sriov_release(struct pci_dev *dev)
531{
532	BUG_ON(dev->sriov->nr_virtfn);
533
534	if (dev != dev->sriov->dev)
535		pci_dev_put(dev->sriov->dev);
536
537	mutex_destroy(&dev->sriov->lock);
538
539	kfree(dev->sriov);
540	dev->sriov = NULL;
541}
542
543static void sriov_restore_state(struct pci_dev *dev)
544{
545	int i;
546	u16 ctrl;
547	struct pci_sriov *iov = dev->sriov;
548
549	pci_read_config_word(dev, iov->pos + PCI_SRIOV_CTRL, &ctrl);
550	if (ctrl & PCI_SRIOV_CTRL_VFE)
551		return;
552
553	for (i = PCI_IOV_RESOURCES; i <= PCI_IOV_RESOURCE_END; i++)
554		pci_update_resource(dev, i);
555
556	pci_write_config_dword(dev, iov->pos + PCI_SRIOV_SYS_PGSIZE, iov->pgsz);
557	pci_write_config_word(dev, iov->pos + PCI_SRIOV_NUM_VF, iov->nr_virtfn);
558	pci_write_config_word(dev, iov->pos + PCI_SRIOV_CTRL, iov->ctrl);
559	if (iov->ctrl & PCI_SRIOV_CTRL_VFE)
560		msleep(100);
561}
562
563/**
564 * pci_iov_init - initialize the IOV capability
565 * @dev: the PCI device
566 *
567 * Returns 0 on success, or negative on failure.
568 */
569int pci_iov_init(struct pci_dev *dev)
570{
571	int pos;
572
573	if (!pci_is_pcie(dev))
574		return -ENODEV;
575
576	pos = pci_find_ext_capability(dev, PCI_EXT_CAP_ID_SRIOV);
577	if (pos)
578		return sriov_init(dev, pos);
579
580	return -ENODEV;
581}
582
583/**
584 * pci_iov_release - release resources used by the IOV capability
585 * @dev: the PCI device
586 */
587void pci_iov_release(struct pci_dev *dev)
588{
589	if (dev->is_physfn)
590		sriov_release(dev);
591}
592
593/**
594 * pci_iov_resource_bar - get position of the SR-IOV BAR
595 * @dev: the PCI device
596 * @resno: the resource number
597 * @type: the BAR type to be filled in
598 *
599 * Returns position of the BAR encapsulated in the SR-IOV capability.
600 */
601int pci_iov_resource_bar(struct pci_dev *dev, int resno,
602			 enum pci_bar_type *type)
603{
604	if (resno < PCI_IOV_RESOURCES || resno > PCI_IOV_RESOURCE_END)
605		return 0;
606
607	BUG_ON(!dev->is_physfn);
608
609	*type = pci_bar_unknown;
610
611	return dev->sriov->pos + PCI_SRIOV_BAR +
612		4 * (resno - PCI_IOV_RESOURCES);
613}
614
615/**
616 * pci_sriov_resource_alignment - get resource alignment for VF BAR
617 * @dev: the PCI device
618 * @resno: the resource number
619 *
620 * Returns the alignment of the VF BAR found in the SR-IOV capability.
621 * This is not the same as the resource size which is defined as
622 * the VF BAR size multiplied by the number of VFs.  The alignment
623 * is just the VF BAR size.
624 */
625resource_size_t pci_sriov_resource_alignment(struct pci_dev *dev, int resno)
626{
627	struct resource tmp;
628	enum pci_bar_type type;
629	int reg = pci_iov_resource_bar(dev, resno, &type);
630	
631	if (!reg)
632		return 0;
633
634	 __pci_read_base(dev, type, &tmp, reg);
635	return resource_alignment(&tmp);
636}
637
638/**
639 * pci_restore_iov_state - restore the state of the IOV capability
640 * @dev: the PCI device
641 */
642void pci_restore_iov_state(struct pci_dev *dev)
643{
644	if (dev->is_physfn)
645		sriov_restore_state(dev);
646}
647
648/**
649 * pci_iov_bus_range - find bus range used by Virtual Function
650 * @bus: the PCI bus
651 *
652 * Returns max number of buses (exclude current one) used by Virtual
653 * Functions.
654 */
655int pci_iov_bus_range(struct pci_bus *bus)
656{
657	int max = 0;
658	u8 busnr;
659	struct pci_dev *dev;
660
661	list_for_each_entry(dev, &bus->devices, bus_list) {
662		if (!dev->is_physfn)
663			continue;
664		busnr = virtfn_bus(dev, dev->sriov->total - 1);
665		if (busnr > max)
666			max = busnr;
667	}
668
669	return max ? max - bus->number : 0;
670}
671
672/**
673 * pci_enable_sriov - enable the SR-IOV capability
674 * @dev: the PCI device
675 * @nr_virtfn: number of virtual functions to enable
676 *
677 * Returns 0 on success, or negative on failure.
678 */
679int pci_enable_sriov(struct pci_dev *dev, int nr_virtfn)
680{
681	might_sleep();
682
683	if (!dev->is_physfn)
684		return -ENODEV;
685
686	return sriov_enable(dev, nr_virtfn);
687}
688EXPORT_SYMBOL_GPL(pci_enable_sriov);
689
690/**
691 * pci_disable_sriov - disable the SR-IOV capability
692 * @dev: the PCI device
693 */
694void pci_disable_sriov(struct pci_dev *dev)
695{
696	might_sleep();
697
698	if (!dev->is_physfn)
699		return;
700
701	sriov_disable(dev);
702}
703EXPORT_SYMBOL_GPL(pci_disable_sriov);
704
705/**
706 * pci_sriov_migration - notify SR-IOV core of Virtual Function Migration
707 * @dev: the PCI device
708 *
709 * Returns IRQ_HANDLED if the IRQ is handled, or IRQ_NONE if not.
710 *
711 * Physical Function driver is responsible to register IRQ handler using
712 * VF Migration Interrupt Message Number, and call this function when the
713 * interrupt is generated by the hardware.
714 */
715irqreturn_t pci_sriov_migration(struct pci_dev *dev)
716{
717	if (!dev->is_physfn)
718		return IRQ_NONE;
719
720	return sriov_migration(dev) ? IRQ_HANDLED : IRQ_NONE;
721}
722EXPORT_SYMBOL_GPL(pci_sriov_migration);
723
724/**
725 * pci_num_vf - return number of VFs associated with a PF device_release_driver
726 * @dev: the PCI device
727 *
728 * Returns number of VFs, or 0 if SR-IOV is not enabled.
 
729 */
730int pci_num_vf(struct pci_dev *dev)
731{
732	if (!dev || !dev->is_physfn)
 
 
 
 
 
733		return 0;
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
734	else
735		return dev->sriov->nr_virtfn;
 
 
736}
737EXPORT_SYMBOL_GPL(pci_num_vf);
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
v3.15
  1/*
  2 * drivers/pci/iov.c
  3 *
  4 * Copyright (C) 2009 Intel Corporation, Yu Zhao <yu.zhao@intel.com>
  5 *
  6 * PCI Express I/O Virtualization (IOV) support.
  7 *   Single Root IOV 1.0
  8 *   Address Translation Service 1.0
  9 */
 10
 11#include <linux/pci.h>
 12#include <linux/slab.h>
 13#include <linux/mutex.h>
 14#include <linux/export.h>
 15#include <linux/string.h>
 16#include <linux/delay.h>
 17#include <linux/pci-ats.h>
 18#include "pci.h"
 19
 20#define VIRTFN_ID_LEN	16
 21
 22static inline u8 virtfn_bus(struct pci_dev *dev, int id)
 23{
 24	return dev->bus->number + ((dev->devfn + dev->sriov->offset +
 25				    dev->sriov->stride * id) >> 8);
 26}
 27
 28static inline u8 virtfn_devfn(struct pci_dev *dev, int id)
 29{
 30	return (dev->devfn + dev->sriov->offset +
 31		dev->sriov->stride * id) & 0xff;
 32}
 33
 34static struct pci_bus *virtfn_add_bus(struct pci_bus *bus, int busnr)
 35{
 
 36	struct pci_bus *child;
 37
 38	if (bus->number == busnr)
 39		return bus;
 40
 41	child = pci_find_bus(pci_domain_nr(bus), busnr);
 42	if (child)
 43		return child;
 44
 45	child = pci_add_new_bus(bus, NULL, busnr);
 46	if (!child)
 47		return NULL;
 48
 49	pci_bus_insert_busn_res(child, busnr, busnr);
 
 
 
 
 
 
 50
 51	return child;
 52}
 53
 54static void virtfn_remove_bus(struct pci_bus *physbus, struct pci_bus *virtbus)
 55{
 56	if (physbus != virtbus && list_empty(&virtbus->devices))
 57		pci_remove_bus(virtbus);
 
 
 
 
 
 
 
 
 58}
 59
 60static int virtfn_add(struct pci_dev *dev, int id, int reset)
 61{
 62	int i;
 63	int rc = -ENOMEM;
 64	u64 size;
 65	char buf[VIRTFN_ID_LEN];
 66	struct pci_dev *virtfn;
 67	struct resource *res;
 68	struct pci_sriov *iov = dev->sriov;
 69	struct pci_bus *bus;
 70
 71	mutex_lock(&iov->dev->sriov->lock);
 72	bus = virtfn_add_bus(dev->bus, virtfn_bus(dev, id));
 73	if (!bus)
 74		goto failed;
 75
 76	virtfn = pci_alloc_dev(bus);
 77	if (!virtfn)
 78		goto failed0;
 79
 
 
 
 
 
 
 
 80	virtfn->devfn = virtfn_devfn(dev, id);
 81	virtfn->vendor = dev->vendor;
 82	pci_read_config_word(dev, iov->pos + PCI_SRIOV_VF_DID, &virtfn->device);
 83	pci_setup_device(virtfn);
 84	virtfn->dev.parent = dev->dev.parent;
 85	virtfn->physfn = pci_dev_get(dev);
 86	virtfn->is_virtfn = 1;
 87	virtfn->multifunction = 0;
 88
 89	for (i = 0; i < PCI_SRIOV_NUM_BARS; i++) {
 90		res = dev->resource + PCI_IOV_RESOURCES + i;
 91		if (!res->parent)
 92			continue;
 93		virtfn->resource[i].name = pci_name(virtfn);
 94		virtfn->resource[i].flags = res->flags;
 95		size = resource_size(res);
 96		do_div(size, iov->total_VFs);
 97		virtfn->resource[i].start = res->start + size * id;
 98		virtfn->resource[i].end = virtfn->resource[i].start + size - 1;
 99		rc = request_resource(res, &virtfn->resource[i]);
100		BUG_ON(rc);
101	}
102
103	if (reset)
104		__pci_reset_function(virtfn);
105
106	pci_device_add(virtfn, virtfn->bus);
107	mutex_unlock(&iov->dev->sriov->lock);
108
 
 
 
109	rc = pci_bus_add_device(virtfn);
 
 
110	sprintf(buf, "virtfn%u", id);
111	rc = sysfs_create_link(&dev->dev.kobj, &virtfn->dev.kobj, buf);
112	if (rc)
113		goto failed1;
114	rc = sysfs_create_link(&virtfn->dev.kobj, &dev->dev.kobj, "physfn");
115	if (rc)
116		goto failed2;
117
118	kobject_uevent(&virtfn->dev.kobj, KOBJ_CHANGE);
119
120	return 0;
121
122failed2:
123	sysfs_remove_link(&dev->dev.kobj, buf);
124failed1:
125	pci_dev_put(dev);
126	mutex_lock(&iov->dev->sriov->lock);
127	pci_stop_and_remove_bus_device(virtfn);
128failed0:
129	virtfn_remove_bus(dev->bus, bus);
130failed:
131	mutex_unlock(&iov->dev->sriov->lock);
132
133	return rc;
134}
135
136static void virtfn_remove(struct pci_dev *dev, int id, int reset)
137{
138	char buf[VIRTFN_ID_LEN];
 
139	struct pci_dev *virtfn;
140	struct pci_sriov *iov = dev->sriov;
141
142	virtfn = pci_get_domain_bus_and_slot(pci_domain_nr(dev->bus),
143					     virtfn_bus(dev, id),
144					     virtfn_devfn(dev, id));
 
 
145	if (!virtfn)
146		return;
147
 
 
148	if (reset) {
149		device_release_driver(&virtfn->dev);
150		__pci_reset_function(virtfn);
151	}
152
153	sprintf(buf, "virtfn%u", id);
154	sysfs_remove_link(&dev->dev.kobj, buf);
155	/*
156	 * pci_stop_dev() could have been called for this virtfn already,
157	 * so the directory for the virtfn may have been removed before.
158	 * Double check to avoid spurious sysfs warnings.
159	 */
160	if (virtfn->dev.kobj.sd)
161		sysfs_remove_link(&virtfn->dev.kobj, "physfn");
162
163	mutex_lock(&iov->dev->sriov->lock);
164	pci_stop_and_remove_bus_device(virtfn);
165	virtfn_remove_bus(dev->bus, virtfn->bus);
166	mutex_unlock(&iov->dev->sriov->lock);
167
168	/* balance pci_get_domain_bus_and_slot() */
169	pci_dev_put(virtfn);
170	pci_dev_put(dev);
171}
172
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
173static int sriov_enable(struct pci_dev *dev, int nr_virtfn)
174{
175	int rc;
176	int i, j;
177	int nres;
178	u16 offset, stride, initial;
179	struct resource *res;
180	struct pci_dev *pdev;
181	struct pci_sriov *iov = dev->sriov;
182	int bars = 0;
183
184	if (!nr_virtfn)
185		return 0;
186
187	if (iov->num_VFs)
188		return -EINVAL;
189
190	pci_read_config_word(dev, iov->pos + PCI_SRIOV_INITIAL_VF, &initial);
191	if (initial > iov->total_VFs ||
192	    (!(iov->cap & PCI_SRIOV_CAP_VFM) && (initial != iov->total_VFs)))
193		return -EIO;
194
195	if (nr_virtfn < 0 || nr_virtfn > iov->total_VFs ||
196	    (!(iov->cap & PCI_SRIOV_CAP_VFM) && (nr_virtfn > initial)))
197		return -EINVAL;
198
 
199	pci_read_config_word(dev, iov->pos + PCI_SRIOV_VF_OFFSET, &offset);
200	pci_read_config_word(dev, iov->pos + PCI_SRIOV_VF_STRIDE, &stride);
201	if (!offset || (nr_virtfn > 1 && !stride))
202		return -EIO;
203
204	nres = 0;
205	for (i = 0; i < PCI_SRIOV_NUM_BARS; i++) {
206		bars |= (1 << (i + PCI_IOV_RESOURCES));
207		res = dev->resource + PCI_IOV_RESOURCES + i;
208		if (res->parent)
209			nres++;
210	}
211	if (nres != iov->nres) {
212		dev_err(&dev->dev, "not enough MMIO resources for SR-IOV\n");
213		return -ENOMEM;
214	}
215
216	iov->offset = offset;
217	iov->stride = stride;
218
219	if (virtfn_bus(dev, nr_virtfn - 1) > dev->bus->busn_res.end) {
220		dev_err(&dev->dev, "SR-IOV: bus number out of range\n");
221		return -ENOMEM;
222	}
223
224	if (pci_enable_resources(dev, bars)) {
225		dev_err(&dev->dev, "SR-IOV: IOV BARS not allocated\n");
226		return -ENOMEM;
227	}
228
229	if (iov->link != dev->devfn) {
230		pdev = pci_get_slot(dev->bus, iov->link);
231		if (!pdev)
232			return -ENODEV;
233
234		if (!pdev->is_physfn) {
235			pci_dev_put(pdev);
236			return -ENOSYS;
237		}
238
239		rc = sysfs_create_link(&dev->dev.kobj,
240					&pdev->dev.kobj, "dep_link");
241		pci_dev_put(pdev);
242		if (rc)
243			return rc;
244	}
245
246	pci_write_config_word(dev, iov->pos + PCI_SRIOV_NUM_VF, nr_virtfn);
247	iov->ctrl |= PCI_SRIOV_CTRL_VFE | PCI_SRIOV_CTRL_MSE;
248	pci_cfg_access_lock(dev);
249	pci_write_config_word(dev, iov->pos + PCI_SRIOV_CTRL, iov->ctrl);
250	msleep(100);
251	pci_cfg_access_unlock(dev);
252
253	iov->initial_VFs = initial;
254	if (nr_virtfn < initial)
255		initial = nr_virtfn;
256
257	for (i = 0; i < initial; i++) {
258		rc = virtfn_add(dev, i, 0);
259		if (rc)
260			goto failed;
261	}
262
 
 
 
 
 
 
263	kobject_uevent(&dev->dev.kobj, KOBJ_CHANGE);
264	iov->num_VFs = nr_virtfn;
265
266	return 0;
267
268failed:
269	for (j = 0; j < i; j++)
270		virtfn_remove(dev, j, 0);
271
272	iov->ctrl &= ~(PCI_SRIOV_CTRL_VFE | PCI_SRIOV_CTRL_MSE);
273	pci_cfg_access_lock(dev);
274	pci_write_config_word(dev, iov->pos + PCI_SRIOV_CTRL, iov->ctrl);
275	pci_write_config_word(dev, iov->pos + PCI_SRIOV_NUM_VF, 0);
276	ssleep(1);
277	pci_cfg_access_unlock(dev);
278
279	if (iov->link != dev->devfn)
280		sysfs_remove_link(&dev->dev.kobj, "dep_link");
281
282	return rc;
283}
284
285static void sriov_disable(struct pci_dev *dev)
286{
287	int i;
288	struct pci_sriov *iov = dev->sriov;
289
290	if (!iov->num_VFs)
291		return;
292
293	for (i = 0; i < iov->num_VFs; i++)
 
 
 
294		virtfn_remove(dev, i, 0);
295
296	iov->ctrl &= ~(PCI_SRIOV_CTRL_VFE | PCI_SRIOV_CTRL_MSE);
297	pci_cfg_access_lock(dev);
298	pci_write_config_word(dev, iov->pos + PCI_SRIOV_CTRL, iov->ctrl);
299	ssleep(1);
300	pci_cfg_access_unlock(dev);
301
302	if (iov->link != dev->devfn)
303		sysfs_remove_link(&dev->dev.kobj, "dep_link");
304
305	iov->num_VFs = 0;
306	pci_write_config_word(dev, iov->pos + PCI_SRIOV_NUM_VF, 0);
307}
308
309static int sriov_init(struct pci_dev *dev, int pos)
310{
311	int i;
312	int rc;
313	int nres;
314	u32 pgsz;
315	u16 ctrl, total, offset, stride;
316	struct pci_sriov *iov;
317	struct resource *res;
318	struct pci_dev *pdev;
319
320	if (pci_pcie_type(dev) != PCI_EXP_TYPE_RC_END &&
321	    pci_pcie_type(dev) != PCI_EXP_TYPE_ENDPOINT)
322		return -ENODEV;
323
324	pci_read_config_word(dev, pos + PCI_SRIOV_CTRL, &ctrl);
325	if (ctrl & PCI_SRIOV_CTRL_VFE) {
326		pci_write_config_word(dev, pos + PCI_SRIOV_CTRL, 0);
327		ssleep(1);
328	}
329
330	pci_read_config_word(dev, pos + PCI_SRIOV_TOTAL_VF, &total);
331	if (!total)
332		return 0;
333
334	ctrl = 0;
335	list_for_each_entry(pdev, &dev->bus->devices, bus_list)
336		if (pdev->is_physfn)
337			goto found;
338
339	pdev = NULL;
340	if (pci_ari_enabled(dev->bus))
341		ctrl |= PCI_SRIOV_CTRL_ARI;
342
343found:
344	pci_write_config_word(dev, pos + PCI_SRIOV_CTRL, ctrl);
345	pci_write_config_word(dev, pos + PCI_SRIOV_NUM_VF, 0);
346	pci_read_config_word(dev, pos + PCI_SRIOV_VF_OFFSET, &offset);
347	pci_read_config_word(dev, pos + PCI_SRIOV_VF_STRIDE, &stride);
348	if (!offset || (total > 1 && !stride))
349		return -EIO;
350
351	pci_read_config_dword(dev, pos + PCI_SRIOV_SUP_PGSIZE, &pgsz);
352	i = PAGE_SHIFT > 12 ? PAGE_SHIFT - 12 : 0;
353	pgsz &= ~((1 << i) - 1);
354	if (!pgsz)
355		return -EIO;
356
357	pgsz &= ~(pgsz - 1);
358	pci_write_config_dword(dev, pos + PCI_SRIOV_SYS_PGSIZE, pgsz);
359
360	nres = 0;
361	for (i = 0; i < PCI_SRIOV_NUM_BARS; i++) {
362		res = dev->resource + PCI_IOV_RESOURCES + i;
363		i += __pci_read_base(dev, pci_bar_unknown, res,
364				     pos + PCI_SRIOV_BAR + i * 4);
365		if (!res->flags)
366			continue;
367		if (resource_size(res) & (PAGE_SIZE - 1)) {
368			rc = -EIO;
369			goto failed;
370		}
371		res->end = res->start + resource_size(res) * total - 1;
372		nres++;
373	}
374
375	iov = kzalloc(sizeof(*iov), GFP_KERNEL);
376	if (!iov) {
377		rc = -ENOMEM;
378		goto failed;
379	}
380
381	iov->pos = pos;
382	iov->nres = nres;
383	iov->ctrl = ctrl;
384	iov->total_VFs = total;
385	iov->offset = offset;
386	iov->stride = stride;
387	iov->pgsz = pgsz;
388	iov->self = dev;
389	pci_read_config_dword(dev, pos + PCI_SRIOV_CAP, &iov->cap);
390	pci_read_config_byte(dev, pos + PCI_SRIOV_FUNC_LINK, &iov->link);
391	if (pci_pcie_type(dev) == PCI_EXP_TYPE_RC_END)
392		iov->link = PCI_DEVFN(PCI_SLOT(dev->devfn), iov->link);
393
394	if (pdev)
395		iov->dev = pci_dev_get(pdev);
396	else
397		iov->dev = dev;
398
399	mutex_init(&iov->lock);
400
401	dev->sriov = iov;
402	dev->is_physfn = 1;
403
404	return 0;
405
406failed:
407	for (i = 0; i < PCI_SRIOV_NUM_BARS; i++) {
408		res = dev->resource + PCI_IOV_RESOURCES + i;
409		res->flags = 0;
410	}
411
412	return rc;
413}
414
415static void sriov_release(struct pci_dev *dev)
416{
417	BUG_ON(dev->sriov->num_VFs);
418
419	if (dev != dev->sriov->dev)
420		pci_dev_put(dev->sriov->dev);
421
422	mutex_destroy(&dev->sriov->lock);
423
424	kfree(dev->sriov);
425	dev->sriov = NULL;
426}
427
428static void sriov_restore_state(struct pci_dev *dev)
429{
430	int i;
431	u16 ctrl;
432	struct pci_sriov *iov = dev->sriov;
433
434	pci_read_config_word(dev, iov->pos + PCI_SRIOV_CTRL, &ctrl);
435	if (ctrl & PCI_SRIOV_CTRL_VFE)
436		return;
437
438	for (i = PCI_IOV_RESOURCES; i <= PCI_IOV_RESOURCE_END; i++)
439		pci_update_resource(dev, i);
440
441	pci_write_config_dword(dev, iov->pos + PCI_SRIOV_SYS_PGSIZE, iov->pgsz);
442	pci_write_config_word(dev, iov->pos + PCI_SRIOV_NUM_VF, iov->num_VFs);
443	pci_write_config_word(dev, iov->pos + PCI_SRIOV_CTRL, iov->ctrl);
444	if (iov->ctrl & PCI_SRIOV_CTRL_VFE)
445		msleep(100);
446}
447
448/**
449 * pci_iov_init - initialize the IOV capability
450 * @dev: the PCI device
451 *
452 * Returns 0 on success, or negative on failure.
453 */
454int pci_iov_init(struct pci_dev *dev)
455{
456	int pos;
457
458	if (!pci_is_pcie(dev))
459		return -ENODEV;
460
461	pos = pci_find_ext_capability(dev, PCI_EXT_CAP_ID_SRIOV);
462	if (pos)
463		return sriov_init(dev, pos);
464
465	return -ENODEV;
466}
467
468/**
469 * pci_iov_release - release resources used by the IOV capability
470 * @dev: the PCI device
471 */
472void pci_iov_release(struct pci_dev *dev)
473{
474	if (dev->is_physfn)
475		sriov_release(dev);
476}
477
478/**
479 * pci_iov_resource_bar - get position of the SR-IOV BAR
480 * @dev: the PCI device
481 * @resno: the resource number
482 * @type: the BAR type to be filled in
483 *
484 * Returns position of the BAR encapsulated in the SR-IOV capability.
485 */
486int pci_iov_resource_bar(struct pci_dev *dev, int resno,
487			 enum pci_bar_type *type)
488{
489	if (resno < PCI_IOV_RESOURCES || resno > PCI_IOV_RESOURCE_END)
490		return 0;
491
492	BUG_ON(!dev->is_physfn);
493
494	*type = pci_bar_unknown;
495
496	return dev->sriov->pos + PCI_SRIOV_BAR +
497		4 * (resno - PCI_IOV_RESOURCES);
498}
499
500/**
501 * pci_sriov_resource_alignment - get resource alignment for VF BAR
502 * @dev: the PCI device
503 * @resno: the resource number
504 *
505 * Returns the alignment of the VF BAR found in the SR-IOV capability.
506 * This is not the same as the resource size which is defined as
507 * the VF BAR size multiplied by the number of VFs.  The alignment
508 * is just the VF BAR size.
509 */
510resource_size_t pci_sriov_resource_alignment(struct pci_dev *dev, int resno)
511{
512	struct resource tmp;
513	enum pci_bar_type type;
514	int reg = pci_iov_resource_bar(dev, resno, &type);
515
516	if (!reg)
517		return 0;
518
519	 __pci_read_base(dev, type, &tmp, reg);
520	return resource_alignment(&tmp);
521}
522
523/**
524 * pci_restore_iov_state - restore the state of the IOV capability
525 * @dev: the PCI device
526 */
527void pci_restore_iov_state(struct pci_dev *dev)
528{
529	if (dev->is_physfn)
530		sriov_restore_state(dev);
531}
532
533/**
534 * pci_iov_bus_range - find bus range used by Virtual Function
535 * @bus: the PCI bus
536 *
537 * Returns max number of buses (exclude current one) used by Virtual
538 * Functions.
539 */
540int pci_iov_bus_range(struct pci_bus *bus)
541{
542	int max = 0;
543	u8 busnr;
544	struct pci_dev *dev;
545
546	list_for_each_entry(dev, &bus->devices, bus_list) {
547		if (!dev->is_physfn)
548			continue;
549		busnr = virtfn_bus(dev, dev->sriov->total_VFs - 1);
550		if (busnr > max)
551			max = busnr;
552	}
553
554	return max ? max - bus->number : 0;
555}
556
557/**
558 * pci_enable_sriov - enable the SR-IOV capability
559 * @dev: the PCI device
560 * @nr_virtfn: number of virtual functions to enable
561 *
562 * Returns 0 on success, or negative on failure.
563 */
564int pci_enable_sriov(struct pci_dev *dev, int nr_virtfn)
565{
566	might_sleep();
567
568	if (!dev->is_physfn)
569		return -ENOSYS;
570
571	return sriov_enable(dev, nr_virtfn);
572}
573EXPORT_SYMBOL_GPL(pci_enable_sriov);
574
575/**
576 * pci_disable_sriov - disable the SR-IOV capability
577 * @dev: the PCI device
578 */
579void pci_disable_sriov(struct pci_dev *dev)
580{
581	might_sleep();
582
583	if (!dev->is_physfn)
584		return;
585
586	sriov_disable(dev);
587}
588EXPORT_SYMBOL_GPL(pci_disable_sriov);
589
590/**
591 * pci_num_vf - return number of VFs associated with a PF device_release_driver
592 * @dev: the PCI device
593 *
594 * Returns number of VFs, or 0 if SR-IOV is not enabled.
 
 
 
 
595 */
596int pci_num_vf(struct pci_dev *dev)
597{
598	if (!dev->is_physfn)
599		return 0;
600
601	return dev->sriov->num_VFs;
602}
603EXPORT_SYMBOL_GPL(pci_num_vf);
604
605/**
606 * pci_vfs_assigned - returns number of VFs are assigned to a guest
607 * @dev: the PCI device
608 *
609 * Returns number of VFs belonging to this device that are assigned to a guest.
610 * If device is not a physical function returns 0.
611 */
612int pci_vfs_assigned(struct pci_dev *dev)
613{
614	struct pci_dev *vfdev;
615	unsigned int vfs_assigned = 0;
616	unsigned short dev_id;
617
618	/* only search if we are a PF */
619	if (!dev->is_physfn)
620		return 0;
621
622	/*
623	 * determine the device ID for the VFs, the vendor ID will be the
624	 * same as the PF so there is no need to check for that one
625	 */
626	pci_read_config_word(dev, dev->sriov->pos + PCI_SRIOV_VF_DID, &dev_id);
627
628	/* loop through all the VFs to see if we own any that are assigned */
629	vfdev = pci_get_device(dev->vendor, dev_id, NULL);
630	while (vfdev) {
631		/*
632		 * It is considered assigned if it is a virtual function with
633		 * our dev as the physical function and the assigned bit is set
634		 */
635		if (vfdev->is_virtfn && (vfdev->physfn == dev) &&
636		    (vfdev->dev_flags & PCI_DEV_FLAGS_ASSIGNED))
637			vfs_assigned++;
638
639		vfdev = pci_get_device(dev->vendor, dev_id, vfdev);
640	}
641
642	return vfs_assigned;
643}
644EXPORT_SYMBOL_GPL(pci_vfs_assigned);
645
646/**
647 * pci_sriov_set_totalvfs -- reduce the TotalVFs available
648 * @dev: the PCI PF device
649 * @numvfs: number that should be used for TotalVFs supported
650 *
651 * Should be called from PF driver's probe routine with
652 * device's mutex held.
653 *
654 * Returns 0 if PF is an SRIOV-capable device and
655 * value of numvfs valid. If not a PF return -ENOSYS;
656 * if numvfs is invalid return -EINVAL;
657 * if VFs already enabled, return -EBUSY.
658 */
659int pci_sriov_set_totalvfs(struct pci_dev *dev, u16 numvfs)
660{
661	if (!dev->is_physfn)
662		return -ENOSYS;
663	if (numvfs > dev->sriov->total_VFs)
664		return -EINVAL;
665
666	/* Shouldn't change if VFs already enabled */
667	if (dev->sriov->ctrl & PCI_SRIOV_CTRL_VFE)
668		return -EBUSY;
669	else
670		dev->sriov->driver_max_VFs = numvfs;
671
672	return 0;
673}
674EXPORT_SYMBOL_GPL(pci_sriov_set_totalvfs);
675
676/**
677 * pci_sriov_get_totalvfs -- get total VFs supported on this device
678 * @dev: the PCI PF device
679 *
680 * For a PCIe device with SRIOV support, return the PCIe
681 * SRIOV capability value of TotalVFs or the value of driver_max_VFs
682 * if the driver reduced it.  Otherwise 0.
683 */
684int pci_sriov_get_totalvfs(struct pci_dev *dev)
685{
686	if (!dev->is_physfn)
687		return 0;
688
689	if (dev->sriov->driver_max_VFs)
690		return dev->sriov->driver_max_VFs;
691
692	return dev->sriov->total_VFs;
693}
694EXPORT_SYMBOL_GPL(pci_sriov_get_totalvfs);