blob: 1fe2d6fb19d5046ae78aaa6df3f26f37c12bf6bf [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;
Yu Zhaodd7cc442009-03-20 11:25:15 +080087
88 for (i = 0; i < PCI_SRIOV_NUM_BARS; i++) {
89 res = dev->resource + PCI_IOV_RESOURCES + i;
90 if (!res->parent)
91 continue;
92 virtfn->resource[i].name = pci_name(virtfn);
93 virtfn->resource[i].flags = res->flags;
94 size = resource_size(res);
Bjorn Helgaas6b136722012-11-09 20:27:53 -070095 do_div(size, iov->total_VFs);
Yu Zhaodd7cc442009-03-20 11:25:15 +080096 virtfn->resource[i].start = res->start + size * id;
97 virtfn->resource[i].end = virtfn->resource[i].start + size - 1;
98 rc = request_resource(res, &virtfn->resource[i]);
99 BUG_ON(rc);
100 }
101
102 if (reset)
Yu Zhao8c1c6992009-06-13 15:52:13 +0800103 __pci_reset_function(virtfn);
Yu Zhaodd7cc442009-03-20 11:25:15 +0800104
105 pci_device_add(virtfn, virtfn->bus);
106 mutex_unlock(&iov->dev->sriov->lock);
107
Yu Zhaodd7cc442009-03-20 11:25:15 +0800108 rc = pci_bus_add_device(virtfn);
Yu Zhaodd7cc442009-03-20 11:25:15 +0800109 sprintf(buf, "virtfn%u", id);
110 rc = sysfs_create_link(&dev->dev.kobj, &virtfn->dev.kobj, buf);
111 if (rc)
112 goto failed1;
113 rc = sysfs_create_link(&virtfn->dev.kobj, &dev->dev.kobj, "physfn");
114 if (rc)
115 goto failed2;
116
117 kobject_uevent(&virtfn->dev.kobj, KOBJ_CHANGE);
118
119 return 0;
120
121failed2:
122 sysfs_remove_link(&dev->dev.kobj, buf);
123failed1:
124 pci_dev_put(dev);
125 mutex_lock(&iov->dev->sriov->lock);
Yinghai Lu210647a2012-02-25 13:54:20 -0800126 pci_stop_and_remove_bus_device(virtfn);
Jiang Liudc087f22013-05-25 21:48:37 +0800127failed0:
128 virtfn_remove_bus(dev->bus, bus);
129failed:
Yu Zhaodd7cc442009-03-20 11:25:15 +0800130 mutex_unlock(&iov->dev->sriov->lock);
131
132 return rc;
133}
134
135static void virtfn_remove(struct pci_dev *dev, int id, int reset)
136{
137 char buf[VIRTFN_ID_LEN];
Yu Zhaodd7cc442009-03-20 11:25:15 +0800138 struct pci_dev *virtfn;
139 struct pci_sriov *iov = dev->sriov;
140
Jiang Liudc087f22013-05-25 21:48:37 +0800141 virtfn = pci_get_domain_bus_and_slot(pci_domain_nr(dev->bus),
142 virtfn_bus(dev, id),
143 virtfn_devfn(dev, id));
Yu Zhaodd7cc442009-03-20 11:25:15 +0800144 if (!virtfn)
145 return;
146
Yu Zhaodd7cc442009-03-20 11:25:15 +0800147 if (reset) {
148 device_release_driver(&virtfn->dev);
Yu Zhao8c1c6992009-06-13 15:52:13 +0800149 __pci_reset_function(virtfn);
Yu Zhaodd7cc442009-03-20 11:25:15 +0800150 }
151
152 sprintf(buf, "virtfn%u", id);
153 sysfs_remove_link(&dev->dev.kobj, buf);
Yinghai Lu09cedbe2012-02-04 22:55:01 -0800154 /*
155 * pci_stop_dev() could have been called for this virtfn already,
156 * so the directory for the virtfn may have been removed before.
157 * Double check to avoid spurious sysfs warnings.
158 */
159 if (virtfn->dev.kobj.sd)
160 sysfs_remove_link(&virtfn->dev.kobj, "physfn");
Yu Zhaodd7cc442009-03-20 11:25:15 +0800161
162 mutex_lock(&iov->dev->sriov->lock);
Yinghai Lu210647a2012-02-25 13:54:20 -0800163 pci_stop_and_remove_bus_device(virtfn);
Jiang Liudc087f22013-05-25 21:48:37 +0800164 virtfn_remove_bus(dev->bus, virtfn->bus);
Yu Zhaodd7cc442009-03-20 11:25:15 +0800165 mutex_unlock(&iov->dev->sriov->lock);
166
Jiang Liudc087f22013-05-25 21:48:37 +0800167 /* balance pci_get_domain_bus_and_slot() */
168 pci_dev_put(virtfn);
Yu Zhaodd7cc442009-03-20 11:25:15 +0800169 pci_dev_put(dev);
170}
171
Yu Zhao74bb1bc2009-03-20 11:25:16 +0800172static int sriov_migration(struct pci_dev *dev)
173{
174 u16 status;
175 struct pci_sriov *iov = dev->sriov;
176
Bjorn Helgaas6b136722012-11-09 20:27:53 -0700177 if (!iov->num_VFs)
Yu Zhao74bb1bc2009-03-20 11:25:16 +0800178 return 0;
179
180 if (!(iov->cap & PCI_SRIOV_CAP_VFM))
181 return 0;
182
183 pci_read_config_word(dev, iov->pos + PCI_SRIOV_STATUS, &status);
184 if (!(status & PCI_SRIOV_STATUS_VFM))
185 return 0;
186
187 schedule_work(&iov->mtask);
188
189 return 1;
190}
191
192static void sriov_migration_task(struct work_struct *work)
193{
194 int i;
195 u8 state;
196 u16 status;
197 struct pci_sriov *iov = container_of(work, struct pci_sriov, mtask);
198
Bjorn Helgaas6b136722012-11-09 20:27:53 -0700199 for (i = iov->initial_VFs; i < iov->num_VFs; i++) {
Yu Zhao74bb1bc2009-03-20 11:25:16 +0800200 state = readb(iov->mstate + i);
201 if (state == PCI_SRIOV_VFM_MI) {
202 writeb(PCI_SRIOV_VFM_AV, iov->mstate + i);
203 state = readb(iov->mstate + i);
204 if (state == PCI_SRIOV_VFM_AV)
205 virtfn_add(iov->self, i, 1);
206 } else if (state == PCI_SRIOV_VFM_MO) {
207 virtfn_remove(iov->self, i, 1);
208 writeb(PCI_SRIOV_VFM_UA, iov->mstate + i);
209 state = readb(iov->mstate + i);
210 if (state == PCI_SRIOV_VFM_AV)
211 virtfn_add(iov->self, i, 0);
212 }
213 }
214
215 pci_read_config_word(iov->self, iov->pos + PCI_SRIOV_STATUS, &status);
216 status &= ~PCI_SRIOV_STATUS_VFM;
217 pci_write_config_word(iov->self, iov->pos + PCI_SRIOV_STATUS, status);
218}
219
220static int sriov_enable_migration(struct pci_dev *dev, int nr_virtfn)
221{
222 int bir;
223 u32 table;
224 resource_size_t pa;
225 struct pci_sriov *iov = dev->sriov;
226
Bjorn Helgaas6b136722012-11-09 20:27:53 -0700227 if (nr_virtfn <= iov->initial_VFs)
Yu Zhao74bb1bc2009-03-20 11:25:16 +0800228 return 0;
229
230 pci_read_config_dword(dev, iov->pos + PCI_SRIOV_VFM, &table);
231 bir = PCI_SRIOV_VFM_BIR(table);
232 if (bir > PCI_STD_RESOURCE_END)
233 return -EIO;
234
235 table = PCI_SRIOV_VFM_OFFSET(table);
236 if (table + nr_virtfn > pci_resource_len(dev, bir))
237 return -EIO;
238
239 pa = pci_resource_start(dev, bir) + table;
240 iov->mstate = ioremap(pa, nr_virtfn);
241 if (!iov->mstate)
242 return -ENOMEM;
243
244 INIT_WORK(&iov->mtask, sriov_migration_task);
245
246 iov->ctrl |= PCI_SRIOV_CTRL_VFM | PCI_SRIOV_CTRL_INTR;
247 pci_write_config_word(dev, iov->pos + PCI_SRIOV_CTRL, iov->ctrl);
248
249 return 0;
250}
251
252static void sriov_disable_migration(struct pci_dev *dev)
253{
254 struct pci_sriov *iov = dev->sriov;
255
256 iov->ctrl &= ~(PCI_SRIOV_CTRL_VFM | PCI_SRIOV_CTRL_INTR);
257 pci_write_config_word(dev, iov->pos + PCI_SRIOV_CTRL, iov->ctrl);
258
259 cancel_work_sync(&iov->mtask);
260 iounmap(iov->mstate);
261}
262
Yu Zhaodd7cc442009-03-20 11:25:15 +0800263static int sriov_enable(struct pci_dev *dev, int nr_virtfn)
264{
265 int rc;
266 int i, j;
267 int nres;
268 u16 offset, stride, initial;
269 struct resource *res;
270 struct pci_dev *pdev;
271 struct pci_sriov *iov = dev->sriov;
Ram Paibbef98a2011-11-06 10:33:10 +0800272 int bars = 0;
Yu Zhaodd7cc442009-03-20 11:25:15 +0800273
274 if (!nr_virtfn)
275 return 0;
276
Bjorn Helgaas6b136722012-11-09 20:27:53 -0700277 if (iov->num_VFs)
Yu Zhaodd7cc442009-03-20 11:25:15 +0800278 return -EINVAL;
279
280 pci_read_config_word(dev, iov->pos + PCI_SRIOV_INITIAL_VF, &initial);
Bjorn Helgaas6b136722012-11-09 20:27:53 -0700281 if (initial > iov->total_VFs ||
282 (!(iov->cap & PCI_SRIOV_CAP_VFM) && (initial != iov->total_VFs)))
Yu Zhaodd7cc442009-03-20 11:25:15 +0800283 return -EIO;
284
Bjorn Helgaas6b136722012-11-09 20:27:53 -0700285 if (nr_virtfn < 0 || nr_virtfn > iov->total_VFs ||
Yu Zhaodd7cc442009-03-20 11:25:15 +0800286 (!(iov->cap & PCI_SRIOV_CAP_VFM) && (nr_virtfn > initial)))
287 return -EINVAL;
288
Yu Zhaodd7cc442009-03-20 11:25:15 +0800289 pci_read_config_word(dev, iov->pos + PCI_SRIOV_VF_OFFSET, &offset);
290 pci_read_config_word(dev, iov->pos + PCI_SRIOV_VF_STRIDE, &stride);
291 if (!offset || (nr_virtfn > 1 && !stride))
292 return -EIO;
293
294 nres = 0;
295 for (i = 0; i < PCI_SRIOV_NUM_BARS; i++) {
Ram Paibbef98a2011-11-06 10:33:10 +0800296 bars |= (1 << (i + PCI_IOV_RESOURCES));
Yu Zhaodd7cc442009-03-20 11:25:15 +0800297 res = dev->resource + PCI_IOV_RESOURCES + i;
298 if (res->parent)
299 nres++;
300 }
301 if (nres != iov->nres) {
302 dev_err(&dev->dev, "not enough MMIO resources for SR-IOV\n");
303 return -ENOMEM;
304 }
305
306 iov->offset = offset;
307 iov->stride = stride;
308
Yinghai Lub918c622012-05-17 18:51:11 -0700309 if (virtfn_bus(dev, nr_virtfn - 1) > dev->bus->busn_res.end) {
Yu Zhaodd7cc442009-03-20 11:25:15 +0800310 dev_err(&dev->dev, "SR-IOV: bus number out of range\n");
311 return -ENOMEM;
312 }
313
Ram Paibbef98a2011-11-06 10:33:10 +0800314 if (pci_enable_resources(dev, bars)) {
315 dev_err(&dev->dev, "SR-IOV: IOV BARS not allocated\n");
316 return -ENOMEM;
317 }
318
Yu Zhaodd7cc442009-03-20 11:25:15 +0800319 if (iov->link != dev->devfn) {
320 pdev = pci_get_slot(dev->bus, iov->link);
321 if (!pdev)
322 return -ENODEV;
323
Jiang Liudc087f22013-05-25 21:48:37 +0800324 if (!pdev->is_physfn) {
325 pci_dev_put(pdev);
Stefan Assmann652d1102013-07-31 16:47:56 -0600326 return -ENOSYS;
Jiang Liudc087f22013-05-25 21:48:37 +0800327 }
Yu Zhaodd7cc442009-03-20 11:25:15 +0800328
329 rc = sysfs_create_link(&dev->dev.kobj,
330 &pdev->dev.kobj, "dep_link");
Jiang Liudc087f22013-05-25 21:48:37 +0800331 pci_dev_put(pdev);
Yu Zhaodd7cc442009-03-20 11:25:15 +0800332 if (rc)
333 return rc;
334 }
335
Yijing Wang19b69842013-07-24 17:26:12 +0800336 pci_write_config_word(dev, iov->pos + PCI_SRIOV_NUM_VF, nr_virtfn);
Yu Zhaodd7cc442009-03-20 11:25:15 +0800337 iov->ctrl |= PCI_SRIOV_CTRL_VFE | PCI_SRIOV_CTRL_MSE;
Jan Kiszkafb51ccb2011-11-04 09:45:59 +0100338 pci_cfg_access_lock(dev);
Yu Zhaodd7cc442009-03-20 11:25:15 +0800339 pci_write_config_word(dev, iov->pos + PCI_SRIOV_CTRL, iov->ctrl);
340 msleep(100);
Jan Kiszkafb51ccb2011-11-04 09:45:59 +0100341 pci_cfg_access_unlock(dev);
Yu Zhaodd7cc442009-03-20 11:25:15 +0800342
Bjorn Helgaas6b136722012-11-09 20:27:53 -0700343 iov->initial_VFs = initial;
Yu Zhaodd7cc442009-03-20 11:25:15 +0800344 if (nr_virtfn < initial)
345 initial = nr_virtfn;
346
347 for (i = 0; i < initial; i++) {
348 rc = virtfn_add(dev, i, 0);
349 if (rc)
350 goto failed;
351 }
352
Yu Zhao74bb1bc2009-03-20 11:25:16 +0800353 if (iov->cap & PCI_SRIOV_CAP_VFM) {
354 rc = sriov_enable_migration(dev, nr_virtfn);
355 if (rc)
356 goto failed;
357 }
358
Yu Zhaodd7cc442009-03-20 11:25:15 +0800359 kobject_uevent(&dev->dev.kobj, KOBJ_CHANGE);
Bjorn Helgaas6b136722012-11-09 20:27:53 -0700360 iov->num_VFs = nr_virtfn;
Yu Zhaodd7cc442009-03-20 11:25:15 +0800361
362 return 0;
363
364failed:
365 for (j = 0; j < i; j++)
366 virtfn_remove(dev, j, 0);
367
368 iov->ctrl &= ~(PCI_SRIOV_CTRL_VFE | PCI_SRIOV_CTRL_MSE);
Jan Kiszkafb51ccb2011-11-04 09:45:59 +0100369 pci_cfg_access_lock(dev);
Yu Zhaodd7cc442009-03-20 11:25:15 +0800370 pci_write_config_word(dev, iov->pos + PCI_SRIOV_CTRL, iov->ctrl);
Yijing Wang19b69842013-07-24 17:26:12 +0800371 pci_write_config_word(dev, iov->pos + PCI_SRIOV_NUM_VF, 0);
Yu Zhaodd7cc442009-03-20 11:25:15 +0800372 ssleep(1);
Jan Kiszkafb51ccb2011-11-04 09:45:59 +0100373 pci_cfg_access_unlock(dev);
Yu Zhaodd7cc442009-03-20 11:25:15 +0800374
375 if (iov->link != dev->devfn)
376 sysfs_remove_link(&dev->dev.kobj, "dep_link");
377
378 return rc;
379}
380
381static void sriov_disable(struct pci_dev *dev)
382{
383 int i;
384 struct pci_sriov *iov = dev->sriov;
385
Bjorn Helgaas6b136722012-11-09 20:27:53 -0700386 if (!iov->num_VFs)
Yu Zhaodd7cc442009-03-20 11:25:15 +0800387 return;
388
Yu Zhao74bb1bc2009-03-20 11:25:16 +0800389 if (iov->cap & PCI_SRIOV_CAP_VFM)
390 sriov_disable_migration(dev);
391
Bjorn Helgaas6b136722012-11-09 20:27:53 -0700392 for (i = 0; i < iov->num_VFs; i++)
Yu Zhaodd7cc442009-03-20 11:25:15 +0800393 virtfn_remove(dev, i, 0);
394
395 iov->ctrl &= ~(PCI_SRIOV_CTRL_VFE | PCI_SRIOV_CTRL_MSE);
Jan Kiszkafb51ccb2011-11-04 09:45:59 +0100396 pci_cfg_access_lock(dev);
Yu Zhaodd7cc442009-03-20 11:25:15 +0800397 pci_write_config_word(dev, iov->pos + PCI_SRIOV_CTRL, iov->ctrl);
398 ssleep(1);
Jan Kiszkafb51ccb2011-11-04 09:45:59 +0100399 pci_cfg_access_unlock(dev);
Yu Zhaodd7cc442009-03-20 11:25:15 +0800400
401 if (iov->link != dev->devfn)
402 sysfs_remove_link(&dev->dev.kobj, "dep_link");
403
Bjorn Helgaas6b136722012-11-09 20:27:53 -0700404 iov->num_VFs = 0;
Yijing Wang19b69842013-07-24 17:26:12 +0800405 pci_write_config_word(dev, iov->pos + PCI_SRIOV_NUM_VF, 0);
Yu Zhaodd7cc442009-03-20 11:25:15 +0800406}
407
Yu Zhaod1b054d2009-03-20 11:25:11 +0800408static int sriov_init(struct pci_dev *dev, int pos)
409{
410 int i;
411 int rc;
412 int nres;
413 u32 pgsz;
414 u16 ctrl, total, offset, stride;
415 struct pci_sriov *iov;
416 struct resource *res;
417 struct pci_dev *pdev;
418
Yijing Wang62f87c02012-07-24 17:20:03 +0800419 if (pci_pcie_type(dev) != PCI_EXP_TYPE_RC_END &&
420 pci_pcie_type(dev) != PCI_EXP_TYPE_ENDPOINT)
Yu Zhaod1b054d2009-03-20 11:25:11 +0800421 return -ENODEV;
422
423 pci_read_config_word(dev, pos + PCI_SRIOV_CTRL, &ctrl);
424 if (ctrl & PCI_SRIOV_CTRL_VFE) {
425 pci_write_config_word(dev, pos + PCI_SRIOV_CTRL, 0);
426 ssleep(1);
427 }
428
429 pci_read_config_word(dev, pos + PCI_SRIOV_TOTAL_VF, &total);
430 if (!total)
431 return 0;
432
433 ctrl = 0;
434 list_for_each_entry(pdev, &dev->bus->devices, bus_list)
435 if (pdev->is_physfn)
436 goto found;
437
438 pdev = NULL;
439 if (pci_ari_enabled(dev->bus))
440 ctrl |= PCI_SRIOV_CTRL_ARI;
441
442found:
443 pci_write_config_word(dev, pos + PCI_SRIOV_CTRL, ctrl);
Yu Zhaod1b054d2009-03-20 11:25:11 +0800444 pci_read_config_word(dev, pos + PCI_SRIOV_VF_OFFSET, &offset);
445 pci_read_config_word(dev, pos + PCI_SRIOV_VF_STRIDE, &stride);
446 if (!offset || (total > 1 && !stride))
447 return -EIO;
448
449 pci_read_config_dword(dev, pos + PCI_SRIOV_SUP_PGSIZE, &pgsz);
450 i = PAGE_SHIFT > 12 ? PAGE_SHIFT - 12 : 0;
451 pgsz &= ~((1 << i) - 1);
452 if (!pgsz)
453 return -EIO;
454
455 pgsz &= ~(pgsz - 1);
Vaidyanathan Srinivasan8161fe92012-02-02 23:11:20 +0530456 pci_write_config_dword(dev, pos + PCI_SRIOV_SYS_PGSIZE, pgsz);
Yu Zhaod1b054d2009-03-20 11:25:11 +0800457
458 nres = 0;
459 for (i = 0; i < PCI_SRIOV_NUM_BARS; i++) {
460 res = dev->resource + PCI_IOV_RESOURCES + i;
461 i += __pci_read_base(dev, pci_bar_unknown, res,
462 pos + PCI_SRIOV_BAR + i * 4);
463 if (!res->flags)
464 continue;
465 if (resource_size(res) & (PAGE_SIZE - 1)) {
466 rc = -EIO;
467 goto failed;
468 }
469 res->end = res->start + resource_size(res) * total - 1;
470 nres++;
471 }
472
473 iov = kzalloc(sizeof(*iov), GFP_KERNEL);
474 if (!iov) {
475 rc = -ENOMEM;
476 goto failed;
477 }
478
479 iov->pos = pos;
480 iov->nres = nres;
481 iov->ctrl = ctrl;
Bjorn Helgaas6b136722012-11-09 20:27:53 -0700482 iov->total_VFs = total;
Yu Zhaod1b054d2009-03-20 11:25:11 +0800483 iov->offset = offset;
484 iov->stride = stride;
485 iov->pgsz = pgsz;
486 iov->self = dev;
487 pci_read_config_dword(dev, pos + PCI_SRIOV_CAP, &iov->cap);
488 pci_read_config_byte(dev, pos + PCI_SRIOV_FUNC_LINK, &iov->link);
Yijing Wang62f87c02012-07-24 17:20:03 +0800489 if (pci_pcie_type(dev) == PCI_EXP_TYPE_RC_END)
Yu Zhao4d135db2009-05-20 17:11:57 +0800490 iov->link = PCI_DEVFN(PCI_SLOT(dev->devfn), iov->link);
Yu Zhaod1b054d2009-03-20 11:25:11 +0800491
492 if (pdev)
493 iov->dev = pci_dev_get(pdev);
Yu Zhaoe277d2f2009-05-18 13:51:33 +0800494 else
Yu Zhaod1b054d2009-03-20 11:25:11 +0800495 iov->dev = dev;
Yu Zhaoe277d2f2009-05-18 13:51:33 +0800496
497 mutex_init(&iov->lock);
Yu Zhaod1b054d2009-03-20 11:25:11 +0800498
499 dev->sriov = iov;
500 dev->is_physfn = 1;
501
502 return 0;
503
504failed:
505 for (i = 0; i < PCI_SRIOV_NUM_BARS; i++) {
506 res = dev->resource + PCI_IOV_RESOURCES + i;
507 res->flags = 0;
508 }
509
510 return rc;
511}
512
513static void sriov_release(struct pci_dev *dev)
514{
Bjorn Helgaas6b136722012-11-09 20:27:53 -0700515 BUG_ON(dev->sriov->num_VFs);
Yu Zhaodd7cc442009-03-20 11:25:15 +0800516
Yu Zhaoe277d2f2009-05-18 13:51:33 +0800517 if (dev != dev->sriov->dev)
Yu Zhaod1b054d2009-03-20 11:25:11 +0800518 pci_dev_put(dev->sriov->dev);
519
Yu Zhaoe277d2f2009-05-18 13:51:33 +0800520 mutex_destroy(&dev->sriov->lock);
521
Yu Zhaod1b054d2009-03-20 11:25:11 +0800522 kfree(dev->sriov);
523 dev->sriov = NULL;
524}
525
Yu Zhao8c5cdb62009-03-20 11:25:12 +0800526static void sriov_restore_state(struct pci_dev *dev)
527{
528 int i;
529 u16 ctrl;
530 struct pci_sriov *iov = dev->sriov;
531
532 pci_read_config_word(dev, iov->pos + PCI_SRIOV_CTRL, &ctrl);
533 if (ctrl & PCI_SRIOV_CTRL_VFE)
534 return;
535
536 for (i = PCI_IOV_RESOURCES; i <= PCI_IOV_RESOURCE_END; i++)
537 pci_update_resource(dev, i);
538
539 pci_write_config_dword(dev, iov->pos + PCI_SRIOV_SYS_PGSIZE, iov->pgsz);
Bjorn Helgaas6b136722012-11-09 20:27:53 -0700540 pci_write_config_word(dev, iov->pos + PCI_SRIOV_NUM_VF, iov->num_VFs);
Yu Zhao8c5cdb62009-03-20 11:25:12 +0800541 pci_write_config_word(dev, iov->pos + PCI_SRIOV_CTRL, iov->ctrl);
542 if (iov->ctrl & PCI_SRIOV_CTRL_VFE)
543 msleep(100);
544}
545
Yu Zhaod1b054d2009-03-20 11:25:11 +0800546/**
547 * pci_iov_init - initialize the IOV capability
548 * @dev: the PCI device
549 *
550 * Returns 0 on success, or negative on failure.
551 */
552int pci_iov_init(struct pci_dev *dev)
553{
554 int pos;
555
Kenji Kaneshige5f4d91a2009-11-11 14:36:17 +0900556 if (!pci_is_pcie(dev))
Yu Zhaod1b054d2009-03-20 11:25:11 +0800557 return -ENODEV;
558
559 pos = pci_find_ext_capability(dev, PCI_EXT_CAP_ID_SRIOV);
560 if (pos)
561 return sriov_init(dev, pos);
562
563 return -ENODEV;
564}
565
566/**
567 * pci_iov_release - release resources used by the IOV capability
568 * @dev: the PCI device
569 */
570void pci_iov_release(struct pci_dev *dev)
571{
572 if (dev->is_physfn)
573 sriov_release(dev);
574}
575
576/**
577 * pci_iov_resource_bar - get position of the SR-IOV BAR
578 * @dev: the PCI device
579 * @resno: the resource number
580 * @type: the BAR type to be filled in
581 *
582 * Returns position of the BAR encapsulated in the SR-IOV capability.
583 */
584int pci_iov_resource_bar(struct pci_dev *dev, int resno,
585 enum pci_bar_type *type)
586{
587 if (resno < PCI_IOV_RESOURCES || resno > PCI_IOV_RESOURCE_END)
588 return 0;
589
590 BUG_ON(!dev->is_physfn);
591
592 *type = pci_bar_unknown;
593
594 return dev->sriov->pos + PCI_SRIOV_BAR +
595 4 * (resno - PCI_IOV_RESOURCES);
596}
Yu Zhao8c5cdb62009-03-20 11:25:12 +0800597
598/**
Chris Wright6faf17f2009-08-28 13:00:06 -0700599 * pci_sriov_resource_alignment - get resource alignment for VF BAR
600 * @dev: the PCI device
601 * @resno: the resource number
602 *
603 * Returns the alignment of the VF BAR found in the SR-IOV capability.
604 * This is not the same as the resource size which is defined as
605 * the VF BAR size multiplied by the number of VFs. The alignment
606 * is just the VF BAR size.
607 */
Cam Macdonell0e522472010-09-07 17:25:20 -0700608resource_size_t pci_sriov_resource_alignment(struct pci_dev *dev, int resno)
Chris Wright6faf17f2009-08-28 13:00:06 -0700609{
610 struct resource tmp;
611 enum pci_bar_type type;
612 int reg = pci_iov_resource_bar(dev, resno, &type);
Bjorn Helgaasf7625982013-11-14 11:28:18 -0700613
Chris Wright6faf17f2009-08-28 13:00:06 -0700614 if (!reg)
615 return 0;
616
617 __pci_read_base(dev, type, &tmp, reg);
618 return resource_alignment(&tmp);
619}
620
621/**
Yu Zhao8c5cdb62009-03-20 11:25:12 +0800622 * pci_restore_iov_state - restore the state of the IOV capability
623 * @dev: the PCI device
624 */
625void pci_restore_iov_state(struct pci_dev *dev)
626{
627 if (dev->is_physfn)
628 sriov_restore_state(dev);
629}
Yu Zhaoa28724b2009-03-20 11:25:13 +0800630
631/**
632 * pci_iov_bus_range - find bus range used by Virtual Function
633 * @bus: the PCI bus
634 *
635 * Returns max number of buses (exclude current one) used by Virtual
636 * Functions.
637 */
638int pci_iov_bus_range(struct pci_bus *bus)
639{
640 int max = 0;
641 u8 busnr;
642 struct pci_dev *dev;
643
644 list_for_each_entry(dev, &bus->devices, bus_list) {
645 if (!dev->is_physfn)
646 continue;
Bjorn Helgaas6b136722012-11-09 20:27:53 -0700647 busnr = virtfn_bus(dev, dev->sriov->total_VFs - 1);
Yu Zhaoa28724b2009-03-20 11:25:13 +0800648 if (busnr > max)
649 max = busnr;
650 }
651
652 return max ? max - bus->number : 0;
653}
Yu Zhaodd7cc442009-03-20 11:25:15 +0800654
655/**
656 * pci_enable_sriov - enable the SR-IOV capability
657 * @dev: the PCI device
Randy Dunlap52a88732009-04-01 17:45:30 -0700658 * @nr_virtfn: number of virtual functions to enable
Yu Zhaodd7cc442009-03-20 11:25:15 +0800659 *
660 * Returns 0 on success, or negative on failure.
661 */
662int pci_enable_sriov(struct pci_dev *dev, int nr_virtfn)
663{
664 might_sleep();
665
666 if (!dev->is_physfn)
Stefan Assmann652d1102013-07-31 16:47:56 -0600667 return -ENOSYS;
Yu Zhaodd7cc442009-03-20 11:25:15 +0800668
669 return sriov_enable(dev, nr_virtfn);
670}
671EXPORT_SYMBOL_GPL(pci_enable_sriov);
672
673/**
674 * pci_disable_sriov - disable the SR-IOV capability
675 * @dev: the PCI device
676 */
677void pci_disable_sriov(struct pci_dev *dev)
678{
679 might_sleep();
680
681 if (!dev->is_physfn)
682 return;
683
684 sriov_disable(dev);
685}
686EXPORT_SYMBOL_GPL(pci_disable_sriov);
Yu Zhao74bb1bc2009-03-20 11:25:16 +0800687
688/**
689 * pci_sriov_migration - notify SR-IOV core of Virtual Function Migration
690 * @dev: the PCI device
691 *
692 * Returns IRQ_HANDLED if the IRQ is handled, or IRQ_NONE if not.
693 *
694 * Physical Function driver is responsible to register IRQ handler using
695 * VF Migration Interrupt Message Number, and call this function when the
696 * interrupt is generated by the hardware.
697 */
698irqreturn_t pci_sriov_migration(struct pci_dev *dev)
699{
700 if (!dev->is_physfn)
701 return IRQ_NONE;
702
703 return sriov_migration(dev) ? IRQ_HANDLED : IRQ_NONE;
704}
705EXPORT_SYMBOL_GPL(pci_sriov_migration);
Yu Zhao302b4212009-05-18 13:51:32 +0800706
Williams, Mitch Afb8a0d92010-02-10 01:43:04 +0000707/**
708 * pci_num_vf - return number of VFs associated with a PF device_release_driver
709 * @dev: the PCI device
710 *
711 * Returns number of VFs, or 0 if SR-IOV is not enabled.
712 */
713int pci_num_vf(struct pci_dev *dev)
714{
Bjorn Helgaas1452cd72012-11-09 20:35:01 -0700715 if (!dev->is_physfn)
Williams, Mitch Afb8a0d92010-02-10 01:43:04 +0000716 return 0;
Bjorn Helgaas1452cd72012-11-09 20:35:01 -0700717
718 return dev->sriov->num_VFs;
Williams, Mitch Afb8a0d92010-02-10 01:43:04 +0000719}
720EXPORT_SYMBOL_GPL(pci_num_vf);
Donald Dutilebff73152012-11-05 15:20:37 -0500721
722/**
Alexander Duyck5a8eb242013-04-25 04:42:29 +0000723 * pci_vfs_assigned - returns number of VFs are assigned to a guest
724 * @dev: the PCI device
725 *
726 * Returns number of VFs belonging to this device that are assigned to a guest.
Stefan Assmann652d1102013-07-31 16:47:56 -0600727 * If device is not a physical function returns 0.
Alexander Duyck5a8eb242013-04-25 04:42:29 +0000728 */
729int pci_vfs_assigned(struct pci_dev *dev)
730{
731 struct pci_dev *vfdev;
732 unsigned int vfs_assigned = 0;
733 unsigned short dev_id;
734
735 /* only search if we are a PF */
736 if (!dev->is_physfn)
737 return 0;
738
739 /*
740 * determine the device ID for the VFs, the vendor ID will be the
741 * same as the PF so there is no need to check for that one
742 */
743 pci_read_config_word(dev, dev->sriov->pos + PCI_SRIOV_VF_DID, &dev_id);
744
745 /* loop through all the VFs to see if we own any that are assigned */
746 vfdev = pci_get_device(dev->vendor, dev_id, NULL);
747 while (vfdev) {
748 /*
749 * It is considered assigned if it is a virtual function with
750 * our dev as the physical function and the assigned bit is set
751 */
752 if (vfdev->is_virtfn && (vfdev->physfn == dev) &&
753 (vfdev->dev_flags & PCI_DEV_FLAGS_ASSIGNED))
754 vfs_assigned++;
755
756 vfdev = pci_get_device(dev->vendor, dev_id, vfdev);
757 }
758
759 return vfs_assigned;
760}
761EXPORT_SYMBOL_GPL(pci_vfs_assigned);
762
763/**
Donald Dutilebff73152012-11-05 15:20:37 -0500764 * pci_sriov_set_totalvfs -- reduce the TotalVFs available
765 * @dev: the PCI PF device
Randy Dunlap2094f162013-01-09 17:12:52 -0800766 * @numvfs: number that should be used for TotalVFs supported
Donald Dutilebff73152012-11-05 15:20:37 -0500767 *
768 * Should be called from PF driver's probe routine with
769 * device's mutex held.
770 *
771 * Returns 0 if PF is an SRIOV-capable device and
Stefan Assmann652d1102013-07-31 16:47:56 -0600772 * value of numvfs valid. If not a PF return -ENOSYS;
773 * if numvfs is invalid return -EINVAL;
Donald Dutilebff73152012-11-05 15:20:37 -0500774 * if VFs already enabled, return -EBUSY.
775 */
776int pci_sriov_set_totalvfs(struct pci_dev *dev, u16 numvfs)
777{
Stefan Assmann652d1102013-07-31 16:47:56 -0600778 if (!dev->is_physfn)
779 return -ENOSYS;
780 if (numvfs > dev->sriov->total_VFs)
Donald Dutilebff73152012-11-05 15:20:37 -0500781 return -EINVAL;
782
783 /* Shouldn't change if VFs already enabled */
784 if (dev->sriov->ctrl & PCI_SRIOV_CTRL_VFE)
785 return -EBUSY;
786 else
Bjorn Helgaas6b136722012-11-09 20:27:53 -0700787 dev->sriov->driver_max_VFs = numvfs;
Donald Dutilebff73152012-11-05 15:20:37 -0500788
789 return 0;
790}
791EXPORT_SYMBOL_GPL(pci_sriov_set_totalvfs);
792
793/**
Jonghwan Choiddc191f2013-07-08 14:02:43 -0600794 * pci_sriov_get_totalvfs -- get total VFs supported on this device
Donald Dutilebff73152012-11-05 15:20:37 -0500795 * @dev: the PCI PF device
796 *
797 * For a PCIe device with SRIOV support, return the PCIe
Bjorn Helgaas6b136722012-11-09 20:27:53 -0700798 * SRIOV capability value of TotalVFs or the value of driver_max_VFs
Stefan Assmann652d1102013-07-31 16:47:56 -0600799 * if the driver reduced it. Otherwise 0.
Donald Dutilebff73152012-11-05 15:20:37 -0500800 */
801int pci_sriov_get_totalvfs(struct pci_dev *dev)
802{
Bjorn Helgaas1452cd72012-11-09 20:35:01 -0700803 if (!dev->is_physfn)
Stefan Assmann652d1102013-07-31 16:47:56 -0600804 return 0;
Donald Dutilebff73152012-11-05 15:20:37 -0500805
Bjorn Helgaas6b136722012-11-09 20:27:53 -0700806 if (dev->sriov->driver_max_VFs)
807 return dev->sriov->driver_max_VFs;
Bjorn Helgaas1452cd72012-11-09 20:35:01 -0700808
809 return dev->sriov->total_VFs;
Donald Dutilebff73152012-11-05 15:20:37 -0500810}
811EXPORT_SYMBOL_GPL(pci_sriov_get_totalvfs);