[deliverable/linux.git] / arch / powerpc / platforms / pseries / mobility.c

/*
 * Support for Partition Mobility/Migration
 *
 * Copyright (C) 2010 Nathan Fontenot
 * Copyright (C) 2010 IBM Corporation
 *
 * This program is free software; you can redistribute it and/or
 * modify it under the terms of the GNU General Public License version
 * 2 as published by the Free Software Foundation.
 */

#include <linux/kernel.h>
#include <linux/kobject.h>
#include <linux/smp.h>
#include <linux/stat.h>
#include <linux/completion.h>
#include <linux/device.h>
#include <linux/delay.h>
#include <linux/slab.h>

#include <asm/machdep.h>
#include <asm/rtas.h>
#include "pseries.h"

static struct kobject *mobility_kobj;

struct update_props_workarea {
	__be32 phandle;
	__be32 state;
	__be64 reserved;
	__be32 nprops;
} __packed;

#define NODE_ACTION_MASK	0xff000000
#define NODE_COUNT_MASK		0x00ffffff

#define DELETE_DT_NODE	0x01000000
#define UPDATE_DT_NODE	0x02000000
#define ADD_DT_NODE	0x03000000

#define MIGRATION_SCOPE	(1)

static int mobility_rtas_call(int token, char *buf, s32 scope)
{
	int rc;

	spin_lock(&rtas_data_buf_lock);

	memcpy(rtas_data_buf, buf, RTAS_DATA_BUF_SIZE);
	rc = rtas_call(token, 2, 1, NULL, rtas_data_buf, scope);
	memcpy(buf, rtas_data_buf, RTAS_DATA_BUF_SIZE);

	spin_unlock(&rtas_data_buf_lock);
	return rc;
}

static int delete_dt_node(__be32 phandle)
{
	struct device_node *dn;

	dn = of_find_node_by_phandle(be32_to_cpu(phandle));
	if (!dn)
		return -ENOENT;

	dlpar_detach_node(dn);
	of_node_put(dn);
	return 0;
}

static int update_dt_property(struct device_node *dn, struct property **prop,
			      const char *name, u32 vd, char *value)
{
	struct property *new_prop = *prop;
	int more = 0;

	/* A negative 'vd' value indicates that only part of the new property
	 * value is contained in the buffer and we need to call
	 * ibm,update-properties again to get the rest of the value.
	 *
	 * A negative value is also the two's compliment of the actual value.
	 */
	if (vd & 0x80000000) {
		vd = ~vd + 1;
		more = 1;
	}

	if (new_prop) {
		/* partial property fixup */
		char *new_data = kzalloc(new_prop->length + vd, GFP_KERNEL);
		if (!new_data)
			return -ENOMEM;

		memcpy(new_data, new_prop->value, new_prop->length);
		memcpy(new_data + new_prop->length, value, vd);

		kfree(new_prop->value);
		new_prop->value = new_data;
		new_prop->length += vd;
	} else {
		new_prop = kzalloc(sizeof(*new_prop), GFP_KERNEL);
		if (!new_prop)
			return -ENOMEM;

		new_prop->name = kstrdup(name, GFP_KERNEL);
		if (!new_prop->name) {
			kfree(new_prop);
			return -ENOMEM;
		}

		new_prop->length = vd;
		new_prop->value = kzalloc(new_prop->length, GFP_KERNEL);
		if (!new_prop->value) {
			kfree(new_prop->name);
			kfree(new_prop);
			return -ENOMEM;
		}

		memcpy(new_prop->value, value, vd);
		*prop = new_prop;
	}

	if (!more) {
		of_update_property(dn, new_prop);
		*prop = NULL;
	}

	return 0;
}

