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>
21 #include <asm/machdep.h>
25 static struct kobject *mobility_kobj;
27 struct update_props_workarea {
34 #define NODE_ACTION_MASK 0xff000000
35 #define NODE_COUNT_MASK 0x00ffffff
37 #define DELETE_DT_NODE 0x01000000
38 #define UPDATE_DT_NODE 0x02000000
39 #define ADD_DT_NODE 0x03000000
41 #define MIGRATION_SCOPE (1)
43 static int mobility_rtas_call(int token, char *buf, s32 scope)
47 spin_lock(&rtas_data_buf_lock);
49 memcpy(rtas_data_buf, buf, RTAS_DATA_BUF_SIZE);
50 rc = rtas_call(token, 2, 1, NULL, rtas_data_buf, scope);
51 memcpy(buf, rtas_data_buf, RTAS_DATA_BUF_SIZE);
53 spin_unlock(&rtas_data_buf_lock);
57 static int delete_dt_node(__be32 phandle)
59 struct device_node *dn;
61 dn = of_find_node_by_phandle(be32_to_cpu(phandle));
65 dlpar_detach_node(dn);
70 static int update_dt_property(struct device_node *dn, struct property **prop,
71 const char *name, u32 vd, char *value)
73 struct property *new_prop = *prop;
76 /* A negative 'vd' value indicates that only part of the new property
77 * value is contained in the buffer and we need to call
78 * ibm,update-properties again to get the rest of the value.
80 * A negative value is also the two's compliment of the actual value.
82 if (vd & 0x80000000) {
88 /* partial property fixup */
89 char *new_data = kzalloc(new_prop->length + vd, GFP_KERNEL);
93 memcpy(new_data, new_prop->value, new_prop->length);
94 memcpy(new_data + new_prop->length, value, vd);
96 kfree(new_prop->value);
97 new_prop->value = new_data;
98 new_prop->length += vd;
100 new_prop = kzalloc(sizeof(*new_prop), GFP_KERNEL);
104 new_prop->name = kstrdup(name, GFP_KERNEL);
105 if (!new_prop->name) {
110 new_prop->length = vd;
111 new_prop->value = kzalloc(new_prop->length, GFP_KERNEL);
112 if (!new_prop->value) {
113 kfree(new_prop->name);
118 memcpy(new_prop->value, value, vd);
123 of_update_property(dn, new_prop);
130 static int update_dt_node(__be32 phandle, s32 scope)
132 struct update_props_workarea *upwa;
133 struct device_node *dn;
134 struct property *prop = NULL;
138 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(be32_to_cpu(phandle));
156 upwa = (struct update_props_workarea *)&rtas_buf[0];
157 upwa->phandle = phandle;
160 rtas_rc = mobility_rtas_call(update_properties_token, rtas_buf,
165 prop_data = rtas_buf + sizeof(*upwa);
166 nprops = be32_to_cpu(upwa->nprops);
168 /* On the first call to ibm,update-properties for a node the
169 * the first property value descriptor contains an empty
170 * property name, the property value length encoded as u32,
171 * and the property value is the node path being updated.
173 if (*prop_data == 0) {
175 vd = be32_to_cpu(*(__be32 *)prop_data);
176 prop_data += vd + sizeof(vd);
180 for (i = 0; i < nprops; i++) {
183 prop_name = prop_data;
184 prop_data += strlen(prop_name) + 1;
185 vd = be32_to_cpu(*(__be32 *)prop_data);
186 prop_data += sizeof(vd);
190 /* name only property, nothing to do */
194 prop = of_find_property(dn, prop_name, NULL);
195 of_remove_property(dn, prop);
200 rc = update_dt_property(dn, &prop, prop_name,
203 printk(KERN_ERR "Could not update %s"
204 " property\n", prop_name);
210 } while (rtas_rc == 1);
217 static int add_dt_node(__be32 parent_phandle, __be32 drc_index)
219 struct device_node *dn;
220 struct device_node *parent_dn;
223 parent_dn = of_find_node_by_phandle(be32_to_cpu(parent_phandle));
227 dn = dlpar_configure_connector(drc_index, parent_dn);
231 rc = dlpar_attach_node(dn);
233 dlpar_free_cc_nodes(dn);
235 of_node_put(parent_dn);
239 int pseries_devicetree_update(s32 scope)
243 int update_nodes_token;
246 update_nodes_token = rtas_token("ibm,update-nodes");
247 if (update_nodes_token == RTAS_UNKNOWN_SERVICE)
250 rtas_buf = kzalloc(RTAS_DATA_BUF_SIZE, GFP_KERNEL);
255 rc = mobility_rtas_call(update_nodes_token, rtas_buf, scope);
259 data = (__be32 *)rtas_buf + 4;
260 while (be32_to_cpu(*data) & NODE_ACTION_MASK) {
262 u32 action = be32_to_cpu(*data) & NODE_ACTION_MASK;
263 u32 node_count = be32_to_cpu(*data) & NODE_COUNT_MASK;
267 for (i = 0; i < node_count; i++) {
268 __be32 phandle = *data++;
273 delete_dt_node(phandle);
276 update_dt_node(phandle, scope);
280 add_dt_node(phandle, drc_index);
291 void post_mobility_fixup(void)
294 int activate_fw_token;
296 activate_fw_token = rtas_token("ibm,activate-firmware");
297 if (activate_fw_token == RTAS_UNKNOWN_SERVICE) {
298 printk(KERN_ERR "Could not make post-mobility "
299 "activate-fw call.\n");
304 rc = rtas_call(activate_fw_token, 0, 1, NULL);
305 } while (rtas_busy_delay(rc));
308 printk(KERN_ERR "Post-mobility activate-fw failed: %d\n", rc);
310 rc = pseries_devicetree_update(MIGRATION_SCOPE);
312 printk(KERN_ERR "Post-mobility device tree update "
318 static ssize_t migrate_store(struct class *class, struct class_attribute *attr,
319 const char *buf, size_t count)
324 rc = kstrtou64(buf, 0, &streamid);
329 rc = rtas_ibm_suspend_me(streamid);
332 } while (rc == -EAGAIN);
337 post_mobility_fixup();
342 * Used by drmgr to determine the kernel behavior of the migration interface.
344 * Version 1: Performs all PAPR requirements for migration including
345 * firmware activation and device tree update.
347 #define MIGRATION_API_VERSION 1
349 static CLASS_ATTR(migration, S_IWUSR, NULL, migrate_store);
350 static CLASS_ATTR_STRING(api_version, S_IRUGO, __stringify(MIGRATION_API_VERSION));
352 static int __init mobility_sysfs_init(void)
356 mobility_kobj = kobject_create_and_add("mobility", kernel_kobj);
360 rc = sysfs_create_file(mobility_kobj, &class_attr_migration.attr);
362 pr_err("mobility: unable to create migration sysfs file (%d)\n", rc);
364 rc = sysfs_create_file(mobility_kobj, &class_attr_api_version.attr.attr);
366 pr_err("mobility: unable to create api_version sysfs file (%d)\n", rc);
370 machine_device_initcall(pseries, mobility_sysfs_init);