diff options
Diffstat (limited to 'Swiften')
| -rw-r--r-- | Swiften/Avatars/AvatarManager.cpp | 5 | ||||
| -rw-r--r-- | Swiften/Disco/CapsInfoGenerator.cpp | 2 | ||||
| -rw-r--r-- | Swiften/SASL/SCRAMSHA1ClientAuthenticator.cpp | 2 | ||||
| -rw-r--r-- | Swiften/SConscript | 4 | ||||
| -rw-r--r-- | Swiften/StringCodecs/Base64.h | 8 | ||||
| -rw-r--r-- | Swiften/StringCodecs/HMACSHA1.cpp | 4 | ||||
| -rw-r--r-- | Swiften/StringCodecs/Hexify.cpp | 22 | ||||
| -rw-r--r-- | Swiften/StringCodecs/Hexify.h | 11 | ||||
| -rw-r--r-- | Swiften/StringCodecs/MD5.cpp | 359 | ||||
| -rw-r--r-- | Swiften/StringCodecs/MD5.h | 10 | ||||
| -rw-r--r-- | Swiften/StringCodecs/SHA1.cpp | 17 | ||||
| -rw-r--r-- | Swiften/StringCodecs/SHA1.h | 10 | ||||
| -rw-r--r-- | Swiften/StringCodecs/UnitTest/Base64Test.cpp | 5 | ||||
| -rw-r--r-- | Swiften/StringCodecs/UnitTest/HexifyTest.cpp | 21 | ||||
| -rw-r--r-- | Swiften/StringCodecs/UnitTest/MD5Test.cpp | 29 | ||||
| -rw-r--r-- | Swiften/StringCodecs/UnitTest/SHA1Test.cpp | 33 | 
16 files changed, 482 insertions, 60 deletions
| diff --git a/Swiften/Avatars/AvatarManager.cpp b/Swiften/Avatars/AvatarManager.cpp index 574e199..599d1d4 100644 --- a/Swiften/Avatars/AvatarManager.cpp +++ b/Swiften/Avatars/AvatarManager.cpp @@ -3,12 +3,13 @@  #include <boost/bind.hpp>  #include "Swiften/Client/StanzaChannel.h"  #include "Swiften/Elements/VCardUpdate.h"  #include "Swiften/Queries/Requests/GetVCardRequest.h"  #include "Swiften/StringCodecs/SHA1.h" +#include "Swiften/StringCodecs/Hexify.h"  #include "Swiften/Avatars/AvatarStorage.h"  #include "Swiften/MUC/MUCRegistry.h"  namespace Swift {  AvatarManager::AvatarManager(StanzaChannel* stanzaChannel, IQRouter* iqRouter, AvatarStorage* avatarStorage, MUCRegistry* mucRegistry) : stanzaChannel_(stanzaChannel), iqRouter_(iqRouter), avatarStorage_(avatarStorage), mucRegistry_(mucRegistry) { @@ -38,22 +39,22 @@ void AvatarManager::handlePresenceReceived(boost::shared_ptr<Presence> presence)  void AvatarManager::handleVCardReceived(const JID& from, const String& promisedHash, boost::shared_ptr<VCard> vCard, const boost::optional<ErrorPayload>& error) {  	if (error) {  		// FIXME: What to do here?  		std::cerr << "Warning: " << from << ": Could not get vCard" << std::endl;  		return;  	} -	String realHash = SHA1::getHexHash(vCard->getPhoto()); +	String realHash = Hexify::hexify(SHA1::getHash(vCard->getPhoto()));  	if (promisedHash != realHash) {  		std::cerr << "Warning: " << from << ": Got different vCard photo hash (" << promisedHash << " != " << realHash << ")" << std::endl;  	}  	avatarStorage_->addAvatar(realHash, vCard->getPhoto());  	setAvatarHash(from, realHash);  }  void AvatarManager::setAvatar(const JID& jid, const ByteArray& avatar) { -	String hash = SHA1::getHexHash(avatar); +	String hash = Hexify::hexify(SHA1::getHash(avatar));  	avatarStorage_->addAvatar(hash, avatar);  	setAvatarHash(getAvatarJID(jid), hash);  }  void AvatarManager::setAvatarHash(const JID& from, const String& hash) {  	avatarHashes_[from] = hash; diff --git a/Swiften/Disco/CapsInfoGenerator.cpp b/Swiften/Disco/CapsInfoGenerator.cpp index 339d76e..8d9e407 100644 --- a/Swiften/Disco/CapsInfoGenerator.cpp +++ b/Swiften/Disco/CapsInfoGenerator.cpp @@ -24,11 +24,11 @@ CapsInfo CapsInfoGenerator::generateCapsInfo(const DiscoInfo& discoInfo) const {  	std::vector<String> features(discoInfo.getFeatures());  	std::sort(features.begin(), features.end());  	foreach (const String& feature, features) {  		serializedCaps += feature + "<";  	} -	String version(Base64::encode(SHA1::getBinaryHash(serializedCaps))); +	String version(Base64::encode(SHA1::getHash(serializedCaps)));  	return CapsInfo(node_, version, "sha-1");  }  } diff --git a/Swiften/SASL/SCRAMSHA1ClientAuthenticator.cpp b/Swiften/SASL/SCRAMSHA1ClientAuthenticator.cpp index a261810..5e6179f 100644 --- a/Swiften/SASL/SCRAMSHA1ClientAuthenticator.cpp +++ b/Swiften/SASL/SCRAMSHA1ClientAuthenticator.cpp @@ -35,13 +35,13 @@ SCRAMSHA1ClientAuthenticator::SCRAMSHA1ClientAuthenticator(const String& nonce)  ByteArray SCRAMSHA1ClientAuthenticator::getResponse() const {  	if (step == Initial) {  		return getGS2Header() + getInitialBareClientMessage();  	}  	else {  		ByteArray clientKey = HMACSHA1::getResult(saltedPassword, "Client Key"); -		ByteArray storedKey = SHA1::getBinaryHash(clientKey); +		ByteArray storedKey = SHA1::getHash(clientKey);  		ByteArray clientSignature = HMACSHA1::getResult(storedKey, authMessage);  		ByteArray clientProof = clientKey;  		for (unsigned int i = 0; i < clientProof.getSize(); ++i) {  			clientProof[i] ^= clientSignature[i];  		}  		ByteArray result = ByteArray("c=") + Base64::encode(getGS2Header()) + ",r=" + clientnonce + serverNonce + ",p=" + Base64::encode(clientProof); diff --git a/Swiften/SConscript b/Swiften/SConscript index e7e1582..6c13edb 100644 --- a/Swiften/SConscript +++ b/Swiften/SConscript @@ -80,13 +80,15 @@ sources = [  		"Session/Session.cpp",  		"Session/SessionStream.cpp",  		"Session/BasicSessionStream.cpp",  		"StringCodecs/Base64.cpp",  		"StringCodecs/SHA1.cpp",  		"StringCodecs/HMACSHA1.cpp", +		"StringCodecs/MD5.cpp",  		"StringCodecs/PBKDF2.cpp", +		"StringCodecs/Hexify.cpp",  	]  # "Notifier/GrowlNotifier.cpp",  if myenv.get("HAVE_OPENSSL", 0) :  	sources += ["TLS/OpenSSL/OpenSSLContext.cpp"] @@ -187,9 +189,11 @@ env.Append(UNITTEST_SOURCES = [  		File("Serializer/XML/UnitTest/XMLElementTest.cpp"),  		File("Server/UnitTest/ServerStanzaRouterTest.cpp"),  		File("StreamStack/UnitTest/StreamStackTest.cpp"),  		File("StreamStack/UnitTest/XMPPLayerTest.cpp"),  		File("StringCodecs/UnitTest/Base64Test.cpp"),  		File("StringCodecs/UnitTest/SHA1Test.cpp"), +		File("StringCodecs/UnitTest/MD5Test.cpp"), +		File("StringCodecs/UnitTest/HexifyTest.cpp"),  		File("StringCodecs/UnitTest/HMACSHA1Test.cpp"),  		File("StringCodecs/UnitTest/PBKDF2Test.cpp"),  	]) diff --git a/Swiften/StringCodecs/Base64.h b/Swiften/StringCodecs/Base64.h index 41ff62e..1ea378c 100644 --- a/Swiften/StringCodecs/Base64.h +++ b/Swiften/StringCodecs/Base64.h @@ -1,18 +1,14 @@ -#ifndef SWIFTEN_STRINGCODECS_BASE64_H -#define SWIFTEN_STRINGCODECS_BASE64_H +#pragma once  #include <vector>  #include "Swiften/Base/String.h"  #include "Swiften/Base/ByteArray.h"  namespace Swift { -	class Base64 -	{ +	class Base64 {  		public:  			static String encode(const ByteArray& s);  			static ByteArray decode(const String &s);  	};  } - -#endif diff --git a/Swiften/StringCodecs/HMACSHA1.cpp b/Swiften/StringCodecs/HMACSHA1.cpp index 59f1482..f4066cb 100644 --- a/Swiften/StringCodecs/HMACSHA1.cpp +++ b/Swiften/StringCodecs/HMACSHA1.cpp @@ -28,12 +28,12 @@ ByteArray HMACSHA1::getResult(const ByteArray& key, const ByteArray& data) {  	// Create the second value  	ByteArray y(paddedKey);  	for (unsigned int i = 0; i < y.getSize(); ++i) {  		y[i] ^= 0x5c;  	} -	y += SHA1::getBinaryHash(x); +	y += SHA1::getHash(x); -	return SHA1::getBinaryHash(y); +	return SHA1::getHash(y);  }  } diff --git a/Swiften/StringCodecs/Hexify.cpp b/Swiften/StringCodecs/Hexify.cpp new file mode 100644 index 0000000..504e314 --- /dev/null +++ b/Swiften/StringCodecs/Hexify.cpp @@ -0,0 +1,22 @@ +#include "Swiften/StringCodecs/Hexify.h" + +#include <sstream> +#include <iomanip> +#include <boost/numeric/conversion/cast.hpp> + +#include "Swiften/Base/String.h" +#include "Swiften/Base/ByteArray.h" + +namespace Swift { + +String Hexify::hexify(const ByteArray& data) { +	std::ostringstream result; +	result << std::hex; + +	for (unsigned int i = 0; i < data.getSize(); ++i) { +		result << std::setw(2) << std::setfill('0') << boost::numeric_cast<unsigned int>(static_cast<unsigned char>(data[i])); +	} +	return String(result.str()); +} + +} diff --git a/Swiften/StringCodecs/Hexify.h b/Swiften/StringCodecs/Hexify.h new file mode 100644 index 0000000..7bd5bee --- /dev/null +++ b/Swiften/StringCodecs/Hexify.h @@ -0,0 +1,11 @@ +#pragma once + +namespace Swift { +	class String; +	class ByteArray; + +	class Hexify { +		public: +			static String hexify(const ByteArray& data); +	}; +} diff --git a/Swiften/StringCodecs/MD5.cpp b/Swiften/StringCodecs/MD5.cpp new file mode 100644 index 0000000..937d128 --- /dev/null +++ b/Swiften/StringCodecs/MD5.cpp @@ -0,0 +1,359 @@ +/*  + * This implementation is shamelessly copied from L. Peter Deutsch's  + * implementation, and altered to use our own defines and datastructures.  + * Original license below. + */ + +/* +	Copyright (C) 1999, 2002 Aladdin Enterprises.  All rights reserved. + +	This software is provided 'as-is', without any express or implied +	warranty.  In no event will the authors be held liable for any damages +	arising from the use of this software. + +	Permission is granted to anyone to use this software for any purpose, +	including commercial applications, and to alter it and redistribute it +	freely, subject to the following restrictions: + +	1. The origin of this software must not be misrepresented; you must not +		 claim that you wrote the original software. If you use this software +		 in a product, an acknowledgment in the product documentation would be +		 appreciated but is not required. +	2. Altered source versions must be plainly marked as such, and must not be +		 misrepresented as being the original software. +	3. This notice may not be removed or altered from any source distribution. + +	L. Peter Deutsch +	ghost@aladdin.com + */ + +#include "Swiften/StringCodecs/MD5.h" + +#include <cassert> + +#include "Swiften/Base/ByteArray.h" +#include "Swiften/Base/Platform.h" + +namespace Swift { + +typedef unsigned char md5_byte_t; /* 8-bit byte */ +typedef unsigned int md5_word_t; /* 32-bit word */ + +/* Define the state of the MD5 Algorithm. */ +typedef struct md5_state_s { +		md5_word_t count[2];				/* message length in bits, lsw first */ +		md5_word_t abcd[4];					/* digest buffer */ +		md5_byte_t buf[64];					/* accumulate block */ +} md5_state_t; + +#define T_MASK ((md5_word_t)~0) +#define T1 /* 0xd76aa478 */ (T_MASK ^ 0x28955b87) +#define T2 /* 0xe8c7b756 */ (T_MASK ^ 0x173848a9) +#define T3		0x242070db +#define T4 /* 0xc1bdceee */ (T_MASK ^ 0x3e423111) +#define T5 /* 0xf57c0faf */ (T_MASK ^ 0x0a83f050) +#define T6		0x4787c62a +#define T7 /* 0xa8304613 */ (T_MASK ^ 0x57cfb9ec) +#define T8 /* 0xfd469501 */ (T_MASK ^ 0x02b96afe) +#define T9		0x698098d8 +#define T10 /* 0x8b44f7af */ (T_MASK ^ 0x74bb0850) +#define T11 /* 0xffff5bb1 */ (T_MASK ^ 0x0000a44e) +#define T12 /* 0x895cd7be */ (T_MASK ^ 0x76a32841) +#define T13		 0x6b901122 +#define T14 /* 0xfd987193 */ (T_MASK ^ 0x02678e6c) +#define T15 /* 0xa679438e */ (T_MASK ^ 0x5986bc71) +#define T16		 0x49b40821 +#define T17 /* 0xf61e2562 */ (T_MASK ^ 0x09e1da9d) +#define T18 /* 0xc040b340 */ (T_MASK ^ 0x3fbf4cbf) +#define T19		 0x265e5a51 +#define T20 /* 0xe9b6c7aa */ (T_MASK ^ 0x16493855) +#define T21 /* 0xd62f105d */ (T_MASK ^ 0x29d0efa2) +#define T22		 0x02441453 +#define T23 /* 0xd8a1e681 */ (T_MASK ^ 0x275e197e) +#define T24 /* 0xe7d3fbc8 */ (T_MASK ^ 0x182c0437) +#define T25		 0x21e1cde6 +#define T26 /* 0xc33707d6 */ (T_MASK ^ 0x3cc8f829) +#define T27 /* 0xf4d50d87 */ (T_MASK ^ 0x0b2af278) +#define T28		 0x455a14ed +#define T29 /* 0xa9e3e905 */ (T_MASK ^ 0x561c16fa) +#define T30 /* 0xfcefa3f8 */ (T_MASK ^ 0x03105c07) +#define T31		 0x676f02d9 +#define T32 /* 0x8d2a4c8a */ (T_MASK ^ 0x72d5b375) +#define T33 /* 0xfffa3942 */ (T_MASK ^ 0x0005c6bd) +#define T34 /* 0x8771f681 */ (T_MASK ^ 0x788e097e) +#define T35		 0x6d9d6122 +#define T36 /* 0xfde5380c */ (T_MASK ^ 0x021ac7f3) +#define T37 /* 0xa4beea44 */ (T_MASK ^ 0x5b4115bb) +#define T38		 0x4bdecfa9 +#define T39 /* 0xf6bb4b60 */ (T_MASK ^ 0x0944b49f) +#define T40 /* 0xbebfbc70 */ (T_MASK ^ 0x4140438f) +#define T41		 0x289b7ec6 +#define T42 /* 0xeaa127fa */ (T_MASK ^ 0x155ed805) +#define T43 /* 0xd4ef3085 */ (T_MASK ^ 0x2b10cf7a) +#define T44		 0x04881d05 +#define T45 /* 0xd9d4d039 */ (T_MASK ^ 0x262b2fc6) +#define T46 /* 0xe6db99e5 */ (T_MASK ^ 0x1924661a) +#define T47		 0x1fa27cf8 +#define T48 /* 0xc4ac5665 */ (T_MASK ^ 0x3b53a99a) +#define T49 /* 0xf4292244 */ (T_MASK ^ 0x0bd6ddbb) +#define T50		 0x432aff97 +#define T51 /* 0xab9423a7 */ (T_MASK ^ 0x546bdc58) +#define T52 /* 0xfc93a039 */ (T_MASK ^ 0x036c5fc6) +#define T53		 0x655b59c3 +#define T54 /* 0x8f0ccc92 */ (T_MASK ^ 0x70f3336d) +#define T55 /* 0xffeff47d */ (T_MASK ^ 0x00100b82) +#define T56 /* 0x85845dd1 */ (T_MASK ^ 0x7a7ba22e) +#define T57		 0x6fa87e4f +#define T58 /* 0xfe2ce6e0 */ (T_MASK ^ 0x01d3191f) +#define T59 /* 0xa3014314 */ (T_MASK ^ 0x5cfebceb) +#define T60		 0x4e0811a1 +#define T61 /* 0xf7537e82 */ (T_MASK ^ 0x08ac817d) +#define T62 /* 0xbd3af235 */ (T_MASK ^ 0x42c50dca) +#define T63		 0x2ad7d2bb +#define T64 /* 0xeb86d391 */ (T_MASK ^ 0x14792c6e) + + +static void md5_process(md5_state_t *pms, const md5_byte_t *data /*[64]*/) { +		md5_word_t +		a = pms->abcd[0], b = pms->abcd[1], +		c = pms->abcd[2], d = pms->abcd[3]; +		md5_word_t t; +#ifdef SWIFTEN_BIG_ENDIAN +		/* Define storage only for big-endian CPUs. */ +		md5_word_t X[16]; +#else +		/* Define storage for little-endian or both types of CPUs. */ +		md5_word_t xbuf[16]; +		const md5_word_t *X; +#endif + +		{ +#ifdef SWIFTEN_LITTLE_ENDIAN +				{ +						/* +						 * On little-endian machines, we can process properly aligned +						 * data without copying it. +						 */ +						if (!((data - (const md5_byte_t *)0) & 3)) { +								/* data are properly aligned */ +								X = (const md5_word_t *)data; +						} else { +								/* not aligned */ +								memcpy(xbuf, data, 64); +								X = xbuf; +						} +				} +#else +				{ +						/* +						 * On big-endian machines, we must arrange the bytes in the +						 * right order. +						 */ +						const md5_byte_t *xp = data; +						int i; + +						for (i = 0; i < 16; ++i, xp += 4) +								X[i] = xp[0] + (xp[1] << 8) + (xp[2] << 16) + (xp[3] << 24); +				} +#endif +		} + +#define ROTATE_LEFT(x, n) (((x) << (n)) | ((x) >> (32 - (n)))) + +		/* Round 1. */ +		/* Let [abcd k s i] denote the operation +			 a = b + ((a + F(b,c,d) + X[k] + T[i]) <<< s). */ +#define F(x, y, z) (((x) & (y)) | (~(x) & (z))) +#define SET(a, b, c, d, k, s, Ti)\ +	t = a + F(b,c,d) + X[k] + Ti;\ +	a = ROTATE_LEFT(t, s) + b +		/* Do the following 16 operations. */ +		SET(a, b, c, d,  0,  7,  T1); +		SET(d, a, b, c,  1, 12,  T2); +		SET(c, d, a, b,  2, 17,  T3); +		SET(b, c, d, a,  3, 22,  T4); +		SET(a, b, c, d,  4,  7,  T5); +		SET(d, a, b, c,  5, 12,  T6); +		SET(c, d, a, b,  6, 17,  T7); +		SET(b, c, d, a,  7, 22,  T8); +		SET(a, b, c, d,  8,  7,  T9); +		SET(d, a, b, c,  9, 12, T10); +		SET(c, d, a, b, 10, 17, T11); +		SET(b, c, d, a, 11, 22, T12); +		SET(a, b, c, d, 12,  7, T13); +		SET(d, a, b, c, 13, 12, T14); +		SET(c, d, a, b, 14, 17, T15); +		SET(b, c, d, a, 15, 22, T16); +#undef SET + +		 /* Round 2. */ +		 /* Let [abcd k s i] denote the operation +					a = b + ((a + G(b,c,d) + X[k] + T[i]) <<< s). */ +#define G(x, y, z) (((x) & (z)) | ((y) & ~(z))) +#define SET(a, b, c, d, k, s, Ti)\ +	t = a + G(b,c,d) + X[k] + Ti;\ +	a = ROTATE_LEFT(t, s) + b +		 /* Do the following 16 operations. */ +		SET(a, b, c, d,  1,  5, T17); +		SET(d, a, b, c,  6,  9, T18); +		SET(c, d, a, b, 11, 14, T19); +		SET(b, c, d, a,  0, 20, T20); +		SET(a, b, c, d,  5,  5, T21); +		SET(d, a, b, c, 10,  9, T22); +		SET(c, d, a, b, 15, 14, T23); +		SET(b, c, d, a,  4, 20, T24); +		SET(a, b, c, d,  9,  5, T25); +		SET(d, a, b, c, 14,  9, T26); +		SET(c, d, a, b,  3, 14, T27); +		SET(b, c, d, a,  8, 20, T28); +		SET(a, b, c, d, 13,  5, T29); +		SET(d, a, b, c,  2,  9, T30); +		SET(c, d, a, b,  7, 14, T31); +		SET(b, c, d, a, 12, 20, T32); +#undef SET + +		 /* Round 3. */ +		 /* Let [abcd k s t] denote the operation +					a = b + ((a + H(b,c,d) + X[k] + T[i]) <<< s). */ +#define H(x, y, z) ((x) ^ (y) ^ (z)) +#define SET(a, b, c, d, k, s, Ti)\ +	t = a + H(b,c,d) + X[k] + Ti;\ +	a = ROTATE_LEFT(t, s) + b +		 /* Do the following 16 operations. */ +		SET(a, b, c, d,  5,  4, T33); +		SET(d, a, b, c,  8, 11, T34); +		SET(c, d, a, b, 11, 16, T35); +		SET(b, c, d, a, 14, 23, T36); +		SET(a, b, c, d,  1,  4, T37); +		SET(d, a, b, c,  4, 11, T38); +		SET(c, d, a, b,  7, 16, T39); +		SET(b, c, d, a, 10, 23, T40); +		SET(a, b, c, d, 13,  4, T41); +		SET(d, a, b, c,  0, 11, T42); +		SET(c, d, a, b,  3, 16, T43); +		SET(b, c, d, a,  6, 23, T44); +		SET(a, b, c, d,  9,  4, T45); +		SET(d, a, b, c, 12, 11, T46); +		SET(c, d, a, b, 15, 16, T47); +		SET(b, c, d, a,  2, 23, T48); +#undef SET + +		 /* Round 4. */ +		 /* Let [abcd k s t] denote the operation +					a = b + ((a + I(b,c,d) + X[k] + T[i]) <<< s). */ +#define I(x, y, z) ((y) ^ ((x) | ~(z))) +#define SET(a, b, c, d, k, s, Ti)\ +	t = a + I(b,c,d) + X[k] + Ti;\ +	a = ROTATE_LEFT(t, s) + b +		 /* Do the following 16 operations. */ +		SET(a, b, c, d,  0,  6, T49); +		SET(d, a, b, c,  7, 10, T50); +		SET(c, d, a, b, 14, 15, T51); +		SET(b, c, d, a,  5, 21, T52); +		SET(a, b, c, d, 12,  6, T53); +		SET(d, a, b, c,  3, 10, T54); +		SET(c, d, a, b, 10, 15, T55); +		SET(b, c, d, a,  1, 21, T56); +		SET(a, b, c, d,  8,  6, T57); +		SET(d, a, b, c, 15, 10, T58); +		SET(c, d, a, b,  6, 15, T59); +		SET(b, c, d, a, 13, 21, T60); +		SET(a, b, c, d,  4,  6, T61); +		SET(d, a, b, c, 11, 10, T62); +		SET(c, d, a, b,  2, 15, T63); +		SET(b, c, d, a,  9, 21, T64); +#undef SET + +		 /* Then perform the following additions. (That is increment each +				of the four registers by the value it had before this block +				was started.) */ +		pms->abcd[0] += a; +		pms->abcd[1] += b; +		pms->abcd[2] += c; +		pms->abcd[3] += d; +} + +void +md5_init(md5_state_t *pms) +{ +		pms->count[0] = pms->count[1] = 0; +		pms->abcd[0] = 0x67452301; +		pms->abcd[1] = /*0xefcdab89*/ T_MASK ^ 0x10325476; +		pms->abcd[2] = /*0x98badcfe*/ T_MASK ^ 0x67452301; +		pms->abcd[3] = 0x10325476; +} + +void +md5_append(md5_state_t *pms, const md5_byte_t *data, int nbytes) +{ +		const md5_byte_t *p = data; +		int left = nbytes; +		int offset = (pms->count[0] >> 3) & 63; +		md5_word_t nbits = (md5_word_t)(nbytes << 3); + +		if (nbytes <= 0) +				return; + +		/* Update the message length. */ +		pms->count[1] += nbytes >> 29; +		pms->count[0] += nbits; +		if (pms->count[0] < nbits) +				pms->count[1]++; + +		/* Process an initial partial block. */ +		if (offset) { +				int copy = (offset + nbytes > 64 ? 64 - offset : nbytes); + +				memcpy(pms->buf + offset, p, copy); +				if (offset + copy < 64) +						return; +				p += copy; +				left -= copy; +				md5_process(pms, pms->buf); +		} + +		/* Process full blocks. */ +		for (; left >= 64; p += 64, left -= 64) +				md5_process(pms, p); + +		/* Process a final partial block. */ +		if (left) +				memcpy(pms->buf, p, left); +} + +void +md5_finish(md5_state_t *pms, md5_byte_t digest[16]) +{ +		static const md5_byte_t pad[64] = { +				0x80, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, +				0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, +				0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, +				0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0 +		}; +		md5_byte_t data[8]; +		int i; + +		/* Save the length before padding. */ +		for (i = 0; i < 8; ++i) +				data[i] = (md5_byte_t)(pms->count[i >> 2] >> ((i & 3) << 3)); +		/* Pad to 56 bytes mod 64. */ +		md5_append(pms, pad, ((55 - (pms->count[0] >> 3)) & 63) + 1); +		/* Append the length. */ +		md5_append(pms, data, 8); +		for (i = 0; i < 16; ++i) +				digest[i] = (md5_byte_t)(pms->abcd[i >> 2] >> ((i & 3) << 3)); +} + +ByteArray MD5::getHash(const ByteArray& data) { +	ByteArray digest; +	digest.resize(16); + +	md5_state_t state; +	md5_init(&state); +	md5_append(&state, reinterpret_cast<const md5_byte_t*>(data.getData()), data.getSize()); +	md5_finish(&state, reinterpret_cast<md5_byte_t*>(digest.getData())); + +	return digest; +} + +} diff --git a/Swiften/StringCodecs/MD5.h b/Swiften/StringCodecs/MD5.h new file mode 100644 index 0000000..8773409 --- /dev/null +++ b/Swiften/StringCodecs/MD5.h @@ -0,0 +1,10 @@ +#pragma once + +namespace Swift { +	class ByteArray; + +	class MD5 { +		public: +			static ByteArray getHash(const ByteArray& data); +	}; +} diff --git a/Swiften/StringCodecs/SHA1.cpp b/Swiften/StringCodecs/SHA1.cpp index 70256e9..ef99d9a 100644 --- a/Swiften/StringCodecs/SHA1.cpp +++ b/Swiften/StringCodecs/SHA1.cpp @@ -1,10 +1,6 @@ -#include <sstream> -#include <iomanip> -#include <boost/numeric/conversion/cast.hpp> -  #include "Swiften/Base/Platform.h"  #pragma GCC diagnostic ignored "-Wold-style-cast"  /*  SHA-1 in C @@ -184,29 +180,18 @@ boost::uint8_t finalcount[8];  // -----------------------------------------------------------------------------  #include "Swiften/StringCodecs/SHA1.h"  namespace Swift { -ByteArray SHA1::getBinaryHash(const ByteArray& input) { +ByteArray SHA1::getHash(const ByteArray& input) {  	ByteArray inputCopy(input);  	ByteArray digest;  	digest.resize(20);  	SHA1_CTX context;  	SHA1Init(&context);  	SHA1Update(&context, (boost::uint8_t*) inputCopy.getData(), inputCopy.getSize());  	SHA1Final((boost::uint8_t*) digest.getData(), &context);  	return digest;  } -String SHA1::getHexHash(const ByteArray& input) { -	ByteArray digest = getBinaryHash(input); -	std::ostringstream result; -	result << std::hex; - -	for (unsigned int i = 0; i < digest.getSize(); ++i) { -		result << std::setw(2) << std::setfill('0') << boost::numeric_cast<unsigned int>(static_cast<unsigned char>(digest[i])); -	} -	return String(result.str()); -} -  } diff --git a/Swiften/StringCodecs/SHA1.h b/Swiften/StringCodecs/SHA1.h index 0b0f434..9f54e8e 100644 --- a/Swiften/StringCodecs/SHA1.h +++ b/Swiften/StringCodecs/SHA1.h @@ -1,14 +1,10 @@ -#ifndef SWIFTEN_SHA1_H -#define SWIFTEN_SHA1_H +#pragma once  #include "Swiften/Base/ByteArray.h"  namespace Swift {  	class SHA1 {  		public: -			static ByteArray getBinaryHash(const ByteArray& data); -			static String getHexHash(const ByteArray& data); -		}; +			static ByteArray getHash(const ByteArray& data); +	};  } - -#endif diff --git a/Swiften/StringCodecs/UnitTest/Base64Test.cpp b/Swiften/StringCodecs/UnitTest/Base64Test.cpp index 0217758..a28a9ab 100644 --- a/Swiften/StringCodecs/UnitTest/Base64Test.cpp +++ b/Swiften/StringCodecs/UnitTest/Base64Test.cpp @@ -2,25 +2,22 @@  #include <cppunit/extensions/TestFactoryRegistry.h>  #include "Swiften/StringCodecs/Base64.h"  using namespace Swift; -class Base64Test : public CppUnit::TestFixture -{ +class Base64Test : public CppUnit::TestFixture {  		CPPUNIT_TEST_SUITE(Base64Test);  		CPPUNIT_TEST(testEncode);  		CPPUNIT_TEST(testEncode_NonAscii);  		CPPUNIT_TEST(testEncode_NoData);  		CPPUNIT_TEST(testDecode);  		CPPUNIT_TEST(testDecode_NoData);  		CPPUNIT_TEST_SUITE_END();  	public: -		Base64Test() {} -  		void testEncode() {  			String result(Base64::encode("ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz1234567890"));  			CPPUNIT_ASSERT_EQUAL(String("QUJDREVGR0hJSktMTU5PUFFSU1RVVldYWVphYmNkZWZnaGlqa2xtbm9wcXJzdHV2d3h5ejEyMzQ1Njc4OTA="), result);  		}  		void testEncode_NonAscii() { diff --git a/Swiften/StringCodecs/UnitTest/HexifyTest.cpp b/Swiften/StringCodecs/UnitTest/HexifyTest.cpp new file mode 100644 index 0000000..bf032a3 --- /dev/null +++ b/Swiften/StringCodecs/UnitTest/HexifyTest.cpp @@ -0,0 +1,21 @@ +#include <cppunit/extensions/HelperMacros.h> +#include <cppunit/extensions/TestFactoryRegistry.h> + +#include "Swiften/StringCodecs/Hexify.h" +#include "Swiften/Base/String.h" +#include "Swiften/Base/ByteArray.h" + +using namespace Swift; + +class HexifyTest : public CppUnit::TestFixture { +		CPPUNIT_TEST_SUITE(HexifyTest); +		CPPUNIT_TEST(testHexify); +		CPPUNIT_TEST_SUITE_END(); + +	public: +		void testHexify() { +			CPPUNIT_ASSERT_EQUAL(String("4206b23ca6b0a643d20d89b04ff58cf78b8096ed"), Hexify::hexify(ByteArray("\x42\x06\xb2\x3c\xa6\xb0\xa6\x43\xd2\x0d\x89\xb0\x4f\xf5\x8c\xf7\x8b\x80\x96\xed"))); +		} +}; + +CPPUNIT_TEST_SUITE_REGISTRATION(HexifyTest); diff --git a/Swiften/StringCodecs/UnitTest/MD5Test.cpp b/Swiften/StringCodecs/UnitTest/MD5Test.cpp new file mode 100644 index 0000000..cad8754 --- /dev/null +++ b/Swiften/StringCodecs/UnitTest/MD5Test.cpp @@ -0,0 +1,29 @@ +#include <cppunit/extensions/HelperMacros.h> +#include <cppunit/extensions/TestFactoryRegistry.h> + +#include "Swiften/StringCodecs/MD5.h" +#include "Swiften/Base/ByteArray.h" + +using namespace Swift; + +class MD5Test : public CppUnit::TestFixture { +		CPPUNIT_TEST_SUITE(MD5Test); +		CPPUNIT_TEST(testGetHash_Empty); +		CPPUNIT_TEST(testGetHash_Alphabet); +		CPPUNIT_TEST_SUITE_END(); + +	public: +		void testGetHash_Empty() { +			ByteArray result(MD5::getHash("")); + +			CPPUNIT_ASSERT_EQUAL(ByteArray("\xd4\x1d\x8c\xd9\x8f\x00\xb2\x04\xe9\x80\x09\x98\xec\xf8\x42\x7e", 16), result); +		} + +		void testGetHash_Alphabet() { +			ByteArray result(MD5::getHash("ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789")); + +			CPPUNIT_ASSERT_EQUAL(ByteArray("\xd1\x74\xab\x98\xd2\x77\xd9\xf5\xa5\x61\x1c\x2c\x9f\x41\x9d\x9f", 16), result); +		} +}; + +CPPUNIT_TEST_SUITE_REGISTRATION(MD5Test); diff --git a/Swiften/StringCodecs/UnitTest/SHA1Test.cpp b/Swiften/StringCodecs/UnitTest/SHA1Test.cpp index 849dd6d..3dbf341 100644 --- a/Swiften/StringCodecs/UnitTest/SHA1Test.cpp +++ b/Swiften/StringCodecs/UnitTest/SHA1Test.cpp @@ -2,45 +2,36 @@  #include <cppunit/extensions/TestFactoryRegistry.h>  #include "Swiften/StringCodecs/SHA1.h"  using namespace Swift; -class SHA1Test : public CppUnit::TestFixture -{ +class SHA1Test : public CppUnit::TestFixture {  		CPPUNIT_TEST_SUITE(SHA1Test); -		CPPUNIT_TEST(testGetBinaryHash); -		CPPUNIT_TEST(testGetBinaryHash_Twice); -		CPPUNIT_TEST(testGetHexHash); -		CPPUNIT_TEST(testGetHexHash_NoData); +		CPPUNIT_TEST(testGetHash); +		CPPUNIT_TEST(testGetHash_Twice); +		CPPUNIT_TEST(testGetHash_NoData);  		CPPUNIT_TEST_SUITE_END();  	public: -		SHA1Test() {} - -		void testGetBinaryHash() { -			ByteArray result(SHA1::getBinaryHash("client/pc//Exodus 0.9.1<http://jabber.org/protocol/caps<http://jabber.org/protocol/disco#info<http://jabber.org/protocol/disco#items<http://jabber.org/protocol/muc<")); +		void testGetHash() { +			ByteArray result(SHA1::getHash("client/pc//Exodus 0.9.1<http://jabber.org/protocol/caps<http://jabber.org/protocol/disco#info<http://jabber.org/protocol/disco#items<http://jabber.org/protocol/muc<"));  			CPPUNIT_ASSERT_EQUAL(ByteArray("\x42\x06\xb2\x3c\xa6\xb0\xa6\x43\xd2\x0d\x89\xb0\x4f\xf5\x8c\xf7\x8b\x80\x96\xed"), result);  		} -		void testGetBinaryHash_Twice() { +		void testGetHash_Twice() {  			ByteArray input("client/pc//Exodus 0.9.1<http://jabber.org/protocol/caps<http://jabber.org/protocol/disco#info<http://jabber.org/protocol/disco#items<http://jabber.org/protocol/muc<"); -			SHA1::getBinaryHash(input); -			ByteArray result(SHA1::getBinaryHash(input)); +			SHA1::getHash(input); +			ByteArray result(SHA1::getHash(input));  			CPPUNIT_ASSERT_EQUAL(ByteArray("\x42\x06\xb2\x3c\xa6\xb0\xa6\x43\xd2\x0d\x89\xb0\x4f\xf5\x8c\xf7\x8b\x80\x96\xed"), result);  		} -		void testGetHexHash() { -			String result(SHA1::getHexHash("client/pc//Exodus 0.9.1<http://jabber.org/protocol/caps<http://jabber.org/protocol/disco#info<http://jabber.org/protocol/disco#items<http://jabber.org/protocol/muc<")); -			CPPUNIT_ASSERT_EQUAL(String("4206b23ca6b0a643d20d89b04ff58cf78b8096ed"), result); -		} - -		void testGetHexHash_NoData() { -			String result(SHA1::getHexHash(ByteArray())); +		void testGetHash_NoData() { +			ByteArray result(SHA1::getHash(ByteArray())); -			CPPUNIT_ASSERT_EQUAL(String("da39a3ee5e6b4b0d3255bfef95601890afd80709"), result); +			CPPUNIT_ASSERT_EQUAL(ByteArray("\xda\x39\xa3\xee\x5e\x6b\x4b\x0d\x32\x55\xbf\xef\x95\x60\x18\x90\xaf\xd8\x07\x09"), result);  		}  };  CPPUNIT_TEST_SUITE_REGISTRATION(SHA1Test); | 
 Swift
 Swift