Cs[a� � Q @ s
d Z d d l m Z d d l Z d d l Z d d l Z d d l Z d d l Z d d l Z d d l Z d d l
Z
d d l Z d d l Z d d l
Z
d d l Z d d l Z d d l Z d d l 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 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' d� k r d d l( Z) n d d l) Z) y d d l) m* Z* Wn? e+ k
rae j, � � e j- d � d d l* Z* Wd QRXYn Xe j' d� k r�d d l. Z. n d Z. d d d d d d d d d d d! d" d# d$ d% d& d' d( d) d* d+ d, d- d. d/ d0 d1 d2 d3 d4 d5 d6 d7 d8 d9 d: d; d< d= d> d? d@ dA dB dC dD dE dF dG dH dI dJ dK dL dM dN dO dP dQ dR dS dT dU dV dW dX dY dZ gD Z/ e j0 d# � ppd[ d� k Z1 d^ e j2 k Z3 e r�d� n d Z4 e5 e j6 j7 d$ � � Z8 e5 e j6 j7 d � � Z9 d` Z: d� Z; e8 s�e9 r�d n dc Z< e j0 dd � s�e1 rde n d Z= e8 se9 r)e: d 9Z: e< d 9Z< df Z> e j? dg k rGdh Z> n di Z> e j@ jA e j@ jB e jC � � e> � ZD eD eE e jF � � ZD eD dj ZG eD e$ dk � ZH e jI � jJ � d� k ZK e j@ jL e j@ jA e j@ jM eN � dn dn � � ZO e j@ jA eO do � ZP e j@ jL e j@ jM eN � � ZQ eR e jS dp � ZT eR e dq � ZU e oJe ZV eR e jS dr � ZW eR e jS ds � ZX eR e jS dt � ZY du e jS � jZ � j[ k Z\ eR e jS dv � Z] eR e jS dw � Z^ eR e jS dx � Z_ eR e jS dy � Z` eR e dz � Za ea o
e5 e jb � � Zc eR e d{ � Zd eR e d| � Ze d} d~ � Zf ef � Zg eh e ji d � Zj d� d� � ek e � D� Zl em e
d� en � � Zo em e
d� en � � Zp eq � Zr eq � Zs eq � Zt d� d� � Zu e jv d� d� � � Zw e jv d� d� � � Zx Gd� d� � d� e jy � Zz d� d� � Z{ e{ d d� d6 � � Z| e{ d� d8 � � Z} d� d7 � Z~ e{ d� d4 � � Z e{ d� d� � � Z� d� d� d5 � Z� d� dI � Z� d� dH � Z� Gd� d� � d� en � Z� e� d� e j� d� d d� e< d� d� � d� dK � � Z� e� d� e� e� f d� d d� e< d� d� � d� d� d� dL � � Z� e� d� e� d� d d� e< d� d� � d� dJ � � Z� d� dC � Z� d� d� � Z� e j� d� dB � � Z� d d� dD � Z� e> d[ d� dG � Z� Gd� d� � d� e) j� � Z� e� e) _� d� d� � Z� d� d> � Z� d� d? � Z� d� d= � Z� e: d� d< � Z� d d� d: � Z� d d� d; � Z� d� d� dO � Z� e j� d[ d� dP � � Z� e e d d� dQ � Z� e
j d� dR � Z� d� d� dS � Z� d� dT � Z� e j� d� dU � � Z� d� dN � Z� d� dM � Z� d� dV � Z� d� dW � Z� d� dX � Z� d� dZ � Z� e r� e j� e> d� dY � � Z� n e j� e> d� dY � � Z� d S)�z
Test utilities.
� )�print_functionN)�AF_INET)�AF_INET6)�
SOCK_DGRAM)� SOCK_STREAM)�MACOS)�POSIX)�SUNOS)�WINDOWS)�
supports_ipv6)�PY3)�u)�unicode)�which� � )�mock�ignore� � �APPVEYOR�DEVNULL�GLOBAL_TIMEOUT�MEMORY_TOLERANCE�
NO_RETRIES�PYPY�
PYTHON_EXE�ROOT_DIR� SCRIPTS_DIR�TESTFILE_PREFIX�TESTFN�TESTFN_UNICODE�TOX�TRAVIS�VALID_PROC_STATUSES� VERBOSITY�HAS_CPU_AFFINITY� HAS_CPU_FREQ� HAS_ENVIRON�HAS_PROC_IO_COUNTERS�
HAS_IONICE�HAS_MEMORY_MAPS�HAS_PROC_CPU_NUM�
HAS_RLIMIT�HAS_SENSORS_BATTERY� HAS_BATTERY�HAS_SENSORS_FANS�HAS_SENSORS_TEMPERATURES�HAS_MEMORY_FULL_INFO�pyrun�
reap_children�get_test_subprocess�create_zombie_proc�create_proc_children_pairZThreadTaskunittest�skip_on_access_denied�skip_on_not_implemented�retry_before_failing�run_test_module_by_name� get_suite� run_suiteZ install_pipZinstall_test_deps�chdir� safe_rmpath�
create_exeZ decode_pathZ encode_path�unique_filename�
get_winver�get_kernel_version�
call_until� wait_for_pid�
wait_for_file�check_connection_ntuple�check_net_address�
get_free_port�unix_socket_path� bind_socket�bind_unix_socket�tcp_socketpair�unix_socketpair�create_sockets�
reload_module�import_module_by_path�warn�copyload_shared_lib�
is_namedtuple� �1�trueZ__pypy__� �
i� i g �?ZSILENT� z$testfn�javaz
$psutil-test-z
@psutil-test-z -internalu -ƒőő�ascii�us-asciiz..�scriptsZ cpu_affinityZcpu_freq�environZ io_countersZioniceZuss� memory_mapsZcpu_numZrlimit�threads�sensors_batteryZ sensors_fansZsensors_temperaturesc C s� d d � } t r� | t j � pt | t j j t j � � pt | t d t j d d � � � pt | t j � j
� � } | s� t d � � | St j j t j � } t j j | � s� t
| � � | Sd S)Nc S sJ y) t j | d g d t j d t j �Wn t k
rA d SYn X| Sd S)Nz-V�stdout�stderr)�
subprocess�
check_call�PIPE� Exception)�exe� rj � /__init__.py�attempt� s #
z _get_py_exe.<locals>.attemptz python%s.%sr z"can't find python exe real abspath)r �sys�
executable�os�path�realpathr � version_info�psutil�Processri �
ValueError�exists�AssertionError)rl ri rj rj rk � _get_py_exe� s # rx zr+c C s. g | ]$ } | j d � r t t | � � q S)ZSTATUS_)�
startswith�getattrrs )�.0�xrj rj rk �
<listcomp>� s r} �AF_UNIX�SOCK_SEQPACKETc C s t | d t j �d S)N�file)�printrm rd )�srj rj rk � logstderr� s r� c C s� t d � t j � x� t j t d � � D]t } t | t � rN t t � } n t } | j | � r* t d | � y t
| � Wq* t k
r� t j
� Yq* Xq* WxI t D]A } t d | � y t
| � Wq� t k
r� t j
� Yq� Xq� Wd S)Nz.executing cleanup_test_files() atexit function�.z removing temporary test file %r)r� r �closero �listdirr
�
isinstancer r ry r? rh � traceback� print_exc�_testfiles_created)�name�prefixrp rj rj rk �cleanup_test_files� s$
r� c C s t d � t d d � d S)Nz.executing cleanup_test_procs() atexit function� recursiveT)r� r4 rj rj rj rk �cleanup_test_procs� s
r� c @ sj e Z d Z d Z d d � Z d d � Z d d � Z d d � Z d
d � Z d d
� Z d d � Z
d S)�
ThreadTaskz6A thread task which does nothing expect staying alive.c C s5 t j j | � d | _ d | _ t j � | _ d S)NFg����MbP?)� threading�Thread�__init__�_running� _interval�Event�_flag)�selfrj rj rk r� s zThreadTask.__init__c C s& | j j } d | | j t | � f S)Nz<%s running=%s at %#x>)� __class__�__name__r� �id)r� r� rj rj rk �__repr__
s zThreadTask.__repr__c C s | j � | S)N)�start)r� rj rj rk � __enter__ s
zThreadTask.__enter__c O s | j � d S)N)�stop)r� �args�kwargsrj rj rk �__exit__ s zThreadTask.__exit__c C s6 | j r t d � � t j j | � | j j � d S)zStart thread and keep it running until an explicit
stop() request. Polls for shutdown every 'timeout' seconds.
zalready startedN)r� ru r� r� r� r� �wait)r� rj rj rk r� s zThreadTask.startc C s: d | _ | j j � x | j r5 t j | j � q Wd S)NT)r� r� �set�time�sleepr� )r� rj rj rk �run s
zThreadTask.runc C s, | j s t d � � d | _ | j � d S)z8Stop thread execution and and waits until it is stopped.zalready stoppedFN)r� ru �join)r� rj rj rk r� $ s zThreadTask.stopN) r� �
__module__� __qualname__�__doc__r� r� r� r� r� r� r� rj rj rj rk r� s r� c s% t j � � � f d d � � } | S)Nc s4 y � | | � SWn t k
r/ t � � Yn Xd S)N)rh r4 )r� r� )�funrj rk �wrapper2 s
z&_reap_children_on_err.<locals>.wrapper)� functools�wraps)r� r� rj )r� rk �_reap_children_on_err1 s !r� c K s� | j d t � | j d t � | j d t j � � | j d t j � t r_ | j d d � | d k r� t t � d t } t d | g } t j
| | � } t j | � t
t d
d d d �n, t j
| | � } t j | � t | j � | S)
a{ Creates a python subprocess which does nothing for 60 secs and
return it as subprocess.Popen instance.
If "cmd" is specified that is used instead of python.
By default stdin and stdout are redirected to /dev/null.
It also attemps to make sure the process is in a reasonably
initialized state.
The process is registered for cleanup on reap_children().
�stdinrc �cwd�env�
creationflagsi Nz:from time import sleep;open(r'%s', 'w').close();sleep(60);z-c�deleteT�empty)�
setdefaultr ro �getcwdr_ r
r? �_TESTFNr re �Popen�_subprocesses_started�addrF rE �pid)�cmd�kwdsZpylineZsprocrj rj rk r5 < s$
c C s� t j j t � d } t j d | t f � } t rJ t | d d �} n t | � } t j
| j � } t | d d d d �} t
| � t | � } t j | � t j
| � } | | f S)a Create a subprocess which creates another one as in:
A (us) -> B (child) -> C (grandchild).
Return a (child, grandchild) tuple.
The 2 processes are fully initialized and will live for 60 secs
and are registered for cleanup on reap_children().
�2a import subprocess, os, sys, time
s = "import os, time;"
s += "f = open('%s', 'w');"
s += "f.write(str(os.getpid()));"
s += "f.close();"
s += "time.sleep(60);"
subprocess.Popen(['%s', '-c', s])
time.sleep(60)
r� r r� Fr� )ro rp �basenamer� �textwrap�dedentr r
r3 rs rt r� rF r? �int�
_pids_startedr� )Z_TESTFN2r� �subpZchild1�dataZ
child2_pidZchild2rj rj rk r7 ] s
c s+ t j s t � t r' t j d t � n t } t j d | � } t
j t j t j
� � �� } | j t � | j | � | j d � t | � | j � \ } } zq t j | j � g g g t � t | j d � � } t j | � t j | � � t � f d d � d � | SWd | j � XWd QRXd S) z+Create a zombie process and return its PID.r� a import os, sys, time, socket, contextlib
child_pid = os.fork()
if child_pid > 0:
time.sleep(3000)
else:
# this is the zombie process
s = socket.socket(socket.AF_UNIX)
with contextlib.closing(s):
s.connect('%s')
if sys.version_info < (3, ):
pid = str(os.getpid())
else:
pid = bytes(str(os.getpid()), 'ascii')
s.sendall(pid)
rZ i c s
� j � S)N)�statusrj )�zprocrj rk �<lambda>� s z$create_zombie_proc.<locals>.<lambda>zret == psutil.STATUS_ZOMBIEN) rs r rw r �tempfile�mktempr r r� r� �
contextlib�closing�socketr~ �
settimeoutr �bind�listenr3 �accept�select�filenor� �recvr� r� rt rD r� )Z unix_file�src�sock�conn�_Zzpidrj )r� rk r6 � s$
c
K s� | j d d � | j d d � t j d t d d d d � �S } t j | j � | j | � | j � t t
| j g | � } t | j � Wd QRX| S) zeRun python 'src' code string in a separate interpreter.
Returns a subprocess.Popen instance.
rc Nrd r� �mode�wtr� F)
r� r� �NamedTemporaryFiler r� r� r� �write�flushr5 r rE r� )r� r� �fr� rj rj rk r3 � s
c K s t | t t f � r d n d } t r3 | r3 d n d } | j d | � | j d t j � | j d t j � | j d d � | j d | � t j | | � } t j | � | j
� \ } } | j d k r� t | � � | r� t
| � | j d
� r
| d d
� } | S)zUrun cmd in a subprocess and return its output.
raises RuntimeError on error.
TFi r �shellrc rd �universal_newlinesr� �
NrZ ���)r� �strr r
r� re rg r� r� r� � communicate�
returncode� RuntimeErrorrR �endswith)r� r� r� �flags�prc rd rj rj rk �sh� s" !
r� Fc ' C s� d d � } | r3 t t j � j d d � � } n t � } x t r\t j � } t j | j � y | j � WnR t
k
r� } z2 t r� | j d k r� n | j t j
k r� � WYd d } ~ Xn X| j r� | j j � | j r� | j j � z | j r | j j � Wd y | j � Wn: t
k
rW} z | j t j k rE� WYd d } ~ Xn XXq? WxX t r�t j � } y t j | � } Wn t j k
r�| | � Yq`X| j | � q`W| r�x4 | D], } y | j � Wq�t j k
r�Yq�Xq�Wt j | d t �\ } } xB | D]: } t d | � y | j � Wqt j k
rPYqXqWt j | d t �\ } } | r�x | D] } t d | � q}Wx | D] } | | j � q�Wd S)
a# Terminate and wait() any subprocess started by this test suite
and ensure that no zombies stick around to hog resources and
create problems when looking for refleaks.
If resursive is True it also tries to terminate and wait()
all grandchildren started by this process.
c S s� t j | � s t | � � | t j � k s: t | � � y, t j | � } | j � se t | � � Wn t j k
r} Yn Xd s� t d | � � d S)Nr zpid %s is not gone)rs Z
pid_existsrw Zpidsrt Z
is_running�
NoSuchProcess)r� r� rj rj rk � assert_gone� s z"reap_children.<locals>.assert_goner� TrX N�timeoutz0couldn't terminate process %r; attempting kill()zprocess %r survived kill())r� rs rt �childrenr� �popr� r� r� � terminate�OSErrorr
�errnoZESRCHrc r� rd r� r� ZECHILDr� Z
wait_procsr rR �kill) r� r� r� r� �errr� r� Zgone�aliverj rj rk r4 � sh !
c C s� t s t d � � d } t j � d } x4 | D], } | j � sM | d k rZ | | 7} q/ Pq/ W| su t d | � � d } d } | j d � } t | d � } t | � d k r� t | d � } t | � d k r� t | d � } | | | f S) z"Return a tuple such as (2, 6, 36).z not POSIXrU r r� zcan't parse %rr rZ r ) r �NotImplementedErrorro �uname�isdigitru �splitr� �len)r� r� �c�minor�microZnums�majorrj rj rk rC 0 s&
c C s� t s t d � � t j � } t | d � r? | j p9 d } n: t j d | d � } | rs t | j d � � } n d } | d | d | f S)Nz not WINDOWS�service_pack_majorr z\s\d$r rZ )
r
r� rm Zgetwindowsversion�hasattrr �re�searchr� �group)ZwvZsp�rrj rj rk rB H s c @ s[ e Z d Z d Z e d d d d d � d d � Z d d � Z d
d � Z d d
� Z d S)�retryzA retry decorator.Ng����MbP?c C s t | d t j �S)Nr� )r� rm rd )r� rj rj rk r� d s zretry.<lambda>c C sI | r | r t d � � | | _ | | _ | | _ | | _ | | _ d S)Nz/timeout and retries args are mutually exclusive)ru � exceptionr� �retries�interval�logfun)r� r r� r r
r rj rj rk r� _ s zretry.__init__c c sx | j r= t j � | j } xU t j � | k r9 d Vq Wn7 | j rh x+ t | j � D] } d VqV Wn x d Vqk Wd S)N)r� r� r �range)r� �stop_atr� rj rj rk �__iter__n s zretry.__iter__c C s# | j d k r t j | j � d S)N)r
r� r� )r� rj rj rk r� z s z retry.sleepc s1 t j � � � � f d d � � } � | _ | S)Nc s� d } xy � D]q } y � | | � SWq
� j k
r} } z4 | } � j d k r^ � j | � � j � w
WYd d } ~ Xq
Xq
Wt r� | � n � d S)N)r r r� r )r� r� �excr� )r� r� rj rk r� s
z retry.__call__.<locals>.wrapper)r� r� � decorator)r� r� r� rj )r� r� rk �__call__~ s $ zretry.__call__) r� r� r� r� rh r� r r� r rj rj rj rk r \ s
r r r r� r
g����MbP?c C s$ t j | � t r t j d � d S)z�Wait for pid to show up in the process list then return.
Used in the test suite to give time the sub process to initialize.
g{�G�z�?N)rs rt r
r� r� )r� rj rj rk rE � s
Tc C sK t | d � � } | j � } Wd QRX| s7 | s7 t � | rG t | � | S)z8Wait for a file to be written on disk with some content.�rbN)�open�readrw r? )Zfnamer� r� r� r� rj rj rk rF � s
c C s | � } t | � s t � | S)zVKeep calling function for timeout secs and exit if eval()
expression is True.
)�evalrw )r� �expr�retrj rj rk rD � s c C s� d d � } yl t j | � } t j | j � rH t j t j | � } n t j t j | � } t rm | � n
| | � Wn: t
k
r� } z | j t j k r� � WYd d } ~ Xn Xd S)z>Convenience function for removing temporary test files or dirsc S s� t j � d } x� t j � | k r� y | � SWq t k
r� } zD | } | j t j k rc � n! t d t | � � t j d � WYd d } ~ Xq Xq W| � d S)NrZ z ignoring %sg{�G�z�?)r� Z WindowsErrorr� �ENOENTrR r� r� )r� r
r� r� rj rj rk � retry_fun� s $z safe_rmpath.<locals>.retry_funN)
ro �stat�S_ISDIR�st_moder� �partial�shutil�rmtree�remover r� r� r )rp r �str� r� rj rj rk r? � s
c C sR y t j | � Wn: t k
rM } z | j t j k r; � WYd d } ~ Xn Xd S)z-Convenience function for creating a directoryN)ro �mkdirr� r� �EEXIST)�dirr� rj rj rk �
safe_mkdir� s
r% c c s7 t j � } z t j | � d VWd t j | � Xd S)z@Context manager which temporarily changes the current directory.N)ro r� r> )�dirname�curdirrj rj rk r> � s
c C s" t j j | � s t | � � | r� t d � s= t d � � t | t � r[ t j d � } t | t
� sv t | � � t j d d d d d d � � } | j
| � Wd
QRXz t j d | j d | g � Wd
t | j � Xn? t j t | � t r t j | � } t j | | j t j B� d
S) z1Creates an executable file in the given location.�gcczgcc is not installedz�
#include <unistd.h>
int main() {
pause();
return 1;
}
�suffixz.cr� Fr� r� Nz-o)ro rp rv rw r ru r� �boolr� r� r� r� r� r� re rf r� r? r �copyfiler r r �chmodr �S_IEXEC)ZoutpathZc_coder� r! rj rj rk r@ � s$ c C s t j d | d | � S)Nr� r) )r� r� )r� r) rj rj rk rA s c @ s: e Z d Z d d � Z e e j d � s6 e j j Z d S)�TestCasec C s d | j j | j j | j f S)Nz%s.%s.%s)r� r� r� Z_testMethodName)r� rj rj rk �__str__" s zTestCase.__str__�assertRaisesRegexN) r� r� r� r/ r �unittestr. ZassertRaisesRegexpr0 rj rj rj rk r. s r. c C s0 d t j k r d t j d <t j j j � d S)NZPSUTIL_TESTINGrV )ro r_ rs Z _psplatformZcextZ set_testingrj rj rj rk � _setup_tests1 s
r2 c C s� d d � t j t � D� } d t j k r> d d � | D� } t j � } x1 | D]) } d | } | j t j j | � � qQ W| S)Nc S sT g | ]J } | j d � r | j d � r | j d � r t j j | � d � q S)z.pyZtest_Ztest_memory_leaksr )r� ry ro rp �splitext)r{ r| rj rj rk r} 9 s z get_suite.<locals>.<listcomp>Z WHEELHOUSE_UPLOADER_USERNAMEc S s% g | ] } | j d � s | � q S)�osx�posix�linux)r4 r5 r6 )r� )r{ r| rj rj rk r} = s zpsutil.tests.%s) ro r� �HEREr_ r1 � TestSuite�addTest�defaultTestLoader�loadTestsFromName)Ztestmods�suiteZtmrj rj rk r< 8 s
c C sN t � t j d t � j t � � } | j � } t j | rC d n d � d S)N� verbosityr rZ ) r2 r1 �TextTestRunnerr% r� r< �
wasSuccessfulrm �exit)�result�successrj rj rk r= G s c C s� t � t j j t j j | � � d } t j � } | j t j j | � � t j
d t � j | � } | j
� } t j | r� d n d � d S)Nr r= rZ )r2 ro rp r3 r� r1 r8 r9 r: r; r> r% r� r? rm r@ )r� r< rA rB rj rj rk r; N s " c C s t d t d d d | � S)zZDecorator which runs a test function and retries N times before
actually failing.
r r� Nr )r rw )r rj rj rk r: Z s c s � f d d � } | S)z,Decorator to Ignore AccessDenied exceptions.c s( t j � � � � f d d � � } | S)Nc sQ y � | | � SWn9 t j k
rL � d k r9 � s9 � t j d � � Yn Xd S)Nzraises AccessDenied)rs Z AccessDeniedr1 �SkipTest)r� r� )r� �only_ifrj rk r� d s z9skip_on_access_denied.<locals>.decorator.<locals>.wrapper)r� r� )r� r� )rD )r� rk r c s $ z(skip_on_access_denied.<locals>.decoratorrj )rD r rj )rD rk r8 a s c s � f d d � } | S)z3Decorator to Ignore NotImplementedError exceptions.c s( t j � � � � f d d � � } | S)Nc s[ y � | | � SWnC t k
rV � d k r6 � s6 � d � j } t j | � � Yn Xd S)Nz4%r was skipped because it raised NotImplementedError)r� r� r1 rC )r� r� �msg)r� rD rj rk r� t s
z;skip_on_not_implemented.<locals>.decorator.<locals>.wrapper)r� r� )r� r� )rD )r� rk r s s $ z*skip_on_not_implemented.<locals>.decoratorrj )rD r rj )rD rk r9 q s
z 127.0.0.1c C s] t j t j � � �A } | j t j t j d � | j | d f � | j � d SWd QRXd S)zReturn an unused TCP port.rZ r N)r� r� r� �
setsockopt�
SOL_SOCKET� SO_REUSEADDRr� � getsockname)�hostr� rj rj rk rI � s c c sU t j s t � t d | � } z | VWd y t j | � Wn t k
rO Yn XXd S)zaA context manager which returns a non-existent file name
and tries to delete it on exit.
r) N)rs r rw rA ro �unlinkr� )r) rp rj rj rk rJ � s
c C s� | d k r$ | t t f k r$ d } t j | | � } yJ | j t j t j d � | j | � | t j k r{ | j d � | SWn t k
r� | j
� � Yn Xd S)zBinds a generic socket.NrU r rZ rY )rU r ) r r r� rF rG rH r� r r� rh r� )�family�type�addrr� rj rj rk rK � s
c C s� t j s t � t j j | � s. t | � � t j t j | � } y- | j | � | t j k ro | j
d � Wn t k
r� | j � � Yn X| S)zBind a UNIX socket.rY )
rs r rw ro rp rv r� r~ r� r r� rh r� )r� rM r� rj rj rk rL � s
c C s� t j t j | t � � �� } | j | � | j d � | j � } t j | t � } yV | j | � | j � } x6 | j � \ } } | | k r� | | f S| j � qu WWn t
k
r� | j � � Yn XWd QRXd S)z^Build a pair of TCP sockets connected to each other.
Return a (server, client) tuple.
rY N) r� r� r� r r� r� rI �connectr� r� r� )rL rN Zllr� Zcaddr�arj rj rk rM � s
c C s� t j s t � d } } yX t | d t j �} | j d � t j t j t j � } | j d � | j | � WnA t k
r� | d k r� | j
� | d k r� | j
� � Yn X| | f S)z�Build a pair of UNIX sockets connected to each other through
the same UNIX file name.
Return a (server, client) tuple.
NrM r ) rs r rw rL r� r � setblockingr~ rO rh r� )r� Zserver�clientrj rj rk rN � s
c c sY g } d } } z� | j t t j t j � � | j t t j t j � � t � r� | j t t j t j � � | j t t j t j � � t rt rt
� j � } t
� j � } t | � \ } } t
| d t j �} x$ | | | f D] } | j | � q� W| VWd x | D] } | j � qW| d k r>t | � | d k rTt | � Xd S)z1Open as many socket families / types as possible.NrM )�appendrK r� r r r r r r �HAS_CONNECTIONS_UNIXrJ r� rN rL r� r? )ZsocksZfname1Zfname2�s1�s2Zs3r� rj rj rk rO � s,
c C so d d l } t r6 t r6 t | t j � s6 t | � � | t j k r� d d � | j d � D� } t | � d k s t | � � x6 | D]. } d | k o� d k n s� t | � � q� Wt s� t
| � } | j | � n� | t j k r&t | t
� st | � � t st
| � } | j | � nE | t j k r\t j d | � d k skt | � � n t d | � � d S)
z[Check a net address validity. Supported families are IPv4,
IPv6 and MAC addresses.
r Nc S s g | ] } t | � � q Srj )r� )r{ r| rj rj rk r} s z%check_net_address.<locals>.<listcomp>r� r � z([a-fA-F0-9]{2}[:|\-]?){6}zunknown family %r)� ipaddress�enumr r� �IntEnumrw r� r r� r� r Z IPv4Addressr r� Z IPv6Addressrs ZAF_LINKr �matchru )rN rL rX Zocts�numrj rj rk rH s&
, 'c ( C s� t | � d k s t | � � t | � d k } t | d d � d k } | d | j k sa t � | d | j k sz t � | d | j k s� t � | d | j k s� t � | d | j k s� t � | d | j k s� t � | r� | d | j k s� t � | r�| j d k s t | � � t
t d
� r�t r�y" t j
| j | j | j � } WnG t j t f k
r�} z | j d t j k r�� WYd d } ~ XnH Xt j | � �5 | j | j k s�t � | j | j k s�t � Wd QRX| j t t t f k st t | j � � � | j t t f k r�| j t k r�t j | j | j � } t j | � �c y | j | j d d f � Wn= t j k
r�} z | j t j k r�� WYd d } ~ Xn XWd QRXn0 | j t k r�| j t j k s�t | j � � | j t t t f k s%t t | j � � � | j t k rU| j t j k sUt | j � � x� | j | j f D]� } | j t t f k r t | t! � s�t | � � | s�qht | j" t# � s�t | j" � � d | j" k o�d k n s�t | j" � � t$ | j% | j � qh| j t k rht | t& � sht | � � qhWt | j t& � sXt | � � d
d � t' t � D� } | j | k s�t | � � d S)z*Check validity of a connection namedtuple.rX r �fdrZ r r r r � �fromfdNi�� c S s. g | ]$ } | j d � r t t | � � q S)ZCONN_)ry rz rs )r{ r| rj rj rk r} h s z+check_connection_ntuple.<locals>.<listcomp>)rX r r� r� )(r� rw rz r] rL rM �laddr�raddrr� r� r r� r
r_ �errorr� r� r� �EBADFr� r� r r r~ �reprr� Z
EADDRNOTAVAILrs Z CONN_NONEr r r r� �tuple�portr� rH Zipr� r$ )r� Zhas_pidZhas_fdZdupsockr� r� rN Zvalidsrj rj rk rG % sf " - !!!. c C sd y% d d l } t | d � s$ t � Wn+ t k
rR d d l } | j | � SYn X| j | � Sd S)z,Backport of importlib.reload of Python 3.3+.r N�reload)� importlibr � ImportError�imprg )�modulerh rj rj rj rk rP q s
c C s� t j j t j j | � � d } t j d d k rQ d d l } | j | | � St j d d � d k r� d d l m } | | | � j
� Sd d l } | j j
| | � } | j j | � } | j j | � | Sd S)Nr r r r )�SourceFileLoader)r r )ro rp r3 r� rm rr rj Z load_sourceZimportlib.machineryrl � load_module�importlib.util�util�spec_from_file_location�module_from_spec�loader� exec_module)rp r� rj rl rh �spec�modrj rj rk rQ ~ s " c C s t j | t � d S)zRaise a warning msg.N)�warningsrR � UserWarning)rE rj rj rk rR � s c C sw t | � } | j } t | � d k s7 | d t k r; d St | d d � } t | t � s` d St d d � | D� � S)z-Check if object is an instance of namedtuple.rZ r F�_fieldsNc s s! | ] } t | � t k Vq d S)N)rM r� )r{ �nrj rj rk � <genexpr>� s z is_namedtuple.<locals>.<genexpr>)rM � __bases__r� re rz r� �all)r| �t�br� rj rj rk rT � s "c # s� d � t j d | d � � } � f d d � t j � j � D� } t j | � } t j | | � z t j
| � | VWd t | � Xd S)z�Ctx manager which picks up a random shared CO lib used
by this process, copies it in another location and loads it
in memory via ctypes. Return the new absolutized path.
z.sor� r) c sM g | ]C } t j j | j � d � k r d | j j � k r | j � q S)rZ �python)ro rp r3 �lower)r{ r| )�extrj rk r} � s z'copyload_shared_lib.<locals>.<listcomp>N) r� r� rs rt r` �random�choicer r+ �ctypesZCDLLr? )�
dst_prefix�dst�libsr� rj )r� rk rS � s %
c # s� d d l m } d d l m } d � t j d | d � � } � f d d � t j � j � D� } t j | � } t
j | | � d } z t j | � } | VWd | d k r� t j
j j } | j g | _ | | j � } | d k r� | � t | � Xd S)
z�Ctx manager which picks up a random shared DLL lib used
by this process, copies it in another location and loads it
in memory via ctypes.
Return the new absolutized, normcased path.
r )�wintypes)�WinErrorz.dllr� r) c st g | ]j } t j j | j � d j � � k r d t j j | j � j � k r d | j j � k r | j � q S)rZ r Zwow64)ro rp r3 r� r� )r{ r| )r� rj rk r} � s %!z'copyload_shared_lib.<locals>.<listcomp>N)r� r� r� r� r� rs rt r` r� r� r r+ ZWinDLLZwindllZkernel32� FreeLibraryZHMODULEZargtypes�_handler? ) r� r� r� r� r� r� �cfiler� r rj )r� rk rS � s$ % )r r )r r )rV rW )rX r r i � )r\ r] )rU r )�r� Z
__future__r �atexitr� r� r� r� ro r� r r� r r� r re rm r� r� r� r� r� rv r r r r rs r r r r
Zpsutil._commonr Zpsutil._compatr r
r r rr Z unittest2r1 r ri �catch_warnings� simplefilterrY �__all__�getenvr" �builtin_module_namesr Z WIN_VISTAr* r_ �getr# r r r r r% r r� rp r� rq r� r r� �getpidr� r! �getfilesystemencodingr� ZASCII_FS�abspathr& �__file__r r r7 r rt r& r' rT r( r) r* Zmemory_full_inforx r2 r+ r, r- Z HAS_THREADSr. rb r/ r0 r1 rx r r �devnullr r$ r$ rz �objectr~ r r� r� r� r� r� �registerr� r� r� r� r� r5 r7 r6 r3 r� r4 rC rB r r� rE �EnvironmentErrorrw rF rD r? r% �contextmanagerr> r@ rA r. r2 r< r= r; r: r8 r9 rI rJ rK rL rM rN rO rH rG rP rQ rR rT rS rj rj rj rk �<module> s\
!
'
0
0 # %a : # L