3
# Copyright (c) 2009 Twisted Matrix Laboratories.
4
# See LICENSE for details.
6
from twisted.cred import portal, checkers
7
from twisted.conch import error, avatar
8
from twisted.conch.checkers import SSHPublicKeyDatabase
9
from twisted.conch.ssh import factory, userauth, connection, keys, session
10
from twisted.internet import reactor, protocol, defer
11
from twisted.python import log
12
from zope.interface import implements
14
log.startLogging(sys.stderr)
17
Example of running another protocol over an SSH channel.
18
log in with username "user" and password "password".
21
class ExampleAvatar(avatar.ConchUser):
23
def __init__(self, username):
24
avatar.ConchUser.__init__(self)
25
self.username = username
26
self.channelLookup.update({'session':session.SSHSession})
29
implements(portal.IRealm)
31
def requestAvatar(self, avatarId, mind, *interfaces):
32
return interfaces[0], ExampleAvatar(avatarId), lambda: None
34
class EchoProtocol(protocol.Protocol):
35
"""this is our example protocol that we will run over SSH
37
def dataReceived(self, data):
40
elif data == '\x03': #^C
41
self.transport.loseConnection()
43
self.transport.write(data)
45
publicKey = 'ssh-rsa AAAAB3NzaC1yc2EAAAABIwAAAGEArzJx8OYOnJmzf4tfBEvLi8DVPrJ3/c9k2I/Az64fxjHf9imyRJbixtQhlH9lfNjUIx+4LmrJH5QNRsFporcHDKOTwTTYLh5KmRpslkYHRivcJSkbh/C+BR3utDS555mV'
47
privateKey = """-----BEGIN RSA PRIVATE KEY-----
48
MIIByAIBAAJhAK8ycfDmDpyZs3+LXwRLy4vA1T6yd/3PZNiPwM+uH8Yx3/YpskSW
49
4sbUIZR/ZXzY1CMfuC5qyR+UDUbBaaK3Bwyjk8E02C4eSpkabJZGB0Yr3CUpG4fw
50
vgUd7rQ0ueeZlQIBIwJgbh+1VZfr7WftK5lu7MHtqE1S1vPWZQYE3+VUn8yJADyb
51
Z4fsZaCrzW9lkIqXkE3GIY+ojdhZhkO1gbG0118sIgphwSWKRxK0mvh6ERxKqIt1
52
xJEJO74EykXZV4oNJ8sjAjEA3J9r2ZghVhGN6V8DnQrTk24Td0E8hU8AcP0FVP+8
53
PQm/g/aXf2QQkQT+omdHVEJrAjEAy0pL0EBH6EVS98evDCBtQw22OZT52qXlAwZ2
54
gyTriKFVoqjeEjt3SZKKqXHSApP/AjBLpF99zcJJZRq2abgYlf9lv1chkrWqDHUu
55
DZttmYJeEfiFBBavVYIF1dOlZT0G8jMCMBc7sOSZodFnAiryP+Qg9otSBjJ3bQML
56
pSTqy7c3a2AScC/YyOwkDaICHnnD3XyjMwIxALRzl0tQEKMXs6hH8ToUdlLROCrP
57
EhQ0wahUTCk1gKA4uPD6TMTChavbh4K63OvbKg==
58
-----END RSA PRIVATE KEY-----"""
61
class InMemoryPublicKeyChecker(SSHPublicKeyDatabase):
63
def checkKey(self, credentials):
64
return credentials.username == 'user' and \
65
keys.getPublicKeyString(data=publicKey) == credentials.blob
69
def __init__(self, avatar):
71
We don't use it, but the adapter is passed the avatar as its first
75
def getPty(self, term, windowSize, attrs):
78
def execCommand(self, proto, cmd):
79
raise Exception("no executing commands")
81
def openShell(self, trans):
83
ep.makeConnection(trans)
84
trans.makeConnection(session.wrapProtocol(ep))
86
def eofReceived(self):
92
from twisted.python import components
93
components.registerAdapter(ExampleSession, ExampleAvatar, session.ISession)
95
class ExampleFactory(factory.SSHFactory):
97
'ssh-rsa': keys.Key.fromString(data=publicKey)
100
'ssh-rsa': keys.Key.fromString(data=privateKey)
103
'ssh-userauth': userauth.SSHUserAuthServer,
104
'ssh-connection': connection.SSHConnection
108
portal = portal.Portal(ExampleRealm())
109
passwdDB = checkers.InMemoryUsernamePasswordDatabaseDontUse()
110
passwdDB.addUser('user', 'password')
111
portal.registerChecker(passwdDB)
112
portal.registerChecker(InMemoryPublicKeyChecker())
113
ExampleFactory.portal = portal
115
if __name__ == '__main__':
116
reactor.listenTCP(5022, ExampleFactory())