Add test cases for add server for NULL cases.
[m6w6/libmemcached] / clients / memdump.cc
1 /* LibMemcached
2 * Copyright (C) 2006-2009 Brian Aker
3 * All rights reserved.
4 *
5 * Use and distribution licensed under the BSD license. See
6 * the COPYING file in the parent directory for full text.
7 *
8 * Summary:
9 *
10 */
11
12 #include "config.h"
13
14 #include <cerrno>
15 #include <cstdio>
16 #include <cstdlib>
17 #include <cstring>
18 #include <fcntl.h>
19 #include <getopt.h>
20 #include <inttypes.h>
21 #include <iostream>
22 #include <sys/stat.h>
23 #include <sys/types.h>
24 #include <sys/types.h>
25 #include <unistd.h>
26
27 #include <libmemcached/memcached.h>
28
29 #include "client_options.h"
30 #include "utilities.h"
31
32 #define PROGRAM_NAME "memdump"
33 #define PROGRAM_DESCRIPTION "Dump all values from one or many servers."
34
35 /* Prototypes */
36 static void options_parse(int argc, char *argv[]);
37
38 static int opt_binary=0;
39 static int opt_verbose= 0;
40 static char *opt_servers= NULL;
41 static char *opt_hash= NULL;
42 static char *opt_username;
43 static char *opt_passwd;
44
45 /* Print the keys and counter how many were found */
46 static memcached_return_t key_printer(const memcached_st *ptr,
47 const char *key, size_t key_length,
48 void *context)
49 {
50 (void)ptr;(void)context;
51 printf("%.*s\n", (uint32_t)key_length, key);
52
53 return MEMCACHED_SUCCESS;
54 }
55
56 int main(int argc, char *argv[])
57 {
58 memcached_st *memc;
59 memcached_return_t rc;
60 memcached_server_st *servers;
61 memcached_dump_fn callbacks[1];
62
63 callbacks[0]= &key_printer;
64
65 options_parse(argc, argv);
66
67 memc= memcached_create(NULL);
68 process_hash_option(memc, opt_hash);
69
70 if (!opt_servers)
71 {
72 char *temp;
73
74 if ((temp= getenv("MEMCACHED_SERVERS")))
75 opt_servers= strdup(temp);
76 else
77 {
78 fprintf(stderr, "No Servers provided\n");
79 exit(1);
80 }
81 }
82
83 if (opt_servers)
84 servers= memcached_servers_parse(opt_servers);
85 else
86 servers= memcached_servers_parse(argv[--argc]);
87
88 memcached_server_push(memc, servers);
89 memcached_server_list_free(servers);
90 memcached_behavior_set(memc, MEMCACHED_BEHAVIOR_BINARY_PROTOCOL,
91 (uint64_t)opt_binary);
92
93 if (opt_username and LIBMEMCACHED_WITH_SASL_SUPPORT == 0)
94 {
95 memcached_free(memc);
96 std::cerr << "--username was supplied, but binary was not built with SASL support." << std::endl;
97 return EXIT_FAILURE;
98 }
99
100 if (opt_username and initialize_sasl(memc, opt_username, opt_passwd) == false)
101 {
102 std::cerr << "Failed to initialize SASL support." << std::endl;
103
104 memcached_free(memc);
105 return EXIT_FAILURE;
106 }
107
108 rc= memcached_dump(memc, callbacks, NULL, 1);
109
110 if (rc != MEMCACHED_SUCCESS)
111 {
112 fprintf(stderr, "memdump: memcache error %s", memcached_strerror(memc, rc));
113 if (memcached_last_error_errno(memc))
114 fprintf(stderr, " system error %s", strerror(memcached_last_error_errno(memc)));
115 fprintf(stderr, "\n");
116 }
117
118 memcached_free(memc);
119
120 if (opt_servers)
121 free(opt_servers);
122 if (opt_hash)
123 free(opt_hash);
124
125 shutdown_sasl();
126
127 return EXIT_SUCCESS;
128 }
129
130 static void options_parse(int argc, char *argv[])
131 {
132 int option_index= 0;
133 int option_rv;
134
135 static struct option long_options[]=
136 {
137 {(OPTIONSTRING)"version", no_argument, NULL, OPT_VERSION},
138 {(OPTIONSTRING)"help", no_argument, NULL, OPT_HELP},
139 {(OPTIONSTRING)"verbose", no_argument, &opt_verbose, OPT_VERBOSE},
140 {(OPTIONSTRING)"debug", no_argument, &opt_verbose, OPT_DEBUG},
141 {(OPTIONSTRING)"servers", required_argument, NULL, OPT_SERVERS},
142 {(OPTIONSTRING)"hash", required_argument, NULL, OPT_HASH},
143 {(OPTIONSTRING)"binary", no_argument, NULL, OPT_BINARY},
144 {(OPTIONSTRING)"username", required_argument, NULL, OPT_USERNAME},
145 {(OPTIONSTRING)"password", required_argument, NULL, OPT_PASSWD},
146 {0, 0, 0, 0}
147 };
148
149 while (1)
150 {
151 option_rv= getopt_long(argc, argv, "Vhvds:", long_options, &option_index);
152
153 if (option_rv == -1) break;
154
155 switch (option_rv)
156 {
157 case 0:
158 break;
159 case OPT_BINARY:
160 opt_binary = 1;
161 break;
162 case OPT_VERBOSE: /* --verbose or -v */
163 opt_verbose = OPT_VERBOSE;
164 break;
165 case OPT_DEBUG: /* --debug or -d */
166 opt_verbose = OPT_DEBUG;
167 break;
168 case OPT_VERSION: /* --version or -V */
169 version_command(PROGRAM_NAME);
170 break;
171 case OPT_HELP: /* --help or -h */
172 help_command(PROGRAM_NAME, PROGRAM_DESCRIPTION, long_options, NULL);
173 break;
174 case OPT_SERVERS: /* --servers or -s */
175 opt_servers= strdup(optarg);
176 break;
177 case OPT_HASH:
178 opt_hash= strdup(optarg);
179 break;
180 case OPT_USERNAME:
181 opt_username= optarg;
182 break;
183 case OPT_PASSWD:
184 opt_passwd= optarg;
185 break;
186 case '?':
187 /* getopt_long already printed an error message. */
188 exit(1);
189 default:
190 abort();
191 }
192 }
193 }