[deliverable/linux.git] / fs / cachefiles / daemon.c

/* Daemon interface
 *
 * Copyright (C) 2007 Red Hat, Inc. All Rights Reserved.
 * Written by David Howells (dhowells@redhat.com)
 *
 * This program is free software; you can redistribute it and/or
 * modify it under the terms of the GNU General Public Licence
 * as published by the Free Software Foundation; either version
 * 2 of the Licence, or (at your option) any later version.
 */

#include <linux/module.h>
#include <linux/init.h>
#include <linux/sched.h>
#include <linux/completion.h>
#include <linux/slab.h>
#include <linux/fs.h>
#include <linux/file.h>
#include <linux/namei.h>
#include <linux/poll.h>
#include <linux/mount.h>
#include <linux/statfs.h>
#include <linux/ctype.h>
#include <linux/string.h>
#include <linux/fs_struct.h>
#include "internal.h"

static int cachefiles_daemon_open(struct inode *, struct file *);
static int cachefiles_daemon_release(struct inode *, struct file *);
static ssize_t cachefiles_daemon_read(struct file *, char __user *, size_t,
				      loff_t *);
static ssize_t cachefiles_daemon_write(struct file *, const char __user *,
				       size_t, loff_t *);
static unsigned int cachefiles_daemon_poll(struct file *,
					   struct poll_table_struct *);
static int cachefiles_daemon_frun(struct cachefiles_cache *, char *);
static int cachefiles_daemon_fcull(struct cachefiles_cache *, char *);
static int cachefiles_daemon_fstop(struct cachefiles_cache *, char *);
static int cachefiles_daemon_brun(struct cachefiles_cache *, char *);
static int cachefiles_daemon_bcull(struct cachefiles_cache *, char *);
static int cachefiles_daemon_bstop(struct cachefiles_cache *, char *);
static int cachefiles_daemon_cull(struct cachefiles_cache *, char *);
static int cachefiles_daemon_debug(struct cachefiles_cache *, char *);
static int cachefiles_daemon_dir(struct cachefiles_cache *, char *);
static int cachefiles_daemon_inuse(struct cachefiles_cache *, char *);
static int cachefiles_daemon_secctx(struct cachefiles_cache *, char *);
static int cachefiles_daemon_tag(struct cachefiles_cache *, char *);

static unsigned long cachefiles_open;

const struct file_operations cachefiles_daemon_fops = {
	.owner		= THIS_MODULE,
	.open		= cachefiles_daemon_open,
	.release	= cachefiles_daemon_release,
	.read		= cachefiles_daemon_read,
	.write		= cachefiles_daemon_write,
	.poll		= cachefiles_daemon_poll,
	.llseek		= noop_llseek,
};

struct cachefiles_daemon_cmd {
	char name[8];
	int (*handler)(struct cachefiles_cache *cache, char *args);
};

static const struct cachefiles_daemon_cmd cachefiles_daemon_cmds[] = {
	{ "bind",	cachefiles_daemon_bind		},
	{ "brun",	cachefiles_daemon_brun		},
	{ "bcull",	cachefiles_daemon_bcull		},
	{ "bstop",	cachefiles_daemon_bstop		},
	{ "cull",	cachefiles_daemon_cull		},
	{ "debug",	cachefiles_daemon_debug		},
	{ "dir",	cachefiles_daemon_dir		},
	{ "frun",	cachefiles_daemon_frun		},
	{ "fcull",	cachefiles_daemon_fcull		},
	{ "fstop",	cachefiles_daemon_fstop		},
	{ "inuse",	cachefiles_daemon_inuse		},
	{ "secctx",	cachefiles_daemon_secctx	},
	{ "tag",	cachefiles_daemon_tag		},
	{ "",		NULL				}
};


/*
 * do various checks
 */
static int cachefiles_daemon_open(struct inode *inode, struct file *file)
{
	struct cachefiles_cache *cache;

	_enter("");

	/* only the superuser may do this */
	if (!capable(CAP_SYS_ADMIN))
		return -EPERM;

	/* the cachefiles device may only be open once at a time */
	if (xchg(&cachefiles_open, 1) == 1)
		return -EBUSY;

	/* allocate a cache record */
	cache = kzalloc(sizeof(struct cachefiles_cache), GFP_KERNEL);
	if (!cache) {
		cachefiles_open = 0;
		return -ENOMEM;
	}

	mutex_init(&cache->daemon_mutex);
	cache->active_nodes = RB_ROOT;
	rwlock_init(&cache->active_lock);
	init_waitqueue_head(&cache->daemon_pollwq);

	/* set default caching limits
	 * - limit at 1% free space and/or free files
	 * - cull below 5% free space and/or free files
	 * - cease culling above 7% free space and/or free files
	 */
	cache->frun_percent = 7;
	cache->fcull_percent = 5;
	cache->fstop_percent = 1;
	cache->brun_percent = 7;
	cache->bcull_percent = 5;
	cache->bstop_percent = 1;

	file->private_data = cache;
	cache->cachefilesd = file;
	return 0;
}

/*
 * release a cache
 */
static int cachefiles_daemon_release(struct inode *inode, struct file *file)
{
	struct cachefiles_cache *cache = file->private_data;

	_enter("");

	ASSERT(cache);

	set_bit(CACHEFILES_DEAD, &cache->flags);

	cachefiles_daemon_unbind(cache);

	ASSERT(!cache->active_nodes.rb_node);

	/* clean up the control file interface */
	cache->cachefilesd = NULL;
	file->private_data = NULL;
	cachefiles_open = 0;

	kfree(cache);

	_leave("");
	return 0;
}

/*
 * read the cache state
 */
