Update for mavericks
[m6w6/libmemcached] / clients / utilities.cc
1 /* LibMemcached
2 * Copyright (C) 2011-2012 Data Differential, http://datadifferential.com/
3 * Copyright (C) 2006-2009 Brian Aker
4 * All rights reserved.
5 *
6 * Use and distribution licensed under the BSD license. See
7 * the COPYING file in the parent directory for full text.
8 *
9 * Summary:
10 *
11 */
12 #include <mem_config.h>
13
14 #include <clients/utilities.h>
15
16 #include <cstdio>
17 #include <cassert>
18 #include <cstdlib>
19 #include <cstring>
20 #include <ctype.h>
21 #include <fcntl.h>
22 #include <sys/stat.h>
23 #include <sys/types.h>
24 #include <unistd.h>
25
26
27 long int timedif(struct timeval a, struct timeval b)
28 {
29 long us, s;
30
31 us = (int)(a.tv_usec - b.tv_usec);
32 us /= 1000;
33 s = (int)(a.tv_sec - b.tv_sec);
34 s *= 1000;
35 return s + us;
36 }
37
38 void version_command(const char *command_name)
39 {
40 printf("%s v%u.%u\n", command_name, 1U, 0U);
41 exit(EXIT_SUCCESS);
42 }
43
44 void close_stdio(void)
45 {
46 int fd;
47 if ((fd = open("/dev/null", O_RDWR, 0)) < 0)
48 {
49 return;
50 }
51 else
52 {
53 if (dup2(fd, STDIN_FILENO) < 0)
54 {
55 return;
56 }
57
58 if (dup2(fd, STDOUT_FILENO) < 0)
59 {
60 return;
61 }
62
63 if (dup2(fd, STDERR_FILENO) < 0)
64 {
65 return;
66 }
67
68 if (fd > STDERR_FILENO)
69 {
70 close(fd);
71 }
72 }
73 }
74
75
76 static const char *lookup_help(memcached_options option)
77 {
78 switch (option)
79 {
80 case OPT_SERVERS: return("List which servers you wish to connect to.");
81 case OPT_VERSION: return("Display the version of the application and then exit.");
82 case OPT_HELP: return("Display this message and then exit.");
83 case OPT_VERBOSE: return("Give more details on the progression of the application.");
84 case OPT_QUIET: return("stderr and stdin will be closed at application startup.");
85 case OPT_DEBUG: return("Provide output only useful for debugging.");
86 case OPT_FLAG: return("Provide flag information for storage operation.");
87 case OPT_EXPIRE: return("Set the expire option for the object.");
88 case OPT_SET: return("Use set command with memcached when storing.");
89 case OPT_REPLACE: return("Use replace command with memcached when storing.");
90 case OPT_ADD: return("Use add command with memcached when storing.");
91 case OPT_SLAP_EXECUTE_NUMBER: return("Number of times to execute the given test.");
92 case OPT_SLAP_INITIAL_LOAD: return("Number of key pairs to load before executing tests.");
93 case OPT_SLAP_TEST: return("Test to run (currently \"get\" or \"set\").");
94 case OPT_SLAP_CONCURRENCY: return("Number of users to simulate with load.");
95 case OPT_SLAP_NON_BLOCK: return("Set TCP up to use non-blocking IO.");
96 case OPT_SLAP_TCP_NODELAY: return("Set TCP socket up to use nodelay.");
97 case OPT_FLUSH: return("Flush servers before running tests.");
98 case OPT_HASH: return("Select hash type.");
99 case OPT_BINARY: return("Switch to binary protocol.");
100 case OPT_ANALYZE: return("Analyze the provided servers.");
101 case OPT_UDP: return("Use UDP protocol when communicating with server.");
102 case OPT_BUFFER: return("Enable request buffering.");
103 case OPT_USERNAME: return "Username to use for SASL authentication";
104 case OPT_PASSWD: return "Password to use for SASL authentication";
105 case OPT_FILE: return "Path to file in which to save result";
106 case OPT_STAT_ARGS: return "Argument for statistics";
107 case OPT_SERVER_VERSION: return "Memcached daemon software version";
108 default:
109 break;
110 };
111
112 assert(0);
113 return "forgot to document this function :)";
114 }
115
116 void help_command(const char *command_name, const char *description,
117 const struct option *long_options,
118 memcached_programs_help_st *options)
119 {
120 unsigned int x;
121 (void)options;
122
123 printf("%s v%u.%u\n\n", command_name, 1U, 0U);
124 printf("\t%s\n\n", description);
125 printf("Current options. A '=' means the option takes a value.\n\n");
126
127 for (x= 0; long_options[x].name; x++)
128 {
129 const char *help_message;
130
131 printf("\t --%s%c\n", long_options[x].name,
132 long_options[x].has_arg ? '=' : ' ');
133 if ((help_message= lookup_help(memcached_options(long_options[x].val))))
134 printf("\t\t%s\n", help_message);
135 }
136
137 printf("\n");
138 exit(EXIT_SUCCESS);
139 }
140
141 void process_hash_option(memcached_st *memc, char *opt_hash)
142 {
143 uint64_t set;
144 memcached_return_t rc;
145
146 if (opt_hash == NULL)
147 {
148 return;
149 }
150
151 set= MEMCACHED_HASH_DEFAULT; /* Just here to solve warning */
152 if (!strcasecmp(opt_hash, "CRC"))
153 {
154 set= MEMCACHED_HASH_CRC;
155 }
156 else if (!strcasecmp(opt_hash, "FNV1_64"))
157 {
158 set= MEMCACHED_HASH_FNV1_64;
159 }
160 else if (!strcasecmp(opt_hash, "FNV1A_64"))
161 {
162 set= MEMCACHED_HASH_FNV1A_64;
163 }
164 else if (!strcasecmp(opt_hash, "FNV1_32"))
165 {
166 set= MEMCACHED_HASH_FNV1_32;
167 }
168 else if (!strcasecmp(opt_hash, "FNV1A_32"))
169 {
170 set= MEMCACHED_HASH_FNV1A_32;
171 }
172 else
173 {
174 fprintf(stderr, "hash: type not recognized %s\n", opt_hash);
175 exit(EXIT_FAILURE);
176 }
177
178 rc= memcached_behavior_set(memc, MEMCACHED_BEHAVIOR_HASH, set);
179 if (rc != MEMCACHED_SUCCESS)
180 {
181 fprintf(stderr, "hash: memcache error %s\n", memcached_strerror(memc, rc));
182 exit(EXIT_FAILURE);
183 }
184 }
185
186 void initialize_sockets(void)
187 {
188 /* Define the function for all platforms to avoid #ifdefs in each program */
189 #if defined(_WIN32)
190 WSADATA wsaData;
191 if (WSAStartup(MAKEWORD(2,0), &wsaData) != 0)
192 {
193 fprintf(stderr, "Socket Initialization Error. Program aborted\n");
194 exit(EXIT_FAILURE);
195 }
196 #endif // #if defined(_WIN32)
197 }