grpc
/
grpc-java
mirror of https://github.com/grpc/grpc-java.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

netty: Initialize ProtocolNegotiators eagerly 

This simplifies the construction paradigm and leads to the eventual
removal of TransportCreationParamsFilterFactory. The eventual end goal
is to be able to shut down ProtocolNegotiators as is necessary for ALTS.

The only reason the initialization was delayed was for 'authority', so
we now plumb the authority through GrpcHttp2ConnectionHandler.
This commit is contained in:
Eric Anderson 2018-08-31 14:49:20 -07:00
parent c8975e987b
commit 7efbc36c72
11 changed files with 191 additions and 175 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
alts/src/main/java/io/grpc/alts/GoogleDefaultChannelBuilder.java
View File

@ -144,7 +144,7 @@ public final class GoogleDefaultChannelBuilder
"%s must be a InetSocketAddress",
serverAddress);
final GoogleDefaultProtocolNegotiator negotiator =
new GoogleDefaultProtocolNegotiator(altsHandshakerFactory, sslContext, authority);
new GoogleDefaultProtocolNegotiator(altsHandshakerFactory, sslContext);
return new TransportCreationParamsFilter() {
@Override
public SocketAddress getTargetServerAddress() {

5
alts/src/main/java/io/grpc/alts/internal/GoogleDefaultProtocolNegotiator.java
View File

@ -28,10 +28,9 @@ public final class GoogleDefaultProtocolNegotiator implements ProtocolNegotiator
private final ProtocolNegotiator altsProtocolNegotiator;
private final ProtocolNegotiator tlsProtocolNegotiator;
public GoogleDefaultProtocolNegotiator(
TsiHandshakerFactory altsFactory, SslContext sslContext, String authority) {
public GoogleDefaultProtocolNegotiator(TsiHandshakerFactory altsFactory, SslContext sslContext) {
altsProtocolNegotiator = AltsProtocolNegotiator.create(altsFactory);
tlsProtocolNegotiator = ProtocolNegotiators.tls(sslContext, authority);
tlsProtocolNegotiator = ProtocolNegotiators.tls(sslContext);
}
@VisibleForTesting

9
netty/src/main/java/io/grpc/netty/GrpcHttp2ConnectionHandler.java
View File

@ -84,4 +84,13 @@ public abstract class GrpcHttp2ConnectionHandler extends Http2ConnectionHandler
public Attributes getEagAttributes() {
return Attributes.EMPTY;
}
/**
* Returns the authority of the server. Only available on the client-side.
*
* @throws UnsupportedOperationException if on server-side
*/
public String getAuthority() {
throw new UnsupportedOperationException();
}
}

167
netty/src/main/java/io/grpc/netty/NettyChannelBuilder.java
View File

@ -331,8 +331,25 @@ public final class NettyChannelBuilder
@CheckReturnValue
@Internal
protected ClientTransportFactory buildTransportFactory() {
return new NettyTransportFactory(dynamicParamsFactory, channelType, channelOptions,
negotiationType, sslContext, eventLoopGroup, flowControlWindow, maxInboundMessageSize(),
TransportCreationParamsFilterFactory transportCreationParamsFilterFactory =
dynamicParamsFactory;
if (transportCreationParamsFilterFactory == null) {
SslContext localSslContext = sslContext;
if (negotiationType == NegotiationType.TLS && localSslContext == null) {
try {
localSslContext = GrpcSslContexts.forClient().build();
} catch (SSLException ex) {
throw new RuntimeException(ex);
}
}
ProtocolNegotiator negotiator =
createProtocolNegotiatorByType(negotiationType, localSslContext);
transportCreationParamsFilterFactory =
new DefaultNettyTransportCreationParamsFilterFactory(negotiator);
}
return new NettyTransportFactory(
transportCreationParamsFilterFactory, channelType, channelOptions,
eventLoopGroup, flowControlWindow, maxInboundMessageSize(),
maxHeaderListSize, keepAliveTimeNanos, keepAliveTimeoutNanos, keepAliveWithoutCalls,
transportTracerFactory.create());
}
@ -362,23 +379,7 @@ public final class NettyChannelBuilder
@VisibleForTesting
@CheckReturnValue
static ProtocolNegotiator createProtocolNegotiator(
String authority,
NegotiationType negotiationType,
SslContext sslContext,
ProxyParameters proxy) {
ProtocolNegotiator negotiator =
createProtocolNegotiatorByType(authority, negotiationType, sslContext);
if (proxy != null) {
negotiator = ProtocolNegotiators.httpProxy(
proxy.proxyAddress, proxy.username, proxy.password, negotiator);
}
return negotiator;
}
@CheckReturnValue
private static ProtocolNegotiator createProtocolNegotiatorByType(
String authority,
static ProtocolNegotiator createProtocolNegotiatorByType(
NegotiationType negotiationType,
SslContext sslContext) {
switch (negotiationType) {
@ -387,7 +388,7 @@ public final class NettyChannelBuilder
case PLAINTEXT_UPGRADE:
return ProtocolNegotiators.plaintextUpgrade();
case TLS:
return ProtocolNegotiators.tls(sslContext, authority);
return ProtocolNegotiators.tls(sslContext);
default:
throw new IllegalArgumentException("Unsupported negotiationType: " + negotiationType);
}
@ -461,7 +462,6 @@ public final class NettyChannelBuilder
private final TransportCreationParamsFilterFactory transportCreationParamsFilterFactory;
private final Class<? extends Channel> channelType;
private final Map<ChannelOption<?>, ?> channelOptions;
private final NegotiationType negotiationType;
private final EventLoopGroup group;
private final boolean usingSharedGroup;
private final int flowControlWindow;
@ -476,27 +476,20 @@ public final class NettyChannelBuilder
NettyTransportFactory(TransportCreationParamsFilterFactory transportCreationParamsFilterFactory,
Class<? extends Channel> channelType, Map<ChannelOption<?>, ?> channelOptions,
NegotiationType negotiationType, SslContext sslContext, EventLoopGroup group,
int flowControlWindow, int maxMessageSize, int maxHeaderListSize,
EventLoopGroup group, int flowControlWindow, int maxMessageSize, int maxHeaderListSize,
long keepAliveTimeNanos, long keepAliveTimeoutNanos, boolean keepAliveWithoutCalls,
TransportTracer transportTracer) {
this.channelType = channelType;
this.negotiationType = negotiationType;
this.channelOptions = new HashMap<ChannelOption<?>, Object>(channelOptions);
this.transportTracer = transportTracer;
if (transportCreationParamsFilterFactory == null) {
transportCreationParamsFilterFactory =
new DefaultNettyTransportCreationParamsFilterFactory(sslContext);
}
this.transportCreationParamsFilterFactory = transportCreationParamsFilterFactory;
this.channelType = channelType;
this.channelOptions = new HashMap<ChannelOption<?>, Object>(channelOptions);
this.flowControlWindow = flowControlWindow;
this.maxMessageSize = maxMessageSize;
this.maxHeaderListSize = maxHeaderListSize;
this.keepAliveTimeNanos = new AtomicBackoff("keepalive time nanos", keepAliveTimeNanos);
this.keepAliveTimeoutNanos = keepAliveTimeoutNanos;
this.keepAliveWithoutCalls = keepAliveWithoutCalls;
this.transportTracer = transportTracer;
usingSharedGroup = group == null;
if (usingSharedGroup) {
// The group was unspecified, using the shared group.
@ -550,71 +543,69 @@ public final class NettyChannelBuilder
SharedResourceHolder.release(Utils.DEFAULT_WORKER_EVENT_LOOP_GROUP, group);
}
}
}
private final class DefaultNettyTransportCreationParamsFilterFactory
implements TransportCreationParamsFilterFactory {
private final SslContext sslContext;
private static final class DefaultNettyTransportCreationParamsFilterFactory
implements TransportCreationParamsFilterFactory {
final ProtocolNegotiator negotiator;
private DefaultNettyTransportCreationParamsFilterFactory(SslContext sslContext) {
if (negotiationType == NegotiationType.TLS && sslContext == null) {
try {
sslContext = GrpcSslContexts.forClient().build();
} catch (SSLException ex) {
throw new RuntimeException(ex);
}
}
this.sslContext = sslContext;
DefaultNettyTransportCreationParamsFilterFactory(ProtocolNegotiator negotiator) {
this.negotiator = negotiator;
}
@Override
public TransportCreationParamsFilter create(
SocketAddress targetServerAddress,
String authority,
String userAgent,
ProxyParameters proxyParams) {
ProtocolNegotiator localNegotiator = negotiator;
if (proxyParams != null) {
localNegotiator = ProtocolNegotiators.httpProxy(
proxyParams.proxyAddress, proxyParams.username, proxyParams.password, negotiator);
}
return new DynamicNettyTransportParams(
targetServerAddress, authority, userAgent, localNegotiator);
}
}
@Override
public TransportCreationParamsFilter create(
SocketAddress targetServerAddress,
String authority,
String userAgent,
ProxyParameters proxyParams) {
return new DynamicNettyTransportParams(
targetServerAddress, authority, userAgent, proxyParams);
}
@CheckReturnValue
private static final class DynamicNettyTransportParams implements TransportCreationParamsFilter {
@CheckReturnValue
private final class DynamicNettyTransportParams implements TransportCreationParamsFilter {
private final SocketAddress targetServerAddress;
private final String authority;
@Nullable private final String userAgent;
private final ProtocolNegotiator protocolNegotiator;
private final SocketAddress targetServerAddress;
private final String authority;
@Nullable private final String userAgent;
private ProxyParameters proxyParams;
private DynamicNettyTransportParams(
SocketAddress targetServerAddress,
String authority,
String userAgent,
ProtocolNegotiator protocolNegotiator) {
this.targetServerAddress = targetServerAddress;
this.authority = authority;
this.userAgent = userAgent;
this.protocolNegotiator = protocolNegotiator;
}
private DynamicNettyTransportParams(
SocketAddress targetServerAddress,
String authority,
String userAgent,
ProxyParameters proxyParams) {
this.targetServerAddress = targetServerAddress;
this.authority = authority;
this.userAgent = userAgent;
this.proxyParams = proxyParams;
}
@Override
public SocketAddress getTargetServerAddress() {
return targetServerAddress;
}
@Override
public SocketAddress getTargetServerAddress() {
return targetServerAddress;
}
@Override
public String getAuthority() {
return authority;
}
@Override
public String getAuthority() {
return authority;
}
@Override
public String getUserAgent() {
return userAgent;
}
@Override
public String getUserAgent() {
return userAgent;
}
@Override
public ProtocolNegotiator getProtocolNegotiator() {
return createProtocolNegotiator(authority, negotiationType, sslContext, proxyParams);
}
}
@Override
public ProtocolNegotiator getProtocolNegotiator() {
return protocolNegotiator;
}
}
}

23
netty/src/main/java/io/grpc/netty/NettyClientHandler.java
View File

@ -103,6 +103,7 @@ class NettyClientHandler extends AbstractNettyHandler {
private final Supplier<Stopwatch> stopwatchFactory;
private final TransportTracer transportTracer;
private final Attributes eagAttributes;
private final String authority;
private WriteQueue clientWriteQueue;
private Http2Ping ping;
private Attributes attributes = Attributes.EMPTY;
@ -116,7 +117,8 @@ class NettyClientHandler extends AbstractNettyHandler {
Supplier<Stopwatch> stopwatchFactory,
Runnable tooManyPingsRunnable,
TransportTracer transportTracer,
Attributes eagAttributes) {
Attributes eagAttributes,
String authority) {
Preconditions.checkArgument(maxHeaderListSize > 0, "maxHeaderListSize must be positive");
Http2HeadersDecoder headersDecoder = new GrpcHttp2ClientHeadersDecoder(maxHeaderListSize);
Http2FrameReader frameReader = new DefaultHttp2FrameReader(headersDecoder);
@ -139,7 +141,8 @@ class NettyClientHandler extends AbstractNettyHandler {
stopwatchFactory,
tooManyPingsRunnable,
transportTracer,
eagAttributes);
eagAttributes,
authority);
}
@VisibleForTesting
@ -154,7 +157,8 @@ class NettyClientHandler extends AbstractNettyHandler {
Supplier<Stopwatch> stopwatchFactory,
Runnable tooManyPingsRunnable,
TransportTracer transportTracer,
Attributes eagAttributes) {
Attributes eagAttributes,
String authority) {
Preconditions.checkNotNull(connection, "connection");
Preconditions.checkNotNull(frameReader, "frameReader");
Preconditions.checkNotNull(lifecycleManager, "lifecycleManager");
@ -163,6 +167,7 @@ class NettyClientHandler extends AbstractNettyHandler {
Preconditions.checkNotNull(stopwatchFactory, "stopwatchFactory");
Preconditions.checkNotNull(tooManyPingsRunnable, "tooManyPingsRunnable");
Preconditions.checkNotNull(eagAttributes, "eagAttributes");
Preconditions.checkNotNull(authority, "authority");
Http2FrameLogger frameLogger = new Http2FrameLogger(LogLevel.DEBUG, NettyClientHandler.class);
frameReader = new Http2InboundFrameLogger(frameReader, frameLogger);
@ -205,7 +210,8 @@ class NettyClientHandler extends AbstractNettyHandler {
stopwatchFactory,
tooManyPingsRunnable,
transportTracer,
eagAttributes);
eagAttributes,
authority);
}
private NettyClientHandler(
@ -217,13 +223,15 @@ class NettyClientHandler extends AbstractNettyHandler {
Supplier<Stopwatch> stopwatchFactory,
final Runnable tooManyPingsRunnable,
TransportTracer transportTracer,
Attributes eagAttributes) {
Attributes eagAttributes,
String authority) {
super(/* channelUnused= */ null, decoder, encoder, settings);
this.lifecycleManager = lifecycleManager;
this.keepAliveManager = keepAliveManager;
this.stopwatchFactory = stopwatchFactory;
this.transportTracer = Preconditions.checkNotNull(transportTracer);
this.eagAttributes = eagAttributes;
this.authority = authority;
// Set the frame listener on the decoder.
decoder().frameListener(new FrameListener());
@ -429,6 +437,11 @@ class NettyClientHandler extends AbstractNettyHandler {
return eagAttributes;
}
@Override
public String getAuthority() {
return authority;
}
InternalChannelz.Security getSecurityInfo() {
return securityInfo;
}

5
netty/src/main/java/io/grpc/netty/NettyClientTransport.java
View File

@ -68,6 +68,7 @@ class NettyClientTransport implements ConnectionClientTransport {
private final Class<? extends Channel> channelType;
private final EventLoopGroup group;
private final ProtocolNegotiator negotiator;
private final String authorityString;
private final AsciiString authority;
private final AsciiString userAgent;
private final int flowControlWindow;
@ -109,6 +110,7 @@ class NettyClientTransport implements ConnectionClientTransport {
this.keepAliveTimeNanos = keepAliveTimeNanos;
this.keepAliveTimeoutNanos = keepAliveTimeoutNanos;
this.keepAliveWithoutCalls = keepAliveWithoutCalls;
this.authorityString = authority;
this.authority = new AsciiString(authority);
this.userAgent = new AsciiString(GrpcUtil.getGrpcUserAgent("netty", userAgent));
this.tooManyPingsRunnable =
@ -195,7 +197,8 @@ class NettyClientTransport implements ConnectionClientTransport {
GrpcUtil.STOPWATCH_SUPPLIER,
tooManyPingsRunnable,
transportTracer,
eagAttributes);
eagAttributes,
authorityString);
NettyHandlerSettings.setAutoWindow(handler);
negotiationHandler = negotiator.newHandler(handler);

74
netty/src/main/java/io/grpc/netty/ProtocolNegotiators.java
View File

@ -259,57 +259,49 @@ public final class ProtocolNegotiators {
* be negotiated, the {@code handler} is added and writes to the {@link io.netty.channel.Channel}
* may happen immediately, even before the TLS Handshake is complete.
*/
public static ProtocolNegotiator tls(SslContext sslContext, String authority) {
Preconditions.checkNotNull(sslContext, "sslContext");
URI uri = GrpcUtil.authorityToUri(Preconditions.checkNotNull(authority, "authority"));
String host;
int port;
if (uri.getHost() != null) {
host = uri.getHost();
port = uri.getPort();
} else {
/*
* Implementation note: We pick -1 as the port here rather than deriving it from the original
* socket address. The SSL engine doens't use this port number when contacting the remote
* server, but rather it is used for other things like SSL Session caching. When an invalid
* authority is provided (like "bad_cert"), picking the original port and passing it in would
* mean that the port might used under the assumption that it was correct. By using -1 here,
* it forces the SSL implementation to treat it as invalid.
*/
host = authority;
port = -1;
}
return new TlsNegotiator(sslContext, host, port);
public static ProtocolNegotiator tls(SslContext sslContext) {
return new TlsNegotiator(sslContext);
}
@VisibleForTesting
static final class TlsNegotiator implements ProtocolNegotiator {
private final SslContext sslContext;
private final String host;
private final int port;
TlsNegotiator(SslContext sslContext, String host, int port) {
TlsNegotiator(SslContext sslContext) {
this.sslContext = checkNotNull(sslContext, "sslContext");
this.host = checkNotNull(host, "host");
this.port = port;
}
@VisibleForTesting
String getHost() {
return host;
}
@VisibleForTesting
int getPort() {
return port;
HostPort parseAuthority(String authority) {
URI uri = GrpcUtil.authorityToUri(Preconditions.checkNotNull(authority, "authority"));
String host;
int port;
if (uri.getHost() != null) {
host = uri.getHost();
port = uri.getPort();
} else {
/*
* Implementation note: We pick -1 as the port here rather than deriving it from the
* original socket address. The SSL engine doens't use this port number when contacting the
* remote server, but rather it is used for other things like SSL Session caching. When an
* invalid authority is provided (like "bad_cert"), picking the original port and passing it
* in would mean that the port might used under the assumption that it was correct. By
* using -1 here, it forces the SSL implementation to treat it as invalid.
*/
host = authority;
port = -1;
}
return new HostPort(host, port);
}
@Override
public Handler newHandler(GrpcHttp2ConnectionHandler handler) {
final HostPort hostPort = parseAuthority(handler.getAuthority());
ChannelHandler sslBootstrap = new ChannelHandlerAdapter() {
@Override
public void handlerAdded(ChannelHandlerContext ctx) throws Exception {
SSLEngine sslEngine = sslContext.newEngine(ctx.alloc(), host, port);
SSLEngine sslEngine = sslContext.newEngine(ctx.alloc(), hostPort.host, hostPort.port);
SSLParameters sslParams = sslEngine.getSSLParameters();
sslParams.setEndpointIdentificationAlgorithm("HTTPS");
sslEngine.setSSLParameters(sslParams);
@ -320,6 +312,18 @@ public final class ProtocolNegotiators {
}
}
/** A tuple of (host, port). */
@VisibleForTesting
static final class HostPort {
final String host;
final int port;
public HostPort(String host, int port) {
this.host = host;
this.port = port;
}
}
/**
* Returns a {@link ProtocolNegotiator} used for upgrading to HTTP/2 from HTTP/1.x.
*/

50
netty/src/test/java/io/grpc/netty/NettyChannelBuilderTest.java
View File

@ -139,65 +139,57 @@ public class NettyChannelBuilderTest {
}
@Test
public void createProtocolNegotiator_plaintext() {
ProtocolNegotiator negotiator = NettyChannelBuilder.createProtocolNegotiator(
"authority",
public void createProtocolNegotiatorByType_plaintext() {
ProtocolNegotiator negotiator = NettyChannelBuilder.createProtocolNegotiatorByType(
NegotiationType.PLAINTEXT,
noSslContext,
noProxy);
noSslContext);
// just check that the classes are the same, and that negotiator is not null.
assertTrue(negotiator instanceof ProtocolNegotiators.PlaintextNegotiator);
}
@Test
public void createProtocolNegotiator_plaintextUpgrade() {
ProtocolNegotiator negotiator = NettyChannelBuilder.createProtocolNegotiator(
"authority",
public void createProtocolNegotiatorByType_plaintextUpgrade() {
ProtocolNegotiator negotiator = NettyChannelBuilder.createProtocolNegotiatorByType(
NegotiationType.PLAINTEXT_UPGRADE,
noSslContext,
noProxy);
noSslContext);
// just check that the classes are the same, and that negotiator is not null.
assertTrue(negotiator instanceof ProtocolNegotiators.PlaintextUpgradeNegotiator);
}
@Test
public void createProtocolNegotiator_tlsWithNoContext() {
public void createProtocolNegotiatorByType_tlsWithNoContext() {
thrown.expect(NullPointerException.class);
NettyChannelBuilder.createProtocolNegotiator(
"authority:1234",
NettyChannelBuilder.createProtocolNegotiatorByType(
NegotiationType.TLS,
noSslContext,
noProxy);
noSslContext);
}
@Test
public void createProtocolNegotiator_tlsWithClientContext() throws SSLException {
ProtocolNegotiator negotiator = NettyChannelBuilder.createProtocolNegotiator(
"authority:1234",
public void createProtocolNegotiatorByType_tlsWithClientContext() throws SSLException {
ProtocolNegotiator negotiator = NettyChannelBuilder.createProtocolNegotiatorByType(
NegotiationType.TLS,
GrpcSslContexts.forClient().build(),
noProxy);
GrpcSslContexts.forClient().build());
assertTrue(negotiator instanceof ProtocolNegotiators.TlsNegotiator);
ProtocolNegotiators.TlsNegotiator n = (TlsNegotiator) negotiator;
ProtocolNegotiators.HostPort hostPort = n.parseAuthority("authority:1234");
assertEquals("authority", n.getHost());
assertEquals(1234, n.getPort());
assertEquals("authority", hostPort.host);
assertEquals(1234, hostPort.port);
}
@Test
public void createProtocolNegotiator_tlsWithAuthorityFallback() throws SSLException {
ProtocolNegotiator negotiator = NettyChannelBuilder.createProtocolNegotiator(
"bad_authority",
public void createProtocolNegotiatorByType_tlsWithAuthorityFallback() throws SSLException {
ProtocolNegotiator negotiator = NettyChannelBuilder.createProtocolNegotiatorByType(
NegotiationType.TLS,
GrpcSslContexts.forClient().build(),
noProxy);
GrpcSslContexts.forClient().build());
assertTrue(negotiator instanceof ProtocolNegotiators.TlsNegotiator);
ProtocolNegotiators.TlsNegotiator n = (TlsNegotiator) negotiator;
ProtocolNegotiators.HostPort hostPort = n.parseAuthority("bad_authority");
assertEquals("bad_authority", n.getHost());
assertEquals(-1, n.getPort());
assertEquals("bad_authority", hostPort.host);
assertEquals(-1, hostPort.port);
}
@Test

3
netty/src/test/java/io/grpc/netty/NettyClientHandlerTest.java
View File

@ -722,7 +722,8 @@ public class NettyClientHandlerTest extends NettyHandlerTestBase<NettyClientHand
stopwatchSupplier,
tooManyPingsRunnable,
transportTracer,
Attributes.EMPTY);
Attributes.EMPTY,
"someauthority");
}
@Override

4
netty/src/test/java/io/grpc/netty/NettyClientTransportTest.java
View File

@ -293,7 +293,7 @@ public class NettyClientTransportTest {
.ciphers(TestUtils.preferredTestCiphers(), SupportedCipherSuiteFilter.INSTANCE)
.keyManager(clientCert, clientKey)
.build();
ProtocolNegotiator negotiator = ProtocolNegotiators.tls(clientContext, authority);
ProtocolNegotiator negotiator = ProtocolNegotiators.tls(clientContext);
final NettyClientTransport transport = newTransport(negotiator);
callMeMaybe(transport.start(clientTransportListener));
@ -573,7 +573,7 @@ public class NettyClientTransportTest {
File caCert = TestUtils.loadCert("ca.pem");
SslContext clientContext = GrpcSslContexts.forClient().trustManager(caCert)
.ciphers(TestUtils.preferredTestCiphers(), SupportedCipherSuiteFilter.INSTANCE).build();
return ProtocolNegotiators.tls(clientContext, authority);
return ProtocolNegotiators.tls(clientContext);
}
private NettyClientTransport newTransport(ProtocolNegotiator negotiator) {

24
netty/src/test/java/io/grpc/netty/ProtocolNegotiatorsTest.java
View File

@ -27,6 +27,7 @@ import static org.mockito.Mockito.mock;
import static org.mockito.Mockito.times;
import io.grpc.internal.testing.TestUtils;
import io.grpc.netty.ProtocolNegotiators.HostPort;
import io.grpc.netty.ProtocolNegotiators.ServerTlsHandler;
import io.grpc.netty.ProtocolNegotiators.TlsNegotiator;
import io.netty.bootstrap.Bootstrap;
@ -253,36 +254,39 @@ public class ProtocolNegotiatorsTest {
public void tls_failsOnNullSslContext() {
thrown.expect(NullPointerException.class);
Object unused = ProtocolNegotiators.tls(null, "authority");
Object unused = ProtocolNegotiators.tls(null);
}
@Test
public void tls_hostAndPort() throws SSLException {
SslContext ctx = GrpcSslContexts.forClient().build();
TlsNegotiator negotiator = (TlsNegotiator) ProtocolNegotiators.tls(ctx, "authority:1234");
TlsNegotiator negotiator = (TlsNegotiator) ProtocolNegotiators.tls(ctx);
HostPort hostPort = negotiator.parseAuthority("authority:1234");
assertEquals("authority", negotiator.getHost());
assertEquals(1234, negotiator.getPort());
assertEquals("authority", hostPort.host);
assertEquals(1234, hostPort.port);
}
@Test
public void tls_host() throws SSLException {
SslContext ctx = GrpcSslContexts.forClient().build();
TlsNegotiator negotiator = (TlsNegotiator) ProtocolNegotiators.tls(ctx, "[::1]");
TlsNegotiator negotiator = (TlsNegotiator) ProtocolNegotiators.tls(ctx);
HostPort hostPort = negotiator.parseAuthority("[::1]");
assertEquals("[::1]", negotiator.getHost());
assertEquals(-1, negotiator.getPort());
assertEquals("[::1]", hostPort.host);
assertEquals(-1, hostPort.port);
}
@Test
public void tls_invalidHost() throws SSLException {
SslContext ctx = GrpcSslContexts.forClient().build();
TlsNegotiator negotiator = (TlsNegotiator) ProtocolNegotiators.tls(ctx, "bad_host:1234");
TlsNegotiator negotiator = (TlsNegotiator) ProtocolNegotiators.tls(ctx);
HostPort hostPort = negotiator.parseAuthority("bad_host:1234");
// Even though it looks like a port, we treat it as part of the authority, since the host is
// invalid.
assertEquals("bad_host:1234", negotiator.getHost());
assertEquals(-1, negotiator.getPort());
assertEquals("bad_host:1234", hostPort.host);
assertEquals(-1, hostPort.port);
}
@Test