Subversion Repositories shark

/*
 * bus.c - bus driver management
 *
 * Copyright (c) 2002-3 Patrick Mochel
 * Copyright (c) 2002-3 Open Source Development Labs
 *
 * This file is released under the GPLv2
 *
 */

#undef DEBUG

#include <linuxcomp.h>

#include <linux/device.h>
#include <linux/module.h>
#include <linux/errno.h>
#include <linux/init.h>
#include <linux/string.h>
#include "base.h"

#define to_dev(node) container_of(node,struct device,bus_list)
#define to_drv(node) container_of(node,struct device_driver,kobj.entry)

#define to_bus_attr(_attr) container_of(_attr,struct bus_attribute,attr)
#define to_bus(obj) container_of(obj,struct bus_type,subsys.kset.kobj)

/*
 * sysfs bindings for drivers
 */

#define to_drv_attr(_attr) container_of(_attr,struct driver_attribute,attr)
#define to_driver(obj) container_of(obj, struct device_driver, kobj)


static ssize_t
drv_attr_show(struct kobject * kobj, struct attribute * attr, char * buf)
{
        struct driver_attribute * drv_attr = to_drv_attr(attr);
        struct device_driver * drv = to_driver(kobj);
        ssize_t ret = 0;

        if (drv_attr->show)
                ret = drv_attr->show(drv,buf);
        return ret;
}

static ssize_t
drv_attr_store(struct kobject * kobj, struct attribute * attr,
               const char * buf, size_t count)
{
        struct driver_attribute * drv_attr = to_drv_attr(attr);
        struct device_driver * drv = to_driver(kobj);
        ssize_t ret = 0;

        if (drv_attr->store)
                ret = drv_attr->store(drv,buf,count);
        return ret;
}

static struct sysfs_ops driver_sysfs_ops = {
        .show   = drv_attr_show,
        .store  = drv_attr_store,
};


static void driver_release(struct kobject * kobj)
{
        //struct device_driver * drv = to_driver(kobj);
        //up(&drv->unload_sem);
}

static struct kobj_type ktype_driver = {
        .sysfs_ops      = &driver_sysfs_ops,
        .release        = driver_release,
};


/*
 * sysfs bindings for buses
 */


static ssize_t
bus_attr_show(struct kobject * kobj, struct attribute * attr, char * buf)
{
        struct bus_attribute * bus_attr = to_bus_attr(attr);
        struct bus_type * bus = to_bus(kobj);
        ssize_t ret = 0;

        if (bus_attr->show)
                ret = bus_attr->show(bus,buf);
        return ret;
}

static ssize_t
bus_attr_store(struct kobject * kobj, struct attribute * attr,
               const char * buf, size_t count)
{
        struct bus_attribute * bus_attr = to_bus_attr(attr);
        struct bus_type * bus = to_bus(kobj);
        ssize_t ret = 0;

        if (bus_attr->store)
                ret = bus_attr->store(bus,buf,count);
        return ret;
}

static struct sysfs_ops bus_sysfs_ops = {
        .show   = bus_attr_show,
        .store  = bus_attr_store,
};

int bus_create_file(struct bus_type * bus, struct bus_attribute * attr)
{
        int error;
        if (get_bus(bus)) {
                error = 0;//sysfs_create_file(&bus->subsys.kset.kobj,&attr->attr);
                put_bus(bus);
        } else
                error = -EINVAL;
        return error;
}

void bus_remove_file(struct bus_type * bus, struct bus_attribute * attr)
{
        if (get_bus(bus)) {
                //sysfs_remove_file(&bus->subsys.kset.kobj,&attr->attr);
                put_bus(bus);
        }
}

static struct kobj_type ktype_bus = {
        .sysfs_ops      = &bus_sysfs_ops,

};

decl_subsys(bus,&ktype_bus,NULL);

/**
 *      bus_for_each_dev - device iterator.
 *      @bus:   bus type.
 *      @start: device to start iterating from.
 *      @data:  data for the callback.
 *      @fn:    function to be called for each device.
 *
 *      Iterate over @bus's list of devices, and call @fn for each,
 *      passing it @data. If @start is not NULL, we use that device to
 *      begin iterating from.
 *
 *      We check the return of @fn each time. If it returns anything
 *      other than 0, we break out and return that value.
 *
 *      NOTE: The device that returns a non-zero value is not retained
 *      in any way, nor is its refcount incremented. If the caller needs
 *      to retain this data, it should do, and increment the reference
 *      count in the supplied callback.
 */
