osdmap.c 28.0 KB
Newer Older
S
Sage Weil 已提交
1

2
#include <linux/ceph/ceph_debug.h>
3

4
#include <linux/module.h>
5
#include <linux/slab.h>
S
Sage Weil 已提交
6 7
#include <asm/div64.h>

8 9 10 11 12
#include <linux/ceph/libceph.h>
#include <linux/ceph/osdmap.h>
#include <linux/ceph/decode.h>
#include <linux/crush/hash.h>
#include <linux/crush/mapper.h>
S
Sage Weil 已提交
13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40

char *ceph_osdmap_state_str(char *str, int len, int state)
{
	int flag = 0;

	if (!len)
		goto done;

	*str = '\0';
	if (state) {
		if (state & CEPH_OSD_EXISTS) {
			snprintf(str, len, "exists");
			flag = 1;
		}
		if (state & CEPH_OSD_UP) {
			snprintf(str, len, "%s%s%s", str, (flag ? ", " : ""),
				 "up");
			flag = 1;
		}
	} else {
		snprintf(str, len, "doesn't exist");
	}
done:
	return str;
}

/* maps */

41
static int calc_bits_of(unsigned int t)
S
Sage Weil 已提交
42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72
{
	int b = 0;
	while (t) {
		t = t >> 1;
		b++;
	}
	return b;
}

/*
 * the foo_mask is the smallest value 2^n-1 that is >= foo.
 */
static void calc_pg_masks(struct ceph_pg_pool_info *pi)
{
	pi->pg_num_mask = (1 << calc_bits_of(le32_to_cpu(pi->v.pg_num)-1)) - 1;
	pi->pgp_num_mask =
		(1 << calc_bits_of(le32_to_cpu(pi->v.pgp_num)-1)) - 1;
	pi->lpg_num_mask =
		(1 << calc_bits_of(le32_to_cpu(pi->v.lpg_num)-1)) - 1;
	pi->lpgp_num_mask =
		(1 << calc_bits_of(le32_to_cpu(pi->v.lpgp_num)-1)) - 1;
}

/*
 * decode crush map
 */
static int crush_decode_uniform_bucket(void **p, void *end,
				       struct crush_bucket_uniform *b)
{
	dout("crush_decode_uniform_bucket %p to %p\n", *p, end);
	ceph_decode_need(p, end, (1+b->h.size) * sizeof(u32), bad);
73
	b->item_weight = ceph_decode_32(p);
S
Sage Weil 已提交
74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91
	return 0;
bad:
	return -EINVAL;
}

static int crush_decode_list_bucket(void **p, void *end,
				    struct crush_bucket_list *b)
{
	int j;
	dout("crush_decode_list_bucket %p to %p\n", *p, end);
	b->item_weights = kcalloc(b->h.size, sizeof(u32), GFP_NOFS);
	if (b->item_weights == NULL)
		return -ENOMEM;
	b->sum_weights = kcalloc(b->h.size, sizeof(u32), GFP_NOFS);
	if (b->sum_weights == NULL)
		return -ENOMEM;
	ceph_decode_need(p, end, 2 * b->h.size * sizeof(u32), bad);
	for (j = 0; j < b->h.size; j++) {
92 93
		b->item_weights[j] = ceph_decode_32(p);
		b->sum_weights[j] = ceph_decode_32(p);
S
Sage Weil 已提交
94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110
	}
	return 0;
bad:
	return -EINVAL;
}

static int crush_decode_tree_bucket(void **p, void *end,
				    struct crush_bucket_tree *b)
{
	int j;
	dout("crush_decode_tree_bucket %p to %p\n", *p, end);
	ceph_decode_32_safe(p, end, b->num_nodes, bad);
	b->node_weights = kcalloc(b->num_nodes, sizeof(u32), GFP_NOFS);
	if (b->node_weights == NULL)
		return -ENOMEM;
	ceph_decode_need(p, end, b->num_nodes * sizeof(u32), bad);
	for (j = 0; j < b->num_nodes; j++)
111
		b->node_weights[j] = ceph_decode_32(p);
S
Sage Weil 已提交
112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129
	return 0;
bad:
	return -EINVAL;
}

static int crush_decode_straw_bucket(void **p, void *end,
				     struct crush_bucket_straw *b)
{
	int j;
	dout("crush_decode_straw_bucket %p to %p\n", *p, end);
	b->item_weights = kcalloc(b->h.size, sizeof(u32), GFP_NOFS);
	if (b->item_weights == NULL)
		return -ENOMEM;
	b->straws = kcalloc(b->h.size, sizeof(u32), GFP_NOFS);
	if (b->straws == NULL)
		return -ENOMEM;
	ceph_decode_need(p, end, 2 * b->h.size * sizeof(u32), bad);
	for (j = 0; j < b->h.size; j++) {
130 131
		b->item_weights[j] = ceph_decode_32(p);
		b->straws[j] = ceph_decode_32(p);
S
Sage Weil 已提交
132 133 134 135 136 137
	}
	return 0;
bad:
	return -EINVAL;
}

S
Sage Weil 已提交
138 139 140 141 142 143 144 145 146 147 148 149 150 151 152
static int skip_name_map(void **p, void *end)
{
        int len;
        ceph_decode_32_safe(p, end, len ,bad);
        while (len--) {
                int strlen;
                *p += sizeof(u32);
                ceph_decode_32_safe(p, end, strlen, bad);
                *p += strlen;
}
        return 0;
bad:
        return -EINVAL;
}

S
Sage Weil 已提交
153 154 155 156 157 158 159 160
static struct crush_map *crush_decode(void *pbyval, void *end)
{
	struct crush_map *c;
	int err = -EINVAL;
	int i, j;
	void **p = &pbyval;
	void *start = pbyval;
	u32 magic;
S
Sage Weil 已提交
161
	u32 num_name_maps;
S
Sage Weil 已提交
162 163 164 165 166 167 168

	dout("crush_decode %p to %p len %d\n", *p, end, (int)(end - *p));

