iovec.h

Go to the documentation of this file.

#ifndef __SOLAR_CAPTURE_IOVEC_H__
#define __SOLAR_CAPTURE_IOVEC_H__

#include <string.h>


struct sc_iovec_ptr {
  const struct iovec* iov;
  int                 iovlen;
  struct iovec        io;
};

static inline void __sc_iovec_ptr_init(struct sc_iovec_ptr* iovp,
                                       const struct iovec* iov, int iovlen)
{
  iovp->iovlen = iovlen;
  iovp->iov = iov;
  iovp->io = iov[0];
}
static inline void sc_iovec_ptr_init(struct sc_iovec_ptr* iovp,
                                     const struct iovec* iov, int iovlen)
{
  iovp->iovlen = iovlen;
  iovp->iov = iov;
  if( iovlen )
    iovp->io = iov[0];
}


static inline void sc_iovec_ptr_init_buf(struct sc_iovec_ptr* iovp,
                                         void* buf, int len)
{
  iovp->iovlen = 1;
  iovp->iov = NULL;
  iovp->io.iov_base = buf;
  iovp->io.iov_len = len;
}


static inline void sc_iovec_ptr_init_packet(struct sc_iovec_ptr* iovp,
                                            const struct sc_packet* packet)
{
  __sc_iovec_ptr_init(iovp, packet->iov, packet->iovlen);
}


static inline int sc_iovec_ptr_bytes(const struct sc_iovec_ptr* iovp)
{
  int i, bytes = 0;
  if( iovp->iovlen ) {
    bytes += iovp->io.iov_len;
    for( i = 1; i < iovp->iovlen; ++i )
      bytes += iovp->iov[i].iov_len;
  }
  return bytes;
}


#if SC_API_VER >= 1
extern int sc_iovec_ptr_skip(struct sc_iovec_ptr* iovp, int bytes_to_skip);
#endif


#if SC_API_VER >= 1
extern int sc_iovec_ptr_find_chr(const struct sc_iovec_ptr* iovp, int c);
#endif


#if SC_API_VER >= 1
extern int sc_iovec_ptr_copy_out(void* dest, struct sc_iovec_ptr* iovp,
                                 int max_bytes);
#endif


static inline void sc_iovec_copy_from_end(void* dest_buf,
                                          const struct iovec* iov,
                                          int iovlen, int bytes)
{
  iov += iovlen - 1;
  while( 1 ) {
    unsigned n = bytes;
    if( n > iov->iov_len )
      n = iov->iov_len;
    memcpy((char*) dest_buf + bytes - n,
           (const char*) iov->iov_base + iov->iov_len - n, n);
    if( (bytes -= n) == 0 )
      break;
    --iov;
    --iovlen;
    assert(iovlen > 0);
  }
}

static inline void sc_iovec_trim_end(struct iovec* iov, uint8_t* iovlen,
                                     int bytes)
{
  iov += *iovlen - 1;
  while( 1 ) {
    if( bytes < (int) iov->iov_len ) {
      iov->iov_len -= bytes;
      return;
    }
    bytes -= iov->iov_len;
    --iov;
    --(*iovlen);
    if( bytes == 0 )
      break;
    assert(*iovlen > 0);
  }
}


#endif  /* __SOLAR_CAPTURE_IOVEC_H__ */