45 lines
1.5 KiB
Python
Executable File
45 lines
1.5 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
# -*- coding: utf-8 -*-
|
|
|
|
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.is_alive():
|
|
try:
|
|
self._stop()
|
|
except:
|
|
pass
|