Fix bad if path
[m6w6/libmemcached] / 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 }