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

/*
 * Support for dynamic reconfiguration for PCI, Memory, and CPU
 * Hotplug and Dynamic Logical Partitioning on RPA platforms.
 *
 * Copyright (C) 2009 Nathan Fontenot
 * Copyright (C) 2009 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/kref.h>
#include <linux/notifier.h>
#include <linux/spinlock.h>
#include <linux/cpu.h>
#include <linux/slab.h>
#include <linux/of.h>
#include "offline_states.h"

#include <asm/prom.h>
#include <asm/machdep.h>
#include <asm/uaccess.h>
#include <asm/rtas.h>

struct cc_workarea {
	u32	drc_index;
	u32	zero;
	u32	name_offset;
	u32	prop_length;
	u32	prop_offset;
};

void dlpar_free_cc_property(struct property *prop)
{
	kfree(prop->name);
	kfree(prop->value);
	kfree(prop);
}

static struct property *dlpar_parse_cc_property(struct cc_workarea *ccwa)
{
	struct property *prop;
	char *name;
	char *value;

	prop = kzalloc(sizeof(*prop), GFP_KERNEL);
	if (!prop)
		return NULL;

	name = (char *)ccwa + ccwa->name_offset;
	prop->name = kstrdup(name, GFP_KERNEL);

	prop->length = ccwa->prop_length;
	value = (char *)ccwa + ccwa->prop_offset;
	prop->value = kmemdup(value, prop->length, GFP_KERNEL);
	if (!prop->value) {
		dlpar_free_cc_property(prop);
		return NULL;
	}

	return prop;
}

static struct device_node *dlpar_parse_cc_node(struct cc_workarea *ccwa)
{
	struct device_node *dn;
	char *name;

	dn = kzalloc(sizeof(*dn), GFP_KERNEL);
	if (!dn)
		return NULL;

	/* The configure connector reported name does not contain a
	 * preceding '/', so we allocate a buffer large enough to
	 * prepend this to the full_name.
	 */
	name = (char *)ccwa + ccwa->name_offset;
	dn->full_name = kasprintf(GFP_KERNEL, "/%s", name);
	if (!dn->full_name) {
		kfree(dn);
		return NULL;
	}

	return dn;
}

static void dlpar_free_one_cc_node(struct device_node *dn)
{
	struct property *prop;

	while (dn->properties) {
		prop = dn->properties;
		dn->properties = prop->next;
		dlpar_free_cc_property(prop);
	}

	kfree(dn->full_name);
	kfree(dn);
}

void dlpar_free_cc_nodes(struct device_node *dn)
{
	if (dn->child)
		dlpar_free_cc_nodes(dn->child);

	if (dn->sibling)
		dlpar_free_cc_nodes(dn->sibling);

	dlpar_free_one_cc_node(dn);
}

#define COMPLETE	0
#define NEXT_SIBLING    1
#define NEXT_CHILD      2
#define NEXT_PROPERTY   3
#define PREV_PARENT     4
#define MORE_MEMORY     5
#define CALL_AGAIN	-2
#define ERR_CFG_USE     -9003

struct device_node *dlpar_configure_connector(u32 drc_index)
{
	struct device_node *dn;
	struct device_node *first_dn = NULL;
	struct device_node *last_dn = NULL;
	struct property *property;
	struct property *last_property = NULL;
	struct cc_workarea *ccwa;
	char *data_buf;
	int cc_token;
	int rc = -1;

	cc_token = rtas_token("ibm,configure-connector");
	if (cc_token == RTAS_UNKNOWN_SERVICE)
		return NULL;

	data_buf = kzalloc(RTAS_DATA_BUF_SIZE, GFP_KERNEL);
	if (!data_buf)
		return NULL;

	ccwa = (struct cc_workarea *)&data_buf[0];
	ccwa->drc_index = drc_index;
	ccwa->zero = 0;

