1// Copyright (c) 2016-present The Bitcoin Core developers
2// Distributed under the MIT software license, see the accompanying
3// file COPYING or http://www.opensource.org/licenses/mit-license.php.
4
5#include <crypto/siphash.h>
6
7#include <uint256.h>
8
9#include <cassert>
10#include <span>
11
12CSipHasher::CSipHasher(uint64_t k0, uint64_t k1) : m_state{k0, k1} {}
13
14CSipHasher& CSipHasher::Write(uint64_t data)
15{
16    assert(m_count % 8 == 0);
17    m_state.Compress2(data);
18    m_count += 8;
19    return *this;
20}
21
22CSipHasher& CSipHasher::Write(std::span<const unsigned char> data)
23{
24    SipHashState state{m_state.Copy()};
25    uint64_t t{m_tmp};
26    uint8_t c{m_count};
27
28    while (data.size() > 0) {
29        t |= uint64_t{data.front()} << (8 * (c % 8));
30        c++;
31        if ((c & 7) == 0) {
32            state.Compress2(t);
33            t = 0;
34        }
35        data = data.subspan(1);
36    }
37
38    m_state = state;
39    m_count = c;
40    m_tmp = t;
41
42    return *this;
43}
44
45uint64_t CSipHasher::Finalize() const
46{
47    return m_state.Copy()
48                  .Compress2(m_tmp | (uint64_t{m_count} << 56))
49                  .Finalize4();
50}
51
52SipHasher13UJ& SipHasher13UJ::Write(uint64_t data) noexcept
53{
54    m_state.Compress1(data);
55    return *this;
56}
57
58SipHasher13UJ& SipHasher13UJ::WriteJumbo(const uint256& hash) noexcept
59{
60    m_state.Compress1Jumbo(hash);
61    return *this;
62}
63
64uint64_t SipHasher13UJ::Finalize() const noexcept
65{
66    return m_state.Copy().Finalize3U();
67}
68
69uint64_t PresaltedSipHasher::operator()(const uint256& val) const noexcept
70{
71    return m_state.Copy()
72                  .Compress2(val.GetUint64(0))
73                  .Compress2(val.GetUint64(1))
74                  .Compress2(val.GetUint64(2))
75                  .Compress2(val.GetUint64(3))
76                  .Compress2(uint64_t{32} << 56)
77                  .Finalize4();
78}
79
80uint64_t PresaltedSipHasher::operator()(const uint256& val, uint32_t extra) const noexcept
81{
82    return m_state.Copy()
83                  .Compress2(val.GetUint64(0))
84                  .Compress2(val.GetUint64(1))
85                  .Compress2(val.GetUint64(2))
86                  .Compress2(val.GetUint64(3))
87                  .Compress2((uint64_t{36} << 56) | extra)
88                  .Finalize4();
89}
90