class/System.ServiceModel/System.ServiceModel.Channels/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.Now;
  89                         if (reqctx != null)
  90                                 reqctx.Close (timeout);
  91
  92                         // FIXME: consider timeout
  93                         AbortConnections (timeout - (DateTime.Now - start));
  94
  95                         base.OnClose (timeout - (DateTime.Now - 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, 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 }