X-Git-Url: https://git.m6w6.name/?a=blobdiff_plain;f=lib%2Fmemcached_delete.c;h=b50aaef0f9be5c7fdb4c459d0ee79b31f0c3f22d;hb=3e08799c9d80873a66e7e0b08c6524b0d5c47c28;hp=d24c0f82a0b7d16ffff58e922a09a05f97817010;hpb=875200dc0c5668bba4f8a68b487f9d2f2a1eedb0;p=m6w6%2Flibmemcached diff --git a/lib/memcached_delete.c b/lib/memcached_delete.c index d24c0f82..b50aaef0 100644 --- a/lib/memcached_delete.c +++ b/lib/memcached_delete.c @@ -3,7 +3,7 @@ memcached_return memcached_delete(memcached_st *ptr, char *key, size_t key_length, time_t expiration) { - size_t send_length; + size_t send_length, sent_length; memcached_return rc; char buffer[MEMCACHED_DEFAULT_COMMAND_SIZE]; unsigned int server_key; @@ -23,7 +23,12 @@ memcached_return memcached_delete(memcached_st *ptr, char *key, size_t key_lengt send_length= snprintf(buffer, MEMCACHED_DEFAULT_COMMAND_SIZE, "delete %.*s\r\n", (int)key_length, key); - if ((write(ptr->hosts[server_key].fd, buffer, send_length) == -1)) + if (send_length >= MEMCACHED_DEFAULT_COMMAND_SIZE) + return MEMCACHED_WRITE_FAILURE; + + sent_length= write(ptr->hosts[server_key].fd, buffer, send_length); + + if (sent_length == -1 || sent_length != send_length) return MEMCACHED_WRITE_FAILURE; return memcached_response(ptr, buffer, MEMCACHED_DEFAULT_COMMAND_SIZE, server_key);