drivers/scsi/cxlflash/vlun.c

   1 /*
   2  * CXL Flash Device Driver
   3  *
   4  * Written by: Manoj N. Kumar <manoj@linux.vnet.ibm.com>, IBM Corporation
   5  *             Matthew R. Ochs <mrochs@linux.vnet.ibm.com>, IBM Corporation
   6  *
   7  * Copyright (C) 2015 IBM Corporation
   8  *
   9  * This program is free software; you can redistribute it and/or
  10  * modify it under the terms of the GNU General Public License
  11  * as published by the Free Software Foundation; either version
  12  * 2 of the License, or (at your option) any later version.
  13  */
  14
  15 #include <linux/syscalls.h>
  16 #include <misc/cxl.h>
  17 #include <asm/unaligned.h>
  18 #include <asm/bitsperlong.h>
  19
  20 #include <scsi/scsi_cmnd.h>
  21 #include <scsi/scsi_host.h>
  22 #include <uapi/scsi/cxlflash_ioctl.h>
  23
  24 #include "sislite.h"
  25 #include "common.h"
  26 #include "vlun.h"
  27 #include "superpipe.h"
  28
  29 /**
  30  * marshal_virt_to_resize() - translate uvirtual to resize structure
  31  * @virt:       Source structure from which to translate/copy.
  32  * @resize:     Destination structure for the translate/copy.
  33  */
  34 static void marshal_virt_to_resize(struct dk_cxlflash_uvirtual *virt,
  35                                    struct dk_cxlflash_resize *resize)
  36 {
  37         resize->hdr = virt->hdr;
  38         resize->context_id = virt->context_id;
  39         resize->rsrc_handle = virt->rsrc_handle;
  40         resize->req_size = virt->lun_size;
  41         resize->last_lba = virt->last_lba;
  42 }
  43
  44 /**
  45  * marshal_clone_to_rele() - translate clone to release structure
  46  * @clone:      Source structure from which to translate/copy.
  47  * @rele:       Destination structure for the translate/copy.
  48  */
  49 static void marshal_clone_to_rele(struct dk_cxlflash_clone *clone,
  50                                   struct dk_cxlflash_release *release)
  51 {
  52         release->hdr = clone->hdr;
  53         release->context_id = clone->context_id_dst;
  54 }
  55
  56 /**
  57  * ba_init() - initializes a block allocator
  58  * @ba_lun:     Block allocator to initialize.
  59  *
  60  * Return: 0 on success, -errno on failure
  61  */
  62 static int ba_init(struct ba_lun *ba_lun)
  63 {
  64         struct ba_lun_info *bali = NULL;
  65         int lun_size_au = 0, i = 0;
  66         int last_word_underflow = 0;
  67         u64 *lam;
  68
  69         pr_debug("%s: Initializing LUN: lun_id=%016llx "
  70                  "ba_lun->lsize=%lx ba_lun->au_size=%lX\n",
  71                 __func__, ba_lun->lun_id, ba_lun->lsize, ba_lun->au_size);
  72
  73         /* Calculate bit map size */
  74         lun_size_au = ba_lun->lsize / ba_lun->au_size;
  75         if (lun_size_au == 0) {
  76                 pr_debug("%s: Requested LUN size of 0!\n", __func__);
  77                 return -EINVAL;
  78         }
  79
  80         /* Allocate lun information container */
  81         bali = kzalloc(sizeof(struct ba_lun_info), GFP_KERNEL);
  82         if (unlikely(!bali)) {
  83                 pr_err("%s: Failed to allocate lun_info lun_id=%016llx\n",
  84                        __func__, ba_lun->lun_id);
  85                 return -ENOMEM;
  86         }
  87
  88         bali->total_aus = lun_size_au;
  89         bali->lun_bmap_size = lun_size_au / BITS_PER_LONG;
  90
  91         if (lun_size_au % BITS_PER_LONG)
  92                 bali->lun_bmap_size++;
  93
  94         /* Allocate bitmap space */
  95         bali->lun_alloc_map = kzalloc((bali->lun_bmap_size * sizeof(u64)),
  96                                       GFP_KERNEL);
  97         if (unlikely(!bali->lun_alloc_map)) {
  98                 pr_err("%s: Failed to allocate lun allocation map: "
  99                        "lun_id=%016llx\n", __func__, ba_lun->lun_id);
 100                 kfree(bali);
 101                 return -ENOMEM;
 102         }
 103
 104         /* Initialize the bit map size and set all bits to '1' */
 105         bali->free_aun_cnt = lun_size_au;
 106
 107         for (i = 0; i < bali->lun_bmap_size; i++)
 108                 bali->lun_alloc_map[i] = 0xFFFFFFFFFFFFFFFFULL;
 109
 110         /* If the last word not fully utilized, mark extra bits as allocated */
 111         last_word_underflow = (bali->lun_bmap_size * BITS_PER_LONG);
 112         last_word_underflow -= bali->free_aun_cnt;
 113         if (last_word_underflow > 0) {
 114                 lam = &bali->lun_alloc_map[bali->lun_bmap_size - 1];
 115                 for (i = (HIBIT - last_word_underflow + 1);
 116                      i < BITS_PER_LONG;
 117                      i++)
 118                         clear_bit(i, (ulong *)lam);
 119         }
 120
 121         /* Initialize high elevator index, low/curr already at 0 from kzalloc */
 122         bali->free_high_idx = bali->lun_bmap_size;
 123
 124         /* Allocate clone map */
 125         bali->aun_clone_map = kzalloc((bali->total_aus * sizeof(u8)),
 126                                       GFP_KERNEL);
 127         if (unlikely(!bali->aun_clone_map)) {
 128                 pr_err("%s: Failed to allocate clone map: lun_id=%016llx\n",
 129                        __func__, ba_lun->lun_id);
 130                 kfree(bali->lun_alloc_map);
 131                 kfree(bali);
 132                 return -ENOMEM;
 133         }
 134
 135         /* Pass the allocated LUN info as a handle to the user */
 136         ba_lun->ba_lun_handle = bali;
 137
 138         pr_debug("%s: Successfully initialized the LUN: "
 139                  "lun_id=%016llx bitmap size=%x, free_aun_cnt=%llx\n",
 140                 __func__, ba_lun->lun_id, bali->lun_bmap_size,
 141                 bali->free_aun_cnt);
 142         return 0;
 143 }
 144
 145 /**
 146  * find_free_range() - locates a free bit within the block allocator
 147  * @low:        First word in block allocator to start search.
 148  * @high:       Last word in block allocator to search.
 149  * @bali:       LUN information structure owning the block allocator to search.
 150  * @bit_word:   Passes back the word in the block allocator owning the free bit.
 151  *
 152  * Return: The bit position within the passed back word, -1 on failure
 153  */
 154 static int find_free_range(u32 low,
 155                            u32 high,
 156                            struct ba_lun_info *bali, int *bit_word)
 157 {
 158         int i;
 159         u64 bit_pos = -1;
 160         ulong *lam, num_bits;
 161
 162         for (i = low; i < high; i++)
 163                 if (bali->lun_alloc_map[i] != 0) {
 164                         lam = (ulong *)&bali->lun_alloc_map[i];
 165                         num_bits = (sizeof(*lam) * BITS_PER_BYTE);
 166                         bit_pos = find_first_bit(lam, num_bits);
 167
 168                         pr_devel("%s: Found free bit %llu in LUN "
 169                                  "map entry %016llx at bitmap index = %d\n",
 170                                  __func__, bit_pos, bali->lun_alloc_map[i], i);
 171
 172                         *bit_word = i;
 173                         bali->free_aun_cnt--;
 174                         clear_bit(bit_pos, lam);
 175                         break;
 176                 }
 177
 178         return bit_pos;
 179 }
 180
 181 /**
 182  * ba_alloc() - allocates a block from the block allocator
 183  * @ba_lun:     Block allocator from which to allocate a block.
 184  *
 185  * Return: The allocated block, -1 on failure
 186  */
 187 static u64 ba_alloc(struct ba_lun *ba_lun)
 188 {
 189         u64 bit_pos = -1;
 190         int bit_word = 0;
 191         struct ba_lun_info *bali = NULL;
 192
 193         bali = ba_lun->ba_lun_handle;
 194
 195         pr_debug("%s: Received block allocation request: "
 196                  "lun_id=%016llx free_aun_cnt=%llx\n",
 197                  __func__, ba_lun->lun_id, bali->free_aun_cnt);
 198
 199         if (bali->free_aun_cnt == 0) {
 200                 pr_debug("%s: No space left on LUN: lun_id=%016llx\n",
 201                          __func__, ba_lun->lun_id);
 202                 return -1ULL;
 203         }
 204
 205         /* Search to find a free entry, curr->high then low->curr */
 206         bit_pos = find_free_range(bali->free_curr_idx,
 207                                   bali->free_high_idx, bali, &bit_word);
 208         if (bit_pos == -1) {
 209                 bit_pos = find_free_range(bali->free_low_idx,
 210                                           bali->free_curr_idx,
 211                                           bali, &bit_word);
 212                 if (bit_pos == -1) {
 213                         pr_debug("%s: Could not find an allocation unit on LUN:"
 214                                  " lun_id=%016llx\n", __func__, ba_lun->lun_id);
 215                         return -1ULL;
 216                 }
 217         }
 218
 219         /* Update the free_curr_idx */
 220         if (bit_pos == HIBIT)
 221                 bali->free_curr_idx = bit_word + 1;
 222         else
 223                 bali->free_curr_idx = bit_word;
 224
 225         pr_debug("%s: Allocating AU number=%llx lun_id=%016llx "
 226                  "free_aun_cnt=%llx\n", __func__,
 227                  ((bit_word * BITS_PER_LONG) + bit_pos), ba_lun->lun_id,
 228                  bali->free_aun_cnt);
 229
 230         return (u64) ((bit_word * BITS_PER_LONG) + bit_pos);
 231 }
 232
 233 /**
 234  * validate_alloc() - validates the specified block has been allocated
 235  * @ba_lun_info:        LUN info owning the block allocator.
 236  * @aun:                Block to validate.
 237  *
 238  * Return: 0 on success, -1 on failure
 239  */
 240 static int validate_alloc(struct ba_lun_info *bali, u64 aun)
 241 {
 242         int idx = 0, bit_pos = 0;
 243
 244         idx = aun / BITS_PER_LONG;
 245         bit_pos = aun % BITS_PER_LONG;
 246
 247         if (test_bit(bit_pos, (ulong *)&bali->lun_alloc_map[idx]))
 248                 return -1;
 249
 250         return 0;
 251 }
 252
 253 /**
 254  * ba_free() - frees a block from the block allocator
 255  * @ba_lun:     Block allocator from which to allocate a block.
 256  * @to_free:    Block to free.
 257  *
 258  * Return: 0 on success, -1 on failure
 259  */
 260 static int ba_free(struct ba_lun *ba_lun, u64 to_free)
 261 {
 262         int idx = 0, bit_pos = 0;
 263         struct ba_lun_info *bali = NULL;
 264
 265         bali = ba_lun->ba_lun_handle;
 266
 267         if (validate_alloc(bali, to_free)) {
 268                 pr_debug("%s: AUN %llx is not allocated on lun_id=%016llx\n",
 269                          __func__, to_free, ba_lun->lun_id);
 270                 return -1;
 271         }
 272
 273         pr_debug("%s: Received a request to free AU=%llx lun_id=%016llx "
 274                  "free_aun_cnt=%llx\n", __func__, to_free, ba_lun->lun_id,
 275                  bali->free_aun_cnt);
 276
 277         if (bali->aun_clone_map[to_free] > 0) {
 278                 pr_debug("%s: AUN %llx lun_id=%016llx cloned. Clone count=%x\n",
 279                          __func__, to_free, ba_lun->lun_id,
 280                          bali->aun_clone_map[to_free]);
 281                 bali->aun_clone_map[to_free]--;
 282                 return 0;
 283         }
 284
 285         idx = to_free / BITS_PER_LONG;
 286         bit_pos = to_free % BITS_PER_LONG;
 287
 288         set_bit(bit_pos, (ulong *)&bali->lun_alloc_map[idx]);
 289         bali->free_aun_cnt++;
 290
 291         if (idx < bali->free_low_idx)
 292                 bali->free_low_idx = idx;
 293         else if (idx > bali->free_high_idx)
 294                 bali->free_high_idx = idx;
 295
 296         pr_debug("%s: Successfully freed AU bit_pos=%x bit map index=%x "
 297                  "lun_id=%016llx free_aun_cnt=%llx\n", __func__, bit_pos, idx,
 298                  ba_lun->lun_id, bali->free_aun_cnt);
 299
 300         return 0;
 301 }
 302
 303 /**
 304  * ba_clone() - Clone a chunk of the block allocation table
 305  * @ba_lun:     Block allocator from which to allocate a block.
 306  * @to_free:    Block to free.
 307  *
 308  * Return: 0 on success, -1 on failure
 309  */
 310 static int ba_clone(struct ba_lun *ba_lun, u64 to_clone)
 311 {
 312         struct ba_lun_info *bali = ba_lun->ba_lun_handle;
 313
 314         if (validate_alloc(bali, to_clone)) {
 315                 pr_debug("%s: AUN=%llx not allocated on lun_id=%016llx\n",
 316                          __func__, to_clone, ba_lun->lun_id);
 317                 return -1;
 318         }
 319
 320         pr_debug("%s: Received a request to clone AUN %llx on lun_id=%016llx\n",
 321                  __func__, to_clone, ba_lun->lun_id);
 322
 323         if (bali->aun_clone_map[to_clone] == MAX_AUN_CLONE_CNT) {
 324                 pr_debug("%s: AUN %llx on lun_id=%016llx hit max clones already\n",
 325                          __func__, to_clone, ba_lun->lun_id);
 326                 return -1;
 327         }
 328
 329         bali->aun_clone_map[to_clone]++;
 330
 331         return 0;
 332 }
 333
 334 /**
 335  * ba_space() - returns the amount of free space left in the block allocator
 336  * @ba_lun:     Block allocator.
 337  *
 338  * Return: Amount of free space in block allocator
 339  */
 340 static u64 ba_space(struct ba_lun *ba_lun)
 341 {
 342         struct ba_lun_info *bali = ba_lun->ba_lun_handle;
 343
 344         return bali->free_aun_cnt;
 345 }
 346
 347 /**
 348  * cxlflash_ba_terminate() - frees resources associated with the block allocator
 349  * @ba_lun:     Block allocator.
 350  *
 351  * Safe to call in a partially allocated state.
 352  */
 353 void cxlflash_ba_terminate(struct ba_lun *ba_lun)
 354 {
 355         struct ba_lun_info *bali = ba_lun->ba_lun_handle;
 356
 357         if (bali) {
 358                 kfree(bali->aun_clone_map);
 359                 kfree(bali->lun_alloc_map);
 360                 kfree(bali);
 361                 ba_lun->ba_lun_handle = NULL;
 362         }
 363 }
 364
 365 /**
 366  * init_vlun() - initializes a LUN for virtual use
 367  * @lun_info:   LUN information structure that owns the block allocator.
 368  *
 369  * Return: 0 on success, -errno on failure
 370  */
 371 static int init_vlun(struct llun_info *lli)
 372 {
 373         int rc = 0;
 374         struct glun_info *gli = lli->parent;
 375         struct blka *blka = &gli->blka;
 376
 377         memset(blka, 0, sizeof(*blka));
 378         mutex_init(&blka->mutex);
 379
 380         /* LUN IDs are unique per port, save the index instead */
 381         blka->ba_lun.lun_id = lli->lun_index;
 382         blka->ba_lun.lsize = gli->max_lba + 1;
 383         blka->ba_lun.lba_size = gli->blk_len;
 384
 385         blka->ba_lun.au_size = MC_CHUNK_SIZE;
 386         blka->nchunk = blka->ba_lun.lsize / MC_CHUNK_SIZE;
 387
 388         rc = ba_init(&blka->ba_lun);
 389         if (unlikely(rc))
 390                 pr_debug("%s: cannot init block_alloc, rc=%d\n", __func__, rc);
 391
 392         pr_debug("%s: returning rc=%d lli=%p\n", __func__, rc, lli);
 393         return rc;
 394 }
 395
 396 /**
 397  * write_same16() - sends a SCSI WRITE_SAME16 (0) command to specified LUN
 398  * @sdev:       SCSI device associated with LUN.
 399  * @lba:        Logical block address to start write same.
 400  * @nblks:      Number of logical blocks to write same.
 401  *
 402  * The SCSI WRITE_SAME16 can take quite a while to complete. Should an EEH occur
 403  * while in scsi_execute(), the EEH handler will attempt to recover. As part of
 404  * the recovery, the handler drains all currently running ioctls, waiting until
 405  * they have completed before proceeding with a reset. As this routine is used
 406  * on the ioctl path, this can create a condition where the EEH handler becomes
 407  * stuck, infinitely waiting for this ioctl thread. To avoid this behavior,
 408  * temporarily unmark this thread as an ioctl thread by releasing the ioctl read
 409  * semaphore. This will allow the EEH handler to proceed with a recovery while
 410  * this thread is still running. Once the scsi_execute() returns, reacquire the
 411  * ioctl read semaphore and check the adapter state in case it changed while
 412  * inside of scsi_execute(). The state check will wait if the adapter is still
 413  * being recovered or return a failure if the recovery failed. In the event that
 414  * the adapter reset failed, simply return the failure as the ioctl would be
 415  * unable to continue.
 416  *
 417  * Note that the above puts a requirement on this routine to only be called on
 418  * an ioctl thread.
 419  *
 420  * Return: 0 on success, -errno on failure
 421  */
 422 static int write_same16(struct scsi_device *sdev,
 423                         u64 lba,
 424                         u32 nblks)
 425 {
 426         u8 *cmd_buf = NULL;
 427         u8 *scsi_cmd = NULL;
 428         u8 *sense_buf = NULL;
 429         int rc = 0;
 430         int result = 0;
 431         int ws_limit = SISLITE_MAX_WS_BLOCKS;
 432         u64 offset = lba;
 433         int left = nblks;
 434         u32 to = sdev->request_queue->rq_timeout;
 435         struct cxlflash_cfg *cfg = shost_priv(sdev->host);
 436         struct device *dev = &cfg->dev->dev;
 437
 438         cmd_buf = kzalloc(CMD_BUFSIZE, GFP_KERNEL);
 439         scsi_cmd = kzalloc(MAX_COMMAND_SIZE, GFP_KERNEL);
 440         sense_buf = kzalloc(SCSI_SENSE_BUFFERSIZE, GFP_KERNEL);
 441         if (unlikely(!cmd_buf || !scsi_cmd || !sense_buf)) {
 442                 rc = -ENOMEM;
 443                 goto out;
 444         }
 445
 446         while (left > 0) {
 447
 448                 scsi_cmd[0] = WRITE_SAME_16;
 449                 put_unaligned_be64(offset, &scsi_cmd[2]);
 450                 put_unaligned_be32(ws_limit < left ? ws_limit : left,
 451                                    &scsi_cmd[10]);
 452
 453                 /* Drop the ioctl read semahpore across lengthy call */
 454                 up_read(&cfg->ioctl_rwsem);
 455                 result = scsi_execute(sdev, scsi_cmd, DMA_TO_DEVICE, cmd_buf,
 456                                       CMD_BUFSIZE, sense_buf, to, CMD_RETRIES,
 457                                       0, NULL);
 458                 down_read(&cfg->ioctl_rwsem);
 459                 rc = check_state(cfg);
 460                 if (rc) {
 461                         dev_err(dev, "%s: Failed state result=%08x\n",
 462                                 __func__, result);
 463                         rc = -ENODEV;
 464                         goto out;
 465                 }
 466
 467                 if (result) {
 468                         dev_err_ratelimited(dev, "%s: command failed for "
 469                                             "offset=%lld result=%08x\n",
 470                                             __func__, offset, result);
 471                         rc = -EIO;
 472                         goto out;
 473                 }
 474                 left -= ws_limit;
 475                 offset += ws_limit;
 476         }
 477
 478 out:
 479         kfree(cmd_buf);
 480         kfree(scsi_cmd);
 481         kfree(sense_buf);
 482         dev_dbg(dev, "%s: returning rc=%d\n", __func__, rc);
 483         return rc;
 484 }
 485
 486 /**
 487  * grow_lxt() - expands the translation table associated with the specified RHTE
 488  * @afu:        AFU associated with the host.
 489  * @sdev:       SCSI device associated with LUN.
 490  * @ctxid:      Context ID of context owning the RHTE.
 491  * @rhndl:      Resource handle associated with the RHTE.
 492  * @rhte:       Resource handle entry (RHTE).
 493  * @new_size:   Number of translation entries associated with RHTE.
 494  *
 495  * By design, this routine employs a 'best attempt' allocation and will
 496  * truncate the requested size down if there is not sufficient space in
 497  * the block allocator to satisfy the request but there does exist some
 498  * amount of space. The user is made aware of this by returning the size
 499  * allocated.
 500  *
 501  * Return: 0 on success, -errno on failure
 502  */
 503 static int grow_lxt(struct afu *afu,
 504                     struct scsi_device *sdev,
 505                     ctx_hndl_t ctxid,
 506                     res_hndl_t rhndl,
 507                     struct sisl_rht_entry *rhte,
 508                     u64 *new_size)
 509 {
 510         struct cxlflash_cfg *cfg = shost_priv(sdev->host);
 511         struct device *dev = &cfg->dev->dev;
 512         struct sisl_lxt_entry *lxt = NULL, *lxt_old = NULL;
 513         struct llun_info *lli = sdev->hostdata;
 514         struct glun_info *gli = lli->parent;
 515         struct blka *blka = &gli->blka;
 516         u32 av_size;
 517         u32 ngrps, ngrps_old;
 518         u64 aun;                /* chunk# allocated by block allocator */
 519         u64 delta = *new_size - rhte->lxt_cnt;
 520         u64 my_new_size;
 521         int i, rc = 0;
 522
 523         /*
 524          * Check what is available in the block allocator before re-allocating
 525          * LXT array. This is done up front under the mutex which must not be
 526          * released until after allocation is complete.
 527          */
 528         mutex_lock(&blka->mutex);
 529         av_size = ba_space(&blka->ba_lun);
 530         if (unlikely(av_size <= 0)) {
 531                 dev_dbg(dev, "%s: ba_space error av_size=%d\n",
 532                         __func__, av_size);
 533                 mutex_unlock(&blka->mutex);
 534                 rc = -ENOSPC;
 535                 goto out;
 536         }
 537
 538         if (av_size < delta)
 539                 delta = av_size;
 540
 541         lxt_old = rhte->lxt_start;
 542         ngrps_old = LXT_NUM_GROUPS(rhte->lxt_cnt);
 543         ngrps = LXT_NUM_GROUPS(rhte->lxt_cnt + delta);
 544
 545         if (ngrps != ngrps_old) {
 546                 /* reallocate to fit new size */
 547                 lxt = kzalloc((sizeof(*lxt) * LXT_GROUP_SIZE * ngrps),
 548                               GFP_KERNEL);
 549                 if (unlikely(!lxt)) {
 550                         mutex_unlock(&blka->mutex);
 551                         rc = -ENOMEM;
 552                         goto out;
 553                 }
 554
 555                 /* copy over all old entries */
 556                 memcpy(lxt, lxt_old, (sizeof(*lxt) * rhte->lxt_cnt));
 557         } else
 558                 lxt = lxt_old;
 559
 560         /* nothing can fail from now on */
 561         my_new_size = rhte->lxt_cnt + delta;
 562
 563         /* add new entries to the end */
 564         for (i = rhte->lxt_cnt; i < my_new_size; i++) {
 565                 /*
 566                  * Due to the earlier check of available space, ba_alloc
 567                  * cannot fail here. If it did due to internal error,
 568                  * leave a rlba_base of -1u which will likely be a
 569                  * invalid LUN (too large).
 570                  */
 571                 aun = ba_alloc(&blka->ba_lun);
 572                 if ((aun == -1ULL) || (aun >= blka->nchunk))
 573                         dev_dbg(dev, "%s: ba_alloc error allocated chunk=%llu "
 574                                 "max=%llu\n", __func__, aun, blka->nchunk - 1);
 575
 576                 /* select both ports, use r/w perms from RHT */
 577                 lxt[i].rlba_base = ((aun << MC_CHUNK_SHIFT) |
 578                                     (lli->lun_index << LXT_LUNIDX_SHIFT) |
 579                                     (RHT_PERM_RW << LXT_PERM_SHIFT |
 580                                      lli->port_sel));
 581         }
 582
 583         mutex_unlock(&blka->mutex);
 584
 585         /*
 586          * The following sequence is prescribed in the SISlite spec
 587          * for syncing up with the AFU when adding LXT entries.
 588          */
 589         dma_wmb(); /* Make LXT updates are visible */
 590
 591         rhte->lxt_start = lxt;
 592         dma_wmb(); /* Make RHT entry's LXT table update visible */
 593
 594         rhte->lxt_cnt = my_new_size;
 595         dma_wmb(); /* Make RHT entry's LXT table size update visible */
 596
 597         cxlflash_afu_sync(afu, ctxid, rhndl, AFU_LW_SYNC);
 598
 599         /* free old lxt if reallocated */
 600         if (lxt != lxt_old)
 601                 kfree(lxt_old);
 602         *new_size = my_new_size;
 603 out:
 604         dev_dbg(dev, "%s: returning rc=%d\n", __func__, rc);
 605         return rc;
 606 }
 607
 608 /**
 609  * shrink_lxt() - reduces translation table associated with the specified RHTE
 610  * @afu:        AFU associated with the host.
 611  * @sdev:       SCSI device associated with LUN.
 612  * @rhndl:      Resource handle associated with the RHTE.
 613  * @rhte:       Resource handle entry (RHTE).
 614  * @ctxi:       Context owning resources.
 615  * @new_size:   Number of translation entries associated with RHTE.
 616  *
 617  * Return: 0 on success, -errno on failure
 618  */
 619 static int shrink_lxt(struct afu *afu,
 620                       struct scsi_device *sdev,
 621                       res_hndl_t rhndl,
 622                       struct sisl_rht_entry *rhte,
 623                       struct ctx_info *ctxi,
 624                       u64 *new_size)
 625 {
 626         struct cxlflash_cfg *cfg = shost_priv(sdev->host);
 627         struct device *dev = &cfg->dev->dev;
 628         struct sisl_lxt_entry *lxt, *lxt_old;
 629         struct llun_info *lli = sdev->hostdata;
 630         struct glun_info *gli = lli->parent;
 631         struct blka *blka = &gli->blka;
 632         ctx_hndl_t ctxid = DECODE_CTXID(ctxi->ctxid);
 633         bool needs_ws = ctxi->rht_needs_ws[rhndl];
 634         bool needs_sync = !ctxi->err_recovery_active;
 635         u32 ngrps, ngrps_old;
 636         u64 aun;                /* chunk# allocated by block allocator */
 637         u64 delta = rhte->lxt_cnt - *new_size;
 638         u64 my_new_size;
 639         int i, rc = 0;
 640
 641         lxt_old = rhte->lxt_start;
 642         ngrps_old = LXT_NUM_GROUPS(rhte->lxt_cnt);
 643         ngrps = LXT_NUM_GROUPS(rhte->lxt_cnt - delta);
 644
 645         if (ngrps != ngrps_old) {
 646                 /* Reallocate to fit new size unless new size is 0 */
 647                 if (ngrps) {
 648                         lxt = kzalloc((sizeof(*lxt) * LXT_GROUP_SIZE * ngrps),
 649                                       GFP_KERNEL);
 650                         if (unlikely(!lxt)) {
 651                                 rc = -ENOMEM;
 652                                 goto out;
 653                         }
 654
 655                         /* Copy over old entries that will remain */
 656                         memcpy(lxt, lxt_old,
 657                                (sizeof(*lxt) * (rhte->lxt_cnt - delta)));
 658                 } else
 659                         lxt = NULL;
 660         } else
 661                 lxt = lxt_old;
 662
 663         /* Nothing can fail from now on */
 664         my_new_size = rhte->lxt_cnt - delta;
 665
 666         /*
 667          * The following sequence is prescribed in the SISlite spec
 668          * for syncing up with the AFU when removing LXT entries.
 669          */
 670         rhte->lxt_cnt = my_new_size;
 671         dma_wmb(); /* Make RHT entry's LXT table size update visible */
 672
 673         rhte->lxt_start = lxt;
 674         dma_wmb(); /* Make RHT entry's LXT table update visible */
 675
 676         if (needs_sync)
 677                 cxlflash_afu_sync(afu, ctxid, rhndl, AFU_HW_SYNC);
 678
 679         if (needs_ws) {
 680                 /*
 681                  * Mark the context as unavailable, so that we can release
 682                  * the mutex safely.
 683                  */
 684                 ctxi->unavail = true;
 685                 mutex_unlock(&ctxi->mutex);
 686         }
 687
 688         /* Free LBAs allocated to freed chunks */
 689         mutex_lock(&blka->mutex);
 690         for (i = delta - 1; i >= 0; i--) {
 691                 /* Mask the higher 48 bits before shifting, even though
 692                  * it is a noop
 693                  */
 694                 aun = (lxt_old[my_new_size + i].rlba_base & SISL_ASTATUS_MASK);
 695                 aun = (aun >> MC_CHUNK_SHIFT);
 696                 if (needs_ws)
 697                         write_same16(sdev, aun, MC_CHUNK_SIZE);
 698                 ba_free(&blka->ba_lun, aun);
 699         }
 700         mutex_unlock(&blka->mutex);
 701
 702         if (needs_ws) {
 703                 /* Make the context visible again */
 704                 mutex_lock(&ctxi->mutex);
 705                 ctxi->unavail = false;
 706         }
 707
 708         /* Free old lxt if reallocated */
 709         if (lxt != lxt_old)
 710                 kfree(lxt_old);
 711         *new_size = my_new_size;
 712 out:
 713         dev_dbg(dev, "%s: returning rc=%d\n", __func__, rc);
 714         return rc;
 715 }
 716
 717 /**
 718  * _cxlflash_vlun_resize() - changes the size of a virtual LUN
 719  * @sdev:       SCSI device associated with LUN owning virtual LUN.
 720  * @ctxi:       Context owning resources.
 721  * @resize:     Resize ioctl data structure.
 722  *
 723  * On successful return, the user is informed of the new size (in blocks)
 724  * of the virtual LUN in last LBA format. When the size of the virtual
 725  * LUN is zero, the last LBA is reflected as -1. See comment in the
 726  * prologue for _cxlflash_disk_release() regarding AFU syncs and contexts
 727  * on the error recovery list.
 728  *
 729  * Return: 0 on success, -errno on failure
 730  */
 731 int _cxlflash_vlun_resize(struct scsi_device *sdev,
 732                           struct ctx_info *ctxi,
 733                           struct dk_cxlflash_resize *resize)
 734 {
 735         struct cxlflash_cfg *cfg = shost_priv(sdev->host);
 736         struct device *dev = &cfg->dev->dev;
 737         struct llun_info *lli = sdev->hostdata;
 738         struct glun_info *gli = lli->parent;
 739         struct afu *afu = cfg->afu;
 740         bool put_ctx = false;
 741
 742         res_hndl_t rhndl = resize->rsrc_handle;
 743         u64 new_size;
 744         u64 nsectors;
 745         u64 ctxid = DECODE_CTXID(resize->context_id),
 746             rctxid = resize->context_id;
 747
 748         struct sisl_rht_entry *rhte;
 749
 750         int rc = 0;
 751
 752         /*
 753          * The requested size (req_size) is always assumed to be in 4k blocks,
 754          * so we have to convert it here from 4k to chunk size.
 755          */
 756         nsectors = (resize->req_size * CXLFLASH_BLOCK_SIZE) / gli->blk_len;
 757         new_size = DIV_ROUND_UP(nsectors, MC_CHUNK_SIZE);
 758
 759         dev_dbg(dev, "%s: ctxid=%llu rhndl=%llu req_size=%llu new_size=%llu\n",
 760                 __func__, ctxid, resize->rsrc_handle, resize->req_size,
 761                 new_size);
 762
 763         if (unlikely(gli->mode != MODE_VIRTUAL)) {
 764                 dev_dbg(dev, "%s: LUN mode does not support resize mode=%d\n",
 765                         __func__, gli->mode);
 766                 rc = -EINVAL;
 767                 goto out;
 768
 769         }
 770
 771         if (!ctxi) {
 772                 ctxi = get_context(cfg, rctxid, lli, CTX_CTRL_ERR_FALLBACK);
 773                 if (unlikely(!ctxi)) {
 774                         dev_dbg(dev, "%s: Bad context ctxid=%llu\n",
 775                                 __func__, ctxid);
 776                         rc = -EINVAL;
 777                         goto out;
 778                 }
 779
 780                 put_ctx = true;
 781         }
 782
 783         rhte = get_rhte(ctxi, rhndl, lli);
 784         if (unlikely(!rhte)) {
 785                 dev_dbg(dev, "%s: Bad resource handle rhndl=%u\n",
 786                         __func__, rhndl);
 787                 rc = -EINVAL;
 788                 goto out;
 789         }
 790
 791         if (new_size > rhte->lxt_cnt)
 792                 rc = grow_lxt(afu, sdev, ctxid, rhndl, rhte, &new_size);
 793         else if (new_size < rhte->lxt_cnt)
 794                 rc = shrink_lxt(afu, sdev, rhndl, rhte, ctxi, &new_size);
 795
 796         resize->hdr.return_flags = 0;
 797         resize->last_lba = (new_size * MC_CHUNK_SIZE * gli->blk_len);
 798         resize->last_lba /= CXLFLASH_BLOCK_SIZE;
 799         resize->last_lba--;
 800
 801 out:
 802         if (put_ctx)
 803                 put_context(ctxi);
 804         dev_dbg(dev, "%s: resized to %llu returning rc=%d\n",
 805                 __func__, resize->last_lba, rc);
 806         return rc;
 807 }
 808
 809 int cxlflash_vlun_resize(struct scsi_device *sdev,
 810                          struct dk_cxlflash_resize *resize)
 811 {
 812         return _cxlflash_vlun_resize(sdev, NULL, resize);
 813 }
 814
 815 /**
 816  * cxlflash_restore_luntable() - Restore LUN table to prior state
 817  * @cfg:        Internal structure associated with the host.
 818  */
 819 void cxlflash_restore_luntable(struct cxlflash_cfg *cfg)
 820 {
 821         struct llun_info *lli, *temp;
 822         u32 chan;
 823         u32 lind;
 824         struct afu *afu = cfg->afu;
 825         struct device *dev = &cfg->dev->dev;
 826         struct sisl_global_map __iomem *agm = &afu->afu_map->global;
 827
 828         mutex_lock(&global.mutex);
 829
 830         list_for_each_entry_safe(lli, temp, &cfg->lluns, list) {
 831                 if (!lli->in_table)
 832                         continue;
 833
 834                 lind = lli->lun_index;
 835
 836                 if (lli->port_sel == BOTH_PORTS) {
 837                         writeq_be(lli->lun_id[0], &agm->fc_port[0][lind]);
 838                         writeq_be(lli->lun_id[1], &agm->fc_port[1][lind]);
 839                         dev_dbg(dev, "%s: Virtual LUN on slot %d  id0=%llx "
 840                                 "id1=%llx\n", __func__, lind,
 841                                 lli->lun_id[0], lli->lun_id[1]);
 842                 } else {
 843                         chan = PORT2CHAN(lli->port_sel);
 844                         writeq_be(lli->lun_id[chan], &agm->fc_port[chan][lind]);
 845                         dev_dbg(dev, "%s: Virtual LUN on slot %d chan=%d "
 846                                 "id=%llx\n", __func__, lind, chan,
 847                                 lli->lun_id[chan]);
 848                 }
 849         }
 850
 851         mutex_unlock(&global.mutex);
 852 }
 853
 854 /**
 855  * init_luntable() - write an entry in the LUN table
 856  * @cfg:        Internal structure associated with the host.
 857  * @lli:        Per adapter LUN information structure.
 858  *
 859  * On successful return, a LUN table entry is created.
 860  * At the top for LUNs visible on both ports.
 861  * At the bottom for LUNs visible only on one port.
 862  *
 863  * Return: 0 on success, -errno on failure
 864  */
 865 static int init_luntable(struct cxlflash_cfg *cfg, struct llun_info *lli)
 866 {
 867         u32 chan;
 868         u32 lind;
 869         int rc = 0;
 870         struct afu *afu = cfg->afu;
 871         struct device *dev = &cfg->dev->dev;
 872         struct sisl_global_map __iomem *agm = &afu->afu_map->global;
 873
 874         mutex_lock(&global.mutex);
 875
 876         if (lli->in_table)
 877                 goto out;
 878
 879         if (lli->port_sel == BOTH_PORTS) {
 880                 /*
 881                  * If this LUN is visible from both ports, we will put
 882                  * it in the top half of the LUN table.
 883                  */
 884                 if ((cfg->promote_lun_index == cfg->last_lun_index[0]) ||
 885                     (cfg->promote_lun_index == cfg->last_lun_index[1])) {
 886                         rc = -ENOSPC;
 887                         goto out;
 888                 }
 889
 890                 lind = lli->lun_index = cfg->promote_lun_index;
 891                 writeq_be(lli->lun_id[0], &agm->fc_port[0][lind]);
 892                 writeq_be(lli->lun_id[1], &agm->fc_port[1][lind]);
 893                 cfg->promote_lun_index++;
 894                 dev_dbg(dev, "%s: Virtual LUN on slot %d  id0=%llx id1=%llx\n",
 895                         __func__, lind, lli->lun_id[0], lli->lun_id[1]);
 896         } else {
 897                 /*
 898                  * If this LUN is visible only from one port, we will put
 899                  * it in the bottom half of the LUN table.
 900                  */
 901                 chan = PORT2CHAN(lli->port_sel);
 902                 if (cfg->promote_lun_index == cfg->last_lun_index[chan]) {
 903                         rc = -ENOSPC;
 904                         goto out;
 905                 }
 906
 907                 lind = lli->lun_index = cfg->last_lun_index[chan];
 908                 writeq_be(lli->lun_id[chan], &agm->fc_port[chan][lind]);
 909                 cfg->last_lun_index[chan]--;
 910                 dev_dbg(dev, "%s: Virtual LUN on slot %d  chan=%d id=%llx\n",
 911                         __func__, lind, chan, lli->lun_id[chan]);
 912         }
 913
 914         lli->in_table = true;
 915 out:
 916         mutex_unlock(&global.mutex);
 917         dev_dbg(dev, "%s: returning rc=%d\n", __func__, rc);
 918         return rc;
 919 }
 920
 921 /**
 922  * cxlflash_disk_virtual_open() - open a virtual disk of specified size
 923  * @sdev:       SCSI device associated with LUN owning virtual LUN.
 924  * @arg:        UVirtual ioctl data structure.
 925  *
 926  * On successful return, the user is informed of the resource handle
 927  * to be used to identify the virtual LUN and the size (in blocks) of
 928  * the virtual LUN in last LBA format. When the size of the virtual LUN
 929  * is zero, the last LBA is reflected as -1.
 930  *
 931  * Return: 0 on success, -errno on failure
 932  */
 933 int cxlflash_disk_virtual_open(struct scsi_device *sdev, void *arg)
 934 {
 935         struct cxlflash_cfg *cfg = shost_priv(sdev->host);
 936         struct device *dev = &cfg->dev->dev;
 937         struct llun_info *lli = sdev->hostdata;
 938         struct glun_info *gli = lli->parent;
 939
 940         struct dk_cxlflash_uvirtual *virt = (struct dk_cxlflash_uvirtual *)arg;
 941         struct dk_cxlflash_resize resize;
 942
 943         u64 ctxid = DECODE_CTXID(virt->context_id),
 944             rctxid = virt->context_id;
 945         u64 lun_size = virt->lun_size;
 946         u64 last_lba = 0;
 947         u64 rsrc_handle = -1;
 948
 949         int rc = 0;
 950
 951         struct ctx_info *ctxi = NULL;
 952         struct sisl_rht_entry *rhte = NULL;
 953
 954         dev_dbg(dev, "%s: ctxid=%llu ls=%llu\n", __func__, ctxid, lun_size);
 955
 956         /* Setup the LUNs block allocator on first call */
 957         mutex_lock(&gli->mutex);
 958         if (gli->mode == MODE_NONE) {
 959                 rc = init_vlun(lli);
 960                 if (rc) {
 961                         dev_err(dev, "%s: init_vlun failed rc=%d\n",
 962                                 __func__, rc);
 963                         rc = -ENOMEM;
 964                         goto err0;
 965                 }
 966         }
 967
 968         rc = cxlflash_lun_attach(gli, MODE_VIRTUAL, true);
 969         if (unlikely(rc)) {
 970                 dev_err(dev, "%s: Failed attach to LUN (VIRTUAL)\n", __func__);
 971                 goto err0;
 972         }
 973         mutex_unlock(&gli->mutex);
 974
 975         rc = init_luntable(cfg, lli);
 976         if (rc) {
 977                 dev_err(dev, "%s: init_luntable failed rc=%d\n", __func__, rc);
 978                 goto err1;
 979         }
 980
 981         ctxi = get_context(cfg, rctxid, lli, 0);
 982         if (unlikely(!ctxi)) {
 983                 dev_err(dev, "%s: Bad context ctxid=%llu\n", __func__, ctxid);
 984                 rc = -EINVAL;
 985                 goto err1;
 986         }
 987
 988         rhte = rhte_checkout(ctxi, lli);
 989         if (unlikely(!rhte)) {
 990                 dev_err(dev, "%s: too many opens ctxid=%llu\n",
 991                         __func__, ctxid);
 992                 rc = -EMFILE;   /* too many opens  */
 993                 goto err1;
 994         }
 995
 996         rsrc_handle = (rhte - ctxi->rht_start);
 997
 998         /* Populate RHT format 0 */
 999         rhte->nmask = MC_RHT_NMASK;
1000         rhte->fp = SISL_RHT_FP(0U, ctxi->rht_perms);
1001
1002         /* Resize even if requested size is 0 */
1003         marshal_virt_to_resize(virt, &resize);
1004         resize.rsrc_handle = rsrc_handle;
1005         rc = _cxlflash_vlun_resize(sdev, ctxi, &resize);
1006         if (rc) {
1007                 dev_err(dev, "%s: resize failed rc=%d\n", __func__, rc);
1008                 goto err2;
1009         }
1010         last_lba = resize.last_lba;
1011
1012         if (virt->hdr.flags & DK_CXLFLASH_UVIRTUAL_NEED_WRITE_SAME)
1013                 ctxi->rht_needs_ws[rsrc_handle] = true;
1014
1015         virt->hdr.return_flags = 0;
1016         virt->last_lba = last_lba;
1017         virt->rsrc_handle = rsrc_handle;
1018
1019         if (lli->port_sel == BOTH_PORTS)
1020                 virt->hdr.return_flags |= DK_CXLFLASH_ALL_PORTS_ACTIVE;
1021 out:
1022         if (likely(ctxi))
1023                 put_context(ctxi);
1024         dev_dbg(dev, "%s: returning handle=%llu rc=%d llba=%llu\n",
1025                 __func__, rsrc_handle, rc, last_lba);
1026         return rc;
1027
1028 err2:
1029         rhte_checkin(ctxi, rhte);
1030 err1:
1031         cxlflash_lun_detach(gli);
1032         goto out;
1033 err0:
1034         /* Special common cleanup prior to successful LUN attach */
1035         cxlflash_ba_terminate(&gli->blka.ba_lun);
1036         mutex_unlock(&gli->mutex);
1037         goto out;
1038 }
1039
1040 /**
1041  * clone_lxt() - copies translation tables from source to destination RHTE
1042  * @afu:        AFU associated with the host.
1043  * @blka:       Block allocator associated with LUN.
1044  * @ctxid:      Context ID of context owning the RHTE.
1045  * @rhndl:      Resource handle associated with the RHTE.
1046  * @rhte:       Destination resource handle entry (RHTE).
1047  * @rhte_src:   Source resource handle entry (RHTE).
1048  *
1049  * Return: 0 on success, -errno on failure
1050  */
1051 static int clone_lxt(struct afu *afu,
1052                      struct blka *blka,
1053                      ctx_hndl_t ctxid,
1054                      res_hndl_t rhndl,
1055                      struct sisl_rht_entry *rhte,
1056                      struct sisl_rht_entry *rhte_src)
1057 {
1058         struct cxlflash_cfg *cfg = afu->parent;
1059         struct device *dev = &cfg->dev->dev;
1060         struct sisl_lxt_entry *lxt;
1061         u32 ngrps;
1062         u64 aun;                /* chunk# allocated by block allocator */
1063         int i, j;
1064
1065         ngrps = LXT_NUM_GROUPS(rhte_src->lxt_cnt);
1066
1067         if (ngrps) {
1068                 /* allocate new LXTs for clone */
1069                 lxt = kzalloc((sizeof(*lxt) * LXT_GROUP_SIZE * ngrps),
1070                                 GFP_KERNEL);
1071                 if (unlikely(!lxt))
1072                         return -ENOMEM;
1073
1074                 /* copy over */
1075                 memcpy(lxt, rhte_src->lxt_start,
1076                        (sizeof(*lxt) * rhte_src->lxt_cnt));
1077
1078                 /* clone the LBAs in block allocator via ref_cnt */
1079                 mutex_lock(&blka->mutex);
1080                 for (i = 0; i < rhte_src->lxt_cnt; i++) {
1081                         aun = (lxt[i].rlba_base >> MC_CHUNK_SHIFT);
1082                         if (ba_clone(&blka->ba_lun, aun) == -1ULL) {
1083                                 /* free the clones already made */
1084                                 for (j = 0; j < i; j++) {
1085                                         aun = (lxt[j].rlba_base >>
1086                                                MC_CHUNK_SHIFT);
1087                                         ba_free(&blka->ba_lun, aun);
1088                                 }
1089
1090                                 mutex_unlock(&blka->mutex);
1091                                 kfree(lxt);
1092                                 return -EIO;
1093                         }
1094                 }
1095                 mutex_unlock(&blka->mutex);
1096         } else {
1097                 lxt = NULL;
1098         }
1099
1100         /*
1101          * The following sequence is prescribed in the SISlite spec
1102          * for syncing up with the AFU when adding LXT entries.
1103          */
1104         dma_wmb(); /* Make LXT updates are visible */
1105
1106         rhte->lxt_start = lxt;
1107         dma_wmb(); /* Make RHT entry's LXT table update visible */
1108
1109         rhte->lxt_cnt = rhte_src->lxt_cnt;
1110         dma_wmb(); /* Make RHT entry's LXT table size update visible */
1111
1112         cxlflash_afu_sync(afu, ctxid, rhndl, AFU_LW_SYNC);
1113
1114         dev_dbg(dev, "%s: returning\n", __func__);
1115         return 0;
1116 }
1117
1118 /**
1119  * cxlflash_disk_clone() - clone a context by making snapshot of another
1120  * @sdev:       SCSI device associated with LUN owning virtual LUN.
1121  * @clone:      Clone ioctl data structure.
1122  *
1123  * This routine effectively performs cxlflash_disk_open operation for each
1124  * in-use virtual resource in the source context. Note that the destination
1125  * context must be in pristine state and cannot have any resource handles
1126  * open at the time of the clone.
1127  *
1128  * Return: 0 on success, -errno on failure
1129  */
1130 int cxlflash_disk_clone(struct scsi_device *sdev,
1131                         struct dk_cxlflash_clone *clone)
1132 {
1133         struct cxlflash_cfg *cfg = shost_priv(sdev->host);
1134         struct device *dev = &cfg->dev->dev;
1135         struct llun_info *lli = sdev->hostdata;
1136         struct glun_info *gli = lli->parent;
1137         struct blka *blka = &gli->blka;
1138         struct afu *afu = cfg->afu;
1139         struct dk_cxlflash_release release = { { 0 }, 0 };
1140
1141         struct ctx_info *ctxi_src = NULL,
1142                         *ctxi_dst = NULL;
1143         struct lun_access *lun_access_src, *lun_access_dst;
1144         u32 perms;
1145         u64 ctxid_src = DECODE_CTXID(clone->context_id_src),
1146             ctxid_dst = DECODE_CTXID(clone->context_id_dst),
1147             rctxid_src = clone->context_id_src,
1148             rctxid_dst = clone->context_id_dst;
1149         int i, j;
1150         int rc = 0;
1151         bool found;
1152         LIST_HEAD(sidecar);
1153
1154         dev_dbg(dev, "%s: ctxid_src=%llu ctxid_dst=%llu\n",
1155                 __func__, ctxid_src, ctxid_dst);
1156
1157         /* Do not clone yourself */
1158         if (unlikely(rctxid_src == rctxid_dst)) {
1159                 rc = -EINVAL;
1160                 goto out;
1161         }
1162
1163         if (unlikely(gli->mode != MODE_VIRTUAL)) {
1164                 rc = -EINVAL;
1165                 dev_dbg(dev, "%s: Only supported on virtual LUNs mode=%u\n",
1166                         __func__, gli->mode);
1167                 goto out;
1168         }
1169
1170         ctxi_src = get_context(cfg, rctxid_src, lli, CTX_CTRL_CLONE);
1171         ctxi_dst = get_context(cfg, rctxid_dst, lli, 0);
1172         if (unlikely(!ctxi_src || !ctxi_dst)) {
1173                 dev_dbg(dev, "%s: Bad context ctxid_src=%llu ctxid_dst=%llu\n",
1174                         __func__, ctxid_src, ctxid_dst);
1175                 rc = -EINVAL;
1176                 goto out;
1177         }
1178
1179         /* Verify there is no open resource handle in the destination context */
1180         for (i = 0; i < MAX_RHT_PER_CONTEXT; i++)
1181                 if (ctxi_dst->rht_start[i].nmask != 0) {
1182                         rc = -EINVAL;
1183                         goto out;
1184                 }
1185
1186         /* Clone LUN access list */
1187         list_for_each_entry(lun_access_src, &ctxi_src->luns, list) {
1188                 found = false;
1189                 list_for_each_entry(lun_access_dst, &ctxi_dst->luns, list)
1190                         if (lun_access_dst->sdev == lun_access_src->sdev) {
1191                                 found = true;
1192                                 break;
1193                         }
1194
1195                 if (!found) {
1196                         lun_access_dst = kzalloc(sizeof(*lun_access_dst),
1197                                                  GFP_KERNEL);
1198                         if (unlikely(!lun_access_dst)) {
1199                                 dev_err(dev, "%s: lun_access allocation fail\n",
1200                                         __func__);
1201                                 rc = -ENOMEM;
1202                                 goto out;
1203                         }
1204
1205                         *lun_access_dst = *lun_access_src;
1206                         list_add(&lun_access_dst->list, &sidecar);
1207                 }
1208         }
1209
1210         if (unlikely(!ctxi_src->rht_out)) {
1211                 dev_dbg(dev, "%s: Nothing to clone\n", __func__);
1212                 goto out_success;
1213         }
1214
1215         /* User specified permission on attach */
1216         perms = ctxi_dst->rht_perms;
1217
1218         /*
1219          * Copy over checked-out RHT (and their associated LXT) entries by
1220          * hand, stopping after we've copied all outstanding entries and
1221          * cleaning up if the clone fails.
1222          *
1223          * Note: This loop is equivalent to performing cxlflash_disk_open and
1224          * cxlflash_vlun_resize. As such, LUN accounting needs to be taken into
1225          * account by attaching after each successful RHT entry clone. In the
1226          * event that a clone failure is experienced, the LUN detach is handled
1227          * via the cleanup performed by _cxlflash_disk_release.
1228          */
1229         for (i = 0; i < MAX_RHT_PER_CONTEXT; i++) {
1230                 if (ctxi_src->rht_out == ctxi_dst->rht_out)
1231                         break;
1232                 if (ctxi_src->rht_start[i].nmask == 0)
1233                         continue;
1234
1235                 /* Consume a destination RHT entry */
1236                 ctxi_dst->rht_out++;
1237                 ctxi_dst->rht_start[i].nmask = ctxi_src->rht_start[i].nmask;
1238                 ctxi_dst->rht_start[i].fp =
1239                     SISL_RHT_FP_CLONE(ctxi_src->rht_start[i].fp, perms);
1240                 ctxi_dst->rht_lun[i] = ctxi_src->rht_lun[i];
1241
1242                 rc = clone_lxt(afu, blka, ctxid_dst, i,
1243                                &ctxi_dst->rht_start[i],
1244                                &ctxi_src->rht_start[i]);
1245                 if (rc) {
1246                         marshal_clone_to_rele(clone, &release);
1247                         for (j = 0; j < i; j++) {
1248                                 release.rsrc_handle = j;
1249                                 _cxlflash_disk_release(sdev, ctxi_dst,
1250                                                        &release);
1251                         }
1252
1253                         /* Put back the one we failed on */
1254                         rhte_checkin(ctxi_dst, &ctxi_dst->rht_start[i]);
1255                         goto err;
1256                 }
1257
1258                 cxlflash_lun_attach(gli, gli->mode, false);
1259         }
1260
1261 out_success:
1262         list_splice(&sidecar, &ctxi_dst->luns);
1263
1264         /* fall through */
1265 out:
1266         if (ctxi_src)
1267                 put_context(ctxi_src);
1268         if (ctxi_dst)
1269                 put_context(ctxi_dst);
1270         dev_dbg(dev, "%s: returning rc=%d\n", __func__, rc);
1271         return rc;
1272
1273 err:
1274         list_for_each_entry_safe(lun_access_src, lun_access_dst, &sidecar, list)
1275                 kfree(lun_access_src);
1276         goto out;
1277 }