ConnectionModeReader.cs source code in C# .NET

Source code for the .NET framework in C#

                        

Code:

/ WCF / WCF / 3.5.30729.1 / untmp / Orcas / SP / ndp / cdf / src / WCF / ServiceModel / System / ServiceModel / Channels / ConnectionModeReader.cs / 1 / ConnectionModeReader.cs

                            //------------------------------------------------------------ 
// Copyright (c) Microsoft Corporation.  All rights reserved.
//-----------------------------------------------------------
namespace System.ServiceModel.Channels
{ 
    using System.Threading;
    using System.ServiceModel; 
    using System.Collections.Generic; 
    using System.Diagnostics;
 
    delegate void ConnectionModeCallback(ConnectionModeReader connectionModeReader);

    sealed class ConnectionModeReader : InitialServerConnectionReader
    { 
        Exception readException;
        ServerModeDecoder decoder; 
        byte[] buffer; 
        int offset;
        int size; 
        ConnectionModeCallback callback;
        static WaitCallback readCallback;
        TimeoutHelper receiveTimeoutHelper;
 
        public ConnectionModeReader(IConnection connection, ConnectionModeCallback callback, ConnectionClosedCallback closedCallback)
            : base(connection, closedCallback) 
        { 
            this.callback = callback;
        } 

        public int BufferOffset
        {
            get { return offset; } 
        }
 
        public int BufferSize 
        {
            get { return size; } 
        }

        public long StreamPosition
        { 
            get { return decoder.StreamPosition; }
        } 
 
        public TimeSpan GetRemainingTimeout()
        { 
            return this.receiveTimeoutHelper.RemainingTime();
        }

        void Complete(Exception e) 
        {
            // exception will be logged by the caller 
            readException = e; 
            Complete();
        } 

        void Complete()
        {
            callback(this); 
        }
 
        bool ContinueReading() 
        {
            for (; ; ) 
            {
                if (size == 0)
                {
                    if (readCallback == null) 
                    {
                        readCallback = new WaitCallback(ReadCallback); 
                    } 

                    if (Connection.BeginRead(0, Connection.AsyncReadBufferSize, GetRemainingTimeout(), 
                        readCallback, this) == AsyncReadResult.Queued)
                    {
                        break;
                    } 
                    if (!GetReadResult()) // we're at EOF, bail
                    { 
                        return false; 
                    }
                } 

                for (; ; )
                {
                    int bytesDecoded; 
                    try
                    { 
                        bytesDecoded = decoder.Decode(buffer, offset, size); 
                    }
                    catch (CommunicationException e) 
                    {
                        // see if we need to send back a framing fault
                        string framingFault;
                        if (FramingEncodingString.TryGetFaultString(e, out framingFault)) 
                        {
                            byte[] drainBuffer = new byte[128]; 
                            InitialServerConnectionReader.SendFault( 
                                Connection, framingFault, drainBuffer, GetRemainingTimeout(),
                                MaxViaSize + MaxContentTypeSize); 
                            base.Close(GetRemainingTimeout());
                        }
                        throw;
                    } 

                    if (bytesDecoded > 0) 
                    { 
                        offset += bytesDecoded;
                        size -= bytesDecoded; 
                    }
                    if (decoder.CurrentState == ServerModeDecoder.State.Done)
                    {
                        return true; 
                    }
                    if (size == 0) 
                    { 
                        break;
                    } 
                }
            }

            return false; 
        }
 
        static void ReadCallback(object state) 
        {
            ConnectionModeReader reader = (ConnectionModeReader)state; 
            bool completeSelf = false;
            Exception completionException = null;
            try
            { 
                if (reader.GetReadResult())
                { 
                    completeSelf = reader.ContinueReading(); 
                }
            } 
#pragma warning suppress 56500 // [....], transferring exception to caller
            catch (Exception e)
            {
                if (DiagnosticUtility.IsFatal(e)) 
                {
                    throw; 
                } 

                completeSelf = true; 
                completionException = e;
            }

            if (completeSelf) 
            {
                reader.Complete(completionException); 
            } 
        }
 
        bool GetReadResult()
        {
            offset = 0;
            size = Connection.EndRead(); 
            if (size == 0)
            { 
                if (this.decoder.StreamPosition == 0) // client timed out a cached connection 
                {
                    base.Close(GetRemainingTimeout()); 
                    return false;
                }
                else
                { 
                    throw DiagnosticUtility.ExceptionUtility.ThrowHelperError(decoder.CreatePrematureEOFException());
                } 
            } 

            // restore ExceptionEventType to Error after the initial read for cached connections 
            Connection.ExceptionEventType = TraceEventType.Error;

            if (buffer == null)
            { 
                buffer = Connection.AsyncReadBuffer;
            } 
 
            return true;
        } 

        public FramingMode GetConnectionMode()
        {
            if (readException != null) 
            {
                throw DiagnosticUtility.ExceptionUtility.ThrowHelper(readException, Connection.ExceptionEventType); 
            } 

            return decoder.Mode; 
        }

        public void StartReading(TimeSpan receiveTimeout, ItemDequeuedCallback connectionDequeuedCallback)
        { 
            this.decoder = new ServerModeDecoder();
            this.receiveTimeoutHelper = new TimeoutHelper(receiveTimeout); 
            this.ConnectionDequeuedCallback = connectionDequeuedCallback; 
            bool completeSelf = false;
            Exception completionException = null; 
            try
            {
                completeSelf = ContinueReading();
            } 
#pragma warning suppress 56500 // [....], transferring exception to caller
            catch (Exception e) 
            { 
                if (DiagnosticUtility.IsFatal(e))
                { 
                    throw;
                }

                completeSelf = true; 
                completionException = e;
            } 
 
            if (completeSelf)
            { 
                Complete(completionException);
            }
        }
    } 
}

// File provided for Reference Use Only by Microsoft Corporation (c) 2007.
// Copyright (c) Microsoft Corporation. All rights reserved.
                        

Link Menu

Network programming in C#, Network Programming in VB.NET, Network Programming in .NET
This book is available now!
Buy at Amazon US or
Buy at Amazon UK