Skip to content

Commit

Permalink
Grpc: Use eventloop for server calls to avoid hanging
Browse files Browse the repository at this point in the history
  • Loading branch information
@michalvavrik
michalvavrik committed Jul 5, 2023
1 parent 8a06248 commit 1788293
Show file tree
Hide file tree
Showing 5 changed files with 376 additions and 215 deletions.
32 changes: 32 additions & 0 deletions ...nsions/grpc/deployment/src/test/java/io/quarkus/grpc/auth/BlockingHttpSecurityPolicy.java
View file
Original file line number Diff line number Diff line change
@@ -0,0 +1,32 @@
package io.quarkus.grpc.auth;

import java.util.function.BiFunction;

import jakarta.inject.Singleton;

import io.quarkus.security.identity.SecurityIdentity;
import io.quarkus.vertx.http.runtime.security.HttpSecurityPolicy;
import io.smallrye.mutiny.Uni;
import io.vertx.ext.web.RoutingContext;

@Singleton
public class BlockingHttpSecurityPolicy implements HttpSecurityPolicy {

final static String BLOCK_REQUEST = "block-request";

@Override
public Uni<CheckResult> checkPermission(RoutingContext request, Uni<SecurityIdentity> identity,
AuthorizationRequestContext requestContext) {
if (request.request().headers().get(BLOCK_REQUEST) != null) {
return requestContext.runBlocking(request, identity,
new BiFunction<RoutingContext, SecurityIdentity, CheckResult>() {
@Override
public CheckResult apply(RoutingContext routingContext, SecurityIdentity securityIdentity) {
return CheckResult.PERMIT;
}
});
} else {
return Uni.createFrom().item(CheckResult.PERMIT);
}
}
}
217 changes: 2 additions & 215 deletions extensions/grpc/deployment/src/test/java/io/quarkus/grpc/auth/GrpcAuthTest.java
View file
Original file line number Diff line number Diff line change
@@ -1,225 +1,12 @@
package io.quarkus.grpc.auth;

import static com.example.security.Security.ThreadInfo.newBuilder;
import static org.assertj.core.api.Assertions.assertThat;
import static org.awaitility.Awaitility.await;

import java.nio.charset.StandardCharsets;
import java.util.Base64;
import java.util.List;
import java.util.concurrent.CopyOnWriteArrayList;
import java.util.concurrent.TimeUnit;
import java.util.concurrent.atomic.AtomicInteger;
import java.util.concurrent.atomic.AtomicReference;

import jakarta.annotation.security.RolesAllowed;
import jakarta.inject.Singleton;

import org.jboss.shrinkwrap.api.ShrinkWrap;
import org.jboss.shrinkwrap.api.asset.StringAsset;
import org.jboss.shrinkwrap.api.spec.JavaArchive;
import org.junit.jupiter.api.Assertions;
import org.junit.jupiter.api.Test;
import org.junit.jupiter.api.extension.RegisterExtension;

import com.example.security.SecuredService;
import com.example.security.Security;

import io.grpc.Metadata;
import io.quarkus.grpc.GrpcClient;
import io.quarkus.grpc.GrpcClientUtils;
import io.quarkus.grpc.GrpcService;
import io.quarkus.security.credential.PasswordCredential;
import io.quarkus.security.identity.request.AuthenticationRequest;
import io.quarkus.security.identity.request.UsernamePasswordAuthenticationRequest;
import io.quarkus.test.QuarkusUnitTest;
import io.smallrye.common.annotation.Blocking;
import io.smallrye.mutiny.Multi;
import io.smallrye.mutiny.Uni;
import io.vertx.core.Context;

