From a7e24c736267d0cbf9ab2eedf2027b7d26e317bb Mon Sep 17 00:00:00 2001 From: Marc Hoersken Date: Thu, 12 Mar 2020 09:26:56 +0100 Subject: impacket: some more Python 3 code compatibility updates This makes smbserver load on Python 3, but still not work completely. --- tests/python_dependencies/impacket/smbserver.py | 17 +++++++++-------- 1 file changed, 9 insertions(+), 8 deletions(-) (limited to 'tests/python_dependencies/impacket/smbserver.py') diff --git a/tests/python_dependencies/impacket/smbserver.py b/tests/python_dependencies/impacket/smbserver.py index 902d22f9c..a74bd9414 100644 --- a/tests/python_dependencies/impacket/smbserver.py +++ b/tests/python_dependencies/impacket/smbserver.py @@ -29,9 +29,10 @@ import struct import sys if sys.version_info.major >= 3: import configparser + import socketserver else: import ConfigParser as configparser -import SocketServer + import SocketServer as socketserver import threading import logging import logging.config @@ -3519,7 +3520,7 @@ class Ioctls: return validateNegotiateInfo.getData(), errorCode -class SMBSERVERHandler(SocketServer.BaseRequestHandler): +class SMBSERVERHandler(socketserver.BaseRequestHandler): def __init__(self, request, client_address, server, select_poll = False): self.__SMB = server self.__ip, self.__port = client_address @@ -3528,7 +3529,7 @@ class SMBSERVERHandler(SocketServer.BaseRequestHandler): self.__timeOut = 60*5 self.__select_poll = select_poll #self.__connId = os.getpid() - SocketServer.BaseRequestHandler.__init__(self, request, client_address, server) + socketserver.BaseRequestHandler.__init__(self, request, client_address, server) def handle(self): self.__SMB.log("Incoming connection (%s,%d)" % (self.__ip, self.__port)) @@ -3570,13 +3571,13 @@ class SMBSERVERHandler(SocketServer.BaseRequestHandler): # Thread/process is dying, we should tell the main SMB thread to remove all this thread data self.__SMB.log("Closing down connection (%s,%d)" % (self.__ip, self.__port)) self.__SMB.removeConnection(self.__connId) - return SocketServer.BaseRequestHandler.finish(self) + return socketserver.BaseRequestHandler.finish(self) -class SMBSERVER(SocketServer.ThreadingMixIn, SocketServer.TCPServer): -#class SMBSERVER(SocketServer.ForkingMixIn, SocketServer.TCPServer): +class SMBSERVER(socketserver.ThreadingMixIn, socketserver.TCPServer): +#class SMBSERVER(socketserver.ForkingMixIn, socketserver.TCPServer): def __init__(self, server_address, handler_class=SMBSERVERHandler, config_parser = None): - SocketServer.TCPServer.allow_reuse_address = True - SocketServer.TCPServer.__init__(self, server_address, handler_class) + socketserver.TCPServer.allow_reuse_address = True + socketserver.TCPServer.__init__(self, server_address, handler_class) # Server name and OS to be presented whenever is necessary self.__serverName = '' -- cgit v1.2.3