summaryrefslogtreecommitdiff
path: root/src
diff options
context:
space:
mode:
Diffstat (limited to 'src')
-rw-r--r--src/multihash.cpp85
-rw-r--r--src/multihash.h37
-rw-r--r--src/multihash_wrapper.cpp14
3 files changed, 136 insertions, 0 deletions
diff --git a/src/multihash.cpp b/src/multihash.cpp
new file mode 100644
index 0000000..64dff76
--- /dev/null
+++ b/src/multihash.cpp
@@ -0,0 +1,85 @@
+#include "multihash.h"
+#include "crc32.h"
+
+#include <stdexcept>
+
+namespace Hex {
+ static char* digits = "0123456789abcdef";
+ std::string hex(char* bin, int length) {
+ std::string s(length * 2, ' ');
+ for(int i = 0; i < length; i++) {
+ s[i*2] = digits[(bin[i] >> 4) & 0xf];
+ s[i*2+1] = digits[bin[i] & 0xf];
+ }
+ return s;
+ }
+ std::string hex(int bin) {
+ std::string s(sizeof(int) * 2, ' ');
+ for(int i = 0; i < sizeof(int) * 2; i++) {
+ s[sizeof(int) * 2 - 1 - i] = digits[bin & 0xf];
+ bin = bin >> 4;
+ }
+ return s;
+ }
+}
+
+Multihash::Multihash() {
+ finished = false;
+
+ crc32_ctx = 0;
+ crc32_str = "";
+
+ ed2k_str = "";
+
+ MD5_Init(&md5_ctx);
+ md5_str = "";
+
+ SHA1_Init(&sha1_ctx);
+ sha1_str = "";
+}
+
+void Multihash::update(std::string data) {
+ if(finished) {
+ throw std::runtime_error("Can't update after digest.");
+ }
+ crc32_ctx = CRC32::crc32(crc32_ctx, data.c_str(), data.length());
+ ed2k_ctx.update(data.c_str(), data.length());
+ MD5_Update(&md5_ctx, data.c_str(), data.length());
+ SHA1_Update(&sha1_ctx, data.c_str(), data.length());
+}
+
+std::string Multihash::crc32() {
+ return Hex::hex(crc32_ctx);
+}
+
+std::string Multihash::ed2k() {
+ if(!ed2k_str.length()) {
+ finished = true;
+ char* digest = ed2k_ctx.digest();
+ ed2k_str = Hex::hex(digest, 16);
+ delete digest;
+ }
+ return ed2k_str;
+}
+
+std::string Multihash::md5() {
+ if(!md5_str.length()) {
+ finished = true;
+ char* digest = new char[16];
+ MD5_Final((unsigned char*)digest, &md5_ctx);
+ md5_str = Hex::hex(digest, 16);
+ delete digest;
+ }
+ return md5_str;
+}
+
+std::string Multihash::sha1() {
+ if(!sha1_str.length()) {
+ finished = true;
+ char* digest = new char[20];
+ SHA1_Final((unsigned char*)digest, &sha1_ctx);
+ sha1_str = Hex::hex(digest, 20);
+ delete digest;
+ }
+ return sha1_str;
+}
diff --git a/src/multihash.h b/src/multihash.h
new file mode 100644
index 0000000..5af339f
--- /dev/null
+++ b/src/multihash.h
@@ -0,0 +1,37 @@
+#ifndef _MULTIHASH_H_
+#define _MULTIHASH_H_
+
+#include <string>
+
+#include "ed2k.h"
+
+#include <openssl/md4.h>
+#include <openssl/md5.h>
+#include <openssl/sha.h>
+
+class Multihash {
+ private:
+ bool finished;
+
+ int crc32_ctx;
+ std::string crc32_str;
+
+ Ed2k ed2k_ctx;
+ std::string ed2k_str;
+
+ MD5_CTX md5_ctx;
+ std::string md5_str;
+
+ SHA_CTX sha1_ctx;
+ std::string sha1_str;
+
+ public:
+ Multihash();
+ void update(std::string data);
+ std::string crc32();
+ std::string ed2k();
+ std::string md5();
+ std::string sha1();
+};
+
+#endif // _HASH_H_
diff --git a/src/multihash_wrapper.cpp b/src/multihash_wrapper.cpp
new file mode 100644
index 0000000..aa6f615
--- /dev/null
+++ b/src/multihash_wrapper.cpp
@@ -0,0 +1,14 @@
+#include "multihash.h"
+
+#include <boost/python.hpp>
+using namespace boost::python;
+
+BOOST_PYTHON_MODULE(_multihash)
+{
+ class_<Multihash>("Multihash")
+ .def("update", &Multihash::update)
+ .def("crc32", &Multihash::crc32)
+ .def("ed2k", &Multihash::ed2k)
+ .def("md5", &Multihash::md5)
+ .def("sha1", &Multihash::sha1);
+}