git.m6w6.name Git - awesomized/libmemcached/blob - 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 bool 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 *,
  47                                       const char *key, size_t key_length,
  48                                       void *)
  49 {
  50   std::cout.write(key, key_length);
  51   std::cout << std::endl;
  52
  53   return MEMCACHED_SUCCESS;
  54 }
  55
  56 int main(int argc, char *argv[])
  57 {
  58   memcached_dump_fn callbacks[1];
  59
  60   callbacks[0]= &key_printer;
  61
  62   options_parse(argc, argv);
  63
  64   memcached_st *memc= memcached_create(NULL);
  65   process_hash_option(memc, opt_hash);
  66
  67   if (opt_servers == NULL)
  68   {
  69     char *temp;
  70
  71     if ((temp= getenv("MEMCACHED_SERVERS")))
  72     {
  73       opt_servers= strdup(temp);
  74     }
  75     else
  76     {
  77       std::cerr << "No Servers provided" << std::endl;
  78       exit(EXIT_FAILURE);
  79     }
  80   }
  81
  82   memcached_server_st *servers;
  83   if (opt_servers)
  84   {
  85     servers= memcached_servers_parse(opt_servers);
  86   }
  87   else
  88   {
  89     servers= memcached_servers_parse(argv[--argc]);
  90   }
  91
  92   memcached_server_push(memc, servers);
  93   memcached_server_list_free(servers);
  94   memcached_behavior_set(memc, MEMCACHED_BEHAVIOR_BINARY_PROTOCOL,
  95                          (uint64_t)opt_binary);
  96
  97   if (opt_username and LIBMEMCACHED_WITH_SASL_SUPPORT == 0)
  98   {
  99     memcached_free(memc);
 100     std::cerr << "--username was supplied, but binary was not built with SASL support." << std::endl;
 101     return EXIT_FAILURE;
 102   }
 103
 104   if (opt_username)
 105   {
 106     memcached_return_t ret;
 107     if (memcached_failed(ret= memcached_set_sasl_auth_data(memc, opt_username, opt_passwd)))
 108     {
 109       std::cerr << memcached_last_error_message(memc) << std::endl;
 110       memcached_free(memc);
 111       return EXIT_FAILURE;
 112     }
 113   }
 114
 115   memcached_return_t rc= memcached_dump(memc, callbacks, NULL, 1);
 116
 117   int exit_code= EXIT_SUCCESS;
 118   if (memcached_failed(rc))
 119   {
 120     if (opt_verbose)
 121     {
 122       std::cerr << "Failed to dump keys: " << memcached_last_error_message(memc) << std::endl;
 123     }
 124     exit_code= EXIT_FAILURE;
 125   }
 126
 127   memcached_free(memc);
 128
 129   if (opt_servers)
 130   {
 131     free(opt_servers);
 132   }
 133   if (opt_hash)
 134   {
 135     free(opt_hash);
 136   }
 137
 138   return exit_code;
 139 }
 140
 141 static void options_parse(int argc, char *argv[])
 142 {
 143   static struct option long_options[]=
 144     {
 145       {(OPTIONSTRING)"version", no_argument, NULL, OPT_VERSION},
 146       {(OPTIONSTRING)"help", no_argument, NULL, OPT_HELP},
 147       {(OPTIONSTRING)"quiet", no_argument, NULL, OPT_QUIET},
 148       {(OPTIONSTRING)"verbose", no_argument, &opt_verbose, OPT_VERBOSE},
 149       {(OPTIONSTRING)"debug", no_argument, &opt_verbose, OPT_DEBUG},
 150       {(OPTIONSTRING)"servers", required_argument, NULL, OPT_SERVERS},
 151       {(OPTIONSTRING)"hash", required_argument, NULL, OPT_HASH},
 152       {(OPTIONSTRING)"binary", no_argument, NULL, OPT_BINARY},
 153       {(OPTIONSTRING)"username", required_argument, NULL, OPT_USERNAME},
 154       {(OPTIONSTRING)"password", required_argument, NULL, OPT_PASSWD},
 155       {0, 0, 0, 0}
 156     };
 157
 158   int option_index= 0;
 159   bool opt_version= false;
 160   bool opt_help= false;
 161   while (1)
 162   {
 163     int option_rv= getopt_long(argc, argv, "Vhvds:", long_options, &option_index);
 164
 165     if (option_rv == -1) break;
 166
 167     switch (option_rv)
 168     {
 169     case 0:
 170       break;
 171
 172     case OPT_BINARY:
 173       opt_binary= true;
 174       break;
 175
 176     case OPT_VERBOSE: /* --verbose or -v */
 177       opt_verbose= OPT_VERBOSE;
 178       break;
 179
 180     case OPT_DEBUG: /* --debug or -d */
 181       opt_verbose= OPT_DEBUG;
 182       break;
 183
 184     case OPT_VERSION: /* --version or -V */
 185       opt_verbose= true;
 186       break;
 187
 188     case OPT_HELP: /* --help or -h */
 189       opt_help= true;
 190       break;
 191
 192     case OPT_SERVERS: /* --servers or -s */
 193       opt_servers= strdup(optarg);
 194       break;
 195
 196     case OPT_HASH:
 197       opt_hash= strdup(optarg);
 198       break;
 199
 200     case OPT_USERNAME:
 201        opt_username= optarg;
 202        break;
 203
 204     case OPT_PASSWD:
 205        opt_passwd= optarg;
 206        break;
 207
 208     case OPT_QUIET:
 209       close_stdio();
 210       break;
 211
 212     case '?':
 213       /* getopt_long already printed an error message. */
 214       exit(1);
 215     default:
 216       abort();
 217     }
 218   }
 219
 220   if (opt_version)
 221   {
 222     version_command(PROGRAM_NAME);
 223     exit(EXIT_SUCCESS);
 224   }
 225
 226   if (opt_help)
 227   {
 228     help_command(PROGRAM_NAME, PROGRAM_DESCRIPTION, long_options, NULL);
 229     exit(EXIT_SUCCESS);
 230   }
 231 }