46 lines
1.4 KiB
Python
46 lines
1.4 KiB
Python
# -*- test-case-name: twisted.conch.test.test_conch -*-
|
|
|
|
from __future__ import absolute_import, division
|
|
|
|
from zope.interface import implementer
|
|
|
|
from twisted.conch.error import ConchError
|
|
from twisted.conch.interfaces import IConchUser
|
|
from twisted.conch.ssh.connection import OPEN_UNKNOWN_CHANNEL_TYPE
|
|
from twisted.python import log
|
|
from twisted.python.compat import nativeString
|
|
|
|
|
|
@implementer(IConchUser)
|
|
class ConchUser:
|
|
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 = nativeString(requestType.replace(b'-', b'_'))
|
|
f = getattr(self, "global_%s" % requestType, None)
|
|
if not f:
|
|
return 0
|
|
return f(data)
|