ceph: use kref for ceph_buffer

Signed-off-by: Sage Weil <sage@newdream.net>

1 files changed, 13 insertions, 1 deletions

diff --git a/fs/ceph/buffer.c b/fs/ceph/buffer.c
index cf9aaccef22b..847c5da9a0db 100644
--- a/fs/ceph/buffer.c
+++ b/fs/ceph/buffer.c
@@ -9,13 +9,25 @@ struct ceph_buffer *ceph_buffer_new(gfp_t gfp)
 	b = kmalloc(sizeof(*b), gfp);
 	if (!b)
 		return NULL;
-	atomic_set(&b->nref, 1);
+	kref_init(&b->kref);
 	b->vec.iov_base = NULL;
 	b->vec.iov_len = 0;
 	b->alloc_len = 0;
 	return b;
 }
 
+void ceph_buffer_release(struct kref *kref)
+{
+	struct ceph_buffer *b = container_of(kref, struct ceph_buffer, kref);
+	if (b->vec.iov_base) {
+		if (b->is_vmalloc)
+			vfree(b->vec.iov_base);
+		else
+			kfree(b->vec.iov_base);
+	}
+	kfree(b);
+}
+
 int ceph_buffer_alloc(struct ceph_buffer *b, int len, gfp_t gfp)
 {
 	b->vec.iov_base = kmalloc(len, gfp | __GFP_NOWARN);