	do {
		/* Since we release the rtas_data_buf lock between configure
		 * connector calls we want to re-populate the rtas_data_buffer
		 * with the contents of the previous call.
		 */
		spin_lock(&rtas_data_buf_lock);

		memcpy(rtas_data_buf, data_buf, RTAS_DATA_BUF_SIZE);
		rc = rtas_call(cc_token, 2, 1, NULL, rtas_data_buf, NULL);
		memcpy(data_buf, rtas_data_buf, RTAS_DATA_BUF_SIZE);

		spin_unlock(&rtas_data_buf_lock);

		switch (rc) {
		case COMPLETE:
			break;

		case NEXT_SIBLING:
			dn = dlpar_parse_cc_node(ccwa);
			if (!dn)
				goto cc_error;

			dn->parent = last_dn->parent;
			last_dn->sibling = dn;
			last_dn = dn;
			break;

		case NEXT_CHILD:
			dn = dlpar_parse_cc_node(ccwa);
			if (!dn)
				goto cc_error;

			if (!first_dn)
				first_dn = dn;
			else {
				dn->parent = last_dn;
				if (last_dn)
					last_dn->child = dn;
			}

			last_dn = dn;
			break;

		case NEXT_PROPERTY:
			property = dlpar_parse_cc_property(ccwa);
			if (!property)
				goto cc_error;

			if (!last_dn->properties)
				last_dn->properties = property;
			else
				last_property->next = property;

			last_property = property;
			break;

		case PREV_PARENT:
			last_dn = last_dn->parent;
			break;

		case CALL_AGAIN:
			break;

		case MORE_MEMORY:
		case ERR_CFG_USE:
		default:
			printk(KERN_ERR "Unexpected Error (%d) "
			       "returned from configure-connector\n", rc);
			goto cc_error;
		}
	} while (rc);

cc_error:
	kfree(data_buf);

	if (rc) {
		if (first_dn)
			dlpar_free_cc_nodes(first_dn);

		return NULL;
	}

	return first_dn;
}

static struct device_node *derive_parent(const char *path)
{
	struct device_node *parent;
	char *last_slash;

	last_slash = strrchr(path, '/');
	if (last_slash == path) {
		parent = of_find_node_by_path("/");
	} else {
		char *parent_path;
		int parent_path_len = last_slash - path + 1;
		parent_path = kmalloc(parent_path_len, GFP_KERNEL);
		if (!parent_path)
			return NULL;

		strlcpy(parent_path, path, parent_path_len);
		parent = of_find_node_by_path(parent_path);
		kfree(parent_path);
	}

	return parent;
}

int dlpar_attach_node(struct device_node *dn)
{
	int rc;

	of_node_set_flag(dn, OF_DYNAMIC);
	kref_init(&dn->kref);
	dn->parent = derive_parent(dn->full_name);
	if (!dn->parent)
		return -ENOMEM;

	rc = of_attach_node(dn);
	if (rc) {
		printk(KERN_ERR "Failed to add device node %s\n",
		       dn->full_name);
		return rc;
	}

	of_node_put(dn->parent);
	return 0;
}

int dlpar_detach_node(struct device_node *dn)
{
	int rc;

	rc = of_detach_node(dn);
	if (rc)
		return rc;

	of_node_put(dn); /* Must decrement the refcount */
	return 0;
}

#define DR_ENTITY_SENSE		9003
#define DR_ENTITY_PRESENT	1
#define DR_ENTITY_UNUSABLE	2
#define ALLOCATION_STATE	9003
#define ALLOC_UNUSABLE		0
#define ALLOC_USABLE		1
#define ISOLATION_STATE		9001
#define ISOLATE			0
#define UNISOLATE		1

int dlpar_acquire_drc(u32 drc_index)
{
	int dr_status, rc;

	rc = rtas_call(rtas_token("get-sensor-state"), 2, 2, &dr_status,
		       DR_ENTITY_SENSE, drc_index);
	if (rc || dr_status != DR_ENTITY_UNUSABLE)
		return -1;

	rc = rtas_set_indicator(ALLOCATION_STATE, drc_index, ALLOC_USABLE);
	if (rc)
		return rc;

	rc = rtas_set_indicator(ISOLATION_STATE, drc_index, UNISOLATE);
	if (rc) {
		rtas_set_indicator(ALLOCATION_STATE, drc_index, ALLOC_UNUSABLE);
		return rc;
	}

	return 0;
}

