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