From df9c15d40a50ed17baa7bdbb11ef6cca9d410cf3 Mon Sep 17 00:00:00 2001 From: zyp Date: Mon, 8 May 2006 13:23:20 +0000 Subject: [project @ zyp-20060508132320-2c118d1a7e2505db] [project @ 24] Converted from make to distutils. --- src/ed2k.cpp | 41 +++++++++++++++++++++++++++++++++++++++++ 1 file changed, 41 insertions(+) create mode 100644 src/ed2k.cpp (limited to 'src/ed2k.cpp') diff --git a/src/ed2k.cpp b/src/ed2k.cpp new file mode 100644 index 0000000..92e7b15 --- /dev/null +++ b/src/ed2k.cpp @@ -0,0 +1,41 @@ +#include "ed2k.h" + +template +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); + } + int size = min(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 digest; +} -- cgit v1.2.3