Handle negative length and eliminate a condition that is always true.
[asterisk/asterisk.git] / main / strings.c
1 /*
2  * Asterisk -- An open source telephony toolkit.
3  *
4  * Copyright (C) 2008, Digium, Inc.
5  *
6  * Tilghman Lesher <tlesher@digium.com>
7  *
8  * See http://www.asterisk.org for more information about
9  * the Asterisk project. Please do not directly contact
10  * any of the maintainers of this project for assistance;
11  * the project provides a web site, mailing lists and IRC
12  * channels for your use.
13  *
14  * This program is free software, distributed under the terms of
15 * the GNU General Public License Version 2. See the LICENSE file
16  * at the top of the source tree.
17  */
18
19 /*! \file
20  *
21  * \brief String manipulation API
22  *
23  * \author Tilghman Lesher <tilghman@digium.com>
24  */
25
26 /*** MAKEOPTS
27 <category name="MENUSELECT_CFLAGS" displayname="Compiler Flags" positive_output="yes" remove_on_change=".lastclean">
28         <member name="DEBUG_OPAQUE" displayname="Change ast_str internals to detect improper usage">
29                 <defaultenabled>yes</defaultenabled>
30         </member>
31 </category>
32  ***/
33
34 #include "asterisk.h"
35
36 ASTERISK_FILE_VERSION(__FILE__, "$Revision$")
37
38 #include "asterisk/strings.h"
39 #include "asterisk/pbx.h"
40
41 /*!
42  * core handler for dynamic strings.
43  * This is not meant to be called directly, but rather through the
44  * various wrapper macros
45  *      ast_str_set(...)
46  *      ast_str_append(...)
47  *      ast_str_set_va(...)
48  *      ast_str_append_va(...)
49  */
50
51 int __ast_str_helper(struct ast_str **buf, size_t max_len,
52         int append, const char *fmt, va_list ap)
53 {
54         int res, need;
55         int offset = (append && (*buf)->__AST_STR_LEN) ? (*buf)->__AST_STR_USED : 0;
56         va_list aq;
57
58         do {
59                 if (max_len < 0) {
60                         max_len = (*buf)->__AST_STR_LEN;        /* don't exceed the allocated space */
61                 }
62                 /*
63                  * Ask vsnprintf how much space we need. Remember that vsnprintf
64                  * does not count the final '\0' so we must add 1.
65                  */
66                 va_copy(aq, ap);
67                 res = vsnprintf((*buf)->__AST_STR_STR + offset, (*buf)->__AST_STR_LEN - offset, fmt, aq);
68
69                 need = res + offset + 1;
70                 /*
71                  * If there is not enough space and we are below the max length,
72                  * reallocate the buffer and return a message telling to retry.
73                  */
74                 if (need > (*buf)->__AST_STR_LEN && (max_len == 0 || (*buf)->__AST_STR_LEN < max_len) ) {
75                         if (max_len && max_len < need) {        /* truncate as needed */
76                                 need = max_len;
77                         } else if (max_len == 0) {      /* if unbounded, give more room for next time */
78                                 need += 16 + need / 4;
79                         }
80                         if (0) {        /* debugging */
81                                 ast_verbose("extend from %d to %d\n", (int)(*buf)->__AST_STR_LEN, need);
82                         }
83                         if (ast_str_make_space(buf, need)) {
84                                 ast_verbose("failed to extend from %d to %d\n", (int)(*buf)->__AST_STR_LEN, need);
85                                 return AST_DYNSTR_BUILD_FAILED;
86                         }
87                         (*buf)->__AST_STR_STR[offset] = '\0';   /* Truncate the partial write. */
88
89                         /* Restart va_copy before calling vsnprintf() again. */
90                         va_end(aq);
91                         continue;
92                 }
93                 break;
94         } while (1);
95         /* update space used, keep in mind the truncation */
96         (*buf)->__AST_STR_USED = (res + offset > (*buf)->__AST_STR_LEN) ? (*buf)->__AST_STR_LEN : res + offset;
97
98         return res;
99 }
100
101 void ast_str_substitute_variables(struct ast_str **buf, size_t maxlen, struct ast_channel *chan, const char *template)
102 {
103         int first = 1;
104         do {
105                 ast_str_make_space(buf, maxlen ? maxlen :
106                         (first ? strlen(template) * 2 : (*buf)->__AST_STR_LEN * 2));
107                 pbx_substitute_variables_helper_full(chan, NULL, template, (*buf)->__AST_STR_STR, (*buf)->__AST_STR_LEN - 1, &((*buf)->__AST_STR_USED));
108                 first = 0;
109         } while (maxlen == 0 && (*buf)->__AST_STR_LEN - 5 < (*buf)->__AST_STR_USED);
110 }
111
112 char *__ast_str_helper2(struct ast_str **buf, size_t maxlen, const char *src, size_t maxsrc, int append, int escapecommas)
113 {
114         int dynamic = 0;
115         char *ptr = append ? &((*buf)->__AST_STR_STR[(*buf)->__AST_STR_USED]) : (*buf)->__AST_STR_STR;
116
117         if (maxlen < 1) {
118                 if (maxlen == 0) {
119                         dynamic = 1;
120                 }
121                 maxlen = (*buf)->__AST_STR_LEN;
122         }
123
124         while (*src && maxsrc && maxlen && (!escapecommas || (maxlen - 1))) {
125                 if (escapecommas && (*src == '\\' || *src == ',')) {
126                         *ptr++ = '\\';
127                         maxlen--;
128                         (*buf)->__AST_STR_USED++;
129                 }
130                 *ptr++ = *src++;
131                 maxsrc--;
132                 maxlen--;
133                 (*buf)->__AST_STR_USED++;
134
135                 if (dynamic && (!maxlen || (escapecommas && !(maxlen - 1)))) {
136                         char *oldbase = (*buf)->__AST_STR_STR;
137                         size_t old = (*buf)->__AST_STR_LEN;
138                         if (ast_str_make_space(buf, (*buf)->__AST_STR_LEN * 2)) {
139                                 /* If the buffer can't be extended, end it. */
140                                 break;
141                         }
142                         /* What we extended the buffer by */
143                         maxlen = old;
144
145                         ptr += (*buf)->__AST_STR_STR - oldbase;
146                 }
147         }
148         if (__builtin_expect(!(maxsrc && maxlen), 0)) {
149                 ptr--;
150         }
151         *ptr = '\0';
152         (*buf)->__AST_STR_USED--;
153         return (*buf)->__AST_STR_STR;
154 }
155