gXdZddlZddlZddlZddlZddlZddlZddlZddlm Z ddl Z ddl m Z dZGddeZGdd eZGd d eZed fd Zed fdZdZGddeZddZedk(xr eyy)z: A simple library for managing the availability of ports. N)abc)timingc|dk(ry|dvry|S)z Return the host on which a client can connect to the given listener. >>> client_host('192.168.0.1') '192.168.0.1' >>> client_host('0.0.0.0') '127.0.0.1' >>> client_host('::') '::1' z0.0.0.0z 127.0.0.1)z::z::0z ::0.0.0.0z::1) server_hosts /opt/Tautulli/lib/portend.py client_hostr s#i00 c"eZdZddZddZdZy)Checkerc||_yNtimeout)selfrs r__init__zChecker.__init__)s  r NcJ|"t|tjr|dd\}}tjdk(r t |}t j||t jt j}ttj|j|y)a Assert that the given addr is free in that all attempts to connect fail within the timeout or raise a PortNotFree exception. >>> free_port = find_available_local_port() >>> Checker().assert_free('localhost', free_port) >>> Checker().assert_free('127.0.0.1', free_port) >>> Checker().assert_free('::1', free_port) Also accepts an addr tuple >>> addr = '::1', free_port, 0, 0 >>> Checker().assert_free(addr) Host might refer to a server bind address like '::', which should use localhost to perform the check. >>> Checker().assert_free('::', free_port) NWindows) isinstancerSequenceplatformsystemr socket getaddrinfo AF_UNSPEC SOCK_STREAMlist itertoolsstarmap_connect)rhostportinfos r assert_freezChecker.assert_free,sy, .>@R@RS Y  t}}d 34r c|tj|||}|j|jtj|5 |j | ddd|dd\}}d} t| jdit#tj $r YdddywxYw#1swYWxYw)Nrz Port {port} is in use on {host}.r) r settimeoutr contextlibclosingconnecterror PortNotFreeformatlocals) rafsocktypeproto canonnamesasr"r#tmpls rr!zChecker._connectIs MM"h . T\\"    "   "  V d1+$++1122 <<       s*B2 BB/$B2.B//B22B;)g?r)__name__ __module__ __qualname__rr%r!rr rr r (s5:3r r c eZdZy)TimeoutNr6r7r8rr rr:r:Zr r:c eZdZy)r,Nr;rr rr,r,^r<r r,Infc 4|s tdtj|} tdj ||y#t $rK|j r#tdjdittjdYnwxYwv)u Wait for the specified port to become free (dropping or rejecting requests). Return when the port is free or raise a Timeout if timeout has elapsed. Timeout may be specified in seconds or as a timedelta. If timeout is None or ∞, the routine will run indefinitely. >>> free('localhost', find_available_local_port()) >>> free(None, None) Traceback (most recent call last): ... ValueError: Host values of '' or None are not allowed. *Host values of '' or None are not allowed.皙?rNzPort {port} not free on {host}.r) ValueErrorrTimerr r%r,expiredr:r-r.timesleepr"r#rtimers rfreerIbs EFF LL !E   C , ,T4 8  }}F?FFRRSS JJsO   sAABBc2|s tdtj|} tdj |||j r#t djdittjdf#t$rYywxYw)uE Wait for the specified port to become occupied (accepting requests). Return when the port is occupied or raise a Timeout if timeout has elapsed. Timeout may be specified in seconds or as a timedelta. If timeout is None or ∞, the routine will run indefinitely. >>> occupied('localhost', find_available_local_port(), .1) Traceback (most recent call last): ... Timeout: Port ... not bound on localhost. >>> occupied(None, None) Traceback (most recent call last): ... ValueError: Host values of '' or None are not allowed. r@g?rz Port {port} not bound on {host}.rANr) rBrrCr r%rDr:r-r.rErFr,rGs roccupiedrKs& EFF LL !E   C , ,T4 8}}G@GGS&(STT JJsO    sA$B BBc<tjddtjtj}t t |\}}}}}tj||}|j ||jdd\}}|j|S)zb Find a free port on localhost. >>> 0 < find_available_local_port() < 65536 True Nrr) rrrrnextiterbind getsocknameclose)infosfamilyr1_addrsockr#s rfind_available_local_portrWs   tQ(8(8&:L:L ME $T%[ 1FE1a == 'DIIdO!!#BQ'JD$JJL Kr c@eZdZdZedZedZedZy)HostPorta A simple representation of a host/port pair as a string >>> hp = HostPort('localhost:32768') >>> hp.host 'localhost' >>> hp.port 32768 >>> len(hp) 15 >>> hp = HostPort('[::1]:32768') >>> hp.host '::1' >>> hp.port 32768 cZtjjd|jSNz//)urllibparseurlparsehostnamers rr"z HostPort.hosts#||$$r$[1:::r cZtjjd|jSr[)r\r]r^r#r`s rr#z HostPort.ports#||$$r$[1666r c|dd\}}t|}d|vrd|dn|}|dj|t|gS)Nr:[])r joinstr)clsrU listen_hostr# plain_hostr"s r from_addrzHostPort.from_addrsP !H T - $':$5:,a :388T3t9-.//r N) r6r7r8__doc__propertyr"r# classmethodrkrr rrYrYsC.;;7700r rYctj}d}|jddt|jdd||jdddt |j |} |j |jj|jj|j y#t$r+}t|tj td d}~wwxYw) Nct|Sr)globals)keys r global_lookupz_main..global_lookupsy~r targetz host:port)metavartypefuncstatez-tz --timeout)defaultrvr)file)argparseArgumentParser add_argumentrYfloat parse_argsrwrtr"r#rr:printsysstderr SystemExit)argsparserrsrs r_mainrs  $ $ &F +HE mD k4eD   T "D $++""DKK$4$4dll K  gCJJ'ms/AB66 C*?&C%%C*__main__r)rlrErr|rrr(r collectionsr urllib.parser\temporarr objectr IOErrorr:r,rrIrKrWrgrYrr6rr rrs  ,/3f/3d g  ' #5\B"'u!H  %0s%0P" J"57"r