class/System.Runtime.Remoting/Test/ContextHookAttribute.cs

   1 //
   2 // MonoTests.Remoting.ContextHookAttribute.cs
   3 //
   4 // Author: Lluis Sanchez Gual (lluis@ximian.com)
   5 //
   6 // 2003 (C) Copyright, Ximian, Inc.
   7 //
   8
   9 using System;
  10 using System.Runtime.Remoting.Contexts;
  11 using System.Runtime.Remoting.Activation;
  12 using System.Runtime.Remoting.Messaging;
  13 using System.Runtime.Remoting.Proxies;
  14 using System.Runtime.Remoting;
  15 using System.Threading;
  16
  17 namespace MonoTests.Remoting
  18 {
  19         [Serializable,AttributeUsage(AttributeTargets.Class)]
  20         public class ContextHookAttribute: Attribute, IContextAttribute, IContextProperty, IContributeObjectSink, IContributeServerContextSink, IContributeEnvoySink, IContributeClientContextSink
  21         {
  22                 bool newContext = false;
  23
  24                 string id = "";
  25                 public ContextHookAttribute()
  26                 {
  27                 }
  28
  29                 public ContextHookAttribute(string idp, bool newContext)
  30                 {
  31                         id = idp;
  32                         if (id != "") id += ".";
  33                         id += "d" + CallSeq.CommonDomainId;
  34                         this.newContext = newContext;
  35                 }
  36
  37                 public override object TypeId
  38                 {
  39
  40                         get { return "ContextHook"; }
  41                 }
  42
  43                 bool IContextAttribute.IsContextOK(Context ctx, IConstructionCallMessage ctor)
  44                 {
  45                         CallSeq.Add("ContextHookAttribute(" + id + ").IsContextOK");
  46                         return !newContext;
  47                 }
  48
  49                 public bool IsNewContextOK(Context ctx)
  50                 {
  51                         CallSeq.Add("ContextHookAttribute(" + id + ").IsNewContextOK");
  52                         return true;
  53                 }
  54
  55                 public void Freeze(Context ctx)
  56                 {
  57                         CallSeq.Add("ContextHookAttribute(" + id + ").Freeze");
  58                 }
  59
  60                 public String Name
  61                 {
  62                         get { return "ContextHook(" + id + ")"; }
  63                 }
  64
  65
  66                 void IContextAttribute.GetPropertiesForNewContext(IConstructionCallMessage ctor)
  67                 {
  68                         CallSeq.Add("IContextAttribute(" + id + ").GetPropertiesForNewContext");
  69                         ctor.ContextProperties.Add(this);
  70                 }
  71
  72                 IMessageSink IContributeObjectSink.GetObjectSink(MarshalByRefObject o, IMessageSink next)
  73                 {
  74                         CallSeq.Add("IContributeObjectSink(" + id + ").GetObjectSink");
  75                         return new GenericMessageSink(o,next,"ObjectSink(" + id + ")");
  76                 }
  77
  78                 IMessageSink IContributeServerContextSink.GetServerContextSink(IMessageSink next)
  79                 {
  80                         CallSeq.Add("IContributeServerContextSink(" + id + ").GetServerContextSink");
  81                         return new GenericMessageSink(null,next,"ServerContextSink(" + id + ")");
  82                 }
  83
  84                 IMessageSink IContributeEnvoySink.GetEnvoySink(MarshalByRefObject obj, IMessageSink nextSink)
  85                 {
  86                         CallSeq.Add("IContributeEnvoySink(" + id + ").GetEnvoySink");
  87                         return new GenericMessageSink(obj,nextSink,"EnvoySink(" + id + ")");
  88                 }
  89
  90                 IMessageSink IContributeClientContextSink.GetClientContextSink (IMessageSink nextSink )
  91                 {
  92                         CallSeq.Add("IContributeClientContextSink(" + id + ").GetClientContextSink");
  93                         return new GenericMessageSink(null,nextSink,"ClientContextSink(" + id + ")");
  94                 }
  95         }
  96
  97         [Serializable]
  98         class GenericMessageSink: IMessageSink
  99         {
 100                 IMessageSink _next;
 101                 string _type;
 102
 103                 public GenericMessageSink(MarshalByRefObject obj, IMessageSink nextSink, string type)
 104                 {
 105                         _type = type;
 106                         _next = nextSink;
 107                 }
 108
 109                 public IMessageSink NextSink
 110                 {
 111                         get { return _next; }
 112                 }
 113
 114                 public IMessage SyncProcessMessage(IMessage imCall)
 115                 {
 116                         CallSeq.Add("--> " + _type + " SyncProcessMessage " + imCall.Properties["__MethodName"]);
 117                         IMessage ret = _next.SyncProcessMessage(imCall);
 118                         CallSeq.Add("<-- " + _type + " SyncProcessMessage " + imCall.Properties["__MethodName"]);
 119                         return ret;
 120                 }
 121
 122                 public IMessageCtrl AsyncProcessMessage(IMessage im, IMessageSink ims)
 123                 {
 124                         CallSeq.Add("--> " + _type + " AsyncProcessMessage " + im.Properties["__MethodName"]);
 125                         IMessageCtrl ret = _next.AsyncProcessMessage(im, ims);
 126                         CallSeq.Add("<-- " + _type + " AsyncProcessMessage " + im.Properties["__MethodName"]);
 127                         return ret;
 128                 }
 129         }
 130
 131         [Serializable]
 132         class GenericDynamicSink: IDynamicMessageSink
 133         {
 134                 string _name;
 135
 136                 public GenericDynamicSink (string name)
 137                 {
 138                         _name = name;
 139                 }
 140
 141                 void IDynamicMessageSink.ProcessMessageFinish(IMessage replyMsg, bool bCliSide, bool bAsync)
 142                 {
 143                         CallSeq.Add("<-> " + _name + " DynamicSink Finish " + replyMsg.Properties["__MethodName"] + " client:" + bCliSide);
 144                 }
 145
 146                 void IDynamicMessageSink.ProcessMessageStart(IMessage replyMsg, bool bCliSide, bool bAsync)
 147                 {
 148                         CallSeq.Add("<-> " + _name + " DynamicSink Start " + replyMsg.Properties["__MethodName"] + " client:" + bCliSide);
 149                 }
 150         }
 151
 152         public class DynProperty: IDynamicProperty, IContributeDynamicSink
 153         {
 154                 string _name;
 155                 public DynProperty (string name)
 156                 {
 157                         _name = name;
 158                 }
 159
 160                 public string Name
 161                 {
 162                         get { return _name; }
 163                 }
 164
 165                 public IDynamicMessageSink GetDynamicSink()
 166                 {
 167                         CallSeq.Add("IContributeDynamicSink(" + _name + ").GetDynamicSink");
 168                         return new GenericDynamicSink(_name);
 169                 }
 170         }
 171 }