Whamcloud - gitweb
LU-5577 ptlrpc: fix comparison between signed and unsigned
[fs/lustre-release.git] / lustre / ldlm / ldlm_request.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) 2002, 2010, Oracle and/or its affiliates. All rights reserved.
28  * Use is subject to license terms.
29  *
30  * Copyright (c) 2010, 2013, Intel Corporation.
31  */
32 /*
33  * This file is part of Lustre, http://www.lustre.org/
34  * Lustre is a trademark of Sun Microsystems, Inc.
35  */
36 /**
37  * This file contains Asynchronous System Trap (AST) handlers and related
38  * LDLM request-processing routines.
39  *
40  * An AST is a callback issued on a lock when its state is changed. There are
41  * several different types of ASTs (callbacks) registered for each lock:
42  *
43  * - completion AST: when a lock is enqueued by some process, but cannot be
44  *   granted immediately due to other conflicting locks on the same resource,
45  *   the completion AST is sent to notify the caller when the lock is
46  *   eventually granted
47  *
48  * - blocking AST: when a lock is granted to some process, if another process
49  *   enqueues a conflicting (blocking) lock on a resource, a blocking AST is
50  *   sent to notify the holder(s) of the lock(s) of the conflicting lock
51  *   request. The lock holder(s) must release their lock(s) on that resource in
52  *   a timely manner or be evicted by the server.
53  *
54  * - glimpse AST: this is used when a process wants information about a lock
55  *   (i.e. the lock value block (LVB)) but does not necessarily require holding
56  *   the lock. If the resource is locked, the lock holder(s) are sent glimpse
57  *   ASTs and the LVB is returned to the caller, and lock holder(s) may CANCEL
58  *   their lock(s) if they are idle. If the resource is not locked, the server
59  *   may grant the lock.
60  */
61
62 #define DEBUG_SUBSYSTEM S_LDLM
63
64 #include <lustre_dlm.h>
65 #include <obd_class.h>
66 #include <obd.h>
67
68 #include "ldlm_internal.h"
69
70 unsigned int ldlm_enqueue_min = OBD_TIMEOUT_DEFAULT;
71 CFS_MODULE_PARM(ldlm_enqueue_min, "i", uint, 0644,
72                 "lock enqueue timeout minimum");
73
74 /* in client side, whether the cached locks will be canceled before replay */
75 unsigned int ldlm_cancel_unused_locks_before_replay = 1;
76
77 static void interrupted_completion_wait(void *data)
78 {
79 }
80
81 struct lock_wait_data {
82         struct ldlm_lock *lwd_lock;
83         __u32             lwd_conn_cnt;
84 };
85
86 struct ldlm_async_args {
87         struct lustre_handle lock_handle;
88 };
89
90 int ldlm_expired_completion_wait(void *data)
91 {
92         struct lock_wait_data *lwd = data;
93         struct ldlm_lock *lock = lwd->lwd_lock;
94         struct obd_import *imp;
95         struct obd_device *obd;
96
97         ENTRY;
98         if (lock->l_conn_export == NULL) {
99                 static cfs_time_t next_dump = 0, last_dump = 0;
100
101                 LCONSOLE_WARN("lock timed out (enqueued at "CFS_TIME_T", "
102                               CFS_DURATION_T"s ago)\n",
103                               lock->l_last_activity,
104                               cfs_time_sub(cfs_time_current_sec(),
105                                            lock->l_last_activity));
106                 LDLM_DEBUG(lock, "lock timed out (enqueued at "CFS_TIME_T", "
107                            CFS_DURATION_T"s ago); not entering recovery in "
108                            "server code, just going back to sleep",
109                            lock->l_last_activity,
110                            cfs_time_sub(cfs_time_current_sec(),
111                                         lock->l_last_activity));
112                 if (cfs_time_after(cfs_time_current(), next_dump)) {
113                         last_dump = next_dump;
114                         next_dump = cfs_time_shift(300);
115                         ldlm_namespace_dump(D_DLMTRACE,
116                                             ldlm_lock_to_ns(lock));
117                         if (last_dump == 0)
118                                 libcfs_debug_dumplog();
119                 }
120                 RETURN(0);
121         }
122
123         obd = lock->l_conn_export->exp_obd;
124         imp = obd->u.cli.cl_import;
125         ptlrpc_fail_import(imp, lwd->lwd_conn_cnt);
126         LDLM_ERROR(lock, "lock timed out (enqueued at "CFS_TIME_T", "
127                   CFS_DURATION_T"s ago), entering recovery for %s@%s",
128                   lock->l_last_activity,
129                   cfs_time_sub(cfs_time_current_sec(), lock->l_last_activity),
130                   obd2cli_tgt(obd), imp->imp_connection->c_remote_uuid.uuid);
131
132         RETURN(0);
133 }
134 EXPORT_SYMBOL(ldlm_expired_completion_wait);
135
136 /**
137  * Calculate the Completion timeout (covering enqueue, BL AST, data flush,
138  * lock cancel, and their replies). Used for lock completion timeout on the
139  * client side.
140  *
141  * \param[in] lock        lock which is waiting the completion callback
142  *
143  * \retval            timeout in seconds to wait for the server reply
144  */
145
146 /* We use the same basis for both server side and client side functions
147    from a single node. */
148 static unsigned int ldlm_cp_timeout(struct ldlm_lock *lock)
149 {
150         unsigned int timeout;
151
152         if (AT_OFF)
153                 return obd_timeout;
154
155         /* Wait a long time for enqueue - server may have to callback a
156          * lock from another client.  Server will evict the other client if it
157          * doesn't respond reasonably, and then give us the lock. */
158         timeout = at_get(ldlm_lock_to_ns_at(lock));
159         return max(3 * timeout, ldlm_enqueue_min);
160 }
161
162 /**
163  * Helper function for ldlm_completion_ast(), updating timings when lock is
164  * actually granted.
165  */
166 static int ldlm_completion_tail(struct ldlm_lock *lock, void *data)
167 {
168         long delay;
169         int  result = 0;
170
171         if (ldlm_is_destroyed(lock) || ldlm_is_failed(lock)) {
172                 LDLM_DEBUG(lock, "client-side enqueue: destroyed");
173                 result = -EIO;
174         } else if (data == NULL) {
175                 LDLM_DEBUG(lock, "client-side enqueue: granted");
176         } else {
177                 /* Take into AT only CP RPC, not immediately granted locks */
178                 delay = cfs_time_sub(cfs_time_current_sec(),
179                                      lock->l_last_activity);
180                 LDLM_DEBUG(lock, "client-side enqueue: granted after "
181                            CFS_DURATION_T"s", delay);
182
183                 /* Update our time estimate */
184                 at_measured(ldlm_lock_to_ns_at(lock), delay);
185         }
186         return result;
187 }
188
189 /**
190  * Implementation of ->l_completion_ast() for a client, that doesn't wait
191  * until lock is granted. Suitable for locks enqueued through ptlrpcd, of
192  * other threads that cannot block for long.
193  */
194 int ldlm_completion_ast_async(struct ldlm_lock *lock, __u64 flags, void *data)
195 {
196         ENTRY;
197
198         if (flags == LDLM_FL_WAIT_NOREPROC) {
199                 LDLM_DEBUG(lock, "client-side enqueue waiting on pending lock");
200                 RETURN(0);
201         }
202
203         if (!(flags & LDLM_FL_BLOCKED_MASK)) {
204                 wake_up(&lock->l_waitq);
205                 RETURN(ldlm_completion_tail(lock, data));
206         }
207
208         LDLM_DEBUG(lock, "client-side enqueue returned a blocked lock, "
209                    "going forward");
210         ldlm_reprocess_all(lock->l_resource);
211         RETURN(0);
212 }
213 EXPORT_SYMBOL(ldlm_completion_ast_async);
214
215 /**
216  * Generic LDLM "completion" AST. This is called in several cases:
217  *
218  *     - when a reply to an ENQUEUE RPC is received from the server
219  *       (ldlm_cli_enqueue_fini()). Lock might be granted or not granted at
220  *       this point (determined by flags);
221  *
222  *     - when LDLM_CP_CALLBACK RPC comes to client to notify it that lock has
223  *       been granted;
224  *
225  *     - when ldlm_lock_match(LDLM_FL_LVB_READY) is about to wait until lock
226  *       gets correct lvb;
227  *
228  *     - to force all locks when resource is destroyed (cleanup_resource());
229  *
230  *     - during lock conversion (not used currently).
231  *
232  * If lock is not granted in the first case, this function waits until second
233  * or penultimate cases happen in some other thread.
234  *
235  */
236 int ldlm_completion_ast(struct ldlm_lock *lock, __u64 flags, void *data)
237 {
238         /* XXX ALLOCATE - 160 bytes */
239         struct lock_wait_data lwd;
240         struct obd_device *obd;
241         struct obd_import *imp = NULL;
242         struct l_wait_info lwi;
243         __u32 timeout;
244         int rc = 0;
245         ENTRY;
246
247         if (flags == LDLM_FL_WAIT_NOREPROC) {
248                 LDLM_DEBUG(lock, "client-side enqueue waiting on pending lock");
249                 goto noreproc;
250         }
251
252         if (!(flags & LDLM_FL_BLOCKED_MASK)) {
253                 wake_up(&lock->l_waitq);
254                 RETURN(0);
255         }
256
257         LDLM_DEBUG(lock, "client-side enqueue returned a blocked lock, "
258                    "sleeping");
259
260 noreproc:
261
262         obd = class_exp2obd(lock->l_conn_export);
263
264         /* if this is a local lock, then there is no import */
265         if (obd != NULL) {
266                 imp = obd->u.cli.cl_import;
267         }
268
269         timeout = ldlm_cp_timeout(lock);
270
271         lwd.lwd_lock = lock;
272         lock->l_last_activity = cfs_time_current_sec();
273
274         if (ldlm_is_no_timeout(lock)) {
275                 LDLM_DEBUG(lock, "waiting indefinitely because of NO_TIMEOUT");
276                 lwi = LWI_INTR(interrupted_completion_wait, &lwd);
277         } else {
278                 lwi = LWI_TIMEOUT_INTR(cfs_time_seconds(timeout),
279                                        ldlm_expired_completion_wait,
280                                        interrupted_completion_wait, &lwd);
281         }
282
283         if (imp != NULL) {
284                 spin_lock(&imp->imp_lock);
285                 lwd.lwd_conn_cnt = imp->imp_conn_cnt;
286                 spin_unlock(&imp->imp_lock);
287         }
288
289         if (ns_is_client(ldlm_lock_to_ns(lock)) &&
290             OBD_FAIL_CHECK_RESET(OBD_FAIL_LDLM_INTR_CP_AST,
291                                  OBD_FAIL_LDLM_CP_BL_RACE | OBD_FAIL_ONCE)) {
292                 ldlm_set_fail_loc(lock);
293                 rc = -EINTR;
294         } else {
295                 /* Go to sleep until the lock is granted or cancelled. */
296                 rc = l_wait_event(lock->l_waitq,
297                                   is_granted_or_cancelled(lock), &lwi);
298         }
299
300         if (rc) {
301                 LDLM_DEBUG(lock, "client-side enqueue waking up: failed (%d)",
302                            rc);
303                 RETURN(rc);
304         }
305
306         RETURN(ldlm_completion_tail(lock, data));
307 }
308 EXPORT_SYMBOL(ldlm_completion_ast);
309
310 /**
311  * A helper to build a blocking AST function
312  *
313  * Perform a common operation for blocking ASTs:
314  * defferred lock cancellation.
315  *
316  * \param lock the lock blocking or canceling AST was called on
317  * \retval 0
318  * \see mdt_blocking_ast
319  * \see ldlm_blocking_ast
320  */
321 int ldlm_blocking_ast_nocheck(struct ldlm_lock *lock)
322 {
323         int do_ast;
324         ENTRY;
325
326         ldlm_set_cbpending(lock);
327         do_ast = (!lock->l_readers && !lock->l_writers);
328         unlock_res_and_lock(lock);
329
330         if (do_ast) {
331                 struct lustre_handle lockh;
332                 int rc;
333
334                 LDLM_DEBUG(lock, "already unused, calling ldlm_cli_cancel");
335                 ldlm_lock2handle(lock, &lockh);
336                 rc = ldlm_cli_cancel(&lockh, LCF_ASYNC);
337                 if (rc < 0)
338                         CERROR("ldlm_cli_cancel: %d\n", rc);
339         } else {
340                 LDLM_DEBUG(lock, "Lock still has references, will be "
341                            "cancelled later");
342         }
343         RETURN(0);
344 }
345 EXPORT_SYMBOL(ldlm_blocking_ast_nocheck);
346
347 /**
348  * Server blocking AST
349  *
350  * ->l_blocking_ast() callback for LDLM locks acquired by server-side
351  * OBDs.
352  *
353  * \param lock the lock which blocks a request or cancelling lock
354  * \param desc unused
355  * \param data unused
356  * \param flag indicates whether this cancelling or blocking callback
357  * \retval 0
358  * \see ldlm_blocking_ast_nocheck
359  */
360 int ldlm_blocking_ast(struct ldlm_lock *lock, struct ldlm_lock_desc *desc,
361                       void *data, int flag)
362 {
363         ENTRY;
364
365         if (flag == LDLM_CB_CANCELING) {
366                 /* Don't need to do anything here. */
367                 RETURN(0);
368         }
369
370         lock_res_and_lock(lock);
371         /* Get this: if ldlm_blocking_ast is racing with intent_policy, such
372          * that ldlm_blocking_ast is called just before intent_policy method
373          * takes the lr_lock, then by the time we get the lock, we might not
374          * be the correct blocking function anymore.  So check, and return
375          * early, if so. */
376         if (lock->l_blocking_ast != ldlm_blocking_ast) {
377                 unlock_res_and_lock(lock);
378                 RETURN(0);
379         }
380         RETURN(ldlm_blocking_ast_nocheck(lock));
381 }
382 EXPORT_SYMBOL(ldlm_blocking_ast);
383
384 /**
385  * Implements ldlm_lock::l_glimpse_ast for extent locks acquired on the server.
386  *
387  * Returning -ELDLM_NO_LOCK_DATA actually works, but the reason for that is
388  * rather subtle: with OST-side locking, it may so happen that _all_ extent
389  * locks are held by the OST. If client wants to obtain the current file size
390  * it calls ll_glimpse_size(), and (as all locks are held only on the server),
391  * this dummy glimpse callback fires and does nothing. The client still
392  * receives the correct file size due to the following fragment of code in
393  * ldlm_cb_interpret():
394  *
395  *      if (rc == -ELDLM_NO_LOCK_DATA) {
396  *              LDLM_DEBUG(lock, "lost race - client has a lock but no"
397  *                         "inode");
398  *              ldlm_res_lvbo_update(lock->l_resource, NULL, 1);
399  *      }
400  *
401  * That is, after the glimpse returns this error, ofd_lvbo_update() is called
402  * and returns the updated file attributes from the inode to the client.
403  *
404  * See also comment in ofd_intent_policy() on why servers must set a non-NULL
405  * l_glimpse_ast when grabbing DLM locks.  Otherwise, the server will assume
406  * that the object is in the process of being destroyed.
407  *
408  * \param[in] lock      DLM lock being glimpsed, unused
409  * \param[in] reqp      pointer to ptlrpc_request, unused
410  *
411  * \retval              -ELDLM_NO_LOCK_DATA to get attributes from disk object
412  */
413 int ldlm_glimpse_ast(struct ldlm_lock *lock, void *reqp)
414 {
415         return -ELDLM_NO_LOCK_DATA;
416 }
417 EXPORT_SYMBOL(ldlm_glimpse_ast);
418
419 /**
420  * Enqueue a local lock (typically on a server).
421  */
422 int ldlm_cli_enqueue_local(struct ldlm_namespace *ns,
423                            const struct ldlm_res_id *res_id,
424                            ldlm_type_t type, ldlm_policy_data_t *policy,
425                            ldlm_mode_t mode, __u64 *flags,
426                            ldlm_blocking_callback blocking,
427                            ldlm_completion_callback completion,
428                            ldlm_glimpse_callback glimpse,
429                            void *data, __u32 lvb_len, enum lvb_type lvb_type,
430                            const __u64 *client_cookie,
431                            struct lustre_handle *lockh)
432 {
433         struct ldlm_lock *lock;
434         int err;
435         const struct ldlm_callback_suite cbs = { .lcs_completion = completion,
436                                                  .lcs_blocking   = blocking,
437                                                  .lcs_glimpse    = glimpse,
438         };
439         ENTRY;
440
441         LASSERT(!(*flags & LDLM_FL_REPLAY));
442         if (unlikely(ns_is_client(ns))) {
443                 CERROR("Trying to enqueue local lock in a shadow namespace\n");
444                 LBUG();
445         }
446
447         lock = ldlm_lock_create(ns, res_id, type, mode, &cbs, data, lvb_len,
448                                 lvb_type);
449         if (IS_ERR(lock))
450                 GOTO(out_nolock, err = PTR_ERR(lock));
451
452         err = ldlm_lvbo_init(lock->l_resource);
453         if (err < 0) {
454                 LDLM_ERROR(lock, "delayed lvb init failed (rc %d)", err);
455                 GOTO(out, err);
456         }
457
458         ldlm_lock2handle(lock, lockh);
459
460         /* NB: we don't have any lock now (lock_res_and_lock)
461          * because it's a new lock */
462         ldlm_lock_addref_internal_nolock(lock, mode);
463         ldlm_set_local(lock);
464         if (*flags & LDLM_FL_ATOMIC_CB)
465                 ldlm_set_atomic_cb(lock);
466
467         if (policy != NULL)
468                 lock->l_policy_data = *policy;
469         if (client_cookie != NULL)
470                 lock->l_client_cookie = *client_cookie;
471         if (type == LDLM_EXTENT) {
472                 /* extent lock without policy is a bug */
473                 if (policy == NULL)
474                         LBUG();
475
476                 lock->l_req_extent = policy->l_extent;
477         }
478
479         err = ldlm_lock_enqueue(ns, &lock, policy, flags);
480         if (unlikely(err != ELDLM_OK))
481                 GOTO(out, err);
482
483         if (policy != NULL)
484                 *policy = lock->l_policy_data;
485
486         if (lock->l_completion_ast)
487                 lock->l_completion_ast(lock, *flags, NULL);
488
489         LDLM_DEBUG(lock, "client-side local enqueue handler, new lock created");
490         EXIT;
491  out:
492         LDLM_LOCK_RELEASE(lock);
493  out_nolock:
494         return err;
495 }
496 EXPORT_SYMBOL(ldlm_cli_enqueue_local);
497
498 static void failed_lock_cleanup(struct ldlm_namespace *ns,
499                                 struct ldlm_lock *lock, int mode)
500 {
501         int need_cancel = 0;
502
503         /* Set a flag to prevent us from sending a CANCEL (bug 407) */
504         lock_res_and_lock(lock);
505         /* Check that lock is not granted or failed, we might race. */
506         if ((lock->l_req_mode != lock->l_granted_mode) &&
507             !ldlm_is_failed(lock)) {
508                 /* Make sure that this lock will not be found by raced
509                  * bl_ast and -EINVAL reply is sent to server anyways.
510                  * b=17645*/
511                 lock->l_flags |= LDLM_FL_LOCAL_ONLY | LDLM_FL_FAILED |
512                                  LDLM_FL_ATOMIC_CB | LDLM_FL_CBPENDING;
513                 need_cancel = 1;
514         }
515         unlock_res_and_lock(lock);
516
517         if (need_cancel)
518                 LDLM_DEBUG(lock,
519                            "setting FL_LOCAL_ONLY | LDLM_FL_FAILED | "
520                            "LDLM_FL_ATOMIC_CB | LDLM_FL_CBPENDING");
521         else
522                 LDLM_DEBUG(lock, "lock was granted or failed in race");
523
524         /* XXX - HACK because we shouldn't call ldlm_lock_destroy()
525          *       from llite/file.c/ll_file_flock(). */
526         /* This code makes for the fact that we do not have blocking handler on
527          * a client for flock locks. As such this is the place where we must
528          * completely kill failed locks. (interrupted and those that
529          * were waiting to be granted when server evicted us. */
530         if (lock->l_resource->lr_type == LDLM_FLOCK) {
531                 lock_res_and_lock(lock);
532                 if (!ldlm_is_destroyed(lock)) {
533                         ldlm_resource_unlink_lock(lock);
534                         ldlm_lock_decref_internal_nolock(lock, mode);
535                         ldlm_lock_destroy_nolock(lock);
536                 }
537                 unlock_res_and_lock(lock);
538         } else {
539                 ldlm_lock_decref_internal(lock, mode);
540         }
541 }
542
543 /**
544  * Finishing portion of client lock enqueue code.
545  *
546  * Called after receiving reply from server.
547  */
548 int ldlm_cli_enqueue_fini(struct obd_export *exp, struct ptlrpc_request *req,
549                           ldlm_type_t type, __u8 with_policy, ldlm_mode_t mode,
550                           __u64 *flags, void *lvb, __u32 lvb_len,
551                           struct lustre_handle *lockh,int rc)
552 {
553         struct ldlm_namespace *ns = exp->exp_obd->obd_namespace;
554         int is_replay = *flags & LDLM_FL_REPLAY;
555         struct ldlm_lock *lock;
556         struct ldlm_reply *reply;
557         int cleanup_phase = 1;
558         ENTRY;
559
560         lock = ldlm_handle2lock(lockh);
561         /* ldlm_cli_enqueue is holding a reference on this lock. */
562         if (!lock) {
563                 LASSERT(type == LDLM_FLOCK);
564                 RETURN(-ENOLCK);
565         }
566
567         LASSERTF(ergo(lvb_len != 0, lvb_len == lock->l_lvb_len),
568                  "lvb_len = %d, l_lvb_len = %d\n", lvb_len, lock->l_lvb_len);
569
570         if (rc != ELDLM_OK) {
571                 LASSERT(!is_replay);
572                 LDLM_DEBUG(lock, "client-side enqueue END (%s)",
573                            rc == ELDLM_LOCK_ABORTED ? "ABORTED" : "FAILED");
574
575                 if (rc != ELDLM_LOCK_ABORTED)
576                         GOTO(cleanup, rc);
577         }
578
579         /* Before we return, swab the reply */
580         reply = req_capsule_server_get(&req->rq_pill, &RMF_DLM_REP);
581         if (reply == NULL)
582                 GOTO(cleanup, rc = -EPROTO);
583
584         if (lvb_len > 0) {
585                 int size = 0;
586
587                 size = req_capsule_get_size(&req->rq_pill, &RMF_DLM_LVB,
588                                             RCL_SERVER);
589                 if (size < 0) {
590                         LDLM_ERROR(lock, "Fail to get lvb_len, rc = %d", size);
591                         GOTO(cleanup, rc = size);
592                 } else if (unlikely(size > lvb_len)) {
593                         LDLM_ERROR(lock, "Replied LVB is larger than "
594                                    "expectation, expected = %d, replied = %d",
595                                    lvb_len, size);
596                         GOTO(cleanup, rc = -EINVAL);
597                 }
598                 lvb_len = size;
599         }
600
601         if (rc == ELDLM_LOCK_ABORTED) {
602                 if (lvb_len > 0 && lvb != NULL)
603                         rc = ldlm_fill_lvb(lock, &req->rq_pill, RCL_SERVER,
604                                            lvb, lvb_len);
605                 GOTO(cleanup, rc = rc ? : ELDLM_LOCK_ABORTED);
606         }
607
608         /* lock enqueued on the server */
609         cleanup_phase = 0;
610
611         lock_res_and_lock(lock);
612         /* Key change rehash lock in per-export hash with new key */
613         if (exp->exp_lock_hash) {
614                 /* In the function below, .hs_keycmp resolves to
615                  * ldlm_export_lock_keycmp() */
616                 /* coverity[overrun-buffer-val] */
617                 cfs_hash_rehash_key(exp->exp_lock_hash,
618                                     &lock->l_remote_handle,
619                                     &reply->lock_handle,
620                                     &lock->l_exp_hash);
621         } else {
622                 lock->l_remote_handle = reply->lock_handle;
623         }
624
625         *flags = ldlm_flags_from_wire(reply->lock_flags);
626         lock->l_flags |= ldlm_flags_from_wire(reply->lock_flags &
627                                               LDLM_FL_INHERIT_MASK);
628         unlock_res_and_lock(lock);
629
630         CDEBUG(D_INFO, "local: %p, remote cookie: "LPX64", flags: "LPX64"\n",
631                lock, reply->lock_handle.cookie, *flags);
632
633         /* If enqueue returned a blocked lock but the completion handler has
634          * already run, then it fixed up the resource and we don't need to do it
635          * again. */
636         if ((*flags) & LDLM_FL_LOCK_CHANGED) {
637                 int newmode = reply->lock_desc.l_req_mode;
638                 LASSERT(!is_replay);
639                 if (newmode && newmode != lock->l_req_mode) {
640                         LDLM_DEBUG(lock, "server returned different mode %s",
641                                    ldlm_lockname[newmode]);
642                         lock->l_req_mode = newmode;
643                 }
644
645                 if (!ldlm_res_eq(&reply->lock_desc.l_resource.lr_name,
646                                  &lock->l_resource->lr_name)) {
647                         CDEBUG(D_INFO, "remote intent success, locking "DLDLMRES
648                                        " instead of "DLDLMRES"\n",
649                                PLDLMRES(&reply->lock_desc.l_resource),
650                                PLDLMRES(lock->l_resource));
651
652                         rc = ldlm_lock_change_resource(ns, lock,
653                                         &reply->lock_desc.l_resource.lr_name);
654                         if (rc || lock->l_resource == NULL)
655                                 GOTO(cleanup, rc = -ENOMEM);
656                         LDLM_DEBUG(lock, "client-side enqueue, new resource");
657                 }
658                 if (with_policy)
659                         if (!(type == LDLM_IBITS &&
660                               !(exp_connect_flags(exp) & OBD_CONNECT_IBITS)))
661                                 /* We assume lock type cannot change on server*/
662                                 ldlm_convert_policy_to_local(exp,
663                                                 lock->l_resource->lr_type,
664                                                 &reply->lock_desc.l_policy_data,
665                                                 &lock->l_policy_data);
666                 if (type != LDLM_PLAIN)
667                         LDLM_DEBUG(lock,"client-side enqueue, new policy data");
668         }
669
670         if ((*flags) & LDLM_FL_AST_SENT) {
671                 lock_res_and_lock(lock);
672                 lock->l_flags |= LDLM_FL_CBPENDING | LDLM_FL_BL_AST;
673                 unlock_res_and_lock(lock);
674                 LDLM_DEBUG(lock, "enqueue reply includes blocking AST");
675         }
676
677         /* If the lock has already been granted by a completion AST, don't
678          * clobber the LVB with an older one. */
679         if (lvb_len > 0) {
680                 /* We must lock or a racing completion might update lvb without
681                  * letting us know and we'll clobber the correct value.
682                  * Cannot unlock after the check either, a that still leaves
683                  * a tiny window for completion to get in */
684                 lock_res_and_lock(lock);
685                 if (lock->l_req_mode != lock->l_granted_mode)
686                         rc = ldlm_fill_lvb(lock, &req->rq_pill, RCL_SERVER,
687                                            lock->l_lvb_data, lvb_len);
688                 unlock_res_and_lock(lock);
689                 if (rc < 0) {
690                         cleanup_phase = 1;
691                         GOTO(cleanup, rc);
692                 }
693         }
694
695         if (!is_replay) {
696                 rc = ldlm_lock_enqueue(ns, &lock, NULL, flags);
697                 if (lock->l_completion_ast != NULL) {
698                         int err = lock->l_completion_ast(lock, *flags, NULL);
699                         if (!rc)
700                                 rc = err;
701                         if (rc)
702                                 cleanup_phase = 1;
703                 }
704         }
705
706         if (lvb_len > 0 && lvb != NULL) {
707                 /* Copy the LVB here, and not earlier, because the completion
708                  * AST (if any) can override what we got in the reply */
709                 memcpy(lvb, lock->l_lvb_data, lvb_len);
710         }
711
712         LDLM_DEBUG(lock, "client-side enqueue END");
713         EXIT;
714 cleanup:
715         if (cleanup_phase == 1 && rc)
716                 failed_lock_cleanup(ns, lock, mode);
717         /* Put lock 2 times, the second reference is held by ldlm_cli_enqueue */
718         LDLM_LOCK_PUT(lock);
719         LDLM_LOCK_RELEASE(lock);
720         return rc;
721 }
722 EXPORT_SYMBOL(ldlm_cli_enqueue_fini);
723
724 /**
725  * Estimate number of lock handles that would fit into request of given
726  * size.  PAGE_SIZE-512 is to allow TCP/IP and LNET headers to fit into
727  * a single page on the send/receive side. XXX: 512 should be changed to
728  * more adequate value.
729  */
730 static inline int ldlm_req_handles_avail(int req_size, int off)
731 {
732         int avail;
733
734         avail = min_t(int, LDLM_MAXREQSIZE, PAGE_CACHE_SIZE - 512) - req_size;
735         if (likely(avail >= 0))
736                 avail /= (int)sizeof(struct lustre_handle);
737         else
738                 avail = 0;
739         avail += LDLM_LOCKREQ_HANDLES - off;
740
741         return avail;
742 }
743
744 static inline int ldlm_capsule_handles_avail(struct req_capsule *pill,
745                                              enum req_location loc,
746                                              int off)
747 {
748         __u32 size = req_capsule_msg_size(pill, loc);
749         return ldlm_req_handles_avail(size, off);
750 }
751
752 static inline int ldlm_format_handles_avail(struct obd_import *imp,
753                                             const struct req_format *fmt,
754                                             enum req_location loc, int off)
755 {
756         __u32 size = req_capsule_fmt_size(imp->imp_msg_magic, fmt, loc);
757         return ldlm_req_handles_avail(size, off);
758 }
759
760 /**
761  * Cancel LRU locks and pack them into the enqueue request. Pack there the given
762  * \a count locks in \a cancels.
763  *
764  * This is to be called by functions preparing their own requests that
765  * might contain lists of locks to cancel in addition to actual operation
766  * that needs to be performed.
767  */
768 int ldlm_prep_elc_req(struct obd_export *exp, struct ptlrpc_request *req,
769                       int version, int opc, int canceloff,
770                       struct list_head *cancels, int count)
771         {
772         struct ldlm_namespace   *ns = exp->exp_obd->obd_namespace;
773         struct req_capsule      *pill = &req->rq_pill;
774         struct ldlm_request     *dlm = NULL;
775         struct list_head        head = LIST_HEAD_INIT(head);
776         int flags, avail, to_free, pack = 0;
777         int rc;
778         ENTRY;
779
780         if (cancels == NULL)
781                 cancels = &head;
782         if (ns_connect_cancelset(ns)) {
783                 /* Estimate the amount of available space in the request. */
784                 req_capsule_filled_sizes(pill, RCL_CLIENT);
785                 avail = ldlm_capsule_handles_avail(pill, RCL_CLIENT, canceloff);
786
787                 flags = ns_connect_lru_resize(ns) ?
788                         LDLM_CANCEL_LRUR : LDLM_CANCEL_AGED;
789                 to_free = !ns_connect_lru_resize(ns) &&
790                           opc == LDLM_ENQUEUE ? 1 : 0;
791
792                 /* Cancel LRU locks here _only_ if the server supports
793                  * EARLY_CANCEL. Otherwise we have to send extra CANCEL
794                  * RPC, which will make us slower. */
795                 if (avail > count)
796                         count += ldlm_cancel_lru_local(ns, cancels, to_free,
797                                                        avail - count, 0, flags);
798                 if (avail > count)
799                         pack = count;
800                 else
801                         pack = avail;
802                 req_capsule_set_size(pill, &RMF_DLM_REQ, RCL_CLIENT,
803                                      ldlm_request_bufsize(pack, opc));
804         }
805
806         rc = ptlrpc_request_pack(req, version, opc);
807         if (rc) {
808                 ldlm_lock_list_put(cancels, l_bl_ast, count);
809                 RETURN(rc);
810         }
811
812         if (ns_connect_cancelset(ns)) {
813                 if (canceloff) {
814                         dlm = req_capsule_client_get(pill, &RMF_DLM_REQ);
815                         LASSERT(dlm);
816                         /* Skip first lock handler in ldlm_request_pack(),
817                          * this method will incrment @lock_count according
818                          * to the lock handle amount actually written to
819                          * the buffer. */
820                         dlm->lock_count = canceloff;
821                 }
822                 /* Pack into the request @pack lock handles. */
823                 ldlm_cli_cancel_list(cancels, pack, req, 0);
824                 /* Prepare and send separate cancel RPC for others. */
825                 ldlm_cli_cancel_list(cancels, count - pack, NULL, 0);
826         } else {
827                 ldlm_lock_list_put(cancels, l_bl_ast, count);
828         }
829         RETURN(0);
830 }
831 EXPORT_SYMBOL(ldlm_prep_elc_req);
832
833 int ldlm_prep_enqueue_req(struct obd_export *exp, struct ptlrpc_request *req,
834                           struct list_head *cancels, int count)
835 {
836         return ldlm_prep_elc_req(exp, req, LUSTRE_DLM_VERSION, LDLM_ENQUEUE,
837                                  LDLM_ENQUEUE_CANCEL_OFF, cancels, count);
838 }
839 EXPORT_SYMBOL(ldlm_prep_enqueue_req);
840
841 struct ptlrpc_request *ldlm_enqueue_pack(struct obd_export *exp, int lvb_len)
842 {
843         struct ptlrpc_request *req;
844         int rc;
845         ENTRY;
846
847         req = ptlrpc_request_alloc(class_exp2cliimp(exp), &RQF_LDLM_ENQUEUE);
848         if (req == NULL)
849                 RETURN(ERR_PTR(-ENOMEM));
850
851         rc = ldlm_prep_enqueue_req(exp, req, NULL, 0);
852         if (rc) {
853                 ptlrpc_request_free(req);
854                 RETURN(ERR_PTR(rc));
855         }
856
857         req_capsule_set_size(&req->rq_pill, &RMF_DLM_LVB, RCL_SERVER, lvb_len);
858         ptlrpc_request_set_replen(req);
859         RETURN(req);
860 }
861 EXPORT_SYMBOL(ldlm_enqueue_pack);
862
863 /**
864  * Client-side lock enqueue.
865  *
866  * If a request has some specific initialisation it is passed in \a reqp,
867  * otherwise it is created in ldlm_cli_enqueue.
868  *
869  * Supports sync and async requests, pass \a async flag accordingly. If a
870  * request was created in ldlm_cli_enqueue and it is the async request,
871  * pass it to the caller in \a reqp.
872  */
873 int ldlm_cli_enqueue(struct obd_export *exp, struct ptlrpc_request **reqp,
874                      struct ldlm_enqueue_info *einfo,
875                      const struct ldlm_res_id *res_id,
876                      ldlm_policy_data_t const *policy, __u64 *flags,
877                      void *lvb, __u32 lvb_len, enum lvb_type lvb_type,
878                      struct lustre_handle *lockh, int async)
879 {
880         struct ldlm_namespace *ns;
881         struct ldlm_lock      *lock;
882         struct ldlm_request   *body;
883         int                    is_replay = *flags & LDLM_FL_REPLAY;
884         int                    req_passed_in = 1;
885         int                    rc, err;
886         struct ptlrpc_request *req;
887         ENTRY;
888
889         LASSERT(exp != NULL);
890
891         ns = exp->exp_obd->obd_namespace;
892
893         /* If we're replaying this lock, just check some invariants.
894          * If we're creating a new lock, get everything all setup nice. */
895         if (is_replay) {
896                 lock = ldlm_handle2lock_long(lockh, 0);
897                 LASSERT(lock != NULL);
898                 LDLM_DEBUG(lock, "client-side enqueue START");
899                 LASSERT(exp == lock->l_conn_export);
900         } else {
901                 const struct ldlm_callback_suite cbs = {
902                         .lcs_completion = einfo->ei_cb_cp,
903                         .lcs_blocking   = einfo->ei_cb_bl,
904                         .lcs_glimpse    = einfo->ei_cb_gl
905                 };
906                 lock = ldlm_lock_create(ns, res_id, einfo->ei_type,
907                                         einfo->ei_mode, &cbs, einfo->ei_cbdata,
908                                         lvb_len, lvb_type);
909                 if (IS_ERR(lock))
910                         RETURN(PTR_ERR(lock));
911                 /* for the local lock, add the reference */
912                 ldlm_lock_addref_internal(lock, einfo->ei_mode);
913                 ldlm_lock2handle(lock, lockh);
914                 if (policy != NULL)
915                         lock->l_policy_data = *policy;
916
917                 if (einfo->ei_type == LDLM_EXTENT) {
918                         /* extent lock without policy is a bug */
919                         if (policy == NULL)
920                                 LBUG();
921
922                         lock->l_req_extent = policy->l_extent;
923                 }
924                 LDLM_DEBUG(lock, "client-side enqueue START, flags "LPX64"\n",
925                            *flags);
926         }
927
928         lock->l_conn_export = exp;
929         lock->l_export = NULL;
930         lock->l_blocking_ast = einfo->ei_cb_bl;
931         lock->l_flags |= (*flags & (LDLM_FL_NO_LRU | LDLM_FL_EXCL));
932         lock->l_last_activity = cfs_time_current_sec();
933
934         /* lock not sent to server yet */
935
936         if (reqp == NULL || *reqp == NULL) {
937                 req = ptlrpc_request_alloc_pack(class_exp2cliimp(exp),
938                                                 &RQF_LDLM_ENQUEUE,
939                                                 LUSTRE_DLM_VERSION,
940                                                 LDLM_ENQUEUE);
941                 if (req == NULL) {
942                         failed_lock_cleanup(ns, lock, einfo->ei_mode);
943                         LDLM_LOCK_RELEASE(lock);
944                         RETURN(-ENOMEM);
945                 }
946                 req_passed_in = 0;
947                 if (reqp)
948                         *reqp = req;
949         } else {
950                 int len;
951
952                 req = *reqp;
953                 len = req_capsule_get_size(&req->rq_pill, &RMF_DLM_REQ,
954                                            RCL_CLIENT);
955                 LASSERTF(len >= sizeof(*body), "buflen[%d] = %d, not %d\n",
956                          DLM_LOCKREQ_OFF, len, (int)sizeof(*body));
957         }
958
959         /* Dump lock data into the request buffer */
960         body = req_capsule_client_get(&req->rq_pill, &RMF_DLM_REQ);
961         ldlm_lock2desc(lock, &body->lock_desc);
962         body->lock_flags = ldlm_flags_to_wire(*flags);
963         body->lock_handle[0] = *lockh;
964
965         /* Continue as normal. */
966         if (!req_passed_in) {
967                 if (lvb_len > 0)
968                         req_capsule_extend(&req->rq_pill,
969                                            &RQF_LDLM_ENQUEUE_LVB);
970                 req_capsule_set_size(&req->rq_pill, &RMF_DLM_LVB, RCL_SERVER,
971                                      lvb_len);
972                 ptlrpc_request_set_replen(req);
973         }
974
975         if (async) {
976                 LASSERT(reqp != NULL);
977                 RETURN(0);
978         }
979
980         LDLM_DEBUG(lock, "sending request");
981
982         rc = ptlrpc_queue_wait(req);
983
984         err = ldlm_cli_enqueue_fini(exp, req, einfo->ei_type, policy ? 1 : 0,
985                                     einfo->ei_mode, flags, lvb, lvb_len,
986                                     lockh, rc);
987
988         /* If ldlm_cli_enqueue_fini did not find the lock, we need to free
989          * one reference that we took */
990         if (err == -ENOLCK)
991                 LDLM_LOCK_RELEASE(lock);
992         else
993                 rc = err;
994
995         if (!req_passed_in && req != NULL) {
996                 ptlrpc_req_finished(req);
997                 if (reqp)
998                         *reqp = NULL;
999         }
1000
1001         RETURN(rc);
1002 }
1003 EXPORT_SYMBOL(ldlm_cli_enqueue);
1004
1005 static int ldlm_cli_convert_local(struct ldlm_lock *lock, int new_mode,
1006                                   __u32 *flags)
1007 {
1008         struct ldlm_resource *res;
1009         int rc;
1010         ENTRY;
1011         if (ns_is_client(ldlm_lock_to_ns(lock))) {
1012                 CERROR("Trying to cancel local lock\n");
1013                 LBUG();
1014         }
1015         LDLM_DEBUG(lock, "client-side local convert");
1016
1017         res = ldlm_lock_convert(lock, new_mode, flags);
1018         if (res) {
1019                 ldlm_reprocess_all(res);
1020                 rc = 0;
1021         } else {
1022                 rc = LUSTRE_EDEADLK;
1023         }
1024         LDLM_DEBUG(lock, "client-side local convert handler END");
1025         LDLM_LOCK_PUT(lock);
1026         RETURN(rc);
1027 }
1028
1029 /* FIXME: one of ldlm_cli_convert or the server side should reject attempted
1030  * conversion of locks which are on the waiting or converting queue */
1031 /* Caller of this code is supposed to take care of lock readers/writers
1032    accounting */
1033 int ldlm_cli_convert(struct lustre_handle *lockh, int new_mode, __u32 *flags)
1034 {
1035         struct ldlm_request   *body;
1036         struct ldlm_reply     *reply;
1037         struct ldlm_lock      *lock;
1038         struct ldlm_resource  *res;
1039         struct ptlrpc_request *req;
1040         int                    rc;
1041         ENTRY;
1042
1043         lock = ldlm_handle2lock(lockh);
1044         if (!lock) {
1045                 LBUG();
1046                 RETURN(-EINVAL);
1047         }
1048         *flags = 0;
1049
1050         if (lock->l_conn_export == NULL)
1051                 RETURN(ldlm_cli_convert_local(lock, new_mode, flags));
1052
1053         LDLM_DEBUG(lock, "client-side convert");
1054
1055         req = ptlrpc_request_alloc_pack(class_exp2cliimp(lock->l_conn_export),
1056                                         &RQF_LDLM_CONVERT, LUSTRE_DLM_VERSION,
1057                                         LDLM_CONVERT);
1058         if (req == NULL) {
1059                 LDLM_LOCK_PUT(lock);
1060                 RETURN(-ENOMEM);
1061         }
1062
1063         body = req_capsule_client_get(&req->rq_pill, &RMF_DLM_REQ);
1064         body->lock_handle[0] = lock->l_remote_handle;
1065
1066         body->lock_desc.l_req_mode = new_mode;
1067         body->lock_flags = ldlm_flags_to_wire(*flags);
1068
1069
1070         ptlrpc_request_set_replen(req);
1071         rc = ptlrpc_queue_wait(req);
1072         if (rc != ELDLM_OK)
1073                 GOTO(out, rc);
1074
1075         reply = req_capsule_server_get(&req->rq_pill, &RMF_DLM_REP);
1076         if (reply == NULL)
1077                 GOTO(out, rc = -EPROTO);
1078
1079         if (req->rq_status)
1080                 GOTO(out, rc = req->rq_status);
1081
1082         res = ldlm_lock_convert(lock, new_mode, &reply->lock_flags);
1083         if (res != NULL) {
1084                 ldlm_reprocess_all(res);
1085                 /* Go to sleep until the lock is granted. */
1086                 /* FIXME: or cancelled. */
1087                 if (lock->l_completion_ast) {
1088                         rc = lock->l_completion_ast(lock, LDLM_FL_WAIT_NOREPROC,
1089                                                     NULL);
1090                         if (rc)
1091                                 GOTO(out, rc);
1092                 }
1093         } else {
1094                 rc = LUSTRE_EDEADLK;
1095         }
1096         EXIT;
1097  out:
1098         LDLM_LOCK_PUT(lock);
1099         ptlrpc_req_finished(req);
1100         return rc;
1101 }
1102 EXPORT_SYMBOL(ldlm_cli_convert);
1103
1104 /**
1105  * Cancel locks locally.
1106  * Returns:
1107  * \retval LDLM_FL_LOCAL_ONLY if there is no need for a CANCEL RPC to the server
1108  * \retval LDLM_FL_CANCELING otherwise;
1109  * \retval LDLM_FL_BL_AST if there is a need for a separate CANCEL RPC.
1110  */
1111 static __u64 ldlm_cli_cancel_local(struct ldlm_lock *lock)
1112 {
1113         __u64 rc = LDLM_FL_LOCAL_ONLY;
1114         ENTRY;
1115
1116         if (lock->l_conn_export) {
1117                 bool local_only;
1118
1119                 LDLM_DEBUG(lock, "client-side cancel");
1120                 /* Set this flag to prevent others from getting new references*/
1121                 lock_res_and_lock(lock);
1122                 ldlm_set_cbpending(lock);
1123                 local_only = !!(lock->l_flags &
1124                                 (LDLM_FL_LOCAL_ONLY|LDLM_FL_CANCEL_ON_BLOCK));
1125                 ldlm_cancel_callback(lock);
1126                 rc = (ldlm_is_bl_ast(lock)) ?
1127                         LDLM_FL_BL_AST : LDLM_FL_CANCELING;
1128                 unlock_res_and_lock(lock);
1129
1130                 if (local_only) {
1131                         CDEBUG(D_DLMTRACE, "not sending request (at caller's "
1132                                "instruction)\n");
1133                         rc = LDLM_FL_LOCAL_ONLY;
1134                 }
1135                 ldlm_lock_cancel(lock);
1136         } else {
1137                 if (ns_is_client(ldlm_lock_to_ns(lock))) {
1138                         LDLM_ERROR(lock, "Trying to cancel local lock");
1139                         LBUG();
1140                 }
1141                 LDLM_DEBUG(lock, "server-side local cancel");
1142                 ldlm_lock_cancel(lock);
1143                 ldlm_reprocess_all(lock->l_resource);
1144         }
1145
1146         RETURN(rc);
1147 }
1148
1149 /**
1150  * Pack \a count locks in \a head into ldlm_request buffer of request \a req.
1151  */
1152 static void ldlm_cancel_pack(struct ptlrpc_request *req,
1153                              struct list_head *head, int count)
1154 {
1155         struct ldlm_request *dlm;
1156         struct ldlm_lock *lock;
1157         int max, packed = 0;
1158         ENTRY;
1159
1160         dlm = req_capsule_client_get(&req->rq_pill, &RMF_DLM_REQ);
1161         LASSERT(dlm != NULL);
1162
1163         /* Check the room in the request buffer. */
1164         max = req_capsule_get_size(&req->rq_pill, &RMF_DLM_REQ, RCL_CLIENT) -
1165                 sizeof(struct ldlm_request);
1166         max /= sizeof(struct lustre_handle);
1167         max += LDLM_LOCKREQ_HANDLES;
1168         LASSERT(max >= dlm->lock_count + count);
1169
1170         /* XXX: it would be better to pack lock handles grouped by resource.
1171          * so that the server cancel would call filter_lvbo_update() less
1172          * frequently. */
1173         list_for_each_entry(lock, head, l_bl_ast) {
1174                 if (!count--)
1175                         break;
1176                 LASSERT(lock->l_conn_export);
1177                 /* Pack the lock handle to the given request buffer. */
1178                 LDLM_DEBUG(lock, "packing");
1179                 dlm->lock_handle[dlm->lock_count++] = lock->l_remote_handle;
1180                 packed++;
1181         }
1182         CDEBUG(D_DLMTRACE, "%d locks packed\n", packed);
1183         EXIT;
1184 }
1185
1186 /**
1187  * Prepare and send a batched cancel RPC. It will include \a count lock
1188  * handles of locks given in \a cancels list. */
1189 int ldlm_cli_cancel_req(struct obd_export *exp, struct list_head *cancels,
1190                         int count, ldlm_cancel_flags_t flags)
1191 {
1192         struct ptlrpc_request *req = NULL;
1193         struct obd_import *imp;
1194         int free, sent = 0;
1195         int rc = 0;
1196         ENTRY;
1197
1198         LASSERT(exp != NULL);
1199         LASSERT(count > 0);
1200
1201         CFS_FAIL_TIMEOUT(OBD_FAIL_LDLM_PAUSE_CANCEL, cfs_fail_val);
1202
1203         if (CFS_FAIL_CHECK(OBD_FAIL_LDLM_CANCEL_RACE))
1204                 RETURN(count);
1205
1206         free = ldlm_format_handles_avail(class_exp2cliimp(exp),
1207                                          &RQF_LDLM_CANCEL, RCL_CLIENT, 0);
1208         if (count > free)
1209                 count = free;
1210
1211         while (1) {
1212                 imp = class_exp2cliimp(exp);
1213                 if (imp == NULL || imp->imp_invalid) {
1214                         CDEBUG(D_DLMTRACE,
1215                                "skipping cancel on invalid import %p\n", imp);
1216                         RETURN(count);
1217                 }
1218
1219                 req = ptlrpc_request_alloc(imp, &RQF_LDLM_CANCEL);
1220                 if (req == NULL)
1221                         GOTO(out, rc = -ENOMEM);
1222
1223                 req_capsule_filled_sizes(&req->rq_pill, RCL_CLIENT);
1224                 req_capsule_set_size(&req->rq_pill, &RMF_DLM_REQ, RCL_CLIENT,
1225                                      ldlm_request_bufsize(count, LDLM_CANCEL));
1226
1227                 rc = ptlrpc_request_pack(req, LUSTRE_DLM_VERSION, LDLM_CANCEL);
1228                 if (rc) {
1229                         ptlrpc_request_free(req);
1230                         GOTO(out, rc);
1231                 }
1232
1233                 req->rq_request_portal = LDLM_CANCEL_REQUEST_PORTAL;
1234                 req->rq_reply_portal = LDLM_CANCEL_REPLY_PORTAL;
1235                 ptlrpc_at_set_req_timeout(req);
1236
1237                 ldlm_cancel_pack(req, cancels, count);
1238
1239                 ptlrpc_request_set_replen(req);
1240                 if (flags & LCF_ASYNC) {
1241                         ptlrpcd_add_req(req, PDL_POLICY_LOCAL, -1);
1242                         sent = count;
1243                         GOTO(out, 0);
1244                 } else {
1245                         rc = ptlrpc_queue_wait(req);
1246                 }
1247                 if (rc == LUSTRE_ESTALE) {
1248                         CDEBUG(D_DLMTRACE, "client/server (nid %s) "
1249                                "out of sync -- not fatal\n",
1250                                libcfs_nid2str(req->rq_import->
1251                                               imp_connection->c_peer.nid));
1252                         rc = 0;
1253                 } else if (rc == -ETIMEDOUT && /* check there was no reconnect*/
1254                            req->rq_import_generation == imp->imp_generation) {
1255                         ptlrpc_req_finished(req);
1256                         continue;
1257                 } else if (rc != ELDLM_OK) {
1258                         /* -ESHUTDOWN is common on umount */
1259                         CDEBUG_LIMIT(rc == -ESHUTDOWN ? D_DLMTRACE : D_ERROR,
1260                                      "Got rc %d from cancel RPC: "
1261                                      "canceling anyway\n", rc);
1262                         break;
1263                 }
1264                 sent = count;
1265                 break;
1266         }
1267
1268         ptlrpc_req_finished(req);
1269         EXIT;
1270 out:
1271         return sent ? sent : rc;
1272 }
1273 EXPORT_SYMBOL(ldlm_cli_cancel_req);
1274
1275 static inline struct ldlm_pool *ldlm_imp2pl(struct obd_import *imp)
1276 {
1277         LASSERT(imp != NULL);
1278         return &imp->imp_obd->obd_namespace->ns_pool;
1279 }
1280
1281 /**
1282  * Update client's OBD pool related fields with new SLV and Limit from \a req.
1283  */
1284 int ldlm_cli_update_pool(struct ptlrpc_request *req)
1285 {
1286         struct obd_device *obd;
1287         __u64 new_slv;
1288         __u32 new_limit;
1289         ENTRY;
1290         if (unlikely(!req->rq_import || !req->rq_import->imp_obd ||
1291                      !imp_connect_lru_resize(req->rq_import)))
1292         {
1293                 /*
1294                  * Do nothing for corner cases.
1295                  */
1296                 RETURN(0);
1297         }
1298
1299         /* In some cases RPC may contain SLV and limit zeroed out. This
1300          * is the case when server does not support LRU resize feature.
1301          * This is also possible in some recovery cases when server-side
1302          * reqs have no reference to the OBD export and thus access to
1303          * server-side namespace is not possible. */
1304         if (lustre_msg_get_slv(req->rq_repmsg) == 0 ||
1305             lustre_msg_get_limit(req->rq_repmsg) == 0) {
1306                 DEBUG_REQ(D_HA, req, "Zero SLV or Limit found "
1307                           "(SLV: "LPU64", Limit: %u)",
1308                           lustre_msg_get_slv(req->rq_repmsg),
1309                           lustre_msg_get_limit(req->rq_repmsg));
1310                 RETURN(0);
1311         }
1312
1313         new_limit = lustre_msg_get_limit(req->rq_repmsg);
1314         new_slv = lustre_msg_get_slv(req->rq_repmsg);
1315         obd = req->rq_import->imp_obd;
1316
1317         /* Set new SLV and limit in OBD fields to make them accessible
1318          * to the pool thread. We do not access obd_namespace and pool
1319          * directly here as there is no reliable way to make sure that
1320          * they are still alive at cleanup time. Evil races are possible
1321          * which may cause Oops at that time. */
1322         write_lock(&obd->obd_pool_lock);
1323         obd->obd_pool_slv = new_slv;
1324         obd->obd_pool_limit = new_limit;
1325         write_unlock(&obd->obd_pool_lock);
1326
1327         RETURN(0);
1328 }
1329 EXPORT_SYMBOL(ldlm_cli_update_pool);
1330
1331 /**
1332  * Client side lock cancel.
1333  *
1334  * Lock must not have any readers or writers by this time.
1335  */
1336 int ldlm_cli_cancel(struct lustre_handle *lockh,
1337                     ldlm_cancel_flags_t cancel_flags)
1338 {
1339         struct obd_export *exp;
1340         int avail, flags, count = 1;
1341         __u64 rc = 0;
1342         struct ldlm_namespace *ns;
1343         struct ldlm_lock *lock;
1344         struct list_head cancels = LIST_HEAD_INIT(cancels);
1345         ENTRY;
1346
1347         /* concurrent cancels on the same handle can happen */
1348         lock = ldlm_handle2lock_long(lockh, LDLM_FL_CANCELING);
1349         if (lock == NULL) {
1350                 LDLM_DEBUG_NOLOCK("lock is already being destroyed");
1351                 RETURN(0);
1352         }
1353
1354         rc = ldlm_cli_cancel_local(lock);
1355         if (rc == LDLM_FL_LOCAL_ONLY || cancel_flags & LCF_LOCAL) {
1356                 LDLM_LOCK_RELEASE(lock);
1357                 RETURN(0);
1358         }
1359         /* Even if the lock is marked as LDLM_FL_BL_AST, this is a LDLM_CANCEL
1360          * RPC which goes to canceld portal, so we can cancel other LRU locks
1361          * here and send them all as one LDLM_CANCEL RPC. */
1362         LASSERT(list_empty(&lock->l_bl_ast));
1363         list_add(&lock->l_bl_ast, &cancels);
1364
1365         exp = lock->l_conn_export;
1366         if (exp_connect_cancelset(exp)) {
1367                 avail = ldlm_format_handles_avail(class_exp2cliimp(exp),
1368                                                   &RQF_LDLM_CANCEL,
1369                                                   RCL_CLIENT, 0);
1370                 LASSERT(avail > 0);
1371
1372                 ns = ldlm_lock_to_ns(lock);
1373                 flags = ns_connect_lru_resize(ns) ?
1374                         LDLM_CANCEL_LRUR : LDLM_CANCEL_AGED;
1375                 count += ldlm_cancel_lru_local(ns, &cancels, 0, avail - 1,
1376                                                LCF_BL_AST, flags);
1377         }
1378         ldlm_cli_cancel_list(&cancels, count, NULL, cancel_flags);
1379         RETURN(0);
1380 }
1381 EXPORT_SYMBOL(ldlm_cli_cancel);
1382
1383 /**
1384  * Locally cancel up to \a count locks in list \a cancels.
1385  * Return the number of cancelled locks.
1386  */
1387 int ldlm_cli_cancel_list_local(struct list_head *cancels, int count,
1388                                ldlm_cancel_flags_t flags)
1389 {
1390         struct list_head head = LIST_HEAD_INIT(head);
1391         struct ldlm_lock *lock, *next;
1392         int left = 0, bl_ast = 0;
1393         __u64 rc;
1394
1395         left = count;
1396         list_for_each_entry_safe(lock, next, cancels, l_bl_ast) {
1397                 if (left-- == 0)
1398                         break;
1399
1400                 if (flags & LCF_LOCAL) {
1401                         rc = LDLM_FL_LOCAL_ONLY;
1402                         ldlm_lock_cancel(lock);
1403                 } else {
1404                         rc = ldlm_cli_cancel_local(lock);
1405                 }
1406                 /* Until we have compound requests and can send LDLM_CANCEL
1407                  * requests batched with generic RPCs, we need to send cancels
1408                  * with the LDLM_FL_BL_AST flag in a separate RPC from
1409                  * the one being generated now. */
1410                 if (!(flags & LCF_BL_AST) && (rc == LDLM_FL_BL_AST)) {
1411                         LDLM_DEBUG(lock, "Cancel lock separately");
1412                         list_del_init(&lock->l_bl_ast);
1413                         list_add(&lock->l_bl_ast, &head);
1414                         bl_ast++;
1415                         continue;
1416                 }
1417                 if (rc == LDLM_FL_LOCAL_ONLY) {
1418                         /* CANCEL RPC should not be sent to server. */
1419                         list_del_init(&lock->l_bl_ast);
1420                         LDLM_LOCK_RELEASE(lock);
1421                         count--;
1422                 }
1423         }
1424         if (bl_ast > 0) {
1425                 count -= bl_ast;
1426                 ldlm_cli_cancel_list(&head, bl_ast, NULL, 0);
1427         }
1428
1429         RETURN(count);
1430 }
1431 EXPORT_SYMBOL(ldlm_cli_cancel_list_local);
1432
1433 /**
1434  * Cancel as many locks as possible w/o sending any RPCs (e.g. to write back
1435  * dirty data, to close a file, ...) or waiting for any RPCs in-flight (e.g.
1436  * readahead requests, ...)
1437  */
1438 static ldlm_policy_res_t ldlm_cancel_no_wait_policy(struct ldlm_namespace *ns,
1439                                                     struct ldlm_lock *lock,
1440                                                     int unused, int added,
1441                                                     int count)
1442 {
1443         ldlm_policy_res_t result = LDLM_POLICY_CANCEL_LOCK;
1444
1445         /* don't check added & count since we want to process all locks
1446          * from unused list.
1447          * It's fine to not take lock to access lock->l_resource since
1448          * the lock has already been granted so it won't change. */
1449         switch (lock->l_resource->lr_type) {
1450                 case LDLM_EXTENT:
1451                 case LDLM_IBITS:
1452                         if (ns->ns_cancel != NULL && ns->ns_cancel(lock) != 0)
1453                                 break;
1454                 default:
1455                         result = LDLM_POLICY_SKIP_LOCK;
1456                         lock_res_and_lock(lock);
1457                         ldlm_set_skipped(lock);
1458                         unlock_res_and_lock(lock);
1459                         break;
1460         }
1461
1462         RETURN(result);
1463 }
1464
1465 /**
1466  * Callback function for LRU-resize policy. Decides whether to keep
1467  * \a lock in LRU for current \a LRU size \a unused, added in current
1468  * scan \a added and number of locks to be preferably canceled \a count.
1469  *
1470  * \retval LDLM_POLICY_KEEP_LOCK keep lock in LRU in stop scanning
1471  *
1472  * \retval LDLM_POLICY_CANCEL_LOCK cancel lock from LRU
1473  */
1474 static ldlm_policy_res_t ldlm_cancel_lrur_policy(struct ldlm_namespace *ns,
1475                                                  struct ldlm_lock *lock,
1476                                                  int unused, int added,
1477                                                  int count)
1478 {
1479         cfs_time_t cur = cfs_time_current();
1480         struct ldlm_pool *pl = &ns->ns_pool;
1481         __u64 slv, lvf, lv;
1482         cfs_time_t la;
1483
1484         /* Stop LRU processing when we reach past @count or have checked all
1485          * locks in LRU. */
1486         if (count && added >= count)
1487                 return LDLM_POLICY_KEEP_LOCK;
1488
1489         slv = ldlm_pool_get_slv(pl);
1490         lvf = ldlm_pool_get_lvf(pl);
1491         la = cfs_duration_sec(cfs_time_sub(cur,
1492                               lock->l_last_used));
1493         lv = lvf * la * unused;
1494
1495         /* Inform pool about current CLV to see it via proc. */
1496         ldlm_pool_set_clv(pl, lv);
1497
1498         /* Stop when SLV is not yet come from server or lv is smaller than
1499          * it is. */
1500         if (slv == 0 || lv < slv)
1501                 return LDLM_POLICY_KEEP_LOCK;
1502
1503         return LDLM_POLICY_CANCEL_LOCK;
1504 }
1505
1506 /**
1507  * Callback function for proc used policy. Makes decision whether to keep
1508  * \a lock in LRU for current \a LRU size \a unused, added in current scan \a
1509  * added and number of locks to be preferably canceled \a count.
1510  *
1511  * \retval LDLM_POLICY_KEEP_LOCK keep lock in LRU in stop scanning
1512  *
1513  * \retval LDLM_POLICY_CANCEL_LOCK cancel lock from LRU
1514  */
1515 static ldlm_policy_res_t ldlm_cancel_passed_policy(struct ldlm_namespace *ns,
1516                                                    struct ldlm_lock *lock,
1517                                                    int unused, int added,
1518                                                    int count)
1519 {
1520         /* Stop LRU processing when we reach past @count or have checked all
1521          * locks in LRU. */
1522         return (added >= count) ?
1523                 LDLM_POLICY_KEEP_LOCK : LDLM_POLICY_CANCEL_LOCK;
1524 }
1525
1526 /**
1527  * Callback function for aged policy. Makes decision whether to keep \a lock in
1528  * LRU for current LRU size \a unused, added in current scan \a added and
1529  * number of locks to be preferably canceled \a count.
1530  *
1531  * \retval LDLM_POLICY_KEEP_LOCK keep lock in LRU in stop scanning
1532  *
1533  * \retval LDLM_POLICY_CANCEL_LOCK cancel lock from LRU
1534  */
1535 static ldlm_policy_res_t ldlm_cancel_aged_policy(struct ldlm_namespace *ns,
1536                                                  struct ldlm_lock *lock,
1537                                                  int unused, int added,
1538                                                  int count)
1539 {
1540         if ((added >= count) &&
1541             cfs_time_before(cfs_time_current(),
1542                             cfs_time_add(lock->l_last_used, ns->ns_max_age)))
1543                 return LDLM_POLICY_KEEP_LOCK;
1544
1545         return LDLM_POLICY_CANCEL_LOCK;
1546 }
1547
1548 /**
1549  * Callback function for default policy. Makes decision whether to keep \a lock
1550  * in LRU for current LRU size \a unused, added in current scan \a added and
1551  * number of locks to be preferably canceled \a count.
1552  *
1553  * \retval LDLM_POLICY_KEEP_LOCK keep lock in LRU in stop scanning
1554  *
1555  * \retval LDLM_POLICY_CANCEL_LOCK cancel lock from LRU
1556  */
1557 static ldlm_policy_res_t ldlm_cancel_default_policy(struct ldlm_namespace *ns,
1558                                                     struct ldlm_lock *lock,
1559                                                     int unused, int added,
1560                                                     int count)
1561 {
1562         /* Stop LRU processing when we reach past count or have checked all
1563          * locks in LRU. */
1564         return (added >= count) ?
1565                 LDLM_POLICY_KEEP_LOCK : LDLM_POLICY_CANCEL_LOCK;
1566 }
1567
1568 typedef ldlm_policy_res_t (*ldlm_cancel_lru_policy_t)(struct ldlm_namespace *,
1569                                                       struct ldlm_lock *, int,
1570                                                       int, int);
1571
1572 static ldlm_cancel_lru_policy_t
1573 ldlm_cancel_lru_policy(struct ldlm_namespace *ns, int flags)
1574 {
1575         if (flags & LDLM_CANCEL_NO_WAIT)
1576                 return ldlm_cancel_no_wait_policy;
1577
1578         if (ns_connect_lru_resize(ns)) {
1579                 if (flags & LDLM_CANCEL_SHRINK)
1580                         /* We kill passed number of old locks. */
1581                         return ldlm_cancel_passed_policy;
1582                 else if (flags & LDLM_CANCEL_LRUR)
1583                         return ldlm_cancel_lrur_policy;
1584                 else if (flags & LDLM_CANCEL_PASSED)
1585                         return ldlm_cancel_passed_policy;
1586         } else {
1587                 if (flags & LDLM_CANCEL_AGED)
1588                         return ldlm_cancel_aged_policy;
1589         }
1590
1591         return ldlm_cancel_default_policy;
1592 }
1593
1594 /**
1595  * - Free space in LRU for \a count new locks,
1596  *   redundant unused locks are canceled locally;
1597  * - also cancel locally unused aged locks;
1598  * - do not cancel more than \a max locks;
1599  * - GET the found locks and add them into the \a cancels list.
1600  *
1601  * A client lock can be added to the l_bl_ast list only when it is
1602  * marked LDLM_FL_CANCELING. Otherwise, somebody is already doing
1603  * CANCEL.  There are the following use cases:
1604  * ldlm_cancel_resource_local(), ldlm_cancel_lru_local() and
1605  * ldlm_cli_cancel(), which check and set this flag properly. As any
1606  * attempt to cancel a lock rely on this flag, l_bl_ast list is accessed
1607  * later without any special locking.
1608  *
1609  * Calling policies for enabled LRU resize:
1610  * ----------------------------------------
1611  * flags & LDLM_CANCEL_LRUR - use LRU resize policy (SLV from server) to
1612  *                            cancel not more than \a count locks;
1613  *
1614  * flags & LDLM_CANCEL_PASSED - cancel \a count number of old locks (located at
1615  *                              the beginning of LRU list);
1616  *
1617  * flags & LDLM_CANCEL_SHRINK - cancel not more than \a count locks according to
1618  *                              memory pressre policy function;
1619  *
1620  * flags & LDLM_CANCEL_AGED - cancel \a count locks according to "aged policy".
1621  *
1622  * flags & LDLM_CANCEL_NO_WAIT - cancel as many unused locks as possible
1623  *                               (typically before replaying locks) w/o
1624  *                               sending any RPCs or waiting for any
1625  *                               outstanding RPC to complete.
1626  */
1627 static int ldlm_prepare_lru_list(struct ldlm_namespace *ns,
1628                                  struct list_head *cancels, int count, int max,
1629                                  int flags)
1630 {
1631         ldlm_cancel_lru_policy_t pf;
1632         struct ldlm_lock *lock, *next;
1633         int added = 0, unused, remained;
1634         ENTRY;
1635
1636         spin_lock(&ns->ns_lock);
1637         unused = ns->ns_nr_unused;
1638         remained = unused;
1639
1640         if (!ns_connect_lru_resize(ns))
1641                 count += unused - ns->ns_max_unused;
1642
1643         pf = ldlm_cancel_lru_policy(ns, flags);
1644         LASSERT(pf != NULL);
1645
1646         while (!list_empty(&ns->ns_unused_list)) {
1647                 ldlm_policy_res_t result;
1648
1649                 /* all unused locks */
1650                 if (remained-- <= 0)
1651                         break;
1652
1653                 /* For any flags, stop scanning if @max is reached. */
1654                 if (max && added >= max)
1655                         break;
1656
1657                 list_for_each_entry_safe(lock, next, &ns->ns_unused_list,
1658                                              l_lru) {
1659                         /* No locks which got blocking requests. */
1660                         LASSERT(!ldlm_is_bl_ast(lock));
1661
1662                         if (flags & LDLM_CANCEL_NO_WAIT &&
1663                             ldlm_is_skipped(lock))
1664                                 /* already processed */
1665                                 continue;
1666
1667                         /* Somebody is already doing CANCEL. No need for this
1668                          * lock in LRU, do not traverse it again. */
1669                         if (!ldlm_is_canceling(lock))
1670                                 break;
1671
1672                         ldlm_lock_remove_from_lru_nolock(lock);
1673                 }
1674                 if (&lock->l_lru == &ns->ns_unused_list)
1675                         break;
1676
1677                 LDLM_LOCK_GET(lock);
1678                 spin_unlock(&ns->ns_lock);
1679                 lu_ref_add(&lock->l_reference, __FUNCTION__, current);
1680
1681                 /* Pass the lock through the policy filter and see if it
1682                  * should stay in LRU.
1683                  *
1684                  * Even for shrinker policy we stop scanning if
1685                  * we find a lock that should stay in the cache.
1686                  * We should take into account lock age anyway
1687                  * as a new lock is a valuable resource even if
1688                  * it has a low weight.
1689                  *
1690                  * That is, for shrinker policy we drop only
1691                  * old locks, but additionally choose them by
1692                  * their weight. Big extent locks will stay in
1693                  * the cache. */
1694                 result = pf(ns, lock, unused, added, count);
1695                 if (result == LDLM_POLICY_KEEP_LOCK) {
1696                         lu_ref_del(&lock->l_reference,
1697                                    __FUNCTION__, current);
1698                         LDLM_LOCK_RELEASE(lock);
1699                         spin_lock(&ns->ns_lock);
1700                         break;
1701                 }
1702                 if (result == LDLM_POLICY_SKIP_LOCK) {
1703                         lu_ref_del(&lock->l_reference,
1704                                    __func__, current);
1705                         LDLM_LOCK_RELEASE(lock);
1706                         spin_lock(&ns->ns_lock);
1707                         continue;
1708                 }
1709
1710                 lock_res_and_lock(lock);
1711                 /* Check flags again under the lock. */
1712                 if (ldlm_is_canceling(lock) ||
1713                     (ldlm_lock_remove_from_lru(lock) == 0)) {
1714                         /* Another thread is removing lock from LRU, or
1715                          * somebody is already doing CANCEL, or there
1716                          * is a blocking request which will send cancel
1717                          * by itself, or the lock is no longer unused. */
1718                         unlock_res_and_lock(lock);
1719                         lu_ref_del(&lock->l_reference, __FUNCTION__, current);
1720                         LDLM_LOCK_RELEASE(lock);
1721                         spin_lock(&ns->ns_lock);
1722                         continue;
1723                 }
1724                 LASSERT(!lock->l_readers && !lock->l_writers);
1725
1726                 /* If we have chosen to cancel this lock voluntarily, we
1727                  * better send cancel notification to server, so that it
1728                  * frees appropriate state. This might lead to a race
1729                  * where while we are doing cancel here, server is also
1730                  * silently cancelling this lock. */
1731                 ldlm_clear_cancel_on_block(lock);
1732
1733                 /* Setting the CBPENDING flag is a little misleading,
1734                  * but prevents an important race; namely, once
1735                  * CBPENDING is set, the lock can accumulate no more
1736                  * readers/writers. Since readers and writers are
1737                  * already zero here, ldlm_lock_decref() won't see
1738                  * this flag and call l_blocking_ast */
1739                 lock->l_flags |= LDLM_FL_CBPENDING | LDLM_FL_CANCELING;
1740
1741                 /* We can't re-add to l_lru as it confuses the
1742                  * refcounting in ldlm_lock_remove_from_lru() if an AST
1743                  * arrives after we drop lr_lock below. We use l_bl_ast
1744                  * and can't use l_pending_chain as it is used both on
1745                  * server and client nevertheless bug 5666 says it is
1746                  * used only on server */
1747                 LASSERT(list_empty(&lock->l_bl_ast));
1748                 list_add(&lock->l_bl_ast, cancels);
1749                 unlock_res_and_lock(lock);
1750                 lu_ref_del(&lock->l_reference, __FUNCTION__, current);
1751                 spin_lock(&ns->ns_lock);
1752                 added++;
1753                 unused--;
1754         }
1755         spin_unlock(&ns->ns_lock);
1756         RETURN(added);
1757 }
1758
1759 int ldlm_cancel_lru_local(struct ldlm_namespace *ns, struct list_head *cancels,
1760                           int count, int max, ldlm_cancel_flags_t cancel_flags,
1761                           int flags)
1762 {
1763         int added;
1764         added = ldlm_prepare_lru_list(ns, cancels, count, max, flags);
1765         if (added <= 0)
1766                 return added;
1767         return ldlm_cli_cancel_list_local(cancels, added, cancel_flags);
1768 }
1769
1770 /**
1771  * Cancel at least \a nr locks from given namespace LRU.
1772  *
1773  * When called with LCF_ASYNC the blocking callback will be handled
1774  * in a thread and this function will return after the thread has been
1775  * asked to call the callback.  When called with LCF_ASYNC the blocking
1776  * callback will be performed in this function.
1777  */
1778 int ldlm_cancel_lru(struct ldlm_namespace *ns, int nr,
1779                     ldlm_cancel_flags_t cancel_flags,
1780                     int flags)
1781 {
1782         struct list_head cancels = LIST_HEAD_INIT(cancels);
1783         int count, rc;
1784         ENTRY;
1785
1786         /* Just prepare the list of locks, do not actually cancel them yet.
1787          * Locks are cancelled later in a separate thread. */
1788         count = ldlm_prepare_lru_list(ns, &cancels, nr, 0, flags);
1789         rc = ldlm_bl_to_thread_list(ns, NULL, &cancels, count, cancel_flags);
1790         if (rc == 0)
1791                 RETURN(count);
1792
1793         RETURN(0);
1794 }
1795
1796 /**
1797  * Find and cancel locally unused locks found on resource, matched to the
1798  * given policy, mode. GET the found locks and add them into the \a cancels
1799  * list.
1800  */
1801 int ldlm_cancel_resource_local(struct ldlm_resource *res,
1802                                struct list_head *cancels,
1803                                ldlm_policy_data_t *policy,
1804                                ldlm_mode_t mode, __u64 lock_flags,
1805                                ldlm_cancel_flags_t cancel_flags, void *opaque)
1806 {
1807         struct ldlm_lock *lock;
1808         int count = 0;
1809         ENTRY;
1810
1811         lock_res(res);
1812         list_for_each_entry(lock, &res->lr_granted, l_res_link) {
1813                 if (opaque != NULL && lock->l_ast_data != opaque) {
1814                         LDLM_ERROR(lock, "data %p doesn't match opaque %p",
1815                                    lock->l_ast_data, opaque);
1816                         //LBUG();
1817                         continue;
1818                 }
1819
1820                 if (lock->l_readers || lock->l_writers)
1821                         continue;
1822
1823                 /* If somebody is already doing CANCEL, or blocking AST came,
1824                  * skip this lock. */
1825                 if (ldlm_is_bl_ast(lock) || ldlm_is_canceling(lock))
1826                         continue;
1827
1828                 if (lockmode_compat(lock->l_granted_mode, mode))
1829                         continue;
1830
1831                 /* If policy is given and this is IBITS lock, add to list only
1832                  * those locks that match by policy. */
1833                 if (policy && (lock->l_resource->lr_type == LDLM_IBITS) &&
1834                     !(lock->l_policy_data.l_inodebits.bits &
1835                       policy->l_inodebits.bits))
1836                         continue;
1837
1838                 /* See CBPENDING comment in ldlm_cancel_lru */
1839                 lock->l_flags |= LDLM_FL_CBPENDING | LDLM_FL_CANCELING |
1840                                  lock_flags;
1841
1842                 LASSERT(list_empty(&lock->l_bl_ast));
1843                 list_add(&lock->l_bl_ast, cancels);
1844                 LDLM_LOCK_GET(lock);
1845                 count++;
1846         }
1847         unlock_res(res);
1848
1849         RETURN(ldlm_cli_cancel_list_local(cancels, count, cancel_flags));
1850 }
1851 EXPORT_SYMBOL(ldlm_cancel_resource_local);
1852
1853 /**
1854  * Cancel client-side locks from a list and send/prepare cancel RPCs to the
1855  * server.
1856  * If \a req is NULL, send CANCEL request to server with handles of locks
1857  * in the \a cancels. If EARLY_CANCEL is not supported, send CANCEL requests
1858  * separately per lock.
1859  * If \a req is not NULL, put handles of locks in \a cancels into the request
1860  * buffer at the offset \a off.
1861  * Destroy \a cancels at the end.
1862  */
1863 int ldlm_cli_cancel_list(struct list_head *cancels, int count,
1864                          struct ptlrpc_request *req, ldlm_cancel_flags_t flags)
1865 {
1866         struct ldlm_lock *lock;
1867         int res = 0;
1868         ENTRY;
1869
1870         if (list_empty(cancels) || count == 0)
1871                 RETURN(0);
1872
1873         /* XXX: requests (both batched and not) could be sent in parallel.
1874          * Usually it is enough to have just 1 RPC, but it is possible that
1875          * there are too many locks to be cancelled in LRU or on a resource.
1876          * It would also speed up the case when the server does not support
1877          * the feature. */
1878         while (count > 0) {
1879                 LASSERT(!list_empty(cancels));
1880                 lock = list_entry(cancels->next, struct ldlm_lock,
1881                                       l_bl_ast);
1882                 LASSERT(lock->l_conn_export);
1883
1884                 if (exp_connect_cancelset(lock->l_conn_export)) {
1885                         res = count;
1886                         if (req)
1887                                 ldlm_cancel_pack(req, cancels, count);
1888                         else
1889                                 res = ldlm_cli_cancel_req(lock->l_conn_export,
1890                                                           cancels, count,
1891                                                           flags);
1892                 } else {
1893                         res = ldlm_cli_cancel_req(lock->l_conn_export,
1894                                                   cancels, 1, flags);
1895                 }
1896
1897                 if (res < 0) {
1898                         CDEBUG_LIMIT(res == -ESHUTDOWN ? D_DLMTRACE : D_ERROR,
1899                                      "ldlm_cli_cancel_list: %d\n", res);
1900                         res = count;
1901                 }
1902
1903                 count -= res;
1904                 ldlm_lock_list_put(cancels, l_bl_ast, res);
1905         }
1906         LASSERT(count == 0);
1907         RETURN(0);
1908 }
1909 EXPORT_SYMBOL(ldlm_cli_cancel_list);
1910
1911 /**
1912  * Cancel all locks on a resource that have 0 readers/writers.
1913  *
1914  * If flags & LDLM_FL_LOCAL_ONLY, throw the locks away without trying
1915  * to notify the server. */
1916 int ldlm_cli_cancel_unused_resource(struct ldlm_namespace *ns,
1917                                     const struct ldlm_res_id *res_id,
1918                                     ldlm_policy_data_t *policy,
1919                                     ldlm_mode_t mode,
1920                                     ldlm_cancel_flags_t flags,
1921                                     void *opaque)
1922 {
1923         struct ldlm_resource *res;
1924         struct list_head cancels = LIST_HEAD_INIT(cancels);
1925         int count;
1926         int rc;
1927         ENTRY;
1928
1929         res = ldlm_resource_get(ns, NULL, res_id, 0, 0);
1930         if (IS_ERR(res)) {
1931                 /* This is not a problem. */
1932                 CDEBUG(D_INFO, "No resource "LPU64"\n", res_id->name[0]);
1933                 RETURN(0);
1934         }
1935
1936         LDLM_RESOURCE_ADDREF(res);
1937         count = ldlm_cancel_resource_local(res, &cancels, policy, mode,
1938                                            0, flags | LCF_BL_AST, opaque);
1939         rc = ldlm_cli_cancel_list(&cancels, count, NULL, flags);
1940         if (rc != ELDLM_OK)
1941                 CERROR("canceling unused lock "DLDLMRES": rc = %d\n",
1942                        PLDLMRES(res), rc);
1943
1944         LDLM_RESOURCE_DELREF(res);
1945         ldlm_resource_putref(res);
1946         RETURN(0);
1947 }
1948 EXPORT_SYMBOL(ldlm_cli_cancel_unused_resource);
1949
1950 struct ldlm_cli_cancel_arg {
1951         int     lc_flags;
1952         void   *lc_opaque;
1953 };
1954
1955 static int ldlm_cli_hash_cancel_unused(cfs_hash_t *hs, cfs_hash_bd_t *bd,
1956                                        struct hlist_node *hnode, void *arg)
1957 {
1958         struct ldlm_resource           *res = cfs_hash_object(hs, hnode);
1959         struct ldlm_cli_cancel_arg     *lc = arg;
1960
1961         ldlm_cli_cancel_unused_resource(ldlm_res_to_ns(res), &res->lr_name,
1962                                         NULL, LCK_MINMODE, lc->lc_flags,
1963                                         lc->lc_opaque);
1964         /* must return 0 for hash iteration */
1965         return 0;
1966 }
1967
1968 /**
1969  * Cancel all locks on a namespace (or a specific resource, if given)
1970  * that have 0 readers/writers.
1971  *
1972  * If flags & LCF_LOCAL, throw the locks away without trying
1973  * to notify the server. */
1974 int ldlm_cli_cancel_unused(struct ldlm_namespace *ns,
1975                            const struct ldlm_res_id *res_id,
1976                            ldlm_cancel_flags_t flags, void *opaque)
1977 {
1978         struct ldlm_cli_cancel_arg arg = {
1979                 .lc_flags       = flags,
1980                 .lc_opaque      = opaque,
1981         };
1982
1983         ENTRY;
1984
1985         if (ns == NULL)
1986                 RETURN(ELDLM_OK);
1987
1988         if (res_id != NULL) {
1989                 RETURN(ldlm_cli_cancel_unused_resource(ns, res_id, NULL,
1990                                                        LCK_MINMODE, flags,
1991                                                        opaque));
1992         } else {
1993                 cfs_hash_for_each_nolock(ns->ns_rs_hash,
1994                                          ldlm_cli_hash_cancel_unused, &arg);
1995                 RETURN(ELDLM_OK);
1996         }
1997 }
1998 EXPORT_SYMBOL(ldlm_cli_cancel_unused);
1999
2000 /* Lock iterators. */
2001
2002 int ldlm_resource_foreach(struct ldlm_resource *res, ldlm_iterator_t iter,
2003                           void *closure)
2004 {
2005         struct list_head *tmp, *next;
2006         struct ldlm_lock *lock;
2007         int rc = LDLM_ITER_CONTINUE;
2008
2009         ENTRY;
2010
2011         if (!res)
2012                 RETURN(LDLM_ITER_CONTINUE);
2013
2014         lock_res(res);
2015         list_for_each_safe(tmp, next, &res->lr_granted) {
2016                 lock = list_entry(tmp, struct ldlm_lock, l_res_link);
2017
2018                 if (iter(lock, closure) == LDLM_ITER_STOP)
2019                         GOTO(out, rc = LDLM_ITER_STOP);
2020         }
2021
2022         list_for_each_safe(tmp, next, &res->lr_converting) {
2023                 lock = list_entry(tmp, struct ldlm_lock, l_res_link);
2024
2025                 if (iter(lock, closure) == LDLM_ITER_STOP)
2026                         GOTO(out, rc = LDLM_ITER_STOP);
2027         }
2028
2029         list_for_each_safe(tmp, next, &res->lr_waiting) {
2030                 lock = list_entry(tmp, struct ldlm_lock, l_res_link);
2031
2032                 if (iter(lock, closure) == LDLM_ITER_STOP)
2033                         GOTO(out, rc = LDLM_ITER_STOP);
2034         }
2035  out:
2036         unlock_res(res);
2037         RETURN(rc);
2038 }
2039 EXPORT_SYMBOL(ldlm_resource_foreach);
2040
2041 struct iter_helper_data {
2042         ldlm_iterator_t iter;
2043         void *closure;
2044 };
2045
2046 static int ldlm_iter_helper(struct ldlm_lock *lock, void *closure)
2047 {
2048         struct iter_helper_data *helper = closure;
2049         return helper->iter(lock, helper->closure);
2050 }
2051
2052 static int ldlm_res_iter_helper(cfs_hash_t *hs, cfs_hash_bd_t *bd,
2053                                 struct hlist_node *hnode, void *arg)
2054
2055 {
2056         struct ldlm_resource *res = cfs_hash_object(hs, hnode);
2057
2058         return ldlm_resource_foreach(res, ldlm_iter_helper, arg) ==
2059                LDLM_ITER_STOP;
2060 }
2061
2062 void ldlm_namespace_foreach(struct ldlm_namespace *ns,
2063                             ldlm_iterator_t iter, void *closure)
2064
2065 {
2066         struct iter_helper_data helper = { .iter = iter, .closure = closure };
2067
2068         cfs_hash_for_each_nolock(ns->ns_rs_hash,
2069                                  ldlm_res_iter_helper, &helper);
2070
2071 }
2072 EXPORT_SYMBOL(ldlm_namespace_foreach);
2073
2074 /* non-blocking function to manipulate a lock whose cb_data is being put away.
2075  * return  0:  find no resource
2076  *       > 0:  must be LDLM_ITER_STOP/LDLM_ITER_CONTINUE.
2077  *       < 0:  errors
2078  */
2079 int ldlm_resource_iterate(struct ldlm_namespace *ns,
2080                           const struct ldlm_res_id *res_id,
2081                           ldlm_iterator_t iter, void *data)
2082 {
2083         struct ldlm_resource *res;
2084         int rc;
2085         ENTRY;
2086
2087         LASSERTF(ns != NULL, "must pass in namespace\n");
2088
2089         res = ldlm_resource_get(ns, NULL, res_id, 0, 0);
2090         if (IS_ERR(res))
2091                 RETURN(0);
2092
2093         LDLM_RESOURCE_ADDREF(res);
2094         rc = ldlm_resource_foreach(res, iter, data);
2095         LDLM_RESOURCE_DELREF(res);
2096         ldlm_resource_putref(res);
2097         RETURN(rc);
2098 }
2099 EXPORT_SYMBOL(ldlm_resource_iterate);
2100
2101 /* Lock replay */
2102
2103 static int ldlm_chain_lock_for_replay(struct ldlm_lock *lock, void *closure)
2104 {
2105         struct list_head *list = closure;
2106
2107         /* we use l_pending_chain here, because it's unused on clients. */
2108         LASSERTF(list_empty(&lock->l_pending_chain),
2109                  "lock %p next %p prev %p\n",
2110                  lock, &lock->l_pending_chain.next,&lock->l_pending_chain.prev);
2111         /* bug 9573: don't replay locks left after eviction, or
2112          * bug 17614: locks being actively cancelled. Get a reference
2113          * on a lock so that it does not disapear under us (e.g. due to cancel)
2114          */
2115         if (!(lock->l_flags & (LDLM_FL_FAILED|LDLM_FL_CANCELING))) {
2116                 list_add(&lock->l_pending_chain, list);
2117                 LDLM_LOCK_GET(lock);
2118         }
2119
2120         return LDLM_ITER_CONTINUE;
2121 }
2122
2123 static int replay_lock_interpret(const struct lu_env *env,
2124                                  struct ptlrpc_request *req,
2125                                  struct ldlm_async_args *aa, int rc)
2126 {
2127         struct ldlm_lock     *lock;
2128         struct ldlm_reply    *reply;
2129         struct obd_export    *exp;
2130
2131         ENTRY;
2132         atomic_dec(&req->rq_import->imp_replay_inflight);
2133         if (rc != ELDLM_OK)
2134                 GOTO(out, rc);
2135
2136         reply = req_capsule_server_get(&req->rq_pill, &RMF_DLM_REP);
2137         if (reply == NULL)
2138                 GOTO(out, rc = -EPROTO);
2139
2140         lock = ldlm_handle2lock(&aa->lock_handle);
2141         if (!lock) {
2142                 CERROR("received replay ack for unknown local cookie "LPX64
2143                        " remote cookie "LPX64 " from server %s id %s\n",
2144                        aa->lock_handle.cookie, reply->lock_handle.cookie,
2145                        req->rq_export->exp_client_uuid.uuid,
2146                        libcfs_id2str(req->rq_peer));
2147                 GOTO(out, rc = -ESTALE);
2148         }
2149
2150         /* Key change rehash lock in per-export hash with new key */
2151         exp = req->rq_export;
2152         if (exp && exp->exp_lock_hash) {
2153                 /* In the function below, .hs_keycmp resolves to
2154                  * ldlm_export_lock_keycmp() */
2155                 /* coverity[overrun-buffer-val] */
2156                 cfs_hash_rehash_key(exp->exp_lock_hash,
2157                                     &lock->l_remote_handle,
2158                                     &reply->lock_handle,
2159                                     &lock->l_exp_hash);
2160         } else {
2161                 lock->l_remote_handle = reply->lock_handle;
2162         }
2163
2164         LDLM_DEBUG(lock, "replayed lock:");
2165         ptlrpc_import_recovery_state_machine(req->rq_import);
2166         LDLM_LOCK_PUT(lock);
2167 out:
2168         if (rc != ELDLM_OK)
2169                 ptlrpc_connect_import(req->rq_import);
2170
2171         RETURN(rc);
2172 }
2173
2174 static int replay_one_lock(struct obd_import *imp, struct ldlm_lock *lock)
2175 {
2176         struct ptlrpc_request *req;
2177         struct ldlm_async_args *aa;
2178         struct ldlm_request   *body;
2179         int flags;
2180         ENTRY;
2181
2182
2183         /* Bug 11974: Do not replay a lock which is actively being canceled */
2184         if (ldlm_is_canceling(lock)) {
2185                 LDLM_DEBUG(lock, "Not replaying canceled lock:");
2186                 RETURN(0);
2187         }
2188
2189         /* If this is reply-less callback lock, we cannot replay it, since
2190          * server might have long dropped it, but notification of that event was
2191          * lost by network. (and server granted conflicting lock already) */
2192         if (ldlm_is_cancel_on_block(lock)) {
2193                 LDLM_DEBUG(lock, "Not replaying reply-less lock:");
2194                 ldlm_lock_cancel(lock);
2195                 RETURN(0);
2196         }
2197
2198         /*
2199          * If granted mode matches the requested mode, this lock is granted.
2200          *
2201          * If they differ, but we have a granted mode, then we were granted
2202          * one mode and now want another: ergo, converting.
2203          *
2204          * If we haven't been granted anything and are on a resource list,
2205          * then we're blocked/waiting.
2206          *
2207          * If we haven't been granted anything and we're NOT on a resource list,
2208          * then we haven't got a reply yet and don't have a known disposition.
2209          * This happens whenever a lock enqueue is the request that triggers
2210          * recovery.
2211          */
2212         if (lock->l_granted_mode == lock->l_req_mode)
2213                 flags = LDLM_FL_REPLAY | LDLM_FL_BLOCK_GRANTED;
2214         else if (lock->l_granted_mode)
2215                 flags = LDLM_FL_REPLAY | LDLM_FL_BLOCK_CONV;
2216         else if (!list_empty(&lock->l_res_link))
2217                 flags = LDLM_FL_REPLAY | LDLM_FL_BLOCK_WAIT;
2218         else
2219                 flags = LDLM_FL_REPLAY;
2220
2221         req = ptlrpc_request_alloc_pack(imp, &RQF_LDLM_ENQUEUE,
2222                                         LUSTRE_DLM_VERSION, LDLM_ENQUEUE);
2223         if (req == NULL)
2224                 RETURN(-ENOMEM);
2225
2226         /* We're part of recovery, so don't wait for it. */
2227         req->rq_send_state = LUSTRE_IMP_REPLAY_LOCKS;
2228
2229         body = req_capsule_client_get(&req->rq_pill, &RMF_DLM_REQ);
2230         ldlm_lock2desc(lock, &body->lock_desc);
2231         body->lock_flags = ldlm_flags_to_wire(flags);
2232
2233         ldlm_lock2handle(lock, &body->lock_handle[0]);
2234         if (lock->l_lvb_len > 0)
2235                 req_capsule_extend(&req->rq_pill, &RQF_LDLM_ENQUEUE_LVB);
2236         req_capsule_set_size(&req->rq_pill, &RMF_DLM_LVB, RCL_SERVER,
2237                              lock->l_lvb_len);
2238         ptlrpc_request_set_replen(req);
2239         /* notify the server we've replayed all requests.
2240          * also, we mark the request to be put on a dedicated
2241          * queue to be processed after all request replayes.
2242          * bug 6063 */
2243         lustre_msg_set_flags(req->rq_reqmsg, MSG_REQ_REPLAY_DONE);
2244
2245         LDLM_DEBUG(lock, "replaying lock:");
2246
2247         atomic_inc(&req->rq_import->imp_replay_inflight);
2248         CLASSERT(sizeof(*aa) <= sizeof(req->rq_async_args));
2249         aa = ptlrpc_req_async_args(req);
2250         aa->lock_handle = body->lock_handle[0];
2251         req->rq_interpret_reply = (ptlrpc_interpterer_t)replay_lock_interpret;
2252         ptlrpcd_add_req(req, PDL_POLICY_LOCAL, -1);
2253
2254         RETURN(0);
2255 }
2256
2257 /**
2258  * Cancel as many unused locks as possible before replay. since we are
2259  * in recovery, we can't wait for any outstanding RPCs to send any RPC
2260  * to the server.
2261  *
2262  * Called only in recovery before replaying locks. there is no need to
2263  * replay locks that are unused. since the clients may hold thousands of
2264  * cached unused locks, dropping the unused locks can greatly reduce the
2265  * load on the servers at recovery time.
2266  */
2267 static void ldlm_cancel_unused_locks_for_replay(struct ldlm_namespace *ns)
2268 {
2269         int canceled;
2270         struct list_head cancels = LIST_HEAD_INIT(cancels);
2271
2272         CDEBUG(D_DLMTRACE, "Dropping as many unused locks as possible before"
2273                            "replay for namespace %s (%d)\n",
2274                            ldlm_ns_name(ns), ns->ns_nr_unused);
2275
2276         /* We don't need to care whether or not LRU resize is enabled
2277          * because the LDLM_CANCEL_NO_WAIT policy doesn't use the
2278          * count parameter */
2279         canceled = ldlm_cancel_lru_local(ns, &cancels, ns->ns_nr_unused, 0,
2280                                          LCF_LOCAL, LDLM_CANCEL_NO_WAIT);
2281
2282         CDEBUG(D_DLMTRACE, "Canceled %d unused locks from namespace %s\n",
2283                            canceled, ldlm_ns_name(ns));
2284 }
2285
2286 int ldlm_replay_locks(struct obd_import *imp)
2287 {
2288         struct ldlm_namespace *ns = imp->imp_obd->obd_namespace;
2289         struct list_head list = LIST_HEAD_INIT(list);
2290         struct ldlm_lock *lock, *next;
2291         int rc = 0;
2292
2293         ENTRY;
2294
2295         LASSERT(atomic_read(&imp->imp_replay_inflight) == 0);
2296
2297         /* don't replay locks if import failed recovery */
2298         if (imp->imp_vbr_failed)
2299                 RETURN(0);
2300
2301         /* ensure this doesn't fall to 0 before all have been queued */
2302         atomic_inc(&imp->imp_replay_inflight);
2303
2304         if (ldlm_cancel_unused_locks_before_replay)
2305                 ldlm_cancel_unused_locks_for_replay(ns);
2306
2307         ldlm_namespace_foreach(ns, ldlm_chain_lock_for_replay, &list);
2308
2309         list_for_each_entry_safe(lock, next, &list, l_pending_chain) {
2310                 list_del_init(&lock->l_pending_chain);
2311                 if (rc) {
2312                         LDLM_LOCK_RELEASE(lock);
2313                         continue; /* or try to do the rest? */
2314                 }
2315                 rc = replay_one_lock(imp, lock);
2316                 LDLM_LOCK_RELEASE(lock);
2317         }
2318
2319         atomic_dec(&imp->imp_replay_inflight);
2320
2321         RETURN(rc);
2322 }
2323 EXPORT_SYMBOL(ldlm_replay_locks);