275 lines
10 KiB
Python
275 lines
10 KiB
Python
#!/usr/bin/python
|
|
|
|
# Copyright (C) 2003-2004 Robey Pointer <robey@lag.net>
|
|
#
|
|
# This file is part of paramiko.
|
|
#
|
|
# Paramiko is free software; you can redistribute it and/or modify it under the
|
|
# terms of the GNU Lesser General Public License as published by the Free
|
|
# Software Foundation; either version 2.1 of the License, or (at your option)
|
|
# any later version.
|
|
#
|
|
# Paramiko is distrubuted in the hope that it will be useful, but WITHOUT ANY
|
|
# WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR
|
|
# A PARTICULAR PURPOSE. See the GNU Lesser General Public License for more
|
|
# details.
|
|
#
|
|
# You should have received a copy of the GNU Lesser General Public License
|
|
# along with Paramiko; if not, write to the Free Software Foundation, Inc.,
|
|
# 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA.
|
|
|
|
"""
|
|
Some unit tests for the ssh2 protocol in Transport.
|
|
"""
|
|
|
|
import sys, unittest, threading
|
|
from paramiko import Transport, SecurityOptions, ServerInterface, RSAKey, DSSKey, \
|
|
SSHException, BadAuthenticationType
|
|
from paramiko import AUTH_FAILED, AUTH_PARTIALLY_SUCCESSFUL, AUTH_SUCCESSFUL
|
|
from paramiko import OPEN_SUCCEEDED
|
|
from loop import LoopSocket
|
|
|
|
|
|
class NullServer (ServerInterface):
|
|
paranoid_did_password = False
|
|
paranoid_did_public_key = False
|
|
paranoid_key = DSSKey.from_private_key_file('tests/test_dss.key')
|
|
|
|
def get_allowed_auths(self, username):
|
|
if username == 'slowdive':
|
|
return 'publickey,password'
|
|
if username == 'paranoid':
|
|
if not self.paranoid_did_password and not self.paranoid_did_public_key:
|
|
return 'publickey,password'
|
|
elif self.paranoid_did_password:
|
|
return 'publickey'
|
|
else:
|
|
return 'password'
|
|
return 'publickey'
|
|
|
|
def check_auth_password(self, username, password):
|
|
if (username == 'slowdive') and (password == 'pygmalion'):
|
|
return AUTH_SUCCESSFUL
|
|
if (username == 'paranoid') and (password == 'paranoid'):
|
|
# 2-part auth (even openssh doesn't support this)
|
|
self.paranoid_did_password = True
|
|
if self.paranoid_did_public_key:
|
|
return AUTH_SUCCESSFUL
|
|
return AUTH_PARTIALLY_SUCCESSFUL
|
|
return AUTH_FAILED
|
|
|
|
def check_auth_publickey(self, username, key):
|
|
if (username == 'paranoid') and (key == self.paranoid_key):
|
|
# 2-part auth
|
|
self.paranoid_did_public_key = True
|
|
if self.paranoid_did_password:
|
|
return AUTH_SUCCESSFUL
|
|
return AUTH_PARTIALLY_SUCCESSFUL
|
|
return AUTH_FAILED
|
|
|
|
def check_channel_request(self, kind, chanid):
|
|
return OPEN_SUCCEEDED
|
|
|
|
def check_channel_exec_request(self, channel, command):
|
|
if command != 'yes':
|
|
return False
|
|
self.exec_channel = channel
|
|
return True
|
|
|
|
def check_channel_shell_request(self, channel):
|
|
self.shell_channel = channel
|
|
return True
|
|
|
|
|
|
class TransportTest (unittest.TestCase):
|
|
|
|
def setUp(self):
|
|
self.socks = LoopSocket()
|
|
self.sockc = LoopSocket()
|
|
self.sockc.link(self.socks)
|
|
self.tc = Transport(self.sockc)
|
|
self.ts = Transport(self.socks)
|
|
|
|
def tearDown(self):
|
|
self.tc.close()
|
|
self.ts.close()
|
|
self.socks.close()
|
|
self.sockc.close()
|
|
|
|
def test_1_security_options(self):
|
|
o = self.tc.get_security_options()
|
|
self.assertEquals(type(o), SecurityOptions)
|
|
self.assert_(('aes256-cbc', 'blowfish-cbc') != o.ciphers)
|
|
o.ciphers = ('aes256-cbc', 'blowfish-cbc')
|
|
self.assertEquals(('aes256-cbc', 'blowfish-cbc'), o.ciphers)
|
|
try:
|
|
o.ciphers = ('aes256-cbc', 'made-up-cipher')
|
|
self.assert_(False)
|
|
except ValueError:
|
|
pass
|
|
try:
|
|
o.ciphers = 23
|
|
self.assert_(False)
|
|
except TypeError:
|
|
pass
|
|
|
|
def test_2_simple(self):
|
|
"""
|
|
verify that we can establish an ssh link with ourselves across the
|
|
loopback sockets. this is hardly "simple" but it's simpler than the
|
|
later tests. :)
|
|
"""
|
|
host_key = RSAKey.from_private_key_file('tests/test_rsa.key')
|
|
public_host_key = RSAKey(data=str(host_key))
|
|
self.ts.add_server_key(host_key)
|
|
event = threading.Event()
|
|
server = NullServer()
|
|
self.assert_(not event.isSet())
|
|
self.ts.start_server(event, server)
|
|
self.tc.connect(hostkey=public_host_key,
|
|
username='slowdive', password='pygmalion')
|
|
event.wait(1.0)
|
|
self.assert_(event.isSet())
|
|
self.assert_(self.ts.is_active())
|
|
|
|
def test_3_bad_auth_type(self):
|
|
"""
|
|
verify that we get the right exception when an unsupported auth
|
|
type is requested.
|
|
"""
|
|
host_key = RSAKey.from_private_key_file('tests/test_rsa.key')
|
|
public_host_key = RSAKey(data=str(host_key))
|
|
self.ts.add_server_key(host_key)
|
|
event = threading.Event()
|
|
server = NullServer()
|
|
self.assert_(not event.isSet())
|
|
self.ts.start_server(event, server)
|
|
try:
|
|
self.tc.connect(hostkey=public_host_key,
|
|
username='unknown', password='error')
|
|
self.assert_(False)
|
|
except:
|
|
etype, evalue, etb = sys.exc_info()
|
|
self.assertEquals(BadAuthenticationType, etype)
|
|
self.assertEquals(['publickey'], evalue.allowed_types)
|
|
|
|
def test_4_bad_password(self):
|
|
"""
|
|
verify that a bad password gets the right exception, and that a retry
|
|
with the right password works.
|
|
"""
|
|
host_key = RSAKey.from_private_key_file('tests/test_rsa.key')
|
|
public_host_key = RSAKey(data=str(host_key))
|
|
self.ts.add_server_key(host_key)
|
|
event = threading.Event()
|
|
server = NullServer()
|
|
self.assert_(not event.isSet())
|
|
self.ts.start_server(event, server)
|
|
self.tc.ultra_debug = True
|
|
self.tc.connect(hostkey=public_host_key)
|
|
try:
|
|
self.tc.auth_password(username='slowdive', password='error')
|
|
self.assert_(False)
|
|
except:
|
|
etype, evalue, etb = sys.exc_info()
|
|
self.assertEquals(SSHException, etype)
|
|
self.tc.auth_password(username='slowdive', password='pygmalion')
|
|
event.wait(1.0)
|
|
self.assert_(event.isSet())
|
|
self.assert_(self.ts.is_active())
|
|
|
|
def test_5_multipart_auth(self):
|
|
"""
|
|
verify that multipart auth works.
|
|
"""
|
|
host_key = RSAKey.from_private_key_file('tests/test_rsa.key')
|
|
public_host_key = RSAKey(data=str(host_key))
|
|
self.ts.add_server_key(host_key)
|
|
event = threading.Event()
|
|
server = NullServer()
|
|
self.assert_(not event.isSet())
|
|
self.ts.start_server(event, server)
|
|
self.tc.ultra_debug = True
|
|
self.tc.connect(hostkey=public_host_key)
|
|
remain = self.tc.auth_password(username='paranoid', password='paranoid')
|
|
self.assertEquals(['publickey'], remain)
|
|
key = DSSKey.from_private_key_file('tests/test_dss.key')
|
|
remain = self.tc.auth_publickey(username='paranoid', key=key)
|
|
self.assertEquals([], remain)
|
|
event.wait(1.0)
|
|
self.assert_(event.isSet())
|
|
self.assert_(self.ts.is_active())
|
|
|
|
def test_6_exec_command(self):
|
|
"""
|
|
verify that exec_command() does something reasonable.
|
|
"""
|
|
host_key = RSAKey.from_private_key_file('tests/test_rsa.key')
|
|
public_host_key = RSAKey(data=str(host_key))
|
|
self.ts.add_server_key(host_key)
|
|
event = threading.Event()
|
|
server = NullServer()
|
|
self.assert_(not event.isSet())
|
|
self.ts.start_server(event, server)
|
|
self.tc.ultra_debug = True
|
|
self.tc.connect(hostkey=public_host_key)
|
|
self.tc.auth_password(username='slowdive', password='pygmalion')
|
|
event.wait(1.0)
|
|
self.assert_(event.isSet())
|
|
self.assert_(self.ts.is_active())
|
|
|
|
chan = self.tc.open_session()
|
|
self.assert_(not chan.exec_command('no'))
|
|
|
|
chan = self.tc.open_session()
|
|
self.assert_(chan.exec_command('yes'))
|
|
server.exec_channel.send('Hello there.\n')
|
|
server.exec_channel.send_stderr('This is on stderr.\n')
|
|
server.exec_channel.close()
|
|
|
|
f = chan.makefile()
|
|
self.assertEquals('Hello there.\n', f.readline())
|
|
self.assertEquals('', f.readline())
|
|
f = chan.makefile_stderr()
|
|
self.assertEquals('This is on stderr.\n', f.readline())
|
|
self.assertEquals('', f.readline())
|
|
|
|
# now try it with combined stdout/stderr
|
|
chan = self.tc.open_session()
|
|
chan.exec_command('yes')
|
|
server.exec_channel.send('Hello there.\n')
|
|
server.exec_channel.send_stderr('This is on stderr.\n')
|
|
server.exec_channel.close()
|
|
|
|
chan.set_combine_stderr(True)
|
|
f = chan.makefile()
|
|
self.assertEquals('Hello there.\n', f.readline())
|
|
self.assertEquals('This is on stderr.\n', f.readline())
|
|
self.assertEquals('', f.readline())
|
|
|
|
def test_7_invoke_shell(self):
|
|
"""
|
|
verify that invoke_shell() does something reasonable.
|
|
"""
|
|
host_key = RSAKey.from_private_key_file('tests/test_rsa.key')
|
|
public_host_key = RSAKey(data=str(host_key))
|
|
self.ts.add_server_key(host_key)
|
|
event = threading.Event()
|
|
server = NullServer()
|
|
self.assert_(not event.isSet())
|
|
self.ts.start_server(event, server)
|
|
self.tc.ultra_debug = True
|
|
self.tc.connect(hostkey=public_host_key)
|
|
self.tc.auth_password(username='slowdive', password='pygmalion')
|
|
event.wait(1.0)
|
|
self.assert_(event.isSet())
|
|
self.assert_(self.ts.is_active())
|
|
|
|
chan = self.tc.open_session()
|
|
self.assert_(chan.invoke_shell())
|
|
chan.send('communist j. cat\n')
|
|
f = server.shell_channel.makefile()
|
|
self.assertEquals('communist j. cat\n', f.readline())
|
|
chan.close()
|
|
self.assertEquals('', f.readline())
|