Test mtls over amqp1.0

Author: MarcialRosales

selenium/amqp10-roundtriptest/src/main/java/com/rabbitmq/amqp1_0/RoundTripTest.java

@@ -15,15 +15,39 @@ public class RoundTripTest {
   public static String getEnv(String property, String defaultValue) {
     return System.getenv(property) == null ? defaultValue : System.getenv(property);
   }
+  public static String getEnv(String property) {
+    String value = System.getenv(property);
+    if (value == null) {
+      throw new IllegalArgumentException("Missing env variable " + property);
+    }
+    return value;
+  }
   public static void main(String args[]) throws Exception {
     String hostname = getEnv("RABBITMQ_HOSTNAME", "localhost");
     String port = getEnv("RABBITMQ_AMQP_PORT", "5672");
     String scheme = getEnv("RABBITMQ_AMQP_SCHEME", "amqp");
     String username = args.length > 0 ? args[0] : getEnv("RABBITMQ_AMQP_USERNAME", "guest");
     String password = args.length > 1 ? args[1] : getEnv("RABBITMQ_AMQP_PASSWORD", "guest");
     String uri = scheme + "://" + hostname + ":" + port;
+    boolean usemtls = Boolean.parseBoolean(getEnv("AMQP_USE_MTLS", "false"));
+    String certsLocation = getEnv("RABBITMQ_CERTS");
+
+    if ("amqps".equals(scheme)) {
+      List<String> connectionParams = new ArrayList<String>();
 
-    System.out.println("AMQPS Roundrip using uri " + uri);
+      connectionParams.add("transport.trustStoreLocation=" + certsLocation + "/truststore.jks");
+      connectionParams.add("transport.trustStorePassword=foobar");
+      connectionParams.add("transport.verifyHost=true");      
+
+      if (usemtls) {
+        connectionParams.add("transport.keyStoreLocation=" + certsLocation + "/rabbitmq.jks");
+        connectionParams.add("transport.keyStorePassword=foobar");
+        connectionParams.add("transport.keyAlias=rabbitmq-tls");
+      }
+      if (!connectionParams.isEmpty()) {
+        uri += "?" + java.net.URLEncoder.encode(String.join("&", connectionParams));       
+      }
+    }
 
     Hashtable<Object, Object> env = new Hashtable<>();
     env.put(Context.INITIAL_CONTEXT_FACTORY, "org.apache.qpid.jms.jndi.JmsInitialContextFactory");

selenium/bin/components/rabbitmq

@@ -16,6 +16,7 @@ init_rabbitmq() {
     mkdir ${RABBITMQ_CONFIG_DIR}/certs
   fi 
   generate-ca-server-client-kpi rabbitmq $RABBITMQ_CONFIG_DIR/certs
+  generate-keystore-if-required rabbitmq $RABBITMQ_CONFIG_DIR/certs
 }
 
 start_rabbitmq() {
@@ -186,8 +187,6 @@ start_docker_rabbitmq() {
     ${RABBITMQ_DOCKER_IMAGE}
 #-v ${TEST_DIR}:/config \
 
-  docker exec rabbitmq ls /etc/rabbitmq/certs
   wait_for_message rabbitmq "Server startup complete"
-
   end "RabbitMQ ready"
 }

selenium/test/authnz-msg-protocols/env.auth-mtls

@@ -1 +1,2 @@
 export MQTT_USE_MTLS=true 
+export AMQP_USE_MTLS=true