2004-04-14 Sebastien Pouliot <sebastien@ximian.com>
[mono.git] / mcs / class / corlib / System.Runtime.Remoting / RemotingServices.cs
index b3dbef05a623636e84547c3790a1906d4ebfa094..09f649274a0636be6c138edf3a968b792c24fd31 100644 (file)
@@ -10,6 +10,7 @@
 //
 
 using System;
+using System.Text;
 using System.Reflection;
 using System.Threading;
 using System.Collections;
@@ -36,7 +37,7 @@ namespace System.Runtime.Remoting
                static RemotingServices ()
                {
                        RegisterInternalChannels ();
-                       app_id = "/" + Guid.NewGuid().ToString().Replace('-', '_') + "/";
+                       app_id = Guid.NewGuid().ToString().Replace('-', '_') + "/";
                        CreateWellKnownServerIdentity (typeof(RemoteActivator), "RemoteActivationService.rem", WellKnownObjectMode.Singleton);
                }
        
@@ -55,17 +56,36 @@ namespace System.Runtime.Remoting
                        ReturnMessage result;
                        
                        MonoMethod method = (MonoMethod) target.GetType().GetMethod(reqMsg.MethodName, BindingFlags.Public|BindingFlags.NonPublic|BindingFlags.Instance, null, (Type[]) reqMsg.MethodSignature, null);
-
-                       try {
+                       object oldContext = CallContext.SetCurrentCallContext (reqMsg.LogicalCallContext);
+                       
+                       try 
+                       {
                                object [] out_args;
                                object rval = InternalExecute (method, target, reqMsg.Args, out out_args);
-                               result = new ReturnMessage (rval, out_args, out_args.Length,
-                                                           reqMsg.LogicalCallContext, reqMsg);
                        
-                       } catch (Exception e) {
+                               // Collect parameters with Out flag from the request message
+
+                               ParameterInfo[] parameters = method.GetParameters();
+                               object[] returnArgs = new object [parameters.Length];
+                               
+                               int n = 0;
+                               int noa = 0;
+                               foreach (ParameterInfo par in parameters)
+                               {
+                                       if (par.IsOut && !par.ParameterType.IsByRef) 
+                                               returnArgs [n++] = reqMsg.GetArg (par.Position);
+                                       else if (par.ParameterType.IsByRef)
+                                               returnArgs [n++] = out_args [noa++]; 
+                               }
+                               
+                               result = new ReturnMessage (rval, returnArgs, n, CallContext.CreateLogicalCallContext(), reqMsg);
+                       } 
+                       catch (Exception e) 
+                       {
                                result = new ReturnMessage (e, reqMsg);
                        }
-
+                       
+                       CallContext.RestoreCallContext (oldContext);
                        return result;
                }
 
@@ -109,7 +129,7 @@ namespace System.Runtime.Remoting
                                if (proxy.GetProxiedType().IsContextful && (proxy.ObjectIdentity is ServerIdentity))
                                        identity = proxy.ObjectIdentity as ServerIdentity;
                                else
-                                       throw new ArgumentException ("The obj parameter is a proxy");
+                                       throw new ArgumentException ("The obj parameter is a proxy.");
                        }
                        else
                                identity = obj.ObjectIdentity;
@@ -126,7 +146,7 @@ namespace System.Runtime.Remoting
 
                public static Type GetServerTypeForUri (string uri)
                {
-                       Identity ident = GetIdentityForUri (uri);
+                       ServerIdentity ident = GetIdentityForUri (uri) as ServerIdentity;
                        if (ident == null) return null;
                        return ident.ObjectType;
                }
