]>
Commit | Line | Data |
---|---|---|
6cc8844f | 1 | /* Copyright (C) 1995,1997,1999-2002,2004,2006,2008,2009 |
dff37515 UD |
2 | Free Software Foundation, Inc. |
3 | This file is part of the GNU C Library. | |
4 | ||
5 | The GNU C Library is free software; you can redistribute it and/or | |
6 | modify it under the terms of the GNU Lesser General Public | |
7 | License as published by the Free Software Foundation; either | |
8 | version 2.1 of the License, or (at your option) any later version. | |
9 | ||
10 | The GNU C Library 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 GNU | |
13 | Lesser General Public License for more details. | |
14 | ||
15 | You should have received a copy of the GNU Lesser General Public | |
16 | License along with the GNU C Library; if not, write to the Free | |
17 | Software Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA | |
18 | 02111-1307 USA. | |
19 | ||
20 | As a special exception, if you link the code in this file with | |
21 | files compiled with a GNU compiler to produce an executable, | |
22 | that does not cause the resulting executable to be covered by | |
23 | the GNU Lesser General Public License. This exception does not | |
24 | however invalidate any other reasons why the executable file | |
25 | might be covered by the GNU Lesser General Public License. | |
26 | This exception applies to code released by its copyright holders | |
27 | in files containing the exception. */ | |
28 | ||
29 | #include <malloc.h> | |
30 | #include <string.h> | |
31 | #include <stdio.h> | |
32 | #include <stdio_ext.h> | |
33 | #include "../libio/libioP.h" | |
34 | #include "../libio/strfile.h" | |
35 | ||
36 | int | |
37 | __vasprintf_chk (char **result_ptr, int flags, const char *format, | |
38 | va_list args) | |
39 | { | |
40 | /* Initial size of the buffer to be used. Will be doubled each time an | |
41 | overflow occurs. */ | |
42 | const _IO_size_t init_string_size = 100; | |
43 | char *string; | |
44 | _IO_strfile sf; | |
45 | int ret; | |
46 | _IO_size_t needed; | |
47 | _IO_size_t allocated; | |
48 | /* No need to clear the memory here (unlike for open_memstream) since | |
49 | we know we will never seek on the stream. */ | |
50 | string = (char *) malloc (init_string_size); | |
51 | if (string == NULL) | |
52 | return -1; | |
53 | #ifdef _IO_MTSAFE_IO | |
54 | sf._sbf._f._lock = NULL; | |
55 | #endif | |
6cc8844f UD |
56 | _IO_no_init (&sf._sbf._f, _IO_USER_LOCK, -1, NULL, NULL); |
57 | _IO_JUMPS (&sf._sbf) = &_IO_str_jumps; | |
dff37515 UD |
58 | _IO_str_init_static_internal (&sf, string, init_string_size, string); |
59 | sf._sbf._f._flags &= ~_IO_USER_BUF; | |
60 | sf._s._allocate_buffer = (_IO_alloc_type) malloc; | |
61 | sf._s._free_buffer = (_IO_free_type) free; | |
62 | ||
63 | /* For flags > 0 (i.e. __USE_FORTIFY_LEVEL > 1) request that %n | |
64 | can only come from read-only format strings. */ | |
65 | if (flags > 0) | |
66 | sf._sbf._f._flags2 |= _IO_FLAGS2_FORTIFY; | |
67 | ||
68 | ret = INTUSE(_IO_vfprintf) (&sf._sbf._f, format, args); | |
69 | if (ret < 0) | |
70 | { | |
71 | free (sf._sbf._f._IO_buf_base); | |
72 | return ret; | |
73 | } | |
74 | /* Only use realloc if the size we need is of the same (binary) | |
75 | order of magnitude then the memory we allocated. */ | |
76 | needed = sf._sbf._f._IO_write_ptr - sf._sbf._f._IO_write_base + 1; | |
77 | allocated = sf._sbf._f._IO_write_end - sf._sbf._f._IO_write_base; | |
78 | if ((allocated >> 1) <= needed) | |
79 | *result_ptr = (char *) realloc (sf._sbf._f._IO_buf_base, needed); | |
80 | else | |
81 | { | |
82 | *result_ptr = (char *) malloc (needed); | |
83 | if (*result_ptr != NULL) | |
84 | { | |
85 | memcpy (*result_ptr, sf._sbf._f._IO_buf_base, needed - 1); | |
86 | free (sf._sbf._f._IO_buf_base); | |
87 | } | |
88 | else | |
89 | /* We have no choice, use the buffer we already have. */ | |
90 | *result_ptr = (char *) realloc (sf._sbf._f._IO_buf_base, needed); | |
91 | } | |
92 | if (*result_ptr == NULL) | |
93 | *result_ptr = sf._sbf._f._IO_buf_base; | |
94 | (*result_ptr)[needed - 1] = '\0'; | |
95 | return ret; | |
96 | } | |
97 | libc_hidden_def (__vasprintf_chk) |