O؇HN6j d Z ddlmZ ddlZddlZddlZddlZddlZddlZddlm Z ddlm
Z
ddlm Z ddlm Z ddlm
Z
dd lmZ dd
lmZ dd lmZ dd lmZ dd
lmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlm Z ddlm Z ddlm Z ddlm Z ddlm Z ddlm!Z! ddlm"Z" dd lm#Z# dd lm$Z$ dd lm%Z% dd lm&Z& ej' j( Z( ej) Z*d Z+d! Z, G d" d#e Z- G d$ d%e- Z. ej/ e d& G d' d(e Z0 G d) d*e Z1e2d+k rdd,l3m4Z4 e4e5 dS 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)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)skip_on_access_denied)spawn_testproc)system_namespace) terminate c d } | S )zsDecorator for those Linux functions which are implemented in pure
Python, and which we want to run faster.
c F t j fd }|S )Nc t rG| j j } t | j _ | g|R i ||| j _ S # || j _ w xY w | g|R i |S N)r __class__times FEW_TIMES)selfargskwargsbeforefuns n/builddir/build/BUILD/cloudlinux-venv-1.0.10/venv/lib64/python3.11/site-packages/psutil/tests/test_memleaks.pywrapperz5fewtimes_if_linux..decorator..wrapperC s
2 -2+4DN( 3t5d555f55+1DN((6DN(1111s41$111&111s ? A
) functoolswraps)r, r. s ` r- decoratorz$fewtimes_if_linux..decoratorB s8 2 2 2 2
2 )r1 s r- fewtimes_if_linuxr4 > s r2 c e Zd ZdZeZd Z e d Z e d Z e d Z
e d Z e j
e d e d Z e j
e d e d Z e d
Zd Zd Z e j
e d
d Z e j
e d
d Z e j
e d
e d Z e j
ed d Z e d Z e ee d Z e j
e d d Z e j
e d e d Z e d Z! e ee d Z" e d Z# e e j
e$ d
d Z% e d Z& e d Z' e j
e d e d Z(d Z) e d! Z* e j
e+ d
d" Z, e j
e+ d
d# Z- e d$ Z. e j
e/ d
e d% Z0 e j
e1 d& e j
e2 d
d' Z3 e j
e1 d& e j
e2 d
d( Z4 e e j
e d) d* Z5 e j
e6 d
d+ Z7 e j
e d d, Z8d-S ).TestProcessObjectLeaksz$Test leaks of Process class methods.c j t d }| | |j |j z d S r$ )r test_class_coveragegetterssettersr( nss r-
test_coveragez$TestProcessObjectLeaks.test_coverage\ s4
t
$
$
tRZ"*%<=====r2 c D | | j j d S r$ )executeprocnamer( s r- test_namez TestProcessObjectLeaks.test_name` TY^$$$$$r2 c D | | j j d S r$ )r? r@ cmdlinerB s r- test_cmdlinez#TestProcessObjectLeaks.test_cmdlined TY&'''''r2 c D | | j j d S r$ )r? r@ exerB s r- test_exez TestProcessObjectLeaks.test_exeh TY]#####r2 c D | | j j d S r$ )r? r@ ppidrB s r- test_ppidz TestProcessObjectLeaks.test_ppidl rD r2 z
POSIX onlyc D | | j j d S r$ )r? r@ uidsrB s r- test_uidsz TestProcessObjectLeaks.test_uidsp
TY^$$$$$r2 c D | | j j d S r$ )r? r@ gidsrB s r- test_gidsz TestProcessObjectLeaks.test_gidsu rS r2 c D | | j j d S r$ )r? r@ statusrB s r- test_statusz"TestProcessObjectLeaks.test_statusz TY%&&&&&r2 c D | | j j d S r$ )r? r@ nicerB s r- test_nicez TestProcessObjectLeaks.test_nice~ s TY^$$$$$r2 c n t fd d S )Nc 8 j S r$ )r@ r\ )nicenessr( s r- z6TestProcessObjectLeaks.test_nice_set.. s TY^^H 5 5 r2 )thisprocr\ r? )r( r` s `@r-
test_nice_setz$TestProcessObjectLeaks.test_nice_set s6 ==?? 5555566666r2
not supportedc D | | j j d S r$ )r? r@ ionicerB s r- test_ionicez"TestProcessObjectLeaks.test_ionice rZ r2 c J t r4t fd d S fd t j t j t j dd }
t | d S )Nc 8 j S r$ )r@ rf )r( values r- ra z8TestProcessObjectLeaks.test_ionice_set.. s !1!1%!8!8 r2 c L j t j S r$ )r@ rf psutilIOPRIO_CLASS_NONErB s r- ra z8TestProcessObjectLeaks.test_ionice_set.. s !1!1&2J!K!K r2 r ) r rb rf r? r/ partialcextproc_ioprio_setosgetpid
execute_w_excOSError)r( r, rj s ` @r- test_ionice_setz&TestProcessObjectLeaks.test_ionice_set s - OO%%E LL88888 9 9 9 9 9 LLKKKK L L L#D$8")++r1MMC w , , , , ,r2 c D | | j j d S r$ )r? r@ io_countersrB s r- test_io_countersz'TestProcessObjectLeaks.test_io_counters !
TY*+++++r2 zworthless on POSIXc t j | | j j d S r$ )rl Processusernamer? r@ rB s r-
test_usernamez$TestProcessObjectLeaks.test_username s; !!### TY'(((((r2 c D | | j j d S r$ )r? r@ create_timerB s r- test_create_timez'TestProcessObjectLeaks.test_create_time TY*+++++r2 )only_ifc D | | j j d S r$ )r? r@ num_threadsrB s r- test_num_threadsz'TestProcessObjectLeaks.test_num_threads rz r2 WINDOWS onlyc D | | j j d S r$ )r? r@ num_handlesrB s r- test_num_handlesz'TestProcessObjectLeaks.test_num_handles r r2 c D | | j j d S r$ )r? r@ num_fdsrB s r- test_num_fdsz#TestProcessObjectLeaks.test_num_fds !
TY&'''''r2 c D | | j j d S r$ )r? r@ num_ctx_switchesrB s r- test_num_ctx_switchesz,TestProcessObjectLeaks.test_num_ctx_switches TY/00000r2 c D | | j j d S r$ )r? r@ threadsrB s r- test_threadsz#TestProcessObjectLeaks.test_threads r r2 c D | | j j d S r$ )r? r@ cpu_timesrB s r- test_cpu_timesz%TestProcessObjectLeaks.test_cpu_times s TY()))))r2 c D | | j j d S r$ )r? r@ cpu_numrB s r- test_cpu_numz#TestProcessObjectLeaks.test_cpu_num r r2 c D | | j j d S r$ )r? r@ memory_inforB s r- test_memory_infoz'TestProcessObjectLeaks.test_memory_info r r2 c D | | j j d S r$ )r? r@ memory_full_inforB s r- test_memory_full_infoz,TestProcessObjectLeaks.test_memory_full_info r r2 c D | | j j d S r$ )r? r@ terminalrB s r-
test_terminalz$TestProcessObjectLeaks.test_terminal s!
TY'(((((r2 c r t rt n| j }| | j j | d S )Nr& )r r' r& r? r@ resumer( r& s r- test_resumez"TestProcessObjectLeaks.test_resume s3 "2
TY%U 33333r2 c D | | j j d S r$ )r? r@ cwdrB s r- test_cwdz TestProcessObjectLeaks.test_cwd rL r2 c D | | j j d S r$ )r? r@ cpu_affinityrB s r- test_cpu_affinityz(TestProcessObjectLeaks.test_cpu_affinity s TY+,,,,,r2 c t fd t fd d S )Nc 8 j S r$ r@ r )affinityr( s r- ra z>TestProcessObjectLeaks.test_cpu_affinity_set.. s TY 3 3H = = r2 c : j dg S Nrn r rB s r- ra z>TestProcessObjectLeaks.test_cpu_affinity_set.. s 6 6t < < r2 )rb r r? rt
ValueError)r( r s `@r- test_cpu_affinity_setz,TestProcessObjectLeaks.test_cpu_affinity_set se ((** =====>>> <<<< > > > > >r2 c t t d 5 | | j j d d d d S # 1 swxY w Y d S )Nw)openr r? r@
open_filesrB s r- test_open_filesz&TestProcessObjectLeaks.test_open_files s
*,,
$
$ / / LL- . . . / / / / / / / / / / / / / / / / / /s A
AAc D | | j j d S r$ )r? r@ memory_mapsrB s r- test_memory_mapsz'TestProcessObjectLeaks.test_memory_maps rz r2 z
LINUX onlyc 8 fd d S )Nc L j t j S r$ r@ rlimitrl
RLIMIT_NOFILErB s r- ra z4TestProcessObjectLeaks.test_rlimit.. s TY - -f.B C C r2 r? rB s `r- test_rlimitz"TestProcessObjectLeaks.test_rlimit s'
CCCCDDDDDr2 c t t j fd t t f fd d S )Nc N j t j S r$ r )limitr( s r- ra z8TestProcessObjectLeaks.test_rlimit_set.. s TY - -f.BE J J r2 c 8 j d S r )r@ r rB s r- ra z8TestProcessObjectLeaks.test_rlimit_set.. s $):J:J2:N:N r2 )rb r rl r r? rt ru r )r( r s `@r- test_rlimit_setz&TestProcessObjectLeaks.test_rlimit_set sd 455 JJJJJKKK GZ02N2N2N2NOOOOOr2 zworthless on WINDOWSc t 5 t rdnd fd d d d d S # 1 swxY w Y d S )Ninetallc 8 j S r$ )r@ connections)kindr( s r- ra z9TestProcessObjectLeaks.test_connections.. s !6!6t! > "-66D LL<<<<< = = = > > > > > > > > > > > > > > > > > >s %AA
Ac D | | j j d S r$ )r? r@ environrB s r- test_environz#TestProcessObjectLeaks.test_environ rH r2 c 2 | d d S )Nc X t t j S r$ )rp proc_inforr rs r3 r2 r- ra z7TestProcessObjectLeaks.test_proc_info..
s T^^BIKK 8 8 r2 r rB s r- test_proc_infoz%TestProcessObjectLeaks.test_proc_info s 8899999r2 N)9__name__
__module__ __qualname____doc__rb r@ r= r4 rC rG rK rO unittestskipIfr rR rV rY r] rc r rg rv r ry r~ r r r r r r r r r r r r r r r r r r r r r r r r r r r r r r r r3 r2 r- r6 r6 W s .. D> > > % % % ( ( ( $ $ $ % % % X_Y --% % .-% X_Y --% % .-% ' ' '% % %7 7 7 X_^_55' ' 65' X_^_55- - 65- X_--??, , @?, X_U 011) ) 21)
, , , 7+++, , ,+ , X_[.11, , 21, X_Y --( ( .-( 1 1 1 7+++( ( ,+ ( * * * X_))?;;( ( <; ( , , , 1 1 1 X_Y --) ) .-)4 4 4 $ $ $ X_))?;;- - <;- X_))?;;> > <;> / / / X_ (/::, , ;:, X_Y --X_^_55E E 65 .-E X_Y --X_^_55P P 65 .-P
X_W 455> > 65 > X__o66( ( 76( X_[.11: : 21: : :r2 r6 c e Zd ZdZe fd Ze fd Zd Zerd Z d Z
d Z d Z d Z
d
Z xZS xZS ) 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 t t | _ t j | j j | _ | j | j d S r$ )
r
setUpClassr subprl r| pidr@ killwaitclsr% s r- r z%TestTerminatedProcessLeaks.setUpClass s]
!##>#(,//
r2 c p t t | j d S r$ )r
tearDownClassr r r s r- r z(TestTerminatedProcessLeaks.tearDownClass s/
#(r2 c H | d S # t j $ r Y d S w xY wr$ )rl
NoSuchProcess)r( r, s r- callz TestTerminatedProcessLeaks.call! s; CEEEEE# DD s
!!c D | | j j d S r$ )r? r@ r rB s r- test_killz$TestTerminatedProcessLeaks.test_kill) LL ( ( ( ( (r2 c D | | j j d S r$ )r? r@ r rB s r- test_terminatez)TestTerminatedProcessLeaks.test_terminate, s LL, - - - - -r2 c D | | j j d S r$ )r? r@ suspendrB s r- test_suspendz'TestTerminatedProcessLeaks.test_suspend/ s LL* + + + + +r2 c D | | j j d S r$ )r? r@ r rB s r- r z&TestTerminatedProcessLeaks.test_resume2 s LL) * * * * *r2 c D | | j j d S r$ )r? r@ r rB s r- test_waitz$TestTerminatedProcessLeaks.test_wait5 r r2 c < fd} | d S )Nc p t j j S # t $ r Y d S w xY wr$ )rp r r@ r r
rB s r- r z7TestTerminatedProcessLeaks.test_proc_info..call: s@ >>$)-888) DDs #'
55r )r( r s ` r- r z)TestTerminatedProcessLeaks.test_proc_info8 s6
LL r2 )r r r r classmethodr r r r r r r r r r
__classcell__)r% s @r- r r
s [ [ ) ) ) . . . , , , + + + ) ) ) # r2 r r c e Zd Zd Zd ZdS ) TestProcessDualImplementationc 2 | d d S )Nc \ t t j d S )NTuse_pebrp proc_cmdlinerr rs r3 r2 r- ra zETestProcessDualImplementation.test_cmdline_peb_true..G s T . .ry{{D . I I r2 r rB s r- test_cmdline_peb_truez3TestProcessDualImplementation.test_cmdline_peb_trueF s IIJJJJJr2 c 2 | d d S )Nc \ t t j d S )NFr r r3 r2 r- ra zFTestProcessDualImplementation.test_cmdline_peb_false..J s T . .ry{{E . J J r2 r rB s r- test_cmdline_peb_falsez4TestProcessDualImplementation.test_cmdline_peb_falseI s JJKKKKKr2 N)r r r r r r3 r2 r- r r C s: K K KL L L L Lr2 r c e Zd ZdZd Z e d Z e d Z e d Z e d Z e d Z
e e j e
o ej dk d e j e d
d Z e j e d d
Zd Z e j ed d Zd Zd Zd Z e j eoej d d e d Z e d Z e e j e d
d Z! e e j e
o ej" dk d d Z#d Z$d Z% e e j e& d
d Z' e e j e( d
d Z) e e j e* d
d Z+ e d! Z,d" Z-d# Z.erd$ Z/d% Z0d& Z1d' Z2d( Z3d)S d)S )*TestModuleFunctionsLeaksz&Test leaks of psutil module functions.c X t }| | |j d S r$ )r r8 r r; s r- r= z&TestModuleFunctionsLeaks.test_coverageU s+
tRV,,,,,r2 c 2 | d d S )Nc , t j d S )NTlogicalrl cpu_countr3 r2 r- ra z9TestModuleFunctionsLeaks.test_cpu_count..] s V -d ; ; ; r2 r rB s r- test_cpu_countz'TestModuleFunctionsLeaks.test_cpu_count[ s ;;<<<<.a s V -e < < < r2 r rB s r- test_cpu_count_coresz-TestModuleFunctionsLeaks.test_cpu_count_cores_ s <<=====r2 c D | t j d S r$ )r? rl r rB s r- r z'TestModuleFunctionsLeaks.test_cpu_timesc V%&&&&&r2 c 2 | d d S )Nc , t j d S )NT)percpu)rl r r3 r2 r- ra z=TestModuleFunctionsLeaks.test_per_cpu_times..i s V -T : : : r2 r rB s r- test_per_cpu_timesz+TestModuleFunctionsLeaks.test_per_cpu_timesg s ::;;;;;r2 c D | t j d S r$ )r? rl cpu_statsrB s r- test_cpu_statsz'TestModuleFunctionsLeaks.test_cpu_statsk r r2 arm64zskipped due to #1892rd c D | t j d S r$ )r? rl cpu_freqrB s r-
test_cpu_freqz&TestModuleFunctionsLeaks.test_cpu_freqo s
V_%%%%%r2 r c j t j | t j d S r$ )rl
getloadavgr? rB s r- test_getloadavgz(TestModuleFunctionsLeaks.test_getloadavgw s, V&'''''r2 c D | t j d S r$ )r? rl virtual_memoryrB s r- test_virtual_memoryz,TestModuleFunctionsLeaks.test_virtual_memory~ s V*+++++r2 z&worthless on SUNOS (uses a subprocess)c D | t j d S r$ )r? rl swap_memoryrB s r- test_swap_memoryz)TestModuleFunctionsLeaks.test_swap_memory s V'(((((r2 c ` t rt n| j }| d | d S )Nc L t j t j S r$ )rl
pid_existsrr rs r3 r2 r- ra z:TestModuleFunctionsLeaks.test_pid_exists.. s V .ry{{ ; ; r2 r r r' r& r? r s r- test_pid_existsz(TestModuleFunctionsLeaks.test_pid_exists s2 "2
;;5 IIIIIr2 c ` t rt n| j }| d | d S )Nc * t j d S )N.)rl
disk_usager3 r2 r- ra z:TestModuleFunctionsLeaks.test_disk_usage.. s V .s 3 3 r2 r r0 r s r- test_disk_usagez(TestModuleFunctionsLeaks.test_disk_usage s2 "2
335 AAAAAr2 c D | t j d S r$ )r? rl disk_partitionsrB s r- test_disk_partitionsz-TestModuleFunctionsLeaks.test_disk_partitions s V+,,,,,r2 z/proc/diskstatsz3/proc/diskstats not available on this Linux versionc 2 | d d S )Nc , t j d S NF)nowrap)rl disk_io_countersr3 r2 r- ra z@TestModuleFunctionsLeaks.test_disk_io_counters.. s V 4E B B B r2 r rB s r- test_disk_io_countersz.TestModuleFunctionsLeaks.test_disk_io_counters s
BBCCCCCr2 c D | t j d S r$ )r? rl pidsrB s r- test_pidsz"TestModuleFunctionsLeaks.test_pids s V[!!!!!r2 c 2 | d d S )Nc , t j d S r< )rl net_io_countersr3 r2 r- ra z?TestModuleFunctionsLeaks.test_net_io_counters.. s V 35 A A A r2 r rB s r- test_net_io_countersz-TestModuleFunctionsLeaks.test_net_io_counters s
AABBBBBr2 r zneed root accessc t j d t 5 | d d d d d S # 1 swxY w Y d S )Nr r c , t j d S )Nr rH )rl net_connectionsr3 r2 r- ra z?TestModuleFunctionsLeaks.test_net_connections.. s !7U!C!C!C r2 )rl rJ r r? rB s r- test_net_connectionsz-TestModuleFunctionsLeaks.test_net_connections s E****
E E LLCC D D D E E E E E E E E E E E E E E E E E Es AA A c h t rdn| j }| t j | d S )Ni @ ) tolerance)r rM r? rl net_if_addrs)r( rM s r- test_net_if_addrsz*TestModuleFunctionsLeaks.test_net_if_addrs s1 !(>>>>r2 c D | t j d S r$ )r? rl net_if_statsrB s r- test_net_if_statsz*TestModuleFunctionsLeaks.test_net_if_stats s V()))))r2 c D | t j d S r$ )r? rl sensors_batteryrB s r- test_sensors_batteryz-TestModuleFunctionsLeaks.test_sensors_battery s
V+,,,,,r2 c D | t j d S r$ )r? rl sensors_temperaturesrB s r- test_sensors_temperaturesz2TestModuleFunctionsLeaks.test_sensors_temperatures s
V011111r2 c D | t j d S r$ )r? rl sensors_fansrB s r- test_sensors_fansz*TestModuleFunctionsLeaks.test_sensors_fans s
V()))))r2 c D | t j d S r$ )r? rl boot_timerB s r- test_boot_timez'TestModuleFunctionsLeaks.test_boot_time r r2 c D | t j d S r$ )r? rl usersrB s r-
test_usersz#TestModuleFunctionsLeaks.test_users s V\"""""r2 c 2 | d d S )Nc * t j d S )NF)rl
_set_debugr3 r2 r- ra z9TestModuleFunctionsLeaks.test_set_debug.. s V .u 5 5 r2 r rB s r- test_set_debugz'TestModuleFunctionsLeaks.test_set_debug s 5566666r2 c D | t j d S r$ )r? rp winservice_enumeraterB s r- test_win_service_iterz.TestModuleFunctionsLeaks.test_win_service_iter s LL2 3 3 3 3 3r2 c d S r$ r3 rB s r- test_win_service_getz-TestModuleFunctionsLeaks.test_win_service_get s Dr2 c t t j | fd d S )Nc 8 t S r$ )rp winservice_query_configrA s r- ra zFTestModuleFunctionsLeaks.test_win_service_get_config.. !=!=d!C!C r2 nextrl win_service_iterrA r? r( rA s @r- test_win_service_get_configz4TestModuleFunctionsLeaks.test_win_service_get_config G /11227799D LLCCCC D D D D Dr2 c t t j | fd d S )Nc 8 t S r$ )rp winservice_query_statusrn s r- ra zFTestModuleFunctionsLeaks.test_win_service_get_status.. ro r2 rp rs s @r- test_win_service_get_statusz4TestModuleFunctionsLeaks.test_win_service_get_status ru r2 c t t j | fd d S )Nc 8 t S r$ )rp winservice_query_descrrn s r- ra zKTestModuleFunctionsLeaks.test_win_service_get_description.. s ! > > ' ' ' < < < ' ' ' X_U< /x / 1 1W <+- -X_ %77& & 87- -
& X_[.11( ( 21( , , , X_U DEE) ) FE)J J J B B B- - - X_UD27>>2C#D#D DJL LD D L LD
" " "
X_,,o>>C C ?> C X_U/yry{{a /1CDDE E ED E ? ? ?
* * *
X_,,o>>- - ?> - X_11?CC2 2 DC 2 X_))?;;* * <; *
' ' '# # #7 7 7 D 4 4 4 E E E E E E D D D D D%D Dr2 r __main__)
run_from_name)6r
__future__r r/ rr r~ r rl psutil._commonr r r r r r psutil._compatr
r psutil.testsr r
r r r r r r r r r r r r r r r r r r _psplatformrp r| rb r' r4 r6 r r r r r psutil.tests.runnerr __file__r3 r2 r- r s & % % % % %
- - - - - - ) ) ) ) ) ) % % % % % % $ $ $ $ $ $ # # # # # # ( ( ( ( ( ( , , , , , , ) ) ) ) ) ) - - - - - - # # # # # # , , , , , , ) ) ) ) ) ) 1 1 1 1 1 1 ' ' ' ' ' ' ' ' ' ' ' ' # # # # # # * * * * * * . . . . . . ' ' ' ' ' ' ) ) ) ) ) ) " " " " " " 6>
2s: s: s: s: s:^ s: s: s:l3 3 3 3 3 !7 3 3 3 l Wn--L L L L LN L L .-L UD UD UD UD UD~ UD UD UDp z111111M( r2