o i@s<ddlZddlZddlZGdddeZGdddeZdS)Nc@seZdZdZdS)UnresolvableContainerIDz1 Unable to get container ID from cgroup. N)__name__ __module__ __qualname____doc__rrO/home/ubuntu/.local/lib/python3.10/site-packages/datadog/dogstatsd/container.pyr src@sreZdZdZdZdZdZdZdZdZ dZ d Z d Z e d Ze d e e e Zd dZddZddZddZdS)CgroupaM A reader class that retrieves either: - The current container ID parsed from the cgroup file - The cgroup controller inode. Returns: object: Cgroup Raises: `NotImplementedError`: No proc filesystem is found (non-Linux systems) `UnresolvableContainerID`: Unable to read the container ID z/proc/self/cgroupz/sys/fs/cgroupz/proc/self/ns/cgroupmemoryl_zH[0-9a-f]{8}[-_][0-9a-f]{4}[-_][0-9a-f]{4}[-_][0-9a-f]{4}[-_][0-9a-f]{12}z [0-9a-f]{64}z[0-9a-f]{32}-\d+z^(\d+):([^:]*):(.+)$z!(?:.+)?({0}|{1}|{2})(?:\.scope)?$cCs$|r ||_dS||_dS)N)_is_host_cgroup_namespace_read_cgroup_path container_id_get_cgroup_from_inodeselfrrr__init__,s zCgroup.__init__cCs@ztj|jrt|jj|jkWSdWStyYdSw)z;Check if the current process is in a host cgroup namespace.F)ospathexistsCGROUP_NS_PATHstatst_inoHOST_CGROUP_NAMESPACE_INODE Exceptionrrrrr 2s  z Cgroup._is_host_cgroup_namespacec Cs(z_t|jddM}|D]@}|}|j|}|sq |\}}}dd|dD}t|rK|j| }|rKd | dWdWSq WdWdS1sXwYWdSt y}z|j t jkrutd |jWYd}~dSd}~wty}ztd t|d}~ww) z+Read the container ID from the cgroup file.rmodecSsg|]}|qSrr).0prrr Gsz,Cgroup._read_cgroup_path../zci-{0}NzUnable to open {}.z!Unable to read the container ID: )open CGROUP_PATHstripLINE_REmatchgroupssplitlen CONTAINER_REpopformatgroupIOErrorerrnoENOENTNotImplementedErrorrrstr)rfpliner'_rpartserrrr =s<    zCgroup._read_cgroup_pathcCsi}t|jdd1}|D]&}|d}t|dkrq |d|jks*|d|jkr2|d||d<q Wdn1s=wY|j|jfD]*}||vrrtj |j |||dkr^||nd }t |j }|dkrrd |SqHdS) z,Read the container ID from the cgroup inode.rr:r"Nr!r zin-{0})r#r$r%r)r*CGROUPV1_BASE_CONTROLLERCGROUPV2_BASE_CONTROLLERrrjoinCGROUP_MOUNT_PATHrrr-)rcgroup_controllers_pathsr4r5tokens controller inode_pathinoderrrrSs2   zCgroup._get_cgroup_from_inodeN)rrrrr$r?rr<r=r UUID_SOURCECONTAINER_SOURCE TASK_SOURCErecompiler&r-r+rr r rrrrrr s"   r )r0rrHrrobjectr rrrrs