Add --server-name-indication option to performance tool

Fixes #12

Author: acogoluegnes

src/main/java/com/rabbitmq/stream/perf/StreamPerfTest.java

@@ -26,6 +26,7 @@
 import com.rabbitmq.stream.ConsumerBuilder;
 import com.rabbitmq.stream.Environment;
 import com.rabbitmq.stream.EnvironmentBuilder;
+import com.rabbitmq.stream.EnvironmentBuilder.TlsConfiguration;
 import com.rabbitmq.stream.OffsetSpecification;
 import com.rabbitmq.stream.Producer;
 import com.rabbitmq.stream.StreamCreator;
@@ -71,6 +72,8 @@
 import java.util.function.Supplier;
 import java.util.stream.Collectors;
 import java.util.stream.IntStream;
+import javax.net.ssl.SNIServerName;
+import javax.net.ssl.SSLParameters;
 import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;
 import picocli.CommandLine;
@@ -296,6 +299,13 @@ public class StreamPerfTest implements Callable<Integer> {
       defaultValue = "false")
   private boolean memoryReport;
 
+  @CommandLine.Option(
+      names = {"--server-name-indication", "-sni"},
+      description = "server names for Server Name Indication TLS parameter, separated by commas",
+      defaultValue = "",
+      converter = Utils.SniServerNamesConverter.class)
+  private List<SNIServerName> sniServerNames;
+
   private MetricsCollector metricsCollector;
   private PerformanceMetrics performanceMetrics;
 
@@ -492,14 +502,18 @@ public Integer call() throws Exception {
             .maxConsumersByConnection(this.consumersByConnection);
 
     if (tls) {
-      environmentBuilder =
-          environmentBuilder
-              .tls()
-              .sslContext(
-                  SslContextBuilder.forClient()
-                      .trustManager(Utils.TRUST_EVERYTHING_TRUST_MANAGER)
-                      .build())
-              .environmentBuilder();
+      TlsConfiguration tlsConfiguration = environmentBuilder.tls();
+      tlsConfiguration =
+          tlsConfiguration.sslContext(
+              SslContextBuilder.forClient()
+                  .trustManager(Utils.TRUST_EVERYTHING_TRUST_MANAGER)
+                  .build());
+      if (!this.sniServerNames.isEmpty()) {
+        SSLParameters sslParameters = new SSLParameters();
+        sslParameters.setServerNames(this.sniServerNames);
+        tlsConfiguration = tlsConfiguration.sslParameters(sslParameters);
+      }
+      environmentBuilder = tlsConfiguration.environmentBuilder();
     }
 
     Environment environment = environmentBuilder.build();

src/main/java/com/rabbitmq/stream/perf/Utils.java

@@ -38,10 +38,13 @@
 import java.util.function.BiFunction;
 import java.util.stream.Collectors;
 import java.util.stream.IntStream;
+import javax.net.ssl.SNIHostName;
+import javax.net.ssl.SNIServerName;
 import javax.net.ssl.X509TrustManager;
 import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;
 import picocli.CommandLine;
+import picocli.CommandLine.ITypeConverter;
 
 class Utils {
 
@@ -161,6 +164,21 @@ public BiFunction<String, Integer, String> convert(String input) {
     }
   }
 
+  static class SniServerNamesConverter implements ITypeConverter<List<SNIServerName>> {
+
+    @Override
+    public List<SNIServerName> convert(String value) throws Exception {
+      if (value == null || value.trim().isEmpty()) {
+        return Collections.emptyList();
+      } else {
+        return Arrays.stream(value.split(","))
+            .map(s -> s.trim())
+            .map(s -> new SNIHostName(s))
+            .collect(Collectors.toList());
+      }
+    }
+  }
+
   static class RangeTypeConverter implements CommandLine.ITypeConverter<String> {
 
     @Override

src/test/java/com/rabbitmq/stream/perf/StreamPerfTestTest.java

@@ -198,7 +198,11 @@ void offsetShouldNotBeStoredWhenOptionIsNotEnabled() throws Exception {
   @Test
   @DisabledIfTlsNotEnabled
   void shouldConnectWithTls() throws Exception {
-    Future<?> run = run(builder().uris("rabbitmq-stream+tls://guest:guest@localhost:5551/%2f"));
+    Future<?> run =
+        run(
+            builder()
+                .uris("rabbitmq-stream+tls://guest:guest@localhost:5551/%2f")
+                .serverNameIndication("localhost"));
     waitUntilStreamExists(s);
     waitOneSecond();
     run.cancel(true);
@@ -310,6 +314,11 @@ ArgumentsBuilder uris(String url) {
       return this;
     }
 
+    ArgumentsBuilder serverNameIndication(String sni) {
+      arguments.put("server-name-indication", sni);
+      return this;
+    }
+
     ArgumentsBuilder byteRates() {
       arguments.put("metrics-byte-rates", "");
       return this;

src/test/java/com/rabbitmq/stream/perf/UtilsTest.java

@@ -24,6 +24,7 @@
 import com.rabbitmq.stream.perf.Utils.CompressionTypeConverter;
 import com.rabbitmq.stream.perf.Utils.PatternConsumerNameStrategy;
 import com.rabbitmq.stream.perf.Utils.RangeTypeConverter;
+import com.rabbitmq.stream.perf.Utils.SniServerNamesConverter;
 import java.util.Arrays;
 import java.util.Collections;
 import java.util.List;
@@ -33,6 +34,7 @@
 import java.util.stream.IntStream;
 import java.util.stream.LongStream;
 import java.util.stream.Stream;
+import javax.net.ssl.SNIHostName;
 import org.junit.jupiter.api.Test;
 import org.junit.jupiter.params.ParameterizedTest;
 import org.junit.jupiter.params.provider.Arguments;
@@ -145,6 +147,16 @@ void consumerNameStrategy(String pattern, String expected) {
     assertThat(strategy.apply("s1", 2)).isEqualTo(expected);
   }
 
+  @Test
+  void sniServerNamesConverter() throws Exception {
+    SniServerNamesConverter converter = new SniServerNamesConverter();
+    assertThat(converter.convert("")).isEmpty();
+    assertThat(converter.convert("localhost,dummy"))
+        .hasSize(2)
+        .contains(new SNIHostName("localhost"))
+        .contains(new SNIHostName("dummy"));
+  }
+
   @Test
   void writeReadLongInByteArray() {
     byte[] array = new byte[8];