int bus_for_each_dev(struct bus_type * bus, struct device * start,
                     void * data, int (*fn)(struct device *, void *))
{
        struct list_head * head, * entry;
        int error = 0;

        if (!(bus = get_bus(bus)))
                return -EINVAL;

        head = start ? &start->bus_list : &bus->devices.list;

        //down_read(&bus->subsys.rwsem);
        list_for_each(entry,head) {
                struct device * dev = get_device(to_dev(entry));
                error = fn(dev,data);
                put_device(dev);
                if (error)
                        break;
        }
        //up_read(&bus->subsys.rwsem);
        put_bus(bus);
        return error;
}

/**
 *      bus_for_each_drv - driver iterator
 *      @bus:   bus we're dealing with.
 *      @start: driver to start iterating on.
 *      @data:  data to pass to the callback.
 *      @fn:    function to call for each driver.
 *
 *      This is nearly identical to the device iterator above.
 *      We iterate over each driver that belongs to @bus, and call
 *      @fn for each. If @fn returns anything but 0, we break out
 *      and return it. If @start is not NULL, we use it as the head
 *      of the list.
 *
 *      NOTE: we don't return the driver that returns a non-zero
 *      value, nor do we leave the reference count incremented for that
 *      driver. If the caller needs to know that info, it must set it
 *      in the callback. It must also be sure to increment the refcount
 *      so it doesn't disappear before returning to the caller.
 */

int bus_for_each_drv(struct bus_type * bus, struct device_driver * start,
                     void * data, int (*fn)(struct device_driver *, void *))
{
        struct list_head * head, * entry;
        int error = 0;

        if(!(bus = get_bus(bus)))
                return -EINVAL;

        head = start ? &start->kobj.entry : &bus->drivers.list;

        //down_read(&bus->subsys.rwsem);
        list_for_each(entry,head) {
                struct device_driver * drv = get_driver(to_drv(entry));
                error = fn(drv,data);
                put_driver(drv);
                if(error)
                        break;
        }
        //up_read(&bus->subsys.rwsem);
        put_bus(bus);
        return error;
}

/**
 *      device_bind_driver - bind a driver to one device.
 *      @dev:   device.
 *
 *      Allow manual attachment of a driver to a deivce.
 *      Caller must have already set @dev->driver.
 *
 *      Note that this does not modify the bus reference count
 *      nor take the bus's rwsem. Please verify those are accounted
 *      for before calling this. (It is ok to call with no other effort
 *      from a driver's probe() method.)
 */

void device_bind_driver(struct device * dev)
{
        pr_debug("bound device '%s' to driver '%s'\n",
                 dev->bus_id,dev->driver->name);
        list_add_tail(&dev->driver_list,&dev->driver->devices);
        //sysfs_create_link(&dev->driver->kobj,&dev->kobj,kobject_name(&dev->kobj));
}


/**
 *      bus_match - check compatibility between device & driver.
 *      @dev:   device.
 *      @drv:   driver.
 *
 *      First, we call the bus's match function, which should compare
 *      the device IDs the driver supports with the device IDs of the
 *      device. Note we don't do this ourselves because we don't know
 *      the format of the ID structures, nor what is to be considered
 *      a match and what is not.
 *     
 *      If we find a match, we call @drv->probe(@dev) if it exists, and
 *      call attach() above.
 */
static int bus_match(struct device * dev, struct device_driver * drv)
{
        int error = -ENODEV;

        if (dev->bus->match(dev,drv)) {
                dev->driver = drv;
                if (drv->probe) {
                        if ((error = drv->probe(dev))) {
                                dev->driver = NULL;
                                return error;
                        }
                }
                device_bind_driver(dev);
                error = 0;
        }
        return error;
}


/**
 *      device_attach - try to attach device to a driver.
 *      @dev:   device.
 *
 *      Walk the list of drivers that the bus has and call bus_match()
 *      for each pair. If a compatible pair is found, break out and return.
 */
static int device_attach(struct device * dev)
{
        struct bus_type * bus = dev->bus;
        struct list_head * entry;
        int error;

        if (dev->driver) {
                device_bind_driver(dev);
                return 1;
        }

        if (bus->match) {
                list_for_each(entry,&bus->drivers.list) {
                        struct device_driver * drv = to_drv(entry);
                        error = bus_match(dev,drv);
                        if (!error )  
                                /* success, driver matched */
                                return 1;
                        if (error != -ENODEV)
                                /* driver matched but the probe failed */
                                printk(KERN_WARNING
                                    "%s: probe of %s failed with error %d\n",
                                    drv->name, dev->bus_id, error);
                }
        }

        return 0;
}


