f ddlZddlmZddlmZmZmZmZddlZddlm Z ddl m Z ddl m Z mZejeZdZGdd Zd d d d ededededef dZd ededefdZ ddedeedefdZdefdZy)N)time)DictOptionalTypeUnion)util)report_diagnostic_event)UrlErrorreadurlzhttp://169.254.169.254/metadatac LeZdZdZddddddedeedeed dfd Zd efd Z y) 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. ?N)ilogging_backoffmax_connection_errors retry_codesretry_deadlinerrrreturncf||_||_||_||_d|_d|_d|_y)Nrr)rrrr_logging_threshold_request_count _last_error)selfrrrrs >/usr/lib/python3/dist-packages/cloudinit/sources/azure/imds.py__init__zReadUrlRetryHandler.__init__s< /%:"&,"%37c|xjdz c_t|tstd|ztj yd}|j t|j k\rd}nd}|jKt|jtjr'|xjdzc_|jdkr)d}n&|j|j|jvrd}|j|jk\r |xj|jzc_nd}|j-|j |jk7rvd}|j|_nb|j :t|j t"r t|j|j sd}t#|j|_|s|s*td|j|fztj |S)Nz1Polling IMDS failed with unexpected exception: %r logger_funcFTrz1Polling IMDS failed attempt %d with exception: %r)r isinstancer r LOGwarningrrrcauserequestsConnectionErrorcoderrrrtype)rreq_args exceptionlogretrys rexception_callbackz&ReadUrlRetryHandler.exception_callback2s q )X. #CKK      *tv9L9L/LEE  % % 1j OOX557   & &! + &))Q. NN &d&6&66E   $"9"9 9  # #t';'; ; #C >> %9>>1#,>>     $d..5ioot/?/?@C#IOO4D  e #C&& 23KK   r) __name__ __module__ __qualname____doc__floatrintrboolr0rrr r sZ"%/3 +/88 (} 8!8 8*;;rr T) log_responsetimeouturl retry_handlerr:r;rc t||jddid||}|j S#t$r$}td|ztj d}~wwxYw)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. MetadatatrueT exception_cbheadersinfinite log_req_respr;z&Failed to fetch metadata from IMDS: %sr"N)r r0r r r%r&contents)r<r=r:r;responseerrors r _fetch_urlrIpsj  &99(%       4u <    s+ AAAct||} tj|jdS#t$r$}t d|zt jd}~wwxYw)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. r=zutf-8z&Failed to parse metadata from IMDS: %sr"N)rIr load_jsondecode ValueErrorr r%r&)r<r=metadatarHs r_fetch_metadatarPsY#];H~~hoog677  4u <    s#3 A AA rrct||} tdz}t||S#t$rX}|jdk(rCt dt jt||}tdz}t||cYd}~Sd}~wwxYw) 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. )rrz./instance?api-version=2021-08-01&extended=truerKiz,Falling back to IMDS api-version: 2019-06-01r"z /instance?api-version=2019-06-01N)r IMDS_URLrPr r*r r%r&)rrr=r<rHs r fetch_metadata_with_api_fallbackrSs(3%MIIs-@@   ::  #>KK 0&;-M??C"3mD D  s!% BA B:BBBctdz}tdddd}t||jddid d d }t d |j dzdt j|jS)zhFetch extended metadata, falling back to non-extended as required. :raises UrlError: on error. z'/reprovisiondata?api-version=2019-06-01g@r!)rrrNrr?r@TFr9rAz Polled IMDS z time(s)r") rRr r r0r rr%debugrF)r<handlerrGs rfetch_reprovision_datarWs > >C!  G //V$ H w--a/09II   r)N)loggingrtypingrrrrr( cloudinitrcloudinit.sources.helpers.azurer cloudinit.url_helperr r getLoggerr1r%rRr strr7r6bytesrIrPr5rSrWr8rrr`s ..C2g! ,YY@   '      F ' 4CG2:3- @r