0x1949 Team - FAZEMRX - MANAGER
Edit File: url_helper.cpython-310.pyc
o ���fO� � @ s� d dl Z d dlZd dlZd dlZd dlZd dlZd dlZd dlZd dlm Z m Z m Z d dl m Z d dlmZ d dlmZ d dlmZ d dlmZ d dlmZmZmZmZmZm Z m Z d d l m Z m Z m!Z!m"Z" d dl#Z#d d l#m$Z$ d d l%m&Z&m'Z' e�(e)�Z*d Z+d d� Z,dd� Z-de.fdd�Z/dPde0de1de2ddfdd�Z3de0dd fd d �Z4de d fd d!�Z5G d"d#� d#�Z6G d$d � d e6�Z7G d%d� de6�Z8G d&d'� d'�Z9G d(d)� d)e:�Z;d*d+� Z< , - - . - / .dQd0e=de9fd1d2�Z> dRd3ed4ef d5e0de.d6ej?d7ee1 def d8d9�Z@ : ;dSd3ed4ef d<ee0 d=e1de.de f d>d?�ZAe1d@�de*jBdddddd/d-d:f dAe1dee1 dBedCee dDee1 dEee dFeeee1ge1f dGe0dHe=dIe1fdJdK�ZCG dLdM� dM�ZD dRdNdO�ZEdS )T� N)�ThreadPoolExecutor� TimeoutError� as_completed)� parsedate)�partial)� NOT_FOUND)�count)�create_default_context)�Any�Callable�Iterator�List�Optional�Tuple�Union)�quote�urlparse�urlsplit� urlunparse)� exceptions)�util�version�REDACTEDc C s< t t| dd��}|d s|d r|d |d< d|d< t|�S )N�http)�scheme� � � )�listr r )�url� parsed_url� r! �6/usr/lib/python3/dist-packages/cloudinit/url_helper.py� _cleanurl& 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_on� url_parsed�pathr! r! r" �combine_single1 s z#combine_url.<locals>.combine_singler! )�base�add_onsr+ r r( r! r! r" � combine_url0 s r. �returnc C sX t jdt jdt jdt jdtdi}|�t| ��}|s*t| t �r#| j }|S t � d� d}|S )z4helper for read_ftps to map return codes to a number�, � �� iX i� z9Unexpected exception type while connecting to ftp server.i����) �ftplib� error_reply� error_temp� error_perm� error_proto�EOFError�get�type� isinstance�OSError�errno�LOG�warning)�exc�ftp_error_codes�coder! r! r" �"ftp_get_return_code_from_exception@ s � ��rC � @r �timeout�kwargs� FtpResponsec K s� t | �}|jstdtd| d��t�� ��_}|jpd}|jp d}d|jkr�zt j t � d�}t � d| |� |j|j||p;d d � W n! t jy` } zt|�} td | � d | � d �| d| d�|�d}~ww t � d|� z|j||jpodd� t � d� |�� W n t jy� } zt �d� tddd| d�|�d}~ww zNz,t � d|j� |jd|j� �|jd� t|�� | �W W t � d� |�� W d � S t jy� } zt|�} td| � d | � d �| d| d�|�d}~ww t � d� |�� w zwzTt � � } t � d| |� | j|j||�pd d � t � d|� | j||j�pdd� t � d|j� | jd|j� �|jd� t|�� | �W W t � d� | �� W d � S t j�yh } zt|�} td| � d | � d �| d| d�|�d}~ww t � d� | �� w 1 �sww Y dS )az connect to URL using ftp over TLS and read a file when using strict mode (ftps://), raise exception in event of failure when not using strict mode (ftp://), fall back to using unencrypted ftp url: string containing the desination to read a file from. The url is parsed with urllib.urlsplit to identify username, password, host, path, and port in the following format: ftps://[username:password@]host[:port]/[path] host is the only required component timeout: maximum time for the connection to take kwargs: unused, for compatibility with read_url returns: UrlResponse zInvalid url providedN��causerB �headersr � � anonymous�ftps)�contextz3Attempting to connect to %s via port [%s] over tls.rD )�host�portrE z1Reading file from server over tls failed for url z [�]z"Attempting to login with user [%s]r )�user�passwdz Creating a secure connectionz�Attempted to connect to an insecure ftp server but used a scheme of ftps://, which is not allowed. Use ftp:// to allow connecting to insecure ftp servers.r2 zReading file: %szRETR )�callbackzClosing connectionz,Reading file from ftp server failed for url z(Attempting to connect to %s via port %s.) r �hostname�UrlErrorr �io�BytesIOrP �usernamer r3 �FTP_TLSr r> �debug�connect� all_errorsrC �login�password�prot_pr6 r? r* � retrbinary�writerG �getvalue�close�FTP) r rE rF � url_parts�bufferrP rR �ftp_tls�erB �ftpr! r! r" � read_ftpsW s � �� ������� � �� ���� �<������ �� � �]������ �rk r* � FileResponsec K s� |� d�r t�d| � z t�| �}t|| �W S ty+ } z t|td| d�|�d}~w t y@ } z t||j d| d�|�d}~ww )zmread a binary file and return a FileResponse matches function signature with read_ftps and read_url �dataz'Unable to post data to file resource %sNrH ) r9 r> r? r �load_binary_filerl �FileNotFoundErrorrV r �IOErrorr= )r* rF �contentsri r! r! r" � _read_file� s ���rr )rl � UrlResponserG c K s� | � � } zt| �}W n ty } zt|| d�|�d}~ww |j}|dks-| r6d| d kr6t|jfi |��S |dv rBt| fi |��S |dv rNt| fi |��S t � d|� t| fi |��S ) 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. )rI r N�filer$ r )rj rM )r �httpsz Attempting unknown protocol %s) �lstripr � ValueErrorrV r rr r* rk �readurlr> r? )r rF �parsedri r r! r! r" �read_file_or_url� s �� rz c @ s& e Zd Zd dd�Zdd� Zdd� ZdS ) �StringResponse�� c C s || _ i | _|| _|| _d S �N)rB rJ rq r ��selfrq r rB r! r! r" �__init__ s zStringResponse.__init__c O s | j dkrdS dS )Nr| FT�rB �r �argsrF r! r! r" �ok s zStringResponse.okc C s | j �d�S )Nzutf-8)rq �decode�r r! r! r" �__str__ s zStringResponse.__str__N�r| )�__name__� __module__� __qualname__r� r� r� r! r! r! r" r{ s r{ c s( e Zd Zddedef� fdd� Z� ZS )rl r| rq r c s t � j|||d� d S )Nr� ��superr� r~ �� __class__r! r" r� s zFileResponse.__init__r� �r� r� r� �bytesr' r� � __classcell__r! r! r� r" rl s c s& e Zd Zdedef� fdd� Z� ZS )rG rq r c s t � �||� d S r} r� )r rq r r� r! r" r� s zFtpResponse.__init__r� r! r! r� r" rG s c @ s� e Zd Zdejfdd�Zedefdd��Zede fdd��Z dde fd d �Z ed d � �Z edefdd��Zdd� Z ddee de dee fdd�ZdS )rs �responsec C s || _ d S r} )� _response)r r� r! r! r" r� " s zUrlResponse.__init__r/ c C s | j jd u rdS | j jS )N� )r� �contentr� r! r! r" rq % s zUrlResponse.contentsc C � | j jS r} )r� r r� r! r! r" r + � zUrlResponse.urlFc C s. d}|rd}d| j kr|k rdS dS dS )Nr0 r1 r| TFr� )r � redirects_ok�upperr! r! r" r� / s �zUrlResponse.okc C r� r} )r� rJ r� r! r! r" rJ 8 r� zUrlResponse.headersc C r� r} )r� � status_coder� r! r! r" rB < r� zUrlResponse.codec C r� r} )r� �textr� r! r! r" r� @ s zUrlResponse.__str__r � chunk_size�decode_unicodec c s � | j �||�E dH dS )at Iterates over the response data. When stream=True is set on the request, this avoids reading the content at once into memory for large responses. :param chunk_size: Number of bytes it should read into memory. :param decode_unicode: If True, content will be decoded using the best available encoding based on the response. N)r� � iter_content)r r� r� r! r! r" r� C s � zUrlResponse.iter_contentN)F)r F)r� r� r� �requests�Responser� �propertyr� rq r' r �boolr� rJ �intrB r� r r r� r! r! r! r" rs ! s( ����rs c @ s e Zd Zddd�ZdS )rV Nc C s< t �| t|�� || _|| _|| _| jd u ri | _|| _d S r} )rp r� r' rI rB rJ r )r rI rB rJ r r! r! r" r� S s zUrlError.__init__)NNN)r� r� r� r� r! r! r! r" rV R s rV c C s� i }t | �j}|dkr@|r@d|v r |d r |d |d< nd|d< d|v r4d|v r4|d |d g|d< |S d|v r@t|d �|d< |S )Nru �ca_certs�verifyT� cert_file�key_file�cert)r r r' )r � ssl_details�ssl_argsr r! r! r" � _get_ssl_args] s ��r� r TFr �streamc s� t | �} | |d�}|�t| |�� | |d< |s |rdnd}||d< |dur8t|t�r/||d< n tt|�d�|d< |du r>g }d }|rKtt|�d d �}d t� � }|durZ|� � }ni }|rb||d < |du rhd }| du rpt � � } t � D ]�}|r{|| �}d |vr�||d < ||d< i }|�� D ]1\}� |d kr�q�|dkr�|r�� fdd�|D �}|r�t �� �||< |D ]}t|| |< q�q�� ||< q�z3|r�t�d|| r�dn|| |� | jdi |��}| r�|�� t�d| |jt|j�|d � t|�W S tj�y } zt|| d�|�d}~w tj �yq } z]t|tj ��r1t |d��r1t |j d��r1t||j j|j j | d�}nt|| d�}| �rE| |� � |��sE||�| �pM|d |k } | �sT||�|dk�rg|�rbt�d|� t!�"|� W Y d}~qsd}~ww t#d��)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. May be a tuple if specifying (connection timeout, read timeout). :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. :param stream: if False, the response content will be immediately downloaded. )r r� �allow_redirects�POST�GET�methodNrE r r z Cloud-Init/%srm ���z User-AgentrJ c s g | ] }� � |�r|�qS r! )r9 )�.0�k��vr! r" � <listcomp>� s zreadurl.<locals>.<listcomp>z'[%s/%s] open '%s' with %s configuration�infinitez(Read from %s (%s, %sb) after %s attempts�r r� r� �rB rJ r �1Please wait %s seconds while we wait to try againz"This path should be unreachable...r! )$r# �updater� r; �tuple�max�floatr� r �version_string�copyr� �Sessionr �items�deepcopyr r> r[ �request�raise_for_statusr� �lenr� rs r �SSLErrorrV �RequestException� HTTPError�hasattrr� rJ �time�sleep� RuntimeError) r rm rE �retries� sec_betweenrJ � headers_cb�headers_redactr� � check_statusr� � exception_cb�sessionr� � log_req_resp�request_methodr� �req_args� manual_tries� user_agent�i�filtered_req_argsr� �matched_headers�key�rri � url_error� will_retryr! r� r" rx o s� 4� � �� � ����� � ��#rx �func.�addr�event�delayc C s |r |j |d�r dS | ||�S )z Execute func with optional delay�rE N��wait�r� r� rE r� r� r! r! r" �_run_func_with_delay s r� �333333�?� � addresses� stagger_delayc s. d}d}d}g }t �� �tt|�d�� z|zW� ����fdd�t|�D �}t|�d�D ],} || }| �� } | r>| }|�|� q*| �� }|rV�� � ||f W W � j dd� S q*|rbt � d||� |�t � d |� td �� ty� t � d d �|�d �tt|��� Y nw W � j dd� ||fS � j dd� w ) a execute multiple callbacks in parallel Run blocking func against two different addresses staggered with a delay. The first call to return successfully is returned from this function and remaining unfinished calls are cancelled if they have not yet started N)� max_workersc s, i | ]\}}� j t�|��|� d �|�qS )r� )�submitr� )r� r� r� ��executorr� �is_doner� rE r! r" � <dictcomp>K s ���z dual_stack.<locals>.<dictcomp>r� Fr� z<Exception(s) %s during request to %s, raising last exceptionzEmpty result for address %szNo result returnedzJTimed out waiting for addresses: %s, exception(s) raised while waiting: %s� )� threading�Eventr r� � enumerater � exception�append�result�set�shutdownr> r? �errorrw r �join�mapr' ) r� r� r� rE � return_result�returned_address�last_exceptionr �futures�future�return_exceptionr! r� r" � dual_stack1 sT � � � �� �r �inf�max_wait� status_cbr� � sleep_timer� � sleep_time_cbr� �connect_synchronously� async_delayc sP dt dtf� fdd� } d dtdtdtfd d �� d d � ���fd d�����fdd������� � fdd�} � ��� fdd�}t�� }� rJ|rJtd��| rN| n|}|pS| }d�d} ||��}|||||�}|rp|\}}||jfS � �||�rwn)�d �t�d|� t�|� t�� }|r�|| |� kr��|| }|dkr�nqYt� d� � dS ) a1 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 sleep_time: Amount of time to sleep between retries. If this and sleep_time_cb are None, the default sleep time defaults to 1 second and increases by 1 seconds every 5 tries. Cannot be specified along with `sleep_time_cb`. 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. Cannot be specified along with 'sleep_time`. request_method: indicate the type of HTTP request, GET, PUT, or POST connect_synchronously: if false, enables executing requests in parallel async_delay: delay before parallel metadata requests, see RFC 6555 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. The default value for max_wait will retry indefinitely. � loop_numberr/ c s � d ur� S |d d S )N� r r! )�_r )r r! r" �default_sleep_time� s z(wait_for_url.<locals>.default_sleep_timer r � start_timer c S s0 | t d�dfv r dS | dkpt�� | | | kS )z4Check if time is up based on start time and max waitr NFr )r� r� � monotonic)r r r r! r! r" �timeup� s �z wait_for_url.<locals>.timeupc S sr | j sd| j }tt|�| j| j|d�}||fS | �� s1d| j }tt|�| j| j|d�}||fS d}d}||fS )z?Map requests response code/contents to internal "UrlError" typezempty response [%s]r� zbad status code [%s]r N)rq rB rV rw rJ r� )r� r �reason�url_excr! r! r" �handle_url_response� s* �� � �z)wait_for_url.<locals>.handle_url_responsec s� d}d}z| |�\}}� ||�\}}|s||fW S W n/ t y2 } z d| }| }W Y d} ~ n d} ~ w tyI } z d| }| }W Y d} ~ nd} ~ ww tt�� | �} �rXd� nd} d|pbt|dd�| | |f } || � |rv|| |d � dS dS ) z:Execute request, handle response, optionally log exceptionr Nzrequest error [%s]zunexpected error [%s]z%ss� unlimitedz Calling '%s' failed [%s/%s]: %sr )�msgr� )rV � Exceptionr� r� r �getattr) � url_reader_cb�urlsr �exc_cb�log_cbr r r� r ri � time_taken� max_wait_str� status_msg)r r r! r" �read_url_handle_exceptions� s: ������z0wait_for_url.<locals>.read_url_handle_exceptionsc s$ t | � d u ri n� | ��|d�d�S )NF)rJ r� rE r� r� )rx )r rE )r� r� r� r! r" � read_url_cb� s �z!wait_for_url.<locals>.read_url_cbc s� �� fdd�}�D ]6}t �� }�dkr1��| �r dS �dur1� r1|� | � kr1t| � | �� �||| ||�}|r?| S q dS )z|iterate over list of urls, request each one and handle responses and thrown exceptions individually per url c s | � | ��fS r} r! r� )r% rE r! r" �url_reader_serial � z@wait_for_url.<locals>.read_url_serial.<locals>.url_reader_serialr N)r� r r� )r rE r r r& r �now�out)�loop_nr r% r$ r r r� r" �read_url_serial s$ � ���z%wait_for_url.<locals>.read_url_serialc s, t t�� |d�}�|�| ||�}|r|S dS )z�pass list of urls to dual_stack which sends requests in parallel handle response and exceptions of the first endpoint to respond )r� rE N)r r )r rE r r �url_reader_parallelr) )r r% r$ r r! r" �read_url_parallel$ s � ��z'wait_for_url.<locals>.read_url_parallelz3sleep_time and sleep_time_cb are mutually exclusiveNTr r� z$Timed out, no response from urls: %s)FN)r ) r� r� r� r rw rq r> r[ r� r� )r r rE r r� r� r r� r r� r r r r+ r- r � do_read_url�calculate_sleep_timer� �current_sleep_timer �address� current_timer! ) r r r� r� r* r r% r$ r� r r r r" � wait_for_url� sJ 9 � � � r3 c @ sh e Zd Z ddd�Zdd� Zdd� Zd d � Zd d � Zd d� Zdd� Z dd� Z dd� Z dd� Z dS )�OauthUrlHelperN�/run/oauth_skew.jsonc C st || _ |pd| _|| _|| _|| _d| _d| _| j| j| j f}t|�s'd| _nt|�s/t d��| � � }|p6i | _ 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�allrw �read_skew_file� skew_data)r r6 r8 r9 r7 r: �required�oldr! r! r" r� d s �zOauthUrlHelper.__init__c C sP | j r&tj�| j �r&t| j dd�� }t�|�W d � S 1 s!w Y d S )Nr� ��mode)r: �osr* �isfile�open�json�load)r �fpr! r! r" r? s �z OauthUrlHelper.read_skew_filec C sj | j sd S | �� }|d u ri }|||< t| j dd��}|�t�|�� W d � d S 1 s.w Y d S )N�wrC )r: r? rG rb rH �dumps)r rO �value�currJ r! r! r" �update_skew_file� s "�z OauthUrlHelper.update_skew_filec C s� t |t�r|jdks|jdksd S d|jvr t�d|j� d S |jd }z t�t|��}W n t yG } zt�d||� W Y d }~d S d }~ww t |t�� �}t |j �j }| j�|d�}t|| �| jkrs| �||� 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)r; rV rB rJ r> r? r� �mktimer r r� r r �netlocr@ r9 �absr< rO ) r r r� rP � remote_timeri �skewrO �old_skewr! r! r" r� � s. � �� zOauthUrlHelper.exception_cbc C sZ | j si S d }t|�j}| jr || jv r tt�� �| j| }t|| j| j| j | j |d�S )N)r r6 r8 r9 r7 � timestamp) r; r rR r@ r� r� � oauth_headersr6 r8 r9 r7 )r r rW rO r! r! r" r� � s �zOauthUrlHelper.headers_cbc C s: t | j|�d��|d< t | j|�d��|d< ||i |��S )Nr� r� )r � _headers_cbr9 � _exception_cb)r � wrapped_funcr� rF r! r! r" �_wrapped� s � �zOauthUrlHelper._wrappedc O � | � t||�S r} )r\ r3 r� r! r! r" r3 � r' zOauthUrlHelper.wait_for_urlc O r] r} )r\ rx r� r! r! r" rx � r' zOauthUrlHelper.readurlc C sF d }z|r|||�}W | � ||� |S W | � ||� |S | � ||� w r} )r� )r �extra_exception_cbr r� �retr! r! r" rZ � s � �z OauthUrlHelper._exception_cbc C s$ i }|r||�}|� | �|�� |S r} )r� r� )r �extra_headers_cbr rJ r! r! r" rY � s zOauthUrlHelper._headers_cb)NNNNr5 ) r� r� r� r� r? rO r� r� r\ r3 rx rZ rY r! r! r! r" r4 c s � r4 c C sr zdd l m} W n ty } ztd�|�d }~ww |r!t|�}nd }|j|||||j|d�}|�| �\} } } | S )Nr z oauth support is not available)� client_secret�resource_owner_key�resource_owner_secret�signature_methodrW )�oauthlib.oauth1�oauth1� ImportError�NotImplementedErrorr' �Client�SIGNATURE_PLAINTEXT�sign) r r6 r8 r9 r7 rW rf ri �client�_uri�signed_headers�_bodyr! r! r" rX � s& �� �rX )rD )NNr r NNNNTTNNFTr Fr} )r� r� )Fr� r3 rW rH �loggingrE r� r� �concurrent.futuresr r r � email.utilsr � functoolsr � http.clientr � itertoolsr �sslr �typingr r r r r r r � urllib.parser r r r r� r � cloudinitr r � getLoggerr� r> r r# r. r� rC r' r� �dictrk rr rz r{ rl rG rs rp rV r� r� rx r� r� r r[ r3 r4 rX r! r! r! r" �<module> s� $ � 1 �� � 6� ����� �� ���� �U������� � � � � `|