using System;
using System.Security.Cryptography;
using System.Text;
using NewLife;
using Xunit;

namespace XUnitTest.Security
{
    public class SecurityHelperTests
    {
        [Fact]
        public void DES_Test()
        {
            var text = "111111";
            var key = "16621235";

            var des = DES.Create();
            var text2 = des.Encrypt(text.GetBytes(), key.GetBytes(), CipherMode.ECB).ToBase64();
            Assert.Equal("kgAdQRZ6w20=", text2);

            var des2 = DES.Create();
            var text3 = des2.Decrypt(text2.ToBase64(), key.GetBytes(), CipherMode.ECB).ToStr();
            Assert.Equal(text, text3);
        }

        [Fact]
        public void AES_CBC_Test()
        {
            var buf = "123456".ToHex();
            var key = "86CAD727DEB54263B73960AA79C5D9B7".ToHex();
            var data = "";

            // CBC加密解密
            {
                var aes = Aes.Create();
                data = aes.Encrypt(buf, key, CipherMode.CBC, PaddingMode.PKCS7).ToHex();

                Assert.Equal("50A7CF869354EC317327671B34543AD8", data);
            }
            {
                var aes = Aes.Create();
                data = aes.Decrypt(data.ToHex(), key, CipherMode.CBC, PaddingMode.PKCS7).ToHex();

                Assert.Equal("123456", data);
            }
        }

        [Fact]
        public void AES_CBC_Test2()
        {
            var buf = "123456".ToHex();
            var key = "86CAD727DEB54263B73960AA79C5D9B7".ToHex();
            var data = "";

            // 直接CBC加密解密
            {
                // CBC需要两边一致的IV
                var aes = Aes.Create();
                aes.Key = key;
                aes.IV = key;
                aes.Mode = CipherMode.CBC;
                aes.Padding = PaddingMode.PKCS7;
                var transform = aes.CreateEncryptor();
                data = transform.TransformFinalBlock(buf, 0, buf.Length).ToHex();

                Assert.Equal("50A7CF869354EC317327671B34543AD8", data);
            }
            {
                buf = data.ToHex();
                var aes = Aes.Create();
                aes.Key = key;
                aes.IV = key;
                aes.Mode = CipherMode.CBC;
                aes.Padding = PaddingMode.PKCS7;
                var transform = aes.CreateDecryptor();
                data = transform.TransformFinalBlock(buf, 0, buf.Length).ToHex();

                Assert.Equal("123456", data);
            }
        }

    [Fact(Skip = "仅开发使用")]
    public void AES_CBC_Test3()
    {
        var plain = "1234567890123456789012345678901234567890123456789012345678901234567890".GetBytes();
        var key = "86CAD727DEB54263B73960AA79C5D9B7".ToHex();
        var cipher = "EOo1EmuK4dJLS4Sb8tD73G68lqLr2yrwq3HtmNKLGnNduAH7yeybrmaEJWAiyKQ+yYWgAKAJ9YSLjfhRURHQT9oBNa101DeRWENSywO4XRA=";

        // 直接ECB加密解密
        {
            var aes = Aes.Create();
            var data = aes.Encrypt(plain, key, CipherMode.CBC, PaddingMode.PKCS7);

            Assert.Equal(cipher, data.ToBase64());
        }
        {
            var aes = Aes.Create();
            var data = aes.Decrypt(cipher.ToBase64(), key, CipherMode.CBC, PaddingMode.PKCS7);

            Assert.Equal(plain.ToStr(), data.ToStr());
        }

        // 内部加密解密
        {
            var aes = Aes.Create();
            aes.Key = key;
            aes.IV = key;
            aes.Mode = CipherMode.CBC;
            aes.Padding = PaddingMode.PKCS7;
            var data = aes.CreateEncryptor().Transform(plain);

            Assert.Equal(cipher, data.ToBase64());
        }
        {
            var aes = Aes.Create();
            aes.Key = key;
            aes.IV = key;
            aes.Mode = CipherMode.CBC;
            aes.Padding = PaddingMode.PKCS7;
            var data = aes.CreateDecryptor().Transform(cipher.ToBase64());

            Assert.Equal(plain.ToStr(), data.ToStr());
        }
    }

