1 /* Main function for test programs.
2    Copyright (C) 2016-2022 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, see
17    <https://www.gnu.org/licenses/>.  */
18 
19 /* This file should be included from test cases.  It will define a
20    main function which provides the test wrapper.
21 
22    It assumes that the test case defines a function
23 
24      int do_test (void);
25 
26    and arranges for that function being called under the test wrapper.
27    The do_test function should return 0 to indicate a passing test, 1
28    to indicate a failing test, or 77 to indicate an unsupported test.
29    Other result values could be used to indicate a failing test, but
30    the result of the expression is passed to exit and exit only
31    returns the lower 8 bits of its input.  A non-zero return with some
32    values could cause a test to incorrectly be considered passing when
33    it really failed.  For this reason, the function should always
34    return 0 (EXIT_SUCCESS), 1 (EXIT_FAILURE), or 77
35    (EXIT_UNSUPPORTED).
36 
37    The test function may print out diagnostic or warning messages as well
38    as messages about failures.  These messages should be printed to stdout
39    and not stderr so that the output is properly ordered with respect to
40    the rest of the glibc testsuite run output.
41 
42    Several preprocessors macros can be defined before including this
43    file.
44 
45    The name of the do_test function can be changed with the
46    TEST_FUNCTION macro.  It must expand to the desired function name.
47 
48    If the test case needs access to command line parameters, it must
49    define the TEST_FUNCTION_ARGV macro with the name of the test
50    function.  It must have the following type:
51 
52      int TEST_FUNCTION_ARGV (int argc, char **argv);
53 
54    This overrides the do_test default function and is incompatible
55    with the TEST_FUNCTION macro.
56 
57    If PREPARE is defined, it must expand to the name of a function of
58    the type
59 
60      void PREPARE (int argc, char **);
61 
62    This function will be called early, after parsing the command line,
63    but before running the test, in the parent process which acts as
64    the test supervisor.
65 
66    If CLEANUP_HANDLER is defined, it must expand to the name of a
67    function of the type
68 
69      void CLEANUP_HANDLER (void);
70 
71    This function will be called from the timeout (SIGALRM) signal
72    handler.
73 
74    If EXPECTED_SIGNAL is defined, it must expanded to a constant which
75    denotes the expected signal number.
76 
77    If EXPECTED_STATUS is defined, it must expand to the expected exit
78    status.
79 
80    If TIMEOUT is defined, it must be positive constant.  It overrides
81    the default test timeout and is measured in seconds.
82 
83    If TEST_NO_MALLOPT is defined, the test wrapper will not call
84    mallopt.
85 
86    Custom command line handling can be implemented by defining the
87    CMDLINE_OPTION macro (after including the <getopt.h> header; this
88    requires _GNU_SOURCE to be defined).  This macro must expand to a
89    to a comma-separated list of braced initializers for struct option
90    from <getopt.h>, with a trailing comma.  CMDLINE_PROCESS can be
91    defined as the name of a function which is called to process these
92    options.  The function is passed the option character/number and
93    has this type:
94 
95      void CMDLINE_PROCESS (int);
96 
97    If the program also to process custom default short command line
98    argument (similar to getopt) it must define CMDLINE_OPTSTRING
99    with the expected options (for instance "vb").
100 */
101 
102 #include <support/test-driver.h>
103 
104 #include <string.h>
105 
106 int
main(int argc,char ** argv)107 main (int argc, char **argv)
108 {
109   struct test_config test_config;
110   memset (&test_config, 0, sizeof (test_config));
111 
112 #ifdef PREPARE
113   test_config.prepare_function = (PREPARE);
114 #endif
115 
116 #if defined (TEST_FUNCTION) && defined (TEST_FUNCTON_ARGV)
117 # error TEST_FUNCTION and TEST_FUNCTION_ARGV cannot be defined at the same time
118 #endif
119 #ifdef RUN_COMMAND_MODE
120   test_config.run_command_mode = 1;
121 #elif defined (TEST_FUNCTION)
122   test_config.test_function = TEST_FUNCTION;
123 #elif defined (TEST_FUNCTION_ARGV)
124   test_config.test_function_argv = TEST_FUNCTION_ARGV;
125 #else
126   test_config.test_function = do_test;
127 #endif
128 
129 #ifdef CLEANUP_HANDLER
130   test_config.cleanup_function = CLEANUP_HANDLER;
131 #endif
132 
133 #ifdef EXPECTED_SIGNAL
134   test_config.expected_signal = (EXPECTED_SIGNAL);
135 #endif
136 
137 #ifdef EXPECTED_STATUS
138   test_config.expected_status = (EXPECTED_STATUS);
139 #endif
140 
141 #ifdef TEST_NO_MALLOPT
142   test_config.no_mallopt = 1;
143 #endif
144 
145 #ifdef TEST_NO_SETVBUF
146   test_config.no_setvbuf = 1;
147 #endif
148 
149 #ifdef TIMEOUT
150   test_config.timeout = TIMEOUT;
151 #endif
152 
153 #ifdef CMDLINE_OPTIONS
154   struct option options[] =
155     {
156       CMDLINE_OPTIONS
157       TEST_DEFAULT_OPTIONS
158     };
159   test_config.options = &options;
160 #endif
161 #ifdef CMDLINE_PROCESS
162   test_config.cmdline_function = CMDLINE_PROCESS;
163 #endif
164 #ifdef CMDLINE_OPTSTRING
165   test_config.optstring = "+" CMDLINE_OPTSTRING;
166 #else
167   test_config.optstring = "+";
168 #endif
169 
170   return support_test_main (argc, argv, &test_config);
171 }
172