caps.c 117.4 KB
Newer Older
1
// SPDX-License-Identifier: GPL-2.0
2
#include <linux/ceph/ceph_debug.h>
S
Sage Weil 已提交
3 4 5

#include <linux/fs.h>
#include <linux/kernel.h>
6
#include <linux/sched/signal.h>
7
#include <linux/slab.h>
S
Sage Weil 已提交
8 9
#include <linux/vmalloc.h>
#include <linux/wait.h>
10
#include <linux/writeback.h>
11
#include <linux/iversion.h>
S
Sage Weil 已提交
12 13

#include "super.h"
14
#include "mds_client.h"
15
#include "cache.h"
16 17
#include <linux/ceph/decode.h>
#include <linux/ceph/messenger.h>
S
Sage Weil 已提交
18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44

/*
 * Capability management
 *
 * The Ceph metadata servers control client access to inode metadata
 * and file data by issuing capabilities, granting clients permission
 * to read and/or write both inode field and file data to OSDs
 * (storage nodes).  Each capability consists of a set of bits
 * indicating which operations are allowed.
 *
 * If the client holds a *_SHARED cap, the client has a coherent value
 * that can be safely read from the cached inode.
 *
 * In the case of a *_EXCL (exclusive) or FILE_WR capabilities, the
 * client is allowed to change inode attributes (e.g., file size,
 * mtime), note its dirty state in the ceph_cap, and asynchronously
 * flush that metadata change to the MDS.
 *
 * In the event of a conflicting operation (perhaps by another
 * client), the MDS will revoke the conflicting client capabilities.
 *
 * In order for a client to cache an inode, it must hold a capability
 * with at least one MDS server.  When inodes are released, release
 * notifications are batched and periodically sent en masse to the MDS
 * cluster to release server state.
 */

45
static u64 __get_oldest_flush_tid(struct ceph_mds_client *mdsc);
46 47 48 49
static void __kick_flushing_caps(struct ceph_mds_client *mdsc,
				 struct ceph_mds_session *session,
				 struct ceph_inode_info *ci,
				 u64 oldest_flush_tid);
S
Sage Weil 已提交
50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72

/*
 * Generate readable cap strings for debugging output.
 */
#define MAX_CAP_STR 20
static char cap_str[MAX_CAP_STR][40];
static DEFINE_SPINLOCK(cap_str_lock);
static int last_cap_str;

static char *gcap_string(char *s, int c)
{
	if (c & CEPH_CAP_GSHARED)
		*s++ = 's';
	if (c & CEPH_CAP_GEXCL)
		*s++ = 'x';
	if (c & CEPH_CAP_GCACHE)
		*s++ = 'c';
	if (c & CEPH_CAP_GRD)
		*s++ = 'r';
	if (c & CEPH_CAP_GWR)
		*s++ = 'w';
	if (c & CEPH_CAP_GBUFFER)
		*s++ = 'b';
73 74
	if (c & CEPH_CAP_GWREXTEND)
		*s++ = 'a';
S
Sage Weil 已提交
75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126
	if (c & CEPH_CAP_GLAZYIO)
		*s++ = 'l';
	return s;
}

const char *ceph_cap_string(int caps)
{
	int i;
	char *s;
	int c;

	spin_lock(&cap_str_lock);
	i = last_cap_str++;
	if (last_cap_str == MAX_CAP_STR)
		last_cap_str = 0;
	spin_unlock(&cap_str_lock);

	s = cap_str[i];

	if (caps & CEPH_CAP_PIN)
		*s++ = 'p';

	c = (caps >> CEPH_CAP_SAUTH) & 3;
	if (c) {
		*s++ = 'A';
		s = gcap_string(s, c);
	}

	c = (caps >> CEPH_CAP_SLINK) & 3;
	if (c) {
		*s++ = 'L';
		s = gcap_string(s, c);
	}

	c = (caps >> CEPH_CAP_SXATTR) & 3;
	if (c) {
		*s++ = 'X';
		s = gcap_string(s, c);
	}

	c = caps >> CEPH_CAP_SFILE;
	if (c) {
		*s++ = 'F';
		s = gcap_string(s, c);
	}

	if (s == cap_str[i])
		*s++ = '-';
	*s = 0;
	return cap_str[i];
}

127
void ceph_caps_init(struct ceph_mds_client *mdsc)
S
Sage Weil 已提交
128
{
129 130
	INIT_LIST_HEAD(&mdsc->caps_list);
	spin_lock_init(&mdsc->caps_list_lock);
S
Sage Weil 已提交
131 132
}

133
void ceph_caps_finalize(struct ceph_mds_client *mdsc)
S
Sage Weil 已提交
134 135 136
{
	struct ceph_cap *cap;

137 138 139 140
	spin_lock(&mdsc->caps_list_lock);
	while (!list_empty(&mdsc->caps_list)) {
		cap = list_first_entry(&mdsc->caps_list,
				       struct ceph_cap, caps_item);
S
Sage Weil 已提交
141 142 143
		list_del(&cap->caps_item);
		kmem_cache_free(ceph_cap_cachep, cap);
	}
144 145 146 147 148 149
	mdsc->caps_total_count = 0;
	mdsc->caps_avail_count = 0;
	mdsc->caps_use_count = 0;
	mdsc->caps_reserve_count = 0;
	mdsc->caps_min_count = 0;
	spin_unlock(&mdsc->caps_list_lock);
150 151
}

152 153
void ceph_adjust_caps_max_min(struct ceph_mds_client *mdsc,
			      struct ceph_mount_options *fsopt)
154
{
155
	spin_lock(&mdsc->caps_list_lock);
156 157 158 159 160 161 162
	mdsc->caps_min_count = fsopt->max_readdir;
	if (mdsc->caps_min_count < 1024)
		mdsc->caps_min_count = 1024;
	mdsc->caps_use_max = fsopt->caps_max;
	if (mdsc->caps_use_max > 0 &&
	    mdsc->caps_use_max < mdsc->caps_min_count)
		mdsc->caps_use_max = mdsc->caps_min_count;
163
	spin_unlock(&mdsc->caps_list_lock);
S
Sage Weil 已提交
164 165
}

166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 185 186 187 188 189 190 191 192 193 194 195 196
static void __ceph_unreserve_caps(struct ceph_mds_client *mdsc, int nr_caps)
{
	struct ceph_cap *cap;
	int i;

	if (nr_caps) {
		BUG_ON(mdsc->caps_reserve_count < nr_caps);
		mdsc->caps_reserve_count -= nr_caps;
		if (mdsc->caps_avail_count >=
		    mdsc->caps_reserve_count + mdsc->caps_min_count) {
			mdsc->caps_total_count -= nr_caps;
			for (i = 0; i < nr_caps; i++) {
				cap = list_first_entry(&mdsc->caps_list,
					struct ceph_cap, caps_item);
				list_del(&cap->caps_item);
				kmem_cache_free(ceph_cap_cachep, cap);
			}
		} else {
			mdsc->caps_avail_count += nr_caps;
		}

		dout("%s: caps %d = %d used + %d resv + %d avail\n",
		     __func__,
		     mdsc->caps_total_count, mdsc->caps_use_count,
		     mdsc->caps_reserve_count, mdsc->caps_avail_count);
		BUG_ON(mdsc->caps_total_count != mdsc->caps_use_count +
						 mdsc->caps_reserve_count +
						 mdsc->caps_avail_count);
	}
}

197 198 199 200
/*
 * Called under mdsc->mutex.
 */
int ceph_reserve_caps(struct ceph_mds_client *mdsc,
201
		      struct ceph_cap_reservation *ctx, int need)
S
Sage Weil 已提交
202
{
203
	int i, j;
S
Sage Weil 已提交
204 205 206
	struct ceph_cap *cap;
	int have;
	int alloc = 0;
207
	int max_caps;
208
	int err = 0;
209 210
	bool trimmed = false;
	struct ceph_mds_session *s;
S
Sage Weil 已提交
211 212 213 214 215
	LIST_HEAD(newcaps);

	dout("reserve caps ctx=%p need=%d\n", ctx, need);

	/* first reserve any caps that are already allocated */
216 217
	spin_lock(&mdsc->caps_list_lock);
	if (mdsc->caps_avail_count >= need)
S
Sage Weil 已提交
218 219
		have = need;
	else
220 221 222 223 224 225 226
		have = mdsc->caps_avail_count;
	mdsc->caps_avail_count -= have;
	mdsc->caps_reserve_count += have;
	BUG_ON(mdsc->caps_total_count != mdsc->caps_use_count +
					 mdsc->caps_reserve_count +
					 mdsc->caps_avail_count);
	spin_unlock(&mdsc->caps_list_lock);
S
Sage Weil 已提交
227

C
Chengguang Xu 已提交
228
	for (i = have; i < need; ) {
S
Sage Weil 已提交
229
		cap = kmem_cache_alloc(ceph_cap_cachep, GFP_NOFS);
C
Chengguang Xu 已提交
230 231 232 233 234 235 236 237 238 239 240 241 242 243 244 245 246 247 248 249 250
		if (cap) {
			list_add(&cap->caps_item, &newcaps);
			alloc++;
			i++;
			continue;
		}

		if (!trimmed) {
			for (j = 0; j < mdsc->max_sessions; j++) {
				s = __ceph_lookup_mds_session(mdsc, j);
				if (!s)
					continue;
				mutex_unlock(&mdsc->mutex);

				mutex_lock(&s->s_mutex);
				max_caps = s->s_nr_caps - (need - i);
				ceph_trim_caps(mdsc, s, max_caps);
				mutex_unlock(&s->s_mutex);

				ceph_put_mds_session(s);
				mutex_lock(&mdsc->mutex);
251
			}
C
Chengguang Xu 已提交
252 253 254 255 256 257 258 259 260 261 262 263 264 265 266 267 268 269 270
			trimmed = true;

			spin_lock(&mdsc->caps_list_lock);
			if (mdsc->caps_avail_count) {
				int more_have;
				if (mdsc->caps_avail_count >= need - i)
					more_have = need - i;
				else
					more_have = mdsc->caps_avail_count;

				i += more_have;
				have += more_have;
				mdsc->caps_avail_count -= more_have;
				mdsc->caps_reserve_count += more_have;

			}
			spin_unlock(&mdsc->caps_list_lock);

			continue;
271
		}
C
Chengguang Xu 已提交
272 273 274

		pr_warn("reserve caps ctx=%p ENOMEM need=%d got=%d\n",
			ctx, need, have + alloc);
275 276 277 278 279 280 281
		err = -ENOMEM;
		break;
	}

	if (!err) {
		BUG_ON(have + alloc != need);
		ctx->count = need;
282
		ctx->used = 0;
S
Sage Weil 已提交
283 284
	}

285 286 287 288
	spin_lock(&mdsc->caps_list_lock);
	mdsc->caps_total_count += alloc;
	mdsc->caps_reserve_count += alloc;
	list_splice(&newcaps, &mdsc->caps_list);
S
Sage Weil 已提交
289

290 291 292
	BUG_ON(mdsc->caps_total_count != mdsc->caps_use_count +
					 mdsc->caps_reserve_count +
					 mdsc->caps_avail_count);
293 294 295 296

	if (err)
		__ceph_unreserve_caps(mdsc, have + alloc);

297
	spin_unlock(&mdsc->caps_list_lock);
S
Sage Weil 已提交
298 299

	dout("reserve caps ctx=%p %d = %d used + %d resv + %d avail\n",
300 301
	     ctx, mdsc->caps_total_count, mdsc->caps_use_count,
	     mdsc->caps_reserve_count, mdsc->caps_avail_count);
302
	return err;
S
Sage Weil 已提交
303 304
}

305
void ceph_unreserve_caps(struct ceph_mds_client *mdsc,
306
			 struct ceph_cap_reservation *ctx)
S
Sage Weil 已提交
307
{
308 309 310 311
	bool reclaim = false;
	if (!ctx->count)
		return;

S
Sage Weil 已提交
312
	dout("unreserve caps ctx=%p count=%d\n", ctx, ctx->count);
313 314 315
	spin_lock(&mdsc->caps_list_lock);
	__ceph_unreserve_caps(mdsc, ctx->count);
	ctx->count = 0;
316 317 318 319

	if (mdsc->caps_use_max > 0 &&
	    mdsc->caps_use_count > mdsc->caps_use_max)
		reclaim = true;
320
	spin_unlock(&mdsc->caps_list_lock);
321 322 323

	if (reclaim)
		ceph_reclaim_caps_nr(mdsc, ctx->used);
S
Sage Weil 已提交
324 325
}

326 327
struct ceph_cap *ceph_get_cap(struct ceph_mds_client *mdsc,
			      struct ceph_cap_reservation *ctx)
S
Sage Weil 已提交
328 329 330 331
{
	struct ceph_cap *cap = NULL;

	/* temporary, until we do something about cap import/export */
332 333 334
	if (!ctx) {
		cap = kmem_cache_alloc(ceph_cap_cachep, GFP_NOFS);
		if (cap) {
335
			spin_lock(&mdsc->caps_list_lock);
336 337
			mdsc->caps_use_count++;
			mdsc->caps_total_count++;
338
			spin_unlock(&mdsc->caps_list_lock);
C
Chengguang Xu 已提交
339 340 341 342 343 344 345 346 347 348 349 350 351 352 353
		} else {
			spin_lock(&mdsc->caps_list_lock);
			if (mdsc->caps_avail_count) {
				BUG_ON(list_empty(&mdsc->caps_list));

				mdsc->caps_avail_count--;
				mdsc->caps_use_count++;
				cap = list_first_entry(&mdsc->caps_list,
						struct ceph_cap, caps_item);
				list_del(&cap->caps_item);

				BUG_ON(mdsc->caps_total_count != mdsc->caps_use_count +
				       mdsc->caps_reserve_count + mdsc->caps_avail_count);
			}
			spin_unlock(&mdsc->caps_list_lock);
354
		}
C
Chengguang Xu 已提交
355

356 357
		return cap;
	}
S
Sage Weil 已提交
358

359
	spin_lock(&mdsc->caps_list_lock);
S
Sage Weil 已提交
360
	dout("get_cap ctx=%p (%d) %d = %d used + %d resv + %d avail\n",
361 362
	     ctx, ctx->count, mdsc->caps_total_count, mdsc->caps_use_count,
	     mdsc->caps_reserve_count, mdsc->caps_avail_count);
S
Sage Weil 已提交
363
	BUG_ON(!ctx->count);
364 365
	BUG_ON(ctx->count > mdsc->caps_reserve_count);
	BUG_ON(list_empty(&mdsc->caps_list));
S
Sage Weil 已提交
366 367

	ctx->count--;
368
	ctx->used++;
369 370
	mdsc->caps_reserve_count--;
	mdsc->caps_use_count++;
S
Sage Weil 已提交
371

372
	cap = list_first_entry(&mdsc->caps_list, struct ceph_cap, caps_item);
S
Sage Weil 已提交
373 374
	list_del(&cap->caps_item);

375 376 377
	BUG_ON(mdsc->caps_total_count != mdsc->caps_use_count +
	       mdsc->caps_reserve_count + mdsc->caps_avail_count);
	spin_unlock(&mdsc->caps_list_lock);
S
Sage Weil 已提交
378 379 380
	return cap;
}

381
void ceph_put_cap(struct ceph_mds_client *mdsc, struct ceph_cap *cap)
S
Sage Weil 已提交
382
{
383
	spin_lock(&mdsc->caps_list_lock);
S
Sage Weil 已提交
384
	dout("put_cap %p %d = %d used + %d resv + %d avail\n",
385 386 387
	     cap, mdsc->caps_total_count, mdsc->caps_use_count,
	     mdsc->caps_reserve_count, mdsc->caps_avail_count);
	mdsc->caps_use_count--;
S
Sage Weil 已提交
388
	/*
389 390
	 * Keep some preallocated caps around (ceph_min_count), to
	 * avoid lots of free/alloc churn.
S
Sage Weil 已提交
391
	 */
392 393 394
	if (mdsc->caps_avail_count >= mdsc->caps_reserve_count +
				      mdsc->caps_min_count) {
		mdsc->caps_total_count--;
S
Sage Weil 已提交
395 396
		kmem_cache_free(ceph_cap_cachep, cap);
	} else {
397 398
		mdsc->caps_avail_count++;
		list_add(&cap->caps_item, &mdsc->caps_list);
S
Sage Weil 已提交
399 400
	}

401 402 403
	BUG_ON(mdsc->caps_total_count != mdsc->caps_use_count +
	       mdsc->caps_reserve_count + mdsc->caps_avail_count);
	spin_unlock(&mdsc->caps_list_lock);
S
Sage Weil 已提交
404 405
}

406
void ceph_reservation_status(struct ceph_fs_client *fsc,
407 408
			     int *total, int *avail, int *used, int *reserved,
			     int *min)
S
Sage Weil 已提交
409
{
410
	struct ceph_mds_client *mdsc = fsc->mdsc;
411

412 413
	spin_lock(&mdsc->caps_list_lock);

S
Sage Weil 已提交
414
	if (total)
415
		*total = mdsc->caps_total_count;
S
Sage Weil 已提交
416
	if (avail)
417
		*avail = mdsc->caps_avail_count;
S
Sage Weil 已提交
418
	if (used)
419
		*used = mdsc->caps_use_count;
S
Sage Weil 已提交
420
	if (reserved)
421
		*reserved = mdsc->caps_reserve_count;
422
	if (min)
423
		*min = mdsc->caps_min_count;
424 425

	spin_unlock(&mdsc->caps_list_lock);
S
Sage Weil 已提交
426 427 428 429 430
}

/*
 * Find ceph_cap for given mds, if any.
 *
431
 * Called with i_ceph_lock held.
S
Sage Weil 已提交
432 433 434 435 436 437 438 439 440 441 442 443 444 445 446 447 448 449
 */
static struct ceph_cap *__get_cap_for_mds(struct ceph_inode_info *ci, int mds)
{
	struct ceph_cap *cap;
	struct rb_node *n = ci->i_caps.rb_node;

	while (n) {
		cap = rb_entry(n, struct ceph_cap, ci_node);
		if (mds < cap->mds)
			n = n->rb_left;
		else if (mds > cap->mds)
			n = n->rb_right;
		else
			return cap;
	}
	return NULL;
}

450 451 452 453
struct ceph_cap *ceph_get_cap_for_mds(struct ceph_inode_info *ci, int mds)
{
	struct ceph_cap *cap;

454
	spin_lock(&ci->i_ceph_lock);
455
	cap = __get_cap_for_mds(ci, mds);
456
	spin_unlock(&ci->i_ceph_lock);
457 458 459
	return cap;
}

S
Sage Weil 已提交
460
/*
461
 * Called under i_ceph_lock.
S
Sage Weil 已提交
462 463 464 465 466 467 468 469 470 471 472 473 474 475 476 477 478 479 480 481 482 483 484 485 486 487 488 489 490 491
 */
static void __insert_cap_node(struct ceph_inode_info *ci,
			      struct ceph_cap *new)
{
	struct rb_node **p = &ci->i_caps.rb_node;
	struct rb_node *parent = NULL;
	struct ceph_cap *cap = NULL;

	while (*p) {
		parent = *p;
		cap = rb_entry(parent, struct ceph_cap, ci_node);
		if (new->mds < cap->mds)
			p = &(*p)->rb_left;
		else if (new->mds > cap->mds)
			p = &(*p)->rb_right;
		else
			BUG();
	}

	rb_link_node(&new->ci_node, parent, p);
	rb_insert_color(&new->ci_node, &ci->i_caps);
}

/*
 * (re)set cap hold timeouts, which control the delayed release
 * of unused caps back to the MDS.  Should be called on cap use.
 */
static void __cap_set_timeouts(struct ceph_mds_client *mdsc,
			       struct ceph_inode_info *ci)
{
492
	struct ceph_mount_options *opt = mdsc->fsc->mount_options;
S
Sage Weil 已提交
493
	ci->i_hold_caps_max = round_jiffies(jiffies +
494
					    opt->caps_wanted_delay_max * HZ);
495 496
	dout("__cap_set_timeouts %p %lu\n", &ci->vfs_inode,
	     ci->i_hold_caps_max - jiffies);
S
Sage Weil 已提交
497 498 499 500 501 502 503
}

/*
 * (Re)queue cap at the end of the delayed cap release list.
 *
 * If I_FLUSH is set, leave the inode at the front of the list.
 *
504
 * Caller holds i_ceph_lock
S
Sage Weil 已提交
505 506 507
 *    -> we take mdsc->cap_delay_lock
 */
static void __cap_delay_requeue(struct ceph_mds_client *mdsc,
508
				struct ceph_inode_info *ci)
S
Sage Weil 已提交
509
{
510
	dout("__cap_delay_requeue %p flags 0x%lx at %lu\n", &ci->vfs_inode,
S
Sage Weil 已提交
511 512 513 514 515 516 517 518
	     ci->i_ceph_flags, ci->i_hold_caps_max);
	if (!mdsc->stopping) {
		spin_lock(&mdsc->cap_delay_lock);
		if (!list_empty(&ci->i_cap_delay_list)) {
			if (ci->i_ceph_flags & CEPH_I_FLUSH)
				goto no_change;
			list_del_init(&ci->i_cap_delay_list);
		}
519
		__cap_set_timeouts(mdsc, ci);
S
Sage Weil 已提交
520 521 522 523 524 525 526 527 528 529 530 531 532 533 534 535 536 537 538 539 540 541 542 543 544 545
		list_add_tail(&ci->i_cap_delay_list, &mdsc->cap_delay_list);
no_change:
		spin_unlock(&mdsc->cap_delay_lock);
	}
}

/*
 * Queue an inode for immediate writeback.  Mark inode with I_FLUSH,
 * indicating we should send a cap message to flush dirty metadata
 * asap, and move to the front of the delayed cap list.
 */
static void __cap_delay_requeue_front(struct ceph_mds_client *mdsc,
				      struct ceph_inode_info *ci)
{
	dout("__cap_delay_requeue_front %p\n", &ci->vfs_inode);
	spin_lock(&mdsc->cap_delay_lock);
	ci->i_ceph_flags |= CEPH_I_FLUSH;
	if (!list_empty(&ci->i_cap_delay_list))
		list_del_init(&ci->i_cap_delay_list);
	list_add(&ci->i_cap_delay_list, &mdsc->cap_delay_list);
	spin_unlock(&mdsc->cap_delay_lock);
}

/*
 * Cancel delayed work on cap.
 *
546
 * Caller must hold i_ceph_lock.
S
Sage Weil 已提交
547 548 549 550 551 552 553 554 555 556 557 558
 */
static void __cap_delay_cancel(struct ceph_mds_client *mdsc,
			       struct ceph_inode_info *ci)
{
	dout("__cap_delay_cancel %p\n", &ci->vfs_inode);
	if (list_empty(&ci->i_cap_delay_list))
		return;
	spin_lock(&mdsc->cap_delay_lock);
	list_del_init(&ci->i_cap_delay_list);
	spin_unlock(&mdsc->cap_delay_lock);
}

559
/* Common issue checks for add_cap, handle_cap_grant. */
S
Sage Weil 已提交
560 561 562 563 564
static void __check_cap_issue(struct ceph_inode_info *ci, struct ceph_cap *cap,
			      unsigned issued)
{
	unsigned had = __ceph_caps_issued(ci, NULL);

565 566
	lockdep_assert_held(&ci->i_ceph_lock);

S
Sage Weil 已提交
567 568 569 570
	/*
	 * Each time we receive FILE_CACHE anew, we increment
	 * i_rdcache_gen.
	 */
571 572
	if (S_ISREG(ci->vfs_inode.i_mode) &&
	    (issued & (CEPH_CAP_FILE_CACHE|CEPH_CAP_FILE_LAZYIO)) &&
573
	    (had & (CEPH_CAP_FILE_CACHE|CEPH_CAP_FILE_LAZYIO)) == 0) {
S
Sage Weil 已提交
574
		ci->i_rdcache_gen++;
575
	}
S
Sage Weil 已提交
576 577

	/*
578 579 580 581
	 * If FILE_SHARED is newly issued, mark dir not complete. We don't
	 * know what happened to this directory while we didn't have the cap.
	 * If FILE_SHARED is being revoked, also mark dir not complete. It
	 * stops on-going cached readdir.
S
Sage Weil 已提交
582
	 */
583 584
	if ((issued & CEPH_CAP_FILE_SHARED) != (had & CEPH_CAP_FILE_SHARED)) {
		if (issued & CEPH_CAP_FILE_SHARED)
585
			atomic_inc(&ci->i_shared_gen);
586 587
		if (S_ISDIR(ci->vfs_inode.i_mode)) {
			dout(" marking %p NOT complete\n", &ci->vfs_inode);
588
			__ceph_dir_clear_complete(ci);
589
		}
S
Sage Weil 已提交
590
	}
591 592 593 594 595 596 597

	/* Wipe saved layout if we're losing DIR_CREATE caps */
	if (S_ISDIR(ci->vfs_inode.i_mode) && (had & CEPH_CAP_DIR_CREATE) &&
		!(issued & CEPH_CAP_DIR_CREATE)) {
	     ceph_put_string(rcu_dereference_raw(ci->i_cached_layout.pool_ns));
	     memset(&ci->i_cached_layout, 0, sizeof(ci->i_cached_layout));
	}
S
Sage Weil 已提交
598 599 600 601 602
}

/*
 * Add a capability under the given MDS session.
 *
603
 * Caller should hold session snap_rwsem (read) and ci->i_ceph_lock
S
Sage Weil 已提交
604 605 606 607 608
 *
 * @fmode is the open file mode, if we are opening a file, otherwise
 * it is < 0.  (This is so we can atomically add the cap and add an
 * open file reference to it.)
 */
609 610
void ceph_add_cap(struct inode *inode,
		  struct ceph_mds_session *session, u64 cap_id,
611
		  unsigned issued, unsigned wanted,
612 613
		  unsigned seq, unsigned mseq, u64 realmino, int flags,
		  struct ceph_cap **new_cap)
