ext/digest/murmurhash/murmurhash2.c in digest-murmurhash-0.2.3 vs ext/digest/murmurhash/murmurhash2.c in digest-murmurhash-0.3.0

@@ -2,35 +2,28 @@
  * MurmurHash2 (C) Austin Appleby
  */
 
 #include "murmurhash2.h"
 
-static inline size_t
-murmur2(uint32_t h, uint32_t k, const uint8_t r)
-{
-	const uint32_t m = MURMURHASH_MAGIC;
-	k *= m;
-	k ^= k >> r;
-	k *= m;
-
-	h *= m;
-	h ^= k;
-	return h;
-}
-
 static uint32_t
-murmur_hash_process2(const char *data, uint32_t length)
+murmur_hash_process2(const char *data, uint32_t length, uint32_t seed)
 {
 	const uint32_t m = MURMURHASH_MAGIC;
 	const uint8_t r = 24;
 	uint32_t h, k;
 
-	h = length * m;
+	h = seed ^ length;
 
 	while (4 <= length) {
 		k = *(uint32_t*)data;
-		h = murmur2(h, k, r);
+		k *= m;
+		k ^= k >> r;
+		k *= m;
+
+		h *= m;
+		h ^= k;
+
 		data += 4;
 		length -= 4;
 	}
 
 	switch (length) {
@@ -48,53 +41,31 @@
 }
 
 VALUE
 murmur2_finish(VALUE self)
 {
-	uint32_t h;
 	uint8_t digest[4];
-	MURMURHASH(self, ptr);
+	uint64_t h;
 
-	h = murmur_hash_process2(ptr->buffer, ptr->p - ptr->buffer);
-
-	digest[0] = h >> 24;
-	digest[1] = h >> 16;
-	digest[2] = h >> 8;
-	digest[3] = h;
-
+	h = _murmur_finish32(self, murmur_hash_process2);
+	ASSINE_BY_ENDIAN_32(digest, h);
 	return rb_str_new((const char*) digest, 4);
 }
 
 VALUE
 murmur2_to_i(VALUE self)
 {
-	MURMURHASH(self, ptr);
-	return UINT2NUM(murmur_hash_process2(ptr->buffer, ptr->p - ptr->buffer));
+	return ULONG2NUM(_murmur_finish32(self, murmur_hash_process2));
 }
 
 VALUE
 murmur2_s_digest(int argc, VALUE *argv, VALUE klass)
 {
-	VALUE str;
-	uint32_t h;
 	uint8_t digest[4];
-
-	if (argc < 1)
-		rb_raise(rb_eArgError, "no data given");
-
-	str = *argv++;
-	argc--;
-
-	StringValue(str);
-
-	h = murmur_hash_process2(RSTRING_PTR(str), RSTRING_LEN(str));
-
-	digest[0] = h >> 24;
-	digest[1] = h >> 16;
-	digest[2] = h >> 8;
-	digest[3] = h;
-
+	uint32_t h;
+	h = _murmur_s_digest32(argc, argv, klass, murmur_hash_process2);
+	ASSINE_BY_ENDIAN_32(digest, h);
 	return rb_str_new((const char*) digest, 4);
 }
 
 VALUE
 murmur2_s_hexdigest(int argc, VALUE *argv, VALUE klass)
@@ -103,18 +74,8 @@
 }
 
 VALUE
 murmur2_s_rawdigest(int argc, VALUE *argv, VALUE klass)
 {
-	VALUE str;
-
-	if (argc < 1)
-		rb_raise(rb_eArgError, "no data given");
-
-	str = *argv++;
-	argc--;
-
-	StringValue(str);
-
-	return UINT2NUM(murmur_hash_process2(RSTRING_PTR(str), RSTRING_LEN(str)));
+	return ULONG2NUM(_murmur_s_digest32(argc, argv, klass, murmur_hash_process2));
 }