[deliverable/linux.git] / arch / s390 / appldata / appldata_base.c

/*
 * arch/s390/appldata/appldata_base.c
 *
 * Base infrastructure for Linux-z/VM Monitor Stream, Stage 1.
 * Exports appldata_register_ops() and appldata_unregister_ops() for the
 * data gathering modules.
 *
 * Copyright IBM Corp. 2003, 2009
 *
 * Author: Gerald Schaefer <gerald.schaefer@de.ibm.com>
 */

#define KMSG_COMPONENT	"appldata"
#define pr_fmt(fmt) KMSG_COMPONENT ": " fmt

#include <linux/module.h>
#include <linux/init.h>
#include <linux/slab.h>
#include <linux/errno.h>
#include <linux/interrupt.h>
#include <linux/proc_fs.h>
#include <linux/mm.h>
#include <linux/swap.h>
#include <linux/pagemap.h>
#include <linux/sysctl.h>
#include <linux/notifier.h>
#include <linux/cpu.h>
#include <linux/workqueue.h>
#include <linux/suspend.h>
#include <linux/platform_device.h>
#include <asm/appldata.h>
#include <asm/timer.h>
#include <asm/uaccess.h>
#include <asm/io.h>
#include <asm/smp.h>

#include "appldata.h"


#define APPLDATA_CPU_INTERVAL	10000		/* default (CPU) time for
						   sampling interval in
						   milliseconds */

#define TOD_MICRO	0x01000			/* nr. of TOD clock units
						   for 1 microsecond */

static struct platform_device *appldata_pdev;

/*
 * /proc entries (sysctl)
 */
static const char appldata_proc_name[APPLDATA_PROC_NAME_LENGTH] = "appldata";
static int appldata_timer_handler(ctl_table *ctl, int write,
				  void __user *buffer, size_t *lenp, loff_t *ppos);
static int appldata_interval_handler(ctl_table *ctl, int write,
					 void __user *buffer,
					 size_t *lenp, loff_t *ppos);

static struct ctl_table_header *appldata_sysctl_header;
static struct ctl_table appldata_table[] = {
	{
		.procname	= "timer",
		.mode		= S_IRUGO | S_IWUSR,
		.proc_handler	= appldata_timer_handler,
	},
	{
		.procname	= "interval",
		.mode		= S_IRUGO | S_IWUSR,
		.proc_handler	= appldata_interval_handler,
	},
	{ },
};

static struct ctl_table appldata_dir_table[] = {
	{
		.procname	= appldata_proc_name,
		.maxlen		= 0,
		.mode		= S_IRUGO | S_IXUGO,
		.child		= appldata_table,
	},
	{ },
};

/*
 * Timer
 */
static DEFINE_PER_CPU(struct vtimer_list, appldata_timer);
static atomic_t appldata_expire_count = ATOMIC_INIT(0);

static DEFINE_SPINLOCK(appldata_timer_lock);
static int appldata_interval = APPLDATA_CPU_INTERVAL;
static int appldata_timer_active;
static int appldata_timer_suspended = 0;

/*
 * Work queue
 */
static struct workqueue_struct *appldata_wq;
static void appldata_work_fn(struct work_struct *work);
static DECLARE_WORK(appldata_work, appldata_work_fn);


/*
 * Ops list
 */
static DEFINE_MUTEX(appldata_ops_mutex);
static LIST_HEAD(appldata_ops_list);


/*************************** timer, work, DIAG *******************************/
/*
 * appldata_timer_function()
 *
 * schedule work and reschedule timer
 */
static void appldata_timer_function(unsigned long data)
{
	if (atomic_dec_and_test(&appldata_expire_count)) {
		atomic_set(&appldata_expire_count, num_online_cpus());
		queue_work(appldata_wq, (struct work_struct *) data);
	}
}

/*
 * appldata_work_fn()
 *
 * call data gathering function for each (active) module
 */
static void appldata_work_fn(struct work_struct *work)
{
	struct list_head *lh;
	struct appldata_ops *ops;

	get_online_cpus();
	mutex_lock(&appldata_ops_mutex);
	list_for_each(lh, &appldata_ops_list) {
		ops = list_entry(lh, struct appldata_ops, list);
		if (ops->active == 1) {
			ops->callback(ops->data);
		}
	}
	mutex_unlock(&appldata_ops_mutex);
	put_online_cpus();
}

/*
 * appldata_diag()
 *
 * prepare parameter list, issue DIAG 0xDC
 */
int appldata_diag(char record_nr, u16 function, unsigned long buffer,
			u16 length, char *mod_lvl)
{
	struct appldata_product_id id = {
		.prod_nr    = {0xD3, 0xC9, 0xD5, 0xE4,
			       0xE7, 0xD2, 0xD9},	/* "LINUXKR" */
		.prod_fn    = 0xD5D3,			/* "NL" */
		.version_nr = 0xF2F6,			/* "26" */
		.release_nr = 0xF0F1,			/* "01" */
	};

	id.record_nr = record_nr;
	id.mod_lvl = (mod_lvl[0]) << 8 | mod_lvl[1];
	return appldata_asm(&id, function, (void *) buffer, length);
}
/************************ timer, work, DIAG <END> ****************************/


/****************************** /proc stuff **********************************/

/*
 * appldata_mod_vtimer_wrap()
 *
 * wrapper function for mod_virt_timer(), because smp_call_function_single()
 * accepts only one parameter.
 */
static void __appldata_mod_vtimer_wrap(void *p) {
	struct {
		struct vtimer_list *timer;
		u64    expires;
	} *args = p;
	mod_virt_timer_periodic(args->timer, args->expires);
}

#define APPLDATA_ADD_TIMER	0
#define APPLDATA_DEL_TIMER	1
#define APPLDATA_MOD_TIMER	2

/*
 * __appldata_vtimer_setup()
 *
 * Add, delete or modify virtual timers on all online cpus.
 * The caller needs to get the appldata_timer_lock spinlock.
 */
