mcs/class/System.ServiceModel/System.ServiceModel.Channels.NetTcp/TcpReplyChannel.cs

   1 //
   2 // TcpReplyChannel.cs
   3 //
   4 // Author:
   5 //      Atsushi Enomoto <atsushi@ximian.com>
   6 //
   7 // Copyright (C) 2009 Novell, Inc.  http://www.novell.com
   8 //
   9 // Permission is hereby granted, free of charge, to any person obtaining
  10 // a copy of this software and associated documentation files (the
  11 // "Software"), to deal in the Software without restriction, including
  12 // without limitation the rights to use, copy, modify, merge, publish,
  13 // distribute, sublicense, and/or sell copies of the Software, and to
  14 // permit persons to whom the Software is furnished to do so, subject to
  15 // the following conditions:
  16 //
  17 // The above copyright notice and this permission notice shall be
  18 // included in all copies or substantial portions of the Software.
  19 //
  20 // THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
  21 // EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
  22 // MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
  23 // NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
  24 // LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
  25 // OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
  26 // WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
  27 //
  28 using System;
  29 using System.Collections.Generic;
  30 using System.IO;
  31 using System.Net;
  32 using System.Net.Sockets;
  33 using System.ServiceModel;
  34 using System.Text;
  35 using System.Threading;
  36
  37 namespace System.ServiceModel.Channels.NetTcp
  38 {
  39         internal class TcpReplyChannel : InternalReplyChannelBase
  40         {
  41                 TcpClient client;
  42                 TcpChannelInfo info;
  43                 TcpBinaryFrameManager frame;
  44
  45                 public TcpReplyChannel (ChannelListenerBase listener, TcpChannelInfo info, TcpClient client)
  46                         : base (listener)
  47                 {
  48                         this.client = client;
  49                         this.info = info;
  50                 }
  51
  52                 public MessageEncoder Encoder {
  53                         get { return info.MessageEncoder; }
  54                 }
  55
  56                 public override RequestContext ReceiveRequest (TimeSpan timeout)
  57                 {
  58                         if (timeout <= TimeSpan.Zero)
  59                                 throw new ArgumentException (String.Format ("Timeout value must be positive value. It was {0}", timeout));
  60
  61                         DateTime start = DateTime.Now;
  62
  63                         // FIXME: use timeout
  64                         if (client == null)
  65                                 client = ((TcpChannelListener<IReplyChannel>) Manager).AcceptTcpClient (timeout);
  66                         NetworkStream ns = client.GetStream ();
  67                         frame = new TcpBinaryFrameManager (TcpBinaryFrameManager.SingletonUnsizedMode, ns, true) { Encoder = this.Encoder };
  68
  69                         // FIXME: use timeout
  70                         if (!frame.ProcessPreambleRecipient ())
  71                                 return null;
  72                         frame.ProcessPreambleAckRecipient ();
  73
  74                         var msg = frame.ReadUnsizedMessage (timeout);
  75
  76                         Logger.LogMessage (MessageLogSourceKind.TransportReceive, ref msg, info.BindingElement.MaxReceivedMessageSize);
  77
  78                         // LAMESPEC: it contradicts the protocol explanation at section 3.1.1.1.1 in [MC-NMF].
  79                         // Moving ReadEndRecord() after context's WriteUnsizedMessage() causes TCP connection blocking.
  80                         frame.ReadEndRecord ();
  81                         return new TcpRequestContext (this, msg);
  82                 }
  83
  84                 class TcpRequestContext : InternalRequestContext
  85                 {
  86                         public TcpRequestContext (TcpReplyChannel owner, Message request)
  87                                 : base (owner.Manager)
  88                         {
  89                                 this.owner = owner;
  90                                 this.request = request;
  91                         }
  92
  93                         TcpReplyChannel owner;
  94                         Message request;
  95
  96                         public override Message RequestMessage {
  97                                 get { return request; }
  98                         }
  99
 100                         public override void Abort ()
 101                         {
 102                                 Close (TimeSpan.Zero);
 103                         }
 104
 105                         public override void Close (TimeSpan timeout)
 106                         {
 107                         }
 108
 109                         public override void Reply (Message message, TimeSpan timeout)
 110                         {
 111                                 Logger.LogMessage (MessageLogSourceKind.TransportSend, ref message, owner.info.BindingElement.MaxReceivedMessageSize);
 112
 113                                 DateTime start = DateTime.Now;
 114                                 owner.frame.WriteUnsizedMessage (message, timeout);
 115                                 // FIXME: consider timeout here too.
 116                                 owner.frame.WriteEndRecord ();
 117                         }
 118                 }
 119
 120                 public override bool TryReceiveRequest (TimeSpan timeout, out RequestContext context)
 121                 {
 122                         try {
 123                                 DateTime start = DateTime.Now;
 124                                 context = ReceiveRequest (timeout);
 125                                 return context != null;
 126                         } catch (Exception ex) {
 127                                 // FIXME: log it?
 128                                 // Console.WriteLine (ex);
 129                                 context = null;
 130                                 return false;
 131                         }
 132                 }
 133
 134                 public override bool WaitForRequest (TimeSpan timeout)
 135                 {
 136                         throw new NotImplementedException ();
 137                 }
 138
 139                 bool close_started;
 140                 object close_lock = new object ();
 141
 142                 protected override void OnClose (TimeSpan timeout)
 143                 {
 144                         lock (close_lock) {
 145                                 if (close_started)
 146                                         return;
 147                                 close_started = true;
 148                         }
 149
 150                         client.Close ();
 151                         client = null;
 152                         base.OnClose (timeout);
 153                 }
 154
 155                 protected override void OnOpen (TimeSpan timeout)
 156                 {
 157                 }
 158         }
 159 }