Merge tag 'optee-bus-for-v5.9' of git://git.linaro.org/people/jens.wiklander/linux...
[linux-2.6-microblaze.git] / drivers / pci / pci-driver.c
1 // SPDX-License-Identifier: GPL-2.0
2 /*
3  * (C) Copyright 2002-2004, 2007 Greg Kroah-Hartman <greg@kroah.com>
4  * (C) Copyright 2007 Novell Inc.
5  */
6
7 #include <linux/pci.h>
8 #include <linux/module.h>
9 #include <linux/init.h>
10 #include <linux/device.h>
11 #include <linux/mempolicy.h>
12 #include <linux/string.h>
13 #include <linux/slab.h>
14 #include <linux/sched.h>
15 #include <linux/cpu.h>
16 #include <linux/pm_runtime.h>
17 #include <linux/suspend.h>
18 #include <linux/kexec.h>
19 #include <linux/of_device.h>
20 #include <linux/acpi.h>
21 #include "pci.h"
22 #include "pcie/portdrv.h"
23
24 struct pci_dynid {
25         struct list_head node;
26         struct pci_device_id id;
27 };
28
29 /**
30  * pci_add_dynid - add a new PCI device ID to this driver and re-probe devices
31  * @drv: target pci driver
32  * @vendor: PCI vendor ID
33  * @device: PCI device ID
34  * @subvendor: PCI subvendor ID
35  * @subdevice: PCI subdevice ID
36  * @class: PCI class
37  * @class_mask: PCI class mask
38  * @driver_data: private driver data
39  *
40  * Adds a new dynamic pci device ID to this driver and causes the
41  * driver to probe for all devices again.  @drv must have been
42  * registered prior to calling this function.
43  *
44  * CONTEXT:
45  * Does GFP_KERNEL allocation.
46  *
47  * RETURNS:
48  * 0 on success, -errno on failure.
49  */
50 int pci_add_dynid(struct pci_driver *drv,
51                   unsigned int vendor, unsigned int device,
52                   unsigned int subvendor, unsigned int subdevice,
53                   unsigned int class, unsigned int class_mask,
54                   unsigned long driver_data)
55 {
56         struct pci_dynid *dynid;
57
58         dynid = kzalloc(sizeof(*dynid), GFP_KERNEL);
59         if (!dynid)
60                 return -ENOMEM;
61
62         dynid->id.vendor = vendor;
63         dynid->id.device = device;
64         dynid->id.subvendor = subvendor;
65         dynid->id.subdevice = subdevice;
66         dynid->id.class = class;
67         dynid->id.class_mask = class_mask;
68         dynid->id.driver_data = driver_data;
69
70         spin_lock(&drv->dynids.lock);
71         list_add_tail(&dynid->node, &drv->dynids.list);
72         spin_unlock(&drv->dynids.lock);
73
74         return driver_attach(&drv->driver);
75 }
76 EXPORT_SYMBOL_GPL(pci_add_dynid);
77
78 static void pci_free_dynids(struct pci_driver *drv)
79 {
80         struct pci_dynid *dynid, *n;
81
82         spin_lock(&drv->dynids.lock);
83         list_for_each_entry_safe(dynid, n, &drv->dynids.list, node) {
84                 list_del(&dynid->node);
85                 kfree(dynid);
86         }
87         spin_unlock(&drv->dynids.lock);
88 }
89
90 /**
91  * store_new_id - sysfs frontend to pci_add_dynid()
92  * @driver: target device driver
93  * @buf: buffer for scanning device ID data
94  * @count: input size
95  *
96  * Allow PCI IDs to be added to an existing driver via sysfs.
97  */
98 static ssize_t new_id_store(struct device_driver *driver, const char *buf,
99                             size_t count)
100 {
101         struct pci_driver *pdrv = to_pci_driver(driver);
102         const struct pci_device_id *ids = pdrv->id_table;
103         u32 vendor, device, subvendor = PCI_ANY_ID,
104                 subdevice = PCI_ANY_ID, class = 0, class_mask = 0;
105         unsigned long driver_data = 0;
106         int fields = 0;
107         int retval = 0;
108
109         fields = sscanf(buf, "%x %x %x %x %x %x %lx",
110                         &vendor, &device, &subvendor, &subdevice,
111                         &class, &class_mask, &driver_data);
112         if (fields < 2)
113                 return -EINVAL;
114
115         if (fields != 7) {
116                 struct pci_dev *pdev = kzalloc(sizeof(*pdev), GFP_KERNEL);
117                 if (!pdev)
118                         return -ENOMEM;
119
120                 pdev->vendor = vendor;
121                 pdev->device = device;
122                 pdev->subsystem_vendor = subvendor;
123                 pdev->subsystem_device = subdevice;
124                 pdev->class = class;
125
126                 if (pci_match_id(pdrv->id_table, pdev))
127                         retval = -EEXIST;
128
129                 kfree(pdev);
130
131                 if (retval)
132                         return retval;
133         }
134
135         /* Only accept driver_data values that match an existing id_table
136            entry */
137         if (ids) {
138                 retval = -EINVAL;
139                 while (ids->vendor || ids->subvendor || ids->class_mask) {
140                         if (driver_data == ids->driver_data) {
141                                 retval = 0;
142                                 break;
143                         }
144                         ids++;
145                 }
146                 if (retval)     /* No match */
147                         return retval;
148         }
149
150         retval = pci_add_dynid(pdrv, vendor, device, subvendor, subdevice,
151                                class, class_mask, driver_data);
152         if (retval)
153                 return retval;
154         return count;
155 }
156 static DRIVER_ATTR_WO(new_id);
157
158 /**
159  * store_remove_id - remove a PCI device ID from this driver
160  * @driver: target device driver
161  * @buf: buffer for scanning device ID data
162  * @count: input size
163  *
164  * Removes a dynamic pci device ID to this driver.
165  */
166 static ssize_t remove_id_store(struct device_driver *driver, const char *buf,
167                                size_t count)
168 {
169         struct pci_dynid *dynid, *n;
170         struct pci_driver *pdrv = to_pci_driver(driver);
171         u32 vendor, device, subvendor = PCI_ANY_ID,
172                 subdevice = PCI_ANY_ID, class = 0, class_mask = 0;
173         int fields = 0;
174         size_t retval = -ENODEV;
175
176         fields = sscanf(buf, "%x %x %x %x %x %x",
177                         &vendor, &device, &subvendor, &subdevice,
178                         &class, &class_mask);
179         if (fields < 2)
180                 return -EINVAL;
181
182         spin_lock(&pdrv->dynids.lock);
183         list_for_each_entry_safe(dynid, n, &pdrv->dynids.list, node) {
184                 struct pci_device_id *id = &dynid->id;
185                 if ((id->vendor == vendor) &&
186                     (id->device == device) &&
187                     (subvendor == PCI_ANY_ID || id->subvendor == subvendor) &&
188                     (subdevice == PCI_ANY_ID || id->subdevice == subdevice) &&
189                     !((id->class ^ class) & class_mask)) {
190                         list_del(&dynid->node);
191                         kfree(dynid);
192                         retval = count;
193                         break;
194                 }
195         }
196         spin_unlock(&pdrv->dynids.lock);
197
198         return retval;
199 }
200 static DRIVER_ATTR_WO(remove_id);
201
202 static struct attribute *pci_drv_attrs[] = {
203         &driver_attr_new_id.attr,
204         &driver_attr_remove_id.attr,
205         NULL,
206 };
207 ATTRIBUTE_GROUPS(pci_drv);
208
209 /**
210  * pci_match_id - See if a pci device matches a given pci_id table
211  * @ids: array of PCI device id structures to search in
212  * @dev: the PCI device structure to match against.
213  *
214  * Used by a driver to check whether a PCI device present in the
215  * system is in its list of supported devices.  Returns the matching
216  * pci_device_id structure or %NULL if there is no match.
217  *
218  * Deprecated, don't use this as it will not catch any dynamic ids
219  * that a driver might want to check for.
220  */
221 const struct pci_device_id *pci_match_id(const struct pci_device_id *ids,
222                                          struct pci_dev *dev)
223 {
224         if (ids) {
225                 while (ids->vendor || ids->subvendor || ids->class_mask) {
226                         if (pci_match_one_device(ids, dev))
227                                 return ids;
228                         ids++;
229                 }
230         }
231         return NULL;
232 }
233 EXPORT_SYMBOL(pci_match_id);
234
235 static const struct pci_device_id pci_device_id_any = {
236         .vendor = PCI_ANY_ID,
237         .device = PCI_ANY_ID,
238         .subvendor = PCI_ANY_ID,
239         .subdevice = PCI_ANY_ID,
240 };
241
242 /**
243  * pci_match_device - Tell if a PCI device structure has a matching PCI device id structure
244  * @drv: the PCI driver to match against
245  * @dev: the PCI device structure to match against
246  *
247  * Used by a driver to check whether a PCI device present in the
248  * system is in its list of supported devices.  Returns the matching
249  * pci_device_id structure or %NULL if there is no match.
250  */
251 static const struct pci_device_id *pci_match_device(struct pci_driver *drv,
252                                                     struct pci_dev *dev)
253 {
254         struct pci_dynid *dynid;
255         const struct pci_device_id *found_id = NULL;
256
257         /* When driver_override is set, only bind to the matching driver */
258         if (dev->driver_override && strcmp(dev->driver_override, drv->name))
259                 return NULL;
260
261         /* Look at the dynamic ids first, before the static ones */
262         spin_lock(&drv->dynids.lock);
263         list_for_each_entry(dynid, &drv->dynids.list, node) {
264                 if (pci_match_one_device(&dynid->id, dev)) {
265                         found_id = &dynid->id;
266                         break;
267                 }
268         }
269         spin_unlock(&drv->dynids.lock);
270
271         if (!found_id)
272                 found_id = pci_match_id(drv->id_table, dev);
273
274         /* driver_override will always match, send a dummy id */
275         if (!found_id && dev->driver_override)
276                 found_id = &pci_device_id_any;
277
278         return found_id;
279 }
280
281 struct drv_dev_and_id {
282         struct pci_driver *drv;
283         struct pci_dev *dev;
284         const struct pci_device_id *id;
285 };
286
287 static long local_pci_probe(void *_ddi)
288 {
289         struct drv_dev_and_id *ddi = _ddi;
290         struct pci_dev *pci_dev = ddi->dev;
291         struct pci_driver *pci_drv = ddi->drv;
292         struct device *dev = &pci_dev->dev;
293         int rc;
294
295         /*
296          * Unbound PCI devices are always put in D0, regardless of
297          * runtime PM status.  During probe, the device is set to
298          * active and the usage count is incremented.  If the driver
299          * supports runtime PM, it should call pm_runtime_put_noidle(),
300          * or any other runtime PM helper function decrementing the usage
301          * count, in its probe routine and pm_runtime_get_noresume() in
302          * its remove routine.
303          */
304         pm_runtime_get_sync(dev);
305         pci_dev->driver = pci_drv;
306         rc = pci_drv->probe(pci_dev, ddi->id);
307         if (!rc)
308                 return rc;
309         if (rc < 0) {
310                 pci_dev->driver = NULL;
311                 pm_runtime_put_sync(dev);
312                 return rc;
313         }
314         /*
315          * Probe function should return < 0 for failure, 0 for success
316          * Treat values > 0 as success, but warn.
317          */
318         pci_warn(pci_dev, "Driver probe function unexpectedly returned %d\n",
319                  rc);
320         return 0;
321 }
322
323 static bool pci_physfn_is_probed(struct pci_dev *dev)
324 {
325 #ifdef CONFIG_PCI_IOV
326         return dev->is_virtfn && dev->physfn->is_probed;
327 #else
328         return false;
329 #endif
330 }
331
332 static int pci_call_probe(struct pci_driver *drv, struct pci_dev *dev,
333                           const struct pci_device_id *id)
334 {
335         int error, node, cpu;
336         struct drv_dev_and_id ddi = { drv, dev, id };
337
338         /*
339          * Execute driver initialization on node where the device is
340          * attached.  This way the driver likely allocates its local memory
341          * on the right node.
342          */
343         node = dev_to_node(&dev->dev);
344         dev->is_probed = 1;
345
346         cpu_hotplug_disable();
347
348         /*
349          * Prevent nesting work_on_cpu() for the case where a Virtual Function
350          * device is probed from work_on_cpu() of the Physical device.
351          */
352         if (node < 0 || node >= MAX_NUMNODES || !node_online(node) ||
353             pci_physfn_is_probed(dev))
354                 cpu = nr_cpu_ids;
355         else
356                 cpu = cpumask_any_and(cpumask_of_node(node), cpu_online_mask);
357
358         if (cpu < nr_cpu_ids)
359                 error = work_on_cpu(cpu, local_pci_probe, &ddi);
360         else
361                 error = local_pci_probe(&ddi);
362
363         dev->is_probed = 0;
364         cpu_hotplug_enable();
365         return error;
366 }
367
368 /**
369  * __pci_device_probe - check if a driver wants to claim a specific PCI device
370  * @drv: driver to call to check if it wants the PCI device
371  * @pci_dev: PCI device being probed
372  *
373  * returns 0 on success, else error.
374  * side-effect: pci_dev->driver is set to drv when drv claims pci_dev.
375  */
376 static int __pci_device_probe(struct pci_driver *drv, struct pci_dev *pci_dev)
377 {
378         const struct pci_device_id *id;
379         int error = 0;
380
381         if (!pci_dev->driver && drv->probe) {
382                 error = -ENODEV;
383
384                 id = pci_match_device(drv, pci_dev);
385                 if (id)
386                         error = pci_call_probe(drv, pci_dev, id);
387         }
388         return error;
389 }
390
391 int __weak pcibios_alloc_irq(struct pci_dev *dev)
392 {
393         return 0;
394 }
395
396 void __weak pcibios_free_irq(struct pci_dev *dev)
397 {
398 }
399
400 #ifdef CONFIG_PCI_IOV
401 static inline bool pci_device_can_probe(struct pci_dev *pdev)
402 {
403         return (!pdev->is_virtfn || pdev->physfn->sriov->drivers_autoprobe ||
404                 pdev->driver_override);
405 }
406 #else
407 static inline bool pci_device_can_probe(struct pci_dev *pdev)
408 {
409         return true;
410 }
411 #endif
412
413 static int pci_device_probe(struct device *dev)
414 {
415         int error;
416         struct pci_dev *pci_dev = to_pci_dev(dev);
417         struct pci_driver *drv = to_pci_driver(dev->driver);
418
419         if (!pci_device_can_probe(pci_dev))
420                 return -ENODEV;
421
422         pci_assign_irq(pci_dev);
423
424         error = pcibios_alloc_irq(pci_dev);
425         if (error < 0)
426                 return error;
427
428         pci_dev_get(pci_dev);
429         error = __pci_device_probe(drv, pci_dev);
430         if (error) {
431                 pcibios_free_irq(pci_dev);
432                 pci_dev_put(pci_dev);
433         }
434
435         return error;
436 }
437
438 static int pci_device_remove(struct device *dev)
439 {
440         struct pci_dev *pci_dev = to_pci_dev(dev);
441         struct pci_driver *drv = pci_dev->driver;
442
443         if (drv) {
444                 if (drv->remove) {
445                         pm_runtime_get_sync(dev);
446                         drv->remove(pci_dev);
447                         pm_runtime_put_noidle(dev);
448                 }
449                 pcibios_free_irq(pci_dev);
450                 pci_dev->driver = NULL;
451                 pci_iov_remove(pci_dev);
452         }
453
454         /* Undo the runtime PM settings in local_pci_probe() */
455         pm_runtime_put_sync(dev);
456
457         /*
458          * If the device is still on, set the power state as "unknown",
459          * since it might change by the next time we load the driver.
460          */
461         if (pci_dev->current_state == PCI_D0)
462                 pci_dev->current_state = PCI_UNKNOWN;
463
464         /*
465          * We would love to complain here if pci_dev->is_enabled is set, that
466          * the driver should have called pci_disable_device(), but the
467          * unfortunate fact is there are too many odd BIOS and bridge setups
468          * that don't like drivers doing that all of the time.
469          * Oh well, we can dream of sane hardware when we sleep, no matter how
470          * horrible the crap we have to deal with is when we are awake...
471          */
472
473         pci_dev_put(pci_dev);
474         return 0;
475 }
476
477 static void pci_device_shutdown(struct device *dev)
478 {
479         struct pci_dev *pci_dev = to_pci_dev(dev);
480         struct pci_driver *drv = pci_dev->driver;
481
482         pm_runtime_resume(dev);
483
484         if (drv && drv->shutdown)
485                 drv->shutdown(pci_dev);
486
487         /*
488          * If this is a kexec reboot, turn off Bus Master bit on the
489          * device to tell it to not continue to do DMA. Don't touch
490          * devices in D3cold or unknown states.
491          * If it is not a kexec reboot, firmware will hit the PCI
492          * devices with big hammer and stop their DMA any way.
493          */
494         if (kexec_in_progress && (pci_dev->current_state <= PCI_D3hot))
495                 pci_clear_master(pci_dev);
496 }
497
498 #ifdef CONFIG_PM
499
500 /* Auxiliary functions used for system resume and run-time resume. */
501
502 /**
503  * pci_restore_standard_config - restore standard config registers of PCI device
504  * @pci_dev: PCI device to handle
505  */
506 static int pci_restore_standard_config(struct pci_dev *pci_dev)
507 {
508         pci_update_current_state(pci_dev, PCI_UNKNOWN);
509
510         if (pci_dev->current_state != PCI_D0) {
511                 int error = pci_set_power_state(pci_dev, PCI_D0);
512                 if (error)
513                         return error;
514         }
515
516         pci_restore_state(pci_dev);
517         pci_pme_restore(pci_dev);
518         return 0;
519 }
520
521 static void pci_pm_default_resume(struct pci_dev *pci_dev)
522 {
523         pci_fixup_device(pci_fixup_resume, pci_dev);
524         pci_enable_wake(pci_dev, PCI_D0, false);
525 }
526
527 #endif
528
529 #ifdef CONFIG_PM_SLEEP
530
531 static void pci_pm_default_resume_early(struct pci_dev *pci_dev)
532 {
533         pci_power_up(pci_dev);
534         pci_update_current_state(pci_dev, PCI_D0);
535         pci_restore_state(pci_dev);
536         pci_pme_restore(pci_dev);
537 }
538
539 /*
540  * Default "suspend" method for devices that have no driver provided suspend,
541  * or not even a driver at all (second part).
542  */
543 static void pci_pm_set_unknown_state(struct pci_dev *pci_dev)
544 {
545         /*
546          * mark its power state as "unknown", since we don't know if
547          * e.g. the BIOS will change its device state when we suspend.
548          */
549         if (pci_dev->current_state == PCI_D0)
550                 pci_dev->current_state = PCI_UNKNOWN;
551 }
552
553 /*
554  * Default "resume" method for devices that have no driver provided resume,
555  * or not even a driver at all (second part).
556  */
557 static int pci_pm_reenable_device(struct pci_dev *pci_dev)
558 {
559         int retval;
560
561         /* if the device was enabled before suspend, reenable */
562         retval = pci_reenable_device(pci_dev);
563         /*
564          * if the device was busmaster before the suspend, make it busmaster
565          * again
566          */
567         if (pci_dev->is_busmaster)
568                 pci_set_master(pci_dev);
569
570         return retval;
571 }
572
573 static int pci_legacy_suspend(struct device *dev, pm_message_t state)
574 {
575         struct pci_dev *pci_dev = to_pci_dev(dev);
576         struct pci_driver *drv = pci_dev->driver;
577
578         if (drv && drv->suspend) {
579                 pci_power_t prev = pci_dev->current_state;
580                 int error;
581
582                 error = drv->suspend(pci_dev, state);
583                 suspend_report_result(drv->suspend, error);
584                 if (error)
585                         return error;
586
587                 if (!pci_dev->state_saved && pci_dev->current_state != PCI_D0
588                     && pci_dev->current_state != PCI_UNKNOWN) {
589                         pci_WARN_ONCE(pci_dev, pci_dev->current_state != prev,
590                                       "PCI PM: Device state not saved by %pS\n",
591                                       drv->suspend);
592                 }
593         }
594
595         pci_fixup_device(pci_fixup_suspend, pci_dev);
596
597         return 0;
598 }
599
600 static int pci_legacy_suspend_late(struct device *dev, pm_message_t state)
601 {
602         struct pci_dev *pci_dev = to_pci_dev(dev);
603
604         if (!pci_dev->state_saved)
605                 pci_save_state(pci_dev);
606
607         pci_pm_set_unknown_state(pci_dev);
608
609         pci_fixup_device(pci_fixup_suspend_late, pci_dev);
610
611         return 0;
612 }
613
614 static int pci_legacy_resume(struct device *dev)
615 {
616         struct pci_dev *pci_dev = to_pci_dev(dev);
617         struct pci_driver *drv = pci_dev->driver;
618
619         pci_fixup_device(pci_fixup_resume, pci_dev);
620
621         return drv && drv->resume ?
622                         drv->resume(pci_dev) : pci_pm_reenable_device(pci_dev);
623 }
624
625 /* Auxiliary functions used by the new power management framework */
626
627 static void pci_pm_default_suspend(struct pci_dev *pci_dev)
628 {
629         /* Disable non-bridge devices without PM support */
630         if (!pci_has_subordinate(pci_dev))
631                 pci_disable_enabled_device(pci_dev);
632 }
633
634 static bool pci_has_legacy_pm_support(struct pci_dev *pci_dev)
635 {
636         struct pci_driver *drv = pci_dev->driver;
637         bool ret = drv && (drv->suspend || drv->resume);
638
639         /*
640          * Legacy PM support is used by default, so warn if the new framework is
641          * supported as well.  Drivers are supposed to support either the
642          * former, or the latter, but not both at the same time.
643          */
644         pci_WARN(pci_dev, ret && drv->driver.pm, "device %04x:%04x\n",
645                  pci_dev->vendor, pci_dev->device);
646
647         return ret;
648 }
649
650 /* New power management framework */
651
652 static int pci_pm_prepare(struct device *dev)
653 {
654         struct pci_dev *pci_dev = to_pci_dev(dev);
655         const struct dev_pm_ops *pm = dev->driver ? dev->driver->pm : NULL;
656
657         if (pm && pm->prepare) {
658                 int error = pm->prepare(dev);
659                 if (error < 0)
660                         return error;
661
662                 if (!error && dev_pm_test_driver_flags(dev, DPM_FLAG_SMART_PREPARE))
663                         return 0;
664         }
665         if (pci_dev_need_resume(pci_dev))
666                 return 0;
667
668         /*
669          * The PME setting needs to be adjusted here in case the direct-complete
670          * optimization is used with respect to this device.
671          */
672         pci_dev_adjust_pme(pci_dev);
673         return 1;
674 }
675
676 static void pci_pm_complete(struct device *dev)
677 {
678         struct pci_dev *pci_dev = to_pci_dev(dev);
679
680         pci_dev_complete_resume(pci_dev);
681         pm_generic_complete(dev);
682
683         /* Resume device if platform firmware has put it in reset-power-on */
684         if (pm_runtime_suspended(dev) && pm_resume_via_firmware()) {
685                 pci_power_t pre_sleep_state = pci_dev->current_state;
686
687                 pci_refresh_power_state(pci_dev);
688                 /*
689                  * On platforms with ACPI this check may also trigger for
690                  * devices sharing power resources if one of those power
691                  * resources has been activated as a result of a change of the
692                  * power state of another device sharing it.  However, in that
693                  * case it is also better to resume the device, in general.
694                  */
695                 if (pci_dev->current_state < pre_sleep_state)
696                         pm_request_resume(dev);
697         }
698 }
699
700 #else /* !CONFIG_PM_SLEEP */
701
702 #define pci_pm_prepare  NULL
703 #define pci_pm_complete NULL
704
705 #endif /* !CONFIG_PM_SLEEP */
706
707 #ifdef CONFIG_SUSPEND
708 static void pcie_pme_root_status_cleanup(struct pci_dev *pci_dev)
709 {
710         /*
711          * Some BIOSes forget to clear Root PME Status bits after system
712          * wakeup, which breaks ACPI-based runtime wakeup on PCI Express.
713          * Clear those bits now just in case (shouldn't hurt).
714          */
715         if (pci_is_pcie(pci_dev) &&
716             (pci_pcie_type(pci_dev) == PCI_EXP_TYPE_ROOT_PORT ||
717              pci_pcie_type(pci_dev) == PCI_EXP_TYPE_RC_EC))
718                 pcie_clear_root_pme_status(pci_dev);
719 }
720
721 static int pci_pm_suspend(struct device *dev)
722 {
723         struct pci_dev *pci_dev = to_pci_dev(dev);
724         const struct dev_pm_ops *pm = dev->driver ? dev->driver->pm : NULL;
725
726         pci_dev->skip_bus_pm = false;
727
728         if (pci_has_legacy_pm_support(pci_dev))
729                 return pci_legacy_suspend(dev, PMSG_SUSPEND);
730
731         if (!pm) {
732                 pci_pm_default_suspend(pci_dev);
733                 return 0;
734         }
735
736         /*
737          * PCI devices suspended at run time may need to be resumed at this
738          * point, because in general it may be necessary to reconfigure them for
739          * system suspend.  Namely, if the device is expected to wake up the
740          * system from the sleep state, it may have to be reconfigured for this
741          * purpose, or if the device is not expected to wake up the system from
742          * the sleep state, it should be prevented from signaling wakeup events
743          * going forward.
744          *
745          * Also if the driver of the device does not indicate that its system
746          * suspend callbacks can cope with runtime-suspended devices, it is
747          * better to resume the device from runtime suspend here.
748          */
749         if (!dev_pm_test_driver_flags(dev, DPM_FLAG_SMART_SUSPEND) ||
750             pci_dev_need_resume(pci_dev)) {
751                 pm_runtime_resume(dev);
752                 pci_dev->state_saved = false;
753         } else {
754                 pci_dev_adjust_pme(pci_dev);
755         }
756
757         if (pm->suspend) {
758                 pci_power_t prev = pci_dev->current_state;
759                 int error;
760
761                 error = pm->suspend(dev);
762                 suspend_report_result(pm->suspend, error);
763                 if (error)
764                         return error;
765
766                 if (!pci_dev->state_saved && pci_dev->current_state != PCI_D0
767                     && pci_dev->current_state != PCI_UNKNOWN) {
768                         pci_WARN_ONCE(pci_dev, pci_dev->current_state != prev,
769                                       "PCI PM: State of device not saved by %pS\n",
770                                       pm->suspend);
771                 }
772         }
773
774         return 0;
775 }
776
777 static int pci_pm_suspend_late(struct device *dev)
778 {
779         if (dev_pm_skip_suspend(dev))
780                 return 0;
781
782         pci_fixup_device(pci_fixup_suspend, to_pci_dev(dev));
783
784         return pm_generic_suspend_late(dev);
785 }
786
787 static int pci_pm_suspend_noirq(struct device *dev)
788 {
789         struct pci_dev *pci_dev = to_pci_dev(dev);
790         const struct dev_pm_ops *pm = dev->driver ? dev->driver->pm : NULL;
791
792         if (dev_pm_skip_suspend(dev))
793                 return 0;
794
795         if (pci_has_legacy_pm_support(pci_dev))
796                 return pci_legacy_suspend_late(dev, PMSG_SUSPEND);
797
798         if (!pm) {
799                 pci_save_state(pci_dev);
800                 goto Fixup;
801         }
802
803         if (pm->suspend_noirq) {
804                 pci_power_t prev = pci_dev->current_state;
805                 int error;
806
807                 error = pm->suspend_noirq(dev);
808                 suspend_report_result(pm->suspend_noirq, error);
809                 if (error)
810                         return error;
811
812                 if (!pci_dev->state_saved && pci_dev->current_state != PCI_D0
813                     && pci_dev->current_state != PCI_UNKNOWN) {
814                         pci_WARN_ONCE(pci_dev, pci_dev->current_state != prev,
815                                       "PCI PM: State of device not saved by %pS\n",
816                                       pm->suspend_noirq);
817                         goto Fixup;
818                 }
819         }
820
821         if (pci_dev->skip_bus_pm) {
822                 /*
823                  * Either the device is a bridge with a child in D0 below it, or
824                  * the function is running for the second time in a row without
825                  * going through full resume, which is possible only during
826                  * suspend-to-idle in a spurious wakeup case.  The device should
827                  * be in D0 at this point, but if it is a bridge, it may be
828                  * necessary to save its state.
829                  */
830                 if (!pci_dev->state_saved)
831                         pci_save_state(pci_dev);
832         } else if (!pci_dev->state_saved) {
833                 pci_save_state(pci_dev);
834                 if (pci_power_manageable(pci_dev))
835                         pci_prepare_to_sleep(pci_dev);
836         }
837
838         pci_dbg(pci_dev, "PCI PM: Suspend power state: %s\n",
839                 pci_power_name(pci_dev->current_state));
840
841         if (pci_dev->current_state == PCI_D0) {
842                 pci_dev->skip_bus_pm = true;
843                 /*
844                  * Per PCI PM r1.2, table 6-1, a bridge must be in D0 if any
845                  * downstream device is in D0, so avoid changing the power state
846                  * of the parent bridge by setting the skip_bus_pm flag for it.
847                  */
848                 if (pci_dev->bus->self)
849                         pci_dev->bus->self->skip_bus_pm = true;
850         }
851
852         if (pci_dev->skip_bus_pm && pm_suspend_no_platform()) {
853                 pci_dbg(pci_dev, "PCI PM: Skipped\n");
854                 goto Fixup;
855         }
856
857         pci_pm_set_unknown_state(pci_dev);
858
859         /*
860          * Some BIOSes from ASUS have a bug: If a USB EHCI host controller's
861          * PCI COMMAND register isn't 0, the BIOS assumes that the controller
862          * hasn't been quiesced and tries to turn it off.  If the controller
863          * is already in D3, this can hang or cause memory corruption.
864          *
865          * Since the value of the COMMAND register doesn't matter once the
866          * device has been suspended, we can safely set it to 0 here.
867          */
868         if (pci_dev->class == PCI_CLASS_SERIAL_USB_EHCI)
869                 pci_write_config_word(pci_dev, PCI_COMMAND, 0);
870
871 Fixup:
872         pci_fixup_device(pci_fixup_suspend_late, pci_dev);
873
874         /*
875          * If the target system sleep state is suspend-to-idle, it is sufficient
876          * to check whether or not the device's wakeup settings are good for
877          * runtime PM.  Otherwise, the pm_resume_via_firmware() check will cause
878          * pci_pm_complete() to take care of fixing up the device's state
879          * anyway, if need be.
880          */
881         if (device_can_wakeup(dev) && !device_may_wakeup(dev))
882                 dev->power.may_skip_resume = false;
883
884         return 0;
885 }
886
887 static int pci_pm_resume_noirq(struct device *dev)
888 {
889         struct pci_dev *pci_dev = to_pci_dev(dev);
890         const struct dev_pm_ops *pm = dev->driver ? dev->driver->pm : NULL;
891         pci_power_t prev_state = pci_dev->current_state;
892         bool skip_bus_pm = pci_dev->skip_bus_pm;
893
894         if (dev_pm_skip_resume(dev))
895                 return 0;
896
897         /*
898          * In the suspend-to-idle case, devices left in D0 during suspend will
899          * stay in D0, so it is not necessary to restore or update their
900          * configuration here and attempting to put them into D0 again is
901          * pointless, so avoid doing that.
902          */
903         if (!(skip_bus_pm && pm_suspend_no_platform()))
904                 pci_pm_default_resume_early(pci_dev);
905
906         pci_fixup_device(pci_fixup_resume_early, pci_dev);
907         pcie_pme_root_status_cleanup(pci_dev);
908
909         if (!skip_bus_pm && prev_state == PCI_D3cold)
910                 pci_bridge_wait_for_secondary_bus(pci_dev);
911
912         if (pci_has_legacy_pm_support(pci_dev))
913                 return 0;
914
915         if (pm && pm->resume_noirq)
916                 return pm->resume_noirq(dev);
917
918         return 0;
919 }
920
921 static int pci_pm_resume_early(struct device *dev)
922 {
923         if (dev_pm_skip_resume(dev))
924                 return 0;
925
926         return pm_generic_resume_early(dev);
927 }
928
929 static int pci_pm_resume(struct device *dev)
930 {
931         struct pci_dev *pci_dev = to_pci_dev(dev);
932         const struct dev_pm_ops *pm = dev->driver ? dev->driver->pm : NULL;
933
934         /*
935          * This is necessary for the suspend error path in which resume is
936          * called without restoring the standard config registers of the device.
937          */
938         if (pci_dev->state_saved)
939                 pci_restore_standard_config(pci_dev);
940
941         if (pci_has_legacy_pm_support(pci_dev))
942                 return pci_legacy_resume(dev);
943
944         pci_pm_default_resume(pci_dev);
945
946         if (pm) {
947                 if (pm->resume)
948                         return pm->resume(dev);
949         } else {
950                 pci_pm_reenable_device(pci_dev);
951         }
952
953         return 0;
954 }
955
956 #else /* !CONFIG_SUSPEND */
957
958 #define pci_pm_suspend          NULL
959 #define pci_pm_suspend_late     NULL
960 #define pci_pm_suspend_noirq    NULL
961 #define pci_pm_resume           NULL
962 #define pci_pm_resume_early     NULL
963 #define pci_pm_resume_noirq     NULL
964
965 #endif /* !CONFIG_SUSPEND */
966
967 #ifdef CONFIG_HIBERNATE_CALLBACKS
968
969 /*
970  * pcibios_pm_ops - provide arch-specific hooks when a PCI device is doing
971  * a hibernate transition
972  */
973 struct dev_pm_ops __weak pcibios_pm_ops;
974
975 static int pci_pm_freeze(struct device *dev)
976 {
977         struct pci_dev *pci_dev = to_pci_dev(dev);
978         const struct dev_pm_ops *pm = dev->driver ? dev->driver->pm : NULL;
979
980         if (pci_has_legacy_pm_support(pci_dev))
981                 return pci_legacy_suspend(dev, PMSG_FREEZE);
982
983         if (!pm) {
984                 pci_pm_default_suspend(pci_dev);
985                 return 0;
986         }
987
988         /*
989          * Resume all runtime-suspended devices before creating a snapshot
990          * image of system memory, because the restore kernel generally cannot
991          * be expected to always handle them consistently and they need to be
992          * put into the runtime-active metastate during system resume anyway,
993          * so it is better to ensure that the state saved in the image will be
994          * always consistent with that.
995          */
996         pm_runtime_resume(dev);
997         pci_dev->state_saved = false;
998
999         if (pm->freeze) {
1000                 int error;
1001
1002                 error = pm->freeze(dev);
1003                 suspend_report_result(pm->freeze, error);
1004                 if (error)
1005                         return error;
1006         }
1007
1008         return 0;
1009 }
1010
1011 static int pci_pm_freeze_noirq(struct device *dev)
1012 {
1013         struct pci_dev *pci_dev = to_pci_dev(dev);
1014         const struct dev_pm_ops *pm = dev->driver ? dev->driver->pm : NULL;
1015
1016         if (pci_has_legacy_pm_support(pci_dev))
1017                 return pci_legacy_suspend_late(dev, PMSG_FREEZE);
1018
1019         if (pm && pm->freeze_noirq) {
1020                 int error;
1021
1022                 error = pm->freeze_noirq(dev);
1023                 suspend_report_result(pm->freeze_noirq, error);
1024                 if (error)
1025                         return error;
1026         }
1027
1028         if (!pci_dev->state_saved)
1029                 pci_save_state(pci_dev);
1030
1031         pci_pm_set_unknown_state(pci_dev);
1032
1033         if (pcibios_pm_ops.freeze_noirq)
1034                 return pcibios_pm_ops.freeze_noirq(dev);
1035
1036         return 0;
1037 }
1038
1039 static int pci_pm_thaw_noirq(struct device *dev)
1040 {
1041         struct pci_dev *pci_dev = to_pci_dev(dev);
1042         const struct dev_pm_ops *pm = dev->driver ? dev->driver->pm : NULL;
1043         int error;
1044
1045         if (pcibios_pm_ops.thaw_noirq) {
1046                 error = pcibios_pm_ops.thaw_noirq(dev);
1047                 if (error)
1048                         return error;
1049         }
1050
1051         /*
1052          * The pm->thaw_noirq() callback assumes the device has been
1053          * returned to D0 and its config state has been restored.
1054          *
1055          * In addition, pci_restore_state() restores MSI-X state in MMIO
1056          * space, which requires the device to be in D0, so return it to D0
1057          * in case the driver's "freeze" callbacks put it into a low-power
1058          * state.
1059          */
1060         pci_set_power_state(pci_dev, PCI_D0);
1061         pci_restore_state(pci_dev);
1062
1063         if (pci_has_legacy_pm_support(pci_dev))
1064                 return 0;
1065
1066         if (pm && pm->thaw_noirq)
1067                 return pm->thaw_noirq(dev);
1068
1069         return 0;
1070 }
1071
1072 static int pci_pm_thaw(struct device *dev)
1073 {
1074         struct pci_dev *pci_dev = to_pci_dev(dev);
1075         const struct dev_pm_ops *pm = dev->driver ? dev->driver->pm : NULL;
1076         int error = 0;
1077
1078         if (pci_has_legacy_pm_support(pci_dev))
1079                 return pci_legacy_resume(dev);
1080
1081         if (pm) {
1082                 if (pm->thaw)
1083                         error = pm->thaw(dev);
1084         } else {
1085                 pci_pm_reenable_device(pci_dev);
1086         }
1087
1088         pci_dev->state_saved = false;
1089
1090         return error;
1091 }
1092
1093 static int pci_pm_poweroff(struct device *dev)
1094 {
1095         struct pci_dev *pci_dev = to_pci_dev(dev);
1096         const struct dev_pm_ops *pm = dev->driver ? dev->driver->pm : NULL;
1097
1098         if (pci_has_legacy_pm_support(pci_dev))
1099                 return pci_legacy_suspend(dev, PMSG_HIBERNATE);
1100
1101         if (!pm) {
1102                 pci_pm_default_suspend(pci_dev);
1103                 return 0;
1104         }
1105
1106         /* The reason to do that is the same as in pci_pm_suspend(). */
1107         if (!dev_pm_test_driver_flags(dev, DPM_FLAG_SMART_SUSPEND) ||
1108             pci_dev_need_resume(pci_dev)) {
1109                 pm_runtime_resume(dev);
1110                 pci_dev->state_saved = false;
1111         } else {
1112                 pci_dev_adjust_pme(pci_dev);
1113         }
1114
1115         if (pm->poweroff) {
1116                 int error;
1117
1118                 error = pm->poweroff(dev);
1119                 suspend_report_result(pm->poweroff, error);
1120                 if (error)
1121                         return error;
1122         }
1123
1124         return 0;
1125 }
1126
1127 static int pci_pm_poweroff_late(struct device *dev)
1128 {
1129         if (dev_pm_skip_suspend(dev))
1130                 return 0;
1131
1132         pci_fixup_device(pci_fixup_suspend, to_pci_dev(dev));
1133
1134         return pm_generic_poweroff_late(dev);
1135 }
1136
1137 static int pci_pm_poweroff_noirq(struct device *dev)
1138 {
1139         struct pci_dev *pci_dev = to_pci_dev(dev);
1140         const struct dev_pm_ops *pm = dev->driver ? dev->driver->pm : NULL;
1141
1142         if (dev_pm_skip_suspend(dev))
1143                 return 0;
1144
1145         if (pci_has_legacy_pm_support(pci_dev))
1146                 return pci_legacy_suspend_late(dev, PMSG_HIBERNATE);
1147
1148         if (!pm) {
1149                 pci_fixup_device(pci_fixup_suspend_late, pci_dev);
1150                 return 0;
1151         }
1152
1153         if (pm->poweroff_noirq) {
1154                 int error;
1155
1156                 error = pm->poweroff_noirq(dev);
1157                 suspend_report_result(pm->poweroff_noirq, error);
1158                 if (error)
1159                         return error;
1160         }
1161
1162         if (!pci_dev->state_saved && !pci_has_subordinate(pci_dev))
1163                 pci_prepare_to_sleep(pci_dev);
1164
1165         /*
1166          * The reason for doing this here is the same as for the analogous code
1167          * in pci_pm_suspend_noirq().
1168          */
1169         if (pci_dev->class == PCI_CLASS_SERIAL_USB_EHCI)
1170                 pci_write_config_word(pci_dev, PCI_COMMAND, 0);
1171
1172         pci_fixup_device(pci_fixup_suspend_late, pci_dev);
1173
1174         if (pcibios_pm_ops.poweroff_noirq)
1175                 return pcibios_pm_ops.poweroff_noirq(dev);
1176
1177         return 0;
1178 }
1179
1180 static int pci_pm_restore_noirq(struct device *dev)
1181 {
1182         struct pci_dev *pci_dev = to_pci_dev(dev);
1183         const struct dev_pm_ops *pm = dev->driver ? dev->driver->pm : NULL;
1184         int error;
1185
1186         if (pcibios_pm_ops.restore_noirq) {
1187                 error = pcibios_pm_ops.restore_noirq(dev);
1188                 if (error)
1189                         return error;
1190         }
1191
1192         pci_pm_default_resume_early(pci_dev);
1193         pci_fixup_device(pci_fixup_resume_early, pci_dev);
1194
1195         if (pci_has_legacy_pm_support(pci_dev))
1196                 return 0;
1197
1198         if (pm && pm->restore_noirq)
1199                 return pm->restore_noirq(dev);
1200
1201         return 0;
1202 }
1203
1204 static int pci_pm_restore(struct device *dev)
1205 {
1206         struct pci_dev *pci_dev = to_pci_dev(dev);
1207         const struct dev_pm_ops *pm = dev->driver ? dev->driver->pm : NULL;
1208
1209         /*
1210          * This is necessary for the hibernation error path in which restore is
1211          * called without restoring the standard config registers of the device.
1212          */
1213         if (pci_dev->state_saved)
1214                 pci_restore_standard_config(pci_dev);
1215
1216         if (pci_has_legacy_pm_support(pci_dev))
1217                 return pci_legacy_resume(dev);
1218
1219         pci_pm_default_resume(pci_dev);
1220
1221         if (pm) {
1222                 if (pm->restore)
1223                         return pm->restore(dev);
1224         } else {
1225                 pci_pm_reenable_device(pci_dev);
1226         }
1227
1228         return 0;
1229 }
1230
1231 #else /* !CONFIG_HIBERNATE_CALLBACKS */
1232
1233 #define pci_pm_freeze           NULL
1234 #define pci_pm_freeze_noirq     NULL
1235 #define pci_pm_thaw             NULL
1236 #define pci_pm_thaw_noirq       NULL
1237 #define pci_pm_poweroff         NULL
1238 #define pci_pm_poweroff_late    NULL
1239 #define pci_pm_poweroff_noirq   NULL
1240 #define pci_pm_restore          NULL
1241 #define pci_pm_restore_noirq    NULL
1242
1243 #endif /* !CONFIG_HIBERNATE_CALLBACKS */
1244
1245 #ifdef CONFIG_PM
1246
1247 static int pci_pm_runtime_suspend(struct device *dev)
1248 {
1249         struct pci_dev *pci_dev = to_pci_dev(dev);
1250         const struct dev_pm_ops *pm = dev->driver ? dev->driver->pm : NULL;
1251         pci_power_t prev = pci_dev->current_state;
1252         int error;
1253
1254         /*
1255          * If pci_dev->driver is not set (unbound), we leave the device in D0,
1256          * but it may go to D3cold when the bridge above it runtime suspends.
1257          * Save its config space in case that happens.
1258          */
1259         if (!pci_dev->driver) {
1260                 pci_save_state(pci_dev);
1261                 return 0;
1262         }
1263
1264         pci_dev->state_saved = false;
1265         if (pm && pm->runtime_suspend) {
1266                 error = pm->runtime_suspend(dev);
1267                 /*
1268                  * -EBUSY and -EAGAIN is used to request the runtime PM core
1269                  * to schedule a new suspend, so log the event only with debug
1270                  * log level.
1271                  */
1272                 if (error == -EBUSY || error == -EAGAIN) {
1273                         pci_dbg(pci_dev, "can't suspend now (%ps returned %d)\n",
1274                                 pm->runtime_suspend, error);
1275                         return error;
1276                 } else if (error) {
1277                         pci_err(pci_dev, "can't suspend (%ps returned %d)\n",
1278                                 pm->runtime_suspend, error);
1279                         return error;
1280                 }
1281         }
1282
1283         pci_fixup_device(pci_fixup_suspend, pci_dev);
1284
1285         if (pm && pm->runtime_suspend
1286             && !pci_dev->state_saved && pci_dev->current_state != PCI_D0
1287             && pci_dev->current_state != PCI_UNKNOWN) {
1288                 pci_WARN_ONCE(pci_dev, pci_dev->current_state != prev,
1289                               "PCI PM: State of device not saved by %pS\n",
1290                               pm->runtime_suspend);
1291                 return 0;
1292         }
1293
1294         if (!pci_dev->state_saved) {
1295                 pci_save_state(pci_dev);
1296                 pci_finish_runtime_suspend(pci_dev);
1297         }
1298
1299         return 0;
1300 }
1301
1302 static int pci_pm_runtime_resume(struct device *dev)
1303 {
1304         struct pci_dev *pci_dev = to_pci_dev(dev);
1305         const struct dev_pm_ops *pm = dev->driver ? dev->driver->pm : NULL;
1306         pci_power_t prev_state = pci_dev->current_state;
1307         int error = 0;
1308
1309         /*
1310          * Restoring config space is necessary even if the device is not bound
1311          * to a driver because although we left it in D0, it may have gone to
1312          * D3cold when the bridge above it runtime suspended.
1313          */
1314         pci_restore_standard_config(pci_dev);
1315
1316         if (!pci_dev->driver)
1317                 return 0;
1318
1319         pci_fixup_device(pci_fixup_resume_early, pci_dev);
1320         pci_pm_default_resume(pci_dev);
1321
1322         if (prev_state == PCI_D3cold)
1323                 pci_bridge_wait_for_secondary_bus(pci_dev);
1324
1325         if (pm && pm->runtime_resume)
1326                 error = pm->runtime_resume(dev);
1327
1328         pci_dev->runtime_d3cold = false;
1329
1330         return error;
1331 }
1332
1333 static int pci_pm_runtime_idle(struct device *dev)
1334 {
1335         struct pci_dev *pci_dev = to_pci_dev(dev);
1336         const struct dev_pm_ops *pm = dev->driver ? dev->driver->pm : NULL;
1337
1338         /*
1339          * If pci_dev->driver is not set (unbound), the device should
1340          * always remain in D0 regardless of the runtime PM status
1341          */
1342         if (!pci_dev->driver)
1343                 return 0;
1344
1345         if (!pm)
1346                 return -ENOSYS;
1347
1348         if (pm->runtime_idle)
1349                 return pm->runtime_idle(dev);
1350
1351         return 0;
1352 }
1353
1354 static const struct dev_pm_ops pci_dev_pm_ops = {
1355         .prepare = pci_pm_prepare,
1356         .complete = pci_pm_complete,
1357         .suspend = pci_pm_suspend,
1358         .suspend_late = pci_pm_suspend_late,
1359         .resume = pci_pm_resume,
1360         .resume_early = pci_pm_resume_early,
1361         .freeze = pci_pm_freeze,
1362         .thaw = pci_pm_thaw,
1363         .poweroff = pci_pm_poweroff,
1364         .poweroff_late = pci_pm_poweroff_late,
1365         .restore = pci_pm_restore,
1366         .suspend_noirq = pci_pm_suspend_noirq,
1367         .resume_noirq = pci_pm_resume_noirq,
1368         .freeze_noirq = pci_pm_freeze_noirq,
1369         .thaw_noirq = pci_pm_thaw_noirq,
1370         .poweroff_noirq = pci_pm_poweroff_noirq,
1371         .restore_noirq = pci_pm_restore_noirq,
1372         .runtime_suspend = pci_pm_runtime_suspend,
1373         .runtime_resume = pci_pm_runtime_resume,
1374         .runtime_idle = pci_pm_runtime_idle,
1375 };
1376
1377 #define PCI_PM_OPS_PTR  (&pci_dev_pm_ops)
1378
1379 #else /* !CONFIG_PM */
1380
1381 #define pci_pm_runtime_suspend  NULL
1382 #define pci_pm_runtime_resume   NULL
1383 #define pci_pm_runtime_idle     NULL
1384
1385 #define PCI_PM_OPS_PTR  NULL
1386
1387 #endif /* !CONFIG_PM */
1388
1389 /**
1390  * __pci_register_driver - register a new pci driver
1391  * @drv: the driver structure to register
1392  * @owner: owner module of drv
1393  * @mod_name: module name string
1394  *
1395  * Adds the driver structure to the list of registered drivers.
1396  * Returns a negative value on error, otherwise 0.
1397  * If no error occurred, the driver remains registered even if
1398  * no device was claimed during registration.
1399  */
1400 int __pci_register_driver(struct pci_driver *drv, struct module *owner,
1401                           const char *mod_name)
1402 {
1403         /* initialize common driver fields */
1404         drv->driver.name = drv->name;
1405         drv->driver.bus = &pci_bus_type;
1406         drv->driver.owner = owner;
1407         drv->driver.mod_name = mod_name;
1408         drv->driver.groups = drv->groups;
1409
1410         spin_lock_init(&drv->dynids.lock);
1411         INIT_LIST_HEAD(&drv->dynids.list);
1412
1413         /* register with core */
1414         return driver_register(&drv->driver);
1415 }
1416 EXPORT_SYMBOL(__pci_register_driver);
1417
1418 /**
1419  * pci_unregister_driver - unregister a pci driver
1420  * @drv: the driver structure to unregister
1421  *
1422  * Deletes the driver structure from the list of registered PCI drivers,
1423  * gives it a chance to clean up by calling its remove() function for
1424  * each device it was responsible for, and marks those devices as
1425  * driverless.
1426  */
1427
1428 void pci_unregister_driver(struct pci_driver *drv)
1429 {
1430         driver_unregister(&drv->driver);
1431         pci_free_dynids(drv);
1432 }
1433 EXPORT_SYMBOL(pci_unregister_driver);
1434
1435 static struct pci_driver pci_compat_driver = {
1436         .name = "compat"
1437 };
1438
1439 /**
1440  * pci_dev_driver - get the pci_driver of a device
1441  * @dev: the device to query
1442  *
1443  * Returns the appropriate pci_driver structure or %NULL if there is no
1444  * registered driver for the device.
1445  */
1446 struct pci_driver *pci_dev_driver(const struct pci_dev *dev)
1447 {
1448         if (dev->driver)
1449                 return dev->driver;
1450         else {
1451                 int i;
1452                 for (i = 0; i <= PCI_ROM_RESOURCE; i++)
1453                         if (dev->resource[i].flags & IORESOURCE_BUSY)
1454                                 return &pci_compat_driver;
1455         }
1456         return NULL;
1457 }
1458 EXPORT_SYMBOL(pci_dev_driver);
1459
1460 /**
1461  * pci_bus_match - Tell if a PCI device structure has a matching PCI device id structure
1462  * @dev: the PCI device structure to match against
1463  * @drv: the device driver to search for matching PCI device id structures
1464  *
1465  * Used by a driver to check whether a PCI device present in the
1466  * system is in its list of supported devices. Returns the matching
1467  * pci_device_id structure or %NULL if there is no match.
1468  */
1469 static int pci_bus_match(struct device *dev, struct device_driver *drv)
1470 {
1471         struct pci_dev *pci_dev = to_pci_dev(dev);
1472         struct pci_driver *pci_drv;
1473         const struct pci_device_id *found_id;
1474
1475         if (!pci_dev->match_driver)
1476                 return 0;
1477
1478         pci_drv = to_pci_driver(drv);
1479         found_id = pci_match_device(pci_drv, pci_dev);
1480         if (found_id)
1481                 return 1;
1482
1483         return 0;
1484 }
1485
1486 /**
1487  * pci_dev_get - increments the reference count of the pci device structure
1488  * @dev: the device being referenced
1489  *
1490  * Each live reference to a device should be refcounted.
1491  *
1492  * Drivers for PCI devices should normally record such references in
1493  * their probe() methods, when they bind to a device, and release
1494  * them by calling pci_dev_put(), in their disconnect() methods.
1495  *
1496  * A pointer to the device with the incremented reference counter is returned.
1497  */
1498 struct pci_dev *pci_dev_get(struct pci_dev *dev)
1499 {
1500         if (dev)
1501                 get_device(&dev->dev);
1502         return dev;
1503 }
1504 EXPORT_SYMBOL(pci_dev_get);
1505
1506 /**
1507  * pci_dev_put - release a use of the pci device structure
1508  * @dev: device that's been disconnected
1509  *
1510  * Must be called when a user of a device is finished with it.  When the last
1511  * user of the device calls this function, the memory of the device is freed.
1512  */
1513 void pci_dev_put(struct pci_dev *dev)
1514 {
1515         if (dev)
1516                 put_device(&dev->dev);
1517 }
1518 EXPORT_SYMBOL(pci_dev_put);
1519
1520 static int pci_uevent(struct device *dev, struct kobj_uevent_env *env)
1521 {
1522         struct pci_dev *pdev;
1523
1524         if (!dev)
1525                 return -ENODEV;
1526
1527         pdev = to_pci_dev(dev);
1528
1529         if (add_uevent_var(env, "PCI_CLASS=%04X", pdev->class))
1530                 return -ENOMEM;
1531
1532         if (add_uevent_var(env, "PCI_ID=%04X:%04X", pdev->vendor, pdev->device))
1533                 return -ENOMEM;
1534
1535         if (add_uevent_var(env, "PCI_SUBSYS_ID=%04X:%04X", pdev->subsystem_vendor,
1536                            pdev->subsystem_device))
1537                 return -ENOMEM;
1538
1539         if (add_uevent_var(env, "PCI_SLOT_NAME=%s", pci_name(pdev)))
1540                 return -ENOMEM;
1541
1542         if (add_uevent_var(env, "MODALIAS=pci:v%08Xd%08Xsv%08Xsd%08Xbc%02Xsc%02Xi%02X",
1543                            pdev->vendor, pdev->device,
1544                            pdev->subsystem_vendor, pdev->subsystem_device,
1545                            (u8)(pdev->class >> 16), (u8)(pdev->class >> 8),
1546                            (u8)(pdev->class)))
1547                 return -ENOMEM;
1548
1549         return 0;
1550 }
1551
1552 #if defined(CONFIG_PCIEPORTBUS) || defined(CONFIG_EEH)
1553 /**
1554  * pci_uevent_ers - emit a uevent during recovery path of PCI device
1555  * @pdev: PCI device undergoing error recovery
1556  * @err_type: type of error event
1557  */
1558 void pci_uevent_ers(struct pci_dev *pdev, enum pci_ers_result err_type)
1559 {
1560         int idx = 0;
1561         char *envp[3];
1562
1563         switch (err_type) {
1564         case PCI_ERS_RESULT_NONE:
1565         case PCI_ERS_RESULT_CAN_RECOVER:
1566                 envp[idx++] = "ERROR_EVENT=BEGIN_RECOVERY";
1567                 envp[idx++] = "DEVICE_ONLINE=0";
1568                 break;
1569         case PCI_ERS_RESULT_RECOVERED:
1570                 envp[idx++] = "ERROR_EVENT=SUCCESSFUL_RECOVERY";
1571                 envp[idx++] = "DEVICE_ONLINE=1";
1572                 break;
1573         case PCI_ERS_RESULT_DISCONNECT:
1574                 envp[idx++] = "ERROR_EVENT=FAILED_RECOVERY";
1575                 envp[idx++] = "DEVICE_ONLINE=0";
1576                 break;
1577         default:
1578                 break;
1579         }
1580
1581         if (idx > 0) {
1582                 envp[idx++] = NULL;
1583                 kobject_uevent_env(&pdev->dev.kobj, KOBJ_CHANGE, envp);
1584         }
1585 }
1586 #endif
1587
1588 static int pci_bus_num_vf(struct device *dev)
1589 {
1590         return pci_num_vf(to_pci_dev(dev));
1591 }
1592
1593 /**
1594  * pci_dma_configure - Setup DMA configuration
1595  * @dev: ptr to dev structure
1596  *
1597  * Function to update PCI devices's DMA configuration using the same
1598  * info from the OF node or ACPI node of host bridge's parent (if any).
1599  */
1600 static int pci_dma_configure(struct device *dev)
1601 {
1602         struct device *bridge;
1603         int ret = 0;
1604
1605         bridge = pci_get_host_bridge_device(to_pci_dev(dev));
1606
1607         if (IS_ENABLED(CONFIG_OF) && bridge->parent &&
1608             bridge->parent->of_node) {
1609                 ret = of_dma_configure(dev, bridge->parent->of_node, true);
1610         } else if (has_acpi_companion(bridge)) {
1611                 struct acpi_device *adev = to_acpi_device_node(bridge->fwnode);
1612
1613                 ret = acpi_dma_configure(dev, acpi_get_dma_attr(adev));
1614         }
1615
1616         pci_put_host_bridge_device(bridge);
1617         return ret;
1618 }
1619
1620 struct bus_type pci_bus_type = {
1621         .name           = "pci",
1622         .match          = pci_bus_match,
1623         .uevent         = pci_uevent,
1624         .probe          = pci_device_probe,
1625         .remove         = pci_device_remove,
1626         .shutdown       = pci_device_shutdown,
1627         .dev_groups     = pci_dev_groups,
1628         .bus_groups     = pci_bus_groups,
1629         .drv_groups     = pci_drv_groups,
1630         .pm             = PCI_PM_OPS_PTR,
1631         .num_vf         = pci_bus_num_vf,
1632         .dma_configure  = pci_dma_configure,
1633 };
1634 EXPORT_SYMBOL(pci_bus_type);
1635
1636 #ifdef CONFIG_PCIEPORTBUS
1637 static int pcie_port_bus_match(struct device *dev, struct device_driver *drv)
1638 {
1639         struct pcie_device *pciedev;
1640         struct pcie_port_service_driver *driver;
1641
1642         if (drv->bus != &pcie_port_bus_type || dev->bus != &pcie_port_bus_type)
1643                 return 0;
1644
1645         pciedev = to_pcie_device(dev);
1646         driver = to_service_driver(drv);
1647
1648         if (driver->service != pciedev->service)
1649                 return 0;
1650
1651         if (driver->port_type != PCIE_ANY_PORT &&
1652             driver->port_type != pci_pcie_type(pciedev->port))
1653                 return 0;
1654
1655         return 1;
1656 }
1657
1658 struct bus_type pcie_port_bus_type = {
1659         .name           = "pci_express",
1660         .match          = pcie_port_bus_match,
1661 };
1662 EXPORT_SYMBOL_GPL(pcie_port_bus_type);
1663 #endif
1664
1665 static int __init pci_driver_init(void)
1666 {
1667         int ret;
1668
1669         ret = bus_register(&pci_bus_type);
1670         if (ret)
1671                 return ret;
1672
1673 #ifdef CONFIG_PCIEPORTBUS
1674         ret = bus_register(&pcie_port_bus_type);
1675         if (ret)
1676                 return ret;
1677 #endif
1678         dma_debug_add_bus(&pci_bus_type);
1679         return 0;
1680 }
1681 postcore_initcall(pci_driver_init);