P h 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' dd!lm(Z( e j) Z* e+ed" e, Z-ej. dd# d$k Z/e" G d% d&e Z0 G d' d(e0 Z1e" G d) d*e0 Z2e" G d+ d,e0 Z3 G d- d.e0 Z4 e&j5 e$d/ G d0 d1e0 Z6 G d2 d3e Z7e8d4k rdd5l9m:Z: e:e; dS dS )6z;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) bind_socket)bind_unix_socket)check_connection_ntuple)create_sockets)HAS_CONNECTIONS_UNIX)PsutilTestCase)
reap_children)retry_on_failure) serialrun)skip_on_access_denied) SKIP_SYSCONS)tcp_socketpair)unittest)unix_socketpair)
wait_for_fileSOCK_SEQPACKET ) c " 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 /builddir/build/BUILD/imunify360-venv-2.6.2/opt/imunify360/venv/lib64/python3.11/site-packages/psutil/tests/test_connections.pysetUpzConnectionTestCase.setUp8 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 r2 tearDownzConnectionTestCase.tearDown> sX "6 " ''U'33D ! !T ! !8 " " " "
" !r4 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).0cr; s r2
zBConnectionTestCase.compare_procsys_connections..T s( ===q AcrcF r4 )psutilnet_connections AccessDeniedr
sort assertEqual)r0 r; proc_consr, sys_conss ` r2 compare_procsys_connectionsz.ConnectionTestCase.compare_procsys_connectionsE s
-4888HH"
>===H===
H-----s 44N)r* )__name__
__module__ __qualname__r3 r6 rF r4 r2 r' r' 5 sF " " " " " ". . . . . .r4 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 r0 conns r2 test_systemz TestBasicOperations.test_system\ 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 rO s r2 test_processz TestBasicOperations.test_processb s
. . ((44%4@@
.
.'----
. . . . . . . . . . . . . . . . . . .s :AA Ac | t t j d | t t j d d S )Nz???r+ ) assertRaises
ValueErrorr- r. r? r@ )r0 s r2 test_invalid_kindz%TestBasicOperations.test_invalid_kindg sD *h&:GGG *f&<5IIIIIr4 N) rG rH rI r skipIfr rQ rT rX rJ r4 r2 rL rL Z s_ X_\?33. . 43.
. . .
J J J J Jr4 rL 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 rJ )fdr< r= s r2 r> z=TestUnconnectedSockets.get_conn_from_sock..r s ---1adAY---r4 r r9 ) r- r. dictr r filenorC lenr^ )r0 sockr1 smaps r2 get_conn_from_sockz)TestUnconnectedSockets.get_conn_from_sockp s ###//-----.. W
& & SYY * * *AwzR dkkmm!4!7 GGG7Nr4 c ` | | }t | |j dk r-| |j | | |j |j | |j | t j t j
| }|s0t r)t |t r| }|j t k r
|dd }|j t" k rt$ rn| |j | |j t" k rKt( rDt* d }| t1 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).
r9 Nr# r* r+ )rf r r^ rC rb familytype
getsockoptsocket
SOL_SOCKETSO_TYPE getsocknamer
isinstancebytesdecoder r r laddrr r- r. rF osgetpid)r0 rd rP rr r1 s r2 check_socketz#TestUnconnectedSockets.check_socket} sy
&&t,, %%% 7b== TWdkkmm 4 4 4 dk222 Itv'8&.II K K K "" # #E5!9!9 #LLNNE ;( " ""1"IE ;' ! !g ! TZ / / / ;' ! !&: !''U'33D , ,RY[[$U , K K K r4 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 ru raddrrC statusr? CONN_LISTENr0 r{ rd rP s r2 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 rz )
r r r r ru r| rC r} r? r~ r s r2 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 rw )
r r r r ru r| rC r} r? CONN_NONEr s r2 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 r2
test_unix_udpz$TestUnconnectedSockets.test_unix_udp r r N)rG rH rI __doc__rf ru r r rY r r r r r
r r rJ r4 r2 r[ r[ l s EE ! ! !F> > > X_ (*>??> > @?>< < < X_ (*>??< < @?< X_Y --< < .-< X_Y --< < .-< < z1TestConnectedSocket.test_unix.. s$ EEEa17n+D+D+D+D+Dr4 r# msg )r r r- r. rr r| r r rC rc r r r r )r0 r r r r1 r{ s r2 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
I a 333 a 333 a )G$q'-HHHH
I !!W]DGM !!W]DGM ; / /D$$T2..../ a !?$q'-HHH a !?$q'-HHH LLNNN LLNNNNN
LLNNN LLNNNNs GH *IN)
rG rH rI r r rY r r r
r rJ r4 r2 r r su X_U 122 32& X_Y -- .- r4 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. assertInrh ri r r? r@ )r, familiestypesrP r0 s r2 checkz'TestFilters.test_filters..check s ,,$,77
0
0
dk8444
di////
4 " 2 = = = 4 4DMM$+x888MM$)U3333
4
44 4r4 r* inetinet4tcpr tcp6udpudp4udp6r )r r r r r r r" r )r0 r s ` r2 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 rC rh ri rr r| r} r. r rF r; ) procrP rh ri rr r| r} kinds all_kindsr, r1 r0 s r2
check_connz+TestFilters.test_combos..check_conn> s 8I #D ) ) ) T[& 1 1 1 TY - - - TZ / / / TZ / / / T[& 1 1 1!
*
*''T'225== KK4K #))T))8) $
C00D6BBBBB
C
Cr4 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)
)dirry )rh r{ r r T)deleter` rJ )r* r r r r )r* r r r r r; )r* r r r r )r* r r r r ) r textwrapdedentrs pathbasenamer getcwdformatintr r pyrunevalr! r r- childrenr. rC rc r; r r? r~ r r getattr)r0 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_addrpr1 rP s ` r2 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 Hr4 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+ r# r` r r r r r r r r r r$ )r r- r. rC rc r r rh r r ri r r r r r r )r0 r1 rP s r2
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)rG rH rI r r r
r r rJ r4 r2 r r
sl )A )A )AV 5)))^H ^H *)^H@8H 8H 8H 8H 8Hr4 r rM 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 rh r ri r )r1 r types_rP r0 s r2 r z0TestSystemWideConnections.test_it..check sn
.
.
dk8
>>>;'))MM$)VM>>>'----
.
.r4 r ) conn_tmapr )
r psutil._commonr itemsr r? r@ rC rc set)r0 r r r, groupsr r r1 s ` r2 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 rJ r: )r< xpidss r2 r> zFTestSystemWideConnections.test_multi_sockets_procs.. s; AEE E1 E E Er4 r* r+ c * g | ]}|j k
|S rJ r: )r< r r; s r2 r> zFTestSystemWideConnections.test_multi_sockets_procs.. s !E!E!E ! r4 )r rc ranger appendr r r r; r! r? r@ rC rS r. )
r0 socksexpectedtimesfnamesifnamesrcsprocsysconsr r; r s
@@r2 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)rG rH rI r r r r rJ r4 r2 r r sS &&. . .& $B $B $B $B $Br4 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)r0 intsstrsnamenumstr_s r2 test_connection_constantsz"TestMisc.test_connection_constants s KK " "Dw''
" fd++3xx||~~++t++~ d+++ d+++ C D!!! # " " " " # #r4 N)rG rH rI r rJ r4 r2 r r s# # # # # #r4 r __main__)
run_from_name)r s~ B A
( ( ( ( ( ( $ $ $ $ $ $ ) ) ) ) ) ) 0 0 0 0 0 0 ' ' ' ' ' ' - - - - - - ' ' ' ' ' ' & & & & & & ) ) ) ) ) ) " " " " " " . . . . . . % % % % % % ' ' ' ' ' ' ! ! ! ! ! ! ( ( ( ( ( ( & & & & & & 6> !16688<< RaR F * !. !. !. !. !. !. !. !.HJ J J J J, J J J$ _< _< _< _< _< / _< _< _