X-Git-Url: https://git.m6w6.name/?a=blobdiff_plain;f=lib%2Fmemcached_hash.c;h=073b0c5d81b606f6a9f7730dd326b448eee5c88e;hb=ffe371b7498a26133a892591c7dc56d3e6212d10;hp=a8ecebf5c419cb912778e899a08c9cc6687ef246;hpb=c93c9ddbc37c21f33ab152b5121bb69baf437ad1;p=awesomized%2Flibmemcached

diff --git a/lib/memcached_hash.c b/lib/memcached_hash.c
index a8ecebf5..073b0c5d 100644
--- a/lib/memcached_hash.c
+++ b/lib/memcached_hash.c
@@ -1,6 +1,30 @@
 #include "common.h"
 
-unsigned int memcached_generate_hash(char *key, size_t key_length)
+/* Prototypes */
+static unsigned int internal_generate_hash(char *key, size_t key_length);
+static uint32_t internal_generate_md5(char *key, size_t key_length);
+
+unsigned int memcached_generate_hash(memcached_st *ptr, char *key, size_t key_length)
+{
+  uint32_t hash;
+
+  if (ptr->flags & MEM_USE_MD5)
+    hash= internal_generate_md5(key, key_length);
+  else if (ptr->flags & MEM_USE_CRC)
+    hash= hash_crc32(key, key_length);
+  else
+    hash= internal_generate_hash(key, key_length);
+
+  if (ptr->flags & MEM_USE_KETAMA)
+  {
+    WATCHPOINT_ASSERT(0);
+    return 0;
+  }
+  else
+    return hash % ptr->number_of_hosts;
+}
+
+static uint32_t internal_generate_hash(char *key, size_t key_length)
 {
   char *ptr= key;
   unsigned int value= 0;
@@ -17,3 +41,15 @@ unsigned int memcached_generate_hash(char *key, size_t key_length)
 
   return value == 0 ? 1 : value;
 }
+
+static uint32_t internal_generate_md5(char *key, size_t key_length)
+{
+  unsigned char results[16];
+
+  md5_signature((unsigned char*)key, (unsigned int)key_length, results);
+
+  return (uint32_t)(( results[3] << 24 )
+                    | ( results[2] << 16 )
+                    | ( results[1] <<  8 )
+                    |   results[0] );
+}