test(security): cover gateway token authentication
This commit is contained in:
@@ -12,6 +12,8 @@ import org.springframework.security.oauth2.jwt.JwtDecoder;
|
|||||||
import org.springframework.security.oauth2.jwt.JwtException;
|
import org.springframework.security.oauth2.jwt.JwtException;
|
||||||
import org.springframework.security.web.AuthenticationEntryPoint;
|
import org.springframework.security.web.AuthenticationEntryPoint;
|
||||||
import org.springframework.web.filter.OncePerRequestFilter;
|
import org.springframework.web.filter.OncePerRequestFilter;
|
||||||
|
import org.slf4j.Logger;
|
||||||
|
import org.slf4j.LoggerFactory;
|
||||||
|
|
||||||
import java.io.IOException;
|
import java.io.IOException;
|
||||||
|
|
||||||
@@ -19,6 +21,7 @@ import java.io.IOException;
|
|||||||
* Servlet JWT 认证过滤器,解析 Bearer 令牌、校验会话并建立当前请求的安全上下文。
|
* Servlet JWT 认证过滤器,解析 Bearer 令牌、校验会话并建立当前请求的安全上下文。
|
||||||
*/
|
*/
|
||||||
public class JwtAuthenticationFilter extends OncePerRequestFilter {
|
public class JwtAuthenticationFilter extends OncePerRequestFilter {
|
||||||
|
private static final Logger LOG = LoggerFactory.getLogger(JwtAuthenticationFilter.class);
|
||||||
private static final String BEARER = "Bearer ";
|
private static final String BEARER = "Bearer ";
|
||||||
private final JwtDecoder jwtDecoder;
|
private final JwtDecoder jwtDecoder;
|
||||||
private final JwtClaimsMapper claimsMapper;
|
private final JwtClaimsMapper claimsMapper;
|
||||||
@@ -60,6 +63,7 @@ public class JwtAuthenticationFilter extends OncePerRequestFilter {
|
|||||||
SecurityContextHolder.getContext().setAuthentication(authentication);
|
SecurityContextHolder.getContext().setAuthentication(authentication);
|
||||||
filterChain.doFilter(request, response);
|
filterChain.doFilter(request, response);
|
||||||
} catch (JwtException | IllegalArgumentException exception) {
|
} catch (JwtException | IllegalArgumentException exception) {
|
||||||
|
LOG.warn("JWT authentication rejected for {}: {}", request.getRequestURI(), exception.getMessage());
|
||||||
SecurityContextHolder.clearContext();
|
SecurityContextHolder.clearContext();
|
||||||
authenticationEntryPoint.commence(request, response,
|
authenticationEntryPoint.commence(request, response,
|
||||||
new InvalidJwtAuthenticationException("Invalid or expired access token", exception));
|
new InvalidJwtAuthenticationException("Invalid or expired access token", exception));
|
||||||
|
|||||||
+30
@@ -41,4 +41,34 @@ class JwtAuthenticationFilterTest {
|
|||||||
assertThat(invoked).isTrue();
|
assertThat(invoked).isTrue();
|
||||||
assertThat(SecurityContextHolder.getContext().getAuthentication()).isNull();
|
assertThat(SecurityContextHolder.getContext().getAuthentication()).isNull();
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
void authenticatesMinimalOverseasTokenWithGatewaySnapshot() throws Exception {
|
||||||
|
Jwt jwt = Jwt.withTokenValue("overseas").header("alg", "RS256").subject("42")
|
||||||
|
.issuedAt(Instant.now()).expiresAt(Instant.now().plusSeconds(60))
|
||||||
|
.claim("tenant_id", 0L).claim("app_code", "PLATFORM")
|
||||||
|
.claim("permission_version", 7L).claim("token_version", 1).build();
|
||||||
|
JwtAuthenticationFilter filter = new JwtAuthenticationFilter(token -> jwt, new JwtClaimsMapper(),
|
||||||
|
new JsonAuthenticationEntryPoint(new ObjectMapper()),
|
||||||
|
new GatewayAccessTokenSessionValidator("gateway-secret"));
|
||||||
|
MockHttpServletRequest request = new MockHttpServletRequest();
|
||||||
|
request.setRequestURI("/api/platform/v1/last-leg/orders");
|
||||||
|
request.addHeader(HttpHeaders.AUTHORIZATION, "Bearer overseas");
|
||||||
|
request.addHeader(GatewayAccessTokenSessionValidator.TOKEN_HEADER, "gateway-secret");
|
||||||
|
request.addHeader(GatewayAccessTokenSessionValidator.ROLES_HEADER, "PLATFORM_ADMIN");
|
||||||
|
request.addHeader(GatewayAccessTokenSessionValidator.AUTHORITIES_HEADER, "platform:lastleg:view");
|
||||||
|
MockHttpServletResponse response = new MockHttpServletResponse();
|
||||||
|
AtomicBoolean invoked = new AtomicBoolean();
|
||||||
|
|
||||||
|
filter.doFilter(request, response, (request1, response1) -> {
|
||||||
|
var authentication = SecurityContextHolder.getContext().getAuthentication();
|
||||||
|
assertThat(authentication).isNotNull();
|
||||||
|
assertThat(authentication.getAuthorities()).extracting("authority")
|
||||||
|
.contains("APP_PLATFORM", "ROLE_PLATFORM_ADMIN", "platform:lastleg:view");
|
||||||
|
invoked.set(true);
|
||||||
|
});
|
||||||
|
|
||||||
|
assertThat(invoked).isTrue();
|
||||||
|
assertThat(response.getStatus()).isEqualTo(200);
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
Reference in New Issue
Block a user