	c = kzalloc(sizeof(*c), GFP_NOFS);
	if (c == NULL)
		return ERR_PTR(-ENOMEM);

S
Sage Weil 已提交
169 170 171 172 173
        /* set tunables to default values */
        c->choose_local_tries = 2;
        c->choose_local_fallback_tries = 5;
        c->choose_total_tries = 19;

S
Sage Weil 已提交
174
	ceph_decode_need(p, end, 4*sizeof(u32), bad);
175
	magic = ceph_decode_32(p);
S
Sage Weil 已提交
176 177
	if (magic != CRUSH_MAGIC) {
		pr_err("crush_decode magic %x != current %x\n",
178
		       (unsigned int)magic, (unsigned int)CRUSH_MAGIC);
S
Sage Weil 已提交
179 180
		goto bad;
	}
181 182 183
	c->max_buckets = ceph_decode_32(p);
	c->max_rules = ceph_decode_32(p);
	c->max_devices = ceph_decode_32(p);
S
Sage Weil 已提交
184 185 186 187 188 189 190 191 192 193 194 195 196 197 198 199 200 201 202 203 204 205 206 207 208 209 210 211 212 213 214 215 216 217 218 219

	c->buckets = kcalloc(c->max_buckets, sizeof(*c->buckets), GFP_NOFS);
	if (c->buckets == NULL)
		goto badmem;
	c->rules = kcalloc(c->max_rules, sizeof(*c->rules), GFP_NOFS);
	if (c->rules == NULL)
		goto badmem;

	/* buckets */
	for (i = 0; i < c->max_buckets; i++) {
		int size = 0;
		u32 alg;
		struct crush_bucket *b;

		ceph_decode_32_safe(p, end, alg, bad);
		if (alg == 0) {
			c->buckets[i] = NULL;
			continue;
		}
		dout("crush_decode bucket %d off %x %p to %p\n",
		     i, (int)(*p-start), *p, end);

		switch (alg) {
		case CRUSH_BUCKET_UNIFORM:
			size = sizeof(struct crush_bucket_uniform);
			break;
		case CRUSH_BUCKET_LIST:
			size = sizeof(struct crush_bucket_list);
			break;
		case CRUSH_BUCKET_TREE:
			size = sizeof(struct crush_bucket_tree);
			break;
		case CRUSH_BUCKET_STRAW:
			size = sizeof(struct crush_bucket_straw);
			break;
		default:
220
			err = -EINVAL;
S
Sage Weil 已提交
221 222 223 224 225 226 227 228
			goto bad;
		}
		BUG_ON(size == 0);
		b = c->buckets[i] = kzalloc(size, GFP_NOFS);
		if (b == NULL)
			goto badmem;

		ceph_decode_need(p, end, 4*sizeof(u32), bad);
229 230
		b->id = ceph_decode_32(p);
		b->type = ceph_decode_16(p);
231 232
		b->alg = ceph_decode_8(p);
		b->hash = ceph_decode_8(p);
233 234
		b->weight = ceph_decode_32(p);
		b->size = ceph_decode_32(p);
S
Sage Weil 已提交
235 236 237 238 239 240 241 242 243 244 245 246 247 248

		dout("crush_decode bucket size %d off %x %p to %p\n",
		     b->size, (int)(*p-start), *p, end);

		b->items = kcalloc(b->size, sizeof(__s32), GFP_NOFS);
		if (b->items == NULL)
			goto badmem;
		b->perm = kcalloc(b->size, sizeof(u32), GFP_NOFS);
		if (b->perm == NULL)
			goto badmem;
		b->perm_n = 0;

		ceph_decode_need(p, end, b->size*sizeof(u32), bad);
		for (j = 0; j < b->size; j++)
249
			b->items[j] = ceph_decode_32(p);
S
Sage Weil 已提交
250 251 252 253 254 255 256 257 258 259 260 261 262 263 264 265 266 267 268 269 270 271 272 273 274 275 276 277 278 279 280 281 282 283 284 285 286 287 288 289 290 291 292 293 294 295 296 297 298

		switch (b->alg) {
		case CRUSH_BUCKET_UNIFORM:
			err = crush_decode_uniform_bucket(p, end,
				  (struct crush_bucket_uniform *)b);
			if (err < 0)
				goto bad;
			break;
		case CRUSH_BUCKET_LIST:
			err = crush_decode_list_bucket(p, end,
			       (struct crush_bucket_list *)b);
			if (err < 0)
				goto bad;
			break;
		case CRUSH_BUCKET_TREE:
			err = crush_decode_tree_bucket(p, end,
				(struct crush_bucket_tree *)b);
			if (err < 0)
				goto bad;
			break;
		case CRUSH_BUCKET_STRAW:
			err = crush_decode_straw_bucket(p, end,
				(struct crush_bucket_straw *)b);
			if (err < 0)
				goto bad;
			break;
		}
	}

	/* rules */
	dout("rule vec is %p\n", c->rules);
	for (i = 0; i < c->max_rules; i++) {
		u32 yes;
		struct crush_rule *r;

		ceph_decode_32_safe(p, end, yes, bad);
		if (!yes) {
			dout("crush_decode NO rule %d off %x %p to %p\n",
			     i, (int)(*p-start), *p, end);
			c->rules[i] = NULL;
			continue;
		}

		dout("crush_decode rule %d off %x %p to %p\n",
		     i, (int)(*p-start), *p, end);

		/* len */
		ceph_decode_32_safe(p, end, yes, bad);
#if BITS_PER_LONG == 32
299
		err = -EINVAL;
300 301
		if (yes > (ULONG_MAX - sizeof(*r))
			  / sizeof(struct crush_rule_step))
S
Sage Weil 已提交
302 303 304 305 306 307 308 309 310 311 312 313
			goto bad;
#endif
		r = c->rules[i] = kmalloc(sizeof(*r) +
					  yes*sizeof(struct crush_rule_step),
					  GFP_NOFS);
		if (r == NULL)
			goto badmem;
		dout(" rule %d is at %p\n", i, r);
		r->len = yes;
		ceph_decode_copy_safe(p, end, &r->mask, 4, bad); /* 4 u8's */
		ceph_decode_need(p, end, r->len*3*sizeof(u32), bad);
		for (j = 0; j < r->len; j++) {
314 315 316
			r->steps[j].op = ceph_decode_32(p);
			r->steps[j].arg1 = ceph_decode_32(p);
			r->steps[j].arg2 = ceph_decode_32(p);
S
Sage Weil 已提交
317 318 319 320
		}
	}

