root / trunk / twisted / conch / manhole_ssh.py

Revision 24441, 5.0 kB (checked in by thijs, 1 year ago)

Merge maintainer-email-2438: Get rid of references to maintainer email addresses from code.

Author: thijs
Reviewer: exarkun
Fixes: #2438

Line 
1 # -*- test-case-name: twisted.conch.test.test_manhole -*-
2 # Copyright (c) 2001-2004 Twisted Matrix Laboratories.
3 # See LICENSE for details.
4
5 """
6 insults/SSH integration support.
7
8 @author: Jp Calderone
9 """
10
11 from zope.interface import implements
12
13 from twisted.conch import avatar, interfaces as iconch, error as econch
14 from twisted.conch.ssh import factory, keys, session
15 from twisted.cred import credentials, checkers, portal
16 from twisted.python import components
17
18 from twisted.conch.insults import insults
19
20 class _Glue:
21     """A feeble class for making one attribute look like another.
22
23     This should be replaced with a real class at some point, probably.
24     Try not to write new code that uses it.
25     """
26     def __init__(self, **kw):
27         self.__dict__.update(kw)
28
29     def __getattr__(self, name):
30         raise AttributeError(self.name, "has no attribute", name)
31
32 class TerminalSessionTransport:
33     def __init__(self, proto, chainedProtocol, avatar, width, height):
34         self.proto = proto
35         self.avatar = avatar
36         self.chainedProtocol = chainedProtocol
37
38         session = self.proto.session
39
40         self.proto.makeConnection(
41             _Glue(write=self.chainedProtocol.dataReceived,
42                   loseConnection=lambda: avatar.conn.sendClose(session),
43                   name="SSH Proto Transport"))
44
45         def loseConnection():
46             self.proto.loseConnection()
47
48         self.chainedProtocol.makeConnection(
49             _Glue(write=self.proto.write,
50                   loseConnection=loseConnection,
51                   name="Chained Proto Transport"))
52
53         # XXX TODO
54         # chainedProtocol is supposed to be an ITerminalTransport,
55         # maybe.  That means perhaps its terminalProtocol attribute is
56         # an ITerminalProtocol, it could be.  So calling terminalSize
57         # on that should do the right thing But it'd be nice to clean
58         # this bit up.
59         self.chainedProtocol.terminalProtocol.terminalSize(width, height)
60
61 class TerminalSession(components.Adapter):
62     implements(iconch.ISession)
63
64     transportFactory = TerminalSessionTransport
65     chainedProtocolFactory = insults.ServerProtocol
66
67     def getPty(self, term, windowSize, attrs):
68         self.height, self.width = windowSize[:2]
69
70     def openShell(self, proto):
71         self.transportFactory(
72             proto, self.chainedProtocolFactory(),
73             iconch.IConchUser(self.original),
74             self.width, self.height)
75
76     def execCommand(self, proto, cmd):
77         raise econch.ConchError("Cannot execute commands")
78
79     def closed(self):
80         pass
81
82 class TerminalUser(avatar.ConchUser, components.Adapter):
83     def __init__(self, original, avatarId):
84         components.Adapter.__init__(self, original)
85         avatar.ConchUser.__init__(self)
86         self.channelLookup['session'] = session.SSHSession
87
88 class TerminalRealm:
89     userFactory = TerminalUser
90     sessionFactory = TerminalSession
91
92     transportFactory = TerminalSessionTransport
93     chainedProtocolFactory = insults.ServerProtocol
94
95     def _getAvatar(self, avatarId):
96         comp = components.Componentized()
97         user = self.userFactory(comp, avatarId)
98         sess = self.sessionFactory(comp)
99
100         sess.transportFactory = self.transportFactory
101         sess.chainedProtocolFactory = self.chainedProtocolFactory
102
103         comp.setComponent(iconch.IConchUser, user)
104         comp.setComponent(iconch.ISession, sess)
105
106         return user
107
108     def __init__(self, transportFactory=None):
109         if transportFactory is not None:
110             self.transportFactory = transportFactory
111
112     def requestAvatar(self, avatarId, mind, *interfaces):
113         for i in interfaces:
114             if i is iconch.IConchUser:
115                 return (iconch.IConchUser,
116                         self._getAvatar(avatarId),
117                         lambda: None)
118         raise NotImplementedError()
119
120 class ConchFactory(factory.SSHFactory):
121     publicKey = 'ssh-rsa AAAAB3NzaC1yc2EAAAABIwAAAGEArzJx8OYOnJmzf4tfBEvLi8DVPrJ3/c9k2I/Az64fxjHf9imyRJbixtQhlH9lfNjUIx+4LmrJH5QNRsFporcHDKOTwTTYLh5KmRpslkYHRivcJSkbh/C+BR3utDS555mV'
122
123     publicKeys = {
124         'ssh-rsa' : keys.Key.fromString(publicKey)
125     }
126     del publicKey
127
128     privateKey = """-----BEGIN RSA PRIVATE KEY-----
129 MIIByAIBAAJhAK8ycfDmDpyZs3+LXwRLy4vA1T6yd/3PZNiPwM+uH8Yx3/YpskSW
130 4sbUIZR/ZXzY1CMfuC5qyR+UDUbBaaK3Bwyjk8E02C4eSpkabJZGB0Yr3CUpG4fw
131 vgUd7rQ0ueeZlQIBIwJgbh+1VZfr7WftK5lu7MHtqE1S1vPWZQYE3+VUn8yJADyb
132 Z4fsZaCrzW9lkIqXkE3GIY+ojdhZhkO1gbG0118sIgphwSWKRxK0mvh6ERxKqIt1
133 xJEJO74EykXZV4oNJ8sjAjEA3J9r2ZghVhGN6V8DnQrTk24Td0E8hU8AcP0FVP+8
134 PQm/g/aXf2QQkQT+omdHVEJrAjEAy0pL0EBH6EVS98evDCBtQw22OZT52qXlAwZ2
135 gyTriKFVoqjeEjt3SZKKqXHSApP/AjBLpF99zcJJZRq2abgYlf9lv1chkrWqDHUu
136 DZttmYJeEfiFBBavVYIF1dOlZT0G8jMCMBc7sOSZodFnAiryP+Qg9otSBjJ3bQML
137 pSTqy7c3a2AScC/YyOwkDaICHnnD3XyjMwIxALRzl0tQEKMXs6hH8ToUdlLROCrP
138 EhQ0wahUTCk1gKA4uPD6TMTChavbh4K63OvbKg==
139 -----END RSA PRIVATE KEY-----"""
140     privateKeys = {
141         'ssh-rsa' : keys.Key.fromString(privateKey)
142     }
143     del privateKey
144
145     def __init__(self, portal):
146         self.portal = portal
Note: See TracBrowser for help on using the browser.