int dlpar_release_drc(u32 drc_index)
{
	int dr_status, rc;

	rc = rtas_call(rtas_token("get-sensor-state"), 2, 2, &dr_status,
		       DR_ENTITY_SENSE, drc_index);
	if (rc || dr_status != DR_ENTITY_PRESENT)
		return -1;

	rc = rtas_set_indicator(ISOLATION_STATE, drc_index, ISOLATE);
	if (rc)
		return rc;

	rc = rtas_set_indicator(ALLOCATION_STATE, drc_index, ALLOC_UNUSABLE);
	if (rc) {
		rtas_set_indicator(ISOLATION_STATE, drc_index, UNISOLATE);
		return rc;
	}

	return 0;
}

#ifdef CONFIG_ARCH_CPU_PROBE_RELEASE

static int dlpar_online_cpu(struct device_node *dn)
{
	int rc = 0;
	unsigned int cpu;
	int len, nthreads, i;
	const u32 *intserv;

	intserv = of_get_property(dn, "ibm,ppc-interrupt-server#s", &len);
	if (!intserv)
		return -EINVAL;

	nthreads = len / sizeof(u32);

	cpu_maps_update_begin();
	for (i = 0; i < nthreads; i++) {
		for_each_present_cpu(cpu) {
			if (get_hard_smp_processor_id(cpu) != intserv[i])
				continue;
			BUG_ON(get_cpu_current_state(cpu)
					!= CPU_STATE_OFFLINE);
			cpu_maps_update_done();
			rc = cpu_up(cpu);
			if (rc)
				goto out;
			cpu_maps_update_begin();

			break;
		}
		if (cpu == num_possible_cpus())
			printk(KERN_WARNING "Could not find cpu to online "
			       "with physical id 0x%x\n", intserv[i]);
	}
	cpu_maps_update_done();

out:
	return rc;

}

static ssize_t dlpar_cpu_probe(const char *buf, size_t count)
{
	struct device_node *dn;
	unsigned long drc_index;
	char *cpu_name;
	int rc;

	cpu_hotplug_driver_lock();
	rc = strict_strtoul(buf, 0, &drc_index);
	if (rc) {
		rc = -EINVAL;
		goto out;
	}

	dn = dlpar_configure_connector(drc_index);
	if (!dn) {
		rc = -EINVAL;
		goto out;
	}

	/* configure-connector reports cpus as living in the base
	 * directory of the device tree.  CPUs actually live in the
	 * cpus directory so we need to fixup the full_name.
	 */
	cpu_name = kasprintf(GFP_KERNEL, "/cpus%s", dn->full_name);
	if (!cpu_name) {
		dlpar_free_cc_nodes(dn);
		rc = -ENOMEM;
		goto out;
	}

	kfree(dn->full_name);
	dn->full_name = cpu_name;

	rc = dlpar_acquire_drc(drc_index);
	if (rc) {
		dlpar_free_cc_nodes(dn);
		rc = -EINVAL;
		goto out;
	}

	rc = dlpar_attach_node(dn);
	if (rc) {
		dlpar_release_drc(drc_index);
		dlpar_free_cc_nodes(dn);
		goto out;
	}

	rc = dlpar_online_cpu(dn);
out:
	cpu_hotplug_driver_unlock();

	return rc ? rc : count;
}

static int dlpar_offline_cpu(struct device_node *dn)
{
	int rc = 0;
	unsigned int cpu;
	int len, nthreads, i;
	const u32 *intserv;

	intserv = of_get_property(dn, "ibm,ppc-interrupt-server#s", &len);
	if (!intserv)
		return -EINVAL;

	nthreads = len / sizeof(u32);

	cpu_maps_update_begin();
	for (i = 0; i < nthreads; i++) {
		for_each_present_cpu(cpu) {
			if (get_hard_smp_processor_id(cpu) != intserv[i])
				continue;

			if (get_cpu_current_state(cpu) == CPU_STATE_OFFLINE)
				break;

			if (get_cpu_current_state(cpu) == CPU_STATE_ONLINE) {
				set_preferred_offline_state(cpu, CPU_STATE_OFFLINE);
				cpu_maps_update_done();
				rc = cpu_down(cpu);
				if (rc)
					goto out;
				cpu_maps_update_begin();
				break;

			}

			/*
			 * The cpu is in CPU_STATE_INACTIVE.
			 * Upgrade it's state to CPU_STATE_OFFLINE.
			 */
			set_preferred_offline_state(cpu, CPU_STATE_OFFLINE);
			BUG_ON(plpar_hcall_norets(H_PROD, intserv[i])
								!= H_SUCCESS);
			__cpu_die(cpu);
			break;
		}
		if (cpu == num_possible_cpus())
			printk(KERN_WARNING "Could not find cpu to offline "
			       "with physical id 0x%x\n", intserv[i]);
	}
	cpu_maps_update_done();

out:
	return rc;

}

