From: James Bottomley The problem is that klists claim to provide semantics for safe traversal of lists which are being modified. The failure case is when traversal of a list causes element removal (a fairly common case). The issue is that although the list node is refcounted, if it is embedded in an object (which is universally the case), then the object will be freed regardless of the klist refcount leading to slab corruption because the klist iterator refers to the prior element to get the next. The solution is to make the klist take and release references to the embedding object meaning that the embedding object won't be released until the list relinquishes the reference to it. Signed-off-by: James Bottomley Signed-off-by: Andrew Morton --- drivers/base/bus.c | 34 ++++++++++++++++++++++++++++++++-- drivers/base/core.c | 17 ++++++++++++++++- drivers/base/driver.c | 15 ++++++++++++++- include/linux/klist.h | 8 +++++--- lib/klist.c | 18 +++++++++++++++++- 5 files changed, 84 insertions(+), 8 deletions(-) diff -puN drivers/base/bus.c~fix-klist-semantics-for-lists-which-have-elements-removed drivers/base/bus.c --- devel/drivers/base/bus.c~fix-klist-semantics-for-lists-which-have-elements-removed 2005-08-30 18:40:38.000000000 -0700 +++ devel-akpm/drivers/base/bus.c 2005-08-30 18:40:38.000000000 -0700 @@ -566,6 +566,36 @@ static void bus_remove_attrs(struct bus_ } } +static void klist_devices_get(struct klist_node *n) +{ + struct device *dev = container_of(n, struct device, knode_bus); + + get_device(dev); +} + +static void klist_devices_put(struct klist_node *n) +{ + struct device *dev = container_of(n, struct device, knode_bus); + + put_device(dev); +} + +static void klist_drivers_get(struct klist_node *n) +{ + struct device_driver *drv = container_of(n, struct device_driver, + knode_bus); + + get_driver(drv); +} + +static void klist_drivers_put(struct klist_node *n) +{ + struct device_driver *drv = container_of(n, struct device_driver, + knode_bus); + + put_driver(drv); +} + /** * bus_register - register a bus with the system. * @bus: bus. @@ -600,8 +630,8 @@ int bus_register(struct bus_type * bus) if (retval) goto bus_drivers_fail; - klist_init(&bus->klist_devices); - klist_init(&bus->klist_drivers); + klist_init(&bus->klist_devices, klist_devices_get, klist_devices_put); + klist_init(&bus->klist_drivers, klist_drivers_get, klist_drivers_put); bus_add_attrs(bus); pr_debug("bus type '%s' registered\n", bus->name); diff -puN drivers/base/core.c~fix-klist-semantics-for-lists-which-have-elements-removed drivers/base/core.c --- devel/drivers/base/core.c~fix-klist-semantics-for-lists-which-have-elements-removed 2005-08-30 18:40:38.000000000 -0700 +++ devel-akpm/drivers/base/core.c 2005-08-30 18:40:38.000000000 -0700 @@ -191,6 +191,20 @@ void device_remove_file(struct device * } } +static void klist_children_get(struct klist_node *n) +{ + struct device *dev = container_of(n, struct device, knode_parent); + + get_device(dev); +} + +static void klist_children_put(struct klist_node *n) +{ + struct device *dev = container_of(n, struct device, knode_parent); + + put_device(dev); +} + /** * device_initialize - init device structure. @@ -207,7 +221,8 @@ void device_initialize(struct device *de { kobj_set_kset_s(dev, devices_subsys); kobject_init(&dev->kobj); - klist_init(&dev->klist_children); + klist_init(&dev->klist_children, klist_children_get, + klist_children_put); INIT_LIST_HEAD(&dev->dma_pools); init_MUTEX(&dev->sem); } diff -puN drivers/base/driver.c~fix-klist-semantics-for-lists-which-have-elements-removed drivers/base/driver.c --- devel/drivers/base/driver.c~fix-klist-semantics-for-lists-which-have-elements-removed 2005-08-30 18:40:38.000000000 -0700 +++ devel-akpm/drivers/base/driver.c 2005-08-30 18:40:38.000000000 -0700 @@ -142,6 +142,19 @@ void put_driver(struct device_driver * d kobject_put(&drv->kobj); } +static void klist_devices_get(struct klist_node *n) +{ + struct device *dev = container_of(n, struct device, knode_driver); + + get_device(dev); +} + +static void klist_devices_put(struct klist_node *n) +{ + struct device *dev = container_of(n, struct device, knode_driver); + + put_device(dev); +} /** * driver_register - register driver with bus @@ -157,7 +170,7 @@ void put_driver(struct device_driver * d */ int driver_register(struct device_driver * drv) { - klist_init(&drv->klist_devices); + klist_init(&drv->klist_devices, klist_devices_get, klist_devices_put); init_completion(&drv->unloaded); return bus_add_driver(drv); } diff -puN include/linux/klist.h~fix-klist-semantics-for-lists-which-have-elements-removed include/linux/klist.h --- devel/include/linux/klist.h~fix-klist-semantics-for-lists-which-have-elements-removed 2005-08-30 18:40:38.000000000 -0700 +++ devel-akpm/include/linux/klist.h 2005-08-30 18:40:38.000000000 -0700 @@ -14,15 +14,17 @@ #include #include - +struct klist_node; struct klist { spinlock_t k_lock; struct list_head k_list; + void (*get)(struct klist_node *); + void (*put)(struct klist_node *); }; -extern void klist_init(struct klist * k); - +extern void klist_init(struct klist * k, void (*get)(struct klist_node *), + void (*put)(struct klist_node *)); struct klist_node { struct klist * n_klist; diff -puN lib/klist.c~fix-klist-semantics-for-lists-which-have-elements-removed lib/klist.c --- devel/lib/klist.c~fix-klist-semantics-for-lists-which-have-elements-removed 2005-08-30 18:40:38.000000000 -0700 +++ devel-akpm/lib/klist.c 2005-08-30 18:40:38.000000000 -0700 @@ -42,12 +42,23 @@ /** * klist_init - Initialize a klist structure. * @k: The klist we're initializing. + * @get: The get function for the embedding object (NULL if none) + * @put: The put function for the embedding object (NULL if none) + * + * Initialises the klist structure. If the klist_node structures are + * going to be embedded in refcounted objects (necessary for safe + * deletion) then the get/put arguments are used to initialise + * functions that take and release references on the embedding + * objects. */ -void klist_init(struct klist * k) +void klist_init(struct klist * k, void (*get)(struct klist_node *), + void (*put)(struct klist_node *)) { INIT_LIST_HEAD(&k->k_list); spin_lock_init(&k->k_lock); + k->get = get; + k->put = put; } EXPORT_SYMBOL_GPL(klist_init); @@ -74,6 +85,8 @@ static void klist_node_init(struct klist init_completion(&n->n_removed); kref_init(&n->n_ref); n->n_klist = k; + if (k->get) + k->get(n); } @@ -110,9 +123,12 @@ EXPORT_SYMBOL_GPL(klist_add_tail); static void klist_release(struct kref * kref) { struct klist_node * n = container_of(kref, struct klist_node, n_ref); + void (*put)(struct klist_node *) = n->n_klist->put; list_del(&n->n_node); complete(&n->n_removed); n->n_klist = NULL; + if (put) + put(n); } static int klist_dec_and_del(struct klist_node * n) _