File: System\ServiceModel\Channels\DuplicateMessageDetector.cs
Project: ndp\cdf\src\NetFx40\System.ServiceModel.Channels\System.ServiceModel.Channels.csproj (System.ServiceModel.Channels)
//------------------------------------------------------------
// Copyright (c) Microsoft Corporation.  All rights reserved.
//------------------------------------------------------------
 
namespace System.ServiceModel.Channels
{
    using System;
    using System.Runtime;
    using System.Security.Cryptography;
 
    sealed class DuplicateMessageDetector : IDisposable
    {
        HashAlgorithm hashAlgorithm;
 
        [Fx.Tag.Cache(typeof(string), Fx.Tag.CacheAttrition.PartialPurgeOnEachAccess, SizeLimit = "maxListLength parameter to constructor")]
        DuplicateDetector<string> duplicateDetector;
        [Fx.Tag.SynchronizationObject()]
        object thisLock;
        bool disposed;
 
        public DuplicateMessageDetector(int maxListLength)
        {
            Fx.Assert(maxListLength > 0, "maxListLength must be > 0");
 
            this.disposed = false;
            this.hashAlgorithm = HashAlgorithm.Create();
            this.thisLock = new object();
 
            this.duplicateDetector = new DuplicateDetector<string>(maxListLength);
        }
 
        public bool IsDuplicate(ArraySegment<byte> msgBytes, out string hashString)
        {
            Fx.Assert(msgBytes != null, "messageBytes can't be null");
            Fx.Assert(msgBytes.Count > 0, "messageBytes.Count must be > 0");
 
            byte[] hash;
 
            bool notDuplicate = true;
            lock (this.thisLock)
            {
                if (disposed)
                {
                    throw FxTrace.Exception.AsError(new ObjectDisposedException(this.GetType().ToString()));
                }
 
                hash = this.hashAlgorithm.ComputeHash(msgBytes.Array, msgBytes.Offset, msgBytes.Count);
            }
 
            hashString = Convert.ToBase64String(hash);
 
            Fx.Assert(string.IsNullOrEmpty(hashString) == false, "computed hashstring is null or empty");
 
            lock (this.thisLock)
            {
                //the act of retreiving an existing item pushes it to the front of the MRU list, ensuring
                //that the oldest hashes are trimmed first when we hit the max length.
                notDuplicate = this.duplicateDetector.AddIfNotDuplicate(hashString);
            }
 
            return !notDuplicate;
        }
 
        public void RemoveEntry(string msgHash)
        {
            Fx.Assert(!string.IsNullOrEmpty(msgHash), "Message hash should never be null or empty");
 
            lock (this.thisLock)
            {
                if (this.disposed)
                {
                    throw FxTrace.Exception.AsError(new ObjectDisposedException(this.GetType().ToString()));
                }
 
                this.duplicateDetector.Remove(msgHash);
            }
        }
 
        public void Dispose()
        {
            if (this.disposed)
            {
                return;
            }
 
            lock (this.thisLock)
            {
                if (!this.disposed)
                {
                    this.disposed = true;
                    if (this.duplicateDetector != null)
                    {
                        this.duplicateDetector.Clear();
                    }
 
                    this.hashAlgorithm.Clear();
                    this.hashAlgorithm = null;
                }
            }
        }
    }
}