static void
__appldata_vtimer_setup(int cmd)
{
	u64 per_cpu_interval;
	int i;

	switch (cmd) {
	case APPLDATA_ADD_TIMER:
		if (appldata_timer_active)
			break;
		per_cpu_interval = (u64) (appldata_interval*1000 /
					  num_online_cpus()) * TOD_MICRO;
		for_each_online_cpu(i) {
			per_cpu(appldata_timer, i).expires = per_cpu_interval;
			smp_call_function_single(i, add_virt_timer_periodic,
						 &per_cpu(appldata_timer, i),
						 1);
		}
		appldata_timer_active = 1;
		break;
	case APPLDATA_DEL_TIMER:
		for_each_online_cpu(i)
			del_virt_timer(&per_cpu(appldata_timer, i));
		if (!appldata_timer_active)
			break;
		appldata_timer_active = 0;
		atomic_set(&appldata_expire_count, num_online_cpus());
		break;
	case APPLDATA_MOD_TIMER:
		per_cpu_interval = (u64) (appldata_interval*1000 /
					  num_online_cpus()) * TOD_MICRO;
		if (!appldata_timer_active)
			break;
		for_each_online_cpu(i) {
			struct {
				struct vtimer_list *timer;
				u64    expires;
			} args;
			args.timer = &per_cpu(appldata_timer, i);
			args.expires = per_cpu_interval;
			smp_call_function_single(i, __appldata_mod_vtimer_wrap,
						 &args, 1);
		}
	}
}

/*
 * appldata_timer_handler()
 *
 * Start/Stop timer, show status of timer (0 = not active, 1 = active)
 */
static int
appldata_timer_handler(ctl_table *ctl, int write,
			   void __user *buffer, size_t *lenp, loff_t *ppos)
{
	int len;
	char buf[2];

	if (!*lenp || *ppos) {
		*lenp = 0;
		return 0;
	}
	if (!write) {
		len = sprintf(buf, appldata_timer_active ? "1\n" : "0\n");
		if (len > *lenp)
			len = *lenp;
		if (copy_to_user(buffer, buf, len))
			return -EFAULT;
		goto out;
	}
	len = *lenp;
	if (copy_from_user(buf, buffer, len > sizeof(buf) ? sizeof(buf) : len))
		return -EFAULT;
	get_online_cpus();
	spin_lock(&appldata_timer_lock);
	if (buf[0] == '1')
		__appldata_vtimer_setup(APPLDATA_ADD_TIMER);
	else if (buf[0] == '0')
		__appldata_vtimer_setup(APPLDATA_DEL_TIMER);
	spin_unlock(&appldata_timer_lock);
	put_online_cpus();
out:
	*lenp = len;
	*ppos += len;
	return 0;
}

/*
 * appldata_interval_handler()
 *
 * Set (CPU) timer interval for collection of data (in milliseconds), show
 * current timer interval.
 */
static int
appldata_interval_handler(ctl_table *ctl, int write,
			   void __user *buffer, size_t *lenp, loff_t *ppos)
{
	int len, interval;
	char buf[16];

	if (!*lenp || *ppos) {
		*lenp = 0;
		return 0;
	}
	if (!write) {
		len = sprintf(buf, "%i\n", appldata_interval);
		if (len > *lenp)
			len = *lenp;
		if (copy_to_user(buffer, buf, len))
			return -EFAULT;
		goto out;
	}
	len = *lenp;
	if (copy_from_user(buf, buffer, len > sizeof(buf) ? sizeof(buf) : len)) {
		return -EFAULT;
	}
	interval = 0;
	sscanf(buf, "%i", &interval);
	if (interval <= 0)
		return -EINVAL;

	get_online_cpus();
	spin_lock(&appldata_timer_lock);
	appldata_interval = interval;
	__appldata_vtimer_setup(APPLDATA_MOD_TIMER);
	spin_unlock(&appldata_timer_lock);
	put_online_cpus();
out:
	*lenp = len;
	*ppos += len;
	return 0;
}

/*
 * appldata_generic_handler()
 *
 * Generic start/stop monitoring and DIAG, show status of
 * monitoring (0 = not in process, 1 = in process)
 */
static int
appldata_generic_handler(ctl_table *ctl, int write,
			   void __user *buffer, size_t *lenp, loff_t *ppos)
{
	struct appldata_ops *ops = NULL, *tmp_ops;
	int rc, len, found;
	char buf[2];
	struct list_head *lh;

	found = 0;
	mutex_lock(&appldata_ops_mutex);
	list_for_each(lh, &appldata_ops_list) {
		tmp_ops = list_entry(lh, struct appldata_ops, list);
		if (&tmp_ops->ctl_table[2] == ctl) {
			found = 1;
		}
	}
	if (!found) {
		mutex_unlock(&appldata_ops_mutex);
		return -ENODEV;
	}
	ops = ctl->data;
	if (!try_module_get(ops->owner)) {	// protect this function
		mutex_unlock(&appldata_ops_mutex);
		return -ENODEV;
	}
	mutex_unlock(&appldata_ops_mutex);

	if (!*lenp || *ppos) {
		*lenp = 0;
		module_put(ops->owner);
		return 0;
	}
	if (!write) {
		len = sprintf(buf, ops->active ? "1\n" : "0\n");
		if (len > *lenp)
			len = *lenp;
		if (copy_to_user(buffer, buf, len)) {
			module_put(ops->owner);
			return -EFAULT;
		}
		goto out;
	}
	len = *lenp;
	if (copy_from_user(buf, buffer,
			   len > sizeof(buf) ? sizeof(buf) : len)) {
		module_put(ops->owner);
		return -EFAULT;
	}

	mutex_lock(&appldata_ops_mutex);
	if ((buf[0] == '1') && (ops->active == 0)) {
		// protect work queue callback
		if (!try_module_get(ops->owner)) {
			mutex_unlock(&appldata_ops_mutex);
			module_put(ops->owner);
			return -ENODEV;
		}
		ops->callback(ops->data);	// init record
		rc = appldata_diag(ops->record_nr,
					APPLDATA_START_INTERVAL_REC,
					(unsigned long) ops->data, ops->size,
					ops->mod_lvl);
		if (rc != 0) {
			pr_err("Starting the data collection for %s "
			       "failed with rc=%d\n", ops->name, rc);
			module_put(ops->owner);
		} else
			ops->active = 1;
	} else if ((buf[0] == '0') && (ops->active == 1)) {
		ops->active = 0;
		rc = appldata_diag(ops->record_nr, APPLDATA_STOP_REC,
				(unsigned long) ops->data, ops->size,
				ops->mod_lvl);
		if (rc != 0)
			pr_err("Stopping the data collection for %s "
			       "failed with rc=%d\n", ops->name, rc);
		module_put(ops->owner);
	}
	mutex_unlock(&appldata_ops_mutex);
out:
	*lenp = len;
	*ppos += len;
	module_put(ops->owner);
	return 0;
}

