New tests.
[mono.git] / mcs / class / System.ServiceModel / System.ServiceModel / ServiceHostBase.cs
index 3a87f6022e25a070467d11d57b5a737411ff0cfb..81b5608aabdaa9e6acbd19776e59d7d5e5c23a3a 100644 (file)
@@ -28,6 +28,7 @@
 using System;
 using System.Collections.Generic;
 using System.Collections.ObjectModel;
+using System.Linq;
 using System.ServiceModel.Channels;
 using System.ServiceModel.Configuration;
 using System.ServiceModel.Description;
@@ -40,6 +41,9 @@ namespace System.ServiceModel
        public abstract partial class ServiceHostBase
                : CommunicationObject, IExtensibleObject<ServiceHostBase>, IDisposable
        {
+               // It is used for mapping a ServiceHostBase to HttpChannelListener precisely.
+               internal static ServiceHostBase CurrentServiceHostHack;
+
                ServiceCredentials credentials;
                ServiceDescription description;
                UriSchemeKeyedCollection base_addresses;
@@ -84,16 +88,26 @@ namespace System.ServiceModel
                        }
                }
 
-               internal Uri CreateUri (string sheme, Uri relatieUri) {
-                       Uri baseUri = base_addresses.Contains (sheme) ? base_addresses [sheme] : null;
+               internal Uri CreateUri (string scheme, Uri relativeUri)
+               {
+                       Uri baseUri = base_addresses.Contains (scheme) ? base_addresses [scheme] : null;
 
-                       if (relatieUri == null)
+                       if (relativeUri == null)
                                return baseUri;
-                       if (relatieUri.IsAbsoluteUri)
-                               return relatieUri;
+                       if (relativeUri.IsAbsoluteUri)
+                               return relativeUri;
                        if (baseUri == null)
                                return null;
-                       return new Uri (baseUri, relatieUri);
+                       var s = relativeUri.ToString ();
+                       if (s.Length == 0)
+                               return baseUri;
+                       var l = baseUri.LocalPath;
+                       var r = relativeUri.ToString ();
+
+                       if (l.Length > 0 && l [l.Length - 1] != '/' && r [0] != '/')
+                               return new Uri (String.Concat (baseUri.ToString (), "/", r));
+                       else
+                               return new Uri (String.Concat (baseUri.ToString (), r));
                }
 
                public ChannelDispatcherCollection ChannelDispatchers {
@@ -152,7 +166,7 @@ namespace System.ServiceModel
 
                protected void AddBaseAddress (Uri baseAddress)
                {
-                       if (Description != null)
+                       if (base_addresses == null)
                                throw new InvalidOperationException ("Base addresses must be added before the service description is initialized");
                        base_addresses.Add (baseAddress);
                }
@@ -186,7 +200,7 @@ namespace System.ServiceModel
                        Uri address, Uri listenUri)
                {
                        EndpointAddress ea = BuildEndpointAddress (address, binding);
-                       ContractDescription cd = GetContract (implementedContract);
+                       ContractDescription cd = GetContract (implementedContract, binding.Namespace == "http://schemas.microsoft.com/ws/2005/02/mex/bindings");
                        if (cd == null)
                                throw new InvalidOperationException (String.Format ("Contract '{0}' was not found in the implemented contracts in this service host.", implementedContract));
                        return AddServiceEndpointCore (cd, binding, ea, listenUri);
@@ -205,43 +219,40 @@ namespace System.ServiceModel
                        return null;
                }
 
-               ContractDescription GetContract (string name)
-               {
-                       //FIXME: hack hack hack
-                       ImplementedContracts ["IHttpGetHelpPageAndMetadataContract"] =
-                               ContractDescription.GetContract (typeof (IHttpGetHelpPageAndMetadataContract));
-
-                       // FIXME: As long as I tried, *only* IMetadataExchange
-                       // is the exception case that does not require full
-                       // type name. Hence I treat it as a special case.
-                       if (name == ServiceMetadataBehavior.MexContractName) {
-                               if (!Description.Behaviors.Contains (typeof (ServiceMetadataBehavior)) && Array.IndexOf (Description.ServiceType.GetInterfaces (), typeof (IMetadataExchange)) < 0)
-                                       throw new InvalidOperationException (
-                                               "Add ServiceMetadataBehavior to the ServiceHost to add a endpoint for IMetadataExchange contract.");
-                                       
-                               ImplementedContracts [ServiceMetadataBehavior.MexContractName] =
-                                       ContractDescription.GetContract (typeof (IMetadataExchange));
-
-                               foreach (ContractDescription cd in ImplementedContracts.Values)
-                                       if (cd.ContractType == typeof (IMetadataExchange))
-                                               return cd;
-                               return null;
+               ContractDescription mex_contract, help_page_contract;
+
+               ContractDescription GetContract (string name, bool mexBinding)
+               {
+                       // FIXME: not sure if they should really be special cases.
+                       switch (name) {
+                       case "IHttpGetHelpPageAndMetadataContract":
+                               if (help_page_contract == null)
+                                       help_page_contract = ContractDescription.GetContract (typeof (IHttpGetHelpPageAndMetadataContract));
+                               return help_page_contract;
+                       case "IMetadataExchange":
+                               // this is certainly looking special (or we may 
+                               // be missing something around ServiceMetadataExtension).
+                               // It seems .NET WCF has some "infrastructure"
+                               // endpoints. .NET ServiceHost fails to Open()
+                               // if it was added only IMetadataExchange 
+                               // endpoint (and you'll see the word
+                               // "infrastructure" in the exception message).
+                               if (mexBinding && Description.Behaviors.Find<ServiceMetadataBehavior> () == null)
+                                       break;
+                               if (mex_contract == null)
+                                       mex_contract = ContractDescription.GetContract (typeof (IMetadataExchange));
+                               return mex_contract;
                        }
 
                        Type type = PopulateType (name);
+                       if (type == null)
+                               return null;
 
                        foreach (ContractDescription cd in ImplementedContracts.Values) {
-                               // FIXME: This check is a negative side effect 
-                               // of the above hack.
+                               // This check is a negative side effect of the above match-by-name design.
                                if (cd.ContractType == typeof (IMetadataExchange))
                                        continue;
 
-                               if (type == null) {
-                                       if (cd.Name == name)
-                                               return cd;
-                                       continue;
-                               }
-
                                if (cd.ContractType == type ||
                                    cd.ContractType.IsSubclassOf (type) ||
                                    type.IsInterface && cd.ContractType.GetInterface (type.FullName) == type)
@@ -271,7 +282,7 @@ namespace System.ServiceModel
                        ContractDescription cd, Binding binding, EndpointAddress address, Uri listenUri)
                {
                        foreach (ServiceEndpoint e in Description.Endpoints)
-                               if (e.Contract == cd)
+                               if (e.Contract == cd && e.Binding == binding && e.Address == address && e.ListenUri.Equals (listenUri))
                                        return e;
                        ServiceEndpoint se = new ServiceEndpoint (cd, binding, address);
                        se.ListenUri = listenUri.IsAbsoluteUri ? listenUri : new Uri (address.Uri, listenUri);
@@ -279,7 +290,6 @@ namespace System.ServiceModel
                        return se;
                }
 
-               [MonoTODO]
                protected virtual void ApplyConfiguration ()
                {
                        if (Description == null)
@@ -296,23 +306,28 @@ namespace System.ServiceModel
                                        AddBaseAddress (new Uri (baseAddress.BaseAddress));
                                }
 
+                               // behaviors
+                               // TODO: use EvaluationContext of ServiceElement.
+                               ServiceBehaviorElement behavior = ConfigUtil.BehaviorsSection.ServiceBehaviors [service.BehaviorConfiguration];
+                               if (behavior != null) {
+                                       foreach (var bxe in behavior) {
+                                               IServiceBehavior b = (IServiceBehavior) bxe.CreateBehavior ();
+                                               Description.Behaviors.Add (b);
+                                       }
+                               }
+
                                // services
                                foreach (ServiceEndpointElement endpoint in service.Endpoints) {
-                                       // FIXME: consider BindingName as well
                                        ServiceEndpoint se = AddServiceEndpoint (
                                                endpoint.Contract,
                                                ConfigUtil.CreateBinding (endpoint.Binding, endpoint.BindingConfiguration),
                                                endpoint.Address.ToString ());
-                               }
-                               // behaviors
-                               // TODO: use EvaluationContext of ServiceElement.
-                               ServiceBehaviorElement behavior = ConfigUtil.BehaviorsSection.ServiceBehaviors.Find (service.BehaviorConfiguration);
-                               if (behavior != null) {
-                                       for (int i = 0; i < behavior.Count; i++) {
-                                               BehaviorExtensionElement bxel = behavior [i];
-                                               IServiceBehavior b = (IServiceBehavior) behavior [i].CreateBehavior ();
-                                               if (b != null)
-                                                       Description.Behaviors.Add (b);
+                                       // endpoint behaviors
+                                       EndpointBehaviorElement epbehavior = ConfigUtil.BehaviorsSection.EndpointBehaviors [endpoint.BehaviorConfiguration];
+                                       if (epbehavior != null)
+                                               foreach (var bxe in epbehavior) {
+                                                       IEndpointBehavior b = (IEndpointBehavior) bxe.CreateBehavior ();
+                                                       se.Behaviors.Add (b);
                                        }
                                }
                        }
@@ -341,14 +356,6 @@ namespace System.ServiceModel
                        return ConfigUtil.ServicesSection.Services [serviceType.FullName];                      
                }
 
-               internal ContractDescription GetContract (string name, string ns)
-               {
-                       foreach (ContractDescription d in ImplementedContracts.Values)
-                               if (d.Name == name && d.Namespace == ns)
-                                       return d;
-                       return null;
-               }
-
                protected abstract ServiceDescription CreateDescription (
                        out IDictionary<string,ContractDescription> implementedContracts);
 
@@ -371,18 +378,18 @@ namespace System.ServiceModel
                        //Build all ChannelDispatchers, one dispatcher per user configured EndPoint.
                        //We must keep thet ServiceEndpoints as a seperate collection, since the user
                        //can change the collection in the description during the behaviors events.
-                       Dictionary<ServiceEndpoint, ChannelDispatcher> endPointToDispatcher = new Dictionary<ServiceEndpoint,ChannelDispatcher>();
                        ServiceEndpoint[] endPoints = new ServiceEndpoint[Description.Endpoints.Count];
                        Description.Endpoints.CopyTo (endPoints, 0);
+                       var builder = new DispatcherBuilder (this);
                        foreach (ServiceEndpoint se in endPoints) {
 
                                var commonParams = new BindingParameterCollection ();
                                foreach (IServiceBehavior b in Description.Behaviors)
                                        b.AddBindingParameters (Description, this, Description.Endpoints, commonParams);
 
-                               ChannelDispatcher channel = BuildChannelDispatcher (se, commonParams);
-                               ChannelDispatchers.Add (channel);
-                               endPointToDispatcher[se] = channel;
+                               var channel = builder.BuildChannelDispatcher (Description.ServiceType, se, commonParams);
+                               if (!ChannelDispatchers.Contains (channel))
+                                       ChannelDispatchers.Add (channel);
                        }
 
                        //After the ChannelDispatchers are created, and attached to the service host
@@ -390,9 +397,7 @@ namespace System.ServiceModel
                        foreach (IServiceBehavior b in Description.Behaviors)
                                b.ApplyDispatchBehavior (Description, this);
 
-                       foreach(KeyValuePair<ServiceEndpoint, ChannelDispatcher> val in endPointToDispatcher)
-                               foreach (var ed in val.Value.Endpoints)
-                                       ApplyDispatchBehavior (ed, val.Key);                    
+                       builder.ApplyDispatchBehaviors ();
                }
 
                private void ValidateDescription ()
@@ -401,24 +406,9 @@ namespace System.ServiceModel
                                b.Validate (Description, this);
                        foreach (ServiceEndpoint endPoint in Description.Endpoints)
                                endPoint.Validate ();
-               }               
-
-               private void ApplyDispatchBehavior (EndpointDispatcher ed, ServiceEndpoint endPoint)
-               {
-                       foreach (IContractBehavior b in endPoint.Contract.Behaviors)
-                               b.ApplyDispatchBehavior (endPoint.Contract, endPoint, ed.DispatchRuntime);
-                       foreach (IEndpointBehavior b in endPoint.Behaviors)
-                               b.ApplyDispatchBehavior (endPoint, ed);
-                       foreach (OperationDescription operation in endPoint.Contract.Operations) {
-                               foreach (IOperationBehavior b in operation.Behaviors)
-                                       b.ApplyDispatchBehavior (operation, ed.DispatchRuntime.Operations [operation.Name]);
-                       }
 
-               }
-
-               internal ChannelDispatcher BuildChannelDispatcher (ServiceEndpoint se, BindingParameterCollection commonParams)
-               {
-                       return new DispatcherBuilder ().BuildChannelDispatcher (Description.ServiceType, se, commonParams);
+                       if (Description.Endpoints.FirstOrDefault (e => e.Contract != mex_contract) == null)
+                               throw new InvalidOperationException ("The ServiceHost must have at least one application endpoint (that does not include metadata exchange contract) defined by either configuration, behaviors or call to AddServiceEndpoint methods.");
                }
 
                [MonoTODO]
@@ -427,16 +417,6 @@ namespace System.ServiceModel
                        ServicesSection services = ConfigUtil.ServicesSection;
                }
 
