diff options
Diffstat (limited to 'gnulib/tests/test-dprintf-posix2.c')
m--------- | gnulib | 0 | ||||
-rw-r--r-- | gnulib/tests/test-dprintf-posix2.c | 112 |
2 files changed, 112 insertions, 0 deletions
diff --git a/gnulib b/gnulib deleted file mode 160000 -Subproject 443bc5ffcf7429e557f4a371b0661abe98ddbc1 diff --git a/gnulib/tests/test-dprintf-posix2.c b/gnulib/tests/test-dprintf-posix2.c new file mode 100644 index 0000000..5172499 --- /dev/null +++ b/gnulib/tests/test-dprintf-posix2.c @@ -0,0 +1,112 @@ +/* Test of POSIX compatible dprintf() function. + Copyright (C) 2009-2011 Free Software Foundation, Inc. + + This program is free software: you can redistribute it and/or modify + it under the terms of the GNU General Public License as published by + the Free Software Foundation; either version 3 of the License, or + (at your option) any later version. + + This program 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 General Public License for more details. + + You should have received a copy of the GNU General Public License + along with this program. If not, see <http://www.gnu.org/licenses/>. */ + +/* Written by Bruno Haible <bruno@clisp.org>, 2009. */ + +#include <config.h> + +#include <stdio.h> + +#include <stdlib.h> +#include <string.h> +#include <unistd.h> +#include <errno.h> + +#if HAVE_GETRLIMIT && HAVE_SETRLIMIT +# include <sys/types.h> +# include <sys/time.h> +# include <sys/resource.h> +#endif + +#include "resource-ext.h" + +/* Test against a memory leak in the fprintf replacement. */ + +/* Number of iterations across the loop. */ +#define NUM_ROUNDS 1000 + +/* Number of bytes that are allowed to escape per round. */ +#define MAX_ALLOC_ROUND 10000 + +/* Number of bytes that are allowed to escape in total. + This should be at least 10 MB, since it includes the normal memory + or address space of the test program. */ +#define MAX_ALLOC_TOTAL (NUM_ROUNDS * MAX_ALLOC_ROUND) + +int +main (int argc, char *argv[]) +{ + uintptr_t initial_rusage_as; + int arg; + int result; + + /* Limit the amount of malloc()ed memory to MAX_ALLOC_TOTAL or less. */ + + /* On AIX systems, malloc() is limited by RLIMIT_DATA. */ +#if HAVE_GETRLIMIT && HAVE_SETRLIMIT && defined RLIMIT_DATA + { + struct rlimit limit; + + if (getrlimit (RLIMIT_DATA, &limit) >= 0) + { + if (limit.rlim_max == RLIM_INFINITY || limit.rlim_max > MAX_ALLOC_TOTAL) + limit.rlim_max = MAX_ALLOC_TOTAL; + limit.rlim_cur = limit.rlim_max; + (void) setrlimit (RLIMIT_DATA, &limit); + } + } +#endif + /* On all systems except AIX and OpenBSD, malloc() is limited by RLIMIT_AS. + On some systems, setrlimit of RLIMIT_AS doesn't work but get_rusage_as () + does. Allow the address space size to grow by at most MAX_ALLOC_TOTAL. */ + initial_rusage_as = get_rusage_as (); +#if !defined _AIX + if (initial_rusage_as == 0) + return 77; +#endif + + arg = atoi (argv[1]); + if (arg == 0) + { + void *memory = malloc (MAX_ALLOC_TOTAL); + if (memory == NULL) + return 1; + memset (memory, 17, MAX_ALLOC_TOTAL); + result = 78; + } + else + { + /* Perform the test and test whether it triggers a permanent memory + allocation of more than MAX_ALLOC_TOTAL bytes. */ + int repeat; + + for (repeat = 0; repeat < NUM_ROUNDS; repeat++) + { + /* This may produce a temporary memory allocation of 11000 bytes. + but should not result in a permanent memory allocation. */ + if (dprintf (STDOUT_FILENO, "%011000d\n", 17) == -1 + && errno == ENOMEM) + return 1; + } + + result = 0; + } + + if (get_rusage_as () > initial_rusage_as + MAX_ALLOC_TOTAL) + return 1; + + return result; +} |