U
    ;g;                     @   s  d Z ddlZddlZddlZddlZddlZddlmZ ddlmZ ddlmZ ddlm	Z	 ddlm
Z
 ddlmZ dd	lmZ dd
lmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlm Z  ddlm!Z! ej"j#Z#e$ Z%dZ&dd  Z'G d!d" d"eZ(G d#d$ d$e(Z)ej*j+e d%d&G d'd( d(eZ,G d)d* d*eZ-dS )+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.
    N)LINUX)MACOS)OPENBSD)POSIX)SUNOS)WINDOWS)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)TestMemoryLeak)create_sockets)
get_testfn)process_namespace)pytest)skip_on_access_denied)spawn_testproc)system_namespace)	terminate   c                  C   s   dd } | S )zsDecorator for those Linux functions which are implemented in pure
    Python, and which we want to run faster.
    c                    s   t   fdd}|S )Nc                    sJ   t r6| jj}zt| j_ | f||W S || j_X n | f||S d S N)r   	__class__times	FEW_TIMES)selfargskwargsbeforefun Z/opt/bot/vods_filmes/script/venv/lib/python3.8/site-packages/psutil/tests/test_memleaks.pywrapper@   s    z5fewtimes_if_linux.<locals>.decorator.<locals>.wrapper)	functoolswraps)r'   r*   r(   r&   r)   	decorator?   s    z$fewtimes_if_linux.<locals>.decoratorr(   )r-   r(   r(   r)   fewtimes_if_linux:   s    r.   c                   @   s0  e Zd ZdZeZdd Ze dd Ze dd Z	e dd	 Z
