|; i^HdZddlmZddlZddlZddlmZddlmZddl m Z ddl m Z m Z mZdd lmZdd lmZdd lmZmZmZmZmZmZmZmZmZmZd d lmZe Z!e"Z#e$Z%e&Z'e Z(eZ)edjTZ+edjTZ,edjTZ-dZ.e/de0dDZ1e/de0e.DZ2eedfDcic] }|e-| c}Z3GddejhZ5e5jljnZ8e5jrjnZ:ejvZGddZ?ycc}w)aMulticast DNS Service Discovery for Python, v0.14-wmcbrine Copyright 2003 Paul Scott-Murphy, 2014 William McBrine This module provides a framework for the use of DNS Service Discovery using IP multicast. This library is free software; you can redistribute it and/or modify it under the terms of the GNU Lesser General Public License as published by the Free Software Foundation; either version 2.1 of the License, or (at your option) any later version. This library is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU Lesser General Public License for more details. You should have received a copy of the GNU Lesser General Public License along with this library; if not, write to the Free Software Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA ) annotationsN)Sequence)Struct) TYPE_CHECKING) DNSPointer DNSQuestion DNSRecord)NamePartTooLongException)log) _CLASS_UNIQUE _DNS_HOST_TTL_DNS_OTHER_TTL_DNS_PACKET_HEADER_LEN_FLAGS_QR_MASK_FLAGS_QR_QUERY_FLAGS_QR_RESPONSE _FLAGS_TC_MAX_MSG_ABSOLUTE_MAX_MSG_TYPICAL) DNSIncomingz>Bz>Hz>Lc#2K|]}t|ywN) PACK_BYTE.0is N/home/ubuntu/myenv/lib/python3.12/site-packages/zeroconf/_protocol/outgoing.py r!>s4A9Q<4c#2K|]}t|ywr) PACK_SHORTrs r r!r!?sCqZ]Cr"ceZdZdZdZy)StaterrN)__name__ __module__ __qualname__initfinishedr r'r'Cs DHr.r'c"eZdZdZdZd#d$dZd%dZd%dZd&dZd'dZ d(dZ d)d Z d*d Z d+d Z d,d Zd-d Zd.dZd-dZd/dZd-dZd0dZd1dZd2dZd3dZd4dZd5dZd6dZd7dZd8dZd9dZd:dZd;dZdd Z"d?d!Z#y")@ DNSOutgoingz+Object representation of an outgoing packet) additionals allow_longanswers authoritiesdatar,flagsid multicastnames packets_data questionssizestatec||_d|_||_||_g|_i|_g|_t|_d|_ t|_ g|_ g|_ g|_g|_y)NFT)r6r,r7r8r:r9r5rr<r2 STATE_INITr=r;r3r4r1)selfr6r8id_s r __init__zDNSOutgoing.__init__csi  ")+&( !# / $ ,.68 -/,.r.c6|jtztk(S)z Returns true if this is a query.)r6rrr@s r is_queryzDNSOutgoing.is_queryws ^+??r.c6|jtztk(S)z#Returns true if this is a response.)r6rrrDs r is_responsezDNSOutgoing.is_response{s ^+0BBBr.cDi|_g|_t|_d|_y)NT)r9r5rr<r2rDs r _reset_for_next_packetz"DNSOutgoing._reset_for_next_packets  * r.c djdjd|jd|jd|jd|j d|j d|jgS) Nzz, z multicast=zflags=z questions=zanswers=z authorities=z additionals=)formatjoinr8r6r;r3r4r1rDs r __repr__zDNSOutgoing.__repr__s!(( II  01TZZL)  01t||n-"4#3#3"45"4#3#3"45    r.c:|jj|y)zAdds a questionN)r;appendr@records r add_questionzDNSOutgoing.add_questions f%r.cL|j|s|j|dyy)zAdds an answergN) suppressed_byadd_answer_at_time)r@inprQs r add_answerzDNSOutgoing.add_answers%##C(  # #FC 0)r.cv|}|5|dk(s|j|s|jj||fyyy)z6Adds an answer if it does not expire by a certain timeNr) is_expiredr3rO)r@rQnow now_doubles r rUzDNSOutgoing.add_answer_at_times?  :?&:K:KJ:W LL   .;X r.c:|jj|y)zAdds an authoritative answerN)r4rOrPs r add_authorative_answerz"DNSOutgoing.add_authorative_answers 'r.c:|jj|y)aAdds an additional answer From: RFC 6763, DNS-Based Service Discovery, February 2013 12. DNS Additional Record Generation DNS has an efficiency feature whereby a DNS server may place additional records in the additional section of the DNS message. These additional records are records that the client did not explicitly request, but the server has reasonable grounds to expect that the client might request them shortly, so including them can save the client from having to issue additional queries. This section recommends which additional records SHOULD be generated to improve network efficiency, for both Unicast and Multicast DNS-SD responses. 12.1. PTR Records When including a DNS-SD Service Instance Enumeration or Selective Instance Enumeration (subtype) PTR record in a response packet, the server/responder SHOULD include the following additional records: o The SRV record(s) named in the PTR rdata. o The TXT record(s) named in the PTR rdata. o All address records (type "A" and "AAAA") named in the SRV rdata. 12.2. SRV Records When including an SRV record in a response packet, the server/responder SHOULD include the following additional records: o All address records (type "A" and "AAAA") named in the SRV rdata. N)r1rOrPs r add_additional_answerz!DNSOutgoing.add_additional_answersH 'r.cr|jjt||xjdz c_y)z"Writes a single byte to the packetrN)r5rO BYTE_TABLEr<r@values r _write_bytezDNSOutgoing._write_bytes& E*+ Q r.c<|tkr t|St|S)z%Convert an unsigned short to 2 bytes.)SHORT_CACHE_MAX SHORT_LOOKUPr%rbs r _get_shortzDNSOutgoing._get_shorts&+o&=|E"T:eCTTr.cZ|jjd|j|y)z4Inserts an unsigned short at the start of the packetrN)r5insertrhrbs r _insert_short_at_startz"DNSOutgoing._insert_short_at_starts DOOE23r.c@|j||j|<y)z>Replaces an unsigned short in a certain position in the packetN)rhr5)r@indexrcs r _replace_shortzDNSOutgoing._replace_shorts??51 %r.c|jj|j||xjdz c_y)z&Writes an unsigned short to the packetrN)r5rOrhr<rbs r write_shortzDNSOutgoing.write_shorts* /0 Q r.ct|}tj|}||jj |n$|jj t ||xj dz c_y)z(Writes an unsigned integer to the packetN)int LONG_LOOKUPgetr5rO PACK_LONGr<)r@rc value_as_int long_bytess r _write_intzDNSOutgoing._write_intsU5z  __\2  ! II  Z ( II  Y|4 5 Q r.ctrt|tsJ|jj ||xj t |z c_y)zWrites a string to the packetN)r isinstancebytesr5rOr<lenrbs r write_stringzDNSOutgoing.write_strings9 eU+ ++  SZ r.c|jd}t|}|dkDrt|j||j |y)z5Writes a UTF-8 string of a given length to the packetutf-8@N)encoder}r rdr~)r@sutfstrlengths r _write_utfzDNSOutgoing._write_utfsB'"V B;* *   &!r.ctrt|tsJt|}|dkDrt|j ||j |y)Nr#)rr{r|r}r rdr~)r@rcrs r write_character_stringz"DNSOutgoing.write_character_stringsG eU+ ++U C<* *   % r.c|r |ddk(r|dd}|jj|d}|r|j|y|j}|j d}||j|<|j |dd}t dt|D]}dj||d}|jj|d}|r|j|y|dk(rt|jd}||zt|jdz |j|<|j |||jdy)a Write names to packet 18.14. Name Compression When generating Multicast DNS messages, implementations SHOULD use name compression wherever possible to compress the names of resource records, by replacing some or all of the resource record name with a compact two-byte reference to an appearance of that data somewhere earlier in the message [RFC1035]. .Nrrr) r9ru_write_link_to_namer<splitrranger}rLrrd)r@namerm start_sizelabels name_lengthcount partial_names r write_namezDNSOutgoing.write_names7 DHO9D tQ'   $ $U + YY C% 4 q " 1c&k* +E88F56N3LJJNN<3E((/a!$++g"67 '1K'?#lFYFYZaFbBc'cDJJ| $ OOF5M * + r.cZ|j|dz dz|j|dzy)N)rd)r@rms r rzDNSOutgoing._write_link_to_name0s- %1*,- &r.ct|j}|j}|j|j|j |j |j||j||S)zWrites a question to the packet) r}r5r<rrrptype_write_record_class_check_data_limit_or_rollback)r@questionstart_data_lengthrs r _write_questionzDNSOutgoing._write_question6s] NYY   & '   *112CZPPr.c|j}|jdur%|jr|j|tzy|j|y)zAWrite out the record class including the unique/unicast (QU) bit.TN)class_uniquer8rpr )r@rQrs r rzDNSOutgoing._write_record_class?s> ==D T^^   Vm3 4   V $r.cp|j|dk(r|jy|j|y)zWrite out the record ttl.rN)ryttlget_remaining_ttl)r@rQrZs r _write_ttlzDNSOutgoing._write_ttlGs( cQh RF4L4LS4QRr.ct|j}|j}|j|j|j |j |j||j||t|j}|j d|j|d}|j|dzdD]}|t|z }|j|||j||S)zWrites a record (answer, authoritative answer, additional) to the packet. Returns True on success, or False if we did not because the packet because the record does not fit.rrN) r}r5r<rrrprrrwriternr)r@rQrZrrrmrds r _write_recordzDNSOutgoing._write_recordKs  NYY   $ %   ( $DII  T519;' A c!f F  E6*112CZPPr.c|jrtnt}d|_|j|kryt t r!t jd|j||j|d=||_|}|jjDcgc] \}}||k\s |}}}|D]}|j|=ycc}}w)z@Check data limit, if we go over, then rollback and return False.FTz8Reached data limit (size=%d) > (limit=%d) - rolling backN) r2rrr<LOGGING_IS_ENABLED_FOR LOGGING_DEBUGr debugr5r9items)r@rr len_limitstart_size_intridxrollback_namess r rz)DNSOutgoing._check_data_limit_or_rollbackas)-%>N  99 ! !- 0 IIJ   II'( ) #04 0@0@0B\94c^F[$\\" !D 4  !]s  B?#B?cdd}|j|dD]}|j|s|S|dz }|SNrr)r;r)r@questions_offsetquestions_writtenrs r _write_questions_from_offsetz(DNSOutgoing._write_questions_from_offsetxsP'7'89 #H''1    "  #! r.cld}|j|dD]\}}|j||s|S|dz }!|Sr)r3r)r@ answer_offsetanswers_writtenanswertime_s r _write_answers_from_offsetz&DNSOutgoing._write_answers_from_offsetsR!\\-.9 !MFE%%fe4 q O !r.cRd}||dD]}|j|ds|S|dz }|Sr)r)r@recordsoffsetrecords_writtenrQs r _write_records_from_offsetz&DNSOutgoing._write_records_from_offsetsJfg& !F%%fa0 q O !r.c|t|jkxsL|t|jkxs2|t|jkxs|t|jkS)z\Check if all questions, answers, authority, and additionals have been written to the packet.)r}r;r3r4r1)r@rrauthority_offsetadditional_offsets r _has_more_to_addzDNSOutgoing._has_more_to_addsb s4>>2 2 9s4<<00 9#d&6&6"77 9!3t'7'7#88  r.c v|j}|jtk(r|Sd}d}d}d}ttdu}d}|rn|r~t j d||||t j dt|jt|jt|jt|j|j|}|j|} |j|j|} |j|j|} t|j } |j#| |j#| |j#| |j#|||z }|| z }|| z }|| z }|rt j d|||||j%||||}|rJ|j'r:|rt j d|j#|j(t*zn|j#|j(|j,r|j#dn|j#|j.|j1dj3|j | st j4dn|r|j7|rnt|_|S) aReturns a list of bytestrings containing the packets' bytes No further parts should be added to the packet once this is done. The packets are each restricted to _MAX_MSG_TYPICAL or less in length, except for the case of a single answer which will be written out to a single oversized packet no more than _MAX_MSG_ABSOLUTE in length (and hence will be subject to IP fragmentation potentially).rTzBoffsets = questions=%d, answers=%d, authorities=%d, additionals=%dzBlengths = questions=%d, answers=%d, authorities=%d, additionals=%dzFnow offsets = questions=%d, answers=%d, authorities=%d, additionals=%dzSetting TC flagr.z4packets() made no progress adding records; returning)r:r=STATE_FINISHEDrrr rr}r;r3r4r1rrrboolr5rkrrEr6rr8r7rOrLwarningrI) r@r:rrrr debug_enablehas_more_to_addrrauthorities_writtenadditionals_written made_progresss r packetszDNSOutgoing.packetss}(( :: '  -m<D  X$!$%  X' %(()(() !% A ABR S "==mLO"&"A"A$BRBRTd"e "&"A"A$BRBRTe"f  OM  ' '(; <  ' '(; <  ' ' 8  ' '(9 :  1 1  _ ,M  3 3  !4 4  \$!$% #33 -1ACTO4==?II/0++DJJ,BC++DJJ7~~++A.++DGG4    3 4  RS++-MP$ r.N)Tr)r6rsr8rrArsreturnNone)rr)rr)rstr)rQr rr)rVrrQr rr)rQzDNSRecord | NonerZfloat_rr)rQrrr)rQr rr)rcint_rr)rcrrr|)rmrrcrrr)rcz float | intrr)rcbytes_rr)rstr_rr)rcr|rr)rrrr)rmrrr)r DNSQuestion_rr)rQzDNSQuestion_ | DNSRecord_rr)rQ DNSRecord_rZrrr)rQrrZrrr)rrrrrr)rrrrs)rrrrs)rzSequence[DNSRecord]rrrrs) rrrrrrrrrr)rz list[bytes])$r(r)r*__doc__ __slots__rBrErGrIrMrRrWrUr]r_rdrhrkrnrpryr~rrrrrrrrrrrrrrr-r.r r0r0Os5I"/(@C  &1 / ($(L U42  "!)V' Q%SQ,.!            _r.r0)@r __future__renumloggingcollections.abcrstructrtypingr_dnsrr r _exceptionsr _loggerr constr rrrrrrrrrincomingrrrfloatrrsrr|rrrpackrr%rvrftuplerrargrtEnumr'r+rcr?r,r isEnabledForrDEBUGrr0)rs0r rs/,# $ 552   "       4L   D\   4L    4s4 4 CE/,BCC )7(JK1q)A,K DII ZZ   %%)) ooLs E