diff --git a/airbyte-integrations/bases/base-java/src/main/java/io/airbyte/integrations/base/ssh/SshBastionContainer.java b/airbyte-integrations/bases/base-java/src/main/java/io/airbyte/integrations/base/ssh/SshBastionContainer.java index f7acac0f0f5f..7b6032061ec7 100644 --- a/airbyte-integrations/bases/base-java/src/main/java/io/airbyte/integrations/base/ssh/SshBastionContainer.java +++ b/airbyte-integrations/bases/base-java/src/main/java/io/airbyte/integrations/base/ssh/SshBastionContainer.java @@ -40,9 +40,13 @@ public JsonNode getTunnelConfig(final SshTunnel.TunnelMethod tunnelMethod, final return Jsons.jsonNode(builderWithSchema .put("tunnel_method", Jsons.jsonNode(ImmutableMap.builder() - .put("tunnel_host", bastion.getHost()) + .put("tunnel_host", + Objects.requireNonNull(bastion.getContainerInfo().getNetworkSettings() + .getNetworks() + .get(((Network.NetworkImpl) network).getName()) + .getIpAddress())) .put("tunnel_method", tunnelMethod) - .put("tunnel_port", bastion.getFirstMappedPort()) + .put("tunnel_port", bastion.getExposedPorts().get(0)) .put("tunnel_user", SSH_USER) .put("tunnel_user_password", tunnelMethod.equals(SSH_PASSWORD_AUTH) ? SSH_PASSWORD : "") .put("ssh_key", tunnelMethod.equals(SSH_KEY_AUTH) ? bastion.execInContainer("cat", "var/bastion/id_rsa").getStdout() : "")