/*************************** /proc stuff <END> *******************************/


/************************* module-ops management *****************************/
/*
 * appldata_register_ops()
 *
 * update ops list, register /proc/sys entries
 */
int appldata_register_ops(struct appldata_ops *ops)
{
	if (ops->size > APPLDATA_MAX_REC_SIZE)
		return -EINVAL;

	ops->ctl_table = kzalloc(4 * sizeof(struct ctl_table), GFP_KERNEL);
	if (!ops->ctl_table)
		return -ENOMEM;

	mutex_lock(&appldata_ops_mutex);
	list_add(&ops->list, &appldata_ops_list);
	mutex_unlock(&appldata_ops_mutex);

	ops->ctl_table[0].procname = appldata_proc_name;
	ops->ctl_table[0].maxlen   = 0;
	ops->ctl_table[0].mode     = S_IRUGO | S_IXUGO;
	ops->ctl_table[0].child    = &ops->ctl_table[2];

	ops->ctl_table[2].procname = ops->name;
	ops->ctl_table[2].mode     = S_IRUGO | S_IWUSR;
	ops->ctl_table[2].proc_handler = appldata_generic_handler;
	ops->ctl_table[2].data = ops;

	ops->sysctl_header = register_sysctl_table(ops->ctl_table);
	if (!ops->sysctl_header)
		goto out;
	return 0;
out:
	mutex_lock(&appldata_ops_mutex);
	list_del(&ops->list);
	mutex_unlock(&appldata_ops_mutex);
	kfree(ops->ctl_table);
	return -ENOMEM;
}

/*
 * appldata_unregister_ops()
 *
 * update ops list, unregister /proc entries, stop DIAG if necessary
 */
void appldata_unregister_ops(struct appldata_ops *ops)
{
	mutex_lock(&appldata_ops_mutex);
	list_del(&ops->list);
	mutex_unlock(&appldata_ops_mutex);
	unregister_sysctl_table(ops->sysctl_header);
	kfree(ops->ctl_table);
}
/********************** module-ops management <END> **************************/


/**************************** suspend / resume *******************************/
static int appldata_freeze(struct device *dev)
{
	struct appldata_ops *ops;
	int rc;
	struct list_head *lh;

	get_online_cpus();
	spin_lock(&appldata_timer_lock);
	if (appldata_timer_active) {
		__appldata_vtimer_setup(APPLDATA_DEL_TIMER);
		appldata_timer_suspended = 1;
	}
	spin_unlock(&appldata_timer_lock);
	put_online_cpus();

	mutex_lock(&appldata_ops_mutex);
	list_for_each(lh, &appldata_ops_list) {
		ops = list_entry(lh, struct appldata_ops, list);
		if (ops->active == 1) {
			rc = appldata_diag(ops->record_nr, APPLDATA_STOP_REC,
					(unsigned long) ops->data, ops->size,
					ops->mod_lvl);
			if (rc != 0)
				pr_err("Stopping the data collection for %s "
				       "failed with rc=%d\n", ops->name, rc);
		}
	}
	mutex_unlock(&appldata_ops_mutex);
	return 0;
}

static int appldata_restore(struct device *dev)
{
	struct appldata_ops *ops;
	int rc;
	struct list_head *lh;

	get_online_cpus();
	spin_lock(&appldata_timer_lock);
	if (appldata_timer_suspended) {
		__appldata_vtimer_setup(APPLDATA_ADD_TIMER);
		appldata_timer_suspended = 0;
	}
	spin_unlock(&appldata_timer_lock);
	put_online_cpus();

	mutex_lock(&appldata_ops_mutex);
	list_for_each(lh, &appldata_ops_list) {
		ops = list_entry(lh, struct appldata_ops, list);
		if (ops->active == 1) {
			ops->callback(ops->data);	// init record
			rc = appldata_diag(ops->record_nr,
					APPLDATA_START_INTERVAL_REC,
					(unsigned long) ops->data, ops->size,
					ops->mod_lvl);
			if (rc != 0) {
				pr_err("Starting the data collection for %s "
				       "failed with rc=%d\n", ops->name, rc);
			}
		}
	}
	mutex_unlock(&appldata_ops_mutex);
	return 0;
}

static int appldata_thaw(struct device *dev)
{
	return appldata_restore(dev);
}

static const struct dev_pm_ops appldata_pm_ops = {
	.freeze		= appldata_freeze,
	.thaw		= appldata_thaw,
	.restore	= appldata_restore,
};

static struct platform_driver appldata_pdrv = {
	.driver = {
		.name	= "appldata",
		.owner	= THIS_MODULE,
		.pm	= &appldata_pm_ops,
	},
};
/************************* suspend / resume <END> ****************************/


/******************************* init / exit *********************************/

static void __cpuinit appldata_online_cpu(int cpu)
{
	init_virt_timer(&per_cpu(appldata_timer, cpu));
	per_cpu(appldata_timer, cpu).function = appldata_timer_function;
	per_cpu(appldata_timer, cpu).data = (unsigned long)
		&appldata_work;
	atomic_inc(&appldata_expire_count);
	spin_lock(&appldata_timer_lock);
	__appldata_vtimer_setup(APPLDATA_MOD_TIMER);
	spin_unlock(&appldata_timer_lock);
}

static void __cpuinit appldata_offline_cpu(int cpu)
{
	del_virt_timer(&per_cpu(appldata_timer, cpu));
	if (atomic_dec_and_test(&appldata_expire_count)) {
		atomic_set(&appldata_expire_count, num_online_cpus());
		queue_work(appldata_wq, &appldata_work);
	}
	spin_lock(&appldata_timer_lock);
	__appldata_vtimer_setup(APPLDATA_MOD_TIMER);
	spin_unlock(&appldata_timer_lock);
}

