using Org.BouncyCastle.Crypto.Parameters; using Org.BouncyCastle.Crypto.Utilities; using System; namespace Org.BouncyCastle.Crypto.Generators { public sealed class Mgf1BytesGenerator : IDerivationFunction { private readonly IDigest m_digest; private readonly int m_hLen; private byte[] m_buffer; public IDigest Digest => m_digest; public Mgf1BytesGenerator(IDigest digest) { m_digest = digest; m_hLen = digest.GetDigestSize(); } public void Init(IDerivationParameters parameters) { MgfParameters mgfParameters = parameters as MgfParameters; if (mgfParameters == null) throw new ArgumentException("MGF parameters required for MGF1Generator"); m_buffer = new byte[mgfParameters.SeedLength + 4 + m_hLen]; mgfParameters.GetSeed(m_buffer, 0); } public int GenerateBytes(byte[] output, int outOff, int length) { Check.OutputLength(output, outOff, length, "output buffer too short"); return GenerateBytes(output.AsSpan(outOff, length)); } public int GenerateBytes(Span<byte> output) { int num = m_buffer.Length - m_hLen; int start = num - 4; uint num2 = 0; m_digest.Reset(); int i = 0; int length = output.Length; for (int num3 = length - m_hLen; i <= num3; i += m_hLen) { Pack.UInt32_To_BE(num2++, m_buffer.AsSpan(start)); m_digest.BlockUpdate(m_buffer.AsSpan(0, num)); IDigest digest = m_digest; int num5 = i; digest.DoFinal(output.Slice(num5, output.Length - num5)); } if (i < length) { Pack.UInt32_To_BE(num2, m_buffer.AsSpan(start)); m_digest.BlockUpdate(m_buffer.AsSpan(0, num)); m_digest.DoFinal(m_buffer.AsSpan(num)); Span<byte> span = m_buffer.AsSpan(num, length - i); int num5 = i; span.CopyTo(output.Slice(num5, output.Length - num5)); } return length; } } }