granicus.if.org Git - icinga2/blob - lib/base/fifo.cpp

   1 /* Icinga 2 | (c) 2012 Icinga GmbH | GPLv2+ */
   2
   3 #include "base/fifo.hpp"
   4
   5 using namespace icinga;
   6
   7 /**
   8  * Destructor for the FIFO class.
   9  */
  10 FIFO::~FIFO()
  11 {
  12         free(m_Buffer);
  13 }
  14
  15 /**
  16  * Resizes the FIFO's buffer so that it is at least newSize bytes long.
  17  *
  18  * @param newSize The minimum new size of the FIFO buffer.
  19  */
  20 void FIFO::ResizeBuffer(size_t newSize, bool decrease)
  21 {
  22         if (m_AllocSize >= newSize && !decrease)
  23                 return;
  24
  25         newSize = (newSize / FIFO::BlockSize + 1) * FIFO::BlockSize;
  26
  27         if (newSize == m_AllocSize)
  28                 return;
  29
  30         auto *newBuffer = static_cast<char *>(realloc(m_Buffer, newSize));
  31
  32         if (!newBuffer)
  33                 BOOST_THROW_EXCEPTION(std::bad_alloc());
  34
  35         m_Buffer = newBuffer;
  36
  37         m_AllocSize = newSize;
  38 }
  39
  40 /**
  41  * Optimizes memory usage of the FIFO buffer by reallocating
  42  * and moving the buffer.
  43  */
  44 void FIFO::Optimize()
  45 {
  46         if (m_Offset > m_DataSize / 10 && m_Offset - m_DataSize > 1024) {
  47                 std::memmove(m_Buffer, m_Buffer + m_Offset, m_DataSize);
  48                 m_Offset = 0;
  49
  50                 if (m_DataSize > 0)
  51                         ResizeBuffer(m_DataSize, true);
  52
  53                 return;
  54         }
  55 }
  56
  57 size_t FIFO::Peek(void *buffer, size_t count, bool allow_partial)
  58 {
  59         ASSERT(allow_partial);
  60
  61         if (count > m_DataSize)
  62                 count = m_DataSize;
  63
  64         if (buffer)
  65                 std::memcpy(buffer, m_Buffer + m_Offset, count);
  66
  67         return count;
  68 }
  69
  70 /**
  71  * Implements IOQueue::Read.
  72  */
  73 size_t FIFO::Read(void *buffer, size_t count, bool allow_partial)
  74 {
  75         ASSERT(allow_partial);
  76
  77         if (count > m_DataSize)
  78                 count = m_DataSize;
  79
  80         if (buffer)
  81                 std::memcpy(buffer, m_Buffer + m_Offset, count);
  82
  83         m_DataSize -= count;
  84         m_Offset += count;
  85
  86         Optimize();
  87
  88         return count;
  89 }
  90
  91 /**
  92  * Implements IOQueue::Write.
  93  */
  94 void FIFO::Write(const void *buffer, size_t count)
  95 {
  96         ResizeBuffer(m_Offset + m_DataSize + count, false);
  97         std::memcpy(m_Buffer + m_Offset + m_DataSize, buffer, count);
  98         m_DataSize += count;
  99
 100         SignalDataAvailable();
 101 }
 102
 103 void FIFO::Close()
 104 { }
 105
 106 bool FIFO::IsEof() const
 107 {
 108         return false;
 109 }
 110
 111 size_t FIFO::GetAvailableBytes() const
 112 {
 113         return m_DataSize;
 114 }
 115
 116 bool FIFO::SupportsWaiting() const
 117 {
 118         return true;
 119 }
 120
 121 bool FIFO::IsDataAvailable() const
 122 {
 123         return m_DataSize > 0;
 124 }