X-Git-Url: https://git.m6w6.name/?a=blobdiff_plain;f=src%2Fmemcat.c;h=2db0c8ee04e93d05bd8a8c4289874532f653ba54;hb=42a7cc99572d9cf161e9db04f31e123b5431df10;hp=bb00ca66d494341cbbfdd6a6d37c5ba5fb3676cc;hpb=44152c2062812b74edbcad6b7bf15e78d93ed12c;p=m6w6%2Flibmemcached diff --git a/src/memcat.c b/src/memcat.c index bb00ca66..2db0c8ee 100644 --- a/src/memcat.c +++ b/src/memcat.c @@ -1,9 +1,10 @@ #include +#include +#include #include #include #include -#include "client_options.h" #include "utilities.h" #define PROGRAM_NAME "memcat" @@ -15,25 +16,35 @@ void options_parse(int argc, char *argv[]); static int opt_verbose= 0; static int opt_displayflag= 0; -static char *opt_servers; +static char *opt_servers= NULL; +static char *opt_hash= NULL; int main(int argc, char *argv[]) { memcached_st *memc; char *string; size_t string_length; - uint16_t flags; + uint32_t flags; memcached_return rc; memcached_server_st *servers; options_parse(argc, argv); if (!opt_servers) - return 0; + { + char *temp; + + if ((temp= getenv("MEMCACHED_SERVERS"))) + opt_servers= strdup(temp); + else + exit(1); + } memc= memcached_create(NULL); + process_hash_option(memc, opt_hash); + + servers= memcached_servers_parse(opt_servers); - servers= parse_opt_servers(opt_servers); memcached_server_push(memc, servers); memcached_server_list_free(servers); @@ -58,17 +69,23 @@ int main(int argc, char *argv[]) free(string); } } - else + else if (rc != MEMCACHED_NOTFOUND) { - fprintf(stderr, "memcat: %s: memcache error %s\n", + fprintf(stderr, "memcat: %s: memcache error %s", argv[optind], memcached_strerror(memc, rc)); + if (memc->cached_errno) + fprintf(stderr, " system error %s", strerror(memc->cached_errno)); + fprintf(stderr, "\n"); } optind++; } memcached_free(memc); - free(opt_servers); + if (opt_servers) + free(opt_servers); + if (opt_hash) + free(opt_hash); return 0; } @@ -92,6 +109,7 @@ void options_parse(int argc, char *argv[]) {"debug", no_argument, &opt_verbose, OPT_DEBUG}, {"servers", required_argument, NULL, OPT_SERVERS}, {"flag", no_argument, &opt_displayflag, OPT_FLAG}, + {"hash", required_argument, NULL, OPT_HASH}, {0, 0, 0, 0}, }; @@ -118,6 +136,9 @@ void options_parse(int argc, char *argv[]) case OPT_SERVERS: /* --servers or -s */ opt_servers= strdup(optarg); break; + case OPT_HASH: + opt_hash= strdup(optarg); + break; case '?': /* getopt_long already printed an error message. */ exit(1);