2 * Support for Partition Mobility/Migration
4 * Copyright (C) 2010 Nathan Fontenot
5 * Copyright (C) 2010 IBM Corporation
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.
12 #include <linux/kernel.h>
13 #include <linux/kobject.h>
14 #include <linux/smp.h>
15 #include <linux/stat.h>
16 #include <linux/completion.h>
17 #include <linux/device.h>
18 #include <linux/delay.h>
19 #include <linux/slab.h>
24 static struct kobject *mobility_kobj;
26 struct update_props_workarea {
33 #define NODE_ACTION_MASK 0xff000000
34 #define NODE_COUNT_MASK 0x00ffffff
36 #define DELETE_DT_NODE 0x01000000
37 #define UPDATE_DT_NODE 0x02000000
38 #define ADD_DT_NODE 0x03000000
40 static int mobility_rtas_call(int token, char *buf)
44 spin_lock(&rtas_data_buf_lock);
46 memcpy(rtas_data_buf, buf, RTAS_DATA_BUF_SIZE);
47 rc = rtas_call(token, 2, 1, NULL, rtas_data_buf, 1);
48 memcpy(buf, rtas_data_buf, RTAS_DATA_BUF_SIZE);
50 spin_unlock(&rtas_data_buf_lock);
54 static int delete_dt_node(u32 phandle)
56 struct device_node *dn;
58 dn = of_find_node_by_phandle(phandle);
62 dlpar_detach_node(dn);
66 static int update_dt_property(struct device_node *dn, struct property **prop,
67 const char *name, u32 vd, char *value)
69 struct property *new_prop = *prop;
70 struct property *old_prop;
73 /* A negative 'vd' value indicates that only part of the new property
74 * value is contained in the buffer and we need to call
75 * ibm,update-properties again to get the rest of the value.
77 * A negative value is also the two's compliment of the actual value.
79 if (vd & 0x80000000) {
85 /* partial property fixup */
86 char *new_data = kzalloc(new_prop->length + vd, GFP_KERNEL);
90 memcpy(new_data, new_prop->value, new_prop->length);
91 memcpy(new_data + new_prop->length, value, vd);
93 kfree(new_prop->value);
94 new_prop->value = new_data;
95 new_prop->length += vd;
97 new_prop = kzalloc(sizeof(*new_prop), GFP_KERNEL);
101 new_prop->name = kstrdup(name, GFP_KERNEL);
102 if (!new_prop->name) {
107 new_prop->length = vd;
108 new_prop->value = kzalloc(new_prop->length, GFP_KERNEL);
109 if (!new_prop->value) {
110 kfree(new_prop->name);
115 memcpy(new_prop->value, value, vd);
120 old_prop = of_find_property(dn, new_prop->name, NULL);
122 prom_update_property(dn, new_prop, old_prop);
124 prom_add_property(dn, new_prop);
132 static int update_dt_node(u32 phandle)
134 struct update_props_workarea *upwa;
135 struct device_node *dn;
136 struct property *prop = NULL;
140 int update_properties_token;
142 update_properties_token = rtas_token("ibm,update-properties");
143 if (update_properties_token == RTAS_UNKNOWN_SERVICE)
146 rtas_buf = kzalloc(RTAS_DATA_BUF_SIZE, GFP_KERNEL);
150 dn = of_find_node_by_phandle(phandle);
156 upwa = (struct update_props_workarea *)&rtas_buf[0];
157 upwa->phandle = phandle;
160 rc = mobility_rtas_call(update_properties_token, rtas_buf);
164 prop_data = rtas_buf + sizeof(*upwa);
166 for (i = 0; i < upwa->nprops; i++) {
170 prop_name = prop_data + 1;
171 prop_data += strlen(prop_name) + 1;
176 /* name only property, nothing to do */
180 prop = of_find_property(dn, prop_name, NULL);
181 prom_remove_property(dn, prop);
186 rc = update_dt_property(dn, &prop, prop_name,
189 printk(KERN_ERR "Could not update %s"
190 " property\n", prop_name);
203 static int add_dt_node(u32 parent_phandle, u32 drc_index)
205 struct device_node *dn;
206 struct device_node *parent_dn;
209 dn = dlpar_configure_connector(drc_index);
213 parent_dn = of_find_node_by_phandle(parent_phandle);
215 dlpar_free_cc_nodes(dn);
219 dn->parent = parent_dn;
220 rc = dlpar_attach_node(dn);
222 dlpar_free_cc_nodes(dn);
224 of_node_put(parent_dn);
228 static int pseries_devicetree_update(void)
232 int update_nodes_token;
235 update_nodes_token = rtas_token("ibm,update-nodes");
236 if (update_nodes_token == RTAS_UNKNOWN_SERVICE)
239 rtas_buf = kzalloc(RTAS_DATA_BUF_SIZE, GFP_KERNEL);
244 rc = mobility_rtas_call(update_nodes_token, rtas_buf);
248 data = (u32 *)rtas_buf + 4;
249 while (*data & NODE_ACTION_MASK) {
251 u32 action = *data & NODE_ACTION_MASK;
252 int node_count = *data & NODE_COUNT_MASK;
256 for (i = 0; i < node_count; i++) {
257 u32 phandle = *data++;
262 delete_dt_node(phandle);
265 update_dt_node(phandle);
269 add_dt_node(phandle, drc_index);
280 void post_mobility_fixup(void)
283 int activate_fw_token;
285 rc = pseries_devicetree_update();
287 printk(KERN_ERR "Initial post-mobility device tree update "
292 activate_fw_token = rtas_token("ibm,activate-firmware");
293 if (activate_fw_token == RTAS_UNKNOWN_SERVICE) {
294 printk(KERN_ERR "Could not make post-mobility "
295 "activate-fw call.\n");
299 rc = rtas_call(activate_fw_token, 0, 1, NULL);
301 rc = pseries_devicetree_update();
303 printk(KERN_ERR "Secondary post-mobility device tree "
304 "update failed: %d\n", rc);
306 printk(KERN_ERR "Post-mobility activate-fw failed: %d\n", rc);
313 static ssize_t migrate_store(struct class *class, struct class_attribute *attr,
314 const char *buf, size_t count)
316 struct rtas_args args;
320 rc = strict_strtoull(buf, 0, &streamid);
324 memset(&args, 0, sizeof(args));
325 args.token = rtas_token("ibm,suspend-me");
329 args.args[0] = streamid >> 32 ;
330 args.args[1] = streamid & 0xffffffff;
331 args.rets = &args.args[args.nargs];
335 rc = rtas_ibm_suspend_me(&args);
336 if (!rc && args.rets[0] == RTAS_NOT_SUSPENDABLE)
338 } while (!rc && args.rets[0] == RTAS_NOT_SUSPENDABLE);
342 else if (args.rets[0])
345 post_mobility_fixup();
349 static CLASS_ATTR(migration, S_IWUSR, NULL, migrate_store);
351 static int __init mobility_sysfs_init(void)
355 mobility_kobj = kobject_create_and_add("mobility", kernel_kobj);
359 rc = sysfs_create_file(mobility_kobj, &class_attr_migration.attr);
363 device_initcall(mobility_sysfs_init);