diff --git a/lib/ansible/runner/__init__.py b/lib/ansible/runner/__init__.py index 105ca31ba26..52f58555e76 100644 --- a/lib/ansible/runner/__init__.py +++ b/lib/ansible/runner/__init__.py @@ -406,7 +406,7 @@ class Runner(object): host_variables = self.inventory.get_variables(host) host_connection = host_variables.get('ansible_connection', self.transport) - if host_connection in [ 'paramiko', 'ssh' ]: + if host_connection in [ 'paramiko', 'ssh', 'fireball2' ]: port = host_variables.get('ansible_ssh_port', self.remote_port) if port is None: port = C.DEFAULT_REMOTE_PORT @@ -569,7 +569,7 @@ class Runner(object): actual_pass = inject.get('ansible_ssh_pass', self.remote_pass) actual_transport = inject.get('ansible_connection', self.transport) actual_private_key_file = inject.get('ansible_ssh_private_key_file', self.private_key_file) - if actual_transport in [ 'paramiko', 'ssh' ]: + if actual_transport in [ 'paramiko', 'ssh', 'fireball2' ]: actual_port = inject.get('ansible_ssh_port', port) # the delegated host may have different SSH port configured, etc @@ -614,7 +614,7 @@ class Runner(object): # for fireball2, we stuff both ports into a single # variable so that we don't have to mangle other function # calls just to accomodate this one case - actual_port = [port, self.accelerate_port] + actual_port = [actual_port, self.accelerate_port] elif actual_port is not None: actual_port = int(actual_port) except ValueError, e: diff --git a/lib/ansible/runner/connection_plugins/fireball2.py b/lib/ansible/runner/connection_plugins/fireball2.py index 55da006540c..47e92e49a27 100644 --- a/lib/ansible/runner/connection_plugins/fireball2.py +++ b/lib/ansible/runner/connection_plugins/fireball2.py @@ -20,6 +20,7 @@ import os import base64 import socket import struct +import time from ansible.callbacks import vvv from ansible.runner.connection_plugins.ssh import Connection as SSHConnection from ansible import utils