e d
d Zejje dde dd Zejje dde dd Ze dd Zdd Zdd Zejje dddd Zejje dddd Zejje dde dd Zejjeddd d! Ze d"d# Ze eed$d%d& Zejje d'dd(d) Z ejje dde d*d+ Z!e d,d- Z"e eed$d.d/ Z#e d0d1 Z$e ejje% ddd2d3 Z&e d4d5 Z'e d6d7 Z(ejje dde d8d9 Z)d:d; Z*e d<d= Z+ejje, ddd>d? Z-ejje, ddd@dA Z.e dBdC Z/ejje0 dde dDdE Z1ejje2 dFdejje3 dddGdH Z4ejje2 dFdejje3 dddIdJ Z5e ejjedKddLdM Z6ejje7 dddNdO Z8ejje d'ddPdQ Z9dRS )STestProcessObjectLeaksz$Test leaks of Process class methods.c                 C   s    t d }|| |j|j  d S r   )r   test_class_coveragegettersZsettersr"   nsr(   r(   r)   test_coverage[   s    z$TestProcessObjectLeaks.test_coveragec                 C   s   |  | jj d S r   )executeprocnamer"   r(   r(   r)   	test_name_   s    z TestProcessObjectLeaks.test_namec                 C   s   |  | jj d S r   )r5   r6   Zcmdliner8   r(   r(   r)   test_cmdlinec   s    z#TestProcessObjectLeaks.test_cmdlinec                 C   s   |  | jj d S r   )r5   r6   exer8   r(   r(   r)   test_exeg   s    zTestProcessObjectLeaks.test_exec                 C   s   |  | jj d S r   )r5   r6   Zppidr8   r(   r(   r)   	test_ppidk   s    z TestProcessObjectLeaks.test_ppidz
POSIX onlyreasonc                 C   s   |  | jj d S r   )r5   r6   Zuidsr8   r(   r(   r)   	test_uidso   s    z TestProcessObjectLeaks.test_uidsc                 C   s   |  | jj d S r   )r5   r6   Zgidsr8   r(   r(   r)   	test_gidst   s    z TestProcessObjectLeaks.test_gidsc                 C   s   |  | jj d S r   )r5   r6   statusr8   r(   r(   r)   test_statusy   s    z"TestProcessObjectLeaks.test_statusc                 C   s   |  | jj d S r   )r5   r6   nicer8   r(   r(   r)   	test_nice}   s    z TestProcessObjectLeaks.test_nicec                    s    t    fdd d S )Nc                      s   j  S r   )r6   rD   r(   Znicenessr"   r(   r)   <lambda>       z6TestProcessObjectLeaks.test_nice_set.<locals>.<lambda>)thisprocrD   r5   r8   r(   rF   r)   test_nice_set   s    z$TestProcessObjectLeaks.test_nice_setnot supportedc                 C   s   |  | jj d S r   )r5   r6   ionicer8   r(   r(   r)   test_ionice   s    z"TestProcessObjectLeaks.test_ionicec                    sZ   t r"t   fdd n4  fdd ttjt	 dd} 
t| d S )Nc                      s    j S r   )r6   rL   r(   r"   valuer(   r)   rG      rH   z8TestProcessObjectLeaks.test_ionice_set.<locals>.<lambda>c                      s    j tjS r   )r6   rL   psutilZIOPRIO_CLASS_NONEr(   r8   r(   r)   rG      rH   r   )r   rI   rL   r5   r+   partialcextZproc_ioprio_setosgetpidexecute_w_excOSErrorr"   r'   r(   rN   r)   test_ionice_set   s    z&TestProcessObjectLeaks.test_ionice_setc                 C   s   |  | jj d S r   )r5   r6   Zio_countersr8   r(   r(   r)   test_io_counters   s    z'TestProcessObjectLeaks.test_io_counterszworthless on POSIXc                 C   s   t    | | jj d S r   )rP   Processusernamer5   r6   r8   r(   r(   r)   test_username   s    z$TestProcessObjectLeaks.test_usernamec                 C   s   |  | jj d S r   )r5   r6   Zcreate_timer8   r(   r(   r)   test_create_time   s    z'TestProcessObjectLeaks.test_create_time)Zonly_ifc                 C   s   |  | jj d S r   )r5   r6   Znum_threadsr8   r(   r(   r)   test_num_threads   s    z'TestProcessObjectLeaks.test_num_threadsWINDOWS onlyc                 C   s   |  | jj d S r   )r5   r6   Znum_handlesr8   r(   r(   r)   test_num_handles   s    z'TestProcessObjectLeaks.test_num_handlesc                 C   s   |  | jj d S r   )r5   r6   Znum_fdsr8   r(   r(   r)   test_num_fds   s    z#TestProcessObjectLeaks.test_num_fdsc                 C   s   |  | jj d S r   )r5   r6   Znum_ctx_switchesr8   r(   r(   r)   test_num_ctx_switches   s    z,TestProcessObjectLeaks.test_num_ctx_switchesc                 C   s   |  | jj d S r   )r5   r6   threadsr8   r(   r(   r)   test_threads   s    z#TestProcessObjectLeaks.test_threadsc                 C   s   |  | jj d S r   )r5   r6   	cpu_timesr8   r(   r(   r)   test_cpu_times   s    z%TestProcessObjectLeaks.test_cpu_timesc                 C   s   |  | jj d S r   )r5   r6   Zcpu_numr8   r(   r(   r)   test_cpu_num   s    z#TestProcessObjectLeaks.test_cpu_numc                 C   s   |  | jj d S r   )r5   r6   Zmemory_infor8   r(   r(   r)   test_memory_info   s    z'TestProcessObjectLeaks.test_memory_infoc                 C   s   |  | jj d S r   )r5   r6   Zmemory_full_infor8   r(   r(   r)   test_memory_full_info   s    z,TestProcessObjectLeaks.test_memory_full_infoc                 C   s   |  | jj d S r   )r5   r6   Zterminalr8   r(   r(   r)   test_terminal   s    z$TestProcessObjectLeaks.test_terminalc                 C   s$   t rtn| j}| j| jj|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j d S r   )r5   r6   cwdr8   r(   r(   r)   test_cwd   s    zTestProcessObjectLeaks.test_cwdc                 C   s   |  | jj d S r   )r5   r6   cpu_affinityr8   r(   r(   r)   test_cpu_affinity   s    z(TestProcessObjectLeaks.test_cpu_affinityc                    s4   t    fdd tfdd d S )Nc                      s   j  S r   r6   rr   r(   Zaffinityr"   r(   r)   rG      rH   z>TestProcessObjectLeaks.test_cpu_affinity_set.<locals>.<lambda>c                      s    j dgS NrQ   rt   r(   r8   r(   r)   rG      rH   )rI   rr   r5   rV   
ValueErrorr8   r(   ru   r)   test_cpu_affinity_set   s    z,TestProcessObjectLeaks.test_cpu_affinity_setc              	   C   s*   t t d | | jj W 5 Q R X d S )Nw)openr   r5   r6   Z
open_filesr8   r(   r(   r)   test_open_files   s    z&TestProcessObjectLeaks.test_open_filesc                 C   s   |  | jj d S r   )r5   r6   Zmemory_mapsr8   r(   r(   r)   test_memory_maps   s    z'TestProcessObjectLeaks.test_memory_mapsz
LINUX onlyc                    s      fdd d S )Nc                      s    j tjS r   r6   rlimitrP   RLIMIT_NOFILEr(   r8   r(   r)   rG      rH   z4TestProcessObjectLeaks.test_rlimit.<locals>.<lambda>r5   r8   r(   r8   r)   test_rlimit   s    z"TestProcessObjectLeaks.test_rlimitc                    s<   t tj  fdd ttffdd d S )Nc                      s   j tj S r   r}   r(   limitr"   r(   r)   rG      rH   z8TestProcessObjectLeaks.test_rlimit_set.<locals>.<lambda>c                      s    j dS rv   )r6   r~   r(   r8   r(   r)   rG      rH   )rI   r~   rP   r   r5   rV   rW   rw   r8   r(   r   r)   test_rlimit_set   s    z&TestProcessObjectLeaks.test_rlimit_setzworthless on WINDOWSc              	      s6   t  & trdnd  fdd W 5 Q R X d S )NZinetallc                      s   j  S r   )r6   net_connectionsr(   kindr"   r(   r)   rG      rH   z=TestProcessObjectLeaks.test_net_connections.<locals>.<lambda>)r   r   r5   r8   r(   r   r)   test_net_connections   s    z+TestProcessObjectLeaks.test_net_connectionsc                 C   s   |  | jj d S r   )r5   r6   environr8   r(   r(   r)   test_environ  s    z#TestProcessObjectLeaks.test_environc                 C   s   |  dd  d S )Nc                   S   s   t t S r   )rS   	proc_inforT   rU   r(   r(   r(   r)   rG     rH   z7TestProcessObjectLeaks.test_proc_info.<locals>.<lambda>r   r8   r(   r(   r)   test_proc_info  s    z%TestProcessObjectLeaks.test_proc_infoN):__name__
__module____qualname____doc__rI   r6   r4   r.   r9   r:   r<   r=   r   markskipifr   r@   rA   rC   rE   rJ   r   rM   rY   r   rZ   r]   r^   r   r   r_   r   ra   rb   rc   re   rg   r   rh   ri   rj   rk   ro   rq   r   rs   rx   r{   r   r|   r   r   r   r   r   r
   r   r   r(   r(   r(   r)   r/   V   s   






	











r/   c                       sp   e Zd ZdZe fddZe fddZdd Zerhdd	 Z	d
d Z
dd Zdd Zdd Zdd Z  ZS )TestTerminatedProcessLeakszRepeat 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    t | _t| jj| _| j  | j	  d S r   )
super
setUpClassr   subprP   r[   pidr6   killwaitclsr   r(   r)   r     s
    

z%TestTerminatedProcessLeaks.setUpClassc                    s   t    t| j d S r   )r   tearDownClassr   r   r   r   r(   r)   r     s    
z(TestTerminatedProcessLeaks.tearDownClassc                 C   s&   z
|  W n t jk
r    Y nX d S r   )rP   ZNoSuchProcessrX   r(   r(   r)   call  s    
zTestTerminatedProcessLeaks.callc                 C   s   |  | jj d S r   )r5   r6   r   r8   r(   r(   r)   	test_kill'  s    z$TestTerminatedProcessLeaks.test_killc                 C   s   |  | jj d S r   )r5   r6   r   r8   r(   r(   r)   test_terminate*  s    z)TestTerminatedProcessLeaks.test_terminatec                 C   s   |  | jj d S r   )r5   r6   Zsuspendr8   r(   r(   r)   test_suspend-  s    z'TestTerminatedProcessLeaks.test_suspendc                 C   s   |  | jj d S r   )r5   r6   rm   r8   r(   r(   r)   ro   0  s    z&TestTerminatedProcessLeaks.test_resumec                 C   s   |  | jj d S r   )r5   r6   r   r8   r(   r(   r)   	test_wait3  s    z$TestTerminatedProcessLeaks.test_waitc                    s    fdd}  | d S )Nc                      s*   zt  jjW S  tk
r$   Y nX d S r   )rS   r   r6   r   ProcessLookupErrorr(   r8   r(   r)   r   8  s    z7TestTerminatedProcessLeaks.test_proc_info.<locals>.callr   )r"   r   r(   r8   r)   r   6  s    z)TestTerminatedProcessLeaks.test_proc_info)r   r   r   r   classmethodr   r   r   r   r   r   r   ro   r   r   __classcell__r(   r(   r   r)   r     s   r   r`   r>   c                   @   s   e Zd Zdd Zdd ZdS )TestProcessDualImplementationc                 C   s   |  dd  d S )Nc                   S   s   t jt ddS )NTZuse_pebrS   Zproc_cmdlinerT   rU   r(   r(   r(   r)   rG   D  rH   zETestProcessDualImplementation.test_cmdline_peb_true.<locals>.<lambda>r   r8   r(   r(   r)   test_cmdline_peb_trueC  s    z3TestProcessDualImplementation.test_cmdline_peb_truec                 C   s   |  dd  d S )Nc                   S   s   t jt ddS )NFr   r   r(   r(   r(   r)   rG   G  rH   zFTestProcessDualImplementation.test_cmdline_peb_false.<locals>.<lambda>r   r8   r(   r(   r)   test_cmdline_peb_falseF  s    z4TestProcessDualImplementation.test_cmdline_peb_falseN)r   r   r   r   r   r(   r(   r(   r)   r   A  s   r   c                   @   s.  e Zd ZdZdd Ze dd Ze dd Ze dd	 Ze d
d Z	e dd Z
e ejjeore dkddejje dddd Zejje dddd Zdd Zejjedddd Zdd Zdd Zd d! Zejjeoejd" d#de d$d% Ze d&d' Z e ejje! ddd(d) Z"e ejjeo\e# d*kd+dd,d- Z$d.d/ Z%d0d1 Z&e ejje' ddd2d3 Z(e ejje) ddd4d5 Z*e ejje+ ddd6d7 Z,e d8d9 Z-d:d; Z.d<d= Z/er*d>d? Z0d@dA Z1dBdC Z2dDdE Z3dFdG Z4dHS )ITestModuleFunctionsLeaksz&Test leaks of psutil module functions.c                 C   s   t  }|| |j d S r   )r   r0   r   r2   r(   r(   r)   r4   R  s    z&TestModuleFunctionsLeaks.test_coveragec                 C   s   |  dd  d S )Nc                   S   s   t jddS )NTZlogicalrP   	cpu_countr(   r(   r(   r)   rG   Z  rH   z9TestModuleFunctionsLeaks.test_cpu_count.<locals>.<lambda>r   r8   r(   r(   r)   test_cpu_countX  s    z'TestModuleFunctionsLeaks.test_cpu_countc                 C   s   |  dd  d S )Nc                   S   s   t jddS )NFr   r   r(   r(   r(   r)   rG   ^  rH   z?TestModuleFunctionsLeaks.test_cpu_count_cores.<locals>.<lambda>r   r8   r(   r(   r)   test_cpu_count_cores\  s    z-TestModuleFunctionsLeaks.test_cpu_count_coresc                 C   s   |  tj d S r   )r5   rP   rf   r8   r(   r(   r)   rg   `  s    z'TestModuleFunctionsLeaks.test_cpu_timesc                 C   s   |  dd  d S )Nc                   S   s   t jddS )NT)Zpercpu)rP   rf   r(   r(   r(   r)   rG   f  rH   z=TestModuleFunctionsLeaks.test_per_cpu_times.<locals>.<lambda>r   r8   r(   r(   r)   test_per_cpu_timesd  s    z+TestModuleFunctionsLeaks.test_per_cpu_timesc                 C   s   |  tj d S r   )r5   rP   Z	cpu_statsr8   r(   r(   r)   test_cpu_statsh  s    z'TestModuleFunctionsLeaks.test_cpu_statsarm64zskipped due to #1892r>   rK   c                 C   s   |  tj d S r   )r5   rP   Zcpu_freqr8   r(   r(   r)   test_cpu_freql  s    z&TestModuleFunctionsLeaks.test_cpu_freqr`   c                 C   s   t   | t j d S r   )rP   
getloadavgr5   r8   r(   r(   r)   test_getloadavgu  s    z(TestModuleFunctionsLeaks.test_getloadavgc                 C   s   |  tj d S r   )r5   rP   Zvirtual_memoryr8   r(   r(   r)   test_virtual_memory|  s    z,TestModuleFunctionsLeaks.test_virtual_memoryz&worthless on SUNOS (uses a subprocess)c                 C   s   |  tj d S r   )r5   rP   Zswap_memoryr8   r(   r(   r)   test_swap_memory  s    z)TestModuleFunctionsLeaks.test_swap_memoryc                 C   s$   t rtn| j}| jdd |d d S )Nc                   S   s   t t S r   )rP   Z
pid_existsrT   rU   r(   r(   r(   r)   rG     rH   z:TestModuleFunctionsLeaks.test_pid_exists.<locals>.<lambda>rl   r   r!   r    r5   rn   r(   r(   r)   test_pid_exists  s    z(TestModuleFunctionsLeaks.test_pid_existsc                 C   s$   t rtn| j}| jdd |d d S )Nc                   S   s
   t dS )N.)rP   