static int __cpuinit appldata_cpu_notify(struct notifier_block *self,
					 unsigned long action,
					 void *hcpu)
{
	switch (action) {
	case CPU_ONLINE:
	case CPU_ONLINE_FROZEN:
		appldata_online_cpu((long) hcpu);
		break;
	case CPU_DEAD:
	case CPU_DEAD_FROZEN:
		appldata_offline_cpu((long) hcpu);
		break;
	default:
		break;
	}
	return NOTIFY_OK;
}

static struct notifier_block __cpuinitdata appldata_nb = {
	.notifier_call = appldata_cpu_notify,
};

/*
 * appldata_init()
 *
 * init timer, register /proc entries
 */
static int __init appldata_init(void)
{
	int i, rc;

	rc = platform_driver_register(&appldata_pdrv);
	if (rc)
		return rc;

	appldata_pdev = platform_device_register_simple("appldata", -1, NULL,
							0);
	if (IS_ERR(appldata_pdev)) {
		rc = PTR_ERR(appldata_pdev);
		goto out_driver;
	}
	appldata_wq = create_singlethread_workqueue("appldata");
	if (!appldata_wq) {
		rc = -ENOMEM;
		goto out_device;
	}

	get_online_cpus();
	for_each_online_cpu(i)
		appldata_online_cpu(i);
	put_online_cpus();

	/* Register cpu hotplug notifier */
	register_hotcpu_notifier(&appldata_nb);

	appldata_sysctl_header = register_sysctl_table(appldata_dir_table);
	return 0;

out_device:
	platform_device_unregister(appldata_pdev);
out_driver:
	platform_driver_unregister(&appldata_pdrv);
	return rc;
}

__initcall(appldata_init);

/**************************** init / exit <END> ******************************/

EXPORT_SYMBOL_GPL(appldata_register_ops);
EXPORT_SYMBOL_GPL(appldata_unregister_ops);
EXPORT_SYMBOL_GPL(appldata_diag);

#ifdef CONFIG_SWAP
EXPORT_SYMBOL_GPL(si_swapinfo);
#endif
EXPORT_SYMBOL_GPL(nr_threads);
EXPORT_SYMBOL_GPL(nr_running);
EXPORT_SYMBOL_GPL(nr_iowait);
Commit	Line	Data
1da177e4 LT	1	/*
	2	* arch/s390/appldata/appldata_base.c
	3	*
	4	* Base infrastructure for Linux-z/VM Monitor Stream, Stage 1.
	5	* Exports appldata_register_ops() and appldata_unregister_ops() for the
	6	* data gathering modules.
	7	*
524dbcda	8	* Copyright IBM Corp. 2003, 2009
1da177e4	9	*
5b5dd21a	10	* Author: Gerald Schaefer <gerald.schaefer@de.ibm.com>
1da177e4 LT	11	*/
1da177e4 LT	12
e7534b0e GS	13	#define KMSG_COMPONENT "appldata"
	14	#define pr_fmt(fmt) KMSG_COMPONENT ": " fmt
	15
1da177e4 LT	16	#include <linux/module.h>
	17	#include <linux/init.h>
	18	#include <linux/slab.h>
	19	#include <linux/errno.h>
1da177e4 LT	20	#include <linux/interrupt.h>
1da177e4 LT	21	#include <linux/proc_fs.h>
2dcea57a	22	#include <linux/mm.h>
1da177e4 LT	23	#include <linux/swap.h>
	24	#include <linux/pagemap.h>
	25	#include <linux/sysctl.h>
1da177e4 LT	26	#include <linux/notifier.h>
1da177e4 LT	27	#include <linux/cpu.h>
f26d583e	28	#include <linux/workqueue.h>
524dbcda GS	29	#include <linux/suspend.h>
524dbcda GS	30	#include <linux/platform_device.h>
1f38d613 GS	31	#include <asm/appldata.h>
	32	#include <asm/timer.h>
	33	#include <asm/uaccess.h>
	34	#include <asm/io.h>
	35	#include <asm/smp.h>
1da177e4 LT	36
	37	#include "appldata.h"
	38
	39
1da177e4 LT	40	#define APPLDATA_CPU_INTERVAL 10000 /* default (CPU) time for
	41	sampling interval in
	42	milliseconds */
	43
	44	#define TOD_MICRO 0x01000 /* nr. of TOD clock units
	45	for 1 microsecond */
524dbcda GS	46
	47	static struct platform_device *appldata_pdev;
	48
1da177e4 LT	49	/*
	50	* /proc entries (sysctl)
	51	*/
	52	static const char appldata_proc_name[APPLDATA_PROC_NAME_LENGTH] = "appldata";
8d65af78	53	static int appldata_timer_handler(ctl_table *ctl, int write,
1da177e4 LT	54	void __user buffer, size_t lenp, loff_t *ppos);
1da177e4 LT	55	static int appldata_interval_handler(ctl_table *ctl, int write,
1da177e4 LT	56	void __user *buffer,
	57	size_t lenp, loff_t ppos);
	58
	59	static struct ctl_table_header *appldata_sysctl_header;
	60	static struct ctl_table appldata_table[] = {
	61	{
1da177e4 LT	62	.procname = "timer",
1da177e4 LT	63	.mode = S_IRUGO \| S_IWUSR,
6d456111	64	.proc_handler = appldata_timer_handler,
1da177e4 LT	65	},
1da177e4 LT	66	{
1da177e4 LT	67	.procname = "interval",
1da177e4 LT	68	.mode = S_IRUGO \| S_IWUSR,
6d456111	69	.proc_handler = appldata_interval_handler,
1da177e4	70	},
37e3a6ac	71	{ },
1da177e4 LT	72	};
	73
	74	static struct ctl_table appldata_dir_table[] = {
	75	{
1da177e4 LT	76	.procname = appldata_proc_name,
	77	.maxlen = 0,
	78	.mode = S_IRUGO \| S_IXUGO,
	79	.child = appldata_table,
	80	},
37e3a6ac	81	{ },
1da177e4 LT	82	};
	83
	84	/*
	85	* Timer
	86	*/
