WCF笔记(6)消息拦截与篡改

要对SOAP消息进行拦截和修改，我们需要实现两个接口，它们都位于System.ServiceModel.Dispatcher （程序集System.ServiceModel）。下面分别价绍。

接口一：IClientMessageInspector

从名字中我们可以猜测，它是用来拦截客户消息的，而看看它的方法，你就更加肯定当初的猜测了。

• BeforeSendRequest：向服务器发送请求前拦截或修改消息（事前控制）
• AfterReceiveReply：接收到服务器的回复消息后，在调用返回之前拦截或修改消息（事后诸葛亮）

接口二：IDispatchMessageInspector

刚才那个接口是针对客户端的，而这个是针对服务器端的。

• AfterReceiveRequest：接收客户端请求后，在进入操作处理代码之前拦截或修改消息（欺上）
• BeforeSendReply：服务器向客户端发送回复消息之前拦截和修改消息（瞒下）。

虽然实现了这两个接口，但你会有新的疑问，怎么用？把它们放到哪儿才能拦截消息？因此，下一步就是要实现IEndpointBehavior按口（System.ServiceModel.Description命名空间，程序集System.ServiceModel），它有四个方法，而我们只需要处理两个就够了。

下面是MSDN的翻译版本说明：

使用 ApplyClientBehavior 方法可以在客户端应用程序中修改、检查或插入对终结点中的扩展。

使用 ApplyDispatchBehavior 方法可以在服务应用程序中修改、检查或插入对终结点范围执行的扩展。

我想不用额外解释了，说白了就是一个在客户拦截和修改消息，另一个在服务器端拦截和修改消息。

在实现这两个方法时，和前面我们实现的IClientMessageInspector和IDispatchMessageInspector联系起来就OK了。

做完了IEndpointBehavior的事情后，把它插入到服务终结点中就行了，无论是服务器端还是客户端，这一步都必须的，因为我们实现的拦截器是包括两个端的，因此，较好的做法是把这些类写到一个独立的类库（dll）中，这样一来，服务器端和客户端都可以引用它。

示例代码

1、实现消息拦截器

