Subversion Repositories or1k

/*

 * firmware_class.c - Multi purpose firmware loading support

 *

 * Copyright (c) 2003 Manuel Estrada Sainz <ranty@debian.org>

 *

 * Please see Documentation/firmware_class/ for more information.

 *

 */

/*

 * Based on kernel/kmod.c and drivers/usb/usb.c

 */

/*

        kernel/kmod.c

        Kirk Petersen

        Reorganized not to be a daemon by Adam Richter, with guidance

        from Greg Zornetzer.

        Modified to avoid chroot and file sharing problems.

        Mikael Pettersson

        Limit the concurrent number of kmod modprobes to catch loops from

        "modprobe needs a service that is in a module".

        Keith Owens <kaos@ocs.com.au> December 1999

        Unblock all signals when we exec a usermode process.

        Shuu Yamaguchi <shuu@wondernetworkresources.com> December 2000

*/

/*

 * drivers/usb/usb.c

 *

 * (C) Copyright Linus Torvalds 1999

 * (C) Copyright Johannes Erdfelt 1999-2001

 * (C) Copyright Andreas Gal 1999

 * (C) Copyright Gregory P. Smith 1999

 * (C) Copyright Deti Fliegl 1999 (new USB architecture)

 * (C) Copyright Randy Dunlap 2000

 * (C) Copyright David Brownell 2000 (kernel hotplug, usb_device_id)

 * (C) Copyright Yggdrasil Computing, Inc. 2000

 *     (usb_device_id matching changes by Adam J. Richter)

 */

#include <linux/config.h>

#include <linux/module.h>

#include <linux/string.h>

#include <linux/types.h>

#include <linux/init.h>

#include <linux/slab.h>

#include <linux/kmod.h>

#include <linux/proc_fs.h>

#include <linux/vmalloc.h>

#include <asm/hardirq.h>

#include "linux/firmware.h"

MODULE_AUTHOR("Manuel Estrada Sainz <ranty@debian.org>");

MODULE_DESCRIPTION("Multi purpose firmware loading support");

MODULE_LICENSE("GPL");

