smsc75xx: add more information to register io failure warnings
[deliverable/linux.git] / drivers / block / rbd.c
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
24 For usage instructions, please refer to:
25
26 Documentation/ABI/testing/sysfs-bus-rbd
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>
34 #include <linux/parser.h>
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
44 /*
45 * The basic unit of block I/O is a sector. It is interpreted in a
46 * number of contexts in Linux (blk, bio, genhd), but the default is
47 * universally 512 bytes. These symbols are just slightly more
48 * meaningful than the bare numbers they represent.
49 */
50 #define SECTOR_SHIFT 9
51 #define SECTOR_SIZE (1ULL << SECTOR_SHIFT)
52
53 #define RBD_DRV_NAME "rbd"
54 #define RBD_DRV_NAME_LONG "rbd (rados block device)"
55
56 #define RBD_MINORS_PER_MAJOR 256 /* max minors per blkdev */
57
58 #define RBD_MAX_MD_NAME_LEN (RBD_MAX_OBJ_NAME_LEN + sizeof(RBD_SUFFIX))
59 #define RBD_MAX_POOL_NAME_LEN 64
60 #define RBD_MAX_SNAP_NAME_LEN 32
61 #define RBD_MAX_OPT_LEN 1024
62
63 #define RBD_SNAP_HEAD_NAME "-"
64
65 /*
66 * An RBD device name will be "rbd#", where the "rbd" comes from
67 * RBD_DRV_NAME above, and # is a unique integer identifier.
68 * MAX_INT_FORMAT_WIDTH is used in ensuring DEV_NAME_LEN is big
69 * enough to hold all possible device names.
70 */
71 #define DEV_NAME_LEN 32
72 #define MAX_INT_FORMAT_WIDTH ((5 * sizeof (int)) / 2 + 1)
73
74 #define RBD_NOTIFY_TIMEOUT_DEFAULT 10
75
76 /*
77 * block device image metadata (in-memory version)
78 */
79 struct rbd_image_header {
80 u64 image_size;
81 char block_name[32];
82 __u8 obj_order;
83 __u8 crypt_type;
84 __u8 comp_type;
85 struct ceph_snap_context *snapc;
86 size_t snap_names_len;
87 u64 snap_seq;
88 u32 total_snaps;
89
90 char *snap_names;
91 u64 *snap_sizes;
92
93 u64 obj_version;
94 };
95
96 struct rbd_options {
97 int notify_timeout;
98 };
99
100 /*
101 * an instance of the client. multiple devices may share an rbd client.
102 */
103 struct rbd_client {
104 struct ceph_client *client;
105 struct rbd_options *rbd_opts;
106 struct kref kref;
107 struct list_head node;
108 };
109
110 /*
111 * a request completion status
112 */
113 struct rbd_req_status {
114 int done;
115 int rc;
116 u64 bytes;
117 };
118
119 /*
120 * a collection of requests
121 */
122 struct rbd_req_coll {
123 int total;
124 int num_done;
125 struct kref kref;
126 struct rbd_req_status status[0];
127 };
128
129 /*
130 * a single io request
131 */
132 struct rbd_request {
133 struct request *rq; /* blk layer request */
134 struct bio *bio; /* cloned bio */
135 struct page **pages; /* list of used pages */
136 u64 len;
137 int coll_index;
138 struct rbd_req_coll *coll;
139 };
140
141 struct rbd_snap {
142 struct device dev;
143 const char *name;
144 size_t size;
145 struct list_head node;
146 u64 id;
147 };
148
149 /*
150 * a single device
151 */
152 struct rbd_device {
153 int id; /* blkdev unique id */
154
155 int major; /* blkdev assigned major */
156 struct gendisk *disk; /* blkdev's gendisk and rq */
157 struct request_queue *q;
158
159 struct rbd_client *rbd_client;
160
161 char name[DEV_NAME_LEN]; /* blkdev name, e.g. rbd3 */
162
163 spinlock_t lock; /* queue lock */
164
165 struct rbd_image_header header;
166 char obj[RBD_MAX_OBJ_NAME_LEN]; /* rbd image name */
167 int obj_len;
168 char obj_md_name[RBD_MAX_MD_NAME_LEN]; /* hdr nm. */
169 char pool_name[RBD_MAX_POOL_NAME_LEN];
170 int poolid;
171
172 struct ceph_osd_event *watch_event;
173 struct ceph_osd_request *watch_request;
174
175 /* protects updating the header */
176 struct rw_semaphore header_rwsem;
177 char snap_name[RBD_MAX_SNAP_NAME_LEN];
178 u32 cur_snap; /* index+1 of current snapshot within snap context
179 0 - for the head */
180 int read_only;
181
182 struct list_head node;
183
184 /* list of snapshots */
185 struct list_head snaps;
186
187 /* sysfs related */
188 struct device dev;
189 };
190
191 static DEFINE_MUTEX(ctl_mutex); /* Serialize open/close/setup/teardown */
192
193 static LIST_HEAD(rbd_dev_list); /* devices */
194 static DEFINE_SPINLOCK(rbd_dev_list_lock);
195
196 static LIST_HEAD(rbd_client_list); /* clients */
197 static DEFINE_SPINLOCK(rbd_client_list_lock);
198
199 static int __rbd_init_snaps_header(struct rbd_device *rbd_dev);
200 static void rbd_dev_release(struct device *dev);
201 static ssize_t rbd_snap_add(struct device *dev,
202 struct device_attribute *attr,
203 const char *buf,
204 size_t count);
205 static void __rbd_remove_snap_dev(struct rbd_device *rbd_dev,
206 struct rbd_snap *snap);
207
208 static ssize_t rbd_add(struct bus_type *bus, const char *buf,
209 size_t count);
210 static ssize_t rbd_remove(struct bus_type *bus, const char *buf,
211 size_t count);
212
213 static struct bus_attribute rbd_bus_attrs[] = {
214 __ATTR(add, S_IWUSR, NULL, rbd_add),
215 __ATTR(remove, S_IWUSR, NULL, rbd_remove),
216 __ATTR_NULL
217 };
218
219 static struct bus_type rbd_bus_type = {
220 .name = "rbd",
221 .bus_attrs = rbd_bus_attrs,
222 };
223
224 static void rbd_root_dev_release(struct device *dev)
225 {
226 }
227
228 static struct device rbd_root_dev = {
229 .init_name = "rbd",
230 .release = rbd_root_dev_release,
231 };
232
233
234 static struct device *rbd_get_dev(struct rbd_device *rbd_dev)
235 {
236 return get_device(&rbd_dev->dev);
237 }
238
239 static void rbd_put_dev(struct rbd_device *rbd_dev)
240 {
241 put_device(&rbd_dev->dev);
242 }
243
244 static int __rbd_update_snaps(struct rbd_device *rbd_dev);
245
246 static int rbd_open(struct block_device *bdev, fmode_t mode)
247 {
248 struct rbd_device *rbd_dev = bdev->bd_disk->private_data;
249
250 rbd_get_dev(rbd_dev);
251
252 set_device_ro(bdev, rbd_dev->read_only);
253
254 if ((mode & FMODE_WRITE) && rbd_dev->read_only)
255 return -EROFS;
256
257 return 0;
258 }
259
260 static int rbd_release(struct gendisk *disk, fmode_t mode)
261 {
262 struct rbd_device *rbd_dev = disk->private_data;
263
264 rbd_put_dev(rbd_dev);
265
266 return 0;
267 }
268
269 static const struct block_device_operations rbd_bd_ops = {
270 .owner = THIS_MODULE,
271 .open = rbd_open,
272 .release = rbd_release,
273 };
274
275 /*
276 * Initialize an rbd client instance.
277 * We own *opt.
278 */
279 static struct rbd_client *rbd_client_create(struct ceph_options *opt,
280 struct rbd_options *rbd_opts)
281 {
282 struct rbd_client *rbdc;
283 int ret = -ENOMEM;
284
285 dout("rbd_client_create\n");
286 rbdc = kmalloc(sizeof(struct rbd_client), GFP_KERNEL);
287 if (!rbdc)
288 goto out_opt;
289
290 kref_init(&rbdc->kref);
291 INIT_LIST_HEAD(&rbdc->node);
292
293 mutex_lock_nested(&ctl_mutex, SINGLE_DEPTH_NESTING);
294
295 rbdc->client = ceph_create_client(opt, rbdc, 0, 0);
296 if (IS_ERR(rbdc->client))
297 goto out_mutex;
298 opt = NULL; /* Now rbdc->client is responsible for opt */
299
300 ret = ceph_open_session(rbdc->client);
301 if (ret < 0)
302 goto out_err;
303
304 rbdc->rbd_opts = rbd_opts;
305
306 spin_lock(&rbd_client_list_lock);
307 list_add_tail(&rbdc->node, &rbd_client_list);
308 spin_unlock(&rbd_client_list_lock);
309
310 mutex_unlock(&ctl_mutex);
311
312 dout("rbd_client_create created %p\n", rbdc);
313 return rbdc;
314
315 out_err:
316 ceph_destroy_client(rbdc->client);
317 out_mutex:
318 mutex_unlock(&ctl_mutex);
319 kfree(rbdc);
320 out_opt:
321 if (opt)
322 ceph_destroy_options(opt);
323 return ERR_PTR(ret);
324 }
325
326 /*
327 * Find a ceph client with specific addr and configuration.
328 */
329 static struct rbd_client *__rbd_client_find(struct ceph_options *opt)
330 {
331 struct rbd_client *client_node;
332
333 if (opt->flags & CEPH_OPT_NOSHARE)
334 return NULL;
335
336 list_for_each_entry(client_node, &rbd_client_list, node)
337 if (ceph_compare_options(opt, client_node->client) == 0)
338 return client_node;
339 return NULL;
340 }
341
342 /*
343 * mount options
344 */
345 enum {
346 Opt_notify_timeout,
347 Opt_last_int,
348 /* int args above */
349 Opt_last_string,
350 /* string args above */
351 };
352
353 static match_table_t rbdopt_tokens = {
354 {Opt_notify_timeout, "notify_timeout=%d"},
355 /* int args above */
356 /* string args above */
357 {-1, NULL}
358 };
359
360 static int parse_rbd_opts_token(char *c, void *private)
361 {
362 struct rbd_options *rbdopt = private;
363 substring_t argstr[MAX_OPT_ARGS];
364 int token, intval, ret;
365
366 token = match_token(c, rbdopt_tokens, argstr);
367 if (token < 0)
368 return -EINVAL;
369
370 if (token < Opt_last_int) {
371 ret = match_int(&argstr[0], &intval);
372 if (ret < 0) {
373 pr_err("bad mount option arg (not int) "
374 "at '%s'\n", c);
375 return ret;
376 }
377 dout("got int token %d val %d\n", token, intval);
378 } else if (token > Opt_last_int && token < Opt_last_string) {
379 dout("got string token %d val %s\n", token,
380 argstr[0].from);
381 } else {
382 dout("got token %d\n", token);
383 }
384
385 switch (token) {
386 case Opt_notify_timeout:
387 rbdopt->notify_timeout = intval;
388 break;
389 default:
390 BUG_ON(token);
391 }
392 return 0;
393 }
394
395 /*
396 * Get a ceph client with specific addr and configuration, if one does
397 * not exist create it.
398 */
399 static struct rbd_client *rbd_get_client(const char *mon_addr,
400 size_t mon_addr_len,
401 char *options)
402 {
403 struct rbd_client *rbdc;
404 struct ceph_options *opt;
405 struct rbd_options *rbd_opts;
406
407 rbd_opts = kzalloc(sizeof(*rbd_opts), GFP_KERNEL);
408 if (!rbd_opts)
409 return ERR_PTR(-ENOMEM);
410
411 rbd_opts->notify_timeout = RBD_NOTIFY_TIMEOUT_DEFAULT;
412
413 opt = ceph_parse_options(options, mon_addr,
414 mon_addr + mon_addr_len,
415 parse_rbd_opts_token, rbd_opts);
416 if (IS_ERR(opt)) {
417 kfree(rbd_opts);
418 return ERR_CAST(opt);
419 }
420
421 spin_lock(&rbd_client_list_lock);
422 rbdc = __rbd_client_find(opt);
423 if (rbdc) {
424 /* using an existing client */
425 kref_get(&rbdc->kref);
426 spin_unlock(&rbd_client_list_lock);
427
428 ceph_destroy_options(opt);
429 kfree(rbd_opts);
430
431 return rbdc;
432 }
433 spin_unlock(&rbd_client_list_lock);
434
435 rbdc = rbd_client_create(opt, rbd_opts);
436
437 if (IS_ERR(rbdc))
438 kfree(rbd_opts);
439
440 return rbdc;
441 }
442
443 /*
444 * Destroy ceph client
445 *
446 * Caller must hold rbd_client_list_lock.
447 */
448 static void rbd_client_release(struct kref *kref)
449 {
450 struct rbd_client *rbdc = container_of(kref, struct rbd_client, kref);
451
452 dout("rbd_release_client %p\n", rbdc);
453 list_del(&rbdc->node);
454
455 ceph_destroy_client(rbdc->client);
456 kfree(rbdc->rbd_opts);
457 kfree(rbdc);
458 }
459
460 /*
461 * Drop reference to ceph client node. If it's not referenced anymore, release
462 * it.
463 */
464 static void rbd_put_client(struct rbd_device *rbd_dev)
465 {
466 spin_lock(&rbd_client_list_lock);
467 kref_put(&rbd_dev->rbd_client->kref, rbd_client_release);
468 spin_unlock(&rbd_client_list_lock);
469 rbd_dev->rbd_client = NULL;
470 }
471
472 /*
473 * Destroy requests collection
474 */
475 static void rbd_coll_release(struct kref *kref)
476 {
477 struct rbd_req_coll *coll =
478 container_of(kref, struct rbd_req_coll, kref);
479
480 dout("rbd_coll_release %p\n", coll);
481 kfree(coll);
482 }
483
484 /*
485 * Create a new header structure, translate header format from the on-disk
486 * header.
487 */
488 static int rbd_header_from_disk(struct rbd_image_header *header,
489 struct rbd_image_header_ondisk *ondisk,
490 int allocated_snaps,
491 gfp_t gfp_flags)
492 {
493 int i;
494 u32 snap_count;
495
496 if (memcmp(ondisk, RBD_HEADER_TEXT, sizeof(RBD_HEADER_TEXT)))
497 return -ENXIO;
498
499 snap_count = le32_to_cpu(ondisk->snap_count);
500 header->snapc = kmalloc(sizeof(struct ceph_snap_context) +
501 snap_count * sizeof (*ondisk),
502 gfp_flags);
503 if (!header->snapc)
504 return -ENOMEM;
505
506 header->snap_names_len = le64_to_cpu(ondisk->snap_names_len);
507 if (snap_count) {
508 header->snap_names = kmalloc(header->snap_names_len,
509 GFP_KERNEL);
510 if (!header->snap_names)
511 goto err_snapc;
512 header->snap_sizes = kmalloc(snap_count * sizeof(u64),
513 GFP_KERNEL);
514 if (!header->snap_sizes)
515 goto err_names;
516 } else {
517 header->snap_names = NULL;
518 header->snap_sizes = NULL;
519 }
520 memcpy(header->block_name, ondisk->block_name,
521 sizeof(ondisk->block_name));
522
523 header->image_size = le64_to_cpu(ondisk->image_size);
524 header->obj_order = ondisk->options.order;
525 header->crypt_type = ondisk->options.crypt_type;
526 header->comp_type = ondisk->options.comp_type;
527
528 atomic_set(&header->snapc->nref, 1);
529 header->snap_seq = le64_to_cpu(ondisk->snap_seq);
530 header->snapc->num_snaps = snap_count;
531 header->total_snaps = snap_count;
532
533 if (snap_count && allocated_snaps == snap_count) {
534 for (i = 0; i < snap_count; i++) {
535 header->snapc->snaps[i] =
536 le64_to_cpu(ondisk->snaps[i].id);
537 header->snap_sizes[i] =
538 le64_to_cpu(ondisk->snaps[i].image_size);
539 }
540
541 /* copy snapshot names */
542 memcpy(header->snap_names, &ondisk->snaps[i],
543 header->snap_names_len);
544 }
545
546 return 0;
547
548 err_names:
549 kfree(header->snap_names);
550 err_snapc:
551 kfree(header->snapc);
552 return -ENOMEM;
553 }
554
555 static int snap_index(struct rbd_image_header *header, int snap_num)
556 {
557 return header->total_snaps - snap_num;
558 }
559
560 static u64 cur_snap_id(struct rbd_device *rbd_dev)
561 {
562 struct rbd_image_header *header = &rbd_dev->header;
563
564 if (!rbd_dev->cur_snap)
565 return 0;
566
567 return header->snapc->snaps[snap_index(header, rbd_dev->cur_snap)];
568 }
569
570 static int snap_by_name(struct rbd_image_header *header, const char *snap_name,
571 u64 *seq, u64 *size)
572 {
573 int i;
574 char *p = header->snap_names;
575
576 for (i = 0; i < header->total_snaps; i++) {
577 if (!strcmp(snap_name, p)) {
578
579 /* Found it. Pass back its id and/or size */
580
581 if (seq)
582 *seq = header->snapc->snaps[i];
583 if (size)
584 *size = header->snap_sizes[i];
585 return i;
586 }
587 p += strlen(p) + 1; /* Skip ahead to the next name */
588 }
589 return -ENOENT;
590 }
591
592 static int rbd_header_set_snap(struct rbd_device *dev, u64 *size)
593 {
594 struct rbd_image_header *header = &dev->header;
595 struct ceph_snap_context *snapc = header->snapc;
596 int ret = -ENOENT;
597
598 BUILD_BUG_ON(sizeof (dev->snap_name) < sizeof (RBD_SNAP_HEAD_NAME));
599
600 down_write(&dev->header_rwsem);
601
602 if (!memcmp(dev->snap_name, RBD_SNAP_HEAD_NAME,
603 sizeof (RBD_SNAP_HEAD_NAME))) {
604 if (header->total_snaps)
605 snapc->seq = header->snap_seq;
606 else
607 snapc->seq = 0;
608 dev->cur_snap = 0;
609 dev->read_only = 0;
610 if (size)
611 *size = header->image_size;
612 } else {
613 ret = snap_by_name(header, dev->snap_name, &snapc->seq, size);
614 if (ret < 0)
615 goto done;
616
617 dev->cur_snap = header->total_snaps - ret;
618 dev->read_only = 1;
619 }
620
621 ret = 0;
622 done:
623 up_write(&dev->header_rwsem);
624 return ret;
625 }
626
627 static void rbd_header_free(struct rbd_image_header *header)
628 {
629 kfree(header->snapc);
630 kfree(header->snap_names);
631 kfree(header->snap_sizes);
632 }
633
634 /*
635 * get the actual striped segment name, offset and length
636 */
637 static u64 rbd_get_segment(struct rbd_image_header *header,
638 const char *block_name,
639 u64 ofs, u64 len,
640 char *seg_name, u64 *segofs)
641 {
642 u64 seg = ofs >> header->obj_order;
643
644 if (seg_name)
645 snprintf(seg_name, RBD_MAX_SEG_NAME_LEN,
646 "%s.%012llx", block_name, seg);
647
648 ofs = ofs & ((1 << header->obj_order) - 1);
649 len = min_t(u64, len, (1 << header->obj_order) - ofs);
650
651 if (segofs)
652 *segofs = ofs;
653
654 return len;
655 }
656
657 static int rbd_get_num_segments(struct rbd_image_header *header,
658 u64 ofs, u64 len)
659 {
660 u64 start_seg = ofs >> header->obj_order;
661 u64 end_seg = (ofs + len - 1) >> header->obj_order;
662 return end_seg - start_seg + 1;
663 }
664
665 /*
666 * returns the size of an object in the image
667 */
668 static u64 rbd_obj_bytes(struct rbd_image_header *header)
669 {
670 return 1 << header->obj_order;
671 }
672
673 /*
674 * bio helpers
675 */
676
677 static void bio_chain_put(struct bio *chain)
678 {
679 struct bio *tmp;
680
681 while (chain) {
682 tmp = chain;
683 chain = chain->bi_next;
684 bio_put(tmp);
685 }
686 }
687
688 /*
689 * zeros a bio chain, starting at specific offset
690 */
691 static void zero_bio_chain(struct bio *chain, int start_ofs)
692 {
693 struct bio_vec *bv;
694 unsigned long flags;
695 void *buf;
696 int i;
697 int pos = 0;
698
699 while (chain) {
700 bio_for_each_segment(bv, chain, i) {
701 if (pos + bv->bv_len > start_ofs) {
702 int remainder = max(start_ofs - pos, 0);
703 buf = bvec_kmap_irq(bv, &flags);
704 memset(buf + remainder, 0,
705 bv->bv_len - remainder);
706 bvec_kunmap_irq(buf, &flags);
707 }
708 pos += bv->bv_len;
709 }
710
711 chain = chain->bi_next;
712 }
713 }
714
715 /*
716 * bio_chain_clone - clone a chain of bios up to a certain length.
717 * might return a bio_pair that will need to be released.
718 */
719 static struct bio *bio_chain_clone(struct bio **old, struct bio **next,
720 struct bio_pair **bp,
721 int len, gfp_t gfpmask)
722 {
723 struct bio *tmp, *old_chain = *old, *new_chain = NULL, *tail = NULL;
724 int total = 0;
725
726 if (*bp) {
727 bio_pair_release(*bp);
728 *bp = NULL;
729 }
730
731 while (old_chain && (total < len)) {
732 tmp = bio_kmalloc(gfpmask, old_chain->bi_max_vecs);
733 if (!tmp)
734 goto err_out;
735
736 if (total + old_chain->bi_size > len) {
737 struct bio_pair *bp;
738
739 /*
740 * this split can only happen with a single paged bio,
741 * split_bio will BUG_ON if this is not the case
742 */
743 dout("bio_chain_clone split! total=%d remaining=%d"
744 "bi_size=%d\n",
745 (int)total, (int)len-total,
746 (int)old_chain->bi_size);
747
748 /* split the bio. We'll release it either in the next
749 call, or it will have to be released outside */
750 bp = bio_split(old_chain, (len - total) / SECTOR_SIZE);
751 if (!bp)
752 goto err_out;
753
754 __bio_clone(tmp, &bp->bio1);
755
756 *next = &bp->bio2;
757 } else {
758 __bio_clone(tmp, old_chain);
759 *next = old_chain->bi_next;
760 }
761
762 tmp->bi_bdev = NULL;
763 gfpmask &= ~__GFP_WAIT;
764 tmp->bi_next = NULL;
765
766 if (!new_chain) {
767 new_chain = tail = tmp;
768 } else {
769 tail->bi_next = tmp;
770 tail = tmp;
771 }
772 old_chain = old_chain->bi_next;
773
774 total += tmp->bi_size;
775 }
776
777 BUG_ON(total < len);
778
779 if (tail)
780 tail->bi_next = NULL;
781
782 *old = old_chain;
783
784 return new_chain;
785
786 err_out:
787 dout("bio_chain_clone with err\n");
788 bio_chain_put(new_chain);
789 return NULL;
790 }
791
792 /*
793 * helpers for osd request op vectors.
794 */
795 static int rbd_create_rw_ops(struct ceph_osd_req_op **ops,
796 int num_ops,
797 int opcode,
798 u32 payload_len)
799 {
800 *ops = kzalloc(sizeof(struct ceph_osd_req_op) * (num_ops + 1),
801 GFP_NOIO);
802 if (!*ops)
803 return -ENOMEM;
804 (*ops)[0].op = opcode;
805 /*
806 * op extent offset and length will be set later on
807 * in calc_raw_layout()
808 */
809 (*ops)[0].payload_len = payload_len;
810 return 0;
811 }
812
813 static void rbd_destroy_ops(struct ceph_osd_req_op *ops)
814 {
815 kfree(ops);
816 }
817
818 static void rbd_coll_end_req_index(struct request *rq,
819 struct rbd_req_coll *coll,
820 int index,
821 int ret, u64 len)
822 {
823 struct request_queue *q;
824 int min, max, i;
825
826 dout("rbd_coll_end_req_index %p index %d ret %d len %lld\n",
827 coll, index, ret, len);
828
829 if (!rq)
830 return;
831
832 if (!coll) {
833 blk_end_request(rq, ret, len);
834 return;
835 }
836
837 q = rq->q;
838
839 spin_lock_irq(q->queue_lock);
840 coll->status[index].done = 1;
841 coll->status[index].rc = ret;
842 coll->status[index].bytes = len;
843 max = min = coll->num_done;
844 while (max < coll->total && coll->status[max].done)
845 max++;
846
847 for (i = min; i<max; i++) {
848 __blk_end_request(rq, coll->status[i].rc,
849 coll->status[i].bytes);
850 coll->num_done++;
851 kref_put(&coll->kref, rbd_coll_release);
852 }
853 spin_unlock_irq(q->queue_lock);
854 }
855
856 static void rbd_coll_end_req(struct rbd_request *req,
857 int ret, u64 len)
858 {
859 rbd_coll_end_req_index(req->rq, req->coll, req->coll_index, ret, len);
860 }
861
862 /*
863 * Send ceph osd request
864 */
865 static int rbd_do_request(struct request *rq,
866 struct rbd_device *dev,
867 struct ceph_snap_context *snapc,
868 u64 snapid,
869 const char *obj, u64 ofs, u64 len,
870 struct bio *bio,
871 struct page **pages,
872 int num_pages,
873 int flags,
874 struct ceph_osd_req_op *ops,
875 int num_reply,
876 struct rbd_req_coll *coll,
877 int coll_index,
878 void (*rbd_cb)(struct ceph_osd_request *req,
879 struct ceph_msg *msg),
880 struct ceph_osd_request **linger_req,
881 u64 *ver)
882 {
883 struct ceph_osd_request *req;
884 struct ceph_file_layout *layout;
885 int ret;
886 u64 bno;
887 struct timespec mtime = CURRENT_TIME;
888 struct rbd_request *req_data;
889 struct ceph_osd_request_head *reqhead;
890 struct ceph_osd_client *osdc;
891
892 req_data = kzalloc(sizeof(*req_data), GFP_NOIO);
893 if (!req_data) {
894 if (coll)
895 rbd_coll_end_req_index(rq, coll, coll_index,
896 -ENOMEM, len);
897 return -ENOMEM;
898 }
899
900 if (coll) {
901 req_data->coll = coll;
902 req_data->coll_index = coll_index;
903 }
904
905 dout("rbd_do_request obj=%s ofs=%lld len=%lld\n", obj, len, ofs);
906
907 down_read(&dev->header_rwsem);
908
909 osdc = &dev->rbd_client->client->osdc;
910 req = ceph_osdc_alloc_request(osdc, flags, snapc, ops,
911 false, GFP_NOIO, pages, bio);
912 if (!req) {
913 up_read(&dev->header_rwsem);
914 ret = -ENOMEM;
915 goto done_pages;
916 }
917
918 req->r_callback = rbd_cb;
919
920 req_data->rq = rq;
921 req_data->bio = bio;
922 req_data->pages = pages;
923 req_data->len = len;
924
925 req->r_priv = req_data;
926
927 reqhead = req->r_request->front.iov_base;
928 reqhead->snapid = cpu_to_le64(CEPH_NOSNAP);
929
930 strncpy(req->r_oid, obj, sizeof(req->r_oid));
931 req->r_oid_len = strlen(req->r_oid);
932
933 layout = &req->r_file_layout;
934 memset(layout, 0, sizeof(*layout));
935 layout->fl_stripe_unit = cpu_to_le32(1 << RBD_MAX_OBJ_ORDER);
936 layout->fl_stripe_count = cpu_to_le32(1);
937 layout->fl_object_size = cpu_to_le32(1 << RBD_MAX_OBJ_ORDER);
938 layout->fl_pg_preferred = cpu_to_le32(-1);
939 layout->fl_pg_pool = cpu_to_le32(dev->poolid);
940 ceph_calc_raw_layout(osdc, layout, snapid, ofs, &len, &bno,
941 req, ops);
942
943 ceph_osdc_build_request(req, ofs, &len,
944 ops,
945 snapc,
946 &mtime,
947 req->r_oid, req->r_oid_len);
948 up_read(&dev->header_rwsem);
949
950 if (linger_req) {
951 ceph_osdc_set_request_linger(osdc, req);
952 *linger_req = req;
953 }
954
955 ret = ceph_osdc_start_request(osdc, req, false);
956 if (ret < 0)
957 goto done_err;
958
959 if (!rbd_cb) {
960 ret = ceph_osdc_wait_request(osdc, req);
961 if (ver)
962 *ver = le64_to_cpu(req->r_reassert_version.version);
963 dout("reassert_ver=%lld\n",
964 le64_to_cpu(req->r_reassert_version.version));
965 ceph_osdc_put_request(req);
966 }
967 return ret;
968
969 done_err:
970 bio_chain_put(req_data->bio);
971 ceph_osdc_put_request(req);
972 done_pages:
973 rbd_coll_end_req(req_data, ret, len);
974 kfree(req_data);
975 return ret;
976 }
977
978 /*
979 * Ceph osd op callback
980 */
981 static void rbd_req_cb(struct ceph_osd_request *req, struct ceph_msg *msg)
982 {
983 struct rbd_request *req_data = req->r_priv;
984 struct ceph_osd_reply_head *replyhead;
985 struct ceph_osd_op *op;
986 __s32 rc;
987 u64 bytes;
988 int read_op;
989
990 /* parse reply */
991 replyhead = msg->front.iov_base;
992 WARN_ON(le32_to_cpu(replyhead->num_ops) == 0);
993 op = (void *)(replyhead + 1);
994 rc = le32_to_cpu(replyhead->result);
995 bytes = le64_to_cpu(op->extent.length);
996 read_op = (le32_to_cpu(op->op) == CEPH_OSD_OP_READ);
997
998 dout("rbd_req_cb bytes=%lld readop=%d rc=%d\n", bytes, read_op, rc);
999
1000 if (rc == -ENOENT && read_op) {
1001 zero_bio_chain(req_data->bio, 0);
1002 rc = 0;
1003 } else if (rc == 0 && read_op && bytes < req_data->len) {
1004 zero_bio_chain(req_data->bio, bytes);
1005 bytes = req_data->len;
1006 }
1007
1008 rbd_coll_end_req(req_data, rc, bytes);
1009
1010 if (req_data->bio)
1011 bio_chain_put(req_data->bio);
1012
1013 ceph_osdc_put_request(req);
1014 kfree(req_data);
1015 }
1016
1017 static void rbd_simple_req_cb(struct ceph_osd_request *req, struct ceph_msg *msg)
1018 {
1019 ceph_osdc_put_request(req);
1020 }
1021
1022 /*
1023 * Do a synchronous ceph osd operation
1024 */
1025 static int rbd_req_sync_op(struct rbd_device *dev,
1026 struct ceph_snap_context *snapc,
1027 u64 snapid,
1028 int opcode,
1029 int flags,
1030 struct ceph_osd_req_op *orig_ops,
1031 int num_reply,
1032 const char *obj,
1033 u64 ofs, u64 len,
1034 char *buf,
1035 struct ceph_osd_request **linger_req,
1036 u64 *ver)
1037 {
1038 int ret;
1039 struct page **pages;
1040 int num_pages;
1041 struct ceph_osd_req_op *ops = orig_ops;
1042 u32 payload_len;
1043
1044 num_pages = calc_pages_for(ofs , len);
1045 pages = ceph_alloc_page_vector(num_pages, GFP_KERNEL);
1046 if (IS_ERR(pages))
1047 return PTR_ERR(pages);
1048
1049 if (!orig_ops) {
1050 payload_len = (flags & CEPH_OSD_FLAG_WRITE ? len : 0);
1051 ret = rbd_create_rw_ops(&ops, 1, opcode, payload_len);
1052 if (ret < 0)
1053 goto done;
1054
1055 if ((flags & CEPH_OSD_FLAG_WRITE) && buf) {
1056 ret = ceph_copy_to_page_vector(pages, buf, ofs, len);
1057 if (ret < 0)
1058 goto done_ops;
1059 }
1060 }
1061
1062 ret = rbd_do_request(NULL, dev, snapc, snapid,
1063 obj, ofs, len, NULL,
1064 pages, num_pages,
1065 flags,
1066 ops,
1067 2,
1068 NULL, 0,
1069 NULL,
1070 linger_req, ver);
1071 if (ret < 0)
1072 goto done_ops;
1073
1074 if ((flags & CEPH_OSD_FLAG_READ) && buf)
1075 ret = ceph_copy_from_page_vector(pages, buf, ofs, ret);
1076
1077 done_ops:
1078 if (!orig_ops)
1079 rbd_destroy_ops(ops);
1080 done:
1081 ceph_release_page_vector(pages, num_pages);
1082 return ret;
1083 }
1084
1085 /*
1086 * Do an asynchronous ceph osd operation
1087 */
1088 static int rbd_do_op(struct request *rq,
1089 struct rbd_device *rbd_dev ,
1090 struct ceph_snap_context *snapc,
1091 u64 snapid,
1092 int opcode, int flags, int num_reply,
1093 u64 ofs, u64 len,
1094 struct bio *bio,
1095 struct rbd_req_coll *coll,
1096 int coll_index)
1097 {
1098 char *seg_name;
1099 u64 seg_ofs;
1100 u64 seg_len;
1101 int ret;
1102 struct ceph_osd_req_op *ops;
1103 u32 payload_len;
1104
1105 seg_name = kmalloc(RBD_MAX_SEG_NAME_LEN + 1, GFP_NOIO);
1106 if (!seg_name)
1107 return -ENOMEM;
1108
1109 seg_len = rbd_get_segment(&rbd_dev->header,
1110 rbd_dev->header.block_name,
1111 ofs, len,
1112 seg_name, &seg_ofs);
1113
1114 payload_len = (flags & CEPH_OSD_FLAG_WRITE ? seg_len : 0);
1115
1116 ret = rbd_create_rw_ops(&ops, 1, opcode, payload_len);
1117 if (ret < 0)
1118 goto done;
1119
1120 /* we've taken care of segment sizes earlier when we
1121 cloned the bios. We should never have a segment
1122 truncated at this point */
1123 BUG_ON(seg_len < len);
1124
1125 ret = rbd_do_request(rq, rbd_dev, snapc, snapid,
1126 seg_name, seg_ofs, seg_len,
1127 bio,
1128 NULL, 0,
1129 flags,
1130 ops,
1131 num_reply,
1132 coll, coll_index,
1133 rbd_req_cb, 0, NULL);
1134
1135 rbd_destroy_ops(ops);
1136 done:
1137 kfree(seg_name);
1138 return ret;
1139 }
1140
1141 /*
1142 * Request async osd write
1143 */
1144 static int rbd_req_write(struct request *rq,
1145 struct rbd_device *rbd_dev,
1146 struct ceph_snap_context *snapc,
1147 u64 ofs, u64 len,
1148 struct bio *bio,
1149 struct rbd_req_coll *coll,
1150 int coll_index)
1151 {
1152 return rbd_do_op(rq, rbd_dev, snapc, CEPH_NOSNAP,
1153 CEPH_OSD_OP_WRITE,
1154 CEPH_OSD_FLAG_WRITE | CEPH_OSD_FLAG_ONDISK,
1155 2,
1156 ofs, len, bio, coll, coll_index);
1157 }
1158
1159 /*
1160 * Request async osd read
1161 */
1162 static int rbd_req_read(struct request *rq,
1163 struct rbd_device *rbd_dev,
1164 u64 snapid,
1165 u64 ofs, u64 len,
1166 struct bio *bio,
1167 struct rbd_req_coll *coll,
1168 int coll_index)
1169 {
1170 return rbd_do_op(rq, rbd_dev, NULL,
1171 (snapid ? snapid : CEPH_NOSNAP),
1172 CEPH_OSD_OP_READ,
1173 CEPH_OSD_FLAG_READ,
1174 2,
1175 ofs, len, bio, coll, coll_index);
1176 }
1177
1178 /*
1179 * Request sync osd read
1180 */
1181 static int rbd_req_sync_read(struct rbd_device *dev,
1182 struct ceph_snap_context *snapc,
1183 u64 snapid,
1184 const char *obj,
1185 u64 ofs, u64 len,
1186 char *buf,
1187 u64 *ver)
1188 {
1189 return rbd_req_sync_op(dev, NULL,
1190 (snapid ? snapid : CEPH_NOSNAP),
1191 CEPH_OSD_OP_READ,
1192 CEPH_OSD_FLAG_READ,
1193 NULL,
1194 1, obj, ofs, len, buf, NULL, ver);
1195 }
1196
1197 /*
1198 * Request sync osd watch
1199 */
1200 static int rbd_req_sync_notify_ack(struct rbd_device *dev,
1201 u64 ver,
1202 u64 notify_id,
1203 const char *obj)
1204 {
1205 struct ceph_osd_req_op *ops;
1206 struct page **pages = NULL;
1207 int ret;
1208
1209 ret = rbd_create_rw_ops(&ops, 1, CEPH_OSD_OP_NOTIFY_ACK, 0);
1210 if (ret < 0)
1211 return ret;
1212
1213 ops[0].watch.ver = cpu_to_le64(dev->header.obj_version);
1214 ops[0].watch.cookie = notify_id;
1215 ops[0].watch.flag = 0;
1216
1217 ret = rbd_do_request(NULL, dev, NULL, CEPH_NOSNAP,
1218 obj, 0, 0, NULL,
1219 pages, 0,
1220 CEPH_OSD_FLAG_READ,
1221 ops,
1222 1,
1223 NULL, 0,
1224 rbd_simple_req_cb, 0, NULL);
1225
1226 rbd_destroy_ops(ops);
1227 return ret;
1228 }
1229
1230 static void rbd_watch_cb(u64 ver, u64 notify_id, u8 opcode, void *data)
1231 {
1232 struct rbd_device *dev = (struct rbd_device *)data;
1233 int rc;
1234
1235 if (!dev)
1236 return;
1237
1238 dout("rbd_watch_cb %s notify_id=%lld opcode=%d\n", dev->obj_md_name,
1239 notify_id, (int)opcode);
1240 mutex_lock_nested(&ctl_mutex, SINGLE_DEPTH_NESTING);
1241 rc = __rbd_update_snaps(dev);
1242 mutex_unlock(&ctl_mutex);
1243 if (rc)
1244 pr_warning(RBD_DRV_NAME "%d got notification but failed to "
1245 " update snaps: %d\n", dev->major, rc);
1246
1247 rbd_req_sync_notify_ack(dev, ver, notify_id, dev->obj_md_name);
1248 }
1249
1250 /*
1251 * Request sync osd watch
1252 */
1253 static int rbd_req_sync_watch(struct rbd_device *dev,
1254 const char *obj,
1255 u64 ver)
1256 {
1257 struct ceph_osd_req_op *ops;
1258 struct ceph_osd_client *osdc = &dev->rbd_client->client->osdc;
1259
1260 int ret = rbd_create_rw_ops(&ops, 1, CEPH_OSD_OP_WATCH, 0);
1261 if (ret < 0)
1262 return ret;
1263
1264 ret = ceph_osdc_create_event(osdc, rbd_watch_cb, 0,
1265 (void *)dev, &dev->watch_event);
1266 if (ret < 0)
1267 goto fail;
1268
1269 ops[0].watch.ver = cpu_to_le64(ver);
1270 ops[0].watch.cookie = cpu_to_le64(dev->watch_event->cookie);
1271 ops[0].watch.flag = 1;
1272
1273 ret = rbd_req_sync_op(dev, NULL,
1274 CEPH_NOSNAP,
1275 0,
1276 CEPH_OSD_FLAG_WRITE | CEPH_OSD_FLAG_ONDISK,
1277 ops,
1278 1, obj, 0, 0, NULL,
1279 &dev->watch_request, NULL);
1280
1281 if (ret < 0)
1282 goto fail_event;
1283
1284 rbd_destroy_ops(ops);
1285 return 0;
1286
1287 fail_event:
1288 ceph_osdc_cancel_event(dev->watch_event);
1289 dev->watch_event = NULL;
1290 fail:
1291 rbd_destroy_ops(ops);
1292 return ret;
1293 }
1294
1295 /*
1296 * Request sync osd unwatch
1297 */
1298 static int rbd_req_sync_unwatch(struct rbd_device *dev,
1299 const char *obj)
1300 {
1301 struct ceph_osd_req_op *ops;
1302
1303 int ret = rbd_create_rw_ops(&ops, 1, CEPH_OSD_OP_WATCH, 0);
1304 if (ret < 0)
1305 return ret;
1306
1307 ops[0].watch.ver = 0;
1308 ops[0].watch.cookie = cpu_to_le64(dev->watch_event->cookie);
1309 ops[0].watch.flag = 0;
1310
1311 ret = rbd_req_sync_op(dev, NULL,
1312 CEPH_NOSNAP,
1313 0,
1314 CEPH_OSD_FLAG_WRITE | CEPH_OSD_FLAG_ONDISK,
1315 ops,
1316 1, obj, 0, 0, NULL, NULL, NULL);
1317
1318 rbd_destroy_ops(ops);
1319 ceph_osdc_cancel_event(dev->watch_event);
1320 dev->watch_event = NULL;
1321 return ret;
1322 }
1323
1324 struct rbd_notify_info {
1325 struct rbd_device *dev;
1326 };
1327
1328 static void rbd_notify_cb(u64 ver, u64 notify_id, u8 opcode, void *data)
1329 {
1330 struct rbd_device *dev = (struct rbd_device *)data;
1331 if (!dev)
1332 return;
1333
1334 dout("rbd_notify_cb %s notify_id=%lld opcode=%d\n", dev->obj_md_name,
1335 notify_id, (int)opcode);
1336 }
1337
1338 /*
1339 * Request sync osd notify
1340 */
1341 static int rbd_req_sync_notify(struct rbd_device *dev,
1342 const char *obj)
1343 {
1344 struct ceph_osd_req_op *ops;
1345 struct ceph_osd_client *osdc = &dev->rbd_client->client->osdc;
1346 struct ceph_osd_event *event;
1347 struct rbd_notify_info info;
1348 int payload_len = sizeof(u32) + sizeof(u32);
1349 int ret;
1350
1351 ret = rbd_create_rw_ops(&ops, 1, CEPH_OSD_OP_NOTIFY, payload_len);
1352 if (ret < 0)
1353 return ret;
1354
1355 info.dev = dev;
1356
1357 ret = ceph_osdc_create_event(osdc, rbd_notify_cb, 1,
1358 (void *)&info, &event);
1359 if (ret < 0)
1360 goto fail;
1361
1362 ops[0].watch.ver = 1;
1363 ops[0].watch.flag = 1;
1364 ops[0].watch.cookie = event->cookie;
1365 ops[0].watch.prot_ver = RADOS_NOTIFY_VER;
1366 ops[0].watch.timeout = 12;
1367
1368 ret = rbd_req_sync_op(dev, NULL,
1369 CEPH_NOSNAP,
1370 0,
1371 CEPH_OSD_FLAG_WRITE | CEPH_OSD_FLAG_ONDISK,
1372 ops,
1373 1, obj, 0, 0, NULL, NULL, NULL);
1374 if (ret < 0)
1375 goto fail_event;
1376
1377 ret = ceph_osdc_wait_event(event, CEPH_OSD_TIMEOUT_DEFAULT);
1378 dout("ceph_osdc_wait_event returned %d\n", ret);
1379 rbd_destroy_ops(ops);
1380 return 0;
1381
1382 fail_event:
1383 ceph_osdc_cancel_event(event);
1384 fail:
1385 rbd_destroy_ops(ops);
1386 return ret;
1387 }
1388
1389 /*
1390 * Request sync osd read
1391 */
1392 static int rbd_req_sync_exec(struct rbd_device *dev,
1393 const char *obj,
1394 const char *cls,
1395 const char *method,
1396 const char *data,
1397 int len,
1398 u64 *ver)
1399 {
1400 struct ceph_osd_req_op *ops;
1401 int cls_len = strlen(cls);
1402 int method_len = strlen(method);
1403 int ret = rbd_create_rw_ops(&ops, 1, CEPH_OSD_OP_CALL,
1404 cls_len + method_len + len);
1405 if (ret < 0)
1406 return ret;
1407
1408 ops[0].cls.class_name = cls;
1409 ops[0].cls.class_len = (__u8)cls_len;
1410 ops[0].cls.method_name = method;
1411 ops[0].cls.method_len = (__u8)method_len;
1412 ops[0].cls.argc = 0;
1413 ops[0].cls.indata = data;
1414 ops[0].cls.indata_len = len;
1415
1416 ret = rbd_req_sync_op(dev, NULL,
1417 CEPH_NOSNAP,
1418 0,
1419 CEPH_OSD_FLAG_WRITE | CEPH_OSD_FLAG_ONDISK,
1420 ops,
1421 1, obj, 0, 0, NULL, NULL, ver);
1422
1423 rbd_destroy_ops(ops);
1424
1425 dout("cls_exec returned %d\n", ret);
1426 return ret;
1427 }
1428
1429 static struct rbd_req_coll *rbd_alloc_coll(int num_reqs)
1430 {
1431 struct rbd_req_coll *coll =
1432 kzalloc(sizeof(struct rbd_req_coll) +
1433 sizeof(struct rbd_req_status) * num_reqs,
1434 GFP_ATOMIC);
1435
1436 if (!coll)
1437 return NULL;
1438 coll->total = num_reqs;
1439 kref_init(&coll->kref);
1440 return coll;
1441 }
1442
1443 /*
1444 * block device queue callback
1445 */
1446 static void rbd_rq_fn(struct request_queue *q)
1447 {
1448 struct rbd_device *rbd_dev = q->queuedata;
1449 struct request *rq;
1450 struct bio_pair *bp = NULL;
1451
1452 while ((rq = blk_fetch_request(q))) {
1453 struct bio *bio;
1454 struct bio *rq_bio, *next_bio = NULL;
1455 bool do_write;
1456 int size, op_size = 0;
1457 u64 ofs;
1458 int num_segs, cur_seg = 0;
1459 struct rbd_req_coll *coll;
1460
1461 /* peek at request from block layer */
1462 if (!rq)
1463 break;
1464
1465 dout("fetched request\n");
1466
1467 /* filter out block requests we don't understand */
1468 if ((rq->cmd_type != REQ_TYPE_FS)) {
1469 __blk_end_request_all(rq, 0);
1470 continue;
1471 }
1472
1473 /* deduce our operation (read, write) */
1474 do_write = (rq_data_dir(rq) == WRITE);
1475
1476 size = blk_rq_bytes(rq);
1477 ofs = blk_rq_pos(rq) * SECTOR_SIZE;
1478 rq_bio = rq->bio;
1479 if (do_write && rbd_dev->read_only) {
1480 __blk_end_request_all(rq, -EROFS);
1481 continue;
1482 }
1483
1484 spin_unlock_irq(q->queue_lock);
1485
1486 dout("%s 0x%x bytes at 0x%llx\n",
1487 do_write ? "write" : "read",
1488 size, blk_rq_pos(rq) * SECTOR_SIZE);
1489
1490 num_segs = rbd_get_num_segments(&rbd_dev->header, ofs, size);
1491 coll = rbd_alloc_coll(num_segs);
1492 if (!coll) {
1493 spin_lock_irq(q->queue_lock);
1494 __blk_end_request_all(rq, -ENOMEM);
1495 continue;
1496 }
1497
1498 do {
1499 /* a bio clone to be passed down to OSD req */
1500 dout("rq->bio->bi_vcnt=%d\n", rq->bio->bi_vcnt);
1501 op_size = rbd_get_segment(&rbd_dev->header,
1502 rbd_dev->header.block_name,
1503 ofs, size,
1504 NULL, NULL);
1505 kref_get(&coll->kref);
1506 bio = bio_chain_clone(&rq_bio, &next_bio, &bp,
1507 op_size, GFP_ATOMIC);
1508 if (!bio) {
1509 rbd_coll_end_req_index(rq, coll, cur_seg,
1510 -ENOMEM, op_size);
1511 goto next_seg;
1512 }
1513
1514
1515 /* init OSD command: write or read */
1516 if (do_write)
1517 rbd_req_write(rq, rbd_dev,
1518 rbd_dev->header.snapc,
1519 ofs,
1520 op_size, bio,
1521 coll, cur_seg);
1522 else
1523 rbd_req_read(rq, rbd_dev,
1524 cur_snap_id(rbd_dev),
1525 ofs,
1526 op_size, bio,
1527 coll, cur_seg);
1528
1529 next_seg:
1530 size -= op_size;
1531 ofs += op_size;
1532
1533 cur_seg++;
1534 rq_bio = next_bio;
1535 } while (size > 0);
1536 kref_put(&coll->kref, rbd_coll_release);
1537
1538 if (bp)
1539 bio_pair_release(bp);
1540 spin_lock_irq(q->queue_lock);
1541 }
1542 }
1543
1544 /*
1545 * a queue callback. Makes sure that we don't create a bio that spans across
1546 * multiple osd objects. One exception would be with a single page bios,
1547 * which we handle later at bio_chain_clone
1548 */
1549 static int rbd_merge_bvec(struct request_queue *q, struct bvec_merge_data *bmd,
1550 struct bio_vec *bvec)
1551 {
1552 struct rbd_device *rbd_dev = q->queuedata;
1553 unsigned int chunk_sectors;
1554 sector_t sector;
1555 unsigned int bio_sectors;
1556 int max;
1557
1558 chunk_sectors = 1 << (rbd_dev->header.obj_order - SECTOR_SHIFT);
1559 sector = bmd->bi_sector + get_start_sect(bmd->bi_bdev);
1560 bio_sectors = bmd->bi_size >> SECTOR_SHIFT;
1561
1562 max = (chunk_sectors - ((sector & (chunk_sectors - 1))
1563 + bio_sectors)) << SECTOR_SHIFT;
1564 if (max < 0)
1565 max = 0; /* bio_add cannot handle a negative return */
1566 if (max <= bvec->bv_len && bio_sectors == 0)
1567 return bvec->bv_len;
1568 return max;
1569 }
1570
1571 static void rbd_free_disk(struct rbd_device *rbd_dev)
1572 {
1573 struct gendisk *disk = rbd_dev->disk;
1574
1575 if (!disk)
1576 return;
1577
1578 rbd_header_free(&rbd_dev->header);
1579
1580 if (disk->flags & GENHD_FL_UP)
1581 del_gendisk(disk);
1582 if (disk->queue)
1583 blk_cleanup_queue(disk->queue);
1584 put_disk(disk);
1585 }
1586
1587 /*
1588 * reload the ondisk the header
1589 */
1590 static int rbd_read_header(struct rbd_device *rbd_dev,
1591 struct rbd_image_header *header)
1592 {
1593 ssize_t rc;
1594 struct rbd_image_header_ondisk *dh;
1595 int snap_count = 0;
1596 u64 ver;
1597 size_t len;
1598
1599 /*
1600 * First reads the fixed-size header to determine the number
1601 * of snapshots, then re-reads it, along with all snapshot
1602 * records as well as their stored names.
1603 */
1604 len = sizeof (*dh);
1605 while (1) {
1606 dh = kmalloc(len, GFP_KERNEL);
1607 if (!dh)
1608 return -ENOMEM;
1609
1610 rc = rbd_req_sync_read(rbd_dev,
1611 NULL, CEPH_NOSNAP,
1612 rbd_dev->obj_md_name,
1613 0, len,
1614 (char *)dh, &ver);
1615 if (rc < 0)
1616 goto out_dh;
1617
1618 rc = rbd_header_from_disk(header, dh, snap_count, GFP_KERNEL);
1619 if (rc < 0) {
1620 if (rc == -ENXIO)
1621 pr_warning("unrecognized header format"
1622 " for image %s", rbd_dev->obj);
1623 goto out_dh;
1624 }
1625
1626 if (snap_count == header->total_snaps)
1627 break;
1628
1629 snap_count = header->total_snaps;
1630 len = sizeof (*dh) +
1631 snap_count * sizeof(struct rbd_image_snap_ondisk) +
1632 header->snap_names_len;
1633
1634 rbd_header_free(header);
1635 kfree(dh);
1636 }
1637 header->obj_version = ver;
1638
1639 out_dh:
1640 kfree(dh);
1641 return rc;
1642 }
1643
1644 /*
1645 * create a snapshot
1646 */
1647 static int rbd_header_add_snap(struct rbd_device *dev,
1648 const char *snap_name,
1649 gfp_t gfp_flags)
1650 {
1651 int name_len = strlen(snap_name);
1652 u64 new_snapid;
1653 int ret;
1654 void *data, *p, *e;
1655 u64 ver;
1656 struct ceph_mon_client *monc;
1657
1658 /* we should create a snapshot only if we're pointing at the head */
1659 if (dev->cur_snap)
1660 return -EINVAL;
1661
1662 monc = &dev->rbd_client->client->monc;
1663 ret = ceph_monc_create_snapid(monc, dev->poolid, &new_snapid);
1664 dout("created snapid=%lld\n", new_snapid);
1665 if (ret < 0)
1666 return ret;
1667
1668 data = kmalloc(name_len + 16, gfp_flags);
1669 if (!data)
1670 return -ENOMEM;
1671
1672 p = data;
1673 e = data + name_len + 16;
1674
1675 ceph_encode_string_safe(&p, e, snap_name, name_len, bad);
1676 ceph_encode_64_safe(&p, e, new_snapid, bad);
1677
1678 ret = rbd_req_sync_exec(dev, dev->obj_md_name, "rbd", "snap_add",
1679 data, p - data, &ver);
1680
1681 kfree(data);
1682
1683 if (ret < 0)
1684 return ret;
1685
1686 dev->header.snapc->seq = new_snapid;
1687
1688 return 0;
1689 bad:
1690 return -ERANGE;
1691 }
1692
1693 static void __rbd_remove_all_snaps(struct rbd_device *rbd_dev)
1694 {
1695 struct rbd_snap *snap;
1696
1697 while (!list_empty(&rbd_dev->snaps)) {
1698 snap = list_first_entry(&rbd_dev->snaps, struct rbd_snap, node);
1699 __rbd_remove_snap_dev(rbd_dev, snap);
1700 }
1701 }
1702
1703 /*
1704 * only read the first part of the ondisk header, without the snaps info
1705 */
1706 static int __rbd_update_snaps(struct rbd_device *rbd_dev)
1707 {
1708 int ret;
1709 struct rbd_image_header h;
1710 u64 snap_seq;
1711 int follow_seq = 0;
1712
1713 ret = rbd_read_header(rbd_dev, &h);
1714 if (ret < 0)
1715 return ret;
1716
1717 /* resized? */
1718 set_capacity(rbd_dev->disk, h.image_size / SECTOR_SIZE);
1719
1720 down_write(&rbd_dev->header_rwsem);
1721
1722 snap_seq = rbd_dev->header.snapc->seq;
1723 if (rbd_dev->header.total_snaps &&
1724 rbd_dev->header.snapc->snaps[0] == snap_seq)
1725 /* pointing at the head, will need to follow that
1726 if head moves */
1727 follow_seq = 1;
1728
1729 kfree(rbd_dev->header.snapc);
1730 kfree(rbd_dev->header.snap_names);
1731 kfree(rbd_dev->header.snap_sizes);
1732
1733 rbd_dev->header.total_snaps = h.total_snaps;
1734 rbd_dev->header.snapc = h.snapc;
1735 rbd_dev->header.snap_names = h.snap_names;
1736 rbd_dev->header.snap_names_len = h.snap_names_len;
1737 rbd_dev->header.snap_sizes = h.snap_sizes;
1738 if (follow_seq)
1739 rbd_dev->header.snapc->seq = rbd_dev->header.snapc->snaps[0];
1740 else
1741 rbd_dev->header.snapc->seq = snap_seq;
1742
1743 ret = __rbd_init_snaps_header(rbd_dev);
1744
1745 up_write(&rbd_dev->header_rwsem);
1746
1747 return ret;
1748 }
1749
1750 static int rbd_init_disk(struct rbd_device *rbd_dev)
1751 {
1752 struct gendisk *disk;
1753 struct request_queue *q;
1754 int rc;
1755 u64 segment_size;
1756 u64 total_size = 0;
1757
1758 /* contact OSD, request size info about the object being mapped */
1759 rc = rbd_read_header(rbd_dev, &rbd_dev->header);
1760 if (rc)
1761 return rc;
1762
1763 /* no need to lock here, as rbd_dev is not registered yet */
1764 rc = __rbd_init_snaps_header(rbd_dev);
1765 if (rc)
1766 return rc;
1767
1768 rc = rbd_header_set_snap(rbd_dev, &total_size);
1769 if (rc)
1770 return rc;
1771
1772 /* create gendisk info */
1773 rc = -ENOMEM;
1774 disk = alloc_disk(RBD_MINORS_PER_MAJOR);
1775 if (!disk)
1776 goto out;
1777
1778 snprintf(disk->disk_name, sizeof(disk->disk_name), RBD_DRV_NAME "%d",
1779 rbd_dev->id);
1780 disk->major = rbd_dev->major;
1781 disk->first_minor = 0;
1782 disk->fops = &rbd_bd_ops;
1783 disk->private_data = rbd_dev;
1784
1785 /* init rq */
1786 rc = -ENOMEM;
1787 q = blk_init_queue(rbd_rq_fn, &rbd_dev->lock);
1788 if (!q)
1789 goto out_disk;
1790
1791 /* We use the default size, but let's be explicit about it. */
1792 blk_queue_physical_block_size(q, SECTOR_SIZE);
1793
1794 /* set io sizes to object size */
1795 segment_size = rbd_obj_bytes(&rbd_dev->header);
1796 blk_queue_max_hw_sectors(q, segment_size / SECTOR_SIZE);
1797 blk_queue_max_segment_size(q, segment_size);
1798 blk_queue_io_min(q, segment_size);
1799 blk_queue_io_opt(q, segment_size);
1800
1801 blk_queue_merge_bvec(q, rbd_merge_bvec);
1802 disk->queue = q;
1803
1804 q->queuedata = rbd_dev;
1805
1806 rbd_dev->disk = disk;
1807 rbd_dev->q = q;
1808
1809 /* finally, announce the disk to the world */
1810 set_capacity(disk, total_size / SECTOR_SIZE);
1811 add_disk(disk);
1812
1813 pr_info("%s: added with size 0x%llx\n",
1814 disk->disk_name, (unsigned long long)total_size);
1815 return 0;
1816
1817 out_disk:
1818 put_disk(disk);
1819 out:
1820 return rc;
1821 }
1822
1823 /*
1824 sysfs
1825 */
1826
1827 static struct rbd_device *dev_to_rbd_dev(struct device *dev)
1828 {
1829 return container_of(dev, struct rbd_device, dev);
1830 }
1831
1832 static ssize_t rbd_size_show(struct device *dev,
1833 struct device_attribute *attr, char *buf)
1834 {
1835 struct rbd_device *rbd_dev = dev_to_rbd_dev(dev);
1836
1837 return sprintf(buf, "%llu\n", (unsigned long long)rbd_dev->header.image_size);
1838 }
1839
1840 static ssize_t rbd_major_show(struct device *dev,
1841 struct device_attribute *attr, char *buf)
1842 {
1843 struct rbd_device *rbd_dev = dev_to_rbd_dev(dev);
1844
1845 return sprintf(buf, "%d\n", rbd_dev->major);
1846 }
1847
1848 static ssize_t rbd_client_id_show(struct device *dev,
1849 struct device_attribute *attr, char *buf)
1850 {
1851 struct rbd_device *rbd_dev = dev_to_rbd_dev(dev);
1852
1853 return sprintf(buf, "client%lld\n",
1854 ceph_client_id(rbd_dev->rbd_client->client));
1855 }
1856
1857 static ssize_t rbd_pool_show(struct device *dev,
1858 struct device_attribute *attr, char *buf)
1859 {
1860 struct rbd_device *rbd_dev = dev_to_rbd_dev(dev);
1861
1862 return sprintf(buf, "%s\n", rbd_dev->pool_name);
1863 }
1864
1865 static ssize_t rbd_name_show(struct device *dev,
1866 struct device_attribute *attr, char *buf)
1867 {
1868 struct rbd_device *rbd_dev = dev_to_rbd_dev(dev);
1869
1870 return sprintf(buf, "%s\n", rbd_dev->obj);
1871 }
1872
1873 static ssize_t rbd_snap_show(struct device *dev,
1874 struct device_attribute *attr,
1875 char *buf)
1876 {
1877 struct rbd_device *rbd_dev = dev_to_rbd_dev(dev);
1878
1879 return sprintf(buf, "%s\n", rbd_dev->snap_name);
1880 }
1881
1882 static ssize_t rbd_image_refresh(struct device *dev,
1883 struct device_attribute *attr,
1884 const char *buf,
1885 size_t size)
1886 {
1887 struct rbd_device *rbd_dev = dev_to_rbd_dev(dev);
1888 int rc;
1889 int ret = size;
1890
1891 mutex_lock_nested(&ctl_mutex, SINGLE_DEPTH_NESTING);
1892
1893 rc = __rbd_update_snaps(rbd_dev);
1894 if (rc < 0)
1895 ret = rc;
1896
1897 mutex_unlock(&ctl_mutex);
1898 return ret;
1899 }
1900
1901 static DEVICE_ATTR(size, S_IRUGO, rbd_size_show, NULL);
1902 static DEVICE_ATTR(major, S_IRUGO, rbd_major_show, NULL);
1903 static DEVICE_ATTR(client_id, S_IRUGO, rbd_client_id_show, NULL);
1904 static DEVICE_ATTR(pool, S_IRUGO, rbd_pool_show, NULL);
1905 static DEVICE_ATTR(name, S_IRUGO, rbd_name_show, NULL);
1906 static DEVICE_ATTR(refresh, S_IWUSR, NULL, rbd_image_refresh);
1907 static DEVICE_ATTR(current_snap, S_IRUGO, rbd_snap_show, NULL);
1908 static DEVICE_ATTR(create_snap, S_IWUSR, NULL, rbd_snap_add);
1909
1910 static struct attribute *rbd_attrs[] = {
1911 &dev_attr_size.attr,
1912 &dev_attr_major.attr,
1913 &dev_attr_client_id.attr,
1914 &dev_attr_pool.attr,
1915 &dev_attr_name.attr,
1916 &dev_attr_current_snap.attr,
1917 &dev_attr_refresh.attr,
1918 &dev_attr_create_snap.attr,
1919 NULL
1920 };
1921
1922 static struct attribute_group rbd_attr_group = {
1923 .attrs = rbd_attrs,
1924 };
1925
1926 static const struct attribute_group *rbd_attr_groups[] = {
1927 &rbd_attr_group,
1928 NULL
1929 };
1930
1931 static void rbd_sysfs_dev_release(struct device *dev)
1932 {
1933 }
1934
1935 static struct device_type rbd_device_type = {
1936 .name = "rbd",
1937 .groups = rbd_attr_groups,
1938 .release = rbd_sysfs_dev_release,
1939 };
1940
1941
1942 /*
1943 sysfs - snapshots
1944 */
1945
1946 static ssize_t rbd_snap_size_show(struct device *dev,
1947 struct device_attribute *attr,
1948 char *buf)
1949 {
1950 struct rbd_snap *snap = container_of(dev, struct rbd_snap, dev);
1951
1952 return sprintf(buf, "%zd\n", snap->size);
1953 }
1954
1955 static ssize_t rbd_snap_id_show(struct device *dev,
1956 struct device_attribute *attr,
1957 char *buf)
1958 {
1959 struct rbd_snap *snap = container_of(dev, struct rbd_snap, dev);
1960
1961 return sprintf(buf, "%llu\n", (unsigned long long) snap->id);
1962 }
1963
1964 static DEVICE_ATTR(snap_size, S_IRUGO, rbd_snap_size_show, NULL);
1965 static DEVICE_ATTR(snap_id, S_IRUGO, rbd_snap_id_show, NULL);
1966
1967 static struct attribute *rbd_snap_attrs[] = {
1968 &dev_attr_snap_size.attr,
1969 &dev_attr_snap_id.attr,
1970 NULL,
1971 };
1972
1973 static struct attribute_group rbd_snap_attr_group = {
1974 .attrs = rbd_snap_attrs,
1975 };
1976
1977 static void rbd_snap_dev_release(struct device *dev)
1978 {
1979 struct rbd_snap *snap = container_of(dev, struct rbd_snap, dev);
1980 kfree(snap->name);
1981 kfree(snap);
1982 }
1983
1984 static const struct attribute_group *rbd_snap_attr_groups[] = {
1985 &rbd_snap_attr_group,
1986 NULL
1987 };
1988
1989 static struct device_type rbd_snap_device_type = {
1990 .groups = rbd_snap_attr_groups,
1991 .release = rbd_snap_dev_release,
1992 };
1993
1994 static void __rbd_remove_snap_dev(struct rbd_device *rbd_dev,
1995 struct rbd_snap *snap)
1996 {
1997 list_del(&snap->node);
1998 device_unregister(&snap->dev);
1999 }
2000
2001 static int rbd_register_snap_dev(struct rbd_device *rbd_dev,
2002 struct rbd_snap *snap,
2003 struct device *parent)
2004 {
2005 struct device *dev = &snap->dev;
2006 int ret;
2007
2008 dev->type = &rbd_snap_device_type;
2009 dev->parent = parent;
2010 dev->release = rbd_snap_dev_release;
2011 dev_set_name(dev, "snap_%s", snap->name);
2012 ret = device_register(dev);
2013
2014 return ret;
2015 }
2016
2017 static int __rbd_add_snap_dev(struct rbd_device *rbd_dev,
2018 int i, const char *name,
2019 struct rbd_snap **snapp)
2020 {
2021 int ret;
2022 struct rbd_snap *snap = kzalloc(sizeof(*snap), GFP_KERNEL);
2023 if (!snap)
2024 return -ENOMEM;
2025 snap->name = kstrdup(name, GFP_KERNEL);
2026 snap->size = rbd_dev->header.snap_sizes[i];
2027 snap->id = rbd_dev->header.snapc->snaps[i];
2028 if (device_is_registered(&rbd_dev->dev)) {
2029 ret = rbd_register_snap_dev(rbd_dev, snap,
2030 &rbd_dev->dev);
2031 if (ret < 0)
2032 goto err;
2033 }
2034 *snapp = snap;
2035 return 0;
2036 err:
2037 kfree(snap->name);
2038 kfree(snap);
2039 return ret;
2040 }
2041
2042 /*
2043 * search for the previous snap in a null delimited string list
2044 */
2045 const char *rbd_prev_snap_name(const char *name, const char *start)
2046 {
2047 if (name < start + 2)
2048 return NULL;
2049
2050 name -= 2;
2051 while (*name) {
2052 if (name == start)
2053 return start;
2054 name--;
2055 }
2056 return name + 1;
2057 }
2058
2059 /*
2060 * compare the old list of snapshots that we have to what's in the header
2061 * and update it accordingly. Note that the header holds the snapshots
2062 * in a reverse order (from newest to oldest) and we need to go from
2063 * older to new so that we don't get a duplicate snap name when
2064 * doing the process (e.g., removed snapshot and recreated a new
2065 * one with the same name.
2066 */
2067 static int __rbd_init_snaps_header(struct rbd_device *rbd_dev)
2068 {
2069 const char *name, *first_name;
2070 int i = rbd_dev->header.total_snaps;
2071 struct rbd_snap *snap, *old_snap = NULL;
2072 int ret;
2073 struct list_head *p, *n;
2074
2075 first_name = rbd_dev->header.snap_names;
2076 name = first_name + rbd_dev->header.snap_names_len;
2077
2078 list_for_each_prev_safe(p, n, &rbd_dev->snaps) {
2079 u64 cur_id;
2080
2081 old_snap = list_entry(p, struct rbd_snap, node);
2082
2083 if (i)
2084 cur_id = rbd_dev->header.snapc->snaps[i - 1];
2085
2086 if (!i || old_snap->id < cur_id) {
2087 /* old_snap->id was skipped, thus was removed */
2088 __rbd_remove_snap_dev(rbd_dev, old_snap);
2089 continue;
2090 }
2091 if (old_snap->id == cur_id) {
2092 /* we have this snapshot already */
2093 i--;
2094 name = rbd_prev_snap_name(name, first_name);
2095 continue;
2096 }
2097 for (; i > 0;
2098 i--, name = rbd_prev_snap_name(name, first_name)) {
2099 if (!name) {
2100 WARN_ON(1);
2101 return -EINVAL;
2102 }
2103 cur_id = rbd_dev->header.snapc->snaps[i];
2104 /* snapshot removal? handle it above */
2105 if (cur_id >= old_snap->id)
2106 break;
2107 /* a new snapshot */
2108 ret = __rbd_add_snap_dev(rbd_dev, i - 1, name, &snap);
2109 if (ret < 0)
2110 return ret;
2111
2112 /* note that we add it backward so using n and not p */
2113 list_add(&snap->node, n);
2114 p = &snap->node;
2115 }
2116 }
2117 /* we're done going over the old snap list, just add what's left */
2118 for (; i > 0; i--) {
2119 name = rbd_prev_snap_name(name, first_name);
2120 if (!name) {
2121 WARN_ON(1);
2122 return -EINVAL;
2123 }
2124 ret = __rbd_add_snap_dev(rbd_dev, i - 1, name, &snap);
2125 if (ret < 0)
2126 return ret;
2127 list_add(&snap->node, &rbd_dev->snaps);
2128 }
2129
2130 return 0;
2131 }
2132
2133 static int rbd_bus_add_dev(struct rbd_device *rbd_dev)
2134 {
2135 int ret;
2136 struct device *dev;
2137 struct rbd_snap *snap;
2138
2139 mutex_lock_nested(&ctl_mutex, SINGLE_DEPTH_NESTING);
2140 dev = &rbd_dev->dev;
2141
2142 dev->bus = &rbd_bus_type;
2143 dev->type = &rbd_device_type;
2144 dev->parent = &rbd_root_dev;
2145 dev->release = rbd_dev_release;
2146 dev_set_name(dev, "%d", rbd_dev->id);
2147 ret = device_register(dev);
2148 if (ret < 0)
2149 goto out;
2150
2151 list_for_each_entry(snap, &rbd_dev->snaps, node) {
2152 ret = rbd_register_snap_dev(rbd_dev, snap,
2153 &rbd_dev->dev);
2154 if (ret < 0)
2155 break;
2156 }
2157 out:
2158 mutex_unlock(&ctl_mutex);
2159 return ret;
2160 }
2161
2162 static void rbd_bus_del_dev(struct rbd_device *rbd_dev)
2163 {
2164 device_unregister(&rbd_dev->dev);
2165 }
2166
2167 static int rbd_init_watch_dev(struct rbd_device *rbd_dev)
2168 {
2169 int ret, rc;
2170
2171 do {
2172 ret = rbd_req_sync_watch(rbd_dev, rbd_dev->obj_md_name,
2173 rbd_dev->header.obj_version);
2174 if (ret == -ERANGE) {
2175 mutex_lock_nested(&ctl_mutex, SINGLE_DEPTH_NESTING);
2176 rc = __rbd_update_snaps(rbd_dev);
2177 mutex_unlock(&ctl_mutex);
2178 if (rc < 0)
2179 return rc;
2180 }
2181 } while (ret == -ERANGE);
2182
2183 return ret;
2184 }
2185
2186 static atomic64_t rbd_id_max = ATOMIC64_INIT(0);
2187
2188 /*
2189 * Get a unique rbd identifier for the given new rbd_dev, and add
2190 * the rbd_dev to the global list. The minimum rbd id is 1.
2191 */
2192 static void rbd_id_get(struct rbd_device *rbd_dev)
2193 {
2194 rbd_dev->id = atomic64_inc_return(&rbd_id_max);
2195
2196 spin_lock(&rbd_dev_list_lock);
2197 list_add_tail(&rbd_dev->node, &rbd_dev_list);
2198 spin_unlock(&rbd_dev_list_lock);
2199 }
2200
2201 /*
2202 * Remove an rbd_dev from the global list, and record that its
2203 * identifier is no longer in use.
2204 */
2205 static void rbd_id_put(struct rbd_device *rbd_dev)
2206 {
2207 struct list_head *tmp;
2208 int rbd_id = rbd_dev->id;
2209 int max_id;
2210
2211 BUG_ON(rbd_id < 1);
2212
2213 spin_lock(&rbd_dev_list_lock);
2214 list_del_init(&rbd_dev->node);
2215
2216 /*
2217 * If the id being "put" is not the current maximum, there
2218 * is nothing special we need to do.
2219 */
2220 if (rbd_id != atomic64_read(&rbd_id_max)) {
2221 spin_unlock(&rbd_dev_list_lock);
2222 return;
2223 }
2224
2225 /*
2226 * We need to update the current maximum id. Search the
2227 * list to find out what it is. We're more likely to find
2228 * the maximum at the end, so search the list backward.
2229 */
2230 max_id = 0;
2231 list_for_each_prev(tmp, &rbd_dev_list) {
2232 struct rbd_device *rbd_dev;
2233
2234 rbd_dev = list_entry(tmp, struct rbd_device, node);
2235 if (rbd_id > max_id)
2236 max_id = rbd_id;
2237 }
2238 spin_unlock(&rbd_dev_list_lock);
2239
2240 /*
2241 * The max id could have been updated by rbd_id_get(), in
2242 * which case it now accurately reflects the new maximum.
2243 * Be careful not to overwrite the maximum value in that
2244 * case.
2245 */
2246 atomic64_cmpxchg(&rbd_id_max, rbd_id, max_id);
2247 }
2248
2249 /*
2250 * Skips over white space at *buf, and updates *buf to point to the
2251 * first found non-space character (if any). Returns the length of
2252 * the token (string of non-white space characters) found. Note
2253 * that *buf must be terminated with '\0'.
2254 */
2255 static inline size_t next_token(const char **buf)
2256 {
2257 /*
2258 * These are the characters that produce nonzero for
2259 * isspace() in the "C" and "POSIX" locales.
2260 */
2261 const char *spaces = " \f\n\r\t\v";
2262
2263 *buf += strspn(*buf, spaces); /* Find start of token */
2264
2265 return strcspn(*buf, spaces); /* Return token length */
2266 }
2267
2268 /*
2269 * Finds the next token in *buf, and if the provided token buffer is
2270 * big enough, copies the found token into it. The result, if
2271 * copied, is guaranteed to be terminated with '\0'. Note that *buf
2272 * must be terminated with '\0' on entry.
2273 *
2274 * Returns the length of the token found (not including the '\0').
2275 * Return value will be 0 if no token is found, and it will be >=
2276 * token_size if the token would not fit.
2277 *
2278 * The *buf pointer will be updated to point beyond the end of the
2279 * found token. Note that this occurs even if the token buffer is
2280 * too small to hold it.
2281 */
2282 static inline size_t copy_token(const char **buf,
2283 char *token,
2284 size_t token_size)
2285 {
2286 size_t len;
2287
2288 len = next_token(buf);
2289 if (len < token_size) {
2290 memcpy(token, *buf, len);
2291 *(token + len) = '\0';
2292 }
2293 *buf += len;
2294
2295 return len;
2296 }
2297
2298 /*
2299 * This fills in the pool_name, obj, obj_len, snap_name, obj_len,
2300 * rbd_dev, rbd_md_name, and name fields of the given rbd_dev, based
2301 * on the list of monitor addresses and other options provided via
2302 * /sys/bus/rbd/add.
2303 */
2304 static int rbd_add_parse_args(struct rbd_device *rbd_dev,
2305 const char *buf,
2306 const char **mon_addrs,
2307 size_t *mon_addrs_size,
2308 char *options,
2309 size_t options_size)
2310 {
2311 size_t len;
2312
2313 /* The first four tokens are required */
2314
2315 len = next_token(&buf);
2316 if (!len)
2317 return -EINVAL;
2318 *mon_addrs_size = len + 1;
2319 *mon_addrs = buf;
2320
2321 buf += len;
2322
2323 len = copy_token(&buf, options, options_size);
2324 if (!len || len >= options_size)
2325 return -EINVAL;
2326
2327 len = copy_token(&buf, rbd_dev->pool_name, sizeof (rbd_dev->pool_name));
2328 if (!len || len >= sizeof (rbd_dev->pool_name))
2329 return -EINVAL;
2330
2331 len = copy_token(&buf, rbd_dev->obj, sizeof (rbd_dev->obj));
2332 if (!len || len >= sizeof (rbd_dev->obj))
2333 return -EINVAL;
2334
2335 /* We have the object length in hand, save it. */
2336
2337 rbd_dev->obj_len = len;
2338
2339 BUILD_BUG_ON(RBD_MAX_MD_NAME_LEN
2340 < RBD_MAX_OBJ_NAME_LEN + sizeof (RBD_SUFFIX));
2341 sprintf(rbd_dev->obj_md_name, "%s%s", rbd_dev->obj, RBD_SUFFIX);
2342
2343 /*
2344 * The snapshot name is optional, but it's an error if it's
2345 * too long. If no snapshot is supplied, fill in the default.
2346 */
2347 len = copy_token(&buf, rbd_dev->snap_name, sizeof (rbd_dev->snap_name));
2348 if (!len)
2349 memcpy(rbd_dev->snap_name, RBD_SNAP_HEAD_NAME,
2350 sizeof (RBD_SNAP_HEAD_NAME));
2351 else if (len >= sizeof (rbd_dev->snap_name))
2352 return -EINVAL;
2353
2354 return 0;
2355 }
2356
2357 static ssize_t rbd_add(struct bus_type *bus,
2358 const char *buf,
2359 size_t count)
2360 {
2361 struct rbd_device *rbd_dev;
2362 const char *mon_addrs = NULL;
2363 size_t mon_addrs_size = 0;
2364 char *options = NULL;
2365 struct ceph_osd_client *osdc;
2366 int rc = -ENOMEM;
2367
2368 if (!try_module_get(THIS_MODULE))
2369 return -ENODEV;
2370
2371 rbd_dev = kzalloc(sizeof(*rbd_dev), GFP_KERNEL);
2372 if (!rbd_dev)
2373 goto err_nomem;
2374 options = kmalloc(count, GFP_KERNEL);
2375 if (!options)
2376 goto err_nomem;
2377
2378 /* static rbd_device initialization */
2379 spin_lock_init(&rbd_dev->lock);
2380 INIT_LIST_HEAD(&rbd_dev->node);
2381 INIT_LIST_HEAD(&rbd_dev->snaps);
2382 init_rwsem(&rbd_dev->header_rwsem);
2383
2384 init_rwsem(&rbd_dev->header_rwsem);
2385
2386 /* generate unique id: find highest unique id, add one */
2387 rbd_id_get(rbd_dev);
2388
2389 /* Fill in the device name, now that we have its id. */
2390 BUILD_BUG_ON(DEV_NAME_LEN
2391 < sizeof (RBD_DRV_NAME) + MAX_INT_FORMAT_WIDTH);
2392 sprintf(rbd_dev->name, "%s%d", RBD_DRV_NAME, rbd_dev->id);
2393
2394 /* parse add command */
2395 rc = rbd_add_parse_args(rbd_dev, buf, &mon_addrs, &mon_addrs_size,
2396 options, count);
2397 if (rc)
2398 goto err_put_id;
2399
2400 rbd_dev->rbd_client = rbd_get_client(mon_addrs, mon_addrs_size - 1,
2401 options);
2402 if (IS_ERR(rbd_dev->rbd_client)) {
2403 rc = PTR_ERR(rbd_dev->rbd_client);
2404 goto err_put_id;
2405 }
2406
2407 /* pick the pool */
2408 osdc = &rbd_dev->rbd_client->client->osdc;
2409 rc = ceph_pg_poolid_by_name(osdc->osdmap, rbd_dev->pool_name);
2410 if (rc < 0)
2411 goto err_out_client;
2412 rbd_dev->poolid = rc;
2413
2414 /* register our block device */
2415 rc = register_blkdev(0, rbd_dev->name);
2416 if (rc < 0)
2417 goto err_out_client;
2418 rbd_dev->major = rc;
2419
2420 rc = rbd_bus_add_dev(rbd_dev);
2421 if (rc)
2422 goto err_out_blkdev;
2423
2424 /*
2425 * At this point cleanup in the event of an error is the job
2426 * of the sysfs code (initiated by rbd_bus_del_dev()).
2427 *
2428 * Set up and announce blkdev mapping.
2429 */
2430 rc = rbd_init_disk(rbd_dev);
2431 if (rc)
2432 goto err_out_bus;
2433
2434 rc = rbd_init_watch_dev(rbd_dev);
2435 if (rc)
2436 goto err_out_bus;
2437
2438 return count;
2439
2440 err_out_bus:
2441 /* this will also clean up rest of rbd_dev stuff */
2442
2443 rbd_bus_del_dev(rbd_dev);
2444 kfree(options);
2445 return rc;
2446
2447 err_out_blkdev:
2448 unregister_blkdev(rbd_dev->major, rbd_dev->name);
2449 err_out_client:
2450 rbd_put_client(rbd_dev);
2451 err_put_id:
2452 rbd_id_put(rbd_dev);
2453 err_nomem:
2454 kfree(options);
2455 kfree(rbd_dev);
2456
2457 dout("Error adding device %s\n", buf);
2458 module_put(THIS_MODULE);
2459
2460 return (ssize_t) rc;
2461 }
2462
2463 static struct rbd_device *__rbd_get_dev(unsigned long id)
2464 {
2465 struct list_head *tmp;
2466 struct rbd_device *rbd_dev;
2467
2468 spin_lock(&rbd_dev_list_lock);
2469 list_for_each(tmp, &rbd_dev_list) {
2470 rbd_dev = list_entry(tmp, struct rbd_device, node);
2471 if (rbd_dev->id == id) {
2472 spin_unlock(&rbd_dev_list_lock);
2473 return rbd_dev;
2474 }
2475 }
2476 spin_unlock(&rbd_dev_list_lock);
2477 return NULL;
2478 }
2479
2480 static void rbd_dev_release(struct device *dev)
2481 {
2482 struct rbd_device *rbd_dev = dev_to_rbd_dev(dev);
2483
2484 if (rbd_dev->watch_request) {
2485 struct ceph_client *client = rbd_dev->rbd_client->client;
2486
2487 ceph_osdc_unregister_linger_request(&client->osdc,
2488 rbd_dev->watch_request);
2489 }
2490 if (rbd_dev->watch_event)
2491 rbd_req_sync_unwatch(rbd_dev, rbd_dev->obj_md_name);
2492
2493 rbd_put_client(rbd_dev);
2494
2495 /* clean up and free blkdev */
2496 rbd_free_disk(rbd_dev);
2497 unregister_blkdev(rbd_dev->major, rbd_dev->name);
2498
2499 /* done with the id, and with the rbd_dev */
2500 rbd_id_put(rbd_dev);
2501 kfree(rbd_dev);
2502
2503 /* release module ref */
2504 module_put(THIS_MODULE);
2505 }
2506
2507 static ssize_t rbd_remove(struct bus_type *bus,
2508 const char *buf,
2509 size_t count)
2510 {
2511 struct rbd_device *rbd_dev = NULL;
2512 int target_id, rc;
2513 unsigned long ul;
2514 int ret = count;
2515
2516 rc = strict_strtoul(buf, 10, &ul);
2517 if (rc)
2518 return rc;
2519
2520 /* convert to int; abort if we lost anything in the conversion */
2521 target_id = (int) ul;
2522 if (target_id != ul)
2523 return -EINVAL;
2524
2525 mutex_lock_nested(&ctl_mutex, SINGLE_DEPTH_NESTING);
2526
2527 rbd_dev = __rbd_get_dev(target_id);
2528 if (!rbd_dev) {
2529 ret = -ENOENT;
2530 goto done;
2531 }
2532
2533 __rbd_remove_all_snaps(rbd_dev);
2534 rbd_bus_del_dev(rbd_dev);
2535
2536 done:
2537 mutex_unlock(&ctl_mutex);
2538 return ret;
2539 }
2540
2541 static ssize_t rbd_snap_add(struct device *dev,
2542 struct device_attribute *attr,
2543 const char *buf,
2544 size_t count)
2545 {
2546 struct rbd_device *rbd_dev = dev_to_rbd_dev(dev);
2547 int ret;
2548 char *name = kmalloc(count + 1, GFP_KERNEL);
2549 if (!name)
2550 return -ENOMEM;
2551
2552 snprintf(name, count, "%s", buf);
2553
2554 mutex_lock_nested(&ctl_mutex, SINGLE_DEPTH_NESTING);
2555
2556 ret = rbd_header_add_snap(rbd_dev,
2557 name, GFP_KERNEL);
2558 if (ret < 0)
2559 goto err_unlock;
2560
2561 ret = __rbd_update_snaps(rbd_dev);
2562 if (ret < 0)
2563 goto err_unlock;
2564
2565 /* shouldn't hold ctl_mutex when notifying.. notify might
2566 trigger a watch callback that would need to get that mutex */
2567 mutex_unlock(&ctl_mutex);
2568
2569 /* make a best effort, don't error if failed */
2570 rbd_req_sync_notify(rbd_dev, rbd_dev->obj_md_name);
2571
2572 ret = count;
2573 kfree(name);
2574 return ret;
2575
2576 err_unlock:
2577 mutex_unlock(&ctl_mutex);
2578 kfree(name);
2579 return ret;
2580 }
2581
2582 /*
2583 * create control files in sysfs
2584 * /sys/bus/rbd/...
2585 */
2586 static int rbd_sysfs_init(void)
2587 {
2588 int ret;
2589
2590 ret = device_register(&rbd_root_dev);
2591 if (ret < 0)
2592 return ret;
2593
2594 ret = bus_register(&rbd_bus_type);
2595 if (ret < 0)
2596 device_unregister(&rbd_root_dev);
2597
2598 return ret;
2599 }
2600
2601 static void rbd_sysfs_cleanup(void)
2602 {
2603 bus_unregister(&rbd_bus_type);
2604 device_unregister(&rbd_root_dev);
2605 }
2606
2607 int __init rbd_init(void)
2608 {
2609 int rc;
2610
2611 rc = rbd_sysfs_init();
2612 if (rc)
2613 return rc;
2614 pr_info("loaded " RBD_DRV_NAME_LONG "\n");
2615 return 0;
2616 }
2617
2618 void __exit rbd_exit(void)
2619 {
2620 rbd_sysfs_cleanup();
2621 }
2622
2623 module_init(rbd_init);
2624 module_exit(rbd_exit);
2625
2626 MODULE_AUTHOR("Sage Weil <sage@newdream.net>");
2627 MODULE_AUTHOR("Yehuda Sadeh <yehuda@hq.newdream.net>");
2628 MODULE_DESCRIPTION("rados block device");
2629
2630 /* following authorship retained from original osdblk.c */
2631 MODULE_AUTHOR("Jeff Garzik <jeff@garzik.org>");
2632
2633 MODULE_LICENSE("GPL");
This page took 0.084719 seconds and 5 git commands to generate.