GNU Linux-libre 4.14.332-gnu1
[releases.git] / drivers / pci / pci.c
1 /*
2  *      PCI Bus Services, see include/linux/pci.h for further explanation.
3  *
4  *      Copyright 1993 -- 1997 Drew Eckhardt, Frederic Potter,
5  *      David Mosberger-Tang
6  *
7  *      Copyright 1997 -- 2000 Martin Mares <mj@ucw.cz>
8  */
9
10 #include <linux/acpi.h>
11 #include <linux/kernel.h>
12 #include <linux/delay.h>
13 #include <linux/dmi.h>
14 #include <linux/init.h>
15 #include <linux/of.h>
16 #include <linux/of_pci.h>
17 #include <linux/pci.h>
18 #include <linux/pm.h>
19 #include <linux/slab.h>
20 #include <linux/module.h>
21 #include <linux/spinlock.h>
22 #include <linux/string.h>
23 #include <linux/log2.h>
24 #include <linux/pci-aspm.h>
25 #include <linux/pm_wakeup.h>
26 #include <linux/interrupt.h>
27 #include <linux/device.h>
28 #include <linux/pm_runtime.h>
29 #include <linux/pci_hotplug.h>
30 #include <linux/vmalloc.h>
31 #include <linux/pci-ats.h>
32 #include <asm/setup.h>
33 #include <asm/dma.h>
34 #include <linux/aer.h>
35 #include "pci.h"
36
37 const char *pci_power_names[] = {
38         "error", "D0", "D1", "D2", "D3hot", "D3cold", "unknown",
39 };
40 EXPORT_SYMBOL_GPL(pci_power_names);
41
42 int isa_dma_bridge_buggy;
43 EXPORT_SYMBOL(isa_dma_bridge_buggy);
44
45 int pci_pci_problems;
46 EXPORT_SYMBOL(pci_pci_problems);
47
48 unsigned int pci_pm_d3_delay;
49
50 static void pci_pme_list_scan(struct work_struct *work);
51
52 static LIST_HEAD(pci_pme_list);
53 static DEFINE_MUTEX(pci_pme_list_mutex);
54 static DECLARE_DELAYED_WORK(pci_pme_work, pci_pme_list_scan);
55
56 struct pci_pme_device {
57         struct list_head list;
58         struct pci_dev *dev;
59 };
60
61 #define PME_TIMEOUT 1000 /* How long between PME checks */
62
63 static void pci_dev_d3_sleep(struct pci_dev *dev)
64 {
65         unsigned int delay = dev->d3_delay;
66
67         if (delay < pci_pm_d3_delay)
68                 delay = pci_pm_d3_delay;
69
70         if (delay)
71                 msleep(delay);
72 }
73
74 #ifdef CONFIG_PCI_DOMAINS
75 int pci_domains_supported = 1;
76 #endif
77
78 #define DEFAULT_CARDBUS_IO_SIZE         (256)
79 #define DEFAULT_CARDBUS_MEM_SIZE        (64*1024*1024)
80 /* pci=cbmemsize=nnM,cbiosize=nn can override this */
81 unsigned long pci_cardbus_io_size = DEFAULT_CARDBUS_IO_SIZE;
82 unsigned long pci_cardbus_mem_size = DEFAULT_CARDBUS_MEM_SIZE;
83
84 #define DEFAULT_HOTPLUG_IO_SIZE         (256)
85 #define DEFAULT_HOTPLUG_MEM_SIZE        (2*1024*1024)
86 /* pci=hpmemsize=nnM,hpiosize=nn can override this */
87 unsigned long pci_hotplug_io_size  = DEFAULT_HOTPLUG_IO_SIZE;
88 unsigned long pci_hotplug_mem_size = DEFAULT_HOTPLUG_MEM_SIZE;
89
90 #define DEFAULT_HOTPLUG_BUS_SIZE        1
91 unsigned long pci_hotplug_bus_size = DEFAULT_HOTPLUG_BUS_SIZE;
92
93 enum pcie_bus_config_types pcie_bus_config = PCIE_BUS_DEFAULT;
94
95 /*
96  * The default CLS is used if arch didn't set CLS explicitly and not
97  * all pci devices agree on the same value.  Arch can override either
98  * the dfl or actual value as it sees fit.  Don't forget this is
99  * measured in 32-bit words, not bytes.
100  */
101 u8 pci_dfl_cache_line_size = L1_CACHE_BYTES >> 2;
102 u8 pci_cache_line_size;
103
104 /*
105  * If we set up a device for bus mastering, we need to check the latency
106  * timer as certain BIOSes forget to set it properly.
107  */
108 unsigned int pcibios_max_latency = 255;
109
110 /* If set, the PCIe ARI capability will not be used. */
111 static bool pcie_ari_disabled;
112
113 /* Disable bridge_d3 for all PCIe ports */
114 static bool pci_bridge_d3_disable;
115 /* Force bridge_d3 for all PCIe ports */
116 static bool pci_bridge_d3_force;
117
118 static int __init pcie_port_pm_setup(char *str)
119 {
120         if (!strcmp(str, "off"))
121                 pci_bridge_d3_disable = true;
122         else if (!strcmp(str, "force"))
123                 pci_bridge_d3_force = true;
124         return 1;
125 }
126 __setup("pcie_port_pm=", pcie_port_pm_setup);
127
128 /**
129  * pci_bus_max_busnr - returns maximum PCI bus number of given bus' children
130  * @bus: pointer to PCI bus structure to search
131  *
132  * Given a PCI bus, returns the highest PCI bus number present in the set
133  * including the given PCI bus and its list of child PCI buses.
134  */
135 unsigned char pci_bus_max_busnr(struct pci_bus *bus)
136 {
137         struct pci_bus *tmp;
138         unsigned char max, n;
139
140         max = bus->busn_res.end;
141         list_for_each_entry(tmp, &bus->children, node) {
142                 n = pci_bus_max_busnr(tmp);
143                 if (n > max)
144                         max = n;
145         }
146         return max;
147 }
148 EXPORT_SYMBOL_GPL(pci_bus_max_busnr);
149
150 #ifdef CONFIG_HAS_IOMEM
151 void __iomem *pci_ioremap_bar(struct pci_dev *pdev, int bar)
152 {
153         struct resource *res = &pdev->resource[bar];
154
155         /*
156          * Make sure the BAR is actually a memory resource, not an IO resource
157          */
158         if (res->flags & IORESOURCE_UNSET || !(res->flags & IORESOURCE_MEM)) {
159                 dev_warn(&pdev->dev, "can't ioremap BAR %d: %pR\n", bar, res);
160                 return NULL;
161         }
162         return ioremap_nocache(res->start, resource_size(res));
163 }
164 EXPORT_SYMBOL_GPL(pci_ioremap_bar);
165
166 void __iomem *pci_ioremap_wc_bar(struct pci_dev *pdev, int bar)
167 {
168         /*
169          * Make sure the BAR is actually a memory resource, not an IO resource
170          */
171         if (!(pci_resource_flags(pdev, bar) & IORESOURCE_MEM)) {
172                 WARN_ON(1);
173                 return NULL;
174         }
175         return ioremap_wc(pci_resource_start(pdev, bar),
176                           pci_resource_len(pdev, bar));
177 }
178 EXPORT_SYMBOL_GPL(pci_ioremap_wc_bar);
179 #endif
180
181
182 static int __pci_find_next_cap_ttl(struct pci_bus *bus, unsigned int devfn,
183                                    u8 pos, int cap, int *ttl)
184 {
185         u8 id;
186         u16 ent;
187
188         pci_bus_read_config_byte(bus, devfn, pos, &pos);
189
190         while ((*ttl)--) {
191                 if (pos < 0x40)
192                         break;
193                 pos &= ~3;
194                 pci_bus_read_config_word(bus, devfn, pos, &ent);
195
196                 id = ent & 0xff;
197                 if (id == 0xff)
198                         break;
199                 if (id == cap)
200                         return pos;
201                 pos = (ent >> 8);
202         }
203         return 0;
204 }
205
206 static int __pci_find_next_cap(struct pci_bus *bus, unsigned int devfn,
207                                u8 pos, int cap)
208 {
209         int ttl = PCI_FIND_CAP_TTL;
210
211         return __pci_find_next_cap_ttl(bus, devfn, pos, cap, &ttl);
212 }
213
214 int pci_find_next_capability(struct pci_dev *dev, u8 pos, int cap)
215 {
216         return __pci_find_next_cap(dev->bus, dev->devfn,
217                                    pos + PCI_CAP_LIST_NEXT, cap);
218 }
219 EXPORT_SYMBOL_GPL(pci_find_next_capability);
220
221 static int __pci_bus_find_cap_start(struct pci_bus *bus,
222                                     unsigned int devfn, u8 hdr_type)
223 {
224         u16 status;
225
226         pci_bus_read_config_word(bus, devfn, PCI_STATUS, &status);
227         if (!(status & PCI_STATUS_CAP_LIST))
228                 return 0;
229
230         switch (hdr_type) {
231         case PCI_HEADER_TYPE_NORMAL:
232         case PCI_HEADER_TYPE_BRIDGE:
233                 return PCI_CAPABILITY_LIST;
234         case PCI_HEADER_TYPE_CARDBUS:
235                 return PCI_CB_CAPABILITY_LIST;
236         }
237
238         return 0;
239 }
240
241 /**
242  * pci_find_capability - query for devices' capabilities
243  * @dev: PCI device to query
244  * @cap: capability code
245  *
246  * Tell if a device supports a given PCI capability.
247  * Returns the address of the requested capability structure within the
248  * device's PCI configuration space or 0 in case the device does not
249  * support it.  Possible values for @cap:
250  *
251  *  %PCI_CAP_ID_PM           Power Management
252  *  %PCI_CAP_ID_AGP          Accelerated Graphics Port
253  *  %PCI_CAP_ID_VPD          Vital Product Data
254  *  %PCI_CAP_ID_SLOTID       Slot Identification
255  *  %PCI_CAP_ID_MSI          Message Signalled Interrupts
256  *  %PCI_CAP_ID_CHSWP        CompactPCI HotSwap
257  *  %PCI_CAP_ID_PCIX         PCI-X
258  *  %PCI_CAP_ID_EXP          PCI Express
259  */
260 int pci_find_capability(struct pci_dev *dev, int cap)
261 {
262         int pos;
263
264         pos = __pci_bus_find_cap_start(dev->bus, dev->devfn, dev->hdr_type);
265         if (pos)
266                 pos = __pci_find_next_cap(dev->bus, dev->devfn, pos, cap);
267
268         return pos;
269 }
270 EXPORT_SYMBOL(pci_find_capability);
271
272 /**
273  * pci_bus_find_capability - query for devices' capabilities
274  * @bus:   the PCI bus to query
275  * @devfn: PCI device to query
276  * @cap:   capability code
277  *
278  * Like pci_find_capability() but works for pci devices that do not have a
279  * pci_dev structure set up yet.
280  *
281  * Returns the address of the requested capability structure within the
282  * device's PCI configuration space or 0 in case the device does not
283  * support it.
284  */
285 int pci_bus_find_capability(struct pci_bus *bus, unsigned int devfn, int cap)
286 {
287         int pos;
288         u8 hdr_type;
289
290         pci_bus_read_config_byte(bus, devfn, PCI_HEADER_TYPE, &hdr_type);
291
292         pos = __pci_bus_find_cap_start(bus, devfn, hdr_type & 0x7f);
293         if (pos)
294                 pos = __pci_find_next_cap(bus, devfn, pos, cap);
295
296         return pos;
297 }
298 EXPORT_SYMBOL(pci_bus_find_capability);
299
300 /**
301  * pci_find_next_ext_capability - Find an extended capability
302  * @dev: PCI device to query
303  * @start: address at which to start looking (0 to start at beginning of list)
304  * @cap: capability code
305  *
306  * Returns the address of the next matching extended capability structure
307  * within the device's PCI configuration space or 0 if the device does
308  * not support it.  Some capabilities can occur several times, e.g., the
309  * vendor-specific capability, and this provides a way to find them all.
310  */
311 int pci_find_next_ext_capability(struct pci_dev *dev, int start, int cap)
312 {
313         u32 header;
314         int ttl;
315         int pos = PCI_CFG_SPACE_SIZE;
316
317         /* minimum 8 bytes per capability */
318         ttl = (PCI_CFG_SPACE_EXP_SIZE - PCI_CFG_SPACE_SIZE) / 8;
319
320         if (dev->cfg_size <= PCI_CFG_SPACE_SIZE)
321                 return 0;
322
323         if (start)
324                 pos = start;
325
326         if (pci_read_config_dword(dev, pos, &header) != PCIBIOS_SUCCESSFUL)
327                 return 0;
328
329         /*
330          * If we have no capabilities, this is indicated by cap ID,
331          * cap version and next pointer all being 0.
332          */
333         if (header == 0)
334                 return 0;
335
336         while (ttl-- > 0) {
337                 if (PCI_EXT_CAP_ID(header) == cap && pos != start)
338                         return pos;
339
340                 pos = PCI_EXT_CAP_NEXT(header);
341                 if (pos < PCI_CFG_SPACE_SIZE)
342                         break;
343
344                 if (pci_read_config_dword(dev, pos, &header) != PCIBIOS_SUCCESSFUL)
345                         break;
346         }
347
348         return 0;
349 }
350 EXPORT_SYMBOL_GPL(pci_find_next_ext_capability);
351
352 /**
353  * pci_find_ext_capability - Find an extended capability
354  * @dev: PCI device to query
355  * @cap: capability code
356  *
357  * Returns the address of the requested extended capability structure
358  * within the device's PCI configuration space or 0 if the device does
359  * not support it.  Possible values for @cap:
360  *
361  *  %PCI_EXT_CAP_ID_ERR         Advanced Error Reporting
362  *  %PCI_EXT_CAP_ID_VC          Virtual Channel
363  *  %PCI_EXT_CAP_ID_DSN         Device Serial Number
364  *  %PCI_EXT_CAP_ID_PWR         Power Budgeting
365  */
366 int pci_find_ext_capability(struct pci_dev *dev, int cap)
367 {
368         return pci_find_next_ext_capability(dev, 0, cap);
369 }
370 EXPORT_SYMBOL_GPL(pci_find_ext_capability);
371
372 static int __pci_find_next_ht_cap(struct pci_dev *dev, int pos, int ht_cap)
373 {
374         int rc, ttl = PCI_FIND_CAP_TTL;
375         u8 cap, mask;
376
377         if (ht_cap == HT_CAPTYPE_SLAVE || ht_cap == HT_CAPTYPE_HOST)
378                 mask = HT_3BIT_CAP_MASK;
379         else
380                 mask = HT_5BIT_CAP_MASK;
381
382         pos = __pci_find_next_cap_ttl(dev->bus, dev->devfn, pos,
383                                       PCI_CAP_ID_HT, &ttl);
384         while (pos) {
385                 rc = pci_read_config_byte(dev, pos + 3, &cap);
386                 if (rc != PCIBIOS_SUCCESSFUL)
387                         return 0;
388
389                 if ((cap & mask) == ht_cap)
390                         return pos;
391
392                 pos = __pci_find_next_cap_ttl(dev->bus, dev->devfn,
393                                               pos + PCI_CAP_LIST_NEXT,
394                                               PCI_CAP_ID_HT, &ttl);
395         }
396
397         return 0;
398 }
399 /**
400  * pci_find_next_ht_capability - query a device's Hypertransport capabilities
401  * @dev: PCI device to query
402  * @pos: Position from which to continue searching
403  * @ht_cap: Hypertransport capability code
404  *
405  * To be used in conjunction with pci_find_ht_capability() to search for
406  * all capabilities matching @ht_cap. @pos should always be a value returned
407  * from pci_find_ht_capability().
408  *
409  * NB. To be 100% safe against broken PCI devices, the caller should take
410  * steps to avoid an infinite loop.
411  */
412 int pci_find_next_ht_capability(struct pci_dev *dev, int pos, int ht_cap)
413 {
414         return __pci_find_next_ht_cap(dev, pos + PCI_CAP_LIST_NEXT, ht_cap);
415 }
416 EXPORT_SYMBOL_GPL(pci_find_next_ht_capability);
417
418 /**
419  * pci_find_ht_capability - query a device's Hypertransport capabilities
420  * @dev: PCI device to query
421  * @ht_cap: Hypertransport capability code
422  *
423  * Tell if a device supports a given Hypertransport capability.
424  * Returns an address within the device's PCI configuration space
425  * or 0 in case the device does not support the request capability.
426  * The address points to the PCI capability, of type PCI_CAP_ID_HT,
427  * which has a Hypertransport capability matching @ht_cap.
428  */
429 int pci_find_ht_capability(struct pci_dev *dev, int ht_cap)
430 {
431         int pos;
432
433         pos = __pci_bus_find_cap_start(dev->bus, dev->devfn, dev->hdr_type);
434         if (pos)
435                 pos = __pci_find_next_ht_cap(dev, pos, ht_cap);
436
437         return pos;
438 }
439 EXPORT_SYMBOL_GPL(pci_find_ht_capability);
440
441 /**
442  * pci_find_parent_resource - return resource region of parent bus of given region
443  * @dev: PCI device structure contains resources to be searched
444  * @res: child resource record for which parent is sought
445  *
446  *  For given resource region of given device, return the resource
447  *  region of parent bus the given region is contained in.
448  */
449 struct resource *pci_find_parent_resource(const struct pci_dev *dev,
450                                           struct resource *res)
451 {
452         const struct pci_bus *bus = dev->bus;
453         struct resource *r;
454         int i;
455
456         pci_bus_for_each_resource(bus, r, i) {
457                 if (!r)
458                         continue;
459                 if (resource_contains(r, res)) {
460
461                         /*
462                          * If the window is prefetchable but the BAR is
463                          * not, the allocator made a mistake.
464                          */
465                         if (r->flags & IORESOURCE_PREFETCH &&
466                             !(res->flags & IORESOURCE_PREFETCH))
467                                 return NULL;
468
469                         /*
470                          * If we're below a transparent bridge, there may
471                          * be both a positively-decoded aperture and a
472                          * subtractively-decoded region that contain the BAR.
473                          * We want the positively-decoded one, so this depends
474                          * on pci_bus_for_each_resource() giving us those
475                          * first.
476                          */
477                         return r;
478                 }
479         }
480         return NULL;
481 }
482 EXPORT_SYMBOL(pci_find_parent_resource);
483
484 /**
485  * pci_find_resource - Return matching PCI device resource
486  * @dev: PCI device to query
487  * @res: Resource to look for
488  *
489  * Goes over standard PCI resources (BARs) and checks if the given resource
490  * is partially or fully contained in any of them. In that case the
491  * matching resource is returned, %NULL otherwise.
492  */
493 struct resource *pci_find_resource(struct pci_dev *dev, struct resource *res)
494 {
495         int i;
496
497         for (i = 0; i < PCI_ROM_RESOURCE; i++) {
498                 struct resource *r = &dev->resource[i];
499
500                 if (r->start && resource_contains(r, res))
501                         return r;
502         }
503
504         return NULL;
505 }
506 EXPORT_SYMBOL(pci_find_resource);
507
508 /**
509  * pci_find_pcie_root_port - return PCIe Root Port
510  * @dev: PCI device to query
511  *
512  * Traverse up the parent chain and return the PCIe Root Port PCI Device
513  * for a given PCI Device.
514  */
515 struct pci_dev *pci_find_pcie_root_port(struct pci_dev *dev)
516 {
517         struct pci_dev *bridge, *highest_pcie_bridge = dev;
518
519         bridge = pci_upstream_bridge(dev);
520         while (bridge && pci_is_pcie(bridge)) {
521                 highest_pcie_bridge = bridge;
522                 bridge = pci_upstream_bridge(bridge);
523         }
524
525         if (pci_pcie_type(highest_pcie_bridge) != PCI_EXP_TYPE_ROOT_PORT)
526                 return NULL;
527
528         return highest_pcie_bridge;
529 }
530 EXPORT_SYMBOL(pci_find_pcie_root_port);
531
532 /**
533  * pci_wait_for_pending - wait for @mask bit(s) to clear in status word @pos
534  * @dev: the PCI device to operate on
535  * @pos: config space offset of status word
536  * @mask: mask of bit(s) to care about in status word
537  *
538  * Return 1 when mask bit(s) in status word clear, 0 otherwise.
539  */
540 int pci_wait_for_pending(struct pci_dev *dev, int pos, u16 mask)
541 {
542         int i;
543
544         /* Wait for Transaction Pending bit clean */
545         for (i = 0; i < 4; i++) {
546                 u16 status;
547                 if (i)
548                         msleep((1 << (i - 1)) * 100);
549
550                 pci_read_config_word(dev, pos, &status);
551                 if (!(status & mask))
552                         return 1;
553         }
554
555         return 0;
556 }
557
558 /**
559  * pci_restore_bars - restore a device's BAR values (e.g. after wake-up)
560  * @dev: PCI device to have its BARs restored
561  *
562  * Restore the BAR values for a given device, so as to make it
563  * accessible by its driver.
564  */
565 static void pci_restore_bars(struct pci_dev *dev)
566 {
567         int i;
568
569         for (i = 0; i < PCI_BRIDGE_RESOURCES; i++)
570                 pci_update_resource(dev, i);
571 }
572
573 static const struct pci_platform_pm_ops *pci_platform_pm;
574
575 int pci_set_platform_pm(const struct pci_platform_pm_ops *ops)
576 {
577         if (!ops->is_manageable || !ops->set_state  || !ops->get_state ||
578             !ops->choose_state  || !ops->set_wakeup || !ops->need_resume)
579                 return -EINVAL;
580         pci_platform_pm = ops;
581         return 0;
582 }
583
584 static inline bool platform_pci_power_manageable(struct pci_dev *dev)
585 {
586         return pci_platform_pm ? pci_platform_pm->is_manageable(dev) : false;
587 }
588
589 static inline int platform_pci_set_power_state(struct pci_dev *dev,
590                                                pci_power_t t)
591 {
592         return pci_platform_pm ? pci_platform_pm->set_state(dev, t) : -ENOSYS;
593 }
594
595 static inline pci_power_t platform_pci_get_power_state(struct pci_dev *dev)
596 {
597         return pci_platform_pm ? pci_platform_pm->get_state(dev) : PCI_UNKNOWN;
598 }
599
600 static inline pci_power_t platform_pci_choose_state(struct pci_dev *dev)
601 {
602         return pci_platform_pm ?
603                         pci_platform_pm->choose_state(dev) : PCI_POWER_ERROR;
604 }
605
606 static inline int platform_pci_set_wakeup(struct pci_dev *dev, bool enable)
607 {
608         return pci_platform_pm ?
609                         pci_platform_pm->set_wakeup(dev, enable) : -ENODEV;
610 }
611
612 static inline bool platform_pci_need_resume(struct pci_dev *dev)
613 {
614         return pci_platform_pm ? pci_platform_pm->need_resume(dev) : false;
615 }
616
617 /**
618  * pci_raw_set_power_state - Use PCI PM registers to set the power state of
619  *                           given PCI device
620  * @dev: PCI device to handle.
621  * @state: PCI power state (D0, D1, D2, D3hot) to put the device into.
622  *
623  * RETURN VALUE:
624  * -EINVAL if the requested state is invalid.
625  * -EIO if device does not support PCI PM or its PM capabilities register has a
626  * wrong version, or device doesn't support the requested state.
627  * 0 if device already is in the requested state.
628  * 0 if device's power state has been successfully changed.
629  */
630 static int pci_raw_set_power_state(struct pci_dev *dev, pci_power_t state)
631 {
632         u16 pmcsr;
633         bool need_restore = false;
634
635         /* Check if we're already there */
636         if (dev->current_state == state)
637                 return 0;
638
639         if (!dev->pm_cap)
640                 return -EIO;
641
642         if (state < PCI_D0 || state > PCI_D3hot)
643                 return -EINVAL;
644
645         /* Validate current state:
646          * Can enter D0 from any state, but if we can only go deeper
647          * to sleep if we're already in a low power state
648          */
649         if (state != PCI_D0 && dev->current_state <= PCI_D3cold
650             && dev->current_state > state) {
651                 dev_err(&dev->dev, "invalid power transition (from state %d to %d)\n",
652                         dev->current_state, state);
653                 return -EINVAL;
654         }
655
656         /* check if this device supports the desired state */
657         if ((state == PCI_D1 && !dev->d1_support)
658            || (state == PCI_D2 && !dev->d2_support))
659                 return -EIO;
660
661         pci_read_config_word(dev, dev->pm_cap + PCI_PM_CTRL, &pmcsr);
662
663         /* If we're (effectively) in D3, force entire word to 0.
664          * This doesn't affect PME_Status, disables PME_En, and
665          * sets PowerState to 0.
666          */
667         switch (dev->current_state) {
668         case PCI_D0:
669         case PCI_D1:
670         case PCI_D2:
671                 pmcsr &= ~PCI_PM_CTRL_STATE_MASK;
672                 pmcsr |= state;
673                 break;
674         case PCI_D3hot:
675         case PCI_D3cold:
676         case PCI_UNKNOWN: /* Boot-up */
677                 if ((pmcsr & PCI_PM_CTRL_STATE_MASK) == PCI_D3hot
678                  && !(pmcsr & PCI_PM_CTRL_NO_SOFT_RESET))
679                         need_restore = true;
680                 /* Fall-through: force to D0 */
681         default:
682                 pmcsr = 0;
683                 break;
684         }
685
686         /* enter specified state */
687         pci_write_config_word(dev, dev->pm_cap + PCI_PM_CTRL, pmcsr);
688
689         /* Mandatory power management transition delays */
690         /* see PCI PM 1.1 5.6.1 table 18 */
691         if (state == PCI_D3hot || dev->current_state == PCI_D3hot)
692                 pci_dev_d3_sleep(dev);
693         else if (state == PCI_D2 || dev->current_state == PCI_D2)
694                 udelay(PCI_PM_D2_DELAY);
695
696         pci_read_config_word(dev, dev->pm_cap + PCI_PM_CTRL, &pmcsr);
697         dev->current_state = (pmcsr & PCI_PM_CTRL_STATE_MASK);
698         if (dev->current_state != state && printk_ratelimit())
699                 dev_info(&dev->dev, "Refused to change power state, currently in D%d\n",
700                          dev->current_state);
701
702         /*
703          * According to section 5.4.1 of the "PCI BUS POWER MANAGEMENT
704          * INTERFACE SPECIFICATION, REV. 1.2", a device transitioning
705          * from D3hot to D0 _may_ perform an internal reset, thereby
706          * going to "D0 Uninitialized" rather than "D0 Initialized".
707          * For example, at least some versions of the 3c905B and the
708          * 3c556B exhibit this behaviour.
709          *
710          * At least some laptop BIOSen (e.g. the Thinkpad T21) leave
711          * devices in a D3hot state at boot.  Consequently, we need to
712          * restore at least the BARs so that the device will be
713          * accessible to its driver.
714          */
715         if (need_restore)
716                 pci_restore_bars(dev);
717
718         if (dev->bus->self)
719                 pcie_aspm_pm_state_change(dev->bus->self);
720
721         return 0;
722 }
723
724 /**
725  * pci_update_current_state - Read power state of given device and cache it
726  * @dev: PCI device to handle.
727  * @state: State to cache in case the device doesn't have the PM capability
728  *
729  * The power state is read from the PMCSR register, which however is
730  * inaccessible in D3cold.  The platform firmware is therefore queried first
731  * to detect accessibility of the register.  In case the platform firmware
732  * reports an incorrect state or the device isn't power manageable by the
733  * platform at all, we try to detect D3cold by testing accessibility of the
734  * vendor ID in config space.
735  */
736 void pci_update_current_state(struct pci_dev *dev, pci_power_t state)
737 {
738         if (platform_pci_get_power_state(dev) == PCI_D3cold ||
739             !pci_device_is_present(dev)) {
740                 dev->current_state = PCI_D3cold;
741         } else if (dev->pm_cap) {
742                 u16 pmcsr;
743
744                 pci_read_config_word(dev, dev->pm_cap + PCI_PM_CTRL, &pmcsr);
745                 dev->current_state = (pmcsr & PCI_PM_CTRL_STATE_MASK);
746         } else {
747                 dev->current_state = state;
748         }
749 }
750
751 /**
752  * pci_platform_power_transition - Use platform to change device power state
753  * @dev: PCI device to handle.
754  * @state: State to put the device into.
755  */
756 static int pci_platform_power_transition(struct pci_dev *dev, pci_power_t state)
757 {
758         int error;
759
760         if (platform_pci_power_manageable(dev)) {
761                 error = platform_pci_set_power_state(dev, state);
762                 if (!error)
763                         pci_update_current_state(dev, state);
764         } else
765                 error = -ENODEV;
766
767         if (error && !dev->pm_cap) /* Fall back to PCI_D0 */
768                 dev->current_state = PCI_D0;
769
770         return error;
771 }
772
773 /**
774  * pci_wakeup - Wake up a PCI device
775  * @pci_dev: Device to handle.
776  * @ign: ignored parameter
777  */
778 static int pci_wakeup(struct pci_dev *pci_dev, void *ign)
779 {
780         pci_wakeup_event(pci_dev);
781         pm_request_resume(&pci_dev->dev);
782         return 0;
783 }
784
785 /**
786  * pci_wakeup_bus - Walk given bus and wake up devices on it
787  * @bus: Top bus of the subtree to walk.
788  */
789 static void pci_wakeup_bus(struct pci_bus *bus)
790 {
791         if (bus)
792                 pci_walk_bus(bus, pci_wakeup, NULL);
793 }
794
795 /**
796  * __pci_start_power_transition - Start power transition of a PCI device
797  * @dev: PCI device to handle.
798  * @state: State to put the device into.
799  */
800 static void __pci_start_power_transition(struct pci_dev *dev, pci_power_t state)
801 {
802         if (state == PCI_D0) {
803                 pci_platform_power_transition(dev, PCI_D0);
804                 /*
805                  * Mandatory power management transition delays, see
806                  * PCI Express Base Specification Revision 2.0 Section
807                  * 6.6.1: Conventional Reset.  Do not delay for
808                  * devices powered on/off by corresponding bridge,
809                  * because have already delayed for the bridge.
810                  */
811                 if (dev->runtime_d3cold) {
812                         if (dev->d3cold_delay)
813                                 msleep(dev->d3cold_delay);
814                         /*
815                          * When powering on a bridge from D3cold, the
816                          * whole hierarchy may be powered on into
817                          * D0uninitialized state, resume them to give
818                          * them a chance to suspend again
819                          */
820                         pci_wakeup_bus(dev->subordinate);
821                 }
822         }
823 }
824
825 /**
826  * __pci_dev_set_current_state - Set current state of a PCI device
827  * @dev: Device to handle
828  * @data: pointer to state to be set
829  */
830 static int __pci_dev_set_current_state(struct pci_dev *dev, void *data)
831 {
832         pci_power_t state = *(pci_power_t *)data;
833
834         dev->current_state = state;
835         return 0;
836 }
837
838 /**
839  * __pci_bus_set_current_state - Walk given bus and set current state of devices
840  * @bus: Top bus of the subtree to walk.
841  * @state: state to be set
842  */
843 static void __pci_bus_set_current_state(struct pci_bus *bus, pci_power_t state)
844 {
845         if (bus)
846                 pci_walk_bus(bus, __pci_dev_set_current_state, &state);
847 }
848
849 /**
850  * __pci_complete_power_transition - Complete power transition of a PCI device
851  * @dev: PCI device to handle.
852  * @state: State to put the device into.
853  *
854  * This function should not be called directly by device drivers.
855  */
856 int __pci_complete_power_transition(struct pci_dev *dev, pci_power_t state)
857 {
858         int ret;
859
860         if (state <= PCI_D0)
861                 return -EINVAL;
862         ret = pci_platform_power_transition(dev, state);
863         /* Power off the bridge may power off the whole hierarchy */
864         if (!ret && state == PCI_D3cold)
865                 __pci_bus_set_current_state(dev->subordinate, PCI_D3cold);
866         return ret;
867 }
868 EXPORT_SYMBOL_GPL(__pci_complete_power_transition);
869
870 /**
871  * pci_set_power_state - Set the power state of a PCI device
872  * @dev: PCI device to handle.
873  * @state: PCI power state (D0, D1, D2, D3hot) to put the device into.
874  *
875  * Transition a device to a new power state, using the platform firmware and/or
876  * the device's PCI PM registers.
877  *
878  * RETURN VALUE:
879  * -EINVAL if the requested state is invalid.
880  * -EIO if device does not support PCI PM or its PM capabilities register has a
881  * wrong version, or device doesn't support the requested state.
882  * 0 if the transition is to D1 or D2 but D1 and D2 are not supported.
883  * 0 if device already is in the requested state.
884  * 0 if the transition is to D3 but D3 is not supported.
885  * 0 if device's power state has been successfully changed.
886  */
887 int pci_set_power_state(struct pci_dev *dev, pci_power_t state)
888 {
889         int error;
890
891         /* bound the state we're entering */
892         if (state > PCI_D3cold)
893                 state = PCI_D3cold;
894         else if (state < PCI_D0)
895                 state = PCI_D0;
896         else if ((state == PCI_D1 || state == PCI_D2) && pci_no_d1d2(dev))
897                 /*
898                  * If the device or the parent bridge do not support PCI PM,
899                  * ignore the request if we're doing anything other than putting
900                  * it into D0 (which would only happen on boot).
901                  */
902                 return 0;
903
904         /* Check if we're already there */
905         if (dev->current_state == state)
906                 return 0;
907
908         __pci_start_power_transition(dev, state);
909
910         /* This device is quirked not to be put into D3, so
911            don't put it in D3 */
912         if (state >= PCI_D3hot && (dev->dev_flags & PCI_DEV_FLAGS_NO_D3))
913                 return 0;
914
915         /*
916          * To put device in D3cold, we put device into D3hot in native
917          * way, then put device into D3cold with platform ops
918          */
919         error = pci_raw_set_power_state(dev, state > PCI_D3hot ?
920                                         PCI_D3hot : state);
921
922         if (!__pci_complete_power_transition(dev, state))
923                 error = 0;
924
925         return error;
926 }
927 EXPORT_SYMBOL(pci_set_power_state);
928
929 /**
930  * pci_power_up - Put the given device into D0 forcibly
931  * @dev: PCI device to power up
932  */
933 void pci_power_up(struct pci_dev *dev)
934 {
935         __pci_start_power_transition(dev, PCI_D0);
936         pci_raw_set_power_state(dev, PCI_D0);
937         pci_update_current_state(dev, PCI_D0);
938 }
939
940 /**
941  * pci_choose_state - Choose the power state of a PCI device
942  * @dev: PCI device to be suspended
943  * @state: target sleep state for the whole system. This is the value
944  *      that is passed to suspend() function.
945  *
946  * Returns PCI power state suitable for given device and given system
947  * message.
948  */
949
950 pci_power_t pci_choose_state(struct pci_dev *dev, pm_message_t state)
951 {
952         pci_power_t ret;
953
954         if (!dev->pm_cap)
955                 return PCI_D0;
956
957         ret = platform_pci_choose_state(dev);
958         if (ret != PCI_POWER_ERROR)
959                 return ret;
960
961         switch (state.event) {
962         case PM_EVENT_ON:
963                 return PCI_D0;
964         case PM_EVENT_FREEZE:
965         case PM_EVENT_PRETHAW:
966                 /* REVISIT both freeze and pre-thaw "should" use D0 */
967         case PM_EVENT_SUSPEND:
968         case PM_EVENT_HIBERNATE:
969                 return PCI_D3hot;
970         default:
971                 dev_info(&dev->dev, "unrecognized suspend event %d\n",
972                          state.event);
973                 BUG();
974         }
975         return PCI_D0;
976 }
977 EXPORT_SYMBOL(pci_choose_state);
978
979 #define PCI_EXP_SAVE_REGS       7
980
981 static struct pci_cap_saved_state *_pci_find_saved_cap(struct pci_dev *pci_dev,
982                                                        u16 cap, bool extended)
983 {
984         struct pci_cap_saved_state *tmp;
985
986         hlist_for_each_entry(tmp, &pci_dev->saved_cap_space, next) {
987                 if (tmp->cap.cap_extended == extended && tmp->cap.cap_nr == cap)
988                         return tmp;
989         }
990         return NULL;
991 }
992
993 struct pci_cap_saved_state *pci_find_saved_cap(struct pci_dev *dev, char cap)
994 {
995         return _pci_find_saved_cap(dev, cap, false);
996 }
997
998 struct pci_cap_saved_state *pci_find_saved_ext_cap(struct pci_dev *dev, u16 cap)
999 {
1000         return _pci_find_saved_cap(dev, cap, true);
1001 }
1002
1003 static int pci_save_pcie_state(struct pci_dev *dev)
1004 {
1005         int i = 0;
1006         struct pci_cap_saved_state *save_state;
1007         u16 *cap;
1008
1009         if (!pci_is_pcie(dev))
1010                 return 0;
1011
1012         save_state = pci_find_saved_cap(dev, PCI_CAP_ID_EXP);
1013         if (!save_state) {
1014                 dev_err(&dev->dev, "buffer not found in %s\n", __func__);
1015                 return -ENOMEM;
1016         }
1017
1018         cap = (u16 *)&save_state->cap.data[0];
1019         pcie_capability_read_word(dev, PCI_EXP_DEVCTL, &cap[i++]);
1020         pcie_capability_read_word(dev, PCI_EXP_LNKCTL, &cap[i++]);
1021         pcie_capability_read_word(dev, PCI_EXP_SLTCTL, &cap[i++]);
1022         pcie_capability_read_word(dev, PCI_EXP_RTCTL,  &cap[i++]);
1023         pcie_capability_read_word(dev, PCI_EXP_DEVCTL2, &cap[i++]);
1024         pcie_capability_read_word(dev, PCI_EXP_LNKCTL2, &cap[i++]);
1025         pcie_capability_read_word(dev, PCI_EXP_SLTCTL2, &cap[i++]);
1026
1027         return 0;
1028 }
1029
1030 static void pci_restore_pcie_state(struct pci_dev *dev)
1031 {
1032         int i = 0;
1033         struct pci_cap_saved_state *save_state;
1034         u16 *cap;
1035
1036         save_state = pci_find_saved_cap(dev, PCI_CAP_ID_EXP);
1037         if (!save_state)
1038                 return;
1039
1040         cap = (u16 *)&save_state->cap.data[0];
1041         pcie_capability_write_word(dev, PCI_EXP_DEVCTL, cap[i++]);
1042         pcie_capability_write_word(dev, PCI_EXP_LNKCTL, cap[i++]);
1043         pcie_capability_write_word(dev, PCI_EXP_SLTCTL, cap[i++]);
1044         pcie_capability_write_word(dev, PCI_EXP_RTCTL, cap[i++]);
1045         pcie_capability_write_word(dev, PCI_EXP_DEVCTL2, cap[i++]);
1046         pcie_capability_write_word(dev, PCI_EXP_LNKCTL2, cap[i++]);
1047         pcie_capability_write_word(dev, PCI_EXP_SLTCTL2, cap[i++]);
1048 }
1049
1050
1051 static int pci_save_pcix_state(struct pci_dev *dev)
1052 {
1053         int pos;
1054         struct pci_cap_saved_state *save_state;
1055
1056         pos = pci_find_capability(dev, PCI_CAP_ID_PCIX);
1057         if (!pos)
1058                 return 0;
1059
1060         save_state = pci_find_saved_cap(dev, PCI_CAP_ID_PCIX);
1061         if (!save_state) {
1062                 dev_err(&dev->dev, "buffer not found in %s\n", __func__);
1063                 return -ENOMEM;
1064         }
1065
1066         pci_read_config_word(dev, pos + PCI_X_CMD,
1067                              (u16 *)save_state->cap.data);
1068
1069         return 0;
1070 }
1071
1072 static void pci_restore_pcix_state(struct pci_dev *dev)
1073 {
1074         int i = 0, pos;
1075         struct pci_cap_saved_state *save_state;
1076         u16 *cap;
1077
1078         save_state = pci_find_saved_cap(dev, PCI_CAP_ID_PCIX);
1079         pos = pci_find_capability(dev, PCI_CAP_ID_PCIX);
1080         if (!save_state || !pos)
1081                 return;
1082         cap = (u16 *)&save_state->cap.data[0];
1083
1084         pci_write_config_word(dev, pos + PCI_X_CMD, cap[i++]);
1085 }
1086
1087
1088 /**
1089  * pci_save_state - save the PCI configuration space of a device before suspending
1090  * @dev: - PCI device that we're dealing with
1091  */
1092 int pci_save_state(struct pci_dev *dev)
1093 {
1094         int i;
1095         /* XXX: 100% dword access ok here? */
1096         for (i = 0; i < 16; i++)
1097                 pci_read_config_dword(dev, i * 4, &dev->saved_config_space[i]);
1098         dev->state_saved = true;
1099
1100         i = pci_save_pcie_state(dev);
1101         if (i != 0)
1102                 return i;
1103
1104         i = pci_save_pcix_state(dev);
1105         if (i != 0)
1106                 return i;
1107
1108         return pci_save_vc_state(dev);
1109 }
1110 EXPORT_SYMBOL(pci_save_state);
1111
1112 static void pci_restore_config_dword(struct pci_dev *pdev, int offset,
1113                                      u32 saved_val, int retry, bool force)
1114 {
1115         u32 val;
1116
1117         pci_read_config_dword(pdev, offset, &val);
1118         if (!force && val == saved_val)
1119                 return;
1120
1121         for (;;) {
1122                 dev_dbg(&pdev->dev, "restoring config space at offset %#x (was %#x, writing %#x)\n",
1123                         offset, val, saved_val);
1124                 pci_write_config_dword(pdev, offset, saved_val);
1125                 if (retry-- <= 0)
1126                         return;
1127
1128                 pci_read_config_dword(pdev, offset, &val);
1129                 if (val == saved_val)
1130                         return;
1131
1132                 mdelay(1);
1133         }
1134 }
1135
1136 static void pci_restore_config_space_range(struct pci_dev *pdev,
1137                                            int start, int end, int retry,
1138                                            bool force)
1139 {
1140         int index;
1141
1142         for (index = end; index >= start; index--)
1143                 pci_restore_config_dword(pdev, 4 * index,
1144                                          pdev->saved_config_space[index],
1145                                          retry, force);
1146 }
1147
1148 static void pci_restore_config_space(struct pci_dev *pdev)
1149 {
1150         if (pdev->hdr_type == PCI_HEADER_TYPE_NORMAL) {
1151                 pci_restore_config_space_range(pdev, 10, 15, 0, false);
1152                 /* Restore BARs before the command register. */
1153                 pci_restore_config_space_range(pdev, 4, 9, 10, false);
1154                 pci_restore_config_space_range(pdev, 0, 3, 0, false);
1155         } else if (pdev->hdr_type == PCI_HEADER_TYPE_BRIDGE) {
1156                 pci_restore_config_space_range(pdev, 12, 15, 0, false);
1157
1158                 /*
1159                  * Force rewriting of prefetch registers to avoid S3 resume
1160                  * issues on Intel PCI bridges that occur when these
1161                  * registers are not explicitly written.
1162                  */
1163                 pci_restore_config_space_range(pdev, 9, 11, 0, true);
1164                 pci_restore_config_space_range(pdev, 0, 8, 0, false);
1165         } else {
1166                 pci_restore_config_space_range(pdev, 0, 15, 0, false);
1167         }
1168 }
1169
1170 /**
1171  * pci_restore_state - Restore the saved state of a PCI device
1172  * @dev: - PCI device that we're dealing with
1173  */
1174 void pci_restore_state(struct pci_dev *dev)
1175 {
1176         if (!dev->state_saved)
1177                 return;
1178
1179         /* PCI Express register must be restored first */
1180         pci_restore_pcie_state(dev);
1181         pci_restore_pasid_state(dev);
1182         pci_restore_pri_state(dev);
1183         pci_restore_ats_state(dev);
1184         pci_restore_vc_state(dev);
1185
1186         pci_cleanup_aer_error_status_regs(dev);
1187
1188         pci_restore_config_space(dev);
1189
1190         pci_restore_pcix_state(dev);
1191         pci_restore_msi_state(dev);
1192
1193         /* Restore ACS and IOV configuration state */
1194         pci_enable_acs(dev);
1195         pci_restore_iov_state(dev);
1196
1197         dev->state_saved = false;
1198 }
1199 EXPORT_SYMBOL(pci_restore_state);
1200
1201 struct pci_saved_state {
1202         u32 config_space[16];
1203         struct pci_cap_saved_data cap[0];
1204 };
1205
1206 /**
1207  * pci_store_saved_state - Allocate and return an opaque struct containing
1208  *                         the device saved state.
1209  * @dev: PCI device that we're dealing with
1210  *
1211  * Return NULL if no state or error.
1212  */
1213 struct pci_saved_state *pci_store_saved_state(struct pci_dev *dev)
1214 {
1215         struct pci_saved_state *state;
1216         struct pci_cap_saved_state *tmp;
1217         struct pci_cap_saved_data *cap;
1218         size_t size;
1219
1220         if (!dev->state_saved)
1221                 return NULL;
1222
1223         size = sizeof(*state) + sizeof(struct pci_cap_saved_data);
1224
1225         hlist_for_each_entry(tmp, &dev->saved_cap_space, next)
1226                 size += sizeof(struct pci_cap_saved_data) + tmp->cap.size;
1227
1228         state = kzalloc(size, GFP_KERNEL);
1229         if (!state)
1230                 return NULL;
1231
1232         memcpy(state->config_space, dev->saved_config_space,
1233                sizeof(state->config_space));
1234
1235         cap = state->cap;
1236         hlist_for_each_entry(tmp, &dev->saved_cap_space, next) {
1237                 size_t len = sizeof(struct pci_cap_saved_data) + tmp->cap.size;
1238                 memcpy(cap, &tmp->cap, len);
1239                 cap = (struct pci_cap_saved_data *)((u8 *)cap + len);
1240         }
1241         /* Empty cap_save terminates list */
1242
1243         return state;
1244 }
1245 EXPORT_SYMBOL_GPL(pci_store_saved_state);
1246
1247 /**
1248  * pci_load_saved_state - Reload the provided save state into struct pci_dev.
1249  * @dev: PCI device that we're dealing with
1250  * @state: Saved state returned from pci_store_saved_state()
1251  */
1252 int pci_load_saved_state(struct pci_dev *dev,
1253                          struct pci_saved_state *state)
1254 {
1255         struct pci_cap_saved_data *cap;
1256
1257         dev->state_saved = false;
1258
1259         if (!state)
1260                 return 0;
1261
1262         memcpy(dev->saved_config_space, state->config_space,
1263                sizeof(state->config_space));
1264
1265         cap = state->cap;
1266         while (cap->size) {
1267                 struct pci_cap_saved_state *tmp;
1268
1269                 tmp = _pci_find_saved_cap(dev, cap->cap_nr, cap->cap_extended);
1270                 if (!tmp || tmp->cap.size != cap->size)
1271                         return -EINVAL;
1272
1273                 memcpy(tmp->cap.data, cap->data, tmp->cap.size);
1274                 cap = (struct pci_cap_saved_data *)((u8 *)cap +
1275                        sizeof(struct pci_cap_saved_data) + cap->size);
1276         }
1277
1278         dev->state_saved = true;
1279         return 0;
1280 }
1281 EXPORT_SYMBOL_GPL(pci_load_saved_state);
1282
1283 /**
1284  * pci_load_and_free_saved_state - Reload the save state pointed to by state,
1285  *                                 and free the memory allocated for it.
1286  * @dev: PCI device that we're dealing with
1287  * @state: Pointer to saved state returned from pci_store_saved_state()
1288  */
1289 int pci_load_and_free_saved_state(struct pci_dev *dev,
1290                                   struct pci_saved_state **state)
1291 {
1292         int ret = pci_load_saved_state(dev, *state);
1293         kfree(*state);
1294         *state = NULL;
1295         return ret;
1296 }
1297 EXPORT_SYMBOL_GPL(pci_load_and_free_saved_state);
1298
1299 int __weak pcibios_enable_device(struct pci_dev *dev, int bars)
1300 {
1301         return pci_enable_resources(dev, bars);
1302 }
1303
1304 static int do_pci_enable_device(struct pci_dev *dev, int bars)
1305 {
1306         int err;
1307         struct pci_dev *bridge;
1308         u16 cmd;
1309         u8 pin;
1310
1311         err = pci_set_power_state(dev, PCI_D0);
1312         if (err < 0 && err != -EIO)
1313                 return err;
1314
1315         bridge = pci_upstream_bridge(dev);
1316         if (bridge)
1317                 pcie_aspm_powersave_config_link(bridge);
1318
1319         err = pcibios_enable_device(dev, bars);
1320         if (err < 0)
1321                 return err;
1322         pci_fixup_device(pci_fixup_enable, dev);
1323
1324         if (dev->msi_enabled || dev->msix_enabled)
1325                 return 0;
1326
1327         pci_read_config_byte(dev, PCI_INTERRUPT_PIN, &pin);
1328         if (pin) {
1329                 pci_read_config_word(dev, PCI_COMMAND, &cmd);
1330                 if (cmd & PCI_COMMAND_INTX_DISABLE)
1331                         pci_write_config_word(dev, PCI_COMMAND,
1332                                               cmd & ~PCI_COMMAND_INTX_DISABLE);
1333         }
1334
1335         return 0;
1336 }
1337
1338 /**
1339  * pci_reenable_device - Resume abandoned device
1340  * @dev: PCI device to be resumed
1341  *
1342  *  Note this function is a backend of pci_default_resume and is not supposed
1343  *  to be called by normal code, write proper resume handler and use it instead.
1344  */
1345 int pci_reenable_device(struct pci_dev *dev)
1346 {
1347         if (pci_is_enabled(dev))
1348                 return do_pci_enable_device(dev, (1 << PCI_NUM_RESOURCES) - 1);
1349         return 0;
1350 }
1351 EXPORT_SYMBOL(pci_reenable_device);
1352
1353 static void pci_enable_bridge(struct pci_dev *dev)
1354 {
1355         struct pci_dev *bridge;
1356         int retval;
1357
1358         bridge = pci_upstream_bridge(dev);
1359         if (bridge)
1360                 pci_enable_bridge(bridge);
1361
1362         if (pci_is_enabled(dev)) {
1363                 if (!dev->is_busmaster)
1364                         pci_set_master(dev);
1365                 return;
1366         }
1367
1368         retval = pci_enable_device(dev);
1369         if (retval)
1370                 dev_err(&dev->dev, "Error enabling bridge (%d), continuing\n",
1371                         retval);
1372         pci_set_master(dev);
1373 }
1374
1375 static int pci_enable_device_flags(struct pci_dev *dev, unsigned long flags)
1376 {
1377         struct pci_dev *bridge;
1378         int err;
1379         int i, bars = 0;
1380
1381         /*
1382          * Power state could be unknown at this point, either due to a fresh
1383          * boot or a device removal call.  So get the current power state
1384          * so that things like MSI message writing will behave as expected
1385          * (e.g. if the device really is in D0 at enable time).
1386          */
1387         pci_update_current_state(dev, dev->current_state);
1388
1389         if (atomic_inc_return(&dev->enable_cnt) > 1)
1390                 return 0;               /* already enabled */
1391
1392         bridge = pci_upstream_bridge(dev);
1393         if (bridge)
1394                 pci_enable_bridge(bridge);
1395
1396         /* only skip sriov related */
1397         for (i = 0; i <= PCI_ROM_RESOURCE; i++)
1398                 if (dev->resource[i].flags & flags)
1399                         bars |= (1 << i);
1400         for (i = PCI_BRIDGE_RESOURCES; i < DEVICE_COUNT_RESOURCE; i++)
1401                 if (dev->resource[i].flags & flags)
1402                         bars |= (1 << i);
1403
1404         err = do_pci_enable_device(dev, bars);
1405         if (err < 0)
1406                 atomic_dec(&dev->enable_cnt);
1407         return err;
1408 }
1409
1410 /**
1411  * pci_enable_device_io - Initialize a device for use with IO space
1412  * @dev: PCI device to be initialized
1413  *
1414  *  Initialize device before it's used by a driver. Ask low-level code
1415  *  to enable I/O resources. Wake up the device if it was suspended.
1416  *  Beware, this function can fail.
1417  */
1418 int pci_enable_device_io(struct pci_dev *dev)
1419 {
1420         return pci_enable_device_flags(dev, IORESOURCE_IO);
1421 }
1422 EXPORT_SYMBOL(pci_enable_device_io);
1423
1424 /**
1425  * pci_enable_device_mem - Initialize a device for use with Memory space
1426  * @dev: PCI device to be initialized
1427  *
1428  *  Initialize device before it's used by a driver. Ask low-level code
1429  *  to enable Memory resources. Wake up the device if it was suspended.
1430  *  Beware, this function can fail.
1431  */
1432 int pci_enable_device_mem(struct pci_dev *dev)
1433 {
1434         return pci_enable_device_flags(dev, IORESOURCE_MEM);
1435 }
1436 EXPORT_SYMBOL(pci_enable_device_mem);
1437
1438 /**
1439  * pci_enable_device - Initialize device before it's used by a driver.
1440  * @dev: PCI device to be initialized
1441  *
1442  *  Initialize device before it's used by a driver. Ask low-level code
1443  *  to enable I/O and memory. Wake up the device if it was suspended.
1444  *  Beware, this function can fail.
1445  *
1446  *  Note we don't actually enable the device many times if we call
1447  *  this function repeatedly (we just increment the count).
1448  */
1449 int pci_enable_device(struct pci_dev *dev)
1450 {
1451         return pci_enable_device_flags(dev, IORESOURCE_MEM | IORESOURCE_IO);
1452 }
1453 EXPORT_SYMBOL(pci_enable_device);
1454
1455 /*
1456  * Managed PCI resources.  This manages device on/off, intx/msi/msix
1457  * on/off and BAR regions.  pci_dev itself records msi/msix status, so
1458  * there's no need to track it separately.  pci_devres is initialized
1459  * when a device is enabled using managed PCI device enable interface.
1460  */
1461 struct pci_devres {
1462         unsigned int enabled:1;
1463         unsigned int pinned:1;
1464         unsigned int orig_intx:1;
1465         unsigned int restore_intx:1;
1466         u32 region_mask;
1467 };
1468
1469 static void pcim_release(struct device *gendev, void *res)
1470 {
1471         struct pci_dev *dev = to_pci_dev(gendev);
1472         struct pci_devres *this = res;
1473         int i;
1474
1475         if (dev->msi_enabled)
1476                 pci_disable_msi(dev);
1477         if (dev->msix_enabled)
1478                 pci_disable_msix(dev);
1479
1480         for (i = 0; i < DEVICE_COUNT_RESOURCE; i++)
1481                 if (this->region_mask & (1 << i))
1482                         pci_release_region(dev, i);
1483
1484         if (this->restore_intx)
1485                 pci_intx(dev, this->orig_intx);
1486
1487         if (this->enabled && !this->pinned)
1488                 pci_disable_device(dev);
1489 }
1490
1491 static struct pci_devres *get_pci_dr(struct pci_dev *pdev)
1492 {
1493         struct pci_devres *dr, *new_dr;
1494
1495         dr = devres_find(&pdev->dev, pcim_release, NULL, NULL);
1496         if (dr)
1497                 return dr;
1498
1499         new_dr = devres_alloc(pcim_release, sizeof(*new_dr), GFP_KERNEL);
1500         if (!new_dr)
1501                 return NULL;
1502         return devres_get(&pdev->dev, new_dr, NULL, NULL);
1503 }
1504
1505 static struct pci_devres *find_pci_dr(struct pci_dev *pdev)
1506 {
1507         if (pci_is_managed(pdev))
1508                 return devres_find(&pdev->dev, pcim_release, NULL, NULL);
1509         return NULL;
1510 }
1511
1512 /**
1513  * pcim_enable_device - Managed pci_enable_device()
1514  * @pdev: PCI device to be initialized
1515  *
1516  * Managed pci_enable_device().
1517  */
1518 int pcim_enable_device(struct pci_dev *pdev)
1519 {
1520         struct pci_devres *dr;
1521         int rc;
1522
1523         dr = get_pci_dr(pdev);
1524         if (unlikely(!dr))
1525                 return -ENOMEM;
1526         if (dr->enabled)
1527                 return 0;
1528
1529         rc = pci_enable_device(pdev);
1530         if (!rc) {
1531                 pdev->is_managed = 1;
1532                 dr->enabled = 1;
1533         }
1534         return rc;
1535 }
1536 EXPORT_SYMBOL(pcim_enable_device);
1537
1538 /**
1539  * pcim_pin_device - Pin managed PCI device
1540  * @pdev: PCI device to pin
1541  *
1542  * Pin managed PCI device @pdev.  Pinned device won't be disabled on
1543  * driver detach.  @pdev must have been enabled with
1544  * pcim_enable_device().
1545  */
1546 void pcim_pin_device(struct pci_dev *pdev)
1547 {
1548         struct pci_devres *dr;
1549
1550         dr = find_pci_dr(pdev);
1551         WARN_ON(!dr || !dr->enabled);
1552         if (dr)
1553                 dr->pinned = 1;
1554 }
1555 EXPORT_SYMBOL(pcim_pin_device);
1556
1557 /*
1558  * pcibios_add_device - provide arch specific hooks when adding device dev
1559  * @dev: the PCI device being added
1560  *
1561  * Permits the platform to provide architecture specific functionality when
1562  * devices are added. This is the default implementation. Architecture
1563  * implementations can override this.
1564  */
1565 int __weak pcibios_add_device(struct pci_dev *dev)
1566 {
1567         return 0;
1568 }
1569
1570 /**
1571  * pcibios_release_device - provide arch specific hooks when releasing device dev
1572  * @dev: the PCI device being released
1573  *
1574  * Permits the platform to provide architecture specific functionality when
1575  * devices are released. This is the default implementation. Architecture
1576  * implementations can override this.
1577  */
1578 void __weak pcibios_release_device(struct pci_dev *dev) {}
1579
1580 /**
1581  * pcibios_disable_device - disable arch specific PCI resources for device dev
1582  * @dev: the PCI device to disable
1583  *
1584  * Disables architecture specific PCI resources for the device. This
1585  * is the default implementation. Architecture implementations can
1586  * override this.
1587  */
1588 void __weak pcibios_disable_device(struct pci_dev *dev) {}
1589
1590 /**
1591  * pcibios_penalize_isa_irq - penalize an ISA IRQ
1592  * @irq: ISA IRQ to penalize
1593  * @active: IRQ active or not
1594  *
1595  * Permits the platform to provide architecture-specific functionality when
1596  * penalizing ISA IRQs. This is the default implementation. Architecture
1597  * implementations can override this.
1598  */
1599 void __weak pcibios_penalize_isa_irq(int irq, int active) {}
1600
1601 static void do_pci_disable_device(struct pci_dev *dev)
1602 {
1603         u16 pci_command;
1604
1605         pci_read_config_word(dev, PCI_COMMAND, &pci_command);
1606         if (pci_command & PCI_COMMAND_MASTER) {
1607                 pci_command &= ~PCI_COMMAND_MASTER;
1608                 pci_write_config_word(dev, PCI_COMMAND, pci_command);
1609         }
1610
1611         pcibios_disable_device(dev);
1612 }
1613
1614 /**
1615  * pci_disable_enabled_device - Disable device without updating enable_cnt
1616  * @dev: PCI device to disable
1617  *
1618  * NOTE: This function is a backend of PCI power management routines and is
1619  * not supposed to be called drivers.
1620  */
1621 void pci_disable_enabled_device(struct pci_dev *dev)
1622 {
1623         if (pci_is_enabled(dev))
1624                 do_pci_disable_device(dev);
1625 }
1626
1627 /**
1628  * pci_disable_device - Disable PCI device after use
1629  * @dev: PCI device to be disabled
1630  *
1631  * Signal to the system that the PCI device is not in use by the system
1632  * anymore.  This only involves disabling PCI bus-mastering, if active.
1633  *
1634  * Note we don't actually disable the device until all callers of
1635  * pci_enable_device() have called pci_disable_device().
1636  */
1637 void pci_disable_device(struct pci_dev *dev)
1638 {
1639         struct pci_devres *dr;
1640
1641         dr = find_pci_dr(dev);
1642         if (dr)
1643                 dr->enabled = 0;
1644
1645         dev_WARN_ONCE(&dev->dev, atomic_read(&dev->enable_cnt) <= 0,
1646                       "disabling already-disabled device");
1647
1648         if (atomic_dec_return(&dev->enable_cnt) != 0)
1649                 return;
1650
1651         do_pci_disable_device(dev);
1652
1653         dev->is_busmaster = 0;
1654 }
1655 EXPORT_SYMBOL(pci_disable_device);
1656
1657 /**
1658  * pcibios_set_pcie_reset_state - set reset state for device dev
1659  * @dev: the PCIe device reset
1660  * @state: Reset state to enter into
1661  *
1662  *
1663  * Sets the PCIe reset state for the device. This is the default
1664  * implementation. Architecture implementations can override this.
1665  */
1666 int __weak pcibios_set_pcie_reset_state(struct pci_dev *dev,
1667                                         enum pcie_reset_state state)
1668 {
1669         return -EINVAL;
1670 }
1671
1672 /**
1673  * pci_set_pcie_reset_state - set reset state for device dev
1674  * @dev: the PCIe device reset
1675  * @state: Reset state to enter into
1676  *
1677  *
1678  * Sets the PCI reset state for the device.
1679  */
1680 int pci_set_pcie_reset_state(struct pci_dev *dev, enum pcie_reset_state state)
1681 {
1682         return pcibios_set_pcie_reset_state(dev, state);
1683 }
1684 EXPORT_SYMBOL_GPL(pci_set_pcie_reset_state);
1685
1686 /**
1687  * pci_check_pme_status - Check if given device has generated PME.
1688  * @dev: Device to check.
1689  *
1690  * Check the PME status of the device and if set, clear it and clear PME enable
1691  * (if set).  Return 'true' if PME status and PME enable were both set or
1692  * 'false' otherwise.
1693  */
1694 bool pci_check_pme_status(struct pci_dev *dev)
1695 {
1696         int pmcsr_pos;
1697         u16 pmcsr;
1698         bool ret = false;
1699
1700         if (!dev->pm_cap)
1701                 return false;
1702
1703         pmcsr_pos = dev->pm_cap + PCI_PM_CTRL;
1704         pci_read_config_word(dev, pmcsr_pos, &pmcsr);
1705         if (!(pmcsr & PCI_PM_CTRL_PME_STATUS))
1706                 return false;
1707
1708         /* Clear PME status. */
1709         pmcsr |= PCI_PM_CTRL_PME_STATUS;
1710         if (pmcsr & PCI_PM_CTRL_PME_ENABLE) {
1711                 /* Disable PME to avoid interrupt flood. */
1712                 pmcsr &= ~PCI_PM_CTRL_PME_ENABLE;
1713                 ret = true;
1714         }
1715
1716         pci_write_config_word(dev, pmcsr_pos, pmcsr);
1717
1718         return ret;
1719 }
1720
1721 /**
1722  * pci_pme_wakeup - Wake up a PCI device if its PME Status bit is set.
1723  * @dev: Device to handle.
1724  * @pme_poll_reset: Whether or not to reset the device's pme_poll flag.
1725  *
1726  * Check if @dev has generated PME and queue a resume request for it in that
1727  * case.
1728  */
1729 static int pci_pme_wakeup(struct pci_dev *dev, void *pme_poll_reset)
1730 {
1731         if (pme_poll_reset && dev->pme_poll)
1732                 dev->pme_poll = false;
1733
1734         if (pci_check_pme_status(dev)) {
1735                 pci_wakeup_event(dev);
1736                 pm_request_resume(&dev->dev);
1737         }
1738         return 0;
1739 }
1740
1741 /**
1742  * pci_pme_wakeup_bus - Walk given bus and wake up devices on it, if necessary.
1743  * @bus: Top bus of the subtree to walk.
1744  */
1745 void pci_pme_wakeup_bus(struct pci_bus *bus)
1746 {
1747         if (bus)
1748                 pci_walk_bus(bus, pci_pme_wakeup, (void *)true);
1749 }
1750
1751
1752 /**
1753  * pci_pme_capable - check the capability of PCI device to generate PME#
1754  * @dev: PCI device to handle.
1755  * @state: PCI state from which device will issue PME#.
1756  */
1757 bool pci_pme_capable(struct pci_dev *dev, pci_power_t state)
1758 {
1759         if (!dev->pm_cap)
1760                 return false;
1761
1762         return !!(dev->pme_support & (1 << state));
1763 }
1764 EXPORT_SYMBOL(pci_pme_capable);
1765
1766 static void pci_pme_list_scan(struct work_struct *work)
1767 {
1768         struct pci_pme_device *pme_dev, *n;
1769
1770         mutex_lock(&pci_pme_list_mutex);
1771         list_for_each_entry_safe(pme_dev, n, &pci_pme_list, list) {
1772                 if (pme_dev->dev->pme_poll) {
1773                         struct pci_dev *bridge;
1774
1775                         bridge = pme_dev->dev->bus->self;
1776                         /*
1777                          * If bridge is in low power state, the
1778                          * configuration space of subordinate devices
1779                          * may be not accessible
1780                          */
1781                         if (bridge && bridge->current_state != PCI_D0)
1782                                 continue;
1783                         /*
1784                          * If the device is in D3cold it should not be
1785                          * polled either.
1786                          */
1787                         if (pme_dev->dev->current_state == PCI_D3cold)
1788                                 continue;
1789
1790                         pci_pme_wakeup(pme_dev->dev, NULL);
1791                 } else {
1792                         list_del(&pme_dev->list);
1793                         kfree(pme_dev);
1794                 }
1795         }
1796         if (!list_empty(&pci_pme_list))
1797                 queue_delayed_work(system_freezable_wq, &pci_pme_work,
1798                                    msecs_to_jiffies(PME_TIMEOUT));
1799         mutex_unlock(&pci_pme_list_mutex);
1800 }
1801
1802 static void __pci_pme_active(struct pci_dev *dev, bool enable)
1803 {
1804         u16 pmcsr;
1805
1806         if (!dev->pme_support)
1807                 return;
1808
1809         pci_read_config_word(dev, dev->pm_cap + PCI_PM_CTRL, &pmcsr);
1810         /* Clear PME_Status by writing 1 to it and enable PME# */
1811         pmcsr |= PCI_PM_CTRL_PME_STATUS | PCI_PM_CTRL_PME_ENABLE;
1812         if (!enable)
1813                 pmcsr &= ~PCI_PM_CTRL_PME_ENABLE;
1814
1815         pci_write_config_word(dev, dev->pm_cap + PCI_PM_CTRL, pmcsr);
1816 }
1817
1818 /**
1819  * pci_pme_restore - Restore PME configuration after config space restore.
1820  * @dev: PCI device to update.
1821  */
1822 void pci_pme_restore(struct pci_dev *dev)
1823 {
1824         u16 pmcsr;
1825
1826         if (!dev->pme_support)
1827                 return;
1828
1829         pci_read_config_word(dev, dev->pm_cap + PCI_PM_CTRL, &pmcsr);
1830         if (dev->wakeup_prepared) {
1831                 pmcsr |= PCI_PM_CTRL_PME_ENABLE;
1832                 pmcsr &= ~PCI_PM_CTRL_PME_STATUS;
1833         } else {
1834                 pmcsr &= ~PCI_PM_CTRL_PME_ENABLE;
1835                 pmcsr |= PCI_PM_CTRL_PME_STATUS;
1836         }
1837         pci_write_config_word(dev, dev->pm_cap + PCI_PM_CTRL, pmcsr);
1838 }
1839
1840 /**
1841  * pci_pme_active - enable or disable PCI device's PME# function
1842  * @dev: PCI device to handle.
1843  * @enable: 'true' to enable PME# generation; 'false' to disable it.
1844  *
1845  * The caller must verify that the device is capable of generating PME# before
1846  * calling this function with @enable equal to 'true'.
1847  */
1848 void pci_pme_active(struct pci_dev *dev, bool enable)
1849 {
1850         __pci_pme_active(dev, enable);
1851
1852         /*
1853          * PCI (as opposed to PCIe) PME requires that the device have
1854          * its PME# line hooked up correctly. Not all hardware vendors
1855          * do this, so the PME never gets delivered and the device
1856          * remains asleep. The easiest way around this is to
1857          * periodically walk the list of suspended devices and check
1858          * whether any have their PME flag set. The assumption is that
1859          * we'll wake up often enough anyway that this won't be a huge
1860          * hit, and the power savings from the devices will still be a
1861          * win.
1862          *
1863          * Although PCIe uses in-band PME message instead of PME# line
1864          * to report PME, PME does not work for some PCIe devices in
1865          * reality.  For example, there are devices that set their PME
1866          * status bits, but don't really bother to send a PME message;
1867          * there are PCI Express Root Ports that don't bother to
1868          * trigger interrupts when they receive PME messages from the
1869          * devices below.  So PME poll is used for PCIe devices too.
1870          */
1871
1872         if (dev->pme_poll) {
1873                 struct pci_pme_device *pme_dev;
1874                 if (enable) {
1875                         pme_dev = kmalloc(sizeof(struct pci_pme_device),
1876                                           GFP_KERNEL);
1877                         if (!pme_dev) {
1878                                 dev_warn(&dev->dev, "can't enable PME#\n");
1879                                 return;
1880                         }
1881                         pme_dev->dev = dev;
1882                         mutex_lock(&pci_pme_list_mutex);
1883                         list_add(&pme_dev->list, &pci_pme_list);
1884                         if (list_is_singular(&pci_pme_list))
1885                                 queue_delayed_work(system_freezable_wq,
1886                                                    &pci_pme_work,
1887                                                    msecs_to_jiffies(PME_TIMEOUT));
1888                         mutex_unlock(&pci_pme_list_mutex);
1889                 } else {
1890                         mutex_lock(&pci_pme_list_mutex);
1891                         list_for_each_entry(pme_dev, &pci_pme_list, list) {
1892                                 if (pme_dev->dev == dev) {
1893                                         list_del(&pme_dev->list);
1894                                         kfree(pme_dev);
1895                                         break;
1896                                 }
1897                         }
1898                         mutex_unlock(&pci_pme_list_mutex);
1899                 }
1900         }
1901
1902         dev_dbg(&dev->dev, "PME# %s\n", enable ? "enabled" : "disabled");
1903 }
1904 EXPORT_SYMBOL(pci_pme_active);
1905
1906 /**
1907  * __pci_enable_wake - enable PCI device as wakeup event source
1908  * @dev: PCI device affected
1909  * @state: PCI state from which device will issue wakeup events
1910  * @enable: True to enable event generation; false to disable
1911  *
1912  * This enables the device as a wakeup event source, or disables it.
1913  * When such events involves platform-specific hooks, those hooks are
1914  * called automatically by this routine.
1915  *
1916  * Devices with legacy power management (no standard PCI PM capabilities)
1917  * always require such platform hooks.
1918  *
1919  * RETURN VALUE:
1920  * 0 is returned on success
1921  * -EINVAL is returned if device is not supposed to wake up the system
1922  * Error code depending on the platform is returned if both the platform and
1923  * the native mechanism fail to enable the generation of wake-up events
1924  */
1925 static int __pci_enable_wake(struct pci_dev *dev, pci_power_t state, bool enable)
1926 {
1927         int ret = 0;
1928
1929         /*
1930          * Bridges can only signal wakeup on behalf of subordinate devices,
1931          * but that is set up elsewhere, so skip them.
1932          */
1933         if (pci_has_subordinate(dev))
1934                 return 0;
1935
1936         /* Don't do the same thing twice in a row for one device. */
1937         if (!!enable == !!dev->wakeup_prepared)
1938                 return 0;
1939
1940         /*
1941          * According to "PCI System Architecture" 4th ed. by Tom Shanley & Don
1942          * Anderson we should be doing PME# wake enable followed by ACPI wake
1943          * enable.  To disable wake-up we call the platform first, for symmetry.
1944          */
1945
1946         if (enable) {
1947                 int error;
1948
1949                 /*
1950                  * Enable PME signaling if the device can signal PME from
1951                  * D3cold regardless of whether or not it can signal PME from
1952                  * the current target state, because that will allow it to
1953                  * signal PME when the hierarchy above it goes into D3cold and
1954                  * the device itself ends up in D3cold as a result of that.
1955                  */
1956                 if (pci_pme_capable(dev, state) || pci_pme_capable(dev, PCI_D3cold))
1957                         pci_pme_active(dev, true);
1958                 else
1959                         ret = 1;
1960                 error = platform_pci_set_wakeup(dev, true);
1961                 if (ret)
1962                         ret = error;
1963                 if (!ret)
1964                         dev->wakeup_prepared = true;
1965         } else {
1966                 platform_pci_set_wakeup(dev, false);
1967                 pci_pme_active(dev, false);
1968                 dev->wakeup_prepared = false;
1969         }
1970
1971         return ret;
1972 }
1973
1974 /**
1975  * pci_enable_wake - change wakeup settings for a PCI device
1976  * @pci_dev: Target device
1977  * @state: PCI state from which device will issue wakeup events
1978  * @enable: Whether or not to enable event generation
1979  *
1980  * If @enable is set, check device_may_wakeup() for the device before calling
1981  * __pci_enable_wake() for it.
1982  */
1983 int pci_enable_wake(struct pci_dev *pci_dev, pci_power_t state, bool enable)
1984 {
1985         if (enable && !device_may_wakeup(&pci_dev->dev))
1986                 return -EINVAL;
1987
1988         return __pci_enable_wake(pci_dev, state, enable);
1989 }
1990 EXPORT_SYMBOL(pci_enable_wake);
1991
1992 /**
1993  * pci_wake_from_d3 - enable/disable device to wake up from D3_hot or D3_cold
1994  * @dev: PCI device to prepare
1995  * @enable: True to enable wake-up event generation; false to disable
1996  *
1997  * Many drivers want the device to wake up the system from D3_hot or D3_cold
1998  * and this function allows them to set that up cleanly - pci_enable_wake()
1999  * should not be called twice in a row to enable wake-up due to PCI PM vs ACPI
2000  * ordering constraints.
2001  *
2002  * This function only returns error code if the device is not allowed to wake
2003  * up the system from sleep or it is not capable of generating PME# from both
2004  * D3_hot and D3_cold and the platform is unable to enable wake-up power for it.
2005  */
2006 int pci_wake_from_d3(struct pci_dev *dev, bool enable)
2007 {
2008         return pci_pme_capable(dev, PCI_D3cold) ?
2009                         pci_enable_wake(dev, PCI_D3cold, enable) :
2010                         pci_enable_wake(dev, PCI_D3hot, enable);
2011 }
2012 EXPORT_SYMBOL(pci_wake_from_d3);
2013
2014 /**
2015  * pci_target_state - find an appropriate low power state for a given PCI dev
2016  * @dev: PCI device
2017  * @wakeup: Whether or not wakeup functionality will be enabled for the device.
2018  *
2019  * Use underlying platform code to find a supported low power state for @dev.
2020  * If the platform can't manage @dev, return the deepest state from which it
2021  * can generate wake events, based on any available PME info.
2022  */
2023 static pci_power_t pci_target_state(struct pci_dev *dev, bool wakeup)
2024 {
2025         pci_power_t target_state = PCI_D3hot;
2026
2027         if (platform_pci_power_manageable(dev)) {
2028                 /*
2029                  * Call the platform to choose the target state of the device
2030                  * and enable wake-up from this state if supported.
2031                  */
2032                 pci_power_t state = platform_pci_choose_state(dev);
2033
2034                 switch (state) {
2035                 case PCI_POWER_ERROR:
2036                 case PCI_UNKNOWN:
2037                         break;
2038                 case PCI_D1:
2039                 case PCI_D2:
2040                         if (pci_no_d1d2(dev))
2041                                 break;
2042                 default:
2043                         target_state = state;
2044                 }
2045
2046                 return target_state;
2047         }
2048
2049         if (!dev->pm_cap)
2050                 target_state = PCI_D0;
2051
2052         /*
2053          * If the device is in D3cold even though it's not power-manageable by
2054          * the platform, it may have been powered down by non-standard means.
2055          * Best to let it slumber.
2056          */
2057         if (dev->current_state == PCI_D3cold)
2058                 target_state = PCI_D3cold;
2059
2060         if (wakeup && dev->pme_support) {
2061                 pci_power_t state = target_state;
2062
2063                 /*
2064                  * Find the deepest state from which the device can generate
2065                  * wake-up events, make it the target state and enable device
2066                  * to generate PME#.
2067                  */
2068                 while (state && !(dev->pme_support & (1 << state)))
2069                         state--;
2070
2071                 if (state)
2072                         return state;
2073                 else if (dev->pme_support & 1)
2074                         return PCI_D0;
2075         }
2076
2077         return target_state;
2078 }
2079
2080 /**
2081  * pci_prepare_to_sleep - prepare PCI device for system-wide transition into a sleep state
2082  * @dev: Device to handle.
2083  *
2084  * Choose the power state appropriate for the device depending on whether
2085  * it can wake up the system and/or is power manageable by the platform
2086  * (PCI_D3hot is the default) and put the device into that state.
2087  */
2088 int pci_prepare_to_sleep(struct pci_dev *dev)
2089 {
2090         bool wakeup = device_may_wakeup(&dev->dev);
2091         pci_power_t target_state = pci_target_state(dev, wakeup);
2092         int error;
2093
2094         if (target_state == PCI_POWER_ERROR)
2095                 return -EIO;
2096
2097         pci_enable_wake(dev, target_state, wakeup);
2098
2099         error = pci_set_power_state(dev, target_state);
2100
2101         if (error)
2102                 pci_enable_wake(dev, target_state, false);
2103
2104         return error;
2105 }
2106 EXPORT_SYMBOL(pci_prepare_to_sleep);
2107
2108 /**
2109  * pci_back_from_sleep - turn PCI device on during system-wide transition into working state
2110  * @dev: Device to handle.
2111  *
2112  * Disable device's system wake-up capability and put it into D0.
2113  */
2114 int pci_back_from_sleep(struct pci_dev *dev)
2115 {
2116         pci_enable_wake(dev, PCI_D0, false);
2117         return pci_set_power_state(dev, PCI_D0);
2118 }
2119 EXPORT_SYMBOL(pci_back_from_sleep);
2120
2121 /**
2122  * pci_finish_runtime_suspend - Carry out PCI-specific part of runtime suspend.
2123  * @dev: PCI device being suspended.
2124  *
2125  * Prepare @dev to generate wake-up events at run time and put it into a low
2126  * power state.
2127  */
2128 int pci_finish_runtime_suspend(struct pci_dev *dev)
2129 {
2130         pci_power_t target_state;
2131         int error;
2132
2133         target_state = pci_target_state(dev, device_can_wakeup(&dev->dev));
2134         if (target_state == PCI_POWER_ERROR)
2135                 return -EIO;
2136
2137         dev->runtime_d3cold = target_state == PCI_D3cold;
2138
2139         __pci_enable_wake(dev, target_state, pci_dev_run_wake(dev));
2140
2141         error = pci_set_power_state(dev, target_state);
2142
2143         if (error) {
2144                 pci_enable_wake(dev, target_state, false);
2145                 dev->runtime_d3cold = false;
2146         }
2147
2148         return error;
2149 }
2150
2151 /**
2152  * pci_dev_run_wake - Check if device can generate run-time wake-up events.
2153  * @dev: Device to check.
2154  *
2155  * Return true if the device itself is capable of generating wake-up events
2156  * (through the platform or using the native PCIe PME) or if the device supports
2157  * PME and one of its upstream bridges can generate wake-up events.
2158  */
2159 bool pci_dev_run_wake(struct pci_dev *dev)
2160 {
2161         struct pci_bus *bus = dev->bus;
2162
2163         if (!dev->pme_support)
2164                 return false;
2165
2166         /* PME-capable in principle, but not from the target power state */
2167         if (!pci_pme_capable(dev, pci_target_state(dev, true)))
2168                 return false;
2169
2170         if (device_can_wakeup(&dev->dev))
2171                 return true;
2172
2173         while (bus->parent) {
2174                 struct pci_dev *bridge = bus->self;
2175
2176                 if (device_can_wakeup(&bridge->dev))
2177                         return true;
2178
2179                 bus = bus->parent;
2180         }
2181
2182         /* We have reached the root bus. */
2183         if (bus->bridge)
2184                 return device_can_wakeup(bus->bridge);
2185
2186         return false;
2187 }
2188 EXPORT_SYMBOL_GPL(pci_dev_run_wake);
2189
2190 /**
2191  * pci_dev_keep_suspended - Check if the device can stay in the suspended state.
2192  * @pci_dev: Device to check.
2193  *
2194  * Return 'true' if the device is runtime-suspended, it doesn't have to be
2195  * reconfigured due to wakeup settings difference between system and runtime
2196  * suspend and the current power state of it is suitable for the upcoming
2197  * (system) transition.
2198  *
2199  * If the device is not configured for system wakeup, disable PME for it before
2200  * returning 'true' to prevent it from waking up the system unnecessarily.
2201  */
2202 bool pci_dev_keep_suspended(struct pci_dev *pci_dev)
2203 {
2204         struct device *dev = &pci_dev->dev;
2205         bool wakeup = device_may_wakeup(dev);
2206
2207         if (!pm_runtime_suspended(dev)
2208             || pci_target_state(pci_dev, wakeup) != pci_dev->current_state
2209             || platform_pci_need_resume(pci_dev)
2210             || (pci_dev->dev_flags & PCI_DEV_FLAGS_NEEDS_RESUME))
2211                 return false;
2212
2213         /*
2214          * At this point the device is good to go unless it's been configured
2215          * to generate PME at the runtime suspend time, but it is not supposed
2216          * to wake up the system.  In that case, simply disable PME for it
2217          * (it will have to be re-enabled on exit from system resume).
2218          *
2219          * If the device's power state is D3cold and the platform check above
2220          * hasn't triggered, the device's configuration is suitable and we don't
2221          * need to manipulate it at all.
2222          */
2223         spin_lock_irq(&dev->power.lock);
2224
2225         if (pm_runtime_suspended(dev) && pci_dev->current_state < PCI_D3cold &&
2226             !wakeup)
2227                 __pci_pme_active(pci_dev, false);
2228
2229         spin_unlock_irq(&dev->power.lock);
2230         return true;
2231 }
2232
2233 /**
2234  * pci_dev_complete_resume - Finalize resume from system sleep for a device.
2235  * @pci_dev: Device to handle.
2236  *
2237  * If the device is runtime suspended and wakeup-capable, enable PME for it as
2238  * it might have been disabled during the prepare phase of system suspend if
2239  * the device was not configured for system wakeup.
2240  */
2241 void pci_dev_complete_resume(struct pci_dev *pci_dev)
2242 {
2243         struct device *dev = &pci_dev->dev;
2244
2245         if (!pci_dev_run_wake(pci_dev))
2246                 return;
2247
2248         spin_lock_irq(&dev->power.lock);
2249
2250         if (pm_runtime_suspended(dev) && pci_dev->current_state < PCI_D3cold)
2251                 __pci_pme_active(pci_dev, true);
2252
2253         spin_unlock_irq(&dev->power.lock);
2254 }
2255
2256 void pci_config_pm_runtime_get(struct pci_dev *pdev)
2257 {
2258         struct device *dev = &pdev->dev;
2259         struct device *parent = dev->parent;
2260
2261         if (parent)
2262                 pm_runtime_get_sync(parent);
2263         pm_runtime_get_noresume(dev);
2264         /*
2265          * pdev->current_state is set to PCI_D3cold during suspending,
2266          * so wait until suspending completes
2267          */
2268         pm_runtime_barrier(dev);
2269         /*
2270          * Only need to resume devices in D3cold, because config
2271          * registers are still accessible for devices suspended but
2272          * not in D3cold.
2273          */
2274         if (pdev->current_state == PCI_D3cold)
2275                 pm_runtime_resume(dev);
2276 }
2277
2278 void pci_config_pm_runtime_put(struct pci_dev *pdev)
2279 {
2280         struct device *dev = &pdev->dev;
2281         struct device *parent = dev->parent;
2282
2283         pm_runtime_put(dev);
2284         if (parent)
2285                 pm_runtime_put_sync(parent);
2286 }
2287
2288 /**
2289  * pci_bridge_d3_possible - Is it possible to put the bridge into D3
2290  * @bridge: Bridge to check
2291  *
2292  * This function checks if it is possible to move the bridge to D3.
2293  * Currently we only allow D3 for recent enough PCIe ports.
2294  */
2295 bool pci_bridge_d3_possible(struct pci_dev *bridge)
2296 {
2297         unsigned int year;
2298
2299         if (!pci_is_pcie(bridge))
2300                 return false;
2301
2302         switch (pci_pcie_type(bridge)) {
2303         case PCI_EXP_TYPE_ROOT_PORT:
2304         case PCI_EXP_TYPE_UPSTREAM:
2305         case PCI_EXP_TYPE_DOWNSTREAM:
2306                 if (pci_bridge_d3_disable)
2307                         return false;
2308
2309                 /*
2310                  * Hotplug interrupts cannot be delivered if the link is down,
2311                  * so parents of a hotplug port must stay awake. In addition,
2312                  * hotplug ports handled by firmware in System Management Mode
2313                  * may not be put into D3 by the OS (Thunderbolt on non-Macs).
2314                  * For simplicity, disallow in general for now.
2315                  */
2316                 if (bridge->is_hotplug_bridge)
2317                         return false;
2318
2319                 if (pci_bridge_d3_force)
2320                         return true;
2321
2322                 /*
2323                  * It should be safe to put PCIe ports from 2015 or newer
2324                  * to D3.
2325                  */
2326                 if (dmi_get_date(DMI_BIOS_DATE, &year, NULL, NULL) &&
2327                     year >= 2015) {
2328                         return true;
2329                 }
2330                 break;
2331         }
2332
2333         return false;
2334 }
2335
2336 static int pci_dev_check_d3cold(struct pci_dev *dev, void *data)
2337 {
2338         bool *d3cold_ok = data;
2339
2340         if (/* The device needs to be allowed to go D3cold ... */
2341             dev->no_d3cold || !dev->d3cold_allowed ||
2342
2343             /* ... and if it is wakeup capable to do so from D3cold. */
2344             (device_may_wakeup(&dev->dev) &&
2345              !pci_pme_capable(dev, PCI_D3cold)) ||
2346
2347             /* If it is a bridge it must be allowed to go to D3. */
2348             !pci_power_manageable(dev))
2349
2350                 *d3cold_ok = false;
2351
2352         return !*d3cold_ok;
2353 }
2354
2355 /*
2356  * pci_bridge_d3_update - Update bridge D3 capabilities
2357  * @dev: PCI device which is changed
2358  *
2359  * Update upstream bridge PM capabilities accordingly depending on if the
2360  * device PM configuration was changed or the device is being removed.  The
2361  * change is also propagated upstream.
2362  */
2363 void pci_bridge_d3_update(struct pci_dev *dev)
2364 {
2365         bool remove = !device_is_registered(&dev->dev);
2366         struct pci_dev *bridge;
2367         bool d3cold_ok = true;
2368
2369         bridge = pci_upstream_bridge(dev);
2370         if (!bridge || !pci_bridge_d3_possible(bridge))
2371                 return;
2372
2373         /*
2374          * If D3 is currently allowed for the bridge, removing one of its
2375          * children won't change that.
2376          */
2377         if (remove && bridge->bridge_d3)
2378                 return;
2379
2380         /*
2381          * If D3 is currently allowed for the bridge and a child is added or
2382          * changed, disallowance of D3 can only be caused by that child, so
2383          * we only need to check that single device, not any of its siblings.
2384          *
2385          * If D3 is currently not allowed for the bridge, checking the device
2386          * first may allow us to skip checking its siblings.
2387          */
2388         if (!remove)
2389                 pci_dev_check_d3cold(dev, &d3cold_ok);
2390
2391         /*
2392          * If D3 is currently not allowed for the bridge, this may be caused
2393          * either by the device being changed/removed or any of its siblings,
2394          * so we need to go through all children to find out if one of them
2395          * continues to block D3.
2396          */
2397         if (d3cold_ok && !bridge->bridge_d3)
2398                 pci_walk_bus(bridge->subordinate, pci_dev_check_d3cold,
2399                              &d3cold_ok);
2400
2401         if (bridge->bridge_d3 != d3cold_ok) {
2402                 bridge->bridge_d3 = d3cold_ok;
2403                 /* Propagate change to upstream bridges */
2404                 pci_bridge_d3_update(bridge);
2405         }
2406 }
2407
2408 /**
2409  * pci_d3cold_enable - Enable D3cold for device
2410  * @dev: PCI device to handle
2411  *
2412  * This function can be used in drivers to enable D3cold from the device
2413  * they handle.  It also updates upstream PCI bridge PM capabilities
2414  * accordingly.
2415  */
2416 void pci_d3cold_enable(struct pci_dev *dev)
2417 {
2418         if (dev->no_d3cold) {
2419                 dev->no_d3cold = false;
2420                 pci_bridge_d3_update(dev);
2421         }
2422 }
2423 EXPORT_SYMBOL_GPL(pci_d3cold_enable);
2424
2425 /**
2426  * pci_d3cold_disable - Disable D3cold for device
2427  * @dev: PCI device to handle
2428  *
2429  * This function can be used in drivers to disable D3cold from the device
2430  * they handle.  It also updates upstream PCI bridge PM capabilities
2431  * accordingly.
2432  */
2433 void pci_d3cold_disable(struct pci_dev *dev)
2434 {
2435         if (!dev->no_d3cold) {
2436                 dev->no_d3cold = true;
2437                 pci_bridge_d3_update(dev);
2438         }
2439 }
2440 EXPORT_SYMBOL_GPL(pci_d3cold_disable);
2441
2442 /**
2443  * pci_pm_init - Initialize PM functions of given PCI device
2444  * @dev: PCI device to handle.
2445  */
2446 void pci_pm_init(struct pci_dev *dev)
2447 {
2448         int pm;
2449         u16 pmc;
2450
2451         pm_runtime_forbid(&dev->dev);
2452         pm_runtime_set_active(&dev->dev);
2453         pm_runtime_enable(&dev->dev);
2454         device_enable_async_suspend(&dev->dev);
2455         dev->wakeup_prepared = false;
2456
2457         dev->pm_cap = 0;
2458         dev->pme_support = 0;
2459
2460         /* find PCI PM capability in list */
2461         pm = pci_find_capability(dev, PCI_CAP_ID_PM);
2462         if (!pm)
2463                 return;
2464         /* Check device's ability to generate PME# */
2465         pci_read_config_word(dev, pm + PCI_PM_PMC, &pmc);
2466
2467         if ((pmc & PCI_PM_CAP_VER_MASK) > 3) {
2468                 dev_err(&dev->dev, "unsupported PM cap regs version (%u)\n",
2469                         pmc & PCI_PM_CAP_VER_MASK);
2470                 return;
2471         }
2472
2473         dev->pm_cap = pm;
2474         dev->d3_delay = PCI_PM_D3_WAIT;
2475         dev->d3cold_delay = PCI_PM_D3COLD_WAIT;
2476         dev->bridge_d3 = pci_bridge_d3_possible(dev);
2477         dev->d3cold_allowed = true;
2478
2479         dev->d1_support = false;
2480         dev->d2_support = false;
2481         if (!pci_no_d1d2(dev)) {
2482                 if (pmc & PCI_PM_CAP_D1)
2483                         dev->d1_support = true;
2484                 if (pmc & PCI_PM_CAP_D2)
2485                         dev->d2_support = true;
2486
2487                 if (dev->d1_support || dev->d2_support)
2488                         dev_printk(KERN_DEBUG, &dev->dev, "supports%s%s\n",
2489                                    dev->d1_support ? " D1" : "",
2490                                    dev->d2_support ? " D2" : "");
2491         }
2492
2493         pmc &= PCI_PM_CAP_PME_MASK;
2494         if (pmc) {
2495                 dev_printk(KERN_DEBUG, &dev->dev,
2496                          "PME# supported from%s%s%s%s%s\n",
2497                          (pmc & PCI_PM_CAP_PME_D0) ? " D0" : "",
2498                          (pmc & PCI_PM_CAP_PME_D1) ? " D1" : "",
2499                          (pmc & PCI_PM_CAP_PME_D2) ? " D2" : "",
2500                          (pmc & PCI_PM_CAP_PME_D3) ? " D3hot" : "",
2501                          (pmc & PCI_PM_CAP_PME_D3cold) ? " D3cold" : "");
2502                 dev->pme_support = pmc >> PCI_PM_CAP_PME_SHIFT;
2503                 dev->pme_poll = true;
2504                 /*
2505                  * Make device's PM flags reflect the wake-up capability, but
2506                  * let the user space enable it to wake up the system as needed.
2507                  */
2508                 device_set_wakeup_capable(&dev->dev, true);
2509                 /* Disable the PME# generation functionality */
2510                 pci_pme_active(dev, false);
2511         }
2512 }
2513
2514 static unsigned long pci_ea_flags(struct pci_dev *dev, u8 prop)
2515 {
2516         unsigned long flags = IORESOURCE_PCI_FIXED | IORESOURCE_PCI_EA_BEI;
2517
2518         switch (prop) {
2519         case PCI_EA_P_MEM:
2520         case PCI_EA_P_VF_MEM:
2521                 flags |= IORESOURCE_MEM;
2522                 break;
2523         case PCI_EA_P_MEM_PREFETCH:
2524         case PCI_EA_P_VF_MEM_PREFETCH:
2525                 flags |= IORESOURCE_MEM | IORESOURCE_PREFETCH;
2526                 break;
2527         case PCI_EA_P_IO:
2528                 flags |= IORESOURCE_IO;
2529                 break;
2530         default:
2531                 return 0;
2532         }
2533
2534         return flags;
2535 }
2536
2537 static struct resource *pci_ea_get_resource(struct pci_dev *dev, u8 bei,
2538                                             u8 prop)
2539 {
2540         if (bei <= PCI_EA_BEI_BAR5 && prop <= PCI_EA_P_IO)
2541                 return &dev->resource[bei];
2542 #ifdef CONFIG_PCI_IOV
2543         else if (bei >= PCI_EA_BEI_VF_BAR0 && bei <= PCI_EA_BEI_VF_BAR5 &&
2544                  (prop == PCI_EA_P_VF_MEM || prop == PCI_EA_P_VF_MEM_PREFETCH))
2545                 return &dev->resource[PCI_IOV_RESOURCES +
2546                                       bei - PCI_EA_BEI_VF_BAR0];
2547 #endif
2548         else if (bei == PCI_EA_BEI_ROM)
2549                 return &dev->resource[PCI_ROM_RESOURCE];
2550         else
2551                 return NULL;
2552 }
2553
2554 /* Read an Enhanced Allocation (EA) entry */
2555 static int pci_ea_read(struct pci_dev *dev, int offset)
2556 {
2557         struct resource *res;
2558         int ent_size, ent_offset = offset;
2559         resource_size_t start, end;
2560         unsigned long flags;
2561         u32 dw0, bei, base, max_offset;
2562         u8 prop;
2563         bool support_64 = (sizeof(resource_size_t) >= 8);
2564
2565         pci_read_config_dword(dev, ent_offset, &dw0);
2566         ent_offset += 4;
2567
2568         /* Entry size field indicates DWORDs after 1st */
2569         ent_size = ((dw0 & PCI_EA_ES) + 1) << 2;
2570
2571         if (!(dw0 & PCI_EA_ENABLE)) /* Entry not enabled */
2572                 goto out;
2573
2574         bei = (dw0 & PCI_EA_BEI) >> 4;
2575         prop = (dw0 & PCI_EA_PP) >> 8;
2576
2577         /*
2578          * If the Property is in the reserved range, try the Secondary
2579          * Property instead.
2580          */
2581         if (prop > PCI_EA_P_BRIDGE_IO && prop < PCI_EA_P_MEM_RESERVED)
2582                 prop = (dw0 & PCI_EA_SP) >> 16;
2583         if (prop > PCI_EA_P_BRIDGE_IO)
2584                 goto out;
2585
2586         res = pci_ea_get_resource(dev, bei, prop);
2587         if (!res) {
2588                 dev_err(&dev->dev, "Unsupported EA entry BEI: %u\n", bei);
2589                 goto out;
2590         }
2591
2592         flags = pci_ea_flags(dev, prop);
2593         if (!flags) {
2594                 dev_err(&dev->dev, "Unsupported EA properties: %#x\n", prop);
2595                 goto out;
2596         }
2597
2598         /* Read Base */
2599         pci_read_config_dword(dev, ent_offset, &base);
2600         start = (base & PCI_EA_FIELD_MASK);
2601         ent_offset += 4;
2602
2603         /* Read MaxOffset */
2604         pci_read_config_dword(dev, ent_offset, &max_offset);
2605         ent_offset += 4;
2606
2607         /* Read Base MSBs (if 64-bit entry) */
2608         if (base & PCI_EA_IS_64) {
2609                 u32 base_upper;
2610
2611                 pci_read_config_dword(dev, ent_offset, &base_upper);
2612                 ent_offset += 4;
2613
2614                 flags |= IORESOURCE_MEM_64;
2615
2616                 /* entry starts above 32-bit boundary, can't use */
2617                 if (!support_64 && base_upper)
2618                         goto out;
2619
2620                 if (support_64)
2621                         start |= ((u64)base_upper << 32);
2622         }
2623
2624         end = start + (max_offset | 0x03);
2625
2626         /* Read MaxOffset MSBs (if 64-bit entry) */
2627         if (max_offset & PCI_EA_IS_64) {
2628                 u32 max_offset_upper;
2629
2630                 pci_read_config_dword(dev, ent_offset, &max_offset_upper);
2631                 ent_offset += 4;
2632
2633                 flags |= IORESOURCE_MEM_64;
2634
2635                 /* entry too big, can't use */
2636                 if (!support_64 && max_offset_upper)
2637                         goto out;
2638
2639                 if (support_64)
2640                         end += ((u64)max_offset_upper << 32);
2641         }
2642
2643         if (end < start) {
2644                 dev_err(&dev->dev, "EA Entry crosses address boundary\n");
2645                 goto out;
2646         }
2647
2648         if (ent_size != ent_offset - offset) {
2649                 dev_err(&dev->dev,
2650                         "EA Entry Size (%d) does not match length read (%d)\n",
2651                         ent_size, ent_offset - offset);
2652                 goto out;
2653         }
2654
2655         res->name = pci_name(dev);
2656         res->start = start;
2657         res->end = end;
2658         res->flags = flags;
2659
2660         if (bei <= PCI_EA_BEI_BAR5)
2661                 dev_printk(KERN_DEBUG, &dev->dev, "BAR %d: %pR (from Enhanced Allocation, properties %#02x)\n",
2662                            bei, res, prop);
2663         else if (bei == PCI_EA_BEI_ROM)
2664                 dev_printk(KERN_DEBUG, &dev->dev, "ROM: %pR (from Enhanced Allocation, properties %#02x)\n",
2665                            res, prop);
2666         else if (bei >= PCI_EA_BEI_VF_BAR0 && bei <= PCI_EA_BEI_VF_BAR5)
2667                 dev_printk(KERN_DEBUG, &dev->dev, "VF BAR %d: %pR (from Enhanced Allocation, properties %#02x)\n",
2668                            bei - PCI_EA_BEI_VF_BAR0, res, prop);
2669         else
2670                 dev_printk(KERN_DEBUG, &dev->dev, "BEI %d res: %pR (from Enhanced Allocation, properties %#02x)\n",
2671                            bei, res, prop);
2672
2673 out:
2674         return offset + ent_size;
2675 }
2676
2677 /* Enhanced Allocation Initialization */
2678 void pci_ea_init(struct pci_dev *dev)
2679 {
2680         int ea;
2681         u8 num_ent;
2682         int offset;
2683         int i;
2684
2685         /* find PCI EA capability in list */
2686         ea = pci_find_capability(dev, PCI_CAP_ID_EA);
2687         if (!ea)
2688                 return;
2689
2690         /* determine the number of entries */
2691         pci_bus_read_config_byte(dev->bus, dev->devfn, ea + PCI_EA_NUM_ENT,
2692                                         &num_ent);
2693         num_ent &= PCI_EA_NUM_ENT_MASK;
2694
2695         offset = ea + PCI_EA_FIRST_ENT;
2696
2697         /* Skip DWORD 2 for type 1 functions */
2698         if (dev->hdr_type == PCI_HEADER_TYPE_BRIDGE)
2699                 offset += 4;
2700
2701         /* parse each EA entry */
2702         for (i = 0; i < num_ent; ++i)
2703                 offset = pci_ea_read(dev, offset);
2704 }
2705
2706 static void pci_add_saved_cap(struct pci_dev *pci_dev,
2707         struct pci_cap_saved_state *new_cap)
2708 {
2709         hlist_add_head(&new_cap->next, &pci_dev->saved_cap_space);
2710 }
2711
2712 /**
2713  * _pci_add_cap_save_buffer - allocate buffer for saving given
2714  *                            capability registers
2715  * @dev: the PCI device
2716  * @cap: the capability to allocate the buffer for
2717  * @extended: Standard or Extended capability ID
2718  * @size: requested size of the buffer
2719  */
2720 static int _pci_add_cap_save_buffer(struct pci_dev *dev, u16 cap,
2721                                     bool extended, unsigned int size)
2722 {
2723         int pos;
2724         struct pci_cap_saved_state *save_state;
2725
2726         if (extended)
2727                 pos = pci_find_ext_capability(dev, cap);
2728         else
2729                 pos = pci_find_capability(dev, cap);
2730
2731         if (!pos)
2732                 return 0;
2733
2734         save_state = kzalloc(sizeof(*save_state) + size, GFP_KERNEL);
2735         if (!save_state)
2736                 return -ENOMEM;
2737
2738         save_state->cap.cap_nr = cap;
2739         save_state->cap.cap_extended = extended;
2740         save_state->cap.size = size;
2741         pci_add_saved_cap(dev, save_state);
2742
2743         return 0;
2744 }
2745
2746 int pci_add_cap_save_buffer(struct pci_dev *dev, char cap, unsigned int size)
2747 {
2748         return _pci_add_cap_save_buffer(dev, cap, false, size);
2749 }
2750
2751 int pci_add_ext_cap_save_buffer(struct pci_dev *dev, u16 cap, unsigned int size)
2752 {
2753         return _pci_add_cap_save_buffer(dev, cap, true, size);
2754 }
2755
2756 /**
2757  * pci_allocate_cap_save_buffers - allocate buffers for saving capabilities
2758  * @dev: the PCI device
2759  */
2760 void pci_allocate_cap_save_buffers(struct pci_dev *dev)
2761 {
2762         int error;
2763
2764         error = pci_add_cap_save_buffer(dev, PCI_CAP_ID_EXP,
2765                                         PCI_EXP_SAVE_REGS * sizeof(u16));
2766         if (error)
2767                 dev_err(&dev->dev,
2768                         "unable to preallocate PCI Express save buffer\n");
2769
2770         error = pci_add_cap_save_buffer(dev, PCI_CAP_ID_PCIX, sizeof(u16));
2771         if (error)
2772                 dev_err(&dev->dev,
2773                         "unable to preallocate PCI-X save buffer\n");
2774
2775         pci_allocate_vc_save_buffers(dev);
2776 }
2777
2778 void pci_free_cap_save_buffers(struct pci_dev *dev)
2779 {
2780         struct pci_cap_saved_state *tmp;
2781         struct hlist_node *n;
2782
2783         hlist_for_each_entry_safe(tmp, n, &dev->saved_cap_space, next)
2784                 kfree(tmp);
2785 }
2786
2787 /**
2788  * pci_configure_ari - enable or disable ARI forwarding
2789  * @dev: the PCI device
2790  *
2791  * If @dev and its upstream bridge both support ARI, enable ARI in the
2792  * bridge.  Otherwise, disable ARI in the bridge.
2793  */
2794 void pci_configure_ari(struct pci_dev *dev)
2795 {
2796         u32 cap;
2797         struct pci_dev *bridge;
2798
2799         if (pcie_ari_disabled || !pci_is_pcie(dev) || dev->devfn)
2800                 return;
2801
2802         bridge = dev->bus->self;
2803         if (!bridge)
2804                 return;
2805
2806         pcie_capability_read_dword(bridge, PCI_EXP_DEVCAP2, &cap);
2807         if (!(cap & PCI_EXP_DEVCAP2_ARI))
2808                 return;
2809
2810         if (pci_find_ext_capability(dev, PCI_EXT_CAP_ID_ARI)) {
2811                 pcie_capability_set_word(bridge, PCI_EXP_DEVCTL2,
2812                                          PCI_EXP_DEVCTL2_ARI);
2813                 bridge->ari_enabled = 1;
2814         } else {
2815                 pcie_capability_clear_word(bridge, PCI_EXP_DEVCTL2,
2816                                            PCI_EXP_DEVCTL2_ARI);
2817                 bridge->ari_enabled = 0;
2818         }
2819 }
2820
2821 static int pci_acs_enable;
2822
2823 /**
2824  * pci_request_acs - ask for ACS to be enabled if supported
2825  */
2826 void pci_request_acs(void)
2827 {
2828         pci_acs_enable = 1;
2829 }
2830
2831 /**
2832  * pci_std_enable_acs - enable ACS on devices using standard ACS capabilites
2833  * @dev: the PCI device
2834  */
2835 static void pci_std_enable_acs(struct pci_dev *dev)
2836 {
2837         int pos;
2838         u16 cap;
2839         u16 ctrl;
2840
2841         pos = pci_find_ext_capability(dev, PCI_EXT_CAP_ID_ACS);
2842         if (!pos)
2843                 return;
2844
2845         pci_read_config_word(dev, pos + PCI_ACS_CAP, &cap);
2846         pci_read_config_word(dev, pos + PCI_ACS_CTRL, &ctrl);
2847
2848         /* Source Validation */
2849         ctrl |= (cap & PCI_ACS_SV);
2850
2851         /* P2P Request Redirect */
2852         ctrl |= (cap & PCI_ACS_RR);
2853
2854         /* P2P Completion Redirect */
2855         ctrl |= (cap & PCI_ACS_CR);
2856
2857         /* Upstream Forwarding */
2858         ctrl |= (cap & PCI_ACS_UF);
2859
2860         pci_write_config_word(dev, pos + PCI_ACS_CTRL, ctrl);
2861 }
2862
2863 /**
2864  * pci_enable_acs - enable ACS if hardware support it
2865  * @dev: the PCI device
2866  */
2867 void pci_enable_acs(struct pci_dev *dev)
2868 {
2869         if (!pci_acs_enable)
2870                 return;
2871
2872         if (!pci_dev_specific_enable_acs(dev))
2873                 return;
2874
2875         pci_std_enable_acs(dev);
2876 }
2877
2878 static bool pci_acs_flags_enabled(struct pci_dev *pdev, u16 acs_flags)
2879 {
2880         int pos;
2881         u16 cap, ctrl;
2882
2883         pos = pci_find_ext_capability(pdev, PCI_EXT_CAP_ID_ACS);
2884         if (!pos)
2885                 return false;
2886
2887         /*
2888          * Except for egress control, capabilities are either required
2889          * or only required if controllable.  Features missing from the
2890          * capability field can therefore be assumed as hard-wired enabled.
2891          */
2892         pci_read_config_word(pdev, pos + PCI_ACS_CAP, &cap);
2893         acs_flags &= (cap | PCI_ACS_EC);
2894
2895         pci_read_config_word(pdev, pos + PCI_ACS_CTRL, &ctrl);
2896         return (ctrl & acs_flags) == acs_flags;
2897 }
2898
2899 /**
2900  * pci_acs_enabled - test ACS against required flags for a given device
2901  * @pdev: device to test
2902  * @acs_flags: required PCI ACS flags
2903  *
2904  * Return true if the device supports the provided flags.  Automatically
2905  * filters out flags that are not implemented on multifunction devices.
2906  *
2907  * Note that this interface checks the effective ACS capabilities of the
2908  * device rather than the actual capabilities.  For instance, most single
2909  * function endpoints are not required to support ACS because they have no
2910  * opportunity for peer-to-peer access.  We therefore return 'true'
2911  * regardless of whether the device exposes an ACS capability.  This makes
2912  * it much easier for callers of this function to ignore the actual type
2913  * or topology of the device when testing ACS support.
2914  */
2915 bool pci_acs_enabled(struct pci_dev *pdev, u16 acs_flags)
2916 {
2917         int ret;
2918
2919         ret = pci_dev_specific_acs_enabled(pdev, acs_flags);
2920         if (ret >= 0)
2921                 return ret > 0;
2922
2923         /*
2924          * Conventional PCI and PCI-X devices never support ACS, either
2925          * effectively or actually.  The shared bus topology implies that
2926          * any device on the bus can receive or snoop DMA.
2927          */
2928         if (!pci_is_pcie(pdev))
2929                 return false;
2930
2931         switch (pci_pcie_type(pdev)) {
2932         /*
2933          * PCI/X-to-PCIe bridges are not specifically mentioned by the spec,
2934          * but since their primary interface is PCI/X, we conservatively
2935          * handle them as we would a non-PCIe device.
2936          */
2937         case PCI_EXP_TYPE_PCIE_BRIDGE:
2938         /*
2939          * PCIe 3.0, 6.12.1 excludes ACS on these devices.  "ACS is never
2940          * applicable... must never implement an ACS Extended Capability...".
2941          * This seems arbitrary, but we take a conservative interpretation
2942          * of this statement.
2943          */
2944         case PCI_EXP_TYPE_PCI_BRIDGE:
2945         case PCI_EXP_TYPE_RC_EC:
2946                 return false;
2947         /*
2948          * PCIe 3.0, 6.12.1.1 specifies that downstream and root ports should
2949          * implement ACS in order to indicate their peer-to-peer capabilities,
2950          * regardless of whether they are single- or multi-function devices.
2951          */
2952         case PCI_EXP_TYPE_DOWNSTREAM:
2953         case PCI_EXP_TYPE_ROOT_PORT:
2954                 return pci_acs_flags_enabled(pdev, acs_flags);
2955         /*
2956          * PCIe 3.0, 6.12.1.2 specifies ACS capabilities that should be
2957          * implemented by the remaining PCIe types to indicate peer-to-peer
2958          * capabilities, but only when they are part of a multifunction
2959          * device.  The footnote for section 6.12 indicates the specific
2960          * PCIe types included here.
2961          */
2962         case PCI_EXP_TYPE_ENDPOINT:
2963         case PCI_EXP_TYPE_UPSTREAM:
2964         case PCI_EXP_TYPE_LEG_END:
2965         case PCI_EXP_TYPE_RC_END:
2966                 if (!pdev->multifunction)
2967                         break;
2968
2969                 return pci_acs_flags_enabled(pdev, acs_flags);
2970         }
2971
2972         /*
2973          * PCIe 3.0, 6.12.1.3 specifies no ACS capabilities are applicable
2974          * to single function devices with the exception of downstream ports.
2975          */
2976         return true;
2977 }
2978
2979 /**
2980  * pci_acs_path_enable - test ACS flags from start to end in a hierarchy
2981  * @start: starting downstream device
2982  * @end: ending upstream device or NULL to search to the root bus
2983  * @acs_flags: required flags
2984  *
2985  * Walk up a device tree from start to end testing PCI ACS support.  If
2986  * any step along the way does not support the required flags, return false.
2987  */
2988 bool pci_acs_path_enabled(struct pci_dev *start,
2989                           struct pci_dev *end, u16 acs_flags)
2990 {
2991         struct pci_dev *pdev, *parent = start;
2992
2993         do {
2994                 pdev = parent;
2995
2996                 if (!pci_acs_enabled(pdev, acs_flags))
2997                         return false;
2998
2999                 if (pci_is_root_bus(pdev->bus))
3000                         return (end == NULL);
3001
3002                 parent = pdev->bus->self;
3003         } while (pdev != end);
3004
3005         return true;
3006 }
3007
3008 /**
3009  * pci_swizzle_interrupt_pin - swizzle INTx for device behind bridge
3010  * @dev: the PCI device
3011  * @pin: the INTx pin (1=INTA, 2=INTB, 3=INTC, 4=INTD)
3012  *
3013  * Perform INTx swizzling for a device behind one level of bridge.  This is
3014  * required by section 9.1 of the PCI-to-PCI bridge specification for devices
3015  * behind bridges on add-in cards.  For devices with ARI enabled, the slot
3016  * number is always 0 (see the Implementation Note in section 2.2.8.1 of
3017  * the PCI Express Base Specification, Revision 2.1)
3018  */
3019 u8 pci_swizzle_interrupt_pin(const struct pci_dev *dev, u8 pin)
3020 {
3021         int slot;
3022
3023         if (pci_ari_enabled(dev->bus))
3024                 slot = 0;
3025         else
3026                 slot = PCI_SLOT(dev->devfn);
3027
3028         return (((pin - 1) + slot) % 4) + 1;
3029 }
3030
3031 int pci_get_interrupt_pin(struct pci_dev *dev, struct pci_dev **bridge)
3032 {
3033         u8 pin;
3034
3035         pin = dev->pin;
3036         if (!pin)
3037                 return -1;
3038
3039         while (!pci_is_root_bus(dev->bus)) {
3040                 pin = pci_swizzle_interrupt_pin(dev, pin);
3041                 dev = dev->bus->self;
3042         }
3043         *bridge = dev;
3044         return pin;
3045 }
3046
3047 /**
3048  * pci_common_swizzle - swizzle INTx all the way to root bridge
3049  * @dev: the PCI device
3050  * @pinp: pointer to the INTx pin value (1=INTA, 2=INTB, 3=INTD, 4=INTD)
3051  *
3052  * Perform INTx swizzling for a device.  This traverses through all PCI-to-PCI
3053  * bridges all the way up to a PCI root bus.
3054  */
3055 u8 pci_common_swizzle(struct pci_dev *dev, u8 *pinp)
3056 {
3057         u8 pin = *pinp;
3058
3059         while (!pci_is_root_bus(dev->bus)) {
3060                 pin = pci_swizzle_interrupt_pin(dev, pin);
3061                 dev = dev->bus->self;
3062         }
3063         *pinp = pin;
3064         return PCI_SLOT(dev->devfn);
3065 }
3066 EXPORT_SYMBOL_GPL(pci_common_swizzle);
3067
3068 /**
3069  *      pci_release_region - Release a PCI bar
3070  *      @pdev: PCI device whose resources were previously reserved by pci_request_region
3071  *      @bar: BAR to release
3072  *
3073  *      Releases the PCI I/O and memory resources previously reserved by a
3074  *      successful call to pci_request_region.  Call this function only
3075  *      after all use of the PCI regions has ceased.
3076  */
3077 void pci_release_region(struct pci_dev *pdev, int bar)
3078 {
3079         struct pci_devres *dr;
3080
3081         if (pci_resource_len(pdev, bar) == 0)
3082                 return;
3083         if (pci_resource_flags(pdev, bar) & IORESOURCE_IO)
3084                 release_region(pci_resource_start(pdev, bar),
3085                                 pci_resource_len(pdev, bar));
3086         else if (pci_resource_flags(pdev, bar) & IORESOURCE_MEM)
3087                 release_mem_region(pci_resource_start(pdev, bar),
3088                                 pci_resource_len(pdev, bar));
3089
3090         dr = find_pci_dr(pdev);
3091         if (dr)
3092                 dr->region_mask &= ~(1 << bar);
3093 }
3094 EXPORT_SYMBOL(pci_release_region);
3095
3096 /**
3097  *      __pci_request_region - Reserved PCI I/O and memory resource
3098  *      @pdev: PCI device whose resources are to be reserved
3099  *      @bar: BAR to be reserved
3100  *      @res_name: Name to be associated with resource.
3101  *      @exclusive: whether the region access is exclusive or not
3102  *
3103  *      Mark the PCI region associated with PCI device @pdev BR @bar as
3104  *      being reserved by owner @res_name.  Do not access any
3105  *      address inside the PCI regions unless this call returns
3106  *      successfully.
3107  *
3108  *      If @exclusive is set, then the region is marked so that userspace
3109  *      is explicitly not allowed to map the resource via /dev/mem or
3110  *      sysfs MMIO access.
3111  *
3112  *      Returns 0 on success, or %EBUSY on error.  A warning
3113  *      message is also printed on failure.
3114  */
3115 static int __pci_request_region(struct pci_dev *pdev, int bar,
3116                                 const char *res_name, int exclusive)
3117 {
3118         struct pci_devres *dr;
3119
3120         if (pci_resource_len(pdev, bar) == 0)
3121                 return 0;
3122
3123         if (pci_resource_flags(pdev, bar) & IORESOURCE_IO) {
3124                 if (!request_region(pci_resource_start(pdev, bar),
3125                             pci_resource_len(pdev, bar), res_name))
3126                         goto err_out;
3127         } else if (pci_resource_flags(pdev, bar) & IORESOURCE_MEM) {
3128                 if (!__request_mem_region(pci_resource_start(pdev, bar),
3129                                         pci_resource_len(pdev, bar), res_name,
3130                                         exclusive))
3131                         goto err_out;
3132         }
3133
3134         dr = find_pci_dr(pdev);
3135         if (dr)
3136                 dr->region_mask |= 1 << bar;
3137
3138         return 0;
3139
3140 err_out:
3141         dev_warn(&pdev->dev, "BAR %d: can't reserve %pR\n", bar,
3142                  &pdev->resource[bar]);
3143         return -EBUSY;
3144 }
3145
3146 /**
3147  *      pci_request_region - Reserve PCI I/O and memory resource
3148  *      @pdev: PCI device whose resources are to be reserved
3149  *      @bar: BAR to be reserved
3150  *      @res_name: Name to be associated with resource
3151  *
3152  *      Mark the PCI region associated with PCI device @pdev BAR @bar as
3153  *      being reserved by owner @res_name.  Do not access any
3154  *      address inside the PCI regions unless this call returns
3155  *      successfully.
3156  *
3157  *      Returns 0 on success, or %EBUSY on error.  A warning
3158  *      message is also printed on failure.
3159  */
3160 int pci_request_region(struct pci_dev *pdev, int bar, const char *res_name)
3161 {
3162         return __pci_request_region(pdev, bar, res_name, 0);
3163 }
3164 EXPORT_SYMBOL(pci_request_region);
3165
3166 /**
3167  *      pci_request_region_exclusive - Reserved PCI I/O and memory resource
3168  *      @pdev: PCI device whose resources are to be reserved
3169  *      @bar: BAR to be reserved
3170  *      @res_name: Name to be associated with resource.
3171  *
3172  *      Mark the PCI region associated with PCI device @pdev BR @bar as
3173  *      being reserved by owner @res_name.  Do not access any
3174  *      address inside the PCI regions unless this call returns
3175  *      successfully.
3176  *
3177  *      Returns 0 on success, or %EBUSY on error.  A warning
3178  *      message is also printed on failure.
3179  *
3180  *      The key difference that _exclusive makes it that userspace is
3181  *      explicitly not allowed to map the resource via /dev/mem or
3182  *      sysfs.
3183  */
3184 int pci_request_region_exclusive(struct pci_dev *pdev, int bar,
3185                                  const char *res_name)
3186 {
3187         return __pci_request_region(pdev, bar, res_name, IORESOURCE_EXCLUSIVE);
3188 }
3189 EXPORT_SYMBOL(pci_request_region_exclusive);
3190
3191 /**
3192  * pci_release_selected_regions - Release selected PCI I/O and memory resources
3193  * @pdev: PCI device whose resources were previously reserved
3194  * @bars: Bitmask of BARs to be released
3195  *
3196  * Release selected PCI I/O and memory resources previously reserved.
3197  * Call this function only after all use of the PCI regions has ceased.
3198  */
3199 void pci_release_selected_regions(struct pci_dev *pdev, int bars)
3200 {
3201         int i;
3202
3203         for (i = 0; i < 6; i++)
3204                 if (bars & (1 << i))
3205                         pci_release_region(pdev, i);
3206 }
3207 EXPORT_SYMBOL(pci_release_selected_regions);
3208
3209 static int __pci_request_selected_regions(struct pci_dev *pdev, int bars,
3210                                           const char *res_name, int excl)
3211 {
3212         int i;
3213
3214         for (i = 0; i < 6; i++)
3215                 if (bars & (1 << i))
3216                         if (__pci_request_region(pdev, i, res_name, excl))
3217                                 goto err_out;
3218         return 0;
3219
3220 err_out:
3221         while (--i >= 0)
3222                 if (bars & (1 << i))
3223                         pci_release_region(pdev, i);
3224
3225         return -EBUSY;
3226 }
3227
3228
3229 /**
3230  * pci_request_selected_regions - Reserve selected PCI I/O and memory resources
3231  * @pdev: PCI device whose resources are to be reserved
3232  * @bars: Bitmask of BARs to be requested
3233  * @res_name: Name to be associated with resource
3234  */
3235 int pci_request_selected_regions(struct pci_dev *pdev, int bars,
3236                                  const char *res_name)
3237 {
3238         return __pci_request_selected_regions(pdev, bars, res_name, 0);
3239 }
3240 EXPORT_SYMBOL(pci_request_selected_regions);
3241
3242 int pci_request_selected_regions_exclusive(struct pci_dev *pdev, int bars,
3243                                            const char *res_name)
3244 {
3245         return __pci_request_selected_regions(pdev, bars, res_name,
3246                         IORESOURCE_EXCLUSIVE);
3247 }
3248 EXPORT_SYMBOL(pci_request_selected_regions_exclusive);
3249
3250 /**
3251  *      pci_release_regions - Release reserved PCI I/O and memory resources
3252  *      @pdev: PCI device whose resources were previously reserved by pci_request_regions
3253  *
3254  *      Releases all PCI I/O and memory resources previously reserved by a
3255  *      successful call to pci_request_regions.  Call this function only
3256  *      after all use of the PCI regions has ceased.
3257  */
3258
3259 void pci_release_regions(struct pci_dev *pdev)
3260 {
3261         pci_release_selected_regions(pdev, (1 << 6) - 1);
3262 }
3263 EXPORT_SYMBOL(pci_release_regions);
3264
3265 /**
3266  *      pci_request_regions - Reserved PCI I/O and memory resources
3267  *      @pdev: PCI device whose resources are to be reserved
3268  *      @res_name: Name to be associated with resource.
3269  *
3270  *      Mark all PCI regions associated with PCI device @pdev as
3271  *      being reserved by owner @res_name.  Do not access any
3272  *      address inside the PCI regions unless this call returns
3273  *      successfully.
3274  *
3275  *      Returns 0 on success, or %EBUSY on error.  A warning
3276  *      message is also printed on failure.
3277  */
3278 int pci_request_regions(struct pci_dev *pdev, const char *res_name)
3279 {
3280         return pci_request_selected_regions(pdev, ((1 << 6) - 1), res_name);
3281 }
3282 EXPORT_SYMBOL(pci_request_regions);
3283
3284 /**
3285  *      pci_request_regions_exclusive - Reserved PCI I/O and memory resources
3286  *      @pdev: PCI device whose resources are to be reserved
3287  *      @res_name: Name to be associated with resource.
3288  *
3289  *      Mark all PCI regions associated with PCI device @pdev as
3290  *      being reserved by owner @res_name.  Do not access any
3291  *      address inside the PCI regions unless this call returns
3292  *      successfully.
3293  *
3294  *      pci_request_regions_exclusive() will mark the region so that
3295  *      /dev/mem and the sysfs MMIO access will not be allowed.
3296  *
3297  *      Returns 0 on success, or %EBUSY on error.  A warning
3298  *      message is also printed on failure.
3299  */
3300 int pci_request_regions_exclusive(struct pci_dev *pdev, const char *res_name)
3301 {
3302         return pci_request_selected_regions_exclusive(pdev,
3303                                         ((1 << 6) - 1), res_name);
3304 }
3305 EXPORT_SYMBOL(pci_request_regions_exclusive);
3306
3307 #ifdef PCI_IOBASE
3308 struct io_range {
3309         struct list_head list;
3310         phys_addr_t start;
3311         resource_size_t size;
3312 };
3313
3314 static LIST_HEAD(io_range_list);
3315 static DEFINE_SPINLOCK(io_range_lock);
3316 #endif
3317
3318 /*
3319  * Record the PCI IO range (expressed as CPU physical address + size).
3320  * Return a negative value if an error has occured, zero otherwise
3321  */
3322 int __weak pci_register_io_range(phys_addr_t addr, resource_size_t size)
3323 {
3324         int err = 0;
3325
3326 #ifdef PCI_IOBASE
3327         struct io_range *range;
3328         resource_size_t allocated_size = 0;
3329
3330         /* check if the range hasn't been previously recorded */
3331         spin_lock(&io_range_lock);
3332         list_for_each_entry(range, &io_range_list, list) {
3333                 if (addr >= range->start && addr + size <= range->start + size) {
3334                         /* range already registered, bail out */
3335                         goto end_register;
3336                 }
3337                 allocated_size += range->size;
3338         }
3339
3340         /* range not registed yet, check for available space */
3341         if (allocated_size + size - 1 > IO_SPACE_LIMIT) {
3342                 /* if it's too big check if 64K space can be reserved */
3343                 if (allocated_size + SZ_64K - 1 > IO_SPACE_LIMIT) {
3344                         err = -E2BIG;
3345                         goto end_register;
3346                 }
3347
3348                 size = SZ_64K;
3349                 pr_warn("Requested IO range too big, new size set to 64K\n");
3350         }
3351
3352         /* add the range to the list */
3353         range = kzalloc(sizeof(*range), GFP_ATOMIC);
3354         if (!range) {
3355                 err = -ENOMEM;
3356                 goto end_register;
3357         }
3358
3359         range->start = addr;
3360         range->size = size;
3361
3362         list_add_tail(&range->list, &io_range_list);
3363
3364 end_register:
3365         spin_unlock(&io_range_lock);
3366 #endif
3367
3368         return err;
3369 }
3370
3371 phys_addr_t pci_pio_to_address(unsigned long pio)
3372 {
3373         phys_addr_t address = (phys_addr_t)OF_BAD_ADDR;
3374
3375 #ifdef PCI_IOBASE
3376         struct io_range *range;
3377         resource_size_t allocated_size = 0;
3378
3379         if (pio > IO_SPACE_LIMIT)
3380                 return address;
3381
3382         spin_lock(&io_range_lock);
3383         list_for_each_entry(range, &io_range_list, list) {
3384                 if (pio >= allocated_size && pio < allocated_size + range->size) {
3385                         address = range->start + pio - allocated_size;
3386                         break;
3387                 }
3388                 allocated_size += range->size;
3389         }
3390         spin_unlock(&io_range_lock);
3391 #endif
3392
3393         return address;
3394 }
3395
3396 unsigned long __weak pci_address_to_pio(phys_addr_t address)
3397 {
3398 #ifdef PCI_IOBASE
3399         struct io_range *res;
3400         resource_size_t offset = 0;
3401         unsigned long addr = -1;
3402
3403         spin_lock(&io_range_lock);
3404         list_for_each_entry(res, &io_range_list, list) {
3405                 if (address >= res->start && address < res->start + res->size) {
3406                         addr = address - res->start + offset;
3407                         break;
3408                 }
3409                 offset += res->size;
3410         }
3411         spin_unlock(&io_range_lock);
3412
3413         return addr;
3414 #else
3415         if (address > IO_SPACE_LIMIT)
3416                 return (unsigned long)-1;
3417
3418         return (unsigned long) address;
3419 #endif
3420 }
3421
3422 /**
3423  *      pci_remap_iospace - Remap the memory mapped I/O space
3424  *      @res: Resource describing the I/O space
3425  *      @phys_addr: physical address of range to be mapped
3426  *
3427  *      Remap the memory mapped I/O space described by the @res
3428  *      and the CPU physical address @phys_addr into virtual address space.
3429  *      Only architectures that have memory mapped IO functions defined
3430  *      (and the PCI_IOBASE value defined) should call this function.
3431  */
3432 int pci_remap_iospace(const struct resource *res, phys_addr_t phys_addr)
3433 {
3434 #if defined(PCI_IOBASE) && defined(CONFIG_MMU)
3435         unsigned long vaddr = (unsigned long)PCI_IOBASE + res->start;
3436
3437         if (!(res->flags & IORESOURCE_IO))
3438                 return -EINVAL;
3439
3440         if (res->end > IO_SPACE_LIMIT)
3441                 return -EINVAL;
3442
3443         return ioremap_page_range(vaddr, vaddr + resource_size(res), phys_addr,
3444                                   pgprot_device(PAGE_KERNEL));
3445 #else
3446         /* this architecture does not have memory mapped I/O space,
3447            so this function should never be called */
3448         WARN_ONCE(1, "This architecture does not support memory mapped I/O\n");
3449         return -ENODEV;
3450 #endif
3451 }
3452 EXPORT_SYMBOL(pci_remap_iospace);
3453
3454 /**
3455  *      pci_unmap_iospace - Unmap the memory mapped I/O space
3456  *      @res: resource to be unmapped
3457  *
3458  *      Unmap the CPU virtual address @res from virtual address space.
3459  *      Only architectures that have memory mapped IO functions defined
3460  *      (and the PCI_IOBASE value defined) should call this function.
3461  */
3462 void pci_unmap_iospace(struct resource *res)
3463 {
3464 #if defined(PCI_IOBASE) && defined(CONFIG_MMU)
3465         unsigned long vaddr = (unsigned long)PCI_IOBASE + res->start;
3466
3467         unmap_kernel_range(vaddr, resource_size(res));
3468 #endif
3469 }
3470 EXPORT_SYMBOL(pci_unmap_iospace);
3471
3472 static void devm_pci_unmap_iospace(struct device *dev, void *ptr)
3473 {
3474         struct resource **res = ptr;
3475
3476         pci_unmap_iospace(*res);
3477 }
3478
3479 /**
3480  * devm_pci_remap_iospace - Managed pci_remap_iospace()
3481  * @dev: Generic device to remap IO address for
3482  * @res: Resource describing the I/O space
3483  * @phys_addr: physical address of range to be mapped
3484  *
3485  * Managed pci_remap_iospace().  Map is automatically unmapped on driver
3486  * detach.
3487  */
3488 int devm_pci_remap_iospace(struct device *dev, const struct resource *res,
3489                            phys_addr_t phys_addr)
3490 {
3491         const struct resource **ptr;
3492         int error;
3493
3494         ptr = devres_alloc(devm_pci_unmap_iospace, sizeof(*ptr), GFP_KERNEL);
3495         if (!ptr)
3496                 return -ENOMEM;
3497
3498         error = pci_remap_iospace(res, phys_addr);
3499         if (error) {
3500                 devres_free(ptr);
3501         } else  {
3502                 *ptr = res;
3503                 devres_add(dev, ptr);
3504         }
3505
3506         return error;
3507 }
3508 EXPORT_SYMBOL(devm_pci_remap_iospace);
3509
3510 /**
3511  * devm_pci_remap_cfgspace - Managed pci_remap_cfgspace()
3512  * @dev: Generic device to remap IO address for
3513  * @offset: Resource address to map
3514  * @size: Size of map
3515  *
3516  * Managed pci_remap_cfgspace().  Map is automatically unmapped on driver
3517  * detach.
3518  */
3519 void __iomem *devm_pci_remap_cfgspace(struct device *dev,
3520                                       resource_size_t offset,
3521                                       resource_size_t size)
3522 {
3523         void __iomem **ptr, *addr;
3524
3525         ptr = devres_alloc(devm_ioremap_release, sizeof(*ptr), GFP_KERNEL);
3526         if (!ptr)
3527                 return NULL;
3528
3529         addr = pci_remap_cfgspace(offset, size);
3530         if (addr) {
3531                 *ptr = addr;
3532                 devres_add(dev, ptr);
3533         } else
3534                 devres_free(ptr);
3535
3536         return addr;
3537 }
3538 EXPORT_SYMBOL(devm_pci_remap_cfgspace);
3539
3540 /**
3541  * devm_pci_remap_cfg_resource - check, request region and ioremap cfg resource
3542  * @dev: generic device to handle the resource for
3543  * @res: configuration space resource to be handled
3544  *
3545  * Checks that a resource is a valid memory region, requests the memory
3546  * region and ioremaps with pci_remap_cfgspace() API that ensures the
3547  * proper PCI configuration space memory attributes are guaranteed.
3548  *
3549  * All operations are managed and will be undone on driver detach.
3550  *
3551  * Returns a pointer to the remapped memory or an ERR_PTR() encoded error code
3552  * on failure. Usage example:
3553  *
3554  *      res = platform_get_resource(pdev, IORESOURCE_MEM, 0);
3555  *      base = devm_pci_remap_cfg_resource(&pdev->dev, res);
3556  *      if (IS_ERR(base))
3557  *              return PTR_ERR(base);
3558  */
3559 void __iomem *devm_pci_remap_cfg_resource(struct device *dev,
3560                                           struct resource *res)
3561 {
3562         resource_size_t size;
3563         const char *name;
3564         void __iomem *dest_ptr;
3565
3566         BUG_ON(!dev);
3567
3568         if (!res || resource_type(res) != IORESOURCE_MEM) {
3569                 dev_err(dev, "invalid resource\n");
3570                 return IOMEM_ERR_PTR(-EINVAL);
3571         }
3572
3573         size = resource_size(res);
3574         name = res->name ?: dev_name(dev);
3575
3576         if (!devm_request_mem_region(dev, res->start, size, name)) {
3577                 dev_err(dev, "can't request region for resource %pR\n", res);
3578                 return IOMEM_ERR_PTR(-EBUSY);
3579         }
3580
3581         dest_ptr = devm_pci_remap_cfgspace(dev, res->start, size);
3582         if (!dest_ptr) {
3583                 dev_err(dev, "ioremap failed for resource %pR\n", res);
3584                 devm_release_mem_region(dev, res->start, size);
3585                 dest_ptr = IOMEM_ERR_PTR(-ENOMEM);
3586         }
3587
3588         return dest_ptr;
3589 }
3590 EXPORT_SYMBOL(devm_pci_remap_cfg_resource);
3591
3592 static void __pci_set_master(struct pci_dev *dev, bool enable)
3593 {
3594         u16 old_cmd, cmd;
3595
3596         pci_read_config_word(dev, PCI_COMMAND, &old_cmd);
3597         if (enable)
3598                 cmd = old_cmd | PCI_COMMAND_MASTER;
3599         else
3600                 cmd = old_cmd & ~PCI_COMMAND_MASTER;
3601         if (cmd != old_cmd) {
3602                 dev_dbg(&dev->dev, "%s bus mastering\n",
3603                         enable ? "enabling" : "disabling");
3604                 pci_write_config_word(dev, PCI_COMMAND, cmd);
3605         }
3606         dev->is_busmaster = enable;
3607 }
3608
3609 /**
3610  * pcibios_setup - process "pci=" kernel boot arguments
3611  * @str: string used to pass in "pci=" kernel boot arguments
3612  *
3613  * Process kernel boot arguments.  This is the default implementation.
3614  * Architecture specific implementations can override this as necessary.
3615  */
3616 char * __weak __init pcibios_setup(char *str)
3617 {
3618         return str;
3619 }
3620
3621 /**
3622  * pcibios_set_master - enable PCI bus-mastering for device dev
3623  * @dev: the PCI device to enable
3624  *
3625  * Enables PCI bus-mastering for the device.  This is the default
3626  * implementation.  Architecture specific implementations can override
3627  * this if necessary.
3628  */
3629 void __weak pcibios_set_master(struct pci_dev *dev)
3630 {
3631         u8 lat;
3632
3633         /* The latency timer doesn't apply to PCIe (either Type 0 or Type 1) */
3634         if (pci_is_pcie(dev))
3635                 return;
3636
3637         pci_read_config_byte(dev, PCI_LATENCY_TIMER, &lat);
3638         if (lat < 16)
3639                 lat = (64 <= pcibios_max_latency) ? 64 : pcibios_max_latency;
3640         else if (lat > pcibios_max_latency)
3641                 lat = pcibios_max_latency;
3642         else
3643                 return;
3644
3645         pci_write_config_byte(dev, PCI_LATENCY_TIMER, lat);
3646 }
3647
3648 /**
3649  * pci_set_master - enables bus-mastering for device dev
3650  * @dev: the PCI device to enable
3651  *
3652  * Enables bus-mastering on the device and calls pcibios_set_master()
3653  * to do the needed arch specific settings.
3654  */
3655 void pci_set_master(struct pci_dev *dev)
3656 {
3657         __pci_set_master(dev, true);
3658         pcibios_set_master(dev);
3659 }
3660 EXPORT_SYMBOL(pci_set_master);
3661
3662 /**
3663  * pci_clear_master - disables bus-mastering for device dev
3664  * @dev: the PCI device to disable
3665  */
3666 void pci_clear_master(struct pci_dev *dev)
3667 {
3668         __pci_set_master(dev, false);
3669 }
3670 EXPORT_SYMBOL(pci_clear_master);
3671
3672 /**
3673  * pci_set_cacheline_size - ensure the CACHE_LINE_SIZE register is programmed
3674  * @dev: the PCI device for which MWI is to be enabled
3675  *
3676  * Helper function for pci_set_mwi.
3677  * Originally copied from drivers/net/acenic.c.
3678  * Copyright 1998-2001 by Jes Sorensen, <jes@trained-monkey.org>.
3679  *
3680  * RETURNS: An appropriate -ERRNO error value on error, or zero for success.
3681  */
3682 int pci_set_cacheline_size(struct pci_dev *dev)
3683 {
3684         u8 cacheline_size;
3685
3686         if (!pci_cache_line_size)
3687                 return -EINVAL;
3688
3689         /* Validate current setting: the PCI_CACHE_LINE_SIZE must be
3690            equal to or multiple of the right value. */
3691         pci_read_config_byte(dev, PCI_CACHE_LINE_SIZE, &cacheline_size);
3692         if (cacheline_size >= pci_cache_line_size &&
3693             (cacheline_size % pci_cache_line_size) == 0)
3694                 return 0;
3695
3696         /* Write the correct value. */
3697         pci_write_config_byte(dev, PCI_CACHE_LINE_SIZE, pci_cache_line_size);
3698         /* Read it back. */
3699         pci_read_config_byte(dev, PCI_CACHE_LINE_SIZE, &cacheline_size);
3700         if (cacheline_size == pci_cache_line_size)
3701                 return 0;
3702
3703         dev_printk(KERN_DEBUG, &dev->dev, "cache line size of %d is not supported\n",
3704                    pci_cache_line_size << 2);
3705
3706         return -EINVAL;
3707 }
3708 EXPORT_SYMBOL_GPL(pci_set_cacheline_size);
3709
3710 /**
3711  * pci_set_mwi - enables memory-write-invalidate PCI transaction
3712  * @dev: the PCI device for which MWI is enabled
3713  *
3714  * Enables the Memory-Write-Invalidate transaction in %PCI_COMMAND.
3715  *
3716  * RETURNS: An appropriate -ERRNO error value on error, or zero for success.
3717  */
3718 int pci_set_mwi(struct pci_dev *dev)
3719 {
3720 #ifdef PCI_DISABLE_MWI
3721         return 0;
3722 #else
3723         int rc;
3724         u16 cmd;
3725
3726         rc = pci_set_cacheline_size(dev);
3727         if (rc)
3728                 return rc;
3729
3730         pci_read_config_word(dev, PCI_COMMAND, &cmd);
3731         if (!(cmd & PCI_COMMAND_INVALIDATE)) {
3732                 dev_dbg(&dev->dev, "enabling Mem-Wr-Inval\n");
3733                 cmd |= PCI_COMMAND_INVALIDATE;
3734                 pci_write_config_word(dev, PCI_COMMAND, cmd);
3735         }
3736         return 0;
3737 #endif
3738 }
3739 EXPORT_SYMBOL(pci_set_mwi);
3740
3741 /**
3742  * pci_try_set_mwi - enables memory-write-invalidate PCI transaction
3743  * @dev: the PCI device for which MWI is enabled
3744  *
3745  * Enables the Memory-Write-Invalidate transaction in %PCI_COMMAND.
3746  * Callers are not required to check the return value.
3747  *
3748  * RETURNS: An appropriate -ERRNO error value on error, or zero for success.
3749  */
3750 int pci_try_set_mwi(struct pci_dev *dev)
3751 {
3752 #ifdef PCI_DISABLE_MWI
3753         return 0;
3754 #else
3755         return pci_set_mwi(dev);
3756 #endif
3757 }
3758 EXPORT_SYMBOL(pci_try_set_mwi);
3759
3760 /**
3761  * pci_clear_mwi - disables Memory-Write-Invalidate for device dev
3762  * @dev: the PCI device to disable
3763  *
3764  * Disables PCI Memory-Write-Invalidate transaction on the device
3765  */
3766 void pci_clear_mwi(struct pci_dev *dev)
3767 {
3768 #ifndef PCI_DISABLE_MWI
3769         u16 cmd;
3770
3771         pci_read_config_word(dev, PCI_COMMAND, &cmd);
3772         if (cmd & PCI_COMMAND_INVALIDATE) {
3773                 cmd &= ~PCI_COMMAND_INVALIDATE;
3774                 pci_write_config_word(dev, PCI_COMMAND, cmd);
3775         }
3776 #endif
3777 }
3778 EXPORT_SYMBOL(pci_clear_mwi);
3779
3780 /**
3781  * pci_intx - enables/disables PCI INTx for device dev
3782  * @pdev: the PCI device to operate on
3783  * @enable: boolean: whether to enable or disable PCI INTx
3784  *
3785  * Enables/disables PCI INTx for device dev
3786  */
3787 void pci_intx(struct pci_dev *pdev, int enable)
3788 {
3789         u16 pci_command, new;
3790
3791         pci_read_config_word(pdev, PCI_COMMAND, &pci_command);
3792
3793         if (enable)
3794                 new = pci_command & ~PCI_COMMAND_INTX_DISABLE;
3795         else
3796                 new = pci_command | PCI_COMMAND_INTX_DISABLE;
3797
3798         if (new != pci_command) {
3799                 struct pci_devres *dr;
3800
3801                 pci_write_config_word(pdev, PCI_COMMAND, new);
3802
3803                 dr = find_pci_dr(pdev);
3804                 if (dr && !dr->restore_intx) {
3805                         dr->restore_intx = 1;
3806                         dr->orig_intx = !enable;
3807                 }
3808         }
3809 }
3810 EXPORT_SYMBOL_GPL(pci_intx);
3811
3812 static bool pci_check_and_set_intx_mask(struct pci_dev *dev, bool mask)
3813 {
3814         struct pci_bus *bus = dev->bus;
3815         bool mask_updated = true;
3816         u32 cmd_status_dword;
3817         u16 origcmd, newcmd;
3818         unsigned long flags;
3819         bool irq_pending;
3820
3821         /*
3822          * We do a single dword read to retrieve both command and status.
3823          * Document assumptions that make this possible.
3824          */
3825         BUILD_BUG_ON(PCI_COMMAND % 4);
3826         BUILD_BUG_ON(PCI_COMMAND + 2 != PCI_STATUS);
3827
3828         raw_spin_lock_irqsave(&pci_lock, flags);
3829
3830         bus->ops->read(bus, dev->devfn, PCI_COMMAND, 4, &cmd_status_dword);
3831
3832         irq_pending = (cmd_status_dword >> 16) & PCI_STATUS_INTERRUPT;
3833
3834         /*
3835          * Check interrupt status register to see whether our device
3836          * triggered the interrupt (when masking) or the next IRQ is
3837          * already pending (when unmasking).
3838          */
3839         if (mask != irq_pending) {
3840                 mask_updated = false;
3841                 goto done;
3842         }
3843
3844         origcmd = cmd_status_dword;
3845         newcmd = origcmd & ~PCI_COMMAND_INTX_DISABLE;
3846         if (mask)
3847                 newcmd |= PCI_COMMAND_INTX_DISABLE;
3848         if (newcmd != origcmd)
3849                 bus->ops->write(bus, dev->devfn, PCI_COMMAND, 2, newcmd);
3850
3851 done:
3852         raw_spin_unlock_irqrestore(&pci_lock, flags);
3853
3854         return mask_updated;
3855 }
3856
3857 /**
3858  * pci_check_and_mask_intx - mask INTx on pending interrupt
3859  * @dev: the PCI device to operate on
3860  *
3861  * Check if the device dev has its INTx line asserted, mask it and
3862  * return true in that case. False is returned if no interrupt was
3863  * pending.
3864  */
3865 bool pci_check_and_mask_intx(struct pci_dev *dev)
3866 {
3867         return pci_check_and_set_intx_mask(dev, true);
3868 }
3869 EXPORT_SYMBOL_GPL(pci_check_and_mask_intx);
3870
3871 /**
3872  * pci_check_and_unmask_intx - unmask INTx if no interrupt is pending
3873  * @dev: the PCI device to operate on
3874  *
3875  * Check if the device dev has its INTx line asserted, unmask it if not
3876  * and return true. False is returned and the mask remains active if
3877  * there was still an interrupt pending.
3878  */
3879 bool pci_check_and_unmask_intx(struct pci_dev *dev)
3880 {
3881         return pci_check_and_set_intx_mask(dev, false);
3882 }
3883 EXPORT_SYMBOL_GPL(pci_check_and_unmask_intx);
3884
3885 /**
3886  * pci_wait_for_pending_transaction - waits for pending transaction
3887  * @dev: the PCI device to operate on
3888  *
3889  * Return 0 if transaction is pending 1 otherwise.
3890  */
3891 int pci_wait_for_pending_transaction(struct pci_dev *dev)
3892 {
3893         if (!pci_is_pcie(dev))
3894                 return 1;
3895
3896         return pci_wait_for_pending(dev, pci_pcie_cap(dev) + PCI_EXP_DEVSTA,
3897                                     PCI_EXP_DEVSTA_TRPND);
3898 }
3899 EXPORT_SYMBOL(pci_wait_for_pending_transaction);
3900
3901 static void pci_flr_wait(struct pci_dev *dev)
3902 {
3903         int delay = 1, timeout = 60000;
3904         u32 id;
3905
3906         /*
3907          * Per PCIe r3.1, sec 6.6.2, a device must complete an FLR within
3908          * 100ms, but may silently discard requests while the FLR is in
3909          * progress.  Wait 100ms before trying to access the device.
3910          */
3911         msleep(100);
3912
3913         /*
3914          * After 100ms, the device should not silently discard config
3915          * requests, but it may still indicate that it needs more time by
3916          * responding to them with CRS completions.  The Root Port will
3917          * generally synthesize ~0 data to complete the read (except when
3918          * CRS SV is enabled and the read was for the Vendor ID; in that
3919          * case it synthesizes 0x0001 data).
3920          *
3921          * Wait for the device to return a non-CRS completion.  Read the
3922          * Command register instead of Vendor ID so we don't have to
3923          * contend with the CRS SV value.
3924          */
3925         pci_read_config_dword(dev, PCI_COMMAND, &id);
3926         while (id == ~0) {
3927                 if (delay > timeout) {
3928                         dev_warn(&dev->dev, "not ready %dms after FLR; giving up\n",
3929                                  100 + delay - 1);
3930                         return;
3931                 }
3932
3933                 if (delay > 1000)
3934                         dev_info(&dev->dev, "not ready %dms after FLR; waiting\n",
3935                                  100 + delay - 1);
3936
3937                 msleep(delay);
3938                 delay *= 2;
3939                 pci_read_config_dword(dev, PCI_COMMAND, &id);
3940         }
3941
3942         if (delay > 1000)
3943                 dev_info(&dev->dev, "ready %dms after FLR\n", 100 + delay - 1);
3944 }
3945
3946 /**
3947  * pcie_has_flr - check if a device supports function level resets
3948  * @dev:        device to check
3949  *
3950  * Returns true if the device advertises support for PCIe function level
3951  * resets.
3952  */
3953 static bool pcie_has_flr(struct pci_dev *dev)
3954 {
3955         u32 cap;
3956
3957         if (dev->dev_flags & PCI_DEV_FLAGS_NO_FLR_RESET)
3958                 return false;
3959
3960         pcie_capability_read_dword(dev, PCI_EXP_DEVCAP, &cap);
3961         return cap & PCI_EXP_DEVCAP_FLR;
3962 }
3963
3964 /**
3965  * pcie_flr - initiate a PCIe function level reset
3966  * @dev:        device to reset
3967  *
3968  * Initiate a function level reset on @dev.  The caller should ensure the
3969  * device supports FLR before calling this function, e.g. by using the
3970  * pcie_has_flr() helper.
3971  */
3972 void pcie_flr(struct pci_dev *dev)
3973 {
3974         if (!pci_wait_for_pending_transaction(dev))
3975                 dev_err(&dev->dev, "timed out waiting for pending transaction; performing function level reset anyway\n");
3976
3977         pcie_capability_set_word(dev, PCI_EXP_DEVCTL, PCI_EXP_DEVCTL_BCR_FLR);
3978         pci_flr_wait(dev);
3979 }
3980 EXPORT_SYMBOL_GPL(pcie_flr);
3981
3982 static int pci_af_flr(struct pci_dev *dev, int probe)
3983 {
3984         int pos;
3985         u8 cap;
3986
3987         pos = pci_find_capability(dev, PCI_CAP_ID_AF);
3988         if (!pos)
3989                 return -ENOTTY;
3990
3991         if (dev->dev_flags & PCI_DEV_FLAGS_NO_FLR_RESET)
3992                 return -ENOTTY;
3993
3994         pci_read_config_byte(dev, pos + PCI_AF_CAP, &cap);
3995         if (!(cap & PCI_AF_CAP_TP) || !(cap & PCI_AF_CAP_FLR))
3996                 return -ENOTTY;
3997
3998         if (probe)
3999                 return 0;
4000
4001         /*
4002          * Wait for Transaction Pending bit to clear.  A word-aligned test
4003          * is used, so we use the conrol offset rather than status and shift
4004          * the test bit to match.
4005          */
4006         if (!pci_wait_for_pending(dev, pos + PCI_AF_CTRL,
4007                                  PCI_AF_STATUS_TP << 8))
4008                 dev_err(&dev->dev, "timed out waiting for pending transaction; performing AF function level reset anyway\n");
4009
4010         pci_write_config_byte(dev, pos + PCI_AF_CTRL, PCI_AF_CTRL_FLR);
4011         pci_flr_wait(dev);
4012         return 0;
4013 }
4014
4015 /**
4016  * pci_pm_reset - Put device into PCI_D3 and back into PCI_D0.
4017  * @dev: Device to reset.
4018  * @probe: If set, only check if the device can be reset this way.
4019  *
4020  * If @dev supports native PCI PM and its PCI_PM_CTRL_NO_SOFT_RESET flag is
4021  * unset, it will be reinitialized internally when going from PCI_D3hot to
4022  * PCI_D0.  If that's the case and the device is not in a low-power state
4023  * already, force it into PCI_D3hot and back to PCI_D0, causing it to be reset.
4024  *
4025  * NOTE: This causes the caller to sleep for twice the device power transition
4026  * cooldown period, which for the D0->D3hot and D3hot->D0 transitions is 10 ms
4027  * by default (i.e. unless the @dev's d3_delay field has a different value).
4028  * Moreover, only devices in D0 can be reset by this function.
4029  */
4030 static int pci_pm_reset(struct pci_dev *dev, int probe)
4031 {
4032         u16 csr;
4033
4034         if (!dev->pm_cap || dev->dev_flags & PCI_DEV_FLAGS_NO_PM_RESET)
4035                 return -ENOTTY;
4036
4037         pci_read_config_word(dev, dev->pm_cap + PCI_PM_CTRL, &csr);
4038         if (csr & PCI_PM_CTRL_NO_SOFT_RESET)
4039                 return -ENOTTY;
4040
4041         if (probe)
4042                 return 0;
4043
4044         if (dev->current_state != PCI_D0)
4045                 return -EINVAL;
4046
4047         csr &= ~PCI_PM_CTRL_STATE_MASK;
4048         csr |= PCI_D3hot;
4049         pci_write_config_word(dev, dev->pm_cap + PCI_PM_CTRL, csr);
4050         pci_dev_d3_sleep(dev);
4051
4052         csr &= ~PCI_PM_CTRL_STATE_MASK;
4053         csr |= PCI_D0;
4054         pci_write_config_word(dev, dev->pm_cap + PCI_PM_CTRL, csr);
4055         pci_dev_d3_sleep(dev);
4056
4057         return 0;
4058 }
4059
4060 void pci_reset_secondary_bus(struct pci_dev *dev)
4061 {
4062         u16 ctrl;
4063
4064         pci_read_config_word(dev, PCI_BRIDGE_CONTROL, &ctrl);
4065         ctrl |= PCI_BRIDGE_CTL_BUS_RESET;
4066         pci_write_config_word(dev, PCI_BRIDGE_CONTROL, ctrl);
4067         /*
4068          * PCI spec v3.0 7.6.4.2 requires minimum Trst of 1ms.  Double
4069          * this to 2ms to ensure that we meet the minimum requirement.
4070          */
4071         msleep(2);
4072
4073         ctrl &= ~PCI_BRIDGE_CTL_BUS_RESET;
4074         pci_write_config_word(dev, PCI_BRIDGE_CONTROL, ctrl);
4075
4076         /*
4077          * Trhfa for conventional PCI is 2^25 clock cycles.
4078          * Assuming a minimum 33MHz clock this results in a 1s
4079          * delay before we can consider subordinate devices to
4080          * be re-initialized.  PCIe has some ways to shorten this,
4081          * but we don't make use of them yet.
4082          */
4083         ssleep(1);
4084 }
4085
4086 void __weak pcibios_reset_secondary_bus(struct pci_dev *dev)
4087 {
4088         pci_reset_secondary_bus(dev);
4089 }
4090
4091 /**
4092  * pci_reset_bridge_secondary_bus - Reset the secondary bus on a PCI bridge.
4093  * @dev: Bridge device
4094  *
4095  * Use the bridge control register to assert reset on the secondary bus.
4096  * Devices on the secondary bus are left in power-on state.
4097  */
4098 void pci_reset_bridge_secondary_bus(struct pci_dev *dev)
4099 {
4100         pcibios_reset_secondary_bus(dev);
4101 }
4102 EXPORT_SYMBOL_GPL(pci_reset_bridge_secondary_bus);
4103
4104 static int pci_parent_bus_reset(struct pci_dev *dev, int probe)
4105 {
4106         struct pci_dev *pdev;
4107
4108         if (pci_is_root_bus(dev->bus) || dev->subordinate ||
4109             !dev->bus->self || dev->dev_flags & PCI_DEV_FLAGS_NO_BUS_RESET)
4110                 return -ENOTTY;
4111
4112         list_for_each_entry(pdev, &dev->bus->devices, bus_list)
4113                 if (pdev != dev)
4114                         return -ENOTTY;
4115
4116         if (probe)
4117                 return 0;
4118
4119         pci_reset_bridge_secondary_bus(dev->bus->self);
4120
4121         return 0;
4122 }
4123
4124 static int pci_reset_hotplug_slot(struct hotplug_slot *hotplug, int probe)
4125 {
4126         int rc = -ENOTTY;
4127
4128         if (!hotplug || !try_module_get(hotplug->ops->owner))
4129                 return rc;
4130
4131         if (hotplug->ops->reset_slot)
4132                 rc = hotplug->ops->reset_slot(hotplug, probe);
4133
4134         module_put(hotplug->ops->owner);
4135
4136         return rc;
4137 }
4138
4139 static int pci_dev_reset_slot_function(struct pci_dev *dev, int probe)
4140 {
4141         struct pci_dev *pdev;
4142
4143         if (dev->subordinate || !dev->slot ||
4144             dev->dev_flags & PCI_DEV_FLAGS_NO_BUS_RESET)
4145                 return -ENOTTY;
4146
4147         list_for_each_entry(pdev, &dev->bus->devices, bus_list)
4148                 if (pdev != dev && pdev->slot == dev->slot)
4149                         return -ENOTTY;
4150
4151         return pci_reset_hotplug_slot(dev->slot->hotplug, probe);
4152 }
4153
4154 static void pci_dev_lock(struct pci_dev *dev)
4155 {
4156         /* block PM suspend, driver probe, etc. */
4157         device_lock(&dev->dev);
4158         pci_cfg_access_lock(dev);
4159 }
4160
4161 /* Return 1 on successful lock, 0 on contention */
4162 static int pci_dev_trylock(struct pci_dev *dev)
4163 {
4164         if (device_trylock(&dev->dev)) {
4165                 if (pci_cfg_access_trylock(dev))
4166                         return 1;
4167                 device_unlock(&dev->dev);
4168         }
4169
4170         return 0;
4171 }
4172
4173 static void pci_dev_unlock(struct pci_dev *dev)
4174 {
4175         pci_cfg_access_unlock(dev);
4176         device_unlock(&dev->dev);
4177 }
4178
4179 static void pci_dev_save_and_disable(struct pci_dev *dev)
4180 {
4181         const struct pci_error_handlers *err_handler =
4182                         dev->driver ? dev->driver->err_handler : NULL;
4183
4184         /*
4185          * dev->driver->err_handler->reset_prepare() is protected against
4186          * races with ->remove() by the device lock, which must be held by
4187          * the caller.
4188          */
4189         if (err_handler && err_handler->reset_prepare)
4190                 err_handler->reset_prepare(dev);
4191
4192         /*
4193          * Wake-up device prior to save.  PM registers default to D0 after
4194          * reset and a simple register restore doesn't reliably return
4195          * to a non-D0 state anyway.
4196          */
4197         pci_set_power_state(dev, PCI_D0);
4198
4199         pci_save_state(dev);
4200         /*
4201          * Disable the device by clearing the Command register, except for
4202          * INTx-disable which is set.  This not only disables MMIO and I/O port
4203          * BARs, but also prevents the device from being Bus Master, preventing
4204          * DMA from the device including MSI/MSI-X interrupts.  For PCI 2.3
4205          * compliant devices, INTx-disable prevents legacy interrupts.
4206          */
4207         pci_write_config_word(dev, PCI_COMMAND, PCI_COMMAND_INTX_DISABLE);
4208 }
4209
4210 static void pci_dev_restore(struct pci_dev *dev)
4211 {
4212         const struct pci_error_handlers *err_handler =
4213                         dev->driver ? dev->driver->err_handler : NULL;
4214
4215         pci_restore_state(dev);
4216
4217         /*
4218          * dev->driver->err_handler->reset_done() is protected against
4219          * races with ->remove() by the device lock, which must be held by
4220          * the caller.
4221          */
4222         if (err_handler && err_handler->reset_done)
4223                 err_handler->reset_done(dev);
4224 }
4225
4226 /**
4227  * __pci_reset_function - reset a PCI device function
4228  * @dev: PCI device to reset
4229  *
4230  * Some devices allow an individual function to be reset without affecting
4231  * other functions in the same device.  The PCI device must be responsive
4232  * to PCI config space in order to use this function.
4233  *
4234  * The device function is presumed to be unused when this function is called.
4235  * Resetting the device will make the contents of PCI configuration space
4236  * random, so any caller of this must be prepared to reinitialise the
4237  * device including MSI, bus mastering, BARs, decoding IO and memory spaces,
4238  * etc.
4239  *
4240  * Returns 0 if the device function was successfully reset or negative if the
4241  * device doesn't support resetting a single function.
4242  */
4243 int __pci_reset_function(struct pci_dev *dev)
4244 {
4245         int ret;
4246
4247         pci_dev_lock(dev);
4248         ret = __pci_reset_function_locked(dev);
4249         pci_dev_unlock(dev);
4250
4251         return ret;
4252 }
4253 EXPORT_SYMBOL_GPL(__pci_reset_function);
4254
4255 /**
4256  * __pci_reset_function_locked - reset a PCI device function while holding
4257  * the @dev mutex lock.
4258  * @dev: PCI device to reset
4259  *
4260  * Some devices allow an individual function to be reset without affecting
4261  * other functions in the same device.  The PCI device must be responsive
4262  * to PCI config space in order to use this function.
4263  *
4264  * The device function is presumed to be unused and the caller is holding
4265  * the device mutex lock when this function is called.
4266  * Resetting the device will make the contents of PCI configuration space
4267  * random, so any caller of this must be prepared to reinitialise the
4268  * device including MSI, bus mastering, BARs, decoding IO and memory spaces,
4269  * etc.
4270  *
4271  * Returns 0 if the device function was successfully reset or negative if the
4272  * device doesn't support resetting a single function.
4273  */
4274 int __pci_reset_function_locked(struct pci_dev *dev)
4275 {
4276         int rc;
4277
4278         might_sleep();
4279
4280         rc = pci_dev_specific_reset(dev, 0);
4281         if (rc != -ENOTTY)
4282                 return rc;
4283         if (pcie_has_flr(dev)) {
4284                 pcie_flr(dev);
4285                 return 0;
4286         }
4287         rc = pci_af_flr(dev, 0);
4288         if (rc != -ENOTTY)
4289                 return rc;
4290         rc = pci_pm_reset(dev, 0);
4291         if (rc != -ENOTTY)
4292                 return rc;
4293         rc = pci_dev_reset_slot_function(dev, 0);
4294         if (rc != -ENOTTY)
4295                 return rc;
4296         return pci_parent_bus_reset(dev, 0);
4297 }
4298 EXPORT_SYMBOL_GPL(__pci_reset_function_locked);
4299
4300 /**
4301  * pci_probe_reset_function - check whether the device can be safely reset
4302  * @dev: PCI device to reset
4303  *
4304  * Some devices allow an individual function to be reset without affecting
4305  * other functions in the same device.  The PCI device must be responsive
4306  * to PCI config space in order to use this function.
4307  *
4308  * Returns 0 if the device function can be reset or negative if the
4309  * device doesn't support resetting a single function.
4310  */
4311 int pci_probe_reset_function(struct pci_dev *dev)
4312 {
4313         int rc;
4314
4315         might_sleep();
4316
4317         rc = pci_dev_specific_reset(dev, 1);
4318         if (rc != -ENOTTY)
4319                 return rc;
4320         if (pcie_has_flr(dev))
4321                 return 0;
4322         rc = pci_af_flr(dev, 1);
4323         if (rc != -ENOTTY)
4324                 return rc;
4325         rc = pci_pm_reset(dev, 1);
4326         if (rc != -ENOTTY)
4327                 return rc;
4328         rc = pci_dev_reset_slot_function(dev, 1);
4329         if (rc != -ENOTTY)
4330                 return rc;
4331
4332         return pci_parent_bus_reset(dev, 1);
4333 }
4334
4335 /**
4336  * pci_reset_function - quiesce and reset a PCI device function
4337  * @dev: PCI device to reset
4338  *
4339  * Some devices allow an individual function to be reset without affecting
4340  * other functions in the same device.  The PCI device must be responsive
4341  * to PCI config space in order to use this function.
4342  *
4343  * This function does not just reset the PCI portion of a device, but
4344  * clears all the state associated with the device.  This function differs
4345  * from __pci_reset_function in that it saves and restores device state
4346  * over the reset.
4347  *
4348  * Returns 0 if the device function was successfully reset or negative if the
4349  * device doesn't support resetting a single function.
4350  */
4351 int pci_reset_function(struct pci_dev *dev)
4352 {
4353         int rc;
4354
4355         rc = pci_probe_reset_function(dev);
4356         if (rc)
4357                 return rc;
4358
4359         pci_dev_lock(dev);
4360         pci_dev_save_and_disable(dev);
4361
4362         rc = __pci_reset_function_locked(dev);
4363
4364         pci_dev_restore(dev);
4365         pci_dev_unlock(dev);
4366
4367         return rc;
4368 }
4369 EXPORT_SYMBOL_GPL(pci_reset_function);
4370
4371 /**
4372  * pci_reset_function_locked - quiesce and reset a PCI device function
4373  * @dev: PCI device to reset
4374  *
4375  * Some devices allow an individual function to be reset without affecting
4376  * other functions in the same device.  The PCI device must be responsive
4377  * to PCI config space in order to use this function.
4378  *
4379  * This function does not just reset the PCI portion of a device, but
4380  * clears all the state associated with the device.  This function differs
4381  * from __pci_reset_function() in that it saves and restores device state
4382  * over the reset.  It also differs from pci_reset_function() in that it
4383  * requires the PCI device lock to be held.
4384  *
4385  * Returns 0 if the device function was successfully reset or negative if the
4386  * device doesn't support resetting a single function.
4387  */
4388 int pci_reset_function_locked(struct pci_dev *dev)
4389 {
4390         int rc;
4391
4392         rc = pci_probe_reset_function(dev);
4393         if (rc)
4394                 return rc;
4395
4396         pci_dev_save_and_disable(dev);
4397
4398         rc = __pci_reset_function_locked(dev);
4399
4400         pci_dev_restore(dev);
4401
4402         return rc;
4403 }
4404 EXPORT_SYMBOL_GPL(pci_reset_function_locked);
4405
4406 /**
4407  * pci_try_reset_function - quiesce and reset a PCI device function
4408  * @dev: PCI device to reset
4409  *
4410  * Same as above, except return -EAGAIN if unable to lock device.
4411  */
4412 int pci_try_reset_function(struct pci_dev *dev)
4413 {
4414         int rc;
4415
4416         rc = pci_probe_reset_function(dev);
4417         if (rc)
4418                 return rc;
4419
4420         if (!pci_dev_trylock(dev))
4421                 return -EAGAIN;
4422
4423         pci_dev_save_and_disable(dev);
4424         rc = __pci_reset_function_locked(dev);
4425         pci_dev_unlock(dev);
4426
4427         pci_dev_restore(dev);
4428         return rc;
4429 }
4430 EXPORT_SYMBOL_GPL(pci_try_reset_function);
4431
4432 /* Do any devices on or below this bus prevent a bus reset? */
4433 static bool pci_bus_resetable(struct pci_bus *bus)
4434 {
4435         struct pci_dev *dev;
4436
4437
4438         if (bus->self && (bus->self->dev_flags & PCI_DEV_FLAGS_NO_BUS_RESET))
4439                 return false;
4440
4441         list_for_each_entry(dev, &bus->devices, bus_list) {
4442                 if (dev->dev_flags & PCI_DEV_FLAGS_NO_BUS_RESET ||
4443                     (dev->subordinate && !pci_bus_resetable(dev->subordinate)))
4444                         return false;
4445         }
4446
4447         return true;
4448 }
4449
4450 /* Lock devices from the top of the tree down */
4451 static void pci_bus_lock(struct pci_bus *bus)
4452 {
4453         struct pci_dev *dev;
4454
4455         list_for_each_entry(dev, &bus->devices, bus_list) {
4456                 pci_dev_lock(dev);
4457                 if (dev->subordinate)
4458                         pci_bus_lock(dev->subordinate);
4459         }
4460 }
4461
4462 /* Unlock devices from the bottom of the tree up */
4463 static void pci_bus_unlock(struct pci_bus *bus)
4464 {
4465         struct pci_dev *dev;
4466
4467         list_for_each_entry(dev, &bus->devices, bus_list) {
4468                 if (dev->subordinate)
4469                         pci_bus_unlock(dev->subordinate);
4470                 pci_dev_unlock(dev);
4471         }
4472 }
4473
4474 /* Return 1 on successful lock, 0 on contention */
4475 static int pci_bus_trylock(struct pci_bus *bus)
4476 {
4477         struct pci_dev *dev;
4478
4479         list_for_each_entry(dev, &bus->devices, bus_list) {
4480                 if (!pci_dev_trylock(dev))
4481                         goto unlock;
4482                 if (dev->subordinate) {
4483                         if (!pci_bus_trylock(dev->subordinate)) {
4484                                 pci_dev_unlock(dev);
4485                                 goto unlock;
4486                         }
4487                 }
4488         }
4489         return 1;
4490
4491 unlock:
4492         list_for_each_entry_continue_reverse(dev, &bus->devices, bus_list) {
4493                 if (dev->subordinate)
4494                         pci_bus_unlock(dev->subordinate);
4495                 pci_dev_unlock(dev);
4496         }
4497         return 0;
4498 }
4499
4500 /* Do any devices on or below this slot prevent a bus reset? */
4501 static bool pci_slot_resetable(struct pci_slot *slot)
4502 {
4503         struct pci_dev *dev;
4504
4505         list_for_each_entry(dev, &slot->bus->devices, bus_list) {
4506                 if (!dev->slot || dev->slot != slot)
4507                         continue;
4508                 if (dev->dev_flags & PCI_DEV_FLAGS_NO_BUS_RESET ||
4509                     (dev->subordinate && !pci_bus_resetable(dev->subordinate)))
4510                         return false;
4511         }
4512
4513         return true;
4514 }
4515
4516 /* Lock devices from the top of the tree down */
4517 static void pci_slot_lock(struct pci_slot *slot)
4518 {
4519         struct pci_dev *dev;
4520
4521         list_for_each_entry(dev, &slot->bus->devices, bus_list) {
4522                 if (!dev->slot || dev->slot != slot)
4523                         continue;
4524                 pci_dev_lock(dev);
4525                 if (dev->subordinate)
4526                         pci_bus_lock(dev->subordinate);
4527         }
4528 }
4529
4530 /* Unlock devices from the bottom of the tree up */
4531 static void pci_slot_unlock(struct pci_slot *slot)
4532 {
4533         struct pci_dev *dev;
4534
4535         list_for_each_entry(dev, &slot->bus->devices, bus_list) {
4536                 if (!dev->slot || dev->slot != slot)
4537                         continue;
4538                 if (dev->subordinate)
4539                         pci_bus_unlock(dev->subordinate);
4540                 pci_dev_unlock(dev);
4541         }
4542 }
4543
4544 /* Return 1 on successful lock, 0 on contention */
4545 static int pci_slot_trylock(struct pci_slot *slot)
4546 {
4547         struct pci_dev *dev;
4548
4549         list_for_each_entry(dev, &slot->bus->devices, bus_list) {
4550                 if (!dev->slot || dev->slot != slot)
4551                         continue;
4552                 if (!pci_dev_trylock(dev))
4553                         goto unlock;
4554                 if (dev->subordinate) {
4555                         if (!pci_bus_trylock(dev->subordinate)) {
4556                                 pci_dev_unlock(dev);
4557                                 goto unlock;
4558                         }
4559                 }
4560         }
4561         return 1;
4562
4563 unlock:
4564         list_for_each_entry_continue_reverse(dev,
4565                                              &slot->bus->devices, bus_list) {
4566                 if (!dev->slot || dev->slot != slot)
4567                         continue;
4568                 if (dev->subordinate)
4569                         pci_bus_unlock(dev->subordinate);
4570                 pci_dev_unlock(dev);
4571         }
4572         return 0;
4573 }
4574
4575 /* Save and disable devices from the top of the tree down */
4576 static void pci_bus_save_and_disable(struct pci_bus *bus)
4577 {
4578         struct pci_dev *dev;
4579
4580         list_for_each_entry(dev, &bus->devices, bus_list) {
4581                 pci_dev_lock(dev);
4582                 pci_dev_save_and_disable(dev);
4583                 pci_dev_unlock(dev);
4584                 if (dev->subordinate)
4585                         pci_bus_save_and_disable(dev->subordinate);
4586         }
4587 }
4588
4589 /*
4590  * Restore devices from top of the tree down - parent bridges need to be
4591  * restored before we can get to subordinate devices.
4592  */
4593 static void pci_bus_restore(struct pci_bus *bus)
4594 {
4595         struct pci_dev *dev;
4596
4597         list_for_each_entry(dev, &bus->devices, bus_list) {
4598                 pci_dev_lock(dev);
4599                 pci_dev_restore(dev);
4600                 pci_dev_unlock(dev);
4601                 if (dev->subordinate)
4602                         pci_bus_restore(dev->subordinate);
4603         }
4604 }
4605
4606 /* Save and disable devices from the top of the tree down */
4607 static void pci_slot_save_and_disable(struct pci_slot *slot)
4608 {
4609         struct pci_dev *dev;
4610
4611         list_for_each_entry(dev, &slot->bus->devices, bus_list) {
4612                 if (!dev->slot || dev->slot != slot)
4613                         continue;
4614                 pci_dev_save_and_disable(dev);
4615                 if (dev->subordinate)
4616                         pci_bus_save_and_disable(dev->subordinate);
4617         }
4618 }
4619
4620 /*
4621  * Restore devices from top of the tree down - parent bridges need to be
4622  * restored before we can get to subordinate devices.
4623  */
4624 static void pci_slot_restore(struct pci_slot *slot)
4625 {
4626         struct pci_dev *dev;
4627
4628         list_for_each_entry(dev, &slot->bus->devices, bus_list) {
4629                 if (!dev->slot || dev->slot != slot)
4630                         continue;
4631                 pci_dev_restore(dev);
4632                 if (dev->subordinate)
4633                         pci_bus_restore(dev->subordinate);
4634         }
4635 }
4636
4637 static int pci_slot_reset(struct pci_slot *slot, int probe)
4638 {
4639         int rc;
4640
4641         if (!slot || !pci_slot_resetable(slot))
4642                 return -ENOTTY;
4643
4644         if (!probe)
4645                 pci_slot_lock(slot);
4646
4647         might_sleep();
4648
4649         rc = pci_reset_hotplug_slot(slot->hotplug, probe);
4650
4651         if (!probe)
4652                 pci_slot_unlock(slot);
4653
4654         return rc;
4655 }
4656
4657 /**
4658  * pci_probe_reset_slot - probe whether a PCI slot can be reset
4659  * @slot: PCI slot to probe
4660  *
4661  * Return 0 if slot can be reset, negative if a slot reset is not supported.
4662  */
4663 int pci_probe_reset_slot(struct pci_slot *slot)
4664 {
4665         return pci_slot_reset(slot, 1);
4666 }
4667 EXPORT_SYMBOL_GPL(pci_probe_reset_slot);
4668
4669 /**
4670  * pci_reset_slot - reset a PCI slot
4671  * @slot: PCI slot to reset
4672  *
4673  * A PCI bus may host multiple slots, each slot may support a reset mechanism
4674  * independent of other slots.  For instance, some slots may support slot power
4675  * control.  In the case of a 1:1 bus to slot architecture, this function may
4676  * wrap the bus reset to avoid spurious slot related events such as hotplug.
4677  * Generally a slot reset should be attempted before a bus reset.  All of the
4678  * function of the slot and any subordinate buses behind the slot are reset
4679  * through this function.  PCI config space of all devices in the slot and
4680  * behind the slot is saved before and restored after reset.
4681  *
4682  * Return 0 on success, non-zero on error.
4683  */
4684 int pci_reset_slot(struct pci_slot *slot)
4685 {
4686         int rc;
4687
4688         rc = pci_slot_reset(slot, 1);
4689         if (rc)
4690                 return rc;
4691
4692         pci_slot_save_and_disable(slot);
4693
4694         rc = pci_slot_reset(slot, 0);
4695
4696         pci_slot_restore(slot);
4697
4698         return rc;
4699 }
4700 EXPORT_SYMBOL_GPL(pci_reset_slot);
4701
4702 /**
4703  * pci_try_reset_slot - Try to reset a PCI slot
4704  * @slot: PCI slot to reset
4705  *
4706  * Same as above except return -EAGAIN if the slot cannot be locked
4707  */
4708 int pci_try_reset_slot(struct pci_slot *slot)
4709 {
4710         int rc;
4711
4712         rc = pci_slot_reset(slot, 1);
4713         if (rc)
4714                 return rc;
4715
4716         pci_slot_save_and_disable(slot);
4717
4718         if (pci_slot_trylock(slot)) {
4719                 might_sleep();
4720                 rc = pci_reset_hotplug_slot(slot->hotplug, 0);
4721                 pci_slot_unlock(slot);
4722         } else
4723                 rc = -EAGAIN;
4724
4725         pci_slot_restore(slot);
4726
4727         return rc;
4728 }
4729 EXPORT_SYMBOL_GPL(pci_try_reset_slot);
4730
4731 static int pci_bus_reset(struct pci_bus *bus, int probe)
4732 {
4733         if (!bus->self || !pci_bus_resetable(bus))
4734                 return -ENOTTY;
4735
4736         if (probe)
4737                 return 0;
4738
4739         pci_bus_lock(bus);
4740
4741         might_sleep();
4742
4743         pci_reset_bridge_secondary_bus(bus->self);
4744
4745         pci_bus_unlock(bus);
4746
4747         return 0;
4748 }
4749
4750 /**
4751  * pci_probe_reset_bus - probe whether a PCI bus can be reset
4752  * @bus: PCI bus to probe
4753  *
4754  * Return 0 if bus can be reset, negative if a bus reset is not supported.
4755  */
4756 int pci_probe_reset_bus(struct pci_bus *bus)
4757 {
4758         return pci_bus_reset(bus, 1);
4759 }
4760 EXPORT_SYMBOL_GPL(pci_probe_reset_bus);
4761
4762 /**
4763  * pci_reset_bus - reset a PCI bus
4764  * @bus: top level PCI bus to reset
4765  *
4766  * Do a bus reset on the given bus and any subordinate buses, saving
4767  * and restoring state of all devices.
4768  *
4769  * Return 0 on success, non-zero on error.
4770  */
4771 int pci_reset_bus(struct pci_bus *bus)
4772 {
4773         int rc;
4774
4775         rc = pci_bus_reset(bus, 1);
4776         if (rc)
4777                 return rc;
4778
4779         pci_bus_save_and_disable(bus);
4780
4781         rc = pci_bus_reset(bus, 0);
4782
4783         pci_bus_restore(bus);
4784
4785         return rc;
4786 }
4787 EXPORT_SYMBOL_GPL(pci_reset_bus);
4788
4789 /**
4790  * pci_try_reset_bus - Try to reset a PCI bus
4791  * @bus: top level PCI bus to reset
4792  *
4793  * Same as above except return -EAGAIN if the bus cannot be locked
4794  */
4795 int pci_try_reset_bus(struct pci_bus *bus)
4796 {
4797         int rc;
4798
4799         rc = pci_bus_reset(bus, 1);
4800         if (rc)
4801                 return rc;
4802
4803         pci_bus_save_and_disable(bus);
4804
4805         if (pci_bus_trylock(bus)) {
4806                 might_sleep();
4807                 pci_reset_bridge_secondary_bus(bus->self);
4808                 pci_bus_unlock(bus);
4809         } else
4810                 rc = -EAGAIN;
4811
4812         pci_bus_restore(bus);
4813
4814         return rc;
4815 }
4816 EXPORT_SYMBOL_GPL(pci_try_reset_bus);
4817
4818 /**
4819  * pcix_get_max_mmrbc - get PCI-X maximum designed memory read byte count
4820  * @dev: PCI device to query
4821  *
4822  * Returns mmrbc: maximum designed memory read count in bytes
4823  *    or appropriate error value.
4824  */
4825 int pcix_get_max_mmrbc(struct pci_dev *dev)
4826 {
4827         int cap;
4828         u32 stat;
4829
4830         cap = pci_find_capability(dev, PCI_CAP_ID_PCIX);
4831         if (!cap)
4832                 return -EINVAL;
4833
4834         if (pci_read_config_dword(dev, cap + PCI_X_STATUS, &stat))
4835                 return -EINVAL;
4836
4837         return 512 << ((stat & PCI_X_STATUS_MAX_READ) >> 21);
4838 }
4839 EXPORT_SYMBOL(pcix_get_max_mmrbc);
4840
4841 /**
4842  * pcix_get_mmrbc - get PCI-X maximum memory read byte count
4843  * @dev: PCI device to query
4844  *
4845  * Returns mmrbc: maximum memory read count in bytes
4846  *    or appropriate error value.
4847  */
4848 int pcix_get_mmrbc(struct pci_dev *dev)
4849 {
4850         int cap;
4851         u16 cmd;
4852
4853         cap = pci_find_capability(dev, PCI_CAP_ID_PCIX);
4854         if (!cap)
4855                 return -EINVAL;
4856
4857         if (pci_read_config_word(dev, cap + PCI_X_CMD, &cmd))
4858                 return -EINVAL;
4859
4860         return 512 << ((cmd & PCI_X_CMD_MAX_READ) >> 2);
4861 }
4862 EXPORT_SYMBOL(pcix_get_mmrbc);
4863
4864 /**
4865  * pcix_set_mmrbc - set PCI-X maximum memory read byte count
4866  * @dev: PCI device to query
4867  * @mmrbc: maximum memory read count in bytes
4868  *    valid values are 512, 1024, 2048, 4096
4869  *
4870  * If possible sets maximum memory read byte count, some bridges have erratas
4871  * that prevent this.
4872  */
4873 int pcix_set_mmrbc(struct pci_dev *dev, int mmrbc)
4874 {
4875         int cap;
4876         u32 stat, v, o;
4877         u16 cmd;
4878
4879         if (mmrbc < 512 || mmrbc > 4096 || !is_power_of_2(mmrbc))
4880                 return -EINVAL;
4881
4882         v = ffs(mmrbc) - 10;
4883
4884         cap = pci_find_capability(dev, PCI_CAP_ID_PCIX);
4885         if (!cap)
4886                 return -EINVAL;
4887
4888         if (pci_read_config_dword(dev, cap + PCI_X_STATUS, &stat))
4889                 return -EINVAL;
4890
4891         if (v > (stat & PCI_X_STATUS_MAX_READ) >> 21)
4892                 return -E2BIG;
4893
4894         if (pci_read_config_word(dev, cap + PCI_X_CMD, &cmd))
4895                 return -EINVAL;
4896
4897         o = (cmd & PCI_X_CMD_MAX_READ) >> 2;
4898         if (o != v) {
4899                 if (v > o && (dev->bus->bus_flags & PCI_BUS_FLAGS_NO_MMRBC))
4900                         return -EIO;
4901
4902                 cmd &= ~PCI_X_CMD_MAX_READ;
4903                 cmd |= v << 2;
4904                 if (pci_write_config_word(dev, cap + PCI_X_CMD, cmd))
4905                         return -EIO;
4906         }
4907         return 0;
4908 }
4909 EXPORT_SYMBOL(pcix_set_mmrbc);
4910
4911 /**
4912  * pcie_get_readrq - get PCI Express read request size
4913  * @dev: PCI device to query
4914  *
4915  * Returns maximum memory read request in bytes
4916  *    or appropriate error value.
4917  */
4918 int pcie_get_readrq(struct pci_dev *dev)
4919 {
4920         u16 ctl;
4921
4922         pcie_capability_read_word(dev, PCI_EXP_DEVCTL, &ctl);
4923
4924         return 128 << ((ctl & PCI_EXP_DEVCTL_READRQ) >> 12);
4925 }
4926 EXPORT_SYMBOL(pcie_get_readrq);
4927
4928 /**
4929  * pcie_set_readrq - set PCI Express maximum memory read request
4930  * @dev: PCI device to query
4931  * @rq: maximum memory read count in bytes
4932  *    valid values are 128, 256, 512, 1024, 2048, 4096
4933  *
4934  * If possible sets maximum memory read request in bytes
4935  */
4936 int pcie_set_readrq(struct pci_dev *dev, int rq)
4937 {
4938         u16 v;
4939
4940         if (rq < 128 || rq > 4096 || !is_power_of_2(rq))
4941                 return -EINVAL;
4942
4943         /*
4944          * If using the "performance" PCIe config, we clamp the
4945          * read rq size to the max packet size to prevent the
4946          * host bridge generating requests larger than we can
4947          * cope with
4948          */
4949         if (pcie_bus_config == PCIE_BUS_PERFORMANCE) {
4950                 int mps = pcie_get_mps(dev);
4951
4952                 if (mps < rq)
4953                         rq = mps;
4954         }
4955
4956         v = (ffs(rq) - 8) << 12;
4957
4958         return pcie_capability_clear_and_set_word(dev, PCI_EXP_DEVCTL,
4959                                                   PCI_EXP_DEVCTL_READRQ, v);
4960 }
4961 EXPORT_SYMBOL(pcie_set_readrq);
4962
4963 /**
4964  * pcie_get_mps - get PCI Express maximum payload size
4965  * @dev: PCI device to query
4966  *
4967  * Returns maximum payload size in bytes
4968  */
4969 int pcie_get_mps(struct pci_dev *dev)
4970 {
4971         u16 ctl;
4972
4973         pcie_capability_read_word(dev, PCI_EXP_DEVCTL, &ctl);
4974
4975         return 128 << ((ctl & PCI_EXP_DEVCTL_PAYLOAD) >> 5);
4976 }
4977 EXPORT_SYMBOL(pcie_get_mps);
4978
4979 /**
4980  * pcie_set_mps - set PCI Express maximum payload size
4981  * @dev: PCI device to query
4982  * @mps: maximum payload size in bytes
4983  *    valid values are 128, 256, 512, 1024, 2048, 4096
4984  *
4985  * If possible sets maximum payload size
4986  */
4987 int pcie_set_mps(struct pci_dev *dev, int mps)
4988 {
4989         u16 v;
4990
4991         if (mps < 128 || mps > 4096 || !is_power_of_2(mps))
4992                 return -EINVAL;
4993
4994         v = ffs(mps) - 8;
4995         if (v > dev->pcie_mpss)
4996                 return -EINVAL;
4997         v <<= 5;
4998
4999         return pcie_capability_clear_and_set_word(dev, PCI_EXP_DEVCTL,
5000                                                   PCI_EXP_DEVCTL_PAYLOAD, v);
5001 }
5002 EXPORT_SYMBOL(pcie_set_mps);
5003
5004 /**
5005  * pcie_get_minimum_link - determine minimum link settings of a PCI device
5006  * @dev: PCI device to query
5007  * @speed: storage for minimum speed
5008  * @width: storage for minimum width
5009  *
5010  * This function will walk up the PCI device chain and determine the minimum
5011  * link width and speed of the device.
5012  */
5013 int pcie_get_minimum_link(struct pci_dev *dev, enum pci_bus_speed *speed,
5014                           enum pcie_link_width *width)
5015 {
5016         int ret;
5017
5018         *speed = PCI_SPEED_UNKNOWN;
5019         *width = PCIE_LNK_WIDTH_UNKNOWN;
5020
5021         while (dev) {
5022                 u16 lnksta;
5023                 enum pci_bus_speed next_speed;
5024                 enum pcie_link_width next_width;
5025
5026                 ret = pcie_capability_read_word(dev, PCI_EXP_LNKSTA, &lnksta);
5027                 if (ret)
5028                         return ret;
5029
5030                 next_speed = pcie_link_speed[lnksta & PCI_EXP_LNKSTA_CLS];
5031                 next_width = (lnksta & PCI_EXP_LNKSTA_NLW) >>
5032                         PCI_EXP_LNKSTA_NLW_SHIFT;
5033
5034                 if (next_speed < *speed)
5035                         *speed = next_speed;
5036
5037                 if (next_width < *width)
5038                         *width = next_width;
5039
5040                 dev = dev->bus->self;
5041         }
5042
5043         return 0;
5044 }
5045 EXPORT_SYMBOL(pcie_get_minimum_link);
5046
5047 /**
5048  * pci_select_bars - Make BAR mask from the type of resource
5049  * @dev: the PCI device for which BAR mask is made
5050  * @flags: resource type mask to be selected
5051  *
5052  * This helper routine makes bar mask from the type of resource.
5053  */
5054 int pci_select_bars(struct pci_dev *dev, unsigned long flags)
5055 {
5056         int i, bars = 0;
5057         for (i = 0; i < PCI_NUM_RESOURCES; i++)
5058                 if (pci_resource_flags(dev, i) & flags)
5059                         bars |= (1 << i);
5060         return bars;
5061 }
5062 EXPORT_SYMBOL(pci_select_bars);
5063
5064 /* Some architectures require additional programming to enable VGA */
5065 static arch_set_vga_state_t arch_set_vga_state;
5066
5067 void __init pci_register_set_vga_state(arch_set_vga_state_t func)
5068 {
5069         arch_set_vga_state = func;      /* NULL disables */
5070 }
5071
5072 static int pci_set_vga_state_arch(struct pci_dev *dev, bool decode,
5073                                   unsigned int command_bits, u32 flags)
5074 {
5075         if (arch_set_vga_state)
5076                 return arch_set_vga_state(dev, decode, command_bits,
5077                                                 flags);
5078         return 0;
5079 }
5080
5081 /**
5082  * pci_set_vga_state - set VGA decode state on device and parents if requested
5083  * @dev: the PCI device
5084  * @decode: true = enable decoding, false = disable decoding
5085  * @command_bits: PCI_COMMAND_IO and/or PCI_COMMAND_MEMORY
5086  * @flags: traverse ancestors and change bridges
5087  * CHANGE_BRIDGE_ONLY / CHANGE_BRIDGE
5088  */
5089 int pci_set_vga_state(struct pci_dev *dev, bool decode,
5090                       unsigned int command_bits, u32 flags)
5091 {
5092         struct pci_bus *bus;
5093         struct pci_dev *bridge;
5094         u16 cmd;
5095         int rc;
5096
5097         WARN_ON((flags & PCI_VGA_STATE_CHANGE_DECODES) && (command_bits & ~(PCI_COMMAND_IO|PCI_COMMAND_MEMORY)));
5098
5099         /* ARCH specific VGA enables */
5100         rc = pci_set_vga_state_arch(dev, decode, command_bits, flags);
5101         if (rc)
5102                 return rc;
5103
5104         if (flags & PCI_VGA_STATE_CHANGE_DECODES) {
5105                 pci_read_config_word(dev, PCI_COMMAND, &cmd);
5106                 if (decode == true)
5107                         cmd |= command_bits;
5108                 else
5109                         cmd &= ~command_bits;
5110                 pci_write_config_word(dev, PCI_COMMAND, cmd);
5111         }
5112
5113         if (!(flags & PCI_VGA_STATE_CHANGE_BRIDGE))
5114                 return 0;
5115
5116         bus = dev->bus;
5117         while (bus) {
5118                 bridge = bus->self;
5119                 if (bridge) {
5120                         pci_read_config_word(bridge, PCI_BRIDGE_CONTROL,
5121                                              &cmd);
5122                         if (decode == true)
5123                                 cmd |= PCI_BRIDGE_CTL_VGA;
5124                         else
5125                                 cmd &= ~PCI_BRIDGE_CTL_VGA;
5126                         pci_write_config_word(bridge, PCI_BRIDGE_CONTROL,
5127                                               cmd);
5128                 }
5129                 bus = bus->parent;
5130         }
5131         return 0;
5132 }
5133
5134 /**
5135  * pci_add_dma_alias - Add a DMA devfn alias for a device
5136  * @dev: the PCI device for which alias is added
5137  * @devfn: alias slot and function
5138  *
5139  * This helper encodes 8-bit devfn as bit number in dma_alias_mask.
5140  * It should be called early, preferably as PCI fixup header quirk.
5141  */
5142 void pci_add_dma_alias(struct pci_dev *dev, u8 devfn)
5143 {
5144         if (!dev->dma_alias_mask)
5145                 dev->dma_alias_mask = kcalloc(BITS_TO_LONGS(U8_MAX),
5146                                               sizeof(long), GFP_KERNEL);
5147         if (!dev->dma_alias_mask) {
5148                 dev_warn(&dev->dev, "Unable to allocate DMA alias mask\n");
5149                 return;
5150         }
5151
5152         set_bit(devfn, dev->dma_alias_mask);
5153         dev_info(&dev->dev, "Enabling fixed DMA alias to %02x.%d\n",
5154                  PCI_SLOT(devfn), PCI_FUNC(devfn));
5155 }
5156
5157 bool pci_devs_are_dma_aliases(struct pci_dev *dev1, struct pci_dev *dev2)
5158 {
5159         return (dev1->dma_alias_mask &&
5160                 test_bit(dev2->devfn, dev1->dma_alias_mask)) ||
5161                (dev2->dma_alias_mask &&
5162                 test_bit(dev1->devfn, dev2->dma_alias_mask));
5163 }
5164
5165 bool pci_device_is_present(struct pci_dev *pdev)
5166 {
5167         u32 v;
5168
5169         /* Check PF if pdev is a VF, since VF Vendor/Device IDs are 0xffff */
5170         pdev = pci_physfn(pdev);
5171         if (pci_dev_is_disconnected(pdev))
5172                 return false;
5173         return pci_bus_read_dev_vendor_id(pdev->bus, pdev->devfn, &v, 0);
5174 }
5175 EXPORT_SYMBOL_GPL(pci_device_is_present);
5176
5177 void pci_ignore_hotplug(struct pci_dev *dev)
5178 {
5179         struct pci_dev *bridge = dev->bus->self;
5180
5181         dev->ignore_hotplug = 1;
5182         /* Propagate the "ignore hotplug" setting to the parent bridge. */
5183         if (bridge)
5184                 bridge->ignore_hotplug = 1;
5185 }
5186 EXPORT_SYMBOL_GPL(pci_ignore_hotplug);
5187
5188 resource_size_t __weak pcibios_default_alignment(void)
5189 {
5190         return 0;
5191 }
5192
5193 #define RESOURCE_ALIGNMENT_PARAM_SIZE COMMAND_LINE_SIZE
5194 static char resource_alignment_param[RESOURCE_ALIGNMENT_PARAM_SIZE] = {0};
5195 static DEFINE_SPINLOCK(resource_alignment_lock);
5196
5197 /**
5198  * pci_specified_resource_alignment - get resource alignment specified by user.
5199  * @dev: the PCI device to get
5200  * @resize: whether or not to change resources' size when reassigning alignment
5201  *
5202  * RETURNS: Resource alignment if it is specified.
5203  *          Zero if it is not specified.
5204  */
5205 static resource_size_t pci_specified_resource_alignment(struct pci_dev *dev,
5206                                                         bool *resize)
5207 {
5208         int seg, bus, slot, func, align_order, count;
5209         unsigned short vendor, device, subsystem_vendor, subsystem_device;
5210         resource_size_t align = pcibios_default_alignment();
5211         char *p;
5212
5213         spin_lock(&resource_alignment_lock);
5214         p = resource_alignment_param;
5215         if (!*p && !align)
5216                 goto out;
5217         if (pci_has_flag(PCI_PROBE_ONLY)) {
5218                 align = 0;
5219                 pr_info_once("PCI: Ignoring requested alignments (PCI_PROBE_ONLY)\n");
5220                 goto out;
5221         }
5222
5223         while (*p) {
5224                 count = 0;
5225                 if (sscanf(p, "%d%n", &align_order, &count) == 1 &&
5226                                                         p[count] == '@') {
5227                         p += count + 1;
5228                 } else {
5229                         align_order = -1;
5230                 }
5231                 if (strncmp(p, "pci:", 4) == 0) {
5232                         /* PCI vendor/device (subvendor/subdevice) ids are specified */
5233                         p += 4;
5234                         if (sscanf(p, "%hx:%hx:%hx:%hx%n",
5235                                 &vendor, &device, &subsystem_vendor, &subsystem_device, &count) != 4) {
5236                                 if (sscanf(p, "%hx:%hx%n", &vendor, &device, &count) != 2) {
5237                                         printk(KERN_ERR "PCI: Can't parse resource_alignment parameter: pci:%s\n",
5238                                                 p);
5239                                         break;
5240                                 }
5241                                 subsystem_vendor = subsystem_device = 0;
5242                         }
5243                         p += count;
5244                         if ((!vendor || (vendor == dev->vendor)) &&
5245                                 (!device || (device == dev->device)) &&
5246                                 (!subsystem_vendor || (subsystem_vendor == dev->subsystem_vendor)) &&
5247                                 (!subsystem_device || (subsystem_device == dev->subsystem_device))) {
5248                                 *resize = true;
5249                                 if (align_order == -1)
5250                                         align = PAGE_SIZE;
5251                                 else
5252                                         align = 1 << align_order;
5253                                 /* Found */
5254                                 break;
5255                         }
5256                 }
5257                 else {
5258                         if (sscanf(p, "%x:%x:%x.%x%n",
5259                                 &seg, &bus, &slot, &func, &count) != 4) {
5260                                 seg = 0;
5261                                 if (sscanf(p, "%x:%x.%x%n",
5262                                                 &bus, &slot, &func, &count) != 3) {
5263                                         /* Invalid format */
5264                                         printk(KERN_ERR "PCI: Can't parse resource_alignment parameter: %s\n",
5265                                                 p);
5266                                         break;
5267                                 }
5268                         }
5269                         p += count;
5270                         if (seg == pci_domain_nr(dev->bus) &&
5271                                 bus == dev->bus->number &&
5272                                 slot == PCI_SLOT(dev->devfn) &&
5273                                 func == PCI_FUNC(dev->devfn)) {
5274                                 *resize = true;
5275                                 if (align_order == -1)
5276                                         align = PAGE_SIZE;
5277                                 else
5278                                         align = 1 << align_order;
5279                                 /* Found */
5280                                 break;
5281                         }
5282                 }
5283                 if (*p != ';' && *p != ',') {
5284                         /* End of param or invalid format */
5285                         break;
5286                 }
5287                 p++;
5288         }
5289 out:
5290         spin_unlock(&resource_alignment_lock);
5291         return align;
5292 }
5293
5294 static void pci_request_resource_alignment(struct pci_dev *dev, int bar,
5295                                            resource_size_t align, bool resize)
5296 {
5297         struct resource *r = &dev->resource[bar];
5298         resource_size_t size;
5299
5300         if (!(r->flags & IORESOURCE_MEM))
5301                 return;
5302
5303         if (r->flags & IORESOURCE_PCI_FIXED) {
5304                 dev_info(&dev->dev, "BAR%d %pR: ignoring requested alignment %#llx\n",
5305                          bar, r, (unsigned long long)align);
5306                 return;
5307         }
5308
5309         size = resource_size(r);
5310         if (size >= align)
5311                 return;
5312
5313         /*
5314          * Increase the alignment of the resource.  There are two ways we
5315          * can do this:
5316          *
5317          * 1) Increase the size of the resource.  BARs are aligned on their
5318          *    size, so when we reallocate space for this resource, we'll
5319          *    allocate it with the larger alignment.  This also prevents
5320          *    assignment of any other BARs inside the alignment region, so
5321          *    if we're requesting page alignment, this means no other BARs
5322          *    will share the page.
5323          *
5324          *    The disadvantage is that this makes the resource larger than
5325          *    the hardware BAR, which may break drivers that compute things
5326          *    based on the resource size, e.g., to find registers at a
5327          *    fixed offset before the end of the BAR.
5328          *
5329          * 2) Retain the resource size, but use IORESOURCE_STARTALIGN and
5330          *    set r->start to the desired alignment.  By itself this
5331          *    doesn't prevent other BARs being put inside the alignment
5332          *    region, but if we realign *every* resource of every device in
5333          *    the system, none of them will share an alignment region.
5334          *
5335          * When the user has requested alignment for only some devices via
5336          * the "pci=resource_alignment" argument, "resize" is true and we
5337          * use the first method.  Otherwise we assume we're aligning all
5338          * devices and we use the second.
5339          */
5340
5341         dev_info(&dev->dev, "BAR%d %pR: requesting alignment to %#llx\n",
5342                  bar, r, (unsigned long long)align);
5343
5344         if (resize) {
5345                 r->start = 0;
5346                 r->end = align - 1;
5347         } else {
5348                 r->flags &= ~IORESOURCE_SIZEALIGN;
5349                 r->flags |= IORESOURCE_STARTALIGN;
5350                 r->start = align;
5351                 r->end = r->start + size - 1;
5352         }
5353         r->flags |= IORESOURCE_UNSET;
5354 }
5355
5356 /*
5357  * This function disables memory decoding and releases memory resources
5358  * of the device specified by kernel's boot parameter 'pci=resource_alignment='.
5359  * It also rounds up size to specified alignment.
5360  * Later on, the kernel will assign page-aligned memory resource back
5361  * to the device.
5362  */
5363 void pci_reassigndev_resource_alignment(struct pci_dev *dev)
5364 {
5365         int i;
5366         struct resource *r;
5367         resource_size_t align;
5368         u16 command;
5369         bool resize = false;
5370
5371         /*
5372          * VF BARs are read-only zero according to SR-IOV spec r1.1, sec
5373          * 3.4.1.11.  Their resources are allocated from the space
5374          * described by the VF BARx register in the PF's SR-IOV capability.
5375          * We can't influence their alignment here.
5376          */
5377         if (dev->is_virtfn)
5378                 return;
5379
5380         /* check if specified PCI is target device to reassign */
5381         align = pci_specified_resource_alignment(dev, &resize);
5382         if (!align)
5383                 return;
5384
5385         if (dev->hdr_type == PCI_HEADER_TYPE_NORMAL &&
5386             (dev->class >> 8) == PCI_CLASS_BRIDGE_HOST) {
5387                 dev_warn(&dev->dev,
5388                         "Can't reassign resources to host bridge.\n");
5389                 return;
5390         }
5391
5392         dev_info(&dev->dev,
5393                 "Disabling memory decoding and releasing memory resources.\n");
5394         pci_read_config_word(dev, PCI_COMMAND, &command);
5395         command &= ~PCI_COMMAND_MEMORY;
5396         pci_write_config_word(dev, PCI_COMMAND, command);
5397
5398         for (i = 0; i <= PCI_ROM_RESOURCE; i++)
5399                 pci_request_resource_alignment(dev, i, align, resize);
5400
5401         /*
5402          * Need to disable bridge's resource window,
5403          * to enable the kernel to reassign new resource
5404          * window later on.
5405          */
5406         if (dev->hdr_type == PCI_HEADER_TYPE_BRIDGE &&
5407             (dev->class >> 8) == PCI_CLASS_BRIDGE_PCI) {
5408                 for (i = PCI_BRIDGE_RESOURCES; i < PCI_NUM_RESOURCES; i++) {
5409                         r = &dev->resource[i];
5410                         if (!(r->flags & IORESOURCE_MEM))
5411                                 continue;
5412                         r->flags |= IORESOURCE_UNSET;
5413                         r->end = resource_size(r) - 1;
5414                         r->start = 0;
5415                 }
5416                 pci_disable_bridge_window(dev);
5417         }
5418 }
5419
5420 static ssize_t pci_set_resource_alignment_param(const char *buf, size_t count)
5421 {
5422         if (count > RESOURCE_ALIGNMENT_PARAM_SIZE - 1)
5423                 count = RESOURCE_ALIGNMENT_PARAM_SIZE - 1;
5424         spin_lock(&resource_alignment_lock);
5425         strncpy(resource_alignment_param, buf, count);
5426         resource_alignment_param[count] = '\0';
5427         spin_unlock(&resource_alignment_lock);
5428         return count;
5429 }
5430
5431 static ssize_t pci_get_resource_alignment_param(char *buf, size_t size)
5432 {
5433         size_t count;
5434         spin_lock(&resource_alignment_lock);
5435         count = snprintf(buf, size, "%s", resource_alignment_param);
5436         spin_unlock(&resource_alignment_lock);
5437         return count;
5438 }
5439
5440 static ssize_t pci_resource_alignment_show(struct bus_type *bus, char *buf)
5441 {
5442         return pci_get_resource_alignment_param(buf, PAGE_SIZE);
5443 }
5444
5445 static ssize_t pci_resource_alignment_store(struct bus_type *bus,
5446                                         const char *buf, size_t count)
5447 {
5448         return pci_set_resource_alignment_param(buf, count);
5449 }
5450
5451 static BUS_ATTR(resource_alignment, 0644, pci_resource_alignment_show,
5452                                         pci_resource_alignment_store);
5453
5454 static int __init pci_resource_alignment_sysfs_init(void)
5455 {
5456         return bus_create_file(&pci_bus_type,
5457                                         &bus_attr_resource_alignment);
5458 }
5459 late_initcall(pci_resource_alignment_sysfs_init);
5460
5461 static void pci_no_domains(void)
5462 {
5463 #ifdef CONFIG_PCI_DOMAINS
5464         pci_domains_supported = 0;
5465 #endif
5466 }
5467
5468 #ifdef CONFIG_PCI_DOMAINS
5469 static atomic_t __domain_nr = ATOMIC_INIT(-1);
5470
5471 int pci_get_new_domain_nr(void)
5472 {
5473         return atomic_inc_return(&__domain_nr);
5474 }
5475
5476 #ifdef CONFIG_PCI_DOMAINS_GENERIC
5477 static int of_pci_bus_find_domain_nr(struct device *parent)
5478 {
5479         static int use_dt_domains = -1;
5480         int domain = -1;
5481
5482         if (parent)
5483                 domain = of_get_pci_domain_nr(parent->of_node);
5484         /*
5485          * Check DT domain and use_dt_domains values.
5486          *
5487          * If DT domain property is valid (domain >= 0) and
5488          * use_dt_domains != 0, the DT assignment is valid since this means
5489          * we have not previously allocated a domain number by using
5490          * pci_get_new_domain_nr(); we should also update use_dt_domains to
5491          * 1, to indicate that we have just assigned a domain number from
5492          * DT.
5493          *
5494          * If DT domain property value is not valid (ie domain < 0), and we
5495          * have not previously assigned a domain number from DT
5496          * (use_dt_domains != 1) we should assign a domain number by
5497          * using the:
5498          *
5499          * pci_get_new_domain_nr()
5500          *
5501          * API and update the use_dt_domains value to keep track of method we
5502          * are using to assign domain numbers (use_dt_domains = 0).
5503          *
5504          * All other combinations imply we have a platform that is trying
5505          * to mix domain numbers obtained from DT and pci_get_new_domain_nr(),
5506          * which is a recipe for domain mishandling and it is prevented by
5507          * invalidating the domain value (domain = -1) and printing a
5508          * corresponding error.
5509          */
5510         if (domain >= 0 && use_dt_domains) {
5511                 use_dt_domains = 1;
5512         } else if (domain < 0 && use_dt_domains != 1) {
5513                 use_dt_domains = 0;
5514                 domain = pci_get_new_domain_nr();
5515         } else {
5516                 dev_err(parent, "Node %pOF has inconsistent \"linux,pci-domain\" property in DT\n",
5517                         parent->of_node);
5518                 domain = -1;
5519         }
5520
5521         return domain;
5522 }
5523
5524 int pci_bus_find_domain_nr(struct pci_bus *bus, struct device *parent)
5525 {
5526         return acpi_disabled ? of_pci_bus_find_domain_nr(parent) :
5527                                acpi_pci_bus_find_domain_nr(bus);
5528 }
5529 #endif
5530 #endif
5531
5532 /**
5533  * pci_ext_cfg_avail - can we access extended PCI config space?
5534  *
5535  * Returns 1 if we can access PCI extended config space (offsets
5536  * greater than 0xff). This is the default implementation. Architecture
5537  * implementations can override this.
5538  */
5539 int __weak pci_ext_cfg_avail(void)
5540 {
5541         return 1;
5542 }
5543
5544 void __weak pci_fixup_cardbus(struct pci_bus *bus)
5545 {
5546 }
5547 EXPORT_SYMBOL(pci_fixup_cardbus);
5548
5549 static int __init pci_setup(char *str)
5550 {
5551         while (str) {
5552                 char *k = strchr(str, ',');
5553                 if (k)
5554                         *k++ = 0;
5555                 if (*str && (str = pcibios_setup(str)) && *str) {
5556                         if (!strcmp(str, "nomsi")) {
5557                                 pci_no_msi();
5558                         } else if (!strcmp(str, "noaer")) {
5559                                 pci_no_aer();
5560                         } else if (!strncmp(str, "realloc=", 8)) {
5561                                 pci_realloc_get_opt(str + 8);
5562                         } else if (!strncmp(str, "realloc", 7)) {
5563                                 pci_realloc_get_opt("on");
5564                         } else if (!strcmp(str, "nodomains")) {
5565                                 pci_no_domains();
5566                         } else if (!strncmp(str, "noari", 5)) {
5567                                 pcie_ari_disabled = true;
5568                         } else if (!strncmp(str, "cbiosize=", 9)) {
5569                                 pci_cardbus_io_size = memparse(str + 9, &str);
5570                         } else if (!strncmp(str, "cbmemsize=", 10)) {
5571                                 pci_cardbus_mem_size = memparse(str + 10, &str);
5572                         } else if (!strncmp(str, "resource_alignment=", 19)) {
5573                                 pci_set_resource_alignment_param(str + 19,
5574                                                         strlen(str + 19));
5575                         } else if (!strncmp(str, "ecrc=", 5)) {
5576                                 pcie_ecrc_get_policy(str + 5);
5577                         } else if (!strncmp(str, "hpiosize=", 9)) {
5578                                 pci_hotplug_io_size = memparse(str + 9, &str);
5579                         } else if (!strncmp(str, "hpmemsize=", 10)) {
5580                                 pci_hotplug_mem_size = memparse(str + 10, &str);
5581                         } else if (!strncmp(str, "hpbussize=", 10)) {
5582                                 pci_hotplug_bus_size =
5583                                         simple_strtoul(str + 10, &str, 0);
5584                                 if (pci_hotplug_bus_size > 0xff)
5585                                         pci_hotplug_bus_size = DEFAULT_HOTPLUG_BUS_SIZE;
5586                         } else if (!strncmp(str, "pcie_bus_tune_off", 17)) {
5587                                 pcie_bus_config = PCIE_BUS_TUNE_OFF;
5588                         } else if (!strncmp(str, "pcie_bus_safe", 13)) {
5589                                 pcie_bus_config = PCIE_BUS_SAFE;
5590                         } else if (!strncmp(str, "pcie_bus_perf", 13)) {
5591                                 pcie_bus_config = PCIE_BUS_PERFORMANCE;
5592                         } else if (!strncmp(str, "pcie_bus_peer2peer", 18)) {
5593                                 pcie_bus_config = PCIE_BUS_PEER2PEER;
5594                         } else if (!strncmp(str, "pcie_scan_all", 13)) {
5595                                 pci_add_flags(PCI_SCAN_ALL_PCIE_DEVS);
5596                         } else {
5597                                 printk(KERN_ERR "PCI: Unknown option `%s'\n",
5598                                                 str);
5599                         }
5600                 }
5601                 str = k;
5602         }
5603         return 0;
5604 }
5605 early_param("pci", pci_setup);