@@ -146,8 +166,6 @@ namespace System.Runtime.Remoting
 
                public static object Unmarshal (ObjRef objref, bool fRefine)
                {
-                       // FIXME: use type name when fRefine==true
-
                        Type classToProxy = fRefine ? objref.ServerType : typeof (MarshalByRefObject);
                        if (classToProxy == null) classToProxy = typeof (MarshalByRefObject);
 
@@ -185,7 +203,7 @@ namespace System.Runtime.Remoting
 
                                if (identity != null)
                                {
-                                       if (identity.ObjectType.IsContextful && !identity.IsConnected)
+                                       if (proxy.GetProxiedType().IsContextful && !identity.IsConnected)
                                        {
                                                // Unregistered local contextbound object. Register now.
                                                ClientActivatedIdentity cboundIdentity = (ClientActivatedIdentity)identity;
@@ -196,7 +214,7 @@ namespace System.Runtime.Remoting
                                                return cboundIdentity.CreateObjRef(requested_type);
                                        }
                                        else if (uri != null)
-                                               throw new RemotingException ("It is not possible marshal a proxy of a remote object");
+                                               throw new RemotingException ("It is not possible marshal a proxy of a remote object.");
 
                                        return proxy.ObjectIdentity.CreateObjRef(requested_type);
                                }
@@ -206,27 +224,34 @@ namespace System.Runtime.Remoting
 
                        if (uri == null) 
                        {
-                               uri = NewUri();
-                               CreateClientActivatedServerIdentity (obj, requested_type, uri);
+                               if (obj.ObjectIdentity == null)
+                               {
+                                       uri = NewUri();
+                                       CreateClientActivatedServerIdentity (obj, requested_type, uri);
+                               }
                        }
                        else
                        {
-                               ClientActivatedIdentity identity = uri_hash [uri] as ClientActivatedIdentity;
+                               ClientActivatedIdentity identity = GetIdentityForUri ("/" + uri) as ClientActivatedIdentity;
                                if (identity == null || obj != identity.GetServerObject()) 
                                        CreateClientActivatedServerIdentity (obj, requested_type, uri);
                        }
 
-                       return obj.CreateObjRef(requested_type);
+                       if (IsTransparentProxy (obj))
+                               return RemotingServices.GetRealProxy(obj).ObjectIdentity.CreateObjRef (requested_type);
+                       else
+                               return obj.CreateObjRef(requested_type);
                }
 
                static string NewUri ()
                {
-                       return app_id + Environment.TickCount + "_" + next_id++;
+                       int n = Interlocked.Increment (ref next_id);
+                       return app_id + Environment.TickCount + "_" + n + ".rem";
                }
 
                public static RealProxy GetRealProxy (object proxy)
                {
-                       if (!IsTransparentProxy(proxy)) throw new RemotingException("Cannot get the real proxy from an object that is not a transparent proxy");
+                       if (!IsTransparentProxy(proxy)) throw new RemotingException("Cannot get the real proxy from an object that is not a transparent proxy.");
                        return (RealProxy)((TransparentProxy)proxy)._rp;
                }
 
@@ -234,7 +259,7 @@ namespace System.Runtime.Remoting
                {
                        Type type = Type.GetType (msg.TypeName);
                        if (type == null)
-                               throw new RemotingException ("Type '" + msg.TypeName + "' not found!");
+                               throw new RemotingException ("Type '" + msg.TypeName + "' not found.");
 
                        BindingFlags bflags = BindingFlags.Instance | BindingFlags.Public | BindingFlags.NonPublic;
                        if (msg.MethodSignature == null)
@@ -258,14 +283,34 @@ namespace System.Runtime.Remoting
                        else return ident.CreateObjRef(null);
                }
 
