tal/str: add tal_append_fmt() and tal_append_vfmt() helpers.
[ccan] / 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 }