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
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
|
/*
* Merkle-Damgard Hash Function
* (C) 1999-2008 Jack Lloyd
*
* Distributed under the terms of the Botan license
*/
#include <botan/mdx_hash.h>
#include <botan/exceptn.h>
#include <botan/loadstor.h>
namespace Botan {
/*
* MDx_HashFunction Constructor
*/
MDx_HashFunction::MDx_HashFunction(size_t block_len,
bool byte_end,
bool bit_end,
size_t cnt_size) :
buffer(block_len),
BIG_BYTE_ENDIAN(byte_end),
BIG_BIT_ENDIAN(bit_end),
COUNT_SIZE(cnt_size)
{
count = position = 0;
}
/*
* Clear memory of sensitive data
*/
void MDx_HashFunction::clear()
{
zeroise(buffer);
count = position = 0;
}
/*
* Update the hash
*/
void MDx_HashFunction::add_data(const byte input[], size_t length)
{
count += length;
if(position)
{
buffer.copy(position, input, length);
if(position + length >= buffer.size())
{
compress_n(&buffer[0], 1);
input += (buffer.size() - position);
length -= (buffer.size() - position);
position = 0;
}
}
const size_t full_blocks = length / buffer.size();
const size_t remaining = length % buffer.size();
if(full_blocks)
compress_n(input, full_blocks);
buffer.copy(position, input + full_blocks * buffer.size(), remaining);
position += remaining;
}
/*
* Finalize a hash
*/
void MDx_HashFunction::final_result(byte output[])
{
buffer[position] = (BIG_BIT_ENDIAN ? 0x80 : 0x01);
for(size_t i = position+1; i != buffer.size(); ++i)
buffer[i] = 0;
if(position >= buffer.size() - COUNT_SIZE)
{
compress_n(&buffer[0], 1);
zeroise(buffer);
}
write_count(&buffer[buffer.size() - COUNT_SIZE]);
compress_n(&buffer[0], 1);
copy_out(output);
clear();
}
/*
* Write the count bits to the buffer
*/
void MDx_HashFunction::write_count(byte out[])
{
if(COUNT_SIZE < 8)
throw Invalid_State("MDx_HashFunction::write_count: COUNT_SIZE < 8");
if(COUNT_SIZE >= output_length() || COUNT_SIZE >= hash_block_size())
throw Invalid_Argument("MDx_HashFunction: COUNT_SIZE is too big");
const u64bit bit_count = count * 8;
if(BIG_BYTE_ENDIAN)
store_be(bit_count, out + COUNT_SIZE - 8);
else
store_le(bit_count, out + COUNT_SIZE - 8);
}
}
|