S
Sage Weil 已提交
614
{
615
	struct ceph_mds_client *mdsc = ceph_inode_to_client(inode)->mdsc;
S
Sage Weil 已提交
616 617 618 619
	struct ceph_inode_info *ci = ceph_inode(inode);
	struct ceph_cap *cap;
	int mds = session->s_mds;
	int actual_wanted;
620
	u32 gen;
S
Sage Weil 已提交
621

622 623
	lockdep_assert_held(&ci->i_ceph_lock);

S
Sage Weil 已提交
624 625 626
	dout("add_cap %p mds%d cap %llx %s seq %d\n", inode,
	     session->s_mds, cap_id, ceph_cap_string(issued), seq);

627 628 629 630
	spin_lock(&session->s_gen_ttl_lock);
	gen = session->s_cap_gen;
	spin_unlock(&session->s_gen_ttl_lock);

S
Sage Weil 已提交
631 632
	cap = __get_cap_for_mds(ci, mds);
	if (!cap) {
633 634
		cap = *new_cap;
		*new_cap = NULL;
S
Sage Weil 已提交
635 636 637 638 639

		cap->issued = 0;
		cap->implemented = 0;
		cap->mds = mds;
		cap->mds_wanted = 0;
640
		cap->mseq = 0;
S
Sage Weil 已提交
641 642 643 644 645 646 647 648 649 650

		cap->ci = ci;
		__insert_cap_node(ci, cap);

		/* add to session cap list */
		cap->session = session;
		spin_lock(&session->s_cap_lock);
		list_add_tail(&cap->session_caps, &session->s_caps);
		session->s_nr_caps++;
		spin_unlock(&session->s_cap_lock);
651
	} else {
652 653 654 655
		spin_lock(&session->s_cap_lock);
		list_move_tail(&cap->session_caps, &session->s_caps);
		spin_unlock(&session->s_cap_lock);

656
		if (cap->cap_gen < gen)
657 658
			cap->issued = cap->implemented = CEPH_CAP_PIN;

659 660 661 662 663 664 665 666 667 668 669 670 671 672 673 674 675 676
		/*
		 * auth mds of the inode changed. we received the cap export
		 * message, but still haven't received the cap import message.
		 * handle_cap_export() updated the new auth MDS' cap.
		 *
		 * "ceph_seq_cmp(seq, cap->seq) <= 0" means we are processing
		 * a message that was send before the cap import message. So
		 * don't remove caps.
		 */
		if (ceph_seq_cmp(seq, cap->seq) <= 0) {
			WARN_ON(cap != ci->i_auth_cap);
			WARN_ON(cap->cap_id != cap_id);
			seq = cap->seq;
			mseq = cap->mseq;
			issued |= cap->issued;
			flags |= CEPH_CAP_FLAG_AUTH;
		}
	}
S
Sage Weil 已提交
677

678 679 680
	if (!ci->i_snap_realm ||
	    ((flags & CEPH_CAP_FLAG_AUTH) &&
	     realmino != (u64)-1 && ci->i_snap_realm->ino != realmino)) {
S
Sage Weil 已提交
681 682 683 684 685 686
		/*
		 * add this inode to the appropriate snap realm
		 */
		struct ceph_snap_realm *realm = ceph_lookup_snap_realm(mdsc,
							       realmino);
		if (realm) {
687 688 689 690 691 692 693
			struct ceph_snap_realm *oldrealm = ci->i_snap_realm;
			if (oldrealm) {
				spin_lock(&oldrealm->inodes_with_caps_lock);
				list_del_init(&ci->i_snap_realm_item);
				spin_unlock(&oldrealm->inodes_with_caps_lock);
			}

S
Sage Weil 已提交
694 695 696
			spin_lock(&realm->inodes_with_caps_lock);
			list_add(&ci->i_snap_realm_item,
				 &realm->inodes_with_caps);
697 698 699
			ci->i_snap_realm = realm;
			if (realm->ino == ci->i_vino.ino)
				realm->inode = inode;
S
Sage Weil 已提交
700
			spin_unlock(&realm->inodes_with_caps_lock);
701 702 703

			if (oldrealm)
				ceph_put_snap_realm(mdsc, oldrealm);
S
Sage Weil 已提交
704 705 706
		} else {
			pr_err("ceph_add_cap: couldn't find snap realm %llx\n",
			       realmino);
S
Sage Weil 已提交
707
			WARN_ON(!realm);
S
Sage Weil 已提交
708 709 710 711 712 713 714 715 716 717 718 719 720 721 722 723
		}
	}

	__check_cap_issue(ci, cap, issued);

	/*
	 * If we are issued caps we don't want, or the mds' wanted
	 * value appears to be off, queue a check so we'll release
	 * later and/or update the mds wanted value.
	 */
	actual_wanted = __ceph_caps_wanted(ci);
	if ((wanted & ~actual_wanted) ||
	    (issued & ~actual_wanted & CEPH_CAP_ANY_WR)) {
		dout(" issued %s, mds wanted %s, actual %s, queueing\n",
		     ceph_cap_string(issued), ceph_cap_string(wanted),
		     ceph_cap_string(actual_wanted));
724
		__cap_delay_requeue(mdsc, ci);
S
Sage Weil 已提交
725 726
	}

727
	if (flags & CEPH_CAP_FLAG_AUTH) {
728
		if (!ci->i_auth_cap ||
729
		    ceph_seq_cmp(ci->i_auth_cap->mseq, mseq) < 0) {
730
			ci->i_auth_cap = cap;
731 732
			cap->mds_wanted = wanted;
		}
733 734
	} else {
		WARN_ON(ci->i_auth_cap == cap);
735
	}
S
Sage Weil 已提交
736 737 738 739 740 741 742

	dout("add_cap inode %p (%llx.%llx) cap %p %s now %s seq %d mds%d\n",
	     inode, ceph_vinop(inode), cap, ceph_cap_string(issued),
	     ceph_cap_string(issued|cap->issued), seq, mds);
	cap->cap_id = cap_id;
	cap->issued = issued;
	cap->implemented |= issued;
743
	if (ceph_seq_cmp(mseq, cap->mseq) > 0)
744 745 746
		cap->mds_wanted = wanted;
	else
		cap->mds_wanted |= wanted;
S
Sage Weil 已提交
747 748 749
	cap->seq = seq;
	cap->issue_seq = seq;
	cap->mseq = mseq;
750
	cap->cap_gen = gen;
S
Sage Weil 已提交
751 752 753 754 755 756 757 758 759 760
}

/*
 * Return true if cap has not timed out and belongs to the current
 * generation of the MDS session (i.e. has not gone 'stale' due to
 * us losing touch with the mds).
 */
static int __cap_is_valid(struct ceph_cap *cap)
{
	unsigned long ttl;
S
Sage Weil 已提交
761
	u32 gen;
S
Sage Weil 已提交
762

763
	spin_lock(&cap->session->s_gen_ttl_lock);
S
Sage Weil 已提交
764 765
	gen = cap->session->s_cap_gen;
	ttl = cap->session->s_cap_ttl;
766
	spin_unlock(&cap->session->s_gen_ttl_lock);
S
Sage Weil 已提交
767

768
	if (cap->cap_gen < gen || time_after_eq(jiffies, ttl)) {
S
Sage Weil 已提交
769 770
		dout("__cap_is_valid %p cap %p issued %s "
		     "but STALE (gen %u vs %u)\n", &cap->ci->vfs_inode,
771
		     cap, ceph_cap_string(cap->issued), cap->cap_gen, gen);
S
Sage Weil 已提交
772 773 774 775 776 777 778 779 780 781 782 783 784
		return 0;
	}

	return 1;
}

/*
 * Return set of valid cap bits issued to us.  Note that caps time
 * out, and may be invalidated in bulk if the client session times out
 * and session->s_cap_gen is bumped.
 */
int __ceph_caps_issued(struct ceph_inode_info *ci, int *implemented)
{
785
	int have = ci->i_snap_caps;
S
Sage Weil 已提交
786 787 788 789 790 791 792 793 794 795 796 797 798 799 800
	struct ceph_cap *cap;
	struct rb_node *p;

	if (implemented)
		*implemented = 0;
	for (p = rb_first(&ci->i_caps); p; p = rb_next(p)) {
		cap = rb_entry(p, struct ceph_cap, ci_node);
		if (!__cap_is_valid(cap))
			continue;
		dout("__ceph_caps_issued %p cap %p issued %s\n",
		     &ci->vfs_inode, cap, ceph_cap_string(cap->issued));
		have |= cap->issued;
		if (implemented)
			*implemented |= cap->implemented;
	}
Y
Yan, Zheng 已提交
801 802 803 804 805 806 807 808 809
	/*
	 * exclude caps issued by non-auth MDS, but are been revoking
	 * by the auth MDS. The non-auth MDS should be revoking/exporting
	 * these caps, but the message is delayed.
	 */
	if (ci->i_auth_cap) {
		cap = ci->i_auth_cap;
		have &= ~cap->implemented | cap->issued;
	}
S
Sage Weil 已提交
810 811 812 813 814 815 816 817 818 819 820 821 822 823 824 825 826 827 828 829 830 831 832 833 834 835 836 837 838 839 840 841
	return have;
}

/*
 * Get cap bits issued by caps other than @ocap
 */
int __ceph_caps_issued_other(struct ceph_inode_info *ci, struct ceph_cap *ocap)
{
	int have = ci->i_snap_caps;
	struct ceph_cap *cap;
	struct rb_node *p;

	for (p = rb_first(&ci->i_caps); p; p = rb_next(p)) {
		cap = rb_entry(p, struct ceph_cap, ci_node);
		if (cap == ocap)
			continue;
		if (!__cap_is_valid(cap))
			continue;
		have |= cap->issued;
	}
	return have;
}

/*
 * Move a cap to the end of the LRU (oldest caps at list head, newest
 * at list tail).
 */
static void __touch_cap(struct ceph_cap *cap)
{
	struct ceph_mds_session *s = cap->session;

	spin_lock(&s->s_cap_lock);
842
	if (!s->s_cap_iterator) {
843 844 845 846 847 848 849
		dout("__touch_cap %p cap %p mds%d\n", &cap->ci->vfs_inode, cap,
		     s->s_mds);
		list_move_tail(&cap->session_caps, &s->s_caps);
	} else {
		dout("__touch_cap %p cap %p mds%d NOP, iterating over caps\n",
		     &cap->ci->vfs_inode, cap, s->s_mds);
	}
S
Sage Weil 已提交
850 851 852 853 854 855 856 857 858 859 860 861 862 863 864
	spin_unlock(&s->s_cap_lock);
}

/*
 * Check if we hold the given mask.  If so, move the cap(s) to the
 * front of their respective LRUs.  (This is the preferred way for
 * callers to check for caps they want.)
 */
int __ceph_caps_issued_mask(struct ceph_inode_info *ci, int mask, int touch)
{
	struct ceph_cap *cap;
	struct rb_node *p;
	int have = ci->i_snap_caps;

	if ((have & mask) == mask) {
865 866
		dout("__ceph_caps_issued_mask ino 0x%lx snap issued %s"
		     " (mask %s)\n", ci->vfs_inode.i_ino,
S
Sage Weil 已提交
867 868 869 870 871 872 873 874 875 876
		     ceph_cap_string(have),
		     ceph_cap_string(mask));
		return 1;
	}

	for (p = rb_first(&ci->i_caps); p; p = rb_next(p)) {
		cap = rb_entry(p, struct ceph_cap, ci_node);
		if (!__cap_is_valid(cap))
			continue;
		if ((cap->issued & mask) == mask) {
877 878
			dout("__ceph_caps_issued_mask ino 0x%lx cap %p issued %s"
			     " (mask %s)\n", ci->vfs_inode.i_ino, cap,
S
Sage Weil 已提交
879 880 881 882 883 884 885 886 887 888
			     ceph_cap_string(cap->issued),
			     ceph_cap_string(mask));
			if (touch)
				__touch_cap(cap);
			return 1;
		}

		/* does a combination of caps satisfy mask? */
		have |= cap->issued;
		if ((have & mask) == mask) {
889 890
			dout("__ceph_caps_issued_mask ino 0x%lx combo issued %s"
			     " (mask %s)\n", ci->vfs_inode.i_ino,
S
Sage Weil 已提交
891 892 893 894 895
			     ceph_cap_string(cap->issued),
			     ceph_cap_string(mask));
			if (touch) {
				struct rb_node *q;

L
Lucas De Marchi 已提交
896
				/* touch this + preceding caps */
S
Sage Weil 已提交
897 898 899 900 901 902 903
				__touch_cap(cap);
				for (q = rb_first(&ci->i_caps); q != p;
				     q = rb_next(q)) {
					cap = rb_entry(q, struct ceph_cap,
						       ci_node);
					if (!__cap_is_valid(cap))
						continue;
904 905
					if (cap->issued & mask)
						__touch_cap(cap);
S
Sage Weil 已提交
906 907 908 909 910 911 912 913 914
				}
			}
			return 1;
		}
	}

	return 0;
}

915 916 917 918 919 920 921 922 923 924 925 926 927 928
int __ceph_caps_issued_mask_metric(struct ceph_inode_info *ci, int mask,
				   int touch)
{
	struct ceph_fs_client *fsc = ceph_sb_to_client(ci->vfs_inode.i_sb);
	int r;

	r = __ceph_caps_issued_mask(ci, mask, touch);
	if (r)
		ceph_update_cap_hit(&fsc->mdsc->metric);
	else
		ceph_update_cap_mis(&fsc->mdsc->metric);
	return r;
}

S
Sage Weil 已提交
929 930 931
/*
 * Return true if mask caps are currently being revoked by an MDS.
 */
932 933
int __ceph_caps_revoking_other(struct ceph_inode_info *ci,
			       struct ceph_cap *ocap, int mask)
S
Sage Weil 已提交
934 935 936 937 938 939
{
	struct ceph_cap *cap;
	struct rb_node *p;

	for (p = rb_first(&ci->i_caps); p; p = rb_next(p)) {
		cap = rb_entry(p, struct ceph_cap, ci_node);
Y
Yan, Zheng 已提交
940
		if (cap != ocap &&
941 942
		    (cap->implemented & ~cap->issued & mask))
			return 1;
S
Sage Weil 已提交
943
	}
944 945 946 947 948 949 950 951 952 953
	return 0;
}

int ceph_caps_revoking(struct ceph_inode_info *ci, int mask)
{
	struct inode *inode = &ci->vfs_inode;
	int ret;

	spin_lock(&ci->i_ceph_lock);
	ret = __ceph_caps_revoking_other(ci, NULL, mask);
954
	spin_unlock(&ci->i_ceph_lock);
S
Sage Weil 已提交
955 956 957 958 959 960 961 962 963 964 965 966
	dout("ceph_caps_revoking %p %s = %d\n", inode,
	     ceph_cap_string(mask), ret);
	return ret;
}

int __ceph_caps_used(struct ceph_inode_info *ci)
{
	int used = 0;
	if (ci->i_pin_ref)
		used |= CEPH_CAP_PIN;
	if (ci->i_rd_ref)
		used |= CEPH_CAP_FILE_RD;
Y
Yan, Zheng 已提交
967
	if (ci->i_rdcache_ref ||
968
	    (S_ISREG(ci->vfs_inode.i_mode) &&
Y
Yan, Zheng 已提交
969
	     ci->vfs_inode.i_data.nrpages))
S
Sage Weil 已提交
970 971 972
		used |= CEPH_CAP_FILE_CACHE;
	if (ci->i_wr_ref)
		used |= CEPH_CAP_FILE_WR;
973
	if (ci->i_wb_ref || ci->i_wrbuffer_ref)
S
Sage Weil 已提交
974
		used |= CEPH_CAP_FILE_BUFFER;
J
Jeff Layton 已提交
975 976
	if (ci->i_fx_ref)
		used |= CEPH_CAP_FILE_EXCL;
S
Sage Weil 已提交
977 978 979
	return used;
}

980 981
#define FMODE_WAIT_BIAS 1000

S
Sage Weil 已提交
982 983 984 985 986
/*
 * wanted, by virtue of open file modes
 */
int __ceph_caps_file_wanted(struct ceph_inode_info *ci)
{
987 988 989 990 991 992 993 994 995 996 997 998 999 1000 1001 1002 1003 1004 1005 1006 1007 1008 1009 1010 1011 1012 1013 1014 1015 1016 1017 1018 1019 1020 1021 1022 1023 1024 1025 1026 1027 1028 1029 1030 1031 1032 1033 1034 1035 1036 1037 1038 1039
	const int PIN_SHIFT = ffs(CEPH_FILE_MODE_PIN);
	const int RD_SHIFT = ffs(CEPH_FILE_MODE_RD);
	const int WR_SHIFT = ffs(CEPH_FILE_MODE_WR);
	const int LAZY_SHIFT = ffs(CEPH_FILE_MODE_LAZY);
	struct ceph_mount_options *opt =
		ceph_inode_to_client(&ci->vfs_inode)->mount_options;
	unsigned long used_cutoff = jiffies - opt->caps_wanted_delay_max * HZ;
	unsigned long idle_cutoff = jiffies - opt->caps_wanted_delay_min * HZ;

	if (S_ISDIR(ci->vfs_inode.i_mode)) {
		int want = 0;

		/* use used_cutoff here, to keep dir's wanted caps longer */
		if (ci->i_nr_by_mode[RD_SHIFT] > 0 ||
		    time_after(ci->i_last_rd, used_cutoff))
			want |= CEPH_CAP_ANY_SHARED;

		if (ci->i_nr_by_mode[WR_SHIFT] > 0 ||
		    time_after(ci->i_last_wr, used_cutoff)) {
			want |= CEPH_CAP_ANY_SHARED | CEPH_CAP_FILE_EXCL;
			if (opt->flags & CEPH_MOUNT_OPT_ASYNC_DIROPS)
				want |= CEPH_CAP_ANY_DIR_OPS;
		}

		if (want || ci->i_nr_by_mode[PIN_SHIFT] > 0)
			want |= CEPH_CAP_PIN;

		return want;
	} else {
		int bits = 0;

		if (ci->i_nr_by_mode[RD_SHIFT] > 0) {
			if (ci->i_nr_by_mode[RD_SHIFT] >= FMODE_WAIT_BIAS ||
			    time_after(ci->i_last_rd, used_cutoff))
				bits |= 1 << RD_SHIFT;
		} else if (time_after(ci->i_last_rd, idle_cutoff)) {
			bits |= 1 << RD_SHIFT;
		}

		if (ci->i_nr_by_mode[WR_SHIFT] > 0) {
			if (ci->i_nr_by_mode[WR_SHIFT] >= FMODE_WAIT_BIAS ||
			    time_after(ci->i_last_wr, used_cutoff))
				bits |= 1 << WR_SHIFT;
		} else if (time_after(ci->i_last_wr, idle_cutoff)) {
			bits |= 1 << WR_SHIFT;
		}

		/* check lazyio only when read/write is wanted */
		if ((bits & (CEPH_FILE_MODE_RDWR << 1)) &&
		    ci->i_nr_by_mode[LAZY_SHIFT] > 0)
			bits |= 1 << LAZY_SHIFT;

		return bits ? ceph_caps_for_mode(bits >> 1) : 0;
1040
	}
S
Sage Weil 已提交
1041 1042
}

1043 1044 1045 1046 1047 1048
/*
 * wanted, by virtue of open file modes AND cap refs (buffered/cached data)
 */
int __ceph_caps_wanted(struct ceph_inode_info *ci)
{
	int w = __ceph_caps_file_wanted(ci) | __ceph_caps_used(ci);
1049 1050 1051 1052 1053
	if (S_ISDIR(ci->vfs_inode.i_mode)) {
		/* we want EXCL if holding caps of dir ops */
		if (w & CEPH_CAP_ANY_DIR_OPS)
			w |= CEPH_CAP_FILE_EXCL;
	} else {
1054 1055 1056 1057 1058 1059 1060
		/* we want EXCL if dirty data */
		if (w & CEPH_CAP_FILE_BUFFER)
			w |= CEPH_CAP_FILE_EXCL;
	}
	return w;
}

S
Sage Weil 已提交
1061 1062 1063
/*
 * Return caps we have registered with the MDS(s) as 'wanted'.
 */
1064
int __ceph_caps_mds_wanted(struct ceph_inode_info *ci, bool check)
S
Sage Weil 已提交
1065 1066 1067 1068 1069 1070 1071
{
	struct ceph_cap *cap;
	struct rb_node *p;
	int mds_wanted = 0;

	for (p = rb_first(&ci->i_caps); p; p = rb_next(p)) {
		cap = rb_entry(p, struct ceph_cap, ci_node);
1072
		if (check && !__cap_is_valid(cap))
S
Sage Weil 已提交
1073
			continue;
1074 1075 1076 1077
		if (cap == ci->i_auth_cap)
			mds_wanted |= cap->mds_wanted;
		else
			mds_wanted |= (cap->mds_wanted & ~CEPH_CAP_ANY_FILE_WR);
S
Sage Weil 已提交
1078 1079 1080 1081
	}
	return mds_wanted;
}

1082 1083 1084 1085 1086 1087
int ceph_is_any_caps(struct inode *inode)
{
	struct ceph_inode_info *ci = ceph_inode(inode);
	int ret;

	spin_lock(&ci->i_ceph_lock);
X
Xiubo Li 已提交
1088
	ret = __ceph_is_any_real_caps(ci);
1089 1090 1091 1092 1093
	spin_unlock(&ci->i_ceph_lock);

	return ret;
}

1094 1095 1096 1097 1098 1099 1100
static void drop_inode_snap_realm(struct ceph_inode_info *ci)
{
	struct ceph_snap_realm *realm = ci->i_snap_realm;
	spin_lock(&realm->inodes_with_caps_lock);
	list_del_init(&ci->i_snap_realm_item);
	ci->i_snap_realm_counter++;
	ci->i_snap_realm = NULL;
1101 1102
	if (realm->ino == ci->i_vino.ino)
		realm->inode = NULL;
1103 1104 1105 1106 1107
	spin_unlock(&realm->inodes_with_caps_lock);
	ceph_put_snap_realm(ceph_sb_to_client(ci->vfs_inode.i_sb)->mdsc,
			    realm);
}

S
Sage Weil 已提交
1108
/*
S
Sage Weil 已提交
1109 1110
 * Remove a cap.  Take steps to deal with a racing iterate_session_caps.
 *
1111
 * caller should hold i_ceph_lock.
1112
 * caller will not hold session s_mutex if called from destroy_inode.
S
Sage Weil 已提交
1113
 */
1114
void __ceph_remove_cap(struct ceph_cap *cap, bool queue_release)
S
Sage Weil 已提交
1115 1116 1117
{
	struct ceph_mds_session *session = cap->session;
	struct ceph_inode_info *ci = cap->ci;
1118
	struct ceph_mds_client *mdsc =
1119
		ceph_sb_to_client(ci->vfs_inode.i_sb)->mdsc;
S
Sage Weil 已提交
1120
	int removed = 0;
S
Sage Weil 已提交
1121 1122 1123

	dout("__ceph_remove_cap %p from %p\n", cap, &ci->vfs_inode);

1124 1125 1126 1127 1128
	/* remove from inode's cap rbtree, and clear auth cap */
	rb_erase(&cap->ci_node, &ci->i_caps);
	if (ci->i_auth_cap == cap)
		ci->i_auth_cap = NULL;

S
Sage Weil 已提交
1129 1130 1131 1132 1133 1134 1135 1136 1137 1138
	/* remove from session list */
	spin_lock(&session->s_cap_lock);
	if (session->s_cap_iterator == cap) {
		/* not yet, we are iterating over this very cap */
		dout("__ceph_remove_cap  delaying %p removal from session %p\n",
		     cap, cap->session);
	} else {
		list_del_init(&cap->session_caps);
		session->s_nr_caps--;
		cap->session = NULL;
S
Sage Weil 已提交
1139
		removed = 1;
S
Sage Weil 已提交
1140
	}
S
Sage Weil 已提交
1141 1142
	/* protect backpointer with s_cap_lock: see iterate_session_caps */
	cap->ci = NULL;
1143 1144 1145 1146 1147 1148 1149 1150 1151

	/*
	 * s_cap_reconnect is protected by s_cap_lock. no one changes
	 * s_cap_gen while session is in the reconnect state.
	 */
	if (queue_release &&
	    (!session->s_cap_reconnect || cap->cap_gen == session->s_cap_gen)) {
		cap->queue_release = 1;
		if (removed) {
1152
			__ceph_queue_cap_release(session, cap);
1153 1154 1155 1156 1157 1158 1159
			removed = 0;
		}
	} else {
		cap->queue_release = 0;
	}
	cap->cap_ino = ci->i_vino.ino;

S
Sage Weil 已提交
1160 1161
	spin_unlock(&session->s_cap_lock);

S
Sage Weil 已提交
1162
	if (removed)
1163
		ceph_put_cap(mdsc, cap);
S
Sage Weil 已提交
1164

X
Xiubo Li 已提交
1165 1166 1167 1168 1169 1170 1171
	if (!__ceph_is_any_real_caps(ci)) {
		/* when reconnect denied, we remove session caps forcibly,
		 * i_wr_ref can be non-zero. If there are ongoing write,
		 * keep i_snap_realm.
		 */
		if (ci->i_wr_ref == 0 && ci->i_snap_realm)
			drop_inode_snap_realm(ci);
1172

S
Sage Weil 已提交
1173
		__cap_delay_cancel(mdsc, ci);
X
Xiubo Li 已提交
1174
	}
S
Sage Weil 已提交
1175 1176
}