disk_usager(   r(   r(   r)   rG     rH   z:TestModuleFunctionsLeaks.test_disk_usage.<locals>.<lambda>rl   r   rn   r(   r(   r)   test_disk_usage  s    z(TestModuleFunctionsLeaks.test_disk_usagec                 C   s   |  tj d S r   )r5   rP   Zdisk_partitionsr8   r(   r(   r)   test_disk_partitions  s    z-TestModuleFunctionsLeaks.test_disk_partitionsz/proc/diskstatsz3/proc/diskstats not available on this Linux versionc                 C   s   |  dd  d S )Nc                   S   s   t jddS NF)Znowrap)rP   Zdisk_io_countersr(   r(   r(   r)   rG     rH   z@TestModuleFunctionsLeaks.test_disk_io_counters.<locals>.<lambda>r   r8   r(   r(   r)   test_disk_io_counters  s    z.TestModuleFunctionsLeaks.test_disk_io_countersc                 C   s   |  tj d S r   )r5   rP   Zpidsr8   r(   r(   r)   	test_pids  s    z"TestModuleFunctionsLeaks.test_pidsc                 C   s   |  dd  d S )Nc                   S   s   t jddS r   )rP   Znet_io_countersr(   r(   r(   r)   rG     rH   z?TestModuleFunctionsLeaks.test_net_io_counters.<locals>.<lambda>r   r8   r(   r(   r)   test_net_io_counters  s    z-TestModuleFunctionsLeaks.test_net_io_countersr   zneed root accessc              	   C   s0   t jdd t  | dd  W 5 Q R X d S )Nr   r   c                   S   s   t jddS )Nr   r   )rP   r   r(   r(   r(   r)   rG     rH   z?TestModuleFunctionsLeaks.test_net_connections.<locals>.<lambda>)rP   r   r   r5   r8   r(   r(   r)   r     s    z-TestModuleFunctionsLeaks.test_net_connectionsc                 C   s"   t rdn| j}| jtj|d d S )Ni @ )	tolerance)r   r   r5   rP   Znet_if_addrs)r"   r   r(   r(   r)   test_net_if_addrs  s    z*TestModuleFunctionsLeaks.test_net_if_addrsc                 C   s   |  tj d S r   )r5   rP   Znet_if_statsr8   r(   r(   r)   test_net_if_stats  s    z*TestModuleFunctionsLeaks.test_net_if_statsc                 C   s   |  tj d S r   )r5   rP   Zsensors_batteryr8   r(   r(   r)   test_sensors_battery  s    z-TestModuleFunctionsLeaks.test_sensors_batteryc                 C   s   |  tj d S r   )r5   rP   Zsensors_temperaturesr8   r(   r(   r)   test_sensors_temperatures  s    z2TestModuleFunctionsLeaks.test_sensors_temperaturesc                 C   s   |  tj d S r   )r5   rP   Zsensors_fansr8   r(   r(   r)   test_sensors_fans  s    z*TestModuleFunctionsLeaks.test_sensors_fansc                 C   s   |  tj d S r   )r5   rP   Z	boot_timer8   r(   r(   r)   test_boot_time  s    z'TestModuleFunctionsLeaks.test_boot_timec                 C   s   |  tj d S r   )r5   rP   Zusersr8   r(   r(   r)   
test_users  s    z#TestModuleFunctionsLeaks.test_usersc                 C   s   |  dd  d S )Nc                   S   s
   t dS )NF)rP   Z
