WCF方法拦截及OperationInvoker传递参数到WCF方法的实现
鉴于一些理由需要拦截WCF方法,比如参数的检测。一个实际的场景是CSRF防范中需要对CsrfToken的检测。
要实现拦截,在形式上要实现两点:
1:为WCF方法增加特性类;
2:捕获WCF方法;
第一点很好实现,实现一个Attribute类。第二点则可以利用WCF框架中的IOperationInvoker接口,让该Attribute类同时也实现该接口,同时在该接口方法的ApplyDispatchBehavior方法中,为dispatchOperation实现一个新的OperationInvoker。
public class MyOperationInterceptorAttribute : Attribute, IOperationBehavior { public void AddBindingParameters(OperationDescription operationDescription, BindingParameterCollection bindingParameters) { } public void ApplyClientBehavior(OperationDescription operationDescription, ClientOperation clientOperation) { } public void ApplyDispatchBehavior(OperationDescription operationDescription, DispatchOperation dispatchOperation) { dispatchOperation.Invoker = new } public void Validate(OperationDescription operationDescription) { } } public class MyInvoker : IOperationInvoker { IOperationInvoker m_OldInvoker; public object Invoke(object instance, object[] inputs, out object[] outputs) { try { //do something object result = m_OldInvoker.Invoke(instance, inputs, out outputParams); return result; } catch (Exception err) { outputs = new object[0]; //do catch return null; } finally { } } public virtual object[] AllocateInputs() { return m_OldInvoker.AllocateInputs(); } public IAsyncResult InvokeBegin(object instance, object[] inputs, AsyncCallback callback, object state) { return m_OldInvoker.InvokeBegin(instance, inputs, callback, state); } public object InvokeEnd(object instance, out object[] outputs, IAsyncResult result) { return m_OldInvoker.InvokeEnd(instance, out outputs, result); } public bool IsSynchronous { get { return m_OldInvoker.IsSynchronous; } } }
如果我们将服务宿主在IIS中,则这种实现的好处是我们可以完全不需要更改配置文件就能实现对WCF方法的拦截。
这种方式还有一种好处是,我们可以解析客户端发送过来的内容,并统一进行处理,而无需为WCF方法增加参数。如,我们对CSRF方法的时候,客户端发送过来的内容中会含有CSRFToken,传统的做法是为WCF方法一个CSRFToken的参数,如下:
public void DoSomething(string arg0, string csrfToken) { Check(csrfToken); ... }
并且每个需要处理CSRF的方法中都要进行如上编码,而现在只要变成:
[MyOperationInterceptor] public void DoSomething(string arg0) { ... }
可以看到少了参数,少了方法内的处理,当然,我们需要在MyInvoker类的Invoke方法的//do something处解析前台内容,获取CSRFToken,并进行Check,获取CSRFToken如下实现:
string GetToken() { //rest or soap string token = HttpContext.Current.Request.QueryString["token"]; if(string.IsNullOrEmpty(token)) { XDocument doc = XDocument.Parse(OperationContext.Current.RequestContext.RequestMessage.ToString()); var rootElements = (from p in doc.Descendants("root") select p).Elements(); var args = (from p in rootElements select new {Name=p.Name.LocalName, Value=p.Value}).ToList(0; token = (from p in args where p.Name="token" select p.Value).First(); } return token; }
在这种拦截的应用中,有时候我们难免要将拦截过程中的一些内容传递到WCF方法中,如果有这样的需求,则可以将这些内容放置到OperationContet.Current.IncomingMessageProperties中,它是一个IDictionary<string, object>,当然,要对这个参数进行存取的时候,我们需要用到OperationContextScope,于是,整个MyInvoker类的Invoke方法变成如下的形式:
try { using(OperationContextScope scope = new OperationContextScope()) { //get token and check OperationContet.Current.IncomingMessageProperties.Add("SomeKey", SomeObjectInstance); //do something object result = m_OldInvoker.Invoke(instance, inputs, out outputParams); return result; } } catch (Exception err) { outputs = new object[0]; //do catch return null; } finally { }
在具体的WCF方法中,如果要使用这个SomeObjectInstance,则可以这样获得:
SomeObjectInstance = OperationContet.Current.IncomingMessageProperties["SomeKeyas TheInstanceType;