1177 1178 1179 1180 1181
struct cap_msg_args {
	struct ceph_mds_session	*session;
	u64			ino, cid, follows;
	u64			flush_tid, oldest_flush_tid, size, max_size;
	u64			xattr_version;
1182
	u64			change_attr;
1183
	struct ceph_buffer	*xattr_buf;
1184
	struct ceph_buffer	*old_xattr_buf;
J
Jeff Layton 已提交
1185
	struct timespec64	atime, mtime, ctime, btime;
1186 1187
	int			op, caps, wanted, dirty;
	u32			seq, issue_seq, mseq, time_warp_seq;
1188
	u32			flags;
1189 1190 1191 1192
	kuid_t			uid;
	kgid_t			gid;
	umode_t			mode;
	bool			inline_data;
1193
	bool			wake;
1194 1195
};

S
Sage Weil 已提交
1196 1197 1198 1199 1200
/*
 * Build and send a cap message to the given MDS.
 *
 * Caller should be holding s_mutex.
 */
1201
static int send_cap_msg(struct cap_msg_args *arg)
S
Sage Weil 已提交
1202 1203 1204
{
	struct ceph_mds_caps *fc;
	struct ceph_msg *msg;
Y
Yan, Zheng 已提交
1205 1206
	void *p;
	size_t extra_len;
1207
	struct ceph_osd_client *osdc = &arg->session->s_mdsc->fsc->client->osdc;
S
Sage Weil 已提交
1208 1209

	dout("send_cap_msg %s %llx %llx caps %s wanted %s dirty %s"
1210
	     " seq %u/%u tid %llu/%llu mseq %u follows %lld size %llu/%llu"
1211 1212 1213 1214 1215 1216 1217
	     " xattr_ver %llu xattr_len %d\n", ceph_cap_op_name(arg->op),
	     arg->cid, arg->ino, ceph_cap_string(arg->caps),
	     ceph_cap_string(arg->wanted), ceph_cap_string(arg->dirty),
	     arg->seq, arg->issue_seq, arg->flush_tid, arg->oldest_flush_tid,
	     arg->mseq, arg->follows, arg->size, arg->max_size,
	     arg->xattr_version,
	     arg->xattr_buf ? (int)arg->xattr_buf->vec.iov_len : 0);
S
Sage Weil 已提交
1218

1219 1220
	/* flock buffer size + inline version + inline data size +
	 * osd_epoch_barrier + oldest_flush_tid */
1221
	extra_len = 4 + 8 + 4 + 4 + 8 + 4 + 4 + 4 + 8 + 8 + 4;
Y
Yan, Zheng 已提交
1222 1223
	msg = ceph_msg_new(CEPH_MSG_CLIENT_CAPS, sizeof(*fc) + extra_len,
			   GFP_NOFS, false);
1224 1225
	if (!msg)
		return -ENOMEM;
S
Sage Weil 已提交
1226

1227
	msg->hdr.version = cpu_to_le16(10);
1228
	msg->hdr.tid = cpu_to_le64(arg->flush_tid);
S
Sage Weil 已提交
1229

1230
	fc = msg->front.iov_base;
S
Sage Weil 已提交
1231 1232
	memset(fc, 0, sizeof(*fc));

1233 1234 1235 1236 1237 1238 1239 1240 1241 1242 1243 1244 1245
	fc->cap_id = cpu_to_le64(arg->cid);
	fc->op = cpu_to_le32(arg->op);
	fc->seq = cpu_to_le32(arg->seq);
	fc->issue_seq = cpu_to_le32(arg->issue_seq);
	fc->migrate_seq = cpu_to_le32(arg->mseq);
	fc->caps = cpu_to_le32(arg->caps);
	fc->wanted = cpu_to_le32(arg->wanted);
	fc->dirty = cpu_to_le32(arg->dirty);
	fc->ino = cpu_to_le64(arg->ino);
	fc->snap_follows = cpu_to_le64(arg->follows);

	fc->size = cpu_to_le64(arg->size);
	fc->max_size = cpu_to_le64(arg->max_size);
1246 1247 1248
	ceph_encode_timespec64(&fc->mtime, &arg->mtime);
	ceph_encode_timespec64(&fc->atime, &arg->atime);
	ceph_encode_timespec64(&fc->ctime, &arg->ctime);
1249 1250 1251 1252 1253 1254 1255 1256 1257 1258 1259
	fc->time_warp_seq = cpu_to_le32(arg->time_warp_seq);

	fc->uid = cpu_to_le32(from_kuid(&init_user_ns, arg->uid));
	fc->gid = cpu_to_le32(from_kgid(&init_user_ns, arg->gid));
	fc->mode = cpu_to_le32(arg->mode);

	fc->xattr_version = cpu_to_le64(arg->xattr_version);
	if (arg->xattr_buf) {
		msg->middle = ceph_buffer_get(arg->xattr_buf);
		fc->xattr_len = cpu_to_le32(arg->xattr_buf->vec.iov_len);
		msg->hdr.middle_len = cpu_to_le32(arg->xattr_buf->vec.iov_len);
1260 1261
	}

Y
Yan, Zheng 已提交
1262
	p = fc + 1;
1263
	/* flock buffer size (version 2) */
Y
Yan, Zheng 已提交
1264
	ceph_encode_32(&p, 0);
1265
	/* inline version (version 4) */
1266
	ceph_encode_64(&p, arg->inline_data ? 0 : CEPH_INLINE_NONE);
Y
Yan, Zheng 已提交
1267 1268
	/* inline data size */
	ceph_encode_32(&p, 0);
1269 1270 1271 1272 1273 1274
	/*
	 * osd_epoch_barrier (version 5)
	 * The epoch_barrier is protected osdc->lock, so READ_ONCE here in
	 * case it was recently changed
	 */
	ceph_encode_32(&p, READ_ONCE(osdc->epoch_barrier));
1275
	/* oldest_flush_tid (version 6) */
1276
	ceph_encode_64(&p, arg->oldest_flush_tid);
Y
Yan, Zheng 已提交
1277

1278 1279 1280 1281 1282 1283 1284 1285 1286 1287 1288 1289 1290
	/*
	 * caller_uid/caller_gid (version 7)
	 *
	 * Currently, we don't properly track which caller dirtied the caps
	 * last, and force a flush of them when there is a conflict. For now,
	 * just set this to 0:0, to emulate how the MDS has worked up to now.
	 */
	ceph_encode_32(&p, 0);
	ceph_encode_32(&p, 0);

	/* pool namespace (version 8) (mds always ignores this) */
	ceph_encode_32(&p, 0);

1291
	/* btime and change_attr (version 9) */
J
Jeff Layton 已提交
1292
	ceph_encode_timespec64(p, &arg->btime);
1293
	p += sizeof(struct ceph_timespec);
1294
	ceph_encode_64(&p, arg->change_attr);
1295 1296

	/* Advisory flags (version 10) */
1297
	ceph_encode_32(&p, arg->flags);
1298

1299
	ceph_con_send(&arg->session->s_con, msg);
S
Sage Weil 已提交
1300 1301 1302 1303
	return 0;
}

/*
1304
 * Queue cap releases when an inode is dropped from our cache.
S
Sage Weil 已提交
1305
 */
1306
void __ceph_remove_caps(struct ceph_inode_info *ci)
S
Sage Weil 已提交
1307 1308 1309
{
	struct rb_node *p;

1310 1311 1312
	/* lock i_ceph_lock, because ceph_d_revalidate(..., LOOKUP_RCU)
	 * may call __ceph_caps_issued_mask() on a freeing inode. */
	spin_lock(&ci->i_ceph_lock);
S
Sage Weil 已提交
1313 1314 1315 1316
	p = rb_first(&ci->i_caps);
	while (p) {
		struct ceph_cap *cap = rb_entry(p, struct ceph_cap, ci_node);
		p = rb_next(p);
1317
		__ceph_remove_cap(cap, true);
S
Sage Weil 已提交
1318
	}
1319
	spin_unlock(&ci->i_ceph_lock);
S
Sage Weil 已提交
1320 1321 1322
}

/*
1323 1324 1325
 * Prepare to send a cap message to an MDS. Update the cap state, and populate
 * the arg struct with the parameters that will need to be sent. This should
 * be done under the i_ceph_lock to guard against changes to cap state.
S
Sage Weil 已提交
1326 1327 1328 1329
 *
 * Make note of max_size reported/requested from mds, revoked caps
 * that have now been implemented.
 */
1330 1331 1332
static void __prep_cap(struct cap_msg_args *arg, struct ceph_cap *cap,
		       int op, int flags, int used, int want, int retain,
		       int flushing, u64 flush_tid, u64 oldest_flush_tid)
S
Sage Weil 已提交
1333 1334 1335
{
	struct ceph_inode_info *ci = cap->ci;
	struct inode *inode = &ci->vfs_inode;
1336
	int held, revoking;
S
Sage Weil 已提交
1337

1338
	lockdep_assert_held(&ci->i_ceph_lock);
1339

1340 1341 1342 1343
	held = cap->issued | cap->implemented;
	revoking = cap->implemented & ~cap->issued;
	retain &= ~revoking;

1344 1345
	dout("%s %p cap %p session %p %s -> %s (revoking %s)\n",
	     __func__, inode, cap, cap->session,
S
Sage Weil 已提交
1346 1347 1348 1349
	     ceph_cap_string(held), ceph_cap_string(held & retain),
	     ceph_cap_string(revoking));
	BUG_ON((retain & CEPH_CAP_PIN) == 0);

1350
	ci->i_ceph_flags &= ~CEPH_I_FLUSH;
S
Sage Weil 已提交
1351 1352

	cap->issued &= retain;  /* drop bits we don't want */
1353 1354 1355 1356 1357 1358
	/*
	 * Wake up any waiters on wanted -> needed transition. This is due to
	 * the weird transition from buffered to sync IO... we need to flush
	 * dirty pages _before_ allowing sync writes to avoid reordering.
	 */
	arg->wake = cap->implemented & ~cap->issued;
S
Sage Weil 已提交
1359 1360 1361
	cap->implemented &= cap->issued | used;
	cap->mds_wanted = want;

1362 1363 1364 1365 1366 1367
	arg->session = cap->session;
	arg->ino = ceph_vino(inode).ino;
	arg->cid = cap->cap_id;
	arg->follows = flushing ? ci->i_head_snapc->seq : 0;
	arg->flush_tid = flush_tid;
	arg->oldest_flush_tid = oldest_flush_tid;
1368

1369 1370 1371
	arg->size = inode->i_size;
	ci->i_reported_size = arg->size;
	arg->max_size = ci->i_wanted_max_size;
1372
	if (cap == ci->i_auth_cap)
1373
		ci->i_requested_max_size = arg->max_size;
S
Sage Weil 已提交
1374

S
Sage Weil 已提交
1375
	if (flushing & CEPH_CAP_XATTR_EXCL) {
1376 1377 1378
		arg->old_xattr_buf = __ceph_build_xattrs_blob(ci);
		arg->xattr_version = ci->i_xattrs.version;
		arg->xattr_buf = ci->i_xattrs.blob;
1379
	} else {
1380 1381
		arg->xattr_buf = NULL;
		arg->old_xattr_buf = NULL;
S
Sage Weil 已提交
1382 1383
	}

1384 1385 1386 1387 1388
	arg->mtime = inode->i_mtime;
	arg->atime = inode->i_atime;
	arg->ctime = inode->i_ctime;
	arg->btime = ci->i_btime;
	arg->change_attr = inode_peek_iversion_raw(inode);
1389

1390 1391 1392 1393
	arg->op = op;
	arg->caps = cap->implemented;
	arg->wanted = want;
	arg->dirty = flushing;
1394

1395 1396 1397 1398
	arg->seq = cap->seq;
	arg->issue_seq = cap->issue_seq;
	arg->mseq = cap->mseq;
	arg->time_warp_seq = ci->i_time_warp_seq;
1399

1400 1401 1402
	arg->uid = inode->i_uid;
	arg->gid = inode->i_gid;
	arg->mode = inode->i_mode;
1403

1404
	arg->inline_data = ci->i_inline_version != CEPH_INLINE_NONE;
1405 1406 1407 1408 1409 1410 1411 1412 1413 1414 1415 1416
	if (!(flags & CEPH_CLIENT_CAPS_PENDING_CAPSNAP) &&
	    !list_empty(&ci->i_cap_snaps)) {
		struct ceph_cap_snap *capsnap;
		list_for_each_entry_reverse(capsnap, &ci->i_cap_snaps, ci_item) {
			if (capsnap->cap_flush.tid)
				break;
			if (capsnap->need_flush) {
				flags |= CEPH_CLIENT_CAPS_PENDING_CAPSNAP;
				break;
			}
		}
	}
1417 1418
	arg->flags = flags;
}
S
Sage Weil 已提交
1419

1420 1421 1422 1423 1424 1425 1426 1427 1428 1429
/*
 * Send a cap msg on the given inode.
 *
 * Caller should hold snap_rwsem (read), s_mutex.
 */
static void __send_cap(struct ceph_mds_client *mdsc, struct cap_msg_args *arg,
		       struct ceph_inode_info *ci)
{
	struct inode *inode = &ci->vfs_inode;
	int ret;
1430

1431
	ret = send_cap_msg(arg);
S
Sage Weil 已提交
1432
	if (ret < 0) {
1433 1434
		pr_err("error sending cap msg, ino (%llx.%llx) "
		       "flushing %s tid %llu, requeue\n",
1435 1436
		       ceph_vinop(inode), ceph_cap_string(arg->dirty),
		       arg->flush_tid);
1437 1438 1439
		spin_lock(&ci->i_ceph_lock);
		__cap_delay_requeue(mdsc, ci);
		spin_unlock(&ci->i_ceph_lock);
S
Sage Weil 已提交
1440 1441
	}

1442
	ceph_buffer_put(arg->old_xattr_buf);
S
Sage Weil 已提交
1443

1444 1445
	if (arg->wake)
		wake_up_all(&ci->i_cap_wq);
S
Sage Weil 已提交
1446 1447
}

1448 1449 1450 1451 1452
static inline int __send_flush_snap(struct inode *inode,
				    struct ceph_mds_session *session,
				    struct ceph_cap_snap *capsnap,
				    u32 mseq, u64 oldest_flush_tid)
{
1453 1454 1455 1456 1457 1458 1459 1460 1461 1462 1463 1464 1465
	struct cap_msg_args	arg;

	arg.session = session;
	arg.ino = ceph_vino(inode).ino;
	arg.cid = 0;
	arg.follows = capsnap->follows;
	arg.flush_tid = capsnap->cap_flush.tid;
	arg.oldest_flush_tid = oldest_flush_tid;

	arg.size = capsnap->size;
	arg.max_size = 0;
	arg.xattr_version = capsnap->xattr_version;
	arg.xattr_buf = capsnap->xattr_blob;
1466
	arg.old_xattr_buf = NULL;
1467 1468 1469 1470

	arg.atime = capsnap->atime;
	arg.mtime = capsnap->mtime;
	arg.ctime = capsnap->ctime;
J
Jeff Layton 已提交
1471
	arg.btime = capsnap->btime;
1472
	arg.change_attr = capsnap->change_attr;
1473 1474 1475 1476 1477 1478 1479 1480 1481 1482 1483 1484 1485 1486 1487 1488

	arg.op = CEPH_CAP_OP_FLUSHSNAP;
	arg.caps = capsnap->issued;
	arg.wanted = 0;
	arg.dirty = capsnap->dirty;

	arg.seq = 0;
	arg.issue_seq = 0;
	arg.mseq = mseq;
	arg.time_warp_seq = capsnap->time_warp_seq;

	arg.uid = capsnap->uid;
	arg.gid = capsnap->gid;
	arg.mode = capsnap->mode;

	arg.inline_data = capsnap->inline_data;
1489
	arg.flags = 0;
1490
	arg.wake = false;
1491 1492

	return send_cap_msg(&arg);
1493 1494
}

S
Sage Weil 已提交
1495 1496 1497 1498 1499 1500 1501
/*
 * When a snapshot is taken, clients accumulate dirty metadata on
 * inodes with capabilities in ceph_cap_snaps to describe the file
 * state at the time the snapshot was taken.  This must be flushed
 * asynchronously back to the MDS once sync writes complete and dirty
 * data is written out.
 *
1502
 * Called under i_ceph_lock.  Takes s_mutex as needed.
S
Sage Weil 已提交
1503
 */
Y
Yan, Zheng 已提交
1504 1505
static void __ceph_flush_snaps(struct ceph_inode_info *ci,
			       struct ceph_mds_session *session)
1506 1507
		__releases(ci->i_ceph_lock)
		__acquires(ci->i_ceph_lock)
S
Sage Weil 已提交
1508 1509
{
	struct inode *inode = &ci->vfs_inode;
Y
Yan, Zheng 已提交
1510
	struct ceph_mds_client *mdsc = session->s_mdsc;
S
Sage Weil 已提交
1511
	struct ceph_cap_snap *capsnap;
Y
Yan, Zheng 已提交
1512 1513
	u64 oldest_flush_tid = 0;
	u64 first_tid = 1, last_tid = 0;
S
Sage Weil 已提交
1514

Y
Yan, Zheng 已提交
1515
	dout("__flush_snaps %p session %p\n", inode, session);
S
Sage Weil 已提交
1516 1517 1518 1519 1520 1521 1522

	list_for_each_entry(capsnap, &ci->i_cap_snaps, ci_item) {
		/*
		 * we need to wait for sync writes to complete and for dirty
		 * pages to be written out.
		 */
		if (capsnap->dirty_pages || capsnap->writing)
1523
			break;
S
Sage Weil 已提交
1524

1525 1526
		/* should be removed by ceph_try_drop_cap_snap() */
		BUG_ON(!capsnap->need_flush);
1527

1528
		/* only flush each capsnap once */
1529
		if (capsnap->cap_flush.tid > 0) {
Y
Yan, Zheng 已提交
1530
			dout(" already flushed %p, skipping\n", capsnap);
1531 1532 1533
			continue;
		}

1534
		spin_lock(&mdsc->cap_dirty_lock);
1535 1536 1537
		capsnap->cap_flush.tid = ++mdsc->last_cap_flush_tid;
		list_add_tail(&capsnap->cap_flush.g_list,
			      &mdsc->cap_flush_list);
Y
Yan, Zheng 已提交
1538 1539
		if (oldest_flush_tid == 0)
			oldest_flush_tid = __get_oldest_flush_tid(mdsc);
1540 1541 1542 1543
		if (list_empty(&ci->i_flushing_item)) {
			list_add_tail(&ci->i_flushing_item,
				      &session->s_cap_flushing);
		}
1544 1545
		spin_unlock(&mdsc->cap_dirty_lock);

1546 1547 1548
		list_add_tail(&capsnap->cap_flush.i_list,
			      &ci->i_cap_flush_list);

Y
Yan, Zheng 已提交
1549 1550 1551 1552 1553 1554 1555 1556 1557 1558 1559 1560 1561 1562 1563 1564 1565 1566 1567 1568 1569 1570 1571 1572 1573 1574 1575 1576 1577 1578 1579
		if (first_tid == 1)
			first_tid = capsnap->cap_flush.tid;
		last_tid = capsnap->cap_flush.tid;
	}

	ci->i_ceph_flags &= ~CEPH_I_FLUSH_SNAPS;

	while (first_tid <= last_tid) {
		struct ceph_cap *cap = ci->i_auth_cap;
		struct ceph_cap_flush *cf;
		int ret;

		if (!(cap && cap->session == session)) {
			dout("__flush_snaps %p auth cap %p not mds%d, "
			     "stop\n", inode, cap, session->s_mds);
			break;
		}

		ret = -ENOENT;
		list_for_each_entry(cf, &ci->i_cap_flush_list, i_list) {
			if (cf->tid >= first_tid) {
				ret = 0;
				break;
			}
		}
		if (ret < 0)
			break;

		first_tid = cf->tid + 1;

		capsnap = container_of(cf, struct ceph_cap_snap, cap_flush);
1580
		refcount_inc(&capsnap->nref);
1581
		spin_unlock(&ci->i_ceph_lock);
S
Sage Weil 已提交
1582

Y
Yan, Zheng 已提交
1583 1584
		dout("__flush_snaps %p capsnap %p tid %llu %s\n",
		     inode, capsnap, cf->tid, ceph_cap_string(capsnap->dirty));
S
Sage Weil 已提交
1585

Y
Yan, Zheng 已提交
1586 1587 1588 1589 1590 1591 1592
		ret = __send_flush_snap(inode, session, capsnap, cap->mseq,
					oldest_flush_tid);
		if (ret < 0) {
			pr_err("__flush_snaps: error sending cap flushsnap, "
			       "ino (%llx.%llx) tid %llu follows %llu\n",
				ceph_vinop(inode), cf->tid, capsnap->follows);
		}
S
Sage Weil 已提交
1593

Y
Yan, Zheng 已提交
1594
		ceph_put_cap_snap(capsnap);
1595
		spin_lock(&ci->i_ceph_lock);
S
Sage Weil 已提交
1596
	}
Y
Yan, Zheng 已提交
1597
}
S
Sage Weil 已提交
1598

Y
Yan, Zheng 已提交
1599 1600 1601 1602 1603
void ceph_flush_snaps(struct ceph_inode_info *ci,
		      struct ceph_mds_session **psession)
{
	struct inode *inode = &ci->vfs_inode;
	struct ceph_mds_client *mdsc = ceph_inode_to_client(inode)->mdsc;
1604
	struct ceph_mds_session *session = NULL;
Y
Yan, Zheng 已提交
1605
	int mds;
1606

Y
Yan, Zheng 已提交
1607
	dout("ceph_flush_snaps %p\n", inode);
1608 1609
	if (psession)
		session = *psession;
Y
Yan, Zheng 已提交
1610 1611 1612 1613 1614 1615 1616 1617 1618 1619
retry:
	spin_lock(&ci->i_ceph_lock);
	if (!(ci->i_ceph_flags & CEPH_I_FLUSH_SNAPS)) {
		dout(" no capsnap needs flush, doing nothing\n");
		goto out;
	}
	if (!ci->i_auth_cap) {
		dout(" no auth cap (migrating?), doing nothing\n");
		goto out;
	}
S
Sage Weil 已提交
1620

Y
Yan, Zheng 已提交
1621 1622 1623
	mds = ci->i_auth_cap->session->s_mds;
	if (session && session->s_mds != mds) {
		dout(" oops, wrong session %p mutex\n", session);
S
Sage Weil 已提交
1624 1625
		mutex_unlock(&session->s_mutex);
		ceph_put_mds_session(session);
Y
Yan, Zheng 已提交
1626 1627 1628 1629 1630 1631 1632 1633 1634 1635 1636 1637
		session = NULL;
	}
	if (!session) {
		spin_unlock(&ci->i_ceph_lock);
		mutex_lock(&mdsc->mutex);
		session = __ceph_lookup_mds_session(mdsc, mds);
		mutex_unlock(&mdsc->mutex);
		if (session) {
			dout(" inverting session/ino locks on %p\n", session);
			mutex_lock(&session->s_mutex);
		}
		goto retry;
S
Sage Weil 已提交
1638 1639
	}

1640
	// make sure flushsnap messages are sent in proper order.
1641
	if (ci->i_ceph_flags & CEPH_I_KICK_FLUSH)
1642 1643
		__kick_flushing_caps(mdsc, session, ci, 0);

Y
Yan, Zheng 已提交
1644 1645
	__ceph_flush_snaps(ci, session);
out:
1646
	spin_unlock(&ci->i_ceph_lock);
Y
Yan, Zheng 已提交
1647 1648 1649

	if (psession) {
		*psession = session;
1650
	} else if (session) {
Y
Yan, Zheng 已提交
1651 1652 1653 1654 1655 1656 1657
		mutex_unlock(&session->s_mutex);
		ceph_put_mds_session(session);
	}
	/* we flushed them all; remove this inode from the queue */
	spin_lock(&mdsc->snap_flush_lock);
	list_del_init(&ci->i_snap_flush_item);
	spin_unlock(&mdsc->snap_flush_lock);
S
Sage Weil 已提交
1658 1659
}

S
Sage Weil 已提交
1660
/*
1661 1662 1663
 * Mark caps dirty.  If inode is newly dirty, return the dirty flags.
 * Caller is then responsible for calling __mark_inode_dirty with the
 * returned flags value.
S
Sage Weil 已提交
1664
 */
1665 1666
int __ceph_mark_dirty_caps(struct ceph_inode_info *ci, int mask,
			   struct ceph_cap_flush **pcf)
S
Sage Weil 已提交
1667
{
1668
	struct ceph_mds_client *mdsc =
1669
		ceph_sb_to_client(ci->vfs_inode.i_sb)->mdsc;
S
Sage Weil 已提交
1670 1671 1672 1673
	struct inode *inode = &ci->vfs_inode;
	int was = ci->i_dirty_caps;
	int dirty = 0;

1674 1675
	lockdep_assert_held(&ci->i_ceph_lock);

1676 1677 1678 1679 1680 1681 1682
	if (!ci->i_auth_cap) {
		pr_warn("__mark_dirty_caps %p %llx mask %s, "
			"but no auth cap (session was closed?)\n",
			inode, ceph_ino(inode), ceph_cap_string(mask));
		return 0;
	}

S
Sage Weil 已提交
1683 1684 1685 1686 1687
	dout("__mark_dirty_caps %p %s dirty %s -> %s\n", &ci->vfs_inode,
	     ceph_cap_string(mask), ceph_cap_string(was),
	     ceph_cap_string(was | mask));
	ci->i_dirty_caps |= mask;
	if (was == 0) {
1688 1689 1690
		WARN_ON_ONCE(ci->i_prealloc_cap_flush);
		swap(ci->i_prealloc_cap_flush, *pcf);

1691 1692
		if (!ci->i_head_snapc) {
			WARN_ON_ONCE(!rwsem_is_locked(&mdsc->snap_rwsem));
1693 1694
			ci->i_head_snapc = ceph_get_snap_context(
				ci->i_snap_realm->cached_context);
1695
		}
1696 1697
		dout(" inode %p now dirty snapc %p auth cap %p\n",
		     &ci->vfs_inode, ci->i_head_snapc, ci->i_auth_cap);
S
Sage Weil 已提交
1698 1699
		BUG_ON(!list_empty(&ci->i_dirty_item));
		spin_lock(&mdsc->cap_dirty_lock);
1700
		list_add(&ci->i_dirty_item, &mdsc->cap_dirty);
S
Sage Weil 已提交
1701 1702
		spin_unlock(&mdsc->cap_dirty_lock);
		if (ci->i_flushing_caps == 0) {
1703
			ihold(inode);
S
Sage Weil 已提交
1704 1705
			dirty |= I_DIRTY_SYNC;
		}
1706 1707
	} else {
		WARN_ON_ONCE(!ci->i_prealloc_cap_flush);
S
Sage Weil 已提交
1708 1709 1710 1711 1712
	}
	BUG_ON(list_empty(&ci->i_dirty_item));
	if (((was | ci->i_flushing_caps) & CEPH_CAP_FILE_BUFFER) &&
	    (mask & CEPH_CAP_FILE_BUFFER))
		dirty |= I_DIRTY_DATASYNC;
1713
	__cap_delay_requeue(mdsc, ci);
1714
	return dirty;
S
Sage Weil 已提交
1715 1716
}

