修补 Asp.net Mvc2 AntiForgeryToken 异常

Question

我的问题的一些背景：

Mvc2 中似乎有一个关于ValidateAntiForgeryTokenAttribute 的更改/错误。

从 Mvc1 升级到 Mvc2 时，具有活动会话的用户在使用 ValidateAntiForgeryTokenAttribute 请求页面时会收到以下错误：

无法将“System.Web.UI.Triplet”类型的对象转换为“System.Object[]”类型。

该问题已记录在 here。

升级到 Mvc2 后，我们预计我们将受到此问题的严重影响。我编写了一个从 cmets 中的代码派生的修复程序（为后代记录如下）。目前，通过创建 Controller 和 AsyncController 的子类来调用此代码，并覆盖 Initialize 方法以纠正问题。例如

public class FixedController:Controller
{
    protected override void Initialize(RequestContext requestContext)
    {
        base.Initialize(requestContext);
        this.FixAntiForgeryTokenMvc1ToMvc2(requestContext); //extension
    }
}

internal static class ControllerEx
{
    public static void FixAntiForgeryTokenMvc1ToMvc2(
        this Controller controller,
        RequestContext requestContext)
    {
        var cc = new ControllerContext(requestContext,
                                       controller);
        var antiForgeryAttribute = new ValidateAntiForgeryTokenAttribute();
        try
        {
            antiForgeryAttribute.OnAuthorization(new AuthorizationContext(cc));
        }
        catch (HttpAntiForgeryException forgeryException)
        {
            var castException = forgeryException.InnerException;
            if (castException != null
                && castException is InvalidCastException
                && castException.Message.StartsWith(
                       "Unable to cast object of type"
                       + " 'System.Web.UI.Triplet' to type"
                       + " 'System.Object[]'"))
            {
                var responseTokenCookieNames =
                    controller
                        .Response
                        .Cookies
                        .Cast<Cookie>()
                        .Select(c => c.Name)
                        .Where(n => n.Contains("RequestVerificationToken"));
                foreach (var s in responseTokenCookieNames)
                {
                    var cookie = controller.Response.Cookies[s];
                    if (cookie != null)
                    {
                        cookie.Value = "";
                    }
                }
                var requestTokenCookieNames =
                    controller
                        .Request
                        .Cookies
                        .Cast<String>()
                        .Where(n => n.Contains("RequestVerificationToken"))
                        .ToList();
                foreach (var c in requestTokenCookieNames)
                {
                    controller.Request.Cookies.Remove(c);
                }
            }
        }
    }
}

这样做的连锁反应是我必须更改我的所有控制器类以从我的新的、更正的控制器子类派生。对于打算在大约一个月内弃用的代码来说，这似乎很麻烦。

所以，关于我的问题，我想知道是否有一种侵入性较小的方法来修补现有类，从而不必更改该类的下游用户，也许使用反射？

Answer 1

花费者，

与其对每个控制器进行编码，不如使用基本控制器并按照以下方式继承：

public abstract class BaseController : Controller
{
    protected override void Initialize(RequestContext requestContext)
    {
        base.Initialize(requestContext);
        FixAntiForgeryTokenMvc1ToMvc2(this, requestContext);
    }
    private static void FixAntiForgeryTokenMvc1ToMvc2(
        Controller controller, RequestContext requestContext)
    {
        // stuff ....
    }
}

在您的“普通”控制器中，只需：

public class NormalController : BaseController
{
    // all the previous stuff
}

试一试……