U Ld@sddlmZddlmZmZddlZddlmZddlmZddl m Z ddl m Z m Z eeZdZGd d d Zd d d eeeeedddZeeedddZeedddZedddZdS))time)DictOptionalN)log)util)report_diagnostic_event)UrlErrorreadurlzhttp://169.254.169.254/metadatac@sBeZdZdZdddddeeeedddd Zed d d Z dS) ReadUrlRetryHandleraFManager for readurl retry behavior using exception_callback(). :param logging_backoff: Backoff to limit logging. :param max_connection_errors: Number of connection errors to retry on. :param retry_codes: Set of http codes to retry on. :param retry_deadline: Optional time()-based deadline to retry until. ? )iiNlogging_backoffmax_connection_errors retry_codesretry_deadline)rrrreturncCs(||_||_||_||_d|_d|_dS)Nr r)rrrr_logging_threshold_request_count)selfrrrrr>/usr/lib/python3/dist-packages/cloudinit/sources/azure/imds.py__init__s zReadUrlRetryHandler.__init__rcCs|jd7_t|ts.td|tjddSd}|jdk rNt|jkrNd}nd}t|jt j r~|j d8_ |j dkrd}n|j dk r|j |j krd}|j|jkr|j|j9_nd}|s|std|j|ftjd|S)Nz1Polling IMDS failed with unexpected exception: %rZ logger_funcFTrz1Polling IMDS failed attempt %d with exception: %r)r isinstancerrLOGwarningrrcauserequestsConnectionErrorrcoderrrinfo)rZreq_argsZ exceptionrZretryrrrexception_callback1sB    z&ReadUrlRetryHandler.exception_callback) __name__ __module__ __qualname____doc__floatintrrboolr&rrrrr s  r T) log_responsetimeout)urlrr/r0rc Csft|d}zt||jddid||d}Wn6tk r^}ztd|tjdW5d}~XYnX|jS) a'Fetch URL from IMDS. :param url: url to fetch. :param log_response: log responses in readurl(). :param retry_deadline: time()-based deadline to retry until. :param timeout: Read/connection timeout in seconds for readurl(). :raises UrlError: on error fetching metadata. rMetadatatrueTZ exception_cbZheadersZinfiniteZ log_req_respr0z&Failed to fetch metadata from IMDS: %srN)r r r&rrrr contents)r1rr/r0handlerresponseerrorrrr _fetch_url\s"  r:)r1rrc CsTt||d}z t|WStk rN}ztd|tjdW5d}~XYnXdS)zFetch IMDS metadata. :param url: url to fetch. :param retry_deadline: time()-based deadline to retry until. :raises UrlError: on error fetching metadata. :raises ValueError: on error parsing metadata. r2z&Failed to parse metadata from IMDS: %srN)r:rZ load_json ValueErrorrrr )r1rZmetadatar9rrr_fetch_metadatas  r<)rrc Cstztd}t||dWStk rn}z:|jdkr\tdtjdtd}t||dWYSW5d}~XYnXdS)zFetch extended metadata, falling back to non-extended as required. :param retry_deadline: time()-based deadline to retry until. :raises UrlError: on error fetching metadata. :raises ValueError: on error parsing metadata. z./instance?api-version=2021-08-01&extended=truer2iz,Falling back to IMDS api-version: 2019-06-01rz /instance?api-version=2019-06-01N)IMDS_URLr<rr$rrr )rr1r9rrr fetch_metadata_with_api_fallbacks r>rcCsTtd}tddddd}t||jddid d d d }td |jddtjd|jS)zhFetch extended metadata, falling back to non-extended as required. :raises UrlError: on error. z'/reprovisiondata?api-version=2019-06-01g@r)r rNrr3r4TFr.r5z Polled IMDS rz time(s)r) r=r r r&rrrdebugr6)r1r7r8rrrfetch_reprovision_datas(  r@)rtypingrrr"Z cloudinitrZloggingrZcloudinit.sources.helpers.azurerZcloudinit.url_helperrr Z getLoggerr'rr=r strr+r-r,bytesr:r<r>r@rrrrs0     L &