mcs/class/System.ServiceModel/System.ServiceModel.Dispatcher/ChannelDispatcher.cs

   1 //
   2 // ChannelDispatcher.cs
   3 //
   4 // Author:
   5 //      Atsushi Enomoto <atsushi@ximian.com>
   6 //
   7 // Copyright (C) 2005,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.Collections.ObjectModel;
  31 using System.Reflection;
  32 using System.ServiceModel.Channels;
  33 using System.Threading;
  34 using System.Transactions;
  35 using System.ServiceModel;
  36 using System.ServiceModel.Description;
  37
  38 namespace System.ServiceModel.Dispatcher
  39 {
  40         public class ChannelDispatcher : ChannelDispatcherBase
  41         {
  42                 class EndpointDispatcherCollection : SynchronizedCollection<EndpointDispatcher>
  43                 {
  44                         public EndpointDispatcherCollection (ChannelDispatcher owner)
  45                         {
  46                                 this.owner = owner;
  47                         }
  48
  49                         ChannelDispatcher owner;
  50
  51                         protected override void ClearItems ()
  52                         {
  53                                 foreach (var ed in this)
  54                                         ed.ChannelDispatcher = null;
  55                                 base.ClearItems ();
  56                         }
  57
  58                         protected override void InsertItem (int index, EndpointDispatcher item)
  59                         {
  60                                 item.ChannelDispatcher = owner;
  61                                 base.InsertItem (index, item);
  62                         }
  63
  64                         protected virtual void RemoveItem (int index)
  65                         {
  66                                 if (index < Count)
  67                                         this [index].ChannelDispatcher = null;
  68                                 base.RemoveItem (index);
  69                         }
  70
  71                         protected virtual void SetItem (int index, EndpointDispatcher item)
  72                         {
  73                                 item.ChannelDispatcher = owner;
  74                                 base.SetItem (index, item);
  75                         }
  76                 }
  77
  78                 ServiceHostBase host;
  79
  80                 string binding_name;
  81                 Collection<IErrorHandler> error_handlers
  82                         = new Collection<IErrorHandler> ();
  83                 IChannelListener listener;
  84                 internal IDefaultCommunicationTimeouts timeouts; // FIXME: remove internal
  85                 MessageVersion message_version;
  86                 bool receive_sync, include_exception_detail_in_faults,
  87                         manual_addressing, is_tx_receive;
  88                 int max_tx_batch_size;
  89                 SynchronizedCollection<IChannelInitializer> initializers
  90                         = new SynchronizedCollection<IChannelInitializer> ();
  91                 IsolationLevel tx_isolation_level;
  92                 TimeSpan tx_timeout;
  93                 ServiceThrottle throttle;
  94
  95                 Guid identifier = Guid.NewGuid ();
  96                 ManualResetEvent async_event = new ManualResetEvent (false);
  97
  98                 ListenerLoopManager loop_manager;
  99                 SynchronizedCollection<EndpointDispatcher> endpoints;
 100
 101                 [MonoTODO ("get binding info from config")]
 102                 public ChannelDispatcher (IChannelListener listener)
 103                         : this (listener, null)
 104                 {
 105                 }
 106
 107                 public ChannelDispatcher (
 108                         IChannelListener listener, string bindingName)
 109                         : this (listener, bindingName, null)
 110                 {
 111                 }
 112
 113                 public ChannelDispatcher (
 114                         IChannelListener listener, string bindingName,
 115                         IDefaultCommunicationTimeouts timeouts)
 116                 {
 117                         if (listener == null)
 118                                 throw new ArgumentNullException ("listener");
 119                         Init (listener, bindingName, timeouts);
 120                 }
 121
 122                 private void Init (IChannelListener listener, string bindingName,
 123                         IDefaultCommunicationTimeouts timeouts)
 124                 {
 125                         this.listener = listener;
 126                         this.binding_name = bindingName;
 127                         // IChannelListener is often a ChannelListenerBase
 128                         // which implements IDefaultCommunicationTimeouts.
 129                         this.timeouts = timeouts ?? listener as IDefaultCommunicationTimeouts ?? DefaultCommunicationTimeouts.Instance;
 130                         endpoints = new EndpointDispatcherCollection (this);
 131                 }
 132
 133                 internal void InitializeServiceEndpoint (Type serviceType, ServiceEndpoint se)
 134                 {
 135                         this.MessageVersion = se.Binding.MessageVersion;
 136                         if (this.MessageVersion == null)
 137                                 this.MessageVersion = MessageVersion.Default;
 138
 139                         //Attach one EndpointDispacher to the ChannelDispatcher
 140                         EndpointDispatcher ed = new EndpointDispatcher (se.Address, se.Contract.Name, se.Contract.Namespace);
 141                         this.Endpoints.Add (ed);
 142                         ed.InitializeServiceEndpoint (false, serviceType, se);
 143                 }
 144
 145                 public string BindingName {
 146                         get { return binding_name; }
 147                 }
 148
 149                 public SynchronizedCollection<IChannelInitializer> ChannelInitializers {
 150                         get { return initializers; }
 151                 }
 152
 153                 protected internal override TimeSpan DefaultCloseTimeout {
 154                         get { return timeouts.CloseTimeout; }
 155                 }
 156
 157                 protected internal override TimeSpan DefaultOpenTimeout {
 158                         get { return timeouts.OpenTimeout; }
 159                 }
 160
 161                 public Collection<IErrorHandler> ErrorHandlers {
 162                         get { return error_handlers; }
 163                 }
 164
 165                 public SynchronizedCollection<EndpointDispatcher> Endpoints {
 166                         get { return endpoints; }
 167                 }
 168
 169                 [MonoTODO]
 170                 public bool IsTransactedAccept {
 171                         get { throw new NotImplementedException (); }
 172                 }
 173
 174                 public bool IsTransactedReceive {
 175                         get { return is_tx_receive; }
 176                         set { is_tx_receive = value; }
 177                 }
 178
 179                 public bool ManualAddressing {
 180                         get { return manual_addressing; }
 181                         set { manual_addressing = value; }
 182                 }
 183
 184                 public int MaxTransactedBatchSize {
 185                         get { return max_tx_batch_size; }
 186                         set { max_tx_batch_size = value; }
 187                 }
 188
 189                 public override ServiceHostBase Host {
 190                         get { return host; }
 191                 }
 192
 193                 public override IChannelListener Listener {
 194                         get { return listener; }
 195                 }
 196
 197                 public MessageVersion MessageVersion {
 198                         get { return message_version; }
 199                         set { message_version = value; }
 200                 }
 201
 202                 public bool ReceiveSynchronously {
 203                         get { return receive_sync; }
 204                         set {
 205                                 ThrowIfDisposedOrImmutable ();
 206                                 receive_sync = value;
 207                         }
 208                 }
 209
 210                 public bool IncludeExceptionDetailInFaults {
 211                         get { return include_exception_detail_in_faults; }
 212                         set { include_exception_detail_in_faults = value; }
 213                 }
 214
 215                 public ServiceThrottle ServiceThrottle {
 216                         get { return throttle; }
 217                         set { throttle = value; }
 218                 }
 219
 220                 public IsolationLevel TransactionIsolationLevel {
 221                         get { return tx_isolation_level; }
 222                         set { tx_isolation_level = value; }
 223                 }
 224
 225                 public TimeSpan TransactionTimeout {
 226                         get { return tx_timeout; }
 227                         set { tx_timeout = value; }
 228                 }
 229
 230                 protected internal override void Attach (ServiceHostBase host)
 231                 {
 232                         this.host = host;
 233                 }
 234
 235                 public override void CloseInput ()
 236                 {
 237                         if (loop_manager != null)
 238                                 loop_manager.CloseInput ();
 239                 }
 240
 241                 protected internal override void Detach (ServiceHostBase host)
 242                 {
 243                         this.host = null;
 244                 }
 245
 246                 protected override void OnAbort ()
 247                 {
 248                         if (loop_manager != null)
 249                                 loop_manager.Stop (TimeSpan.FromTicks (1));
 250                 }
 251
 252                 Action<TimeSpan> open_delegate;
 253                 Action<TimeSpan> close_delegate;
 254
 255                 protected override IAsyncResult OnBeginClose (TimeSpan timeout,
 256                         AsyncCallback callback, object state)
 257                 {
 258                         if (close_delegate == null)
 259                                 close_delegate = new Action<TimeSpan> (OnClose);
 260                         return close_delegate.BeginInvoke (timeout, callback, state);
 261                 }
 262
 263                 protected override IAsyncResult OnBeginOpen (TimeSpan timeout,
 264                         AsyncCallback callback, object state)
 265                 {
 266                         if (open_delegate == null)
 267                                 open_delegate = new Action<TimeSpan> (OnClose);
 268                         return open_delegate.BeginInvoke (timeout, callback, state);
 269                 }
 270
 271                 protected override void OnClose (TimeSpan timeout)
 272                 {
 273                         if (loop_manager != null)
 274                                 loop_manager.Stop (timeout);
 275                 }
 276
 277                 protected override void OnClosed ()
 278                 {
 279                         if (host != null)
 280                                 host.ChannelDispatchers.Remove (this);
 281                         base.OnClosed ();
 282                 }
 283
 284                 protected override void OnEndClose (IAsyncResult result)
 285                 {
 286                         close_delegate.EndInvoke (result);
 287                 }
 288
 289                 protected override void OnEndOpen (IAsyncResult result)
 290                 {
 291                         open_delegate.EndInvoke (result);
 292                 }
 293
 294                 protected override void OnOpen (TimeSpan timeout)
 295                 {
 296                         if (Host == null || MessageVersion == null)
 297                                 throw new InvalidOperationException ("Service host is not attached to this ChannelDispatcher.");
 298
 299                         loop_manager = new ListenerLoopManager (this, timeout);
 300                 }
 301
 302                 [MonoTODO ("what to do here?")]
 303                 protected override void OnOpening ()
 304                 {
 305                 }
 306
 307                 protected override void OnOpened ()
 308                 {
 309                         loop_manager.Setup ();
 310                 }
 311
 312                 internal void StartLoop ()
 313                 {
 314                         // FIXME: not sure if it should be filled here.
 315                         if (ServiceThrottle == null)
 316                                 ServiceThrottle = new ServiceThrottle ();
 317
 318                         loop_manager.Start ();
 319                 }
 320         }
 321
 322                 // isolated from ChannelDispatcher
 323                 class ListenerLoopManager
 324                 {
 325                         ChannelDispatcher owner;
 326                         AutoResetEvent handle = new AutoResetEvent (false);
 327                         AutoResetEvent creator_handle = new AutoResetEvent (false);
 328                         ManualResetEvent stop_handle = new ManualResetEvent (false);
 329                         bool loop;
 330                         Thread loop_thread;
 331                         TimeSpan open_timeout;
 332                         Func<IAsyncResult> channel_acceptor;
 333                         List<IChannel> channels = new List<IChannel> ();
 334
 335                         public ListenerLoopManager (ChannelDispatcher owner, TimeSpan openTimeout)
 336                         {
 337                                 this.owner = owner;
 338                                 open_timeout = openTimeout;
 339                         }
 340
 341                         public void Setup ()
 342                         {
 343                                 if (owner.Listener.State != CommunicationState.Opened)
 344                                         owner.Listener.Open (open_timeout);
 345
 346                                 // It is tested at Open(), but strangely it is not instantiated at this point.
 347                                 foreach (var ed in owner.Endpoints)
 348                                         if (ed.DispatchRuntime.InstanceContextProvider == null && (ed.DispatchRuntime.Type == null || ed.DispatchRuntime.Type.GetConstructor (Type.EmptyTypes) == null))
 349                                                 throw new InvalidOperationException ("There is no default constructor for the service Type in the DispatchRuntime");
 350                                 SetupChannelAcceptor ();
 351                         }
 352
 353                         public void Start ()
 354                         {
 355                                 foreach (var ed in owner.Endpoints)
 356                                         if (ed.DispatchRuntime.InstanceContextProvider == null)
 357                                                 ed.DispatchRuntime.InstanceContextProvider = new DefaultInstanceContextProvider ();
 358
 359                                 if (loop_thread == null)
 360                                         loop_thread = new Thread (new ThreadStart (Loop));
 361                                 loop_thread.Start ();
 362                         }
 363
 364                         Func<IAsyncResult> CreateAcceptor<TChannel> (IChannelListener l) where TChannel : class, IChannel
 365                         {
 366                                 IChannelListener<TChannel> r = l as IChannelListener<TChannel>;
 367                                 if (r == null)
 368                                         return null;
 369                                 AsyncCallback callback = delegate (IAsyncResult result) {
 370                                         try {
 371                                                 ChannelAccepted (r.EndAcceptChannel (result));
 372                                         } catch (Exception ex) {
 373                                                 Console.WriteLine ("Exception during finishing channel acceptance.");
 374                                                 Console.WriteLine (ex);
 375                                         }
 376                                 };
 377                                 return delegate {
 378                                         try {
 379                                                 return r.BeginAcceptChannel (callback, null);
 380                                         } catch (Exception ex) {
 381                                                 Console.WriteLine ("Exception during accepting channel.");
 382                                                 Console.WriteLine (ex);
 383                                                 throw;
 384                                         }
 385                                 };
 386                         }
 387
 388                         void SetupChannelAcceptor ()
 389                         {
 390                                 var l = owner.Listener;
 391                                 channel_acceptor =
 392                                         CreateAcceptor<IReplyChannel> (l) ??
 393                                         CreateAcceptor<IReplySessionChannel> (l) ??
 394                                         CreateAcceptor<IInputChannel> (l) ??
 395                                         CreateAcceptor<IInputSessionChannel> (l) ??
 396                                         CreateAcceptor<IDuplexChannel> (l) ??
 397                                         CreateAcceptor<IDuplexSessionChannel> (l);
 398                                 if (channel_acceptor == null)
 399                                         throw new InvalidOperationException (String.Format ("Unrecognized channel listener type: {0}", l.GetType ()));
 400                         }
 401
 402                         public void Stop (TimeSpan timeout)
 403                         {
 404                                 if (loop_thread == null)
 405                                         return;
 406
 407                                 loop = false;
 408                                 creator_handle.Set ();
 409                                 handle.Set ();
 410                                 if (stop_handle != null) {
 411                                         stop_handle.WaitOne (timeout > TimeSpan.Zero ? timeout : TimeSpan.FromTicks (1));
 412                                         stop_handle.Close ();
 413                                         stop_handle = null;
 414                                 }
 415                                 if (owner.Listener.State != CommunicationState.Closed)
 416                                         owner.Listener.Abort ();
 417                                 if (loop_thread != null && loop_thread.IsAlive)
 418                                         loop_thread.Abort ();
 419                                 loop_thread = null;
 420                         }
 421
 422                         public void CloseInput ()
 423                         {
 424                                 foreach (var ch in channels.ToArray ()) {
 425                                         if (ch.State == CommunicationState.Closed)
 426                                                 channels.Remove (ch); // zonbie, if exists
 427                                         else
 428                                                 ch.Close ();
 429                                 }
 430                         }
 431
 432                         void Loop ()
 433                         {
 434                                 try {
 435                                         LoopCore ();
 436                                 } catch (Exception ex) {
 437                                         // FIXME: log it
 438                                         Console.WriteLine ("ChannelDispatcher caught an exception inside dispatcher loop, which is likely thrown by the channel listener {0}", owner.Listener);
 439                                         Console.WriteLine (ex);
 440                                 } finally {
 441                                         if (stop_handle != null)
 442                                                 stop_handle.Set ();
 443                                 }
 444                         }
 445
 446                         void LoopCore ()
 447                         {
 448                                 loop = true;
 449
 450                                 // FIXME: use WaitForChannel() for (*only* for) transacted channel listeners.
 451                                 // http://social.msdn.microsoft.com/Forums/en-US/wcf/thread/3faa4a5e-8602-4dbe-a181-73b3f581835e
 452
 453                                 //FIXME: The logic here should be somewhat different as follows:
 454                                 //1. Get the message
 455                                 //2. Get the appropriate EndPointDispatcher that can handle the message
 456                                 //   which is done using the filters (AddressFilter, ContractFilter).
 457                                 //3. Let the appropriate endpoint handle the request.
 458
 459                                 while (loop) {
 460                                         while (loop && channels.Count < owner.ServiceThrottle.MaxConcurrentSessions) {
 461                                                 channel_acceptor ();
 462                                                 creator_handle.WaitOne (); // released by ChannelAccepted()
 463                                                 creator_handle.Reset ();
 464                                         }
 465                                         if (!loop)
 466                                                 break;
 467                                         handle.WaitOne (); // released by IChannel.Close()
 468                                         handle.Reset ();
 469                                 }
 470                                 owner.Listener.Close ();
 471                                 owner.CloseInput ();
 472                         }
 473
 474                         void ChannelAccepted (IChannel ch)
 475                         {
 476                         try {
 477                                 if (ch == null) // could happen when it was aborted
 478                                         return;
 479                                 if (!loop) {
 480                                         var dis = ch as IDisposable;
 481                                         if (dis != null)
 482                                                 dis.Dispose ();
 483                                         return;
 484                                 }
 485
 486                                 channels.Add (ch);
 487                                 ch.Opened += delegate {
 488                                         ch.Faulted += delegate {
 489                                                 if (channels.Contains (ch))
 490                                                         channels.Remove (ch);
 491                                                 handle.Set (); // release loop wait lock.
 492                                                 };
 493                                         ch.Closed += delegate {
 494                                                 if (channels.Contains (ch))
 495                                                         channels.Remove (ch);
 496                                                 handle.Set (); // release loop wait lock.
 497                                                 };
 498                                         };
 499                                 ch.Open ();
 500                         } finally {
 501                                 creator_handle.Set ();
 502                         }
 503
 504                                 ProcessRequestOrInput (ch);
 505                         }
 506
 507                         void ProcessRequestOrInput (IChannel ch)
 508                         {
 509                                 var reply = ch as IReplyChannel;
 510                                 var input = ch as IInputChannel;
 511
 512                                 if (reply != null) {
 513                                         if (owner.ReceiveSynchronously) {
 514                                                 RequestContext rc;
 515                                                 if (reply.TryReceiveRequest (owner.timeouts.ReceiveTimeout, out rc))
 516                                                         ProcessRequest (reply, rc);
 517                                         } else {
 518                                                 reply.BeginTryReceiveRequest (owner.timeouts.ReceiveTimeout, TryReceiveRequestDone, reply);
 519                                         }
 520                                 } else if (input != null) {
 521                                         if (owner.ReceiveSynchronously) {
 522                                                 Message msg;
 523                                                 if (input.TryReceive (owner.timeouts.ReceiveTimeout, out msg))
 524                                                         ProcessInput (input, msg);
 525                                         } else {
 526                                                 input.BeginTryReceive (owner.timeouts.ReceiveTimeout, TryReceiveDone, input);
 527                                         }
 528                                 }
 529                         }
 530
 531                         void TryReceiveRequestDone (IAsyncResult result)
 532                         {
 533                                 RequestContext rc;
 534                                 var reply = (IReplyChannel) result.AsyncState;
 535                                 if (reply.EndTryReceiveRequest (result, out rc))
 536                                         ProcessRequest (reply, rc);
 537                         }
 538
 539                         void TryReceiveDone (IAsyncResult result)
 540                         {
 541                                 Message msg;
 542                                 var input = (IInputChannel) result.AsyncState;
 543                                 if (input.EndTryReceive (result, out msg))
 544                                         ProcessInput (input, msg);
 545                         }
 546
 547                         void SendEndpointNotFound (RequestContext rc, EndpointNotFoundException ex)
 548                         {
 549                                 try {
 550
 551                                         MessageVersion version = rc.RequestMessage.Version;
 552                                         FaultCode fc = new FaultCode ("DestinationUnreachable", version.Addressing.Namespace);
 553                                         Message res = Message.CreateMessage (version, fc, "error occured", rc.RequestMessage.Headers.Action);
 554                                         rc.Reply (res);
 555                                 }
 556                                 catch (Exception e) { }
 557                         }
 558
 559                         void ProcessRequest (IReplyChannel reply, RequestContext rc)
 560                         {
 561                                 try {
 562                                         EndpointDispatcher candidate = FindEndpointDispatcher (rc.RequestMessage);
 563                                         new InputOrReplyRequestProcessor (candidate.DispatchRuntime, reply).
 564                                                 ProcessReply (rc);
 565                                 } catch (EndpointNotFoundException ex) {
 566                                         SendEndpointNotFound (rc, ex);
 567                                 } catch (Exception ex) {
 568                                         // FIXME: log it.
 569                                         Console.WriteLine (ex);
 570                                 } finally {
 571                                         if (rc != null)
 572                                                 rc.Close ();
 573                                         // unless it is closed by session/call manager, move it back to the loop to receive the next message.
 574                                         if (reply.State != CommunicationState.Closed)
 575                                                 ProcessRequestOrInput (reply);
 576                                 }
 577                         }
 578
 579                         void ProcessInput (IInputChannel input, Message message)
 580                         {
 581                                 try {
 582                                         EndpointDispatcher candidate = null;
 583                                         candidate = FindEndpointDispatcher (message);
 584                                         new InputOrReplyRequestProcessor (candidate.DispatchRuntime, input).
 585                                                 ProcessInput (message);
 586                                 }
 587                                 catch (Exception ex) {
 588                                         // FIXME: log it.
 589                                         Console.WriteLine (ex);
 590                                 } finally {
 591                                         // unless it is closed by session/call manager, move it back to the loop to receive the next message.
 592                                         if (input.State != CommunicationState.Closed)
 593                                                 ProcessRequestOrInput (input);
 594                                 }
 595                         }
 596
 597                         EndpointDispatcher FindEndpointDispatcher (Message message) {
 598                                 EndpointDispatcher candidate = null;
 599                                 for (int i = 0; i < owner.Endpoints.Count; i++) {
 600                                         if (MessageMatchesEndpointDispatcher (message, owner.Endpoints [i])) {
 601                                                 var newdis = owner.Endpoints [i];
 602                                                 if (candidate == null || candidate.FilterPriority < newdis.FilterPriority)
 603                                                         candidate = newdis;
 604                                                 else if (candidate.FilterPriority == newdis.FilterPriority)
 605                                                         throw new MultipleFilterMatchesException ();
 606                                         }
 607                                 }
 608                                 if (candidate == null && owner.Host != null)
 609                                         owner.Host.OnUnknownMessageReceived (message);
 610                                 return candidate;
 611                         }
 612
 613                         bool MessageMatchesEndpointDispatcher (Message req, EndpointDispatcher endpoint)
 614                         {
 615                                 Uri to = req.Headers.To;
 616                                 if (to == null)
 617                                         return false;
 618                                 if (to.AbsoluteUri == Constants.WsaAnonymousUri)
 619                                         return false;
 620                                 return endpoint.AddressFilter.Match (req) && endpoint.ContractFilter.Match (req);
 621                         }
 622                 }
 623 }