static ssize_t dlpar_cpu_release(const char *buf, size_t count)
{
	struct device_node *dn;
	const u32 *drc_index;
	int rc;

	dn = of_find_node_by_path(buf);
	if (!dn)
		return -EINVAL;

	drc_index = of_get_property(dn, "ibm,my-drc-index", NULL);
	if (!drc_index) {
		of_node_put(dn);
		return -EINVAL;
	}

	cpu_hotplug_driver_lock();
	rc = dlpar_offline_cpu(dn);
	if (rc) {
		of_node_put(dn);
		rc = -EINVAL;
		goto out;
	}

	rc = dlpar_release_drc(*drc_index);
	if (rc) {
		of_node_put(dn);
		goto out;
	}

	rc = dlpar_detach_node(dn);
	if (rc) {
		dlpar_acquire_drc(*drc_index);
		goto out;
	}

	of_node_put(dn);
out:
	cpu_hotplug_driver_unlock();
	return rc ? rc : count;
}

static int __init pseries_dlpar_init(void)
{
	ppc_md.cpu_probe = dlpar_cpu_probe;
	ppc_md.cpu_release = dlpar_cpu_release;

	return 0;
}
machine_device_initcall(pseries, pseries_dlpar_init);

#endif /* CONFIG_ARCH_CPU_PROBE_RELEASE */
Commit	Line	Data
ab519a01 NF	1	/*
	2	* Support for dynamic reconfiguration for PCI, Memory, and CPU
	3	* Hotplug and Dynamic Logical Partitioning on RPA platforms.
	4	*
	5	* Copyright (C) 2009 Nathan Fontenot
	6	* Copyright (C) 2009 IBM Corporation
	7	*
	8	* This program is free software; you can redistribute it and/or
	9	* modify it under the terms of the GNU General Public License version
	10	* 2 as published by the Free Software Foundation.
	11	*/
	12
	13	#include <linux/kernel.h>
	14	#include <linux/kref.h>
	15	#include <linux/notifier.h>
ab519a01 NF	16	#include <linux/spinlock.h>
ab519a01 NF	17	#include <linux/cpu.h>
5a0e3ad6	18	#include <linux/slab.h>
1cf3d8b3	19	#include <linux/of.h>
b6db63d1	20	#include "offline_states.h"
ab519a01 NF	21
	22	#include <asm/prom.h>
	23	#include <asm/machdep.h>
	24	#include <asm/uaccess.h>
	25	#include <asm/rtas.h>
ab519a01 NF	26
	27	struct cc_workarea {
	28	u32 drc_index;
	29	u32 zero;
	30	u32 name_offset;
	31	u32 prop_length;
	32	u32 prop_offset;
	33	};
	34
20648974	35	void dlpar_free_cc_property(struct property *prop)
ab519a01 NF	36	{
	37	kfree(prop->name);
	38	kfree(prop->value);
	39	kfree(prop);
	40	}
	41
	42	static struct property dlpar_parse_cc_property(struct cc_workarea ccwa)
	43	{
	44	struct property *prop;
	45	char *name;
	46	char *value;
	47
	48	prop = kzalloc(sizeof(*prop), GFP_KERNEL);
	49	if (!prop)
	50	return NULL;
	51
	52	name = (char *)ccwa + ccwa->name_offset;
	53	prop->name = kstrdup(name, GFP_KERNEL);
	54
	55	prop->length = ccwa->prop_length;
	56	value = (char *)ccwa + ccwa->prop_offset;
e72ed6b5	57	prop->value = kmemdup(value, prop->length, GFP_KERNEL);
ab519a01 NF	58	if (!prop->value) {
	59	dlpar_free_cc_property(prop);
	60	return NULL;
	61	}
	62
ab519a01 NF	63	return prop;
	64	}
	65
	66	static struct device_node dlpar_parse_cc_node(struct cc_workarea ccwa)
	67	{
	68	struct device_node *dn;
	69	char *name;
	70
	71	dn = kzalloc(sizeof(*dn), GFP_KERNEL);
	72	if (!dn)
	73	return NULL;
	74
	75	/* The configure connector reported name does not contain a
25985edc	76	* preceding '/', so we allocate a buffer large enough to
ab519a01 NF	77	* prepend this to the full_name.
	78	*/
	79	name = (char *)ccwa + ccwa->name_offset;
43caa61f	80	dn->full_name = kasprintf(GFP_KERNEL, "/%s", name);
ab519a01 NF	81	if (!dn->full_name) {
	82	kfree(dn);
	83	return NULL;
	84	}
	85
ab519a01 NF	86	return dn;
	87	}
	88
	89	static void dlpar_free_one_cc_node(struct device_node *dn)
	90	{
	91	struct property *prop;
	92
	93	while (dn->properties) {
	94	prop = dn->properties;
	95	dn->properties = prop->next;
	96	dlpar_free_cc_property(prop);
	97	}
	98
	99	kfree(dn->full_name);
	100	kfree(dn);
	101	}
	102
