arch/powerpc/platforms/pseries/mobility.c

   1 /*
   2  * Support for Partition Mobility/Migration
   3  *
   4  * Copyright (C) 2010 Nathan Fontenot
   5  * Copyright (C) 2010 IBM Corporation
   6  *
   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.
  10  */
  11
  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>
  20
  21 #include <asm/machdep.h>
  22 #include <asm/rtas.h>
  23 #include "pseries.h"
  24
  25 static struct kobject *mobility_kobj;
  26
  27 struct update_props_workarea {
  28         u32 phandle;
  29         u32 state;
  30         u64 reserved;
  31         u32 nprops;
  32 } __packed;
  33
  34 #define NODE_ACTION_MASK        0xff000000
  35 #define NODE_COUNT_MASK         0x00ffffff
  36
  37 #define DELETE_DT_NODE  0x01000000
  38 #define UPDATE_DT_NODE  0x02000000
  39 #define ADD_DT_NODE     0x03000000
  40
  41 #define MIGRATION_SCOPE (1)
  42
  43 static int mobility_rtas_call(int token, char *buf, s32 scope)
  44 {
  45         int rc;
  46
  47         spin_lock(&rtas_data_buf_lock);
  48
  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);
  52
  53         spin_unlock(&rtas_data_buf_lock);
  54         return rc;
  55 }
  56
  57 static int delete_dt_node(u32 phandle)
  58 {
  59         struct device_node *dn;
  60
  61         dn = of_find_node_by_phandle(phandle);
  62         if (!dn)
  63                 return -ENOENT;
  64
  65         dlpar_detach_node(dn);
  66         of_node_put(dn);
  67         return 0;
  68 }
  69
  70 static int update_dt_property(struct device_node *dn, struct property **prop,
  71                               const char *name, u32 vd, char *value)
  72 {
  73         struct property *new_prop = *prop;
  74         int more = 0;
  75
  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.
  79          *
  80          * A negative value is also the two's compliment of the actual value.
  81          */
  82         if (vd & 0x80000000) {
  83                 vd = ~vd + 1;
  84                 more = 1;
  85         }
  86
  87         if (new_prop) {
  88                 /* partial property fixup */
  89                 char *new_data = kzalloc(new_prop->length + vd, GFP_KERNEL);
  90                 if (!new_data)
  91                         return -ENOMEM;
  92
  93                 memcpy(new_data, new_prop->value, new_prop->length);
  94                 memcpy(new_data + new_prop->length, value, vd);
  95
  96                 kfree(new_prop->value);
  97                 new_prop->value = new_data;
  98                 new_prop->length += vd;
  99         } else {
 100                 new_prop = kzalloc(sizeof(*new_prop), GFP_KERNEL);
 101                 if (!new_prop)
 102                         return -ENOMEM;
 103
 104                 new_prop->name = kstrdup(name, GFP_KERNEL);
 105                 if (!new_prop->name) {
 106                         kfree(new_prop);
 107                         return -ENOMEM;
 108                 }
 109
 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);
 114                         kfree(new_prop);
 115                         return -ENOMEM;
 116                 }
 117
 118                 memcpy(new_prop->value, value, vd);
 119                 *prop = new_prop;
 120         }
 121
 122         if (!more) {
 123                 of_update_property(dn, new_prop);
 124                 *prop = NULL;
 125         }
 126
 127         return 0;
 128 }
 129
 130 static int update_dt_node(u32 phandle, s32 scope)
 131 {
 132         struct update_props_workarea *upwa;
 133         struct device_node *dn;
 134         struct property *prop = NULL;
 135         int i, rc, rtas_rc;
 136         char *prop_data;
 137         char *rtas_buf;
 138         int update_properties_token;
 139         u32 vd;
 140
 141         update_properties_token = rtas_token("ibm,update-properties");
 142         if (update_properties_token == RTAS_UNKNOWN_SERVICE)
 143                 return -EINVAL;
 144
 145         rtas_buf = kzalloc(RTAS_DATA_BUF_SIZE, GFP_KERNEL);
 146         if (!rtas_buf)
 147                 return -ENOMEM;
 148
 149         dn = of_find_node_by_phandle(phandle);
 150         if (!dn) {
 151                 kfree(rtas_buf);
 152                 return -ENOENT;
 153         }
 154
 155         upwa = (struct update_props_workarea *)&rtas_buf[0];
 156         upwa->phandle = phandle;
 157
 158         do {
 159                 rtas_rc = mobility_rtas_call(update_properties_token, rtas_buf,
 160                                         scope);
 161                 if (rtas_rc < 0)
 162                         break;
 163
 164                 prop_data = rtas_buf + sizeof(*upwa);
 165
 166                 /* On the first call to ibm,update-properties for a node the
 167                  * the first property value descriptor contains an empty
 168                  * property name, the property value length encoded as u32,
 169                  * and the property value is the node path being updated.
 170                  */
 171                 if (*prop_data == 0) {
 172                         prop_data++;
 173                         vd = *(u32 *)prop_data;
 174                         prop_data += vd + sizeof(vd);
 175                         upwa->nprops--;
 176                 }
 177
 178                 for (i = 0; i < upwa->nprops; i++) {
 179                         char *prop_name;
 180
 181                         prop_name = prop_data;
 182                         prop_data += strlen(prop_name) + 1;
 183                         vd = *(u32 *)prop_data;
 184                         prop_data += sizeof(vd);
 185
 186                         switch (vd) {
 187                         case 0x00000000:
 188                                 /* name only property, nothing to do */
 189                                 break;
 190
 191                         case 0x80000000:
 192                                 prop = of_find_property(dn, prop_name, NULL);
 193                                 of_remove_property(dn, prop);
 194                                 prop = NULL;
 195                                 break;
 196
 197                         default:
 198                                 rc = update_dt_property(dn, &prop, prop_name,
 199                                                         vd, prop_data);
 200                                 if (rc) {
 201                                         printk(KERN_ERR "Could not update %s"
 202                                                " property\n", prop_name);
 203                                 }
 204
 205                                 prop_data += vd;
 206                         }
 207                 }
 208         } while (rtas_rc == 1);
 209
 210         of_node_put(dn);
 211         kfree(rtas_buf);
 212         return 0;
 213 }
 214
 215 static int add_dt_node(u32 parent_phandle, u32 drc_index)
 216 {
 217         struct device_node *dn;
 218         struct device_node *parent_dn;
 219         int rc;
 220
 221         parent_dn = of_find_node_by_phandle(parent_phandle);
 222         if (!parent_dn)
 223                 return -ENOENT;
 224
 225         dn = dlpar_configure_connector(drc_index, parent_dn);
 226         if (!dn)
 227                 return -ENOENT;
 228
 229         rc = dlpar_attach_node(dn);
 230         if (rc)
 231                 dlpar_free_cc_nodes(dn);
 232
 233         of_node_put(parent_dn);
 234         return rc;
 235 }
 236
 237 int pseries_devicetree_update(s32 scope)
 238 {
 239         char *rtas_buf;
 240         u32 *data;
 241         int update_nodes_token;
 242         int rc;
 243
 244         update_nodes_token = rtas_token("ibm,update-nodes");
 245         if (update_nodes_token == RTAS_UNKNOWN_SERVICE)
 246                 return -EINVAL;
 247
 248         rtas_buf = kzalloc(RTAS_DATA_BUF_SIZE, GFP_KERNEL);
 249         if (!rtas_buf)
 250                 return -ENOMEM;
 251
 252         do {
 253                 rc = mobility_rtas_call(update_nodes_token, rtas_buf, scope);
 254                 if (rc && rc != 1)
 255                         break;
 256
 257                 data = (u32 *)rtas_buf + 4;
 258                 while (*data & NODE_ACTION_MASK) {
 259                         int i;
 260                         u32 action = *data & NODE_ACTION_MASK;
 261                         int node_count = *data & NODE_COUNT_MASK;
 262
 263                         data++;
 264
 265                         for (i = 0; i < node_count; i++) {
 266                                 u32 phandle = *data++;
 267                                 u32 drc_index;
 268
 269                                 switch (action) {
 270                                 case DELETE_DT_NODE:
 271                                         delete_dt_node(phandle);
 272                                         break;
 273                                 case UPDATE_DT_NODE:
 274                                         update_dt_node(phandle, scope);
 275                                         break;
 276                                 case ADD_DT_NODE:
 277                                         drc_index = *data++;
 278                                         add_dt_node(phandle, drc_index);
 279                                         break;
 280                                 }
 281                         }
 282                 }
 283         } while (rc == 1);
 284
 285         kfree(rtas_buf);
 286         return rc;
 287 }
 288
 289 void post_mobility_fixup(void)
 290 {
 291         int rc;
 292         int activate_fw_token;
 293
 294         activate_fw_token = rtas_token("ibm,activate-firmware");
 295         if (activate_fw_token == RTAS_UNKNOWN_SERVICE) {
 296                 printk(KERN_ERR "Could not make post-mobility "
 297                        "activate-fw call.\n");
 298                 return;
 299         }
 300
 301         do {
 302                 rc = rtas_call(activate_fw_token, 0, 1, NULL);
 303         } while (rtas_busy_delay(rc));
 304
 305         if (rc)
 306                 printk(KERN_ERR "Post-mobility activate-fw failed: %d\n", rc);
 307
 308         rc = pseries_devicetree_update(MIGRATION_SCOPE);
 309         if (rc)
 310                 printk(KERN_ERR "Post-mobility device tree update "
 311                         "failed: %d\n", rc);
 312
 313         return;
 314 }
 315
 316 static ssize_t migrate_store(struct class *class, struct class_attribute *attr,
 317                              const char *buf, size_t count)
 318 {
 319         struct rtas_args args;
 320         u64 streamid;
 321         int rc;
 322
 323         rc = kstrtou64(buf, 0, &streamid);
 324         if (rc)
 325                 return rc;
 326
 327         memset(&args, 0, sizeof(args));
 328         args.token = rtas_token("ibm,suspend-me");
 329         args.nargs = 2;
 330         args.nret = 1;
 331
 332         args.args[0] = streamid >> 32 ;
 333         args.args[1] = streamid & 0xffffffff;
 334         args.rets = &args.args[args.nargs];
 335
 336         do {
 337                 args.rets[0] = 0;
 338                 rc = rtas_ibm_suspend_me(&args);
 339                 if (!rc && args.rets[0] == RTAS_NOT_SUSPENDABLE)
 340                         ssleep(1);
 341         } while (!rc && args.rets[0] == RTAS_NOT_SUSPENDABLE);
 342
 343         if (rc)
 344                 return rc;
 345         else if (args.rets[0])
 346                 return args.rets[0];
 347
 348         post_mobility_fixup();
 349         return count;
 350 }
 351
 352 static CLASS_ATTR(migration, S_IWUSR, NULL, migrate_store);
 353
 354 static int __init mobility_sysfs_init(void)
 355 {
 356         int rc;
 357
 358         mobility_kobj = kobject_create_and_add("mobility", kernel_kobj);
 359         if (!mobility_kobj)
 360                 return -ENOMEM;
 361
 362         rc = sysfs_create_file(mobility_kobj, &class_attr_migration.attr);
 363
 364         return rc;
 365 }
 366 machine_device_initcall(pseries, mobility_sysfs_init);