[mirror_ubuntu-bionic-kernel.git] / arch / s390 / appldata / appldata_base.c

// SPDX-License-Identifier: GPL-2.0
/*
 * 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/sched/stat.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/vtimer.h>
#include <linux/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(struct ctl_table *ctl, int write,
				  void __user *buffer, size_t *lenp, loff_t *ppos);
static int appldata_interval_handler(struct 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 struct vtimer_list appldata_timer;

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)
{
	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;

	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);
}

/*
 * 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 **********************************/

#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 timer_interval = (u64) appldata_interval * 1000 * TOD_MICRO;

	switch (cmd) {
	case APPLDATA_ADD_TIMER:
		if (appldata_timer_active)
			break;
		appldata_timer.expires = timer_interval;
		add_virt_timer_periodic(&appldata_timer);
		appldata_timer_active = 1;
		break;
	case APPLDATA_DEL_TIMER:
		del_virt_timer(&appldata_timer);
		if (!appldata_timer_active)
			break;
		appldata_timer_active = 0;
		break;
	case APPLDATA_MOD_TIMER:
		if (!appldata_timer_active)
			break;
		mod_virt_timer_periodic(&appldata_timer, timer_interval);
	}
}

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

	if (!*lenp || *ppos) {
		*lenp = 0;
		return 0;
	}
	if (!write) {
		strncpy(buf, appldata_timer_active ? "1\n" : "0\n",
			ARRAY_SIZE(buf));
		len = strnlen(buf, ARRAY_SIZE(buf));
		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;
	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);
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(struct ctl_table *ctl, int write,
			   void __user *buffer, size_t *lenp, loff_t *ppos)
{
	unsigned int len;
	int 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;

	spin_lock(&appldata_timer_lock);
	appldata_interval = interval;
	__appldata_vtimer_setup(APPLDATA_MOD_TIMER);
	spin_unlock(&appldata_timer_lock);
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(struct ctl_table *ctl, int write,
			   void __user *buffer, size_t *lenp, loff_t *ppos)
{
	struct appldata_ops *ops = NULL, *tmp_ops;
	unsigned int len;
	int rc, 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) {
		strncpy(buf, ops->active ? "1\n" : "0\n", ARRAY_SIZE(buf));
		len = strnlen(buf, ARRAY_SIZE(buf));
		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;

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

	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;

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

	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",
		.pm	= &appldata_pm_ops,
	},
};
/************************* suspend / resume <END> ****************************/


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

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

	init_virt_timer(&appldata_timer);
	appldata_timer.function = appldata_timer_function;
	appldata_timer.data = (unsigned long) &appldata_work;

	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 = alloc_ordered_workqueue("appldata", 0);
	if (!appldata_wq) {
		rc = -ENOMEM;
		goto out_device;
	}

	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
0caa8cdf	1	// SPDX-License-Identifier: GPL-2.0
1da177e4	2	/*
1da177e4 LT	3	* Base infrastructure for Linux-z/VM Monitor Stream, Stage 1.
	4	* Exports appldata_register_ops() and appldata_unregister_ops() for the
	5	* data gathering modules.
	6	*
524dbcda	7	* Copyright IBM Corp. 2003, 2009
1da177e4	8	*
5b5dd21a	9	* Author: Gerald Schaefer <gerald.schaefer@de.ibm.com>
1da177e4 LT	10	*/
1da177e4 LT	11
e7534b0e GS	12	#define KMSG_COMPONENT "appldata"
	13	#define pr_fmt(fmt) KMSG_COMPONENT ": " fmt
	14
