Standardize the snprintf() failure messages, and add tests for libmemcached_util_getpid()
[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 <stdio.h>
15 #include <stdlib.h>
16 #include <inttypes.h>
17 #include <unistd.h>
18 #include <getopt.h>
19 #include <sys/types.h>
20 #include <sys/stat.h>
21 #include <sys/types.h>
22 #include <fcntl.h>
23 #include <errno.h>
24 #include <strings.h>
25 #include <string.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 if (!initialize_sasl(memc, opt_username, opt_passwd))
93 {
94 memcached_free(memc);
95 return EXIT_FAILURE;
96 }
97
98 rc= memcached_dump(memc, callbacks, NULL, 1);
99
100 if (rc != MEMCACHED_SUCCESS)
101 {
102 fprintf(stderr, "memdump: memcache error %s", memcached_strerror(memc, rc));
103 if (memcached_last_error_errno(memc))
104 fprintf(stderr, " system error %s", strerror(memcached_last_error_errno(memc)));
105 fprintf(stderr, "\n");
106 }
107
108 memcached_free(memc);
109
110 if (opt_servers)
111 free(opt_servers);
112 if (opt_hash)
113 free(opt_hash);
114
115 shutdown_sasl();
116
117 return EXIT_SUCCESS;
118 }
119
120 static void options_parse(int argc, char *argv[])
121 {
122 int option_index= 0;
123 int option_rv;
124
125 static struct option long_options[]=
126 {
127 {(OPTIONSTRING)"version", no_argument, NULL, OPT_VERSION},
128 {(OPTIONSTRING)"help", no_argument, NULL, OPT_HELP},
129 {(OPTIONSTRING)"verbose", no_argument, &opt_verbose, OPT_VERBOSE},
130 {(OPTIONSTRING)"debug", no_argument, &opt_verbose, OPT_DEBUG},
131 {(OPTIONSTRING)"servers", required_argument, NULL, OPT_SERVERS},
132 {(OPTIONSTRING)"hash", required_argument, NULL, OPT_HASH},
133 {(OPTIONSTRING)"binary", no_argument, NULL, OPT_BINARY},
134 {(OPTIONSTRING)"username", required_argument, NULL, OPT_USERNAME},
135 {(OPTIONSTRING)"password", required_argument, NULL, OPT_PASSWD},
136 {0, 0, 0, 0}
137 };
138
139 while (1)
140 {
141 option_rv= getopt_long(argc, argv, "Vhvds:", long_options, &option_index);
142
143 if (option_rv == -1) break;
144
145 switch (option_rv)
146 {
147 case 0:
148 break;
149 case OPT_BINARY:
150 opt_binary = 1;
151 break;
152 case OPT_VERBOSE: /* --verbose or -v */
153 opt_verbose = OPT_VERBOSE;
154 break;
155 case OPT_DEBUG: /* --debug or -d */
156 opt_verbose = OPT_DEBUG;
157 break;
158 case OPT_VERSION: /* --version or -V */
159 version_command(PROGRAM_NAME);
160 break;
161 case OPT_HELP: /* --help or -h */
162 help_command(PROGRAM_NAME, PROGRAM_DESCRIPTION, long_options, NULL);
163 break;
164 case OPT_SERVERS: /* --servers or -s */
165 opt_servers= strdup(optarg);
166 break;
167 case OPT_HASH:
168 opt_hash= strdup(optarg);
169 break;
170 case OPT_USERNAME:
171 opt_username= optarg;
172 break;
173 case OPT_PASSWD:
174 opt_passwd= optarg;
175 break;
176 case '?':
177 /* getopt_long already printed an error message. */
178 exit(1);
179 default:
180 abort();
181 }
182 }
183 }