robot/include/asio2/util/sha1.hpp

/*
*****
sha1.hpp is a repackaging of the sha1.cpp and sha1.h files from the smallsha1
library (http://code.google.com/p/smallsha1/) into a single header suitable for
use as a header only library. This conversion was done by Peter Thorson
(webmaster@zaphoyd.com) in 2013. All modifications to the code are redistributed
under the same license as the original, which is listed below.
*****

 Copyright (c) 2011, Micael Hildenborg
 All rights reserved.

 Redistribution and use in source and binary forms, with or without
 modification, are permitted provided that the following conditions are met:
    * Redistributions of source code must retain the above copyright
      notice, this list of conditions and the following disclaimer.
    * Redistributions in binary form must reproduce the above copyright
      notice, this list of conditions and the following disclaimer in the
      documentation and/or other materials provided with the distribution.
    * Neither the name of Micael Hildenborg nor the
      names of its contributors may be used to endorse or promote products
      derived from this software without specific prior written permission.

 THIS SOFTWARE IS PROVIDED BY Micael Hildenborg ''AS IS'' AND ANY
 EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
 WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
 DISCLAIMED. IN NO EVENT SHALL Micael Hildenborg BE LIABLE FOR ANY
 DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
 (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
 LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
 ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
 (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
 SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 */

#ifndef __ASIO2_SHA1_IMPL_HPP__
#define __ASIO2_SHA1_IMPL_HPP__

#include <cstring>
#include <cstdint>
#include <cstddef>

#include <string>

namespace asio2
{
	class sha1
	{
	protected:
		// Rotate an integer value to left.
		inline unsigned int rol(unsigned int value, unsigned int steps) {
			return ((value << steps) | (value >> (32 - steps)));
		}

		// Sets the first 16 integers in the buffert to zero.
		// Used for clearing the W buffert.
		inline void clearWBuffert(unsigned int * buffert)
		{
			for (int pos = 16; --pos >= 0;)
			{
				buffert[pos] = 0;
			}
		}

		inline void innerHash(unsigned int * result, unsigned int * w)
		{
			unsigned int a = result[0];
			unsigned int b = result[1];
			unsigned int c = result[2];
			unsigned int d = result[3];
			unsigned int e = result[4];

			int round = 0;

#define ASIO2_SHA1MACRO(func,val) \
			{ \
				const unsigned int t = rol(a, 5) + (func) + e + val + w[round]; \
				e = d; \
				d = c; \
				c = rol(b, 30); \
				b = a; \
				a = t; \
			}

			while (round < 16)
			{
				ASIO2_SHA1MACRO((b & c) | (~b & d), 0x5a827999)
					++round;
			}
			while (round < 20)
			{
				w[round] = rol((w[round - 3] ^ w[round - 8] ^ w[round - 14] ^ w[round - 16]), 1);
				ASIO2_SHA1MACRO((b & c) | (~b & d), 0x5a827999)
					++round;
			}
			while (round < 40)
			{
				w[round] = rol((w[round - 3] ^ w[round - 8] ^ w[round - 14] ^ w[round - 16]), 1);
				ASIO2_SHA1MACRO(b ^ c ^ d, 0x6ed9eba1)
					++round;
			}
			while (round < 60)
			{
				w[round] = rol((w[round - 3] ^ w[round - 8] ^ w[round - 14] ^ w[round - 16]), 1);
				ASIO2_SHA1MACRO((b & c) | (b & d) | (c & d), 0x8f1bbcdc)
					++round;
			}
			while (round < 80)
			{
				w[round] = rol((w[round - 3] ^ w[round - 8] ^ w[round - 14] ^ w[round - 16]), 1);
				ASIO2_SHA1MACRO(b ^ c ^ d, 0xca62c1d6)
					++round;
			}

#undef ASIO2_SHA1MACRO

			result[0] += a;
			result[1] += b;
			result[2] += c;
			result[3] += d;
			result[4] += e;
		}

