cmueller/ansible
1
0
Fork 0
Code Issues Pull requests Projects Releases Wiki Activity

Work in progress on sudo, hitting some issues with paramiko saying the connection is closed.

Browse source
This commit is contained in:
Michael DeHaan 2012-03-28 22:51:16 -04:00
parent 7133734d87
commit c2a6e2f97b
2 changed files with 53 additions and 40 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

48
lib/ansible/connection.py
View file

@ -19,7 +19,9 @@
################################################ ################################################
import paramiko import paramiko
import traceback
import os import os
import time
from ansible import errors from ansible import errors
################################################ ################################################
@ -53,14 +55,12 @@ class ParamikoConnection(object):
self.runner = runner self.runner = runner
self.host = host self.host = host
def connect(self): def _get_conn(self):
''' connect to the remote host ''' ssh = paramiko.SSHClient()
ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy())
self.ssh = paramiko.SSHClient()
self.ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy())
try: try:
self.ssh.connect( ssh.connect(
self.host, username=self.runner.remote_user, self.host, username=self.runner.remote_user,
allow_agent=True, look_for_keys=True, password=self.runner.remote_pass, allow_agent=True, look_for_keys=True, password=self.runner.remote_pass,
timeout=self.runner.timeout, port=self.runner.remote_port timeout=self.runner.timeout, port=self.runner.remote_port
@ -71,23 +71,34 @@ class ParamikoConnection(object):
else: else:
raise errors.AnsibleConnectionFailed(str(e)) raise errors.AnsibleConnectionFailed(str(e))
return ssh
def connect(self):
''' connect to the remote host '''
self.ssh = self._get_conn()
return self return self
def exec_command(self, cmd): def exec_command(self, cmd, sudoable=True):
''' run a command on the remote host ''' ''' run a command on the remote host '''
#if not False: if not False: # if not self.runner.sudo or not sudoable:
stdin, stdout, stderr = self.ssh.exec_command(cmd) stdin, stdout, stderr = self.ssh.exec_command(cmd)
return (stdin, stdout, stderr) return (stdin, stdout, stderr)
#else: else:
# sudo_chan = self.ssh.get_transport().open_session() # this code is a work in progress, so it's disabled...
# sudo_chan = chan.get_pty() self.ssh.close()
# sudo_chan.exec_command("sudo %s" % cmd) ssh_sudo = self._get_conn()
# output = channel.makefile('rb', -1).readlines() sudo_chan = ssh_sudo.invoke_shell()
# if not output: sudo_chan.exec_command("sudo -s")
# output = channel.makefile_stderr('rb', -1).readlines() sudo_chan.recv(1024)
# print "DEBUG: output: %s" % output sudo_chan.send("%s\n" % cmd)
# channel.close() # TODO: wait for ready...
# return (None, '', output) out = sudo_chan.recv(1024)
sudo_chan.close()
self.ssh = self._get_conn()
return (None, "\n".join(out), '')
def put_file(self, in_path, out_path): def put_file(self, in_path, out_path):
@ -98,6 +109,7 @@ class ParamikoConnection(object):
try: try:
sftp.put(in_path, out_path) sftp.put(in_path, out_path)
except IOError: except IOError:
traceback.print_exc()
raise errors.AnsibleError("failed to transfer file to %s" % out_path) raise errors.AnsibleError("failed to transfer file to %s" % out_path)
sftp.close() sftp.close()

31
lib/ansible/runner.py
View file

@ -73,7 +73,8 @@ class Runner(object):
forks=C.DEFAULT_FORKS, timeout=C.DEFAULT_TIMEOUT, pattern=C.DEFAULT_PATTERN, forks=C.DEFAULT_FORKS, timeout=C.DEFAULT_TIMEOUT, pattern=C.DEFAULT_PATTERN,
remote_user=C.DEFAULT_REMOTE_USER, remote_pass=C.DEFAULT_REMOTE_PASS, remote_user=C.DEFAULT_REMOTE_USER, remote_pass=C.DEFAULT_REMOTE_PASS,
remote_port=C.DEFAULT_REMOTE_PORT, background=0, basedir=None, setup_cache=None, remote_port=C.DEFAULT_REMOTE_PORT, background=0, basedir=None, setup_cache=None,
transport='paramiko', conditional='True', groups={}, callbacks=None, verbose=False): transport='paramiko', conditional='True', groups={}, callbacks=None, verbose=False,
sudo=True): # FIXME FIXME FIXME
if setup_cache is None: if setup_cache is None:
setup_cache = {} setup_cache = {}
@ -107,6 +108,7 @@ class Runner(object):
self.remote_port = remote_port self.remote_port = remote_port
self.background = background self.background = background
self.basedir = basedir self.basedir = basedir
self.sudo = sudo
self._tmp_paths = {} self._tmp_paths = {}
random.seed() random.seed()
@ -243,13 +245,6 @@ class Runner(object):
# ***************************************************** # *****************************************************
def _transfer_file(self, conn, source, dest):
''' transfers a remote file '''
conn.put_file(source, dest)
# *****************************************************
def _transfer_module(self, conn, tmp, module): def _transfer_module(self, conn, tmp, module):
''' transfers a module file to the remote side to execute it, but does not execute it yet ''' ''' transfers a module file to the remote side to execute it, but does not execute it yet '''
@ -269,7 +264,7 @@ class Runner(object):
args_fo.close() args_fo.close()
args_remote = os.path.join(tmp, 'arguments') args_remote = os.path.join(tmp, 'arguments')
self._transfer_file(conn, args_file, args_remote) conn.put_file(args_file, args_remote)
os.unlink(args_file) os.unlink(args_file)
return args_remote return args_remote
@ -435,7 +430,7 @@ class Runner(object):
# transfer the file to a remote tmp location # transfer the file to a remote tmp location
tmp_src = tmp + source.split('/')[-1] tmp_src = tmp + source.split('/')[-1]
self._transfer_file(conn, utils.path_dwim(self.basedir, source), tmp_src) conn.put_file(utils.path_dwim(self.basedir, source), tmp_src)
# install the copy module # install the copy module
self.module_name = 'copy' self.module_name = 'copy'
@ -487,7 +482,7 @@ class Runner(object):
# first copy the source template over # first copy the source template over
temppath = tmp + os.path.split(source)[-1] temppath = tmp + os.path.split(source)[-1]
self._transfer_file(conn, utils.path_dwim(self.basedir, source), temppath) conn.put_file(utils.path_dwim(self.basedir, source), temppath)
# install the template module # install the template module
template_module = self._transfer_module(conn, tmp, 'template') template_module = self._transfer_module(conn, tmp, 'template')
@ -531,6 +526,7 @@ class Runner(object):
tmp = self._get_tmp_path(conn) tmp = self._get_tmp_path(conn)
result = None result = None
if self.module_name == 'copy': if self.module_name == 'copy':
result = self._execute_copy(conn, host, tmp) result = self._execute_copy(conn, host, tmp)
elif self.module_name == 'template': elif self.module_name == 'template':
@ -559,23 +555,28 @@ class Runner(object):
# ***************************************************** # *****************************************************
def _exec_command(self, conn, cmd): def _exec_command(self, conn, cmd, sudoable=False):
''' execute a command string over SSH, return the output ''' ''' execute a command string over SSH, return the output '''
msg = '%s: %s' % (self.module_name, cmd) msg = '%s: %s' % (self.module_name, cmd)
# log remote command execution # log remote command execution
conn.exec_command('/usr/bin/logger -t ansible -p auth.info "%s"' % msg) conn.exec_command('/usr/bin/logger -t ansible -p auth.info "%s"' % msg)
# now run actual command # now run actual command
stdin, stdout, stderr = conn.exec_command(cmd) stdin, stdout, stderr = conn.exec_command(cmd, sudoable=sudoable)
if type(stdout) != str:
return "\n".join(stdout.readlines()) return "\n".join(stdout.readlines())
else:
return stdout
# ***************************************************** # *****************************************************
def _get_tmp_path(self, conn): def _get_tmp_path(self, conn):
''' gets a temporary path on a remote box ''' ''' gets a temporary path on a remote box '''
result = self._exec_command(conn, "mktemp -d /tmp/ansible.XXXXXX") result = self._exec_command(conn, "mktemp -d /tmp/ansible.XXXXXX", sudoable=False)
return result.split("\n")[0] + '/' cleaned = result.split("\n")[0].strip() + '/'
return cleaned
# ***************************************************** # *****************************************************