_set_debugr(   r(   r(   r)   rG     rH   z9TestModuleFunctionsLeaks.test_set_debug.<locals>.<lambda>r   r8   r(   r(   r)   test_set_debug  s    z'TestModuleFunctionsLeaks.test_set_debugc                 C   s   |  tj d S r   )r5   rS   Zwinservice_enumerater8   r(   r(   r)   test_win_service_iter  s    z.TestModuleFunctionsLeaks.test_win_service_iterc                 C   s   d S r   r(   r8   r(   r(   r)   test_win_service_get  s    z-TestModuleFunctionsLeaks.test_win_service_getc                    s&   t t   |  fdd d S )Nc                      s
   t  S r   )rS   Zwinservice_query_configr(   r7   r(   r)   rG     rH   zFTestModuleFunctionsLeaks.test_win_service_get_config.<locals>.<lambda>nextrP   Zwin_service_iterr7   r5   r8   r(   r   r)   test_win_service_get_config  s    z4TestModuleFunctionsLeaks.test_win_service_get_configc                    s&   t t   |  fdd d S )Nc                      s
   t  S r   )rS   Zwinservice_query_statusr(   r   r(   r)   rG     rH   zFTestModuleFunctionsLeaks.test_win_service_get_status.<locals>.<lambda>r   r8   r(   r   r)   test_win_service_get_status  s    z4TestModuleFunctionsLeaks.test_win_service_get_statusc                    s&   t t   |  fdd d S )Nc                      s
   t  S r   )rS   Zwinservice_query_descrr(   r   r(   r)   rG     rH   zKTestModuleFunctionsLeaks.test_win_service_get_description.<locals>.<lambda>r   r8   r(   r   r)    test_win_service_get_description  s    z9TestModuleFunctionsLeaks.test_win_service_get_descriptionN)5r   r   r   r   r4   r.   r   r   rg   r   r   r   r   r   r   platformmachiner	   r   r   r   r   r   r   r   r   r   r   rT   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   O  s|   




 



r   ).r   r+   rT   r   rP   Zpsutil._commonr   r   r   r   r   r   Zpsutil.testsr   r	   r
   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   Z_psplatformrS   r[   rI   r!   r.   r/   r   r   r   r   r   r(   r(   r(   r)   <module>   sT    66