14a826c83SDan Williams /* 24a826c83SDan Williams * Copyright(c) 2013-2015 Intel Corporation. All rights reserved. 34a826c83SDan Williams * 44a826c83SDan Williams * This program is free software; you can redistribute it and/or modify 54a826c83SDan Williams * it under the terms of version 2 of the GNU General Public License as 64a826c83SDan Williams * published by the Free Software Foundation. 74a826c83SDan Williams * 84a826c83SDan Williams * This program is distributed in the hope that it will be useful, but 94a826c83SDan Williams * WITHOUT ANY WARRANTY; without even the implied warranty of 104a826c83SDan Williams * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU 114a826c83SDan Williams * General Public License for more details. 124a826c83SDan Williams */ 134a826c83SDan Williams #include <linux/device.h> 144a826c83SDan Williams #include <linux/ndctl.h> 15f524bf27SDan Williams #include <linux/slab.h> 164a826c83SDan Williams #include <linux/io.h> 174a826c83SDan Williams #include <linux/nd.h> 184a826c83SDan Williams #include "nd-core.h" 194a826c83SDan Williams #include "label.h" 204a826c83SDan Williams #include "nd.h" 214a826c83SDan Williams 224a826c83SDan Williams static u32 best_seq(u32 a, u32 b) 234a826c83SDan Williams { 244a826c83SDan Williams a &= NSINDEX_SEQ_MASK; 254a826c83SDan Williams b &= NSINDEX_SEQ_MASK; 264a826c83SDan Williams 274a826c83SDan Williams if (a == 0 || a == b) 284a826c83SDan Williams return b; 294a826c83SDan Williams else if (b == 0) 304a826c83SDan Williams return a; 314a826c83SDan Williams else if (nd_inc_seq(a) == b) 324a826c83SDan Williams return b; 334a826c83SDan Williams else 344a826c83SDan Williams return a; 354a826c83SDan Williams } 364a826c83SDan Williams 37564e871aSDan Williams unsigned sizeof_namespace_label(struct nvdimm_drvdata *ndd) 38564e871aSDan Williams { 39564e871aSDan Williams return ndd->nslabel_size; 40564e871aSDan Williams } 41564e871aSDan Williams 424a826c83SDan Williams size_t sizeof_namespace_index(struct nvdimm_drvdata *ndd) 434a826c83SDan Williams { 444a826c83SDan Williams u32 index_span; 454a826c83SDan Williams 464a826c83SDan Williams if (ndd->nsindex_size) 474a826c83SDan Williams return ndd->nsindex_size; 484a826c83SDan Williams 494a826c83SDan Williams /* 504a826c83SDan Williams * The minimum index space is 512 bytes, with that amount of 514a826c83SDan Williams * index we can describe ~1400 labels which is less than a byte 524a826c83SDan Williams * of overhead per label. Round up to a byte of overhead per 534a826c83SDan Williams * label and determine the size of the index region. Yes, this 544a826c83SDan Williams * starts to waste space at larger config_sizes, but it's 554a826c83SDan Williams * unlikely we'll ever see anything but 128K. 564a826c83SDan Williams */ 57564e871aSDan Williams index_span = ndd->nsarea.config_size / (sizeof_namespace_label(ndd) + 1); 584a826c83SDan Williams index_span /= NSINDEX_ALIGN * 2; 594a826c83SDan Williams ndd->nsindex_size = index_span * NSINDEX_ALIGN; 604a826c83SDan Williams 614a826c83SDan Williams return ndd->nsindex_size; 624a826c83SDan Williams } 634a826c83SDan Williams 640ba1c634SDan Williams int nvdimm_num_label_slots(struct nvdimm_drvdata *ndd) 65f524bf27SDan Williams { 66564e871aSDan Williams return ndd->nsarea.config_size / (sizeof_namespace_label(ndd) + 1); 67f524bf27SDan Williams } 68f524bf27SDan Williams 69564e871aSDan Williams static int __nd_label_validate(struct nvdimm_drvdata *ndd) 704a826c83SDan Williams { 714a826c83SDan Williams /* 724a826c83SDan Williams * On media label format consists of two index blocks followed 734a826c83SDan Williams * by an array of labels. None of these structures are ever 744a826c83SDan Williams * updated in place. A sequence number tracks the current 754a826c83SDan Williams * active index and the next one to write, while labels are 764a826c83SDan Williams * written to free slots. 774a826c83SDan Williams * 784a826c83SDan Williams * +------------+ 794a826c83SDan Williams * | | 804a826c83SDan Williams * | nsindex0 | 814a826c83SDan Williams * | | 824a826c83SDan Williams * +------------+ 834a826c83SDan Williams * | | 844a826c83SDan Williams * | nsindex1 | 854a826c83SDan Williams * | | 864a826c83SDan Williams * +------------+ 874a826c83SDan Williams * | label0 | 884a826c83SDan Williams * +------------+ 894a826c83SDan Williams * | label1 | 904a826c83SDan Williams * +------------+ 914a826c83SDan Williams * | | 924a826c83SDan Williams * ....nslot... 934a826c83SDan Williams * | | 944a826c83SDan Williams * +------------+ 954a826c83SDan Williams * | labelN | 964a826c83SDan Williams * +------------+ 974a826c83SDan Williams */ 984a826c83SDan Williams struct nd_namespace_index *nsindex[] = { 994a826c83SDan Williams to_namespace_index(ndd, 0), 1004a826c83SDan Williams to_namespace_index(ndd, 1), 1014a826c83SDan Williams }; 1024a826c83SDan Williams const int num_index = ARRAY_SIZE(nsindex); 1034a826c83SDan Williams struct device *dev = ndd->dev; 1044a826c83SDan Williams bool valid[2] = { 0 }; 1054a826c83SDan Williams int i, num_valid = 0; 1064a826c83SDan Williams u32 seq; 1074a826c83SDan Williams 1084a826c83SDan Williams for (i = 0; i < num_index; i++) { 1094a826c83SDan Williams u32 nslot; 1104a826c83SDan Williams u8 sig[NSINDEX_SIG_LEN]; 1114a826c83SDan Williams u64 sum_save, sum, size; 112564e871aSDan Williams unsigned int version, labelsize; 1134a826c83SDan Williams 1144a826c83SDan Williams memcpy(sig, nsindex[i]->sig, NSINDEX_SIG_LEN); 1154a826c83SDan Williams if (memcmp(sig, NSINDEX_SIGNATURE, NSINDEX_SIG_LEN) != 0) { 1164a826c83SDan Williams dev_dbg(dev, "%s: nsindex%d signature invalid\n", 1174a826c83SDan Williams __func__, i); 1184a826c83SDan Williams continue; 1194a826c83SDan Williams } 120564e871aSDan Williams 121564e871aSDan Williams /* label sizes larger than 128 arrived with v1.2 */ 122564e871aSDan Williams version = __le16_to_cpu(nsindex[i]->major) * 100 123564e871aSDan Williams + __le16_to_cpu(nsindex[i]->minor); 124564e871aSDan Williams if (version >= 102) 125564e871aSDan Williams labelsize = 1 << (7 + nsindex[i]->labelsize); 126564e871aSDan Williams else 127564e871aSDan Williams labelsize = 128; 128564e871aSDan Williams 129564e871aSDan Williams if (labelsize != sizeof_namespace_label(ndd)) { 130564e871aSDan Williams dev_dbg(dev, "%s: nsindex%d labelsize %d invalid\n", 131564e871aSDan Williams __func__, i, nsindex[i]->labelsize); 132564e871aSDan Williams continue; 133564e871aSDan Williams } 134564e871aSDan Williams 1354a826c83SDan Williams sum_save = __le64_to_cpu(nsindex[i]->checksum); 1364a826c83SDan Williams nsindex[i]->checksum = __cpu_to_le64(0); 1374a826c83SDan Williams sum = nd_fletcher64(nsindex[i], sizeof_namespace_index(ndd), 1); 1384a826c83SDan Williams nsindex[i]->checksum = __cpu_to_le64(sum_save); 1394a826c83SDan Williams if (sum != sum_save) { 1404a826c83SDan Williams dev_dbg(dev, "%s: nsindex%d checksum invalid\n", 1414a826c83SDan Williams __func__, i); 1424a826c83SDan Williams continue; 1434a826c83SDan Williams } 1444a826c83SDan Williams 1454a826c83SDan Williams seq = __le32_to_cpu(nsindex[i]->seq); 1464a826c83SDan Williams if ((seq & NSINDEX_SEQ_MASK) == 0) { 1474a826c83SDan Williams dev_dbg(dev, "%s: nsindex%d sequence: %#x invalid\n", 1484a826c83SDan Williams __func__, i, seq); 1494a826c83SDan Williams continue; 1504a826c83SDan Williams } 1514a826c83SDan Williams 1524a826c83SDan Williams /* sanity check the index against expected values */ 1534a826c83SDan Williams if (__le64_to_cpu(nsindex[i]->myoff) 1544a826c83SDan Williams != i * sizeof_namespace_index(ndd)) { 1554a826c83SDan Williams dev_dbg(dev, "%s: nsindex%d myoff: %#llx invalid\n", 1564a826c83SDan Williams __func__, i, (unsigned long long) 1574a826c83SDan Williams __le64_to_cpu(nsindex[i]->myoff)); 1584a826c83SDan Williams continue; 1594a826c83SDan Williams } 1604a826c83SDan Williams if (__le64_to_cpu(nsindex[i]->otheroff) 1614a826c83SDan Williams != (!i) * sizeof_namespace_index(ndd)) { 1624a826c83SDan Williams dev_dbg(dev, "%s: nsindex%d otheroff: %#llx invalid\n", 1634a826c83SDan Williams __func__, i, (unsigned long long) 1644a826c83SDan Williams __le64_to_cpu(nsindex[i]->otheroff)); 1654a826c83SDan Williams continue; 1664a826c83SDan Williams } 1674a826c83SDan Williams 1684a826c83SDan Williams size = __le64_to_cpu(nsindex[i]->mysize); 1694a826c83SDan Williams if (size > sizeof_namespace_index(ndd) 1704a826c83SDan Williams || size < sizeof(struct nd_namespace_index)) { 1714a826c83SDan Williams dev_dbg(dev, "%s: nsindex%d mysize: %#llx invalid\n", 1724a826c83SDan Williams __func__, i, size); 1734a826c83SDan Williams continue; 1744a826c83SDan Williams } 1754a826c83SDan Williams 1764a826c83SDan Williams nslot = __le32_to_cpu(nsindex[i]->nslot); 177564e871aSDan Williams if (nslot * sizeof_namespace_label(ndd) 1784a826c83SDan Williams + 2 * sizeof_namespace_index(ndd) 1794a826c83SDan Williams > ndd->nsarea.config_size) { 1804a826c83SDan Williams dev_dbg(dev, "%s: nsindex%d nslot: %u invalid, config_size: %#x\n", 1814a826c83SDan Williams __func__, i, nslot, 1824a826c83SDan Williams ndd->nsarea.config_size); 1834a826c83SDan Williams continue; 1844a826c83SDan Williams } 1854a826c83SDan Williams valid[i] = true; 1864a826c83SDan Williams num_valid++; 1874a826c83SDan Williams } 1884a826c83SDan Williams 1894a826c83SDan Williams switch (num_valid) { 1904a826c83SDan Williams case 0: 1914a826c83SDan Williams break; 1924a826c83SDan Williams case 1: 1934a826c83SDan Williams for (i = 0; i < num_index; i++) 1944a826c83SDan Williams if (valid[i]) 1954a826c83SDan Williams return i; 1964a826c83SDan Williams /* can't have num_valid > 0 but valid[] = { false, false } */ 1974a826c83SDan Williams WARN_ON(1); 1984a826c83SDan Williams break; 1994a826c83SDan Williams default: 2004a826c83SDan Williams /* pick the best index... */ 2014a826c83SDan Williams seq = best_seq(__le32_to_cpu(nsindex[0]->seq), 2024a826c83SDan Williams __le32_to_cpu(nsindex[1]->seq)); 2034a826c83SDan Williams if (seq == (__le32_to_cpu(nsindex[1]->seq) & NSINDEX_SEQ_MASK)) 2044a826c83SDan Williams return 1; 2054a826c83SDan Williams else 2064a826c83SDan Williams return 0; 2074a826c83SDan Williams break; 2084a826c83SDan Williams } 2094a826c83SDan Williams 2104a826c83SDan Williams return -1; 2114a826c83SDan Williams } 2124a826c83SDan Williams 213564e871aSDan Williams int nd_label_validate(struct nvdimm_drvdata *ndd) 214564e871aSDan Williams { 215564e871aSDan Williams /* 216564e871aSDan Williams * In order to probe for and validate namespace index blocks we 217564e871aSDan Williams * need to know the size of the labels, and we can't trust the 218564e871aSDan Williams * size of the labels until we validate the index blocks. 219564e871aSDan Williams * Resolve this dependency loop by probing for known label 220564e871aSDan Williams * sizes. 221564e871aSDan Williams */ 222564e871aSDan Williams int label_size[] = { 256, 128 }; 223564e871aSDan Williams int i, rc; 224564e871aSDan Williams 225564e871aSDan Williams for (i = 0; i < ARRAY_SIZE(label_size); i++) { 226564e871aSDan Williams ndd->nslabel_size = label_size[i]; 227564e871aSDan Williams rc = __nd_label_validate(ndd); 228564e871aSDan Williams if (rc >= 0) 229564e871aSDan Williams return rc; 230564e871aSDan Williams } 231564e871aSDan Williams 232564e871aSDan Williams return -1; 233564e871aSDan Williams } 234564e871aSDan Williams 2354a826c83SDan Williams void nd_label_copy(struct nvdimm_drvdata *ndd, struct nd_namespace_index *dst, 2364a826c83SDan Williams struct nd_namespace_index *src) 2374a826c83SDan Williams { 2384a826c83SDan Williams if (dst && src) 2394a826c83SDan Williams /* pass */; 2404a826c83SDan Williams else 2414a826c83SDan Williams return; 2424a826c83SDan Williams 2434a826c83SDan Williams memcpy(dst, src, sizeof_namespace_index(ndd)); 2444a826c83SDan Williams } 2454a826c83SDan Williams 2464a826c83SDan Williams static struct nd_namespace_label *nd_label_base(struct nvdimm_drvdata *ndd) 2474a826c83SDan Williams { 2484a826c83SDan Williams void *base = to_namespace_index(ndd, 0); 2494a826c83SDan Williams 2504a826c83SDan Williams return base + 2 * sizeof_namespace_index(ndd); 2514a826c83SDan Williams } 2524a826c83SDan Williams 253f524bf27SDan Williams static int to_slot(struct nvdimm_drvdata *ndd, 254f524bf27SDan Williams struct nd_namespace_label *nd_label) 255f524bf27SDan Williams { 256564e871aSDan Williams unsigned long label, base; 257564e871aSDan Williams 258564e871aSDan Williams label = (unsigned long) nd_label; 259564e871aSDan Williams base = (unsigned long) nd_label_base(ndd); 260564e871aSDan Williams 261564e871aSDan Williams return (label - base) / sizeof_namespace_label(ndd); 262564e871aSDan Williams } 263564e871aSDan Williams 264564e871aSDan Williams static struct nd_namespace_label *to_label(struct nvdimm_drvdata *ndd, int slot) 265564e871aSDan Williams { 266564e871aSDan Williams unsigned long label, base; 267564e871aSDan Williams 268564e871aSDan Williams base = (unsigned long) nd_label_base(ndd); 269564e871aSDan Williams label = base + sizeof_namespace_label(ndd) * slot; 270564e871aSDan Williams 271564e871aSDan Williams return (struct nd_namespace_label *) label; 272f524bf27SDan Williams } 273f524bf27SDan Williams 2744a826c83SDan Williams #define for_each_clear_bit_le(bit, addr, size) \ 2754a826c83SDan Williams for ((bit) = find_next_zero_bit_le((addr), (size), 0); \ 2764a826c83SDan Williams (bit) < (size); \ 2774a826c83SDan Williams (bit) = find_next_zero_bit_le((addr), (size), (bit) + 1)) 2784a826c83SDan Williams 2794a826c83SDan Williams /** 280f524bf27SDan Williams * preamble_index - common variable initialization for nd_label_* routines 2814a826c83SDan Williams * @ndd: dimm container for the relevant label set 282f524bf27SDan Williams * @idx: namespace_index index 2834a826c83SDan Williams * @nsindex_out: on return set to the currently active namespace index 2844a826c83SDan Williams * @free: on return set to the free label bitmap in the index 2854a826c83SDan Williams * @nslot: on return set to the number of slots in the label space 2864a826c83SDan Williams */ 287f524bf27SDan Williams static bool preamble_index(struct nvdimm_drvdata *ndd, int idx, 2884a826c83SDan Williams struct nd_namespace_index **nsindex_out, 2894a826c83SDan Williams unsigned long **free, u32 *nslot) 2904a826c83SDan Williams { 2914a826c83SDan Williams struct nd_namespace_index *nsindex; 2924a826c83SDan Williams 293f524bf27SDan Williams nsindex = to_namespace_index(ndd, idx); 2944a826c83SDan Williams if (nsindex == NULL) 2954a826c83SDan Williams return false; 2964a826c83SDan Williams 2974a826c83SDan Williams *free = (unsigned long *) nsindex->free; 2984a826c83SDan Williams *nslot = __le32_to_cpu(nsindex->nslot); 2994a826c83SDan Williams *nsindex_out = nsindex; 3004a826c83SDan Williams 3014a826c83SDan Williams return true; 3024a826c83SDan Williams } 3034a826c83SDan Williams 304bf9bccc1SDan Williams char *nd_label_gen_id(struct nd_label_id *label_id, u8 *uuid, u32 flags) 3054a826c83SDan Williams { 3064a826c83SDan Williams if (!label_id || !uuid) 3074a826c83SDan Williams return NULL; 3084a826c83SDan Williams snprintf(label_id->id, ND_LABEL_ID_SIZE, "%s-%pUb", 3094a826c83SDan Williams flags & NSLABEL_FLAG_LOCAL ? "blk" : "pmem", uuid); 3104a826c83SDan Williams return label_id->id; 3114a826c83SDan Williams } 3124a826c83SDan Williams 313f524bf27SDan Williams static bool preamble_current(struct nvdimm_drvdata *ndd, 314f524bf27SDan Williams struct nd_namespace_index **nsindex, 315f524bf27SDan Williams unsigned long **free, u32 *nslot) 316f524bf27SDan Williams { 317f524bf27SDan Williams return preamble_index(ndd, ndd->ns_current, nsindex, 318f524bf27SDan Williams free, nslot); 319f524bf27SDan Williams } 320f524bf27SDan Williams 321f524bf27SDan Williams static bool preamble_next(struct nvdimm_drvdata *ndd, 322f524bf27SDan Williams struct nd_namespace_index **nsindex, 323f524bf27SDan Williams unsigned long **free, u32 *nslot) 324f524bf27SDan Williams { 325f524bf27SDan Williams return preamble_index(ndd, ndd->ns_next, nsindex, 326f524bf27SDan Williams free, nslot); 327f524bf27SDan Williams } 328f524bf27SDan Williams 3294a826c83SDan Williams static bool slot_valid(struct nd_namespace_label *nd_label, u32 slot) 3304a826c83SDan Williams { 3314a826c83SDan Williams /* check that we are written where we expect to be written */ 3324a826c83SDan Williams if (slot != __le32_to_cpu(nd_label->slot)) 3334a826c83SDan Williams return false; 3344a826c83SDan Williams 3354a826c83SDan Williams /* check that DPA allocations are page aligned */ 3364a826c83SDan Williams if ((__le64_to_cpu(nd_label->dpa) 3374a826c83SDan Williams | __le64_to_cpu(nd_label->rawsize)) % SZ_4K) 3384a826c83SDan Williams return false; 3394a826c83SDan Williams 3404a826c83SDan Williams return true; 3414a826c83SDan Williams } 3424a826c83SDan Williams 3434a826c83SDan Williams int nd_label_reserve_dpa(struct nvdimm_drvdata *ndd) 3444a826c83SDan Williams { 3454a826c83SDan Williams struct nd_namespace_index *nsindex; 3464a826c83SDan Williams unsigned long *free; 3474a826c83SDan Williams u32 nslot, slot; 3484a826c83SDan Williams 3494a826c83SDan Williams if (!preamble_current(ndd, &nsindex, &free, &nslot)) 3504a826c83SDan Williams return 0; /* no label, nothing to reserve */ 3514a826c83SDan Williams 3524a826c83SDan Williams for_each_clear_bit_le(slot, free, nslot) { 3534a826c83SDan Williams struct nd_namespace_label *nd_label; 3544a826c83SDan Williams struct nd_region *nd_region = NULL; 3554a826c83SDan Williams u8 label_uuid[NSLABEL_UUID_LEN]; 3564a826c83SDan Williams struct nd_label_id label_id; 3574a826c83SDan Williams struct resource *res; 3584a826c83SDan Williams u32 flags; 3594a826c83SDan Williams 360564e871aSDan Williams nd_label = to_label(ndd, slot); 3614a826c83SDan Williams 3624a826c83SDan Williams if (!slot_valid(nd_label, slot)) 3634a826c83SDan Williams continue; 3644a826c83SDan Williams 3654a826c83SDan Williams memcpy(label_uuid, nd_label->uuid, NSLABEL_UUID_LEN); 3664a826c83SDan Williams flags = __le32_to_cpu(nd_label->flags); 3674a826c83SDan Williams nd_label_gen_id(&label_id, label_uuid, flags); 3684a826c83SDan Williams res = nvdimm_allocate_dpa(ndd, &label_id, 3694a826c83SDan Williams __le64_to_cpu(nd_label->dpa), 3704a826c83SDan Williams __le64_to_cpu(nd_label->rawsize)); 3714a826c83SDan Williams nd_dbg_dpa(nd_region, ndd, res, "reserve\n"); 3724a826c83SDan Williams if (!res) 3734a826c83SDan Williams return -EBUSY; 3744a826c83SDan Williams } 3754a826c83SDan Williams 3764a826c83SDan Williams return 0; 3774a826c83SDan Williams } 378bf9bccc1SDan Williams 379bf9bccc1SDan Williams int nd_label_active_count(struct nvdimm_drvdata *ndd) 380bf9bccc1SDan Williams { 381bf9bccc1SDan Williams struct nd_namespace_index *nsindex; 382bf9bccc1SDan Williams unsigned long *free; 383bf9bccc1SDan Williams u32 nslot, slot; 384bf9bccc1SDan Williams int count = 0; 385bf9bccc1SDan Williams 386bf9bccc1SDan Williams if (!preamble_current(ndd, &nsindex, &free, &nslot)) 387bf9bccc1SDan Williams return 0; 388bf9bccc1SDan Williams 389bf9bccc1SDan Williams for_each_clear_bit_le(slot, free, nslot) { 390bf9bccc1SDan Williams struct nd_namespace_label *nd_label; 391bf9bccc1SDan Williams 392564e871aSDan Williams nd_label = to_label(ndd, slot); 393bf9bccc1SDan Williams 394bf9bccc1SDan Williams if (!slot_valid(nd_label, slot)) { 395bf9bccc1SDan Williams u32 label_slot = __le32_to_cpu(nd_label->slot); 396bf9bccc1SDan Williams u64 size = __le64_to_cpu(nd_label->rawsize); 397bf9bccc1SDan Williams u64 dpa = __le64_to_cpu(nd_label->dpa); 398bf9bccc1SDan Williams 399bf9bccc1SDan Williams dev_dbg(ndd->dev, 400bf9bccc1SDan Williams "%s: slot%d invalid slot: %d dpa: %llx size: %llx\n", 401bf9bccc1SDan Williams __func__, slot, label_slot, dpa, size); 402bf9bccc1SDan Williams continue; 403bf9bccc1SDan Williams } 404bf9bccc1SDan Williams count++; 405bf9bccc1SDan Williams } 406bf9bccc1SDan Williams return count; 407bf9bccc1SDan Williams } 408bf9bccc1SDan Williams 409bf9bccc1SDan Williams struct nd_namespace_label *nd_label_active(struct nvdimm_drvdata *ndd, int n) 410bf9bccc1SDan Williams { 411bf9bccc1SDan Williams struct nd_namespace_index *nsindex; 412bf9bccc1SDan Williams unsigned long *free; 413bf9bccc1SDan Williams u32 nslot, slot; 414bf9bccc1SDan Williams 415bf9bccc1SDan Williams if (!preamble_current(ndd, &nsindex, &free, &nslot)) 416bf9bccc1SDan Williams return NULL; 417bf9bccc1SDan Williams 418bf9bccc1SDan Williams for_each_clear_bit_le(slot, free, nslot) { 419bf9bccc1SDan Williams struct nd_namespace_label *nd_label; 420bf9bccc1SDan Williams 421564e871aSDan Williams nd_label = to_label(ndd, slot); 422bf9bccc1SDan Williams if (!slot_valid(nd_label, slot)) 423bf9bccc1SDan Williams continue; 424bf9bccc1SDan Williams 425bf9bccc1SDan Williams if (n-- == 0) 426564e871aSDan Williams return to_label(ndd, slot); 427bf9bccc1SDan Williams } 428bf9bccc1SDan Williams 429bf9bccc1SDan Williams return NULL; 430bf9bccc1SDan Williams } 431f524bf27SDan Williams 4320ba1c634SDan Williams u32 nd_label_alloc_slot(struct nvdimm_drvdata *ndd) 433f524bf27SDan Williams { 434f524bf27SDan Williams struct nd_namespace_index *nsindex; 435f524bf27SDan Williams unsigned long *free; 436f524bf27SDan Williams u32 nslot, slot; 437f524bf27SDan Williams 438f524bf27SDan Williams if (!preamble_next(ndd, &nsindex, &free, &nslot)) 439f524bf27SDan Williams return UINT_MAX; 440f524bf27SDan Williams 441f524bf27SDan Williams WARN_ON(!is_nvdimm_bus_locked(ndd->dev)); 442f524bf27SDan Williams 443f524bf27SDan Williams slot = find_next_bit_le(free, nslot, 0); 444f524bf27SDan Williams if (slot == nslot) 445f524bf27SDan Williams return UINT_MAX; 446f524bf27SDan Williams 447f524bf27SDan Williams clear_bit_le(slot, free); 448f524bf27SDan Williams 449f524bf27SDan Williams return slot; 450f524bf27SDan Williams } 451f524bf27SDan Williams 4520ba1c634SDan Williams bool nd_label_free_slot(struct nvdimm_drvdata *ndd, u32 slot) 453f524bf27SDan Williams { 454f524bf27SDan Williams struct nd_namespace_index *nsindex; 455f524bf27SDan Williams unsigned long *free; 456f524bf27SDan Williams u32 nslot; 457f524bf27SDan Williams 458f524bf27SDan Williams if (!preamble_next(ndd, &nsindex, &free, &nslot)) 459f524bf27SDan Williams return false; 460f524bf27SDan Williams 461f524bf27SDan Williams WARN_ON(!is_nvdimm_bus_locked(ndd->dev)); 462f524bf27SDan Williams 463f524bf27SDan Williams if (slot < nslot) 464f524bf27SDan Williams return !test_and_set_bit_le(slot, free); 465f524bf27SDan Williams return false; 466f524bf27SDan Williams } 467f524bf27SDan Williams 468f524bf27SDan Williams u32 nd_label_nfree(struct nvdimm_drvdata *ndd) 469f524bf27SDan Williams { 470f524bf27SDan Williams struct nd_namespace_index *nsindex; 471f524bf27SDan Williams unsigned long *free; 472f524bf27SDan Williams u32 nslot; 473f524bf27SDan Williams 474f524bf27SDan Williams WARN_ON(!is_nvdimm_bus_locked(ndd->dev)); 475f524bf27SDan Williams 476f524bf27SDan Williams if (!preamble_next(ndd, &nsindex, &free, &nslot)) 4770ba1c634SDan Williams return nvdimm_num_label_slots(ndd); 478f524bf27SDan Williams 479f524bf27SDan Williams return bitmap_weight(free, nslot); 480f524bf27SDan Williams } 481f524bf27SDan Williams 482f524bf27SDan Williams static int nd_label_write_index(struct nvdimm_drvdata *ndd, int index, u32 seq, 483f524bf27SDan Williams unsigned long flags) 484f524bf27SDan Williams { 485f524bf27SDan Williams struct nd_namespace_index *nsindex; 486f524bf27SDan Williams unsigned long offset; 487f524bf27SDan Williams u64 checksum; 488f524bf27SDan Williams u32 nslot; 489f524bf27SDan Williams int rc; 490f524bf27SDan Williams 491f524bf27SDan Williams nsindex = to_namespace_index(ndd, index); 492f524bf27SDan Williams if (flags & ND_NSINDEX_INIT) 493f524bf27SDan Williams nslot = nvdimm_num_label_slots(ndd); 494f524bf27SDan Williams else 495f524bf27SDan Williams nslot = __le32_to_cpu(nsindex->nslot); 496f524bf27SDan Williams 497f524bf27SDan Williams memcpy(nsindex->sig, NSINDEX_SIGNATURE, NSINDEX_SIG_LEN); 498564e871aSDan Williams memset(&nsindex->flags, 0, 3); 499564e871aSDan Williams nsindex->labelsize = sizeof_namespace_label(ndd) >> 8; 500f524bf27SDan Williams nsindex->seq = __cpu_to_le32(seq); 501f524bf27SDan Williams offset = (unsigned long) nsindex 502f524bf27SDan Williams - (unsigned long) to_namespace_index(ndd, 0); 503f524bf27SDan Williams nsindex->myoff = __cpu_to_le64(offset); 504f524bf27SDan Williams nsindex->mysize = __cpu_to_le64(sizeof_namespace_index(ndd)); 505f524bf27SDan Williams offset = (unsigned long) to_namespace_index(ndd, 506f524bf27SDan Williams nd_label_next_nsindex(index)) 507f524bf27SDan Williams - (unsigned long) to_namespace_index(ndd, 0); 508f524bf27SDan Williams nsindex->otheroff = __cpu_to_le64(offset); 509f524bf27SDan Williams offset = (unsigned long) nd_label_base(ndd) 510f524bf27SDan Williams - (unsigned long) to_namespace_index(ndd, 0); 511f524bf27SDan Williams nsindex->labeloff = __cpu_to_le64(offset); 512f524bf27SDan Williams nsindex->nslot = __cpu_to_le32(nslot); 513f524bf27SDan Williams nsindex->major = __cpu_to_le16(1); 514f524bf27SDan Williams nsindex->minor = __cpu_to_le16(1); 515f524bf27SDan Williams nsindex->checksum = __cpu_to_le64(0); 516f524bf27SDan Williams if (flags & ND_NSINDEX_INIT) { 517f524bf27SDan Williams unsigned long *free = (unsigned long *) nsindex->free; 518f524bf27SDan Williams u32 nfree = ALIGN(nslot, BITS_PER_LONG); 519f524bf27SDan Williams int last_bits, i; 520f524bf27SDan Williams 521f524bf27SDan Williams memset(nsindex->free, 0xff, nfree / 8); 522f524bf27SDan Williams for (i = 0, last_bits = nfree - nslot; i < last_bits; i++) 523f524bf27SDan Williams clear_bit_le(nslot + i, free); 524f524bf27SDan Williams } 525f524bf27SDan Williams checksum = nd_fletcher64(nsindex, sizeof_namespace_index(ndd), 1); 526f524bf27SDan Williams nsindex->checksum = __cpu_to_le64(checksum); 527f524bf27SDan Williams rc = nvdimm_set_config_data(ndd, __le64_to_cpu(nsindex->myoff), 528f524bf27SDan Williams nsindex, sizeof_namespace_index(ndd)); 529f524bf27SDan Williams if (rc < 0) 530f524bf27SDan Williams return rc; 531f524bf27SDan Williams 532f524bf27SDan Williams if (flags & ND_NSINDEX_INIT) 533f524bf27SDan Williams return 0; 534f524bf27SDan Williams 535f524bf27SDan Williams /* copy the index we just wrote to the new 'next' */ 536f524bf27SDan Williams WARN_ON(index != ndd->ns_next); 537f524bf27SDan Williams nd_label_copy(ndd, to_current_namespace_index(ndd), nsindex); 538f524bf27SDan Williams ndd->ns_current = nd_label_next_nsindex(ndd->ns_current); 539f524bf27SDan Williams ndd->ns_next = nd_label_next_nsindex(ndd->ns_next); 540f524bf27SDan Williams WARN_ON(ndd->ns_current == ndd->ns_next); 541f524bf27SDan Williams 542f524bf27SDan Williams return 0; 543f524bf27SDan Williams } 544f524bf27SDan Williams 545f524bf27SDan Williams static unsigned long nd_label_offset(struct nvdimm_drvdata *ndd, 546f524bf27SDan Williams struct nd_namespace_label *nd_label) 547f524bf27SDan Williams { 548f524bf27SDan Williams return (unsigned long) nd_label 549f524bf27SDan Williams - (unsigned long) to_namespace_index(ndd, 0); 550f524bf27SDan Williams } 551f524bf27SDan Williams 552f524bf27SDan Williams static int __pmem_label_update(struct nd_region *nd_region, 553f524bf27SDan Williams struct nd_mapping *nd_mapping, struct nd_namespace_pmem *nspm, 554f524bf27SDan Williams int pos) 555f524bf27SDan Williams { 556*faec6f8aSDan Williams struct nd_interleave_set *nd_set = nd_region->nd_set; 557f524bf27SDan Williams struct nvdimm_drvdata *ndd = to_ndd(nd_mapping); 55816660eaeSDan Williams struct nd_label_ent *label_ent, *victim = NULL; 559f524bf27SDan Williams struct nd_namespace_label *nd_label; 560f524bf27SDan Williams struct nd_namespace_index *nsindex; 56116660eaeSDan Williams struct nd_label_id label_id; 56216660eaeSDan Williams struct resource *res; 563f524bf27SDan Williams unsigned long *free; 564f524bf27SDan Williams u32 nslot, slot; 565f524bf27SDan Williams size_t offset; 566c12c48ceSDan Williams u64 cookie; 567f524bf27SDan Williams int rc; 568f524bf27SDan Williams 569f524bf27SDan Williams if (!preamble_next(ndd, &nsindex, &free, &nslot)) 570f524bf27SDan Williams return -ENXIO; 571f524bf27SDan Williams 572c12c48ceSDan Williams cookie = nd_region_interleave_set_cookie(nd_region, nsindex); 57316660eaeSDan Williams nd_label_gen_id(&label_id, nspm->uuid, 0); 57416660eaeSDan Williams for_each_dpa_resource(ndd, res) 57516660eaeSDan Williams if (strcmp(res->name, label_id.id) == 0) 57616660eaeSDan Williams break; 57716660eaeSDan Williams 57816660eaeSDan Williams if (!res) { 57916660eaeSDan Williams WARN_ON_ONCE(1); 58016660eaeSDan Williams return -ENXIO; 58116660eaeSDan Williams } 58216660eaeSDan Williams 583f524bf27SDan Williams /* allocate and write the label to the staging (next) index */ 584f524bf27SDan Williams slot = nd_label_alloc_slot(ndd); 585f524bf27SDan Williams if (slot == UINT_MAX) 586f524bf27SDan Williams return -ENXIO; 587f524bf27SDan Williams dev_dbg(ndd->dev, "%s: allocated: %d\n", __func__, slot); 588f524bf27SDan Williams 589564e871aSDan Williams nd_label = to_label(ndd, slot); 590564e871aSDan Williams memset(nd_label, 0, sizeof_namespace_label(ndd)); 591f524bf27SDan Williams memcpy(nd_label->uuid, nspm->uuid, NSLABEL_UUID_LEN); 592f524bf27SDan Williams if (nspm->alt_name) 593f524bf27SDan Williams memcpy(nd_label->name, nspm->alt_name, NSLABEL_NAME_LEN); 594f524bf27SDan Williams nd_label->flags = __cpu_to_le32(NSLABEL_FLAG_UPDATING); 595f524bf27SDan Williams nd_label->nlabel = __cpu_to_le16(nd_region->ndr_mappings); 596f524bf27SDan Williams nd_label->position = __cpu_to_le16(pos); 597f524bf27SDan Williams nd_label->isetcookie = __cpu_to_le64(cookie); 59816660eaeSDan Williams nd_label->rawsize = __cpu_to_le64(resource_size(res)); 59916660eaeSDan Williams nd_label->dpa = __cpu_to_le64(res->start); 600f524bf27SDan Williams nd_label->slot = __cpu_to_le32(slot); 601*faec6f8aSDan Williams if (namespace_label_has(ndd, type_guid)) 602*faec6f8aSDan Williams guid_copy(&nd_label->type_guid, &nd_set->type_guid); 60316660eaeSDan Williams nd_dbg_dpa(nd_region, ndd, res, "%s\n", __func__); 604f524bf27SDan Williams 605f524bf27SDan Williams /* update label */ 606f524bf27SDan Williams offset = nd_label_offset(ndd, nd_label); 607f524bf27SDan Williams rc = nvdimm_set_config_data(ndd, offset, nd_label, 608564e871aSDan Williams sizeof_namespace_label(ndd)); 609f524bf27SDan Williams if (rc < 0) 610f524bf27SDan Williams return rc; 611f524bf27SDan Williams 612f524bf27SDan Williams /* Garbage collect the previous label */ 613ae8219f1SDan Williams mutex_lock(&nd_mapping->lock); 61416660eaeSDan Williams list_for_each_entry(label_ent, &nd_mapping->labels, list) { 61516660eaeSDan Williams if (!label_ent->label) 61616660eaeSDan Williams continue; 61716660eaeSDan Williams if (memcmp(nspm->uuid, label_ent->label->uuid, 61816660eaeSDan Williams NSLABEL_UUID_LEN) != 0) 61916660eaeSDan Williams continue; 62016660eaeSDan Williams victim = label_ent; 62116660eaeSDan Williams list_move_tail(&victim->list, &nd_mapping->labels); 62216660eaeSDan Williams break; 62316660eaeSDan Williams } 62416660eaeSDan Williams if (victim) { 625f524bf27SDan Williams dev_dbg(ndd->dev, "%s: free: %d\n", __func__, slot); 62616660eaeSDan Williams slot = to_slot(ndd, victim->label); 62716660eaeSDan Williams nd_label_free_slot(ndd, slot); 62816660eaeSDan Williams victim->label = NULL; 629f524bf27SDan Williams } 630f524bf27SDan Williams 631f524bf27SDan Williams /* update index */ 632f524bf27SDan Williams rc = nd_label_write_index(ndd, ndd->ns_next, 633f524bf27SDan Williams nd_inc_seq(__le32_to_cpu(nsindex->seq)), 0); 63416660eaeSDan Williams if (rc == 0) { 63516660eaeSDan Williams list_for_each_entry(label_ent, &nd_mapping->labels, list) 63616660eaeSDan Williams if (!label_ent->label) { 637ae8219f1SDan Williams label_ent->label = nd_label; 63816660eaeSDan Williams nd_label = NULL; 63916660eaeSDan Williams break; 64016660eaeSDan Williams } 64116660eaeSDan Williams dev_WARN_ONCE(&nspm->nsio.common.dev, nd_label, 64216660eaeSDan Williams "failed to track label: %d\n", 64316660eaeSDan Williams to_slot(ndd, nd_label)); 64416660eaeSDan Williams if (nd_label) 64516660eaeSDan Williams rc = -ENXIO; 64616660eaeSDan Williams } 647ae8219f1SDan Williams mutex_unlock(&nd_mapping->lock); 648ae8219f1SDan Williams 649f524bf27SDan Williams return rc; 6500ba1c634SDan Williams } 6510ba1c634SDan Williams 6520ba1c634SDan Williams static bool is_old_resource(struct resource *res, struct resource **list, int n) 653f524bf27SDan Williams { 654f524bf27SDan Williams int i; 6550ba1c634SDan Williams 6560ba1c634SDan Williams if (res->flags & DPA_RESOURCE_ADJUSTED) 6570ba1c634SDan Williams return false; 6580ba1c634SDan Williams for (i = 0; i < n; i++) 6590ba1c634SDan Williams if (res == list[i]) 6600ba1c634SDan Williams return true; 6610ba1c634SDan Williams return false; 6620ba1c634SDan Williams } 6630ba1c634SDan Williams 6640ba1c634SDan Williams static struct resource *to_resource(struct nvdimm_drvdata *ndd, 6650ba1c634SDan Williams struct nd_namespace_label *nd_label) 6660ba1c634SDan Williams { 6670ba1c634SDan Williams struct resource *res; 6680ba1c634SDan Williams 6690ba1c634SDan Williams for_each_dpa_resource(ndd, res) { 6700ba1c634SDan Williams if (res->start != __le64_to_cpu(nd_label->dpa)) 6710ba1c634SDan Williams continue; 6720ba1c634SDan Williams if (resource_size(res) != __le64_to_cpu(nd_label->rawsize)) 6730ba1c634SDan Williams continue; 6740ba1c634SDan Williams return res; 6750ba1c634SDan Williams } 6760ba1c634SDan Williams 6770ba1c634SDan Williams return NULL; 6780ba1c634SDan Williams } 6790ba1c634SDan Williams 6800ba1c634SDan Williams /* 6810ba1c634SDan Williams * 1/ Account all the labels that can be freed after this update 6820ba1c634SDan Williams * 2/ Allocate and write the label to the staging (next) index 6830ba1c634SDan Williams * 3/ Record the resources in the namespace device 6840ba1c634SDan Williams */ 6850ba1c634SDan Williams static int __blk_label_update(struct nd_region *nd_region, 6860ba1c634SDan Williams struct nd_mapping *nd_mapping, struct nd_namespace_blk *nsblk, 6870ba1c634SDan Williams int num_labels) 6880ba1c634SDan Williams { 689ae8219f1SDan Williams int i, alloc, victims, nfree, old_num_resources, nlabel, rc = -ENXIO; 690*faec6f8aSDan Williams struct nd_interleave_set *nd_set = nd_region->nd_set; 691f524bf27SDan Williams struct nvdimm_drvdata *ndd = to_ndd(nd_mapping); 6920ba1c634SDan Williams struct nd_namespace_label *nd_label; 693ae8219f1SDan Williams struct nd_label_ent *label_ent, *e; 6940ba1c634SDan Williams struct nd_namespace_index *nsindex; 6950ba1c634SDan Williams unsigned long *free, *victim_map = NULL; 6960ba1c634SDan Williams struct resource *res, **old_res_list; 6970ba1c634SDan Williams struct nd_label_id label_id; 6980ba1c634SDan Williams u8 uuid[NSLABEL_UUID_LEN]; 699ae8219f1SDan Williams LIST_HEAD(list); 7000ba1c634SDan Williams u32 nslot, slot; 701f524bf27SDan Williams 7020ba1c634SDan Williams if (!preamble_next(ndd, &nsindex, &free, &nslot)) 7030ba1c634SDan Williams return -ENXIO; 704f524bf27SDan Williams 7050ba1c634SDan Williams old_res_list = nsblk->res; 7060ba1c634SDan Williams nfree = nd_label_nfree(ndd); 7070ba1c634SDan Williams old_num_resources = nsblk->num_resources; 7080ba1c634SDan Williams nd_label_gen_id(&label_id, nsblk->uuid, NSLABEL_FLAG_LOCAL); 7090ba1c634SDan Williams 7100ba1c634SDan Williams /* 7110ba1c634SDan Williams * We need to loop over the old resources a few times, which seems a 7120ba1c634SDan Williams * bit inefficient, but we need to know that we have the label 7130ba1c634SDan Williams * space before we start mutating the tracking structures. 7140ba1c634SDan Williams * Otherwise the recovery method of last resort for userspace is 7150ba1c634SDan Williams * disable and re-enable the parent region. 7160ba1c634SDan Williams */ 7170ba1c634SDan Williams alloc = 0; 7180ba1c634SDan Williams for_each_dpa_resource(ndd, res) { 7190ba1c634SDan Williams if (strcmp(res->name, label_id.id) != 0) 7200ba1c634SDan Williams continue; 7210ba1c634SDan Williams if (!is_old_resource(res, old_res_list, old_num_resources)) 7220ba1c634SDan Williams alloc++; 7230ba1c634SDan Williams } 7240ba1c634SDan Williams 7250ba1c634SDan Williams victims = 0; 7260ba1c634SDan Williams if (old_num_resources) { 7270ba1c634SDan Williams /* convert old local-label-map to dimm-slot victim-map */ 7280ba1c634SDan Williams victim_map = kcalloc(BITS_TO_LONGS(nslot), sizeof(long), 7290ba1c634SDan Williams GFP_KERNEL); 7300ba1c634SDan Williams if (!victim_map) 7310ba1c634SDan Williams return -ENOMEM; 7320ba1c634SDan Williams 7330ba1c634SDan Williams /* mark unused labels for garbage collection */ 7340ba1c634SDan Williams for_each_clear_bit_le(slot, free, nslot) { 735564e871aSDan Williams nd_label = to_label(ndd, slot); 7360ba1c634SDan Williams memcpy(uuid, nd_label->uuid, NSLABEL_UUID_LEN); 7370ba1c634SDan Williams if (memcmp(uuid, nsblk->uuid, NSLABEL_UUID_LEN) != 0) 7380ba1c634SDan Williams continue; 7390ba1c634SDan Williams res = to_resource(ndd, nd_label); 7400ba1c634SDan Williams if (res && is_old_resource(res, old_res_list, 7410ba1c634SDan Williams old_num_resources)) 7420ba1c634SDan Williams continue; 7430ba1c634SDan Williams slot = to_slot(ndd, nd_label); 7440ba1c634SDan Williams set_bit(slot, victim_map); 7450ba1c634SDan Williams victims++; 7460ba1c634SDan Williams } 7470ba1c634SDan Williams } 7480ba1c634SDan Williams 7490ba1c634SDan Williams /* don't allow updates that consume the last label */ 7500ba1c634SDan Williams if (nfree - alloc < 0 || nfree - alloc + victims < 1) { 7518c2f7e86SDan Williams dev_info(&nsblk->common.dev, "insufficient label space\n"); 7520ba1c634SDan Williams kfree(victim_map); 7530ba1c634SDan Williams return -ENOSPC; 7540ba1c634SDan Williams } 7550ba1c634SDan Williams /* from here on we need to abort on error */ 7560ba1c634SDan Williams 7570ba1c634SDan Williams 7580ba1c634SDan Williams /* assign all resources to the namespace before writing the labels */ 7590ba1c634SDan Williams nsblk->res = NULL; 7600ba1c634SDan Williams nsblk->num_resources = 0; 7610ba1c634SDan Williams for_each_dpa_resource(ndd, res) { 7620ba1c634SDan Williams if (strcmp(res->name, label_id.id) != 0) 7630ba1c634SDan Williams continue; 7640ba1c634SDan Williams if (!nsblk_add_resource(nd_region, ndd, nsblk, res->start)) { 7650ba1c634SDan Williams rc = -ENOMEM; 7660ba1c634SDan Williams goto abort; 7670ba1c634SDan Williams } 7680ba1c634SDan Williams } 7690ba1c634SDan Williams 7700ba1c634SDan Williams for (i = 0; i < nsblk->num_resources; i++) { 7710ba1c634SDan Williams size_t offset; 7720ba1c634SDan Williams 7730ba1c634SDan Williams res = nsblk->res[i]; 7740ba1c634SDan Williams if (is_old_resource(res, old_res_list, old_num_resources)) 7750ba1c634SDan Williams continue; /* carry-over */ 7760ba1c634SDan Williams slot = nd_label_alloc_slot(ndd); 7770ba1c634SDan Williams if (slot == UINT_MAX) 7780ba1c634SDan Williams goto abort; 7790ba1c634SDan Williams dev_dbg(ndd->dev, "%s: allocated: %d\n", __func__, slot); 7800ba1c634SDan Williams 781564e871aSDan Williams nd_label = to_label(ndd, slot); 782564e871aSDan Williams memset(nd_label, 0, sizeof_namespace_label(ndd)); 7830ba1c634SDan Williams memcpy(nd_label->uuid, nsblk->uuid, NSLABEL_UUID_LEN); 7840ba1c634SDan Williams if (nsblk->alt_name) 7850ba1c634SDan Williams memcpy(nd_label->name, nsblk->alt_name, 7860ba1c634SDan Williams NSLABEL_NAME_LEN); 7870ba1c634SDan Williams nd_label->flags = __cpu_to_le32(NSLABEL_FLAG_LOCAL); 7880ba1c634SDan Williams nd_label->nlabel = __cpu_to_le16(0); /* N/A */ 7890ba1c634SDan Williams nd_label->position = __cpu_to_le16(0); /* N/A */ 7900ba1c634SDan Williams nd_label->isetcookie = __cpu_to_le64(0); /* N/A */ 7910ba1c634SDan Williams nd_label->dpa = __cpu_to_le64(res->start); 7920ba1c634SDan Williams nd_label->rawsize = __cpu_to_le64(resource_size(res)); 7930ba1c634SDan Williams nd_label->lbasize = __cpu_to_le64(nsblk->lbasize); 7940ba1c634SDan Williams nd_label->slot = __cpu_to_le32(slot); 795*faec6f8aSDan Williams if (namespace_label_has(ndd, type_guid)) 796*faec6f8aSDan Williams guid_copy(&nd_label->type_guid, &nd_set->type_guid); 7970ba1c634SDan Williams 7980ba1c634SDan Williams /* update label */ 7990ba1c634SDan Williams offset = nd_label_offset(ndd, nd_label); 8000ba1c634SDan Williams rc = nvdimm_set_config_data(ndd, offset, nd_label, 801564e871aSDan Williams sizeof_namespace_label(ndd)); 8020ba1c634SDan Williams if (rc < 0) 8030ba1c634SDan Williams goto abort; 8040ba1c634SDan Williams } 8050ba1c634SDan Williams 8060ba1c634SDan Williams /* free up now unused slots in the new index */ 8070ba1c634SDan Williams for_each_set_bit(slot, victim_map, victim_map ? nslot : 0) { 8080ba1c634SDan Williams dev_dbg(ndd->dev, "%s: free: %d\n", __func__, slot); 8090ba1c634SDan Williams nd_label_free_slot(ndd, slot); 8100ba1c634SDan Williams } 8110ba1c634SDan Williams 8120ba1c634SDan Williams /* update index */ 8130ba1c634SDan Williams rc = nd_label_write_index(ndd, ndd->ns_next, 8140ba1c634SDan Williams nd_inc_seq(__le32_to_cpu(nsindex->seq)), 0); 8150ba1c634SDan Williams if (rc) 8160ba1c634SDan Williams goto abort; 8170ba1c634SDan Williams 8180ba1c634SDan Williams /* 8190ba1c634SDan Williams * Now that the on-dimm labels are up to date, fix up the tracking 8200ba1c634SDan Williams * entries in nd_mapping->labels 8210ba1c634SDan Williams */ 8220ba1c634SDan Williams nlabel = 0; 823ae8219f1SDan Williams mutex_lock(&nd_mapping->lock); 824ae8219f1SDan Williams list_for_each_entry_safe(label_ent, e, &nd_mapping->labels, list) { 825ae8219f1SDan Williams nd_label = label_ent->label; 826ae8219f1SDan Williams if (!nd_label) 827ae8219f1SDan Williams continue; 8280ba1c634SDan Williams nlabel++; 8290ba1c634SDan Williams memcpy(uuid, nd_label->uuid, NSLABEL_UUID_LEN); 8300ba1c634SDan Williams if (memcmp(uuid, nsblk->uuid, NSLABEL_UUID_LEN) != 0) 8310ba1c634SDan Williams continue; 8320ba1c634SDan Williams nlabel--; 833ae8219f1SDan Williams list_move(&label_ent->list, &list); 834ae8219f1SDan Williams label_ent->label = NULL; 8350ba1c634SDan Williams } 836ae8219f1SDan Williams list_splice_tail_init(&list, &nd_mapping->labels); 837ae8219f1SDan Williams mutex_unlock(&nd_mapping->lock); 838ae8219f1SDan Williams 8390ba1c634SDan Williams if (nlabel + nsblk->num_resources > num_labels) { 8400ba1c634SDan Williams /* 8410ba1c634SDan Williams * Bug, we can't end up with more resources than 8420ba1c634SDan Williams * available labels 8430ba1c634SDan Williams */ 8440ba1c634SDan Williams WARN_ON_ONCE(1); 8450ba1c634SDan Williams rc = -ENXIO; 8460ba1c634SDan Williams goto out; 8470ba1c634SDan Williams } 8480ba1c634SDan Williams 849ae8219f1SDan Williams mutex_lock(&nd_mapping->lock); 850ae8219f1SDan Williams label_ent = list_first_entry_or_null(&nd_mapping->labels, 851ae8219f1SDan Williams typeof(*label_ent), list); 852ae8219f1SDan Williams if (!label_ent) { 853ae8219f1SDan Williams WARN_ON(1); 854ae8219f1SDan Williams mutex_unlock(&nd_mapping->lock); 855ae8219f1SDan Williams rc = -ENXIO; 856ae8219f1SDan Williams goto out; 857ae8219f1SDan Williams } 8580ba1c634SDan Williams for_each_clear_bit_le(slot, free, nslot) { 859564e871aSDan Williams nd_label = to_label(ndd, slot); 8600ba1c634SDan Williams memcpy(uuid, nd_label->uuid, NSLABEL_UUID_LEN); 8610ba1c634SDan Williams if (memcmp(uuid, nsblk->uuid, NSLABEL_UUID_LEN) != 0) 8620ba1c634SDan Williams continue; 8630ba1c634SDan Williams res = to_resource(ndd, nd_label); 8640ba1c634SDan Williams res->flags &= ~DPA_RESOURCE_ADJUSTED; 865ae8219f1SDan Williams dev_vdbg(&nsblk->common.dev, "assign label slot: %d\n", slot); 866ae8219f1SDan Williams list_for_each_entry_from(label_ent, &nd_mapping->labels, list) { 867ae8219f1SDan Williams if (label_ent->label) 868ae8219f1SDan Williams continue; 869ae8219f1SDan Williams label_ent->label = nd_label; 870ae8219f1SDan Williams nd_label = NULL; 871ae8219f1SDan Williams break; 8720ba1c634SDan Williams } 873ae8219f1SDan Williams if (nd_label) 874ae8219f1SDan Williams dev_WARN(&nsblk->common.dev, 875ae8219f1SDan Williams "failed to track label slot%d\n", slot); 876ae8219f1SDan Williams } 877ae8219f1SDan Williams mutex_unlock(&nd_mapping->lock); 8780ba1c634SDan Williams 8790ba1c634SDan Williams out: 8800ba1c634SDan Williams kfree(old_res_list); 8810ba1c634SDan Williams kfree(victim_map); 8820ba1c634SDan Williams return rc; 8830ba1c634SDan Williams 8840ba1c634SDan Williams abort: 8850ba1c634SDan Williams /* 8860ba1c634SDan Williams * 1/ repair the allocated label bitmap in the index 8870ba1c634SDan Williams * 2/ restore the resource list 8880ba1c634SDan Williams */ 8890ba1c634SDan Williams nd_label_copy(ndd, nsindex, to_current_namespace_index(ndd)); 8900ba1c634SDan Williams kfree(nsblk->res); 8910ba1c634SDan Williams nsblk->res = old_res_list; 8920ba1c634SDan Williams nsblk->num_resources = old_num_resources; 8930ba1c634SDan Williams old_res_list = NULL; 8940ba1c634SDan Williams goto out; 8950ba1c634SDan Williams } 8960ba1c634SDan Williams 8970ba1c634SDan Williams static int init_labels(struct nd_mapping *nd_mapping, int num_labels) 8980ba1c634SDan Williams { 899ae8219f1SDan Williams int i, old_num_labels = 0; 900ae8219f1SDan Williams struct nd_label_ent *label_ent; 9010ba1c634SDan Williams struct nd_namespace_index *nsindex; 9020ba1c634SDan Williams struct nvdimm_drvdata *ndd = to_ndd(nd_mapping); 9030ba1c634SDan Williams 904ae8219f1SDan Williams mutex_lock(&nd_mapping->lock); 905ae8219f1SDan Williams list_for_each_entry(label_ent, &nd_mapping->labels, list) 9060ba1c634SDan Williams old_num_labels++; 907ae8219f1SDan Williams mutex_unlock(&nd_mapping->lock); 9080ba1c634SDan Williams 9090ba1c634SDan Williams /* 9100ba1c634SDan Williams * We need to preserve all the old labels for the mapping so 9110ba1c634SDan Williams * they can be garbage collected after writing the new labels. 9120ba1c634SDan Williams */ 913ae8219f1SDan Williams for (i = old_num_labels; i < num_labels; i++) { 914ae8219f1SDan Williams label_ent = kzalloc(sizeof(*label_ent), GFP_KERNEL); 915ae8219f1SDan Williams if (!label_ent) 9160ba1c634SDan Williams return -ENOMEM; 917ae8219f1SDan Williams mutex_lock(&nd_mapping->lock); 918ae8219f1SDan Williams list_add_tail(&label_ent->list, &nd_mapping->labels); 919ae8219f1SDan Williams mutex_unlock(&nd_mapping->lock); 9200ba1c634SDan Williams } 9210ba1c634SDan Williams 922f524bf27SDan Williams if (ndd->ns_current == -1 || ndd->ns_next == -1) 923f524bf27SDan Williams /* pass */; 924f524bf27SDan Williams else 9250ba1c634SDan Williams return max(num_labels, old_num_labels); 926f524bf27SDan Williams 927f524bf27SDan Williams nsindex = to_namespace_index(ndd, 0); 928f524bf27SDan Williams memset(nsindex, 0, ndd->nsarea.config_size); 929f524bf27SDan Williams for (i = 0; i < 2; i++) { 930f524bf27SDan Williams int rc = nd_label_write_index(ndd, i, i*2, ND_NSINDEX_INIT); 931f524bf27SDan Williams 932f524bf27SDan Williams if (rc) 933f524bf27SDan Williams return rc; 934f524bf27SDan Williams } 935f524bf27SDan Williams ndd->ns_next = 1; 936f524bf27SDan Williams ndd->ns_current = 0; 937f524bf27SDan Williams 9380ba1c634SDan Williams return max(num_labels, old_num_labels); 939f524bf27SDan Williams } 940f524bf27SDan Williams 941f524bf27SDan Williams static int del_labels(struct nd_mapping *nd_mapping, u8 *uuid) 942f524bf27SDan Williams { 943f524bf27SDan Williams struct nvdimm_drvdata *ndd = to_ndd(nd_mapping); 944ae8219f1SDan Williams struct nd_label_ent *label_ent, *e; 945f524bf27SDan Williams struct nd_namespace_index *nsindex; 946f524bf27SDan Williams u8 label_uuid[NSLABEL_UUID_LEN]; 947f524bf27SDan Williams unsigned long *free; 948ae8219f1SDan Williams LIST_HEAD(list); 949f524bf27SDan Williams u32 nslot, slot; 950ae8219f1SDan Williams int active = 0; 951f524bf27SDan Williams 952f524bf27SDan Williams if (!uuid) 953f524bf27SDan Williams return 0; 954f524bf27SDan Williams 955f524bf27SDan Williams /* no index || no labels == nothing to delete */ 956ae8219f1SDan Williams if (!preamble_next(ndd, &nsindex, &free, &nslot)) 957f524bf27SDan Williams return 0; 958f524bf27SDan Williams 959ae8219f1SDan Williams mutex_lock(&nd_mapping->lock); 960ae8219f1SDan Williams list_for_each_entry_safe(label_ent, e, &nd_mapping->labels, list) { 961ae8219f1SDan Williams struct nd_namespace_label *nd_label = label_ent->label; 962ae8219f1SDan Williams 963ae8219f1SDan Williams if (!nd_label) 964ae8219f1SDan Williams continue; 965ae8219f1SDan Williams active++; 966f524bf27SDan Williams memcpy(label_uuid, nd_label->uuid, NSLABEL_UUID_LEN); 967f524bf27SDan Williams if (memcmp(label_uuid, uuid, NSLABEL_UUID_LEN) != 0) 968f524bf27SDan Williams continue; 969ae8219f1SDan Williams active--; 970f524bf27SDan Williams slot = to_slot(ndd, nd_label); 971f524bf27SDan Williams nd_label_free_slot(ndd, slot); 972f524bf27SDan Williams dev_dbg(ndd->dev, "%s: free: %d\n", __func__, slot); 973ae8219f1SDan Williams list_move_tail(&label_ent->list, &list); 974ae8219f1SDan Williams label_ent->label = NULL; 975f524bf27SDan Williams } 976ae8219f1SDan Williams list_splice_tail_init(&list, &nd_mapping->labels); 977f524bf27SDan Williams 978ae8219f1SDan Williams if (active == 0) { 979ae8219f1SDan Williams nd_mapping_free_labels(nd_mapping); 980ae8219f1SDan Williams dev_dbg(ndd->dev, "%s: no more active labels\n", __func__); 981f524bf27SDan Williams } 982ae8219f1SDan Williams mutex_unlock(&nd_mapping->lock); 983f524bf27SDan Williams 984f524bf27SDan Williams return nd_label_write_index(ndd, ndd->ns_next, 985f524bf27SDan Williams nd_inc_seq(__le32_to_cpu(nsindex->seq)), 0); 986f524bf27SDan Williams } 987f524bf27SDan Williams 988f524bf27SDan Williams int nd_pmem_namespace_label_update(struct nd_region *nd_region, 989f524bf27SDan Williams struct nd_namespace_pmem *nspm, resource_size_t size) 990f524bf27SDan Williams { 991f524bf27SDan Williams int i; 992f524bf27SDan Williams 993f524bf27SDan Williams for (i = 0; i < nd_region->ndr_mappings; i++) { 994f524bf27SDan Williams struct nd_mapping *nd_mapping = &nd_region->mapping[i]; 99516660eaeSDan Williams struct nvdimm_drvdata *ndd = to_ndd(nd_mapping); 99616660eaeSDan Williams struct resource *res; 99716660eaeSDan Williams int rc, count = 0; 998f524bf27SDan Williams 999f524bf27SDan Williams if (size == 0) { 1000f524bf27SDan Williams rc = del_labels(nd_mapping, nspm->uuid); 1001f524bf27SDan Williams if (rc) 1002f524bf27SDan Williams return rc; 1003f524bf27SDan Williams continue; 1004f524bf27SDan Williams } 1005f524bf27SDan Williams 100616660eaeSDan Williams for_each_dpa_resource(ndd, res) 10072d9a0274SNicolas Iooss if (strncmp(res->name, "pmem", 4) == 0) 100816660eaeSDan Williams count++; 100916660eaeSDan Williams WARN_ON_ONCE(!count); 101016660eaeSDan Williams 101116660eaeSDan Williams rc = init_labels(nd_mapping, count); 10120ba1c634SDan Williams if (rc < 0) 1013f524bf27SDan Williams return rc; 1014f524bf27SDan Williams 1015f524bf27SDan Williams rc = __pmem_label_update(nd_region, nd_mapping, nspm, i); 1016f524bf27SDan Williams if (rc) 1017f524bf27SDan Williams return rc; 1018f524bf27SDan Williams } 1019f524bf27SDan Williams 1020f524bf27SDan Williams return 0; 1021f524bf27SDan Williams } 10220ba1c634SDan Williams 10230ba1c634SDan Williams int nd_blk_namespace_label_update(struct nd_region *nd_region, 10240ba1c634SDan Williams struct nd_namespace_blk *nsblk, resource_size_t size) 10250ba1c634SDan Williams { 10260ba1c634SDan Williams struct nd_mapping *nd_mapping = &nd_region->mapping[0]; 10270ba1c634SDan Williams struct resource *res; 10280ba1c634SDan Williams int count = 0; 10290ba1c634SDan Williams 10300ba1c634SDan Williams if (size == 0) 10310ba1c634SDan Williams return del_labels(nd_mapping, nsblk->uuid); 10320ba1c634SDan Williams 10330ba1c634SDan Williams for_each_dpa_resource(to_ndd(nd_mapping), res) 10340ba1c634SDan Williams count++; 10350ba1c634SDan Williams 10360ba1c634SDan Williams count = init_labels(nd_mapping, count); 10370ba1c634SDan Williams if (count < 0) 10380ba1c634SDan Williams return count; 10390ba1c634SDan Williams 10400ba1c634SDan Williams return __blk_label_update(nd_region, nd_mapping, nsblk, count); 10410ba1c634SDan Williams } 1042