#!/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