mcs/class/System.Runtime.Remoting/System.Runtime.Remoting.Channels/BinaryServerFormatterSink.cs

   1 //
   2 // System.Runtime.Remoting.Channels.BinaryServerFormatterSink.cs
   3 //
   4 // Author: Duncan Mak (duncan@ximian.com)
   5 //         Lluis Sanchez Gual (lluis@ideary.com)
   6 //
   7 // 2002 (C) Copyright, Ximian, Inc.
   8 //
   9
  10 //
  11 // Permission is hereby granted, free of charge, to any person obtaining
  12 // a copy of this software and associated documentation files (the
  13 // "Software"), to deal in the Software without restriction, including
  14 // without limitation the rights to use, copy, modify, merge, publish,
  15 // distribute, sublicense, and/or sell copies of the Software, and to
  16 // permit persons to whom the Software is furnished to do so, subject to
  17 // the following conditions:
  18 //
  19 // The above copyright notice and this permission notice shall be
  20 // included in all copies or substantial portions of the Software.
  21 //
  22 // THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
  23 // EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
  24 // MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
  25 // NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
  26 // LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
  27 // OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
  28 // WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
  29 //
  30
  31 using System.Collections;
  32 using System.IO;
  33 using System.Runtime.Remoting.Messaging;
  34 using System.Runtime.Serialization;
  35 using System.Runtime.Serialization.Formatters;
  36 using System.Runtime.Serialization.Formatters.Binary;
  37 using System.Runtime.InteropServices;
  38
  39 namespace System.Runtime.Remoting.Channels {
  40
  41         public class BinaryServerFormatterSink : IServerChannelSink, IChannelSinkBase
  42         {
  43                 [Serializable]
  44                 public enum Protocol
  45                 {
  46                         Http = 0,
  47                         Other = 1,
  48                 }
  49
  50                 BinaryCore _binaryCore = BinaryCore.DefaultInstance;
  51
  52                 IServerChannelSink next_sink;
  53                 Protocol protocol;
  54                 IChannelReceiver receiver;
  55
  56                 public BinaryServerFormatterSink (BinaryServerFormatterSink.Protocol protocol,
  57                                                   IServerChannelSink nextSink,
  58                                                   IChannelReceiver receiver)
  59                 {
  60                         this.protocol = protocol;
  61                         this.next_sink = nextSink;
  62                         this.receiver = receiver;
  63                 }
  64
  65                 internal BinaryCore BinaryCore
  66                 {
  67                         get { return _binaryCore; }
  68                         set { _binaryCore = value; }
  69                 }
  70
  71                 public IServerChannelSink NextChannelSink {
  72                         get {
  73                                 return next_sink;
  74                         }
  75                 }
  76
  77                 public IDictionary Properties {
  78                         get {
  79                                 return null;
  80                         }
  81                 }
  82
  83 #if NET_1_1
  84                 [ComVisible(false)]
  85                 public TypeFilterLevel TypeFilterLevel
  86                 {
  87                         get { return _binaryCore.TypeFilterLevel; }
  88                         set
  89                         {
  90                                 IDictionary props = (IDictionary) ((ICloneable)_binaryCore.Properties).Clone ();
  91                                 props ["typeFilterLevel"] = value;
  92                                 _binaryCore = new BinaryCore (this, props, BinaryServerFormatterSinkProvider.AllowedProperties);
  93                         }
  94                 }
  95 #endif
  96
  97                 public void AsyncProcessResponse (IServerResponseChannelSinkStack sinkStack, object state,
  98                                                   IMessage msg, ITransportHeaders headers, Stream stream)
  99                 {
 100                         ITransportHeaders responseHeaders = new TransportHeaders();
 101
 102                         if (sinkStack != null) stream = sinkStack.GetResponseStream (msg, responseHeaders);
 103                         if (stream == null) stream = new MemoryStream();
 104
 105                         _binaryCore.Serializer.Serialize (stream, msg, null);
 106                         if (stream is MemoryStream) stream.Position = 0;
 107
 108                         sinkStack.AsyncProcessResponse (msg, responseHeaders, stream);
 109                 }
 110
 111                 public Stream GetResponseStream (IServerResponseChannelSinkStack sinkStack, object state,
 112                                                 IMessage msg, ITransportHeaders headers)
 113                 {
 114                         return null;
 115                 }
 116
 117                 public ServerProcessing ProcessMessage (IServerChannelSinkStack sinkStack,
 118                                                         IMessage requestMsg, ITransportHeaders requestHeaders, Stream requestStream,
 119                                                         out IMessage responseMsg, out ITransportHeaders responseHeaders, out Stream responseStream)
 120                 {
 121                         // Check whether the request was already processed by another
 122                         // formatter sink and pass the request to the next sink if so.
 123                         if (requestMsg != null)
 124                                 return next_sink.ProcessMessage (sinkStack,
 125                                                                  requestMsg,
 126                                                                  requestHeaders,
 127                                                                  requestStream,
 128                                                                  out responseMsg,
 129                                                                  out responseHeaders,
 130                                                                  out responseStream);
 131
 132                         // Check whether the request is suitable for this formatter
 133                         // and pass the request to the next sink if not.
 134                         // Note that a null content-type is handled as suitable,
 135                         // otherwise no other sink will be able to handle the request.
 136                         string contentType = requestHeaders["Content-Type"] as string;
 137                         if (contentType != null && contentType != "application/octet-stream") {
 138                                 try {
 139                                         return next_sink.ProcessMessage (sinkStack,
 140                                                 requestMsg,
 141                                                 requestHeaders,
 142                                                 requestStream,
 143                                                 out responseMsg,
 144                                                 out responseHeaders,
 145                                                 out responseStream);
 146                                 } catch {
 147                                         // Let this formatter handle the exception.
 148                                 }
 149                         }
 150
 151                         sinkStack.Push (this, null);
 152                         ServerProcessing res;
 153
 154                         try
 155                         {
 156                                 string url = (string)requestHeaders[CommonTransportKeys.RequestUri];
 157                                 string uri;
 158                                 receiver.Parse (url, out uri);
 159                                 if (uri == null) uri = url;
 160
 161                                 MethodCallHeaderHandler mhh = new MethodCallHeaderHandler(uri);
 162                                 requestMsg = (IMessage) _binaryCore.Deserializer.Deserialize (requestStream, new HeaderHandler(mhh.HandleHeaders));
 163
 164                                 res = next_sink.ProcessMessage (sinkStack, requestMsg, requestHeaders, null, out responseMsg, out responseHeaders, out responseStream);
 165                         }
 166                         catch (Exception ex)
 167                         {
 168                                 responseMsg = new ReturnMessage (ex, (IMethodCallMessage)requestMsg);
 169                                 res = ServerProcessing.Complete;
 170                                 responseHeaders = null;
 171                                 responseStream = null;
 172                         }
 173
 174                         if (res == ServerProcessing.Complete)
 175                         {
 176                                 for (int n=0; n<3; n++) {
 177                                         responseStream = null;
 178                                         responseHeaders = new TransportHeaders();
 179
 180                                         if (sinkStack != null) responseStream = sinkStack.GetResponseStream (responseMsg, responseHeaders);
 181                                         if (responseStream == null) responseStream = new MemoryStream();
 182
 183                                         try {
 184                                                 _binaryCore.Serializer.Serialize (responseStream, responseMsg);
 185                                                 break;
 186                                         } catch (Exception ex) {
 187                                                 if (n == 2) throw ex;
 188                                                 else responseMsg = new ReturnMessage (ex, (IMethodCallMessage)requestMsg);
 189                                         }
 190                                 }
 191
 192                                 if (responseStream is MemoryStream) responseStream.Position = 0;
 193
 194
 195                                 sinkStack.Pop (this);
 196                         }
 197                         return res;
 198                 }
 199
 200         }
 201
 202         internal class MethodCallHeaderHandler
 203         {
 204                 string _uri;
 205
 206                 public MethodCallHeaderHandler (string uri)
 207                 {
 208                         _uri = uri;
 209                 }
 210
 211                 public object HandleHeaders (Header[] headers)
 212                 {
 213                         return _uri;
 214                 }
 215         }
 216 }