dW ddlZddlZddlZddlZddlZddlZddlmcmZddl mcm Z ddl mcm cm Z ddlmZmZmZddlmZmZmZddlmZmZmZdadZdZdZdZd Zej@d Z!ejDejFejHejJejLejNejPejRejTd g Z+e+ddZ,e,j[ej\ej^gZ0ejbejdejfgZ4ejjgZ6d gZ7ejHejPd gZ8ejrejtejvejxgZ=d Z>dZ?dZ@djeeZBdjeBZCdZDdZEdZFdZGGddeHZIdefdZJd3dZKdZLdZMdZNdZOdZPdZQd ZRd!ZSd"ZTd#ZUd4d$ZVd%ZW d5d&ZXddddeddfd'ZYddddedd(fd)ZZdddded(fd*Z[dddded(fd+Z\ddddedd(fd,Z]dddded(fd-Z^d3d.Z_d3d/Z`d0Zad3d1Zbd2Zcy)6N) HttpErrorResourceGoneErrorInvalidContainerError) httpclienturlparseustr)PY_VERSION_MAJOR AGENT_NAMEGOAL_STATE_AGENT_VERSIONFzz3^(https?://[a-zA-Z0-9.].*sig=)([a-zA-Z0-9%-]*)(.*)$ii http_proxy https_proxyno_proxyz{0}/{1}z {0}+healthInvalidContainerConfigurationRequestRoleConfigFileNotFoundz 168.63.129.16iceZdZejZeZddddZe ddZ e dZ e dZ e efdZ y) IOErrorCounterr hostpluginprotocolotherNc,tj5|tjk(rA|tk(rtjdxxdz cc<n7tjdxxdz cc<ntjdxxdz cc<dddy#1swYyxYw)Nrr rr)r_lock_protocol_endpointHOST_PLUGIN_PORT_countshostports G/usr/lib/python3/dist-packages/azurelinuxagent/common/utils/restutil.py incrementzIOErrorCounter.incrementvsy  ! ! 5~888++"**<8A=8"**:6!;6&&w/14/ 5 5 5s A0B  Bctj5tjj}tj |cdddS#1swYyxYwN)rrrcopyreset)countss r" get_and_resetzIOErrorCounter.get_and_resetsC  ! ! #++002F  "   s 4AAcltj5ddddt_dddy#1swYyxYw)Nrr)rrrr"r'zIOErrorCounter.resets3  ! ! O34A%NN " O O Os*3c|t_yr%)rr)endpoints r"set_protocol_endpointz$IOErrorCounter.set_protocol_endpoints ,4)r,NN)__name__ __module__ __qualname__ threadingRLockrKNOWN_WIRESERVER_IPrr staticmethodr#r)r'r/r+r,r"rrqsu IOO E,!Q7G55 OO':55r,rcVd}t|D]}|d|d|dzf}||dzS)N)r r r r)range) retry_attemptdelayfib_s r"_compute_delayr>sC C = !&1vs1vc!f}%& Q<r,c|t}||vSr%) RETRY_CODES)status retry_codess r"_is_retry_statusrCs! [  r,c fttDcgc]}t||s|c}dkDScc}w)Nr)lenRETRY_EXCEPTIONS isinstance)exs r"_is_retry_exceptionrJs* +@az!Q/?@ AA EE@s..c|tvSr%)THROTTLE_CODES)rAs r"_is_throttle_statusrMs ^ ##r,cFt|}|j}|jrdj||j}|jrdj||j}d}|j j dk(rd}|j|j||fS)zt Parse URL to get the components of the URL broken down to host, port :rtype: string, int, bool, string z{0}#{1}z{0}?{1}FhttpsT) rpathfragmentformatqueryschemelowerhostnamer!)urlorel_urisecures r" _parse_urlr[s  AffGzz""7AJJ7ww""7AGG4 Fxx~~7" ::qvvvw ..r,c&t|}|jry|jrI)rf inet_ntoastructpack)rjbitss r"dotted_netmaskrqs7 b4i1, ,D   FKKd3 44r,c rtjdtj|d}|j d\}}tjdtjt t |d}tjdtj|d|z}||z||zk(S)zThis function allows you to check if an IP belongs to a network subnet Example: returns True if ip = 192.168.1.1 and net = 192.168.1.0/24 returns False if ip = 192.168.1.1 and net = 192.168.100.0/24 :rtype: bool z=Lrr`)rnunpackrfrgrdrqrc)ipnetipaddrnetaddrrpnetmasknetworks r"address_in_networkrzs ]]4!1!1"!5 6q 9FIIcNMGTmmD&"2"2>#d)3L"MNqQGmmD&"2"27";P>P>R/SH%-%5%5c2%>%D%DS%IRTTDRR OSs <B B ct}|rjt|r)|D]#}t|rt||sy||k(s#yy|D]1}|j j |j s1yy)NTF)rr}rkrzrUendswith)r rproxy_ip proxy_domains r" bypass_proxyrs~H 4 $  *)$9#X%    !) ::<((););)=>  r,c*tj}d}|tj}||fS|rtnt}d}||j fD])}|t jvst j|}n|t|\}}}}||fSr%) confget_httpproxy_hostget_httpproxy_portHTTPS_PROXY_ENVHTTP_PROXY_ENVrrrr[)rZr r!http_proxy_envhttp_proxy_urlvr=s r"_get_http_proxyrs  " " $D D <&&( :-3 ."6"6"89 ABJJ!#A  %).9 D$1 :r,cBtjdtzdz|S)Nz\1z\3)SAS_TOKEN_RETRIEVAL_REGEXsub REDACTED_TEXT)rWs r"redact_sas_tokens_in_urlsr4s $ ( ()>)F LLr,c |in|}d|d<|duxr| du} ||rdnd}d|vr t|d<| r|| } } |rdnd}dj||||}n||} } |}|r-tj| | | }| r+|j ||ntj | | | }|}| rd }t jd |t|tj|||j|||| |jS) Nclose ConnectioniPz User-AgentrOhttpr])timeoutz [REDACTED]z'HTTP connection [{0}] [{1}] [{2}] [{3}])methodrWbodyheaders) HTTP_USER_AGENTrRrHTTPSConnection set_tunnelHTTPConnectionloggerverbosertextutilstr_to_encoded_ustrrequest getresponse)rr rYrr!datarZr proxy_host proxy_port redact_data use_proxy conn_host conn_portrTrWconnpayloads r" _http_requestr8s-ObG#GL$&A:T+AI |s"7" / ):9 " ''dGD#T9  )))*329;  OOD$ '(()218:G NN<,S1//8   LLCdGLD    r,c N|t}|t}t|\} } } }d\}}|r5t| s*t | \}}|s|rt j d||| rNttds>tjs tdd} tst jdd a | r\|Z|Xttjd s>tjs td d} tst jd d a d }d}d}d}||krW|dkDrF|rtn t!||}t j d|dz|||t#j$||dz } t'|| ||| || ||||  }t j d|j(| r|St+|r\t-|j(|rEdj/|||j(}t1|j(rd }t3|t4}|j(t6vrt9|}t;||j(tj<k(rt9|}t>|vr tA||Stdj/||#tjB$r<}| rtE|}dj/|||}tG|rYd}~Yd}~ed}~wtH$rC}| rtJjM| | tE|}dj/|||}Yd}~ d}~wwxYw)a NOTE: This method provides some logic to handle errors in the HTTP request, including checking the HTTP status of the response and handling some exceptions. If return_raw_response is set to True all the error handling will be skipped and the method will return the actual HTTP response and bubble up any exceptions while issuing the request. Also note that if return_raw_response is True no retries will be done. Nr0)rZzHTTP proxy: [{0}:{1}]rz!HTTPS is unavailable and requiredFz#Python does not include SSL supportTrz,HTTPS tunnelling is unavailable and requiredz(Python does not support HTTPS tunnellingrr)r:r;z;[HTTP Retry] Attempt {0} of {1} will delay {2} seconds: {3}r )r!rrZrrrrz[HTTP Response] Status Code {0})rBz'[HTTP Retry] {0} {1} -- Status Code {2}z*[HTTP Failed] {0} {1} -- HttpException {2}rz$[HTTP Failed] {0} {1} -- IOError {2}z{0} -- {1} attempts made)'DEFAULT_RETRIESr@r[rrrrhasattrrrget_allow_httprSECURE_WARNING_EMITTEDwarnrTHROTTLE_DELAY_IN_SECONDSr>timesleeprrArequest_failedrCrRrMmaxTHROTTLE_RETRIESRESOURCE_GONE_CODESread_response_errorr BAD_REQUESTINVALID_CONTAINER_CONFIGURATIONr HTTPExceptionr^rJIOErrorrr#)rrWrrrr max_retryrB retry_delayrreturn_raw_responser r!rZrYrrmsgattemptr; was_throttledrespresponse_errorrH clean_urls r" http_requestrhs # ! #-S/D$*J d+!0!? J  NN2J Kgj*;<""$?@ @% KK= >%) " 22LA""$JK K% KKB C%) " CG EM I  Q;).+'2=?  NNI !   JJu 1 :  !%!(!(&*&*(.)0,6,6-8 :D NNEEfiYZ[C"1%  "  $ $$T $ :,S1I8?? STUC   s2*9J $A&J A$J L$,K L$!8LL$ c L|t}|t}td|d||||||| S)a NOTE: This method provides some logic to handle errors in the HTTP request, including checking the HTTP status of the response and handling some exceptions. If return_raw_response is set to True all the error handling will be skipped and the method will return the actual HTTP response and bubble up any exceptions while issuing the request. Also note that if return_raw_response is True no retries will be done. NGET)rrrrBrrrr@r)rWrrrrBrrrs r"http_getrsD# ! T7 '"+"+$/$/,? AAr,c J|t}|t}td|d|||||| S)NHEADrrrrBrrrWrrrrBrrs r" http_headrs?# ! T7 '"+"+$/$/  11r,c J|t}|t}td|||||||| S)NPOSTrr)rWrrrrrBrrs r" http_postr/s?# ! T7 '"+"+$/$/  11r,c L|t}|t}td||||||||| S)NPUT)rrrrBrrr) rWrrrrrBrrrs r"http_putrEsB# ! T7 '"+"+$/$/$/ 11r,c J|t}|t}td|d|||||| S)NDELETErrrs r" http_deleter]s?# ! T7 '"+"+$/$/  11r,c.|t}t|| S)N)ok_codes)OK_CODESrequest_succeededrrs r"rrrs 9 99r,c:|t}|duxr|j|vSr%)rrArs r"rrxs& t  7 x 77r,c2|duxr|jtvSr%)rANOT_MODIFIED_CODES)rs r"request_not_modifiedr~s t  A /A AAr,c\|t}|duxr|jdk\xr|j|vS)z` Host plugin will return 502 for any upstream issue, so a failure is any 5xx except 502 Ni)!HOSTPLUGIN_UPSTREAM_FAILURE_CODESrA)rupstream_failure_codess r"request_failed_at_hostpluginrs8%!B t  ` s 2 `t{{J`7``r,cd}| dj|j|j|j}tdkrt |dd}n$|j ddjdd}tj|}|S|S#t$r3}tjtj|Yd}~|Sd}~wwxYw)Nrz[HTTP Failed] [{0}: {1}] {2}asciiignore)encodingerrors)rRrAreasonreadr rencodedecoderreplace_non_ascii Exceptionrrformat_exception)rresultrHs r"rrs F  63::   %F !#fwxHVWXV>VWXV>//7F M6M 6 KK11!4 5 5 M 6sBB C (CC r%)F)NNFNNNF)drrer4rrfrnazurelinuxagent.common.confcommonrazurelinuxagent.common.loggerr%azurelinuxagent.common.utils.textutilutilsr azurelinuxagent.common.exceptionrrrazurelinuxagent.common.futurerrrazurelinuxagent.common.versionr r r rrDELAY_IN_SECONDSrrrcompiler RESET_CONTENTPARTIAL_CONTENT FORBIDDENINTERNAL_SERVER_ERRORNOT_IMPLEMENTED BAD_GATEWAYSERVICE_UNAVAILABLEGATEWAY_TIMEOUTINSUFFICIENT_STORAGEr@'HGAP_GET_EXTENSION_ARTIFACT_RETRY_CODESappendrGONErOKCREATEDACCEPTEDr NOT_MODIFIEDrrrL NotConnectedIncompleteReadImproperConnectionState BadStatusLinerFrrrrRrHTTP_USER_AGENT_HEALTHr"REQUEST_ROLE_CONFIG_FILE_NOT_FOUNDr6robjectrr>rCrJrMr[r^rkrqrzr}rrrrrrrrrrrrrrrrr+r,r"r&s(  **..88``DDaa! &BJJ']^$$""##  &+6a.''..z/E/EFOO MM  %! ""&&  "":/GH%,,_="A%D"%5V5B"#*:! F$/ .5 5.,MPUNS-d !-"%*REl)!&A<* 1.*10)12 , 1*: 8 Bar,