[mirror_ubuntu-hirsute-kernel.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/rtas.h>
#include "pseries.h"

static struct kobject *mobility_kobj;

struct update_props_workarea {
	u32 phandle;
	u32 state;
	u64 reserved;
	u32 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(u32 phandle)
{
	struct device_node *dn;

	dn = of_find_node_by_phandle(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(u32 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 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(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);

		/* 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 = *(u32 *)prop_data;
			prop_data += vd + sizeof(vd);
			upwa->nprops--;
		}

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

			prop_name = prop_data;
			prop_data += strlen(prop_name) + 1;
			vd = *(u32 *)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(u32 parent_phandle, u32 drc_index)
{
	struct device_node *dn;
	struct device_node *parent_dn;
	int rc;

	parent_dn = of_find_node_by_phandle(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;
	u32 *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 = (u32 *)rtas_buf + 4;
		while (*data & NODE_ACTION_MASK) {
			int i;
			u32 action = *data & NODE_ACTION_MASK;
			int node_count = *data & NODE_COUNT_MASK;

			data++;

			for (i = 0; i < node_count; i++) {
				u32 phandle = *data++;
				u32 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;

	rc = pseries_devicetree_update(MIGRATION_SCOPE);
	if (rc) {
		printk(KERN_ERR "Initial post-mobility device tree update "
		       "failed: %d\n", rc);
		return;
	}

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

	rc = rtas_call(activate_fw_token, 0, 1, NULL);
	if (!rc) {
		rc = pseries_devicetree_update(MIGRATION_SCOPE);
		if (rc)
			printk(KERN_ERR "Secondary post-mobility device tree "
			       "update failed: %d\n", rc);
	} else {
		printk(KERN_ERR "Post-mobility activate-fw failed: %d\n", rc);
		return;
	}

	return;
}

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

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

	memset(&args, 0, sizeof(args));
	args.token = rtas_token("ibm,suspend-me");
	args.nargs = 2;
	args.nret = 1;

	args.args[0] = streamid >> 32 ;
	args.args[1] = streamid & 0xffffffff;
	args.rets = &args.args[args.nargs];

	do {
		args.rets[0] = 0;
		rc = rtas_ibm_suspend_me(&args);
		if (!rc && args.rets[0] == RTAS_NOT_SUSPENDABLE)
			ssleep(1);
	} while (!rc && args.rets[0] == RTAS_NOT_SUSPENDABLE);

	if (rc)
		return rc;
	else if (args.rets[0])
		return args.rets[0];

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