1 /*
2  * PCI Stub Driver - Grabs devices in backend to be exported later
3  *
4  * Ryan Wilson <hap9@epoch.ncsc.mil>
5  * Chris Bookholt <hap10@epoch.ncsc.mil>
6  */
7 #include <linux/module.h>
8 #include <linux/init.h>
9 #include <linux/rwsem.h>
10 #include <linux/list.h>
11 #include <linux/spinlock.h>
12 #include <linux/kref.h>
13 #include <linux/pci.h>
14 #include <linux/wait.h>
15 #include <linux/sched.h>
16 #include <linux/atomic.h>
17 #include <xen/events.h>
18 #include <asm/xen/pci.h>
19 #include <asm/xen/hypervisor.h>
20 #include "pciback.h"
21 #include "conf_space.h"
22 #include "conf_space_quirks.h"
23 
24 #define DRV_NAME	"xen-pciback"
25 
26 static char *pci_devs_to_hide;
27 wait_queue_head_t xen_pcibk_aer_wait_queue;
28 /*Add sem for sync AER handling and xen_pcibk remove/reconfigue ops,
29 * We want to avoid in middle of AER ops, xen_pcibk devices is being removed
30 */
31 static DECLARE_RWSEM(pcistub_sem);
32 module_param_named(hide, pci_devs_to_hide, charp, 0444);
33 
34 struct pcistub_device_id {
35 	struct list_head slot_list;
36 	int domain;
37 	unsigned char bus;
38 	unsigned int devfn;
39 };
40 static LIST_HEAD(pcistub_device_ids);
41 static DEFINE_SPINLOCK(device_ids_lock);
42 
43 struct pcistub_device {
44 	struct kref kref;
45 	struct list_head dev_list;
46 	spinlock_t lock;
47 
48 	struct pci_dev *dev;
49 	struct xen_pcibk_device *pdev;/* non-NULL if struct pci_dev is in use */
50 };
51 
52 /* Access to pcistub_devices & seized_devices lists and the initialize_devices
53  * flag must be locked with pcistub_devices_lock
54  */
55 static DEFINE_SPINLOCK(pcistub_devices_lock);
56 static LIST_HEAD(pcistub_devices);
57 
58 /* wait for device_initcall before initializing our devices
59  * (see pcistub_init_devices_late)
60  */
61 static int initialize_devices;
62 static LIST_HEAD(seized_devices);
63 
64 static struct pcistub_device *pcistub_device_alloc(struct pci_dev *dev)
65 {
66 	struct pcistub_device *psdev;
67 
68 	dev_dbg(&dev->dev, "pcistub_device_alloc\n");
69 
70 	psdev = kzalloc(sizeof(*psdev), GFP_ATOMIC);
71 	if (!psdev)
72 		return NULL;
73 
74 	psdev->dev = pci_dev_get(dev);
75 	if (!psdev->dev) {
76 		kfree(psdev);
77 		return NULL;
78 	}
79 
80 	kref_init(&psdev->kref);
81 	spin_lock_init(&psdev->lock);
82 
83 	return psdev;
84 }
85 
86 /* Don't call this directly as it's called by pcistub_device_put */
87 static void pcistub_device_release(struct kref *kref)
88 {
89 	struct pcistub_device *psdev;
90 
91 	psdev = container_of(kref, struct pcistub_device, kref);
92 
93 	dev_dbg(&psdev->dev->dev, "pcistub_device_release\n");
94 
95 	xen_unregister_device_domain_owner(psdev->dev);
96 
97 	/* Clean-up the device */
98 	xen_pcibk_reset_device(psdev->dev);
99 	xen_pcibk_config_free_dyn_fields(psdev->dev);
100 	xen_pcibk_config_free_dev(psdev->dev);
101 	kfree(pci_get_drvdata(psdev->dev));
102 	pci_set_drvdata(psdev->dev, NULL);
103 
104 	pci_dev_put(psdev->dev);
105 
106 	kfree(psdev);
107 }
108 
109 static inline void pcistub_device_get(struct pcistub_device *psdev)
110 {
111 	kref_get(&psdev->kref);
112 }
113 
114 static inline void pcistub_device_put(struct pcistub_device *psdev)
115 {
116 	kref_put(&psdev->kref, pcistub_device_release);
117 }
118 
119 static struct pcistub_device *pcistub_device_find(int domain, int bus,
120 						  int slot, int func)
121 {
122 	struct pcistub_device *psdev = NULL;
123 	unsigned long flags;
124 
125 	spin_lock_irqsave(&pcistub_devices_lock, flags);
126 
127 	list_for_each_entry(psdev, &pcistub_devices, dev_list) {
128 		if (psdev->dev != NULL
129 		    && domain == pci_domain_nr(psdev->dev->bus)
130 		    && bus == psdev->dev->bus->number
131 		    && PCI_DEVFN(slot, func) == psdev->dev->devfn) {
132 			pcistub_device_get(psdev);
133 			goto out;
134 		}
135 	}
136 
137 	/* didn't find it */
138 	psdev = NULL;
139 
140 out:
141 	spin_unlock_irqrestore(&pcistub_devices_lock, flags);
142 	return psdev;
143 }
144 
145 static struct pci_dev *pcistub_device_get_pci_dev(struct xen_pcibk_device *pdev,
146 						  struct pcistub_device *psdev)
147 {
148 	struct pci_dev *pci_dev = NULL;
149 	unsigned long flags;
150 
151 	pcistub_device_get(psdev);
152 
153 	spin_lock_irqsave(&psdev->lock, flags);
154 	if (!psdev->pdev) {
155 		psdev->pdev = pdev;
156 		pci_dev = psdev->dev;
157 	}
158 	spin_unlock_irqrestore(&psdev->lock, flags);
159 
160 	if (!pci_dev)
161 		pcistub_device_put(psdev);
162 
163 	return pci_dev;
164 }
165 
166 struct pci_dev *pcistub_get_pci_dev_by_slot(struct xen_pcibk_device *pdev,
167 					    int domain, int bus,
168 					    int slot, int func)
169 {
170 	struct pcistub_device *psdev;
171 	struct pci_dev *found_dev = NULL;
172 	unsigned long flags;
173 
174 	spin_lock_irqsave(&pcistub_devices_lock, flags);
175 
176 	list_for_each_entry(psdev, &pcistub_devices, dev_list) {
177 		if (psdev->dev != NULL
178 		    && domain == pci_domain_nr(psdev->dev->bus)
179 		    && bus == psdev->dev->bus->number
180 		    && PCI_DEVFN(slot, func) == psdev->dev->devfn) {
181 			found_dev = pcistub_device_get_pci_dev(pdev, psdev);
182 			break;
183 		}
184 	}
185 
186 	spin_unlock_irqrestore(&pcistub_devices_lock, flags);
187 	return found_dev;
188 }
189 
190 struct pci_dev *pcistub_get_pci_dev(struct xen_pcibk_device *pdev,
191 				    struct pci_dev *dev)
192 {
193 	struct pcistub_device *psdev;
194 	struct pci_dev *found_dev = NULL;
195 	unsigned long flags;
196 
197 	spin_lock_irqsave(&pcistub_devices_lock, flags);
198 
199 	list_for_each_entry(psdev, &pcistub_devices, dev_list) {
200 		if (psdev->dev == dev) {
201 			found_dev = pcistub_device_get_pci_dev(pdev, psdev);
202 			break;
203 		}
204 	}
205 
206 	spin_unlock_irqrestore(&pcistub_devices_lock, flags);
207 	return found_dev;
208 }
209 
210 void pcistub_put_pci_dev(struct pci_dev *dev)
211 {
212 	struct pcistub_device *psdev, *found_psdev = NULL;
213 	unsigned long flags;
214 
215 	spin_lock_irqsave(&pcistub_devices_lock, flags);
216 
217 	list_for_each_entry(psdev, &pcistub_devices, dev_list) {
218 		if (psdev->dev == dev) {
219 			found_psdev = psdev;
220 			break;
221 		}
222 	}
223 
224 	spin_unlock_irqrestore(&pcistub_devices_lock, flags);
225 
226 	/*hold this lock for avoiding breaking link between
227 	* pcistub and xen_pcibk when AER is in processing
228 	*/
229 	down_write(&pcistub_sem);
230 	/* Cleanup our device
231 	 * (so it's ready for the next domain)
232 	 */
233 	xen_pcibk_reset_device(found_psdev->dev);
234 	xen_pcibk_config_free_dyn_fields(found_psdev->dev);
235 	xen_pcibk_config_reset_dev(found_psdev->dev);
236 
237 	spin_lock_irqsave(&found_psdev->lock, flags);
238 	found_psdev->pdev = NULL;
239 	spin_unlock_irqrestore(&found_psdev->lock, flags);
240 
241 	pcistub_device_put(found_psdev);
242 	up_write(&pcistub_sem);
243 }
244 
245 static int __devinit pcistub_match_one(struct pci_dev *dev,
246 				       struct pcistub_device_id *pdev_id)
247 {
248 	/* Match the specified device by domain, bus, slot, func and also if
249 	 * any of the device's parent bridges match.
250 	 */
251 	for (; dev != NULL; dev = dev->bus->self) {
252 		if (pci_domain_nr(dev->bus) == pdev_id->domain
253 		    && dev->bus->number == pdev_id->bus
254 		    && dev->devfn == pdev_id->devfn)
255 			return 1;
256 
257 		/* Sometimes topmost bridge links to itself. */
258 		if (dev == dev->bus->self)
259 			break;
260 	}
261 
262 	return 0;
263 }
264 
265 static int __devinit pcistub_match(struct pci_dev *dev)
266 {
267 	struct pcistub_device_id *pdev_id;
268 	unsigned long flags;
269 	int found = 0;
270 
271 	spin_lock_irqsave(&device_ids_lock, flags);
272 	list_for_each_entry(pdev_id, &pcistub_device_ids, slot_list) {
273 		if (pcistub_match_one(dev, pdev_id)) {
274 			found = 1;
275 			break;
276 		}
277 	}
278 	spin_unlock_irqrestore(&device_ids_lock, flags);
279 
280 	return found;
281 }
282 
283 static int __devinit pcistub_init_device(struct pci_dev *dev)
284 {
285 	struct xen_pcibk_dev_data *dev_data;
286 	int err = 0;
287 
288 	dev_dbg(&dev->dev, "initializing...\n");
289 
290 	/* The PCI backend is not intended to be a module (or to work with
291 	 * removable PCI devices (yet). If it were, xen_pcibk_config_free()
292 	 * would need to be called somewhere to free the memory allocated
293 	 * here and then to call kfree(pci_get_drvdata(psdev->dev)).
294 	 */
295 	dev_data = kzalloc(sizeof(*dev_data) +  strlen(DRV_NAME "[]")
296 				+ strlen(pci_name(dev)) + 1, GFP_ATOMIC);
297 	if (!dev_data) {
298 		err = -ENOMEM;
299 		goto out;
300 	}
301 	pci_set_drvdata(dev, dev_data);
302 
303 	/*
304 	 * Setup name for fake IRQ handler. It will only be enabled
305 	 * once the device is turned on by the guest.
306 	 */
307 	sprintf(dev_data->irq_name, DRV_NAME "[%s]", pci_name(dev));
308 
309 	dev_dbg(&dev->dev, "initializing config\n");
310 
311 	init_waitqueue_head(&xen_pcibk_aer_wait_queue);
312 	err = xen_pcibk_config_init_dev(dev);
313 	if (err)
314 		goto out;
315 
316 	/* HACK: Force device (& ACPI) to determine what IRQ it's on - we
317 	 * must do this here because pcibios_enable_device may specify
318 	 * the pci device's true irq (and possibly its other resources)
319 	 * if they differ from what's in the configuration space.
320 	 * This makes the assumption that the device's resources won't
321 	 * change after this point (otherwise this code may break!)
322 	 */
323 	dev_dbg(&dev->dev, "enabling device\n");
324 	err = pci_enable_device(dev);
325 	if (err)
326 		goto config_release;
327 
328 	/* Now disable the device (this also ensures some private device
329 	 * data is setup before we export)
330 	 */
331 	dev_dbg(&dev->dev, "reset device\n");
332 	xen_pcibk_reset_device(dev);
333 
334 	return 0;
335 
336 config_release:
337 	xen_pcibk_config_free_dev(dev);
338 
339 out:
340 	pci_set_drvdata(dev, NULL);
341 	kfree(dev_data);
342 	return err;
343 }
344 
345 /*
346  * Because some initialization still happens on
347  * devices during fs_initcall, we need to defer
348  * full initialization of our devices until
349  * device_initcall.
350  */
351 static int __init pcistub_init_devices_late(void)
352 {
353 	struct pcistub_device *psdev;
354 	unsigned long flags;
355 	int err = 0;
356 
357 	pr_debug(DRV_NAME ": pcistub_init_devices_late\n");
358 
359 	spin_lock_irqsave(&pcistub_devices_lock, flags);
360 
361 	while (!list_empty(&seized_devices)) {
362 		psdev = container_of(seized_devices.next,
363 				     struct pcistub_device, dev_list);
364 		list_del(&psdev->dev_list);
365 
366 		spin_unlock_irqrestore(&pcistub_devices_lock, flags);
367 
368 		err = pcistub_init_device(psdev->dev);
369 		if (err) {
370 			dev_err(&psdev->dev->dev,
371 				"error %d initializing device\n", err);
372 			kfree(psdev);
373 			psdev = NULL;
374 		}
375 
376 		spin_lock_irqsave(&pcistub_devices_lock, flags);
377 
378 		if (psdev)
379 			list_add_tail(&psdev->dev_list, &pcistub_devices);
380 	}
381 
382 	initialize_devices = 1;
383 
384 	spin_unlock_irqrestore(&pcistub_devices_lock, flags);
385 
386 	return 0;
387 }
388 
389 static int __devinit pcistub_seize(struct pci_dev *dev)
390 {
391 	struct pcistub_device *psdev;
392 	unsigned long flags;
393 	int err = 0;
394 
395 	psdev = pcistub_device_alloc(dev);
396 	if (!psdev)
397 		return -ENOMEM;
398 
399 	spin_lock_irqsave(&pcistub_devices_lock, flags);
400 
401 	if (initialize_devices) {
402 		spin_unlock_irqrestore(&pcistub_devices_lock, flags);
403 
404 		/* don't want irqs disabled when calling pcistub_init_device */
405 		err = pcistub_init_device(psdev->dev);
406 
407 		spin_lock_irqsave(&pcistub_devices_lock, flags);
408 
409 		if (!err)
410 			list_add(&psdev->dev_list, &pcistub_devices);
411 	} else {
412 		dev_dbg(&dev->dev, "deferring initialization\n");
413 		list_add(&psdev->dev_list, &seized_devices);
414 	}
415 
416 	spin_unlock_irqrestore(&pcistub_devices_lock, flags);
417 
418 	if (err)
419 		pcistub_device_put(psdev);
420 
421 	return err;
422 }
423 
424 static int __devinit pcistub_probe(struct pci_dev *dev,
425 				   const struct pci_device_id *id)
426 {
427 	int err = 0;
428 
429 	dev_dbg(&dev->dev, "probing...\n");
430 
431 	if (pcistub_match(dev)) {
432 
433 		if (dev->hdr_type != PCI_HEADER_TYPE_NORMAL
434 		    && dev->hdr_type != PCI_HEADER_TYPE_BRIDGE) {
435 			dev_err(&dev->dev, "can't export pci devices that "
436 				"don't have a normal (0) or bridge (1) "
437 				"header type!\n");
438 			err = -ENODEV;
439 			goto out;
440 		}
441 
442 		dev_info(&dev->dev, "seizing device\n");
443 		err = pcistub_seize(dev);
444 	} else
445 		/* Didn't find the device */
446 		err = -ENODEV;
447 
448 out:
449 	return err;
450 }
451 
452 static void pcistub_remove(struct pci_dev *dev)
453 {
454 	struct pcistub_device *psdev, *found_psdev = NULL;
455 	unsigned long flags;
456 
457 	dev_dbg(&dev->dev, "removing\n");
458 
459 	spin_lock_irqsave(&pcistub_devices_lock, flags);
460 
461 	xen_pcibk_config_quirk_release(dev);
462 
463 	list_for_each_entry(psdev, &pcistub_devices, dev_list) {
464 		if (psdev->dev == dev) {
465 			found_psdev = psdev;
466 			break;
467 		}
468 	}
469 
470 	spin_unlock_irqrestore(&pcistub_devices_lock, flags);
471 
472 	if (found_psdev) {
473 		dev_dbg(&dev->dev, "found device to remove - in use? %p\n",
474 			found_psdev->pdev);
475 
476 		if (found_psdev->pdev) {
477 			printk(KERN_WARNING DRV_NAME ": ****** removing device "
478 			       "%s while still in-use! ******\n",
479 			       pci_name(found_psdev->dev));
480 			printk(KERN_WARNING DRV_NAME ": ****** driver domain may"
481 			       " still access this device's i/o resources!\n");
482 			printk(KERN_WARNING DRV_NAME ": ****** shutdown driver "
483 			       "domain before binding device\n");
484 			printk(KERN_WARNING DRV_NAME ": ****** to other drivers "
485 			       "or domains\n");
486 
487 			xen_pcibk_release_pci_dev(found_psdev->pdev,
488 						found_psdev->dev);
489 		}
490 
491 		spin_lock_irqsave(&pcistub_devices_lock, flags);
492 		list_del(&found_psdev->dev_list);
493 		spin_unlock_irqrestore(&pcistub_devices_lock, flags);
494 
495 		/* the final put for releasing from the list */
496 		pcistub_device_put(found_psdev);
497 	}
498 }
499 
500 static DEFINE_PCI_DEVICE_TABLE(pcistub_ids) = {
501 	{
502 	 .vendor = PCI_ANY_ID,
503 	 .device = PCI_ANY_ID,
504 	 .subvendor = PCI_ANY_ID,
505 	 .subdevice = PCI_ANY_ID,
506 	 },
507 	{0,},
508 };
509 
510 #define PCI_NODENAME_MAX 40
511 static void kill_domain_by_device(struct pcistub_device *psdev)
512 {
513 	struct xenbus_transaction xbt;
514 	int err;
515 	char nodename[PCI_NODENAME_MAX];
516 
517 	if (!psdev)
518 		dev_err(&psdev->dev->dev,
519 			"device is NULL when do AER recovery/kill_domain\n");
520 	snprintf(nodename, PCI_NODENAME_MAX, "/local/domain/0/backend/pci/%d/0",
521 		psdev->pdev->xdev->otherend_id);
522 	nodename[strlen(nodename)] = '\0';
523 
524 again:
525 	err = xenbus_transaction_start(&xbt);
526 	if (err) {
527 		dev_err(&psdev->dev->dev,
528 			"error %d when start xenbus transaction\n", err);
529 		return;
530 	}
531 	/*PV AER handlers will set this flag*/
532 	xenbus_printf(xbt, nodename, "aerState" , "aerfail");
533 	err = xenbus_transaction_end(xbt, 0);
534 	if (err) {
535 		if (err == -EAGAIN)
536 			goto again;
537 		dev_err(&psdev->dev->dev,
538 			"error %d when end xenbus transaction\n", err);
539 		return;
540 	}
541 }
542 
543 /* For each aer recovery step error_detected, mmio_enabled, etc, front_end and
544  * backend need to have cooperation. In xen_pcibk, those steps will do similar
545  * jobs: send service request and waiting for front_end response.
546 */
547 static pci_ers_result_t common_process(struct pcistub_device *psdev,
548 				       pci_channel_state_t state, int aer_cmd,
549 				       pci_ers_result_t result)
550 {
551 	pci_ers_result_t res = result;
552 	struct xen_pcie_aer_op *aer_op;
553 	int ret;
554 
555 	/*with PV AER drivers*/
556 	aer_op = &(psdev->pdev->sh_info->aer_op);
557 	aer_op->cmd = aer_cmd ;
558 	/*useful for error_detected callback*/
559 	aer_op->err = state;
560 	/*pcifront_end BDF*/
561 	ret = xen_pcibk_get_pcifront_dev(psdev->dev, psdev->pdev,
562 		&aer_op->domain, &aer_op->bus, &aer_op->devfn);
563 	if (!ret) {
564 		dev_err(&psdev->dev->dev,
565 			DRV_NAME ": failed to get pcifront device\n");
566 		return PCI_ERS_RESULT_NONE;
567 	}
568 	wmb();
569 
570 	dev_dbg(&psdev->dev->dev,
571 			DRV_NAME ": aer_op %x dom %x bus %x devfn %x\n",
572 			aer_cmd, aer_op->domain, aer_op->bus, aer_op->devfn);
573 	/*local flag to mark there's aer request, xen_pcibk callback will use
574 	* this flag to judge whether we need to check pci-front give aer
575 	* service ack signal
576 	*/
577 	set_bit(_PCIB_op_pending, (unsigned long *)&psdev->pdev->flags);
578 
579 	/*It is possible that a pcifront conf_read_write ops request invokes
580 	* the callback which cause the spurious execution of wake_up.
581 	* Yet it is harmless and better than a spinlock here
582 	*/
583 	set_bit(_XEN_PCIB_active,
584 		(unsigned long *)&psdev->pdev->sh_info->flags);
585 	wmb();
586 	notify_remote_via_irq(psdev->pdev->evtchn_irq);
587 
588 	ret = wait_event_timeout(xen_pcibk_aer_wait_queue,
589 				 !(test_bit(_XEN_PCIB_active, (unsigned long *)
590 				 &psdev->pdev->sh_info->flags)), 300*HZ);
591 
592 	if (!ret) {
593 		if (test_bit(_XEN_PCIB_active,
594 			(unsigned long *)&psdev->pdev->sh_info->flags)) {
595 			dev_err(&psdev->dev->dev,
596 				"pcifront aer process not responding!\n");
597 			clear_bit(_XEN_PCIB_active,
598 			  (unsigned long *)&psdev->pdev->sh_info->flags);
599 			aer_op->err = PCI_ERS_RESULT_NONE;
600 			return res;
601 		}
602 	}
603 	clear_bit(_PCIB_op_pending, (unsigned long *)&psdev->pdev->flags);
604 
605 	if (test_bit(_XEN_PCIF_active,
606 		(unsigned long *)&psdev->pdev->sh_info->flags)) {
607 		dev_dbg(&psdev->dev->dev,
608 			"schedule pci_conf service in xen_pcibk\n");
609 		xen_pcibk_test_and_schedule_op(psdev->pdev);
610 	}
611 
612 	res = (pci_ers_result_t)aer_op->err;
613 	return res;
614 }
615 
616 /*
617 * xen_pcibk_slot_reset: it will send the slot_reset request to  pcifront in case
618 * of the device driver could provide this service, and then wait for pcifront
619 * ack.
620 * @dev: pointer to PCI devices
621 * return value is used by aer_core do_recovery policy
622 */
623 static pci_ers_result_t xen_pcibk_slot_reset(struct pci_dev *dev)
624 {
625 	struct pcistub_device *psdev;
626 	pci_ers_result_t result;
627 
628 	result = PCI_ERS_RESULT_RECOVERED;
629 	dev_dbg(&dev->dev, "xen_pcibk_slot_reset(bus:%x,devfn:%x)\n",
630 		dev->bus->number, dev->devfn);
631 
632 	down_write(&pcistub_sem);
633 	psdev = pcistub_device_find(pci_domain_nr(dev->bus),
634 				dev->bus->number,
635 				PCI_SLOT(dev->devfn),
636 				PCI_FUNC(dev->devfn));
637 
638 	if (!psdev || !psdev->pdev) {
639 		dev_err(&dev->dev,
640 			DRV_NAME " device is not found/assigned\n");
641 		goto end;
642 	}
643 
644 	if (!psdev->pdev->sh_info) {
645 		dev_err(&dev->dev, DRV_NAME " device is not connected or owned"
646 			" by HVM, kill it\n");
647 		kill_domain_by_device(psdev);
648 		goto release;
649 	}
650 
651 	if (!test_bit(_XEN_PCIB_AERHANDLER,
652 		(unsigned long *)&psdev->pdev->sh_info->flags)) {
653 		dev_err(&dev->dev,
654 			"guest with no AER driver should have been killed\n");
655 		goto release;
656 	}
657 	result = common_process(psdev, 1, XEN_PCI_OP_aer_slotreset, result);
658 
659 	if (result == PCI_ERS_RESULT_NONE ||
660 		result == PCI_ERS_RESULT_DISCONNECT) {
661 		dev_dbg(&dev->dev,
662 			"No AER slot_reset service or disconnected!\n");
663 		kill_domain_by_device(psdev);
664 	}
665 release:
666 	pcistub_device_put(psdev);
667 end:
668 	up_write(&pcistub_sem);
669 	return result;
670 
671 }
672 
673 
674 /*xen_pcibk_mmio_enabled: it will send the mmio_enabled request to  pcifront
675 * in case of the device driver could provide this service, and then wait
676 * for pcifront ack
677 * @dev: pointer to PCI devices
678 * return value is used by aer_core do_recovery policy
679 */
680 
681 static pci_ers_result_t xen_pcibk_mmio_enabled(struct pci_dev *dev)
682 {
683 	struct pcistub_device *psdev;
684 	pci_ers_result_t result;
685 
686 	result = PCI_ERS_RESULT_RECOVERED;
687 	dev_dbg(&dev->dev, "xen_pcibk_mmio_enabled(bus:%x,devfn:%x)\n",
688 		dev->bus->number, dev->devfn);
689 
690 	down_write(&pcistub_sem);
691 	psdev = pcistub_device_find(pci_domain_nr(dev->bus),
692 				dev->bus->number,
693 				PCI_SLOT(dev->devfn),
694 				PCI_FUNC(dev->devfn));
695 
696 	if (!psdev || !psdev->pdev) {
697 		dev_err(&dev->dev,
698 			DRV_NAME " device is not found/assigned\n");
699 		goto end;
700 	}
701 
702 	if (!psdev->pdev->sh_info) {
703 		dev_err(&dev->dev, DRV_NAME " device is not connected or owned"
704 			" by HVM, kill it\n");
705 		kill_domain_by_device(psdev);
706 		goto release;
707 	}
708 
709 	if (!test_bit(_XEN_PCIB_AERHANDLER,
710 		(unsigned long *)&psdev->pdev->sh_info->flags)) {
711 		dev_err(&dev->dev,
712 			"guest with no AER driver should have been killed\n");
713 		goto release;
714 	}
715 	result = common_process(psdev, 1, XEN_PCI_OP_aer_mmio, result);
716 
717 	if (result == PCI_ERS_RESULT_NONE ||
718 		result == PCI_ERS_RESULT_DISCONNECT) {
719 		dev_dbg(&dev->dev,
720 			"No AER mmio_enabled service or disconnected!\n");
721 		kill_domain_by_device(psdev);
722 	}
723 release:
724 	pcistub_device_put(psdev);
725 end:
726 	up_write(&pcistub_sem);
727 	return result;
728 }
729 
730 /*xen_pcibk_error_detected: it will send the error_detected request to  pcifront
731 * in case of the device driver could provide this service, and then wait
732 * for pcifront ack.
733 * @dev: pointer to PCI devices
734 * @error: the current PCI connection state
735 * return value is used by aer_core do_recovery policy
736 */
737 
738 static pci_ers_result_t xen_pcibk_error_detected(struct pci_dev *dev,
739 	pci_channel_state_t error)
740 {
741 	struct pcistub_device *psdev;
742 	pci_ers_result_t result;
743 
744 	result = PCI_ERS_RESULT_CAN_RECOVER;
745 	dev_dbg(&dev->dev, "xen_pcibk_error_detected(bus:%x,devfn:%x)\n",
746 		dev->bus->number, dev->devfn);
747 
748 	down_write(&pcistub_sem);
749 	psdev = pcistub_device_find(pci_domain_nr(dev->bus),
750 				dev->bus->number,
751 				PCI_SLOT(dev->devfn),
752 				PCI_FUNC(dev->devfn));
753 
754 	if (!psdev || !psdev->pdev) {
755 		dev_err(&dev->dev,
756 			DRV_NAME " device is not found/assigned\n");
757 		goto end;
758 	}
759 
760 	if (!psdev->pdev->sh_info) {
761 		dev_err(&dev->dev, DRV_NAME " device is not connected or owned"
762 			" by HVM, kill it\n");
763 		kill_domain_by_device(psdev);
764 		goto release;
765 	}
766 
767 	/*Guest owns the device yet no aer handler regiested, kill guest*/
768 	if (!test_bit(_XEN_PCIB_AERHANDLER,
769 		(unsigned long *)&psdev->pdev->sh_info->flags)) {
770 		dev_dbg(&dev->dev, "guest may have no aer driver, kill it\n");
771 		kill_domain_by_device(psdev);
772 		goto release;
773 	}
774 	result = common_process(psdev, error, XEN_PCI_OP_aer_detected, result);
775 
776 	if (result == PCI_ERS_RESULT_NONE ||
777 		result == PCI_ERS_RESULT_DISCONNECT) {
778 		dev_dbg(&dev->dev,
779 			"No AER error_detected service or disconnected!\n");
780 		kill_domain_by_device(psdev);
781 	}
782 release:
783 	pcistub_device_put(psdev);
784 end:
785 	up_write(&pcistub_sem);
786 	return result;
787 }
788 
789 /*xen_pcibk_error_resume: it will send the error_resume request to  pcifront
790 * in case of the device driver could provide this service, and then wait
791 * for pcifront ack.
792 * @dev: pointer to PCI devices
793 */
794 
795 static void xen_pcibk_error_resume(struct pci_dev *dev)
796 {
797 	struct pcistub_device *psdev;
798 
799 	dev_dbg(&dev->dev, "xen_pcibk_error_resume(bus:%x,devfn:%x)\n",
800 		dev->bus->number, dev->devfn);
801 
802 	down_write(&pcistub_sem);
803 	psdev = pcistub_device_find(pci_domain_nr(dev->bus),
804 				dev->bus->number,
805 				PCI_SLOT(dev->devfn),
806 				PCI_FUNC(dev->devfn));
807 
808 	if (!psdev || !psdev->pdev) {
809 		dev_err(&dev->dev,
810 			DRV_NAME " device is not found/assigned\n");
811 		goto end;
812 	}
813 
814 	if (!psdev->pdev->sh_info) {
815 		dev_err(&dev->dev, DRV_NAME " device is not connected or owned"
816 			" by HVM, kill it\n");
817 		kill_domain_by_device(psdev);
818 		goto release;
819 	}
820 
821 	if (!test_bit(_XEN_PCIB_AERHANDLER,
822 		(unsigned long *)&psdev->pdev->sh_info->flags)) {
823 		dev_err(&dev->dev,
824 			"guest with no AER driver should have been killed\n");
825 		kill_domain_by_device(psdev);
826 		goto release;
827 	}
828 	common_process(psdev, 1, XEN_PCI_OP_aer_resume,
829 		       PCI_ERS_RESULT_RECOVERED);
830 release:
831 	pcistub_device_put(psdev);
832 end:
833 	up_write(&pcistub_sem);
834 	return;
835 }
836 
837 /*add xen_pcibk AER handling*/
838 static struct pci_error_handlers xen_pcibk_error_handler = {
839 	.error_detected = xen_pcibk_error_detected,
840 	.mmio_enabled = xen_pcibk_mmio_enabled,
841 	.slot_reset = xen_pcibk_slot_reset,
842 	.resume = xen_pcibk_error_resume,
843 };
844 
845 /*
846  * Note: There is no MODULE_DEVICE_TABLE entry here because this isn't
847  * for a normal device. I don't want it to be loaded automatically.
848  */
849 
850 static struct pci_driver xen_pcibk_pci_driver = {
851 	/* The name should be xen_pciback, but until the tools are updated
852 	 * we will keep it as pciback. */
853 	.name = "pciback",
854 	.id_table = pcistub_ids,
855 	.probe = pcistub_probe,
856 	.remove = pcistub_remove,
857 	.err_handler = &xen_pcibk_error_handler,
858 };
859 
860 static inline int str_to_slot(const char *buf, int *domain, int *bus,
861 			      int *slot, int *func)
862 {
863 	int err;
864 
865 	err = sscanf(buf, " %x:%x:%x.%x", domain, bus, slot, func);
866 	if (err == 4)
867 		return 0;
868 	else if (err < 0)
869 		return -EINVAL;
870 
871 	/* try again without domain */
872 	*domain = 0;
873 	err = sscanf(buf, " %x:%x.%x", bus, slot, func);
874 	if (err == 3)
875 		return 0;
876 
877 	return -EINVAL;
878 }
879 
880 static inline int str_to_quirk(const char *buf, int *domain, int *bus, int
881 			       *slot, int *func, int *reg, int *size, int *mask)
882 {
883 	int err;
884 
885 	err =
886 	    sscanf(buf, " %04x:%02x:%02x.%1x-%08x:%1x:%08x", domain, bus, slot,
887 		   func, reg, size, mask);
888 	if (err == 7)
889 		return 0;
890 	return -EINVAL;
891 }
892 
893 static int pcistub_device_id_add(int domain, int bus, int slot, int func)
894 {
895 	struct pcistub_device_id *pci_dev_id;
896 	unsigned long flags;
897 
898 	pci_dev_id = kmalloc(sizeof(*pci_dev_id), GFP_KERNEL);
899 	if (!pci_dev_id)
900 		return -ENOMEM;
901 
902 	pci_dev_id->domain = domain;
903 	pci_dev_id->bus = bus;
904 	pci_dev_id->devfn = PCI_DEVFN(slot, func);
905 
906 	pr_debug(DRV_NAME ": wants to seize %04x:%02x:%02x.%01x\n",
907 		 domain, bus, slot, func);
908 
909 	spin_lock_irqsave(&device_ids_lock, flags);
910 	list_add_tail(&pci_dev_id->slot_list, &pcistub_device_ids);
911 	spin_unlock_irqrestore(&device_ids_lock, flags);
912 
913 	return 0;
914 }
915 
916 static int pcistub_device_id_remove(int domain, int bus, int slot, int func)
917 {
918 	struct pcistub_device_id *pci_dev_id, *t;
919 	int devfn = PCI_DEVFN(slot, func);
920 	int err = -ENOENT;
921 	unsigned long flags;
922 
923 	spin_lock_irqsave(&device_ids_lock, flags);
924 	list_for_each_entry_safe(pci_dev_id, t, &pcistub_device_ids,
925 				 slot_list) {
926 		if (pci_dev_id->domain == domain
927 		    && pci_dev_id->bus == bus && pci_dev_id->devfn == devfn) {
928 			/* Don't break; here because it's possible the same
929 			 * slot could be in the list more than once
930 			 */
931 			list_del(&pci_dev_id->slot_list);
932 			kfree(pci_dev_id);
933 
934 			err = 0;
935 
936 			pr_debug(DRV_NAME ": removed %04x:%02x:%02x.%01x from "
937 				 "seize list\n", domain, bus, slot, func);
938 		}
939 	}
940 	spin_unlock_irqrestore(&device_ids_lock, flags);
941 
942 	return err;
943 }
944 
945 static int pcistub_reg_add(int domain, int bus, int slot, int func, int reg,
946 			   int size, int mask)
947 {
948 	int err = 0;
949 	struct pcistub_device *psdev;
950 	struct pci_dev *dev;
951 	struct config_field *field;
952 
953 	psdev = pcistub_device_find(domain, bus, slot, func);
954 	if (!psdev || !psdev->dev) {
955 		err = -ENODEV;
956 		goto out;
957 	}
958 	dev = psdev->dev;
959 
960 	field = kzalloc(sizeof(*field), GFP_ATOMIC);
961 	if (!field) {
962 		err = -ENOMEM;
963 		goto out;
964 	}
965 
966 	field->offset = reg;
967 	field->size = size;
968 	field->mask = mask;
969 	field->init = NULL;
970 	field->reset = NULL;
971 	field->release = NULL;
972 	field->clean = xen_pcibk_config_field_free;
973 
974 	err = xen_pcibk_config_quirks_add_field(dev, field);
975 	if (err)
976 		kfree(field);
977 out:
978 	return err;
979 }
980 
981 static ssize_t pcistub_slot_add(struct device_driver *drv, const char *buf,
982 				size_t count)
983 {
984 	int domain, bus, slot, func;
985 	int err;
986 
987 	err = str_to_slot(buf, &domain, &bus, &slot, &func);
988 	if (err)
989 		goto out;
990 
991 	err = pcistub_device_id_add(domain, bus, slot, func);
992 
993 out:
994 	if (!err)
995 		err = count;
996 	return err;
997 }
998 
999 DRIVER_ATTR(new_slot, S_IWUSR, NULL, pcistub_slot_add);
1000 
1001 static ssize_t pcistub_slot_remove(struct device_driver *drv, const char *buf,
1002 				   size_t count)
1003 {
1004 	int domain, bus, slot, func;
1005 	int err;
1006 
1007 	err = str_to_slot(buf, &domain, &bus, &slot, &func);
1008 	if (err)
1009 		goto out;
1010 
1011 	err = pcistub_device_id_remove(domain, bus, slot, func);
1012 
1013 out:
1014 	if (!err)
1015 		err = count;
1016 	return err;
1017 }
1018 
1019 DRIVER_ATTR(remove_slot, S_IWUSR, NULL, pcistub_slot_remove);
1020 
1021 static ssize_t pcistub_slot_show(struct device_driver *drv, char *buf)
1022 {
1023 	struct pcistub_device_id *pci_dev_id;
1024 	size_t count = 0;
1025 	unsigned long flags;
1026 
1027 	spin_lock_irqsave(&device_ids_lock, flags);
1028 	list_for_each_entry(pci_dev_id, &pcistub_device_ids, slot_list) {
1029 		if (count >= PAGE_SIZE)
1030 			break;
1031 
1032 		count += scnprintf(buf + count, PAGE_SIZE - count,
1033 				   "%04x:%02x:%02x.%01x\n",
1034 				   pci_dev_id->domain, pci_dev_id->bus,
1035 				   PCI_SLOT(pci_dev_id->devfn),
1036 				   PCI_FUNC(pci_dev_id->devfn));
1037 	}
1038 	spin_unlock_irqrestore(&device_ids_lock, flags);
1039 
1040 	return count;
1041 }
1042 
1043 DRIVER_ATTR(slots, S_IRUSR, pcistub_slot_show, NULL);
1044 
1045 static ssize_t pcistub_irq_handler_show(struct device_driver *drv, char *buf)
1046 {
1047 	struct pcistub_device *psdev;
1048 	struct xen_pcibk_dev_data *dev_data;
1049 	size_t count = 0;
1050 	unsigned long flags;
1051 
1052 	spin_lock_irqsave(&pcistub_devices_lock, flags);
1053 	list_for_each_entry(psdev, &pcistub_devices, dev_list) {
1054 		if (count >= PAGE_SIZE)
1055 			break;
1056 		if (!psdev->dev)
1057 			continue;
1058 		dev_data = pci_get_drvdata(psdev->dev);
1059 		if (!dev_data)
1060 			continue;
1061 		count +=
1062 		    scnprintf(buf + count, PAGE_SIZE - count,
1063 			      "%s:%s:%sing:%ld\n",
1064 			      pci_name(psdev->dev),
1065 			      dev_data->isr_on ? "on" : "off",
1066 			      dev_data->ack_intr ? "ack" : "not ack",
1067 			      dev_data->handled);
1068 	}
1069 	spin_unlock_irqrestore(&pcistub_devices_lock, flags);
1070 	return count;
1071 }
1072 
1073 DRIVER_ATTR(irq_handlers, S_IRUSR, pcistub_irq_handler_show, NULL);
1074 
1075 static ssize_t pcistub_irq_handler_switch(struct device_driver *drv,
1076 					  const char *buf,
1077 					  size_t count)
1078 {
1079 	struct pcistub_device *psdev;
1080 	struct xen_pcibk_dev_data *dev_data;
1081 	int domain, bus, slot, func;
1082 	int err = -ENOENT;
1083 
1084 	err = str_to_slot(buf, &domain, &bus, &slot, &func);
1085 	if (err)
1086 		goto out;
1087 
1088 	psdev = pcistub_device_find(domain, bus, slot, func);
1089 
1090 	if (!psdev)
1091 		goto out;
1092 
1093 	dev_data = pci_get_drvdata(psdev->dev);
1094 	if (!dev_data)
1095 		goto out;
1096 
1097 	dev_dbg(&psdev->dev->dev, "%s fake irq handler: %d->%d\n",
1098 		dev_data->irq_name, dev_data->isr_on,
1099 		!dev_data->isr_on);
1100 
1101 	dev_data->isr_on = !(dev_data->isr_on);
1102 	if (dev_data->isr_on)
1103 		dev_data->ack_intr = 1;
1104 out:
1105 	if (!err)
1106 		err = count;
1107 	return err;
1108 }
1109 DRIVER_ATTR(irq_handler_state, S_IWUSR, NULL, pcistub_irq_handler_switch);
1110 
1111 static ssize_t pcistub_quirk_add(struct device_driver *drv, const char *buf,
1112 				 size_t count)
1113 {
1114 	int domain, bus, slot, func, reg, size, mask;
1115 	int err;
1116 
1117 	err = str_to_quirk(buf, &domain, &bus, &slot, &func, &reg, &size,
1118 			   &mask);
1119 	if (err)
1120 		goto out;
1121 
1122 	err = pcistub_reg_add(domain, bus, slot, func, reg, size, mask);
1123 
1124 out:
1125 	if (!err)
1126 		err = count;
1127 	return err;
1128 }
1129 
1130 static ssize_t pcistub_quirk_show(struct device_driver *drv, char *buf)
1131 {
1132 	int count = 0;
1133 	unsigned long flags;
1134 	struct xen_pcibk_config_quirk *quirk;
1135 	struct xen_pcibk_dev_data *dev_data;
1136 	const struct config_field *field;
1137 	const struct config_field_entry *cfg_entry;
1138 
1139 	spin_lock_irqsave(&device_ids_lock, flags);
1140 	list_for_each_entry(quirk, &xen_pcibk_quirks, quirks_list) {
1141 		if (count >= PAGE_SIZE)
1142 			goto out;
1143 
1144 		count += scnprintf(buf + count, PAGE_SIZE - count,
1145 				   "%02x:%02x.%01x\n\t%04x:%04x:%04x:%04x\n",
1146 				   quirk->pdev->bus->number,
1147 				   PCI_SLOT(quirk->pdev->devfn),
1148 				   PCI_FUNC(quirk->pdev->devfn),
1149 				   quirk->devid.vendor, quirk->devid.device,
1150 				   quirk->devid.subvendor,
1151 				   quirk->devid.subdevice);
1152 
1153 		dev_data = pci_get_drvdata(quirk->pdev);
1154 
1155 		list_for_each_entry(cfg_entry, &dev_data->config_fields, list) {
1156 			field = cfg_entry->field;
1157 			if (count >= PAGE_SIZE)
1158 				goto out;
1159 
1160 			count += scnprintf(buf + count, PAGE_SIZE - count,
1161 					   "\t\t%08x:%01x:%08x\n",
1162 					   cfg_entry->base_offset +
1163 					   field->offset, field->size,
1164 					   field->mask);
1165 		}
1166 	}
1167 
1168 out:
1169 	spin_unlock_irqrestore(&device_ids_lock, flags);
1170 
1171 	return count;
1172 }
1173 
1174 DRIVER_ATTR(quirks, S_IRUSR | S_IWUSR, pcistub_quirk_show, pcistub_quirk_add);
1175 
1176 static ssize_t permissive_add(struct device_driver *drv, const char *buf,
1177 			      size_t count)
1178 {
1179 	int domain, bus, slot, func;
1180 	int err;
1181 	struct pcistub_device *psdev;
1182 	struct xen_pcibk_dev_data *dev_data;
1183 	err = str_to_slot(buf, &domain, &bus, &slot, &func);
1184 	if (err)
1185 		goto out;
1186 	psdev = pcistub_device_find(domain, bus, slot, func);
1187 	if (!psdev) {
1188 		err = -ENODEV;
1189 		goto out;
1190 	}
1191 	if (!psdev->dev) {
1192 		err = -ENODEV;
1193 		goto release;
1194 	}
1195 	dev_data = pci_get_drvdata(psdev->dev);
1196 	/* the driver data for a device should never be null at this point */
1197 	if (!dev_data) {
1198 		err = -ENXIO;
1199 		goto release;
1200 	}
1201 	if (!dev_data->permissive) {
1202 		dev_data->permissive = 1;
1203 		/* Let user know that what they're doing could be unsafe */
1204 		dev_warn(&psdev->dev->dev, "enabling permissive mode "
1205 			 "configuration space accesses!\n");
1206 		dev_warn(&psdev->dev->dev,
1207 			 "permissive mode is potentially unsafe!\n");
1208 	}
1209 release:
1210 	pcistub_device_put(psdev);
1211 out:
1212 	if (!err)
1213 		err = count;
1214 	return err;
1215 }
1216 
1217 static ssize_t permissive_show(struct device_driver *drv, char *buf)
1218 {
1219 	struct pcistub_device *psdev;
1220 	struct xen_pcibk_dev_data *dev_data;
1221 	size_t count = 0;
1222 	unsigned long flags;
1223 	spin_lock_irqsave(&pcistub_devices_lock, flags);
1224 	list_for_each_entry(psdev, &pcistub_devices, dev_list) {
1225 		if (count >= PAGE_SIZE)
1226 			break;
1227 		if (!psdev->dev)
1228 			continue;
1229 		dev_data = pci_get_drvdata(psdev->dev);
1230 		if (!dev_data || !dev_data->permissive)
1231 			continue;
1232 		count +=
1233 		    scnprintf(buf + count, PAGE_SIZE - count, "%s\n",
1234 			      pci_name(psdev->dev));
1235 	}
1236 	spin_unlock_irqrestore(&pcistub_devices_lock, flags);
1237 	return count;
1238 }
1239 
1240 DRIVER_ATTR(permissive, S_IRUSR | S_IWUSR, permissive_show, permissive_add);
1241 
1242 static void pcistub_exit(void)
1243 {
1244 	driver_remove_file(&xen_pcibk_pci_driver.driver, &driver_attr_new_slot);
1245 	driver_remove_file(&xen_pcibk_pci_driver.driver,
1246 			   &driver_attr_remove_slot);
1247 	driver_remove_file(&xen_pcibk_pci_driver.driver, &driver_attr_slots);
1248 	driver_remove_file(&xen_pcibk_pci_driver.driver, &driver_attr_quirks);
1249 	driver_remove_file(&xen_pcibk_pci_driver.driver,
1250 			   &driver_attr_permissive);
1251 	driver_remove_file(&xen_pcibk_pci_driver.driver,
1252 			   &driver_attr_irq_handlers);
1253 	driver_remove_file(&xen_pcibk_pci_driver.driver,
1254 			   &driver_attr_irq_handler_state);
1255 	pci_unregister_driver(&xen_pcibk_pci_driver);
1256 }
1257 
1258 static int __init pcistub_init(void)
1259 {
1260 	int pos = 0;
1261 	int err = 0;
1262 	int domain, bus, slot, func;
1263 	int parsed;
1264 
1265 	if (pci_devs_to_hide && *pci_devs_to_hide) {
1266 		do {
1267 			parsed = 0;
1268 
1269 			err = sscanf(pci_devs_to_hide + pos,
1270 				     " (%x:%x:%x.%x) %n",
1271 				     &domain, &bus, &slot, &func, &parsed);
1272 			if (err != 4) {
1273 				domain = 0;
1274 				err = sscanf(pci_devs_to_hide + pos,
1275 					     " (%x:%x.%x) %n",
1276 					     &bus, &slot, &func, &parsed);
1277 				if (err != 3)
1278 					goto parse_error;
1279 			}
1280 
1281 			err = pcistub_device_id_add(domain, bus, slot, func);
1282 			if (err)
1283 				goto out;
1284 
1285 			/* if parsed<=0, we've reached the end of the string */
1286 			pos += parsed;
1287 		} while (parsed > 0 && pci_devs_to_hide[pos]);
1288 	}
1289 
1290 	/* If we're the first PCI Device Driver to register, we're the
1291 	 * first one to get offered PCI devices as they become
1292 	 * available (and thus we can be the first to grab them)
1293 	 */
1294 	err = pci_register_driver(&xen_pcibk_pci_driver);
1295 	if (err < 0)
1296 		goto out;
1297 
1298 	err = driver_create_file(&xen_pcibk_pci_driver.driver,
1299 				 &driver_attr_new_slot);
1300 	if (!err)
1301 		err = driver_create_file(&xen_pcibk_pci_driver.driver,
1302 					 &driver_attr_remove_slot);
1303 	if (!err)
1304 		err = driver_create_file(&xen_pcibk_pci_driver.driver,
1305 					 &driver_attr_slots);
1306 	if (!err)
1307 		err = driver_create_file(&xen_pcibk_pci_driver.driver,
1308 					 &driver_attr_quirks);
1309 	if (!err)
1310 		err = driver_create_file(&xen_pcibk_pci_driver.driver,
1311 					 &driver_attr_permissive);
1312 
1313 	if (!err)
1314 		err = driver_create_file(&xen_pcibk_pci_driver.driver,
1315 					 &driver_attr_irq_handlers);
1316 	if (!err)
1317 		err = driver_create_file(&xen_pcibk_pci_driver.driver,
1318 					&driver_attr_irq_handler_state);
1319 	if (err)
1320 		pcistub_exit();
1321 
1322 out:
1323 	return err;
1324 
1325 parse_error:
1326 	printk(KERN_ERR DRV_NAME ": Error parsing pci_devs_to_hide at \"%s\"\n",
1327 	       pci_devs_to_hide + pos);
1328 	return -EINVAL;
1329 }
1330 
1331 #ifndef MODULE
1332 /*
1333  * fs_initcall happens before device_initcall
1334  * so xen_pcibk *should* get called first (b/c we
1335  * want to suck up any device before other drivers
1336  * get a chance by being the first pci device
1337  * driver to register)
1338  */
1339 fs_initcall(pcistub_init);
1340 #endif
1341 
1342 static int __init xen_pcibk_init(void)
1343 {
1344 	int err;
1345 
1346 	if (!xen_initial_domain())
1347 		return -ENODEV;
1348 
1349 	err = xen_pcibk_config_init();
1350 	if (err)
1351 		return err;
1352 
1353 #ifdef MODULE
1354 	err = pcistub_init();
1355 	if (err < 0)
1356 		return err;
1357 #endif
1358 
1359 	pcistub_init_devices_late();
1360 	err = xen_pcibk_xenbus_register();
1361 	if (err)
1362 		pcistub_exit();
1363 
1364 	return err;
1365 }
1366 
1367 static void __exit xen_pcibk_cleanup(void)
1368 {
1369 	xen_pcibk_xenbus_unregister();
1370 	pcistub_exit();
1371 }
1372 
1373 module_init(xen_pcibk_init);
1374 module_exit(xen_pcibk_cleanup);
1375 
1376 MODULE_LICENSE("Dual BSD/GPL");
1377