Whamcloud - gitweb
fc8f9c99a69a92f5bc0473081b04bbdda8ed2d8c
[fs/lustre-release.git] / lustre / obdclass / cl_lock.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) 2008, 2010, Oracle and/or its affiliates. All rights reserved.
28  * Use is subject to license terms.
29  *
30  * Copyright (c) 2011, 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  * Client Extent Lock.
37  *
38  *   Author: Nikita Danilov <nikita.danilov@sun.com>
39  */
40
41 #define DEBUG_SUBSYSTEM S_CLASS
42
43 #include <obd_class.h>
44 #include <obd_support.h>
45 #include <lustre_fid.h>
46 #include <libcfs/list.h>
47 #include <cl_object.h>
48 #include "cl_internal.h"
49
50 /** Lock class of cl_lock::cll_guard */
51 static struct lock_class_key cl_lock_guard_class;
52 static cfs_mem_cache_t *cl_lock_kmem;
53
54 static struct lu_kmem_descr cl_lock_caches[] = {
55         {
56                 .ckd_cache = &cl_lock_kmem,
57                 .ckd_name  = "cl_lock_kmem",
58                 .ckd_size  = sizeof (struct cl_lock)
59         },
60         {
61                 .ckd_cache = NULL
62         }
63 };
64
65 #ifdef CONFIG_DEBUG_PAGESTATE_TRACKING
66 #define CS_LOCK_INC(o, item) \
67         cfs_atomic_inc(&cl_object_site(o)->cs_locks.cs_stats[CS_##item])
68 #define CS_LOCK_DEC(o, item) \
69         cfs_atomic_dec(&cl_object_site(o)->cs_locks.cs_stats[CS_##item])
70 #define CS_LOCKSTATE_INC(o, state) \
71         cfs_atomic_inc(&cl_object_site(o)->cs_locks_state[state])
72 #define CS_LOCKSTATE_DEC(o, state) \
73         cfs_atomic_dec(&cl_object_site(o)->cs_locks_state[state])
74 #else
75 #define CS_LOCK_INC(o, item)
76 #define CS_LOCK_DEC(o, item)
77 #define CS_LOCKSTATE_INC(o, state)
78 #define CS_LOCKSTATE_DEC(o, state)
79 #endif
80
81 /**
82  * Basic lock invariant that is maintained at all times. Caller either has a
83  * reference to \a lock, or somehow assures that \a lock cannot be freed.
84  *
85  * \see cl_lock_invariant()
86  */
87 static int cl_lock_invariant_trusted(const struct lu_env *env,
88                                      const struct cl_lock *lock)
89 {
90         return  ergo(lock->cll_state == CLS_FREEING, lock->cll_holds == 0) &&
91                 cfs_atomic_read(&lock->cll_ref) >= lock->cll_holds &&
92                 lock->cll_holds >= lock->cll_users &&
93                 lock->cll_holds >= 0 &&
94                 lock->cll_users >= 0 &&
95                 lock->cll_depth >= 0;
96 }
97
98 /**
99  * Stronger lock invariant, checking that caller has a reference on a lock.
100  *
101  * \see cl_lock_invariant_trusted()
102  */
103 static int cl_lock_invariant(const struct lu_env *env,
104                              const struct cl_lock *lock)
105 {
106         int result;
107
108         result = cfs_atomic_read(&lock->cll_ref) > 0 &&
109                 cl_lock_invariant_trusted(env, lock);
110         if (!result && env != NULL)
111                 CL_LOCK_DEBUG(D_ERROR, env, lock, "invariant broken");
112         return result;
113 }
114
115 /**
116  * Returns lock "nesting": 0 for a top-lock and 1 for a sub-lock.
117  */
118 static enum clt_nesting_level cl_lock_nesting(const struct cl_lock *lock)
119 {
120         return cl_object_header(lock->cll_descr.cld_obj)->coh_nesting;
121 }
122
123 /**
124  * Returns a set of counters for this lock, depending on a lock nesting.
125  */
126 static struct cl_thread_counters *cl_lock_counters(const struct lu_env *env,
127                                                    const struct cl_lock *lock)
128 {
129         struct cl_thread_info *info;
130         enum clt_nesting_level nesting;
131
132         info = cl_env_info(env);
133         nesting = cl_lock_nesting(lock);
134         LASSERT(nesting < ARRAY_SIZE(info->clt_counters));
135         return &info->clt_counters[nesting];
136 }
137
138 static void cl_lock_trace0(int level, const struct lu_env *env,
139                            const char *prefix, const struct cl_lock *lock,
140                            const char *func, const int line)
141 {
142         struct cl_object_header *h = cl_object_header(lock->cll_descr.cld_obj);
143         CDEBUG(level, "%s: %p@(%d %p %d %d %d %d %d %lx)"
144                       "(%p/%d/%d) at %s():%d\n",
145                prefix, lock, cfs_atomic_read(&lock->cll_ref),
146                lock->cll_guarder, lock->cll_depth,
147                lock->cll_state, lock->cll_error, lock->cll_holds,
148                lock->cll_users, lock->cll_flags,
149                env, h->coh_nesting, cl_lock_nr_mutexed(env),
150                func, line);
151 }
152 #define cl_lock_trace(level, env, prefix, lock)                         \
153         cl_lock_trace0(level, env, prefix, lock, __FUNCTION__, __LINE__)
154
155 #define RETIP ((unsigned long)__builtin_return_address(0))
156
157 #ifdef CONFIG_LOCKDEP
158 static struct lock_class_key cl_lock_key;
159
160 static void cl_lock_lockdep_init(struct cl_lock *lock)
161 {
162         lockdep_set_class_and_name(lock, &cl_lock_key, "EXT");
163 }
164
165 static void cl_lock_lockdep_acquire(const struct lu_env *env,
166                                     struct cl_lock *lock, __u32 enqflags)
167 {
168         cl_lock_counters(env, lock)->ctc_nr_locks_acquired++;
169 #ifdef HAVE_LOCK_MAP_ACQUIRE
170         lock_map_acquire(&lock->dep_map);
171 #else  /* HAVE_LOCK_MAP_ACQUIRE */
172         lock_acquire(&lock->dep_map, !!(enqflags & CEF_ASYNC),
173                      /* try: */ 0, lock->cll_descr.cld_mode <= CLM_READ,
174                      /* check: */ 2, RETIP);
175 #endif /* HAVE_LOCK_MAP_ACQUIRE */
176 }
177
178 static void cl_lock_lockdep_release(const struct lu_env *env,
179                                     struct cl_lock *lock)
180 {
181         cl_lock_counters(env, lock)->ctc_nr_locks_acquired--;
182         lock_release(&lock->dep_map, 0, RETIP);
183 }
184
185 #else /* !CONFIG_LOCKDEP */
186
187 static void cl_lock_lockdep_init(struct cl_lock *lock)
188 {}
189 static void cl_lock_lockdep_acquire(const struct lu_env *env,
190                                     struct cl_lock *lock, __u32 enqflags)
191 {}
192 static void cl_lock_lockdep_release(const struct lu_env *env,
193                                     struct cl_lock *lock)
194 {}
195
196 #endif /* !CONFIG_LOCKDEP */
197
198 /**
199  * Adds lock slice to the compound lock.
200  *
201  * This is called by cl_object_operations::coo_lock_init() methods to add a
202  * per-layer state to the lock. New state is added at the end of
203  * cl_lock::cll_layers list, that is, it is at the bottom of the stack.
204  *
205  * \see cl_req_slice_add(), cl_page_slice_add(), cl_io_slice_add()
206  */
207 void cl_lock_slice_add(struct cl_lock *lock, struct cl_lock_slice *slice,
208                        struct cl_object *obj,
209                        const struct cl_lock_operations *ops)
210 {
211         ENTRY;
212         slice->cls_lock = lock;
213         cfs_list_add_tail(&slice->cls_linkage, &lock->cll_layers);
214         slice->cls_obj = obj;
215         slice->cls_ops = ops;
216         EXIT;
217 }
218 EXPORT_SYMBOL(cl_lock_slice_add);
219
220 /**
221  * Returns true iff a lock with the mode \a has provides at least the same
222  * guarantees as a lock with the mode \a need.
223  */
224 int cl_lock_mode_match(enum cl_lock_mode has, enum cl_lock_mode need)
225 {
226         LINVRNT(need == CLM_READ || need == CLM_WRITE ||
227                 need == CLM_PHANTOM || need == CLM_GROUP);
228         LINVRNT(has == CLM_READ || has == CLM_WRITE ||
229                 has == CLM_PHANTOM || has == CLM_GROUP);
230         CLASSERT(CLM_PHANTOM < CLM_READ);
231         CLASSERT(CLM_READ < CLM_WRITE);
232         CLASSERT(CLM_WRITE < CLM_GROUP);
233
234         if (has != CLM_GROUP)
235                 return need <= has;
236         else
237                 return need == has;
238 }
239 EXPORT_SYMBOL(cl_lock_mode_match);
240
241 /**
242  * Returns true iff extent portions of lock descriptions match.
243  */
244 int cl_lock_ext_match(const struct cl_lock_descr *has,
245                       const struct cl_lock_descr *need)
246 {
247         return
248                 has->cld_start <= need->cld_start &&
249                 has->cld_end >= need->cld_end &&
250                 cl_lock_mode_match(has->cld_mode, need->cld_mode) &&
251                 (has->cld_mode != CLM_GROUP || has->cld_gid == need->cld_gid);
252 }
253 EXPORT_SYMBOL(cl_lock_ext_match);
254
255 /**
256  * Returns true iff a lock with the description \a has provides at least the
257  * same guarantees as a lock with the description \a need.
258  */
259 int cl_lock_descr_match(const struct cl_lock_descr *has,
260                         const struct cl_lock_descr *need)
261 {
262         return
263                 cl_object_same(has->cld_obj, need->cld_obj) &&
264                 cl_lock_ext_match(has, need);
265 }
266 EXPORT_SYMBOL(cl_lock_descr_match);
267
268 static void cl_lock_free(const struct lu_env *env, struct cl_lock *lock)
269 {
270         struct cl_object *obj = lock->cll_descr.cld_obj;
271
272         LINVRNT(!cl_lock_is_mutexed(lock));
273
274         ENTRY;
275         cl_lock_trace(D_DLMTRACE, env, "free lock", lock);
276         cfs_might_sleep();
277         while (!cfs_list_empty(&lock->cll_layers)) {
278                 struct cl_lock_slice *slice;
279
280                 slice = cfs_list_entry(lock->cll_layers.next,
281                                        struct cl_lock_slice, cls_linkage);
282                 cfs_list_del_init(lock->cll_layers.next);
283                 slice->cls_ops->clo_fini(env, slice);
284         }
285         CS_LOCK_DEC(obj, total);
286         CS_LOCKSTATE_DEC(obj, lock->cll_state);
287         lu_object_ref_del_at(&obj->co_lu, lock->cll_obj_ref, "cl_lock", lock);
288         cl_object_put(env, obj);
289         lu_ref_fini(&lock->cll_reference);
290         lu_ref_fini(&lock->cll_holders);
291         mutex_destroy(&lock->cll_guard);
292         OBD_SLAB_FREE_PTR(lock, cl_lock_kmem);
293         EXIT;
294 }
295
296 /**
297  * Releases a reference on a lock.
298  *
299  * When last reference is released, lock is returned to the cache, unless it
300  * is in cl_lock_state::CLS_FREEING state, in which case it is destroyed
301  * immediately.
302  *
303  * \see cl_object_put(), cl_page_put()
304  */
305 void cl_lock_put(const struct lu_env *env, struct cl_lock *lock)
306 {
307         struct cl_object        *obj;
308
309         LINVRNT(cl_lock_invariant(env, lock));
310         ENTRY;
311         obj = lock->cll_descr.cld_obj;
312         LINVRNT(obj != NULL);
313
314         CDEBUG(D_TRACE, "releasing reference: %d %p %lu\n",
315                cfs_atomic_read(&lock->cll_ref), lock, RETIP);
316
317         if (cfs_atomic_dec_and_test(&lock->cll_ref)) {
318                 if (lock->cll_state == CLS_FREEING) {
319                         LASSERT(cfs_list_empty(&lock->cll_linkage));
320                         cl_lock_free(env, lock);
321                 }
322                 CS_LOCK_DEC(obj, busy);
323         }
324         EXIT;
325 }
326 EXPORT_SYMBOL(cl_lock_put);
327
328 /**
329  * Acquires an additional reference to a lock.
330  *
331  * This can be called only by caller already possessing a reference to \a
332  * lock.
333  *
334  * \see cl_object_get(), cl_page_get()
335  */
336 void cl_lock_get(struct cl_lock *lock)
337 {
338         LINVRNT(cl_lock_invariant(NULL, lock));
339         CDEBUG(D_TRACE, "acquiring reference: %d %p %lu\n",
340                cfs_atomic_read(&lock->cll_ref), lock, RETIP);
341         cfs_atomic_inc(&lock->cll_ref);
342 }
343 EXPORT_SYMBOL(cl_lock_get);
344
345 /**
346  * Acquires a reference to a lock.
347  *
348  * This is much like cl_lock_get(), except that this function can be used to
349  * acquire initial reference to the cached lock. Caller has to deal with all
350  * possible races. Use with care!
351  *
352  * \see cl_page_get_trust()
353  */
354 void cl_lock_get_trust(struct cl_lock *lock)
355 {
356         CDEBUG(D_TRACE, "acquiring trusted reference: %d %p %lu\n",
357                cfs_atomic_read(&lock->cll_ref), lock, RETIP);
358         if (cfs_atomic_inc_return(&lock->cll_ref) == 1)
359                 CS_LOCK_INC(lock->cll_descr.cld_obj, busy);
360 }
361 EXPORT_SYMBOL(cl_lock_get_trust);
362
363 /**
364  * Helper function destroying the lock that wasn't completely initialized.
365  *
366  * Other threads can acquire references to the top-lock through its
367  * sub-locks. Hence, it cannot be cl_lock_free()-ed immediately.
368  */
369 static void cl_lock_finish(const struct lu_env *env, struct cl_lock *lock)
370 {
371         cl_lock_mutex_get(env, lock);
372         cl_lock_cancel(env, lock);
373         cl_lock_delete(env, lock);
374         cl_lock_mutex_put(env, lock);
375         cl_lock_put(env, lock);
376 }
377
378 static struct cl_lock *cl_lock_alloc(const struct lu_env *env,
379                                      struct cl_object *obj,
380                                      const struct cl_io *io,
381                                      const struct cl_lock_descr *descr)
382 {
383         struct cl_lock          *lock;
384         struct lu_object_header *head;
385
386         ENTRY;
387         OBD_SLAB_ALLOC_PTR_GFP(lock, cl_lock_kmem, CFS_ALLOC_IO);
388         if (lock != NULL) {
389                 cfs_atomic_set(&lock->cll_ref, 1);
390                 lock->cll_descr = *descr;
391                 lock->cll_state = CLS_NEW;
392                 cl_object_get(obj);
393                 lock->cll_obj_ref = lu_object_ref_add(&obj->co_lu,
394                                                       "cl_lock", lock);
395                 CFS_INIT_LIST_HEAD(&lock->cll_layers);
396                 CFS_INIT_LIST_HEAD(&lock->cll_linkage);
397                 CFS_INIT_LIST_HEAD(&lock->cll_inclosure);
398                 lu_ref_init(&lock->cll_reference);
399                 lu_ref_init(&lock->cll_holders);
400                 mutex_init(&lock->cll_guard);
401                 lockdep_set_class(&lock->cll_guard, &cl_lock_guard_class);
402                 cfs_waitq_init(&lock->cll_wq);
403                 head = obj->co_lu.lo_header;
404                 CS_LOCKSTATE_INC(obj, CLS_NEW);
405                 CS_LOCK_INC(obj, total);
406                 CS_LOCK_INC(obj, create);
407                 cl_lock_lockdep_init(lock);
408                 cfs_list_for_each_entry(obj, &head->loh_layers,
409                                         co_lu.lo_linkage) {
410                         int err;
411
412                         err = obj->co_ops->coo_lock_init(env, obj, lock, io);
413                         if (err != 0) {
414                                 cl_lock_finish(env, lock);
415                                 lock = ERR_PTR(err);
416                                 break;
417                         }
418                 }
419         } else
420                 lock = ERR_PTR(-ENOMEM);
421         RETURN(lock);
422 }
423
424 /**
425  * Transfer the lock into INTRANSIT state and return the original state.
426  *
427  * \pre  state: CLS_CACHED, CLS_HELD or CLS_ENQUEUED
428  * \post state: CLS_INTRANSIT
429  * \see CLS_INTRANSIT
430  */
431 enum cl_lock_state cl_lock_intransit(const struct lu_env *env,
432                                      struct cl_lock *lock)
433 {
434         enum cl_lock_state state = lock->cll_state;
435
436         LASSERT(cl_lock_is_mutexed(lock));
437         LASSERT(state != CLS_INTRANSIT);
438         LASSERTF(state >= CLS_ENQUEUED && state <= CLS_CACHED,
439                  "Malformed lock state %d.\n", state);
440
441         cl_lock_state_set(env, lock, CLS_INTRANSIT);
442         lock->cll_intransit_owner = cfs_current();
443         cl_lock_hold_add(env, lock, "intransit", cfs_current());
444         return state;
445 }
446 EXPORT_SYMBOL(cl_lock_intransit);
447
448 /**
449  *  Exit the intransit state and restore the lock state to the original state
450  */
451 void cl_lock_extransit(const struct lu_env *env, struct cl_lock *lock,
452                        enum cl_lock_state state)
453 {
454         LASSERT(cl_lock_is_mutexed(lock));
455         LASSERT(lock->cll_state == CLS_INTRANSIT);
456         LASSERT(state != CLS_INTRANSIT);
457         LASSERT(lock->cll_intransit_owner == cfs_current());
458
459         lock->cll_intransit_owner = NULL;
460         cl_lock_state_set(env, lock, state);
461         cl_lock_unhold(env, lock, "intransit", cfs_current());
462 }
463 EXPORT_SYMBOL(cl_lock_extransit);
464
465 /**
466  * Checking whether the lock is intransit state
467  */
468 int cl_lock_is_intransit(struct cl_lock *lock)
469 {
470         LASSERT(cl_lock_is_mutexed(lock));
471         return lock->cll_state == CLS_INTRANSIT &&
472                lock->cll_intransit_owner != cfs_current();
473 }
474 EXPORT_SYMBOL(cl_lock_is_intransit);
475 /**
476  * Returns true iff lock is "suitable" for given io. E.g., locks acquired by
477  * truncate and O_APPEND cannot be reused for read/non-append-write, as they
478  * cover multiple stripes and can trigger cascading timeouts.
479  */
480 static int cl_lock_fits_into(const struct lu_env *env,
481                              const struct cl_lock *lock,
482                              const struct cl_lock_descr *need,
483                              const struct cl_io *io)
484 {
485         const struct cl_lock_slice *slice;
486
487         LINVRNT(cl_lock_invariant_trusted(env, lock));
488         ENTRY;
489         cfs_list_for_each_entry(slice, &lock->cll_layers, cls_linkage) {
490                 if (slice->cls_ops->clo_fits_into != NULL &&
491                     !slice->cls_ops->clo_fits_into(env, slice, need, io))
492                         RETURN(0);
493         }
494         RETURN(1);
495 }
496
497 static struct cl_lock *cl_lock_lookup(const struct lu_env *env,
498                                       struct cl_object *obj,
499                                       const struct cl_io *io,
500                                       const struct cl_lock_descr *need)
501 {
502         struct cl_lock          *lock;
503         struct cl_object_header *head;
504
505         ENTRY;
506
507         head = cl_object_header(obj);
508         LINVRNT_SPIN_LOCKED(&head->coh_lock_guard);
509         CS_LOCK_INC(obj, lookup);
510         cfs_list_for_each_entry(lock, &head->coh_locks, cll_linkage) {
511                 int matched;
512
513                 matched = cl_lock_ext_match(&lock->cll_descr, need) &&
514                           lock->cll_state < CLS_FREEING &&
515                           lock->cll_error == 0 &&
516                           !(lock->cll_flags & CLF_CANCELLED) &&
517                           cl_lock_fits_into(env, lock, need, io);
518                 CDEBUG(D_DLMTRACE, "has: "DDESCR"(%d) need: "DDESCR": %d\n",
519                        PDESCR(&lock->cll_descr), lock->cll_state, PDESCR(need),
520                        matched);
521                 if (matched) {
522                         cl_lock_get_trust(lock);
523                         CS_LOCK_INC(obj, hit);
524                         RETURN(lock);
525                 }
526         }
527         RETURN(NULL);
528 }
529
530 /**
531  * Returns a lock matching description \a need.
532  *
533  * This is the main entry point into the cl_lock caching interface. First, a
534  * cache (implemented as a per-object linked list) is consulted. If lock is
535  * found there, it is returned immediately. Otherwise new lock is allocated
536  * and returned. In any case, additional reference to lock is acquired.
537  *
538  * \see cl_object_find(), cl_page_find()
539  */
540 static struct cl_lock *cl_lock_find(const struct lu_env *env,
541                                     const struct cl_io *io,
542                                     const struct cl_lock_descr *need)
543 {
544         struct cl_object_header *head;
545         struct cl_object        *obj;
546         struct cl_lock          *lock;
547
548         ENTRY;
549
550         obj  = need->cld_obj;
551         head = cl_object_header(obj);
552
553         spin_lock(&head->coh_lock_guard);
554         lock = cl_lock_lookup(env, obj, io, need);
555         spin_unlock(&head->coh_lock_guard);
556
557         if (lock == NULL) {
558                 lock = cl_lock_alloc(env, obj, io, need);
559                 if (!IS_ERR(lock)) {
560                         struct cl_lock *ghost;
561
562                         spin_lock(&head->coh_lock_guard);
563                         ghost = cl_lock_lookup(env, obj, io, need);
564                         if (ghost == NULL) {
565                                 cfs_list_add_tail(&lock->cll_linkage,
566                                                   &head->coh_locks);
567                                 spin_unlock(&head->coh_lock_guard);
568                                 CS_LOCK_INC(obj, busy);
569                         } else {
570                                 spin_unlock(&head->coh_lock_guard);
571                                 /*
572                                  * Other threads can acquire references to the
573                                  * top-lock through its sub-locks. Hence, it
574                                  * cannot be cl_lock_free()-ed immediately.
575                                  */
576                                 cl_lock_finish(env, lock);
577                                 lock = ghost;
578                         }
579                 }
580         }
581         RETURN(lock);
582 }
583
584 /**
585  * Returns existing lock matching given description. This is similar to
586  * cl_lock_find() except that no new lock is created, and returned lock is
587  * guaranteed to be in enum cl_lock_state::CLS_HELD state.
588  */
589 struct cl_lock *cl_lock_peek(const struct lu_env *env, const struct cl_io *io,
590                              const struct cl_lock_descr *need,
591                              const char *scope, const void *source)
592 {
593         struct cl_object_header *head;
594         struct cl_object        *obj;
595         struct cl_lock          *lock;
596
597         obj  = need->cld_obj;
598         head = cl_object_header(obj);
599
600         do {
601                 spin_lock(&head->coh_lock_guard);
602                 lock = cl_lock_lookup(env, obj, io, need);
603                 spin_unlock(&head->coh_lock_guard);
604                 if (lock == NULL)
605                         return NULL;
606
607                 cl_lock_mutex_get(env, lock);
608                 if (lock->cll_state == CLS_INTRANSIT)
609                         /* Don't care return value. */
610                         cl_lock_state_wait(env, lock);
611                 if (lock->cll_state == CLS_FREEING) {
612                         cl_lock_mutex_put(env, lock);
613                         cl_lock_put(env, lock);
614                         lock = NULL;
615                 }
616         } while (lock == NULL);
617
618         cl_lock_hold_add(env, lock, scope, source);
619         cl_lock_user_add(env, lock);
620         if (lock->cll_state == CLS_CACHED)
621                 cl_use_try(env, lock, 1);
622         if (lock->cll_state == CLS_HELD) {
623                 cl_lock_mutex_put(env, lock);
624                 cl_lock_lockdep_acquire(env, lock, 0);
625                 cl_lock_put(env, lock);
626         } else {
627                 cl_unuse_try(env, lock);
628                 cl_lock_unhold(env, lock, scope, source);
629                 cl_lock_mutex_put(env, lock);
630                 cl_lock_put(env, lock);
631                 lock = NULL;
632         }
633
634         return lock;
635 }
636 EXPORT_SYMBOL(cl_lock_peek);
637
638 /**
639  * Returns a slice within a lock, corresponding to the given layer in the
640  * device stack.
641  *
642  * \see cl_page_at()
643  */
644 const struct cl_lock_slice *cl_lock_at(const struct cl_lock *lock,
645                                        const struct lu_device_type *dtype)
646 {
647         const struct cl_lock_slice *slice;
648
649         LINVRNT(cl_lock_invariant_trusted(NULL, lock));
650         ENTRY;
651
652         cfs_list_for_each_entry(slice, &lock->cll_layers, cls_linkage) {
653                 if (slice->cls_obj->co_lu.lo_dev->ld_type == dtype)
654                         RETURN(slice);
655         }
656         RETURN(NULL);
657 }
658 EXPORT_SYMBOL(cl_lock_at);
659
660 static void cl_lock_mutex_tail(const struct lu_env *env, struct cl_lock *lock)
661 {
662         struct cl_thread_counters *counters;
663
664         counters = cl_lock_counters(env, lock);
665         lock->cll_depth++;
666         counters->ctc_nr_locks_locked++;
667         lu_ref_add(&counters->ctc_locks_locked, "cll_guard", lock);
668         cl_lock_trace(D_TRACE, env, "got mutex", lock);
669 }
670
671 /**
672  * Locks cl_lock object.
673  *
674  * This is used to manipulate cl_lock fields, and to serialize state
675  * transitions in the lock state machine.
676  *
677  * \post cl_lock_is_mutexed(lock)
678  *
679  * \see cl_lock_mutex_put()
680  */
681 void cl_lock_mutex_get(const struct lu_env *env, struct cl_lock *lock)
682 {
683         LINVRNT(cl_lock_invariant(env, lock));
684
685         if (lock->cll_guarder == cfs_current()) {
686                 LINVRNT(cl_lock_is_mutexed(lock));
687                 LINVRNT(lock->cll_depth > 0);
688         } else {
689                 struct cl_object_header *hdr;
690                 struct cl_thread_info   *info;
691                 int i;
692
693                 LINVRNT(lock->cll_guarder != cfs_current());
694                 hdr = cl_object_header(lock->cll_descr.cld_obj);
695                 /*
696                  * Check that mutices are taken in the bottom-to-top order.
697                  */
698                 info = cl_env_info(env);
699                 for (i = 0; i < hdr->coh_nesting; ++i)
700                         LASSERT(info->clt_counters[i].ctc_nr_locks_locked == 0);
701                 mutex_lock_nested(&lock->cll_guard, hdr->coh_nesting);
702                 lock->cll_guarder = cfs_current();
703                 LINVRNT(lock->cll_depth == 0);
704         }
705         cl_lock_mutex_tail(env, lock);
706 }
707 EXPORT_SYMBOL(cl_lock_mutex_get);
708
709 /**
710  * Try-locks cl_lock object.
711  *
712  * \retval 0 \a lock was successfully locked
713  *
714  * \retval -EBUSY \a lock cannot be locked right now
715  *
716  * \post ergo(result == 0, cl_lock_is_mutexed(lock))
717  *
718  * \see cl_lock_mutex_get()
719  */
720 int cl_lock_mutex_try(const struct lu_env *env, struct cl_lock *lock)
721 {
722         int result;
723
724         LINVRNT(cl_lock_invariant_trusted(env, lock));
725         ENTRY;
726
727         result = 0;
728         if (lock->cll_guarder == cfs_current()) {
729                 LINVRNT(lock->cll_depth > 0);
730                 cl_lock_mutex_tail(env, lock);
731         } else if (mutex_trylock(&lock->cll_guard)) {
732                 LINVRNT(lock->cll_depth == 0);
733                 lock->cll_guarder = cfs_current();
734                 cl_lock_mutex_tail(env, lock);
735         } else
736                 result = -EBUSY;
737         RETURN(result);
738 }
739 EXPORT_SYMBOL(cl_lock_mutex_try);
740
741 /**
742  {* Unlocks cl_lock object.
743  *
744  * \pre cl_lock_is_mutexed(lock)
745  *
746  * \see cl_lock_mutex_get()
747  */
748 void cl_lock_mutex_put(const struct lu_env *env, struct cl_lock *lock)
749 {
750         struct cl_thread_counters *counters;
751
752         LINVRNT(cl_lock_invariant(env, lock));
753         LINVRNT(cl_lock_is_mutexed(lock));
754         LINVRNT(lock->cll_guarder == cfs_current());
755         LINVRNT(lock->cll_depth > 0);
756
757         counters = cl_lock_counters(env, lock);
758         LINVRNT(counters->ctc_nr_locks_locked > 0);
759
760         cl_lock_trace(D_TRACE, env, "put mutex", lock);
761         lu_ref_del(&counters->ctc_locks_locked, "cll_guard", lock);
762         counters->ctc_nr_locks_locked--;
763         if (--lock->cll_depth == 0) {
764                 lock->cll_guarder = NULL;
765                 mutex_unlock(&lock->cll_guard);
766         }
767 }
768 EXPORT_SYMBOL(cl_lock_mutex_put);
769
770 /**
771  * Returns true iff lock's mutex is owned by the current thread.
772  */
773 int cl_lock_is_mutexed(struct cl_lock *lock)
774 {
775         return lock->cll_guarder == cfs_current();
776 }
777 EXPORT_SYMBOL(cl_lock_is_mutexed);
778
779 /**
780  * Returns number of cl_lock mutices held by the current thread (environment).
781  */
782 int cl_lock_nr_mutexed(const struct lu_env *env)
783 {
784         struct cl_thread_info *info;
785         int i;
786         int locked;
787
788         /*
789          * NOTE: if summation across all nesting levels (currently 2) proves
790          *       too expensive, a summary counter can be added to
791          *       struct cl_thread_info.
792          */
793         info = cl_env_info(env);
794         for (i = 0, locked = 0; i < ARRAY_SIZE(info->clt_counters); ++i)
795                 locked += info->clt_counters[i].ctc_nr_locks_locked;
796         return locked;
797 }
798 EXPORT_SYMBOL(cl_lock_nr_mutexed);
799
800 static void cl_lock_cancel0(const struct lu_env *env, struct cl_lock *lock)
801 {
802         LINVRNT(cl_lock_is_mutexed(lock));
803         LINVRNT(cl_lock_invariant(env, lock));
804         ENTRY;
805         if (!(lock->cll_flags & CLF_CANCELLED)) {
806                 const struct cl_lock_slice *slice;
807
808                 lock->cll_flags |= CLF_CANCELLED;
809                 cfs_list_for_each_entry_reverse(slice, &lock->cll_layers,
810                                                 cls_linkage) {
811                         if (slice->cls_ops->clo_cancel != NULL)
812                                 slice->cls_ops->clo_cancel(env, slice);
813                 }
814         }
815         EXIT;
816 }
817
818 static void cl_lock_delete0(const struct lu_env *env, struct cl_lock *lock)
819 {
820         struct cl_object_header    *head;
821         const struct cl_lock_slice *slice;
822
823         LINVRNT(cl_lock_is_mutexed(lock));
824         LINVRNT(cl_lock_invariant(env, lock));
825
826         ENTRY;
827         if (lock->cll_state < CLS_FREEING) {
828                 LASSERT(lock->cll_state != CLS_INTRANSIT);
829                 cl_lock_state_set(env, lock, CLS_FREEING);
830
831                 head = cl_object_header(lock->cll_descr.cld_obj);
832
833                 spin_lock(&head->coh_lock_guard);
834                 cfs_list_del_init(&lock->cll_linkage);
835                 spin_unlock(&head->coh_lock_guard);
836
837                 /*
838                  * From now on, no new references to this lock can be acquired
839                  * by cl_lock_lookup().
840                  */
841                 cfs_list_for_each_entry_reverse(slice, &lock->cll_layers,
842                                                 cls_linkage) {
843                         if (slice->cls_ops->clo_delete != NULL)
844                                 slice->cls_ops->clo_delete(env, slice);
845                 }
846                 /*
847                  * From now on, no new references to this lock can be acquired
848                  * by layer-specific means (like a pointer from struct
849                  * ldlm_lock in osc, or a pointer from top-lock to sub-lock in
850                  * lov).
851                  *
852                  * Lock will be finally freed in cl_lock_put() when last of
853                  * existing references goes away.
854                  */
855         }
856         EXIT;
857 }
858
859 /**
860  * Mod(ifie)s cl_lock::cll_holds counter for a given lock. Also, for a
861  * top-lock (nesting == 0) accounts for this modification in the per-thread
862  * debugging counters. Sub-lock holds can be released by a thread different
863  * from one that acquired it.
864  */
865 static void cl_lock_hold_mod(const struct lu_env *env, struct cl_lock *lock,
866                              int delta)
867 {
868         struct cl_thread_counters *counters;
869         enum clt_nesting_level     nesting;
870
871         lock->cll_holds += delta;
872         nesting = cl_lock_nesting(lock);
873         if (nesting == CNL_TOP) {
874                 counters = &cl_env_info(env)->clt_counters[CNL_TOP];
875                 counters->ctc_nr_held += delta;
876                 LASSERT(counters->ctc_nr_held >= 0);
877         }
878 }
879
880 /**
881  * Mod(ifie)s cl_lock::cll_users counter for a given lock. See
882  * cl_lock_hold_mod() for the explanation of the debugging code.
883  */
884 static void cl_lock_used_mod(const struct lu_env *env, struct cl_lock *lock,
885                              int delta)
886 {
887         struct cl_thread_counters *counters;
888         enum clt_nesting_level     nesting;
889
890         lock->cll_users += delta;
891         nesting = cl_lock_nesting(lock);
892         if (nesting == CNL_TOP) {
893                 counters = &cl_env_info(env)->clt_counters[CNL_TOP];
894                 counters->ctc_nr_used += delta;
895                 LASSERT(counters->ctc_nr_used >= 0);
896         }
897 }
898
899 void cl_lock_hold_release(const struct lu_env *env, struct cl_lock *lock,
900                           const char *scope, const void *source)
901 {
902         LINVRNT(cl_lock_is_mutexed(lock));
903         LINVRNT(cl_lock_invariant(env, lock));
904         LASSERT(lock->cll_holds > 0);
905
906         ENTRY;
907         cl_lock_trace(D_DLMTRACE, env, "hold release lock", lock);
908         lu_ref_del(&lock->cll_holders, scope, source);
909         cl_lock_hold_mod(env, lock, -1);
910         if (lock->cll_holds == 0) {
911                 CL_LOCK_ASSERT(lock->cll_state != CLS_HELD, env, lock);
912                 if (lock->cll_descr.cld_mode == CLM_PHANTOM ||
913                     lock->cll_descr.cld_mode == CLM_GROUP ||
914                     lock->cll_state != CLS_CACHED)
915                         /*
916                          * If lock is still phantom or grouplock when user is
917                          * done with it---destroy the lock.
918                          */
919                         lock->cll_flags |= CLF_CANCELPEND|CLF_DOOMED;
920                 if (lock->cll_flags & CLF_CANCELPEND) {
921                         lock->cll_flags &= ~CLF_CANCELPEND;
922                         cl_lock_cancel0(env, lock);
923                 }
924                 if (lock->cll_flags & CLF_DOOMED) {
925                         /* no longer doomed: it's dead... Jim. */
926                         lock->cll_flags &= ~CLF_DOOMED;
927                         cl_lock_delete0(env, lock);
928                 }
929         }
930         EXIT;
931 }
932 EXPORT_SYMBOL(cl_lock_hold_release);
933
934 /**
935  * Waits until lock state is changed.
936  *
937  * This function is called with cl_lock mutex locked, atomically releases
938  * mutex and goes to sleep, waiting for a lock state change (signaled by
939  * cl_lock_signal()), and re-acquires the mutex before return.
940  *
941  * This function is used to wait until lock state machine makes some progress
942  * and to emulate synchronous operations on top of asynchronous lock
943  * interface.
944  *
945  * \retval -EINTR wait was interrupted
946  *
947  * \retval 0 wait wasn't interrupted
948  *
949  * \pre cl_lock_is_mutexed(lock)
950  *
951  * \see cl_lock_signal()
952  */
953 int cl_lock_state_wait(const struct lu_env *env, struct cl_lock *lock)
954 {
955         cfs_waitlink_t waiter;
956         cfs_sigset_t blocked;
957         int result;
958
959         ENTRY;
960         LINVRNT(cl_lock_is_mutexed(lock));
961         LINVRNT(cl_lock_invariant(env, lock));
962         LASSERT(lock->cll_depth == 1);
963         LASSERT(lock->cll_state != CLS_FREEING); /* too late to wait */
964
965         cl_lock_trace(D_DLMTRACE, env, "state wait lock", lock);
966         result = lock->cll_error;
967         if (result == 0) {
968                 /* To avoid being interrupted by the 'non-fatal' signals
969                  * (SIGCHLD, for instance), we'd block them temporarily.
970                  * LU-305 */
971                 blocked = cfs_block_sigsinv(LUSTRE_FATAL_SIGS);
972
973                 cfs_waitlink_init(&waiter);
974                 cfs_waitq_add(&lock->cll_wq, &waiter);
975                 cfs_set_current_state(CFS_TASK_INTERRUPTIBLE);
976                 cl_lock_mutex_put(env, lock);
977
978                 LASSERT(cl_lock_nr_mutexed(env) == 0);
979
980                 /* Returning ERESTARTSYS instead of EINTR so syscalls
981                  * can be restarted if signals are pending here */
982                 result = -ERESTARTSYS;
983                 if (likely(!OBD_FAIL_CHECK(OBD_FAIL_LOCK_STATE_WAIT_INTR))) {
984                         cfs_waitq_wait(&waiter, CFS_TASK_INTERRUPTIBLE);
985                         if (!cfs_signal_pending())
986                                 result = 0;
987                 }
988
989                 cl_lock_mutex_get(env, lock);
990                 cfs_set_current_state(CFS_TASK_RUNNING);
991                 cfs_waitq_del(&lock->cll_wq, &waiter);
992
993                 /* Restore old blocked signals */
994                 cfs_restore_sigs(blocked);
995         }
996         RETURN(result);
997 }
998 EXPORT_SYMBOL(cl_lock_state_wait);
999
1000 static void cl_lock_state_signal(const struct lu_env *env, struct cl_lock *lock,
1001                                  enum cl_lock_state state)
1002 {
1003         const struct cl_lock_slice *slice;
1004
1005         ENTRY;
1006         LINVRNT(cl_lock_is_mutexed(lock));
1007         LINVRNT(cl_lock_invariant(env, lock));
1008
1009         cfs_list_for_each_entry(slice, &lock->cll_layers, cls_linkage)
1010                 if (slice->cls_ops->clo_state != NULL)
1011                         slice->cls_ops->clo_state(env, slice, state);
1012         cfs_waitq_broadcast(&lock->cll_wq);
1013         EXIT;
1014 }
1015
1016 /**
1017  * Notifies waiters that lock state changed.
1018  *
1019  * Wakes up all waiters sleeping in cl_lock_state_wait(), also notifies all
1020  * layers about state change by calling cl_lock_operations::clo_state()
1021  * top-to-bottom.
1022  */
1023 void cl_lock_signal(const struct lu_env *env, struct cl_lock *lock)
1024 {
1025         ENTRY;
1026         cl_lock_trace(D_DLMTRACE, env, "state signal lock", lock);
1027         cl_lock_state_signal(env, lock, lock->cll_state);
1028         EXIT;
1029 }
1030 EXPORT_SYMBOL(cl_lock_signal);
1031
1032 /**
1033  * Changes lock state.
1034  *
1035  * This function is invoked to notify layers that lock state changed, possible
1036  * as a result of an asynchronous event such as call-back reception.
1037  *
1038  * \post lock->cll_state == state
1039  *
1040  * \see cl_lock_operations::clo_state()
1041  */
1042 void cl_lock_state_set(const struct lu_env *env, struct cl_lock *lock,
1043                        enum cl_lock_state state)
1044 {
1045         ENTRY;
1046         LASSERT(lock->cll_state <= state ||
1047                 (lock->cll_state == CLS_CACHED &&
1048                  (state == CLS_HELD || /* lock found in cache */
1049                   state == CLS_NEW  ||   /* sub-lock canceled */
1050                   state == CLS_INTRANSIT)) ||
1051                 /* lock is in transit state */
1052                 lock->cll_state == CLS_INTRANSIT);
1053
1054         if (lock->cll_state != state) {
1055                 CS_LOCKSTATE_DEC(lock->cll_descr.cld_obj, lock->cll_state);
1056                 CS_LOCKSTATE_INC(lock->cll_descr.cld_obj, state);
1057
1058                 cl_lock_state_signal(env, lock, state);
1059                 lock->cll_state = state;
1060         }
1061         EXIT;
1062 }
1063 EXPORT_SYMBOL(cl_lock_state_set);
1064
1065 static int cl_unuse_try_internal(const struct lu_env *env, struct cl_lock *lock)
1066 {
1067         const struct cl_lock_slice *slice;
1068         int result;
1069
1070         do {
1071                 result = 0;
1072
1073                 LINVRNT(cl_lock_is_mutexed(lock));
1074                 LINVRNT(cl_lock_invariant(env, lock));
1075                 LASSERT(lock->cll_state == CLS_INTRANSIT);
1076
1077                 result = -ENOSYS;
1078                 cfs_list_for_each_entry_reverse(slice, &lock->cll_layers,
1079                                                 cls_linkage) {
1080                         if (slice->cls_ops->clo_unuse != NULL) {
1081                                 result = slice->cls_ops->clo_unuse(env, slice);
1082                                 if (result != 0)
1083                                         break;
1084                         }
1085                 }
1086                 LASSERT(result != -ENOSYS);
1087         } while (result == CLO_REPEAT);
1088
1089         return result;
1090 }
1091
1092 /**
1093  * Yanks lock from the cache (cl_lock_state::CLS_CACHED state) by calling
1094  * cl_lock_operations::clo_use() top-to-bottom to notify layers.
1095  * @atomic = 1, it must unuse the lock to recovery the lock to keep the
1096  *  use process atomic
1097  */
1098 int cl_use_try(const struct lu_env *env, struct cl_lock *lock, int atomic)
1099 {
1100         const struct cl_lock_slice *slice;
1101         int result;
1102         enum cl_lock_state state;
1103
1104         ENTRY;
1105         cl_lock_trace(D_DLMTRACE, env, "use lock", lock);
1106
1107         LASSERT(lock->cll_state == CLS_CACHED);
1108         if (lock->cll_error)
1109                 RETURN(lock->cll_error);
1110
1111         result = -ENOSYS;
1112         state = cl_lock_intransit(env, lock);
1113         cfs_list_for_each_entry(slice, &lock->cll_layers, cls_linkage) {
1114                 if (slice->cls_ops->clo_use != NULL) {
1115                         result = slice->cls_ops->clo_use(env, slice);
1116                         if (result != 0)
1117                                 break;
1118                 }
1119         }
1120         LASSERT(result != -ENOSYS);
1121
1122         LASSERTF(lock->cll_state == CLS_INTRANSIT, "Wrong state %d.\n",
1123                  lock->cll_state);
1124
1125         if (result == 0) {
1126                 state = CLS_HELD;
1127         } else {
1128                 if (result == -ESTALE) {
1129                         /*
1130                          * ESTALE means sublock being cancelled
1131                          * at this time, and set lock state to
1132                          * be NEW here and ask the caller to repeat.
1133                          */
1134                         state = CLS_NEW;
1135                         result = CLO_REPEAT;
1136                 }
1137
1138                 /* @atomic means back-off-on-failure. */
1139                 if (atomic) {
1140                         int rc;
1141                         rc = cl_unuse_try_internal(env, lock);
1142                         /* Vet the results. */
1143                         if (rc < 0 && result > 0)
1144                                 result = rc;
1145                 }
1146
1147         }
1148         cl_lock_extransit(env, lock, state);
1149         RETURN(result);
1150 }
1151 EXPORT_SYMBOL(cl_use_try);
1152
1153 /**
1154  * Helper for cl_enqueue_try() that calls ->clo_enqueue() across all layers
1155  * top-to-bottom.
1156  */
1157 static int cl_enqueue_kick(const struct lu_env *env,
1158                            struct cl_lock *lock,
1159                            struct cl_io *io, __u32 flags)
1160 {
1161         int result;
1162         const struct cl_lock_slice *slice;
1163
1164         ENTRY;
1165         result = -ENOSYS;
1166         cfs_list_for_each_entry(slice, &lock->cll_layers, cls_linkage) {
1167                 if (slice->cls_ops->clo_enqueue != NULL) {
1168                         result = slice->cls_ops->clo_enqueue(env,
1169                                                              slice, io, flags);
1170                         if (result != 0)
1171                                 break;
1172                 }
1173         }
1174         LASSERT(result != -ENOSYS);
1175         RETURN(result);
1176 }
1177
1178 /**
1179  * Tries to enqueue a lock.
1180  *
1181  * This function is called repeatedly by cl_enqueue() until either lock is
1182  * enqueued, or error occurs. This function does not block waiting for
1183  * networking communication to complete.
1184  *
1185  * \post ergo(result == 0, lock->cll_state == CLS_ENQUEUED ||
1186  *                         lock->cll_state == CLS_HELD)
1187  *
1188  * \see cl_enqueue() cl_lock_operations::clo_enqueue()
1189  * \see cl_lock_state::CLS_ENQUEUED
1190  */
1191 int cl_enqueue_try(const struct lu_env *env, struct cl_lock *lock,
1192                    struct cl_io *io, __u32 flags)
1193 {
1194         int result;
1195
1196         ENTRY;
1197         cl_lock_trace(D_DLMTRACE, env, "enqueue lock", lock);
1198         do {
1199                 LINVRNT(cl_lock_is_mutexed(lock));
1200
1201                 result = lock->cll_error;
1202                 if (result != 0)
1203                         break;
1204
1205                 switch (lock->cll_state) {
1206                 case CLS_NEW:
1207                         cl_lock_state_set(env, lock, CLS_QUEUING);
1208                         /* fall-through */
1209                 case CLS_QUEUING:
1210                         /* kick layers. */
1211                         result = cl_enqueue_kick(env, lock, io, flags);
1212                         /* For AGL case, the cl_lock::cll_state may
1213                          * become CLS_HELD already. */
1214                         if (result == 0 && lock->cll_state == CLS_QUEUING)
1215                                 cl_lock_state_set(env, lock, CLS_ENQUEUED);
1216                         break;
1217                 case CLS_INTRANSIT:
1218                         LASSERT(cl_lock_is_intransit(lock));
1219                         result = CLO_WAIT;
1220                         break;
1221                 case CLS_CACHED:
1222                         /* yank lock from the cache. */
1223                         result = cl_use_try(env, lock, 0);
1224                         break;
1225                 case CLS_ENQUEUED:
1226                 case CLS_HELD:
1227                         result = 0;
1228                         break;
1229                 default:
1230                 case CLS_FREEING:
1231                         /*
1232                          * impossible, only held locks with increased
1233                          * ->cll_holds can be enqueued, and they cannot be
1234                          * freed.
1235                          */
1236                         LBUG();
1237                 }
1238         } while (result == CLO_REPEAT);
1239         RETURN(result);
1240 }
1241 EXPORT_SYMBOL(cl_enqueue_try);
1242
1243 /**
1244  * Cancel the conflicting lock found during previous enqueue.
1245  *
1246  * \retval 0 conflicting lock has been canceled.
1247  * \retval -ve error code.
1248  */
1249 int cl_lock_enqueue_wait(const struct lu_env *env,
1250                          struct cl_lock *lock,
1251                          int keep_mutex)
1252 {
1253         struct cl_lock  *conflict;
1254         int              rc = 0;
1255         ENTRY;
1256
1257         LASSERT(cl_lock_is_mutexed(lock));
1258         LASSERT(lock->cll_state == CLS_QUEUING);
1259         LASSERT(lock->cll_conflict != NULL);
1260
1261         conflict = lock->cll_conflict;
1262         lock->cll_conflict = NULL;
1263
1264         cl_lock_mutex_put(env, lock);
1265         LASSERT(cl_lock_nr_mutexed(env) == 0);
1266
1267         cl_lock_mutex_get(env, conflict);
1268         cl_lock_trace(D_DLMTRACE, env, "enqueue wait", conflict);
1269         cl_lock_cancel(env, conflict);
1270         cl_lock_delete(env, conflict);
1271
1272         while (conflict->cll_state != CLS_FREEING) {
1273                 rc = cl_lock_state_wait(env, conflict);
1274                 if (rc != 0)
1275                         break;
1276         }
1277         cl_lock_mutex_put(env, conflict);
1278         lu_ref_del(&conflict->cll_reference, "cancel-wait", lock);
1279         cl_lock_put(env, conflict);
1280
1281         if (keep_mutex)
1282                 cl_lock_mutex_get(env, lock);
1283
1284         LASSERT(rc <= 0);
1285         RETURN(rc);
1286 }
1287 EXPORT_SYMBOL(cl_lock_enqueue_wait);
1288
1289 static int cl_enqueue_locked(const struct lu_env *env, struct cl_lock *lock,
1290                              struct cl_io *io, __u32 enqflags)
1291 {
1292         int result;
1293
1294         ENTRY;
1295
1296         LINVRNT(cl_lock_is_mutexed(lock));
1297         LINVRNT(cl_lock_invariant(env, lock));
1298         LASSERT(lock->cll_holds > 0);
1299
1300         cl_lock_user_add(env, lock);
1301         do {
1302                 result = cl_enqueue_try(env, lock, io, enqflags);
1303                 if (result == CLO_WAIT) {
1304                         if (lock->cll_conflict != NULL)
1305                                 result = cl_lock_enqueue_wait(env, lock, 1);
1306                         else
1307                                 result = cl_lock_state_wait(env, lock);
1308                         if (result == 0)
1309                                 continue;
1310                 }
1311                 break;
1312         } while (1);
1313         if (result != 0)
1314                 cl_unuse_try(env, lock);
1315         LASSERT(ergo(result == 0 && !(enqflags & CEF_AGL),
1316                      lock->cll_state == CLS_ENQUEUED ||
1317                      lock->cll_state == CLS_HELD));
1318         RETURN(result);
1319 }
1320
1321 /**
1322  * Enqueues a lock.
1323  *
1324  * \pre current thread or io owns a hold on lock.
1325  *
1326  * \post ergo(result == 0, lock->users increased)
1327  * \post ergo(result == 0, lock->cll_state == CLS_ENQUEUED ||
1328  *                         lock->cll_state == CLS_HELD)
1329  */
1330 int cl_enqueue(const struct lu_env *env, struct cl_lock *lock,
1331                struct cl_io *io, __u32 enqflags)
1332 {
1333         int result;
1334
1335         ENTRY;
1336
1337         cl_lock_lockdep_acquire(env, lock, enqflags);
1338         cl_lock_mutex_get(env, lock);
1339         result = cl_enqueue_locked(env, lock, io, enqflags);
1340         cl_lock_mutex_put(env, lock);
1341         if (result != 0)
1342                 cl_lock_lockdep_release(env, lock);
1343         LASSERT(ergo(result == 0, lock->cll_state == CLS_ENQUEUED ||
1344                      lock->cll_state == CLS_HELD));
1345         RETURN(result);
1346 }
1347 EXPORT_SYMBOL(cl_enqueue);
1348
1349 /**
1350  * Tries to unlock a lock.
1351  *
1352  * This function is called to release underlying resource:
1353  * 1. for top lock, the resource is sublocks it held;
1354  * 2. for sublock, the resource is the reference to dlmlock.
1355  *
1356  * cl_unuse_try is a one-shot operation, so it must NOT return CLO_WAIT.
1357  *
1358  * \see cl_unuse() cl_lock_operations::clo_unuse()
1359  * \see cl_lock_state::CLS_CACHED
1360  */
1361 int cl_unuse_try(const struct lu_env *env, struct cl_lock *lock)
1362 {
1363         int                         result;
1364         enum cl_lock_state          state = CLS_NEW;
1365
1366         ENTRY;
1367         cl_lock_trace(D_DLMTRACE, env, "unuse lock", lock);
1368
1369         if (lock->cll_users > 1) {
1370                 cl_lock_user_del(env, lock);
1371                 RETURN(0);
1372         }
1373
1374         /* Only if the lock is in CLS_HELD or CLS_ENQUEUED state, it can hold
1375          * underlying resources. */
1376         if (!(lock->cll_state == CLS_HELD || lock->cll_state == CLS_ENQUEUED)) {
1377                 cl_lock_user_del(env, lock);
1378                 RETURN(0);
1379         }
1380
1381         /*
1382          * New lock users (->cll_users) are not protecting unlocking
1383          * from proceeding. From this point, lock eventually reaches
1384          * CLS_CACHED, is reinitialized to CLS_NEW or fails into
1385          * CLS_FREEING.
1386          */
1387         state = cl_lock_intransit(env, lock);
1388
1389         result = cl_unuse_try_internal(env, lock);
1390         LASSERT(lock->cll_state == CLS_INTRANSIT);
1391         LASSERT(result != CLO_WAIT);
1392         cl_lock_user_del(env, lock);
1393         if (result == 0 || result == -ESTALE) {
1394                 /*
1395                  * Return lock back to the cache. This is the only
1396                  * place where lock is moved into CLS_CACHED state.
1397                  *
1398                  * If one of ->clo_unuse() methods returned -ESTALE, lock
1399                  * cannot be placed into cache and has to be
1400                  * re-initialized. This happens e.g., when a sub-lock was
1401                  * canceled while unlocking was in progress.
1402                  */
1403                 if (state == CLS_HELD && result == 0)
1404                         state = CLS_CACHED;
1405                 else
1406                         state = CLS_NEW;
1407                 cl_lock_extransit(env, lock, state);
1408
1409                 /*
1410                  * Hide -ESTALE error.
1411                  * If the lock is a glimpse lock, and it has multiple
1412                  * stripes. Assuming that one of its sublock returned -ENAVAIL,
1413                  * and other sublocks are matched write locks. In this case,
1414                  * we can't set this lock to error because otherwise some of
1415                  * its sublocks may not be canceled. This causes some dirty
1416                  * pages won't be written to OSTs. -jay
1417                  */
1418                 result = 0;
1419         } else {
1420                 CERROR("result = %d, this is unlikely!\n", result);
1421                 state = CLS_NEW;
1422                 cl_lock_extransit(env, lock, state);
1423         }
1424         RETURN(result ?: lock->cll_error);
1425 }
1426 EXPORT_SYMBOL(cl_unuse_try);
1427
1428 static void cl_unuse_locked(const struct lu_env *env, struct cl_lock *lock)
1429 {
1430         int result;
1431         ENTRY;
1432
1433         result = cl_unuse_try(env, lock);
1434         if (result)
1435                 CL_LOCK_DEBUG(D_ERROR, env, lock, "unuse return %d\n", result);
1436
1437         EXIT;
1438 }
1439
1440 /**
1441  * Unlocks a lock.
1442  */
1443 void cl_unuse(const struct lu_env *env, struct cl_lock *lock)
1444 {
1445         ENTRY;
1446         cl_lock_mutex_get(env, lock);
1447         cl_unuse_locked(env, lock);
1448         cl_lock_mutex_put(env, lock);
1449         cl_lock_lockdep_release(env, lock);
1450         EXIT;
1451 }
1452 EXPORT_SYMBOL(cl_unuse);
1453
1454 /**
1455  * Tries to wait for a lock.
1456  *
1457  * This function is called repeatedly by cl_wait() until either lock is
1458  * granted, or error occurs. This function does not block waiting for network
1459  * communication to complete.
1460  *
1461  * \see cl_wait() cl_lock_operations::clo_wait()
1462  * \see cl_lock_state::CLS_HELD
1463  */
1464 int cl_wait_try(const struct lu_env *env, struct cl_lock *lock)
1465 {
1466         const struct cl_lock_slice *slice;
1467         int                         result;
1468
1469         ENTRY;
1470         cl_lock_trace(D_DLMTRACE, env, "wait lock try", lock);
1471         do {
1472                 LINVRNT(cl_lock_is_mutexed(lock));
1473                 LINVRNT(cl_lock_invariant(env, lock));
1474                 LASSERTF(lock->cll_state == CLS_QUEUING ||
1475                          lock->cll_state == CLS_ENQUEUED ||
1476                          lock->cll_state == CLS_HELD ||
1477                          lock->cll_state == CLS_INTRANSIT,
1478                          "lock state: %d\n", lock->cll_state);
1479                 LASSERT(lock->cll_users > 0);
1480                 LASSERT(lock->cll_holds > 0);
1481
1482                 result = lock->cll_error;
1483                 if (result != 0)
1484                         break;
1485
1486                 if (cl_lock_is_intransit(lock)) {
1487                         result = CLO_WAIT;
1488                         break;
1489                 }
1490
1491                 if (lock->cll_state == CLS_HELD)
1492                         /* nothing to do */
1493                         break;
1494
1495                 result = -ENOSYS;
1496                 cfs_list_for_each_entry(slice, &lock->cll_layers, cls_linkage) {
1497                         if (slice->cls_ops->clo_wait != NULL) {
1498                                 result = slice->cls_ops->clo_wait(env, slice);
1499                                 if (result != 0)
1500                                         break;
1501                         }
1502                 }
1503                 LASSERT(result != -ENOSYS);
1504                 if (result == 0) {
1505                         LASSERT(lock->cll_state != CLS_INTRANSIT);
1506                         cl_lock_state_set(env, lock, CLS_HELD);
1507                 }
1508         } while (result == CLO_REPEAT);
1509         RETURN(result);
1510 }
1511 EXPORT_SYMBOL(cl_wait_try);
1512
1513 /**
1514  * Waits until enqueued lock is granted.
1515  *
1516  * \pre current thread or io owns a hold on the lock
1517  * \pre ergo(result == 0, lock->cll_state == CLS_ENQUEUED ||
1518  *                        lock->cll_state == CLS_HELD)
1519  *
1520  * \post ergo(result == 0, lock->cll_state == CLS_HELD)
1521  */
1522 int cl_wait(const struct lu_env *env, struct cl_lock *lock)
1523 {
1524         int result;
1525
1526         ENTRY;
1527         cl_lock_mutex_get(env, lock);
1528
1529         LINVRNT(cl_lock_invariant(env, lock));
1530         LASSERTF(lock->cll_state == CLS_ENQUEUED || lock->cll_state == CLS_HELD,
1531                  "Wrong state %d \n", lock->cll_state);
1532         LASSERT(lock->cll_holds > 0);
1533
1534         do {
1535                 result = cl_wait_try(env, lock);
1536                 if (result == CLO_WAIT) {
1537                         result = cl_lock_state_wait(env, lock);
1538                         if (result == 0)
1539                                 continue;
1540                 }
1541                 break;
1542         } while (1);
1543         if (result < 0) {
1544                 cl_unuse_try(env, lock);
1545                 cl_lock_lockdep_release(env, lock);
1546         }
1547         cl_lock_trace(D_DLMTRACE, env, "wait lock", lock);
1548         cl_lock_mutex_put(env, lock);
1549         LASSERT(ergo(result == 0, lock->cll_state == CLS_HELD));
1550         RETURN(result);
1551 }
1552 EXPORT_SYMBOL(cl_wait);
1553
1554 /**
1555  * Executes cl_lock_operations::clo_weigh(), and sums results to estimate lock
1556  * value.
1557  */
1558 unsigned long cl_lock_weigh(const struct lu_env *env, struct cl_lock *lock)
1559 {
1560         const struct cl_lock_slice *slice;
1561         unsigned long pound;
1562         unsigned long ounce;
1563
1564         ENTRY;
1565         LINVRNT(cl_lock_is_mutexed(lock));
1566         LINVRNT(cl_lock_invariant(env, lock));
1567
1568         pound = 0;
1569         cfs_list_for_each_entry_reverse(slice, &lock->cll_layers, cls_linkage) {
1570                 if (slice->cls_ops->clo_weigh != NULL) {
1571                         ounce = slice->cls_ops->clo_weigh(env, slice);
1572                         pound += ounce;
1573                         if (pound < ounce) /* over-weight^Wflow */
1574                                 pound = ~0UL;
1575                 }
1576         }
1577         RETURN(pound);
1578 }
1579 EXPORT_SYMBOL(cl_lock_weigh);
1580
1581 /**
1582  * Notifies layers that lock description changed.
1583  *
1584  * The server can grant client a lock different from one that was requested
1585  * (e.g., larger in extent). This method is called when actually granted lock
1586  * description becomes known to let layers to accommodate for changed lock
1587  * description.
1588  *
1589  * \see cl_lock_operations::clo_modify()
1590  */
1591 int cl_lock_modify(const struct lu_env *env, struct cl_lock *lock,
1592                    const struct cl_lock_descr *desc)
1593 {
1594         const struct cl_lock_slice *slice;
1595         struct cl_object           *obj = lock->cll_descr.cld_obj;
1596         struct cl_object_header    *hdr = cl_object_header(obj);
1597         int result;
1598
1599         ENTRY;
1600         cl_lock_trace(D_DLMTRACE, env, "modify lock", lock);
1601         /* don't allow object to change */
1602         LASSERT(obj == desc->cld_obj);
1603         LINVRNT(cl_lock_is_mutexed(lock));
1604         LINVRNT(cl_lock_invariant(env, lock));
1605
1606         cfs_list_for_each_entry_reverse(slice, &lock->cll_layers, cls_linkage) {
1607                 if (slice->cls_ops->clo_modify != NULL) {
1608                         result = slice->cls_ops->clo_modify(env, slice, desc);
1609                         if (result != 0)
1610                                 RETURN(result);
1611                 }
1612         }
1613         CL_LOCK_DEBUG(D_DLMTRACE, env, lock, " -> "DDESCR"@"DFID"\n",
1614                       PDESCR(desc), PFID(lu_object_fid(&desc->cld_obj->co_lu)));
1615         /*
1616          * Just replace description in place. Nothing more is needed for
1617          * now. If locks were indexed according to their extent and/or mode,
1618          * that index would have to be updated here.
1619          */
1620         spin_lock(&hdr->coh_lock_guard);
1621         lock->cll_descr = *desc;
1622         spin_unlock(&hdr->coh_lock_guard);
1623         RETURN(0);
1624 }
1625 EXPORT_SYMBOL(cl_lock_modify);
1626
1627 /**
1628  * Initializes lock closure with a given origin.
1629  *
1630  * \see cl_lock_closure
1631  */
1632 void cl_lock_closure_init(const struct lu_env *env,
1633                           struct cl_lock_closure *closure,
1634                           struct cl_lock *origin, int wait)
1635 {
1636         LINVRNT(cl_lock_is_mutexed(origin));
1637         LINVRNT(cl_lock_invariant(env, origin));
1638
1639         CFS_INIT_LIST_HEAD(&closure->clc_list);
1640         closure->clc_origin = origin;
1641         closure->clc_wait   = wait;
1642         closure->clc_nr     = 0;
1643 }
1644 EXPORT_SYMBOL(cl_lock_closure_init);
1645
1646 /**
1647  * Builds a closure of \a lock.
1648  *
1649  * Building of a closure consists of adding initial lock (\a lock) into it,
1650  * and calling cl_lock_operations::clo_closure() methods of \a lock. These
1651  * methods might call cl_lock_closure_build() recursively again, adding more
1652  * locks to the closure, etc.
1653  *
1654  * \see cl_lock_closure
1655  */
1656 int cl_lock_closure_build(const struct lu_env *env, struct cl_lock *lock,
1657                           struct cl_lock_closure *closure)
1658 {
1659         const struct cl_lock_slice *slice;
1660         int result;
1661
1662         ENTRY;
1663         LINVRNT(cl_lock_is_mutexed(closure->clc_origin));
1664         LINVRNT(cl_lock_invariant(env, closure->clc_origin));
1665
1666         result = cl_lock_enclosure(env, lock, closure);
1667         if (result == 0) {
1668                 cfs_list_for_each_entry(slice, &lock->cll_layers, cls_linkage) {
1669                         if (slice->cls_ops->clo_closure != NULL) {
1670                                 result = slice->cls_ops->clo_closure(env, slice,
1671                                                                      closure);
1672                                 if (result != 0)
1673                                         break;
1674                         }
1675                 }
1676         }
1677         if (result != 0)
1678                 cl_lock_disclosure(env, closure);
1679         RETURN(result);
1680 }
1681 EXPORT_SYMBOL(cl_lock_closure_build);
1682
1683 /**
1684  * Adds new lock to a closure.
1685  *
1686  * Try-locks \a lock and if succeeded, adds it to the closure (never more than
1687  * once). If try-lock failed, returns CLO_REPEAT, after optionally waiting
1688  * until next try-lock is likely to succeed.
1689  */
1690 int cl_lock_enclosure(const struct lu_env *env, struct cl_lock *lock,
1691                       struct cl_lock_closure *closure)
1692 {
1693         int result = 0;
1694         ENTRY;
1695         cl_lock_trace(D_DLMTRACE, env, "enclosure lock", lock);
1696         if (!cl_lock_mutex_try(env, lock)) {
1697                 /*
1698                  * If lock->cll_inclosure is not empty, lock is already in
1699                  * this closure.
1700                  */
1701                 if (cfs_list_empty(&lock->cll_inclosure)) {
1702                         cl_lock_get_trust(lock);
1703                         lu_ref_add(&lock->cll_reference, "closure", closure);
1704                         cfs_list_add(&lock->cll_inclosure, &closure->clc_list);
1705                         closure->clc_nr++;
1706                 } else
1707                         cl_lock_mutex_put(env, lock);
1708                 result = 0;
1709         } else {
1710                 cl_lock_disclosure(env, closure);
1711                 if (closure->clc_wait) {
1712                         cl_lock_get_trust(lock);
1713                         lu_ref_add(&lock->cll_reference, "closure-w", closure);
1714                         cl_lock_mutex_put(env, closure->clc_origin);
1715
1716                         LASSERT(cl_lock_nr_mutexed(env) == 0);
1717                         cl_lock_mutex_get(env, lock);
1718                         cl_lock_mutex_put(env, lock);
1719
1720                         cl_lock_mutex_get(env, closure->clc_origin);
1721                         lu_ref_del(&lock->cll_reference, "closure-w", closure);
1722                         cl_lock_put(env, lock);
1723                 }
1724                 result = CLO_REPEAT;
1725         }
1726         RETURN(result);
1727 }
1728 EXPORT_SYMBOL(cl_lock_enclosure);
1729
1730 /** Releases mutices of enclosed locks. */
1731 void cl_lock_disclosure(const struct lu_env *env,
1732                         struct cl_lock_closure *closure)
1733 {
1734         struct cl_lock *scan;
1735         struct cl_lock *temp;
1736
1737         cl_lock_trace(D_DLMTRACE, env, "disclosure lock", closure->clc_origin);
1738         cfs_list_for_each_entry_safe(scan, temp, &closure->clc_list,
1739                                      cll_inclosure){
1740                 cfs_list_del_init(&scan->cll_inclosure);
1741                 cl_lock_mutex_put(env, scan);
1742                 lu_ref_del(&scan->cll_reference, "closure", closure);
1743                 cl_lock_put(env, scan);
1744                 closure->clc_nr--;
1745         }
1746         LASSERT(closure->clc_nr == 0);
1747 }
1748 EXPORT_SYMBOL(cl_lock_disclosure);
1749
1750 /** Finalizes a closure. */
1751 void cl_lock_closure_fini(struct cl_lock_closure *closure)
1752 {
1753         LASSERT(closure->clc_nr == 0);
1754         LASSERT(cfs_list_empty(&closure->clc_list));
1755 }
1756 EXPORT_SYMBOL(cl_lock_closure_fini);
1757
1758 /**
1759  * Destroys this lock. Notifies layers (bottom-to-top) that lock is being
1760  * destroyed, then destroy the lock. If there are holds on the lock, postpone
1761  * destruction until all holds are released. This is called when a decision is
1762  * made to destroy the lock in the future. E.g., when a blocking AST is
1763  * received on it, or fatal communication error happens.
1764  *
1765  * Caller must have a reference on this lock to prevent a situation, when
1766  * deleted lock lingers in memory for indefinite time, because nobody calls
1767  * cl_lock_put() to finish it.
1768  *
1769  * \pre atomic_read(&lock->cll_ref) > 0
1770  * \pre ergo(cl_lock_nesting(lock) == CNL_TOP,
1771  *           cl_lock_nr_mutexed(env) == 1)
1772  *      [i.e., if a top-lock is deleted, mutices of no other locks can be
1773  *      held, as deletion of sub-locks might require releasing a top-lock
1774  *      mutex]
1775  *
1776  * \see cl_lock_operations::clo_delete()
1777  * \see cl_lock::cll_holds
1778  */
1779 void cl_lock_delete(const struct lu_env *env, struct cl_lock *lock)
1780 {
1781         LINVRNT(cl_lock_is_mutexed(lock));
1782         LINVRNT(cl_lock_invariant(env, lock));
1783         LASSERT(ergo(cl_lock_nesting(lock) == CNL_TOP,
1784                      cl_lock_nr_mutexed(env) == 1));
1785
1786         ENTRY;
1787         cl_lock_trace(D_DLMTRACE, env, "delete lock", lock);
1788         if (lock->cll_holds == 0)
1789                 cl_lock_delete0(env, lock);
1790         else
1791                 lock->cll_flags |= CLF_DOOMED;
1792         EXIT;
1793 }
1794 EXPORT_SYMBOL(cl_lock_delete);
1795
1796 /**
1797  * Mark lock as irrecoverably failed, and mark it for destruction. This
1798  * happens when, e.g., server fails to grant a lock to us, or networking
1799  * time-out happens.
1800  *
1801  * \pre atomic_read(&lock->cll_ref) > 0
1802  *
1803  * \see clo_lock_delete()
1804  * \see cl_lock::cll_holds
1805  */
1806 void cl_lock_error(const struct lu_env *env, struct cl_lock *lock, int error)
1807 {
1808         LINVRNT(cl_lock_is_mutexed(lock));
1809         LINVRNT(cl_lock_invariant(env, lock));
1810
1811         ENTRY;
1812         if (lock->cll_error == 0 && error != 0) {
1813                 cl_lock_trace(D_DLMTRACE, env, "set lock error", lock);
1814                 lock->cll_error = error;
1815                 cl_lock_signal(env, lock);
1816                 cl_lock_cancel(env, lock);
1817                 cl_lock_delete(env, lock);
1818         }
1819         EXIT;
1820 }
1821 EXPORT_SYMBOL(cl_lock_error);
1822
1823 /**
1824  * Cancels this lock. Notifies layers
1825  * (bottom-to-top) that lock is being cancelled, then destroy the lock. If
1826  * there are holds on the lock, postpone cancellation until
1827  * all holds are released.
1828  *
1829  * Cancellation notification is delivered to layers at most once.
1830  *
1831  * \see cl_lock_operations::clo_cancel()
1832  * \see cl_lock::cll_holds
1833  */
1834 void cl_lock_cancel(const struct lu_env *env, struct cl_lock *lock)
1835 {
1836         LINVRNT(cl_lock_is_mutexed(lock));
1837         LINVRNT(cl_lock_invariant(env, lock));
1838
1839         ENTRY;
1840         cl_lock_trace(D_DLMTRACE, env, "cancel lock", lock);
1841         if (lock->cll_holds == 0)
1842                 cl_lock_cancel0(env, lock);
1843         else
1844                 lock->cll_flags |= CLF_CANCELPEND;
1845         EXIT;
1846 }
1847 EXPORT_SYMBOL(cl_lock_cancel);
1848
1849 /**
1850  * Finds an existing lock covering given index and optionally different from a
1851  * given \a except lock.
1852  */
1853 struct cl_lock *cl_lock_at_pgoff(const struct lu_env *env,
1854                                  struct cl_object *obj, pgoff_t index,
1855                                  struct cl_lock *except,
1856                                  int pending, int canceld)
1857 {
1858         struct cl_object_header *head;
1859         struct cl_lock          *scan;
1860         struct cl_lock          *lock;
1861         struct cl_lock_descr    *need;
1862
1863         ENTRY;
1864
1865         head = cl_object_header(obj);
1866         need = &cl_env_info(env)->clt_descr;
1867         lock = NULL;
1868
1869         need->cld_mode = CLM_READ; /* CLM_READ matches both READ & WRITE, but
1870                                     * not PHANTOM */
1871         need->cld_start = need->cld_end = index;
1872         need->cld_enq_flags = 0;
1873
1874         spin_lock(&head->coh_lock_guard);
1875         /* It is fine to match any group lock since there could be only one
1876          * with a uniq gid and it conflicts with all other lock modes too */
1877         cfs_list_for_each_entry(scan, &head->coh_locks, cll_linkage) {
1878                 if (scan != except &&
1879                     (scan->cll_descr.cld_mode == CLM_GROUP ||
1880                     cl_lock_ext_match(&scan->cll_descr, need)) &&
1881                     scan->cll_state >= CLS_HELD &&
1882                     scan->cll_state < CLS_FREEING &&
1883                     /*
1884                      * This check is racy as the lock can be canceled right
1885                      * after it is done, but this is fine, because page exists
1886                      * already.
1887                      */
1888                     (canceld || !(scan->cll_flags & CLF_CANCELLED)) &&
1889                     (pending || !(scan->cll_flags & CLF_CANCELPEND))) {
1890                         /* Don't increase cs_hit here since this
1891                          * is just a helper function. */
1892                         cl_lock_get_trust(scan);
1893                         lock = scan;
1894                         break;
1895                 }
1896         }
1897         spin_unlock(&head->coh_lock_guard);
1898         RETURN(lock);
1899 }
1900 EXPORT_SYMBOL(cl_lock_at_pgoff);
1901
1902 /**
1903  * Calculate the page offset at the layer of @lock.
1904  * At the time of this writing, @page is top page and @lock is sub lock.
1905  */
1906 static pgoff_t pgoff_at_lock(struct cl_page *page, struct cl_lock *lock)
1907 {
1908         struct lu_device_type *dtype;
1909         const struct cl_page_slice *slice;
1910
1911         dtype = lock->cll_descr.cld_obj->co_lu.lo_dev->ld_type;
1912         slice = cl_page_at(page, dtype);
1913         LASSERT(slice != NULL);
1914         return slice->cpl_page->cp_index;
1915 }
1916
1917 /**
1918  * Check if page @page is covered by an extra lock or discard it.
1919  */
1920 static int check_and_discard_cb(const struct lu_env *env, struct cl_io *io,
1921                                 struct cl_page *page, void *cbdata)
1922 {
1923         struct cl_thread_info *info = cl_env_info(env);
1924         struct cl_lock *lock = cbdata;
1925         pgoff_t index = pgoff_at_lock(page, lock);
1926
1927         if (index >= info->clt_fn_index) {
1928                 struct cl_lock *tmp;
1929
1930                 /* refresh non-overlapped index */
1931                 tmp = cl_lock_at_pgoff(env, lock->cll_descr.cld_obj, index,
1932                                         lock, 1, 0);
1933                 if (tmp != NULL) {
1934                         /* Cache the first-non-overlapped index so as to skip
1935                          * all pages within [index, clt_fn_index). This
1936                          * is safe because if tmp lock is canceled, it will
1937                          * discard these pages. */
1938                         info->clt_fn_index = tmp->cll_descr.cld_end + 1;
1939                         if (tmp->cll_descr.cld_end == CL_PAGE_EOF)
1940                                 info->clt_fn_index = CL_PAGE_EOF;
1941                         cl_lock_put(env, tmp);
1942                 } else if (cl_page_own(env, io, page) == 0) {
1943                         /* discard the page */
1944                         cl_page_unmap(env, io, page);
1945                         cl_page_discard(env, io, page);
1946                         cl_page_disown(env, io, page);
1947                 } else {
1948                         LASSERT(page->cp_state == CPS_FREEING);
1949                 }
1950         }
1951
1952         info->clt_next_index = index + 1;
1953         return CLP_GANG_OKAY;
1954 }
1955
1956 static int discard_cb(const struct lu_env *env, struct cl_io *io,
1957                       struct cl_page *page, void *cbdata)
1958 {
1959         struct cl_thread_info *info = cl_env_info(env);
1960         struct cl_lock *lock   = cbdata;
1961
1962         LASSERT(lock->cll_descr.cld_mode >= CLM_WRITE);
1963         KLASSERT(ergo(page->cp_type == CPT_CACHEABLE,
1964                       !PageWriteback(cl_page_vmpage(env, page))));
1965         KLASSERT(ergo(page->cp_type == CPT_CACHEABLE,
1966                       !PageDirty(cl_page_vmpage(env, page))));
1967
1968         info->clt_next_index = pgoff_at_lock(page, lock) + 1;
1969         if (cl_page_own(env, io, page) == 0) {
1970                 /* discard the page */
1971                 cl_page_unmap(env, io, page);
1972                 cl_page_discard(env, io, page);
1973                 cl_page_disown(env, io, page);
1974         } else {
1975                 LASSERT(page->cp_state == CPS_FREEING);
1976         }
1977
1978         return CLP_GANG_OKAY;
1979 }
1980
1981 /**
1982  * Discard pages protected by the given lock. This function traverses radix
1983  * tree to find all covering pages and discard them. If a page is being covered
1984  * by other locks, it should remain in cache.
1985  *
1986  * If error happens on any step, the process continues anyway (the reasoning
1987  * behind this being that lock cancellation cannot be delayed indefinitely).
1988  */
1989 int cl_lock_discard_pages(const struct lu_env *env, struct cl_lock *lock)
1990 {
1991         struct cl_thread_info *info  = cl_env_info(env);
1992         struct cl_io          *io    = &info->clt_io;
1993         struct cl_lock_descr  *descr = &lock->cll_descr;
1994         cl_page_gang_cb_t      cb;
1995         int res;
1996         int result;
1997
1998         LINVRNT(cl_lock_invariant(env, lock));
1999         ENTRY;
2000
2001         io->ci_obj = cl_object_top(descr->cld_obj);
2002         io->ci_ignore_layout = 1;
2003         result = cl_io_init(env, io, CIT_MISC, io->ci_obj);
2004         if (result != 0)
2005                 GOTO(out, result);
2006
2007         cb = descr->cld_mode == CLM_READ ? check_and_discard_cb : discard_cb;
2008         info->clt_fn_index = info->clt_next_index = descr->cld_start;
2009         do {
2010                 res = cl_page_gang_lookup(env, descr->cld_obj, io,
2011                                           info->clt_next_index, descr->cld_end,
2012                                           cb, (void *)lock);
2013                 if (info->clt_next_index > descr->cld_end)
2014                         break;
2015
2016                 if (res == CLP_GANG_RESCHED)
2017                         cfs_cond_resched();
2018         } while (res != CLP_GANG_OKAY);
2019 out:
2020         cl_io_fini(env, io);
2021         RETURN(result);
2022 }
2023 EXPORT_SYMBOL(cl_lock_discard_pages);
2024
2025 /**
2026  * Eliminate all locks for a given object.
2027  *
2028  * Caller has to guarantee that no lock is in active use.
2029  *
2030  * \param cancel when this is set, cl_locks_prune() cancels locks before
2031  *               destroying.
2032  */
2033 void cl_locks_prune(const struct lu_env *env, struct cl_object *obj, int cancel)
2034 {
2035         struct cl_object_header *head;
2036         struct cl_lock          *lock;
2037
2038         ENTRY;
2039         head = cl_object_header(obj);
2040         /*
2041          * If locks are destroyed without cancellation, all pages must be
2042          * already destroyed (as otherwise they will be left unprotected).
2043          */
2044         LASSERT(ergo(!cancel,
2045                      head->coh_tree.rnode == NULL && head->coh_pages == 0));
2046
2047         spin_lock(&head->coh_lock_guard);
2048         while (!cfs_list_empty(&head->coh_locks)) {
2049                 lock = container_of(head->coh_locks.next,
2050                                     struct cl_lock, cll_linkage);
2051                 cl_lock_get_trust(lock);
2052                 spin_unlock(&head->coh_lock_guard);
2053                 lu_ref_add(&lock->cll_reference, "prune", cfs_current());
2054
2055 again:
2056                 cl_lock_mutex_get(env, lock);
2057                 if (lock->cll_state < CLS_FREEING) {
2058                         LASSERT(lock->cll_users <= 1);
2059                         if (unlikely(lock->cll_users == 1)) {
2060                                 struct l_wait_info lwi = { 0 };
2061
2062                                 cl_lock_mutex_put(env, lock);
2063                                 l_wait_event(lock->cll_wq,
2064                                              lock->cll_users == 0,
2065                                              &lwi);
2066                                 goto again;
2067                         }
2068
2069                         if (cancel)
2070                                 cl_lock_cancel(env, lock);
2071                         cl_lock_delete(env, lock);
2072                 }
2073                 cl_lock_mutex_put(env, lock);
2074                 lu_ref_del(&lock->cll_reference, "prune", cfs_current());
2075                 cl_lock_put(env, lock);
2076                 spin_lock(&head->coh_lock_guard);
2077         }
2078         spin_unlock(&head->coh_lock_guard);
2079         EXIT;
2080 }
2081 EXPORT_SYMBOL(cl_locks_prune);
2082
2083 static struct cl_lock *cl_lock_hold_mutex(const struct lu_env *env,
2084                                           const struct cl_io *io,
2085                                           const struct cl_lock_descr *need,
2086                                           const char *scope, const void *source)
2087 {
2088         struct cl_lock *lock;
2089
2090         ENTRY;
2091
2092         while (1) {
2093                 lock = cl_lock_find(env, io, need);
2094                 if (IS_ERR(lock))
2095                         break;
2096                 cl_lock_mutex_get(env, lock);
2097                 if (lock->cll_state < CLS_FREEING &&
2098                     !(lock->cll_flags & CLF_CANCELLED)) {
2099                         cl_lock_hold_mod(env, lock, +1);
2100                         lu_ref_add(&lock->cll_holders, scope, source);
2101                         lu_ref_add(&lock->cll_reference, scope, source);
2102                         break;
2103                 }
2104                 cl_lock_mutex_put(env, lock);
2105                 cl_lock_put(env, lock);
2106         }
2107         RETURN(lock);
2108 }
2109
2110 /**
2111  * Returns a lock matching \a need description with a reference and a hold on
2112  * it.
2113  *
2114  * This is much like cl_lock_find(), except that cl_lock_hold() additionally
2115  * guarantees that lock is not in the CLS_FREEING state on return.
2116  */
2117 struct cl_lock *cl_lock_hold(const struct lu_env *env, const struct cl_io *io,
2118                              const struct cl_lock_descr *need,
2119                              const char *scope, const void *source)
2120 {
2121         struct cl_lock *lock;
2122
2123         ENTRY;
2124
2125         lock = cl_lock_hold_mutex(env, io, need, scope, source);
2126         if (!IS_ERR(lock))
2127                 cl_lock_mutex_put(env, lock);
2128         RETURN(lock);
2129 }
2130 EXPORT_SYMBOL(cl_lock_hold);
2131
2132 /**
2133  * Main high-level entry point of cl_lock interface that finds existing or
2134  * enqueues new lock matching given description.
2135  */
2136 struct cl_lock *cl_lock_request(const struct lu_env *env, struct cl_io *io,
2137                                 const struct cl_lock_descr *need,
2138                                 const char *scope, const void *source)
2139 {
2140         struct cl_lock       *lock;
2141         int                   rc;
2142         __u32                 enqflags = need->cld_enq_flags;
2143
2144         ENTRY;
2145         do {
2146                 lock = cl_lock_hold_mutex(env, io, need, scope, source);
2147                 if (IS_ERR(lock))
2148                         break;
2149
2150                 rc = cl_enqueue_locked(env, lock, io, enqflags);
2151                 if (rc == 0) {
2152                         if (cl_lock_fits_into(env, lock, need, io)) {
2153                                 if (!(enqflags & CEF_AGL)) {
2154                                         cl_lock_mutex_put(env, lock);
2155                                         cl_lock_lockdep_acquire(env, lock,
2156                                                                 enqflags);
2157                                         break;
2158                                 }
2159                                 rc = 1;
2160                         }
2161                         cl_unuse_locked(env, lock);
2162                 }
2163                 cl_lock_trace(D_DLMTRACE, env,
2164                               rc <= 0 ? "enqueue failed" : "agl succeed", lock);
2165                 cl_lock_hold_release(env, lock, scope, source);
2166                 cl_lock_mutex_put(env, lock);
2167                 lu_ref_del(&lock->cll_reference, scope, source);
2168                 cl_lock_put(env, lock);
2169                 if (rc > 0) {
2170                         LASSERT(enqflags & CEF_AGL);
2171                         lock = NULL;
2172                 } else if (rc != 0) {
2173                         lock = ERR_PTR(rc);
2174                 }
2175         } while (rc == 0);
2176         RETURN(lock);
2177 }
2178 EXPORT_SYMBOL(cl_lock_request);
2179
2180 /**
2181  * Adds a hold to a known lock.
2182  */
2183 void cl_lock_hold_add(const struct lu_env *env, struct cl_lock *lock,
2184                       const char *scope, const void *source)
2185 {
2186         LINVRNT(cl_lock_is_mutexed(lock));
2187         LINVRNT(cl_lock_invariant(env, lock));
2188         LASSERT(lock->cll_state != CLS_FREEING);
2189
2190         ENTRY;
2191         cl_lock_hold_mod(env, lock, +1);
2192         cl_lock_get(lock);
2193         lu_ref_add(&lock->cll_holders, scope, source);
2194         lu_ref_add(&lock->cll_reference, scope, source);
2195         EXIT;
2196 }
2197 EXPORT_SYMBOL(cl_lock_hold_add);
2198
2199 /**
2200  * Releases a hold and a reference on a lock, on which caller acquired a
2201  * mutex.
2202  */
2203 void cl_lock_unhold(const struct lu_env *env, struct cl_lock *lock,
2204                     const char *scope, const void *source)
2205 {
2206         LINVRNT(cl_lock_invariant(env, lock));
2207         ENTRY;
2208         cl_lock_hold_release(env, lock, scope, source);
2209         lu_ref_del(&lock->cll_reference, scope, source);
2210         cl_lock_put(env, lock);
2211         EXIT;
2212 }
2213 EXPORT_SYMBOL(cl_lock_unhold);
2214
2215 /**
2216  * Releases a hold and a reference on a lock, obtained by cl_lock_hold().
2217  */
2218 void cl_lock_release(const struct lu_env *env, struct cl_lock *lock,
2219                      const char *scope, const void *source)
2220 {
2221         LINVRNT(cl_lock_invariant(env, lock));
2222         ENTRY;
2223         cl_lock_trace(D_DLMTRACE, env, "release lock", lock);
2224         cl_lock_mutex_get(env, lock);
2225         cl_lock_hold_release(env, lock, scope, source);
2226         cl_lock_mutex_put(env, lock);
2227         lu_ref_del(&lock->cll_reference, scope, source);
2228         cl_lock_put(env, lock);
2229         EXIT;
2230 }
2231 EXPORT_SYMBOL(cl_lock_release);
2232
2233 void cl_lock_user_add(const struct lu_env *env, struct cl_lock *lock)
2234 {
2235         LINVRNT(cl_lock_is_mutexed(lock));
2236         LINVRNT(cl_lock_invariant(env, lock));
2237
2238         ENTRY;
2239         cl_lock_used_mod(env, lock, +1);
2240         EXIT;
2241 }
2242 EXPORT_SYMBOL(cl_lock_user_add);
2243
2244 void cl_lock_user_del(const struct lu_env *env, struct cl_lock *lock)
2245 {
2246         LINVRNT(cl_lock_is_mutexed(lock));
2247         LINVRNT(cl_lock_invariant(env, lock));
2248         LASSERT(lock->cll_users > 0);
2249
2250         ENTRY;
2251         cl_lock_used_mod(env, lock, -1);
2252         if (lock->cll_users == 0)
2253                 cfs_waitq_broadcast(&lock->cll_wq);
2254         EXIT;
2255 }
2256 EXPORT_SYMBOL(cl_lock_user_del);
2257
2258 const char *cl_lock_mode_name(const enum cl_lock_mode mode)
2259 {
2260         static const char *names[] = {
2261                 [CLM_PHANTOM] = "P",
2262                 [CLM_READ]    = "R",
2263                 [CLM_WRITE]   = "W",
2264                 [CLM_GROUP]   = "G"
2265         };
2266         if (0 <= mode && mode < ARRAY_SIZE(names))
2267                 return names[mode];
2268         else
2269                 return "U";
2270 }
2271 EXPORT_SYMBOL(cl_lock_mode_name);
2272
2273 /**
2274  * Prints human readable representation of a lock description.
2275  */
2276 void cl_lock_descr_print(const struct lu_env *env, void *cookie,
2277                        lu_printer_t printer,
2278                        const struct cl_lock_descr *descr)
2279 {
2280         const struct lu_fid  *fid;
2281
2282         fid = lu_object_fid(&descr->cld_obj->co_lu);
2283         (*printer)(env, cookie, DDESCR"@"DFID, PDESCR(descr), PFID(fid));
2284 }
2285 EXPORT_SYMBOL(cl_lock_descr_print);
2286
2287 /**
2288  * Prints human readable representation of \a lock to the \a f.
2289  */
2290 void cl_lock_print(const struct lu_env *env, void *cookie,
2291                    lu_printer_t printer, const struct cl_lock *lock)
2292 {
2293         const struct cl_lock_slice *slice;
2294         (*printer)(env, cookie, "lock@%p[%d %d %d %d %d %08lx] ",
2295                    lock, cfs_atomic_read(&lock->cll_ref),
2296                    lock->cll_state, lock->cll_error, lock->cll_holds,
2297                    lock->cll_users, lock->cll_flags);
2298         cl_lock_descr_print(env, cookie, printer, &lock->cll_descr);
2299         (*printer)(env, cookie, " {\n");
2300
2301         cfs_list_for_each_entry(slice, &lock->cll_layers, cls_linkage) {
2302                 (*printer)(env, cookie, "    %s@%p: ",
2303                            slice->cls_obj->co_lu.lo_dev->ld_type->ldt_name,
2304                            slice);
2305                 if (slice->cls_ops->clo_print != NULL)
2306                         slice->cls_ops->clo_print(env, cookie, printer, slice);
2307                 (*printer)(env, cookie, "\n");
2308         }
2309         (*printer)(env, cookie, "} lock@%p\n", lock);
2310 }
2311 EXPORT_SYMBOL(cl_lock_print);
2312
2313 int cl_lock_init(void)
2314 {
2315         return lu_kmem_init(cl_lock_caches);
2316 }
2317
2318 void cl_lock_fini(void)
2319 {
2320         lu_kmem_fini(cl_lock_caches);
2321 }