File: //lib/python2.7/site-packages/DNS/Base.pyc
�
.� Yc @ 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 Z d d l Z d d l Z d d l Z y d d l
m Z e � Z
Wn d d l
Z
n Xd e f d � � YZ
d e
f d � � YZ d e
f d � � YZ d
e
f d � � YZ d e
f d
� � YZ d e
f d � � YZ d d l Z i d d 6d d 6e j d 6e j d 6d d 6d d 6d d 6d d 6a g t d <d d � Z d � Z d! d% d" � � YZ d# e e j f d$ � � YZ d S(& s.
$Id: Base.py,v 1.12.2.19 2011/11/23 17:14:11 customdesigned Exp $
This file is part of the pydns project.
Homepage: http://pydns.sourceforge.net
This code is covered by the standard Python License. See LICENSE for details.
Base functionality. Request and Response classes, that sort of thing.
i����N( t SystemRandomt DNSErrorc B s e Z RS( ( t __name__t
__module__( ( ( s, /usr/lib/python2.7/site-packages/DNS/Base.pyR s t
ArgumentErrorc B s e Z RS( ( R R ( ( ( s, /usr/lib/python2.7/site-packages/DNS/Base.pyR s t SocketErrorc B s e Z RS( ( R R ( ( ( s, /usr/lib/python2.7/site-packages/DNS/Base.pyR s t TimeoutErrorc B s e Z RS( ( R R ( ( ( s, /usr/lib/python2.7/site-packages/DNS/Base.pyR s t ServerErrorc B s e Z d � Z RS( c C s) t j | | | � | | _ | | _ d S( N( R t __init__t messaget rcode( t selfR R
( ( s, /usr/lib/python2.7/site-packages/DNS/Base.pyR " s ( R R R ( ( ( s, /usr/lib/python2.7/site-packages/DNS/Base.pyR ! s t IncompleteReplyErrorc B s e Z RS( ( R R ( ( ( s, /usr/lib/python2.7/site-packages/DNS/Base.pyR ' s t udpt protocoli5 t portt opcodet qtypei t rdt timingi t timeouti t
server_rotatet servers /etc/resolv.confc C s t | � j � } x| D]� } t j | � } | s | d d k s | d d k r[ q n t j | � } t | � d k r� q n | d d k r� t | � d k r� | d t d <n | d d k r� n | d d k r� n | d d k r� n | d d
k r t d j | d � q q Wd S(
sC parses the /etc/resolv.conf file and sets defaults for name serversi t ;t #i t domaini t searcht optionst sortlistt
nameserverR N( t opent readlinest stringt stript splitt lent defaultst append( t resolv_patht linest linet fields( ( s, /usr/lib/python2.7/site-packages/DNS/Base.pyt ParseResolvConf2 s$
'"c C sE d d l } | j d k r: d d l } | j � t d <n t � Sd S( Ni����t win32t ntR ( s win32s nt( t syst platformt win32dnst RegistryResolveR$ R* ( R- R/ ( ( s, /usr/lib/python2.7/site-packages/DNS/Base.pyt DiscoverNameServersH s
t
DnsRequestc B sz e Z d Z d � Z d � Z d � Z d � Z d � Z d � Z d � Z d � Z
d � Z d
� Z d � Z
d � Z RS(
s high level Request object c O sD d | _ d | _ i | _ | j | | � | j | _ d | _ d S( Ni ( t Nonet donefunct asyncR$ t argparset argst tid( R t nameR7 ( ( s, /usr/lib/python2.7/site-packages/DNS/Base.pyR R s c C sZ | r- | j j d � r- | j d | d <n t | � t j k rO | | d <n0 t | � d k r | d r | d | d <q n t d r� t t d � t j k r� t d d t d d t d <n xZ t j � D]L } | j | � s� | j j | � r | j | | | <q t | | | <q� q� Wt | d � t j k rM| d g | d <n | | _ d S( NR9 i i R R ( R$ t has_keyt typet typest
StringTypeR# t ListTypet keysR7 ( R R9 R7 t i( ( s, /usr/lib/python2.7/site-packages/DNS/Base.pyR6 Z s$
!c C s t j | | � | _ d S( N( t sockett s( R t at b( ( s, /usr/lib/python2.7/site-packages/DNS/Base.pyt
socketInitp s c C s� | j d k rT t j | j g g g | j � \ } } } t | � sT t d � qT n | j j d � \ | _ | _ t j � | _ | j
| j d <| j � S( Ni t Timeouti�� R (
R t selectRB R# R t recvfromt replyt from_addresst timet time_finisht nsR7 t processReply( R t rt wt e( ( s, /usr/lib/python2.7/site-packages/DNS/Base.pyt processUDPReplys s * c C s� | j | � } x� t | � | k r� | j d k rx | j | j t j � } | d k re t d � n | j j | � n | j | t | � � } | s� t d t | � | f � n | | 7} q W| S( Ni RF s incomplete reply - %d of %d read( t readR# R t
time_startRK R RB t
settimeoutR ( R t ft countt rest remt buf( ( s, /usr/lib/python2.7/site-packages/DNS/Base.pyt _readall} s c C s� | j d k r% | j j | j � n | j j d � | j j d � } z: | j | d � } t j | � } | j | | � | _ Wd | j � Xt
j
� | _ | j | j
d <| j � S( Ni t rbi R ( R RB RU R3 t makefileR[ t Libt unpack16bitRI t closeRK RL RM R7 RN ( R RV t headerRW ( ( s, /usr/lib/python2.7/site-packages/DNS/Base.pyt processTCPReply� s c C sR | j | j d | j d <t j | j � } t j | | j � } | j | _ | S( Ni� t elapsed( RL RT R7 R^ t MunpackerRI t DnsResult( R t uRO ( ( s, /usr/lib/python2.7/site-packages/DNS/Base.pyRN � s
c C sm xf t rh y- t j d d � } | j j d | f � PWq t j k
rd } | d d k re � qe q Xq Wd S( s< Pick random source port to avoid DNS cache poisoning attack.i i�� t i ib N( t Truet randomt randintRB t bindRA t error( R t source_portt msg( ( s, /usr/lib/python2.7/site-packages/DNS/Base.pyt getSource� s c C s* | j � | j j | j | j f � d S( N( Ro RB t connectRM R ( R ( ( s, /usr/lib/python2.7/site-packages/DNS/Base.pyt conn� s
c O s
| j | | � | j d } | j d | _ t j d d � | _ | j d | _ | j d } | j d } | j d } t | j d � t j k r� y# t
t t j
| j d � � } Wq� t k
r� t d
� q� Xn
| j d } | j j d � s | j GHt d � n | j d } | t j k rA| d
k rAd GHd
} n t j � } | j | j d | d d | d d d d d d d �
| j | | t j � | j � | _ y- | d k r�| j | � n
| j | � Wn t j k
r�}
t |
� n X| j rd S| j! Sd S( s needs a refactoring R R i i�� R R R R R s unknown query typeR9 s nothing to lookupt tcps' Query type AXFR, protocol forced to TCPi R
N(" R6 R7 R Ri Rj R8 R R; R< R= t getattrt TypeR t uppert AttributeErrorR R: t AXFRR^ t Mpackert addHeadert addQuestiont Classt INt getbuft requestt sendUDPRequestt sendTCPRequestRA Rl R R5 R3 t response( R R9 R7 R R R R R t qnamet mt reason( ( s, /usr/lib/python2.7/site-packages/DNS/Base.pyt req� sH
#
c C s� d } d | _ xw| D]o| _ y| j j d � rk t t d � re t j re | j t j t j � q� w n | j t j
t j � z� t j � | _ | j
� | j s| j j | j � | j � } x< | j d | j k s� | j d | j k r| j � } q� W| | _ n Wd | j s-| j j � n XWnF t j k
rV} | pM| } q n" t k
rw} | pn| } q n X| j r Pq q W| j r�| r�| � n d S( s refactor met :t has_ipv6t idi N( R3 R� RM RW t hasattrRA R� RE t AF_INET6t
SOCK_DGRAMt AF_INETRK RT Rq R5 RB t sendR~ RR Ra R8 RJ R R` Rl R ( R R t first_socket_errorRO RQ t t( ( s, /usr/lib/python2.7/site-packages/DNS/Base.pyR � s>
c C s� d } d | _ xf| D]^| _ y� | j j d � rk t t d � re t j re | j t j t j � q� w n | j t j
t j � z� t j � | _ | j
� t j t | j � � | j } | j j d � | j j | � | j � } | j d | j k r | | _ Pn Wd | j j � XWnF t j k
rE} | p<| } q n" t k
rf} | p]| } q n X| j r Pq q W| j r�| r�| � n d S( s& do the work of sending a TCP request R� R� i R� N( R3 R� RM RW R� RA R� RE R� t SOCK_STREAMR� RK RT Rq R^ t pack16bitR# R~ RB t setblockingt sendallRb Ra R8 R` Rl R ( R R R� RZ RO RQ R� ( ( s, /usr/lib/python2.7/site-packages/DNS/Base.pyR� s<
( R R t __doc__R R6 RE RR R[ Rb RN Ro Rq R� R R� ( ( ( s, /usr/lib/python2.7/site-packages/DNS/Base.pyR2 P s
. .t DnsAsyncRequestc B sM e Z d Z d � Z d � Z d � Z d � Z d � Z d � Z d � Z RS( s> an asynchronous request object. out of date, probably broken c O sU t j | | | � | j d � r< | d r< | d | _ n | j | _ d | _ d S( Nt donei ( R2 R R: R4 t
showResultR5 ( R R9 R7 ( ( s, /usr/lib/python2.7/site-packages/DNS/Base.pyR L s
c C sh | j � | j | j | j f � t j � | _ | j j d � rd | j d rd t j j
| � n d S( Nt start( Ro Rp RM R RK RT R7 R: t asyncoret
dispatchert go( R ( ( s, /usr/lib/python2.7/site-packages/DNS/Base.pyRq U s
c C s- | j | | � t j j | � | | _ d S( N( t
create_socketR� R� R RB ( R RC RD ( ( s, /usr/lib/python2.7/site-packages/DNS/Base.pyRE [ s c C sH | j d d k rD | j � | _ | j rD t | j | f � qD n d S( NR R
( R7 RR R� R4 t apply( R ( ( s, /usr/lib/python2.7/site-packages/DNS/Base.pyt handle_read_ s c C s | j | j � d S( N( R� R~ ( R ( ( s, /usr/lib/python2.7/site-packages/DNS/Base.pyt handle_connectd s c C s d S( N( ( R ( ( s, /usr/lib/python2.7/site-packages/DNS/Base.pyt handle_writef s c G s | j j � d S( N( R� t show( R RB ( ( s, /usr/lib/python2.7/site-packages/DNS/Base.pyR� h s (
R R R� R Rq RE R� R� R� R� ( ( ( s, /usr/lib/python2.7/site-packages/DNS/Base.pyR� J s ( ( R� RA R R< RK RG Rt R{ t OpcodeR� Ri R t ExceptionR R R R R R R^ t QUERYt AR$ R* R1 R2 t dispatcher_with_sendR� ( ( ( s, /usr/lib/python2.7/site-packages/DNS/Base.pyt <module>
s. <$
