Tùy chỉnh mã trạng thái HTTP OWIN/OAuth khi từ chối yêu cầu mã thông báo

Question

Tôi đã xuất phát OAuthAuthorizationServerProvider để xác thực cả khách hàng và chủ sở hữu tài nguyên.

Khi tôi xác nhận chủ sở hữu tài nguyên tôi thấy thông tin của họ là không hợp lệ, tôi gọi context.Rejected(), và HTTP response đi kèm với HTTP/400 Bad Request mã trạng thái trong khi tôi mong chờ HTTP/401 Unauthorized.

Tôi làm cách nào để tùy chỉnh mã trạng thái HTTP phản hồi của OAuthAuthorizationServerProvider?

Answer 1

Đây là cách chúng tôi ghi đè lên OwinMiddleware ... đầu tiên chúng tôi tạo phần mềm trung gian của riêng mình trên đầu trang của Owin ... Tôi nghĩ chúng tôi có vấn đề tương tự như bạn đã làm.

Trước tiên cần phải tạo ra một hằng số:

public class Constants 
{ 
    public const string OwinChallengeFlag = "X-Challenge"; 
} 

Và chúng tôi ghi đè OwinMiddleware

public class AuthenticationMiddleware : OwinMiddleware 
{ 
    public AuthenticationMiddleware(OwinMiddleware next) : base(next) { } 

    public override async Task Invoke(IOwinContext context) 
    { 
     await Next.Invoke(context); 

     if (context.Response.StatusCode == 400 && context.Response.Headers.ContainsKey(Constants.OwinChallengeFlag)) 
     { 
      var headerValues = context.Response.Headers.GetValues(Constants.OwinChallengeFlag); 
      context.Response.StatusCode = Convert.ToInt16(headerValues.FirstOrDefault()); 
      context.Response.Headers.Remove(Constants.OwinChallengeFlag); 
     } 

    } 
} 

Trong tập tin startup.Auth, chúng tôi cho phép overrid của Invoke Owin Commands

public void ConfigureAuth(IAppBuilder app) 
    .... 
     app.Use<AuthenticationMiddleware>(); //Allows override of Invoke OWIN commands 
    .... 

    } 

Và trong ApplicationOAuthProvider, chúng tôi đã sửa đổi GrantResourceOwnerCredentials.

public override async Task GrantResourceOwnerCredentials(OAuthGrantResourceOwnerCredentialsContext context) 
    { 
     using (UserManager<IdentityUser> userManager = _userManagerFactory()) 
     { 
      IdentityUser user = await userManager.FindAsync(context.UserName, context.Password); 

      if (user == null) 
      { 
       context.SetError("invalid_grant", "The user name or password is incorrect."); 
       context.Response.Headers.Add(Constants.OwinChallengeFlag, new[] { ((int)HttpStatusCode.Unauthorized).ToString() }); //Little trick to get this to throw 401, refer to AuthenticationMiddleware for more 
       //return; 
      } 
      ....