git.m6w6.name Git - awesomized/libmemcached/blob - 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 "libtest/yatlcon.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 }