qdevices/dynar-str.c

   1 /*
   2  * Copyright (c) 2015-2016 Red Hat, Inc.
   3  *
   4  * All rights reserved.
   5  *
   6  * Author: Jan Friesse (jfriesse@redhat.com)
   7  *
   8  * This software licensed under BSD license, the text of which follows:
   9  *
  10  * Redistribution and use in source and binary forms, with or without
  11  * modification, are permitted provided that the following conditions are met:
  12  *
  13  * - Redistributions of source code must retain the above copyright notice,
  14  *   this list of conditions and the following disclaimer.
  15  * - Redistributions in binary form must reproduce the above copyright notice,
  16  *   this list of conditions and the following disclaimer in the documentation
  17  *   and/or other materials provided with the distribution.
  18  * - Neither the name of the Red Hat, Inc. nor the names of its
  19  *   contributors may be used to endorse or promote products derived from this
  20  *   software without specific prior written permission.
  21  *
  22  * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
  23  * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  24  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
  25  * ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE
  26  * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
  27  * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
  28  * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
  29  * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
  30  * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
  31  * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF
  32  * THE POSSIBILITY OF SUCH DAMAGE.
  33  */
  34
  35 #include <string.h>
  36 #include <stdarg.h>
  37 #include <stdio.h>
  38
  39 #include "dynar-str.h"
  40
  41 int
  42 dynar_str_cpy(struct dynar *dest, const char *str)
  43 {
  44
  45         if (strlen(str) > dynar_max_size(dest)) {
  46                 return (-1);
  47         }
  48
  49         dynar_clean(dest);
  50
  51         return (dynar_str_cat(dest, str));
  52 }
  53
  54 int
  55 dynar_str_cat(struct dynar *dest, const char *str)
  56 {
  57
  58         return (dynar_cat(dest, str, strlen(str)));
  59 }
  60
  61 int
  62 dynar_str_prepend(struct dynar *dest, const char *str)
  63 {
  64
  65         return (dynar_prepend(dest, str, strlen(str)));
  66 }
  67
  68 int
  69 dynar_str_vcatf(struct dynar *dest, const char *format, va_list ap)
  70 {
  71         int to_write;
  72         int written;
  73         va_list ap_copy;
  74         size_t allocated;
  75         char buf;
  76         char *p;
  77
  78         /*
  79          * Find out how much bytes is needed
  80          */
  81         va_copy(ap_copy, ap);
  82         to_write = vsnprintf(&buf, sizeof(buf), format, ap_copy);
  83         va_end(ap_copy);
  84
  85         if (to_write < 0) {
  86                 return (-1);
  87         }
  88
  89         if ((size_t)to_write < sizeof(buf)) {
  90                 /*
  91                  * Writing 1 byte string (snprintf writes also '\0') means string is empty
  92                  */
  93
  94                 return (0);
  95         }
  96
  97         allocated = to_write + 1;
  98         if (dynar_prealloc(dest, allocated) != 0) {
  99                 return (-1);
 100         }
 101
 102         p = dynar_data(dest) + dynar_size(dest);
 103
 104         va_copy(ap_copy, ap);
 105         written = vsnprintf(p, allocated, format, ap_copy);
 106         va_end(ap_copy);
 107
 108         if (written < 0) {
 109                 return (-1);
 110         }
 111
 112         if ((size_t)written >= allocated) {
 113                 return (-1);
 114         }
 115
 116         dest->size += written;
 117
 118         return (written);
 119 }
 120
 121 int
 122 dynar_str_catf(struct dynar *dest, const char *format, ...)
 123 {
 124         va_list ap;
 125         int res;
 126
 127         va_start(ap, format);
 128         res = dynar_str_vcatf(dest, format, ap);
 129         va_end(ap);
 130
 131         return (res);
 132 }
 133
 134 int
 135 dynar_str_quote_cat(struct dynar *dest, const char *str)
 136 {
 137         size_t zi;
 138
 139         if (dynar_str_cat(dest, "\"") != 0) {
 140                 return (-1);
 141         }
 142
 143         for (zi = 0; zi < strlen(str); zi++) {
 144                 if (str[zi] == '"' || str[zi] == '\\') {
 145                         if (dynar_str_cat(dest, "\\") != 0) {
 146                                 return (-1);
 147                         }
 148                 }
 149
 150                 if (dynar_cat(dest, &str[zi], sizeof(str[zi])) != 0) {
 151                         return (-1);
 152                 }
 153         }
 154
 155         if (dynar_str_cat(dest, "\"") != 0) {
 156                 return (-1);
 157         }
 158
 159         return (0);
 160 }
 161
 162 int
 163 dynar_str_quote_cpy(struct dynar *dest, const char *str)
 164 {
 165
 166         dynar_clean(dest);
 167
 168         return (dynar_str_quote_cat(dest, str));
 169 }