淺談基于URL的權限控制ASP.NET MVC中的實現

在數據庫中新建2個表。PermissionItem表用于保存權限ID和頁面路徑的關系，一個權限ID可以有多個頁面，一般同一個權限ID下的頁面是為了實現同一個功能。PermissionList表用于保存用戶所具有的權限。

Code  
USE [UrlAuthorize]  
GO  
/****** Object:  Table [dbo].[PermissionList]    Script Date: 07/07/2009 00:07:10 ******/  
SET ANSI_NULLS ON  
GO  
SET QUOTED_IDENTIFIER ON  
GO  
CREATE TABLE [dbo].[PermissionList](  
    [ID] [int] IDENTITY(1,1) NOT NULL,  
    [PermissionID] [int] NOT NULL,  
    [UserID] [int] NOT NULL,  
 CONSTRAINT [PK_PermissionList] PRIMARY KEY CLUSTERED   
(  
    [ID] ASC  
)WITH (PAD_INDEX  = OFF, STATISTICS_NORECOMPUTE  = OFF, IGNORE_DUP_KEY = OFF, ALLOW_ROW_LOCKS  = ON, ALLOW_PAGE_LOCKS  = ON) ON [PRIMARY]  
) ON [PRIMARY]  
GO  
SET IDENTITY_INSERT [dbo].[PermissionList] ON  
INSERT [dbo].[PermissionList] ([ID], [PermissionID], [UserID]) VALUES (1, 2, 1)  
INSERT [dbo].[PermissionList] ([ID], [PermissionID], [UserID]) VALUES (2, 3, 1)  
SET IDENTITY_INSERT [dbo].[PermissionList] OFF  
/****** Object:  Table [dbo].[PermissionItem]    Script Date: 07/07/2009 00:07:10 ******/  
SET ANSI_NULLS ON  
GO  
SET QUOTED_IDENTIFIER ON  
GO  
SET ANSI_PADDING ON  
GO  
CREATE TABLE [dbo].[PermissionItem](  
    [ID] [int] IDENTITY(1,1) NOT NULL,  
    [PermissionID] [int] NOT NULL,  
    [Name] [nvarchar](50) NOT NULL,  
    [Route] [varchar](100) NOT NULL,  
 CONSTRAINT [PK_PermissionItem] PRIMARY KEY CLUSTERED   
(  
    [ID] ASC  
)WITH (PAD_INDEX  = OFF, STATISTICS_NORECOMPUTE  = OFF, IGNORE_DUP_KEY = OFF, ALLOW_ROW_LOCKS  = ON, ALLOW_PAGE_LOCKS  = ON) ON [PRIMARY]  
) ON [PRIMARY]  
GO  
SET ANSI_PADDING OFF  
GO  
SET IDENTITY_INSERT [dbo].[PermissionItem] ON  
INSERT [dbo].[PermissionItem] ([ID], [PermissionID], [Name], [Route]) VALUES (1, 1, N'測試頁1', N'/Test/Page1')  
INSERT [dbo].[PermissionItem] ([ID], [PermissionID], [Name], [Route]) VALUES (2, 2, N'測試頁2', N'/Test/Page2')  
INSERT [dbo].[PermissionItem] ([ID], [PermissionID], [Name], [Route]) VALUES (3, 3, N'測試頁3', N'/Test/Page3')  
INSERT [dbo].[PermissionItem] ([ID], [PermissionID], [Name], [Route]) VALUES (5, 1, N'測試頁4', N'/Test/Page4')  
INSERT [dbo].[PermissionItem] ([ID], [PermissionID], [Name], [Route]) VALUES (6, 2, N'測試頁5', N'/Test/Page5')  
SET IDENTITY_INSERT [dbo].[PermissionItem] OFF 

數據庫中的示例表示Page1和Page4同屬于權限1，Page2和Page5同屬于權限2，Page3屬于權限3。用戶ID為1的用戶具有權限2和3。

