summary refs log tree commit diff
path: root/crypto/src/pqc/crypto/lms/SeedDerive.cs
blob: 38b72fd3f2aace0b5f1007ac1858661475cb6a91 (plain) (blame)
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
using System;

using Org.BouncyCastle.Crypto;
using Org.BouncyCastle.Crypto.Utilities;

namespace Org.BouncyCastle.Pqc.Crypto.Lms
{
    public sealed class SeedDerive
    {
        private readonly byte[] m_I;
        private readonly byte[] m_masterSeed;
        private readonly IDigest m_digest;

        public SeedDerive(byte[] I, byte[] masterSeed, IDigest digest)
        {
            m_I = I;
            m_masterSeed = masterSeed;
            m_digest = digest;
        }

        public int Q { get; set; }

        public int J { get; set; }

        // FIXME
        public byte[] I => m_I;

        // FIXME
        public byte[] MasterSeed => m_masterSeed;

        public byte[] DeriveSeed(bool incJ, byte[] target, int offset)
        {
            if (target.Length - offset < m_digest.GetDigestSize())
                throw new ArgumentException("target length is less than digest size.", nameof(target));

            int q = Q, j = J;

            m_digest.BlockUpdate(I, 0, I.Length);

#if NETCOREAPP2_1_OR_GREATER || NETSTANDARD2_1_OR_GREATER
            Span<byte> qj = stackalloc byte[7];
            Pack.UInt32_To_BE((uint)q, qj);
            Pack.UInt16_To_BE((ushort)j, qj[4..]);
            qj[6] = 0xFF;
            m_digest.BlockUpdate(qj);
#else
            m_digest.Update((byte)(q >> 24));
            m_digest.Update((byte)(q >> 16));
            m_digest.Update((byte)(q >> 8));
            m_digest.Update((byte)(q));

            m_digest.Update((byte)(j >> 8));
            m_digest.Update((byte)(j));
            m_digest.Update(0xFF);
#endif

            m_digest.BlockUpdate(m_masterSeed, 0, m_masterSeed.Length);

            m_digest.DoFinal(target, offset); // Digest resets here.

            if (incJ)
            {
                ++J;
            }

            return target;
        }
    }
}