OwlCyberSecurity - MANAGER
Edit File: auth.cpython-34.pyc
� �Re� � @ s� d Z d d l Z d d l Z d d l Z d d l Z d d l m Z d d l m Z m Z d d l m Z d d l m Z d Z d Z d d � Z Gd d � d e � Z Gd d � d e � Z Gd d � d e � Z Gd d � d e � Z d S)z] requests.auth ~~~~~~~~~~~~~ This module contains the authentication handlers for Requests. � N)� b64encode� )�urlparse�str)�extract_cookies_to_jar)�parse_dict_headerz!application/x-www-form-urlencodedzmultipart/form-datac C s0 d t d | | f j d � � j � j d � S)z Returns a Basic Auth string.zBasic z%s:%s�latin1)r �encode�strip�decode)�username�password� r �4/tmp/pip-ztkk0jow-build/pip/_vendor/requests/auth.py�_basic_auth_str s r c @ s" e Z d Z d Z d d � Z d S)�AuthBasez4Base class that all auth implementations derive fromc C s t d � � d S)Nz Auth hooks must be callable.)�NotImplementedError)�self�rr r r �__call__" s zAuthBase.__call__N)�__name__� __module__� __qualname__�__doc__r r r r r r s r c @ s. e Z d Z d Z d d � Z d d � Z d S)� HTTPBasicAuthz?Attaches HTTP Basic Authentication to the given Request object.c C s | | _ | | _ d S)N)r r )r r r r r r �__init__( s zHTTPBasicAuth.__init__c C s t | j | j � | j d <| S)N� Authorization)r r r �headers)r r r r r r , s zHTTPBasicAuth.__call__N)r r r r r r r r r r r & s r c @ s" e Z d Z d Z d d � Z d S)� HTTPProxyAuthz=Attaches HTTP Proxy Authentication to a given Request object.c C s t | j | j � | j d <| S)NzProxy-Authorization)r r r r )r r r r r r 3 s zHTTPProxyAuth.__call__N)r r r r r r r r r r 1 s r c @ sF e Z d Z d Z d d � Z d d � Z d d � Z d d � Z d S) �HTTPDigestAuthz@Attaches HTTP Digest Authentication to the given Request object.c C s: | | _ | | _ d | _ d | _ i | _ d | _ d S)N� r )r r � last_nonce� nonce_count�chal�pos)r r r r r r r : s zHTTPDigestAuth.__init__c s6 | j d } | j d } | j j d � } | j j d � } | j j d � } | d k re d } n | j � } | d k s� | d k r� d d � } | � n! | d k r� d d � } | � n � f d d � } � d k r� d Sd } t | � } | j } | j r| d | j 7} n d | j | | j f } d | | f } � | � } � | � } | | j k r{| j d 7_ n d | _ d | j } t | j � j d � } | | j d � 7} | t j � j d � 7} | t j d � 7} t j | � j � d d � } d | | | | | f } | d k rH� d | | | f � } n | d k rp| | d | | f � } n7 | d k s�d | j d � k r�| | | � } n d S| | _ d | j | | | | f } | r�| d | 7} n | r�| d | 7} n | r| d | 7} n | r.| d | | f 7} n d | S) N�realm�nonce�qop� algorithm�opaque�MD5zMD5-SESSc S s4 t | t � r! | j d � } n t j | � j � S)Nzutf-8)� isinstancer r �hashlib�md5� hexdigest)�xr r r �md5_utf8P s z4HTTPDigestAuth.build_digest_header.<locals>.md5_utf8�SHAc S s4 t | t � r! | j d � } n t j | � j � S)Nzutf-8)r+ r r r, �sha1r. )r/ r r r �sha_utf8V s z4HTTPDigestAuth.build_digest_header.<locals>.sha_utf8c s � d | | f � S)Nz%s:%sr )�s�d)� hash_utf8r r �<lambda>\ s z4HTTPDigestAuth.build_digest_header.<locals>.<lambda>�?z%s:%s:%sz%s:%sr z%08xzutf-8� � z%s:%s:%s:%s:%s�auth�,z>username="%s", realm="%s", nonce="%s", uri="%s", response="%s"z , opaque="%s"z, algorithm="%s"z , digest="%s"z , qop="auth", nc=%s, cnonce="%s"z Digest %s)r# �get�upperr �path�queryr r r! r" r r �time�ctime�os�urandomr, r2 r. �split)r �method�urlr% r& r' r( r) � _algorithmr0 r3 �KD�entdig�p_parsedr? �A1�A2�HA1�HA2�ncvaluer4 �cnonce�noncebit�respdig�baser )r6 r �build_digest_headerB sn ! z"HTTPDigestAuth.build_digest_headerc K sk | j d k r( | j j j | j � n t | d d � } | j j d d � } d | j � k rW| d k rWt | d | d � t j d d t j �} t | j d | d d �� | _ | j | j j � | j j � } t | j | j | j � | j | j � | j | j | j � | j d <| j j | | � } | j j | � | | _ | St | d d � | S) z:Takes the given response and tries digest-auth, if needed.N� num_401_callsr zwww-authenticater �digest� zdigest �flags�countr ) r$ �request�body�seek�getattrr r= �lower�setattr�re�compile� IGNORECASEr �subr# �content�raw� release_conn�copyr �_cookies�prepare_cookiesrU rF rG � connection�send�history�append)r r �kwargsrV �s_auth�pat�prep�_rr r r � handle_401� s* ! zHTTPDigestAuth.handle_401c C sm | j r+ | j | j | j � | j d <n y | j j � | _ Wn t k rU Yn X| j d | j � | S)Nr �response) r! rU rF rG r r\ �tellr$ �AttributeError� register_hookrt )r r r r r r � s " zHTTPDigestAuth.__call__N)r r r r r rU rt r r r r r r 8 s S #r )r rC ra rA r, �base64r �compatr r �cookiesr �utilsr � CONTENT_TYPE_FORM_URLENCODED�CONTENT_TYPE_MULTI_PARTr �objectr r r r r r r r �<module> s