public class GrpcAuthTest {

public static final Metadata.Key<String> AUTHORIZATION = Metadata.Key.of("Authorization",
Metadata.ASCII_STRING_MARSHALLER);
public class GrpcAuthTest extends GrpcAuthTestBase {

@RegisterExtension
static final QuarkusUnitTest config = new QuarkusUnitTest().setArchiveProducer(
() -> ShrinkWrap.create(JavaArchive.class)
.addClasses(Service.class, BasicGrpcSecurityMechanism.class)
.addPackage(SecuredService.class.getPackage())
.add(new StringAsset("quarkus.security.users.embedded.enabled=true\n" +
"quarkus.security.users.embedded.users.john=john\n" +
"quarkus.security.users.embedded.roles.john=employees\n" +
"quarkus.security.users.embedded.users.paul=paul\n" +
"quarkus.security.users.embedded.roles.paul=interns\n" +
"quarkus.security.users.embedded.plain-text=true\n" +
"quarkus.http.auth.basic=true"), "application.properties"));
public static final String JOHN_BASIC_CREDS = "am9objpqb2hu";
public static final String PAUL_BASIC_CREDS = "cGF1bDpwYXVs";

@GrpcClient
SecuredService securityClient;

@Test
void shouldSecureUniEndpoint() {
Metadata headers = new Metadata();
headers.put(AUTHORIZATION, "Basic " + JOHN_BASIC_CREDS);
SecuredService client = GrpcClientUtils.attachHeaders(securityClient, headers);
AtomicInteger resultCount = new AtomicInteger();
client.unaryCall(Security.Container.newBuilder().setText("woo-hoo").build())
.subscribe().with(e -> {
if (!e.getIsOnEventLoop()) {
Assertions.fail("Secured method should be run on event loop");
}
resultCount.incrementAndGet();
});

await().atMost(10, TimeUnit.SECONDS)
.until(() -> resultCount.get() == 1);
}

@Test
void shouldSecureBlockingUniEndpoint() {
Metadata headers = new Metadata();
headers.put(AUTHORIZATION, "Basic " + JOHN_BASIC_CREDS);
SecuredService client = GrpcClientUtils.attachHeaders(securityClient, headers);
AtomicInteger resultCount = new AtomicInteger();
client.unaryCallBlocking(Security.Container.newBuilder().setText("woo-hoo").build())
.subscribe().with(e -> {
if (e.getIsOnEventLoop()) {
Assertions.fail("Secured method annotated with @Blocking should be executed on worker thread");
}
resultCount.incrementAndGet();
});

await().atMost(10, TimeUnit.SECONDS)
.until(() -> resultCount.get() == 1);
}

@Test
void shouldSecureMultiEndpoint() {
Metadata headers = new Metadata();
headers.put(AUTHORIZATION, "Basic " + PAUL_BASIC_CREDS);
SecuredService client = GrpcClientUtils.attachHeaders(securityClient, headers);
List<Boolean> results = new CopyOnWriteArrayList<>();
client.streamCall(Multi.createBy().repeating()
.supplier(() -> (Security.Container.newBuilder().setText("woo-hoo").build())).atMost(4))
.subscribe().with(e -> results.add(e.getIsOnEventLoop()));

await().atMost(10, TimeUnit.SECONDS)
.until(() -> results.size() == 5);

assertThat(results.stream().filter(e -> !e)).isEmpty();
}

@Test
void shouldSecureBlockingMultiEndpoint() {
Metadata headers = new Metadata();
headers.put(AUTHORIZATION, "Basic " + PAUL_BASIC_CREDS);
SecuredService client = GrpcClientUtils.attachHeaders(securityClient, headers);
List<Boolean> results = new CopyOnWriteArrayList<>();
client.streamCallBlocking(Multi.createBy().repeating()
.supplier(() -> (Security.Container.newBuilder().setText("woo-hoo").build())).atMost(4))
.subscribe().with(e -> results.add(e.getIsOnEventLoop()));

await().atMost(10, TimeUnit.SECONDS)
.until(() -> results.size() == 5);

assertThat(results.stream().filter(e -> e)).isEmpty();
}

@Test
void shouldFailWithInvalidCredentials() {
Metadata headers = new Metadata();
headers.put(AUTHORIZATION, "Basic invalid creds");
SecuredService client = GrpcClientUtils.attachHeaders(securityClient, headers);

AtomicReference<Throwable> error = new AtomicReference<>();

AtomicInteger resultCount = new AtomicInteger();
client.unaryCall(Security.Container.newBuilder().setText("woo-hoo").build())
.onFailure().invoke(error::set)
.subscribe().with(e -> resultCount.incrementAndGet());

await().atMost(10, TimeUnit.SECONDS)
.until(() -> error.get() != null);
}

@Test
void shouldFailWithInvalidInsufficientRole() {
Metadata headers = new Metadata();
headers.put(AUTHORIZATION, PAUL_BASIC_CREDS);
SecuredService client = GrpcClientUtils.attachHeaders(securityClient, headers);

AtomicReference<Throwable> error = new AtomicReference<>();

AtomicInteger resultCount = new AtomicInteger();
client.unaryCall(Security.Container.newBuilder().setText("woo-hoo").build())
.onFailure().invoke(error::set)
.subscribe().with(e -> resultCount.incrementAndGet());

await().atMost(10, TimeUnit.SECONDS)
.until(() -> error.get() != null);
}

@GrpcService
public static class Service implements SecuredService {
@Override
@RolesAllowed("employees")
public Uni<Security.ThreadInfo> unaryCall(Security.Container request) {
return Uni.createFrom()
.item(newBuilder().setIsOnEventLoop(Context.isOnEventLoopThread()).build());
}

@Override
@RolesAllowed("interns")
public Multi<Security.ThreadInfo> streamCall(Multi<Security.Container> request) {
return Multi.createBy()
.repeating().supplier(() -> newBuilder().setIsOnEventLoop(Context.isOnEventLoopThread()).build())
.atMost(5);
}

@Blocking
@Override
@RolesAllowed("employees")
public Uni<Security.ThreadInfo> unaryCallBlocking(Security.Container request) {
return Uni.createFrom()
.item(newBuilder().setIsOnEventLoop(Context.isOnEventLoopThread()).build());
}

@Blocking
@Override
@RolesAllowed("interns")
public Multi<Security.ThreadInfo> streamCallBlocking(Multi<Security.Container> request) {
return Multi.createBy()
.repeating().supplier(() -> newBuilder().setIsOnEventLoop(Context.isOnEventLoopThread()).build())
.atMost(5);
}
}

@Singleton
public static class BasicGrpcSecurityMechanism implements GrpcSecurityMechanism {
@Override
public boolean handles(Metadata metadata) {
String authString = metadata.get(AUTHORIZATION);
return authString != null && authString.startsWith("Basic ");
}
static final QuarkusUnitTest config = createQuarkusUnitTest(null);

@Override
public AuthenticationRequest createAuthenticationRequest(Metadata metadata) {
String authString = metadata.get(AUTHORIZATION);
authString = authString.substring("Basic ".length());
byte[] decode = Base64.getDecoder().decode(authString);
String plainChallenge = new String(decode, StandardCharsets.UTF_8);
int colonPos;
if ((colonPos = plainChallenge.indexOf(':')) > -1) {
String userName = plainChallenge.substring(0, colonPos);
char[] password = plainChallenge.substring(colonPos + 1).toCharArray();
return new UsernamePasswordAuthenticationRequest(userName, new PasswordCredential(password));
} else {
return null;
}
}
}
}
Loading

0 comments on commit 1788293

Please sign in to comment.