/**
 *      driver_attach - try to bind driver to devices.
 *      @drv:   driver.
 *
 *      Walk the list of devices that the bus has on it and try to match
 *      the driver with each one.
 *      If bus_match() returns 0 and the @dev->driver is set, we've found
 *      a compatible pair.
 *
 *      Note that we ignore the -ENODEV error from bus_match(), since it's
 *      perfectly valid for a driver not to bind to any devices.
 */
void driver_attach(struct device_driver * drv)
{
        struct bus_type * bus = drv->bus;
        struct list_head * entry;
        int error;

        if (!bus->match)
                return;

        list_for_each(entry,&bus->devices.list) {
                struct device * dev = container_of(entry,struct device,bus_list);
                if (!dev->driver) {
                        error = bus_match(dev,drv);
                        if (error && (error != -ENODEV))
                                /* driver matched but the probe failed */
                                printk(KERN_WARNING
                                    "%s: probe of %s failed with error %d\n",
                                    drv->name, dev->bus_id, error);
                }
        }
}


/**
 *      device_release_driver - manually detach device from driver.
 *      @dev:   device.
 *
 *      Manually detach device from driver.
 *      Note that this is called without incrementing the bus
 *      reference count nor taking the bus's rwsem. Be sure that
 *      those are accounted for before calling this function.
 */

void device_release_driver(struct device * dev)
{
        struct device_driver * drv = dev->driver;
        if (drv) {
                //sysfs_remove_link(&drv->kobj,kobject_name(&dev->kobj));
                list_del_init(&dev->driver_list);
                //device_detach_shutdown(dev);
                if (drv->remove)
                        drv->remove(dev);
                dev->driver = NULL;
        }
}


/**
 *      driver_detach - detach driver from all devices it controls.
 *      @drv:   driver.
 */

static void driver_detach(struct device_driver * drv)
{
        struct list_head * entry, * next;
        list_for_each_safe(entry,next,&drv->devices) {
                struct device * dev = container_of(entry,struct device,driver_list);
                device_release_driver(dev);
        }
       
}

/**
 *      bus_add_device - add device to bus
 *      @dev:   device being added
 *
 *      - Add the device to its bus's list of devices.
 *      - Try to attach to driver.
 *      - Create link to device's physical location.
 */
int bus_add_device(struct device * dev)
{
        struct bus_type * bus = get_bus(dev->bus);
        int error = 0;

        if (bus) {
                //down_write(&dev->bus->subsys.rwsem);
                pr_debug("bus %s: add device %s\n",bus->name,dev->bus_id);
                list_add_tail(&dev->bus_list,&dev->bus->devices.list);
                device_attach(dev);
                //up_write(&dev->bus->subsys.rwsem);
                //sysfs_create_link(&bus->devices.kobj,&dev->kobj,dev->bus_id);
        }
        return error;
}

/**
 *      bus_remove_device - remove device from bus
 *      @dev:   device to be removed
 *
 *      - Remove symlink from bus's directory.
 *      - Delete device from bus's list.
 *      - Detach from its driver.
 *      - Drop reference taken in bus_add_device().
 */
void bus_remove_device(struct device * dev)
{
        if (dev->bus) {
                //sysfs_remove_link(&dev->bus->devices.kobj,dev->bus_id);
                //down_write(&dev->bus->subsys.rwsem);
                pr_debug("bus %s: remove device %s\n",dev->bus->name,dev->bus_id);
                device_release_driver(dev);
                list_del_init(&dev->bus_list);
                //up_write(&dev->bus->subsys.rwsem);
                put_bus(dev->bus);
        }
}


/**
 *      bus_add_driver - Add a driver to the bus.
 *      @drv:   driver.
 *
 */
int bus_add_driver(struct device_driver * drv)
{
        struct bus_type * bus = get_bus(drv->bus);
        int error = 0;

        if (bus) {
                pr_debug("bus %s: add driver %s\n",bus->name,drv->name);
                kobject_set_name(&drv->kobj,drv->name);
                drv->kobj.kset = &bus->drivers;
                if ((error = kobject_register(&drv->kobj))) {
                        put_bus(bus);
                        return error;
                }

                //down_write(&bus->subsys.rwsem);
                driver_attach(drv);
                //up_write(&bus->subsys.rwsem);

        }
        return error;
}