1717 1718 1719 1720 1721 1722 1723 1724 1725 1726 1727
struct ceph_cap_flush *ceph_alloc_cap_flush(void)
{
	return kmem_cache_alloc(ceph_cap_flush_cachep, GFP_KERNEL);
}

void ceph_free_cap_flush(struct ceph_cap_flush *cf)
{
	if (cf)
		kmem_cache_free(ceph_cap_flush_cachep, cf);
}

1728 1729
static u64 __get_oldest_flush_tid(struct ceph_mds_client *mdsc)
{
1730
	if (!list_empty(&mdsc->cap_flush_list)) {
1731
		struct ceph_cap_flush *cf =
1732 1733
			list_first_entry(&mdsc->cap_flush_list,
					 struct ceph_cap_flush, g_list);
1734 1735 1736 1737 1738
		return cf->tid;
	}
	return 0;
}

Y
Yan, Zheng 已提交
1739 1740 1741 1742
/*
 * Remove cap_flush from the mdsc's or inode's flushing cap list.
 * Return true if caller needs to wake up flush waiters.
 */
J
Jeff Layton 已提交
1743 1744
static bool __detach_cap_flush_from_mdsc(struct ceph_mds_client *mdsc,
					 struct ceph_cap_flush *cf)
Y
Yan, Zheng 已提交
1745 1746 1747
{
	struct ceph_cap_flush *prev;
	bool wake = cf->wake;
J
Jeff Layton 已提交
1748 1749 1750 1751 1752 1753 1754 1755 1756 1757 1758 1759 1760 1761 1762 1763 1764 1765 1766 1767

	if (wake && cf->g_list.prev != &mdsc->cap_flush_list) {
		prev = list_prev_entry(cf, g_list);
		prev->wake = true;
		wake = false;
	}
	list_del(&cf->g_list);
	return wake;
}

static bool __detach_cap_flush_from_ci(struct ceph_inode_info *ci,
				       struct ceph_cap_flush *cf)
{
	struct ceph_cap_flush *prev;
	bool wake = cf->wake;

	if (wake && cf->i_list.prev != &ci->i_cap_flush_list) {
		prev = list_prev_entry(cf, i_list);
		prev->wake = true;
		wake = false;
Y
Yan, Zheng 已提交
1768
	}
J
Jeff Layton 已提交
1769
	list_del(&cf->i_list);
Y
Yan, Zheng 已提交
1770 1771 1772
	return wake;
}

S
Sage Weil 已提交
1773 1774 1775
/*
 * Add dirty inode to the flushing list.  Assigned a seq number so we
 * can wait for caps to flush without starving.
1776
 *
1777
 * Called under i_ceph_lock. Returns the flush tid.
S
Sage Weil 已提交
1778
 */
1779
static u64 __mark_caps_flushing(struct inode *inode,
Y
Yan, Zheng 已提交
1780
				struct ceph_mds_session *session, bool wake,
1781
				u64 *oldest_flush_tid)
S
Sage Weil 已提交
1782
{
1783
	struct ceph_mds_client *mdsc = ceph_sb_to_client(inode->i_sb)->mdsc;
S
Sage Weil 已提交
1784
	struct ceph_inode_info *ci = ceph_inode(inode);
1785
	struct ceph_cap_flush *cf = NULL;
1786
	int flushing;
S
Sage Weil 已提交
1787

1788
	lockdep_assert_held(&ci->i_ceph_lock);
1789
	BUG_ON(ci->i_dirty_caps == 0);
S
Sage Weil 已提交
1790
	BUG_ON(list_empty(&ci->i_dirty_item));
1791
	BUG_ON(!ci->i_prealloc_cap_flush);
1792 1793 1794 1795 1796 1797 1798 1799

	flushing = ci->i_dirty_caps;
	dout("__mark_caps_flushing flushing %s, flushing_caps %s -> %s\n",
	     ceph_cap_string(flushing),
	     ceph_cap_string(ci->i_flushing_caps),
	     ceph_cap_string(ci->i_flushing_caps | flushing));
	ci->i_flushing_caps |= flushing;
	ci->i_dirty_caps = 0;
1800
	dout(" inode %p now !dirty\n", inode);
1801

1802
	swap(cf, ci->i_prealloc_cap_flush);
1803
	cf->caps = flushing;
Y
Yan, Zheng 已提交
1804
	cf->wake = wake;
1805

S
Sage Weil 已提交
1806
	spin_lock(&mdsc->cap_dirty_lock);
1807 1808
	list_del_init(&ci->i_dirty_item);

1809
	cf->tid = ++mdsc->last_cap_flush_tid;
1810
	list_add_tail(&cf->g_list, &mdsc->cap_flush_list);
1811
	*oldest_flush_tid = __get_oldest_flush_tid(mdsc);
1812

S
Sage Weil 已提交
1813 1814 1815 1816 1817
	if (list_empty(&ci->i_flushing_item)) {
		list_add_tail(&ci->i_flushing_item, &session->s_cap_flushing);
		mdsc->num_cap_flushing++;
	}
	spin_unlock(&mdsc->cap_dirty_lock);
1818

1819
	list_add_tail(&cf->i_list, &ci->i_cap_flush_list);
1820

1821
	return cf->tid;
S
Sage Weil 已提交
1822 1823
}

1824 1825 1826 1827 1828 1829 1830 1831
/*
 * try to invalidate mapping pages without blocking.
 */
static int try_nonblocking_invalidate(struct inode *inode)
{
	struct ceph_inode_info *ci = ceph_inode(inode);
	u32 invalidating_gen = ci->i_rdcache_gen;

1832
	spin_unlock(&ci->i_ceph_lock);
1833
	invalidate_mapping_pages(&inode->i_data, 0, -1);
1834
	spin_lock(&ci->i_ceph_lock);
1835

1836
	if (inode->i_data.nrpages == 0 &&
1837 1838 1839
	    invalidating_gen == ci->i_rdcache_gen) {
		/* success. */
		dout("try_nonblocking_invalidate %p success\n", inode);
1840 1841
		/* save any racing async invalidate some trouble */
		ci->i_rdcache_revoking = ci->i_rdcache_gen - 1;
1842 1843 1844 1845 1846 1847
		return 0;
	}
	dout("try_nonblocking_invalidate %p failed\n", inode);
	return -1;
}

1848 1849 1850 1851 1852 1853 1854 1855 1856 1857 1858 1859 1860 1861 1862
bool __ceph_should_report_size(struct ceph_inode_info *ci)
{
	loff_t size = ci->vfs_inode.i_size;
	/* mds will adjust max size according to the reported size */
	if (ci->i_flushing_caps & CEPH_CAP_FILE_WR)
		return false;
	if (size >= ci->i_max_size)
		return true;
	/* half of previous max_size increment has been used */
	if (ci->i_max_size > ci->i_reported_size &&
	    (size << 1) >= ci->i_max_size + ci->i_reported_size)
		return true;
	return false;
}

S
Sage Weil 已提交
1863 1864 1865 1866 1867 1868 1869 1870 1871 1872 1873 1874
/*
 * Swiss army knife function to examine currently used and wanted
 * versus held caps.  Release, flush, ack revoked caps to mds as
 * appropriate.
 *
 *  CHECK_CAPS_AUTHONLY - we should only check the auth cap
 *  CHECK_CAPS_FLUSH - we should flush any dirty caps immediately, without
 *    further delay.
 */
void ceph_check_caps(struct ceph_inode_info *ci, int flags,
		     struct ceph_mds_session *session)
{
1875 1876
	struct ceph_fs_client *fsc = ceph_inode_to_client(&ci->vfs_inode);
	struct ceph_mds_client *mdsc = fsc->mdsc;
S
Sage Weil 已提交
1877 1878
	struct inode *inode = &ci->vfs_inode;
	struct ceph_cap *cap;
1879
	u64 flush_tid, oldest_flush_tid;
1880
	int file_wanted, used, cap_used;
S
Sage Weil 已提交
1881
	int took_snap_rwsem = 0;             /* true if mdsc->snap_rwsem held */
S
Sage Weil 已提交
1882
	int issued, implemented, want, retain, revoking, flushing = 0;
S
Sage Weil 已提交
1883 1884 1885
	int mds = -1;   /* keep track of how far we've gone through i_caps list
			   to avoid an infinite loop on retry */
	struct rb_node *p;
1886 1887
	bool queue_invalidate = false;
	bool tried_invalidate = false;
S
Sage Weil 已提交
1888

1889
	spin_lock(&ci->i_ceph_lock);
S
Sage Weil 已提交
1890 1891 1892 1893 1894
	if (ci->i_ceph_flags & CEPH_I_FLUSH)
		flags |= CHECK_CAPS_FLUSH;

	goto retry_locked;
retry:
1895
	spin_lock(&ci->i_ceph_lock);
S
Sage Weil 已提交
1896 1897 1898
retry_locked:
	file_wanted = __ceph_caps_file_wanted(ci);
	used = __ceph_caps_used(ci);
S
Sage Weil 已提交
1899 1900
	issued = __ceph_caps_issued(ci, &implemented);
	revoking = implemented & ~issued;
S
Sage Weil 已提交
1901

1902 1903
	want = file_wanted;
	retain = file_wanted | used | CEPH_CAP_PIN;
S
Sage Weil 已提交
1904
	if (!mdsc->stopping && inode->i_nlink > 0) {
1905
		if (file_wanted) {
S
Sage Weil 已提交
1906
			retain |= CEPH_CAP_ANY;       /* be greedy */
1907 1908
		} else if (S_ISDIR(inode->i_mode) &&
			   (issued & CEPH_CAP_FILE_SHARED) &&
1909
			   __ceph_dir_is_complete(ci)) {
1910 1911 1912 1913 1914 1915
			/*
			 * If a directory is complete, we want to keep
			 * the exclusive cap. So that MDS does not end up
			 * revoking the shared cap on every create/unlink
			 * operation.
			 */
1916
			if (IS_RDONLY(inode)) {
1917
				want = CEPH_CAP_ANY_SHARED;
1918
			} else {
1919
				want |= CEPH_CAP_ANY_SHARED | CEPH_CAP_FILE_EXCL;
1920
			}
1921
			retain |= want;
S
Sage Weil 已提交
1922
		} else {
1923

S
Sage Weil 已提交
1924 1925 1926 1927 1928 1929 1930 1931 1932 1933 1934 1935
			retain |= CEPH_CAP_ANY_SHARED;
			/*
			 * keep RD only if we didn't have the file open RW,
			 * because then the mds would revoke it anyway to
			 * journal max_size=0.
			 */
			if (ci->i_max_size == 0)
				retain |= CEPH_CAP_ANY_RD;
		}
	}

	dout("check_caps %p file_want %s used %s dirty %s flushing %s"
1936
	     " issued %s revoking %s retain %s %s%s\n", inode,
S
Sage Weil 已提交
1937 1938 1939
	     ceph_cap_string(file_wanted),
	     ceph_cap_string(used), ceph_cap_string(ci->i_dirty_caps),
	     ceph_cap_string(ci->i_flushing_caps),
S
Sage Weil 已提交
1940
	     ceph_cap_string(issued), ceph_cap_string(revoking),
S
Sage Weil 已提交
1941 1942 1943 1944 1945 1946 1947 1948 1949
	     ceph_cap_string(retain),
	     (flags & CHECK_CAPS_AUTHONLY) ? " AUTHONLY" : "",
	     (flags & CHECK_CAPS_FLUSH) ? " FLUSH" : "");

	/*
	 * If we no longer need to hold onto old our caps, and we may
	 * have cached pages, but don't want them, then try to invalidate.
	 * If we fail, it's because pages are locked.... try again later.
	 */
1950
	if ((!(flags & CHECK_CAPS_NOINVAL) || mdsc->stopping) &&
1951
	    S_ISREG(inode->i_mode) &&
1952
	    !(ci->i_wb_ref || ci->i_wrbuffer_ref) &&   /* no dirty pages... */
Y
Yan, Zheng 已提交
1953
	    inode->i_data.nrpages &&		/* have cached pages */
1954 1955
	    (revoking & (CEPH_CAP_FILE_CACHE|
			 CEPH_CAP_FILE_LAZYIO)) && /*  or revoking cache */
S
Sage Weil 已提交
1956 1957
	    !tried_invalidate) {
		dout("check_caps trying to invalidate on %p\n", inode);
1958
		if (try_nonblocking_invalidate(inode) < 0) {
1959 1960 1961
			dout("check_caps queuing invalidate\n");
			queue_invalidate = true;
			ci->i_rdcache_revoking = ci->i_rdcache_gen;
S
Sage Weil 已提交
1962
		}
1963
		tried_invalidate = true;
S
Sage Weil 已提交
1964 1965 1966 1967
		goto retry_locked;
	}

	for (p = rb_first(&ci->i_caps); p; p = rb_next(p)) {
1968 1969
		struct cap_msg_args arg;

S
Sage Weil 已提交
1970 1971 1972 1973 1974 1975 1976 1977 1978
		cap = rb_entry(p, struct ceph_cap, ci_node);

		/* avoid looping forever */
		if (mds >= cap->mds ||
		    ((flags & CHECK_CAPS_AUTHONLY) && cap != ci->i_auth_cap))
			continue;

		/* NOTE: no side-effects allowed, until we take s_mutex */

1979 1980 1981 1982
		cap_used = used;
		if (ci->i_auth_cap && cap != ci->i_auth_cap)
			cap_used &= ~ci->i_auth_cap->issued;

S
Sage Weil 已提交
1983
		revoking = cap->implemented & ~cap->issued;
1984
		dout(" mds%d cap %p used %s issued %s implemented %s revoking %s\n",
1985 1986
		     cap->mds, cap, ceph_cap_string(cap_used),
		     ceph_cap_string(cap->issued),
1987 1988
		     ceph_cap_string(cap->implemented),
		     ceph_cap_string(revoking));
S
Sage Weil 已提交
1989 1990 1991 1992 1993 1994 1995 1996 1997 1998 1999

		if (cap == ci->i_auth_cap &&
		    (cap->issued & CEPH_CAP_FILE_WR)) {
			/* request larger max_size from MDS? */
			if (ci->i_wanted_max_size > ci->i_max_size &&
			    ci->i_wanted_max_size > ci->i_requested_max_size) {
				dout("requesting new max_size\n");
				goto ack;
			}

			/* approaching file_max? */
2000
			if (__ceph_should_report_size(ci)) {
S
Sage Weil 已提交
2001 2002 2003 2004 2005
				dout("i_size approaching max_size\n");
				goto ack;
			}
		}
		/* flush anything dirty? */
2006 2007 2008 2009 2010 2011 2012 2013 2014
		if (cap == ci->i_auth_cap) {
			if ((flags & CHECK_CAPS_FLUSH) && ci->i_dirty_caps) {
				dout("flushing dirty caps\n");
				goto ack;
			}
			if (ci->i_ceph_flags & CEPH_I_FLUSH_SNAPS) {
				dout("flushing snap caps\n");
				goto ack;
			}
S
Sage Weil 已提交
2015 2016 2017
		}

		/* completed revocation? going down and there are no caps? */
2018
		if (revoking && (revoking & cap_used) == 0) {
S
Sage Weil 已提交
2019 2020 2021 2022 2023 2024
			dout("completed revocation of %s\n",
			     ceph_cap_string(cap->implemented & ~cap->issued));
			goto ack;
		}

		/* want more caps from mds? */
2025 2026 2027 2028 2029 2030
		if (want & ~cap->mds_wanted) {
			if (want & ~(cap->mds_wanted | cap->issued))
				goto ack;
			if (!__cap_is_valid(cap))
				goto ack;
		}
S
Sage Weil 已提交
2031 2032

		/* things we might delay */
2033
		if ((cap->issued & ~retain) == 0)
S
Sage Weil 已提交
2034 2035 2036 2037 2038 2039 2040 2041 2042 2043 2044 2045 2046
			continue;     /* nope, all good */

ack:
		if (session && session != cap->session) {
			dout("oops, wrong session %p mutex\n", session);
			mutex_unlock(&session->s_mutex);
			session = NULL;
		}
		if (!session) {
			session = cap->session;
			if (mutex_trylock(&session->s_mutex) == 0) {
				dout("inverting session/ino locks on %p\n",
				     session);
2047
				spin_unlock(&ci->i_ceph_lock);
S
Sage Weil 已提交
2048 2049 2050 2051 2052 2053 2054 2055
				if (took_snap_rwsem) {
					up_read(&mdsc->snap_rwsem);
					took_snap_rwsem = 0;
				}
				mutex_lock(&session->s_mutex);
				goto retry;
			}
		}
2056 2057 2058 2059 2060 2061

		/* kick flushing and flush snaps before sending normal
		 * cap message */
		if (cap == ci->i_auth_cap &&
		    (ci->i_ceph_flags &
		     (CEPH_I_KICK_FLUSH | CEPH_I_FLUSH_SNAPS))) {
2062
			if (ci->i_ceph_flags & CEPH_I_KICK_FLUSH)
2063
				__kick_flushing_caps(mdsc, session, ci, 0);
Y
Yan, Zheng 已提交
2064 2065 2066
			if (ci->i_ceph_flags & CEPH_I_FLUSH_SNAPS)
				__ceph_flush_snaps(ci, session);

2067 2068 2069
			goto retry_locked;
		}

S
Sage Weil 已提交
2070 2071 2072 2073 2074
		/* take snap_rwsem after session mutex */
		if (!took_snap_rwsem) {
			if (down_read_trylock(&mdsc->snap_rwsem) == 0) {
				dout("inverting snap/in locks on %p\n",
				     inode);
2075
				spin_unlock(&ci->i_ceph_lock);
S
Sage Weil 已提交
2076 2077 2078 2079 2080 2081 2082
				down_read(&mdsc->snap_rwsem);
				took_snap_rwsem = 1;
				goto retry;
			}
			took_snap_rwsem = 1;
		}

2083
		if (cap == ci->i_auth_cap && ci->i_dirty_caps) {
2084 2085 2086
			flushing = ci->i_dirty_caps;
			flush_tid = __mark_caps_flushing(inode, session, false,
							 &oldest_flush_tid);
2087
		} else {
2088
			flushing = 0;
2089
			flush_tid = 0;
2090 2091 2092
			spin_lock(&mdsc->cap_dirty_lock);
			oldest_flush_tid = __get_oldest_flush_tid(mdsc);
			spin_unlock(&mdsc->cap_dirty_lock);
2093
		}
S
Sage Weil 已提交
2094 2095 2096

		mds = cap->mds;  /* remember mds, so we don't repeat */

2097
		__prep_cap(&arg, cap, CEPH_CAP_OP_UPDATE, 0, cap_used, want,
2098
			   retain, flushing, flush_tid, oldest_flush_tid);
2099 2100 2101 2102
		spin_unlock(&ci->i_ceph_lock);

		__send_cap(mdsc, &arg, ci);

2103
		goto retry; /* retake i_ceph_lock and restart our cap scan. */
S
Sage Weil 已提交
2104 2105
	}

2106 2107 2108 2109 2110 2111
	/* periodically re-calculate caps wanted by open files */
	if (__ceph_is_any_real_caps(ci) &&
	    list_empty(&ci->i_cap_delay_list) &&
	    (file_wanted & ~CEPH_CAP_PIN) &&
	    !(used & (CEPH_CAP_FILE_RD | CEPH_CAP_ANY_FILE_WR))) {
		__cap_delay_requeue(mdsc, ci);
2112
	}
S
Sage Weil 已提交
2113

2114
	spin_unlock(&ci->i_ceph_lock);
S
Sage Weil 已提交
2115

S
Sage Weil 已提交
2116
	if (queue_invalidate)
2117
		ceph_queue_invalidate(inode);
S
Sage Weil 已提交
2118

2119
	if (session)
S
Sage Weil 已提交
2120 2121 2122 2123 2124 2125 2126 2127
		mutex_unlock(&session->s_mutex);
	if (took_snap_rwsem)
		up_read(&mdsc->snap_rwsem);
}

/*
 * Try to flush dirty caps back to the auth mds.
 */
2128
static int try_flush_caps(struct inode *inode, u64 *ptid)
S
Sage Weil 已提交
2129
{
2130
	struct ceph_mds_client *mdsc = ceph_sb_to_client(inode->i_sb)->mdsc;
S
Sage Weil 已提交
2131
	struct ceph_inode_info *ci = ceph_inode(inode);
2132
	struct ceph_mds_session *session = NULL;
Y
Yan, Zheng 已提交
2133
	int flushing = 0;
2134
	u64 flush_tid = 0, oldest_flush_tid = 0;
S
Sage Weil 已提交
2135 2136

retry:
2137
	spin_lock(&ci->i_ceph_lock);
2138
retry_locked:
S
Sage Weil 已提交
2139 2140
	if (ci->i_dirty_caps && ci->i_auth_cap) {
		struct ceph_cap *cap = ci->i_auth_cap;
2141
		struct cap_msg_args arg;
S
Sage Weil 已提交
2142

2143
		if (session != cap->session) {
2144
			spin_unlock(&ci->i_ceph_lock);
2145 2146
			if (session)
				mutex_unlock(&session->s_mutex);
S
Sage Weil 已提交
2147 2148 2149 2150
			session = cap->session;
			mutex_lock(&session->s_mutex);
			goto retry;
		}
2151 2152
		if (cap->session->s_state < CEPH_MDS_SESSION_OPEN) {
			spin_unlock(&ci->i_ceph_lock);
S
Sage Weil 已提交
2153
			goto out;
2154
		}
S
Sage Weil 已提交
2155

2156 2157 2158 2159 2160 2161 2162 2163 2164
		if (ci->i_ceph_flags &
		    (CEPH_I_KICK_FLUSH | CEPH_I_FLUSH_SNAPS)) {
			if (ci->i_ceph_flags & CEPH_I_KICK_FLUSH)
				__kick_flushing_caps(mdsc, session, ci, 0);
			if (ci->i_ceph_flags & CEPH_I_FLUSH_SNAPS)
				__ceph_flush_snaps(ci, session);
			goto retry_locked;
		}

2165 2166 2167
		flushing = ci->i_dirty_caps;
		flush_tid = __mark_caps_flushing(inode, session, true,
						 &oldest_flush_tid);
S
Sage Weil 已提交
2168

2169
		__prep_cap(&arg, cap, CEPH_CAP_OP_FLUSH, CEPH_CLIENT_CAPS_SYNC,
2170 2171 2172
			   __ceph_caps_used(ci), __ceph_caps_wanted(ci),
			   (cap->issued | cap->implemented),
			   flushing, flush_tid, oldest_flush_tid);
2173 2174 2175
		spin_unlock(&ci->i_ceph_lock);

		__send_cap(mdsc, &arg, ci);
2176
	} else {
2177
		if (!list_empty(&ci->i_cap_flush_list)) {
2178
			struct ceph_cap_flush *cf =
2179
				list_last_entry(&ci->i_cap_flush_list,
Y
Yan, Zheng 已提交
2180 2181
						struct ceph_cap_flush, i_list);
			cf->wake = true;
2182 2183 2184 2185
			flush_tid = cf->tid;
		}
		flushing = ci->i_flushing_caps;
		spin_unlock(&ci->i_ceph_lock);
S
Sage Weil 已提交
2186 2187
	}
out:
2188
	if (session)
S
Sage Weil 已提交
2189
		mutex_unlock(&session->s_mutex);
2190 2191

	*ptid = flush_tid;
S
Sage Weil 已提交
2192 2193 2194 2195 2196 2197
	return flushing;
}

/*
 * Return true if we've flushed caps through the given flush_tid.
 */
2198
static int caps_are_flushed(struct inode *inode, u64 flush_tid)
S
Sage Weil 已提交
2199 2200
{
	struct ceph_inode_info *ci = ceph_inode(inode);
2201
	int ret = 1;
S
Sage Weil 已提交
2202

2203
	spin_lock(&ci->i_ceph_lock);
2204 2205 2206 2207
	if (!list_empty(&ci->i_cap_flush_list)) {
		struct ceph_cap_flush * cf =
			list_first_entry(&ci->i_cap_flush_list,
					 struct ceph_cap_flush, i_list);
2208
		if (cf->tid <= flush_tid)
S
Sage Weil 已提交
2209
			ret = 0;
Y
Yan, Zheng 已提交
2210
	}
2211
	spin_unlock(&ci->i_ceph_lock);
S
Sage Weil 已提交
2212 2213 2214
	return ret;
}

Y
Yan, Zheng 已提交
2215
/*
2216
 * wait for any unsafe requests to complete.
Y
Yan, Zheng 已提交
2217
 */
