1 /*-
2  * Copyright (c) 2015-2016 Mellanox Technologies, Ltd.
3  * All rights reserved.
4  *
5  * Redistribution and use in source and binary forms, with or without
6  * modification, are permitted provided that the following conditions
7  * are met:
8  * 1. Redistributions of source code must retain the above copyright
9  *    notice unmodified, this list of conditions, and the following
10  *    disclaimer.
11  * 2. Redistributions in binary form must reproduce the above copyright
12  *    notice, this list of conditions and the following disclaimer in the
13  *    documentation and/or other materials provided with the distribution.
14  *
15  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
16  * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
17  * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
18  * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
19  * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
20  * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
21  * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
22  * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
23  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
24  * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
25  */
26 
27 #include <sys/cdefs.h>
28 __FBSDID("$FreeBSD$");
29 
30 #include <sys/param.h>
31 #include <sys/systm.h>
32 #include <sys/malloc.h>
33 #include <sys/kernel.h>
34 #include <sys/sysctl.h>
35 #include <sys/lock.h>
36 #include <sys/mutex.h>
37 #include <sys/bus.h>
38 #include <sys/fcntl.h>
39 #include <sys/file.h>
40 #include <sys/filio.h>
41 #include <sys/rwlock.h>
42 
43 #include <vm/vm.h>
44 #include <vm/pmap.h>
45 
46 #include <machine/stdarg.h>
47 
48 #include <linux/kobject.h>
49 #include <linux/device.h>
50 #include <linux/slab.h>
51 #include <linux/module.h>
52 #include <linux/cdev.h>
53 #include <linux/file.h>
54 #include <linux/sysfs.h>
55 #include <linux/mm.h>
56 #include <linux/io.h>
57 #include <linux/vmalloc.h>
58 #include <linux/pci.h>
59 #include <linux/compat.h>
60 
61 static device_probe_t linux_pci_probe;
62 static device_attach_t linux_pci_attach;
63 static device_detach_t linux_pci_detach;
64 static device_suspend_t linux_pci_suspend;
65 static device_resume_t linux_pci_resume;
66 static device_shutdown_t linux_pci_shutdown;
67 
68 static device_method_t pci_methods[] = {
69 	DEVMETHOD(device_probe, linux_pci_probe),
70 	DEVMETHOD(device_attach, linux_pci_attach),
71 	DEVMETHOD(device_detach, linux_pci_detach),
72 	DEVMETHOD(device_suspend, linux_pci_suspend),
73 	DEVMETHOD(device_resume, linux_pci_resume),
74 	DEVMETHOD(device_shutdown, linux_pci_shutdown),
75 	DEVMETHOD_END
76 };
77 
78 static struct pci_driver *
79 linux_pci_find(device_t dev, const struct pci_device_id **idp)
80 {
81 	const struct pci_device_id *id;
82 	struct pci_driver *pdrv;
83 	uint16_t vendor;
84 	uint16_t device;
85 
86 	vendor = pci_get_vendor(dev);
87 	device = pci_get_device(dev);
88 
89 	spin_lock(&pci_lock);
90 	list_for_each_entry(pdrv, &pci_drivers, links) {
91 		for (id = pdrv->id_table; id->vendor != 0; id++) {
92 			if (vendor == id->vendor && device == id->device) {
93 				*idp = id;
94 				spin_unlock(&pci_lock);
95 				return (pdrv);
96 			}
97 		}
98 	}
99 	spin_unlock(&pci_lock);
100 	return (NULL);
101 }
102 
103 static int
104 linux_pci_probe(device_t dev)
105 {
106 	const struct pci_device_id *id;
107 	struct pci_driver *pdrv;
108 
109 	if ((pdrv = linux_pci_find(dev, &id)) == NULL)
110 		return (ENXIO);
111 	if (device_get_driver(dev) != &pdrv->driver)
112 		return (ENXIO);
113 	device_set_desc(dev, pdrv->name);
114 	return (0);
115 }
116 
117 static int
118 linux_pci_attach(device_t dev)
119 {
120 	struct resource_list_entry *rle;
121 	struct pci_dev *pdev;
122 	struct pci_driver *pdrv;
123 	const struct pci_device_id *id;
124 	int error;
125 
126 	linux_set_current(curthread);
127 	pdrv = linux_pci_find(dev, &id);
128 	pdev = device_get_softc(dev);
129 	pdev->dev.parent = &linux_root_device;
130 	pdev->dev.bsddev = dev;
131 	INIT_LIST_HEAD(&pdev->dev.irqents);
132 	pdev->device = id->device;
133 	pdev->vendor = id->vendor;
134 	pdev->dev.dma_mask = &pdev->dma_mask;
135 	pdev->pdrv = pdrv;
136 	kobject_init(&pdev->dev.kobj, &linux_dev_ktype);
137 	kobject_set_name(&pdev->dev.kobj, device_get_nameunit(dev));
138 	kobject_add(&pdev->dev.kobj, &linux_root_device.kobj,
139 	    kobject_name(&pdev->dev.kobj));
140 	rle = linux_pci_get_rle(pdev, SYS_RES_IRQ, 0);
141 	if (rle != NULL)
142 		pdev->dev.irq = rle->start;
143 	else
144 		pdev->dev.irq = LINUX_IRQ_INVALID;
145 	pdev->irq = pdev->dev.irq;
146 	DROP_GIANT();
147 	spin_lock(&pci_lock);
148 	list_add(&pdev->links, &pci_devices);
149 	spin_unlock(&pci_lock);
150 	error = pdrv->probe(pdev, id);
151 	PICKUP_GIANT();
152 	if (error) {
153 		spin_lock(&pci_lock);
154 		list_del(&pdev->links);
155 		spin_unlock(&pci_lock);
156 		put_device(&pdev->dev);
157 		error = -error;
158 	}
159 	return (error);
160 }
161 
162 static int
163 linux_pci_detach(device_t dev)
164 {
165 	struct pci_dev *pdev;
166 
167 	linux_set_current(curthread);
168 	pdev = device_get_softc(dev);
169 	DROP_GIANT();
170 	pdev->pdrv->remove(pdev);
171 	PICKUP_GIANT();
172 	spin_lock(&pci_lock);
173 	list_del(&pdev->links);
174 	spin_unlock(&pci_lock);
175 	put_device(&pdev->dev);
176 
177 	return (0);
178 }
179 
180 static int
181 linux_pci_suspend(device_t dev)
182 {
183 	struct pm_message pm = { };
184 	struct pci_dev *pdev;
185 	int err;
186 
187 	linux_set_current(curthread);
188 	pdev = device_get_softc(dev);
189 	if (pdev->pdrv->suspend != NULL)
190 		err = -pdev->pdrv->suspend(pdev, pm);
191 	else
192 		err = 0;
193 	return (err);
194 }
195 
196 static int
197 linux_pci_resume(device_t dev)
198 {
199 	struct pci_dev *pdev;
200 	int err;
201 
202 	linux_set_current(curthread);
203 	pdev = device_get_softc(dev);
204 	if (pdev->pdrv->resume != NULL)
205 		err = -pdev->pdrv->resume(pdev);
206 	else
207 		err = 0;
208 	return (err);
209 }
210 
211 static int
212 linux_pci_shutdown(device_t dev)
213 {
214 	struct pci_dev *pdev;
215 
216 	linux_set_current(curthread);
217 	pdev = device_get_softc(dev);
218 	if (pdev->pdrv->shutdown != NULL) {
219 		DROP_GIANT();
220 		pdev->pdrv->shutdown(pdev);
221 		PICKUP_GIANT();
222 	}
223 	return (0);
224 }
225 
226 int
227 pci_register_driver(struct pci_driver *pdrv)
228 {
229 	devclass_t bus;
230 	int error = 0;
231 
232 	bus = devclass_find("pci");
233 
234 	linux_set_current(curthread);
235 	spin_lock(&pci_lock);
236 	list_add(&pdrv->links, &pci_drivers);
237 	spin_unlock(&pci_lock);
238 	pdrv->driver.name = pdrv->name;
239 	pdrv->driver.methods = pci_methods;
240 	pdrv->driver.size = sizeof(struct pci_dev);
241 	mtx_lock(&Giant);
242 	if (bus != NULL) {
243 		error = devclass_add_driver(bus, &pdrv->driver, BUS_PASS_DEFAULT,
244 		    &pdrv->bsdclass);
245 	}
246 	mtx_unlock(&Giant);
247 	return (-error);
248 }
249 
250 void
251 pci_unregister_driver(struct pci_driver *pdrv)
252 {
253 	devclass_t bus;
254 
255 	bus = devclass_find("pci");
256 
257 	list_del(&pdrv->links);
258 	mtx_lock(&Giant);
259 	if (bus != NULL)
260 		devclass_delete_driver(bus, &pdrv->driver);
261 	mtx_unlock(&Giant);
262 }
263 
264