仅使用标准库的python http NTLM身份验证

时间:2014-07-22 20:37:43

标签: python windows-authentication ntlm ntlmv2

我对没有使用第三方python ntlm库的http ntlm身份验证的python实现感兴趣。

1 个答案:

答案 0 :(得分:1)

只要用户名不是域的一部分,这就有效。在某些情况下,您可能需要传递ntlm='NTLM'而不是Negotiate

def ntlm_authenticate_message(nonce, username, password):
    # http://davenport.sourceforge.net/ntlm.html#theLmv2Response
    import hashlib
    import hmac
    import os
    ntlm_hash = hashlib.new('md4', password.encode('utf-16le')).digest()
    ntlm_v2 = hmac.new(ntlm_hash, username.upper().encode('utf-16le')).digest()
    rand = os.urandom(8)
    lm = hmac.new(ntlm_v2, nonce + rand).digest() + rand
    return ('NTLMSSP\0\3\0\0\0' + '\x18\0\x18\0\x40\0\0\0' + '\0' * 8 + '\0' * 8 +
            chr(len(username)) + '\0' + chr(len(username)) + '\0' + '\x58\0\0\0' +
            '\0' * 8 + '\0' * 8 + '\0' * 4 + lm + username).encode('base64').replace('\n', '')


def http_ntlm_auth(url, data, headers, username, password, ntlm='Negotiate'):
    # http://davenport.sourceforge.net/ntlm.html#ntlmHttpAuthentication
    u = urlparse(url)
    h = HTTPSConnection(u.netloc)
    # h.set_debuglevel(1)
    h.request('GET', u.path, None, {'Authorization': ntlm + ' ' + 'NTLMSSP\0\1\0\0\0\2\0\0\0'.encode('base64')})  # 1
    resp = h.getresponse()
    error = resp.read()
    assert resp.status == 401 and not error, (resp.status, resp.reason, error)
    nonce = resp.getheader('WWW-Authenticate').split()[1].decode('base64')[24:32]  # type 2
    headers['Authorization'] = ntlm + ' ' + ntlm_authenticate_message(nonce, username, password)  # type 3
    h.request('POST' if data else 'GET', u.path, data, headers)
    return h.getresponse()