git.ozlabs.org Git - ccan/blob - ccan/tal/str/str.c

   1 /* Licensed under BSD-MIT - see LICENSE file for details */
   2 #include <unistd.h>
   3 #include <stdint.h>
   4 #include <string.h>
   5 #include <limits.h>
   6 #include <stdlib.h>
   7 #include "str.h"
   8 #include <sys/types.h>
   9 #include <regex.h>
  10 #include <stdarg.h>
  11 #include <unistd.h>
  12 #include <stdio.h>
  13 #include <ccan/str/str.h>
  14 #include <ccan/tal/tal.h>
  15 #include <ccan/take/take.h>
  16
  17 char *tal_strdup(const tal_t *ctx, const char *p)
  18 {
  19         /* We have to let through NULL for take(). */
  20         return tal_dup_(ctx, p, 1, p ? strlen(p) + 1: 1, 0, false,
  21                         TAL_LABEL(char, "[]"));
  22 }
  23
  24 char *tal_strndup(const tal_t *ctx, const char *p, size_t n)
  25 {
  26         size_t len;
  27         char *ret;
  28
  29         /* We have to let through NULL for take(). */
  30         if (likely(p)) {
  31                 len = strlen(p);
  32                 if (len > n)
  33                         len = n;
  34         } else
  35                 len = n;
  36
  37         ret = tal_dup_(ctx, p, 1, len, 1, false, TAL_LABEL(char, "[]"));
  38         if (ret)
  39                 ret[len] = '\0';
  40         return ret;
  41 }
  42
  43 char *tal_fmt(const tal_t *ctx, const char *fmt, ...)
  44 {
  45         va_list ap;
  46         char *ret;
  47
  48         va_start(ap, fmt);
  49         ret = tal_vfmt(ctx, fmt, ap);
  50         va_end(ap);
  51
  52         return ret;
  53 }
  54
  55 static bool do_vfmt(char **buf, size_t off, const char *fmt, va_list ap)
  56 {
  57         /* A decent guess to start. */
  58         size_t max = strlen(fmt) * 2;
  59         bool ok;
  60
  61         for (;;) {
  62                 va_list ap2;
  63                 int ret;
  64
  65                 if (!tal_resize(buf, off + max)) {
  66                         ok = false;
  67                         break;
  68                 }
  69
  70                 va_copy(ap2, ap);
  71                 ret = vsnprintf(*buf + off, max, fmt, ap2);
  72                 va_end(ap2);
  73
  74                 if (ret < max) {
  75                         ok = true;
  76                         break;
  77                 }
  78                 max *= 2;
  79         }
  80
  81         if (taken(fmt))
  82                 tal_free(fmt);
  83         return ok;
  84 }
  85
  86 char *tal_vfmt(const tal_t *ctx, const char *fmt, va_list ap)
  87 {
  88         char *buf;
  89
  90         if (!fmt && taken(fmt))
  91                 return NULL;
  92
  93         /* A decent guess to start. */
  94         buf = tal_arr(ctx, char, strlen(fmt) * 2);
  95         if (!do_vfmt(&buf, 0, fmt, ap))
  96                 buf = tal_free(buf);
  97         return buf;
  98 }
  99
 100 bool tal_append_vfmt(char **baseptr, const char *fmt, va_list ap)
 101 {
 102         if (!fmt && taken(fmt))
 103                 return false;
 104
 105         return do_vfmt(baseptr, strlen(*baseptr), fmt, ap);
 106 }
 107
 108 bool tal_append_fmt(char **baseptr, const char *fmt, ...)
 109 {
 110         va_list ap;
 111         bool ret;
 112
 113         va_start(ap, fmt);
 114         ret = tal_append_vfmt(baseptr, fmt, ap);
 115         va_end(ap);
 116
 117         return ret;
 118 }
 119
 120 char *tal_strcat(const tal_t *ctx, const char *s1, const char *s2)
 121 {
 122         size_t len1, len2;
 123         char *ret;
 124
 125         if (unlikely(!s2) && taken(s2)) {
 126                 if (taken(s1))
 127                         tal_free(s1);
 128                 return NULL;
 129         }
 130         /* We have to let through NULL for take(). */
 131         len1 = s1 ? strlen(s1) : 0;
 132         len2 = strlen(s2);
 133
 134         /* We use tal_dup_ here to avoid attaching a length property. */
 135         ret = tal_dup_(ctx, s1, 1, len1, len2 + 1, false,
 136                        TAL_LABEL(char, "[]"));
 137         if (likely(ret))
 138                 memcpy(ret + len1, s2, len2 + 1);
 139
 140         if (taken(s2))
 141                 tal_free(s2);
 142         return ret;
 143 }
 144
 145 char **tal_strsplit(const tal_t *ctx,
 146                     const char *string, const char *delims, enum strsplit flags)
 147 {
 148         char **parts, *str;
 149         size_t max = 64, num = 0;
 150
 151         parts = tal_arr(ctx, char *, max + 1);
 152         if (unlikely(!parts)) {
 153                 if (taken(string))
 154                         tal_free(string);
 155                 if (taken(delims))
 156                         tal_free(delims);
 157                 return NULL;
 158         }
 159         str = tal_strdup(parts, string);
 160         if (unlikely(!str))
 161                 goto fail;
 162         if (unlikely(!delims) && is_taken(delims))
 163                 goto fail;
 164
 165         if (flags == STR_NO_EMPTY)
 166                 str += strspn(str, delims);
 167
 168         while (*str != '\0') {
 169                 size_t len = strcspn(str, delims), dlen;
 170
 171                 parts[num] = str;
 172                 dlen = strspn(str + len, delims);
 173                 parts[num][len] = '\0';
 174                 if (flags == STR_EMPTY_OK && dlen)
 175                         dlen = 1;
 176                 str += len + dlen;
 177                 if (++num == max && !tal_resize(&parts, max*=2 + 1))
 178                         goto fail;
 179         }
 180         parts[num] = NULL;
 181
 182         /* Ensure that tal_count() is correct. */
 183         if (unlikely(!tal_resize(&parts, num+1)))
 184                 goto fail;
 185
 186         if (taken(delims))
 187                 tal_free(delims);
 188         return parts;
 189
 190 fail:
 191         tal_free(parts);
 192         if (taken(delims))
 193                 tal_free(delims);
 194         return NULL;
 195 }
 196
 197 char *tal_strjoin(const tal_t *ctx,
 198                   char *strings[], const char *delim, enum strjoin flags)
 199 {
 200         unsigned int i;
 201         char *ret = NULL;
 202         size_t totlen = 0, dlen;
 203
 204         if (unlikely(!strings) && is_taken(strings))
 205                 goto fail;
 206
 207         if (unlikely(!delim) && is_taken(delim))
 208                 goto fail;
 209
 210         dlen = strlen(delim);
 211         ret = tal_arr(ctx, char, dlen*2+1);
 212         if (!ret)
 213                 goto fail;
 214
 215         ret[0] = '\0';
 216         for (i = 0; strings[i]; i++) {
 217                 size_t len = strlen(strings[i]);
 218
 219                 if (flags == STR_NO_TRAIL && !strings[i+1])
 220                         dlen = 0;
 221                 if (!tal_resize(&ret, totlen + len + dlen + 1))
 222                         goto fail;
 223                 memcpy(ret + totlen, strings[i], len);
 224                 totlen += len;
 225                 memcpy(ret + totlen, delim, dlen);
 226                 totlen += dlen;
 227         }
 228         ret[totlen] = '\0';
 229 out:
 230         if (taken(strings))
 231                 tal_free(strings);
 232         if (taken(delim))
 233                 tal_free(delim);
 234         return ret;
 235 fail:
 236         ret = tal_free(ret);
 237         goto out;
 238 }
 239
 240 bool tal_strreg(const tal_t *ctx, const char *string, const char *regex, ...)
 241 {
 242         size_t nmatch = 1 + strcount(regex, "(");
 243         regmatch_t matches[nmatch];
 244         regex_t r;
 245         bool ret = false;
 246         unsigned int i;
 247         va_list ap;
 248
 249         if (unlikely(!regex) && is_taken(regex))
 250                 goto fail_no_re;
 251
 252         if (regcomp(&r, regex, REG_EXTENDED) != 0)
 253                 goto fail_no_re;
 254
 255         if (unlikely(!string) && is_taken(string))
 256                 goto fail;
 257
 258         if (regexec(&r, string, nmatch, matches, 0) != 0)
 259                 goto fail;
 260
 261         ret = true;
 262         va_start(ap, regex);
 263         for (i = 1; i < nmatch; i++) {
 264                 char **arg = va_arg(ap, char **);
 265                 if (arg) {
 266                         /* eg. ([a-z])? can give "no match". */
 267                         if (matches[i].rm_so == -1)
 268                                 *arg = NULL;
 269                         else {
 270                                 *arg = tal_strndup(ctx,
 271                                                    string + matches[i].rm_so,
 272                                                    matches[i].rm_eo
 273                                                    - matches[i].rm_so);
 274                                 /* FIXME: If we fail, we set some and leak! */
 275                                 if (!*arg) {
 276                                         ret = false;
 277                                         break;
 278                                 }
 279                         }
 280                 }
 281         }
 282         va_end(ap);
 283 fail:
 284         regfree(&r);
 285 fail_no_re:
 286         if (taken(regex))
 287                 tal_free(regex);
 288         if (taken(string))
 289                 tal_free(string);
 290         return ret;
 291 }