Imported Upstream version 4.6.0.125

Former-commit-id: a2155e9bd80020e49e72e86c44da02a8ac0e57a4

mcs/class/referencesource/System.ServiceModel/System/ServiceModel/Channels/MsmqInputMessage.cs

@@ -0,0 +1,176 @@
+//------------------------------------------------------------  
+// Copyright (c) Microsoft Corporation.  All rights reserved.   
+//------------------------------------------------------------  
+
+namespace System.ServiceModel.Channels
+{
+    class MsmqInputMessage : NativeMsmqMessage
+    {
+        BufferProperty senderId;
+        IntProperty senderIdLength;
+        LongProperty lookupId;
+        IntProperty abortCount;
+        IntProperty moveCount;
+        BufferProperty senderCertificate;
+        IntProperty senderCertificateLength;
+        IntProperty lastMovedTime;
+        BufferProperty body;
+        IntProperty bodyLength;
+        BufferProperty messageId;
+        ShortProperty cls;
+        int maxBufferSize;
+        const int maxSize = 4 * 1024 * 1024;
+        const int initialBodySize = 4096;
+        const int initialSenderIdSize = 256;
+        const int initialCertificateSize = 4096;
+
+        public MsmqInputMessage()
+            : this(0, maxSize)
+        {
+        }
+
+        public MsmqInputMessage(int maxBufferSize)
+            : this(0, maxBufferSize)
+        {
+        }
+
+        protected MsmqInputMessage(int additionalPropertyCount, int maxBufferSize)
+            : this(additionalPropertyCount, new SizeQuota(maxBufferSize))
+        {
+        }
+
+        protected MsmqInputMessage(int additionalPropertyCount, SizeQuota bufferSizeQuota)
+            : base(12 + additionalPropertyCount)
+        {
+            this.maxBufferSize = bufferSizeQuota.MaxSize;
+            this.body = new BufferProperty(this, UnsafeNativeMethods.PROPID_M_BODY,
+                bufferSizeQuota.AllocIfAvailable(initialBodySize));
+            this.bodyLength = new IntProperty(this, UnsafeNativeMethods.PROPID_M_BODY_SIZE);
+            this.messageId = new BufferProperty(this, UnsafeNativeMethods.PROPID_M_MSGID,
+                UnsafeNativeMethods.PROPID_M_MSGID_SIZE);
+            this.lookupId = new LongProperty(this, UnsafeNativeMethods.PROPID_M_LOOKUPID);
+            this.cls = new ShortProperty(this, UnsafeNativeMethods.PROPID_M_CLASS);
+            this.senderId = new BufferProperty(this, UnsafeNativeMethods.PROPID_M_SENDERID, initialSenderIdSize);
+            this.senderIdLength = new IntProperty(this, UnsafeNativeMethods.PROPID_M_SENDERID_LEN);
+            this.senderCertificate = new BufferProperty(this, UnsafeNativeMethods.PROPID_M_SENDER_CERT,
+                bufferSizeQuota.AllocIfAvailable(initialCertificateSize));
+            this.senderCertificateLength = new IntProperty(this, UnsafeNativeMethods.PROPID_M_SENDER_CERT_LEN);
+            if (Msmq.IsAdvancedPoisonHandlingSupported)
+            {
+                this.lastMovedTime = new IntProperty(this, UnsafeNativeMethods.PROPID_M_LAST_MOVE_TIME);
+                this.abortCount = new IntProperty(this, UnsafeNativeMethods.PROPID_M_ABORT_COUNT);
+                this.moveCount = new IntProperty(this, UnsafeNativeMethods.PROPID_M_MOVE_COUNT);
+            }
+        }
+
+        public override void GrowBuffers()
+        {
+            OnGrowBuffers(new SizeQuota(this.maxBufferSize));
+        }
+
+        protected virtual void OnGrowBuffers(SizeQuota bufferSizeQuota)
+        {
+            bufferSizeQuota.Alloc(this.senderIdLength.Value);
+            this.senderId.EnsureBufferLength(this.senderIdLength.Value);
+
+            bufferSizeQuota.Alloc(this.senderCertificateLength.Value);
+            this.senderCertificate.EnsureBufferLength(this.senderCertificateLength.Value);
+
+            bufferSizeQuota.Alloc(this.bodyLength.Value);
+            this.body.EnsureBufferLength(this.bodyLength.Value);
+        }
+
+        public BufferProperty SenderId
+        {
+            get { return this.senderId; }
+        }
+
+        public IntProperty SenderIdLength
+        {
+            get { return this.senderIdLength; }
+        }
+
+        public LongProperty LookupId
+        {
+            get { return this.lookupId; }
+        }
+
+        public IntProperty AbortCount
+        {
+            get { return this.abortCount; }
+        }
+
+        public IntProperty MoveCount
+        {
+            get { return this.moveCount; }
+        }
+
+        public BufferProperty SenderCertificate
+        {
+            get { return this.senderCertificate; }
+        }
+
+        public IntProperty SenderCertificateLength
+        {
+            get { return this.senderCertificateLength; }
+        }
+
+        public IntProperty LastMovedTime
+        {
+            get { return this.lastMovedTime; }
+        }
+
+        public BufferProperty Body
+        {
+            get { return this.body; }
+        }
+
+        public IntProperty BodyLength
+        {
+            get { return this.bodyLength; }
+        }
+
+        public BufferProperty MessageId
+        {
+            get { return this.messageId; }
+        }
+
+        public ShortProperty Class
+        {
+            get { return this.cls; }
+        }
+
+        protected class SizeQuota
+        {
+            int remainingSize;
+            int maxSize;
+
+            public SizeQuota(int maxSize)
+            {
+                this.maxSize = maxSize;
+                this.remainingSize = maxSize;
+            }
+
+            public int MaxSize
+            {
+                get { return this.maxSize; }
+            }
+
+            public void Alloc(int requiredSize)
+            {
+                if (requiredSize > this.remainingSize)
+                {
+                    throw DiagnosticUtility.ExceptionUtility.ThrowHelperError(MaxMessageSizeStream.CreateMaxReceivedMessageSizeExceededException(this.maxSize));
+                }
+                this.remainingSize -= requiredSize;
+            }
+
+            public int AllocIfAvailable(int desiredSize)
+            {
+                int availableSize = Math.Min(desiredSize, this.remainingSize);
+                this.remainingSize -= availableSize;
+                return availableSize;
+            }
+        }
+    }
+}