2b67fc46	87	static DEFINE_PER_CPU(struct vtimer_list, appldata_timer);
1da177e4 LT	88	static atomic_t appldata_expire_count = ATOMIC_INIT(0);
	89
	90	static DEFINE_SPINLOCK(appldata_timer_lock);
	91	static int appldata_interval = APPLDATA_CPU_INTERVAL;
	92	static int appldata_timer_active;
524dbcda	93	static int appldata_timer_suspended = 0;
1da177e4 LT	94
1da177e4 LT	95	/*
f26d583e	96	* Work queue
1da177e4	97	*/
f26d583e	98	static struct workqueue_struct *appldata_wq;
6d5aefb8 DH	99	static void appldata_work_fn(struct work_struct *work);
6d5aefb8 DH	100	static DECLARE_WORK(appldata_work, appldata_work_fn);
f26d583e	101
1da177e4 LT	102
	103	/*
	104	* Ops list
	105	*/
b1ad171e	106	static DEFINE_MUTEX(appldata_ops_mutex);
1da177e4 LT	107	static LIST_HEAD(appldata_ops_list);
	108
	109
f26d583e	110	/************************* timer, work, DIAG *****************************/
1da177e4 LT	111	/*
	112	* appldata_timer_function()
	113	*
f26d583e	114	* schedule work and reschedule timer
1da177e4	115	*/
9d0a57cb	116	static void appldata_timer_function(unsigned long data)
1da177e4	117	{
1da177e4 LT	118	if (atomic_dec_and_test(&appldata_expire_count)) {
1da177e4 LT	119	atomic_set(&appldata_expire_count, num_online_cpus());
f26d583e	120	queue_work(appldata_wq, (struct work_struct *) data);
1da177e4 LT	121	}
	122	}
	123
	124	/*
f26d583e	125	* appldata_work_fn()
1da177e4 LT	126	*
	127	* call data gathering function for each (active) module
	128	*/
6d5aefb8	129	static void appldata_work_fn(struct work_struct *work)
1da177e4 LT	130	{
	131	struct list_head *lh;
	132	struct appldata_ops *ops;
1da177e4	133
1760537b	134	get_online_cpus();
b1ad171e	135	mutex_lock(&appldata_ops_mutex);
1da177e4 LT	136	list_for_each(lh, &appldata_ops_list) {
1da177e4 LT	137	ops = list_entry(lh, struct appldata_ops, list);
1da177e4 LT	138	if (ops->active == 1) {
	139	ops->callback(ops->data);
	140	}
	141	}
b1ad171e	142	mutex_unlock(&appldata_ops_mutex);
1760537b	143	put_online_cpus();
1da177e4 LT	144	}
	145
	146	/*
	147	* appldata_diag()
	148	*
	149	* prepare parameter list, issue DIAG 0xDC
	150	*/
5b5dd21a GS	151	int appldata_diag(char record_nr, u16 function, unsigned long buffer,
5b5dd21a GS	152	u16 length, char *mod_lvl)
1da177e4	153	{
1f38d613	154	struct appldata_product_id id = {
1da177e4	155	.prod_nr = {0xD3, 0xC9, 0xD5, 0xE4,
1f38d613 GS	156	0xE7, 0xD2, 0xD9}, /* "LINUXKR" */
1f38d613 GS	157	.prod_fn = 0xD5D3, /* "NL" */
1f38d613 GS	158	.version_nr = 0xF2F6, /* "26" */
1f38d613 GS	159	.release_nr = 0xF0F1, /* "01" */
1da177e4 LT	160	};
1da177e4 LT	161
925afbd6 GS	162	id.record_nr = record_nr;
925afbd6 GS	163	id.mod_lvl = (mod_lvl[0]) << 8 \| mod_lvl[1];
1f38d613	164	return appldata_asm(&id, function, (void *) buffer, length);
1da177e4	165	}
f26d583e	166	/********************** timer, work, DIAG <END> **************************/
1da177e4 LT	167
	168
	169	/**************************** /proc stuff ********************************/
	170
	171	/*
	172	* appldata_mod_vtimer_wrap()
	173	*
3bb447fc	174	* wrapper function for mod_virt_timer(), because smp_call_function_single()
1da177e4 LT	175	* accepts only one parameter.
	176	*/
	177	static void __appldata_mod_vtimer_wrap(void *p) {
	178	struct {
	179	struct vtimer_list *timer;
	180	u64 expires;
	181	} *args = p;
43ae8a1b	182	mod_virt_timer_periodic(args->timer, args->expires);
1da177e4 LT	183	}
	184
	185	#define APPLDATA_ADD_TIMER 0
	186	#define APPLDATA_DEL_TIMER 1
	187	#define APPLDATA_MOD_TIMER 2
	188
	189	/*
	190	* __appldata_vtimer_setup()
	191	*
	192	* Add, delete or modify virtual timers on all online cpus.
	193	* The caller needs to get the appldata_timer_lock spinlock.
	194	*/
	195	static void
	196	__appldata_vtimer_setup(int cmd)
	197	{
	198	u64 per_cpu_interval;
	199	int i;
	200
	201	switch (cmd) {
	202	case APPLDATA_ADD_TIMER:
	203	if (appldata_timer_active)
	204	break;
	205	per_cpu_interval = (u64) (appldata_interval*1000 /
	206	num_online_cpus()) * TOD_MICRO;
	207	for_each_online_cpu(i) {
	208	per_cpu(appldata_timer, i).expires = per_cpu_interval;
3bb447fc HC	209	smp_call_function_single(i, add_virt_timer_periodic,
3bb447fc HC	210	&per_cpu(appldata_timer, i),
8691e5a8	211	1);
1da177e4 LT	212	}
1da177e4 LT	213	appldata_timer_active = 1;
1da177e4 LT	214	break;
	215	case APPLDATA_DEL_TIMER:
	216	for_each_online_cpu(i)
	217	del_virt_timer(&per_cpu(appldata_timer, i));
	218	if (!appldata_timer_active)
	219	break;
	220	appldata_timer_active = 0;
	221	atomic_set(&appldata_expire_count, num_online_cpus());
1da177e4 LT	222	break;
	223	case APPLDATA_MOD_TIMER:
	224	per_cpu_interval = (u64) (appldata_interval*1000 /
	225	num_online_cpus()) * TOD_MICRO;
	226	if (!appldata_timer_active)
	227	break;
	228	for_each_online_cpu(i) {
	229	struct {
	230	struct vtimer_list *timer;
	231	u64 expires;
	232	} args;
	233	args.timer = &per_cpu(appldata_timer, i);
	234	args.expires = per_cpu_interval;
3bb447fc	235	smp_call_function_single(i, __appldata_mod_vtimer_wrap,
8691e5a8	236	&args, 1);
1da177e4 LT	237	}
	238	}
	239	}
	240
	241	/*
	242	* appldata_timer_handler()
	243	*
	244	* Start/Stop timer, show status of timer (0 = not active, 1 = active)
	245	*/
	246	static int
