|; iI1dZddlmZddlmZddlmZmZmZddl m Z m Z ddl m Z mZmZmZmZmZmZmZddlmZdd lmZmZe eeeefZe e eeeefZeeeeeffZeZeZ e!Z"e#Z$d Z%dd Z&Gd d Z'y)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 ) annotations)Iterable)heapifyheappopheappush)Unioncast) DNSAddressDNSEntryDNSHinfoDNSNsec DNSPointer DNSRecord DNSServiceDNSText)current_time_millis) _ONE_SECOND _TYPE_PTRdc ||}||=|s||=yy)z[Remove a key from a DNSRecord cache This function must be run in from event loop. N)cachekeyrecord record_caches B/home/ubuntu/myenv/lib/python3.12/site-packages/zeroconf/_cache.py _remove_keyr8s$ :LV  #J ceZdZdZddZddZddZddZddZddZ ddZ dd Z dd Z dd Z dd Zd d Zd!dZd"dZddZd#dZd$dZ d%dZd&dZy)'DNSCachezA cache of DNS entries.c<i|_g|_i|_i|_yN)r _expire_heap _expirations service_cacheselfs r__init__zDNSCache.__init__Fs!*, ;=4624rc6|jj|jx}ix}|j|j<||vxrt|t }|||<|j |j dzz}|jj||k7r't|j||f||j|<t|trI|}|jj|jx}ix}|j|j<|||<|S)zAdds an entry. Returns true if the entry was not already in the cache. This function must be run in from event loop. i) rgetr isinstancercreatedttlr%rr$rr& server_key)r(rstorenewwhenservice_record service_stores r _async_addzDNSCache._async_addOsZZ^^FJJ/ /E 8-/ /EDJJvzz*E!E*VW*E&Ef ~~d!23     (D 0 T&&v 7(,D  f % fj )#N!%!3!3!7!78Q8Q!RR [PRR  2 2>3L3L M,:M. ) rc@d}|D]}|j|sd}|S)zAdd multiple records. Returns true if any of the records were not in the cache. This function must be run in from event loop. FT)r5)r(entriesr1entrys rasync_add_recordszDNSCache.async_add_recordsms0 Eu%  rct|tr#|}t|j|j|t|j |j ||jj|dy)zQRemoves an entry. This function must be run in from event loop. N) r,rrr&r/rrr%pop)r(rr3s r _async_removezDNSCache._async_removezsU fj )#N **N,E,E~ VDJJ F3 fd+rc4|D]}|j|y)zXRemove multiple records. This function must be run in from event loop. N)r<)r(r7r8s rasync_remove_recordszDNSCache.async_remove_recordss!  &E   u % &rcZt|jx}sgSg}|jrp|jd}|d}||kDrnVt|j|d}|jj ||k(r|j ||jrp|t kDrt|t|jdzkDrY|jDcgc])}|jj |d|dk(s(|+c}|_t|j|j||Scc}w)zPurge expired entries from the cache. This function must be run in from event loop. :param now: The current time in milliseconds. rr ) lenr$rr%r+append _MIN_SCHEDULED_RECORD_EXPIRATIONrr>)r(nowexpire_heap_lenexpired when_recordr2rr8s r async_expirezDNSCache.async_expires($'t'8'8#999I#%++A.Kq>Dcz D%% & !^F  $$V,4v&( > >#d&7&7"81"<<$(#4#4!8I8I8M8MeTUh8W[`ab[c8c!D  D%% & !!'* !s )D(4D(ct|jj|j}|y|j|S)zGets a unique entry by key. Will return None if there is no matching entry. This function is not threadsafe and must be called from the event loop. N)rr+r)r(r8r0s rasync_get_uniquezDNSCache.async_get_uniques1 uyy) =yyrc|j}|jj|}g}||S|jD]3}||jk(s||j k(s#|j |5|S)zGets all matching entries by details. This function is not thread-safe and must be called from the event loop. )lowerrr+valuestypeclass_rB)r(nametype_rOrrecordsmatchesrs rasync_all_by_detailszDNSCache.async_all_by_detailssp jjl**..%#% ?Nnn& 'F #&--(?v& 'rc$|j|S)zReturns a dict of entries whose key matches the name. This function is not threadsafe and must be called from the event loop. )entries_with_namer(rPs rasync_entries_with_namez DNSCache.async_entries_with_names %%d++rc$|j|S)zReturns a dict of entries whose key matches the server. This function is not threadsafe and must be called from the event loop. )entries_with_serverrWs rasync_entries_with_serverz"DNSCache.async_entries_with_servers ''--rcNt|tr5|jj|jij|St t |jj|jijD]}|j|s|cSy)zNGets an entry by key. Will return None if there is no matching entry.N) r,_UNIQUE_RECORD_TYPESrr+rreversedlistrM__eq__)r(r8 cached_entrys rr+z DNSCache.gets e1 2::>>%))R044U; ;$T$**..B*G*N*N*P%QR $L||L)## $rc|j}|jj|}|ytt |j D]%}||j k(s||jk(s#|cSy)aGets the first matching entry by details. Returns None if no entries match. Calling this function is not recommended as it will only return one record even if there are multiple entries. For example if there are multiple A or AAAA addresses this function will return the last one that was added to the cache which may not be the one you expect. Use get_all_by_details instead. N)rLrr+r^r_rMrNrO)r(rPrQrOrrRras rget_by_detailszDNSCache.get_by_detailssojjl**..% ?$T'..*:%;< $L )))f 8K8K.K## $rc|j}|jj|}|gSt|j Dcgc]$}||j k(s||j k(s#|&c}Scc}w)z%Gets all matching entries by details.)rLrr+r_rMrNrO)r(rPrQrOrrRr8s rget_all_by_detailszDNSCache.get_all_by_details scjjl**..% ?I#'(8#9l%Uejj=PU[_d_k_kUkllls A7 A70A7c|jj|jx}rt|j SgS)z8Returns a list of entries whose server matches the name.)r&r+rLr_rM)r(serverr7s rrZzDNSCache.entries_with_servers;((,,V\\^< <7 <() ) rc|jj|jx}rt|j SgS)z5Returns a list of entries whose key matches the name.)rr+rLr_rM)r(rPr7s rrVzDNSCache.entries_with_names7jjnnTZZ\2 27 2() ) rct}t|j|D]I}|jtk(s|j |r)t t|j|k(sG|cSyr#) rr^rVrNr is_expiredr ralias)r(rPrkrDrs r!current_entry_with_name_and_aliasz*DNSCache.current_entry_with_name_and_aliassb!#t55d;< F y())#.V,22e;  rc,t|jS)z1Return a copy of the list of current cache names.)r_rr's rnameszDNSCache.names*sDJJrct|}|D]O\}}}|j|||D]3}|j} || z tkDs||vs!|j ||d5Qy)Nr )setrTr-r_async_set_created_ttl) r( unique_typesanswersrD answers_rrsetrPrQrOrcreated_doubles r1async_mark_unique_records_older_than_1s_to_expirez:DNSCache.async_mark_unique_records_older_than_1s_to_expire.stG #/ @ D%33D%H @!'.(;6F-rHrJrTrXr[r+rcrerZrVrlrnrvrqrrrr!r!Cs!5<  ,&-^   ,.*m   @2@%@ @  @$ rr!N)r_DNSRecordCacheTyperrrr{ryrz)(r __future__rcollections.abcrheapqrrrtypingrr _dnsr r r rrrrr _utils.timerconstrrr]rdictrrr{rfloatr~intrrCrr!rrrrs,#$,,   -)"Hj':N:xWjPQ3Y %9 ::;      $' B B r