diff options
-rw-r--r-- | include/GMPHashFunction.h | 42 | ||||
-rw-r--r-- | include/HashFunction.h | 80 | ||||
-rw-r--r-- | lib/GMPHashFunction.cc | 31 | ||||
-rw-r--r-- | lib/HashFunction.cc | 145 |
4 files changed, 298 insertions, 0 deletions
diff --git a/include/GMPHashFunction.h b/include/GMPHashFunction.h new file mode 100644 index 0000000..3aa6738 --- /dev/null +++ b/include/GMPHashFunction.h @@ -0,0 +1,42 @@ +/* + * Baltisot + * Copyright (C) 1999-2007 Nicolas "Pixel" Noble + * + * This program is free software; you can redistribute it and/or modify + * it under the terms of the GNU General Public License as published by + * the Free Software Foundation; either version 2 of the License, or + * (at your option) any later version. + * + * This program is distributed in the hope that it will be useful, + * but WITHOUT ANY WARRANTY; without even the implied warranty of + * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the + * GNU General Public License for more details. + * + * You should have received a copy of the GNU General Public License + * along with this program; if not, write to the Free Software + * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA + */ + +/* $Id: GMPHashFunction.h,v 1.1 2007-09-05 14:11:54 pixel Exp $ */ + +#ifndef __GMPHASHFUNCTION_H__ +#define __GMPHASHFUNCTION_H__ + +#include <gmpxx.h> +#include <HashFunction.h> + +class GMPHashFunction : public HashFunction { + public: + mpz_class GFinish(); +}; + +class GMPSHA1 : public GMPHashFunction, SHA1 { +}; + +class GMPSHA256 : public GMPHashFunction, SHA1 { +}; + +class GMPMD5 : public GMPHashFunction, SHA1 { +}; + +#endif diff --git a/include/HashFunction.h b/include/HashFunction.h new file mode 100644 index 0000000..66c185b --- /dev/null +++ b/include/HashFunction.h @@ -0,0 +1,80 @@ +/* + * Baltisot + * Copyright (C) 1999-2007 Nicolas "Pixel" Noble + * + * This program is free software; you can redistribute it and/or modify + * it under the terms of the GNU General Public License as published by + * the Free Software Foundation; either version 2 of the License, or + * (at your option) any later version. + * + * This program is distributed in the hope that it will be useful, + * but WITHOUT ANY WARRANTY; without even the implied warranty of + * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the + * GNU General Public License for more details. + * + * You should have received a copy of the GNU General Public License + * along with this program; if not, write to the Free Software + * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA + */ + +/* $Id: HashFunction.h,v 1.1 2007-09-05 14:11:54 pixel Exp $ */ + +#ifndef __HASHFUNCTION_H__ +#define __HASHFUNCTION_H__ + +#include <Exceptions.h> + +class HashFunction : public Base { + public: + HashFunction(); + void Update(const String &); + void Update(const unsigned char * data, size_t len); + String Finish(); + size_t RFinish(Uint8 * digest); + protected: + size_t finish(); + virtual void update(const unsigned char * data, size_t len) = 0; + virtual size_t finish(Uint8 * digest) = 0; + Uint8 digest[256]; + size_t l; + private: + bool finished; +}; + +class SHA1 : public HashFunction { + public: + SHA1(); + virtual ~SHA1(); + size_t size(); + protected: + virtual void update(const unsigned char * data, size_t len); + virtual size_t finish(Uint8 * digest); + private: + void * opaque; +}; + +class SHA256 : public HashFunction { + public: + SHA256(); + virtual ~SHA256(); + size_t size(); + protected: + virtual void update(const unsigned char * data, size_t len); + virtual size_t finish(Uint8 * digest); + private: + void * opaque; +}; + +class MD5 : public HashFunction { + public: + MD5(); + virtual ~MD5(); + size_t size(); + protected: + virtual void update(const unsigned char * data, size_t len); + virtual size_t finish(Uint8 * digest); + private: + void * opaque; +}; + +#endif diff --git a/lib/GMPHashFunction.cc b/lib/GMPHashFunction.cc new file mode 100644 index 0000000..f7a0535 --- /dev/null +++ b/lib/GMPHashFunction.cc @@ -0,0 +1,31 @@ +/* + * Baltisot + * Copyright (C) 1999-2007 Nicolas "Pixel" Noble + * + * This program is free software; you can redistribute it and/or modify + * it under the terms of the GNU General Public License as published by + * the Free Software Foundation; either version 2 of the License, or + * (at your option) any later version. + * + * This program is distributed in the hope that it will be useful, + * but WITHOUT ANY WARRANTY; without even the implied warranty of + * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the + * GNU General Public License for more details. + * + * You should have received a copy of the GNU General Public License + * along with this program; if not, write to the Free Software + * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA + */ + +/* $Id: GMPHashFunction.cc,v 1.1 2007-09-05 14:11:54 pixel Exp $ */ + +#include <GMPHashFunction.h> + +mpz_class GMPHashFunction::GFinish() { + size_t l = finish(); + mpz_t mp; + + mpz_import(mp, l, 1, 1, 1, 0, digest); + + return mpz_class(mp); +} diff --git a/lib/HashFunction.cc b/lib/HashFunction.cc new file mode 100644 index 0000000..1904f49 --- /dev/null +++ b/lib/HashFunction.cc @@ -0,0 +1,145 @@ +/* + * Baltisot + * Copyright (C) 1999-2007 Nicolas "Pixel" Noble + * + * This program is free software; you can redistribute it and/or modify + * it under the terms of the GNU General Public License as published by + * the Free Software Foundation; either version 2 of the License, or + * (at your option) any later version. + * + * This program is distributed in the hope that it will be useful, + * but WITHOUT ANY WARRANTY; without even the implied warranty of + * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the + * GNU General Public License for more details. + * + * You should have received a copy of the GNU General Public License + * along with this program; if not, write to the Free Software + * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA + */ + +/* $Id: HashFunction.cc,v 1.1 2007-09-05 14:11:54 pixel Exp $ */ + +#include <sha1.h> +#include <sha256.h> +#include <md5.h> + +#include <HashFunction.h> + +static const char hconv[] = "0123456789ABCDEF"; + +HashFunction::HashFunction() { + finished = false; + l = 0; +} + +void HashFunction::Update(const String & str) { + update((unsigned char *) str.to_charp(), str.strlen()); +} + +void HashFunction::Update(const unsigned char * data, size_t len) { + update(data, len); +} + +String HashFunction::Finish() { + char digest_r[513]; + int i; + + if (!finished) + l = finish(digest); + finished = true; + + for (i = 0; i < l; i++) { + digest_r[i * 2 + 0] = hconv[digest[i] >> 4]; + digest_r[i * 2 + 1] = hconv[digest[i] % 16]; + } + digest_r[l * 2] = 0; + + return digest_r; +} + +size_t HashFunction::RFinish(Uint8 * _digest) { + if (!finished) + l = finish(digest); + finished = true; + + memcpy(_digest, digest, l); + + return l; +} + +size_t HashFunction::finish() { + if (!finished) + l = finish(digest); + finished = true; + + return l; +} + +SHA1::SHA1() { + opaque = (sha1_context *) malloc(sizeof(sha1_context)); + sha1_starts((sha1_context *) opaque); +} + +SHA1::~SHA1() { + free(opaque); +} + +void SHA1::update(const unsigned char * data, size_t len) { + sha1_update((sha1_context *) opaque, data, len); +} + +size_t SHA1::finish(Uint8 * digest) { + sha1_finish((sha1_context *) opaque, digest); + + return size(); +} + +size_t SHA1::size() { + return 20; +} + +SHA256::SHA256() { + opaque = (sha256_context *) malloc(sizeof(sha256_context)); + sha256_starts((sha256_context *) opaque); +} + +SHA256::~SHA256() { + free(opaque); +} + +void SHA256::update(const unsigned char * data, size_t len) { + sha256_update((sha256_context *) opaque, data, len); +} + +size_t SHA256::finish(Uint8 * digest) { + sha256_finish((sha256_context *) opaque, digest); + + return size(); +} + +size_t SHA256::size() { + return 32; +} + +MD5::MD5() { + opaque = (md5_context *) malloc(sizeof(md5_context)); + md5_starts((md5_context *) opaque); +} + +MD5::~MD5() { + free(opaque); +} + +void MD5::update(const unsigned char * data, size_t len) { + md5_update((md5_context *) opaque, data, len); +} + +size_t MD5::finish(Uint8 * digest) { + md5_finish((md5_context *) opaque, digest); + + return size(); +} + +size_t MD5::size() { + return 16; +} |