rbd: implement feature checks
[linux-block.git] / drivers / block / rbd.c
CommitLineData
602adf40
YS
1/*
2 rbd.c -- Export ceph rados objects as a Linux block device
3
4
5 based on drivers/block/osdblk.c:
6
7 Copyright 2009 Red Hat, Inc.
8
9 This program is free software; you can redistribute it and/or modify
10 it under the terms of the GNU General Public License as published by
11 the Free Software Foundation.
12
13 This program is distributed in the hope that it will be useful,
14 but WITHOUT ANY WARRANTY; without even the implied warranty of
15 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
16 GNU General Public License for more details.
17
18 You should have received a copy of the GNU General Public License
19 along with this program; see the file COPYING. If not, write to
20 the Free Software Foundation, 675 Mass Ave, Cambridge, MA 02139, USA.
21
22
23
dfc5606d 24 For usage instructions, please refer to:
602adf40 25
dfc5606d 26 Documentation/ABI/testing/sysfs-bus-rbd
602adf40
YS
27
28 */
29
30#include <linux/ceph/libceph.h>
31#include <linux/ceph/osd_client.h>
32#include <linux/ceph/mon_client.h>
33#include <linux/ceph/decode.h>
59c2be1e 34#include <linux/parser.h>
602adf40
YS
35
36#include <linux/kernel.h>
37#include <linux/device.h>
38#include <linux/module.h>
39#include <linux/fs.h>
40#include <linux/blkdev.h>
41
42#include "rbd_types.h"
43
aafb230e
AE
44#define RBD_DEBUG /* Activate rbd_assert() calls */
45
593a9e7b
AE
46/*
47 * The basic unit of block I/O is a sector. It is interpreted in a
48 * number of contexts in Linux (blk, bio, genhd), but the default is
49 * universally 512 bytes. These symbols are just slightly more
50 * meaningful than the bare numbers they represent.
51 */
52#define SECTOR_SHIFT 9
53#define SECTOR_SIZE (1ULL << SECTOR_SHIFT)
54
df111be6
AE
55/* It might be useful to have this defined elsewhere too */
56
57#define U64_MAX ((u64) (~0ULL))
58
f0f8cef5
AE
59#define RBD_DRV_NAME "rbd"
60#define RBD_DRV_NAME_LONG "rbd (rados block device)"
602adf40
YS
61
62#define RBD_MINORS_PER_MAJOR 256 /* max minors per blkdev */
63
602adf40 64#define RBD_MAX_SNAP_NAME_LEN 32
35d489f9 65#define RBD_MAX_SNAP_COUNT 510 /* allows max snapc to fit in 4KB */
602adf40
YS
66#define RBD_MAX_OPT_LEN 1024
67
68#define RBD_SNAP_HEAD_NAME "-"
69
1e130199
AE
70#define RBD_IMAGE_ID_LEN_MAX 64
71#define RBD_OBJ_PREFIX_LEN_MAX 64
589d30e0 72
d889140c
AE
73/* Feature bits */
74
75#define RBD_FEATURE_LAYERING 1
76
77/* Features supported by this (client software) implementation. */
78
79#define RBD_FEATURES_ALL (0)
80
81a89793
AE
81/*
82 * An RBD device name will be "rbd#", where the "rbd" comes from
83 * RBD_DRV_NAME above, and # is a unique integer identifier.
84 * MAX_INT_FORMAT_WIDTH is used in ensuring DEV_NAME_LEN is big
85 * enough to hold all possible device names.
86 */
602adf40 87#define DEV_NAME_LEN 32
81a89793 88#define MAX_INT_FORMAT_WIDTH ((5 * sizeof (int)) / 2 + 1)
602adf40 89
cc0538b6 90#define RBD_READ_ONLY_DEFAULT false
59c2be1e 91
602adf40
YS
92/*
93 * block device image metadata (in-memory version)
94 */
95struct rbd_image_header {
f84344f3 96 /* These four fields never change for a given rbd image */
849b4260 97 char *object_prefix;
34b13184 98 u64 features;
602adf40
YS
99 __u8 obj_order;
100 __u8 crypt_type;
101 __u8 comp_type;
602adf40 102
f84344f3
AE
103 /* The remaining fields need to be updated occasionally */
104 u64 image_size;
105 struct ceph_snap_context *snapc;
602adf40
YS
106 char *snap_names;
107 u64 *snap_sizes;
59c2be1e
YS
108
109 u64 obj_version;
110};
111
112struct rbd_options {
cc0538b6 113 bool read_only;
602adf40
YS
114};
115
116/*
f0f8cef5 117 * an instance of the client. multiple devices may share an rbd client.
602adf40
YS
118 */
119struct rbd_client {
120 struct ceph_client *client;
121 struct kref kref;
122 struct list_head node;
123};
124
125/*
f0f8cef5 126 * a request completion status
602adf40 127 */
1fec7093
YS
128struct rbd_req_status {
129 int done;
130 int rc;
131 u64 bytes;
132};
133
134/*
135 * a collection of requests
136 */
137struct rbd_req_coll {
138 int total;
139 int num_done;
140 struct kref kref;
141 struct rbd_req_status status[0];
602adf40
YS
142};
143
f0f8cef5
AE
144/*
145 * a single io request
146 */
147struct rbd_request {
148 struct request *rq; /* blk layer request */
149 struct bio *bio; /* cloned bio */
150 struct page **pages; /* list of used pages */
151 u64 len;
152 int coll_index;
153 struct rbd_req_coll *coll;
154};
155
dfc5606d
YS
156struct rbd_snap {
157 struct device dev;
158 const char *name;
3591538f 159 u64 size;
dfc5606d
YS
160 struct list_head node;
161 u64 id;
34b13184 162 u64 features;
dfc5606d
YS
163};
164
f84344f3
AE
165struct rbd_mapping {
166 char *snap_name;
167 u64 snap_id;
99c1f08f 168 u64 size;
34b13184 169 u64 features;
f84344f3
AE
170 bool snap_exists;
171 bool read_only;
172};
173
602adf40
YS
174/*
175 * a single device
176 */
177struct rbd_device {
de71a297 178 int dev_id; /* blkdev unique id */
602adf40
YS
179
180 int major; /* blkdev assigned major */
181 struct gendisk *disk; /* blkdev's gendisk and rq */
602adf40 182
a30b71b9 183 u32 image_format; /* Either 1 or 2 */
f8c38929 184 struct rbd_options rbd_opts;
602adf40
YS
185 struct rbd_client *rbd_client;
186
187 char name[DEV_NAME_LEN]; /* blkdev name, e.g. rbd3 */
188
189 spinlock_t lock; /* queue lock */
190
191 struct rbd_image_header header;
589d30e0
AE
192 char *image_id;
193 size_t image_id_len;
0bed54dc
AE
194 char *image_name;
195 size_t image_name_len;
196 char *header_name;
d22f76e7 197 char *pool_name;
9bb2f334 198 int pool_id;
602adf40 199
59c2be1e
YS
200 struct ceph_osd_event *watch_event;
201 struct ceph_osd_request *watch_request;
202
c666601a
JD
203 /* protects updating the header */
204 struct rw_semaphore header_rwsem;
f84344f3
AE
205
206 struct rbd_mapping mapping;
602adf40
YS
207
208 struct list_head node;
dfc5606d
YS
209
210 /* list of snapshots */
211 struct list_head snaps;
212
213 /* sysfs related */
214 struct device dev;
215};
216
602adf40 217static DEFINE_MUTEX(ctl_mutex); /* Serialize open/close/setup/teardown */
e124a82f 218
602adf40 219static LIST_HEAD(rbd_dev_list); /* devices */
e124a82f
AE
220static DEFINE_SPINLOCK(rbd_dev_list_lock);
221
432b8587
AE
222static LIST_HEAD(rbd_client_list); /* clients */
223static DEFINE_SPINLOCK(rbd_client_list_lock);
602adf40 224
304f6808
AE
225static int rbd_dev_snaps_update(struct rbd_device *rbd_dev);
226static int rbd_dev_snaps_register(struct rbd_device *rbd_dev);
227
dfc5606d 228static void rbd_dev_release(struct device *dev);
14e7085d 229static void __rbd_remove_snap_dev(struct rbd_snap *snap);
dfc5606d 230
f0f8cef5
AE
231static ssize_t rbd_add(struct bus_type *bus, const char *buf,
232 size_t count);
233static ssize_t rbd_remove(struct bus_type *bus, const char *buf,
234 size_t count);
235
236static struct bus_attribute rbd_bus_attrs[] = {
237 __ATTR(add, S_IWUSR, NULL, rbd_add),
238 __ATTR(remove, S_IWUSR, NULL, rbd_remove),
239 __ATTR_NULL
240};
241
242static struct bus_type rbd_bus_type = {
243 .name = "rbd",
244 .bus_attrs = rbd_bus_attrs,
245};
246
247static void rbd_root_dev_release(struct device *dev)
248{
249}
250
251static struct device rbd_root_dev = {
252 .init_name = "rbd",
253 .release = rbd_root_dev_release,
254};
255
aafb230e
AE
256#ifdef RBD_DEBUG
257#define rbd_assert(expr) \
258 if (unlikely(!(expr))) { \
259 printk(KERN_ERR "\nAssertion failure in %s() " \
260 "at line %d:\n\n" \
261 "\trbd_assert(%s);\n\n", \
262 __func__, __LINE__, #expr); \
263 BUG(); \
264 }
265#else /* !RBD_DEBUG */
266# define rbd_assert(expr) ((void) 0)
267#endif /* !RBD_DEBUG */
dfc5606d 268
dfc5606d
YS
269static struct device *rbd_get_dev(struct rbd_device *rbd_dev)
270{
271 return get_device(&rbd_dev->dev);
272}
273
274static void rbd_put_dev(struct rbd_device *rbd_dev)
275{
276 put_device(&rbd_dev->dev);
277}
602adf40 278
117973fb
AE
279static int rbd_dev_refresh(struct rbd_device *rbd_dev, u64 *hver);
280static int rbd_dev_v2_refresh(struct rbd_device *rbd_dev, u64 *hver);
59c2be1e 281
602adf40
YS
282static int rbd_open(struct block_device *bdev, fmode_t mode)
283{
f0f8cef5 284 struct rbd_device *rbd_dev = bdev->bd_disk->private_data;
602adf40 285
f84344f3 286 if ((mode & FMODE_WRITE) && rbd_dev->mapping.read_only)
602adf40
YS
287 return -EROFS;
288
340c7a2b 289 rbd_get_dev(rbd_dev);
f84344f3 290 set_device_ro(bdev, rbd_dev->mapping.read_only);
340c7a2b 291
602adf40
YS
292 return 0;
293}
294
dfc5606d
YS
295static int rbd_release(struct gendisk *disk, fmode_t mode)
296{
297 struct rbd_device *rbd_dev = disk->private_data;
298
299 rbd_put_dev(rbd_dev);
300
301 return 0;
302}
303
602adf40
YS
304static const struct block_device_operations rbd_bd_ops = {
305 .owner = THIS_MODULE,
306 .open = rbd_open,
dfc5606d 307 .release = rbd_release,
602adf40
YS
308};
309
310/*
311 * Initialize an rbd client instance.
43ae4701 312 * We own *ceph_opts.
602adf40 313 */
f8c38929 314static struct rbd_client *rbd_client_create(struct ceph_options *ceph_opts)
602adf40
YS
315{
316 struct rbd_client *rbdc;
317 int ret = -ENOMEM;
318
319 dout("rbd_client_create\n");
320 rbdc = kmalloc(sizeof(struct rbd_client), GFP_KERNEL);
321 if (!rbdc)
322 goto out_opt;
323
324 kref_init(&rbdc->kref);
325 INIT_LIST_HEAD(&rbdc->node);
326
bc534d86
AE
327 mutex_lock_nested(&ctl_mutex, SINGLE_DEPTH_NESTING);
328
43ae4701 329 rbdc->client = ceph_create_client(ceph_opts, rbdc, 0, 0);
602adf40 330 if (IS_ERR(rbdc->client))
bc534d86 331 goto out_mutex;
43ae4701 332 ceph_opts = NULL; /* Now rbdc->client is responsible for ceph_opts */
602adf40
YS
333
334 ret = ceph_open_session(rbdc->client);
335 if (ret < 0)
336 goto out_err;
337
432b8587 338 spin_lock(&rbd_client_list_lock);
602adf40 339 list_add_tail(&rbdc->node, &rbd_client_list);
432b8587 340 spin_unlock(&rbd_client_list_lock);
602adf40 341
bc534d86
AE
342 mutex_unlock(&ctl_mutex);
343
602adf40
YS
344 dout("rbd_client_create created %p\n", rbdc);
345 return rbdc;
346
347out_err:
348 ceph_destroy_client(rbdc->client);
bc534d86
AE
349out_mutex:
350 mutex_unlock(&ctl_mutex);
602adf40
YS
351 kfree(rbdc);
352out_opt:
43ae4701
AE
353 if (ceph_opts)
354 ceph_destroy_options(ceph_opts);
28f259b7 355 return ERR_PTR(ret);
602adf40
YS
356}
357
358/*
1f7ba331
AE
359 * Find a ceph client with specific addr and configuration. If
360 * found, bump its reference count.
602adf40 361 */
1f7ba331 362static struct rbd_client *rbd_client_find(struct ceph_options *ceph_opts)
602adf40
YS
363{
364 struct rbd_client *client_node;
1f7ba331 365 bool found = false;
602adf40 366
43ae4701 367 if (ceph_opts->flags & CEPH_OPT_NOSHARE)
602adf40
YS
368 return NULL;
369
1f7ba331
AE
370 spin_lock(&rbd_client_list_lock);
371 list_for_each_entry(client_node, &rbd_client_list, node) {
372 if (!ceph_compare_options(ceph_opts, client_node->client)) {
373 kref_get(&client_node->kref);
374 found = true;
375 break;
376 }
377 }
378 spin_unlock(&rbd_client_list_lock);
379
380 return found ? client_node : NULL;
602adf40
YS
381}
382
59c2be1e
YS
383/*
384 * mount options
385 */
386enum {
59c2be1e
YS
387 Opt_last_int,
388 /* int args above */
389 Opt_last_string,
390 /* string args above */
cc0538b6
AE
391 Opt_read_only,
392 Opt_read_write,
393 /* Boolean args above */
394 Opt_last_bool,
59c2be1e
YS
395};
396
43ae4701 397static match_table_t rbd_opts_tokens = {
59c2be1e
YS
398 /* int args above */
399 /* string args above */
f84344f3 400 {Opt_read_only, "mapping.read_only"},
cc0538b6
AE
401 {Opt_read_only, "ro"}, /* Alternate spelling */
402 {Opt_read_write, "read_write"},
403 {Opt_read_write, "rw"}, /* Alternate spelling */
404 /* Boolean args above */
59c2be1e
YS
405 {-1, NULL}
406};
407
408static int parse_rbd_opts_token(char *c, void *private)
409{
43ae4701 410 struct rbd_options *rbd_opts = private;
59c2be1e
YS
411 substring_t argstr[MAX_OPT_ARGS];
412 int token, intval, ret;
413
43ae4701 414 token = match_token(c, rbd_opts_tokens, argstr);
59c2be1e
YS
415 if (token < 0)
416 return -EINVAL;
417
418 if (token < Opt_last_int) {
419 ret = match_int(&argstr[0], &intval);
420 if (ret < 0) {
421 pr_err("bad mount option arg (not int) "
422 "at '%s'\n", c);
423 return ret;
424 }
425 dout("got int token %d val %d\n", token, intval);
426 } else if (token > Opt_last_int && token < Opt_last_string) {
427 dout("got string token %d val %s\n", token,
428 argstr[0].from);
cc0538b6
AE
429 } else if (token > Opt_last_string && token < Opt_last_bool) {
430 dout("got Boolean token %d\n", token);
59c2be1e
YS
431 } else {
432 dout("got token %d\n", token);
433 }
434
435 switch (token) {
cc0538b6
AE
436 case Opt_read_only:
437 rbd_opts->read_only = true;
438 break;
439 case Opt_read_write:
440 rbd_opts->read_only = false;
441 break;
59c2be1e 442 default:
aafb230e
AE
443 rbd_assert(false);
444 break;
59c2be1e
YS
445 }
446 return 0;
447}
448
602adf40
YS
449/*
450 * Get a ceph client with specific addr and configuration, if one does
451 * not exist create it.
452 */
f8c38929
AE
453static int rbd_get_client(struct rbd_device *rbd_dev, const char *mon_addr,
454 size_t mon_addr_len, char *options)
602adf40 455{
f8c38929 456 struct rbd_options *rbd_opts = &rbd_dev->rbd_opts;
43ae4701 457 struct ceph_options *ceph_opts;
f8c38929 458 struct rbd_client *rbdc;
59c2be1e 459
cc0538b6 460 rbd_opts->read_only = RBD_READ_ONLY_DEFAULT;
602adf40 461
43ae4701
AE
462 ceph_opts = ceph_parse_options(options, mon_addr,
463 mon_addr + mon_addr_len,
464 parse_rbd_opts_token, rbd_opts);
f8c38929
AE
465 if (IS_ERR(ceph_opts))
466 return PTR_ERR(ceph_opts);
602adf40 467
1f7ba331 468 rbdc = rbd_client_find(ceph_opts);
602adf40 469 if (rbdc) {
602adf40 470 /* using an existing client */
43ae4701 471 ceph_destroy_options(ceph_opts);
f8c38929
AE
472 } else {
473 rbdc = rbd_client_create(ceph_opts);
474 if (IS_ERR(rbdc))
475 return PTR_ERR(rbdc);
602adf40 476 }
f8c38929 477 rbd_dev->rbd_client = rbdc;
602adf40 478
f8c38929 479 return 0;
602adf40
YS
480}
481
482/*
483 * Destroy ceph client
d23a4b3f 484 *
432b8587 485 * Caller must hold rbd_client_list_lock.
602adf40
YS
486 */
487static void rbd_client_release(struct kref *kref)
488{
489 struct rbd_client *rbdc = container_of(kref, struct rbd_client, kref);
490
491 dout("rbd_release_client %p\n", rbdc);
cd9d9f5d 492 spin_lock(&rbd_client_list_lock);
602adf40 493 list_del(&rbdc->node);
cd9d9f5d 494 spin_unlock(&rbd_client_list_lock);
602adf40
YS
495
496 ceph_destroy_client(rbdc->client);
497 kfree(rbdc);
498}
499
500/*
501 * Drop reference to ceph client node. If it's not referenced anymore, release
502 * it.
503 */
504static void rbd_put_client(struct rbd_device *rbd_dev)
505{
506 kref_put(&rbd_dev->rbd_client->kref, rbd_client_release);
507 rbd_dev->rbd_client = NULL;
602adf40
YS
508}
509
1fec7093
YS
510/*
511 * Destroy requests collection
512 */
513static void rbd_coll_release(struct kref *kref)
514{
515 struct rbd_req_coll *coll =
516 container_of(kref, struct rbd_req_coll, kref);
517
518 dout("rbd_coll_release %p\n", coll);
519 kfree(coll);
520}
602adf40 521
a30b71b9
AE
522static bool rbd_image_format_valid(u32 image_format)
523{
524 return image_format == 1 || image_format == 2;
525}
526
8e94af8e
AE
527static bool rbd_dev_ondisk_valid(struct rbd_image_header_ondisk *ondisk)
528{
103a150f
AE
529 size_t size;
530 u32 snap_count;
531
532 /* The header has to start with the magic rbd header text */
533 if (memcmp(&ondisk->text, RBD_HEADER_TEXT, sizeof (RBD_HEADER_TEXT)))
534 return false;
535
536 /*
537 * The size of a snapshot header has to fit in a size_t, and
538 * that limits the number of snapshots.
539 */
540 snap_count = le32_to_cpu(ondisk->snap_count);
541 size = SIZE_MAX - sizeof (struct ceph_snap_context);
542 if (snap_count > size / sizeof (__le64))
543 return false;
544
545 /*
546 * Not only that, but the size of the entire the snapshot
547 * header must also be representable in a size_t.
548 */
549 size -= snap_count * sizeof (__le64);
550 if ((u64) size < le64_to_cpu(ondisk->snap_names_len))
551 return false;
552
553 return true;
8e94af8e
AE
554}
555
602adf40
YS
556/*
557 * Create a new header structure, translate header format from the on-disk
558 * header.
559 */
560static int rbd_header_from_disk(struct rbd_image_header *header,
4156d998 561 struct rbd_image_header_ondisk *ondisk)
602adf40 562{
ccece235 563 u32 snap_count;
58c17b0e 564 size_t len;
d2bb24e5 565 size_t size;
621901d6 566 u32 i;
602adf40 567
6a52325f
AE
568 memset(header, 0, sizeof (*header));
569
103a150f
AE
570 snap_count = le32_to_cpu(ondisk->snap_count);
571
58c17b0e
AE
572 len = strnlen(ondisk->object_prefix, sizeof (ondisk->object_prefix));
573 header->object_prefix = kmalloc(len + 1, GFP_KERNEL);
6a52325f 574 if (!header->object_prefix)
602adf40 575 return -ENOMEM;
58c17b0e
AE
576 memcpy(header->object_prefix, ondisk->object_prefix, len);
577 header->object_prefix[len] = '\0';
00f1f36f 578
602adf40 579 if (snap_count) {
f785cc1d
AE
580 u64 snap_names_len = le64_to_cpu(ondisk->snap_names_len);
581
621901d6
AE
582 /* Save a copy of the snapshot names */
583
f785cc1d
AE
584 if (snap_names_len > (u64) SIZE_MAX)
585 return -EIO;
586 header->snap_names = kmalloc(snap_names_len, GFP_KERNEL);
602adf40 587 if (!header->snap_names)
6a52325f 588 goto out_err;
f785cc1d
AE
589 /*
590 * Note that rbd_dev_v1_header_read() guarantees
591 * the ondisk buffer we're working with has
592 * snap_names_len bytes beyond the end of the
593 * snapshot id array, this memcpy() is safe.
594 */
595 memcpy(header->snap_names, &ondisk->snaps[snap_count],
596 snap_names_len);
6a52325f 597
621901d6
AE
598 /* Record each snapshot's size */
599
d2bb24e5
AE
600 size = snap_count * sizeof (*header->snap_sizes);
601 header->snap_sizes = kmalloc(size, GFP_KERNEL);
602adf40 602 if (!header->snap_sizes)
6a52325f 603 goto out_err;
621901d6
AE
604 for (i = 0; i < snap_count; i++)
605 header->snap_sizes[i] =
606 le64_to_cpu(ondisk->snaps[i].image_size);
602adf40 607 } else {
ccece235 608 WARN_ON(ondisk->snap_names_len);
602adf40
YS
609 header->snap_names = NULL;
610 header->snap_sizes = NULL;
611 }
849b4260 612
34b13184 613 header->features = 0; /* No features support in v1 images */
602adf40
YS
614 header->obj_order = ondisk->options.order;
615 header->crypt_type = ondisk->options.crypt_type;
616 header->comp_type = ondisk->options.comp_type;
6a52325f 617
621901d6
AE
618 /* Allocate and fill in the snapshot context */
619
f84344f3 620 header->image_size = le64_to_cpu(ondisk->image_size);
6a52325f
AE
621 size = sizeof (struct ceph_snap_context);
622 size += snap_count * sizeof (header->snapc->snaps[0]);
623 header->snapc = kzalloc(size, GFP_KERNEL);
624 if (!header->snapc)
625 goto out_err;
602adf40
YS
626
627 atomic_set(&header->snapc->nref, 1);
505cbb9b 628 header->snapc->seq = le64_to_cpu(ondisk->snap_seq);
602adf40 629 header->snapc->num_snaps = snap_count;
621901d6
AE
630 for (i = 0; i < snap_count; i++)
631 header->snapc->snaps[i] =
632 le64_to_cpu(ondisk->snaps[i].id);
602adf40
YS
633
634 return 0;
635
6a52325f 636out_err:
849b4260 637 kfree(header->snap_sizes);
ccece235 638 header->snap_sizes = NULL;
602adf40 639 kfree(header->snap_names);
ccece235 640 header->snap_names = NULL;
6a52325f
AE
641 kfree(header->object_prefix);
642 header->object_prefix = NULL;
ccece235 643
00f1f36f 644 return -ENOMEM;
602adf40
YS
645}
646
8836b995 647static int snap_by_name(struct rbd_device *rbd_dev, const char *snap_name)
602adf40 648{
602adf40 649
e86924a8 650 struct rbd_snap *snap;
602adf40 651
e86924a8
AE
652 list_for_each_entry(snap, &rbd_dev->snaps, node) {
653 if (!strcmp(snap_name, snap->name)) {
654 rbd_dev->mapping.snap_id = snap->id;
655 rbd_dev->mapping.size = snap->size;
34b13184 656 rbd_dev->mapping.features = snap->features;
602adf40 657
e86924a8 658 return 0;
00f1f36f 659 }
00f1f36f 660 }
e86924a8 661
00f1f36f 662 return -ENOENT;
602adf40
YS
663}
664
5ed16177 665static int rbd_dev_set_mapping(struct rbd_device *rbd_dev, char *snap_name)
602adf40 666{
78dc447d 667 int ret;
602adf40 668
4e1105a2 669 if (!memcmp(snap_name, RBD_SNAP_HEAD_NAME,
cc9d734c 670 sizeof (RBD_SNAP_HEAD_NAME))) {
f84344f3 671 rbd_dev->mapping.snap_id = CEPH_NOSNAP;
99c1f08f 672 rbd_dev->mapping.size = rbd_dev->header.image_size;
34b13184 673 rbd_dev->mapping.features = rbd_dev->header.features;
f84344f3
AE
674 rbd_dev->mapping.snap_exists = false;
675 rbd_dev->mapping.read_only = rbd_dev->rbd_opts.read_only;
e86924a8 676 ret = 0;
602adf40 677 } else {
8836b995 678 ret = snap_by_name(rbd_dev, snap_name);
602adf40
YS
679 if (ret < 0)
680 goto done;
f84344f3
AE
681 rbd_dev->mapping.snap_exists = true;
682 rbd_dev->mapping.read_only = true;
602adf40 683 }
4e1105a2 684 rbd_dev->mapping.snap_name = snap_name;
602adf40 685done:
602adf40
YS
686 return ret;
687}
688
689static void rbd_header_free(struct rbd_image_header *header)
690{
849b4260 691 kfree(header->object_prefix);
d78fd7ae 692 header->object_prefix = NULL;
602adf40 693 kfree(header->snap_sizes);
d78fd7ae 694 header->snap_sizes = NULL;
849b4260 695 kfree(header->snap_names);
d78fd7ae 696 header->snap_names = NULL;
d1d25646 697 ceph_put_snap_context(header->snapc);
d78fd7ae 698 header->snapc = NULL;
602adf40
YS
699}
700
65ccfe21 701static char *rbd_segment_name(struct rbd_device *rbd_dev, u64 offset)
602adf40 702{
65ccfe21
AE
703 char *name;
704 u64 segment;
705 int ret;
602adf40 706
65ccfe21
AE
707 name = kmalloc(RBD_MAX_SEG_NAME_LEN + 1, GFP_NOIO);
708 if (!name)
709 return NULL;
710 segment = offset >> rbd_dev->header.obj_order;
711 ret = snprintf(name, RBD_MAX_SEG_NAME_LEN, "%s.%012llx",
712 rbd_dev->header.object_prefix, segment);
713 if (ret < 0 || ret >= RBD_MAX_SEG_NAME_LEN) {
714 pr_err("error formatting segment name for #%llu (%d)\n",
715 segment, ret);
716 kfree(name);
717 name = NULL;
718 }
602adf40 719
65ccfe21
AE
720 return name;
721}
602adf40 722
65ccfe21
AE
723static u64 rbd_segment_offset(struct rbd_device *rbd_dev, u64 offset)
724{
725 u64 segment_size = (u64) 1 << rbd_dev->header.obj_order;
602adf40 726
65ccfe21
AE
727 return offset & (segment_size - 1);
728}
729
730static u64 rbd_segment_length(struct rbd_device *rbd_dev,
731 u64 offset, u64 length)
732{
733 u64 segment_size = (u64) 1 << rbd_dev->header.obj_order;
734
735 offset &= segment_size - 1;
736
aafb230e 737 rbd_assert(length <= U64_MAX - offset);
65ccfe21
AE
738 if (offset + length > segment_size)
739 length = segment_size - offset;
740
741 return length;
602adf40
YS
742}
743
1fec7093
YS
744static int rbd_get_num_segments(struct rbd_image_header *header,
745 u64 ofs, u64 len)
746{
df111be6
AE
747 u64 start_seg;
748 u64 end_seg;
749
750 if (!len)
751 return 0;
752 if (len - 1 > U64_MAX - ofs)
753 return -ERANGE;
754
755 start_seg = ofs >> header->obj_order;
756 end_seg = (ofs + len - 1) >> header->obj_order;
757
1fec7093
YS
758 return end_seg - start_seg + 1;
759}
760
029bcbd8
JD
761/*
762 * returns the size of an object in the image
763 */
764static u64 rbd_obj_bytes(struct rbd_image_header *header)
765{
766 return 1 << header->obj_order;
767}
768
602adf40
YS
769/*
770 * bio helpers
771 */
772
773static void bio_chain_put(struct bio *chain)
774{
775 struct bio *tmp;
776
777 while (chain) {
778 tmp = chain;
779 chain = chain->bi_next;
780 bio_put(tmp);
781 }
782}
783
784/*
785 * zeros a bio chain, starting at specific offset
786 */
787static void zero_bio_chain(struct bio *chain, int start_ofs)
788{
789 struct bio_vec *bv;
790 unsigned long flags;
791 void *buf;
792 int i;
793 int pos = 0;
794
795 while (chain) {
796 bio_for_each_segment(bv, chain, i) {
797 if (pos + bv->bv_len > start_ofs) {
798 int remainder = max(start_ofs - pos, 0);
799 buf = bvec_kmap_irq(bv, &flags);
800 memset(buf + remainder, 0,
801 bv->bv_len - remainder);
85b5aaa6 802 bvec_kunmap_irq(buf, &flags);
602adf40
YS
803 }
804 pos += bv->bv_len;
805 }
806
807 chain = chain->bi_next;
808 }
809}
810
811/*
812 * bio_chain_clone - clone a chain of bios up to a certain length.
813 * might return a bio_pair that will need to be released.
814 */
815static struct bio *bio_chain_clone(struct bio **old, struct bio **next,
816 struct bio_pair **bp,
817 int len, gfp_t gfpmask)
818{
542582fc
AE
819 struct bio *old_chain = *old;
820 struct bio *new_chain = NULL;
821 struct bio *tail;
602adf40
YS
822 int total = 0;
823
824 if (*bp) {
825 bio_pair_release(*bp);
826 *bp = NULL;
827 }
828
829 while (old_chain && (total < len)) {
542582fc
AE
830 struct bio *tmp;
831
602adf40
YS
832 tmp = bio_kmalloc(gfpmask, old_chain->bi_max_vecs);
833 if (!tmp)
834 goto err_out;
542582fc 835 gfpmask &= ~__GFP_WAIT; /* can't wait after the first */
602adf40
YS
836
837 if (total + old_chain->bi_size > len) {
838 struct bio_pair *bp;
839
840 /*
841 * this split can only happen with a single paged bio,
842 * split_bio will BUG_ON if this is not the case
843 */
844 dout("bio_chain_clone split! total=%d remaining=%d"
bd919d45
AE
845 "bi_size=%u\n",
846 total, len - total, old_chain->bi_size);
602adf40
YS
847
848 /* split the bio. We'll release it either in the next
849 call, or it will have to be released outside */
593a9e7b 850 bp = bio_split(old_chain, (len - total) / SECTOR_SIZE);
602adf40
YS
851 if (!bp)
852 goto err_out;
853
854 __bio_clone(tmp, &bp->bio1);
855
856 *next = &bp->bio2;
857 } else {
858 __bio_clone(tmp, old_chain);
859 *next = old_chain->bi_next;
860 }
861
862 tmp->bi_bdev = NULL;
602adf40 863 tmp->bi_next = NULL;
542582fc 864 if (new_chain)
602adf40 865 tail->bi_next = tmp;
542582fc
AE
866 else
867 new_chain = tmp;
868 tail = tmp;
602adf40
YS
869 old_chain = old_chain->bi_next;
870
871 total += tmp->bi_size;
872 }
873
aafb230e 874 rbd_assert(total == len);
602adf40 875
602adf40
YS
876 *old = old_chain;
877
878 return new_chain;
879
880err_out:
881 dout("bio_chain_clone with err\n");
882 bio_chain_put(new_chain);
883 return NULL;
884}
885
886/*
887 * helpers for osd request op vectors.
888 */
57cfc106
AE
889static struct ceph_osd_req_op *rbd_create_rw_ops(int num_ops,
890 int opcode, u32 payload_len)
602adf40 891{
57cfc106
AE
892 struct ceph_osd_req_op *ops;
893
894 ops = kzalloc(sizeof (*ops) * (num_ops + 1), GFP_NOIO);
895 if (!ops)
896 return NULL;
897
898 ops[0].op = opcode;
899
602adf40
YS
900 /*
901 * op extent offset and length will be set later on
902 * in calc_raw_layout()
903 */
57cfc106
AE
904 ops[0].payload_len = payload_len;
905
906 return ops;
602adf40
YS
907}
908
909static void rbd_destroy_ops(struct ceph_osd_req_op *ops)
910{
911 kfree(ops);
912}
913
1fec7093
YS
914static void rbd_coll_end_req_index(struct request *rq,
915 struct rbd_req_coll *coll,
916 int index,
917 int ret, u64 len)
918{
919 struct request_queue *q;
920 int min, max, i;
921
bd919d45
AE
922 dout("rbd_coll_end_req_index %p index %d ret %d len %llu\n",
923 coll, index, ret, (unsigned long long) len);
1fec7093
YS
924
925 if (!rq)
926 return;
927
928 if (!coll) {
929 blk_end_request(rq, ret, len);
930 return;
931 }
932
933 q = rq->q;
934
935 spin_lock_irq(q->queue_lock);
936 coll->status[index].done = 1;
937 coll->status[index].rc = ret;
938 coll->status[index].bytes = len;
939 max = min = coll->num_done;
940 while (max < coll->total && coll->status[max].done)
941 max++;
942
943 for (i = min; i<max; i++) {
944 __blk_end_request(rq, coll->status[i].rc,
945 coll->status[i].bytes);
946 coll->num_done++;
947 kref_put(&coll->kref, rbd_coll_release);
948 }
949 spin_unlock_irq(q->queue_lock);
950}
951
952static void rbd_coll_end_req(struct rbd_request *req,
953 int ret, u64 len)
954{
955 rbd_coll_end_req_index(req->rq, req->coll, req->coll_index, ret, len);
956}
957
602adf40
YS
958/*
959 * Send ceph osd request
960 */
961static int rbd_do_request(struct request *rq,
0ce1a794 962 struct rbd_device *rbd_dev,
602adf40
YS
963 struct ceph_snap_context *snapc,
964 u64 snapid,
aded07ea 965 const char *object_name, u64 ofs, u64 len,
602adf40
YS
966 struct bio *bio,
967 struct page **pages,
968 int num_pages,
969 int flags,
970 struct ceph_osd_req_op *ops,
1fec7093
YS
971 struct rbd_req_coll *coll,
972 int coll_index,
602adf40 973 void (*rbd_cb)(struct ceph_osd_request *req,
59c2be1e
YS
974 struct ceph_msg *msg),
975 struct ceph_osd_request **linger_req,
976 u64 *ver)
602adf40
YS
977{
978 struct ceph_osd_request *req;
979 struct ceph_file_layout *layout;
980 int ret;
981 u64 bno;
982 struct timespec mtime = CURRENT_TIME;
983 struct rbd_request *req_data;
984 struct ceph_osd_request_head *reqhead;
1dbb4399 985 struct ceph_osd_client *osdc;
602adf40 986
602adf40 987 req_data = kzalloc(sizeof(*req_data), GFP_NOIO);
1fec7093
YS
988 if (!req_data) {
989 if (coll)
990 rbd_coll_end_req_index(rq, coll, coll_index,
991 -ENOMEM, len);
992 return -ENOMEM;
993 }
994
995 if (coll) {
996 req_data->coll = coll;
997 req_data->coll_index = coll_index;
998 }
602adf40 999
bd919d45
AE
1000 dout("rbd_do_request object_name=%s ofs=%llu len=%llu\n", object_name,
1001 (unsigned long long) ofs, (unsigned long long) len);
602adf40 1002
0ce1a794 1003 osdc = &rbd_dev->rbd_client->client->osdc;
1dbb4399
AE
1004 req = ceph_osdc_alloc_request(osdc, flags, snapc, ops,
1005 false, GFP_NOIO, pages, bio);
4ad12621 1006 if (!req) {
4ad12621 1007 ret = -ENOMEM;
602adf40
YS
1008 goto done_pages;
1009 }
1010
1011 req->r_callback = rbd_cb;
1012
1013 req_data->rq = rq;
1014 req_data->bio = bio;
1015 req_data->pages = pages;
1016 req_data->len = len;
1017
1018 req->r_priv = req_data;
1019
1020 reqhead = req->r_request->front.iov_base;
1021 reqhead->snapid = cpu_to_le64(CEPH_NOSNAP);
1022
aded07ea 1023 strncpy(req->r_oid, object_name, sizeof(req->r_oid));
602adf40
YS
1024 req->r_oid_len = strlen(req->r_oid);
1025
1026 layout = &req->r_file_layout;
1027 memset(layout, 0, sizeof(*layout));
1028 layout->fl_stripe_unit = cpu_to_le32(1 << RBD_MAX_OBJ_ORDER);
1029 layout->fl_stripe_count = cpu_to_le32(1);
1030 layout->fl_object_size = cpu_to_le32(1 << RBD_MAX_OBJ_ORDER);
0ce1a794 1031 layout->fl_pg_pool = cpu_to_le32(rbd_dev->pool_id);
6cae3717
SW
1032 ret = ceph_calc_raw_layout(osdc, layout, snapid, ofs, &len, &bno,
1033 req, ops);
1034 rbd_assert(ret == 0);
602adf40
YS
1035
1036 ceph_osdc_build_request(req, ofs, &len,
1037 ops,
1038 snapc,
1039 &mtime,
1040 req->r_oid, req->r_oid_len);
602adf40 1041
59c2be1e 1042 if (linger_req) {
1dbb4399 1043 ceph_osdc_set_request_linger(osdc, req);
59c2be1e
YS
1044 *linger_req = req;
1045 }
1046
1dbb4399 1047 ret = ceph_osdc_start_request(osdc, req, false);
602adf40
YS
1048 if (ret < 0)
1049 goto done_err;
1050
1051 if (!rbd_cb) {
1dbb4399 1052 ret = ceph_osdc_wait_request(osdc, req);
59c2be1e
YS
1053 if (ver)
1054 *ver = le64_to_cpu(req->r_reassert_version.version);
bd919d45
AE
1055 dout("reassert_ver=%llu\n",
1056 (unsigned long long)
1057 le64_to_cpu(req->r_reassert_version.version));
602adf40
YS
1058 ceph_osdc_put_request(req);
1059 }
1060 return ret;
1061
1062done_err:
1063 bio_chain_put(req_data->bio);
1064 ceph_osdc_put_request(req);
1065done_pages:
1fec7093 1066 rbd_coll_end_req(req_data, ret, len);
602adf40 1067 kfree(req_data);
602adf40
YS
1068 return ret;
1069}
1070
1071/*
1072 * Ceph osd op callback
1073 */
1074static void rbd_req_cb(struct ceph_osd_request *req, struct ceph_msg *msg)
1075{
1076 struct rbd_request *req_data = req->r_priv;
1077 struct ceph_osd_reply_head *replyhead;
1078 struct ceph_osd_op *op;
1079 __s32 rc;
1080 u64 bytes;
1081 int read_op;
1082
1083 /* parse reply */
1084 replyhead = msg->front.iov_base;
1085 WARN_ON(le32_to_cpu(replyhead->num_ops) == 0);
1086 op = (void *)(replyhead + 1);
1087 rc = le32_to_cpu(replyhead->result);
1088 bytes = le64_to_cpu(op->extent.length);
895cfcc8 1089 read_op = (le16_to_cpu(op->op) == CEPH_OSD_OP_READ);
602adf40 1090
bd919d45
AE
1091 dout("rbd_req_cb bytes=%llu readop=%d rc=%d\n",
1092 (unsigned long long) bytes, read_op, (int) rc);
602adf40
YS
1093
1094 if (rc == -ENOENT && read_op) {
1095 zero_bio_chain(req_data->bio, 0);
1096 rc = 0;
1097 } else if (rc == 0 && read_op && bytes < req_data->len) {
1098 zero_bio_chain(req_data->bio, bytes);
1099 bytes = req_data->len;
1100 }
1101
1fec7093 1102 rbd_coll_end_req(req_data, rc, bytes);
602adf40
YS
1103
1104 if (req_data->bio)
1105 bio_chain_put(req_data->bio);
1106
1107 ceph_osdc_put_request(req);
1108 kfree(req_data);
1109}
1110
59c2be1e
YS
1111static void rbd_simple_req_cb(struct ceph_osd_request *req, struct ceph_msg *msg)
1112{
1113 ceph_osdc_put_request(req);
1114}
1115
602adf40
YS
1116/*
1117 * Do a synchronous ceph osd operation
1118 */
0ce1a794 1119static int rbd_req_sync_op(struct rbd_device *rbd_dev,
602adf40
YS
1120 struct ceph_snap_context *snapc,
1121 u64 snapid,
602adf40 1122 int flags,
913d2fdc 1123 struct ceph_osd_req_op *ops,
aded07ea 1124 const char *object_name,
f8d4de6e
AE
1125 u64 ofs, u64 inbound_size,
1126 char *inbound,
59c2be1e
YS
1127 struct ceph_osd_request **linger_req,
1128 u64 *ver)
602adf40
YS
1129{
1130 int ret;
1131 struct page **pages;
1132 int num_pages;
913d2fdc 1133
aafb230e 1134 rbd_assert(ops != NULL);
602adf40 1135
f8d4de6e 1136 num_pages = calc_pages_for(ofs, inbound_size);
602adf40 1137 pages = ceph_alloc_page_vector(num_pages, GFP_KERNEL);
b8d0638a
DC
1138 if (IS_ERR(pages))
1139 return PTR_ERR(pages);
602adf40 1140
0ce1a794 1141 ret = rbd_do_request(NULL, rbd_dev, snapc, snapid,
f8d4de6e 1142 object_name, ofs, inbound_size, NULL,
602adf40
YS
1143 pages, num_pages,
1144 flags,
1145 ops,
1fec7093 1146 NULL, 0,
59c2be1e
YS
1147 NULL,
1148 linger_req, ver);
602adf40 1149 if (ret < 0)
913d2fdc 1150 goto done;
602adf40 1151
f8d4de6e
AE
1152 if ((flags & CEPH_OSD_FLAG_READ) && inbound)
1153 ret = ceph_copy_from_page_vector(pages, inbound, ofs, ret);
602adf40 1154
602adf40
YS
1155done:
1156 ceph_release_page_vector(pages, num_pages);
1157 return ret;
1158}
1159
1160/*
1161 * Do an asynchronous ceph osd operation
1162 */
1163static int rbd_do_op(struct request *rq,
0ce1a794 1164 struct rbd_device *rbd_dev,
602adf40
YS
1165 struct ceph_snap_context *snapc,
1166 u64 snapid,
d1f57ea6 1167 int opcode, int flags,
602adf40 1168 u64 ofs, u64 len,
1fec7093
YS
1169 struct bio *bio,
1170 struct rbd_req_coll *coll,
1171 int coll_index)
602adf40
YS
1172{
1173 char *seg_name;
1174 u64 seg_ofs;
1175 u64 seg_len;
1176 int ret;
1177 struct ceph_osd_req_op *ops;
1178 u32 payload_len;
1179
65ccfe21 1180 seg_name = rbd_segment_name(rbd_dev, ofs);
602adf40
YS
1181 if (!seg_name)
1182 return -ENOMEM;
65ccfe21
AE
1183 seg_len = rbd_segment_length(rbd_dev, ofs, len);
1184 seg_ofs = rbd_segment_offset(rbd_dev, ofs);
602adf40
YS
1185
1186 payload_len = (flags & CEPH_OSD_FLAG_WRITE ? seg_len : 0);
1187
57cfc106
AE
1188 ret = -ENOMEM;
1189 ops = rbd_create_rw_ops(1, opcode, payload_len);
1190 if (!ops)
602adf40
YS
1191 goto done;
1192
1193 /* we've taken care of segment sizes earlier when we
1194 cloned the bios. We should never have a segment
1195 truncated at this point */
aafb230e 1196 rbd_assert(seg_len == len);
602adf40
YS
1197
1198 ret = rbd_do_request(rq, rbd_dev, snapc, snapid,
1199 seg_name, seg_ofs, seg_len,
1200 bio,
1201 NULL, 0,
1202 flags,
1203 ops,
1fec7093 1204 coll, coll_index,
59c2be1e 1205 rbd_req_cb, 0, NULL);
11f77002
SW
1206
1207 rbd_destroy_ops(ops);
602adf40
YS
1208done:
1209 kfree(seg_name);
1210 return ret;
1211}
1212
1213/*
1214 * Request async osd write
1215 */
1216static int rbd_req_write(struct request *rq,
1217 struct rbd_device *rbd_dev,
1218 struct ceph_snap_context *snapc,
1219 u64 ofs, u64 len,
1fec7093
YS
1220 struct bio *bio,
1221 struct rbd_req_coll *coll,
1222 int coll_index)
602adf40
YS
1223{
1224 return rbd_do_op(rq, rbd_dev, snapc, CEPH_NOSNAP,
1225 CEPH_OSD_OP_WRITE,
1226 CEPH_OSD_FLAG_WRITE | CEPH_OSD_FLAG_ONDISK,
1fec7093 1227 ofs, len, bio, coll, coll_index);
602adf40
YS
1228}
1229
1230/*
1231 * Request async osd read
1232 */
1233static int rbd_req_read(struct request *rq,
1234 struct rbd_device *rbd_dev,
1235 u64 snapid,
1236 u64 ofs, u64 len,
1fec7093
YS
1237 struct bio *bio,
1238 struct rbd_req_coll *coll,
1239 int coll_index)
602adf40
YS
1240{
1241 return rbd_do_op(rq, rbd_dev, NULL,
b06e6a6b 1242 snapid,
602adf40
YS
1243 CEPH_OSD_OP_READ,
1244 CEPH_OSD_FLAG_READ,
1fec7093 1245 ofs, len, bio, coll, coll_index);
602adf40
YS
1246}
1247
1248/*
1249 * Request sync osd read
1250 */
0ce1a794 1251static int rbd_req_sync_read(struct rbd_device *rbd_dev,
602adf40 1252 u64 snapid,
aded07ea 1253 const char *object_name,
602adf40 1254 u64 ofs, u64 len,
59c2be1e
YS
1255 char *buf,
1256 u64 *ver)
602adf40 1257{
913d2fdc
AE
1258 struct ceph_osd_req_op *ops;
1259 int ret;
1260
1261 ops = rbd_create_rw_ops(1, CEPH_OSD_OP_READ, 0);
1262 if (!ops)
1263 return -ENOMEM;
1264
1265 ret = rbd_req_sync_op(rbd_dev, NULL,
b06e6a6b 1266 snapid,
602adf40 1267 CEPH_OSD_FLAG_READ,
913d2fdc
AE
1268 ops, object_name, ofs, len, buf, NULL, ver);
1269 rbd_destroy_ops(ops);
1270
1271 return ret;
602adf40
YS
1272}
1273
1274/*
59c2be1e
YS
1275 * Request sync osd watch
1276 */
0ce1a794 1277static int rbd_req_sync_notify_ack(struct rbd_device *rbd_dev,
59c2be1e 1278 u64 ver,
7f0a24d8 1279 u64 notify_id)
59c2be1e
YS
1280{
1281 struct ceph_osd_req_op *ops;
11f77002
SW
1282 int ret;
1283
57cfc106
AE
1284 ops = rbd_create_rw_ops(1, CEPH_OSD_OP_NOTIFY_ACK, 0);
1285 if (!ops)
1286 return -ENOMEM;
59c2be1e 1287
a71b891b 1288 ops[0].watch.ver = cpu_to_le64(ver);
59c2be1e
YS
1289 ops[0].watch.cookie = notify_id;
1290 ops[0].watch.flag = 0;
1291
0ce1a794 1292 ret = rbd_do_request(NULL, rbd_dev, NULL, CEPH_NOSNAP,
7f0a24d8 1293 rbd_dev->header_name, 0, 0, NULL,
ad4f232f 1294 NULL, 0,
59c2be1e
YS
1295 CEPH_OSD_FLAG_READ,
1296 ops,
1fec7093 1297 NULL, 0,
59c2be1e
YS
1298 rbd_simple_req_cb, 0, NULL);
1299
1300 rbd_destroy_ops(ops);
1301 return ret;
1302}
1303
1304static void rbd_watch_cb(u64 ver, u64 notify_id, u8 opcode, void *data)
1305{
0ce1a794 1306 struct rbd_device *rbd_dev = (struct rbd_device *)data;
a71b891b 1307 u64 hver;
13143d2d
SW
1308 int rc;
1309
0ce1a794 1310 if (!rbd_dev)
59c2be1e
YS
1311 return;
1312
bd919d45
AE
1313 dout("rbd_watch_cb %s notify_id=%llu opcode=%u\n",
1314 rbd_dev->header_name, (unsigned long long) notify_id,
1315 (unsigned int) opcode);
117973fb 1316 rc = rbd_dev_refresh(rbd_dev, &hver);
13143d2d 1317 if (rc)
f0f8cef5 1318 pr_warning(RBD_DRV_NAME "%d got notification but failed to "
0ce1a794 1319 " update snaps: %d\n", rbd_dev->major, rc);
59c2be1e 1320
7f0a24d8 1321 rbd_req_sync_notify_ack(rbd_dev, hver, notify_id);
59c2be1e
YS
1322}
1323
1324/*
1325 * Request sync osd watch
1326 */
0e6f322d 1327static int rbd_req_sync_watch(struct rbd_device *rbd_dev)
59c2be1e
YS
1328{
1329 struct ceph_osd_req_op *ops;
0ce1a794 1330 struct ceph_osd_client *osdc = &rbd_dev->rbd_client->client->osdc;
57cfc106 1331 int ret;
59c2be1e 1332
57cfc106
AE
1333 ops = rbd_create_rw_ops(1, CEPH_OSD_OP_WATCH, 0);
1334 if (!ops)
1335 return -ENOMEM;
59c2be1e
YS
1336
1337 ret = ceph_osdc_create_event(osdc, rbd_watch_cb, 0,
0ce1a794 1338 (void *)rbd_dev, &rbd_dev->watch_event);
59c2be1e
YS
1339 if (ret < 0)
1340 goto fail;
1341
0e6f322d 1342 ops[0].watch.ver = cpu_to_le64(rbd_dev->header.obj_version);
0ce1a794 1343 ops[0].watch.cookie = cpu_to_le64(rbd_dev->watch_event->cookie);
59c2be1e
YS
1344 ops[0].watch.flag = 1;
1345
0ce1a794 1346 ret = rbd_req_sync_op(rbd_dev, NULL,
59c2be1e 1347 CEPH_NOSNAP,
59c2be1e
YS
1348 CEPH_OSD_FLAG_WRITE | CEPH_OSD_FLAG_ONDISK,
1349 ops,
0e6f322d
AE
1350 rbd_dev->header_name,
1351 0, 0, NULL,
0ce1a794 1352 &rbd_dev->watch_request, NULL);
59c2be1e
YS
1353
1354 if (ret < 0)
1355 goto fail_event;
1356
1357 rbd_destroy_ops(ops);
1358 return 0;
1359
1360fail_event:
0ce1a794
AE
1361 ceph_osdc_cancel_event(rbd_dev->watch_event);
1362 rbd_dev->watch_event = NULL;
59c2be1e
YS
1363fail:
1364 rbd_destroy_ops(ops);
1365 return ret;
1366}
1367
79e3057c
YS
1368/*
1369 * Request sync osd unwatch
1370 */
070c633f 1371static int rbd_req_sync_unwatch(struct rbd_device *rbd_dev)
79e3057c
YS
1372{
1373 struct ceph_osd_req_op *ops;
57cfc106 1374 int ret;
79e3057c 1375
57cfc106
AE
1376 ops = rbd_create_rw_ops(1, CEPH_OSD_OP_WATCH, 0);
1377 if (!ops)
1378 return -ENOMEM;
79e3057c
YS
1379
1380 ops[0].watch.ver = 0;
0ce1a794 1381 ops[0].watch.cookie = cpu_to_le64(rbd_dev->watch_event->cookie);
79e3057c
YS
1382 ops[0].watch.flag = 0;
1383
0ce1a794 1384 ret = rbd_req_sync_op(rbd_dev, NULL,
79e3057c 1385 CEPH_NOSNAP,
79e3057c
YS
1386 CEPH_OSD_FLAG_WRITE | CEPH_OSD_FLAG_ONDISK,
1387 ops,
070c633f
AE
1388 rbd_dev->header_name,
1389 0, 0, NULL, NULL, NULL);
1390
79e3057c
YS
1391
1392 rbd_destroy_ops(ops);
0ce1a794
AE
1393 ceph_osdc_cancel_event(rbd_dev->watch_event);
1394 rbd_dev->watch_event = NULL;
79e3057c
YS
1395 return ret;
1396}
1397
602adf40 1398/*
3cb4a687 1399 * Synchronous osd object method call
602adf40 1400 */
0ce1a794 1401static int rbd_req_sync_exec(struct rbd_device *rbd_dev,
aded07ea
AE
1402 const char *object_name,
1403 const char *class_name,
1404 const char *method_name,
3cb4a687
AE
1405 const char *outbound,
1406 size_t outbound_size,
f8d4de6e
AE
1407 char *inbound,
1408 size_t inbound_size,
3cb4a687 1409 int flags,
59c2be1e 1410 u64 *ver)
602adf40
YS
1411{
1412 struct ceph_osd_req_op *ops;
aded07ea
AE
1413 int class_name_len = strlen(class_name);
1414 int method_name_len = strlen(method_name);
3cb4a687 1415 int payload_size;
57cfc106
AE
1416 int ret;
1417
3cb4a687
AE
1418 /*
1419 * Any input parameters required by the method we're calling
1420 * will be sent along with the class and method names as
1421 * part of the message payload. That data and its size are
1422 * supplied via the indata and indata_len fields (named from
1423 * the perspective of the server side) in the OSD request
1424 * operation.
1425 */
1426 payload_size = class_name_len + method_name_len + outbound_size;
1427 ops = rbd_create_rw_ops(1, CEPH_OSD_OP_CALL, payload_size);
57cfc106
AE
1428 if (!ops)
1429 return -ENOMEM;
602adf40 1430
aded07ea
AE
1431 ops[0].cls.class_name = class_name;
1432 ops[0].cls.class_len = (__u8) class_name_len;
1433 ops[0].cls.method_name = method_name;
1434 ops[0].cls.method_len = (__u8) method_name_len;
602adf40 1435 ops[0].cls.argc = 0;
3cb4a687
AE
1436 ops[0].cls.indata = outbound;
1437 ops[0].cls.indata_len = outbound_size;
602adf40 1438
0ce1a794 1439 ret = rbd_req_sync_op(rbd_dev, NULL,
602adf40 1440 CEPH_NOSNAP,
3cb4a687 1441 flags, ops,
f8d4de6e
AE
1442 object_name, 0, inbound_size, inbound,
1443 NULL, ver);
602adf40
YS
1444
1445 rbd_destroy_ops(ops);
1446
1447 dout("cls_exec returned %d\n", ret);
1448 return ret;
1449}
1450
1fec7093
YS
1451static struct rbd_req_coll *rbd_alloc_coll(int num_reqs)
1452{
1453 struct rbd_req_coll *coll =
1454 kzalloc(sizeof(struct rbd_req_coll) +
1455 sizeof(struct rbd_req_status) * num_reqs,
1456 GFP_ATOMIC);
1457
1458 if (!coll)
1459 return NULL;
1460 coll->total = num_reqs;
1461 kref_init(&coll->kref);
1462 return coll;
1463}
1464
602adf40
YS
1465/*
1466 * block device queue callback
1467 */
1468static void rbd_rq_fn(struct request_queue *q)
1469{
1470 struct rbd_device *rbd_dev = q->queuedata;
1471 struct request *rq;
1472 struct bio_pair *bp = NULL;
1473
00f1f36f 1474 while ((rq = blk_fetch_request(q))) {
602adf40
YS
1475 struct bio *bio;
1476 struct bio *rq_bio, *next_bio = NULL;
1477 bool do_write;
bd919d45
AE
1478 unsigned int size;
1479 u64 op_size = 0;
602adf40 1480 u64 ofs;
1fec7093
YS
1481 int num_segs, cur_seg = 0;
1482 struct rbd_req_coll *coll;
d1d25646 1483 struct ceph_snap_context *snapc;
602adf40 1484
602adf40
YS
1485 dout("fetched request\n");
1486
1487 /* filter out block requests we don't understand */
1488 if ((rq->cmd_type != REQ_TYPE_FS)) {
1489 __blk_end_request_all(rq, 0);
00f1f36f 1490 continue;
602adf40
YS
1491 }
1492
1493 /* deduce our operation (read, write) */
1494 do_write = (rq_data_dir(rq) == WRITE);
1495
1496 size = blk_rq_bytes(rq);
593a9e7b 1497 ofs = blk_rq_pos(rq) * SECTOR_SIZE;
602adf40 1498 rq_bio = rq->bio;
f84344f3 1499 if (do_write && rbd_dev->mapping.read_only) {
602adf40 1500 __blk_end_request_all(rq, -EROFS);
00f1f36f 1501 continue;
602adf40
YS
1502 }
1503
1504 spin_unlock_irq(q->queue_lock);
1505
d1d25646 1506 down_read(&rbd_dev->header_rwsem);
e88a36ec 1507
f84344f3
AE
1508 if (rbd_dev->mapping.snap_id != CEPH_NOSNAP &&
1509 !rbd_dev->mapping.snap_exists) {
e88a36ec 1510 up_read(&rbd_dev->header_rwsem);
d1d25646
JD
1511 dout("request for non-existent snapshot");
1512 spin_lock_irq(q->queue_lock);
1513 __blk_end_request_all(rq, -ENXIO);
1514 continue;
e88a36ec
JD
1515 }
1516
d1d25646
JD
1517 snapc = ceph_get_snap_context(rbd_dev->header.snapc);
1518
1519 up_read(&rbd_dev->header_rwsem);
1520
602adf40
YS
1521 dout("%s 0x%x bytes at 0x%llx\n",
1522 do_write ? "write" : "read",
bd919d45 1523 size, (unsigned long long) blk_rq_pos(rq) * SECTOR_SIZE);
602adf40 1524
1fec7093 1525 num_segs = rbd_get_num_segments(&rbd_dev->header, ofs, size);
df111be6
AE
1526 if (num_segs <= 0) {
1527 spin_lock_irq(q->queue_lock);
1528 __blk_end_request_all(rq, num_segs);
1529 ceph_put_snap_context(snapc);
1530 continue;
1531 }
1fec7093
YS
1532 coll = rbd_alloc_coll(num_segs);
1533 if (!coll) {
1534 spin_lock_irq(q->queue_lock);
1535 __blk_end_request_all(rq, -ENOMEM);
d1d25646 1536 ceph_put_snap_context(snapc);
00f1f36f 1537 continue;
1fec7093
YS
1538 }
1539
602adf40
YS
1540 do {
1541 /* a bio clone to be passed down to OSD req */
bd919d45 1542 dout("rq->bio->bi_vcnt=%hu\n", rq->bio->bi_vcnt);
65ccfe21 1543 op_size = rbd_segment_length(rbd_dev, ofs, size);
1fec7093 1544 kref_get(&coll->kref);
602adf40
YS
1545 bio = bio_chain_clone(&rq_bio, &next_bio, &bp,
1546 op_size, GFP_ATOMIC);
1547 if (!bio) {
1fec7093
YS
1548 rbd_coll_end_req_index(rq, coll, cur_seg,
1549 -ENOMEM, op_size);
1550 goto next_seg;
602adf40
YS
1551 }
1552
1fec7093 1553
602adf40
YS
1554 /* init OSD command: write or read */
1555 if (do_write)
1556 rbd_req_write(rq, rbd_dev,
d1d25646 1557 snapc,
602adf40 1558 ofs,
1fec7093
YS
1559 op_size, bio,
1560 coll, cur_seg);
602adf40
YS
1561 else
1562 rbd_req_read(rq, rbd_dev,
f84344f3 1563 rbd_dev->mapping.snap_id,
602adf40 1564 ofs,
1fec7093
YS
1565 op_size, bio,
1566 coll, cur_seg);
602adf40 1567
1fec7093 1568next_seg:
602adf40
YS
1569 size -= op_size;
1570 ofs += op_size;
1571
1fec7093 1572 cur_seg++;
602adf40
YS
1573 rq_bio = next_bio;
1574 } while (size > 0);
1fec7093 1575 kref_put(&coll->kref, rbd_coll_release);
602adf40
YS
1576
1577 if (bp)
1578 bio_pair_release(bp);
602adf40 1579 spin_lock_irq(q->queue_lock);
d1d25646
JD
1580
1581 ceph_put_snap_context(snapc);
602adf40
YS
1582 }
1583}
1584
1585/*
1586 * a queue callback. Makes sure that we don't create a bio that spans across
1587 * multiple osd objects. One exception would be with a single page bios,
1588 * which we handle later at bio_chain_clone
1589 */
1590static int rbd_merge_bvec(struct request_queue *q, struct bvec_merge_data *bmd,
1591 struct bio_vec *bvec)
1592{
1593 struct rbd_device *rbd_dev = q->queuedata;
593a9e7b
AE
1594 unsigned int chunk_sectors;
1595 sector_t sector;
1596 unsigned int bio_sectors;
602adf40
YS
1597 int max;
1598
593a9e7b
AE
1599 chunk_sectors = 1 << (rbd_dev->header.obj_order - SECTOR_SHIFT);
1600 sector = bmd->bi_sector + get_start_sect(bmd->bi_bdev);
1601 bio_sectors = bmd->bi_size >> SECTOR_SHIFT;
1602
602adf40 1603 max = (chunk_sectors - ((sector & (chunk_sectors - 1))
593a9e7b 1604 + bio_sectors)) << SECTOR_SHIFT;
602adf40
YS
1605 if (max < 0)
1606 max = 0; /* bio_add cannot handle a negative return */
1607 if (max <= bvec->bv_len && bio_sectors == 0)
1608 return bvec->bv_len;
1609 return max;
1610}
1611
1612static void rbd_free_disk(struct rbd_device *rbd_dev)
1613{
1614 struct gendisk *disk = rbd_dev->disk;
1615
1616 if (!disk)
1617 return;
1618
602adf40
YS
1619 if (disk->flags & GENHD_FL_UP)
1620 del_gendisk(disk);
1621 if (disk->queue)
1622 blk_cleanup_queue(disk->queue);
1623 put_disk(disk);
1624}
1625
1626/*
4156d998
AE
1627 * Read the complete header for the given rbd device.
1628 *
1629 * Returns a pointer to a dynamically-allocated buffer containing
1630 * the complete and validated header. Caller can pass the address
1631 * of a variable that will be filled in with the version of the
1632 * header object at the time it was read.
1633 *
1634 * Returns a pointer-coded errno if a failure occurs.
602adf40 1635 */
4156d998
AE
1636static struct rbd_image_header_ondisk *
1637rbd_dev_v1_header_read(struct rbd_device *rbd_dev, u64 *version)
602adf40 1638{
4156d998 1639 struct rbd_image_header_ondisk *ondisk = NULL;
50f7c4c9 1640 u32 snap_count = 0;
4156d998
AE
1641 u64 names_size = 0;
1642 u32 want_count;
1643 int ret;
602adf40 1644
00f1f36f 1645 /*
4156d998
AE
1646 * The complete header will include an array of its 64-bit
1647 * snapshot ids, followed by the names of those snapshots as
1648 * a contiguous block of NUL-terminated strings. Note that
1649 * the number of snapshots could change by the time we read
1650 * it in, in which case we re-read it.
00f1f36f 1651 */
4156d998
AE
1652 do {
1653 size_t size;
1654
1655 kfree(ondisk);
1656
1657 size = sizeof (*ondisk);
1658 size += snap_count * sizeof (struct rbd_image_snap_ondisk);
1659 size += names_size;
1660 ondisk = kmalloc(size, GFP_KERNEL);
1661 if (!ondisk)
1662 return ERR_PTR(-ENOMEM);
1663
1664 ret = rbd_req_sync_read(rbd_dev, CEPH_NOSNAP,
0bed54dc 1665 rbd_dev->header_name,
4156d998
AE
1666 0, size,
1667 (char *) ondisk, version);
1668
1669 if (ret < 0)
1670 goto out_err;
1671 if (WARN_ON((size_t) ret < size)) {
1672 ret = -ENXIO;
1673 pr_warning("short header read for image %s"
1674 " (want %zd got %d)\n",
1675 rbd_dev->image_name, size, ret);
1676 goto out_err;
1677 }
1678 if (!rbd_dev_ondisk_valid(ondisk)) {
1679 ret = -ENXIO;
1680 pr_warning("invalid header for image %s\n",
1681 rbd_dev->image_name);
1682 goto out_err;
81e759fb 1683 }
602adf40 1684
4156d998
AE
1685 names_size = le64_to_cpu(ondisk->snap_names_len);
1686 want_count = snap_count;
1687 snap_count = le32_to_cpu(ondisk->snap_count);
1688 } while (snap_count != want_count);
00f1f36f 1689
4156d998 1690 return ondisk;
00f1f36f 1691
4156d998
AE
1692out_err:
1693 kfree(ondisk);
1694
1695 return ERR_PTR(ret);
1696}
1697
1698/*
1699 * reload the ondisk the header
1700 */
1701static int rbd_read_header(struct rbd_device *rbd_dev,
1702 struct rbd_image_header *header)
1703{
1704 struct rbd_image_header_ondisk *ondisk;
1705 u64 ver = 0;
1706 int ret;
602adf40 1707
4156d998
AE
1708 ondisk = rbd_dev_v1_header_read(rbd_dev, &ver);
1709 if (IS_ERR(ondisk))
1710 return PTR_ERR(ondisk);
1711 ret = rbd_header_from_disk(header, ondisk);
1712 if (ret >= 0)
1713 header->obj_version = ver;
1714 kfree(ondisk);
1715
1716 return ret;
602adf40
YS
1717}
1718
dfc5606d
YS
1719static void __rbd_remove_all_snaps(struct rbd_device *rbd_dev)
1720{
1721 struct rbd_snap *snap;
a0593290 1722 struct rbd_snap *next;
dfc5606d 1723
a0593290 1724 list_for_each_entry_safe(snap, next, &rbd_dev->snaps, node)
14e7085d 1725 __rbd_remove_snap_dev(snap);
dfc5606d
YS
1726}
1727
9478554a
AE
1728static void rbd_update_mapping_size(struct rbd_device *rbd_dev)
1729{
1730 sector_t size;
1731
1732 if (rbd_dev->mapping.snap_id != CEPH_NOSNAP)
1733 return;
1734
1735 size = (sector_t) rbd_dev->header.image_size / SECTOR_SIZE;
1736 dout("setting size to %llu sectors", (unsigned long long) size);
1737 rbd_dev->mapping.size = (u64) size;
1738 set_capacity(rbd_dev->disk, size);
1739}
1740
602adf40
YS
1741/*
1742 * only read the first part of the ondisk header, without the snaps info
1743 */
117973fb 1744static int rbd_dev_v1_refresh(struct rbd_device *rbd_dev, u64 *hver)
602adf40
YS
1745{
1746 int ret;
1747 struct rbd_image_header h;
602adf40
YS
1748
1749 ret = rbd_read_header(rbd_dev, &h);
1750 if (ret < 0)
1751 return ret;
1752
a51aa0c0
JD
1753 down_write(&rbd_dev->header_rwsem);
1754
9478554a
AE
1755 /* Update image size, and check for resize of mapped image */
1756 rbd_dev->header.image_size = h.image_size;
1757 rbd_update_mapping_size(rbd_dev);
9db4b3e3 1758
849b4260 1759 /* rbd_dev->header.object_prefix shouldn't change */
602adf40 1760 kfree(rbd_dev->header.snap_sizes);
849b4260 1761 kfree(rbd_dev->header.snap_names);
d1d25646
JD
1762 /* osd requests may still refer to snapc */
1763 ceph_put_snap_context(rbd_dev->header.snapc);
602adf40 1764
b813623a
AE
1765 if (hver)
1766 *hver = h.obj_version;
a71b891b 1767 rbd_dev->header.obj_version = h.obj_version;
93a24e08 1768 rbd_dev->header.image_size = h.image_size;
602adf40
YS
1769 rbd_dev->header.snapc = h.snapc;
1770 rbd_dev->header.snap_names = h.snap_names;
1771 rbd_dev->header.snap_sizes = h.snap_sizes;
849b4260
AE
1772 /* Free the extra copy of the object prefix */
1773 WARN_ON(strcmp(rbd_dev->header.object_prefix, h.object_prefix));
1774 kfree(h.object_prefix);
1775
304f6808
AE
1776 ret = rbd_dev_snaps_update(rbd_dev);
1777 if (!ret)
1778 ret = rbd_dev_snaps_register(rbd_dev);
dfc5606d 1779
c666601a 1780 up_write(&rbd_dev->header_rwsem);
602adf40 1781
dfc5606d 1782 return ret;
602adf40
YS
1783}
1784
117973fb 1785static int rbd_dev_refresh(struct rbd_device *rbd_dev, u64 *hver)
1fe5e993
AE
1786{
1787 int ret;
1788
117973fb 1789 rbd_assert(rbd_image_format_valid(rbd_dev->image_format));
1fe5e993 1790 mutex_lock_nested(&ctl_mutex, SINGLE_DEPTH_NESTING);
117973fb
AE
1791 if (rbd_dev->image_format == 1)
1792 ret = rbd_dev_v1_refresh(rbd_dev, hver);
1793 else
1794 ret = rbd_dev_v2_refresh(rbd_dev, hver);
1fe5e993
AE
1795 mutex_unlock(&ctl_mutex);
1796
1797 return ret;
1798}
1799
602adf40
YS
1800static int rbd_init_disk(struct rbd_device *rbd_dev)
1801{
1802 struct gendisk *disk;
1803 struct request_queue *q;
593a9e7b 1804 u64 segment_size;
602adf40 1805
602adf40 1806 /* create gendisk info */
602adf40
YS
1807 disk = alloc_disk(RBD_MINORS_PER_MAJOR);
1808 if (!disk)
1fcdb8aa 1809 return -ENOMEM;
602adf40 1810
f0f8cef5 1811 snprintf(disk->disk_name, sizeof(disk->disk_name), RBD_DRV_NAME "%d",
de71a297 1812 rbd_dev->dev_id);
602adf40
YS
1813 disk->major = rbd_dev->major;
1814 disk->first_minor = 0;
1815 disk->fops = &rbd_bd_ops;
1816 disk->private_data = rbd_dev;
1817
1818 /* init rq */
602adf40
YS
1819 q = blk_init_queue(rbd_rq_fn, &rbd_dev->lock);
1820 if (!q)
1821 goto out_disk;
029bcbd8 1822
593a9e7b
AE
1823 /* We use the default size, but let's be explicit about it. */
1824 blk_queue_physical_block_size(q, SECTOR_SIZE);
1825
029bcbd8 1826 /* set io sizes to object size */
593a9e7b
AE
1827 segment_size = rbd_obj_bytes(&rbd_dev->header);
1828 blk_queue_max_hw_sectors(q, segment_size / SECTOR_SIZE);
1829 blk_queue_max_segment_size(q, segment_size);
1830 blk_queue_io_min(q, segment_size);
1831 blk_queue_io_opt(q, segment_size);
029bcbd8 1832
602adf40
YS
1833 blk_queue_merge_bvec(q, rbd_merge_bvec);
1834 disk->queue = q;
1835
1836 q->queuedata = rbd_dev;
1837
1838 rbd_dev->disk = disk;
602adf40 1839
12f02944
AE
1840 set_capacity(rbd_dev->disk, rbd_dev->mapping.size / SECTOR_SIZE);
1841
602adf40 1842 return 0;
602adf40
YS
1843out_disk:
1844 put_disk(disk);
1fcdb8aa
AE
1845
1846 return -ENOMEM;
602adf40
YS
1847}
1848
dfc5606d
YS
1849/*
1850 sysfs
1851*/
1852
593a9e7b
AE
1853static struct rbd_device *dev_to_rbd_dev(struct device *dev)
1854{
1855 return container_of(dev, struct rbd_device, dev);
1856}
1857
dfc5606d
YS
1858static ssize_t rbd_size_show(struct device *dev,
1859 struct device_attribute *attr, char *buf)
1860{
593a9e7b 1861 struct rbd_device *rbd_dev = dev_to_rbd_dev(dev);
a51aa0c0
JD
1862 sector_t size;
1863
1864 down_read(&rbd_dev->header_rwsem);
1865 size = get_capacity(rbd_dev->disk);
1866 up_read(&rbd_dev->header_rwsem);
dfc5606d 1867
a51aa0c0 1868 return sprintf(buf, "%llu\n", (unsigned long long) size * SECTOR_SIZE);
dfc5606d
YS
1869}
1870
34b13184
AE
1871/*
1872 * Note this shows the features for whatever's mapped, which is not
1873 * necessarily the base image.
1874 */
1875static ssize_t rbd_features_show(struct device *dev,
1876 struct device_attribute *attr, char *buf)
1877{
1878 struct rbd_device *rbd_dev = dev_to_rbd_dev(dev);
1879
1880 return sprintf(buf, "0x%016llx\n",
1881 (unsigned long long) rbd_dev->mapping.features);
1882}
1883
dfc5606d
YS
1884static ssize_t rbd_major_show(struct device *dev,
1885 struct device_attribute *attr, char *buf)
1886{
593a9e7b 1887 struct rbd_device *rbd_dev = dev_to_rbd_dev(dev);
602adf40 1888
dfc5606d
YS
1889 return sprintf(buf, "%d\n", rbd_dev->major);
1890}
1891
1892static ssize_t rbd_client_id_show(struct device *dev,
1893 struct device_attribute *attr, char *buf)
602adf40 1894{
593a9e7b 1895 struct rbd_device *rbd_dev = dev_to_rbd_dev(dev);
dfc5606d 1896
1dbb4399
AE
1897 return sprintf(buf, "client%lld\n",
1898 ceph_client_id(rbd_dev->rbd_client->client));
602adf40
YS
1899}
1900
dfc5606d
YS
1901static ssize_t rbd_pool_show(struct device *dev,
1902 struct device_attribute *attr, char *buf)
602adf40 1903{
593a9e7b 1904 struct rbd_device *rbd_dev = dev_to_rbd_dev(dev);
dfc5606d
YS
1905
1906 return sprintf(buf, "%s\n", rbd_dev->pool_name);
1907}
1908
9bb2f334
AE
1909static ssize_t rbd_pool_id_show(struct device *dev,
1910 struct device_attribute *attr, char *buf)
1911{
1912 struct rbd_device *rbd_dev = dev_to_rbd_dev(dev);
1913
1914 return sprintf(buf, "%d\n", rbd_dev->pool_id);
1915}
1916
dfc5606d
YS
1917static ssize_t rbd_name_show(struct device *dev,
1918 struct device_attribute *attr, char *buf)
1919{
593a9e7b 1920 struct rbd_device *rbd_dev = dev_to_rbd_dev(dev);
dfc5606d 1921
0bed54dc 1922 return sprintf(buf, "%s\n", rbd_dev->image_name);
dfc5606d
YS
1923}
1924
589d30e0
AE
1925static ssize_t rbd_image_id_show(struct device *dev,
1926 struct device_attribute *attr, char *buf)
1927{
1928 struct rbd_device *rbd_dev = dev_to_rbd_dev(dev);
1929
1930 return sprintf(buf, "%s\n", rbd_dev->image_id);
1931}
1932
34b13184
AE
1933/*
1934 * Shows the name of the currently-mapped snapshot (or
1935 * RBD_SNAP_HEAD_NAME for the base image).
1936 */
dfc5606d
YS
1937static ssize_t rbd_snap_show(struct device *dev,
1938 struct device_attribute *attr,
1939 char *buf)
1940{
593a9e7b 1941 struct rbd_device *rbd_dev = dev_to_rbd_dev(dev);
dfc5606d 1942
f84344f3 1943 return sprintf(buf, "%s\n", rbd_dev->mapping.snap_name);
dfc5606d
YS
1944}
1945
1946static ssize_t rbd_image_refresh(struct device *dev,
1947 struct device_attribute *attr,
1948 const char *buf,
1949 size_t size)
1950{
593a9e7b 1951 struct rbd_device *rbd_dev = dev_to_rbd_dev(dev);
b813623a 1952 int ret;
602adf40 1953
117973fb 1954 ret = rbd_dev_refresh(rbd_dev, NULL);
b813623a
AE
1955
1956 return ret < 0 ? ret : size;
dfc5606d 1957}
602adf40 1958
dfc5606d 1959static DEVICE_ATTR(size, S_IRUGO, rbd_size_show, NULL);
34b13184 1960static DEVICE_ATTR(features, S_IRUGO, rbd_features_show, NULL);
dfc5606d
YS
1961static DEVICE_ATTR(major, S_IRUGO, rbd_major_show, NULL);
1962static DEVICE_ATTR(client_id, S_IRUGO, rbd_client_id_show, NULL);
1963static DEVICE_ATTR(pool, S_IRUGO, rbd_pool_show, NULL);
9bb2f334 1964static DEVICE_ATTR(pool_id, S_IRUGO, rbd_pool_id_show, NULL);
dfc5606d 1965static DEVICE_ATTR(name, S_IRUGO, rbd_name_show, NULL);
589d30e0 1966static DEVICE_ATTR(image_id, S_IRUGO, rbd_image_id_show, NULL);
dfc5606d
YS
1967static DEVICE_ATTR(refresh, S_IWUSR, NULL, rbd_image_refresh);
1968static DEVICE_ATTR(current_snap, S_IRUGO, rbd_snap_show, NULL);
dfc5606d
YS
1969
1970static struct attribute *rbd_attrs[] = {
1971 &dev_attr_size.attr,
34b13184 1972 &dev_attr_features.attr,
dfc5606d
YS
1973 &dev_attr_major.attr,
1974 &dev_attr_client_id.attr,
1975 &dev_attr_pool.attr,
9bb2f334 1976 &dev_attr_pool_id.attr,
dfc5606d 1977 &dev_attr_name.attr,
589d30e0 1978 &dev_attr_image_id.attr,
dfc5606d
YS
1979 &dev_attr_current_snap.attr,
1980 &dev_attr_refresh.attr,
dfc5606d
YS
1981 NULL
1982};
1983
1984static struct attribute_group rbd_attr_group = {
1985 .attrs = rbd_attrs,
1986};
1987
1988static const struct attribute_group *rbd_attr_groups[] = {
1989 &rbd_attr_group,
1990 NULL
1991};
1992
1993static void rbd_sysfs_dev_release(struct device *dev)
1994{
1995}
1996
1997static struct device_type rbd_device_type = {
1998 .name = "rbd",
1999 .groups = rbd_attr_groups,
2000 .release = rbd_sysfs_dev_release,
2001};
2002
2003
2004/*
2005 sysfs - snapshots
2006*/
2007
2008static ssize_t rbd_snap_size_show(struct device *dev,
2009 struct device_attribute *attr,
2010 char *buf)
2011{
2012 struct rbd_snap *snap = container_of(dev, struct rbd_snap, dev);
2013
3591538f 2014 return sprintf(buf, "%llu\n", (unsigned long long)snap->size);
dfc5606d
YS
2015}
2016
2017static ssize_t rbd_snap_id_show(struct device *dev,
2018 struct device_attribute *attr,
2019 char *buf)
2020{
2021 struct rbd_snap *snap = container_of(dev, struct rbd_snap, dev);
2022
3591538f 2023 return sprintf(buf, "%llu\n", (unsigned long long)snap->id);
dfc5606d
YS
2024}
2025
34b13184
AE
2026static ssize_t rbd_snap_features_show(struct device *dev,
2027 struct device_attribute *attr,
2028 char *buf)
2029{
2030 struct rbd_snap *snap = container_of(dev, struct rbd_snap, dev);
2031
2032 return sprintf(buf, "0x%016llx\n",
2033 (unsigned long long) snap->features);
2034}
2035
dfc5606d
YS
2036static DEVICE_ATTR(snap_size, S_IRUGO, rbd_snap_size_show, NULL);
2037static DEVICE_ATTR(snap_id, S_IRUGO, rbd_snap_id_show, NULL);
34b13184 2038static DEVICE_ATTR(snap_features, S_IRUGO, rbd_snap_features_show, NULL);
dfc5606d
YS
2039
2040static struct attribute *rbd_snap_attrs[] = {
2041 &dev_attr_snap_size.attr,
2042 &dev_attr_snap_id.attr,
34b13184 2043 &dev_attr_snap_features.attr,
dfc5606d
YS
2044 NULL,
2045};
2046
2047static struct attribute_group rbd_snap_attr_group = {
2048 .attrs = rbd_snap_attrs,
2049};
2050
2051static void rbd_snap_dev_release(struct device *dev)
2052{
2053 struct rbd_snap *snap = container_of(dev, struct rbd_snap, dev);
2054 kfree(snap->name);
2055 kfree(snap);
2056}
2057
2058static const struct attribute_group *rbd_snap_attr_groups[] = {
2059 &rbd_snap_attr_group,
2060 NULL
2061};
2062
2063static struct device_type rbd_snap_device_type = {
2064 .groups = rbd_snap_attr_groups,
2065 .release = rbd_snap_dev_release,
2066};
2067
304f6808
AE
2068static bool rbd_snap_registered(struct rbd_snap *snap)
2069{
2070 bool ret = snap->dev.type == &rbd_snap_device_type;
2071 bool reg = device_is_registered(&snap->dev);
2072
2073 rbd_assert(!ret ^ reg);
2074
2075 return ret;
2076}
2077
14e7085d 2078static void __rbd_remove_snap_dev(struct rbd_snap *snap)
dfc5606d
YS
2079{
2080 list_del(&snap->node);
304f6808
AE
2081 if (device_is_registered(&snap->dev))
2082 device_unregister(&snap->dev);
dfc5606d
YS
2083}
2084
14e7085d 2085static int rbd_register_snap_dev(struct rbd_snap *snap,
dfc5606d
YS
2086 struct device *parent)
2087{
2088 struct device *dev = &snap->dev;
2089 int ret;
2090
2091 dev->type = &rbd_snap_device_type;
2092 dev->parent = parent;
2093 dev->release = rbd_snap_dev_release;
2094 dev_set_name(dev, "snap_%s", snap->name);
304f6808
AE
2095 dout("%s: registering device for snapshot %s\n", __func__, snap->name);
2096
dfc5606d
YS
2097 ret = device_register(dev);
2098
2099 return ret;
2100}
2101
4e891e0a 2102static struct rbd_snap *__rbd_add_snap_dev(struct rbd_device *rbd_dev,
c8d18425 2103 const char *snap_name,
34b13184
AE
2104 u64 snap_id, u64 snap_size,
2105 u64 snap_features)
dfc5606d 2106{
4e891e0a 2107 struct rbd_snap *snap;
dfc5606d 2108 int ret;
4e891e0a
AE
2109
2110 snap = kzalloc(sizeof (*snap), GFP_KERNEL);
dfc5606d 2111 if (!snap)
4e891e0a
AE
2112 return ERR_PTR(-ENOMEM);
2113
2114 ret = -ENOMEM;
c8d18425 2115 snap->name = kstrdup(snap_name, GFP_KERNEL);
4e891e0a
AE
2116 if (!snap->name)
2117 goto err;
2118
c8d18425
AE
2119 snap->id = snap_id;
2120 snap->size = snap_size;
34b13184 2121 snap->features = snap_features;
4e891e0a
AE
2122
2123 return snap;
2124
dfc5606d
YS
2125err:
2126 kfree(snap->name);
2127 kfree(snap);
4e891e0a
AE
2128
2129 return ERR_PTR(ret);
dfc5606d
YS
2130}
2131
cd892126
AE
2132static char *rbd_dev_v1_snap_info(struct rbd_device *rbd_dev, u32 which,
2133 u64 *snap_size, u64 *snap_features)
2134{
2135 char *snap_name;
2136
2137 rbd_assert(which < rbd_dev->header.snapc->num_snaps);
2138
2139 *snap_size = rbd_dev->header.snap_sizes[which];
2140 *snap_features = 0; /* No features for v1 */
2141
2142 /* Skip over names until we find the one we are looking for */
2143
2144 snap_name = rbd_dev->header.snap_names;
2145 while (which--)
2146 snap_name += strlen(snap_name) + 1;
2147
2148 return snap_name;
2149}
2150
9d475de5
AE
2151/*
2152 * Get the size and object order for an image snapshot, or if
2153 * snap_id is CEPH_NOSNAP, gets this information for the base
2154 * image.
2155 */
2156static int _rbd_dev_v2_snap_size(struct rbd_device *rbd_dev, u64 snap_id,
2157 u8 *order, u64 *snap_size)
2158{
2159 __le64 snapid = cpu_to_le64(snap_id);
2160 int ret;
2161 struct {
2162 u8 order;
2163 __le64 size;
2164 } __attribute__ ((packed)) size_buf = { 0 };
2165
2166 ret = rbd_req_sync_exec(rbd_dev, rbd_dev->header_name,
2167 "rbd", "get_size",
2168 (char *) &snapid, sizeof (snapid),
2169 (char *) &size_buf, sizeof (size_buf),
2170 CEPH_OSD_FLAG_READ, NULL);
2171 dout("%s: rbd_req_sync_exec returned %d\n", __func__, ret);
2172 if (ret < 0)
2173 return ret;
2174
2175 *order = size_buf.order;
2176 *snap_size = le64_to_cpu(size_buf.size);
2177
2178 dout(" snap_id 0x%016llx order = %u, snap_size = %llu\n",
2179 (unsigned long long) snap_id, (unsigned int) *order,
2180 (unsigned long long) *snap_size);
2181
2182 return 0;
2183}
2184
2185static int rbd_dev_v2_image_size(struct rbd_device *rbd_dev)
2186{
2187 return _rbd_dev_v2_snap_size(rbd_dev, CEPH_NOSNAP,
2188 &rbd_dev->header.obj_order,
2189 &rbd_dev->header.image_size);
2190}
2191
1e130199
AE
2192static int rbd_dev_v2_object_prefix(struct rbd_device *rbd_dev)
2193{
2194 void *reply_buf;
2195 int ret;
2196 void *p;
2197
2198 reply_buf = kzalloc(RBD_OBJ_PREFIX_LEN_MAX, GFP_KERNEL);
2199 if (!reply_buf)
2200 return -ENOMEM;
2201
2202 ret = rbd_req_sync_exec(rbd_dev, rbd_dev->header_name,
2203 "rbd", "get_object_prefix",
2204 NULL, 0,
2205 reply_buf, RBD_OBJ_PREFIX_LEN_MAX,
2206 CEPH_OSD_FLAG_READ, NULL);
2207 dout("%s: rbd_req_sync_exec returned %d\n", __func__, ret);
2208 if (ret < 0)
2209 goto out;
2210
2211 p = reply_buf;
2212 rbd_dev->header.object_prefix = ceph_extract_encoded_string(&p,
2213 p + RBD_OBJ_PREFIX_LEN_MAX,
2214 NULL, GFP_NOIO);
2215
2216 if (IS_ERR(rbd_dev->header.object_prefix)) {
2217 ret = PTR_ERR(rbd_dev->header.object_prefix);
2218 rbd_dev->header.object_prefix = NULL;
2219 } else {
2220 dout(" object_prefix = %s\n", rbd_dev->header.object_prefix);
2221 }
2222
2223out:
2224 kfree(reply_buf);
2225
2226 return ret;
2227}
2228
b1b5402a
AE
2229static int _rbd_dev_v2_snap_features(struct rbd_device *rbd_dev, u64 snap_id,
2230 u64 *snap_features)
2231{
2232 __le64 snapid = cpu_to_le64(snap_id);
2233 struct {
2234 __le64 features;
2235 __le64 incompat;
2236 } features_buf = { 0 };
d889140c 2237 u64 incompat;
b1b5402a
AE
2238 int ret;
2239
2240 ret = rbd_req_sync_exec(rbd_dev, rbd_dev->header_name,
2241 "rbd", "get_features",
2242 (char *) &snapid, sizeof (snapid),
2243 (char *) &features_buf, sizeof (features_buf),
2244 CEPH_OSD_FLAG_READ, NULL);
2245 dout("%s: rbd_req_sync_exec returned %d\n", __func__, ret);
2246 if (ret < 0)
2247 return ret;
d889140c
AE
2248
2249 incompat = le64_to_cpu(features_buf.incompat);
2250 if (incompat & ~RBD_FEATURES_ALL)
2251 return -ENOTSUPP;
2252
b1b5402a
AE
2253 *snap_features = le64_to_cpu(features_buf.features);
2254
2255 dout(" snap_id 0x%016llx features = 0x%016llx incompat = 0x%016llx\n",
2256 (unsigned long long) snap_id,
2257 (unsigned long long) *snap_features,
2258 (unsigned long long) le64_to_cpu(features_buf.incompat));
2259
2260 return 0;
2261}
2262
2263static int rbd_dev_v2_features(struct rbd_device *rbd_dev)
2264{
2265 return _rbd_dev_v2_snap_features(rbd_dev, CEPH_NOSNAP,
2266 &rbd_dev->header.features);
2267}
2268
6e14b1a6 2269static int rbd_dev_v2_snap_context(struct rbd_device *rbd_dev, u64 *ver)
35d489f9
AE
2270{
2271 size_t size;
2272 int ret;
2273 void *reply_buf;
2274 void *p;
2275 void *end;
2276 u64 seq;
2277 u32 snap_count;
2278 struct ceph_snap_context *snapc;
2279 u32 i;
2280
2281 /*
2282 * We'll need room for the seq value (maximum snapshot id),
2283 * snapshot count, and array of that many snapshot ids.
2284 * For now we have a fixed upper limit on the number we're
2285 * prepared to receive.
2286 */
2287 size = sizeof (__le64) + sizeof (__le32) +
2288 RBD_MAX_SNAP_COUNT * sizeof (__le64);
2289 reply_buf = kzalloc(size, GFP_KERNEL);
2290 if (!reply_buf)
2291 return -ENOMEM;
2292
2293 ret = rbd_req_sync_exec(rbd_dev, rbd_dev->header_name,
2294 "rbd", "get_snapcontext",
2295 NULL, 0,
2296 reply_buf, size,
6e14b1a6 2297 CEPH_OSD_FLAG_READ, ver);
35d489f9
AE
2298 dout("%s: rbd_req_sync_exec returned %d\n", __func__, ret);
2299 if (ret < 0)
2300 goto out;
2301
2302 ret = -ERANGE;
2303 p = reply_buf;
2304 end = (char *) reply_buf + size;
2305 ceph_decode_64_safe(&p, end, seq, out);
2306 ceph_decode_32_safe(&p, end, snap_count, out);
2307
2308 /*
2309 * Make sure the reported number of snapshot ids wouldn't go
2310 * beyond the end of our buffer. But before checking that,
2311 * make sure the computed size of the snapshot context we
2312 * allocate is representable in a size_t.
2313 */
2314 if (snap_count > (SIZE_MAX - sizeof (struct ceph_snap_context))
2315 / sizeof (u64)) {
2316 ret = -EINVAL;
2317 goto out;
2318 }
2319 if (!ceph_has_room(&p, end, snap_count * sizeof (__le64)))
2320 goto out;
2321
2322 size = sizeof (struct ceph_snap_context) +
2323 snap_count * sizeof (snapc->snaps[0]);
2324 snapc = kmalloc(size, GFP_KERNEL);
2325 if (!snapc) {
2326 ret = -ENOMEM;
2327 goto out;
2328 }
2329
2330 atomic_set(&snapc->nref, 1);
2331 snapc->seq = seq;
2332 snapc->num_snaps = snap_count;
2333 for (i = 0; i < snap_count; i++)
2334 snapc->snaps[i] = ceph_decode_64(&p);
2335
2336 rbd_dev->header.snapc = snapc;
2337
2338 dout(" snap context seq = %llu, snap_count = %u\n",
2339 (unsigned long long) seq, (unsigned int) snap_count);
2340
2341out:
2342 kfree(reply_buf);
2343
2344 return 0;
2345}
2346
b8b1e2db
AE
2347static char *rbd_dev_v2_snap_name(struct rbd_device *rbd_dev, u32 which)
2348{
2349 size_t size;
2350 void *reply_buf;
2351 __le64 snap_id;
2352 int ret;
2353 void *p;
2354 void *end;
2355 size_t snap_name_len;
2356 char *snap_name;
2357
2358 size = sizeof (__le32) + RBD_MAX_SNAP_NAME_LEN;
2359 reply_buf = kmalloc(size, GFP_KERNEL);
2360 if (!reply_buf)
2361 return ERR_PTR(-ENOMEM);
2362
2363 snap_id = cpu_to_le64(rbd_dev->header.snapc->snaps[which]);
2364 ret = rbd_req_sync_exec(rbd_dev, rbd_dev->header_name,
2365 "rbd", "get_snapshot_name",
2366 (char *) &snap_id, sizeof (snap_id),
2367 reply_buf, size,
2368 CEPH_OSD_FLAG_READ, NULL);
2369 dout("%s: rbd_req_sync_exec returned %d\n", __func__, ret);
2370 if (ret < 0)
2371 goto out;
2372
2373 p = reply_buf;
2374 end = (char *) reply_buf + size;
2375 snap_name_len = 0;
2376 snap_name = ceph_extract_encoded_string(&p, end, &snap_name_len,
2377 GFP_KERNEL);
2378 if (IS_ERR(snap_name)) {
2379 ret = PTR_ERR(snap_name);
2380 goto out;
2381 } else {
2382 dout(" snap_id 0x%016llx snap_name = %s\n",
2383 (unsigned long long) le64_to_cpu(snap_id), snap_name);
2384 }
2385 kfree(reply_buf);
2386
2387 return snap_name;
2388out:
2389 kfree(reply_buf);
2390
2391 return ERR_PTR(ret);
2392}
2393
2394static char *rbd_dev_v2_snap_info(struct rbd_device *rbd_dev, u32 which,
2395 u64 *snap_size, u64 *snap_features)
2396{
2397 __le64 snap_id;
2398 u8 order;
2399 int ret;
2400
2401 snap_id = rbd_dev->header.snapc->snaps[which];
2402 ret = _rbd_dev_v2_snap_size(rbd_dev, snap_id, &order, snap_size);
2403 if (ret)
2404 return ERR_PTR(ret);
2405 ret = _rbd_dev_v2_snap_features(rbd_dev, snap_id, snap_features);
2406 if (ret)
2407 return ERR_PTR(ret);
2408
2409 return rbd_dev_v2_snap_name(rbd_dev, which);
2410}
2411
2412static char *rbd_dev_snap_info(struct rbd_device *rbd_dev, u32 which,
2413 u64 *snap_size, u64 *snap_features)
2414{
2415 if (rbd_dev->image_format == 1)
2416 return rbd_dev_v1_snap_info(rbd_dev, which,
2417 snap_size, snap_features);
2418 if (rbd_dev->image_format == 2)
2419 return rbd_dev_v2_snap_info(rbd_dev, which,
2420 snap_size, snap_features);
2421 return ERR_PTR(-EINVAL);
2422}
2423
117973fb
AE
2424static int rbd_dev_v2_refresh(struct rbd_device *rbd_dev, u64 *hver)
2425{
2426 int ret;
2427 __u8 obj_order;
2428
2429 down_write(&rbd_dev->header_rwsem);
2430
2431 /* Grab old order first, to see if it changes */
2432
2433 obj_order = rbd_dev->header.obj_order,
2434 ret = rbd_dev_v2_image_size(rbd_dev);
2435 if (ret)
2436 goto out;
2437 if (rbd_dev->header.obj_order != obj_order) {
2438 ret = -EIO;
2439 goto out;
2440 }
2441 rbd_update_mapping_size(rbd_dev);
2442
2443 ret = rbd_dev_v2_snap_context(rbd_dev, hver);
2444 dout("rbd_dev_v2_snap_context returned %d\n", ret);
2445 if (ret)
2446 goto out;
2447 ret = rbd_dev_snaps_update(rbd_dev);
2448 dout("rbd_dev_snaps_update returned %d\n", ret);
2449 if (ret)
2450 goto out;
2451 ret = rbd_dev_snaps_register(rbd_dev);
2452 dout("rbd_dev_snaps_register returned %d\n", ret);
2453out:
2454 up_write(&rbd_dev->header_rwsem);
2455
2456 return ret;
2457}
2458
dfc5606d 2459/*
35938150
AE
2460 * Scan the rbd device's current snapshot list and compare it to the
2461 * newly-received snapshot context. Remove any existing snapshots
2462 * not present in the new snapshot context. Add a new snapshot for
2463 * any snaphots in the snapshot context not in the current list.
2464 * And verify there are no changes to snapshots we already know
2465 * about.
2466 *
2467 * Assumes the snapshots in the snapshot context are sorted by
2468 * snapshot id, highest id first. (Snapshots in the rbd_dev's list
2469 * are also maintained in that order.)
dfc5606d 2470 */
304f6808 2471static int rbd_dev_snaps_update(struct rbd_device *rbd_dev)
dfc5606d 2472{
35938150
AE
2473 struct ceph_snap_context *snapc = rbd_dev->header.snapc;
2474 const u32 snap_count = snapc->num_snaps;
35938150
AE
2475 struct list_head *head = &rbd_dev->snaps;
2476 struct list_head *links = head->next;
2477 u32 index = 0;
dfc5606d 2478
9fcbb800 2479 dout("%s: snap count is %u\n", __func__, (unsigned int) snap_count);
35938150
AE
2480 while (index < snap_count || links != head) {
2481 u64 snap_id;
2482 struct rbd_snap *snap;
cd892126
AE
2483 char *snap_name;
2484 u64 snap_size = 0;
2485 u64 snap_features = 0;
dfc5606d 2486
35938150
AE
2487 snap_id = index < snap_count ? snapc->snaps[index]
2488 : CEPH_NOSNAP;
2489 snap = links != head ? list_entry(links, struct rbd_snap, node)
2490 : NULL;
aafb230e 2491 rbd_assert(!snap || snap->id != CEPH_NOSNAP);
dfc5606d 2492
35938150
AE
2493 if (snap_id == CEPH_NOSNAP || (snap && snap->id > snap_id)) {
2494 struct list_head *next = links->next;
dfc5606d 2495
35938150 2496 /* Existing snapshot not in the new snap context */
dfc5606d 2497
f84344f3
AE
2498 if (rbd_dev->mapping.snap_id == snap->id)
2499 rbd_dev->mapping.snap_exists = false;
35938150 2500 __rbd_remove_snap_dev(snap);
9fcbb800 2501 dout("%ssnap id %llu has been removed\n",
f84344f3
AE
2502 rbd_dev->mapping.snap_id == snap->id ?
2503 "mapped " : "",
9fcbb800 2504 (unsigned long long) snap->id);
35938150
AE
2505
2506 /* Done with this list entry; advance */
2507
2508 links = next;
dfc5606d
YS
2509 continue;
2510 }
35938150 2511
b8b1e2db
AE
2512 snap_name = rbd_dev_snap_info(rbd_dev, index,
2513 &snap_size, &snap_features);
cd892126
AE
2514 if (IS_ERR(snap_name))
2515 return PTR_ERR(snap_name);
2516
9fcbb800
AE
2517 dout("entry %u: snap_id = %llu\n", (unsigned int) snap_count,
2518 (unsigned long long) snap_id);
35938150
AE
2519 if (!snap || (snap_id != CEPH_NOSNAP && snap->id < snap_id)) {
2520 struct rbd_snap *new_snap;
2521
2522 /* We haven't seen this snapshot before */
2523
c8d18425 2524 new_snap = __rbd_add_snap_dev(rbd_dev, snap_name,
cd892126 2525 snap_id, snap_size, snap_features);
9fcbb800
AE
2526 if (IS_ERR(new_snap)) {
2527 int err = PTR_ERR(new_snap);
2528
2529 dout(" failed to add dev, error %d\n", err);
2530
2531 return err;
2532 }
35938150
AE
2533
2534 /* New goes before existing, or at end of list */
2535
9fcbb800 2536 dout(" added dev%s\n", snap ? "" : " at end\n");
35938150
AE
2537 if (snap)
2538 list_add_tail(&new_snap->node, &snap->node);
2539 else
523f3258 2540 list_add_tail(&new_snap->node, head);
35938150
AE
2541 } else {
2542 /* Already have this one */
2543
9fcbb800
AE
2544 dout(" already present\n");
2545
cd892126 2546 rbd_assert(snap->size == snap_size);
aafb230e 2547 rbd_assert(!strcmp(snap->name, snap_name));
cd892126 2548 rbd_assert(snap->features == snap_features);
35938150
AE
2549
2550 /* Done with this list entry; advance */
2551
2552 links = links->next;
dfc5606d 2553 }
35938150
AE
2554
2555 /* Advance to the next entry in the snapshot context */
2556
2557 index++;
dfc5606d 2558 }
9fcbb800 2559 dout("%s: done\n", __func__);
dfc5606d
YS
2560
2561 return 0;
2562}
2563
304f6808
AE
2564/*
2565 * Scan the list of snapshots and register the devices for any that
2566 * have not already been registered.
2567 */
2568static int rbd_dev_snaps_register(struct rbd_device *rbd_dev)
2569{
2570 struct rbd_snap *snap;
2571 int ret = 0;
2572
2573 dout("%s called\n", __func__);
86ff77bb
AE
2574 if (WARN_ON(!device_is_registered(&rbd_dev->dev)))
2575 return -EIO;
304f6808
AE
2576
2577 list_for_each_entry(snap, &rbd_dev->snaps, node) {
2578 if (!rbd_snap_registered(snap)) {
2579 ret = rbd_register_snap_dev(snap, &rbd_dev->dev);
2580 if (ret < 0)
2581 break;
2582 }
2583 }
2584 dout("%s: returning %d\n", __func__, ret);
2585
2586 return ret;
2587}
2588
dfc5606d
YS
2589static int rbd_bus_add_dev(struct rbd_device *rbd_dev)
2590{
dfc5606d 2591 struct device *dev;
cd789ab9 2592 int ret;
dfc5606d
YS
2593
2594 mutex_lock_nested(&ctl_mutex, SINGLE_DEPTH_NESTING);
dfc5606d 2595
cd789ab9 2596 dev = &rbd_dev->dev;
dfc5606d
YS
2597 dev->bus = &rbd_bus_type;
2598 dev->type = &rbd_device_type;
2599 dev->parent = &rbd_root_dev;
2600 dev->release = rbd_dev_release;
de71a297 2601 dev_set_name(dev, "%d", rbd_dev->dev_id);
dfc5606d 2602 ret = device_register(dev);
dfc5606d 2603
dfc5606d 2604 mutex_unlock(&ctl_mutex);
cd789ab9 2605
dfc5606d 2606 return ret;
602adf40
YS
2607}
2608
dfc5606d
YS
2609static void rbd_bus_del_dev(struct rbd_device *rbd_dev)
2610{
2611 device_unregister(&rbd_dev->dev);
2612}
2613
59c2be1e
YS
2614static int rbd_init_watch_dev(struct rbd_device *rbd_dev)
2615{
2616 int ret, rc;
2617
2618 do {
0e6f322d 2619 ret = rbd_req_sync_watch(rbd_dev);
59c2be1e 2620 if (ret == -ERANGE) {
117973fb 2621 rc = rbd_dev_refresh(rbd_dev, NULL);
59c2be1e
YS
2622 if (rc < 0)
2623 return rc;
2624 }
2625 } while (ret == -ERANGE);
2626
2627 return ret;
2628}
2629
e2839308 2630static atomic64_t rbd_dev_id_max = ATOMIC64_INIT(0);
1ddbe94e
AE
2631
2632/*
499afd5b
AE
2633 * Get a unique rbd identifier for the given new rbd_dev, and add
2634 * the rbd_dev to the global list. The minimum rbd id is 1.
1ddbe94e 2635 */
e2839308 2636static void rbd_dev_id_get(struct rbd_device *rbd_dev)
b7f23c36 2637{
e2839308 2638 rbd_dev->dev_id = atomic64_inc_return(&rbd_dev_id_max);
499afd5b
AE
2639
2640 spin_lock(&rbd_dev_list_lock);
2641 list_add_tail(&rbd_dev->node, &rbd_dev_list);
2642 spin_unlock(&rbd_dev_list_lock);
e2839308
AE
2643 dout("rbd_dev %p given dev id %llu\n", rbd_dev,
2644 (unsigned long long) rbd_dev->dev_id);
1ddbe94e 2645}
b7f23c36 2646
1ddbe94e 2647/*
499afd5b
AE
2648 * Remove an rbd_dev from the global list, and record that its
2649 * identifier is no longer in use.
1ddbe94e 2650 */
e2839308 2651static void rbd_dev_id_put(struct rbd_device *rbd_dev)
1ddbe94e 2652{
d184f6bf 2653 struct list_head *tmp;
de71a297 2654 int rbd_id = rbd_dev->dev_id;
d184f6bf
AE
2655 int max_id;
2656
aafb230e 2657 rbd_assert(rbd_id > 0);
499afd5b 2658
e2839308
AE
2659 dout("rbd_dev %p released dev id %llu\n", rbd_dev,
2660 (unsigned long long) rbd_dev->dev_id);
499afd5b
AE
2661 spin_lock(&rbd_dev_list_lock);
2662 list_del_init(&rbd_dev->node);
d184f6bf
AE
2663
2664 /*
2665 * If the id being "put" is not the current maximum, there
2666 * is nothing special we need to do.
2667 */
e2839308 2668 if (rbd_id != atomic64_read(&rbd_dev_id_max)) {
d184f6bf
AE
2669 spin_unlock(&rbd_dev_list_lock);
2670 return;
2671 }
2672
2673 /*
2674 * We need to update the current maximum id. Search the
2675 * list to find out what it is. We're more likely to find
2676 * the maximum at the end, so search the list backward.
2677 */
2678 max_id = 0;
2679 list_for_each_prev(tmp, &rbd_dev_list) {
2680 struct rbd_device *rbd_dev;
2681
2682 rbd_dev = list_entry(tmp, struct rbd_device, node);
2683 if (rbd_id > max_id)
2684 max_id = rbd_id;
2685 }
499afd5b 2686 spin_unlock(&rbd_dev_list_lock);
b7f23c36 2687
1ddbe94e 2688 /*
e2839308 2689 * The max id could have been updated by rbd_dev_id_get(), in
d184f6bf
AE
2690 * which case it now accurately reflects the new maximum.
2691 * Be careful not to overwrite the maximum value in that
2692 * case.
1ddbe94e 2693 */
e2839308
AE
2694 atomic64_cmpxchg(&rbd_dev_id_max, rbd_id, max_id);
2695 dout(" max dev id has been reset\n");
b7f23c36
AE
2696}
2697
e28fff26
AE
2698/*
2699 * Skips over white space at *buf, and updates *buf to point to the
2700 * first found non-space character (if any). Returns the length of
593a9e7b
AE
2701 * the token (string of non-white space characters) found. Note
2702 * that *buf must be terminated with '\0'.
e28fff26
AE
2703 */
2704static inline size_t next_token(const char **buf)
2705{
2706 /*
2707 * These are the characters that produce nonzero for
2708 * isspace() in the "C" and "POSIX" locales.
2709 */
2710 const char *spaces = " \f\n\r\t\v";
2711
2712 *buf += strspn(*buf, spaces); /* Find start of token */
2713
2714 return strcspn(*buf, spaces); /* Return token length */
2715}
2716
2717/*
2718 * Finds the next token in *buf, and if the provided token buffer is
2719 * big enough, copies the found token into it. The result, if
593a9e7b
AE
2720 * copied, is guaranteed to be terminated with '\0'. Note that *buf
2721 * must be terminated with '\0' on entry.
e28fff26
AE
2722 *
2723 * Returns the length of the token found (not including the '\0').
2724 * Return value will be 0 if no token is found, and it will be >=
2725 * token_size if the token would not fit.
2726 *
593a9e7b 2727 * The *buf pointer will be updated to point beyond the end of the
e28fff26
AE
2728 * found token. Note that this occurs even if the token buffer is
2729 * too small to hold it.
2730 */
2731static inline size_t copy_token(const char **buf,
2732 char *token,
2733 size_t token_size)
2734{
2735 size_t len;
2736
2737 len = next_token(buf);
2738 if (len < token_size) {
2739 memcpy(token, *buf, len);
2740 *(token + len) = '\0';
2741 }
2742 *buf += len;
2743
2744 return len;
2745}
2746
ea3352f4
AE
2747/*
2748 * Finds the next token in *buf, dynamically allocates a buffer big
2749 * enough to hold a copy of it, and copies the token into the new
2750 * buffer. The copy is guaranteed to be terminated with '\0'. Note
2751 * that a duplicate buffer is created even for a zero-length token.
2752 *
2753 * Returns a pointer to the newly-allocated duplicate, or a null
2754 * pointer if memory for the duplicate was not available. If
2755 * the lenp argument is a non-null pointer, the length of the token
2756 * (not including the '\0') is returned in *lenp.
2757 *
2758 * If successful, the *buf pointer will be updated to point beyond
2759 * the end of the found token.
2760 *
2761 * Note: uses GFP_KERNEL for allocation.
2762 */
2763static inline char *dup_token(const char **buf, size_t *lenp)
2764{
2765 char *dup;
2766 size_t len;
2767
2768 len = next_token(buf);
2769 dup = kmalloc(len + 1, GFP_KERNEL);
2770 if (!dup)
2771 return NULL;
2772
2773 memcpy(dup, *buf, len);
2774 *(dup + len) = '\0';
2775 *buf += len;
2776
2777 if (lenp)
2778 *lenp = len;
2779
2780 return dup;
2781}
2782
a725f65e 2783/*
3feeb894
AE
2784 * This fills in the pool_name, image_name, image_name_len, rbd_dev,
2785 * rbd_md_name, and name fields of the given rbd_dev, based on the
2786 * list of monitor addresses and other options provided via
2787 * /sys/bus/rbd/add. Returns a pointer to a dynamically-allocated
2788 * copy of the snapshot name to map if successful, or a
2789 * pointer-coded error otherwise.
d22f76e7
AE
2790 *
2791 * Note: rbd_dev is assumed to have been initially zero-filled.
a725f65e 2792 */
3feeb894
AE
2793static char *rbd_add_parse_args(struct rbd_device *rbd_dev,
2794 const char *buf,
2795 const char **mon_addrs,
2796 size_t *mon_addrs_size,
2797 char *options,
2798 size_t options_size)
e28fff26 2799{
d22f76e7 2800 size_t len;
3feeb894
AE
2801 char *err_ptr = ERR_PTR(-EINVAL);
2802 char *snap_name;
e28fff26
AE
2803
2804 /* The first four tokens are required */
2805
7ef3214a
AE
2806 len = next_token(&buf);
2807 if (!len)
3feeb894 2808 return err_ptr;
5214ecc4 2809 *mon_addrs_size = len + 1;
7ef3214a
AE
2810 *mon_addrs = buf;
2811
2812 buf += len;
a725f65e 2813
e28fff26
AE
2814 len = copy_token(&buf, options, options_size);
2815 if (!len || len >= options_size)
3feeb894 2816 return err_ptr;
e28fff26 2817
3feeb894 2818 err_ptr = ERR_PTR(-ENOMEM);
d22f76e7
AE
2819 rbd_dev->pool_name = dup_token(&buf, NULL);
2820 if (!rbd_dev->pool_name)
d22f76e7 2821 goto out_err;
e28fff26 2822
0bed54dc
AE
2823 rbd_dev->image_name = dup_token(&buf, &rbd_dev->image_name_len);
2824 if (!rbd_dev->image_name)
bf3e5ae1 2825 goto out_err;
a725f65e 2826
3feeb894
AE
2827 /* Snapshot name is optional */
2828 len = next_token(&buf);
820a5f3e 2829 if (!len) {
3feeb894
AE
2830 buf = RBD_SNAP_HEAD_NAME; /* No snapshot supplied */
2831 len = sizeof (RBD_SNAP_HEAD_NAME) - 1;
849b4260 2832 }
3feeb894
AE
2833 snap_name = kmalloc(len + 1, GFP_KERNEL);
2834 if (!snap_name)
2835 goto out_err;
2836 memcpy(snap_name, buf, len);
2837 *(snap_name + len) = '\0';
e28fff26 2838
3feeb894
AE
2839dout(" SNAP_NAME is <%s>, len is %zd\n", snap_name, len);
2840
2841 return snap_name;
d22f76e7
AE
2842
2843out_err:
0bed54dc 2844 kfree(rbd_dev->image_name);
d78fd7ae
AE
2845 rbd_dev->image_name = NULL;
2846 rbd_dev->image_name_len = 0;
d22f76e7
AE
2847 kfree(rbd_dev->pool_name);
2848 rbd_dev->pool_name = NULL;
2849
3feeb894 2850 return err_ptr;
a725f65e
AE
2851}
2852
589d30e0
AE
2853/*
2854 * An rbd format 2 image has a unique identifier, distinct from the
2855 * name given to it by the user. Internally, that identifier is
2856 * what's used to specify the names of objects related to the image.
2857 *
2858 * A special "rbd id" object is used to map an rbd image name to its
2859 * id. If that object doesn't exist, then there is no v2 rbd image
2860 * with the supplied name.
2861 *
2862 * This function will record the given rbd_dev's image_id field if
2863 * it can be determined, and in that case will return 0. If any
2864 * errors occur a negative errno will be returned and the rbd_dev's
2865 * image_id field will be unchanged (and should be NULL).
2866 */
2867static int rbd_dev_image_id(struct rbd_device *rbd_dev)
2868{
2869 int ret;
2870 size_t size;
2871 char *object_name;
2872 void *response;
2873 void *p;
2874
2875 /*
2876 * First, see if the format 2 image id file exists, and if
2877 * so, get the image's persistent id from it.
2878 */
2879 size = sizeof (RBD_ID_PREFIX) + rbd_dev->image_name_len;
2880 object_name = kmalloc(size, GFP_NOIO);
2881 if (!object_name)
2882 return -ENOMEM;
2883 sprintf(object_name, "%s%s", RBD_ID_PREFIX, rbd_dev->image_name);
2884 dout("rbd id object name is %s\n", object_name);
2885
2886 /* Response will be an encoded string, which includes a length */
2887
2888 size = sizeof (__le32) + RBD_IMAGE_ID_LEN_MAX;
2889 response = kzalloc(size, GFP_NOIO);
2890 if (!response) {
2891 ret = -ENOMEM;
2892 goto out;
2893 }
2894
2895 ret = rbd_req_sync_exec(rbd_dev, object_name,
2896 "rbd", "get_id",
2897 NULL, 0,
2898 response, RBD_IMAGE_ID_LEN_MAX,
2899 CEPH_OSD_FLAG_READ, NULL);
2900 dout("%s: rbd_req_sync_exec returned %d\n", __func__, ret);
2901 if (ret < 0)
2902 goto out;
2903
2904 p = response;
2905 rbd_dev->image_id = ceph_extract_encoded_string(&p,
2906 p + RBD_IMAGE_ID_LEN_MAX,
2907 &rbd_dev->image_id_len,
2908 GFP_NOIO);
2909 if (IS_ERR(rbd_dev->image_id)) {
2910 ret = PTR_ERR(rbd_dev->image_id);
2911 rbd_dev->image_id = NULL;
2912 } else {
2913 dout("image_id is %s\n", rbd_dev->image_id);
2914 }
2915out:
2916 kfree(response);
2917 kfree(object_name);
2918
2919 return ret;
2920}
2921
a30b71b9
AE
2922static int rbd_dev_v1_probe(struct rbd_device *rbd_dev)
2923{
2924 int ret;
2925 size_t size;
2926
2927 /* Version 1 images have no id; empty string is used */
2928
2929 rbd_dev->image_id = kstrdup("", GFP_KERNEL);
2930 if (!rbd_dev->image_id)
2931 return -ENOMEM;
2932 rbd_dev->image_id_len = 0;
2933
2934 /* Record the header object name for this rbd image. */
2935
2936 size = rbd_dev->image_name_len + sizeof (RBD_SUFFIX);
2937 rbd_dev->header_name = kmalloc(size, GFP_KERNEL);
2938 if (!rbd_dev->header_name) {
2939 ret = -ENOMEM;
2940 goto out_err;
2941 }
2942 sprintf(rbd_dev->header_name, "%s%s", rbd_dev->image_name, RBD_SUFFIX);
2943
2944 /* Populate rbd image metadata */
2945
2946 ret = rbd_read_header(rbd_dev, &rbd_dev->header);
2947 if (ret < 0)
2948 goto out_err;
2949 rbd_dev->image_format = 1;
2950
2951 dout("discovered version 1 image, header name is %s\n",
2952 rbd_dev->header_name);
2953
2954 return 0;
2955
2956out_err:
2957 kfree(rbd_dev->header_name);
2958 rbd_dev->header_name = NULL;
2959 kfree(rbd_dev->image_id);
2960 rbd_dev->image_id = NULL;
2961
2962 return ret;
2963}
2964
2965static int rbd_dev_v2_probe(struct rbd_device *rbd_dev)
2966{
2967 size_t size;
9d475de5 2968 int ret;
6e14b1a6 2969 u64 ver = 0;
a30b71b9
AE
2970
2971 /*
2972 * Image id was filled in by the caller. Record the header
2973 * object name for this rbd image.
2974 */
2975 size = sizeof (RBD_HEADER_PREFIX) + rbd_dev->image_id_len;
2976 rbd_dev->header_name = kmalloc(size, GFP_KERNEL);
2977 if (!rbd_dev->header_name)
2978 return -ENOMEM;
2979 sprintf(rbd_dev->header_name, "%s%s",
2980 RBD_HEADER_PREFIX, rbd_dev->image_id);
9d475de5
AE
2981
2982 /* Get the size and object order for the image */
2983
2984 ret = rbd_dev_v2_image_size(rbd_dev);
1e130199
AE
2985 if (ret < 0)
2986 goto out_err;
2987
2988 /* Get the object prefix (a.k.a. block_name) for the image */
2989
2990 ret = rbd_dev_v2_object_prefix(rbd_dev);
b1b5402a
AE
2991 if (ret < 0)
2992 goto out_err;
2993
d889140c 2994 /* Get the and check features for the image */
b1b5402a
AE
2995
2996 ret = rbd_dev_v2_features(rbd_dev);
9d475de5
AE
2997 if (ret < 0)
2998 goto out_err;
35d489f9 2999
6e14b1a6
AE
3000 /* crypto and compression type aren't (yet) supported for v2 images */
3001
3002 rbd_dev->header.crypt_type = 0;
3003 rbd_dev->header.comp_type = 0;
35d489f9 3004
6e14b1a6
AE
3005 /* Get the snapshot context, plus the header version */
3006
3007 ret = rbd_dev_v2_snap_context(rbd_dev, &ver);
35d489f9
AE
3008 if (ret)
3009 goto out_err;
6e14b1a6
AE
3010 rbd_dev->header.obj_version = ver;
3011
a30b71b9
AE
3012 rbd_dev->image_format = 2;
3013
3014 dout("discovered version 2 image, header name is %s\n",
3015 rbd_dev->header_name);
3016
3017 return -ENOTSUPP;
9d475de5
AE
3018out_err:
3019 kfree(rbd_dev->header_name);
3020 rbd_dev->header_name = NULL;
1e130199
AE
3021 kfree(rbd_dev->header.object_prefix);
3022 rbd_dev->header.object_prefix = NULL;
9d475de5
AE
3023
3024 return ret;
a30b71b9
AE
3025}
3026
3027/*
3028 * Probe for the existence of the header object for the given rbd
3029 * device. For format 2 images this includes determining the image
3030 * id.
3031 */
3032static int rbd_dev_probe(struct rbd_device *rbd_dev)
3033{
3034 int ret;
3035
3036 /*
3037 * Get the id from the image id object. If it's not a
3038 * format 2 image, we'll get ENOENT back, and we'll assume
3039 * it's a format 1 image.
3040 */
3041 ret = rbd_dev_image_id(rbd_dev);
3042 if (ret)
3043 ret = rbd_dev_v1_probe(rbd_dev);
3044 else
3045 ret = rbd_dev_v2_probe(rbd_dev);
3046 if (ret)
3047 dout("probe failed, returning %d\n", ret);
3048
3049 return ret;
3050}
3051
59c2be1e
YS
3052static ssize_t rbd_add(struct bus_type *bus,
3053 const char *buf,
3054 size_t count)
602adf40 3055{
cb8627c7
AE
3056 char *options;
3057 struct rbd_device *rbd_dev = NULL;
7ef3214a
AE
3058 const char *mon_addrs = NULL;
3059 size_t mon_addrs_size = 0;
27cc2594
AE
3060 struct ceph_osd_client *osdc;
3061 int rc = -ENOMEM;
3feeb894 3062 char *snap_name;
602adf40
YS
3063
3064 if (!try_module_get(THIS_MODULE))
3065 return -ENODEV;
3066
60571c7d 3067 options = kmalloc(count, GFP_KERNEL);
602adf40 3068 if (!options)
85ae8926 3069 goto err_out_mem;
cb8627c7
AE
3070 rbd_dev = kzalloc(sizeof(*rbd_dev), GFP_KERNEL);
3071 if (!rbd_dev)
85ae8926 3072 goto err_out_mem;
602adf40
YS
3073
3074 /* static rbd_device initialization */
3075 spin_lock_init(&rbd_dev->lock);
3076 INIT_LIST_HEAD(&rbd_dev->node);
dfc5606d 3077 INIT_LIST_HEAD(&rbd_dev->snaps);
c666601a 3078 init_rwsem(&rbd_dev->header_rwsem);
602adf40 3079
602adf40 3080 /* parse add command */
3feeb894
AE
3081 snap_name = rbd_add_parse_args(rbd_dev, buf,
3082 &mon_addrs, &mon_addrs_size, options, count);
3083 if (IS_ERR(snap_name)) {
3084 rc = PTR_ERR(snap_name);
85ae8926 3085 goto err_out_mem;
3feeb894 3086 }
e124a82f 3087
f8c38929
AE
3088 rc = rbd_get_client(rbd_dev, mon_addrs, mon_addrs_size - 1, options);
3089 if (rc < 0)
85ae8926 3090 goto err_out_args;
602adf40 3091
602adf40 3092 /* pick the pool */
1dbb4399 3093 osdc = &rbd_dev->rbd_client->client->osdc;
602adf40
YS
3094 rc = ceph_pg_poolid_by_name(osdc->osdmap, rbd_dev->pool_name);
3095 if (rc < 0)
3096 goto err_out_client;
9bb2f334 3097 rbd_dev->pool_id = rc;
602adf40 3098
a30b71b9
AE
3099 rc = rbd_dev_probe(rbd_dev);
3100 if (rc < 0)
05fd6f6f
AE
3101 goto err_out_client;
3102
3103 /* no need to lock here, as rbd_dev is not registered yet */
3104 rc = rbd_dev_snaps_update(rbd_dev);
3105 if (rc)
3106 goto err_out_header;
3107
3108 rc = rbd_dev_set_mapping(rbd_dev, snap_name);
3109 if (rc)
3110 goto err_out_header;
3111
85ae8926
AE
3112 /* generate unique id: find highest unique id, add one */
3113 rbd_dev_id_get(rbd_dev);
3114
3115 /* Fill in the device name, now that we have its id. */
3116 BUILD_BUG_ON(DEV_NAME_LEN
3117 < sizeof (RBD_DRV_NAME) + MAX_INT_FORMAT_WIDTH);
3118 sprintf(rbd_dev->name, "%s%d", RBD_DRV_NAME, rbd_dev->dev_id);
3119
3120 /* Get our block major device number. */
3121
27cc2594
AE
3122 rc = register_blkdev(0, rbd_dev->name);
3123 if (rc < 0)
85ae8926 3124 goto err_out_id;
27cc2594 3125 rbd_dev->major = rc;
602adf40 3126
0f308a31
AE
3127 /* Set up the blkdev mapping. */
3128
3129 rc = rbd_init_disk(rbd_dev);
dfc5606d 3130 if (rc)
766fc439
YS
3131 goto err_out_blkdev;
3132
0f308a31
AE
3133 rc = rbd_bus_add_dev(rbd_dev);
3134 if (rc)
3135 goto err_out_disk;
3136
32eec68d
AE
3137 /*
3138 * At this point cleanup in the event of an error is the job
3139 * of the sysfs code (initiated by rbd_bus_del_dev()).
32eec68d 3140 */
2ac4e75d 3141
4bb1f1ed 3142 down_write(&rbd_dev->header_rwsem);
5ed16177 3143 rc = rbd_dev_snaps_register(rbd_dev);
4bb1f1ed 3144 up_write(&rbd_dev->header_rwsem);
2ac4e75d
AE
3145 if (rc)
3146 goto err_out_bus;
3147
3ee4001e
AE
3148 rc = rbd_init_watch_dev(rbd_dev);
3149 if (rc)
3150 goto err_out_bus;
3151
2ac4e75d
AE
3152 /* Everything's ready. Announce the disk to the world. */
3153
2ac4e75d 3154 add_disk(rbd_dev->disk);
3ee4001e 3155
2ac4e75d
AE
3156 pr_info("%s: added with size 0x%llx\n", rbd_dev->disk->disk_name,
3157 (unsigned long long) rbd_dev->mapping.size);
3158
602adf40
YS
3159 return count;
3160
766fc439 3161err_out_bus:
766fc439
YS
3162 /* this will also clean up rest of rbd_dev stuff */
3163
3164 rbd_bus_del_dev(rbd_dev);
3165 kfree(options);
766fc439
YS
3166 return rc;
3167
0f308a31
AE
3168err_out_disk:
3169 rbd_free_disk(rbd_dev);
602adf40
YS
3170err_out_blkdev:
3171 unregister_blkdev(rbd_dev->major, rbd_dev->name);
85ae8926
AE
3172err_out_id:
3173 rbd_dev_id_put(rbd_dev);
05fd6f6f
AE
3174err_out_header:
3175 rbd_header_free(&rbd_dev->header);
602adf40 3176err_out_client:
3fcf2581 3177 kfree(rbd_dev->header_name);
602adf40 3178 rbd_put_client(rbd_dev);
589d30e0 3179 kfree(rbd_dev->image_id);
85ae8926
AE
3180err_out_args:
3181 kfree(rbd_dev->mapping.snap_name);
3182 kfree(rbd_dev->image_name);
3183 kfree(rbd_dev->pool_name);
3184err_out_mem:
27cc2594 3185 kfree(rbd_dev);
cb8627c7 3186 kfree(options);
27cc2594 3187
602adf40
YS
3188 dout("Error adding device %s\n", buf);
3189 module_put(THIS_MODULE);
27cc2594
AE
3190
3191 return (ssize_t) rc;
602adf40
YS
3192}
3193
de71a297 3194static struct rbd_device *__rbd_get_dev(unsigned long dev_id)
602adf40
YS
3195{
3196 struct list_head *tmp;
3197 struct rbd_device *rbd_dev;
3198
e124a82f 3199 spin_lock(&rbd_dev_list_lock);
602adf40
YS
3200 list_for_each(tmp, &rbd_dev_list) {
3201 rbd_dev = list_entry(tmp, struct rbd_device, node);
de71a297 3202 if (rbd_dev->dev_id == dev_id) {
e124a82f 3203 spin_unlock(&rbd_dev_list_lock);
602adf40 3204 return rbd_dev;
e124a82f 3205 }
602adf40 3206 }
e124a82f 3207 spin_unlock(&rbd_dev_list_lock);
602adf40
YS
3208 return NULL;
3209}
3210
dfc5606d 3211static void rbd_dev_release(struct device *dev)
602adf40 3212{
593a9e7b 3213 struct rbd_device *rbd_dev = dev_to_rbd_dev(dev);
602adf40 3214
1dbb4399
AE
3215 if (rbd_dev->watch_request) {
3216 struct ceph_client *client = rbd_dev->rbd_client->client;
3217
3218 ceph_osdc_unregister_linger_request(&client->osdc,
59c2be1e 3219 rbd_dev->watch_request);
1dbb4399 3220 }
59c2be1e 3221 if (rbd_dev->watch_event)
070c633f 3222 rbd_req_sync_unwatch(rbd_dev);
59c2be1e 3223
602adf40
YS
3224 rbd_put_client(rbd_dev);
3225
3226 /* clean up and free blkdev */
3227 rbd_free_disk(rbd_dev);
3228 unregister_blkdev(rbd_dev->major, rbd_dev->name);
32eec68d 3229
2ac4e75d
AE
3230 /* release allocated disk header fields */
3231 rbd_header_free(&rbd_dev->header);
3232
32eec68d 3233 /* done with the id, and with the rbd_dev */
f84344f3 3234 kfree(rbd_dev->mapping.snap_name);
589d30e0 3235 kfree(rbd_dev->image_id);
0bed54dc 3236 kfree(rbd_dev->header_name);
d22f76e7 3237 kfree(rbd_dev->pool_name);
0bed54dc 3238 kfree(rbd_dev->image_name);
e2839308 3239 rbd_dev_id_put(rbd_dev);
602adf40
YS
3240 kfree(rbd_dev);
3241
3242 /* release module ref */
3243 module_put(THIS_MODULE);
602adf40
YS
3244}
3245
dfc5606d
YS
3246static ssize_t rbd_remove(struct bus_type *bus,
3247 const char *buf,
3248 size_t count)
602adf40
YS
3249{
3250 struct rbd_device *rbd_dev = NULL;
3251 int target_id, rc;
3252 unsigned long ul;
3253 int ret = count;
3254
3255 rc = strict_strtoul(buf, 10, &ul);
3256 if (rc)
3257 return rc;
3258
3259 /* convert to int; abort if we lost anything in the conversion */
3260 target_id = (int) ul;
3261 if (target_id != ul)
3262 return -EINVAL;
3263
3264 mutex_lock_nested(&ctl_mutex, SINGLE_DEPTH_NESTING);
3265
3266 rbd_dev = __rbd_get_dev(target_id);
3267 if (!rbd_dev) {
3268 ret = -ENOENT;
3269 goto done;
3270 }
3271
dfc5606d
YS
3272 __rbd_remove_all_snaps(rbd_dev);
3273 rbd_bus_del_dev(rbd_dev);
602adf40
YS
3274
3275done:
3276 mutex_unlock(&ctl_mutex);
aafb230e 3277
602adf40
YS
3278 return ret;
3279}
3280
602adf40
YS
3281/*
3282 * create control files in sysfs
dfc5606d 3283 * /sys/bus/rbd/...
602adf40
YS
3284 */
3285static int rbd_sysfs_init(void)
3286{
dfc5606d 3287 int ret;
602adf40 3288
fed4c143 3289 ret = device_register(&rbd_root_dev);
21079786 3290 if (ret < 0)
dfc5606d 3291 return ret;
602adf40 3292
fed4c143
AE
3293 ret = bus_register(&rbd_bus_type);
3294 if (ret < 0)
3295 device_unregister(&rbd_root_dev);
602adf40 3296
602adf40
YS
3297 return ret;
3298}
3299
3300static void rbd_sysfs_cleanup(void)
3301{
dfc5606d 3302 bus_unregister(&rbd_bus_type);
fed4c143 3303 device_unregister(&rbd_root_dev);
602adf40
YS
3304}
3305
3306int __init rbd_init(void)
3307{
3308 int rc;
3309
3310 rc = rbd_sysfs_init();
3311 if (rc)
3312 return rc;
f0f8cef5 3313 pr_info("loaded " RBD_DRV_NAME_LONG "\n");
602adf40
YS
3314 return 0;
3315}
3316
3317void __exit rbd_exit(void)
3318{
3319 rbd_sysfs_cleanup();
3320}
3321
3322module_init(rbd_init);
3323module_exit(rbd_exit);
3324
3325MODULE_AUTHOR("Sage Weil <sage@newdream.net>");
3326MODULE_AUTHOR("Yehuda Sadeh <yehuda@hq.newdream.net>");
3327MODULE_DESCRIPTION("rados block device");
3328
3329/* following authorship retained from original osdblk.c */
3330MODULE_AUTHOR("Jeff Garzik <jeff@garzik.org>");
3331
3332MODULE_LICENSE("GPL");