o b^@sdZddlZddlZddlZddlmZmZddlmZddl m Z ddl m Z m Z mZddlmZddlmZmZmZmZmZdd lmZdd lmZmZmZmZmZmZdd l m!Z!dd l"m#Z#m$Z$dd l%m&Z&ddl'm(Z(ddl)m*Z*m+Z+m,Z,ddl-m.Z.ddZ/Gdddej0Z1ejdddddddddZ2ejdddd d!d"d#d$dZ3ejd%d&d'd(dd)dd*Z4e1de2fe2ed+e5d,e6d-e6d.d/e7d0d1ej8d2d3d4e9d5e:d6e;d7ee=d?d@dAedCej?dDdEdFej@ddGdHejAdIdJdKeBdLejCdMeDdd0dNdOdPeEdQgeFdRdSdTdUgedVedWgedXedYge9dgedZgd[d\ZGe1de3fd]eHd^e3gid\ZIe1d_e4fd_e4ejdZd`daej5dbdcdaej5dddedaejFdRdSdTdUdfdagid\ZJGdgdhdhe.jKZLGdidjdje.jKZMGdkdldle.jKZNGdmdndne.jKZOGdodpdpe.jKZPGdqdrdre.jKZQGdsdtdte.jKZRGdudvdve.jKZSGdwdxdxe.jKZTdyZUGdzd{d{e.jKZVd|ZWGd}d~d~e.jKZXdS)z Test cases for twisted.names. N)partialreduce)BytesIO)pack)defererrorreactor)succeed) authorityclientcommondnsserver)Resolver)SOAMessageQueryRecord_A Record_SOARRHeader) DomainError)SecondaryAuthoritySecondaryAuthorityService) nativeString)FilePath)MemoryReactorClockStringTransportwaitUntilAllDisconnected)unittestcCsdd|dDS)NcSsg|]}|jqSpayload.0rrr?/usr/lib/python3/dist-packages/twisted/names/test/test_names.py "zjustPayload..rrresultsrrr% justPayload!sr*c@eZdZddZdS)NoFileAuthoritycCstj||||_|_dSN)r ResolverBase__init__soarecords)selfr0r1rrr%r/&s zNoFileAuthority.__init__N)__name__ __module__ __qualname__r/rrrr%r,% r,test-domain.comzroot.test-domain.comdiH?&mnamernameserialrefreshminimumexpireretryttls93.84.28.in-addr.arpaxi1i~Ϊ my-domain.comspostmaster.test-domain.comi90i?B)r>r?r@rArBrCrDs 127.0.0.1s 39.28.189.39#v=spf1 mx/30 mx:example.org/30 -allv=spf1 +mx a:colo.example.com/28 -all not valid host.test-domain.comLinuxA Fast One, Dontcha know)oscpuscanonical.name.comsmailbox.test-domain.comsmail.group.someplaceA First piece of Texta SecoNd pieces ABCD::4321 s 0:0069::0ssome.network.tlds$0:5634:1294:AFCB:56AC:48EF:34C3:01FFs tra.la.la.net+Some more text, haha! Yes. Still here?smail.redirect.or.whatevers r mail boxs e mail box)rmailbxemailbxsafsdb.test-domain.com)subtypehostnameswhatever.i.dunnossome.more.text)mboxtxts 12.54.78.12usip+E2U!!^.*$!sip:information@domain.tld!s'AF43:5634:1294:AFCB:56AC:48EF:34C3:01FF?骪ssome.other.place.fool 123.242.1.5 0.255.0.255s255.255.255.254s0.0.0.0s1.2.3.4)r7shttp.tcp.test-domain.comhost.test-domain.comshost-two.test-domain.comscname.test-domain.comsanothertest-domain.comr0r1s123.93.84.28.in-addr.arpastest.host-reverse.lookup.coms my-domain.com1SrEs ns1.domains2Ms ns2.domain3H4Dc@seZdZdZddZddZddZdd Zd d Zd d Z ddZ ddZ ddZ ddZ ddZddZddZddZddZd d!Zd"d#Zd$d%Zd&d'Zd(d)Zd*d+Zd,d-Zd.d/Zd0d1Zd2d3Zd4d5Zd6d7Zd8d9Zd:d;Z d.checkResults) addCallback)r2queryingrrrrr% namesTests zServerDNSTests.namesTestcC ||jdtjdddgS)z"Test simple DNS 'A' record queriestest-domain.comrsr:rmrr lookupAddressr rr2rrr%test_addressRecord1 z"ServerDNSTests.test_addressRecord1cC,||jdtjdddtjdddgS)z1Test DNS 'A' record queries with multiple answersrPz 123.242.1.5r:rmz 0.255.0.255rrrrr%test_addressRecord2    z"ServerDNSTests.test_addressRecord2cCr)z+Test DNS 'A' record queries with edge caseszhost-two.test-domain.comz255.255.255.254r:rm0.0.0.0rrrrr%test_addressRecord3rz"ServerDNSTests.test_addressRecord3cCs||jdtgS)zTest DNS 'SOA' record queriesr)rrlookupAuthority soa_recordrrrr%test_authorityszServerDNSTests.test_authorityc Cs:||jdtjddddtjdddtjdddgS)z The DNS client can issue an MX query and receive a response including an MX record as well as any A record hints. r7rOrjr:rmrhri)rrlookupMailExchanger Record_MXrrrrr%test_mailExchangeRecords   z&ServerDNSTests.test_mailExchangeRecordcCr)zTest DNS 'NS' record queriesrz 39.28.189.39r:rm)rrlookupNameserversr Record_NSrrrr%test_nameserverrzServerDNSTests.test_nameservercC"||jdtjddddgS)zTest DNS 'HINFO' record queriesrrQrRr:)rSrTrE)rrlookupHostInfor Record_HINFOrrrr% test_HINFO zServerDNSTests.test_HINFOcCr)zTest DNS 'PTR' record queriesz123.93.84.28.in-addr.arpaztest.host-reverse.lookup.comrGrm)rr lookupPointerr Record_PTRrrrr%test_PTR rzServerDNSTests.test_PTRcCr)zTest DNS 'CNAME' record queriesrzcanonical.name.comr:rm)rrlookupCanonicalNamer Record_CNAMErrrr% test_CNAME'rzServerDNSTests.test_CNAMEcCr)zTest DNS 'MB' record queriesrzmailbox.test-domain.comr:rm)rr lookupMailBoxr Record_MBrrrr%test_MB.rzServerDNSTests.test_MBcCr)zTest DNS 'MG' record queriesrzmail.group.someplacer:rm)rrlookupMailGroupr Record_MGrrrr%test_MG5rzServerDNSTests.test_MGcCr)zTest DNS 'MR' record queriesrzmail.redirect.or.whateverr:rm)rrlookupMailRenamer Record_MRrrrr%test_MR<rzServerDNSTests.test_MRcCr)zTest DNS 'MINFO' record queriesrz r mail boxz e mail boxr:)r[r\rE)rrlookupMailboxInfor Record_MINFOrrrr% test_MINFOCrzServerDNSTests.test_MINFOc Cs&||jdtjddddddgS)zTest DNS 'SRV' record querieszhttp.tcp.test-domain.comrerfrgsome.other.place.foolr:rm)rr lookupServicer Record_SRVrrrr%test_SRVNs zServerDNSTests.test_SRVcCr)zTest DNS 'AFSDB' record queriesrr<zafsdb.test-domain.comr:)r]r^rE)rrlookupAFSDatabaser Record_AFSDBrrrr% test_AFSDBUrzServerDNSTests.test_AFSDBcCr)zTest DNS 'RP' record queriesrzwhatever.i.dunnozsome.more.textr:)r_r`rE)rrlookupResponsibilityr Record_RPrrrr%test_RP`rzServerDNSTests.test_RPcCs.||jdtjddddtjdddgS)zTest DNS 'TXT' record queriesrrUrVr:rmrZ)rr lookupTextr Record_TXTrrrr%test_TXTks zServerDNSTests.test_TXTc Cs.||jdtjdddtjddddgS)zH L{DNSServerFactory} can serve I{SPF} resource records. rrLr:rmrMrN)rrlookupSenderPolicyr Record_SPFrrrr%test_spfys  zServerDNSTests.test_spfc Cs&||jdtjdtjdddgS)zTest DNS 'WKS' record queriesrz 12.54.78.12rar:rm)rrlookupWellKnownServicesr Record_WKSsocket IPPROTO_TCPrrrr%test_WKSs zServerDNSTests.test_WKScCs^tt}tj|_||jd|tjdddtj dddtj dddtj d d d d d dgS)NrJ1.2.3.4rlrmz ns1.domain2Mz ns2.domainrnrerfrgrro) copymy_soarCrErrlookupAllRecordsr rrr)r2 result_soarrr%test_someRecordsWithTTLss     z'ServerDNSTests.test_someRecordsWithTTLscCr)z%Test DNS 'AAAA' record queries (IPv6)rz'AF43:5634:1294:AFCB:56AC:48EF:34C3:01FFr:rm)rrlookupIPV6Addressr Record_AAAArrrr% test_AAAArzServerDNSTests.test_AAAAc CsD||jdtjdddddtjddd ddtjd d d ddgS) z#Test DNS 'A6' record queries (IPv6)rrz ABCD::4321r:rmrXz 0:0069::0zsome.network.tldrYz$0:5634:1294:AFCB:56AC:48EF:34C3:01FFz tra.la.la.net)rrlookupAddress6r Record_A6rrrr%test_A6s zServerDNSTests.test_A6cCsZtj}ddttjtjD}|D] }|jdur||_q| |j d dd|S)z9 Test DNS 'AXFR' queries (Zone transfer) cSsg|]}t|qSr)rr"rrr%r&s z4ServerDNSTests.test_zoneTransfer..NrcSs|dddfS)Nrr)r$rrr%r'z2ServerDNSTests.test_zoneTransfer..) rrCroperatoraddrxr1valuesrErr lookupZoner)r2 default_ttlr)r$rrr%test_zoneTransfers  z ServerDNSTests.test_zoneTransfercCs tdgd}||dtjS)zu A failed AXFR TCP connection errbacks the L{Deferred} returned from L{Resolver.lookupZone}. )znameserver.invalid5ruzimpossible.invalid)r assertFailurerrDNSLookupError)r2rrrr% test_zoneTransferConnectionFailss  z/ServerDNSTests.test_zoneTransferConnectionFailscCr)z6Tests that unrelated zones don't mess with each other.zanothertest-domain.comrr:rmrrrrr%test_similarZonesDontInterfererz-ServerDNSTests.test_similarZonesDontInterferec Cs(||jdtjdddddddgS) z2 Test DNS 'NAPTR' record queries. rr8rOrbrcrdr:rm)rrlookupNamingAuthorityPointerr Record_NAPTRrrrr% test_NAPTRs zServerDNSTests.test_NAPTRN)!r3r4r5__doc__rrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrr%rps>        rpc@r+) HelperTestscCs@|}t|}tdD]}t|}|||k|}q dS)N)mktempr getSerialrange assertTrue)r2faibrrr%test_serialGenerators   z HelperTests.test_serialGeneratorN)r3r4r5rrrrr%rr6rc@s4eZdZddZddZddZddZd d Zd S) AXFRTestscCsd|_t|_|j|jtd|j|_t j dt j t j ddt j ddddddd d d d |_|jt j dt jt j d dt jdd dd t j dt jt j d dt jddd dd t j dt jt j d dt jdd dd |jg|_dS)N fooby.comiQFzhooj.fooby.comr8i,iiiX)r>r?r@rArDrCrBrEnametypeclsrEauthr!ns.twistedmatrix.comrrErOz mail.mv3d.com) preferenceexchangerEz 64.123.27.105)addressrE)r)rDeferreddr _gotResultsr AXFRController controllerr rrINrr0NSrMXrArr1rrrr%rs`    zAXFRTests.setUpc Cstjdddddddddd S)Nr<r idansweropCoderecDesrecAvr rCodetruncmaxSize)r rrrrr% _makeMessage2szAXFRTests._makeMessagecCsH|}tdtjtjg|_|j|_|j |d| |j |jdS)Nr) r$r rAXFRrqueriesr1answersrmessageReceivedrr))r2mrrr%test_bindAndTNamesStyle@s z!AXFRTests.test_bindAndTNamesStylecCs ||_dSr-r()r2resultrrr%rHs zAXFRTests._gotResultscCsT|jdd}|r |}g|_|dg|_|j|d|s ||j|jdSNr) r1r$r&popr'rr(rr))r2r1r)rrr% test_DJBStyleKszAXFRTests.test_DJBStyleN)r3r4r5rr$r*rr.rrrr%rs 5 rc@seZdZddZddZdS)ResolvConfHandlingTestscCs2|}tj|d}||jdg|jdS)Nresolvrsr)rr rr dynServers _parseCallcancelr2 resolvConfr$rrr% test_missingWs z$ResolvConfHandlingTests.test_missingcCs@|}t|dtj|d}||jdg|jdS)Nwr0r2) ropencloser rrr3r4r5r6rrr% test_empty]s  z"ResolvConfHandlingTests.test_emptyN)r3r4r5r8r<rrrr%r/Vs r/c@sHeZdZdZddZddZddZdd Zd d Zd d Z ddZ dS)AuthorityTestszs Tests for the basic response record selection code in L{FileAuthority} (independent of its fileness). cCs4t}d|jd}|||}||jtdS)aJ L{FileAuthority} lookup methods errback with L{DomainError} if the requested C{name} shares a common suffix with its zone but is not actually a descendant of its zone, in terms of its sequence of DNS name labels. eg www.the-example.com has nothing to do with the zone example.com. nonexistent.prefix-rN)rxr0failureResultOfrassertIsInstancevaluer)r2 testDomaintestDomainNamerrrr%'test_domainErrorForNameWithCommonSuffixksz6AuthorityTests.test_domainErrorForNameWithCommonSuffixc Csttjjtftjjttdgid}||tjj\}}}||g||tj tjjtj tj tddg||gdS)a! If a L{FileAuthority} has a zone which includes an I{NS} record for a particular name and that authority is asked for another record for the same name which does not exist, the I{NS} record is not included in the authority section of the response. rrkTrEr!r N) r,rr>rr rsuccessResultOfrrrTYPErC)r2r r additionalrrr%test_recordMissingxs0    z!AuthorityTests.test_recordMissingcCsJt}d|jd}ttjd}||t||d}||j t dS)z Requesting a record of unknown type where no records exist for the name in question results in L{DomainError}. r>rr<rrN) rxr0maxr typeToMethodr?queryrr@rAr)r2rBrC unknownTyperrrr%test_unknownTypeNXDOMAINsz'AuthorityTests.test_unknownTypeNXDOMAINcCs<ttjd}|ttd|d\}}}||gdS)z Requesting a record of unknown type where other records exist for the name in question results in an empty answer set. r<rJrJN)rKr rLrFrzrMrr)r2rNrr rHrrr%test_unknownTypeMissings  z&AuthorityTests.test_unknownTypeMissingc Csdtjj}td}ttjjtf||gid}t|||}||\}}}||g||tj |tj tj |ddg||gdS)z Create an authority and make a request against it. Then verify that the result is a referral, including no records in the answers or additional sections, but with an I{NS} record in the authority section. sexample.rrkFrEN) rr>rr rr,getattrrFrrrrC)r2method subdomain nameserverr rrrHrrr% _referralTests0     zAuthorityTests._referralTestcC|ddS)a  When an I{NS} record is found for a child zone, it is included in the authority section of the response. It is marked as non-authoritative if the authority is not also authoritative for the child zone (RFC 2181, section 6.1). rNrUrrrr% test_referralszAuthorityTests.test_referralcCrV)zT A referral is also generated for a request of type C{ALL_RECORDS}. rNrWrrrr%test_allRecordsReferralsz&AuthorityTests.test_allRecordsReferralN) r3r4r5rrDrIrOrPrUrXrYrrrr%r=es "  ! r=c@seZdZdZedZedZddZ ddZ dd Z d d Z d d Z ddZddZddZddZddZddZddZddZddZd d!Zd"S)#AdditionalProcessingTestszz Tests for L{FileAuthority}'s additional processing for those record types which require it (MX, CNAME, etc). s10.0.0.1sf080::1cCs8t|jj|f|jj||g||id}t||tjjS)a Perform a DNS lookup against a L{FileAuthority} configured with records as defined by C{makeRecord} and C{addresses}. @param method: The name of the lookup method to use; for example, C{"lookupNameservers"}. @type method: L{str} @param soa: A L{Record_SOA} for the zone for which the L{FileAuthority} is authoritative. @param makeRecord: A one-argument callable which accepts a name and returns an L{IRecord} provider. L{FileAuthority} is constructed with this record. The L{FileAuthority} is queried for a record of the resulting type with the given name. @param target: The extra name which the record returned by C{makeRecord} will be pointed at; this is the name which might require extra processing by the server so that all the available, useful information is returned. For example, this is the target of a CNAME record or the mail exchange host pointed to by an MX record. @type target: L{bytes} @param addresses: A L{list} of records giving addresses of C{target}. @return: A L{Deferred} that fires with the result of the resolver method give by C{method}. rk)r,r>rrQr)r2rRr0 makeRecordtarget addressesr rrr%_lookupSomeRecordss z,AdditionalProcessingTests._lookupSomeRecordscCs(dd}|t||dt||ddS)a  Assert that the L{RRHeader} instances given by C{expected} and C{computed} carry all the same information but without requiring the records appear in the same order. @param expected: A L{list} of L{RRHeader} instances giving the expected records. @param computed: A L{list} of L{RRHeader} instances giving the records computed by the scenario under test. @raise self.failureException: If the two collections of records disagree. cSs|jSr-)r)rrrrr%r$sz>AdditionalProcessingTests.assertRecordsMatch..)keyN)rsorted)r2expectedcomputedr`rrr%assertRecordsMatchs z,AdditionalProcessingTests.assertRecordsMatchcsLdtjj||t||}||\}}}|fdd|D|dS)aT Verify that certain address records are included in the I{additional} section of a response generated by L{FileAuthority}. @param method: See L{_lookupSomeRecords} @param makeRecord: See L{_lookupSomeRecords} @param addresses: A L{list} of L{IRecord} providers which the I{additional} section of the response is required to match (ignoring order). @raise self.failureException: If the I{additional} section of the response consists of different records than those given by C{addresses}. smail.c $g|]}tj|jtj|ddqSTrEr rrGrrCr#rr\rr%r&=z=AdditionalProcessingTests._additionalTest..N)rr>rr^rFrd)r2rRr[r]rrr rHrrir%_additionalTest's   z)AdditionalProcessingTests._additionalTestcCs|dttjd|dS)z Verify that a response to an MX query has certain records in the I{additional} section. @param addresses: See C{_additionalTest} rrON)rkrr rr2r]rrr%_additionalMXTestJsz+AdditionalProcessingTests._additionalMXTestcC||jgdS)z If the name of the MX response has A records, they are included in the additional section of the response. N)rm_Arrrr%test_mailExchangeAdditionalAUz6AdditionalProcessingTests.test_mailExchangeAdditionalAcCrn)z If the name of the MX response has AAAA records, they are included in the additional section of the response. N)rm_AAAArrrr%test_mailExchangeAdditionalAAAA\rqz9AdditionalProcessingTests.test_mailExchangeAdditionalAAAAcC||j|jgdS)z If the name of the MX response has both A and AAAA records, they are all included in the additional section of the response. N)rmrorrrrrr%test_mailExchangeAdditionalBothcz9AdditionalProcessingTests.test_mailExchangeAdditionalBothcCs|dtj|dS)z Verify that a response to an NS query has certain records in the I{additional} section. @param addresses: See C{_additionalTest} rN)rkr rrlrrr%_additionalNSTestjsz+AdditionalProcessingTests._additionalNSTestcCrn)z If the name of the NS response has A records, they are included in the additional section of the response. N)rwrorrrr%test_nameserverAdditionalAsrqz4AdditionalProcessingTests.test_nameserverAdditionalAcCrn)z If the name of the NS response has AAAA records, they are included in the additional section of the response. N)rwrrrrrr%test_nameserverAdditionalAAAAzrqz7AdditionalProcessingTests.test_nameserverAdditionalAAAAcCrt)z If the name of the NS response has both A and AAAA records, they are all included in the additional section of the response. N)rwrorrrrrr%test_nameserverAdditionalBothrvz7AdditionalProcessingTests.test_nameserverAdditionalBothcsvdtjj|dttj|}||\}}}tjtjjtjtj tdd}| fdd|D|g|dS)z Verify that a response to a CNAME query has certain records in the I{answer} section. @param addresses: See C{_additionalTest} swww.rTrEc rerfrgrhrirr%r&rjz>AdditionalProcessingTests._answerCNAMETest..N) rr>rr^r rrFrCNAMErCrd)r2r]rrr rHaliasrrir%_answerCNAMETests(     z*AdditionalProcessingTests._answerCNAMETestcCrn)z If the name of the CNAME response has A records, they are included in the answer section of the response. N)r}rorrrr%test_canonicalNameAnswerArqz3AdditionalProcessingTests.test_canonicalNameAnswerAcCrn)z If the name of the CNAME response has AAAA records, they are included in the answer section of the response. N)r}rrrrrr%test_canonicalNameAnswerAAAArqz6AdditionalProcessingTests.test_canonicalNameAnswerAAAAcCrt)z If the name of the CNAME response has both A and AAAA records, they are all included in the answer section of the response. N)r}rorrrrrr%test_canonicalNameAnswerBothrvz6AdditionalProcessingTests.test_canonicalNameAnswerBothN)r3r4r5rr rrorrrr^rdrkrmrprsrurwrxryrzr}r~rrrrrr%rZs&  &#  # rZc@r+)NoInitialResponseTestsc sfdd}dd}tdgd}||_g|tjd|tjdtjtjd d tjd d d d g}tj dddddddddd }||_ || | dtjS)z If a request returns a L{dns.NS} response, but we can't connect to the given server, the request fails with the error returned at connection. cstdSr,)r r-)r2argsmessagesrr%rMsz3NoInitialResponseTests.test_noAnswer..queryc_sttdS)NzCouldn't connect)rfailrgaierror)r2rkwargsrrr% queryProtocolz;NoInitialResponseTests.test_noAnswer..queryProtocol)rrrurMz fooba.comr Fr r rrr<rrr)r_querypatchr r|rrrrrr'appendr getHostByNamerr)r2rMrrr1r)rrr% test_noAnswers:     z$NoInitialResponseTests.test_noAnswerN)r3r4r5rrrrr%rr6rc@s(eZdZdZddZddZddZdS) SecondaryAuthorityServiceTestsz Tests for L{SecondaryAuthorityService}, a service which keeps one or more authorities up to date by doing zone transfers from a master. cCsd}t|ddg}||j|||jd||jdj|||jdjd||jdjd||jdj|||jdjd||jdjddS) a L{SecondaryAuthorityService} can be constructed with a C{str} giving a master server address and several domains, causing the creation of a secondary authority for each domain and that master server address and the default DNS port. 192.168.1.2 example.comz example.orgrrr<s example.orgN)rrprimary_portdomainsdomainr2rservicerrr%test_constructAuthorityFromHostsz>SecondaryAuthorityServiceTests.test_constructAuthorityFromHostcCsd}d}t||fddg}||j|||jd||jdj|||jdj|||jdjd||jdj|||jdj|||jdjddS) a\ L{SecondaryAuthorityService.fromServerAddressAndDomains} constructs a new L{SecondaryAuthorityService} from a C{str} giving a master server address and DNS port and several domains, causing the creation of a secondary authority for each domain and that master server address and the given DNS port. 192.168.1.3iz example.net example.edur example.netr<N)rfromServerAddressAndDomainsrrrrr)r2rrrrrr%&test_constructAuthorityFromHostAndPort s zESecondaryAuthorityServiceTests.test_constructAuthorityFromHostAndPortcCsxd}t|ddg}||j|||jdj|||jdjd||jdj|||jdjddS)aQ L{SecondaryAuthorityService.fromServerAddressAndDomains} constructs a new L{SecondaryAuthorityService} from a C{bytes} giving a master server address and several domains, causing the creation of a secondary authority for each domain and that master server address and the given DNS port. rrz example.edurr<rN)rencoderrrrrrrr% test_constructAuthorityFromBytes#sz?SecondaryAuthorityServiceTests.test_constructAuthorityFromBytesN)r3r4r5rrrrrrrr%rs  rc@s0eZdZdZddZddZddZdd Zd S) SecondaryAuthorityTestsz L{twisted.names.secondary.SecondaryAuthority} correctly constructs objects with a specified IP address and optionally specified DNS port. cCs8tdd}||jd||jd||jddS)zp When constructed using L{SecondaryAuthority.__init__}, the default port of 53 is used. 192.168.1.1 inside.comr inside.comN)rrrrrr2 secondaryrrr%test_defaultPort?s z(SecondaryAuthorityTests.test_defaultPortcCs:tdd}||jd||jd||jddS)z~ When constructed using L{SecondaryAuthority.fromServerAddressAndDomain}, the specified port is used. )rrrrrN)rfromServerAddressAndDomainrrrrrrrr%test_explicitPortIs z)SecondaryAuthorityTests.test_explicitPortc Cstdd}t|_}||jd\}}}}}||d||d|||f}t } | | t } | t | dd|tdtjtjg| jdS)z An attempt is made to transfer the zone for the domain the L{SecondaryAuthority} was constructed with from the server address it was constructed with when L{SecondaryAuthority.transfer} is called. rr9 example.comrrr9rqN)rrr_reactortransfer tcpClientsr-r buildProtocolrmakeConnectionrdecoderrAr rr%rr&) r2rrhostrr{timeout bindAddressprotormsgrrr% test_transferUs    "z%SecondaryAuthorityTests.test_transferc Cs.tdd}t|_}||jd\}}}}}|||f}t} | | t ddd} | t | ddtdd d d d d ddd} tddd} t | jddd} | jtdt| dtd| dtdt| dg| }|tdt||||d}|td| ddgggf|dS)z L{SecondaryAuthority.lookupAddress} returns a L{Deferred} that fires with the I{A} records the authority has cached from the primary. rrrr<)rr rqNsns1.example.comzadmin.example.comi@iii i%i.r=s 192.168.1.2rm)rrr )rr!r z!HrT)r!r )rrrrrrr-rrrrrrrArrrr'extendrrtoStr dataReceivedrlenrFrr)r2rrrrr{rrrrrMr0rrdatar+rrr%test_lookupAddressssF       z*SecondaryAuthorityTests.test_lookupAddressN)r3r4r5rrrrrrrrr%r9s   rsY$ORIGIN example.com. $TTL 1w example.com. IN SOA dns.example.com ( 2013120201 ; serial number of this zone file 1d ; slave refresh 2h ; slave retry time in case of a problem 4w ; slave expiration time 1h ; maximum caching time in case of failed lookups ) ; A comment. @ IN AAAA 2001:db8:10::1 example.com. IN A 10.0.0.1 no-in.example.com. A 10.0.0.2 ; technically wrong but used to work not-fqdn IN MX 10 mx.example.com www IN CNAME example.comc@s`eZdZdZddZddZddZdd Zd d Zd d Z ddZ ddZ ddZ ddZ dS)BindAuthorityTestsz= Tests for L{twisted.names.authority.BindAuthority}. cCs(t|d}||t|jS)z Create a new L{twisted.names.authority.BindAuthority} from C{s}. @param s: A string with BIND zone data. @type s: bytes @return: a new bind authority @rtype: L{twisted.names.authority.BindAuthority} ascii)rrr setContentr BindAuthoritypath)r2sfprrr%loadBindStrings  z!BindAuthorityTests.loadBindStringcC|t|_dSr-)rsampleBindZoner rrrr%rrzBindAuthorityTests.setUpcCs8|jjD]}|jj|D] }|d|jkqqdS)zG Loads the default $TTL and applies it to all records. : N)r r1keysrrE)r2domrecrrr%test_ttls zBindAuthorityTests.test_ttlcCs$|d|jj|d|jjdS)z, Loads the default $ORIGIN. s example.com.not-fqdn.example.comN)rr originassertInr1rrrr%test_originFromFilesz&BindAuthorityTests.test_originFromFilecCsBdD]\}}||j|\\}\\|t|d|jqdS)' A records are loaded. )rz10.0.0.1)no-in.example.comz10.0.0.2rNrFr rrr rr!r2ripr_rrr% test_aRecordss z BindAuthorityTests.test_aRecordscCs4||jd\\}\\|tdd|jdS)* AAAA records are loaded. r2001:db8:10::1rNrFr rrr rr!r2r_rrr%test_aaaaRecordss  z#BindAuthorityTests.test_aaaaRecordscCs8||jd\\}\\|tjdddd|jdS)( MX records are loaded. rrOzmx.example.comr)rrrENrFr rrr rr!rrrr%test_mxRecordss  z!BindAuthorityTests.test_mxRecordscCs<||jd\}\\|d}|tjddd|jdS)+ CNAME records are loaded. www.example.comrrrr NrFr rrr rr!r2r'r_rrr%test_cnameRecordss  z$BindAuthorityTests.test_cnameRecordscCsL|t }|dWdn1swY|d|jjddS)zO loadBindString raises NotImplementedError on invalid records. sexample.com. IN LOL 192.168.0.1NzRecord type 'LOL' not supportedr) assertRaisesNotImplementedErrorrr exceptionr)r2errr%test_invalidRecordClass%s  z*BindAuthorityTests.test_invalidRecordClassc CsbdD],}|t}||dWdn1swY|t|d|jjdqdS)zC $INCLUDE and $GENERATE raise NotImplementedError. )s$INCLUDEs $GENERATEs doesNotMatterNs directive not implementedr)rrrrrrr)r2 directiverrrr%test_invalidDirectives-s   z)BindAuthorityTests.test_invalidDirectivesN)r3r4r5rrrrrrrrrrrrrrr%rs  razone = [ SOA( # For whom we are the authority 'example.com', # This nameserver's name mname = "dns.example.com", # Mailbox of individual who handles this rname = "root.example.com", # Unique serial identifying this SOA data serial = 86400, # Time interval before zone should be refreshed refresh = "2H", # Interval before failed refresh should be retried retry = "1H", # Upper limit on time interval before expiry expire = "1H", # Minimum TTL minimum = "3H" ), AAAA('example.com', '2001:db8:10::1'), A('example.com', '10.0.0.1'), NS('example.com', 'dns.example.com'), A('no-in.example.com', '10.0.0.2'), PTR('2.0.0.10.in-addr.arpa', 'no-in.example.com'), CNAME('www.example.com', 'example.com'), CNAME('ftp.example.com', 'example.com'), MX('not-fqdn.example.com', 10, 'mail.example.com'), ] c@sPeZdZdZddZddZddZdd Zd d Zd d Z ddZ ddZ dS)PySourceAuthorityTestszA Tests for L{twisted.names.authority.PySourceAuthority}. cCsNt|}t|jd }||Wdn1swYt|jS)a Create a new L{twisted.names.authority.PySourceAuthority} from C{s}. @param s: A string with BIND zone data in a Python source file. @type s: L{str} @return: a new bind authority @rtype: L{twisted.names.authority.PySourceAuthority} r9N)rrr:rwriter PySourceAuthority)r2rrrrrr%loadPySourceStringjs   z)PySourceAuthorityTests.loadPySourceStringcCrr-)rsamplePySourcer rrrr%rzrzPySourceAuthorityTests.setUpcCs@dD]\}}||j|\\}\\|t||jqdS)rrNrrrrr%r}s z$PySourceAuthorityTests.test_aRecordscCs2||jd\\}\\|td|jdS)rrrNrrrrr%rs  z'PySourceAuthorityTests.test_aaaaRecordscCs6||jd\\}\\|tjddd|jdS)rrrOzmail.example.com)rrNrrrrr%rs  z%PySourceAuthorityTests.test_mxRecordscC:||jd\}\\|d}|tjdd|jdS)rrrrrNrrrrr%r  z(PySourceAuthorityTests.test_cnameRecordscCr)z) PTR records are loaded. s2.0.0.10.in-addr.arparrrN)rFr rrr rr!rrrr%rrzPySourceAuthorityTests.test_PTRcCsd}|t|j|dS)z1 Input file has no zone variable z nothing = []N)r ValueErrorr)r2 badPySourcerrr%test_badInputNoZonesz*PySourceAuthorityTests.test_badInputNoZoneN) r3r4r5rrrrrrrrrrrrr%res  r)Yrrrr functoolsrriorstructrtwisted.internetrrrtwisted.internet.deferr twisted.namesr r r r rtwisted.names.clientrtwisted.names.dnsrrrrrrtwisted.names.errorrtwisted.names.secondaryrrtwisted.python.compatrtwisted.python.filepathrtwisted.test.proto_helpersrrr twisted.trialrr* FileAuthorityr,r reverse_soarrrrrrrrrrrrrrrrrrrrxrryrzTestCaserprrr/r=rZrrrrrrrrrrr%s                          3    \ Z~_0Hm+