File: //usr/lib/python2.7/site-packages/azurelinuxagent/agent.pyc
�
d�bc @ s� d Z d d l m Z d d l Z d d l Z d d l Z d d l Z d d l Z d d l m Z m
Z
d d l m Z d d l
j j Z d d l j j Z d d l j j Z d d l m Z d d l m Z m Z d d l m Z d d l m Z m Z d d
l m Z d d l m! Z! d d l" m# Z# m$ Z$ m% Z% m& Z& m' Z' m( Z( m) Z) m* Z* m+ Z+ m, Z, m- Z- d d
l. m/ Z/ d d l0 m1 Z1 d e2 f d � � YZ3 d e2 f d � � YZ4 d d � Z6 d � Z7 d � Z8 d � Z9 d d � Z: e; d k r�e6 � n d S( s
Module agent
i����( t print_functionN( t cgroupconfiguratort logcollector( t SystemdCgroupsApi( t ustr( t LogCollectort OUTPUT_RESULTS_FILE_PATH( t
get_osutil( t fileutilt textutil( t FlexibleVersion( t AddFirewallRules( t
AGENT_NAMEt AGENT_LONG_VERSIONt
AGENT_VERSIONt DISTRO_NAMEt DISTRO_VERSIONt PY_VERSION_MAJORt PY_VERSION_MINORt PY_VERSION_MICROt GOAL_STATE_AGENT_VERSIONt get_daemon_versiont set_daemon_version( t CollectLogsHandler( t ProvisionHandlert
AgentCommandsc B sV e Z d Z d Z d Z d Z d Z d Z d Z d Z d Z
d Z d
Z d Z
d Z RS(
sN
This is the list of all commands that the Linux Guest Agent supports
s deprovision+usert deprovisiont daemont starts register-services run-exthandlerst versions show-configurationt helps collect-logss setup-firewallt provision( t __name__t
__module__t __doc__t DeprovisionUsert Deprovisiont Daemont Startt RegisterServicet RunExthandlerst Versiont
ShowConfigt Helpt CollectLogst
SetupFirewallt Provision( ( ( s9 /usr/lib/python2.7/site-packages/azurelinuxagent/agent.pyR 4 s t Agentc B st e Z d
d � Z d � Z d � Z d � Z e e d � Z d � Z e d � Z
d � Z d � Z e
d � � Z RS( c C s | | _ t � | _ | r' t j j n t j j } t j t j j | � | j d k ra | j n | j j � } t j
| � | p� t j � } | r� t j j n t j j } t j t j j | d t j � �t j � r� t j � r� | j | � n t j � r,t j t j j t j j d t j �n t j � } y\ t j j | � ret d j | � � � n t j j | � s�t! j" | d d d d �n Wn, t k
r�} t j# d j | | � � n Xt j$ t j% � � t j j& t j% � t j' � } t j( | � t j) d � d S(
s7
Initialize agent running environment.
t paths
{0} is a filet modei� t ownert rootsB Exception occurred while creating extension log directory {0}: {1}t WALAN(* t conf_file_pathR t osutilt loggert LogLevelt VERBOSEt INFOt add_logger_appendert AppenderTypet STDOUTt Nonet get_agent_conf_file_patht conft load_conf_from_filet get_logs_verboset FILEt get_agent_log_filet get_logs_consoleR t is_provisionedt _Agent__add_console_appendert eventt send_logs_to_telemetryt TELEMETRYt WARNINGt
add_log_eventt get_ext_log_dirt osR0 t isfilet Exceptiont formatt isdirR t mkdirt errort init_event_statust get_lib_dirt joint EVENTS_DIRECTORYt init_event_loggert enable_unhandled_err_dump( t selft verboseR5 t levelt ext_log_dirt et event_dir( ( s9 /usr/lib/python2.7/site-packages/azurelinuxagent/agent.pyt __init__G s<
"
c C s t j t j j | d d �d S( NR0 s /dev/console( R7 R; R< t CONSOLE( R[ R] ( ( s9 /usr/lib/python2.7/site-packages/azurelinuxagent/agent.pyt __add_console_appender| s c C s~ t t � t j d � t j � j d � | j d k r? d n d j | j � } d d l
m } | � } | j d | � d S( s"
Run agent daemon
R% s -configuration-path:{0}i����( t get_daemon_handlert
child_argsN(
R R R7 t
set_prefixt threadingt current_threadt setNameR5 R> RQ t azurelinuxagent.daemonRd t run( R[ Re Rd t daemon_handler( ( s9 /usr/lib/python2.7/site-packages/azurelinuxagent/agent.pyR s
c C s' d d l m } | � } | j � d S( s'
Run provision command
i����( t get_provision_handlerN( t azurelinuxagent.pa.provisionRm Rk ( R[ Rm t provision_handler( ( s9 /usr/lib/python2.7/site-packages/azurelinuxagent/agent.pyR � s c C s3 d d l m } | � } | j d | d | � d S( s)
Run deprovision command
i����( t get_deprovision_handlert forcet deluserN( t azurelinuxagent.pa.deprovisionRp Rk ( R[ Rq Rr Rp t deprovision_handler( ( s9 /usr/lib/python2.7/site-packages/azurelinuxagent/agent.pyR � s c C sd t d j t � � | j j � t d j t � � | j j � t d j t � � | j j � d S( s-
Register agent as a service
s Register {0} services Stop {0} services Start {0} serviceN( t printRQ R R6 t register_agent_servicet stop_agent_servicet start_agent_service( R[ ( ( s9 /usr/lib/python2.7/site-packages/azurelinuxagent/agent.pyt register_service� s
c C s� t j d � t j � j d � t j � rv t � t d � k rv | j t j
j � z t j d � Wd t j
� Xn d d l m } | � } | j | � d S( s6
Run the update and extension handler
t
ExtHandlers 2.2.53u� The agent will now check for updates and then will process extensions. Output to /dev/console will be suspended during those operations.Ni����( t get_update_handler( R7 Rf Rg Rh Ri R@ RE R R
RG R8 R: t infot disable_console_outputt azurelinuxagent.ga.updateR{ Rk ( R[ t debugR{ t update_handler( ( s9 /usr/lib/python2.7/site-packages/azurelinuxagent/agent.pyt run_exthandlers� s
! c C sG t j � } x4 t | j � � D] } t d j | | | � � q Wd S( Ns {0} = {1}( R@ t get_configurationt sortedt keysRu RQ ( R[ t
configurationt k( ( s9 /usr/lib/python2.7/site-packages/azurelinuxagent/agent.pyt show_configuration� s c C s5 | r t d � n
t d � t j � r� t j d � \ } } t j | k } t j | k } | sj | r� t d � | s� t d � n | s� t d � n t j t j
� q� n y2 t | � } | j � } t d j
| t � � WnL t k
r0} t d j
t | � � � t d j
t � � t j d
� n Xd S( Ns Running log collector mode fulls! Running log collector mode normalR[ s7 The Log Collector process is not in the proper cgroups:s unexpected cpu slices unexpected memory slicesn Log collection successfully completed. Archive can be found at {0} and detailed log output can be found at {1}s3 Log collection completed unsuccessfully. Error: {0}s' Detailed log output can be found at {0}i ( Ru R t should_validate_cgroupsR t! get_process_cgroup_relative_pathsR t LOGCOLLECTOR_SLICEt syst exitR t INVALID_CGROUPS_ERRCODER t collect_logs_and_get_archiveRQ R RP R ( R[ t is_full_modet cpu_cgroup_patht memory_cgroup_patht cpu_slice_matchest memory_slice_matchest
log_collectort archiveR_ ( ( s9 /usr/lib/python2.7/site-packages/azurelinuxagent/agent.pyt collect_logs� s.
c C s� t d j | � � y- t j | d | d | d � t d � Wn9 t k
r{ } t d j t | � � � t j d � n Xd S( Ns8 Setting up firewall for the WALinux Agent with args: {0}t waitt dst_ipt uids# Successfully set the firewall ruless( Unable to add firewall rules. Error: {0}i ( Ru RQ R t add_iptables_rulesRP R R� R� ( t firewall_metadataRT ( ( s9 /usr/lib/python2.7/site-packages/azurelinuxagent/agent.pyt setup_firewall� s N( R R! R> Ra RG R R t FalseR Ry R� R� R� t staticmethodR� ( ( ( s9 /usr/lib/python2.7/site-packages/azurelinuxagent/agent.pyR/ F s 5 c
C s | d k r g } n t | � d k r7 t j d } n t | � \ } } } } } } } | t j k rq t � n�| t j k r� t t
� � n~| t j k r� t d | � n_y-t
| d | �} | t j k r� | j | d t �n� | t j k r| j | d t �n� | t j k r*| j � n� | t j k rF| j � n� | t j k rb| j � ny | t j k r�| j | � nZ | t j k r�| j � n> | t j k r�| j | � n | t j k r�| j | � n Wn/ t! k
r
} t" j# d | t$ j% | � � n Xd S( s
Parse command line arguments, exit with usage() on error.
Invoke different methods according to different command
i i R5 Rr u Failed to run '{0}': {1}N(& R> t lenR� t argvt
parse_argsR R) R R+ Ru t usageR& R R/ R# R t TrueR$ R� R. R R' Ry R% R R( R� R* R� R, R� R- R� RP R7 RT R t format_exception(
t argst commandRq R\ R R5 t log_collector_full_modeR� t agentR_ ( ( s9 /usr/lib/python2.7/site-packages/azurelinuxagent/agent.pyt main� sF !
c C s t j } t } t } t } d } t } i d d 6d d 6d d 6} d } x�| D]�} | d k rg qO n t j d | � }
|
d k r� |
j d � } t j j | � s�t
d j | � d t j
�t
t � � t j d � q�qO t j d
| � r t j } qO t j | j t j � | � r*t j } qO t j | j t j � | � rTt j } qO t j | j t j � | � r~t j } qO t j | j t j � | � r�t j } qO t j | j t j � | � r�t j } qO t j | j t j � | � r�t j } qO t j | j d � | � r t } qO t j | j d � | � rDt } qO t j | j d
� | � rht } qO t j | j t j � | � r�t j } qO t j d | � r�t j } qO t j | j t j � | � r�t j } qO t j | j d � | � r�t } qO t j | j t j � | � r(t j } qO t j | j d � | � rnt j | j d � | � j d � | d <qO t j | j d � | � r�t j | j d � | � j d � | d <qO t j | j d � | � r�d | d <qO t j } PqO W| | | | | | | f S( s&
Parse command line arguments
R� R� t R� s ^([-/]*){0}s* ^(?:[-/]*)configuration-path:([\w/\.\-_]+)i s, Error: Configuration file {0} does not existt files ^([-/]*)deprovision\+userR\ R Rq s ^([-/]*)(help|usage|\?)t fulls dst_ip=(?P<dst_ip>[\d.]{7,})s uid=(?P<uid>[\d]+)s (w|wait)$s -wN( R R+ R� R> t ret matcht groupRN R0 t existsRu RQ R� t stderrR� R� R# R$ R% R& R' R( R) R� R* R, R- ( t sys_argst cmdRq R\ R R5 R� R� t regex_cmd_formatt argt m( ( s9 /usr/lib/python2.7/site-packages/azurelinuxagent/agent.pyR� s|
+
c C sI t d j t t t � � t d j t t t � � t d j t � � d S( s
Show agent version
s {0} running on {1} {2}s Python: {0}.{1}.{2}s Goal state agent: {0}N( Ru RQ R
R R R R R R ( ( ( s9 /usr/lib/python2.7/site-packages/azurelinuxagent/agent.pyR h s
c C s. d } | d j t j d � 7} | d 7} | S( s$
Return agent usage message
s
s usage: {0} [-verbose] [-force] [-help] -configuration-path:<path to configuration file>-deprovision[+user]|-register-service|-version|-daemon|-start|-run-exthandlers|-show-configuration|-collect-logs [-full]|-setup-firewall [-dst_ip=<IP> -uid=<UID> [-w/--wait]]i ( RQ R� R� ( t s( ( s9 /usr/lib/python2.7/site-packages/azurelinuxagent/agent.pyR� u s
c
C sp t j d d g } | d k r8 | j d j | � � n t t j d � � } t j | d | d | �Wd QXd S( s[
Start agent daemon in a background process and set stdout/stderr to
/dev/null
i s -daemons -configuration-path:{0}t wt stdoutR� N(
R� R� R> t appendRQ t openRN t devnullt
subprocesst Popen( R5 R� R� ( ( s9 /usr/lib/python2.7/site-packages/azurelinuxagent/agent.pyR � s
t __main__(< R" t
__future__R RN R� R� R� Rg t azurelinuxagent.commonR R t azurelinuxagent.common.cgroupapiR t azurelinuxagent.common.conft commonR@ t azurelinuxagent.common.eventRH t azurelinuxagent.common.loggerR7 t azurelinuxagent.common.futureR t# azurelinuxagent.common.logcollectorR R t azurelinuxagent.common.osutilR t azurelinuxagent.common.utilsR R t- azurelinuxagent.common.utils.flexible_versionR
t( azurelinuxagent.common.utils.networkutilR t azurelinuxagent.common.versionR R
R R R R R R R R R t azurelinuxagent.ga.collect_logsR t$ azurelinuxagent.pa.provision.defaultR t objectR R/ R> R� R� R R� R R ( ( ( s9 /usr/lib/python2.7/site-packages/azurelinuxagent/agent.pyt <module> s: L� + J