asp.net MVC之AuthorizeAttribute浅析

using System;
using System.Collections.Generic;
using System.Linq;
using System.Security.Principal;
using System.Web.Mvc.Properties;
namespace System.Web.Mvc
{
    /// <summary>Specifies that access to a controller or action method is restricted to users who meet the authorization requirement.</summary>
    [AttributeUsage(AttributeTargets.Class | AttributeTargets.Method, Inherited = true, AllowMultiple = true)]
    public class AuthorizeAttribute : FilterAttribute, IAuthorizationFilter
    {
        private static readonly char[] _splitParameter = new char[]
        {
            ‘,‘
        };
        private readonly object _typeId = new object();
        private string _roles;
        private string[] _rolesSplit = new string[0];
        private string _users;
        private string[] _usersSplit = new string[0];
        /// <summary>Gets or sets the user roles that are authorized to access the controller or action method.</summary>
        /// <returns>The user roles that are authorized to access the controller or action method.</returns>
        public string Roles
        {
            get
            {
                return this._roles ?? string.Empty;
            }
            set
            {
                this._roles = value;
                this._rolesSplit = AuthorizeAttribute.SplitString(value);
            }
        }
        /// <summary>Gets the unique identifier for this attribute.</summary>
        /// <returns>The unique identifier for this attribute.</returns>
        public override object TypeId
        {
            get
            {
                return this._typeId;
            }
        }
        /// <summary>Gets or sets the users that are authorized to access the controller or action method.</summary>
        /// <returns>The users that are authorized to access the controller or action method.</returns>
        public string Users
        {
            get
            {
                return this._users ?? string.Empty;
            }
            set
            {
                this._users = value;
                this._usersSplit = AuthorizeAttribute.SplitString(value);
            }
        }
        /// <summary>When overridden, provides an entry point for custom authorization checks.</summary>
        /// <returns>true if the user is authorized; otherwise, false.</returns>
        /// <param name="httpContext">The HTTP context, which encapsulates all HTTP-specific information about an individual HTTP request.</param>
        /// <exception cref="T:System.ArgumentNullException">The <paramref name="httpContext" /> parameter is null.</exception>
        protected virtual bool AuthorizeCore(HttpContextBase httpContext)
        {
            if (httpContext == null)
            {
                throw new ArgumentNullException("httpContext");
            }
            IPrincipal user = httpContext.User;
            return user.Identity.IsAuthenticated && (this._usersSplit.Length <= 0 || this._usersSplit.Contains(user.Identity.Name, StringComparer.OrdinalIgnoreCase)) && (this._rolesSplit.Length <= 0 || this._rolesSplit.Any(new Func<string, bool>(user.IsInRole)));
        }
        private void CacheValidateHandler(HttpContext context, object data, ref HttpValidationStatus validationStatus)
        {
            validationStatus = this.OnCacheAuthorization(new HttpContextWrapper(context));
        }
        /// <summary>Called when a process requests authorization.</summary>
        /// <param name="filterContext">The filter context, which encapsulates information for using <see cref="T:System.Web.Mvc.AuthorizeAttribute" />.</param>
        /// <exception cref="T:System.ArgumentNullException">The <paramref name="filterContext" /> parameter is null.</exception>
        public virtual void OnAuthorization(AuthorizationContext filterContext)
        {
            if (filterContext == null)
            {
                throw new ArgumentNullException("filterContext");
            }
            if (OutputCacheAttribute.IsChildActionCacheActive(filterContext))
            {
                throw new InvalidOperationException(MvcResources.AuthorizeAttribute_CannotUseWithinChildActionCache);
            }
            bool flag = filterContext.ActionDescriptor.IsDefined(typeof(AllowAnonymousAttribute), true) || filterContext.ActionDescriptor.ControllerDescriptor.IsDefined(typeof(AllowAnonymousAttribute), true);
            if (flag)
            {
                return;
            }
            if (this.AuthorizeCore(filterContext.HttpContext))
            {
                HttpCachePolicyBase cache = filterContext.HttpContext.Response.Cache;
                cache.SetProxyMaxAge(new TimeSpan(0L));
                cache.AddValidationCallback(new HttpCacheValidateHandler(this.CacheValidateHandler), null);
                return;
            }
            this.HandleUnauthorizedRequest(filterContext);
        }
        /// <summary>Processes HTTP requests that fail authorization.</summary>
        /// <param name="filterContext">Encapsulates the information for using <see cref="T:System.Web.Mvc.AuthorizeAttribute" />. The <paramref name="filterContext" /> object contains the controller, HTTP context, request context, action result, and route data.</param>
        protected virtual void HandleUnauthorizedRequest(AuthorizationContext filterContext)
        {
            filterContext.Result = new HttpUnauthorizedResult();
        }
        /// <summary>Called when the caching module requests authorization.</summary>
        /// <returns>A reference to the validation status.</returns>
        /// <param name="httpContext">The HTTP context, which encapsulates all HTTP-specific information about an individual HTTP request.</param>
        /// <exception cref="T:System.ArgumentNullException">The <paramref name="httpContext" /> parameter is null.</exception>
        protected virtual HttpValidationStatus OnCacheAuthorization(HttpContextBase httpContext)
        {
            if (httpContext == null)
            {
                throw new ArgumentNullException("httpContext");
            }
            if (!this.AuthorizeCore(httpContext))
            {
                return HttpValidationStatus.IgnoreThisRequest;
            }
            return HttpValidationStatus.Valid;
        }
        internal static string[] SplitString(string original)
        {
            if (string.IsNullOrEmpty(original))
            {
                return new string[0];
            }
            IEnumerable<string> source = 
                from piece in original.Split(AuthorizeAttribute._splitParameter)
                let trimmed = piece.Trim()
                where !string.IsNullOrEmpty(trimmed)
                select trimmed;
            return source.ToArray<string>();
        }
    }
}

