ASP.NET Core使用自定義驗證屬性控制訪問權限詳解

更新時間：2018年10月21日 14:57:05 作者：ROTA

這篇文章主要給大家介紹了關于ASP.NET Core使用自定義驗證屬性控制訪問權限的相關資料，這是我們在日常工作中經(jīng)常會遇到的一個需求，文中通過示例代碼介紹的非常詳細，需要的朋友可以參考下

前言

大家都知道在應用中，有時我們需要對訪問的客戶端進行有效性驗證，只有提供有效憑證（AccessToken）的終端應用能訪問我們的受控站點（如WebAPI站點），此時我們可以通過驗證屬性的方法來解決。

本文將詳細介紹ASP.NET Core使用自定義驗證屬性控制訪問權限的相關內(nèi)容，分享出來供大家參考學習，下面話不多說了，來一起看看詳細的介紹吧

方法如下

一、public class Startup的配置：

//啟用跨域訪問（不同端口也是跨域）
services.AddCors(options =>
{
options.AddPolicy("AllowOriginOtherBis",
builder => builder.WithOrigins("https://1.16.9.12:4432", "https://pc12.ato.biz:4432", "https://localhost:44384", "https://1.16.9.12:4432", "https://pc12.ato.biz:4432").AllowAnyMethod().AllowAnyHeader());
});

//啟用自定義屬性以便對控制器或Action進行[TerminalApp()]定義。

services.AddSingleton<IAuthorizationHandler, TerminalAppAuthorizationHandler>();
services.AddAuthorization(options =>
{
options.AddPolicy("TerminalApp", policyBuilder =>
{
policyBuilder.Requirements.Add(new TerminalAppAuthorizationRequirement());
});
});

二、public void Configure(IApplicationBuilder app, IHostingEnvironment env)中的配置：

app.UseHttpsRedirection();　　//使用Https傳輸
app.UseCors("AllowOriginOtherBis"); //根據(jù)定義啟用跨域設置

三、示例WebApi項目結(jié)構(gòu)：

四、主要代碼（我采用的從數(shù)據(jù)庫進行驗證）：

[AttributeUsage(AttributeTargets.Class | AttributeTargets.Method, AllowMultiple = true)]
 internal class TerminalAppAttribute : AuthorizeAttribute
 {
  public string AppID { get; }

  /// <summary>
  /// 指定客戶端訪問API
  /// </summary>
  /// <param name="appID"></param>
  public TerminalAppAttribute(string appID="") : base("TerminalApp")
  {
   AppID = appID;
  }
 }

