2 * Copyright (C) 2006-2009 Brian Aker
5 * Use and distribution licensed under the BSD license. See
6 * the COPYING file in the parent directory for full text.
13 #include <clients/utilities.h>
21 #include <sys/types.h>
25 long int timedif(struct timeval a
, struct timeval b
)
29 us
= (int)(a
.tv_usec
- b
.tv_usec
);
31 s
= (int)(a
.tv_sec
- b
.tv_sec
);
36 void version_command(const char *command_name
)
38 printf("%s v%u.%u\n", command_name
, 1U, 0U);
42 void close_stdio(void)
45 if ((fd
= open("/dev/null", O_RDWR
, 0)) < 0)
51 if (dup2(fd
, STDIN_FILENO
) < 0)
56 if (dup2(fd
, STDOUT_FILENO
) < 0)
61 if (dup2(fd
, STDERR_FILENO
) < 0)
66 if (fd
> STDERR_FILENO
)
74 static const char *lookup_help(memcached_options option
)
78 case OPT_SERVERS
: return("List which servers you wish to connect to.");
79 case OPT_VERSION
: return("Display the version of the application and then exit.");
80 case OPT_HELP
: return("Display this message and then exit.");
81 case OPT_VERBOSE
: return("Give more details on the progression of the application.");
82 case OPT_QUIET
: return("stderr and stdin will be closed at application startup.");
83 case OPT_DEBUG
: return("Provide output only useful for debugging.");
84 case OPT_FLAG
: return("Provide flag information for storage operation.");
85 case OPT_EXPIRE
: return("Set the expire option for the object.");
86 case OPT_SET
: return("Use set command with memcached when storing.");
87 case OPT_REPLACE
: return("Use replace command with memcached when storing.");
88 case OPT_ADD
: return("Use add command with memcached when storing.");
89 case OPT_SLAP_EXECUTE_NUMBER
: return("Number of times to execute the given test.");
90 case OPT_SLAP_INITIAL_LOAD
: return("Number of key pairs to load before executing tests.");
91 case OPT_SLAP_TEST
: return("Test to run (currently \"get\" or \"set\").");
92 case OPT_SLAP_CONCURRENCY
: return("Number of users to simulate with load.");
93 case OPT_SLAP_NON_BLOCK
: return("Set TCP up to use non-blocking IO.");
94 case OPT_SLAP_TCP_NODELAY
: return("Set TCP socket up to use nodelay.");
95 case OPT_FLUSH
: return("Flush servers before running tests.");
96 case OPT_HASH
: return("Select hash type.");
97 case OPT_BINARY
: return("Switch to binary protocol.");
98 case OPT_ANALYZE
: return("Analyze the provided servers.");
99 case OPT_UDP
: return("Use UDP protocol when communicating with server.");
100 case OPT_USERNAME
: return "Username to use for SASL authentication";
101 case OPT_PASSWD
: return "Password to use for SASL authentication";
102 case OPT_FILE
: return "Path to file in which to save result";
103 case OPT_STAT_ARGS
: return "Argument for statistics";
104 default: WATCHPOINT_ASSERT(0);
107 WATCHPOINT_ASSERT(0);
108 return "forgot to document this function :)";
111 void help_command(const char *command_name
, const char *description
,
112 const struct option
*long_options
,
113 memcached_programs_help_st
*options
)
118 printf("%s v%u.%u\n\n", command_name
, 1U, 0U);
119 printf("\t%s\n\n", description
);
120 printf("Current options. A '=' means the option takes a value.\n\n");
122 for (x
= 0; long_options
[x
].name
; x
++)
124 const char *help_message
;
126 printf("\t --%s%c\n", long_options
[x
].name
,
127 long_options
[x
].has_arg
? '=' : ' ');
128 if ((help_message
= lookup_help(memcached_options(long_options
[x
].val
))))
129 printf("\t\t%s\n", help_message
);
136 void process_hash_option(memcached_st
*memc
, char *opt_hash
)
139 memcached_return_t rc
;
141 if (opt_hash
== NULL
)
144 set
= MEMCACHED_HASH_DEFAULT
; /* Just here to solve warning */
145 if (!strcasecmp(opt_hash
, "CRC"))
147 set
= MEMCACHED_HASH_CRC
;
149 else if (!strcasecmp(opt_hash
, "FNV1_64"))
151 set
= MEMCACHED_HASH_FNV1_64
;
153 else if (!strcasecmp(opt_hash
, "FNV1A_64"))
155 set
= MEMCACHED_HASH_FNV1A_64
;
157 else if (!strcasecmp(opt_hash
, "FNV1_32"))
159 set
= MEMCACHED_HASH_FNV1_32
;
161 else if (!strcasecmp(opt_hash
, "FNV1A_32"))
163 set
= MEMCACHED_HASH_FNV1A_32
;
167 fprintf(stderr
, "hash: type not recognized %s\n", opt_hash
);
171 rc
= memcached_behavior_set(memc
, MEMCACHED_BEHAVIOR_HASH
, set
);
172 if (rc
!= MEMCACHED_SUCCESS
)
174 fprintf(stderr
, "hash: memcache error %s\n", memcached_strerror(memc
, rc
));
179 void initialize_sockets(void)
181 /* Define the function for all platforms to avoid #ifdefs in each program */
182 #if defined(WIN32) && WIN32
184 if (WSAStartup(MAKEWORD(2,0), &wsaData
) != 0)
186 fprintf(stderr
, "Socket Initialization Error. Program aborted\n");