B H/\Pb@sddlmZmZmZddlZddlZddlZddlZddlZddlm Z ddlm Z ddlm Z ddlm Z ddlm Z ddlmZddlZdd lmZdd lmZdd lmZdd lmZdd lmZddlmZdgZddZeZddZeej_ej Z ej j!Z"e#edddZ$ddedddfddZ%e%e _%dZ&ej'j(fddZ)e)Z*ddZ+Gddde,Z-Gd d!d!ej j.Z/Gd"d#d#e,Z0Gd$d%d%e,Z1d&d'Z2Gd(ddeZ3dS)))absolute_importprint_functiondivisionN)AI_NUMERICHOST)error)NI_NUMERICSERV)AF_INET)AF_INET6) AF_UNSPEC)AbstractResolver)hostname_types) string_types) iteritems)import_patched)configResolvercCsJtd}xysr cCs |tjtjB@dkrt|dkr4|dkr4ttjg}g}d}y|dkrd}|tj@dkrr|d|dq|d|dn>|d} t | dkr| d} n|} t j | } |||}Wnt k r`ty t j |} |||}WnXt k rZt|tj@dkrVy|tjksD|tjkrtjj|t jjd d } | j}| jd }| jdk rx| jD]} || jqW|tjks|tjkrtjj|t jjd d }|j}|jd }|jdk rx|jD]} || jqWWnRt jjk r.tttjYn(t k rTtttj YnXYnXYnXd}y|dkrxd}nt!|}WnXt k rt|tj"@dkryt#|}Wnt k rtYnXYnX|dkrttjg}|dkrtj$tj%g}n|g}|tj&@dkr*|}nd }|tjksF|tjkrxR|D]J} xB|D]:}x2tj'|D]$}|tj|||| |ddffqfWqVWqLW|tjks|tjkrxN|D]F} x>|D]6}x.tj'|D] }|tj|||| |ffqWqWqWt |dkrttj|S) Nr localhostz::z0.0.0.0z::1z 127.0.0.1%F)raise_on_no_answerT)(socketZ AI_ADDRCONFIGZ AI_V4MAPPEDNotImplementedErrorgaierrorZ EAI_NONAMEZ AI_PASSIVEappendsplitlenripv6 inet_aton Exception _exc_clearipv4rr r resolver _resolverquery rdatatypeAAAAqnamecanonical_nameZto_textrrsetZaddressrANXDOMAINZ EAI_SYSTEMintZAI_NUMERICSERVZ getservbynameZ SOCK_DGRAMZ SOCK_STREAMZ AI_CANONNAMEZ_protocols_for_socktype)hostZservicefamilysocktypeprotoflagsZv6addrsZv4addrsr7partsZahostaddrv6rdatav4portZtuplesZ socktypescnamerrr _getaddrinfos                            $     rHgr@cCsL|sdSt|tstt|y ||Wntjjk rBdSXdSdS)NFT) isinstancer rreprrZ exception SyntaxError)r<parserrr_is_addrs rMcCs4|r&t|trdnd}||dd}t|tjjS)Nr"%r)rIstrr*rMrr,r-)r<srrr _is_ipv6_addrsrRc@s@eZdZdZedejZd ddZddZ dd Z d d Z dS) HostsFilezD A class to read the contents of a hosts file (/etc/hosts). z \s* # Leading space ([^\r\n#]+?) # The actual match, non-greedy so as not to include trailing space \s* # Trailing space (?:[#][^\r\n]+)? # Comments (?:$|[\r\n]+) # EOF or newline NcCs`i|_i|_i|_i|_|dkrFtjdkr0d}ntjdkrFtjd}||_|jsVt d|_ dS)Nposixz /etc/hostsntz'%SystemRoot%\system32\drivers\etc\hostsr) rErCaliasesreverseosrpath expandvarsfnamer _last_load)selfr[rrr__init__s   zHostsFile.__init__c CsFt|jd}|}WdQRXt|ts6|jddn|}|j|S)Nrbignore)errors)openr[readrIrPdecodeLINES_REfindall)r]fpZfdataZudatarrr _readlines%szHostsFile._readlinesc Csjyt|jj}||jk}Wn>ttfk rZddlm}|j |ft d}YnX|sddSi}i}i}i}x| D]}| } t| dkrq~| d} t| r|} nt| r~| drq~|} nq~| d} | | | <x$| D]} | } | | | <| || <qW| |kr,dt| dd}n| d}||kr~| ||<q~W||_||_||_||_||_dS) Nr)get_hubFr#fe80rz .in-addr.arpaz .ip6.arpa.)rXstatr[st_mtimer\IOErrorOSErrorgeventri handle_errorsysexc_inforhr*r+pop _is_ipv4_addrrRrlowerjoinreversedrErCrVrW)r]Z load_timeZ needs_loadrirErCrVrWlinerAZipZipmaprGaliasZptrrrrload4sR         zHostsFile.loadccsP|x t|jD]\}}||fVqWx t|jD]\}}||fVq6WdS)N)rzrrErC)r]rrBrrriter_all_host_addr_pairsms z"HostsFile.iter_all_host_addr_pairs)N) __name__ __module__ __qualname____doc__recompileVERBOSErer^rhrzr{rrrrrSs  9rSc@seZdZdddZdS) _HostsAnswerTcCsXd|_||_||_||_||_|s0|r0tj||_t |drNt |j nd|_ dS)Nttlr) Zresponser6rdtyperdclassr7rr1NoAnswerr8hasattrtimerZ expiration)r]r6rrr8r$rrrr^ws z_HostsAnswer.__init__N)T)r|r}r~r^rrrrrtsrc@sBeZdZdZdefddZejjej j dddfddZ d d Z dS) _HostsResolverz' Class to parse the hosts file NcCst||_||_d|_dS)Nr)rS hosts_fileintervalr\)r]r[rrrrr^s z_HostsResolver.__init__FTcCsLt}|j}|j|j|kr,||_|tjj}t|t rP|} tj |}nt |} | } tj|||} |j|j|| _|tjjkr|j} tjjjj} nV|tjjkr|j} tjjjj} n6|tjjkr|j} dd} n|tjjkr|j} dd} | | } | s$|r$| | dd} | r<| | ||| t|||| |S)NcSstjjj||tj|S)N)rrdtypesANYCNAMEr from_text)ctrBrrrr sz&_HostsResolver.query..cSstjjj||tj|S)N)rrrPTRrr)rrrBrrrr s)rrr\rrzr rdataclassINrIr rrrPrur8ZRRsetrr4r9rErr5rCrrVrrWgetZ is_absoluteaddr)r]r6rrtcpsourcer$Znowrrr8mappingZkindrBrrrr3s>        z_HostsResolver.querycCsb|jj}g}||kr||}n|}||x&t|D]\}}||kr6||q6W|||S)N)rrVr)rremove)r]hostnamerVresultZcannonryrGrrr getaliasess   z_HostsResolver.getaliases) r|r}r~r HOSTS_TTLr^rr4r9rrr3rrrrrrs  0rc@sDeZdZddZejjejjdddejjejj ejj ffddZ dS) _DualResolvercCs"t|_t|_t|j_dS)N)rhosts_resolverr1Zget_default_resolvernetwork_resolverZLRUCachecache)r]rrrr^s z_DualResolver.__init__FNTc Cs|dkr d}t|ts*t|tr*|d}t|trBtj|d}t|trXtj|}|tjj kr||kry|j j ||dd}Wn6t k rddl m} | j|ftYn X|jr|S|jj ||||||dS)Nz0.0.0.0idnaF)r$r)ri)rIr bytesrdrrrr4rrrr3r.rorirprqrrr8r) r]r6rrrrr$Z_hosts_rdtypesZanswerrirrrr3s&        z_DualResolver.query) r|r}r~r^rr4r9rrr5rr3rrrrrs rcCs:|tjkrtjj}n"|tjkr(tjj}nttjd|S)NzAddress family not supported) r&rrr4r9r r5r(Z EAI_FAMILY)r=rrrr_family_to_rdtypes    rc@sPeZdZdZdddZeddZddZd d Zdd d Z ddZ ddZ dS)ra An *experimental* resolver that uses `dnspython`_. This is typically slower than the default threaded resolver (unless there's a cache hit, in which case it can be much faster). It is usually much faster than the c-ares resolver. It tends to scale well as more concurrent resolutions are attempted. Under Python 2, if the ``idna`` package is installed, this resolver can resolve Unicode host names that the system resolver cannot. .. note:: This **does not** use dnspython's default resolver object, or share any classes with ``import dns``. A separate copy of the objects is imported to be able to function in a non monkey-patched process. The documentation for the resolver object still applies. The resolver that we use is available as the :attr:`resolver` attribute of this object (typically ``gevent.get_hub().resolver.resolver``). .. caution:: Many of the same caveats about DNS results apply here as are documented for :class:`gevent.resolver.ares.Resolver`. .. caution:: This resolver is experimental. It may be removed or modified in the future. As always, feedback is welcome. .. versionadded:: 1.3a2 .. _dnspython: http://www.dnspython.org NcCsZtjdkr>t}t_tjr.tj|jjdd<tjr>tj|j_t tjtsNt tj|_dS)N) r1r2rrZresolver_nameserversrZ nameserversZresolver_timeoutZlifetimerIr)r]Zhubr2rrrr^+s   zResolver.__init__cCs|jjS)a The dnspython resolver object we use. This object has several useful attributes that can be used to adjust the behaviour of the DNS system: * ``cache`` is a :class:`dns.resolver.LRUCache`. Its maximum size can be configured by calling :meth:`resolver.cache.set_max_size` * ``nameservers`` controls which nameservers to talk to * ``lifetime`` configures a timeout for each individual query. )r2r)r]rrrr17s zResolver.resolvercCsdS)Nr)r]rrrcloseFszResolver.closec Cst|tst|tr|d}|jj|}|jj}t|}xxy| |t j j |}Wn<t j jt j jt j jfk r|PYq>tk rPYq>X|dd|jD|dj}q>W|S)Nrcss|]}t|jVqdS)N)rPtarget).0Zrrrrr Xsz'Resolver._getaliases..r)rIrPrrdr2rrrrr3rr4rr1rr:Z NoNameserversdTimeoutextendr8r)r]rr=rVZ net_resolverrZansrrr _getaliasesIs     zResolver._getaliasesrc Cs|dks,t|r|ds,t|tr,|t@r@t||||||S|tkryt||||||St j k ryt||t |||St j k rt||t |||SXYqXnt||||||SdS)N)r!s localhostrj) rRrrIrPr_socket getaddrinfor rHr&r(r r)r]r<rFr=r>r?r@rrrr\s zResolver.getaddrinfocCs|r*t|ttfr*|ddkr*t||St|ttfrNt|dtsNtdy t||St k r~|szt|t SYnXdS)Nr)z::1z 127.0.0.1r!z(getnameinfo(): illegal sockaddr argument) rIlisttupler getnameinfor TypeErrorr1Z _getnameinforr)r]Zsockaddrr@rrrr}s   zResolver.getnameinfocCs8|dkrt|St|ts.tdt|ft|S)N)z 127.0.0.1z::1s 127.0.0.1s::1r!z2argument 1 must be str, bytes or bytearray, not %s)r gethostbyaddrrIr rtyper1Z_gethostbyaddr)r]Z ip_addressrrrrs   zResolver.gethostbyaddr)N)rrrr) r|r}r~rr^propertyr1rrrrrrrrrrs$  !)4Z __future__rrrrrrXrqrrrrrr r r&Zgevent.resolverr r Zgevent._compatr rZgevent._patcherrZgevent._configrrrrrrD __import__r1ZTimeoutrgetattrr/rHrr0r-rMrtrRobjectrSZAnswerrrrrrrrrr=sN              m lN)