a a2@sdddlZddlZddlmZddlmZddlmZmZddl m Z ddl m Z Gddde Z dS) N)system_session)SamDB) ndr_unpackndr_pack)dnsp)SambaToolCmdTestcseZdZfddZfddZddZddZd d Zd d Zd dZ ddZ ddZ ddZ ddZ ddZddZddZddZdd Zd!d"Zd#d$Zd%d&Zd'd(Zd)d*Zd+d,ZZS)-DnsCmdTestCasec sBtt|dtjd|_dtjdtjdf|_|d|j|j|_t |j |_ d|_ d|_ |d d d d d ddddg}gd}ddg}gd}ddg}gd}|D] }|d||d|q|D] }|d||d|qddggd|||||gdd|_d d!ggd"|||||d#|_dS)$Nz ldap://%sSERVERz-U%s%%%sZ DC_USERNAMEZ DC_PASSWORDz-Hz 192.168.0.193z 192.168.0.194SAMDOM.EXAMPLE.COMz 1.EXAMPLE.COMz %sEXAMPLE.COMz1.1.1.1.1.1.1.1.1.1.1.1.1.1.1.1.1.1.1.1.1.1.1.1.1.1.1.1.1.1.1.1.1.1.1.1.1.1.1.1.1.1.1.1.1.1.1.1.1.1.1.1.1.1.1.1.1.1.1.1.1.1.1.1.1.1.1.1.1.1.1.1.1.1.1.1.1.1.1.1.1.1.1.1.1.1.1.1.1.1.1.1.1.1.1.1.1.1.1.1.ZEXAMPLEz !@#$%^&*()_u HIGHÿBYTEz @.EXAMPLE.COM.)z...z .EXAMPLE.COMz .EXAMPLE.zSAMDOM..EXAMPLE.COMzSAMDOM.EXAMPLE.COM 65530zSAMDOM.EXAMPLE.COM 0)zSAMDOM.EXAMPLE.COM -1r  zSAMDOM.EXAMPLE.COM 1 1z%SAMDOM.EXAMPLE.COM SAMDOM.EXAMPLE.COMz$SAMDOM.EXAMPLE.COM 65530 65530 65530z$SAMDOM.EXAMPLE.COM 1 1 1)zSAMDOM.EXAMPLE.COM 0 65536 0zSAMDOM.EXAMPLE.COM 0 0 65536zSAMDOM.EXAMPLE.COM 65536 0 0z%s 1z%s 0 0 0z 192.168.0.1z255.255.255.255) z'1234:5678:9ABC:DEF0:0000:0000:0000:0000z'0000:0000:0000:0000:0000:0000:0000:0000z'1234:5678:9ABC:DEF0:1234:5678:9ABC:DEF0z1234:1234:1234::z1234:5678:9ABC:DEF0::z0000:0000::0000z#1234::5678:9ABC:0000:0000:0000:0000z::1z::z1:1:1:1:1:1:1:1)textr z@#! )AAAAAPTRCNAMENSMXSRVZTXTz 192.168.0.500z255.255.255.255/32)z'GGGG:1234:5678:9ABC:0000:0000:0000:0000z)0000:0000:0000:0000:0000:0000:0000:0000/1z'AAAA:AAAA:AAAA:AAAA:G000:0000:0000:1234z,1234:5678:9ABC:DEF0:1234:5678:9ABC:DEF0:1234z"1234:5678:9ABC:DEF0:1234:5678:9ABCz1111::1111::1111)rrrrrrr)superrsetUposenvironZdburl creds_stringZgetSamDBsamdbstrZget_config_basednZ config_dntestipZtestip2addZoneappend good_records bad_records) selfZgood_dnsZbad_dnsZgood_mxZbad_mxZgood_srvZbad_srvZbad_dnZgood_dn __class__?/usr/lib/python3/dist-packages/samba/tests/samba_tool/dnscmd.pyrsj  zDnsCmdTestCase.setUpcs|tt|dSN) deleteZonerrtearDownr#r$r&r'r*uszDnsCmdTestCase.tearDowncCs||dSr()r)rr+r&r&r' resetZoneyszDnsCmdTestCase.resetZonecCs:d|_|ddtjd|j|j\}}}||||dS)NzonednsZ zonecreater )r- runsubcmdrrrassertCmdSuccessr#resultouterrr&r&r'r}s zDnsCmdTestCase.addZonecCs4|ddtjd|j|j\}}}||||dS)Nr.Z zonedeleter )r/rrr-rr0r1r&r&r'r)s zDnsCmdTestCase.deleteZonecCs|jjd|jtjddgd}|D]}|t|jvr&|j}qDq&|jj|tjddgd}|D]0}|t|jvr`|jttj |ddfSq`dS)NzDC=DomainDnsZones,%sz(objectClass=dnsZone)ZcnbaseZscopeZ expressionattrsz(objectClass=dnsNode) dnsRecordr) rsearchget_default_basednldb SCOPE_SUBTREErdnrrDnssrvRpcRecord)r#Z zone_name record_nameZzonesr-Zzone_dnrecordsZold_packed_recordr&r&r'get_record_from_dbs,  z!DnsCmdTestCase.get_record_from_dbc Csxd}d}|ddtjd|jd|||j\}}}||||d||f||jd\}}d|_|j ||g}|dur| d g} |dd tjd|jd||j\}}}z,||||d | d|vo||vd Wn$t y} z WYd} ~ n d} ~ 00|ddtjd|jd|||j\}}}z| |d Wn0t yf} z| | WYd} ~ n d} ~ 00|ddtjd|jd|||j\}}}z||||dWn0t y} z| | WYd} ~ n d} ~ 00|dd tjd|jd||j\}}}z| |dWn0t y<} z| | WYd} ~ n d} ~ 00t| dkrtd} | D]} | dt| } qTt | dS)Nz 192.168.50.50rr.addr testrecord'Failed to add record '%s' with type %s.rz2Unable to update dns record to have DNS_RANK_NONE.queryz4Failed to query for a recordwhich had DNS_RANK_NONE.zVQuery for a record which had DNS_RANK_NONEsucceeded but produced no resulting records.zBSuccessfully added duplicate recordof one which had DNS_RANK_NONE.deletez/Failed to delete recordwhich had DNS_RANK_NONE.z?Successfully queried for deleted recordwhich had DNS_RANK_NONE.z0Failed appropriate behaviour with DNS_RANK_NONE:r)r/rrr-rr0rAZrankrZdns_replace_by_dnfail assertTrueAssertionError assertCmdFailr lenr) r#Z record_strZrecord_type_strr2r3r4r=recordreserrorseZerr_strerrorr&r&r'test_rank_nonesv           zDnsCmdTestCase.test_rank_nonec CsTd}g}|jD]}|j|D]}z|ddtjd|jd|||j\}}}||||d||f|ddtjd|jd||j\}}}||||d||f|dd tjd|jd|||j\}}}||||d ||fWqty}z|d }||WYd }~qd }~00qq|dkrP|D]} t | q2| d |d S)zN For all good records, attempt to add, query and delete them. rr.rBr rCz$Failed to addrecord %s with type %s.rEz+Failed to queryrecord %s with qualifier %s.rFz'Failed to removerecord %s with type %s.N@Failed to accept valid commands. %d total failures.Errors above.) r!r/rrr-rr0rIr printrG) r# num_failures failure_msgsdnstyperLr2r3r4rOmsgr&r&r'test_accept_valid_commandssR        $  z)DnsCmdTestCase.test_accept_valid_commandsc Csd}g}|jD]}|j|D]}z@|ddtjd|jd|||j\}}}||d||fWn>ty}z&|d}||| WYd}~n d}~00z@|dd tjd|jd|||j\}}}||d ||fWqty$}z&|d}||| WYd}~qd}~00qq|jD].}|j|D]} |j |d} z|ddtjd|jd|| |j\}}}| |||d ||f|dd tjd|jd|| | |j \}}}||d | || f|dd tjd|jd|| |j\}}}| |||d| |fWn@tyX}z&|d}||| WYd}~n d}~00q@q0|dkr|D]} t | qp| d|dS)z} For all bad records, attempt to add them and update to them, making sure that both operations fail. rr.rBr rCz3Successfully added invalidrecord '%s' of type '%s'.rRNrFzJSuccessfully deleted invalidrecord '%s' of type '%s' whichshouldn't exist.rDupdatez\Successfully updated valid record '%s' of type '%s' to invalid record '%s' of the same type.0Could not delete valid record '%s' of type '%s'.zCFailed to reject invalid commands. %d total failures. Errors above.)r"r/rrr-rrJrIr r,r!r0rTrG) r#rUrVrWrLr2r3r4rOZ bad_recordZ good_recordrXr&r&r'test_reject_invalid_commandss       "        &  z+DnsCmdTestCase.test_reject_invalid_commandsc CsL|jD]>}|j|d}|ddtjd|jd|||j\}}}||||d||f|jD]}|j|d}|ddtjd|jd|||j\}}}z||WntyYq`Yn0|ddtjd|jd||||j \}}}||d||||f|ddtjd|jd||||j \}}}||d ||||fq`qd S) zV Make sure that a record can't be updated to one of a different type. rr.rBr rC%Failed to add record %s with type %s.rZziSuccessfully updated record '%s' to '%s', even though the latter is of type '%s' where '%s' was expected.ziSuccessfully updated record '%s' to '%s', even though the former is of type '%s' where '%s' was expected.N) r!r/rrr-rr0rJrI)r#dnstype1record1r2r3r4dnstype2record2r&r&r'test_update_invalid_typeksR           z'DnsCmdTestCase.test_update_invalid_typec Cs|jD]}|j|D]}|ddtjd|jd|||j\}}}||||d||f|ddtjd|jd||||j \}}}||d||ddtjd|jd|||j\}}}||||d ||fqq|jd D]:}|ddtjd|jdd ||j\}}}||||d ||}t t |d d }d|d|d |d|f}|ddtjd|jdd |||j \}}}||||d|d |f|ddtjd|jdd |j\}}}||||d|d f|ddtjd|jdd ||j\}}}||||d |d fq|jD]}t |j|d kr@q$|j|d} |j|d } |j|d}|ddtjd|jd|| |j\}}}||||d| |f|ddtjd|jd|| ||j \}}}||d| | |fq$dS)Nr.rBr rCr]rZz8Successfully updated record '%s' to be exactly the same.rFr[rz(Failed to add record %s with type 'SRV'.rRz %s %s %s %srz2Failed to update record '%s' of type '%s' to '%s'.rEz-Failed to query for record '%s' of type '%s'.zeSuccessfully updated record '%s' from '%s' to '%s', even though the given source record is incorrect.) r!r/rrr-rr0rJsplitrintrK) r#rWrLr2r3r4reZnew_bitZ new_recordZ used_recordZ unused_recordr&r&r'test_update_valid_types                   z%DnsCmdTestCase.test_update_valid_typec CsL|ddtjd|jddd|j\}}}||d|d|vd |dS) Nr.rBr rCZSOAtestzNSuccessfully added record of type SOA, when this type should not be available.ztype SOA is not supportedzEInvalid error message '%s' when attempting to add record of type SOA.)r/rrr-rrJrHr1r&r&r'test_invalid_typess   z!DnsCmdTestCase.test_invalid_typesc Csd}|jD]}|j|d}|jD]}|d7}|ddkr@q$||krJq$|j|d}|ddtjd|jd|||j\}}}||||d||f|ddtjd|jd|||j\}}}||||d ||||f|dd tjd|jd||j\}}}||||d ||||f|dd tjd|jd||j\}}}||||d ||||f|dd tjd|jd|||j\}}}||||d||f|dd tjd|jd|||j\}}}||||d||fq$q dS)zp Make sure that we can add an entry with the same name as an existing one but a different type. rrRr.rBr rCz'Failed to add record '%s' of type '%s'.zaFailed to add record '%s' of type '%s' when a record '%s' of type '%s' with the same name exists.rEznFailed to query for record '%s' of type '%s' when a new record '%s' of type '%s' with the same name was added.zuFailed to query record '%s' of type '%s' which should have been added with the same name as record '%s' of type '%s'.rFz*Failed to delete record '%s' of type '%s'.N)r!r/rrr-rr0) r#ir^r_r`rar2r3r4r&r&r'#test_add_overlapping_different_type s|                  z2DnsCmdTestCase.test_add_overlapping_different_typec Cs||ddtjd|jdd|j|j|ddtjd|jdd|j|j|ddtjd|jdd|j\}}}||dS)Nr.rBr rCrrFrEr/rrr-rrrJr1r&r&r'test_query_deleted_recordXs   z(DnsCmdTestCase.test_query_deleted_recordc Cs|jD]}|ddtjd|jd||j|d|j\}}}|||||ddtjd|jd||j|d|j\}}}|||ddtjd|jd||j\}}}|||||ddtjd|jd||j|d|j\}}}||||qdS)Nr.rBr rCrrErF)r!r/rrr-rr0rJ)r#Z record_typer2r3r4r&r&r'test_add_duplicate_recordds>         z(DnsCmdTestCase.test_add_duplicate_recordc Cs|ddtjd|jdd|j|j|ddtjd|jdd|j|j|ddtjd|jdd|j|j\}}}|||ddtjd|jdd|j\}}}|||ddtjd|jdd|j|j\}}}||dS) Nr.rBr rCrrFrEZ testrecord2rmr1r&r&r'test_remove_deleted_records0       z)DnsCmdTestCase.test_remove_deleted_recordc CsJ|ddtjd|jdd|j|jdd|j}|ddtjd|jdd||j|ddtjd|jd d ||j|ddtjd|jd d ||jd |}|ddtjd|jdd||j|ddtjd||jdD]^}|jj d|j t j d|dgd}| t|d|D]}| t|ddq(qdS)z; Test dns cleanup command is working fine. r.rBr testar{0}.{1} testcnamertestnsrtestptrr{0} 65530 65530 65530testsrvrcleanup)rqrsrtrurwDC=DomainDnsZones,{0}"(&(objectClass=dnsNode)(name={0})) dNSTombstonedr5rRTRUENr/rrr-rrformatlowerrr9r:r;r< assertEqualrKrr# dnshostname srv_recordr?r@rLr&r&r'test_cleanup_records<      z"DnsCmdTestCase.test_cleanup_recordc Csp|ddtjd|jdd|j|jdd|j}|ddtjd|jdd ||j|ddtjd|jd d ||j|ddtjd|jd d ||jd|}|ddtjd|jdd||j|ddtjd|jdd|j|j|ddtjd||jdD]`}|jj d|j t j d|dgd}| t|d|D]}| t|ddqLq dS)zB Test dns cleanup command works with no A record. r.rBr Znotestarrrrq notestcnamernotestnsr notestptrrrv notestsrvrrFrx)rrrrryrzr{r5rRr|Nr}rr&r&r'test_cleanup_record_no_A_recordsB       z.DnsCmdTestCase.test_cleanup_record_no_A_recordc Csddg}d}|D]f}|ddtjd|j|d|j|jd||j}d |}|ddtjd|j|d ||jq|jj d |j t j d |d gd}| t|dd dd|j}|ddtjd||j|jj d |j t j d |d dgd}| t|dd d|dd d}ttj|} d|j} | | jj| |d|dvpt|dddkdS)a? Test dns cleanup command for multi-valued SRV record. Steps: - Add 2 A records host1 and host2 - Add a SRV record srv1 and points to both host1 and host2 - Run cleanup command for host1 - Check records for srv1, data for host1 should be gone and host2 is kept. Zhost1Zhost2Zsrv1r.rBr rrrrvrryrzr8r5rrdz host1.{0}rxr{rRz host2.{0}ZFALSEN)r/rrr-rrr~rrr9r:r;r<rrKrrr>dataZ nameTargetrHr) r#ZhostsZsrv_nameZhostrrr@Z dnshostname1Zdns_record_binZdns_record_objZ dnshostname2r&r&r'test_cleanup_multi_srv_recordsH     z,DnsCmdTestCase.test_cleanup_multi_srv_recordc Csd}g}ddg}|D]D\}}}}z|ddtjd|j||||j\}} } ||| | d|||f|ddtjd|j|||j\}} } ||| | d ||f|ddtjd|j|||j\}} } ||d ||f|dd tjd|j||||j\}} } ||| | d ||fWqtyX} z|d }|| WYd } ~ qd } ~ 00q|dkr|D]} t | qj| d|d S)zT Ensure that DNS wild card entries can be added deleted and queried r)z*.ZMISSrz1.1.1.1)z*.SAMDOMz MISS.SAMDOMrz1.1.1.2r.rBr z*Failed to add record %s (%s) with type %s.rEz,Failed to query record %s with qualifier %s.rFz(Failed to remove record %s with type %s.rRNrS) r/rrr-rr0rJrIr rTrG) r#rUrVr@nameZmissrWrLr2r3r4rOrXr&r&r'test_dns_wildcards5s     "  z!DnsCmdTestCase.test_dns_wildcardsc CsRdD]H}|ddd|tjd|j\}}}||||d|||dkqdS)N)Zw2kZdotnetZlonghornr.Z serverinfoz--client-versionr z1Failed to print serverinfo with client version %sr )r/rrrr0rH)r#vr2r3r4r&r&r'test_serverinfoxs zDnsCmdTestCase.test_serverinfocCsD|ddtjd|j|j\}}}||||d||dkdS)Nr.Zzoneinfor zFailed to print zoneinfor )r/rrr-rr0rHr1r&r&r' test_zoneinfos zDnsCmdTestCase.test_zoneinfo)__name__ __module__ __qualname__rr*r,rr)rArQrYr\rbrgrirlrnrorprrrrrr __classcell__r&r&r$r'rs, X  F-T4b K .2=Cr)rr;Z samba.authrZ samba.samdbrZ samba.ndrrrZ samba.dcerpcrZsamba.tests.samba_tool.baserrr&r&r&r's