File: //usr/lib/python3/dist-packages/cloudinit/__pycache__/url_helper.cpython-39.pyc
a
��`�V � @ s� d dl Z d dlZd dlZd dlZd dlmZ d dlmZ d dlm Z d dl
m Z d dl m
Z
d dlmZmZmZ d dlZd dlmZ d d lmZ d d
lmZ e�e�Zd Zd Z dZ d Z zdd d
l m Z d dl!Z!e!�"d�Z#e e#j�Z e e d�kr�dZe d�e k�re d�k �rn ndZ W n e$�y. Y n0 dd� Z%dd� Z&dd� Z'G dd� de(�Z)G dd � d e)�Z*G d d � d e(�Z+G d d � d e,�Z-d!d"� Z.d.d$d%�Z/d/d&d'�Z0G d(d)� d)e(�Z1d0d*d+�Z2d,d-� Z3dS )1� N)� parsedate)�ENOENT)�partial)� NOT_FOUND)�count)�urlparse�
urlunparse�quote)�
exceptions)�log)�versionF�REDACTED)� LooseVersion�requestsz0.8.8Tz0.7.0z1.0.0c C s< t t| dd��}|d s4|d r4|d |d< d|d< t|�S )N�http)�scheme� � � )�listr r )�urlZ
parsed_url� r �6/usr/lib/python3/dist-packages/cloudinit/url_helper.py� _cleanurl1 s
r c G s$ dd� }| }|D ]}|||�}q|S )Nc S sN t t| ��}|d }|r*|�d�s*|d7 }|tt|�dd�7 }||d< t|�S )Nr �/z/:)�safe)r r �endswithr �strr )r �add_onZ
url_parsed�pathr r r �combine_single= s z#combine_url.<locals>.combine_singler )�baseZadd_onsr r r r r r � combine_url; s
r" c
K s� | � � } | �d�rd| } | �� �d�r�|�d�r>t�d| � | td�d� }z6t|d��}|�� }W d � n1 sx0 Y W nJ t y� } z2|j
}|j
t kr�t }t
||d| d�|�W Y d}~n
d}~0 0 t||d �S t| fi |��S dS )
a0 Wrapper function around readurl to allow passing a file path as url.
When url is not a local file path, passthrough any kwargs to readurl.
In the case of parameter passthrough to readurl, default values for some
parameters. See: call-signature of readurl in this module for param docs.
r z file://%szfile://�dataz'Unable to post data to file resource %sN�rb)�cause�code�headersr )�contents)�lstrip�
startswith�lower�get�LOG�warning�len�open�read�IOError�errnor r �UrlError� FileResponse�readurl)r �kwargs� file_path�fpr( �er& r r r �read_file_or_urlL s"
*
( r; c @ s& e Zd Zd dd�Zdd� Zdd� ZdS )
�StringResponse�� c C s || _ i | _|| _d | _d S �N)r& r' r( r )�selfr( r& r r r �__init__l s zStringResponse.__init__c O s | j dkrdS dS )Nr= FT�r& �r? �argsr7 r r r �okr s
zStringResponse.okc C s | j �d�S )Nzutf-8)r( �decode�r? r r r �__str__w s zStringResponse.__str__N)r= )�__name__�
__module__� __qualname__r@ rD rG r r r r r< k s
r< c @ s e Zd Zddd�ZdS )r5 r= c C s t j| ||d� || _d S )NrA )r< r@ r )r? r r( r& r r r r@ | s zFileResponse.__init__N)r= �rH rI rJ r@ r r r r r5 { s r5 c @ sV e Zd Zdd� Zedd� �Zedd� �Zddd �Zed
d � �Zed d
� �Z dd� Z
dS )� UrlResponsec C s
|| _ d S r> )� _response)r? �responser r r r@ � s zUrlResponse.__init__c C s | j jS r> )rM �contentrF r r r r( � s zUrlResponse.contentsc C s | j jS r> )rM r rF r r r r � s zUrlResponse.urlFc C s2 d}|r d}d| j kr"|k r*n ndS dS d S )Ni, i� r= TFrA )r? Z redirects_ok�upperr r r rD � s zUrlResponse.okc C s | j jS r> )rM r' rF r r r r' � s zUrlResponse.headersc C s | j jS r> )rM � status_coderF r r r r& � s zUrlResponse.codec C s | j jS r> )rM �textrF r r r rG � s zUrlResponse.__str__N)F) rH rI rJ r@ �propertyr( r rD r' r& rG r r r r rL � s
rL c @ s e Zd Zddd�ZdS )r4 Nc C s< t �| t|�� || _|| _|| _| jd u r2i | _|| _d S r> )r2 r@ r r% r& r' r )r? r% r& r' r r r r r@ � s
zUrlError.__init__)NNNrK r r r r r4 � s r4 c C s� i }t | �j}|dkr�|r�ts,t�dt� ndd|v rJ|d rJ|d |d< nd|d< d|v rxd|v rx|d |d g|d< nd|v r�t|d �|d< |S ) N�httpszGSSL is not supported in requests v%s, cert. verification can not occur!Zca_certs�verifyT� cert_file�key_fileZcert)r r � SSL_ENABLEDr- r. �_REQ_VERr )r � ssl_detailsZssl_argsr r r r �
_get_ssl_args� s"
��
r[ r c sX t | �} d| i}|�t| |�� |
|d< |s8|r4dnd}||d< |durZtt|�d�|d< |du rfg }trzd d
i}||d < d }|r�tt|�d d �}d
dt�� i}� r�|�� � |� |sʇ fdd�}|}|r�||d< |du r�d}g }|
r�t � nt
d|�D �]H}|| �|d< i }|� � D ]p\}�|dk�r.�q|dk�r~|�r~�fdd�|D �}|�r�t �
��||< |D ]}t|| |< �qhn�||< �qz�|�r�t�d||
�r�dn|| |� | du �r�t�� } | � }|jf i |��} W d � n1 �s�0 Y | �r| �� t�d| | jt| j�|d � t| �W S tj�yB } z�t| tj ��r�t | d��r�t | j d��r�|� t | | j j| j j!| d�� n6|� t | | d �� t"�r�t| tj#��r�W Y d} ~ �qF| �r�| |� � |d ��s�W Y d} ~ �qF|
�r�|dk�s|d |k �r.|dk�r.|�r$t�d |� t$�%|� W Y d} ~ q�d} ~ 0 0 q�|�rT|d �dS ) a� Wrapper around requests.Session to read the url and retry if necessary
:param url: Mandatory url to request.
:param data: Optional form data to post the URL. Will set request_method
to 'POST' if present.
:param timeout: Timeout in seconds to wait for a response
:param retries: Number of times to retry on exception if exception_cb is
None or exception_cb returns True for the exception caught. Default is
to fail with 0 retries on exception.
:param sec_between: Default 1: amount of seconds passed to time.sleep
between retries. None or -1 means don't sleep.
:param headers: Optional dict of headers to send during request
:param headers_cb: Optional callable returning a dict of values to send as
headers during request
:param headers_redact: Optional list of header names to redact from the log
:param ssl_details: Optional dict providing key_file, ca_certs, and
cert_file keys for use on in ssl connections.
:param check_status: Optional boolean set True to raise when HTTPError
occurs. Default: True.
:param allow_redirects: Optional boolean passed straight to Session.request
as 'allow_redirects'. Default: True.
:param exception_cb: Optional callable which accepts the params
msg and exception and returns a boolean True if retries are permitted.
:param session: Optional exiting requests.Session instance to reuse.
:param infinite: Bool, set True to retry indefinitely. Default: False.
:param log_req_resp: Set False to turn off verbose debug messages.
:param request_method: String passed as 'method' to Session.request.
Typically GET, or POST. Default: POST if data is provided, GET
otherwise.
r �allow_redirects�POST�GET�methodNr �timeoutZ
store_cookiesF�configr z
User-Agentz
Cloud-Init/%sc s � S r> r �r )r' r r �_cb
s zreadurl.<locals>._cbr# ���r' c s g | ]}� � |�r|�qS r )r, )�.0�k)�vr r �
<listcomp> � zreadurl.<locals>.<listcomp>z'[%s/%s] open '%s' with %s configuration�infinitez(Read from %s (%s, %sb) after %s attemptsrN rQ �r& r' r rb �1Please wait %s seconds while we wait to try again)&r �updater[ �max�float�CONFIG_ENABLED�intr �version_stringr �range�items�copy�deepcopyr
r- �debugr ZSession�requestZraise_for_statusrQ r/ rO rL r
ZRequestException�
isinstance� HTTPError�hasattrrN �appendr4 r' rX ZSSLError�time�sleep) r r# r` �retries� sec_betweenr' �
headers_cb�headers_redactrZ � check_statusr\ � exception_cbZsessionrj Z log_req_resp�request_methodZreq_argsZ
req_configZ manual_triesZ def_headersrc Zexcps�iZfiltered_req_argsrf Zmatched_headers�keyZsess�rr: r )r' rg r r6 � s� #� � �
�
0��
��
���"r6 c
C s, t � � }
ddd�} |du r | }dd� } d}
d}|durB|||
�}nt|
d �d }| D �]�}t � � }|
dkr�| ||
�r| �q�|dur�|r�|| |
| kr�t|
| | �}d }d}z�|dur�||�}ni }t||||d
| d �}|j�sd |j }tt|�|j|j|d
�}n>|�� �s6d|j }tt|�|j|j|d
�}n||jfW S W nb t�yz } zd| }|}W Y d}~n:d}~0 t �y� } zd| }|}W Y d}~n
d}~0 0 tt � � |
�}|�r�d| nd}d||||f }||� |rV|||d� qV| ||
��r�q(|
d }
t
� d|� t � |� q.dS )aI
urls: a list of urls to try
max_wait: roughly the maximum time to wait before giving up
The max time is *actually* len(urls)*timeout as each url will
be tried once and given the timeout provided.
a number <= 0 will always result in only one try
timeout: the timeout provided to urlopen
status_cb: call method with string message when a url is not available
headers_cb: call method with single argument of url to get headers
for request.
headers_redact: a list of header names to redact from the log
exception_cb: call method with 2 arguments 'msg' (per status_cb) and
'exception', the exception that occurred.
sleep_time_cb: call method with 2 arguments (response, loop_n) that
generates the next sleep time.
request_method: indicate the type of HTTP request, GET, PUT, or POST
returns: tuple of (url, response contents), on failure, (False, None)
the idea of this routine is to wait for the EC2 metadata service to
come up. On both Eucalyptus and EC2 we have seen the case where
the instance hit the MD before the MD service was up. EC2 seems
to have permanently fixed this, though.
In openstack, the metadata service might be painfully slow, and
unable to avoid hitting a timeout of even up to 10 seconds or more
(LP: #894279) for a simple GET.
Offset those needs with the need to not hang forever (and block boot)
on a system where cloud-init is configured to look for EC2 Metadata
service but is not going to find one. It is possible that the instance
data host (169.254.169.254) may be firewalled off Entirely for a system,
meaning that the connection will block forever unless a timeout is set.
A value of None for max_wait will retry indefinitely.
Nc S s t �| � d S r> )r- rw ��msg�excr r r �
log_status_cb� s z#wait_for_url.<locals>.log_status_cbc S s$ | d u r dS | dkp"t � � | | kS )NFr )r} )�max_wait�
start_timer r r �timeup� s z wait_for_url.<locals>.timeupr � r r F)r' r� r` r� r� zempty response [%s]rk zbad status code [%s]zrequest error [%s]zunexpected error [%s]z%ssZ unlimitedz Calling '%s' failed [%s/%s]: %s)r� � exceptionrl )FN)N)
r} rq r6 r( r& r4 �
ValueErrorr' rD � Exceptionr- rw r~ )Zurlsr� r` � status_cbr� r� Z
sleep_timer� Z
sleep_time_cbr� r� r� r� Zloop_nrN r �now�reasonZurl_excr' r: Z
time_takenZ max_wait_strZ
status_msgr r r � wait_for_urlX s� &
��
�
�
�� � r� c @ s^ e Zd Zddd�Zdd� Zdd� Zd d
� Zd d � Zd
d� Zdd� Z dd� Z
dd� Z dd� Z dS )�OauthUrlHelperN�/run/oauth_skew.jsonc C st || _ |p d| _|| _|| _|| _d| _d| _| j| j| j f}t|�sNd| _nt|�s^t d��| �
� }|pli | _ d S )Nr Tr� FzBall or none of token_key, token_secret, or consumer_key can be set) � consumer_key�consumer_secret� token_key� token_secret�skew_data_file� _do_oauth�skew_change_limit�any�allr� �read_skew_file� skew_data)r? r� r� r� r� r� �required�oldr r r r@ � s
zOauthUrlHelper.__init__c C sP | j rLtj�| j �rLt| j dd��}t�|�W d � S 1 sB0 Y d S )Nr� ��mode)r� �osr �isfiler0 �json�load)r? r9 r r r r� � s (z OauthUrlHelper.read_skew_filec C sh | j s
d S | �� }|d u r i }|||< t| j dd�� }|�t�|�� W d � n1 sZ0 Y d S )N�wr� )r� r� r0 �writer� �dumps)r? �host�value�curr9 r r r �update_skew_file� s z OauthUrlHelper.update_skew_filec
C s� t |t�r |jdks"|jdks"d S d|jvr>t�d|j� d S |jd }zt�t|��}W n4 t y� } z t�d||� W Y d }~d S d }~0 0 t
|t�� �}t |j �j
}| j�|d�}t|| �| jkr�| �||� t�d||� || j|< d S )Ni� i� �datez$Missing header 'date' in %s responsez#Failed to convert datetime '%s': %sr z$Setting oauth clockskew for %s to %d)ry r4 r& r' r- r. r} �mktimer r� rq r r �netlocr� r, �absr� r� ) r? r� r� r� Z remote_timer: Zskewr� Zold_skewr r r r� � s2
��
�
zOauthUrlHelper.exception_cbc C sZ | j s
i S d }t|�j}| jr>|| jv r>tt�� �| j| }t|| j| j| j | j
|d�S )N)r r� r� r� r� � timestamp) r� r r� r� rq r} �
oauth_headersr� r� r� r� )r? r r� r� r r r r� s
�zOauthUrlHelper.headers_cbc C s: t | j|�d��|d< t | j|�d��|d< ||i |��S )Nr� r� )r � _headers_cbr, �
_exception_cb)r? Z wrapped_funcrC r7 r r r �_wrapped s � �zOauthUrlHelper._wrappedc O s | � t||�S r> )r� r� rB r r r r� ' s zOauthUrlHelper.wait_for_urlc O s | � t||�S r> )r� r6 rB r r r r6 * s zOauthUrlHelper.readurlc
C s6 d }z |r|||�}W | � ||� n| � ||� 0 |S r> )r� )r? Zextra_exception_cbr� r� �retr r r r� - s z OauthUrlHelper._exception_cbc C s$ i }|r||�}|� | �|�� |S r> )rm r� )r? Zextra_headers_cbr r' r r r r� 6 s
zOauthUrlHelper._headers_cb)NNNNr� )
rH rI rJ r@ r� r� r� r� r� r� r6 r� r� r r r r r� � s �
r� c
C s~ zdd l m} W n. ty> } ztd�|�W Y d }~n
d }~0 0 |rNt|�}nd }|j|||||j|d�}|�| �\} }
} |
S )Nr z oauth support is not available)Z
client_secretZresource_owner_keyZresource_owner_secretZsignature_methodr� )Zoauthlib.oauth1�oauth1� ImportError�NotImplementedErrorr ZClientZSIGNATURE_PLAINTEXT�sign) r r� r� r� r� r� r� r: �clientZ_uriZsigned_headersZ_bodyr r r r� > s"
�r� c C s8 t |t�sdS |jtkr dS |jr4t |jtj�r4dS dS )z�readurl exception_cb that will retry on NOT_FOUND and Timeout.
Returns False to raise the exception from readurl, True to retry.
FT)ry r4 r& r r% r ZTimeoutr� r r r �retry_on_url_excU s
r� )NNr r NNNNTTNNFTN) NNNNNr NNN)N)4ru r� r� r} � email.utilsr r3 r � functoolsr � http.clientr � itertoolsr � urllib.parser r r r r
� cloudinitr �loggingr � getLoggerrH r- rX rp rY r
Zdistutils.versionr �
pkg_resourcesZget_distributionZ_REQr� r r" r; �objectr< r5 rL r2 r4 r[ r6 r� r� r� r� r r r r �<module> sh
$
! �
�
yn �