fs/ceph/msgpool.c

   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
  10 static void *alloc_fn(gfp_t gfp_mask, void *arg)
  11 {
  12         struct ceph_msgpool *pool = arg;
  13         struct ceph_msg *m;
  14
  15         m = ceph_msg_new(0, pool->front_len, 0, 0, NULL);
  16         if (IS_ERR(m))
  17                 return NULL;
  18         return m;
  19 }
  20
  21 static void free_fn(void *element, void *arg)
  22 {
  23         ceph_msg_put(element);
  24 }
  25
  26 int ceph_msgpool_init(struct ceph_msgpool *pool,
  27                       int front_len, int size, bool blocking)
  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         return 0;
  34 }
  35
  36 void ceph_msgpool_destroy(struct ceph_msgpool *pool)
  37 {
  38         mempool_destroy(pool->pool);
  39 }
  40
  41 struct ceph_msg *ceph_msgpool_get(struct ceph_msgpool *pool,
  42                                   int front_len)
  43 {
  44         if (front_len > pool->front_len) {
  45                 struct ceph_msg *msg;
  46
  47                 pr_err("msgpool_get pool %p need front %d, pool size is %d\n",
  48                        pool, front_len, pool->front_len);
  49                 WARN_ON(1);
  50
  51                 /* try to alloc a fresh message */
  52                 msg = ceph_msg_new(0, front_len, 0, 0, NULL);
  53                 if (!IS_ERR(msg))
  54                         return msg;
  55                 return NULL;
  56         }
  57
  58         return mempool_alloc(pool->pool, GFP_NOFS);
  59 }
  60
  61 void ceph_msgpool_put(struct ceph_msgpool *pool, struct ceph_msg *msg)
  62 {
  63         /* reset msg front_len; user may have changed it */
  64         msg->front.iov_len = pool->front_len;
  65         msg->hdr.front_len = cpu_to_le32(pool->front_len);
  66
  67         kref_init(&msg->kref);  /* retake single ref */
  68 }