X-Git-Url: https://git.m6w6.name/?a=blobdiff_plain;f=libmemcached%2Fmemcached_auto.c;h=9808ab65e8e98f815f5f3218d4f2b91f76d1d652;hb=29c13aacae8a79791e8912ff7520601aa8136d83;hp=449d4c4e23b5f6f341986f4885143c9fe63a9366;hpb=2fe4e1635c70f4b2815aaf9bc147fee0b2c93de3;p=awesomized%2Flibmemcached diff --git a/libmemcached/memcached_auto.c b/libmemcached/memcached_auto.c index 449d4c4e..9808ab65 100644 --- a/libmemcached/memcached_auto.c +++ b/libmemcached/memcached_auto.c @@ -11,9 +11,6 @@ static memcached_return memcached_auto(memcached_st *ptr, char buffer[MEMCACHED_DEFAULT_COMMAND_SIZE]; unsigned int server_key; - unlikely (key_length == 0) - return MEMCACHED_NO_KEY_PROVIDED; - unlikely (ptr->hosts == NULL || ptr->number_of_hosts == 0) return MEMCACHED_NO_SERVERS; @@ -68,18 +65,12 @@ static memcached_return binary_incr_decr(memcached_st *ptr, uint8_t cmd, { unsigned int server_key; - unlikely (key_length == 0) - return MEMCACHED_NO_KEY_PROVIDED; - unlikely (ptr->hosts == NULL || ptr->number_of_hosts == 0) return MEMCACHED_NO_SERVERS; - if ((ptr->flags & MEM_VERIFY_KEY) && (memcachd_key_test(&key, &key_length, 1) == MEMCACHED_BAD_KEY_PROVIDED)) - return MEMCACHED_BAD_KEY_PROVIDED; - server_key= memcached_generate_hash(ptr, key, key_length); - protocol_binary_request_incr request= {0}; + protocol_binary_request_incr request= {.bytes= {0}}; request.message.header.request.magic= PROTOCOL_BINARY_REQ; request.message.header.request.opcode= cmd; @@ -98,7 +89,7 @@ static memcached_return binary_incr_decr(memcached_st *ptr, uint8_t cmd, return MEMCACHED_WRITE_FAILURE; } - return memcached_response(&ptr->hosts[server_key], value, sizeof(*value), NULL); + return memcached_response(&ptr->hosts[server_key], (char*)value, sizeof(*value), NULL); } memcached_return memcached_increment(memcached_st *ptr, @@ -106,7 +97,9 @@ memcached_return memcached_increment(memcached_st *ptr, uint32_t offset, uint64_t *value) { - memcached_return rc; + memcached_return rc= memcached_validate_key_length(key_length, ptr->flags & MEM_BINARY_PROTOCOL); + unlikely (rc != MEMCACHED_SUCCESS) + return rc; LIBMEMCACHED_MEMCACHED_INCREMENT_START(); if (ptr->flags & MEM_BINARY_PROTOCOL) @@ -125,7 +118,9 @@ memcached_return memcached_decrement(memcached_st *ptr, uint32_t offset, uint64_t *value) { - memcached_return rc; + memcached_return rc= memcached_validate_key_length(key_length, ptr->flags & MEM_BINARY_PROTOCOL); + unlikely (rc != MEMCACHED_SUCCESS) + return rc; LIBMEMCACHED_MEMCACHED_DECREMENT_START(); if (ptr->flags & MEM_BINARY_PROTOCOL)