Driver Core: add ability for drivers to do a threaded probe
[safe/jmp/linux-2.6] / drivers / base / dd.c
1 /*
2  *      drivers/base/dd.c - The core device/driver interactions.
3  *
4  *      This file contains the (sometimes tricky) code that controls the
5  *      interactions between devices and drivers, which primarily includes
6  *      driver binding and unbinding.
7  *
8  *      All of this code used to exist in drivers/base/bus.c, but was
9  *      relocated to here in the name of compartmentalization (since it wasn't
10  *      strictly code just for the 'struct bus_type'.
11  *
12  *      Copyright (c) 2002-5 Patrick Mochel
13  *      Copyright (c) 2002-3 Open Source Development Labs
14  *
15  *      This file is released under the GPLv2
16  */
17
18 #include <linux/device.h>
19 #include <linux/module.h>
20 #include <linux/kthread.h>
21
22 #include "base.h"
23 #include "power/power.h"
24
25 #define to_drv(node) container_of(node, struct device_driver, kobj.entry)
26
27
28 /**
29  *      device_bind_driver - bind a driver to one device.
30  *      @dev:   device.
31  *
32  *      Allow manual attachment of a driver to a device.
33  *      Caller must have already set @dev->driver.
34  *
35  *      Note that this does not modify the bus reference count
36  *      nor take the bus's rwsem. Please verify those are accounted
37  *      for before calling this. (It is ok to call with no other effort
38  *      from a driver's probe() method.)
39  *
40  *      This function must be called with @dev->sem held.
41  */
42 int device_bind_driver(struct device *dev)
43 {
44         int ret;
45
46         if (klist_node_attached(&dev->knode_driver)) {
47                 printk(KERN_WARNING "%s: device %s already bound\n",
48                         __FUNCTION__, kobject_name(&dev->kobj));
49                 return 0;
50         }
51
52         pr_debug("bound device '%s' to driver '%s'\n",
53                  dev->bus_id, dev->driver->name);
54         klist_add_tail(&dev->knode_driver, &dev->driver->klist_devices);
55         ret = sysfs_create_link(&dev->driver->kobj, &dev->kobj,
56                           kobject_name(&dev->kobj));
57         if (ret == 0) {
58                 ret = sysfs_create_link(&dev->kobj, &dev->driver->kobj,
59                                         "driver");
60                 if (ret)
61                         sysfs_remove_link(&dev->driver->kobj,
62                                         kobject_name(&dev->kobj));
63         }
64         return ret;
65 }
66
67 struct stupid_thread_structure {
68         struct device_driver *drv;
69         struct device *dev;
70 };
71
72 static atomic_t probe_count = ATOMIC_INIT(0);
73 static int really_probe(void *void_data)
74 {
75         struct stupid_thread_structure *data = void_data;
76         struct device_driver *drv = data->drv;
77         struct device *dev = data->dev;
78         int ret = 0;
79
80         atomic_inc(&probe_count);
81         pr_debug("%s: Probing driver %s with device %s\n",
82                  drv->bus->name, drv->name, dev->bus_id);
83
84         dev->driver = drv;
85         if (dev->bus->probe) {
86                 ret = dev->bus->probe(dev);
87                 if (ret) {
88                         dev->driver = NULL;
89                         goto probe_failed;
90                 }
91         } else if (drv->probe) {
92                 ret = drv->probe(dev);
93                 if (ret) {
94                         dev->driver = NULL;
95                         goto probe_failed;
96                 }
97         }
98         if (device_bind_driver(dev)) {
99                 printk(KERN_ERR "%s: device_bind_driver(%s) failed\n",
100                         __FUNCTION__, dev->bus_id);
101                 /* How does undo a ->probe?  We're screwed. */
102         }
103         ret = 1;
104         pr_debug("%s: Bound Device %s to Driver %s\n",
105                  drv->bus->name, dev->bus_id, drv->name);
106         goto done;
107
108 probe_failed:
109         if (ret == -ENODEV || ret == -ENXIO) {
110                 /* Driver matched, but didn't support device
111                  * or device not found.
112                  * Not an error; keep going.
113                  */
114                 ret = 0;
115         } else {
116                 /* driver matched but the probe failed */
117                 printk(KERN_WARNING
118                        "%s: probe of %s failed with error %d\n",
119                        drv->name, dev->bus_id, ret);
120         }
121 done:
122         kfree(data);
123         atomic_dec(&probe_count);
124         return ret;
125 }
126
127 /**
128  * driver_probe_done
129  * Determine if the probe sequence is finished or not.
130  *
131  * Should somehow figure out how to use a semaphore, not an atomic variable...
132  */
133 int driver_probe_done(void)
134 {
135         pr_debug("%s: probe_count = %d\n", __FUNCTION__,
136                  atomic_read(&probe_count));
137         if (atomic_read(&probe_count))
138                 return -EBUSY;
139         return 0;
140 }
141
142 /**
143  * driver_probe_device - attempt to bind device & driver together
144  * @drv: driver to bind a device to
145  * @dev: device to try to bind to the driver
146  *
147  * First, we call the bus's match function, if one present, which should
148  * compare the device IDs the driver supports with the device IDs of the
149  * device. Note we don't do this ourselves because we don't know the
150  * format of the ID structures, nor what is to be considered a match and
151  * what is not.
152  *
153  * This function returns 1 if a match is found, an error if one occurs
154  * (that is not -ENODEV or -ENXIO), and 0 otherwise.
155  *
156  * This function must be called with @dev->sem held.  When called for a
157  * USB interface, @dev->parent->sem must be held as well.
158  */
159 int driver_probe_device(struct device_driver * drv, struct device * dev)
160 {
161         struct stupid_thread_structure *data;
162         struct task_struct *probe_task;
163         int ret = 0;
164
165         if (drv->bus->match && !drv->bus->match(dev, drv))
166                 goto done;
167
168         pr_debug("%s: Matched Device %s with Driver %s\n",
169                  drv->bus->name, dev->bus_id, drv->name);
170
171         data = kmalloc(sizeof(*data), GFP_KERNEL);
172         data->drv = drv;
173         data->dev = dev;
174
175         if (drv->multithread_probe) {
176                 probe_task = kthread_run(really_probe, data,
177                                          "probe-%s", dev->bus_id);
178                 if (IS_ERR(probe_task))
179                         ret = PTR_ERR(probe_task);
180         } else
181                 ret = really_probe(data);
182
183 done:
184         return ret;
185 }
186
187 static int __device_attach(struct device_driver * drv, void * data)
188 {
189         struct device * dev = data;
190         return driver_probe_device(drv, dev);
191 }
192
193 /**
194  *      device_attach - try to attach device to a driver.
195  *      @dev:   device.
196  *
197  *      Walk the list of drivers that the bus has and call
198  *      driver_probe_device() for each pair. If a compatible
199  *      pair is found, break out and return.
200  *
201  *      Returns 1 if the device was bound to a driver;
202  *      0 if no matching device was found; error code otherwise.
203  *
204  *      When called for a USB interface, @dev->parent->sem must be held.
205  */
206 int device_attach(struct device * dev)
207 {
208         int ret = 0;
209
210         down(&dev->sem);
211         if (dev->driver) {
212                 ret = device_bind_driver(dev);
213                 if (ret == 0)
214                         ret = 1;
215         } else
216                 ret = bus_for_each_drv(dev->bus, NULL, dev, __device_attach);
217         up(&dev->sem);
218         return ret;
219 }
220
221 static int __driver_attach(struct device * dev, void * data)
222 {
223         struct device_driver * drv = data;
224
225         /*
226          * Lock device and try to bind to it. We drop the error
227          * here and always return 0, because we need to keep trying
228          * to bind to devices and some drivers will return an error
229          * simply if it didn't support the device.
230          *
231          * driver_probe_device() will spit a warning if there
232          * is an error.
233          */
234
235         if (dev->parent)        /* Needed for USB */
236                 down(&dev->parent->sem);
237         down(&dev->sem);
238         if (!dev->driver)
239                 driver_probe_device(drv, dev);
240         up(&dev->sem);
241         if (dev->parent)
242                 up(&dev->parent->sem);
243
244         return 0;
245 }
246
247 /**
248  *      driver_attach - try to bind driver to devices.
249  *      @drv:   driver.
250  *
251  *      Walk the list of devices that the bus has on it and try to
252  *      match the driver with each one.  If driver_probe_device()
253  *      returns 0 and the @dev->driver is set, we've found a
254  *      compatible pair.
255  */
256 int driver_attach(struct device_driver * drv)
257 {
258         return bus_for_each_dev(drv->bus, NULL, drv, __driver_attach);
259 }
260
261 /**
262  *      device_release_driver - manually detach device from driver.
263  *      @dev:   device.
264  *
265  *      Manually detach device from driver.
266  *
267  *      __device_release_driver() must be called with @dev->sem held.
268  *      When called for a USB interface, @dev->parent->sem must be held
269  *      as well.
270  */
271
272 static void __device_release_driver(struct device * dev)
273 {
274         struct device_driver * drv;
275
276         drv = dev->driver;
277         if (drv) {
278                 get_driver(drv);
279                 sysfs_remove_link(&drv->kobj, kobject_name(&dev->kobj));
280                 sysfs_remove_link(&dev->kobj, "driver");
281                 klist_remove(&dev->knode_driver);
282
283                 if (dev->bus && dev->bus->remove)
284                         dev->bus->remove(dev);
285                 else if (drv->remove)
286                         drv->remove(dev);
287                 dev->driver = NULL;
288                 put_driver(drv);
289         }
290 }
291
292 void device_release_driver(struct device * dev)
293 {
294         /*
295          * If anyone calls device_release_driver() recursively from
296          * within their ->remove callback for the same device, they
297          * will deadlock right here.
298          */
299         down(&dev->sem);
300         __device_release_driver(dev);
301         up(&dev->sem);
302 }
303
304
305 /**
306  * driver_detach - detach driver from all devices it controls.
307  * @drv: driver.
308  */
309 void driver_detach(struct device_driver * drv)
310 {
311         struct device * dev;
312
313         for (;;) {
314                 spin_lock(&drv->klist_devices.k_lock);
315                 if (list_empty(&drv->klist_devices.k_list)) {
316                         spin_unlock(&drv->klist_devices.k_lock);
317                         break;
318                 }
319                 dev = list_entry(drv->klist_devices.k_list.prev,
320                                 struct device, knode_driver.n_node);
321                 get_device(dev);
322                 spin_unlock(&drv->klist_devices.k_lock);
323
324                 if (dev->parent)        /* Needed for USB */
325                         down(&dev->parent->sem);
326                 down(&dev->sem);
327                 if (dev->driver == drv)
328                         __device_release_driver(dev);
329                 up(&dev->sem);
330                 if (dev->parent)
331                         up(&dev->parent->sem);
332                 put_device(dev);
333         }
334 }
335
336
337 EXPORT_SYMBOL_GPL(device_bind_driver);
338 EXPORT_SYMBOL_GPL(device_release_driver);
339 EXPORT_SYMBOL_GPL(device_attach);
340 EXPORT_SYMBOL_GPL(driver_attach);
341