public abstract class AttributeAuthorizationHandler<TRequirement, TAttribute> : AuthorizationHandler<TRequirement> where TRequirement : IAuthorizationRequirement where TAttribute : Attribute
 {
  protected override Task HandleRequirementAsync(AuthorizationHandlerContext context, TRequirement requirement)
  {
   var attributes = new List<TAttribute>();

   if ((context.Resource as AuthorizationFilterContext)?.ActionDescriptor is ControllerActionDescriptor action)
   {
    attributes.AddRange(GetAttributes(action.ControllerTypeInfo.UnderlyingSystemType));
    attributes.AddRange(GetAttributes(action.MethodInfo));
   }

   return HandleRequirementAsync(context, requirement, attributes);
  }

  protected abstract Task HandleRequirementAsync(AuthorizationHandlerContext context, TRequirement requirement, IEnumerable<TAttribute> attributes);

  private static IEnumerable<TAttribute> GetAttributes(MemberInfo memberInfo)
  {
   return memberInfo.GetCustomAttributes(typeof(TAttribute), false).Cast<TAttribute>();
  }
 }

 internal class TerminalAppAuthorizationHandler : AttributeAuthorizationHandler<TerminalAppAuthorizationRequirement,TerminalAppAttribute>
 {
  protected override Task HandleRequirementAsync(AuthorizationHandlerContext context, TerminalAppAuthorizationRequirement requirement, IEnumerable<TerminalAppAttribute> attributes)
  {
   object errorMsg = string.Empty;
   //如果取不到身份驗證信息，并且不允許匿名訪問，則返回未驗證403
   if (context.Resource is AuthorizationFilterContext filterContext &&
filterContext.ActionDescriptor is ControllerActionDescriptor descriptor)
   {
    //先判斷是否是匿名訪問,
    if (descriptor != null)
    {
     var actionAttributes = descriptor.MethodInfo.GetCustomAttributes(inherit: true);
     bool isAnonymous = actionAttributes.Any(a => a is AllowAnonymousAttribute);
     //非匿名的方法,鏈接中添加accesstoken值
     if (isAnonymous)
     {
      context.Succeed(requirement);
      return Task.CompletedTask;
     }
     else
     {
      //url獲取access_token
      //從AuthorizationHandlerContext轉(zhuǎn)成HttpContext，以便取出表求信息
      var httpContext = (context.Resource as AuthorizationFilterContext).HttpContext;
      //var questUrl = httpContext.Request.Path.Value.ToLower();
      string requestAppID = httpContext.Request.Headers["appid"];
      string requestAccessToken = httpContext.Request.Headers["access_token"];
      if ((!string.IsNullOrEmpty(requestAppID)) && (!string.IsNullOrEmpty(requestAccessToken)))
      {
       if (attributes != null)
       {
        //當不指定具體的客戶端AppID僅運用驗證屬性時默認所有客戶端都接受
        if (attributes.ToArray().ToString()=="") 
        {
         //任意一個在數(shù)據(jù)庫列表中的App都可以運行,否則先判斷提交的APPID與需要ID是否相符
         bool mat = false;
         foreach (var terminalAppAttribute in attributes)
         {
          if (terminalAppAttribute.AppID == requestAppID)
          {
           mat = true;
           break;
          }
         }
         if (!mat)
         {
          errorMsg = ReturnStd.NotAuthorize("客戶端應用未在服務端登記或未被授權運用當前功能.");
          return HandleBlockedAsync(context, requirement, errorMsg);
         }
        }
       }

       //如果未指定attributes，則表示任何一個終端服務都可以調(diào)用服務, 在驗證區(qū)域驗證終端提供的ID是否匹配數(shù)據(jù)庫記錄
       string valRst = ValidateToken(requestAppID, requestAccessToken);
       if (string.IsNullOrEmpty(valRst))
       {
        context.Succeed(requirement);
        return Task.CompletedTask;
       }
       else
       {
        errorMsg = ReturnStd.NotAuthorize("AccessToken驗證失敗(" + valRst + ")","91");
        return HandleBlockedAsync(context, requirement, errorMsg);
       }
      }
      else
      {
       errorMsg = ReturnStd.NotAuthorize("未提供AppID或Token."); 
       return HandleBlockedAsync(context, requirement, errorMsg);
       //return Task.CompletedTask;
      }
     }
    }
   }
   else
   {
    errorMsg = ReturnStd.NotAuthorize("FilterContext類型不匹配.");
    return HandleBlockedAsync(context, requirement, errorMsg);
   }

   errorMsg = ReturnStd.NotAuthorize("未知錯誤.");
   return HandleBlockedAsync(context,requirement, errorMsg);
  }


  //校驗票據(jù)（數(shù)據(jù)庫數(shù)據(jù)匹配）
  /// <summary>
  /// 驗證終端服務程序提供的AccessToken是否合法
  /// </summary>
  /// <param name="appID">終端APP的ID</param>
  /// <param name="accessToken">終端APP利用其自身AppKEY運算出來的AccessToken,與服務器生成的進行比對</param>
  /// <returns></returns>
  private string ValidateToken(string appID,string accessToken)
  {
   try
   {
    DBContextMain dBContext = new DBContextMain();
    string appKeyOnServer = string.Empty;
    //從數(shù)據(jù)庫讀取AppID對應的KEY(此KEY為加解密算法的AES_KEY
    AuthApp authApp = dBContext.AuthApps.FirstOrDefault(a => a.AppID == appID);
    if (authApp == null)
    {
     return "客戶端應用沒有在云端登記!";
    }
    else
    {
     appKeyOnServer = authApp.APPKey;
    }
    if (string.IsNullOrEmpty(appKeyOnServer))
    {
     return "客戶端應用基礎信息有誤!"; 
    }

    string tmpToken = string.Empty;
    tmpToken = System.Net.WebUtility.UrlDecode(accessToken);//解碼相應的Token到原始字符(因其中可能會有+=等特殊字符,必須編碼后傳遞)
    tmpToken = OCrypto.AES16Decrypt(tmpToken, appKeyOnServer); //使用APPKEY解密并分析

    if (string.IsNullOrEmpty(tmpToken))
    {
     return "客戶端提交的身份令牌運算為空!";
    }
    else
    {
     try
     {
      //原始驗證碼為im_cloud_sv001-appid-ticks格式
      //取出時間,與服務器時間對比,超過10秒即拒絕服務
      long tmpTime =Convert.ToInt64(tmpToken.Substring(tmpToken.LastIndexOf("-")+1));
      //DateTime dt = DateTime.ParseExact(tmpTime, "yyyyMMddHHmmss", CultureInfo.CurrentCulture);
      DateTime dt= new DateTime(tmpTime);
      bool IsInTimeSpan = (Convert.ToDouble(ODateTime.DateDiffSeconds(dt, DateTime.Now)) <= 7200);
      bool IsInternalApp = (tmpToken.IndexOf("im_cloud_sv001-") >= 0);
      if (!IsInternalApp || !IsInTimeSpan)
      {
       return "令牌未被許可或已經(jīng)失效!";
      }
      else
      {
       return string.Empty; //成功驗證
      }
     }
     catch (Exception ex)
     {
      return "令牌解析出錯(" + ex.Message + ")";
     }

    }
   }
   catch (Exception ex)
   {
    return "令牌解析出錯(" + ex.Message + ")";
   }
  }

  private Task HandleBlockedAsync(AuthorizationHandlerContext context, TerminalAppAuthorizationRequirement requirement, object errorMsg)
  {
   var authorizationFilterContext = context.Resource as AuthorizationFilterContext;
   authorizationFilterContext.Result = new JsonResult(errorMsg) { StatusCode = 202 };
   //設置為403會顯示不了自定義信息,改為Accepted202,由客戶端處理
   context.Succeed(requirement);
   return Task.CompletedTask;
  }
 }

 internal class TerminalAppAuthorizationRequirement : IAuthorizationRequirement
 {
  public TerminalAppAuthorizationRequirement()
  {
  }
 }