static int update_dt_node(__be32 phandle, s32 scope)
{
	struct update_props_workarea *upwa;
	struct device_node *dn;
	struct property *prop = NULL;
	int i, rc, rtas_rc;
	char *prop_data;
	char *rtas_buf;
	int update_properties_token;
	u32 nprops;
	u32 vd;

	update_properties_token = rtas_token("ibm,update-properties");
	if (update_properties_token == RTAS_UNKNOWN_SERVICE)
		return -EINVAL;

	rtas_buf = kzalloc(RTAS_DATA_BUF_SIZE, GFP_KERNEL);
	if (!rtas_buf)
		return -ENOMEM;

	dn = of_find_node_by_phandle(be32_to_cpu(phandle));
	if (!dn) {
		kfree(rtas_buf);
		return -ENOENT;
	}

	upwa = (struct update_props_workarea *)&rtas_buf[0];
	upwa->phandle = phandle;

	do {
		rtas_rc = mobility_rtas_call(update_properties_token, rtas_buf,
					scope);
		if (rtas_rc < 0)
			break;

		prop_data = rtas_buf + sizeof(*upwa);
		nprops = be32_to_cpu(upwa->nprops);

		/* On the first call to ibm,update-properties for a node the
		 * the first property value descriptor contains an empty
		 * property name, the property value length encoded as u32,
		 * and the property value is the node path being updated.
		 */
		if (*prop_data == 0) {
			prop_data++;
			vd = be32_to_cpu(*(__be32 *)prop_data);
			prop_data += vd + sizeof(vd);
			nprops--;
		}

		for (i = 0; i < nprops; i++) {
			char *prop_name;

			prop_name = prop_data;
			prop_data += strlen(prop_name) + 1;
			vd = be32_to_cpu(*(__be32 *)prop_data);
			prop_data += sizeof(vd);

			switch (vd) {
			case 0x00000000:
				/* name only property, nothing to do */
				break;

			case 0x80000000:
				prop = of_find_property(dn, prop_name, NULL);
				of_remove_property(dn, prop);
				prop = NULL;
				break;

			default:
				rc = update_dt_property(dn, &prop, prop_name,
							vd, prop_data);
				if (rc) {
					printk(KERN_ERR "Could not update %s"
					       " property\n", prop_name);
				}

				prop_data += vd;
			}
		}
	} while (rtas_rc == 1);

	of_node_put(dn);
	kfree(rtas_buf);
	return 0;
}

static int add_dt_node(__be32 parent_phandle, __be32 drc_index)
{
	struct device_node *dn;
	struct device_node *parent_dn;
	int rc;

	parent_dn = of_find_node_by_phandle(be32_to_cpu(parent_phandle));
	if (!parent_dn)
		return -ENOENT;

	dn = dlpar_configure_connector(drc_index, parent_dn);
	if (!dn)
		return -ENOENT;

	rc = dlpar_attach_node(dn);
	if (rc)
		dlpar_free_cc_nodes(dn);

	of_node_put(parent_dn);
	return rc;
}

int pseries_devicetree_update(s32 scope)
{
	char *rtas_buf;
	__be32 *data;
	int update_nodes_token;
	int rc;

	update_nodes_token = rtas_token("ibm,update-nodes");
	if (update_nodes_token == RTAS_UNKNOWN_SERVICE)
		return -EINVAL;

	rtas_buf = kzalloc(RTAS_DATA_BUF_SIZE, GFP_KERNEL);
	if (!rtas_buf)
		return -ENOMEM;

	do {
		rc = mobility_rtas_call(update_nodes_token, rtas_buf, scope);
		if (rc && rc != 1)
			break;

		data = (__be32 *)rtas_buf + 4;
		while (be32_to_cpu(*data) & NODE_ACTION_MASK) {
			int i;
			u32 action = be32_to_cpu(*data) & NODE_ACTION_MASK;
			u32 node_count = be32_to_cpu(*data) & NODE_COUNT_MASK;

			data++;

			for (i = 0; i < node_count; i++) {
				__be32 phandle = *data++;
				__be32 drc_index;

				switch (action) {
				case DELETE_DT_NODE:
					delete_dt_node(phandle);
					break;
				case UPDATE_DT_NODE:
					update_dt_node(phandle, scope);
					break;
				case ADD_DT_NODE:
					drc_index = *data++;
					add_dt_node(phandle, drc_index);
					break;
				}
			}
		}
	} while (rc == 1);

	kfree(rtas_buf);
	return rc;
}