static ssize_t cachefiles_daemon_read(struct file *file, char __user *_buffer,
				      size_t buflen, loff_t *pos)
{
	struct cachefiles_cache *cache = file->private_data;
	unsigned long long b_released;
	unsigned f_released;
	char buffer[256];
	int n;

	//_enter(",,%zu,", buflen);

	if (!test_bit(CACHEFILES_READY, &cache->flags))
		return 0;

	/* check how much space the cache has */
	cachefiles_has_space(cache, 0, 0);

	/* summarise */
	f_released = atomic_xchg(&cache->f_released, 0);
	b_released = atomic_long_xchg(&cache->b_released, 0);
	clear_bit(CACHEFILES_STATE_CHANGED, &cache->flags);

	n = snprintf(buffer, sizeof(buffer),
		     "cull=%c"
		     " frun=%llx"
		     " fcull=%llx"
		     " fstop=%llx"
		     " brun=%llx"
		     " bcull=%llx"
		     " bstop=%llx"
		     " freleased=%x"
		     " breleased=%llx",
		     test_bit(CACHEFILES_CULLING, &cache->flags) ? '1' : '0',
		     (unsigned long long) cache->frun,
		     (unsigned long long) cache->fcull,
		     (unsigned long long) cache->fstop,
		     (unsigned long long) cache->brun,
		     (unsigned long long) cache->bcull,
		     (unsigned long long) cache->bstop,
		     f_released,
		     b_released);

	if (n > buflen)
		return -EMSGSIZE;

	if (copy_to_user(_buffer, buffer, n) != 0)
		return -EFAULT;

	return n;
}

/*
 * command the cache
 */
static ssize_t cachefiles_daemon_write(struct file *file,
				       const char __user *_data,
				       size_t datalen,
				       loff_t *pos)
{
	const struct cachefiles_daemon_cmd *cmd;
	struct cachefiles_cache *cache = file->private_data;
	ssize_t ret;
	char *data, *args, *cp;

	//_enter(",,%zu,", datalen);

	ASSERT(cache);

	if (test_bit(CACHEFILES_DEAD, &cache->flags))
		return -EIO;

	if (datalen < 0 || datalen > PAGE_SIZE - 1)
		return -EOPNOTSUPP;

	/* drag the command string into the kernel so we can parse it */
	data = memdup_user_nul(_data, datalen);
	if (IS_ERR(data))
		return PTR_ERR(data);

	ret = -EINVAL;
	if (memchr(data, '\0', datalen))
		goto error;

	/* strip any newline */
	cp = memchr(data, '\n', datalen);
	if (cp) {
		if (cp == data)
			goto error;

		*cp = '\0';
	}

	/* parse the command */
	ret = -EOPNOTSUPP;

	for (args = data; *args; args++)
		if (isspace(*args))
			break;
	if (*args) {
		if (args == data)
			goto error;
		*args = '\0';
		args = skip_spaces(++args);
	}

	/* run the appropriate command handler */
	for (cmd = cachefiles_daemon_cmds; cmd->name[0]; cmd++)
		if (strcmp(cmd->name, data) == 0)
			goto found_command;

error:
	kfree(data);
	//_leave(" = %zd", ret);
	return ret;

found_command:
	mutex_lock(&cache->daemon_mutex);

	ret = -EIO;
	if (!test_bit(CACHEFILES_DEAD, &cache->flags))
		ret = cmd->handler(cache, args);

	mutex_unlock(&cache->daemon_mutex);

	if (ret == 0)
		ret = datalen;
	goto error;
}

/*
 * poll for culling state
 * - use POLLOUT to indicate culling state
 */
static unsigned int cachefiles_daemon_poll(struct file *file,
					   struct poll_table_struct *poll)
{
	struct cachefiles_cache *cache = file->private_data;
	unsigned int mask;

	poll_wait(file, &cache->daemon_pollwq, poll);
	mask = 0;

	if (test_bit(CACHEFILES_STATE_CHANGED, &cache->flags))
		mask |= POLLIN;

	if (test_bit(CACHEFILES_CULLING, &cache->flags))
		mask |= POLLOUT;

	return mask;
}

/*
 * give a range error for cache space constraints
 * - can be tail-called
 */
static int cachefiles_daemon_range_error(struct cachefiles_cache *cache,
					 char *args)
{
	pr_err("Free space limits must be in range 0%%<=stop<cull<run<100%%\n");

	return -EINVAL;
}

/*
 * set the percentage of files at which to stop culling
 * - command: "frun <N>%"
 */
static int cachefiles_daemon_frun(struct cachefiles_cache *cache, char *args)
{
	unsigned long frun;

	_enter(",%s", args);

	if (!*args)
		return -EINVAL;

	frun = simple_strtoul(args, &args, 10);
	if (args[0] != '%' || args[1] != '\0')
		return -EINVAL;

	if (frun <= cache->fcull_percent || frun >= 100)
		return cachefiles_daemon_range_error(cache, args);

	cache->frun_percent = frun;
	return 0;
}

/*
 * set the percentage of files at which to start culling
 * - command: "fcull <N>%"
 */
static int cachefiles_daemon_fcull(struct cachefiles_cache *cache, char *args)
{
	unsigned long fcull;

	_enter(",%s", args);

	if (!*args)
		return -EINVAL;

	fcull = simple_strtoul(args, &args, 10);
	if (args[0] != '%' || args[1] != '\0')
		return -EINVAL;

	if (fcull <= cache->fstop_percent || fcull >= cache->frun_percent)
		return cachefiles_daemon_range_error(cache, args);

	cache->fcull_percent = fcull;
	return 0;
}

/*
 * set the percentage of files at which to stop allocating
 * - command: "fstop <N>%"
 */
static int cachefiles_daemon_fstop(struct cachefiles_cache *cache, char *args)
{
	unsigned long fstop;

	_enter(",%s", args);

	if (!*args)
		return -EINVAL;

	fstop = simple_strtoul(args, &args, 10);
	if (args[0] != '%' || args[1] != '\0')
		return -EINVAL;

	if (fstop < 0 || fstop >= cache->fcull_percent)
		return cachefiles_daemon_range_error(cache, args);

	cache->fstop_percent = fstop;
	return 0;
}

