ZerethShell
File Manager
SQL Manager
/
usr
/
lib
/
python3.6
/
site-packages
/
dns
/
__pycache__
message.cpython-36.pyc
3 �@�Wy� � @ s� d Z ddlmZ ddlmZ ddlZddlZddlZddl Zddl Zddl Zddl Zddl ZddlZddlZddlZddlZddlZddlZddlZddlZddlmZmZmZ G dd� dejj�Z G d d � d ejj�Z G d d � d ejj �Z G d d� dejj�Z G dd� dejj�Z!G dd� dejj �Z"G dd� de#�Z$G dd� de#�Z%d(dd�Z&G d d � d e#�Z'd d � Z(d d!� Z)ej*j+ddddddfd"d#�Z,d)d&d'�Z-dS )*z DNS Messages� )�absolute_import)�StringION� )�long�xrange� string_typesc @ s e Zd ZdZdS )� ShortHeaderz2The DNS packet passed to from_wire() is too short.N)�__name__� __module__� __qualname__�__doc__� r r � /usr/lib/python3.6/message.pyr * s r c @ s e Zd ZdZdS )� TrailingJunkzEThe DNS packet passed to from_wire() has extra junk at the end of it.N)r r r r r r r r r / s r c @ s e Zd ZdZdS )�UnknownHeaderFieldzVThe header field name was not recognized when converting from text into a message.N)r r r r r r r r r 4 s r c @ s e Zd ZdZdS )�BadEDNSzVOPT record occurred somewhere other than the start of the additional data section.N)r r r r r r r r r : s r c @ s e Zd ZdZdS )�BadTSIGzWA TSIG record occurred somewhere other than the end of the additional data section.N)r r r r r r r r r @ s r c @ s e Zd ZdZdS )�UnknownTSIGKeyz(A TSIG with an unknown key was received.N)r r r r r r r r r F s r c @ s� e Zd ZdZd-dd�Zdd� Zdd� Zd.d d �Zd d � Zdd� Z dd� Z dd� Z e j jdddfdd�Ze j jdddfdd�Zd/dd�Zdd ddd e jjfd d �Zd0d!d"�Zd1d#d$�Zd%d&� Zd'd(� Zd)d*� Zd+d,� ZdS )2�Messagea, A DNS message. @ivar id: The query id; the default is a randomly chosen id. @type id: int @ivar flags: The DNS flags of the message. @see: RFC 1035 for an explanation of these flags. @type flags: int @ivar question: The question section. @type question: list of dns.rrset.RRset objects @ivar answer: The answer section. @type answer: list of dns.rrset.RRset objects @ivar authority: The authority section. @type authority: list of dns.rrset.RRset objects @ivar additional: The additional data section. @type additional: list of dns.rrset.RRset objects @ivar edns: The EDNS level to use. The default is -1, no Edns. @type edns: int @ivar ednsflags: The EDNS flags @type ednsflags: long @ivar payload: The EDNS payload size. The default is 0. @type payload: int @ivar options: The EDNS options @type options: list of dns.edns.Option objects @ivar request_payload: The associated request's EDNS payload size. @type request_payload: int @ivar keyring: The TSIG keyring to use. The default is None. @type keyring: dict @ivar keyname: The TSIG keyname to use. The default is None. @type keyname: dns.name.Name object @ivar keyalgorithm: The TSIG algorithm to use; defaults to dns.tsig.default_algorithm. Constants for TSIG algorithms are defined in dns.tsig, and the currently implemented algorithms are HMAC_MD5, HMAC_SHA1, HMAC_SHA224, HMAC_SHA256, HMAC_SHA384, and HMAC_SHA512. @type keyalgorithm: string @ivar request_mac: The TSIG MAC of the request message associated with this message; used when validating TSIG signatures. @see: RFC 2845 for more information on TSIG fields. @type request_mac: string @ivar fudge: TSIG time fudge; default is 300 seconds. @type fudge: int @ivar original_id: TSIG original id; defaults to the message's id @type original_id: int @ivar tsig_error: TSIG error code; default is 0. @type tsig_error: int @ivar other_data: TSIG other data. @type other_data: string @ivar mac: The TSIG MAC for this message. @type mac: string @ivar xfr: Is the message being used to contain the results of a DNS zone transfer? The default is False. @type xfr: bool @ivar origin: The origin of the zone in messages which are used for zone transfers or for DNS dynamic updates. The default is None. @type origin: dns.name.Name object @ivar tsig_ctx: The TSIG signature context associated with this message. The default is None. @type tsig_ctx: hmac.HMAC object @ivar had_tsig: Did the message decoded from wire format have a TSIG signature? @type had_tsig: bool @ivar multi: Is this message part of a multi-message sequence? The default is false. This variable is used when validating TSIG signatures on messages which are part of a zone transfer. @type multi: bool @ivar first: Is this message standalone, or the first of a multi message sequence? This variable is used when validating TSIG signatures on messages which are part of a zone transfer. @type first: bool @ivar index: An index of rrsets in the message. The index key is (section, name, rdclass, rdtype, covers, deleting). Indexing can be disabled by setting the index to None. @type index: dict Nc C s� |d krt jj� | _n|| _d| _g | _g | _g | _g | _d| _ d| _ d| _ g | _ d| _ d | _d | _t jj| _d| _d| _d| _d| _| j| _d| _d| _d | _d | _d| _ d| _ d| _ i | _ d S )Nr r � i, FT���) �dnsZentropyZ random_16�id�flags�question�answer� authority� additional�edns� ednsflags�payload�options�request_payload�keyring�keyname�tsig�default_algorithm� keyalgorithm� request_mac� other_data� tsig_error�fudge� original_id�mac�xfr�origin�tsig_ctx�had_tsig�multi�first�index)�selfr r r r �__init__� s: zMessage.__init__c C s dt | j� d S )Nz<DNS message, ID �>)�reprr )r5 r r r �__repr__� s zMessage.__repr__c C s | j � S )N)�to_text)r5 r r r �__str__� s zMessage.__str__Tc K s t � }|jd| j � |jdtjjtjj| j�� � tjj| j| j �}|jdtjj|� � |jdtjj| j� � | j dkr�|jd| j � | j dkr�|jdtjj | j � � |jd| j � tjj | j�}|r�|jd � n |jd � x.| jD ]$}|j|j||f|�� |jd � q�W |�r,|jd � n |jd � x0| jD ]&}|j|j||f|�� |jd � �q>W |�rz|jd� n |jd� x0| jD ]&}|j|j||f|�� |jd � �q�W |jd� x0| jD ]&}|j|j||f|�� |jd � �q�W |j� dd� S )z�Convert the message to text. The I{origin}, I{relativize}, and any other keyword arguments are passed to the rrset to_wire() method. @rtype: string zid %d z opcode %s z rcode %s z flags %s r zedns %s z eflags %s z payload %d z;ZONE z ;QUESTION � z;PREREQ z;ANSWER z;UPDATE z ;AUTHORITY z ;ADDITIONAL Nr r )r �writer r �opcoder: � from_flagsr �rcoder r Z edns_to_textr � is_updater r r r �getvalue)r5 r/ � relativize�kw�sZrcrA �rrsetr r r r: � sJ zMessage.to_textc C s� t |t�sdS | j|jkr dS | j|jkr.dS x | jD ]}||jkr6dS q6W x |jD ]}|| jkrTdS qTW x | jD ]}||jkrrdS qrW x |jD ]}|| jkr�dS q�W x | jD ]}||jkr�dS q�W x |jD ]}|| jkr�dS q�W dS )z�Two messages are equal if they have the same content in the header, question, answer, and authority sections. @rtype: boolFT)� isinstancer r r r r r )r5 �other�nr r r �__eq__� s2 zMessage.__eq__c C s | j |� S )z0Are two messages not equal? @rtype: bool)rJ )r5 rH r r r �__ne__ s zMessage.__ne__c C s� |j tj j@ dks:| j|jks:tjj| j �tjj|j �kr>dS tjj|j |j�tjjkr\dS tjj | j �rndS x | j D ]}||j krvdS qvW x |j D ]}|| j kr�dS q�W dS )z1Is other a response to self? @rtype: boolr FT) r r �QRr r> r? r@ r ZNOERRORrA r )r5 rH rI r r r � is_response s" zMessage.is_responsec C sD || j krdS || jkr dS || jkr*dS || jkr8dS td��d S )Nr r � � zunknown section)r r r r � ValueError)r5 �sectionr r r �section_number- s zMessage.section_numberFc C s� | j |�|||||f} |sb| jdk r>| jj| �} | dk rb| S n$x"|D ]} | j|||||�rD| S qDW |sjt�tjj|||||�} |j| � | jdk r�| | j| < | S )a Find the RRset with the given attributes in the specified section. @param section: the section of the message to look in, e.g. self.answer. @type section: list of dns.rrset.RRset objects @param name: the name of the RRset @type name: dns.name.Name object @param rdclass: the class of the RRset @type rdclass: int @param rdtype: the type of the RRset @type rdtype: int @param covers: the covers value of the RRset @type covers: int @param deleting: the deleting value of the RRset @type deleting: int @param create: If True, create the RRset if it is not found. The created RRset is appended to I{section}. @type create: bool @param force_unique: If True and create is also True, create a new RRset regardless of whether a matching RRset exists already. @type force_unique: bool @raises KeyError: the RRset was not found and create was False @rtype: dns.rrset.RRset objectN) rR r4 �get�match�KeyErrorr rF ZRRset�append) r5 rQ �name�rdclass�rdtype�covers�deleting�create� force_unique�keyrF r r r � find_rrset9 s" zMessage.find_rrsetc C s: y | j ||||||||�} W n tk r4 d} Y nX | S )a Get the RRset with the given attributes in the specified section. If the RRset is not found, None is returned. @param section: the section of the message to look in, e.g. self.answer. @type section: list of dns.rrset.RRset objects @param name: the name of the RRset @type name: dns.name.Name object @param rdclass: the class of the RRset @type rdclass: int @param rdtype: the type of the RRset @type rdtype: int @param covers: the covers value of the RRset @type covers: int @param deleting: the deleting value of the RRset @type deleting: int @param create: If True, create the RRset if it is not found. The created RRset is appended to I{section}. @type create: bool @param force_unique: If True and create is also True, create a new RRset regardless of whether a matching RRset exists already. @type force_unique: bool @rtype: dns.rrset.RRset object or NoneN)r_ rU ) r5 rQ rW rX rY rZ r[ r\ r] rF r r r � get_rrsetg s zMessage.get_rrsetr c K sR |dkr | j dkr| j }nd}|dk r,d}n |dkr8d}tjj| j| j||�}x"| jD ]}|j|j|j |j � qVW x"| j D ]}|j tjj |f|� qzW x"| jD ]}|j tjj|f|� q�W | jdkr�|j| j| j| j| j� x"| jD ]}|j tjj|f|� q�W |j� | jdk �rJ|j| j| j| j | j| j | j | j | j | j � |j!| _!|j"� S )a7 Return a string containing the message in DNS compressed wire format. Additional keyword arguments are passed to the rrset to_wire() method. @param origin: The origin to be appended to any relative names. @type origin: dns.name.Name object @param max_size: The maximum size of the wire format output; default is 0, which means 'the message's request payload, if nonzero, or 65536'. @type max_size: int @raises dns.exception.TooBig: max_size was exceeded @rtype: string r i�� i N)#r" r ZrendererZRendererr r r Z add_questionrW rY rX r Z add_rrset�ANSWERr � AUTHORITYr Zadd_ednsr r r! r � ADDITIONALZ write_headerr$ Zadd_tsigr# r+ r, r* r) r( r' r- Zget_wire)r5 r/ Zmax_sizerD �rrF r r r �to_wire� s6 zMessage.to_wirei, r c C st || _ |dkr$t| j j� �d | _n t|t�r:tjj|�}|| _|| _ || _ |dkr^| j | _ n|| _ || _ || _dS )aw When sending, a TSIG signature using the specified keyring and keyname should be added. @param keyring: The TSIG keyring to use; defaults to None. @type keyring: dict @param keyname: The name of the TSIG key to use; defaults to None. The key must be defined in the keyring. If a keyring is specified but a keyname is not, then the key used will be the first key in the keyring. Note that the order of keys in a dictionary is not defined, so applications should supply a keyname when a keyring is used, unless they know the keyring contains only one key. @type keyname: dns.name.Name or string @param fudge: TSIG time fudge; default is 300 seconds. @type fudge: int @param original_id: TSIG original id; defaults to the message's id @type original_id: int @param tsig_error: TSIG error code; default is 0. @type tsig_error: int @param other_data: TSIG other data. @type other_data: string @param algorithm: The TSIG algorithm to use; defaults to dns.tsig.default_algorithm Nr )r# �list�keysr$ rG r r rW � from_textr' r+ r r, r* r) )r5 r# r$ r+ r, r* r) � algorithmr r r �use_tsig� s zMessage.use_tsig� c C s� |dks|dkrd}|dkr d}|dkr,|}|dk rFd}d}d}g }n$|t d�M }||d> O }|dkrjg }|| _|| _|| _|| _|| _dS ) a_ Configure EDNS behavior. @param edns: The EDNS level to use. Specifying None, False, or -1 means 'do not use EDNS', and in this case the other parameters are ignored. Specifying True is equivalent to specifying 0, i.e. 'use EDNS0'. @type edns: int or bool or None @param ednsflags: EDNS flag values. @type ednsflags: int @param payload: The EDNS sender's payload field, which is the maximum size of UDP datagram the sender can handle. @type payload: int @param request_payload: The EDNS payload size to use when sending this message. If not specified, defaults to the value of payload. @type request_payload: int or None @param options: The EDNS options @type options: None or list of dns.edns.Option objects @see: RFC 2671 NFr Tr l �~ � r )r r r r r! r" )r5 r r r r" r! r r r �use_edns� s( zMessage.use_ednsc C sL |r*| j dk r| j� | jtjjO _n | j dkrH| jtjj M _dS )a Enable or disable 'DNSSEC desired' flag in requests. @param wanted: Is DNSSEC desired? If True, EDNS is enabled if required, and then the DO bit is set. If False, the DO bit is cleared if EDNS is enabled. @type wanted: bool r N)r rm r r r ZDO)r5 Zwantedr r r � want_dnssec s zMessage.want_dnssecc C s t jj| j| j�S )z.Return the rcode. @rtype: int )r r@ r? r r )r5 r r r r@ s z Message.rcodec C sj t jj|�\}}| jdM _| j|O _| jtd�M _| j|O _| jdkrf| jdk rfd| _dS )zPSet the rcode. @param rcode: the rcode @type rcode: int i� i��� r N)r r@ �to_flagsr r r r )r5 r@ �valueZevaluer r r � set_rcode" s zMessage.set_rcodec C s t jj| j�S )z/Return the opcode. @rtype: int )r r> r? r )r5 r r r r> / s zMessage.opcodec C s( | j dM _ | j tjj|�O _ dS )zTSet the opcode. @param opcode: the opcode @type opcode: int i�� N)r r r> ro )r5 r> r r r � set_opcode5 s zMessage.set_opcode)N)NT)Nr )r r rk NN)T)r r r r r6 r9 r; r: rJ rK rM rR r � rdatatype�NONEr_ r` re r% r& rj rm rn r@ rq r> rr r r r r r K s0 K 7 - ! .) * r c @ s2 e Zd ZdZd dd�Zdd� Zdd� Zd d � Zd S ) � _WireReadera� Wire format reader. @ivar wire: the wire-format message. @type wire: string @ivar message: The message object being built @type message: dns.message.Message object @ivar current: When building a message object from wire format, this variable contains the offset from the beginning of wire of the next octet to be read. @type current: int @ivar updating: Is the message a dynamic update? @type updating: bool @ivar one_rr_per_rrset: Put each RR into its own RRset? @type one_rr_per_rrset: bool @ivar ignore_trailing: Ignore trailing junk at end of request? @type ignore_trailing: bool @ivar zone_rdclass: The class of the zone in messages which are DNS dynamic updates. @type zone_rdclass: int Fc C s@ t jj|�| _|| _d| _d| _t jj| _ || _ || _ || _ d S )Nr F) r ZwiredataZ maybe_wrap�wire�message�current�updating� rdataclass�IN� zone_rdclass� question_only�one_rr_per_rrset�ignore_trailing)r5 rv rw r} r~ r r r r r6 U s z_WireReader.__init__c C s� | j r|dkrtjj�x�td|�D ]�}tjj| j| j�\}}| j j dk rV|j | j j �}| j| | _t j d| j| j| jd � �\}}| jd | _| j j| j j|||ddd� | j r"|| _q"W dS )z�Read the next I{qcount} records from the wire data and add them to the question section. @param qcount: the number of questions in the message @type qcount: intr r Nz!HH� T)r\ r] )ry r � exception� FormErrorr rW � from_wirerv rx rw r/ rC �struct�unpackr_ r r| )r5 �qcount�i�qname�usedrY rX r r r � _get_question` s z_WireReader._get_questionc C sH | j s | jrd}nd}d}�x&td|�D �]}| j}tjj| j| j�\}}|} | jj dk rh|j | jj �}| j| | _t j d| j| j| jd � �\} } } } | jd | _| tj jk�rn|| jjk s�|r�t�| | j_| | j_| d@ d? | j_g | j_| j}| }xj|dk�rdt j d | j||d � �\}}|d }tjj|| j||�}| jjj|� || }|d | }q�W d}�n�| tj jk�rH|| jjk�o�||d k�s�t�| jjdk�r�td ��| jjj| �}|dk�r�td | ��| | j_ tj j | j| j| �\| j_ | j_ tj j!| j| |t"t#j#� �| jj$|| j| | jj%| jj&| jj'� | j_%d| j_(n�| dk �rVd} | j �r�| tj)j*k�sz| tj)j+k�r�| }| j,} nd}|tj)j*k�s�|tj)j+k�r�|| jj-k�r�tj j+}d}n&tj.j| | | j| j| | jj �}|j/� }| jj0�r| tj j1k�rd}| jj2||| | ||d|�}|dk �r4|j3|| � | j| | _q(W dS )a/ Read the next I{count} records from the wire data and add them to the specified section. @param section: the section of the message to which to add records @type section: list of dns.rrset.RRset objects @param count: the number of records to read @type count: intTFr Nz!HHIH� i � rl z!HHr� r z"got signed message without keyringzkey '%s' unknown)4ry r~ r rx r rW r� rv rw r/ rC r� r� rs �OPTr r r r r r! Zoption_from_wirerV ZTSIGr r# r rS r$ r% Zget_algorithm_and_macr' r- Zvalidate�int�timer( r0 r2 r3 r1 rz �ANYrt r| r �rdatarZ r. ZSOAr_ �add)r5 rQ �countr] Zseen_optr� Zrr_startrW r� Z absolute_namerY rX �ttlZrdlenrx ZoptslenZotypeZolen�optZsecretr[ rZ �rdrF r r r � _get_sectionx s� z_WireReader._get_sectionc C s� t | j�}|dk rt�tjd| jdd� �\| j_| j_}}}}d| _t j j | jj�r\d| _ | j |� | jrpdS | j| jj|� | j| jj|� | j| jj|� | j r�| j|kr�t�| jjr�| jjr�| jj r�| jjj| j� dS )zNRead a wire format DNS message and build a dns.message.Message object.� z!HHHHHHNT)�lenrv r r� r� rw r r rx r r> rA ry r� r} r� r r r r r r2 r0 r1 �update)r5 �lr� ZancountZaucountZadcountr r r �read� s$ * z_WireReader.readN)FFF)r r r r r6 r� r� r� r r r r ru > s cru r FTc C sP t dd�} || _|| _|| _|| _|| _|| _|| _t| | || | �} | j � | S )a) Convert a DNS wire format message into a message object. @param keyring: The keyring to use if the message is signed. @type keyring: dict @param request_mac: If the message is a response to a TSIG-signed request, I{request_mac} should be set to the MAC of that request. @type request_mac: string @param xfr: Is this message part of a zone transfer? @type xfr: bool @param origin: If the message is part of a zone transfer, I{origin} should be the origin name of the zone. @type origin: dns.name.Name object @param tsig_ctx: The ongoing TSIG context, used when validating zone transfers. @type tsig_ctx: hmac.HMAC object @param multi: Is this message part of a multiple message sequence? @type multi: bool @param first: Is this message standalone, or the first of a multi message sequence? @type first: bool @param question_only: Read only up to the end of the question section? @type question_only: bool @param one_rr_per_rrset: Put each RR into its own RRset @type one_rr_per_rrset: bool @param ignore_trailing: Ignore trailing junk at end of request? @type ignore_trailing: bool @raises ShortHeader: The message is less than 12 octets long. @raises TrailingJunk: There were octets in the message past the end of the proper DNS message. @raises BadEDNS: An OPT record was in the wrong section, or occurred more than once. @raises BadTSIG: A TSIG record was not the last record of the additional data section. @rtype: dns.message.Message objectr )r ) r r# r( r. r/ r0 r2 r3 ru r� ) rv r# r( r. r/ r0 r2 r3 r} r~ r �m�readerr r r r� � s ( r� c @ s8 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd d � Zd S ) � _TextReadera Text format reader. @ivar tok: the tokenizer @type tok: dns.tokenizer.Tokenizer object @ivar message: The message object being built @type message: dns.message.Message object @ivar updating: Is the message a dynamic update? @type updating: bool @ivar zone_rdclass: The class of the zone in messages which are DNS dynamic updates. @type zone_rdclass: int @ivar last_name: The most recently read name when building a message object from text format. @type last_name: dns.name.Name object c C s. || _ tjj|�| _d | _tjj| _d| _ d S )NF) rw r Z tokenizer� Tokenizer�tok� last_namerz r{ r| ry )r5 �textrw r r r r6 > s z_TextReader.__init__c C s� | j j� }|j}|dkr*| j j� | j_�n�|dkr�x>| j j� }|j� sT| j j|� P | jjt jj |j�B | j_q4W t j j | jj�r�d| _ �n&|dkr�| j j� | j_| jj| jjd> B | j_n�|dk�r | jjdk r�d| j_x�| j j� }|j� � s�| j j|� P | jjt jj|j�B | j_q�W n�|dk�rN| j j� | j_| jjdk �r�d| j_nd|d k�r�| j j� }| jjt j jt j j |��B | j_n.|d k�r�| j j� }| jjt jj |�� nt�| j j� d S ) z5Process one line from the text format header section.r r Tr rl Zeflagsr r r> r@ N)r� rS rp Zget_intrw r � is_identifier�ungetr r rh r> rA ry r r Zedns_from_textr Z get_stringro rq r@ r �get_eol)r5 rQ �tokenZwhatr� r r r � _header_lineE sR z_TextReader._header_linec C s� | j jdd�}|j� s(tjj|jd�| _| j}| j j� }|j� sHtj j �y,tj j|j�}| j j� }|j� srtj j �W n: tj j k r� tj j �Y n t k r� tj j }Y nX tjj|j�}| jj| jj|||ddd� | jr�|| _| j j� dS )z7Process one line from the text format question section.T)� want_leadingN)r\ r] )r� rS � is_whitespacer rW rh rp r� r� r� � SyntaxErrorrz � Exceptionr{ rs rw r_ r ry r| r� )r5 rQ r� rW rX rY r r r �_question_lines s. z_TextReader._question_linec C s� d}| j jdd�}|j� s,tjj|jd�| _| j}| j j� }|j� sLtj j �y*t |jd�}| j j� }|j� sttj j �W n6 tj j k r� tj j �Y n t k r� d}Y nX yPtj j|j�}| j j� }|j� s�tj j �|tj jks�|tj jk� r�|}| j}W n> tj j k �r tj j �Y n t k �r< tj j}Y nX tjj|j�}| j j� }|j� �s�| j j|� tjj||| j d�}|j� } n d}tjj} | jj||||| |d| j�} |dk �r�| j||� dS )zfProcess one line from the text format answer, authority, or additional data sections. NT)r� r )r� rS r� r rW rh rp r� r� r� r� r� r� rz r� rt r| r{ rs � is_eol_or_eofr� r� rZ rw r_ ry r� ) r5 rQ r[ r� rW r� rX rY r� rZ rF r r r �_rr_line� sT z_TextReader._rr_linec C s� | j }d}x�| jjdd�}|j� r$P |j� r�|jj� }|dkrF| j }nv|dksV|dkrf| j}| jj }nV|dksv|dkr�| j }| jj }n6|dks�|d kr�| j }| jj }n|d kr�| j }| jj }| jj� q | jj|� ||� q W dS ) zNRead a text format DNS message and build a dns.message.Message object.NTZHEADERZQUESTIONZZONEra ZPREREQrb ZUPDATErc )r� r� rS r� Z is_commentrp �upperr� rw r r� r r r r� r� )r5 Z line_methodrQ r� �ur r r r� � s4 z_TextReader.readN) r r r r r6 r� r� r� r� r r r r r� , s . 5r� c C s t � }t| |�}|j� |S )z�Convert the text format message into a message object. @param text: The text format message. @type text: string @raises UnknownHeaderField: @raises dns.exception.SyntaxError: @rtype: dns.message.Message object)r r� r� )r� r� r� r r r rh � s rh c C sF t }d}t| |�r"t| |�} d}nd}z t| �}W d|r@| j� X |S )a Read the next text format message from the specified file. @param f: file or string. If I{f} is a string, it is treated as the name of a file to open. @raises UnknownHeaderField: @raises dns.exception.SyntaxError: @rtype: dns.message.Message objectZrUTFN)r rG �openrh �close)�fZstr_typeZoptsZ want_closer� r r r � from_file� s r� c C s t | t�rtjj| �} t |t�r,tjj|�}t |t�rBtjj|�}t� } | jtjj O _| j | j | ||ddd� i } |dk r�|| d< |dkr�d}|dk r�|| d< |dkr�d}|dk r�|| d< |dkr�d}|dk r�|| d< |dkr�d}|| d < | j f | � | j |� | S ) aV Make a query message. The query name, type, and class may all be specified either as objects of the appropriate type, or as strings. The query will have a randomly chosen query id, and its DNS flags will be set to dns.flags.RD. @param qname: The query name. @type qname: dns.name.Name object or string @param rdtype: The desired rdata type. @type rdtype: int @param rdclass: The desired rdata class; the default is class IN. @type rdclass: int @param use_edns: The EDNS level to use; the default is None (no EDNS). See the description of dns.message.Message.use_edns() for the possible values for use_edns and their meanings. @type use_edns: int or bool or None @param want_dnssec: Should the query indicate that DNSSEC is desired? @type want_dnssec: bool @param ednsflags: EDNS flag values. @type ednsflags: int @param payload: The EDNS sender's payload field, which is the maximum size of UDP datagram the sender can handle. @type payload: int @param request_payload: The EDNS payload size to use when sending this message. If not specified, defaults to the value of payload. @type request_payload: int or None @param options: The EDNS options @type options: None or list of dns.edns.Option objects @see: RFC 2671 @rtype: dns.message.Message objectT)r\ r] Nr r r r" r! r )rG r r rW rh rs rz r r �RDr_ r rm rn ) r� rY rX rm rn r r r" r! r� �kwargsr r r � make_query s> $ r� � �, c C s� | j tj j@ rtjjd��tjj| j�}tj j| j tj j@ B |_ |rV| j tj j O _ |j | j � � t | j �|_ | jdkr�|jdd|| j� | jr�|j| j| j|ddd| j� | j|_|S )a Make a message which is a response for the specified query. The message returned is really a response skeleton; it has all of the infrastructure required of a response, but none of the content. The response's question section is a shallow copy of the query's question section, so the query's question RRsets should not be changed. @param query: the query to respond to @type query: dns.message.Message object @param recursion_available: should RA be set in the response? @type recursion_available: bool @param our_payload: payload size to advertise in EDNS responses; default is 8192. @type our_payload: int @param fudge: TSIG time fudge; default is 300 seconds. @type fudge: int @rtype: dns.message.Message objectz&specified query message is not a queryr Nr )r r rL r� r� rw r r r� ZRArr r> rf r r rm r r1 rj r# r$ r' r- r( )ZqueryZrecursion_availableZ our_payloadr+ Zresponser r r � make_response\ s r� ) Nr FNNFTFFF)Fr� r� ).r Z __future__r �ior r� r� Zdns.ednsr Z dns.exceptionZ dns.flagsZdns.nameZ dns.opcodeZ dns.entropyZ dns.rcodeZ dns.rdataZdns.rdataclassZ dns.rdatatypeZ dns.rrsetZ dns.rendererZdns.tsigZ dns.wiredataZ_compatr r r r� r� r r Z DNSExceptionr r r r �objectr ru r� r� rh r� rz r{ r� r� r r r r �<module> sX v 7 5 :F
Kaydet
Vazgeç