	/* ignore trailing name maps. */
S
Sage Weil 已提交
321 322 323 324 325 326 327 328 329 330 331 332 333 334 335 336 337
        for (num_name_maps = 0; num_name_maps < 3; num_name_maps++) {
                err = skip_name_map(p, end);
                if (err < 0)
                        goto done;
        }

        /* tunables */
        ceph_decode_need(p, end, 3*sizeof(u32), done);
        c->choose_local_tries = ceph_decode_32(p);
        c->choose_local_fallback_tries =  ceph_decode_32(p);
        c->choose_total_tries = ceph_decode_32(p);
        dout("crush decode tunable choose_local_tries = %d",
             c->choose_local_tries);
        dout("crush decode tunable choose_local_fallback_tries = %d",
             c->choose_local_fallback_tries);
        dout("crush decode tunable choose_total_tries = %d",
             c->choose_total_tries);
S
Sage Weil 已提交
338

S
Sage Weil 已提交
339
done:
S
Sage Weil 已提交
340 341 342 343 344 345 346 347 348 349 350 351
	dout("crush_decode success\n");
	return c;

badmem:
	err = -ENOMEM;
bad:
	dout("crush_decode fail %d\n", err);
	crush_destroy(c);
	return ERR_PTR(err);
}

/*
352 353
 * rbtree of pg_mapping for handling pg_temp (explicit mapping of pgid
 * to a set of osds)
S
Sage Weil 已提交
354
 */
355 356 357 358 359 360 361 362 363 364 365 366
static int pgid_cmp(struct ceph_pg l, struct ceph_pg r)
{
	u64 a = *(u64 *)&l;
	u64 b = *(u64 *)&r;

	if (a < b)
		return -1;
	if (a > b)
		return 1;
	return 0;
}

367 368
static int __insert_pg_mapping(struct ceph_pg_mapping *new,
			       struct rb_root *root)
S
Sage Weil 已提交
369 370 371 372
{
	struct rb_node **p = &root->rb_node;
	struct rb_node *parent = NULL;
	struct ceph_pg_mapping *pg = NULL;
373
	int c;
S
Sage Weil 已提交
374

S
Sage Weil 已提交
375
	dout("__insert_pg_mapping %llx %p\n", *(u64 *)&new->pgid, new);
S
Sage Weil 已提交
376 377 378
	while (*p) {
		parent = *p;
		pg = rb_entry(parent, struct ceph_pg_mapping, node);
379 380
		c = pgid_cmp(new->pgid, pg->pgid);
		if (c < 0)
S
Sage Weil 已提交
381
			p = &(*p)->rb_left;
382
		else if (c > 0)
S
Sage Weil 已提交
383 384
			p = &(*p)->rb_right;
		else
385
			return -EEXIST;
S
Sage Weil 已提交
386 387 388 389
	}

	rb_link_node(&new->node, parent, p);
	rb_insert_color(&new->node, root);
390
	return 0;
S
Sage Weil 已提交
391 392
}

393 394 395 396 397 398 399 400 401 402
static struct ceph_pg_mapping *__lookup_pg_mapping(struct rb_root *root,
						   struct ceph_pg pgid)
{
	struct rb_node *n = root->rb_node;
	struct ceph_pg_mapping *pg;
	int c;

	while (n) {
		pg = rb_entry(n, struct ceph_pg_mapping, node);
		c = pgid_cmp(pgid, pg->pgid);
S
Sage Weil 已提交
403
		if (c < 0) {
404
			n = n->rb_left;
S
Sage Weil 已提交
405
		} else if (c > 0) {
406
			n = n->rb_right;
S
Sage Weil 已提交
407 408 409
		} else {
			dout("__lookup_pg_mapping %llx got %p\n",
			     *(u64 *)&pgid, pg);
410
			return pg;
S
Sage Weil 已提交
411
		}
412 413 414 415
	}
	return NULL;
}

S
Sage Weil 已提交
416 417 418 419 420 421 422 423 424 425 426 427 428 429
static int __remove_pg_mapping(struct rb_root *root, struct ceph_pg pgid)
{
	struct ceph_pg_mapping *pg = __lookup_pg_mapping(root, pgid);

	if (pg) {
		dout("__remove_pg_mapping %llx %p\n", *(u64 *)&pgid, pg);
		rb_erase(&pg->node, root);
		kfree(pg);
		return 0;
	}
	dout("__remove_pg_mapping %llx dne\n", *(u64 *)&pgid);
	return -ENOENT;
}

430 431 432 433 434 435 436 437 438 439 440 441 442 443 444 445 446 447 448 449 450 451 452 453 454 455 456 457 458 459 460 461 462 463 464 465 466 467 468 469 470 471
/*
 * rbtree of pg pool info
 */
static int __insert_pg_pool(struct rb_root *root, struct ceph_pg_pool_info *new)
{
	struct rb_node **p = &root->rb_node;
	struct rb_node *parent = NULL;
	struct ceph_pg_pool_info *pi = NULL;

	while (*p) {
		parent = *p;
		pi = rb_entry(parent, struct ceph_pg_pool_info, node);
		if (new->id < pi->id)
			p = &(*p)->rb_left;
		else if (new->id > pi->id)
			p = &(*p)->rb_right;
		else
			return -EEXIST;
	}

	rb_link_node(&new->node, parent, p);
	rb_insert_color(&new->node, root);
	return 0;
}

static struct ceph_pg_pool_info *__lookup_pg_pool(struct rb_root *root, int id)
{
	struct ceph_pg_pool_info *pi;
	struct rb_node *n = root->rb_node;

	while (n) {
		pi = rb_entry(n, struct ceph_pg_pool_info, node);
		if (id < pi->id)
			n = n->rb_left;
		else if (id > pi->id)
			n = n->rb_right;
		else
			return pi;
	}
	return NULL;
}

Y
Yehuda Sadeh 已提交
472 473 474 475 476 477 478 479 480 481 482 483
int ceph_pg_poolid_by_name(struct ceph_osdmap *map, const char *name)
{
	struct rb_node *rbp;

	for (rbp = rb_first(&map->pg_pools); rbp; rbp = rb_next(rbp)) {
		struct ceph_pg_pool_info *pi =
			rb_entry(rbp, struct ceph_pg_pool_info, node);
		if (pi->name && strcmp(pi->name, name) == 0)
			return pi->id;
	}
	return -ENOENT;
}
484
EXPORT_SYMBOL(ceph_pg_poolid_by_name);
Y
Yehuda Sadeh 已提交
485

