X-Git-Url: https://git.m6w6.name/?a=blobdiff_plain;f=lib%2Fmemcached_auto.c;h=acb554b149bd4dd220ce0eaf3899681503152f96;hb=806525899ce3dff894b829416307a4290cd882e2;hp=268f95b8b5b6474c4a5505f56c62233fe7b88a45;hpb=3efc2e6fb6a17e8eb52dbe3590a5dcb063b30537;p=awesomized%2Flibmemcached diff --git a/lib/memcached_auto.c b/lib/memcached_auto.c index 268f95b8..acb554b1 100644 --- a/lib/memcached_auto.c +++ b/lib/memcached_auto.c @@ -6,7 +6,7 @@ static memcached_return memcached_auto(memcached_st *ptr, unsigned int offset, unsigned int *value) { - size_t send_length; + size_t send_length, sent_length; memcached_return rc; char buffer[MEMCACHED_DEFAULT_COMMAND_SIZE]; unsigned int server_key; @@ -20,15 +20,14 @@ static memcached_return memcached_auto(memcached_st *ptr, send_length= snprintf(buffer, MEMCACHED_DEFAULT_COMMAND_SIZE, "%s %.*s %u\r\n", verb, - key_length, key, + (int)key_length, key, offset); + if (send_length >= MEMCACHED_DEFAULT_COMMAND_SIZE) + return MEMCACHED_WRITE_FAILURE; + sent_length= write(ptr->hosts[server_key].fd, buffer, send_length); - if ((send(ptr->hosts[server_key].fd, buffer, send_length, 0) == -1)) - { - fprintf(stderr, "failed set on %.*s TCP\n", key_length+1, key); - + if (sent_length == -1 || sent_length != send_length) return MEMCACHED_WRITE_FAILURE; - } memset(buffer, 0, MEMCACHED_DEFAULT_COMMAND_SIZE); send_length= read(ptr->hosts[server_key].fd, buffer, MEMCACHED_DEFAULT_COMMAND_SIZE);