Esempio n. 1
0
class SMBNTLMSSPI:
    def __init__(self, settings):
        self.settings = settings
        self.mode = None  #'CLIENT'
        self.sspi = NTLMSMBSSPI()
        self.client = None
        self.target = None

        self.setup()

    def setup(self):
        self.mode = self.settings.mode.upper()
        self.client = self.settings.client
        self.password = self.settings.password

    def get_session_key(self):
        return self.sspi.get_session_key()

    async def encrypt(self, data, message_no):
        return self.sspi.encrypt(data, message_no)

    async def decrypt(self, data, message_no):
        return self.sspi.decrypt(data, message_no)

    async def authenticate(self, authData=None):
        if self.mode == 'CLIENT':
            if authData is None:
                return self.sspi.negotiate()
            else:
                return self.sspi.authenticate(authData)
Esempio n. 2
0
    def __init__(self, settings):
        self.settings = settings
        self.mode = None  #'CLIENT'
        self.sspi = NTLMSMBSSPI()
        self.client = None
        self.target = None

        self.setup()
Esempio n. 3
0
    def __init__(self, settings):
        self.settings = settings
        self.mode = None  #'CLIENT'
        self.sspi = NTLMSMBSSPI()
        self.client = None
        self.target = None
        #self.ntlmChallenge = None

        self.session_key = None
        self.ntlm_ctx = NTLMAUTHHandler(NTLMHandlerSettings(None, 'MANUAL'))

        self.setup()
Esempio n. 4
0
class SMBNTLMSSPI:
    def __init__(self, settings):
        self.settings = settings
        self.mode = None  #'CLIENT'
        self.sspi = NTLMSMBSSPI()
        self.client = None
        self.target = None
        #self.ntlmChallenge = None

        self.session_key = None
        self.ntlm_ctx = NTLMAUTHHandler(NTLMHandlerSettings(None, 'MANUAL'))

        self.setup()

    @property
    def ntlmChallenge(self):
        return self.ntlm_ctx.ntlmChallenge

    def setup(self):
        self.mode = self.settings.mode.upper()
        self.client = self.settings.client
        self.password = self.settings.password

    def get_sealkey(self, mode='Client'):
        return self.ntlm_ctx.get_sealkey(mode=mode)

    def get_signkey(self, mode='Client'):
        return self.ntlm_ctx.get_signkey(mode=mode)

    def SEAL(self, signingKey, sealingKey, messageToSign, messageToEncrypt,
             seqNum, cipher_encrypt):
        return self.ntlm_ctx.SEAL(signingKey, sealingKey, messageToSign,
                                  messageToEncrypt, seqNum, cipher_encrypt)

    def SIGN(self, signingKey, message, seqNum, cipher_encrypt):
        return self.ntlm_ctx.SIGN(signingKey, message, seqNum, cipher_encrypt)

    def get_session_key(self):
        if not self.session_key:
            self.session_key = self.sspi.get_session_key()

        return self.session_key

    def get_extra_info(self):
        return self.ntlm_ctx.get_extra_info()

    def is_extended_security(self):
        return self.ntlm_ctx.is_extended_security()

    async def encrypt(self, data, message_no):
        return self.sspi.encrypt(data, message_no)

    async def decrypt(self, data, message_no):
        return self.sspi.decrypt(data, message_no)

    async def authenticate(self,
                           authData=None,
                           flags=None,
                           seq_number=0,
                           is_rpc=False):
        if self.mode == 'CLIENT':
            if authData is None:
                data, res = self.sspi.negotiate(is_rpc=is_rpc)
                self.ntlm_ctx.load_negotiate(data)
                return data, res
            else:
                self.ntlm_ctx.load_challenge(authData)
                data, res = self.sspi.authenticate(authData, is_rpc=is_rpc)
                self.ntlm_ctx.load_authenticate(data)
                self.ntlm_ctx.load_sessionkey(self.get_session_key())

                input(self.ntlm_ctx.get_extra_info())
                return data, res