2 +--------------------------------------------------------------------+
3 | libmemcached - C/C++ Client Library for memcached |
4 +--------------------------------------------------------------------+
5 | Redistribution and use in source and binary forms, with or without |
6 | modification, are permitted under the terms of the BSD license. |
7 | You should have received a copy of the license in a bundled file |
8 | named LICENSE; in case you did not receive a copy you can review |
9 | the terms online at: https://opensource.org/licenses/BSD-3-Clause |
10 +--------------------------------------------------------------------+
11 | Copyright (c) 2006-2014 Brian Aker https://datadifferential.com/ |
12 | Copyright (c) 2020 Michael Wallner <mike@php.net> |
13 +--------------------------------------------------------------------+
16 #include "mem_config.h"
18 #include "utilities.h"
27 #include <sys/types.h>
30 long int timedif(struct timeval a
, struct timeval b
) {
33 us
= (int) (a
.tv_usec
- b
.tv_usec
);
35 s
= (int) (a
.tv_sec
- b
.tv_sec
);
40 void version_command(const char *command_name
) {
41 printf("%s v%u.%u\n", command_name
, 1U, 0U);
45 void close_stdio(void) {
47 if ((fd
= open("/dev/null", O_RDWR
, 0)) < 0) {
50 if (dup2(fd
, STDIN_FILENO
) < 0) {
54 if (dup2(fd
, STDOUT_FILENO
) < 0) {
58 if (dup2(fd
, STDERR_FILENO
) < 0) {
62 if (fd
> STDERR_FILENO
) {
68 static const char *lookup_help(memcached_options option
) {
71 return ("List which servers you wish to connect to.");
73 return ("Display the version of the application and then exit.");
75 return ("Display this message and then exit.");
77 return ("Give more details on the progression of the application.");
79 return ("stderr and stdin will be closed at application startup.");
81 return ("Provide output only useful for debugging.");
83 return ("Provide flag information for storage operation.");
85 return ("Set the expire option for the object.");
87 return ("Use set command with memcached when storing.");
89 return ("Use replace command with memcached when storing.");
91 return ("Use add command with memcached when storing.");
92 case OPT_SLAP_EXECUTE_NUMBER
:
93 return ("Number of times to execute the given test.");
94 case OPT_SLAP_INITIAL_LOAD
:
95 return ("Number of key pairs to load before executing tests.");
97 return ("Test to run (currently \"get\" or \"set\").");
98 case OPT_SLAP_CONCURRENCY
:
99 return ("Number of users to simulate with load.");
100 case OPT_SLAP_NON_BLOCK
:
101 return ("Set TCP up to use non-blocking IO.");
102 case OPT_SLAP_TCP_NODELAY
:
103 return ("Set TCP socket up to use nodelay.");
105 return ("Flush servers before running tests.");
107 return ("Select hash type.");
109 return ("Switch to binary protocol.");
111 return ("Analyze the provided servers.");
113 return ("Use UDP protocol when communicating with server.");
115 return ("Enable request buffering.");
117 return "Username to use for SASL authentication";
119 return "Password to use for SASL authentication";
121 return "Path to file in which to save result";
123 return "Argument for statistics";
124 case OPT_SERVER_VERSION
:
125 return "Memcached daemon software version";
131 return "forgot to document this function :)";
134 void help_command(const char *command_name
, const char *description
,
135 const struct option
*long_options
, memcached_programs_help_st
*options
) {
139 printf("%s v%u.%u\n\n", command_name
, 1U, 0U);
140 printf("\t%s\n\n", description
);
141 printf("Current options. A '=' means the option takes a value.\n\n");
143 for (x
= 0; long_options
[x
].name
; x
++) {
144 const char *help_message
;
146 printf("\t --%s%c\n", long_options
[x
].name
, long_options
[x
].has_arg
? '=' : ' ');
147 if ((help_message
= lookup_help(memcached_options(long_options
[x
].val
))))
148 printf("\t\t%s\n", help_message
);
155 void process_hash_option(memcached_st
*memc
, char *opt_hash
) {
157 memcached_return_t rc
;
159 if (opt_hash
== NULL
) {
163 set
= MEMCACHED_HASH_DEFAULT
; /* Just here to solve warning */
164 if (!strcasecmp(opt_hash
, "CRC")) {
165 set
= MEMCACHED_HASH_CRC
;
166 } else if (!strcasecmp(opt_hash
, "FNV1_64")) {
167 set
= MEMCACHED_HASH_FNV1_64
;
168 } else if (!strcasecmp(opt_hash
, "FNV1A_64")) {
169 set
= MEMCACHED_HASH_FNV1A_64
;
170 } else if (!strcasecmp(opt_hash
, "FNV1_32")) {
171 set
= MEMCACHED_HASH_FNV1_32
;
172 } else if (!strcasecmp(opt_hash
, "FNV1A_32")) {
173 set
= MEMCACHED_HASH_FNV1A_32
;
175 fprintf(stderr
, "hash: type not recognized %s\n", opt_hash
);
179 rc
= memcached_behavior_set(memc
, MEMCACHED_BEHAVIOR_HASH
, set
);
180 if (rc
!= MEMCACHED_SUCCESS
) {
181 fprintf(stderr
, "hash: memcache error %s\n", memcached_strerror(memc
, rc
));
186 void initialize_sockets(void) {
187 /* Define the function for all platforms to avoid #ifdefs in each program */
190 if (WSAStartup(MAKEWORD(2, 0), &wsaData
)) {
191 fprintf(stderr
, "Socket Initialization Error. Program aborted\n");
194 #endif // #if defined(_WIN32)