1 /* Test of POSIX compatible dprintf() function.
2 Copyright (C) 2009-2021 Free Software Foundation, Inc.
3
4 This program is free software: you can redistribute it and/or modify
5 it under the terms of the GNU General Public License as published by
6 the Free Software Foundation; either version 3 of the License, or
7 (at your option) any later version.
8
9 This program is distributed in the hope that it will be useful,
10 but WITHOUT ANY WARRANTY; without even the implied warranty of
11 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
12 GNU General Public License for more details.
13
14 You should have received a copy of the GNU General Public License
15 along with this program. If not, see <https://www.gnu.org/licenses/>. */
16
17 /* Written by Bruno Haible <bruno@clisp.org>, 2009. */
18
19 #include <config.h>
20
21 #include <stdio.h>
22
23 #include <stdlib.h>
24 #include <string.h>
25 #include <unistd.h>
26 #include <errno.h>
27
28 #if HAVE_GETRLIMIT && HAVE_SETRLIMIT
29 # include <sys/types.h>
30 # include <sys/time.h>
31 # include <sys/resource.h>
32 #endif
33
34 #include "qemu.h"
35 #include "resource-ext.h"
36
37 /* Test against a memory leak in the fprintf replacement. */
38
39 /* Number of iterations across the loop. */
40 #define NUM_ROUNDS 1000
41
42 /* Number of bytes that are allowed to escape per round. */
43 #define MAX_ALLOC_ROUND 10000
44
45 /* Number of bytes that are allowed to escape in total.
46 This should be at least 10 MB, since it includes the normal memory
47 or address space of the test program. */
48 #define MAX_ALLOC_TOTAL (NUM_ROUNDS * MAX_ALLOC_ROUND)
49
50 int
main(int argc,char * argv[])51 main (int argc, char *argv[])
52 {
53 uintptr_t initial_rusage_as;
54 int arg;
55 int result;
56
57 if (is_running_under_qemu_user ())
58 {
59 fprintf (stderr, "Skipping test: cannot trust address space size when running under QEMU\n");
60 return 79;
61 }
62
63 /* Limit the amount of malloc()ed memory to MAX_ALLOC_TOTAL or less. */
64
65 /* On AIX systems, malloc() is limited by RLIMIT_DATA. */
66 #if HAVE_GETRLIMIT && HAVE_SETRLIMIT && defined RLIMIT_DATA
67 {
68 struct rlimit limit;
69
70 if (getrlimit (RLIMIT_DATA, &limit) >= 0)
71 {
72 if (limit.rlim_max == RLIM_INFINITY || limit.rlim_max > MAX_ALLOC_TOTAL)
73 limit.rlim_max = MAX_ALLOC_TOTAL;
74 limit.rlim_cur = limit.rlim_max;
75 (void) setrlimit (RLIMIT_DATA, &limit);
76 }
77 }
78 #endif
79 /* On all systems except AIX and OpenBSD, malloc() is limited by RLIMIT_AS.
80 On some systems, setrlimit of RLIMIT_AS doesn't work but get_rusage_as ()
81 does. Allow the address space size to grow by at most MAX_ALLOC_TOTAL. */
82 initial_rusage_as = get_rusage_as ();
83 #if !defined _AIX
84 if (initial_rusage_as == 0)
85 return 77;
86 #endif
87
88 arg = atoi (argv[1]);
89 if (arg == 0)
90 {
91 void *memory = malloc (MAX_ALLOC_TOTAL);
92 if (memory == NULL)
93 return 1;
94 memset (memory, 17, MAX_ALLOC_TOTAL);
95 result = 80;
96 }
97 else
98 {
99 /* Perform the test and test whether it triggers a permanent memory
100 allocation of more than MAX_ALLOC_TOTAL bytes. */
101 int repeat;
102
103 for (repeat = 0; repeat < NUM_ROUNDS; repeat++)
104 {
105 /* This may produce a temporary memory allocation of 11000 bytes.
106 but should not result in a permanent memory allocation. */
107 if (dprintf (STDOUT_FILENO, "%011000d\n", 17) == -1
108 && errno == ENOMEM)
109 return 1;
110 }
111
112 result = 0;
113 }
114
115 if (get_rusage_as () > initial_rusage_as + MAX_ALLOC_TOTAL)
116 return 1;
117
118 return result;
119 }
120