Edit File: linux_proc.cpython-36.pyc
3 � ey � @ s� d Z ddlmZmZ ddlmZ ddlmZ ddlm Z e� Z G dd� de �Z dd � Z d d � Zd d � ZG dd� de�Zdd� Zdd� Zdd� Zdd� Zdd� Zd&dd �Zd d � Zd d � Zd!d"� Zd#d$� Zd%e_ d%S )'zn Functions and variables to access to Linux proc directory. Constant: - PAGE_SIZE: size of a memory page � )�readlink�listdir)� getpagesize)� timestampUNIX)� timedeltac @ s e Zd ZdZdS )� ProcErrorz% Linux proc directory error. N)�__name__� __module__� __qualname__�__doc__� r r �:/tmp/pip-build-uv1wutoy/python-ptrace/ptrace/linux_proc.pyr s r c C sH d| }yt |�S tk rB } ztd||f ��W Y dd}~X nX dS )z. Open a proc entry in read only mode. z/proc/%szUnable to open %r: %sN)�open�IOErrorr )�path�filename�errr r r �openProc s r c C s t | �� }|j� S Q R X dS )zX Read the content of a proc entry. Eg. readProc("stat") to read /proc/stat. N)r �read)r Zprocfiler r r �readProc" s r c C s` y(d| |f }t |�� }|j� S Q R X W n2 tk rZ } ztd| |f ��W Y dd}~X nX dS )z� Read the content of a process entry in the proc directory. Eg. readProcessProc(pid, "status") to read /proc/pid/status. z /proc/%s/%sNz Process %s doesn't exist: %s)r r r r )�pid�keyr �procr r r r �readProcessProc+ s r c @ s* e Zd ZdZddddddd�Zd d � Zd S ) � ProcessStatea� Processus state. Attributes: - state (str): process status ('R', 'S', 'T', ...) - program (str): program name - pid (int): process identifier - ppid (int): parent process identifier - pgrp (int): process group - session (int): session identifier - tty_nr (int): tty number - tpgid (int) - utime (int): user space time (jiffies) - stime (int): kernel space time (jiffies) - starttime (int): start time ZrunningZsleepingZdiskZzombieZtracedZpaging)�R�S�D�Z�T�Wc C s� |j dd�\}}|jdd�\| _| _t| j�| _|j� }|d | _dd� |dd � D �}|d | _|d | _|d | _|d | _ |d | _ |d | _ |d | _ |d | _ d S ) N�)� �(r c S s g | ] }t |��qS r )�int)�.0�itemr r r � <listcomp>Y s z)ProcessState.__init__.<locals>.<listcomp>� � � � � � )�rsplit�splitr �programr$ �stateZppidZpgrp�sessionZtty_nrZtpgid�utimeZstimeZ starttime)�self�stat�partr r r �__init__P s zProcessState.__init__N)r r r r Z STATE_NAMESr7 r r r r r 8 s r c C s t | d�}t|�S )zC Read the process state ('stat') as a ProcessState object. r5 )r r )r r5 r r r �readProcessStatf s r8 c C s t | d�}dd� |j� D �}|S )zu Read the process memory status ('statm') as a list of integers. Values are in bytes (and not in pages). �statmc S s g | ]}t |�t �qS r )r$ � PAGE_SIZE)r% r& r r r r' t s z$readProcessStatm.<locals>.<listcomp>)r r/ )r r9 r r r �readProcessStatmn s r; c C s. t | |�}|sg S |jd�}|d s*|d= |S )z4 Read a process entry as a list of strings. � r" ���r= )r r/ )r r �datar r r �readProcessProcListx s r? c C sL yd| |f }t |�S tk rF } ztd||f ��W Y dd}~X nX dS )z Read a process link. z /proc/%s/%sz Unable to read proc link %r: %sN)r �OSErrorr )r r r r r r r �readProcessLink� s rA c c s< x6t d�D ]*} yt| �V W q tk r2 w Y q X q W dS )z� Read all processes identifiers. The function is a generator, use it with: :: for pid in readProcesses(): ... z/procN)r r$ � ValueError)r r r r � readProcesses� s rC Tc C sf yt | d�}|r|S W n tk r* Y nX y t| �}|j}|rHd| }|gS tk r` dS X dS )z� Read the process command line. If escape_stat is True, format program name with "[%s]" if the process has no command line, e.g. "[khelper]". �cmdlinez[%s]N)r? r r8 r0 )r Z escape_statrD r5 r0 r r r �readProcessCmdline� s rE c c sJ d| }x<t � D ]2}t|�}|s"q|d }|| ks<|j|�r|V qW dS )a Find all processes matching the program name pattern. Eg. pattern "ssh" will find the program "/usr/bin/ssh". This function is a generator yielding the process identifier, use it with: :: for pid in searchProcessByName(pattern): ... �/r N)rC rE �endswith)� process_name�suffixr rD r0 r r r �searchProcessesByName� s rJ c C s$ xt | �D ]}|S W td| ��dS )z� Function similar to searchProcessesByName() but only return the identifier of the first matching process. Raise a ProcError if there is no matching process. zUnable to find process: %rN)rJ r )rH r r r r �searchProcessByName� s rK c C s* t d�} | j� j� } t| d �} t| d�S )z? Get the system uptime as a datetime.timedelta object. �uptimer )�seconds)r �stripr/ �floatr )rL r r r � getUptime� s rP c C sn t jdkrhtd�} x:| D ]2}|jd�s(qt|dd� �}t|d�}|t _P qW | j� t jdkrhtd��t jS )zA Get the system boot date as a datetime.datetime object. Nr5 zbtime � Tz Unable to read system boot time!)� getSystemBoot�valuer � startswithr$ r �closer )Z stat_file�linerM Zbtimer r r rR � s rR N)T) r �osr r �resourcer Z ptrace.toolsr �datetimer r: � Exceptionr r r r �objectr r8 r; r? rA rC rE rJ rK rP rR rS r r r r �<module> s* .