o w7e @sddlmZddlmZmZddlZddlmZddlmZddl m Z ddl m Z m Z eeZdZGd d d Zd d d dededededef ddZdededefddZdedefddZdefddZdS))time)DictOptionalN)log)util)report_diagnostic_event)UrlErrorreadurlzhttp://169.254.169.254/metadatac @sHeZdZdZddddddeded eed dfd d Zd efd dZ 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_deadlinerrrreturncCs(||_||_||_||_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__cCs|jd7_t|tstd|tjddSd}|jdur't|jkr'd}nd}t|jt j r?|j d8_ |j dkr>d}n |j durL|j |j vrLd}|j|jkr[|j|j9_nd}|sa|smtd|j|ftjd|S)Nz1Polling IMDS failed with unexpected exception: %r logger_funcFTrz1Polling IMDS failed attempt %d with exception: %r)r isinstancerrLOGwarningrrcauserequestsConnectionErrorrcoderrrinfo)rreq_args exceptionrretryrrrexception_callback1s@     z&ReadUrlRetryHandler.exception_callback) __name__ __module__ __qualname____doc__floatintrrboolr)rrrrr s    r T) log_responsetimeouturlrr2r3rc CsZt|d}zt||jddid||d}W|jSty,}z td|tjdd}~ww) 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. rMetadatatrueT exception_cbheadersinfinite log_req_respr3z&Failed to fetch metadata from IMDS: %srN)r r r)rrrr contents)r4rr2r3handlerresponseerrorrrr _fetch_url\s& rAc CsFt||d}zt|WSty"}z td|tjdd}~ww)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. r5z&Failed to parse metadata from IMDS: %srN)rAr load_json ValueErrorrrr )r4rmetadatar@rrr_fetch_metadatas  rEc Csjz td}t||dWSty4}z|jdkr/tdtjdtd}t||dWYd}~Sd}~ww)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=truer5iz,Falling back to IMDS api-version: 2019-06-01rz /instance?api-version=2019-06-01N)IMDS_URLrErr$rrr )rr4r@rrr fetch_metadata_with_api_fallbacks rGcCsTtd}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 rNrr6r7TFr1r8z Polled IMDS rz time(s)r) rFr r r)rrrdebugr=)r4r>r?rrrfetch_reprovision_datas(  rI)rtypingrrr" cloudinitrloggingrcloudinit.sources.helpers.azurercloudinit.url_helperrr getLoggerr*rrFr strr.r0r/bytesrArErGrIrrrrs@     L %