public class StudentViewModel {
        public int ID { get; set; }
        public string Name { get; set; }
        public string Tel { get; set; }
    }

    public class UserViewModel {
        [DisplayName("用户名")]
        public string UserName { get; set; }
        [DisplayName("密  码")]
        public string Pwd { get; set; }       
        public Role Role { get; set; }
    }

    public enum Role {
        Admin=1,
        Normal=2,
        System=3
    }

    public class Data {
        public static List<StudentViewModel> students = new List<StudentViewModel> {
            new StudentViewModel { ID=1, Name="张三", Tel="15808038502" },
            new StudentViewModel { ID=2, Name="李四", Tel="15708032302" },
            new StudentViewModel { ID=3, Name="王五", Tel="15562438502" },
            new StudentViewModel { ID=4, Name="赵六", Tel="15064534502" },
            new StudentViewModel { ID=5, Name="孙琦", Tel="15185465402" }
        };

        public static List<UserViewModel> users = new List<UserViewModel> {
            new UserViewModel { UserName="admin", Pwd="1", Role=Role.Admin },
            new UserViewModel { UserName="normal", Pwd="1", Role=Role.Normal},
            new UserViewModel { UserName="system", Pwd="1", Role=Role.System}
        };       
    }

    public class StudentController : Controller {
        
        public ActionResult Index() {
            return View(Data.Data.students);
        }
        public ActionResult FindOne(int? id) {
            return View("Index", Data.Data.students.Where(s => s.ID == id));
        }
    }

    public class AccountController : Controller {
        //用于获取登录页面
        [HttpGet]
        public ActionResult Login() {
            return View();
        }

        //用于登录验证
        [HttpPost]
        public ActionResult Login(LoginViewModel login) {
            if (ModelState.IsValid) {
                UserViewModel clogin = Data.Data.users.Where(u => u.UserName == login.UserName && u.Pwd == login.Pwd).FirstOrDefault();
                if (clogin != null) {
                    //FormsAuthentication.SetAuthCookie有两个参数
                    //第一个参数：设置当前登录用户的标识，可以自定义
                    //第二个参数：
                    //true:设置永久的。可以在web.config中设置过期时间，喏，下面的timeout就是过期时间，以分钟为单位。如下所示：
                    //     < system.web >
                    //         < authentication mode = "Forms" >
                    //           < forms loginUrl = "~/Account/Login" timeout = "2" />
                    //        </ authentication >
                    //     </ system.web >
                    //false：设置当前会话时间为有效时间，浏览器或页面已关闭就需要重新登录。
                    FormsAuthentication.SetAuthCookie(clogin.UserName, false);
                    return RedirectToAction("FindOne", "Student", new { id = 1 });
                }
            }
            return View();
        }

        //用于注销登录
        [Authorize]
        [NoCache]
        public ActionResult LoginOut() {
            FormsAuthentication.SignOut();
            return RedirectToAction("Login", "Account");
        }

        //用于显示无权限消息
        public ActionResult NoPemission() {
            return View();
        }
    }

@model MVCDemo2.ViewModels.UserViewModel
@{
    Layout = null;
}

