Вложенная сессия SSH с Paramiko
Я переписываю скрипт Bash, который я написал в Python. Суть этого сценария была
ssh -t first.com "ssh second.com very_remote_command"
У меня проблема с вложенной аутентификацией с paramiko. Я не смог найти ни одного примера, связанного с моей конкретной ситуацией, но я смог найти примеры с sudo на удаленном хосте.
Первый метод пишет в стандартный ввод
ssh.connect('127.0.0.1', username='jesse', password='lol')
stdin, stdout, stderr = ssh.exec_command("sudo dmesg")
stdin.write('lol\n')
stdin.flush()
Второй создает канал и использует сокет-подобный send и recv.
Мне удалось заставить stdin.write работать с sudo, но он не работает с ssh на удаленном хосте.
import paramiko
ssh = paramiko.SSHClient()
ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy())
ssh.connect('first.com', username='luser', password='secret')
stdin, stdout, stderr = ssh.exec_command('ssh luser@second.com')
stdin.write('secret')
stdin.flush()
print '---- out ----'
print stdout.readlines()
print '---- error ----'
print stderr.readlines()
ssh.close()
... печать...
---- out ----
[]
---- error ----
['Pseudo-terminal will not be allocated because stdin is not a terminal.\r\n', 'Permission denied, please try again.\r\n', 'Permission denied, please try again.\r\n', 'Permission denied (publickey,password,keyboard-interactive).\r\n']
Ошибка псевдотерминала напомнила мне флаг -t в моей исходной команде, поэтому я переключился на второй метод, используя канал. Вместо ssh.exec_command и позже у меня есть:
t = ssh.get_transport()
chan = t.open_session()
chan.get_pty()
print '---- send ssh cmd ----'
print chan.send('ssh luser@second.com')
print '---- recv ----'
print chan.recv(9999)
chan = t.open_session()
print '---- send password ----'
print chan.send('secret')
print '---- recv ----'
print chan.recv(9999)
... но он печатает "---- send ssh cmd ----" и просто зависает, пока я не убью процесс.
Я новичок в Python и не слишком осведомлен о сетях. В первом случае, почему отправка пароля работает с sudo, но не с ssh? Различаются ли подсказки? Является ли paramiko подходящей библиотекой для этого?
5 ответов
Мне удалось найти решение, но оно требует небольшой ручной работы. Если у кого-нибудь есть лучшее решение, пожалуйста, скажите мне.
ssh = paramiko.SSHClient()
ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy())
ssh.connect('first.com', username='luser', password='secret')
chan = ssh.invoke_shell()
# Ssh and wait for the password prompt.
chan.send('ssh second.com\n')
buff = ''
while not buff.endswith('\'s password: '):
resp = chan.recv(9999)
buff += resp
# Send the password and wait for a prompt.
chan.send('secret\n')
buff = ''
while not buff.endswith('some-prompt$ '):
resp = chan.recv(9999)
buff += resp
# Execute whatever command and wait for a prompt again.
chan.send('ls\n')
buff = ''
while not buff.endswith('some-prompt$ '):
resp = chan.recv(9999)
buff += resp
# Now buff has the data I need.
print 'buff', buff
ssh.close()
Стоит отметить, что вместо этого
t = ssh.get_transport()
chan = t.open_session()
chan.get_pty()
...Вы хотите это
chan = ssh.invoke_shell()
Это напоминает мне о том, когда я пытался написать сценарий TradeWars, когда я был ребенком, и бросил программирование в течение десяти лет.:)
Вот небольшой пример использования только paramiko (и переадресации портов):
import paramiko as ssh
class SSHTool():
def __init__(self, host, user, auth,
via=None, via_user=None, via_auth=None):
if via:
t0 = ssh.Transport(via)
t0.start_client()
t0.auth_password(via_user, via_auth)
# setup forwarding from 127.0.0.1:<free_random_port> to |host|
channel = t0.open_channel('direct-tcpip', host, ('127.0.0.1', 0))
self.transport = ssh.Transport(channel)
else:
self.transport = ssh.Transport(host)
self.transport.start_client()
self.transport.auth_password(user, auth)
def run(self, cmd):
ch = self.transport.open_session()
ch.set_combine_stderr(True)
ch.exec_command(cmd)
retcode = ch.recv_exit_status()
buf = ''
while ch.recv_ready():
buf += ch.recv(1024)
return (buf, retcode)
# The example below is equivalent to
# $ ssh 10.10.10.10 ssh 192.168.1.1 uname -a
# The code above works as if these 2 commands were executed:
# $ ssh -L <free_random_port>:192.168.1.1:22 10.10.10.10
# $ ssh 127.0.0.1:<free_random_port> uname -a
host = ('192.168.1.1', 22)
via_host = ('10.10.10.10', 22)
ssht = SSHTool(host, 'user1', 'pass1',
via=via_host, via_user='user2', via_auth='pass2')
print ssht.run('uname -a')
Вы можете создать ssh-соединение, используя канал из другого ssh-соединения. Смотрите здесь для более подробной информации.
Для готового решения посмотрите pxssh из проекта pxpect. Посмотрите на примеры sshls.py и ssh_tunnel.py.
Ответ Синаса работает хорошо, но не предоставил мне весь вывод очень длинных команд. Однако использование chan.makefile() позволяет мне получить все выходные данные.
Ниже работает в системе, которая требует tty, а также запрашивает пароль sudo
ssh = paramiko.SSHClient()
ssh.load_system_host_keys()
ssh.set_missing_host_key_policy(paramiko.WarningPolicy())
ssh.connect("10.10.10.1", 22, "user", "password")
chan=ssh.get_transport().open_session()
chan.get_pty()
f = chan.makefile()
chan.exec_command("sudo dmesg")
chan.send("password\n")
print f.read()
ssh.close()