summaryrefslogtreecommitdiffstats
blob: 28a0e05a66f4d3866b9d7d788e2841a4329171c9 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
/*
 * Copyright (c) 2010 Remko Tronçon
 * Licensed under the GNU General Public License v3.
 * See Documentation/Licenses/GPLv3.txt for more information.
 */

#pragma once

#include <vector>
#include <boost/cstdint.hpp>

#include <Swiften/Base/ByteArray.h>
#include <Swiften/Base/SafeByteArray.h>

namespace Swift {
	class SHA256 {
		public:
			SHA256();

			SHA256& update(const std::vector<unsigned char>& data);
			std::vector<unsigned char> getHash() const;

			/**
			 * Equivalent of:
			 *	SHA256().update(data),getHash(), but slightly more efficient and
			 *	convenient.
			 */
			static ByteArray getHash(const ByteArray& data);
			static ByteArray getHash(const SafeByteArray& data);

			ByteArray operator()(const SafeByteArray& data) {
				return getHash(data);
			}

			ByteArray operator()(const ByteArray& data) {
				return getHash(data);
			}

		private:
			struct State {
					boost::uint64_t length;
					boost::uint32_t state[8], curlen;
					unsigned char buf[64];
			};

			static int init(State *md);
			static int process(State * md, const unsigned char *in, unsigned int inlen);
			static int compress(State *md, unsigned char *buf);
			static int done(State * md, unsigned char *out);

			template<typename Container> static	ByteArray getHashInternal(const Container& input);

		private:
			State state;
	};
}