* roottypes.cs: Rename from tree.cs.

[mono.git] / mcs / class / System.Runtime.Remoting / System.Runtime.Remoting.Channels / BinaryServerFormatterSink.cs

diff --git a/mcs/class/System.Runtime.Remoting/System.Runtime.Remoting.Channels/BinaryServerFormatterSink.cs b/mcs/class/System.Runtime.Remoting/System.Runtime.Remoting.Channels/BinaryServerFormatterSink.cs
index fdcf649c47b0cb26cf33347ee32423bf39b1e43d..a7ba64956e5dd881467a686b4d7df95aeae1f4d9 100644 (file)
--- a/mcs/class/System.Runtime.Remoting/System.Runtime.Remoting.Channels/BinaryServerFormatterSink.cs
+++ b/mcs/class/System.Runtime.Remoting/System.Runtime.Remoting.Channels/BinaryServerFormatterSink.cs
@@ -7,6 +7,27 @@
 // 2002 (C) Copyright, Ximian, Inc.
 //
 
+//
+// Permission is hereby granted, free of charge, to any person obtaining
+// a copy of this software and associated documentation files (the
+// "Software"), to deal in the Software without restriction, including
+// without limitation the rights to use, copy, modify, merge, publish,
+// distribute, sublicense, and/or sell copies of the Software, and to
+// permit persons to whom the Software is furnished to do so, subject to
+// the following conditions:
+// 
+// The above copyright notice and this permission notice shall be
+// included in all copies or substantial portions of the Software.
+// 
+// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
+// EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
+// MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
+// NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
+// LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
+// OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
+// WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
+//
+
 using System.Collections;
 using System.IO;
 using System.Runtime.Remoting.Messaging;
@@ -97,6 +118,36 @@ namespace System.Runtime.Remoting.Channels {
                                                        IMessage requestMsg, ITransportHeaders requestHeaders, Stream requestStream,
                                                        out IMessage responseMsg, out ITransportHeaders responseHeaders, out Stream responseStream)
                {
+                       // Check whether the request was already processed by another
+                       // formatter sink and pass the request to the next sink if so.
+                       if (requestMsg != null)
+                               return next_sink.ProcessMessage (sinkStack,
+                                                                requestMsg,
+                                                                requestHeaders,
+                                                                requestStream,
+                                                                out responseMsg,
+                                                                out responseHeaders,
+                                                                out responseStream);
+
+                       // Check whether the request is suitable for this formatter
+                       // and pass the request to the next sink if not.
+                       // Note that a null content-type is handled as suitable,
+                       // otherwise no other sink will be able to handle the request.
+                       string contentType = requestHeaders["Content-Type"] as string;
+                       if (contentType != null && contentType != "application/octet-stream") {
+                               try {
+                                       return next_sink.ProcessMessage (sinkStack,
+                                               requestMsg,
+                                               requestHeaders,
+                                               requestStream,
+                                               out responseMsg,
+                                               out responseHeaders,
+                                               out responseStream);
+                               } catch {
+                                       // Let this formatter handle the exception.
+                               }
+                       }
+
                        sinkStack.Push (this, null);
                        ServerProcessing res;
 
@@ -119,17 +170,27 @@ namespace System.Runtime.Remoting.Channels {
                                responseHeaders = null;
                                responseStream = null;
                        }
-
+                       
                        if (res == ServerProcessing.Complete)
                        {
-                               responseStream = null;
-                               responseHeaders = new TransportHeaders();
-
-                               if (sinkStack != null) responseStream = sinkStack.GetResponseStream (responseMsg, responseHeaders);
-                               if (responseStream == null) responseStream = new MemoryStream();
-
-                               _binaryCore.Serializer.Serialize (responseStream, responseMsg);
+                               for (int n=0; n<3; n++) {
+                                       responseStream = null;
+                                       responseHeaders = new TransportHeaders();
+
+                                       if (sinkStack != null) responseStream = sinkStack.GetResponseStream (responseMsg, responseHeaders);
+                                       if (responseStream == null) responseStream = new MemoryStream();
+
+                                       try {
+                                               _binaryCore.Serializer.Serialize (responseStream, responseMsg);
+                                               break;
+                                       } catch (Exception ex) {
+                                               if (n == 2) throw ex;
+                                               else responseMsg = new ReturnMessage (ex, (IMethodCallMessage)requestMsg);
+                                       }
+                               }
+                               
                                if (responseStream is MemoryStream) responseStream.Position = 0;
+                               
 
                                sinkStack.Pop (this);
                        }