This source file includes following definitions.
- is_hole
- alloc_parallel
- get_parallel
- destroy_parallel
- put_parallel
- bl_submit_bio
- bl_alloc_init_bio
- offset_in_map
- do_add_page_to_bio
- bl_mark_devices_unavailable
- bl_end_io_read
- bl_read_cleanup
- bl_end_par_io_read
- bl_read_pagelist
- bl_end_io_write
- bl_write_cleanup
- bl_end_par_io_write
- bl_write_pagelist
- bl_free_layout_hdr
- __bl_alloc_layout_hdr
- bl_alloc_layout_hdr
- sl_alloc_layout_hdr
- bl_free_lseg
- verify_extent
- decode_sector_number
- bl_find_get_deviceid
- bl_alloc_extent
- bl_alloc_lseg
- bl_return_range
- bl_prepare_layoutcommit
- bl_cleanup_layoutcommit
- bl_set_layoutdriver
- is_aligned_req
- bl_pg_init_read
- bl_pg_test_read
- pnfs_num_cont_bytes
- bl_pg_init_write
- bl_pg_test_write
- nfs4blocklayout_init
- nfs4blocklayout_exit
   1 
   2 
   3 
   4 
   5 
   6 
   7 
   8 
   9 
  10 
  11 
  12 
  13 
  14 
  15 
  16 
  17 
  18 
  19 
  20 
  21 
  22 
  23 
  24 
  25 
  26 
  27 
  28 
  29 
  30 
  31 
  32 
  33 #include <linux/module.h>
  34 #include <linux/init.h>
  35 #include <linux/mount.h>
  36 #include <linux/namei.h>
  37 #include <linux/bio.h>          
  38 #include <linux/prefetch.h>
  39 #include <linux/pagevec.h>
  40 
  41 #include "../pnfs.h"
  42 #include "../nfs4session.h"
  43 #include "../internal.h"
  44 #include "blocklayout.h"
  45 
  46 #define NFSDBG_FACILITY NFSDBG_PNFS_LD
  47 
  48 MODULE_LICENSE("GPL");
  49 MODULE_AUTHOR("Andy Adamson <andros@citi.umich.edu>");
  50 MODULE_DESCRIPTION("The NFSv4.1 pNFS Block layout driver");
  51 
  52 static bool is_hole(struct pnfs_block_extent *be)
  53 {
  54         switch (be->be_state) {
  55         case PNFS_BLOCK_NONE_DATA:
  56                 return true;
  57         case PNFS_BLOCK_INVALID_DATA:
  58                 return be->be_tag ? false : true;
  59         default:
  60                 return false;
  61         }
  62 }
  63 
  64 
  65 
  66 
  67 struct parallel_io {
  68         struct kref refcnt;
  69         void (*pnfs_callback) (void *data);
  70         void *data;
  71 };
  72 
  73 static inline struct parallel_io *alloc_parallel(void *data)
  74 {
  75         struct parallel_io *rv;
  76 
  77         rv  = kmalloc(sizeof(*rv), GFP_NOFS);
  78         if (rv) {
  79                 rv->data = data;
  80                 kref_init(&rv->refcnt);
  81         }
  82         return rv;
  83 }
  84 
  85 static inline void get_parallel(struct parallel_io *p)
  86 {
  87         kref_get(&p->refcnt);
  88 }
  89 
  90 static void destroy_parallel(struct kref *kref)
  91 {
  92         struct parallel_io *p = container_of(kref, struct parallel_io, refcnt);
  93 
  94         dprintk("%s enter\n", __func__);
  95         p->pnfs_callback(p->data);
  96         kfree(p);
  97 }
  98 
  99 static inline void put_parallel(struct parallel_io *p)
 100 {
 101         kref_put(&p->refcnt, destroy_parallel);
 102 }
 103 
 104 static struct bio *
 105 bl_submit_bio(struct bio *bio)
 106 {
 107         if (bio) {
 108                 get_parallel(bio->bi_private);
 109                 dprintk("%s submitting %s bio %u@%llu\n", __func__,
 110                         bio_op(bio) == READ ? "read" : "write",
 111                         bio->bi_iter.bi_size,
 112                         (unsigned long long)bio->bi_iter.bi_sector);
 113                 submit_bio(bio);
 114         }
 115         return NULL;
 116 }
 117 
 118 static struct bio *
 119 bl_alloc_init_bio(int npg, struct block_device *bdev, sector_t disk_sector,
 120                 bio_end_io_t end_io, struct parallel_io *par)
 121 {
 122         struct bio *bio;
 123 
 124         npg = min(npg, BIO_MAX_PAGES);
 125         bio = bio_alloc(GFP_NOIO, npg);
 126         if (!bio && (current->flags & PF_MEMALLOC)) {
 127                 while (!bio && (npg /= 2))
 128                         bio = bio_alloc(GFP_NOIO, npg);
 129         }
 130 
 131         if (bio) {
 132                 bio->bi_iter.bi_sector = disk_sector;
 133                 bio_set_dev(bio, bdev);
 134                 bio->bi_end_io = end_io;
 135                 bio->bi_private = par;
 136         }
 137         return bio;
 138 }
 139 
 140 static bool offset_in_map(u64 offset, struct pnfs_block_dev_map *map)
 141 {
 142         return offset >= map->start && offset < map->start + map->len;
 143 }
 144 
 145 static struct bio *
 146 do_add_page_to_bio(struct bio *bio, int npg, int rw, sector_t isect,
 147                 struct page *page, struct pnfs_block_dev_map *map,
 148                 struct pnfs_block_extent *be, bio_end_io_t end_io,
 149                 struct parallel_io *par, unsigned int offset, int *len)
 150 {
 151         struct pnfs_block_dev *dev =
 152                 container_of(be->be_device, struct pnfs_block_dev, node);
 153         u64 disk_addr, end;
 154 
 155         dprintk("%s: npg %d rw %d isect %llu offset %u len %d\n", __func__,
 156                 npg, rw, (unsigned long long)isect, offset, *len);
 157 
 158         
 159         isect += be->be_v_offset;
 160         isect -= be->be_f_offset;
 161 
 162         
 163         disk_addr = (u64)isect << SECTOR_SHIFT;
 164         if (!offset_in_map(disk_addr, map)) {
 165                 if (!dev->map(dev, disk_addr, map) || !offset_in_map(disk_addr, map))
 166                         return ERR_PTR(-EIO);
 167                 bio = bl_submit_bio(bio);
 168         }
 169         disk_addr += map->disk_offset;
 170         disk_addr -= map->start;
 171 
 172         
 173         end = disk_addr + *len;
 174         if (end >= map->start + map->len)
 175                 *len = map->start + map->len - disk_addr;
 176 
 177 retry:
 178         if (!bio) {
 179                 bio = bl_alloc_init_bio(npg, map->bdev,
 180                                 disk_addr >> SECTOR_SHIFT, end_io, par);
 181                 if (!bio)
 182                         return ERR_PTR(-ENOMEM);
 183                 bio_set_op_attrs(bio, rw, 0);
 184         }
 185         if (bio_add_page(bio, page, *len, offset) < *len) {
 186                 bio = bl_submit_bio(bio);
 187                 goto retry;
 188         }
 189         return bio;
 190 }
 191 
 192 static void bl_mark_devices_unavailable(struct nfs_pgio_header *header, bool rw)
 193 {
 194         struct pnfs_block_layout *bl = BLK_LSEG2EXT(header->lseg);
 195         size_t bytes_left = header->args.count;
 196         sector_t isect, extent_length = 0;
 197         struct pnfs_block_extent be;
 198 
 199         isect = header->args.offset >> SECTOR_SHIFT;
 200         bytes_left += header->args.offset - (isect << SECTOR_SHIFT);
 201 
 202         while (bytes_left > 0) {
 203                 if (!ext_tree_lookup(bl, isect, &be, rw))
 204                                 return;
 205                 extent_length = be.be_length - (isect - be.be_f_offset);
 206                 nfs4_mark_deviceid_unavailable(be.be_device);
 207                 isect += extent_length;
 208                 if (bytes_left > extent_length << SECTOR_SHIFT)
 209                         bytes_left -= extent_length << SECTOR_SHIFT;
 210                 else
 211                         bytes_left = 0;
 212         }
 213 }
 214 
 215 static void bl_end_io_read(struct bio *bio)
 216 {
 217         struct parallel_io *par = bio->bi_private;
 218 
 219         if (bio->bi_status) {
 220                 struct nfs_pgio_header *header = par->data;
 221 
 222                 if (!header->pnfs_error)
 223                         header->pnfs_error = -EIO;
 224                 pnfs_set_lo_fail(header->lseg);
 225                 bl_mark_devices_unavailable(header, false);
 226         }
 227 
 228         bio_put(bio);
 229         put_parallel(par);
 230 }
 231 
 232 static void bl_read_cleanup(struct work_struct *work)
 233 {
 234         struct rpc_task *task;
 235         struct nfs_pgio_header *hdr;
 236         dprintk("%s enter\n", __func__);
 237         task = container_of(work, struct rpc_task, u.tk_work);
 238         hdr = container_of(task, struct nfs_pgio_header, task);
 239         pnfs_ld_read_done(hdr);
 240 }
 241 
 242 static void
 243 bl_end_par_io_read(void *data)
 244 {
 245         struct nfs_pgio_header *hdr = data;
 246 
 247         hdr->task.tk_status = hdr->pnfs_error;
 248         INIT_WORK(&hdr->task.u.tk_work, bl_read_cleanup);
 249         schedule_work(&hdr->task.u.tk_work);
 250 }
 251 
 252 static enum pnfs_try_status
 253 bl_read_pagelist(struct nfs_pgio_header *header)
 254 {
 255         struct pnfs_block_layout *bl = BLK_LSEG2EXT(header->lseg);
 256         struct pnfs_block_dev_map map = { .start = NFS4_MAX_UINT64 };
 257         struct bio *bio = NULL;
 258         struct pnfs_block_extent be;
 259         sector_t isect, extent_length = 0;
 260         struct parallel_io *par;
 261         loff_t f_offset = header->args.offset;
 262         size_t bytes_left = header->args.count;
 263         unsigned int pg_offset = header->args.pgbase, pg_len;
 264         struct page **pages = header->args.pages;
 265         int pg_index = header->args.pgbase >> PAGE_SHIFT;
 266         const bool is_dio = (header->dreq != NULL);
 267         struct blk_plug plug;
 268         int i;
 269 
 270         dprintk("%s enter nr_pages %u offset %lld count %u\n", __func__,
 271                 header->page_array.npages, f_offset,
 272                 (unsigned int)header->args.count);
 273 
 274         par = alloc_parallel(header);
 275         if (!par)
 276                 return PNFS_NOT_ATTEMPTED;
 277         par->pnfs_callback = bl_end_par_io_read;
 278 
 279         blk_start_plug(&plug);
 280 
 281         isect = (sector_t) (f_offset >> SECTOR_SHIFT);
 282         
 283         for (i = pg_index; i < header->page_array.npages; i++) {
 284                 if (extent_length <= 0) {
 285                         
 286                         bio = bl_submit_bio(bio);
 287 
 288                         
 289                         if (!ext_tree_lookup(bl, isect, &be, false)) {
 290                                 header->pnfs_error = -EIO;
 291                                 goto out;
 292                         }
 293                         extent_length = be.be_length - (isect - be.be_f_offset);
 294                 }
 295 
 296                 if (is_dio) {
 297                         if (pg_offset + bytes_left > PAGE_SIZE)
 298                                 pg_len = PAGE_SIZE - pg_offset;
 299                         else
 300                                 pg_len = bytes_left;
 301                 } else {
 302                         BUG_ON(pg_offset != 0);
 303                         pg_len = PAGE_SIZE;
 304                 }
 305 
 306                 if (is_hole(&be)) {
 307                         bio = bl_submit_bio(bio);
 308                         
 309                         dprintk("%s Zeroing page for hole\n", __func__);
 310                         zero_user_segment(pages[i], pg_offset, pg_len);
 311 
 312                         
 313                         map.start = NFS4_MAX_UINT64;
 314                 } else {
 315                         bio = do_add_page_to_bio(bio,
 316                                                  header->page_array.npages - i,
 317                                                  READ,
 318                                                  isect, pages[i], &map, &be,
 319                                                  bl_end_io_read, par,
 320                                                  pg_offset, &pg_len);
 321                         if (IS_ERR(bio)) {
 322                                 header->pnfs_error = PTR_ERR(bio);
 323                                 bio = NULL;
 324                                 goto out;
 325                         }
 326                 }
 327                 isect += (pg_len >> SECTOR_SHIFT);
 328                 extent_length -= (pg_len >> SECTOR_SHIFT);
 329                 f_offset += pg_len;
 330                 bytes_left -= pg_len;
 331                 pg_offset = 0;
 332         }
 333         if ((isect << SECTOR_SHIFT) >= header->inode->i_size) {
 334                 header->res.eof = 1;
 335                 header->res.count = header->inode->i_size - header->args.offset;
 336         } else {
 337                 header->res.count = (isect << SECTOR_SHIFT) - header->args.offset;
 338         }
 339 out:
 340         bl_submit_bio(bio);
 341         blk_finish_plug(&plug);
 342         put_parallel(par);
 343         return PNFS_ATTEMPTED;
 344 }
 345 
 346 static void bl_end_io_write(struct bio *bio)
 347 {
 348         struct parallel_io *par = bio->bi_private;
 349         struct nfs_pgio_header *header = par->data;
 350 
 351         if (bio->bi_status) {
 352                 if (!header->pnfs_error)
 353                         header->pnfs_error = -EIO;
 354                 pnfs_set_lo_fail(header->lseg);
 355                 bl_mark_devices_unavailable(header, true);
 356         }
 357         bio_put(bio);
 358         put_parallel(par);
 359 }
 360 
 361 
 362 
 363 
 364 static void bl_write_cleanup(struct work_struct *work)
 365 {
 366         struct rpc_task *task = container_of(work, struct rpc_task, u.tk_work);
 367         struct nfs_pgio_header *hdr =
 368                         container_of(task, struct nfs_pgio_header, task);
 369 
 370         dprintk("%s enter\n", __func__);
 371 
 372         if (likely(!hdr->pnfs_error)) {
 373                 struct pnfs_block_layout *bl = BLK_LSEG2EXT(hdr->lseg);
 374                 u64 start = hdr->args.offset & (loff_t)PAGE_MASK;
 375                 u64 end = (hdr->args.offset + hdr->args.count +
 376                         PAGE_SIZE - 1) & (loff_t)PAGE_MASK;
 377                 u64 lwb = hdr->args.offset + hdr->args.count;
 378 
 379                 ext_tree_mark_written(bl, start >> SECTOR_SHIFT,
 380                                         (end - start) >> SECTOR_SHIFT, lwb);
 381         }
 382 
 383         pnfs_ld_write_done(hdr);
 384 }
 385 
 386 
 387 static void bl_end_par_io_write(void *data)
 388 {
 389         struct nfs_pgio_header *hdr = data;
 390 
 391         hdr->task.tk_status = hdr->pnfs_error;
 392         hdr->verf.committed = NFS_FILE_SYNC;
 393         INIT_WORK(&hdr->task.u.tk_work, bl_write_cleanup);
 394         schedule_work(&hdr->task.u.tk_work);
 395 }
 396 
 397 static enum pnfs_try_status
 398 bl_write_pagelist(struct nfs_pgio_header *header, int sync)
 399 {
 400         struct pnfs_block_layout *bl = BLK_LSEG2EXT(header->lseg);
 401         struct pnfs_block_dev_map map = { .start = NFS4_MAX_UINT64 };
 402         struct bio *bio = NULL;
 403         struct pnfs_block_extent be;
 404         sector_t isect, extent_length = 0;
 405         struct parallel_io *par = NULL;
 406         loff_t offset = header->args.offset;
 407         size_t count = header->args.count;
 408         struct page **pages = header->args.pages;
 409         int pg_index = header->args.pgbase >> PAGE_SHIFT;
 410         unsigned int pg_len;
 411         struct blk_plug plug;
 412         int i;
 413 
 414         dprintk("%s enter, %zu@%lld\n", __func__, count, offset);
 415 
 416         
 417 
 418 
 419 
 420         par = alloc_parallel(header);
 421         if (!par)
 422                 return PNFS_NOT_ATTEMPTED;
 423         par->pnfs_callback = bl_end_par_io_write;
 424 
 425         blk_start_plug(&plug);
 426 
 427         
 428         offset = offset & (loff_t)PAGE_MASK;
 429         isect = offset >> SECTOR_SHIFT;
 430 
 431         for (i = pg_index; i < header->page_array.npages; i++) {
 432                 if (extent_length <= 0) {
 433                         
 434                         bio = bl_submit_bio(bio);
 435                         
 436                         if (!ext_tree_lookup(bl, isect, &be, true)) {
 437                                 header->pnfs_error = -EINVAL;
 438                                 goto out;
 439                         }
 440 
 441                         extent_length = be.be_length - (isect - be.be_f_offset);
 442                 }
 443 
 444                 pg_len = PAGE_SIZE;
 445                 bio = do_add_page_to_bio(bio, header->page_array.npages - i,
 446                                          WRITE, isect, pages[i], &map, &be,
 447                                          bl_end_io_write, par,
 448                                          0, &pg_len);
 449                 if (IS_ERR(bio)) {
 450                         header->pnfs_error = PTR_ERR(bio);
 451                         bio = NULL;
 452                         goto out;
 453                 }
 454 
 455                 offset += pg_len;
 456                 count -= pg_len;
 457                 isect += (pg_len >> SECTOR_SHIFT);
 458                 extent_length -= (pg_len >> SECTOR_SHIFT);
 459         }
 460 
 461         header->res.count = header->args.count;
 462 out:
 463         bl_submit_bio(bio);
 464         blk_finish_plug(&plug);
 465         put_parallel(par);
 466         return PNFS_ATTEMPTED;
 467 }
 468 
 469 static void bl_free_layout_hdr(struct pnfs_layout_hdr *lo)
 470 {
 471         struct pnfs_block_layout *bl = BLK_LO2EXT(lo);
 472         int err;
 473 
 474         dprintk("%s enter\n", __func__);
 475 
 476         err = ext_tree_remove(bl, true, 0, LLONG_MAX);
 477         WARN_ON(err);
 478 
 479         kfree(bl);
 480 }
 481 
 482 static struct pnfs_layout_hdr *__bl_alloc_layout_hdr(struct inode *inode,
 483                 gfp_t gfp_flags, bool is_scsi_layout)
 484 {
 485         struct pnfs_block_layout *bl;
 486 
 487         dprintk("%s enter\n", __func__);
 488         bl = kzalloc(sizeof(*bl), gfp_flags);
 489         if (!bl)
 490                 return NULL;
 491 
 492         bl->bl_ext_rw = RB_ROOT;
 493         bl->bl_ext_ro = RB_ROOT;
 494         spin_lock_init(&bl->bl_ext_lock);
 495 
 496         bl->bl_scsi_layout = is_scsi_layout;
 497         return &bl->bl_layout;
 498 }
 499 
 500 static struct pnfs_layout_hdr *bl_alloc_layout_hdr(struct inode *inode,
 501                                                    gfp_t gfp_flags)
 502 {
 503         return __bl_alloc_layout_hdr(inode, gfp_flags, false);
 504 }
 505 
 506 static struct pnfs_layout_hdr *sl_alloc_layout_hdr(struct inode *inode,
 507                                                    gfp_t gfp_flags)
 508 {
 509         return __bl_alloc_layout_hdr(inode, gfp_flags, true);
 510 }
 511 
 512 static void bl_free_lseg(struct pnfs_layout_segment *lseg)
 513 {
 514         dprintk("%s enter\n", __func__);
 515         kfree(lseg);
 516 }
 517 
 518 
 519 struct layout_verification {
 520         u32 mode;       
 521         u64 start;      
 522         u64 inval;      
 523         u64 cowread;    
 524 };
 525 
 526 
 527 
 528 
 529 static int verify_extent(struct pnfs_block_extent *be,
 530                          struct layout_verification *lv)
 531 {
 532         if (lv->mode == IOMODE_READ) {
 533                 if (be->be_state == PNFS_BLOCK_READWRITE_DATA ||
 534                     be->be_state == PNFS_BLOCK_INVALID_DATA)
 535                         return -EIO;
 536                 if (be->be_f_offset != lv->start)
 537                         return -EIO;
 538                 lv->start += be->be_length;
 539                 return 0;
 540         }
 541         
 542         if (be->be_state == PNFS_BLOCK_READWRITE_DATA) {
 543                 if (be->be_f_offset != lv->start)
 544                         return -EIO;
 545                 if (lv->cowread > lv->start)
 546                         return -EIO;
 547                 lv->start += be->be_length;
 548                 lv->inval = lv->start;
 549                 return 0;
 550         } else if (be->be_state == PNFS_BLOCK_INVALID_DATA) {
 551                 if (be->be_f_offset != lv->start)
 552                         return -EIO;
 553                 lv->start += be->be_length;
 554                 return 0;
 555         } else if (be->be_state == PNFS_BLOCK_READ_DATA) {
 556                 if (be->be_f_offset > lv->start)
 557                         return -EIO;
 558                 if (be->be_f_offset < lv->inval)
 559                         return -EIO;
 560                 if (be->be_f_offset < lv->cowread)
 561                         return -EIO;
 562                 
 563 
 564 
 565                 lv->inval = lv->inval + be->be_length;
 566                 lv->cowread = be->be_f_offset + be->be_length;
 567                 return 0;
 568         } else
 569                 return -EIO;
 570 }
 571 
 572 static int decode_sector_number(__be32 **rp, sector_t *sp)
 573 {
 574         uint64_t s;
 575 
 576         *rp = xdr_decode_hyper(*rp, &s);
 577         if (s & 0x1ff) {
 578                 printk(KERN_WARNING "NFS: %s: sector not aligned\n", __func__);
 579                 return -1;
 580         }
 581         *sp = s >> SECTOR_SHIFT;
 582         return 0;
 583 }
 584 
 585 static struct nfs4_deviceid_node *
 586 bl_find_get_deviceid(struct nfs_server *server,
 587                 const struct nfs4_deviceid *id, const struct cred *cred,
 588                 gfp_t gfp_mask)
 589 {
 590         struct nfs4_deviceid_node *node;
 591         unsigned long start, end;
 592 
 593 retry:
 594         node = nfs4_find_get_deviceid(server, id, cred, gfp_mask);
 595         if (!node)
 596                 return ERR_PTR(-ENODEV);
 597 
 598         if (test_bit(NFS_DEVICEID_UNAVAILABLE, &node->flags) == 0)
 599                 return node;
 600 
 601         end = jiffies;
 602         start = end - PNFS_DEVICE_RETRY_TIMEOUT;
 603         if (!time_in_range(node->timestamp_unavailable, start, end)) {
 604                 nfs4_delete_deviceid(node->ld, node->nfs_client, id);
 605                 goto retry;
 606         }
 607         return ERR_PTR(-ENODEV);
 608 }
 609 
 610 static int
 611 bl_alloc_extent(struct xdr_stream *xdr, struct pnfs_layout_hdr *lo,
 612                 struct layout_verification *lv, struct list_head *extents,
 613                 gfp_t gfp_mask)
 614 {
 615         struct pnfs_block_extent *be;
 616         struct nfs4_deviceid id;
 617         int error;
 618         __be32 *p;
 619 
 620         p = xdr_inline_decode(xdr, 28 + NFS4_DEVICEID4_SIZE);
 621         if (!p)
 622                 return -EIO;
 623 
 624         be = kzalloc(sizeof(*be), GFP_NOFS);
 625         if (!be)
 626                 return -ENOMEM;
 627 
 628         memcpy(&id, p, NFS4_DEVICEID4_SIZE);
 629         p += XDR_QUADLEN(NFS4_DEVICEID4_SIZE);
 630 
 631         be->be_device = bl_find_get_deviceid(NFS_SERVER(lo->plh_inode), &id,
 632                                                 lo->plh_lc_cred, gfp_mask);
 633         if (IS_ERR(be->be_device)) {
 634                 error = PTR_ERR(be->be_device);
 635                 goto out_free_be;
 636         }
 637 
 638         
 639 
 640 
 641 
 642         error = -EIO;
 643         if (decode_sector_number(&p, &be->be_f_offset) < 0)
 644                 goto out_put_deviceid;
 645         if (decode_sector_number(&p, &be->be_length) < 0)
 646                 goto out_put_deviceid;
 647         if (decode_sector_number(&p, &be->be_v_offset) < 0)
 648                 goto out_put_deviceid;
 649         be->be_state = be32_to_cpup(p++);
 650 
 651         error = verify_extent(be, lv);
 652         if (error) {
 653                 dprintk("%s: extent verification failed\n", __func__);
 654                 goto out_put_deviceid;
 655         }
 656 
 657         list_add_tail(&be->be_list, extents);
 658         return 0;
 659 
 660 out_put_deviceid:
 661         nfs4_put_deviceid_node(be->be_device);
 662 out_free_be:
 663         kfree(be);
 664         return error;
 665 }
 666 
 667 static struct pnfs_layout_segment *
 668 bl_alloc_lseg(struct pnfs_layout_hdr *lo, struct nfs4_layoutget_res *lgr,
 669                 gfp_t gfp_mask)
 670 {
 671         struct layout_verification lv = {
 672                 .mode = lgr->range.iomode,
 673                 .start = lgr->range.offset >> SECTOR_SHIFT,
 674                 .inval = lgr->range.offset >> SECTOR_SHIFT,
 675                 .cowread = lgr->range.offset >> SECTOR_SHIFT,
 676         };
 677         struct pnfs_block_layout *bl = BLK_LO2EXT(lo);
 678         struct pnfs_layout_segment *lseg;
 679         struct xdr_buf buf;
 680         struct xdr_stream xdr;
 681         struct page *scratch;
 682         int status, i;
 683         uint32_t count;
 684         __be32 *p;
 685         LIST_HEAD(extents);
 686 
 687         dprintk("---> %s\n", __func__);
 688 
 689         lseg = kzalloc(sizeof(*lseg), gfp_mask);
 690         if (!lseg)
 691                 return ERR_PTR(-ENOMEM);
 692 
 693         status = -ENOMEM;
 694         scratch = alloc_page(gfp_mask);
 695         if (!scratch)
 696                 goto out;
 697 
 698         xdr_init_decode_pages(&xdr, &buf,
 699                         lgr->layoutp->pages, lgr->layoutp->len);
 700         xdr_set_scratch_buffer(&xdr, page_address(scratch), PAGE_SIZE);
 701 
 702         status = -EIO;
 703         p = xdr_inline_decode(&xdr, 4);
 704         if (unlikely(!p))
 705                 goto out_free_scratch;
 706 
 707         count = be32_to_cpup(p++);
 708         dprintk("%s: number of extents %d\n", __func__, count);
 709 
 710         
 711 
 712 
 713 
 714         for (i = 0; i < count; i++) {
 715                 status = bl_alloc_extent(&xdr, lo, &lv, &extents, gfp_mask);
 716                 if (status)
 717                         goto process_extents;
 718         }
 719 
 720         if (lgr->range.offset + lgr->range.length !=
 721                         lv.start << SECTOR_SHIFT) {
 722                 dprintk("%s Final length mismatch\n", __func__);
 723                 status = -EIO;
 724                 goto process_extents;
 725         }
 726 
 727         if (lv.start < lv.cowread) {
 728                 dprintk("%s Final uncovered COW extent\n", __func__);
 729                 status = -EIO;
 730         }
 731 
 732 process_extents:
 733         while (!list_empty(&extents)) {
 734                 struct pnfs_block_extent *be =
 735                         list_first_entry(&extents, struct pnfs_block_extent,
 736                                          be_list);
 737                 list_del(&be->be_list);
 738 
 739                 if (!status)
 740                         status = ext_tree_insert(bl, be);
 741 
 742                 if (status) {
 743                         nfs4_put_deviceid_node(be->be_device);
 744                         kfree(be);
 745                 }
 746         }
 747 
 748 out_free_scratch:
 749         __free_page(scratch);
 750 out:
 751         dprintk("%s returns %d\n", __func__, status);
 752         switch (status) {
 753         case -ENODEV:
 754                 
 755                 set_bit(NFS_LSEG_UNAVAILABLE, &lseg->pls_flags);
 756                 
 757         case 0:
 758                 return lseg;
 759         default:
 760                 kfree(lseg);
 761                 return ERR_PTR(status);
 762         }
 763 }
 764 
 765 static void
 766 bl_return_range(struct pnfs_layout_hdr *lo,
 767                 struct pnfs_layout_range *range)
 768 {
 769         struct pnfs_block_layout *bl = BLK_LO2EXT(lo);
 770         sector_t offset = range->offset >> SECTOR_SHIFT, end;
 771 
 772         if (range->offset % 8) {
 773                 dprintk("%s: offset %lld not block size aligned\n",
 774                         __func__, range->offset);
 775                 return;
 776         }
 777 
 778         if (range->length != NFS4_MAX_UINT64) {
 779                 if (range->length % 8) {
 780                         dprintk("%s: length %lld not block size aligned\n",
 781                                 __func__, range->length);
 782                         return;
 783                 }
 784 
 785                 end = offset + (range->length >> SECTOR_SHIFT);
 786         } else {
 787                 end = round_down(NFS4_MAX_UINT64, PAGE_SIZE);
 788         }
 789 
 790         ext_tree_remove(bl, range->iomode & IOMODE_RW, offset, end);
 791 }
 792 
 793 static int
 794 bl_prepare_layoutcommit(struct nfs4_layoutcommit_args *arg)
 795 {
 796         return ext_tree_prepare_commit(arg);
 797 }
 798 
 799 static void
 800 bl_cleanup_layoutcommit(struct nfs4_layoutcommit_data *lcdata)
 801 {
 802         ext_tree_mark_committed(&lcdata->args, lcdata->res.status);
 803 }
 804 
 805 static int
 806 bl_set_layoutdriver(struct nfs_server *server, const struct nfs_fh *fh)
 807 {
 808         dprintk("%s enter\n", __func__);
 809 
 810         if (server->pnfs_blksize == 0) {
 811                 dprintk("%s Server did not return blksize\n", __func__);
 812                 return -EINVAL;
 813         }
 814         if (server->pnfs_blksize > PAGE_SIZE) {
 815                 printk(KERN_ERR "%s: pNFS blksize %d not supported.\n",
 816                         __func__, server->pnfs_blksize);
 817                 return -EINVAL;
 818         }
 819 
 820         return 0;
 821 }
 822 
 823 static bool
 824 is_aligned_req(struct nfs_pageio_descriptor *pgio,
 825                 struct nfs_page *req, unsigned int alignment, bool is_write)
 826 {
 827         
 828 
 829 
 830 
 831         if (pgio->pg_dreq == NULL)
 832                 return true;
 833 
 834         if (!IS_ALIGNED(req->wb_offset, alignment))
 835                 return false;
 836 
 837         if (IS_ALIGNED(req->wb_bytes, alignment))
 838                 return true;
 839 
 840         if (is_write &&
 841             (req_offset(req) + req->wb_bytes == i_size_read(pgio->pg_inode))) {
 842                 
 843 
 844 
 845 
 846 
 847 
 848 
 849                 return true;
 850         }
 851 
 852         return false;
 853 }
 854 
 855 static void
 856 bl_pg_init_read(struct nfs_pageio_descriptor *pgio, struct nfs_page *req)
 857 {
 858         if (!is_aligned_req(pgio, req, SECTOR_SIZE, false)) {
 859                 nfs_pageio_reset_read_mds(pgio);
 860                 return;
 861         }
 862 
 863         pnfs_generic_pg_init_read(pgio, req);
 864 
 865         if (pgio->pg_lseg &&
 866                 test_bit(NFS_LSEG_UNAVAILABLE, &pgio->pg_lseg->pls_flags)) {
 867                 pnfs_error_mark_layout_for_return(pgio->pg_inode, pgio->pg_lseg);
 868                 pnfs_set_lo_fail(pgio->pg_lseg);
 869                 nfs_pageio_reset_read_mds(pgio);
 870         }
 871 }
 872 
 873 
 874 
 875 
 876 
 877 static size_t
 878 bl_pg_test_read(struct nfs_pageio_descriptor *pgio, struct nfs_page *prev,
 879                 struct nfs_page *req)
 880 {
 881         if (!is_aligned_req(pgio, req, SECTOR_SIZE, false))
 882                 return 0;
 883         return pnfs_generic_pg_test(pgio, prev, req);
 884 }
 885 
 886 
 887 
 888 
 889 
 890 static u64 pnfs_num_cont_bytes(struct inode *inode, pgoff_t idx)
 891 {
 892         struct address_space *mapping = inode->i_mapping;
 893         pgoff_t end;
 894 
 895         
 896         end = DIV_ROUND_UP(i_size_read(inode), PAGE_SIZE);
 897         if (end != inode->i_mapping->nrpages) {
 898                 rcu_read_lock();
 899                 end = page_cache_next_miss(mapping, idx + 1, ULONG_MAX);
 900                 rcu_read_unlock();
 901         }
 902 
 903         if (!end)
 904                 return i_size_read(inode) - (idx << PAGE_SHIFT);
 905         else
 906                 return (end - idx) << PAGE_SHIFT;
 907 }
 908 
 909 static void
 910 bl_pg_init_write(struct nfs_pageio_descriptor *pgio, struct nfs_page *req)
 911 {
 912         u64 wb_size;
 913 
 914         if (!is_aligned_req(pgio, req, PAGE_SIZE, true)) {
 915                 nfs_pageio_reset_write_mds(pgio);
 916                 return;
 917         }
 918 
 919         if (pgio->pg_dreq == NULL)
 920                 wb_size = pnfs_num_cont_bytes(pgio->pg_inode,
 921                                               req->wb_index);
 922         else
 923                 wb_size = nfs_dreq_bytes_left(pgio->pg_dreq);
 924 
 925         pnfs_generic_pg_init_write(pgio, req, wb_size);
 926 
 927         if (pgio->pg_lseg &&
 928                 test_bit(NFS_LSEG_UNAVAILABLE, &pgio->pg_lseg->pls_flags)) {
 929 
 930                 pnfs_error_mark_layout_for_return(pgio->pg_inode, pgio->pg_lseg);
 931                 pnfs_set_lo_fail(pgio->pg_lseg);
 932                 nfs_pageio_reset_write_mds(pgio);
 933         }
 934 }
 935 
 936 
 937 
 938 
 939 
 940 static size_t
 941 bl_pg_test_write(struct nfs_pageio_descriptor *pgio, struct nfs_page *prev,
 942                  struct nfs_page *req)
 943 {
 944         if (!is_aligned_req(pgio, req, PAGE_SIZE, true))
 945                 return 0;
 946         return pnfs_generic_pg_test(pgio, prev, req);
 947 }
 948 
 949 static const struct nfs_pageio_ops bl_pg_read_ops = {
 950         .pg_init = bl_pg_init_read,
 951         .pg_test = bl_pg_test_read,
 952         .pg_doio = pnfs_generic_pg_readpages,
 953         .pg_cleanup = pnfs_generic_pg_cleanup,
 954 };
 955 
 956 static const struct nfs_pageio_ops bl_pg_write_ops = {
 957         .pg_init = bl_pg_init_write,
 958         .pg_test = bl_pg_test_write,
 959         .pg_doio = pnfs_generic_pg_writepages,
 960         .pg_cleanup = pnfs_generic_pg_cleanup,
 961 };
 962 
 963 static struct pnfs_layoutdriver_type blocklayout_type = {
 964         .id                             = LAYOUT_BLOCK_VOLUME,
 965         .name                           = "LAYOUT_BLOCK_VOLUME",
 966         .owner                          = THIS_MODULE,
 967         .flags                          = PNFS_LAYOUTRET_ON_SETATTR |
 968                                           PNFS_LAYOUTRET_ON_ERROR |
 969                                           PNFS_READ_WHOLE_PAGE,
 970         .read_pagelist                  = bl_read_pagelist,
 971         .write_pagelist                 = bl_write_pagelist,
 972         .alloc_layout_hdr               = bl_alloc_layout_hdr,
 973         .free_layout_hdr                = bl_free_layout_hdr,
 974         .alloc_lseg                     = bl_alloc_lseg,
 975         .free_lseg                      = bl_free_lseg,
 976         .return_range                   = bl_return_range,
 977         .prepare_layoutcommit           = bl_prepare_layoutcommit,
 978         .cleanup_layoutcommit           = bl_cleanup_layoutcommit,
 979         .set_layoutdriver               = bl_set_layoutdriver,
 980         .alloc_deviceid_node            = bl_alloc_deviceid_node,
 981         .free_deviceid_node             = bl_free_deviceid_node,
 982         .pg_read_ops                    = &bl_pg_read_ops,
 983         .pg_write_ops                   = &bl_pg_write_ops,
 984         .sync                           = pnfs_generic_sync,
 985 };
 986 
 987 static struct pnfs_layoutdriver_type scsilayout_type = {
 988         .id                             = LAYOUT_SCSI,
 989         .name                           = "LAYOUT_SCSI",
 990         .owner                          = THIS_MODULE,
 991         .flags                          = PNFS_LAYOUTRET_ON_SETATTR |
 992                                           PNFS_LAYOUTRET_ON_ERROR |
 993                                           PNFS_READ_WHOLE_PAGE,
 994         .read_pagelist                  = bl_read_pagelist,
 995         .write_pagelist                 = bl_write_pagelist,
 996         .alloc_layout_hdr               = sl_alloc_layout_hdr,
 997         .free_layout_hdr                = bl_free_layout_hdr,
 998         .alloc_lseg                     = bl_alloc_lseg,
 999         .free_lseg                      = bl_free_lseg,
1000         .return_range                   = bl_return_range,
1001         .prepare_layoutcommit           = bl_prepare_layoutcommit,
1002         .cleanup_layoutcommit           = bl_cleanup_layoutcommit,
1003         .set_layoutdriver               = bl_set_layoutdriver,
1004         .alloc_deviceid_node            = bl_alloc_deviceid_node,
1005         .free_deviceid_node             = bl_free_deviceid_node,
1006         .pg_read_ops                    = &bl_pg_read_ops,
1007         .pg_write_ops                   = &bl_pg_write_ops,
1008         .sync                           = pnfs_generic_sync,
1009 };
1010 
1011 
1012 static int __init nfs4blocklayout_init(void)
1013 {
1014         int ret;
1015 
1016         dprintk("%s: NFSv4 Block Layout Driver Registering...\n", __func__);
1017 
1018         ret = bl_init_pipefs();
1019         if (ret)
1020                 goto out;
1021 
1022         ret = pnfs_register_layoutdriver(&blocklayout_type);
1023         if (ret)
1024                 goto out_cleanup_pipe;
1025 
1026         ret = pnfs_register_layoutdriver(&scsilayout_type);
1027         if (ret)
1028                 goto out_unregister_block;
1029         return 0;
1030 
1031 out_unregister_block:
1032         pnfs_unregister_layoutdriver(&blocklayout_type);
1033 out_cleanup_pipe:
1034         bl_cleanup_pipefs();
1035 out:
1036         return ret;
1037 }
1038 
1039 static void __exit nfs4blocklayout_exit(void)
1040 {
1041         dprintk("%s: NFSv4 Block Layout Driver Unregistering...\n",
1042                __func__);
1043 
1044         pnfs_unregister_layoutdriver(&scsilayout_type);
1045         pnfs_unregister_layoutdriver(&blocklayout_type);
1046         bl_cleanup_pipefs();
1047 }
1048 
1049 MODULE_ALIAS("nfs-layouttype4-3");
1050 MODULE_ALIAS("nfs-layouttype4-5");
1051 
1052 module_init(nfs4blocklayout_init);
1053 module_exit(nfs4blocklayout_exit);