From 0c4fc1e72770fc78d87891e5c031272fca59e409 Mon Sep 17 00:00:00 2001
From: zyp <zyp@localhost>
Date: Tue, 9 May 2006 20:33:27 +0000
Subject: [project @ zyp-20060509203327-c8c93b489da2ea46] [project @ 27]
 Multihash library rewrite complete.

---
 src/ed2k.cpp | 74 +++++++++++++++++++++++++++++++-----------------------------
 1 file changed, 38 insertions(+), 36 deletions(-)

(limited to 'src/ed2k.cpp')

diff --git a/src/ed2k.cpp b/src/ed2k.cpp
index 92e7b15..2d4b2f7 100644
--- a/src/ed2k.cpp
+++ b/src/ed2k.cpp
@@ -1,41 +1,43 @@
 #include "ed2k.h"
 
-template<class T>
-inline T min(T a, T b) {
-	return (a > b) ? b : a;
-}
-
-Ed2k::Ed2k() {
-	MD4_Init(&md4_partial);
-	MD4_Init(&md4_final);
-	size_total = 0;
-}
-
-void Ed2k::update(const char* data, int length) {
-	while(length) {
-		if(!(size_total % (9500 * 1024)) && size_total) {
-			unsigned char digest[16];
-			MD4_Final(digest, &md4_partial);
-			MD4_Update(&md4_final, digest, 16);
-			MD4_Init(&md4_partial);
+namespace Multihash {
+	template<class T>
+	inline T min(T a, T b) {
+		return (a > b) ? b : a;
+	}
+	
+	Ed2k::Ed2k(Hash* n) : Hash(n) {
+		MD4_Init(&md4_partial);
+		MD4_Init(&md4_final);
+		size_total = 0;
+	}
+	
+	void Ed2k::hash_update(const char* data, int length) {
+		while(length) {
+			if(!(size_total % (9500 * 1024)) && size_total) {
+				unsigned char digest[16];
+				MD4_Final(digest, &md4_partial);
+				MD4_Update(&md4_final, digest, 16);
+				MD4_Init(&md4_partial);
+			}
+			int size = min<int>(length, (9500 * 1024) - (size_total % (9500 * 1024)));
+			MD4_Update(&md4_partial, data, size);
+			length -= size;
+			data += size;
+			size_total += size;
+		};
+	}
+	
+	std::string Ed2k::hash_digest() {
+		char digest[16];
+		if(size_total > (9500 * 1024)) {
+			unsigned char digest_partial[16];
+			MD4_Final(digest_partial, &md4_partial);
+			MD4_Update(&md4_final, digest_partial, 16);
+			MD4_Final((unsigned char*)digest, &md4_final);
+		} else {
+			MD4_Final((unsigned char*)digest, &md4_partial);
 		}
-		int size = min<int>(length, (9500 * 1024) - (size_total % (9500 * 1024)));
-		MD4_Update(&md4_partial, data, size);
-		length -= size;
-		data += size;
-		size_total += size;
-	};
-}
-
-char* Ed2k::digest() {
-	char* digest = new char[16];
-	if(size_total > (9500 * 1024)) {
-		unsigned char digest_partial[16];
-		MD4_Final(digest_partial, &md4_partial);
-		MD4_Update(&md4_final, digest_partial, 16);
-		MD4_Final((unsigned char*)digest, &md4_final);
-	} else {
-		MD4_Final((unsigned char*)digest, &md4_partial);
+		return Hex::hex(digest, 16);
 	}
-	return digest;
 }
-- 
cgit v1.2.3