3
�[g� � b @ s� d Z ddlmZ ddlZddlZddlZddlZddlZddlZddl Z ddl
Z
ddl Z ddl Z ddl
Z
ddlZddlZddlZddlZddlZddlZddlZddlZddlZddlZddlZddlZddlZddlmZ ddlm Z ddlm Z y ddl Z W n e k
�r dZ Y nX ddl Z ddl m!Z! ddl m"Z" dd l m#Z# dd
l m$Z$ dd l m%Z% dd l m&Z& dd
l m'Z' ddl m(Z( ddl)m*Z* ddl)m+Z+ ddl)m,Z, ddl)m-Z- ddl)m.Z. ddl/m0Z0 ddl/m1Z1 ddl/m2Z2 ddl/m3Z3 ddl/m4Z4 ddl/m5Z5 ddl/m6Z6 yddlm7Z7 W n< e k
�r` ej8� � ej9d � ddl7Z7W dQ R X Y nX e0�rrddl:Z:n ddl;ZdZ:e&�r�dd l<m=Z= d d d d!d"d#d$d%d&d'd(d)d*d+d,d-d.d/d0d1d2d3d4d5d6d7d8d9d:d;d<d=d>d?d@dAdBdCdDdEdFdGdHdIdJdKdLdMdNdOdPdQdRdSdTdUdVdWdXdYdZd[d\d]d^d_d`dadbdcdddedfdgdhdidjgMZ>dkej?kZ@d e jAkZBdle jAk�pVdme jAkZCeB�p`eCZDdne jAkZEdoe jAkZFe"�r�eC�r�eGdp��ZHdqeHjI� kZJW dQ R X ndrZJejKd�kZLe
jM� dukZNe,dvdw� �ZOe#�r�eO� d�kZPeO� d�kZQndrZPdrZQdxZRd�ZSd�ZTd{ZUeD�r$eRd}9 ZReUd}9 ZUeSd~9 ZSeTd}9 ZTe jVdk�r>d�e jW� ZXn d�e jW� ZXd�ZYe0�rbd�jZd�d��Z[nd�Z[ej\� j]� d�kZ^e j_j`e j_jae j_jbec�d�d���Zde jAjed�e j_jaedd���Zfe j_j`e j_jbec��Zgehe jid��Zjehe d��Zkehe jid��Zlehe d��Zmehe jid��Znehe jid��Zoe&�oe' Zpehe d��Zqehe jid��Zrehe jid��Zsehe jid��Ztehe d��Zuyeu�o\eve jw� �ZxW n eyk
�rz drZxY nX ehe d��Zzehe d��Z{ehe jid��Z|e#�s�e!�o�e j}� dkZ~d�d�� Ze� \Z�Z�eGe j�d��Z�ej�e�j�� d�d�� e�e �D �Z�e�ed�e�� �Z�e�� Z�e�� Z�G d�dH� dHej��Z�d�d�� Z�e�d�d�dE��Z�e�d�dG� �Z�d�dF� Z�e�d�dB� �Z�e�d�d�� �Z�ej�eUfd�dC�Z�d�d�dD�Z�d�dZ� Z�d�dY� Z�G d�d�� d��Z�e�e j�deUd�d��d�d\� �Z�e�e2e�fdeUd�d��d�d�d]��Z�e�e�deUd�d��d�d[� �Z�d�dU� Z�d�d�� Z�ej�d�dT� �Z�d�dV� Z�d�d�dW�Z�d�d�dX�Z�G d�dS� dS�Z�e dk�r<e�Z G d�d�� d�ej��Z�e�e_�G d�dN� dNe��Z�e j�j�e@d�dG d�dM� dMe���Z�d�dQ� Z�d�dR� Z�d�dDŽ Z�G d�dO� dO�Z�G d�dP� dP�Z�eRfd�dL�Z�d�d�dJ�Z�d�d�dK�Z�d�d�d`�Z�ee dfd�da�Z�ej fd�db�Z�d�d�dc�Z�d�dd� Z�ej�d�de� �Z�d�d^� Z�d�dք Z�d�d_� Z�d�df� Z�d�dg� Z�d�dh� Z�d�dj� Z�e&�rpej�d�d�di��Z�nej�d�d�di��Z�ej�d�d߄ �Z�e&�r�ejej�d�d� � dS )�zTest utilities.� )�print_functionN)�AF_INET)�AF_INET6)� SOCK_STREAM)�AIX)�LINUX)�MACOS)�NETBSD)�OPENBSD)�POSIX)�SUNOS)�WINDOWS)� bytes2human)�debug)�memoize)� print_color)�
supports_ipv6)�PY3)�FileExistsError)�FileNotFoundError)�range)�super)�unicode)�which)�mock�ignore)�wait_pid�APPVEYOR�DEVNULL�GLOBAL_TIMEOUT�TOLERANCE_SYS_MEM�
NO_RETRIES�PYPY�
PYTHON_EXE�PYTHON_EXE_ENV�ROOT_DIR� SCRIPTS_DIR�
TESTFN_PREFIX�UNICODE_SUFFIX�INVALID_UNICODE_SUFFIX�
CI_TESTING�VALID_PROC_STATUSES�TOLERANCE_DISK_USAGE�IS_64BIT�HAS_CPU_AFFINITY� HAS_CPU_FREQ� HAS_ENVIRON�HAS_PROC_IO_COUNTERS�
HAS_IONICE�HAS_MEMORY_MAPS�HAS_PROC_CPU_NUM�
HAS_RLIMIT�HAS_SENSORS_BATTERY� HAS_BATTERY�HAS_SENSORS_FANS�HAS_SENSORS_TEMPERATURES�HAS_NET_CONNECTIONS_UNIX� MACOS_11PLUS� MACOS_12PLUS�COVERAGE�AARCH64� QEMU_USER�PYTEST_PARALLEL�pyrun� terminate�
reap_children�spawn_testproc� spawn_zombie�spawn_children_pair�
ThreadTask�unittest�skip_on_access_denied�skip_on_not_implemented�retry_on_failure�TestMemoryLeak�PsutilTestCase�process_namespace�system_namespace�
print_sysinfo�is_win_secure_system_proc� fake_pytest�chdir� safe_rmpath�
create_py_exe� create_c_exe�
get_testfn�
get_winver�kernel_version�
call_until� wait_for_pid�
wait_for_file�check_net_address�filter_proc_net_connections�
get_free_port� bind_socket�bind_unix_socket�tcp_socketpair�unix_socketpair�create_sockets�
reload_module�import_module_by_path�warn�copyload_shared_lib�
is_namedtuple�__pypy__�GITHUB_ACTIONS� CIBUILDWHEELZ COVERAGE_RUNZPYTEST_XDIST_WORKERz/proc/1/cmdlinez
/bin/qemu-F� � Zaarch64c C sn t j� d } ttt| jd�d d� ��}|d
krjtjtj dddgd d
id d �} ttt| jd�d d� ��}|S )Nr �.rm �
� z-sSz-cz-import platform; print(platform.mac_ver()[0])ZSYSTEM_VERSION_COMPAT�0T)�env�universal_newlines)rp rq )
�platform�mac_ver�tuple�map�int�split�
subprocess� check_output�sys�
executable)Z version_str�version� r� �A/usr/local/lib64/python3.6/site-packages/psutil/tests/__init__.py�
macos_version� s r� rp � � � i � � �javaz $psutil-%s-z @psutil-%s-u -ƒőős f���utf8�surrogateescapeu fÀ�ascii�us-asciiz..ZPSUTIL_SCRIPTS_DIR�scripts� cpu_affinity�cpu_freq�environ�
getloadavg�ionice� memory_maps�net_io_counters�cpu_num� io_counters�rlimit�sensors_battery� sensors_fans�sensors_temperatures�threadsc C s� dd� } t jj� }ttdd �}trFtjd
krF|d k rFtj|d< ||fS trTtj|fS t r�| tj�p�| t j
j tj��p�| t dtjd d� ��p�| t
j� j� �}|s�td ��||fS t j
j tj�}t j
j|�s�t|��||fS d S ) Nc S s< y t j| dgt jt jd� W n t jk
r2 d S X | S d S )Nz-V)�stdout�stderr)r{ �
check_call�PIPE�CalledProcessError)�exer� r� r� �attempt� s z _get_py_exe.<locals>.attemptZ_base_executabler� � �__PYVENV_LAUNCHER__z python%s.%srm z"can't find python exe real abspath)r� r� )�osr� �copy�getattrr} r
� version_infor~ rk r �path�realpathr �psutil�Processr� �
ValueError�exists�AssertionError)r� rs �baser� r� r� r� � _get_py_exe� s&
r� zr+c C s g | ]}|j d �rtt|��qS )ZSTATUS_)�
startswithr� r� )�.0�xr� r� r� �
<listcomp>% s r� �AF_UNIXc sP e Zd ZdZ� fdd�Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dd� Z
� Z S )rG z6A thread task which does nothing expect staying alive.c s$ t � j� d| _d| _tj� | _d S )NFg����MbP?)r �__init__�_running� _interval� threading�Event�_flag)�self)� __class__r� r� r� 5 s
zThreadTask.__init__c C s | j j}d|| jt| �f S )Nz<%s running=%s at %#x>)r� �__name__r� �id)r� �namer� r� r� �__repr__; s zThreadTask.__repr__c C s | j � | S )N)�start)r� r� r� r� � __enter__? s zThreadTask.__enter__c O s | j � d S )N)�stop)r� �args�kwargsr� r� r� �__exit__C s zThreadTask.__exit__c C s( | j rtd��tjj| � | jj� dS )zStart thread and keep it running until an explicit
stop() request. Polls for shutdown every 'timeout' seconds.
zalready startedN)r� r� r� �Threadr� r� �wait)r� r� r� r� r� F s zThreadTask.startc C s, d| _ | jj� x| j r&tj| j� qW d S )NT)r� r� �set�time�sleepr� )r� r� r� r� �runO s
zThreadTask.runc C s | j std��d| _ | j� dS )z8Stop thread execution and and waits until it is stopped.zalready stoppedFN)r� r� �join)r� r� r� r� r� U s zThreadTask.stop) r� �
__module__� __qualname__�__doc__r� r� r� r� r� r� r� �
__classcell__r� r� )r� r� rG 2 s c s t j� �� fdd��}|S )Nc s, y
� | |�S t k
r& t� � Y nX d S )N)� ExceptionrC )r� r� )�funr� r� �wrapperc s
z&_reap_children_on_err.<locals>.wrapper)� functools�wraps)r� r� r� )r� r� �_reap_children_on_errb s r� c K s� |j dt� |j dt� |j dtj� � |j dt� trHd}|j d|� | dkr�ttj� d�}zLt|� d d
| d }td |g} t j
| f|�}t j |� t
|d
d
d� W dt|� X n"t j
| f|�}t j |� t|j� |S )a Create a python subprocess which does nothing for some secs and
return it as a subprocess.Popen instance.
If "cmd" is specified that is used instead of python.
By default stdin and stdout are redirected to /dev/null.
It also attempts to make sure the process is in a reasonably
initialized state.
The process is registered for cleanup on reap_children().
�stdinr� �cwdrs i �
creationflagsN)�dirz import time;zopen(r'%s', 'w').close();z&[time.sleep(0.1) for x in range(100)];z-cT)�delete�empty)�
setdefaultr r� �getcwdr$ r
rW rT r# r{ �Popen�_subprocesses_started�addr\ r[ �pid)�cmd�kwdsZCREATE_NO_WINDOW�testfnZpyline�sprocr� r� r� rD n s*
c C s� d} t tj� d�}zvtjdtjj|�tf �}trDt |dd�\}} n t |�\}} t
j |j �}t
t|ddd��}tj|� t
j |�}||fS t|� | dk r�t| � X dS ) a Create a subprocess which creates another one as in:
A (us) -> B (child) -> C (grandchild).
Return a (child, grandchild) tuple.
The 2 processes are fully initialized and will live for 60 secs
and are registered for cleanup on reap_children().
N)r� aV import subprocess, os, sys, time
s = "import os, time;"
s += "f = open('%s', 'w');"
s += "f.write(str(os.getpid()));"
s += "f.close();"
s += "[time.sleep(0.1) for x in range(100 * 6)];"
p = subprocess.Popen([r'%s', '-c', s])
p.wait()
r )r� TF)r� r� )rW r� r� �textwrap�dedentr� �basenamer# r
rA r� r� r� ry r\ �
_pids_startedr� rT )�tfiler� �s�subp�childZgrandchild_pidZ
grandchildr� r� r� rF � s
c s� t js
t�t� } tjd| �}d}t| �}z�|jt� t |�\}}|j
� \}}zPt j |j � gg g t� t
|jd��}tj|� t j|�� t� fdd�� |� fS |j� X W d|j� t| � |dk r�t|� X dS )z�Create a zombie process and return a (parent, zombie) process tuple.
In order to kill the zombie parent must be terminate()d first, then
zombie must be wait()ed on.
a import os, sys, time, socket, contextlib
child_pid = os.fork()
if child_pid > 0:
time.sleep(3000)
else:
# this is the zombie process
s = socket.socket(socket.AF_UNIX)
with contextlib.closing(s):
s.connect('%s')
if sys.version_info < (3, ):
pid = str(os.getpid())
else:
pid = bytes(str(os.getpid()), 'ascii')
s.sendall(pid)
Ni c s � j � tjkS )N)�statusr� �
STATUS_ZOMBIEr� )�zombier� r� �<lambda>� s z spawn_zombie.<locals>.<lambda>)r� r r� rW r� r� ra �
settimeoutr rA �accept�select�filenory �recvr� r� r� rZ �closerT )Z unix_file�srcr� �sock�parent�conn�_Zzpidr� )r� r� rE � s,
c K s� |j dd� |j dd� t� }yDt|d��}|j| � W dQ R X tt|jgf|�}t|j� ||fS t k
r� t
|� � Y nX dS )z�Run python 'src' code string in a separate interpreter.
Returns a subprocess.Popen instance and the test file where the source
code was written.
r� Nr� �w) r� rW �open�writerD r# r� r[ r� r� rT )r� r� �srcfile�fr� r� r� r� rA � s
c K s� t rdnd}|jdtj� |jdtj� |jdd� |jd|� t| t�rTtj| �} tj| f|�}t j
|� t r�|j t
d�\}}n |j � \}}|jdkr�t|| ��|r�t|� |jd �r�|d
d � }|S )
zURun cmd in a subprocess and return its output.
raises RuntimeError on error.
i r r� r� rt Tr� )�timeout�
N� ���)r
r� r{ r� �
isinstance�str�shlexrz r� r� r� r � communicater �
returncode� RuntimeErrorrg �endswith)r� r� �flags�pr� r� r� r� r� �sh s&
r c
s� dd� �dd� � � ��fdd�}� ��fdd���fd d
�}d d � }| }zVt |t�r^|||�S t |tjtjf�rz�||�S t |tj�r�|||�S td
| ��W dt |tjtjf�r�||� t |t�r�|n|j}tj|� s�t |��X dS )a� Terminate a process and wait() for it.
Process can be a PID or an instance of psutil.Process(),
subprocess.Popen() or psutil.Popen().
If it's a subprocess.Popen() or psutil.Popen() instance also closes
its stdin / stdout / stderr fds.
PID is wait()ed even if the process is already gone (kills zombies).
Does nothing if the process does not exist.
Return process exit status.
c S sd t | tj�r t r | j� n
| j|� tr`t | tj�r`ytj| j�j|�S tj k
r^ Y nX d S )N)
r
r{ r� r r� r
r� r� r� �
NoSuchProcess)�procr r� r� r� r� ' s
zterminate.<locals>.waitc S s6 t rtrtj}tr(|tjkr(| jtj� | j|� d S )N)r rk �signal�SIGKILLr � send_signal�SIGCONT)r �sigr� r� r� �sendsig3 s
zterminate.<locals>.sendsigc sZ y� | �� W n@ t k
rN } z$tr0|jdkr0n|jtjkr>� W Y d d }~X nX �| |�S )N� )�OSErrorr
�winerror�errnoZESRCH)r r �err)r r r� r� r� �term_subprocess_proc= s z'terminate.<locals>.term_subprocess_procc s0 y� | �� W n t jk
r$ Y nX �| |�S )N)r� r )r r )r r r� r� r� �term_psutil_procG s
z#terminate.<locals>.term_psutil_procc sB yt j| �}W n$ t jk
r2 tr.t| |�S Y n X � ||�S d S )N)r� r� r r r )r� r r )r" r� r� �term_pidN s zterminate.<locals>.term_pidc S s4 | j r| j j� | jr | jj� | jr0| jj� d S )N)r� r� r� r� )r r� r� r� � flush_popenX s
z terminate.<locals>.flush_popenz
wrong type %rN)
r
ry r� r� r� r{ � TypeErrorr� �
pid_existsr� )Z proc_or_pidr � wait_timeoutr! r# r$ r r� r� )r r r" r� r� rB s&
c C s� t j� j| d�}xtr(tj� }t|� qW xtrBtj� }t|� q,W |r�x|D ]}t|dd� qNW t j|td�\}}x&|D ] }t d| � t|t
j d� qzW dS )a Terminate and wait() any subprocess started by this test suite
and any children currently running, ensuring that no processes stick
around to hog resources.
If recursive is True it also tries to terminate and wait()
all grandchildren started by this process.
)� recursiveN)r' )r z0couldn't terminate process %r; attempting kill())r ) r� r� �childrenr� �poprB r� �
wait_procsr rg r r )r( r) r� r� r r �aliver� r� r� rC r s
c C s� t s td��d} tj� d }x(|D ] }|j� s6|dkr@| |7 } q"P q"W | sVtd| ��d}d}| jd�}t|d �}t|�dkr�t|d �}t|�dkr�t|d �}|||fS ) z"Return a tuple such as (2, 6, 36).z not POSIX� rm ro zcan't parse %rr r r� ) r �NotImplementedErrorr� �uname�isdigitr� rz ry �len)r� r/ �c�minor�micro�nums�majorr� r� r� rY � s&
c C sb t s td��tj� } t| d�r*| jp&d}n&tjd| d �}|rLt|j d��nd}| d | d |fS )Nz not WINDOWS�service_pack_majorr z\s\d$r� r )
r
r. r} �getwindowsversion�hasattrr7 �re�searchry �group)Zwv�sp�rr� r� r� rX � s
c @ s< e Zd ZdZeddddfdd�Zdd� Zdd � Zd
d � ZdS ) �retryzA retry decorator.Ng����MbP?c C s2 |r|rt d��|| _|| _|| _|| _|| _d S )Nz/timeout and retries args are mutually exclusive)r� � exceptionr �retries�interval�logfun)r� r@ r rA rB rC r� r� r� r� � s zretry.__init__c c s` | j r.tj� | j }xFtj� |k r*d V qW n.| jrPx&t| j�D ]
}d V q@W n x
d V qRW d S )N)r r� rA r )r� �stop_atr r� r� r� �__iter__� s zretry.__iter__c C s | j d k rtj| j � d S )N)rB r� r� )r� r� r� r� r� � s
z retry.sleepc s" t j� �� �fdd��}�|_|S )Nc st d }x^�D ]V}y
� | |�S �j k
r^ } z(|}�jd k rD�j|� �j� w
W Y d d }~X q
X q
W trn|�n� d S )N)r@ rC r� r )r� r� �excr )r� r� r� r� r� � s
z retry.__call__.<locals>.wrapper)r� r� � decorator)r� r� r� r� )r� r� r� �__call__� s zretry.__call__) r� r� r� r� r� r� rE r� rH r� r� r� r� r? � s
r? g����MbP?)r@ rC r rB c C s$ | t j� krt j| ��t j| � dS )z�Wait for pid to show up in the process list then return.
Used in the test suite to give time the sub process to initialize.
N)r� �pidsr r� )r� r� r� r� r[ � s
Tc C s: t | d��}|j� }W dQ R X |s*|s*t�|r6t| � |S )z8Wait for a file to be written on disk with some content.�rbN)r �readr� rT )�fnamer� r� r �datar� r� r� r\
s c C s | � }|st �|S )z1Keep calling function until it evaluates to True.)r� )r� �retr� r� r� rZ s c C sn dd� }yLt j| �}tj|j�r0tjtj| �}ntjt j| �}t rJ|� n||� W n t
k
rh Y nX dS )z?Convenience function for removing temporary test files or dirs.c S s� t j � t }xjt j � |k rvy| � S tk
r4 Y n6 tk
rh } z|}tdt|� � W Y d d }~X nX t jd� qW |�d S )Nz ignoring %sg{�G�z�?)r� r r Z WindowsErrorrg r r� )r� rD r r r� r� r� � retry_fun0 s "z safe_rmpath.<locals>.retry_funN) r� �stat�S_ISDIR�st_moder� �partial�shutil�rmtree�remover r )r� rO �str� r� r� r� rT - s
c C s( yt j| � W n tk
r" Y nX dS )z.Convenience function for creating a directory.N)r� �mkdirr )r� r� r� r� �
safe_mkdirO s rY c c s. t j� }zt j| � dV W dt j|� X dS )z@Context manager which temporarily changes the current directory.N)r� r� rS )�dirname�curdirr� r� r� rS W s
c C sT t jj| � st| ��tjt| � tjt | � t
rPt j | �}t j | |j
t jB � | S )z6Create a Python executable file in the given location.)r� r� r� r� �atexit�registerrT rT �copyfiler# r rP �chmodrR �S_IEXEC)r� rW r� r� r� rU b s
c C s� t jj| � st| ��td�s(tjd��|dkr<tjd�}nt |t
�sNt|��t j t
| � ttdd�d��}|j|� W dQ R X ztjd|jd| g� W dt
|j� X | S ) z5Create a compiled C executable in the given location.�gcczgcc is not installedNz�
#include <unistd.h>
int main() {
pause();
return 1;
}
z.c)�suffixr z-o)r� r� r� r� r �pytest�skipr� r� r
r r\ r] rT r rW r r{ r� r� )r� �c_coder r� r� r� rV m s
r- c C sB x<t jt| |d�}tjj|�stjj|�}tjt |� |S qW dS )z�Return an absolute pathname of a file or dir that did not
exist at the time this call is made. Also schedule it for safe
deletion at interpreter exit. It's technically racy but probably
not really due to the time variant.
)�prefixrb r� N)
�tempfile�mktempr' r� r� r� r� r\ r] rT )rb r� r� r� r� r� r� rW � s c @ sT e Zd ZdZedd� �Zeddd��Zeddd��Zedd
d ��ZG d d
� d
�Z dS )rR z�A class that mimics some basic pytest APIs. This is meant for
when unit tests are run in production, where pytest may not be
installed. Still, the user can test psutil installation via:
$ python3 -m psutil.tests
c O s4 t j� jt�}t jdd�j|� tjdtdd� |S )z�Mimics pytest.main(). It has the same effect as running
`python3 -m unittest -v` from the project root directory.
rm )� verbosityz<Fake pytest module was used. Test results may be inaccurate.r )�
stacklevel) rH Z
TestLoaderZdiscover�HEREZTextTestRunnerr� �warningsrg � UserWarning)r� �kw�suiter� r� r� �main� s zfake_pytest.mainNc s. G dd� d�� t jd� fdd� �}|| |d�S )zMimics `pytest.raises`.c @ s e Zd ZdZedd� �ZdS )z)fake_pytest.raises.<locals>.ExceptionInfoNc S s | j S )N)�_exc)r� r� r� r� �value� s z/fake_pytest.raises.<locals>.ExceptionInfo.value)r� r� r� rq �propertyrr r� r� r� r� �
ExceptionInfo� s rt Nc 3 sx � � }y
|V W nV | k
rf } z:|rPt j|t|�� rPdj|t|��}t|��||_W Y d d }~X nX td| ��d S )Nz"{}" does not match "{}"z
%r not raised)r: r; r �formatr� rq )rF �matchZeinfor �msg)rt r� r� �context� s
z#fake_pytest.raises.<locals>.context)rv )N)�
contextlib�contextmanager)rF rv rx r� )rt r� �raises� s zfake_pytest.raisesc C s" |rt j� j| |�S t j� j| �S )zMimics `pytest.warns`.)rH �TestCaseZassertWarnsRegexZ assertWarns)�warningrv r� r� r� �warns� s zfake_pytest.warnsr- c C s t j| ��dS )zMimics `unittest.SkipTest`.N)rH ZSkipTest)�reasonr� r� r� rd � s zfake_pytest.skipc @ s( e Zd Zeddd��ZG dd� d�ZdS )zfake_pytest.markr- c C s t j| |�S )z'Mimics `@pytest.mark.skipif` decorator.)rH ZskipIf)Z conditionr r� r� r� �skipif� s zfake_pytest.mark.skipifc @ s" e Zd ZdZddd�Zdd� ZdS )z fake_pytest.mark.xdist_groupz4Mimics `@pytest.mark.xdist_group` decorator (no-op).Nc C s d S )Nr� )r� r� r� r� r� r� � s z%fake_pytest.mark.xdist_group.__init__c C s |S )Nr� )r� Z cls_or_methr� r� r� rH � s z%fake_pytest.mark.xdist_group.__call__)N)r� r� r� r� r� rH r� r� r� r� � xdist_group� s
r� N)r- )r� r� r� � staticmethodr� r� r� r� r� r� �mark� s r� )N)N)r- )
r� r� r� r� r� rp r{ r~ rd r� r� r� r� r� rR � s c @ s& e Zd Zes"dd� Zejdd� �ZdS )r| c C s d S )Nr� )r� r� r� r� �runTest� s zTestCase.runTestc o s
d V d S )Nr� )r� r� rn r� r� r� �subTest� s zTestCase.subTestN)r� r� r� r r� ry rz r� r� r� r� r� r| � s r| c @ sZ e Zd ZdZddd�Zdd� Zdd � Zd
d � Zd d
� Zdd� Z dd� Z
dd� Z dd� Z dS )rM z�Test class providing auto-cleanup wrappers on top of process
test utilities. All test classes should derive from this one, even
if we use pytest.
r- Nc C s t ||d�}| jt|� |S )N)rb r� )rW �
addCleanuprT )r� rb r� rL r� r� r� rW s zPsutilTestCase.get_testfnc O s t ||�}| jt|� |S )N)rD r� rB )r� r� r� r� r� r� r� rD s
z PsutilTestCase.spawn_testprocc C s* t � \}}| jt|� | jt|� ||fS )N)rF r� rB )r� Zchild1Zchild2r� r� r� rF s
z"PsutilTestCase.spawn_children_pairc C s* t � \}}| jt|� | jt|� ||fS )N)rE r� rB )r� r� r� r� r� r� rE s
zPsutilTestCase.spawn_zombiec O s* t ||�\}}| jt|� | jt|� |S )N)rA r� rT rB )r� r� r� r� r r� r� r� rA s zPsutilTestCase.pyrunc C s� t |tj�st�|j|jks t�|j|jks0t�|jr@|js@t�t |tj�rt|j|j ks\t�|jd k rt|jdkstt�t
|� t |� d S )Nr ) r
r� �Errorr� r� r� �_name�
ZombieProcess�ppidZ_ppidr �repr)r� r rF r� r� r� �_check_proc_exc s
z PsutilTestCase._check_proc_excc C s� t jtj��4}ytj|� W n tjk
r: td��Y nX W d Q R X |jj|ksVt�|jj d ksft�tj
|� szt|��|tj � ks�t�|dd� tj � D �ks�t�d S )Nz&wasn't supposed to raise ZombieProcessc S s g | ]
}|j �qS r� )r� )r� r� r� r� r� r� 5 s z0PsutilTestCase.assertPidGone.<locals>.<listcomp>)
rc r{ r� r r� r� r� rr r� r� r& rI � process_iter)r� r� �cmr� r� r� �
assertPidGone+ s z PsutilTestCase.assertPidGonec C s� | j |j� t|�}x�|j|jdd�D ]�\}}| j||d��l y
|� }W nF tjk
r` � Y nD tjk
r� } z| j ||� W Y d d }~X nX d||f }t
|��W d Q R X q&W |j dd� d S )NT)� clear_cache)r r� z-Process.%s() didn't raise NSP and returned %rr )r ) r� r� rN �iter�allr� r� r� r r� r� r� )r� r �nsr� r� rN rF rw r� r� r� �assertProcessGone7 s
z PsutilTestCase.assertProcessGonec C sh t j|j�}||kst�tp ts4t|�t|�ks4t�|j� t jksFt�|j � sRt�t j
|j�sbt�|j � |jt j � ks|t�|jdd� t j
� D �ks�t�i t _|jdd� t j
� D �ks�t�t|�}xz|j|jdd�D ]f\}}| j||d��J y
|� W n8 t jt jfk
�r. } z| j||� W Y d d }~X nX W d Q R X q�W t�r�tjt j��}|j� W d Q R X | j||j� tjt j��}|j� W d Q R X | j||j� tjt j��}|j � W d Q R X | j||j� |j � |j � |j � |j � |j � �s�t�t j
|j��st�|jt j � k�s"t�|jdd� t j
� D �k�s@t�i t _|jdd� t j
� D �k�sdt�d S ) Nc S s g | ]
}|j �qS r� )r� )r� r� r� r� r� r� [ s z6PsutilTestCase.assertProcessZombie.<locals>.<listcomp>c S s g | ]
}|j �qS r� )r� )r� r� r� r� r� r� ] s T)r� )r r� c S s g | ]
}|j �qS r� )r� )r� r� r� r� r� r� y s c S s g | ]
}|j �qS r� )r� )r� r� r� r� r� r� { s )!r� r� r� r� r
r �hashr� r� �
is_runningr& �as_dictrI r� Z_pmaprN r� r� r� r� � AccessDeniedr� r rc r{ �cmdlinerr r� r� �suspend�resumerB �kill)r� r �cloner� r� r� rF r� r� r� r� �assertProcessZombieJ sN
, z"PsutilTestCase.assertProcessZombie)r- N)
r� r� r� r� rW rD rF rE rA r� r� r� r� r� r� r� r� rM � s
zunreliable on PYPY)r c @ s� e Zd ZdZdZdZdZer dndZdZ e
j � Z e
ejd��Zedd � �Zed
d � �Zd d
� Zdd� Zdd� Zdd� Zdd� Zdd� Zdd� Zd dd �Zd d � Z dS ) rL a� Test framework class for detecting function memory leaks,
typically functions implemented in C which forgot to free() memory
from the heap. It does so by checking whether the process memory
usage increased before and after calling the function many times.
Note that this is hard (probably impossible) to do reliably, due
to how the OS handles memory, the GC and so on (memory can even
decrease!). In order to avoid false positives, in case of failure
(mem > 0) we retry the test for up to 5 times, increasing call
repetitions each time. If the memory keeps increasing then it's a
failure.
If available (Linux, OSX, Windows), USS memory is used for comparison,
since it's supposed to be more precise, see:
https://gmpy.dev/blog/2016/real-process-memory-and-environ-in-python
If not, RSS memory is used. mallinfo() on Linux and _heapwalk() on
Windows may give even more precision, but at the moment are not
implemented.
PyPy appears to be completely unstable for this framework, probably
because of its JIT, so tests on PYPY are skipped.
Usage:
class TestLeaks(psutil.tests.TestMemoryLeak):
def test_fun(self):
self.execute(some_function)
�� rp r r� T� PSUTIL_DEBUGc C s t jd� d S )NF)r� �
_set_debug)�clsr� r� r� �
setUpClass� s zTestMemoryLeak.setUpClassc C s t j| j� d S )N)r� r� �_psutil_debug_orig)r� r� r� r� �
tearDownClass� s z TestMemoryLeak.tearDownClassc C s | j j� }t|d|j�S )NZuss)� _thisproc�memory_full_infor� Zrss)r� �memr� r� r� �_get_mem� s
zTestMemoryLeak._get_memc C s t r| jj� S | jj� S d S )N)r r� �num_fds� num_handles)r� r� r� r� � _get_num_fds� s
zTestMemoryLeak._get_num_fdsc C s | j rt|dtjd� d S )N�yellow)�color�file)�verboser r} r� )r� rw r� r� r� �_log� s zTestMemoryLeak._logc C sx | j � }| j|� | j � }|| }|dk r8| jd| ��|dkrttrHdnd}|dkr\|d7 }d|||f }| j|��dS ) z�Makes sure num_fds() (POSIX) or num_handles() (Windows) does
not increase after calling a function. Used to discover forgotten
close(2) and CloseHandle syscalls.
r zHnegative diff %r (gc probably collected a resource from a previous test)�fd�handler r� z %s unclosed %s after calling %rN)r� �call�failr )r� r� �before�after�diff�type_rw r� r� r� �
_check_fds� s
zTestMemoryLeak._check_fdsc C sb t jdd� | j� }x t|�D ]}| j|�}~~q W t jdd� | j� }t jg ksVt�|| }|S )z�Get 2 distinct memory samples, before and after having
called fun repeatedly, and return the memory difference.
r )Z
generation)�gcZcollectr� r r� Zgarbager� )r� r� �timesZmem1r� rN Zmem2r� r� r� r� � _call_ntimes� s
zTestMemoryLeak._call_ntimesc C s� g }d}|}x�t d|d �D ]�}| j||�} d|t| �t| | �|f }
|j|
� | |kp`| |k} | r||dkrx| j|
� d S |dkr�t� | j|
� ||7 }| }q W | jdj|���d S )Nr r z,Run #%s: extra-mem=%s, per-call=%s, calls=%sz. )r r� r �appendr� �printr� r� ) r� r� r� rA � tolerance�messagesZprev_memZincrease�idxr� rw �successr� r� r� �
_check_mem� s,
zTestMemoryLeak._check_memc C s |� S )Nr� )r� r� r� r� r� r�
s zTestMemoryLeak.callNc C s� |dk r |n| j }|dk r |n| j}|dk r0|n| j}|dk rB|n| j}yD|dksZtd��|dksjtd��|dksztd��|dks�td��W n. tk
r� } ztt|���W Y dd}~X nX | j||� | j|� | j ||||d� dS ) zTest a callable.Nr ztimes must be >= 1r zwarmup_times must be >= 0zretries must be >= 0ztolerance must be >= 0)r� rA r� )
r� � warmup_timesrA r� r� r� r r� r� r� )r� r� r� r� rA r� r r� r� r� �execute s
zTestMemoryLeak.executec s" � ��fdd�}�j |f|� dS )znConvenience method to test a callable while making sure it
raises an exception on every call.
c s �j � �� d S )N)Z assertRaisesr� )rF r� r� r� r� r� + s z*TestMemoryLeak.execute_w_exc.<locals>.callN)r� )r� rF r� r� r� r� )rF r� r� r� �
execute_w_exc& s z TestMemoryLeak.execute_w_exc)NNNN) r� r� r� r� r� r� r� r* rA r� r� r� r� �boolr� �getenvr� � classmethodr� r� r� r� r� r� r� r� r� r� r� r� r� r� r� rL � s&
c C s� dd l } dd l}dd l}dd l}dd l}y dd l}W n tk
rL d }Y nX y dd l}W n tk
rr d }Y nX | j� }t j
r�t d�r�t d�|d< nzt j
r�dtj� d |d< n^t jr�ddjtttj� �� |d< ttd�r�|d d tj� 7 < nd
tj� tj� f |d< d jttj� �tj� g �|d < t j �rJtj � d |d
< d jtj � tj � tj � g�|d< t!|dd�|d< |d k �r�|d d|j" 7 < t j �r�t d��r�t ddg�}t|�j#d�d |d< nd|d< tj$� d } | �r�| |d< t%j&� |d< |j'� }
d|
d |
d f |d< |jj(t j)� �j*d�|d < |jj+� j*d�|d < |j,� |d < t-j.j/d �|d < t-j0� |d!< t1|d"< tj2� |d#< t-j3� |d$< t j4� |d%< d&t5d'd(� t j6� D �� |d)< t j7� } d*t8| j9�t:| j;�t:| j<�f |d+< t j=� } d*t8| j9�t:| j;�t:| j<�f |d,< t>t j?� �|d-< t j@� jA� }
|
jBd.d � |jC|
�|d/< tDd0d1 t%jEd2� x0|jF� D ]$\}}tDd3|d4 |f t%jEd2� �qhW tDd0d1 t%jEd2� t%jGjH� t%jGjH� d S )5Nr Z lsb_releasezlsb_release -d -sZOSz Darwin %szWindows � �
win32_editionz, z%s %s�archrm �kernel�python� __version__z
not installed�pipz (wheel=%s)ra z --versionr r �glibcz fs-encodingz%s, %s�langz%Y-%m-%d %H:%M:%Sz boot-timer� �user�~�homer� Zpyexe�hostnameZPIDZcpusz%.1f%%, %.1f%%, %.1f%%c S s g | ]}|t j� d �qS )�d )r� � cpu_count)r� r� r� r� r� r� { s z!print_sysinfo.<locals>.<listcomp>Zloadavgz%s%%, used=%s, total=%sZmemory�swaprI r� r �=�F )r� z%-17s %s�:)I� collections�datetime�getpass�locale�pprintr� � ImportError�wheel� OrderedDictr� r r r �OSXru rv r
r� rx r � win32_verr9 r� �systemr �list� architecture�machiner r/ �python_implementation�python_version�python_compilerr� r� rz �libc_verr} �getfilesystemencoding� getlocale�
fromtimestamp� boot_time�strftime�nowZgetuserr� r� �
expanduserr� r# �node�getpidr� rw r� �virtual_memoryry �percentr �used�total� swap_memoryr1 rI r� r� r* �pformatr� r� �itemsr� �flush)r� r� r� r� r� r� r� �info�outr� r� r� r� Zpinfo�k�vr� r� r� rP 1 s�
c C s4 t dd� �}y|� | dkS tk
r. dS X d S )Nc S s\ i } t d�}xJ|j� dd � D ]6}dd� |jd�D �}|d t|d � }}|| |< q W | S )Nztasklist.exe /NH /FO csvr c S s g | ]}|j d d��qS )�"r- )�replace)r� r� r� r� r� r� � s z@is_win_secure_system_proc.<locals>.get_procs.<locals>.<listcomp>�,r )r �
splitlinesrz ry )rN r �line�bitsr� r� r� r� r� � get_procs� s z,is_win_secure_system_proc.<locals>.get_procsz
Secure SystemF)r �KeyError)r� r r� r� r� rQ � s
c C s6 t j� } t| d�r| j� S t| d�r2tj| j� �S dS )Nr� r� r )r� r� r9 r� �random�choicer� )r r� r� r� �_get_eligible_cpu� s
r c @ sJ e Zd ZdZdf i fdf i fgZdf i fdf ddifdf i fd f i fd
f i fd f i fd f i fd
f i fdf i fddDi fg
Zdf i fdf i fdf i fdf i fdf i fdf i fdf i fdf i fdf ddifd f i fd f i fd f i fd f i fd f i fd!f i fd"f i fd#f i fgZe�rNed$f i fg7 Zed%f i fg7 Zed&f i fg7 Zed'f i fg7 Ze�rded(f i fg7 Ze �rzed)f i fg7 Ze
�r�ed*e j fi fg7 Ze
�r�ed+f i fg7 Ze�r�ed,f i fg7 Ze�r�ed-f i fg7 Ze�r�ed.f i fg7 Ze�red/f d0d1ifg7 Zg Ze�r"ed dEi fg7 Zned e jfi fg7 Ze
�rRed*e j dFfi fg7 Ze �r�e�rved)e jdfi fg7 Zned)e jfi fg7 Ze
�r�ed+e� gfi fg7 Zd4ejfi fd5f i fd6f i fd7f i fd8f i fgZe�red4ejfi fg7 Zed4ej fi fg7 Zee e e Z d9d:� Z dGd;d<�Z d=d>� Z e!d?d@� �Z"e!dAdB� �Z#dCS )HrN a A container that lists all Process class method names + some
reasonable parameters to be called with. Utility methods (parent(),
children(), ...) are excluded.
>>> ns = process_namespace(psutil.Process())
>>> for fun, name in ns.iter(ns.getters):
... fun()
� cpu_percentZmemory_percentr� r) r( TZ connectionsr� Zmemory_info_exZoneshotr� �parentsr� r� r r� � cpu_timesZ create_timer� r� r� Z memory_infor� �net_connections�kindr� �niceZnum_ctx_switchesZ num_threadsZ
open_filesr� r� r� �usernameZuidsZgids�terminalr� r� r� r� r� r� r� r� r� ZgroupedF� � r r� r� rB r� c C s
|| _ d S )N)�_proc)r� r r� r� r� r� s zprocess_namespace.__init__c c s^ t |�}tj|� xF|D ]>\}}}|r.| j� t| j|�}tj|f|�|�}||fV qW dS )z_Given a list of tuples yields a set of (fun, fun_name) tuples
in random order.
N)r� r
�shuffler� r� r r� rS )r� �lsr� �fun_namer� r� r� r� r� r� r� s
zprocess_namespace.iterc C s | j j| j jdd� dS )z&Clear the cache of a Process instance.T)Z _ignore_nspN)r �_initr� )r� r� r� r� r� & s z process_namespace.clear_cachec C sB x<|D ]4\}}}d| }t ||�sd|jj|f }t|��qW dS )z}Given a TestCase instance and a list of tuples checks that
the class defines the required test method names.
Ztest_z$%r class should define a '%s' methodN)r9 r� r� �AttributeError)r� Z
test_classr r r Z meth_namerw r� r� r� �test_class_coverage* s
z%process_namespace.test_class_coveragec C s` t dd� | jD ��}t dd� | jD ��}t dd� ttj�D ��}||B |A }|r\td| ��d S )Nc S s g | ] }|d �qS )r r� )r� r� r� r� r� r� : s z*process_namespace.test.<locals>.<listcomp>c S s g | ] }|d �qS )r r� )r� r� r� r� r� r� ; s c S s g | ]}|d dkr|�qS )r r r� )r� r� r� r� r� r� < s z!uncovered Process class names: %r)r� r� �ignoredr� r� r� r� )r� �thisr! �klassZleftoutr� r� r� �test8 s zprocess_namespace.testN)r )r )r r )T)$r� r� r� r� �utilsr! �gettersr r1 r2 r5 r� Z
RLIMIT_NOFILEr. r4 r0 r
r3 ZsettersZNORMAL_PRIORITY_CLASSr ZIOPRIO_CLASS_NONEZ
IOPRIO_NORMALr r �SIGTERMZkillersZ CTRL_C_EVENTZCTRL_BREAK_EVENTr� r� r� r� r� r r$ r� r� r� r� rN � s�
c @ s� e Zd ZdZdf i fdf ddifdf ddifdf i fdf d difdf d difd
f d difd f d
difdej� fi fdf dd
ifdf i fdf i fdf ddifdej� fi fdf i fdf i fdf i fdf i fgZer�e r�e
j � dkr�nedf d difg7 Ze �r ed f i fg7 Ze
�r"ed f i fg7 Ze�r8ed f i fg7 Ze�rNed f i fg7 Ze�rted f i fg7 Zed!d*i fg7 Zd#f i fd$ej� gfi fd%f i fd&f i fgZeZed'd(� �ZejZd)S )+rO z�A container that lists all the module-level, system-related APIs.
Utilities such as cpu_percent() are excluded. Usage:
>>> ns = system_namespace
>>> for fun, name in ns.iter(ns.getters):
... fun()
r� r� ZlogicalFTZ cpu_statsr ZpercpuZdisk_io_countersZperdiskZdisk_partitionsr� �
disk_usager r Z net_if_addrsZ net_if_statsr� Zpernicr& rI r� Zusersr� Zarm64r� r� r� r� r� Zwin_service_iterZwin_service_get�algr� r+ r Zcpu_times_percentc c sP t | �} tj| � x8| D ]0\}}}tt|�}tj|f|�|�}||fV qW dS )z_Given a list of tuples yields a set of (fun, fun_name) tuples
in random order.
N)r� r
r r� r� r� rS )r r r� r� r� r� r� r� r� y s
zsystem_namespace.iterN)r) )r� r� r� r� r� r� r� r& r/ r ru r� �HAS_GETLOADAVGr9 r8 r6 r
r� r� r! r� r� r� rN r r� r� r� r� rO B sR c C s dd� }t td| |d�S )zZDecorator which runs a test function and retries N times before
actually failing.
c S s t d| tjd� d S )Nz %r, retrying)r� )r� r} r� )rF r� r� r� rC � s z retry_on_failure.<locals>.logfunN)r@ r rA rC )r? r� )rA rC r� r� r� rK � s c s � fdd�}|S )z,Decorator to Ignore AccessDenied exceptions.c s t j� �� �fdd��}|S )Nc s> y
� | |�S t jk
r8 �d k r*�s*� tjd��Y nX d S )Nzraises AccessDenied)r� r� rc rd )r� r� )r� �only_ifr� r� r� � s
z9skip_on_access_denied.<locals>.decorator.<locals>.wrapper)r� r� )r� r� )r+ )r� r� rG � s
z(skip_on_access_denied.<locals>.decoratorr� )r+ rG r� )r+ r� rI � s
c s � fdd�}|S )z3Decorator to Ignore NotImplementedError exceptions.c s t j� �� �fdd��}|S )Nc sF y
� | |�S t k
r@ �d k r(�s(� d� j }tj|��Y nX d S )Nz4%r was skipped because it raised NotImplementedError)r. r� rc rd )r� r� rw )r� r+ r� r� r� � s
z;skip_on_not_implemented.<locals>.decorator.<locals>.wrapper)r� r� )r� r� )r+ )r� r� rG � s z*skip_on_not_implemented.<locals>.decoratorr� )r+ rG r� )r+ r� rJ � s � 127.0.0.1c
C s4 t jtj� �� }|j| df� |j� d S Q R X dS )z6Return an unused TCP port. Subject to race conditions.r r N)ry �closing�socket�bind� getsockname)�hostr� r� r� r� r_ � s c C s� |dkr| t tfkrd}tj| |�}y>tjd krB|jtjtjd� |j|� |tj kr`|j
d� |S t k
r� |j � � Y nX dS )
zBinds a generic socket.Nr- r �nt�cygwinr r� )r- r )r2 r3 )
r r r. r� r� �
setsockopt�
SOL_SOCKET� SO_REUSEADDRr/ r �listenr� r� )�family�type�addrr� r� r� r� r` � s
c C st t js
t�tjj| � s t| ��tjtj|�}y"|j| � |tj krN|j
d� W n t k
rn |j � � Y nX |S )zBind a UNIX socket.r� )
r� r r� r� r� r� r. r� r/ r r7 r� r� )r� r9 r� r� r� r� ra � s
c C s� t jtj| t����}|j|� |jd� |j� }tj| t�}y@|j|� |j� }x(|j� \}}||krn||fS |j � qRW W n t
k
r� |j � � Y nX W dQ R X dS )z^Build a pair of TCP sockets connected to each other.
Return a (server, client) tuple.
r� N) ry r- r. r r/ r7 r0 �connectr� r� r )r8 r: Zllr2 Zcaddr�ar� r� r� rb � s
c C s� t js
t�d }}y@t| tjd�}|jd� tjtjtj�}|jd� |j| � W n6 t k
r� |dk rr|j
� |dk r�|j
� � Y nX ||fS )z�Build a pair of UNIX sockets connected to each other through
the same UNIX file name.
Return a (server, client) tuple.
N)r9 r ) r� r r� ra r. r � setblockingr� r; r� r� )r� �serverZclientr� r� r� rc s
c c s� g } d }}z�| j ttjtj�� | j ttjtj�� t� rd| j ttjtj�� | j ttjtj�� tr�t r�t
� }t
� }t |�\}}t |tjd�}x |||fD ]}| j |� q�W | V W dx| D ] }|j
� q�W x ||fD ]}|dk r�t|� q�W X dS )z1Open as many socket families / types as possible.N)r9 )r� r` r. r r �
SOCK_DGRAMr r r r: rW rc ra r� rT )ZsocksZfname1Zfname2�s1�s2Zs3r� rL r� r� r� rd s*
c C s
ddl }tr*tr*t r*t|tj�s*t|��|tjkr�dd� | j d�D �}t
|�dks\t| ��x,|D ]$}d| koxdkn sbt| ��qbW ts�t | �} |j | � nd|tj
kr�t| t�s�t| ��ts�t | �} |j| � n0|tjkr�tjd| �dk s�t| ��n td | ��dS )
z[Check a net address validity. Supported families are IPv4,
IPv6 and MAC addresses.
r Nc S s g | ] }t |��qS r� )ry )r� r� r� r� r� r� > s z%check_net_address.<locals>.<listcomp>ro r� � z([a-fA-F0-9]{2}[:|\-]?){6}zunknown family %r)� ipaddress�enumr r" r
�IntEnumr� r. r rz r1 r � IPv4Addressr r � IPv6Addressr� ZAF_LINKr: rv r� )r: r8 rC Zocts�numr� r� r� r] 5 s&
$
c C sT dd� }dd� }dd� }dd� }d d
� }|| � || � || � || � || � d S ) z*Check validity of a connection namedtuple.c S s� t | �dk}t | �d ks$tt | ���| d | jks<t| j��| d | jksTt| j��| d | jkslt| j��| d | jks�t| j��| d | jks�t| j��| d | jks�t| j��|r�| d | jks�t| j��d S )
Nr� r r r rm r� r� r� )r r� ) r1 r� r� r8 r9 �laddr�raddrr� r� )r� Zhas_pidr� r� r� � check_ntupleS s z-check_connection_ntuple.<locals>.check_ntuplec S s� | j tttfkst| j ��td k r:t| j tj�sNt| ��nt| j t�sNt| ��| j tkr�t j | j | j
�}t j |��R y|j
| jd df� W n2 t jk
r� } z|jtjkr�� W Y d d }~X nX W d Q R X n | j tkr�| jtjks�t| j��d S )Nr )r8 r r r� r� rD r
rE ry r. r9 ry r- r/ rI �errorr Z
EADDRNOTAVAILr� r� � CONN_NONE)r� r� r r� r� r� � check_family_ s
z-check_connection_ntuple.<locals>.check_familyc S s� t tdt� �}| jtjtj|fks,t| j��td k rLt| jtj �s`t| ��nt| jt
�s`t| ��| jtjkr�| j t j
ks�t| j ��d S )N�SOCK_SEQPACKET)r� r. �objectr9 r r? r� rD r
rE ry r� r� rM )r� rO r� r� r� �
check_typet s
z+check_connection_ntuple.<locals>.check_typec S s� x�| j | jfD ]�}| jttfkr�t|t�s6tt|���|s<qt|j t
�sVtt|j ���d|j kojdkn szt|j ��t |j | j� q| jt
krt|t�stt|���qW d S )Nr i�� )rI rJ r8 r r r
rw r� r9 �portry r] �ipr� r )r� r: r� r� r� � check_addrs� s $
z,check_connection_ntuple.<locals>.check_addrsc S s� t | jt�st| j��dd� tt�D �}| j|ks<t| j��| jttfkrl| j t
krl| jtj ks�t| j��n| jtj ks�t| j��d S )Nc S s g | ]}|j d �rtt|��qS )ZCONN_)r� r� r� )r� r� r� r� r� r� � s zAcheck_connection_ntuple.<locals>.check_status.<locals>.<listcomp>) r
r� r r� r� r� r8 r r r9 r rM )r� Zvalidsr� r� r� � check_status� s z-check_connection_ntuple.<locals>.check_statusNr� )r� rK rN rQ rT rU r� r� r� �check_connection_ntupleP s
rV c C sP g }xF| D ]>}t r>|jtjkr>tr>d|jkr>tdt|� � q
|j|� q
W |S )ztOur process may start with some open UNIX sockets which are not
initialized by us, invalidating unit tests.
z/syslogz skipping %s) r r8 r. r� r rJ r r r� )Zcons�newr� r� r� r� r^ � s
c C sL yddl }t|d�st�W n" tk
r< ddl}|j| �S X |j| �S dS )z,Backport of importlib.reload of Python 3.3+.r N�reload)� importlibr9 r� �imprX )�modulerY rZ r� r� r� re � s
c C sp t jjt jj| ��d }tjd dk r:dd l}|j|| �S dd l}|j j
|| �}|j j |�}|j j
|� |S d S )Nr r� )r� r� �splitextr� r} r� rZ � load_source�importlib.util�util�spec_from_file_location�module_from_spec�loader� exec_module)r� r� rZ rY �spec�modr� r� r� rf � s c C s t j| tdd� dS )zRaise a warning msg.rm )rj N)rl rg rm )rw r� r� r� rg � s c C sV t | �}|j}t|�dks&|d tk r*dS t|dd�}t|t�sDdS tdd� |D ��S )z-Check if object is an instance of namedtuple.r r F�_fieldsNc s s | ]}t |t�V qd S )N)r
r )r� �nr� r� r� � <genexpr>� s z is_namedtuple.<locals>.<genexpr>)r9 � __bases__r1 rw r� r
r� )r� �t�br r� r� r� ri � s
c # st t rdnd� d�t| � d�}� �fdd�tj� j� D �}tj|�}tj||� zt j
|� |V W dt |� X dS )z�Ctx manager which picks up a random shared CO lib used
by this process, copies it in another location and loads it
in memory via ctypes. Return the new absolutized path.
�pypyr� z.so)rb c s6 g | ].}t jj|j�d �kr� |jj� kr|j�qS )r )r� r� r\ �lower)r� r� )r� �extr� r� r� � s z'copyload_shared_lib.<locals>.<listcomp>N) r" rW r� r� r� r
r rT r^ �ctypes�CDLLrT )rb �dst�libsr� r� )r� rn r� rh � s
c # s� ddl m} ddl m} d� t| � d�}� fdd�tj� j� D �}trd| rddd� tj� j� D �}tj |�}t
j ||� d }zt j |�}|V W d |d k r�t j
jj}|jg|_||j�}|dkr�|� t|� X d S )
z�Ctx manager which picks up a random shared DLL lib used
by this process, copies it in another location and loads it
in memory via ctypes.
Return the new absolutized, normcased path.
r )�WinError)�wintypesz.dll)rb c sF g | ]>}|j j� j� �rd tj j|j �j� krd|j j� kr|j �qS )r� Zwow64)r� rm r r� r� )r� r� )rn r� r� r� s z'copyload_shared_lib.<locals>.<listcomp>c S s( g | ] }d t jj|j�j� kr|j�qS )rl )r� r� r� rm )r� r� r� r� r� r� s N)ro rs rt rW r� r� r� r" r
r rT r^ ZWinDLL�windll�kernel32� FreeLibraryZHMODULE�argtypes�_handlerT ) rb rs rt rq rr r� �cfilerw rN r� )rn r� rh s,
c C s t dd� d S )NT)r( )rC r� r� r� r� �cleanup_test_procs7 s r{ c C s
t j| �S )N)r} �exit)r r r� r� r� r� A s r� l )rp r� )r� r i i P i ( i � )r� r� )N)F)TF)N)r- N)N)N)r, �r- r )r} )r- )r- )�r� �
__future__r r\ ry ro r r� r� r� ru r
r: r� r rT r r. rP r{ r} rg r� r� r� rH rl r r r rc r� r� r r r r r
r r r
Zpsutil._commonr r r r r Zpsutil._compatr r r r r r r r �catch_warnings� simplefilterrD � unittest2Zpsutil._psposixr �__all__�builtin_module_namesr" r� r rk r* r= r@ r r rK r? �maxsizer- r� r> r� r; r< r! r r, r r� r� r' r( �decoder) r� rm ZASCII_FSr� r� r� rZ �__file__r% �getr& rk r9 r� r. r/ r0 r* r2 r3 r: ZHAS_NET_IO_COUNTERSr4 r1 r5 r6 r� r� r7 r� r8 r9 Z HAS_THREADS�getuidZ SKIP_SYSCONSr� r# r$ �devnullr r] r� r� r+ r� rP r� r� r� r� r� rG r� rD rF rE rA r r' rB rC rY rX r? r r[ r� r\ rZ rT rY rz rS rU rV rW rR r| rM r� r� rL rP rQ r rN rO rK rI rJ r_ r` ra rb rc rd r] rV r^ re rf rg ri rh r{ r� r� r� r� �<module> s�
*
0 ) '- V
%; "
M
%l F
R 3 |