void post_mobility_fixup(void)
{
	int rc;
	int activate_fw_token;

	activate_fw_token = rtas_token("ibm,activate-firmware");
	if (activate_fw_token == RTAS_UNKNOWN_SERVICE) {
		printk(KERN_ERR "Could not make post-mobility "
		       "activate-fw call.\n");
		return;
	}

	do {
		rc = rtas_call(activate_fw_token, 0, 1, NULL);
	} while (rtas_busy_delay(rc));

	if (rc)
		printk(KERN_ERR "Post-mobility activate-fw failed: %d\n", rc);

	rc = pseries_devicetree_update(MIGRATION_SCOPE);
	if (rc)
		printk(KERN_ERR "Post-mobility device tree update "
			"failed: %d\n", rc);

	return;
}

static ssize_t migrate_store(struct class *class, struct class_attribute *attr,
			     const char *buf, size_t count)
{
	u64 streamid;
	int rc;
	int vasi_rc = 0;

	rc = kstrtou64(buf, 0, &streamid);
	if (rc)
		return rc;

	do {
		rc = rtas_ibm_suspend_me(streamid, &vasi_rc);
		if (!rc && vasi_rc == RTAS_NOT_SUSPENDABLE)
			ssleep(1);
	} while (!rc && vasi_rc == RTAS_NOT_SUSPENDABLE);

	if (rc)
		return rc;
	if (vasi_rc)
		return vasi_rc;

	post_mobility_fixup();
	return count;
}

static CLASS_ATTR(migration, S_IWUSR, NULL, migrate_store);

static int __init mobility_sysfs_init(void)
{
	int rc;

	mobility_kobj = kobject_create_and_add("mobility", kernel_kobj);
	if (!mobility_kobj)
		return -ENOMEM;

	rc = sysfs_create_file(mobility_kobj, &class_attr_migration.attr);

	return rc;
}
machine_device_initcall(pseries, mobility_sysfs_init);
Commit	Line	Data
410bccf9 NF	1	/*
	2	* Support for Partition Mobility/Migration
	3	*
	4	* Copyright (C) 2010 Nathan Fontenot
	5	* Copyright (C) 2010 IBM Corporation
	6	*
	7	* This program is free software; you can redistribute it and/or
	8	* modify it under the terms of the GNU General Public License version
	9	* 2 as published by the Free Software Foundation.
	10	*/
	11
	12	#include <linux/kernel.h>
	13	#include <linux/kobject.h>
	14	#include <linux/smp.h>
b56eade5	15	#include <linux/stat.h>
410bccf9 NF	16	#include <linux/completion.h>
	17	#include <linux/device.h>
	18	#include <linux/delay.h>
	19	#include <linux/slab.h>
	20
8e83e905	21	#include <asm/machdep.h>
410bccf9 NF	22	#include <asm/rtas.h>
	23	#include "pseries.h"
	24
	25	static struct kobject *mobility_kobj;
	26
	27	struct update_props_workarea {
f6ff0414 TD	28	__be32 phandle;
	29	__be32 state;
	30	__be64 reserved;
	31	__be32 nprops;
d0ef4403	32	} __packed;
410bccf9 NF	33
	34	#define NODE_ACTION_MASK 0xff000000
	35	#define NODE_COUNT_MASK 0x00ffffff
	36
	37	#define DELETE_DT_NODE 0x01000000
	38	#define UPDATE_DT_NODE 0x02000000
	39	#define ADD_DT_NODE 0x03000000
	40
762ec157 NF	41	#define MIGRATION_SCOPE (1)
	42
	43	static int mobility_rtas_call(int token, char *buf, s32 scope)
