p fAdZddlmZddlZddlZddlZddlZddlZddlm Z m Z ddl m Z ddl mcmZddlmcmZddlmcmZddlmZddlmZmZddlmZdd lmZmZdd lmZdd l m!Z!dd l"m#Z#m$Z$m%Z%m&Z&m'Z'm(Z(m)Z)m*Z*m+Z+m,Z,m-Z-dd l.m/Z/m0Z0ddl1m2Z2Gdde3Z4Gdde3Z5ddZ6dZ7dZ8dZ9ddZ:e;dk(re6yy)z Module agent )print_functionN)cgroupconfigurator logcollector)SystemdCgroupsApi)ustr) LogCollectorOUTPUT_RESULTS_FILE_PATH) get_osutil)fileutiltextutil)FlexibleVersion)AddFirewallRules) AGENT_NAMEAGENT_LONG_VERSION AGENT_VERSION DISTRO_NAMEDISTRO_VERSIONPY_VERSION_MAJORPY_VERSION_MINORPY_VERSION_MICROGOAL_STATE_AGENT_VERSIONget_daemon_versionset_daemon_version)CollectLogsHandler!get_log_collector_monitor_handler)ProvisionHandlerc@eZdZdZdZdZdZdZdZdZ dZ d Z d Z d Z d Zd Zy) AgentCommandszN This is the list of all commands that the Linux Guest Agent supports zdeprovision+user deprovisiondaemonstartzregister-servicezrun-exthandlersversionzshow-configurationhelpz collect-logszsetup-firewall provisionN)__name__ __module__ __qualname____doc__DeprovisionUser DeprovisionDaemonStartRegisterServiceRunExthandlersVersion ShowConfigHelp CollectLogs SetupFirewall Provision7/usr/lib/python3/dist-packages/azurelinuxagent/agent.pyrr4sE)OK F E(O&NG%J D K$MIr6rcXeZdZd dZdZdZdZd dZdZddZ d Z d Z e d Z y)AgentNc||_t|_|rtjj ntjj }tjtjj||j |jn|jj}tj||xstj}|rtjj ntjj }tjtjj|tjtj r%t#j$s|j'|t)j*rVtjtjj,tjj.t(j0tj2} t4j6j9|rt;dj=|t4j6j?|stAjB|ddt)jFtjHt4j6jKtjHt(jL}t)jN|t)jPdy#t:$r/}tjDdj=||Yd}~d}~wwxYw) z7 Initialize agent running environment. Npathz {0} is a fileiroot)modeownerzBException occurred while creating extension log directory {0}: {1}WALA))conf_file_pathr osutilloggerLogLevelVERBOSEINFOadd_logger_appender AppenderTypeSTDOUTget_agent_conf_file_pathconfload_conf_from_fileget_logs_verboseFILEget_agent_log_fileget_logs_consoleris_provisioned_Agent__add_console_appendereventsend_logs_to_telemetry TELEMETRYWARNING add_log_eventget_ext_log_dirosr<isfile Exceptionformatisdirr mkdirerrorinit_event_status get_lib_dirjoinEVENTS_DIRECTORYinit_event_loggerenable_unhandled_err_dump)selfverboserAlevel ext_log_dire event_dirs r7__init__zAgent.__init__Gs:- l ,3''8L8L""6#6#6#=#=uE&&2,,==?    04T224+2''8L8L""6#6#6#;#;UI`I`Ibc  "+;+J+J+L  ' ' .  ' ' )  & &v':':'D'D'-'>'>,1,?,? A**,  Aww~~k* 6 6{ CDD77==-{fE  0 0 23GGLL!1!1!3U5K5KL   * ''/ A LL))/ Q)? A A As A0L M%L??Mcdtjtjj|dy)Nz /dev/consoler;)rCrGrHCONSOLE)rfrhs r7__add_console_appenderzAgent.__add_console_appender|s ""6#6#6#>#>N[r6c"tttjdt j j d|jdndj|j}ddl m }|}|j|y)z" Run agent daemon r+N-configuration-path:{0}r)get_daemon_handler) child_args) rrrC set_prefix threadingcurrent_threadsetNamerAr\azurelinuxagent.daemonrrrun)rfrsrrdaemon_handlers r7r z Agent.daemonsy =)(#  "**84""*.55d6I6IJ  >+-j1r6c>ddlm}|}|jy)z' Run provision command r)get_provision_handlerN)azurelinuxagent.pa.provisionr|ry)rfr|provision_handlers r7r$zAgent.provisions G13r6cDddlm}|}|j||y)z) Run deprovision command r)get_deprovision_handler)forcedeluserN)azurelinuxagent.pa.deprovisionrry)rfrrrdeprovision_handlers r7rzAgent.deprovisions$ K57eW=r6cTtdjt|jj tdjt|jj tdjt|jj y)z- Register agent as a service zRegister {0} servicezStop {0} servicezStart {0} serviceN)printr\rrBregister_agent_servicestop_agent_servicestart_agent_service)rfs r7register_servicezAgent.register_servicesp $++J78 **,  '' 34 &&( !((45 '')r6ctjdtjj dt j rittdkrS|jtjj tjdtjddlm}|}|j!|y#tjwxYw)z6 Run the update and extension handler ExtHandlerz2.2.53zThe agent will now check for updates and then will process extensions. Output to /dev/console will be suspended during those operations.r)get_update_handlerN)rCrtrurvrwrKrPrr rRrDrFinfodisable_console_outputazurelinuxagent.ga.updaterry)rfdebugrupdate_handlers r7run_exthandlerszAgent.run_exthandlerss ,'  "**<8  "'9';oh>W'W  ' '(<(< = 0 hi--/@+-5! --/s CC+ctj}t|jD] }t dj |||"y)Nz {0} = {1})rKget_configurationsortedkeysrr\)rf configurationks r7show_configurationzAgent.show_configurationsG..0  **,- ;A +$$Q a(89 : ;r6cBtjd|rtjdntjdd\}}}tjrt }|j d\}}tj|v}tj|v}|r|sftjd|stjd|stjdtjtj t|||}t|j}|j!|j#} tjd j%| t&||j/yy#t($rt} tj*d j%t-| tjd j%t&tjd Yd} ~ d} ~ wwxYw#||j/wwxYw) NrzRunning log collector mode fullz!Running log collector mode normal)NNNrfz7The Log Collector process is not in the proper cgroups:z unexpected cpu slicez unexpected memory sliceznLog collection successfully completed. Archive can be found at {0} and detailed log output can be found at {1}z3Log collection completed unsuccessfully. Error: {0}z'Detailed log output can be found at {0})rCrtrrshould_validate_cgroupsrget_process_cgroup_pathsrLOGCOLLECTOR_SLICEsysexitrINVALID_CGROUPS_ERRCODErrcgroupsrycollect_logs_and_get_archiver\r r[r_rstop) rf is_full_modecpu_cgroup_pathmemory_cgroup_pathlog_collector_monitor cgroups_apicpu_slice_matchesmemory_slice_matches log_collectorarchiverjs r7 collect_logszAgent.collect_logss.)  KK9 : KK; <FVB+-B  5 5 7+-K2=2V2VW]2^ /O/!3!F!F/!Y $6$I$IM_$_ $,@ UV(KK 89+KK ;<==> -(HZ[M$EmF[F[$\ ! ! % % '#@@BG KK@@FwPh@i k%0%**,1   LLNUUVZ[\V]^ _ KKAHHIab c HHQKK  %0%**,1s, A+F HA*H<H HH Hc,tdj| tj|d|d|dtdy#t$rB}tdjt |t jdYd}~yd}~wwxYw)Nz8Setting up firewall for the WALinux Agent with args: {0}waitdst_ipuidz#Successfully set the firewall rulesz(Unable to add firewall rules. Error: {0}r)rr\radd_iptables_rulesr[rrr)firewall_metadatar_s r7setup_firewallzAgent.setup_firewalls HOOPabc   / /0A&0IK\]eKf0A%0H J 7 8  <CCDKP Q HHQKK s+A B8BBN)FF)F)r%r&r'rlrRr r$rrrrr staticmethodrr5r6r7r9r9FsF30j\ 2 > *"2; '-R  r6r9ct|g}t|dkrtjdd}t|\}}}}}}}|tj k(r t y|tjk(rtty|tjk(r t|y t||}|tjk(r|j|dy|tjk(r|j|dy|tj k(r|j#y|tj$k(r|j'y|tj(k(r|j+y|tj,k(r|j/|y|tj0k(r|j3y|tj4k(r|j7|y|tj8k(r|j;|yy#t<$r4} t?j@d|tCjD| Yd} ~ yd} ~ wwxYw) z Parse command line arguments, exit with usage() on error. Invoke different methods according to different command Nrr)rAT)rFzFailed to run '{0}': {1})#lenrargv parse_argsrr/r"r1rusager,r!r9r)rr*r4r$r-rr+r r.rr0rr2rr3rr[rCr_r format_exception) argscommandrrgrrAlog_collector_full_moderagentrjs r7mainrs  | 4yA~xx|aklpaq^GUGUN4KM^-''' M&& & eg M'' ' ^, 7'.AE-777!!%!6M555!!%!7M333!M999&&(M000 M888%%e,M444((*M555""#:;M777$$%678 7 LL4 !2215 7 7 7sH3G:&G::#G:#G:#G:&$G: #G:/$G:$G:: H7*H22H7c^ tj}d}d}d}d}d}dddd}d}|D]} | dk(r tjd| } | | j d}t j j|rStdj|tj tttjdtjd | rtj}tj|jtj| rtj}tj|jtj | rtj }Wtj|jtj"| rtj"}tj|jtj$| rtj$}tj|jtj&| rtj&}&tj|jtj(| rtj(}ktj|jd | rd }tj|jd | rd }tj|jd| rd }tj|jtj*| rtj*}+tjd| rtj}Stj|jtj,| rtj,}tj|jd| rd }tj|jtj.| rtj.}tj|jd| r9tj|jd| j d|d<dtj|jd| r9tj|jd| j d|d<tj|jd| rd|d<tj}n|||||||fS)z& Parse command line arguments FN)rrrz ^([-/]*){0}z*^(?:[-/]*)configuration-path:([\w/\.\-_]+)rz,Error: Configuration file {0} does not exist)filez^([-/]*)deprovision\+userrgTrrz^([-/]*)(help|usage|\?)fullzdst_ip=(?P[\d.]{7,})rzuid=(?P[\d]+)rz (w|wait)$z-wr)rr1rematchgrouprYr<existsrr\rstderrrrr)r*r+r,r-r.r/r0r2r3) sys_argscmdrrgrrArrregex_cmd_formatargms r7rr's   C EG EN# %3 "9  HHBC HyWWQZN77>>.1DKK&(.1jj:eg XX2C 8//C XX&--m.G.GH# N++C XX&--m.B.BCS I&&C XX&--m.A.ABC H%%C XX&--m.K.KLc R//C XX&--m.J.JKS Q..C XX&--m.C.CDc J''C XX&--i8# >G XX&--g6 <E XX&--g6 <E XX&--m.F.FG M**C XX0# 6$$C XX&--m.G.GH# N++C XX&--f5s ;&* # XX&--m.I.IJC P--C XX&--.MNPS T*,((3C3J3JKj3kmp*q*w*w+ h ' XX&--.CDc J')xx0@0G0GH]0^`c'd'j'jkp'q e $ XX&--k:C @(, f %$$C g3j w~7NPa aar6ctdjttttdjt t ttdjty)z Show agent version z{0} running on {1} {2}zPython: {0}.{1}.{2}zGoal state agent: {0}N) rr\rrrrrrrr5r6r7r"r"qs[ # * *+=+6+9 ;=  & &'7'7'7 9: ! ( ()A BCr6c^d}|djtjdz }|dz }|S)z$ Return agent usage message  ausage: {0} [-verbose] [-force] [-help] -configuration-path:-deprovision[+user]|-register-service|-version|-daemon|-start|-run-exthandlers|-show-configuration|-collect-logs [-full]|-setup-firewall [-dst_ip= -uid= [-w/--wait]]r)r\rr)ss r7rr~s< A vchhqk" #A IA Hr6ctjddg}| |jdj|t t j d5}tj|||dddy#1swYyxYw)z[ Start agent daemon in a background process and set stdout/stderr to /dev/null rz-daemonNrqw)stdoutr) rrappendr\openrYdevnull subprocessPopen)rArrs r7r!r!sk HHQK #D! -44^DE bjj# ?'gg>???s A44A=__main__r)rs(&  C>**,,...V4;IE++++ bAF$sFsl(7VGbT D   ? zFr6