git.m6w6.name Git - m6w6/libmemcached/blob - libtest/core_count.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/cpu.hpp>
  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 << "Prints the number of cores found on the local host." << 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 enum {
  70   OPT_HELP,
  71   OPT_VERSION
  72 };
  73
  74 static void options_parse(int argc, char *argv[])
  75 {
  76   static struct option long_options[]=
  77   {
  78     { "version", no_argument, NULL, OPT_VERSION},
  79     { "help", no_argument, NULL, OPT_HELP},
  80     {0, 0, 0, 0},
  81   };
  82
  83   bool opt_version= false;
  84   bool opt_help= false;
  85   int option_index= 0;
  86
  87   while (1)
  88   {
  89     int option_rv= getopt_long(argc, argv, "", long_options, &option_index);
  90     if (option_rv == -1)
  91     {
  92       break;
  93     }
  94
  95     switch (option_rv)
  96     {
  97     case OPT_HELP: /* --help or -h */
  98       opt_help= true;
  99       break;
 100
 101     case OPT_VERSION: /* --version or -v */
 102       opt_version= true;
 103       break;
 104
 105     case '?':
 106       /* getopt_long already printed an error message. */
 107       exit(EXIT_FAILURE);
 108
 109     default:
 110       help_command(argv[0], 1, 0, long_options);
 111       exit(EXIT_FAILURE);
 112     }
 113   }
 114
 115   if (opt_version)
 116   {
 117     version_command(argv[0], 1, 0);
 118     exit(EXIT_SUCCESS);
 119   }
 120
 121   if (opt_help)
 122   {
 123     help_command(argv[0], 1, 0, long_options);
 124     exit(EXIT_SUCCESS);
 125   }
 126 }
 127
 128 int main(int argc, char *argv[])
 129 {
 130   options_parse(argc, argv);
 131
 132   std::cout << libtest::number_of_cpus() << std::endl;
 133
 134   return EXIT_SUCCESS;
 135 }