1 #include <linux/delay.h>
2 #include <linux/pci.h>
3 #include <linux/module.h>
4 #include <linux/sched.h>
5 #include <linux/slab.h>
6 #include <linux/ioport.h>
7 #include <linux/wait.h>
8
9 #include "pci.h"
10
11 /*
12 * This interrupt-safe spinlock protects all accesses to PCI
13 * configuration space.
14 */
15
16 DEFINE_RAW_SPINLOCK(pci_lock);
17
18 /*
19 * Wrappers for all PCI configuration access functions. They just check
20 * alignment, do locking and call the low-level functions pointed to
21 * by pci_dev->ops.
22 */
23
24 #define PCI_byte_BAD 0
25 #define PCI_word_BAD (pos & 1)
26 #define PCI_dword_BAD (pos & 3)
27
28 #define PCI_OP_READ(size,type,len) \
29 int pci_bus_read_config_##size \
30 (struct pci_bus *bus, unsigned int devfn, int pos, type *value) \
31 { \
32 int res; \
33 unsigned long flags; \
34 u32 data = 0; \
35 if (PCI_##size##_BAD) return PCIBIOS_BAD_REGISTER_NUMBER; \
36 raw_spin_lock_irqsave(&pci_lock, flags); \
37 res = bus->ops->read(bus, devfn, pos, len, &data); \
38 *value = (type)data; \
39 raw_spin_unlock_irqrestore(&pci_lock, flags); \
40 return res; \
41 }
42
43 #define PCI_OP_WRITE(size,type,len) \
44 int pci_bus_write_config_##size \
45 (struct pci_bus *bus, unsigned int devfn, int pos, type value) \
46 { \
47 int res; \
48 unsigned long flags; \
49 if (PCI_##size##_BAD) return PCIBIOS_BAD_REGISTER_NUMBER; \
50 raw_spin_lock_irqsave(&pci_lock, flags); \
51 res = bus->ops->write(bus, devfn, pos, len, value); \
52 raw_spin_unlock_irqrestore(&pci_lock, flags); \
53 return res; \
54 }
55
56 PCI_OP_READ(byte, u8, 1)
57 PCI_OP_READ(word, u16, 2)
58 PCI_OP_READ(dword, u32, 4)
59 PCI_OP_WRITE(byte, u8, 1)
60 PCI_OP_WRITE(word, u16, 2)
61 PCI_OP_WRITE(dword, u32, 4)
62
63 EXPORT_SYMBOL(pci_bus_read_config_byte);
64 EXPORT_SYMBOL(pci_bus_read_config_word);
65 EXPORT_SYMBOL(pci_bus_read_config_dword);
66 EXPORT_SYMBOL(pci_bus_write_config_byte);
67 EXPORT_SYMBOL(pci_bus_write_config_word);
68 EXPORT_SYMBOL(pci_bus_write_config_dword);
69
pci_generic_config_read(struct pci_bus * bus,unsigned int devfn,int where,int size,u32 * val)70 int pci_generic_config_read(struct pci_bus *bus, unsigned int devfn,
71 int where, int size, u32 *val)
72 {
73 void __iomem *addr;
74
75 addr = bus->ops->map_bus(bus, devfn, where);
76 if (!addr) {
77 *val = ~0;
78 return PCIBIOS_DEVICE_NOT_FOUND;
79 }
80
81 if (size == 1)
82 *val = readb(addr);
83 else if (size == 2)
84 *val = readw(addr);
85 else
86 *val = readl(addr);
87
88 return PCIBIOS_SUCCESSFUL;
89 }
90 EXPORT_SYMBOL_GPL(pci_generic_config_read);
91
pci_generic_config_write(struct pci_bus * bus,unsigned int devfn,int where,int size,u32 val)92 int pci_generic_config_write(struct pci_bus *bus, unsigned int devfn,
93 int where, int size, u32 val)
94 {
95 void __iomem *addr;
96
97 addr = bus->ops->map_bus(bus, devfn, where);
98 if (!addr)
99 return PCIBIOS_DEVICE_NOT_FOUND;
100
101 if (size == 1)
102 writeb(val, addr);
103 else if (size == 2)
104 writew(val, addr);
105 else
106 writel(val, addr);
107
108 return PCIBIOS_SUCCESSFUL;
109 }
110 EXPORT_SYMBOL_GPL(pci_generic_config_write);
111
pci_generic_config_read32(struct pci_bus * bus,unsigned int devfn,int where,int size,u32 * val)112 int pci_generic_config_read32(struct pci_bus *bus, unsigned int devfn,
113 int where, int size, u32 *val)
114 {
115 void __iomem *addr;
116
117 addr = bus->ops->map_bus(bus, devfn, where & ~0x3);
118 if (!addr) {
119 *val = ~0;
120 return PCIBIOS_DEVICE_NOT_FOUND;
121 }
122
123 *val = readl(addr);
124
125 if (size <= 2)
126 *val = (*val >> (8 * (where & 3))) & ((1 << (size * 8)) - 1);
127
128 return PCIBIOS_SUCCESSFUL;
129 }
130 EXPORT_SYMBOL_GPL(pci_generic_config_read32);
131
pci_generic_config_write32(struct pci_bus * bus,unsigned int devfn,int where,int size,u32 val)132 int pci_generic_config_write32(struct pci_bus *bus, unsigned int devfn,
133 int where, int size, u32 val)
134 {
135 void __iomem *addr;
136 u32 mask, tmp;
137
138 addr = bus->ops->map_bus(bus, devfn, where & ~0x3);
139 if (!addr)
140 return PCIBIOS_DEVICE_NOT_FOUND;
141
142 if (size == 4) {
143 writel(val, addr);
144 return PCIBIOS_SUCCESSFUL;
145 } else {
146 mask = ~(((1 << (size * 8)) - 1) << ((where & 0x3) * 8));
147 }
148
149 tmp = readl(addr) & mask;
150 tmp |= val << ((where & 0x3) * 8);
151 writel(tmp, addr);
152
153 return PCIBIOS_SUCCESSFUL;
154 }
155 EXPORT_SYMBOL_GPL(pci_generic_config_write32);
156
157 /**
158 * pci_bus_set_ops - Set raw operations of pci bus
159 * @bus: pci bus struct
160 * @ops: new raw operations
161 *
162 * Return previous raw operations
163 */
pci_bus_set_ops(struct pci_bus * bus,struct pci_ops * ops)164 struct pci_ops *pci_bus_set_ops(struct pci_bus *bus, struct pci_ops *ops)
165 {
166 struct pci_ops *old_ops;
167 unsigned long flags;
168
169 raw_spin_lock_irqsave(&pci_lock, flags);
170 old_ops = bus->ops;
171 bus->ops = ops;
172 raw_spin_unlock_irqrestore(&pci_lock, flags);
173 return old_ops;
174 }
175 EXPORT_SYMBOL(pci_bus_set_ops);
176
177 /**
178 * pci_read_vpd - Read one entry from Vital Product Data
179 * @dev: pci device struct
180 * @pos: offset in vpd space
181 * @count: number of bytes to read
182 * @buf: pointer to where to store result
183 *
184 */
pci_read_vpd(struct pci_dev * dev,loff_t pos,size_t count,void * buf)185 ssize_t pci_read_vpd(struct pci_dev *dev, loff_t pos, size_t count, void *buf)
186 {
187 if (!dev->vpd || !dev->vpd->ops)
188 return -ENODEV;
189 return dev->vpd->ops->read(dev, pos, count, buf);
190 }
191 EXPORT_SYMBOL(pci_read_vpd);
192
193 /**
194 * pci_write_vpd - Write entry to Vital Product Data
195 * @dev: pci device struct
196 * @pos: offset in vpd space
197 * @count: number of bytes to write
198 * @buf: buffer containing write data
199 *
200 */
pci_write_vpd(struct pci_dev * dev,loff_t pos,size_t count,const void * buf)201 ssize_t pci_write_vpd(struct pci_dev *dev, loff_t pos, size_t count, const void *buf)
202 {
203 if (!dev->vpd || !dev->vpd->ops)
204 return -ENODEV;
205 return dev->vpd->ops->write(dev, pos, count, buf);
206 }
207 EXPORT_SYMBOL(pci_write_vpd);
208
209 /*
210 * The following routines are to prevent the user from accessing PCI config
211 * space when it's unsafe to do so. Some devices require this during BIST and
212 * we're required to prevent it during D-state transitions.
213 *
214 * We have a bit per device to indicate it's blocked and a global wait queue
215 * for callers to sleep on until devices are unblocked.
216 */
217 static DECLARE_WAIT_QUEUE_HEAD(pci_cfg_wait);
218
pci_wait_cfg(struct pci_dev * dev)219 static noinline void pci_wait_cfg(struct pci_dev *dev)
220 {
221 DECLARE_WAITQUEUE(wait, current);
222
223 __add_wait_queue(&pci_cfg_wait, &wait);
224 do {
225 set_current_state(TASK_UNINTERRUPTIBLE);
226 raw_spin_unlock_irq(&pci_lock);
227 schedule();
228 raw_spin_lock_irq(&pci_lock);
229 } while (dev->block_cfg_access);
230 __remove_wait_queue(&pci_cfg_wait, &wait);
231 }
232
233 /* Returns 0 on success, negative values indicate error. */
234 #define PCI_USER_READ_CONFIG(size,type) \
235 int pci_user_read_config_##size \
236 (struct pci_dev *dev, int pos, type *val) \
237 { \
238 int ret = PCIBIOS_SUCCESSFUL; \
239 u32 data = -1; \
240 if (PCI_##size##_BAD) \
241 return -EINVAL; \
242 raw_spin_lock_irq(&pci_lock); \
243 if (unlikely(dev->block_cfg_access)) \
244 pci_wait_cfg(dev); \
245 ret = dev->bus->ops->read(dev->bus, dev->devfn, \
246 pos, sizeof(type), &data); \
247 raw_spin_unlock_irq(&pci_lock); \
248 *val = (type)data; \
249 return pcibios_err_to_errno(ret); \
250 } \
251 EXPORT_SYMBOL_GPL(pci_user_read_config_##size);
252
253 /* Returns 0 on success, negative values indicate error. */
254 #define PCI_USER_WRITE_CONFIG(size,type) \
255 int pci_user_write_config_##size \
256 (struct pci_dev *dev, int pos, type val) \
257 { \
258 int ret = PCIBIOS_SUCCESSFUL; \
259 if (PCI_##size##_BAD) \
260 return -EINVAL; \
261 raw_spin_lock_irq(&pci_lock); \
262 if (unlikely(dev->block_cfg_access)) \
263 pci_wait_cfg(dev); \
264 ret = dev->bus->ops->write(dev->bus, dev->devfn, \
265 pos, sizeof(type), val); \
266 raw_spin_unlock_irq(&pci_lock); \
267 return pcibios_err_to_errno(ret); \
268 } \
269 EXPORT_SYMBOL_GPL(pci_user_write_config_##size);
270
271 PCI_USER_READ_CONFIG(byte, u8)
272 PCI_USER_READ_CONFIG(word, u16)
273 PCI_USER_READ_CONFIG(dword, u32)
274 PCI_USER_WRITE_CONFIG(byte, u8)
275 PCI_USER_WRITE_CONFIG(word, u16)
276 PCI_USER_WRITE_CONFIG(dword, u32)
277
278 /* VPD access through PCI 2.2+ VPD capability */
279
280 #define PCI_VPD_PCI22_SIZE (PCI_VPD_ADDR_MASK + 1)
281
282 struct pci_vpd_pci22 {
283 struct pci_vpd base;
284 struct mutex lock;
285 u16 flag;
286 bool busy;
287 u8 cap;
288 };
289
290 /*
291 * Wait for last operation to complete.
292 * This code has to spin since there is no other notification from the PCI
293 * hardware. Since the VPD is often implemented by serial attachment to an
294 * EEPROM, it may take many milliseconds to complete.
295 *
296 * Returns 0 on success, negative values indicate error.
297 */
pci_vpd_pci22_wait(struct pci_dev * dev)298 static int pci_vpd_pci22_wait(struct pci_dev *dev)
299 {
300 struct pci_vpd_pci22 *vpd =
301 container_of(dev->vpd, struct pci_vpd_pci22, base);
302 unsigned long timeout = jiffies + HZ/20 + 2;
303 u16 status;
304 int ret;
305
306 if (!vpd->busy)
307 return 0;
308
309 for (;;) {
310 ret = pci_user_read_config_word(dev, vpd->cap + PCI_VPD_ADDR,
311 &status);
312 if (ret < 0)
313 return ret;
314
315 if ((status & PCI_VPD_ADDR_F) == vpd->flag) {
316 vpd->busy = false;
317 return 0;
318 }
319
320 if (time_after(jiffies, timeout)) {
321 dev_printk(KERN_DEBUG, &dev->dev, "vpd r/w failed. This is likely a firmware bug on this device. Contact the card vendor for a firmware update\n");
322 return -ETIMEDOUT;
323 }
324 if (fatal_signal_pending(current))
325 return -EINTR;
326 if (!cond_resched())
327 udelay(10);
328 }
329 }
330
pci_vpd_pci22_read(struct pci_dev * dev,loff_t pos,size_t count,void * arg)331 static ssize_t pci_vpd_pci22_read(struct pci_dev *dev, loff_t pos, size_t count,
332 void *arg)
333 {
334 struct pci_vpd_pci22 *vpd =
335 container_of(dev->vpd, struct pci_vpd_pci22, base);
336 int ret;
337 loff_t end = pos + count;
338 u8 *buf = arg;
339
340 if (pos < 0 || pos > vpd->base.len || end > vpd->base.len)
341 return -EINVAL;
342
343 if (mutex_lock_killable(&vpd->lock))
344 return -EINTR;
345
346 ret = pci_vpd_pci22_wait(dev);
347 if (ret < 0)
348 goto out;
349
350 while (pos < end) {
351 u32 val;
352 unsigned int i, skip;
353
354 ret = pci_user_write_config_word(dev, vpd->cap + PCI_VPD_ADDR,
355 pos & ~3);
356 if (ret < 0)
357 break;
358 vpd->busy = true;
359 vpd->flag = PCI_VPD_ADDR_F;
360 ret = pci_vpd_pci22_wait(dev);
361 if (ret < 0)
362 break;
363
364 ret = pci_user_read_config_dword(dev, vpd->cap + PCI_VPD_DATA, &val);
365 if (ret < 0)
366 break;
367
368 skip = pos & 3;
369 for (i = 0; i < sizeof(u32); i++) {
370 if (i >= skip) {
371 *buf++ = val;
372 if (++pos == end)
373 break;
374 }
375 val >>= 8;
376 }
377 }
378 out:
379 mutex_unlock(&vpd->lock);
380 return ret ? ret : count;
381 }
382
pci_vpd_pci22_write(struct pci_dev * dev,loff_t pos,size_t count,const void * arg)383 static ssize_t pci_vpd_pci22_write(struct pci_dev *dev, loff_t pos, size_t count,
384 const void *arg)
385 {
386 struct pci_vpd_pci22 *vpd =
387 container_of(dev->vpd, struct pci_vpd_pci22, base);
388 const u8 *buf = arg;
389 loff_t end = pos + count;
390 int ret = 0;
391
392 if (pos < 0 || (pos & 3) || (count & 3) || end > vpd->base.len)
393 return -EINVAL;
394
395 if (mutex_lock_killable(&vpd->lock))
396 return -EINTR;
397
398 ret = pci_vpd_pci22_wait(dev);
399 if (ret < 0)
400 goto out;
401
402 while (pos < end) {
403 u32 val;
404
405 val = *buf++;
406 val |= *buf++ << 8;
407 val |= *buf++ << 16;
408 val |= *buf++ << 24;
409
410 ret = pci_user_write_config_dword(dev, vpd->cap + PCI_VPD_DATA, val);
411 if (ret < 0)
412 break;
413 ret = pci_user_write_config_word(dev, vpd->cap + PCI_VPD_ADDR,
414 pos | PCI_VPD_ADDR_F);
415 if (ret < 0)
416 break;
417
418 vpd->busy = true;
419 vpd->flag = 0;
420 ret = pci_vpd_pci22_wait(dev);
421 if (ret < 0)
422 break;
423
424 pos += sizeof(u32);
425 }
426 out:
427 mutex_unlock(&vpd->lock);
428 return ret ? ret : count;
429 }
430
pci_vpd_pci22_release(struct pci_dev * dev)431 static void pci_vpd_pci22_release(struct pci_dev *dev)
432 {
433 kfree(container_of(dev->vpd, struct pci_vpd_pci22, base));
434 }
435
436 static const struct pci_vpd_ops pci_vpd_pci22_ops = {
437 .read = pci_vpd_pci22_read,
438 .write = pci_vpd_pci22_write,
439 .release = pci_vpd_pci22_release,
440 };
441
pci_vpd_f0_read(struct pci_dev * dev,loff_t pos,size_t count,void * arg)442 static ssize_t pci_vpd_f0_read(struct pci_dev *dev, loff_t pos, size_t count,
443 void *arg)
444 {
445 struct pci_dev *tdev = pci_get_slot(dev->bus,
446 PCI_DEVFN(PCI_SLOT(dev->devfn), 0));
447 ssize_t ret;
448
449 if (!tdev)
450 return -ENODEV;
451
452 ret = pci_read_vpd(tdev, pos, count, arg);
453 pci_dev_put(tdev);
454 return ret;
455 }
456
pci_vpd_f0_write(struct pci_dev * dev,loff_t pos,size_t count,const void * arg)457 static ssize_t pci_vpd_f0_write(struct pci_dev *dev, loff_t pos, size_t count,
458 const void *arg)
459 {
460 struct pci_dev *tdev = pci_get_slot(dev->bus,
461 PCI_DEVFN(PCI_SLOT(dev->devfn), 0));
462 ssize_t ret;
463
464 if (!tdev)
465 return -ENODEV;
466
467 ret = pci_write_vpd(tdev, pos, count, arg);
468 pci_dev_put(tdev);
469 return ret;
470 }
471
472 static const struct pci_vpd_ops pci_vpd_f0_ops = {
473 .read = pci_vpd_f0_read,
474 .write = pci_vpd_f0_write,
475 .release = pci_vpd_pci22_release,
476 };
477
pci_vpd_pci22_init(struct pci_dev * dev)478 int pci_vpd_pci22_init(struct pci_dev *dev)
479 {
480 struct pci_vpd_pci22 *vpd;
481 u8 cap;
482
483 cap = pci_find_capability(dev, PCI_CAP_ID_VPD);
484 if (!cap)
485 return -ENODEV;
486
487 vpd = kzalloc(sizeof(*vpd), GFP_ATOMIC);
488 if (!vpd)
489 return -ENOMEM;
490
491 vpd->base.len = PCI_VPD_PCI22_SIZE;
492 if (dev->dev_flags & PCI_DEV_FLAGS_VPD_REF_F0)
493 vpd->base.ops = &pci_vpd_f0_ops;
494 else
495 vpd->base.ops = &pci_vpd_pci22_ops;
496 mutex_init(&vpd->lock);
497 vpd->cap = cap;
498 vpd->busy = false;
499 dev->vpd = &vpd->base;
500 return 0;
501 }
502
503 /**
504 * pci_cfg_access_lock - Lock PCI config reads/writes
505 * @dev: pci device struct
506 *
507 * When access is locked, any userspace reads or writes to config
508 * space and concurrent lock requests will sleep until access is
509 * allowed via pci_cfg_access_unlocked again.
510 */
pci_cfg_access_lock(struct pci_dev * dev)511 void pci_cfg_access_lock(struct pci_dev *dev)
512 {
513 might_sleep();
514
515 raw_spin_lock_irq(&pci_lock);
516 if (dev->block_cfg_access)
517 pci_wait_cfg(dev);
518 dev->block_cfg_access = 1;
519 raw_spin_unlock_irq(&pci_lock);
520 }
521 EXPORT_SYMBOL_GPL(pci_cfg_access_lock);
522
523 /**
524 * pci_cfg_access_trylock - try to lock PCI config reads/writes
525 * @dev: pci device struct
526 *
527 * Same as pci_cfg_access_lock, but will return 0 if access is
528 * already locked, 1 otherwise. This function can be used from
529 * atomic contexts.
530 */
pci_cfg_access_trylock(struct pci_dev * dev)531 bool pci_cfg_access_trylock(struct pci_dev *dev)
532 {
533 unsigned long flags;
534 bool locked = true;
535
536 raw_spin_lock_irqsave(&pci_lock, flags);
537 if (dev->block_cfg_access)
538 locked = false;
539 else
540 dev->block_cfg_access = 1;
541 raw_spin_unlock_irqrestore(&pci_lock, flags);
542
543 return locked;
544 }
545 EXPORT_SYMBOL_GPL(pci_cfg_access_trylock);
546
547 /**
548 * pci_cfg_access_unlock - Unlock PCI config reads/writes
549 * @dev: pci device struct
550 *
551 * This function allows PCI config accesses to resume.
552 */
pci_cfg_access_unlock(struct pci_dev * dev)553 void pci_cfg_access_unlock(struct pci_dev *dev)
554 {
555 unsigned long flags;
556
557 raw_spin_lock_irqsave(&pci_lock, flags);
558
559 /* This indicates a problem in the caller, but we don't need
560 * to kill them, unlike a double-block above. */
561 WARN_ON(!dev->block_cfg_access);
562
563 dev->block_cfg_access = 0;
564 wake_up_all(&pci_cfg_wait);
565 raw_spin_unlock_irqrestore(&pci_lock, flags);
566 }
567 EXPORT_SYMBOL_GPL(pci_cfg_access_unlock);
568
pcie_cap_version(const struct pci_dev * dev)569 static inline int pcie_cap_version(const struct pci_dev *dev)
570 {
571 return pcie_caps_reg(dev) & PCI_EXP_FLAGS_VERS;
572 }
573
pcie_cap_has_lnkctl(const struct pci_dev * dev)574 bool pcie_cap_has_lnkctl(const struct pci_dev *dev)
575 {
576 int type = pci_pcie_type(dev);
577
578 return type == PCI_EXP_TYPE_ENDPOINT ||
579 type == PCI_EXP_TYPE_LEG_END ||
580 type == PCI_EXP_TYPE_ROOT_PORT ||
581 type == PCI_EXP_TYPE_UPSTREAM ||
582 type == PCI_EXP_TYPE_DOWNSTREAM ||
583 type == PCI_EXP_TYPE_PCI_BRIDGE ||
584 type == PCI_EXP_TYPE_PCIE_BRIDGE;
585 }
586
pcie_cap_has_sltctl(const struct pci_dev * dev)587 static inline bool pcie_cap_has_sltctl(const struct pci_dev *dev)
588 {
589 int type = pci_pcie_type(dev);
590
591 return (type == PCI_EXP_TYPE_ROOT_PORT ||
592 type == PCI_EXP_TYPE_DOWNSTREAM) &&
593 pcie_caps_reg(dev) & PCI_EXP_FLAGS_SLOT;
594 }
595
pcie_cap_has_rtctl(const struct pci_dev * dev)596 static inline bool pcie_cap_has_rtctl(const struct pci_dev *dev)
597 {
598 int type = pci_pcie_type(dev);
599
600 return type == PCI_EXP_TYPE_ROOT_PORT ||
601 type == PCI_EXP_TYPE_RC_EC;
602 }
603
pcie_capability_reg_implemented(struct pci_dev * dev,int pos)604 static bool pcie_capability_reg_implemented(struct pci_dev *dev, int pos)
605 {
606 if (!pci_is_pcie(dev))
607 return false;
608
609 switch (pos) {
610 case PCI_EXP_FLAGS:
611 return true;
612 case PCI_EXP_DEVCAP:
613 case PCI_EXP_DEVCTL:
614 case PCI_EXP_DEVSTA:
615 return true;
616 case PCI_EXP_LNKCAP:
617 case PCI_EXP_LNKCTL:
618 case PCI_EXP_LNKSTA:
619 return pcie_cap_has_lnkctl(dev);
620 case PCI_EXP_SLTCAP:
621 case PCI_EXP_SLTCTL:
622 case PCI_EXP_SLTSTA:
623 return pcie_cap_has_sltctl(dev);
624 case PCI_EXP_RTCTL:
625 case PCI_EXP_RTCAP:
626 case PCI_EXP_RTSTA:
627 return pcie_cap_has_rtctl(dev);
628 case PCI_EXP_DEVCAP2:
629 case PCI_EXP_DEVCTL2:
630 case PCI_EXP_LNKCAP2:
631 case PCI_EXP_LNKCTL2:
632 case PCI_EXP_LNKSTA2:
633 return pcie_cap_version(dev) > 1;
634 default:
635 return false;
636 }
637 }
638
639 /*
640 * Note that these accessor functions are only for the "PCI Express
641 * Capability" (see PCIe spec r3.0, sec 7.8). They do not apply to the
642 * other "PCI Express Extended Capabilities" (AER, VC, ACS, MFVC, etc.)
643 */
pcie_capability_read_word(struct pci_dev * dev,int pos,u16 * val)644 int pcie_capability_read_word(struct pci_dev *dev, int pos, u16 *val)
645 {
646 int ret;
647
648 *val = 0;
649 if (pos & 1)
650 return -EINVAL;
651
652 if (pcie_capability_reg_implemented(dev, pos)) {
653 ret = pci_read_config_word(dev, pci_pcie_cap(dev) + pos, val);
654 /*
655 * Reset *val to 0 if pci_read_config_word() fails, it may
656 * have been written as 0xFFFF if hardware error happens
657 * during pci_read_config_word().
658 */
659 if (ret)
660 *val = 0;
661 return ret;
662 }
663
664 /*
665 * For Functions that do not implement the Slot Capabilities,
666 * Slot Status, and Slot Control registers, these spaces must
667 * be hardwired to 0b, with the exception of the Presence Detect
668 * State bit in the Slot Status register of Downstream Ports,
669 * which must be hardwired to 1b. (PCIe Base Spec 3.0, sec 7.8)
670 */
671 if (pci_is_pcie(dev) && pos == PCI_EXP_SLTSTA &&
672 pci_pcie_type(dev) == PCI_EXP_TYPE_DOWNSTREAM) {
673 *val = PCI_EXP_SLTSTA_PDS;
674 }
675
676 return 0;
677 }
678 EXPORT_SYMBOL(pcie_capability_read_word);
679
pcie_capability_read_dword(struct pci_dev * dev,int pos,u32 * val)680 int pcie_capability_read_dword(struct pci_dev *dev, int pos, u32 *val)
681 {
682 int ret;
683
684 *val = 0;
685 if (pos & 3)
686 return -EINVAL;
687
688 if (pcie_capability_reg_implemented(dev, pos)) {
689 ret = pci_read_config_dword(dev, pci_pcie_cap(dev) + pos, val);
690 /*
691 * Reset *val to 0 if pci_read_config_dword() fails, it may
692 * have been written as 0xFFFFFFFF if hardware error happens
693 * during pci_read_config_dword().
694 */
695 if (ret)
696 *val = 0;
697 return ret;
698 }
699
700 if (pci_is_pcie(dev) && pos == PCI_EXP_SLTCTL &&
701 pci_pcie_type(dev) == PCI_EXP_TYPE_DOWNSTREAM) {
702 *val = PCI_EXP_SLTSTA_PDS;
703 }
704
705 return 0;
706 }
707 EXPORT_SYMBOL(pcie_capability_read_dword);
708
pcie_capability_write_word(struct pci_dev * dev,int pos,u16 val)709 int pcie_capability_write_word(struct pci_dev *dev, int pos, u16 val)
710 {
711 if (pos & 1)
712 return -EINVAL;
713
714 if (!pcie_capability_reg_implemented(dev, pos))
715 return 0;
716
717 return pci_write_config_word(dev, pci_pcie_cap(dev) + pos, val);
718 }
719 EXPORT_SYMBOL(pcie_capability_write_word);
720
pcie_capability_write_dword(struct pci_dev * dev,int pos,u32 val)721 int pcie_capability_write_dword(struct pci_dev *dev, int pos, u32 val)
722 {
723 if (pos & 3)
724 return -EINVAL;
725
726 if (!pcie_capability_reg_implemented(dev, pos))
727 return 0;
728
729 return pci_write_config_dword(dev, pci_pcie_cap(dev) + pos, val);
730 }
731 EXPORT_SYMBOL(pcie_capability_write_dword);
732
pcie_capability_clear_and_set_word(struct pci_dev * dev,int pos,u16 clear,u16 set)733 int pcie_capability_clear_and_set_word(struct pci_dev *dev, int pos,
734 u16 clear, u16 set)
735 {
736 int ret;
737 u16 val;
738
739 ret = pcie_capability_read_word(dev, pos, &val);
740 if (!ret) {
741 val &= ~clear;
742 val |= set;
743 ret = pcie_capability_write_word(dev, pos, val);
744 }
745
746 return ret;
747 }
748 EXPORT_SYMBOL(pcie_capability_clear_and_set_word);
749
pcie_capability_clear_and_set_dword(struct pci_dev * dev,int pos,u32 clear,u32 set)750 int pcie_capability_clear_and_set_dword(struct pci_dev *dev, int pos,
751 u32 clear, u32 set)
752 {
753 int ret;
754 u32 val;
755
756 ret = pcie_capability_read_dword(dev, pos, &val);
757 if (!ret) {
758 val &= ~clear;
759 val |= set;
760 ret = pcie_capability_write_dword(dev, pos, val);
761 }
762
763 return ret;
764 }
765 EXPORT_SYMBOL(pcie_capability_clear_and_set_dword);
766