1da177e4	15	#include <linux/module.h>
3605df49	16	#include <linux/sched/stat.h>
1da177e4 LT	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	31	#include <asm/appldata.h>
27f6b416	32	#include <asm/vtimer.h>
7c0f6ba6	33	#include <linux/uaccess.h>
1f38d613 GS	34	#include <asm/io.h>
1f38d613 GS	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";
302bfe20	53	static int appldata_timer_handler(struct ctl_table *ctl, int write,
1da177e4	54	void __user buffer, size_t lenp, loff_t *ppos);
302bfe20	55	static int appldata_interval_handler(struct 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	*/
27f6b416	87	static struct vtimer_list appldata_timer;
1da177e4 LT	88
	89	static DEFINE_SPINLOCK(appldata_timer_lock);
	90	static int appldata_interval = APPLDATA_CPU_INTERVAL;
	91	static int appldata_timer_active;
524dbcda	92	static int appldata_timer_suspended = 0;
1da177e4 LT	93
1da177e4 LT	94	/*
f26d583e	95	* Work queue
1da177e4	96	*/
f26d583e	97	static struct workqueue_struct *appldata_wq;
6d5aefb8 DH	98	static void appldata_work_fn(struct work_struct *work);
6d5aefb8 DH	99	static DECLARE_WORK(appldata_work, appldata_work_fn);
f26d583e	100
1da177e4 LT	101
	102	/*
	103	* Ops list
	104	*/
b1ad171e	105	static DEFINE_MUTEX(appldata_ops_mutex);
1da177e4 LT	106	static LIST_HEAD(appldata_ops_list);
	107
	108
f26d583e	109	/************************* timer, work, DIAG *****************************/
1da177e4 LT	110	/*
	111	* appldata_timer_function()
	112	*
f26d583e	113	* schedule work and reschedule timer
1da177e4	114	*/
9d0a57cb	115	static void appldata_timer_function(unsigned long data)
1da177e4	116	{
27f6b416	117	queue_work(appldata_wq, (struct work_struct *) data);
1da177e4 LT	118	}
	119
	120	/*
f26d583e	121	* appldata_work_fn()
1da177e4 LT	122	*
	123	* call data gathering function for each (active) module
	124	*/
6d5aefb8	125	static void appldata_work_fn(struct work_struct *work)
1da177e4 LT	126	{
	127	struct list_head *lh;
	128	struct appldata_ops *ops;
1da177e4	129
b1ad171e	130	mutex_lock(&appldata_ops_mutex);
1da177e4 LT	131	list_for_each(lh, &appldata_ops_list) {
1da177e4 LT	132	ops = list_entry(lh, struct appldata_ops, list);
1da177e4 LT	133	if (ops->active == 1) {
	134	ops->callback(ops->data);
	135	}
	136	}
b1ad171e	137	mutex_unlock(&appldata_ops_mutex);
1da177e4 LT	138	}
	139
	140	/*
	141	* appldata_diag()
	142	*
	143	* prepare parameter list, issue DIAG 0xDC
	144	*/
5b5dd21a GS	145	int appldata_diag(char record_nr, u16 function, unsigned long buffer,
5b5dd21a GS	146	u16 length, char *mod_lvl)
1da177e4	147	{
1f38d613	148	struct appldata_product_id id = {
1da177e4	149	.prod_nr = {0xD3, 0xC9, 0xD5, 0xE4,
1f38d613 GS	150	0xE7, 0xD2, 0xD9}, /* "LINUXKR" */
1f38d613 GS	151	.prod_fn = 0xD5D3, /* "NL" */
1f38d613 GS	152	.version_nr = 0xF2F6, /* "26" */
1f38d613 GS	153	.release_nr = 0xF0F1, /* "01" */
1da177e4 LT	154	};
1da177e4 LT	155
925afbd6 GS	156	id.record_nr = record_nr;
925afbd6 GS	157	id.mod_lvl = (mod_lvl[0]) << 8 \| mod_lvl[1];
1f38d613	158	return appldata_asm(&id, function, (void *) buffer, length);
1da177e4	159	}
f26d583e	160	/********************** timer, work, DIAG <END> **************************/
1da177e4 LT	161
	162
	163	/**************************** /proc stuff ********************************/
	164
1da177e4 LT	165	#define APPLDATA_ADD_TIMER 0
	166	#define APPLDATA_DEL_TIMER 1
	167	#define APPLDATA_MOD_TIMER 2
	168
	169	/*
	170	* __appldata_vtimer_setup()
	171	*
	172	* Add, delete or modify virtual timers on all online cpus.
	173	* The caller needs to get the appldata_timer_lock spinlock.
	174	*/
27f6b416	175	static void __appldata_vtimer_setup(int cmd)
1da177e4	176	{
27f6b416	177	u64 timer_interval = (u64) appldata_interval * 1000 * TOD_MICRO;
1da177e4 LT	178
	179	switch (cmd) {
	180	case APPLDATA_ADD_TIMER:
	181	if (appldata_timer_active)
	182	break;
27f6b416 MS	183	appldata_timer.expires = timer_interval;
27f6b416 MS	184	add_virt_timer_periodic(&appldata_timer);
1da177e4	185	appldata_timer_active = 1;
1da177e4 LT	186	break;
1da177e4 LT	187	case APPLDATA_DEL_TIMER:
27f6b416	188	del_virt_timer(&appldata_timer);
1da177e4 LT	189	if (!appldata_timer_active)
	190	break;
	191	appldata_timer_active = 0;
1da177e4 LT	192	break;
1da177e4 LT	193	case APPLDATA_MOD_TIMER:
1da177e4 LT	194	if (!appldata_timer_active)
1da177e4 LT	195	break;
27f6b416	196	mod_virt_timer_periodic(&appldata_timer, timer_interval);
1da177e4 LT	197	}
	198	}
	199
	200	/*
	201	* appldata_timer_handler()
	202	*
	203	* Start/Stop timer, show status of timer (0 = not active, 1 = active)
	204	*/
	205	static int
302bfe20	206	appldata_timer_handler(struct ctl_table *ctl, int write,
1da177e4 LT	207	void __user buffer, size_t lenp, loff_t *ppos)
1da177e4 LT	208	{
57f47ba0	209	unsigned int len;
1da177e4 LT	210	char buf[2];
	211
	212	if (!lenp \|\| ppos) {
	213	*lenp = 0;
	214	return 0;
	215	}
	216	if (!write) {
d5b4c2f4 CG	217	strncpy(buf, appldata_timer_active ? "1\n" : "0\n",
	218	ARRAY_SIZE(buf));
	219	len = strnlen(buf, ARRAY_SIZE(buf));
1da177e4 LT	220	if (len > *lenp)
	221	len = *lenp;
	222	if (copy_to_user(buffer, buf, len))
	223	return -EFAULT;
	224	goto out;
	225	}
	226	len = *lenp;
	227	if (copy_from_user(buf, buffer, len > sizeof(buf) ? sizeof(buf) : len))
	228	return -EFAULT;
	229	spin_lock(&appldata_timer_lock);
	230	if (buf[0] == '1')
	231	__appldata_vtimer_setup(APPLDATA_ADD_TIMER);
	232	else if (buf[0] == '0')
	233	__appldata_vtimer_setup(APPLDATA_DEL_TIMER);
	234	spin_unlock(&appldata_timer_lock);
	235	out:
	236	*lenp = len;
	237	*ppos += len;
	238	return 0;
	239	}
	240
	241	/*
	242	* appldata_interval_handler()
	243	*
	244	* Set (CPU) timer interval for collection of data (in milliseconds), show
	245	* current timer interval.
	246	*/
	247	static int
302bfe20	248	appldata_interval_handler(struct ctl_table *ctl, int write,
1da177e4 LT	249	void __user buffer, size_t lenp, loff_t *ppos)
1da177e4 LT	250	{
57f47ba0 HC	251	unsigned int len;
57f47ba0 HC	252	int interval;
1da177e4 LT	253	char buf[16];
	254
	255	if (!lenp \|\| ppos) {
	256	*lenp = 0;
	257	return 0;
	258	}
	259	if (!write) {
	260	len = sprintf(buf, "%i\n", appldata_interval);
	261	if (len > *lenp)
	262	len = *lenp;
	263	if (copy_to_user(buffer, buf, len))
	264	return -EFAULT;
	265	goto out;
	266	}
	267	len = *lenp;
27f6b416	268	if (copy_from_user(buf, buffer, len > sizeof(buf) ? sizeof(buf) : len))
1da177e4	269	return -EFAULT;
95425f19	270	interval = 0;
1da177e4	271	sscanf(buf, "%i", &interval);
d3ae942d	272	if (interval <= 0)
1da177e4	273	return -EINVAL;
1da177e4 LT	274
	275	spin_lock(&appldata_timer_lock);
	276	appldata_interval = interval;
	277	__appldata_vtimer_setup(APPLDATA_MOD_TIMER);
	278	spin_unlock(&appldata_timer_lock);
1da177e4 LT	279	out:
	280	*lenp = len;
	281	*ppos += len;
	282	return 0;
	283	}
	284
	285	/*
	286	* appldata_generic_handler()
	287	*
	288	* Generic start/stop monitoring and DIAG, show status of
	289	* monitoring (0 = not in process, 1 = in process)
	290	*/
	291	static int
302bfe20	292	appldata_generic_handler(struct ctl_table *ctl, int write,
1da177e4 LT	293	void __user buffer, size_t lenp, loff_t *ppos)
	294	{
	295	struct appldata_ops ops = NULL, tmp_ops;
57f47ba0 HC	296	unsigned int len;
57f47ba0 HC	297	int rc, found;
1da177e4 LT	298	char buf[2];
	299	struct list_head *lh;
	300
	301	found = 0;
b1ad171e	302	mutex_lock(&appldata_ops_mutex);
1da177e4 LT	303	list_for_each(lh, &appldata_ops_list) {
	304	tmp_ops = list_entry(lh, struct appldata_ops, list);
	305	if (&tmp_ops->ctl_table[2] == ctl) {
	306	found = 1;
	307	}
	308	}
	309	if (!found) {
b1ad171e	310	mutex_unlock(&appldata_ops_mutex);
1da177e4 LT	311	return -ENODEV;
	312	}
	313	ops = ctl->data;
	314	if (!try_module_get(ops->owner)) { // protect this function
b1ad171e	315	mutex_unlock(&appldata_ops_mutex);
1da177e4 LT	316	return -ENODEV;
1da177e4 LT	317	}
b1ad171e	318	mutex_unlock(&appldata_ops_mutex);
1da177e4 LT	319
	320	if (!lenp \|\| ppos) {
	321	*lenp = 0;
	322	module_put(ops->owner);
	323	return 0;
	324	}
	325	if (!write) {
d5b4c2f4 CG	326	strncpy(buf, ops->active ? "1\n" : "0\n", ARRAY_SIZE(buf));
d5b4c2f4 CG	327	len = strnlen(buf, ARRAY_SIZE(buf));
1da177e4 LT	328	if (len > *lenp)
	329	len = *lenp;
	330	if (copy_to_user(buffer, buf, len)) {
	331	module_put(ops->owner);
	332	return -EFAULT;
	333	}
	334	goto out;
	335	}
	336	len = *lenp;
	337	if (copy_from_user(buf, buffer,
	338	len > sizeof(buf) ? sizeof(buf) : len)) {
	339	module_put(ops->owner);
	340	return -EFAULT;
	341	}
	342
b1ad171e	343	mutex_lock(&appldata_ops_mutex);
1da177e4	344	if ((buf[0] == '1') && (ops->active == 0)) {
f26d583e GS	345	// protect work queue callback
f26d583e GS	346	if (!try_module_get(ops->owner)) {
b1ad171e	347	mutex_unlock(&appldata_ops_mutex);
1da177e4 LT	348	module_put(ops->owner);
	349	return -ENODEV;
	350	}
1da177e4 LT	351	ops->callback(ops->data); // init record
	352	rc = appldata_diag(ops->record_nr,
	353	APPLDATA_START_INTERVAL_REC,
5b5dd21a GS	354	(unsigned long) ops->data, ops->size,
5b5dd21a GS	355	ops->mod_lvl);
1da177e4	356	if (rc != 0) {
e7534b0e GS	357	pr_err("Starting the data collection for %s "
e7534b0e GS	358	"failed with rc=%d\n", ops->name, rc);
1da177e4	359	module_put(ops->owner);
d3ae942d	360	} else
5b5dd21a	361	ops->active = 1;
1da177e4 LT	362	} else if ((buf[0] == '0') && (ops->active == 1)) {
	363	ops->active = 0;
	364	rc = appldata_diag(ops->record_nr, APPLDATA_STOP_REC,
5b5dd21a GS	365	(unsigned long) ops->data, ops->size,
5b5dd21a GS	366	ops->mod_lvl);
d3ae942d	367	if (rc != 0)
e7534b0e GS	368	pr_err("Stopping the data collection for %s "
e7534b0e GS	369	"failed with rc=%d\n", ops->name, rc);
1da177e4 LT	370	module_put(ops->owner);
1da177e4 LT	371	}
b1ad171e	372	mutex_unlock(&appldata_ops_mutex);
1da177e4 LT	373	out:
	374	*lenp = len;
	375	*ppos += len;
	376	module_put(ops->owner);
	377	return 0;
	378	}
	379
	380	/************************* /proc stuff <END> *****************************/
	381
	382
	383	/*********************** module-ops management ***************************/
	384	/*
	385	* appldata_register_ops()
	386	*
	387	* update ops list, register /proc/sys entries
	388	*/
	389	int appldata_register_ops(struct appldata_ops *ops)
	390	{
13f8b7c5	391	if (ops->size > APPLDATA_MAX_REC_SIZE)
37e3a6ac	392	return -EINVAL;
1da177e4	393
37e3a6ac HC	394	ops->ctl_table = kzalloc(4 * sizeof(struct ctl_table), GFP_KERNEL);
37e3a6ac HC	395	if (!ops->ctl_table)
1da177e4	396	return -ENOMEM;
1da177e4	397
b1ad171e	398	mutex_lock(&appldata_ops_mutex);
1da177e4	399	list_add(&ops->list, &appldata_ops_list);
b1ad171e	400	mutex_unlock(&appldata_ops_mutex);
1da177e4	401
1da177e4 LT	402	ops->ctl_table[0].procname = appldata_proc_name;
	403	ops->ctl_table[0].maxlen = 0;
	404	ops->ctl_table[0].mode = S_IRUGO \| S_IXUGO;
	405	ops->ctl_table[0].child = &ops->ctl_table[2];
	406
1da177e4 LT	407	ops->ctl_table[2].procname = ops->name;
	408	ops->ctl_table[2].mode = S_IRUGO \| S_IWUSR;
	409	ops->ctl_table[2].proc_handler = appldata_generic_handler;
	410	ops->ctl_table[2].data = ops;
	411
0b4d4147	412	ops->sysctl_header = register_sysctl_table(ops->ctl_table);
37e3a6ac HC	413	if (!ops->sysctl_header)
37e3a6ac HC	414	goto out;
1da177e4	415	return 0;
37e3a6ac	416	out:
b1ad171e	417	mutex_lock(&appldata_ops_mutex);
37e3a6ac	418	list_del(&ops->list);
b1ad171e	419	mutex_unlock(&appldata_ops_mutex);
37e3a6ac HC	420	kfree(ops->ctl_table);
37e3a6ac HC	421	return -ENOMEM;
1da177e4 LT	422	}
	423
	424	/*
	425	* appldata_unregister_ops()
	426	*
	427	* update ops list, unregister /proc entries, stop DIAG if necessary
	428	*/
	429	void appldata_unregister_ops(struct appldata_ops *ops)
	430	{
b1ad171e	431	mutex_lock(&appldata_ops_mutex);
1da177e4	432	list_del(&ops->list);
b1ad171e	433	mutex_unlock(&appldata_ops_mutex);
330d57fb	434	unregister_sysctl_table(ops->sysctl_header);
37e3a6ac	435	kfree(ops->ctl_table);
1da177e4 LT	436	}
	437	/******************** module-ops management <END> ************************/
	438
	439
524dbcda GS	440	/************************** suspend / resume *****************************/
	441	static int appldata_freeze(struct device *dev)
	442	{
	443	struct appldata_ops *ops;
	444	int rc;
	445	struct list_head *lh;
	446
524dbcda GS	447	spin_lock(&appldata_timer_lock);
	448	if (appldata_timer_active) {
	449	__appldata_vtimer_setup(APPLDATA_DEL_TIMER);
	450	appldata_timer_suspended = 1;
	451	}
	452	spin_unlock(&appldata_timer_lock);
524dbcda GS	453
	454	mutex_lock(&appldata_ops_mutex);
	455	list_for_each(lh, &appldata_ops_list) {
	456	ops = list_entry(lh, struct appldata_ops, list);
	457	if (ops->active == 1) {
	458	rc = appldata_diag(ops->record_nr, APPLDATA_STOP_REC,
	459	(unsigned long) ops->data, ops->size,
	460	ops->mod_lvl);
	461	if (rc != 0)
	462	pr_err("Stopping the data collection for %s "
	463	"failed with rc=%d\n", ops->name, rc);
	464	}
	465	}
	466	mutex_unlock(&appldata_ops_mutex);
	467	return 0;
	468	}
	469
	470	static int appldata_restore(struct device *dev)
	471	{
	472	struct appldata_ops *ops;
	473	int rc;
	474	struct list_head *lh;
	475
524dbcda GS	476	spin_lock(&appldata_timer_lock);
	477	if (appldata_timer_suspended) {
	478	__appldata_vtimer_setup(APPLDATA_ADD_TIMER);
	479	appldata_timer_suspended = 0;
	480	}
	481	spin_unlock(&appldata_timer_lock);
524dbcda GS	482
	483	mutex_lock(&appldata_ops_mutex);
	484	list_for_each(lh, &appldata_ops_list) {
	485	ops = list_entry(lh, struct appldata_ops, list);
	486	if (ops->active == 1) {
	487	ops->callback(ops->data); // init record
	488	rc = appldata_diag(ops->record_nr,
	489	APPLDATA_START_INTERVAL_REC,
	490	(unsigned long) ops->data, ops->size,
	491	ops->mod_lvl);
	492	if (rc != 0) {
	493	pr_err("Starting the data collection for %s "
	494	"failed with rc=%d\n", ops->name, rc);
	495	}
	496	}
	497	}
	498	mutex_unlock(&appldata_ops_mutex);
	499	return 0;
	500	}
	501
	502	static int appldata_thaw(struct device *dev)
	503	{
	504	return appldata_restore(dev);
	505	}
	506
47145210	507	static const struct dev_pm_ops appldata_pm_ops = {
524dbcda GS	508	.freeze = appldata_freeze,
	509	.thaw = appldata_thaw,
	510	.restore = appldata_restore,
	511	};
	512
	513	static struct platform_driver appldata_pdrv = {
	514	.driver = {
	515	.name = "appldata",
524dbcda GS	516	.pm = &appldata_pm_ops,
	517	},
	518	};
	519	/*********************** suspend / resume <END> **************************/
	520
	521
1da177e4 LT	522	/***************************** init / exit *******************************/
1da177e4 LT	523
1da177e4 LT	524	/*
	525	* appldata_init()
	526	*
f26d583e	527	* init timer, register /proc entries
1da177e4 LT	528	*/
	529	static int __init appldata_init(void)
	530	{
27f6b416 MS	531	int rc;
27f6b416 MS	532
b7c5b1aa	533	init_virt_timer(&appldata_timer);
27f6b416 MS	534	appldata_timer.function = appldata_timer_function;
27f6b416 MS	535	appldata_timer.data = (unsigned long) &appldata_work;
524dbcda GS	536
	537	rc = platform_driver_register(&appldata_pdrv);
	538	if (rc)
	539	return rc;
1da177e4	540
524dbcda GS	541	appldata_pdev = platform_device_register_simple("appldata", -1, NULL,
	542	0);
	543	if (IS_ERR(appldata_pdev)) {
	544	rc = PTR_ERR(appldata_pdev);
	545	goto out_driver;
	546	}
e68f1d4c	547	appldata_wq = alloc_ordered_workqueue("appldata", 0);
524dbcda GS	548	if (!appldata_wq) {
	549	rc = -ENOMEM;
	550	goto out_device;
	551	}
f26d583e	552
0b4d4147	553	appldata_sysctl_header = register_sysctl_table(appldata_dir_table);
1da177e4	554	return 0;
524dbcda GS	555
	556	out_device:
	557	platform_device_unregister(appldata_pdev);
	558	out_driver:
	559	platform_driver_unregister(&appldata_pdrv);
	560	return rc;
1da177e4 LT	561	}
1da177e4 LT	562
076fc808	563	__initcall(appldata_init);
1da177e4	564
1da177e4 LT	565	/************************** init / exit <END> ****************************/
1da177e4 LT	566
1da177e4 LT	567	EXPORT_SYMBOL_GPL(appldata_register_ops);
1da177e4 LT	568	EXPORT_SYMBOL_GPL(appldata_unregister_ops);
5b5dd21a	569	EXPORT_SYMBOL_GPL(appldata_diag);
1da177e4	570
0c3252d5	571	#ifdef CONFIG_SWAP
1da177e4	572	EXPORT_SYMBOL_GPL(si_swapinfo);
0c3252d5	573	#endif
1da177e4	574	EXPORT_SYMBOL_GPL(nr_threads);
1da177e4 LT	575	EXPORT_SYMBOL_GPL(nr_running);
1da177e4 LT	576	EXPORT_SYMBOL_GPL(nr_iowait);