o j@sRddlmZddlmZmZmZmZddlmZddl Z ddl Z GdddZ dS))calcualte_length) LDAPMessageAuthenticationChoice protocolOp BindResponse)loggerNc@s<eZdZddZddZddZddZd d Zd d Zd S)LDAPRelayServerConnectioncCs2||_|j|_|jjd|_||_d|_d|_dS)Nz2NTLMSSP - Microsoft NTLM Security Support Providerr)settingsgssapiauthentication_contextsntlm connection%_LDAPRelayServerConnection__auth_type%_LDAPRelayServerConnection__auth_iter)selfr r r/root/aizidognhua/tmp/workspace/projects/ec89d86c-575f-41c9-af57-ac45cbdbf775/venv/lib/python3.10/site-packages/msldap/relay/serverconnection.py__init__s  z"LDAPRelayServerConnection.__init__csL|jjdurd|j|jf}|jj|||fIdHdSt||dS)Nz LDAPCON-%s:%s)r log_q client_ip client_portputrlog)rlevelmsgsrcrrr log_asyncs  z#LDAPRelayServerConnection.log_asynccs|jdSN)handle_in_taskcancelrrrr terminatesz#LDAPRelayServerConnection.terminatec s|j2zC3dHW}zt|}|djdkr"||IdHntd|jWqtyI}z|dt |IdHWYd}~dSd}~ww6dS)NrrzUnknown LDAP message! %s) r readrload_choice#_LDAPRelayServerConnection__bindreq Exceptionnativerstr)rmsg_datarerrr__handle_ldap_ins z*LDAPRelayServerConnection.__handle_ldap_inc sz:|jd}|jdd}t|tdurd|_|jdkrLdddd}d t|i}|t|d }t|}|j | IdH|jd 7_WdS|j |IdH\}} } | dur^| |durd dd d}d t|i}|t|d }t|}|j | IdH|IdHWdSd|dd}d t|i}|t|d }t|}|j | IdH|jd 7_WdSt|tdur7d|_|j |dIdH\}} } | dur| |dur d dd d}d t|i}|t|d }t|}|j | IdH|IdHWdSddd|d}d t|i}|t|d }t|}|j | IdH|jd 7_WdStd|tyT} z tWYd} ~ dSd} ~ ww)N messageIDrauthenticationTNTLMr) resultCode matchedDNdiagnosticMessage bindResponse)r-rr"1sX8009030C: LdapErr: DSID-0C090569, comment: AcceptSecurityContext error, data 52e, v4563GSSAPI credentials)r1r2r3serverSaslCredszUnknown auth method %s)r( isinstancebytesrrencoderrrr writedumpr authenticate_relay_serverr!dictr r' traceback print_exc) rrmsg_id authdata_rawtpobrespresdata to_conitnueerrr+rrr __bindreq's         z#LDAPRelayServerConnection.__bindreqcs"t||_|jIdHdSr)asyncio create_task*_LDAPRelayServerConnection__handle_ldap_inrr rrrrunszLDAPRelayServerConnection.runN) __name__ __module__ __qualname__rrr!rOr&rPrrrrrs kr) msldap.protocol.utilsrmsldap.protocol.messagesrrrrmsldaprrArMrrrrrs