/*
 * set the percentage of blocks at which to stop culling
 * - command: "brun <N>%"
 */
static int cachefiles_daemon_brun(struct cachefiles_cache *cache, char *args)
{
	unsigned long brun;

	_enter(",%s", args);

	if (!*args)
		return -EINVAL;

	brun = simple_strtoul(args, &args, 10);
	if (args[0] != '%' || args[1] != '\0')
		return -EINVAL;

	if (brun <= cache->bcull_percent || brun >= 100)
		return cachefiles_daemon_range_error(cache, args);

	cache->brun_percent = brun;
	return 0;
}

/*
 * set the percentage of blocks at which to start culling
 * - command: "bcull <N>%"
 */
static int cachefiles_daemon_bcull(struct cachefiles_cache *cache, char *args)
{
	unsigned long bcull;

	_enter(",%s", args);

	if (!*args)
		return -EINVAL;

	bcull = simple_strtoul(args, &args, 10);
	if (args[0] != '%' || args[1] != '\0')
		return -EINVAL;

	if (bcull <= cache->bstop_percent || bcull >= cache->brun_percent)
		return cachefiles_daemon_range_error(cache, args);

	cache->bcull_percent = bcull;
	return 0;
}

/*
 * set the percentage of blocks at which to stop allocating
 * - command: "bstop <N>%"
 */
static int cachefiles_daemon_bstop(struct cachefiles_cache *cache, char *args)
{
	unsigned long bstop;

	_enter(",%s", args);

	if (!*args)
		return -EINVAL;

	bstop = simple_strtoul(args, &args, 10);
	if (args[0] != '%' || args[1] != '\0')
		return -EINVAL;

	if (bstop < 0 || bstop >= cache->bcull_percent)
		return cachefiles_daemon_range_error(cache, args);

	cache->bstop_percent = bstop;
	return 0;
}

/*
 * set the cache directory
 * - command: "dir <name>"
 */
static int cachefiles_daemon_dir(struct cachefiles_cache *cache, char *args)
{
	char *dir;

	_enter(",%s", args);

	if (!*args) {
		pr_err("Empty directory specified\n");
		return -EINVAL;
	}

	if (cache->rootdirname) {
		pr_err("Second cache directory specified\n");
		return -EEXIST;
	}

	dir = kstrdup(args, GFP_KERNEL);
	if (!dir)
		return -ENOMEM;

	cache->rootdirname = dir;
	return 0;
}

/*
 * set the cache security context
 * - command: "secctx <ctx>"
 */
static int cachefiles_daemon_secctx(struct cachefiles_cache *cache, char *args)
{
	char *secctx;

	_enter(",%s", args);

	if (!*args) {
		pr_err("Empty security context specified\n");
		return -EINVAL;
	}

	if (cache->secctx) {
		pr_err("Second security context specified\n");
		return -EINVAL;
	}

	secctx = kstrdup(args, GFP_KERNEL);
	if (!secctx)
		return -ENOMEM;

	cache->secctx = secctx;
	return 0;
}

/*
 * set the cache tag
 * - command: "tag <name>"
 */
static int cachefiles_daemon_tag(struct cachefiles_cache *cache, char *args)
{
	char *tag;

	_enter(",%s", args);

	if (!*args) {
		pr_err("Empty tag specified\n");
		return -EINVAL;
	}

	if (cache->tag)
		return -EEXIST;

	tag = kstrdup(args, GFP_KERNEL);
	if (!tag)
		return -ENOMEM;

	cache->tag = tag;
	return 0;
}

/*
 * request a node in the cache be culled from the current working directory
 * - command: "cull <name>"
 */
static int cachefiles_daemon_cull(struct cachefiles_cache *cache, char *args)
{
	struct path path;
	const struct cred *saved_cred;
	int ret;

	_enter(",%s", args);

	if (strchr(args, '/'))
		goto inval;

	if (!test_bit(CACHEFILES_READY, &cache->flags)) {
		pr_err("cull applied to unready cache\n");
		return -EIO;
	}

	if (test_bit(CACHEFILES_DEAD, &cache->flags)) {
		pr_err("cull applied to dead cache\n");
		return -EIO;
	}

	/* extract the directory dentry from the cwd */
	get_fs_pwd(current->fs, &path);

	if (!d_can_lookup(path.dentry))
		goto notdir;

	cachefiles_begin_secure(cache, &saved_cred);
	ret = cachefiles_cull(cache, path.dentry, args);
	cachefiles_end_secure(cache, saved_cred);

	path_put(&path);
	_leave(" = %d", ret);
	return ret;

notdir:
	path_put(&path);
	pr_err("cull command requires dirfd to be a directory\n");
	return -ENOTDIR;

inval:
	pr_err("cull command requires dirfd and filename\n");
	return -EINVAL;
}

/*
 * set debugging mode
 * - command: "debug <mask>"
 */
static int cachefiles_daemon_debug(struct cachefiles_cache *cache, char *args)
{
	unsigned long mask;

	_enter(",%s", args);

	mask = simple_strtoul(args, &args, 0);
	if (args[0] != '\0')
		goto inval;

	cachefiles_debug = mask;
	_leave(" = 0");
	return 0;

inval:
	pr_err("debug command requires mask\n");
	return -EINVAL;
}

/*
 * find out whether an object in the current working directory is in use or not
 * - command: "inuse <name>"
 */
