1
# Copyright (c) 2001-2004 Twisted Matrix Laboratories.
2
# See LICENSE for details.
5
from twisted.internet import defer, protocol, reactor
6
from twisted.conch import error
7
from twisted.conch.ssh import transport
8
from twisted.python import log
14
class SSHClientFactory(protocol.ClientFactory):
17
def __init__(self, d, options, verifyHostKey, userAuthObject):
19
self.options = options
20
self.verifyHostKey = verifyHostKey
21
self.userAuthObject = userAuthObject
23
def clientConnectionLost(self, connector, reason):
24
if self.options['reconnect']:
27
def clientConnectionFailed(self, connector, reason):
33
def buildProtocol(self, addr):
34
trans = SSHClientTransport(self)
35
if self.options['ciphers']:
36
trans.supportedCiphers = self.options['ciphers']
37
if self.options['macs']:
38
trans.supportedMACs = self.options['macs']
39
if self.options['compress']:
40
trans.supportedCompressions[0:1] = ['zlib']
41
if self.options['host-key-algorithms']:
42
trans.supportedPublicKeys = self.options['host-key-algorithms']
45
class SSHClientTransport(transport.SSHClientTransport):
47
def __init__(self, factory):
48
self.factory = factory
49
self.unixServer = None
51
def connectionLost(self, reason):
52
transport.SSHClientTransport.connectionLost(self, reason)
54
self.unixServer.stopListening()
55
self.unixServer = None
57
def receiveError(self, code, desc):
58
if not self.factory.d: return
61
d.errback(error.ConchError(desc, code))
63
def sendDisconnect(self, code, reason):
64
if not self.factory.d: return
67
transport.SSHClientTransport.sendDisconnect(self, code, reason)
68
d.errback(error.ConchError(reason, code))
70
def receiveDebug(self, alwaysDisplay, message, lang):
71
log.msg('Received Debug Message: %s' % message)
72
if alwaysDisplay: # XXX what should happen here?
75
def verifyHostKey(self, pubKey, fingerprint):
76
return self.factory.verifyHostKey(self, self.transport.getPeer().host, pubKey,
79
def setService(self, service):
80
log.msg('setting client server to %s' % service)
81
transport.SSHClientTransport.setService(self, service)
82
if service.name != 'ssh-userauth' and self.factory.d:
86
if service.name == 'ssh-connection':
88
if not self.factory.options['nocache']:
89
user = self.factory.userAuthObject.user
90
peer = self.transport.getPeer()
91
filename = os.path.expanduser("~/.conch-%s-%s-%i" % (user, peer.host, peer.port))
93
u = unix.SSHUnixServerFactory(service)
98
self.unixServer = reactor.listenUNIX(filename, u, mode=0600, wantPID=1)
100
log.msg('error trying to listen on %s' % filename)
103
def connectionSecure(self):
104
self.requestService(self.factory.userAuthObject)
107
def connect(host, port, options, verifyHostKey, userAuthObject):
109
factory = SSHClientFactory(d, options, verifyHostKey, userAuthObject)
110
reactor.connectTCP(host, port, factory)