/*
 * 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 "kcompat.h"
#include "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;

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;

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

	retval = call_usermodehelper(argv[0], argv, envp);
	if (retval) {
		printk("call_usermodehelper return %d\n", retval);
	}

	kfree(buf);
	kfree(envp);
	return retval;
}

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 ssize_t
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 ssize_t
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 ssize_t
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*)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 ssize_t
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);

EXPORT_SYMBOL(release_firmware);
EXPORT_SYMBOL(request_firmware);
EXPORT_SYMBOL(request_firmware_nowait);
EXPORT_SYMBOL(register_firmware);