/**
 *      bus_remove_driver - delete driver from bus's knowledge.
 *      @drv:   driver.
 *
 *      Detach the driver from the devices it controls, and remove
 *      it from its bus's list of drivers. Finally, we drop the reference
 *      to the bus we took in bus_add_driver().
 */

void bus_remove_driver(struct device_driver * drv)
{
        if (drv->bus) {
                //down_write(&drv->bus->subsys.rwsem);
                pr_debug("bus %s: remove driver %s\n",drv->bus->name,drv->name);
                driver_detach(drv);
                //up_write(&drv->bus->subsys.rwsem);
                kobject_unregister(&drv->kobj);
                put_bus(drv->bus);
        }
}


/* Helper for bus_rescan_devices's iter */
static int bus_rescan_devices_helper(struct device *dev, void *data)
{
        int *count = data;

        if (!dev->driver && device_attach(dev))
                (*count)++;

        return 0;
}


/**
 *      bus_rescan_devices - rescan devices on the bus for possible drivers
 *      @bus:   the bus to scan.
 *
 *      This function will look for devices on the bus with no driver
 *      attached and rescan it against existing drivers to see if it
 *      matches any. Calls device_attach(). Returns the number of devices
 *      that were sucessfully bound to a driver.
 */
int bus_rescan_devices(struct bus_type * bus)
{
        int count = 0;

        bus_for_each_dev(bus, NULL, &count, bus_rescan_devices_helper);

        return count;
}


struct bus_type * get_bus(struct bus_type * bus)
{
        return bus ? container_of(subsys_get(&bus->subsys),struct bus_type,subsys) : NULL;
}

void put_bus(struct bus_type * bus)
{
        subsys_put(&bus->subsys);
}


/**
 *      find_bus - locate bus by name.
 *      @name:  name of bus.
 *
 *      Call kset_find_obj() to iterate over list of buses to
 *      find a bus by name. Return bus if found.
 */

struct bus_type * find_bus(char * name)
{
        struct kobject * k = kset_find_obj(&bus_subsys.kset,name);
        return k ? to_bus(k) : NULL;
}

/**
 *      bus_register - register a bus with the system.
 *      @bus:   bus.
 *
 *      Once we have that, we registered the bus with the kobject
 *      infrastructure, then register the children subsystems it has:
 *      the devices and drivers that belong to the bus.
 */
int bus_register(struct bus_type * bus)
{
        kobject_set_name(&bus->subsys.kset.kobj,bus->name);
        subsys_set_kset(bus,bus_subsys);
        subsystem_register(&bus->subsys);

        kobject_set_name(&bus->devices.kobj, "devices");
        bus->devices.subsys = &bus->subsys;
        kset_register(&bus->devices);

        kobject_set_name(&bus->drivers.kobj, "drivers");
        bus->drivers.subsys = &bus->subsys;
        bus->drivers.ktype = &ktype_driver;
        kset_register(&bus->drivers);
 
        pr_debug("bus type '%s' registered\n",bus->name);
        return 0;
}


/**
 *      bus_unregister - remove a bus from the system
 *      @bus:   bus.
 *
 *      Unregister the child subsystems and the bus itself.
 *      Finally, we call put_bus() to release the refcount
 */
void bus_unregister(struct bus_type * bus)
{
        pr_debug("bus %s: unregistering\n",bus->name);
        kset_unregister(&bus->drivers);
        kset_unregister(&bus->devices);
        subsystem_unregister(&bus->subsys);
}

int __init buses_init(void)
{
        return subsystem_register(&bus_subsys);
}


EXPORT_SYMBOL(bus_for_each_dev);
EXPORT_SYMBOL(bus_for_each_drv);

EXPORT_SYMBOL(device_bind_driver);
EXPORT_SYMBOL(device_release_driver);

EXPORT_SYMBOL(bus_add_device);
EXPORT_SYMBOL(bus_remove_device);
EXPORT_SYMBOL(bus_register);
EXPORT_SYMBOL(bus_unregister);
EXPORT_SYMBOL(bus_rescan_devices);
EXPORT_SYMBOL(get_bus);
EXPORT_SYMBOL(put_bus);
EXPORT_SYMBOL(find_bus);

EXPORT_SYMBOL(bus_create_file);
EXPORT_SYMBOL(bus_remove_file);