import logging
import sys
import threading
import os

from socket import error as socketerror
from mitmflib.impacket import version, smbserver, LOG
from core.servers.smb.KarmaSMB import KarmaSMBServer
from core.configwatcher import ConfigWatcher
from core.utils import shutdown

#Logging is something I'm going to have to clean up in the future

class SMBserver(ConfigWatcher):

    _instance    = None
    impacket_ver = version.VER_MINOR
    server_type  = ConfigWatcher.config["MITMf"]["SMB"]["type"].lower()
    smbchallenge = ConfigWatcher.config["MITMf"]["SMB"]["Challenge"]
    smb_port     = int(ConfigWatcher.config["MITMf"]["SMB"]["port"])

    @staticmethod
    def getInstance():
        if SMBserver._instance == None:
            SMBserver._instance = SMBserver()

        return SMBserver._instance

    def parseConfig(self):
        server = None
        try:
            if self.server_type == 'normal':

                formatter = logging.Formatter("%(asctime)s [SMBserver] %(message)s", datefmt="%Y-%m-%d %H:%M:%S")
                self.configureLogging(formatter)

                server = smbserver.SimpleSMBServer(listenPort=self.smb_port)
                
                for share in self.config["MITMf"]["SMB"]["Shares"]:
                    path = self.config["MITMf"]["SMB"]["Shares"][share]['path']
                    readonly = self.config["MITMf"]["SMB"]["Shares"][share]['readonly'].lower()
                    server.addShare(share.upper(), path, readOnly=readonly)

                server.setSMBChallenge(self.smbchallenge)
                server.setLogFile('')

            elif self.server_type == 'karma':

                formatter = logging.Formatter("%(asctime)s [KarmaSMB] %(message)s", datefmt="%Y-%m-%d %H:%M:%S")
                self.configureLogging(formatter)

                server = KarmaSMBServer(self.smbchallenge, self.smb_port)
                server.defaultFile = self.config["MITMf"]["SMB"]["Karma"]["defaultfile"]
                
                for extension, path in self.config["MITMf"]["SMB"]["Karma"].iteritems():
                    server.extensions[extension.upper()] = os.path.normpath(path)

            else:
                shutdown("\n[-] Invalid SMB server type specified in config file!")

            return server
        
        except socketerror as e:
            if "Address already in use" in e:
                shutdown("\n[-] Unable to start SMB server on port {}: port already in use".format(listenPort))

    def configureLogging(self, formatter):
        #yes I know this looks awful, yuck

        LOG.setLevel(logging.INFO)
        LOG.propagate = False
        logging.getLogger('smbserver').setLevel(logging.INFO)
        logging.getLogger('impacket').setLevel(logging.INFO)

        fileHandler = logging.FileHandler("./logs/mitmf.log")
        streamHandler = logging.StreamHandler(sys.stdout)
        fileHandler.setFormatter(formatter)
        streamHandler.setFormatter(formatter)
        LOG.addHandler(fileHandler)
        LOG.addHandler(streamHandler)

    def start(self):
        t = threading.Thread(name='SMBserver', target=self.parseConfig().start)
        t.setDaemon(True)
        t.start()