mcs/class/corlib/System.Runtime.Remoting/ServerIdentity.cs

   1 //
   2 // System.Runtime.Remoting.Identity.cs
   3 //
   4 // Author: Lluis Sanchez Gual (lluis@ideary.com)
   5 //
   6 // (C) 2002, Lluis Sanchez Gual
   7 //
   8
   9 //
  10 // Copyright (C) 2004 Novell, Inc (http://www.novell.com)
  11 //
  12 // Permission is hereby granted, free of charge, to any person obtaining
  13 // a copy of this software and associated documentation files (the
  14 // "Software"), to deal in the Software without restriction, including
  15 // without limitation the rights to use, copy, modify, merge, publish,
  16 // distribute, sublicense, and/or sell copies of the Software, and to
  17 // permit persons to whom the Software is furnished to do so, subject to
  18 // the following conditions:
  19 //
  20 // The above copyright notice and this permission notice shall be
  21 // included in all copies or substantial portions of the Software.
  22 //
  23 // THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
  24 // EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
  25 // MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
  26 // NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
  27 // LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
  28 // OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
  29 // WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
  30 //
  31
  32 using System;
  33 using System.Runtime.Remoting.Messaging;
  34 using System.Runtime.Remoting.Contexts;
  35 using System.Runtime.Remoting.Lifetime;
  36 using System.Runtime.Remoting.Proxies;
  37
  38 namespace System.Runtime.Remoting
  39 {
  40         internal abstract class ServerIdentity : Identity
  41         {
  42                 protected Type _objectType;
  43
  44                 protected MarshalByRefObject _serverObject;
  45
  46                 // Message sink used in the server to dispatch a message
  47                 // to the server object
  48                 protected IMessageSink _serverSink = null;
  49
  50                 protected Context _context;
  51                 protected Lease _lease;
  52
  53                 public ServerIdentity (string objectUri, Context context, Type objectType): base (objectUri)
  54                 {
  55                         _objectType = objectType;
  56                         _context = context;
  57                 }
  58
  59                 public Type ObjectType
  60                 {
  61                         get { return _objectType; }
  62                 }
  63
  64                 public void StartTrackingLifetime (ILease lease)
  65                 {
  66                         // Adds this identity to the LeaseManager.
  67                         // _serverObject must be set.
  68
  69                         if (lease != null && lease.CurrentState == LeaseState.Null) lease = null;
  70
  71                         if (lease != null)
  72                         {
  73                                 if (! (lease is Lease)) lease = new Lease();  // This seems to be MS behavior
  74                                 _lease = (Lease) lease;
  75                                 LifetimeServices.TrackLifetime (this);
  76                         }
  77                 }
  78
  79                 public virtual void OnLifetimeExpired()
  80                 {
  81                         DisposeServerObject();
  82                 }
  83
  84                 public override ObjRef CreateObjRef (Type requestedType)
  85                 {
  86                         if (_objRef != null)
  87                         {
  88                                 // Just update channel info. It may have changed.
  89                                 _objRef.UpdateChannelInfo();
  90                                 return _objRef;
  91                         }
  92
  93                         if (requestedType == null) requestedType = _objectType;
  94                         _objRef = new ObjRef ();
  95                         _objRef.TypeInfo = new TypeInfo(requestedType);
  96                         _objRef.URI = _objectUri;
  97
  98                         if (_envoySink != null && !(_envoySink is EnvoyTerminatorSink))
  99                                 _objRef.EnvoyInfo = new EnvoyInfo (_envoySink);
 100
 101                         return _objRef;
 102                 }
 103
 104                 public void AttachServerObject (MarshalByRefObject serverObject, Context context)
 105                 {
 106                         _context = context;
 107                         _serverObject = serverObject;
 108
 109                         if (RemotingServices.IsTransparentProxy (serverObject))
 110                         {
 111                                 RealProxy rp = RemotingServices.GetRealProxy (serverObject);
 112                                 if (rp.ObjectIdentity == null)
 113                                         rp.ObjectIdentity = this;
 114                         }
 115                         else
 116                         {
 117                                 if (_objectType.IsContextful)
 118                                         _envoySink = context.CreateEnvoySink (serverObject);
 119
 120                                 _serverObject.ObjectIdentity = this;
 121                         }
 122                 }
 123
 124                 public Lease Lease
 125                 {
 126                         get { return _lease; }
 127                 }
 128
 129                 public Context Context
 130                 {
 131                         get { return _context; }
 132                         set { _context = value; }
 133                 }
 134
 135                 public abstract IMessage SyncObjectProcessMessage (IMessage msg);
 136                 public abstract IMessageCtrl AsyncObjectProcessMessage (IMessage msg, IMessageSink replySink);
 137
 138                 protected void DisposeServerObject()
 139                 {
 140                         // Detach identity from server object to avoid problems if the
 141                         // object is marshalled again.
 142
 143                         if (_serverObject != null) {
 144                                 _serverObject.ObjectIdentity = null;
 145                                 _serverObject = null;
 146                         }
 147                 }
 148         }
 149
 150         internal class ClientActivatedIdentity : ServerIdentity
 151         {
 152                 MarshalByRefObject _targetThis;
 153
 154                 public ClientActivatedIdentity (string objectUri, Type objectType): base (objectUri, null, objectType)
 155                 {
 156                 }
 157
 158                 public MarshalByRefObject GetServerObject ()
 159                 {
 160                         return _serverObject;
 161                 }
 162
 163                 public MarshalByRefObject GetClientProxy ()
 164                 {
 165                         return _targetThis;
 166                 }
 167
 168                 public void SetClientProxy (MarshalByRefObject obj)
 169                 {
 170                         _targetThis = obj;
 171                 }
 172
 173                 public override void OnLifetimeExpired()
 174                 {
 175                         base.OnLifetimeExpired();
 176                         RemotingServices.DisposeIdentity (this);
 177                 }
 178
 179                 public override IMessage SyncObjectProcessMessage (IMessage msg)
 180                 {
 181                         if (_serverSink == null) {
 182                                 bool useProxy = _targetThis != null;
 183                                 _serverSink = _context.CreateServerObjectSinkChain ((useProxy ? _targetThis : _serverObject), useProxy);
 184                         }
 185                         return _serverSink.SyncProcessMessage (msg);
 186                 }
 187
 188                 public override IMessageCtrl AsyncObjectProcessMessage (IMessage msg, IMessageSink replySink)
 189                 {
 190                         if (_serverSink == null) {
 191                                 bool useProxy = _targetThis != null;
 192                                 _serverSink = _context.CreateServerObjectSinkChain ((useProxy ? _targetThis : _serverObject), useProxy);
 193                         }
 194                         return _serverSink.AsyncProcessMessage (msg, replySink);
 195                 }
 196         }
 197
 198         internal class SingletonIdentity : ServerIdentity
 199         {
 200                 public SingletonIdentity (string objectUri, Context context, Type objectType): base (objectUri, context, objectType)
 201                 {
 202                 }
 203
 204                 public MarshalByRefObject GetServerObject ()
 205                 {
 206                         if (_serverObject != null) return _serverObject;
 207
 208                         lock (this)
 209                         {
 210                                 if (_serverObject == null) {
 211                                         MarshalByRefObject server = (MarshalByRefObject) Activator.CreateInstance (_objectType, true);
 212                                         AttachServerObject (server, Context.DefaultContext);
 213                                         StartTrackingLifetime ((ILease)server.InitializeLifetimeService ());
 214                                 }
 215                         }
 216                         return _serverObject;
 217                 }
 218
 219                 public override IMessage SyncObjectProcessMessage (IMessage msg)
 220                 {
 221                         MarshalByRefObject obj = GetServerObject ();
 222                         if (_serverSink == null) _serverSink = _context.CreateServerObjectSinkChain (obj, false);
 223                         return _serverSink.SyncProcessMessage (msg);
 224                 }
 225
 226                 public override IMessageCtrl AsyncObjectProcessMessage (IMessage msg, IMessageSink replySink)
 227                 {
 228                         MarshalByRefObject obj = GetServerObject ();
 229                         if (_serverSink == null) _serverSink = _context.CreateServerObjectSinkChain (obj, false);
 230                         return _serverSink.AsyncProcessMessage (msg, replySink);
 231                 }
 232         }
 233
 234         internal class SingleCallIdentity : ServerIdentity
 235         {
 236                 public SingleCallIdentity (string objectUri, Context context, Type objectType): base (objectUri, context, objectType)
 237                 {
 238                 }
 239
 240                 public override IMessage SyncObjectProcessMessage (IMessage msg)
 241                 {
 242                         // SingleCallIdentity creates and disposes an instance in each call
 243
 244                         MarshalByRefObject obj = (MarshalByRefObject)Activator.CreateInstance (_objectType, true);
 245                         if (obj.ObjectIdentity == null) obj.ObjectIdentity = this;
 246                         IMessageSink serverSink = _context.CreateServerObjectSinkChain (obj, false);
 247                         IMessage result = serverSink.SyncProcessMessage (msg);
 248                         if (obj is IDisposable) ((IDisposable)obj).Dispose();
 249                         return result;
 250                 }
 251
 252                 public override IMessageCtrl AsyncObjectProcessMessage (IMessage msg, IMessageSink replySink)
 253                 {
 254                         MarshalByRefObject obj = (MarshalByRefObject)Activator.CreateInstance (_objectType, true);
 255                         IMessageSink serverSink = _context.CreateServerObjectSinkChain (obj, false);
 256                         if (obj is IDisposable) replySink = new DisposerReplySink(replySink, ((IDisposable)obj));
 257                         return serverSink.AsyncProcessMessage (msg, replySink);
 258                 }
 259         }
 260
 261         internal class DisposerReplySink : IMessageSink
 262         {
 263                 IMessageSink _next;
 264                 IDisposable _disposable;
 265
 266                 public DisposerReplySink (IMessageSink next, IDisposable disposable)
 267                 {
 268                         _next = next;
 269                         _disposable = disposable;
 270                 }
 271
 272                 public IMessage SyncProcessMessage (IMessage msg)
 273                 {
 274                         _disposable.Dispose();
 275                         return _next.SyncProcessMessage (msg);
 276                 }
 277
 278                 public IMessageCtrl AsyncProcessMessage (IMessage msg, IMessageSink replySink)
 279                 {
 280                         throw new NotSupportedException();
 281                 }
 282
 283                 public IMessageSink NextSink
 284                 {
 285                         get { return _next; }
 286                 }
 287         }
 288 }