U Ad@sddlmZddlZddlmZddlmZddlmZddl m Z m Z e e ZdZGdd d Zd d d d eeeeedddZdeeedddZdeedddZedddZdS))DictN)log)util)report_diagnostic_event)UrlErrorreadurlzhttp://169.254.169.254/metadatac@s6eZdZddddeeddddZed d d ZdS) ReadUrlRetryHandler)ii ?) retry_codesmax_connection_errorslogging_backoffN)rrreturncCs"||_||_||_d|_d|_dS)Nr r)rrr _logging_threshold_request_count)selfr rrr>/usr/lib/python3/dist-packages/cloudinit/sources/azure/imds.py__init__s zReadUrlRetryHandler.__init__rcCs|jd7_t|ts.td|tjddSd}d}t|jtjtj frh|j d8_ |j dkrxd}n|j |j krxd}|j|j kr|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 isinstancerrLOGwarningcauserequestsConnectionErrorZTimeoutrcoder rrinfo)rZreq_argsZ exceptionrZretryrrrexception_callback&s>     z&ReadUrlRetryHandler.exception_callback)__name__ __module__ __qualname__intfloatrboolr"rrrrrs  rTr ) log_responseretriestimeout)urlr*r+r,rc Csdt}z t||jddid|||d}Wn6tk r\}ztd|tjdW5d}~XYnX|jS)zLFetch URL from IMDS. :raises UrlError: on error fetching metadata. MetadatatrueF) exception_cbheadersinfinite log_req_respr+r,z&Failed to fetch metadata from IMDS: %srN)rrr"rrrrcontents)r-r*r+r,handlerresponseerrorrrr _fetch_urlMs$ r8)r-r+rc CsTt||d}z t|WStk rN}ztd|tjdW5d}~XYnXdS)zFetch IMDS metadata. :raises UrlError: on error fetching metadata. :raises ValueError: on error parsing metadata. r+z&Failed to parse metadata from IMDS: %srN)r8rZ load_json ValueErrorrrr)r-r+Zmetadatar7rrr_fetch_metadatajs  r;)r+rc Cstztd}t||dWStk rn}z:|jdkr\tdtjdtd}t||dWYSW5d}~XYnXdS)zFetch extended metadata, falling back to non-extended as required. :raises UrlError: on error fetching metadata. :raises ValueError: on error parsing metadata. z./instance?api-version=2021-08-01&extended=truer9iz,Falling back to IMDS api-version: 2019-06-01rz /instance?api-version=2019-06-01N)IMDS_URLr;rr rrr)r+r-r7rrr fetch_metadata_with_api_fallbacks r=rcCsRtd}tdddd}t||jddidd d d }td |jd dtjd|jS)zhFetch extended metadata, falling back to non-extended as required. :raises UrlError: on error. z'/reprovisiondata?api-version=2019-06-01g@r)r r )rrr r.r/TFr))r0r1r2r3r,z Polled IMDS rz time(s)r) r<rrr"rrrdebugr4)r-r5r6rrrfetch_reprovision_datas& r?)r )r )typingrrZ cloudinitrZloggingrZcloudinit.sources.helpers.azurerZcloudinit.url_helperrrZ getLoggerr#rr<rstrr(r&bytesr8r;r=r?rrrrs4     ;