static int cachefiles_daemon_inuse(struct cachefiles_cache *cache, char *args)
{
	struct path path;
	const struct cred *saved_cred;
	int ret;

	//_enter(",%s", args);

	if (strchr(args, '/'))
		goto inval;

	if (!test_bit(CACHEFILES_READY, &cache->flags)) {
		pr_err("inuse applied to unready cache\n");
		return -EIO;
	}

	if (test_bit(CACHEFILES_DEAD, &cache->flags)) {
		pr_err("inuse applied to dead cache\n");
		return -EIO;
	}

	/* extract the directory dentry from the cwd */
	get_fs_pwd(current->fs, &path);

	if (!d_can_lookup(path.dentry))
		goto notdir;

	cachefiles_begin_secure(cache, &saved_cred);
	ret = cachefiles_check_in_use(cache, path.dentry, args);
	cachefiles_end_secure(cache, saved_cred);

	path_put(&path);
	//_leave(" = %d", ret);
	return ret;

notdir:
	path_put(&path);
	pr_err("inuse command requires dirfd to be a directory\n");
	return -ENOTDIR;

inval:
	pr_err("inuse command requires dirfd and filename\n");
	return -EINVAL;
}

/*
 * see if we have space for a number of pages and/or a number of files in the
 * cache
 */
int cachefiles_has_space(struct cachefiles_cache *cache,
			 unsigned fnr, unsigned bnr)
{
	struct kstatfs stats;
	struct path path = {
		.mnt	= cache->mnt,
		.dentry	= cache->mnt->mnt_root,
	};
	int ret;

	//_enter("{%llu,%llu,%llu,%llu,%llu,%llu},%u,%u",
	//       (unsigned long long) cache->frun,
	//       (unsigned long long) cache->fcull,
	//       (unsigned long long) cache->fstop,
	//       (unsigned long long) cache->brun,
	//       (unsigned long long) cache->bcull,
	//       (unsigned long long) cache->bstop,
	//       fnr, bnr);

	/* find out how many pages of blockdev are available */
	memset(&stats, 0, sizeof(stats));

	ret = vfs_statfs(&path, &stats);
	if (ret < 0) {
		if (ret == -EIO)
			cachefiles_io_error(cache, "statfs failed");
		_leave(" = %d", ret);
		return ret;
	}

	stats.f_bavail >>= cache->bshift;

	//_debug("avail %llu,%llu",
	//       (unsigned long long) stats.f_ffree,
	//       (unsigned long long) stats.f_bavail);

	/* see if there is sufficient space */
	if (stats.f_ffree > fnr)
		stats.f_ffree -= fnr;
	else
		stats.f_ffree = 0;

	if (stats.f_bavail > bnr)
		stats.f_bavail -= bnr;
	else
		stats.f_bavail = 0;

	ret = -ENOBUFS;
	if (stats.f_ffree < cache->fstop ||
	    stats.f_bavail < cache->bstop)
		goto begin_cull;

	ret = 0;
	if (stats.f_ffree < cache->fcull ||
	    stats.f_bavail < cache->bcull)
		goto begin_cull;

	if (test_bit(CACHEFILES_CULLING, &cache->flags) &&
	    stats.f_ffree >= cache->frun &&
	    stats.f_bavail >= cache->brun &&
	    test_and_clear_bit(CACHEFILES_CULLING, &cache->flags)
	    ) {
		_debug("cease culling");
		cachefiles_state_changed(cache);
	}

	//_leave(" = 0");
	return 0;

begin_cull:
	if (!test_and_set_bit(CACHEFILES_CULLING, &cache->flags)) {
		_debug("### CULL CACHE ###");
		cachefiles_state_changed(cache);
	}

	_leave(" = %d", ret);
	return ret;
}
Commit	Line	Data
9ae326a6 DH	1	/* Daemon interface
	2	*
	3	* Copyright (C) 2007 Red Hat, Inc. All Rights Reserved.
	4	* Written by David Howells (dhowells@redhat.com)
	5	*
	6	* This program is free software; you can redistribute it and/or
	7	* modify it under the terms of the GNU General Public Licence
	8	* as published by the Free Software Foundation; either version
	9	* 2 of the Licence, or (at your option) any later version.
	10	*/
	11
	12	#include <linux/module.h>
	13	#include <linux/init.h>
	14	#include <linux/sched.h>
	15	#include <linux/completion.h>
	16	#include <linux/slab.h>
	17	#include <linux/fs.h>
	18	#include <linux/file.h>
	19	#include <linux/namei.h>
	20	#include <linux/poll.h>
	21	#include <linux/mount.h>
	22	#include <linux/statfs.h>
	23	#include <linux/ctype.h>