-               [MonoTODO]
+               public static object GetLifetimeService (MarshalByRefObject obj)
+               {
+                       if (obj == null) return null;
+                       return obj.GetLifetimeService ();
+               }
+
+               public static IMessageSink GetEnvoyChainForProxy (MarshalByRefObject obj)
+               {
+                       if (IsTransparentProxy(obj))
+                               return ((ClientIdentity)GetRealProxy (obj).ObjectIdentity).EnvoySink;
+                       else
+                               throw new ArgumentException ("obj must be a proxy.","obj");                     
+               }
+
+               public static void LogRemotingStage (int stage)
+               {
+                       throw new NotImplementedException ();
+               }
+
                public static string GetSessionIdForMethodMessage(IMethodMessage msg)
                {
-                       throw new NotImplementedException (); 
+                       // It seems that this it what MS returns.
+                       return msg.Uri;
                }
 
                public static bool IsMethodOverloaded(IMethodMessage msg)
                {
+                       // TODO: use internal call for better performance
                        Type type = msg.MethodBase.DeclaringType;
                        MemberInfo[] members = type.GetMember (msg.MethodName, MemberTypes.Method, BindingFlags.NonPublic | BindingFlags.Public | BindingFlags.Instance);
                        return members.Length > 1;
@@ -273,6 +318,7 @@ namespace System.Runtime.Remoting
 
                public static bool IsObjectOutOfAppDomain(object tp)
                {
+                       // TODO: use internal call for better performance
                        Identity ident = GetObjectIdentity((MarshalByRefObject)tp);
                        if (ident != null) return !ident.IsFromThisAppDomain;
                        else return false;
@@ -280,6 +326,7 @@ namespace System.Runtime.Remoting
 
                public static bool IsObjectOutOfContext(object tp)
                {
+                       // TODO: use internal call for better performance
                        ServerIdentity ident = GetObjectIdentity((MarshalByRefObject)tp) as ServerIdentity;
                        if (ident != null) return ident.Context != System.Threading.Thread.CurrentContext;
                        else return false;
@@ -292,7 +339,7 @@ namespace System.Runtime.Remoting
                        return atts.Length > 0;
                }
 
-               public static bool IsAsyncMessage(IMessage msg)
+               internal static bool IsAsyncMessage(IMessage msg)
                {
                        if (! (msg is MonoMethodMessage)) return false;
                        else if (((MonoMethodMessage)msg).IsAsync) return true;
@@ -354,7 +401,7 @@ namespace System.Runtime.Remoting
                {
                        lock (uri_hash)
                        {
-                               return (Identity)uri_hash [uri];
+                               return (Identity)uri_hash [GetNormalizedUri(uri)];
                        }
                }
 
@@ -366,7 +413,7 @@ namespace System.Runtime.Remoting
                                return obj.ObjectIdentity;
                }
 
-               internal static ClientIdentity GetOrCreateClientIdentity(ObjRef objRef, Type proxyType)
+               internal static ClientIdentity GetOrCreateClientIdentity(ObjRef objRef, Type proxyType, out object clientProxy)
                {
                        // This method looks for an identity for the given url. 
                        // If an identity is not found, it creates the identity and 
@@ -385,23 +432,36 @@ namespace System.Runtime.Remoting
 
                        lock (uri_hash)
                        {
-                               ClientIdentity identity = uri_hash [objRef.URI] as ClientIdentity;
-                               if (identity != null) 
-                                       return identity;        // Object already registered
+                               clientProxy = null;
+                               string uri = GetNormalizedUri (objRef.URI);
+                               
+                               ClientIdentity identity = uri_hash [uri] as ClientIdentity;
+                               if (identity != null)
+                               {
+                                       // Object already registered
+                                       clientProxy = identity.ClientProxy;
+                                       if (clientProxy != null) return identity;
+                                       
+                                       // The proxy has just been GCed, so its identity cannot
+                                       // be reused. Just dispose it.
+                                       DisposeIdentity (identity);
+                               }
 
                                // Creates an identity and a proxy for the remote object
 
                                identity = new ClientIdentity (objectUri, objRef);
                                identity.ChannelSink = sink;
 
+                               // Registers the identity
+                               uri_hash [uri] = identity;
+                               
                                if (proxyType != null)
                                {
                                        RemotingProxy proxy = new RemotingProxy (proxyType, identity);
-                                       identity.ClientProxy = (MarshalByRefObject) proxy.GetTransparentProxy();
+                                       clientProxy = proxy.GetTransparentProxy();
+                                       identity.ClientProxy = (MarshalByRefObject) clientProxy;
                                }
 
-                               // Registers the identity
-                               uri_hash [objRef.URI] = identity;
                                return identity;
                        }
                }
@@ -459,7 +519,7 @@ namespace System.Runtime.Remoting
                        lock (uri_hash)
                        {
                                if (uri_hash.ContainsKey (identity.ObjectUri)) 
-                                       throw new RemotingException ("Uri already in use: " + identity.ObjectUri);
+                                       throw new RemotingException ("Uri already in use: " + identity.ObjectUri + ".");
 
                                uri_hash[identity.ObjectUri] = identity;
                        }
@@ -467,15 +527,16 @@ namespace System.Runtime.Remoting
 
                internal static object GetProxyForRemoteObject (ObjRef objref, Type classToProxy)
                {
-                       ClientActivatedIdentity identity = uri_hash [objref.URI] as ClientActivatedIdentity;
+                       ClientActivatedIdentity identity = GetIdentityForUri (objref.URI) as ClientActivatedIdentity;
                        if (identity != null) return identity.GetServerObject ();
                        else return GetRemoteObject (objref, classToProxy);
                }
 
                internal static object GetRemoteObject(ObjRef objRef, Type proxyType)
                {
-                       ClientIdentity id = GetOrCreateClientIdentity (objRef, proxyType);
-                       return id.ClientProxy;
+                       object proxy;
+                       GetOrCreateClientIdentity (objRef, proxyType, out proxy);
+                       return proxy;
                }
 
                internal static object GetDomainProxy(AppDomain domain) 
@@ -483,16 +544,15 @@ namespace System.Runtime.Remoting
                        byte[] data = null;
 
                        Context currentContext = Thread.CurrentContext;
-                       AppDomain currentDomain = AppDomain.InternalSetDomain (domain);
-                       try 
+
+                       try
                        {
-                               data = domain.GetMarshalledDomainObjRef ();
+                               data = (byte[])AppDomain.InvokeInDomain (domain, typeof (AppDomain).GetMethod ("GetMarshalledDomainObjRef", BindingFlags.Instance|BindingFlags.NonPublic), domain, null);
                        }
-                       finally 
+                       finally
                        {
-                               AppDomain.InternalSetDomain (currentDomain);
                                AppDomain.InternalSetContext (currentContext);
-                       }
+                       }                               
 
                        MemoryStream stream = new MemoryStream (data);
                        ObjRef appref = (ObjRef) CADSerializer.DeserializeObject (stream);
@@ -504,9 +564,20 @@ namespace System.Runtime.Remoting
                        CrossAppDomainChannel.RegisterCrossAppDomainChannel();
                }
                