2218
static int unsafe_request_wait(struct inode *inode)
Y
Yan, Zheng 已提交
2219 2220
{
	struct ceph_inode_info *ci = ceph_inode(inode);
2221 2222
	struct ceph_mds_request *req1 = NULL, *req2 = NULL;
	int ret, err = 0;
Y
Yan, Zheng 已提交
2223 2224

	spin_lock(&ci->i_unsafe_lock);
2225 2226 2227 2228 2229 2230 2231 2232 2233 2234 2235 2236 2237
	if (S_ISDIR(inode->i_mode) && !list_empty(&ci->i_unsafe_dirops)) {
		req1 = list_last_entry(&ci->i_unsafe_dirops,
					struct ceph_mds_request,
					r_unsafe_dir_item);
		ceph_mdsc_get_request(req1);
	}
	if (!list_empty(&ci->i_unsafe_iops)) {
		req2 = list_last_entry(&ci->i_unsafe_iops,
					struct ceph_mds_request,
					r_unsafe_target_item);
		ceph_mdsc_get_request(req2);
	}
	spin_unlock(&ci->i_unsafe_lock);
Y
Yan, Zheng 已提交
2238

2239
	dout("unsafe_request_wait %p wait on tid %llu %llu\n",
2240 2241 2242 2243
	     inode, req1 ? req1->r_tid : 0ULL, req2 ? req2->r_tid : 0ULL);
	if (req1) {
		ret = !wait_for_completion_timeout(&req1->r_safe_completion,
					ceph_timeout_jiffies(req1->r_timeout));
Y
Yan, Zheng 已提交
2244
		if (ret)
2245 2246 2247 2248 2249 2250 2251 2252 2253 2254 2255
			err = -EIO;
		ceph_mdsc_put_request(req1);
	}
	if (req2) {
		ret = !wait_for_completion_timeout(&req2->r_safe_completion,
					ceph_timeout_jiffies(req2->r_timeout));
		if (ret)
			err = -EIO;
		ceph_mdsc_put_request(req2);
	}
	return err;
Y
Yan, Zheng 已提交
2256 2257
}

2258
int ceph_fsync(struct file *file, loff_t start, loff_t end, int datasync)
S
Sage Weil 已提交
2259
{
2260
	struct ceph_file_info *fi = file->private_data;
2261
	struct inode *inode = file->f_mapping->host;
S
Sage Weil 已提交
2262
	struct ceph_inode_info *ci = ceph_inode(inode);
2263
	u64 flush_tid;
2264
	int ret, err;
S
Sage Weil 已提交
2265 2266 2267
	int dirty;

	dout("fsync %p%s\n", inode, datasync ? " datasync" : "");
2268

2269
	ret = file_write_and_wait_range(file, start, end);
Y
Yan, Zheng 已提交
2270 2271 2272
	if (datasync)
		goto out;

2273 2274 2275 2276
	ret = ceph_wait_on_async_create(inode);
	if (ret)
		goto out;

2277
	dirty = try_flush_caps(inode, &flush_tid);
S
Sage Weil 已提交
2278 2279
	dout("fsync dirty caps are %s\n", ceph_cap_string(dirty));

2280
	err = unsafe_request_wait(inode);
Y
Yan, Zheng 已提交
2281

S
Sage Weil 已提交
2282 2283 2284 2285 2286
	/*
	 * only wait on non-file metadata writeback (the mds
	 * can recover size and mtime, so we don't need to
	 * wait for that)
	 */
2287 2288
	if (!err && (dirty & ~CEPH_CAP_ANY_FILE_WR)) {
		err = wait_event_interruptible(ci->i_cap_wq,
Y
Yan, Zheng 已提交
2289
					caps_are_flushed(inode, flush_tid));
S
Sage Weil 已提交
2290
	}
2291 2292 2293 2294 2295 2296 2297 2298 2299 2300 2301 2302

	if (err < 0)
		ret = err;

	if (errseq_check(&ci->i_meta_err, READ_ONCE(fi->meta_err))) {
		spin_lock(&file->f_lock);
		err = errseq_check_and_advance(&ci->i_meta_err,
					       &fi->meta_err);
		spin_unlock(&file->f_lock);
		if (err < 0)
			ret = err;
	}
Y
Yan, Zheng 已提交
2303 2304
out:
	dout("fsync %p%s result=%d\n", inode, datasync ? " datasync" : "", ret);
S
Sage Weil 已提交
2305 2306 2307 2308 2309 2310 2311 2312 2313
	return ret;
}

/*
 * Flush any dirty caps back to the mds.  If we aren't asked to wait,
 * queue inode for flush but don't do so immediately, because we can
 * get by with fewer MDS messages if we wait for data writeback to
 * complete first.
 */
2314
int ceph_write_inode(struct inode *inode, struct writeback_control *wbc)
S
Sage Weil 已提交
2315 2316
{
	struct ceph_inode_info *ci = ceph_inode(inode);
2317
	u64 flush_tid;
S
Sage Weil 已提交
2318 2319
	int err = 0;
	int dirty;
2320
	int wait = (wbc->sync_mode == WB_SYNC_ALL && !wbc->for_sync);
S
Sage Weil 已提交
2321 2322 2323

	dout("write_inode %p wait=%d\n", inode, wait);
	if (wait) {
2324
		dirty = try_flush_caps(inode, &flush_tid);
S
Sage Weil 已提交
2325 2326 2327 2328
		if (dirty)
			err = wait_event_interruptible(ci->i_cap_wq,
				       caps_are_flushed(inode, flush_tid));
	} else {
2329
		struct ceph_mds_client *mdsc =
2330
			ceph_sb_to_client(inode->i_sb)->mdsc;
S
Sage Weil 已提交
2331

2332
		spin_lock(&ci->i_ceph_lock);
S
Sage Weil 已提交
2333 2334
		if (__ceph_caps_dirty(ci))
			__cap_delay_requeue_front(mdsc, ci);
2335
		spin_unlock(&ci->i_ceph_lock);
S
Sage Weil 已提交
2336 2337 2338 2339
	}
	return err;
}

2340 2341 2342 2343 2344 2345
static void __kick_flushing_caps(struct ceph_mds_client *mdsc,
				 struct ceph_mds_session *session,
				 struct ceph_inode_info *ci,
				 u64 oldest_flush_tid)
	__releases(ci->i_ceph_lock)
	__acquires(ci->i_ceph_lock)
2346 2347 2348 2349
{
	struct inode *inode = &ci->vfs_inode;
	struct ceph_cap *cap;
	struct ceph_cap_flush *cf;
2350
	int ret;
2351
	u64 first_tid = 0;
2352
	u64 last_snap_flush = 0;
2353

2354 2355
	ci->i_ceph_flags &= ~CEPH_I_KICK_FLUSH;

2356 2357 2358 2359 2360 2361 2362
	list_for_each_entry_reverse(cf, &ci->i_cap_flush_list, i_list) {
		if (!cf->caps) {
			last_snap_flush = cf->tid;
			break;
		}
	}

2363 2364 2365 2366
	list_for_each_entry(cf, &ci->i_cap_flush_list, i_list) {
		if (cf->tid < first_tid)
			continue;

2367 2368
		cap = ci->i_auth_cap;
		if (!(cap && cap->session == session)) {
2369 2370
			pr_err("%p auth cap %p not mds%d ???\n",
			       inode, cap, session->s_mds);
2371 2372 2373 2374 2375
			break;
		}

		first_tid = cf->tid + 1;

2376
		if (cf->caps) {
2377 2378
			struct cap_msg_args arg;

2379 2380
			dout("kick_flushing_caps %p cap %p tid %llu %s\n",
			     inode, cap, cf->tid, ceph_cap_string(cf->caps));
2381
			__prep_cap(&arg, cap, CEPH_CAP_OP_FLUSH,
2382 2383 2384
					 (cf->tid < last_snap_flush ?
					  CEPH_CLIENT_CAPS_PENDING_CAPSNAP : 0),
					  __ceph_caps_used(ci),
2385
					  __ceph_caps_wanted(ci),
2386
					  (cap->issued | cap->implemented),
2387
					  cf->caps, cf->tid, oldest_flush_tid);
2388 2389
			spin_unlock(&ci->i_ceph_lock);
			__send_cap(mdsc, &arg, ci);
2390 2391 2392 2393 2394 2395 2396 2397
		} else {
			struct ceph_cap_snap *capsnap =
					container_of(cf, struct ceph_cap_snap,
						    cap_flush);
			dout("kick_flushing_caps %p capsnap %p tid %llu %s\n",
			     inode, capsnap, cf->tid,
			     ceph_cap_string(capsnap->dirty));

2398
			refcount_inc(&capsnap->nref);
2399 2400 2401 2402 2403 2404 2405 2406 2407 2408 2409 2410 2411 2412
			spin_unlock(&ci->i_ceph_lock);

			ret = __send_flush_snap(inode, session, capsnap, cap->mseq,
						oldest_flush_tid);
			if (ret < 0) {
				pr_err("kick_flushing_caps: error sending "
					"cap flushsnap, ino (%llx.%llx) "
					"tid %llu follows %llu\n",
					ceph_vinop(inode), cf->tid,
					capsnap->follows);
			}

			ceph_put_cap_snap(capsnap);
		}
2413 2414

		spin_lock(&ci->i_ceph_lock);
2415 2416 2417
	}
}

2418 2419 2420 2421 2422
void ceph_early_kick_flushing_caps(struct ceph_mds_client *mdsc,
				   struct ceph_mds_session *session)
{
	struct ceph_inode_info *ci;
	struct ceph_cap *cap;
2423
	u64 oldest_flush_tid;
2424 2425

	dout("early_kick_flushing_caps mds%d\n", session->s_mds);
2426 2427 2428 2429 2430

	spin_lock(&mdsc->cap_dirty_lock);
	oldest_flush_tid = __get_oldest_flush_tid(mdsc);
	spin_unlock(&mdsc->cap_dirty_lock);

2431 2432 2433 2434 2435 2436 2437 2438 2439 2440 2441 2442 2443 2444 2445 2446 2447 2448 2449
	list_for_each_entry(ci, &session->s_cap_flushing, i_flushing_item) {
		spin_lock(&ci->i_ceph_lock);
		cap = ci->i_auth_cap;
		if (!(cap && cap->session == session)) {
			pr_err("%p auth cap %p not mds%d ???\n",
				&ci->vfs_inode, cap, session->s_mds);
			spin_unlock(&ci->i_ceph_lock);
			continue;
		}


		/*
		 * if flushing caps were revoked, we re-send the cap flush
		 * in client reconnect stage. This guarantees MDS * processes
		 * the cap flush message before issuing the flushing caps to
		 * other client.
		 */
		if ((cap->issued & ci->i_flushing_caps) !=
		    ci->i_flushing_caps) {
2450 2451 2452 2453 2454 2455
			/* encode_caps_cb() also will reset these sequence
			 * numbers. make sure sequence numbers in cap flush
			 * message match later reconnect message */
			cap->seq = 0;
			cap->issue_seq = 0;
			cap->mseq = 0;
2456 2457
			__kick_flushing_caps(mdsc, session, ci,
					     oldest_flush_tid);
2458 2459
		} else {
			ci->i_ceph_flags |= CEPH_I_KICK_FLUSH;
2460 2461 2462 2463 2464 2465
		}

		spin_unlock(&ci->i_ceph_lock);
	}
}

S
Sage Weil 已提交
2466 2467 2468 2469
void ceph_kick_flushing_caps(struct ceph_mds_client *mdsc,
			     struct ceph_mds_session *session)
{
	struct ceph_inode_info *ci;
2470
	struct ceph_cap *cap;
2471
	u64 oldest_flush_tid;
S
Sage Weil 已提交
2472 2473

	dout("kick_flushing_caps mds%d\n", session->s_mds);
2474 2475 2476 2477 2478

	spin_lock(&mdsc->cap_dirty_lock);
	oldest_flush_tid = __get_oldest_flush_tid(mdsc);
	spin_unlock(&mdsc->cap_dirty_lock);

S
Sage Weil 已提交
2479
	list_for_each_entry(ci, &session->s_cap_flushing, i_flushing_item) {
2480
		spin_lock(&ci->i_ceph_lock);
2481 2482 2483 2484 2485 2486 2487 2488 2489 2490 2491
		cap = ci->i_auth_cap;
		if (!(cap && cap->session == session)) {
			pr_err("%p auth cap %p not mds%d ???\n",
				&ci->vfs_inode, cap, session->s_mds);
			spin_unlock(&ci->i_ceph_lock);
			continue;
		}
		if (ci->i_ceph_flags & CEPH_I_KICK_FLUSH) {
			__kick_flushing_caps(mdsc, session, ci,
					     oldest_flush_tid);
		}
2492
		spin_unlock(&ci->i_ceph_lock);
S
Sage Weil 已提交
2493 2494 2495
	}
}

2496 2497
void ceph_kick_flushing_inode_caps(struct ceph_mds_session *session,
				   struct ceph_inode_info *ci)
2498
{
2499 2500 2501 2502
	struct ceph_mds_client *mdsc = session->s_mdsc;
	struct ceph_cap *cap = ci->i_auth_cap;

	lockdep_assert_held(&ci->i_ceph_lock);
2503

2504
	dout("%s %p flushing %s\n", __func__, &ci->vfs_inode,
2505
	     ceph_cap_string(ci->i_flushing_caps));
2506

2507 2508
	if (!list_empty(&ci->i_cap_flush_list)) {
		u64 oldest_flush_tid;
2509 2510 2511
		spin_lock(&mdsc->cap_dirty_lock);
		list_move_tail(&ci->i_flushing_item,
			       &cap->session->s_cap_flushing);
2512
		oldest_flush_tid = __get_oldest_flush_tid(mdsc);
2513 2514
		spin_unlock(&mdsc->cap_dirty_lock);

2515
		__kick_flushing_caps(mdsc, session, ci, oldest_flush_tid);
2516 2517 2518
	}
}

S
Sage Weil 已提交
2519 2520 2521 2522 2523

/*
 * Take references to capabilities we hold, so that we don't release
 * them to the MDS prematurely.
 */
2524
void ceph_take_cap_refs(struct ceph_inode_info *ci, int got,
2525
			    bool snap_rwsem_locked)
S
Sage Weil 已提交
2526
{
2527 2528
	lockdep_assert_held(&ci->i_ceph_lock);

S
Sage Weil 已提交
2529 2530 2531 2532 2533 2534
	if (got & CEPH_CAP_PIN)
		ci->i_pin_ref++;
	if (got & CEPH_CAP_FILE_RD)
		ci->i_rd_ref++;
	if (got & CEPH_CAP_FILE_CACHE)
		ci->i_rdcache_ref++;
J
Jeff Layton 已提交
2535 2536
	if (got & CEPH_CAP_FILE_EXCL)
		ci->i_fx_ref++;
2537 2538 2539 2540 2541 2542
	if (got & CEPH_CAP_FILE_WR) {
		if (ci->i_wr_ref == 0 && !ci->i_head_snapc) {
			BUG_ON(!snap_rwsem_locked);
			ci->i_head_snapc = ceph_get_snap_context(
					ci->i_snap_realm->cached_context);
		}
S
Sage Weil 已提交
2543
		ci->i_wr_ref++;
2544
	}
S
Sage Weil 已提交
2545
	if (got & CEPH_CAP_FILE_BUFFER) {
2546
		if (ci->i_wb_ref == 0)
2547
			ihold(&ci->vfs_inode);
2548
		ci->i_wb_ref++;
2549
		dout("%s %p wb %d -> %d (?)\n", __func__,
2550
		     &ci->vfs_inode, ci->i_wb_ref-1, ci->i_wb_ref);
S
Sage Weil 已提交
2551 2552 2553 2554 2555 2556 2557 2558 2559
	}
}

/*
 * Try to grab cap references.  Specify those refs we @want, and the
 * minimal set we @need.  Also include the larger offset we are writing
 * to (when applicable), and check against max_size here as well.
 * Note that caller is responsible for ensuring max_size increases are
 * requested from the MDS.
2560
 *
2561 2562 2563 2564 2565
 * Returns 0 if caps were not able to be acquired (yet), 1 if succeed,
 * or a negative error code. There are 3 speical error codes:
 *  -EAGAIN: need to sleep but non-blocking is specified
 *  -EFBIG:  ask caller to call check_max_size() and try again.
 *  -ESTALE: ask caller to call ceph_renew_caps() and try again.
S
Sage Weil 已提交
2566
 */
2567
enum {
2568 2569 2570
	/* first 8 bits are reserved for CEPH_FILE_MODE_FOO */
	NON_BLOCKING	= (1 << 8),
	CHECK_FILELOCK	= (1 << 9),
2571 2572
};

Y
Yan, Zheng 已提交
2573
static int try_get_cap_refs(struct inode *inode, int need, int want,
2574
			    loff_t endoff, int flags, int *got)
S
Sage Weil 已提交
2575
{
Y
Yan, Zheng 已提交
2576
	struct ceph_inode_info *ci = ceph_inode(inode);
2577
	struct ceph_mds_client *mdsc = ceph_inode_to_client(inode)->mdsc;
S
Sage Weil 已提交
2578
	int ret = 0;
2579
	int have, implemented;
2580
	bool snap_rwsem_locked = false;
S
Sage Weil 已提交
2581 2582 2583

	dout("get_cap_refs %p need %s want %s\n", inode,
	     ceph_cap_string(need), ceph_cap_string(want));
2584

2585
again:
2586
	spin_lock(&ci->i_ceph_lock);
S
Sage Weil 已提交
2587

2588 2589 2590 2591 2592 2593 2594
	if ((flags & CHECK_FILELOCK) &&
	    (ci->i_ceph_flags & CEPH_I_ERROR_FILELOCK)) {
		dout("try_get_cap_refs %p error filelock\n", inode);
		ret = -EIO;
		goto out_unlock;
	}

2595 2596 2597
	/* finish pending truncate */
	while (ci->i_truncate_pending) {
		spin_unlock(&ci->i_ceph_lock);
2598 2599 2600 2601
		if (snap_rwsem_locked) {
			up_read(&mdsc->snap_rwsem);
			snap_rwsem_locked = false;
		}
Y
Yan, Zheng 已提交
2602
		__ceph_do_pending_vmtruncate(inode);
2603 2604 2605
		spin_lock(&ci->i_ceph_lock);
	}

Y
Yan, Zheng 已提交
2606 2607 2608
	have = __ceph_caps_issued(ci, &implemented);

	if (have & need & CEPH_CAP_FILE_WR) {
S
Sage Weil 已提交
2609 2610 2611
		if (endoff >= 0 && endoff > (loff_t)ci->i_max_size) {
			dout("get_cap_refs %p endoff %llu > maxsize %llu\n",
			     inode, endoff, ci->i_max_size);
2612
			if (endoff > ci->i_requested_max_size)
2613
				ret = ci->i_auth_cap ? -EFBIG : -ESTALE;
2614
			goto out_unlock;
S
Sage Weil 已提交
2615 2616 2617 2618 2619 2620 2621
		}
		/*
		 * If a sync write is in progress, we must wait, so that we
		 * can get a final snapshot value for size+mtime.
		 */
		if (__ceph_have_pending_cap_snap(ci)) {
			dout("get_cap_refs %p cap_snap_pending\n", inode);
2622
			goto out_unlock;
S
Sage Weil 已提交
2623 2624 2625 2626 2627 2628 2629 2630 2631 2632 2633 2634 2635 2636 2637 2638
		}
	}

	if ((have & need) == need) {
		/*
		 * Look at (implemented & ~have & not) so that we keep waiting
		 * on transition from wanted -> needed caps.  This is needed
		 * for WRBUFFER|WR -> WR to avoid a new WR sync write from
		 * going before a prior buffered writeback happens.
		 */
		int not = want & ~(have & need);
		int revoking = implemented & ~have;
		dout("get_cap_refs %p have %s but not %s (revoking %s)\n",
		     inode, ceph_cap_string(have), ceph_cap_string(not),
		     ceph_cap_string(revoking));
		if ((revoking & not) == 0) {
2639 2640 2641 2642 2643 2644 2645 2646
			if (!snap_rwsem_locked &&
			    !ci->i_head_snapc &&
			    (need & CEPH_CAP_FILE_WR)) {
				if (!down_read_trylock(&mdsc->snap_rwsem)) {
					/*
					 * we can not call down_read() when
					 * task isn't in TASK_RUNNING state
					 */
2647
					if (flags & NON_BLOCKING) {
2648
						ret = -EAGAIN;
2649 2650 2651 2652 2653 2654 2655 2656 2657 2658
						goto out_unlock;
					}

					spin_unlock(&ci->i_ceph_lock);
					down_read(&mdsc->snap_rwsem);
					snap_rwsem_locked = true;
					goto again;
				}
				snap_rwsem_locked = true;
			}
2659 2660 2661 2662
			if ((have & want) == want)
				*got = need | want;
			else
				*got = need;
2663 2664
			if (S_ISREG(inode->i_mode) &&
			    (need & CEPH_CAP_FILE_RD) &&
Y
Yan, Zheng 已提交
2665 2666
			    !(*got & CEPH_CAP_FILE_CACHE))
				ceph_disable_fscache_readpage(ci);
2667
			ceph_take_cap_refs(ci, *got, true);
S
Sage Weil 已提交
2668 2669 2670
			ret = 1;
		}
	} else {
2671
		int session_readonly = false;
2672
		int mds_wanted;
2673 2674
		if (ci->i_auth_cap &&
		    (need & (CEPH_CAP_FILE_WR | CEPH_CAP_FILE_EXCL))) {
2675 2676 2677 2678 2679 2680
			struct ceph_mds_session *s = ci->i_auth_cap->session;
			spin_lock(&s->s_cap_lock);
			session_readonly = s->s_readonly;
			spin_unlock(&s->s_cap_lock);
		}
		if (session_readonly) {
2681
			dout("get_cap_refs %p need %s but mds%d readonly\n",
2682
			     inode, ceph_cap_string(need), ci->i_auth_cap->mds);
2683
			ret = -EROFS;
2684 2685 2686
			goto out_unlock;
		}

2687 2688 2689 2690 2691 2692 2693 2694 2695 2696 2697 2698
		if (READ_ONCE(mdsc->fsc->mount_state) == CEPH_MOUNT_SHUTDOWN) {
			dout("get_cap_refs %p forced umount\n", inode);
			ret = -EIO;
			goto out_unlock;
		}
		mds_wanted = __ceph_caps_mds_wanted(ci, false);
		if (need & ~mds_wanted) {
			dout("get_cap_refs %p need %s > mds_wanted %s\n",
			     inode, ceph_cap_string(need),
			     ceph_cap_string(mds_wanted));
			ret = -ESTALE;
			goto out_unlock;
2699 2700
		}

2701
		dout("get_cap_refs %p have %s need %s\n", inode,
S
Sage Weil 已提交
2702 2703
		     ceph_cap_string(have), ceph_cap_string(need));
	}
2704
out_unlock:
2705 2706 2707

	__ceph_touch_fmode(ci, mdsc, flags);

2708
	spin_unlock(&ci->i_ceph_lock);
2709 2710
	if (snap_rwsem_locked)
		up_read(&mdsc->snap_rwsem);
2711

2712 2713 2714 2715 2716
	if (!ret)
		ceph_update_cap_mis(&mdsc->metric);
	else if (ret == 1)
		ceph_update_cap_hit(&mdsc->metric);

S
Sage Weil 已提交
2717
	dout("get_cap_refs %p ret %d got %s\n", inode,
2718
	     ret, ceph_cap_string(*got));
S
Sage Weil 已提交
2719 2720 2721 2722 2723 2724 2725 2726 2727 2728 2729 2730 2731 2732
	return ret;
}

/*
 * Check the offset we are writing up to against our current
 * max_size.  If necessary, tell the MDS we want to write to
 * a larger offset.
 */
static void check_max_size(struct inode *inode, loff_t endoff)
{
	struct ceph_inode_info *ci = ceph_inode(inode);
	int check = 0;

	/* do we need to explicitly request a larger max_size? */
2733
	spin_lock(&ci->i_ceph_lock);
Y
Yan, Zheng 已提交
2734
	if (endoff >= ci->i_max_size && endoff > ci->i_wanted_max_size) {
S
Sage Weil 已提交
2735 2736 2737 2738
		dout("write %p at large endoff %llu, req max_size\n",
		     inode, endoff);
		ci->i_wanted_max_size = endoff;
	}
Y
Yan, Zheng 已提交
2739 2740 2741 2742 2743 2744
	/* duplicate ceph_check_caps()'s logic */
	if (ci->i_auth_cap &&
	    (ci->i_auth_cap->issued & CEPH_CAP_FILE_WR) &&
	    ci->i_wanted_max_size > ci->i_max_size &&
	    ci->i_wanted_max_size > ci->i_requested_max_size)
		check = 1;
2745
	spin_unlock(&ci->i_ceph_lock);
S
Sage Weil 已提交
2746 2747 2748 2749
	if (check)
		ceph_check_caps(ci, CHECK_CAPS_AUTHONLY, NULL);
}

2750 2751 2752 2753 2754 2755 2756 2757 2758 2759
static inline int get_used_fmode(int caps)
{
	int fmode = 0;
	if (caps & CEPH_CAP_FILE_RD)
		fmode |= CEPH_FILE_MODE_RD;
	if (caps & CEPH_CAP_FILE_WR)
		fmode |= CEPH_FILE_MODE_WR;
	return fmode;
}

Y
Yan, Zheng 已提交
2760
int ceph_try_get_caps(struct inode *inode, int need, int want,
2761
		      bool nonblock, int *got)
2762
{
2763
	int ret, flags;
2764 2765

	BUG_ON(need & ~CEPH_CAP_FILE_RD);
2766 2767 2768 2769 2770 2771 2772 2773
	BUG_ON(want & ~(CEPH_CAP_FILE_CACHE | CEPH_CAP_FILE_LAZYIO |
			CEPH_CAP_FILE_SHARED | CEPH_CAP_FILE_EXCL |
			CEPH_CAP_ANY_DIR_OPS));
	if (need) {
		ret = ceph_pool_perm_check(inode, need);
		if (ret < 0)
			return ret;
	}
2774

2775 2776 2777 2778 2779
	flags = get_used_fmode(need | want);
	if (nonblock)
		flags |= NON_BLOCKING;

	ret = try_get_cap_refs(inode, need, want, 0, flags, got);
2780
	/* three special error codes */
2781
	if (ret == -EAGAIN || ret == -EFBIG || ret == -ESTALE)
2782 2783
		ret = 0;
	return ret;
2784 2785
}

