TestCounter.cs

using System.Threading.Tasks;
using System.Xml;
using Waher.Content.Xml.Attributes;
using Waher.Networking.HTTP.Interfaces;
using Waher.Persistence;
using Waher.Runtime.Counters;
using Waher.Script;
 
namespace Waher.Security.WAF.Model.Actions
{
    public class TestCounter : WafActions
    {
        private readonly StringAttribute name;
        private readonly Int64Attribute delta;
        private readonly Int64Attribute above;
        private readonly Int64Attribute below;
 
        public TestCounter()
            : base()
        {
        }
 
        public TestCounter(XmlElement Xml, WafAction Parent, WebApplicationFirewall Document)
            : base(Xml, Parent, Document)
        {
            this.name = new StringAttribute(Xml, "name");
            this.delta = new Int64Attribute(Xml, "delta");
            this.above = new Int64Attribute(Xml, "above");
            this.below = new Int64Attribute(Xml, "below");
        }
 
        public override string LocalName => nameof(TestCounter);
 
        public override WafAction Create(XmlElement Xml, WafAction Parent, WebApplicationFirewall Document) => new TestCounter(Xml, Parent, Document);
 
        public override async Task<WafResult?> Review(ProcessingState State)
        {
            Variables Variables = State.Variables;
            CaseInsensitiveString Name = await this.name.EvaluateAsync(Variables, string.Empty);
            long Delta = await this.delta.EvaluateAsync(Variables, 0L);
            long Count, Above, Below;
 
            if (Delta == 0)
                Count = await RuntimeCounters.GetCount(Name);
            else
                Count = await RuntimeCounters.IncrementCounter(Name, Delta);
 
            if (this.above.IsEmpty)
            {
                if (this.below.IsEmpty)
                    return null;
 
                Below = await this.below.EvaluateAsync(Variables, 0L);
 
                if (Count < Below)
                    return await this.ReviewChildren(State);
            }
            else
            {
                Above = await this.above.EvaluateAsync(Variables, 0L);
 
                if (this.below.IsEmpty)
                {
                    if (Count > Above)
                        return await this.ReviewChildren(State);
                }
                else
                {
                    Below = await this.below.EvaluateAsync(Variables, 0L);
 
                    if (Above < Below)
                    {
                        if (Count > Above && Count < Below)
                            return await this.ReviewChildren(State);
                    }
                    else if (Above > Below)
                    {
                        if (Count > Above || Count < Below)
                            return await this.ReviewChildren(State);
                    }
                }
            }
 
            return null;
        }
    }
}