| 1 |
|
| 2 | #include <linux/ceph/ceph_debug.h>
|
| 3 |
|
| 4 | #include <linux/module.h>
|
| 5 | #include <linux/slab.h>
|
| 6 |
|
| 7 | #include <linux/ceph/buffer.h>
|
| 8 | #include <linux/ceph/decode.h>
|
| 9 |
|
| 10 | struct ceph_buffer *ceph_buffer_new(size_t len, gfp_t gfp)
|
| 11 | {
|
| 12 | struct ceph_buffer *b;
|
| 13 |
|
| 14 | b = kmalloc(sizeof(*b), gfp);
|
| 15 | if (!b)
|
| 16 | return NULL;
|
| 17 |
|
| 18 | b->vec.iov_base = kmalloc(len, gfp | __GFP_NOWARN);
|
| 19 | if (b->vec.iov_base) {
|
| 20 | b->is_vmalloc = false;
|
| 21 | } else {
|
| 22 | b->vec.iov_base = __vmalloc(len, gfp | __GFP_HIGHMEM, PAGE_KERNEL);
|
| 23 | if (!b->vec.iov_base) {
|
| 24 | kfree(b);
|
| 25 | return NULL;
|
| 26 | }
|
| 27 | b->is_vmalloc = true;
|
| 28 | }
|
| 29 |
|
| 30 | kref_init(&b->kref);
|
| 31 | b->alloc_len = len;
|
| 32 | b->vec.iov_len = len;
|
| 33 | dout("buffer_new %p\n", b);
|
| 34 | return b;
|
| 35 | }
|
| 36 | EXPORT_SYMBOL(ceph_buffer_new);
|
| 37 |
|
| 38 | void ceph_buffer_release(struct kref *kref)
|
| 39 | {
|
| 40 | struct ceph_buffer *b = container_of(kref, struct ceph_buffer, kref);
|
| 41 |
|
| 42 | dout("buffer_release %p\n", b);
|
| 43 | if (b->vec.iov_base) {
|
| 44 | if (b->is_vmalloc)
|
| 45 | vfree(b->vec.iov_base);
|
| 46 | else
|
| 47 | kfree(b->vec.iov_base);
|
| 48 | }
|
| 49 | kfree(b);
|
| 50 | }
|
| 51 | EXPORT_SYMBOL(ceph_buffer_release);
|
| 52 |
|
| 53 | int ceph_decode_buffer(struct ceph_buffer **b, void **p, void *end)
|
| 54 | {
|
| 55 | size_t len;
|
| 56 |
|
| 57 | ceph_decode_need(p, end, sizeof(u32), bad);
|
| 58 | len = ceph_decode_32(p);
|
| 59 | dout("decode_buffer len %d\n", (int)len);
|
| 60 | ceph_decode_need(p, end, len, bad);
|
| 61 | *b = ceph_buffer_new(len, GFP_NOFS);
|
| 62 | if (!*b)
|
| 63 | return -ENOMEM;
|
| 64 | ceph_decode_copy(p, (*b)->vec.iov_base, len);
|
| 65 | return 0;
|
| 66 | bad:
|
| 67 | return -EINVAL;
|
| 68 | }
|