drivers/nvdimm/dimm_devs.c

   1 /*
   2  * Copyright(c) 2013-2015 Intel Corporation. All rights reserved.
   3  *
   4  * This program is free software; you can redistribute it and/or modify
   5  * it under the terms of version 2 of the GNU General Public License as
   6  * published by the Free Software Foundation.
   7  *
   8  * This program is distributed in the hope that it will be useful, but
   9  * WITHOUT ANY WARRANTY; without even the implied warranty of
  10  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  11  * General Public License for more details.
  12  */
  13 #define pr_fmt(fmt) KBUILD_MODNAME ": " fmt
  14 #include <linux/moduleparam.h>
  15 #include <linux/vmalloc.h>
  16 #include <linux/device.h>
  17 #include <linux/ndctl.h>
  18 #include <linux/slab.h>
  19 #include <linux/io.h>
  20 #include <linux/fs.h>
  21 #include <linux/mm.h>
  22 #include "nd-core.h"
  23 #include "label.h"
  24 #include "pmem.h"
  25 #include "nd.h"
  26
  27 static DEFINE_IDA(dimm_ida);
  28
  29 static bool noblk;
  30 module_param(noblk, bool, 0444);
  31 MODULE_PARM_DESC(noblk, "force disable BLK / local alias support");
  32
  33 /*
  34  * Retrieve bus and dimm handle and return if this bus supports
  35  * get_config_data commands
  36  */
  37 int nvdimm_check_config_data(struct device *dev)
  38 {
  39         struct nvdimm *nvdimm = to_nvdimm(dev);
  40
  41         if (!nvdimm->cmd_mask ||
  42             !test_bit(ND_CMD_GET_CONFIG_DATA, &nvdimm->cmd_mask)) {
  43                 if (test_bit(NDD_ALIASING, &nvdimm->flags))
  44                         return -ENXIO;
  45                 else
  46                         return -ENOTTY;
  47         }
  48
  49         return 0;
  50 }
  51
  52 static int validate_dimm(struct nvdimm_drvdata *ndd)
  53 {
  54         int rc;
  55
  56         if (!ndd)
  57                 return -EINVAL;
  58
  59         rc = nvdimm_check_config_data(ndd->dev);
  60         if (rc)
  61                 dev_dbg(ndd->dev, "%pf: %s error: %d\n",
  62                                 __builtin_return_address(0), __func__, rc);
  63         return rc;
  64 }
  65
  66 /**
  67  * nvdimm_init_nsarea - determine the geometry of a dimm's namespace area
  68  * @nvdimm: dimm to initialize
  69  */
  70 int nvdimm_init_nsarea(struct nvdimm_drvdata *ndd)
  71 {
  72         struct nd_cmd_get_config_size *cmd = &ndd->nsarea;
  73         struct nvdimm_bus *nvdimm_bus = walk_to_nvdimm_bus(ndd->dev);
  74         struct nvdimm_bus_descriptor *nd_desc;
  75         int rc = validate_dimm(ndd);
  76         int cmd_rc = 0;
  77
  78         if (rc)
  79                 return rc;
  80
  81         if (cmd->config_size)
  82                 return 0; /* already valid */
  83
  84         memset(cmd, 0, sizeof(*cmd));
  85         nd_desc = nvdimm_bus->nd_desc;
  86         rc = nd_desc->ndctl(nd_desc, to_nvdimm(ndd->dev),
  87                         ND_CMD_GET_CONFIG_SIZE, cmd, sizeof(*cmd), &cmd_rc);
  88         if (rc < 0)
  89                 return rc;
  90         return cmd_rc;
  91 }
  92
  93 int nvdimm_get_config_data(struct nvdimm_drvdata *ndd, void *buf,
  94                            size_t offset, size_t len)
  95 {
  96         struct nvdimm_bus *nvdimm_bus = walk_to_nvdimm_bus(ndd->dev);
  97         struct nvdimm_bus_descriptor *nd_desc = nvdimm_bus->nd_desc;
  98         int rc = validate_dimm(ndd), cmd_rc = 0;
  99         struct nd_cmd_get_config_data_hdr *cmd;
 100         size_t max_cmd_size, buf_offset;
 101
 102         if (rc)
 103                 return rc;
 104
 105         if (offset + len > ndd->nsarea.config_size)
 106                 return -ENXIO;
 107
 108         max_cmd_size = min_t(u32, len, ndd->nsarea.max_xfer);
 109         cmd = kvzalloc(max_cmd_size + sizeof(*cmd), GFP_KERNEL);
 110         if (!cmd)
 111                 return -ENOMEM;
 112
 113         for (buf_offset = 0; len;
 114              len -= cmd->in_length, buf_offset += cmd->in_length) {
 115                 size_t cmd_size;
 116
 117                 cmd->in_offset = offset + buf_offset;
 118                 cmd->in_length = min(max_cmd_size, len);
 119
 120                 cmd_size = sizeof(*cmd) + cmd->in_length;
 121
 122                 rc = nd_desc->ndctl(nd_desc, to_nvdimm(ndd->dev),
 123                                 ND_CMD_GET_CONFIG_DATA, cmd, cmd_size, &cmd_rc);
 124                 if (rc < 0)
 125                         break;
 126                 if (cmd_rc < 0) {
 127                         rc = cmd_rc;
 128                         break;
 129                 }
 130
 131                 /* out_buf should be valid, copy it into our output buffer */
 132                 memcpy(buf + buf_offset, cmd->out_buf, cmd->in_length);
 133         }
 134         kvfree(cmd);
 135
 136         return rc;
 137 }
 138
 139 int nvdimm_set_config_data(struct nvdimm_drvdata *ndd, size_t offset,
 140                 void *buf, size_t len)
 141 {
 142         size_t max_cmd_size, buf_offset;
 143         struct nd_cmd_set_config_hdr *cmd;
 144         int rc = validate_dimm(ndd), cmd_rc = 0;
 145         struct nvdimm_bus *nvdimm_bus = walk_to_nvdimm_bus(ndd->dev);
 146         struct nvdimm_bus_descriptor *nd_desc = nvdimm_bus->nd_desc;
 147
 148         if (rc)
 149                 return rc;
 150
 151         if (offset + len > ndd->nsarea.config_size)
 152                 return -ENXIO;
 153
 154         max_cmd_size = min_t(u32, len, ndd->nsarea.max_xfer);
 155         cmd = kvzalloc(max_cmd_size + sizeof(*cmd) + sizeof(u32), GFP_KERNEL);
 156         if (!cmd)
 157                 return -ENOMEM;
 158
 159         for (buf_offset = 0; len; len -= cmd->in_length,
 160                         buf_offset += cmd->in_length) {
 161                 size_t cmd_size;
 162
 163                 cmd->in_offset = offset + buf_offset;
 164                 cmd->in_length = min(max_cmd_size, len);
 165                 memcpy(cmd->in_buf, buf + buf_offset, cmd->in_length);
 166
 167                 /* status is output in the last 4-bytes of the command buffer */
 168                 cmd_size = sizeof(*cmd) + cmd->in_length + sizeof(u32);
 169
 170                 rc = nd_desc->ndctl(nd_desc, to_nvdimm(ndd->dev),
 171                                 ND_CMD_SET_CONFIG_DATA, cmd, cmd_size, &cmd_rc);
 172                 if (rc < 0)
 173                         break;
 174                 if (cmd_rc < 0) {
 175                         rc = cmd_rc;
 176                         break;
 177                 }
 178         }
 179         kvfree(cmd);
 180
 181         return rc;
 182 }
 183
 184 void nvdimm_set_aliasing(struct device *dev)
 185 {
 186         struct nvdimm *nvdimm = to_nvdimm(dev);
 187
 188         set_bit(NDD_ALIASING, &nvdimm->flags);
 189 }
 190
 191 void nvdimm_set_locked(struct device *dev)
 192 {
 193         struct nvdimm *nvdimm = to_nvdimm(dev);
 194
 195         set_bit(NDD_LOCKED, &nvdimm->flags);
 196 }
 197
 198 void nvdimm_clear_locked(struct device *dev)
 199 {
 200         struct nvdimm *nvdimm = to_nvdimm(dev);
 201
 202         clear_bit(NDD_LOCKED, &nvdimm->flags);
 203 }
 204
 205 static void nvdimm_release(struct device *dev)
 206 {
 207         struct nvdimm *nvdimm = to_nvdimm(dev);
 208
 209         ida_simple_remove(&dimm_ida, nvdimm->id);
 210         kfree(nvdimm);
 211 }
 212
 213 static struct device_type nvdimm_device_type = {
 214         .name = "nvdimm",
 215         .release = nvdimm_release,
 216 };
 217
 218 bool is_nvdimm(struct device *dev)
 219 {
 220         return dev->type == &nvdimm_device_type;
 221 }
 222
 223 struct nvdimm *to_nvdimm(struct device *dev)
 224 {
 225         struct nvdimm *nvdimm = container_of(dev, struct nvdimm, dev);
 226
 227         WARN_ON(!is_nvdimm(dev));
 228         return nvdimm;
 229 }
 230 EXPORT_SYMBOL_GPL(to_nvdimm);
 231
 232 struct nvdimm *nd_blk_region_to_dimm(struct nd_blk_region *ndbr)
 233 {
 234         struct nd_region *nd_region = &ndbr->nd_region;
 235         struct nd_mapping *nd_mapping = &nd_region->mapping[0];
 236
 237         return nd_mapping->nvdimm;
 238 }
 239 EXPORT_SYMBOL_GPL(nd_blk_region_to_dimm);
 240
 241 unsigned long nd_blk_memremap_flags(struct nd_blk_region *ndbr)
 242 {
 243         /* pmem mapping properties are private to libnvdimm */
 244         return ARCH_MEMREMAP_PMEM;
 245 }
 246 EXPORT_SYMBOL_GPL(nd_blk_memremap_flags);
 247
 248 struct nvdimm_drvdata *to_ndd(struct nd_mapping *nd_mapping)
 249 {
 250         struct nvdimm *nvdimm = nd_mapping->nvdimm;
 251
 252         WARN_ON_ONCE(!is_nvdimm_bus_locked(&nvdimm->dev));
 253
 254         return dev_get_drvdata(&nvdimm->dev);
 255 }
 256 EXPORT_SYMBOL(to_ndd);
 257
 258 void nvdimm_drvdata_release(struct kref *kref)
 259 {
 260         struct nvdimm_drvdata *ndd = container_of(kref, typeof(*ndd), kref);
 261         struct device *dev = ndd->dev;
 262         struct resource *res, *_r;
 263
 264         dev_dbg(dev, "trace\n");
 265         nvdimm_bus_lock(dev);
 266         for_each_dpa_resource_safe(ndd, res, _r)
 267                 nvdimm_free_dpa(ndd, res);
 268         nvdimm_bus_unlock(dev);
 269
 270         kvfree(ndd->data);
 271         kfree(ndd);
 272         put_device(dev);
 273 }
 274
 275 void get_ndd(struct nvdimm_drvdata *ndd)
 276 {
 277         kref_get(&ndd->kref);
 278 }
 279
 280 void put_ndd(struct nvdimm_drvdata *ndd)
 281 {
 282         if (ndd)
 283                 kref_put(&ndd->kref, nvdimm_drvdata_release);
 284 }
 285
 286 const char *nvdimm_name(struct nvdimm *nvdimm)
 287 {
 288         return dev_name(&nvdimm->dev);
 289 }
 290 EXPORT_SYMBOL_GPL(nvdimm_name);
 291
 292 struct kobject *nvdimm_kobj(struct nvdimm *nvdimm)
 293 {
 294         return &nvdimm->dev.kobj;
 295 }
 296 EXPORT_SYMBOL_GPL(nvdimm_kobj);
 297
 298 unsigned long nvdimm_cmd_mask(struct nvdimm *nvdimm)
 299 {
 300         return nvdimm->cmd_mask;
 301 }
 302 EXPORT_SYMBOL_GPL(nvdimm_cmd_mask);
 303
 304 void *nvdimm_provider_data(struct nvdimm *nvdimm)
 305 {
 306         if (nvdimm)
 307                 return nvdimm->provider_data;
 308         return NULL;
 309 }
 310 EXPORT_SYMBOL_GPL(nvdimm_provider_data);
 311
 312 static ssize_t commands_show(struct device *dev,
 313                 struct device_attribute *attr, char *buf)
 314 {
 315         struct nvdimm *nvdimm = to_nvdimm(dev);
 316         int cmd, len = 0;
 317
 318         if (!nvdimm->cmd_mask)
 319                 return sprintf(buf, "\n");
 320
 321         for_each_set_bit(cmd, &nvdimm->cmd_mask, BITS_PER_LONG)
 322                 len += sprintf(buf + len, "%s ", nvdimm_cmd_name(cmd));
 323         len += sprintf(buf + len, "\n");
 324         return len;
 325 }
 326 static DEVICE_ATTR_RO(commands);
 327
 328 static ssize_t flags_show(struct device *dev,
 329                 struct device_attribute *attr, char *buf)
 330 {
 331         struct nvdimm *nvdimm = to_nvdimm(dev);
 332
 333         return sprintf(buf, "%s%s\n",
 334                         test_bit(NDD_ALIASING, &nvdimm->flags) ? "alias " : "",
 335                         test_bit(NDD_LOCKED, &nvdimm->flags) ? "lock " : "");
 336 }
 337 static DEVICE_ATTR_RO(flags);
 338
 339 static ssize_t state_show(struct device *dev, struct device_attribute *attr,
 340                 char *buf)
 341 {
 342         struct nvdimm *nvdimm = to_nvdimm(dev);
 343
 344         /*
 345          * The state may be in the process of changing, userspace should
 346          * quiesce probing if it wants a static answer
 347          */
 348         nvdimm_bus_lock(dev);
 349         nvdimm_bus_unlock(dev);
 350         return sprintf(buf, "%s\n", atomic_read(&nvdimm->busy)
 351                         ? "active" : "idle");
 352 }
 353 static DEVICE_ATTR_RO(state);
 354
 355 static ssize_t available_slots_show(struct device *dev,
 356                 struct device_attribute *attr, char *buf)
 357 {
 358         struct nvdimm_drvdata *ndd = dev_get_drvdata(dev);
 359         ssize_t rc;
 360         u32 nfree;
 361
 362         if (!ndd)
 363                 return -ENXIO;
 364
 365         nvdimm_bus_lock(dev);
 366         nfree = nd_label_nfree(ndd);
 367         if (nfree - 1 > nfree) {
 368                 dev_WARN_ONCE(dev, 1, "we ate our last label?\n");
 369                 nfree = 0;
 370         } else
 371                 nfree--;
 372         rc = sprintf(buf, "%d\n", nfree);
 373         nvdimm_bus_unlock(dev);
 374         return rc;
 375 }
 376 static DEVICE_ATTR_RO(available_slots);
 377
 378 __weak ssize_t security_show(struct device *dev,
 379                 struct device_attribute *attr, char *buf)
 380 {
 381         struct nvdimm *nvdimm = to_nvdimm(dev);
 382
 383         switch (nvdimm->sec.state) {
 384         case NVDIMM_SECURITY_DISABLED:
 385                 return sprintf(buf, "disabled\n");
 386         case NVDIMM_SECURITY_UNLOCKED:
 387                 return sprintf(buf, "unlocked\n");
 388         case NVDIMM_SECURITY_LOCKED:
 389                 return sprintf(buf, "locked\n");
 390         case NVDIMM_SECURITY_FROZEN:
 391                 return sprintf(buf, "frozen\n");
 392         case NVDIMM_SECURITY_OVERWRITE:
 393                 return sprintf(buf, "overwrite\n");
 394         default:
 395                 return -ENOTTY;
 396         }
 397
 398         return -ENOTTY;
 399 }
 400
 401 #define OPS                                                     \
 402         C( OP_FREEZE,           "freeze",               1),     \
 403         C( OP_DISABLE,          "disable",              2),     \
 404         C( OP_UPDATE,           "update",               3),     \
 405         C( OP_ERASE,            "erase",                2),     \
 406         C( OP_OVERWRITE,        "overwrite",            2),     \
 407         C( OP_MASTER_UPDATE,    "master_update",        3),     \
 408         C( OP_MASTER_ERASE,     "master_erase",         2)
 409 #undef C
 410 #define C(a, b, c) a
 411 enum nvdimmsec_op_ids { OPS };
 412 #undef C
 413 #define C(a, b, c) { b, c }
 414 static struct {
 415         const char *name;
 416         int args;
 417 } ops[] = { OPS };
 418 #undef C
 419
 420 #define SEC_CMD_SIZE 32
 421 #define KEY_ID_SIZE 10
 422
 423 static ssize_t __security_store(struct device *dev, const char *buf, size_t len)
 424 {
 425         struct nvdimm *nvdimm = to_nvdimm(dev);
 426         ssize_t rc;
 427         char cmd[SEC_CMD_SIZE+1], keystr[KEY_ID_SIZE+1],
 428                 nkeystr[KEY_ID_SIZE+1];
 429         unsigned int key, newkey;
 430         int i;
 431
 432         if (atomic_read(&nvdimm->busy))
 433                 return -EBUSY;
 434
 435         rc = sscanf(buf, "%"__stringify(SEC_CMD_SIZE)"s"
 436                         " %"__stringify(KEY_ID_SIZE)"s"
 437                         " %"__stringify(KEY_ID_SIZE)"s",
 438                         cmd, keystr, nkeystr);
 439         if (rc < 1)
 440                 return -EINVAL;
 441         for (i = 0; i < ARRAY_SIZE(ops); i++)
 442                 if (sysfs_streq(cmd, ops[i].name))
 443                         break;
 444         if (i >= ARRAY_SIZE(ops))
 445                 return -EINVAL;
 446         if (ops[i].args > 1)
 447                 rc = kstrtouint(keystr, 0, &key);
 448         if (rc >= 0 && ops[i].args > 2)
 449                 rc = kstrtouint(nkeystr, 0, &newkey);
 450         if (rc < 0)
 451                 return rc;
 452
 453         if (i == OP_FREEZE) {
 454                 dev_dbg(dev, "freeze\n");
 455                 rc = nvdimm_security_freeze(nvdimm);
 456         } else if (i == OP_DISABLE) {
 457                 dev_dbg(dev, "disable %u\n", key);
 458                 rc = nvdimm_security_disable(nvdimm, key);
 459         } else if (i == OP_UPDATE) {
 460                 dev_dbg(dev, "update %u %u\n", key, newkey);
 461                 rc = nvdimm_security_update(nvdimm, key, newkey, NVDIMM_USER);
 462         } else if (i == OP_ERASE) {
 463                 dev_dbg(dev, "erase %u\n", key);
 464                 rc = nvdimm_security_erase(nvdimm, key, NVDIMM_USER);
 465         } else if (i == OP_OVERWRITE) {
 466                 dev_dbg(dev, "overwrite %u\n", key);
 467                 rc = nvdimm_security_overwrite(nvdimm, key);
 468         } else if (i == OP_MASTER_UPDATE) {
 469                 dev_dbg(dev, "master_update %u %u\n", key, newkey);
 470                 rc = nvdimm_security_update(nvdimm, key, newkey,
 471                                 NVDIMM_MASTER);
 472         } else if (i == OP_MASTER_ERASE) {
 473                 dev_dbg(dev, "master_erase %u\n", key);
 474                 rc = nvdimm_security_erase(nvdimm, key,
 475                                 NVDIMM_MASTER);
 476         } else
 477                 return -EINVAL;
 478
 479         if (rc == 0)
 480                 rc = len;
 481         return rc;
 482 }
 483
 484 static ssize_t security_store(struct device *dev,
 485                 struct device_attribute *attr, const char *buf, size_t len)
 486
 487 {
 488         ssize_t rc;
 489
 490         /*
 491          * Require all userspace triggered security management to be
 492          * done while probing is idle and the DIMM is not in active use
 493          * in any region.
 494          */
 495         device_lock(dev);
 496         nvdimm_bus_lock(dev);
 497         wait_nvdimm_bus_probe_idle(dev);
 498         rc = __security_store(dev, buf, len);
 499         nvdimm_bus_unlock(dev);
 500         device_unlock(dev);
 501
 502         return rc;
 503 }
 504 static DEVICE_ATTR_RW(security);
 505
 506 static struct attribute *nvdimm_attributes[] = {
 507         &dev_attr_state.attr,
 508         &dev_attr_flags.attr,
 509         &dev_attr_commands.attr,
 510         &dev_attr_available_slots.attr,
 511         &dev_attr_security.attr,
 512         NULL,
 513 };
 514
 515 static umode_t nvdimm_visible(struct kobject *kobj, struct attribute *a, int n)
 516 {
 517         struct device *dev = container_of(kobj, typeof(*dev), kobj);
 518         struct nvdimm *nvdimm = to_nvdimm(dev);
 519
 520         if (a != &dev_attr_security.attr)
 521                 return a->mode;
 522         if (nvdimm->sec.state < 0)
 523                 return 0;
 524         /* Are there any state mutation ops? */
 525         if (nvdimm->sec.ops->freeze || nvdimm->sec.ops->disable
 526                         || nvdimm->sec.ops->change_key
 527                         || nvdimm->sec.ops->erase
 528                         || nvdimm->sec.ops->overwrite)
 529                 return a->mode;
 530         return 0444;
 531 }
 532
 533 struct attribute_group nvdimm_attribute_group = {
 534         .attrs = nvdimm_attributes,
 535         .is_visible = nvdimm_visible,
 536 };
 537 EXPORT_SYMBOL_GPL(nvdimm_attribute_group);
 538
 539 struct nvdimm *__nvdimm_create(struct nvdimm_bus *nvdimm_bus,
 540                 void *provider_data, const struct attribute_group **groups,
 541                 unsigned long flags, unsigned long cmd_mask, int num_flush,
 542                 struct resource *flush_wpq, const char *dimm_id,
 543                 const struct nvdimm_security_ops *sec_ops)
 544 {
 545         struct nvdimm *nvdimm = kzalloc(sizeof(*nvdimm), GFP_KERNEL);
 546         struct device *dev;
 547
 548         if (!nvdimm)
 549                 return NULL;
 550
 551         nvdimm->id = ida_simple_get(&dimm_ida, 0, 0, GFP_KERNEL);
 552         if (nvdimm->id < 0) {
 553                 kfree(nvdimm);
 554                 return NULL;
 555         }
 556
 557         nvdimm->dimm_id = dimm_id;
 558         nvdimm->provider_data = provider_data;
 559         if (noblk)
 560                 flags |= 1 << NDD_NOBLK;
 561         nvdimm->flags = flags;
 562         nvdimm->cmd_mask = cmd_mask;
 563         nvdimm->num_flush = num_flush;
 564         nvdimm->flush_wpq = flush_wpq;
 565         atomic_set(&nvdimm->busy, 0);
 566         dev = &nvdimm->dev;
 567         dev_set_name(dev, "nmem%d", nvdimm->id);
 568         dev->parent = &nvdimm_bus->dev;
 569         dev->type = &nvdimm_device_type;
 570         dev->devt = MKDEV(nvdimm_major, nvdimm->id);
 571         dev->groups = groups;
 572         nvdimm->sec.ops = sec_ops;
 573         nvdimm->sec.overwrite_tmo = 0;
 574         INIT_DELAYED_WORK(&nvdimm->dwork, nvdimm_security_overwrite_query);
 575         /*
 576          * Security state must be initialized before device_add() for
 577          * attribute visibility.
 578          */
 579         /* get security state and extended (master) state */
 580         nvdimm->sec.state = nvdimm_security_state(nvdimm, NVDIMM_USER);
 581         nvdimm->sec.ext_state = nvdimm_security_state(nvdimm, NVDIMM_MASTER);
 582         nd_device_register(dev);
 583
 584         return nvdimm;
 585 }
 586 EXPORT_SYMBOL_GPL(__nvdimm_create);
 587
 588 static void shutdown_security_notify(void *data)
 589 {
 590         struct nvdimm *nvdimm = data;
 591
 592         sysfs_put(nvdimm->sec.overwrite_state);
 593 }
 594
 595 int nvdimm_security_setup_events(struct device *dev)
 596 {
 597         struct nvdimm *nvdimm = to_nvdimm(dev);
 598
 599         if (nvdimm->sec.state < 0 || !nvdimm->sec.ops
 600                         || !nvdimm->sec.ops->overwrite)
 601                 return 0;
 602         nvdimm->sec.overwrite_state = sysfs_get_dirent(dev->kobj.sd, "security");
 603         if (!nvdimm->sec.overwrite_state)
 604                 return -ENOMEM;
 605
 606         return devm_add_action_or_reset(dev, shutdown_security_notify, nvdimm);
 607 }
 608 EXPORT_SYMBOL_GPL(nvdimm_security_setup_events);
 609
 610 int nvdimm_in_overwrite(struct nvdimm *nvdimm)
 611 {
 612         return test_bit(NDD_SECURITY_OVERWRITE, &nvdimm->flags);
 613 }
 614 EXPORT_SYMBOL_GPL(nvdimm_in_overwrite);
 615
 616 int nvdimm_security_freeze(struct nvdimm *nvdimm)
 617 {
 618         int rc;
 619
 620         WARN_ON_ONCE(!is_nvdimm_bus_locked(&nvdimm->dev));
 621
 622         if (!nvdimm->sec.ops || !nvdimm->sec.ops->freeze)
 623                 return -EOPNOTSUPP;
 624
 625         if (nvdimm->sec.state < 0)
 626                 return -EIO;
 627
 628         if (test_bit(NDD_SECURITY_OVERWRITE, &nvdimm->flags)) {
 629                 dev_warn(&nvdimm->dev, "Overwrite operation in progress.\n");
 630                 return -EBUSY;
 631         }
 632
 633         rc = nvdimm->sec.ops->freeze(nvdimm);
 634         nvdimm->sec.state = nvdimm_security_state(nvdimm, NVDIMM_USER);
 635
 636         return rc;
 637 }
 638
 639 int alias_dpa_busy(struct device *dev, void *data)
 640 {
 641         resource_size_t map_end, blk_start, new;
 642         struct blk_alloc_info *info = data;
 643         struct nd_mapping *nd_mapping;
 644         struct nd_region *nd_region;
 645         struct nvdimm_drvdata *ndd;
 646         struct resource *res;
 647         int i;
 648
 649         if (!is_memory(dev))
 650                 return 0;
 651
 652         nd_region = to_nd_region(dev);
 653         for (i = 0; i < nd_region->ndr_mappings; i++) {
 654                 nd_mapping  = &nd_region->mapping[i];
 655                 if (nd_mapping->nvdimm == info->nd_mapping->nvdimm)
 656                         break;
 657         }
 658
 659         if (i >= nd_region->ndr_mappings)
 660                 return 0;
 661
 662         ndd = to_ndd(nd_mapping);
 663         map_end = nd_mapping->start + nd_mapping->size - 1;
 664         blk_start = nd_mapping->start;
 665
 666         /*
 667          * In the allocation case ->res is set to free space that we are
 668          * looking to validate against PMEM aliasing collision rules
 669          * (i.e. BLK is allocated after all aliased PMEM).
 670          */
 671         if (info->res) {
 672                 if (info->res->start >= nd_mapping->start
 673                                 && info->res->start < map_end)
 674                         /* pass */;
 675                 else
 676                         return 0;
 677         }
 678
 679  retry:
 680         /*
 681          * Find the free dpa from the end of the last pmem allocation to
 682          * the end of the interleave-set mapping.
 683          */
 684         for_each_dpa_resource(ndd, res) {
 685                 if (strncmp(res->name, "pmem", 4) != 0)
 686                         continue;
 687                 if ((res->start >= blk_start && res->start < map_end)
 688                                 || (res->end >= blk_start
 689                                         && res->end <= map_end)) {
 690                         new = max(blk_start, min(map_end + 1, res->end + 1));
 691                         if (new != blk_start) {
 692                                 blk_start = new;
 693                                 goto retry;
 694                         }
 695                 }
 696         }
 697
 698         /* update the free space range with the probed blk_start */
 699         if (info->res && blk_start > info->res->start) {
 700                 info->res->start = max(info->res->start, blk_start);
 701                 if (info->res->start > info->res->end)
 702                         info->res->end = info->res->start - 1;
 703                 return 1;
 704         }
 705
 706         info->available -= blk_start - nd_mapping->start;
 707
 708         return 0;
 709 }
 710
 711 /**
 712  * nd_blk_available_dpa - account the unused dpa of BLK region
 713  * @nd_mapping: container of dpa-resource-root + labels
 714  *
 715  * Unlike PMEM, BLK namespaces can occupy discontiguous DPA ranges, but
 716  * we arrange for them to never start at an lower dpa than the last
 717  * PMEM allocation in an aliased region.
 718  */
 719 resource_size_t nd_blk_available_dpa(struct nd_region *nd_region)
 720 {
 721         struct nvdimm_bus *nvdimm_bus = walk_to_nvdimm_bus(&nd_region->dev);
 722         struct nd_mapping *nd_mapping = &nd_region->mapping[0];
 723         struct nvdimm_drvdata *ndd = to_ndd(nd_mapping);
 724         struct blk_alloc_info info = {
 725                 .nd_mapping = nd_mapping,
 726                 .available = nd_mapping->size,
 727                 .res = NULL,
 728         };
 729         struct resource *res;
 730
 731         if (!ndd)
 732                 return 0;
 733
 734         device_for_each_child(&nvdimm_bus->dev, &info, alias_dpa_busy);
 735
 736         /* now account for busy blk allocations in unaliased dpa */
 737         for_each_dpa_resource(ndd, res) {
 738                 if (strncmp(res->name, "blk", 3) != 0)
 739                         continue;
 740                 info.available -= resource_size(res);
 741         }
 742
 743         return info.available;
 744 }
 745
 746 /**
 747  * nd_pmem_max_contiguous_dpa - For the given dimm+region, return the max
 748  *                         contiguous unallocated dpa range.
 749  * @nd_region: constrain available space check to this reference region
 750  * @nd_mapping: container of dpa-resource-root + labels
 751  */
 752 resource_size_t nd_pmem_max_contiguous_dpa(struct nd_region *nd_region,
 753                                            struct nd_mapping *nd_mapping)
 754 {
 755         struct nvdimm_drvdata *ndd = to_ndd(nd_mapping);
 756         struct nvdimm_bus *nvdimm_bus;
 757         resource_size_t max = 0;
 758         struct resource *res;
 759
 760         /* if a dimm is disabled the available capacity is zero */
 761         if (!ndd)
 762                 return 0;
 763
 764         nvdimm_bus = walk_to_nvdimm_bus(ndd->dev);
 765         if (__reserve_free_pmem(&nd_region->dev, nd_mapping->nvdimm))
 766                 return 0;
 767         for_each_dpa_resource(ndd, res) {
 768                 if (strcmp(res->name, "pmem-reserve") != 0)
 769                         continue;
 770                 if (resource_size(res) > max)
 771                         max = resource_size(res);
 772         }
 773         release_free_pmem(nvdimm_bus, nd_mapping);
 774         return max;
 775 }
 776
 777 /**
 778  * nd_pmem_available_dpa - for the given dimm+region account unallocated dpa
 779  * @nd_mapping: container of dpa-resource-root + labels
 780  * @nd_region: constrain available space check to this reference region
 781  * @overlap: calculate available space assuming this level of overlap
 782  *
 783  * Validate that a PMEM label, if present, aligns with the start of an
 784  * interleave set and truncate the available size at the lowest BLK
 785  * overlap point.
 786  *
 787  * The expectation is that this routine is called multiple times as it
 788  * probes for the largest BLK encroachment for any single member DIMM of
 789  * the interleave set.  Once that value is determined the PMEM-limit for
 790  * the set can be established.
 791  */
 792 resource_size_t nd_pmem_available_dpa(struct nd_region *nd_region,
 793                 struct nd_mapping *nd_mapping, resource_size_t *overlap)
 794 {
 795         resource_size_t map_start, map_end, busy = 0, available, blk_start;
 796         struct nvdimm_drvdata *ndd = to_ndd(nd_mapping);
 797         struct resource *res;
 798         const char *reason;
 799
 800         if (!ndd)
 801                 return 0;
 802
 803         map_start = nd_mapping->start;
 804         map_end = map_start + nd_mapping->size - 1;
 805         blk_start = max(map_start, map_end + 1 - *overlap);
 806         for_each_dpa_resource(ndd, res) {
 807                 if (res->start >= map_start && res->start < map_end) {
 808                         if (strncmp(res->name, "blk", 3) == 0)
 809                                 blk_start = min(blk_start,
 810                                                 max(map_start, res->start));
 811                         else if (res->end > map_end) {
 812                                 reason = "misaligned to iset";
 813                                 goto err;
 814                         } else
 815                                 busy += resource_size(res);
 816                 } else if (res->end >= map_start && res->end <= map_end) {
 817                         if (strncmp(res->name, "blk", 3) == 0) {
 818                                 /*
 819                                  * If a BLK allocation overlaps the start of
 820                                  * PMEM the entire interleave set may now only
 821                                  * be used for BLK.
 822                                  */
 823                                 blk_start = map_start;
 824                         } else
 825                                 busy += resource_size(res);
 826                 } else if (map_start > res->start && map_start < res->end) {
 827                         /* total eclipse of the mapping */
 828                         busy += nd_mapping->size;
 829                         blk_start = map_start;
 830                 }
 831         }
 832
 833         *overlap = map_end + 1 - blk_start;
 834         available = blk_start - map_start;
 835         if (busy < available)
 836                 return available - busy;
 837         return 0;
 838
 839  err:
 840         nd_dbg_dpa(nd_region, ndd, res, "%s\n", reason);
 841         return 0;
 842 }
 843
 844 void nvdimm_free_dpa(struct nvdimm_drvdata *ndd, struct resource *res)
 845 {
 846         WARN_ON_ONCE(!is_nvdimm_bus_locked(ndd->dev));
 847         kfree(res->name);
 848         __release_region(&ndd->dpa, res->start, resource_size(res));
 849 }
 850
 851 struct resource *nvdimm_allocate_dpa(struct nvdimm_drvdata *ndd,
 852                 struct nd_label_id *label_id, resource_size_t start,
 853                 resource_size_t n)
 854 {
 855         char *name = kmemdup(label_id, sizeof(*label_id), GFP_KERNEL);
 856         struct resource *res;
 857
 858         if (!name)
 859                 return NULL;
 860
 861         WARN_ON_ONCE(!is_nvdimm_bus_locked(ndd->dev));
 862         res = __request_region(&ndd->dpa, start, n, name, 0);
 863         if (!res)
 864                 kfree(name);
 865         return res;
 866 }
 867
 868 /**
 869  * nvdimm_allocated_dpa - sum up the dpa currently allocated to this label_id
 870  * @nvdimm: container of dpa-resource-root + labels
 871  * @label_id: dpa resource name of the form {pmem|blk}-<human readable uuid>
 872  */
 873 resource_size_t nvdimm_allocated_dpa(struct nvdimm_drvdata *ndd,
 874                 struct nd_label_id *label_id)
 875 {
 876         resource_size_t allocated = 0;
 877         struct resource *res;
 878
 879         for_each_dpa_resource(ndd, res)
 880                 if (strcmp(res->name, label_id->id) == 0)
 881                         allocated += resource_size(res);
 882
 883         return allocated;
 884 }
 885
 886 static int count_dimms(struct device *dev, void *c)
 887 {
 888         int *count = c;
 889
 890         if (is_nvdimm(dev))
 891                 (*count)++;
 892         return 0;
 893 }
 894
 895 int nvdimm_bus_check_dimm_count(struct nvdimm_bus *nvdimm_bus, int dimm_count)
 896 {
 897         int count = 0;
 898         /* Flush any possible dimm registration failures */
 899         nd_synchronize();
 900
 901         device_for_each_child(&nvdimm_bus->dev, &count, count_dimms);
 902         dev_dbg(&nvdimm_bus->dev, "count: %d\n", count);
 903         if (count != dimm_count)
 904                 return -ENXIO;
 905         return 0;
 906 }
 907 EXPORT_SYMBOL_GPL(nvdimm_bus_check_dimm_count);
 908
 909 void __exit nvdimm_devs_exit(void)
 910 {
 911         ida_destroy(&dimm_ida);
 912 }