5 // Atsushi Enomoto <atsushi@ximian.com>
7 // Copyright (C) 2006 Novell, Inc. http://www.novell.com
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:
17 // The above copyright notice and this permission notice shall be
18 // included in all copies or substantial portions of the Software.
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.
29 using System.Collections.Generic;
32 using System.Net.Security;
33 using System.ServiceModel;
34 using System.ServiceModel.Description;
35 using System.ServiceModel.Security;
36 using System.Threading;
38 namespace System.ServiceModel.Channels
40 internal abstract class InternalReplyChannelBase : ReplyChannelBase
42 public InternalReplyChannelBase (ChannelListenerBase listener)
45 local_address = new EndpointAddress (listener.Uri);
48 EndpointAddress local_address;
50 public override EndpointAddress LocalAddress {
51 get { return local_address; }
55 internal abstract class ReplyChannelBase : ChannelBase, IReplyChannel
57 public ReplyChannelBase (ChannelListenerBase listener)
60 this.listener = listener;
63 ChannelListenerBase listener;
65 public ChannelListenerBase Listener {
66 get { return listener; }
69 public abstract EndpointAddress LocalAddress { get; }
71 public override T GetProperty<T> ()
73 if (typeof (T) == typeof (MessageVersion) && listener is IHasMessageEncoder)
74 return (T) (object) ((IHasMessageEncoder) listener).MessageEncoder.MessageVersion;
75 if (typeof (T) == typeof (IChannelListener))
76 return (T) (object) listener;
77 return base.GetProperty<T> ();
80 // FIXME: this is wrong. Implement all of them in each channel.
81 protected override void OnAbort ()
83 OnClose (TimeSpan.Zero);
86 protected override void OnClose (TimeSpan timeout)
88 if (currentAsyncThreads.Count > 0)
89 if (!CancelAsync (timeout))
90 foreach (Thread asyncThread in currentAsyncThreads)
94 public virtual bool CancelAsync (TimeSpan timeout)
96 // FIXME: It should wait for the actual completion.
97 return currentAsyncResults.Count > 0;
98 //return CurrentAsyncResult == null || CurrentAsyncResult.AsyncWaitHandle.WaitOne (timeout);
101 public virtual bool TryReceiveRequest ()
103 RequestContext dummy;
104 return TryReceiveRequest (DefaultReceiveTimeout, out dummy);
107 public abstract bool TryReceiveRequest (TimeSpan timeout, out RequestContext context);
109 delegate bool TryReceiveDelegate (TimeSpan timeout, out RequestContext context);
110 TryReceiveDelegate try_recv_delegate;
112 object async_result_lock = new object ();
113 HashSet<Thread> currentAsyncThreads = new HashSet<Thread>();
114 HashSet<IAsyncResult> currentAsyncResults = new HashSet<IAsyncResult>();
116 public virtual IAsyncResult BeginTryReceiveRequest (TimeSpan timeout, AsyncCallback callback, object state)
118 IAsyncResult result = null;
120 if (try_recv_delegate == null)
121 try_recv_delegate = new TryReceiveDelegate (delegate (TimeSpan tout, out RequestContext ctx) {
122 lock (async_result_lock) {
123 if (currentAsyncResults.Contains (result))
124 currentAsyncThreads.Add (Thread.CurrentThread);
127 return TryReceiveRequest (tout, out ctx);
129 lock (async_result_lock) {
130 currentAsyncResults.Remove (result);
131 currentAsyncThreads.Remove (Thread.CurrentThread);
135 RequestContext dummy;
136 lock (async_result_lock) {
137 result = try_recv_delegate.BeginInvoke (timeout, out dummy, callback, state);
138 currentAsyncResults.Add (result);
140 // Note that at this point result can be missing from currentAsyncResults here if delegate has run to completion
144 public virtual bool EndTryReceiveRequest (IAsyncResult result)
146 RequestContext dummy;
147 return EndTryReceiveRequest (result, out dummy);
150 public virtual bool EndTryReceiveRequest (IAsyncResult result, out RequestContext context)
152 if (try_recv_delegate == null)
153 throw new InvalidOperationException ("BeginTryReceiveRequest operation has not started");
154 return try_recv_delegate.EndInvoke (out context, result);
157 public virtual bool WaitForRequest ()
159 return WaitForRequest (DefaultReceiveTimeout);
162 public abstract bool WaitForRequest (TimeSpan timeout);
164 Func<TimeSpan,bool> wait_delegate;
166 public virtual IAsyncResult BeginWaitForRequest (TimeSpan timeout, AsyncCallback callback, object state)
168 if (wait_delegate == null)
169 wait_delegate = new Func<TimeSpan,bool> (WaitForRequest);
170 return wait_delegate.BeginInvoke (timeout, callback, state);
173 public virtual bool EndWaitForRequest (IAsyncResult result)
175 if (wait_delegate == null)
176 throw new InvalidOperationException ("BeginWaitForRequest operation has not started");
177 return wait_delegate.EndInvoke (result);
180 public virtual RequestContext ReceiveRequest ()
182 return ReceiveRequest (DefaultReceiveTimeout);
185 public abstract RequestContext ReceiveRequest (TimeSpan timeout);
187 public virtual IAsyncResult BeginReceiveRequest (AsyncCallback callback, object state)
189 return BeginReceiveRequest (DefaultReceiveTimeout, callback, state);
192 Func<TimeSpan,RequestContext> recv_delegate;
193 public virtual IAsyncResult BeginReceiveRequest (TimeSpan timeout, AsyncCallback callback, object state)
195 if (recv_delegate == null)
196 recv_delegate = new Func<TimeSpan,RequestContext> (ReceiveRequest);
197 return recv_delegate.BeginInvoke (timeout, callback, state);
200 public virtual RequestContext EndReceiveRequest (IAsyncResult result)
202 if (recv_delegate == null)
203 throw new InvalidOperationException ("BeginReceiveRequest operation has not started");
204 return recv_delegate.EndInvoke (result);
207 Action<TimeSpan> open_delegate, close_delegate;
209 protected override IAsyncResult OnBeginOpen (TimeSpan timeout,
210 AsyncCallback callback, object state)
212 if (open_delegate == null)
213 open_delegate = new Action<TimeSpan> (OnOpen);
214 return open_delegate.BeginInvoke (timeout, callback, state);
217 protected override void OnEndOpen (IAsyncResult result)
219 if (open_delegate == null)
220 throw new InvalidOperationException ("async open operation has not started");
221 open_delegate.EndInvoke (result);
224 protected override IAsyncResult OnBeginClose (TimeSpan timeout,
225 AsyncCallback callback, object state)
227 if (close_delegate == null)
228 close_delegate = new Action<TimeSpan> (OnClose);
229 return close_delegate.BeginInvoke (timeout, callback, state);
232 protected override void OnEndClose (IAsyncResult result)
234 if (close_delegate == null)
235 throw new InvalidOperationException ("async close operation has not started");
236 close_delegate.EndInvoke (result);