blob: cb6f24740ee3e5b3f6763f7b25c2bf8700e6a1b2 [file] [log] [blame]
Yu Zhaod1b054d2009-03-20 11:25:11 +08001/*
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
Yu Zhao302b4212009-05-18 13:51:32 +08008 * Address Translation Service 1.0
Yu Zhaod1b054d2009-03-20 11:25:11 +08009 */
10
11#include <linux/pci.h>
Tejun Heo5a0e3ad2010-03-24 17:04:11 +090012#include <linux/slab.h>
Yu Zhaod1b054d2009-03-20 11:25:11 +080013#include <linux/mutex.h>
Paul Gortmaker363c75d2011-05-27 09:37:25 -040014#include <linux/export.h>
Yu Zhaod1b054d2009-03-20 11:25:11 +080015#include <linux/string.h>
16#include <linux/delay.h>
Joerg Roedel5cdede22011-04-04 15:55:18 +020017#include <linux/pci-ats.h>
Yu Zhaod1b054d2009-03-20 11:25:11 +080018#include "pci.h"
19
Yu Zhaodd7cc442009-03-20 11:25:15 +080020#define VIRTFN_ID_LEN 16
Yu Zhaod1b054d2009-03-20 11:25:11 +080021
Yu Zhaoa28724b2009-03-20 11:25:13 +080022static 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
Yu Zhaodd7cc442009-03-20 11:25:15 +080034static struct pci_bus *virtfn_add_bus(struct pci_bus *bus, int busnr)
35{
Yu Zhaodd7cc442009-03-20 11:25:15 +080036 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
Yinghai Lub7eac052012-05-17 18:51:13 -070049 pci_bus_insert_busn_res(child, busnr, busnr);
Yu Zhaodd7cc442009-03-20 11:25:15 +080050
51 return child;
52}
53
Jiang Liudc087f22013-05-25 21:48:37 +080054static void virtfn_remove_bus(struct pci_bus *physbus, struct pci_bus *virtbus)
Yu Zhaodd7cc442009-03-20 11:25:15 +080055{
Jiang Liudc087f22013-05-25 21:48:37 +080056 if (physbus != virtbus && list_empty(&virtbus->devices))
57 pci_remove_bus(virtbus);
Yu Zhaodd7cc442009-03-20 11:25:15 +080058}
59
60static int virtfn_add(struct pci_dev *dev, int id, int reset)
61{
62 int i;
Jiang Liudc087f22013-05-25 21:48:37 +080063 int rc = -ENOMEM;
Yu Zhaodd7cc442009-03-20 11:25:15 +080064 u64 size;
65 char buf[VIRTFN_ID_LEN];
66 struct pci_dev *virtfn;
67 struct resource *res;
68 struct pci_sriov *iov = dev->sriov;
Gu Zheng8b1fce02013-05-25 21:48:31 +080069 struct pci_bus *bus;
Yu Zhaodd7cc442009-03-20 11:25:15 +080070
71 mutex_lock(&iov->dev->sriov->lock);
Gu Zheng8b1fce02013-05-25 21:48:31 +080072 bus = virtfn_add_bus(dev->bus, virtfn_bus(dev, id));
Jiang Liudc087f22013-05-25 21:48:37 +080073 if (!bus)
74 goto failed;
75
76 virtfn = pci_alloc_dev(bus);
77 if (!virtfn)
78 goto failed0;
79
Yu Zhaodd7cc442009-03-20 11:25:15 +080080 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;
Xudong Haofbf33f52013-05-31 12:21:29 +080085 virtfn->physfn = pci_dev_get(dev);
86 virtfn->is_virtfn = 1;
Alex Williamsonaa9319772014-01-09 08:36:08 -070087 virtfn->multifunction = 0;
Yu Zhaodd7cc442009-03-20 11:25:15 +080088
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);
Bjorn Helgaas6b136722012-11-09 20:27:53 -070096 do_div(size, iov->total_VFs);
Yu Zhaodd7cc442009-03-20 11:25:15 +080097 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)
Yu Zhao8c1c6992009-06-13 15:52:13 +0800104 __pci_reset_function(virtfn);
Yu Zhaodd7cc442009-03-20 11:25:15 +0800105
106 pci_device_add(virtfn, virtfn->bus);
107 mutex_unlock(&iov->dev->sriov->lock);
108
Yijing Wangc893d132014-05-30 11:01:03 +0800109 pci_bus_add_device(virtfn);
Yu Zhaodd7cc442009-03-20 11:25:15 +0800110 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);
Yinghai Lu210647a2012-02-25 13:54:20 -0800127 pci_stop_and_remove_bus_device(virtfn);
Jiang Liudc087f22013-05-25 21:48:37 +0800128failed0:
129 virtfn_remove_bus(dev->bus, bus);
130failed:
Yu Zhaodd7cc442009-03-20 11:25:15 +0800131 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];
Yu Zhaodd7cc442009-03-20 11:25:15 +0800139 struct pci_dev *virtfn;
140 struct pci_sriov *iov = dev->sriov;
141
Jiang Liudc087f22013-05-25 21:48:37 +0800142 virtfn = pci_get_domain_bus_and_slot(pci_domain_nr(dev->bus),
143 virtfn_bus(dev, id),
144 virtfn_devfn(dev, id));
Yu Zhaodd7cc442009-03-20 11:25:15 +0800145 if (!virtfn)
146 return;
147
Yu Zhaodd7cc442009-03-20 11:25:15 +0800148 if (reset) {
149 device_release_driver(&virtfn->dev);
Yu Zhao8c1c6992009-06-13 15:52:13 +0800150 __pci_reset_function(virtfn);
Yu Zhaodd7cc442009-03-20 11:25:15 +0800151 }
152
153 sprintf(buf, "virtfn%u", id);
154 sysfs_remove_link(&dev->dev.kobj, buf);
Yinghai Lu09cedbe2012-02-04 22:55:01 -0800155 /*
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");
Yu Zhaodd7cc442009-03-20 11:25:15 +0800162
163 mutex_lock(&iov->dev->sriov->lock);
Yinghai Lu210647a2012-02-25 13:54:20 -0800164 pci_stop_and_remove_bus_device(virtfn);
Jiang Liudc087f22013-05-25 21:48:37 +0800165 virtfn_remove_bus(dev->bus, virtfn->bus);
Yu Zhaodd7cc442009-03-20 11:25:15 +0800166 mutex_unlock(&iov->dev->sriov->lock);
167
Jiang Liudc087f22013-05-25 21:48:37 +0800168 /* balance pci_get_domain_bus_and_slot() */
169 pci_dev_put(virtfn);
Yu Zhaodd7cc442009-03-20 11:25:15 +0800170 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;
Ram Paibbef98a2011-11-06 10:33:10 +0800182 int bars = 0;
Yu Zhaodd7cc442009-03-20 11:25:15 +0800183
184 if (!nr_virtfn)
185 return 0;
186
Bjorn Helgaas6b136722012-11-09 20:27:53 -0700187 if (iov->num_VFs)
Yu Zhaodd7cc442009-03-20 11:25:15 +0800188 return -EINVAL;
189
190 pci_read_config_word(dev, iov->pos + PCI_SRIOV_INITIAL_VF, &initial);
Bjorn Helgaas6b136722012-11-09 20:27:53 -0700191 if (initial > iov->total_VFs ||
192 (!(iov->cap & PCI_SRIOV_CAP_VFM) && (initial != iov->total_VFs)))
Yu Zhaodd7cc442009-03-20 11:25:15 +0800193 return -EIO;
194
Bjorn Helgaas6b136722012-11-09 20:27:53 -0700195 if (nr_virtfn < 0 || nr_virtfn > iov->total_VFs ||
Yu Zhaodd7cc442009-03-20 11:25:15 +0800196 (!(iov->cap & PCI_SRIOV_CAP_VFM) && (nr_virtfn > initial)))
197 return -EINVAL;
198
Yu Zhaodd7cc442009-03-20 11:25:15 +0800199 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++) {
Ram Paibbef98a2011-11-06 10:33:10 +0800206 bars |= (1 << (i + PCI_IOV_RESOURCES));
Yu Zhaodd7cc442009-03-20 11:25:15 +0800207 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
Yinghai Lub918c622012-05-17 18:51:11 -0700219 if (virtfn_bus(dev, nr_virtfn - 1) > dev->bus->busn_res.end) {
Yu Zhaodd7cc442009-03-20 11:25:15 +0800220 dev_err(&dev->dev, "SR-IOV: bus number out of range\n");
221 return -ENOMEM;
222 }
223
Ram Paibbef98a2011-11-06 10:33:10 +0800224 if (pci_enable_resources(dev, bars)) {
225 dev_err(&dev->dev, "SR-IOV: IOV BARS not allocated\n");
226 return -ENOMEM;
227 }
228
Yu Zhaodd7cc442009-03-20 11:25:15 +0800229 if (iov->link != dev->devfn) {
230 pdev = pci_get_slot(dev->bus, iov->link);
231 if (!pdev)
232 return -ENODEV;
233
Jiang Liudc087f22013-05-25 21:48:37 +0800234 if (!pdev->is_physfn) {
235 pci_dev_put(pdev);
Stefan Assmann652d1102013-07-31 16:47:56 -0600236 return -ENOSYS;
Jiang Liudc087f22013-05-25 21:48:37 +0800237 }
Yu Zhaodd7cc442009-03-20 11:25:15 +0800238
239 rc = sysfs_create_link(&dev->dev.kobj,
240 &pdev->dev.kobj, "dep_link");
Jiang Liudc087f22013-05-25 21:48:37 +0800241 pci_dev_put(pdev);
Yu Zhaodd7cc442009-03-20 11:25:15 +0800242 if (rc)
243 return rc;
244 }
245
Yijing Wang19b69842013-07-24 17:26:12 +0800246 pci_write_config_word(dev, iov->pos + PCI_SRIOV_NUM_VF, nr_virtfn);
Yu Zhaodd7cc442009-03-20 11:25:15 +0800247 iov->ctrl |= PCI_SRIOV_CTRL_VFE | PCI_SRIOV_CTRL_MSE;
Jan Kiszkafb51ccb2011-11-04 09:45:59 +0100248 pci_cfg_access_lock(dev);
Yu Zhaodd7cc442009-03-20 11:25:15 +0800249 pci_write_config_word(dev, iov->pos + PCI_SRIOV_CTRL, iov->ctrl);
250 msleep(100);
Jan Kiszkafb51ccb2011-11-04 09:45:59 +0100251 pci_cfg_access_unlock(dev);
Yu Zhaodd7cc442009-03-20 11:25:15 +0800252
Bjorn Helgaas6b136722012-11-09 20:27:53 -0700253 iov->initial_VFs = initial;
Yu Zhaodd7cc442009-03-20 11:25:15 +0800254 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);
Bjorn Helgaas6b136722012-11-09 20:27:53 -0700264 iov->num_VFs = nr_virtfn;
Yu Zhaodd7cc442009-03-20 11:25:15 +0800265
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);
Jan Kiszkafb51ccb2011-11-04 09:45:59 +0100273 pci_cfg_access_lock(dev);
Yu Zhaodd7cc442009-03-20 11:25:15 +0800274 pci_write_config_word(dev, iov->pos + PCI_SRIOV_CTRL, iov->ctrl);
Yijing Wang19b69842013-07-24 17:26:12 +0800275 pci_write_config_word(dev, iov->pos + PCI_SRIOV_NUM_VF, 0);
Yu Zhaodd7cc442009-03-20 11:25:15 +0800276 ssleep(1);
Jan Kiszkafb51ccb2011-11-04 09:45:59 +0100277 pci_cfg_access_unlock(dev);
Yu Zhaodd7cc442009-03-20 11:25:15 +0800278
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
Bjorn Helgaas6b136722012-11-09 20:27:53 -0700290 if (!iov->num_VFs)
Yu Zhaodd7cc442009-03-20 11:25:15 +0800291 return;
292
Bjorn Helgaas6b136722012-11-09 20:27:53 -0700293 for (i = 0; i < iov->num_VFs; i++)
Yu Zhaodd7cc442009-03-20 11:25:15 +0800294 virtfn_remove(dev, i, 0);
295
296 iov->ctrl &= ~(PCI_SRIOV_CTRL_VFE | PCI_SRIOV_CTRL_MSE);
Jan Kiszkafb51ccb2011-11-04 09:45:59 +0100297 pci_cfg_access_lock(dev);
Yu Zhaodd7cc442009-03-20 11:25:15 +0800298 pci_write_config_word(dev, iov->pos + PCI_SRIOV_CTRL, iov->ctrl);
299 ssleep(1);
Jan Kiszkafb51ccb2011-11-04 09:45:59 +0100300 pci_cfg_access_unlock(dev);
Yu Zhaodd7cc442009-03-20 11:25:15 +0800301
302 if (iov->link != dev->devfn)
303 sysfs_remove_link(&dev->dev.kobj, "dep_link");
304
Bjorn Helgaas6b136722012-11-09 20:27:53 -0700305 iov->num_VFs = 0;
Yijing Wang19b69842013-07-24 17:26:12 +0800306 pci_write_config_word(dev, iov->pos + PCI_SRIOV_NUM_VF, 0);
Yu Zhaodd7cc442009-03-20 11:25:15 +0800307}
308
Yu Zhaod1b054d2009-03-20 11:25:11 +0800309static 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
Yijing Wang62f87c02012-07-24 17:20:03 +0800320 if (pci_pcie_type(dev) != PCI_EXP_TYPE_RC_END &&
321 pci_pcie_type(dev) != PCI_EXP_TYPE_ENDPOINT)
Yu Zhaod1b054d2009-03-20 11:25:11 +0800322 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);
ethan.zhao045cc222013-11-06 22:49:13 +0800345 pci_write_config_word(dev, pos + PCI_SRIOV_NUM_VF, 0);
Yu Zhaod1b054d2009-03-20 11:25:11 +0800346 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);
Vaidyanathan Srinivasan8161fe92012-02-02 23:11:20 +0530358 pci_write_config_dword(dev, pos + PCI_SRIOV_SYS_PGSIZE, pgsz);
Yu Zhaod1b054d2009-03-20 11:25:11 +0800359
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;
Bjorn Helgaas6b136722012-11-09 20:27:53 -0700384 iov->total_VFs = total;
Yu Zhaod1b054d2009-03-20 11:25:11 +0800385 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);
Yijing Wang62f87c02012-07-24 17:20:03 +0800391 if (pci_pcie_type(dev) == PCI_EXP_TYPE_RC_END)
Yu Zhao4d135db2009-05-20 17:11:57 +0800392 iov->link = PCI_DEVFN(PCI_SLOT(dev->devfn), iov->link);
Yu Zhaod1b054d2009-03-20 11:25:11 +0800393
394 if (pdev)
395 iov->dev = pci_dev_get(pdev);
Yu Zhaoe277d2f2009-05-18 13:51:33 +0800396 else
Yu Zhaod1b054d2009-03-20 11:25:11 +0800397 iov->dev = dev;
Yu Zhaoe277d2f2009-05-18 13:51:33 +0800398
399 mutex_init(&iov->lock);
Yu Zhaod1b054d2009-03-20 11:25:11 +0800400
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{
Bjorn Helgaas6b136722012-11-09 20:27:53 -0700417 BUG_ON(dev->sriov->num_VFs);
Yu Zhaodd7cc442009-03-20 11:25:15 +0800418
Yu Zhaoe277d2f2009-05-18 13:51:33 +0800419 if (dev != dev->sriov->dev)
Yu Zhaod1b054d2009-03-20 11:25:11 +0800420 pci_dev_put(dev->sriov->dev);
421
Yu Zhaoe277d2f2009-05-18 13:51:33 +0800422 mutex_destroy(&dev->sriov->lock);
423
Yu Zhaod1b054d2009-03-20 11:25:11 +0800424 kfree(dev->sriov);
425 dev->sriov = NULL;
426}
427
Yu Zhao8c5cdb62009-03-20 11:25:12 +0800428static 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);
Bjorn Helgaas6b136722012-11-09 20:27:53 -0700442 pci_write_config_word(dev, iov->pos + PCI_SRIOV_NUM_VF, iov->num_VFs);
Yu Zhao8c5cdb62009-03-20 11:25:12 +0800443 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
Yu Zhaod1b054d2009-03-20 11:25:11 +0800448/**
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
Kenji Kaneshige5f4d91a2009-11-11 14:36:17 +0900458 if (!pci_is_pcie(dev))
Yu Zhaod1b054d2009-03-20 11:25:11 +0800459 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}
Yu Zhao8c5cdb62009-03-20 11:25:12 +0800499
500/**
Chris Wright6faf17f2009-08-28 13:00:06 -0700501 * 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 */
Cam Macdonell0e522472010-09-07 17:25:20 -0700510resource_size_t pci_sriov_resource_alignment(struct pci_dev *dev, int resno)
Chris Wright6faf17f2009-08-28 13:00:06 -0700511{
512 struct resource tmp;
513 enum pci_bar_type type;
514 int reg = pci_iov_resource_bar(dev, resno, &type);
Bjorn Helgaasf7625982013-11-14 11:28:18 -0700515
Chris Wright6faf17f2009-08-28 13:00:06 -0700516 if (!reg)
517 return 0;
518
519 __pci_read_base(dev, type, &tmp, reg);
520 return resource_alignment(&tmp);
521}
522
523/**
Yu Zhao8c5cdb62009-03-20 11:25:12 +0800524 * 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}
Yu Zhaoa28724b2009-03-20 11:25:13 +0800532
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;
Bjorn Helgaas6b136722012-11-09 20:27:53 -0700549 busnr = virtfn_bus(dev, dev->sriov->total_VFs - 1);
Yu Zhaoa28724b2009-03-20 11:25:13 +0800550 if (busnr > max)
551 max = busnr;
552 }
553
554 return max ? max - bus->number : 0;
555}
Yu Zhaodd7cc442009-03-20 11:25:15 +0800556
557/**
558 * pci_enable_sriov - enable the SR-IOV capability
559 * @dev: the PCI device
Randy Dunlap52a88732009-04-01 17:45:30 -0700560 * @nr_virtfn: number of virtual functions to enable
Yu Zhaodd7cc442009-03-20 11:25:15 +0800561 *
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)
Stefan Assmann652d1102013-07-31 16:47:56 -0600569 return -ENOSYS;
Yu Zhaodd7cc442009-03-20 11:25:15 +0800570
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);
Yu Zhao74bb1bc2009-03-20 11:25:16 +0800589
590/**
Williams, Mitch Afb8a0d92010-02-10 01:43:04 +0000591 * 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{
Bjorn Helgaas1452cd72012-11-09 20:35:01 -0700598 if (!dev->is_physfn)
Williams, Mitch Afb8a0d92010-02-10 01:43:04 +0000599 return 0;
Bjorn Helgaas1452cd72012-11-09 20:35:01 -0700600
601 return dev->sriov->num_VFs;
Williams, Mitch Afb8a0d92010-02-10 01:43:04 +0000602}
603EXPORT_SYMBOL_GPL(pci_num_vf);
Donald Dutilebff73152012-11-05 15:20:37 -0500604
605/**
Alexander Duyck5a8eb242013-04-25 04:42:29 +0000606 * 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.
Stefan Assmann652d1102013-07-31 16:47:56 -0600610 * If device is not a physical function returns 0.
Alexander Duyck5a8eb242013-04-25 04:42:29 +0000611 */
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/**
Donald Dutilebff73152012-11-05 15:20:37 -0500647 * pci_sriov_set_totalvfs -- reduce the TotalVFs available
648 * @dev: the PCI PF device
Randy Dunlap2094f162013-01-09 17:12:52 -0800649 * @numvfs: number that should be used for TotalVFs supported
Donald Dutilebff73152012-11-05 15:20:37 -0500650 *
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
Stefan Assmann652d1102013-07-31 16:47:56 -0600655 * value of numvfs valid. If not a PF return -ENOSYS;
656 * if numvfs is invalid return -EINVAL;
Donald Dutilebff73152012-11-05 15:20:37 -0500657 * if VFs already enabled, return -EBUSY.
658 */
659int pci_sriov_set_totalvfs(struct pci_dev *dev, u16 numvfs)
660{
Stefan Assmann652d1102013-07-31 16:47:56 -0600661 if (!dev->is_physfn)
662 return -ENOSYS;
663 if (numvfs > dev->sriov->total_VFs)
Donald Dutilebff73152012-11-05 15:20:37 -0500664 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
Bjorn Helgaas6b136722012-11-09 20:27:53 -0700670 dev->sriov->driver_max_VFs = numvfs;
Donald Dutilebff73152012-11-05 15:20:37 -0500671
672 return 0;
673}
674EXPORT_SYMBOL_GPL(pci_sriov_set_totalvfs);
675
676/**
Jonghwan Choiddc191f2013-07-08 14:02:43 -0600677 * pci_sriov_get_totalvfs -- get total VFs supported on this device
Donald Dutilebff73152012-11-05 15:20:37 -0500678 * @dev: the PCI PF device
679 *
680 * For a PCIe device with SRIOV support, return the PCIe
Bjorn Helgaas6b136722012-11-09 20:27:53 -0700681 * SRIOV capability value of TotalVFs or the value of driver_max_VFs
Stefan Assmann652d1102013-07-31 16:47:56 -0600682 * if the driver reduced it. Otherwise 0.
Donald Dutilebff73152012-11-05 15:20:37 -0500683 */
684int pci_sriov_get_totalvfs(struct pci_dev *dev)
685{
Bjorn Helgaas1452cd72012-11-09 20:35:01 -0700686 if (!dev->is_physfn)
Stefan Assmann652d1102013-07-31 16:47:56 -0600687 return 0;
Donald Dutilebff73152012-11-05 15:20:37 -0500688
Bjorn Helgaas6b136722012-11-09 20:27:53 -0700689 if (dev->sriov->driver_max_VFs)
690 return dev->sriov->driver_max_VFs;
Bjorn Helgaas1452cd72012-11-09 20:35:01 -0700691
692 return dev->sriov->total_VFs;
Donald Dutilebff73152012-11-05 15:20:37 -0500693}
694EXPORT_SYMBOL_GPL(pci_sriov_get_totalvfs);