#define err(format, arg...) \

     printk(KERN_ERR  "%s:%s: " format "\n",__FILE__, __FUNCTION__ , ## arg)

#define warn(format, arg...) \

     printk(KERN_WARNING "%s:%s: " format "\n",__FILE__, __FUNCTION__ , ## arg)

#define dbg(format, arg...) \

     printk(KERN_DEBUG "%s:%s: " format "\n",__FILE__, __FUNCTION__ , ## arg)

static int loading_timeout = 10;        /* In seconds */

static struct proc_dir_entry *proc_dir_timeout;

static struct proc_dir_entry *proc_dir;

#ifdef CONFIG_HOTPLUG

static int

call_helper(char *verb, const char *name, const char *device)

{

        char *argv[3], **envp, *buf, *scratch;

        int i = 0;

        int retval = 0;

        if (!hotplug_path[0])

                return -ENOENT;

        if (in_interrupt()) {

                err("in_interrupt");

                return -EFAULT;

        }

        if (!current->fs->root) {

                warn("call_policy %s -- no FS yet", verb);

                return -EPERM;

        }

        if (!(envp = (char **) kmalloc(20 * sizeof (char *), GFP_KERNEL))) {

                err("unable to allocate envp");

                return -ENOMEM;

        }

        if (!(buf = kmalloc(256, GFP_KERNEL))) {

                kfree(envp);

                err("unable to allocate buf");

                return -ENOMEM;

        }

        /* only one standardized param to hotplug command: type */

        argv[0] = hotplug_path;

        argv[1] = "firmware";

        argv[2] = 0;

        /* minimal command environment */

        envp[i++] = "HOME=/";

        envp[i++] = "PATH=/sbin:/bin:/usr/sbin:/usr/bin";

#ifdef  DEBUG

        /* hint that policy agent should enter no-stdout debug mode */

        envp[i++] = "DEBUG=kernel";

#endif

        scratch = buf;

        if (device) {

                envp[i++] = scratch;

                scratch += snprintf(scratch, FIRMWARE_NAME_MAX+25,

                                    "DEVPATH=/driver/firmware/%s", device) + 1;

        }

        envp[i++] = scratch;

        scratch += sprintf(scratch, "ACTION=%s", verb) + 1;

        envp[i++] = scratch;

        scratch += snprintf(scratch, FIRMWARE_NAME_MAX,

                            "FIRMWARE=%s", name) + 1;

        envp[i++] = 0;

#ifdef  DEBUG

        dbg("firmware: %s %s %s", argv[0], argv[1], verb);

#endif

        retval = call_usermodehelper(argv[0], argv, envp);

        if (retval) {

                printk("call_usermodehelper return %d\n", retval);

        }

        kfree(buf);

        kfree(envp);

        return retval;

}

#else

static inline int

call_helper(char *verb, const char *name, const char *device)

{

        return -ENOENT;

}

#endif /* CONFIG_HOTPLUG */

struct firmware_priv {

        struct completion completion;

        struct proc_dir_entry *proc_dir;

        struct proc_dir_entry *attr_data;

        struct proc_dir_entry *attr_loading;

        struct firmware *fw;

        int loading;

        int abort;

        int alloc_size;

        struct timer_list timeout;

};

static int

firmware_timeout_show(char *buf, char **start, off_t off,

                      int count, int *eof, void *data)

{

        return sprintf(buf, "%d\n", loading_timeout);

}

/**

 * firmware_timeout_store:

 * Description:

 *      Sets the number of seconds to wait for the firmware.  Once

 *      this expires an error will be return to the driver and no

 *      firmware will be provided.

 *

 *      Note: zero means 'wait for ever'

 *

 **/

static int

firmware_timeout_store(struct file *file, const char *buf,

                       unsigned long count, void *data)

{

        loading_timeout = simple_strtol(buf, NULL, 10);

        return count;

}

static int

firmware_loading_show(char *buf, char **start, off_t off,

                      int count, int *eof, void *data)

{

        struct firmware_priv *fw_priv = data;

        return sprintf(buf, "%d\n", fw_priv->loading);

}

/**

 * firmware_loading_store: - loading control file

 * Description:

 *      The relevant values are:

 *

 *       1: Start a load, discarding any previous partial load.

 *       0: Conclude the load and handle the data to the driver code.

 *      -1: Conclude the load with an error and discard any written data.

 **/

static int

firmware_loading_store(struct file *file, const char *buf,

                       unsigned long count, void *data)

{

        struct firmware_priv *fw_priv = data;

        int prev_loading = fw_priv->loading;

        fw_priv->loading = simple_strtol(buf, NULL, 10);

        switch (fw_priv->loading) {

        case -1:

                fw_priv->abort = 1;

                wmb();

                complete(&fw_priv->completion);

                break;

        case 1:

                kfree(fw_priv->fw->data);

                fw_priv->fw->data = NULL;

                fw_priv->fw->size = 0;

                fw_priv->alloc_size = 0;

                break;

        case 0:

                if (prev_loading == 1)

                        complete(&fw_priv->completion);

                break;

        }

        return count;

}

static int

firmware_data_read(char *buffer, char **start, off_t offset,

                   int count, int *eof, void *data)

{

        struct firmware_priv *fw_priv = data;

        struct firmware *fw = fw_priv->fw;

        if (offset > fw->size)

                return 0;

        if (offset + count > fw->size)

                count = fw->size - offset;

        memcpy(buffer, fw->data + offset, count);

        *start = (void *) ((long) count);

        return count;

}

static int

fw_realloc_buffer(struct firmware_priv *fw_priv, int min_size)

{

        u8 *new_data;

        int new_size;

        if (min_size <= fw_priv->alloc_size)

                return 0;

        if((min_size % PAGE_SIZE) == 0)

                new_size = min_size;

        else

                new_size = (min_size + PAGE_SIZE) & PAGE_MASK;

        new_data = vmalloc(new_size);

        if (!new_data) {

                printk(KERN_ERR "%s: unable to alloc buffer\n", __FUNCTION__);

                /* Make sure that we don't keep incomplete data */

                fw_priv->abort = 1;

                return -ENOMEM;

        }

        fw_priv->alloc_size = new_size;

        if (fw_priv->fw->data) {

                memcpy(new_data, fw_priv->fw->data, fw_priv->fw->size);

                vfree(fw_priv->fw->data);

        }

        fw_priv->fw->data = new_data;

        BUG_ON(min_size > fw_priv->alloc_size);

        return 0;

}

/**

 * firmware_data_write:

 *

 * Description:

 *

 *      Data written to the 'data' attribute will be later handled to

 *      the driver as a firmware image.

 **/

static int

firmware_data_write(struct file *file, const char *buffer,

                    unsigned long count, void *data)

{

        struct firmware_priv *fw_priv = data;

        struct firmware *fw = fw_priv->fw;

        int offset = file->f_pos;

        int retval;

        retval = fw_realloc_buffer(fw_priv, offset + count);

        if (retval) {

                printk("%s: retval:%d\n", __FUNCTION__, retval);

                return retval;

        }

        memcpy(fw->data + offset, buffer, count);

        fw->size = max_t(size_t, offset + count, fw->size);

        file->f_pos += count;

        return count;

}

static void

firmware_class_timeout(u_long data)

{

        struct firmware_priv *fw_priv = (struct firmware_priv *) data;

        fw_priv->abort = 1;

        wmb();

        complete(&fw_priv->completion);

}

static int

fw_setup_class_device(struct firmware_priv **fw_priv_p,

                      const char *fw_name, const char *device)

{

        int retval;

        struct firmware_priv *fw_priv = kmalloc(sizeof (struct firmware_priv),

                                                GFP_KERNEL);

        *fw_priv_p = fw_priv;

        if (!fw_priv) {

                retval = -ENOMEM;

                goto out;

        }

        memset(fw_priv, 0, sizeof (*fw_priv));

        init_completion(&fw_priv->completion);

        fw_priv->timeout.function = firmware_class_timeout;

        fw_priv->timeout.data = (u_long) fw_priv;

        init_timer(&fw_priv->timeout);

        retval = -EAGAIN;

        fw_priv->proc_dir = create_proc_entry(device, 0644 | S_IFDIR, proc_dir);

        if (!fw_priv->proc_dir)

                goto err_free_fw_priv;

        fw_priv->attr_data = create_proc_entry("data", 0644 | S_IFREG,

                                               fw_priv->proc_dir);

        if (!fw_priv->attr_data)

                goto err_remove_dir;

        fw_priv->attr_data->read_proc = firmware_data_read;

        fw_priv->attr_data->write_proc = firmware_data_write;

        fw_priv->attr_data->data = fw_priv;

        fw_priv->attr_loading = create_proc_entry("loading", 0644 | S_IFREG,

                                                  fw_priv->proc_dir);

        if (!fw_priv->attr_loading)

                goto err_remove_data;

        fw_priv->attr_loading->read_proc = firmware_loading_show;

        fw_priv->attr_loading->write_proc = firmware_loading_store;

        fw_priv->attr_loading->data = fw_priv;

        retval = 0;

        fw_priv->fw = kmalloc(sizeof (struct firmware), GFP_KERNEL);

        if (!fw_priv->fw) {

                printk(KERN_ERR "%s: kmalloc(struct firmware) failed\n",

                       __FUNCTION__);

                retval = -ENOMEM;

                goto err_remove_loading;

        }

        memset(fw_priv->fw, 0, sizeof (*fw_priv->fw));

        goto out;

err_remove_loading:

        remove_proc_entry("loading", fw_priv->proc_dir);

err_remove_data:

        remove_proc_entry("data", fw_priv->proc_dir);

err_remove_dir:

        remove_proc_entry(device, proc_dir);

err_free_fw_priv:

        kfree(fw_priv);

out:

        return retval;

}

static void

fw_remove_class_device(struct firmware_priv *fw_priv)

{

        remove_proc_entry("loading", fw_priv->proc_dir);

        remove_proc_entry("data", fw_priv->proc_dir);

        remove_proc_entry(fw_priv->proc_dir->name, proc_dir);

}

/**

 * request_firmware: - request firmware to hotplug and wait for it

 * Description:

 *      @firmware will be used to return a firmware image by the name

 *      of @name for device @device.

 *

 *      Should be called from user context where sleeping is allowed.

 *

 *      @name will be use as $FIRMWARE in the hotplug environment and

 *      should be distinctive enough not to be confused with any other

 *      firmware image for this or any other device.

 **/

int

request_firmware(const struct firmware **firmware, const char *name,

                 const char *device)

{

        struct firmware_priv *fw_priv;

        int retval;

        if (!firmware) {

                retval = -EINVAL;

                goto out;

        }

        *firmware = NULL;

        retval = fw_setup_class_device(&fw_priv, name, device);

        if (retval)

                goto out;

        retval = call_helper("add", name, device);

        if (retval)

                goto out;

        if (loading_timeout) {

                fw_priv->timeout.expires = jiffies + loading_timeout * HZ;

                add_timer(&fw_priv->timeout);

        }

        wait_for_completion(&fw_priv->completion);

        del_timer(&fw_priv->timeout);

        fw_remove_class_device(fw_priv);

        if (fw_priv->fw->size && !fw_priv->abort) {

                *firmware = fw_priv->fw;

        } else {

                retval = -ENOENT;

                vfree(fw_priv->fw->data);

                kfree(fw_priv->fw);

        }

out:

        kfree(fw_priv);

        return retval;

}

void

release_firmware(const struct firmware *fw)

{

        if (fw) {

                vfree(fw->data);

                kfree(fw);

        }

}

/**

 * register_firmware: - provide a firmware image for later usage

 *

 * Description:

 *      Make sure that @data will be available by requesting firmware @name.

 *

 *      Note: This will not be possible until some kind of persistence

 *      is available.

 **/

void

register_firmware(const char *name, const u8 *data, size_t size)

{

        /* This is meaningless without firmware caching, so until we

         * decide if firmware caching is reasonable just leave it as a

         * noop */

}

/* Async support */

struct firmware_work {

        struct tq_struct work;

        struct module *module;

        const char *name;

        const char *device;

        void *context;

        void (*cont)(const struct firmware *fw, void *context);

};

static void

request_firmware_work_func(void *arg)

{

        struct firmware_work *fw_work = arg;

        const struct firmware *fw;

        if (!arg)

                return;

        request_firmware(&fw, fw_work->name, fw_work->device);

        fw_work->cont(fw, fw_work->context);

        release_firmware(fw);

        __MOD_DEC_USE_COUNT(fw_work->module);

        kfree(fw_work);

}

/**

 * request_firmware_nowait:

 *

 * Description:

 *      Asynchronous variant of request_firmware() for contexts where

 *      it is not possible to sleep.

 *

 *      @cont will be called asynchronously when the firmware request is over.

 *

 *      @context will be passed over to @cont.

 *

 *      @fw may be %NULL if firmware request fails.

 *

 **/

int

request_firmware_nowait(

        struct module *module,

        const char *name, const char *device, void *context,

        void (*cont)(const struct firmware *fw, void *context))

{

        struct firmware_work *fw_work = kmalloc(sizeof (struct firmware_work),

                                                GFP_ATOMIC);

        if (!fw_work)

                return -ENOMEM;

        if (!try_inc_mod_count(module)) {

                kfree(fw_work);

                return -EFAULT;

        }

        *fw_work = (struct firmware_work) {

                .module = module,

                .name = name,

                .device = device,

                .context = context,

                .cont = cont,

        };

        INIT_TQUEUE(&fw_work->work, request_firmware_work_func, fw_work);

        schedule_task(&fw_work->work);

        return 0;

}

static int __init

firmware_class_init(void)

{

        proc_dir = create_proc_entry("driver/firmware", 0755 | S_IFDIR, NULL);

        if (!proc_dir)

                return -EAGAIN;

        proc_dir_timeout = create_proc_entry("timeout",

                                             0644 | S_IFREG, proc_dir);

        if (!proc_dir_timeout) {

                remove_proc_entry("driver/firmware", NULL);

                return -EAGAIN;

        }

        proc_dir_timeout->read_proc = firmware_timeout_show;

        proc_dir_timeout->write_proc = firmware_timeout_store;

        return 0;

}

static void __exit

firmware_class_exit(void)

{

        remove_proc_entry("timeout", proc_dir);

        remove_proc_entry("driver/firmware", NULL);

}

module_init(firmware_class_init);

module_exit(firmware_class_exit);

#ifndef CONFIG_FW_LOADER

EXPORT_SYMBOL(release_firmware);

EXPORT_SYMBOL(request_firmware);

EXPORT_SYMBOL(request_firmware_nowait);

EXPORT_SYMBOL(register_firmware);

#endif