]> git.cworth.org Git - obsolete/notmuch-old/blob - util/string-util.c
contrib: pick: pass tab through to the message pane
[obsolete/notmuch-old] / util / string-util.c
1 /* string-util.c -  Extra or enhanced routines for null terminated strings.
2  *
3  * Copyright (c) 2012 Jani Nikula
4  *
5  * This program is free software: you can redistribute it and/or modify
6  * it under the terms of the GNU General Public License as published by
7  * the Free Software Foundation, either version 3 of the License, or
8  * (at your option) any later version.
9  *
10  * This program is distributed in the hope that it will be useful,
11  * but WITHOUT ANY WARRANTY; without even the implied warranty of
12  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
13  * GNU General Public License for more details.
14  *
15  * You should have received a copy of the GNU General Public License
16  * along with this program.  If not, see http://www.gnu.org/licenses/ .
17  *
18  * Author: Jani Nikula <jani@nikula.org>
19  */
20
21
22 #include "string-util.h"
23 #include "talloc.h"
24
25 #include <ctype.h>
26 #include <errno.h>
27
28 char *
29 strtok_len (char *s, const char *delim, size_t *len)
30 {
31     /* skip initial delims */
32     s += strspn (s, delim);
33
34     /* length of token */
35     *len = strcspn (s, delim);
36
37     return *len ? s : NULL;
38 }
39
40 static int
41 is_unquoted_terminator (unsigned char c)
42 {
43     return c == 0 || c <= ' ' || c == ')';
44 }
45
46 int
47 make_boolean_term (void *ctx, const char *prefix, const char *term,
48                    char **buf, size_t *len)
49 {
50     const char *in;
51     char *out;
52     size_t needed = 3;
53     int need_quoting = 0;
54
55     /* Do we need quoting?  To be paranoid, we quote anything
56      * containing a quote, even though it only matters at the
57      * beginning, and anything containing non-ASCII text. */
58     for (in = term; *in && !need_quoting; in++)
59         if (is_unquoted_terminator (*in) || *in == '"'
60             || (unsigned char)*in > 127)
61             need_quoting = 1;
62
63     if (need_quoting)
64         for (in = term; *in; in++)
65             needed += (*in == '"') ? 2 : 1;
66     else
67         needed = strlen (term) + 1;
68
69     /* Reserve space for the prefix */
70     if (prefix)
71         needed += strlen (prefix) + 1;
72
73     if ((*buf == NULL) || (needed > *len)) {
74         *len = 2 * needed;
75         *buf = talloc_realloc (ctx, *buf, char, *len);
76     }
77
78     if (! *buf) {
79         errno = ENOMEM;
80         return -1;
81     }
82
83     out = *buf;
84
85     /* Copy in the prefix */
86     if (prefix) {
87         strcpy (out, prefix);
88         out += strlen (prefix);
89         *out++ = ':';
90     }
91
92     if (! need_quoting) {
93         strcpy (out, term);
94         return 0;
95     }
96
97     /* Quote term by enclosing it in double quotes and doubling any
98      * internal double quotes. */
99     *out++ = '"';
100     in = term;
101     while (*in) {
102         if (*in == '"')
103             *out++ = '"';
104         *out++ = *in++;
105     }
106     *out++ = '"';
107     *out = '\0';
108
109     return 0;
110 }
111
112 static const char*
113 skip_space (const char *str)
114 {
115     while (*str && isspace ((unsigned char) *str))
116         ++str;
117     return str;
118 }
119
120 int
121 parse_boolean_term (void *ctx, const char *str,
122                     char **prefix_out, char **term_out)
123 {
124     int err = EINVAL;
125     *prefix_out = *term_out = NULL;
126
127     /* Parse prefix */
128     str = skip_space (str);
129     const char *pos = strchr (str, ':');
130     if (! pos || pos == str)
131         goto FAIL;
132     *prefix_out = talloc_strndup (ctx, str, pos - str);
133     if (! *prefix_out) {
134         err = ENOMEM;
135         goto FAIL;
136     }
137     ++pos;
138
139     /* Implement de-quoting compatible with make_boolean_term. */
140     if (*pos == '"') {
141         char *out = talloc_array (ctx, char, strlen (pos));
142         int closed = 0;
143         if (! out) {
144             err = ENOMEM;
145             goto FAIL;
146         }
147         *term_out = out;
148         /* Skip the opening quote, find the closing quote, and
149          * un-double doubled internal quotes. */
150         for (++pos; *pos; ) {
151             if (*pos == '"') {
152                 ++pos;
153                 if (*pos != '"') {
154                     /* Found the closing quote. */
155                     closed = 1;
156                     pos = skip_space (pos);
157                     break;
158                 }
159             }
160             *out++ = *pos++;
161         }
162         /* Did the term terminate without a closing quote or is there
163          * trailing text after the closing quote? */
164         if (!closed || *pos)
165             goto FAIL;
166         *out = '\0';
167     } else {
168         const char *start = pos;
169         /* Check for text after the boolean term. */
170         while (! is_unquoted_terminator (*pos))
171             ++pos;
172         if (*skip_space (pos)) {
173             err = EINVAL;
174             goto FAIL;
175         }
176         /* No trailing text; dup the string so the caller can free
177          * it. */
178         *term_out = talloc_strndup (ctx, start, pos - start);
179         if (! *term_out) {
180             err = ENOMEM;
181             goto FAIL;
182         }
183     }
184     return 0;
185
186  FAIL:
187     talloc_free (*prefix_out);
188     talloc_free (*term_out);
189     errno = err;
190     return -1;
191 }