agxbuf.h

Go to the documentation of this file.

 
/*************************************************************************
 * Copyright (c) 2011 AT&T Intellectual Property
 * All rights reserved. This program and the accompanying materials
 * are made available under the terms of the Eclipse Public License v1.0
 * which accompanies this distribution, and is available at
 * https://www.eclipse.org/legal/epl-v10.html
 *
 * Contributors: Details at https://graphviz.org
 *************************************************************************/
 
#pragma once
 
#include <assert.h>
#include <limits.h>
#include <stdarg.h>
#include <stdbool.h>
#include <stddef.h>
#include <stdint.h>
#include <stdio.h>
#include <string.h>
#include <util/alloc.h>
#include <util/unused.h>
 
typedef enum {
  AGXBUF_INLINE_SIZE_0 = 0,
  AGXBUF_ON_HEAP = 255, 
} agxbuf_loc_t;
 
typedef struct {
  union {
    struct {
      char *buf;                        
      size_t size;                      
      size_t capacity;                  
      char padding[sizeof(size_t) - 1]; 
      unsigned char
          located; 
    } s;
    char store[sizeof(char *) + sizeof(size_t) * 3 -
               1]; 
  } u;
} agxbuf;
 
static inline bool agxbuf_is_inline(const agxbuf *xb) {
  assert((xb->u.s.located == AGXBUF_ON_HEAP ||
          xb->u.s.located <= sizeof(xb->u.store)) &&
         "corrupted agxbuf type");
  return xb->u.s.located < AGXBUF_ON_HEAP;
}
 
static inline void agxbfree(agxbuf *xb) {
  if (xb->u.s.located == AGXBUF_ON_HEAP)
    free(xb->u.s.buf);
}
 
static inline char *agxbstart(agxbuf *xb) {
  return agxbuf_is_inline(xb) ? xb->u.store : xb->u.s.buf;
}
 
static inline size_t agxblen(const agxbuf *xb) {
  if (agxbuf_is_inline(xb)) {
    return xb->u.s.located - AGXBUF_INLINE_SIZE_0;
  }
  return xb->u.s.size;
}
 
static inline size_t agxbsizeof(const agxbuf *xb) {
  if (agxbuf_is_inline(xb)) {
    return sizeof(xb->u.store);
  }
  return xb->u.s.capacity;
}
 
static inline int agxbpop(agxbuf *xb) {
 
  size_t len = agxblen(xb);
  if (len == 0) {
    return -1;
  }
 
  if (agxbuf_is_inline(xb)) {
    assert(xb->u.s.located > AGXBUF_INLINE_SIZE_0);
    int c = xb->u.store[len - 1];
    --xb->u.s.located;
    return c;
  }
 
  int c = xb->u.s.buf[xb->u.s.size - 1];
  --xb->u.s.size;
  return c;
}
 
static inline void agxbmore(agxbuf *xb, size_t ssz) {
  size_t cnt = 0;   // current no. of characters in buffer
  size_t size = 0;  // current buffer size
  size_t nsize = 0; // new buffer size
  char *nbuf;       // new buffer
 
  size = agxbsizeof(xb);
  nsize = size == 0 ? BUFSIZ : (2 * size);
  if (size + ssz > nsize)
    nsize = size + ssz;
  cnt = agxblen(xb);
 
  if (xb->u.s.located == AGXBUF_ON_HEAP) {
    nbuf = (char *)gv_recalloc(xb->u.s.buf, size, nsize, sizeof(char));
  } else {
    nbuf = (char *)gv_calloc(nsize, sizeof(char));
    memcpy(nbuf, xb->u.store, cnt);
    xb->u.s.size = cnt;
  }
  xb->u.s.buf = nbuf;
  xb->u.s.capacity = nsize;
  xb->u.s.located = AGXBUF_ON_HEAP;
}
 
static inline char *agxbnext(agxbuf *xb) {
  size_t len = agxblen(xb);
  return agxbuf_is_inline(xb) ? &xb->u.store[len] : &xb->u.s.buf[len];
}
 
static inline int vagxbprint(agxbuf *xb, const char *fmt, va_list ap) {
  size_t size;
  int result;
 
  // determine how many bytes we need to print
  {
    va_list ap2;
    int rc;
    va_copy(ap2, ap);
    rc = vsnprintf(NULL, 0, fmt, ap2);
    va_end(ap2);
    if (rc < 0) {
      va_end(ap);
      return rc;
    }
    size = (size_t)rc + 1; // account for NUL terminator
  }
 
  // should we use double buffering?
  bool use_stage = false;
 
  // do we need to expand the buffer?
  {
    size_t unused_space = agxbsizeof(xb) - agxblen(xb);
    if (unused_space < size) {
      size_t extra = size - unused_space;
      if (agxbuf_is_inline(xb) && extra == 1) {
        // The content is currently stored inline, but this print will push it
        // over into being heap-allocated by a single byte. This last byte is a
        // '\0' that `vsnprintf` unavoidably writes but we do not need. So lets
        // avoid this by printing to an intermediate, larger buffer, and then
        // copying the content minus the trailing '\0' to the final destination.
        use_stage = true;
      } else {
        agxbmore(xb, extra);
      }
    }
  }
 
  // a buffer one byte larger than inline storage to fit the trailing '\0'
  char stage[sizeof(xb->u.store) + 1] = {0};
  assert(!use_stage || size <= sizeof(stage));
 
  // we can now safely print into the buffer
  char *dst = use_stage ? stage : agxbnext(xb);
  result = vsnprintf(dst, size, fmt, ap);
  assert(result == (int)(size - 1) || result < 0);
  if (result > 0) {
    if (agxbuf_is_inline(xb)) {
      assert(result <= (int)UCHAR_MAX);
      if (use_stage) {
        memcpy(agxbnext(xb), stage, (size_t)result);
      }
      xb->u.s.located += (unsigned char)result;
      assert(agxblen(xb) <= sizeof(xb->u.store) && "agxbuf corruption");
    } else {
      assert(!use_stage);
      xb->u.s.size += (size_t)result;
    }
  }
 
  return result;
}
 
