Fix for md5

[awesomized/libmemcached] / clients / memtouch.cc

diff --git a/clients/memtouch.cc b/clients/memtouch.cc
index 4f3be80e8a9b57b68c6d1408a85369ee2ae3f5d3..19213569c5a1a2fff23e583575bd69cd87236ef3 100644 (file)
--- a/clients/memtouch.cc
+++ b/clients/memtouch.cc
@@ -1,4 +1,5 @@
 /* LibMemcached
+ * Copyright (C) 2011-2012 Data Differential, http://datadifferential.com/
  * Copyright (C) 2006-2009 Brian Aker
  * All rights reserved.
  *
@@ -9,14 +10,16 @@
  *
  */
 
-#include <config.h>
+#include <mem_config.h>
 
+#include <cerrno>
 #include <cstdio>
 #include <cstring>
 #include <getopt.h>
 #include <iostream>
 #include <unistd.h>
-#include <libmemcached/memcached.h>
+
+#include <libmemcached-1.0/memcached.h>
 
 #include "utilities.h"
 
@@ -211,7 +214,13 @@ void options_parse(int argc, char *argv[])
       break;
 
     case OPT_EXPIRE:
+      errno= 0;
       expiration= time_t(strtoul(optarg, (char **)NULL, 10));
+      if (errno != 0)
+      {
+        fprintf(stderr, "Invalid value for --expire: %s\n", optarg);
+        exit(EXIT_FAILURE);
+      }
       break;
 
     case OPT_QUIET: