lustre/osd-zfs/osd_io.c

   1 /*
   2  * GPL HEADER START
   3  *
   4  * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
   5  *
   6  * This program is free software; you can redistribute it and/or modify
   7  * it under the terms of the GNU General Public License version 2 only,
   8  * as published by the Free Software Foundation.
   9  *
  10  * This program is distributed in the hope that it will be useful, but
  11  * WITHOUT ANY WARRANTY; without even the implied warranty of
  12  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  13  * General Public License version 2 for more details (a copy is included
  14  * in the LICENSE file that accompanied this code).
  15  *
  16  * You should have received a copy of the GNU General Public License
  17  * version 2 along with this program; If not, see
  18  * http://www.sun.com/software/products/lustre/docs/GPLv2.pdf
  19  *
  20  * Please contact Sun Microsystems, Inc., 4150 Network Circle, Santa Clara,
  21  * CA 95054 USA or visit www.sun.com if you need additional information or
  22  * have any questions.
  23  *
  24  * GPL HEADER END
  25  */
  26 /*
  27  * Copyright (c) 2009, 2010, Oracle and/or its affiliates. All rights reserved.
  28  * Use is subject to license terms.
  29  */
  30 /*
  31  * Copyright (c) 2012, Intel Corporation.
  32  * Use is subject to license terms.
  33  */
  34 /*
  35  * This file is part of Lustre, http://www.lustre.org/
  36  * Lustre is a trademark of Sun Microsystems, Inc.
  37  *
  38  * lustre/osd-zfs/osd_io.c
  39  *
  40  * Author: Alex Zhuravlev <bzzz@whamcloud.com>
  41  * Author: Mike Pershin <tappro@whamcloud.com>
  42  */
  43
  44 #ifndef EXPORT_SYMTAB
  45 # define EXPORT_SYMTAB
  46 #endif
  47 #define DEBUG_SUBSYSTEM S_OSD
  48
  49 #include <lustre_ver.h>
  50 #include <libcfs/libcfs.h>
  51 #include <lustre_fsfilt.h>
  52 #include <obd_support.h>
  53 #include <lustre_net.h>
  54 #include <obd.h>
  55 #include <obd_class.h>
  56 #include <lustre_disk.h>
  57 #include <lustre_fid.h>
  58
  59 #include "osd_internal.h"
  60
  61 #include <sys/dnode.h>
  62 #include <sys/dbuf.h>
  63 #include <sys/spa.h>
  64 #include <sys/stat.h>
  65 #include <sys/zap.h>
  66 #include <sys/spa_impl.h>
  67 #include <sys/zfs_znode.h>
  68 #include <sys/dmu_tx.h>
  69 #include <sys/dmu_objset.h>
  70 #include <sys/dsl_prop.h>
  71 #include <sys/sa_impl.h>
  72 #include <sys/txg.h>
  73
  74 static char *osd_zerocopy_tag = "zerocopy";
  75
  76 static ssize_t osd_read(const struct lu_env *env, struct dt_object *dt,
  77                         struct lu_buf *buf, loff_t *pos,
  78                         struct lustre_capa *capa)
  79 {
  80         struct osd_object *obj  = osd_dt_obj(dt);
  81         struct osd_device *osd = osd_obj2dev(obj);
  82         uint64_t           old_size;
  83         int                size = buf->lb_len;
  84         int                rc;
  85
  86         LASSERT(dt_object_exists(dt));
  87         LASSERT(obj->oo_db);
  88
  89         read_lock(&obj->oo_attr_lock);
  90         old_size = obj->oo_attr.la_size;
  91         read_unlock(&obj->oo_attr_lock);
  92
  93         if (*pos + size > old_size) {
  94                 if (old_size < *pos)
  95                         return 0;
  96                 else
  97                         size = old_size - *pos;
  98         }
  99
 100         rc = -dmu_read(osd->od_objset.os, obj->oo_db->db_object, *pos, size,
 101                         buf->lb_buf, DMU_READ_PREFETCH);
 102         if (rc == 0) {
 103                 rc = size;
 104                 *pos += size;
 105
 106                 /* XXX: workaround for bug in HEAD: fsfilt_ldiskfs_read() returns
 107                  * requested number of bytes, not actually read ones */
 108                 if (S_ISLNK(obj->oo_dt.do_lu.lo_header->loh_attr))
 109                         rc = buf->lb_len;
 110         }
 111         return rc;
 112 }
 113
 114 static ssize_t osd_declare_write(const struct lu_env *env, struct dt_object *dt,
 115                                 const loff_t size, loff_t pos,
 116                                 struct thandle *th)
 117 {
 118         struct osd_object  *obj  = osd_dt_obj(dt);
 119         struct osd_device  *osd = osd_obj2dev(obj);
 120         struct osd_thandle *oh;
 121         uint64_t            oid;
 122         ENTRY;
 123
 124         oh = container_of0(th, struct osd_thandle, ot_super);
 125
 126         /* in some cases declare can race with creation (e.g. llog)
 127          * and we need to wait till object is initialized. notice
 128          * LOHA_EXISTs is supposed to be the last step in the
 129          * initialization */
 130
 131         /* declare possible size change. notice we can't check
 132          * current size here as another thread can change it */
 133
 134         if (dt_object_exists(dt)) {
 135                 LASSERT(obj->oo_db);
 136                 oid = obj->oo_db->db_object;
 137
 138                 dmu_tx_hold_sa(oh->ot_tx, obj->oo_sa_hdl, 0);
 139         } else {
 140                 oid = DMU_NEW_OBJECT;
 141                 dmu_tx_hold_sa_create(oh->ot_tx, ZFS_SA_BASE_ATTR_SIZE);
 142         }
 143
 144         dmu_tx_hold_write(oh->ot_tx, oid, pos, size);
 145
 146         /* dt_declare_write() is usually called for system objects, such
 147          * as llog or last_rcvd files. We needn't enforce quota on those
 148          * objects, so always set the lqi_space as 0. */
 149         RETURN(osd_declare_quota(env, osd, obj->oo_attr.la_uid,
 150                                  obj->oo_attr.la_gid, 0, oh, true, NULL,
 151                                  false));
 152 }
 153
 154 static ssize_t osd_write(const struct lu_env *env, struct dt_object *dt,
 155                         const struct lu_buf *buf, loff_t *pos,
 156                         struct thandle *th, struct lustre_capa *capa,
 157                         int ignore_quota)
 158 {
 159         struct osd_object  *obj  = osd_dt_obj(dt);
 160         struct osd_device  *osd = osd_obj2dev(obj);
 161         udmu_objset_t      *uos = &osd->od_objset;
 162         struct osd_thandle *oh;
 163         uint64_t            offset = *pos;
 164         int                 rc;
 165         ENTRY;
 166
 167         LASSERT(dt_object_exists(dt));
 168         LASSERT(obj->oo_db);
 169
 170         LASSERT(th != NULL);
 171         oh = container_of0(th, struct osd_thandle, ot_super);
 172
 173         dmu_write(osd->od_objset.os, obj->oo_db->db_object, offset,
 174                 (uint64_t)buf->lb_len, buf->lb_buf, oh->ot_tx);
 175         write_lock(&obj->oo_attr_lock);
 176         if (obj->oo_attr.la_size < offset + buf->lb_len) {
 177                 obj->oo_attr.la_size = offset + buf->lb_len;
 178                 write_unlock(&obj->oo_attr_lock);
 179                 /* osd_object_sa_update() will be copying directly from oo_attr
 180                  * into dbuf.  any update within a single txg will copy the
 181                  * most actual */
 182                 rc = osd_object_sa_update(obj, SA_ZPL_SIZE(uos),
 183                                         &obj->oo_attr.la_size, 8, oh);
 184                 if (unlikely(rc))
 185                         GOTO(out, rc);
 186         } else {
 187                 write_unlock(&obj->oo_attr_lock);
 188         }
 189
 190         *pos += buf->lb_len;
 191         rc = buf->lb_len;
 192
 193 out:
 194         RETURN(rc);
 195 }
 196
 197 /*
 198  * XXX: for the moment I don't want to use lnb_flags for osd-internal
 199  *      purposes as it's not very well defined ...
 200  *      instead I use the lowest bit of the address so that:
 201  *        arc buffer:  .lnb_obj = abuf          (arc we loan for write)
 202  *        dbuf buffer: .lnb_obj = dbuf | 1      (dbuf we get for read)
 203  *        copy buffer: .lnb_page->mapping = obj (page we allocate for write)
 204  *
 205  *      bzzz, to blame
 206  */
 207 static int osd_bufs_put(const struct lu_env *env, struct dt_object *dt,
 208                         struct niobuf_local *lnb, int npages)
 209 {
 210         struct osd_object *obj  = osd_dt_obj(dt);
 211         struct osd_device *osd = osd_obj2dev(obj);
 212         unsigned long      ptr;
 213         int                i;
 214
 215         LASSERT(dt_object_exists(dt));
 216         LASSERT(obj->oo_db);
 217
 218         for (i = 0; i < npages; i++) {
 219                 if (lnb[i].page == NULL)
 220                         continue;
 221                 if (lnb[i].page->mapping == (void *)obj) {
 222                         /* this is anonymous page allocated for copy-write */
 223                         lnb[i].page->mapping = NULL;
 224                         __free_page(lnb[i].page);
 225                         cfs_atomic_dec(&osd->od_zerocopy_alloc);
 226                 } else {
 227                         /* see comment in osd_bufs_get_read() */
 228                         ptr = (unsigned long)lnb[i].dentry;
 229                         if (ptr & 1UL) {
 230                                 ptr &= ~1UL;
 231                                 dmu_buf_rele((void *)ptr, osd_zerocopy_tag);
 232                                 cfs_atomic_dec(&osd->od_zerocopy_pin);
 233                         } else if (lnb[i].dentry != NULL) {
 234                                 dmu_return_arcbuf((void *)lnb[i].dentry);
 235                                 cfs_atomic_dec(&osd->od_zerocopy_loan);
 236                         }
 237                 }
 238                 lnb[i].page = NULL;
 239                 lnb[i].dentry = NULL;
 240         }
 241
 242         return 0;
 243 }
 244
 245 static struct page *kmem_to_page(void *addr)
 246 {
 247         struct page *page;
 248
 249         if (kmem_virt(addr))
 250                 page = vmalloc_to_page(addr);
 251         else
 252                 page = virt_to_page(addr);
 253
 254         return page;
 255 }
 256
 257 static int osd_bufs_get_read(const struct lu_env *env, struct osd_object *obj,
 258                                 loff_t off, ssize_t len, struct niobuf_local *lnb)
 259 {
 260         struct osd_device *osd = osd_obj2dev(obj);
 261         dmu_buf_t        **dbp;
 262         int                rc, i, numbufs, npages = 0;
 263         ENTRY;
 264
 265         /* grab buffers for read:
 266          * OSD API let us to grab buffers first, then initiate IO(s)
 267          * so that all required IOs will be done in parallel, but at the
 268          * moment DMU doesn't provide us with a method to grab buffers.
 269          * If we discover this is a vital for good performance we
 270          * can get own replacement for dmu_buf_hold_array_by_bonus().
 271          */
 272         while (len > 0) {
 273                 rc = -dmu_buf_hold_array_by_bonus(obj->oo_db, off, len, TRUE,
 274                                                   osd_zerocopy_tag, &numbufs,
 275                                                   &dbp);
 276                 if (unlikely(rc))
 277                         GOTO(err, rc);
 278
 279                 for (i = 0; i < numbufs; i++) {
 280                         int bufoff, tocpy, thispage;
 281                         void *dbf = dbp[i];
 282
 283                         LASSERT(len > 0);
 284
 285                         cfs_atomic_inc(&osd->od_zerocopy_pin);
 286
 287                         bufoff = off - dbp[i]->db_offset;
 288                         tocpy = min_t(int, dbp[i]->db_size - bufoff, len);
 289
 290                         /* kind of trick to differentiate dbuf vs. arcbuf */
 291                         LASSERT(((unsigned long)dbp[i] & 1) == 0);
 292                         dbf = (void *) ((unsigned long)dbp[i] | 1);
 293
 294                         while (tocpy > 0) {
 295                                 thispage = CFS_PAGE_SIZE;
 296                                 thispage -= bufoff & (CFS_PAGE_SIZE - 1);
 297                                 thispage = min(tocpy, thispage);
 298
 299                                 lnb->rc = 0;
 300                                 lnb->lnb_file_offset = off;
 301                                 lnb->lnb_page_offset = bufoff & ~CFS_PAGE_MASK;
 302                                 lnb->len = thispage;
 303                                 lnb->page = kmem_to_page(dbp[i]->db_data +
 304                                                                 bufoff);
 305                                 /* mark just a single slot: we need this
 306                                  * reference to dbuf to be release once */
 307                                 lnb->dentry = dbf;
 308                                 dbf = NULL;
 309
 310                                 tocpy -= thispage;
 311                                 len -= thispage;
 312                                 bufoff += thispage;
 313                                 off += thispage;
 314
 315                                 npages++;
 316                                 lnb++;
 317                         }
 318
 319                         /* steal dbuf so dmu_buf_rele_array() cant release it */
 320                         dbp[i] = NULL;
 321                 }
 322
 323                 dmu_buf_rele_array(dbp, numbufs, osd_zerocopy_tag);
 324         }
 325
 326         RETURN(npages);
 327
 328 err:
 329         LASSERT(rc < 0);
 330         osd_bufs_put(env, &obj->oo_dt, lnb - npages, npages);
 331         RETURN(rc);
 332 }
 333
 334 static int osd_bufs_get_write(const struct lu_env *env, struct osd_object *obj,
 335                                 loff_t off, ssize_t len, struct niobuf_local *lnb)
 336 {
 337         struct osd_device *osd = osd_obj2dev(obj);
 338         int                plen, off_in_block, sz_in_block;
 339         int                i = 0, npages = 0;
 340         arc_buf_t         *abuf;
 341         uint32_t           bs;
 342         uint64_t           dummy;
 343         ENTRY;
 344
 345         dmu_object_size_from_db(obj->oo_db, &bs, &dummy);
 346
 347         /*
 348          * currently only full blocks are subject to zerocopy approach:
 349          * so that we're sure nobody is trying to update the same block
 350          */
 351         while (len > 0) {
 352                 LASSERT(npages < PTLRPC_MAX_BRW_PAGES);
 353
 354                 off_in_block = off & (bs - 1);
 355                 sz_in_block = min_t(int, bs - off_in_block, len);
 356
 357                 if (sz_in_block == bs) {
 358                         /* full block, try to use zerocopy */
 359
 360                         abuf = dmu_request_arcbuf(obj->oo_db, bs);
 361                         if (unlikely(abuf == NULL))
 362                                 GOTO(out_err, -ENOMEM);
 363
 364                         cfs_atomic_inc(&osd->od_zerocopy_loan);
 365
 366                         /* go over pages arcbuf contains, put them as
 367                          * local niobufs for ptlrpc's bulks */
 368                         while (sz_in_block > 0) {
 369                                 plen = min_t(int, sz_in_block, CFS_PAGE_SIZE);
 370
 371                                 lnb[i].lnb_file_offset = off;
 372                                 lnb[i].lnb_page_offset = 0;
 373                                 lnb[i].len = plen;
 374                                 lnb[i].rc = 0;
 375                                 if (sz_in_block == bs)
 376                                         lnb[i].dentry = (void *)abuf;
 377                                 else
 378                                         lnb[i].dentry = NULL;
 379
 380                                 /* this one is not supposed to fail */
 381                                 lnb[i].page = kmem_to_page(abuf->b_data +
 382                                                         off_in_block);
 383                                 LASSERT(lnb[i].page);
 384
 385                                 lprocfs_counter_add(osd->od_stats,
 386                                                 LPROC_OSD_ZEROCOPY_IO, 1);
 387
 388                                 sz_in_block -= plen;
 389                                 len -= plen;
 390                                 off += plen;
 391                                 off_in_block += plen;
 392                                 i++;
 393                                 npages++;
 394                         }
 395                 } else {
 396                         if (off_in_block == 0 && len < bs &&
 397                                         off + len >= obj->oo_attr.la_size)
 398                                 lprocfs_counter_add(osd->od_stats,
 399                                                 LPROC_OSD_TAIL_IO, 1);
 400
 401                         /* can't use zerocopy, allocate temp. buffers */
 402                         while (sz_in_block > 0) {
 403                                 plen = min_t(int, sz_in_block, CFS_PAGE_SIZE);
 404
 405                                 lnb[i].lnb_file_offset = off;
 406                                 lnb[i].lnb_page_offset = 0;
 407                                 lnb[i].len = plen;
 408                                 lnb[i].rc = 0;
 409                                 lnb[i].dentry = NULL;
 410
 411                                 lnb[i].page = alloc_page(OSD_GFP_IO);
 412                                 if (unlikely(lnb[i].page == NULL))
 413                                         GOTO(out_err, -ENOMEM);
 414
 415                                 LASSERT(lnb[i].page->mapping == NULL);
 416                                 lnb[i].page->mapping = (void *)obj;
 417
 418                                 cfs_atomic_inc(&osd->od_zerocopy_alloc);
 419                                 lprocfs_counter_add(osd->od_stats,
 420                                                 LPROC_OSD_COPY_IO, 1);
 421
 422                                 sz_in_block -= plen;
 423                                 len -= plen;
 424                                 off += plen;
 425                                 i++;
 426                                 npages++;
 427                         }
 428                 }
 429         }
 430
 431         RETURN(npages);
 432
 433 out_err:
 434         osd_bufs_put(env, &obj->oo_dt, lnb, npages);
 435         RETURN(-ENOMEM);
 436 }
 437
 438 static int osd_bufs_get(const struct lu_env *env, struct dt_object *dt,
 439                         loff_t offset, ssize_t len, struct niobuf_local *lnb,
 440                         int rw, struct lustre_capa *capa)
 441 {
 442         struct osd_object *obj  = osd_dt_obj(dt);
 443         int                rc;
 444
 445         LASSERT(dt_object_exists(dt));
 446         LASSERT(obj->oo_db);
 447
 448         if (rw == 0)
 449                 rc = osd_bufs_get_read(env, obj, offset, len, lnb);
 450         else
 451                 rc = osd_bufs_get_write(env, obj, offset, len, lnb);
 452
 453         return rc;
 454 }
 455
 456 static int osd_write_prep(const struct lu_env *env, struct dt_object *dt,
 457                         struct niobuf_local *lnb, int npages)
 458 {
 459         struct osd_object *obj = osd_dt_obj(dt);
 460
 461         LASSERT(dt_object_exists(dt));
 462         LASSERT(obj->oo_db);
 463
 464         return 0;
 465 }
 466
 467 /* Return number of blocks that aren't mapped in the [start, start + size]
 468  * region */
 469 static int osd_count_not_mapped(struct osd_object *obj, uint64_t start,
 470                                 uint32_t size)
 471 {
 472         dmu_buf_impl_t  *dbi = (dmu_buf_impl_t *)obj->oo_db;
 473         dmu_buf_impl_t  *db;
 474         dnode_t         *dn;
 475         uint32_t         blkshift;
 476         uint64_t         end, blkid;
 477         int              rc;
 478         ENTRY;
 479
 480         DB_DNODE_ENTER(dbi);
 481         dn = DB_DNODE(dbi);
 482
 483         if (dn->dn_maxblkid == 0) {
 484                 if (start + size <= dn->dn_datablksz)
 485                         GOTO(out, size = 0);
 486                 if (start < dn->dn_datablksz)
 487                         start = dn->dn_datablksz;
 488                 /* assume largest block size */
 489                 blkshift = SPA_MAXBLOCKSHIFT;
 490         } else {
 491                 /* blocksize can't change */
 492                 blkshift = dn->dn_datablkshift;
 493         }
 494
 495         /* compute address of last block */
 496         end = (start + size - 1) >> blkshift;
 497         /* align start on block boundaries */
 498         start >>= blkshift;
 499
 500         /* size is null, can't be mapped */
 501         if (obj->oo_attr.la_size == 0 || dn->dn_maxblkid == 0)
 502                 GOTO(out, size = (end - start + 1) << blkshift);
 503
 504         /* beyond EOF, can't be mapped */
 505         if (start > dn->dn_maxblkid)
 506                 GOTO(out, size = (end - start + 1) << blkshift);
 507
 508         size = 0;
 509         for (blkid = start; blkid <= end; blkid++) {
 510                 if (blkid == dn->dn_maxblkid)
 511                         /* this one is mapped for sure */
 512                         continue;
 513                 if (blkid > dn->dn_maxblkid) {
 514                         size += (end - blkid + 1) << blkshift;
 515                         GOTO(out, size);
 516                 }
 517
 518                 rc = dbuf_hold_impl(dn, 0, blkid, TRUE, FTAG, &db);
 519                 if (rc) {
 520                         /* for ENOENT (block not mapped) and any other errors,
 521                          * assume the block isn't mapped */
 522                         size += 1 << blkshift;
 523                         continue;
 524                 }
 525                 dbuf_rele(db, FTAG);
 526         }
 527
 528         GOTO(out, size);
 529 out:
 530         DB_DNODE_EXIT(dbi);
 531         return size;
 532 }
 533
 534 static int osd_declare_write_commit(const struct lu_env *env,
 535                                 struct dt_object *dt,
 536                                 struct niobuf_local *lnb, int npages,
 537                                 struct thandle *th)
 538 {
 539         struct osd_object  *obj = osd_dt_obj(dt);
 540         struct osd_device  *osd = osd_obj2dev(obj);
 541         struct osd_thandle *oh;
 542         uint64_t            offset = 0;
 543         uint32_t            size = 0;
 544         int                 i, rc, flags = 0;
 545         bool                ignore_quota = false, synced = false;
 546         long long           space = 0;
 547         ENTRY;
 548
 549         LASSERT(dt_object_exists(dt));
 550         LASSERT(obj->oo_db);
 551
 552         LASSERT(lnb);
 553         LASSERT(npages > 0);
 554
 555         oh = container_of0(th, struct osd_thandle, ot_super);
 556
 557         for (i = 0; i < npages; i++) {
 558                 if (lnb[i].rc)
 559                         /* ENOSPC, network RPC error, etc.
 560                          * We don't want to book space for pages which will be
 561                          * skipped in osd_write_commit(). Hence we skip pages
 562                          * with lnb_rc != 0 here too */
 563                         continue;
 564                 /* ignore quota for the whole request if any page is from
 565                  * client cache or written by root.
 566                  *
 567                  * XXX once we drop the 1.8 client support, the checking
 568                  * for whether page is from cache can be simplified as:
 569                  * !(lnb[i].flags & OBD_BRW_SYNC)
 570                  *
 571                  * XXX we could handle this on per-lnb basis as done by
 572                  * grant. */
 573                 if ((lnb[i].flags & OBD_BRW_NOQUOTA) ||
 574                     (lnb[i].flags & (OBD_BRW_FROM_GRANT | OBD_BRW_SYNC)) ==
 575                     OBD_BRW_FROM_GRANT)
 576                         ignore_quota = true;
 577                 if (size == 0) {
 578                         /* first valid lnb */
 579                         offset = lnb[i].lnb_file_offset;
 580                         size = lnb[i].len;
 581                         continue;
 582                 }
 583                 if (offset + size == lnb[i].lnb_file_offset) {
 584                         /* this lnb is contiguous to the previous one */
 585                         size += lnb[i].len;
 586                         continue;
 587                 }
 588
 589                 dmu_tx_hold_write(oh->ot_tx, obj->oo_db->db_object, offset,size);
 590
 591                 /* estimating space that will be consumed by a write is rather
 592                  * complicated with ZFS. As a consequence, we don't account for
 593                  * indirect blocks and quota overrun will be adjusted once the
 594                  * operation is committed, if required. */
 595                 space += osd_count_not_mapped(obj, offset, size);
 596
 597                 offset = lnb->lnb_file_offset;
 598                 size = lnb->len;
 599         }
 600
 601         if (size) {
 602                 dmu_tx_hold_write(oh->ot_tx, obj->oo_db->db_object, offset,size);
 603                 space += osd_count_not_mapped(obj, offset, size);
 604         }
 605
 606         dmu_tx_hold_sa(oh->ot_tx, obj->oo_sa_hdl, 0);
 607
 608         oh->ot_write_commit = 1; /* used in osd_trans_start() for fail_loc */
 609
 610         /* backend zfs filesystem might be configured to store multiple data
 611          * copies */
 612         space  *= osd->od_objset.os->os_copies;
 613         space   = toqb(space);
 614         CDEBUG(D_QUOTA, "writting %d pages, reserving "LPD64"K of quota "
 615                "space\n", npages, space);
 616
 617 retry:
 618         /* acquire quota space if needed */
 619         rc = osd_declare_quota(env, osd, obj->oo_attr.la_uid,
 620                                obj->oo_attr.la_gid, space, oh, true, &flags,
 621                                ignore_quota);
 622
 623         if (!synced && rc == -EDQUOT && (flags & QUOTA_FL_SYNC) != 0) {
 624                 dt_sync(env, th->th_dev);
 625                 synced = true;
 626                 CDEBUG(D_QUOTA, "retry after sync\n");
 627                 flags = 0;
 628                 goto retry;
 629         }
 630
 631         /* we need only to store the overquota flags in the first lnb for
 632          * now, once we support multiple objects BRW, this code needs be
 633          * revised. */
 634         if (flags & QUOTA_FL_OVER_USRQUOTA)
 635                 lnb[0].flags |= OBD_BRW_OVER_USRQUOTA;
 636         if (flags & QUOTA_FL_OVER_GRPQUOTA)
 637                 lnb[0].flags |= OBD_BRW_OVER_GRPQUOTA;
 638
 639         RETURN(rc);
 640 }
 641
 642 static int osd_write_commit(const struct lu_env *env, struct dt_object *dt,
 643                         struct niobuf_local *lnb, int npages,
 644                         struct thandle *th)
 645 {
 646         struct osd_object  *obj  = osd_dt_obj(dt);
 647         struct osd_device  *osd = osd_obj2dev(obj);
 648         udmu_objset_t      *uos = &osd->od_objset;
 649         struct osd_thandle *oh;
 650         uint64_t            new_size = 0;
 651         int                 i, rc = 0;
 652         ENTRY;
 653
 654         LASSERT(dt_object_exists(dt));
 655         LASSERT(obj->oo_db);
 656
 657         LASSERT(th != NULL);
 658         oh = container_of0(th, struct osd_thandle, ot_super);
 659
 660         for (i = 0; i < npages; i++) {
 661                 CDEBUG(D_INODE, "write %u bytes at %u\n",
 662                         (unsigned) lnb[i].len,
 663                         (unsigned) lnb[i].lnb_file_offset);
 664
 665                 if (lnb[i].rc) {
 666                         /* ENOSPC, network RPC error, etc.
 667                          * Unlike ldiskfs, zfs allocates new blocks on rewrite,
 668                          * so we skip this page if lnb_rc is set to -ENOSPC */
 669                         CDEBUG(D_INODE, "obj "DFID": skipping lnb[%u]: rc=%d\n",
 670                                 PFID(lu_object_fid(&dt->do_lu)), i,
 671                                 lnb[i].rc);
 672                         continue;
 673                 }
 674
 675                 if (lnb[i].page->mapping == (void *)obj) {
 676                         dmu_write(osd->od_objset.os, obj->oo_db->db_object,
 677                                 lnb[i].lnb_file_offset, lnb[i].len,
 678                                 kmap(lnb[i].page), oh->ot_tx);
 679                         kunmap(lnb[i].page);
 680                 } else if (lnb[i].dentry) {
 681                         LASSERT(((unsigned long)lnb[i].dentry & 1) == 0);
 682                         /* buffer loaned for zerocopy, try to use it.
 683                          * notice that dmu_assign_arcbuf() is smart
 684                          * enough to recognize changed blocksize
 685                          * in this case it fallbacks to dmu_write() */
 686                         dmu_assign_arcbuf(obj->oo_db, lnb[i].lnb_file_offset,
 687                                         (void *)lnb[i].dentry, oh->ot_tx);
 688                         /* drop the reference, otherwise osd_put_bufs()
 689                          * will be releasing it - bad! */
 690                         lnb[i].dentry = NULL;
 691                         cfs_atomic_dec(&osd->od_zerocopy_loan);
 692                 }
 693
 694                 if (new_size < lnb[i].lnb_file_offset + lnb[i].len)
 695                         new_size = lnb[i].lnb_file_offset + lnb[i].len;
 696         }
 697
 698         if (unlikely(new_size == 0)) {
 699                 /* no pages to write, no transno is needed */
 700                 th->th_local = 1;
 701                 /* it is important to return 0 even when all lnb_rc == -ENOSPC
 702                  * since ofd_commitrw_write() retries several times on ENOSPC */
 703                 RETURN(0);
 704         }
 705
 706         write_lock(&obj->oo_attr_lock);
 707         if (obj->oo_attr.la_size < new_size) {
 708                 obj->oo_attr.la_size = new_size;
 709                 write_unlock(&obj->oo_attr_lock);
 710                 /* osd_object_sa_update() will be copying directly from
 711                  * oo_attr into dbuf. any update within a single txg will copy
 712                  * the most actual */
 713                 rc = osd_object_sa_update(obj, SA_ZPL_SIZE(uos),
 714                                         &obj->oo_attr.la_size, 8, oh);
 715         } else {
 716                 write_unlock(&obj->oo_attr_lock);
 717         }
 718
 719         RETURN(rc);
 720 }
 721
 722 static int osd_read_prep(const struct lu_env *env, struct dt_object *dt,
 723                         struct niobuf_local *lnb, int npages)
 724 {
 725         struct osd_object *obj  = osd_dt_obj(dt);
 726         struct lu_buf      buf;
 727         loff_t             offset;
 728         int                i;
 729
 730         LASSERT(dt_object_exists(dt));
 731         LASSERT(obj->oo_db);
 732
 733         for (i = 0; i < npages; i++) {
 734                 buf.lb_buf = kmap(lnb[i].page);
 735                 buf.lb_len = lnb[i].len;
 736                 offset = lnb[i].lnb_file_offset;
 737
 738                 CDEBUG(D_OTHER, "read %u bytes at %u\n",
 739                         (unsigned) lnb[i].len,
 740                         (unsigned) lnb[i].lnb_file_offset);
 741                 lnb[i].rc = osd_read(env, dt, &buf, &offset, NULL);
 742                 kunmap(lnb[i].page);
 743
 744                 if (lnb[i].rc < buf.lb_len) {
 745                         /* all subsequent rc should be 0 */
 746                         while (++i < npages)
 747                                 lnb[i].rc = 0;
 748                         break;
 749                 }
 750         }
 751
 752         return 0;
 753 }
 754
 755 /*
 756  * Punch/truncate an object
 757  *
 758  *      IN:     db  - dmu_buf of the object to free data in.
 759  *              off - start of section to free.
 760  *              len - length of section to free (DMU_OBJECT_END => to EOF).
 761  *
 762  *      RETURN: 0 if success
 763  *              error code if failure
 764  *
 765  * The transaction passed to this routine must have
 766  * dmu_tx_hold_sa() and if off < size, dmu_tx_hold_free()
 767  * called and then assigned to a transaction group.
 768  */
 769 static int __osd_object_punch(objset_t *os, dmu_buf_t *db, dmu_tx_t *tx,
 770                                 uint64_t size, uint64_t off, uint64_t len)
 771 {
 772         int rc = 0;
 773
 774         /* Assert that the transaction has been assigned to a
 775            transaction group. */
 776         LASSERT(tx->tx_txg != 0);
 777         /*
 778          * Nothing to do if file already at desired length.
 779          */
 780         if (len == DMU_OBJECT_END && size == off)
 781                 return 0;
 782
 783         if (off < size)
 784                 rc = -dmu_free_range(os, db->db_object, off, len, tx);
 785
 786         return rc;
 787 }
 788
 789 static int osd_punch(const struct lu_env *env, struct dt_object *dt,
 790                         __u64 start, __u64 end, struct thandle *th,
 791                         struct lustre_capa *capa)
 792 {
 793         struct osd_object  *obj = osd_dt_obj(dt);
 794         struct osd_device  *osd = osd_obj2dev(obj);
 795         udmu_objset_t      *uos = &osd->od_objset;
 796         struct osd_thandle *oh;
 797         __u64               len;
 798         int                 rc = 0;
 799         ENTRY;
 800
 801         LASSERT(dt_object_exists(dt));
 802         LASSERT(osd_invariant(obj));
 803
 804         LASSERT(th != NULL);
 805         oh = container_of0(th, struct osd_thandle, ot_super);
 806
 807         write_lock(&obj->oo_attr_lock);
 808         /* truncate */
 809         if (end == OBD_OBJECT_EOF || end >= obj->oo_attr.la_size)
 810                 len = DMU_OBJECT_END;
 811         else
 812                 len = end - start;
 813         write_unlock(&obj->oo_attr_lock);
 814
 815         rc = __osd_object_punch(osd->od_objset.os, obj->oo_db, oh->ot_tx,
 816                                 obj->oo_attr.la_size, start, len);
 817         /* set new size */
 818         if (len == DMU_OBJECT_END) {
 819                 write_lock(&obj->oo_attr_lock);
 820                 obj->oo_attr.la_size = start;
 821                 write_unlock(&obj->oo_attr_lock);
 822                 rc = osd_object_sa_update(obj, SA_ZPL_SIZE(uos),
 823                                         &obj->oo_attr.la_size, 8, oh);
 824         }
 825         RETURN(rc);
 826 }
 827
 828 static int osd_declare_punch(const struct lu_env *env, struct dt_object *dt,
 829                         __u64 start, __u64 end, struct thandle *handle)
 830 {
 831         struct osd_object  *obj = osd_dt_obj(dt);
 832         struct osd_device  *osd = osd_obj2dev(obj);
 833         struct osd_thandle *oh;
 834         __u64               len;
 835         ENTRY;
 836
 837         oh = container_of0(handle, struct osd_thandle, ot_super);
 838
 839         read_lock(&obj->oo_attr_lock);
 840         if (end == OBD_OBJECT_EOF || end >= obj->oo_attr.la_size)
 841                 len = DMU_OBJECT_END;
 842         else
 843                 len = end - start;
 844
 845         /* declare we'll free some blocks ... */
 846         if (start < obj->oo_attr.la_size) {
 847                 read_unlock(&obj->oo_attr_lock);
 848                 dmu_tx_hold_free(oh->ot_tx, obj->oo_db->db_object, start, len);
 849         } else {
 850                 read_unlock(&obj->oo_attr_lock);
 851         }
 852
 853         /* ... and we'll modify size attribute */
 854         dmu_tx_hold_sa(oh->ot_tx, obj->oo_sa_hdl, 0);
 855
 856         RETURN(osd_declare_quota(env, osd, obj->oo_attr.la_uid,
 857                                  obj->oo_attr.la_gid, 0, oh, true, NULL,
 858                                  false));
 859 }
 860
 861
 862 struct dt_body_operations osd_body_ops = {
 863         .dbo_read                       = osd_read,
 864         .dbo_declare_write              = osd_declare_write,
 865         .dbo_write                      = osd_write,
 866         .dbo_bufs_get                   = osd_bufs_get,
 867         .dbo_bufs_put                   = osd_bufs_put,
 868         .dbo_write_prep                 = osd_write_prep,
 869         .dbo_declare_write_commit       = osd_declare_write_commit,
 870         .dbo_write_commit               = osd_write_commit,
 871         .dbo_read_prep                  = osd_read_prep,
 872         .dbo_declare_punch              = osd_declare_punch,
 873         .dbo_punch                      = osd_punch,
 874 };
 875