S
Sage Weil 已提交
2786 2787 2788 2789 2790
/*
 * Wait for caps, and take cap references.  If we can't get a WR cap
 * due to a small max_size, make sure we check_max_size (and possibly
 * ask the mds) so we don't get hung up indefinitely.
 */
Y
Yan, Zheng 已提交
2791
int ceph_get_caps(struct file *filp, int need, int want,
2792
		  loff_t endoff, int *got, struct page **pinned_page)
S
Sage Weil 已提交
2793
{
2794
	struct ceph_file_info *fi = filp->private_data;
Y
Yan, Zheng 已提交
2795 2796
	struct inode *inode = file_inode(filp);
	struct ceph_inode_info *ci = ceph_inode(inode);
2797
	struct ceph_fs_client *fsc = ceph_inode_to_client(inode);
2798
	int ret, _got, flags;
S
Sage Weil 已提交
2799

Y
Yan, Zheng 已提交
2800
	ret = ceph_pool_perm_check(inode, need);
2801 2802 2803
	if (ret < 0)
		return ret;

2804 2805 2806 2807
	if ((fi->fmode & CEPH_FILE_MODE_WR) &&
	    fi->filp_gen != READ_ONCE(fsc->filp_gen))
		return -EBADF;

2808 2809
	flags = get_used_fmode(need | want);

2810
	while (true) {
2811 2812 2813
		flags &= CEPH_FILE_MODE_MASK;
		if (atomic_read(&fi->num_locks))
			flags |= CHECK_FILELOCK;
2814
		_got = 0;
Y
Yan, Zheng 已提交
2815
		ret = try_get_cap_refs(inode, need, want, endoff,
2816
				       flags, &_got);
2817
		WARN_ON_ONCE(ret == -EAGAIN);
2818
		if (!ret) {
2819 2820
			struct ceph_mds_client *mdsc = fsc->mdsc;
			struct cap_wait cw;
2821
			DEFINE_WAIT_FUNC(wait, woken_wake_function);
2822 2823 2824 2825 2826 2827 2828 2829 2830 2831

			cw.ino = inode->i_ino;
			cw.tgid = current->tgid;
			cw.need = need;
			cw.want = want;

			spin_lock(&mdsc->caps_list_lock);
			list_add(&cw.list, &mdsc->cap_wait_list);
			spin_unlock(&mdsc->caps_list_lock);

2832 2833
			/* make sure used fmode not timeout */
			ceph_get_fmode(ci, flags, FMODE_WAIT_BIAS);
2834 2835
			add_wait_queue(&ci->i_cap_wq, &wait);

2836
			flags |= NON_BLOCKING;
Y
Yan, Zheng 已提交
2837
			while (!(ret = try_get_cap_refs(inode, need, want,
2838
							endoff, flags, &_got))) {
2839 2840 2841 2842
				if (signal_pending(current)) {
					ret = -ERESTARTSYS;
					break;
				}
2843
				wait_woken(&wait, TASK_INTERRUPTIBLE, MAX_SCHEDULE_TIMEOUT);
2844
			}
2845 2846

			remove_wait_queue(&ci->i_cap_wq, &wait);
2847
			ceph_put_fmode(ci, flags, FMODE_WAIT_BIAS);
2848 2849 2850 2851 2852

			spin_lock(&mdsc->caps_list_lock);
			list_del(&cw.list);
			spin_unlock(&mdsc->caps_list_lock);

2853
			if (ret == -EAGAIN)
2854
				continue;
2855
		}
2856 2857 2858 2859 2860 2861 2862 2863

		if ((fi->fmode & CEPH_FILE_MODE_WR) &&
		    fi->filp_gen != READ_ONCE(fsc->filp_gen)) {
			if (ret >= 0 && _got)
				ceph_put_cap_refs(ci, _got);
			return -EBADF;
		}

2864
		if (ret < 0) {
2865 2866 2867 2868 2869
			if (ret == -EFBIG || ret == -ESTALE) {
				int ret2 = ceph_wait_on_async_create(inode);
				if (ret2 < 0)
					return ret2;
			}
2870 2871 2872 2873
			if (ret == -EFBIG) {
				check_max_size(inode, endoff);
				continue;
			}
2874 2875
			if (ret == -ESTALE) {
				/* session was killed, try renew caps */
2876
				ret = ceph_renew_caps(inode, flags);
2877 2878 2879
				if (ret == 0)
					continue;
			}
2880
			return ret;
2881
		}
2882

2883 2884
		if (S_ISREG(ci->vfs_inode.i_mode) &&
		    ci->i_inline_version != CEPH_INLINE_NONE &&
2885
		    (_got & (CEPH_CAP_FILE_CACHE|CEPH_CAP_FILE_LAZYIO)) &&
Y
Yan, Zheng 已提交
2886
		    i_size_read(inode) > 0) {
2887
			struct page *page =
Y
Yan, Zheng 已提交
2888
				find_get_page(inode->i_mapping, 0);
2889 2890 2891 2892 2893
			if (page) {
				if (PageUptodate(page)) {
					*pinned_page = page;
					break;
				}
2894
				put_page(page);
2895
			}
2896 2897 2898 2899 2900 2901
			/*
			 * drop cap refs first because getattr while
			 * holding * caps refs can cause deadlock.
			 */
			ceph_put_cap_refs(ci, _got);
			_got = 0;
2902

2903 2904 2905 2906
			/*
			 * getattr request will bring inline data into
			 * page cache
			 */
Y
Yan, Zheng 已提交
2907
			ret = __ceph_do_getattr(inode, NULL,
2908 2909 2910 2911 2912 2913 2914
						CEPH_STAT_CAP_INLINE_DATA,
						true);
			if (ret < 0)
				return ret;
			continue;
		}
		break;
2915
	}
2916

2917 2918
	if (S_ISREG(ci->vfs_inode.i_mode) &&
	    (_got & CEPH_CAP_FILE_RD) && (_got & CEPH_CAP_FILE_CACHE))
Y
Yan, Zheng 已提交
2919 2920
		ceph_fscache_revalidate_cookie(ci);

2921 2922
	*got = _got;
	return 0;
S
Sage Weil 已提交
2923 2924 2925 2926 2927 2928 2929 2930
}

/*
 * Take cap refs.  Caller must already know we hold at least one ref
 * on the caps in question or we don't know this is safe.
 */
void ceph_get_cap_refs(struct ceph_inode_info *ci, int caps)
{
2931
	spin_lock(&ci->i_ceph_lock);
2932
	ceph_take_cap_refs(ci, caps, false);
2933
	spin_unlock(&ci->i_ceph_lock);
S
Sage Weil 已提交
2934 2935
}

2936 2937 2938 2939 2940

/*
 * drop cap_snap that is not associated with any snapshot.
 * we don't need to send FLUSHSNAP message for it.
 */
2941 2942
static int ceph_try_drop_cap_snap(struct ceph_inode_info *ci,
				  struct ceph_cap_snap *capsnap)
2943 2944 2945 2946 2947
{
	if (!capsnap->need_flush &&
	    !capsnap->writing && !capsnap->dirty_pages) {
		dout("dropping cap_snap %p follows %llu\n",
		     capsnap, capsnap->follows);
2948
		BUG_ON(capsnap->cap_flush.tid > 0);
2949
		ceph_put_snap_context(capsnap->context);
2950 2951 2952
		if (!list_is_last(&capsnap->ci_item, &ci->i_cap_snaps))
			ci->i_ceph_flags |= CEPH_I_FLUSH_SNAPS;

2953 2954 2955 2956 2957 2958 2959
		list_del(&capsnap->ci_item);
		ceph_put_cap_snap(capsnap);
		return 1;
	}
	return 0;
}

S
Sage Weil 已提交
2960 2961 2962 2963 2964 2965 2966 2967 2968 2969 2970 2971 2972 2973
/*
 * Release cap refs.
 *
 * If we released the last ref on any given cap, call ceph_check_caps
 * to release (or schedule a release).
 *
 * If we are releasing a WR cap (from a sync write), finalize any affected
 * cap_snap, and wake up any waiters.
 */
void ceph_put_cap_refs(struct ceph_inode_info *ci, int had)
{
	struct inode *inode = &ci->vfs_inode;
	int last = 0, put = 0, flushsnaps = 0, wake = 0;

2974
	spin_lock(&ci->i_ceph_lock);
S
Sage Weil 已提交
2975 2976 2977 2978 2979 2980 2981 2982
	if (had & CEPH_CAP_PIN)
		--ci->i_pin_ref;
	if (had & CEPH_CAP_FILE_RD)
		if (--ci->i_rd_ref == 0)
			last++;
	if (had & CEPH_CAP_FILE_CACHE)
		if (--ci->i_rdcache_ref == 0)
			last++;
J
Jeff Layton 已提交
2983 2984 2985
	if (had & CEPH_CAP_FILE_EXCL)
		if (--ci->i_fx_ref == 0)
			last++;
S
Sage Weil 已提交
2986
	if (had & CEPH_CAP_FILE_BUFFER) {
2987
		if (--ci->i_wb_ref == 0) {
S
Sage Weil 已提交
2988 2989 2990
			last++;
			put++;
		}
2991 2992
		dout("put_cap_refs %p wb %d -> %d (?)\n",
		     inode, ci->i_wb_ref+1, ci->i_wb_ref);
S
Sage Weil 已提交
2993 2994 2995 2996
	}
	if (had & CEPH_CAP_FILE_WR)
		if (--ci->i_wr_ref == 0) {
			last++;
2997 2998 2999 3000 3001 3002
			if (__ceph_have_pending_cap_snap(ci)) {
				struct ceph_cap_snap *capsnap =
					list_last_entry(&ci->i_cap_snaps,
							struct ceph_cap_snap,
							ci_item);
				capsnap->writing = 0;
3003
				if (ceph_try_drop_cap_snap(ci, capsnap))
3004 3005 3006 3007
					put++;
				else if (__ceph_finish_cap_snap(ci, capsnap))
					flushsnaps = 1;
				wake = 1;
S
Sage Weil 已提交
3008
			}
3009 3010 3011 3012 3013 3014 3015
			if (ci->i_wrbuffer_ref_head == 0 &&
			    ci->i_dirty_caps == 0 &&
			    ci->i_flushing_caps == 0) {
				BUG_ON(!ci->i_head_snapc);
				ceph_put_snap_context(ci->i_head_snapc);
				ci->i_head_snapc = NULL;
			}
3016
			/* see comment in __ceph_remove_cap() */
X
Xiubo Li 已提交
3017
			if (!__ceph_is_any_real_caps(ci) && ci->i_snap_realm)
3018
				drop_inode_snap_realm(ci);
S
Sage Weil 已提交
3019
		}
3020
	spin_unlock(&ci->i_ceph_lock);
S
Sage Weil 已提交
3021

3022 3023
	dout("put_cap_refs %p had %s%s%s\n", inode, ceph_cap_string(had),
	     last ? " last" : "", put ? " put" : "");
S
Sage Weil 已提交
3024

3025
	if (last)
S
Sage Weil 已提交
3026 3027
		ceph_check_caps(ci, 0, NULL);
	else if (flushsnaps)
Y
Yan, Zheng 已提交
3028
		ceph_flush_snaps(ci, NULL);
S
Sage Weil 已提交
3029
	if (wake)
3030
		wake_up_all(&ci->i_cap_wq);
3031
	while (put-- > 0)
S
Sage Weil 已提交
3032 3033 3034 3035 3036 3037 3038 3039 3040 3041 3042 3043 3044 3045 3046
		iput(inode);
}

/*
 * Release @nr WRBUFFER refs on dirty pages for the given @snapc snap
 * context.  Adjust per-snap dirty page accounting as appropriate.
 * Once all dirty data for a cap_snap is flushed, flush snapped file
 * metadata back to the MDS.  If we dropped the last ref, call
 * ceph_check_caps.
 */
void ceph_put_wrbuffer_cap_refs(struct ceph_inode_info *ci, int nr,
				struct ceph_snap_context *snapc)
{
	struct inode *inode = &ci->vfs_inode;
	struct ceph_cap_snap *capsnap = NULL;
3047 3048 3049 3050 3051
	int put = 0;
	bool last = false;
	bool found = false;
	bool flush_snaps = false;
	bool complete_capsnap = false;
S
Sage Weil 已提交
3052

3053
	spin_lock(&ci->i_ceph_lock);
S
Sage Weil 已提交
3054
	ci->i_wrbuffer_ref -= nr;
3055 3056 3057 3058
	if (ci->i_wrbuffer_ref == 0) {
		last = true;
		put++;
	}
S
Sage Weil 已提交
3059 3060 3061

	if (ci->i_head_snapc == snapc) {
		ci->i_wrbuffer_ref_head -= nr;
3062
		if (ci->i_wrbuffer_ref_head == 0 &&
3063 3064 3065
		    ci->i_wr_ref == 0 &&
		    ci->i_dirty_caps == 0 &&
		    ci->i_flushing_caps == 0) {
3066
			BUG_ON(!ci->i_head_snapc);
S
Sage Weil 已提交
3067 3068 3069 3070 3071 3072 3073 3074 3075 3076 3077
			ceph_put_snap_context(ci->i_head_snapc);
			ci->i_head_snapc = NULL;
		}
		dout("put_wrbuffer_cap_refs on %p head %d/%d -> %d/%d %s\n",
		     inode,
		     ci->i_wrbuffer_ref+nr, ci->i_wrbuffer_ref_head+nr,
		     ci->i_wrbuffer_ref, ci->i_wrbuffer_ref_head,
		     last ? " LAST" : "");
	} else {
		list_for_each_entry(capsnap, &ci->i_cap_snaps, ci_item) {
			if (capsnap->context == snapc) {
3078
				found = true;
S
Sage Weil 已提交
3079 3080 3081 3082
				break;
			}
		}
		BUG_ON(!found);
3083 3084
		capsnap->dirty_pages -= nr;
		if (capsnap->dirty_pages == 0) {
3085 3086 3087 3088 3089 3090 3091 3092 3093
			complete_capsnap = true;
			if (!capsnap->writing) {
				if (ceph_try_drop_cap_snap(ci, capsnap)) {
					put++;
				} else {
					ci->i_ceph_flags |= CEPH_I_FLUSH_SNAPS;
					flush_snaps = true;
				}
			}
3094
		}
S
Sage Weil 已提交
3095
		dout("put_wrbuffer_cap_refs on %p cap_snap %p "
3096
		     " snap %lld %d/%d -> %d/%d %s%s\n",
S
Sage Weil 已提交
3097 3098 3099 3100
		     inode, capsnap, capsnap->context->seq,
		     ci->i_wrbuffer_ref+nr, capsnap->dirty_pages + nr,
		     ci->i_wrbuffer_ref, capsnap->dirty_pages,
		     last ? " (wrbuffer last)" : "",
3101
		     complete_capsnap ? " (complete capsnap)" : "");
S
Sage Weil 已提交
3102 3103
	}

3104
	spin_unlock(&ci->i_ceph_lock);
S
Sage Weil 已提交
3105 3106

	if (last) {
3107
		ceph_check_caps(ci, 0, NULL);
3108
	} else if (flush_snaps) {
Y
Yan, Zheng 已提交
3109
		ceph_flush_snaps(ci, NULL);
S
Sage Weil 已提交
3110
	}
3111 3112
	if (complete_capsnap)
		wake_up_all(&ci->i_cap_wq);
3113 3114 3115 3116
	while (put-- > 0) {
		/* avoid calling iput_final() in osd dispatch threads */
		ceph_async_iput(inode);
	}
S
Sage Weil 已提交
3117 3118
}

Y
Yan, Zheng 已提交
3119 3120 3121 3122 3123 3124 3125 3126 3127 3128 3129
/*
 * Invalidate unlinked inode's aliases, so we can drop the inode ASAP.
 */
static void invalidate_aliases(struct inode *inode)
{
	struct dentry *dn, *prev = NULL;

	dout("invalidate_aliases inode %p\n", inode);
	d_prune_aliases(inode);
	/*
	 * For non-directory inode, d_find_alias() only returns
J
J. Bruce Fields 已提交
3130 3131
	 * hashed dentry. After calling d_invalidate(), the
	 * dentry becomes unhashed.
Y
Yan, Zheng 已提交
3132
	 *
3133
	 * For directory inode, d_find_alias() can return
J
J. Bruce Fields 已提交
3134
	 * unhashed dentry. But directory inode should have
Y
Yan, Zheng 已提交
3135 3136 3137 3138 3139 3140 3141
	 * one alias at most.
	 */
	while ((dn = d_find_alias(inode))) {
		if (dn == prev) {
			dput(dn);
			break;
		}
3142
		d_invalidate(dn);
Y
Yan, Zheng 已提交
3143 3144 3145 3146 3147 3148 3149 3150
		if (prev)
			dput(prev);
		prev = dn;
	}
	if (prev)
		dput(prev);
}

3151 3152 3153 3154 3155 3156
struct cap_extra_info {
	struct ceph_string *pool_ns;
	/* inline data */
	u64 inline_version;
	void *inline_data;
	u32 inline_len;
3157 3158 3159 3160
	/* dirstat */
	bool dirstat_valid;
	u64 nfiles;
	u64 nsubdirs;
3161
	u64 change_attr;
3162 3163
	/* currently issued */
	int issued;
J
Jeff Layton 已提交
3164
	struct timespec64 btime;
3165 3166
};

S
Sage Weil 已提交
3167 3168 3169 3170
/*
 * Handle a cap GRANT message from the MDS.  (Note that a GRANT may
 * actually be a revocation if it specifies a smaller cap set.)
 *
3171
 * caller holds s_mutex and i_ceph_lock, we drop both.
S
Sage Weil 已提交
3172
 */
3173
static void handle_cap_grant(struct inode *inode,
3174
			     struct ceph_mds_session *session,
3175 3176 3177 3178
			     struct ceph_cap *cap,
			     struct ceph_mds_caps *grant,
			     struct ceph_buffer *xattr_buf,
			     struct cap_extra_info *extra_info)
Y
Yan, Zheng 已提交
3179
	__releases(ci->i_ceph_lock)
3180
	__releases(session->s_mdsc->snap_rwsem)
S
Sage Weil 已提交
3181 3182
{
	struct ceph_inode_info *ci = ceph_inode(inode);
3183
	int seq = le32_to_cpu(grant->seq);
S
Sage Weil 已提交
3184
	int newcaps = le32_to_cpu(grant->caps);
Y
Yan, Zheng 已提交
3185
	int used, wanted, dirty;
S
Sage Weil 已提交
3186 3187
	u64 size = le64_to_cpu(grant->size);
	u64 max_size = le64_to_cpu(grant->max_size);
3188 3189
	unsigned char check_caps = 0;
	bool was_stale = cap->cap_gen < session->s_cap_gen;
F
Fabian Frederick 已提交
3190 3191 3192 3193 3194
	bool wake = false;
	bool writeback = false;
	bool queue_trunc = false;
	bool queue_invalidate = false;
	bool deleted_inode = false;
Y
Yan, Zheng 已提交
3195
	bool fill_inline = false;
S
Sage Weil 已提交
3196

3197
	dout("handle_cap_grant inode %p cap %p mds%d seq %d %s\n",
3198
	     inode, cap, session->s_mds, seq, ceph_cap_string(newcaps));
S
Sage Weil 已提交
3199 3200 3201
	dout(" size %llu max_size %llu, i_size %llu\n", size, max_size,
		inode->i_size);

3202

S
Sage Weil 已提交
3203 3204 3205 3206 3207
	/*
	 * If CACHE is being revoked, and we have no dirty buffers,
	 * try to invalidate (once).  (If there are dirty buffers, we
	 * will invalidate _after_ writeback.)
	 */
3208
	if (S_ISREG(inode->i_mode) && /* don't invalidate readdir cache */
Y
Yan, Zheng 已提交
3209
	    ((cap->issued & ~newcaps) & CEPH_CAP_FILE_CACHE) &&
3210
	    (newcaps & CEPH_CAP_FILE_LAZYIO) == 0 &&
3211
	    !(ci->i_wrbuffer_ref || ci->i_wb_ref)) {
3212
		if (try_nonblocking_invalidate(inode)) {
S
Sage Weil 已提交
3213 3214 3215
			/* there were locked pages.. invalidate later
			   in a separate thread. */
			if (ci->i_rdcache_revoking != ci->i_rdcache_gen) {
F
Fabian Frederick 已提交
3216
				queue_invalidate = true;
S
Sage Weil 已提交
3217 3218 3219 3220 3221
				ci->i_rdcache_revoking = ci->i_rdcache_gen;
			}
		}
	}

3222 3223 3224 3225 3226 3227 3228 3229 3230 3231 3232 3233 3234 3235 3236 3237 3238 3239
	if (was_stale)
		cap->issued = cap->implemented = CEPH_CAP_PIN;

	/*
	 * auth mds of the inode changed. we received the cap export message,
	 * but still haven't received the cap import message. handle_cap_export
	 * updated the new auth MDS' cap.
	 *
	 * "ceph_seq_cmp(seq, cap->seq) <= 0" means we are processing a message
	 * that was sent before the cap import message. So don't remove caps.
	 */
	if (ceph_seq_cmp(seq, cap->seq) <= 0) {
		WARN_ON(cap != ci->i_auth_cap);
		WARN_ON(cap->cap_id != le64_to_cpu(grant->cap_id));
		seq = cap->seq;
		newcaps |= cap->issued;
	}

S
Sage Weil 已提交
3240
	/* side effects now are allowed */
3241
	cap->cap_gen = session->s_cap_gen;
3242
	cap->seq = seq;
S
Sage Weil 已提交
3243 3244 3245

	__check_cap_issue(ci, cap, newcaps);

3246 3247
	inode_set_max_iversion_raw(inode, extra_info->change_attr);

3248
	if ((newcaps & CEPH_CAP_AUTH_SHARED) &&
3249
	    (extra_info->issued & CEPH_CAP_AUTH_EXCL) == 0) {
S
Sage Weil 已提交
3250
		inode->i_mode = le32_to_cpu(grant->mode);
3251 3252
		inode->i_uid = make_kuid(&init_user_ns, le32_to_cpu(grant->uid));
		inode->i_gid = make_kgid(&init_user_ns, le32_to_cpu(grant->gid));
J
Jeff Layton 已提交
3253
		ci->i_btime = extra_info->btime;
S
Sage Weil 已提交
3254
		dout("%p mode 0%o uid.gid %d.%d\n", inode, inode->i_mode,
3255 3256
		     from_kuid(&init_user_ns, inode->i_uid),
		     from_kgid(&init_user_ns, inode->i_gid));
S
Sage Weil 已提交
3257 3258
	}

3259
	if ((newcaps & CEPH_CAP_LINK_SHARED) &&
3260
	    (extra_info->issued & CEPH_CAP_LINK_EXCL) == 0) {
M
Miklos Szeredi 已提交
3261
		set_nlink(inode, le32_to_cpu(grant->nlink));
Y
Yan, Zheng 已提交
3262 3263
		if (inode->i_nlink == 0 &&
		    (newcaps & (CEPH_CAP_LINK_SHARED | CEPH_CAP_LINK_EXCL)))
F
Fabian Frederick 已提交
3264
			deleted_inode = true;
Y
Yan, Zheng 已提交
3265
	}
S
Sage Weil 已提交
3266

3267 3268
	if ((extra_info->issued & CEPH_CAP_XATTR_EXCL) == 0 &&
	    grant->xattr_len) {
S
Sage Weil 已提交
3269 3270 3271 3272 3273 3274 3275 3276 3277 3278
		int len = le32_to_cpu(grant->xattr_len);
		u64 version = le64_to_cpu(grant->xattr_version);

		if (version > ci->i_xattrs.version) {
			dout(" got new xattrs v%llu on %p len %d\n",
			     version, inode, len);
			if (ci->i_xattrs.blob)
				ceph_buffer_put(ci->i_xattrs.blob);
			ci->i_xattrs.blob = ceph_buffer_get(xattr_buf);
			ci->i_xattrs.version = version;
G
Guangliang Zhao 已提交
3279
			ceph_forget_all_cached_acls(inode);
Y
Yan, Zheng 已提交
3280
			ceph_security_invalidate_secctx(inode);
S
Sage Weil 已提交
3281 3282 3283
		}
	}

3284
	if (newcaps & CEPH_CAP_ANY_RD) {
3285
		struct timespec64 mtime, atime, ctime;
3286
		/* ctime/mtime/atime? */
3287 3288 3289
		ceph_decode_timespec64(&mtime, &grant->mtime);
		ceph_decode_timespec64(&atime, &grant->atime);
		ceph_decode_timespec64(&ctime, &grant->ctime);
3290
		ceph_fill_file_time(inode, extra_info->issued,
3291 3292 3293 3294
				    le32_to_cpu(grant->time_warp_seq),
				    &ctime, &mtime, &atime);
	}

3295 3296 3297 3298 3299
	if ((newcaps & CEPH_CAP_FILE_SHARED) && extra_info->dirstat_valid) {
		ci->i_files = extra_info->nfiles;
		ci->i_subdirs = extra_info->nsubdirs;
	}

3300 3301
	if (newcaps & (CEPH_CAP_ANY_FILE_RD | CEPH_CAP_ANY_FILE_WR)) {
		/* file layout may have changed */
3302
		s64 old_pool = ci->i_layout.pool_id;
Y
Yan, Zheng 已提交
3303 3304
		struct ceph_string *old_ns;

3305
		ceph_file_layout_from_legacy(&ci->i_layout, &grant->layout);
Y
Yan, Zheng 已提交
3306 3307
		old_ns = rcu_dereference_protected(ci->i_layout.pool_ns,
					lockdep_is_held(&ci->i_ceph_lock));
3308
		rcu_assign_pointer(ci->i_layout.pool_ns, extra_info->pool_ns);
Y
Yan, Zheng 已提交
3309

3310 3311
		if (ci->i_layout.pool_id != old_pool ||
		    extra_info->pool_ns != old_ns)
3312
			ci->i_ceph_flags &= ~CEPH_I_POOL_PERM;
3313

3314
		extra_info->pool_ns = old_ns;
Y
Yan, Zheng 已提交
3315

3316
		/* size/truncate_seq? */
3317
		queue_trunc = ceph_fill_file_size(inode, extra_info->issued,
3318 3319 3320
					le32_to_cpu(grant->truncate_seq),
					le64_to_cpu(grant->truncate_size),
					size);
3321 3322 3323 3324
	}

	if (ci->i_auth_cap == cap && (newcaps & CEPH_CAP_ANY_FILE_WR)) {
		if (max_size != ci->i_max_size) {
3325 3326 3327 3328 3329 3330 3331
			dout("max_size %lld -> %llu\n",
			     ci->i_max_size, max_size);
			ci->i_max_size = max_size;
			if (max_size >= ci->i_wanted_max_size) {
				ci->i_wanted_max_size = 0;  /* reset */
				ci->i_requested_max_size = 0;
			}
F
Fabian Frederick 已提交
3332
			wake = true;
3333 3334 3335 3336
		} else if (ci->i_wanted_max_size > ci->i_max_size &&
			   ci->i_wanted_max_size > ci->i_requested_max_size) {
			/* CEPH_CAP_OP_IMPORT */
			wake = true;
S
Sage Weil 已提交
3337 3338 3339 3340 3341 3342 3343 3344 3345 3346 3347
		}
	}

	/* check cap bits */
	wanted = __ceph_caps_wanted(ci);
	used = __ceph_caps_used(ci);
	dirty = __ceph_caps_dirty(ci);
	dout(" my wanted = %s, used = %s, dirty %s\n",
	     ceph_cap_string(wanted),
	     ceph_cap_string(used),
	     ceph_cap_string(dirty));
3348 3349 3350 3351 3352 3353 3354 3355 3356 3357 3358 3359 3360 3361

	if ((was_stale || le32_to_cpu(grant->op) == CEPH_CAP_OP_IMPORT) &&
	    (wanted & ~(cap->mds_wanted | newcaps))) {
		/*
		 * If mds is importing cap, prior cap messages that update
		 * 'wanted' may get dropped by mds (migrate seq mismatch).
		 *
		 * We don't send cap message to update 'wanted' if what we
		 * want are already issued. If mds revokes caps, cap message
		 * that releases caps also tells mds what we want. But if
		 * caps got revoked by mds forcedly (session stale). We may
		 * haven't told mds what we want.
		 */
		check_caps = 1;
S
Sage Weil 已提交
3362 3363 3364 3365
	}

	/* revocation, grant, or no-op? */
	if (cap->issued & ~newcaps) {
3366 3367 3368 3369 3370 3371
		int revoking = cap->issued & ~newcaps;

		dout("revocation: %s -> %s (revoking %s)\n",
		     ceph_cap_string(cap->issued),
		     ceph_cap_string(newcaps),
		     ceph_cap_string(revoking));
3372 3373
		if (S_ISREG(inode->i_mode) &&
		    (revoking & used & CEPH_CAP_FILE_BUFFER))
F
Fabian Frederick 已提交
3374
			writeback = true;  /* initiate writeback; will delay ack */
3375 3376 3377
		else if (queue_invalidate &&
			 revoking == CEPH_CAP_FILE_CACHE &&
			 (newcaps & CEPH_CAP_FILE_LAZYIO) == 0)
3378 3379 3380 3381 3382
			; /* do nothing yet, invalidation will be queued */
		else if (cap == ci->i_auth_cap)
			check_caps = 1; /* check auth cap only */
		else
			check_caps = 2; /* check all caps */
S
Sage Weil 已提交
3383
		cap->issued = newcaps;
3384
		cap->implemented |= newcaps;
S
Sage Weil 已提交
3385 3386 3387 3388 3389 3390
	} else if (cap->issued == newcaps) {
		dout("caps unchanged: %s -> %s\n",
		     ceph_cap_string(cap->issued), ceph_cap_string(newcaps));
	} else {
		dout("grant: %s -> %s\n", ceph_cap_string(cap->issued),
		     ceph_cap_string(newcaps));
3391 3392 3393 3394 3395
		/* non-auth MDS is revoking the newly grant caps ? */
		if (cap == ci->i_auth_cap &&
		    __ceph_caps_revoking_other(ci, cap, newcaps))
		    check_caps = 2;

S
Sage Weil 已提交
3396 3397 3398 3399
		cap->issued = newcaps;
		cap->implemented |= newcaps; /* add bits only, to
					      * avoid stepping on a
					      * pending revocation */
F
Fabian Frederick 已提交
3400
		wake = true;
S
Sage Weil 已提交
3401
	}
3402
	BUG_ON(cap->issued & ~cap->implemented);
S
Sage Weil 已提交
3403

3404 3405 3406
	if (extra_info->inline_version > 0 &&
	    extra_info->inline_version >= ci->i_inline_version) {
		ci->i_inline_version = extra_info->inline_version;
Y
Yan, Zheng 已提交
3407 3408 3409 3410 3411
		if (ci->i_inline_version != CEPH_INLINE_NONE &&
		    (newcaps & (CEPH_CAP_FILE_CACHE|CEPH_CAP_FILE_LAZYIO)))
			fill_inline = true;
	}

Y
Yan, Zheng 已提交
3412
	if (le32_to_cpu(grant->op) == CEPH_CAP_OP_IMPORT) {
3413
		if (newcaps & ~extra_info->issued)
F
Fabian Frederick 已提交
3414
			wake = true;
3415 3416
		ceph_kick_flushing_inode_caps(session, ci);
		spin_unlock(&ci->i_ceph_lock);
3417
		up_read(&session->s_mdsc->snap_rwsem);
3418 3419
	} else {
		spin_unlock(&ci->i_ceph_lock);
Y
Yan, Zheng 已提交
3420 3421
	}

Y
Yan, Zheng 已提交
3422
	if (fill_inline)
3423 3424
		ceph_fill_inline_data(inode, NULL, extra_info->inline_data,
				      extra_info->inline_len);
Y
Yan, Zheng 已提交
3425

3426
	if (queue_trunc)
3427 3428
		ceph_queue_vmtruncate(inode);

3429
	if (writeback)
S
Sage Weil 已提交
3430 3431 3432 3433 3434
		/*
		 * queue inode for writeback: we can't actually call
		 * filemap_write_and_wait, etc. from message handler
		 * context.
		 */
3435 3436 3437
		ceph_queue_writeback(inode);
	if (queue_invalidate)
		ceph_queue_invalidate(inode);
Y
Yan, Zheng 已提交
3438 3439
	if (deleted_inode)
		invalidate_aliases(inode);
S
Sage Weil 已提交
3440
	if (wake)
3441
		wake_up_all(&ci->i_cap_wq);
3442 3443

	if (check_caps == 1)
3444
		ceph_check_caps(ci, CHECK_CAPS_AUTHONLY | CHECK_CAPS_NOINVAL,
3445 3446
				session);
	else if (check_caps == 2)
3447
		ceph_check_caps(ci, CHECK_CAPS_NOINVAL, session);
3448 3449
	else
		mutex_unlock(&session->s_mutex);
S
Sage Weil 已提交
3450 3451 3452 3453 3454 3455
}