<!DOCTYPE html>
<html>
<head>
    <meta name="viewport" content="width=device-width" />
    <title>Login</title>
    <link href="~/Content/w3.css" rel="stylesheet" />
    <style>
        html, body {
            margin: 0;
            padding: 0;
        }

        .main {
            width: 400px;
            height: 300px;
            position: absolute;
            left: 0;
            top: 0;
            right: 0;
            bottom: 0;
            margin: auto;
        }
    </style>
</head>
<body>

    <div class="main">
        <h3 class="w3-teal w3-center w3-margin-0" style="width:100%;">权限管理系统</h3>
        @using (Html.BeginForm("Login", "Account", FormMethod.Post, new { @class = "w3-container w3-card-4 w3-light-grey", style = "width:100%;" })) {           
                <table class="w3-table">
                    <tr>
                        <td style="width:20%; vertical-align:middle;text-align:right;">@Html.LabelFor(x => x.UserName)</td>
                        <td style="width:80%;">@Html.TextBoxFor(x => x.UserName, new { @class = "w3-input w3-border", placeholder = "用户名" })</td>
                    </tr>
                    <tr class="w3-padding">
                        <td style="vertical-align:middle;text-align:right;">@Html.LabelFor(x => x.Pwd)</td>
                        <td>@Html.PasswordFor(x => x.Pwd, new { @class = "w3-input w3-border", placeholder = "密  码" })</td>
                    </tr>
                    <tr style="display:none;">
                        <td><input type="text" id="returnUrl" name="returnUrl" value="@Url.Encode(Request.Url.ToString())"/></td>
                    </tr>
                    <tr>
                        <td></td>
                        <td style="text-align:left;">
                            <div style="width:30%;"><button type="submit" class="w3-btn-block w3-teal">登录</button></div>
                        </td>
                    </tr>
                </table>
        }
    </div>
    @*<script>
        function login() {

        }
    </script>*@
</body>
</html>

@{
    Layout = null;
}
<!DOCTYPE html>
<html>
<head>
    <meta name="viewport" content="width=device-width" />
    <title>NoPemission</title>
</head>
<body>
    <div> 
        您无权限操作该页面！
    </div>
</body>
</html>

    public class MyAuthorizeAttribute : AuthorizeAttribute {
        //设置是否拥有某些权限的状态码
        private int _status = 0;
        protected override bool AuthorizeCore(HttpContextBase httpContext) {
            //判断当前登录的用户 是否已经登录过且身份认证被通过
            if (httpContext.User.Identity.IsAuthenticated) {
                //判断当前的Roles字段是否是空，是空的，则说明该控制器或Action不需要角色权限控制
                if (string.IsNullOrWhiteSpace(Roles)) return true;
                //通过半角逗号去获取一个权限数组
                string[] roles = Roles.Split(new string[] { "," }, StringSplitOptions.RemoveEmptyEntries);
                if (roles.Length <= 0) return true;
                //找到当前登录的用户
                //httpContext.User.Identity.Name就是我们在Login验证登录时候设置的clogin.UserName。设置代码为FormsAuthentication.SetAuthCookie(clogin.UserName, false);
                UserViewModel user = Data.Data.users.Where(u => u.UserName == httpContext.User.Identity.Name).SingleOrDefault();
                //判断当前登录用户是否拥有访问该控制器或Action的权限，如果有，则返回true，可以访问；如果没有，则返回false，需要在OnAuthorization方法中设置该滤过器的Result。
                if (roles.Contains(((int)user.Role).ToString()))
                    return true;
                else {
                    //无角色权限 通过设置_status，可以在OnAuthorization方法中做出相应的判断，并设置相应的filterContext.Result
                    _status = 10;
                    //授权失败
                    return false;
                }
            }
            //没有登录，则返回false，表明身份认证未通过，程序直接会跳转到登录页面，登录页面可在web.config中设置
            //  < system.web >
            //      < authentication mode = "Forms" >
            //         < forms loginUrl = "~/Account/Login" timeout = "2" />
            //      </ authentication >
            //  </ system.web >
            else return false;
        }

        public override void OnAuthorization(AuthorizationContext filterContext) {
            //执行父类的OnAuthorization方法
            base.OnAuthorization(filterContext);
            //如果当前的_status有状态变化，则返回相应的ActionResult。当前返回的是无权限查看的提示界面。
            if (_status == 10)
                filterContext.Result = new RedirectToRouteResult(new System.Web.Routing.RouteValueDictionary(new {
                    //需要跳转的Controller
                    controller = "Account",
                    //需要跳转的Action
                    action = "NoPemission",
                    //返回的字段说明
                    returnUrl = filterContext.HttpContext.Request.Url,
                    returnMessage = "您无权限查看！"
                }));
        }
    }