20648974	103	void dlpar_free_cc_nodes(struct device_node *dn)
ab519a01 NF	104	{
	105	if (dn->child)
	106	dlpar_free_cc_nodes(dn->child);
	107
	108	if (dn->sibling)
	109	dlpar_free_cc_nodes(dn->sibling);
	110
	111	dlpar_free_one_cc_node(dn);
	112	}
	113
9c740025	114	#define COMPLETE 0
ab519a01 NF	115	#define NEXT_SIBLING 1
	116	#define NEXT_CHILD 2
	117	#define NEXT_PROPERTY 3
	118	#define PREV_PARENT 4
	119	#define MORE_MEMORY 5
	120	#define CALL_AGAIN -2
	121	#define ERR_CFG_USE -9003
	122
	123	struct device_node *dlpar_configure_connector(u32 drc_index)
	124	{
	125	struct device_node *dn;
	126	struct device_node *first_dn = NULL;
	127	struct device_node *last_dn = NULL;
	128	struct property *property;
	129	struct property *last_property = NULL;
	130	struct cc_workarea *ccwa;
93f68f1e	131	char *data_buf;
ab519a01	132	int cc_token;
93f68f1e	133	int rc = -1;
ab519a01 NF	134
	135	cc_token = rtas_token("ibm,configure-connector");
	136	if (cc_token == RTAS_UNKNOWN_SERVICE)
	137	return NULL;
	138
93f68f1e NF	139	data_buf = kzalloc(RTAS_DATA_BUF_SIZE, GFP_KERNEL);
	140	if (!data_buf)
	141	return NULL;
	142
	143	ccwa = (struct cc_workarea *)&data_buf[0];
ab519a01 NF	144	ccwa->drc_index = drc_index;
	145	ccwa->zero = 0;
	146
93f68f1e NF	147	do {
	148	/* Since we release the rtas_data_buf lock between configure
	149	* connector calls we want to re-populate the rtas_data_buffer
	150	* with the contents of the previous call.
	151	*/
	152	spin_lock(&rtas_data_buf_lock);
	153
	154	memcpy(rtas_data_buf, data_buf, RTAS_DATA_BUF_SIZE);
	155	rc = rtas_call(cc_token, 2, 1, NULL, rtas_data_buf, NULL);
	156	memcpy(data_buf, rtas_data_buf, RTAS_DATA_BUF_SIZE);
	157
	158	spin_unlock(&rtas_data_buf_lock);
	159
ab519a01	160	switch (rc) {
9c740025 AB	161	case COMPLETE:
	162	break;
	163
ab519a01 NF	164	case NEXT_SIBLING:
	165	dn = dlpar_parse_cc_node(ccwa);
	166	if (!dn)
	167	goto cc_error;
	168
	169	dn->parent = last_dn->parent;
	170	last_dn->sibling = dn;
	171	last_dn = dn;
	172	break;
	173
	174	case NEXT_CHILD:
	175	dn = dlpar_parse_cc_node(ccwa);
	176	if (!dn)
	177	goto cc_error;
	178
	179	if (!first_dn)
	180	first_dn = dn;
	181	else {
	182	dn->parent = last_dn;
	183	if (last_dn)
	184	last_dn->child = dn;
	185	}
	186
	187	last_dn = dn;
	188	break;
	189
	190	case NEXT_PROPERTY:
	191	property = dlpar_parse_cc_property(ccwa);
	192	if (!property)
	193	goto cc_error;
	194
	195	if (!last_dn->properties)
	196	last_dn->properties = property;
	197	else
	198	last_property->next = property;
	199
	200	last_property = property;
	201	break;
	202
	203	case PREV_PARENT:
	204	last_dn = last_dn->parent;
	205	break;
	206
	207	case CALL_AGAIN:
	208	break;
	209
	210	case MORE_MEMORY:
	211	case ERR_CFG_USE:
	212	default:
	213	printk(KERN_ERR "Unexpected Error (%d) "
	214	"returned from configure-connector\n", rc);
	215	goto cc_error;
	216	}
93f68f1e	217	} while (rc);
ab519a01	218
93f68f1e NF	219	cc_error:
	220	kfree(data_buf);
	221
	222	if (rc) {
	223	if (first_dn)
	224	dlpar_free_cc_nodes(first_dn);
	225
	226	return NULL;
ab519a01 NF	227	}
ab519a01 NF	228
ab519a01	229	return first_dn;
ab519a01 NF	230	}
	231
	232	static struct device_node derive_parent(const char path)
	233	{
	234	struct device_node *parent;
	235	char *last_slash;
	236
	237	last_slash = strrchr(path, '/');
	238	if (last_slash == path) {
	239	parent = of_find_node_by_path("/");
	240	} else {
	241	char *parent_path;
	242	int parent_path_len = last_slash - path + 1;
	243	parent_path = kmalloc(parent_path_len, GFP_KERNEL);
	244	if (!parent_path)
	245	return NULL;
	246
	247	strlcpy(parent_path, path, parent_path_len);
	248	parent = of_find_node_by_path(parent_path);
	249	kfree(parent_path);
	250	}
	251
	252	return parent;
	253	}
	254
	255	int dlpar_attach_node(struct device_node *dn)
	256	{
ab519a01 NF	257	int rc;
	258
	259	of_node_set_flag(dn, OF_DYNAMIC);
	260	kref_init(&dn->kref);
	261	dn->parent = derive_parent(dn->full_name);
	262	if (!dn->parent)
	263	return -ENOMEM;
	264
1cf3d8b3	265	rc = of_attach_node(dn);
3aef19f0	266	if (rc) {
ab519a01 NF	267	printk(KERN_ERR "Failed to add device node %s\n",
ab519a01 NF	268	dn->full_name);
3aef19f0	269	return rc;
ab519a01 NF	270	}
ab519a01 NF	271
ab519a01 NF	272	of_node_put(dn->parent);
	273	return 0;
	274	}
	275
	276	int dlpar_detach_node(struct device_node *dn)
	277	{
1cf3d8b3	278	int rc;
ab519a01	279
1cf3d8b3 NF	280	rc = of_detach_node(dn);
	281	if (rc)
	282	return rc;
	283
	284	of_node_put(dn); /* Must decrement the refcount */
ab519a01 NF	285	return 0;
	286	}
	287
	288	#define DR_ENTITY_SENSE 9003
	289	#define DR_ENTITY_PRESENT 1
	290	#define DR_ENTITY_UNUSABLE 2
	291	#define ALLOCATION_STATE 9003
	292	#define ALLOC_UNUSABLE 0
	293	#define ALLOC_USABLE 1
	294	#define ISOLATION_STATE 9001
	295	#define ISOLATE 0
	296	#define UNISOLATE 1
	297
	298	int dlpar_acquire_drc(u32 drc_index)
	299	{
	300	int dr_status, rc;
	301
	302	rc = rtas_call(rtas_token("get-sensor-state"), 2, 2, &dr_status,
	303	DR_ENTITY_SENSE, drc_index);
	304	if (rc \|\| dr_status != DR_ENTITY_UNUSABLE)
	305	return -1;
	306
	307	rc = rtas_set_indicator(ALLOCATION_STATE, drc_index, ALLOC_USABLE);
	308	if (rc)
	309	return rc;
	310
	311	rc = rtas_set_indicator(ISOLATION_STATE, drc_index, UNISOLATE);
	312	if (rc) {
	313	rtas_set_indicator(ALLOCATION_STATE, drc_index, ALLOC_UNUSABLE);
	314	return rc;
	315	}
	316
	317	return 0;
	318	}
	319
	320	int dlpar_release_drc(u32 drc_index)
	321	{
	322	int dr_status, rc;
	323
	324	rc = rtas_call(rtas_token("get-sensor-state"), 2, 2, &dr_status,
	325	DR_ENTITY_SENSE, drc_index);
	326	if (rc \|\| dr_status != DR_ENTITY_PRESENT)
	327	return -1;
	328
	329	rc = rtas_set_indicator(ISOLATION_STATE, drc_index, ISOLATE);
	330	if (rc)
	331	return rc;
	332
	333	rc = rtas_set_indicator(ALLOCATION_STATE, drc_index, ALLOC_UNUSABLE);
	334	if (rc) {
	335	rtas_set_indicator(ISOLATION_STATE, drc_index, UNISOLATE);
	336	return rc;
	337	}
	338
	339	return 0;
	340	}
	341
