Public Access
feat(security): check role markers on services
Every service method a controller calls carries @RequiresRole or @Public, checked by a method interceptor and by ArchUnit. deleteByAdmin needs PLATFORM_ADMIN; the Steam callback's linkVerifiedSteamId stays unmarked.
This commit is contained in:
1 parent
ff26760d40
commit
5341e0b870
32 files changed
+915
-2
No files matched your search
@@ -119,4 +119,17 @@ class ArchitectureTest extends AbstractArchitectureTest {
|
||||
.orShould().beAnnotatedWith(org.springframework.web.bind.annotation.ControllerAdvice.class)
|
||||
.orShould().beAnnotatedWith(org.springframework.web.bind.annotation.RestControllerAdvice.class)
|
||||
.because("Classes in 'api' package should be registered as Spring @RestController or @ControllerAdvice or @RestControllerAdvice beans.");
|
||||
|
||||
// ===================================================================================
|
||||
// 6. ROLE MARKERS (see RoleAnnotationRules)
|
||||
// ===================================================================================
|
||||
|
||||
@ArchTest
|
||||
static final ArchRule controllers_call_only_marked_service_methods = RoleAnnotationRules.controllers_call_only_marked_service_methods;
|
||||
|
||||
@ArchTest
|
||||
static final ArchRule one_role_marker_per_element = RoleAnnotationRules.one_role_marker_per_element;
|
||||
|
||||
@ArchTest
|
||||
static final ArchRule services_never_call_a_method_that_needs_more = RoleAnnotationRules.services_never_call_a_method_that_needs_more;
|
||||
}
|
||||
@@ -0,0 +1,192 @@
|
||||
package auth_service.architecture;
|
||||
|
||||
import com.tngtech.archunit.base.DescribedPredicate;
|
||||
import com.tngtech.archunit.core.domain.JavaClass;
|
||||
import com.tngtech.archunit.core.domain.JavaCodeUnit;
|
||||
import com.tngtech.archunit.core.domain.JavaCodeUnitAccess;
|
||||
import com.tngtech.archunit.core.domain.JavaMethod;
|
||||
import com.tngtech.archunit.core.domain.JavaModifier;
|
||||
import com.tngtech.archunit.core.domain.properties.HasAnnotations;
|
||||
import com.tngtech.archunit.lang.ArchCondition;
|
||||
import com.tngtech.archunit.lang.ArchRule;
|
||||
import com.tngtech.archunit.lang.ConditionEvents;
|
||||
import com.tngtech.archunit.lang.SimpleConditionEvent;
|
||||
import auth_service.shared.security.Public;
|
||||
import auth_service.shared.security.RequiresRole;
|
||||
import auth_service.shared.security.Role;
|
||||
import org.springframework.scheduling.annotation.Async;
|
||||
|
||||
import java.lang.annotation.Annotation;
|
||||
import java.util.ArrayDeque;
|
||||
import java.util.Deque;
|
||||
import java.util.HashSet;
|
||||
import java.util.List;
|
||||
import java.util.Optional;
|
||||
import java.util.Set;
|
||||
import java.util.stream.Stream;
|
||||
|
||||
import static com.tngtech.archunit.lang.syntax.ArchRuleDefinition.classes;
|
||||
|
||||
/**
|
||||
* The rules for {@link RequiresRole} and {@link Public}. Core has the same rules plus its {@code Internal} marker.
|
||||
*/
|
||||
final class RoleAnnotationRules {
|
||||
|
||||
private static final List<Class<? extends Annotation>> MARKERS = List.of(RequiresRole.class, Public.class);
|
||||
|
||||
private static final String APP_PACKAGE = "auth_service.";
|
||||
|
||||
private RoleAnnotationRules() {
|
||||
}
|
||||
|
||||
/**
|
||||
* What a marked method needs from its caller. {@code role} is set only for {@link RequiresRole}.
|
||||
*/
|
||||
private record Need(Class<? extends Annotation> marker, Role role) {
|
||||
|
||||
boolean isMetBy(Need caller) {
|
||||
if (marker == Public.class) {
|
||||
return true;
|
||||
}
|
||||
return caller.marker == RequiresRole.class && caller.role.atLeast(role);
|
||||
}
|
||||
|
||||
@Override
|
||||
public String toString() {
|
||||
return marker == RequiresRole.class ? "@RequiresRole(" + role + ")" : "@" + marker.getSimpleName();
|
||||
}
|
||||
}
|
||||
|
||||
private static long markerCount(HasAnnotations<?> element) {
|
||||
return MARKERS.stream().filter(element::isAnnotatedWith).count();
|
||||
}
|
||||
|
||||
private static Optional<Need> need(HasAnnotations<?> element) {
|
||||
for (Class<? extends Annotation> marker : MARKERS) {
|
||||
if (element.isAnnotatedWith(marker)) {
|
||||
Role role = marker == RequiresRole.class
|
||||
? element.getAnnotationOfType(RequiresRole.class).value() : null;
|
||||
return Optional.of(new Need(marker, role));
|
||||
}
|
||||
}
|
||||
return Optional.empty();
|
||||
}
|
||||
|
||||
private static Optional<Need> effectiveNeed(JavaMethod method) {
|
||||
Optional<Need> own = need(method);
|
||||
return own.isPresent() ? own : need(method.getOwner());
|
||||
}
|
||||
|
||||
private record MethodAccess(JavaCodeUnitAccess<?> access, JavaMethod target) {
|
||||
|
||||
@Override
|
||||
public String toString() {
|
||||
return target.getOwner().getSimpleName() + "." + target.getName() + " called in "
|
||||
+ access.getSourceCodeLocation();
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* Calls and method references ({@code service::method}) to non-static methods: a proxy never sees a static call,
|
||||
* so a marker there would promise a check that does not run.
|
||||
*/
|
||||
private static List<MethodAccess> methodAccesses(JavaCodeUnit from) {
|
||||
return Stream.<JavaCodeUnitAccess<?>>concat(from.getMethodCallsFromSelf().stream(),
|
||||
from.getMethodReferencesFromSelf().stream())
|
||||
.flatMap(access -> access.getTarget().resolveMember().stream()
|
||||
.filter(JavaMethod.class::isInstance).map(JavaMethod.class::cast)
|
||||
.filter(method -> !method.getModifiers().contains(JavaModifier.STATIC))
|
||||
.map(method -> new MethodAccess(access, method)))
|
||||
.toList();
|
||||
}
|
||||
|
||||
private static boolean isService(JavaClass javaClass) {
|
||||
return javaClass.getSimpleName().endsWith("Service") && javaClass.getPackageName().contains(".service");
|
||||
}
|
||||
|
||||
private static boolean isAsync(JavaMethod method) {
|
||||
return method.isAnnotatedWith(Async.class) || method.getOwner().isAnnotatedWith(Async.class);
|
||||
}
|
||||
|
||||
private static final DescribedPredicate<JavaClass> SERVICE_CLASSES =
|
||||
DescribedPredicate.describe("are services", RoleAnnotationRules::isService);
|
||||
|
||||
/**
|
||||
* Every service method a controller calls is marked, and is not {@code @Async}: there the check would never run.
|
||||
*/
|
||||
static final ArchRule controllers_call_only_marked_service_methods = classes()
|
||||
.that().resideInAPackage("..api..")
|
||||
.should(new ArchCondition<>("call only service methods that carry a role marker and are not @Async") {
|
||||
@Override
|
||||
public void check(JavaClass controller, ConditionEvents events) {
|
||||
for (JavaCodeUnit codeUnit : controller.getCodeUnits()) {
|
||||
for (MethodAccess access : methodAccesses(codeUnit)) {
|
||||
if (!isService(access.target().getOwner())) {
|
||||
continue;
|
||||
}
|
||||
if (effectiveNeed(access.target()).isEmpty()) {
|
||||
events.add(SimpleConditionEvent.violated(access.access(), "no role marker on " + access));
|
||||
}
|
||||
if (isAsync(access.target())) {
|
||||
events.add(SimpleConditionEvent.violated(access.access(), "@Async method " + access));
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
})
|
||||
.because("a new endpoint must not forget the role check");
|
||||
|
||||
static final ArchRule one_role_marker_per_element = classes()
|
||||
.should(new ArchCondition<>("carry at most one role marker on the class and on each method") {
|
||||
@Override
|
||||
public void check(JavaClass javaClass, ConditionEvents events) {
|
||||
if (markerCount(javaClass) > 1) {
|
||||
events.add(SimpleConditionEvent.violated(javaClass, javaClass.getName() + " has two role markers"));
|
||||
}
|
||||
for (JavaMethod method : javaClass.getMethods()) {
|
||||
if (markerCount(method) > 1) {
|
||||
events.add(SimpleConditionEvent.violated(method, method.getFullName() + " has two role markers"));
|
||||
}
|
||||
}
|
||||
}
|
||||
})
|
||||
.because("the marker must say one thing");
|
||||
|
||||
/**
|
||||
* Starts at every method the proxy checks (not private, not static) and follows every call that is not checked
|
||||
* again: self calls and calls of unmarked methods. Calls through an interface (a port) end the walk: the
|
||||
* implementation is not known here.
|
||||
*/
|
||||
static final ArchRule services_never_call_a_method_that_needs_more = classes()
|
||||
.that(SERVICE_CLASSES)
|
||||
.should(new ArchCondition<>("never call a service method that needs more than the caller") {
|
||||
@Override
|
||||
public void check(JavaClass service, ConditionEvents events) {
|
||||
for (JavaMethod entry : service.getMethods()) {
|
||||
Optional<Need> caller = effectiveNeed(entry);
|
||||
if (caller.isEmpty() || entry.getModifiers().contains(JavaModifier.PRIVATE)
|
||||
|| entry.getModifiers().contains(JavaModifier.STATIC)) {
|
||||
continue;
|
||||
}
|
||||
Set<JavaCodeUnit> seen = new HashSet<>(List.of(entry));
|
||||
Deque<JavaCodeUnit> toVisit = new ArrayDeque<>(seen);
|
||||
while (!toVisit.isEmpty()) {
|
||||
for (MethodAccess access : methodAccesses(toVisit.pop())) {
|
||||
JavaMethod target = access.target();
|
||||
Optional<Need> callee = target.getOwner().equals(service) ? Optional.empty()
|
||||
: effectiveNeed(target);
|
||||
if (callee.isEmpty()) {
|
||||
if (target.getOwner().getPackageName().startsWith(APP_PACKAGE) && seen.add(target)) {
|
||||
toVisit.push(target);
|
||||
}
|
||||
} else if (!callee.get().isMetBy(caller.get())) {
|
||||
events.add(SimpleConditionEvent.violated(access.access(), entry.getName() + " ("
|
||||
+ caller.get() + ") calls " + access + ", which needs " + callee.get()));
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
})
|
||||
.because("the inner check would refuse a caller the outer check let in");
|
||||
}
|
||||
@@ -0,0 +1,100 @@
|
||||
package auth_service.architecture;
|
||||
|
||||
import com.tngtech.archunit.core.domain.JavaClasses;
|
||||
import com.tngtech.archunit.core.importer.ClassFileImporter;
|
||||
import com.tngtech.archunit.lang.ArchRule;
|
||||
import com.tngtech.archunit.lang.EvaluationResult;
|
||||
import auth_service.architecture.fixture.api.BadController;
|
||||
import auth_service.architecture.fixture.api.GoodController;
|
||||
import auth_service.architecture.fixture.api.MethodReferenceController;
|
||||
import auth_service.architecture.fixture.service.ForwardingService;
|
||||
import auth_service.architecture.fixture.service.MarkedService;
|
||||
import auth_service.architecture.fixture.service.PublicCallerService;
|
||||
import auth_service.architecture.fixture.service.PublicViaUnmarkedCallerService;
|
||||
import auth_service.architecture.fixture.service.PublicHelperCallerService;
|
||||
import auth_service.architecture.fixture.service.PublicMethodReferenceCallerService;
|
||||
import auth_service.architecture.fixture.service.TwoMarkersService;
|
||||
import auth_service.architecture.fixture.service.UnmarkedService;
|
||||
import auth_service.architecture.fixture.service.UserCallerService;
|
||||
import auth_service.shared.test.AbstractArchitectureTest;
|
||||
import org.junit.jupiter.api.Test;
|
||||
|
||||
import java.util.List;
|
||||
|
||||
import static org.assertj.core.api.Assertions.assertThat;
|
||||
|
||||
class RoleAnnotationRulesTest extends AbstractArchitectureTest {
|
||||
|
||||
private static List<String> violations(ArchRule rule, Class<?>... classes) {
|
||||
JavaClasses imported = new ClassFileImporter().importClasses(classes);
|
||||
EvaluationResult result = rule.evaluate(imported);
|
||||
return result.getFailureReport().getDetails();
|
||||
}
|
||||
|
||||
@Test
|
||||
void controllerRule_PassesMarkedMethodsTheMethodMarkerAndStaticHelpers() {
|
||||
assertThat(violations(RoleAnnotationRules.controllers_call_only_marked_service_methods,
|
||||
GoodController.class, MarkedService.class, UnmarkedService.class)).isEmpty();
|
||||
}
|
||||
|
||||
@Test
|
||||
void controllerRule_FindsAnUnmarkedAndAnAsyncMethod() {
|
||||
List<String> found = violations(RoleAnnotationRules.controllers_call_only_marked_service_methods,
|
||||
BadController.class, MarkedService.class, UnmarkedService.class);
|
||||
|
||||
assertThat(found).hasSize(2);
|
||||
assertThat(found).anyMatch(line -> line.contains("no role marker on UnmarkedService.bare"));
|
||||
assertThat(found).anyMatch(line -> line.contains("@Async method MarkedService.later"));
|
||||
}
|
||||
|
||||
@Test
|
||||
void markerRule_FindsTwoMarkersOnOneMethod() {
|
||||
assertThat(violations(RoleAnnotationRules.one_role_marker_per_element, TwoMarkersService.class))
|
||||
.singleElement().asString().contains("both");
|
||||
}
|
||||
|
||||
@Test
|
||||
void escalationRule_FindsAPublicMethodCallingAUserMethod() {
|
||||
List<String> found = violations(RoleAnnotationRules.services_never_call_a_method_that_needs_more,
|
||||
PublicCallerService.class, MarkedService.class);
|
||||
|
||||
assertThat(found).singleElement().asString()
|
||||
.contains("callsUser (@Public)", "MarkedService.plain", "needs @RequiresRole(USER)");
|
||||
}
|
||||
|
||||
@Test
|
||||
void escalationRule_FindsAUserMethodCallingAnAuditorMethod() {
|
||||
assertThat(violations(RoleAnnotationRules.services_never_call_a_method_that_needs_more,
|
||||
UserCallerService.class, MarkedService.class))
|
||||
.singleElement().asString().contains("needs @RequiresRole(AUDITOR)");
|
||||
}
|
||||
|
||||
@Test
|
||||
void controllerRule_FindsAnUnmarkedMethodCalledByMethodReference() {
|
||||
assertThat(violations(RoleAnnotationRules.controllers_call_only_marked_service_methods,
|
||||
MethodReferenceController.class, UnmarkedService.class))
|
||||
.anyMatch(line -> line.contains("no role marker on UnmarkedService.bare"));
|
||||
}
|
||||
|
||||
@Test
|
||||
void escalationRule_FindsAPublicMethodCallingAUserMethodThroughAPrivateHelper() {
|
||||
assertThat(violations(RoleAnnotationRules.services_never_call_a_method_that_needs_more,
|
||||
PublicHelperCallerService.class, MarkedService.class))
|
||||
.anyMatch(line -> line.contains("MarkedService.plain") && line.contains("needs @RequiresRole(USER)"));
|
||||
}
|
||||
|
||||
@Test
|
||||
void escalationRule_FindsAPublicMethodCallingAUserMethodByMethodReference() {
|
||||
assertThat(violations(RoleAnnotationRules.services_never_call_a_method_that_needs_more,
|
||||
PublicMethodReferenceCallerService.class, MarkedService.class))
|
||||
.anyMatch(line -> line.contains("MarkedService.plain") && line.contains("needs @RequiresRole(USER)"));
|
||||
}
|
||||
|
||||
@Test
|
||||
void escalationRule_FindsAPublicMethodReachingAUserMethodThroughAnUnmarkedService() {
|
||||
assertThat(violations(RoleAnnotationRules.services_never_call_a_method_that_needs_more,
|
||||
PublicViaUnmarkedCallerService.class, ForwardingService.class, MarkedService.class))
|
||||
.singleElement().asString()
|
||||
.contains("callsThroughAnUnmarkedService (@Public)", "MarkedService.plain", "needs @RequiresRole(USER)");
|
||||
}
|
||||
}
|
||||
@@ -0,0 +1,15 @@
|
||||
package auth_service.architecture.fixture.api;
|
||||
|
||||
import auth_service.architecture.fixture.service.MarkedService;
|
||||
import auth_service.architecture.fixture.service.UnmarkedService;
|
||||
|
||||
public class BadController {
|
||||
|
||||
private final UnmarkedService unmarkedService = new UnmarkedService();
|
||||
private final MarkedService markedService = new MarkedService();
|
||||
|
||||
public void handle() {
|
||||
unmarkedService.bare();
|
||||
markedService.later();
|
||||
}
|
||||
}
|
||||
@@ -0,0 +1,15 @@
|
||||
package auth_service.architecture.fixture.api;
|
||||
|
||||
import auth_service.architecture.fixture.service.MarkedService;
|
||||
import auth_service.architecture.fixture.service.UnmarkedService;
|
||||
|
||||
public class GoodController {
|
||||
|
||||
private final MarkedService markedService = new MarkedService();
|
||||
|
||||
public void handle() {
|
||||
markedService.plain();
|
||||
markedService.open();
|
||||
UnmarkedService.helper();
|
||||
}
|
||||
}
|
||||
@@ -0,0 +1,13 @@
|
||||
package auth_service.architecture.fixture.api;
|
||||
|
||||
import auth_service.architecture.fixture.service.UnmarkedService;
|
||||
|
||||
public class MethodReferenceController {
|
||||
|
||||
private final UnmarkedService unmarkedService = new UnmarkedService();
|
||||
|
||||
public void handle() {
|
||||
Runnable call = unmarkedService::bare;
|
||||
call.run();
|
||||
}
|
||||
}
|
||||
@@ -0,0 +1,10 @@
|
||||
package auth_service.architecture.fixture.service;
|
||||
|
||||
public class ForwardingService {
|
||||
|
||||
private final MarkedService markedService = new MarkedService();
|
||||
|
||||
public void forward() {
|
||||
markedService.plain();
|
||||
}
|
||||
}
|
||||
@@ -0,0 +1,25 @@
|
||||
package auth_service.architecture.fixture.service;
|
||||
|
||||
import auth_service.shared.security.Public;
|
||||
import auth_service.shared.security.RequiresRole;
|
||||
import auth_service.shared.security.Role;
|
||||
import org.springframework.scheduling.annotation.Async;
|
||||
|
||||
@RequiresRole(Role.USER)
|
||||
public class MarkedService {
|
||||
|
||||
public void plain() {
|
||||
}
|
||||
|
||||
@Public
|
||||
public void open() {
|
||||
}
|
||||
|
||||
@RequiresRole(Role.AUDITOR)
|
||||
public void audit() {
|
||||
}
|
||||
|
||||
@Async
|
||||
public void later() {
|
||||
}
|
||||
}
|
||||
@@ -0,0 +1,17 @@
|
||||
package auth_service.architecture.fixture.service;
|
||||
|
||||
import auth_service.shared.security.Public;
|
||||
|
||||
@Public
|
||||
public class PublicCallerService {
|
||||
|
||||
private final MarkedService markedService = new MarkedService();
|
||||
|
||||
public void callsOpen() {
|
||||
markedService.open();
|
||||
}
|
||||
|
||||
public void callsUser() {
|
||||
markedService.plain();
|
||||
}
|
||||
}
|
||||
+20
@@ -0,0 +1,20 @@
|
||||
package auth_service.architecture.fixture.service;
|
||||
|
||||
import auth_service.shared.security.Public;
|
||||
import auth_service.shared.security.RequiresRole;
|
||||
import auth_service.shared.security.Role;
|
||||
|
||||
@RequiresRole(Role.USER)
|
||||
public class PublicHelperCallerService {
|
||||
|
||||
private final MarkedService markedService = new MarkedService();
|
||||
|
||||
@Public
|
||||
public void open() {
|
||||
helper();
|
||||
}
|
||||
|
||||
private void helper() {
|
||||
markedService.plain();
|
||||
}
|
||||
}
|
||||
+17
@@ -0,0 +1,17 @@
|
||||
package auth_service.architecture.fixture.service;
|
||||
|
||||
import auth_service.shared.security.Public;
|
||||
import auth_service.shared.security.RequiresRole;
|
||||
import auth_service.shared.security.Role;
|
||||
|
||||
@RequiresRole(Role.USER)
|
||||
public class PublicMethodReferenceCallerService {
|
||||
|
||||
private final MarkedService markedService = new MarkedService();
|
||||
|
||||
@Public
|
||||
public void open() {
|
||||
Runnable call = markedService::plain;
|
||||
call.run();
|
||||
}
|
||||
}
|
||||
+13
@@ -0,0 +1,13 @@
|
||||
package auth_service.architecture.fixture.service;
|
||||
|
||||
import auth_service.shared.security.Public;
|
||||
|
||||
@Public
|
||||
public class PublicViaUnmarkedCallerService {
|
||||
|
||||
private final ForwardingService forwardingService = new ForwardingService();
|
||||
|
||||
public void callsThroughAnUnmarkedService() {
|
||||
forwardingService.forward();
|
||||
}
|
||||
}
|
||||
@@ -0,0 +1,13 @@
|
||||
package auth_service.architecture.fixture.service;
|
||||
|
||||
import auth_service.shared.security.Public;
|
||||
import auth_service.shared.security.RequiresRole;
|
||||
import auth_service.shared.security.Role;
|
||||
|
||||
public class TwoMarkersService {
|
||||
|
||||
@Public
|
||||
@RequiresRole(Role.USER)
|
||||
public void both() {
|
||||
}
|
||||
}
|
||||
@@ -0,0 +1,11 @@
|
||||
package auth_service.architecture.fixture.service;
|
||||
|
||||
public class UnmarkedService {
|
||||
|
||||
public void bare() {
|
||||
}
|
||||
|
||||
public static boolean helper() {
|
||||
return true;
|
||||
}
|
||||
}
|
||||
@@ -0,0 +1,14 @@
|
||||
package auth_service.architecture.fixture.service;
|
||||
|
||||
import auth_service.shared.security.RequiresRole;
|
||||
import auth_service.shared.security.Role;
|
||||
|
||||
@RequiresRole(Role.USER)
|
||||
public class UserCallerService {
|
||||
|
||||
private final MarkedService markedService = new MarkedService();
|
||||
|
||||
public void callsAuditor() {
|
||||
markedService.audit();
|
||||
}
|
||||
}
|
||||
@@ -0,0 +1,100 @@
|
||||
package auth_service.configuration.security;
|
||||
|
||||
import auth_service.shared.security.Public;
|
||||
import auth_service.shared.security.RequiresRole;
|
||||
import auth_service.shared.security.Role;
|
||||
import auth_service.shared.test.AbstractUnitTest;
|
||||
import org.aopalliance.intercept.MethodInvocation;
|
||||
import org.junit.jupiter.api.AfterEach;
|
||||
import org.junit.jupiter.api.Test;
|
||||
import org.springframework.mock.web.MockHttpServletRequest;
|
||||
import org.springframework.security.authentication.AnonymousAuthenticationToken;
|
||||
import org.springframework.security.authentication.UsernamePasswordAuthenticationToken;
|
||||
import org.springframework.security.core.Authentication;
|
||||
import org.springframework.security.core.authority.SimpleGrantedAuthority;
|
||||
import org.springframework.web.context.request.RequestContextHolder;
|
||||
import org.springframework.web.context.request.ServletRequestAttributes;
|
||||
|
||||
import java.util.Arrays;
|
||||
import java.util.List;
|
||||
|
||||
import static org.junit.jupiter.api.Assertions.assertFalse;
|
||||
import static org.junit.jupiter.api.Assertions.assertTrue;
|
||||
import static org.mockito.Mockito.lenient;
|
||||
import static org.mockito.Mockito.mock;
|
||||
|
||||
class RoleAuthorizationManagerTest extends AbstractUnitTest {
|
||||
|
||||
private final RoleAuthorizationManager manager = new RoleAuthorizationManager();
|
||||
|
||||
@RequiresRole(Role.USER)
|
||||
static class UserMarked {
|
||||
public void plain() {
|
||||
}
|
||||
|
||||
@RequiresRole(Role.AUDITOR)
|
||||
public void audit() {
|
||||
}
|
||||
|
||||
@Public
|
||||
public void open() {
|
||||
}
|
||||
}
|
||||
|
||||
@AfterEach
|
||||
void clearRequest() {
|
||||
RequestContextHolder.resetRequestAttributes();
|
||||
}
|
||||
|
||||
private static void inRequest() {
|
||||
RequestContextHolder.setRequestAttributes(new ServletRequestAttributes(new MockHttpServletRequest()));
|
||||
}
|
||||
|
||||
// The authorities JwtAuthenticationFilter grants: the role and every role below it.
|
||||
private static Authentication login(Role role) {
|
||||
return new UsernamePasswordAuthenticationToken("42", null, Arrays.stream(Role.values()).filter(role::atLeast)
|
||||
.map(each -> new SimpleGrantedAuthority("ROLE_" + each.name())).toList());
|
||||
}
|
||||
|
||||
private static Authentication anonymous() {
|
||||
return new AnonymousAuthenticationToken("key", "anonymousUser", List.of(new SimpleGrantedAuthority("ROLE_ANONYMOUS")));
|
||||
}
|
||||
|
||||
private boolean granted(Authentication authentication, Object target, String method) throws Exception {
|
||||
MethodInvocation invocation = mock(MethodInvocation.class);
|
||||
// Lenient: without a request the manager does not look at the invocation.
|
||||
lenient().when(invocation.getThis()).thenReturn(target);
|
||||
lenient().when(invocation.getMethod()).thenReturn(target.getClass().getMethod(method));
|
||||
return manager.authorize(() -> authentication, invocation).isGranted();
|
||||
}
|
||||
|
||||
@Test
|
||||
void classMarker_CoversTheMethodsWithoutOwnMarker() throws Exception {
|
||||
inRequest();
|
||||
|
||||
assertTrue(granted(login(Role.USER), new UserMarked(), "plain"));
|
||||
assertFalse(granted(anonymous(), new UserMarked(), "plain"));
|
||||
}
|
||||
|
||||
@Test
|
||||
void methodMarker_ReplacesTheClassMarker() throws Exception {
|
||||
inRequest();
|
||||
|
||||
assertFalse(granted(login(Role.USER), new UserMarked(), "audit"));
|
||||
assertTrue(granted(anonymous(), new UserMarked(), "open"));
|
||||
}
|
||||
|
||||
@Test
|
||||
void higherRoles_PassLowerChecks() throws Exception {
|
||||
inRequest();
|
||||
|
||||
assertTrue(granted(login(Role.AUDITOR), new UserMarked(), "audit"));
|
||||
assertTrue(granted(login(Role.PLATFORM_ADMIN), new UserMarked(), "audit"));
|
||||
assertTrue(granted(login(Role.PLATFORM_ADMIN), new UserMarked(), "plain"));
|
||||
}
|
||||
|
||||
@Test
|
||||
void withoutARequest_EverythingRuns() throws Exception {
|
||||
assertTrue(granted(null, new UserMarked(), "audit"));
|
||||
}
|
||||
}
|
||||
@@ -0,0 +1,131 @@
|
||||
package auth_service.configuration.security;
|
||||
|
||||
import auth_service.shared.security.Public;
|
||||
import auth_service.shared.security.RequiresRole;
|
||||
import auth_service.shared.security.Role;
|
||||
import auth_service.shared.security.TokenGenerator;
|
||||
import auth_service.shared.api.AuthCookies;
|
||||
import auth_service.shared.test.AbstractIntegrationTest;
|
||||
import jakarta.servlet.http.Cookie;
|
||||
import org.junit.jupiter.api.Test;
|
||||
import org.springframework.beans.factory.annotation.Autowired;
|
||||
import org.springframework.boot.webmvc.test.autoconfigure.AutoConfigureMockMvc;
|
||||
import org.springframework.context.annotation.Import;
|
||||
import org.springframework.stereotype.Service;
|
||||
import org.springframework.test.web.servlet.MockMvc;
|
||||
import org.springframework.test.web.servlet.ResultActions;
|
||||
import org.springframework.test.web.servlet.request.MockHttpServletRequestBuilder;
|
||||
import org.springframework.web.bind.annotation.GetMapping;
|
||||
import org.springframework.web.bind.annotation.RestController;
|
||||
|
||||
import static org.junit.jupiter.api.Assertions.assertEquals;
|
||||
import static org.springframework.test.web.servlet.request.MockMvcRequestBuilders.get;
|
||||
import static org.springframework.test.web.servlet.result.MockMvcResultMatchers.status;
|
||||
|
||||
/**
|
||||
* The role markers through the real filter chain, on a probe that only this test sees.
|
||||
*/
|
||||
@AutoConfigureMockMvc
|
||||
@Import({RoleCheckFlowTest.ProbeService.class, RoleCheckFlowTest.ProbeController.class})
|
||||
class RoleCheckFlowTest extends AbstractIntegrationTest {
|
||||
|
||||
@Service
|
||||
@RequiresRole(Role.USER)
|
||||
static class ProbeService {
|
||||
public String user() {
|
||||
return "user";
|
||||
}
|
||||
|
||||
@RequiresRole(Role.AUDITOR)
|
||||
public String auditor() {
|
||||
return "auditor";
|
||||
}
|
||||
|
||||
@Public
|
||||
public String open() {
|
||||
return "open";
|
||||
}
|
||||
}
|
||||
|
||||
@RestController
|
||||
static class ProbeController {
|
||||
private final ProbeService probeService;
|
||||
|
||||
ProbeController(ProbeService probeService) {
|
||||
this.probeService = probeService;
|
||||
}
|
||||
|
||||
@GetMapping("/api/role-probe/user")
|
||||
String user() {
|
||||
return probeService.user();
|
||||
}
|
||||
|
||||
@GetMapping("/api/role-probe/auditor")
|
||||
String auditor() {
|
||||
return probeService.auditor();
|
||||
}
|
||||
|
||||
// Authenticated URL: the marker opens nothing the URL rules close.
|
||||
@GetMapping("/api/role-probe/open")
|
||||
String open() {
|
||||
return probeService.open();
|
||||
}
|
||||
|
||||
// Public URL: only the marker stops an anonymous caller.
|
||||
@GetMapping("/api/public/role-probe/user")
|
||||
String publicUser() {
|
||||
return probeService.user();
|
||||
}
|
||||
|
||||
@GetMapping("/api/public/role-probe/open")
|
||||
String publicOpen() {
|
||||
return probeService.open();
|
||||
}
|
||||
}
|
||||
|
||||
@Autowired
|
||||
private MockMvc mockMvc;
|
||||
|
||||
@Autowired
|
||||
private ProbeService probeService;
|
||||
|
||||
@Autowired
|
||||
private TokenGenerator tokenGenerator;
|
||||
|
||||
private String token(Role role) {
|
||||
return tokenGenerator.generateToken("4711", "probe@example.com", null, role);
|
||||
}
|
||||
|
||||
private ResultActions call(String path, Role role) throws Exception {
|
||||
MockHttpServletRequestBuilder request = get("/api/role-probe/" + path);
|
||||
if (role != null) {
|
||||
request.cookie(new Cookie(AuthCookies.ACCESS_COOKIE, token(role)));
|
||||
}
|
||||
return mockMvc.perform(request);
|
||||
}
|
||||
|
||||
@Test
|
||||
void user_PassesUserAndIsRefusedAuditor() throws Exception {
|
||||
call("user", Role.USER).andExpect(status().isOk());
|
||||
call("open", Role.USER).andExpect(status().isOk());
|
||||
call("auditor", Role.USER).andExpect(status().isForbidden());
|
||||
}
|
||||
|
||||
@Test
|
||||
void higherRoles_PassTheLowerChecks() throws Exception {
|
||||
call("auditor", Role.AUDITOR).andExpect(status().isOk());
|
||||
call("auditor", Role.PLATFORM_ADMIN).andExpect(status().isOk());
|
||||
call("user", Role.PLATFORM_ADMIN).andExpect(status().isOk());
|
||||
}
|
||||
|
||||
@Test
|
||||
void noLogin_OnAPublicUrl_IsStoppedByTheMarker() throws Exception {
|
||||
mockMvc.perform(get("/api/public/role-probe/user")).andExpect(status().isUnauthorized());
|
||||
mockMvc.perform(get("/api/public/role-probe/open")).andExpect(status().isOk());
|
||||
}
|
||||
|
||||
@Test
|
||||
void outsideARequest_TheServiceRuns() {
|
||||
assertEquals("auditor", probeService.auditor());
|
||||
}
|
||||
}
|
||||
@@ -0,0 +1,39 @@
|
||||
package auth_service.modules.account.service;
|
||||
|
||||
import auth_service.shared.account.AccountDeleter;
|
||||
import auth_service.shared.test.AbstractIntegrationTest;
|
||||
import org.junit.jupiter.api.AfterEach;
|
||||
import org.junit.jupiter.api.Test;
|
||||
import org.springframework.beans.factory.annotation.Autowired;
|
||||
import org.springframework.mock.web.MockHttpServletRequest;
|
||||
import org.springframework.security.authentication.UsernamePasswordAuthenticationToken;
|
||||
import org.springframework.security.authorization.AuthorizationDeniedException;
|
||||
import org.springframework.security.core.authority.SimpleGrantedAuthority;
|
||||
import org.springframework.security.core.context.SecurityContextHolder;
|
||||
import org.springframework.web.context.request.RequestContextHolder;
|
||||
import org.springframework.web.context.request.ServletRequestAttributes;
|
||||
|
||||
import java.util.List;
|
||||
|
||||
import static org.junit.jupiter.api.Assertions.assertThrows;
|
||||
|
||||
class AccountServiceRoleTest extends AbstractIntegrationTest {
|
||||
|
||||
@Autowired
|
||||
private AccountDeleter accountDeleter;
|
||||
|
||||
@AfterEach
|
||||
void clear() {
|
||||
RequestContextHolder.resetRequestAttributes();
|
||||
SecurityContextHolder.clearContext();
|
||||
}
|
||||
|
||||
@Test
|
||||
void deleteByAdmin_RefusesAUserLogin() {
|
||||
RequestContextHolder.setRequestAttributes(new ServletRequestAttributes(new MockHttpServletRequest()));
|
||||
SecurityContextHolder.getContext().setAuthentication(new UsernamePasswordAuthenticationToken("42", null,
|
||||
List.of(new SimpleGrantedAuthority("ROLE_USER"))));
|
||||
|
||||
assertThrows(AuthorizationDeniedException.class, () -> accountDeleter.deleteByAdmin(999_999L));
|
||||
}
|
||||
}
|
||||
Reference in new issue
Block a user