o jE5@sddlmZmZmZmZddlZddlZddlZddl Zddl Zddl Zddl Zddl ZddlZddlZddlZddlZddlZddlZddlZddlZddlZGdddejjZGdddejjZGdddejjZGd d d Zddddddddejjf d ej j!d e"dBd e"e#BdBde"dBde"dBde"dBdeej$j%dBdedej&j'dBdej&j'e(Bdeej)j*e"dBffddZ+dej)j,de"dBfddZ-dS))AnyListTuplecastNcs eZdZdZfddZZS) TransferErrorz.A zone transfer response got a non-zero rcode.cs(dtj|}t|||_dS)NzZone transfer error: )dnsrcodeto_textsuper__init__)selfrmessage __class__z/root/aizidognhua/tmp/workspace/projects/ec89d86c-575f-41c9-af57-ac45cbdbf775/venv/lib/python3.10/site-packages/dns/xfr.pyr *s  zTransferError.__init__)__name__ __module__ __qualname____doc__r __classcell__rrrrr'src@eZdZdZdS)SerialWentBackwardsz:The current serial number is less than the serial we know.Nrrrrrrrrr0rc@r)UseTCPz'This IXFR cannot be completed with UDP.Nrrrrrr4rrc @sheZdZdZejjddfdejjdejj de dBde fdd Z d ej jd e fd d ZddZddZdS)Inboundz+ State machine for zone transfers. NF txn_managerrdtypeserialis_udpcCs||_d|_||_|tjjkr|durtdd|_n|tjjkr+|r'tdd|_ntd||_ ||_ | \}}|_ d|_ d|_d|_d|_dS)aqInitialize an inbound zone transfer. *txn_manager* is a :py:class:`dns.transaction.TransactionManager`. *rdtype* can be `dns.rdatatype.AXFR` or `dns.rdatatype.IXFR` *serial* is the base serial number for IXFRs, and is required in that case. *is_udp*, a ``bool`` indidicates if UDP is being used for this XFR. Nz,a starting serial must be supplied for IXFRsTzis_udp specified for AXFRFzrdtype is not IXFR or AXFR)rtxnrr rdatatypeIXFR ValueError incrementalAXFRrr origin_informationorigin soa_rdatasetdone expecting_SOA delete_mode)r rrrr _rrrr =s&   zInbound.__init__r returncCsF|jdur|j|j |_|}|tjjkrt|t|j dkrA|j dj |j kr2tj d|j dj|jkrAtj dd}|jdur|jrT|jdj |j krZtj d|jd}|}|jtjjkrntj dd}||_|jr|jdusJttjjjj|jd}|j|jkrd|_ntj|j|jkrt|jrt|j|ddkrtd|_|j|dD]}|j }|}|jrtj d |jdusJ|jtjjkre||j kre|jr|j |_||jkr.|jr|jr.ttjjjj|d}|jr tj d |jr|j|jkrtj d |j|||j d|_d|_n6d |_ttjjjj|d}|jr^|jrR|j|jkrQtj d n|j|_|j||ntj dq|jr~d |_d |_d |_|j!|jd|_|jr|j"||q|j#||q|jr|jstj d|jS)aZProcess one message in the transfer. The message should have the same relativization as was specified when the `dns.xfr.Inbound` was created. The message should also have been created with `one_rr_per_rrset=True` because order matters. Returns `True` if the transfer is complete, and `False` otherwise. Nrzwrong question namezwrong question rdatatypez&No answer or RRset not for zone originzfirst RRset is not an SOATzanswers after final SOAzempty IXFR sequencezunexpected end of IXFR sequenceFzIXFR base serial mismatchzunexpected origin SOA in AXFRzunexpected end of UDP IXFR)$r!rwriterr%rrNOERRORrlenquestionnamer( exception FormErrorrr)answerr"SOAcopyrrdtypesANYrr*Serialrr rr+r,replacecommitrollback delete_exactadd)r r r answer_indexrrsetrdatasetsoar4rrrprocess_messagees                   zInbound.process_messagecCs|S)Nr)r rrr __enter__szInbound.__enter__cCs|jr|jdS)NF)r!r?)r exc_typeexc_valexc_tbrrr__exit__s zInbound.__exit__)rrrrrr"r& transactionTransactionManager RdataTypeintboolr r MessagerFrGrKrrrrr8s$ ( rrruse_edns ednsflagspayloadrequest_payloadoptionskeyringkeyname keyalgorithmr.c  Csb|\} } } | durtd|durtjj} nTt|ts!td|dkr[|)}|| d}|rEt tj j j j |d}|j }tjj} nd}tjj} Wdn1sUwYn|dkrh|dkrhtjj} ntd|}tj| | ||d|||| }|durtj|dd |d }|j|j| |tjj d d }||d|dur|j||| d ||fS)aMake an AXFR or IXFR query. *txn_manager* is a ``dns.transaction.TransactionManager``, typically a ``dns.zone.Zone``. *serial* is an ``int`` or ``None``. If 0, then IXFR will be attempted using the most recent serial number from the *txn_manager*; it is the caller's responsibility to ensure there are no write transactions active that could invalidate the retrieved serial. If a serial cannot be determined, AXFR will be forced. Other integer values are the starting serial to use. ``None`` forces an AXFR. Please see the documentation for :py:func:`dns.message.make_query` and :py:func:`dns.message.Message.use_tsig` for details on the other parameters to this function. Returns a `(query, serial)` tuple. Nzno zone originzserial is not an integerrr8lzserial out-of-rangeFz. . z 0 0 0 0T)create) algorithm)r'r$rr"r& isinstancerOreadergetrr:r;r8rr# get_classr make_queryrdata from_text find_rrset authorityrAuse_tsig)rrrRrSrTrUrVrWrXrY zone_originr-r(rr!rDrErdclassqrarCrrrr`sV        r`querycCs~t|tjjs td|jd}|jtjjkrdS|jtjj kr$td| |j |j |j tjj}ttjjjj|d}|jS)a7Extract the SOA serial number from query if it is an IXFR and return it, otherwise return None. *query* is a dns.message.QueryMessage that is an IXFR or AXFR request. Raises if the query is not an IXFR or AXFR, or if an IXFR doesn't have an appropriate SOA RRset in the authority section. zquery not a QueryMessagerNzquery is not an AXFR or IXFR)r\rr QueryMessager$r3rr"r&r#rcrdr4rgr8rr:r;r)rir3 soa_rrsetrErrrextract_serial_from_queryPs  rl).typingrrrrdns.ednsr dns.exception dns.messagedns.name dns.rcode dns.rdata dns.rdataset dns.rdatatype dns.rdtypesdns.rdtypes.ANYdns.rdtypes.ANY.SMIMEAdns.rdtypes.ANY.SOAdns.rdtypes.svcbbase dns.serialdns.transactiondns.tsigdns.zoner5 DNSExceptionrr6rrrtsigdefault_algorithmrLrMrOrPednsOptionr4Namestrr rjr`rQrlrrrrsp P    K