1a8061c4	342	#ifdef CONFIG_ARCH_CPU_PROBE_RELEASE
ab519a01	343
275a64f6 NF	344	static int dlpar_online_cpu(struct device_node *dn)
	345	{
	346	int rc = 0;
	347	unsigned int cpu;
	348	int len, nthreads, i;
	349	const u32 *intserv;
	350
	351	intserv = of_get_property(dn, "ibm,ppc-interrupt-server#s", &len);
	352	if (!intserv)
	353	return -EINVAL;
	354
	355	nthreads = len / sizeof(u32);
	356
	357	cpu_maps_update_begin();
	358	for (i = 0; i < nthreads; i++) {
	359	for_each_present_cpu(cpu) {
	360	if (get_hard_smp_processor_id(cpu) != intserv[i])
	361	continue;
	362	BUG_ON(get_cpu_current_state(cpu)
	363	!= CPU_STATE_OFFLINE);
	364	cpu_maps_update_done();
	365	rc = cpu_up(cpu);
	366	if (rc)
	367	goto out;
	368	cpu_maps_update_begin();
	369
	370	break;
	371	}
	372	if (cpu == num_possible_cpus())
	373	printk(KERN_WARNING "Could not find cpu to online "
	374	"with physical id 0x%x\n", intserv[i]);
	375	}
	376	cpu_maps_update_done();
	377
	378	out:
	379	return rc;
	380
	381	}
	382
