common/buffer.c

   1 /******************************************************************************/
   2 /*          pfixtools: a collection of postfix related tools                  */
   3 /*          ~~~~~~~~~                                                         */
   4 /*  ________________________________________________________________________  */
   5 /*                                                                            */
   6 /*  Redistribution and use in source and binary forms, with or without        */
   7 /*  modification, are permitted provided that the following conditions        */
   8 /*  are met:                                                                  */
   9 /*                                                                            */
  10 /*  1. Redistributions of source code must retain the above copyright         */
  11 /*     notice, this list of conditions and the following disclaimer.          */
  12 /*  2. Redistributions in binary form must reproduce the above copyright      */
  13 /*     notice, this list of conditions and the following disclaimer in the    */
  14 /*     documentation and/or other materials provided with the distribution.   */
  15 /*  3. The names of its contributors may not be used to endorse or promote    */
  16 /*     products derived from this software without specific prior written     */
  17 /*     permission.                                                            */
  18 /*                                                                            */
  19 /*  THIS SOFTWARE IS PROVIDED BY THE CONTRIBUTORS ``AS IS'' AND ANY EXPRESS   */
  20 /*  OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED         */
  21 /*  WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE    */
  22 /*  DISCLAIMED.  IN NO EVENT SHALL THE CONTRIBUTORS BE LIABLE FOR ANY         */
  23 /*  DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL        */
  24 /*  DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS   */
  25 /*  OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)     */
  26 /*  HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT,       */
  27 /*  STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN  */
  28 /*  ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE           */
  29 /*  POSSIBILITY OF SUCH DAMAGE.                                               */
  30 /*                                                                            */
  31 /*   Copyright (c) 2006-2008 the Authors                                      */
  32 /*   see AUTHORS and source files for details                                 */
  33 /******************************************************************************/
  34
  35 /*
  36  * Copyright © 2006-2007 Pierre Habouzit
  37  */
  38
  39 #include <errno.h>
  40 #include <stdio.h>
  41 #include <unistd.h>
  42
  43 #include "mem.h"
  44 #include "buffer.h"
  45
  46 int buffer_addvf(buffer_t *buf, const char *fmt, va_list ap)
  47 {
  48     int len, avail = buf->size - buf->len;
  49     va_list ap2;
  50
  51     va_copy(ap2, ap);
  52
  53     len = vsnprintf(buf->data + buf->len, avail, fmt, ap);
  54     if (len >= avail) {
  55         buffer_resize(buf, buf->len + len);
  56         avail = buf->size - buf->len;
  57         len = vsnprintf(buf->data + buf->len, avail, fmt, ap2);
  58     }
  59     buf->len += len;
  60     buf->data[buf->len] = '\0';
  61     va_end(ap2);
  62     return len;
  63 }
  64
  65 void buffer_consume(buffer_t *buf, int len)
  66 {
  67     if (len <= 0)
  68         return;
  69
  70     if (len >= (int)buf->len) {
  71         buffer_reset(buf);
  72         return;
  73     }
  74
  75     memmove(buf->data, buf->data + len, buf->len + 1 - len);
  76     buf->len -= len;
  77 }
  78
  79 int buffer_read(buffer_t *buf, int fd, int count)
  80 {
  81     int res;
  82
  83     if (count < 0)
  84         count = BUFSIZ;
  85
  86     buffer_ensure(buf, count);
  87
  88     res = read(fd, buf->data + buf->len, count);
  89     if (res < 0) {
  90         buf->data[buf->len] = '\0';
  91         return res;
  92     }
  93
  94     buffer_extend(buf, res);
  95     return res;
  96 }
  97
  98 int buffer_write(buffer_t *buf, int fd)
  99 {
 100     int res = write(fd, buf->data, buf->len);
 101     if (res < 0) {
 102         return errno == EINTR || errno == EAGAIN ? 0 : -1;
 103     }
 104     buffer_consume(buf, res);
 105     return res;
 106 }