1 #include "ceph_debug.h"
4 #include <linux/sched.h>
5 #include <linux/types.h>
6 #include <linux/vmalloc.h>
10 static void *alloc_fn(gfp_t gfp_mask, void *arg)
12 struct ceph_msgpool *pool = arg;
14 return ceph_msg_new(0, pool->front_len);
17 static void free_fn(void *element, void *arg)
19 ceph_msg_put(element);
22 int ceph_msgpool_init(struct ceph_msgpool *pool,
23 int front_len, int size, bool blocking)
25 pool->front_len = front_len;
26 pool->pool = mempool_create(size, alloc_fn, free_fn, pool);
32 void ceph_msgpool_destroy(struct ceph_msgpool *pool)
34 mempool_destroy(pool->pool);
37 struct ceph_msg *ceph_msgpool_get(struct ceph_msgpool *pool,
40 if (front_len > pool->front_len) {
41 pr_err("msgpool_get pool %p need front %d, pool size is %d\n",
42 pool, front_len, pool->front_len);
45 /* try to alloc a fresh message */
46 return ceph_msg_new(0, front_len);
49 return mempool_alloc(pool->pool, GFP_NOFS);
52 void ceph_msgpool_put(struct ceph_msgpool *pool, struct ceph_msg *msg)
54 /* reset msg front_len; user may have changed it */
55 msg->front.iov_len = pool->front_len;
56 msg->hdr.front_len = cpu_to_le32(pool->front_len);
58 kref_init(&msg->kref); /* retake single ref */