e7d2860b	24	#include <linux/string.h>
9ae326a6 DH	25	#include <linux/fs_struct.h>
	26	#include "internal.h"
	27
	28	static int cachefiles_daemon_open(struct inode , struct file );
	29	static int cachefiles_daemon_release(struct inode , struct file );
	30	static ssize_t cachefiles_daemon_read(struct file , char __user , size_t,
	31	loff_t *);
	32	static ssize_t cachefiles_daemon_write(struct file , const char __user ,
	33	size_t, loff_t *);
	34	static unsigned int cachefiles_daemon_poll(struct file *,
	35	struct poll_table_struct *);
	36	static int cachefiles_daemon_frun(struct cachefiles_cache , char );
	37	static int cachefiles_daemon_fcull(struct cachefiles_cache , char );
	38	static int cachefiles_daemon_fstop(struct cachefiles_cache , char );
	39	static int cachefiles_daemon_brun(struct cachefiles_cache , char );
	40	static int cachefiles_daemon_bcull(struct cachefiles_cache , char );
	41	static int cachefiles_daemon_bstop(struct cachefiles_cache , char );
	42	static int cachefiles_daemon_cull(struct cachefiles_cache , char );
	43	static int cachefiles_daemon_debug(struct cachefiles_cache , char );
	44	static int cachefiles_daemon_dir(struct cachefiles_cache , char );
	45	static int cachefiles_daemon_inuse(struct cachefiles_cache , char );
	46	static int cachefiles_daemon_secctx(struct cachefiles_cache , char );
	47	static int cachefiles_daemon_tag(struct cachefiles_cache , char );
	48
	49	static unsigned long cachefiles_open;
	50
	51	const struct file_operations cachefiles_daemon_fops = {
	52	.owner = THIS_MODULE,
	53	.open = cachefiles_daemon_open,
	54	.release = cachefiles_daemon_release,
	55	.read = cachefiles_daemon_read,
	56	.write = cachefiles_daemon_write,
	57	.poll = cachefiles_daemon_poll,
6038f373	58	.llseek = noop_llseek,
9ae326a6 DH	59	};
	60
	61	struct cachefiles_daemon_cmd {
	62	char name[8];
	63	int (handler)(struct cachefiles_cache cache, char *args);
	64	};
	65
	66	static const struct cachefiles_daemon_cmd cachefiles_daemon_cmds[] = {
	67	{ "bind", cachefiles_daemon_bind },
	68	{ "brun", cachefiles_daemon_brun },
	69	{ "bcull", cachefiles_daemon_bcull },
	70	{ "bstop", cachefiles_daemon_bstop },
	71	{ "cull", cachefiles_daemon_cull },
	72	{ "debug", cachefiles_daemon_debug },
	73	{ "dir", cachefiles_daemon_dir },
	74	{ "frun", cachefiles_daemon_frun },
	75	{ "fcull", cachefiles_daemon_fcull },
	76	{ "fstop", cachefiles_daemon_fstop },
	77	{ "inuse", cachefiles_daemon_inuse },
	78	{ "secctx", cachefiles_daemon_secctx },
	79	{ "tag", cachefiles_daemon_tag },
	80	{ "", NULL }
	81	};
	82
	83
	84	/*
	85	* do various checks
	86	*/
	87	static int cachefiles_daemon_open(struct inode inode, struct file file)
	88	{
	89	struct cachefiles_cache *cache;
	90
	91	_enter("");
	92
	93	/* only the superuser may do this */
	94	if (!capable(CAP_SYS_ADMIN))
	95	return -EPERM;
	96
	97	/* the cachefiles device may only be open once at a time */
	98	if (xchg(&cachefiles_open, 1) == 1)
	99	return -EBUSY;
	100
	101	/* allocate a cache record */
	102	cache = kzalloc(sizeof(struct cachefiles_cache), GFP_KERNEL);
	103	if (!cache) {
	104	cachefiles_open = 0;
	105	return -ENOMEM;
	106	}
	107
	108	mutex_init(&cache->daemon_mutex);
	109	cache->active_nodes = RB_ROOT;
	110	rwlock_init(&cache->active_lock);
	111	init_waitqueue_head(&cache->daemon_pollwq);
	112
	113	/* set default caching limits
	114	* - limit at 1% free space and/or free files
	115	* - cull below 5% free space and/or free files
	116	* - cease culling above 7% free space and/or free files
	117	*/
	118	cache->frun_percent = 7;
	119	cache->fcull_percent = 5;
	120	cache->fstop_percent = 1;
	121	cache->brun_percent = 7;
	122	cache->bcull_percent = 5;
123	cache->bstop_percent = 1;
124
125	file->private_data = cache;
126	cache->cachefilesd = file;
127	return 0;
128	}
129
130	/*
131	* release a cache
132	*/
133	static int cachefiles_daemon_release(struct inode inode, struct file file)
134	{
135	struct cachefiles_cache *cache = file->private_data;
136
137	_enter("");
138
139	ASSERT(cache);
140
141	set_bit(CACHEFILES_DEAD, &cache->flags);
142
143	cachefiles_daemon_unbind(cache);
144
145	ASSERT(!cache->active_nodes.rb_node);
146
147	/* clean up the control file interface */
148	cache->cachefilesd = NULL;
149	file->private_data = NULL;
150	cachefiles_open = 0;
151
152	kfree(cache);
153
154	_leave("");
155	return 0;
156	}
157
158	/*
159	* read the cache state
160	*/
161	static ssize_t cachefiles_daemon_read(struct file file, char __user _buffer,
162	size_t buflen, loff_t *pos)
163	{
164	struct cachefiles_cache *cache = file->private_data;
a5b3a80b DH	165	unsigned long long b_released;
a5b3a80b DH	166	unsigned f_released;
9ae326a6 DH	167	char buffer[256];
	168	int n;
	169
	170	//_enter(",,%zu,", buflen);
	171
	172	if (!test_bit(CACHEFILES_READY, &cache->flags))
	173	return 0;
	174
	175	/* check how much space the cache has */
	176	cachefiles_has_space(cache, 0, 0);
	177
	178	/* summarise */
a5b3a80b DH	179	f_released = atomic_xchg(&cache->f_released, 0);
a5b3a80b DH	180	b_released = atomic_long_xchg(&cache->b_released, 0);
9ae326a6 DH	181	clear_bit(CACHEFILES_STATE_CHANGED, &cache->flags);
	182
	183	n = snprintf(buffer, sizeof(buffer),
	184	"cull=%c"
	185	" frun=%llx"
	186	" fcull=%llx"
	187	" fstop=%llx"
	188	" brun=%llx"
	189	" bcull=%llx"
a5b3a80b DH	190	" bstop=%llx"
	191	" freleased=%x"
	192	" breleased=%llx",
9ae326a6 DH	193	test_bit(CACHEFILES_CULLING, &cache->flags) ? '1' : '0',
	194	(unsigned long long) cache->frun,
	195	(unsigned long long) cache->fcull,
	196	(unsigned long long) cache->fstop,
	197	(unsigned long long) cache->brun,
	198	(unsigned long long) cache->bcull,
a5b3a80b DH	199	(unsigned long long) cache->bstop,
	200	f_released,
	201	b_released);
9ae326a6 DH	202
	203	if (n > buflen)
	204	return -EMSGSIZE;
	205
	206	if (copy_to_user(_buffer, buffer, n) != 0)
	207	return -EFAULT;
	208
	209	return n;
	210	}
	211
	212	/*
	213	* command the cache
	214	*/
	215	static ssize_t cachefiles_daemon_write(struct file *file,
	216	const char __user *_data,
	217	size_t datalen,
	218	loff_t *pos)
	219	{
	220	const struct cachefiles_daemon_cmd *cmd;
	221	struct cachefiles_cache *cache = file->private_data;
	222	ssize_t ret;
	223	char data, args, *cp;
	224
	225	//_enter(",,%zu,", datalen);
	226
	227	ASSERT(cache);
	228
	229	if (test_bit(CACHEFILES_DEAD, &cache->flags))
	230	return -EIO;
	231
	232	if (datalen < 0 \|\| datalen > PAGE_SIZE - 1)
	233	return -EOPNOTSUPP;
	234
	235	/* drag the command string into the kernel so we can parse it */
16e5c1fc AV	236	data = memdup_user_nul(_data, datalen);
	237	if (IS_ERR(data))
	238	return PTR_ERR(data);
9ae326a6 DH	239
	240	ret = -EINVAL;
	241	if (memchr(data, '\0', datalen))
	242	goto error;
	243
	244	/* strip any newline */
	245	cp = memchr(data, '\n', datalen);
	246	if (cp) {
	247	if (cp == data)
	248	goto error;
	249
	250	*cp = '\0';
	251	}
	252
	253	/* parse the command */
	254	ret = -EOPNOTSUPP;
	255
	256	for (args = data; *args; args++)
	257	if (isspace(*args))
	258	break;
	259	if (*args) {
	260	if (args == data)
	261	goto error;
	262	*args = '\0';
e7d2860b	263	args = skip_spaces(++args);
9ae326a6 DH	264	}
	265
	266	/* run the appropriate command handler */
	267	for (cmd = cachefiles_daemon_cmds; cmd->name[0]; cmd++)
	268	if (strcmp(cmd->name, data) == 0)
	269	goto found_command;
	270
	271	error:
	272	kfree(data);
	273	//_leave(" = %zd", ret);
	274	return ret;
	275
	276	found_command:
	277	mutex_lock(&cache->daemon_mutex);
	278
	279	ret = -EIO;
	280	if (!test_bit(CACHEFILES_DEAD, &cache->flags))
	281	ret = cmd->handler(cache, args);
	282
	283	mutex_unlock(&cache->daemon_mutex);
	284
	285	if (ret == 0)
	286	ret = datalen;
	287	goto error;
	288	}
	289
	290	/*
	291	* poll for culling state
	292	* - use POLLOUT to indicate culling state
	293	*/
	294	static unsigned int cachefiles_daemon_poll(struct file *file,
	295	struct poll_table_struct *poll)
	296	{
	297	struct cachefiles_cache *cache = file->private_data;
	298	unsigned int mask;
	299
	300	poll_wait(file, &cache->daemon_pollwq, poll);
	301	mask = 0;
	302
	303	if (test_bit(CACHEFILES_STATE_CHANGED, &cache->flags))
	304	mask \|= POLLIN;
	305
	306	if (test_bit(CACHEFILES_CULLING, &cache->flags))
	307	mask \|= POLLOUT;
	308
	309	return mask;
	310	}
	311
	312	/*
	313	* give a range error for cache space constraints
	314	* - can be tail-called
	315	*/
	316	static int cachefiles_daemon_range_error(struct cachefiles_cache *cache,
	317	char *args)
	318	{
6ff66ac7	319	pr_err("Free space limits must be in range 0%%<=stop<cull<run<100%%\n");
9ae326a6 DH	320
	321	return -EINVAL;
	322	}
	323
	324	/*
	325	* set the percentage of files at which to stop culling
	326	* - command: "frun <N>%"
	327	*/
	328	static int cachefiles_daemon_frun(struct cachefiles_cache cache, char args)
	329	{
	330	unsigned long frun;
	331
	332	_enter(",%s", args);
	333
	334	if (!*args)
	335	return -EINVAL;
	336
	337	frun = simple_strtoul(args, &args, 10);
	338	if (args[0] != '%' \|\| args[1] != '\0')
	339	return -EINVAL;
	340
	341	if (frun <= cache->fcull_percent \|\| frun >= 100)
	342	return cachefiles_daemon_range_error(cache, args);
	343
	344	cache->frun_percent = frun;
	345	return 0;
	346	}
	347
	348	/*
	349	* set the percentage of files at which to start culling
	350	* - command: "fcull <N>%"
	351	*/
	352	static int cachefiles_daemon_fcull(struct cachefiles_cache cache, char args)
	353	{
	354	unsigned long fcull;
	355
	356	_enter(",%s", args);
	357
	358	if (!*args)
	359	return -EINVAL;
	360
	361	fcull = simple_strtoul(args, &args, 10);
	362	if (args[0] != '%' \|\| args[1] != '\0')
	363	return -EINVAL;
	364
	365	if (fcull <= cache->fstop_percent \|\| fcull >= cache->frun_percent)
	366	return cachefiles_daemon_range_error(cache, args);
	367
	368	cache->fcull_percent = fcull;
	369	return 0;
	370	}
	371
	372	/*
	373	* set the percentage of files at which to stop allocating
	374	* - command: "fstop <N>%"
	375	*/
	376	static int cachefiles_daemon_fstop(struct cachefiles_cache cache, char args)
	377	{
	378	unsigned long fstop;
	379
	380	_enter(",%s", args);
	381
	382	if (!*args)
	383	return -EINVAL;
384
385	fstop = simple_strtoul(args, &args, 10);
386	if (args[0] != '%' \|\| args[1] != '\0')
387	return -EINVAL;
388
389	if (fstop < 0 \|\| fstop >= cache->fcull_percent)
390	return cachefiles_daemon_range_error(cache, args);
391
392	cache->fstop_percent = fstop;
393	return 0;
394	}
395
396	/*
397	* set the percentage of blocks at which to stop culling
398	* - command: "brun <N>%"
399	*/
400	static int cachefiles_daemon_brun(struct cachefiles_cache cache, char args)
401	{
402	unsigned long brun;
403
404	_enter(",%s", args);
405
406	if (!*args)
407	return -EINVAL;
408
409	brun = simple_strtoul(args, &args, 10);
410	if (args[0] != '%' \|\| args[1] != '\0')
411	return -EINVAL;
412
413	if (brun <= cache->bcull_percent \|\| brun >= 100)
414	return cachefiles_daemon_range_error(cache, args);
415
416	cache->brun_percent = brun;
417	return 0;
418	}
419
420	/*
421	* set the percentage of blocks at which to start culling
422	* - command: "bcull <N>%"
423	*/
424	static int cachefiles_daemon_bcull(struct cachefiles_cache cache, char args)
425	{
426	unsigned long bcull;
427
428	_enter(",%s", args);
429
430	if (!*args)
431	return -EINVAL;
432
433	bcull = simple_strtoul(args, &args, 10);
434	if (args[0] != '%' \|\| args[1] != '\0')
435	return -EINVAL;
436
437	if (bcull <= cache->bstop_percent \|\| bcull >= cache->brun_percent)
438	return cachefiles_daemon_range_error(cache, args);
439
440	cache->bcull_percent = bcull;
441	return 0;
442	}
443
444	/*
445	* set the percentage of blocks at which to stop allocating
446	* - command: "bstop <N>%"
447	*/
448	static int cachefiles_daemon_bstop(struct cachefiles_cache cache, char args)
449	{
450	unsigned long bstop;
451
452	_enter(",%s", args);
453
454	if (!*args)
455	return -EINVAL;
456
457	bstop = simple_strtoul(args, &args, 10);
458	if (args[0] != '%' \|\| args[1] != '\0')
459	return -EINVAL;
460
461	if (bstop < 0 \|\| bstop >= cache->bcull_percent)
462	return cachefiles_daemon_range_error(cache, args);
463
464	cache->bstop_percent = bstop;
465	return 0;
466	}
467
468	/*
469	* set the cache directory
470	* - command: "dir <name>"
471	*/
472	static int cachefiles_daemon_dir(struct cachefiles_cache cache, char args)
473	{
474	char *dir;
475
476	_enter(",%s", args);
477
478	if (!*args) {
6ff66ac7	479	pr_err("Empty directory specified\n");
9ae326a6 DH	480	return -EINVAL;
	481	}
	482
	483	if (cache->rootdirname) {
6ff66ac7	484	pr_err("Second cache directory specified\n");
9ae326a6 DH	485	return -EEXIST;
	486	}
	487
	488	dir = kstrdup(args, GFP_KERNEL);
	489	if (!dir)
	490	return -ENOMEM;
	491
	492	cache->rootdirname = dir;
	493	return 0;
	494	}
	495
	496	/*
	497	* set the cache security context
	498	* - command: "secctx <ctx>"
	499	*/
	500	static int cachefiles_daemon_secctx(struct cachefiles_cache cache, char args)
	501	{
	502	char *secctx;
	503
	504	_enter(",%s", args);
	505
	506	if (!*args) {
6ff66ac7	507	pr_err("Empty security context specified\n");
9ae326a6 DH	508	return -EINVAL;
	509	}
	510
	511	if (cache->secctx) {
6ff66ac7	512	pr_err("Second security context specified\n");
9ae326a6 DH	513	return -EINVAL;
	514	}
	515
	516	secctx = kstrdup(args, GFP_KERNEL);
	517	if (!secctx)
	518	return -ENOMEM;
	519
	520	cache->secctx = secctx;
	521	return 0;
	522	}
	523
	524	/*
	525	* set the cache tag
	526	* - command: "tag <name>"
	527	*/
	528	static int cachefiles_daemon_tag(struct cachefiles_cache cache, char args)
	529	{
	530	char *tag;
	531
	532	_enter(",%s", args);
	533
	534	if (!*args) {
6ff66ac7	535	pr_err("Empty tag specified\n");
9ae326a6 DH	536	return -EINVAL;
	537	}
	538
	539	if (cache->tag)
	540	return -EEXIST;
	541
	542	tag = kstrdup(args, GFP_KERNEL);
	543	if (!tag)
	544	return -ENOMEM;
	545
	546	cache->tag = tag;
	547	return 0;
	548	}
	549
	550	/*
	551	* request a node in the cache be culled from the current working directory
	552	* - command: "cull <name>"
	553	*/
	554	static int cachefiles_daemon_cull(struct cachefiles_cache cache, char args)
	555	{
542ce7a9	556	struct path path;
9ae326a6 DH	557	const struct cred *saved_cred;
	558	int ret;
	559
	560	_enter(",%s", args);
	561
	562	if (strchr(args, '/'))
	563	goto inval;
	564
	565	if (!test_bit(CACHEFILES_READY, &cache->flags)) {
6ff66ac7	566	pr_err("cull applied to unready cache\n");
9ae326a6 DH	567	return -EIO;
	568	}
	569
	570	if (test_bit(CACHEFILES_DEAD, &cache->flags)) {
6ff66ac7	571	pr_err("cull applied to dead cache\n");
9ae326a6 DH	572	return -EIO;
	573	}
	574
	575	/* extract the directory dentry from the cwd */
f7ad3c6b	576	get_fs_pwd(current->fs, &path);
9ae326a6	577
ce40fa78	578	if (!d_can_lookup(path.dentry))
9ae326a6 DH	579	goto notdir;
	580
	581	cachefiles_begin_secure(cache, &saved_cred);
542ce7a9	582	ret = cachefiles_cull(cache, path.dentry, args);
9ae326a6 DH	583	cachefiles_end_secure(cache, saved_cred);
9ae326a6 DH	584
542ce7a9	585	path_put(&path);
9ae326a6 DH	586	_leave(" = %d", ret);
	587	return ret;
	588
	589	notdir:
542ce7a9	590	path_put(&path);
6ff66ac7	591	pr_err("cull command requires dirfd to be a directory\n");
9ae326a6 DH	592	return -ENOTDIR;
	593
	594	inval:
6ff66ac7	595	pr_err("cull command requires dirfd and filename\n");
9ae326a6 DH	596	return -EINVAL;
	597	}
	598
	599	/*
	600	* set debugging mode
	601	* - command: "debug <mask>"
	602	*/
	603	static int cachefiles_daemon_debug(struct cachefiles_cache cache, char args)
	604	{
	605	unsigned long mask;
	606
	607	_enter(",%s", args);
	608
	609	mask = simple_strtoul(args, &args, 0);
	610	if (args[0] != '\0')
	611	goto inval;
	612
	613	cachefiles_debug = mask;
	614	_leave(" = 0");
	615	return 0;
	616
	617	inval:
6ff66ac7	618	pr_err("debug command requires mask\n");
9ae326a6 DH	619	return -EINVAL;
	620	}
	621
	622	/*
	623	* find out whether an object in the current working directory is in use or not
	624	* - command: "inuse <name>"
	625	*/
	626	static int cachefiles_daemon_inuse(struct cachefiles_cache cache, char args)
	627	{
542ce7a9	628	struct path path;
9ae326a6 DH	629	const struct cred *saved_cred;
	630	int ret;
	631
	632	//_enter(",%s", args);
	633
	634	if (strchr(args, '/'))
	635	goto inval;
	636
	637	if (!test_bit(CACHEFILES_READY, &cache->flags)) {
6ff66ac7	638	pr_err("inuse applied to unready cache\n");
9ae326a6 DH	639	return -EIO;
	640	}
	641
	642	if (test_bit(CACHEFILES_DEAD, &cache->flags)) {
6ff66ac7	643	pr_err("inuse applied to dead cache\n");
9ae326a6 DH	644	return -EIO;
	645	}
	646
	647	/* extract the directory dentry from the cwd */
f7ad3c6b	648	get_fs_pwd(current->fs, &path);
9ae326a6	649
ce40fa78	650	if (!d_can_lookup(path.dentry))
9ae326a6 DH	651	goto notdir;
	652
	653	cachefiles_begin_secure(cache, &saved_cred);
542ce7a9	654	ret = cachefiles_check_in_use(cache, path.dentry, args);
9ae326a6 DH	655	cachefiles_end_secure(cache, saved_cred);
9ae326a6 DH	656
542ce7a9	657	path_put(&path);
9ae326a6 DH	658	//_leave(" = %d", ret);
	659	return ret;
	660
	661	notdir:
542ce7a9	662	path_put(&path);
6ff66ac7	663	pr_err("inuse command requires dirfd to be a directory\n");
9ae326a6 DH	664	return -ENOTDIR;
	665
	666	inval:
6ff66ac7	667	pr_err("inuse command requires dirfd and filename\n");
9ae326a6 DH	668	return -EINVAL;
	669	}
	670
	671	/*
	672	* see if we have space for a number of pages and/or a number of files in the
	673	* cache
	674	*/
	675	int cachefiles_has_space(struct cachefiles_cache *cache,
	676	unsigned fnr, unsigned bnr)
	677	{
	678	struct kstatfs stats;
ebabe9a9 CH	679	struct path path = {
	680	.mnt = cache->mnt,
	681	.dentry = cache->mnt->mnt_root,
	682	};
9ae326a6 DH	683	int ret;
	684
	685	//_enter("{%llu,%llu,%llu,%llu,%llu,%llu},%u,%u",
	686	// (unsigned long long) cache->frun,
	687	// (unsigned long long) cache->fcull,
	688	// (unsigned long long) cache->fstop,
	689	// (unsigned long long) cache->brun,
	690	// (unsigned long long) cache->bcull,
	691	// (unsigned long long) cache->bstop,
	692	// fnr, bnr);
	693
	694	/* find out how many pages of blockdev are available */
	695	memset(&stats, 0, sizeof(stats));
	696
ebabe9a9	697	ret = vfs_statfs(&path, &stats);
9ae326a6 DH	698	if (ret < 0) {
	699	if (ret == -EIO)
	700	cachefiles_io_error(cache, "statfs failed");
	701	_leave(" = %d", ret);
	702	return ret;
	703	}
	704
	705	stats.f_bavail >>= cache->bshift;
	706
	707	//_debug("avail %llu,%llu",
	708	// (unsigned long long) stats.f_ffree,
	709	// (unsigned long long) stats.f_bavail);
	710
	711	/* see if there is sufficient space */
	712	if (stats.f_ffree > fnr)
	713	stats.f_ffree -= fnr;
	714	else
	715	stats.f_ffree = 0;
	716
	717	if (stats.f_bavail > bnr)
	718	stats.f_bavail -= bnr;
	719	else
	720	stats.f_bavail = 0;
	721
	722	ret = -ENOBUFS;
	723	if (stats.f_ffree < cache->fstop \|\|
	724	stats.f_bavail < cache->bstop)
	725	goto begin_cull;
	726
	727	ret = 0;
	728	if (stats.f_ffree < cache->fcull \|\|
	729	stats.f_bavail < cache->bcull)
	730	goto begin_cull;
	731
	732	if (test_bit(CACHEFILES_CULLING, &cache->flags) &&
	733	stats.f_ffree >= cache->frun &&
	734	stats.f_bavail >= cache->brun &&
	735	test_and_clear_bit(CACHEFILES_CULLING, &cache->flags)
	736	) {
	737	_debug("cease culling");
	738	cachefiles_state_changed(cache);
	739	}
	740
	741	//_leave(" = 0");
	742	return 0;
	743
	744	begin_cull:
	745	if (!test_and_set_bit(CACHEFILES_CULLING, &cache->flags)) {
	746	_debug("### CULL CACHE ###");
	747	cachefiles_state_changed(cache);
	748	}
	749
	750	_leave(" = %d", ret);
	751	return ret;
	752	}