/*
 * Handle FLUSH_ACK from MDS, indicating that metadata we sent to the
 * MDS has been safely committed.
 */
3456
static void handle_cap_flush_ack(struct inode *inode, u64 flush_tid,
S
Sage Weil 已提交
3457 3458 3459
				 struct ceph_mds_caps *m,
				 struct ceph_mds_session *session,
				 struct ceph_cap *cap)
3460
	__releases(ci->i_ceph_lock)
S
Sage Weil 已提交
3461 3462
{
	struct ceph_inode_info *ci = ceph_inode(inode);
3463
	struct ceph_mds_client *mdsc = ceph_sb_to_client(inode->i_sb)->mdsc;
3464
	struct ceph_cap_flush *cf, *tmp_cf;
3465
	LIST_HEAD(to_remove);
S
Sage Weil 已提交
3466 3467 3468
	unsigned seq = le32_to_cpu(m->seq);
	int dirty = le32_to_cpu(m->dirty);
	int cleaned = 0;
Y
Yan, Zheng 已提交
3469
	bool drop = false;
3470 3471
	bool wake_ci = false;
	bool wake_mdsc = false;
S
Sage Weil 已提交
3472

3473
	list_for_each_entry_safe(cf, tmp_cf, &ci->i_cap_flush_list, i_list) {
3474
		/* Is this the one that was flushed? */
3475 3476
		if (cf->tid == flush_tid)
			cleaned = cf->caps;
3477 3478 3479

		/* Is this a capsnap? */
		if (cf->caps == 0)
3480
			continue;
3481

3482
		if (cf->tid <= flush_tid) {
3483 3484 3485 3486
			/*
			 * An earlier or current tid. The FLUSH_ACK should
			 * represent a superset of this flush's caps.
			 */
J
Jeff Layton 已提交
3487
			wake_ci |= __detach_cap_flush_from_ci(ci, cf);
3488
			list_add_tail(&cf->i_list, &to_remove);
3489
		} else {
3490 3491 3492 3493
			/*
			 * This is a later one. Any caps in it are still dirty
			 * so don't count them as cleaned.
			 */
3494 3495 3496 3497 3498
			cleaned &= ~cf->caps;
			if (!cleaned)
				break;
		}
	}
S
Sage Weil 已提交
3499 3500 3501 3502 3503 3504 3505

	dout("handle_cap_flush_ack inode %p mds%d seq %d on %s cleaned %s,"
	     " flushing %s -> %s\n",
	     inode, session->s_mds, seq, ceph_cap_string(dirty),
	     ceph_cap_string(cleaned), ceph_cap_string(ci->i_flushing_caps),
	     ceph_cap_string(ci->i_flushing_caps & ~cleaned));

3506
	if (list_empty(&to_remove) && !cleaned)
S
Sage Weil 已提交
3507 3508 3509 3510 3511
		goto out;

	ci->i_flushing_caps &= ~cleaned;

	spin_lock(&mdsc->cap_dirty_lock);
3512

J
Jeff Layton 已提交
3513 3514
	list_for_each_entry(cf, &to_remove, i_list)
		wake_mdsc |= __detach_cap_flush_from_mdsc(mdsc, cf);
3515

S
Sage Weil 已提交
3516
	if (ci->i_flushing_caps == 0) {
3517 3518 3519 3520 3521 3522 3523 3524 3525 3526
		if (list_empty(&ci->i_cap_flush_list)) {
			list_del_init(&ci->i_flushing_item);
			if (!list_empty(&session->s_cap_flushing)) {
				dout(" mds%d still flushing cap on %p\n",
				     session->s_mds,
				     &list_first_entry(&session->s_cap_flushing,
						struct ceph_inode_info,
						i_flushing_item)->vfs_inode);
			}
		}
S
Sage Weil 已提交
3527 3528
		mdsc->num_cap_flushing--;
		dout(" inode %p now !flushing\n", inode);
3529 3530 3531 3532

		if (ci->i_dirty_caps == 0) {
			dout(" inode %p now clean\n", inode);
			BUG_ON(!list_empty(&ci->i_dirty_item));
Y
Yan, Zheng 已提交
3533
			drop = true;
3534 3535
			if (ci->i_wr_ref == 0 &&
			    ci->i_wrbuffer_ref_head == 0) {
3536 3537 3538 3539
				BUG_ON(!ci->i_head_snapc);
				ceph_put_snap_context(ci->i_head_snapc);
				ci->i_head_snapc = NULL;
			}
S
Sage Weil 已提交
3540 3541
		} else {
			BUG_ON(list_empty(&ci->i_dirty_item));
3542
		}
S
Sage Weil 已提交
3543 3544 3545 3546
	}
	spin_unlock(&mdsc->cap_dirty_lock);

out:
3547
	spin_unlock(&ci->i_ceph_lock);
3548 3549 3550

	while (!list_empty(&to_remove)) {
		cf = list_first_entry(&to_remove,
3551 3552
				      struct ceph_cap_flush, i_list);
		list_del(&cf->i_list);
3553
		ceph_free_cap_flush(cf);
3554
	}
Y
Yan, Zheng 已提交
3555 3556 3557 3558 3559

	if (wake_ci)
		wake_up_all(&ci->i_cap_wq);
	if (wake_mdsc)
		wake_up_all(&mdsc->cap_flushing_wq);
3560
	if (drop)
S
Sage Weil 已提交
3561 3562 3563 3564 3565 3566 3567 3568 3569
		iput(inode);
}

/*
 * Handle FLUSHSNAP_ACK.  MDS has flushed snap data to disk and we can
 * throw away our cap_snap.
 *
 * Caller hold s_mutex.
 */
3570
static void handle_cap_flushsnap_ack(struct inode *inode, u64 flush_tid,
S
Sage Weil 已提交
3571 3572 3573 3574
				     struct ceph_mds_caps *m,
				     struct ceph_mds_session *session)
{
	struct ceph_inode_info *ci = ceph_inode(inode);
3575
	struct ceph_mds_client *mdsc = ceph_sb_to_client(inode->i_sb)->mdsc;
S
Sage Weil 已提交
3576 3577
	u64 follows = le64_to_cpu(m->snap_follows);
	struct ceph_cap_snap *capsnap;
Y
Yan, Zheng 已提交
3578 3579 3580
	bool flushed = false;
	bool wake_ci = false;
	bool wake_mdsc = false;
S
Sage Weil 已提交
3581 3582 3583 3584

	dout("handle_cap_flushsnap_ack inode %p ci %p mds%d follows %lld\n",
	     inode, ci, session->s_mds, follows);

3585
	spin_lock(&ci->i_ceph_lock);
S
Sage Weil 已提交
3586 3587
	list_for_each_entry(capsnap, &ci->i_cap_snaps, ci_item) {
		if (capsnap->follows == follows) {
3588
			if (capsnap->cap_flush.tid != flush_tid) {
S
Sage Weil 已提交
3589 3590
				dout(" cap_snap %p follows %lld tid %lld !="
				     " %lld\n", capsnap, follows,
3591
				     flush_tid, capsnap->cap_flush.tid);
S
Sage Weil 已提交
3592 3593
				break;
			}
Y
Yan, Zheng 已提交
3594
			flushed = true;
S
Sage Weil 已提交
3595 3596 3597 3598 3599 3600
			break;
		} else {
			dout(" skipping cap_snap %p follows %lld\n",
			     capsnap, capsnap->follows);
		}
	}
3601 3602 3603 3604 3605
	if (flushed) {
		WARN_ON(capsnap->dirty_pages || capsnap->writing);
		dout(" removing %p cap_snap %p follows %lld\n",
		     inode, capsnap, follows);
		list_del(&capsnap->ci_item);
J
Jeff Layton 已提交
3606
		wake_ci |= __detach_cap_flush_from_ci(ci, &capsnap->cap_flush);
3607 3608 3609 3610 3611 3612

		spin_lock(&mdsc->cap_dirty_lock);

		if (list_empty(&ci->i_cap_flush_list))
			list_del_init(&ci->i_flushing_item);

J
Jeff Layton 已提交
3613 3614
		wake_mdsc |= __detach_cap_flush_from_mdsc(mdsc,
							  &capsnap->cap_flush);
3615 3616
		spin_unlock(&mdsc->cap_dirty_lock);
	}
3617
	spin_unlock(&ci->i_ceph_lock);
3618 3619 3620
	if (flushed) {
		ceph_put_snap_context(capsnap->context);
		ceph_put_cap_snap(capsnap);
Y
Yan, Zheng 已提交
3621 3622 3623 3624
		if (wake_ci)
			wake_up_all(&ci->i_cap_wq);
		if (wake_mdsc)
			wake_up_all(&mdsc->cap_flushing_wq);
S
Sage Weil 已提交
3625
		iput(inode);
3626
	}
S
Sage Weil 已提交
3627 3628 3629 3630 3631 3632 3633
}

/*
 * Handle TRUNC from MDS, indicating file truncation.
 *
 * caller hold s_mutex.
 */
3634
static bool handle_cap_trunc(struct inode *inode,
S
Sage Weil 已提交
3635 3636 3637 3638 3639 3640 3641 3642 3643 3644 3645 3646
			     struct ceph_mds_caps *trunc,
			     struct ceph_mds_session *session)
{
	struct ceph_inode_info *ci = ceph_inode(inode);
	int mds = session->s_mds;
	int seq = le32_to_cpu(trunc->seq);
	u32 truncate_seq = le32_to_cpu(trunc->truncate_seq);
	u64 truncate_size = le64_to_cpu(trunc->truncate_size);
	u64 size = le64_to_cpu(trunc->size);
	int implemented = 0;
	int dirty = __ceph_caps_dirty(ci);
	int issued = __ceph_caps_issued(ceph_inode(inode), &implemented);
3647 3648 3649
	bool queue_trunc = false;

	lockdep_assert_held(&ci->i_ceph_lock);
S
Sage Weil 已提交
3650 3651 3652 3653 3654 3655 3656

	issued |= implemented | dirty;

	dout("handle_cap_trunc inode %p mds%d seq %d to %lld seq %d\n",
	     inode, mds, seq, truncate_size, truncate_seq);
	queue_trunc = ceph_fill_file_size(inode, issued,
					  truncate_seq, truncate_size, size);
3657
	return queue_trunc;
S
Sage Weil 已提交
3658 3659 3660 3661 3662 3663 3664 3665 3666 3667 3668
}

/*
 * Handle EXPORT from MDS.  Cap is being migrated _from_ this mds to a
 * different one.  If we are the most recent migration we've seen (as
 * indicated by mseq), make note of the migrating cap bits for the
 * duration (until we see the corresponding IMPORT).
 *
 * caller holds s_mutex
 */
static void handle_cap_export(struct inode *inode, struct ceph_mds_caps *ex,
3669 3670
			      struct ceph_mds_cap_peer *ph,
			      struct ceph_mds_session *session)
S
Sage Weil 已提交
3671
{
S
Sage Weil 已提交
3672
	struct ceph_mds_client *mdsc = ceph_inode_to_client(inode)->mdsc;
3673
	struct ceph_mds_session *tsession = NULL;
3674
	struct ceph_cap *cap, *tcap, *new_cap = NULL;
S
Sage Weil 已提交
3675
	struct ceph_inode_info *ci = ceph_inode(inode);
3676
	u64 t_cap_id;
S
Sage Weil 已提交
3677
	unsigned mseq = le32_to_cpu(ex->migrate_seq);
3678 3679 3680
	unsigned t_seq, t_mseq;
	int target, issued;
	int mds = session->s_mds;
S
Sage Weil 已提交
3681

3682 3683 3684 3685 3686 3687 3688 3689 3690
	if (ph) {
		t_cap_id = le64_to_cpu(ph->cap_id);
		t_seq = le32_to_cpu(ph->seq);
		t_mseq = le32_to_cpu(ph->mseq);
		target = le32_to_cpu(ph->mds);
	} else {
		t_cap_id = t_seq = t_mseq = 0;
		target = -1;
	}
S
Sage Weil 已提交
3691

3692 3693 3694
	dout("handle_cap_export inode %p ci %p mds%d mseq %d target %d\n",
	     inode, ci, mds, mseq, target);
retry:
3695
	spin_lock(&ci->i_ceph_lock);
3696
	cap = __get_cap_for_mds(ci, mds);
3697
	if (!cap || cap->cap_id != le64_to_cpu(ex->cap_id))
3698
		goto out_unlock;
S
Sage Weil 已提交
3699

3700
	if (target < 0) {
3701
		__ceph_remove_cap(cap, false);
3702
		goto out_unlock;
S
Sage Weil 已提交
3703 3704
	}

3705 3706 3707 3708
	/*
	 * now we know we haven't received the cap import message yet
	 * because the exported cap still exist.
	 */
S
Sage Weil 已提交
3709

3710
	issued = cap->issued;
3711 3712 3713 3714 3715 3716 3717 3718
	if (issued != cap->implemented)
		pr_err_ratelimited("handle_cap_export: issued != implemented: "
				"ino (%llx.%llx) mds%d seq %d mseq %d "
				"issued %s implemented %s\n",
				ceph_vinop(inode), mds, cap->seq, cap->mseq,
				ceph_cap_string(issued),
				ceph_cap_string(cap->implemented));

3719 3720 3721 3722

	tcap = __get_cap_for_mds(ci, target);
	if (tcap) {
		/* already have caps from the target */
3723
		if (tcap->cap_id == t_cap_id &&
3724 3725 3726 3727 3728 3729 3730 3731 3732
		    ceph_seq_cmp(tcap->seq, t_seq) < 0) {
			dout(" updating import cap %p mds%d\n", tcap, target);
			tcap->cap_id = t_cap_id;
			tcap->seq = t_seq - 1;
			tcap->issue_seq = t_seq - 1;
			tcap->issued |= issued;
			tcap->implemented |= issued;
			if (cap == ci->i_auth_cap)
				ci->i_auth_cap = tcap;
3733

3734 3735
			if (!list_empty(&ci->i_cap_flush_list) &&
			    ci->i_auth_cap == tcap) {
3736 3737 3738 3739
				spin_lock(&mdsc->cap_dirty_lock);
				list_move_tail(&ci->i_flushing_item,
					       &tcap->session->s_cap_flushing);
				spin_unlock(&mdsc->cap_dirty_lock);
S
Sage Weil 已提交
3740
			}
S
Sage Weil 已提交
3741
		}
3742
		__ceph_remove_cap(cap, false);
3743
		goto out_unlock;
3744
	} else if (tsession) {
3745
		/* add placeholder for the export tagert */
3746
		int flag = (cap == ci->i_auth_cap) ? CEPH_CAP_FLAG_AUTH : 0;
3747
		tcap = new_cap;
3748
		ceph_add_cap(inode, tsession, t_cap_id, issued, 0,
3749 3750
			     t_seq - 1, t_mseq, (u64)-1, flag, &new_cap);

3751 3752 3753 3754 3755 3756 3757 3758
		if (!list_empty(&ci->i_cap_flush_list) &&
		    ci->i_auth_cap == tcap) {
			spin_lock(&mdsc->cap_dirty_lock);
			list_move_tail(&ci->i_flushing_item,
				       &tcap->session->s_cap_flushing);
			spin_unlock(&mdsc->cap_dirty_lock);
		}

3759 3760
		__ceph_remove_cap(cap, false);
		goto out_unlock;
S
Sage Weil 已提交
3761 3762
	}

3763
	spin_unlock(&ci->i_ceph_lock);
3764 3765 3766 3767 3768 3769 3770 3771 3772 3773 3774 3775 3776 3777
	mutex_unlock(&session->s_mutex);

	/* open target session */
	tsession = ceph_mdsc_open_export_target_session(mdsc, target);
	if (!IS_ERR(tsession)) {
		if (mds > target) {
			mutex_lock(&session->s_mutex);
			mutex_lock_nested(&tsession->s_mutex,
					  SINGLE_DEPTH_NESTING);
		} else {
			mutex_lock(&tsession->s_mutex);
			mutex_lock_nested(&session->s_mutex,
					  SINGLE_DEPTH_NESTING);
		}
3778
		new_cap = ceph_get_cap(mdsc, NULL);
3779 3780 3781 3782
	} else {
		WARN_ON(1);
		tsession = NULL;
		target = -1;
3783
		mutex_lock(&session->s_mutex);
3784 3785 3786 3787 3788 3789 3790 3791 3792 3793
	}
	goto retry;

out_unlock:
	spin_unlock(&ci->i_ceph_lock);
	mutex_unlock(&session->s_mutex);
	if (tsession) {
		mutex_unlock(&tsession->s_mutex);
		ceph_put_mds_session(tsession);
	}
3794 3795
	if (new_cap)
		ceph_put_cap(mdsc, new_cap);
S
Sage Weil 已提交
3796 3797 3798
}

/*
Y
Yan, Zheng 已提交
3799
 * Handle cap IMPORT.
S
Sage Weil 已提交
3800
 *
Y
Yan, Zheng 已提交
3801
 * caller holds s_mutex. acquires i_ceph_lock
S
Sage Weil 已提交
3802 3803 3804
 */
static void handle_cap_import(struct ceph_mds_client *mdsc,
			      struct inode *inode, struct ceph_mds_caps *im,
3805
			      struct ceph_mds_cap_peer *ph,
S
Sage Weil 已提交
3806
			      struct ceph_mds_session *session,
Y
Yan, Zheng 已提交
3807
			      struct ceph_cap **target_cap, int *old_issued)
