File: //usr/lib/python2.7/site-packages/azurelinuxagent/common/cgroupapi.pyc
�
d�bc @ sz d d l 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 d d l m Z d d l
m Z d d l m
Z
d d l m Z m Z m Z m Z d d l m Z d d l m Z d d l m Z m Z d d
l m Z m Z m Z d d l m Z d d l m! Z! d
Z" d d g Z# d Z$ d e f d � � YZ% d e& f d � � YZ' d e' f d � � YZ( d S( i����N( t logger( t CpuCgroup( t CGroupsTelemetry( t get_agent_pid_file_path( t CGroupsExceptiont ExtensionErrorCodest ExtensionErrort ExtensionOperationError( t ustr( t systemd( t fileutilt shellutil( t handle_process_completiont read_outputt TELEMETRY_MESSAGE_MAX_LEN( t FlexibleVersion( t
get_distros /sys/fs/cgroupt cput memorys azure-vmextensionst SystemdRunErrorc B s e Z d Z d d � Z RS( s'
Raised when systemd-run fails
c C s t t | � j | � d S( N( t superR t __init__( t selft msg( ( sD /usr/lib/python2.7/site-packages/azurelinuxagent/common/cgroupapi.pyR 1 s N( t __name__t
__module__t __doc__t NoneR ( ( ( sD /usr/lib/python2.7/site-packages/azurelinuxagent/common/cgroupapi.pyR , s t
CGroupsApic B sS e Z e d � � Z e d � � Z e d � � Z e d � � Z e d � � Z RS( c C s� t � } | d } y t | d � } Wn t k
r; t SX| j � d k r] | j d k p� | j � d k o� | j d k r� | j d k p� | j d k S(
Ni i t ubuntui t centost redhati i ( R R ( R R t
ValueErrort Falset lowert majort minor( t distro_infot distro_namet distro_version( ( sD /usr/lib/python2.7/site-packages/azurelinuxagent/common/cgroupapi.pyt cgroups_supported6 s
!c C s^ y" x | D] } t j | � q
WWn5 t k
rY } t j d j | j t | � � � n Xd S( NsX Cannot add cgroup '{0}' to tracking list; resource usage will not be tracked. Error: {1}( R t track_cgroupt ExceptionR t warnt formatt pathR ( t extension_cgroupst cgroupt exception( ( sD /usr/lib/python2.7/site-packages/azurelinuxagent/common/cgroupapi.pyt
track_cgroupsB s
c C sT t t j j | d � d � �0 } g | j � j � D] } t | � ^ q4 SWd QXd S( Ns cgroup.procst r( t opent osR- t joint readt splitt int( t cgroup_patht cgroup_procst pid( ( sD /usr/lib/python2.7/site-packages/azurelinuxagent/common/cgroupapi.pyt get_processes_in_cgroupK s !c C s@ g } xg d d g D]Y } t j j t | d d � } t j j | � r t j d | � | j | | f � q q Wz� x� | D]y \ } } t j j | d � } t j j | � rz t j | � j
� } t j � } t
| � | k r� | | | � q� qz qz WWd x7 | D]/ \ } } t j d | � t j | d t �qWXt | � S( sE
Previous versions of the daemon (2.2.31-2.2.40) wrote their PID to /sys/fs/cgroup/{cpu,memory}/WALinuxAgent/WALinuxAgent;
starting from version 2.2.41 we track the agent service in walinuxagent.service instead of WALinuxAgent/WALinuxAgent. Also,
when running under systemd, the PIDs should not be explicitly moved to the cgroup filesystem. The older daemons would
incorrectly do that under certain conditions.
This method checks for the existence of the legacy cgroups and, if the daemon's PID has been added to them, executes the
given operation on the cgroups. After this check, the method attempts to remove the legacy cgroups.
:param operation:
The function to execute on each legacy cgroup. It must take 2 arguments: the controller and the daemon's PID
R R t WALinuxAgents Found legacy cgroup {0}s cgroup.procsNs Removing {0}t
ignore_errors( R4 R- R5 t CGROUPS_FILE_SYSTEM_ROOTt existsR t infot appendR
t read_filet stripR t get_daemon_pidR t shutilt rmtreet Truet len( t operationt legacy_cgroupst
controllerR/ t
procs_filet procs_file_contentst
daemon_pidt _( ( sD /usr/lib/python2.7/site-packages/azurelinuxagent/common/cgroupapi.pyt _foreach_legacy_cgroupP s$ c C s t t j t � � j � � S( N( R8 R
RC R RD ( ( ( sD /usr/lib/python2.7/site-packages/azurelinuxagent/common/cgroupapi.pyRE u s ( R R t staticmethodR( R1 R< RQ RE ( ( ( sD /usr/lib/python2.7/site-packages/azurelinuxagent/common/cgroupapi.pyR 5 s
%t SystemdCgroupsApic B s� e Z d Z d � Z d � Z d � Z e d � � Z d � Z d � Z e d � � Z
e d � � Z e d � � Z e
j d
� Z d � Z RS( s'
Cgroups interface via systemd
c C s. d | _ d | _ g | _ t j � | _ d S( N( R t _cgroup_mountpointst _agent_unit_namet _systemd_run_commandst threadingt RLockt _systemd_run_commands_lock( R ( ( sD /usr/lib/python2.7/site-packages/azurelinuxagent/common/cgroupapi.pyR s c C s | j �
| j SWd QXd S( s^
Returns a list of the systemd-run commands currently running (given as PIDs)
N( RY RV ( R ( ( sD /usr/lib/python2.7/site-packages/azurelinuxagent/common/cgroupapi.pyt get_systemd_run_commands� s
c C s� | j d k r� d } d } xq t j d d d g � j � D]Q } t j d | � } | d k r: | j d � } d | k r� | } q� | } q: q: Wi | d 6| d 6| _ n | j d | j d f S(
s�
Returns a tuple with the mount points for the cpu and memory controllers; the values can be None
if the corresponding controller is not mounted
t mounts -tR/ s% on\s+(?P<path>/\S+(memory|cpuacct))\sR- t cpuacctR R N( RT R R t run_commandt
splitlinest ret searcht group( R R R t linet matchR- ( ( sD /usr/lib/python2.7/site-packages/azurelinuxagent/common/cgroupapi.pyt get_cgroup_mount_points� s %
c C s� d } d } x� t j d j | � � j � D]� } t j d | � } | d k r+ | j d � } | j d � d k r� | j d � j d � n d } | d k r� | } q� | } q+ q+ W| | f S( s�
Returns a tuple with the path of the cpu and memory cgroups for the given process (relative to the mount point of the corresponding
controller).
The 'process_id' can be a numeric PID or the string "self" for the current process.
The values returned can be None if the process is not in a cgroup for that controller (e.g. the controller is not mounted).
s /proc/{0}/cgroups5 \d+:(?P<controller>(memory|.*cpuacct.*)):(?P<path>.+)RL R- t /R N( R R
RC R, R^ R_ Rc Ra t lstrip( t
process_idt cpu_patht memory_pathRb Rc RL R- ( ( sD /usr/lib/python2.7/site-packages/azurelinuxagent/common/cgroupapi.pyt! get_process_cgroup_relative_paths� s
% 3
c C s� | j | � \ } } | j � \ } } | d k rT | d k rT t j j | | � n d } | d k r� | d k r� t j j | | � n d } | | f S( s<
Returns a tuple with the path of the cpu and memory cgroups for the given process. The 'process_id' can be a numeric PID or the string "self" for the current process.
The values returned can be None if the process is not in a cgroup for that controller (e.g. the controller is not mounted).
N( Rj Rd R R4 R- R5 ( R Rg t cpu_cgroup_relative_patht memory_cgroup_relative_patht cpu_mount_pointt memory_mount_pointt cpu_cgroup_patht memory_cgroup_path( ( sD /usr/lib/python2.7/site-packages/azurelinuxagent/common/cgroupapi.pyt get_process_cgroup_paths� s
33c C s� t j | d � } | j � \ } } | d k rI t j j | | d � n d } | d k rt t j j | | d � n d } | | f S( s'
Returns a tuple with the path of the cpu and memory cgroups for the given unit.
The values returned can be None if the controller is not mounted.
Ex: ControlGroup=/azure.slice/walinuxagent.service
controlgroup_path[1:] = azure.slice/walinuxagent.service
t ControlGroupi N( R t get_unit_propertyRd R R4 R- R5 ( R t unit_namet controlgroup_pathRm Rn Ro Rp ( ( sD /usr/lib/python2.7/site-packages/azurelinuxagent/common/cgroupapi.pyt get_unit_cgroup_paths� s
++c C s� x� t j d d d g � j � D]| } t j d | � } | d k r | j d � } d } t j j | d � } t j j
| � r� t j | � } n | | f Sq Wd S( s�
Returns a tuple with the mount point for the cgroups v2 controllers, and the currently mounted controllers;
either value can be None if cgroups v2 or its controllers are not mounted
R[ s -tt cgroup2s on\s+(?P<path>/\S+)\sR- s cgroup.controllersN( NN(
R R] R^ R_ R` R Ra R4 R- R5 R@ R
RC ( Rb Rc t mount_pointt controllerst controllers_file( ( sD /usr/lib/python2.7/site-packages/azurelinuxagent/common/cgroupapi.pyt get_cgroup2_controllers� s
% c C sS | j d � t | j t � d d d d �} d j | � } | | k pR | | k S( Ni t encodings utf-8t errorst backslashreplaces Unit {0} not found.( t seekR R6 R R, ( t
scope_namet stderrt unit_not_found( ( sD /usr/lib/python2.7/site-packages/azurelinuxagent/common/cgroupapi.pyt _is_systemd_failure� s
!c C s t d | j d d � d S( Nt -RP s .slice( t EXTENSION_SLICE_PREFIXt replace( t extension_name( ( sD /usr/lib/python2.7/site-packages/azurelinuxagent/common/cgroupapi.pyt get_extension_slice_name s c C s~ d j | t j � � } | j | � } | j �] t j d j | | | � d | d | d | d | d | d t j �}
| j j
|
j � Wd QX| d } t j
d
| � yr t j j d | � } | j � \ } } | d k r� t j
d � n+ t j j | | � } t j t | | � � Wnp t k
re} | j d
k rLt j
d � n t j
d t | � � n) t k
r�} t j
d t | � � n Xz� y/ t d |
d | d | d | d | d |
� SWn� t k
rQ} | j | | � s�� n t | | � } | j d � | j d � t | t � r8d | j | f } n
d | } t! | � � n XWd | j � | j j" |
j � Wd QXXd S( Ns {0}_{1}s. systemd-run --unit={0} --scope --slice={1} {2}t shellt cwdt stdoutR� t envt
preexec_fns .scopes Started extension in unit '{0}'s$ azure.slice/azure-vmextensions.slices@ The CPU controller is not mounted; will not track resource usagei sF The extension command already completed; will not track resource usages> Failed to start tracking resource usage for the extension: {0}t processt commandt timeoutt
error_codei s1 Systemd process exited with code %s and output %ss Systemd timed-out, output: %s(# R, t uuidt uuid4R� RY t
subprocesst PopenR4 t setsidRV RB R; R RA R- R5 Rd R R R) R t IOErrort errnoR R* R R R� R
t truncatet
isinstanceR t exit_codeR t remove( R R� R� t cmd_nameR� R� R� R� R� R� R� t scopet extension_slice_nameR� R� t cgroup_relative_patht cpu_cgroup_mountpointRP Ro t et process_outputt err_msg( ( sD /usr/lib/python2.7/site-packages/azurelinuxagent/common/cgroupapi.pyt start_extension_command sX
c C s t j d � � S( s�
Previous versions of the daemon (2.2.31-2.2.40) wrote their PID to /sys/fs/cgroup/{cpu,memory}/WALinuxAgent/WALinuxAgent;
starting from version 2.2.41 we track the agent service in walinuxagent.service instead of WALinuxAgent/WALinuxAgent. If
we find that any of the legacy groups include the PID of the daemon then we need to disable data collection for this
instance (under systemd, moving PIDs across the cgroup file system can produce unpredictable results)
c W s d S( N( R ( RP ( ( sD /usr/lib/python2.7/site-packages/azurelinuxagent/common/cgroupapi.pyt <lambda>P s ( R RQ ( R ( ( sD /usr/lib/python2.7/site-packages/azurelinuxagent/common/cgroupapi.pyt cleanup_legacy_cgroupsI s ( R R R R RZ Rd RR Rj Rq Rv R{ R� R� R t PluginUnknownFailureR� R� ( ( ( sD /usr/lib/python2.7/site-packages/azurelinuxagent/common/cgroupapi.pyRS z s B() R4 R_ RF R� RW R� t azurelinuxagent.commonR t azurelinuxagent.common.cgroupR t' azurelinuxagent.common.cgroupstelemetryR t azurelinuxagent.common.confR t azurelinuxagent.common.exceptionR R R R t azurelinuxagent.common.futureR t azurelinuxagent.common.osutilR t azurelinuxagent.common.utilsR
R t1 azurelinuxagent.common.utils.extensionprocessutilR R
R t- azurelinuxagent.common.utils.flexible_versionR t azurelinuxagent.common.versionR R? t CGROUP_CONTROLLERSR� R t objectR RS ( ( ( sD /usr/lib/python2.7/site-packages/azurelinuxagent/common/cgroupapi.pyt <module> s, " E