		/// Calculate a SHA1 hash
		/**
		 * @param src points to any kind of data to be hashed.
		 * @param bytelength the number of bytes to hash from the src pointer.
		 * @param hash should point to a buffer of at least 20 bytes of size for storing
		 * the sha1 result in.
		 */
		inline void calc(void const * src, size_t bytelength, unsigned char * hash) {
			// Init the result array.
			unsigned int result[5] = { 0x67452301, 0xefcdab89, 0x98badcfe,
									   0x10325476, 0xc3d2e1f0 };

			// Cast the void src pointer to be the byte array we can work with.
			unsigned char const * sarray = (unsigned char const *)src;

			// The reusable round buffer
			unsigned int w[80];

			// Loop through all complete 64byte blocks.

			size_t endCurrentBlock;
			size_t currentBlock = 0;

			if (bytelength >= 64) {
				size_t const endOfFullBlocks = bytelength - 64;

				while (currentBlock <= endOfFullBlocks) {
					endCurrentBlock = currentBlock + 64;

					// Init the round buffer with the 64 byte block data.
					for (int roundPos = 0; currentBlock < endCurrentBlock; currentBlock += 4)
					{
						// This line will swap endian on big endian and keep endian on
						// little endian.
						w[roundPos++] = (unsigned int)sarray[currentBlock + 3]
							| (((unsigned int)sarray[currentBlock + 2]) << 8)
							| (((unsigned int)sarray[currentBlock + 1]) << 16)
							| (((unsigned int)sarray[currentBlock]) << 24);
					}
					innerHash(result, w);
				}
			}

			// Handle the last and not full 64 byte block if existing.
			endCurrentBlock = bytelength - currentBlock;
			clearWBuffert(w);
			size_t lastBlockBytes = 0;
			for (; lastBlockBytes < endCurrentBlock; ++lastBlockBytes) {
				w[lastBlockBytes >> 2] |= (unsigned int)sarray[lastBlockBytes + currentBlock] << ((3 - (lastBlockBytes & 3)) << 3);
			}

			w[lastBlockBytes >> 2] |= 0x80 << ((3 - (lastBlockBytes & 3)) << 3);
			if (endCurrentBlock >= 56) {
				innerHash(result, w);
				clearWBuffert(w);
			}
			w[15] = static_cast<unsigned int>(bytelength << 3);
			innerHash(result, w);

			// Store hash in result pointer, and make sure we get in in the correct
			// order on both endian models.
			for (int hashByte = 20; --hashByte >= 0;) {
				hash[hashByte] = (result[hashByte >> 2] >> (((3 - hashByte) & 0x3) << 3)) & 0xff;
			}
		}

		unsigned char hash_[20];

	public:
		/**
		 * @construct Construct a sha1 object with a std::string.
		 */
		sha1(const std::string & message)
		{
			calc((void const *)message.data(), message.size(), hash_);
		}

		/**
		 * @construct Construct a sha1 object with a char pointer.
		 */
		sha1(const char * message)
		{
			calc((const void*)message, std::strlen(message), hash_);
		}

		/**
		 * @construct Construct a sha1 object with a unsigned char pointer.
		 */
		sha1(const void * message, std::size_t size)
		{
			calc((const void*)message, size, hash_);
		}

		/* Convert digest to std::string value */
		std::string str(bool upper = false)
		{
			/* Hex numbers. */
			char hex_upper[16] = {
				'0', '1', '2', '3',
				'4', '5', '6', '7',
				'8', '9', 'A', 'B',
				'C', 'D', 'E', 'F'
			};
			char hex_lower[16] = {
				'0', '1', '2', '3',
				'4', '5', '6', '7',
				'8', '9', 'a', 'b',
				'c', 'd', 'e', 'f'
			};

			std::string str;
			str.reserve(20 << 1);
			for (std::size_t i = 0; i < 20; ++i)
			{
				int t = hash_[i];
				int a = t / 16;
				int b = t % 16;
				str.append(1, upper ? hex_upper[a] : hex_lower[a]);
				str.append(1, upper ? hex_upper[b] : hex_lower[b]);
			}
			return str;
		}
	};

} // namespace asio2

#endif // __ASIO2_SHA1_IMPL_HPP__