1 /* Licensed under BSD-MIT - see LICENSE file for details */
13 #include <ccan/str/str.h>
15 char *tal_strdup_(const tal_t *ctx, const char *p, const char *label)
17 /* We have to let through NULL for take(). */
18 return tal_dup_arr_label(ctx, char, p, p ? strlen(p) + 1: 1, 0, label);
21 char *tal_strndup_(const tal_t *ctx, const char *p, size_t n, const char *label)
26 /* We have to let through NULL for take(). */
32 ret = tal_dup_arr_label(ctx, char, p, len, 1, label);
38 char *tal_fmt_(const tal_t *ctx, const char *label, const char *fmt, ...)
44 ret = tal_vfmt_(ctx, fmt, ap, label);
50 static bool do_vfmt(char **buf, size_t off, const char *fmt, va_list ap)
52 /* A decent guess to start. */
53 size_t max = strlen(fmt) * 2 + 1;
60 if (!tal_resize(buf, off + max)) {
66 ret = vsnprintf(*buf + off, max, fmt, ap2);
71 /* Make sure tal_count() is correct! */
72 tal_resize(buf, off + ret + 1);
83 char *tal_vfmt_(const tal_t *ctx, const char *fmt, va_list ap, const char *label)
87 if (!fmt && taken(fmt))
90 /* A decent guess to start. */
91 buf = tal_arr_label(ctx, char, strlen(fmt) * 2, label);
92 if (!do_vfmt(&buf, 0, fmt, ap))
97 bool tal_append_vfmt(char **baseptr, const char *fmt, va_list ap)
99 if (!fmt && taken(fmt))
102 return do_vfmt(baseptr, strlen(*baseptr), fmt, ap);
105 bool tal_append_fmt(char **baseptr, const char *fmt, ...)
111 ret = tal_append_vfmt(baseptr, fmt, ap);
117 char *tal_strcat_(const tal_t *ctx, const char *s1, const char *s2,
123 if (unlikely(!s2) && taken(s2)) {
128 /* We have to let through NULL for take(). */
129 len1 = s1 ? strlen(s1) : 0;
132 ret = tal_dup_arr_label(ctx, char, s1, len1, len2 + 1, label);
134 memcpy(ret + len1, s2, len2 + 1);
141 char **tal_strsplit_(const tal_t *ctx,
142 const char *string, const char *delims, enum strsplit flags,
146 size_t max = 64, num = 0;
148 parts = tal_arr(ctx, char *, max + 1);
149 if (unlikely(!parts)) {
156 str = tal_strdup(parts, string);
159 if (unlikely(!delims) && is_taken(delims))
162 if (flags == STR_NO_EMPTY)
163 str += strspn(str, delims);
165 while (*str != '\0') {
166 size_t len = strcspn(str, delims), dlen;
169 dlen = strspn(str + len, delims);
170 parts[num][len] = '\0';
171 if (flags == STR_EMPTY_OK && dlen)
174 if (++num == max && !tal_resize(&parts, max*=2 + 1))
179 /* Ensure that tal_count() is correct. */
180 if (unlikely(!tal_resize(&parts, num+1)))
194 char *tal_strjoin_(const tal_t *ctx,
195 char *strings[], const char *delim, enum strjoin flags,
200 size_t totlen = 0, dlen;
202 if (unlikely(!strings) && is_taken(strings))
205 if (unlikely(!delim) && is_taken(delim))
208 dlen = strlen(delim);
209 ret = tal_arr_label(ctx, char, dlen*2+1, label);
214 for (i = 0; strings[i]; i++) {
215 size_t len = strlen(strings[i]);
217 if (flags == STR_NO_TRAIL && !strings[i+1])
219 if (!tal_resize(&ret, totlen + len + dlen + 1))
221 memcpy(ret + totlen, strings[i], len);
223 memcpy(ret + totlen, delim, dlen);
227 /* Make sure tal_count() is correct! */
228 tal_resize(&ret, totlen+1);
240 static size_t count_open_braces(const char *string)
243 size_t num = 0, esc = 0;
249 /* An odd number of \ means it's escaped. */
250 if (*string == '(' && (esc & 1) == 0)
258 return strcount(string, "(");
262 bool tal_strreg_(const tal_t *ctx, const char *string, const char *label,
263 const char *regex, ...)
265 size_t nmatch = 1 + count_open_braces(regex);
266 regmatch_t matches[nmatch];
272 if (unlikely(!regex) && is_taken(regex))
275 if (regcomp(&r, regex, REG_EXTENDED) != 0)
278 if (unlikely(!string) && is_taken(string))
281 if (regexec(&r, string, nmatch, matches, 0) != 0)
286 for (i = 1; i < nmatch; i++) {
287 char **arg = va_arg(ap, char **);
289 /* eg. ([a-z])? can give "no match". */
290 if (matches[i].rm_so == -1)
293 *arg = tal_strndup_(ctx,
294 string + matches[i].rm_so,
298 /* FIXME: If we fail, we set some and leak! */