S
Sage Weil 已提交
3808 3809
{
	struct ceph_inode_info *ci = ceph_inode(inode);
Y
Yan, Zheng 已提交
3810
	struct ceph_cap *cap, *ocap, *new_cap = NULL;
S
Sage Weil 已提交
3811
	int mds = session->s_mds;
Y
Yan, Zheng 已提交
3812 3813
	int issued;
	unsigned caps = le32_to_cpu(im->caps);
S
Sage Weil 已提交
3814 3815 3816 3817 3818
	unsigned wanted = le32_to_cpu(im->wanted);
	unsigned seq = le32_to_cpu(im->seq);
	unsigned mseq = le32_to_cpu(im->migrate_seq);
	u64 realmino = le64_to_cpu(im->realm);
	u64 cap_id = le64_to_cpu(im->cap_id);
3819 3820
	u64 p_cap_id;
	int peer;
S
Sage Weil 已提交
3821

3822 3823 3824 3825 3826 3827 3828
	if (ph) {
		p_cap_id = le64_to_cpu(ph->cap_id);
		peer = le32_to_cpu(ph->mds);
	} else {
		p_cap_id = 0;
		peer = -1;
	}
S
Sage Weil 已提交
3829

3830 3831
	dout("handle_cap_import inode %p ci %p mds%d mseq %d peer %d\n",
	     inode, ci, mds, mseq, peer);
3832 3833 3834 3835 3836 3837
retry:
	cap = __get_cap_for_mds(ci, mds);
	if (!cap) {
		if (!new_cap) {
			spin_unlock(&ci->i_ceph_lock);
			new_cap = ceph_get_cap(mdsc, NULL);
3838
			spin_lock(&ci->i_ceph_lock);
3839 3840
			goto retry;
		}
Y
Yan, Zheng 已提交
3841 3842 3843 3844 3845 3846
		cap = new_cap;
	} else {
		if (new_cap) {
			ceph_put_cap(mdsc, new_cap);
			new_cap = NULL;
		}
3847 3848
	}

Y
Yan, Zheng 已提交
3849 3850 3851
	__ceph_caps_issued(ci, &issued);
	issued |= __ceph_caps_dirty(ci);

3852
	ceph_add_cap(inode, session, cap_id, caps, wanted, seq, mseq,
3853 3854
		     realmino, CEPH_CAP_FLAG_AUTH, &new_cap);

Y
Yan, Zheng 已提交
3855 3856
	ocap = peer >= 0 ? __get_cap_for_mds(ci, peer) : NULL;
	if (ocap && ocap->cap_id == p_cap_id) {
3857
		dout(" remove export cap %p mds%d flags %d\n",
Y
Yan, Zheng 已提交
3858
		     ocap, peer, ph->flags);
3859
		if ((ph->flags & CEPH_CAP_FLAG_AUTH) &&
Y
Yan, Zheng 已提交
3860 3861
		    (ocap->seq != le32_to_cpu(ph->seq) ||
		     ocap->mseq != le32_to_cpu(ph->mseq))) {
3862 3863 3864 3865 3866 3867 3868
			pr_err_ratelimited("handle_cap_import: "
					"mismatched seq/mseq: ino (%llx.%llx) "
					"mds%d seq %d mseq %d importer mds%d "
					"has peer seq %d mseq %d\n",
					ceph_vinop(inode), peer, ocap->seq,
					ocap->mseq, mds, le32_to_cpu(ph->seq),
					le32_to_cpu(ph->mseq));
S
Sage Weil 已提交
3869
		}
Y
Yan, Zheng 已提交
3870
		__ceph_remove_cap(ocap, (ph->flags & CEPH_CAP_FLAG_RELEASE));
S
Sage Weil 已提交
3871 3872
	}

3873 3874
	/* make sure we re-request max_size, if necessary */
	ci->i_requested_max_size = 0;
3875

Y
Yan, Zheng 已提交
3876 3877
	*old_issued = issued;
	*target_cap = cap;
S
Sage Weil 已提交
3878 3879 3880 3881 3882 3883 3884 3885 3886 3887 3888 3889 3890
}

/*
 * Handle a caps message from the MDS.
 *
 * Identify the appropriate session, inode, and call the right handler
 * based on the cap op.
 */
void ceph_handle_caps(struct ceph_mds_session *session,
		      struct ceph_msg *msg)
{
	struct ceph_mds_client *mdsc = session->s_mdsc;
	struct inode *inode;
3891
	struct ceph_inode_info *ci;
S
Sage Weil 已提交
3892 3893
	struct ceph_cap *cap;
	struct ceph_mds_caps *h;
3894
	struct ceph_mds_cap_peer *peer = NULL;
Y
Yan, Zheng 已提交
3895
	struct ceph_snap_realm *realm = NULL;
3896
	int op;
3897
	int msg_version = le16_to_cpu(msg->hdr.version);
3898
	u32 seq, mseq;
S
Sage Weil 已提交
3899
	struct ceph_vino vino;
3900
	void *snaptrace;
3901
	size_t snaptrace_len;
3902
	void *p, *end;
3903
	struct cap_extra_info extra_info = {};
3904
	bool queue_trunc;
S
Sage Weil 已提交
3905

3906
	dout("handle_caps from mds%d\n", session->s_mds);
S
Sage Weil 已提交
3907 3908

	/* decode */
3909
	end = msg->front.iov_base + msg->front.iov_len;
S
Sage Weil 已提交
3910 3911 3912 3913 3914 3915 3916
	if (msg->front.iov_len < sizeof(*h))
		goto bad;
	h = msg->front.iov_base;
	op = le32_to_cpu(h->op);
	vino.ino = le64_to_cpu(h->ino);
	vino.snap = CEPH_NOSNAP;
	seq = le32_to_cpu(h->seq);
3917
	mseq = le32_to_cpu(h->migrate_seq);
S
Sage Weil 已提交
3918

3919 3920
	snaptrace = h + 1;
	snaptrace_len = le32_to_cpu(h->snap_trace_len);
3921
	p = snaptrace + snaptrace_len;
3922

3923
	if (msg_version >= 2) {
3924
		u32 flock_len;
3925
		ceph_decode_32_safe(&p, end, flock_len, bad);
3926 3927
		if (p + flock_len > end)
			goto bad;
3928
		p += flock_len;
3929 3930
	}

3931
	if (msg_version >= 3) {
3932 3933 3934 3935
		if (op == CEPH_CAP_OP_IMPORT) {
			if (p + sizeof(*peer) > end)
				goto bad;
			peer = p;
3936
			p += sizeof(*peer);
3937 3938 3939
		} else if (op == CEPH_CAP_OP_EXPORT) {
			/* recorded in unused fields */
			peer = (void *)&h->size;
3940 3941 3942
		}
	}

3943
	if (msg_version >= 4) {
3944 3945 3946
		ceph_decode_64_safe(&p, end, extra_info.inline_version, bad);
		ceph_decode_32_safe(&p, end, extra_info.inline_len, bad);
		if (p + extra_info.inline_len > end)
3947
			goto bad;
3948 3949
		extra_info.inline_data = p;
		p += extra_info.inline_len;
3950 3951
	}

3952
	if (msg_version >= 5) {
3953 3954 3955 3956 3957 3958 3959
		struct ceph_osd_client	*osdc = &mdsc->fsc->client->osdc;
		u32			epoch_barrier;

		ceph_decode_32_safe(&p, end, epoch_barrier, bad);
		ceph_osdc_update_epoch_barrier(osdc, epoch_barrier);
	}

3960
	if (msg_version >= 8) {
3961 3962
		u64 flush_tid;
		u32 caller_uid, caller_gid;
Y
Yan, Zheng 已提交
3963
		u32 pool_ns_len;
3964

3965 3966 3967 3968 3969 3970 3971
		/* version >= 6 */
		ceph_decode_64_safe(&p, end, flush_tid, bad);
		/* version >= 7 */
		ceph_decode_32_safe(&p, end, caller_uid, bad);
		ceph_decode_32_safe(&p, end, caller_gid, bad);
		/* version >= 8 */
		ceph_decode_32_safe(&p, end, pool_ns_len, bad);
Y
Yan, Zheng 已提交
3972 3973
		if (pool_ns_len > 0) {
			ceph_decode_need(&p, end, pool_ns_len, bad);
3974 3975
			extra_info.pool_ns =
				ceph_find_or_create_string(p, pool_ns_len);
Y
Yan, Zheng 已提交
3976 3977
			p += pool_ns_len;
		}
3978 3979
	}

J
Jeff Layton 已提交
3980
	if (msg_version >= 9) {
3981 3982 3983 3984 3985
		struct ceph_timespec *btime;

		if (p + sizeof(*btime) > end)
			goto bad;
		btime = p;
J
Jeff Layton 已提交
3986
		ceph_decode_timespec64(&extra_info.btime, btime);
3987
		p += sizeof(*btime);
3988
		ceph_decode_64_safe(&p, end, extra_info.change_attr, bad);
J
Jeff Layton 已提交
3989 3990 3991 3992
	}

	if (msg_version >= 11) {
		u32 flags;
3993 3994 3995 3996 3997 3998 3999 4000
		/* version >= 10 */
		ceph_decode_32_safe(&p, end, flags, bad);
		/* version >= 11 */
		extra_info.dirstat_valid = true;
		ceph_decode_64_safe(&p, end, extra_info.nfiles, bad);
		ceph_decode_64_safe(&p, end, extra_info.nsubdirs, bad);
	}

4001
	/* lookup ino */
4002
	inode = ceph_find_inode(mdsc->fsc->sb, vino);
4003 4004 4005 4006
	ci = ceph_inode(inode);
	dout(" op %s ino %llx.%llx inode %p\n", ceph_cap_op_name(op), vino.ino,
	     vino.snap, inode);

S
Sage Weil 已提交
4007 4008 4009 4010 4011 4012 4013
	mutex_lock(&session->s_mutex);
	session->s_seq++;
	dout(" mds%d seq %lld cap seq %u\n", session->s_mds, session->s_seq,
	     (unsigned)seq);

	if (!inode) {
		dout(" i don't have ino %llx\n", vino.ino);
4014

4015
		if (op == CEPH_CAP_OP_IMPORT) {
4016 4017 4018
			cap = ceph_get_cap(mdsc, NULL);
			cap->cap_ino = vino.ino;
			cap->queue_release = 1;
Y
Yan, Zheng 已提交
4019
			cap->cap_id = le64_to_cpu(h->cap_id);
4020 4021
			cap->mseq = mseq;
			cap->seq = seq;
4022
			cap->issue_seq = seq;
4023
			spin_lock(&session->s_cap_lock);
4024
			__ceph_queue_cap_release(session, cap);
4025 4026
			spin_unlock(&session->s_cap_lock);
		}
4027
		goto flush_cap_releases;
S
Sage Weil 已提交
4028 4029 4030 4031 4032
	}

	/* these will work even if we don't have a cap yet */
	switch (op) {
	case CEPH_CAP_OP_FLUSHSNAP_ACK:
4033 4034
		handle_cap_flushsnap_ack(inode, le64_to_cpu(msg->hdr.tid),
					 h, session);
S
Sage Weil 已提交
4035 4036 4037
		goto done;

	case CEPH_CAP_OP_EXPORT:
4038 4039
		handle_cap_export(inode, h, peer, session);
		goto done_unlocked;
S
Sage Weil 已提交
4040 4041

	case CEPH_CAP_OP_IMPORT:
4042 4043 4044 4045 4046 4047 4048 4049 4050 4051
		realm = NULL;
		if (snaptrace_len) {
			down_write(&mdsc->snap_rwsem);
			ceph_update_snap_trace(mdsc, snaptrace,
					       snaptrace + snaptrace_len,
					       false, &realm);
			downgrade_write(&mdsc->snap_rwsem);
		} else {
			down_read(&mdsc->snap_rwsem);
		}
4052
		spin_lock(&ci->i_ceph_lock);
4053
		handle_cap_import(mdsc, inode, h, peer, session,
4054 4055 4056
				  &cap, &extra_info.issued);
		handle_cap_grant(inode, session, cap,
				 h, msg->middle, &extra_info);
4057 4058
		if (realm)
			ceph_put_snap_realm(mdsc, realm);
Y
Yan, Zheng 已提交
4059
		goto done_unlocked;
S
Sage Weil 已提交
4060 4061 4062
	}

	/* the rest require a cap */
4063
	spin_lock(&ci->i_ceph_lock);
4064
	cap = __get_cap_for_mds(ceph_inode(inode), session->s_mds);
S
Sage Weil 已提交
4065
	if (!cap) {
4066
		dout(" no cap on %p ino %llx.%llx from mds%d\n",
4067 4068
		     inode, ceph_ino(inode), ceph_snap(inode),
		     session->s_mds);
4069
		spin_unlock(&ci->i_ceph_lock);
4070
		goto flush_cap_releases;
S
Sage Weil 已提交
4071 4072
	}

4073
	/* note that each of these drops i_ceph_lock for us */
S
Sage Weil 已提交
4074 4075 4076
	switch (op) {
	case CEPH_CAP_OP_REVOKE:
	case CEPH_CAP_OP_GRANT:
4077 4078 4079 4080
		__ceph_caps_issued(ci, &extra_info.issued);
		extra_info.issued |= __ceph_caps_dirty(ci);
		handle_cap_grant(inode, session, cap,
				 h, msg->middle, &extra_info);
4081
		goto done_unlocked;
S
Sage Weil 已提交
4082 4083

	case CEPH_CAP_OP_FLUSH_ACK:
4084 4085
		handle_cap_flush_ack(inode, le64_to_cpu(msg->hdr.tid),
				     h, session, cap);
S
Sage Weil 已提交
4086 4087 4088
		break;

	case CEPH_CAP_OP_TRUNC:
4089 4090 4091 4092
		queue_trunc = handle_cap_trunc(inode, h, session);
		spin_unlock(&ci->i_ceph_lock);
		if (queue_trunc)
			ceph_queue_vmtruncate(inode);
S
Sage Weil 已提交
4093 4094 4095
		break;

	default:
4096
		spin_unlock(&ci->i_ceph_lock);
S
Sage Weil 已提交
4097 4098 4099 4100
		pr_err("ceph_handle_caps: unknown cap op %d %s\n", op,
		       ceph_cap_op_name(op));
	}

4101 4102 4103 4104
done:
	mutex_unlock(&session->s_mutex);
done_unlocked:
	ceph_put_string(extra_info.pool_ns);
4105 4106
	/* avoid calling iput_final() in mds dispatch threads */
	ceph_async_iput(inode);
4107
	return;
4108 4109 4110

flush_cap_releases:
	/*
4111
	 * send any cap release message to try to move things
4112 4113 4114
	 * along for the mds (who clearly thinks we still have this
	 * cap).
	 */
4115 4116
	ceph_flush_cap_releases(mdsc, session);
	goto done;
S
Sage Weil 已提交
4117 4118 4119

bad:
	pr_err("ceph_handle_caps: corrupt message\n");
4120
	ceph_msg_dump(msg);
S
Sage Weil 已提交
4121 4122 4123 4124 4125 4126
	return;
}

/*
 * Delayed work handler to process end of delayed cap release LRU list.
 */
4127
void ceph_check_delayed_caps(struct ceph_mds_client *mdsc)
S
Sage Weil 已提交
4128
{
4129
	struct inode *inode;
S
Sage Weil 已提交
4130 4131 4132 4133 4134 4135 4136 4137 4138 4139 4140 4141 4142 4143
	struct ceph_inode_info *ci;

	dout("check_delayed_caps\n");
	while (1) {
		spin_lock(&mdsc->cap_delay_lock);
		if (list_empty(&mdsc->cap_delay_list))
			break;
		ci = list_first_entry(&mdsc->cap_delay_list,
				      struct ceph_inode_info,
				      i_cap_delay_list);
		if ((ci->i_ceph_flags & CEPH_I_FLUSH) == 0 &&
		    time_before(jiffies, ci->i_hold_caps_max))
			break;
		list_del_init(&ci->i_cap_delay_list);
4144 4145

		inode = igrab(&ci->vfs_inode);
S
Sage Weil 已提交
4146
		spin_unlock(&mdsc->cap_delay_lock);
4147 4148 4149

		if (inode) {
			dout("check_delayed_caps on %p\n", inode);
4150
			ceph_check_caps(ci, 0, NULL);
4151 4152
			/* avoid calling iput_final() in tick thread */
			ceph_async_iput(inode);
4153
		}
S
Sage Weil 已提交
4154 4155 4156 4157
	}
	spin_unlock(&mdsc->cap_delay_lock);
}

4158 4159 4160 4161 4162
/*
 * Flush all dirty caps to the mds
 */
void ceph_flush_dirty_caps(struct ceph_mds_client *mdsc)
{
S
Sage Weil 已提交
4163 4164
	struct ceph_inode_info *ci;
	struct inode *inode;
4165 4166 4167

	dout("flush_dirty_caps\n");
	spin_lock(&mdsc->cap_dirty_lock);
S
Sage Weil 已提交
4168 4169 4170
	while (!list_empty(&mdsc->cap_dirty)) {
		ci = list_first_entry(&mdsc->cap_dirty, struct ceph_inode_info,
				      i_dirty_item);
4171 4172
		inode = &ci->vfs_inode;
		ihold(inode);
S
Sage Weil 已提交
4173
		dout("flush_dirty_caps %p\n", inode);
4174
		spin_unlock(&mdsc->cap_dirty_lock);
4175
		ceph_check_caps(ci, CHECK_CAPS_FLUSH, NULL);
4176
		iput(inode);
4177 4178 4179
		spin_lock(&mdsc->cap_dirty_lock);
	}
	spin_unlock(&mdsc->cap_dirty_lock);
S
Sage Weil 已提交
4180
	dout("flush_dirty_caps done\n");
4181 4182
}

4183 4184 4185 4186 4187 4188 4189 4190 4191 4192 4193 4194
void __ceph_touch_fmode(struct ceph_inode_info *ci,
			struct ceph_mds_client *mdsc, int fmode)
{
	unsigned long now = jiffies;
	if (fmode & CEPH_FILE_MODE_RD)
		ci->i_last_rd = now;
	if (fmode & CEPH_FILE_MODE_WR)
		ci->i_last_wr = now;
	/* queue periodic check */
	if (fmode &&
	    __ceph_is_any_real_caps(ci) &&
	    list_empty(&ci->i_cap_delay_list))
4195
		__cap_delay_requeue(mdsc, ci);
4196 4197 4198 4199 4200 4201 4202 4203 4204 4205 4206 4207 4208 4209
}

void ceph_get_fmode(struct ceph_inode_info *ci, int fmode, int count)
{
	int i;
	int bits = (fmode << 1) | 1;
	spin_lock(&ci->i_ceph_lock);
	for (i = 0; i < CEPH_FILE_MODE_BITS; i++) {
		if (bits & (1 << i))
			ci->i_nr_by_mode[i] += count;
	}
	spin_unlock(&ci->i_ceph_lock);
}

S
Sage Weil 已提交
4210 4211 4212 4213 4214
/*
 * Drop open file reference.  If we were the last open file,
 * we may need to release capabilities to the MDS (or schedule
 * their delayed release).
 */
4215
void ceph_put_fmode(struct ceph_inode_info *ci, int fmode, int count)
S
Sage Weil 已提交
4216
{
4217
	int i;
4218
	int bits = (fmode << 1) | 1;
4219
	spin_lock(&ci->i_ceph_lock);
4220 4221
	for (i = 0; i < CEPH_FILE_MODE_BITS; i++) {
		if (bits & (1 << i)) {
4222 4223
			BUG_ON(ci->i_nr_by_mode[i] < count);
			ci->i_nr_by_mode[i] -= count;
4224 4225
		}
	}
4226
	spin_unlock(&ci->i_ceph_lock);
S
Sage Weil 已提交
4227 4228
}

4229
/*
4230
 * For a soon-to-be unlinked file, drop the LINK caps. If it
4231 4232 4233 4234 4235 4236 4237 4238 4239 4240 4241 4242 4243 4244 4245 4246 4247 4248 4249 4250 4251 4252 4253
 * looks like the link count will hit 0, drop any other caps (other
 * than PIN) we don't specifically want (due to the file still being
 * open).
 */
int ceph_drop_caps_for_unlink(struct inode *inode)
{
	struct ceph_inode_info *ci = ceph_inode(inode);
	int drop = CEPH_CAP_LINK_SHARED | CEPH_CAP_LINK_EXCL;

	spin_lock(&ci->i_ceph_lock);
	if (inode->i_nlink == 1) {
		drop |= ~(__ceph_caps_wanted(ci) | CEPH_CAP_PIN);

		if (__ceph_caps_dirty(ci)) {
			struct ceph_mds_client *mdsc =
				ceph_inode_to_client(inode)->mdsc;
			__cap_delay_requeue_front(mdsc, ci);
		}
	}
	spin_unlock(&ci->i_ceph_lock);
	return drop;
}

S
Sage Weil 已提交
4254 4255 4256 4257 4258 4259 4260 4261 4262 4263 4264 4265 4266 4267
/*
 * Helpers for embedding cap and dentry lease releases into mds
 * requests.
 *
 * @force is used by dentry_release (below) to force inclusion of a
 * record for the directory inode, even when there aren't any caps to
 * drop.
 */
int ceph_encode_inode_release(void **p, struct inode *inode,
			      int mds, int drop, int unless, int force)
{
	struct ceph_inode_info *ci = ceph_inode(inode);
	struct ceph_cap *cap;
	struct ceph_mds_request_release *rel = *p;
4268
	int used, dirty;
S
Sage Weil 已提交
4269 4270
	int ret = 0;

4271
	spin_lock(&ci->i_ceph_lock);
4272
	used = __ceph_caps_used(ci);
4273
	dirty = __ceph_caps_dirty(ci);
4274

4275 4276
	dout("encode_inode_release %p mds%d used|dirty %s drop %s unless %s\n",
	     inode, mds, ceph_cap_string(used|dirty), ceph_cap_string(drop),
4277 4278
	     ceph_cap_string(unless));

4279 4280
	/* only drop unused, clean caps */
	drop &= ~(used | dirty);
4281

S
Sage Weil 已提交
4282 4283
	cap = __get_cap_for_mds(ci, mds);
	if (cap && __cap_is_valid(cap)) {
4284 4285 4286 4287 4288 4289 4290 4291 4292 4293 4294 4295 4296 4297
		unless &= cap->issued;
		if (unless) {
			if (unless & CEPH_CAP_AUTH_EXCL)
				drop &= ~CEPH_CAP_AUTH_SHARED;
			if (unless & CEPH_CAP_LINK_EXCL)
				drop &= ~CEPH_CAP_LINK_SHARED;
			if (unless & CEPH_CAP_XATTR_EXCL)
				drop &= ~CEPH_CAP_XATTR_SHARED;
			if (unless & CEPH_CAP_FILE_EXCL)
				drop &= ~CEPH_CAP_FILE_SHARED;
		}

		if (force || (cap->issued & drop)) {
			if (cap->issued & drop) {
Y
Yan, Zheng 已提交
4298 4299 4300
				int wanted = __ceph_caps_wanted(ci);
				dout("encode_inode_release %p cap %p "
				     "%s -> %s, wanted %s -> %s\n", inode, cap,
S
Sage Weil 已提交
4301
				     ceph_cap_string(cap->issued),
Y
Yan, Zheng 已提交
4302 4303 4304 4305
				     ceph_cap_string(cap->issued & ~drop),
				     ceph_cap_string(cap->mds_wanted),
				     ceph_cap_string(wanted));

S
Sage Weil 已提交
4306 4307
				cap->issued &= ~drop;
				cap->implemented &= ~drop;
Y
Yan, Zheng 已提交
4308
				cap->mds_wanted = wanted;
S
Sage Weil 已提交
4309 4310 4311 4312 4313 4314 4315 4316 4317
			} else {
				dout("encode_inode_release %p cap %p %s"
				     " (force)\n", inode, cap,
				     ceph_cap_string(cap->issued));
			}

			rel->ino = cpu_to_le64(ceph_ino(inode));
			rel->cap_id = cpu_to_le64(cap->cap_id);
			rel->seq = cpu_to_le32(cap->seq);
4318
			rel->issue_seq = cpu_to_le32(cap->issue_seq);
S
Sage Weil 已提交
4319
			rel->mseq = cpu_to_le32(cap->mseq);
4320
			rel->caps = cpu_to_le32(cap->implemented);
S
Sage Weil 已提交
4321 4322 4323 4324 4325 4326
			rel->wanted = cpu_to_le32(cap->mds_wanted);
			rel->dname_len = 0;
			rel->dname_seq = 0;
			*p += sizeof(*rel);
			ret = 1;
		} else {
4327
			dout("encode_inode_release %p cap %p %s (noop)\n",
S
Sage Weil 已提交
4328 4329 4330
			     inode, cap, ceph_cap_string(cap->issued));
		}
	}
4331
	spin_unlock(&ci->i_ceph_lock);
S
Sage Weil 已提交
4332 4333 4334 4335
	return ret;
}

int ceph_encode_dentry_release(void **p, struct dentry *dentry,
4336
			       struct inode *dir,
S
Sage Weil 已提交
4337 4338
			       int mds, int drop, int unless)
{
4339
	struct dentry *parent = NULL;
S
Sage Weil 已提交
4340 4341 4342 4343 4344 4345 4346
	struct ceph_mds_request_release *rel = *p;
	struct ceph_dentry_info *di = ceph_dentry(dentry);
	int force = 0;
	int ret;

	/*
	 * force an record for the directory caps if we have a dentry lease.
4347
	 * this is racy (can't take i_ceph_lock and d_lock together), but it
S
Sage Weil 已提交
4348 4349 4350 4351 4352 4353
	 * doesn't have to be perfect; the mds will revoke anything we don't
	 * release.
	 */
	spin_lock(&dentry->d_lock);
	if (di->lease_session && di->lease_session->s_mds == mds)
		force = 1;
4354 4355 4356 4357
	if (!dir) {
		parent = dget(dentry->d_parent);
		dir = d_inode(parent);
	}
S
Sage Weil 已提交
4358 4359
	spin_unlock(&dentry->d_lock);

4360
	ret = ceph_encode_inode_release(p, dir, mds, drop, unless, force);
4361
	dput(parent);
S
Sage Weil 已提交
4362 4363 4364 4365 4366 4367 4368 4369 4370

	spin_lock(&dentry->d_lock);
	if (ret && di->lease_session && di->lease_session->s_mds == mds) {
		dout("encode_dentry_release %p mds%d seq %d\n",
		     dentry, mds, (int)di->lease_seq);
		rel->dname_len = cpu_to_le32(dentry->d_name.len);
		memcpy(*p, dentry->d_name.name, dentry->d_name.len);
		*p += dentry->d_name.len;
		rel->dname_seq = cpu_to_le32(di->lease_seq);
S
Sage Weil 已提交
4371
		__ceph_mdsc_drop_dentry_lease(dentry);
S
Sage Weil 已提交
4372 4373 4374 4375
	}
	spin_unlock(&dentry->d_lock);
	return ret;
}