drivers/nvdimm/namespace_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 #include <linux/module.h>
  14 #include <linux/device.h>
  15 #include <linux/sort.h>
  16 #include <linux/slab.h>
  17 #include <linux/pmem.h>
  18 #include <linux/list.h>
  19 #include <linux/nd.h>
  20 #include "nd-core.h"
  21 #include "nd.h"
  22
  23 static void namespace_io_release(struct device *dev)
  24 {
  25         struct nd_namespace_io *nsio = to_nd_namespace_io(dev);
  26
  27         kfree(nsio);
  28 }
  29
  30 static void namespace_pmem_release(struct device *dev)
  31 {
  32         struct nd_namespace_pmem *nspm = to_nd_namespace_pmem(dev);
  33         struct nd_region *nd_region = to_nd_region(dev->parent);
  34
  35         if (nspm->id >= 0)
  36                 ida_simple_remove(&nd_region->ns_ida, nspm->id);
  37         kfree(nspm->alt_name);
  38         kfree(nspm->uuid);
  39         kfree(nspm);
  40 }
  41
  42 static void namespace_blk_release(struct device *dev)
  43 {
  44         struct nd_namespace_blk *nsblk = to_nd_namespace_blk(dev);
  45         struct nd_region *nd_region = to_nd_region(dev->parent);
  46
  47         if (nsblk->id >= 0)
  48                 ida_simple_remove(&nd_region->ns_ida, nsblk->id);
  49         kfree(nsblk->alt_name);
  50         kfree(nsblk->uuid);
  51         kfree(nsblk->res);
  52         kfree(nsblk);
  53 }
  54
  55 static const struct device_type namespace_io_device_type = {
  56         .name = "nd_namespace_io",
  57         .release = namespace_io_release,
  58 };
  59
  60 static const struct device_type namespace_pmem_device_type = {
  61         .name = "nd_namespace_pmem",
  62         .release = namespace_pmem_release,
  63 };
  64
  65 static const struct device_type namespace_blk_device_type = {
  66         .name = "nd_namespace_blk",
  67         .release = namespace_blk_release,
  68 };
  69
  70 static bool is_namespace_pmem(const struct device *dev)
  71 {
  72         return dev ? dev->type == &namespace_pmem_device_type : false;
  73 }
  74
  75 static bool is_namespace_blk(const struct device *dev)
  76 {
  77         return dev ? dev->type == &namespace_blk_device_type : false;
  78 }
  79
  80 static bool is_namespace_io(const struct device *dev)
  81 {
  82         return dev ? dev->type == &namespace_io_device_type : false;
  83 }
  84
  85 static int is_uuid_busy(struct device *dev, void *data)
  86 {
  87         u8 *uuid1 = data, *uuid2 = NULL;
  88
  89         if (is_namespace_pmem(dev)) {
  90                 struct nd_namespace_pmem *nspm = to_nd_namespace_pmem(dev);
  91
  92                 uuid2 = nspm->uuid;
  93         } else if (is_namespace_blk(dev)) {
  94                 struct nd_namespace_blk *nsblk = to_nd_namespace_blk(dev);
  95
  96                 uuid2 = nsblk->uuid;
  97         } else if (is_nd_btt(dev)) {
  98                 struct nd_btt *nd_btt = to_nd_btt(dev);
  99
 100                 uuid2 = nd_btt->uuid;
 101         } else if (is_nd_pfn(dev)) {
 102                 struct nd_pfn *nd_pfn = to_nd_pfn(dev);
 103
 104                 uuid2 = nd_pfn->uuid;
 105         }
 106
 107         if (uuid2 && memcmp(uuid1, uuid2, NSLABEL_UUID_LEN) == 0)
 108                 return -EBUSY;
 109
 110         return 0;
 111 }
 112
 113 static int is_namespace_uuid_busy(struct device *dev, void *data)
 114 {
 115         if (is_nd_pmem(dev) || is_nd_blk(dev))
 116                 return device_for_each_child(dev, data, is_uuid_busy);
 117         return 0;
 118 }
 119
 120 /**
 121  * nd_is_uuid_unique - verify that no other namespace has @uuid
 122  * @dev: any device on a nvdimm_bus
 123  * @uuid: uuid to check
 124  */
 125 bool nd_is_uuid_unique(struct device *dev, u8 *uuid)
 126 {
 127         struct nvdimm_bus *nvdimm_bus = walk_to_nvdimm_bus(dev);
 128
 129         if (!nvdimm_bus)
 130                 return false;
 131         WARN_ON_ONCE(!is_nvdimm_bus_locked(&nvdimm_bus->dev));
 132         if (device_for_each_child(&nvdimm_bus->dev, uuid,
 133                                 is_namespace_uuid_busy) != 0)
 134                 return false;
 135         return true;
 136 }
 137
 138 bool pmem_should_map_pages(struct device *dev)
 139 {
 140         struct nd_region *nd_region = to_nd_region(dev->parent);
 141         struct nd_namespace_io *nsio;
 142
 143         if (!IS_ENABLED(CONFIG_ZONE_DEVICE))
 144                 return false;
 145
 146         if (!test_bit(ND_REGION_PAGEMAP, &nd_region->flags))
 147                 return false;
 148
 149         if (is_nd_pfn(dev) || is_nd_btt(dev))
 150                 return false;
 151
 152         nsio = to_nd_namespace_io(dev);
 153         if (region_intersects(nsio->res.start, resource_size(&nsio->res),
 154                                 IORESOURCE_SYSTEM_RAM,
 155                                 IORES_DESC_NONE) == REGION_MIXED)
 156                 return false;
 157
 158 #ifdef ARCH_MEMREMAP_PMEM
 159         return ARCH_MEMREMAP_PMEM == MEMREMAP_WB;
 160 #else
 161         return false;
 162 #endif
 163 }
 164 EXPORT_SYMBOL(pmem_should_map_pages);
 165
 166 const char *nvdimm_namespace_disk_name(struct nd_namespace_common *ndns,
 167                 char *name)
 168 {
 169         struct nd_region *nd_region = to_nd_region(ndns->dev.parent);
 170         const char *suffix = NULL;
 171
 172         if (ndns->claim && is_nd_btt(ndns->claim))
 173                 suffix = "s";
 174
 175         if (is_namespace_pmem(&ndns->dev) || is_namespace_io(&ndns->dev)) {
 176                 int nsidx = 0;
 177
 178                 if (is_namespace_pmem(&ndns->dev)) {
 179                         struct nd_namespace_pmem *nspm;
 180
 181                         nspm = to_nd_namespace_pmem(&ndns->dev);
 182                         nsidx = nspm->id;
 183                 }
 184
 185                 if (nsidx)
 186                         sprintf(name, "pmem%d.%d%s", nd_region->id, nsidx,
 187                                         suffix ? suffix : "");
 188                 else
 189                         sprintf(name, "pmem%d%s", nd_region->id,
 190                                         suffix ? suffix : "");
 191         } else if (is_namespace_blk(&ndns->dev)) {
 192                 struct nd_namespace_blk *nsblk;
 193
 194                 nsblk = to_nd_namespace_blk(&ndns->dev);
 195                 sprintf(name, "ndblk%d.%d%s", nd_region->id, nsblk->id,
 196                                 suffix ? suffix : "");
 197         } else {
 198                 return NULL;
 199         }
 200
 201         return name;
 202 }
 203 EXPORT_SYMBOL(nvdimm_namespace_disk_name);
 204
 205 const u8 *nd_dev_to_uuid(struct device *dev)
 206 {
 207         static const u8 null_uuid[16];
 208
 209         if (!dev)
 210                 return null_uuid;
 211
 212         if (is_namespace_pmem(dev)) {
 213                 struct nd_namespace_pmem *nspm = to_nd_namespace_pmem(dev);
 214
 215                 return nspm->uuid;
 216         } else if (is_namespace_blk(dev)) {
 217                 struct nd_namespace_blk *nsblk = to_nd_namespace_blk(dev);
 218
 219                 return nsblk->uuid;
 220         } else
 221                 return null_uuid;
 222 }
 223 EXPORT_SYMBOL(nd_dev_to_uuid);
 224
 225 static ssize_t nstype_show(struct device *dev,
 226                 struct device_attribute *attr, char *buf)
 227 {
 228         struct nd_region *nd_region = to_nd_region(dev->parent);
 229
 230         return sprintf(buf, "%d\n", nd_region_to_nstype(nd_region));
 231 }
 232 static DEVICE_ATTR_RO(nstype);
 233
 234 static ssize_t __alt_name_store(struct device *dev, const char *buf,
 235                 const size_t len)
 236 {
 237         char *input, *pos, *alt_name, **ns_altname;
 238         ssize_t rc;
 239
 240         if (is_namespace_pmem(dev)) {
 241                 struct nd_namespace_pmem *nspm = to_nd_namespace_pmem(dev);
 242
 243                 ns_altname = &nspm->alt_name;
 244         } else if (is_namespace_blk(dev)) {
 245                 struct nd_namespace_blk *nsblk = to_nd_namespace_blk(dev);
 246
 247                 ns_altname = &nsblk->alt_name;
 248         } else
 249                 return -ENXIO;
 250
 251         if (dev->driver || to_ndns(dev)->claim)
 252                 return -EBUSY;
 253
 254         input = kmemdup(buf, len + 1, GFP_KERNEL);
 255         if (!input)
 256                 return -ENOMEM;
 257
 258         input[len] = '\0';
 259         pos = strim(input);
 260         if (strlen(pos) + 1 > NSLABEL_NAME_LEN) {
 261                 rc = -EINVAL;
 262                 goto out;
 263         }
 264
 265         alt_name = kzalloc(NSLABEL_NAME_LEN, GFP_KERNEL);
 266         if (!alt_name) {
 267                 rc = -ENOMEM;
 268                 goto out;
 269         }
 270         kfree(*ns_altname);
 271         *ns_altname = alt_name;
 272         sprintf(*ns_altname, "%s", pos);
 273         rc = len;
 274
 275 out:
 276         kfree(input);
 277         return rc;
 278 }
 279
 280 static resource_size_t nd_namespace_blk_size(struct nd_namespace_blk *nsblk)
 281 {
 282         struct nd_region *nd_region = to_nd_region(nsblk->common.dev.parent);
 283         struct nd_mapping *nd_mapping = &nd_region->mapping[0];
 284         struct nvdimm_drvdata *ndd = to_ndd(nd_mapping);
 285         struct nd_label_id label_id;
 286         resource_size_t size = 0;
 287         struct resource *res;
 288
 289         if (!nsblk->uuid)
 290                 return 0;
 291         nd_label_gen_id(&label_id, nsblk->uuid, NSLABEL_FLAG_LOCAL);
 292         for_each_dpa_resource(ndd, res)
 293                 if (strcmp(res->name, label_id.id) == 0)
 294                         size += resource_size(res);
 295         return size;
 296 }
 297
 298 static bool __nd_namespace_blk_validate(struct nd_namespace_blk *nsblk)
 299 {
 300         struct nd_region *nd_region = to_nd_region(nsblk->common.dev.parent);
 301         struct nd_mapping *nd_mapping = &nd_region->mapping[0];
 302         struct nvdimm_drvdata *ndd = to_ndd(nd_mapping);
 303         struct nd_label_id label_id;
 304         struct resource *res;
 305         int count, i;
 306
 307         if (!nsblk->uuid || !nsblk->lbasize || !ndd)
 308                 return false;
 309
 310         count = 0;
 311         nd_label_gen_id(&label_id, nsblk->uuid, NSLABEL_FLAG_LOCAL);
 312         for_each_dpa_resource(ndd, res) {
 313                 if (strcmp(res->name, label_id.id) != 0)
 314                         continue;
 315                 /*
 316                  * Resources with unacknowledged adjustments indicate a
 317                  * failure to update labels
 318                  */
 319                 if (res->flags & DPA_RESOURCE_ADJUSTED)
 320                         return false;
 321                 count++;
 322         }
 323
 324         /* These values match after a successful label update */
 325         if (count != nsblk->num_resources)
 326                 return false;
 327
 328         for (i = 0; i < nsblk->num_resources; i++) {
 329                 struct resource *found = NULL;
 330
 331                 for_each_dpa_resource(ndd, res)
 332                         if (res == nsblk->res[i]) {
 333                                 found = res;
 334                                 break;
 335                         }
 336                 /* stale resource */
 337                 if (!found)
 338                         return false;
 339         }
 340
 341         return true;
 342 }
 343
 344 resource_size_t nd_namespace_blk_validate(struct nd_namespace_blk *nsblk)
 345 {
 346         resource_size_t size;
 347
 348         nvdimm_bus_lock(&nsblk->common.dev);
 349         size = __nd_namespace_blk_validate(nsblk);
 350         nvdimm_bus_unlock(&nsblk->common.dev);
 351
 352         return size;
 353 }
 354 EXPORT_SYMBOL(nd_namespace_blk_validate);
 355
 356
 357 static int nd_namespace_label_update(struct nd_region *nd_region,
 358                 struct device *dev)
 359 {
 360         dev_WARN_ONCE(dev, dev->driver || to_ndns(dev)->claim,
 361                         "namespace must be idle during label update\n");
 362         if (dev->driver || to_ndns(dev)->claim)
 363                 return 0;
 364
 365         /*
 366          * Only allow label writes that will result in a valid namespace
 367          * or deletion of an existing namespace.
 368          */
 369         if (is_namespace_pmem(dev)) {
 370                 struct nd_namespace_pmem *nspm = to_nd_namespace_pmem(dev);
 371                 resource_size_t size = resource_size(&nspm->nsio.res);
 372
 373                 if (size == 0 && nspm->uuid)
 374                         /* delete allocation */;
 375                 else if (!nspm->uuid)
 376                         return 0;
 377
 378                 return nd_pmem_namespace_label_update(nd_region, nspm, size);
 379         } else if (is_namespace_blk(dev)) {
 380                 struct nd_namespace_blk *nsblk = to_nd_namespace_blk(dev);
 381                 resource_size_t size = nd_namespace_blk_size(nsblk);
 382
 383                 if (size == 0 && nsblk->uuid)
 384                         /* delete allocation */;
 385                 else if (!nsblk->uuid || !nsblk->lbasize)
 386                         return 0;
 387
 388                 return nd_blk_namespace_label_update(nd_region, nsblk, size);
 389         } else
 390                 return -ENXIO;
 391 }
 392
 393 static ssize_t alt_name_store(struct device *dev,
 394                 struct device_attribute *attr, const char *buf, size_t len)
 395 {
 396         struct nd_region *nd_region = to_nd_region(dev->parent);
 397         ssize_t rc;
 398
 399         device_lock(dev);
 400         nvdimm_bus_lock(dev);
 401         wait_nvdimm_bus_probe_idle(dev);
 402         rc = __alt_name_store(dev, buf, len);
 403         if (rc >= 0)
 404                 rc = nd_namespace_label_update(nd_region, dev);
 405         dev_dbg(dev, "%s: %s(%zd)\n", __func__, rc < 0 ? "fail " : "", rc);
 406         nvdimm_bus_unlock(dev);
 407         device_unlock(dev);
 408
 409         return rc < 0 ? rc : len;
 410 }
 411
 412 static ssize_t alt_name_show(struct device *dev,
 413                 struct device_attribute *attr, char *buf)
 414 {
 415         char *ns_altname;
 416
 417         if (is_namespace_pmem(dev)) {
 418                 struct nd_namespace_pmem *nspm = to_nd_namespace_pmem(dev);
 419
 420                 ns_altname = nspm->alt_name;
 421         } else if (is_namespace_blk(dev)) {
 422                 struct nd_namespace_blk *nsblk = to_nd_namespace_blk(dev);
 423
 424                 ns_altname = nsblk->alt_name;
 425         } else
 426                 return -ENXIO;
 427
 428         return sprintf(buf, "%s\n", ns_altname ? ns_altname : "");
 429 }
 430 static DEVICE_ATTR_RW(alt_name);
 431
 432 static int scan_free(struct nd_region *nd_region,
 433                 struct nd_mapping *nd_mapping, struct nd_label_id *label_id,
 434                 resource_size_t n)
 435 {
 436         bool is_blk = strncmp(label_id->id, "blk", 3) == 0;
 437         struct nvdimm_drvdata *ndd = to_ndd(nd_mapping);
 438         int rc = 0;
 439
 440         while (n) {
 441                 struct resource *res, *last;
 442                 resource_size_t new_start;
 443
 444                 last = NULL;
 445                 for_each_dpa_resource(ndd, res)
 446                         if (strcmp(res->name, label_id->id) == 0)
 447                                 last = res;
 448                 res = last;
 449                 if (!res)
 450                         return 0;
 451
 452                 if (n >= resource_size(res)) {
 453                         n -= resource_size(res);
 454                         nd_dbg_dpa(nd_region, ndd, res, "delete %d\n", rc);
 455                         nvdimm_free_dpa(ndd, res);
 456                         /* retry with last resource deleted */
 457                         continue;
 458                 }
 459
 460                 /*
 461                  * Keep BLK allocations relegated to high DPA as much as
 462                  * possible
 463                  */
 464                 if (is_blk)
 465                         new_start = res->start + n;
 466                 else
 467                         new_start = res->start;
 468
 469                 rc = adjust_resource(res, new_start, resource_size(res) - n);
 470                 if (rc == 0)
 471                         res->flags |= DPA_RESOURCE_ADJUSTED;
 472                 nd_dbg_dpa(nd_region, ndd, res, "shrink %d\n", rc);
 473                 break;
 474         }
 475
 476         return rc;
 477 }
 478
 479 /**
 480  * shrink_dpa_allocation - for each dimm in region free n bytes for label_id
 481  * @nd_region: the set of dimms to reclaim @n bytes from
 482  * @label_id: unique identifier for the namespace consuming this dpa range
 483  * @n: number of bytes per-dimm to release
 484  *
 485  * Assumes resources are ordered.  Starting from the end try to
 486  * adjust_resource() the allocation to @n, but if @n is larger than the
 487  * allocation delete it and find the 'new' last allocation in the label
 488  * set.
 489  */
 490 static int shrink_dpa_allocation(struct nd_region *nd_region,
 491                 struct nd_label_id *label_id, resource_size_t n)
 492 {
 493         int i;
 494
 495         for (i = 0; i < nd_region->ndr_mappings; i++) {
 496                 struct nd_mapping *nd_mapping = &nd_region->mapping[i];
 497                 int rc;
 498
 499                 rc = scan_free(nd_region, nd_mapping, label_id, n);
 500                 if (rc)
 501                         return rc;
 502         }
 503
 504         return 0;
 505 }
 506
 507 static resource_size_t init_dpa_allocation(struct nd_label_id *label_id,
 508                 struct nd_region *nd_region, struct nd_mapping *nd_mapping,
 509                 resource_size_t n)
 510 {
 511         bool is_blk = strncmp(label_id->id, "blk", 3) == 0;
 512         struct nvdimm_drvdata *ndd = to_ndd(nd_mapping);
 513         resource_size_t first_dpa;
 514         struct resource *res;
 515         int rc = 0;
 516
 517         /* allocate blk from highest dpa first */
 518         if (is_blk)
 519                 first_dpa = nd_mapping->start + nd_mapping->size - n;
 520         else
 521                 first_dpa = nd_mapping->start;
 522
 523         /* first resource allocation for this label-id or dimm */
 524         res = nvdimm_allocate_dpa(ndd, label_id, first_dpa, n);
 525         if (!res)
 526                 rc = -EBUSY;
 527
 528         nd_dbg_dpa(nd_region, ndd, res, "init %d\n", rc);
 529         return rc ? n : 0;
 530 }
 531
 532
 533 /**
 534  * space_valid() - validate free dpa space against constraints
 535  * @nd_region: hosting region of the free space
 536  * @ndd: dimm device data for debug
 537  * @label_id: namespace id to allocate space
 538  * @prev: potential allocation that precedes free space
 539  * @next: allocation that follows the given free space range
 540  * @exist: first allocation with same id in the mapping
 541  * @n: range that must satisfied for pmem allocations
 542  * @valid: free space range to validate
 543  *
 544  * BLK-space is valid as long as it does not precede a PMEM
 545  * allocation in a given region. PMEM-space must be contiguous
 546  * and adjacent to an existing existing allocation (if one
 547  * exists).  If reserving PMEM any space is valid.
 548  */
 549 static void space_valid(struct nd_region *nd_region, struct nvdimm_drvdata *ndd,
 550                 struct nd_label_id *label_id, struct resource *prev,
 551                 struct resource *next, struct resource *exist,
 552                 resource_size_t n, struct resource *valid)
 553 {
 554         bool is_reserve = strcmp(label_id->id, "pmem-reserve") == 0;
 555         bool is_pmem = strncmp(label_id->id, "pmem", 4) == 0;
 556
 557         if (valid->start >= valid->end)
 558                 goto invalid;
 559
 560         if (is_reserve)
 561                 return;
 562
 563         if (!is_pmem) {
 564                 struct nd_mapping *nd_mapping = &nd_region->mapping[0];
 565                 struct nvdimm_bus *nvdimm_bus;
 566                 struct blk_alloc_info info = {
 567                         .nd_mapping = nd_mapping,
 568                         .available = nd_mapping->size,
 569                         .res = valid,
 570                 };
 571
 572                 WARN_ON(!is_nd_blk(&nd_region->dev));
 573                 nvdimm_bus = walk_to_nvdimm_bus(&nd_region->dev);
 574                 device_for_each_child(&nvdimm_bus->dev, &info, alias_dpa_busy);
 575                 return;
 576         }
 577
 578         /* allocation needs to be contiguous, so this is all or nothing */
 579         if (resource_size(valid) < n)
 580                 goto invalid;
 581
 582         /* we've got all the space we need and no existing allocation */
 583         if (!exist)
 584                 return;
 585
 586         /* allocation needs to be contiguous with the existing namespace */
 587         if (valid->start == exist->end + 1
 588                         || valid->end == exist->start - 1)
 589                 return;
 590
 591  invalid:
 592         /* truncate @valid size to 0 */
 593         valid->end = valid->start - 1;
 594 }
 595
 596 enum alloc_loc {
 597         ALLOC_ERR = 0, ALLOC_BEFORE, ALLOC_MID, ALLOC_AFTER,
 598 };
 599
 600 static resource_size_t scan_allocate(struct nd_region *nd_region,
 601                 struct nd_mapping *nd_mapping, struct nd_label_id *label_id,
 602                 resource_size_t n)
 603 {
 604         resource_size_t mapping_end = nd_mapping->start + nd_mapping->size - 1;
 605         bool is_pmem = strncmp(label_id->id, "pmem", 4) == 0;
 606         struct nvdimm_drvdata *ndd = to_ndd(nd_mapping);
 607         struct resource *res, *exist = NULL, valid;
 608         const resource_size_t to_allocate = n;
 609         int first;
 610
 611         for_each_dpa_resource(ndd, res)
 612                 if (strcmp(label_id->id, res->name) == 0)
 613                         exist = res;
 614
 615         valid.start = nd_mapping->start;
 616         valid.end = mapping_end;
 617         valid.name = "free space";
 618  retry:
 619         first = 0;
 620         for_each_dpa_resource(ndd, res) {
 621                 struct resource *next = res->sibling, *new_res = NULL;
 622                 resource_size_t allocate, available = 0;
 623                 enum alloc_loc loc = ALLOC_ERR;
 624                 const char *action;
 625                 int rc = 0;
 626
 627                 /* ignore resources outside this nd_mapping */
 628                 if (res->start > mapping_end)
 629                         continue;
 630                 if (res->end < nd_mapping->start)
 631                         continue;
 632
 633                 /* space at the beginning of the mapping */
 634                 if (!first++ && res->start > nd_mapping->start) {
 635                         valid.start = nd_mapping->start;
 636                         valid.end = res->start - 1;
 637                         space_valid(nd_region, ndd, label_id, NULL, next, exist,
 638                                         to_allocate, &valid);
 639                         available = resource_size(&valid);
 640                         if (available)
 641                                 loc = ALLOC_BEFORE;
 642                 }
 643
 644                 /* space between allocations */
 645                 if (!loc && next) {
 646                         valid.start = res->start + resource_size(res);
 647                         valid.end = min(mapping_end, next->start - 1);
 648                         space_valid(nd_region, ndd, label_id, res, next, exist,
 649                                         to_allocate, &valid);
 650                         available = resource_size(&valid);
 651                         if (available)
 652                                 loc = ALLOC_MID;
 653                 }
 654
 655                 /* space at the end of the mapping */
 656                 if (!loc && !next) {
 657                         valid.start = res->start + resource_size(res);
 658                         valid.end = mapping_end;
 659                         space_valid(nd_region, ndd, label_id, res, next, exist,
 660                                         to_allocate, &valid);
 661                         available = resource_size(&valid);
 662                         if (available)
 663                                 loc = ALLOC_AFTER;
 664                 }
 665
 666                 if (!loc || !available)
 667                         continue;
 668                 allocate = min(available, n);
 669                 switch (loc) {
 670                 case ALLOC_BEFORE:
 671                         if (strcmp(res->name, label_id->id) == 0) {
 672                                 /* adjust current resource up */
 673                                 rc = adjust_resource(res, res->start - allocate,
 674                                                 resource_size(res) + allocate);
 675                                 action = "cur grow up";
 676                         } else
 677                                 action = "allocate";
 678                         break;
 679                 case ALLOC_MID:
 680                         if (strcmp(next->name, label_id->id) == 0) {
 681                                 /* adjust next resource up */
 682                                 rc = adjust_resource(next, next->start
 683                                                 - allocate, resource_size(next)
 684                                                 + allocate);
 685                                 new_res = next;
 686                                 action = "next grow up";
 687                         } else if (strcmp(res->name, label_id->id) == 0) {
 688                                 action = "grow down";
 689                         } else
 690                                 action = "allocate";
 691                         break;
 692                 case ALLOC_AFTER:
 693                         if (strcmp(res->name, label_id->id) == 0)
 694                                 action = "grow down";
 695                         else
 696                                 action = "allocate";
 697                         break;
 698                 default:
 699                         return n;
 700                 }
 701
 702                 if (strcmp(action, "allocate") == 0) {
 703                         /* BLK allocate bottom up */
 704                         if (!is_pmem)
 705                                 valid.start += available - allocate;
 706
 707                         new_res = nvdimm_allocate_dpa(ndd, label_id,
 708                                         valid.start, allocate);
 709                         if (!new_res)
 710                                 rc = -EBUSY;
 711                 } else if (strcmp(action, "grow down") == 0) {
 712                         /* adjust current resource down */
 713                         rc = adjust_resource(res, res->start, resource_size(res)
 714                                         + allocate);
 715                         if (rc == 0)
 716                                 res->flags |= DPA_RESOURCE_ADJUSTED;
 717                 }
 718
 719                 if (!new_res)
 720                         new_res = res;
 721
 722                 nd_dbg_dpa(nd_region, ndd, new_res, "%s(%d) %d\n",
 723                                 action, loc, rc);
 724
 725                 if (rc)
 726                         return n;
 727
 728                 n -= allocate;
 729                 if (n) {
 730                         /*
 731                          * Retry scan with newly inserted resources.
 732                          * For example, if we did an ALLOC_BEFORE
 733                          * insertion there may also have been space
 734                          * available for an ALLOC_AFTER insertion, so we
 735                          * need to check this same resource again
 736                          */
 737                         goto retry;
 738                 } else
 739                         return 0;
 740         }
 741
 742         /*
 743          * If we allocated nothing in the BLK case it may be because we are in
 744          * an initial "pmem-reserve pass".  Only do an initial BLK allocation
 745          * when none of the DPA space is reserved.
 746          */
 747         if ((is_pmem || !ndd->dpa.child) && n == to_allocate)
 748                 return init_dpa_allocation(label_id, nd_region, nd_mapping, n);
 749         return n;
 750 }
 751
 752 static int merge_dpa(struct nd_region *nd_region,
 753                 struct nd_mapping *nd_mapping, struct nd_label_id *label_id)
 754 {
 755         struct nvdimm_drvdata *ndd = to_ndd(nd_mapping);
 756         struct resource *res;
 757
 758         if (strncmp("pmem", label_id->id, 4) == 0)
 759                 return 0;
 760  retry:
 761         for_each_dpa_resource(ndd, res) {
 762                 int rc;
 763                 struct resource *next = res->sibling;
 764                 resource_size_t end = res->start + resource_size(res);
 765
 766                 if (!next || strcmp(res->name, label_id->id) != 0
 767                                 || strcmp(next->name, label_id->id) != 0
 768                                 || end != next->start)
 769                         continue;
 770                 end += resource_size(next);
 771                 nvdimm_free_dpa(ndd, next);
 772                 rc = adjust_resource(res, res->start, end - res->start);
 773                 nd_dbg_dpa(nd_region, ndd, res, "merge %d\n", rc);
 774                 if (rc)
 775                         return rc;
 776                 res->flags |= DPA_RESOURCE_ADJUSTED;
 777                 goto retry;
 778         }
 779
 780         return 0;
 781 }
 782
 783 static int __reserve_free_pmem(struct device *dev, void *data)
 784 {
 785         struct nvdimm *nvdimm = data;
 786         struct nd_region *nd_region;
 787         struct nd_label_id label_id;
 788         int i;
 789
 790         if (!is_nd_pmem(dev))
 791                 return 0;
 792
 793         nd_region = to_nd_region(dev);
 794         if (nd_region->ndr_mappings == 0)
 795                 return 0;
 796
 797         memset(&label_id, 0, sizeof(label_id));
 798         strcat(label_id.id, "pmem-reserve");
 799         for (i = 0; i < nd_region->ndr_mappings; i++) {
 800                 struct nd_mapping *nd_mapping = &nd_region->mapping[i];
 801                 resource_size_t n, rem = 0;
 802
 803                 if (nd_mapping->nvdimm != nvdimm)
 804                         continue;
 805
 806                 n = nd_pmem_available_dpa(nd_region, nd_mapping, &rem);
 807                 if (n == 0)
 808                         return 0;
 809                 rem = scan_allocate(nd_region, nd_mapping, &label_id, n);
 810                 dev_WARN_ONCE(&nd_region->dev, rem,
 811                                 "pmem reserve underrun: %#llx of %#llx bytes\n",
 812                                 (unsigned long long) n - rem,
 813                                 (unsigned long long) n);
 814                 return rem ? -ENXIO : 0;
 815         }
 816
 817         return 0;
 818 }
 819
 820 static void release_free_pmem(struct nvdimm_bus *nvdimm_bus,
 821                 struct nd_mapping *nd_mapping)
 822 {
 823         struct nvdimm_drvdata *ndd = to_ndd(nd_mapping);
 824         struct resource *res, *_res;
 825
 826         for_each_dpa_resource_safe(ndd, res, _res)
 827                 if (strcmp(res->name, "pmem-reserve") == 0)
 828                         nvdimm_free_dpa(ndd, res);
 829 }
 830
 831 static int reserve_free_pmem(struct nvdimm_bus *nvdimm_bus,
 832                 struct nd_mapping *nd_mapping)
 833 {
 834         struct nvdimm *nvdimm = nd_mapping->nvdimm;
 835         int rc;
 836
 837         rc = device_for_each_child(&nvdimm_bus->dev, nvdimm,
 838                         __reserve_free_pmem);
 839         if (rc)
 840                 release_free_pmem(nvdimm_bus, nd_mapping);
 841         return rc;
 842 }
 843
 844 /**
 845  * grow_dpa_allocation - for each dimm allocate n bytes for @label_id
 846  * @nd_region: the set of dimms to allocate @n more bytes from
 847  * @label_id: unique identifier for the namespace consuming this dpa range
 848  * @n: number of bytes per-dimm to add to the existing allocation
 849  *
 850  * Assumes resources are ordered.  For BLK regions, first consume
 851  * BLK-only available DPA free space, then consume PMEM-aliased DPA
 852  * space starting at the highest DPA.  For PMEM regions start
 853  * allocations from the start of an interleave set and end at the first
 854  * BLK allocation or the end of the interleave set, whichever comes
 855  * first.
 856  */
 857 static int grow_dpa_allocation(struct nd_region *nd_region,
 858                 struct nd_label_id *label_id, resource_size_t n)
 859 {
 860         struct nvdimm_bus *nvdimm_bus = walk_to_nvdimm_bus(&nd_region->dev);
 861         bool is_pmem = strncmp(label_id->id, "pmem", 4) == 0;
 862         int i;
 863
 864         for (i = 0; i < nd_region->ndr_mappings; i++) {
 865                 struct nd_mapping *nd_mapping = &nd_region->mapping[i];
 866                 resource_size_t rem = n;
 867                 int rc, j;
 868
 869                 /*
 870                  * In the BLK case try once with all unallocated PMEM
 871                  * reserved, and once without
 872                  */
 873                 for (j = is_pmem; j < 2; j++) {
 874                         bool blk_only = j == 0;
 875
 876                         if (blk_only) {
 877                                 rc = reserve_free_pmem(nvdimm_bus, nd_mapping);
 878                                 if (rc)
 879                                         return rc;
 880                         }
 881                         rem = scan_allocate(nd_region, nd_mapping,
 882                                         label_id, rem);
 883                         if (blk_only)
 884                                 release_free_pmem(nvdimm_bus, nd_mapping);
 885
 886                         /* try again and allow encroachments into PMEM */
 887                         if (rem == 0)
 888                                 break;
 889                 }
 890
 891                 dev_WARN_ONCE(&nd_region->dev, rem,
 892                                 "allocation underrun: %#llx of %#llx bytes\n",
 893                                 (unsigned long long) n - rem,
 894                                 (unsigned long long) n);
 895                 if (rem)
 896                         return -ENXIO;
 897
 898                 rc = merge_dpa(nd_region, nd_mapping, label_id);
 899                 if (rc)
 900                         return rc;
 901         }
 902
 903         return 0;
 904 }
 905
 906 static void nd_namespace_pmem_set_resource(struct nd_region *nd_region,
 907                 struct nd_namespace_pmem *nspm, resource_size_t size)
 908 {
 909         struct resource *res = &nspm->nsio.res;
 910         resource_size_t offset = 0;
 911
 912         if (size && !nspm->uuid) {
 913                 WARN_ON_ONCE(1);
 914                 size = 0;
 915         }
 916
 917         if (size && nspm->uuid) {
 918                 struct nd_mapping *nd_mapping = &nd_region->mapping[0];
 919                 struct nvdimm_drvdata *ndd = to_ndd(nd_mapping);
 920                 struct nd_label_id label_id;
 921                 struct resource *res;
 922
 923                 if (!ndd) {
 924                         size = 0;
 925                         goto out;
 926                 }
 927
 928                 nd_label_gen_id(&label_id, nspm->uuid, 0);
 929
 930                 /* calculate a spa offset from the dpa allocation offset */
 931                 for_each_dpa_resource(ndd, res)
 932                         if (strcmp(res->name, label_id.id) == 0) {
 933                                 offset = (res->start - nd_mapping->start)
 934                                         * nd_region->ndr_mappings;
 935                                 goto out;
 936                         }
 937
 938                 WARN_ON_ONCE(1);
 939                 size = 0;
 940         }
 941
 942  out:
 943         res->start = nd_region->ndr_start + offset;
 944         res->end = res->start + size - 1;
 945 }
 946
 947 static bool uuid_not_set(const u8 *uuid, struct device *dev, const char *where)
 948 {
 949         if (!uuid) {
 950                 dev_dbg(dev, "%s: uuid not set\n", where);
 951                 return true;
 952         }
 953         return false;
 954 }
 955
 956 static ssize_t __size_store(struct device *dev, unsigned long long val)
 957 {
 958         resource_size_t allocated = 0, available = 0;
 959         struct nd_region *nd_region = to_nd_region(dev->parent);
 960         struct nd_namespace_common *ndns = to_ndns(dev);
 961         struct nd_mapping *nd_mapping;
 962         struct nvdimm_drvdata *ndd;
 963         struct nd_label_id label_id;
 964         u32 flags = 0, remainder;
 965         int rc, i, id = -1;
 966         u8 *uuid = NULL;
 967
 968         if (dev->driver || ndns->claim)
 969                 return -EBUSY;
 970
 971         if (is_namespace_pmem(dev)) {
 972                 struct nd_namespace_pmem *nspm = to_nd_namespace_pmem(dev);
 973
 974                 uuid = nspm->uuid;
 975                 id = nspm->id;
 976         } else if (is_namespace_blk(dev)) {
 977                 struct nd_namespace_blk *nsblk = to_nd_namespace_blk(dev);
 978
 979                 uuid = nsblk->uuid;
 980                 flags = NSLABEL_FLAG_LOCAL;
 981                 id = nsblk->id;
 982         }
 983
 984         /*
 985          * We need a uuid for the allocation-label and dimm(s) on which
 986          * to store the label.
 987          */
 988         if (uuid_not_set(uuid, dev, __func__))
 989                 return -ENXIO;
 990         if (nd_region->ndr_mappings == 0) {
 991                 dev_dbg(dev, "%s: not associated with dimm(s)\n", __func__);
 992                 return -ENXIO;
 993         }
 994
 995         div_u64_rem(val, SZ_4K * nd_region->ndr_mappings, &remainder);
 996         if (remainder) {
 997                 dev_dbg(dev, "%llu is not %dK aligned\n", val,
 998                                 (SZ_4K * nd_region->ndr_mappings) / SZ_1K);
 999                 return -EINVAL;
1000         }
1001
1002         nd_label_gen_id(&label_id, uuid, flags);
1003         for (i = 0; i < nd_region->ndr_mappings; i++) {
1004                 nd_mapping = &nd_region->mapping[i];
1005                 ndd = to_ndd(nd_mapping);
1006
1007                 /*
1008                  * All dimms in an interleave set, or the base dimm for a blk
1009                  * region, need to be enabled for the size to be changed.
1010                  */
1011                 if (!ndd)
1012                         return -ENXIO;
1013
1014                 allocated += nvdimm_allocated_dpa(ndd, &label_id);
1015         }
1016         available = nd_region_available_dpa(nd_region);
1017
1018         if (val > available + allocated)
1019                 return -ENOSPC;
1020
1021         if (val == allocated)
1022                 return 0;
1023
1024         val = div_u64(val, nd_region->ndr_mappings);
1025         allocated = div_u64(allocated, nd_region->ndr_mappings);
1026         if (val < allocated)
1027                 rc = shrink_dpa_allocation(nd_region, &label_id,
1028                                 allocated - val);
1029         else
1030                 rc = grow_dpa_allocation(nd_region, &label_id, val - allocated);
1031
1032         if (rc)
1033                 return rc;
1034
1035         if (is_namespace_pmem(dev)) {
1036                 struct nd_namespace_pmem *nspm = to_nd_namespace_pmem(dev);
1037
1038                 nd_namespace_pmem_set_resource(nd_region, nspm,
1039                                 val * nd_region->ndr_mappings);
1040         }
1041
1042         /*
1043          * Try to delete the namespace if we deleted all of its
1044          * allocation, this is not the seed or 0th device for the
1045          * region, and it is not actively claimed by a btt, pfn, or dax
1046          * instance.
1047          */
1048         if (val == 0 && id != 0 && nd_region->ns_seed != dev && !ndns->claim)
1049                 nd_device_unregister(dev, ND_ASYNC);
1050
1051         return rc;
1052 }
1053
1054 static ssize_t size_store(struct device *dev,
1055                 struct device_attribute *attr, const char *buf, size_t len)
1056 {
1057         struct nd_region *nd_region = to_nd_region(dev->parent);
1058         unsigned long long val;
1059         u8 **uuid = NULL;
1060         int rc;
1061
1062         rc = kstrtoull(buf, 0, &val);
1063         if (rc)
1064                 return rc;
1065
1066         device_lock(dev);
1067         nvdimm_bus_lock(dev);
1068         wait_nvdimm_bus_probe_idle(dev);
1069         rc = __size_store(dev, val);
1070         if (rc >= 0)
1071                 rc = nd_namespace_label_update(nd_region, dev);
1072
1073         if (is_namespace_pmem(dev)) {
1074                 struct nd_namespace_pmem *nspm = to_nd_namespace_pmem(dev);
1075
1076                 uuid = &nspm->uuid;
1077         } else if (is_namespace_blk(dev)) {
1078                 struct nd_namespace_blk *nsblk = to_nd_namespace_blk(dev);
1079
1080                 uuid = &nsblk->uuid;
1081         }
1082
1083         if (rc == 0 && val == 0 && uuid) {
1084                 /* setting size zero == 'delete namespace' */
1085                 kfree(*uuid);
1086                 *uuid = NULL;
1087         }
1088
1089         dev_dbg(dev, "%s: %llx %s (%d)\n", __func__, val, rc < 0
1090                         ? "fail" : "success", rc);
1091
1092         nvdimm_bus_unlock(dev);
1093         device_unlock(dev);
1094
1095         return rc < 0 ? rc : len;
1096 }
1097
1098 resource_size_t __nvdimm_namespace_capacity(struct nd_namespace_common *ndns)
1099 {
1100         struct device *dev = &ndns->dev;
1101
1102         if (is_namespace_pmem(dev)) {
1103                 struct nd_namespace_pmem *nspm = to_nd_namespace_pmem(dev);
1104
1105                 return resource_size(&nspm->nsio.res);
1106         } else if (is_namespace_blk(dev)) {
1107                 return nd_namespace_blk_size(to_nd_namespace_blk(dev));
1108         } else if (is_namespace_io(dev)) {
1109                 struct nd_namespace_io *nsio = to_nd_namespace_io(dev);
1110
1111                 return resource_size(&nsio->res);
1112         } else
1113                 WARN_ONCE(1, "unknown namespace type\n");
1114         return 0;
1115 }
1116
1117 resource_size_t nvdimm_namespace_capacity(struct nd_namespace_common *ndns)
1118 {
1119         resource_size_t size;
1120
1121         nvdimm_bus_lock(&ndns->dev);
1122         size = __nvdimm_namespace_capacity(ndns);
1123         nvdimm_bus_unlock(&ndns->dev);
1124
1125         return size;
1126 }
1127 EXPORT_SYMBOL(nvdimm_namespace_capacity);
1128
1129 static ssize_t size_show(struct device *dev,
1130                 struct device_attribute *attr, char *buf)
1131 {
1132         return sprintf(buf, "%llu\n", (unsigned long long)
1133                         nvdimm_namespace_capacity(to_ndns(dev)));
1134 }
1135 static DEVICE_ATTR(size, 0444, size_show, size_store);
1136
1137 static u8 *namespace_to_uuid(struct device *dev)
1138 {
1139         if (is_namespace_pmem(dev)) {
1140                 struct nd_namespace_pmem *nspm = to_nd_namespace_pmem(dev);
1141
1142                 return nspm->uuid;
1143         } else if (is_namespace_blk(dev)) {
1144                 struct nd_namespace_blk *nsblk = to_nd_namespace_blk(dev);
1145
1146                 return nsblk->uuid;
1147         } else
1148                 return ERR_PTR(-ENXIO);
1149 }
1150
1151 static ssize_t uuid_show(struct device *dev,
1152                 struct device_attribute *attr, char *buf)
1153 {
1154         u8 *uuid = namespace_to_uuid(dev);
1155
1156         if (IS_ERR(uuid))
1157                 return PTR_ERR(uuid);
1158         if (uuid)
1159                 return sprintf(buf, "%pUb\n", uuid);
1160         return sprintf(buf, "\n");
1161 }
1162
1163 /**
1164  * namespace_update_uuid - check for a unique uuid and whether we're "renaming"
1165  * @nd_region: parent region so we can updates all dimms in the set
1166  * @dev: namespace type for generating label_id
1167  * @new_uuid: incoming uuid
1168  * @old_uuid: reference to the uuid storage location in the namespace object
1169  */
1170 static int namespace_update_uuid(struct nd_region *nd_region,
1171                 struct device *dev, u8 *new_uuid, u8 **old_uuid)
1172 {
1173         u32 flags = is_namespace_blk(dev) ? NSLABEL_FLAG_LOCAL : 0;
1174         struct nd_label_id old_label_id;
1175         struct nd_label_id new_label_id;
1176         int i;
1177
1178         if (!nd_is_uuid_unique(dev, new_uuid))
1179                 return -EINVAL;
1180
1181         if (*old_uuid == NULL)
1182                 goto out;
1183
1184         /*
1185          * If we've already written a label with this uuid, then it's
1186          * too late to rename because we can't reliably update the uuid
1187          * without losing the old namespace.  Userspace must delete this
1188          * namespace to abandon the old uuid.
1189          */
1190         for (i = 0; i < nd_region->ndr_mappings; i++) {
1191                 struct nd_mapping *nd_mapping = &nd_region->mapping[i];
1192
1193                 /*
1194                  * This check by itself is sufficient because old_uuid
1195                  * would be NULL above if this uuid did not exist in the
1196                  * currently written set.
1197                  *
1198                  * FIXME: can we delete uuid with zero dpa allocated?
1199                  */
1200                 if (list_empty(&nd_mapping->labels))
1201                         return -EBUSY;
1202         }
1203
1204         nd_label_gen_id(&old_label_id, *old_uuid, flags);
1205         nd_label_gen_id(&new_label_id, new_uuid, flags);
1206         for (i = 0; i < nd_region->ndr_mappings; i++) {
1207                 struct nd_mapping *nd_mapping = &nd_region->mapping[i];
1208                 struct nvdimm_drvdata *ndd = to_ndd(nd_mapping);
1209                 struct resource *res;
1210
1211                 for_each_dpa_resource(ndd, res)
1212                         if (strcmp(res->name, old_label_id.id) == 0)
1213                                 sprintf((void *) res->name, "%s",
1214                                                 new_label_id.id);
1215         }
1216         kfree(*old_uuid);
1217  out:
1218         *old_uuid = new_uuid;
1219         return 0;
1220 }
1221
1222 static ssize_t uuid_store(struct device *dev,
1223                 struct device_attribute *attr, const char *buf, size_t len)
1224 {
1225         struct nd_region *nd_region = to_nd_region(dev->parent);
1226         u8 *uuid = NULL;
1227         ssize_t rc = 0;
1228         u8 **ns_uuid;
1229
1230         if (is_namespace_pmem(dev)) {
1231                 struct nd_namespace_pmem *nspm = to_nd_namespace_pmem(dev);
1232
1233                 ns_uuid = &nspm->uuid;
1234         } else if (is_namespace_blk(dev)) {
1235                 struct nd_namespace_blk *nsblk = to_nd_namespace_blk(dev);
1236
1237                 ns_uuid = &nsblk->uuid;
1238         } else
1239                 return -ENXIO;
1240
1241         device_lock(dev);
1242         nvdimm_bus_lock(dev);
1243         wait_nvdimm_bus_probe_idle(dev);
1244         if (to_ndns(dev)->claim)
1245                 rc = -EBUSY;
1246         if (rc >= 0)
1247                 rc = nd_uuid_store(dev, &uuid, buf, len);
1248         if (rc >= 0)
1249                 rc = namespace_update_uuid(nd_region, dev, uuid, ns_uuid);
1250         if (rc >= 0)
1251                 rc = nd_namespace_label_update(nd_region, dev);
1252         else
1253                 kfree(uuid);
1254         dev_dbg(dev, "%s: result: %zd wrote: %s%s", __func__,
1255                         rc, buf, buf[len - 1] == '\n' ? "" : "\n");
1256         nvdimm_bus_unlock(dev);
1257         device_unlock(dev);
1258
1259         return rc < 0 ? rc : len;
1260 }
1261 static DEVICE_ATTR_RW(uuid);
1262
1263 static ssize_t resource_show(struct device *dev,
1264                 struct device_attribute *attr, char *buf)
1265 {
1266         struct resource *res;
1267
1268         if (is_namespace_pmem(dev)) {
1269                 struct nd_namespace_pmem *nspm = to_nd_namespace_pmem(dev);
1270
1271                 res = &nspm->nsio.res;
1272         } else if (is_namespace_io(dev)) {
1273                 struct nd_namespace_io *nsio = to_nd_namespace_io(dev);
1274
1275                 res = &nsio->res;
1276         } else
1277                 return -ENXIO;
1278
1279         /* no address to convey if the namespace has no allocation */
1280         if (resource_size(res) == 0)
1281                 return -ENXIO;
1282         return sprintf(buf, "%#llx\n", (unsigned long long) res->start);
1283 }
1284 static DEVICE_ATTR_RO(resource);
1285
1286 static const unsigned long ns_lbasize_supported[] = { 512, 520, 528,
1287         4096, 4104, 4160, 4224, 0 };
1288
1289 static ssize_t sector_size_show(struct device *dev,
1290                 struct device_attribute *attr, char *buf)
1291 {
1292         struct nd_namespace_blk *nsblk = to_nd_namespace_blk(dev);
1293
1294         if (!is_namespace_blk(dev))
1295                 return -ENXIO;
1296
1297         return nd_sector_size_show(nsblk->lbasize, ns_lbasize_supported, buf);
1298 }
1299
1300 static ssize_t sector_size_store(struct device *dev,
1301                 struct device_attribute *attr, const char *buf, size_t len)
1302 {
1303         struct nd_namespace_blk *nsblk = to_nd_namespace_blk(dev);
1304         struct nd_region *nd_region = to_nd_region(dev->parent);
1305         ssize_t rc = 0;
1306
1307         if (!is_namespace_blk(dev))
1308                 return -ENXIO;
1309
1310         device_lock(dev);
1311         nvdimm_bus_lock(dev);
1312         if (to_ndns(dev)->claim)
1313                 rc = -EBUSY;
1314         if (rc >= 0)
1315                 rc = nd_sector_size_store(dev, buf, &nsblk->lbasize,
1316                                 ns_lbasize_supported);
1317         if (rc >= 0)
1318                 rc = nd_namespace_label_update(nd_region, dev);
1319         dev_dbg(dev, "%s: result: %zd %s: %s%s", __func__,
1320                         rc, rc < 0 ? "tried" : "wrote", buf,
1321                         buf[len - 1] == '\n' ? "" : "\n");
1322         nvdimm_bus_unlock(dev);
1323         device_unlock(dev);
1324
1325         return rc ? rc : len;
1326 }
1327 static DEVICE_ATTR_RW(sector_size);
1328
1329 static ssize_t dpa_extents_show(struct device *dev,
1330                 struct device_attribute *attr, char *buf)
1331 {
1332         struct nd_region *nd_region = to_nd_region(dev->parent);
1333         struct nd_label_id label_id;
1334         int count = 0, i;
1335         u8 *uuid = NULL;
1336         u32 flags = 0;
1337
1338         nvdimm_bus_lock(dev);
1339         if (is_namespace_pmem(dev)) {
1340                 struct nd_namespace_pmem *nspm = to_nd_namespace_pmem(dev);
1341
1342                 uuid = nspm->uuid;
1343                 flags = 0;
1344         } else if (is_namespace_blk(dev)) {
1345                 struct nd_namespace_blk *nsblk = to_nd_namespace_blk(dev);
1346
1347                 uuid = nsblk->uuid;
1348                 flags = NSLABEL_FLAG_LOCAL;
1349         }
1350
1351         if (!uuid)
1352                 goto out;
1353
1354         nd_label_gen_id(&label_id, uuid, flags);
1355         for (i = 0; i < nd_region->ndr_mappings; i++) {
1356                 struct nd_mapping *nd_mapping = &nd_region->mapping[i];
1357                 struct nvdimm_drvdata *ndd = to_ndd(nd_mapping);
1358                 struct resource *res;
1359
1360                 for_each_dpa_resource(ndd, res)
1361                         if (strcmp(res->name, label_id.id) == 0)
1362                                 count++;
1363         }
1364  out:
1365         nvdimm_bus_unlock(dev);
1366
1367         return sprintf(buf, "%d\n", count);
1368 }
1369 static DEVICE_ATTR_RO(dpa_extents);
1370
1371 static ssize_t holder_show(struct device *dev,
1372                 struct device_attribute *attr, char *buf)
1373 {
1374         struct nd_namespace_common *ndns = to_ndns(dev);
1375         ssize_t rc;
1376
1377         device_lock(dev);
1378         rc = sprintf(buf, "%s\n", ndns->claim ? dev_name(ndns->claim) : "");
1379         device_unlock(dev);
1380
1381         return rc;
1382 }
1383 static DEVICE_ATTR_RO(holder);
1384
1385 static ssize_t mode_show(struct device *dev,
1386                 struct device_attribute *attr, char *buf)
1387 {
1388         struct nd_namespace_common *ndns = to_ndns(dev);
1389         struct device *claim;
1390         char *mode;
1391         ssize_t rc;
1392
1393         device_lock(dev);
1394         claim = ndns->claim;
1395         if (claim && is_nd_btt(claim))
1396                 mode = "safe";
1397         else if (claim && is_nd_pfn(claim))
1398                 mode = "memory";
1399         else if (claim && is_nd_dax(claim))
1400                 mode = "dax";
1401         else if (!claim && pmem_should_map_pages(dev))
1402                 mode = "memory";
1403         else
1404                 mode = "raw";
1405         rc = sprintf(buf, "%s\n", mode);
1406         device_unlock(dev);
1407
1408         return rc;
1409 }
1410 static DEVICE_ATTR_RO(mode);
1411
1412 static ssize_t force_raw_store(struct device *dev,
1413                 struct device_attribute *attr, const char *buf, size_t len)
1414 {
1415         bool force_raw;
1416         int rc = strtobool(buf, &force_raw);
1417
1418         if (rc)
1419                 return rc;
1420
1421         to_ndns(dev)->force_raw = force_raw;
1422         return len;
1423 }
1424
1425 static ssize_t force_raw_show(struct device *dev,
1426                 struct device_attribute *attr, char *buf)
1427 {
1428         return sprintf(buf, "%d\n", to_ndns(dev)->force_raw);
1429 }
1430 static DEVICE_ATTR_RW(force_raw);
1431
1432 static struct attribute *nd_namespace_attributes[] = {
1433         &dev_attr_nstype.attr,
1434         &dev_attr_size.attr,
1435         &dev_attr_mode.attr,
1436         &dev_attr_uuid.attr,
1437         &dev_attr_holder.attr,
1438         &dev_attr_resource.attr,
1439         &dev_attr_alt_name.attr,
1440         &dev_attr_force_raw.attr,
1441         &dev_attr_sector_size.attr,
1442         &dev_attr_dpa_extents.attr,
1443         NULL,
1444 };
1445
1446 static umode_t namespace_visible(struct kobject *kobj,
1447                 struct attribute *a, int n)
1448 {
1449         struct device *dev = container_of(kobj, struct device, kobj);
1450
1451         if (a == &dev_attr_resource.attr) {
1452                 if (is_namespace_blk(dev))
1453                         return 0;
1454                 return a->mode;
1455         }
1456
1457         if (is_namespace_pmem(dev) || is_namespace_blk(dev)) {
1458                 if (a == &dev_attr_size.attr)
1459                         return 0644;
1460
1461                 if (is_namespace_pmem(dev) && a == &dev_attr_sector_size.attr)
1462                         return 0;
1463
1464                 return a->mode;
1465         }
1466
1467         if (a == &dev_attr_nstype.attr || a == &dev_attr_size.attr
1468                         || a == &dev_attr_holder.attr
1469                         || a == &dev_attr_force_raw.attr
1470                         || a == &dev_attr_mode.attr)
1471                 return a->mode;
1472
1473         return 0;
1474 }
1475
1476 static struct attribute_group nd_namespace_attribute_group = {
1477         .attrs = nd_namespace_attributes,
1478         .is_visible = namespace_visible,
1479 };
1480
1481 static const struct attribute_group *nd_namespace_attribute_groups[] = {
1482         &nd_device_attribute_group,
1483         &nd_namespace_attribute_group,
1484         &nd_numa_attribute_group,
1485         NULL,
1486 };
1487
1488 struct nd_namespace_common *nvdimm_namespace_common_probe(struct device *dev)
1489 {
1490         struct nd_btt *nd_btt = is_nd_btt(dev) ? to_nd_btt(dev) : NULL;
1491         struct nd_pfn *nd_pfn = is_nd_pfn(dev) ? to_nd_pfn(dev) : NULL;
1492         struct nd_dax *nd_dax = is_nd_dax(dev) ? to_nd_dax(dev) : NULL;
1493         struct nd_namespace_common *ndns = NULL;
1494         resource_size_t size;
1495
1496         if (nd_btt || nd_pfn || nd_dax) {
1497                 if (nd_btt)
1498                         ndns = nd_btt->ndns;
1499                 else if (nd_pfn)
1500                         ndns = nd_pfn->ndns;
1501                 else if (nd_dax)
1502                         ndns = nd_dax->nd_pfn.ndns;
1503
1504                 if (!ndns)
1505                         return ERR_PTR(-ENODEV);
1506
1507                 /*
1508                  * Flush any in-progess probes / removals in the driver
1509                  * for the raw personality of this namespace.
1510                  */
1511                 device_lock(&ndns->dev);
1512                 device_unlock(&ndns->dev);
1513                 if (ndns->dev.driver) {
1514                         dev_dbg(&ndns->dev, "is active, can't bind %s\n",
1515                                         dev_name(dev));
1516                         return ERR_PTR(-EBUSY);
1517                 }
1518                 if (dev_WARN_ONCE(&ndns->dev, ndns->claim != dev,
1519                                         "host (%s) vs claim (%s) mismatch\n",
1520                                         dev_name(dev),
1521                                         dev_name(ndns->claim)))
1522                         return ERR_PTR(-ENXIO);
1523         } else {
1524                 ndns = to_ndns(dev);
1525                 if (ndns->claim) {
1526                         dev_dbg(dev, "claimed by %s, failing probe\n",
1527                                 dev_name(ndns->claim));
1528
1529                         return ERR_PTR(-ENXIO);
1530                 }
1531         }
1532
1533         size = nvdimm_namespace_capacity(ndns);
1534         if (size < ND_MIN_NAMESPACE_SIZE) {
1535                 dev_dbg(&ndns->dev, "%pa, too small must be at least %#x\n",
1536                                 &size, ND_MIN_NAMESPACE_SIZE);
1537                 return ERR_PTR(-ENODEV);
1538         }
1539
1540         if (is_namespace_pmem(&ndns->dev)) {
1541                 struct nd_namespace_pmem *nspm;
1542
1543                 nspm = to_nd_namespace_pmem(&ndns->dev);
1544                 if (uuid_not_set(nspm->uuid, &ndns->dev, __func__))
1545                         return ERR_PTR(-ENODEV);
1546         } else if (is_namespace_blk(&ndns->dev)) {
1547                 struct nd_namespace_blk *nsblk;
1548
1549                 nsblk = to_nd_namespace_blk(&ndns->dev);
1550                 if (uuid_not_set(nsblk->uuid, &ndns->dev, __func__))
1551                         return ERR_PTR(-ENODEV);
1552                 if (!nsblk->lbasize) {
1553                         dev_dbg(&ndns->dev, "%s: sector size not set\n",
1554                                 __func__);
1555                         return ERR_PTR(-ENODEV);
1556                 }
1557                 if (!nd_namespace_blk_validate(nsblk))
1558                         return ERR_PTR(-ENODEV);
1559         }
1560
1561         return ndns;
1562 }
1563 EXPORT_SYMBOL(nvdimm_namespace_common_probe);
1564
1565 static struct device **create_namespace_io(struct nd_region *nd_region)
1566 {
1567         struct nd_namespace_io *nsio;
1568         struct device *dev, **devs;
1569         struct resource *res;
1570
1571         nsio = kzalloc(sizeof(*nsio), GFP_KERNEL);
1572         if (!nsio)
1573                 return NULL;
1574
1575         devs = kcalloc(2, sizeof(struct device *), GFP_KERNEL);
1576         if (!devs) {
1577                 kfree(nsio);
1578                 return NULL;
1579         }
1580
1581         dev = &nsio->common.dev;
1582         dev->type = &namespace_io_device_type;
1583         dev->parent = &nd_region->dev;
1584         res = &nsio->res;
1585         res->name = dev_name(&nd_region->dev);
1586         res->flags = IORESOURCE_MEM;
1587         res->start = nd_region->ndr_start;
1588         res->end = res->start + nd_region->ndr_size - 1;
1589
1590         devs[0] = dev;
1591         return devs;
1592 }
1593
1594 static bool has_uuid_at_pos(struct nd_region *nd_region, u8 *uuid,
1595                 u64 cookie, u16 pos)
1596 {
1597         struct nd_namespace_label *found = NULL;
1598         int i;
1599
1600         for (i = 0; i < nd_region->ndr_mappings; i++) {
1601                 struct nd_mapping *nd_mapping = &nd_region->mapping[i];
1602                 struct nd_label_ent *label_ent;
1603                 bool found_uuid = false;
1604
1605                 list_for_each_entry(label_ent, &nd_mapping->labels, list) {
1606                         struct nd_namespace_label *nd_label = label_ent->label;
1607                         u16 position, nlabel;
1608                         u64 isetcookie;
1609
1610                         if (!nd_label)
1611                                 continue;
1612                         isetcookie = __le64_to_cpu(nd_label->isetcookie);
1613                         position = __le16_to_cpu(nd_label->position);
1614                         nlabel = __le16_to_cpu(nd_label->nlabel);
1615
1616                         if (isetcookie != cookie)
1617                                 continue;
1618
1619                         if (memcmp(nd_label->uuid, uuid, NSLABEL_UUID_LEN) != 0)
1620                                 continue;
1621
1622                         if (found_uuid) {
1623                                 dev_dbg(to_ndd(nd_mapping)->dev,
1624                                                 "%s duplicate entry for uuid\n",
1625                                                 __func__);
1626                                 return false;
1627                         }
1628                         found_uuid = true;
1629                         if (nlabel != nd_region->ndr_mappings)
1630                                 continue;
1631                         if (position != pos)
1632                                 continue;
1633                         found = nd_label;
1634                         break;
1635                 }
1636                 if (found)
1637                         break;
1638         }
1639         return found != NULL;
1640 }
1641
1642 static int select_pmem_id(struct nd_region *nd_region, u8 *pmem_id)
1643 {
1644         int i;
1645
1646         if (!pmem_id)
1647                 return -ENODEV;
1648
1649         for (i = 0; i < nd_region->ndr_mappings; i++) {
1650                 struct nd_mapping *nd_mapping = &nd_region->mapping[i];
1651                 struct nvdimm_drvdata *ndd = to_ndd(nd_mapping);
1652                 struct nd_namespace_label *nd_label = NULL;
1653                 u64 hw_start, hw_end, pmem_start, pmem_end;
1654                 struct nd_label_ent *label_ent;
1655
1656                 lockdep_assert_held(&nd_mapping->lock);
1657                 list_for_each_entry(label_ent, &nd_mapping->labels, list) {
1658                         nd_label = label_ent->label;
1659                         if (!nd_label)
1660                                 continue;
1661                         if (memcmp(nd_label->uuid, pmem_id, NSLABEL_UUID_LEN) == 0)
1662                                 break;
1663                         nd_label = NULL;
1664                 }
1665
1666                 if (!nd_label) {
1667                         WARN_ON(1);
1668                         return -EINVAL;
1669                 }
1670
1671                 /*
1672                  * Check that this label is compliant with the dpa
1673                  * range published in NFIT
1674                  */
1675                 hw_start = nd_mapping->start;
1676                 hw_end = hw_start + nd_mapping->size;
1677                 pmem_start = __le64_to_cpu(nd_label->dpa);
1678                 pmem_end = pmem_start + __le64_to_cpu(nd_label->rawsize);
1679                 if (pmem_start >= hw_start && pmem_start < hw_end
1680                                 && pmem_end <= hw_end && pmem_end > hw_start)
1681                         /* pass */;
1682                 else {
1683                         dev_dbg(&nd_region->dev, "%s invalid label for %pUb\n",
1684                                         dev_name(ndd->dev), nd_label->uuid);
1685                         return -EINVAL;
1686                 }
1687
1688                 /* move recently validated label to the front of the list */
1689                 list_move(&label_ent->list, &nd_mapping->labels);
1690         }
1691         return 0;
1692 }
1693
1694 /**
1695  * create_namespace_pmem - validate interleave set labelling, retrieve label0
1696  * @nd_region: region with mappings to validate
1697  * @nspm: target namespace to create
1698  * @nd_label: target pmem namespace label to evaluate
1699  */
1700 struct device *create_namespace_pmem(struct nd_region *nd_region,
1701                 struct nd_namespace_label *nd_label)
1702 {
1703         u64 altcookie = nd_region_interleave_set_altcookie(nd_region);
1704         u64 cookie = nd_region_interleave_set_cookie(nd_region);
1705         struct nd_label_ent *label_ent;
1706         struct nd_namespace_pmem *nspm;
1707         struct nd_mapping *nd_mapping;
1708         resource_size_t size = 0;
1709         struct resource *res;
1710         struct device *dev;
1711         int rc = 0;
1712         u16 i;
1713
1714         if (cookie == 0) {
1715                 dev_dbg(&nd_region->dev, "invalid interleave-set-cookie\n");
1716                 return ERR_PTR(-ENXIO);
1717         }
1718
1719         if (__le64_to_cpu(nd_label->isetcookie) != cookie) {
1720                 dev_dbg(&nd_region->dev, "invalid cookie in label: %pUb\n",
1721                                 nd_label->uuid);
1722                 if (__le64_to_cpu(nd_label->isetcookie) != altcookie)
1723                         return ERR_PTR(-EAGAIN);
1724
1725                 dev_dbg(&nd_region->dev, "valid altcookie in label: %pUb\n",
1726                                 nd_label->uuid);
1727         }
1728
1729         nspm = kzalloc(sizeof(*nspm), GFP_KERNEL);
1730         if (!nspm)
1731                 return ERR_PTR(-ENOMEM);
1732
1733         nspm->id = -1;
1734         dev = &nspm->nsio.common.dev;
1735         dev->type = &namespace_pmem_device_type;
1736         dev->parent = &nd_region->dev;
1737         res = &nspm->nsio.res;
1738         res->name = dev_name(&nd_region->dev);
1739         res->flags = IORESOURCE_MEM;
1740
1741         for (i = 0; i < nd_region->ndr_mappings; i++) {
1742                 if (has_uuid_at_pos(nd_region, nd_label->uuid, cookie, i))
1743                         continue;
1744                 if (has_uuid_at_pos(nd_region, nd_label->uuid, altcookie, i))
1745                         continue;
1746                 break;
1747         }
1748
1749         if (i < nd_region->ndr_mappings) {
1750                 struct nvdimm_drvdata *ndd = to_ndd(&nd_region->mapping[i]);
1751
1752                 /*
1753                  * Give up if we don't find an instance of a uuid at each
1754                  * position (from 0 to nd_region->ndr_mappings - 1), or if we
1755                  * find a dimm with two instances of the same uuid.
1756                  */
1757                 dev_err(&nd_region->dev, "%s missing label for %pUb\n",
1758                                 dev_name(ndd->dev), nd_label->uuid);
1759                 rc = -EINVAL;
1760                 goto err;
1761         }
1762
1763         /*
1764          * Fix up each mapping's 'labels' to have the validated pmem label for
1765          * that position at labels[0], and NULL at labels[1].  In the process,
1766          * check that the namespace aligns with interleave-set.  We know
1767          * that it does not overlap with any blk namespaces by virtue of
1768          * the dimm being enabled (i.e. nd_label_reserve_dpa()
1769          * succeeded).
1770          */
1771         rc = select_pmem_id(nd_region, nd_label->uuid);
1772         if (rc)
1773                 goto err;
1774
1775         /* Calculate total size and populate namespace properties from label0 */
1776         for (i = 0; i < nd_region->ndr_mappings; i++) {
1777                 struct nd_namespace_label *label0;
1778
1779                 nd_mapping = &nd_region->mapping[i];
1780                 label_ent = list_first_entry_or_null(&nd_mapping->labels,
1781                                 typeof(*label_ent), list);
1782                 label0 = label_ent ? label_ent->label : 0;
1783
1784                 if (!label0) {
1785                         WARN_ON(1);
1786                         continue;
1787                 }
1788
1789                 size += __le64_to_cpu(label0->rawsize);
1790                 if (__le16_to_cpu(label0->position) != 0)
1791                         continue;
1792                 WARN_ON(nspm->alt_name || nspm->uuid);
1793                 nspm->alt_name = kmemdup((void __force *) label0->name,
1794                                 NSLABEL_NAME_LEN, GFP_KERNEL);
1795                 nspm->uuid = kmemdup((void __force *) label0->uuid,
1796                                 NSLABEL_UUID_LEN, GFP_KERNEL);
1797         }
1798
1799         if (!nspm->alt_name || !nspm->uuid) {
1800                 rc = -ENOMEM;
1801                 goto err;
1802         }
1803
1804         nd_namespace_pmem_set_resource(nd_region, nspm, size);
1805
1806         return dev;
1807  err:
1808         namespace_pmem_release(dev);
1809         switch (rc) {
1810         case -EINVAL:
1811                 dev_dbg(&nd_region->dev, "%s: invalid label(s)\n", __func__);
1812                 break;
1813         case -ENODEV:
1814                 dev_dbg(&nd_region->dev, "%s: label not found\n", __func__);
1815                 break;
1816         default:
1817                 dev_dbg(&nd_region->dev, "%s: unexpected err: %d\n",
1818                                 __func__, rc);
1819                 break;
1820         }
1821         return ERR_PTR(rc);
1822 }
1823
1824 struct resource *nsblk_add_resource(struct nd_region *nd_region,
1825                 struct nvdimm_drvdata *ndd, struct nd_namespace_blk *nsblk,
1826                 resource_size_t start)
1827 {
1828         struct nd_label_id label_id;
1829         struct resource *res;
1830
1831         nd_label_gen_id(&label_id, nsblk->uuid, NSLABEL_FLAG_LOCAL);
1832         res = krealloc(nsblk->res,
1833                         sizeof(void *) * (nsblk->num_resources + 1),
1834                         GFP_KERNEL);
1835         if (!res)
1836                 return NULL;
1837         nsblk->res = (struct resource **) res;
1838         for_each_dpa_resource(ndd, res)
1839                 if (strcmp(res->name, label_id.id) == 0
1840                                 && res->start == start) {
1841                         nsblk->res[nsblk->num_resources++] = res;
1842                         return res;
1843                 }
1844         return NULL;
1845 }
1846
1847 static struct device *nd_namespace_blk_create(struct nd_region *nd_region)
1848 {
1849         struct nd_namespace_blk *nsblk;
1850         struct device *dev;
1851
1852         if (!is_nd_blk(&nd_region->dev))
1853                 return NULL;
1854
1855         nsblk = kzalloc(sizeof(*nsblk), GFP_KERNEL);
1856         if (!nsblk)
1857                 return NULL;
1858
1859         dev = &nsblk->common.dev;
1860         dev->type = &namespace_blk_device_type;
1861         nsblk->id = ida_simple_get(&nd_region->ns_ida, 0, 0, GFP_KERNEL);
1862         if (nsblk->id < 0) {
1863                 kfree(nsblk);
1864                 return NULL;
1865         }
1866         dev_set_name(dev, "namespace%d.%d", nd_region->id, nsblk->id);
1867         dev->parent = &nd_region->dev;
1868         dev->groups = nd_namespace_attribute_groups;
1869
1870         return &nsblk->common.dev;
1871 }
1872
1873 static struct device *nd_namespace_pmem_create(struct nd_region *nd_region)
1874 {
1875         struct nd_namespace_pmem *nspm;
1876         struct resource *res;
1877         struct device *dev;
1878
1879         if (!is_nd_pmem(&nd_region->dev))
1880                 return NULL;
1881
1882         nspm = kzalloc(sizeof(*nspm), GFP_KERNEL);
1883         if (!nspm)
1884                 return NULL;
1885
1886         dev = &nspm->nsio.common.dev;
1887         dev->type = &namespace_pmem_device_type;
1888         dev->parent = &nd_region->dev;
1889         res = &nspm->nsio.res;
1890         res->name = dev_name(&nd_region->dev);
1891         res->flags = IORESOURCE_MEM;
1892
1893         nspm->id = ida_simple_get(&nd_region->ns_ida, 0, 0, GFP_KERNEL);
1894         if (nspm->id < 0) {
1895                 kfree(nspm);
1896                 return NULL;
1897         }
1898         dev_set_name(dev, "namespace%d.%d", nd_region->id, nspm->id);
1899         dev->parent = &nd_region->dev;
1900         dev->groups = nd_namespace_attribute_groups;
1901         nd_namespace_pmem_set_resource(nd_region, nspm, 0);
1902
1903         return dev;
1904 }
1905
1906 void nd_region_create_ns_seed(struct nd_region *nd_region)
1907 {
1908         WARN_ON(!is_nvdimm_bus_locked(&nd_region->dev));
1909
1910         if (nd_region_to_nstype(nd_region) == ND_DEVICE_NAMESPACE_IO)
1911                 return;
1912
1913         if (is_nd_blk(&nd_region->dev))
1914                 nd_region->ns_seed = nd_namespace_blk_create(nd_region);
1915         else
1916                 nd_region->ns_seed = nd_namespace_pmem_create(nd_region);
1917
1918         /*
1919          * Seed creation failures are not fatal, provisioning is simply
1920          * disabled until memory becomes available
1921          */
1922         if (!nd_region->ns_seed)
1923                 dev_err(&nd_region->dev, "failed to create %s namespace\n",
1924                                 is_nd_blk(&nd_region->dev) ? "blk" : "pmem");
1925         else
1926                 nd_device_register(nd_region->ns_seed);
1927 }
1928
1929 void nd_region_create_dax_seed(struct nd_region *nd_region)
1930 {
1931         WARN_ON(!is_nvdimm_bus_locked(&nd_region->dev));
1932         nd_region->dax_seed = nd_dax_create(nd_region);
1933         /*
1934          * Seed creation failures are not fatal, provisioning is simply
1935          * disabled until memory becomes available
1936          */
1937         if (!nd_region->dax_seed)
1938                 dev_err(&nd_region->dev, "failed to create dax namespace\n");
1939 }
1940
1941 void nd_region_create_pfn_seed(struct nd_region *nd_region)
1942 {
1943         WARN_ON(!is_nvdimm_bus_locked(&nd_region->dev));
1944         nd_region->pfn_seed = nd_pfn_create(nd_region);
1945         /*
1946          * Seed creation failures are not fatal, provisioning is simply
1947          * disabled until memory becomes available
1948          */
1949         if (!nd_region->pfn_seed)
1950                 dev_err(&nd_region->dev, "failed to create pfn namespace\n");
1951 }
1952
1953 void nd_region_create_btt_seed(struct nd_region *nd_region)
1954 {
1955         WARN_ON(!is_nvdimm_bus_locked(&nd_region->dev));
1956         nd_region->btt_seed = nd_btt_create(nd_region);
1957         /*
1958          * Seed creation failures are not fatal, provisioning is simply
1959          * disabled until memory becomes available
1960          */
1961         if (!nd_region->btt_seed)
1962                 dev_err(&nd_region->dev, "failed to create btt namespace\n");
1963 }
1964
1965 static int add_namespace_resource(struct nd_region *nd_region,
1966                 struct nd_namespace_label *nd_label, struct device **devs,
1967                 int count)
1968 {
1969         struct nd_mapping *nd_mapping = &nd_region->mapping[0];
1970         struct nvdimm_drvdata *ndd = to_ndd(nd_mapping);
1971         int i;
1972
1973         for (i = 0; i < count; i++) {
1974                 u8 *uuid = namespace_to_uuid(devs[i]);
1975                 struct resource *res;
1976
1977                 if (IS_ERR_OR_NULL(uuid)) {
1978                         WARN_ON(1);
1979                         continue;
1980                 }
1981
1982                 if (memcmp(uuid, nd_label->uuid, NSLABEL_UUID_LEN) != 0)
1983                         continue;
1984                 if (is_namespace_blk(devs[i])) {
1985                         res = nsblk_add_resource(nd_region, ndd,
1986                                         to_nd_namespace_blk(devs[i]),
1987                                         __le64_to_cpu(nd_label->dpa));
1988                         if (!res)
1989                                 return -ENXIO;
1990                         nd_dbg_dpa(nd_region, ndd, res, "%d assign\n", count);
1991                 } else {
1992                         dev_err(&nd_region->dev,
1993                                         "error: conflicting extents for uuid: %pUb\n",
1994                                         nd_label->uuid);
1995                         return -ENXIO;
1996                 }
1997                 break;
1998         }
1999
2000         return i;
2001 }
2002
2003 struct device *create_namespace_blk(struct nd_region *nd_region,
2004                 struct nd_namespace_label *nd_label, int count)
2005 {
2006
2007         struct nd_mapping *nd_mapping = &nd_region->mapping[0];
2008         struct nvdimm_drvdata *ndd = to_ndd(nd_mapping);
2009         struct nd_namespace_blk *nsblk;
2010         char name[NSLABEL_NAME_LEN];
2011         struct device *dev = NULL;
2012         struct resource *res;
2013
2014         nsblk = kzalloc(sizeof(*nsblk), GFP_KERNEL);
2015         if (!nsblk)
2016                 return ERR_PTR(-ENOMEM);
2017         dev = &nsblk->common.dev;
2018         dev->type = &namespace_blk_device_type;
2019         dev->parent = &nd_region->dev;
2020         nsblk->id = -1;
2021         nsblk->lbasize = __le64_to_cpu(nd_label->lbasize);
2022         nsblk->uuid = kmemdup(nd_label->uuid, NSLABEL_UUID_LEN,
2023                         GFP_KERNEL);
2024         if (!nsblk->uuid)
2025                 goto blk_err;
2026         memcpy(name, nd_label->name, NSLABEL_NAME_LEN);
2027         if (name[0])
2028                 nsblk->alt_name = kmemdup(name, NSLABEL_NAME_LEN,
2029                                 GFP_KERNEL);
2030         res = nsblk_add_resource(nd_region, ndd, nsblk,
2031                         __le64_to_cpu(nd_label->dpa));
2032         if (!res)
2033                 goto blk_err;
2034         nd_dbg_dpa(nd_region, ndd, res, "%d: assign\n", count);
2035         return dev;
2036  blk_err:
2037         namespace_blk_release(dev);
2038         return ERR_PTR(-ENXIO);
2039 }
2040
2041 static int cmp_dpa(const void *a, const void *b)
2042 {
2043         const struct device *dev_a = *(const struct device **) a;
2044         const struct device *dev_b = *(const struct device **) b;
2045         struct nd_namespace_blk *nsblk_a, *nsblk_b;
2046         struct nd_namespace_pmem *nspm_a, *nspm_b;
2047
2048         if (is_namespace_io(dev_a))
2049                 return 0;
2050
2051         if (is_namespace_blk(dev_a)) {
2052                 nsblk_a = to_nd_namespace_blk(dev_a);
2053                 nsblk_b = to_nd_namespace_blk(dev_b);
2054
2055                 return memcmp(&nsblk_a->res[0]->start, &nsblk_b->res[0]->start,
2056                                 sizeof(resource_size_t));
2057         }
2058
2059         nspm_a = to_nd_namespace_pmem(dev_a);
2060         nspm_b = to_nd_namespace_pmem(dev_b);
2061
2062         return memcmp(&nspm_a->nsio.res.start, &nspm_b->nsio.res.start,
2063                         sizeof(resource_size_t));
2064 }
2065
2066 static struct device **scan_labels(struct nd_region *nd_region)
2067 {
2068         int i, count = 0;
2069         struct device *dev, **devs = NULL;
2070         struct nd_label_ent *label_ent, *e;
2071         struct nd_mapping *nd_mapping = &nd_region->mapping[0];
2072         resource_size_t map_end = nd_mapping->start + nd_mapping->size - 1;
2073
2074         /* "safe" because create_namespace_pmem() might list_move() label_ent */
2075         list_for_each_entry_safe(label_ent, e, &nd_mapping->labels, list) {
2076                 struct nd_namespace_label *nd_label = label_ent->label;
2077                 struct device **__devs;
2078                 u32 flags;
2079
2080                 if (!nd_label)
2081                         continue;
2082                 flags = __le32_to_cpu(nd_label->flags);
2083                 if (is_nd_blk(&nd_region->dev)
2084                                 == !!(flags & NSLABEL_FLAG_LOCAL))
2085                         /* pass, region matches label type */;
2086                 else
2087                         continue;
2088
2089                 /* skip labels that describe extents outside of the region */
2090                 if (nd_label->dpa < nd_mapping->start || nd_label->dpa > map_end)
2091                         continue;
2092
2093                 i = add_namespace_resource(nd_region, nd_label, devs, count);
2094                 if (i < 0)
2095                         goto err;
2096                 if (i < count)
2097                         continue;
2098                 __devs = kcalloc(count + 2, sizeof(dev), GFP_KERNEL);
2099                 if (!__devs)
2100                         goto err;
2101                 memcpy(__devs, devs, sizeof(dev) * count);
2102                 kfree(devs);
2103                 devs = __devs;
2104
2105                 if (is_nd_blk(&nd_region->dev)) {
2106                         dev = create_namespace_blk(nd_region, nd_label, count);
2107                         if (IS_ERR(dev))
2108                                 goto err;
2109                         devs[count++] = dev;
2110                 } else {
2111                         dev = create_namespace_pmem(nd_region, nd_label);
2112                         if (IS_ERR(dev)) {
2113                                 switch (PTR_ERR(dev)) {
2114                                 case -EAGAIN:
2115                                         /* skip invalid labels */
2116                                         continue;
2117                                 case -ENODEV:
2118                                         /* fallthrough to seed creation */
2119                                         break;
2120                                 default:
2121                                         goto err;
2122                                 }
2123                         } else
2124                                 devs[count++] = dev;
2125                 }
2126         }
2127
2128         dev_dbg(&nd_region->dev, "%s: discovered %d %s namespace%s\n",
2129                         __func__, count, is_nd_blk(&nd_region->dev)
2130                         ? "blk" : "pmem", count == 1 ? "" : "s");
2131
2132         if (count == 0) {
2133                 /* Publish a zero-sized namespace for userspace to configure. */
2134                 nd_mapping_free_labels(nd_mapping);
2135
2136                 devs = kcalloc(2, sizeof(dev), GFP_KERNEL);
2137                 if (!devs)
2138                         goto err;
2139                 if (is_nd_blk(&nd_region->dev)) {
2140                         struct nd_namespace_blk *nsblk;
2141
2142                         nsblk = kzalloc(sizeof(*nsblk), GFP_KERNEL);
2143                         if (!nsblk)
2144                                 goto err;
2145                         dev = &nsblk->common.dev;
2146                         dev->type = &namespace_blk_device_type;
2147                 } else {
2148                         struct nd_namespace_pmem *nspm;
2149
2150                         nspm = kzalloc(sizeof(*nspm), GFP_KERNEL);
2151                         if (!nspm)
2152                                 goto err;
2153                         dev = &nspm->nsio.common.dev;
2154                         dev->type = &namespace_pmem_device_type;
2155                         nd_namespace_pmem_set_resource(nd_region, nspm, 0);
2156                 }
2157                 dev->parent = &nd_region->dev;
2158                 devs[count++] = dev;
2159         } else if (is_nd_pmem(&nd_region->dev)) {
2160                 /* clean unselected labels */
2161                 for (i = 0; i < nd_region->ndr_mappings; i++) {
2162                         struct list_head *l, *e;
2163                         LIST_HEAD(list);
2164                         int j;
2165
2166                         nd_mapping = &nd_region->mapping[i];
2167                         if (list_empty(&nd_mapping->labels)) {
2168                                 WARN_ON(1);
2169                                 continue;
2170                         }
2171
2172                         j = count;
2173                         list_for_each_safe(l, e, &nd_mapping->labels) {
2174                                 if (!j--)
2175                                         break;
2176                                 list_move_tail(l, &list);
2177                         }
2178                         nd_mapping_free_labels(nd_mapping);
2179                         list_splice_init(&list, &nd_mapping->labels);
2180                 }
2181         }
2182
2183         if (count > 1)
2184                 sort(devs, count, sizeof(struct device *), cmp_dpa, NULL);
2185
2186         return devs;
2187
2188  err:
2189         if (devs) {
2190                 for (i = 0; devs[i]; i++)
2191                         if (is_nd_blk(&nd_region->dev))
2192                                 namespace_blk_release(devs[i]);
2193                         else
2194                                 namespace_pmem_release(devs[i]);
2195                 kfree(devs);
2196         }
2197         return NULL;
2198 }
2199
2200 static struct device **create_namespaces(struct nd_region *nd_region)
2201 {
2202         struct nd_mapping *nd_mapping = &nd_region->mapping[0];
2203         struct device **devs;
2204         int i;
2205
2206         if (nd_region->ndr_mappings == 0)
2207                 return NULL;
2208
2209         /* lock down all mappings while we scan labels */
2210         for (i = 0; i < nd_region->ndr_mappings; i++) {
2211                 nd_mapping = &nd_region->mapping[i];
2212                 mutex_lock_nested(&nd_mapping->lock, i);
2213         }
2214
2215         devs = scan_labels(nd_region);
2216
2217         for (i = 0; i < nd_region->ndr_mappings; i++) {
2218                 int reverse = nd_region->ndr_mappings - 1 - i;
2219
2220                 nd_mapping = &nd_region->mapping[reverse];
2221                 mutex_unlock(&nd_mapping->lock);
2222         }
2223
2224         return devs;
2225 }
2226
2227 static int init_active_labels(struct nd_region *nd_region)
2228 {
2229         int i;
2230
2231         for (i = 0; i < nd_region->ndr_mappings; i++) {
2232                 struct nd_mapping *nd_mapping = &nd_region->mapping[i];
2233                 struct nvdimm_drvdata *ndd = to_ndd(nd_mapping);
2234                 struct nvdimm *nvdimm = nd_mapping->nvdimm;
2235                 struct nd_label_ent *label_ent;
2236                 int count, j;
2237
2238                 /*
2239                  * If the dimm is disabled then prevent the region from
2240                  * being activated if it aliases DPA.
2241                  */
2242                 if (!ndd) {
2243                         if ((nvdimm->flags & NDD_ALIASING) == 0)
2244                                 return 0;
2245                         dev_dbg(&nd_region->dev, "%s: is disabled, failing probe\n",
2246                                         dev_name(&nd_mapping->nvdimm->dev));
2247                         return -ENXIO;
2248                 }
2249                 nd_mapping->ndd = ndd;
2250                 atomic_inc(&nvdimm->busy);
2251                 get_ndd(ndd);
2252
2253                 count = nd_label_active_count(ndd);
2254                 dev_dbg(ndd->dev, "%s: %d\n", __func__, count);
2255                 if (!count)
2256                         continue;
2257                 for (j = 0; j < count; j++) {
2258                         struct nd_namespace_label *label;
2259
2260                         label_ent = kzalloc(sizeof(*label_ent), GFP_KERNEL);
2261                         if (!label_ent)
2262                                 break;
2263                         label = nd_label_active(ndd, j);
2264                         label_ent->label = label;
2265
2266                         mutex_lock(&nd_mapping->lock);
2267                         list_add_tail(&label_ent->list, &nd_mapping->labels);
2268                         mutex_unlock(&nd_mapping->lock);
2269                 }
2270
2271                 if (j >= count)
2272                         continue;
2273
2274                 mutex_lock(&nd_mapping->lock);
2275                 nd_mapping_free_labels(nd_mapping);
2276                 mutex_unlock(&nd_mapping->lock);
2277                 return -ENOMEM;
2278         }
2279
2280         return 0;
2281 }
2282
2283 int nd_region_register_namespaces(struct nd_region *nd_region, int *err)
2284 {
2285         struct device **devs = NULL;
2286         int i, rc = 0, type;
2287
2288         *err = 0;
2289         nvdimm_bus_lock(&nd_region->dev);
2290         rc = init_active_labels(nd_region);
2291         if (rc) {
2292                 nvdimm_bus_unlock(&nd_region->dev);
2293                 return rc;
2294         }
2295
2296         type = nd_region_to_nstype(nd_region);
2297         switch (type) {
2298         case ND_DEVICE_NAMESPACE_IO:
2299                 devs = create_namespace_io(nd_region);
2300                 break;
2301         case ND_DEVICE_NAMESPACE_PMEM:
2302         case ND_DEVICE_NAMESPACE_BLK:
2303                 devs = create_namespaces(nd_region);
2304                 break;
2305         default:
2306                 break;
2307         }
2308         nvdimm_bus_unlock(&nd_region->dev);
2309
2310         if (!devs)
2311                 return -ENODEV;
2312
2313         for (i = 0; devs[i]; i++) {
2314                 struct device *dev = devs[i];
2315                 int id;
2316
2317                 if (type == ND_DEVICE_NAMESPACE_BLK) {
2318                         struct nd_namespace_blk *nsblk;
2319
2320                         nsblk = to_nd_namespace_blk(dev);
2321                         id = ida_simple_get(&nd_region->ns_ida, 0, 0,
2322                                         GFP_KERNEL);
2323                         nsblk->id = id;
2324                 } else if (type == ND_DEVICE_NAMESPACE_PMEM) {
2325                         struct nd_namespace_pmem *nspm;
2326
2327                         nspm = to_nd_namespace_pmem(dev);
2328                         id = ida_simple_get(&nd_region->ns_ida, 0, 0,
2329                                         GFP_KERNEL);
2330                         nspm->id = id;
2331                 } else
2332                         id = i;
2333
2334                 if (id < 0)
2335                         break;
2336                 dev_set_name(dev, "namespace%d.%d", nd_region->id, id);
2337                 dev->groups = nd_namespace_attribute_groups;
2338                 nd_device_register(dev);
2339         }
2340         if (i)
2341                 nd_region->ns_seed = devs[0];
2342
2343         if (devs[i]) {
2344                 int j;
2345
2346                 for (j = i; devs[j]; j++) {
2347                         struct device *dev = devs[j];
2348
2349                         device_initialize(dev);
2350                         put_device(dev);
2351                 }
2352                 *err = j - i;
2353                 /*
2354                  * All of the namespaces we tried to register failed, so
2355                  * fail region activation.
2356                  */
2357                 if (*err == 0)
2358                         rc = -ENODEV;
2359         }
2360         kfree(devs);
2361
2362         if (rc == -ENODEV)
2363                 return rc;
2364
2365         return i;
2366 }