/* support for extra API misuse warnings if available */
#ifdef __GNUC__
#define PRINTF_LIKE(index, first) __attribute__((format(printf, index, first)))
#else
#define PRINTF_LIKE(index, first) /* nothing */
#endif
 
static inline PRINTF_LIKE(2, 3) int agxbprint(agxbuf *xb, const char *fmt,
                                              ...) {
  va_list ap;
  int result;
 
  va_start(ap, fmt);
 
  result = vagxbprint(xb, fmt, ap);
 
  va_end(ap);
  return result;
}
 
#undef PRINTF_LIKE
 
static inline size_t agxbput_n(agxbuf *xb, const char *s, size_t ssz) {
  if (ssz == 0) {
    return 0;
  }
  if (ssz > agxbsizeof(xb) - agxblen(xb))
    agxbmore(xb, ssz);
  size_t len = agxblen(xb);
  if (agxbuf_is_inline(xb)) {
    memcpy(&xb->u.store[len], s, ssz);
    assert(ssz <= UCHAR_MAX);
    xb->u.s.located += (unsigned char)ssz;
    assert(agxblen(xb) <= sizeof(xb->u.store) && "agxbuf corruption");
  } else {
    memcpy(&xb->u.s.buf[len], s, ssz);
    xb->u.s.size += ssz;
  }
  return ssz;
}
 
static inline size_t agxbput(agxbuf *xb, const char *s) {
  size_t ssz = strlen(s);
 
  return agxbput_n(xb, s, ssz);
}
 
static inline int agxbputc(agxbuf *xb, char c) {
  if (agxblen(xb) >= agxbsizeof(xb)) {
    agxbmore(xb, 1);
  }
  size_t len = agxblen(xb);
  if (agxbuf_is_inline(xb)) {
    xb->u.store[len] = c;
    ++xb->u.s.located;
    assert(agxblen(xb) <= sizeof(xb->u.store) && "agxbuf corruption");
  } else {
    xb->u.s.buf[len] = c;
    ++xb->u.s.size;
  }
  return 0;
}
 
static inline void agxbclear(agxbuf *xb) {
  if (agxbuf_is_inline(xb)) {
    xb->u.s.located = AGXBUF_INLINE_SIZE_0;
  } else {
    xb->u.s.size = 0;
  }
}
 
/* Null-terminates buffer; resets and returns pointer to data. The buffer is
 * still associated with the agxbuf and will be overwritten on the next, e.g.,
 * agxbput. If you want to retrieve and disassociate the buffer, use agxbdisown
 * instead.
 */
static inline WUR char *agxbuse(agxbuf *xb) {
  if (!agxbuf_is_inline(xb) || agxblen(xb) != sizeof(xb->u.store)) {
    (void)agxbputc(xb, '\0');
  } else {
    // we can skip explicitly null-terminating the buffer because `agxbclear`
    // resets the `xb->located` byte such that it naturally forms a terminator
    assert(AGXBUF_INLINE_SIZE_0 == '\0');
  }
 
  agxbclear(xb);
  return agxbstart(xb);
}
 
/* Disassociate the backing buffer from this agxbuf and return it. The buffer is
 * NUL terminated before being returned. If the agxbuf is using stack memory,
 * this will first copy the data to a new heap buffer to then return. If you
 * want to temporarily access the string in the buffer, but have it overwritten
 * and reused the next time, e.g., agxbput is called, use agxbuse instead of
 * agxbdisown.
 */
static inline char *agxbdisown(agxbuf *xb) {
  char *buf;
 
  if (agxbuf_is_inline(xb)) {
    // the string lives in `store`, so we need to copy its contents to heap
    // memory
    buf = gv_strndup(xb->u.store, agxblen(xb));
  } else {
    // the buffer is already dynamically allocated, so terminate it and then
    // take it as-is
    agxbputc(xb, '\0');
    buf = xb->u.s.buf;
  }
 
  // reset xb to a state where it is usable
  memset(xb, 0, sizeof(*xb));
 
  return buf;
}
 
static inline void agxbuf_trim_zeros(agxbuf *xb) {
 
  // find last period
  char *start = agxbstart(xb);
  size_t period;
  for (period = agxblen(xb) - 1;; --period) {
    if (period == SIZE_MAX) {
      // we searched the entire string and did not find a period
      return;
    }
    if (start[period] == '.') {
      break;
    }
  }
 
  // truncate any “0”s that provide no information
  for (size_t follower = agxblen(xb) - 1;; --follower) {
    if (follower == period || start[follower] == '0') {
      // truncate this character
      if (agxbuf_is_inline(xb)) {
        assert(xb->u.s.located > AGXBUF_INLINE_SIZE_0);
        --xb->u.s.located;
      } else {
        --xb->u.s.size;
      }
      if (follower == period) {
        break;
      }
    } else {
      return;
    }
  }
 
  // is the remainder we have left not “-0”?
  const size_t len = agxblen(xb);
  if (len < 2 || start[len - 2] != '-' || start[len - 1] != '0') {
    return;
  }
 
  // turn “-0” into “0”
  start[len - 2] = '0';
  if (agxbuf_is_inline(xb)) {
    assert(xb->u.s.located > AGXBUF_INLINE_SIZE_0);
    --xb->u.s.located;
  } else {
    --xb->u.s.size;
  }
}