| Disable Functions: Path : /lib64/python2.7/site-packages/psutil/tests/ |
| Current File : //lib64/python2.7/site-packages/psutil/tests/test_connections.pyo |
�
S��]c @ 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 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) d d# l m* Z* d d$ l m+ Z+ d d% l m, Z, e j- � Z. e/ e d& e0 � � Z1 d' e0 f d( � � YZ2 d) e2 e) j3 f d* � � YZ4 d+ e2 e) j3 f d, � � YZ5 d- e2 e) j3 f d. � � YZ6 d/ e2 e) j3 f d0 � � YZ7 e) j8 e% d1 � d2 e2 e) j3 f d3 � � Y� Z9 d4 e) j3 f d5 � � YZ: e; d6 k r�d d7 l< m= Z= e= e> � n d S(8 s; Tests for net_connections() and Process.connections() APIs.i����N( t closing( t AF_INET( t AF_INET6( t
SOCK_DGRAM( t SOCK_STREAM( t FREEBSD( t LINUX( t MACOS( t NETBSD( t OPENBSD( t POSIX( t SUNOS( t WINDOWS( t
supports_ipv6( t PY3( t AF_UNIX( t bind_socket( t bind_unix_socket( t check_net_address( t create_sockets( t enum( t
get_free_port( t HAS_CONNECTIONS_UNIX( t pyrun( t
reap_children( t safe_rmpath( t skip_on_access_denied( t SKIP_SYSCONS( t tcp_socketpair( t TESTFN( t TRAVIS( t unittest( t unix_socket_path( t unix_socketpair( t
wait_for_filet SOCK_SEQPACKETt Basec B s/ e Z d � Z d � Z d d � Z d � Z RS( c C s) t t � t s% t j d d � } n d S( Nt kindt all( R R R t thisproct connections( t selft cons( ( sC /usr/lib64/python2.7/site-packages/psutil/tests/test_connections.pyt setUp; s
c C s0 t t � t � t s, t j d d � } n d S( NR% R& ( R R R R R' R( ( R) R* ( ( sC /usr/lib64/python2.7/site-packages/psutil/tests/test_connections.pyt tearDownB s
R& c C s� y t j d | � } Wn! t j k
r9 t r3 d S� n Xg | D] } | j | k rA | d ^ qA } | j � | j � | j | | � d S( s� Given a process PID and its list of connections compare
those against system-wide connections retrieved via
psutil.net_connections.
R% Ni����( t psutilt net_connectionst AccessDeniedR t pidt sortt assertEqual( R) R0 t proc_consR% t sys_const c( ( sC /usr/lib64/python2.7/site-packages/psutil/tests/test_connections.pyt compare_procsys_connectionsK s ,
c s� � f d � } � f d � } � f d � } � f d � } � f d � } | | � | | � | | � | | � | | � d S( s* Check validity of a connection namedtuple.c s� t | � d k } � j t | � d � � j | d | j � � j | d | j � � j | d | j � � j | d | j � � j | d | j � � j | d | j � | r� � j | d | j � n d S(
Ni i i i i i i i ( i i (
t lent assertInR2 t fdt familyt typet laddrt raddrt statusR0 ( t connt has_pid( R) ( sC /usr/lib64/python2.7/site-packages/psutil/tests/test_connections.pyt check_ntuplea s c s� � j | j t t t f � t d k r+ n | j t k r� t j | j | j � } t j
| � �T y | j | j d d f � Wn. t j
k
r� } | j t j k r� � q� n XWd QXn( | j t k r� � j | j t j � n d S( Ni ( R8 R: R R R R t Nonet socketR; t
contextlibR t bindR< t errort errnot
EADDRNOTAVAILR2 R> R- t CONN_NONE( R? t st err( R) ( sC /usr/lib64/python2.7/site-packages/psutil/tests/test_connections.pyt check_familym s c sW � j | j t t t f � t d k r+ n | j t k rS � j | j t j
� n d S( N( R8 R; R R R# R RB R2 R> R- RI ( R? ( R) ( sC /usr/lib64/python2.7/site-packages/psutil/tests/test_connections.pyt
check_type� s
c s� x� | j | j f D]� } | j t t f k rs � j | t � | sJ q n � j | j t � t | j
| j � q | j t k r � j | t � q q Wd S( N(
R< R= R: R R t assertIsInstancet tuplet portt intR t ipR t str( R? t addr( R) ( sC /usr/lib64/python2.7/site-packages/psutil/tests/test_connections.pyt check_addrs� s c s� � j | j t � g t t � D]$ } | j d � r t t | � ^ q } � j | j | � | j t t
f k r� | j t k r� � j
| j t j � n � j | j t j � d S( Nt CONN_( RN R> RS t dirR- t
startswitht getattrR8 R: R R R; R t assertNotEqualRI R2 ( R? t xt valids( R) ( sC /usr/lib64/python2.7/site-packages/psutil/tests/test_connections.pyt check_status� s $$N( ( R) R? RA RL RM RU R] ( ( R) sC /usr/lib64/python2.7/site-packages/psutil/tests/test_connections.pyt check_connection_ntuple_ s
( t __name__t
__module__R+ R, R6 R^ ( ( ( sC /usr/lib64/python2.7/site-packages/psutil/tests/test_connections.pyR$ 9 s t TestBasec B s5 e Z e j e d � d � � Z d � Z d � Z RS( s
requires rootc C s> t � �/ x' t j d d � D] } | j | � q WWd QXd S( NR% R& ( R R- R. R^ ( R) R? ( ( sC /usr/lib64/python2.7/site-packages/psutil/tests/test_connections.pyt test_system� s
c C sD t � �5 x- t j � j d d � D] } | j | � q# WWd QXd S( NR% R& ( R R- t ProcessR( R^ ( R) R? ( ( sC /usr/lib64/python2.7/site-packages/psutil/tests/test_connections.pyt test_process� s
c C s6 | j t t j d d �| j t t j d d �d S( NR% s ???( t assertRaisest
ValueErrorR' R( R- R. ( R) ( ( sC /usr/lib64/python2.7/site-packages/psutil/tests/test_connections.pyt test_invalid_kind� s ( R_ R` R t skipIfR Rb Rd Rg ( ( ( sC /usr/lib64/python2.7/site-packages/psutil/tests/test_connections.pyRa � s t TestUnconnectedSocketsc B s� e Z d Z d � Z d � Z d � Z e j e � d � d � � Z d � Z
e j e � d � d � � Z e j e d � d � � Z
e j e d � d
� � Z RS( s; Tests sockets which are open but not connected to anything.c C s� t j d d � } t g | D] } | j | f ^ q � } t rN | | j � S| j t | � d � | d j d k r� | j | | j � j | j � � n | d Sd S( NR% R& i i i����( R' R( t dictR9 R t filenoR2 R7 ( R) t sockR* R5 t smap( ( sC /usr/lib64/python2.7/site-packages/psutil/tests/test_connections.pyt get_conn_from_sock� s (&c C sJ | j | � } | j | � | j d k rG | j | j | j � � n | j | j | j � | j | j | j t j t j
� � | j � } | r� t r� t
| t � r� | j � } n | j t k r� | d } n | j t k r� t r� n | j | j | � | j t k rFt rFt j d d � } | j t j � | d d �n | S( s� Given 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).
i����i R% R& ( Rn R^ R9 R2 Rk R: R; t
getsockoptRC t
SOL_SOCKETt SO_TYPEt getsocknameR t
isinstancet bytest decodeR R R R< R R' R( R6 t ost getpid( R) Rl R? R< R* ( ( sC /usr/lib64/python2.7/site-packages/psutil/tests/test_connections.pyt check_socket� s&
c C s\ d t � f } t t t t d | �� �, } | j | � } | j | j t j � Wd QXd S( Ns 127.0.0.1RT (
R R R R R Rx R2 R> R- t CONN_LISTEN( R) RT Rl R? ( ( sC /usr/lib64/python2.7/site-packages/psutil/tests/test_connections.pyt test_tcp_v4� s s IPv6 not supportedc C s\ d t � f } t t t t d | �� �, } | j | � } | j | j t j � Wd QXd S( Ns ::1RT (
R R R R R Rx R2 R> R- Ry ( R) RT Rl R? ( ( sC /usr/lib64/python2.7/site-packages/psutil/tests/test_connections.pyt test_tcp_v6� s c C s\ d t � f } t t t t d | �� �, } | j | � } | j | j t j � Wd QXd S( Ns 127.0.0.1RT (
R R R R R Rx R2 R> R- RI ( R) RT Rl R? ( ( sC /usr/lib64/python2.7/site-packages/psutil/tests/test_connections.pyt test_udp_v4� s c C s\ d t � f } t t t t d | �� �, } | j | � } | j | j t j � Wd QXd S( Ns ::1RT (
R R R R R Rx R2 R> R- RI ( R) RT Rl R? ( ( sC /usr/lib64/python2.7/site-packages/psutil/tests/test_connections.pyt test_udp_v6 s s
POSIX onlyc C s\ t � �M } t t | d t �� �, } | j | � } | j | j t j � Wd QXWd QXd S( NR; ( R R R R Rx R2 R> R- RI ( R) t nameRl R? ( ( sC /usr/lib64/python2.7/site-packages/psutil/tests/test_connections.pyt
test_unix_tcp
s c C s\ t � �M } t t | d t �� �, } | j | � } | j | j t j � Wd QXWd QXd S( NR; ( R R R R Rx R2 R> R- RI ( R) R~ Rl R? ( ( sC /usr/lib64/python2.7/site-packages/psutil/tests/test_connections.pyt
test_unix_udp s ( R_ R` t __doc__Rn Rx Rz R Rh R
R{ R| R} R
R R� ( ( ( sC /usr/lib64/python2.7/site-packages/psutil/tests/test_connections.pyRi � s
# t TestConnectedSocketc B sE e Z d Z e j e d � d � � Z e j e d � d � � Z RS( sJ Test socket pairs which are are actually connected to
each other.
s unreliable on SUONSc C s� d t � f } t t d | �\ } } z` t j d d � } | j t | � d � | j | d j t j � | j | d j t j � Wd | j
� | j
� Xd S( Ns 127.0.0.1RT R% t tcp4i i i ( R R R R' R( R2 R7 R> R- t CONN_ESTABLISHEDt close( R) RT t servert clientR* ( ( sC /usr/lib64/python2.7/site-packages/psutil/tests/test_connections.pyt test_tcp% s
s
POSIX onlyc C s� t � ��} t | � \ } } z^t j d d � } t rd g | D] } | j d k r@ | ^ q@ } n | j t | � d � t s� t s� t
r� | j | d j d � | j | d j d � | j | | d j p� | d j � n� t r3x� | d j | d j | d j | d j f D] } | j | d � qWnH | j | d j pP| d j | � | j | d j pt| d j | � Wd | j
� | j
� XWd QXd S( NR% t unixs /var/run/logi i t i ( R R! R' R( R R= R2 R7 R R R R< R R� ( R) R~ R� R� R* R5 RT ( ( sC /usr/lib64/python2.7/site-packages/psutil/tests/test_connections.pyt test_unix9 s& +' $(
( R_ R` R� R Rh R R� R
R� ( ( ( sC /usr/lib64/python2.7/site-packages/psutil/tests/test_connections.pyR� s t TestFiltersc B s2 e Z d � Z e d e � d � � Z d � Z RS( c
s) � f d � } t � � | d t t t g t t t g � | d t t g t t g � | d t g t t g � | d t t g t g � | d t g t g � | d t g t g � | d t t g t g � | d t g t g � | d
t g t g � t r | d t g t t t g � n Wd QXd S( Nc s� x@ t j d | � D], } � j | j | � � j | j | � q Wt s� xC t j d | � D], } � j | j | � � j | j | � q\ Wn d S( NR% ( R' R( R8 R: R; R R- R. ( R% t familiest typesR? ( R) ( sC /usr/lib64/python2.7/site-packages/psutil/tests/test_connections.pyt check^ s R& t inett inet4t tcpR� t tcp6t udpt udp4t udp6R� ( R R R R R R R# R ( R) R� ( ( R) sC /usr/lib64/python2.7/site-packages/psutil/tests/test_connections.pyt test_filters] sB
t only_ifc s� � f d � } t j d � } t j d � } d d l m } t j j t � } | | � j d t t
� d d d | � } | | � j d t t
� d d d | � } | | � j d t t � d d
d | � } | | � j d t t � d d
d | � } t | � }
t
t | � � } t | � } t
t | � � }
t � r{t | � } t
t | � � } t | � } t
t | � � } n d } d } d } d } x5t j � D]'} | j � } � j t | � d � x� | D]� } | j |
j k r | | | t
t | d t j d � q�| j | j k rC| | | t
t |
d t j d � q�| j t | d d � k r�| | | t t | d t j d � q�| j t | d d � k r�| | | t t | d t j d � q�q�Wq�Wd S( Nc
s� d } � j | � � j | j | � � j | j | � � j | j | � � j | j | � � j | j | � x/ | D]' } | j d | � }
| | k ry qy qy Wt r� � j | j
| g � n d S(
NR& R� R� t inet6R� R� R� R� R� R� R% (
s alls inets inet4R� s tcps tcp4s tcp6s udps udp4s udp6( R^ R2 R: R; R< R= R> R( R R6 R0 ( t procR? R: R; R< R= R> t kindst all_kindsR% R* ( R) ( sC /usr/lib64/python2.7/site-packages/psutil/tests/test_connections.pyt
check_conn� s
s
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)
s�
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)
i����( t TemplateR: RT s 127.0.0.1t testfns ::1i R& R� R� R� R� R� R� R0 R� R� R� ( ( s alls inets inet4s tcps tcp4( ( s alls inets inet4s udps udp4( ( s alls inets inet6s tcps tcp6( ( s alls inets inet6s udps udp6( t textwrapt dedentt stringR� Rv t patht basenameR t
substituteRQ R R R t evalR" R
RB R' t childrenR( R2 R7 R0 R R- Ry R RI RY ( R) R� t tcp_templatet udp_templateR� t testfilet
tcp4_templatet
udp4_templatet
tcp6_templatet
udp6_templatet tcp4_proct tcp4_addrt udp4_proct udp4_addrt tcp6_proct tcp6_addrt udp6_proct udp6_addrt pR* R? ( ( R) sC /usr/lib64/python2.7/site-packages/psutil/tests/test_connections.pyt test_combos� s`
c
C s� t � ��t j d d � } | j t | � t � r: d n d � x: | D]2 } | j | j t t f � | j | j
t � qH Wt j d d � } | j t | � d � | j | d j t � | j | d j
t � t � r6t j d d � } | j t | � d � | j | d j t � | j | d j
t � n t j d d � } | j t | � t � rfd n d � x: | D]2 } | j | j t t f � | j | j
t � qtWt j d d � } | j t | � d � | j | d j t � | j | d j
t � t � rbt j d d
� } | j t | � d � | j | d j t � | j | d j
t � n t j d d � } | j t | � t � r�d n d � x@ | D]8 } | j | j t t f � | j | j
t t f � q�Wt � rMt j d d
� } | j t | � d � x= | D]2 } | j | j t � | j | j
t t f � qWn t
r�t r�t j d d � } | j t | � d � x= | D]2 } | j | j t � | j | j
t t f � q�Wn Wd QXd S( NR% R� i i R� i R� R� R� R� R� i R� R� i ( R R' R( R2 R7 R
R8 R: R R R; R R R R R ( R) R* R? ( ( sC /usr/lib64/python2.7/site-packages/psutil/tests/test_connections.pyt
test_count� s\
%
%
%
( R_ R` R� R R R� R� ( ( ( sC /usr/lib64/python2.7/site-packages/psutil/tests/test_connections.pyR� [ s +_s
requires roott TestSystemWideConnectionsc B s8 e Z d Z d � Z e j e o$ e d � d � � Z RS( s Tests for net_connections().c s� � f d � } t � �� d d l m } x� | j � D]r \ } } | d k r[ t r[ q6 n | \ } } t j | � } � j t | � t t | � � � | | | | � q6 WWd QXd S( Nc sf x_ | D]W } � j | j | d | �| j t k rQ � j | j | d | �n � j | � q Wd S( Nt msg( R8 R: R R; R^ ( R* R� t types_R? ( R) ( sC /usr/lib64/python2.7/site-packages/psutil/tests/test_connections.pyR� ' s
i����( t conn_tmapR� (
R t psutil._commonR� t itemsR R- R. R2 R7 t set( R) R� R� R% t groupsR� R� R* ( ( R) sC /usr/lib64/python2.7/site-packages/psutil/tests/test_connections.pyt test_it& s
"s unreliable on MACOS + TRAVISc
C s| t � � } t | � } Wd QXg } d } xo t | � D]a } t j j t � t | � } t j d | � } t
| � } | j | j � | j
t | � q7 Wx. t | � D] } t t | � } t | � q� Wg t j d d � D] } | j | k r� | ^ q� }
xt | D]l } | j t g |
D] } | j | k r | ^ q � | � t j | � } | j t | j d � � | � qWd S( Ni
s import time, os
from psutil.tests import create_sockets
with create_sockets():
with open(r'%s', 'w') as f:
f.write(str(os.getpid()))
time.sleep(60)
R% R& ( R R7 t rangeRv R� t realpathR RS R� R� R t appendR0 t
addCleanupR R" R- R. R2 Rc R( (
R) t sockst expectedt pidst timest it fnamet srct sprocR[ t sysconsR0 R� ( ( sC /usr/lib64/python2.7/site-packages/psutil/tests/test_connections.pyt test_multi_sockets_procs: s*
1( R_ R` R� R� R Rh R R R� ( ( ( sC /usr/lib64/python2.7/site-packages/psutil/tests/test_connections.pyR� " s t TestMiscc B s e Z d � Z RS( c C s� g } g } x{ t t � D]m } | j d � r t t | � } t | � } | j t | � | j | | � | j | � | j | � q q Wt r� t j t j n t
r� t j n d S( NRV ( RW R- RX RY RS t assertNotInR� R t CONN_IDLEt
CONN_BOUNDR t CONN_DELETE_TCB( R) t intst strsR~ t numt str_( ( sC /usr/lib64/python2.7/site-packages/psutil/tests/test_connections.pyt test_connection_constantsd s
( R_ R` R� ( ( ( sC /usr/lib64/python2.7/site-packages/psutil/tests/test_connections.pyR� b s t __main__( t run(? R� RD RG Rv RC R� R R R R R R- R R R R R R
R R R� R
t psutil._compatR t psutil.testsR R R R R R R R R R R R R R R R R R R! R" Rc R' RY t objectR# R$ t TestCaseRa Ri R� R� Rh R� R� R_ t psutil.tests.runnerR� t __file__( ( ( sC /usr/lib64/python2.7/site-packages/psutil/tests/test_connections.pyt <module> sl q b = � ?