486 487 488 489 490 491 492
static void __remove_pg_pool(struct rb_root *root, struct ceph_pg_pool_info *pi)
{
	rb_erase(&pi->node, root);
	kfree(pi->name);
	kfree(pi);
}

S
Sage Weil 已提交
493
static int __decode_pool(void **p, void *end, struct ceph_pg_pool_info *pi)
494
{
495
	unsigned int n, m;
S
Sage Weil 已提交
496

497 498
	ceph_decode_copy(p, &pi->v, sizeof(pi->v));
	calc_pg_masks(pi);
S
Sage Weil 已提交
499 500 501 502 503 504 505 506 507 508 509 510 511

	/* num_snaps * snap_info_t */
	n = le32_to_cpu(pi->v.num_snaps);
	while (n--) {
		ceph_decode_need(p, end, sizeof(u64) + 1 + sizeof(u64) +
				 sizeof(struct ceph_timespec), bad);
		*p += sizeof(u64) +       /* key */
			1 + sizeof(u64) + /* u8, snapid */
			sizeof(struct ceph_timespec);
		m = ceph_decode_32(p);    /* snap name */
		*p += m;
	}

512
	*p += le32_to_cpu(pi->v.num_removed_snap_intervals) * sizeof(u64) * 2;
S
Sage Weil 已提交
513 514 515 516
	return 0;

bad:
	return -EINVAL;
517 518
}

519 520 521 522 523 524 525 526 527 528 529
static int __decode_pool_names(void **p, void *end, struct ceph_osdmap *map)
{
	struct ceph_pg_pool_info *pi;
	u32 num, len, pool;

	ceph_decode_32_safe(p, end, num, bad);
	dout(" %d pool names\n", num);
	while (num--) {
		ceph_decode_32_safe(p, end, pool, bad);
		ceph_decode_32_safe(p, end, len, bad);
		dout("  pool %d len %d\n", pool, len);
530
		ceph_decode_need(p, end, len, bad);
531 532
		pi = __lookup_pg_pool(&map->pg_pools, pool);
		if (pi) {
533 534 535 536
			char *name = kstrndup(*p, len, GFP_NOFS);

			if (!name)
				return -ENOMEM;
537
			kfree(pi->name);
538 539
			pi->name = name;
			dout("  name is %s\n", pi->name);
540 541 542 543 544 545 546 547 548 549 550 551 552 553 554 555 556 557 558 559 560 561 562 563 564 565 566 567 568 569 570 571 572 573 574 575 576 577 578 579 580 581 582 583 584 585 586 587 588 589 590 591 592 593 594 595 596 597 598 599 600 601 602 603 604 605 606 607 608 609 610 611
		}
		*p += len;
	}
	return 0;

bad:
	return -EINVAL;
}

/*
 * osd map
 */
void ceph_osdmap_destroy(struct ceph_osdmap *map)
{
	dout("osdmap_destroy %p\n", map);
	if (map->crush)
		crush_destroy(map->crush);
	while (!RB_EMPTY_ROOT(&map->pg_temp)) {
		struct ceph_pg_mapping *pg =
			rb_entry(rb_first(&map->pg_temp),
				 struct ceph_pg_mapping, node);
		rb_erase(&pg->node, &map->pg_temp);
		kfree(pg);
	}
	while (!RB_EMPTY_ROOT(&map->pg_pools)) {
		struct ceph_pg_pool_info *pi =
			rb_entry(rb_first(&map->pg_pools),
				 struct ceph_pg_pool_info, node);
		__remove_pg_pool(&map->pg_pools, pi);
	}
	kfree(map->osd_state);
	kfree(map->osd_weight);
	kfree(map->osd_addr);
	kfree(map);
}

/*
 * adjust max osd value.  reallocate arrays.
 */
static int osdmap_set_max_osd(struct ceph_osdmap *map, int max)
{
	u8 *state;
	struct ceph_entity_addr *addr;
	u32 *weight;

	state = kcalloc(max, sizeof(*state), GFP_NOFS);
	addr = kcalloc(max, sizeof(*addr), GFP_NOFS);
	weight = kcalloc(max, sizeof(*weight), GFP_NOFS);
	if (state == NULL || addr == NULL || weight == NULL) {
		kfree(state);
		kfree(addr);
		kfree(weight);
		return -ENOMEM;
	}

	/* copy old? */
	if (map->osd_state) {
		memcpy(state, map->osd_state, map->max_osd*sizeof(*state));
		memcpy(addr, map->osd_addr, map->max_osd*sizeof(*addr));
		memcpy(weight, map->osd_weight, map->max_osd*sizeof(*weight));
		kfree(map->osd_state);
		kfree(map->osd_addr);
		kfree(map->osd_weight);
	}

	map->osd_state = state;
	map->osd_weight = weight;
	map->osd_addr = addr;
	map->max_osd = max;
	return 0;
}

S
Sage Weil 已提交
612 613 614 615 616 617 618 619
/*
 * decode a full map.
 */
struct ceph_osdmap *osdmap_decode(void **p, void *end)
{
	struct ceph_osdmap *map;
	u16 version;
	u32 len, max, i;
620
	u8 ev;
S
Sage Weil 已提交
621 622
	int err = -EINVAL;
	void *start = *p;
623
	struct ceph_pg_pool_info *pi;
S
Sage Weil 已提交
624 625 626 627 628 629 630 631 632

	dout("osdmap_decode %p to %p len %d\n", *p, end, (int)(end - *p));

	map = kzalloc(sizeof(*map), GFP_NOFS);
	if (map == NULL)
		return ERR_PTR(-ENOMEM);
	map->pg_temp = RB_ROOT;

	ceph_decode_16_safe(p, end, version, bad);
S
Sage Weil 已提交
633 634 635 636 637
	if (version > CEPH_OSDMAP_VERSION) {
		pr_warning("got unknown v %d > %d of osdmap\n", version,
			   CEPH_OSDMAP_VERSION);
		goto bad;
	}
S
Sage Weil 已提交
638 639 640

