lustre/ost/ost_handler.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) 2001, 2010, Oracle and/or its affiliates. All rights reserved.
  28  * Use is subject to license terms.
  29  *
  30  * Copyright (c) 2011, 2012, Whamcloud, Inc.
  31  */
  32 /*
  33  * This file is part of Lustre, http://www.lustre.org/
  34  * Lustre is a trademark of Sun Microsystems, Inc.
  35  *
  36  * lustre/ost/ost_handler.c
  37  *
  38  * Author: Peter J. Braam <braam@clusterfs.com>
  39  * Author: Phil Schwan <phil@clusterfs.com>
  40  */
  41
  42 #define DEBUG_SUBSYSTEM S_OST
  43
  44 #include <linux/module.h>
  45 #include <obd_cksum.h>
  46 #include <obd_ost.h>
  47 #include <lustre_net.h>
  48 #include <lustre_dlm.h>
  49 #include <lustre_export.h>
  50 #include <lustre_debug.h>
  51 #include <linux/init.h>
  52 #include <lprocfs_status.h>
  53 #include <libcfs/list.h>
  54 #include <lustre_quota.h>
  55 #include "ost_internal.h"
  56
  57 static int oss_num_threads;
  58 CFS_MODULE_PARM(oss_num_threads, "i", int, 0444,
  59                 "number of OSS service threads to start");
  60
  61 static int ost_num_threads;
  62 CFS_MODULE_PARM(ost_num_threads, "i", int, 0444,
  63                 "number of OST service threads to start (deprecated)");
  64
  65 static int oss_num_create_threads;
  66 CFS_MODULE_PARM(oss_num_create_threads, "i", int, 0444,
  67                 "number of OSS create threads to start");
  68
  69 /**
  70  * Do not return server-side uid/gid to remote client
  71  */
  72 static void ost_drop_id(struct obd_export *exp, struct obdo *oa)
  73 {
  74         if (exp_connect_rmtclient(exp)) {
  75                 oa->o_uid = -1;
  76                 oa->o_gid = -1;
  77                 oa->o_valid &= ~(OBD_MD_FLUID | OBD_MD_FLGID);
  78         }
  79 }
  80
  81 /**
  82  * Validate oa from client.
  83  * If the request comes from 2.0 clients, currently only RSVD seq and IDIF
  84  * req are valid.
  85  *    a. for single MDS  seq = FID_SEQ_OST_MDT0,
  86  *    b. for CMD, seq = FID_SEQ_OST_MDT0, FID_SEQ_OST_MDT1 - FID_SEQ_OST_MAX
  87  */
  88 static int ost_validate_obdo(struct obd_export *exp, struct obdo *oa,
  89                              struct obd_ioobj *ioobj)
  90 {
  91         if (oa != NULL && !(oa->o_valid & OBD_MD_FLGROUP)) {
  92                 oa->o_seq = FID_SEQ_OST_MDT0;
  93                 if (ioobj)
  94                         ioobj->ioo_seq = FID_SEQ_OST_MDT0;
  95         /* remove fid_seq_is_rsvd() after FID-on-OST allows SEQ > 9 */
  96         } else if (oa == NULL || !(fid_seq_is_rsvd(oa->o_seq) ||
  97                                    fid_seq_is_mdt0(oa->o_seq))) {
  98                 CERROR("%s: client %s sent invalid object "POSTID"\n",
  99                        exp->exp_obd->obd_name, obd_export_nid2str(exp),
 100                        oa ? oa->o_id : -1, oa ? oa->o_seq : -1);
 101                 return -EPROTO;
 102         }
 103         obdo_from_ostid(oa, &oa->o_oi);
 104         if (ioobj)
 105                 ioobj_from_obdo(ioobj, oa);
 106         return 0;
 107 }
 108
 109 void oti_to_request(struct obd_trans_info *oti, struct ptlrpc_request *req)
 110 {
 111         struct oti_req_ack_lock *ack_lock;
 112         int i;
 113
 114         if (oti == NULL)
 115                 return;
 116
 117         if (req->rq_repmsg) {
 118                 __u64 versions[PTLRPC_NUM_VERSIONS] = { 0 };
 119                 lustre_msg_set_transno(req->rq_repmsg, oti->oti_transno);
 120                 versions[0] = oti->oti_pre_version;
 121                 lustre_msg_set_versions(req->rq_repmsg, versions);
 122         }
 123         req->rq_transno = oti->oti_transno;
 124
 125         /* XXX 4 == entries in oti_ack_locks??? */
 126         for (ack_lock = oti->oti_ack_locks, i = 0; i < 4; i++, ack_lock++) {
 127                 if (!ack_lock->mode)
 128                         break;
 129                 /* XXX not even calling target_send_reply in some cases... */
 130                 ptlrpc_save_lock (req, &ack_lock->lock, ack_lock->mode, 0);
 131         }
 132 }
 133
 134 static int ost_destroy(struct obd_export *exp, struct ptlrpc_request *req,
 135                        struct obd_trans_info *oti)
 136 {
 137         struct ost_body *body, *repbody;
 138         struct lustre_capa *capa = NULL;
 139         int rc;
 140         ENTRY;
 141
 142         /* Get the request body */
 143         body = req_capsule_client_get(&req->rq_pill, &RMF_OST_BODY);
 144         if (body == NULL)
 145                 RETURN(-EFAULT);
 146
 147         if (body->oa.o_id == 0)
 148                 RETURN(-EPROTO);
 149
 150         rc = ost_validate_obdo(exp, &body->oa, NULL);
 151         if (rc)
 152                 RETURN(rc);
 153
 154         /* If there's a DLM request, cancel the locks mentioned in it*/
 155         if (req_capsule_field_present(&req->rq_pill, &RMF_DLM_REQ, RCL_CLIENT)) {
 156                 struct ldlm_request *dlm;
 157
 158                 dlm = req_capsule_client_get(&req->rq_pill, &RMF_DLM_REQ);
 159                 if (dlm == NULL)
 160                         RETURN (-EFAULT);
 161                 ldlm_request_cancel(req, dlm, 0);
 162         }
 163
 164         /* If there's a capability, get it */
 165         if (body->oa.o_valid & OBD_MD_FLOSSCAPA) {
 166                 capa = req_capsule_client_get(&req->rq_pill, &RMF_CAPA1);
 167                 if (capa == NULL) {
 168                         CERROR("Missing capability for OST DESTROY");
 169                         RETURN (-EFAULT);
 170                 }
 171         }
 172
 173         /* Prepare the reply */
 174         rc = req_capsule_server_pack(&req->rq_pill);
 175         if (rc)
 176                 RETURN(rc);
 177
 178         /* Get the log cancellation cookie */
 179         if (body->oa.o_valid & OBD_MD_FLCOOKIE)
 180                 oti->oti_logcookies = &body->oa.o_lcookie;
 181
 182         /* Finish the reply */
 183         repbody = req_capsule_server_get(&req->rq_pill, &RMF_OST_BODY);
 184         memcpy(&repbody->oa, &body->oa, sizeof(body->oa));
 185
 186         /* Do the destroy and set the reply status accordingly  */
 187         req->rq_status = obd_destroy(req->rq_svc_thread->t_env, exp,
 188                                      &repbody->oa, NULL, oti, NULL, capa);
 189         RETURN(0);
 190 }
 191
 192 /**
 193  * Helper function for getting server side [start, start+count] DLM lock
 194  * if asked by client.
 195  */
 196 static int ost_lock_get(struct obd_export *exp, struct obdo *oa,
 197                         __u64 start, __u64 count, struct lustre_handle *lh,
 198                         int mode, int flags)
 199 {
 200         struct ldlm_res_id res_id;
 201         ldlm_policy_data_t policy;
 202         __u64 end = start + count;
 203
 204         ENTRY;
 205
 206         LASSERT(!lustre_handle_is_used(lh));
 207         /* o_id and o_gr are used for localizing resource, if client miss to set
 208          * them, do not trigger ASSERTION. */
 209         if (unlikely((oa->o_valid & (OBD_MD_FLID | OBD_MD_FLGROUP)) !=
 210                      (OBD_MD_FLID | OBD_MD_FLGROUP)))
 211                 RETURN(-EPROTO);
 212
 213         if (!(oa->o_valid & OBD_MD_FLFLAGS) ||
 214             !(oa->o_flags & OBD_FL_SRVLOCK))
 215                 RETURN(0);
 216
 217         osc_build_res_name(oa->o_id, oa->o_seq, &res_id);
 218         CDEBUG(D_INODE, "OST-side extent lock.\n");
 219
 220         policy.l_extent.start = start & CFS_PAGE_MASK;
 221
 222         /* If ->o_blocks is EOF it means "lock till the end of the
 223          * file". Otherwise, it's size of a hole being punched (in bytes) */
 224         if (count == OBD_OBJECT_EOF || end < start)
 225                 policy.l_extent.end = OBD_OBJECT_EOF;
 226         else
 227                 policy.l_extent.end = end | ~CFS_PAGE_MASK;
 228
 229         RETURN(ldlm_cli_enqueue_local(exp->exp_obd->obd_namespace, &res_id,
 230                                       LDLM_EXTENT, &policy, mode, &flags,
 231                                       ldlm_blocking_ast, ldlm_completion_ast,
 232                                       ldlm_glimpse_ast, NULL, 0, NULL, lh));
 233 }
 234
 235 /* Helper function: release lock, if any. */
 236 static void ost_lock_put(struct obd_export *exp,
 237                          struct lustre_handle *lh, int mode)
 238 {
 239         ENTRY;
 240         if (lustre_handle_is_used(lh))
 241                 ldlm_lock_decref(lh, mode);
 242         EXIT;
 243 }
 244
 245 static int ost_getattr(struct obd_export *exp, struct ptlrpc_request *req)
 246 {
 247         struct ost_body *body, *repbody;
 248         struct obd_info *oinfo;
 249         struct lustre_handle lh = { 0 };
 250         struct lustre_capa *capa = NULL;
 251         int rc;
 252         ENTRY;
 253
 254         body = req_capsule_client_get(&req->rq_pill, &RMF_OST_BODY);
 255         if (body == NULL)
 256                 RETURN(-EFAULT);
 257
 258         rc = ost_validate_obdo(exp, &body->oa, NULL);
 259         if (rc)
 260                 RETURN(rc);
 261
 262         if (body->oa.o_valid & OBD_MD_FLOSSCAPA) {
 263                 capa = req_capsule_client_get(&req->rq_pill, &RMF_CAPA1);
 264                 if (capa == NULL) {
 265                         CERROR("Missing capability for OST GETATTR");
 266                         RETURN(-EFAULT);
 267                 }
 268         }
 269
 270         rc = req_capsule_server_pack(&req->rq_pill);
 271         if (rc)
 272                 RETURN(rc);
 273
 274         repbody = req_capsule_server_get(&req->rq_pill, &RMF_OST_BODY);
 275         repbody->oa = body->oa;
 276
 277         rc = ost_lock_get(exp, &repbody->oa, 0, OBD_OBJECT_EOF, &lh, LCK_PR, 0);
 278         if (rc)
 279                 RETURN(rc);
 280
 281         OBD_ALLOC_PTR(oinfo);
 282         if (!oinfo)
 283                 GOTO(unlock, rc = -ENOMEM);
 284         oinfo->oi_oa = &repbody->oa;
 285         oinfo->oi_capa = capa;
 286
 287         req->rq_status = obd_getattr(req->rq_svc_thread->t_env, exp, oinfo);
 288
 289         OBD_FREE_PTR(oinfo);
 290
 291         ost_drop_id(exp, &repbody->oa);
 292
 293 unlock:
 294         ost_lock_put(exp, &lh, LCK_PR);
 295         RETURN(rc);
 296 }
 297
 298 static int ost_statfs(struct ptlrpc_request *req)
 299 {
 300         struct obd_statfs *osfs;
 301         int rc;
 302         ENTRY;
 303
 304         rc = req_capsule_server_pack(&req->rq_pill);
 305         if (rc)
 306                 RETURN(rc);
 307
 308         osfs = req_capsule_server_get(&req->rq_pill, &RMF_OBD_STATFS);
 309
 310         req->rq_status = obd_statfs(req->rq_svc_thread->t_env, req->rq_export,
 311                                     osfs,
 312                                     cfs_time_shift_64(-OBD_STATFS_CACHE_SECONDS),
 313                                     0);
 314         if (req->rq_status != 0)
 315                 CERROR("ost: statfs failed: rc %d\n", req->rq_status);
 316
 317         RETURN(0);
 318 }
 319
 320 static int ost_create(struct obd_export *exp, struct ptlrpc_request *req,
 321                       struct obd_trans_info *oti)
 322 {
 323         struct ost_body *body, *repbody;
 324         int rc;
 325         ENTRY;
 326
 327         body = req_capsule_client_get(&req->rq_pill, &RMF_OST_BODY);
 328         if (body == NULL)
 329                 RETURN(-EFAULT);
 330
 331         rc = ost_validate_obdo(req->rq_export, &body->oa, NULL);
 332         if (rc)
 333                 RETURN(rc);
 334
 335         rc = req_capsule_server_pack(&req->rq_pill);
 336         if (rc)
 337                 RETURN(rc);
 338
 339         repbody = req_capsule_server_get(&req->rq_pill, &RMF_OST_BODY);
 340         repbody->oa = body->oa;
 341         oti->oti_logcookies = &body->oa.o_lcookie;
 342
 343         req->rq_status = obd_create(req->rq_svc_thread->t_env, exp,
 344                                     &repbody->oa, NULL, oti);
 345         //obd_log_cancel(conn, NULL, 1, oti->oti_logcookies, 0);
 346         RETURN(0);
 347 }
 348
 349 static int ost_punch(struct obd_export *exp, struct ptlrpc_request *req,
 350                      struct obd_trans_info *oti)
 351 {
 352         struct ost_body *body, *repbody;
 353         int rc, flags = 0;
 354         struct lustre_handle lh = {0,};
 355         ENTRY;
 356
 357         /* check that we do support OBD_CONNECT_TRUNCLOCK. */
 358         CLASSERT(OST_CONNECT_SUPPORTED & OBD_CONNECT_TRUNCLOCK);
 359
 360         body = req_capsule_client_get(&req->rq_pill, &RMF_OST_BODY);
 361         if (body == NULL)
 362                 RETURN(-EFAULT);
 363
 364         rc = ost_validate_obdo(exp, &body->oa, NULL);
 365         if (rc)
 366                 RETURN(rc);
 367
 368         if ((body->oa.o_valid & (OBD_MD_FLSIZE | OBD_MD_FLBLOCKS)) !=
 369             (OBD_MD_FLSIZE | OBD_MD_FLBLOCKS))
 370                 RETURN(-EPROTO);
 371
 372         rc = req_capsule_server_pack(&req->rq_pill);
 373         if (rc)
 374                 RETURN(rc);
 375
 376         /* standard truncate optimization: if file body is completely
 377          * destroyed, don't send data back to the server. */
 378         if (body->oa.o_size == 0)
 379                 flags |= LDLM_AST_DISCARD_DATA;
 380
 381         repbody = req_capsule_server_get(&req->rq_pill, &RMF_OST_BODY);
 382         repbody->oa = body->oa;
 383
 384         rc = ost_lock_get(exp, &repbody->oa, repbody->oa.o_size,
 385                           repbody->oa.o_blocks, &lh, LCK_PW, flags);
 386         if (rc == 0) {
 387                 struct obd_info *oinfo;
 388                 struct lustre_capa *capa = NULL;
 389
 390                 if (repbody->oa.o_valid & OBD_MD_FLFLAGS &&
 391                     repbody->oa.o_flags == OBD_FL_SRVLOCK)
 392                         /*
 393                          * If OBD_FL_SRVLOCK is the only bit set in
 394                          * ->o_flags, clear OBD_MD_FLFLAGS to avoid falling
 395                          * through filter_setattr() to filter_iocontrol().
 396                          */
 397                         repbody->oa.o_valid &= ~OBD_MD_FLFLAGS;
 398
 399                 if (repbody->oa.o_valid & OBD_MD_FLOSSCAPA) {
 400                         capa = req_capsule_client_get(&req->rq_pill,
 401                                                       &RMF_CAPA1);
 402                         if (capa == NULL) {
 403                                 CERROR("Missing capability for OST PUNCH");
 404                                 GOTO(unlock, rc = -EFAULT);
 405                         }
 406                 }
 407
 408                 OBD_ALLOC_PTR(oinfo);
 409                 if (!oinfo)
 410                         GOTO(unlock, rc = -ENOMEM);
 411                 oinfo->oi_oa = &repbody->oa;
 412                 oinfo->oi_policy.l_extent.start = oinfo->oi_oa->o_size;
 413                 oinfo->oi_policy.l_extent.end = oinfo->oi_oa->o_blocks;
 414                 oinfo->oi_capa = capa;
 415                 oinfo->oi_flags = OBD_FL_PUNCH;
 416
 417                 req->rq_status = obd_punch(req->rq_svc_thread->t_env, exp,
 418                                            oinfo, oti, NULL);
 419                 OBD_FREE_PTR(oinfo);
 420 unlock:
 421                 ost_lock_put(exp, &lh, LCK_PW);
 422         }
 423
 424         ost_drop_id(exp, &repbody->oa);
 425         RETURN(rc);
 426 }
 427
 428 static int ost_sync(struct obd_export *exp, struct ptlrpc_request *req,
 429                     struct obd_trans_info *oti)
 430 {
 431         struct ost_body *body, *repbody;
 432         struct obd_info *oinfo;
 433         struct lustre_capa *capa = NULL;
 434         int rc;
 435         ENTRY;
 436
 437         body = req_capsule_client_get(&req->rq_pill, &RMF_OST_BODY);
 438         if (body == NULL)
 439                 RETURN(-EFAULT);
 440
 441         rc = ost_validate_obdo(exp, &body->oa, NULL);
 442         if (rc)
 443                 RETURN(rc);
 444
 445         if (body->oa.o_valid & OBD_MD_FLOSSCAPA) {
 446                 capa = req_capsule_client_get(&req->rq_pill, &RMF_CAPA1);
 447                 if (capa == NULL) {
 448                         CERROR("Missing capability for OST SYNC");
 449                         RETURN (-EFAULT);
 450                 }
 451         }
 452
 453         rc = req_capsule_server_pack(&req->rq_pill);
 454         if (rc)
 455                 RETURN(rc);
 456
 457         repbody = req_capsule_server_get(&req->rq_pill, &RMF_OST_BODY);
 458         repbody->oa = body->oa;
 459
 460         OBD_ALLOC_PTR(oinfo);
 461         if (!oinfo)
 462                 RETURN(-ENOMEM);
 463
 464         oinfo->oi_oa = &repbody->oa;
 465         oinfo->oi_capa = capa;
 466         oinfo->oi_jobid = oti->oti_jobid;
 467         req->rq_status = obd_sync(req->rq_svc_thread->t_env, exp, oinfo,
 468                                   repbody->oa.o_size, repbody->oa.o_blocks,
 469                                   NULL);
 470         OBD_FREE_PTR(oinfo);
 471
 472         ost_drop_id(exp, &repbody->oa);
 473         RETURN(0);
 474 }
 475
 476 static int ost_setattr(struct obd_export *exp, struct ptlrpc_request *req,
 477                        struct obd_trans_info *oti)
 478 {
 479         struct ost_body *body, *repbody;
 480         struct obd_info *oinfo;
 481         struct lustre_capa *capa = NULL;
 482         int rc;
 483         ENTRY;
 484
 485         body = req_capsule_client_get(&req->rq_pill, &RMF_OST_BODY);
 486         if (body == NULL)
 487                 RETURN(-EFAULT);
 488
 489         rc = ost_validate_obdo(req->rq_export, &body->oa, NULL);
 490         if (rc)
 491                 RETURN(rc);
 492
 493         rc = req_capsule_server_pack(&req->rq_pill);
 494         if (rc)
 495                 RETURN(rc);
 496
 497         if (body->oa.o_valid & OBD_MD_FLOSSCAPA) {
 498                 capa = req_capsule_client_get(&req->rq_pill, &RMF_CAPA1);
 499                 if (capa == NULL) {
 500                         CERROR("Missing capability for OST SETATTR");
 501                         RETURN (-EFAULT);
 502                 }
 503         }
 504
 505         repbody = req_capsule_server_get(&req->rq_pill, &RMF_OST_BODY);
 506         repbody->oa = body->oa;
 507
 508         OBD_ALLOC_PTR(oinfo);
 509         if (!oinfo)
 510                 RETURN(-ENOMEM);
 511         oinfo->oi_oa = &repbody->oa;
 512         oinfo->oi_capa = capa;
 513
 514         req->rq_status = obd_setattr(req->rq_svc_thread->t_env, exp, oinfo,
 515                                      oti);
 516
 517         OBD_FREE_PTR(oinfo);
 518
 519         ost_drop_id(exp, &repbody->oa);
 520         RETURN(0);
 521 }
 522
 523 static __u32 ost_checksum_bulk(struct ptlrpc_bulk_desc *desc, int opc,
 524                                cksum_type_t cksum_type)
 525 {
 526         __u32 cksum;
 527         int i;
 528
 529         cksum = init_checksum(cksum_type);
 530         for (i = 0; i < desc->bd_iov_count; i++) {
 531                 struct page *page = desc->bd_iov[i].kiov_page;
 532                 int off = desc->bd_iov[i].kiov_offset & ~CFS_PAGE_MASK;
 533                 char *ptr = kmap(page) + off;
 534                 int len = desc->bd_iov[i].kiov_len;
 535
 536                 /* corrupt the data before we compute the checksum, to
 537                  * simulate a client->OST data error */
 538                 if (i == 0 && opc == OST_WRITE &&
 539                     OBD_FAIL_CHECK(OBD_FAIL_OST_CHECKSUM_RECEIVE))
 540                         memcpy(ptr, "bad3", min(4, len));
 541                 cksum = compute_checksum(cksum, ptr, len, cksum_type);
 542                 /* corrupt the data after we compute the checksum, to
 543                  * simulate an OST->client data error */
 544                 if (i == 0 && opc == OST_READ &&
 545                     OBD_FAIL_CHECK(OBD_FAIL_OST_CHECKSUM_SEND)) {
 546                         memcpy(ptr, "bad4", min(4, len));
 547                         /* nobody should use corrupted page again */
 548                         ClearPageUptodate(page);
 549                 }
 550                 kunmap(page);
 551         }
 552
 553         return fini_checksum(cksum, cksum_type);
 554 }
 555
 556 static int ost_brw_lock_get(int mode, struct obd_export *exp,
 557                             struct obd_ioobj *obj, struct niobuf_remote *nb,
 558                             struct lustre_handle *lh)
 559 {
 560         int flags                 = 0;
 561         int nrbufs                = obj->ioo_bufcnt;
 562         struct ldlm_res_id res_id;
 563         ldlm_policy_data_t policy;
 564         int i;
 565         ENTRY;
 566
 567         osc_build_res_name(obj->ioo_id, obj->ioo_seq, &res_id);
 568         LASSERT(mode == LCK_PR || mode == LCK_PW);
 569         LASSERT(!lustre_handle_is_used(lh));
 570
 571         if (nrbufs == 0 || !(nb[0].flags & OBD_BRW_SRVLOCK))
 572                 RETURN(0);
 573
 574         for (i = 1; i < nrbufs; i ++)
 575                 if ((nb[0].flags & OBD_BRW_SRVLOCK) !=
 576                     (nb[i].flags & OBD_BRW_SRVLOCK))
 577                         RETURN(-EFAULT);
 578
 579         policy.l_extent.start = nb[0].offset & CFS_PAGE_MASK;
 580         policy.l_extent.end   = (nb[nrbufs - 1].offset +
 581                                  nb[nrbufs - 1].len - 1) | ~CFS_PAGE_MASK;
 582
 583         RETURN(ldlm_cli_enqueue_local(exp->exp_obd->obd_namespace, &res_id,
 584                                       LDLM_EXTENT, &policy, mode, &flags,
 585                                       ldlm_blocking_ast, ldlm_completion_ast,
 586                                       ldlm_glimpse_ast, NULL, 0, NULL, lh));
 587 }
 588
 589 static void ost_brw_lock_put(int mode,
 590                              struct obd_ioobj *obj, struct niobuf_remote *niob,
 591                              struct lustre_handle *lh)
 592 {
 593         ENTRY;
 594         LASSERT(mode == LCK_PR || mode == LCK_PW);
 595         LASSERT((obj->ioo_bufcnt > 0 && (niob[0].flags & OBD_BRW_SRVLOCK)) ==
 596                 lustre_handle_is_used(lh));
 597         if (lustre_handle_is_used(lh))
 598                 ldlm_lock_decref(lh, mode);
 599         EXIT;
 600 }
 601
 602 /* Allocate thread local buffers if needed */
 603 static struct ost_thread_local_cache *ost_tls_get(struct ptlrpc_request *r)
 604 {
 605         struct ost_thread_local_cache *tls =
 606                 (struct ost_thread_local_cache *)(r->rq_svc_thread->t_data);
 607
 608         /* In normal mode of operation an I/O request is serviced only
 609          * by ll_ost_io threads each of them has own tls buffers allocated by
 610          * ost_thread_init().
 611          * During recovery, an I/O request may be queued until any of the ost
 612          * service threads process it. Not necessary it should be one of
 613          * ll_ost_io threads. In that case we dynamically allocating tls
 614          * buffers for the request service time. */
 615         if (unlikely(tls == NULL)) {
 616                 LASSERT(r->rq_export->exp_in_recovery);
 617                 OBD_ALLOC_PTR(tls);
 618                 if (tls != NULL) {
 619                         tls->temporary = 1;
 620                         r->rq_svc_thread->t_data = tls;
 621                 }
 622         }
 623         return  tls;
 624 }
 625
 626 /* Free thread local buffers if they were allocated only for servicing
 627  * this one request */
 628 static void ost_tls_put(struct ptlrpc_request *r)
 629 {
 630         struct ost_thread_local_cache *tls =
 631                 (struct ost_thread_local_cache *)(r->rq_svc_thread->t_data);
 632
 633         if (unlikely(tls->temporary)) {
 634                 OBD_FREE_PTR(tls);
 635                 r->rq_svc_thread->t_data = NULL;
 636         }
 637 }
 638
 639 static int ost_brw_read(struct ptlrpc_request *req, struct obd_trans_info *oti)
 640 {
 641         struct ptlrpc_bulk_desc *desc = NULL;
 642         struct obd_export *exp = req->rq_export;
 643         struct niobuf_remote *remote_nb;
 644         struct niobuf_local *local_nb;
 645         struct obd_ioobj *ioo;
 646         struct ost_body *body, *repbody;
 647         struct lustre_capa *capa = NULL;
 648         struct l_wait_info lwi;
 649         struct lustre_handle lockh = { 0 };
 650         int niocount, npages, nob = 0, rc, i;
 651         int no_reply = 0;
 652         struct ost_thread_local_cache *tls;
 653         ENTRY;
 654
 655         req->rq_bulk_read = 1;
 656
 657         if (OBD_FAIL_CHECK(OBD_FAIL_OST_BRW_READ_BULK))
 658                 GOTO(out, rc = -EIO);
 659
 660         OBD_FAIL_TIMEOUT(OBD_FAIL_OST_BRW_PAUSE_BULK, (obd_timeout + 1) / 4);
 661
 662         /* Check if there is eviction in progress, and if so, wait for it to
 663          * finish */
 664         if (unlikely(cfs_atomic_read(&exp->exp_obd->obd_evict_inprogress))) {
 665                 lwi = LWI_INTR(NULL, NULL); // We do not care how long it takes
 666                 rc = l_wait_event(exp->exp_obd->obd_evict_inprogress_waitq,
 667                         !cfs_atomic_read(&exp->exp_obd->obd_evict_inprogress),
 668                         &lwi);
 669         }
 670         if (exp->exp_failed)
 671                 GOTO(out, rc = -ENOTCONN);
 672
 673         /* ost_body, ioobj & noibuf_remote are verified and swabbed in
 674          * ost_rw_hpreq_check(). */
 675         body = req_capsule_client_get(&req->rq_pill, &RMF_OST_BODY);
 676         if (body == NULL)
 677                 GOTO(out, rc = -EFAULT);
 678
 679         /*
 680          * A req_capsule_X_get_array(pill, field, ptr_to_element_count) function
 681          * would be useful here and wherever we get &RMF_OBD_IOOBJ and
 682          * &RMF_NIOBUF_REMOTE.
 683          */
 684         ioo = req_capsule_client_get(&req->rq_pill, &RMF_OBD_IOOBJ);
 685         if (ioo == NULL)
 686                 GOTO(out, rc = -EFAULT);
 687
 688         rc = ost_validate_obdo(exp, &body->oa, ioo);
 689         if (rc)
 690                 RETURN(rc);
 691
 692         niocount = ioo->ioo_bufcnt;
 693         remote_nb = req_capsule_client_get(&req->rq_pill, &RMF_NIOBUF_REMOTE);
 694         if (remote_nb == NULL)
 695                 GOTO(out, rc = -EFAULT);
 696
 697         if (body->oa.o_valid & OBD_MD_FLOSSCAPA) {
 698                 capa = req_capsule_client_get(&req->rq_pill, &RMF_CAPA1);
 699                 if (capa == NULL) {
 700                         CERROR("Missing capability for OST BRW READ");
 701                         GOTO(out, rc = -EFAULT);
 702                 }
 703         }
 704
 705         rc = req_capsule_server_pack(&req->rq_pill);
 706         if (rc)
 707                 GOTO(out, rc);
 708
 709         tls = ost_tls_get(req);
 710         if (tls == NULL)
 711                 GOTO(out_bulk, rc = -ENOMEM);
 712         local_nb = tls->local;
 713
 714         rc = ost_brw_lock_get(LCK_PR, exp, ioo, remote_nb, &lockh);
 715         if (rc != 0)
 716                 GOTO(out_tls, rc);
 717
 718         /*
 719          * If getting the lock took more time than
 720          * client was willing to wait, drop it. b=11330
 721          */
 722         if (cfs_time_current_sec() > req->rq_deadline ||
 723             OBD_FAIL_CHECK(OBD_FAIL_OST_DROP_REQ)) {
 724                 no_reply = 1;
 725                 CERROR("Dropping timed-out read from %s because locking"
 726                        "object "LPX64" took %ld seconds (limit was %ld).\n",
 727                        libcfs_id2str(req->rq_peer), ioo->ioo_id,
 728                        cfs_time_current_sec() - req->rq_arrival_time.tv_sec,
 729                        req->rq_deadline - req->rq_arrival_time.tv_sec);
 730                 GOTO(out_lock, rc = -ETIMEDOUT);
 731         }
 732
 733         repbody = req_capsule_server_get(&req->rq_pill, &RMF_OST_BODY);
 734         memcpy(&repbody->oa, &body->oa, sizeof(repbody->oa));
 735
 736         npages = OST_THREAD_POOL_SIZE;
 737         rc = obd_preprw(req->rq_svc_thread->t_env, OBD_BRW_READ, exp,
 738                         &repbody->oa, 1, ioo, remote_nb, &npages, local_nb,
 739                         oti, capa);
 740         if (rc != 0)
 741                 GOTO(out_lock, rc);
 742
 743         desc = ptlrpc_prep_bulk_exp(req, npages,
 744                                      BULK_PUT_SOURCE, OST_BULK_PORTAL);
 745         if (desc == NULL)
 746                 GOTO(out_commitrw, rc = -ENOMEM);
 747
 748         nob = 0;
 749         for (i = 0; i < npages; i++) {
 750                 int page_rc = local_nb[i].rc;
 751
 752                 if (page_rc < 0) {              /* error */
 753                         rc = page_rc;
 754                         break;
 755                 }
 756
 757                 nob += page_rc;
 758                 if (page_rc != 0) {             /* some data! */
 759                         LASSERT (local_nb[i].page != NULL);
 760                         ptlrpc_prep_bulk_page(desc, local_nb[i].page,
 761                                               local_nb[i].offset & ~CFS_PAGE_MASK,
 762                                               page_rc);
 763                 }
 764
 765                 if (page_rc != local_nb[i].len) { /* short read */
 766                         /* All subsequent pages should be 0 */
 767                         while(++i < npages)
 768                                 LASSERT(local_nb[i].rc == 0);
 769                         break;
 770                 }
 771         }
 772
 773         if (body->oa.o_valid & OBD_MD_FLCKSUM) {
 774                 cksum_type_t cksum_type =
 775                         cksum_type_unpack(repbody->oa.o_valid & OBD_MD_FLFLAGS ?
 776                                           repbody->oa.o_flags : 0);
 777                 repbody->oa.o_flags = cksum_type_pack(cksum_type);
 778                 repbody->oa.o_valid = OBD_MD_FLCKSUM | OBD_MD_FLFLAGS;
 779                 repbody->oa.o_cksum = ost_checksum_bulk(desc, OST_READ,cksum_type);
 780                 CDEBUG(D_PAGE, "checksum at read origin: %x\n",
 781                        repbody->oa.o_cksum);
 782         } else {
 783                 repbody->oa.o_valid = 0;
 784         }
 785         /* We're finishing using body->oa as an input variable */
 786
 787         /* Check if client was evicted while we were doing i/o before touching
 788            network */
 789         if (rc == 0) {
 790                 if (likely(!CFS_FAIL_PRECHECK(OBD_FAIL_PTLRPC_CLIENT_BULK_CB2)))
 791                         rc = target_bulk_io(exp, desc, &lwi);
 792                 no_reply = rc != 0;
 793         }
 794
 795 out_commitrw:
 796         /* Must commit after prep above in all cases */
 797         rc = obd_commitrw(req->rq_svc_thread->t_env, OBD_BRW_READ, exp,
 798                           &repbody->oa, 1, ioo, remote_nb, npages, local_nb,
 799                           oti, rc);
 800
 801         if (rc == 0)
 802                 ost_drop_id(exp, &repbody->oa);
 803
 804 out_lock:
 805         ost_brw_lock_put(LCK_PR, ioo, remote_nb, &lockh);
 806 out_tls:
 807         ost_tls_put(req);
 808 out_bulk:
 809         if (desc && !CFS_FAIL_PRECHECK(OBD_FAIL_PTLRPC_CLIENT_BULK_CB2))
 810                 ptlrpc_free_bulk(desc);
 811 out:
 812         LASSERT(rc <= 0);
 813         if (rc == 0) {
 814                 req->rq_status = nob;
 815                 ptlrpc_lprocfs_brw(req, nob);
 816                 target_committed_to_req(req);
 817                 ptlrpc_reply(req);
 818         } else if (!no_reply) {
 819                 /* Only reply if there was no comms problem with bulk */
 820                 target_committed_to_req(req);
 821                 req->rq_status = rc;
 822                 ptlrpc_error(req);
 823         } else {
 824                 /* reply out callback would free */
 825                 ptlrpc_req_drop_rs(req);
 826                 LCONSOLE_WARN("%s: Bulk IO read error with %s (at %s), "
 827                               "client will retry: rc %d\n",
 828                               exp->exp_obd->obd_name,
 829                               obd_uuid2str(&exp->exp_client_uuid),
 830                               obd_export_nid2str(exp), rc);
 831         }
 832         /* send a bulk after reply to simulate a network delay or reordering
 833          * by a router */
 834         if (unlikely(CFS_FAIL_PRECHECK(OBD_FAIL_PTLRPC_CLIENT_BULK_CB2))) {
 835                 cfs_waitq_t              waitq;
 836                 struct l_wait_info       lwi1;
 837
 838                 CDEBUG(D_INFO, "reorder BULK\n");
 839                 cfs_waitq_init(&waitq);
 840
 841                 lwi1 = LWI_TIMEOUT_INTR(cfs_time_seconds(3), NULL, NULL, NULL);
 842                 l_wait_event(waitq, 0, &lwi1);
 843                 rc = target_bulk_io(exp, desc, &lwi);
 844                 ptlrpc_free_bulk(desc);
 845         }
 846
 847         RETURN(rc);
 848 }
 849
 850 static int ost_brw_write(struct ptlrpc_request *req, struct obd_trans_info *oti)
 851 {
 852         struct ptlrpc_bulk_desc *desc = NULL;
 853         struct obd_export       *exp = req->rq_export;
 854         struct niobuf_remote    *remote_nb;
 855         struct niobuf_local     *local_nb;
 856         struct obd_ioobj        *ioo;
 857         struct ost_body         *body, *repbody;
 858         struct l_wait_info       lwi;
 859         struct lustre_handle     lockh = {0};
 860         struct lustre_capa      *capa = NULL;
 861         __u32                   *rcs;
 862         int objcount, niocount, npages;
 863         int rc, i, j;
 864         obd_count                client_cksum = 0, server_cksum = 0;
 865         cksum_type_t             cksum_type = OBD_CKSUM_CRC32;
 866         int                      no_reply = 0, mmap = 0;
 867         __u32                    o_uid = 0, o_gid = 0;
 868         struct ost_thread_local_cache *tls;
 869         ENTRY;
 870
 871         req->rq_bulk_write = 1;
 872
 873         if (OBD_FAIL_CHECK(OBD_FAIL_OST_BRW_WRITE_BULK))
 874                 GOTO(out, rc = -EIO);
 875         if (OBD_FAIL_CHECK(OBD_FAIL_OST_BRW_WRITE_BULK2))
 876                 GOTO(out, rc = -EFAULT);
 877
 878         /* pause before transaction has been started */
 879         OBD_FAIL_TIMEOUT(OBD_FAIL_OST_BRW_PAUSE_BULK, (obd_timeout + 1) / 4);
 880
 881         /* ost_body, ioobj & noibuf_remote are verified and swabbed in
 882          * ost_rw_hpreq_check(). */
 883         body = req_capsule_client_get(&req->rq_pill, &RMF_OST_BODY);
 884         if (body == NULL)
 885                 GOTO(out, rc = -EFAULT);
 886
 887         objcount = req_capsule_get_size(&req->rq_pill, &RMF_OBD_IOOBJ,
 888                                         RCL_CLIENT) / sizeof(*ioo);
 889         ioo = req_capsule_client_get(&req->rq_pill, &RMF_OBD_IOOBJ);
 890         if (ioo == NULL)
 891                 GOTO(out, rc = -EFAULT);
 892
 893         rc = ost_validate_obdo(exp, &body->oa, ioo);
 894         if (rc)
 895                 RETURN(rc);
 896
 897         for (niocount = i = 0; i < objcount; i++)
 898                 niocount += ioo[i].ioo_bufcnt;
 899
 900         /*
 901          * It'd be nice to have a capsule function to indicate how many elements
 902          * there were in a buffer for an RMF that's declared to be an array.
 903          * It's easy enough to compute the number of elements here though.
 904          */
 905         remote_nb = req_capsule_client_get(&req->rq_pill, &RMF_NIOBUF_REMOTE);
 906         if (remote_nb == NULL || niocount != (req_capsule_get_size(&req->rq_pill,
 907             &RMF_NIOBUF_REMOTE, RCL_CLIENT) / sizeof(*remote_nb)))
 908                 GOTO(out, rc = -EFAULT);
 909
 910         if ((remote_nb[0].flags & OBD_BRW_MEMALLOC) &&
 911             (exp->exp_connection->c_peer.nid == exp->exp_connection->c_self))
 912                 cfs_memory_pressure_set();
 913
 914         if (body->oa.o_valid & OBD_MD_FLOSSCAPA) {
 915                 capa = req_capsule_client_get(&req->rq_pill, &RMF_CAPA1);
 916                 if (capa == NULL) {
 917                         CERROR("Missing capability for OST BRW WRITE");
 918                         GOTO(out, rc = -EFAULT);
 919                 }
 920         }
 921
 922         req_capsule_set_size(&req->rq_pill, &RMF_RCS, RCL_SERVER,
 923                              niocount * sizeof(*rcs));
 924         rc = req_capsule_server_pack(&req->rq_pill);
 925         if (rc != 0)
 926                 GOTO(out, rc);
 927         CFS_FAIL_TIMEOUT(OBD_FAIL_OST_BRW_PAUSE_PACK, cfs_fail_val);
 928         rcs = req_capsule_server_get(&req->rq_pill, &RMF_RCS);
 929
 930         tls = ost_tls_get(req);
 931         if (tls == NULL)
 932                 GOTO(out_bulk, rc = -ENOMEM);
 933         local_nb = tls->local;
 934
 935         rc = ost_brw_lock_get(LCK_PW, exp, ioo, remote_nb, &lockh);
 936         if (rc != 0)
 937                 GOTO(out_tls, rc);
 938
 939         /*
 940          * If getting the lock took more time than
 941          * client was willing to wait, drop it. b=11330
 942          */
 943         if (cfs_time_current_sec() > req->rq_deadline ||
 944             OBD_FAIL_CHECK(OBD_FAIL_OST_DROP_REQ)) {
 945                 no_reply = 1;
 946                 CERROR("Dropping timed-out write from %s because locking "
 947                        "object "LPX64" took %ld seconds (limit was %ld).\n",
 948                        libcfs_id2str(req->rq_peer), ioo->ioo_id,
 949                        cfs_time_current_sec() - req->rq_arrival_time.tv_sec,
 950                        req->rq_deadline - req->rq_arrival_time.tv_sec);
 951                 GOTO(out_lock, rc = -ETIMEDOUT);
 952         }
 953
 954         /* obd_preprw clobbers oa->valid, so save what we need */
 955         if (body->oa.o_valid & OBD_MD_FLCKSUM) {
 956                 client_cksum = body->oa.o_cksum;
 957                 if (body->oa.o_valid & OBD_MD_FLFLAGS)
 958                         cksum_type = cksum_type_unpack(body->oa.o_flags);
 959         }
 960         if (body->oa.o_valid & OBD_MD_FLFLAGS && body->oa.o_flags & OBD_FL_MMAP)
 961                 mmap = 1;
 962
 963         /* Because we already sync grant info with client when reconnect,
 964          * grant info will be cleared for resent req, then fed_grant and
 965          * total_grant will not be modified in following preprw_write */
 966         if (lustre_msg_get_flags(req->rq_reqmsg) & (MSG_RESENT | MSG_REPLAY)) {
 967                 DEBUG_REQ(D_CACHE, req, "clear resent/replay req grant info");
 968                 body->oa.o_valid &= ~OBD_MD_FLGRANT;
 969         }
 970
 971         if (exp_connect_rmtclient(exp)) {
 972                 o_uid = body->oa.o_uid;
 973                 o_gid = body->oa.o_gid;
 974         }
 975
 976         repbody = req_capsule_server_get(&req->rq_pill, &RMF_OST_BODY);
 977         memcpy(&repbody->oa, &body->oa, sizeof(repbody->oa));
 978
 979         npages = OST_THREAD_POOL_SIZE;
 980         rc = obd_preprw(req->rq_svc_thread->t_env, OBD_BRW_WRITE, exp,
 981                         &repbody->oa, objcount, ioo, remote_nb, &npages,
 982                         local_nb, oti, capa);
 983         if (rc != 0)
 984                 GOTO(out_lock, rc);
 985
 986         desc = ptlrpc_prep_bulk_exp(req, npages,
 987                                      BULK_GET_SINK, OST_BULK_PORTAL);
 988         if (desc == NULL)
 989                 GOTO(skip_transfer, rc = -ENOMEM);
 990
 991         /* NB Having prepped, we must commit... */
 992
 993         for (i = 0; i < npages; i++)
 994                 ptlrpc_prep_bulk_page(desc, local_nb[i].page,
 995                                       local_nb[i].offset & ~CFS_PAGE_MASK,
 996                                       local_nb[i].len);
 997
 998         rc = sptlrpc_svc_prep_bulk(req, desc);
 999         if (rc != 0)
1000                 GOTO(out_lock, rc);
1001
1002         rc = target_bulk_io(exp, desc, &lwi);
1003         no_reply = rc != 0;
1004
1005 skip_transfer:
1006         if (client_cksum != 0 && rc == 0) {
1007                 static int cksum_counter;
1008                 repbody->oa.o_valid |= OBD_MD_FLCKSUM | OBD_MD_FLFLAGS;
1009                 repbody->oa.o_flags &= ~OBD_FL_CKSUM_ALL;
1010                 repbody->oa.o_flags |= cksum_type_pack(cksum_type);
1011                 server_cksum = ost_checksum_bulk(desc, OST_WRITE, cksum_type);
1012                 repbody->oa.o_cksum = server_cksum;
1013                 cksum_counter++;
1014                 if (unlikely(client_cksum != server_cksum)) {
1015                         CDEBUG_LIMIT(mmap ? D_INFO : D_ERROR,
1016                                      "client csum %x, server csum %x\n",
1017                                      client_cksum, server_cksum);
1018                         cksum_counter = 0;
1019                 } else if ((cksum_counter & (-cksum_counter)) == cksum_counter){
1020                         CDEBUG(D_INFO, "Checksum %u from %s OK: %x\n",
1021                                cksum_counter, libcfs_id2str(req->rq_peer),
1022                                server_cksum);
1023                 }
1024         }
1025
1026         /* Must commit after prep above in all cases */
1027         rc = obd_commitrw(req->rq_svc_thread->t_env, OBD_BRW_WRITE, exp,
1028                           &repbody->oa, objcount, ioo, remote_nb, npages,
1029                           local_nb, oti, rc);
1030         if (rc == -ENOTCONN)
1031                 /* quota acquire process has been given up because
1032                  * either the client has been evicted or the client
1033                  * has timed out the request already */
1034                 no_reply = 1;
1035
1036         if (exp_connect_rmtclient(exp)) {
1037                 repbody->oa.o_uid = o_uid;
1038                 repbody->oa.o_gid = o_gid;
1039         }
1040
1041         /*
1042          * Disable sending mtime back to the client. If the client locked the
1043          * whole object, then it has already updated the mtime on its side,
1044          * otherwise it will have to glimpse anyway (see bug 21489, comment 32)
1045          */
1046         repbody->oa.o_valid &= ~(OBD_MD_FLMTIME | OBD_MD_FLATIME);
1047
1048         if (unlikely(client_cksum != server_cksum && rc == 0 && !mmap)) {
1049                 int  new_cksum = ost_checksum_bulk(desc, OST_WRITE, cksum_type);
1050                 char *msg;
1051                 char *via;
1052                 char *router;
1053
1054                 if (new_cksum == server_cksum)
1055                         msg = "changed in transit before arrival at OST";
1056                 else if (new_cksum == client_cksum)
1057                         msg = "initial checksum before message complete";
1058                 else
1059                         msg = "changed in transit AND after initial checksum";
1060
1061                 if (req->rq_peer.nid == desc->bd_sender) {
1062                         via = router = "";
1063                 } else {
1064                         via = " via ";
1065                         router = libcfs_nid2str(desc->bd_sender);
1066                 }
1067
1068                 LCONSOLE_ERROR_MSG(0x168, "%s: BAD WRITE CHECKSUM: %s from "
1069                                    "%s%s%s inode "DFID" object "
1070                                    LPU64"/"LPU64" extent ["LPU64"-"LPU64"]\n",
1071                                    exp->exp_obd->obd_name, msg,
1072                                    libcfs_id2str(req->rq_peer),
1073                                    via, router,
1074                                    body->oa.o_valid & OBD_MD_FLFID ?
1075                                                 body->oa.o_parent_seq : (__u64)0,
1076                                    body->oa.o_valid & OBD_MD_FLFID ?
1077                                                 body->oa.o_parent_oid : 0,
1078                                    body->oa.o_valid & OBD_MD_FLFID ?
1079                                                 body->oa.o_parent_ver : 0,
1080                                    body->oa.o_id,
1081                                    body->oa.o_valid & OBD_MD_FLGROUP ?
1082                                                 body->oa.o_seq : (__u64)0,
1083                                    local_nb[0].offset,
1084                                    local_nb[npages-1].offset +
1085                                    local_nb[npages-1].len - 1 );
1086                 CERROR("client csum %x, original server csum %x, "
1087                        "server csum now %x\n",
1088                        client_cksum, server_cksum, new_cksum);
1089         }
1090
1091         if (rc == 0) {
1092                 int nob = 0;
1093
1094                 /* set per-requested niobuf return codes */
1095                 for (i = j = 0; i < niocount; i++) {
1096                         int len = remote_nb[i].len;
1097
1098                         nob += len;
1099                         rcs[i] = 0;
1100                         do {
1101                                 LASSERT(j < npages);
1102                                 if (local_nb[j].rc < 0)
1103                                         rcs[i] = local_nb[j].rc;
1104                                 len -= local_nb[j].len;
1105                                 j++;
1106                         } while (len > 0);
1107                         LASSERT(len == 0);
1108                 }
1109                 LASSERT(j == npages);
1110                 ptlrpc_lprocfs_brw(req, nob);
1111         }
1112
1113 out_lock:
1114         ost_brw_lock_put(LCK_PW, ioo, remote_nb, &lockh);
1115 out_tls:
1116         ost_tls_put(req);
1117 out_bulk:
1118         if (desc)
1119                 ptlrpc_free_bulk(desc);
1120 out:
1121         if (rc == 0) {
1122                 oti_to_request(oti, req);
1123                 target_committed_to_req(req);
1124                 rc = ptlrpc_reply(req);
1125         } else if (!no_reply) {
1126                 /* Only reply if there was no comms problem with bulk */
1127                 target_committed_to_req(req);
1128                 req->rq_status = rc;
1129                 ptlrpc_error(req);
1130         } else {
1131                 /* reply out callback would free */
1132                 ptlrpc_req_drop_rs(req);
1133                 LCONSOLE_WARN("%s: Bulk IO write error with %s (at %s), "
1134                               "client will retry: rc %d\n",
1135                               exp->exp_obd->obd_name,
1136                               obd_uuid2str(&exp->exp_client_uuid),
1137                               obd_export_nid2str(exp), rc);
1138         }
1139         cfs_memory_pressure_clr();
1140         RETURN(rc);
1141 }
1142
1143 /**
1144  * Implementation of OST_SET_INFO.
1145  *
1146  * OST_SET_INFO is like ioctl(): heavily overloaded.  Specifically, it takes a
1147  * "key" and a value RPC buffers as arguments, with the value's contents
1148  * interpreted according to the key.
1149  *
1150  * Value types that need swabbing have swabbing done explicitly, either here or
1151  * in functions called from here.  This should be corrected: all swabbing should
1152  * be done in the capsule abstraction, as that will then allow us to move
1153  * swabbing exclusively to the client without having to modify server code
1154  * outside the capsule abstraction's implementation itself.  To correct this
1155  * will require minor changes to the capsule abstraction; see the comments for
1156  * req_capsule_extend() in layout.c.
1157  */
1158 static int ost_set_info(struct obd_export *exp, struct ptlrpc_request *req)
1159 {
1160         struct ost_body *body = NULL, *repbody;
1161         char *key, *val = NULL;
1162         int keylen, vallen, rc = 0;
1163         int is_grant_shrink = 0;
1164         ENTRY;
1165
1166         key = req_capsule_client_get(&req->rq_pill, &RMF_SETINFO_KEY);
1167         if (key == NULL) {
1168                 DEBUG_REQ(D_HA, req, "no set_info key");
1169                 RETURN(-EFAULT);
1170         }
1171         keylen = req_capsule_get_size(&req->rq_pill, &RMF_SETINFO_KEY,
1172                                       RCL_CLIENT);
1173
1174         vallen = req_capsule_get_size(&req->rq_pill, &RMF_SETINFO_VAL,
1175                                       RCL_CLIENT);
1176
1177         if ((is_grant_shrink = KEY_IS(KEY_GRANT_SHRINK)))
1178                 /* In this case the value is actually an RMF_OST_BODY, so we
1179                  * transmutate the type of this PTLRPC */
1180                 req_capsule_extend(&req->rq_pill, &RQF_OST_SET_GRANT_INFO);
1181
1182         rc = req_capsule_server_pack(&req->rq_pill);
1183         if (rc)
1184                 RETURN(rc);
1185
1186         if (vallen) {
1187                 if (is_grant_shrink) {
1188                         body = req_capsule_client_get(&req->rq_pill,
1189                                                       &RMF_OST_BODY);
1190                         if (!body)
1191                                 RETURN(-EFAULT);
1192
1193                         repbody = req_capsule_server_get(&req->rq_pill,
1194                                                          &RMF_OST_BODY);
1195                         memcpy(repbody, body, sizeof(*body));
1196                         val = (char*)repbody;
1197                 } else {
1198                         val = req_capsule_client_get(&req->rq_pill,
1199                                                      &RMF_SETINFO_VAL);
1200                 }
1201         }
1202
1203         if (KEY_IS(KEY_EVICT_BY_NID)) {
1204                 if (val && vallen)
1205                         obd_export_evict_by_nid(exp->exp_obd, val);
1206                 GOTO(out, rc = 0);
1207         } else if (KEY_IS(KEY_MDS_CONN) && ptlrpc_req_need_swab(req)) {
1208                 if (vallen < sizeof(__u32))
1209                         RETURN(-EFAULT);
1210                 __swab32s((__u32 *)val);
1211         }
1212
1213         /* OBD will also check if KEY_IS(KEY_GRANT_SHRINK), and will cast val to
1214          * a struct ost_body * value */
1215         rc = obd_set_info_async(req->rq_svc_thread->t_env, exp, keylen,
1216                                 key, vallen, val, NULL);
1217 out:
1218         lustre_msg_set_status(req->rq_repmsg, 0);
1219         RETURN(rc);
1220 }
1221
1222 static int ost_get_info(struct obd_export *exp, struct ptlrpc_request *req)
1223 {
1224         void *key, *reply;
1225         int keylen, replylen, rc = 0;
1226         struct req_capsule *pill = &req->rq_pill;
1227         ENTRY;
1228
1229         /* this common part for get_info rpc */
1230         key = req_capsule_client_get(pill, &RMF_SETINFO_KEY);
1231         if (key == NULL) {
1232                 DEBUG_REQ(D_HA, req, "no get_info key");
1233                 RETURN(-EFAULT);
1234         }
1235         keylen = req_capsule_get_size(pill, &RMF_SETINFO_KEY, RCL_CLIENT);
1236
1237         if (KEY_IS(KEY_FIEMAP)) {
1238                 struct ll_fiemap_info_key *fm_key = key;
1239                 int rc;
1240
1241                 rc = ost_validate_obdo(exp, &fm_key->oa, NULL);
1242                 if (rc)
1243                         RETURN(rc);
1244         }
1245
1246         rc = obd_get_info(req->rq_svc_thread->t_env, exp, keylen, key,
1247                           &replylen, NULL, NULL);
1248         if (rc)
1249                 RETURN(rc);
1250
1251         req_capsule_set_size(pill, &RMF_GENERIC_DATA,
1252                              RCL_SERVER, replylen);
1253
1254         rc = req_capsule_server_pack(pill);
1255         if (rc)
1256                 RETURN(rc);
1257
1258         reply = req_capsule_server_get(pill, &RMF_GENERIC_DATA);
1259         if (reply == NULL)
1260                 RETURN(-ENOMEM);
1261
1262         /* call again to fill in the reply buffer */
1263         rc = obd_get_info(req->rq_svc_thread->t_env, exp, keylen, key,
1264                           &replylen, reply, NULL);
1265
1266         lustre_msg_set_status(req->rq_repmsg, 0);
1267         RETURN(rc);
1268 }
1269
1270 #ifdef HAVE_QUOTA_SUPPORT
1271 static int ost_handle_quotactl(struct ptlrpc_request *req)
1272 {
1273         struct obd_quotactl *oqctl, *repoqc;
1274         int rc;
1275         ENTRY;
1276
1277         oqctl = req_capsule_client_get(&req->rq_pill, &RMF_OBD_QUOTACTL);
1278         if (oqctl == NULL)
1279                 GOTO(out, rc = -EPROTO);
1280
1281         rc = req_capsule_server_pack(&req->rq_pill);
1282         if (rc)
1283                 GOTO(out, rc);
1284
1285         repoqc = req_capsule_server_get(&req->rq_pill, &RMF_OBD_QUOTACTL);
1286         req->rq_status = obd_quotactl(req->rq_export, oqctl);
1287         *repoqc = *oqctl;
1288
1289 out:
1290         RETURN(rc);
1291 }
1292
1293 static int ost_handle_quotacheck(struct ptlrpc_request *req)
1294 {
1295         struct obd_quotactl *oqctl;
1296         int rc;
1297         ENTRY;
1298
1299         oqctl = req_capsule_client_get(&req->rq_pill, &RMF_OBD_QUOTACTL);
1300         if (oqctl == NULL)
1301                 RETURN(-EPROTO);
1302
1303         rc = req_capsule_server_pack(&req->rq_pill);
1304         if (rc)
1305                 RETURN(-ENOMEM);
1306
1307         req->rq_status = obd_quotacheck(req->rq_export, oqctl);
1308         RETURN(0);
1309 }
1310
1311 static int ost_handle_quota_adjust_qunit(struct ptlrpc_request *req)
1312 {
1313         struct quota_adjust_qunit *oqaq, *repoqa;
1314         struct lustre_quota_ctxt *qctxt;
1315         int rc;
1316         ENTRY;
1317
1318         qctxt = &req->rq_export->exp_obd->u.obt.obt_qctxt;
1319         oqaq = req_capsule_client_get(&req->rq_pill, &RMF_QUOTA_ADJUST_QUNIT);
1320         if (oqaq == NULL)
1321                 GOTO(out, rc = -EPROTO);
1322
1323         rc = req_capsule_server_pack(&req->rq_pill);
1324         if (rc)
1325                 GOTO(out, rc);
1326
1327         repoqa = req_capsule_server_get(&req->rq_pill, &RMF_QUOTA_ADJUST_QUNIT);
1328         req->rq_status = obd_quota_adjust_qunit(req->rq_export, oqaq, qctxt, NULL);
1329         *repoqa = *oqaq;
1330
1331  out:
1332         RETURN(rc);
1333 }
1334 #endif
1335
1336 static int ost_llog_handle_connect(struct obd_export *exp,
1337                                    struct ptlrpc_request *req)
1338 {
1339         struct llogd_conn_body *body;
1340         int rc;
1341         ENTRY;
1342
1343         body = req_capsule_client_get(&req->rq_pill, &RMF_LLOGD_CONN_BODY);
1344         rc = obd_llog_connect(exp, body);
1345         RETURN(rc);
1346 }
1347
1348 #define ost_init_sec_none(reply, exp)                                   \
1349 do {                                                                    \
1350         reply->ocd_connect_flags &= ~(OBD_CONNECT_RMT_CLIENT |          \
1351                                       OBD_CONNECT_RMT_CLIENT_FORCE |    \
1352                                       OBD_CONNECT_OSS_CAPA);            \
1353         cfs_spin_lock(&exp->exp_lock);                                  \
1354         exp->exp_connect_flags = reply->ocd_connect_flags;              \
1355         cfs_spin_unlock(&exp->exp_lock);                                \
1356 } while (0)
1357
1358 static int ost_init_sec_level(struct ptlrpc_request *req)
1359 {
1360         struct obd_export *exp = req->rq_export;
1361         struct req_capsule *pill = &req->rq_pill;
1362         struct obd_device *obd = exp->exp_obd;
1363         struct filter_obd *filter = &obd->u.filter;
1364         char *client = libcfs_nid2str(req->rq_peer.nid);
1365         struct obd_connect_data *data, *reply;
1366         int rc = 0, remote;
1367         ENTRY;
1368
1369         data = req_capsule_client_get(pill, &RMF_CONNECT_DATA);
1370         reply = req_capsule_server_get(pill, &RMF_CONNECT_DATA);
1371         if (data == NULL || reply == NULL)
1372                 RETURN(-EFAULT);
1373
1374         /* connection from MDT is always trusted */
1375         if (req->rq_auth_usr_mdt) {
1376                 ost_init_sec_none(reply, exp);
1377                 RETURN(0);
1378         }
1379
1380         /* no GSS support case */
1381         if (!req->rq_auth_gss) {
1382                 if (filter->fo_sec_level > LUSTRE_SEC_NONE) {
1383                         CWARN("client %s -> target %s does not user GSS, "
1384                               "can not run under security level %d.\n",
1385                               client, obd->obd_name, filter->fo_sec_level);
1386                         RETURN(-EACCES);
1387                 } else {
1388                         ost_init_sec_none(reply, exp);
1389                         RETURN(0);
1390                 }
1391         }
1392
1393         /* old version case */
1394         if (unlikely(!(data->ocd_connect_flags & OBD_CONNECT_RMT_CLIENT) ||
1395                      !(data->ocd_connect_flags & OBD_CONNECT_OSS_CAPA))) {
1396                 if (filter->fo_sec_level > LUSTRE_SEC_NONE) {
1397                         CWARN("client %s -> target %s uses old version, "
1398                               "can not run under security level %d.\n",
1399                               client, obd->obd_name, filter->fo_sec_level);
1400                         RETURN(-EACCES);
1401                 } else {
1402                         CWARN("client %s -> target %s uses old version, "
1403                               "run under security level %d.\n",
1404                               client, obd->obd_name, filter->fo_sec_level);
1405                         ost_init_sec_none(reply, exp);
1406                         RETURN(0);
1407                 }
1408         }
1409
1410         remote = data->ocd_connect_flags & OBD_CONNECT_RMT_CLIENT_FORCE;
1411         if (remote) {
1412                 if (!req->rq_auth_remote)
1413                         CDEBUG(D_SEC, "client (local realm) %s -> target %s "
1414                                "asked to be remote.\n", client, obd->obd_name);
1415         } else if (req->rq_auth_remote) {
1416                 remote = 1;
1417                 CDEBUG(D_SEC, "client (remote realm) %s -> target %s is set "
1418                        "as remote by default.\n", client, obd->obd_name);
1419         }
1420
1421         if (remote) {
1422                 if (!filter->fo_fl_oss_capa) {
1423                         CDEBUG(D_SEC, "client %s -> target %s is set as remote,"
1424                                " but OSS capabilities are not enabled: %d.\n",
1425                                client, obd->obd_name, filter->fo_fl_oss_capa);
1426                         RETURN(-EACCES);
1427                 }
1428         }
1429
1430         switch (filter->fo_sec_level) {
1431         case LUSTRE_SEC_NONE:
1432                 if (!remote) {
1433                         ost_init_sec_none(reply, exp);
1434                         break;
1435                 } else {
1436                         CDEBUG(D_SEC, "client %s -> target %s is set as remote, "
1437                                "can not run under security level %d.\n",
1438                                client, obd->obd_name, filter->fo_sec_level);
1439                         RETURN(-EACCES);
1440                 }
1441         case LUSTRE_SEC_REMOTE:
1442                 if (!remote)
1443                         ost_init_sec_none(reply, exp);
1444                 break;
1445         case LUSTRE_SEC_ALL:
1446                 if (!remote) {
1447                         reply->ocd_connect_flags &= ~(OBD_CONNECT_RMT_CLIENT |
1448                                                       OBD_CONNECT_RMT_CLIENT_FORCE);
1449                         if (!filter->fo_fl_oss_capa)
1450                                 reply->ocd_connect_flags &= ~OBD_CONNECT_OSS_CAPA;
1451
1452                         cfs_spin_lock(&exp->exp_lock);
1453                         exp->exp_connect_flags = reply->ocd_connect_flags;
1454                         cfs_spin_unlock(&exp->exp_lock);
1455                 }
1456                 break;
1457         default:
1458                 RETURN(-EINVAL);
1459         }
1460
1461         RETURN(rc);
1462 }
1463
1464 /*
1465  * FIXME
1466  * this should be done in filter_connect()/filter_reconnect(), but
1467  * we can't obtain information like NID, which stored in incoming
1468  * request, thus can't decide what flavor to use. so we do it here.
1469  *
1470  * This hack should be removed after the OST stack be rewritten, just
1471  * like what we are doing in mdt_obd_connect()/mdt_obd_reconnect().
1472  */
1473 static int ost_connect_check_sptlrpc(struct ptlrpc_request *req)
1474 {
1475         struct obd_export     *exp = req->rq_export;
1476         struct filter_obd     *filter = &exp->exp_obd->u.filter;
1477         struct sptlrpc_flavor  flvr;
1478         int                    rc = 0;
1479
1480         if (unlikely(strcmp(exp->exp_obd->obd_type->typ_name,
1481                             LUSTRE_ECHO_NAME) == 0)) {
1482                 exp->exp_flvr.sf_rpc = SPTLRPC_FLVR_ANY;
1483                 return 0;
1484         }
1485
1486         if (exp->exp_flvr.sf_rpc == SPTLRPC_FLVR_INVALID) {
1487                 cfs_read_lock(&filter->fo_sptlrpc_lock);
1488                 sptlrpc_target_choose_flavor(&filter->fo_sptlrpc_rset,
1489                                              req->rq_sp_from,
1490                                              req->rq_peer.nid,
1491                                              &flvr);
1492                 cfs_read_unlock(&filter->fo_sptlrpc_lock);
1493
1494                 cfs_spin_lock(&exp->exp_lock);
1495
1496                 exp->exp_sp_peer = req->rq_sp_from;
1497                 exp->exp_flvr = flvr;
1498
1499                 if (exp->exp_flvr.sf_rpc != SPTLRPC_FLVR_ANY &&
1500                     exp->exp_flvr.sf_rpc != req->rq_flvr.sf_rpc) {
1501                         CERROR("unauthorized rpc flavor %x from %s, "
1502                                "expect %x\n", req->rq_flvr.sf_rpc,
1503                                libcfs_nid2str(req->rq_peer.nid),
1504                                exp->exp_flvr.sf_rpc);
1505                         rc = -EACCES;
1506                 }
1507
1508                 cfs_spin_unlock(&exp->exp_lock);
1509         } else {
1510                 if (exp->exp_sp_peer != req->rq_sp_from) {
1511                         CERROR("RPC source %s doesn't match %s\n",
1512                                sptlrpc_part2name(req->rq_sp_from),
1513                                sptlrpc_part2name(exp->exp_sp_peer));
1514                         rc = -EACCES;
1515                 } else {
1516                         rc = sptlrpc_target_export_check(exp, req);
1517                 }
1518         }
1519
1520         return rc;
1521 }
1522
1523 /* Ensure that data and metadata are synced to the disk when lock is cancelled
1524  * (if requested) */
1525 int ost_blocking_ast(struct ldlm_lock *lock,
1526                              struct ldlm_lock_desc *desc,
1527                              void *data, int flag)
1528 {
1529         __u32 sync_lock_cancel = 0;
1530         __u32 len = sizeof(sync_lock_cancel);
1531         int rc = 0;
1532         ENTRY;
1533
1534         rc = obd_get_info(NULL, lock->l_export, sizeof(KEY_SYNC_LOCK_CANCEL),
1535                           KEY_SYNC_LOCK_CANCEL, &len, &sync_lock_cancel, NULL);
1536
1537         if (!rc && flag == LDLM_CB_CANCELING &&
1538             (lock->l_granted_mode & (LCK_PW|LCK_GROUP)) &&
1539             (sync_lock_cancel == ALWAYS_SYNC_ON_CANCEL ||
1540              (sync_lock_cancel == BLOCKING_SYNC_ON_CANCEL &&
1541               lock->l_flags & LDLM_FL_CBPENDING))) {
1542                 struct obd_info *oinfo;
1543                 struct obdo *oa;
1544                 int rc;
1545
1546                 OBD_ALLOC_PTR(oinfo);
1547                 if (!oinfo)
1548                         RETURN(-ENOMEM);
1549                 OBDO_ALLOC(oa);
1550                 if (!oa) {
1551                         OBD_FREE_PTR(oinfo);
1552                         RETURN(-ENOMEM);
1553                 }
1554                 oa->o_id = lock->l_resource->lr_name.name[0];
1555                 oa->o_seq = lock->l_resource->lr_name.name[1];
1556                 oa->o_valid = OBD_MD_FLID|OBD_MD_FLGROUP;
1557                 oinfo->oi_oa = oa;
1558
1559                 rc = obd_sync(NULL, lock->l_export, oinfo,
1560                               lock->l_policy_data.l_extent.start,
1561                               lock->l_policy_data.l_extent.end, NULL);
1562                 if (rc)
1563                         CERROR("Error %d syncing data on lock cancel\n", rc);
1564
1565                 OBDO_FREE(oa);
1566                 OBD_FREE_PTR(oinfo);
1567         }
1568
1569         rc = ldlm_server_blocking_ast(lock, desc, data, flag);
1570         RETURN(rc);
1571 }
1572
1573 static int ost_filter_recovery_request(struct ptlrpc_request *req,
1574                                        struct obd_device *obd, int *process)
1575 {
1576         switch (lustre_msg_get_opc(req->rq_reqmsg)) {
1577         case OST_CONNECT: /* This will never get here, but for completeness. */
1578         case OST_DISCONNECT:
1579                *process = 1;
1580                RETURN(0);
1581
1582         case OBD_PING:
1583         case OST_CREATE:
1584         case OST_DESTROY:
1585         case OST_PUNCH:
1586         case OST_SETATTR:
1587         case OST_SYNC:
1588         case OST_WRITE:
1589         case OBD_LOG_CANCEL:
1590         case LDLM_ENQUEUE:
1591                 *process = target_queue_recovery_request(req, obd);
1592                 RETURN(0);
1593
1594         default:
1595                 DEBUG_REQ(D_WARNING, req, "not permitted during recovery");
1596                 *process = -EAGAIN;
1597                 RETURN(0);
1598         }
1599 }
1600
1601 int ost_msg_check_version(struct lustre_msg *msg)
1602 {
1603         int rc;
1604
1605         switch(lustre_msg_get_opc(msg)) {
1606         case OST_CONNECT:
1607         case OST_DISCONNECT:
1608         case OBD_PING:
1609         case SEC_CTX_INIT:
1610         case SEC_CTX_INIT_CONT:
1611         case SEC_CTX_FINI:
1612                 rc = lustre_msg_check_version(msg, LUSTRE_OBD_VERSION);
1613                 if (rc)
1614                         CERROR("bad opc %u version %08x, expecting %08x\n",
1615                                lustre_msg_get_opc(msg),
1616                                lustre_msg_get_version(msg),
1617                                LUSTRE_OBD_VERSION);
1618                 break;
1619         case OST_CREATE:
1620         case OST_DESTROY:
1621         case OST_GETATTR:
1622         case OST_SETATTR:
1623         case OST_WRITE:
1624         case OST_READ:
1625         case OST_PUNCH:
1626         case OST_STATFS:
1627         case OST_SYNC:
1628         case OST_SET_INFO:
1629         case OST_GET_INFO:
1630 #ifdef HAVE_QUOTA_SUPPORT
1631         case OST_QUOTACHECK:
1632         case OST_QUOTACTL:
1633         case OST_QUOTA_ADJUST_QUNIT:
1634 #endif
1635                 rc = lustre_msg_check_version(msg, LUSTRE_OST_VERSION);
1636                 if (rc)
1637                         CERROR("bad opc %u version %08x, expecting %08x\n",
1638                                lustre_msg_get_opc(msg),
1639                                lustre_msg_get_version(msg),
1640                                LUSTRE_OST_VERSION);
1641                 break;
1642         case LDLM_ENQUEUE:
1643         case LDLM_CONVERT:
1644         case LDLM_CANCEL:
1645         case LDLM_BL_CALLBACK:
1646         case LDLM_CP_CALLBACK:
1647                 rc = lustre_msg_check_version(msg, LUSTRE_DLM_VERSION);
1648                 if (rc)
1649                         CERROR("bad opc %u version %08x, expecting %08x\n",
1650                                lustre_msg_get_opc(msg),
1651                                lustre_msg_get_version(msg),
1652                                LUSTRE_DLM_VERSION);
1653                 break;
1654         case LLOG_ORIGIN_CONNECT:
1655         case OBD_LOG_CANCEL:
1656                 rc = lustre_msg_check_version(msg, LUSTRE_LOG_VERSION);
1657                 if (rc)
1658                         CERROR("bad opc %u version %08x, expecting %08x\n",
1659                                lustre_msg_get_opc(msg),
1660                                lustre_msg_get_version(msg),
1661                                LUSTRE_LOG_VERSION);
1662                 break;
1663         default:
1664                 CERROR("Unexpected opcode %d\n", lustre_msg_get_opc(msg));
1665                 rc = -ENOTSUPP;
1666         }
1667         return rc;
1668 }
1669
1670 struct ost_prolong_data {
1671         struct ptlrpc_request *opd_req;
1672         struct obd_export     *opd_exp;
1673         struct obdo           *opd_oa;
1674         struct ldlm_res_id     opd_resid;
1675         struct ldlm_extent     opd_extent;
1676         ldlm_mode_t            opd_mode;
1677         unsigned int           opd_locks;
1678         int                    opd_timeout;
1679 };
1680
1681 /* prolong locks for the current service time of the corresponding
1682  * portal (= OST_IO_PORTAL)
1683  */
1684 static inline int prolong_timeout(struct ptlrpc_request *req)
1685 {
1686         struct ptlrpc_service *svc = req->rq_rqbd->rqbd_service;
1687
1688         if (AT_OFF)
1689                 return obd_timeout / 2;
1690
1691         return max(at_est2timeout(at_get(&svc->srv_at_estimate)), ldlm_timeout);
1692 }
1693
1694 static void ost_prolong_lock_one(struct ost_prolong_data *opd,
1695                                  struct ldlm_lock *lock)
1696 {
1697         LASSERT(lock->l_req_mode == lock->l_granted_mode);
1698         LASSERT(lock->l_export == opd->opd_exp);
1699
1700         /* XXX: never try to grab resource lock here because we're inside
1701          * exp_bl_list_lock; in ldlm_lockd.c to handle waiting list we take
1702          * res lock and then exp_bl_list_lock. */
1703
1704         if (!(lock->l_flags & LDLM_FL_AST_SENT))
1705                 /* ignore locks not being cancelled */
1706                 return;
1707
1708         LDLM_DEBUG(lock,
1709                    "refreshed for req x"LPU64" ext("LPU64"->"LPU64") to %ds.\n",
1710                    opd->opd_req->rq_xid, opd->opd_extent.start,
1711                    opd->opd_extent.end, opd->opd_timeout);
1712
1713         /* OK. this is a possible lock the user holds doing I/O
1714          * let's refresh eviction timer for it */
1715         ldlm_refresh_waiting_lock(lock, opd->opd_timeout);
1716         ++opd->opd_locks;
1717 }
1718
1719 static void ost_prolong_locks(struct ost_prolong_data *data)
1720 {
1721         struct obd_export *exp = data->opd_exp;
1722         struct obdo       *oa  = data->opd_oa;
1723         struct ldlm_lock  *lock;
1724         ENTRY;
1725
1726         if (oa->o_valid & OBD_MD_FLHANDLE) {
1727                 /* mostly a request should be covered by only one lock, try
1728                  * fast path. */
1729                 lock = ldlm_handle2lock(&oa->o_handle);
1730                 if (lock != NULL) {
1731                         /* Fast path to check if the lock covers the whole IO
1732                          * region exclusively. */
1733                         if (lock->l_granted_mode == LCK_PW &&
1734                             ldlm_extent_contain(&lock->l_policy_data.l_extent,
1735                                                 &data->opd_extent)) {
1736                                 /* bingo */
1737                                 ost_prolong_lock_one(data, lock);
1738                                 LDLM_LOCK_PUT(lock);
1739                                 RETURN_EXIT;
1740                         }
1741                         LDLM_LOCK_PUT(lock);
1742                 }
1743         }
1744
1745
1746         cfs_spin_lock_bh(&exp->exp_bl_list_lock);
1747         cfs_list_for_each_entry(lock, &exp->exp_bl_list, l_exp_list) {
1748                 LASSERT(lock->l_flags & LDLM_FL_AST_SENT);
1749                 LASSERT(lock->l_resource->lr_type == LDLM_EXTENT);
1750
1751                 if (!ldlm_res_eq(&data->opd_resid, &lock->l_resource->lr_name))
1752                         continue;
1753
1754                 if (!ldlm_extent_overlap(&lock->l_policy_data.l_extent,
1755                                          &data->opd_extent))
1756                         continue;
1757
1758                 ost_prolong_lock_one(data, lock);
1759         }
1760         cfs_spin_unlock_bh(&exp->exp_bl_list_lock);
1761
1762         EXIT;
1763 }
1764
1765 /**
1766  * Returns 1 if the given PTLRPC matches the given LDLM locks, or 0 if it does
1767  * not.
1768  */
1769 static int ost_rw_hpreq_lock_match(struct ptlrpc_request *req,
1770                                    struct ldlm_lock *lock)
1771 {
1772         struct niobuf_remote *nb;
1773         struct obd_ioobj *ioo;
1774         int mode, opc;
1775         struct ldlm_extent ext;
1776         ENTRY;
1777
1778         opc = lustre_msg_get_opc(req->rq_reqmsg);
1779         LASSERT(opc == OST_READ || opc == OST_WRITE);
1780
1781         ioo = req_capsule_client_get(&req->rq_pill, &RMF_OBD_IOOBJ);
1782         LASSERT(ioo != NULL);
1783
1784         nb = req_capsule_client_get(&req->rq_pill, &RMF_NIOBUF_REMOTE);
1785         LASSERT(nb != NULL);
1786
1787         ext.start = nb->offset;
1788         nb += ioo->ioo_bufcnt - 1;
1789         ext.end = nb->offset + nb->len - 1;
1790
1791         LASSERT(lock->l_resource != NULL);
1792         if (!osc_res_name_eq(ioo->ioo_id, ioo->ioo_seq,
1793                              &lock->l_resource->lr_name))
1794                 RETURN(0);
1795
1796         mode = LCK_PW;
1797         if (opc == OST_READ)
1798                 mode |= LCK_PR;
1799         if (!(lock->l_granted_mode & mode))
1800                 RETURN(0);
1801
1802         RETURN(ldlm_extent_overlap(&lock->l_policy_data.l_extent, &ext));
1803 }
1804
1805 /**
1806  * High-priority queue request check for whether the given PTLRPC request (\a
1807  * req) is blocking an LDLM lock cancel.
1808  *
1809  * Returns 1 if the given given PTLRPC request (\a req) is blocking an LDLM lock
1810  * cancel, 0 if it is not, and -EFAULT if the request is malformed.
1811  *
1812  * Only OST_READs, OST_WRITEs and OST_PUNCHes go on the h-p RPC queue.  This
1813  * function looks only at OST_READs and OST_WRITEs.
1814  */
1815 static int ost_rw_hpreq_check(struct ptlrpc_request *req)
1816 {
1817         struct obd_device *obd = req->rq_export->exp_obd;
1818         struct ost_body *body;
1819         struct obd_ioobj *ioo;
1820         struct niobuf_remote *nb;
1821         struct ost_prolong_data opd = { 0 };
1822         int mode, opc;
1823         ENTRY;
1824
1825         /*
1826          * Use LASSERT to do sanity check because malformed RPCs should have
1827          * been filtered out in ost_hpreq_handler().
1828          */
1829         opc = lustre_msg_get_opc(req->rq_reqmsg);
1830         LASSERT(opc == OST_READ || opc == OST_WRITE);
1831
1832         body = req_capsule_client_get(&req->rq_pill, &RMF_OST_BODY);
1833         LASSERT(body != NULL);
1834
1835         ioo = req_capsule_client_get(&req->rq_pill, &RMF_OBD_IOOBJ);
1836         LASSERT(ioo != NULL);
1837
1838         nb = req_capsule_client_get(&req->rq_pill, &RMF_NIOBUF_REMOTE);
1839         LASSERT(nb != NULL);
1840         LASSERT(!(nb->flags & OBD_BRW_SRVLOCK));
1841
1842         osc_build_res_name(ioo->ioo_id, ioo->ioo_seq, &opd.opd_resid);
1843
1844         opd.opd_req = req;
1845         mode = LCK_PW;
1846         if (opc == OST_READ)
1847                 mode |= LCK_PR;
1848         opd.opd_mode = mode;
1849         opd.opd_exp = req->rq_export;
1850         opd.opd_oa  = &body->oa;
1851         opd.opd_extent.start = nb->offset;
1852         nb += ioo->ioo_bufcnt - 1;
1853         opd.opd_extent.end = nb->offset + nb->len - 1;
1854         opd.opd_timeout = prolong_timeout(req);
1855
1856         DEBUG_REQ(D_RPCTRACE, req,
1857                "%s %s: refresh rw locks: " LPU64"/"LPU64" ("LPU64"->"LPU64")\n",
1858                obd->obd_name, cfs_current()->comm,
1859                opd.opd_resid.name[0], opd.opd_resid.name[1],
1860                opd.opd_extent.start, opd.opd_extent.end);
1861
1862         ost_prolong_locks(&opd);
1863
1864         CDEBUG(D_DLMTRACE, "%s: refreshed %u locks timeout for req %p.\n",
1865                obd->obd_name, opd.opd_locks, req);
1866
1867         RETURN(opd.opd_locks);
1868 }
1869
1870 static void ost_rw_hpreq_fini(struct ptlrpc_request *req)
1871 {
1872         (void)ost_rw_hpreq_check(req);
1873 }
1874
1875 /**
1876  * Like ost_rw_hpreq_lock_match(), but for OST_PUNCH RPCs.
1877  */
1878 static int ost_punch_hpreq_lock_match(struct ptlrpc_request *req,
1879                                       struct ldlm_lock *lock)
1880 {
1881         struct ost_body *body;
1882         ENTRY;
1883
1884         body = req_capsule_client_get(&req->rq_pill, &RMF_OST_BODY);
1885         LASSERT(body != NULL);
1886
1887         if (body->oa.o_valid & OBD_MD_FLHANDLE &&
1888             body->oa.o_handle.cookie == lock->l_handle.h_cookie)
1889                 RETURN(1);
1890
1891         RETURN(0);
1892 }
1893
1894 /**
1895  * Like ost_rw_hpreq_check(), but for OST_PUNCH RPCs.
1896  */
1897 static int ost_punch_hpreq_check(struct ptlrpc_request *req)
1898 {
1899         struct obd_device *obd = req->rq_export->exp_obd;
1900         struct ost_body *body;
1901         struct obdo *oa;
1902         struct ost_prolong_data opd = { 0 };
1903         __u64 start, end;
1904         ENTRY;
1905
1906         body = req_capsule_client_get(&req->rq_pill, &RMF_OST_BODY);
1907         LASSERT(body != NULL);
1908
1909         oa = &body->oa;
1910         LASSERT(!(oa->o_valid & OBD_MD_FLFLAGS) ||
1911                 !(oa->o_flags & OBD_FL_SRVLOCK));
1912
1913         start = oa->o_size;
1914         end = start + oa->o_blocks;
1915
1916         opd.opd_req = req;
1917         opd.opd_mode = LCK_PW;
1918         opd.opd_exp = req->rq_export;
1919         opd.opd_oa  = oa;
1920         opd.opd_extent.start = start;
1921         opd.opd_extent.end   = end;
1922         if (oa->o_blocks == OBD_OBJECT_EOF)
1923                 opd.opd_extent.end = OBD_OBJECT_EOF;
1924         opd.opd_timeout = prolong_timeout(req);
1925
1926         osc_build_res_name(oa->o_id, oa->o_seq, &opd.opd_resid);
1927
1928         CDEBUG(D_DLMTRACE,
1929                "%s: refresh locks: "LPU64"/"LPU64" ("LPU64"->"LPU64")\n",
1930                obd->obd_name,
1931                opd.opd_resid.name[0], opd.opd_resid.name[1],
1932                opd.opd_extent.start, opd.opd_extent.end);
1933
1934         ost_prolong_locks(&opd);
1935
1936         CDEBUG(D_DLMTRACE, "%s: refreshed %u locks timeout for req %p.\n",
1937                obd->obd_name, opd.opd_locks, req);
1938
1939         RETURN(opd.opd_locks > 0);
1940 }
1941
1942 static void ost_punch_hpreq_fini(struct ptlrpc_request *req)
1943 {
1944         (void)ost_punch_hpreq_check(req);
1945 }
1946
1947 struct ptlrpc_hpreq_ops ost_hpreq_rw = {
1948         .hpreq_lock_match = ost_rw_hpreq_lock_match,
1949         .hpreq_check      = ost_rw_hpreq_check,
1950         .hpreq_fini       = ost_rw_hpreq_fini
1951 };
1952
1953 struct ptlrpc_hpreq_ops ost_hpreq_punch = {
1954         .hpreq_lock_match = ost_punch_hpreq_lock_match,
1955         .hpreq_check      = ost_punch_hpreq_check,
1956         .hpreq_fini       = ost_punch_hpreq_fini
1957 };
1958
1959 /** Assign high priority operations to the request if needed. */
1960 static int ost_hpreq_handler(struct ptlrpc_request *req)
1961 {
1962         ENTRY;
1963         if (req->rq_export) {
1964                 int opc = lustre_msg_get_opc(req->rq_reqmsg);
1965                 struct ost_body *body;
1966
1967                 if (opc == OST_READ || opc == OST_WRITE) {
1968                         struct niobuf_remote *nb;
1969                         struct obd_ioobj *ioo;
1970                         int objcount, niocount;
1971                         int rc;
1972                         int i;
1973
1974                         /* RPCs on the H-P queue can be inspected before
1975                          * ost_handler() initializes their pills, so we
1976                          * initialize that here.  Capsule initialization is
1977                          * idempotent, as is setting the pill's format (provided
1978                          * it doesn't change).
1979                          */
1980                         req_capsule_init(&req->rq_pill, req, RCL_SERVER);
1981                         if (opc == OST_READ)
1982                                 req_capsule_set(&req->rq_pill,
1983                                                 &RQF_OST_BRW_READ);
1984                         else
1985                                 req_capsule_set(&req->rq_pill,
1986                                                 &RQF_OST_BRW_WRITE);
1987
1988                         body = req_capsule_client_get(&req->rq_pill,
1989                                                       &RMF_OST_BODY);
1990                         if (body == NULL) {
1991                                 CERROR("Missing/short ost_body\n");
1992                                 RETURN(-EFAULT);
1993                         }
1994
1995                         objcount = req_capsule_get_size(&req->rq_pill,
1996                                                         &RMF_OBD_IOOBJ,
1997                                                         RCL_CLIENT) /
1998                                                         sizeof(*ioo);
1999                         if (objcount == 0) {
2000                                 CERROR("Missing/short ioobj\n");
2001                                 RETURN(-EFAULT);
2002                         }
2003                         if (objcount > 1) {
2004                                 CERROR("too many ioobjs (%d)\n", objcount);
2005                                 RETURN(-EFAULT);
2006                         }
2007
2008                         ioo = req_capsule_client_get(&req->rq_pill,
2009                                                      &RMF_OBD_IOOBJ);
2010                         if (ioo == NULL) {
2011                                 CERROR("Missing/short ioobj\n");
2012                                 RETURN(-EFAULT);
2013                         }
2014
2015                         rc = ost_validate_obdo(req->rq_export, &body->oa, ioo);
2016                         if (rc) {
2017                                 CERROR("invalid object ids\n");
2018                                 RETURN(rc);
2019                         }
2020
2021                         for (niocount = i = 0; i < objcount; i++) {
2022                                 if (ioo[i].ioo_bufcnt == 0) {
2023                                         CERROR("ioo[%d] has zero bufcnt\n", i);
2024                                         RETURN(-EFAULT);
2025                                 }
2026                                 niocount += ioo[i].ioo_bufcnt;
2027                         }
2028                         if (niocount > PTLRPC_MAX_BRW_PAGES) {
2029                                 DEBUG_REQ(D_RPCTRACE, req,
2030                                           "bulk has too many pages (%d)",
2031                                           niocount);
2032                                 RETURN(-EFAULT);
2033                         }
2034
2035                         nb = req_capsule_client_get(&req->rq_pill,
2036                                                     &RMF_NIOBUF_REMOTE);
2037                         if (nb == NULL) {
2038                                 CERROR("Missing/short niobuf\n");
2039                                 RETURN(-EFAULT);
2040                         }
2041
2042                         if (niocount == 0 || !(nb[0].flags & OBD_BRW_SRVLOCK))
2043                                 req->rq_ops = &ost_hpreq_rw;
2044                 } else if (opc == OST_PUNCH) {
2045                         req_capsule_init(&req->rq_pill, req, RCL_SERVER);
2046                         req_capsule_set(&req->rq_pill, &RQF_OST_PUNCH);
2047
2048                         body = req_capsule_client_get(&req->rq_pill,
2049                                                       &RMF_OST_BODY);
2050                         if (body == NULL) {
2051                                 CERROR("Missing/short ost_body\n");
2052                                 RETURN(-EFAULT);
2053                         }
2054
2055                         if (!(body->oa.o_valid & OBD_MD_FLFLAGS) ||
2056                             !(body->oa.o_flags & OBD_FL_SRVLOCK))
2057                                 req->rq_ops = &ost_hpreq_punch;
2058                 }
2059         }
2060         RETURN(0);
2061 }
2062
2063 /* TODO: handle requests in a similar way as MDT: see mdt_handle_common() */
2064 int ost_handle(struct ptlrpc_request *req)
2065 {
2066         struct obd_trans_info trans_info = { 0, };
2067         struct obd_trans_info *oti = &trans_info;
2068         int should_process, fail = OBD_FAIL_OST_ALL_REPLY_NET, rc = 0;
2069         struct obd_device *obd = NULL;
2070         ENTRY;
2071
2072         /* OST module is kept between remounts, but the last reference
2073          * to specific module (say, osd or ofd) kills all related keys
2074          * from the environment. so we have to refill it until the root
2075          * cause is fixed properly */
2076         lu_env_refill(req->rq_svc_thread->t_env);
2077
2078         LASSERT(current->journal_info == NULL);
2079
2080         /* primordial rpcs don't affect server recovery */
2081         switch (lustre_msg_get_opc(req->rq_reqmsg)) {
2082         case SEC_CTX_INIT:
2083         case SEC_CTX_INIT_CONT:
2084         case SEC_CTX_FINI:
2085                 GOTO(out, rc = 0);
2086         }
2087
2088         req_capsule_init(&req->rq_pill, req, RCL_SERVER);
2089
2090         if (lustre_msg_get_opc(req->rq_reqmsg) != OST_CONNECT) {
2091                 if (!class_connected_export(req->rq_export)) {
2092                         CDEBUG(D_HA,"operation %d on unconnected OST from %s\n",
2093                                lustre_msg_get_opc(req->rq_reqmsg),
2094                                libcfs_id2str(req->rq_peer));
2095                         req->rq_status = -ENOTCONN;
2096                         GOTO(out, rc = -ENOTCONN);
2097                 }
2098
2099                 obd = req->rq_export->exp_obd;
2100
2101                 /* Check for aborted recovery. */
2102                 if (obd->obd_recovering) {
2103                         rc = ost_filter_recovery_request(req, obd,
2104                                                          &should_process);
2105                         if (rc || !should_process)
2106                                 RETURN(rc);
2107                         else if (should_process < 0) {
2108                                 req->rq_status = should_process;
2109                                 rc = ptlrpc_error(req);
2110                                 RETURN(rc);
2111                         }
2112                 }
2113         }
2114
2115         oti_init(oti, req);
2116
2117         rc = ost_msg_check_version(req->rq_reqmsg);
2118         if (rc)
2119                 RETURN(rc);
2120
2121         if (req && req->rq_reqmsg && req->rq_export &&
2122             (req->rq_export->exp_connect_flags & OBD_CONNECT_JOBSTATS))
2123                 oti->oti_jobid = lustre_msg_get_jobid(req->rq_reqmsg);
2124
2125         switch (lustre_msg_get_opc(req->rq_reqmsg)) {
2126         case OST_CONNECT: {
2127                 CDEBUG(D_INODE, "connect\n");
2128                 req_capsule_set(&req->rq_pill, &RQF_OST_CONNECT);
2129                 if (OBD_FAIL_CHECK(OBD_FAIL_OST_CONNECT_NET))
2130                         RETURN(0);
2131                 rc = target_handle_connect(req);
2132                 if (OBD_FAIL_CHECK(OBD_FAIL_OST_CONNECT_NET2))
2133                         RETURN(0);
2134                 if (!rc) {
2135                         rc = ost_init_sec_level(req);
2136                         if (!rc)
2137                                 rc = ost_connect_check_sptlrpc(req);
2138                 }
2139                 break;
2140         }
2141         case OST_DISCONNECT:
2142                 CDEBUG(D_INODE, "disconnect\n");
2143                 req_capsule_set(&req->rq_pill, &RQF_OST_DISCONNECT);
2144                 if (OBD_FAIL_CHECK(OBD_FAIL_OST_DISCONNECT_NET))
2145                         RETURN(0);
2146                 rc = target_handle_disconnect(req);
2147                 break;
2148         case OST_CREATE:
2149                 CDEBUG(D_INODE, "create\n");
2150                 req_capsule_set(&req->rq_pill, &RQF_OST_CREATE);
2151                 if (OBD_FAIL_CHECK(OBD_FAIL_OST_CREATE_NET))
2152                         RETURN(0);
2153                 if (OBD_FAIL_CHECK(OBD_FAIL_OST_EROFS))
2154                         GOTO(out, rc = -EROFS);
2155                 rc = ost_create(req->rq_export, req, oti);
2156                 break;
2157         case OST_DESTROY:
2158                 CDEBUG(D_INODE, "destroy\n");
2159                 req_capsule_set(&req->rq_pill, &RQF_OST_DESTROY);
2160                 if (OBD_FAIL_CHECK(OBD_FAIL_OST_DESTROY_NET))
2161                         RETURN(0);
2162                 if (OBD_FAIL_CHECK(OBD_FAIL_OST_EROFS))
2163                         GOTO(out, rc = -EROFS);
2164                 rc = ost_destroy(req->rq_export, req, oti);
2165                 break;
2166         case OST_GETATTR:
2167                 CDEBUG(D_INODE, "getattr\n");
2168                 req_capsule_set(&req->rq_pill, &RQF_OST_GETATTR);
2169                 if (OBD_FAIL_CHECK(OBD_FAIL_OST_GETATTR_NET))
2170                         RETURN(0);
2171                 rc = ost_getattr(req->rq_export, req);
2172                 break;
2173         case OST_SETATTR:
2174                 CDEBUG(D_INODE, "setattr\n");
2175                 req_capsule_set(&req->rq_pill, &RQF_OST_SETATTR);
2176                 if (OBD_FAIL_CHECK(OBD_FAIL_OST_SETATTR_NET))
2177                         RETURN(0);
2178                 rc = ost_setattr(req->rq_export, req, oti);
2179                 break;
2180         case OST_WRITE:
2181                 req_capsule_set(&req->rq_pill, &RQF_OST_BRW_WRITE);
2182                 CDEBUG(D_INODE, "write\n");
2183                 /* req->rq_request_portal would be nice, if it was set */
2184                 if (req->rq_rqbd->rqbd_service->srv_req_portal !=OST_IO_PORTAL){
2185                         CERROR("%s: deny write request from %s to portal %u\n",
2186                                req->rq_export->exp_obd->obd_name,
2187                                obd_export_nid2str(req->rq_export),
2188                                req->rq_rqbd->rqbd_service->srv_req_portal);
2189                         GOTO(out, rc = -EPROTO);
2190                 }
2191                 if (OBD_FAIL_CHECK(OBD_FAIL_OST_BRW_NET))
2192                         RETURN(0);
2193                 if (OBD_FAIL_CHECK(OBD_FAIL_OST_ENOSPC))
2194                         GOTO(out, rc = -ENOSPC);
2195                 if (OBD_FAIL_CHECK(OBD_FAIL_OST_EROFS))
2196                         GOTO(out, rc = -EROFS);
2197                 rc = ost_brw_write(req, oti);
2198                 LASSERT(current->journal_info == NULL);
2199                 /* ost_brw_write sends its own replies */
2200                 RETURN(rc);
2201         case OST_READ:
2202                 req_capsule_set(&req->rq_pill, &RQF_OST_BRW_READ);
2203                 CDEBUG(D_INODE, "read\n");
2204                 /* req->rq_request_portal would be nice, if it was set */
2205                 if (req->rq_rqbd->rqbd_service->srv_req_portal !=OST_IO_PORTAL){
2206                         CERROR("%s: deny read request from %s to portal %u\n",
2207                                req->rq_export->exp_obd->obd_name,
2208                                obd_export_nid2str(req->rq_export),
2209                                req->rq_rqbd->rqbd_service->srv_req_portal);
2210                         GOTO(out, rc = -EPROTO);
2211                 }
2212                 if (OBD_FAIL_CHECK(OBD_FAIL_OST_BRW_NET))
2213                         RETURN(0);
2214                 rc = ost_brw_read(req, oti);
2215                 LASSERT(current->journal_info == NULL);
2216                 /* ost_brw_read sends its own replies */
2217                 RETURN(rc);
2218         case OST_PUNCH:
2219                 CDEBUG(D_INODE, "punch\n");
2220                 req_capsule_set(&req->rq_pill, &RQF_OST_PUNCH);
2221                 if (OBD_FAIL_CHECK(OBD_FAIL_OST_PUNCH_NET))
2222                         RETURN(0);
2223                 if (OBD_FAIL_CHECK(OBD_FAIL_OST_EROFS))
2224                         GOTO(out, rc = -EROFS);
2225                 rc = ost_punch(req->rq_export, req, oti);
2226                 break;
2227         case OST_STATFS:
2228                 CDEBUG(D_INODE, "statfs\n");
2229                 req_capsule_set(&req->rq_pill, &RQF_OST_STATFS);
2230                 if (OBD_FAIL_CHECK(OBD_FAIL_OST_STATFS_NET))
2231                         RETURN(0);
2232                 rc = ost_statfs(req);
2233                 break;
2234         case OST_SYNC:
2235                 CDEBUG(D_INODE, "sync\n");
2236                 req_capsule_set(&req->rq_pill, &RQF_OST_SYNC);
2237                 if (OBD_FAIL_CHECK(OBD_FAIL_OST_SYNC_NET))
2238                         RETURN(0);
2239                 rc = ost_sync(req->rq_export, req, oti);
2240                 break;
2241         case OST_SET_INFO:
2242                 DEBUG_REQ(D_INODE, req, "set_info");
2243                 req_capsule_set(&req->rq_pill, &RQF_OBD_SET_INFO);
2244                 rc = ost_set_info(req->rq_export, req);
2245                 break;
2246         case OST_GET_INFO:
2247                 DEBUG_REQ(D_INODE, req, "get_info");
2248                 req_capsule_set(&req->rq_pill, &RQF_OST_GET_INFO_GENERIC);
2249                 rc = ost_get_info(req->rq_export, req);
2250                 break;
2251 #ifdef HAVE_QUOTA_SUPPORT
2252         case OST_QUOTACHECK:
2253                 CDEBUG(D_INODE, "quotacheck\n");
2254                 req_capsule_set(&req->rq_pill, &RQF_OST_QUOTACHECK);
2255                 if (OBD_FAIL_CHECK(OBD_FAIL_OST_QUOTACHECK_NET))
2256                         RETURN(0);
2257                 rc = ost_handle_quotacheck(req);
2258                 break;
2259         case OST_QUOTACTL:
2260                 CDEBUG(D_INODE, "quotactl\n");
2261                 req_capsule_set(&req->rq_pill, &RQF_OST_QUOTACTL);
2262                 if (OBD_FAIL_CHECK(OBD_FAIL_OST_QUOTACTL_NET))
2263                         RETURN(0);
2264                 rc = ost_handle_quotactl(req);
2265                 break;
2266         case OST_QUOTA_ADJUST_QUNIT:
2267                 CDEBUG(D_INODE, "quota_adjust_qunit\n");
2268                 req_capsule_set(&req->rq_pill, &RQF_OST_QUOTA_ADJUST_QUNIT);
2269                 rc = ost_handle_quota_adjust_qunit(req);
2270                 break;
2271 #endif
2272         case OBD_PING:
2273                 DEBUG_REQ(D_INODE, req, "ping");
2274                 req_capsule_set(&req->rq_pill, &RQF_OBD_PING);
2275                 rc = target_handle_ping(req);
2276                 break;
2277         /* FIXME - just reply status */
2278         case LLOG_ORIGIN_CONNECT:
2279                 DEBUG_REQ(D_INODE, req, "log connect");
2280                 req_capsule_set(&req->rq_pill, &RQF_LLOG_ORIGIN_CONNECT);
2281                 rc = ost_llog_handle_connect(req->rq_export, req);
2282                 req->rq_status = rc;
2283                 rc = req_capsule_server_pack(&req->rq_pill);
2284                 if (rc)
2285                         RETURN(rc);
2286                 RETURN(ptlrpc_reply(req));
2287         case OBD_LOG_CANCEL:
2288                 CDEBUG(D_INODE, "log cancel\n");
2289                 req_capsule_set(&req->rq_pill, &RQF_LOG_CANCEL);
2290                 if (OBD_FAIL_CHECK(OBD_FAIL_OBD_LOG_CANCEL_NET))
2291                         RETURN(0);
2292                 rc = llog_origin_handle_cancel(req);
2293                 if (OBD_FAIL_CHECK(OBD_FAIL_OBD_LOG_CANCEL_REP))
2294                         RETURN(0);
2295                 req->rq_status = rc;
2296                 rc = req_capsule_server_pack(&req->rq_pill);
2297                 if (rc)
2298                         RETURN(rc);
2299                 RETURN(ptlrpc_reply(req));
2300         case LDLM_ENQUEUE:
2301                 CDEBUG(D_INODE, "enqueue\n");
2302                 req_capsule_set(&req->rq_pill, &RQF_LDLM_ENQUEUE);
2303                 if (OBD_FAIL_CHECK(OBD_FAIL_LDLM_ENQUEUE))
2304                         RETURN(0);
2305                 rc = ldlm_handle_enqueue(req, ldlm_server_completion_ast,
2306                                          ost_blocking_ast,
2307                                          ldlm_server_glimpse_ast);
2308                 fail = OBD_FAIL_OST_LDLM_REPLY_NET;
2309                 break;
2310         case LDLM_CONVERT:
2311                 CDEBUG(D_INODE, "convert\n");
2312                 req_capsule_set(&req->rq_pill, &RQF_LDLM_CONVERT);
2313                 if (OBD_FAIL_CHECK(OBD_FAIL_LDLM_CONVERT))
2314                         RETURN(0);
2315                 rc = ldlm_handle_convert(req);
2316                 break;
2317         case LDLM_CANCEL:
2318                 CDEBUG(D_INODE, "cancel\n");
2319                 req_capsule_set(&req->rq_pill, &RQF_LDLM_CANCEL);
2320                 if (OBD_FAIL_CHECK(OBD_FAIL_LDLM_CANCEL))
2321                         RETURN(0);
2322                 rc = ldlm_handle_cancel(req);
2323                 break;
2324         case LDLM_BL_CALLBACK:
2325         case LDLM_CP_CALLBACK:
2326                 CDEBUG(D_INODE, "callback\n");
2327                 CERROR("callbacks should not happen on OST\n");
2328                 /* fall through */
2329         default:
2330                 CERROR("Unexpected opcode %d\n",
2331                        lustre_msg_get_opc(req->rq_reqmsg));
2332                 req->rq_status = -ENOTSUPP;
2333                 rc = ptlrpc_error(req);
2334                 RETURN(rc);
2335         }
2336
2337         LASSERT(current->journal_info == NULL);
2338
2339         EXIT;
2340         /* If we're DISCONNECTing, the export_data is already freed */
2341         if (!rc && lustre_msg_get_opc(req->rq_reqmsg) != OST_DISCONNECT)
2342                 target_committed_to_req(req);
2343
2344 out:
2345         if (!rc)
2346                 oti_to_request(oti, req);
2347
2348         target_send_reply(req, rc, fail);
2349         return 0;
2350 }
2351 EXPORT_SYMBOL(ost_handle);
2352 /*
2353  * free per-thread pool created by ost_thread_init().
2354  */
2355 static void ost_thread_done(struct ptlrpc_thread *thread)
2356 {
2357         struct ost_thread_local_cache *tls; /* TLS stands for Thread-Local
2358                                              * Storage */
2359
2360         ENTRY;
2361
2362         LASSERT(thread != NULL);
2363
2364         /*
2365          * be prepared to handle partially-initialized pools (because this is
2366          * called from ost_thread_init() for cleanup.
2367          */
2368         tls = thread->t_data;
2369         if (tls != NULL) {
2370                 OBD_FREE_PTR(tls);
2371                 thread->t_data = NULL;
2372         }
2373         EXIT;
2374 }
2375
2376 /*
2377  * initialize per-thread page pool (bug 5137).
2378  */
2379 static int ost_thread_init(struct ptlrpc_thread *thread)
2380 {
2381         struct ost_thread_local_cache *tls;
2382
2383         ENTRY;
2384
2385         LASSERT(thread != NULL);
2386         LASSERT(thread->t_data == NULL);
2387         LASSERTF(thread->t_id <= OSS_THREADS_MAX, "%u\n", thread->t_id);
2388
2389         OBD_ALLOC_PTR(tls);
2390         if (tls == NULL)
2391                 RETURN(-ENOMEM);
2392         thread->t_data = tls;
2393         RETURN(0);
2394 }
2395
2396 #define OST_WATCHDOG_TIMEOUT (obd_timeout * 1000)
2397
2398 /* Sigh - really, this is an OSS, the _server_, not the _target_ */
2399 static int ost_setup(struct obd_device *obd, struct lustre_cfg* lcfg)
2400 {
2401         struct ost_obd *ost = &obd->u.ost;
2402         struct lprocfs_static_vars lvars;
2403         int oss_min_threads;
2404         int oss_max_threads;
2405         int oss_min_create_threads;
2406         int oss_max_create_threads;
2407         int rc;
2408         ENTRY;
2409
2410         rc = cfs_cleanup_group_info();
2411         if (rc)
2412                 RETURN(rc);
2413
2414         lprocfs_ost_init_vars(&lvars);
2415         lprocfs_obd_setup(obd, lvars.obd_vars);
2416
2417         cfs_mutex_init(&ost->ost_health_mutex);
2418
2419         if (oss_num_threads) {
2420                 /* If oss_num_threads is set, it is the min and the max. */
2421                 if (oss_num_threads > OSS_THREADS_MAX)
2422                         oss_num_threads = OSS_THREADS_MAX;
2423                 if (oss_num_threads < OSS_THREADS_MIN)
2424                         oss_num_threads = OSS_THREADS_MIN;
2425                 oss_max_threads = oss_min_threads = oss_num_threads;
2426         } else {
2427                 /* Base min threads on memory and cpus */
2428                 oss_min_threads =
2429                         cfs_num_possible_cpus() * CFS_NUM_CACHEPAGES >>
2430                         (27 - CFS_PAGE_SHIFT);
2431                 if (oss_min_threads < OSS_THREADS_MIN)
2432                         oss_min_threads = OSS_THREADS_MIN;
2433                 /* Insure a 4x range for dynamic threads */
2434                 if (oss_min_threads > OSS_THREADS_MAX / 4)
2435                         oss_min_threads = OSS_THREADS_MAX / 4;
2436                 oss_max_threads = min(OSS_THREADS_MAX, oss_min_threads * 4 + 1);
2437         }
2438
2439         ost->ost_service =
2440                 ptlrpc_init_svc(OST_NBUFS, OST_BUFSIZE, OST_MAXREQSIZE,
2441                                 OST_MAXREPSIZE, OST_REQUEST_PORTAL,
2442                                 OSC_REPLY_PORTAL, OSS_SERVICE_WATCHDOG_FACTOR,
2443                                 ost_handle, LUSTRE_OSS_NAME,
2444                                 obd->obd_proc_entry, target_print_req,
2445                                 oss_min_threads, oss_max_threads,
2446                                 "ll_ost", LCT_DT_THREAD, NULL);
2447         if (ost->ost_service == NULL) {
2448                 CERROR("failed to start service\n");
2449                 GOTO(out_lprocfs, rc = -ENOMEM);
2450         }
2451
2452         rc = ptlrpc_start_threads(ost->ost_service);
2453         if (rc)
2454                 GOTO(out_service, rc = -EINVAL);
2455
2456         if (oss_num_create_threads) {
2457                 if (oss_num_create_threads > OSS_MAX_CREATE_THREADS)
2458                         oss_num_create_threads = OSS_MAX_CREATE_THREADS;
2459                 if (oss_num_create_threads < OSS_MIN_CREATE_THREADS)
2460                         oss_num_create_threads = OSS_MIN_CREATE_THREADS;
2461                 oss_min_create_threads = oss_max_create_threads =
2462                         oss_num_create_threads;
2463         } else {
2464                 oss_min_create_threads = OSS_MIN_CREATE_THREADS;
2465                 oss_max_create_threads = OSS_MAX_CREATE_THREADS;
2466         }
2467
2468         ost->ost_create_service =
2469                 ptlrpc_init_svc(OST_NBUFS, OST_BUFSIZE, OST_MAXREQSIZE,
2470                                 OST_MAXREPSIZE, OST_CREATE_PORTAL,
2471                                 OSC_REPLY_PORTAL, OSS_SERVICE_WATCHDOG_FACTOR,
2472                                 ost_handle, "ost_create",
2473                                 obd->obd_proc_entry, target_print_req,
2474                                 oss_min_create_threads, oss_max_create_threads,
2475                                 "ll_ost_creat", LCT_DT_THREAD, NULL);
2476         if (ost->ost_create_service == NULL) {
2477                 CERROR("failed to start OST create service\n");
2478                 GOTO(out_service, rc = -ENOMEM);
2479         }
2480
2481         rc = ptlrpc_start_threads(ost->ost_create_service);
2482         if (rc)
2483                 GOTO(out_create, rc = -EINVAL);
2484
2485         ost->ost_io_service =
2486                 ptlrpc_init_svc(OST_NBUFS, OST_BUFSIZE, OST_MAXREQSIZE,
2487                                 OST_MAXREPSIZE, OST_IO_PORTAL,
2488                                 OSC_REPLY_PORTAL, OSS_SERVICE_WATCHDOG_FACTOR,
2489                                 ost_handle, "ost_io",
2490                                 obd->obd_proc_entry, target_print_req,
2491                                 oss_min_threads, oss_max_threads,
2492                                 "ll_ost_io", LCT_DT_THREAD, ost_hpreq_handler);
2493         if (ost->ost_io_service == NULL) {
2494                 CERROR("failed to start OST I/O service\n");
2495                 GOTO(out_create, rc = -ENOMEM);
2496         }
2497
2498         ost->ost_io_service->srv_init = ost_thread_init;
2499         ost->ost_io_service->srv_done = ost_thread_done;
2500         ost->ost_io_service->srv_cpu_affinity = 1;
2501         rc = ptlrpc_start_threads(ost->ost_io_service);
2502         if (rc)
2503                 GOTO(out_io, rc = -EINVAL);
2504
2505         ping_evictor_start();
2506
2507         RETURN(0);
2508
2509 out_io:
2510         ptlrpc_unregister_service(ost->ost_io_service);
2511         ost->ost_io_service = NULL;
2512 out_create:
2513         ptlrpc_unregister_service(ost->ost_create_service);
2514         ost->ost_create_service = NULL;
2515 out_service:
2516         ptlrpc_unregister_service(ost->ost_service);
2517         ost->ost_service = NULL;
2518 out_lprocfs:
2519         lprocfs_obd_cleanup(obd);
2520         RETURN(rc);
2521 }
2522
2523 static int ost_cleanup(struct obd_device *obd)
2524 {
2525         struct ost_obd *ost = &obd->u.ost;
2526         int err = 0;
2527         ENTRY;
2528
2529         ping_evictor_stop();
2530
2531         /* there is no recovery for OST OBD, all recovery is controlled by
2532          * obdfilter OBD */
2533         LASSERT(obd->obd_recovering == 0);
2534         cfs_mutex_lock(&ost->ost_health_mutex);
2535         ptlrpc_unregister_service(ost->ost_service);
2536         ptlrpc_unregister_service(ost->ost_create_service);
2537         ptlrpc_unregister_service(ost->ost_io_service);
2538         ost->ost_service = NULL;
2539         ost->ost_create_service = NULL;
2540         cfs_mutex_unlock(&ost->ost_health_mutex);
2541
2542         lprocfs_obd_cleanup(obd);
2543
2544         RETURN(err);
2545 }
2546
2547 static int ost_health_check(const struct lu_env *env, struct obd_device *obd)
2548 {
2549         struct ost_obd *ost = &obd->u.ost;
2550         int rc = 0;
2551
2552         cfs_mutex_lock(&ost->ost_health_mutex);
2553         rc |= ptlrpc_service_health_check(ost->ost_service);
2554         rc |= ptlrpc_service_health_check(ost->ost_create_service);
2555         rc |= ptlrpc_service_health_check(ost->ost_io_service);
2556         cfs_mutex_unlock(&ost->ost_health_mutex);
2557
2558         /*
2559          * health_check to return 0 on healthy
2560          * and 1 on unhealthy.
2561          */
2562         if( rc != 0)
2563                 rc = 1;
2564
2565         return rc;
2566 }
2567
2568 struct ost_thread_local_cache *ost_tls(struct ptlrpc_request *r)
2569 {
2570         return (struct ost_thread_local_cache *)(r->rq_svc_thread->t_data);
2571 }
2572
2573 /* use obd ops to offer management infrastructure */
2574 static struct obd_ops ost_obd_ops = {
2575         .o_owner        = THIS_MODULE,
2576         .o_setup        = ost_setup,
2577         .o_cleanup      = ost_cleanup,
2578         .o_health_check = ost_health_check,
2579 };
2580
2581
2582 static int __init ost_init(void)
2583 {
2584         struct lprocfs_static_vars lvars;
2585         int rc;
2586         ENTRY;
2587
2588         lprocfs_ost_init_vars(&lvars);
2589         rc = class_register_type(&ost_obd_ops, NULL, lvars.module_vars,
2590                                  LUSTRE_OSS_NAME, NULL);
2591
2592         if (ost_num_threads != 0 && oss_num_threads == 0) {
2593                 LCONSOLE_INFO("ost_num_threads module parameter is deprecated, "
2594                               "use oss_num_threads instead or unset both for "
2595                               "dynamic thread startup\n");
2596                 oss_num_threads = ost_num_threads;
2597         }
2598
2599         RETURN(rc);
2600 }
2601
2602 static void /*__exit*/ ost_exit(void)
2603 {
2604         class_unregister_type(LUSTRE_OSS_NAME);
2605 }
2606
2607 MODULE_AUTHOR("Sun Microsystems, Inc. <http://www.lustre.org/>");
2608 MODULE_DESCRIPTION("Lustre Object Storage Target (OST) v0.01");
2609 MODULE_LICENSE("GPL");
2610
2611 module_init(ost_init);
2612 module_exit(ost_exit);