mirror of
https://sourceware.org/git/glibc.git
synced 2024-11-30 16:50:07 +00:00
383bd1c503
2001-12-06 Ulrich Drepper <drepper@redhat.com> * libio/vasprintf.c (_IO_vasprintf): Free buffer on failure. * assert/assert.c: Check result of __asprintf call and don't use string if it failed. * assert/assert-perr.c: Likewise. * inet/rcmd.c: Likewise. * locale/programs/localedef.c (main): Check result of construct_output_path and exit if it failed. (construct_output_path): Check result of asprintf and mkdir calls and fail if they failed. * posix/getopt.c: Check result of __asprintf calls and fail if they failed. Patch by Dmitry V. Levin <ldv@alt-linux.org>.
94 lines
3.3 KiB
C
94 lines
3.3 KiB
C
/* Copyright (C) 1995, 1997, 1999, 2000, 2001 Free Software Foundation, Inc.
|
|
This file is part of the GNU C Library.
|
|
|
|
The GNU C Library is free software; you can redistribute it and/or
|
|
modify it under the terms of the GNU Lesser General Public
|
|
License as published by the Free Software Foundation; either
|
|
version 2.1 of the License, or (at your option) any later version.
|
|
|
|
The GNU C Library is distributed in the hope that it will be useful,
|
|
but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
|
|
Lesser General Public License for more details.
|
|
|
|
You should have received a copy of the GNU Lesser General Public
|
|
License along with the GNU C Library; if not, write to the Free
|
|
Software Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA
|
|
02111-1307 USA.
|
|
|
|
As a special exception, if you link the code in this file with
|
|
files compiled with a GNU compiler to produce an executable,
|
|
that does not cause the resulting executable to be covered by
|
|
the GNU Lesser General Public License. This exception does not
|
|
however invalidate any other reasons why the executable file
|
|
might be covered by the GNU Lesser General Public License.
|
|
This exception applies to code released by its copyright holders
|
|
in files containing the exception. */
|
|
|
|
#include <malloc.h>
|
|
#include <string.h>
|
|
#include "libioP.h"
|
|
#include "stdio.h"
|
|
#include <stdio_ext.h>
|
|
#include "strfile.h"
|
|
|
|
int
|
|
_IO_vasprintf (result_ptr, format, args)
|
|
char **result_ptr;
|
|
const char *format;
|
|
_IO_va_list args;
|
|
{
|
|
/* Initial size of the buffer to be used. Will be doubled each time an
|
|
overflow occurs. */
|
|
const _IO_size_t init_string_size = 100;
|
|
char *string;
|
|
_IO_strfile sf;
|
|
int ret;
|
|
_IO_size_t needed;
|
|
_IO_size_t allocated;
|
|
string = (char *) malloc (init_string_size);
|
|
if (string == NULL)
|
|
return -1;
|
|
#ifdef _IO_MTSAFE_IO
|
|
sf._sbf._f._lock = NULL;
|
|
#endif
|
|
_IO_no_init ((_IO_FILE *) &sf._sbf, _IO_USER_LOCK, -1, NULL, NULL);
|
|
_IO_JUMPS ((struct _IO_FILE_plus *) &sf._sbf) = &_IO_str_jumps;
|
|
_IO_str_init_static (&sf, string, init_string_size, string);
|
|
sf._sbf._f._flags &= ~_IO_USER_BUF;
|
|
sf._s._allocate_buffer = (_IO_alloc_type) malloc;
|
|
sf._s._free_buffer = (_IO_free_type) free;
|
|
ret = _IO_vfprintf (&sf._sbf._f, format, args);
|
|
if (ret < 0)
|
|
{
|
|
free (sf._sbf._f._IO_buf_base);
|
|
return ret;
|
|
}
|
|
/* Only use realloc if the size we need is of the same order of
|
|
magnitude then the memory we allocated. */
|
|
needed = sf._sbf._f._IO_write_ptr - sf._sbf._f._IO_write_base + 1;
|
|
allocated = sf._sbf._f._IO_write_end - sf._sbf._f._IO_write_base;
|
|
if ((allocated << 1) <= needed)
|
|
*result_ptr = (char *) realloc (sf._sbf._f._IO_buf_base, needed);
|
|
else
|
|
{
|
|
*result_ptr = (char *) malloc (needed);
|
|
if (*result_ptr != NULL)
|
|
{
|
|
memcpy (*result_ptr, sf._sbf._f._IO_buf_base, needed - 1);
|
|
free (sf._sbf._f._IO_buf_base);
|
|
}
|
|
else
|
|
/* We have no choice, use the buffer we already have. */
|
|
*result_ptr = (char *) realloc (sf._sbf._f._IO_buf_base, needed);
|
|
}
|
|
if (*result_ptr == NULL)
|
|
*result_ptr = sf._sbf._f._IO_buf_base;
|
|
(*result_ptr)[needed - 1] = '\0';
|
|
return ret;
|
|
}
|
|
|
|
#ifdef weak_alias
|
|
weak_alias (_IO_vasprintf, vasprintf)
|
|
#endif
|