⊑Wy 6 d Z ddlZddlZddlZddlZddlmZ ddlmZ ddlmZ ddlm Z ddlm
Z
ddl Z ddl m Z dd l m
Z
dd
l mZ dd l mZ dd l mZ dd
l mZ ddl mZ ddl mZ 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& dd lm'Z' e j( Z) e*ed! e+ Z,e# G d" d#e Z- G d$ d%e- Z.e# G d& d'e- Z/e# G d( d)e- Z0 G d* d+e- Z1 ej2 ed, G d- d.e- Z3 G d/ d0e Z4e5d1k rdd2l6m7Z7 e7e8 dS dS )3z;Tests for net_connections() and Process.connections() APIs. N)closing)AF_INET)AF_INET6)
SOCK_DGRAM) SOCK_STREAM)FREEBSD)LINUX)MACOS)NETBSD)OPENBSD)POSIX)SUNOS)WINDOWS)
supports_ipv6)PY3)AF_UNIX)HAS_CONNECTIONS_UNIX) SKIP_SYSCONS)PsutilTestCase) bind_socket)bind_unix_socket)check_connection_ntuple)create_sockets)
reap_children)retry_on_failure) serialrun)skip_on_access_denied)tcp_socketpair)unix_socketpair)
wait_for_fileSOCK_SEQPACKETc " e Zd Zd Zd ZddZdS )ConnectionTestCasec x t s.t s)t d }|rJ | d S d S d S Nallkind)r r thisproc connectionsselfconss q/builddir/build/BUILD/cloudlinux-venv-1.0.10/venv/lib64/python3.11/site-packages/psutil/tests/test_connections.pysetUpzConnectionTestCase.setUp6 sV "' "''U'33D ! !T ! !8 " " " "
" ! c x t s.t s)t d }|rJ | d S d S d S r% )r r r) r* r+ s r. tearDownzConnectionTestCase.tearDown< sX "6 " ''U'33D ! !T ! !8 " " " "
" !r0 r& c
t j | }n # t j $ r t rY dS w xY wfd|D }| | | || dS )zGiven a process PID and its list of connections compare
those against system-wide connections retrieved via
psutil.net_connections.
r' Nc : g | ]}|j k
|d d S )Npid).0cr7 s r.
zBConnectionTestCase.compare_procsys_connections..R s( ===q AcrcF r0 )psutilnet_connections AccessDeniedr
sort assertEqual)r, r7 proc_consr( sys_conss ` r. compare_procsys_connectionsz.ConnectionTestCase.compare_procsys_connectionsC s
-4888HH"
>===H===
H-----s 44N)r& )__name__
__module__ __qualname__r/ r2 rB r0 r. r# r# 3 sF " " " " " ". . . . . .r0 r# c N e Zd Z ej ed d Zd Zd ZdS )TestBasicOperations
requires rootc t 5 t j d D ]}t | d d d d S # 1 swxY w Y d S r% )r r; r< r r, conns r. test_systemz TestBasicOperations.test_systemZ s
. . .E:::
.
.'----
. . . . . . . . . . . . . . . . . . .s (AA A c t 5 t j d D ]}t | d d d d S # 1 swxY w Y d S r% )r r; Processr* r rK s r. test_processz TestBasicOperations.test_process` s
. . ((44%4@@
.
.'----
. . . . . . . . . . . . . . . . . . .s :AA Ac | t t j d | t t j d d S )Nz???r' ) assertRaises
ValueErrorr) r* r; r< )r, s r. test_invalid_kindz%TestBasicOperations.test_invalid_kinde sD *h&:GGG *f&<5IIIIIr0 N) rC rD rE unittestskipIfr rM rP rT rF r0 r. rH rH X s_ X_\?33. . 43.
. . .
J J J J Jr0 rH c " e Zd ZdZd Zd Zd Z ej e d d Z
d Z ej e d d Z ej e
d d
Z ej e
d d Zd S )
TestUnconnectedSocketsz;Tests sockets which are open but not connected to anything.c t d }t d |D }t st r|| S | t | d |d j dk rE| || j | |d S )Nr& r' c g | ] }|j |f S rF )fdr8 r9 s r. r: z=TestUnconnectedSockets.get_conn_from_sock..p s ---1adAY---r0 r r5 ) r) r* dictr r filenor? lenr[ )r, sockr- smaps r. get_conn_from_sockz)TestUnconnectedSockets.get_conn_from_sockn s ###//-----.. W
& & SYY * * *AwzR dkkmm!4!7 GGG7Nr0 c 0 | | }t | |j dk r-| |j | | |j |j | |j | t j t j
| }|s0t r)t |t r| }|j t k r
|dd }| |j | |j t$ k rKt& rDt( d }| t/ j |d |S )zGiven a socket, makes sure it matches the one obtained
via psutil. It assumes this process created one connection
only (the one supposed to be checked).
r5 N r& r' )rc r r[ r? r_ familytype
getsockoptsocket
SOL_SOCKETSO_TYPE getsocknamer
isinstancebytesdecoder laddrr r r) r* rB osgetpid)r, ra rL rp r- s r. check_socketz#TestUnconnectedSockets.check_socket{ sd
&&t,, %%% 7b== TWdkkmm 4 4 4 dk222 Itv'8&.II K K K "" # #E5!9!9 #LLNNE ;( " ""1"IE U+++ ;' ! !&: !''U'33D , ,RY[[$U , K K K r0 c d}t t t t | 5 }| | }|j rJ | |j t j d d d d S # 1 swxY w Y d S N 127.0.0.1r addr)
r r r r rs raddrr? statusr; CONN_LISTENr, ry ra rL s r. test_tcp_v4z"TestUnconnectedSockets.test_tcp_v4 s
[+DAAA
B
B >d$$T**Dz ! !> T[&*< = = = > > > > > > > > > > > > > > > > > > AA==BBzIPv6 not supportedc d}t t t t | 5 }| | }|j rJ | |j t j d d d d S # 1 swxY w Y d S N)::1r rx )
r r r r rs rz r? r{ r; r| r} s r. test_tcp_v6z"TestUnconnectedSockets.test_tcp_v6 s
[;TBBB
C
C >t$$T**Dz ! !> T[&*< = = = > > > > > > > > > > > > > > > > > >r c d}t t t t | 5 }| | }|j rJ | |j t j d d d d S # 1 swxY w Y d S ru )
r r r r rs rz r? r{ r; CONN_NONEr} s r. test_udp_v4z"TestUnconnectedSockets.test_udp_v4 s
[*4@@@
A
A T[&*: ; ; ; < < < < < < < < < < < < < < < < < T[&*: ; ; ; < < < < < < < < < < < < < < < < < T[&*: ; ; ; < < < < < < < < < < < < < < < < < < AB B
B
c , | }t t |t 5 }| | }|j rJ | |j t j d d d d S # 1 swxY w Y d S r r r s r.
test_unix_udpz$TestUnconnectedSockets.test_unix_udp r r N)rC rD rE __doc__rc rs r~ rU rV r r r r r
r r rF r0 r. rX rX j s EE >> > > X_ (*>??> > @?>< < < X_ (*>??< < @?< X_Y --< < .-< X_Y --< < .-< < . s$ EEEa17n+D+D+D+D+Dr0 re msg )r r r) r* rp rz r r r? r` r r r r )r, r r r r- s r. test_unixz TestConnectedSocket.test_unix s ""(00 ''V'44D Q
8$q'- 8 88 Q
8$q'- 8 88
F
F FE4EEE SYYt 4 4 4
I
I5
IG
I a 333 a 333 a )G$q'-HHHH a !?$q'-HHH LLNNN LLNNNNN
LLNNN LLNNNNs E
F *G N)
rC rD rE r rU rV r r r
r rF r0 r. r r su X_U 122 32& X_Y -- .- r0 r c D e Zd Zd Z ee d Zd ZdS ) TestFiltersc fd}t 5 |dt t t gt t
t g |dt t gt t
g |dt gt t
g |dt t gt g |dt gt g |dt gt g |dt t gt
g |d t gt
g |d
t gt
g t r% |d t gt t
t g d d d d S # 1 swxY w Y d S ) Nc ^ t | D ]8} |j | |j | 9t
sNt
j | D ]:} |j | |j | 9d S d S )Nr' )r) r* assertInrf rg r r; r< )r( familiestypesrL r, s r. checkz'TestFilters.test_filters..check s ,,$,77
0
0
dk8444
di////
4 " 2 = = = 4 4DMM$+x888MM$)U3333
4
44 4r0 r& inetinet4tcpr tcp6udpudp4udp6r )r r r r r r r! r )r, r s ` r. test_filterszTestFilters.test_filters s 4 4 4 4 4
A A E%Hg.
N;
=
=
=
E&H%
+
-
-
-
E')
+
-
-
-
E%H% -
!
!
!
E&) -
!
!
!
E&* -
!
!
!
E%H% ,
E&) ,
E&* ,
$
Af i"J?A A A; A A A A A A A A A A A A A A A A A As EE&&E*-E*)only_ifc t fd}t j d }t j d }t j t j }| t t d| }| t t d| }| t t d| }| t t d| } | } t t |d }
| } t t |d } t rg | }
t t |d } | }t t |d }nd }
d }d }d }t D ].}| } t) | d
|D ]}|j | j k r' |||t t, |
d t. j d 9|j | j k r' |||t t2 | d t. j d
p|j t7 |
dd k r' |||t t, |d t. j d |j t7 |dd k r& |||t t2 |d t. j d 0d S )Nc d}t | |j | |j | |j | |j | |j | |D ]-} | | }
| |v r|
sJ !|
r
J |
.t r | j
|g d S d S )N)
r& r r inet6r r r r r r r' ) r r? rf rg rp rz r{ r* r rB r7 ) procrL rf rg rp rz r{ kinds all_kindsr( r- r, s r.
check_connz+TestFilters.test_combos..check_conn2 s 8I #D ) ) ) T[& 1 1 1 TY - - - TZ / / / TZ / / / T[& 1 1 1!
*
*''T'225== KK4K #))T))8) $
C00D6BBBBB
C
Cr0 a
import socket, time
s = socket.socket({family}, socket.SOCK_STREAM)
s.bind(('{addr}', 0))
s.listen(5)
with open('{testfn}', 'w') as f:
f.write(str(s.getsockname()[:2]))
time.sleep(60)
a
import socket, time
s = socket.socket({family}, socket.SOCK_DGRAM)
s.bind(('{addr}', 0))
with open('{testfn}', 'w') as f:
f.write(str(s.getsockname()[:2]))
time.sleep(60)
)dirrw )rf ry r r T)deleter] rF )r& r r r r )r& r r r r r7 )r& r r r r )r& r r r r ) r textwrapdedentrq pathbasenamer getcwdformatintr r pyrunevalr r r) childrenr* r? r` r7 r r; r| r r getattr)r, r tcp_template udp_templatetestfile
tcp4_template
udp4_template
tcp6_template
udp6_template tcp4_proc tcp4_addr udp4_proc udp4_addr tcp6_proc tcp6_addr udp6_proc udp6_addrpr- rL s ` r. test_comboszTestFilters.test_combos. sw C C C C C* ( ( 7##DOO O$D$DEE$++w<> JJ}-- x===>> ??
=11I ]8D A A ABBI
=11I ]8D A A ABBII I I I I""$$ H HA==??D SYY * * *
H
H5IM)) Jq$i % 1 FH H H H Uim++ Jq$Y % / FH H H H Ugi = === Jq$+y" % 1 FH H H H Ugi = === Jq$*i % / FH H H%
H H Hr0 c
t 5 t d }| t | t rdnd |D ]I}| |j t t f | |j
t Jt d }| t | d | |d j t | |d j
t t rt d }| t | d | |d j t | |d j
t t d }| t | t rdnd |D ]I}| |j t t f | |j
t Jt d }| t | d | |d j t | |d j
t t rt d
}| t | d | |d j t | |d j
t t d }| t | t rd nd |D ]P}| |j t t f | |j
t t f Qt rt d
}| t | d |D ]I}| |j t | |j
t t f Jt rt st st d }| t | d |D ]I}| |j t | |j
t t f Jd d d d S # 1 swxY w Y d S )Nr r' re r] r r r r r r r r r )r r) r* r? r` r r rf r r rg r r r r r r )r, r- rL s r.
test_countzTestFilters.test_count s
7 H 7 H''U'33D SYY]__(C! D D D
9
9
dkGX+>??? K8888''V'44D SYY * * * T!W^W 5 5 5 T!W\; 7 7 7
< +++88 TA... a ::: a {;;;''U'33D SYY]__(C! D D D
8
8
dkGX+>??? J7777''V'44D SYY * * * T!W^W 5 5 5 T!W\: 6 6 6
; +++88 TA... a ::: a z:::''V'44D SYY]__(C! D D D
D
D
dkGX+>???
di+z)BCCCC
H +++99 TA... H HD$$T[(;;;MM$)k:-FGGGG $
HW
H
H +++88 TA... H HD$$T[':::MM$)k:-FGGGGo7 H 7 H 7 H 7 H 7 H 7 H 7 H 7 H 7 H 7 H 7 H 7 H 7 H 7 H 7 H 7 H 7 H 7 Hs UU--U14U1N)rC rD rE r r r
r r rF r0 r. r r sl )A )A )AV 5)))^H ^H *)^H@8H 8H 8H 8H 8Hr0 r rI c > e Zd ZdZd Z e d ZdS )TestSystemWideConnectionsz Tests for net_connections().c
fd}t 5 ddlm} | D ]v\ }}|dk rt s|\ }}t j | } t | t t | |||| w d d d d S # 1 swxY w Y d S )Nc | D ][} |j || |j t k r |j || t | \d S )Nr )r rf r rg r )r- r types_rL r, s r. r z0TestSystemWideConnections.test_it..check sn
.
.
dk8
>>>;'))MM$)VM>>>'----
.
.r0 r ) conn_tmapr )
r psutil._commonr itemsr r; r< r? r` set)r, r r r( groupsr r r- s ` r. test_itz!TestSystemWideConnections.test_it s, . . . . .
. . 0 0 0 0 0 0 ) 1 1
.
. f6>>*>> #) & -d33 TCD NN;;;dHf----
. . . . . . . . . . . . . . . . . . .s BB55B9<B9c t 5 }t | }d d d n # 1 swxY w Y g d}g }t | D ]q}| }| | t j d|z }| | } |j r|D ]}t | fdt j d D } D ]{ | t fd| D | t j
}
| t |
d | |d S )N
a import time, os
from psutil.tests import create_sockets
with create_sockets():
with open(r'%s', 'w') as f:
f.write("hello")
time.sleep(60)
c & g | ]
}|j v |S rF r6 )r8 xpidss r. r: zFTestSystemWideConnections.test_multi_sockets_procs.. s; AEE E1 E E Er0 r& r' c * g | ]}|j k
|S rF r6 )r8 r r7 s r. r: zFTestSystemWideConnections.test_multi_sockets_procs.. s !E!E!E ! r0 )r r` ranger appendr r r r7 r r; r< r? rO r* )
r, socksexpectedtimesfnames_fnamesrcsprocsysconsr r7 r s
@@r. test_multi_sockets_procsz2TestSystemWideConnections.test_multi_sockets_procs s
"5zzH " " " " " " " " " " " " " " "u # #AOO%%E MM% / # # C JJsOOE KK " " " " ! !E % f 4% @ @ @ B BC S!E!E!E!EW!E!E!E F F %
'
'
's##A S u!5!5 6 6 A A A A B Bs -11N)rC rD rE r r r r rF r0 r. r r sS &&. . .& $B $B $B $B $Br0 r c e Zd Zd ZdS )TestMiscc g }g }t t D ]}| d rt t | }t | }| s
J | | t | | || | | | | t rt j t j
t rt j d S d S )NCONN_)
r r;
startswithr strisupper assertNotInr r CONN_IDLE
CONN_BOUNDr CONN_DELETE_TCB)r, intsstrsnamenumstr_s r. test_connection_constantsz"TestMisc.test_connection_constants s KK " "Dw''
" fd++3xx||~~++t++~ d+++ d+++ C D!!! # " " " " # #r0 N)rC rD rE r
rF r0 r. r r s# # # # # #r0 r __main__)
run_from_name)9r rq ri r rU
contextlibr r r r r r; r r r
r r r
r r r r psutil._compatr psutil.testsr r r r r r r r r r r r r r r rO r) r objectr! r# rH rX r r rV r r rC psutil.tests.runnerr __file__rF r0 r. r sV B A
( ( ( ( ( ( - - - - - - % % % % % % ' ' ' ' ' ' $ $ $ $ $ $ ) ) ) ) ) ) 0 0 0 0 0 0 ' ' ' ' ' ' & & & & & & ) ) ) ) ) ) " " " " " " . . . . . . ' ' ' ' ' ' ( ( ( ( ( ( & & & & & & 6> !16688<< !. !. !. !. !. !. !. !.HJ J J J J, J J J$ [< [< [< [< [< / [< [< [<| 4 4 4 4 4, 4 4 4nFH FH FH FH FH$ FH FH FHR //;B ;B ;B ;B ;B 2 ;B ;B 0/;B|# # # # #~ # # #* z111111M( r0