How to return UNAUTHORIZED response if Basic headers fail to be decoded?

[JasterV]

Summary

I have the following middleware:

pub async fn authorize(
    State(state): State<AuthState>,
    auth_header: Option<TypedHeader<Authorization<Basic>>>,
    req: Request,
    next: Next,
) -> Result<Response, UnauthorizedError> {
// ...
}

And when I try to call it with the following headers:

Authorization: Basic user:password

The headers fail to decode and returns a 400 error without me being able to handle this case in the code.

Instead, if I encode the credentials in base64 and make the call again with headers:

Authorization: Basic dXNlcjpwYXNzd29yZA==

The middleware then is able to decode the Basic auth header and my code gets executed.

Is there a way to have control over the case when the auth headers fail to decode? Because in that case I would like to return an UNAUTHORIZED error instead of BAD_REQUEST

Thank you :)

axum version

0.8.4

[yanns]

Can you use a Result instead of an Option?

[JasterV]

That sounds interesting, what should I put as the error type though?

[yanns]

By looking at https://docs.rs/axum-extra/latest/axum_extra/struct.TypedHeader.html#impl-FromRequestParts%3CS%3E-for-TypedHeader%3CT%3E, it seems that the error type should be TypedHeaderRejection (https://docs.rs/axum-extra/latest/axum_extra/typed_header/struct.TypedHeaderRejection.html)

[JasterV]

Great, I will try it as soon as possible and get back :D Thank you

[JasterV]

That worked, thank you!

[mladedav]

Alternatively, you can use WithRejection to just map the rejection during extraction.