38 lines
1.2 KiB
Python
38 lines
1.2 KiB
Python
|
# -*- test-case-name: twisted.conch.test.test_conch -*-
|
||
|
|
from interfaces import IConchUser
|
||
|
|
from error import ConchError
|
||
|
|
from ssh.connection import OPEN_UNKNOWN_CHANNEL_TYPE
|
||
|
|
from twisted.python import log
|
||
|
|
from zope import interface
|
||
|
|
|
||
|
|
class ConchUser:
|
||
|
|
interface.implements(IConchUser)
|
||
|
|
|
||
|
|
def __init__(self):
|
||
|
|
self.channelLookup = {}
|
||
|
|
self.subsystemLookup = {}
|
||
|
|
|
||
|
|
def lookupChannel(self, channelType, windowSize, maxPacket, data):
|
||
|
|
klass = self.channelLookup.get(channelType, None)
|
||
|
|
if not klass:
|
||
|
|
raise ConchError(OPEN_UNKNOWN_CHANNEL_TYPE, "unknown channel")
|
||
|
|
else:
|
||
|
|
return klass(remoteWindow=windowSize,
|
||
|
|
remoteMaxPacket=maxPacket,
|
||
|
|
data=data, avatar=self)
|
||
|
|
|
||
|
|
def lookupSubsystem(self, subsystem, data):
|
||
|
|
log.msg(repr(self.subsystemLookup))
|
||
|
|
klass = self.subsystemLookup.get(subsystem, None)
|
||
|
|
if not klass:
|
||
|
|
return False
|
||
|
|
return klass(data, avatar=self)
|
||
|
|
|
||
|
|
def gotGlobalRequest(self, requestType, data):
|
||
|
|
# XXX should this use method dispatch?
|
||
|
|
requestType = requestType.replace('-','_')
|
||
|
|
f = getattr(self, "global_%s" % requestType, None)
|
||
|
|
if not f:
|
||
|
|
return 0
|
||
|
|
return f(data)
|