410bccf9 NF	44	{
	45	int rc;
	46
	47	spin_lock(&rtas_data_buf_lock);
	48
	49	memcpy(rtas_data_buf, buf, RTAS_DATA_BUF_SIZE);
762ec157	50	rc = rtas_call(token, 2, 1, NULL, rtas_data_buf, scope);
410bccf9 NF	51	memcpy(buf, rtas_data_buf, RTAS_DATA_BUF_SIZE);
	52
	53	spin_unlock(&rtas_data_buf_lock);
	54	return rc;
	55	}
	56
f6ff0414	57	static int delete_dt_node(__be32 phandle)
410bccf9 NF	58	{
	59	struct device_node *dn;
	60
f6ff0414	61	dn = of_find_node_by_phandle(be32_to_cpu(phandle));
410bccf9 NF	62	if (!dn)
	63	return -ENOENT;
	64
	65	dlpar_detach_node(dn);
14cd820a	66	of_node_put(dn);
410bccf9 NF	67	return 0;
	68	}
	69
	70	static int update_dt_property(struct device_node dn, struct property *prop,
	71	const char name, u32 vd, char value)
	72	{
	73	struct property new_prop = prop;
410bccf9 NF	74	int more = 0;
	75
	76	/* A negative 'vd' value indicates that only part of the new property
	77	* value is contained in the buffer and we need to call
	78	* ibm,update-properties again to get the rest of the value.
	79	*
	80	* A negative value is also the two's compliment of the actual value.
	81	*/
	82	if (vd & 0x80000000) {
	83	vd = ~vd + 1;
	84	more = 1;
	85	}
	86
	87	if (new_prop) {
	88	/* partial property fixup */
	89	char *new_data = kzalloc(new_prop->length + vd, GFP_KERNEL);
	90	if (!new_data)
	91	return -ENOMEM;
	92
	93	memcpy(new_data, new_prop->value, new_prop->length);
	94	memcpy(new_data + new_prop->length, value, vd);
	95
	96	kfree(new_prop->value);
	97	new_prop->value = new_data;
	98	new_prop->length += vd;
	99	} else {
	100	new_prop = kzalloc(sizeof(*new_prop), GFP_KERNEL);
	101	if (!new_prop)
	102	return -ENOMEM;
	103
	104	new_prop->name = kstrdup(name, GFP_KERNEL);
	105	if (!new_prop->name) {
	106	kfree(new_prop);
	107	return -ENOMEM;
	108	}
	109
	110	new_prop->length = vd;
	111	new_prop->value = kzalloc(new_prop->length, GFP_KERNEL);
	112	if (!new_prop->value) {
	113	kfree(new_prop->name);
	114	kfree(new_prop);
	115	return -ENOMEM;
	116	}
	117
	118	memcpy(new_prop->value, value, vd);
	119	*prop = new_prop;
	120	}
	121
	122	if (!more) {
79d1c712	123	of_update_property(dn, new_prop);
d8e533b4	124	*prop = NULL;
410bccf9 NF	125	}
	126
	127	return 0;
	128	}
	129