	ceph_decode_need(p, end, 2*sizeof(u64)+6*sizeof(u32), bad);
	ceph_decode_copy(p, &map->fsid, sizeof(map->fsid));
641
	map->epoch = ceph_decode_32(p);
S
Sage Weil 已提交
642 643 644 645 646
	ceph_decode_copy(p, &map->created, sizeof(map->created));
	ceph_decode_copy(p, &map->modified, sizeof(map->modified));

	ceph_decode_32_safe(p, end, max, bad);
	while (max--) {
647
		ceph_decode_need(p, end, 4 + 1 + sizeof(pi->v), bad);
S
Sage Weil 已提交
648
		err = -ENOMEM;
649
		pi = kzalloc(sizeof(*pi), GFP_NOFS);
650
		if (!pi)
S
Sage Weil 已提交
651
			goto bad;
652
		pi->id = ceph_decode_32(p);
S
Sage Weil 已提交
653
		err = -EINVAL;
654
		ev = ceph_decode_8(p); /* encoding version */
S
Sage Weil 已提交
655 656 657
		if (ev > CEPH_PG_POOL_VERSION) {
			pr_warning("got unknown v %d > %d of ceph_pg_pool\n",
				   ev, CEPH_PG_POOL_VERSION);
D
Dan Carpenter 已提交
658
			kfree(pi);
S
Sage Weil 已提交
659 660
			goto bad;
		}
S
Sage Weil 已提交
661
		err = __decode_pool(p, end, pi);
662 663
		if (err < 0) {
			kfree(pi);
S
Sage Weil 已提交
664
			goto bad;
665
		}
666
		__insert_pg_pool(&map->pg_pools, pi);
S
Sage Weil 已提交
667
	}
668

S
Sage Weil 已提交
669 670 671 672 673 674 675
	if (version >= 5) {
		err = __decode_pool_names(p, end, map);
		if (err < 0) {
			dout("fail to decode pool names");
			goto bad;
		}
	}
676

677
	ceph_decode_32_safe(p, end, map->pool_max, bad);
S
Sage Weil 已提交
678 679 680

	ceph_decode_32_safe(p, end, map->flags, bad);

681
	max = ceph_decode_32(p);
S
Sage Weil 已提交
682 683 684 685 686 687 688 689 690 691 692 693 694 695 696 697 698

	/* (re)alloc osd arrays */
	err = osdmap_set_max_osd(map, max);
	if (err < 0)
		goto bad;
	dout("osdmap_decode max_osd = %d\n", map->max_osd);

	/* osds */
	err = -EINVAL;
	ceph_decode_need(p, end, 3*sizeof(u32) +
			 map->max_osd*(1 + sizeof(*map->osd_weight) +
				       sizeof(*map->osd_addr)), bad);
	*p += 4; /* skip length field (should match max) */
	ceph_decode_copy(p, map->osd_state, map->max_osd);

	*p += 4; /* skip length field (should match max) */
	for (i = 0; i < map->max_osd; i++)
699
		map->osd_weight[i] = ceph_decode_32(p);
S
Sage Weil 已提交
700 701 702

	*p += 4; /* skip length field (should match max) */
	ceph_decode_copy(p, map->osd_addr, map->max_osd*sizeof(*map->osd_addr));
703 704
	for (i = 0; i < map->max_osd; i++)
		ceph_decode_addr(&map->osd_addr[i]);
S
Sage Weil 已提交
705 706 707 708 709

	/* pg_temp */
	ceph_decode_32_safe(p, end, len, bad);
	for (i = 0; i < len; i++) {
		int n, j;
710
		struct ceph_pg pgid;
S
Sage Weil 已提交
711 712 713
		struct ceph_pg_mapping *pg;

		ceph_decode_need(p, end, sizeof(u32) + sizeof(u64), bad);
714
		ceph_decode_copy(p, &pgid, sizeof(pgid));
715
		n = ceph_decode_32(p);
716 717 718
		err = -EINVAL;
		if (n > (UINT_MAX - sizeof(*pg)) / sizeof(u32))
			goto bad;
S
Sage Weil 已提交
719
		ceph_decode_need(p, end, n * sizeof(u32), bad);
720
		err = -ENOMEM;
S
Sage Weil 已提交
721
		pg = kmalloc(sizeof(*pg) + n*sizeof(u32), GFP_NOFS);
722
		if (!pg)
S
Sage Weil 已提交
723 724 725 726
			goto bad;
		pg->pgid = pgid;
		pg->len = n;
		for (j = 0; j < n; j++)
727
			pg->osds[j] = ceph_decode_32(p);
S
Sage Weil 已提交
728

729 730 731
		err = __insert_pg_mapping(pg, &map->pg_temp);
		if (err)
			goto bad;
732
		dout(" added pg_temp %llx len %d\n", *(u64 *)&pgid, len);
S
Sage Weil 已提交
733 734 735 736 737 738 739 740 741 742 743 744 745 746 747 748 749 750 751 752 753 754
	}

	/* crush */
	ceph_decode_32_safe(p, end, len, bad);
	dout("osdmap_decode crush len %d from off 0x%x\n", len,
	     (int)(*p - start));
	ceph_decode_need(p, end, len, bad);
	map->crush = crush_decode(*p, end);
	*p += len;
	if (IS_ERR(map->crush)) {
		err = PTR_ERR(map->crush);
		map->crush = NULL;
		goto bad;
	}

	/* ignore the rest of the map */
	*p = end;

	dout("osdmap_decode done %p %p\n", *p, end);
	return map;

bad:
S
Sage Weil 已提交
755
	dout("osdmap_decode fail err %d\n", err);
S
Sage Weil 已提交
756 757 758 759 760 761 762 763 764 765 766 767 768 769 770 771
	ceph_osdmap_destroy(map);
	return ERR_PTR(err);
}

/*
 * decode and apply an incremental map update.
 */
struct ceph_osdmap *osdmap_apply_incremental(void **p, void *end,
					     struct ceph_osdmap *map,
					     struct ceph_messenger *msgr)
{
	struct crush_map *newcrush = NULL;
	struct ceph_fsid fsid;
	u32 epoch = 0;
	struct ceph_timespec modified;
	u32 len, pool;
772
	__s32 new_pool_max, new_flags, max;
S
Sage Weil 已提交
773 774 775 776 777
	void *start = *p;
	int err = -EINVAL;
	u16 version;