1a8061c4 NF	383	static ssize_t dlpar_cpu_probe(const char *buf, size_t count)
	384	{
	385	struct device_node *dn;
	386	unsigned long drc_index;
	387	char *cpu_name;
	388	int rc;
	389
51badebd	390	cpu_hotplug_driver_lock();
1a8061c4	391	rc = strict_strtoul(buf, 0, &drc_index);
51badebd GS	392	if (rc) {
	393	rc = -EINVAL;
	394	goto out;
	395	}
1a8061c4 NF	396
1a8061c4 NF	397	dn = dlpar_configure_connector(drc_index);
51badebd GS	398	if (!dn) {
	399	rc = -EINVAL;
	400	goto out;
	401	}
1a8061c4 NF	402
	403	/* configure-connector reports cpus as living in the base
	404	* directory of the device tree. CPUs actually live in the
	405	* cpus directory so we need to fixup the full_name.
	406	*/
43caa61f	407	cpu_name = kasprintf(GFP_KERNEL, "/cpus%s", dn->full_name);
1a8061c4 NF	408	if (!cpu_name) {
1a8061c4 NF	409	dlpar_free_cc_nodes(dn);
51badebd GS	410	rc = -ENOMEM;
51badebd GS	411	goto out;
1a8061c4 NF	412	}
1a8061c4 NF	413
1a8061c4 NF	414	kfree(dn->full_name);
	415	dn->full_name = cpu_name;
	416
	417	rc = dlpar_acquire_drc(drc_index);
	418	if (rc) {
	419	dlpar_free_cc_nodes(dn);
51badebd GS	420	rc = -EINVAL;
51badebd GS	421	goto out;
1a8061c4 NF	422	}
	423
	424	rc = dlpar_attach_node(dn);
	425	if (rc) {
	426	dlpar_release_drc(drc_index);
	427	dlpar_free_cc_nodes(dn);
a7df5c5e	428	goto out;
1a8061c4 NF	429	}
1a8061c4 NF	430
275a64f6	431	rc = dlpar_online_cpu(dn);
51badebd GS	432	out:
51badebd GS	433	cpu_hotplug_driver_unlock();
b6db63d1	434
1a8061c4 NF	435	return rc ? rc : count;
	436	}
	437