f6ff0414	130	static int update_dt_node(__be32 phandle, s32 scope)
410bccf9 NF	131	{
	132	struct update_props_workarea *upwa;
	133	struct device_node *dn;
	134	struct property *prop = NULL;
638a405f	135	int i, rc, rtas_rc;
410bccf9 NF	136	char *prop_data;
	137	char *rtas_buf;
	138	int update_properties_token;
f6ff0414	139	u32 nprops;
2e9b7b02	140	u32 vd;
410bccf9 NF	141
	142	update_properties_token = rtas_token("ibm,update-properties");
	143	if (update_properties_token == RTAS_UNKNOWN_SERVICE)
	144	return -EINVAL;
	145
	146	rtas_buf = kzalloc(RTAS_DATA_BUF_SIZE, GFP_KERNEL);
	147	if (!rtas_buf)
	148	return -ENOMEM;
	149
f6ff0414	150	dn = of_find_node_by_phandle(be32_to_cpu(phandle));
410bccf9 NF	151	if (!dn) {
	152	kfree(rtas_buf);
	153	return -ENOENT;
	154	}
	155
	156	upwa = (struct update_props_workarea *)&rtas_buf[0];
	157	upwa->phandle = phandle;
	158
	159	do {
638a405f	160	rtas_rc = mobility_rtas_call(update_properties_token, rtas_buf,
762ec157	161	scope);
638a405f	162	if (rtas_rc < 0)
410bccf9 NF	163	break;
	164
	165	prop_data = rtas_buf + sizeof(*upwa);
f6ff0414	166	nprops = be32_to_cpu(upwa->nprops);
410bccf9	167
c8f5a57c TD	168	/* On the first call to ibm,update-properties for a node the
	169	* the first property value descriptor contains an empty
	170	* property name, the property value length encoded as u32,
	171	* and the property value is the node path being updated.
2e9b7b02	172	*/
c8f5a57c TD	173	if (*prop_data == 0) {
c8f5a57c TD	174	prop_data++;
f6ff0414	175	vd = be32_to_cpu((__be32 )prop_data);
c8f5a57c	176	prop_data += vd + sizeof(vd);
f6ff0414	177	nprops--;
c8f5a57c	178	}
2e9b7b02	179
f6ff0414	180	for (i = 0; i < nprops; i++) {
410bccf9	181	char *prop_name;
410bccf9	182
2e9b7b02	183	prop_name = prop_data;
410bccf9	184	prop_data += strlen(prop_name) + 1;
f6ff0414	185	vd = be32_to_cpu((__be32 )prop_data);
2e9b7b02	186	prop_data += sizeof(vd);
410bccf9 NF	187
	188	switch (vd) {
	189	case 0x00000000:
	190	/* name only property, nothing to do */
	191	break;
	192
	193	case 0x80000000:
	194	prop = of_find_property(dn, prop_name, NULL);
79d1c712	195	of_remove_property(dn, prop);
410bccf9 NF	196	prop = NULL;
	197	break;
	198
	199	default:
	200	rc = update_dt_property(dn, &prop, prop_name,
	201	vd, prop_data);
	202	if (rc) {
	203	printk(KERN_ERR "Could not update %s"
	204	" property\n", prop_name);
	205	}
	206
	207	prop_data += vd;
	208	}
	209	}
638a405f	210	} while (rtas_rc == 1);
410bccf9 NF	211
	212	of_node_put(dn);
	213	kfree(rtas_buf);
	214	return 0;
	215	}
	216
f6ff0414	217	static int add_dt_node(__be32 parent_phandle, __be32 drc_index)
410bccf9 NF	218	{
	219	struct device_node *dn;
	220	struct device_node *parent_dn;
	221	int rc;
	222
f6ff0414	223	parent_dn = of_find_node_by_phandle(be32_to_cpu(parent_phandle));
8d5ff320	224	if (!parent_dn)
410bccf9 NF	225	return -ENOENT;
410bccf9 NF	226
8d5ff320 TD	227	dn = dlpar_configure_connector(drc_index, parent_dn);
8d5ff320 TD	228	if (!dn)
410bccf9	229	return -ENOENT;
410bccf9	230
410bccf9 NF	231	rc = dlpar_attach_node(dn);
	232	if (rc)
	233	dlpar_free_cc_nodes(dn);
	234
	235	of_node_put(parent_dn);
	236	return rc;
	237	}
	238
