pool.c

Go to the documentation of this file.

#include "config.h"
#include <stdio.h>
#include "buffer.h"
#include "logging.h"
#include "memory.h"

static size_t BufferPoolCount = 0;
static size_t BufferPoolLen = 0;
static size_t BufferPoolIncrement = 20;
static size_t BufferPoolInitialBufferSize = 1024;
static struct Buffer **BufferPool = NULL;

static struct Buffer *buffer_new(void)
{
  struct Buffer *buf = mutt_mem_malloc(sizeof(struct Buffer));
  mutt_buffer_init(buf);
  return buf;
}

static void buffer_free(struct Buffer **p)
{
  if (!p || !*p)
    return;

  mutt_buffer_dealloc(*p);
  FREE(p);
}

static void increase_buffer_pool(void)
{
  BufferPoolLen += BufferPoolIncrement;
  mutt_debug(LL_DEBUG1, "%zu\n", BufferPoolLen);

  mutt_mem_realloc(&BufferPool, BufferPoolLen * sizeof(struct Buffer *));
  while (BufferPoolCount < BufferPoolIncrement)
  {
    struct Buffer *newbuf = buffer_new();
    mutt_buffer_alloc(newbuf, BufferPoolInitialBufferSize);
    BufferPool[BufferPoolCount++] = newbuf;
  }
}

void mutt_buffer_pool_free(void)
{
  mutt_debug(LL_DEBUG1, "%zu of %zu returned to pool\n", BufferPoolCount, BufferPoolLen);

  while (BufferPoolCount)
    buffer_free(&BufferPool[--BufferPoolCount]);
  FREE(&BufferPool);
  BufferPoolLen = 0;
}

struct Buffer *mutt_buffer_pool_get(void)
{
  if (BufferPoolCount == 0)
    increase_buffer_pool();
  return BufferPool[--BufferPoolCount];
}

void mutt_buffer_pool_release(struct Buffer **pbuf)
{
  if (!pbuf || !*pbuf)
    return;

  if (BufferPoolCount >= BufferPoolLen)
  {
    mutt_debug(LL_DEBUG1, "Internal buffer pool error\n");
    buffer_free(pbuf);
    return;
  }

  struct Buffer *buf = *pbuf;
  if (buf->dsize > 2 * BufferPoolInitialBufferSize)
  {
    buf->dsize = BufferPoolInitialBufferSize;
    mutt_mem_realloc(&buf->data, buf->dsize);
  }
  mutt_buffer_reset(buf);
  BufferPool[BufferPoolCount++] = buf;

  *pbuf = NULL;
}