3
�[g3< � @ s d Z ddlmZ ddlZddlZddlZddlZddlZddlmZ ddlm Z ddlm
Z
ddlm Z ddlm Z dd lm
Z
dd
lmZ dd lmZ dd lmZ dd
lmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlm Z ddlm Z ddlm Z ddlm Z ddlm Z ddlm!Z! dd lm"Z" dd lm#Z# dd lm$Z$ dd lm%Z% dd lm&Z& dd!lm'Z' ej(j)Z)ej*� Z+d"Z,d#d$� Z-G d%d&� d&e �Z.G d'd(� d(e.�Z/e#j0j1e
d)d*�G d+d,� d,e ��Z2G d-d.� d.e �Z3dS )/a� Tests for detecting function memory leaks (typically the ones
implemented in C). It does so by calling a function many times and
checking whether process memory usage keeps increasing between
calls or over time.
Note that this may produce false positives (especially on Windows
for some reason).
PyPy appears to be completely unstable for this framework, probably
because of how its JIT handles memory, so tests are skipped.
� )�print_functionN)�LINUX)�MACOS)�OPENBSD)�POSIX)�SUNOS)�WINDOWS)�ProcessLookupError)�super)�HAS_CPU_AFFINITY)� HAS_CPU_FREQ)� HAS_ENVIRON)�
HAS_IONICE)�HAS_MEMORY_MAPS)�HAS_NET_IO_COUNTERS)�HAS_PROC_CPU_NUM)�HAS_PROC_IO_COUNTERS)�
HAS_RLIMIT)�HAS_SENSORS_BATTERY)�HAS_SENSORS_FANS)�HAS_SENSORS_TEMPERATURES)� QEMU_USER)�TestMemoryLeak)�create_sockets)�
get_testfn)�process_namespace)�pytest)�skip_on_access_denied)�spawn_testproc)�system_namespace)� terminate� c C s dd� } | S )zsDecorator for those Linux functions which are implemented in pure
Python, and which we want to run faster.
c s t j� �� fdd��}|S )Nc sF t r2| jj}zt| j_� | f|�|�S || j_X n� | f|�|�S d S )N)r � __class__�times� FEW_TIMES)�self�args�kwargs�before)�fun� �F/usr/local/lib64/python3.6/site-packages/psutil/tests/test_memleaks.py�wrapperD s z5fewtimes_if_linux.<locals>.decorator.<locals>.wrapper)� functools�wraps)r) r, r* )r) r+ � decoratorC s z$fewtimes_if_linux.<locals>.decoratorr* )r/ r* r* r+ �fewtimes_if_linux> s r0 c @ s0 e Zd ZdZeZdd� Ze� dd� �Ze� dd� �Z e� dd � �Z
e� d
d � �Z e j
je d d
�e� dd� ��Ze j
je d d
�e� dd� ��Ze� dd� �Zdd� Zdd� Ze j
je dd
�dd� �Ze j
je dd
�dd � �Ze j
je dd
�e� d d � ��Ze j
jed d
�d d!� �Ze� d"d#� �Ze� e e d$�d%d&� ��Z e j
je d'd
�d(d)� �Z e j
je d d
�e� d*d+� ��Z!e� d,d-� �Z"e� e e d$�d.d/� ��Z#e� d0d1� �Z$e� e j
je% dd
�d2d3� ��Z&e� d4d5� �Z'e� d6d7� �Z(e j
je d d
�e� d8d9� ��Z)d:d;� Z*e� d<d=� �Z+e j
je, dd
�d>d?� �Z-e j
je, dd
�d@dA� �Z.e� dBdC� �Z/e j
je0 dd
�e� dDdE� ��Z1e j
je2 dFd
�e j
je3 dd
�dGdH� ��Z4e j
je2 dFd
�e j
je3 dd
�dIdJ� ��Z5e� e j
je dKd
�dLdM� ��Z6e j
je7 dd
�dNdO� �Z8e j
je d'd
�dPdQ� �Z9dRS )S�TestProcessObjectLeaksz$Test leaks of Process class methods.c C s t d �}|j| |j|j � d S )N)r �test_class_coverageZgettersZsetters)r% �nsr* r* r+ �
test_coverage_ s z$TestProcessObjectLeaks.test_coveragec C s | j | jj� d S )N)�execute�proc�name)r% r* r* r+ � test_namec s z TestProcessObjectLeaks.test_namec C s | j | jj� d S )N)r5 r6 �cmdline)r% r* r* r+ � test_cmdlineg s z#TestProcessObjectLeaks.test_cmdlinec C s | j | jj� d S )N)r5 r6 Zexe)r% r* r* r+ �test_exek s z TestProcessObjectLeaks.test_exec C s | j | jj� d S )N)r5 r6 �ppid)r% r* r* r+ � test_ppido s z TestProcessObjectLeaks.test_ppidz
POSIX only)�reasonc C s | j | jj� d S )N)r5 r6 Zuids)r% r* r* r+ � test_uidss s z TestProcessObjectLeaks.test_uidsc C s | j | jj� d S )N)r5 r6 Zgids)r% r* r* r+ � test_gidsx s z TestProcessObjectLeaks.test_gidsc C s | j | jj� d S )N)r5 r6 �status)r% r* r* r+ � test_status} s z"TestProcessObjectLeaks.test_statusc C s | j | jj� d S )N)r5 r6 �nice)r% r* r* r+ � test_nice� s z TestProcessObjectLeaks.test_nicec s t j� � �j� �fdd�� d S )Nc s �j j� �S )N)r6 rC r* )�nicenessr% r* r+ �<lambda>� s z6TestProcessObjectLeaks.test_nice_set.<locals>.<lambda>)�thisprocrC r5 )r% r* )rE r% r+ �
test_nice_set� s z$TestProcessObjectLeaks.test_nice_setz
not supportedc C s | j | jj� d S )N)r5 r6 �ionice)r% r* r* r+ � test_ionice� s z"TestProcessObjectLeaks.test_ionicec sZ t r"tj� �� j� �fdd�� n4� j� fdd�� tjtjtj � dd�}� j
t |� d S )Nc s � j j��S )N)r6 rI r* )r% �valuer* r+ rF � s z8TestProcessObjectLeaks.test_ionice_set.<locals>.<lambda>c s � j jtj�S )N)r6 rI �psutilZIOPRIO_CLASS_NONEr* )r% r* r+ rF � s � r ���) r rG rI r5 r- �partial�cextZproc_ioprio_set�os�getpid�
execute_w_exc�OSError)r% r) r* )r% rK r+ �test_ionice_set� s z&TestProcessObjectLeaks.test_ionice_setc C s | j | jj� d S )N)r5 r6 Z io_counters)r% r* r* r+ �test_io_counters� s z'TestProcessObjectLeaks.test_io_counterszworthless on POSIXc C s t j� j� | j| jj� d S )N)rL �Process�usernamer5 r6 )r% r* r* r+ �
test_username� s z$TestProcessObjectLeaks.test_usernamec C s | j | jj� d S )N)r5 r6 Z create_time)r% r* r* r+ �test_create_time� s z'TestProcessObjectLeaks.test_create_time)Zonly_ifc C s | j | jj� d S )N)r5 r6 Z num_threads)r% r* r* r+ �test_num_threads� s z'TestProcessObjectLeaks.test_num_threadsz WINDOWS onlyc C s | j | jj� d S )N)r5 r6 Z num_handles)r% r* r* r+ �test_num_handles� s z'TestProcessObjectLeaks.test_num_handlesc C s | j | jj� d S )N)r5 r6 Znum_fds)r% r* r* r+ � test_num_fds� s z#TestProcessObjectLeaks.test_num_fdsc C s | j | jj� d S )N)r5 r6 Znum_ctx_switches)r% r* r* r+ �test_num_ctx_switches� s z,TestProcessObjectLeaks.test_num_ctx_switchesc C s | j | jj� d S )N)r5 r6 �threads)r% r* r* r+ � test_threads� s z#TestProcessObjectLeaks.test_threadsc C s | j | jj� d S )N)r5 r6 � cpu_times)r% r* r* r+ �test_cpu_times� s z%TestProcessObjectLeaks.test_cpu_timesc C s | j | jj� d S )N)r5 r6 Zcpu_num)r% r* r* r+ � test_cpu_num� s z#TestProcessObjectLeaks.test_cpu_numc C s | j | jj� d S )N)r5 r6 Z memory_info)r% r* r* r+ �test_memory_info� s z'TestProcessObjectLeaks.test_memory_infoc C s | j | jj� d S )N)r5 r6 Zmemory_full_info)r% r* r* r+ �test_memory_full_info� s z,TestProcessObjectLeaks.test_memory_full_infoc C s | j | jj� d S )N)r5 r6 �terminal)r% r* r* r+ �
test_terminal� s z$TestProcessObjectLeaks.test_terminalc C s$ t rtn| j}| j| jj|d� d S )N)r# )r r$ r# r5 r6 �resume)r% r# r* r* r+ � test_resume� s z"TestProcessObjectLeaks.test_resumec C s | j | jj� d S )N)r5 r6 �cwd)r% r* r* r+ �test_cwd� s z TestProcessObjectLeaks.test_cwdc C s | j | jj� d S )N)r5 r6 � cpu_affinity)r% r* r* r+ �test_cpu_affinity� s z(TestProcessObjectLeaks.test_cpu_affinityc s4 t j� � �j� �fdd�� �jt�fdd�� d S )Nc s �j j� �S )N)r6 rl r* )�affinityr% r* r+ rF � s z>TestProcessObjectLeaks.test_cpu_affinity_set.<locals>.<lambda>c s � j jdg�S )NrM rN )r6 rl r* )r% r* r+ rF � s )rG rl r5 rS �
ValueError)r% r* )rn r% r+ �test_cpu_affinity_set� s z,TestProcessObjectLeaks.test_cpu_affinity_setc C s* t t� d�� | j| jj� W d Q R X d S )N�w)�openr r5 r6 Z
open_files)r% r* r* r+ �test_open_files� s z&TestProcessObjectLeaks.test_open_filesc C s | j | jj� d S )N)r5 r6 Z memory_maps)r% r* r* r+ �test_memory_maps� s z'TestProcessObjectLeaks.test_memory_mapsz
LINUX onlyc s � j � fdd�� d S )Nc s � j jtj�S )N)r6 �rlimitrL �
RLIMIT_NOFILEr* )r% r* r+ rF � s z4TestProcessObjectLeaks.test_rlimit.<locals>.<lambda>)r5 )r% r* )r% r+ � test_rlimit� s z"TestProcessObjectLeaks.test_rlimitc s< t jtj�� �j� �fdd�� �jttf�fdd�� d S )Nc s �j jtj� �S )N)r6 ru rL rv r* )�limitr% r* r+ rF � s z8TestProcessObjectLeaks.test_rlimit_set.<locals>.<lambda>c s � j jd�S )NrM rN )r6 ru r* )r% r* r+ rF � s )rG ru rL rv r5 rS rT ro )r% r* )rx r% r+ �test_rlimit_set� s z&TestProcessObjectLeaks.test_rlimit_setzworthless on WINDOWSc s6 t � �& trdnd� �j� �fdd�� W d Q R X d S )N�inet�allc s �j j� �S )N)r6 �net_connectionsr* )�kindr% r* r+ rF s z=TestProcessObjectLeaks.test_net_connections.<locals>.<lambda>)r r r5 )r% r* )r} r% r+ �test_net_connections� s z+TestProcessObjectLeaks.test_net_connectionsc C s | j | jj� d S )N)r5 r6 �environ)r% r* r* r+ � test_environ s z#TestProcessObjectLeaks.test_environc C s | j dd� � d S )Nc S s t jtj� �S )N)rP � proc_inforQ rR r* r* r* r+ rF s z7TestProcessObjectLeaks.test_proc_info.<locals>.<lambda>)r5 )r% r* r* r+ �test_proc_info
s z%TestProcessObjectLeaks.test_proc_infoN):�__name__�
__module__� __qualname__�__doc__rG r6 r4 r0 r8 r: r; r= r �mark�skipifr r? r@ rB rD rH r rJ rU r rV rY rZ r r r[ r r\ r] r^ r` rb r rc rd re rg ri rk r rm rp rs r rt r r rw ry r~ r
r� r� r* r* r* r+ r1 Z sd
r1 c sp e Zd ZdZe� fdd��Ze� fdd��Zdd� Zerhdd � Z d
d � Z
d d
� Z dd� Z dd� Z
dd� Z� ZS )�TestTerminatedProcessLeaksz�Repeat the tests above looking for leaks occurring when dealing
with terminated processes raising NoSuchProcess exception.
The C functions are still invoked but will follow different code
paths. We'll check those code paths.
c s: t � j� t� | _tj| jj�| _| jj� | jj � d S )N)
r
�
setUpClassr �subprL rW �pidr6 �kill�wait)�cls)r" r* r+ r� s
z%TestTerminatedProcessLeaks.setUpClassc s t � j� t| j� d S )N)r
�
tearDownClassr r� )r� )r" r* r+ r� s
z(TestTerminatedProcessLeaks.tearDownClassc C s& y
|� W n t jk
r Y nX d S )N)rL �
NoSuchProcess)r% r) r* r* r+ �call# s
z TestTerminatedProcessLeaks.callc C s | j | jj� d S )N)r5 r6 r� )r% r* r* r+ � test_kill+ s z$TestTerminatedProcessLeaks.test_killc C s | j | jj� d S )N)r5 r6 r )r% r* r* r+ �test_terminate. s z)TestTerminatedProcessLeaks.test_terminatec C s | j | jj� d S )N)r5 r6 Zsuspend)r% r* r* r+ � test_suspend1 s z'TestTerminatedProcessLeaks.test_suspendc C s | j | jj� d S )N)r5 r6 rh )r% r* r* r+ ri 4 s z&TestTerminatedProcessLeaks.test_resumec C s | j | jj� d S )N)r5 r6 r� )r% r* r* r+ � test_wait7 s z$TestTerminatedProcessLeaks.test_waitc s � fdd�}� j |� d S )Nc s( yt j� jj�S tk
r" Y nX d S )N)rP r� r6 r� r r* )r% r* r+ r� < s z7TestTerminatedProcessLeaks.test_proc_info.<locals>.call)r5 )r% r� r* )r% r+ r� : s z)TestTerminatedProcessLeaks.test_proc_info)r� r� r� r� � classmethodr� r� r� r r� r� r� ri r� r� �
__classcell__r* r* )r" r+ r� s r� z WINDOWS only)r> c @ s e Zd Zdd� Zdd� ZdS )� TestProcessDualImplementationc C s | j dd� � d S )Nc S s t jtj� dd�S )NT)�use_peb)rP � proc_cmdlinerQ rR r* r* r* r+ rF H s zETestProcessDualImplementation.test_cmdline_peb_true.<locals>.<lambda>)r5 )r% r* r* r+ �test_cmdline_peb_trueG s z3TestProcessDualImplementation.test_cmdline_peb_truec C s | j dd� � d S )Nc S s t jtj� dd�S )NF)r� )rP r� rQ rR r* r* r* r+ rF K s zFTestProcessDualImplementation.test_cmdline_peb_false.<locals>.<lambda>)r5 )r% r* r* r+ �test_cmdline_peb_falseJ s z4TestProcessDualImplementation.test_cmdline_peb_falseN)r� r� r� r� r� r* r* r* r+ r� E s r� c @ sP e Zd ZdZdd� Ze� dd� �Ze� dd� �Ze� dd � �Ze� d
d � �Z e� d d
� �Z
e� e j j
eorej� dkdd�e j j
e dd�dd� ���Ze j j
e dd�dd� �Zdd� Ze j j
edd�dd� �Zd d � Zd d � Ze j j
ed d�d!d"� �Ze j j
e �oe j j d#� d$d�e� d%d&� ��Z e� d'd(� �Z!e� e j j
e" dd�d)d*� ��Z#e� e j j
e�one j$� d+kd,d�d-d.� ��Z%d/d0� Z&e j j
ed d�d1d2� �Z'e� e j j
e( dd�d3d4� ��Z)e� e j j
e* dd�d5d6� ��Z+e� e j j
e, dd�d7d8� ��Z-e� d9d:� �Z.d;d<� Z/d=d>� Z0e�rLd?d@� Z1dAdB� Z2dCdD� Z3dEdF� Z4dGdH� Z5dIS )J�TestModuleFunctionsLeaksz&Test leaks of psutil module functions.c C s t � }|j| |j� d S )N)r r2 r{ )r% r3 r* r* r+ r4 V s z&TestModuleFunctionsLeaks.test_coveragec C s | j dd� � d S )Nc S s t jdd�S )NT)�logical)rL � cpu_countr* r* r* r+ rF ^ s z9TestModuleFunctionsLeaks.test_cpu_count.<locals>.<lambda>)r5 )r% r* r* r+ �test_cpu_count\ s z'TestModuleFunctionsLeaks.test_cpu_countc C s | j dd� � d S )Nc S s t jdd�S )NF)r� )rL r� r* r* r* r+ rF b s z?TestModuleFunctionsLeaks.test_cpu_count_cores.<locals>.<lambda>)r5 )r% r* r* r+ �test_cpu_count_cores` s z-TestModuleFunctionsLeaks.test_cpu_count_coresc C s | j tj� d S )N)r5 rL ra )r% r* r* r+ rb d s z'TestModuleFunctionsLeaks.test_cpu_timesc C s | j dd� � d S )Nc S s t jdd�S )NT)Zpercpu)rL ra r* r* r* r+ rF j s z=TestModuleFunctionsLeaks.test_per_cpu_times.<locals>.<lambda>)r5 )r% r* r* r+ �test_per_cpu_timesh s z+TestModuleFunctionsLeaks.test_per_cpu_timesc C s | j tj� d S )N)r5 rL Z cpu_stats)r% r* r* r+ �test_cpu_statsl s z'TestModuleFunctionsLeaks.test_cpu_statsZarm64zskipped due to #1892)r> z
not supportedc C s | j tj� d S )N)r5 rL Zcpu_freq)r% r* r* r+ �
test_cpu_freqp s z&TestModuleFunctionsLeaks.test_cpu_freqz WINDOWS onlyc C s t j� | jt j� d S )N)rL �
getloadavgr5 )r% r* r* r+ �test_getloadavgy s z(TestModuleFunctionsLeaks.test_getloadavgc C s | j tj� d S )N)r5 rL Zvirtual_memory)r% r* r* r+ �test_virtual_memory� s z,TestModuleFunctionsLeaks.test_virtual_memoryz&worthless on SUNOS (uses a subprocess)c C s | j tj� d S )N)r5 rL Z swap_memory)r% r* r* r+ �test_swap_memory� s z)TestModuleFunctionsLeaks.test_swap_memoryc C s$ t rtn| j}| jdd� |d� d S )Nc S s t jtj� �S )N)rL Z
pid_existsrQ rR r* r* r* r+ rF � s z:TestModuleFunctionsLeaks.test_pid_exists.<locals>.<lambda>)r# )r r$ r# r5 )r% r# r* r* r+ �test_pid_exists� s z(TestModuleFunctionsLeaks.test_pid_existsc C s$ t rtn| j}| jdd� |d� d S )Nc S s
t jd�S )N�.)rL �
disk_usager* r* r* r+ rF � s z:TestModuleFunctionsLeaks.test_disk_usage.<locals>.<lambda>)r# )r r$ r# r5 )r% r# r* r* r+ �test_disk_usage� s z(TestModuleFunctionsLeaks.test_disk_usagezQEMU user not supportedc C s | j tj� d S )N)r5 rL Zdisk_partitions)r% r* r* r+ �test_disk_partitions� s z-TestModuleFunctionsLeaks.test_disk_partitionsz/proc/diskstatsz3/proc/diskstats not available on this Linux versionc C s | j dd� � d S )Nc S s t jdd�S )NF)�nowrap)rL Zdisk_io_countersr* r* r* r+ rF � s z@TestModuleFunctionsLeaks.test_disk_io_counters.<locals>.<lambda>)r5 )r% r* r* r+ �test_disk_io_counters� s z.TestModuleFunctionsLeaks.test_disk_io_countersc C s | j tj� d S )N)r5 rL Zpids)r% r* r* r+ � test_pids� s z"TestModuleFunctionsLeaks.test_pidsc C s | j dd� � d S )Nc S s t jdd�S )NF)r� )rL Znet_io_countersr* r* r* r+ rF � s z?TestModuleFunctionsLeaks.test_net_io_counters.<locals>.<lambda>)r5 )r% r* r* r+ �test_net_io_counters� s z-TestModuleFunctionsLeaks.test_net_io_countersr zneed root accessc
C s0 t jdd� t� � | jdd� � W d Q R X d S )Nr{ )r} c S s t jdd�S )Nr{ )r} )rL r| r* r* r* r+ rF � s z?TestModuleFunctionsLeaks.test_net_connections.<locals>.<lambda>)rL r| r r5 )r% r* r* r+ r~ � s z-TestModuleFunctionsLeaks.test_net_connectionsc C s" t rdn| j}| jtj|d� d S )N�P i )� tolerancei @ )r r� r5 rL Z net_if_addrs)r% r� r* r* r+ �test_net_if_addrs� s z*TestModuleFunctionsLeaks.test_net_if_addrsc C s | j tj� d S )N)r5 rL Z net_if_stats)r% r* r* r+ �test_net_if_stats� s z*TestModuleFunctionsLeaks.test_net_if_statsc C s | j tj� d S )N)r5 rL Zsensors_battery)r% r* r* r+ �test_sensors_battery� s z-TestModuleFunctionsLeaks.test_sensors_batteryc C s | j tj� d S )N)r5 rL Zsensors_temperatures)r% r* r* r+ �test_sensors_temperatures� s z2TestModuleFunctionsLeaks.test_sensors_temperaturesc C s | j tj� d S )N)r5 rL Z sensors_fans)r% r* r* r+ �test_sensors_fans� s z*TestModuleFunctionsLeaks.test_sensors_fansc C s | j tj� d S )N)r5 rL Z boot_time)r% r* r* r+ �test_boot_time� s z'TestModuleFunctionsLeaks.test_boot_timec C s | j tj� d S )N)r5 rL Zusers)r% r* r* r+ �
test_users� s z#TestModuleFunctionsLeaks.test_usersc C s | j dd� � d S )Nc S s
t jd�S )NF)rL Z
_set_debugr* r* r* r+ rF � s z9TestModuleFunctionsLeaks.test_set_debug.<locals>.<lambda>)r5 )r% r* r* r+ �test_set_debug� s z'TestModuleFunctionsLeaks.test_set_debugc C s | j tj� d S )N)r5 rP Zwinservice_enumerate)r% r* r* r+ �test_win_service_iter� s z.TestModuleFunctionsLeaks.test_win_service_iterc C s d S )Nr* )r% r* r* r+ �test_win_service_get� s z-TestModuleFunctionsLeaks.test_win_service_getc s& t tj� �j� � | j� fdd�� d S )Nc s
t j� �S )N)rP Zwinservice_query_configr* )r7 r* r+ rF � s zFTestModuleFunctionsLeaks.test_win_service_get_config.<locals>.<lambda>)�nextrL �win_service_iterr7 r5 )r% r* )r7 r+ �test_win_service_get_config� s z4TestModuleFunctionsLeaks.test_win_service_get_configc s& t tj� �j� � | j� fdd�� d S )Nc s
t j� �S )N)rP Zwinservice_query_statusr* )r7 r* r+ rF � s zFTestModuleFunctionsLeaks.test_win_service_get_status.<locals>.<lambda>)r� rL r� r7 r5 )r% r* )r7 r+ �test_win_service_get_status� s z4TestModuleFunctionsLeaks.test_win_service_get_statusc s& t tj� �j� � | j� fdd�� d S )Nc s
t j� �S )N)rP Zwinservice_query_descrr* )r7 r* r+ rF � s zKTestModuleFunctionsLeaks.test_win_service_get_description.<locals>.<lambda>)r� rL r� r7 r5 )r% r* )r7 r+ � test_win_service_get_description� s z9TestModuleFunctionsLeaks.test_win_service_get_descriptionN)6r� r� r� r� r4 r0 r� r� rb r� r� r r� r� r �platform�machiner r� r r� r� r r� r� r� r r� r rQ �path�existsr� r� r r� �getuidr~ r� r� r r� r r� r r� r� r� r� r� r� r� r� r� r* r* r* r+ r� S sV ( r� )4r� �
__future__r r- rQ r� rL Zpsutil._commonr r r r r r Zpsutil._compatr r
Z psutil.testsr r r
r r r r r r r r r r r r r r r r r r r Z _psplatformrP rW rG r$ r0 r1 r� r� r� r� r� r* r* r* r+ �<module> s\ 66
|