]> pilppa.org Git - linux-2.6-omap-h63xx.git/blob - drivers/pci/pci.c
PCI: restore saved SR-IOV state
[linux-2.6-omap-h63xx.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/kernel.h>
11 #include <linux/delay.h>
12 #include <linux/init.h>
13 #include <linux/pci.h>
14 #include <linux/pm.h>
15 #include <linux/module.h>
16 #include <linux/spinlock.h>
17 #include <linux/string.h>
18 #include <linux/log2.h>
19 #include <linux/pci-aspm.h>
20 #include <linux/pm_wakeup.h>
21 #include <linux/interrupt.h>
22 #include <asm/dma.h>    /* isa_dma_bridge_buggy */
23 #include <linux/device.h>
24 #include <asm/setup.h>
25 #include "pci.h"
26
27 unsigned int pci_pm_d3_delay = PCI_PM_D3_WAIT;
28
29 #ifdef CONFIG_PCI_DOMAINS
30 int pci_domains_supported = 1;
31 #endif
32
33 #define DEFAULT_CARDBUS_IO_SIZE         (256)
34 #define DEFAULT_CARDBUS_MEM_SIZE        (64*1024*1024)
35 /* pci=cbmemsize=nnM,cbiosize=nn can override this */
36 unsigned long pci_cardbus_io_size = DEFAULT_CARDBUS_IO_SIZE;
37 unsigned long pci_cardbus_mem_size = DEFAULT_CARDBUS_MEM_SIZE;
38
39 /**
40  * pci_bus_max_busnr - returns maximum PCI bus number of given bus' children
41  * @bus: pointer to PCI bus structure to search
42  *
43  * Given a PCI bus, returns the highest PCI bus number present in the set
44  * including the given PCI bus and its list of child PCI buses.
45  */
46 unsigned char pci_bus_max_busnr(struct pci_bus* bus)
47 {
48         struct list_head *tmp;
49         unsigned char max, n;
50
51         max = bus->subordinate;
52         list_for_each(tmp, &bus->children) {
53                 n = pci_bus_max_busnr(pci_bus_b(tmp));
54                 if(n > max)
55                         max = n;
56         }
57         return max;
58 }
59 EXPORT_SYMBOL_GPL(pci_bus_max_busnr);
60
61 #ifdef CONFIG_HAS_IOMEM
62 void __iomem *pci_ioremap_bar(struct pci_dev *pdev, int bar)
63 {
64         /*
65          * Make sure the BAR is actually a memory resource, not an IO resource
66          */
67         if (!(pci_resource_flags(pdev, bar) & IORESOURCE_MEM)) {
68                 WARN_ON(1);
69                 return NULL;
70         }
71         return ioremap_nocache(pci_resource_start(pdev, bar),
72                                      pci_resource_len(pdev, bar));
73 }
74 EXPORT_SYMBOL_GPL(pci_ioremap_bar);
75 #endif
76
77 #if 0
78 /**
79  * pci_max_busnr - returns maximum PCI bus number
80  *
81  * Returns the highest PCI bus number present in the system global list of
82  * PCI buses.
83  */
84 unsigned char __devinit
85 pci_max_busnr(void)
86 {
87         struct pci_bus *bus = NULL;
88         unsigned char max, n;
89
90         max = 0;
91         while ((bus = pci_find_next_bus(bus)) != NULL) {
92                 n = pci_bus_max_busnr(bus);
93                 if(n > max)
94                         max = n;
95         }
96         return max;
97 }
98
99 #endif  /*  0  */
100
101 #define PCI_FIND_CAP_TTL        48
102
103 static int __pci_find_next_cap_ttl(struct pci_bus *bus, unsigned int devfn,
104                                    u8 pos, int cap, int *ttl)
105 {
106         u8 id;
107
108         while ((*ttl)--) {
109                 pci_bus_read_config_byte(bus, devfn, pos, &pos);
110                 if (pos < 0x40)
111                         break;
112                 pos &= ~3;
113                 pci_bus_read_config_byte(bus, devfn, pos + PCI_CAP_LIST_ID,
114                                          &id);
115                 if (id == 0xff)
116                         break;
117                 if (id == cap)
118                         return pos;
119                 pos += PCI_CAP_LIST_NEXT;
120         }
121         return 0;
122 }
123
124 static int __pci_find_next_cap(struct pci_bus *bus, unsigned int devfn,
125                                u8 pos, int cap)
126 {
127         int ttl = PCI_FIND_CAP_TTL;
128
129         return __pci_find_next_cap_ttl(bus, devfn, pos, cap, &ttl);
130 }
131
132 int pci_find_next_capability(struct pci_dev *dev, u8 pos, int cap)
133 {
134         return __pci_find_next_cap(dev->bus, dev->devfn,
135                                    pos + PCI_CAP_LIST_NEXT, cap);
136 }
137 EXPORT_SYMBOL_GPL(pci_find_next_capability);
138
139 static int __pci_bus_find_cap_start(struct pci_bus *bus,
140                                     unsigned int devfn, u8 hdr_type)
141 {
142         u16 status;
143
144         pci_bus_read_config_word(bus, devfn, PCI_STATUS, &status);
145         if (!(status & PCI_STATUS_CAP_LIST))
146                 return 0;
147
148         switch (hdr_type) {
149         case PCI_HEADER_TYPE_NORMAL:
150         case PCI_HEADER_TYPE_BRIDGE:
151                 return PCI_CAPABILITY_LIST;
152         case PCI_HEADER_TYPE_CARDBUS:
153                 return PCI_CB_CAPABILITY_LIST;
154         default:
155                 return 0;
156         }
157
158         return 0;
159 }
160
161 /**
162  * pci_find_capability - query for devices' capabilities 
163  * @dev: PCI device to query
164  * @cap: capability code
165  *
166  * Tell if a device supports a given PCI capability.
167  * Returns the address of the requested capability structure within the
168  * device's PCI configuration space or 0 in case the device does not
169  * support it.  Possible values for @cap:
170  *
171  *  %PCI_CAP_ID_PM           Power Management 
172  *  %PCI_CAP_ID_AGP          Accelerated Graphics Port 
173  *  %PCI_CAP_ID_VPD          Vital Product Data 
174  *  %PCI_CAP_ID_SLOTID       Slot Identification 
175  *  %PCI_CAP_ID_MSI          Message Signalled Interrupts
176  *  %PCI_CAP_ID_CHSWP        CompactPCI HotSwap 
177  *  %PCI_CAP_ID_PCIX         PCI-X
178  *  %PCI_CAP_ID_EXP          PCI Express
179  */
180 int pci_find_capability(struct pci_dev *dev, int cap)
181 {
182         int pos;
183
184         pos = __pci_bus_find_cap_start(dev->bus, dev->devfn, dev->hdr_type);
185         if (pos)
186                 pos = __pci_find_next_cap(dev->bus, dev->devfn, pos, cap);
187
188         return pos;
189 }
190
191 /**
192  * pci_bus_find_capability - query for devices' capabilities 
193  * @bus:   the PCI bus to query
194  * @devfn: PCI device to query
195  * @cap:   capability code
196  *
197  * Like pci_find_capability() but works for pci devices that do not have a
198  * pci_dev structure set up yet. 
199  *
200  * Returns the address of the requested capability structure within the
201  * device's PCI configuration space or 0 in case the device does not
202  * support it.
203  */
204 int pci_bus_find_capability(struct pci_bus *bus, unsigned int devfn, int cap)
205 {
206         int pos;
207         u8 hdr_type;
208
209         pci_bus_read_config_byte(bus, devfn, PCI_HEADER_TYPE, &hdr_type);
210
211         pos = __pci_bus_find_cap_start(bus, devfn, hdr_type & 0x7f);
212         if (pos)
213                 pos = __pci_find_next_cap(bus, devfn, pos, cap);
214
215         return pos;
216 }
217
218 /**
219  * pci_find_ext_capability - Find an extended capability
220  * @dev: PCI device to query
221  * @cap: capability code
222  *
223  * Returns the address of the requested extended capability structure
224  * within the device's PCI configuration space or 0 if the device does
225  * not support it.  Possible values for @cap:
226  *
227  *  %PCI_EXT_CAP_ID_ERR         Advanced Error Reporting
228  *  %PCI_EXT_CAP_ID_VC          Virtual Channel
229  *  %PCI_EXT_CAP_ID_DSN         Device Serial Number
230  *  %PCI_EXT_CAP_ID_PWR         Power Budgeting
231  */
232 int pci_find_ext_capability(struct pci_dev *dev, int cap)
233 {
234         u32 header;
235         int ttl;
236         int pos = PCI_CFG_SPACE_SIZE;
237
238         /* minimum 8 bytes per capability */
239         ttl = (PCI_CFG_SPACE_EXP_SIZE - PCI_CFG_SPACE_SIZE) / 8;
240
241         if (dev->cfg_size <= PCI_CFG_SPACE_SIZE)
242                 return 0;
243
244         if (pci_read_config_dword(dev, pos, &header) != PCIBIOS_SUCCESSFUL)
245                 return 0;
246
247         /*
248          * If we have no capabilities, this is indicated by cap ID,
249          * cap version and next pointer all being 0.
250          */
251         if (header == 0)
252                 return 0;
253
254         while (ttl-- > 0) {
255                 if (PCI_EXT_CAP_ID(header) == cap)
256                         return pos;
257
258                 pos = PCI_EXT_CAP_NEXT(header);
259                 if (pos < PCI_CFG_SPACE_SIZE)
260                         break;
261
262                 if (pci_read_config_dword(dev, pos, &header) != PCIBIOS_SUCCESSFUL)
263                         break;
264         }
265
266         return 0;
267 }
268 EXPORT_SYMBOL_GPL(pci_find_ext_capability);
269
270 static int __pci_find_next_ht_cap(struct pci_dev *dev, int pos, int ht_cap)
271 {
272         int rc, ttl = PCI_FIND_CAP_TTL;
273         u8 cap, mask;
274
275         if (ht_cap == HT_CAPTYPE_SLAVE || ht_cap == HT_CAPTYPE_HOST)
276                 mask = HT_3BIT_CAP_MASK;
277         else
278                 mask = HT_5BIT_CAP_MASK;
279
280         pos = __pci_find_next_cap_ttl(dev->bus, dev->devfn, pos,
281                                       PCI_CAP_ID_HT, &ttl);
282         while (pos) {
283                 rc = pci_read_config_byte(dev, pos + 3, &cap);
284                 if (rc != PCIBIOS_SUCCESSFUL)
285                         return 0;
286
287                 if ((cap & mask) == ht_cap)
288                         return pos;
289
290                 pos = __pci_find_next_cap_ttl(dev->bus, dev->devfn,
291                                               pos + PCI_CAP_LIST_NEXT,
292                                               PCI_CAP_ID_HT, &ttl);
293         }
294
295         return 0;
296 }
297 /**
298  * pci_find_next_ht_capability - query a device's Hypertransport capabilities
299  * @dev: PCI device to query
300  * @pos: Position from which to continue searching
301  * @ht_cap: Hypertransport capability code
302  *
303  * To be used in conjunction with pci_find_ht_capability() to search for
304  * all capabilities matching @ht_cap. @pos should always be a value returned
305  * from pci_find_ht_capability().
306  *
307  * NB. To be 100% safe against broken PCI devices, the caller should take
308  * steps to avoid an infinite loop.
309  */
310 int pci_find_next_ht_capability(struct pci_dev *dev, int pos, int ht_cap)
311 {
312         return __pci_find_next_ht_cap(dev, pos + PCI_CAP_LIST_NEXT, ht_cap);
313 }
314 EXPORT_SYMBOL_GPL(pci_find_next_ht_capability);
315
316 /**
317  * pci_find_ht_capability - query a device's Hypertransport capabilities
318  * @dev: PCI device to query
319  * @ht_cap: Hypertransport capability code
320  *
321  * Tell if a device supports a given Hypertransport capability.
322  * Returns an address within the device's PCI configuration space
323  * or 0 in case the device does not support the request capability.
324  * The address points to the PCI capability, of type PCI_CAP_ID_HT,
325  * which has a Hypertransport capability matching @ht_cap.
326  */
327 int pci_find_ht_capability(struct pci_dev *dev, int ht_cap)
328 {
329         int pos;
330
331         pos = __pci_bus_find_cap_start(dev->bus, dev->devfn, dev->hdr_type);
332         if (pos)
333                 pos = __pci_find_next_ht_cap(dev, pos, ht_cap);
334
335         return pos;
336 }
337 EXPORT_SYMBOL_GPL(pci_find_ht_capability);
338
339 /**
340  * pci_find_parent_resource - return resource region of parent bus of given region
341  * @dev: PCI device structure contains resources to be searched
342  * @res: child resource record for which parent is sought
343  *
344  *  For given resource region of given device, return the resource
345  *  region of parent bus the given region is contained in or where
346  *  it should be allocated from.
347  */
348 struct resource *
349 pci_find_parent_resource(const struct pci_dev *dev, struct resource *res)
350 {
351         const struct pci_bus *bus = dev->bus;
352         int i;
353         struct resource *best = NULL;
354
355         for(i = 0; i < PCI_BUS_NUM_RESOURCES; i++) {
356                 struct resource *r = bus->resource[i];
357                 if (!r)
358                         continue;
359                 if (res->start && !(res->start >= r->start && res->end <= r->end))
360                         continue;       /* Not contained */
361                 if ((res->flags ^ r->flags) & (IORESOURCE_IO | IORESOURCE_MEM))
362                         continue;       /* Wrong type */
363                 if (!((res->flags ^ r->flags) & IORESOURCE_PREFETCH))
364                         return r;       /* Exact match */
365                 if ((res->flags & IORESOURCE_PREFETCH) && !(r->flags & IORESOURCE_PREFETCH))
366                         best = r;       /* Approximating prefetchable by non-prefetchable */
367         }
368         return best;
369 }
370
371 /**
372  * pci_restore_bars - restore a devices BAR values (e.g. after wake-up)
373  * @dev: PCI device to have its BARs restored
374  *
375  * Restore the BAR values for a given device, so as to make it
376  * accessible by its driver.
377  */
378 static void
379 pci_restore_bars(struct pci_dev *dev)
380 {
381         int i;
382
383         for (i = 0; i < PCI_BRIDGE_RESOURCES; i++)
384                 pci_update_resource(dev, i);
385 }
386
387 static struct pci_platform_pm_ops *pci_platform_pm;
388
389 int pci_set_platform_pm(struct pci_platform_pm_ops *ops)
390 {
391         if (!ops->is_manageable || !ops->set_state || !ops->choose_state
392             || !ops->sleep_wake || !ops->can_wakeup)
393                 return -EINVAL;
394         pci_platform_pm = ops;
395         return 0;
396 }
397
398 static inline bool platform_pci_power_manageable(struct pci_dev *dev)
399 {
400         return pci_platform_pm ? pci_platform_pm->is_manageable(dev) : false;
401 }
402
403 static inline int platform_pci_set_power_state(struct pci_dev *dev,
404                                                 pci_power_t t)
405 {
406         return pci_platform_pm ? pci_platform_pm->set_state(dev, t) : -ENOSYS;
407 }
408
409 static inline pci_power_t platform_pci_choose_state(struct pci_dev *dev)
410 {
411         return pci_platform_pm ?
412                         pci_platform_pm->choose_state(dev) : PCI_POWER_ERROR;
413 }
414
415 static inline bool platform_pci_can_wakeup(struct pci_dev *dev)
416 {
417         return pci_platform_pm ? pci_platform_pm->can_wakeup(dev) : false;
418 }
419
420 static inline int platform_pci_sleep_wake(struct pci_dev *dev, bool enable)
421 {
422         return pci_platform_pm ?
423                         pci_platform_pm->sleep_wake(dev, enable) : -ENODEV;
424 }
425
426 /**
427  * pci_raw_set_power_state - Use PCI PM registers to set the power state of
428  *                           given PCI device
429  * @dev: PCI device to handle.
430  * @state: PCI power state (D0, D1, D2, D3hot) to put the device into.
431  * @wait: If 'true', wait for the device to change its power state
432  *
433  * RETURN VALUE:
434  * -EINVAL if the requested state is invalid.
435  * -EIO if device does not support PCI PM or its PM capabilities register has a
436  * wrong version, or device doesn't support the requested state.
437  * 0 if device already is in the requested state.
438  * 0 if device's power state has been successfully changed.
439  */
440 static int
441 pci_raw_set_power_state(struct pci_dev *dev, pci_power_t state, bool wait)
442 {
443         u16 pmcsr;
444         bool need_restore = false;
445
446         if (!dev->pm_cap)
447                 return -EIO;
448
449         if (state < PCI_D0 || state > PCI_D3hot)
450                 return -EINVAL;
451
452         /* Validate current state:
453          * Can enter D0 from any state, but if we can only go deeper 
454          * to sleep if we're already in a low power state
455          */
456         if (dev->current_state == state) {
457                 /* we're already there */
458                 return 0;
459         } else if (state != PCI_D0 && dev->current_state <= PCI_D3cold
460             && dev->current_state > state) {
461                 dev_err(&dev->dev, "invalid power transition "
462                         "(from state %d to %d)\n", dev->current_state, state);
463                 return -EINVAL;
464         }
465
466         /* check if this device supports the desired state */
467         if ((state == PCI_D1 && !dev->d1_support)
468            || (state == PCI_D2 && !dev->d2_support))
469                 return -EIO;
470
471         pci_read_config_word(dev, dev->pm_cap + PCI_PM_CTRL, &pmcsr);
472
473         /* If we're (effectively) in D3, force entire word to 0.
474          * This doesn't affect PME_Status, disables PME_En, and
475          * sets PowerState to 0.
476          */
477         switch (dev->current_state) {
478         case PCI_D0:
479         case PCI_D1:
480         case PCI_D2:
481                 pmcsr &= ~PCI_PM_CTRL_STATE_MASK;
482                 pmcsr |= state;
483                 break;
484         case PCI_UNKNOWN: /* Boot-up */
485                 if ((pmcsr & PCI_PM_CTRL_STATE_MASK) == PCI_D3hot
486                  && !(pmcsr & PCI_PM_CTRL_NO_SOFT_RESET)) {
487                         need_restore = true;
488                         wait = true;
489                 }
490                 /* Fall-through: force to D0 */
491         default:
492                 pmcsr = 0;
493                 break;
494         }
495
496         /* enter specified state */
497         pci_write_config_word(dev, dev->pm_cap + PCI_PM_CTRL, pmcsr);
498
499         if (!wait)
500                 return 0;
501
502         /* Mandatory power management transition delays */
503         /* see PCI PM 1.1 5.6.1 table 18 */
504         if (state == PCI_D3hot || dev->current_state == PCI_D3hot)
505                 msleep(pci_pm_d3_delay);
506         else if (state == PCI_D2 || dev->current_state == PCI_D2)
507                 udelay(PCI_PM_D2_DELAY);
508
509         dev->current_state = state;
510
511         /* According to section 5.4.1 of the "PCI BUS POWER MANAGEMENT
512          * INTERFACE SPECIFICATION, REV. 1.2", a device transitioning
513          * from D3hot to D0 _may_ perform an internal reset, thereby
514          * going to "D0 Uninitialized" rather than "D0 Initialized".
515          * For example, at least some versions of the 3c905B and the
516          * 3c556B exhibit this behaviour.
517          *
518          * At least some laptop BIOSen (e.g. the Thinkpad T21) leave
519          * devices in a D3hot state at boot.  Consequently, we need to
520          * restore at least the BARs so that the device will be
521          * accessible to its driver.
522          */
523         if (need_restore)
524                 pci_restore_bars(dev);
525
526         if (wait && dev->bus->self)
527                 pcie_aspm_pm_state_change(dev->bus->self);
528
529         return 0;
530 }
531
532 /**
533  * pci_update_current_state - Read PCI power state of given device from its
534  *                            PCI PM registers and cache it
535  * @dev: PCI device to handle.
536  * @state: State to cache in case the device doesn't have the PM capability
537  */
538 void pci_update_current_state(struct pci_dev *dev, pci_power_t state)
539 {
540         if (dev->pm_cap) {
541                 u16 pmcsr;
542
543                 pci_read_config_word(dev, dev->pm_cap + PCI_PM_CTRL, &pmcsr);
544                 dev->current_state = (pmcsr & PCI_PM_CTRL_STATE_MASK);
545         } else {
546                 dev->current_state = state;
547         }
548 }
549
550 /**
551  * pci_set_power_state - Set the power state of a PCI device
552  * @dev: PCI device to handle.
553  * @state: PCI power state (D0, D1, D2, D3hot) to put the device into.
554  *
555  * Transition a device to a new power state, using the platform formware and/or
556  * the device's PCI PM registers.
557  *
558  * RETURN VALUE:
559  * -EINVAL if the requested state is invalid.
560  * -EIO if device does not support PCI PM or its PM capabilities register has a
561  * wrong version, or device doesn't support the requested state.
562  * 0 if device already is in the requested state.
563  * 0 if device's power state has been successfully changed.
564  */
565 int pci_set_power_state(struct pci_dev *dev, pci_power_t state)
566 {
567         int error;
568
569         /* bound the state we're entering */
570         if (state > PCI_D3hot)
571                 state = PCI_D3hot;
572         else if (state < PCI_D0)
573                 state = PCI_D0;
574         else if ((state == PCI_D1 || state == PCI_D2) && pci_no_d1d2(dev))
575                 /*
576                  * If the device or the parent bridge do not support PCI PM,
577                  * ignore the request if we're doing anything other than putting
578                  * it into D0 (which would only happen on boot).
579                  */
580                 return 0;
581
582         if (state == PCI_D0 && platform_pci_power_manageable(dev)) {
583                 /*
584                  * Allow the platform to change the state, for example via ACPI
585                  * _PR0, _PS0 and some such, but do not trust it.
586                  */
587                 int ret = platform_pci_set_power_state(dev, PCI_D0);
588                 if (!ret)
589                         pci_update_current_state(dev, PCI_D0);
590         }
591         /* This device is quirked not to be put into D3, so
592            don't put it in D3 */
593         if (state == PCI_D3hot && (dev->dev_flags & PCI_DEV_FLAGS_NO_D3))
594                 return 0;
595
596         error = pci_raw_set_power_state(dev, state, true);
597
598         if (state > PCI_D0 && platform_pci_power_manageable(dev)) {
599                 /* Allow the platform to finalize the transition */
600                 int ret = platform_pci_set_power_state(dev, state);
601                 if (!ret) {
602                         pci_update_current_state(dev, state);
603                         error = 0;
604                 }
605         }
606
607         return error;
608 }
609
610 /**
611  * pci_choose_state - Choose the power state of a PCI device
612  * @dev: PCI device to be suspended
613  * @state: target sleep state for the whole system. This is the value
614  *      that is passed to suspend() function.
615  *
616  * Returns PCI power state suitable for given device and given system
617  * message.
618  */
619
620 pci_power_t pci_choose_state(struct pci_dev *dev, pm_message_t state)
621 {
622         pci_power_t ret;
623
624         if (!pci_find_capability(dev, PCI_CAP_ID_PM))
625                 return PCI_D0;
626
627         ret = platform_pci_choose_state(dev);
628         if (ret != PCI_POWER_ERROR)
629                 return ret;
630
631         switch (state.event) {
632         case PM_EVENT_ON:
633                 return PCI_D0;
634         case PM_EVENT_FREEZE:
635         case PM_EVENT_PRETHAW:
636                 /* REVISIT both freeze and pre-thaw "should" use D0 */
637         case PM_EVENT_SUSPEND:
638         case PM_EVENT_HIBERNATE:
639                 return PCI_D3hot;
640         default:
641                 dev_info(&dev->dev, "unrecognized suspend event %d\n",
642                          state.event);
643                 BUG();
644         }
645         return PCI_D0;
646 }
647
648 EXPORT_SYMBOL(pci_choose_state);
649
650 static int pci_save_pcie_state(struct pci_dev *dev)
651 {
652         int pos, i = 0;
653         struct pci_cap_saved_state *save_state;
654         u16 *cap;
655
656         pos = pci_find_capability(dev, PCI_CAP_ID_EXP);
657         if (pos <= 0)
658                 return 0;
659
660         save_state = pci_find_saved_cap(dev, PCI_CAP_ID_EXP);
661         if (!save_state) {
662                 dev_err(&dev->dev, "buffer not found in %s\n", __func__);
663                 return -ENOMEM;
664         }
665         cap = (u16 *)&save_state->data[0];
666
667         pci_read_config_word(dev, pos + PCI_EXP_DEVCTL, &cap[i++]);
668         pci_read_config_word(dev, pos + PCI_EXP_LNKCTL, &cap[i++]);
669         pci_read_config_word(dev, pos + PCI_EXP_SLTCTL, &cap[i++]);
670         pci_read_config_word(dev, pos + PCI_EXP_RTCTL, &cap[i++]);
671
672         return 0;
673 }
674
675 static void pci_restore_pcie_state(struct pci_dev *dev)
676 {
677         int i = 0, pos;
678         struct pci_cap_saved_state *save_state;
679         u16 *cap;
680
681         save_state = pci_find_saved_cap(dev, PCI_CAP_ID_EXP);
682         pos = pci_find_capability(dev, PCI_CAP_ID_EXP);
683         if (!save_state || pos <= 0)
684                 return;
685         cap = (u16 *)&save_state->data[0];
686
687         pci_write_config_word(dev, pos + PCI_EXP_DEVCTL, cap[i++]);
688         pci_write_config_word(dev, pos + PCI_EXP_LNKCTL, cap[i++]);
689         pci_write_config_word(dev, pos + PCI_EXP_SLTCTL, cap[i++]);
690         pci_write_config_word(dev, pos + PCI_EXP_RTCTL, cap[i++]);
691 }
692
693
694 static int pci_save_pcix_state(struct pci_dev *dev)
695 {
696         int pos;
697         struct pci_cap_saved_state *save_state;
698
699         pos = pci_find_capability(dev, PCI_CAP_ID_PCIX);
700         if (pos <= 0)
701                 return 0;
702
703         save_state = pci_find_saved_cap(dev, PCI_CAP_ID_PCIX);
704         if (!save_state) {
705                 dev_err(&dev->dev, "buffer not found in %s\n", __func__);
706                 return -ENOMEM;
707         }
708
709         pci_read_config_word(dev, pos + PCI_X_CMD, (u16 *)save_state->data);
710
711         return 0;
712 }
713
714 static void pci_restore_pcix_state(struct pci_dev *dev)
715 {
716         int i = 0, pos;
717         struct pci_cap_saved_state *save_state;
718         u16 *cap;
719
720         save_state = pci_find_saved_cap(dev, PCI_CAP_ID_PCIX);
721         pos = pci_find_capability(dev, PCI_CAP_ID_PCIX);
722         if (!save_state || pos <= 0)
723                 return;
724         cap = (u16 *)&save_state->data[0];
725
726         pci_write_config_word(dev, pos + PCI_X_CMD, cap[i++]);
727 }
728
729
730 /**
731  * pci_save_state - save the PCI configuration space of a device before suspending
732  * @dev: - PCI device that we're dealing with
733  */
734 int
735 pci_save_state(struct pci_dev *dev)
736 {
737         int i;
738         /* XXX: 100% dword access ok here? */
739         for (i = 0; i < 16; i++)
740                 pci_read_config_dword(dev, i * 4,&dev->saved_config_space[i]);
741         dev->state_saved = true;
742         if ((i = pci_save_pcie_state(dev)) != 0)
743                 return i;
744         if ((i = pci_save_pcix_state(dev)) != 0)
745                 return i;
746         return 0;
747 }
748
749 /** 
750  * pci_restore_state - Restore the saved state of a PCI device
751  * @dev: - PCI device that we're dealing with
752  */
753 int 
754 pci_restore_state(struct pci_dev *dev)
755 {
756         int i;
757         u32 val;
758
759         /* PCI Express register must be restored first */
760         pci_restore_pcie_state(dev);
761
762         /*
763          * The Base Address register should be programmed before the command
764          * register(s)
765          */
766         for (i = 15; i >= 0; i--) {
767                 pci_read_config_dword(dev, i * 4, &val);
768                 if (val != dev->saved_config_space[i]) {
769                         dev_printk(KERN_DEBUG, &dev->dev, "restoring config "
770                                 "space at offset %#x (was %#x, writing %#x)\n",
771                                 i, val, (int)dev->saved_config_space[i]);
772                         pci_write_config_dword(dev,i * 4,
773                                 dev->saved_config_space[i]);
774                 }
775         }
776         pci_restore_pcix_state(dev);
777         pci_restore_msi_state(dev);
778         pci_restore_iov_state(dev);
779
780         return 0;
781 }
782
783 static int do_pci_enable_device(struct pci_dev *dev, int bars)
784 {
785         int err;
786
787         err = pci_set_power_state(dev, PCI_D0);
788         if (err < 0 && err != -EIO)
789                 return err;
790         err = pcibios_enable_device(dev, bars);
791         if (err < 0)
792                 return err;
793         pci_fixup_device(pci_fixup_enable, dev);
794
795         return 0;
796 }
797
798 /**
799  * pci_reenable_device - Resume abandoned device
800  * @dev: PCI device to be resumed
801  *
802  *  Note this function is a backend of pci_default_resume and is not supposed
803  *  to be called by normal code, write proper resume handler and use it instead.
804  */
805 int pci_reenable_device(struct pci_dev *dev)
806 {
807         if (atomic_read(&dev->enable_cnt))
808                 return do_pci_enable_device(dev, (1 << PCI_NUM_RESOURCES) - 1);
809         return 0;
810 }
811
812 static int __pci_enable_device_flags(struct pci_dev *dev,
813                                      resource_size_t flags)
814 {
815         int err;
816         int i, bars = 0;
817
818         if (atomic_add_return(1, &dev->enable_cnt) > 1)
819                 return 0;               /* already enabled */
820
821         for (i = 0; i < DEVICE_COUNT_RESOURCE; i++)
822                 if (dev->resource[i].flags & flags)
823                         bars |= (1 << i);
824
825         err = do_pci_enable_device(dev, bars);
826         if (err < 0)
827                 atomic_dec(&dev->enable_cnt);
828         return err;
829 }
830
831 /**
832  * pci_enable_device_io - Initialize a device for use with IO space
833  * @dev: PCI device to be initialized
834  *
835  *  Initialize device before it's used by a driver. Ask low-level code
836  *  to enable I/O resources. Wake up the device if it was suspended.
837  *  Beware, this function can fail.
838  */
839 int pci_enable_device_io(struct pci_dev *dev)
840 {
841         return __pci_enable_device_flags(dev, IORESOURCE_IO);
842 }
843
844 /**
845  * pci_enable_device_mem - Initialize a device for use with Memory space
846  * @dev: PCI device to be initialized
847  *
848  *  Initialize device before it's used by a driver. Ask low-level code
849  *  to enable Memory resources. Wake up the device if it was suspended.
850  *  Beware, this function can fail.
851  */
852 int pci_enable_device_mem(struct pci_dev *dev)
853 {
854         return __pci_enable_device_flags(dev, IORESOURCE_MEM);
855 }
856
857 /**
858  * pci_enable_device - Initialize device before it's used by a driver.
859  * @dev: PCI device to be initialized
860  *
861  *  Initialize device before it's used by a driver. Ask low-level code
862  *  to enable I/O and memory. Wake up the device if it was suspended.
863  *  Beware, this function can fail.
864  *
865  *  Note we don't actually enable the device many times if we call
866  *  this function repeatedly (we just increment the count).
867  */
868 int pci_enable_device(struct pci_dev *dev)
869 {
870         return __pci_enable_device_flags(dev, IORESOURCE_MEM | IORESOURCE_IO);
871 }
872
873 /*
874  * Managed PCI resources.  This manages device on/off, intx/msi/msix
875  * on/off and BAR regions.  pci_dev itself records msi/msix status, so
876  * there's no need to track it separately.  pci_devres is initialized
877  * when a device is enabled using managed PCI device enable interface.
878  */
879 struct pci_devres {
880         unsigned int enabled:1;
881         unsigned int pinned:1;
882         unsigned int orig_intx:1;
883         unsigned int restore_intx:1;
884         u32 region_mask;
885 };
886
887 static void pcim_release(struct device *gendev, void *res)
888 {
889         struct pci_dev *dev = container_of(gendev, struct pci_dev, dev);
890         struct pci_devres *this = res;
891         int i;
892
893         if (dev->msi_enabled)
894                 pci_disable_msi(dev);
895         if (dev->msix_enabled)
896                 pci_disable_msix(dev);
897
898         for (i = 0; i < DEVICE_COUNT_RESOURCE; i++)
899                 if (this->region_mask & (1 << i))
900                         pci_release_region(dev, i);
901
902         if (this->restore_intx)
903                 pci_intx(dev, this->orig_intx);
904
905         if (this->enabled && !this->pinned)
906                 pci_disable_device(dev);
907 }
908
909 static struct pci_devres * get_pci_dr(struct pci_dev *pdev)
910 {
911         struct pci_devres *dr, *new_dr;
912
913         dr = devres_find(&pdev->dev, pcim_release, NULL, NULL);
914         if (dr)
915                 return dr;
916
917         new_dr = devres_alloc(pcim_release, sizeof(*new_dr), GFP_KERNEL);
918         if (!new_dr)
919                 return NULL;
920         return devres_get(&pdev->dev, new_dr, NULL, NULL);
921 }
922
923 static struct pci_devres * find_pci_dr(struct pci_dev *pdev)
924 {
925         if (pci_is_managed(pdev))
926                 return devres_find(&pdev->dev, pcim_release, NULL, NULL);
927         return NULL;
928 }
929
930 /**
931  * pcim_enable_device - Managed pci_enable_device()
932  * @pdev: PCI device to be initialized
933  *
934  * Managed pci_enable_device().
935  */
936 int pcim_enable_device(struct pci_dev *pdev)
937 {
938         struct pci_devres *dr;
939         int rc;
940
941         dr = get_pci_dr(pdev);
942         if (unlikely(!dr))
943                 return -ENOMEM;
944         if (dr->enabled)
945                 return 0;
946
947         rc = pci_enable_device(pdev);
948         if (!rc) {
949                 pdev->is_managed = 1;
950                 dr->enabled = 1;
951         }
952         return rc;
953 }
954
955 /**
956  * pcim_pin_device - Pin managed PCI device
957  * @pdev: PCI device to pin
958  *
959  * Pin managed PCI device @pdev.  Pinned device won't be disabled on
960  * driver detach.  @pdev must have been enabled with
961  * pcim_enable_device().
962  */
963 void pcim_pin_device(struct pci_dev *pdev)
964 {
965         struct pci_devres *dr;
966
967         dr = find_pci_dr(pdev);
968         WARN_ON(!dr || !dr->enabled);
969         if (dr)
970                 dr->pinned = 1;
971 }
972
973 /**
974  * pcibios_disable_device - disable arch specific PCI resources for device dev
975  * @dev: the PCI device to disable
976  *
977  * Disables architecture specific PCI resources for the device. This
978  * is the default implementation. Architecture implementations can
979  * override this.
980  */
981 void __attribute__ ((weak)) pcibios_disable_device (struct pci_dev *dev) {}
982
983 static void do_pci_disable_device(struct pci_dev *dev)
984 {
985         u16 pci_command;
986
987         pci_read_config_word(dev, PCI_COMMAND, &pci_command);
988         if (pci_command & PCI_COMMAND_MASTER) {
989                 pci_command &= ~PCI_COMMAND_MASTER;
990                 pci_write_config_word(dev, PCI_COMMAND, pci_command);
991         }
992
993         pcibios_disable_device(dev);
994 }
995
996 /**
997  * pci_disable_enabled_device - Disable device without updating enable_cnt
998  * @dev: PCI device to disable
999  *
1000  * NOTE: This function is a backend of PCI power management routines and is
1001  * not supposed to be called drivers.
1002  */
1003 void pci_disable_enabled_device(struct pci_dev *dev)
1004 {
1005         if (atomic_read(&dev->enable_cnt))
1006                 do_pci_disable_device(dev);
1007 }
1008
1009 /**
1010  * pci_disable_device - Disable PCI device after use
1011  * @dev: PCI device to be disabled
1012  *
1013  * Signal to the system that the PCI device is not in use by the system
1014  * anymore.  This only involves disabling PCI bus-mastering, if active.
1015  *
1016  * Note we don't actually disable the device until all callers of
1017  * pci_device_enable() have called pci_device_disable().
1018  */
1019 void
1020 pci_disable_device(struct pci_dev *dev)
1021 {
1022         struct pci_devres *dr;
1023
1024         dr = find_pci_dr(dev);
1025         if (dr)
1026                 dr->enabled = 0;
1027
1028         if (atomic_sub_return(1, &dev->enable_cnt) != 0)
1029                 return;
1030
1031         do_pci_disable_device(dev);
1032
1033         dev->is_busmaster = 0;
1034 }
1035
1036 /**
1037  * pcibios_set_pcie_reset_state - set reset state for device dev
1038  * @dev: the PCI-E device reset
1039  * @state: Reset state to enter into
1040  *
1041  *
1042  * Sets the PCI-E reset state for the device. This is the default
1043  * implementation. Architecture implementations can override this.
1044  */
1045 int __attribute__ ((weak)) pcibios_set_pcie_reset_state(struct pci_dev *dev,
1046                                                         enum pcie_reset_state state)
1047 {
1048         return -EINVAL;
1049 }
1050
1051 /**
1052  * pci_set_pcie_reset_state - set reset state for device dev
1053  * @dev: the PCI-E device reset
1054  * @state: Reset state to enter into
1055  *
1056  *
1057  * Sets the PCI reset state for the device.
1058  */
1059 int pci_set_pcie_reset_state(struct pci_dev *dev, enum pcie_reset_state state)
1060 {
1061         return pcibios_set_pcie_reset_state(dev, state);
1062 }
1063
1064 /**
1065  * pci_pme_capable - check the capability of PCI device to generate PME#
1066  * @dev: PCI device to handle.
1067  * @state: PCI state from which device will issue PME#.
1068  */
1069 bool pci_pme_capable(struct pci_dev *dev, pci_power_t state)
1070 {
1071         if (!dev->pm_cap)
1072                 return false;
1073
1074         return !!(dev->pme_support & (1 << state));
1075 }
1076
1077 /**
1078  * pci_pme_active - enable or disable PCI device's PME# function
1079  * @dev: PCI device to handle.
1080  * @enable: 'true' to enable PME# generation; 'false' to disable it.
1081  *
1082  * The caller must verify that the device is capable of generating PME# before
1083  * calling this function with @enable equal to 'true'.
1084  */
1085 void pci_pme_active(struct pci_dev *dev, bool enable)
1086 {
1087         u16 pmcsr;
1088
1089         if (!dev->pm_cap)
1090                 return;
1091
1092         pci_read_config_word(dev, dev->pm_cap + PCI_PM_CTRL, &pmcsr);
1093         /* Clear PME_Status by writing 1 to it and enable PME# */
1094         pmcsr |= PCI_PM_CTRL_PME_STATUS | PCI_PM_CTRL_PME_ENABLE;
1095         if (!enable)
1096                 pmcsr &= ~PCI_PM_CTRL_PME_ENABLE;
1097
1098         pci_write_config_word(dev, dev->pm_cap + PCI_PM_CTRL, pmcsr);
1099
1100         dev_printk(KERN_INFO, &dev->dev, "PME# %s\n",
1101                         enable ? "enabled" : "disabled");
1102 }
1103
1104 /**
1105  * pci_enable_wake - enable PCI device as wakeup event source
1106  * @dev: PCI device affected
1107  * @state: PCI state from which device will issue wakeup events
1108  * @enable: True to enable event generation; false to disable
1109  *
1110  * This enables the device as a wakeup event source, or disables it.
1111  * When such events involves platform-specific hooks, those hooks are
1112  * called automatically by this routine.
1113  *
1114  * Devices with legacy power management (no standard PCI PM capabilities)
1115  * always require such platform hooks.
1116  *
1117  * RETURN VALUE:
1118  * 0 is returned on success
1119  * -EINVAL is returned if device is not supposed to wake up the system
1120  * Error code depending on the platform is returned if both the platform and
1121  * the native mechanism fail to enable the generation of wake-up events
1122  */
1123 int pci_enable_wake(struct pci_dev *dev, pci_power_t state, int enable)
1124 {
1125         int error = 0;
1126         bool pme_done = false;
1127
1128         if (enable && !device_may_wakeup(&dev->dev))
1129                 return -EINVAL;
1130
1131         /*
1132          * According to "PCI System Architecture" 4th ed. by Tom Shanley & Don
1133          * Anderson we should be doing PME# wake enable followed by ACPI wake
1134          * enable.  To disable wake-up we call the platform first, for symmetry.
1135          */
1136
1137         if (!enable && platform_pci_can_wakeup(dev))
1138                 error = platform_pci_sleep_wake(dev, false);
1139
1140         if (!enable || pci_pme_capable(dev, state)) {
1141                 pci_pme_active(dev, enable);
1142                 pme_done = true;
1143         }
1144
1145         if (enable && platform_pci_can_wakeup(dev))
1146                 error = platform_pci_sleep_wake(dev, true);
1147
1148         return pme_done ? 0 : error;
1149 }
1150
1151 /**
1152  * pci_wake_from_d3 - enable/disable device to wake up from D3_hot or D3_cold
1153  * @dev: PCI device to prepare
1154  * @enable: True to enable wake-up event generation; false to disable
1155  *
1156  * Many drivers want the device to wake up the system from D3_hot or D3_cold
1157  * and this function allows them to set that up cleanly - pci_enable_wake()
1158  * should not be called twice in a row to enable wake-up due to PCI PM vs ACPI
1159  * ordering constraints.
1160  *
1161  * This function only returns error code if the device is not capable of
1162  * generating PME# from both D3_hot and D3_cold, and the platform is unable to
1163  * enable wake-up power for it.
1164  */
1165 int pci_wake_from_d3(struct pci_dev *dev, bool enable)
1166 {
1167         return pci_pme_capable(dev, PCI_D3cold) ?
1168                         pci_enable_wake(dev, PCI_D3cold, enable) :
1169                         pci_enable_wake(dev, PCI_D3hot, enable);
1170 }
1171
1172 /**
1173  * pci_target_state - find an appropriate low power state for a given PCI dev
1174  * @dev: PCI device
1175  *
1176  * Use underlying platform code to find a supported low power state for @dev.
1177  * If the platform can't manage @dev, return the deepest state from which it
1178  * can generate wake events, based on any available PME info.
1179  */
1180 pci_power_t pci_target_state(struct pci_dev *dev)
1181 {
1182         pci_power_t target_state = PCI_D3hot;
1183
1184         if (platform_pci_power_manageable(dev)) {
1185                 /*
1186                  * Call the platform to choose the target state of the device
1187                  * and enable wake-up from this state if supported.
1188                  */
1189                 pci_power_t state = platform_pci_choose_state(dev);
1190
1191                 switch (state) {
1192                 case PCI_POWER_ERROR:
1193                 case PCI_UNKNOWN:
1194                         break;
1195                 case PCI_D1:
1196                 case PCI_D2:
1197                         if (pci_no_d1d2(dev))
1198                                 break;
1199                 default:
1200                         target_state = state;
1201                 }
1202         } else if (device_may_wakeup(&dev->dev)) {
1203                 /*
1204                  * Find the deepest state from which the device can generate
1205                  * wake-up events, make it the target state and enable device
1206                  * to generate PME#.
1207                  */
1208                 if (!dev->pm_cap)
1209                         return PCI_POWER_ERROR;
1210
1211                 if (dev->pme_support) {
1212                         while (target_state
1213                               && !(dev->pme_support & (1 << target_state)))
1214                                 target_state--;
1215                 }
1216         }
1217
1218         return target_state;
1219 }
1220
1221 /**
1222  * pci_prepare_to_sleep - prepare PCI device for system-wide transition into a sleep state
1223  * @dev: Device to handle.
1224  *
1225  * Choose the power state appropriate for the device depending on whether
1226  * it can wake up the system and/or is power manageable by the platform
1227  * (PCI_D3hot is the default) and put the device into that state.
1228  */
1229 int pci_prepare_to_sleep(struct pci_dev *dev)
1230 {
1231         pci_power_t target_state = pci_target_state(dev);
1232         int error;
1233
1234         if (target_state == PCI_POWER_ERROR)
1235                 return -EIO;
1236
1237         pci_enable_wake(dev, target_state, true);
1238
1239         error = pci_set_power_state(dev, target_state);
1240
1241         if (error)
1242                 pci_enable_wake(dev, target_state, false);
1243
1244         return error;
1245 }
1246
1247 /**
1248  * pci_back_from_sleep - turn PCI device on during system-wide transition into working state
1249  * @dev: Device to handle.
1250  *
1251  * Disable device's sytem wake-up capability and put it into D0.
1252  */
1253 int pci_back_from_sleep(struct pci_dev *dev)
1254 {
1255         pci_enable_wake(dev, PCI_D0, false);
1256         return pci_set_power_state(dev, PCI_D0);
1257 }
1258
1259 /**
1260  * pci_pm_init - Initialize PM functions of given PCI device
1261  * @dev: PCI device to handle.
1262  */
1263 void pci_pm_init(struct pci_dev *dev)
1264 {
1265         int pm;
1266         u16 pmc;
1267
1268         dev->pm_cap = 0;
1269
1270         /* find PCI PM capability in list */
1271         pm = pci_find_capability(dev, PCI_CAP_ID_PM);
1272         if (!pm)
1273                 return;
1274         /* Check device's ability to generate PME# */
1275         pci_read_config_word(dev, pm + PCI_PM_PMC, &pmc);
1276
1277         if ((pmc & PCI_PM_CAP_VER_MASK) > 3) {
1278                 dev_err(&dev->dev, "unsupported PM cap regs version (%u)\n",
1279                         pmc & PCI_PM_CAP_VER_MASK);
1280                 return;
1281         }
1282
1283         dev->pm_cap = pm;
1284
1285         dev->d1_support = false;
1286         dev->d2_support = false;
1287         if (!pci_no_d1d2(dev)) {
1288                 if (pmc & PCI_PM_CAP_D1)
1289                         dev->d1_support = true;
1290                 if (pmc & PCI_PM_CAP_D2)
1291                         dev->d2_support = true;
1292
1293                 if (dev->d1_support || dev->d2_support)
1294                         dev_printk(KERN_DEBUG, &dev->dev, "supports%s%s\n",
1295                                    dev->d1_support ? " D1" : "",
1296                                    dev->d2_support ? " D2" : "");
1297         }
1298
1299         pmc &= PCI_PM_CAP_PME_MASK;
1300         if (pmc) {
1301                 dev_info(&dev->dev, "PME# supported from%s%s%s%s%s\n",
1302                          (pmc & PCI_PM_CAP_PME_D0) ? " D0" : "",
1303                          (pmc & PCI_PM_CAP_PME_D1) ? " D1" : "",
1304                          (pmc & PCI_PM_CAP_PME_D2) ? " D2" : "",
1305                          (pmc & PCI_PM_CAP_PME_D3) ? " D3hot" : "",
1306                          (pmc & PCI_PM_CAP_PME_D3cold) ? " D3cold" : "");
1307                 dev->pme_support = pmc >> PCI_PM_CAP_PME_SHIFT;
1308                 /*
1309                  * Make device's PM flags reflect the wake-up capability, but
1310                  * let the user space enable it to wake up the system as needed.
1311                  */
1312                 device_set_wakeup_capable(&dev->dev, true);
1313                 device_set_wakeup_enable(&dev->dev, false);
1314                 /* Disable the PME# generation functionality */
1315                 pci_pme_active(dev, false);
1316         } else {
1317                 dev->pme_support = 0;
1318         }
1319 }
1320
1321 /**
1322  * platform_pci_wakeup_init - init platform wakeup if present
1323  * @dev: PCI device
1324  *
1325  * Some devices don't have PCI PM caps but can still generate wakeup
1326  * events through platform methods (like ACPI events).  If @dev supports
1327  * platform wakeup events, set the device flag to indicate as much.  This
1328  * may be redundant if the device also supports PCI PM caps, but double
1329  * initialization should be safe in that case.
1330  */
1331 void platform_pci_wakeup_init(struct pci_dev *dev)
1332 {
1333         if (!platform_pci_can_wakeup(dev))
1334                 return;
1335
1336         device_set_wakeup_capable(&dev->dev, true);
1337         device_set_wakeup_enable(&dev->dev, false);
1338         platform_pci_sleep_wake(dev, false);
1339 }
1340
1341 /**
1342  * pci_add_save_buffer - allocate buffer for saving given capability registers
1343  * @dev: the PCI device
1344  * @cap: the capability to allocate the buffer for
1345  * @size: requested size of the buffer
1346  */
1347 static int pci_add_cap_save_buffer(
1348         struct pci_dev *dev, char cap, unsigned int size)
1349 {
1350         int pos;
1351         struct pci_cap_saved_state *save_state;
1352
1353         pos = pci_find_capability(dev, cap);
1354         if (pos <= 0)
1355                 return 0;
1356
1357         save_state = kzalloc(sizeof(*save_state) + size, GFP_KERNEL);
1358         if (!save_state)
1359                 return -ENOMEM;
1360
1361         save_state->cap_nr = cap;
1362         pci_add_saved_cap(dev, save_state);
1363
1364         return 0;
1365 }
1366
1367 /**
1368  * pci_allocate_cap_save_buffers - allocate buffers for saving capabilities
1369  * @dev: the PCI device
1370  */
1371 void pci_allocate_cap_save_buffers(struct pci_dev *dev)
1372 {
1373         int error;
1374
1375         error = pci_add_cap_save_buffer(dev, PCI_CAP_ID_EXP, 4 * sizeof(u16));
1376         if (error)
1377                 dev_err(&dev->dev,
1378                         "unable to preallocate PCI Express save buffer\n");
1379
1380         error = pci_add_cap_save_buffer(dev, PCI_CAP_ID_PCIX, sizeof(u16));
1381         if (error)
1382                 dev_err(&dev->dev,
1383                         "unable to preallocate PCI-X save buffer\n");
1384 }
1385
1386 /**
1387  * pci_restore_standard_config - restore standard config registers of PCI device
1388  * @dev: PCI device to handle
1389  *
1390  * This function assumes that the device's configuration space is accessible.
1391  * If the device needs to be powered up, the function will wait for it to
1392  * change the state.
1393  */
1394 int pci_restore_standard_config(struct pci_dev *dev)
1395 {
1396         pci_power_t prev_state;
1397         int error;
1398
1399         pci_update_current_state(dev, PCI_D0);
1400
1401         prev_state = dev->current_state;
1402         if (prev_state == PCI_D0)
1403                 goto Restore;
1404
1405         error = pci_raw_set_power_state(dev, PCI_D0, false);
1406         if (error)
1407                 return error;
1408
1409         /*
1410          * This assumes that we won't get a bus in B2 or B3 from the BIOS, but
1411          * we've made this assumption forever and it appears to be universally
1412          * satisfied.
1413          */
1414         switch(prev_state) {
1415         case PCI_D3cold:
1416         case PCI_D3hot:
1417                 mdelay(pci_pm_d3_delay);
1418                 break;
1419         case PCI_D2:
1420                 udelay(PCI_PM_D2_DELAY);
1421                 break;
1422         }
1423
1424         pci_update_current_state(dev, PCI_D0);
1425
1426  Restore:
1427         return dev->state_saved ? pci_restore_state(dev) : 0;
1428 }
1429
1430 /**
1431  * pci_enable_ari - enable ARI forwarding if hardware support it
1432  * @dev: the PCI device
1433  */
1434 void pci_enable_ari(struct pci_dev *dev)
1435 {
1436         int pos;
1437         u32 cap;
1438         u16 ctrl;
1439         struct pci_dev *bridge;
1440
1441         if (!dev->is_pcie || dev->devfn)
1442                 return;
1443
1444         pos = pci_find_ext_capability(dev, PCI_EXT_CAP_ID_ARI);
1445         if (!pos)
1446                 return;
1447
1448         bridge = dev->bus->self;
1449         if (!bridge || !bridge->is_pcie)
1450                 return;
1451
1452         pos = pci_find_capability(bridge, PCI_CAP_ID_EXP);
1453         if (!pos)
1454                 return;
1455
1456         pci_read_config_dword(bridge, pos + PCI_EXP_DEVCAP2, &cap);
1457         if (!(cap & PCI_EXP_DEVCAP2_ARI))
1458                 return;
1459
1460         pci_read_config_word(bridge, pos + PCI_EXP_DEVCTL2, &ctrl);
1461         ctrl |= PCI_EXP_DEVCTL2_ARI;
1462         pci_write_config_word(bridge, pos + PCI_EXP_DEVCTL2, ctrl);
1463
1464         bridge->ari_enabled = 1;
1465 }
1466
1467 /**
1468  * pci_swizzle_interrupt_pin - swizzle INTx for device behind bridge
1469  * @dev: the PCI device
1470  * @pin: the INTx pin (1=INTA, 2=INTB, 3=INTD, 4=INTD)
1471  *
1472  * Perform INTx swizzling for a device behind one level of bridge.  This is
1473  * required by section 9.1 of the PCI-to-PCI bridge specification for devices
1474  * behind bridges on add-in cards.
1475  */
1476 u8 pci_swizzle_interrupt_pin(struct pci_dev *dev, u8 pin)
1477 {
1478         return (((pin - 1) + PCI_SLOT(dev->devfn)) % 4) + 1;
1479 }
1480
1481 int
1482 pci_get_interrupt_pin(struct pci_dev *dev, struct pci_dev **bridge)
1483 {
1484         u8 pin;
1485
1486         pin = dev->pin;
1487         if (!pin)
1488                 return -1;
1489
1490         while (dev->bus->parent) {
1491                 pin = pci_swizzle_interrupt_pin(dev, pin);
1492                 dev = dev->bus->self;
1493         }
1494         *bridge = dev;
1495         return pin;
1496 }
1497
1498 /**
1499  * pci_common_swizzle - swizzle INTx all the way to root bridge
1500  * @dev: the PCI device
1501  * @pinp: pointer to the INTx pin value (1=INTA, 2=INTB, 3=INTD, 4=INTD)
1502  *
1503  * Perform INTx swizzling for a device.  This traverses through all PCI-to-PCI
1504  * bridges all the way up to a PCI root bus.
1505  */
1506 u8 pci_common_swizzle(struct pci_dev *dev, u8 *pinp)
1507 {
1508         u8 pin = *pinp;
1509
1510         while (dev->bus->parent) {
1511                 pin = pci_swizzle_interrupt_pin(dev, pin);
1512                 dev = dev->bus->self;
1513         }
1514         *pinp = pin;
1515         return PCI_SLOT(dev->devfn);
1516 }
1517
1518 /**
1519  *      pci_release_region - Release a PCI bar
1520  *      @pdev: PCI device whose resources were previously reserved by pci_request_region
1521  *      @bar: BAR to release
1522  *
1523  *      Releases the PCI I/O and memory resources previously reserved by a
1524  *      successful call to pci_request_region.  Call this function only
1525  *      after all use of the PCI regions has ceased.
1526  */
1527 void pci_release_region(struct pci_dev *pdev, int bar)
1528 {
1529         struct pci_devres *dr;
1530
1531         if (pci_resource_len(pdev, bar) == 0)
1532                 return;
1533         if (pci_resource_flags(pdev, bar) & IORESOURCE_IO)
1534                 release_region(pci_resource_start(pdev, bar),
1535                                 pci_resource_len(pdev, bar));
1536         else if (pci_resource_flags(pdev, bar) & IORESOURCE_MEM)
1537                 release_mem_region(pci_resource_start(pdev, bar),
1538                                 pci_resource_len(pdev, bar));
1539
1540         dr = find_pci_dr(pdev);
1541         if (dr)
1542                 dr->region_mask &= ~(1 << bar);
1543 }
1544
1545 /**
1546  *      __pci_request_region - Reserved PCI I/O and memory resource
1547  *      @pdev: PCI device whose resources are to be reserved
1548  *      @bar: BAR to be reserved
1549  *      @res_name: Name to be associated with resource.
1550  *      @exclusive: whether the region access is exclusive or not
1551  *
1552  *      Mark the PCI region associated with PCI device @pdev BR @bar as
1553  *      being reserved by owner @res_name.  Do not access any
1554  *      address inside the PCI regions unless this call returns
1555  *      successfully.
1556  *
1557  *      If @exclusive is set, then the region is marked so that userspace
1558  *      is explicitly not allowed to map the resource via /dev/mem or
1559  *      sysfs MMIO access.
1560  *
1561  *      Returns 0 on success, or %EBUSY on error.  A warning
1562  *      message is also printed on failure.
1563  */
1564 static int __pci_request_region(struct pci_dev *pdev, int bar, const char *res_name,
1565                                                                         int exclusive)
1566 {
1567         struct pci_devres *dr;
1568
1569         if (pci_resource_len(pdev, bar) == 0)
1570                 return 0;
1571                 
1572         if (pci_resource_flags(pdev, bar) & IORESOURCE_IO) {
1573                 if (!request_region(pci_resource_start(pdev, bar),
1574                             pci_resource_len(pdev, bar), res_name))
1575                         goto err_out;
1576         }
1577         else if (pci_resource_flags(pdev, bar) & IORESOURCE_MEM) {
1578                 if (!__request_mem_region(pci_resource_start(pdev, bar),
1579                                         pci_resource_len(pdev, bar), res_name,
1580                                         exclusive))
1581                         goto err_out;
1582         }
1583
1584         dr = find_pci_dr(pdev);
1585         if (dr)
1586                 dr->region_mask |= 1 << bar;
1587
1588         return 0;
1589
1590 err_out:
1591         dev_warn(&pdev->dev, "BAR %d: can't reserve %s region %pR\n",
1592                  bar,
1593                  pci_resource_flags(pdev, bar) & IORESOURCE_IO ? "I/O" : "mem",
1594                  &pdev->resource[bar]);
1595         return -EBUSY;
1596 }
1597
1598 /**
1599  *      pci_request_region - Reserve PCI I/O and memory resource
1600  *      @pdev: PCI device whose resources are to be reserved
1601  *      @bar: BAR to be reserved
1602  *      @res_name: Name to be associated with resource
1603  *
1604  *      Mark the PCI region associated with PCI device @pdev BAR @bar as
1605  *      being reserved by owner @res_name.  Do not access any
1606  *      address inside the PCI regions unless this call returns
1607  *      successfully.
1608  *
1609  *      Returns 0 on success, or %EBUSY on error.  A warning
1610  *      message is also printed on failure.
1611  */
1612 int pci_request_region(struct pci_dev *pdev, int bar, const char *res_name)
1613 {
1614         return __pci_request_region(pdev, bar, res_name, 0);
1615 }
1616
1617 /**
1618  *      pci_request_region_exclusive - Reserved PCI I/O and memory resource
1619  *      @pdev: PCI device whose resources are to be reserved
1620  *      @bar: BAR to be reserved
1621  *      @res_name: Name to be associated with resource.
1622  *
1623  *      Mark the PCI region associated with PCI device @pdev BR @bar as
1624  *      being reserved by owner @res_name.  Do not access any
1625  *      address inside the PCI regions unless this call returns
1626  *      successfully.
1627  *
1628  *      Returns 0 on success, or %EBUSY on error.  A warning
1629  *      message is also printed on failure.
1630  *
1631  *      The key difference that _exclusive makes it that userspace is
1632  *      explicitly not allowed to map the resource via /dev/mem or
1633  *      sysfs.
1634  */
1635 int pci_request_region_exclusive(struct pci_dev *pdev, int bar, const char *res_name)
1636 {
1637         return __pci_request_region(pdev, bar, res_name, IORESOURCE_EXCLUSIVE);
1638 }
1639 /**
1640  * pci_release_selected_regions - Release selected PCI I/O and memory resources
1641  * @pdev: PCI device whose resources were previously reserved
1642  * @bars: Bitmask of BARs to be released
1643  *
1644  * Release selected PCI I/O and memory resources previously reserved.
1645  * Call this function only after all use of the PCI regions has ceased.
1646  */
1647 void pci_release_selected_regions(struct pci_dev *pdev, int bars)
1648 {
1649         int i;
1650
1651         for (i = 0; i < 6; i++)
1652                 if (bars & (1 << i))
1653                         pci_release_region(pdev, i);
1654 }
1655
1656 int __pci_request_selected_regions(struct pci_dev *pdev, int bars,
1657                                  const char *res_name, int excl)
1658 {
1659         int i;
1660
1661         for (i = 0; i < 6; i++)
1662                 if (bars & (1 << i))
1663                         if (__pci_request_region(pdev, i, res_name, excl))
1664                                 goto err_out;
1665         return 0;
1666
1667 err_out:
1668         while(--i >= 0)
1669                 if (bars & (1 << i))
1670                         pci_release_region(pdev, i);
1671
1672         return -EBUSY;
1673 }
1674
1675
1676 /**
1677  * pci_request_selected_regions - Reserve selected PCI I/O and memory resources
1678  * @pdev: PCI device whose resources are to be reserved
1679  * @bars: Bitmask of BARs to be requested
1680  * @res_name: Name to be associated with resource
1681  */
1682 int pci_request_selected_regions(struct pci_dev *pdev, int bars,
1683                                  const char *res_name)
1684 {
1685         return __pci_request_selected_regions(pdev, bars, res_name, 0);
1686 }
1687
1688 int pci_request_selected_regions_exclusive(struct pci_dev *pdev,
1689                                  int bars, const char *res_name)
1690 {
1691         return __pci_request_selected_regions(pdev, bars, res_name,
1692                         IORESOURCE_EXCLUSIVE);
1693 }
1694
1695 /**
1696  *      pci_release_regions - Release reserved PCI I/O and memory resources
1697  *      @pdev: PCI device whose resources were previously reserved by pci_request_regions
1698  *
1699  *      Releases all PCI I/O and memory resources previously reserved by a
1700  *      successful call to pci_request_regions.  Call this function only
1701  *      after all use of the PCI regions has ceased.
1702  */
1703
1704 void pci_release_regions(struct pci_dev *pdev)
1705 {
1706         pci_release_selected_regions(pdev, (1 << 6) - 1);
1707 }
1708
1709 /**
1710  *      pci_request_regions - Reserved PCI I/O and memory resources
1711  *      @pdev: PCI device whose resources are to be reserved
1712  *      @res_name: Name to be associated with resource.
1713  *
1714  *      Mark all PCI regions associated with PCI device @pdev as
1715  *      being reserved by owner @res_name.  Do not access any
1716  *      address inside the PCI regions unless this call returns
1717  *      successfully.
1718  *
1719  *      Returns 0 on success, or %EBUSY on error.  A warning
1720  *      message is also printed on failure.
1721  */
1722 int pci_request_regions(struct pci_dev *pdev, const char *res_name)
1723 {
1724         return pci_request_selected_regions(pdev, ((1 << 6) - 1), res_name);
1725 }
1726
1727 /**
1728  *      pci_request_regions_exclusive - Reserved PCI I/O and memory resources
1729  *      @pdev: PCI device whose resources are to be reserved
1730  *      @res_name: Name to be associated with resource.
1731  *
1732  *      Mark all PCI regions associated with PCI device @pdev as
1733  *      being reserved by owner @res_name.  Do not access any
1734  *      address inside the PCI regions unless this call returns
1735  *      successfully.
1736  *
1737  *      pci_request_regions_exclusive() will mark the region so that
1738  *      /dev/mem and the sysfs MMIO access will not be allowed.
1739  *
1740  *      Returns 0 on success, or %EBUSY on error.  A warning
1741  *      message is also printed on failure.
1742  */
1743 int pci_request_regions_exclusive(struct pci_dev *pdev, const char *res_name)
1744 {
1745         return pci_request_selected_regions_exclusive(pdev,
1746                                         ((1 << 6) - 1), res_name);
1747 }
1748
1749 static void __pci_set_master(struct pci_dev *dev, bool enable)
1750 {
1751         u16 old_cmd, cmd;
1752
1753         pci_read_config_word(dev, PCI_COMMAND, &old_cmd);
1754         if (enable)
1755                 cmd = old_cmd | PCI_COMMAND_MASTER;
1756         else
1757                 cmd = old_cmd & ~PCI_COMMAND_MASTER;
1758         if (cmd != old_cmd) {
1759                 dev_dbg(&dev->dev, "%s bus mastering\n",
1760                         enable ? "enabling" : "disabling");
1761                 pci_write_config_word(dev, PCI_COMMAND, cmd);
1762         }
1763         dev->is_busmaster = enable;
1764 }
1765
1766 /**
1767  * pci_set_master - enables bus-mastering for device dev
1768  * @dev: the PCI device to enable
1769  *
1770  * Enables bus-mastering on the device and calls pcibios_set_master()
1771  * to do the needed arch specific settings.
1772  */
1773 void pci_set_master(struct pci_dev *dev)
1774 {
1775         __pci_set_master(dev, true);
1776         pcibios_set_master(dev);
1777 }
1778
1779 /**
1780  * pci_clear_master - disables bus-mastering for device dev
1781  * @dev: the PCI device to disable
1782  */
1783 void pci_clear_master(struct pci_dev *dev)
1784 {
1785         __pci_set_master(dev, false);
1786 }
1787
1788 #ifdef PCI_DISABLE_MWI
1789 int pci_set_mwi(struct pci_dev *dev)
1790 {
1791         return 0;
1792 }
1793
1794 int pci_try_set_mwi(struct pci_dev *dev)
1795 {
1796         return 0;
1797 }
1798
1799 void pci_clear_mwi(struct pci_dev *dev)
1800 {
1801 }
1802
1803 #else
1804
1805 #ifndef PCI_CACHE_LINE_BYTES
1806 #define PCI_CACHE_LINE_BYTES L1_CACHE_BYTES
1807 #endif
1808
1809 /* This can be overridden by arch code. */
1810 /* Don't forget this is measured in 32-bit words, not bytes */
1811 u8 pci_cache_line_size = PCI_CACHE_LINE_BYTES / 4;
1812
1813 /**
1814  * pci_set_cacheline_size - ensure the CACHE_LINE_SIZE register is programmed
1815  * @dev: the PCI device for which MWI is to be enabled
1816  *
1817  * Helper function for pci_set_mwi.
1818  * Originally copied from drivers/net/acenic.c.
1819  * Copyright 1998-2001 by Jes Sorensen, <jes@trained-monkey.org>.
1820  *
1821  * RETURNS: An appropriate -ERRNO error value on error, or zero for success.
1822  */
1823 static int
1824 pci_set_cacheline_size(struct pci_dev *dev)
1825 {
1826         u8 cacheline_size;
1827
1828         if (!pci_cache_line_size)
1829                 return -EINVAL;         /* The system doesn't support MWI. */
1830
1831         /* Validate current setting: the PCI_CACHE_LINE_SIZE must be
1832            equal to or multiple of the right value. */
1833         pci_read_config_byte(dev, PCI_CACHE_LINE_SIZE, &cacheline_size);
1834         if (cacheline_size >= pci_cache_line_size &&
1835             (cacheline_size % pci_cache_line_size) == 0)
1836                 return 0;
1837
1838         /* Write the correct value. */
1839         pci_write_config_byte(dev, PCI_CACHE_LINE_SIZE, pci_cache_line_size);
1840         /* Read it back. */
1841         pci_read_config_byte(dev, PCI_CACHE_LINE_SIZE, &cacheline_size);
1842         if (cacheline_size == pci_cache_line_size)
1843                 return 0;
1844
1845         dev_printk(KERN_DEBUG, &dev->dev, "cache line size of %d is not "
1846                    "supported\n", pci_cache_line_size << 2);
1847
1848         return -EINVAL;
1849 }
1850
1851 /**
1852  * pci_set_mwi - enables memory-write-invalidate PCI transaction
1853  * @dev: the PCI device for which MWI is enabled
1854  *
1855  * Enables the Memory-Write-Invalidate transaction in %PCI_COMMAND.
1856  *
1857  * RETURNS: An appropriate -ERRNO error value on error, or zero for success.
1858  */
1859 int
1860 pci_set_mwi(struct pci_dev *dev)
1861 {
1862         int rc;
1863         u16 cmd;
1864
1865         rc = pci_set_cacheline_size(dev);
1866         if (rc)
1867                 return rc;
1868
1869         pci_read_config_word(dev, PCI_COMMAND, &cmd);
1870         if (! (cmd & PCI_COMMAND_INVALIDATE)) {
1871                 dev_dbg(&dev->dev, "enabling Mem-Wr-Inval\n");
1872                 cmd |= PCI_COMMAND_INVALIDATE;
1873                 pci_write_config_word(dev, PCI_COMMAND, cmd);
1874         }
1875         
1876         return 0;
1877 }
1878
1879 /**
1880  * pci_try_set_mwi - enables memory-write-invalidate PCI transaction
1881  * @dev: the PCI device for which MWI is enabled
1882  *
1883  * Enables the Memory-Write-Invalidate transaction in %PCI_COMMAND.
1884  * Callers are not required to check the return value.
1885  *
1886  * RETURNS: An appropriate -ERRNO error value on error, or zero for success.
1887  */
1888 int pci_try_set_mwi(struct pci_dev *dev)
1889 {
1890         int rc = pci_set_mwi(dev);
1891         return rc;
1892 }
1893
1894 /**
1895  * pci_clear_mwi - disables Memory-Write-Invalidate for device dev
1896  * @dev: the PCI device to disable
1897  *
1898  * Disables PCI Memory-Write-Invalidate transaction on the device
1899  */
1900 void
1901 pci_clear_mwi(struct pci_dev *dev)
1902 {
1903         u16 cmd;
1904
1905         pci_read_config_word(dev, PCI_COMMAND, &cmd);
1906         if (cmd & PCI_COMMAND_INVALIDATE) {
1907                 cmd &= ~PCI_COMMAND_INVALIDATE;
1908                 pci_write_config_word(dev, PCI_COMMAND, cmd);
1909         }
1910 }
1911 #endif /* ! PCI_DISABLE_MWI */
1912
1913 /**
1914  * pci_intx - enables/disables PCI INTx for device dev
1915  * @pdev: the PCI device to operate on
1916  * @enable: boolean: whether to enable or disable PCI INTx
1917  *
1918  * Enables/disables PCI INTx for device dev
1919  */
1920 void
1921 pci_intx(struct pci_dev *pdev, int enable)
1922 {
1923         u16 pci_command, new;
1924
1925         pci_read_config_word(pdev, PCI_COMMAND, &pci_command);
1926
1927         if (enable) {
1928                 new = pci_command & ~PCI_COMMAND_INTX_DISABLE;
1929         } else {
1930                 new = pci_command | PCI_COMMAND_INTX_DISABLE;
1931         }
1932
1933         if (new != pci_command) {
1934                 struct pci_devres *dr;
1935
1936                 pci_write_config_word(pdev, PCI_COMMAND, new);
1937
1938                 dr = find_pci_dr(pdev);
1939                 if (dr && !dr->restore_intx) {
1940                         dr->restore_intx = 1;
1941                         dr->orig_intx = !enable;
1942                 }
1943         }
1944 }
1945
1946 /**
1947  * pci_msi_off - disables any msi or msix capabilities
1948  * @dev: the PCI device to operate on
1949  *
1950  * If you want to use msi see pci_enable_msi and friends.
1951  * This is a lower level primitive that allows us to disable
1952  * msi operation at the device level.
1953  */
1954 void pci_msi_off(struct pci_dev *dev)
1955 {
1956         int pos;
1957         u16 control;
1958
1959         pos = pci_find_capability(dev, PCI_CAP_ID_MSI);
1960         if (pos) {
1961                 pci_read_config_word(dev, pos + PCI_MSI_FLAGS, &control);
1962                 control &= ~PCI_MSI_FLAGS_ENABLE;
1963                 pci_write_config_word(dev, pos + PCI_MSI_FLAGS, control);
1964         }
1965         pos = pci_find_capability(dev, PCI_CAP_ID_MSIX);
1966         if (pos) {
1967                 pci_read_config_word(dev, pos + PCI_MSIX_FLAGS, &control);
1968                 control &= ~PCI_MSIX_FLAGS_ENABLE;
1969                 pci_write_config_word(dev, pos + PCI_MSIX_FLAGS, control);
1970         }
1971 }
1972
1973 #ifndef HAVE_ARCH_PCI_SET_DMA_MASK
1974 /*
1975  * These can be overridden by arch-specific implementations
1976  */
1977 int
1978 pci_set_dma_mask(struct pci_dev *dev, u64 mask)
1979 {
1980         if (!pci_dma_supported(dev, mask))
1981                 return -EIO;
1982
1983         dev->dma_mask = mask;
1984
1985         return 0;
1986 }
1987     
1988 int
1989 pci_set_consistent_dma_mask(struct pci_dev *dev, u64 mask)
1990 {
1991         if (!pci_dma_supported(dev, mask))
1992                 return -EIO;
1993
1994         dev->dev.coherent_dma_mask = mask;
1995
1996         return 0;
1997 }
1998 #endif
1999
2000 #ifndef HAVE_ARCH_PCI_SET_DMA_MAX_SEGMENT_SIZE
2001 int pci_set_dma_max_seg_size(struct pci_dev *dev, unsigned int size)
2002 {
2003         return dma_set_max_seg_size(&dev->dev, size);
2004 }
2005 EXPORT_SYMBOL(pci_set_dma_max_seg_size);
2006 #endif
2007
2008 #ifndef HAVE_ARCH_PCI_SET_DMA_SEGMENT_BOUNDARY
2009 int pci_set_dma_seg_boundary(struct pci_dev *dev, unsigned long mask)
2010 {
2011         return dma_set_seg_boundary(&dev->dev, mask);
2012 }
2013 EXPORT_SYMBOL(pci_set_dma_seg_boundary);
2014 #endif
2015
2016 static int __pcie_flr(struct pci_dev *dev, int probe)
2017 {
2018         u16 status;
2019         u32 cap;
2020         int exppos = pci_find_capability(dev, PCI_CAP_ID_EXP);
2021
2022         if (!exppos)
2023                 return -ENOTTY;
2024         pci_read_config_dword(dev, exppos + PCI_EXP_DEVCAP, &cap);
2025         if (!(cap & PCI_EXP_DEVCAP_FLR))
2026                 return -ENOTTY;
2027
2028         if (probe)
2029                 return 0;
2030
2031         pci_block_user_cfg_access(dev);
2032
2033         /* Wait for Transaction Pending bit clean */
2034         pci_read_config_word(dev, exppos + PCI_EXP_DEVSTA, &status);
2035         if (!(status & PCI_EXP_DEVSTA_TRPND))
2036                 goto transaction_done;
2037
2038         msleep(100);
2039         pci_read_config_word(dev, exppos + PCI_EXP_DEVSTA, &status);
2040         if (!(status & PCI_EXP_DEVSTA_TRPND))
2041                 goto transaction_done;
2042
2043         dev_info(&dev->dev, "Busy after 100ms while trying to reset; "
2044                         "sleeping for 1 second\n");
2045         ssleep(1);
2046         pci_read_config_word(dev, exppos + PCI_EXP_DEVSTA, &status);
2047         if (status & PCI_EXP_DEVSTA_TRPND)
2048                 dev_info(&dev->dev, "Still busy after 1s; "
2049                                 "proceeding with reset anyway\n");
2050
2051 transaction_done:
2052         pci_write_config_word(dev, exppos + PCI_EXP_DEVCTL,
2053                                 PCI_EXP_DEVCTL_BCR_FLR);
2054         mdelay(100);
2055
2056         pci_unblock_user_cfg_access(dev);
2057         return 0;
2058 }
2059
2060 static int __pci_af_flr(struct pci_dev *dev, int probe)
2061 {
2062         int cappos = pci_find_capability(dev, PCI_CAP_ID_AF);
2063         u8 status;
2064         u8 cap;
2065
2066         if (!cappos)
2067                 return -ENOTTY;
2068         pci_read_config_byte(dev, cappos + PCI_AF_CAP, &cap);
2069         if (!(cap & PCI_AF_CAP_TP) || !(cap & PCI_AF_CAP_FLR))
2070                 return -ENOTTY;
2071
2072         if (probe)
2073                 return 0;
2074
2075         pci_block_user_cfg_access(dev);
2076
2077         /* Wait for Transaction Pending bit clean */
2078         pci_read_config_byte(dev, cappos + PCI_AF_STATUS, &status);
2079         if (!(status & PCI_AF_STATUS_TP))
2080                 goto transaction_done;
2081
2082         msleep(100);
2083         pci_read_config_byte(dev, cappos + PCI_AF_STATUS, &status);
2084         if (!(status & PCI_AF_STATUS_TP))
2085                 goto transaction_done;
2086
2087         dev_info(&dev->dev, "Busy after 100ms while trying to"
2088                         " reset; sleeping for 1 second\n");
2089         ssleep(1);
2090         pci_read_config_byte(dev, cappos + PCI_AF_STATUS, &status);
2091         if (status & PCI_AF_STATUS_TP)
2092                 dev_info(&dev->dev, "Still busy after 1s; "
2093                                 "proceeding with reset anyway\n");
2094
2095 transaction_done:
2096         pci_write_config_byte(dev, cappos + PCI_AF_CTRL, PCI_AF_CTRL_FLR);
2097         mdelay(100);
2098
2099         pci_unblock_user_cfg_access(dev);
2100         return 0;
2101 }
2102
2103 static int __pci_reset_function(struct pci_dev *pdev, int probe)
2104 {
2105         int res;
2106
2107         res = __pcie_flr(pdev, probe);
2108         if (res != -ENOTTY)
2109                 return res;
2110
2111         res = __pci_af_flr(pdev, probe);
2112         if (res != -ENOTTY)
2113                 return res;
2114
2115         return res;
2116 }
2117
2118 /**
2119  * pci_execute_reset_function() - Reset a PCI device function
2120  * @dev: Device function to reset
2121  *
2122  * Some devices allow an individual function to be reset without affecting
2123  * other functions in the same device.  The PCI device must be responsive
2124  * to PCI config space in order to use this function.
2125  *
2126  * The device function is presumed to be unused when this function is called.
2127  * Resetting the device will make the contents of PCI configuration space
2128  * random, so any caller of this must be prepared to reinitialise the
2129  * device including MSI, bus mastering, BARs, decoding IO and memory spaces,
2130  * etc.
2131  *
2132  * Returns 0 if the device function was successfully reset or -ENOTTY if the
2133  * device doesn't support resetting a single function.
2134  */
2135 int pci_execute_reset_function(struct pci_dev *dev)
2136 {
2137         return __pci_reset_function(dev, 0);
2138 }
2139 EXPORT_SYMBOL_GPL(pci_execute_reset_function);
2140
2141 /**
2142  * pci_reset_function() - quiesce and reset a PCI device function
2143  * @dev: Device function to reset
2144  *
2145  * Some devices allow an individual function to be reset without affecting
2146  * other functions in the same device.  The PCI device must be responsive
2147  * to PCI config space in order to use this function.
2148  *
2149  * This function does not just reset the PCI portion of a device, but
2150  * clears all the state associated with the device.  This function differs
2151  * from pci_execute_reset_function in that it saves and restores device state
2152  * over the reset.
2153  *
2154  * Returns 0 if the device function was successfully reset or -ENOTTY if the
2155  * device doesn't support resetting a single function.
2156  */
2157 int pci_reset_function(struct pci_dev *dev)
2158 {
2159         int r = __pci_reset_function(dev, 1);
2160
2161         if (r < 0)
2162                 return r;
2163
2164         if (!dev->msi_enabled && !dev->msix_enabled && dev->irq != 0)
2165                 disable_irq(dev->irq);
2166         pci_save_state(dev);
2167
2168         pci_write_config_word(dev, PCI_COMMAND, PCI_COMMAND_INTX_DISABLE);
2169
2170         r = pci_execute_reset_function(dev);
2171
2172         pci_restore_state(dev);
2173         if (!dev->msi_enabled && !dev->msix_enabled && dev->irq != 0)
2174                 enable_irq(dev->irq);
2175
2176         return r;
2177 }
2178 EXPORT_SYMBOL_GPL(pci_reset_function);
2179
2180 /**
2181  * pcix_get_max_mmrbc - get PCI-X maximum designed memory read byte count
2182  * @dev: PCI device to query
2183  *
2184  * Returns mmrbc: maximum designed memory read count in bytes
2185  *    or appropriate error value.
2186  */
2187 int pcix_get_max_mmrbc(struct pci_dev *dev)
2188 {
2189         int err, cap;
2190         u32 stat;
2191
2192         cap = pci_find_capability(dev, PCI_CAP_ID_PCIX);
2193         if (!cap)
2194                 return -EINVAL;
2195
2196         err = pci_read_config_dword(dev, cap + PCI_X_STATUS, &stat);
2197         if (err)
2198                 return -EINVAL;
2199
2200         return (stat & PCI_X_STATUS_MAX_READ) >> 12;
2201 }
2202 EXPORT_SYMBOL(pcix_get_max_mmrbc);
2203
2204 /**
2205  * pcix_get_mmrbc - get PCI-X maximum memory read byte count
2206  * @dev: PCI device to query
2207  *
2208  * Returns mmrbc: maximum memory read count in bytes
2209  *    or appropriate error value.
2210  */
2211 int pcix_get_mmrbc(struct pci_dev *dev)
2212 {
2213         int ret, cap;
2214         u32 cmd;
2215
2216         cap = pci_find_capability(dev, PCI_CAP_ID_PCIX);
2217         if (!cap)
2218                 return -EINVAL;
2219
2220         ret = pci_read_config_dword(dev, cap + PCI_X_CMD, &cmd);
2221         if (!ret)
2222                 ret = 512 << ((cmd & PCI_X_CMD_MAX_READ) >> 2);
2223
2224         return ret;
2225 }
2226 EXPORT_SYMBOL(pcix_get_mmrbc);
2227
2228 /**
2229  * pcix_set_mmrbc - set PCI-X maximum memory read byte count
2230  * @dev: PCI device to query
2231  * @mmrbc: maximum memory read count in bytes
2232  *    valid values are 512, 1024, 2048, 4096
2233  *
2234  * If possible sets maximum memory read byte count, some bridges have erratas
2235  * that prevent this.
2236  */
2237 int pcix_set_mmrbc(struct pci_dev *dev, int mmrbc)
2238 {
2239         int cap, err = -EINVAL;
2240         u32 stat, cmd, v, o;
2241
2242         if (mmrbc < 512 || mmrbc > 4096 || !is_power_of_2(mmrbc))
2243                 goto out;
2244
2245         v = ffs(mmrbc) - 10;
2246
2247         cap = pci_find_capability(dev, PCI_CAP_ID_PCIX);
2248         if (!cap)
2249                 goto out;
2250
2251         err = pci_read_config_dword(dev, cap + PCI_X_STATUS, &stat);
2252         if (err)
2253                 goto out;
2254
2255         if (v > (stat & PCI_X_STATUS_MAX_READ) >> 21)
2256                 return -E2BIG;
2257
2258         err = pci_read_config_dword(dev, cap + PCI_X_CMD, &cmd);
2259         if (err)
2260                 goto out;
2261
2262         o = (cmd & PCI_X_CMD_MAX_READ) >> 2;
2263         if (o != v) {
2264                 if (v > o && dev->bus &&
2265                    (dev->bus->bus_flags & PCI_BUS_FLAGS_NO_MMRBC))
2266                         return -EIO;
2267
2268                 cmd &= ~PCI_X_CMD_MAX_READ;
2269                 cmd |= v << 2;
2270                 err = pci_write_config_dword(dev, cap + PCI_X_CMD, cmd);
2271         }
2272 out:
2273         return err;
2274 }
2275 EXPORT_SYMBOL(pcix_set_mmrbc);
2276
2277 /**
2278  * pcie_get_readrq - get PCI Express read request size
2279  * @dev: PCI device to query
2280  *
2281  * Returns maximum memory read request in bytes
2282  *    or appropriate error value.
2283  */
2284 int pcie_get_readrq(struct pci_dev *dev)
2285 {
2286         int ret, cap;
2287         u16 ctl;
2288
2289         cap = pci_find_capability(dev, PCI_CAP_ID_EXP);
2290         if (!cap)
2291                 return -EINVAL;
2292
2293         ret = pci_read_config_word(dev, cap + PCI_EXP_DEVCTL, &ctl);
2294         if (!ret)
2295         ret = 128 << ((ctl & PCI_EXP_DEVCTL_READRQ) >> 12);
2296
2297         return ret;
2298 }
2299 EXPORT_SYMBOL(pcie_get_readrq);
2300
2301 /**
2302  * pcie_set_readrq - set PCI Express maximum memory read request
2303  * @dev: PCI device to query
2304  * @rq: maximum memory read count in bytes
2305  *    valid values are 128, 256, 512, 1024, 2048, 4096
2306  *
2307  * If possible sets maximum read byte count
2308  */
2309 int pcie_set_readrq(struct pci_dev *dev, int rq)
2310 {
2311         int cap, err = -EINVAL;
2312         u16 ctl, v;
2313
2314         if (rq < 128 || rq > 4096 || !is_power_of_2(rq))
2315                 goto out;
2316
2317         v = (ffs(rq) - 8) << 12;
2318
2319         cap = pci_find_capability(dev, PCI_CAP_ID_EXP);
2320         if (!cap)
2321                 goto out;
2322
2323         err = pci_read_config_word(dev, cap + PCI_EXP_DEVCTL, &ctl);
2324         if (err)
2325                 goto out;
2326
2327         if ((ctl & PCI_EXP_DEVCTL_READRQ) != v) {
2328                 ctl &= ~PCI_EXP_DEVCTL_READRQ;
2329                 ctl |= v;
2330                 err = pci_write_config_dword(dev, cap + PCI_EXP_DEVCTL, ctl);
2331         }
2332
2333 out:
2334         return err;
2335 }
2336 EXPORT_SYMBOL(pcie_set_readrq);
2337
2338 /**
2339  * pci_select_bars - Make BAR mask from the type of resource
2340  * @dev: the PCI device for which BAR mask is made
2341  * @flags: resource type mask to be selected
2342  *
2343  * This helper routine makes bar mask from the type of resource.
2344  */
2345 int pci_select_bars(struct pci_dev *dev, unsigned long flags)
2346 {
2347         int i, bars = 0;
2348         for (i = 0; i < PCI_NUM_RESOURCES; i++)
2349                 if (pci_resource_flags(dev, i) & flags)
2350                         bars |= (1 << i);
2351         return bars;
2352 }
2353
2354 /**
2355  * pci_resource_bar - get position of the BAR associated with a resource
2356  * @dev: the PCI device
2357  * @resno: the resource number
2358  * @type: the BAR type to be filled in
2359  *
2360  * Returns BAR position in config space, or 0 if the BAR is invalid.
2361  */
2362 int pci_resource_bar(struct pci_dev *dev, int resno, enum pci_bar_type *type)
2363 {
2364         int reg;
2365
2366         if (resno < PCI_ROM_RESOURCE) {
2367                 *type = pci_bar_unknown;
2368                 return PCI_BASE_ADDRESS_0 + 4 * resno;
2369         } else if (resno == PCI_ROM_RESOURCE) {
2370                 *type = pci_bar_mem32;
2371                 return dev->rom_base_reg;
2372         } else if (resno < PCI_BRIDGE_RESOURCES) {
2373                 /* device specific resource */
2374                 reg = pci_iov_resource_bar(dev, resno, type);
2375                 if (reg)
2376                         return reg;
2377         }
2378
2379         dev_err(&dev->dev, "BAR: invalid resource #%d\n", resno);
2380         return 0;
2381 }
2382
2383 #define RESOURCE_ALIGNMENT_PARAM_SIZE COMMAND_LINE_SIZE
2384 static char resource_alignment_param[RESOURCE_ALIGNMENT_PARAM_SIZE] = {0};
2385 spinlock_t resource_alignment_lock = SPIN_LOCK_UNLOCKED;
2386
2387 /**
2388  * pci_specified_resource_alignment - get resource alignment specified by user.
2389  * @dev: the PCI device to get
2390  *
2391  * RETURNS: Resource alignment if it is specified.
2392  *          Zero if it is not specified.
2393  */
2394 resource_size_t pci_specified_resource_alignment(struct pci_dev *dev)
2395 {
2396         int seg, bus, slot, func, align_order, count;
2397         resource_size_t align = 0;
2398         char *p;
2399
2400         spin_lock(&resource_alignment_lock);
2401         p = resource_alignment_param;
2402         while (*p) {
2403                 count = 0;
2404                 if (sscanf(p, "%d%n", &align_order, &count) == 1 &&
2405                                                         p[count] == '@') {
2406                         p += count + 1;
2407                 } else {
2408                         align_order = -1;
2409                 }
2410                 if (sscanf(p, "%x:%x:%x.%x%n",
2411                         &seg, &bus, &slot, &func, &count) != 4) {
2412                         seg = 0;
2413                         if (sscanf(p, "%x:%x.%x%n",
2414                                         &bus, &slot, &func, &count) != 3) {
2415                                 /* Invalid format */
2416                                 printk(KERN_ERR "PCI: Can't parse resource_alignment parameter: %s\n",
2417                                         p);
2418                                 break;
2419                         }
2420                 }
2421                 p += count;
2422                 if (seg == pci_domain_nr(dev->bus) &&
2423                         bus == dev->bus->number &&
2424                         slot == PCI_SLOT(dev->devfn) &&
2425                         func == PCI_FUNC(dev->devfn)) {
2426                         if (align_order == -1) {
2427                                 align = PAGE_SIZE;
2428                         } else {
2429                                 align = 1 << align_order;
2430                         }
2431                         /* Found */
2432                         break;
2433                 }
2434                 if (*p != ';' && *p != ',') {
2435                         /* End of param or invalid format */
2436                         break;
2437                 }
2438                 p++;
2439         }
2440         spin_unlock(&resource_alignment_lock);
2441         return align;
2442 }
2443
2444 /**
2445  * pci_is_reassigndev - check if specified PCI is target device to reassign
2446  * @dev: the PCI device to check
2447  *
2448  * RETURNS: non-zero for PCI device is a target device to reassign,
2449  *          or zero is not.
2450  */
2451 int pci_is_reassigndev(struct pci_dev *dev)
2452 {
2453         return (pci_specified_resource_alignment(dev) != 0);
2454 }
2455
2456 ssize_t pci_set_resource_alignment_param(const char *buf, size_t count)
2457 {
2458         if (count > RESOURCE_ALIGNMENT_PARAM_SIZE - 1)
2459                 count = RESOURCE_ALIGNMENT_PARAM_SIZE - 1;
2460         spin_lock(&resource_alignment_lock);
2461         strncpy(resource_alignment_param, buf, count);
2462         resource_alignment_param[count] = '\0';
2463         spin_unlock(&resource_alignment_lock);
2464         return count;
2465 }
2466
2467 ssize_t pci_get_resource_alignment_param(char *buf, size_t size)
2468 {
2469         size_t count;
2470         spin_lock(&resource_alignment_lock);
2471         count = snprintf(buf, size, "%s", resource_alignment_param);
2472         spin_unlock(&resource_alignment_lock);
2473         return count;
2474 }
2475
2476 static ssize_t pci_resource_alignment_show(struct bus_type *bus, char *buf)
2477 {
2478         return pci_get_resource_alignment_param(buf, PAGE_SIZE);
2479 }
2480
2481 static ssize_t pci_resource_alignment_store(struct bus_type *bus,
2482                                         const char *buf, size_t count)
2483 {
2484         return pci_set_resource_alignment_param(buf, count);
2485 }
2486
2487 BUS_ATTR(resource_alignment, 0644, pci_resource_alignment_show,
2488                                         pci_resource_alignment_store);
2489
2490 static int __init pci_resource_alignment_sysfs_init(void)
2491 {
2492         return bus_create_file(&pci_bus_type,
2493                                         &bus_attr_resource_alignment);
2494 }
2495
2496 late_initcall(pci_resource_alignment_sysfs_init);
2497
2498 static void __devinit pci_no_domains(void)
2499 {
2500 #ifdef CONFIG_PCI_DOMAINS
2501         pci_domains_supported = 0;
2502 #endif
2503 }
2504
2505 /**
2506  * pci_ext_cfg_enabled - can we access extended PCI config space?
2507  * @dev: The PCI device of the root bridge.
2508  *
2509  * Returns 1 if we can access PCI extended config space (offsets
2510  * greater than 0xff). This is the default implementation. Architecture
2511  * implementations can override this.
2512  */
2513 int __attribute__ ((weak)) pci_ext_cfg_avail(struct pci_dev *dev)
2514 {
2515         return 1;
2516 }
2517
2518 static int __devinit pci_init(void)
2519 {
2520         struct pci_dev *dev = NULL;
2521
2522         while ((dev = pci_get_device(PCI_ANY_ID, PCI_ANY_ID, dev)) != NULL) {
2523                 pci_fixup_device(pci_fixup_final, dev);
2524         }
2525
2526         return 0;
2527 }
2528
2529 static int __init pci_setup(char *str)
2530 {
2531         while (str) {
2532                 char *k = strchr(str, ',');
2533                 if (k)
2534                         *k++ = 0;
2535                 if (*str && (str = pcibios_setup(str)) && *str) {
2536                         if (!strcmp(str, "nomsi")) {
2537                                 pci_no_msi();
2538                         } else if (!strcmp(str, "noaer")) {
2539                                 pci_no_aer();
2540                         } else if (!strcmp(str, "nodomains")) {
2541                                 pci_no_domains();
2542                         } else if (!strncmp(str, "cbiosize=", 9)) {
2543                                 pci_cardbus_io_size = memparse(str + 9, &str);
2544                         } else if (!strncmp(str, "cbmemsize=", 10)) {
2545                                 pci_cardbus_mem_size = memparse(str + 10, &str);
2546                         } else if (!strncmp(str, "resource_alignment=", 19)) {
2547                                 pci_set_resource_alignment_param(str + 19,
2548                                                         strlen(str + 19));
2549                         } else {
2550                                 printk(KERN_ERR "PCI: Unknown option `%s'\n",
2551                                                 str);
2552                         }
2553                 }
2554                 str = k;
2555         }
2556         return 0;
2557 }
2558 early_param("pci", pci_setup);
2559
2560 device_initcall(pci_init);
2561
2562 EXPORT_SYMBOL(pci_reenable_device);
2563 EXPORT_SYMBOL(pci_enable_device_io);
2564 EXPORT_SYMBOL(pci_enable_device_mem);
2565 EXPORT_SYMBOL(pci_enable_device);
2566 EXPORT_SYMBOL(pcim_enable_device);
2567 EXPORT_SYMBOL(pcim_pin_device);
2568 EXPORT_SYMBOL(pci_disable_device);
2569 EXPORT_SYMBOL(pci_find_capability);
2570 EXPORT_SYMBOL(pci_bus_find_capability);
2571 EXPORT_SYMBOL(pci_release_regions);
2572 EXPORT_SYMBOL(pci_request_regions);
2573 EXPORT_SYMBOL(pci_request_regions_exclusive);
2574 EXPORT_SYMBOL(pci_release_region);
2575 EXPORT_SYMBOL(pci_request_region);
2576 EXPORT_SYMBOL(pci_request_region_exclusive);
2577 EXPORT_SYMBOL(pci_release_selected_regions);
2578 EXPORT_SYMBOL(pci_request_selected_regions);
2579 EXPORT_SYMBOL(pci_request_selected_regions_exclusive);
2580 EXPORT_SYMBOL(pci_set_master);
2581 EXPORT_SYMBOL(pci_clear_master);
2582 EXPORT_SYMBOL(pci_set_mwi);
2583 EXPORT_SYMBOL(pci_try_set_mwi);
2584 EXPORT_SYMBOL(pci_clear_mwi);
2585 EXPORT_SYMBOL_GPL(pci_intx);
2586 EXPORT_SYMBOL(pci_set_dma_mask);
2587 EXPORT_SYMBOL(pci_set_consistent_dma_mask);
2588 EXPORT_SYMBOL(pci_assign_resource);
2589 EXPORT_SYMBOL(pci_find_parent_resource);
2590 EXPORT_SYMBOL(pci_select_bars);
2591
2592 EXPORT_SYMBOL(pci_set_power_state);
2593 EXPORT_SYMBOL(pci_save_state);
2594 EXPORT_SYMBOL(pci_restore_state);
2595 EXPORT_SYMBOL(pci_pme_capable);
2596 EXPORT_SYMBOL(pci_pme_active);
2597 EXPORT_SYMBOL(pci_enable_wake);
2598 EXPORT_SYMBOL(pci_wake_from_d3);
2599 EXPORT_SYMBOL(pci_target_state);
2600 EXPORT_SYMBOL(pci_prepare_to_sleep);
2601 EXPORT_SYMBOL(pci_back_from_sleep);
2602 EXPORT_SYMBOL_GPL(pci_set_pcie_reset_state);
2603