    [Fact]
    public void AES_ECB_Test()
    {
        var buf = "123456".ToHex();
        var key = "86CAD727DEB54263B73960AA79C5D9B7".ToHex();
        var data = "";

            // ECB加密解密
            {
                var aes = Aes.Create();
                data = aes.Encrypt(buf, key, CipherMode.ECB, PaddingMode.PKCS7).ToHex();

                Assert.Equal("C0041698AD73EADC75FB886CEF6385A5", data);
            }
            {
                var aes = Aes.Create();
                data = aes.Decrypt(data.ToHex(), key, CipherMode.ECB, PaddingMode.PKCS7).ToHex();

                Assert.Equal("123456", data);
            }
        }

        [Fact]
        public void AES_ECB_Test2()
        {
            var buf = "123456".ToHex();
            var key = "86CAD727DEB54263B73960AA79C5D9B7".ToHex();
            var data = "";

            // 直接ECB加密解密
            {
                var aes = Aes.Create();
                aes.Key = key;
                aes.Mode = CipherMode.ECB;
                aes.Padding = PaddingMode.PKCS7;
                var transform = aes.CreateEncryptor();
                data = transform.TransformFinalBlock(buf, 0, buf.Length).ToHex();

                Assert.Equal("C0041698AD73EADC75FB886CEF6385A5", data);
            }
            {
                buf = data.ToHex();
                var aes = Aes.Create();
                aes.Key = key;
                aes.Mode = CipherMode.ECB;
                aes.Padding = PaddingMode.PKCS7;
                var transform = aes.CreateDecryptor();
                //data = transform.TransformFinalBlock(buf, 0, buf.Length).ToHex();

                var buf2 = new Byte[buf.Length];
                var rs = transform.TransformBlock(buf, 0, buf.Length, buf2, 0);
                var buf3 = transform.TransformFinalBlock(new Byte[1024], 0, 0);
                data = buf3.ToHex();

                Assert.Equal("123456", data);
            }
            {
                var aes = Aes.Create();
                data = aes.Decrypt(buf, key, CipherMode.ECB, PaddingMode.PKCS7).ToHex();

            Assert.Equal("123456", data);
        }
    }

    [Fact(Skip = "仅开发使用")]
    public void AES_ECB_Test3()
    {
        var plain = "1234567890123456789012345678901234567890123456789012345678901234567890".GetBytes();
        var key = "86CAD727DEB54263B73960AA79C5D9B7".ToHex();
        var cipher = "qzPmHdAnD1wW4ujt6NEYPHy5oIn6NuuosQtW3srs3z0/uzIf5O3o637qjWyWeJqiHgLjkH8SXx1rETj3Nxu+Lclz+qK5qkwYW1OOK13Ip3Y=";

        // 直接ECB加密解密
        {
            var aes = Aes.Create();
            var data = aes.Encrypt(plain, key, CipherMode.ECB, PaddingMode.PKCS7);

            Assert.Equal(cipher, data.ToBase64());
        }
        {
            var aes = Aes.Create();
            var data = aes.Decrypt(cipher.ToBase64(), key, CipherMode.ECB, PaddingMode.PKCS7);

            Assert.Equal(plain.ToStr(), data.ToStr());
        }

        // 内部加密解密
        {
            var aes = Aes.Create();
            aes.Key = key;
            aes.IV = key;
            aes.Mode = CipherMode.ECB;
            aes.Padding = PaddingMode.PKCS7;
            var data = aes.CreateEncryptor().Transform(plain);

            Assert.Equal(cipher, data.ToBase64());
        }
        {
            var aes = Aes.Create();
            aes.Key = key;
            aes.IV = key;
            aes.Mode = CipherMode.ECB;
            aes.Padding = PaddingMode.PKCS7;
            var data = aes.CreateDecryptor().Transform(cipher.ToBase64());

            Assert.Equal(plain.ToStr(), data.ToStr());
        }
    }
}