xds: handle the handlerRemoved callback to skip updateSslContext processing (#10118)

* xds: handle the handlerRemoved callback to skip updateSslContext processing
     In handlerAdded we submit a callback to updateSslContext but before the
     callback is executed the handler could be removed (e.g. bad connection)
     in which case the callback should skip all of the processing.
     Also added a unit test to check there is no exception.
This commit is contained in:
sanjaypujare 2023-05-03 20:11:15 +05:30 committed by GitHub
parent 545e982afd
commit f24a5a7d0d
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 33 additions and 0 deletions

View File

@ -209,6 +209,9 @@ public final class SecurityProtocolNegotiators {
@Override @Override
public void updateSslContext(SslContext sslContext) { public void updateSslContext(SslContext sslContext) {
if (ctx.isRemoved()) {
return;
}
logger.log( logger.log(
Level.FINEST, Level.FINEST,
"ClientSdsHandler.updateSslContext authority={0}, ctx.name={1}", "ClientSdsHandler.updateSslContext authority={0}, ctx.name={1}",

View File

@ -405,6 +405,36 @@ public class SecurityProtocolNegotiatorsTest {
CommonCertProviderTestUtils.register0(); CommonCertProviderTestUtils.register0();
} }
@Test
public void clientSdsProtocolNegotiatorNewHandler_handleHandlerRemoved() {
FakeClock executor = new FakeClock();
CommonCertProviderTestUtils.register(executor);
Bootstrapper.BootstrapInfo bootstrapInfoForClient = CommonBootstrapperTestUtils
.buildBootstrapInfo("google_cloud_private_spiffe-client", CLIENT_KEY_FILE, CLIENT_PEM_FILE,
CA_PEM_FILE, null, null, null, null);
UpstreamTlsContext upstreamTlsContext =
CommonTlsContextTestsUtil
.buildUpstreamTlsContext("google_cloud_private_spiffe-client", true);
SslContextProviderSupplier sslContextProviderSupplier =
new SslContextProviderSupplier(upstreamTlsContext,
new TlsContextManagerImpl(bootstrapInfoForClient));
SecurityProtocolNegotiators.ClientSdsHandler clientSdsHandler =
new SecurityProtocolNegotiators.ClientSdsHandler(grpcHandler, sslContextProviderSupplier);
pipeline.addLast(clientSdsHandler);
channelHandlerCtx = pipeline.context(clientSdsHandler);
// kick off protocol negotiation.
pipeline.fireUserEventTriggered(InternalProtocolNegotiationEvent.getDefault());
executor.runDueTasks();
pipeline.remove(clientSdsHandler);
channel.runPendingTasks();
channel.checkException();
CommonCertProviderTestUtils.register0();
}
private static final class FakeGrpcHttp2ConnectionHandler extends GrpcHttp2ConnectionHandler { private static final class FakeGrpcHttp2ConnectionHandler extends GrpcHttp2ConnectionHandler {
FakeGrpcHttp2ConnectionHandler( FakeGrpcHttp2ConnectionHandler(