1*6236dae4SAndroid Build Coastguard Worker /***************************************************************************
2*6236dae4SAndroid Build Coastguard Worker * _ _ ____ _
3*6236dae4SAndroid Build Coastguard Worker * Project ___| | | | _ \| |
4*6236dae4SAndroid Build Coastguard Worker * / __| | | | |_) | |
5*6236dae4SAndroid Build Coastguard Worker * | (__| |_| | _ <| |___
6*6236dae4SAndroid Build Coastguard Worker * \___|\___/|_| \_\_____|
7*6236dae4SAndroid Build Coastguard Worker *
8*6236dae4SAndroid Build Coastguard Worker * Copyright (C) Daniel Stenberg, <[email protected]>, et al.
9*6236dae4SAndroid Build Coastguard Worker *
10*6236dae4SAndroid Build Coastguard Worker * This software is licensed as described in the file COPYING, which
11*6236dae4SAndroid Build Coastguard Worker * you should have received as part of this distribution. The terms
12*6236dae4SAndroid Build Coastguard Worker * are also available at https://curl.se/docs/copyright.html.
13*6236dae4SAndroid Build Coastguard Worker *
14*6236dae4SAndroid Build Coastguard Worker * You may opt to use, copy, modify, merge, publish, distribute and/or sell
15*6236dae4SAndroid Build Coastguard Worker * copies of the Software, and permit persons to whom the Software is
16*6236dae4SAndroid Build Coastguard Worker * furnished to do so, under the terms of the COPYING file.
17*6236dae4SAndroid Build Coastguard Worker *
18*6236dae4SAndroid Build Coastguard Worker * This software is distributed on an "AS IS" basis, WITHOUT WARRANTY OF ANY
19*6236dae4SAndroid Build Coastguard Worker * KIND, either express or implied.
20*6236dae4SAndroid Build Coastguard Worker *
21*6236dae4SAndroid Build Coastguard Worker * SPDX-License-Identifier: curl
22*6236dae4SAndroid Build Coastguard Worker *
23*6236dae4SAndroid Build Coastguard Worker ***************************************************************************/
24*6236dae4SAndroid Build Coastguard Worker
25*6236dae4SAndroid Build Coastguard Worker #include "curl_setup.h"
26*6236dae4SAndroid Build Coastguard Worker #include "dynbuf.h"
27*6236dae4SAndroid Build Coastguard Worker #include "curl_printf.h"
28*6236dae4SAndroid Build Coastguard Worker #ifdef BUILDING_LIBCURL
29*6236dae4SAndroid Build Coastguard Worker #include "curl_memory.h"
30*6236dae4SAndroid Build Coastguard Worker #endif
31*6236dae4SAndroid Build Coastguard Worker #include "memdebug.h"
32*6236dae4SAndroid Build Coastguard Worker
33*6236dae4SAndroid Build Coastguard Worker #define MIN_FIRST_ALLOC 32
34*6236dae4SAndroid Build Coastguard Worker
35*6236dae4SAndroid Build Coastguard Worker #define DYNINIT 0xbee51da /* random pattern */
36*6236dae4SAndroid Build Coastguard Worker
37*6236dae4SAndroid Build Coastguard Worker /*
38*6236dae4SAndroid Build Coastguard Worker * Init a dynbuf struct.
39*6236dae4SAndroid Build Coastguard Worker */
Curl_dyn_init(struct dynbuf * s,size_t toobig)40*6236dae4SAndroid Build Coastguard Worker void Curl_dyn_init(struct dynbuf *s, size_t toobig)
41*6236dae4SAndroid Build Coastguard Worker {
42*6236dae4SAndroid Build Coastguard Worker DEBUGASSERT(s);
43*6236dae4SAndroid Build Coastguard Worker DEBUGASSERT(toobig);
44*6236dae4SAndroid Build Coastguard Worker s->bufr = NULL;
45*6236dae4SAndroid Build Coastguard Worker s->leng = 0;
46*6236dae4SAndroid Build Coastguard Worker s->allc = 0;
47*6236dae4SAndroid Build Coastguard Worker s->toobig = toobig;
48*6236dae4SAndroid Build Coastguard Worker #ifdef DEBUGBUILD
49*6236dae4SAndroid Build Coastguard Worker s->init = DYNINIT;
50*6236dae4SAndroid Build Coastguard Worker #endif
51*6236dae4SAndroid Build Coastguard Worker }
52*6236dae4SAndroid Build Coastguard Worker
53*6236dae4SAndroid Build Coastguard Worker /*
54*6236dae4SAndroid Build Coastguard Worker * free the buffer and re-init the necessary fields. It does not touch the
55*6236dae4SAndroid Build Coastguard Worker * 'init' field and thus this buffer can be reused to add data to again.
56*6236dae4SAndroid Build Coastguard Worker */
Curl_dyn_free(struct dynbuf * s)57*6236dae4SAndroid Build Coastguard Worker void Curl_dyn_free(struct dynbuf *s)
58*6236dae4SAndroid Build Coastguard Worker {
59*6236dae4SAndroid Build Coastguard Worker DEBUGASSERT(s);
60*6236dae4SAndroid Build Coastguard Worker Curl_safefree(s->bufr);
61*6236dae4SAndroid Build Coastguard Worker s->leng = s->allc = 0;
62*6236dae4SAndroid Build Coastguard Worker }
63*6236dae4SAndroid Build Coastguard Worker
64*6236dae4SAndroid Build Coastguard Worker /*
65*6236dae4SAndroid Build Coastguard Worker * Store/append an chunk of memory to the dynbuf.
66*6236dae4SAndroid Build Coastguard Worker */
dyn_nappend(struct dynbuf * s,const unsigned char * mem,size_t len)67*6236dae4SAndroid Build Coastguard Worker static CURLcode dyn_nappend(struct dynbuf *s,
68*6236dae4SAndroid Build Coastguard Worker const unsigned char *mem, size_t len)
69*6236dae4SAndroid Build Coastguard Worker {
70*6236dae4SAndroid Build Coastguard Worker size_t indx = s->leng;
71*6236dae4SAndroid Build Coastguard Worker size_t a = s->allc;
72*6236dae4SAndroid Build Coastguard Worker size_t fit = len + indx + 1; /* new string + old string + zero byte */
73*6236dae4SAndroid Build Coastguard Worker
74*6236dae4SAndroid Build Coastguard Worker /* try to detect if there is rubbish in the struct */
75*6236dae4SAndroid Build Coastguard Worker DEBUGASSERT(s->init == DYNINIT);
76*6236dae4SAndroid Build Coastguard Worker DEBUGASSERT(s->toobig);
77*6236dae4SAndroid Build Coastguard Worker DEBUGASSERT(indx < s->toobig);
78*6236dae4SAndroid Build Coastguard Worker DEBUGASSERT(!s->leng || s->bufr);
79*6236dae4SAndroid Build Coastguard Worker DEBUGASSERT(a <= s->toobig);
80*6236dae4SAndroid Build Coastguard Worker DEBUGASSERT(!len || mem);
81*6236dae4SAndroid Build Coastguard Worker
82*6236dae4SAndroid Build Coastguard Worker if(fit > s->toobig) {
83*6236dae4SAndroid Build Coastguard Worker Curl_dyn_free(s);
84*6236dae4SAndroid Build Coastguard Worker return CURLE_TOO_LARGE;
85*6236dae4SAndroid Build Coastguard Worker }
86*6236dae4SAndroid Build Coastguard Worker else if(!a) {
87*6236dae4SAndroid Build Coastguard Worker DEBUGASSERT(!indx);
88*6236dae4SAndroid Build Coastguard Worker /* first invoke */
89*6236dae4SAndroid Build Coastguard Worker if(MIN_FIRST_ALLOC > s->toobig)
90*6236dae4SAndroid Build Coastguard Worker a = s->toobig;
91*6236dae4SAndroid Build Coastguard Worker else if(fit < MIN_FIRST_ALLOC)
92*6236dae4SAndroid Build Coastguard Worker a = MIN_FIRST_ALLOC;
93*6236dae4SAndroid Build Coastguard Worker else
94*6236dae4SAndroid Build Coastguard Worker a = fit;
95*6236dae4SAndroid Build Coastguard Worker }
96*6236dae4SAndroid Build Coastguard Worker else {
97*6236dae4SAndroid Build Coastguard Worker while(a < fit)
98*6236dae4SAndroid Build Coastguard Worker a *= 2;
99*6236dae4SAndroid Build Coastguard Worker if(a > s->toobig)
100*6236dae4SAndroid Build Coastguard Worker /* no point in allocating a larger buffer than this is allowed to use */
101*6236dae4SAndroid Build Coastguard Worker a = s->toobig;
102*6236dae4SAndroid Build Coastguard Worker }
103*6236dae4SAndroid Build Coastguard Worker
104*6236dae4SAndroid Build Coastguard Worker if(a != s->allc) {
105*6236dae4SAndroid Build Coastguard Worker /* this logic is not using Curl_saferealloc() to make the tool not have to
106*6236dae4SAndroid Build Coastguard Worker include that as well when it uses this code */
107*6236dae4SAndroid Build Coastguard Worker void *p = realloc(s->bufr, a);
108*6236dae4SAndroid Build Coastguard Worker if(!p) {
109*6236dae4SAndroid Build Coastguard Worker Curl_dyn_free(s);
110*6236dae4SAndroid Build Coastguard Worker return CURLE_OUT_OF_MEMORY;
111*6236dae4SAndroid Build Coastguard Worker }
112*6236dae4SAndroid Build Coastguard Worker s->bufr = p;
113*6236dae4SAndroid Build Coastguard Worker s->allc = a;
114*6236dae4SAndroid Build Coastguard Worker }
115*6236dae4SAndroid Build Coastguard Worker
116*6236dae4SAndroid Build Coastguard Worker if(len)
117*6236dae4SAndroid Build Coastguard Worker memcpy(&s->bufr[indx], mem, len);
118*6236dae4SAndroid Build Coastguard Worker s->leng = indx + len;
119*6236dae4SAndroid Build Coastguard Worker s->bufr[s->leng] = 0;
120*6236dae4SAndroid Build Coastguard Worker return CURLE_OK;
121*6236dae4SAndroid Build Coastguard Worker }
122*6236dae4SAndroid Build Coastguard Worker
123*6236dae4SAndroid Build Coastguard Worker /*
124*6236dae4SAndroid Build Coastguard Worker * Clears the string, keeps the allocation. This can also be called on a
125*6236dae4SAndroid Build Coastguard Worker * buffer that already was freed.
126*6236dae4SAndroid Build Coastguard Worker */
Curl_dyn_reset(struct dynbuf * s)127*6236dae4SAndroid Build Coastguard Worker void Curl_dyn_reset(struct dynbuf *s)
128*6236dae4SAndroid Build Coastguard Worker {
129*6236dae4SAndroid Build Coastguard Worker DEBUGASSERT(s);
130*6236dae4SAndroid Build Coastguard Worker DEBUGASSERT(s->init == DYNINIT);
131*6236dae4SAndroid Build Coastguard Worker DEBUGASSERT(!s->leng || s->bufr);
132*6236dae4SAndroid Build Coastguard Worker if(s->leng)
133*6236dae4SAndroid Build Coastguard Worker s->bufr[0] = 0;
134*6236dae4SAndroid Build Coastguard Worker s->leng = 0;
135*6236dae4SAndroid Build Coastguard Worker }
136*6236dae4SAndroid Build Coastguard Worker
137*6236dae4SAndroid Build Coastguard Worker /*
138*6236dae4SAndroid Build Coastguard Worker * Specify the size of the tail to keep (number of bytes from the end of the
139*6236dae4SAndroid Build Coastguard Worker * buffer). The rest will be dropped.
140*6236dae4SAndroid Build Coastguard Worker */
Curl_dyn_tail(struct dynbuf * s,size_t trail)141*6236dae4SAndroid Build Coastguard Worker CURLcode Curl_dyn_tail(struct dynbuf *s, size_t trail)
142*6236dae4SAndroid Build Coastguard Worker {
143*6236dae4SAndroid Build Coastguard Worker DEBUGASSERT(s);
144*6236dae4SAndroid Build Coastguard Worker DEBUGASSERT(s->init == DYNINIT);
145*6236dae4SAndroid Build Coastguard Worker DEBUGASSERT(!s->leng || s->bufr);
146*6236dae4SAndroid Build Coastguard Worker if(trail > s->leng)
147*6236dae4SAndroid Build Coastguard Worker return CURLE_BAD_FUNCTION_ARGUMENT;
148*6236dae4SAndroid Build Coastguard Worker else if(trail == s->leng)
149*6236dae4SAndroid Build Coastguard Worker return CURLE_OK;
150*6236dae4SAndroid Build Coastguard Worker else if(!trail) {
151*6236dae4SAndroid Build Coastguard Worker Curl_dyn_reset(s);
152*6236dae4SAndroid Build Coastguard Worker }
153*6236dae4SAndroid Build Coastguard Worker else {
154*6236dae4SAndroid Build Coastguard Worker memmove(&s->bufr[0], &s->bufr[s->leng - trail], trail);
155*6236dae4SAndroid Build Coastguard Worker s->leng = trail;
156*6236dae4SAndroid Build Coastguard Worker s->bufr[s->leng] = 0;
157*6236dae4SAndroid Build Coastguard Worker }
158*6236dae4SAndroid Build Coastguard Worker return CURLE_OK;
159*6236dae4SAndroid Build Coastguard Worker
160*6236dae4SAndroid Build Coastguard Worker }
161*6236dae4SAndroid Build Coastguard Worker
162*6236dae4SAndroid Build Coastguard Worker /*
163*6236dae4SAndroid Build Coastguard Worker * Appends a buffer with length.
164*6236dae4SAndroid Build Coastguard Worker */
Curl_dyn_addn(struct dynbuf * s,const void * mem,size_t len)165*6236dae4SAndroid Build Coastguard Worker CURLcode Curl_dyn_addn(struct dynbuf *s, const void *mem, size_t len)
166*6236dae4SAndroid Build Coastguard Worker {
167*6236dae4SAndroid Build Coastguard Worker DEBUGASSERT(s);
168*6236dae4SAndroid Build Coastguard Worker DEBUGASSERT(s->init == DYNINIT);
169*6236dae4SAndroid Build Coastguard Worker DEBUGASSERT(!s->leng || s->bufr);
170*6236dae4SAndroid Build Coastguard Worker return dyn_nappend(s, mem, len);
171*6236dae4SAndroid Build Coastguard Worker }
172*6236dae4SAndroid Build Coastguard Worker
173*6236dae4SAndroid Build Coastguard Worker /*
174*6236dae4SAndroid Build Coastguard Worker * Append a null-terminated string at the end.
175*6236dae4SAndroid Build Coastguard Worker */
Curl_dyn_add(struct dynbuf * s,const char * str)176*6236dae4SAndroid Build Coastguard Worker CURLcode Curl_dyn_add(struct dynbuf *s, const char *str)
177*6236dae4SAndroid Build Coastguard Worker {
178*6236dae4SAndroid Build Coastguard Worker size_t n;
179*6236dae4SAndroid Build Coastguard Worker DEBUGASSERT(str);
180*6236dae4SAndroid Build Coastguard Worker DEBUGASSERT(s);
181*6236dae4SAndroid Build Coastguard Worker DEBUGASSERT(s->init == DYNINIT);
182*6236dae4SAndroid Build Coastguard Worker DEBUGASSERT(!s->leng || s->bufr);
183*6236dae4SAndroid Build Coastguard Worker n = strlen(str);
184*6236dae4SAndroid Build Coastguard Worker return dyn_nappend(s, (unsigned char *)str, n);
185*6236dae4SAndroid Build Coastguard Worker }
186*6236dae4SAndroid Build Coastguard Worker
187*6236dae4SAndroid Build Coastguard Worker /*
188*6236dae4SAndroid Build Coastguard Worker * Append a string vprintf()-style
189*6236dae4SAndroid Build Coastguard Worker */
Curl_dyn_vaddf(struct dynbuf * s,const char * fmt,va_list ap)190*6236dae4SAndroid Build Coastguard Worker CURLcode Curl_dyn_vaddf(struct dynbuf *s, const char *fmt, va_list ap)
191*6236dae4SAndroid Build Coastguard Worker {
192*6236dae4SAndroid Build Coastguard Worker #ifdef BUILDING_LIBCURL
193*6236dae4SAndroid Build Coastguard Worker int rc;
194*6236dae4SAndroid Build Coastguard Worker DEBUGASSERT(s);
195*6236dae4SAndroid Build Coastguard Worker DEBUGASSERT(s->init == DYNINIT);
196*6236dae4SAndroid Build Coastguard Worker DEBUGASSERT(!s->leng || s->bufr);
197*6236dae4SAndroid Build Coastguard Worker DEBUGASSERT(fmt);
198*6236dae4SAndroid Build Coastguard Worker rc = Curl_dyn_vprintf(s, fmt, ap);
199*6236dae4SAndroid Build Coastguard Worker
200*6236dae4SAndroid Build Coastguard Worker if(!rc)
201*6236dae4SAndroid Build Coastguard Worker return CURLE_OK;
202*6236dae4SAndroid Build Coastguard Worker else if(rc == MERR_TOO_LARGE)
203*6236dae4SAndroid Build Coastguard Worker return CURLE_TOO_LARGE;
204*6236dae4SAndroid Build Coastguard Worker return CURLE_OUT_OF_MEMORY;
205*6236dae4SAndroid Build Coastguard Worker #else
206*6236dae4SAndroid Build Coastguard Worker char *str;
207*6236dae4SAndroid Build Coastguard Worker str = vaprintf(fmt, ap); /* this allocs a new string to append */
208*6236dae4SAndroid Build Coastguard Worker
209*6236dae4SAndroid Build Coastguard Worker if(str) {
210*6236dae4SAndroid Build Coastguard Worker CURLcode result = dyn_nappend(s, (unsigned char *)str, strlen(str));
211*6236dae4SAndroid Build Coastguard Worker free(str);
212*6236dae4SAndroid Build Coastguard Worker return result;
213*6236dae4SAndroid Build Coastguard Worker }
214*6236dae4SAndroid Build Coastguard Worker /* If we failed, we cleanup the whole buffer and return error */
215*6236dae4SAndroid Build Coastguard Worker Curl_dyn_free(s);
216*6236dae4SAndroid Build Coastguard Worker return CURLE_OUT_OF_MEMORY;
217*6236dae4SAndroid Build Coastguard Worker #endif
218*6236dae4SAndroid Build Coastguard Worker }
219*6236dae4SAndroid Build Coastguard Worker
220*6236dae4SAndroid Build Coastguard Worker /*
221*6236dae4SAndroid Build Coastguard Worker * Append a string printf()-style
222*6236dae4SAndroid Build Coastguard Worker */
Curl_dyn_addf(struct dynbuf * s,const char * fmt,...)223*6236dae4SAndroid Build Coastguard Worker CURLcode Curl_dyn_addf(struct dynbuf *s, const char *fmt, ...)
224*6236dae4SAndroid Build Coastguard Worker {
225*6236dae4SAndroid Build Coastguard Worker CURLcode result;
226*6236dae4SAndroid Build Coastguard Worker va_list ap;
227*6236dae4SAndroid Build Coastguard Worker DEBUGASSERT(s);
228*6236dae4SAndroid Build Coastguard Worker DEBUGASSERT(s->init == DYNINIT);
229*6236dae4SAndroid Build Coastguard Worker DEBUGASSERT(!s->leng || s->bufr);
230*6236dae4SAndroid Build Coastguard Worker va_start(ap, fmt);
231*6236dae4SAndroid Build Coastguard Worker result = Curl_dyn_vaddf(s, fmt, ap);
232*6236dae4SAndroid Build Coastguard Worker va_end(ap);
233*6236dae4SAndroid Build Coastguard Worker return result;
234*6236dae4SAndroid Build Coastguard Worker }
235*6236dae4SAndroid Build Coastguard Worker
236*6236dae4SAndroid Build Coastguard Worker /*
237*6236dae4SAndroid Build Coastguard Worker * Returns a pointer to the buffer.
238*6236dae4SAndroid Build Coastguard Worker */
Curl_dyn_ptr(const struct dynbuf * s)239*6236dae4SAndroid Build Coastguard Worker char *Curl_dyn_ptr(const struct dynbuf *s)
240*6236dae4SAndroid Build Coastguard Worker {
241*6236dae4SAndroid Build Coastguard Worker DEBUGASSERT(s);
242*6236dae4SAndroid Build Coastguard Worker DEBUGASSERT(s->init == DYNINIT);
243*6236dae4SAndroid Build Coastguard Worker DEBUGASSERT(!s->leng || s->bufr);
244*6236dae4SAndroid Build Coastguard Worker return s->bufr;
245*6236dae4SAndroid Build Coastguard Worker }
246*6236dae4SAndroid Build Coastguard Worker
247*6236dae4SAndroid Build Coastguard Worker /*
248*6236dae4SAndroid Build Coastguard Worker * Returns an unsigned pointer to the buffer.
249*6236dae4SAndroid Build Coastguard Worker */
Curl_dyn_uptr(const struct dynbuf * s)250*6236dae4SAndroid Build Coastguard Worker unsigned char *Curl_dyn_uptr(const struct dynbuf *s)
251*6236dae4SAndroid Build Coastguard Worker {
252*6236dae4SAndroid Build Coastguard Worker DEBUGASSERT(s);
253*6236dae4SAndroid Build Coastguard Worker DEBUGASSERT(s->init == DYNINIT);
254*6236dae4SAndroid Build Coastguard Worker DEBUGASSERT(!s->leng || s->bufr);
255*6236dae4SAndroid Build Coastguard Worker return (unsigned char *)s->bufr;
256*6236dae4SAndroid Build Coastguard Worker }
257*6236dae4SAndroid Build Coastguard Worker
258*6236dae4SAndroid Build Coastguard Worker /*
259*6236dae4SAndroid Build Coastguard Worker * Returns the length of the buffer.
260*6236dae4SAndroid Build Coastguard Worker */
Curl_dyn_len(const struct dynbuf * s)261*6236dae4SAndroid Build Coastguard Worker size_t Curl_dyn_len(const struct dynbuf *s)
262*6236dae4SAndroid Build Coastguard Worker {
263*6236dae4SAndroid Build Coastguard Worker DEBUGASSERT(s);
264*6236dae4SAndroid Build Coastguard Worker DEBUGASSERT(s->init == DYNINIT);
265*6236dae4SAndroid Build Coastguard Worker DEBUGASSERT(!s->leng || s->bufr);
266*6236dae4SAndroid Build Coastguard Worker return s->leng;
267*6236dae4SAndroid Build Coastguard Worker }
268*6236dae4SAndroid Build Coastguard Worker
269*6236dae4SAndroid Build Coastguard Worker /*
270*6236dae4SAndroid Build Coastguard Worker * Set a new (smaller) length.
271*6236dae4SAndroid Build Coastguard Worker */
Curl_dyn_setlen(struct dynbuf * s,size_t set)272*6236dae4SAndroid Build Coastguard Worker CURLcode Curl_dyn_setlen(struct dynbuf *s, size_t set)
273*6236dae4SAndroid Build Coastguard Worker {
274*6236dae4SAndroid Build Coastguard Worker DEBUGASSERT(s);
275*6236dae4SAndroid Build Coastguard Worker DEBUGASSERT(s->init == DYNINIT);
276*6236dae4SAndroid Build Coastguard Worker DEBUGASSERT(!s->leng || s->bufr);
277*6236dae4SAndroid Build Coastguard Worker if(set > s->leng)
278*6236dae4SAndroid Build Coastguard Worker return CURLE_BAD_FUNCTION_ARGUMENT;
279*6236dae4SAndroid Build Coastguard Worker s->leng = set;
280*6236dae4SAndroid Build Coastguard Worker s->bufr[s->leng] = 0;
281*6236dae4SAndroid Build Coastguard Worker return CURLE_OK;
282*6236dae4SAndroid Build Coastguard Worker }
283