Remove the need for the custom SPHINXBUILD plugin.
[m6w6/libmemcached] / libtest / wait.cc
1 /* vim:expandtab:shiftwidth=2:tabstop=2:smarttab:
2 *
3 * Data Differential YATL (i.e. libtest) library
4 *
5 * Copyright (C) 2012 Data Differential, http://datadifferential.com/
6 *
7 * Redistribution and use in source and binary forms, with or without
8 * modification, are permitted provided that the following conditions are
9 * met:
10 *
11 * * Redistributions of source code must retain the above copyright
12 * notice, this list of conditions and the following disclaimer.
13 *
14 * * Redistributions in binary form must reproduce the above
15 * copyright notice, this list of conditions and the following disclaimer
16 * in the documentation and/or other materials provided with the
17 * distribution.
18 *
19 * * The names of its contributors may not be used to endorse or
20 * promote products derived from this software without specific prior
21 * written permission.
22 *
23 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
24 * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
25 * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
26 * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
27 * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
28 * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
29 * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
30 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
31 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
32 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
33 * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
34 *
35 */
36
37 #include <config.h>
38
39 #include <cstdlib>
40 #include <fcntl.h>
41 #include <getopt.h>
42 #include <iostream>
43 #include <sys/stat.h>
44 #include <sys/types.h>
45 #include <unistd.h>
46
47 #include <libtest/wait.h>
48
49 static void version_command(const char *command_name, int major_version, int minor_version)
50 {
51 std::cout << command_name << " " << major_version << "." << minor_version << std::endl;
52 }
53
54 static void help_command(const char *command_name,
55 int major_version, int minor_version,
56 const struct option *long_options)
57 {
58 std::cout << command_name << " " << major_version << "." << minor_version << std::endl;
59 std::cout << "Current options. A '=' means the option takes a value." << std::endl << std::endl;
60
61 for (uint32_t x= 0; long_options[x].name; x++)
62 {
63 std::cout << "\t --" << long_options[x].name << char(long_options[x].has_arg ? '=' : ' ') << std::endl;
64 }
65
66 std::cout << std::endl;
67 }
68
69 static void close_stdio(void)
70 {
71 int fd;
72 if ((fd = open("/dev/null", O_RDWR, 0)) < 0)
73 {
74 return;
75 }
76 else
77 {
78 if (dup2(fd, STDIN_FILENO) < 0)
79 {
80 return;
81 }
82
83 if (dup2(fd, STDOUT_FILENO) < 0)
84 {
85 return;
86 }
87
88 if (dup2(fd, STDERR_FILENO) < 0)
89 {
90 return;
91 }
92
93 if (fd > STDERR_FILENO)
94 {
95 close(fd);
96 }
97 }
98 }
99
100 enum {
101 OPT_HELP,
102 OPT_QUIET,
103 OPT_VERSION
104 };
105
106 static void options_parse(int argc, char *argv[])
107 {
108 static struct option long_options[]=
109 {
110 { "version", no_argument, NULL, OPT_VERSION},
111 { "help", no_argument, NULL, OPT_HELP},
112 { "quiet", no_argument, NULL, OPT_QUIET},
113 {0, 0, 0, 0},
114 };
115
116 bool opt_version= false;
117 bool opt_help= false;
118 bool opt_quiet= false;
119 int option_index= 0;
120
121 while (1)
122 {
123 int option_rv= getopt_long(argc, argv, "", long_options, &option_index);
124 if (option_rv == -1)
125 {
126 break;
127 }
128
129 switch (option_rv)
130 {
131 case OPT_HELP: /* --help or -h */
132 opt_help= true;
133 break;
134
135 case OPT_VERSION: /* --version or -v */
136 opt_version= true;
137 break;
138
139 case OPT_QUIET:
140 opt_quiet= true;
141 break;
142
143 case '?':
144 /* getopt_long already printed an error message. */
145 exit(EXIT_FAILURE);
146
147 default:
148 help_command(argv[0], 1, 0, long_options);
149 exit(EXIT_FAILURE);
150 }
151 }
152
153 if (opt_quiet)
154 {
155 close_stdio();
156 }
157
158 if (opt_version)
159 {
160 version_command(argv[0], 1, 0);
161 exit(EXIT_SUCCESS);
162 }
163
164 if (opt_help)
165 {
166 help_command(argv[0], 1, 0, long_options);
167 exit(EXIT_SUCCESS);
168 }
169 }
170
171 int main(int argc, char *argv[])
172 {
173 if (argc == 1)
174 {
175 return EXIT_FAILURE;
176 }
177
178 options_parse(argc, argv);
179
180 int ret= EXIT_FAILURE;
181 while (optind < argc)
182 {
183 libtest::Wait wait(argv[optind++]);
184
185 if (wait.successful() == false)
186 {
187 return EXIT_FAILURE;
188 }
189
190 ret= EXIT_SUCCESS;
191 }
192
193 return ret;
194 }