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