8d65af78	247	appldata_timer_handler(ctl_table *ctl, int write,
1da177e4 LT	248	void __user buffer, size_t lenp, loff_t *ppos)
	249	{
	250	int len;
	251	char buf[2];
	252
	253	if (!lenp \|\| ppos) {
	254	*lenp = 0;
	255	return 0;
	256	}
	257	if (!write) {
	258	len = sprintf(buf, appldata_timer_active ? "1\n" : "0\n");
	259	if (len > *lenp)
	260	len = *lenp;
	261	if (copy_to_user(buffer, buf, len))
	262	return -EFAULT;
	263	goto out;
	264	}
	265	len = *lenp;
	266	if (copy_from_user(buf, buffer, len > sizeof(buf) ? sizeof(buf) : len))
	267	return -EFAULT;
1760537b	268	get_online_cpus();
1da177e4 LT	269	spin_lock(&appldata_timer_lock);
	270	if (buf[0] == '1')
	271	__appldata_vtimer_setup(APPLDATA_ADD_TIMER);
	272	else if (buf[0] == '0')
	273	__appldata_vtimer_setup(APPLDATA_DEL_TIMER);
	274	spin_unlock(&appldata_timer_lock);
1760537b	275	put_online_cpus();
1da177e4 LT	276	out:
	277	*lenp = len;
	278	*ppos += len;
	279	return 0;
	280	}
	281
	282	/*
	283	* appldata_interval_handler()
	284	*
	285	* Set (CPU) timer interval for collection of data (in milliseconds), show
	286	* current timer interval.
	287	*/
	288	static int
8d65af78	289	appldata_interval_handler(ctl_table *ctl, int write,
1da177e4 LT	290	void __user buffer, size_t lenp, loff_t *ppos)
	291	{
	292	int len, interval;
	293	char buf[16];
	294
	295	if (!lenp \|\| ppos) {
	296	*lenp = 0;
	297	return 0;
	298	}
	299	if (!write) {
	300	len = sprintf(buf, "%i\n", appldata_interval);
	301	if (len > *lenp)
	302	len = *lenp;
	303	if (copy_to_user(buffer, buf, len))
	304	return -EFAULT;
	305	goto out;
	306	}
	307	len = *lenp;
	308	if (copy_from_user(buf, buffer, len > sizeof(buf) ? sizeof(buf) : len)) {
	309	return -EFAULT;
	310	}
95425f19	311	interval = 0;
1da177e4	312	sscanf(buf, "%i", &interval);
d3ae942d	313	if (interval <= 0)
1da177e4	314	return -EINVAL;
1da177e4	315
1760537b	316	get_online_cpus();
1da177e4 LT	317	spin_lock(&appldata_timer_lock);
	318	appldata_interval = interval;
	319	__appldata_vtimer_setup(APPLDATA_MOD_TIMER);
	320	spin_unlock(&appldata_timer_lock);
1760537b	321	put_online_cpus();
1da177e4 LT	322	out:
	323	*lenp = len;
	324	*ppos += len;
	325	return 0;
	326	}
	327
	328	/*
	329	* appldata_generic_handler()
	330	*
	331	* Generic start/stop monitoring and DIAG, show status of
	332	* monitoring (0 = not in process, 1 = in process)
	333	*/
	334	static int