-               void DoOpen (TimeSpan timeout)
-               {
-                       foreach (var cd in ChannelDispatchers) {
-                               cd.Open (timeout);
-                               // This is likely hack.
-                               if (cd is ChannelDispatcher)
-                                       ((ChannelDispatcher) cd).StartLoop ();
-                       }
-               }
-
                [MonoTODO]
                protected override sealed void OnAbort ()
                {
@@ -482,8 +462,13 @@ namespace System.ServiceModel
 
                protected override sealed void OnOpen (TimeSpan timeout)
                {
+                       DateTime start = DateTime.Now;
                        InitializeRuntime ();
-                       DoOpen (timeout);
+                       foreach (var cd in ChannelDispatchers)
+                               cd.Open (timeout - (DateTime.Now - start));
+
+                       // FIXME: remove this hack. It should make sure that each ChannelDispatcher's loop has started, using WaitHandle.WaitAll() or something similar.
+                       System.Threading.Thread.Sleep (300);
                }
 
                protected override void OnEndClose (IAsyncResult result)
@@ -502,6 +487,7 @@ namespace System.ServiceModel
 
                protected override void OnOpened ()
                {
+                       base.OnOpened ();
                }
 
                [MonoTODO]
@@ -598,19 +584,76 @@ namespace System.ServiceModel
                */
        }
 
