1#include "ceph_debug.h"
2
3#include <linux/err.h>
4#include <linux/sched.h>
5#include <linux/types.h>
6#include <linux/vmalloc.h>
7
8#include "msgpool.h"
9
10static void *alloc_fn(gfp_t gfp_mask, void *arg)
11{
12	struct ceph_msgpool *pool = arg;
13	void *p;
14
15	p = ceph_msg_new(0, pool->front_len, gfp_mask);
16	if (!p)
17		pr_err("msgpool %s alloc failed\n", pool->name);
18	return p;
19}
20
21static void free_fn(void *element, void *arg)
22{
23	ceph_msg_put(element);
24}
25
26int ceph_msgpool_init(struct ceph_msgpool *pool,
27		      int front_len, int size, bool blocking, const char *name)
28{
29	pool->front_len = front_len;
30	pool->pool = mempool_create(size, alloc_fn, free_fn, pool);
31	if (!pool->pool)
32		return -ENOMEM;
33	pool->name = name;
34	return 0;
35}
36
37void ceph_msgpool_destroy(struct ceph_msgpool *pool)
38{
39	mempool_destroy(pool->pool);
40}
41
42struct ceph_msg *ceph_msgpool_get(struct ceph_msgpool *pool,
43				  int front_len)
44{
45	if (front_len > pool->front_len) {
46		pr_err("msgpool_get pool %s need front %d, pool size is %d\n",
47		       pool->name, front_len, pool->front_len);
48		WARN_ON(1);
49
50		/* try to alloc a fresh message */
51		return ceph_msg_new(0, front_len, GFP_NOFS);
52	}
53
54	return mempool_alloc(pool->pool, GFP_NOFS);
55}
56
57void ceph_msgpool_put(struct ceph_msgpool *pool, struct ceph_msg *msg)
58{
59	/* reset msg front_len; user may have changed it */
60	msg->front.iov_len = pool->front_len;
61	msg->hdr.front_len = cpu_to_le32(pool->front_len);
62
63	kref_init(&msg->kref);  /* retake single ref */
64}
65