B �=�_�9�@sdZddlmZddlZddlZddlZddlZddlmZddlmZddlm Z ddlm Z ddlmZdd lmZdd l mZddl mZddlmZdd lmZddlmZddlmZddlmZddlmZddlmZddlmZddlmZddlmZddlmZddlmZddlmZddlmZddlmZddlm Z ddlm!Z!ddlm"Z"ddlm#Z#ddlm$Z$dd lm%Z%ej&j'Z'e�(�Z)d!Z*d"d#�Z+Gd$d%�d%e$�Z,Gd&d'�d'e,�Z-e%�.ed(�Gd)d*�d*e$��Z/Gd+d,�d,e$�Z0e1d-k�rdd.l2m3Z3e3e4�dS)/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)�create_sockets)� get_testfn)�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)�process_namespace)�skip_on_access_denied)�spawn_testproc)�system_namespace)� terminate)�TestMemoryLeak)�unittest�cCsdd�}|S)zsDecorator for those Linux functions which are implemented in pure Python, and which we want to run faster. cst��fdd��}|S)NcsFtr2|jj}zt|j_�|f|�|�S||j_Xn�|f|�|�SdS)N)r� __class__�times� FEW_TIMES)�self�args�kwargs�before)�fun��M/opt/alt/python37/lib64/python3.7/site-packages/psutil/tests/test_memleaks.py�wrapperAsz5fewtimes_if_linux..decorator..wrapper)� functools�wraps)r(r+r))r(r*� decorator@sz$fewtimes_if_linux..decoratorr))r.r)r)r*�fewtimes_if_linux<s r/c@s�eZdZdZeZdd�Ze�dd��Ze�dd��Z e�dd ��Z e�d d��Ze� ed�e�d d��Ze� ed�e�dd��Ze�dd��Zdd�Zdd�Ze� ed�dd��Ze� ed�dd��Ze� ed�e�dd��Ze� ed�dd ��Ze�d!d"��Ze�eed#�d$d%��Ze� ed&�d'd(��Ze� ed�e�d)d*��Z e�d+d,��Z!e�eed#�d-d.��Z"e�d/d0��Z#e�e� e$d�d1d2��Z%e�d3d4��Z&e�d5d6��Z'e� ed�e�d7d8��Z(d9d:�Z)e�d;d<��Z*e� e+d�d=d>��Z,e� e+d�d?d@��Z-e�dAdB��Z.e� e/d�e�dCdD��Z0e� e1dE�e� e2d�dFdG��Z3e� e1dE�e� e2d�dHdI��Z4e�e� edJ�dKdL��Z5e� e6d�dMdN��Z7e� ed&�dOdP��Z8dQS)R�TestProcessObjectLeaksz$Test leaks of Process class methods.cCs td�}|�||j|j�dS)N)r�test_class_coverage�gettersZsetters)r$�nsr)r)r*� test_coverageZsz$TestProcessObjectLeaks.test_coveragecCs|�|jj�dS)N)�execute�proc�name)r$r)r)r*� test_name^sz TestProcessObjectLeaks.test_namecCs|�|jj�dS)N)r5r6Zcmdline)r$r)r)r*�test_cmdlinebsz#TestProcessObjectLeaks.test_cmdlinecCs|�|jj�dS)N)r5r6Zexe)r$r)r)r*�test_exefszTestProcessObjectLeaks.test_execCs|�|jj�dS)N)r5r6�ppid)r$r)r)r*� test_ppidjsz TestProcessObjectLeaks.test_ppidz POSIX onlycCs|�|jj�dS)N)r5r6Zuids)r$r)r)r*� test_uidsnsz TestProcessObjectLeaks.test_uidscCs|�|jj�dS)N)r5r6Zgids)r$r)r)r*� test_gidsssz TestProcessObjectLeaks.test_gidscCs|�|jj�dS)N)r5r6�status)r$r)r)r*�test_statusxsz"TestProcessObjectLeaks.test_statuscCs|�|jj�dS)N)r5r6�nice)r$r)r)r*� test_nice|sz TestProcessObjectLeaks.test_nicecs t��fdd��dS)Ncs�j��S)N)r6rAr))�nicenessr$r)r*��z6TestProcessObjectLeaks.test_nice_set..)�thisprocrAr5)r$r))rCr$r*� test_nice_setsz$TestProcessObjectLeaks.test_nice_setz not supportedcCs|�|jj�dS)N)r5r6�ionice)r$r)r)r*�test_ionice�sz"TestProcessObjectLeaks.test_ionicecsZtr"t��fdd��n4��fdd��t�tjt� �dd�}�� t|�dS)Ncs�j��S)N)r6rHr))r$�valuer)r*rD�rEz8TestProcessObjectLeaks.test_ionice_set..cs�j�tj�S)N)r6rH�psutilZIOPRIO_CLASS_NONEr))r$r)r*rD�rE��r)rrFrHr5r,�partial�cextZproc_ioprio_set�os�getpid� execute_w_exc�OSError)r$r(r))r$rJr*�test_ionice_set�sz&TestProcessObjectLeaks.test_ionice_setcCs|�|jj�dS)N)r5r6Zio_counters)r$r)r)r*�test_io_counters�sz'TestProcessObjectLeaks.test_io_counterszworthless on POSIXcCst��|�|jj�dS)N)rK�Process�usernamer5r6)r$r)r)r*� test_username�sz$TestProcessObjectLeaks.test_usernamecCs|�|jj�dS)N)r5r6Zcreate_time)r$r)r)r*�test_create_time�sz'TestProcessObjectLeaks.test_create_time)Zonly_ifcCs|�|jj�dS)N)r5r6�num_threads)r$r)r)r*�test_num_threads�sz'TestProcessObjectLeaks.test_num_threadszWINDOWS onlycCs|�|jj�dS)N)r5r6Znum_handles)r$r)r)r*�test_num_handles�sz'TestProcessObjectLeaks.test_num_handlescCs|�|jj�dS)N)r5r6Znum_fds)r$r)r)r*�test_num_fds�sz#TestProcessObjectLeaks.test_num_fdscCs|�|jj�dS)N)r5r6Znum_ctx_switches)r$r)r)r*�test_num_ctx_switches�sz,TestProcessObjectLeaks.test_num_ctx_switchescCs|�|jj�dS)N)r5r6�threads)r$r)r)r*�test_threads�sz#TestProcessObjectLeaks.test_threadscCs|�|jj�dS)N)r5r6� cpu_times)r$r)r)r*�test_cpu_times�sz%TestProcessObjectLeaks.test_cpu_timescCs|�|jj�dS)N)r5r6Zcpu_num)r$r)r)r*�test_cpu_num�sz#TestProcessObjectLeaks.test_cpu_numcCs|�|jj�dS)N)r5r6Zmemory_info)r$r)r)r*�test_memory_info�sz'TestProcessObjectLeaks.test_memory_infocCs|�|jj�dS)N)r5r6Zmemory_full_info)r$r)r)r*�test_memory_full_info�sz,TestProcessObjectLeaks.test_memory_full_infocCs|�|jj�dS)N)r5r6�terminal)r$r)r)r*� test_terminal�sz$TestProcessObjectLeaks.test_terminalcCs$trtn|j}|j|jj|d�dS)N)r")rr#r"r5r6�resume)r$r"r)r)r*�test_resume�sz"TestProcessObjectLeaks.test_resumecCs|�|jj�dS)N)r5r6�cwd)r$r)r)r*�test_cwd�szTestProcessObjectLeaks.test_cwdcCs|�|jj�dS)N)r5r6�cpu_affinity)r$r)r)r*�test_cpu_affinity�sz(TestProcessObjectLeaks.test_cpu_affinitycs4t��fdd��t�fdd��dS)Ncs�j��S)N)r6rkr))�affinityr$r)r*rD�rEz>TestProcessObjectLeaks.test_cpu_affinity_set..cs�j�dg�S)NrL)r6rkr))r$r)r*rD�rE)rFrkr5rQ� ValueError)r$r))rmr$r*�test_cpu_affinity_set�sz,TestProcessObjectLeaks.test_cpu_affinity_setc Cs*tt�d��|�|jj�WdQRXdS)N�w)�openrr5r6Z open_files)r$r)r)r*�test_open_files�sz&TestProcessObjectLeaks.test_open_filescCs|�|jj�dS)N)r5r6Zmemory_maps)r$r)r)r*�test_memory_maps�sz'TestProcessObjectLeaks.test_memory_mapsz LINUX onlycs��fdd��dS)Ncs�j�tj�S)N)r6�rlimitrK� RLIMIT_NOFILEr))r$r)r*rD�rEz4TestProcessObjectLeaks.test_rlimit..)r5)r$r))r$r*�test_rlimit�sz"TestProcessObjectLeaks.test_rlimitcs<t�tj��fdd��ttf�fdd��dS)Ncs�j�tj��S)N)r6rtrKrur))�limitr$r)r*rD�rEz8TestProcessObjectLeaks.test_rlimit_set..cs�j�d�S)NrL)r6rtr))r$r)r*rD�rE)rFrtrKrur5rQrRrn)r$r))rwr$r*�test_rlimit_set�sz&TestProcessObjectLeaks.test_rlimit_setzworthless on WINDOWSc s6t��&trdnd��fdd��WdQRXdS)N�inet�allcs�j��S)N)r6Zconnectionsr))�kindr$r)r*rDrEz9TestProcessObjectLeaks.test_connections..)rrr5)r$r))r{r$r*�test_connections�sz'TestProcessObjectLeaks.test_connectionscCs|�|jj�dS)N)r5r6�environ)r$r)r)r*�test_environsz#TestProcessObjectLeaks.test_environcCs|�dd��dS)NcSst�t��S)N)rN� proc_inforOrPr)r)r)r*rDrEz7TestProcessObjectLeaks.test_proc_info..)r5)r$r)r)r*�test_proc_infosz%TestProcessObjectLeaks.test_proc_infoN)9�__name__� __module__�__qualname__�__doc__rFr6r4r/r8r9r:r<r�skipIfrr=r>r@rBrGrrIrSrrTrWrXrrrZrr[r\r]r_rarrbrcrdrfrhrjr rlrorrrrsrrrvrxr|rr~r�r)r)r)r*r0Usd r0cspeZdZdZe�fdd��Ze�fdd��Zdd�Zerhdd �Z d d�Z dd �Zdd�Zdd�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. cs:t��t�|_t�|jj�|_|j��|j� �dS)N) r � setUpClassr�subprKrU�pidr6�kill�wait)�cls)r!r)r*r�s z%TestTerminatedProcessLeaks.setUpClasscst��t|j�dS)N)r � tearDownClassrr�)r�)r!r)r*r�s z(TestTerminatedProcessLeaks.tearDownClasscCs&y |�Wntjk r YnXdS)N)rK� NoSuchProcess)r$r(r)r)r*�calls zTestTerminatedProcessLeaks.callcCs|�|jj�dS)N)r5r6r�)r$r)r)r*� test_kill'sz$TestTerminatedProcessLeaks.test_killcCs|�|jj�dS)N)r5r6r)r$r)r)r*�test_terminate*sz)TestTerminatedProcessLeaks.test_terminatecCs|�|jj�dS)N)r5r6Zsuspend)r$r)r)r*�test_suspend-sz'TestTerminatedProcessLeaks.test_suspendcCs|�|jj�dS)N)r5r6rg)r$r)r)r*rh0sz&TestTerminatedProcessLeaks.test_resumecCs|�|jj�dS)N)r5r6r�)r$r)r)r*� test_wait3sz$TestTerminatedProcessLeaks.test_waitcs�fdd�}��|�dS)Ncs(yt��jj�Stk r"YnXdS)N)rNrr6r�r r))r$r)r*r�8sz7TestTerminatedProcessLeaks.test_proc_info..call)r5)r$r�r))r$r*r�6sz)TestTerminatedProcessLeaks.test_proc_info)r�r�r�r��classmethodr�r�r�rr�r�r�rhr�r�� __classcell__r)r))r!r*r�sr�zWINDOWS onlyc@seZdZdd�Zdd�ZdS)�TestProcessDualImplementationcCs|�dd��dS)NcSstjt��dd�S)NT)�use_peb)rN�proc_cmdlinerOrPr)r)r)r*rDErEzETestProcessDualImplementation.test_cmdline_peb_true..)r5)r$r)r)r*�test_cmdline_peb_trueDsz3TestProcessDualImplementation.test_cmdline_peb_truecCs|�dd��dS)NcSstjt��dd�S)NF)r�)rNr�rOrPr)r)r)r*rDHrEzFTestProcessDualImplementation.test_cmdline_peb_false..)r5)r$r)r)r*�test_cmdline_peb_falseGsz4TestProcessDualImplementation.test_cmdline_peb_falseN)r�r�r�r�r�r)r)r)r*r�Asr�c@s�eZdZdZdd�Ze�dd��Ze�dd��Ze�dd ��Ze�d d��Z e�dd ��Z e�e�e d�dd��Ze�ed�dd��Zdd�Ze�ed�dd��Zdd�Zdd�Zdd�Ze�eo�ej�d�d �e�d!d"��Ze�d#d$��Ze�e�ed�d%d&��Ze�e�e�o*e� �d'kd(�d)d*��Z!d+d,�Z"d-d.�Z#e�e�e$d�d/d0��Z%e�e�e&d�d1d2��Z'e�e�e(d�d3d4��Z)e�d5d6��Z*d7d8�Z+e�r�d9d:�Z,d;d<�Z-d=d>�Z.d?d@�Z/dAdB�Z0dCS)D�TestModuleFunctionsLeaksz&Test leaks of psutil module functions.cCst�}|�||j�dS)N)rr1rz)r$r3r)r)r*r4Ssz&TestModuleFunctionsLeaks.test_coveragecCs|�dd��dS)NcSstjdd�S)NT)�logical)rK� cpu_countr)r)r)r*rD[rEz9TestModuleFunctionsLeaks.test_cpu_count..)r5)r$r)r)r*�test_cpu_countYsz'TestModuleFunctionsLeaks.test_cpu_countcCs|�dd��dS)NcSstjdd�S)NF)r�)rKr�r)r)r)r*rD_rEzBTestModuleFunctionsLeaks.test_cpu_count_physical..)r5)r$r)r)r*�test_cpu_count_physical]sz0TestModuleFunctionsLeaks.test_cpu_count_physicalcCs|�tj�dS)N)r5rKr`)r$r)r)r*raasz'TestModuleFunctionsLeaks.test_cpu_timescCs|�dd��dS)NcSstjdd�S)NT)Zpercpu)rKr`r)r)r)r*rDgrEz=TestModuleFunctionsLeaks.test_per_cpu_times..)r5)r$r)r)r*�test_per_cpu_timesesz+TestModuleFunctionsLeaks.test_per_cpu_timescCs|�tj�dS)N)r5rKZ cpu_stats)r$r)r)r*�test_cpu_statsisz'TestModuleFunctionsLeaks.test_cpu_statsz not supportedcCs|�tj�dS)N)r5rKZcpu_freq)r$r)r)r*� test_cpu_freqmsz&TestModuleFunctionsLeaks.test_cpu_freqzWINDOWS onlycCst��|�tj�dS)N)rK� getloadavgr5)r$r)r)r*�test_getloadavgrsz(TestModuleFunctionsLeaks.test_getloadavgcCs|�tj�dS)N)r5rKZvirtual_memory)r$r)r)r*�test_virtual_memoryysz,TestModuleFunctionsLeaks.test_virtual_memoryz&worthless on SUNOS (uses a subprocess)cCs|�tj�dS)N)r5rKZswap_memory)r$r)r)r*�test_swap_memory}sz)TestModuleFunctionsLeaks.test_swap_memorycCs$trtn|j}|jdd�|d�dS)NcSst�t��S)N)rKZ pid_existsrOrPr)r)r)r*rD�rEz:TestModuleFunctionsLeaks.test_pid_exists..)r")rr#r"r5)r$r"r)r)r*�test_pid_exists�sz(TestModuleFunctionsLeaks.test_pid_existscCs$trtn|j}|jdd�|d�dS)NcSs t�d�S)N�.)rK� disk_usager)r)r)r*rD�rEz:TestModuleFunctionsLeaks.test_disk_usage..)r")rr#r"r5)r$r"r)r)r*�test_disk_usage�sz(TestModuleFunctionsLeaks.test_disk_usagecCs|�tj�dS)N)r5rKZdisk_partitions)r$r)r)r*�test_disk_partitions�sz-TestModuleFunctionsLeaks.test_disk_partitionsz/proc/diskstatsz3/proc/diskstats not available on this Linux versioncCs|�dd��dS)NcSstjdd�S)NF)�nowrap)rKZdisk_io_countersr)r)r)r*rD�rEz@TestModuleFunctionsLeaks.test_disk_io_counters..)r5)r$r)r)r*�test_disk_io_counters�sz.TestModuleFunctionsLeaks.test_disk_io_counterscCs|�tj�dS)N)r5rKZpids)r$r)r)r*� test_pids�sz"TestModuleFunctionsLeaks.test_pidscCs|�dd��dS)NcSstjdd�S)NF)r�)rKZnet_io_countersr)r)r)r*rD�rEz?TestModuleFunctionsLeaks.test_net_io_counters..)r5)r$r)r)r*�test_net_io_counters�sz-TestModuleFunctionsLeaks.test_net_io_countersrzneed root accessc Cs0tjdd�t��|�dd��WdQRXdS)Nrz)r{cSstjdd�S)Nrz)r{)rK�net_connectionsr)r)r)r*rD�rEz?TestModuleFunctionsLeaks.test_net_connections..)rKr�rr5)r$r)r)r*�test_net_connections�sz-TestModuleFunctionsLeaks.test_net_connectionscCs"trdn|j}|jtj|d�dS)Ni@)� tolerance)rr�r5rKZnet_if_addrs)r$r�r)r)r*�test_net_if_addrs�sz*TestModuleFunctionsLeaks.test_net_if_addrscCs|�tj�dS)N)r5rKZnet_if_stats)r$r)r)r*�test_net_if_stats�sz*TestModuleFunctionsLeaks.test_net_if_statscCs|�tj�dS)N)r5rKZsensors_battery)r$r)r)r*�test_sensors_battery�sz-TestModuleFunctionsLeaks.test_sensors_batterycCs|�tj�dS)N)r5rKZsensors_temperatures)r$r)r)r*�test_sensors_temperatures�sz2TestModuleFunctionsLeaks.test_sensors_temperaturescCs|�tj�dS)N)r5rKZsensors_fans)r$r)r)r*�test_sensors_fans�sz*TestModuleFunctionsLeaks.test_sensors_fanscCs|�tj�dS)N)r5rKZ boot_time)r$r)r)r*�test_boot_time�sz'TestModuleFunctionsLeaks.test_boot_timecCs|�tj�dS)N)r5rKZusers)r$r)r)r*� test_users�sz#TestModuleFunctionsLeaks.test_userscCs|�tj�dS)N)r5rNZwinservice_enumerate)r$r)r)r*�test_win_service_iter�sz.TestModuleFunctionsLeaks.test_win_service_itercCsdS)Nr))r$r)r)r*�test_win_service_get�sz-TestModuleFunctionsLeaks.test_win_service_getcs&tt��|��fdd��dS)Ncs t��S)N)rNZwinservice_query_configr))r7r)r*rD�rEzFTestModuleFunctionsLeaks.test_win_service_get_config..)�nextrK�win_service_iterr7r5)r$r))r7r*�test_win_service_get_config�sz4TestModuleFunctionsLeaks.test_win_service_get_configcs&tt��|��fdd��dS)Ncs t��S)N)rNZwinservice_query_statusr))r7r)r*rD�rEzFTestModuleFunctionsLeaks.test_win_service_get_status..)r�rKr�r7r5)r$r))r7r*�test_win_service_get_status�sz4TestModuleFunctionsLeaks.test_win_service_get_statuscs&tt��|��fdd��dS)Ncs t��S)N)rNZwinservice_query_descrr))r7r)r*rD�rEzKTestModuleFunctionsLeaks.test_win_service_get_description..)r�rKr�r7r5)r$r))r7r*� test_win_service_get_description�sz9TestModuleFunctionsLeaks.test_win_service_get_descriptionN)1r�r�r�r�r4r/r�r�rar�r�rr�rr�rr�r�rr�r�r�r�rrO�path�existsr�r�rr�r�getuidr�r�r�rr�rr�rr�r�r�r�r�r�r�r�r)r)r)r*r�PsN$r��__main__)� run_from_name)5r�� __future__rr,rOrKZpsutil._commonrrrrrrZpsutil._compatr r Zpsutil.testsrrr rrrrrrrrrrrrrrrrrrZ_psplatformrNrUrFr#r/r0r�r�r�r�r�Zpsutil.tests.runnerr��__file__r)r)r)r*�s`76