	ceph_decode_16_safe(p, end, version, bad);
S
Sage Weil 已提交
778 779 780 781 782
	if (version > CEPH_OSDMAP_INC_VERSION) {
		pr_warning("got unknown v %d > %d of inc osdmap\n", version,
			   CEPH_OSDMAP_INC_VERSION);
		goto bad;
	}
S
Sage Weil 已提交
783 784 785 786

	ceph_decode_need(p, end, sizeof(fsid)+sizeof(modified)+2*sizeof(u32),
			 bad);
	ceph_decode_copy(p, &fsid, sizeof(fsid));
787
	epoch = ceph_decode_32(p);
S
Sage Weil 已提交
788 789
	BUG_ON(epoch != map->epoch+1);
	ceph_decode_copy(p, &modified, sizeof(modified));
790
	new_pool_max = ceph_decode_32(p);
791
	new_flags = ceph_decode_32(p);
S
Sage Weil 已提交
792 793 794 795 796 797

	/* full map? */
	ceph_decode_32_safe(p, end, len, bad);
	if (len > 0) {
		dout("apply_incremental full map len %d, %p to %p\n",
		     len, *p, end);
798
		return osdmap_decode(p, min(*p+len, end));
S
Sage Weil 已提交
799 800 801 802 803 804 805 806 807
	}

	/* new crush? */
	ceph_decode_32_safe(p, end, len, bad);
	if (len > 0) {
		dout("apply_incremental new crush map len %d, %p to %p\n",
		     len, *p, end);
		newcrush = crush_decode(*p, min(*p+len, end));
		if (IS_ERR(newcrush))
J
Julia Lawall 已提交
808
			return ERR_CAST(newcrush);
S
Sage Weil 已提交
809
		*p += len;
S
Sage Weil 已提交
810 811 812 813 814
	}

	/* new flags? */
	if (new_flags >= 0)
		map->flags = new_flags;
815 816
	if (new_pool_max >= 0)
		map->pool_max = new_pool_max;
S
Sage Weil 已提交
817 818 819 820

	ceph_decode_need(p, end, 5*sizeof(u32), bad);

	/* new max? */
821
	max = ceph_decode_32(p);
S
Sage Weil 已提交
822 823 824 825 826 827 828
	if (max >= 0) {
		err = osdmap_set_max_osd(map, max);
		if (err < 0)
			goto bad;
	}

	map->epoch++;
829
	map->modified = modified;
S
Sage Weil 已提交
830 831 832 833 834 835 836 837 838 839
	if (newcrush) {
		if (map->crush)
			crush_destroy(map->crush);
		map->crush = newcrush;
		newcrush = NULL;
	}

	/* new_pool */
	ceph_decode_32_safe(p, end, len, bad);
	while (len--) {
840
		__u8 ev;
841
		struct ceph_pg_pool_info *pi;
842

S
Sage Weil 已提交
843
		ceph_decode_32_safe(p, end, pool, bad);
844
		ceph_decode_need(p, end, 1 + sizeof(pi->v), bad);
845
		ev = ceph_decode_8(p);  /* encoding version */
S
Sage Weil 已提交
846 847 848
		if (ev > CEPH_PG_POOL_VERSION) {
			pr_warning("got unknown v %d > %d of ceph_pg_pool\n",
				   ev, CEPH_PG_POOL_VERSION);
S
Sage Weil 已提交
849
			err = -EINVAL;
S
Sage Weil 已提交
850 851
			goto bad;
		}
852 853
		pi = __lookup_pg_pool(&map->pg_pools, pool);
		if (!pi) {
854
			pi = kzalloc(sizeof(*pi), GFP_NOFS);
855 856 857 858 859 860 861
			if (!pi) {
				err = -ENOMEM;
				goto bad;
			}
			pi->id = pool;
			__insert_pg_pool(&map->pg_pools, pi);
		}
S
Sage Weil 已提交
862 863 864
		err = __decode_pool(p, end, pi);
		if (err < 0)
			goto bad;
S
Sage Weil 已提交
865
	}
S
Sage Weil 已提交
866 867 868 869 870
	if (version >= 5) {
		err = __decode_pool_names(p, end, map);
		if (err < 0)
			goto bad;
	}
S
Sage Weil 已提交
871

872
	/* old_pool */
S
Sage Weil 已提交
873
	ceph_decode_32_safe(p, end, len, bad);
874 875 876 877 878
	while (len--) {
		struct ceph_pg_pool_info *pi;

		ceph_decode_32_safe(p, end, pool, bad);
		pi = __lookup_pg_pool(&map->pg_pools, pool);
879 880
		if (pi)
			__remove_pg_pool(&map->pg_pools, pi);
881
	}
S
Sage Weil 已提交
882 883 884 885 886 887 888 889 890

	/* new_up */
	err = -EINVAL;
	ceph_decode_32_safe(p, end, len, bad);
	while (len--) {
		u32 osd;
		struct ceph_entity_addr addr;
		ceph_decode_32_safe(p, end, osd, bad);
		ceph_decode_copy_safe(p, end, &addr, sizeof(addr), bad);
891
		ceph_decode_addr(&addr);
S
Sage Weil 已提交
892 893 894 895 896 897
		pr_info("osd%d up\n", osd);
		BUG_ON(osd >= map->max_osd);
		map->osd_state[osd] |= CEPH_OSD_UP;
		map->osd_addr[osd] = addr;
	}

898
	/* new_state */
S
Sage Weil 已提交
899 900 901
	ceph_decode_32_safe(p, end, len, bad);
	while (len--) {
		u32 osd;
902
		u8 xorstate;
S
Sage Weil 已提交
903
		ceph_decode_32_safe(p, end, osd, bad);
904
		xorstate = **(u8 **)p;
S
Sage Weil 已提交
905
		(*p)++;  /* clean flag */
906 907 908 909
		if (xorstate == 0)
			xorstate = CEPH_OSD_UP;
		if (xorstate & CEPH_OSD_UP)
			pr_info("osd%d down\n", osd);
S
Sage Weil 已提交
910
		if (osd < map->max_osd)
911
			map->osd_state[osd] ^= xorstate;
S
Sage Weil 已提交
912 913 914 915 916 917 918
	}

