[mirror_ubuntu-bionic-kernel.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/notifier.h>
#include <linux/spinlock.h>
#include <linux/cpu.h>
#include <linux/slab.h>
#include <linux/of.h>
#include "offline_states.h"
#include "pseries.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,
					       const char *path)
{
	struct device_node *dn;
	char *name;

	/* If parent node path is "/" advance path to NULL terminator to
	 * prevent double leading slashs in full_name.
	 */
	if (!path[1])
		path++;

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

	name = (char *)ccwa + ccwa->name_offset;
	dn->full_name = kasprintf(GFP_KERNEL, "%s/%s", path, name);
	if (!dn->full_name) {
		kfree(dn);
		return NULL;
	}

	of_node_set_flag(dn, OF_DYNAMIC);
	of_node_init(dn);

	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 *parent)
{
	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;
	const char *parent_path = parent->full_name;
	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, parent_path);
			if (!dn)
				goto cc_error;

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

		case NEXT_CHILD:
			if (first_dn)
				parent_path = last_dn->full_name;

			dn = dlpar_parse_cc_node(ccwa, parent_path);
			if (!dn)
				goto cc_error;

			if (!first_dn) {
				dn->parent = parent;
				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;
			parent_path = last_dn->parent->full_name;
			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;

	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)
{
	struct device_node *child;
	int rc;

	child = of_get_next_child(dn, NULL);
	while (child) {
		dlpar_detach_node(child);
		child = of_get_next_child(dn, child);
	}

	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 __be32 *intserv;
	u32 thread;

	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++) {
		thread = be32_to_cpu(intserv[i]);
		for_each_present_cpu(cpu) {
			if (get_hard_smp_processor_id(cpu) != thread)
				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", thread);
	}
	cpu_maps_update_done();

out:
	return rc;

}

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

	rc = kstrtou32(buf, 0, &drc_index);
	if (rc)
		return -EINVAL;

	parent = of_find_node_by_path("/cpus");
	if (!parent)
		return -ENODEV;

	dn = dlpar_configure_connector(drc_index, parent);
	if (!dn)
		return -EINVAL;

	of_node_put(parent);

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

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

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

	return count;
}

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

	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++) {
		thread = be32_to_cpu(intserv[i]);
		for_each_present_cpu(cpu) {
			if (get_hard_smp_processor_id(cpu) != thread)
				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, thread)
								!= 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", thread);
	}
	cpu_maps_update_done();

out:
	return rc;

}

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

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

	rc = of_property_read_u32(dn, "ibm,my-drc-index", &drc_index);
	if (rc) {
		of_node_put(dn);
		return -EINVAL;
	}

	rc = dlpar_offline_cpu(dn);
	if (rc) {
		of_node_put(dn);
		return -EINVAL;
	}

	rc = dlpar_release_drc(drc_index);
	if (rc) {
		of_node_put(dn);
		return rc;
	}

	rc = dlpar_detach_node(dn);
	if (rc) {
		dlpar_acquire_drc(drc_index);
		return rc;
	}

	of_node_put(dn);

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