SAFE public projects git trees. - safe/jmp/linux-2.6/blob - 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
  14         return ceph_msg_new(0, pool->front_len);
  15 }
  16
  17 static void free_fn(void *element, void *arg)
  18 {
  19         ceph_msg_put(element);
  20 }
  21
  22 int ceph_msgpool_init(struct ceph_msgpool *pool,
  23                       int front_len, int size, bool blocking)
  24 {
  25         pool->front_len = front_len;
  26         pool->pool = mempool_create(size, alloc_fn, free_fn, pool);
  27         if (!pool->pool)
  28                 return -ENOMEM;
  29         return 0;
  30 }
  31
  32 void ceph_msgpool_destroy(struct ceph_msgpool *pool)
  33 {
  34         mempool_destroy(pool->pool);
  35 }
  36
  37 struct ceph_msg *ceph_msgpool_get(struct ceph_msgpool *pool,
  38                                   int front_len)
  39 {
  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);
  43                 WARN_ON(1);
  44
  45                 /* try to alloc a fresh message */
  46                 return ceph_msg_new(0, front_len);
  47         }
  48
  49         return mempool_alloc(pool->pool, GFP_NOFS);
  50 }
  51
  52 void ceph_msgpool_put(struct ceph_msgpool *pool, struct ceph_msg *msg)
  53 {
  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);
  57
  58         kref_init(&msg->kref);  /* retake single ref */
  59 }