275a64f6 NF	438	static int dlpar_offline_cpu(struct device_node *dn)
	439	{
	440	int rc = 0;
	441	unsigned int cpu;
	442	int len, nthreads, i;
	443	const u32 *intserv;
	444
	445	intserv = of_get_property(dn, "ibm,ppc-interrupt-server#s", &len);
	446	if (!intserv)
	447	return -EINVAL;
	448
	449	nthreads = len / sizeof(u32);
	450
	451	cpu_maps_update_begin();
	452	for (i = 0; i < nthreads; i++) {
	453	for_each_present_cpu(cpu) {
	454	if (get_hard_smp_processor_id(cpu) != intserv[i])
	455	continue;
	456
	457	if (get_cpu_current_state(cpu) == CPU_STATE_OFFLINE)
	458	break;
	459
	460	if (get_cpu_current_state(cpu) == CPU_STATE_ONLINE) {
ceddee23	461	set_preferred_offline_state(cpu, CPU_STATE_OFFLINE);
275a64f6 NF	462	cpu_maps_update_done();
	463	rc = cpu_down(cpu);
	464	if (rc)
	465	goto out;
	466	cpu_maps_update_begin();
	467	break;
	468
	469	}
	470
	471	/*
	472	* The cpu is in CPU_STATE_INACTIVE.
	473	* Upgrade it's state to CPU_STATE_OFFLINE.
	474	*/
	475	set_preferred_offline_state(cpu, CPU_STATE_OFFLINE);
	476	BUG_ON(plpar_hcall_norets(H_PROD, intserv[i])
	477	!= H_SUCCESS);
	478	__cpu_die(cpu);
	479	break;
	480	}
	481	if (cpu == num_possible_cpus())
	482	printk(KERN_WARNING "Could not find cpu to offline "
	483	"with physical id 0x%x\n", intserv[i]);
	484	}
	485	cpu_maps_update_done();
	486
	487	out:
	488	return rc;
	489
	490	}
	491
1a8061c4 NF	492	static ssize_t dlpar_cpu_release(const char *buf, size_t count)
	493	{
	494	struct device_node *dn;
	495	const u32 *drc_index;
	496	int rc;
	497
	498	dn = of_find_node_by_path(buf);
	499	if (!dn)
	500	return -EINVAL;
	501
	502	drc_index = of_get_property(dn, "ibm,my-drc-index", NULL);
	503	if (!drc_index) {
	504	of_node_put(dn);
	505	return -EINVAL;
	506	}
	507
51badebd	508	cpu_hotplug_driver_lock();
275a64f6	509	rc = dlpar_offline_cpu(dn);
b6db63d1 GS	510	if (rc) {
b6db63d1 GS	511	of_node_put(dn);
51badebd GS	512	rc = -EINVAL;
51badebd GS	513	goto out;
b6db63d1 GS	514	}
b6db63d1 GS	515
1a8061c4 NF	516	rc = dlpar_release_drc(*drc_index);
	517	if (rc) {
	518	of_node_put(dn);
51badebd	519	goto out;
1a8061c4 NF	520	}
	521
	522	rc = dlpar_detach_node(dn);
	523	if (rc) {
	524	dlpar_acquire_drc(*drc_index);
51badebd	525	goto out;
1a8061c4 NF	526	}
	527
	528	of_node_put(dn);
51badebd GS	529	out:
	530	cpu_hotplug_driver_unlock();
	531	return rc ? rc : count;
1a8061c4 NF	532	}
	533
	534	static int __init pseries_dlpar_init(void)
	535	{
	536	ppc_md.cpu_probe = dlpar_cpu_probe;
	537	ppc_md.cpu_release = dlpar_cpu_release;
	538
	539	return 0;
	540	}
	541	machine_device_initcall(pseries, pseries_dlpar_init);
	542
	543	#endif /* CONFIG_ARCH_CPU_PROBE_RELEASE */