	/* new_weight */
	ceph_decode_32_safe(p, end, len, bad);
	while (len--) {
		u32 osd, off;
		ceph_decode_need(p, end, sizeof(u32)*2, bad);
919 920
		osd = ceph_decode_32(p);
		off = ceph_decode_32(p);
S
Sage Weil 已提交
921 922 923 924 925 926 927 928 929 930 931 932
		pr_info("osd%d weight 0x%x %s\n", osd, off,
		     off == CEPH_OSD_IN ? "(in)" :
		     (off == CEPH_OSD_OUT ? "(out)" : ""));
		if (osd < map->max_osd)
			map->osd_weight[osd] = off;
	}

	/* new_pg_temp */
	ceph_decode_32_safe(p, end, len, bad);
	while (len--) {
		struct ceph_pg_mapping *pg;
		int j;
933
		struct ceph_pg pgid;
S
Sage Weil 已提交
934 935
		u32 pglen;
		ceph_decode_need(p, end, sizeof(u64) + sizeof(u32), bad);
936
		ceph_decode_copy(p, &pgid, sizeof(pgid));
937
		pglen = ceph_decode_32(p);
S
Sage Weil 已提交
938 939 940

		if (pglen) {
			ceph_decode_need(p, end, pglen*sizeof(u32), bad);
S
Sage Weil 已提交
941 942 943 944 945

			/* removing existing (if any) */
			(void) __remove_pg_mapping(&map->pg_temp, pgid);

			/* insert */
S
Sage Weil 已提交
946 947
			err = -EINVAL;
			if (pglen > (UINT_MAX - sizeof(*pg)) / sizeof(u32))
948
				goto bad;
S
Sage Weil 已提交
949
			err = -ENOMEM;
S
Sage Weil 已提交
950
			pg = kmalloc(sizeof(*pg) + sizeof(u32)*pglen, GFP_NOFS);
S
Sage Weil 已提交
951
			if (!pg)
S
Sage Weil 已提交
952 953 954
				goto bad;
			pg->pgid = pgid;
			pg->len = pglen;
955
			for (j = 0; j < pglen; j++)
956
				pg->osds[j] = ceph_decode_32(p);
957
			err = __insert_pg_mapping(pg, &map->pg_temp);
958 959
			if (err) {
				kfree(pg);
960
				goto bad;
961
			}
962 963
			dout(" added pg_temp %llx len %d\n", *(u64 *)&pgid,
			     pglen);
S
Sage Weil 已提交
964 965 966
		} else {
			/* remove */
			__remove_pg_mapping(&map->pg_temp, pgid);
S
Sage Weil 已提交
967 968 969 970 971 972 973 974 975 976
		}
	}

	/* ignore the rest */
	*p = end;
	return map;

bad:
	pr_err("corrupt inc osdmap epoch %d off %d (%p of %p-%p)\n",
	       epoch, (int)(*p - start), *p, start, end);
977 978 979
	print_hex_dump(KERN_DEBUG, "osdmap: ",
		       DUMP_PREFIX_OFFSET, 16, 1,
		       start, end - start, true);
S
Sage Weil 已提交
980 981 982 983 984 985 986 987 988 989 990 991 992 993 994 995
	if (newcrush)
		crush_destroy(newcrush);
	return ERR_PTR(err);
}




/*
 * calculate file layout from given offset, length.
 * fill in correct oid, logical length, and object extent
 * offset, length.
 *
 * for now, we write only a single su, until we can
 * pass a stride back to the caller.
 */
S
Sage Weil 已提交
996
int ceph_calc_file_object_mapping(struct ceph_file_layout *layout,
S
Sage Weil 已提交
997
				   u64 off, u64 *plen,
998
				   u64 *ono,
S
Sage Weil 已提交
999 1000 1001 1002 1003 1004 1005
				   u64 *oxoff, u64 *oxlen)
{
	u32 osize = le32_to_cpu(layout->fl_object_size);
	u32 su = le32_to_cpu(layout->fl_stripe_unit);
	u32 sc = le32_to_cpu(layout->fl_stripe_count);
	u32 bl, stripeno, stripepos, objsetno;
	u32 su_per_object;
1006
	u64 t, su_offset;
S
Sage Weil 已提交
1007 1008 1009

	dout("mapping %llu~%llu  osize %u fl_su %u\n", off, *plen,
	     osize, su);
S
Sage Weil 已提交
1010 1011
	if (su == 0 || sc == 0)
		goto invalid;
1012
	su_per_object = osize / su;
S
Sage Weil 已提交
1013 1014
	if (su_per_object == 0)
		goto invalid;
S
Sage Weil 已提交
1015 1016 1017
	dout("osize %u / su %u = su_per_object %u\n", osize, su,
	     su_per_object);

S
Sage Weil 已提交
1018 1019 1020
	if ((su & ~PAGE_MASK) != 0)
		goto invalid;

S
Sage Weil 已提交
1021 1022 1023 1024 1025 1026 1027 1028 1029 1030
	/* bl = *off / su; */
	t = off;
	do_div(t, su);
	bl = t;
	dout("off %llu / su %u = bl %u\n", off, su, bl);

	stripeno = bl / sc;
	stripepos = bl % sc;
	objsetno = stripeno / su_per_object;

1031
	*ono = objsetno * sc + stripepos;
1032
	dout("objset %u * sc %u = ono %u\n", objsetno, sc, (unsigned int)*ono);
1033 1034

	/* *oxoff = *off % layout->fl_stripe_unit;  # offset in su */
S
Sage Weil 已提交
1035
	t = off;
1036 1037 1038 1039 1040 1041 1042 1043 1044
	su_offset = do_div(t, su);
	*oxoff = su_offset + (stripeno % su_per_object) * su;

	/*
	 * Calculate the length of the extent being written to the selected
	 * object. This is the minimum of the full length requested (plen) or
	 * the remainder of the current stripe being written to.
	 */
	*oxlen = min_t(u64, *plen, su - su_offset);
S
Sage Weil 已提交
1045 1046 1047
	*plen = *oxlen;

	dout(" obj extent %llu~%llu\n", *oxoff, *oxlen);
S
Sage Weil 已提交
1048 1049 1050 1051 1052 1053 1054 1055
	return 0;

invalid:
	dout(" invalid layout\n");
	*ono = 0;
	*oxoff = 0;
	*oxlen = 0;
	return -EINVAL;
S
Sage Weil 已提交
1056
}
1057
EXPORT_SYMBOL(ceph_calc_file_object_mapping);
S
Sage Weil 已提交
1058 1059 1060 1061 1062 1063 1064 1065 1066 1067

