NetExec/cme/servers/smb.py

42 lines
1.4 KiB
Python
Executable File

import threading
import logging
from sys import exit
from impacket import smbserver
class CMESMBServer(threading.Thread):
def __init__(self, logger, share_name, share_path='/tmp/cme_hosted', listen_address='0.0.0.0', listen_port=445, verbose=False):
try:
threading.Thread.__init__(self)
self.server = smbserver.SimpleSMBServer(listen_address, listen_port)
self.server.addShare(share_name.upper(), share_path)
if verbose: self.server.setLogFile('')
self.server.setSMB2Support(True)
self.server.setSMBChallenge('')
except Exception as e:
errno, message = e.args
if errno == 98 and message == 'Address already in use':
logger.error('Error starting SMB server on port 445: the port is already in use')
else:
logger.error('Error starting SMB server on port 445: {}'.format(message))
exit(1)
def addShare(self, share_name, share_path):
self.server.addShare(share_name, share_path)
def run(self):
try:
self.server.start()
except:
pass
def shutdown(self):
# TODO: should fine the proper way
# make sure all the threads are killed
for thread in threading.enumerate():
if thread.isAlive():
try:
self._stop()
except:
pass