在ASP.NET MVC項目中新建一個AccountHelper類，這是一個輔助類。GetPermissionItems方法用于獲取權限ID和頁面路徑的對應關系。這是全局的，并且每個用戶在訪問頁面時都會用到這些信息，所以存入Cache中。數據庫的相關操作這里使用的是ADO.NET Entity Framework。

 1/**//// <summary>
 2/// 獲取權限項
 3/// </summary>
 4/// <returns>權限項列表</returns>
 5public static List<PermissionItem> GetPermissionItems()
 6{
 7     // 如果緩存中已經存在權限列表信息，則直接從緩存中讀取。
 8      if (HttpContext.Current.Cache["PermissionItems"] == null)
 9     {
10          // 如果緩存中沒有權限列表信息，則從數據庫獲取并寫入緩存
11           UrlAuthorizeEntities db = new UrlAuthorizeEntities();
12          var items = db.PermissionItem.Where(c => c.PermissionID > 0).ToList();
13          HttpContext.Current.Cache["PermissionItems"] = items;
14     }
15
16     // 這個緩存中保存了所有需要進行權限控制的頁面所對應的權限ID
17     return (List<PermissionItem>)HttpContext.Current.Cache["PermissionItems"];
18}
19

GetUserPermission方法是將用戶所具有的權限ID保存到一個一維Int32數組中。這個信息每個用戶是不同的，但是會經常使用到，所以存入Session。

1/**//// <summary>
 2/// 獲取用戶權限
 3/// </summary>
 4/// <param name="userID">用戶ID</param>
 5/// <returns>用戶權限數組</returns>
 6public static Int32[] GetUserPermission(int userID)
 7{
 8    // 如果緩存中已經存在權限列表信息，則直接從緩存中讀取。
 9    if (HttpContext.Current.Session["Permission"] == null)
10    {
11        // 從數據庫獲取用戶權限并將權限ID放到int數組并存入Session
12        UrlAuthorizeEntities db = new UrlAuthorizeEntities();
13        var permissions = db.PermissionList.Where(c => c.UserID == userID).Select(c=>c.PermissionID).ToArray();
14        HttpContext.Current.Session["Permission"] = permissions;
15    }
16    return (Int32[])HttpContext.Current.Session["Permission"];
17}
18

再新建一個UrlAuthorizeAttribute類，繼承自AuthorizeAttribute，這是一個Filter。我們重寫它的OnAuthorization方法，以在ASP.NET頁生命周期身份驗證階段執行它。

1/**//// <summary>
 2/// 重寫OnAuthorization
 3/// </summary>
 4/// <param name="filterContext"></param>
 5public override void OnAuthorization(AuthorizationContext filterContext)
 6{
 7    // 獲取權限項列表
 8    List<PermissionItem> pItems = AccountHelper.GetPermissionItems();
 9
10    // 獲取當前訪問頁面對應的權限ID。如果item為空則表示當前頁面沒有權限控制信息，不需要進行權限控制
11    var item = pItems.FirstOrDefault(c => c.Route == filterContext.HttpContext.Request.Path);
12
13    if (item != null)
14    {
15        if (Array.IndexOf<Int32>(AccountHelper.GetUserPermission(int.Parse(filterContext.HttpContext.Session["UserID"].ToString())), item.PermissionID) == -1)
16        {
17            // 提示權限不夠，也可以跳轉到其他頁面
18            filterContext.HttpContext.Response.Write("沒有權限訪問該頁面");
19            filterContext.HttpContext.Response.End();
20        }
21    }
22    else
23    {
24        // 如果權限項列表中不存在當前頁面對應的權限ID則所有用戶都不允許訪問，直接提示無權訪問。***注1***
25        filterContext.HttpContext.Response.Write("沒有權限訪問該頁面");
26        filterContext.HttpContext.Response.End();
27    }
28}
29

至此，主要的工作都已經完成了的。接下來我們只需要在需要進行基于URL權限控制的Action或Controller前加上[UrlAuthorize]，這些Action或Controller中的所有Actions就會自動被UrlAuthorize這個Filter進行處理。如果某一個Action被標上了[UrlAuthorize]，而數據庫中又不存在該頁面對應的權限ID，那么根據示例的代碼，所有用戶都將無法訪問這個頁面，如果需要更改這個設置，可以修改上面“注1”下面的2行代碼。

【編輯推薦】

1. ASP.NET的錯誤處理機制
2. ASP.NET多附件上傳和附件編輯的實現
3. ASP.NET中性能和擴展性的秘密
4. ASP.NET 3.5圖表控件親密接觸
5. 詳解ASP.NET的四種狀態