/*
 * calculate an object layout (i.e. pgid) from an oid,
 * file_layout, and osdmap
 */
int ceph_calc_object_layout(struct ceph_object_layout *ol,
			    const char *oid,
			    struct ceph_file_layout *fl,
			    struct ceph_osdmap *osdmap)
{
1068
	unsigned int num, num_mask;
1069
	struct ceph_pg pgid;
S
Sage Weil 已提交
1070 1071
	int poolid = le32_to_cpu(fl->fl_pg_pool);
	struct ceph_pg_pool_info *pool;
1072
	unsigned int ps;
S
Sage Weil 已提交
1073

1074
	BUG_ON(!osdmap);
S
Sage Weil 已提交
1075

1076 1077 1078
	pool = __lookup_pg_pool(&osdmap->pg_pools, poolid);
	if (!pool)
		return -EIO;
1079
	ps = ceph_str_hash(pool->v.object_hash, oid, strlen(oid));
1080 1081
	num = le32_to_cpu(pool->v.pg_num);
	num_mask = pool->pg_num_mask;
S
Sage Weil 已提交
1082

1083
	pgid.ps = cpu_to_le16(ps);
1084
	pgid.preferred = cpu_to_le16(-1);
1085
	pgid.pool = fl->fl_pg_pool;
1086
	dout("calc_object_layout '%s' pgid %d.%x\n", oid, poolid, ps);
S
Sage Weil 已提交
1087

1088
	ol->ol_pgid = pgid;
S
Sage Weil 已提交
1089 1090 1091
	ol->ol_stripe_unit = fl->fl_object_stripe_unit;
	return 0;
}
1092
EXPORT_SYMBOL(ceph_calc_object_layout);
S
Sage Weil 已提交
1093 1094 1095 1096 1097

/*
 * Calculate raw osd vector for the given pgid.  Return pointer to osd
 * array, or NULL on failure.
 */
1098
static int *calc_pg_raw(struct ceph_osdmap *osdmap, struct ceph_pg pgid,
S
Sage Weil 已提交
1099 1100 1101 1102 1103
			int *osds, int *num)
{
	struct ceph_pg_mapping *pg;
	struct ceph_pg_pool_info *pool;
	int ruleno;
1104
	unsigned int poolid, ps, pps, t, r;
S
Sage Weil 已提交
1105

1106 1107 1108 1109 1110 1111 1112
	poolid = le32_to_cpu(pgid.pool);
	ps = le16_to_cpu(pgid.ps);

	pool = __lookup_pg_pool(&osdmap->pg_pools, poolid);
	if (!pool)
		return NULL;

S
Sage Weil 已提交
1113
	/* pg_temp? */
1114 1115
	t = ceph_stable_mod(ps, le32_to_cpu(pool->v.pg_num),
			    pool->pgp_num_mask);
1116
	pgid.ps = cpu_to_le16(t);
1117 1118 1119 1120
	pg = __lookup_pg_mapping(&osdmap->pg_temp, pgid);
	if (pg) {
		*num = pg->len;
		return pg->osds;
S
Sage Weil 已提交
1121 1122 1123 1124 1125 1126
	}

	/* crush */
	ruleno = crush_find_rule(osdmap->crush, pool->v.crush_ruleset,
				 pool->v.type, pool->v.size);
	if (ruleno < 0) {
1127 1128 1129
		pr_err("no crush rule pool %d ruleset %d type %d size %d\n",
		       poolid, pool->v.crush_ruleset, pool->v.type,
		       pool->v.size);
S
Sage Weil 已提交
1130 1131 1132
		return NULL;
	}

1133 1134 1135
	pps = ceph_stable_mod(ps,
			      le32_to_cpu(pool->v.pgp_num),
			      pool->pgp_num_mask);
1136
	pps += poolid;
S
Sage Weil 已提交
1137 1138 1139 1140 1141 1142 1143 1144 1145 1146
	r = crush_do_rule(osdmap->crush, ruleno, pps, osds,
			  min_t(int, pool->v.size, *num),
			  osdmap->osd_weight);
	if (r < 0) {
		pr_err("error %d from crush rule: pool %d ruleset %d type %d"
		       " size %d\n", r, poolid, pool->v.crush_ruleset,
		       pool->v.type, pool->v.size);
		return NULL;
	}
	*num = r;
S
Sage Weil 已提交
1147 1148 1149
	return osds;
}

1150 1151 1152 1153 1154 1155 1156 1157 1158 1159 1160 1161 1162 1163 1164 1165 1166 1167 1168 1169 1170
/*
 * Return acting set for given pgid.
 */
int ceph_calc_pg_acting(struct ceph_osdmap *osdmap, struct ceph_pg pgid,
			int *acting)
{
	int rawosds[CEPH_PG_MAX_SIZE], *osds;
	int i, o, num = CEPH_PG_MAX_SIZE;

	osds = calc_pg_raw(osdmap, pgid, rawosds, &num);
	if (!osds)
		return -1;

	/* primary is first up osd */
	o = 0;
	for (i = 0; i < num; i++)
		if (ceph_osd_is_up(osdmap, osds[i]))
			acting[o++] = osds[i];
	return o;
}

S
Sage Weil 已提交
1171 1172 1173
/*
 * Return primary osd for given pgid, or -1 if none.
 */
1174
int ceph_calc_pg_primary(struct ceph_osdmap *osdmap, struct ceph_pg pgid)
S
Sage Weil 已提交
1175
{
1176 1177
	int rawosds[CEPH_PG_MAX_SIZE], *osds;
	int i, num = CEPH_PG_MAX_SIZE;
S
Sage Weil 已提交
1178 1179 1180 1181 1182 1183 1184

	osds = calc_pg_raw(osdmap, pgid, rawosds, &num);
	if (!osds)
		return -1;

	/* primary is first up osd */
	for (i = 0; i < num; i++)
1185
		if (ceph_osd_is_up(osdmap, osds[i]))
S
Sage Weil 已提交
1186 1187 1188
			return osds[i];
	return -1;
}
1189
EXPORT_SYMBOL(ceph_calc_pg_primary);