762ec157	239	int pseries_devicetree_update(s32 scope)
410bccf9 NF	240	{
410bccf9 NF	241	char *rtas_buf;
f6ff0414	242	__be32 *data;
410bccf9 NF	243	int update_nodes_token;
	244	int rc;
	245
	246	update_nodes_token = rtas_token("ibm,update-nodes");
	247	if (update_nodes_token == RTAS_UNKNOWN_SERVICE)
	248	return -EINVAL;
	249
	250	rtas_buf = kzalloc(RTAS_DATA_BUF_SIZE, GFP_KERNEL);
	251	if (!rtas_buf)
	252	return -ENOMEM;
	253
	254	do {
762ec157	255	rc = mobility_rtas_call(update_nodes_token, rtas_buf, scope);
410bccf9 NF	256	if (rc && rc != 1)
	257	break;
	258
f6ff0414 TD	259	data = (__be32 *)rtas_buf + 4;
f6ff0414 TD	260	while (be32_to_cpu(*data) & NODE_ACTION_MASK) {
410bccf9	261	int i;
f6ff0414 TD	262	u32 action = be32_to_cpu(*data) & NODE_ACTION_MASK;
f6ff0414 TD	263	u32 node_count = be32_to_cpu(*data) & NODE_COUNT_MASK;
410bccf9 NF	264
	265	data++;
	266
	267	for (i = 0; i < node_count; i++) {
f6ff0414 TD	268	__be32 phandle = *data++;
f6ff0414 TD	269	__be32 drc_index;
410bccf9 NF	270
	271	switch (action) {
	272	case DELETE_DT_NODE:
	273	delete_dt_node(phandle);
	274	break;
	275	case UPDATE_DT_NODE:
762ec157	276	update_dt_node(phandle, scope);
410bccf9 NF	277	break;
	278	case ADD_DT_NODE:
	279	drc_index = *data++;
	280	add_dt_node(phandle, drc_index);
	281	break;
	282	}
	283	}
	284	}
	285	} while (rc == 1);
	286
	287	kfree(rtas_buf);
	288	return rc;
	289	}
	290
	291	void post_mobility_fixup(void)
	292	{
	293	int rc;
	294	int activate_fw_token;
	295
410bccf9 NF	296	activate_fw_token = rtas_token("ibm,activate-firmware");
	297	if (activate_fw_token == RTAS_UNKNOWN_SERVICE) {
	298	printk(KERN_ERR "Could not make post-mobility "
	299	"activate-fw call.\n");
	300	return;
	301	}
	302
39a33b59 HM	303	do {
	304	rc = rtas_call(activate_fw_token, 0, 1, NULL);
	305	} while (rtas_busy_delay(rc));
	306
	307	if (rc)
410bccf9	308	printk(KERN_ERR "Post-mobility activate-fw failed: %d\n", rc);
39a33b59 HM	309
	310	rc = pseries_devicetree_update(MIGRATION_SCOPE);
	311	if (rc)
	312	printk(KERN_ERR "Post-mobility device tree update "
	313	"failed: %d\n", rc);
410bccf9 NF	314
	315	return;
	316	}
	317
	318	static ssize_t migrate_store(struct class class, struct class_attribute attr,
	319	const char *buf, size_t count)
	320	{
410bccf9 NF	321	u64 streamid;
410bccf9 NF	322	int rc;
3df76a9d	323	int vasi_rc = 0;
410bccf9	324
1618bd53	325	rc = kstrtou64(buf, 0, &streamid);
410bccf9 NF	326	if (rc)
	327	return rc;
	328
410bccf9	329	do {
3df76a9d CB	330	rc = rtas_ibm_suspend_me(streamid, &vasi_rc);
3df76a9d CB	331	if (!rc && vasi_rc == RTAS_NOT_SUSPENDABLE)
410bccf9	332	ssleep(1);
3df76a9d	333	} while (!rc && vasi_rc == RTAS_NOT_SUSPENDABLE);
410bccf9 NF	334
	335	if (rc)
	336	return rc;
3df76a9d CB	337	if (vasi_rc)
3df76a9d CB	338	return vasi_rc;
410bccf9 NF	339
	340	post_mobility_fixup();
	341	return count;
	342	}
	343
	344	static CLASS_ATTR(migration, S_IWUSR, NULL, migrate_store);
	345
	346	static int __init mobility_sysfs_init(void)
	347	{
	348	int rc;
	349
	350	mobility_kobj = kobject_create_and_add("mobility", kernel_kobj);
	351	if (!mobility_kobj)
	352	return -ENOMEM;
	353
	354	rc = sysfs_create_file(mobility_kobj, &class_attr_migration.attr);
	355
	356	return rc;
	357	}
8e83e905	358	machine_device_initcall(pseries, mobility_sysfs_init);