1 // SPDX-License-Identifier: GPL-2.0+
2 /*
3  * Copyright (c) 2014 Google, Inc
4  * Written by Simon Glass <sjg@chromium.org>
5  */
6 
7 #include <common.h>
8 #include <dm.h>
9 #include <fdtdec.h>
10 #include <linux/libfdt.h>
11 #include <pci.h>
12 #include <dm/lists.h>
13 
14 struct sandbox_pci_priv {
15 	int dev_count;
16 };
17 
18 int sandbox_pci_get_emul(struct udevice *bus, pci_dev_t find_devfn,
19 			 struct udevice **containerp, struct udevice **emulp)
20 {
21 	struct udevice *dev;
22 	int ret;
23 
24 	*containerp = NULL;
25 	ret = pci_bus_find_devfn(bus, PCI_MASK_BUS(find_devfn), &dev);
26 	if (ret) {
27 		debug("%s: Could not find emulator for dev %x\n", __func__,
28 		      find_devfn);
29 		return ret;
30 	}
31 	*containerp = dev;
32 
33 	if (device_get_uclass_id(dev) == UCLASS_PCI_GENERIC) {
34 		ret = device_find_first_child(dev, emulp);
35 		if (ret)
36 			return ret;
37 	} else {
38 		*emulp = dev;
39 	}
40 
41 	return *emulp ? 0 : -ENODEV;
42 }
43 
44 static int sandbox_pci_emul_post_probe(struct udevice *dev)
45 {
46 	struct sandbox_pci_priv *priv = dev->uclass->priv;
47 
48 	priv->dev_count++;
49 	sandbox_set_enable_pci_map(true);
50 
51 	return 0;
52 }
53 
54 static int sandbox_pci_emul_pre_remove(struct udevice *dev)
55 {
56 	struct sandbox_pci_priv *priv = dev->uclass->priv;
57 
58 	priv->dev_count--;
59 	sandbox_set_enable_pci_map(priv->dev_count > 0);
60 
61 	return 0;
62 }
63 
64 UCLASS_DRIVER(pci_emul) = {
65 	.id		= UCLASS_PCI_EMUL,
66 	.name		= "pci_emul",
67 	.post_probe	= sandbox_pci_emul_post_probe,
68 	.pre_remove	= sandbox_pci_emul_pre_remove,
69 	.priv_auto_alloc_size	= sizeof(struct sandbox_pci_priv),
70 };
71