using System;using System.ServiceModel.Dispatcher;using System.ServiceModel.Description;namespace MyMessageIntercept{    /// <summary>    /// 消息拦截器【实现客户端和服务端的消息拦截】    /// </summary>    public class MessageIntercept:IClientMessageInspector,IDispatchMessageInspector     {        public void AfterReceiveReply(ref System.ServiceModel.Channels.Message reply, object correlationState)        {            Console.WriteLine("客户端接收到的回复:{0}\n", reply);        }        public object BeforeSendRequest(ref System.ServiceModel.Channels.Message request, System.ServiceModel.IClientChannel channel)        {            Console.WriteLine("客户端发送请求前的SOAP消息:{0}\n", request);            return null;        }        public object AfterReceiveRequest(ref System.ServiceModel.Channels.Message request, System.ServiceModel.IClientChannel channel, System.ServiceModel.InstanceContext instanceContext)        {            Console.WriteLine("服务器端接收到的请求:{0}\n", request);            return null;        }        public void BeforeSendReply(ref System.ServiceModel.Channels.Message reply, object correlationState)        {            Console.WriteLine("服务器将作出以下回复:{0}\n", reply);        }    }    /// <summary>    /// 将自定义的消息拦截器(MessageIntercept)插入到终结点行为    /// </summary>    public class MyEndPointBehavior : IEndpointBehavior    {           public void AddBindingParameters(ServiceEndpoint endpoint, System.ServiceModel.Channels.BindingParameterCollection bindingParameters)        {            //不需要        }        public void ApplyClientBehavior(ServiceEndpoint endpoint, ClientRuntime clientRuntime)        {           //植入"窃听器"客户端            clientRuntime.ClientMessageInspectors.Add(new MessageIntercept());        }        public void ApplyDispatchBehavior(ServiceEndpoint endpoint, EndpointDispatcher endpointDispatcher)        {            //植入"窃听器"服务端            endpointDispatcher.DispatchRuntime.MessageInspectors.Add(new MessageIntercept());        }        public void Validate(ServiceEndpoint endpoint)        {            //不需要        }    }}

2、实现服务端

using System;using System.ServiceModel;using System.ServiceModel.Description;using System.Runtime.Serialization;using MyMessageIntercept;namespace Server{    class Program    {        //设置服务并启动        static void Main()        {            //服务基地址            Uri baseUri = new Uri("http://localhost:3000/Service");            //声明服务器主机            using (ServiceHost host = new ServiceHost(typeof(MyService), baseUri))            {                //添加绑定和终结点                WSHttpBinding binding = new WSHttpBinding();                host.AddServiceEndpoint(typeof(IService), binding, "test");                //添加服务描述                host.Description.Behaviors.Add(new ServiceMetadataBehavior { HttpGetEnabled = true });                //把自定义的"MyEndPointBehavior"插入到终结点中                foreach (var endPoint in host.Description.Endpoints)                {                    endPoint.EndpointBehaviors.Add(new MyEndPointBehavior());                }                //启动服务                try                {                    host.Open();                    Console.WriteLine("服务已启动");                }                catch (Exception ex)                {                    Console.WriteLine(ex.Message);                }                Console.ReadKey();                //关闭服务                host.Close();            }        }    }    /// <summary>    /// 定义服务协定    /// </summary>    [ServiceContract]    public interface IService    {        [OperationContract]        int AddInt(int a, int b);        [OperationContract]        Student GetStudent();        [OperationContract]        CalculateResultResponse CaculateNumber(CalculateRequest request);    }    /// <summary>    /// 实现服务协定    /// </summary>    public class MyService : IService    {        public int AddInt(int a, int b)        {            return a + b;        }        public Student GetStudent()        {            Student st = new Student                {                    Name = "小明",                    Age = 23                };            return st;        }        public CalculateResultResponse CaculateNumber(CalculateRequest request)        {            CalculateResultResponse result = new CalculateResultResponse();            switch (request.Operation)            {                case "+":                    result.CalculateResult = request.NumberA + request.NumberB;                    break;                case "-":                    result.CalculateResult = request.NumberA - request.NumberB;                    break;                case "*":                      result.CalculateResult = request.NumberA * request.NumberB;                    break;                case "/":                    result.CalculateResult = request.NumberA / request.NumberB;                    break;                default:                    throw new ArgumentException("运算操作符只允许'+'、'-'、'*'、'/'.");            }            return result;        }    }    [DataContract]    public class Student    {        [DataMember]        public string Name { get; set; }        [DataMember]        public int Age { get; set; }    }    [MessageContract]    public class CalculateRequest    {        /// <summary>        /// 操作符        /// </summary>        [MessageHeader]        public string Operation { get; set; }        /// <summary>        /// 第一个数        /// </summary>        [MessageBodyMember]        public int NumberA { get; set; }        /// <summary>        /// 第二个数        /// </summary>        [MessageBodyMember ]        public int NumberB { get; set; }    }        /// <summary>    /// 计算返回的结果    /// </summary>    [MessageContract]    public class CalculateResultResponse    {        /// <summary>        /// 计算结果        /// </summary>        [MessageBodyMember]        public int CalculateResult { get; set; }    }}

3、实现客户端

using System;using MyMessageIntercept;namespace Client{    class Program    {        private static void Main()        {            WS.ServiceClient sc = new WS.ServiceClient();            //记得在客户端也要插入自定义的"MyEndPointBehavior"            sc.Endpoint.EndpointBehaviors.Add(new MyEndPointBehavior());            try            {                // 1、调用带元数据参数和返回值的操作                  Console.WriteLine("10和55相加的结果是:{0}\n", sc.AddInt(10, 55));                //2、调用带有数据协定的操作                WS.Student st = sc.GetStudent();                Console.WriteLine("-----学生信息--------\n姓名:{0}\n年龄:{1}\n", st.Name, st.Age);                //3、调用带有消息协定的操作                Console.WriteLine("12乘以56的结果是:{0}\n", sc.CaculateNumber("*", 12, 56));                sc.Close();            }            catch (Exception ex)            {                Console.WriteLine(ex.Message);            }            Console.ReadKey();        }    }}

消息拦截效果图

修改消息，只需要对消息拦截器(MessageIntercept)类做修改

using System;using System.ServiceModel.Dispatcher;using System.ServiceModel.Description;namespace MyMessageIntercept{    /// <summary>    /// 消息拦截器【实现客户端和服务端的消息拦截】    /// </summary>    public class MessageIntercept:IClientMessageInspector,IDispatchMessageInspector     {        public void AfterReceiveReply(ref System.ServiceModel.Channels.Message reply, object correlationState)        {            //Console.WriteLine("客户端接收到的回复：\n{0}", reply);          }        public object BeforeSendRequest(ref System.ServiceModel.Channels.Message request, System.ServiceModel.IClientChannel channel)        {            //Console.WriteLine("客户端发送请求前的SOAP消息：\n{0}", request);             // (客服端)插入验证信息              System.ServiceModel.Channels.MessageHeader hdUserName = System.ServiceModel.Channels.MessageHeader.CreateHeader("user", "myTest", "admin");            System.ServiceModel.Channels.MessageHeader hdPassWord = System.ServiceModel.Channels.MessageHeader.CreateHeader("pw", "myTest", "123");            request.Headers.Add(hdUserName);            request.Headers.Add(hdPassWord);            return null;          }        public object AfterReceiveRequest(ref System.ServiceModel.Channels.Message request, System.ServiceModel.IClientChannel channel, System.ServiceModel.InstanceContext instanceContext)        {            //Console.WriteLine("服务器端：接收到的请求：\n{0}", request);             // (服务端)栓查验证信息               string un = request.Headers.GetHeader<string>("user", "myTest");             string ps = request.Headers.GetHeader<string>("pw", "myTest");            if (un == "admin" && ps == "123")            {                Console.WriteLine("用户名和密码正确.");            }            else            {                throw new Exception("验证失败!");            }            return null;         }        public void BeforeSendReply(ref System.ServiceModel.Channels.Message reply, object correlationState)        {            //Console.WriteLine("服务器即将作出以下回复：\n{0}", reply);         }    }    /// <summary>    /// 将自定义的消息拦截器(MessageIntercept)插入到终结点行为    /// </summary>    public class MyEndPointBehavior : IEndpointBehavior    {           public void AddBindingParameters(ServiceEndpoint endpoint, System.ServiceModel.Channels.BindingParameterCollection bindingParameters)        {            //不需要        }        public void ApplyClientBehavior(ServiceEndpoint endpoint, ClientRuntime clientRuntime)        {           //植入"窃听器"客户端            clientRuntime.ClientMessageInspectors.Add(new MessageIntercept());        }        public void ApplyDispatchBehavior(ServiceEndpoint endpoint, EndpointDispatcher endpointDispatcher)        {            //植入"窃听器"服务端            endpointDispatcher.DispatchRuntime.MessageInspectors.Add(new MessageIntercept());        }        public void Validate(ServiceEndpoint endpoint)        {            //不需要        }    }}

注意：添加对System.Runtime.Serialization的引用。
前面我们说过，如果安装证书进行身份验证会相当麻烦，而可以通过修改SOAP消息头来验证，但是，上次的做法会有一个麻烦，那就是每次调用操作协定都要手动修改一次，这一次，我们直接在终结点级别进行修改和验证，就省去了许多功夫。