lib/dynamic-string.c

   1 /* Copyright (C) 2008 Board of Trustees, Leland Stanford Jr. University.
   2  *
   3  * Permission is hereby granted, free of charge, to any person obtaining a copy
   4  * of this software and associated documentation files (the "Software"), to
   5  * deal in the Software without restriction, including without limitation the
   6  * rights to use, copy, modify, merge, publish, distribute, sublicense, and/or
   7  * sell copies of the Software, and to permit persons to whom the Software is
   8  * furnished to do so, subject to the following conditions:
   9  *
  10  * The above copyright notice and this permission notice shall be included in
  11  * all copies or substantial portions of the Software.
  12  *
  13  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
  14  * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
  15  * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
  16  * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
  17  * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
  18  * FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS
  19  * IN THE SOFTWARE.
  20  */
  21
  22 #include "dynamic-string.h"
  23 #include <assert.h>
  24 #include <stdlib.h>
  25 #include "util.h"
  26
  27 void
  28 ds_init(struct ds *ds)
  29 {
  30     ds->string = NULL;
  31     ds->length = 0;
  32     ds->allocated = 0;
  33 }
  34
  35 void
  36 ds_reserve(struct ds *ds, size_t min_length)
  37 {
  38     if (min_length > ds->allocated || !ds->string) {
  39         ds->allocated += MAX(min_length, ds->allocated);
  40         ds->allocated = MAX(8, ds->allocated);
  41         ds->string = xrealloc(ds->string, ds->allocated + 1);
  42     }
  43 }
  44
  45 void
  46 ds_put_format(struct ds *ds, const char *format, ...)
  47 {
  48     va_list args;
  49
  50     va_start(args, format);
  51     ds_put_format_valist(ds, format, args);
  52     va_end(args);
  53 }
  54
  55 void
  56 ds_put_format_valist(struct ds *ds, const char *format, va_list args_)
  57 {
  58     va_list args;
  59     size_t available;
  60     int needed;
  61
  62     va_copy(args, args_);
  63     available = ds->string ? ds->allocated - ds->length + 1 : 0;
  64     needed = vsnprintf(&ds->string[ds->length], available, format, args);
  65     va_end(args);
  66
  67     if (needed < available) {
  68         ds->length += needed;
  69     } else {
  70         size_t available;
  71
  72         ds_reserve(ds, ds->length + needed);
  73
  74         va_copy(args, args_);
  75         available = ds->allocated - ds->length + 1;
  76         needed = vsnprintf(&ds->string[ds->length], available, format, args);
  77         va_end(args);
  78
  79         assert(needed < available);
  80         ds->length += needed;
  81     }
  82 }
  83
  84 char *
  85 ds_cstr(struct ds *ds)
  86 {
  87     if (!ds->string) {
  88         ds_reserve(ds, 0);
  89         ds->string[0] = '\0';
  90     }
  91     return ds->string;
  92 }
  93
  94 void
  95 ds_destroy(struct ds *ds)
  96 {
  97     free(ds->string);
  98 }