lustre/mdt/mdt_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.gnu.org/licenses/gpl-2.0.html
  19  *
  20  * GPL HEADER END
  21  */
  22 /*
  23  * Copyright (c) 2017, Intel Corporation.
  24  */
  25 /*
  26  * lustre/mdt/mdt_io.c
  27  *
  28  * Author: Mikhail Pershin <mike.pershin@intel.com>
  29  */
  30
  31 #define DEBUG_SUBSYSTEM S_FILTER
  32
  33 #include <dt_object.h>
  34 #include <linux/falloc.h>
  35 #include <lustre_nodemap.h>
  36
  37 #include "mdt_internal.h"
  38
  39 /* functions below are stubs for now, they will be implemented with
  40  * grant support on MDT */
  41 static inline void mdt_dom_read_lock(struct mdt_object *mo)
  42 {
  43         down_read(&mo->mot_dom_sem);
  44 }
  45
  46 static inline void mdt_dom_read_unlock(struct mdt_object *mo)
  47 {
  48         up_read(&mo->mot_dom_sem);
  49 }
  50
  51 static inline void mdt_dom_write_lock(struct mdt_object *mo)
  52 {
  53         down_write(&mo->mot_dom_sem);
  54 }
  55
  56 static inline void mdt_dom_write_unlock(struct mdt_object *mo)
  57 {
  58         up_write(&mo->mot_dom_sem);
  59 }
  60
  61 static void mdt_dom_resource_prolong(struct ldlm_prolong_args *arg)
  62 {
  63         struct ldlm_resource *res;
  64         struct ldlm_lock *lock;
  65
  66         ENTRY;
  67
  68         res = ldlm_resource_get(arg->lpa_export->exp_obd->obd_namespace,
  69                                 &arg->lpa_resid, LDLM_IBITS, 0);
  70         if (IS_ERR(res)) {
  71                 CDEBUG(D_DLMTRACE,
  72                        "Failed to get resource for resid %llu/%llu\n",
  73                        arg->lpa_resid.name[0], arg->lpa_resid.name[1]);
  74                 RETURN_EXIT;
  75         }
  76
  77         lock_res(res);
  78         list_for_each_entry(lock, &res->lr_granted, l_res_link) {
  79                 if (ldlm_has_dom(lock)) {
  80                         LDLM_DEBUG(lock, "DOM lock to prolong ");
  81                         ldlm_lock_prolong_one(lock, arg);
  82                         /* only one PW or EX lock can be granted,
  83                          * no need to continue search
  84                          */
  85                         if (lock->l_granted_mode & (LCK_PW | LCK_EX))
  86                                 break;
  87                 }
  88         }
  89         unlock_res(res);
  90         ldlm_resource_putref(res);
  91
  92         EXIT;
  93 }
  94
  95 static void mdt_prolong_dom_lock(struct tgt_session_info *tsi,
  96                                  struct ldlm_prolong_args *data)
  97 {
  98         struct obdo *oa = &tsi->tsi_ost_body->oa;
  99         struct ldlm_lock *lock;
 100
 101         ENTRY;
 102
 103         data->lpa_req = tgt_ses_req(tsi);
 104         data->lpa_export = tsi->tsi_exp;
 105         data->lpa_resid = tsi->tsi_resid;
 106
 107         CDEBUG(D_RPCTRACE, "Prolong DOM lock for req %p with x%llu\n",
 108                tgt_ses_req(tsi), tgt_ses_req(tsi)->rq_xid);
 109
 110         if (oa->o_valid & OBD_MD_FLHANDLE) {
 111                 /* mostly a request should be covered by only one lock, try
 112                  * fast path. */
 113                 lock = ldlm_handle2lock(&oa->o_handle);
 114                 if (lock != NULL) {
 115                         LASSERT(lock->l_export == data->lpa_export);
 116                         ldlm_lock_prolong_one(lock, data);
 117                         lock->l_last_used = ktime_get();
 118                         LDLM_LOCK_PUT(lock);
 119                         if (data->lpa_locks_cnt > 0)
 120                                 RETURN_EXIT;
 121                 }
 122         }
 123         mdt_dom_resource_prolong(data);
 124         EXIT;
 125 }
 126
 127 static int mdt_rw_hpreq_lock_match(struct ptlrpc_request *req,
 128                                    struct ldlm_lock *lock)
 129 {
 130         struct obd_ioobj *ioo;
 131         enum ldlm_mode mode;
 132         __u32 opc = lustre_msg_get_opc(req->rq_reqmsg);
 133
 134         ENTRY;
 135
 136         if (!(lock->l_policy_data.l_inodebits.bits & MDS_INODELOCK_DOM))
 137                 RETURN(0);
 138
 139         ioo = req_capsule_client_get(&req->rq_pill, &RMF_OBD_IOOBJ);
 140         LASSERT(ioo != NULL);
 141
 142         LASSERT(lock->l_resource != NULL);
 143         if (!fid_res_name_eq(&ioo->ioo_oid.oi_fid, &lock->l_resource->lr_name))
 144                 RETURN(0);
 145
 146         /* a bulk write can only hold a reference on a PW extent lock. */
 147         mode = LCK_PW | LCK_GROUP;
 148         if (opc == OST_READ)
 149                 /* whereas a bulk read can be protected by either a PR or PW
 150                  * extent lock */
 151                 mode |= LCK_PR;
 152
 153         if (!(lock->l_granted_mode & mode))
 154                 RETURN(0);
 155
 156         RETURN(1);
 157 }
 158
 159 static int mdt_rw_hpreq_check(struct ptlrpc_request *req)
 160 {
 161         struct tgt_session_info *tsi;
 162         struct obd_ioobj *ioo;
 163         struct niobuf_remote *rnb;
 164         int opc;
 165         struct ldlm_prolong_args pa = { 0 };
 166
 167         ENTRY;
 168
 169         /* Don't use tgt_ses_info() to get session info, because lock_match()
 170          * can be called while request has no processing thread yet. */
 171         tsi = lu_context_key_get(&req->rq_session, &tgt_session_key);
 172
 173         /*
 174          * Use LASSERT below because malformed RPCs should have
 175          * been filtered out in tgt_hpreq_handler().
 176          */
 177         opc = lustre_msg_get_opc(req->rq_reqmsg);
 178         LASSERT(opc == OST_READ || opc == OST_WRITE);
 179
 180         ioo = req_capsule_client_get(&req->rq_pill, &RMF_OBD_IOOBJ);
 181         LASSERT(ioo != NULL);
 182
 183         rnb = req_capsule_client_get(&req->rq_pill, &RMF_NIOBUF_REMOTE);
 184         LASSERT(rnb != NULL);
 185         LASSERT(!(rnb->rnb_flags & OBD_BRW_SRVLOCK));
 186
 187         pa.lpa_mode = LCK_PW | LCK_GROUP;
 188         if (opc == OST_READ)
 189                 pa.lpa_mode |= LCK_PR;
 190
 191         DEBUG_REQ(D_RPCTRACE, req, "%s %s: refresh rw locks for " DFID,
 192                   tgt_name(tsi->tsi_tgt), current->comm, PFID(&tsi->tsi_fid));
 193
 194         mdt_prolong_dom_lock(tsi, &pa);
 195
 196         if (pa.lpa_blocks_cnt > 0) {
 197                 CDEBUG(D_DLMTRACE,
 198                        "%s: refreshed %u locks timeout for req %p\n",
 199                        tgt_name(tsi->tsi_tgt), pa.lpa_blocks_cnt, req);
 200                 RETURN(1);
 201         }
 202
 203         RETURN(pa.lpa_locks_cnt > 0 ? 0 : -ESTALE);
 204 }
 205
 206 static void mdt_rw_hpreq_fini(struct ptlrpc_request *req)
 207 {
 208         mdt_rw_hpreq_check(req);
 209 }
 210
 211 static struct ptlrpc_hpreq_ops mdt_hpreq_rw = {
 212         .hpreq_lock_match = mdt_rw_hpreq_lock_match,
 213         .hpreq_check = mdt_rw_hpreq_check,
 214         .hpreq_fini = mdt_rw_hpreq_fini
 215 };
 216
 217 /**
 218  * Assign high priority operations to an IO request.
 219  *
 220  * Check if the incoming request is a candidate for
 221  * high-priority processing. If it is, assign it a high
 222  * priority operations table.
 223  *
 224  * \param[in] tsi       target session environment for this request
 225  */
 226 void mdt_hp_brw(struct tgt_session_info *tsi)
 227 {
 228         struct niobuf_remote    *rnb;
 229         struct obd_ioobj        *ioo;
 230
 231         ENTRY;
 232
 233         ioo = req_capsule_client_get(tsi->tsi_pill, &RMF_OBD_IOOBJ);
 234         LASSERT(ioo != NULL); /* must exist after request preprocessing */
 235         if (ioo->ioo_bufcnt > 0) {
 236                 rnb = req_capsule_client_get(tsi->tsi_pill, &RMF_NIOBUF_REMOTE);
 237                 LASSERT(rnb != NULL); /* must exist after preprocessing */
 238
 239                 /* no high priority if server lock is needed */
 240                 if (rnb->rnb_flags & OBD_BRW_SRVLOCK ||
 241                     (lustre_msg_get_flags(tgt_ses_req(tsi)->rq_reqmsg) &
 242                      MSG_REPLAY))
 243                         return;
 244         }
 245         tgt_ses_req(tsi)->rq_ops = &mdt_hpreq_rw;
 246 }
 247
 248 static int mdt_punch_hpreq_lock_match(struct ptlrpc_request *req,
 249                                       struct ldlm_lock *lock)
 250 {
 251         struct tgt_session_info *tsi;
 252         struct obdo *oa;
 253
 254         ENTRY;
 255
 256         /* Don't use tgt_ses_info() to get session info, because lock_match()
 257          * can be called while request has no processing thread yet. */
 258         tsi = lu_context_key_get(&req->rq_session, &tgt_session_key);
 259
 260         /*
 261          * Use LASSERT below because malformed RPCs should have
 262          * been filtered out in tgt_hpreq_handler().
 263          */
 264         LASSERT(tsi->tsi_ost_body != NULL);
 265         if (tsi->tsi_ost_body->oa.o_valid & OBD_MD_FLHANDLE &&
 266             tsi->tsi_ost_body->oa.o_handle.cookie == lock->l_handle.h_cookie)
 267                 RETURN(1);
 268
 269         oa = &tsi->tsi_ost_body->oa;
 270
 271         LASSERT(lock->l_resource != NULL);
 272         if (!fid_res_name_eq(&oa->o_oi.oi_fid, &lock->l_resource->lr_name))
 273                 RETURN(0);
 274
 275         if (!(lock->l_granted_mode & (LCK_PW | LCK_GROUP)))
 276                 RETURN(0);
 277
 278         RETURN(1);
 279 }
 280
 281 /**
 282  * Implementation of ptlrpc_hpreq_ops::hpreq_lock_check for OST_PUNCH request.
 283  *
 284  * High-priority queue request check for whether the given punch request
 285  * (\a req) is blocking an LDLM lock cancel. Also checks whether the request is
 286  * covered by an LDLM lock.
 287  *
 288
 289  *
 290  * \param[in] req       the incoming request
 291  *
 292  * \retval              1 if \a req is blocking an LDLM lock cancel
 293  * \retval              0 if it is not
 294  * \retval              -ESTALE if lock is not found
 295  */
 296 static int mdt_punch_hpreq_check(struct ptlrpc_request *req)
 297 {
 298         struct tgt_session_info *tsi;
 299         struct obdo *oa;
 300         struct ldlm_prolong_args pa = { 0 };
 301
 302         ENTRY;
 303
 304         /* Don't use tgt_ses_info() to get session info, because lock_match()
 305          * can be called while request has no processing thread yet. */
 306         tsi = lu_context_key_get(&req->rq_session, &tgt_session_key);
 307         LASSERT(tsi != NULL);
 308         oa = &tsi->tsi_ost_body->oa;
 309
 310         LASSERT(!(oa->o_valid & OBD_MD_FLFLAGS &&
 311                   oa->o_flags & OBD_FL_SRVLOCK));
 312
 313         pa.lpa_mode = LCK_PW | LCK_GROUP;
 314
 315         CDEBUG(D_DLMTRACE, "%s: refresh DOM lock for "DFID"\n",
 316                tgt_name(tsi->tsi_tgt), PFID(&tsi->tsi_fid));
 317
 318         mdt_prolong_dom_lock(tsi, &pa);
 319
 320         if (pa.lpa_blocks_cnt > 0) {
 321                 CDEBUG(D_DLMTRACE,
 322                        "%s: refreshed %u locks timeout for req %p.\n",
 323                        tgt_name(tsi->tsi_tgt), pa.lpa_blocks_cnt, req);
 324                 RETURN(1);
 325         }
 326
 327         RETURN(pa.lpa_locks_cnt > 0 ? 0 : -ESTALE);
 328 }
 329
 330 /**
 331  * Implementation of ptlrpc_hpreq_ops::hpreq_lock_fini for OST_PUNCH request.
 332  *
 333  * Called after the request has been handled. It refreshes lock timeout again
 334  * so that client has more time to send lock cancel RPC.
 335  *
 336  * \param[in] req       request which is being processed.
 337  */
 338 static void mdt_punch_hpreq_fini(struct ptlrpc_request *req)
 339 {
 340         mdt_punch_hpreq_check(req);
 341 }
 342
 343 static struct ptlrpc_hpreq_ops mdt_hpreq_punch = {
 344         .hpreq_lock_match = mdt_punch_hpreq_lock_match,
 345         .hpreq_check = mdt_punch_hpreq_check,
 346         .hpreq_fini = mdt_punch_hpreq_fini
 347 };
 348
 349 void mdt_hp_punch(struct tgt_session_info *tsi)
 350 {
 351         LASSERT(tsi->tsi_ost_body != NULL); /* must exists if we are here */
 352         /* no high-priority if server lock is needed */
 353         if ((tsi->tsi_ost_body->oa.o_valid & OBD_MD_FLFLAGS &&
 354              tsi->tsi_ost_body->oa.o_flags & OBD_FL_SRVLOCK) ||
 355             tgt_conn_flags(tsi) & OBD_CONNECT_MDS ||
 356             lustre_msg_get_flags(tgt_ses_req(tsi)->rq_reqmsg) & MSG_REPLAY)
 357                 return;
 358         tgt_ses_req(tsi)->rq_ops = &mdt_hpreq_punch;
 359 }
 360
 361 static int mdt_preprw_read(const struct lu_env *env, struct obd_export *exp,
 362                            struct mdt_device *mdt, struct mdt_object *mo,
 363                            struct lu_attr *la, int niocount,
 364                            struct niobuf_remote *rnb, int *nr_local,
 365                            struct niobuf_local *lnb)
 366 {
 367         struct dt_object *dob;
 368         int i, j, rc, tot_bytes = 0;
 369         int maxlnb = *nr_local;
 370         int level;
 371
 372         ENTRY;
 373
 374         mdt_dom_read_lock(mo);
 375         *nr_local = 0;
 376         /* the only valid case when READ can find object is missing or stale
 377          * when export is just evicted and open files are closed forcefully
 378          * on server while client's READ can be in progress.
 379          * This should not happen on healthy export, object can't be missing
 380          * or dying because both states means it was finally destroyed.
 381          */
 382         level = exp->exp_failed ? D_INFO : D_ERROR;
 383         if (!mdt_object_exists(mo)) {
 384                 CDEBUG_LIMIT(level,
 385                              "%s: READ IO to missing obj "DFID": rc = %d\n",
 386                              exp->exp_obd->obd_name, PFID(mdt_object_fid(mo)),
 387                              -ENOENT);
 388                 /* return 0 and continue with empty commit to skip such READ
 389                  * without more BRW errors.
 390                  */
 391                 RETURN(0);
 392         }
 393         if (lu_object_is_dying(&mo->mot_header)) {
 394                 CDEBUG_LIMIT(level,
 395                              "%s: READ IO to stale obj "DFID": rc = %d\n",
 396                              exp->exp_obd->obd_name, PFID(mdt_object_fid(mo)),
 397                              -ESTALE);
 398                 /* return 0 and continue with empty commit to skip such READ
 399                  * without more BRW errors.
 400                  */
 401                 RETURN(0);
 402         }
 403
 404         dob = mdt_obj2dt(mo);
 405         /* parse remote buffers to local buffers and prepare the latter */
 406         for (i = 0, j = 0; i < niocount; i++) {
 407                 rc = dt_bufs_get(env, dob, rnb + i, lnb + j, maxlnb, 0);
 408                 if (unlikely(rc < 0))
 409                         GOTO(buf_put, rc);
 410                 /* correct index for local buffers to continue with */
 411                 j += rc;
 412                 maxlnb -= rc;
 413                 *nr_local += rc;
 414                 tot_bytes += rnb[i].rnb_len;
 415         }
 416
 417         rc = dt_attr_get(env, dob, la);
 418         if (unlikely(rc))
 419                 GOTO(buf_put, rc);
 420
 421         rc = dt_read_prep(env, dob, lnb, *nr_local);
 422         if (unlikely(rc))
 423                 GOTO(buf_put, rc);
 424
 425         RETURN(0);
 426 buf_put:
 427         dt_bufs_put(env, dob, lnb, *nr_local);
 428         mdt_dom_read_unlock(mo);
 429         return rc;
 430 }
 431
 432 static int mdt_preprw_write(const struct lu_env *env, struct obd_export *exp,
 433                             struct mdt_device *mdt, struct mdt_object *mo,
 434                             struct lu_attr *la, struct obdo *oa,
 435                             int objcount, struct obd_ioobj *obj,
 436                             struct niobuf_remote *rnb, int *nr_local,
 437                             struct niobuf_local *lnb)
 438 {
 439         struct dt_object *dob;
 440         int i, j, k, rc = 0, tot_bytes = 0;
 441         int maxlnb = *nr_local;
 442
 443         ENTRY;
 444
 445         /* Process incoming grant info, set OBD_BRW_GRANTED flag and grant some
 446          * space back if possible */
 447         tgt_grant_prepare_write(env, exp, oa, rnb, obj->ioo_bufcnt);
 448
 449         mdt_dom_read_lock(mo);
 450         *nr_local = 0;
 451         /* don't report error in cases with failed export */
 452         if (!mdt_object_exists(mo)) {
 453                 int level = exp->exp_failed ? D_INFO : D_ERROR;
 454
 455                 rc = -ENOENT;
 456                 CDEBUG_LIMIT(level,
 457                              "%s: WRITE IO to missing obj "DFID": rc = %d\n",
 458                              exp->exp_obd->obd_name, PFID(mdt_object_fid(mo)),
 459                              rc);
 460                 /* exit with no data written, note nr_local = 0 above */
 461                 GOTO(unlock, rc);
 462         }
 463         if (lu_object_is_dying(&mo->mot_header)) {
 464                 /* This is possible race between object destroy followed by
 465                  * discard BL AST and client cache flushing. Object is
 466                  * referenced until discard finish.
 467                  */
 468                 CDEBUG(D_INODE, "WRITE IO to stale object "DFID"\n",
 469                        PFID(mdt_object_fid(mo)));
 470                 /* Note: continue with no error here to don't cause BRW errors
 471                  * but skip transaction in commitrw silently so no data is
 472                  * written.
 473                  */
 474         }
 475
 476         dob = mdt_obj2dt(mo);
 477         /* parse remote buffers to local buffers and prepare the latter */
 478         for (i = 0, j = 0; i < obj->ioo_bufcnt; i++) {
 479                 rc = dt_bufs_get(env, dob, rnb + i, lnb + j, maxlnb, 1);
 480                 if (unlikely(rc < 0))
 481                         GOTO(err, rc);
 482                 /* correct index for local buffers to continue with */
 483                 for (k = 0; k < rc; k++) {
 484                         lnb[j + k].lnb_flags = rnb[i].rnb_flags;
 485                         if (!(rnb[i].rnb_flags & OBD_BRW_GRANTED))
 486                                 lnb[j + k].lnb_rc = -ENOSPC;
 487                 }
 488                 j += rc;
 489                 maxlnb -= rc;
 490                 *nr_local += rc;
 491                 tot_bytes += rnb[i].rnb_len;
 492         }
 493
 494         rc = dt_write_prep(env, dob, lnb, *nr_local);
 495         if (likely(rc))
 496                 GOTO(err, rc);
 497
 498         RETURN(0);
 499 err:
 500         dt_bufs_put(env, dob, lnb, *nr_local);
 501 unlock:
 502         mdt_dom_read_unlock(mo);
 503         /* tgt_grant_prepare_write() was called, so we must commit */
 504         tgt_grant_commit(exp, oa->o_grant_used, rc);
 505         /* let's still process incoming grant information packed in the oa,
 506          * but without enforcing grant since we won't proceed with the write.
 507          * Just like a read request actually. */
 508         tgt_grant_prepare_read(env, exp, oa);
 509         return rc;
 510 }
 511
 512 int mdt_obd_preprw(const struct lu_env *env, int cmd, struct obd_export *exp,
 513                    struct obdo *oa, int objcount, struct obd_ioobj *obj,
 514                    struct niobuf_remote *rnb, int *nr_local,
 515                    struct niobuf_local *lnb)
 516 {
 517         struct tgt_session_info *tsi = tgt_ses_info(env);
 518         struct mdt_thread_info *info = tsi2mdt_info(tsi);
 519         struct lu_attr *la = &info->mti_attr.ma_attr;
 520         struct mdt_device *mdt = mdt_dev(exp->exp_obd->obd_lu_dev);
 521         struct mdt_object *mo;
 522         int rc = 0;
 523
 524         /* The default value PTLRPC_MAX_BRW_PAGES is set in tgt_brw_write()
 525          * but for MDT it is different, correct it here. */
 526         if (*nr_local > MD_MAX_BRW_PAGES)
 527                 *nr_local = MD_MAX_BRW_PAGES;
 528
 529         if (!oa || objcount != 1 || obj->ioo_bufcnt == 0) {
 530                 CERROR("%s: bad parameters %p/%i/%i\n",
 531                        exp->exp_obd->obd_name, oa, objcount, obj->ioo_bufcnt);
 532                 rc = -EPROTO;
 533         }
 534
 535         mo = mdt_object_find(env, mdt, &tsi->tsi_fid);
 536         if (IS_ERR(mo))
 537                 GOTO(out, rc = PTR_ERR(mo));
 538
 539         LASSERT(info->mti_object == NULL);
 540         info->mti_object = mo;
 541
 542         if (cmd == OBD_BRW_WRITE) {
 543                 la_from_obdo(la, oa, OBD_MD_FLGETATTR);
 544                 rc = mdt_preprw_write(env, exp, mdt, mo, la, oa,
 545                                       objcount, obj, rnb, nr_local, lnb);
 546         } else if (cmd == OBD_BRW_READ) {
 547                 tgt_grant_prepare_read(env, exp, oa);
 548                 rc = mdt_preprw_read(env, exp, mdt, mo, la,
 549                                      obj->ioo_bufcnt, rnb, nr_local, lnb);
 550                 obdo_from_la(oa, la, LA_ATIME);
 551         } else {
 552                 CERROR("%s: wrong cmd %d received!\n",
 553                        exp->exp_obd->obd_name, cmd);
 554                 rc = -EPROTO;
 555         }
 556         if (rc) {
 557                 lu_object_put(env, &mo->mot_obj);
 558                 info->mti_object = NULL;
 559         }
 560 out:
 561         RETURN(rc);
 562 }
 563
 564 static int mdt_commitrw_read(const struct lu_env *env, struct mdt_device *mdt,
 565                              struct mdt_object *mo, int objcount, int niocount,
 566                              struct niobuf_local *lnb)
 567 {
 568         struct dt_object *dob;
 569         int rc = 0;
 570
 571         ENTRY;
 572
 573         dob = mdt_obj2dt(mo);
 574
 575         if (niocount)
 576                 dt_bufs_put(env, dob, lnb, niocount);
 577
 578         mdt_dom_read_unlock(mo);
 579         RETURN(rc);
 580 }
 581
 582 static int mdt_commitrw_write(const struct lu_env *env, struct obd_export *exp,
 583                               struct mdt_device *mdt, struct mdt_object *mo,
 584                               struct lu_attr *la, struct obdo *oa, int objcount,
 585                               int niocount, struct niobuf_local *lnb,
 586                               unsigned long granted, int old_rc)
 587 {
 588         struct dt_device *dt = mdt->mdt_bottom;
 589         struct dt_object *dob;
 590         struct thandle *th;
 591         int rc = 0;
 592         int retries = 0;
 593         int i, restart = 0;
 594
 595         ENTRY;
 596
 597         dob = mdt_obj2dt(mo);
 598
 599         if (old_rc)
 600                 GOTO(out, rc = old_rc);
 601
 602         la->la_valid &= LA_ATIME | LA_MTIME | LA_CTIME;
 603 retry:
 604         if (!dt_object_exists(dob))
 605                 GOTO(out, rc = -ENOENT);
 606
 607         if (niocount == 0) {
 608                 rc = -EPROTO;
 609                 DEBUG_REQ(D_WARNING, tgt_ses_req(tgt_ses_info(env)),
 610                           "%s: commit with no pages for "DFID": rc = %d\n",
 611                           exp->exp_obd->obd_name, PFID(mdt_object_fid(mo)), rc);
 612                 GOTO(out, rc);
 613         }
 614
 615         CFS_FAIL_TIMEOUT(OBD_FAIL_MDS_COMMITRW_DELAY, cfs_fail_val);
 616
 617         th = dt_trans_create(env, dt);
 618         if (IS_ERR(th))
 619                 GOTO(out, rc = PTR_ERR(th));
 620
 621         for (i = 0; i < niocount; i++) {
 622                 if (!(lnb[i].lnb_flags & OBD_BRW_ASYNC)) {
 623                         th->th_sync = 1;
 624                         break;
 625                 }
 626         }
 627
 628         if (CFS_FAIL_CHECK(OBD_FAIL_OST_DQACQ_NET))
 629                 GOTO(out_stop, rc = -EINPROGRESS);
 630
 631         rc = dt_declare_write_commit(env, dob, lnb, niocount, th);
 632         if (rc)
 633                 GOTO(out_stop, rc);
 634
 635         if (la->la_valid) {
 636                 /* update [mac]time if needed */
 637                 rc = dt_declare_attr_set(env, dob, la, th);
 638                 if (rc)
 639                         GOTO(out_stop, rc);
 640         }
 641
 642         tgt_vbr_obj_set(env, dob);
 643         rc = dt_trans_start(env, dt, th);
 644         if (rc)
 645                 GOTO(out_stop, rc);
 646
 647         dt_write_lock(env, dob, 0);
 648         if (lu_object_is_dying(&mo->mot_header)) {
 649                 /* Commit to stale object can be just skipped silently. */
 650                 CDEBUG(D_INODE, "skip commit to stale object "DFID"\n",
 651                         PFID(mdt_object_fid(mo)));
 652                 GOTO(unlock, rc = 0);
 653         }
 654         rc = dt_write_commit(env, dob, lnb, niocount, th, oa->o_size);
 655         if (rc) {
 656                 restart = th->th_restart_tran;
 657                 GOTO(unlock, rc);
 658         }
 659
 660         if (la->la_valid) {
 661                 rc = dt_attr_set(env, dob, la, th);
 662                 if (rc)
 663                         GOTO(unlock, rc);
 664         }
 665         /* get attr to return */
 666         rc = dt_attr_get(env, dob, la);
 667 unlock:
 668         dt_write_unlock(env, dob);
 669
 670 out_stop:
 671         /* Force commit to make the just-deleted blocks
 672          * reusable. LU-456 */
 673         if (rc == -ENOSPC)
 674                 th->th_sync = 1;
 675
 676
 677         if (rc == 0 && granted > 0) {
 678                 if (tgt_grant_commit_cb_add(th, exp, granted) == 0)
 679                         granted = 0;
 680         }
 681
 682         th->th_result = restart ? 0 : rc;
 683         dt_trans_stop(env, dt, th);
 684         if (rc == -ENOSPC && retries++ < 3) {
 685                 CDEBUG(D_INODE, "retry after force commit, retries:%d\n",
 686                        retries);
 687                 goto retry;
 688         }
 689         if (restart) {
 690                 retries++;
 691                 restart = 0;
 692                 if (retries % 10000 == 0)
 693                         CERROR("%s: restart IO write too many times: %d\n",
 694                                exp->exp_obd->obd_name, retries);
 695                 CDEBUG(D_INODE, "retry transaction, retries:%d\n",
 696                        retries);
 697                 goto retry;
 698         }
 699
 700 out:
 701         dt_bufs_put(env, dob, lnb, niocount);
 702         mdt_dom_read_unlock(mo);
 703         if (granted > 0)
 704                 tgt_grant_commit(exp, granted, old_rc);
 705         RETURN(rc);
 706 }
 707
 708 void mdt_dom_obj_lvb_update(const struct lu_env *env, struct mdt_object *mo,
 709                             struct obdo *oa, bool increase_only)
 710 {
 711         struct mdt_device *mdt = mdt_dev(mo->mot_obj.lo_dev);
 712         struct ldlm_res_id resid;
 713         struct ldlm_resource *res;
 714
 715         fid_build_reg_res_name(mdt_object_fid(mo), &resid);
 716         res = ldlm_resource_get(mdt->mdt_namespace, &resid, LDLM_IBITS, 1);
 717         if (IS_ERR(res))
 718                 return;
 719
 720         /* Update lvbo data if exists. */
 721         if (mdt_dom_lvb_is_valid(res)) {
 722                 mdt_dom_disk_lvbo_update(env, mo, res, increase_only);
 723                 if (oa) {
 724                         struct ost_lvb *res_lvb = res->lr_lvb_data;
 725
 726                         lock_res(res);
 727                         oa->o_valid |= OBD_MD_FLBLOCKS | OBD_MD_FLSIZE |
 728                                        OBD_MD_FLMTIME | OBD_MD_FLATIME |
 729                                        OBD_MD_FLCTIME;
 730                         oa->o_blocks = res_lvb->lvb_blocks;
 731                         oa->o_size = res_lvb->lvb_size;
 732                         oa->o_atime = res_lvb->lvb_atime;
 733                         oa->o_mtime = res_lvb->lvb_mtime;
 734                         oa->o_ctime = res_lvb->lvb_ctime;
 735                         unlock_res(res);
 736                 }
 737         }
 738         ldlm_resource_putref(res);
 739 }
 740
 741 int mdt_obd_commitrw(const struct lu_env *env, int cmd, struct obd_export *exp,
 742                      struct obdo *oa, int objcount, struct obd_ioobj *obj,
 743                      struct niobuf_remote *rnb, int npages,
 744                      struct niobuf_local *lnb, int old_rc, int nob,
 745                      ktime_t kstart)
 746 {
 747         struct tgt_session_info *tsi = tgt_ses_info(env);
 748         struct ptlrpc_request *req = tgt_ses_req(tsi);
 749         struct mdt_thread_info *info = mdt_th_info(env);
 750         struct mdt_device *mdt = mdt_dev(exp->exp_obd->obd_lu_dev);
 751         struct mdt_object *mo = info->mti_object;
 752         struct lu_attr *la = &info->mti_attr.ma_attr;
 753         __u64 valid;
 754         int rc = 0;
 755         int root_squash = 0;
 756
 757         LASSERT(mo);
 758
 759         if (cmd == OBD_BRW_WRITE) {
 760                 struct lu_nodemap *nodemap;
 761                 __u32 mapped_uid, mapped_gid, mapped_projid;
 762
 763                 mapped_uid = oa->o_uid;
 764                 mapped_gid = oa->o_gid;
 765                 mapped_projid = oa->o_projid;
 766                 nodemap = nodemap_get_from_exp(exp);
 767                 if (!IS_ERR(nodemap)) {
 768                         mapped_uid = nodemap_map_id(nodemap, NODEMAP_UID,
 769                                                     NODEMAP_FS_TO_CLIENT,
 770                                                     oa->o_uid);
 771                         mapped_gid = nodemap_map_id(nodemap, NODEMAP_GID,
 772                                                     NODEMAP_FS_TO_CLIENT,
 773                                                     oa->o_gid);
 774                         mapped_projid = nodemap_map_id(nodemap, NODEMAP_PROJID,
 775                                                        NODEMAP_FS_TO_CLIENT,
 776                                                        oa->o_projid);
 777                 } else if (old_rc == 0) {
 778                         old_rc = PTR_ERR(nodemap);
 779                 }
 780
 781                 if (!IS_ERR_OR_NULL(nodemap)) {
 782                         /* do not bypass quota enforcement if squashed uid */
 783                         if (unlikely(mapped_uid == nodemap->nm_squash_uid)) {
 784                                 int idx;
 785
 786                                 for (idx = 0; idx < npages; idx++)
 787                                         lnb[idx].lnb_flags &=
 788                                                 ~OBD_BRW_SYS_RESOURCE;
 789                                 root_squash = 1;
 790                         }
 791                         nodemap_putref(nodemap);
 792                 }
 793                 /* Don't update timestamps if this write is older than a
 794                  * setattr which modifies the timestamps. b=10150 */
 795
 796                 /* XXX when we start having persistent reservations this needs
 797                  * to be changed to ofd_fmd_get() to create the fmd if it
 798                  * doesn't already exist so we can store the reservation handle
 799                  * there. */
 800                 valid = OBD_MD_FLUID | OBD_MD_FLGID;
 801                 if (tgt_fmd_check(exp, mdt_object_fid(mo),
 802                                   mdt_info_req(info)->rq_xid))
 803                         valid |= OBD_MD_FLATIME | OBD_MD_FLMTIME |
 804                                  OBD_MD_FLCTIME;
 805
 806                 la_from_obdo(la, oa, valid);
 807
 808                 mdt_counter_incr(req, LPROC_MDT_IO_WRITE_BYTES, nob);
 809                 mdt_counter_incr(req, LPROC_MDT_IO_WRITE,
 810                                  ktime_us_delta(ktime_get(), kstart));
 811
 812                 rc = mdt_commitrw_write(env, exp, mdt, mo, la, oa, objcount,
 813                                         npages, lnb, oa->o_grant_used, old_rc);
 814                 if (rc == 0)
 815                         obdo_from_la(oa, la, VALID_FLAGS | LA_GID | LA_UID);
 816                 else
 817                         obdo_from_la(oa, la, LA_GID | LA_UID);
 818
 819                 mdt_dom_obj_lvb_update(env, mo, NULL, false);
 820                 /* don't report overquota flag if we failed before reaching
 821                  * commit */
 822                 if (old_rc == 0 && (rc == 0 || rc == -EDQUOT)) {
 823                         /* return the overquota flags to client */
 824                         if (lnb[0].lnb_flags & OBD_BRW_OVER_USRQUOTA) {
 825                                 if (oa->o_valid & OBD_MD_FLFLAGS)
 826                                         oa->o_flags |= OBD_FL_NO_USRQUOTA;
 827                                 else
 828                                         oa->o_flags = OBD_FL_NO_USRQUOTA;
 829                         }
 830
 831                         if (lnb[0].lnb_flags & OBD_BRW_OVER_GRPQUOTA) {
 832                                 if (oa->o_valid & OBD_MD_FLFLAGS)
 833                                         oa->o_flags |= OBD_FL_NO_GRPQUOTA;
 834                                 else
 835                                         oa->o_flags = OBD_FL_NO_GRPQUOTA;
 836                         }
 837
 838                         if (lnb[0].lnb_flags & OBD_BRW_OVER_PRJQUOTA) {
 839                                 if (oa->o_valid & OBD_MD_FLFLAGS)
 840                                         oa->o_flags |= OBD_FL_NO_PRJQUOTA;
 841                                 else
 842                                         oa->o_flags = OBD_FL_NO_PRJQUOTA;
 843                         }
 844
 845                         if (lnb[0].lnb_flags & OBD_BRW_ROOT_PRJQUOTA)
 846                                 oa->o_flags |= OBD_FL_ROOT_PRJQUOTA;
 847
 848                         if (root_squash)
 849                                 oa->o_flags |= OBD_FL_ROOT_SQUASH;
 850
 851                         oa->o_valid |= OBD_MD_FLFLAGS | OBD_MD_FLUSRQUOTA |
 852                                        OBD_MD_FLGRPQUOTA | OBD_MD_FLPRJQUOTA;
 853                 }
 854                 /* Convert back to client IDs. LU-9671.
 855                  * nodemap_get_from_exp() may fail due to nodemap deactivated,
 856                  * server ID will be returned back to client in that case.
 857                  */
 858                 oa->o_uid = mapped_uid;
 859                 oa->o_gid = mapped_gid;
 860                 oa->o_projid = mapped_projid;
 861         } else if (cmd == OBD_BRW_READ) {
 862                 /* If oa != NULL then mdt_preprw_read updated the inode
 863                  * atime and we should update the lvb so that other glimpses
 864                  * will also get the updated value. bug 5972 */
 865                 if (oa)
 866                         mdt_dom_obj_lvb_update(env, mo, NULL, true);
 867
 868                 mdt_counter_incr(req, LPROC_MDT_IO_READ_BYTES, nob);
 869                 mdt_counter_incr(req, LPROC_MDT_IO_READ,
 870                                  ktime_us_delta(ktime_get(), kstart));
 871
 872                 rc = mdt_commitrw_read(env, mdt, mo, objcount, npages, lnb);
 873                 if (old_rc)
 874                         rc = old_rc;
 875         } else {
 876                 rc = -EPROTO;
 877         }
 878         mdt_thread_info_fini(info);
 879         RETURN(rc);
 880 }
 881
 882 int mdt_object_fallocate(const struct lu_env *env, struct dt_device *dt,
 883                          struct dt_object *dob, __u64 start, __u64 end,
 884                          int mode, struct lu_attr *la)
 885 {
 886         struct thandle *th;
 887         int rc;
 888
 889         ENTRY;
 890
 891         if (!dt_object_exists(dob))
 892                 RETURN(-ENOENT);
 893
 894         th = dt_trans_create(env, dt);
 895         if (IS_ERR(th))
 896                 RETURN(PTR_ERR(th));
 897
 898         rc = dt_declare_attr_set(env, dob, la, th);
 899         if (rc)
 900                 GOTO(stop, rc);
 901
 902         rc = dt_declare_fallocate(env, dob, start, end, mode, th);
 903         if (rc)
 904                 GOTO(stop, rc);
 905
 906         tgt_vbr_obj_set(env, dob);
 907         rc = dt_trans_start(env, dt, th);
 908         if (rc)
 909                 GOTO(stop, rc);
 910
 911         dt_write_lock(env, dob, 0);
 912         rc = dt_falloc(env, dob, start, end, mode, th);
 913         if (rc)
 914                 GOTO(unlock, rc);
 915         rc = dt_attr_set(env, dob, la, th);
 916         if (rc)
 917                 GOTO(unlock, rc);
 918 unlock:
 919         dt_write_unlock(env, dob);
 920 stop:
 921         th->th_result = rc;
 922         dt_trans_stop(env, dt, th);
 923         RETURN(rc);
 924 }
 925
 926 /**
 927  * MDT request handler for OST_FALLOCATE RPC.
 928  *
 929  * This is part of request processing. Validate request fields,
 930  * preallocate the given MDT object and pack reply.
 931  *
 932  * \param[in] tsi       target session environment for this request
 933  *
 934  * \retval              0 if successful
 935  * \retval              negative value on error
 936  */
 937 int mdt_fallocate_hdl(struct tgt_session_info *tsi)
 938 {
 939         struct obdo *oa = &tsi->tsi_ost_body->oa;
 940         struct ptlrpc_request *req = tgt_ses_req(tsi);
 941         struct ost_body *repbody;
 942         struct mdt_thread_info *info;
 943         struct ldlm_namespace *ns = tsi->tsi_tgt->lut_obd->obd_namespace;
 944         struct obd_export *exp = tsi->tsi_exp;
 945         struct mdt_device *mdt = mdt_dev(exp->exp_obd->obd_lu_dev);
 946         struct mdt_object *mo;
 947         struct dt_object *dob;
 948         struct lu_attr *la;
 949         __u64 flags = 0;
 950         struct lustre_handle lh = { 0, };
 951         int rc, mode;
 952         __u64 start, end;
 953         bool srvlock;
 954         ktime_t kstart = ktime_get();
 955
 956         repbody = req_capsule_server_get(tsi->tsi_pill, &RMF_OST_BODY);
 957         if (repbody == NULL)
 958                 RETURN(err_serious(-ENOMEM));
 959
 960         /*
 961          * fallocate() start and end are passed in o_size and o_blocks
 962          * on the wire.  Clients 2.15.0 and newer should always set
 963          * the OBD_MD_FLSIZE and OBD_MD_FLBLOCKS valid flags, but some
 964          * older client versions did not.  We permit older clients to
 965          * not set these flags, checking their version by proxy using
 966          * the lack of OBD_CONNECT_TRUNCLOCK to imply 2.14.0 and older.
 967          *
 968          * Return -EOPNOTSUPP to also work with older clients not
 969          * supporting newer server modes.
 970          *
 971          * fallocate start and end are passed in o_size, o_blocks
 972          * on the wire.
 973          */
 974         if ((oa->o_valid & (OBD_MD_FLSIZE | OBD_MD_FLBLOCKS)) !=
 975             (OBD_MD_FLSIZE | OBD_MD_FLBLOCKS)
 976 #if LUSTRE_VERSION_CODE < OBD_OCD_VERSION(2, 21, 53, 0)
 977             && !tsi->tsi_exp->exp_old_falloc
 978 #endif
 979             )
 980                 RETURN(-EOPNOTSUPP);
 981
 982         start = oa->o_size;
 983         end = oa->o_blocks;
 984         CDEBUG(D_INFO, "%s: start: %llu end: %llu\n",
 985                mdt->mdt_child_exp->exp_obd->obd_name, start, end);
 986
 987 #if LUSTRE_VERSION_CODE < OBD_OCD_VERSION(2, 21, 53, 0)
 988         /*
 989          * For inter-op case with older clients (where exp_old_falloc is true)
 990          * fallocate() start and end are passed in as 0 (For interior case
 991          * where end offset less than file size) This is fixed later.
 992          * For such cases we return -EOPNOTSUPP
 993          */
 994         if (tsi->tsi_exp->exp_old_falloc && start >= end)
 995                 RETURN(-EOPNOTSUPP);
 996 #endif
 997         /* client should already limit len >= 0 */
 998         if (start >= end)
 999                 RETURN(-EINVAL);
1000
1001         mode = oa->o_falloc_mode;
1002
1003         CDEBUG(D_INODE,
1004                "fallocate: "DFID", mode = %#x, start = %lld, end = %lld\n",
1005                PFID(&tsi->tsi_fid), mode, start, end);
1006
1007         /*
1008          * mode == 0 (which is standard prealloc) and PUNCH is supported
1009          * Rest of mode options are not supported yet.
1010          */
1011         if (mode & ~(FALLOC_FL_KEEP_SIZE | FALLOC_FL_PUNCH_HOLE))
1012                 RETURN(-EOPNOTSUPP);
1013
1014         if (mode & FALLOC_FL_PUNCH_HOLE && !(mode & FALLOC_FL_KEEP_SIZE)) {
1015                 CWARN("%s: PUNCH mode misses KEEP_SIZE flag, setting it\n",
1016                       tsi->tsi_tgt->lut_obd->obd_name);
1017                 mode |= FALLOC_FL_KEEP_SIZE;
1018         }
1019
1020         info = tsi2mdt_info(tsi);
1021         la = &info->mti_attr.ma_attr;
1022
1023         repbody->oa.o_oi = oa->o_oi;
1024         repbody->oa.o_valid = OBD_MD_FLID;
1025
1026         srvlock = oa->o_valid & OBD_MD_FLFLAGS &&
1027                   oa->o_flags & OBD_FL_SRVLOCK;
1028
1029         if (srvlock) {
1030                 rc = tgt_mdt_data_lock(ns, &tsi->tsi_resid, &lh, LCK_PW,
1031                                        &flags);
1032                 if (rc != 0)
1033                         GOTO(out, rc);
1034         }
1035
1036         mo = mdt_object_find(tsi->tsi_env, mdt, &tsi->tsi_fid);
1037         if (IS_ERR(mo))
1038                 GOTO(out_unlock, rc = PTR_ERR(mo));
1039
1040         if (!mdt_object_exists(mo))
1041                 GOTO(out_put, rc = -ENOENT);
1042
1043         /* Shouldn't happen on dirs */
1044         if (S_ISDIR(lu_object_attr(&mo->mot_obj))) {
1045                 rc = -EPERM;
1046                 CERROR("%s: fallocate on dir "DFID": rc = %d\n",
1047                        exp->exp_obd->obd_name, PFID(&tsi->tsi_fid), rc);
1048                 GOTO(out_put, rc);
1049         }
1050
1051         la_from_obdo(la, oa, OBD_MD_FLMTIME | OBD_MD_FLATIME | OBD_MD_FLCTIME);
1052
1053         mdt_dom_write_lock(mo);
1054         dob = mdt_obj2dt(mo);
1055
1056         if (la->la_valid & (LA_ATIME | LA_MTIME | LA_CTIME))
1057                 tgt_fmd_update(tsi->tsi_exp, &tsi->tsi_fid,
1058                                tgt_ses_req(tsi)->rq_xid);
1059
1060         rc = mdt_object_fallocate(tsi->tsi_env, mdt->mdt_bottom, dob, start,
1061                                   end, mode, la);
1062         mdt_dom_write_unlock(mo);
1063         if (rc)
1064                 GOTO(out_put, rc);
1065
1066         mdt_dom_obj_lvb_update(tsi->tsi_env, mo, &repbody->oa, false);
1067
1068         mdt_counter_incr(req, LPROC_MDT_FALLOCATE,
1069                          ktime_us_delta(ktime_get(), kstart));
1070
1071         EXIT;
1072 out_put:
1073         lu_object_put(tsi->tsi_env, &mo->mot_obj);
1074 out_unlock:
1075         if (srvlock)
1076                 tgt_data_unlock(&lh, LCK_PW);
1077 out:
1078         mdt_thread_info_fini(info);
1079         return rc;
1080 }
1081
1082 int mdt_object_punch(const struct lu_env *env, struct dt_device *dt,
1083                      struct dt_object *dob, __u64 start, __u64 end,
1084                      struct lu_attr *la)
1085 {
1086         struct thandle *th;
1087         int rc;
1088
1089         ENTRY;
1090
1091         /* we support truncate, not punch yet */
1092         LASSERT(end == OBD_OBJECT_EOF);
1093
1094         if (!dt_object_exists(dob))
1095                 RETURN(-ENOENT);
1096
1097         th = dt_trans_create(env, dt);
1098         if (IS_ERR(th))
1099                 RETURN(PTR_ERR(th));
1100
1101         rc = dt_declare_attr_set(env, dob, la, th);
1102         if (rc)
1103                 GOTO(stop, rc);
1104
1105         rc = dt_declare_punch(env, dob, start, OBD_OBJECT_EOF, th);
1106         if (rc)
1107                 GOTO(stop, rc);
1108
1109         tgt_vbr_obj_set(env, dob);
1110         rc = dt_trans_start(env, dt, th);
1111         if (rc)
1112                 GOTO(stop, rc);
1113
1114         dt_write_lock(env, dob, 0);
1115         rc = dt_punch(env, dob, start, OBD_OBJECT_EOF, th);
1116         if (rc)
1117                 GOTO(unlock, rc);
1118         rc = dt_attr_set(env, dob, la, th);
1119         if (rc)
1120                 GOTO(unlock, rc);
1121 unlock:
1122         dt_write_unlock(env, dob);
1123 stop:
1124         th->th_result = rc;
1125         dt_trans_stop(env, dt, th);
1126         RETURN(rc);
1127 }
1128
1129 int mdt_punch_hdl(struct tgt_session_info *tsi)
1130 {
1131         const struct obdo *oa = &tsi->tsi_ost_body->oa;
1132         struct ptlrpc_request *req = tgt_ses_req(tsi);
1133         struct ost_body *repbody;
1134         struct mdt_thread_info *info;
1135         struct lu_attr *la;
1136         struct ldlm_namespace *ns = tsi->tsi_tgt->lut_obd->obd_namespace;
1137         struct obd_export *exp = tsi->tsi_exp;
1138         struct mdt_device *mdt = mdt_dev(exp->exp_obd->obd_lu_dev);
1139         struct mdt_object *mo;
1140         struct dt_object *dob;
1141         __u64 flags = 0;
1142         struct lustre_handle lh = { 0, };
1143         ktime_t kstart = ktime_get();
1144         __u64 start, end;
1145         int rc;
1146         bool srvlock;
1147
1148         ENTRY;
1149
1150         if ((oa->o_valid & (OBD_MD_FLSIZE | OBD_MD_FLBLOCKS)) !=
1151             (OBD_MD_FLSIZE | OBD_MD_FLBLOCKS))
1152                 RETURN(err_serious(-EPROTO));
1153
1154         repbody = req_capsule_server_get(tsi->tsi_pill, &RMF_OST_BODY);
1155         if (repbody == NULL)
1156                 RETURN(err_serious(-ENOMEM));
1157
1158         /* punch start,end are passed in o_size,o_blocks throught wire */
1159         start = oa->o_size;
1160         end = oa->o_blocks;
1161
1162         if (end != OBD_OBJECT_EOF) /* Only truncate is supported */
1163                 RETURN(-EPROTO);
1164
1165         info = tsi2mdt_info(tsi);
1166         la = &info->mti_attr.ma_attr;
1167         /* standard truncate optimization: if file body is completely
1168          * destroyed, don't send data back to the server. */
1169         if (start == 0)
1170                 flags |= LDLM_FL_AST_DISCARD_DATA;
1171
1172         repbody->oa.o_oi = oa->o_oi;
1173         repbody->oa.o_valid = OBD_MD_FLID;
1174
1175         srvlock = (exp_connect_flags(exp) & OBD_CONNECT_SRVLOCK) &&
1176                   oa->o_valid & OBD_MD_FLFLAGS &&
1177                   oa->o_flags & OBD_FL_SRVLOCK;
1178
1179         if (srvlock) {
1180                 rc = tgt_mdt_data_lock(ns, &tsi->tsi_resid, &lh, LCK_PW,
1181                                        &flags);
1182                 if (rc != 0)
1183                         GOTO(out, rc);
1184         }
1185
1186         CDEBUG(D_INODE, "calling punch for object "DFID", valid = %#llx"
1187                ", start = %lld, end = %lld\n", PFID(&tsi->tsi_fid),
1188                oa->o_valid, start, end);
1189
1190         mo = mdt_object_find(tsi->tsi_env, mdt, &tsi->tsi_fid);
1191         if (IS_ERR(mo))
1192                 GOTO(out_unlock, rc = PTR_ERR(mo));
1193
1194         if (!mdt_object_exists(mo))
1195                 GOTO(out_put, rc = -ENOENT);
1196
1197         /* Shouldn't happen on dirs */
1198         if (S_ISDIR(lu_object_attr(&mo->mot_obj))) {
1199                 rc = -EPERM;
1200                 CERROR("%s: Truncate on dir "DFID": rc = %d\n",
1201                        exp->exp_obd->obd_name, PFID(&tsi->tsi_fid), rc);
1202                 GOTO(out_put, rc);
1203         }
1204
1205         mdt_dom_write_lock(mo);
1206         dob = mdt_obj2dt(mo);
1207
1208         la_from_obdo(la, oa, OBD_MD_FLMTIME | OBD_MD_FLATIME | OBD_MD_FLCTIME);
1209         la->la_size = start;
1210         la->la_valid |= LA_SIZE;
1211
1212         /* MDT supports FMD for Data-on-MDT needs */
1213         if (la->la_valid & (LA_ATIME | LA_MTIME | LA_CTIME))
1214                 tgt_fmd_update(tsi->tsi_exp, &tsi->tsi_fid,
1215                                tgt_ses_req(tsi)->rq_xid);
1216
1217         rc = mdt_object_punch(tsi->tsi_env, mdt->mdt_bottom, dob,
1218                               start, end, la);
1219         mdt_dom_write_unlock(mo);
1220         if (rc)
1221                 GOTO(out_put, rc);
1222
1223         mdt_dom_obj_lvb_update(tsi->tsi_env, mo, &repbody->oa, false);
1224
1225         mdt_counter_incr(req, LPROC_MDT_IO_PUNCH,
1226                          ktime_us_delta(ktime_get(), kstart));
1227         EXIT;
1228 out_put:
1229         lu_object_put(tsi->tsi_env, &mo->mot_obj);
1230 out_unlock:
1231         if (srvlock)
1232                 tgt_data_unlock(&lh, LCK_PW);
1233 out:
1234         mdt_thread_info_fini(info);
1235         return rc;
1236 }
1237
1238 /**
1239  * MDT glimpse for Data-on-MDT
1240  *
1241  * If there is write lock on client then function issues glimpse_ast to get
1242  * an actual size from that client.
1243  *
1244  */
1245 int mdt_do_glimpse(const struct lu_env *env, struct ldlm_namespace *ns,
1246                    struct ldlm_resource *res)
1247 {
1248         union ldlm_policy_data policy;
1249         struct lustre_handle lockh;
1250         enum ldlm_mode mode;
1251         struct ldlm_lock *lock;
1252         struct ldlm_glimpse_work *gl_work;
1253         LIST_HEAD(gl_list);
1254         int rc;
1255
1256         ENTRY;
1257
1258         /* There can be only one write lock covering data, try to match it. */
1259         policy.l_inodebits.bits = MDS_INODELOCK_DOM;
1260         mode = ldlm_lock_match(ns, LDLM_FL_TEST_LOCK,
1261                                &res->lr_name, LDLM_IBITS, &policy,
1262                                LCK_PW, &lockh);
1263
1264         /* There is no PW lock on this object; finished. */
1265         if (mode == 0)
1266                 RETURN(0);
1267
1268         lock = ldlm_handle2lock(&lockh);
1269         if (lock == NULL)
1270                 RETURN(0);
1271
1272         /*
1273          * This check is for lock taken in mdt_reint_unlink() that does
1274          * not have l_glimpse_ast set. So the logic is: if there is a lock
1275          * with no l_glimpse_ast set, this object is being destroyed already.
1276          * Hence, if you are grabbing DLM locks on the server, always set
1277          * non-NULL glimpse_ast (e.g., ldlm_request.c::ldlm_glimpse_ast()).
1278          */
1279         if (lock->l_glimpse_ast == NULL) {
1280                 LDLM_DEBUG(lock, "no l_glimpse_ast");
1281                 GOTO(out, rc = -ENOENT);
1282         }
1283
1284         OBD_SLAB_ALLOC_PTR_GFP(gl_work, ldlm_glimpse_work_kmem, GFP_ATOMIC);
1285         if (!gl_work)
1286                 GOTO(out, rc = -ENOMEM);
1287
1288         /* Populate the gl_work structure.
1289          * Grab additional reference on the lock which will be released in
1290          * ldlm_work_gl_ast_lock() */
1291         gl_work->gl_lock = LDLM_LOCK_GET(lock);
1292         /* The glimpse callback is sent to one single IO lock. As a result,
1293          * the gl_work list is just composed of one element */
1294         list_add_tail(&gl_work->gl_list, &gl_list);
1295         /* There is actually no need for a glimpse descriptor when glimpsing
1296          * IO locks */
1297         gl_work->gl_desc = NULL;
1298         /* the ldlm_glimpse_work structure is allocated on the stack */
1299         gl_work->gl_flags = LDLM_GL_WORK_SLAB_ALLOCATED;
1300
1301         ldlm_glimpse_locks(res, &gl_list); /* this will update the LVB */
1302
1303         /* If the list is not empty, we failed to glimpse a lock and
1304          * must clean it up. Usually due to a race with unlink.*/
1305         if (!list_empty(&gl_list)) {
1306                 LDLM_LOCK_RELEASE(lock);
1307                 OBD_SLAB_FREE_PTR(gl_work, ldlm_glimpse_work_kmem);
1308         }
1309         rc = 0;
1310         EXIT;
1311 out:
1312         LDLM_LOCK_PUT(lock);
1313         return rc;
1314 }
1315
1316 static void mdt_lvb2reply(struct ldlm_resource *res, struct mdt_body *mb,
1317                           struct ost_lvb *lvb)
1318 {
1319         struct ost_lvb *res_lvb;
1320
1321         lock_res(res);
1322         res_lvb = res->lr_lvb_data;
1323         if (res_lvb) {
1324                 if (lvb)
1325                         *lvb = *res_lvb;
1326
1327                 if (mb) {
1328                         mb->mbo_dom_size = res_lvb->lvb_size;
1329                         mb->mbo_dom_blocks = res_lvb->lvb_blocks;
1330                         mb->mbo_mtime = res_lvb->lvb_mtime;
1331                         mb->mbo_ctime = res_lvb->lvb_ctime;
1332                         mb->mbo_atime = res_lvb->lvb_atime;
1333                         mb->mbo_valid |= OBD_MD_FLATIME | OBD_MD_FLCTIME |
1334                                          OBD_MD_FLMTIME | OBD_MD_DOM_SIZE;
1335                 }
1336                 CDEBUG(D_DLMTRACE, "size %llu\n", res_lvb->lvb_size);
1337         }
1338         unlock_res(res);
1339 }
1340
1341 /**
1342  * MDT glimpse for Data-on-MDT
1343  *
1344  * This function is called when MDT get attributes for the DoM object.
1345  * If there is write lock on client then function issues glimpse_ast to get
1346  * an actual size from that client.
1347  */
1348 int mdt_dom_object_size(const struct lu_env *env, struct mdt_device *mdt,
1349                         const struct lu_fid *fid, struct mdt_body *mb,
1350                         bool dom_lock)
1351 {
1352         struct ldlm_res_id resid;
1353         struct ldlm_resource *res;
1354         int rc = 0;
1355
1356         ENTRY;
1357
1358         fid_build_reg_res_name(fid, &resid);
1359         res = ldlm_resource_get(mdt->mdt_namespace, &resid, LDLM_IBITS, 1);
1360         if (IS_ERR(res))
1361                 RETURN(-ENOENT);
1362
1363         /* Update lvbo data if DoM lock returned or if LVB is not yet valid. */
1364         if (dom_lock || !mdt_dom_lvb_is_valid(res))
1365                 mdt_dom_lvbo_update(res, NULL, NULL, false);
1366
1367         mdt_lvb2reply(res, mb, NULL);
1368         ldlm_resource_putref(res);
1369         RETURN(rc);
1370 }
1371
1372 /**
1373  * MDT DoM lock intent policy (glimpse)
1374  *
1375  * Intent policy is called when lock has an intent, for DoM file that
1376  * means glimpse lock and policy fills Lock Value Block (LVB).
1377  *
1378  * If already granted lock is found it will be placed in \a lockp and
1379  * returned back to caller function.
1380  *
1381  * \param[in] tsi        session info
1382  * \param[in,out] lockp  pointer to the lock
1383  * \param[in] flags      LDLM flags
1384  *
1385  * \retval              ELDLM_LOCK_REPLACED if already granted lock was found
1386  *                      and placed in \a lockp
1387  * \retval              ELDLM_LOCK_ABORTED in other cases except error
1388  * \retval              negative value on error
1389  */
1390 int mdt_glimpse_enqueue(struct mdt_thread_info *mti, struct ldlm_namespace *ns,
1391                         struct ldlm_lock **lockp, __u64 flags)
1392 {
1393         struct ldlm_lock *lock = *lockp;
1394         struct ldlm_resource *res = lock->l_resource;
1395         ldlm_processing_policy policy;
1396         struct ldlm_reply *rep;
1397         struct mdt_body *mbo;
1398         struct ost_lvb *lvb;
1399         bool old_client = !exp_connect_dom_lvb(mti->mti_exp);
1400         int rc;
1401
1402         ENTRY;
1403
1404         policy = ldlm_get_processing_policy(res);
1405         LASSERT(policy != NULL);
1406
1407         if (unlikely(old_client)) {
1408                 req_capsule_set_size(mti->mti_pill, &RMF_MDT_MD, RCL_SERVER, 0);
1409                 req_capsule_set_size(mti->mti_pill, &RMF_ACL, RCL_SERVER, 0);
1410         } else {
1411                 req_capsule_set_size(mti->mti_pill, &RMF_DLM_LVB, RCL_SERVER,
1412                                      sizeof(*lvb));
1413         }
1414         rc = req_capsule_server_pack(mti->mti_pill);
1415         if (rc)
1416                 RETURN(err_serious(rc));
1417
1418         rep = req_capsule_server_get(mti->mti_pill, &RMF_DLM_REP);
1419
1420         if (unlikely(old_client)) {
1421                 mbo = req_capsule_server_get(mti->mti_pill, &RMF_MDT_BODY);
1422                 LASSERT(mbo);
1423                 lvb = NULL;
1424         } else {
1425                 lvb = req_capsule_server_get(mti->mti_pill, &RMF_DLM_LVB);
1426                 LASSERT(lvb);
1427                 mbo = NULL;
1428         }
1429
1430         lock_res(res);
1431         /* Check if this is a resend case (MSG_RESENT is set on RPC) and a
1432          * lock was found by ldlm_handle_enqueue(); if so no need to grant
1433          * it again. */
1434         if (flags & LDLM_FL_RESENT) {
1435                 rc = LDLM_ITER_CONTINUE;
1436         } else {
1437                 __u64 tmpflags = LDLM_FL_BLOCK_NOWAIT;
1438                 enum ldlm_error err;
1439
1440                 rc = policy(lock, &tmpflags, LDLM_PROCESS_RESCAN, &err, NULL);
1441                 check_res_locked(res);
1442         }
1443         unlock_res(res);
1444
1445         /* The lock met with no resistance; we're finished. */
1446         if (rc == LDLM_ITER_CONTINUE) {
1447                 GOTO(fill_mbo, rc = ELDLM_LOCK_REPLACED);
1448         } else if (flags & LDLM_FL_BLOCK_NOWAIT) {
1449                 /* LDLM_FL_BLOCK_NOWAIT means it is for AGL. Do not send glimpse
1450                  * callback for glimpse size. The real size user will trigger
1451                  * the glimpse callback when necessary. */
1452                 GOTO(fill_mbo, rc = ELDLM_LOCK_ABORTED);
1453         }
1454
1455         rc = mdt_do_glimpse(mti->mti_env, ns, res);
1456         if (rc == -ENOENT) {
1457                 /* We are racing with unlink(); just return -ENOENT */
1458                 rep->lock_policy_res2 = ptlrpc_status_hton(-ENOENT);
1459         } else if (rc == -EINVAL) {
1460                 /* this is possible is client lock has been cancelled but
1461                  * still exists on server. If that lock was found on server
1462                  * as only conflicting lock then the client has already
1463                  * size authority and glimpse is not needed. */
1464                 CDEBUG(D_DLMTRACE, "Glimpse from the client owning lock\n");
1465         } else if (rc < 0) {
1466                 RETURN(rc);
1467         }
1468         rc = ELDLM_LOCK_ABORTED;
1469 fill_mbo:
1470         /* LVB can be without valid data in case of DOM */
1471         if (!mdt_dom_lvb_is_valid(res))
1472                 mdt_dom_lvbo_update(res, lock, NULL, false);
1473         mdt_lvb2reply(res, mbo, lvb);
1474
1475         RETURN(rc);
1476 }
1477
1478 int mdt_brw_enqueue(struct mdt_thread_info *mti, struct ldlm_namespace *ns,
1479                     struct ldlm_lock **lockp, __u64 flags)
1480 {
1481         struct tgt_session_info *tsi = tgt_ses_info(mti->mti_env);
1482         struct lu_fid *fid = &tsi->tsi_fid;
1483         struct ldlm_lock *lock = *lockp;
1484         struct ldlm_resource *res = lock->l_resource;
1485         struct ldlm_reply *rep;
1486         struct mdt_body *mbo;
1487         struct mdt_lock_handle *lhc = &mti->mti_lh[MDT_LH_RMT];
1488         struct mdt_object *mo;
1489         int rc = 0;
1490
1491         ENTRY;
1492
1493         req_capsule_set_size(mti->mti_pill, &RMF_MDT_MD, RCL_SERVER, 0);
1494         req_capsule_set_size(mti->mti_pill, &RMF_ACL, RCL_SERVER, 0);
1495         rc = req_capsule_server_pack(mti->mti_pill);
1496         if (rc)
1497                 RETURN(err_serious(rc));
1498
1499         rep = req_capsule_server_get(mti->mti_pill, &RMF_DLM_REP);
1500         if (rep == NULL)
1501                 RETURN(-EPROTO);
1502
1503         mbo = req_capsule_server_get(mti->mti_pill, &RMF_MDT_BODY);
1504         if (mbo == NULL)
1505                 RETURN(-EPROTO);
1506
1507         fid_extract_from_res_name(fid, &res->lr_name);
1508         mo = mdt_object_find(mti->mti_env, mti->mti_mdt, fid);
1509         if (unlikely(IS_ERR(mo)))
1510                 RETURN(PTR_ERR(mo));
1511
1512         if (!mdt_object_exists(mo))
1513                 GOTO(out, rc = -ENOENT);
1514
1515         if (mdt_object_remote(mo))
1516                 GOTO(out, rc = -EPROTO);
1517
1518         /* Get lock from request for possible resent case. */
1519         mdt_intent_fixup_resent(mti, *lockp, lhc, flags);
1520         /* resent case */
1521         if (!lustre_handle_is_used(&lhc->mlh_reg_lh)) {
1522                 __u64 ibits = MDS_INODELOCK_DOM;
1523
1524                 mdt_lh_reg_init(lhc, *lockp);
1525
1526                 /* This will block MDT thread but it should be fine until
1527                  * client caches small amount of data for DoM, which should be
1528                  * smaller than one BRW RPC and should be able to be
1529                  * piggybacked by lock cancel RPC.
1530                  * If the client could hold the lock too long, this code can be
1531                  * revised to call mdt_object_lock_try(). And if fails, it will
1532                  * return ELDLM_OK here and fall back into normal lock enqueue
1533                  * process.
1534                  */
1535                 rc = mdt_object_lock_internal(mti, mo, mdt_object_fid(mo), lhc,
1536                                               &ibits, 0, false, false);
1537                 if (rc)
1538                         GOTO(out, rc);
1539         }
1540
1541         if (!mdt_dom_lvb_is_valid(res)) {
1542                 rc = mdt_dom_lvb_alloc(res);
1543                 if (rc)
1544                         GOTO(out_fail, rc);
1545                 mdt_dom_disk_lvbo_update(mti->mti_env, mo, res, false);
1546         }
1547         mdt_lvb2reply(res, mbo, NULL);
1548 out_fail:
1549         rep->lock_policy_res2 = clear_serious(rc);
1550         if (rep->lock_policy_res2) {
1551                 lhc->mlh_reg_lh.cookie = 0ull;
1552                 GOTO(out, rc = ELDLM_LOCK_ABORTED);
1553         }
1554
1555         rc = mdt_intent_lock_replace(mti, lockp, lhc, flags, rc);
1556 out:
1557         if (rc < 0)
1558                 lhc->mlh_reg_lh.cookie = 0ull;
1559         mdt_object_put(mti->mti_env, mo);
1560         RETURN(rc);
1561 }
1562
1563 /* check if client has already DoM lock for given resource */
1564 bool mdt_dom_client_has_lock(struct mdt_thread_info *info,
1565                              const struct lu_fid *fid)
1566 {
1567         struct mdt_device *mdt = info->mti_mdt;
1568         union ldlm_policy_data *policy = &info->mti_policy;
1569         struct ldlm_res_id *res_id = &info->mti_res_id;
1570         __u64 open_flags = info->mti_spec.sp_cr_flags;
1571         struct lustre_handle lockh;
1572         enum ldlm_mode mode;
1573         struct ldlm_lock *lock;
1574         enum ldlm_mode lm;
1575         bool rc;
1576
1577         policy->l_inodebits.bits = MDS_INODELOCK_DOM;
1578         fid_build_reg_res_name(fid, res_id);
1579
1580
1581         lm = (open_flags & MDS_FMODE_WRITE) ? LCK_PW : LCK_PR | LCK_PW;
1582         mode = ldlm_lock_match(mdt->mdt_namespace, LDLM_FL_BLOCK_GRANTED |
1583                                LDLM_FL_TEST_LOCK, res_id, LDLM_IBITS, policy,
1584                                lm, &lockh);
1585
1586         /* There is no other PW lock on this object; finished. */
1587         if (mode == 0)
1588                 return false;
1589
1590         lock = ldlm_handle2lock(&lockh);
1591         if (lock == 0)
1592                 return false;
1593
1594         /* check if lock from the same client */
1595         rc = (lock->l_export->exp_handle.h_cookie ==
1596               info->mti_exp->exp_handle.h_cookie);
1597         LDLM_LOCK_PUT(lock);
1598         return rc;
1599 }
1600
1601 /**
1602  * MDT request handler for OST_GETATTR RPC.
1603  *
1604  * This is data-specific request to get object and layout versions under
1605  * IO lock. It is reliable only for Data-on-MDT files.
1606  *
1607  * \param[in] tsi target session environment for this request
1608  *
1609  * \retval 0 if successful
1610  * \retval negative value on error
1611  */
1612 int mdt_data_version_get(struct tgt_session_info *tsi)
1613 {
1614         struct mdt_thread_info *mti = mdt_th_info(tsi->tsi_env);
1615         struct mdt_device *mdt = mti->mti_mdt;
1616         struct mdt_body *repbody;
1617         struct mdt_object *mo = mti->mti_object;
1618         struct lov_comp_md_v1 *comp;
1619         struct lustre_handle lh = { 0 };
1620         __u64 flags = 0;
1621         __s64 version;
1622         enum ldlm_mode lock_mode = LCK_PR;
1623         bool srvlock;
1624         int rc;
1625
1626         ENTRY;
1627
1628         req_capsule_set_size(tsi->tsi_pill, &RMF_MDT_MD, RCL_SERVER, 0);
1629         req_capsule_set_size(tsi->tsi_pill, &RMF_ACL, RCL_SERVER, 0);
1630         req_capsule_set_size(tsi->tsi_pill, &RMF_FILE_ENCCTX, RCL_SERVER, 0);
1631         rc = req_capsule_server_pack(tsi->tsi_pill);
1632         if (unlikely(rc != 0))
1633                 RETURN(err_serious(rc));
1634
1635         repbody = req_capsule_server_get(tsi->tsi_pill, &RMF_MDT_BODY);
1636         if (repbody == NULL)
1637                 RETURN(-ENOMEM);
1638
1639         srvlock = tsi->tsi_mdt_body->mbo_valid & OBD_MD_FLFLAGS &&
1640                   tsi->tsi_mdt_body->mbo_flags & OBD_FL_SRVLOCK;
1641
1642         if (srvlock) {
1643                 if (unlikely(tsi->tsi_mdt_body->mbo_flags & OBD_FL_FLUSH))
1644                         lock_mode = LCK_PW;
1645
1646                 fid_build_reg_res_name(&tsi->tsi_fid, &tsi->tsi_resid);
1647                 rc = tgt_mdt_data_lock(mdt->mdt_namespace, &tsi->tsi_resid,
1648                                        &lh, lock_mode, &flags);
1649                 if (rc != 0)
1650                         RETURN(rc);
1651         }
1652
1653         if (!mdt_object_exists(mo))
1654                 GOTO(out, rc = -ENOENT);
1655         if (mdt_object_remote(mo))
1656                 GOTO(out, rc = -EREMOTE);
1657         if (!S_ISREG(lu_object_attr(&mo->mot_obj)))
1658                 GOTO(out, rc = -EBADF);
1659
1660         /* Get version first */
1661         version = dt_version_get(tsi->tsi_env, mdt_obj2dt(mo));
1662         if (version && version != -EOPNOTSUPP) {
1663                 repbody->mbo_valid |= OBD_MD_FLDATAVERSION;
1664                 /* re-use mbo_ioepoch to transfer version */
1665                 repbody->mbo_version = version;
1666         }
1667
1668         /* Read layout to get its version */
1669         rc = mdt_big_xattr_get(mti, mo, XATTR_NAME_LOV);
1670         if (rc == -ENODATA) /* File has no layout yet */
1671                 GOTO(out, rc = 0);
1672         else if (rc < 0)
1673                 GOTO(out, rc);
1674
1675         comp = mti->mti_buf.lb_buf;
1676         if (le32_to_cpu(comp->lcm_magic) != LOV_MAGIC_COMP_V1) {
1677                 CDEBUG(D_INFO, DFID" has no composite layout",
1678                        PFID(&tsi->tsi_fid));
1679                 GOTO(out, rc = -ESTALE);
1680         }
1681
1682         CDEBUG(D_INODE, DFID": layout version: %u\n",
1683                PFID(&tsi->tsi_fid), le32_to_cpu(comp->lcm_layout_gen));
1684
1685         repbody->mbo_valid |= OBD_MD_LAYOUT_VERSION;
1686         /* re-use mbo_rdev for that */
1687         repbody->mbo_layout_gen = le32_to_cpu(comp->lcm_layout_gen);
1688         rc = 0;
1689 out:
1690         if (srvlock)
1691                 tgt_data_unlock(&lh, lock_mode);
1692
1693         repbody->mbo_valid |= OBD_MD_FLFLAGS;
1694         repbody->mbo_flags = OBD_FL_FLUSH;
1695         RETURN(rc);
1696 }
1697
1698 /* read file data to the buffer */
1699 int mdt_dom_read_on_open(struct mdt_thread_info *mti, struct mdt_device *mdt,
1700                          struct lustre_handle *lh)
1701 {
1702         const struct lu_env *env = mti->mti_env;
1703         struct tgt_session_info *tsi = tgt_ses_info(env);
1704         struct req_capsule *pill = tsi->tsi_pill;
1705         const struct lu_fid *fid;
1706         struct ptlrpc_request *req = tgt_ses_req(tsi);
1707         struct mdt_body *mbo;
1708         struct dt_device *dt = mdt->mdt_bottom;
1709         struct dt_object *mo;
1710         void *buf;
1711         struct niobuf_remote *rnb = NULL;
1712         struct niobuf_local *lnb;
1713         int rc;
1714         loff_t offset;
1715         unsigned int len, copied = 0;
1716         __u64 real_dom_size;
1717         int lnbs, nr_local, i;
1718         bool dom_lock = false;
1719
1720         ENTRY;
1721
1722         if (!req_capsule_field_present(pill, &RMF_NIOBUF_INLINE, RCL_SERVER)) {
1723                 /* There is no reply buffers for this field, this means that
1724                  * client has no support for data in reply.
1725                  */
1726                 RETURN(0);
1727         }
1728
1729         mbo = req_capsule_server_get(pill, &RMF_MDT_BODY);
1730         if (!(mbo->mbo_valid & OBD_MD_DOM_SIZE))
1731                 RETURN(0);
1732
1733         if (!mbo->mbo_dom_size)
1734                 RETURN(0);
1735
1736         if (lustre_handle_is_used(lh)) {
1737                 struct ldlm_lock *lock;
1738
1739                 lock = ldlm_handle2lock(lh);
1740                 if (lock) {
1741                         dom_lock = ldlm_has_dom(lock) && ldlm_has_layout(lock);
1742                         LDLM_LOCK_PUT(lock);
1743                 }
1744         }
1745
1746         /* return data along with open only along with DoM lock */
1747         if (!dom_lock || !mdt->mdt_opts.mo_dom_read_open)
1748                 RETURN(0);
1749
1750         /* if DoM object holds encrypted content, we need to make sure we
1751          * send whole encryption units, or client will read corrupted content
1752          */
1753         if (mbo->mbo_valid & LA_FLAGS && mbo->mbo_flags & LUSTRE_ENCRYPT_FL &&
1754             mbo->mbo_dom_size & ~LUSTRE_ENCRYPTION_MASK)
1755                 real_dom_size = (mbo->mbo_dom_size & LUSTRE_ENCRYPTION_MASK) +
1756                                 LUSTRE_ENCRYPTION_UNIT_SIZE;
1757         else
1758                 real_dom_size = mbo->mbo_dom_size;
1759
1760         CDEBUG(D_INFO, "File size %llu, reply sizes %d/%d\n",
1761                real_dom_size, req->rq_reqmsg->lm_repsize, req->rq_replen);
1762         len = req->rq_reqmsg->lm_repsize - req->rq_replen;
1763
1764         /* NB: at this moment we have the following sizes:
1765          * - req->rq_replen: used data in reply
1766          * - req->rq_reqmsg->lm_repsize: total allocated reply buffer at client
1767          *
1768          * Ideal case when file size fits in allocated reply buffer,
1769          * that mean we can return whole data in reply. We can also fit more
1770          * data up to max_reply_size in total reply size, but this will cause
1771          * re-allocation on client and resend with larger buffer. This is still
1772          * faster than separate READ IO.
1773          * Third case if file is too big to fit even in maximum size, in that
1774          * case we return just tail to optimize possible append.
1775          *
1776          * At the moment the following strategy is used:
1777          * 1) try to fit into the buffer we have
1778          * 2) return just file tail otherwise.
1779          */
1780         if (real_dom_size <= len) {
1781                 /* can fit whole data */
1782                 len = real_dom_size;
1783                 offset = 0;
1784         } else if (real_dom_size <
1785                    mdt_lmm_dom_stripesize(mti->mti_attr.ma_lmm)) {
1786                 int tail, pgbits;
1787
1788                 /* File tail offset must be aligned with larger page size
1789                  * between client and server, so the maximum page size is
1790                  * used here to align offset.
1791                  *
1792                  * NB: DOM feature was introduced when server supports pagebits
1793                  * already, so it should be always non-zero value. Report error
1794                  * if it is not for some reason.
1795                  */
1796                 if (!req->rq_export->exp_target_data.ted_pagebits) {
1797                         CERROR("%s: client page bits are not saved on server\n",
1798                                mdt_obd_name(mdt));
1799                         RETURN(0);
1800                 }
1801                 pgbits = max_t(int, PAGE_SHIFT,
1802                                req->rq_export->exp_target_data.ted_pagebits);
1803                 tail = real_dom_size % (1 << pgbits);
1804
1805                 /* no partial tail or tail can't fit in reply */
1806                 if (tail == 0 || len < tail)
1807                         RETURN(0);
1808
1809                 len = tail;
1810                 offset = real_dom_size - len;
1811         } else {
1812                 /* DOM stripe is fully written, so don't expect its tail
1813                  * will be used by append.
1814                  */
1815                 RETURN(0);
1816         }
1817
1818         LASSERT((offset & ~PAGE_MASK) == 0);
1819         rc = req_capsule_server_grow(pill, &RMF_NIOBUF_INLINE,
1820                                      sizeof(*rnb) + len);
1821         if (rc != 0) {
1822                 /* failed to grow data buffer, just exit */
1823                 GOTO(out, rc = -E2BIG);
1824         }
1825
1826         /* re-take MDT_BODY and NIOBUF_INLINE buffers after the buffer grow */
1827         mbo = req_capsule_server_get(pill, &RMF_MDT_BODY);
1828         fid = &mbo->mbo_fid1;
1829         if (!fid_is_sane(fid))
1830                 GOTO(out, rc = -EINVAL);
1831
1832         rnb = req_capsule_server_get(tsi->tsi_pill, &RMF_NIOBUF_INLINE);
1833         if (rnb == NULL)
1834                 GOTO(out, rc = -EPROTO);
1835
1836         buf = (char *)rnb + sizeof(*rnb);
1837         rnb->rnb_len = len;
1838         rnb->rnb_offset = offset;
1839
1840         mo = dt_locate(env, dt, fid);
1841         if (IS_ERR(mo))
1842                 GOTO(out_rnb, rc = PTR_ERR(mo));
1843         LASSERT(mo != NULL);
1844
1845         dt_read_lock(env, mo, 0);
1846         if (!dt_object_exists(mo))
1847                 GOTO(unlock, rc = -ENOENT);
1848
1849         /* parse remote buffers to local buffers and prepare the latter */
1850         lnbs = (len >> PAGE_SHIFT) + 1;
1851         OBD_ALLOC_PTR_ARRAY(lnb, lnbs);
1852         if (lnb == NULL)
1853                 GOTO(unlock, rc = -ENOMEM);
1854
1855         rc = dt_bufs_get(env, mo, rnb, lnb, lnbs, 0);
1856         if (unlikely(rc < 0))
1857                 GOTO(free, rc);
1858         LASSERT(rc <= lnbs);
1859         nr_local = rc;
1860         rc = dt_read_prep(env, mo, lnb, nr_local);
1861         if (unlikely(rc))
1862                 GOTO(buf_put, rc);
1863         /* copy data to the buffer finally */
1864         for (i = 0; i < nr_local; i++) {
1865                 char *p = kmap(lnb[i].lnb_page);
1866                 long off;
1867
1868                 LASSERT(lnb[i].lnb_page_offset == 0);
1869                 off = lnb[i].lnb_len & ~PAGE_MASK;
1870                 if (off > 0)
1871                         memset(p + off, 0, PAGE_SIZE - off);
1872
1873                 memcpy(buf + (i << PAGE_SHIFT), p, lnb[i].lnb_len);
1874                 kunmap(lnb[i].lnb_page);
1875                 copied += lnb[i].lnb_len;
1876                 LASSERT(rc <= len);
1877         }
1878         CDEBUG(D_INFO, "Read %i (wanted %u) bytes from %llu\n", copied,
1879                len, offset);
1880         if (copied < len) {
1881                 CWARN("%s: read %i bytes for "DFID
1882                       " but wanted %u, is size wrong?\n",
1883                       tsi->tsi_exp->exp_obd->obd_name, copied,
1884                       PFID(&tsi->tsi_fid), len);
1885                 /* Ignore partially copied data */
1886                 copied = 0;
1887         }
1888         EXIT;
1889 buf_put:
1890         dt_bufs_put(env, mo, lnb, nr_local);
1891 free:
1892         OBD_FREE_PTR_ARRAY(lnb, lnbs);
1893 unlock:
1894         dt_read_unlock(env, mo);
1895         lu_object_put(env, &mo->do_lu);
1896 out_rnb:
1897         rnb->rnb_len = copied;
1898 out:
1899         /* Don't fail OPEN request if read-on-open is failed, but drop
1900          * a message in log about the error.
1901          */
1902         if (rc)
1903                 CDEBUG(D_INFO, "Read-on-open is failed, rc = %d", rc);
1904
1905         RETURN(0);
1906 }
1907
1908 /**
1909  * Completion AST for DOM discard locks:
1910  *
1911  * CP AST an DOM discard lock is called always right after enqueue or from
1912  * reprocess if lock was blocked, in the latest case l_ast_data is set to
1913  * the mdt_object which is kept while there are pending locks on it.
1914  */
1915 int ldlm_dom_discard_cp_ast(struct ldlm_lock *lock, __u64 flags, void *data)
1916 {
1917         struct mdt_object *mo;
1918         struct lustre_handle dom_lh;
1919         struct lu_env *env;
1920
1921         ENTRY;
1922
1923         /* l_ast_data is set when lock was not granted immediately
1924          * in mdt_dom_discard_data() below but put into waiting list,
1925          * so this CP callback means we are finished and corresponding
1926          * MDT object should be released finally as well as lock itself.
1927          */
1928         lock_res_and_lock(lock);
1929         if (!lock->l_ast_data) {
1930                 unlock_res_and_lock(lock);
1931                 RETURN(0);
1932         }
1933
1934         mo = lock->l_ast_data;
1935         lock->l_ast_data = NULL;
1936         unlock_res_and_lock(lock);
1937
1938         ldlm_lock2handle(lock, &dom_lh);
1939         ldlm_lock_decref(&dom_lh, LCK_PW);
1940
1941         env = lu_env_find();
1942         LASSERT(env);
1943
1944         lu_env_refill(env);
1945
1946         mdt_object_put(env, mo);
1947
1948         RETURN(0);
1949 }
1950
1951 void mdt_dom_discard_data(struct mdt_thread_info *info,
1952                           struct mdt_object *mo)
1953 {
1954         struct ptlrpc_request *req = mdt_info_req(info);
1955         struct mdt_device *mdt = mdt_dev(mo->mot_obj.lo_dev);
1956         union ldlm_policy_data policy;
1957         struct ldlm_res_id res_id;
1958         struct lustre_handle dom_lh;
1959         struct ldlm_lock *lock;
1960         __u64 flags = LDLM_FL_AST_DISCARD_DATA;
1961         int rc = 0;
1962         bool old_client;
1963
1964         ENTRY;
1965
1966         if (req && req_is_replay(req))
1967                 RETURN_EXIT;
1968
1969         policy.l_inodebits.bits = MDS_INODELOCK_DOM;
1970         policy.l_inodebits.try_bits = 0;
1971         fid_build_reg_res_name(mdt_object_fid(mo), &res_id);
1972
1973         /* Keep blocking version of discard for an old client to avoid
1974          * crashes on non-patched clients. LU-11359.
1975          */
1976         old_client = req && !(exp_connect_flags2(req->rq_export) &
1977                               OBD_CONNECT2_ASYNC_DISCARD);
1978
1979         /* Tell the clients that the object is gone now and that they should
1980          * throw away any cached pages. */
1981         rc = ldlm_cli_enqueue_local(info->mti_env, mdt->mdt_namespace, &res_id,
1982                                     LDLM_IBITS, &policy, LCK_PW, &flags,
1983                                     ldlm_blocking_ast, old_client ?
1984                                     ldlm_completion_ast :
1985                                     ldlm_dom_discard_cp_ast,
1986                                     NULL, NULL, 0, LVB_T_NONE, NULL, &dom_lh);
1987         if (rc != ELDLM_OK) {
1988                 CDEBUG(D_DLMTRACE,
1989                        "Failed to issue discard lock, rc = %d\n", rc);
1990                 RETURN_EXIT;
1991         }
1992
1993         lock = ldlm_handle2lock(&dom_lh);
1994         lock_res_and_lock(lock);
1995         /* if lock is not granted then there are BL ASTs in progress and
1996          * lock will be granted in result of reprocessing with CP callback
1997          * notifying about that. The mdt object has to be kept until that and
1998          * it is saved in l_ast_data of the lock. Lock reference is kept too
1999          * until that to prevent it from canceling.
2000          */
2001         if (!is_granted_or_cancelled_nolock(lock)) {
2002                 mdt_object_get(info->mti_env, mo);
2003                 lock->l_ast_data = mo;
2004                 unlock_res_and_lock(lock);
2005         } else {
2006                 unlock_res_and_lock(lock);
2007                 ldlm_lock_decref_and_cancel(&dom_lh, LCK_PW);
2008         }
2009         LDLM_LOCK_PUT(lock);
2010
2011         RETURN_EXIT;
2012 }