Merge branch 'pci/sparc'
[muen/linux.git] / drivers / pci / pci.c
1 // SPDX-License-Identifier: GPL-2.0
2 /*
3  * PCI Bus Services, see include/linux/pci.h for further explanation.
4  *
5  * Copyright 1993 -- 1997 Drew Eckhardt, Frederic Potter,
6  * David Mosberger-Tang
7  *
8  * Copyright 1997 -- 2000 Martin Mares <mj@ucw.cz>
9  */
10
11 #include <linux/acpi.h>
12 #include <linux/kernel.h>
13 #include <linux/delay.h>
14 #include <linux/dmi.h>
15 #include <linux/init.h>
16 #include <linux/of.h>
17 #include <linux/of_pci.h>
18 #include <linux/pci.h>
19 #include <linux/pm.h>
20 #include <linux/slab.h>
21 #include <linux/module.h>
22 #include <linux/spinlock.h>
23 #include <linux/string.h>
24 #include <linux/log2.h>
25 #include <linux/logic_pio.h>
26 #include <linux/pci-aspm.h>
27 #include <linux/pm_wakeup.h>
28 #include <linux/interrupt.h>
29 #include <linux/device.h>
30 #include <linux/pm_runtime.h>
31 #include <linux/pci_hotplug.h>
32 #include <linux/vmalloc.h>
33 #include <linux/pci-ats.h>
34 #include <asm/setup.h>
35 #include <asm/dma.h>
36 #include <linux/aer.h>
37 #include "pci.h"
38
39 const char *pci_power_names[] = {
40         "error", "D0", "D1", "D2", "D3hot", "D3cold", "unknown",
41 };
42 EXPORT_SYMBOL_GPL(pci_power_names);
43
44 int isa_dma_bridge_buggy;
45 EXPORT_SYMBOL(isa_dma_bridge_buggy);
46
47 int pci_pci_problems;
48 EXPORT_SYMBOL(pci_pci_problems);
49
50 unsigned int pci_pm_d3_delay;
51
52 static void pci_pme_list_scan(struct work_struct *work);
53
54 static LIST_HEAD(pci_pme_list);
55 static DEFINE_MUTEX(pci_pme_list_mutex);
56 static DECLARE_DELAYED_WORK(pci_pme_work, pci_pme_list_scan);
57
58 struct pci_pme_device {
59         struct list_head list;
60         struct pci_dev *dev;
61 };
62
63 #define PME_TIMEOUT 1000 /* How long between PME checks */
64
65 static void pci_dev_d3_sleep(struct pci_dev *dev)
66 {
67         unsigned int delay = dev->d3_delay;
68
69         if (delay < pci_pm_d3_delay)
70                 delay = pci_pm_d3_delay;
71
72         if (delay)
73                 msleep(delay);
74 }
75
76 #ifdef CONFIG_PCI_DOMAINS
77 int pci_domains_supported = 1;
78 #endif
79
80 #define DEFAULT_CARDBUS_IO_SIZE         (256)
81 #define DEFAULT_CARDBUS_MEM_SIZE        (64*1024*1024)
82 /* pci=cbmemsize=nnM,cbiosize=nn can override this */
83 unsigned long pci_cardbus_io_size = DEFAULT_CARDBUS_IO_SIZE;
84 unsigned long pci_cardbus_mem_size = DEFAULT_CARDBUS_MEM_SIZE;
85
86 #define DEFAULT_HOTPLUG_IO_SIZE         (256)
87 #define DEFAULT_HOTPLUG_MEM_SIZE        (2*1024*1024)
88 /* pci=hpmemsize=nnM,hpiosize=nn can override this */
89 unsigned long pci_hotplug_io_size  = DEFAULT_HOTPLUG_IO_SIZE;
90 unsigned long pci_hotplug_mem_size = DEFAULT_HOTPLUG_MEM_SIZE;
91
92 #define DEFAULT_HOTPLUG_BUS_SIZE        1
93 unsigned long pci_hotplug_bus_size = DEFAULT_HOTPLUG_BUS_SIZE;
94
95 enum pcie_bus_config_types pcie_bus_config = PCIE_BUS_DEFAULT;
96
97 /*
98  * The default CLS is used if arch didn't set CLS explicitly and not
99  * all pci devices agree on the same value.  Arch can override either
100  * the dfl or actual value as it sees fit.  Don't forget this is
101  * measured in 32-bit words, not bytes.
102  */
103 u8 pci_dfl_cache_line_size = L1_CACHE_BYTES >> 2;
104 u8 pci_cache_line_size;
105
106 /*
107  * If we set up a device for bus mastering, we need to check the latency
108  * timer as certain BIOSes forget to set it properly.
109  */
110 unsigned int pcibios_max_latency = 255;
111
112 /* If set, the PCIe ARI capability will not be used. */
113 static bool pcie_ari_disabled;
114
115 /* Disable bridge_d3 for all PCIe ports */
116 static bool pci_bridge_d3_disable;
117 /* Force bridge_d3 for all PCIe ports */
118 static bool pci_bridge_d3_force;
119
120 static int __init pcie_port_pm_setup(char *str)
121 {
122         if (!strcmp(str, "off"))
123                 pci_bridge_d3_disable = true;
124         else if (!strcmp(str, "force"))
125                 pci_bridge_d3_force = true;
126         return 1;
127 }
128 __setup("pcie_port_pm=", pcie_port_pm_setup);
129
130 /**
131  * pci_bus_max_busnr - returns maximum PCI bus number of given bus' children
132  * @bus: pointer to PCI bus structure to search
133  *
134  * Given a PCI bus, returns the highest PCI bus number present in the set
135  * including the given PCI bus and its list of child PCI buses.
136  */
137 unsigned char pci_bus_max_busnr(struct pci_bus *bus)
138 {
139         struct pci_bus *tmp;
140         unsigned char max, n;
141
142         max = bus->busn_res.end;
143         list_for_each_entry(tmp, &bus->children, node) {
144                 n = pci_bus_max_busnr(tmp);
145                 if (n > max)
146                         max = n;
147         }
148         return max;
149 }
150 EXPORT_SYMBOL_GPL(pci_bus_max_busnr);
151
152 #ifdef CONFIG_HAS_IOMEM
153 void __iomem *pci_ioremap_bar(struct pci_dev *pdev, int bar)
154 {
155         struct resource *res = &pdev->resource[bar];
156
157         /*
158          * Make sure the BAR is actually a memory resource, not an IO resource
159          */
160         if (res->flags & IORESOURCE_UNSET || !(res->flags & IORESOURCE_MEM)) {
161                 pci_warn(pdev, "can't ioremap BAR %d: %pR\n", bar, res);
162                 return NULL;
163         }
164         return ioremap_nocache(res->start, resource_size(res));
165 }
166 EXPORT_SYMBOL_GPL(pci_ioremap_bar);
167
168 void __iomem *pci_ioremap_wc_bar(struct pci_dev *pdev, int bar)
169 {
170         /*
171          * Make sure the BAR is actually a memory resource, not an IO resource
172          */
173         if (!(pci_resource_flags(pdev, bar) & IORESOURCE_MEM)) {
174                 WARN_ON(1);
175                 return NULL;
176         }
177         return ioremap_wc(pci_resource_start(pdev, bar),
178                           pci_resource_len(pdev, bar));
179 }
180 EXPORT_SYMBOL_GPL(pci_ioremap_wc_bar);
181 #endif
182
183
184 static int __pci_find_next_cap_ttl(struct pci_bus *bus, unsigned int devfn,
185                                    u8 pos, int cap, int *ttl)
186 {
187         u8 id;
188         u16 ent;
189
190         pci_bus_read_config_byte(bus, devfn, pos, &pos);
191
192         while ((*ttl)--) {
193                 if (pos < 0x40)
194                         break;
195                 pos &= ~3;
196                 pci_bus_read_config_word(bus, devfn, pos, &ent);
197
198                 id = ent & 0xff;
199                 if (id == 0xff)
200                         break;
201                 if (id == cap)
202                         return pos;
203                 pos = (ent >> 8);
204         }
205         return 0;
206 }
207
208 static int __pci_find_next_cap(struct pci_bus *bus, unsigned int devfn,
209                                u8 pos, int cap)
210 {
211         int ttl = PCI_FIND_CAP_TTL;
212
213         return __pci_find_next_cap_ttl(bus, devfn, pos, cap, &ttl);
214 }
215
216 int pci_find_next_capability(struct pci_dev *dev, u8 pos, int cap)
217 {
218         return __pci_find_next_cap(dev->bus, dev->devfn,
219                                    pos + PCI_CAP_LIST_NEXT, cap);
220 }
221 EXPORT_SYMBOL_GPL(pci_find_next_capability);
222
223 static int __pci_bus_find_cap_start(struct pci_bus *bus,
224                                     unsigned int devfn, u8 hdr_type)
225 {
226         u16 status;
227
228         pci_bus_read_config_word(bus, devfn, PCI_STATUS, &status);
229         if (!(status & PCI_STATUS_CAP_LIST))
230                 return 0;
231
232         switch (hdr_type) {
233         case PCI_HEADER_TYPE_NORMAL:
234         case PCI_HEADER_TYPE_BRIDGE:
235                 return PCI_CAPABILITY_LIST;
236         case PCI_HEADER_TYPE_CARDBUS:
237                 return PCI_CB_CAPABILITY_LIST;
238         }
239
240         return 0;
241 }
242
243 /**
244  * pci_find_capability - query for devices' capabilities
245  * @dev: PCI device to query
246  * @cap: capability code
247  *
248  * Tell if a device supports a given PCI capability.
249  * Returns the address of the requested capability structure within the
250  * device's PCI configuration space or 0 in case the device does not
251  * support it.  Possible values for @cap:
252  *
253  *  %PCI_CAP_ID_PM           Power Management
254  *  %PCI_CAP_ID_AGP          Accelerated Graphics Port
255  *  %PCI_CAP_ID_VPD          Vital Product Data
256  *  %PCI_CAP_ID_SLOTID       Slot Identification
257  *  %PCI_CAP_ID_MSI          Message Signalled Interrupts
258  *  %PCI_CAP_ID_CHSWP        CompactPCI HotSwap
259  *  %PCI_CAP_ID_PCIX         PCI-X
260  *  %PCI_CAP_ID_EXP          PCI Express
261  */
262 int pci_find_capability(struct pci_dev *dev, int cap)
263 {
264         int pos;
265
266         pos = __pci_bus_find_cap_start(dev->bus, dev->devfn, dev->hdr_type);
267         if (pos)
268                 pos = __pci_find_next_cap(dev->bus, dev->devfn, pos, cap);
269
270         return pos;
271 }
272 EXPORT_SYMBOL(pci_find_capability);
273
274 /**
275  * pci_bus_find_capability - query for devices' capabilities
276  * @bus:   the PCI bus to query
277  * @devfn: PCI device to query
278  * @cap:   capability code
279  *
280  * Like pci_find_capability() but works for pci devices that do not have a
281  * pci_dev structure set up yet.
282  *
283  * Returns the address of the requested capability structure within the
284  * device's PCI configuration space or 0 in case the device does not
285  * support it.
286  */
287 int pci_bus_find_capability(struct pci_bus *bus, unsigned int devfn, int cap)
288 {
289         int pos;
290         u8 hdr_type;
291
292         pci_bus_read_config_byte(bus, devfn, PCI_HEADER_TYPE, &hdr_type);
293
294         pos = __pci_bus_find_cap_start(bus, devfn, hdr_type & 0x7f);
295         if (pos)
296                 pos = __pci_find_next_cap(bus, devfn, pos, cap);
297
298         return pos;
299 }
300 EXPORT_SYMBOL(pci_bus_find_capability);
301
302 /**
303  * pci_find_next_ext_capability - Find an extended capability
304  * @dev: PCI device to query
305  * @start: address at which to start looking (0 to start at beginning of list)
306  * @cap: capability code
307  *
308  * Returns the address of the next matching extended capability structure
309  * within the device's PCI configuration space or 0 if the device does
310  * not support it.  Some capabilities can occur several times, e.g., the
311  * vendor-specific capability, and this provides a way to find them all.
312  */
313 int pci_find_next_ext_capability(struct pci_dev *dev, int start, int cap)
314 {
315         u32 header;
316         int ttl;
317         int pos = PCI_CFG_SPACE_SIZE;
318
319         /* minimum 8 bytes per capability */
320         ttl = (PCI_CFG_SPACE_EXP_SIZE - PCI_CFG_SPACE_SIZE) / 8;
321
322         if (dev->cfg_size <= PCI_CFG_SPACE_SIZE)
323                 return 0;
324
325         if (start)
326                 pos = start;
327
328         if (pci_read_config_dword(dev, pos, &header) != PCIBIOS_SUCCESSFUL)
329                 return 0;
330
331         /*
332          * If we have no capabilities, this is indicated by cap ID,
333          * cap version and next pointer all being 0.
334          */
335         if (header == 0)
336                 return 0;
337
338         while (ttl-- > 0) {
339                 if (PCI_EXT_CAP_ID(header) == cap && pos != start)
340                         return pos;
341
342                 pos = PCI_EXT_CAP_NEXT(header);
343                 if (pos < PCI_CFG_SPACE_SIZE)
344                         break;
345
346                 if (pci_read_config_dword(dev, pos, &header) != PCIBIOS_SUCCESSFUL)
347                         break;
348         }
349
350         return 0;
351 }
352 EXPORT_SYMBOL_GPL(pci_find_next_ext_capability);
353
354 /**
355  * pci_find_ext_capability - Find an extended capability
356  * @dev: PCI device to query
357  * @cap: capability code
358  *
359  * Returns the address of the requested extended capability structure
360  * within the device's PCI configuration space or 0 if the device does
361  * not support it.  Possible values for @cap:
362  *
363  *  %PCI_EXT_CAP_ID_ERR         Advanced Error Reporting
364  *  %PCI_EXT_CAP_ID_VC          Virtual Channel
365  *  %PCI_EXT_CAP_ID_DSN         Device Serial Number
366  *  %PCI_EXT_CAP_ID_PWR         Power Budgeting
367  */
368 int pci_find_ext_capability(struct pci_dev *dev, int cap)
369 {
370         return pci_find_next_ext_capability(dev, 0, cap);
371 }
372 EXPORT_SYMBOL_GPL(pci_find_ext_capability);
373
374 static int __pci_find_next_ht_cap(struct pci_dev *dev, int pos, int ht_cap)
375 {
376         int rc, ttl = PCI_FIND_CAP_TTL;
377         u8 cap, mask;
378
379         if (ht_cap == HT_CAPTYPE_SLAVE || ht_cap == HT_CAPTYPE_HOST)
380                 mask = HT_3BIT_CAP_MASK;
381         else
382                 mask = HT_5BIT_CAP_MASK;
383
384         pos = __pci_find_next_cap_ttl(dev->bus, dev->devfn, pos,
385                                       PCI_CAP_ID_HT, &ttl);
386         while (pos) {
387                 rc = pci_read_config_byte(dev, pos + 3, &cap);
388                 if (rc != PCIBIOS_SUCCESSFUL)
389                         return 0;
390
391                 if ((cap & mask) == ht_cap)
392                         return pos;
393
394                 pos = __pci_find_next_cap_ttl(dev->bus, dev->devfn,
395                                               pos + PCI_CAP_LIST_NEXT,
396                                               PCI_CAP_ID_HT, &ttl);
397         }
398
399         return 0;
400 }
401 /**
402  * pci_find_next_ht_capability - query a device's Hypertransport capabilities
403  * @dev: PCI device to query
404  * @pos: Position from which to continue searching
405  * @ht_cap: Hypertransport capability code
406  *
407  * To be used in conjunction with pci_find_ht_capability() to search for
408  * all capabilities matching @ht_cap. @pos should always be a value returned
409  * from pci_find_ht_capability().
410  *
411  * NB. To be 100% safe against broken PCI devices, the caller should take
412  * steps to avoid an infinite loop.
413  */
414 int pci_find_next_ht_capability(struct pci_dev *dev, int pos, int ht_cap)
415 {
416         return __pci_find_next_ht_cap(dev, pos + PCI_CAP_LIST_NEXT, ht_cap);
417 }
418 EXPORT_SYMBOL_GPL(pci_find_next_ht_capability);
419
420 /**
421  * pci_find_ht_capability - query a device's Hypertransport capabilities
422  * @dev: PCI device to query
423  * @ht_cap: Hypertransport capability code
424  *
425  * Tell if a device supports a given Hypertransport capability.
426  * Returns an address within the device's PCI configuration space
427  * or 0 in case the device does not support the request capability.
428  * The address points to the PCI capability, of type PCI_CAP_ID_HT,
429  * which has a Hypertransport capability matching @ht_cap.
430  */
431 int pci_find_ht_capability(struct pci_dev *dev, int ht_cap)
432 {
433         int pos;
434
435         pos = __pci_bus_find_cap_start(dev->bus, dev->devfn, dev->hdr_type);
436         if (pos)
437                 pos = __pci_find_next_ht_cap(dev, pos, ht_cap);
438
439         return pos;
440 }
441 EXPORT_SYMBOL_GPL(pci_find_ht_capability);
442
443 /**
444  * pci_find_parent_resource - return resource region of parent bus of given region
445  * @dev: PCI device structure contains resources to be searched
446  * @res: child resource record for which parent is sought
447  *
448  *  For given resource region of given device, return the resource
449  *  region of parent bus the given region is contained in.
450  */
451 struct resource *pci_find_parent_resource(const struct pci_dev *dev,
452                                           struct resource *res)
453 {
454         const struct pci_bus *bus = dev->bus;
455         struct resource *r;
456         int i;
457
458         pci_bus_for_each_resource(bus, r, i) {
459                 if (!r)
460                         continue;
461                 if (resource_contains(r, res)) {
462
463                         /*
464                          * If the window is prefetchable but the BAR is
465                          * not, the allocator made a mistake.
466                          */
467                         if (r->flags & IORESOURCE_PREFETCH &&
468                             !(res->flags & IORESOURCE_PREFETCH))
469                                 return NULL;
470
471                         /*
472                          * If we're below a transparent bridge, there may
473                          * be both a positively-decoded aperture and a
474                          * subtractively-decoded region that contain the BAR.
475                          * We want the positively-decoded one, so this depends
476                          * on pci_bus_for_each_resource() giving us those
477                          * first.
478                          */
479                         return r;
480                 }
481         }
482         return NULL;
483 }
484 EXPORT_SYMBOL(pci_find_parent_resource);
485
486 /**
487  * pci_find_resource - Return matching PCI device resource
488  * @dev: PCI device to query
489  * @res: Resource to look for
490  *
491  * Goes over standard PCI resources (BARs) and checks if the given resource
492  * is partially or fully contained in any of them. In that case the
493  * matching resource is returned, %NULL otherwise.
494  */
495 struct resource *pci_find_resource(struct pci_dev *dev, struct resource *res)
496 {
497         int i;
498
499         for (i = 0; i < PCI_ROM_RESOURCE; i++) {
500                 struct resource *r = &dev->resource[i];
501
502                 if (r->start && resource_contains(r, res))
503                         return r;
504         }
505
506         return NULL;
507 }
508 EXPORT_SYMBOL(pci_find_resource);
509
510 /**
511  * pci_find_pcie_root_port - return PCIe Root Port
512  * @dev: PCI device to query
513  *
514  * Traverse up the parent chain and return the PCIe Root Port PCI Device
515  * for a given PCI Device.
516  */
517 struct pci_dev *pci_find_pcie_root_port(struct pci_dev *dev)
518 {
519         struct pci_dev *bridge, *highest_pcie_bridge = dev;
520
521         bridge = pci_upstream_bridge(dev);
522         while (bridge && pci_is_pcie(bridge)) {
523                 highest_pcie_bridge = bridge;
524                 bridge = pci_upstream_bridge(bridge);
525         }
526
527         if (pci_pcie_type(highest_pcie_bridge) != PCI_EXP_TYPE_ROOT_PORT)
528                 return NULL;
529
530         return highest_pcie_bridge;
531 }
532 EXPORT_SYMBOL(pci_find_pcie_root_port);
533
534 /**
535  * pci_wait_for_pending - wait for @mask bit(s) to clear in status word @pos
536  * @dev: the PCI device to operate on
537  * @pos: config space offset of status word
538  * @mask: mask of bit(s) to care about in status word
539  *
540  * Return 1 when mask bit(s) in status word clear, 0 otherwise.
541  */
542 int pci_wait_for_pending(struct pci_dev *dev, int pos, u16 mask)
543 {
544         int i;
545
546         /* Wait for Transaction Pending bit clean */
547         for (i = 0; i < 4; i++) {
548                 u16 status;
549                 if (i)
550                         msleep((1 << (i - 1)) * 100);
551
552                 pci_read_config_word(dev, pos, &status);
553                 if (!(status & mask))
554                         return 1;
555         }
556
557         return 0;
558 }
559
560 /**
561  * pci_restore_bars - restore a device's BAR values (e.g. after wake-up)
562  * @dev: PCI device to have its BARs restored
563  *
564  * Restore the BAR values for a given device, so as to make it
565  * accessible by its driver.
566  */
567 static void pci_restore_bars(struct pci_dev *dev)
568 {
569         int i;
570
571         for (i = 0; i < PCI_BRIDGE_RESOURCES; i++)
572                 pci_update_resource(dev, i);
573 }
574
575 static const struct pci_platform_pm_ops *pci_platform_pm;
576
577 int pci_set_platform_pm(const struct pci_platform_pm_ops *ops)
578 {
579         if (!ops->is_manageable || !ops->set_state  || !ops->get_state ||
580             !ops->choose_state  || !ops->set_wakeup || !ops->need_resume)
581                 return -EINVAL;
582         pci_platform_pm = ops;
583         return 0;
584 }
585
586 static inline bool platform_pci_power_manageable(struct pci_dev *dev)
587 {
588         return pci_platform_pm ? pci_platform_pm->is_manageable(dev) : false;
589 }
590
591 static inline int platform_pci_set_power_state(struct pci_dev *dev,
592                                                pci_power_t t)
593 {
594         return pci_platform_pm ? pci_platform_pm->set_state(dev, t) : -ENOSYS;
595 }
596
597 static inline pci_power_t platform_pci_get_power_state(struct pci_dev *dev)
598 {
599         return pci_platform_pm ? pci_platform_pm->get_state(dev) : PCI_UNKNOWN;
600 }
601
602 static inline pci_power_t platform_pci_choose_state(struct pci_dev *dev)
603 {
604         return pci_platform_pm ?
605                         pci_platform_pm->choose_state(dev) : PCI_POWER_ERROR;
606 }
607
608 static inline int platform_pci_set_wakeup(struct pci_dev *dev, bool enable)
609 {
610         return pci_platform_pm ?
611                         pci_platform_pm->set_wakeup(dev, enable) : -ENODEV;
612 }
613
614 static inline bool platform_pci_need_resume(struct pci_dev *dev)
615 {
616         return pci_platform_pm ? pci_platform_pm->need_resume(dev) : false;
617 }
618
619 /**
620  * pci_raw_set_power_state - Use PCI PM registers to set the power state of
621  *                           given PCI device
622  * @dev: PCI device to handle.
623  * @state: PCI power state (D0, D1, D2, D3hot) to put the device into.
624  *
625  * RETURN VALUE:
626  * -EINVAL if the requested state is invalid.
627  * -EIO if device does not support PCI PM or its PM capabilities register has a
628  * wrong version, or device doesn't support the requested state.
629  * 0 if device already is in the requested state.
630  * 0 if device's power state has been successfully changed.
631  */
632 static int pci_raw_set_power_state(struct pci_dev *dev, pci_power_t state)
633 {
634         u16 pmcsr;
635         bool need_restore = false;
636
637         /* Check if we're already there */
638         if (dev->current_state == state)
639                 return 0;
640
641         if (!dev->pm_cap)
642                 return -EIO;
643
644         if (state < PCI_D0 || state > PCI_D3hot)
645                 return -EINVAL;
646
647         /* Validate current state:
648          * Can enter D0 from any state, but if we can only go deeper
649          * to sleep if we're already in a low power state
650          */
651         if (state != PCI_D0 && dev->current_state <= PCI_D3cold
652             && dev->current_state > state) {
653                 pci_err(dev, "invalid power transition (from state %d to %d)\n",
654                         dev->current_state, state);
655                 return -EINVAL;
656         }
657
658         /* check if this device supports the desired state */
659         if ((state == PCI_D1 && !dev->d1_support)
660            || (state == PCI_D2 && !dev->d2_support))
661                 return -EIO;
662
663         pci_read_config_word(dev, dev->pm_cap + PCI_PM_CTRL, &pmcsr);
664
665         /* If we're (effectively) in D3, force entire word to 0.
666          * This doesn't affect PME_Status, disables PME_En, and
667          * sets PowerState to 0.
668          */
669         switch (dev->current_state) {
670         case PCI_D0:
671         case PCI_D1:
672         case PCI_D2:
673                 pmcsr &= ~PCI_PM_CTRL_STATE_MASK;
674                 pmcsr |= state;
675                 break;
676         case PCI_D3hot:
677         case PCI_D3cold:
678         case PCI_UNKNOWN: /* Boot-up */
679                 if ((pmcsr & PCI_PM_CTRL_STATE_MASK) == PCI_D3hot
680                  && !(pmcsr & PCI_PM_CTRL_NO_SOFT_RESET))
681                         need_restore = true;
682                 /* Fall-through: force to D0 */
683         default:
684                 pmcsr = 0;
685                 break;
686         }
687
688         /* enter specified state */
689         pci_write_config_word(dev, dev->pm_cap + PCI_PM_CTRL, pmcsr);
690
691         /* Mandatory power management transition delays */
692         /* see PCI PM 1.1 5.6.1 table 18 */
693         if (state == PCI_D3hot || dev->current_state == PCI_D3hot)
694                 pci_dev_d3_sleep(dev);
695         else if (state == PCI_D2 || dev->current_state == PCI_D2)
696                 udelay(PCI_PM_D2_DELAY);
697
698         pci_read_config_word(dev, dev->pm_cap + PCI_PM_CTRL, &pmcsr);
699         dev->current_state = (pmcsr & PCI_PM_CTRL_STATE_MASK);
700         if (dev->current_state != state && printk_ratelimit())
701                 pci_info(dev, "Refused to change power state, currently in D%d\n",
702                          dev->current_state);
703
704         /*
705          * According to section 5.4.1 of the "PCI BUS POWER MANAGEMENT
706          * INTERFACE SPECIFICATION, REV. 1.2", a device transitioning
707          * from D3hot to D0 _may_ perform an internal reset, thereby
708          * going to "D0 Uninitialized" rather than "D0 Initialized".
709          * For example, at least some versions of the 3c905B and the
710          * 3c556B exhibit this behaviour.
711          *
712          * At least some laptop BIOSen (e.g. the Thinkpad T21) leave
713          * devices in a D3hot state at boot.  Consequently, we need to
714          * restore at least the BARs so that the device will be
715          * accessible to its driver.
716          */
717         if (need_restore)
718                 pci_restore_bars(dev);
719
720         if (dev->bus->self)
721                 pcie_aspm_pm_state_change(dev->bus->self);
722
723         return 0;
724 }
725
726 /**
727  * pci_update_current_state - Read power state of given device and cache it
728  * @dev: PCI device to handle.
729  * @state: State to cache in case the device doesn't have the PM capability
730  *
731  * The power state is read from the PMCSR register, which however is
732  * inaccessible in D3cold.  The platform firmware is therefore queried first
733  * to detect accessibility of the register.  In case the platform firmware
734  * reports an incorrect state or the device isn't power manageable by the
735  * platform at all, we try to detect D3cold by testing accessibility of the
736  * vendor ID in config space.
737  */
738 void pci_update_current_state(struct pci_dev *dev, pci_power_t state)
739 {
740         if (platform_pci_get_power_state(dev) == PCI_D3cold ||
741             !pci_device_is_present(dev)) {
742                 dev->current_state = PCI_D3cold;
743         } else if (dev->pm_cap) {
744                 u16 pmcsr;
745
746                 pci_read_config_word(dev, dev->pm_cap + PCI_PM_CTRL, &pmcsr);
747                 dev->current_state = (pmcsr & PCI_PM_CTRL_STATE_MASK);
748         } else {
749                 dev->current_state = state;
750         }
751 }
752
753 /**
754  * pci_power_up - Put the given device into D0 forcibly
755  * @dev: PCI device to power up
756  */
757 void pci_power_up(struct pci_dev *dev)
758 {
759         if (platform_pci_power_manageable(dev))
760                 platform_pci_set_power_state(dev, PCI_D0);
761
762         pci_raw_set_power_state(dev, PCI_D0);
763         pci_update_current_state(dev, PCI_D0);
764 }
765
766 /**
767  * pci_platform_power_transition - Use platform to change device power state
768  * @dev: PCI device to handle.
769  * @state: State to put the device into.
770  */
771 static int pci_platform_power_transition(struct pci_dev *dev, pci_power_t state)
772 {
773         int error;
774
775         if (platform_pci_power_manageable(dev)) {
776                 error = platform_pci_set_power_state(dev, state);
777                 if (!error)
778                         pci_update_current_state(dev, state);
779         } else
780                 error = -ENODEV;
781
782         if (error && !dev->pm_cap) /* Fall back to PCI_D0 */
783                 dev->current_state = PCI_D0;
784
785         return error;
786 }
787
788 /**
789  * pci_wakeup - Wake up a PCI device
790  * @pci_dev: Device to handle.
791  * @ign: ignored parameter
792  */
793 static int pci_wakeup(struct pci_dev *pci_dev, void *ign)
794 {
795         pci_wakeup_event(pci_dev);
796         pm_request_resume(&pci_dev->dev);
797         return 0;
798 }
799
800 /**
801  * pci_wakeup_bus - Walk given bus and wake up devices on it
802  * @bus: Top bus of the subtree to walk.
803  */
804 static void pci_wakeup_bus(struct pci_bus *bus)
805 {
806         if (bus)
807                 pci_walk_bus(bus, pci_wakeup, NULL);
808 }
809
810 /**
811  * __pci_start_power_transition - Start power transition of a PCI device
812  * @dev: PCI device to handle.
813  * @state: State to put the device into.
814  */
815 static void __pci_start_power_transition(struct pci_dev *dev, pci_power_t state)
816 {
817         if (state == PCI_D0) {
818                 pci_platform_power_transition(dev, PCI_D0);
819                 /*
820                  * Mandatory power management transition delays, see
821                  * PCI Express Base Specification Revision 2.0 Section
822                  * 6.6.1: Conventional Reset.  Do not delay for
823                  * devices powered on/off by corresponding bridge,
824                  * because have already delayed for the bridge.
825                  */
826                 if (dev->runtime_d3cold) {
827                         if (dev->d3cold_delay)
828                                 msleep(dev->d3cold_delay);
829                         /*
830                          * When powering on a bridge from D3cold, the
831                          * whole hierarchy may be powered on into
832                          * D0uninitialized state, resume them to give
833                          * them a chance to suspend again
834                          */
835                         pci_wakeup_bus(dev->subordinate);
836                 }
837         }
838 }
839
840 /**
841  * __pci_dev_set_current_state - Set current state of a PCI device
842  * @dev: Device to handle
843  * @data: pointer to state to be set
844  */
845 static int __pci_dev_set_current_state(struct pci_dev *dev, void *data)
846 {
847         pci_power_t state = *(pci_power_t *)data;
848
849         dev->current_state = state;
850         return 0;
851 }
852
853 /**
854  * __pci_bus_set_current_state - Walk given bus and set current state of devices
855  * @bus: Top bus of the subtree to walk.
856  * @state: state to be set
857  */
858 static void __pci_bus_set_current_state(struct pci_bus *bus, pci_power_t state)
859 {
860         if (bus)
861                 pci_walk_bus(bus, __pci_dev_set_current_state, &state);
862 }
863
864 /**
865  * __pci_complete_power_transition - Complete power transition of a PCI device
866  * @dev: PCI device to handle.
867  * @state: State to put the device into.
868  *
869  * This function should not be called directly by device drivers.
870  */
871 int __pci_complete_power_transition(struct pci_dev *dev, pci_power_t state)
872 {
873         int ret;
874
875         if (state <= PCI_D0)
876                 return -EINVAL;
877         ret = pci_platform_power_transition(dev, state);
878         /* Power off the bridge may power off the whole hierarchy */
879         if (!ret && state == PCI_D3cold)
880                 __pci_bus_set_current_state(dev->subordinate, PCI_D3cold);
881         return ret;
882 }
883 EXPORT_SYMBOL_GPL(__pci_complete_power_transition);
884
885 /**
886  * pci_set_power_state - Set the power state of a PCI device
887  * @dev: PCI device to handle.
888  * @state: PCI power state (D0, D1, D2, D3hot) to put the device into.
889  *
890  * Transition a device to a new power state, using the platform firmware and/or
891  * the device's PCI PM registers.
892  *
893  * RETURN VALUE:
894  * -EINVAL if the requested state is invalid.
895  * -EIO if device does not support PCI PM or its PM capabilities register has a
896  * wrong version, or device doesn't support the requested state.
897  * 0 if the transition is to D1 or D2 but D1 and D2 are not supported.
898  * 0 if device already is in the requested state.
899  * 0 if the transition is to D3 but D3 is not supported.
900  * 0 if device's power state has been successfully changed.
901  */
902 int pci_set_power_state(struct pci_dev *dev, pci_power_t state)
903 {
904         int error;
905
906         /* bound the state we're entering */
907         if (state > PCI_D3cold)
908                 state = PCI_D3cold;
909         else if (state < PCI_D0)
910                 state = PCI_D0;
911         else if ((state == PCI_D1 || state == PCI_D2) && pci_no_d1d2(dev))
912                 /*
913                  * If the device or the parent bridge do not support PCI PM,
914                  * ignore the request if we're doing anything other than putting
915                  * it into D0 (which would only happen on boot).
916                  */
917                 return 0;
918
919         /* Check if we're already there */
920         if (dev->current_state == state)
921                 return 0;
922
923         __pci_start_power_transition(dev, state);
924
925         /* This device is quirked not to be put into D3, so
926            don't put it in D3 */
927         if (state >= PCI_D3hot && (dev->dev_flags & PCI_DEV_FLAGS_NO_D3))
928                 return 0;
929
930         /*
931          * To put device in D3cold, we put device into D3hot in native
932          * way, then put device into D3cold with platform ops
933          */
934         error = pci_raw_set_power_state(dev, state > PCI_D3hot ?
935                                         PCI_D3hot : state);
936
937         if (!__pci_complete_power_transition(dev, state))
938                 error = 0;
939
940         return error;
941 }
942 EXPORT_SYMBOL(pci_set_power_state);
943
944 /**
945  * pci_choose_state - Choose the power state of a PCI device
946  * @dev: PCI device to be suspended
947  * @state: target sleep state for the whole system. This is the value
948  *      that is passed to suspend() function.
949  *
950  * Returns PCI power state suitable for given device and given system
951  * message.
952  */
953
954 pci_power_t pci_choose_state(struct pci_dev *dev, pm_message_t state)
955 {
956         pci_power_t ret;
957
958         if (!dev->pm_cap)
959                 return PCI_D0;
960
961         ret = platform_pci_choose_state(dev);
962         if (ret != PCI_POWER_ERROR)
963                 return ret;
964
965         switch (state.event) {
966         case PM_EVENT_ON:
967                 return PCI_D0;
968         case PM_EVENT_FREEZE:
969         case PM_EVENT_PRETHAW:
970                 /* REVISIT both freeze and pre-thaw "should" use D0 */
971         case PM_EVENT_SUSPEND:
972         case PM_EVENT_HIBERNATE:
973                 return PCI_D3hot;
974         default:
975                 pci_info(dev, "unrecognized suspend event %d\n",
976                          state.event);
977                 BUG();
978         }
979         return PCI_D0;
980 }
981 EXPORT_SYMBOL(pci_choose_state);
982
983 #define PCI_EXP_SAVE_REGS       7
984
985 static struct pci_cap_saved_state *_pci_find_saved_cap(struct pci_dev *pci_dev,
986                                                        u16 cap, bool extended)
987 {
988         struct pci_cap_saved_state *tmp;
989
990         hlist_for_each_entry(tmp, &pci_dev->saved_cap_space, next) {
991                 if (tmp->cap.cap_extended == extended && tmp->cap.cap_nr == cap)
992                         return tmp;
993         }
994         return NULL;
995 }
996
997 struct pci_cap_saved_state *pci_find_saved_cap(struct pci_dev *dev, char cap)
998 {
999         return _pci_find_saved_cap(dev, cap, false);
1000 }
1001
1002 struct pci_cap_saved_state *pci_find_saved_ext_cap(struct pci_dev *dev, u16 cap)
1003 {
1004         return _pci_find_saved_cap(dev, cap, true);
1005 }
1006
1007 static int pci_save_pcie_state(struct pci_dev *dev)
1008 {
1009         int i = 0;
1010         struct pci_cap_saved_state *save_state;
1011         u16 *cap;
1012
1013         if (!pci_is_pcie(dev))
1014                 return 0;
1015
1016         save_state = pci_find_saved_cap(dev, PCI_CAP_ID_EXP);
1017         if (!save_state) {
1018                 pci_err(dev, "buffer not found in %s\n", __func__);
1019                 return -ENOMEM;
1020         }
1021
1022         cap = (u16 *)&save_state->cap.data[0];
1023         pcie_capability_read_word(dev, PCI_EXP_DEVCTL, &cap[i++]);
1024         pcie_capability_read_word(dev, PCI_EXP_LNKCTL, &cap[i++]);
1025         pcie_capability_read_word(dev, PCI_EXP_SLTCTL, &cap[i++]);
1026         pcie_capability_read_word(dev, PCI_EXP_RTCTL,  &cap[i++]);
1027         pcie_capability_read_word(dev, PCI_EXP_DEVCTL2, &cap[i++]);
1028         pcie_capability_read_word(dev, PCI_EXP_LNKCTL2, &cap[i++]);
1029         pcie_capability_read_word(dev, PCI_EXP_SLTCTL2, &cap[i++]);
1030
1031         return 0;
1032 }
1033
1034 static void pci_restore_pcie_state(struct pci_dev *dev)
1035 {
1036         int i = 0;
1037         struct pci_cap_saved_state *save_state;
1038         u16 *cap;
1039
1040         save_state = pci_find_saved_cap(dev, PCI_CAP_ID_EXP);
1041         if (!save_state)
1042                 return;
1043
1044         cap = (u16 *)&save_state->cap.data[0];
1045         pcie_capability_write_word(dev, PCI_EXP_DEVCTL, cap[i++]);
1046         pcie_capability_write_word(dev, PCI_EXP_LNKCTL, cap[i++]);
1047         pcie_capability_write_word(dev, PCI_EXP_SLTCTL, cap[i++]);
1048         pcie_capability_write_word(dev, PCI_EXP_RTCTL, cap[i++]);
1049         pcie_capability_write_word(dev, PCI_EXP_DEVCTL2, cap[i++]);
1050         pcie_capability_write_word(dev, PCI_EXP_LNKCTL2, cap[i++]);
1051         pcie_capability_write_word(dev, PCI_EXP_SLTCTL2, cap[i++]);
1052 }
1053
1054
1055 static int pci_save_pcix_state(struct pci_dev *dev)
1056 {
1057         int pos;
1058         struct pci_cap_saved_state *save_state;
1059
1060         pos = pci_find_capability(dev, PCI_CAP_ID_PCIX);
1061         if (!pos)
1062                 return 0;
1063
1064         save_state = pci_find_saved_cap(dev, PCI_CAP_ID_PCIX);
1065         if (!save_state) {
1066                 pci_err(dev, "buffer not found in %s\n", __func__);
1067                 return -ENOMEM;
1068         }
1069
1070         pci_read_config_word(dev, pos + PCI_X_CMD,
1071                              (u16 *)save_state->cap.data);
1072
1073         return 0;
1074 }
1075
1076 static void pci_restore_pcix_state(struct pci_dev *dev)
1077 {
1078         int i = 0, pos;
1079         struct pci_cap_saved_state *save_state;
1080         u16 *cap;
1081
1082         save_state = pci_find_saved_cap(dev, PCI_CAP_ID_PCIX);
1083         pos = pci_find_capability(dev, PCI_CAP_ID_PCIX);
1084         if (!save_state || !pos)
1085                 return;
1086         cap = (u16 *)&save_state->cap.data[0];
1087
1088         pci_write_config_word(dev, pos + PCI_X_CMD, cap[i++]);
1089 }
1090
1091
1092 /**
1093  * pci_save_state - save the PCI configuration space of a device before suspending
1094  * @dev: - PCI device that we're dealing with
1095  */
1096 int pci_save_state(struct pci_dev *dev)
1097 {
1098         int i;
1099         /* XXX: 100% dword access ok here? */
1100         for (i = 0; i < 16; i++)
1101                 pci_read_config_dword(dev, i * 4, &dev->saved_config_space[i]);
1102         dev->state_saved = true;
1103
1104         i = pci_save_pcie_state(dev);
1105         if (i != 0)
1106                 return i;
1107
1108         i = pci_save_pcix_state(dev);
1109         if (i != 0)
1110                 return i;
1111
1112         return pci_save_vc_state(dev);
1113 }
1114 EXPORT_SYMBOL(pci_save_state);
1115
1116 static void pci_restore_config_dword(struct pci_dev *pdev, int offset,
1117                                      u32 saved_val, int retry)
1118 {
1119         u32 val;
1120
1121         pci_read_config_dword(pdev, offset, &val);
1122         if (val == saved_val)
1123                 return;
1124
1125         for (;;) {
1126                 pci_dbg(pdev, "restoring config space at offset %#x (was %#x, writing %#x)\n",
1127                         offset, val, saved_val);
1128                 pci_write_config_dword(pdev, offset, saved_val);
1129                 if (retry-- <= 0)
1130                         return;
1131
1132                 pci_read_config_dword(pdev, offset, &val);
1133                 if (val == saved_val)
1134                         return;
1135
1136                 mdelay(1);
1137         }
1138 }
1139
1140 static void pci_restore_config_space_range(struct pci_dev *pdev,
1141                                            int start, int end, int retry)
1142 {
1143         int index;
1144
1145         for (index = end; index >= start; index--)
1146                 pci_restore_config_dword(pdev, 4 * index,
1147                                          pdev->saved_config_space[index],
1148                                          retry);
1149 }
1150
1151 static void pci_restore_config_space(struct pci_dev *pdev)
1152 {
1153         if (pdev->hdr_type == PCI_HEADER_TYPE_NORMAL) {
1154                 pci_restore_config_space_range(pdev, 10, 15, 0);
1155                 /* Restore BARs before the command register. */
1156                 pci_restore_config_space_range(pdev, 4, 9, 10);
1157                 pci_restore_config_space_range(pdev, 0, 3, 0);
1158         } else {
1159                 pci_restore_config_space_range(pdev, 0, 15, 0);
1160         }
1161 }
1162
1163 /**
1164  * pci_restore_state - Restore the saved state of a PCI device
1165  * @dev: - PCI device that we're dealing with
1166  */
1167 void pci_restore_state(struct pci_dev *dev)
1168 {
1169         if (!dev->state_saved)
1170                 return;
1171
1172         /* PCI Express register must be restored first */
1173         pci_restore_pcie_state(dev);
1174         pci_restore_pasid_state(dev);
1175         pci_restore_pri_state(dev);
1176         pci_restore_ats_state(dev);
1177         pci_restore_vc_state(dev);
1178
1179         pci_cleanup_aer_error_status_regs(dev);
1180
1181         pci_restore_config_space(dev);
1182
1183         pci_restore_pcix_state(dev);
1184         pci_restore_msi_state(dev);
1185
1186         /* Restore ACS and IOV configuration state */
1187         pci_enable_acs(dev);
1188         pci_restore_iov_state(dev);
1189
1190         dev->state_saved = false;
1191 }
1192 EXPORT_SYMBOL(pci_restore_state);
1193
1194 struct pci_saved_state {
1195         u32 config_space[16];
1196         struct pci_cap_saved_data cap[0];
1197 };
1198
1199 /**
1200  * pci_store_saved_state - Allocate and return an opaque struct containing
1201  *                         the device saved state.
1202  * @dev: PCI device that we're dealing with
1203  *
1204  * Return NULL if no state or error.
1205  */
1206 struct pci_saved_state *pci_store_saved_state(struct pci_dev *dev)
1207 {
1208         struct pci_saved_state *state;
1209         struct pci_cap_saved_state *tmp;
1210         struct pci_cap_saved_data *cap;
1211         size_t size;
1212
1213         if (!dev->state_saved)
1214                 return NULL;
1215
1216         size = sizeof(*state) + sizeof(struct pci_cap_saved_data);
1217
1218         hlist_for_each_entry(tmp, &dev->saved_cap_space, next)
1219                 size += sizeof(struct pci_cap_saved_data) + tmp->cap.size;
1220
1221         state = kzalloc(size, GFP_KERNEL);
1222         if (!state)
1223                 return NULL;
1224
1225         memcpy(state->config_space, dev->saved_config_space,
1226                sizeof(state->config_space));
1227
1228         cap = state->cap;
1229         hlist_for_each_entry(tmp, &dev->saved_cap_space, next) {
1230                 size_t len = sizeof(struct pci_cap_saved_data) + tmp->cap.size;
1231                 memcpy(cap, &tmp->cap, len);
1232                 cap = (struct pci_cap_saved_data *)((u8 *)cap + len);
1233         }
1234         /* Empty cap_save terminates list */
1235
1236         return state;
1237 }
1238 EXPORT_SYMBOL_GPL(pci_store_saved_state);
1239
1240 /**
1241  * pci_load_saved_state - Reload the provided save state into struct pci_dev.
1242  * @dev: PCI device that we're dealing with
1243  * @state: Saved state returned from pci_store_saved_state()
1244  */
1245 int pci_load_saved_state(struct pci_dev *dev,
1246                          struct pci_saved_state *state)
1247 {
1248         struct pci_cap_saved_data *cap;
1249
1250         dev->state_saved = false;
1251
1252         if (!state)
1253                 return 0;
1254
1255         memcpy(dev->saved_config_space, state->config_space,
1256                sizeof(state->config_space));
1257
1258         cap = state->cap;
1259         while (cap->size) {
1260                 struct pci_cap_saved_state *tmp;
1261
1262                 tmp = _pci_find_saved_cap(dev, cap->cap_nr, cap->cap_extended);
1263                 if (!tmp || tmp->cap.size != cap->size)
1264                         return -EINVAL;
1265
1266                 memcpy(tmp->cap.data, cap->data, tmp->cap.size);
1267                 cap = (struct pci_cap_saved_data *)((u8 *)cap +
1268                        sizeof(struct pci_cap_saved_data) + cap->size);
1269         }
1270
1271         dev->state_saved = true;
1272         return 0;
1273 }
1274 EXPORT_SYMBOL_GPL(pci_load_saved_state);
1275
1276 /**
1277  * pci_load_and_free_saved_state - Reload the save state pointed to by state,
1278  *                                 and free the memory allocated for it.
1279  * @dev: PCI device that we're dealing with
1280  * @state: Pointer to saved state returned from pci_store_saved_state()
1281  */
1282 int pci_load_and_free_saved_state(struct pci_dev *dev,
1283                                   struct pci_saved_state **state)
1284 {
1285         int ret = pci_load_saved_state(dev, *state);
1286         kfree(*state);
1287         *state = NULL;
1288         return ret;
1289 }
1290 EXPORT_SYMBOL_GPL(pci_load_and_free_saved_state);
1291
1292 int __weak pcibios_enable_device(struct pci_dev *dev, int bars)
1293 {
1294         return pci_enable_resources(dev, bars);
1295 }
1296
1297 static int do_pci_enable_device(struct pci_dev *dev, int bars)
1298 {
1299         int err;
1300         struct pci_dev *bridge;
1301         u16 cmd;
1302         u8 pin;
1303
1304         err = pci_set_power_state(dev, PCI_D0);
1305         if (err < 0 && err != -EIO)
1306                 return err;
1307
1308         bridge = pci_upstream_bridge(dev);
1309         if (bridge)
1310                 pcie_aspm_powersave_config_link(bridge);
1311
1312         err = pcibios_enable_device(dev, bars);
1313         if (err < 0)
1314                 return err;
1315         pci_fixup_device(pci_fixup_enable, dev);
1316
1317         if (dev->msi_enabled || dev->msix_enabled)
1318                 return 0;
1319
1320         pci_read_config_byte(dev, PCI_INTERRUPT_PIN, &pin);
1321         if (pin) {
1322                 pci_read_config_word(dev, PCI_COMMAND, &cmd);
1323                 if (cmd & PCI_COMMAND_INTX_DISABLE)
1324                         pci_write_config_word(dev, PCI_COMMAND,
1325                                               cmd & ~PCI_COMMAND_INTX_DISABLE);
1326         }
1327
1328         return 0;
1329 }
1330
1331 /**
1332  * pci_reenable_device - Resume abandoned device
1333  * @dev: PCI device to be resumed
1334  *
1335  *  Note this function is a backend of pci_default_resume and is not supposed
1336  *  to be called by normal code, write proper resume handler and use it instead.
1337  */
1338 int pci_reenable_device(struct pci_dev *dev)
1339 {
1340         if (pci_is_enabled(dev))
1341                 return do_pci_enable_device(dev, (1 << PCI_NUM_RESOURCES) - 1);
1342         return 0;
1343 }
1344 EXPORT_SYMBOL(pci_reenable_device);
1345
1346 static void pci_enable_bridge(struct pci_dev *dev)
1347 {
1348         struct pci_dev *bridge;
1349         int retval;
1350
1351         bridge = pci_upstream_bridge(dev);
1352         if (bridge)
1353                 pci_enable_bridge(bridge);
1354
1355         if (pci_is_enabled(dev)) {
1356                 if (!dev->is_busmaster)
1357                         pci_set_master(dev);
1358                 return;
1359         }
1360
1361         retval = pci_enable_device(dev);
1362         if (retval)
1363                 pci_err(dev, "Error enabling bridge (%d), continuing\n",
1364                         retval);
1365         pci_set_master(dev);
1366 }
1367
1368 static int pci_enable_device_flags(struct pci_dev *dev, unsigned long flags)
1369 {
1370         struct pci_dev *bridge;
1371         int err;
1372         int i, bars = 0;
1373
1374         /*
1375          * Power state could be unknown at this point, either due to a fresh
1376          * boot or a device removal call.  So get the current power state
1377          * so that things like MSI message writing will behave as expected
1378          * (e.g. if the device really is in D0 at enable time).
1379          */
1380         if (dev->pm_cap) {
1381                 u16 pmcsr;
1382                 pci_read_config_word(dev, dev->pm_cap + PCI_PM_CTRL, &pmcsr);
1383                 dev->current_state = (pmcsr & PCI_PM_CTRL_STATE_MASK);
1384         }
1385
1386         if (atomic_inc_return(&dev->enable_cnt) > 1)
1387                 return 0;               /* already enabled */
1388
1389         bridge = pci_upstream_bridge(dev);
1390         if (bridge)
1391                 pci_enable_bridge(bridge);
1392
1393         /* only skip sriov related */
1394         for (i = 0; i <= PCI_ROM_RESOURCE; i++)
1395                 if (dev->resource[i].flags & flags)
1396                         bars |= (1 << i);
1397         for (i = PCI_BRIDGE_RESOURCES; i < DEVICE_COUNT_RESOURCE; i++)
1398                 if (dev->resource[i].flags & flags)
1399                         bars |= (1 << i);
1400
1401         err = do_pci_enable_device(dev, bars);
1402         if (err < 0)
1403                 atomic_dec(&dev->enable_cnt);
1404         return err;
1405 }
1406
1407 /**
1408  * pci_enable_device_io - Initialize a device for use with IO space
1409  * @dev: PCI device to be initialized
1410  *
1411  *  Initialize device before it's used by a driver. Ask low-level code
1412  *  to enable I/O resources. Wake up the device if it was suspended.
1413  *  Beware, this function can fail.
1414  */
1415 int pci_enable_device_io(struct pci_dev *dev)
1416 {
1417         return pci_enable_device_flags(dev, IORESOURCE_IO);
1418 }
1419 EXPORT_SYMBOL(pci_enable_device_io);
1420
1421 /**
1422  * pci_enable_device_mem - Initialize a device for use with Memory space
1423  * @dev: PCI device to be initialized
1424  *
1425  *  Initialize device before it's used by a driver. Ask low-level code
1426  *  to enable Memory resources. Wake up the device if it was suspended.
1427  *  Beware, this function can fail.
1428  */
1429 int pci_enable_device_mem(struct pci_dev *dev)
1430 {
1431         return pci_enable_device_flags(dev, IORESOURCE_MEM);
1432 }
1433 EXPORT_SYMBOL(pci_enable_device_mem);
1434
1435 /**
1436  * pci_enable_device - Initialize device before it's used by a driver.
1437  * @dev: PCI device to be initialized
1438  *
1439  *  Initialize device before it's used by a driver. Ask low-level code
1440  *  to enable I/O and memory. Wake up the device if it was suspended.
1441  *  Beware, this function can fail.
1442  *
1443  *  Note we don't actually enable the device many times if we call
1444  *  this function repeatedly (we just increment the count).
1445  */
1446 int pci_enable_device(struct pci_dev *dev)
1447 {
1448         return pci_enable_device_flags(dev, IORESOURCE_MEM | IORESOURCE_IO);
1449 }
1450 EXPORT_SYMBOL(pci_enable_device);
1451
1452 /*
1453  * Managed PCI resources.  This manages device on/off, intx/msi/msix
1454  * on/off and BAR regions.  pci_dev itself records msi/msix status, so
1455  * there's no need to track it separately.  pci_devres is initialized
1456  * when a device is enabled using managed PCI device enable interface.
1457  */
1458 struct pci_devres {
1459         unsigned int enabled:1;
1460         unsigned int pinned:1;
1461         unsigned int orig_intx:1;
1462         unsigned int restore_intx:1;
1463         unsigned int mwi:1;
1464         u32 region_mask;
1465 };
1466
1467 static void pcim_release(struct device *gendev, void *res)
1468 {
1469         struct pci_dev *dev = to_pci_dev(gendev);
1470         struct pci_devres *this = res;
1471         int i;
1472
1473         if (dev->msi_enabled)
1474                 pci_disable_msi(dev);
1475         if (dev->msix_enabled)
1476                 pci_disable_msix(dev);
1477
1478         for (i = 0; i < DEVICE_COUNT_RESOURCE; i++)
1479                 if (this->region_mask & (1 << i))
1480                         pci_release_region(dev, i);
1481
1482         if (this->mwi)
1483                 pci_clear_mwi(dev);
1484
1485         if (this->restore_intx)
1486                 pci_intx(dev, this->orig_intx);
1487
1488         if (this->enabled && !this->pinned)
1489                 pci_disable_device(dev);
1490 }
1491
1492 static struct pci_devres *get_pci_dr(struct pci_dev *pdev)
1493 {
1494         struct pci_devres *dr, *new_dr;
1495
1496         dr = devres_find(&pdev->dev, pcim_release, NULL, NULL);
1497         if (dr)
1498                 return dr;
1499
1500         new_dr = devres_alloc(pcim_release, sizeof(*new_dr), GFP_KERNEL);
1501         if (!new_dr)
1502                 return NULL;
1503         return devres_get(&pdev->dev, new_dr, NULL, NULL);
1504 }
1505
1506 static struct pci_devres *find_pci_dr(struct pci_dev *pdev)
1507 {
1508         if (pci_is_managed(pdev))
1509                 return devres_find(&pdev->dev, pcim_release, NULL, NULL);
1510         return NULL;
1511 }
1512
1513 /**
1514  * pcim_enable_device - Managed pci_enable_device()
1515  * @pdev: PCI device to be initialized
1516  *
1517  * Managed pci_enable_device().
1518  */
1519 int pcim_enable_device(struct pci_dev *pdev)
1520 {
1521         struct pci_devres *dr;
1522         int rc;
1523
1524         dr = get_pci_dr(pdev);
1525         if (unlikely(!dr))
1526                 return -ENOMEM;
1527         if (dr->enabled)
1528                 return 0;
1529
1530         rc = pci_enable_device(pdev);
1531         if (!rc) {
1532                 pdev->is_managed = 1;
1533                 dr->enabled = 1;
1534         }
1535         return rc;
1536 }
1537 EXPORT_SYMBOL(pcim_enable_device);
1538
1539 /**
1540  * pcim_pin_device - Pin managed PCI device
1541  * @pdev: PCI device to pin
1542  *
1543  * Pin managed PCI device @pdev.  Pinned device won't be disabled on
1544  * driver detach.  @pdev must have been enabled with
1545  * pcim_enable_device().
1546  */
1547 void pcim_pin_device(struct pci_dev *pdev)
1548 {
1549         struct pci_devres *dr;
1550
1551         dr = find_pci_dr(pdev);
1552         WARN_ON(!dr || !dr->enabled);
1553         if (dr)
1554                 dr->pinned = 1;
1555 }
1556 EXPORT_SYMBOL(pcim_pin_device);
1557
1558 /*
1559  * pcibios_add_device - provide arch specific hooks when adding device dev
1560  * @dev: the PCI device being added
1561  *
1562  * Permits the platform to provide architecture specific functionality when
1563  * devices are added. This is the default implementation. Architecture
1564  * implementations can override this.
1565  */
1566 int __weak pcibios_add_device(struct pci_dev *dev)
1567 {
1568         return 0;
1569 }
1570
1571 /**
1572  * pcibios_release_device - provide arch specific hooks when releasing device dev
1573  * @dev: the PCI device being released
1574  *
1575  * Permits the platform to provide architecture specific functionality when
1576  * devices are released. This is the default implementation. Architecture
1577  * implementations can override this.
1578  */
1579 void __weak pcibios_release_device(struct pci_dev *dev) {}
1580
1581 /**
1582  * pcibios_disable_device - disable arch specific PCI resources for device dev
1583  * @dev: the PCI device to disable
1584  *
1585  * Disables architecture specific PCI resources for the device. This
1586  * is the default implementation. Architecture implementations can
1587  * override this.
1588  */
1589 void __weak pcibios_disable_device(struct pci_dev *dev) {}
1590
1591 /**
1592  * pcibios_penalize_isa_irq - penalize an ISA IRQ
1593  * @irq: ISA IRQ to penalize
1594  * @active: IRQ active or not
1595  *
1596  * Permits the platform to provide architecture-specific functionality when
1597  * penalizing ISA IRQs. This is the default implementation. Architecture
1598  * implementations can override this.
1599  */
1600 void __weak pcibios_penalize_isa_irq(int irq, int active) {}
1601
1602 static void do_pci_disable_device(struct pci_dev *dev)
1603 {
1604         u16 pci_command;
1605
1606         pci_read_config_word(dev, PCI_COMMAND, &pci_command);
1607         if (pci_command & PCI_COMMAND_MASTER) {
1608                 pci_command &= ~PCI_COMMAND_MASTER;
1609                 pci_write_config_word(dev, PCI_COMMAND, pci_command);
1610         }
1611
1612         pcibios_disable_device(dev);
1613 }
1614
1615 /**
1616  * pci_disable_enabled_device - Disable device without updating enable_cnt
1617  * @dev: PCI device to disable
1618  *
1619  * NOTE: This function is a backend of PCI power management routines and is
1620  * not supposed to be called drivers.
1621  */
1622 void pci_disable_enabled_device(struct pci_dev *dev)
1623 {
1624         if (pci_is_enabled(dev))
1625                 do_pci_disable_device(dev);
1626 }
1627
1628 /**
1629  * pci_disable_device - Disable PCI device after use
1630  * @dev: PCI device to be disabled
1631  *
1632  * Signal to the system that the PCI device is not in use by the system
1633  * anymore.  This only involves disabling PCI bus-mastering, if active.
1634  *
1635  * Note we don't actually disable the device until all callers of
1636  * pci_enable_device() have called pci_disable_device().
1637  */
1638 void pci_disable_device(struct pci_dev *dev)
1639 {
1640         struct pci_devres *dr;
1641
1642         dr = find_pci_dr(dev);
1643         if (dr)
1644                 dr->enabled = 0;
1645
1646         dev_WARN_ONCE(&dev->dev, atomic_read(&dev->enable_cnt) <= 0,
1647                       "disabling already-disabled device");
1648
1649         if (atomic_dec_return(&dev->enable_cnt) != 0)
1650                 return;
1651
1652         do_pci_disable_device(dev);
1653
1654         dev->is_busmaster = 0;
1655 }
1656 EXPORT_SYMBOL(pci_disable_device);
1657
1658 /**
1659  * pcibios_set_pcie_reset_state - set reset state for device dev
1660  * @dev: the PCIe device reset
1661  * @state: Reset state to enter into
1662  *
1663  *
1664  * Sets the PCIe reset state for the device. This is the default
1665  * implementation. Architecture implementations can override this.
1666  */
1667 int __weak pcibios_set_pcie_reset_state(struct pci_dev *dev,
1668                                         enum pcie_reset_state state)
1669 {
1670         return -EINVAL;
1671 }
1672
1673 /**
1674  * pci_set_pcie_reset_state - set reset state for device dev
1675  * @dev: the PCIe device reset
1676  * @state: Reset state to enter into
1677  *
1678  *
1679  * Sets the PCI reset state for the device.
1680  */
1681 int pci_set_pcie_reset_state(struct pci_dev *dev, enum pcie_reset_state state)
1682 {
1683         return pcibios_set_pcie_reset_state(dev, state);
1684 }
1685 EXPORT_SYMBOL_GPL(pci_set_pcie_reset_state);
1686
1687 /**
1688  * pcie_clear_root_pme_status - Clear root port PME interrupt status.
1689  * @dev: PCIe root port or event collector.
1690  */
1691 void pcie_clear_root_pme_status(struct pci_dev *dev)
1692 {
1693         pcie_capability_set_dword(dev, PCI_EXP_RTSTA, PCI_EXP_RTSTA_PME);
1694 }
1695
1696 /**
1697  * pci_check_pme_status - Check if given device has generated PME.
1698  * @dev: Device to check.
1699  *
1700  * Check the PME status of the device and if set, clear it and clear PME enable
1701  * (if set).  Return 'true' if PME status and PME enable were both set or
1702  * 'false' otherwise.
1703  */
1704 bool pci_check_pme_status(struct pci_dev *dev)
1705 {
1706         int pmcsr_pos;
1707         u16 pmcsr;
1708         bool ret = false;
1709
1710         if (!dev->pm_cap)
1711                 return false;
1712
1713         pmcsr_pos = dev->pm_cap + PCI_PM_CTRL;
1714         pci_read_config_word(dev, pmcsr_pos, &pmcsr);
1715         if (!(pmcsr & PCI_PM_CTRL_PME_STATUS))
1716                 return false;
1717
1718         /* Clear PME status. */
1719         pmcsr |= PCI_PM_CTRL_PME_STATUS;
1720         if (pmcsr & PCI_PM_CTRL_PME_ENABLE) {
1721                 /* Disable PME to avoid interrupt flood. */
1722                 pmcsr &= ~PCI_PM_CTRL_PME_ENABLE;
1723                 ret = true;
1724         }
1725
1726         pci_write_config_word(dev, pmcsr_pos, pmcsr);
1727
1728         return ret;
1729 }
1730
1731 /**
1732  * pci_pme_wakeup - Wake up a PCI device if its PME Status bit is set.
1733  * @dev: Device to handle.
1734  * @pme_poll_reset: Whether or not to reset the device's pme_poll flag.
1735  *
1736  * Check if @dev has generated PME and queue a resume request for it in that
1737  * case.
1738  */
1739 static int pci_pme_wakeup(struct pci_dev *dev, void *pme_poll_reset)
1740 {
1741         if (pme_poll_reset && dev->pme_poll)
1742                 dev->pme_poll = false;
1743
1744         if (pci_check_pme_status(dev)) {
1745                 pci_wakeup_event(dev);
1746                 pm_request_resume(&dev->dev);
1747         }
1748         return 0;
1749 }
1750
1751 /**
1752  * pci_pme_wakeup_bus - Walk given bus and wake up devices on it, if necessary.
1753  * @bus: Top bus of the subtree to walk.
1754  */
1755 void pci_pme_wakeup_bus(struct pci_bus *bus)
1756 {
1757         if (bus)
1758                 pci_walk_bus(bus, pci_pme_wakeup, (void *)true);
1759 }
1760
1761
1762 /**
1763  * pci_pme_capable - check the capability of PCI device to generate PME#
1764  * @dev: PCI device to handle.
1765  * @state: PCI state from which device will issue PME#.
1766  */
1767 bool pci_pme_capable(struct pci_dev *dev, pci_power_t state)
1768 {
1769         if (!dev->pm_cap)
1770                 return false;
1771
1772         return !!(dev->pme_support & (1 << state));
1773 }
1774 EXPORT_SYMBOL(pci_pme_capable);
1775
1776 static void pci_pme_list_scan(struct work_struct *work)
1777 {
1778         struct pci_pme_device *pme_dev, *n;
1779
1780         mutex_lock(&pci_pme_list_mutex);
1781         list_for_each_entry_safe(pme_dev, n, &pci_pme_list, list) {
1782                 if (pme_dev->dev->pme_poll) {
1783                         struct pci_dev *bridge;
1784
1785                         bridge = pme_dev->dev->bus->self;
1786                         /*
1787                          * If bridge is in low power state, the
1788                          * configuration space of subordinate devices
1789                          * may be not accessible
1790                          */
1791                         if (bridge && bridge->current_state != PCI_D0)
1792                                 continue;
1793                         pci_pme_wakeup(pme_dev->dev, NULL);
1794                 } else {
1795                         list_del(&pme_dev->list);
1796                         kfree(pme_dev);
1797                 }
1798         }
1799         if (!list_empty(&pci_pme_list))
1800                 queue_delayed_work(system_freezable_wq, &pci_pme_work,
1801                                    msecs_to_jiffies(PME_TIMEOUT));
1802         mutex_unlock(&pci_pme_list_mutex);
1803 }
1804
1805 static void __pci_pme_active(struct pci_dev *dev, bool enable)
1806 {
1807         u16 pmcsr;
1808
1809         if (!dev->pme_support)
1810                 return;
1811
1812         pci_read_config_word(dev, dev->pm_cap + PCI_PM_CTRL, &pmcsr);
1813         /* Clear PME_Status by writing 1 to it and enable PME# */
1814         pmcsr |= PCI_PM_CTRL_PME_STATUS | PCI_PM_CTRL_PME_ENABLE;
1815         if (!enable)
1816                 pmcsr &= ~PCI_PM_CTRL_PME_ENABLE;
1817
1818         pci_write_config_word(dev, dev->pm_cap + PCI_PM_CTRL, pmcsr);
1819 }
1820
1821 /**
1822  * pci_pme_restore - Restore PME configuration after config space restore.
1823  * @dev: PCI device to update.
1824  */
1825 void pci_pme_restore(struct pci_dev *dev)
1826 {
1827         u16 pmcsr;
1828
1829         if (!dev->pme_support)
1830                 return;
1831
1832         pci_read_config_word(dev, dev->pm_cap + PCI_PM_CTRL, &pmcsr);
1833         if (dev->wakeup_prepared) {
1834                 pmcsr |= PCI_PM_CTRL_PME_ENABLE;
1835                 pmcsr &= ~PCI_PM_CTRL_PME_STATUS;
1836         } else {
1837                 pmcsr &= ~PCI_PM_CTRL_PME_ENABLE;
1838                 pmcsr |= PCI_PM_CTRL_PME_STATUS;
1839         }
1840         pci_write_config_word(dev, dev->pm_cap + PCI_PM_CTRL, pmcsr);
1841 }
1842
1843 /**
1844  * pci_pme_active - enable or disable PCI device's PME# function
1845  * @dev: PCI device to handle.
1846  * @enable: 'true' to enable PME# generation; 'false' to disable it.
1847  *
1848  * The caller must verify that the device is capable of generating PME# before
1849  * calling this function with @enable equal to 'true'.
1850  */
1851 void pci_pme_active(struct pci_dev *dev, bool enable)
1852 {
1853         __pci_pme_active(dev, enable);
1854
1855         /*
1856          * PCI (as opposed to PCIe) PME requires that the device have
1857          * its PME# line hooked up correctly. Not all hardware vendors
1858          * do this, so the PME never gets delivered and the device
1859          * remains asleep. The easiest way around this is to
1860          * periodically walk the list of suspended devices and check
1861          * whether any have their PME flag set. The assumption is that
1862          * we'll wake up often enough anyway that this won't be a huge
1863          * hit, and the power savings from the devices will still be a
1864          * win.
1865          *
1866          * Although PCIe uses in-band PME message instead of PME# line
1867          * to report PME, PME does not work for some PCIe devices in
1868          * reality.  For example, there are devices that set their PME
1869          * status bits, but don't really bother to send a PME message;
1870          * there are PCI Express Root Ports that don't bother to
1871          * trigger interrupts when they receive PME messages from the
1872          * devices below.  So PME poll is used for PCIe devices too.
1873          */
1874
1875         if (dev->pme_poll) {
1876                 struct pci_pme_device *pme_dev;
1877                 if (enable) {
1878                         pme_dev = kmalloc(sizeof(struct pci_pme_device),
1879                                           GFP_KERNEL);
1880                         if (!pme_dev) {
1881                                 pci_warn(dev, "can't enable PME#\n");
1882                                 return;
1883                         }
1884                         pme_dev->dev = dev;
1885                         mutex_lock(&pci_pme_list_mutex);
1886                         list_add(&pme_dev->list, &pci_pme_list);
1887                         if (list_is_singular(&pci_pme_list))
1888                                 queue_delayed_work(system_freezable_wq,
1889                                                    &pci_pme_work,
1890                                                    msecs_to_jiffies(PME_TIMEOUT));
1891                         mutex_unlock(&pci_pme_list_mutex);
1892                 } else {
1893                         mutex_lock(&pci_pme_list_mutex);
1894                         list_for_each_entry(pme_dev, &pci_pme_list, list) {
1895                                 if (pme_dev->dev == dev) {
1896                                         list_del(&pme_dev->list);
1897                                         kfree(pme_dev);
1898                                         break;
1899                                 }
1900                         }
1901                         mutex_unlock(&pci_pme_list_mutex);
1902                 }
1903         }
1904
1905         pci_dbg(dev, "PME# %s\n", enable ? "enabled" : "disabled");
1906 }
1907 EXPORT_SYMBOL(pci_pme_active);
1908
1909 /**
1910  * pci_enable_wake - enable PCI device as wakeup event source
1911  * @dev: PCI device affected
1912  * @state: PCI state from which device will issue wakeup events
1913  * @enable: True to enable event generation; false to disable
1914  *
1915  * This enables the device as a wakeup event source, or disables it.
1916  * When such events involves platform-specific hooks, those hooks are
1917  * called automatically by this routine.
1918  *
1919  * Devices with legacy power management (no standard PCI PM capabilities)
1920  * always require such platform hooks.
1921  *
1922  * RETURN VALUE:
1923  * 0 is returned on success
1924  * -EINVAL is returned if device is not supposed to wake up the system
1925  * Error code depending on the platform is returned if both the platform and
1926  * the native mechanism fail to enable the generation of wake-up events
1927  */
1928 int pci_enable_wake(struct pci_dev *dev, pci_power_t state, bool enable)
1929 {
1930         int ret = 0;
1931
1932         /*
1933          * Bridges can only signal wakeup on behalf of subordinate devices,
1934          * but that is set up elsewhere, so skip them.
1935          */
1936         if (pci_has_subordinate(dev))
1937                 return 0;
1938
1939         /* Don't do the same thing twice in a row for one device. */
1940         if (!!enable == !!dev->wakeup_prepared)
1941                 return 0;
1942
1943         /*
1944          * According to "PCI System Architecture" 4th ed. by Tom Shanley & Don
1945          * Anderson we should be doing PME# wake enable followed by ACPI wake
1946          * enable.  To disable wake-up we call the platform first, for symmetry.
1947          */
1948
1949         if (enable) {
1950                 int error;
1951
1952                 if (pci_pme_capable(dev, state))
1953                         pci_pme_active(dev, true);
1954                 else
1955                         ret = 1;
1956                 error = platform_pci_set_wakeup(dev, true);
1957                 if (ret)
1958                         ret = error;
1959                 if (!ret)
1960                         dev->wakeup_prepared = true;
1961         } else {
1962                 platform_pci_set_wakeup(dev, false);
1963                 pci_pme_active(dev, false);
1964                 dev->wakeup_prepared = false;
1965         }
1966
1967         return ret;
1968 }
1969 EXPORT_SYMBOL(pci_enable_wake);
1970
1971 /**
1972  * pci_wake_from_d3 - enable/disable device to wake up from D3_hot or D3_cold
1973  * @dev: PCI device to prepare
1974  * @enable: True to enable wake-up event generation; false to disable
1975  *
1976  * Many drivers want the device to wake up the system from D3_hot or D3_cold
1977  * and this function allows them to set that up cleanly - pci_enable_wake()
1978  * should not be called twice in a row to enable wake-up due to PCI PM vs ACPI
1979  * ordering constraints.
1980  *
1981  * This function only returns error code if the device is not capable of
1982  * generating PME# from both D3_hot and D3_cold, and the platform is unable to
1983  * enable wake-up power for it.
1984  */
1985 int pci_wake_from_d3(struct pci_dev *dev, bool enable)
1986 {
1987         return pci_pme_capable(dev, PCI_D3cold) ?
1988                         pci_enable_wake(dev, PCI_D3cold, enable) :
1989                         pci_enable_wake(dev, PCI_D3hot, enable);
1990 }
1991 EXPORT_SYMBOL(pci_wake_from_d3);
1992
1993 /**
1994  * pci_target_state - find an appropriate low power state for a given PCI dev
1995  * @dev: PCI device
1996  * @wakeup: Whether or not wakeup functionality will be enabled for the device.
1997  *
1998  * Use underlying platform code to find a supported low power state for @dev.
1999  * If the platform can't manage @dev, return the deepest state from which it
2000  * can generate wake events, based on any available PME info.
2001  */
2002 static pci_power_t pci_target_state(struct pci_dev *dev, bool wakeup)
2003 {
2004         pci_power_t target_state = PCI_D3hot;
2005
2006         if (platform_pci_power_manageable(dev)) {
2007                 /*
2008                  * Call the platform to choose the target state of the device
2009                  * and enable wake-up from this state if supported.
2010                  */
2011                 pci_power_t state = platform_pci_choose_state(dev);
2012
2013                 switch (state) {
2014                 case PCI_POWER_ERROR:
2015                 case PCI_UNKNOWN:
2016                         break;
2017                 case PCI_D1:
2018                 case PCI_D2:
2019                         if (pci_no_d1d2(dev))
2020                                 break;
2021                 default:
2022                         target_state = state;
2023                 }
2024
2025                 return target_state;
2026         }
2027
2028         if (!dev->pm_cap)
2029                 target_state = PCI_D0;
2030
2031         /*
2032          * If the device is in D3cold even though it's not power-manageable by
2033          * the platform, it may have been powered down by non-standard means.
2034          * Best to let it slumber.
2035          */
2036         if (dev->current_state == PCI_D3cold)
2037                 target_state = PCI_D3cold;
2038
2039         if (wakeup) {
2040                 /*
2041                  * Find the deepest state from which the device can generate
2042                  * wake-up events, make it the target state and enable device
2043                  * to generate PME#.
2044                  */
2045                 if (dev->pme_support) {
2046                         while (target_state
2047                               && !(dev->pme_support & (1 << target_state)))
2048                                 target_state--;
2049                 }
2050         }
2051
2052         return target_state;
2053 }
2054
2055 /**
2056  * pci_prepare_to_sleep - prepare PCI device for system-wide transition into a sleep state
2057  * @dev: Device to handle.
2058  *
2059  * Choose the power state appropriate for the device depending on whether
2060  * it can wake up the system and/or is power manageable by the platform
2061  * (PCI_D3hot is the default) and put the device into that state.
2062  */
2063 int pci_prepare_to_sleep(struct pci_dev *dev)
2064 {
2065         bool wakeup = device_may_wakeup(&dev->dev);
2066         pci_power_t target_state = pci_target_state(dev, wakeup);
2067         int error;
2068
2069         if (target_state == PCI_POWER_ERROR)
2070                 return -EIO;
2071
2072         pci_enable_wake(dev, target_state, wakeup);
2073
2074         error = pci_set_power_state(dev, target_state);
2075
2076         if (error)
2077                 pci_enable_wake(dev, target_state, false);
2078
2079         return error;
2080 }
2081 EXPORT_SYMBOL(pci_prepare_to_sleep);
2082
2083 /**
2084  * pci_back_from_sleep - turn PCI device on during system-wide transition into working state
2085  * @dev: Device to handle.
2086  *
2087  * Disable device's system wake-up capability and put it into D0.
2088  */
2089 int pci_back_from_sleep(struct pci_dev *dev)
2090 {
2091         pci_enable_wake(dev, PCI_D0, false);
2092         return pci_set_power_state(dev, PCI_D0);
2093 }
2094 EXPORT_SYMBOL(pci_back_from_sleep);
2095
2096 /**
2097  * pci_finish_runtime_suspend - Carry out PCI-specific part of runtime suspend.
2098  * @dev: PCI device being suspended.
2099  *
2100  * Prepare @dev to generate wake-up events at run time and put it into a low
2101  * power state.
2102  */
2103 int pci_finish_runtime_suspend(struct pci_dev *dev)
2104 {
2105         pci_power_t target_state;
2106         int error;
2107
2108         target_state = pci_target_state(dev, device_can_wakeup(&dev->dev));
2109         if (target_state == PCI_POWER_ERROR)
2110                 return -EIO;
2111
2112         dev->runtime_d3cold = target_state == PCI_D3cold;
2113
2114         pci_enable_wake(dev, target_state, pci_dev_run_wake(dev));
2115
2116         error = pci_set_power_state(dev, target_state);
2117
2118         if (error) {
2119                 pci_enable_wake(dev, target_state, false);
2120                 dev->runtime_d3cold = false;
2121         }
2122
2123         return error;
2124 }
2125
2126 /**
2127  * pci_dev_run_wake - Check if device can generate run-time wake-up events.
2128  * @dev: Device to check.
2129  *
2130  * Return true if the device itself is capable of generating wake-up events
2131  * (through the platform or using the native PCIe PME) or if the device supports
2132  * PME and one of its upstream bridges can generate wake-up events.
2133  */
2134 bool pci_dev_run_wake(struct pci_dev *dev)
2135 {
2136         struct pci_bus *bus = dev->bus;
2137
2138         if (device_can_wakeup(&dev->dev))
2139                 return true;
2140
2141         if (!dev->pme_support)
2142                 return false;
2143
2144         /* PME-capable in principle, but not from the target power state */
2145         if (!pci_pme_capable(dev, pci_target_state(dev, false)))
2146                 return false;
2147
2148         while (bus->parent) {
2149                 struct pci_dev *bridge = bus->self;
2150
2151                 if (device_can_wakeup(&bridge->dev))
2152                         return true;
2153
2154                 bus = bus->parent;
2155         }
2156
2157         /* We have reached the root bus. */
2158         if (bus->bridge)
2159                 return device_can_wakeup(bus->bridge);
2160
2161         return false;
2162 }
2163 EXPORT_SYMBOL_GPL(pci_dev_run_wake);
2164
2165 /**
2166  * pci_dev_keep_suspended - Check if the device can stay in the suspended state.
2167  * @pci_dev: Device to check.
2168  *
2169  * Return 'true' if the device is runtime-suspended, it doesn't have to be
2170  * reconfigured due to wakeup settings difference between system and runtime
2171  * suspend and the current power state of it is suitable for the upcoming
2172  * (system) transition.
2173  *
2174  * If the device is not configured for system wakeup, disable PME for it before
2175  * returning 'true' to prevent it from waking up the system unnecessarily.
2176  */
2177 bool pci_dev_keep_suspended(struct pci_dev *pci_dev)
2178 {
2179         struct device *dev = &pci_dev->dev;
2180         bool wakeup = device_may_wakeup(dev);
2181
2182         if (!pm_runtime_suspended(dev)
2183             || pci_target_state(pci_dev, wakeup) != pci_dev->current_state
2184             || platform_pci_need_resume(pci_dev))
2185                 return false;
2186
2187         /*
2188          * At this point the device is good to go unless it's been configured
2189          * to generate PME at the runtime suspend time, but it is not supposed
2190          * to wake up the system.  In that case, simply disable PME for it
2191          * (it will have to be re-enabled on exit from system resume).
2192          *
2193          * If the device's power state is D3cold and the platform check above
2194          * hasn't triggered, the device's configuration is suitable and we don't
2195          * need to manipulate it at all.
2196          */
2197         spin_lock_irq(&dev->power.lock);
2198
2199         if (pm_runtime_suspended(dev) && pci_dev->current_state < PCI_D3cold &&
2200             !wakeup)
2201                 __pci_pme_active(pci_dev, false);
2202
2203         spin_unlock_irq(&dev->power.lock);
2204         return true;
2205 }
2206
2207 /**
2208  * pci_dev_complete_resume - Finalize resume from system sleep for a device.
2209  * @pci_dev: Device to handle.
2210  *
2211  * If the device is runtime suspended and wakeup-capable, enable PME for it as
2212  * it might have been disabled during the prepare phase of system suspend if
2213  * the device was not configured for system wakeup.
2214  */
2215 void pci_dev_complete_resume(struct pci_dev *pci_dev)
2216 {
2217         struct device *dev = &pci_dev->dev;
2218
2219         if (!pci_dev_run_wake(pci_dev))
2220                 return;
2221
2222         spin_lock_irq(&dev->power.lock);
2223
2224         if (pm_runtime_suspended(dev) && pci_dev->current_state < PCI_D3cold)
2225                 __pci_pme_active(pci_dev, true);
2226
2227         spin_unlock_irq(&dev->power.lock);
2228 }
2229
2230 void pci_config_pm_runtime_get(struct pci_dev *pdev)
2231 {
2232         struct device *dev = &pdev->dev;
2233         struct device *parent = dev->parent;
2234
2235         if (parent)
2236                 pm_runtime_get_sync(parent);
2237         pm_runtime_get_noresume(dev);
2238         /*
2239          * pdev->current_state is set to PCI_D3cold during suspending,
2240          * so wait until suspending completes
2241          */
2242         pm_runtime_barrier(dev);
2243         /*
2244          * Only need to resume devices in D3cold, because config
2245          * registers are still accessible for devices suspended but
2246          * not in D3cold.
2247          */
2248         if (pdev->current_state == PCI_D3cold)
2249                 pm_runtime_resume(dev);
2250 }
2251
2252 void pci_config_pm_runtime_put(struct pci_dev *pdev)
2253 {
2254         struct device *dev = &pdev->dev;
2255         struct device *parent = dev->parent;
2256
2257         pm_runtime_put(dev);
2258         if (parent)
2259                 pm_runtime_put_sync(parent);
2260 }
2261
2262 /**
2263  * pci_bridge_d3_possible - Is it possible to put the bridge into D3
2264  * @bridge: Bridge to check
2265  *
2266  * This function checks if it is possible to move the bridge to D3.
2267  * Currently we only allow D3 for recent enough PCIe ports.
2268  */
2269 bool pci_bridge_d3_possible(struct pci_dev *bridge)
2270 {
2271         unsigned int year;
2272
2273         if (!pci_is_pcie(bridge))
2274                 return false;
2275
2276         switch (pci_pcie_type(bridge)) {
2277         case PCI_EXP_TYPE_ROOT_PORT:
2278         case PCI_EXP_TYPE_UPSTREAM:
2279         case PCI_EXP_TYPE_DOWNSTREAM:
2280                 if (pci_bridge_d3_disable)
2281                         return false;
2282
2283                 /*
2284                  * Hotplug interrupts cannot be delivered if the link is down,
2285                  * so parents of a hotplug port must stay awake. In addition,
2286                  * hotplug ports handled by firmware in System Management Mode
2287                  * may not be put into D3 by the OS (Thunderbolt on non-Macs).
2288                  * For simplicity, disallow in general for now.
2289                  */
2290                 if (bridge->is_hotplug_bridge)
2291                         return false;
2292
2293                 if (pci_bridge_d3_force)
2294                         return true;
2295
2296                 /*
2297                  * It should be safe to put PCIe ports from 2015 or newer
2298                  * to D3.
2299                  */
2300                 if (dmi_get_date(DMI_BIOS_DATE, &year, NULL, NULL) &&
2301                     year >= 2015) {
2302                         return true;
2303                 }
2304                 break;
2305         }
2306
2307         return false;
2308 }
2309
2310 static int pci_dev_check_d3cold(struct pci_dev *dev, void *data)
2311 {
2312         bool *d3cold_ok = data;
2313
2314         if (/* The device needs to be allowed to go D3cold ... */
2315             dev->no_d3cold || !dev->d3cold_allowed ||
2316
2317             /* ... and if it is wakeup capable to do so from D3cold. */
2318             (device_may_wakeup(&dev->dev) &&
2319              !pci_pme_capable(dev, PCI_D3cold)) ||
2320
2321             /* If it is a bridge it must be allowed to go to D3. */
2322             !pci_power_manageable(dev))
2323
2324                 *d3cold_ok = false;
2325
2326         return !*d3cold_ok;
2327 }
2328
2329 /*
2330  * pci_bridge_d3_update - Update bridge D3 capabilities
2331  * @dev: PCI device which is changed
2332  *
2333  * Update upstream bridge PM capabilities accordingly depending on if the
2334  * device PM configuration was changed or the device is being removed.  The
2335  * change is also propagated upstream.
2336  */
2337 void pci_bridge_d3_update(struct pci_dev *dev)
2338 {
2339         bool remove = !device_is_registered(&dev->dev);
2340         struct pci_dev *bridge;
2341         bool d3cold_ok = true;
2342
2343         bridge = pci_upstream_bridge(dev);
2344         if (!bridge || !pci_bridge_d3_possible(bridge))
2345                 return;
2346
2347         /*
2348          * If D3 is currently allowed for the bridge, removing one of its
2349          * children won't change that.
2350          */
2351         if (remove && bridge->bridge_d3)
2352                 return;
2353
2354         /*
2355          * If D3 is currently allowed for the bridge and a child is added or
2356          * changed, disallowance of D3 can only be caused by that child, so
2357          * we only need to check that single device, not any of its siblings.
2358          *
2359          * If D3 is currently not allowed for the bridge, checking the device
2360          * first may allow us to skip checking its siblings.
2361          */
2362         if (!remove)
2363                 pci_dev_check_d3cold(dev, &d3cold_ok);
2364
2365         /*
2366          * If D3 is currently not allowed for the bridge, this may be caused
2367          * either by the device being changed/removed or any of its siblings,
2368          * so we need to go through all children to find out if one of them
2369          * continues to block D3.
2370          */
2371         if (d3cold_ok && !bridge->bridge_d3)
2372                 pci_walk_bus(bridge->subordinate, pci_dev_check_d3cold,
2373                              &d3cold_ok);
2374
2375         if (bridge->bridge_d3 != d3cold_ok) {
2376                 bridge->bridge_d3 = d3cold_ok;
2377                 /* Propagate change to upstream bridges */
2378                 pci_bridge_d3_update(bridge);
2379         }
2380 }
2381
2382 /**
2383  * pci_d3cold_enable - Enable D3cold for device
2384  * @dev: PCI device to handle
2385  *
2386  * This function can be used in drivers to enable D3cold from the device
2387  * they handle.  It also updates upstream PCI bridge PM capabilities
2388  * accordingly.
2389  */
2390 void pci_d3cold_enable(struct pci_dev *dev)
2391 {
2392         if (dev->no_d3cold) {
2393                 dev->no_d3cold = false;
2394                 pci_bridge_d3_update(dev);
2395         }
2396 }
2397 EXPORT_SYMBOL_GPL(pci_d3cold_enable);
2398
2399 /**
2400  * pci_d3cold_disable - Disable D3cold for device
2401  * @dev: PCI device to handle
2402  *
2403  * This function can be used in drivers to disable D3cold from the device
2404  * they handle.  It also updates upstream PCI bridge PM capabilities
2405  * accordingly.
2406  */
2407 void pci_d3cold_disable(struct pci_dev *dev)
2408 {
2409         if (!dev->no_d3cold) {
2410                 dev->no_d3cold = true;
2411                 pci_bridge_d3_update(dev);
2412         }
2413 }
2414 EXPORT_SYMBOL_GPL(pci_d3cold_disable);
2415
2416 /**
2417  * pci_pm_init - Initialize PM functions of given PCI device
2418  * @dev: PCI device to handle.
2419  */
2420 void pci_pm_init(struct pci_dev *dev)
2421 {
2422         int pm;
2423         u16 pmc;
2424
2425         pm_runtime_forbid(&dev->dev);
2426         pm_runtime_set_active(&dev->dev);
2427         pm_runtime_enable(&dev->dev);
2428         device_enable_async_suspend(&dev->dev);
2429         dev->wakeup_prepared = false;
2430
2431         dev->pm_cap = 0;
2432         dev->pme_support = 0;
2433
2434         /* find PCI PM capability in list */
2435         pm = pci_find_capability(dev, PCI_CAP_ID_PM);
2436         if (!pm)
2437                 return;
2438         /* Check device's ability to generate PME# */
2439         pci_read_config_word(dev, pm + PCI_PM_PMC, &pmc);
2440
2441         if ((pmc & PCI_PM_CAP_VER_MASK) > 3) {
2442                 pci_err(dev, "unsupported PM cap regs version (%u)\n",
2443                         pmc & PCI_PM_CAP_VER_MASK);
2444                 return;
2445         }
2446
2447         dev->pm_cap = pm;
2448         dev->d3_delay = PCI_PM_D3_WAIT;
2449         dev->d3cold_delay = PCI_PM_D3COLD_WAIT;
2450         dev->bridge_d3 = pci_bridge_d3_possible(dev);
2451         dev->d3cold_allowed = true;
2452
2453         dev->d1_support = false;
2454         dev->d2_support = false;
2455         if (!pci_no_d1d2(dev)) {
2456                 if (pmc & PCI_PM_CAP_D1)
2457                         dev->d1_support = true;
2458                 if (pmc & PCI_PM_CAP_D2)
2459                         dev->d2_support = true;
2460
2461                 if (dev->d1_support || dev->d2_support)
2462                         pci_printk(KERN_DEBUG, dev, "supports%s%s\n",
2463                                    dev->d1_support ? " D1" : "",
2464                                    dev->d2_support ? " D2" : "");
2465         }
2466
2467         pmc &= PCI_PM_CAP_PME_MASK;
2468         if (pmc) {
2469                 pci_printk(KERN_DEBUG, dev, "PME# supported from%s%s%s%s%s\n",
2470                          (pmc & PCI_PM_CAP_PME_D0) ? " D0" : "",
2471                          (pmc & PCI_PM_CAP_PME_D1) ? " D1" : "",
2472                          (pmc & PCI_PM_CAP_PME_D2) ? " D2" : "",
2473                          (pmc & PCI_PM_CAP_PME_D3) ? " D3hot" : "",
2474                          (pmc & PCI_PM_CAP_PME_D3cold) ? " D3cold" : "");
2475                 dev->pme_support = pmc >> PCI_PM_CAP_PME_SHIFT;
2476                 dev->pme_poll = true;
2477                 /*
2478                  * Make device's PM flags reflect the wake-up capability, but
2479                  * let the user space enable it to wake up the system as needed.
2480                  */
2481                 device_set_wakeup_capable(&dev->dev, true);
2482                 /* Disable the PME# generation functionality */
2483                 pci_pme_active(dev, false);
2484         }
2485 }
2486
2487 static unsigned long pci_ea_flags(struct pci_dev *dev, u8 prop)
2488 {
2489         unsigned long flags = IORESOURCE_PCI_FIXED | IORESOURCE_PCI_EA_BEI;
2490
2491         switch (prop) {
2492         case PCI_EA_P_MEM:
2493         case PCI_EA_P_VF_MEM:
2494                 flags |= IORESOURCE_MEM;
2495                 break;
2496         case PCI_EA_P_MEM_PREFETCH:
2497         case PCI_EA_P_VF_MEM_PREFETCH:
2498                 flags |= IORESOURCE_MEM | IORESOURCE_PREFETCH;
2499                 break;
2500         case PCI_EA_P_IO:
2501                 flags |= IORESOURCE_IO;
2502                 break;
2503         default:
2504                 return 0;
2505         }
2506
2507         return flags;
2508 }
2509
2510 static struct resource *pci_ea_get_resource(struct pci_dev *dev, u8 bei,
2511                                             u8 prop)
2512 {
2513         if (bei <= PCI_EA_BEI_BAR5 && prop <= PCI_EA_P_IO)
2514                 return &dev->resource[bei];
2515 #ifdef CONFIG_PCI_IOV
2516         else if (bei >= PCI_EA_BEI_VF_BAR0 && bei <= PCI_EA_BEI_VF_BAR5 &&
2517                  (prop == PCI_EA_P_VF_MEM || prop == PCI_EA_P_VF_MEM_PREFETCH))
2518                 return &dev->resource[PCI_IOV_RESOURCES +
2519                                       bei - PCI_EA_BEI_VF_BAR0];
2520 #endif
2521         else if (bei == PCI_EA_BEI_ROM)
2522                 return &dev->resource[PCI_ROM_RESOURCE];
2523         else
2524                 return NULL;
2525 }
2526
2527 /* Read an Enhanced Allocation (EA) entry */
2528 static int pci_ea_read(struct pci_dev *dev, int offset)
2529 {
2530         struct resource *res;
2531         int ent_size, ent_offset = offset;
2532         resource_size_t start, end;
2533         unsigned long flags;
2534         u32 dw0, bei, base, max_offset;
2535         u8 prop;
2536         bool support_64 = (sizeof(resource_size_t) >= 8);
2537
2538         pci_read_config_dword(dev, ent_offset, &dw0);
2539         ent_offset += 4;
2540
2541         /* Entry size field indicates DWORDs after 1st */
2542         ent_size = ((dw0 & PCI_EA_ES) + 1) << 2;
2543
2544         if (!(dw0 & PCI_EA_ENABLE)) /* Entry not enabled */
2545                 goto out;
2546
2547         bei = (dw0 & PCI_EA_BEI) >> 4;
2548         prop = (dw0 & PCI_EA_PP) >> 8;
2549
2550         /*
2551          * If the Property is in the reserved range, try the Secondary
2552          * Property instead.
2553          */
2554         if (prop > PCI_EA_P_BRIDGE_IO && prop < PCI_EA_P_MEM_RESERVED)
2555                 prop = (dw0 & PCI_EA_SP) >> 16;
2556         if (prop > PCI_EA_P_BRIDGE_IO)
2557                 goto out;
2558
2559         res = pci_ea_get_resource(dev, bei, prop);
2560         if (!res) {
2561                 pci_err(dev, "Unsupported EA entry BEI: %u\n", bei);
2562                 goto out;
2563         }
2564
2565         flags = pci_ea_flags(dev, prop);
2566         if (!flags) {
2567                 pci_err(dev, "Unsupported EA properties: %#x\n", prop);
2568                 goto out;
2569         }
2570
2571         /* Read Base */
2572         pci_read_config_dword(dev, ent_offset, &base);
2573         start = (base & PCI_EA_FIELD_MASK);
2574         ent_offset += 4;
2575
2576         /* Read MaxOffset */
2577         pci_read_config_dword(dev, ent_offset, &max_offset);
2578         ent_offset += 4;
2579
2580         /* Read Base MSBs (if 64-bit entry) */
2581         if (base & PCI_EA_IS_64) {
2582                 u32 base_upper;
2583
2584                 pci_read_config_dword(dev, ent_offset, &base_upper);
2585                 ent_offset += 4;
2586
2587                 flags |= IORESOURCE_MEM_64;
2588
2589                 /* entry starts above 32-bit boundary, can't use */
2590                 if (!support_64 && base_upper)
2591                         goto out;
2592
2593                 if (support_64)
2594                         start |= ((u64)base_upper << 32);
2595         }
2596
2597         end = start + (max_offset | 0x03);
2598
2599         /* Read MaxOffset MSBs (if 64-bit entry) */
2600         if (max_offset & PCI_EA_IS_64) {
2601                 u32 max_offset_upper;
2602
2603                 pci_read_config_dword(dev, ent_offset, &max_offset_upper);
2604                 ent_offset += 4;
2605
2606                 flags |= IORESOURCE_MEM_64;
2607
2608                 /* entry too big, can't use */
2609                 if (!support_64 && max_offset_upper)
2610                         goto out;
2611
2612                 if (support_64)
2613                         end += ((u64)max_offset_upper << 32);
2614         }
2615
2616         if (end < start) {
2617                 pci_err(dev, "EA Entry crosses address boundary\n");
2618                 goto out;
2619         }
2620
2621         if (ent_size != ent_offset - offset) {
2622                 pci_err(dev, "EA Entry Size (%d) does not match length read (%d)\n",
2623                         ent_size, ent_offset - offset);
2624                 goto out;
2625         }
2626
2627         res->name = pci_name(dev);
2628         res->start = start;
2629         res->end = end;
2630         res->flags = flags;
2631
2632         if (bei <= PCI_EA_BEI_BAR5)
2633                 pci_printk(KERN_DEBUG, dev, "BAR %d: %pR (from Enhanced Allocation, properties %#02x)\n",
2634                            bei, res, prop);
2635         else if (bei == PCI_EA_BEI_ROM)
2636                 pci_printk(KERN_DEBUG, dev, "ROM: %pR (from Enhanced Allocation, properties %#02x)\n",
2637                            res, prop);
2638         else if (bei >= PCI_EA_BEI_VF_BAR0 && bei <= PCI_EA_BEI_VF_BAR5)
2639                 pci_printk(KERN_DEBUG, dev, "VF BAR %d: %pR (from Enhanced Allocation, properties %#02x)\n",
2640                            bei - PCI_EA_BEI_VF_BAR0, res, prop);
2641         else
2642                 pci_printk(KERN_DEBUG, dev, "BEI %d res: %pR (from Enhanced Allocation, properties %#02x)\n",
2643                            bei, res, prop);
2644
2645 out:
2646         return offset + ent_size;
2647 }
2648
2649 /* Enhanced Allocation Initialization */
2650 void pci_ea_init(struct pci_dev *dev)
2651 {
2652         int ea;
2653         u8 num_ent;
2654         int offset;
2655         int i;
2656
2657         /* find PCI EA capability in list */
2658         ea = pci_find_capability(dev, PCI_CAP_ID_EA);
2659         if (!ea)
2660                 return;
2661
2662         /* determine the number of entries */
2663         pci_bus_read_config_byte(dev->bus, dev->devfn, ea + PCI_EA_NUM_ENT,
2664                                         &num_ent);
2665         num_ent &= PCI_EA_NUM_ENT_MASK;
2666
2667         offset = ea + PCI_EA_FIRST_ENT;
2668
2669         /* Skip DWORD 2 for type 1 functions */
2670         if (dev->hdr_type == PCI_HEADER_TYPE_BRIDGE)
2671                 offset += 4;
2672
2673         /* parse each EA entry */
2674         for (i = 0; i < num_ent; ++i)
2675                 offset = pci_ea_read(dev, offset);
2676 }
2677
2678 static void pci_add_saved_cap(struct pci_dev *pci_dev,
2679         struct pci_cap_saved_state *new_cap)
2680 {
2681         hlist_add_head(&new_cap->next, &pci_dev->saved_cap_space);
2682 }
2683
2684 /**
2685  * _pci_add_cap_save_buffer - allocate buffer for saving given
2686  *                            capability registers
2687  * @dev: the PCI device
2688  * @cap: the capability to allocate the buffer for
2689  * @extended: Standard or Extended capability ID
2690  * @size: requested size of the buffer
2691  */
2692 static int _pci_add_cap_save_buffer(struct pci_dev *dev, u16 cap,
2693                                     bool extended, unsigned int size)
2694 {
2695         int pos;
2696         struct pci_cap_saved_state *save_state;
2697
2698         if (extended)
2699                 pos = pci_find_ext_capability(dev, cap);
2700         else
2701                 pos = pci_find_capability(dev, cap);
2702
2703         if (!pos)
2704                 return 0;
2705
2706         save_state = kzalloc(sizeof(*save_state) + size, GFP_KERNEL);
2707         if (!save_state)
2708                 return -ENOMEM;
2709
2710         save_state->cap.cap_nr = cap;
2711         save_state->cap.cap_extended = extended;
2712         save_state->cap.size = size;
2713         pci_add_saved_cap(dev, save_state);
2714
2715         return 0;
2716 }
2717
2718 int pci_add_cap_save_buffer(struct pci_dev *dev, char cap, unsigned int size)
2719 {
2720         return _pci_add_cap_save_buffer(dev, cap, false, size);
2721 }
2722
2723 int pci_add_ext_cap_save_buffer(struct pci_dev *dev, u16 cap, unsigned int size)
2724 {
2725         return _pci_add_cap_save_buffer(dev, cap, true, size);
2726 }
2727
2728 /**
2729  * pci_allocate_cap_save_buffers - allocate buffers for saving capabilities
2730  * @dev: the PCI device
2731  */
2732 void pci_allocate_cap_save_buffers(struct pci_dev *dev)
2733 {
2734         int error;
2735
2736         error = pci_add_cap_save_buffer(dev, PCI_CAP_ID_EXP,
2737                                         PCI_EXP_SAVE_REGS * sizeof(u16));
2738         if (error)
2739                 pci_err(dev, "unable to preallocate PCI Express save buffer\n");
2740
2741         error = pci_add_cap_save_buffer(dev, PCI_CAP_ID_PCIX, sizeof(u16));
2742         if (error)
2743                 pci_err(dev, "unable to preallocate PCI-X save buffer\n");
2744
2745         pci_allocate_vc_save_buffers(dev);
2746 }
2747
2748 void pci_free_cap_save_buffers(struct pci_dev *dev)
2749 {
2750         struct pci_cap_saved_state *tmp;
2751         struct hlist_node *n;
2752
2753         hlist_for_each_entry_safe(tmp, n, &dev->saved_cap_space, next)
2754                 kfree(tmp);
2755 }
2756
2757 /**
2758  * pci_configure_ari - enable or disable ARI forwarding
2759  * @dev: the PCI device
2760  *
2761  * If @dev and its upstream bridge both support ARI, enable ARI in the
2762  * bridge.  Otherwise, disable ARI in the bridge.
2763  */
2764 void pci_configure_ari(struct pci_dev *dev)
2765 {
2766         u32 cap;
2767         struct pci_dev *bridge;
2768
2769         if (pcie_ari_disabled || !pci_is_pcie(dev) || dev->devfn)
2770                 return;
2771
2772         bridge = dev->bus->self;
2773         if (!bridge)
2774                 return;
2775
2776         pcie_capability_read_dword(bridge, PCI_EXP_DEVCAP2, &cap);
2777         if (!(cap & PCI_EXP_DEVCAP2_ARI))
2778                 return;
2779
2780         if (pci_find_ext_capability(dev, PCI_EXT_CAP_ID_ARI)) {
2781                 pcie_capability_set_word(bridge, PCI_EXP_DEVCTL2,
2782                                          PCI_EXP_DEVCTL2_ARI);
2783                 bridge->ari_enabled = 1;
2784         } else {
2785                 pcie_capability_clear_word(bridge, PCI_EXP_DEVCTL2,
2786                                            PCI_EXP_DEVCTL2_ARI);
2787                 bridge->ari_enabled = 0;
2788         }
2789 }
2790
2791 static int pci_acs_enable;
2792
2793 /**
2794  * pci_request_acs - ask for ACS to be enabled if supported
2795  */
2796 void pci_request_acs(void)
2797 {
2798         pci_acs_enable = 1;
2799 }
2800
2801 /**
2802  * pci_std_enable_acs - enable ACS on devices using standard ACS capabilites
2803  * @dev: the PCI device
2804  */
2805 static void pci_std_enable_acs(struct pci_dev *dev)
2806 {
2807         int pos;
2808         u16 cap;
2809         u16 ctrl;
2810
2811         pos = pci_find_ext_capability(dev, PCI_EXT_CAP_ID_ACS);
2812         if (!pos)
2813                 return;
2814
2815         pci_read_config_word(dev, pos + PCI_ACS_CAP, &cap);
2816         pci_read_config_word(dev, pos + PCI_ACS_CTRL, &ctrl);
2817
2818         /* Source Validation */
2819         ctrl |= (cap & PCI_ACS_SV);
2820
2821         /* P2P Request Redirect */
2822         ctrl |= (cap & PCI_ACS_RR);
2823
2824         /* P2P Completion Redirect */
2825         ctrl |= (cap & PCI_ACS_CR);
2826
2827         /* Upstream Forwarding */
2828         ctrl |= (cap & PCI_ACS_UF);
2829
2830         pci_write_config_word(dev, pos + PCI_ACS_CTRL, ctrl);
2831 }
2832
2833 /**
2834  * pci_enable_acs - enable ACS if hardware support it
2835  * @dev: the PCI device
2836  */
2837 void pci_enable_acs(struct pci_dev *dev)
2838 {
2839         if (!pci_acs_enable)
2840                 return;
2841
2842         if (!pci_dev_specific_enable_acs(dev))
2843                 return;
2844
2845         pci_std_enable_acs(dev);
2846 }
2847
2848 static bool pci_acs_flags_enabled(struct pci_dev *pdev, u16 acs_flags)
2849 {
2850         int pos;
2851         u16 cap, ctrl;
2852
2853         pos = pci_find_ext_capability(pdev, PCI_EXT_CAP_ID_ACS);
2854         if (!pos)
2855                 return false;
2856
2857         /*
2858          * Except for egress control, capabilities are either required
2859          * or only required if controllable.  Features missing from the
2860          * capability field can therefore be assumed as hard-wired enabled.
2861          */
2862         pci_read_config_word(pdev, pos + PCI_ACS_CAP, &cap);
2863         acs_flags &= (cap | PCI_ACS_EC);
2864
2865         pci_read_config_word(pdev, pos + PCI_ACS_CTRL, &ctrl);
2866         return (ctrl & acs_flags) == acs_flags;
2867 }
2868
2869 /**
2870  * pci_acs_enabled - test ACS against required flags for a given device
2871  * @pdev: device to test
2872  * @acs_flags: required PCI ACS flags
2873  *
2874  * Return true if the device supports the provided flags.  Automatically
2875  * filters out flags that are not implemented on multifunction devices.
2876  *
2877  * Note that this interface checks the effective ACS capabilities of the
2878  * device rather than the actual capabilities.  For instance, most single
2879  * function endpoints are not required to support ACS because they have no
2880  * opportunity for peer-to-peer access.  We therefore return 'true'
2881  * regardless of whether the device exposes an ACS capability.  This makes
2882  * it much easier for callers of this function to ignore the actual type
2883  * or topology of the device when testing ACS support.
2884  */
2885 bool pci_acs_enabled(struct pci_dev *pdev, u16 acs_flags)
2886 {
2887         int ret;
2888
2889         ret = pci_dev_specific_acs_enabled(pdev, acs_flags);
2890         if (ret >= 0)
2891                 return ret > 0;
2892
2893         /*
2894          * Conventional PCI and PCI-X devices never support ACS, either
2895          * effectively or actually.  The shared bus topology implies that
2896          * any device on the bus can receive or snoop DMA.
2897          */
2898         if (!pci_is_pcie(pdev))
2899                 return false;
2900
2901         switch (pci_pcie_type(pdev)) {
2902         /*
2903          * PCI/X-to-PCIe bridges are not specifically mentioned by the spec,
2904          * but since their primary interface is PCI/X, we conservatively
2905          * handle them as we would a non-PCIe device.
2906          */
2907         case PCI_EXP_TYPE_PCIE_BRIDGE:
2908         /*
2909          * PCIe 3.0, 6.12.1 excludes ACS on these devices.  "ACS is never
2910          * applicable... must never implement an ACS Extended Capability...".
2911          * This seems arbitrary, but we take a conservative interpretation
2912          * of this statement.
2913          */
2914         case PCI_EXP_TYPE_PCI_BRIDGE:
2915         case PCI_EXP_TYPE_RC_EC:
2916                 return false;
2917         /*
2918          * PCIe 3.0, 6.12.1.1 specifies that downstream and root ports should
2919          * implement ACS in order to indicate their peer-to-peer capabilities,
2920          * regardless of whether they are single- or multi-function devices.
2921          */
2922         case PCI_EXP_TYPE_DOWNSTREAM:
2923         case PCI_EXP_TYPE_ROOT_PORT:
2924                 return pci_acs_flags_enabled(pdev, acs_flags);
2925         /*
2926          * PCIe 3.0, 6.12.1.2 specifies ACS capabilities that should be
2927          * implemented by the remaining PCIe types to indicate peer-to-peer
2928          * capabilities, but only when they are part of a multifunction
2929          * device.  The footnote for section 6.12 indicates the specific
2930          * PCIe types included here.
2931          */
2932         case PCI_EXP_TYPE_ENDPOINT:
2933         case PCI_EXP_TYPE_UPSTREAM:
2934         case PCI_EXP_TYPE_LEG_END:
2935         case PCI_EXP_TYPE_RC_END:
2936                 if (!pdev->multifunction)
2937                         break;
2938
2939                 return pci_acs_flags_enabled(pdev, acs_flags);
2940         }
2941
2942         /*
2943          * PCIe 3.0, 6.12.1.3 specifies no ACS capabilities are applicable
2944          * to single function devices with the exception of downstream ports.
2945          */
2946         return true;
2947 }
2948
2949 /**
2950  * pci_acs_path_enable - test ACS flags from start to end in a hierarchy
2951  * @start: starting downstream device
2952  * @end: ending upstream device or NULL to search to the root bus
2953  * @acs_flags: required flags
2954  *
2955  * Walk up a device tree from start to end testing PCI ACS support.  If
2956  * any step along the way does not support the required flags, return false.
2957  */
2958 bool pci_acs_path_enabled(struct pci_dev *start,
2959                           struct pci_dev *end, u16 acs_flags)
2960 {
2961         struct pci_dev *pdev, *parent = start;
2962
2963         do {
2964                 pdev = parent;
2965
2966                 if (!pci_acs_enabled(pdev, acs_flags))
2967                         return false;
2968
2969                 if (pci_is_root_bus(pdev->bus))
2970                         return (end == NULL);
2971
2972                 parent = pdev->bus->self;
2973         } while (pdev != end);
2974
2975         return true;
2976 }
2977
2978 /**
2979  * pci_rebar_find_pos - find position of resize ctrl reg for BAR
2980  * @pdev: PCI device
2981  * @bar: BAR to find
2982  *
2983  * Helper to find the position of the ctrl register for a BAR.
2984  * Returns -ENOTSUPP if resizable BARs are not supported at all.
2985  * Returns -ENOENT if no ctrl register for the BAR could be found.
2986  */
2987 static int pci_rebar_find_pos(struct pci_dev *pdev, int bar)
2988 {
2989         unsigned int pos, nbars, i;
2990         u32 ctrl;
2991
2992         pos = pci_find_ext_capability(pdev, PCI_EXT_CAP_ID_REBAR);
2993         if (!pos)
2994                 return -ENOTSUPP;
2995
2996         pci_read_config_dword(pdev, pos + PCI_REBAR_CTRL, &ctrl);
2997         nbars = (ctrl & PCI_REBAR_CTRL_NBAR_MASK) >>
2998                     PCI_REBAR_CTRL_NBAR_SHIFT;
2999
3000         for (i = 0; i < nbars; i++, pos += 8) {
3001                 int bar_idx;
3002
3003                 pci_read_config_dword(pdev, pos + PCI_REBAR_CTRL, &ctrl);
3004                 bar_idx = ctrl & PCI_REBAR_CTRL_BAR_IDX;
3005                 if (bar_idx == bar)
3006                         return pos;
3007         }
3008
3009         return -ENOENT;
3010 }
3011
3012 /**
3013  * pci_rebar_get_possible_sizes - get possible sizes for BAR
3014  * @pdev: PCI device
3015  * @bar: BAR to query
3016  *
3017  * Get the possible sizes of a resizable BAR as bitmask defined in the spec
3018  * (bit 0=1MB, bit 19=512GB). Returns 0 if BAR isn't resizable.
3019  */
3020 u32 pci_rebar_get_possible_sizes(struct pci_dev *pdev, int bar)
3021 {
3022         int pos;
3023         u32 cap;
3024
3025         pos = pci_rebar_find_pos(pdev, bar);
3026         if (pos < 0)
3027                 return 0;
3028
3029         pci_read_config_dword(pdev, pos + PCI_REBAR_CAP, &cap);
3030         return (cap & PCI_REBAR_CAP_SIZES) >> 4;
3031 }
3032
3033 /**
3034  * pci_rebar_get_current_size - get the current size of a BAR
3035  * @pdev: PCI device
3036  * @bar: BAR to set size to
3037  *
3038  * Read the size of a BAR from the resizable BAR config.
3039  * Returns size if found or negative error code.
3040  */
3041 int pci_rebar_get_current_size(struct pci_dev *pdev, int bar)
3042 {
3043         int pos;
3044         u32 ctrl;
3045
3046         pos = pci_rebar_find_pos(pdev, bar);
3047         if (pos < 0)
3048                 return pos;
3049
3050         pci_read_config_dword(pdev, pos + PCI_REBAR_CTRL, &ctrl);
3051         return (ctrl & PCI_REBAR_CTRL_BAR_SIZE) >> 8;
3052 }
3053
3054 /**
3055  * pci_rebar_set_size - set a new size for a BAR
3056  * @pdev: PCI device
3057  * @bar: BAR to set size to
3058  * @size: new size as defined in the spec (0=1MB, 19=512GB)
3059  *
3060  * Set the new size of a BAR as defined in the spec.
3061  * Returns zero if resizing was successful, error code otherwise.
3062  */
3063 int pci_rebar_set_size(struct pci_dev *pdev, int bar, int size)
3064 {
3065         int pos;
3066         u32 ctrl;
3067
3068         pos = pci_rebar_find_pos(pdev, bar);
3069         if (pos < 0)
3070                 return pos;
3071
3072         pci_read_config_dword(pdev, pos + PCI_REBAR_CTRL, &ctrl);
3073         ctrl &= ~PCI_REBAR_CTRL_BAR_SIZE;
3074         ctrl |= size << 8;
3075         pci_write_config_dword(pdev, pos + PCI_REBAR_CTRL, ctrl);
3076         return 0;
3077 }
3078
3079 /**
3080  * pci_enable_atomic_ops_to_root - enable AtomicOp requests to root port
3081  * @dev: the PCI device
3082  * @cap_mask: mask of desired AtomicOp sizes, including one or more of:
3083  *      PCI_EXP_DEVCAP2_ATOMIC_COMP32
3084  *      PCI_EXP_DEVCAP2_ATOMIC_COMP64
3085  *      PCI_EXP_DEVCAP2_ATOMIC_COMP128
3086  *
3087  * Return 0 if all upstream bridges support AtomicOp routing, egress
3088  * blocking is disabled on all upstream ports, and the root port supports
3089  * the requested completion capabilities (32-bit, 64-bit and/or 128-bit
3090  * AtomicOp completion), or negative otherwise.
3091  */
3092 int pci_enable_atomic_ops_to_root(struct pci_dev *dev, u32 cap_mask)
3093 {
3094         struct pci_bus *bus = dev->bus;
3095         struct pci_dev *bridge;
3096         u32 cap, ctl2;
3097
3098         if (!pci_is_pcie(dev))
3099                 return -EINVAL;
3100
3101         /*
3102          * Per PCIe r4.0, sec 6.15, endpoints and root ports may be
3103          * AtomicOp requesters.  For now, we only support endpoints as
3104          * requesters and root ports as completers.  No endpoints as
3105          * completers, and no peer-to-peer.
3106          */
3107
3108         switch (pci_pcie_type(dev)) {
3109         case PCI_EXP_TYPE_ENDPOINT:
3110         case PCI_EXP_TYPE_LEG_END:
3111         case PCI_EXP_TYPE_RC_END:
3112                 break;
3113         default:
3114                 return -EINVAL;
3115         }
3116
3117         while (bus->parent) {
3118                 bridge = bus->self;
3119
3120                 pcie_capability_read_dword(bridge, PCI_EXP_DEVCAP2, &cap);
3121
3122                 switch (pci_pcie_type(bridge)) {
3123                 /* Ensure switch ports support AtomicOp routing */
3124                 case PCI_EXP_TYPE_UPSTREAM:
3125                 case PCI_EXP_TYPE_DOWNSTREAM:
3126                         if (!(cap & PCI_EXP_DEVCAP2_ATOMIC_ROUTE))
3127                                 return -EINVAL;
3128                         break;
3129
3130                 /* Ensure root port supports all the sizes we care about */
3131                 case PCI_EXP_TYPE_ROOT_PORT:
3132                         if ((cap & cap_mask) != cap_mask)
3133                                 return -EINVAL;
3134                         break;
3135                 }
3136
3137                 /* Ensure upstream ports don't block AtomicOps on egress */
3138                 if (!bridge->has_secondary_link) {
3139                         pcie_capability_read_dword(bridge, PCI_EXP_DEVCTL2,
3140                                                    &ctl2);
3141                         if (ctl2 & PCI_EXP_DEVCTL2_ATOMIC_EGRESS_BLOCK)
3142                                 return -EINVAL;
3143                 }
3144
3145                 bus = bus->parent;
3146         }
3147
3148         pcie_capability_set_word(dev, PCI_EXP_DEVCTL2,
3149                                  PCI_EXP_DEVCTL2_ATOMIC_REQ);
3150         return 0;
3151 }
3152 EXPORT_SYMBOL(pci_enable_atomic_ops_to_root);
3153
3154 /**
3155  * pci_swizzle_interrupt_pin - swizzle INTx for device behind bridge
3156  * @dev: the PCI device
3157  * @pin: the INTx pin (1=INTA, 2=INTB, 3=INTC, 4=INTD)
3158  *
3159  * Perform INTx swizzling for a device behind one level of bridge.  This is
3160  * required by section 9.1 of the PCI-to-PCI bridge specification for devices
3161  * behind bridges on add-in cards.  For devices with ARI enabled, the slot
3162  * number is always 0 (see the Implementation Note in section 2.2.8.1 of
3163  * the PCI Express Base Specification, Revision 2.1)
3164  */
3165 u8 pci_swizzle_interrupt_pin(const struct pci_dev *dev, u8 pin)
3166 {
3167         int slot;
3168
3169         if (pci_ari_enabled(dev->bus))
3170                 slot = 0;
3171         else
3172                 slot = PCI_SLOT(dev->devfn);
3173
3174         return (((pin - 1) + slot) % 4) + 1;
3175 }
3176
3177 int pci_get_interrupt_pin(struct pci_dev *dev, struct pci_dev **bridge)
3178 {
3179         u8 pin;
3180
3181         pin = dev->pin;
3182         if (!pin)
3183                 return -1;
3184
3185         while (!pci_is_root_bus(dev->bus)) {
3186                 pin = pci_swizzle_interrupt_pin(dev, pin);
3187                 dev = dev->bus->self;
3188         }
3189         *bridge = dev;
3190         return pin;
3191 }
3192
3193 /**
3194  * pci_common_swizzle - swizzle INTx all the way to root bridge
3195  * @dev: the PCI device
3196  * @pinp: pointer to the INTx pin value (1=INTA, 2=INTB, 3=INTD, 4=INTD)
3197  *
3198  * Perform INTx swizzling for a device.  This traverses through all PCI-to-PCI
3199  * bridges all the way up to a PCI root bus.
3200  */
3201 u8 pci_common_swizzle(struct pci_dev *dev, u8 *pinp)
3202 {
3203         u8 pin = *pinp;
3204
3205         while (!pci_is_root_bus(dev->bus)) {
3206                 pin = pci_swizzle_interrupt_pin(dev, pin);
3207                 dev = dev->bus->self;
3208         }
3209         *pinp = pin;
3210         return PCI_SLOT(dev->devfn);
3211 }
3212 EXPORT_SYMBOL_GPL(pci_common_swizzle);
3213
3214 /**
3215  *      pci_release_region - Release a PCI bar
3216  *      @pdev: PCI device whose resources were previously reserved by pci_request_region
3217  *      @bar: BAR to release
3218  *
3219  *      Releases the PCI I/O and memory resources previously reserved by a
3220  *      successful call to pci_request_region.  Call this function only
3221  *      after all use of the PCI regions has ceased.
3222  */
3223 void pci_release_region(struct pci_dev *pdev, int bar)
3224 {
3225         struct pci_devres *dr;
3226
3227         if (pci_resource_len(pdev, bar) == 0)
3228                 return;
3229         if (pci_resource_flags(pdev, bar) & IORESOURCE_IO)
3230                 release_region(pci_resource_start(pdev, bar),
3231                                 pci_resource_len(pdev, bar));
3232         else if (pci_resource_flags(pdev, bar) & IORESOURCE_MEM)
3233                 release_mem_region(pci_resource_start(pdev, bar),
3234                                 pci_resource_len(pdev, bar));
3235
3236         dr = find_pci_dr(pdev);
3237         if (dr)
3238                 dr->region_mask &= ~(1 << bar);
3239 }
3240 EXPORT_SYMBOL(pci_release_region);
3241
3242 /**
3243  *      __pci_request_region - Reserved PCI I/O and memory resource
3244  *      @pdev: PCI device whose resources are to be reserved
3245  *      @bar: BAR to be reserved
3246  *      @res_name: Name to be associated with resource.
3247  *      @exclusive: whether the region access is exclusive or not
3248  *
3249  *      Mark the PCI region associated with PCI device @pdev BR @bar as
3250  *      being reserved by owner @res_name.  Do not access any
3251  *      address inside the PCI regions unless this call returns
3252  *      successfully.
3253  *
3254  *      If @exclusive is set, then the region is marked so that userspace
3255  *      is explicitly not allowed to map the resource via /dev/mem or
3256  *      sysfs MMIO access.
3257  *
3258  *      Returns 0 on success, or %EBUSY on error.  A warning
3259  *      message is also printed on failure.
3260  */
3261 static int __pci_request_region(struct pci_dev *pdev, int bar,
3262                                 const char *res_name, int exclusive)
3263 {
3264         struct pci_devres *dr;
3265
3266         if (pci_resource_len(pdev, bar) == 0)
3267                 return 0;
3268
3269         if (pci_resource_flags(pdev, bar) & IORESOURCE_IO) {
3270                 if (!request_region(pci_resource_start(pdev, bar),
3271                             pci_resource_len(pdev, bar), res_name))
3272                         goto err_out;
3273         } else if (pci_resource_flags(pdev, bar) & IORESOURCE_MEM) {
3274                 if (!__request_mem_region(pci_resource_start(pdev, bar),
3275                                         pci_resource_len(pdev, bar), res_name,
3276                                         exclusive))
3277                         goto err_out;
3278         }
3279
3280         dr = find_pci_dr(pdev);
3281         if (dr)
3282                 dr->region_mask |= 1 << bar;
3283
3284         return 0;
3285
3286 err_out:
3287         pci_warn(pdev, "BAR %d: can't reserve %pR\n", bar,
3288                  &pdev->resource[bar]);
3289         return -EBUSY;
3290 }
3291
3292 /**
3293  *      pci_request_region - Reserve PCI I/O and memory resource
3294  *      @pdev: PCI device whose resources are to be reserved
3295  *      @bar: BAR to be reserved
3296  *      @res_name: Name to be associated with resource
3297  *
3298  *      Mark the PCI region associated with PCI device @pdev BAR @bar as
3299  *      being reserved by owner @res_name.  Do not access any
3300  *      address inside the PCI regions unless this call returns
3301  *      successfully.
3302  *
3303  *      Returns 0 on success, or %EBUSY on error.  A warning
3304  *      message is also printed on failure.
3305  */
3306 int pci_request_region(struct pci_dev *pdev, int bar, const char *res_name)
3307 {
3308         return __pci_request_region(pdev, bar, res_name, 0);
3309 }
3310 EXPORT_SYMBOL(pci_request_region);
3311
3312 /**
3313  *      pci_request_region_exclusive - Reserved PCI I/O and memory resource
3314  *      @pdev: PCI device whose resources are to be reserved
3315  *      @bar: BAR to be reserved
3316  *      @res_name: Name to be associated with resource.
3317  *
3318  *      Mark the PCI region associated with PCI device @pdev BR @bar as
3319  *      being reserved by owner @res_name.  Do not access any
3320  *      address inside the PCI regions unless this call returns
3321  *      successfully.
3322  *
3323  *      Returns 0 on success, or %EBUSY on error.  A warning
3324  *      message is also printed on failure.
3325  *
3326  *      The key difference that _exclusive makes it that userspace is
3327  *      explicitly not allowed to map the resource via /dev/mem or
3328  *      sysfs.
3329  */
3330 int pci_request_region_exclusive(struct pci_dev *pdev, int bar,
3331                                  const char *res_name)
3332 {
3333         return __pci_request_region(pdev, bar, res_name, IORESOURCE_EXCLUSIVE);
3334 }
3335 EXPORT_SYMBOL(pci_request_region_exclusive);
3336
3337 /**
3338  * pci_release_selected_regions - Release selected PCI I/O and memory resources
3339  * @pdev: PCI device whose resources were previously reserved
3340  * @bars: Bitmask of BARs to be released
3341  *
3342  * Release selected PCI I/O and memory resources previously reserved.
3343  * Call this function only after all use of the PCI regions has ceased.
3344  */
3345 void pci_release_selected_regions(struct pci_dev *pdev, int bars)
3346 {
3347         int i;
3348
3349         for (i = 0; i < 6; i++)
3350                 if (bars & (1 << i))
3351                         pci_release_region(pdev, i);
3352 }
3353 EXPORT_SYMBOL(pci_release_selected_regions);
3354
3355 static int __pci_request_selected_regions(struct pci_dev *pdev, int bars,
3356                                           const char *res_name, int excl)
3357 {
3358         int i;
3359
3360         for (i = 0; i < 6; i++)
3361                 if (bars & (1 << i))
3362                         if (__pci_request_region(pdev, i, res_name, excl))
3363                                 goto err_out;
3364         return 0;
3365
3366 err_out:
3367         while (--i >= 0)
3368                 if (bars & (1 << i))
3369                         pci_release_region(pdev, i);
3370
3371         return -EBUSY;
3372 }
3373
3374
3375 /**
3376  * pci_request_selected_regions - Reserve selected PCI I/O and memory resources
3377  * @pdev: PCI device whose resources are to be reserved
3378  * @bars: Bitmask of BARs to be requested
3379  * @res_name: Name to be associated with resource
3380  */
3381 int pci_request_selected_regions(struct pci_dev *pdev, int bars,
3382                                  const char *res_name)
3383 {
3384         return __pci_request_selected_regions(pdev, bars, res_name, 0);
3385 }
3386 EXPORT_SYMBOL(pci_request_selected_regions);
3387
3388 int pci_request_selected_regions_exclusive(struct pci_dev *pdev, int bars,
3389                                            const char *res_name)
3390 {
3391         return __pci_request_selected_regions(pdev, bars, res_name,
3392                         IORESOURCE_EXCLUSIVE);
3393 }
3394 EXPORT_SYMBOL(pci_request_selected_regions_exclusive);
3395
3396 /**
3397  *      pci_release_regions - Release reserved PCI I/O and memory resources
3398  *      @pdev: PCI device whose resources were previously reserved by pci_request_regions
3399  *
3400  *      Releases all PCI I/O and memory resources previously reserved by a
3401  *      successful call to pci_request_regions.  Call this function only
3402  *      after all use of the PCI regions has ceased.
3403  */
3404
3405 void pci_release_regions(struct pci_dev *pdev)
3406 {
3407         pci_release_selected_regions(pdev, (1 << 6) - 1);
3408 }
3409 EXPORT_SYMBOL(pci_release_regions);
3410
3411 /**
3412  *      pci_request_regions - Reserved PCI I/O and memory resources
3413  *      @pdev: PCI device whose resources are to be reserved
3414  *      @res_name: Name to be associated with resource.
3415  *
3416  *      Mark all PCI regions associated with PCI device @pdev as
3417  *      being reserved by owner @res_name.  Do not access any
3418  *      address inside the PCI regions unless this call returns
3419  *      successfully.
3420  *
3421  *      Returns 0 on success, or %EBUSY on error.  A warning
3422  *      message is also printed on failure.
3423  */
3424 int pci_request_regions(struct pci_dev *pdev, const char *res_name)
3425 {
3426         return pci_request_selected_regions(pdev, ((1 << 6) - 1), res_name);
3427 }
3428 EXPORT_SYMBOL(pci_request_regions);
3429
3430 /**
3431  *      pci_request_regions_exclusive - Reserved PCI I/O and memory resources
3432  *      @pdev: PCI device whose resources are to be reserved
3433  *      @res_name: Name to be associated with resource.
3434  *
3435  *      Mark all PCI regions associated with PCI device @pdev as
3436  *      being reserved by owner @res_name.  Do not access any
3437  *      address inside the PCI regions unless this call returns
3438  *      successfully.
3439  *
3440  *      pci_request_regions_exclusive() will mark the region so that
3441  *      /dev/mem and the sysfs MMIO access will not be allowed.
3442  *
3443  *      Returns 0 on success, or %EBUSY on error.  A warning
3444  *      message is also printed on failure.
3445  */
3446 int pci_request_regions_exclusive(struct pci_dev *pdev, const char *res_name)
3447 {
3448         return pci_request_selected_regions_exclusive(pdev,
3449                                         ((1 << 6) - 1), res_name);
3450 }
3451 EXPORT_SYMBOL(pci_request_regions_exclusive);
3452
3453 /*
3454  * Record the PCI IO range (expressed as CPU physical address + size).
3455  * Return a negative value if an error has occured, zero otherwise
3456  */
3457 int pci_register_io_range(struct fwnode_handle *fwnode, phys_addr_t addr,
3458                         resource_size_t size)
3459 {
3460         int ret = 0;
3461 #ifdef PCI_IOBASE
3462         struct logic_pio_hwaddr *range;
3463
3464         if (!size || addr + size < addr)
3465                 return -EINVAL;
3466
3467         range = kzalloc(sizeof(*range), GFP_ATOMIC);
3468         if (!range)
3469                 return -ENOMEM;
3470
3471         range->fwnode = fwnode;
3472         range->size = size;
3473         range->hw_start = addr;
3474         range->flags = LOGIC_PIO_CPU_MMIO;
3475
3476         ret = logic_pio_register_range(range);
3477         if (ret)
3478                 kfree(range);
3479 #endif
3480
3481         return ret;
3482 }
3483
3484 phys_addr_t pci_pio_to_address(unsigned long pio)
3485 {
3486         phys_addr_t address = (phys_addr_t)OF_BAD_ADDR;
3487
3488 #ifdef PCI_IOBASE
3489         if (pio >= MMIO_UPPER_LIMIT)
3490                 return address;
3491
3492         address = logic_pio_to_hwaddr(pio);
3493 #endif
3494
3495         return address;
3496 }
3497
3498 unsigned long __weak pci_address_to_pio(phys_addr_t address)
3499 {
3500 #ifdef PCI_IOBASE
3501         return logic_pio_trans_cpuaddr(address);
3502 #else
3503         if (address > IO_SPACE_LIMIT)
3504                 return (unsigned long)-1;
3505
3506         return (unsigned long) address;
3507 #endif
3508 }
3509
3510 /**
3511  *      pci_remap_iospace - Remap the memory mapped I/O space
3512  *      @res: Resource describing the I/O space
3513  *      @phys_addr: physical address of range to be mapped
3514  *
3515  *      Remap the memory mapped I/O space described by the @res
3516  *      and the CPU physical address @phys_addr into virtual address space.
3517  *      Only architectures that have memory mapped IO functions defined
3518  *      (and the PCI_IOBASE value defined) should call this function.
3519  */
3520 int pci_remap_iospace(const struct resource *res, phys_addr_t phys_addr)
3521 {
3522 #if defined(PCI_IOBASE) && defined(CONFIG_MMU)
3523         unsigned long vaddr = (unsigned long)PCI_IOBASE + res->start;
3524
3525         if (!(res->flags & IORESOURCE_IO))
3526                 return -EINVAL;
3527
3528         if (res->end > IO_SPACE_LIMIT)
3529                 return -EINVAL;
3530
3531         return ioremap_page_range(vaddr, vaddr + resource_size(res), phys_addr,
3532                                   pgprot_device(PAGE_KERNEL));
3533 #else
3534         /* this architecture does not have memory mapped I/O space,
3535            so this function should never be called */
3536         WARN_ONCE(1, "This architecture does not support memory mapped I/O\n");
3537         return -ENODEV;
3538 #endif
3539 }
3540 EXPORT_SYMBOL(pci_remap_iospace);
3541
3542 /**
3543  *      pci_unmap_iospace - Unmap the memory mapped I/O space
3544  *      @res: resource to be unmapped
3545  *
3546  *      Unmap the CPU virtual address @res from virtual address space.
3547  *      Only architectures that have memory mapped IO functions defined
3548  *      (and the PCI_IOBASE value defined) should call this function.
3549  */
3550 void pci_unmap_iospace(struct resource *res)
3551 {
3552 #if defined(PCI_IOBASE) && defined(CONFIG_MMU)
3553         unsigned long vaddr = (unsigned long)PCI_IOBASE + res->start;
3554
3555         unmap_kernel_range(vaddr, resource_size(res));
3556 #endif
3557 }
3558 EXPORT_SYMBOL(pci_unmap_iospace);
3559
3560 /**
3561  * devm_pci_remap_cfgspace - Managed pci_remap_cfgspace()
3562  * @dev: Generic device to remap IO address for
3563  * @offset: Resource address to map
3564  * @size: Size of map
3565  *
3566  * Managed pci_remap_cfgspace().  Map is automatically unmapped on driver
3567  * detach.
3568  */
3569 void __iomem *devm_pci_remap_cfgspace(struct device *dev,
3570                                       resource_size_t offset,
3571                                       resource_size_t size)
3572 {
3573         void __iomem **ptr, *addr;
3574
3575         ptr = devres_alloc(devm_ioremap_release, sizeof(*ptr), GFP_KERNEL);
3576         if (!ptr)
3577                 return NULL;
3578
3579         addr = pci_remap_cfgspace(offset, size);
3580         if (addr) {
3581                 *ptr = addr;
3582                 devres_add(dev, ptr);
3583         } else
3584                 devres_free(ptr);
3585
3586         return addr;
3587 }
3588 EXPORT_SYMBOL(devm_pci_remap_cfgspace);
3589
3590 /**
3591  * devm_pci_remap_cfg_resource - check, request region and ioremap cfg resource
3592  * @dev: generic device to handle the resource for
3593  * @res: configuration space resource to be handled
3594  *
3595  * Checks that a resource is a valid memory region, requests the memory
3596  * region and ioremaps with pci_remap_cfgspace() API that ensures the
3597  * proper PCI configuration space memory attributes are guaranteed.
3598  *
3599  * All operations are managed and will be undone on driver detach.
3600  *
3601  * Returns a pointer to the remapped memory or an ERR_PTR() encoded error code
3602  * on failure. Usage example::
3603  *
3604  *      res = platform_get_resource(pdev, IORESOURCE_MEM, 0);
3605  *      base = devm_pci_remap_cfg_resource(&pdev->dev, res);
3606  *      if (IS_ERR(base))
3607  *              return PTR_ERR(base);
3608  */
3609 void __iomem *devm_pci_remap_cfg_resource(struct device *dev,
3610                                           struct resource *res)
3611 {
3612         resource_size_t size;
3613         const char *name;
3614         void __iomem *dest_ptr;
3615
3616         BUG_ON(!dev);
3617
3618         if (!res || resource_type(res) != IORESOURCE_MEM) {
3619                 dev_err(dev, "invalid resource\n");
3620                 return IOMEM_ERR_PTR(-EINVAL);
3621         }
3622
3623         size = resource_size(res);
3624         name = res->name ?: dev_name(dev);
3625
3626         if (!devm_request_mem_region(dev, res->start, size, name)) {
3627                 dev_err(dev, "can't request region for resource %pR\n", res);
3628                 return IOMEM_ERR_PTR(-EBUSY);
3629         }
3630
3631         dest_ptr = devm_pci_remap_cfgspace(dev, res->start, size);
3632         if (!dest_ptr) {
3633                 dev_err(dev, "ioremap failed for resource %pR\n", res);
3634                 devm_release_mem_region(dev, res->start, size);
3635                 dest_ptr = IOMEM_ERR_PTR(-ENOMEM);
3636         }
3637
3638         return dest_ptr;
3639 }
3640 EXPORT_SYMBOL(devm_pci_remap_cfg_resource);
3641
3642 static void __pci_set_master(struct pci_dev *dev, bool enable)
3643 {
3644         u16 old_cmd, cmd;
3645
3646         pci_read_config_word(dev, PCI_COMMAND, &old_cmd);
3647         if (enable)
3648                 cmd = old_cmd | PCI_COMMAND_MASTER;
3649         else
3650                 cmd = old_cmd & ~PCI_COMMAND_MASTER;
3651         if (cmd != old_cmd) {
3652                 pci_dbg(dev, "%s bus mastering\n",
3653                         enable ? "enabling" : "disabling");
3654                 pci_write_config_word(dev, PCI_COMMAND, cmd);
3655         }
3656         dev->is_busmaster = enable;
3657 }
3658
3659 /**
3660  * pcibios_setup - process "pci=" kernel boot arguments
3661  * @str: string used to pass in "pci=" kernel boot arguments
3662  *
3663  * Process kernel boot arguments.  This is the default implementation.
3664  * Architecture specific implementations can override this as necessary.
3665  */
3666 char * __weak __init pcibios_setup(char *str)
3667 {
3668         return str;