mirror of
https://github.com/aljazceru/cowrie.git
synced 2026-02-05 14:34:24 +01:00
split off session.py
This commit is contained in:
Michel Oosterhof
@@ -15,6 +15,7 @@ from twisted.python import log, components
|
||||
|
||||
from cowrie.core import pwd
|
||||
from cowrie.core import ssh
|
||||
from cowrie.core import session
|
||||
|
||||
|
||||
@implementer(IConchUser)
|
||||
@@ -29,7 +30,7 @@ class CowrieUser(avatar.ConchUser):
|
||||
self.cfg = self.server.cfg
|
||||
|
||||
self.channelLookup.update(
|
||||
{"session": ssh.HoneyPotSSHSession,
|
||||
{"session": session.HoneyPotSSHSession,
|
||||
"direct-tcpip": ssh.CowrieOpenConnectForwardingClient})
|
||||
|
||||
try:
|
||||
@@ -57,6 +58,6 @@ class CowrieUser(avatar.ConchUser):
|
||||
|
||||
|
||||
components.registerAdapter(ssh.SFTPServerForCowrieUser, CowrieUser, ISFTPServer)
|
||||
components.registerAdapter(ssh.SSHSessionForCowrieUser, CowrieUser, ISession)
|
||||
components.registerAdapter(session.SSHSessionForCowrieUser, CowrieUser, ISession)
|
||||
|
||||
|
||||
|
||||
167
cowrie/core/session.py
Normal file
167
cowrie/core/session.py
Normal file
@@ -0,0 +1,167 @@
|
||||
# Copyright (c) 2009-2014 Upi Tamminen <desaster@gmail.com>
|
||||
# See the COPYRIGHT file for more information
|
||||
|
||||
"""
|
||||
This module contains ...
|
||||
"""
|
||||
|
||||
import os
|
||||
|
||||
from zope.interface import implementer
|
||||
|
||||
import twisted
|
||||
from twisted.conch.interfaces import ISession
|
||||
from twisted.conch.ssh import session
|
||||
from twisted.python import log
|
||||
from twisted.conch.ssh.common import getNS
|
||||
|
||||
from cowrie.core import pwd
|
||||
from cowrie.core import protocol
|
||||
|
||||
|
||||
class HoneyPotSSHSession(session.SSHSession):
|
||||
"""
|
||||
This is an SSH channel that's used for SSH sessions
|
||||
"""
|
||||
|
||||
def __init__(self, *args, **kw):
|
||||
session.SSHSession.__init__(self, *args, **kw)
|
||||
#self.__dict__['request_auth_agent_req@openssh.com'] = self.request_agent
|
||||
|
||||
|
||||
def request_env(self, data):
|
||||
"""
|
||||
"""
|
||||
name, rest = getNS(data)
|
||||
value, rest = getNS(rest)
|
||||
if rest:
|
||||
raise ValueError("Bad data given in env request")
|
||||
log.msg(eventid='COW0013', format='request_env: %(name)s=%(value)s',
|
||||
name=name, value=value)
|
||||
# Environment variables come after shell or before exec command
|
||||
if self.session:
|
||||
self.session.environ[name] = value
|
||||
return 0
|
||||
|
||||
|
||||
def request_agent(self, data):
|
||||
"""
|
||||
"""
|
||||
log.msg('request_agent: %s' % (repr(data),))
|
||||
return 0
|
||||
|
||||
|
||||
def request_x11_req(self, data):
|
||||
"""
|
||||
"""
|
||||
log.msg('request_x11: %s' % (repr(data),))
|
||||
return 0
|
||||
|
||||
|
||||
def closed(self):
|
||||
"""
|
||||
This is reliably called on session close/disconnect and calls the avatar
|
||||
"""
|
||||
session.SSHSession.closed(self)
|
||||
|
||||
|
||||
def sendEOF(self):
|
||||
"""
|
||||
Utility function to request to send EOF for this session
|
||||
"""
|
||||
self.conn.sendEOF(self)
|
||||
|
||||
|
||||
def sendClose(self):
|
||||
"""
|
||||
Utility function to request to send close for this session
|
||||
"""
|
||||
self.conn.sendClose(self)
|
||||
|
||||
|
||||
def channelClosed(self):
|
||||
"""
|
||||
"""
|
||||
log.msg("Called channelClosed in SSHSession")
|
||||
|
||||
|
||||
|
||||
@implementer(ISession)
|
||||
class SSHSessionForCowrieUser:
|
||||
"""
|
||||
"""
|
||||
|
||||
def __init__(self, avatar, reactor=None):
|
||||
"""
|
||||
Construct an C{SSHSessionForCowrieUser}.
|
||||
|
||||
@param avatar: The L{CowrieUser} for whom this is an SSH session.
|
||||
@param reactor: An L{IReactorProcess} used to handle shell and exec
|
||||
requests. Uses the default reactor if None.
|
||||
"""
|
||||
self.protocol = None
|
||||
self.avatar = avatar
|
||||
self.server = avatar.server
|
||||
self.cfg = avatar.cfg
|
||||
self.uid = avatar.uid
|
||||
self.gid = avatar.gid
|
||||
self.username = avatar.username
|
||||
self.environ = {
|
||||
'LOGNAME': self.username,
|
||||
'USER': self.username,
|
||||
'HOME': self.avatar.home}
|
||||
if self.uid==0:
|
||||
self.environ['PATH']='/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin'
|
||||
else:
|
||||
self.environ['PATH']='/usr/local/bin:/usr/bin:/bin:/usr/local/games:/usr/games'
|
||||
|
||||
def openShell(self, processprotocol):
|
||||
"""
|
||||
"""
|
||||
self.protocol = protocol.LoggingServerProtocol(
|
||||
protocol.HoneyPotInteractiveProtocol, self)
|
||||
self.protocol.makeConnection(processprotocol)
|
||||
processprotocol.makeConnection(session.wrapProtocol(self.protocol))
|
||||
|
||||
|
||||
def getPty(self, terminal, windowSize, attrs):
|
||||
"""
|
||||
"""
|
||||
self.environ['TERM'] = terminal
|
||||
log.msg(eventid='COW0010', width=windowSize[0], height=windowSize[1],
|
||||
format='Terminal Size: %(width)s %(height)s')
|
||||
self.windowSize = windowSize
|
||||
return None
|
||||
|
||||
|
||||
def execCommand(self, processprotocol, cmd):
|
||||
"""
|
||||
"""
|
||||
self.protocol = protocol.LoggingServerProtocol(
|
||||
protocol.HoneyPotExecProtocol, self, cmd)
|
||||
self.protocol.makeConnection(processprotocol)
|
||||
processprotocol.makeConnection(session.wrapProtocol(self.protocol))
|
||||
|
||||
|
||||
def closed(self):
|
||||
"""
|
||||
this is reliably called on both logout and disconnect
|
||||
we notify the protocol here we lost the connection
|
||||
"""
|
||||
if self.protocol:
|
||||
self.protocol.connectionLost("disconnected")
|
||||
self.protocol = None
|
||||
|
||||
|
||||
def eofReceived(self):
|
||||
"""
|
||||
"""
|
||||
if self.protocol:
|
||||
self.protocol.eofReceived()
|
||||
|
||||
|
||||
def windowChanged(self, windowSize):
|
||||
"""
|
||||
"""
|
||||
self.windowSize = windowSize
|
||||
|
||||
@@ -10,166 +10,12 @@ import os
|
||||
from zope.interface import implementer
|
||||
|
||||
import twisted
|
||||
from twisted.conch.interfaces import ISFTPFile, ISFTPServer, ISession
|
||||
from twisted.conch.ssh import session
|
||||
from twisted.conch.interfaces import ISFTPFile, ISFTPServer
|
||||
from twisted.conch.ssh import filetransfer
|
||||
from twisted.conch.ssh import forwarding
|
||||
from twisted.conch.ssh.filetransfer import FXF_READ, FXF_WRITE, FXF_APPEND, FXF_CREAT, FXF_TRUNC, FXF_EXCL
|
||||
import twisted.conch.ls
|
||||
from twisted.python import log
|
||||
from twisted.conch.ssh.common import getNS
|
||||
|
||||
from cowrie.core import pwd
|
||||
from cowrie.core import protocol
|
||||
|
||||
|
||||
|
||||
class HoneyPotSSHSession(session.SSHSession):
|
||||
"""
|
||||
This is an SSH channel that's used for SSH sessions
|
||||
"""
|
||||
|
||||
def __init__(self, *args, **kw):
|
||||
session.SSHSession.__init__(self, *args, **kw)
|
||||
#self.__dict__['request_auth_agent_req@openssh.com'] = self.request_agent
|
||||
|
||||
|
||||
def request_env(self, data):
|
||||
"""
|
||||
"""
|
||||
name, rest = getNS(data)
|
||||
value, rest = getNS(rest)
|
||||
if rest:
|
||||
raise ValueError("Bad data given in env request")
|
||||
log.msg(eventid='COW0013', format='request_env: %(name)s=%(value)s',
|
||||
name=name, value=value)
|
||||
# Environment variables come after shell or before exec command
|
||||
if self.session:
|
||||
self.session.environ[name] = value
|
||||
return 0
|
||||
|
||||
|
||||
def request_agent(self, data):
|
||||
"""
|
||||
"""
|
||||
log.msg('request_agent: %s' % (repr(data),))
|
||||
return 0
|
||||
|
||||
|
||||
def request_x11_req(self, data):
|
||||
"""
|
||||
"""
|
||||
log.msg('request_x11: %s' % (repr(data),))
|
||||
return 0
|
||||
|
||||
|
||||
def closed(self):
|
||||
"""
|
||||
This is reliably called on session close/disconnect and calls the avatar
|
||||
"""
|
||||
session.SSHSession.closed(self)
|
||||
|
||||
|
||||
def sendEOF(self):
|
||||
"""
|
||||
Utility function to request to send EOF for this session
|
||||
"""
|
||||
self.conn.sendEOF(self)
|
||||
|
||||
|
||||
def sendClose(self):
|
||||
"""
|
||||
Utility function to request to send close for this session
|
||||
"""
|
||||
self.conn.sendClose(self)
|
||||
|
||||
|
||||
def channelClosed(self):
|
||||
"""
|
||||
"""
|
||||
log.msg("Called channelClosed in SSHSession")
|
||||
|
||||
|
||||
|
||||
@implementer(ISession)
|
||||
class SSHSessionForCowrieUser:
|
||||
"""
|
||||
"""
|
||||
|
||||
def __init__(self, avatar, reactor=None):
|
||||
"""
|
||||
Construct an C{SSHSessionForCowrieUser}.
|
||||
|
||||
@param avatar: The L{CowrieUser} for whom this is an SSH session.
|
||||
@param reactor: An L{IReactorProcess} used to handle shell and exec
|
||||
requests. Uses the default reactor if None.
|
||||
"""
|
||||
self.protocol = None
|
||||
self.avatar = avatar
|
||||
self.server = avatar.server
|
||||
self.cfg = avatar.cfg
|
||||
self.uid = avatar.uid
|
||||
self.gid = avatar.gid
|
||||
self.username = avatar.username
|
||||
self.environ = {
|
||||
'LOGNAME': self.username,
|
||||
'USER': self.username,
|
||||
'HOME': self.avatar.home}
|
||||
if self.uid==0:
|
||||
self.environ['PATH']='/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin'
|
||||
else:
|
||||
self.environ['PATH']='/usr/local/bin:/usr/bin:/bin:/usr/local/games:/usr/games'
|
||||
|
||||
def openShell(self, processprotocol):
|
||||
"""
|
||||
"""
|
||||
self.protocol = protocol.LoggingServerProtocol(
|
||||
protocol.HoneyPotInteractiveProtocol, self)
|
||||
self.protocol.makeConnection(processprotocol)
|
||||
processprotocol.makeConnection(session.wrapProtocol(self.protocol))
|
||||
|
||||
|
||||
def getPty(self, terminal, windowSize, attrs):
|
||||
"""
|
||||
"""
|
||||
self.environ['TERM'] = terminal
|
||||
log.msg(eventid='COW0010', width=windowSize[0], height=windowSize[1],
|
||||
format='Terminal Size: %(width)s %(height)s')
|
||||
self.windowSize = windowSize
|
||||
return None
|
||||
|
||||
|
||||
def execCommand(self, processprotocol, cmd):
|
||||
"""
|
||||
"""
|
||||
self.protocol = protocol.LoggingServerProtocol(
|
||||
protocol.HoneyPotExecProtocol, self, cmd)
|
||||
self.protocol.makeConnection(processprotocol)
|
||||
processprotocol.makeConnection(session.wrapProtocol(self.protocol))
|
||||
|
||||
|
||||
def closed(self):
|
||||
"""
|
||||
this is reliably called on both logout and disconnect
|
||||
we notify the protocol here we lost the connection
|
||||
"""
|
||||
if self.protocol:
|
||||
self.protocol.connectionLost("disconnected")
|
||||
self.protocol = None
|
||||
|
||||
|
||||
def eofReceived(self):
|
||||
"""
|
||||
"""
|
||||
if self.protocol:
|
||||
self.protocol.eofReceived()
|
||||
|
||||
|
||||
def windowChanged(self, windowSize):
|
||||
"""
|
||||
"""
|
||||
self.windowSize = windowSize
|
||||
|
||||
|
||||
|
||||
@implementer(ISFTPFile)
|
||||
|
||||
Reference in New Issue
Block a user