githubmirror/shadowsocks-windows

shadowsocks-csharp/Encryption/RNG.cs

50 строк1 KB

Syrone Wong

add AEAD support

31 мар 2017, 16:36

31 мар 2017, 16:363de1409

using System;
using System.Security.Cryptography;

namespace Shadowsocks.Encryption
{
    public static class RNG
    {
        private static RNGCryptoServiceProvider _rng = null;

        public static void Init()
        {
            if (_rng == null)
                _rng = new RNGCryptoServiceProvider();
        }

        public static void Close()
        {
            if (_rng == null) return;
            _rng.Dispose();
            _rng = null;
        }

        public static void Reload()
        {
            Close();
            Init();
        }

        public static void GetBytes(byte[] buf)
        {
            GetBytes(buf, buf.Length);
        }

        public static void GetBytes(byte[] buf, int len)
        {
            if (_rng == null) Reload();
            try
            {
                _rng.GetBytes(buf, 0, len);
            }
            catch (System.Exception)
            {
                // the backup way
                byte[] tmp = new byte[len];
                _rng.GetBytes(tmp);
                Buffer.BlockCopy(tmp, 0, buf, 0, len);
            }
        }
    }
}