-               internal static void DisposeIdentity (ServerIdentity ident)
+               internal static void DisposeIdentity (Identity ident)
                {
-                       uri_hash.Remove (ident.ObjectUri);
+                       lock (uri_hash)
+                       {
+                               if (!ident.Disposed) {
+                                       ClientIdentity clientId = ident as ClientIdentity;
+                                       if (clientId != null)
+                                               uri_hash.Remove (GetNormalizedUri (clientId.TargetUri));
+                                       else
+                                               uri_hash.Remove (ident.ObjectUri);
+                                               
+                                       ident.Disposed = true;
+                               }
+                       }
                }
 
                internal static Identity GetMessageTargetIdentity (IMessage msg)
@@ -518,7 +589,8 @@ namespace System.Runtime.Remoting
 
                        lock (uri_hash)
                        {
-                               return uri_hash [((IMethodMessage)msg).Uri] as ServerIdentity;
+                               string uri = GetNormalizedUri (((IMethodMessage)msg).Uri);
+                               return uri_hash [uri] as ServerIdentity;
                        }
                }
 
@@ -527,6 +599,37 @@ namespace System.Runtime.Remoting
                        if (msg is IInternalMessage) 
                                ((IInternalMessage)msg).TargetIdentity = ident;
                }
+               
+               internal static bool UpdateOutArgObject (ParameterInfo pi, object local, object remote)
+               {
+                       if (local is StringBuilder) 
+                       {
+                               StringBuilder sb = local as StringBuilder;
+                               sb.Remove (0, sb.Length);
+                               sb.Append (remote.ToString());
+                               return true;
+                       }
+                       else if (pi.ParameterType.IsArray && ((Array)local).Rank == 1)
+                       {
+                               Array alocal = (Array) local;
+                               if (alocal.Rank == 1)
+                               {
+                                       Array.Copy ((Array) remote, alocal, alocal.Length);
+                                       return true;
+                               }
+                               else
+                               {
+                                       // TODO
+                               }
+                       }
+                       return false;
+               }
+               
+               static string GetNormalizedUri (string uri)
+               {
+                       if (uri.StartsWith ("/")) return uri.Substring (1);
+                       else return uri;
+               }
 
                #endregion
        }