Aes256Encrypt.cs

using System;
using System.Security.Cryptography;
using Waher.Script.Abstraction.Elements;
using Waher.Script.Exceptions;
using Waher.Script.Model;
using Waher.Script.Objects;
 
namespace Waher.Script.Cryptography.Functions.Encryption
{
    public class Aes256Encrypt : FunctionMultiVariate
    {
        public Aes256Encrypt(ScriptNode Content, ScriptNode Key, ScriptNode IV, int Start, int Length, Expression Expression)
            : base(new ScriptNode[] { Content, Key, IV }, argumentTypes3Normal, Start, Length, Expression)
        {
        }
 
        public Aes256Encrypt(ScriptNode Content, ScriptNode Key, ScriptNode IV, ScriptNode CipherMode, int Start, int Length, Expression Expression)
            : base(new ScriptNode[] { Content, Key, IV, CipherMode }, argumentTypes4Normal, Start, Length, Expression)
        {
        }
 
        public Aes256Encrypt(ScriptNode Content, ScriptNode Key, ScriptNode IV, ScriptNode CipherMode, ScriptNode PaddingMode, int Start, int Length, Expression Expression)
            : base(new ScriptNode[] { Content, Key, IV, CipherMode, PaddingMode }, argumentTypes5Normal, Start, Length, Expression)
        {
        }
 
        public override string FunctionName => nameof(Aes256Encrypt);
 
        public override string[] DefaultArgumentNames => new string[] { "Content", "Key", "IV", "CipherMode", "PaddingMode" };
 
        public override IElement Evaluate(IElement[] Arguments, Variables Variables)
        {
            if (!(Arguments[0].AssociatedObjectValue is byte[] Data))
                throw new ScriptRuntimeException("Data to encrypt must be binary (i.e. an array of bytes).", this);
 
            if (!(Arguments[1].AssociatedObjectValue is byte[] Key))
                throw new ScriptRuntimeException("Key to use for encryption must be binary (i.e. an array of bytes).", this);
 
            if (!(Arguments[2].AssociatedObjectValue is byte[] IV))
                throw new ScriptRuntimeException("Initiation Vector to use for encryption must be binary (i.e. an array of bytes).", this);
 
            int c = Arguments.Length;
            CipherMode CipherMode = c <= 3 ? CipherMode.CBC : this.ToEnum<CipherMode>(Arguments[3]);
            PaddingMode PaddingMode = c <= 4 ? PaddingMode.PKCS7 : this.ToEnum<PaddingMode>(Arguments[4]);
 
            return new ObjectValue(Encrypt(Data, Key, IV, CipherMode, PaddingMode));
        }
 
        public static byte[] Encrypt(byte[] Data, byte[] Key, byte[] IV, CipherMode CipherMode = CipherMode.CBC, PaddingMode PaddingMode = PaddingMode.PKCS7)
        {
            if (Data is null)
                throw new ArgumentNullException(nameof(Data));
 
            if (Key is null)
                throw new ArgumentNullException(nameof(Key));
 
            if (IV is null)
                throw new ArgumentNullException(nameof(IV));
 
            using (Aes Aes = Aes.Create())
            {
                Aes.BlockSize = 128;
                Aes.KeySize = 256;
                Aes.Mode = CipherMode;
                Aes.Padding = PaddingMode;
 
                using (ICryptoTransform Transform = Aes.CreateEncryptor(Key, IV))
                {
                    return Transform.TransformFinalBlock(Data, 0, Data.Length);
                }
            }
        }
 
    }
}