X-Git-Url: https://git.m6w6.name/?a=blobdiff_plain;f=libmemcached%2Fconnect.cc;h=aa2c7def7e4904c7f769d43bdf4c459be9333ee7;hb=1ed39cb2d0601d753ef54be65f24334097f48805;hp=7efdd8fe6d52f3a5b89e9ddaaf94e91d4064a557;hpb=f11bc11e720cdeb4489f2c3ef77d5eb0577f6dd3;p=awesomized%2Flibmemcached

diff --git a/libmemcached/connect.cc b/libmemcached/connect.cc
index 7efdd8fe..aa2c7def 100644
--- a/libmemcached/connect.cc
+++ b/libmemcached/connect.cc
@@ -155,11 +155,12 @@ static memcached_return_t set_hostinfo(org::libmemcached::Instance* server)
     server->address_info_next= NULL;
   }
 
-  char str_port[NI_MAXSERV];
-  int length= snprintf(str_port, NI_MAXSERV, "%u", uint32_t(server->port()));
-  if (length >= NI_MAXSERV or length <= 0)
+  char str_port[MEMCACHED_NI_MAXSERV];
+  int length= snprintf(str_port, MEMCACHED_NI_MAXSERV, "%u", uint32_t(server->port()));
+  if (length >= MEMCACHED_NI_MAXSERV or length <= 0)
   {
-    return MEMCACHED_FAILURE;
+    return memcached_set_error(*server, MEMCACHED_MEMORY_ALLOCATION_FAILURE, MEMCACHED_AT, 
+                               memcached_literal_param("snprintf(NI_MAXSERV)"));
   }
 
   struct addrinfo hints;