8d65af78	335	appldata_generic_handler(ctl_table *ctl, int write,
1da177e4 LT	336	void __user buffer, size_t lenp, loff_t *ppos)
	337	{
	338	struct appldata_ops ops = NULL, tmp_ops;
	339	int rc, len, found;
	340	char buf[2];
	341	struct list_head *lh;
	342
	343	found = 0;
b1ad171e	344	mutex_lock(&appldata_ops_mutex);
1da177e4 LT	345	list_for_each(lh, &appldata_ops_list) {
	346	tmp_ops = list_entry(lh, struct appldata_ops, list);
	347	if (&tmp_ops->ctl_table[2] == ctl) {
	348	found = 1;
	349	}
	350	}
	351	if (!found) {
b1ad171e	352	mutex_unlock(&appldata_ops_mutex);
1da177e4 LT	353	return -ENODEV;
	354	}
	355	ops = ctl->data;
	356	if (!try_module_get(ops->owner)) { // protect this function
b1ad171e	357	mutex_unlock(&appldata_ops_mutex);
1da177e4 LT	358	return -ENODEV;
1da177e4 LT	359	}
b1ad171e	360	mutex_unlock(&appldata_ops_mutex);
1da177e4 LT	361
	362	if (!lenp \|\| ppos) {
	363	*lenp = 0;
	364	module_put(ops->owner);
	365	return 0;
	366	}
	367	if (!write) {
	368	len = sprintf(buf, ops->active ? "1\n" : "0\n");
	369	if (len > *lenp)
	370	len = *lenp;
	371	if (copy_to_user(buffer, buf, len)) {
	372	module_put(ops->owner);
	373	return -EFAULT;
	374	}
	375	goto out;
	376	}
	377	len = *lenp;
	378	if (copy_from_user(buf, buffer,
	379	len > sizeof(buf) ? sizeof(buf) : len)) {
	380	module_put(ops->owner);
	381	return -EFAULT;
	382	}
	383
b1ad171e	384	mutex_lock(&appldata_ops_mutex);
1da177e4	385	if ((buf[0] == '1') && (ops->active == 0)) {
f26d583e GS	386	// protect work queue callback
f26d583e GS	387	if (!try_module_get(ops->owner)) {
b1ad171e	388	mutex_unlock(&appldata_ops_mutex);
1da177e4 LT	389	module_put(ops->owner);
	390	return -ENODEV;
	391	}
1da177e4 LT	392	ops->callback(ops->data); // init record
	393	rc = appldata_diag(ops->record_nr,
	394	APPLDATA_START_INTERVAL_REC,
5b5dd21a GS	395	(unsigned long) ops->data, ops->size,
5b5dd21a GS	396	ops->mod_lvl);
1da177e4	397	if (rc != 0) {
e7534b0e GS	398	pr_err("Starting the data collection for %s "
e7534b0e GS	399	"failed with rc=%d\n", ops->name, rc);
1da177e4	400	module_put(ops->owner);
d3ae942d	401	} else
5b5dd21a	402	ops->active = 1;
1da177e4 LT	403	} else if ((buf[0] == '0') && (ops->active == 1)) {
	404	ops->active = 0;
	405	rc = appldata_diag(ops->record_nr, APPLDATA_STOP_REC,
5b5dd21a GS	406	(unsigned long) ops->data, ops->size,
5b5dd21a GS	407	ops->mod_lvl);
d3ae942d	408	if (rc != 0)
e7534b0e GS	409	pr_err("Stopping the data collection for %s "
e7534b0e GS	410	"failed with rc=%d\n", ops->name, rc);
1da177e4 LT	411	module_put(ops->owner);
1da177e4 LT	412	}
b1ad171e	413	mutex_unlock(&appldata_ops_mutex);
1da177e4 LT	414	out:
	415	*lenp = len;
	416	*ppos += len;
	417	module_put(ops->owner);
	418	return 0;
	419	}
	420
	421	/************************* /proc stuff <END> *****************************/
	422
	423
	424	/*********************** module-ops management ***************************/
	425	/*
	426	* appldata_register_ops()
	427	*
	428	* update ops list, register /proc/sys entries
	429	*/
	430	int appldata_register_ops(struct appldata_ops *ops)
	431	{
13f8b7c5	432	if (ops->size > APPLDATA_MAX_REC_SIZE)
37e3a6ac	433	return -EINVAL;
1da177e4	434
37e3a6ac HC	435	ops->ctl_table = kzalloc(4 * sizeof(struct ctl_table), GFP_KERNEL);
37e3a6ac HC	436	if (!ops->ctl_table)
1da177e4	437	return -ENOMEM;
1da177e4	438
b1ad171e	439	mutex_lock(&appldata_ops_mutex);
1da177e4	440	list_add(&ops->list, &appldata_ops_list);
b1ad171e	441	mutex_unlock(&appldata_ops_mutex);
1da177e4	442
1da177e4 LT	443	ops->ctl_table[0].procname = appldata_proc_name;
	444	ops->ctl_table[0].maxlen = 0;
	445	ops->ctl_table[0].mode = S_IRUGO \| S_IXUGO;
	446	ops->ctl_table[0].child = &ops->ctl_table[2];
	447
1da177e4 LT	448	ops->ctl_table[2].procname = ops->name;
	449	ops->ctl_table[2].mode = S_IRUGO \| S_IWUSR;
	450	ops->ctl_table[2].proc_handler = appldata_generic_handler;
	451	ops->ctl_table[2].data = ops;
	452
0b4d4147	453	ops->sysctl_header = register_sysctl_table(ops->ctl_table);
37e3a6ac HC	454	if (!ops->sysctl_header)
37e3a6ac HC	455	goto out;
1da177e4	456	return 0;
37e3a6ac	457	out:
b1ad171e	458	mutex_lock(&appldata_ops_mutex);
37e3a6ac	459	list_del(&ops->list);
b1ad171e	460	mutex_unlock(&appldata_ops_mutex);
37e3a6ac HC	461	kfree(ops->ctl_table);
37e3a6ac HC	462	return -ENOMEM;
1da177e4 LT	463	}
	464
	465	/*
	466	* appldata_unregister_ops()
	467	*
	468	* update ops list, unregister /proc entries, stop DIAG if necessary
	469	*/
	470	void appldata_unregister_ops(struct appldata_ops *ops)
	471	{
b1ad171e	472	mutex_lock(&appldata_ops_mutex);
1da177e4	473	list_del(&ops->list);
b1ad171e	474	mutex_unlock(&appldata_ops_mutex);
330d57fb	475	unregister_sysctl_table(ops->sysctl_header);
37e3a6ac	476	kfree(ops->ctl_table);
1da177e4 LT	477	}
	478	/******************** module-ops management <END> ************************/
	479
	480
524dbcda GS	481	/************************** suspend / resume *****************************/
	482	static int appldata_freeze(struct device *dev)
	483	{
	484	struct appldata_ops *ops;
	485	int rc;
	486	struct list_head *lh;
	487
	488	get_online_cpus();
	489	spin_lock(&appldata_timer_lock);
	490	if (appldata_timer_active) {
	491	__appldata_vtimer_setup(APPLDATA_DEL_TIMER);
	492	appldata_timer_suspended = 1;
	493	}
	494	spin_unlock(&appldata_timer_lock);
	495	put_online_cpus();
	496
	497	mutex_lock(&appldata_ops_mutex);
	498	list_for_each(lh, &appldata_ops_list) {
	499	ops = list_entry(lh, struct appldata_ops, list);
	500	if (ops->active == 1) {
	501	rc = appldata_diag(ops->record_nr, APPLDATA_STOP_REC,
	502	(unsigned long) ops->data, ops->size,
	503	ops->mod_lvl);
	504	if (rc != 0)
	505	pr_err("Stopping the data collection for %s "
	506	"failed with rc=%d\n", ops->name, rc);
	507	}
	508	}
	509	mutex_unlock(&appldata_ops_mutex);
	510	return 0;
	511	}
	512
	513	static int appldata_restore(struct device *dev)
	514	{
	515	struct appldata_ops *ops;
	516	int rc;
	517	struct list_head *lh;
	518
	519	get_online_cpus();
	520	spin_lock(&appldata_timer_lock);
	521	if (appldata_timer_suspended) {
	522	__appldata_vtimer_setup(APPLDATA_ADD_TIMER);
	523	appldata_timer_suspended = 0;
	524	}
	525	spin_unlock(&appldata_timer_lock);
	526	put_online_cpus();
	527
	528	mutex_lock(&appldata_ops_mutex);
	529	list_for_each(lh, &appldata_ops_list) {
	530	ops = list_entry(lh, struct appldata_ops, list);
	531	if (ops->active == 1) {
	532	ops->callback(ops->data); // init record
	533	rc = appldata_diag(ops->record_nr,
	534	APPLDATA_START_INTERVAL_REC,
	535	(unsigned long) ops->data, ops->size,
	536	ops->mod_lvl);
	537	if (rc != 0) {
	538	pr_err("Starting the data collection for %s "
	539	"failed with rc=%d\n", ops->name, rc);
	540	}
	541	}
	542	}
	543	mutex_unlock(&appldata_ops_mutex);
	544	return 0;
545	}
546
547	static int appldata_thaw(struct device *dev)
548	{
549	return appldata_restore(dev);
550	}
551
47145210	552	static const struct dev_pm_ops appldata_pm_ops = {
524dbcda GS	553	.freeze = appldata_freeze,
	554	.thaw = appldata_thaw,
	555	.restore = appldata_restore,
	556	};
	557
	558	static struct platform_driver appldata_pdrv = {
	559	.driver = {
	560	.name = "appldata",
	561	.owner = THIS_MODULE,
	562	.pm = &appldata_pm_ops,
	563	},
	564	};
	565	/*********************** suspend / resume <END> **************************/
	566
	567