五、相應的Token驗證代碼：

[AutoValidateAntiforgeryToken] //在本控制器內(nèi)自動啟用跨站攻擊防護
 [Route("api/get_accesstoken")]
 public class GetAccessTokenController : Controller
 {
  //尚未限制訪問頻率
  //返回{"access_token":"ACCESS_TOKEN","expires_in":7200} 有效期2個小時
  //錯誤時返回{"errcode":40013,"errmsg":"invalid appid"}
  [AllowAnonymous]
  public ActionResult<string> Get()
  {
   try
   {
    string tmpToken = string.Empty;

    string appID = HttpContext.Request.Headers["appid"];
    string appKey = HttpContext.Request.Headers["appkey"];

    if ((appID.Length < 5) || appKey.Length != 32)
    {
     return "{'errcode':10000,'errmsg':'appid或appkey未提供'}";
    }
    //token采用im_cloud_sv001-appid-ticks數(shù)字
    long timeTk = DateTime.Now.Ticks; //輸出毫微秒：633603924670937500
             //DateTime dt = new DateTime(timeTk);//可以還原時間

    string plToken = "im_cloud1-" + appID + "-" + timeTk;
    tmpToken = OCrypto.AES16Encrypt(plToken, appKey); //使用APPKEY加密

    tmpToken = System.Net.WebUtility.UrlEncode(tmpToken);
    //編碼相應的Token(因其中可能會有+=等特殊字符,必須編碼后傳遞)
    tmpToken = "{'access_token':'" + tmpToken + "','expires_in':7200}";
    return tmpToken;
   }
   catch (Exception ex)
   {
    return "{'errcode':10001,'errmsg':'" + ex.Message +"'}";
   }
  }
 }

