pool.c

Go to the documentation of this file.

 
#include "config.h"
#include <stdio.h>
#include "pool.h"
#include "buffer.h"
#include "logging2.h"
#include "memory.h"
#include "signal2.h"

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

static size_t BufferPoolGetCount = 0;
static size_t BufferPoolGivenOut = 0;
static size_t BufferPoolMaxGivenOut = 0;

static void pool_increase_size(void)
{
  BufferPoolLen += BufferPoolIncrement;
  mutt_debug(LL_DEBUG1, "%zu\n", BufferPoolLen);
 
  MUTT_MEM_REALLOC(&BufferPool, BufferPoolLen, struct Buffer *);
  while (BufferPoolCount < BufferPoolIncrement)
  {
    struct Buffer *newbuf = buf_new(NULL);
    buf_alloc(newbuf, BufferPoolInitialBufferSize);
    BufferPool[BufferPoolCount++] = newbuf;
  }
}

void buf_pool_cleanup(void)
{
  mutt_debug(LL_DEBUG3, "%zu of %zu returned to pool\n", BufferPoolCount, BufferPoolLen);
  mutt_debug(LL_DEBUG3, "pool stats: %zu gets, %zu max buffers in use\n",
             BufferPoolGetCount, BufferPoolMaxGivenOut);
 
  while (BufferPoolCount)
    buf_free(&BufferPool[--BufferPoolCount]);
  FREE(&BufferPool);
  BufferPoolLen = 0;
}

struct Buffer *buf_pool_get(void)
{
  if (BufferPoolCount == 0)
    pool_increase_size();
  ASSERT(BufferPoolCount > 0);
 
  BufferPoolGetCount++;
  BufferPoolGivenOut++;
  if (BufferPoolGivenOut > BufferPoolMaxGivenOut)
    BufferPoolMaxGivenOut = BufferPoolGivenOut;
 
  return BufferPool[--BufferPoolCount];
}

void buf_pool_release(struct Buffer **ptr)
{
  if (!ptr || !*ptr)
    return;
 
  if (BufferPoolGivenOut > 0)
    BufferPoolGivenOut--;
 
  if (BufferPoolCount >= BufferPoolLen)
  {
    // LCOV_EXCL_START
    mutt_debug(LL_DEBUG1, "Internal buffer pool error\n");
    buf_free(ptr);
    return;
    // LCOV_EXCL_STOP
  }
 
  // Reset the size if it's too big or too small
  struct Buffer *buf = *ptr;
  if ((buf->dsize > (2 * BufferPoolInitialBufferSize)) ||
      (buf->dsize < BufferPoolInitialBufferSize))
  {
    buf->dsize = BufferPoolInitialBufferSize;
    MUTT_MEM_REALLOC(&buf->data, buf->dsize, char);
  }
  buf_reset(buf);
  BufferPool[BufferPoolCount++] = buf;
 
  *ptr = NULL;
}