xroad_net_buf.h

Go to the documentation of this file.

#pragma once
#include "xroad_sock_fwd.h"
#include <common/xroad_common_types.h>
#include <logger/xroad_logger.h>
#include <stdint.h>
#include <stdlib.h>
 
#ifdef __cplusplus
extern "C"
{
#endif
 
struct xroad_net_buf_s
{
   char*  begin;      
   char*  data_begin; 
   char*  data_end;   
   char*  end;        
   size_t max_size;   
};
 
xroad_errno_t xroad_net_buf_create(xroad_net_buf_t* buf, size_t size, size_t max_size);
 
xroad_errno_t xroad_net_buf_resize(xroad_net_buf_t* buf, size_t size);
 
xroad_errno_t xroad_net_buf_increase(xroad_net_buf_t* buf, size_t size);
 
void xroad_net_buf_destroy(xroad_net_buf_t* buf);
 
void xroad_net_buf_free_space(xroad_net_buf_t* buf);
 
static inline void xroad_net_buf_reset(xroad_net_buf_t* buf)
{
   buf->data_begin = buf->data_end = buf->begin;
}
 
static inline size_t xroad_net_buf_get_data_size(const xroad_net_buf_t* buf)
{
   return buf->data_end - buf->data_begin;
}
 
static inline size_t xroad_net_buf_get_size(const xroad_net_buf_t* buf)
{
   return buf->end - buf->begin;
}
 
static inline size_t xroad_net_buf_get_free_size(const xroad_net_buf_t* buf)
{
   return buf->end - buf->data_end;
}
 
static inline char* xroad_net_buf_alloc(xroad_net_buf_t* buf, size_t len)
{
   if (xroad_net_buf_get_free_size(buf) < len)
   {
      if (XROAD_ERROR_NO_MORE_RESOURCES == xroad_net_buf_increase(buf, len))
      {
         xroad_log_error("no more space in buffer");
         return NULL;
      }
   }
   char* ptr = buf->data_end;
   buf->data_end += len;
   return ptr;
}
 
static inline char* xroad_net_buf_free(xroad_net_buf_t* buf, ssize_t pop_sz)
{
   if (pop_sz < 0)
   {
      return buf->data_begin;
   }
   if ((size_t)pop_sz >= xroad_net_buf_get_data_size(buf))
   {
      xroad_net_buf_reset(buf);
   }
   else
   {
      buf->data_begin += pop_sz;
   }
   return buf->data_begin;
}
 
#ifdef NDEBUG
#  define xroad_net_buf_check(buf)
#else
#  define xroad_net_buf_check(buf)                      \
   {                                                    \
      xroad_net_buf_t* buf777_ = (buf);                 \
      assert(*(uint32_t*)buf777_->end == 0xDEADBEAF);   \
      assert(buf777_->data_begin >= buf777_->begin);    \
      assert(buf777_->data_begin <= buf777_->end);      \
      assert(buf777_->data_end >= buf777_->begin);      \
      assert(buf777_->data_end <= buf777_->end);        \
      assert(buf777_->data_begin <= buf777_->data_end); \
   }
#endif
 
#ifdef __cplusplus
}
#endif