GetAccessTokenController.cs

六、這樣，在我們需要控制的地方加上[TerminalApp()] 即可，這樣所有授權的App都能訪問，當然，也可以使用[TerminalApp(“app01”)]限定某一個ID為app01的應用訪問。

 [Area("SYS")]  // 路由: api/sys/user
 [Produces("application/json")]
 [TerminalApp()] 
 public class UserController : Controller
{
//
}

七、一個CS客戶端通過Web API上傳數(shù)據(jù)調(diào)用示例：

string postURL = "http://sv12.ato.com/api/sys/user/postnew";
 
Dictionary<string, string> headerDic2 = new Dictionary<string, string>
{
 { "appid", MainFramework.CloudAppID },
 { "access_token", accessToken }
};
string pushRst = OPWeb.Post(postURL, headerDic2, "POST", sYS_Users);
if (string.IsNullOrEmpty(pushRst))
{
 MyMsg.Information("推送成功!");
}
else
{
 MyMsg.Information("推送失敗!", pushRst);
}

string accessToken = MainFramework.CloudAccessToken;
if (accessToken.IndexOf("ERROR:") >= 0)
{
 MyMsg.Information("獲取Token出錯:" + accessToken);
 return;
}

總結(jié)

以上就是這篇文章的全部內(nèi)容了，希望本文的內(nèi)容對大家的學習或者工作具有一定的參考學習價值，如果有疑問大家可以留言交流，謝謝大家對腳本之家的支持。

您可能感興趣的文章:

使用ASP.NET操作IIS7中應用程序
IIS7的操作和IIS5/6有很大的不同，在IIS7里增加了 Microsoft.Web.Administration 命名空間里，增加了ServerManager、Site幾個大類來操作IIS7。本文主要介紹.NET對IIS7的操作。
2015-10-10
在.NET?MAUI應用中配置應用生命周期事件
本文詳細講解了在.NET?MAUI應用中配置應用生命周期事件的方法，文中通過示例代碼介紹的非常詳細。對大家的學習或工作具有一定的參考借鑒價值，需要的朋友可以參考下
2022-03-03
如何在ASP.NET Core 的任意類中注入Configuration
這篇文章主要介紹了如何在 ASP.NET Core 的任意類中注入Configuration ,幫助大家更好的理解和學習使用.net技術，感興趣的朋友可以了解下
2021-04-04
Visual Studio 2017設置版權的方法
這篇文章主要為大家詳細介紹了Visual Studio 2017設置版權的方法，具有一定的參考價值，感興趣的小伙伴們可以參考一下
2017-03-03
asp.net 臨時數(shù)據(jù)保存實現(xiàn)代碼
在一個程序運行的時候往往我們會回到上一個頁面，或者我們會需要當時留在這個頁面的臨時數(shù)據(jù)，例如，我們用百度搜索“腳本”之后我們會看到很多那啥我就不再說了啊！然后我們返回本來頁面在文本框內(nèi)會有“腳本”的字樣，我們一下代碼段就是實現(xiàn)這個功能
2012-04-04
ADO.NET通用數(shù)據(jù)庫訪問類
這篇文章主要為大家介紹了ADO.NET通用數(shù)據(jù)庫訪問類，利用ADO.NET的體系架構(gòu)打造通用的數(shù)據(jù)庫訪問通用類，感興趣的小伙伴們可以參考一下
2016-03-03
分享Visual Studio原生開發(fā)的10個調(diào)試技巧（2）
這篇文章又為大家分享了Visual Studio原生開發(fā)的10個調(diào)試技巧，感興趣的朋友可以參考一下
2015-11-11
asp.net(C#) 開源資源大匯總
asp.net(C#) 開源資源大匯總，想要學習c#的朋友可以參考下
2012-05-05
asp.net(c#)下讀取word文檔的方法小結(jié)
asp.net(c#)下讀取word文檔的方法小結(jié),需要的朋友可以參考下。
2011-07-07
asp.net下計算數(shù)字1至10的總和
老師布置給Insus.NET第四道題目。一開始時，是使用下面的方法解答
2012-05-05