聊聊在 .Net 5.0 中自定义授权响应

ASP.NET Core 授权框架中经常要求的[1]一项功能是能够在授权失败时自定义 HTTP 响应。

以前，唯一的方法是IAuthorizationService直接在您的控制器中(或通过过滤器)调用授权服务，类似于基于资源的授权方法[2]或实现您自己的授权过滤器[3]。

从 .NET 5.0 开始，您现在可以通过实现IAuthorizationMiddlewareResultHandler接口来自定义 HTTP 响应;当授权失败时，授权框架会自动调用中间件。

这是记录[4]在微软文档的网站，但根据我的具体使用情况我花了不少时间才找到。

我一直在采取措施将旧的 ASP.NET Web API 应用程序移植到 .NET Core 5.0。此 API 具有分层 URI 结构，因此大多数端点将位于“站点”资源下，例如：

为了验证用户是否有权访问指定站点，该应用程序以前使用自定义操作过滤器来提取siteId路由参数并根据用户的声明对其进行验证。迁移到 .NET 5.0 我想利用授权框架来实现这种基于资源的授权，但同样不想在每个控制器中复制这个逻辑。

我的解决方案是实现一个执行类似操作的授权处理程序，获取siteId参数并验证用户的访问权限：

public class SiteAccessAuthorizationHandler : AuthorizationHandler<SiteAccessRequirement>
{
private const string SiteIdRouteParameter = "siteId";
private readonly ILogger<SiteAccessAuthorizationHandler> _logger;
public SiteAccessAuthorizationHandler(ILogger<SiteAccessAuthorizationHandler> logger)
{
_logger = logger.NotNull(nameof(logger));
}
protected override Task HandleRequirementAsync(AuthorizationHandlerContext context, SiteAccessRequirement requirement)
{
context.NotNull(nameof(context));
requirement.NotNull(nameof(requirement));
if (context.Resource is HttpContext httpContext
&& httpContext.GetRouteData().Values.TryGetValue(SiteIdRouteParameter, out object? routeValue)
&& routeValue is string siteId)
{
string qualifiedId = $"sites/{siteId}";
AccountPrincipal account = context.User.ToAccount();
_logger.LogDebug("Validating access to Site {SiteId} from User {UserId}.", qualifiedId, account.GetAuthIdentifier());
if (account.CanAccessSite(qualifiedId))
{
context.Succeed(requirement);
}
else
{
_logger.LogWarning("Site validation failed. User {UserId} is not permitted to access {SiteId}.", account.GetAuthIdentifier(), qualifiedId);
}
}
return Task.CompletedTask;
}
}

然后将其注册为授权策略的一部分：

services.AddAuthorization(options =>
{
options.FallbackPolicy = Policies.FallbackPolicy;
options.AddPolicy("SiteAccess", Policies.SiteAccessPolicy);
})
public static AuthorizationPolicy SiteAccessPolicy =>
ConfigureDefaults(new AuthorizationPolicyBuilder())
.AddRequirements(new SiteAccessRequirement())
.Build();
private static AuthorizationPolicyBuilder ConfigureDefaults(AuthorizationPolicyBuilder builder)
=> builder.AddAuthenticationSchemes(JwtBearerDefaults.AuthenticationScheme)
.RequireAuthenticatedUser()
.RequireClaim(JwtClaimTypes.ClientId);

并应用于控制器和/或动作：

[Authorize(Policy = "SiteAccess")]
[HttpGet("{siteId}", Name = RouteNames.SiteRoute)]
public async Task<IActionResult> GetSiteAsync(string siteId, CancellationToken cancellationToken)
{
var site = await _session.LoadAsync<CMS.Domain.Site>($"sites/{siteId}", cancellationToken);
return site is null ? NotFound() : Ok(Enrich(_mapper.Map<Site>(site), true));
}

当我尝试访问未映射到当前用户的站点时，我会收到HTTP 403 - Forbidden响应。

这样虽然达到了保护站点资源的目的，但也存在泄露用户无权访问的站点信息的弊端。因此最好返回一个HTTP 404 - Not Found响应。考虑到该站点不存在于用户的站点资源集合中，这在语义上也是有意义的。

如果您想知道为什么我不只是将用户过滤器作为查询的一部分，那是因为用户/帐户与内容域是分开的，并且由于数据模型的设计以及我使用的事实键值存储，验证访问的责任转移到应用层。

为了实现上述目标，我们可以使用 newIAuthorizationMiddlewareResultHandler并创建一个处理程序，当由于我的站点访问要求未得到满足而导致授权失败时，该处理程序会转换 HTTP 响应：

public class AuthorizationResultTransformer : IAuthorizationMiddlewareResultHandler
{
private readonly IAuthorizationMiddlewareResultHandler _handler;
public AuthorizationResultTransformer()
{
_handler = new AuthorizationMiddlewareResultHandler();
}
public async Task HandleAsync(
RequestDelegate requestDelegate,
HttpContext httpContext,
AuthorizationPolicy authorizationPolicy,
PolicyAuthorizationResult policyAuthorizationResult)
{
if (policyAuthorizationResult.Forbidden && policyAuthorizationResult.AuthorizationFailure != null)
{
if (policyAuthorizationResult.AuthorizationFailure.FailedRequirements.Any(requirement => requirement is SiteAccessRequirement))
{
httpContext.Response.StatusCode = (int)HttpStatusCode.NotFound;
return;
}
// Other transformations here
}
await _handler.HandleAsync(requestDelegate, httpContext, authorizationPolicy, policyAuthorizationResult);
}
}

在上面的代码中，我检查授权失败(结果是禁止)和失败的要求，相应地更改HTTP状态代码;否则我们通过调用内置的AuthorizationMiddlewareResultHandler.

为了连接自定义处理程序，它在启动时注册：

services.AddAuthorization(options =>
{
options.FallbackPolicy = Policies.FallbackPolicy;
options.AddPolicy("SiteAccess", Policies.SiteAccessPolicy);
})
.AddSingleton<IAuthorizationMiddlewareResultHandler, AuthorizationResultTransformer>();