From 75c771a6f6e51c18775e4db8045ef3b6dead8779 Mon Sep 17 00:00:00 2001 From: Brian Aker Date: Wed, 1 Jul 2009 09:18:18 -0700 Subject: [PATCH] Work inspired by matt knox --- libmemcached/memcached_hash.c | 11 +++++++---- 1 file changed, 7 insertions(+), 4 deletions(-) diff --git a/libmemcached/memcached_hash.c b/libmemcached/memcached_hash.c index 5deb5c75..e3249c06 100644 --- a/libmemcached/memcached_hash.c +++ b/libmemcached/memcached_hash.c @@ -170,12 +170,15 @@ uint32_t memcached_generate_hash(memcached_st *ptr, const char *key, size_t key_ if (ptr->flags & MEM_HASH_WITH_PREFIX_KEY) { - int temp_len= ptr->prefix_key_length + key_length; - char *temp= (char *)malloc(temp_len); + size_t temp_length= ptr->prefix_key_length + key_length; + char temp[temp_length]; + + if (temp_length > MEMCACHED_MAX_KEY -1) + return 0; + strncpy(temp, ptr->prefix_key, ptr->prefix_key_length); strncpy(temp + ptr->prefix_key_length, key, key_length); - hash= generate_hash(ptr, temp, temp_len); - free(temp); + hash= generate_hash(ptr, temp, temp_length); } else { -- 2.30.2