Merge pull request #5198 from BrzVlad/fix-binprot-stats
[mono.git] / mcs / class / System.ServiceModel / old-code / HttpReplyChannel.cs
1 //
2 // HttpReplyChannel.cs
3 //
4 // Author:
5 //      Atsushi Enomoto <atsushi@ximian.com>
6 //
7 // Copyright (C) 2006 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.Specialized;
31 using System.IO;
32 using System.Net;
33 using System.ServiceModel;
34 using System.Text;
35 using System.Threading;
36
37 namespace System.ServiceModel.Channels
38 {
39         internal class HttpSimpleReplyChannel : HttpReplyChannel
40         {
41                 HttpSimpleChannelListener<IReplyChannel> source;
42                 List<HttpListenerContext> waiting = new List<HttpListenerContext> ();
43                 RequestContext reqctx;
44
45                 public HttpSimpleReplyChannel (HttpSimpleChannelListener<IReplyChannel> listener)
46                         : base (listener)
47                 {
48                         this.source = listener;
49                 }
50
51                 protected override void OnAbort ()
52                 {
53                         AbortConnections (TimeSpan.Zero);
54                         base.OnAbort (); // FIXME: remove it. The base is wrong. But it is somehow required to not block some tests.
55                 }
56
57                 public override bool CancelAsync (TimeSpan timeout)
58                 {
59                         AbortConnections (timeout);
60                         // FIXME: this wait is sort of hack (because it should not be required), but without it some tests are blocked.
61                         // This hack even had better be moved to base.CancelAsync().
62                         if (CurrentAsyncResult != null)
63                                 CurrentAsyncResult.AsyncWaitHandle.WaitOne (TimeSpan.FromMilliseconds (300));
64                         return base.CancelAsync (timeout);
65                 }
66
67                 void SignalAsyncWait ()
68                 {
69                         if (wait == null)
70                                 return;
71                         var wait_ = wait;
72                         wait = null;
73                         wait_.Set ();
74                 }
75
76                 void AbortConnections (TimeSpan timeout)
77                 {
78                         // FIXME: use timeout
79                         lock (waiting)
80                                 foreach (var ctx in waiting)
81                                         ctx.Response.Close ();
82                         if (wait != null)
83                                 source.ListenerManager.CancelGetHttpContextAsync ();
84                 }
85
86                 protected override void OnClose (TimeSpan timeout)
87                 {
88                         DateTime start = DateTime.UtcNow;
89                         if (reqctx != null)
90                                 reqctx.Close (timeout);
91
92                         // FIXME: consider timeout
93                         AbortConnections (timeout - (DateTime.UtcNow - start));
94
95                         base.OnClose (timeout - (DateTime.UtcNow - start));
96                 }
97
98                 public override bool TryReceiveRequest (TimeSpan timeout, out RequestContext context)
99                 {
100                         context = null;
101                         if (waiting.Count == 0 && !WaitForRequest (timeout))
102                                 return false;
103                         HttpListenerContext ctx = null;
104                         lock (waiting) {
105                                 if (waiting.Count > 0) {
106                                         ctx = waiting [0];
107                                         waiting.RemoveAt (0);
108                                 }
109                         }
110                         if (ctx == null)
111                                 // Though as long as this instance is used
112                                 // synchronously, it should not happen.
113                                 return false;
114                         if (ctx.Response.StatusCode != 200) {
115                                 ctx.Response.Close ();
116                                 return false;
117                         }
118
119                         // FIXME: supply maxSizeOfHeaders.
120                         int maxSizeOfHeaders = 0x10000;
121
122                         Message msg = null;
123
124                         if (ctx.Request.HttpMethod == "POST") {
125                                 if (!Encoder.IsContentTypeSupported (ctx.Request.ContentType)) {
126                                         ctx.Response.StatusCode = (int) HttpStatusCode.UnsupportedMediaType;
127                                         ctx.Response.StatusDescription = String.Format (
128                                                         "Expected content-type '{0}' but got '{1}'", Encoder.ContentType, ctx.Request.ContentType);
129                                         ctx.Response.Close ();
130
131                                         return false;
132                                 }
133
134                                 msg = Encoder.ReadMessage (
135                                         ctx.Request.InputStream, maxSizeOfHeaders);
136
137                                 if (MessageVersion.Envelope.Equals (EnvelopeVersion.Soap11) ||
138                                     MessageVersion.Addressing.Equals (AddressingVersion.None)) {
139                                         string action = GetHeaderItem (ctx.Request.Headers ["SOAPAction"]);
140                                         if (action != null) {
141                                                 if (action.Length > 2 && action [0] == '"' && action [action.Length] == '"')
142                                                         action = action.Substring (1, action.Length - 2);
143                                                 msg.Headers.Action = action;
144                                         }
145                                 }
146                         } else if (ctx.Request.HttpMethod == "GET") {
147                                 msg = Message.CreateMessage (MessageVersion.None, null);
148                         }
149                         if (msg.Headers.To == null)
150                                 msg.Headers.To = ctx.Request.Url;
151                         msg.Properties.Add ("Via", LocalAddress.Uri);
152                         msg.Properties.Add (HttpRequestMessageProperty.Name, CreateRequestProperty (ctx.Request.HttpMethod, ctx.Request.Url.Query, ctx.Request.Headers));
153 /*
154 MessageBuffer buf = msg.CreateBufferedCopy (0x10000);
155 msg = buf.CreateMessage ();
156 Console.WriteLine (buf.CreateMessage ());
157 */
158                         context = new HttpRequestContext (this, msg, ctx);
159                         reqctx = context;
160                         return true;
161                 }
162
163                 ManualResetEvent wait;
164
165                 public override bool WaitForRequest (TimeSpan timeout)
166                 {
167                         if (wait != null)
168                                 throw new InvalidOperationException ("Another wait operation is in progress");
169                         try {
170                                 var wait_ = new ManualResetEvent (false);
171                                 wait = wait_;   // wait can be set to null if HttpContextAcquired runs to completion before we do WaitOne
172                                 source.ListenerManager.GetHttpContextAsync (timeout, HttpContextAcquired);
173                                 return wait_.WaitOne (timeout, false) && waiting.Count > 0;
174                         } catch (HttpListenerException e) {
175                                 // FIXME: does this make sense? I doubt.
176                                 if ((uint) e.ErrorCode == 0x80004005) // invalid handle. Happens during shutdown.
177                                         while (true) Thread.Sleep (1000); // thread is about to be terminated.
178                                 throw;
179                         } catch (ObjectDisposedException) {
180                                 return false;
181                         } finally {
182                                 wait = null;
183                         }
184                 }
185
186                 void HttpContextAcquired (HttpContextInfo ctx)
187                 {
188                         if (wait == null)
189                                 throw new InvalidOperationException ("WaitForRequest operation has not started");
190                         var sctx = (HttpListenerContextInfo) ctx;
191                         if (State == CommunicationState.Opened && ctx != null)
192                                 lock (waiting)
193                                         waiting.Add (sctx.Source);
194                         SignalAsyncWait ();
195                 }
196         }
197
198         internal abstract class HttpReplyChannel : InternalReplyChannelBase
199         {
200                 HttpChannelListenerBase<IReplyChannel> source;
201
202                 public HttpReplyChannel (HttpChannelListenerBase<IReplyChannel> listener)
203                         : base (listener)
204                 {
205                         this.source = listener;
206                 }
207
208                 public MessageEncoder Encoder {
209                         get { return source.MessageEncoder; }
210                 }
211
212                 internal MessageVersion MessageVersion {
213                         get { return source.MessageEncoder.MessageVersion; }
214                 }
215
216                 public override RequestContext ReceiveRequest (TimeSpan timeout)
217                 {
218                         RequestContext ctx;
219                         TryReceiveRequest (timeout, out ctx);
220                         return ctx;
221                 }
222
223                 protected override void OnOpen (TimeSpan timeout)
224                 {
225                 }
226
227                 protected string GetHeaderItem (string raw)
228                 {
229                         if (raw == null || raw.Length == 0)
230                                 return raw;
231                         switch (raw [0]) {
232                         case '\'':
233                         case '"':
234                                 if (raw [raw.Length - 1] == raw [0])
235                                         return raw.Substring (1, raw.Length - 2);
236                                 // FIXME: is it simply an error?
237                                 break;
238                         }
239                         return raw;
240                 }
241
242                 protected HttpRequestMessageProperty CreateRequestProperty (string method, string query, NameValueCollection headers)
243                 {
244                         var prop = new HttpRequestMessageProperty ();
245                         prop.Method = method;
246                         prop.QueryString = query.StartsWith ("?") ? query.Substring (1) : query;
247                         // FIXME: prop.SuppressEntityBody
248                         prop.Headers.Add (headers);
249                         return prop;
250                 }
251         }
252 }