apply fix from r34275 to BLI_dynstr_appendf(), dont chop off last character for ...
[blender-staging.git] / source / blender / blenlib / intern / BLI_dynstr.c
1 /**
2  * $Id$
3  *
4  * ***** BEGIN GPL LICENSE BLOCK *****
5  *
6  * This program is free software; you can redistribute it and/or
7  * modify it under the terms of the GNU General Public License
8  * as published by the Free Software Foundation; either version 2
9  * of the License, or (at your option) any later version.
10  *
11  * This program is distributed in the hope that it will be useful,
12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
14  * GNU General Public License for more details.
15  *
16  * You should have received a copy of the GNU General Public License
17  * along with this program; if not, write to the Free Software Foundation,
18  * Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
19  *
20  * The Original Code is Copyright (C) 2001-2002 by NaN Holding BV.
21  * All rights reserved.
22  *
23  * The Original Code is: all of this file.
24  *
25  * Contributor(s): none yet.
26  *
27  * ***** END GPL LICENSE BLOCK *****
28  * Dynamically sized string ADT
29  */
30
31 #include <stdarg.h>
32 #include <string.h>
33
34 #include "MEM_guardedalloc.h"
35 #include "BLI_blenlib.h"
36 #include "BLI_dynstr.h"
37
38 #ifdef _WIN32
39 #ifndef vsnprintf
40 #define vsnprintf _vsnprintf
41 #endif
42 #endif
43
44 #ifndef va_copy
45 # ifdef __va_copy
46 #  define va_copy(a,b) __va_copy(a,b)
47 # else /* !__va_copy */
48 #  define va_copy(a,b) ((a)=(b))
49 # endif /* __va_copy */
50 #endif /* va_copy */
51
52 /***/
53
54 typedef struct DynStrElem DynStrElem;
55 struct DynStrElem {
56         DynStrElem *next;
57         
58         char *str;
59 };
60
61 struct DynStr {
62         DynStrElem *elems, *last;
63         int curlen;
64 };
65
66 /***/
67
68 DynStr *BLI_dynstr_new(void) {
69         DynStr *ds= MEM_mallocN(sizeof(*ds), "DynStr");
70         ds->elems= ds->last= NULL;
71         ds->curlen= 0;
72         
73         return ds;
74 }
75
76 void BLI_dynstr_append(DynStr *ds, const char *cstr) {
77         DynStrElem *dse= malloc(sizeof(*dse));
78         int cstrlen= strlen(cstr);
79         
80         dse->str= malloc(cstrlen+1);
81         memcpy(dse->str, cstr, cstrlen+1);
82         dse->next= NULL;
83         
84         if (!ds->last)
85                 ds->last= ds->elems= dse;
86         else
87                 ds->last= ds->last->next= dse;
88
89         ds->curlen+= cstrlen;
90 }
91
92 void BLI_dynstr_nappend(DynStr *ds, const char *cstr, int len) {
93         DynStrElem *dse= malloc(sizeof(*dse));
94         int cstrlen= BLI_strnlen(cstr, len);
95
96         dse->str= malloc(cstrlen+1);
97         memcpy(dse->str, cstr, cstrlen);
98         dse->str[cstrlen] = '\0';
99         dse->next= NULL;
100
101         if (!ds->last)
102                 ds->last= ds->elems= dse;
103         else
104                 ds->last= ds->last->next= dse;
105
106         ds->curlen+= cstrlen;
107 }
108
109 void BLI_dynstr_vappendf(DynStr *ds, const char *format, va_list args)
110 {
111         char *message, fixedmessage[256];
112         int len= sizeof(fixedmessage);
113         const int maxlen= 65536;
114         int retval;
115
116         while(1) {
117                 va_list args_cpy;
118                 if(len == sizeof(fixedmessage))
119                         message= fixedmessage;
120                 else
121                         message= MEM_callocN(sizeof(char) * len, "BLI_dynstr_appendf");
122
123                 /* cant reuse the same args, so work on a copy */
124                 va_copy(args_cpy, args);
125                 retval= vsnprintf(message, len, format, args_cpy);
126                 va_end(args_cpy);
127
128                 if(retval == -1) {
129                         /* -1 means not enough space, but on windows it may also mean
130                          * there is a formatting error, so we impose a maximum length */
131                         if(message != fixedmessage)
132                                 MEM_freeN(message);
133                         message= NULL;
134
135                         len *= 2;
136                         if(len > maxlen) {
137                                 fprintf(stderr, "BLI_dynstr_append text too long or format error.\n");
138                                 break;
139                         }
140                 }
141                 else if(retval >= len) {
142                         /* in C99 the actual length required is returned */
143                         if(message != fixedmessage)
144                                 MEM_freeN(message);
145                         message= NULL;
146
147                         /* retval doesnt include \0 terminator */
148                         len= retval + 1;
149                 }
150                 else
151                         break;
152         }
153
154         if(message) {
155                 BLI_dynstr_append(ds, message);
156
157                 if(message != fixedmessage)
158                         MEM_freeN(message);
159         }
160 }
161
162 void BLI_dynstr_appendf(DynStr *ds, const char *format, ...)
163 {
164         va_list args;
165         char *message, fixedmessage[256];
166         int len= sizeof(fixedmessage);
167         const int maxlen= 65536;
168         int retval;
169
170         /* note that it's tempting to just call BLI_dynstr_vappendf here
171          * and avoid code duplication, that crashes on some system because
172          * va_start/va_end have to be called for each vsnprintf call */
173
174         while(1) {
175                 if(len == sizeof(fixedmessage))
176                         message= fixedmessage;
177                 else
178                         message= MEM_callocN(sizeof(char)*(len), "BLI_dynstr_appendf");
179
180                 va_start(args, format);
181                 retval= vsnprintf(message, len, format, args);
182                 va_end(args);
183
184                 if(retval == -1) {
185                         /* -1 means not enough space, but on windows it may also mean
186                          * there is a formatting error, so we impose a maximum length */
187                         if(message != fixedmessage)
188                                 MEM_freeN(message);
189                         message= NULL;
190
191                         len *= 2;
192                         if(len > maxlen) {
193                                 fprintf(stderr, "BLI_dynstr_append text too long or format error.\n");
194                                 break;
195                         }
196                 }
197                 else if(retval >= len) {
198                         /* in C99 the actual length required is returned */
199                         if(message != fixedmessage)
200                                 MEM_freeN(message);
201                         message= NULL;
202
203                         /* retval doesnt include \0 terminator */
204                         len= retval + 1;
205                 }
206                 else
207                         break;
208         }
209
210         if(message) {
211                 BLI_dynstr_append(ds, message);
212
213                 if(message != fixedmessage)
214                         MEM_freeN(message);
215         }
216 }
217
218 int BLI_dynstr_get_len(DynStr *ds) {
219         return ds->curlen;
220 }
221
222 char *BLI_dynstr_get_cstring(DynStr *ds) {
223         char *s, *rets= MEM_mallocN(ds->curlen+1, "dynstr_cstring");
224         DynStrElem *dse;
225         
226         for (s= rets, dse= ds->elems; dse; dse= dse->next) {
227                 int slen= strlen(dse->str);
228
229                 memcpy(s, dse->str, slen);
230
231                 s+= slen;
232         }
233         rets[ds->curlen]= '\0';
234         
235         return rets;
236 }
237
238 void BLI_dynstr_free(DynStr *ds) {
239         DynStrElem *dse;
240         
241         for (dse= ds->elems; dse; ) {
242                 DynStrElem *n= dse->next;
243                 
244                 free(dse->str);
245                 free(dse);
246                 
247                 dse= n;
248         }
249         
250         MEM_freeN(ds);
251 }