+       /// <summary>
+       ///  Builds ChannelDispatchers as appropriate to service the service endpoints. 
+       /// </summary>
+       /// <remarks>Will re-use ChannelDispatchers when two endpoint uris are the same</remarks>
        partial class DispatcherBuilder
        {
+               ServiceHostBase host;
+
+               public DispatcherBuilder (ServiceHostBase host)
+               {
+                       this.host = host;
+               }
+
+               List<ChannelDispatcher> built_dispatchers = new List<ChannelDispatcher> ();
+               Dictionary<ServiceEndpoint, EndpointDispatcher> ep_to_dispatcher_ep = new Dictionary<ServiceEndpoint, EndpointDispatcher> ();
+
+               internal static Action<ChannelDispatcher> ChannelDispatcherSetter;
+
                internal ChannelDispatcher BuildChannelDispatcher (Type serviceType, ServiceEndpoint se, BindingParameterCollection commonParams)
                {
                        //Let all behaviors add their binding parameters
                        AddBindingParameters (commonParams, se);
-                       //User the binding parameters to build the channel listener and Dispatcher
-                       IChannelListener lf = BuildListener (se, commonParams);
-                       ChannelDispatcher cd = new ChannelDispatcher (
-                               lf, se.Binding.Name);
-                       cd.InitializeServiceEndpoint (serviceType, se);
+                       
+                       // See if there's an existing channel that matches this endpoint
+                       ChannelDispatcher cd = FindExistingDispatcher (se);
+                       EndpointDispatcher ep;
+                       if (cd != null) {
+                               ep = cd.InitializeServiceEndpoint (serviceType, se);
+                       } else {
+                               // Use the binding parameters to build the channel listener and Dispatcher.
+                               lock (HttpTransportBindingElement.ListenerBuildLock) {
+                                       ServiceHostBase.CurrentServiceHostHack = host;
+                                       IChannelListener lf = BuildListener (se, commonParams);
+                                       cd = new ChannelDispatcher (lf, se.Binding.Name);
+                                       if (ChannelDispatcherSetter != null) {
+                                               ChannelDispatcherSetter (cd);
+                                               ChannelDispatcherSetter = null;
+                                       }
+                                       ServiceHostBase.CurrentServiceHostHack = null;
+                               }
+                               ep = cd.InitializeServiceEndpoint (serviceType, se);
+                               built_dispatchers.Add (cd);
+                       }
+                       ep_to_dispatcher_ep[se] = ep;
                        return cd;
                }
+               
+               ChannelDispatcher FindExistingDispatcher (ServiceEndpoint se)
+               {
+                       return built_dispatchers.FirstOrDefault ((ChannelDispatcher cd) => (cd.Listener.Uri.Equals (se.ListenUri)) && cd.MessageVersion.Equals (se.Binding.MessageVersion));
+               }
+
+               internal void ApplyDispatchBehaviors ()
+               {
+                       foreach (KeyValuePair<ServiceEndpoint, EndpointDispatcher> val in ep_to_dispatcher_ep)
+                               ApplyDispatchBehavior (val.Value, val.Key);
+               }
+               
+               private void ApplyDispatchBehavior (EndpointDispatcher ed, ServiceEndpoint endPoint)
+               {
+                       foreach (IContractBehavior b in endPoint.Contract.Behaviors)
+                               b.ApplyDispatchBehavior (endPoint.Contract, endPoint, ed.DispatchRuntime);
+                       foreach (IEndpointBehavior b in endPoint.Behaviors)
+                               b.ApplyDispatchBehavior (endPoint, ed);
+                       foreach (OperationDescription operation in endPoint.Contract.Operations) {
+                               foreach (IOperationBehavior b in operation.Behaviors)
+                                       b.ApplyDispatchBehavior (operation, ed.DispatchRuntime.Operations [operation.Name]);
+                       }
+
+               }
 
                private void AddBindingParameters (BindingParameterCollection commonParams, ServiceEndpoint endPoint) {