1 | // |
2 | // SHA1Engine.cpp |
3 | // |
4 | // Library: Foundation |
5 | // Package: Crypt |
6 | // Module: SHA1Engine |
7 | // |
8 | // Based on the public domain implementation by Peter C. Gutmann |
9 | // on 2 Sep 1992, modified by Carl Ellison to be SHA-1. |
10 | // |
11 | // Copyright (c) 2004-2006, Applied Informatics Software Engineering GmbH. |
12 | // and Contributors. |
13 | // |
14 | // SPDX-License-Identifier: BSL-1.0 |
15 | // |
16 | |
17 | |
18 | #include "Poco/SHA1Engine.h" |
19 | #include "Poco/ByteOrder.h" |
20 | #include <cstring> |
21 | |
22 | |
23 | #ifdef POCO_ARCH_LITTLE_ENDIAN |
24 | #define SHA1_BYTE_REVERSE(x, y) byteReverse(x, y) |
25 | #else |
26 | #define SHA1_BYTE_REVERSE(x, y) |
27 | #endif |
28 | |
29 | |
30 | namespace Poco { |
31 | |
32 | |
33 | SHA1Engine::SHA1Engine() |
34 | { |
35 | _digest.reserve(16); |
36 | reset(); |
37 | } |
38 | |
39 | |
40 | SHA1Engine::~SHA1Engine() |
41 | { |
42 | reset(); |
43 | } |
44 | |
45 | |
46 | inline void SHA1Engine::byteReverse(UInt32* buffer, int byteCount) |
47 | { |
48 | #ifdef POCO_ARCH_LITTLE_ENDIAN |
49 | byteCount /= sizeof(UInt32); |
50 | for(int count = 0; count < byteCount; count++) |
51 | { |
52 | buffer[count] = Poco::ByteOrder::toBigEndian(buffer[ count ]); |
53 | } |
54 | #endif // POCO_ARCH_LITTLE_ENDIAN |
55 | } |
56 | |
57 | |
58 | void SHA1Engine::updateImpl(const void* buffer_, std::size_t count) |
59 | { |
60 | const BYTE* buffer = (const BYTE*) buffer_; |
61 | BYTE* db = (BYTE*) &_context.data[0]; |
62 | |
63 | /* Update bitcount */ |
64 | if ((_context.countLo + ((UInt32) count << 3)) < _context.countLo) |
65 | _context.countHi++; /* Carry from low to high bitCount */ |
66 | _context.countLo += ((UInt32) count << 3); |
67 | _context.countHi += ((UInt32 ) count >> 29); |
68 | |
69 | /* Process data in BLOCK_SIZE chunks */ |
70 | while (count-- > 0) |
71 | { |
72 | db[_context.slop++] = *(buffer++); |
73 | if (_context.slop == BLOCK_SIZE) |
74 | { |
75 | /* transform this one block */ |
76 | SHA1_BYTE_REVERSE(_context.data, BLOCK_SIZE); |
77 | transform(); |
78 | _context.slop = 0 ; /* no slop left */ |
79 | } |
80 | } |
81 | } |
82 | |
83 | |
84 | std::size_t SHA1Engine::digestLength() const |
85 | { |
86 | return DIGEST_SIZE; |
87 | } |
88 | |
89 | |
90 | void SHA1Engine::reset() |
91 | { |
92 | _context.digest[0] = 0x67452301L; |
93 | _context.digest[1] = 0xEFCDAB89L; |
94 | _context.digest[2] = 0x98BADCFEL; |
95 | _context.digest[3] = 0x10325476L; |
96 | _context.digest[4] = 0xC3D2E1F0L; |
97 | _context.countLo = 0; |
98 | _context.countHi = 0; |
99 | _context.slop = 0; |
100 | std::memset(_context.data, 0, sizeof(_context.data)); |
101 | } |
102 | |
103 | |
104 | const DigestEngine::Digest& SHA1Engine::digest() |
105 | { |
106 | int count; |
107 | UInt32 lowBitcount = _context.countLo; |
108 | UInt32 highBitcount = _context.countHi; |
109 | |
110 | /* Compute number of bytes mod 64 */ |
111 | count = (int) ((_context.countLo >> 3) & 0x3F); |
112 | |
113 | /* Set the first char of padding to 0x80. This is safe since there is |
114 | always at least one byte free */ |
115 | ((BYTE*) _context.data)[count++] = 0x80; |
116 | |
117 | /* Pad out to 56 mod 64 */ |
118 | if (count > 56) |
119 | { |
120 | /* Two lots of padding: Pad the first block to 64 bytes */ |
121 | std::memset((BYTE*) &_context.data + count, 0, 64 - count); |
122 | SHA1_BYTE_REVERSE(_context.data, BLOCK_SIZE); |
123 | transform(); |
124 | |
125 | /* Now fill the next block with 56 bytes */ |
126 | std::memset(&_context.data, 0, 56); |
127 | } |
128 | else |
129 | { |
130 | /* Pad block to 56 bytes */ |
131 | std::memset((BYTE*) &_context.data + count, 0, 56 - count); |
132 | } |
133 | SHA1_BYTE_REVERSE(_context.data, BLOCK_SIZE); |
134 | |
135 | /* Append length in bits and transform */ |
136 | _context.data[14] = highBitcount; |
137 | _context.data[15] = lowBitcount; |
138 | |
139 | transform(); |
140 | SHA1_BYTE_REVERSE(_context.data, DIGEST_SIZE); |
141 | |
142 | unsigned char hash[DIGEST_SIZE]; |
143 | for (count = 0; count < DIGEST_SIZE; count++) |
144 | hash[count] = (BYTE) ((_context.digest[count>>2]) >> (8*(3-(count & 0x3)))) & 0xff; |
145 | _digest.clear(); |
146 | _digest.insert(_digest.begin(), hash, hash + DIGEST_SIZE); |
147 | reset(); |
148 | return _digest; |
149 | } |
150 | |
151 | |
152 | /* The SHA f()-functions */ |
153 | #define f1(x,y,z) ( ( x & y ) | ( ~x & z ) ) /* Rounds 0-19 */ |
154 | #define f2(x,y,z) ( x ^ y ^ z ) /* Rounds 20-39 */ |
155 | #define f3(x,y,z) ( ( x & y ) | ( x & z ) | ( y & z ) ) /* Rounds 40-59 */ |
156 | #define f4(x,y,z) ( x ^ y ^ z ) /* Rounds 60-79 */ |
157 | |
158 | |
159 | /* The SHA Mysterious Constants */ |
160 | #define K1 0x5A827999L /* Rounds 0-19 */ |
161 | #define K2 0x6ED9EBA1L /* Rounds 20-39 */ |
162 | #define K3 0x8F1BBCDCL /* Rounds 40-59 */ |
163 | #define K4 0xCA62C1D6L /* Rounds 60-79 */ |
164 | |
165 | |
166 | /* 32-bit rotate - kludged with shifts */ |
167 | typedef UInt32 UL; /* to save space */ |
168 | |
169 | |
170 | #define S(n,X) ( ( ((UL)X) << n ) | ( ((UL)X) >> ( 32 - n ) ) ) |
171 | |
172 | |
173 | /* The initial expanding function */ |
174 | #define expand(count) W[ count ] = S(1,(W[ count - 3 ] ^ W[ count - 8 ] ^ W[ count - 14 ] ^ W[ count - 16 ])) /* to make this SHA-1 */ |
175 | |
176 | |
177 | /* The four SHA sub-rounds */ |
178 | #define subRound1(count) \ |
179 | { \ |
180 | temp = S( 5, A ) + f1( B, C, D ) + E + W[ count ] + K1; \ |
181 | E = D; \ |
182 | D = C; \ |
183 | C = S( 30, B ); \ |
184 | B = A; \ |
185 | A = temp; \ |
186 | } |
187 | |
188 | #define subRound2(count) \ |
189 | { \ |
190 | temp = S( 5, A ) + f2( B, C, D ) + E + W[ count ] + K2; \ |
191 | E = D; \ |
192 | D = C; \ |
193 | C = S( 30, B ); \ |
194 | B = A; \ |
195 | A = temp; \ |
196 | } |
197 | |
198 | #define subRound3(count) \ |
199 | { \ |
200 | temp = S( 5, A ) + f3( B, C, D ) + E + W[ count ] + K3; \ |
201 | E = D; \ |
202 | D = C; \ |
203 | C = S( 30, B ); \ |
204 | B = A; \ |
205 | A = temp; \ |
206 | } |
207 | |
208 | #define subRound4(count) \ |
209 | { \ |
210 | temp = S( 5, A ) + f4( B, C, D ) + E + W[ count ] + K4; \ |
211 | E = D; \ |
212 | D = C; \ |
213 | C = S( 30, B ); \ |
214 | B = A; \ |
215 | A = temp; \ |
216 | } |
217 | |
218 | |
219 | void SHA1Engine::transform() |
220 | { |
221 | UInt32 W[80]; |
222 | UInt32 temp; |
223 | UInt32 A, B, C, D, E; |
224 | int i; |
225 | |
226 | /* Step A. Copy the data buffer into the local work buffer */ |
227 | for( i = 0; i < 16; i++ ) |
228 | W[ i ] = _context.data[ i ]; |
229 | |
230 | /* Step B. Expand the 16 words into 64 temporary data words */ |
231 | expand( 16 ); expand( 17 ); expand( 18 ); expand( 19 ); expand( 20 ); |
232 | expand( 21 ); expand( 22 ); expand( 23 ); expand( 24 ); expand( 25 ); |
233 | expand( 26 ); expand( 27 ); expand( 28 ); expand( 29 ); expand( 30 ); |
234 | expand( 31 ); expand( 32 ); expand( 33 ); expand( 34 ); expand( 35 ); |
235 | expand( 36 ); expand( 37 ); expand( 38 ); expand( 39 ); expand( 40 ); |
236 | expand( 41 ); expand( 42 ); expand( 43 ); expand( 44 ); expand( 45 ); |
237 | expand( 46 ); expand( 47 ); expand( 48 ); expand( 49 ); expand( 50 ); |
238 | expand( 51 ); expand( 52 ); expand( 53 ); expand( 54 ); expand( 55 ); |
239 | expand( 56 ); expand( 57 ); expand( 58 ); expand( 59 ); expand( 60 ); |
240 | expand( 61 ); expand( 62 ); expand( 63 ); expand( 64 ); expand( 65 ); |
241 | expand( 66 ); expand( 67 ); expand( 68 ); expand( 69 ); expand( 70 ); |
242 | expand( 71 ); expand( 72 ); expand( 73 ); expand( 74 ); expand( 75 ); |
243 | expand( 76 ); expand( 77 ); expand( 78 ); expand( 79 ); |
244 | |
245 | /* Step C. Set up first buffer */ |
246 | A = _context.digest[ 0 ]; |
247 | B = _context.digest[ 1 ]; |
248 | C = _context.digest[ 2 ]; |
249 | D = _context.digest[ 3 ]; |
250 | E = _context.digest[ 4 ]; |
251 | |
252 | /* Step D. Serious mangling, divided into four sub-rounds */ |
253 | subRound1( 0 ); subRound1( 1 ); subRound1( 2 ); subRound1( 3 ); |
254 | subRound1( 4 ); subRound1( 5 ); subRound1( 6 ); subRound1( 7 ); |
255 | subRound1( 8 ); subRound1( 9 ); subRound1( 10 ); subRound1( 11 ); |
256 | subRound1( 12 ); subRound1( 13 ); subRound1( 14 ); subRound1( 15 ); |
257 | subRound1( 16 ); subRound1( 17 ); subRound1( 18 ); subRound1( 19 ); |
258 | subRound2( 20 ); subRound2( 21 ); subRound2( 22 ); subRound2( 23 ); |
259 | subRound2( 24 ); subRound2( 25 ); subRound2( 26 ); subRound2( 27 ); |
260 | subRound2( 28 ); subRound2( 29 ); subRound2( 30 ); subRound2( 31 ); |
261 | subRound2( 32 ); subRound2( 33 ); subRound2( 34 ); subRound2( 35 ); |
262 | subRound2( 36 ); subRound2( 37 ); subRound2( 38 ); subRound2( 39 ); |
263 | subRound3( 40 ); subRound3( 41 ); subRound3( 42 ); subRound3( 43 ); |
264 | subRound3( 44 ); subRound3( 45 ); subRound3( 46 ); subRound3( 47 ); |
265 | subRound3( 48 ); subRound3( 49 ); subRound3( 50 ); subRound3( 51 ); |
266 | subRound3( 52 ); subRound3( 53 ); subRound3( 54 ); subRound3( 55 ); |
267 | subRound3( 56 ); subRound3( 57 ); subRound3( 58 ); subRound3( 59 ); |
268 | subRound4( 60 ); subRound4( 61 ); subRound4( 62 ); subRound4( 63 ); |
269 | subRound4( 64 ); subRound4( 65 ); subRound4( 66 ); subRound4( 67 ); |
270 | subRound4( 68 ); subRound4( 69 ); subRound4( 70 ); subRound4( 71 ); |
271 | subRound4( 72 ); subRound4( 73 ); subRound4( 74 ); subRound4( 75 ); |
272 | subRound4( 76 ); subRound4( 77 ); subRound4( 78 ); subRound4( 79 ); |
273 | |
274 | /* Step E. Build message digest */ |
275 | _context.digest[ 0 ] += A; |
276 | _context.digest[ 1 ] += B; |
277 | _context.digest[ 2 ] += C; |
278 | _context.digest[ 3 ] += D; |
279 | _context.digest[ 4 ] += E; |
280 | } |
281 | |
282 | |
283 | } // namespace Poco |
284 | |