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
) {
70 case OPT_SERVERS
: return ("List which servers you wish to connect to.");
71 case OPT_VERSION
: return ("Display the version of the application and then exit.");
72 case OPT_HELP
: return ("Display this message and then exit.");
73 case OPT_VERBOSE
: return ("Give more details on the progression of the application.");
74 case OPT_QUIET
: return ("stderr and stdin will be closed at application startup.");
75 case OPT_DEBUG
: return ("Provide output only useful for debugging.");
76 case OPT_FLAG
: return ("Provide flag information for storage operation.");
77 case OPT_EXPIRE
: return ("Set the expire option for the object.");
78 case OPT_SET
: return ("Use set command with memcached when storing.");
79 case OPT_REPLACE
: return ("Use replace command with memcached when storing.");
80 case OPT_ADD
: return ("Use add command with memcached when storing.");
81 case OPT_SLAP_EXECUTE_NUMBER
: return ("Number of times to execute the given test.");
82 case OPT_SLAP_INITIAL_LOAD
: return ("Number of key pairs to load before executing tests.");
83 case OPT_SLAP_TEST
: return ("Test to run (currently \"get\" or \"set\").");
84 case OPT_SLAP_CONCURRENCY
: return ("Number of users to simulate with load.");
85 case OPT_SLAP_NON_BLOCK
: return ("Set TCP up to use non-blocking IO.");
86 case OPT_SLAP_TCP_NODELAY
: return ("Set TCP socket up to use nodelay.");
87 case OPT_FLUSH
: return ("Flush servers before running tests.");
88 case OPT_HASH
: return ("Select hash type.");
89 case OPT_BINARY
: return ("Switch to binary protocol.");
90 case OPT_ANALYZE
: return ("Analyze the provided servers.");
91 case OPT_UDP
: return ("Use UDP protocol when communicating with server.");
92 case OPT_BUFFER
: return ("Enable request buffering.");
93 case OPT_USERNAME
: return "Username to use for SASL authentication";
94 case OPT_PASSWD
: return "Password to use for SASL authentication";
95 case OPT_FILE
: return "Path to file in which to save result";
96 case OPT_STAT_ARGS
: return "Argument for statistics";
97 case OPT_SERVER_VERSION
: return "Memcached daemon software version";
102 return "forgot to document this function :)";
105 void help_command(const char *command_name
, const char *description
,
106 const struct option
*long_options
, memcached_programs_help_st
*options
) {
110 printf("%s v%u.%u\n\n", command_name
, 1U, 0U);
111 printf("\t%s\n\n", description
);
112 printf("Current options. A '=' means the option takes a value.\n\n");
114 for (x
= 0; long_options
[x
].name
; x
++) {
115 const char *help_message
;
117 printf("\t --%s%c\n", long_options
[x
].name
, long_options
[x
].has_arg
? '=' : ' ');
118 if ((help_message
= lookup_help(memcached_options(long_options
[x
].val
))))
119 printf("\t\t%s\n", help_message
);
126 void process_hash_option(memcached_st
*memc
, char *opt_hash
) {
128 memcached_return_t rc
;
130 if (opt_hash
== NULL
) {
134 set
= MEMCACHED_HASH_DEFAULT
; /* Just here to solve warning */
135 if (!strcasecmp(opt_hash
, "CRC")) {
136 set
= MEMCACHED_HASH_CRC
;
137 } else if (!strcasecmp(opt_hash
, "FNV1_64")) {
138 set
= MEMCACHED_HASH_FNV1_64
;
139 } else if (!strcasecmp(opt_hash
, "FNV1A_64")) {
140 set
= MEMCACHED_HASH_FNV1A_64
;
141 } else if (!strcasecmp(opt_hash
, "FNV1_32")) {
142 set
= MEMCACHED_HASH_FNV1_32
;
143 } else if (!strcasecmp(opt_hash
, "FNV1A_32")) {
144 set
= MEMCACHED_HASH_FNV1A_32
;
146 fprintf(stderr
, "hash: type not recognized %s\n", opt_hash
);
150 rc
= memcached_behavior_set(memc
, MEMCACHED_BEHAVIOR_HASH
, set
);
151 if (rc
!= MEMCACHED_SUCCESS
) {
152 fprintf(stderr
, "hash: memcache error %s\n", memcached_strerror(memc
, rc
));
157 void initialize_sockets(void) {
158 /* Define the function for all platforms to avoid #ifdefs in each program */
161 if (WSAStartup(MAKEWORD(2, 0), &wsaData
) != 0) {
162 fprintf(stderr
, "Socket Initialization Error. Program aborted\n");
165 #endif // #if defined(_WIN32)