1da177e4 LT	568	/***************************** init / exit *******************************/
1da177e4 LT	569
84b36a8e	570	static void __cpuinit appldata_online_cpu(int cpu)
1da177e4 LT	571	{
	572	init_virt_timer(&per_cpu(appldata_timer, cpu));
	573	per_cpu(appldata_timer, cpu).function = appldata_timer_function;
	574	per_cpu(appldata_timer, cpu).data = (unsigned long)
f26d583e	575	&appldata_work;
1da177e4 LT	576	atomic_inc(&appldata_expire_count);
	577	spin_lock(&appldata_timer_lock);
	578	__appldata_vtimer_setup(APPLDATA_MOD_TIMER);
	579	spin_unlock(&appldata_timer_lock);
	580	}
	581
076fc808	582	static void __cpuinit appldata_offline_cpu(int cpu)
1da177e4 LT	583	{
	584	del_virt_timer(&per_cpu(appldata_timer, cpu));
	585	if (atomic_dec_and_test(&appldata_expire_count)) {
	586	atomic_set(&appldata_expire_count, num_online_cpus());
f26d583e	587	queue_work(appldata_wq, &appldata_work);
1da177e4 LT	588	}
	589	spin_lock(&appldata_timer_lock);
	590	__appldata_vtimer_setup(APPLDATA_MOD_TIMER);
	591	spin_unlock(&appldata_timer_lock);
	592	}
	593
11b8bf01 SS	594	static int __cpuinit appldata_cpu_notify(struct notifier_block *self,
	595	unsigned long action,
	596	void *hcpu)
1da177e4 LT	597	{
	598	switch (action) {
	599	case CPU_ONLINE:
8bb78442	600	case CPU_ONLINE_FROZEN:
1da177e4 LT	601	appldata_online_cpu((long) hcpu);
1da177e4 LT	602	break;
1da177e4	603	case CPU_DEAD:
8bb78442	604	case CPU_DEAD_FROZEN:
1da177e4 LT	605	appldata_offline_cpu((long) hcpu);
1da177e4 LT	606	break;
1da177e4 LT	607	default:
	608	break;
	609	}
	610	return NOTIFY_OK;
	611	}
	612
84b36a8e	613	static struct notifier_block __cpuinitdata appldata_nb = {
1da177e4 LT	614	.notifier_call = appldata_cpu_notify,
	615	};
	616
	617	/*
	618	* appldata_init()
	619	*
f26d583e	620	* init timer, register /proc entries
1da177e4 LT	621	*/
	622	static int __init appldata_init(void)
	623	{
524dbcda GS	624	int i, rc;
	625
	626	rc = platform_driver_register(&appldata_pdrv);
	627	if (rc)
	628	return rc;
1da177e4	629
524dbcda GS	630	appldata_pdev = platform_device_register_simple("appldata", -1, NULL,
	631	0);
	632	if (IS_ERR(appldata_pdev)) {
	633	rc = PTR_ERR(appldata_pdev);
	634	goto out_driver;
	635	}
f26d583e	636	appldata_wq = create_singlethread_workqueue("appldata");
524dbcda GS	637	if (!appldata_wq) {
	638	rc = -ENOMEM;
	639	goto out_device;
	640	}
f26d583e	641
1760537b	642	get_online_cpus();
1da177e4 LT	643	for_each_online_cpu(i)
1da177e4 LT	644	appldata_online_cpu(i);
1760537b	645	put_online_cpus();
1da177e4 LT	646
1da177e4 LT	647	/* Register cpu hotplug notifier */
be6b5a35	648	register_hotcpu_notifier(&appldata_nb);
1da177e4	649
0b4d4147	650	appldata_sysctl_header = register_sysctl_table(appldata_dir_table);
1da177e4	651	return 0;
524dbcda GS	652
	653	out_device:
	654	platform_device_unregister(appldata_pdev);
	655	out_driver:
	656	platform_driver_unregister(&appldata_pdrv);
	657	return rc;
1da177e4 LT	658	}
1da177e4 LT	659
076fc808	660	__initcall(appldata_init);
1da177e4	661
1da177e4 LT	662	/************************** init / exit <END> ****************************/
1da177e4 LT	663
1da177e4 LT	664	EXPORT_SYMBOL_GPL(appldata_register_ops);
1da177e4 LT	665	EXPORT_SYMBOL_GPL(appldata_unregister_ops);
5b5dd21a	666	EXPORT_SYMBOL_GPL(appldata_diag);
1da177e4	667
0c3252d5	668	#ifdef CONFIG_SWAP
1da177e4	669	EXPORT_SYMBOL_GPL(si_swapinfo);
0c3252d5	670	#endif
1da177e4	671	EXPORT_SYMBOL_GPL(nr_threads);
1da177e4 LT	672	EXPORT_SYMBOL_GPL(nr_running);
1da177e4 LT	673	EXPORT_SYMBOL_GPL(nr_iowait);