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