ruạṛ
a �+�b� � & @ s d dl mZmZ dZdZdZd dlZd dlZd dlZd dl Z d dl Z d dl Z d dl Z d dl Z d dlmZ g d�Zd e� d< d Zd Zd Zd Zd ZdZdZdZdZdZdZ dZ d Z dZ dZ dZ!dZ"dZ#dZ$dZ%dZ&dZ'd Z(d Z)d Z*d Z+d Z,d!Z-dZ.dZ/d Z0d Z1d Z2d"Z3d#Z4d$Z5d%Z6d&Z7d'Z8d(Z9d)Z:d*Z;d+Z<d,Z=dZ>d-Z?d.Z@d ZAe'd/e(d0e)d1e*d2e+d3e,d4iZBe/d5e0d6e1d7e2d8e3d9e4d:e5d;e6d<e7d=e8d>e9d?e:d@e;dAe<dBe=dCe>dDe?dEe@dFeAd4iZCdGdH� ZDG dIdJ� dJeE�ZFG dKdL� dLeE�ZGG dMdN� dNeE�ZHG dOdP� dPeE�ZIG dQdR� dReE�ZJG dSdT� dTeE�ZKG dUdV� dVeK�ZLG dWdX� dXeM�ZNG dYdZ� dZeN�ZOG d[d\� d\eN�ZPG d]d^� d^eP�ZQG d_d`� d`eP�ZRG dadb� dbeP�ZSG dcdd� ddeP�ZTG dedf� dfeP�ZUG dgdh� dheM�ZVG didj� djeM�ZWG dkdl� dleM�ZXG dmdn� dne jY�ZZG dodp� dpeM�Z[G dqdr� dre jY�Z\G dsdt� dte jY�Z]G dudv� dveM�Z^G dwdx� dxeM�Z_e`dyk�readze� e_� Zbead{� d|d}d~d�Zce^d�d�e �dd��d�d d ec�Zeead�� eb�fee� ead�� ead�� ead�egeb�hd�d���� ead�� ead�� ead�egeb�hd�d���� ead�� ead�� eb�iee� ead�� eb�j� dS )�� )�absolute_import�print_functions Paul Scott-Murphys! paul at scott dash murphy dot coms 0.12N)�pycompat)s Zeroconfs ServiceInfos ServiceBrowser� _GLOBAL_DONE�} � �� �� i� z 224.0.0.251i� �5 i i� i # i � i i � �@ � � � � � � � � i� � � � � � � � � � � � � �! s ins css chs hss nones any� as nss mds mfs cnames soas mbs mgs mrs nulls wkss ptrs hinfos minfos mxs txts quadas srvc C s t � � d S )z#Current system time in milliseconds�� )�time� r% r% �=/usr/lib64/python3.9/site-packages/hgext/zeroconf/Zeroconf.py�currentTimeMillis� s r' c @ s e Zd ZdS )�NonLocalNameExceptionN��__name__� __module__� __qualname__r% r% r% r& r( � s r( c @ s e Zd ZdS )�NonUniqueNameExceptionNr) r% r% r% r& r- � s r- c @ s e Zd ZdS )�NamePartTooLongExceptionNr) r% r% r% r& r. � s r. c @ s e Zd ZdS )�AbstractMethodExceptionNr) r% r% r% r& r/ � s r/ c @ s e Zd ZdS )�BadTypeInNameExceptionNr) r% r% r% r& r0 � s r0 c @ s e Zd Zdd� ZdS )� BadDomainNamec C s t �| d| � d S )Ns at position %s)� Exception�__init__)�self�posr% r% r& r3 � s zBadDomainName.__init__N)r* r+ r, r3 r% r% r% r& r1 � s r1 c @ s e Zd ZdS )�BadDomainNameCircularNr) r% r% r% r& r6 � s r6 c @ s@ e Zd ZdZdd� Zdd� Zdd� Zdd � Zd d � Zd d � Z dS )�DNSEntryz A DNS entryc C s2 |� � | _|| _|| _|t@ | _|t@ dk| _d S )Nr )�lower�key�name�type� _CLASS_MASK�clazz� _CLASS_UNIQUE�unique�r4 r: r; r= r% r% r& r3 � s zDNSEntry.__init__c C s2 t |t�r.| j|jko,| j|jko,| j|jkS dS )z&Equality test on name, type, and classr )� isinstancer7 r: r; r= �r4 �otherr% r% r& �__eq__� s � �zDNSEntry.__eq__c C s | � |� S �zNon-equality test�rD rB r% r% r& �__ne__ s zDNSEntry.__ne__c C s* z t | W S ty$ d| Y S 0 dS )zClass accessor� ?(%s)N)�_CLASSES�KeyError)r4 r= r% r% r& �getClazz s zDNSEntry.getClazzc C s* z t | W S ty$ d| Y S 0 dS )� Type accessorrH N)�_TYPESrJ )r4 r; r% r% r& �getType s zDNSEntry.getTypec C sb d|| � | j�| �| j�f }| jr.|d7 }n|d7 }|| j7 }|durV|d| 7 }n|d7 }|S )�1String representation with additional informations %s[%s,%ss -unique,� ,Ns ,%s]� ])rN r; rK r= r? r: )r4 ZhdrrC �resultr% r% r& �toString s � zDNSEntry.toStringN) r* r+ r, �__doc__r3 rD rG rK rN rS r% r% r% r& r7 � s r7 c @ s( e Zd ZdZdd� Zdd� Zdd� ZdS ) � DNSQuestionzA DNS question entryc C s@ t jrt|t�r|�d�}|�d�s,t|��t�| |||� d S )N�asciis .local.) r Zispy3rA �str�encode�endswithr( r7 r3 r@ r% r% r& r3 ( s zDNSQuestion.__init__c C s. | j |j ko,| j|jks"| jtko,| j|jkS )z6Returns true if the question is answered by the record)r= r; � _TYPE_ANYr: )r4 �recr% r% r& � answeredBy/ s � �zDNSQuestion.answeredByc C s t �| dd�S )�String representations questionN)r7 rS �r4 r% r% r& �__repr__7 s zDNSQuestion.__repr__N)r* r+ r, rT r3 r\ r_ r% r% r% r& rU % s rU c @ sh e Zd ZdZdd� Zdd� Zdd� Zdd � Zd d � Zd d � Z dd� Z dd� Z dd� Z dd� Z dd� ZdS )� DNSRecordz.A DNS record - like a DNS entry, but has a TTLc C s" t �| |||� || _t� | _d S �N)r7 r3 �ttlr' �created)r4 r: r; r= rb r% r% r& r3 ? s zDNSRecord.__init__c C s t |t�rt�| |�S dS )z Tests equality as per DNSRecordr )rA r` r7 rD rB r% r% r& rD D s zDNSRecord.__eq__c C s |j D ]}| �|�r dS qdS )zdReturns true if any answer in a message can suffice for the information held in this record.r r )�answers�suppressedByAnswer)r4 �msg�recordr% r% r& � suppressedByJ s zDNSRecord.suppressedByc C s | |kr |j | j d kr dS dS )zwReturns true if another record has same name, type and class, and if its TTL is at least half of this record's.r r r )rb rB r% r% r& re R s z DNSRecord.suppressedByAnswerc C s | j || j d S )zXReturns the time at which this record will have expired by a certain percentage.r �rc rb )r4 �percentr% r% r& �getExpirationTimeY s zDNSRecord.getExpirationTimec C s t d| �d�| d �S )z%Returns the remaining TTL in seconds.r �d r# )�maxrk �r4 �nowr% r% r& �getRemainingTTL^ s zDNSRecord.getRemainingTTLc C s | � d�|kS )z(Returns true if this record has expired.rl �rk rn r% r% r& � isExpiredb s zDNSRecord.isExpiredc C s | � d�|kS )z9Returns true if this record is at least half way expired.�2 rq rn r% r% r& �isStalef s zDNSRecord.isStalec C s |j | _ |j| _dS )zJSets this record's TTL and created time to that of another record.Nri rB r% r% r& �resetTTLj s zDNSRecord.resetTTLc C s t �dS )zAbstract methodN)r/ �r4 �outr% r% r& �writep s zDNSRecord.writec C s& d| j | �t� �|f }t�| d|�S )rO s %s/%s,%ss record)rb rp r' r7 rS )r4 rC �argr% r% r& rS t s �zDNSRecord.toStringN)r* r+ r, rT r3 rD rh re rk rp rr rt ru rx rS r% r% r% r& r` < s r` c @ s0 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd S ) � DNSAddresszA DNS address recordc C s t �| ||||� || _d S ra )r` r3 �address)r4 r: r; r= rb r{ r% r% r& r3 � s zDNSAddress.__init__c C s |� | jt| j�� dS �z'Used in constructing an outgoing packetN)� writeStringr{ �lenrv r% r% r& rx � s zDNSAddress.writec C s t |t�r| j|jkS dS )zTests equality on addressr )rA rz r{ rB r% r% r& rD � s zDNSAddress.__eq__c C s, zt �| j�W S ty& | j Y S 0 dS )r] N)�socket� inet_ntoar{ r2 r^ r% r% r& r_ � s zDNSAddress.__repr__N�r* r+ r, rT r3 rx rD r_ r% r% r% r& rz ~ s rz c @ s0 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd S ) �DNSHinfoz A DNS host information recordc C s" t �| ||||� || _|| _d S ra )r` r3 �cpu�os)r4 r: r; r= rb r� r� r% r% r& r3 � s zDNSHinfo.__init__c C s, |� | jt| j�� |� | jt| j�� dS r| )r} r� r~ r� rv r% r% r& rx � s zDNSHinfo.writec C s& t |t�r"| j|jko | j|jkS dS )z Tests equality on cpu and osr )rA r� r� r� rB r% r% r& rD � s zDNSHinfo.__eq__c C s | j d | j S )r] � )r� r� r^ r% r% r& r_ � s zDNSHinfo.__repr__Nr� r% r% r% r& r� � s r� c @ s0 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd S ) � DNSPointerzA DNS pointer recordc C s t �| ||||� || _d S ra )r` r3 �alias)r4 r: r; r= rb r� r% r% r& r3 � s zDNSPointer.__init__c C s |� | j� dS r| )� writeNamer� rv r% r% r& rx � s zDNSPointer.writec C s t |t�r| j|jkS dS )zTests equality on aliasr )rA r� r� rB r% r% r& rD � s zDNSPointer.__eq__c C s | � | j�S )r] )rS r� r^ r% r% r& r_ � s zDNSPointer.__repr__Nr� r% r% r% r& r� � s r� c @ s0 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd S ) �DNSTextzA DNS text recordc C s t �| ||||� || _d S ra )r` r3 �text)r4 r: r; r= rb r� r% r% r& r3 � s zDNSText.__init__c C s |� | jt| j�� dS r| )r} r� r~ rv r% r% r& rx � s z DNSText.writec C s t |t�r| j|jkS dS )zTests equality on textr )rA r� r� rB r% r% r& rD � s zDNSText.__eq__c C s6 t | j�dkr&| �| jdd� d �S | �| j�S dS )r] r Nr � ...)r~ r� rS r^ r% r% r& r_ � s zDNSText.__repr__Nr� r% r% r% r& r� � s r� c @ s0 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd S ) � DNSServicezA DNS service recordc C s. t �| ||||� || _|| _|| _|| _d S ra )r` r3 �priority�weight�port�server) r4 r: r; r= rb r� r� r� r� r% r% r& r3 � s zDNSService.__init__c C s4 |� | j� |� | j� |� | j� |�| j� dS r| )� writeShortr� r� r� r� r� rv r% r% r& rx � s zDNSService.writec C s> t |t�r:| j|jko8| j|jko8| j|jko8| j|jkS dS )z3Tests equality on priority, weight, port and serverr )rA r� r� r� r� r� rB r% r% r& rD � s � � �zDNSService.__eq__c C s | � d| j| jf �S )r] s %s:%s)rS r� r� r^ r% r% r& r_ � s zDNSService.__repr__Nr� r% r% r% r& r� � s r� c @ sp e Zd ZdZdd� Zdd� Zdd� Zdd � Zd d � Zd d � Z dd� Z dd� Z dd� Z dd� Z dd� Zdd� ZdS )� DNSIncomingz/Object representation of an incoming DNS packetc C sL d| _ || _g | _g | _d| _d| _d| _d| _| �� | � � | � � dS )z/Constructor from string holding bytes of packetr N) �offset�data� questionsrd � numquestions� numanswers�numauthorities�numadditionals� readHeader� readQuestions� readOthers)r4 r� r% r% r& r3 s zDNSIncoming.__init__c C sz d}t �|�}t �|| j| j| j| � �}| j|7 _|d | _|d | _|d | _|d | _|d | _ |d | _ dS ) z Reads header portion of packets !HHHHHHr r r r r r N) �struct�calcsize�unpackr� r� �id�flagsr� r� r� r� �r4 �format�length�infor% r% r& r� s � zDNSIncoming.readHeaderc C s� d}t �|�}td| j�D ]p}| �� }t �|| j| j| j| � �}| j|7 _z$t||d |d �}| j � |� W q t y� Y q0 qdS )z!Reads questions section of packets !HHr r N) r� r� �ranger� �readNamer� r� r� rU r� �appendr( )r4 r� r� �ir: r� �questionr% r% r& r� ! s � zDNSIncoming.readQuestionsc C sB d}t �|�}t �|| j| j| j| � �}| j|7 _|d S )z Reads an integer from the packet� !Ir �r� r� r� r� r� r� r% r% r& �readInt2 s �zDNSIncoming.readIntc C s( t | j| j �}| jd7 _| �|�S )z(Reads a character string from the packetr )�ordr� r� � readString)r4 r� r% r% r& �readCharacterString< s z DNSIncoming.readCharacterStringc C sF d| }t �|�}t �|| j| j| j| � �}| j|7 _|d S )z0Reads a string of a given length from the packets !%dsr r� )r4 r~ r� r� r� r% r% r& r� B s �zDNSIncoming.readStringc C sB d}t �|�}t �|| j| j| j| � �}| j|7 _|d S )z'Reads an unsigned short from the packet� !Hr r� r� r% r% r& �readUnsignedShortL s �z DNSIncoming.readUnsignedShortc C s� d}t �|�}| j| j | j }td|�D �]�}| �� }t �|| j| j | j | � �}| j |7 _ d}|d t kr�t ||d |d |d | � d��}�n,|d t ks�|d tkr�t||d |d |d | �� �}n�|d tk�r t||d |d |d | � |d ��}n�|d tk�rLt||d |d |d | �� | �� | �� | �� �}nz|d tk�r�t||d |d |d | �� | �� �}nD|d tk�r�t ||d |d |d | � d��}n| j |d 7 _ |dur*| j�|� q*dS ) z@Reads answers, authorities and additionals section of the packets !HHiHr Nr r r r r )r� r� r� r� r� r� r� r� r� r� �_TYPE_Arz r� � _TYPE_CNAME� _TYPE_PTRr� � _TYPE_TXTr� � _TYPE_SRVr� r� � _TYPE_HINFOr� r� � _TYPE_AAAArd r� )r4 r� r� �nr� �domainr� r[ r% r% r& r� V sb � �� �� � � zDNSIncoming.readOthersc C s | j t@ tkS )z Returns true if this is a query)r� �_FLAGS_QR_MASK�_FLAGS_QR_QUERYr^ r% r% r& �isQuery� s zDNSIncoming.isQueryc C s | j t@ tkS )z"Returns true if this is a response)r� r� �_FLAGS_QR_RESPONSEr^ r% r% r& � isResponse� s zDNSIncoming.isResponsec C s | j ||| � �d�S )z6Reads a UTF-8 string of a given length from the packet�utf-8)r� �decode)r4 r� r~ r% r% r& �readUTF� s zDNSIncoming.readUTFc C s� d}| j }d}|}t| j||d � �}|d7 }|dkr:q�|d@ }|dkrnd�|| �||�d f�}||7 }q|dkr�|dk r�|d }|d@ d> t| j||d � �B }||kr�t|��|}qt|��q|dkr�|| _ n|| _ |S ) z#Reads a domain name from the packet� ���r r � �.�? r )r� r� r� �joinr� r6 r1 )r4 rR Zoff�next�firstr~ �tr% r% r& r� � s0 " zDNSIncoming.readNameN)r* r+ r, rT r3 r� r� r� r� r� r� r� r� r� r� r� r% r% r% r& r� � s ?r� c @ s� e Zd ZdZd$dd�Zdd� Zdd� Zd d � Zd d � Zd d� Z dd� Z dd� Z dd� Z dd� Z dd� Zdd� Zdd � Zd d � Zd d � Zd!d"� Zd#S )%� DNSOutgoingz+Object representation of an outgoing packetr c C sF d| _ d| _|| _|| _i | _g | _d| _g | _g | _g | _ g | _ d S )Nr r ) �finishedr� � multicastr� �namesr� �sizer� rd � authorities� additionals)r4 r� r� r% r% r& r3 � s zDNSOutgoing.__init__c C s | j �|� dS )zAdds a questionN)r� r� �r4 rg r% r% r& � addQuestion� s zDNSOutgoing.addQuestionc C s |� |�s| �|d� dS )zAdds an answerr N)rh �addAnswerAtTime)r4 �inprg r% r% r& � addAnswer� s zDNSOutgoing.addAnswerc C s. |dur*|dks|� |�s*| j�||f� dS )z6Adds an answer if if does not expire by a certain timeNr )rr rd r� )r4 rg ro r% r% r& r� � s zDNSOutgoing.addAnswerAtTimec C s | j �|� dS )z Adds an authoritative answerN)r� r� r� r% r% r& �addAuthoritativeAnswer� s z"DNSOutgoing.addAuthoritativeAnswerc C s | j �|� dS )zAdds an additional answerN)r� r� r� r% r% r& �addAdditionalAnswer� s z DNSOutgoing.addAdditionalAnswerc C s. d}| j �t�|t|��� | jd7 _dS )z"Writes a single byte to the packet� !cr N)r� r� r� �pack�chrr� �r4 �valuer� r% r% r& � writeByte� s zDNSOutgoing.writeBytec C s, d}| j �|t�||�� | jd7 _dS )z=Inserts an unsigned short in a certain position in the packetr� r N)r� �insertr� r� r� )r4 �indexr� r� r% r% r& � insertShort� s zDNSOutgoing.insertShortc C s* d}| j �t�||�� | jd7 _dS )z&Writes an unsigned short to the packetr� r N)r� r� r� r� r� r� r% r% r& r� � s zDNSOutgoing.writeShortc C s. d}| j �t�|t|��� | jd7 _dS )z(Writes an unsigned integer to the packetr� r N)r� r� r� r� �intr� r� r% r% r& �writeInt� s zDNSOutgoing.writeIntc C s6 dt |� d }| j�t�||�� | j|7 _dS )z Writes a string to the packet�!�sN)rW r� r� r� r� r� )r4 r� r� r� r% r% r& r} s zDNSOutgoing.writeStringc C s8 |� d�}t|�}|dkr t�| �|� | �||� dS )z5Writes a UTF-8 string of a given length to the packetr� r N)rX r~ r. r� r} )r4 r� Zutfstrr� r% r% r& �writeUTF s zDNSOutgoing.writeUTFc C s� z| j | }W n` tyn | j| j |< |�d�}|d dkrJ|dd� }|D ]}| �|� qN| �d� Y dS 0 | �|d? dB � | �|� dS )z"Writes a domain name to the packet� .r� � Nr r r� )r� rJ r� �splitr� r� )r4 r: r� �parts�partr% r% r& r� s zDNSOutgoing.writeNamec C s( | � |j� | �|j� | �|j� dS )z Writes a question to the packetN)r� r: r� r; r= )r4 r� r% r% r& � writeQuestion, s zDNSOutgoing.writeQuestionc C s� | � |j� | �|j� |jr6| jr6| �|jtB � n | �|j� |dkrX| �|j � n| �|� |�� t | j �}| j d7 _ |�| � | j d8 _ t d�| j |d� ��}| �||� dS )zPWrites a record (answer, authoritative answer, additional) to the packetr r r� N)r� r: r� r; r? r� r= r> r� rb rp r~ r� r� rx r� r� )r4 rg ro r� r� r% r% r& � writeRecord2 s zDNSOutgoing.writeRecordc C s� | j s�d| _ | jD ]}| �|� q| jD ]\}}| �||� q(| jD ]}| �|d� qD| jD ]}| �|d� q\| �dt| j�� | �dt| j�� | �dt| j�� | �dt| j�� | �d| j � | j r�| �dd� n| �d| j � d� | j �S )z�Returns a string containing the packet's bytes No further parts should be added to the packet once this is done.r r r� )r� r� r� rd r� r� r� r� r~ r� r� r� r� r� )r4 r� ZanswerZtime_Z authority� additionalr% r% r& �packetI s& zDNSOutgoing.packetN)r )r* r+ r, rT r3 r� r� r� r� r� r� r� r� r� r} r� r� r� r� r� r% r% r% r& r� � s" r� c @ sH e Zd ZdZdd� Zdd� Zdd� Zdd � Zd d � Zd d � Z dd� Z dS )�DNSCachezA cache of DNS entriesc C s i | _ d S ra )�cacher^ r% r% r& r3 h s zDNSCache.__init__c C sB z| j |j }W n" ty2 g }| j |j< Y n0 |�|� dS )z Adds an entryN)r� r9 rJ r� �r4 �entry�listr% r% r& �addk s z DNSCache.addc C s2 z| j |j }|�|� W n ty, Y n0 dS )zRemoves an entryN)r� r9 �removerJ r� r% r% r& r� s s zDNSCache.removec C s: z | j |j }||�|� W S ttfy4 Y dS 0 dS )zNGets an entry by key. Will return None if there is no matching entry.N)r� r9 r� rJ � ValueErrorr� r% r% r& �get{ s z DNSCache.getc C s t |||�}| �|�S )zRGets an entry by details. Will return None if there is no matching entry.)r7 r� )r4 r: r; r= r� r% r% r& � getByDetails� s zDNSCache.getByDetailsc C s( z | j | W S ty" g Y S 0 dS )z5Returns a list of entries whose key matches the name.N)r� rJ )r4 r: r% r% r& �entriesWithName� s zDNSCache.entriesWithNamec C s4 zt tj�| j�� ��W S ty. g Y S 0 dS )z Returns a list of all entriesN)r� � itertools�chain� from_iterabler� �valuesr2 r^ r% r% r& �entries� s zDNSCache.entriesN) r* r+ r, rT r3 r� r� r� r� r� r r% r% r% r& r� e s r� c @ s@ e Zd ZdZdd� Zdd� Zdd� Zdd � Zd d � Zd d � Z dS )�Engineac An engine wraps read access to sockets, allowing objects that need to receive data from sockets to be called back when the sockets are ready. A reader needs a handle_read() method, which is called when the socket it is interested in is ready for reading. Writers are not implemented here, because we only send short packets. c C s4 t j�| � || _i | _d| _t �� | _| �� d S )Nr ) � threading�Threadr3 �zeroconf�readers�timeout� Condition� condition�start�r4 r r% r% r& r3 � s zEngine.__init__c C s� t � d s�| �� }t|�dkrB| j�� | j�| j� | j�� q z^t�|g g | j�\}}}|D ]<}z| j | � � W q` t y� t � d s�t � � Y q`0 q`W q t y� Y q 0 q d S )Nr r )�globals� getReadersr~ r �acquire�waitr �release�selectr � handle_readr2 � traceback� print_exc)r4 ZrsZrr�wrZer�sockr% r% r& �run� s z Engine.runc C s" | j �� | j�� }| j �� |S ra )r r r �keysr �r4 rR r% r% r& r � s zEngine.getReadersc C s, | j �� || j|< | j �� | j �� d S ra �r r r �notifyr )r4 �readerr r% r% r& � addReader� s zEngine.addReaderc C s* | j �� | j|= | j �� | j �� d S ra r )r4 r r% r% r& � delReader� s zEngine.delReaderc C s" | j �� | j �� | j �� d S ra )r r r r r^ r% r% r& r � s z Engine.notifyN) r* r+ r, rT r3 r r r r r r% r% r% r& r � s r c @ s e Zd ZdZdd� Zdd� ZdS )�Listenera4 A Listener is used by this module to listen on the multicast group to which DNS messages are sent, allowing the implementation to cache information as it arrives. It requires registration with an Engine object in order to have the read() method called when a socket is available for reading.c C s || _ | j j�| | j j� d S ra )r �enginer r r r% r% r& r3 � s zListener.__init__c C s� | j j}z|�t�\}\}}W nB tjy` } z(|jtjkrJW Y d }~d S � W Y d }~n d }~0 0 || _t|�}|� � r�|t kr�| j � |t t � q�|t kr�| j � |||� | j � |t t � n | j �|� d S ra )r r Zrecvfrom�_MAX_MSG_ABSOLUTE�error�errno�EBADFr� r� r� � _MDNS_PORT� handleQuery� _MDNS_ADDR� _DNS_PORT�handleResponse)r4 r r� �addrr� �erf r% r% r& r � s zListener.handle_readN)r* r+ r, rT r3 r r% r% r% r& r � s r c @ s e Zd ZdZdd� Zdd� ZdS )�ReaperzNA Reaper is used by this module to remove cache entries that have expired.c C s t j�| � || _| �� d S ra )r r r3 r r r r% r% r& r3 s zReaper.__init__c C s^ | j �d� t� d rd S t� }| j j�� D ]*}|�|�r,| j �||� | j j�|� q,q d S )Ni' r ) r r r r' r� r rr � updateRecordr� )r4 ro rg r% r% r& r s z Reaper.runN)r* r+ r, rT r3 r r% r% r% r& r, s r, c @ s0 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd S ) �ServiceBrowserz�Used to browse for a service of a specific type. The listener object will have its addService() and removeService() methods called when this browser discovers changes in the services availability.c C sb t j�| � || _|| _|| _i | _t� | _t | _ g | _ d| _ | j� | t| jtt�� | �� dS )z%Creates a browser for a specific typer N)r r r3 r r; �listener�servicesr' �nexttime� _BROWSER_TIME�delayr� �done� addListenerrU r� � _CLASS_INr )r4 r r; r/ r% r% r& r3 s �zServiceBrowser.__init__c s� � j tkrΈ j�j krΈ �|�}zR�j� j�� }|sB|�� � n.�j� j�� = � �fdd�}�j� |� W dS W n@ t y� |s�� �j� j�� < � �fdd�}�j� |� Y n0 � � d�}|�j k r�|�_ dS )z�Callback invoked by Zeroconf when new information arrives. Updates information required by browser in the Zeroconf cache.c s �j �| �j� j�S ra )r/ Z removeServicer; r� ��x�rg r4 r% r& �<lambda>< s �z-ServiceBrowser.updateRecord.<locals>.<lambda>Nc s �j �| �j� j�S ra )r/ Z addServicer; r� r7 r9 r% r& r: D s ��K ) r; r� r: rr r0 r� r8 ru r� r� r2 rk r1 )r4 r ro rg �expiredZ oldrecord�callbackZexpiresr% r9 r& r- 0 s$ zServiceBrowser.updateRecordc C s d| _ | j�� d S )Nr )r4 r � notifyAllr^ r% r% r& �cancelM s zServiceBrowser.cancelc C s� d }t � }t| j�dkr4| j|kr4| j�| j| � t� d sD| jrHd S t � }| j|kr�tt �}|� t | j t t�� | j�� D ]}|�|�s~|�||� q~| j�|� || j | _td| jd �| _t| j�dkr�| j�d�}|d ur || j� q d S )Nr r i N r )r' r~ r� r1 r r r r4 r� r� r� rU r; r� r6 r0 r rr r� �sendr3 �min�pop)r4 Zeventro rw rg r% r% r& r Q s( zServiceBrowser.runN)r* r+ r, rT r3 r- r? r r% r% r% r& r. s r. c @ s� e Zd ZdZd&dd�Zdd� Zdd � Zd d � Zd d � Zdd� Z dd� Z dd� Z dd� Z dd� Z dd� Zdd� Zd d � Zd d � Zd d!� Zd"d#� Zd$d%� ZdS )'� ServiceInfozService informationNr c C sR |� |�st�|| _|| _|| _|| _|| _|| _|r>|| _n|| _| � |� dS )a� Create a service description. type: fully qualified service type name name: fully qualified service name address: IP address as unsigned short, network byte order port: port that the service runs on weight: weight of the service priority: priority of the service properties: dictionary of properties (or a string holding the bytes for the text field) server: fully qualified name for service host (defaults to name)N) rY r0 r; r: r{ r� r� r� r� � setProperties) r4 r; r: r{ r� r� r� � propertiesr� r% r% r& r3 o s zServiceInfo.__init__c C s� t |t�r�|| _g }d}|D ]\}|| }|du r6d}n.t |t�rF|}n t |t�r`|rZd}qdd}nd}|�d�||f�� q |D ]&}d�|t�dt � t |���|f�}q~|| _ n|| _ dS )z7Sets properties and text of this info from a dictionaryr� N� true� false� =r� ) rA �dictrE rW r� r� r� r� r� r Zbytechrr~ r� )r4 rE r� rR r9 r� �suffix�itemr% r% r& rD � s2 ��zServiceInfo.setPropertiesc C s || _ z�i }t|�}d}g }||k rXt|| �}|d7 }|�|||| � � ||7 }q |D ]t}|�d�}|dkr||} d} n:|d|� } ||d d� } | dkr�d} n| dks�| s�d} | r\|�| �du r\| || < q\|| _W n ty� t� � d| _Y n0 dS )z+Sets properties and text given a text fieldr r rH r� NrF rG ) r� r~ r� r� �findr� rE r2 r r ) r4 r� rR �endr� Zstrsr� r� Zeindexr9 r� r% r% r& �setText� s8 zServiceInfo.setTextc C s | j S )rL )r; r^ r% r% r& rN � s zServiceInfo.getTypec C sD | j dur>| j�d| j �r>| jdt| j�t| j � d � S | jS )z Name accessorNr� r )r; r: rY r~ r^ r% r% r& �getName� s "zServiceInfo.getNamec C s | j S )zAddress accessor)r{ r^ r% r% r& � getAddress� s zServiceInfo.getAddressc C s | j S )z Port accessor)r� r^ r% r% r& �getPort� s zServiceInfo.getPortc C s | j S )zPriority accessor)r� r^ r% r% r& � getPriority� s zServiceInfo.getPriorityc C s | j S )zWeight accessor)r� r^ r% r% r& � getWeight� s zServiceInfo.getWeightc C s | j S )zProperties accessor)rE r^ r% r% r& � getProperties� s zServiceInfo.getPropertiesc C s | j S )z Text accessor)r� r^ r% r% r& �getText� s zServiceInfo.getTextc C s | j S )zServer accessor)r� r^ r% r% r& � getServer� s zServiceInfo.getServerc C s� |dur�|� |�s�|jtkr2|j| jkr�|j| _nv|jtkr�|j| jkr�|j| _|j| _|j| _|j | _ | � |||j � | jtt �� n"|jtkr�|j| jkr�| �|j� dS )z-Updates service information from a DNS recordN)rr r; r� r: r� r{ r� r� r� r� r- r� r� r6 r� rN r� )r4 r ro rg r% r% r& r- � s( �� zServiceInfo.updateRecordc C sl t � }t}|| }|| }�z>|�| t| jtt�� | jdu sT| jdu sT| j du �rJ||krlW |� | � dS ||k�r.t t �}|� t| jtt�� |�|j�| jtt�|� |� t| jtt�� |�|j�| jtt�|� | jdu�r|� t| jtt�� |�|j�| jtt�|� |�|� || }|d }|�t||�| � t � }q4d}W |� | � n |� | � 0 |S )z�Returns true if the service could be discovered on the network, and updates this object with details discovered. Nr r r )r' �_LISTENER_TIMEr5 rU r: rZ r6 r� r{ r� �removeListenerr� r� r� r� r� r� r� r� r� r@ r rA ) r4 r r ro r3 r� Zlastrw rR r% r% r& �request sp ����' � ��� ��� ��� zServiceInfo.requestc C s t |t�r|j| jkS dS )z Tests equality of service namer )rA rC r: rB r% r% r& rD S s zServiceInfo.__eq__c C s | � |� S rE rF rB r% r% r& rG Y s zServiceInfo.__ne__c C sl d| j t�| �� �| jf }| jdu r0|d7 }n0t| j�dk rJ|| j7 }n|| jdd� d 7 }|d7 }|S )r] s service[%s,%s:%s,Ns None� � r� rQ )r: r r� rP r� r� r~ r r% r% r& r_ ] s � zServiceInfo.__repr__)NNr r NN)r* r+ r, rT r3 rD rN rN rO rP rQ rR rS rT rU rV r- rY rD rG r_ r% r% r% r& rC l s0 � % %;rC c @ s� e Zd ZdZd*dd�Zdd� Zdd� Zd d � Zd d � Zd+dd�Z dd� Z dd� Z e fdd�Z dd� Zdd� Zdd� Zd d � Zd d � Zd d!� Zd"d#� Zd$d%� Zeefd&d'�Zd(d)� ZdS ),�Zeroconfz�Implementation of Zeroconf Multicast DNS Service Discovery Supports registration, unregistration, queries and browsing. Nc C sD dt � d< |du r$t�t�� �| _n|| _dtf| _t�tjtj�| _z,| j� tj tj d� | j� tj tj d� W n t y� Y n0 | j� tjtjd� | j� tjtjd� z| j�| j� W n t y� Y n0 | j� tjtjt�t�t�d� � g | _g | _i | _i | _t� | _t� � | _ t | �| _ t | �| _!t"| �| _#dS ) zxCreates an instance of the Zeroconf class, establishing multicast communications, listening and reaping threads.r r Nr� r � �� �0.0.0.0)$r r Z gethostbynameZ gethostname�intfr% �groupZAF_INETZ SOCK_DGRAM� setsockoptZ SOL_SOCKETZ SO_REUSEADDRZ SO_REUSEPORTr2 �SOL_IPZIP_MULTICAST_TTLZIP_MULTICAST_LOOPZbindZIP_ADD_MEMBERSHIP� inet_atonr' � listeners�browsersr0 � servicetypesr� r� r r r r r r r/ r, Zreaper)r4 Z bindaddressr% r% r& r3 u s> � zZeroconf.__init__c C s | j �d�S )N� 127.0.0.1�r` � startswithr^ r% r% r& � isLoopback� s zZeroconf.isLoopbackc C s | j �d�S )Ns 169.254.ri r^ r% r% r& � isLinklocal� s zZeroconf.isLinklocalc C s( | j �� | j �|d � | j �� dS )zRCalling thread waits for a given number of milliseconds or until notified.r# N)r r r r )r4 r r% r% r& r � s z Zeroconf.waitc C s" | j �� | j �� | j �� dS )z Notifies all waiting threadsN)r r r> r r^ r% r% r& r> � s zZeroconf.notifyAll� c C s t ||�}|�| |�r|S dS )z�Returns network's service information for a particular name and type, or None if no service matches by the timeout, which defaults to 3 seconds.N)rC rY )r4 r; r: r r� r% r% r& �getServiceInfo� s zZeroconf.getServiceInfoc C s" | � |� | j�t| ||�� dS )z�Adds a listener for a particular service type. This object will then have its updateRecord method called when information arrives for that type.N)�removeServiceListenerrf r� r. )r4 r; r/ r% r% r& �addServiceListener� s zZeroconf.addServiceListenerc C s$ | j D ]}|j|kr|�� ~qdS )z<Removes a listener from the set that is currently listening.N)rf r/ r? )r4 r/ Zbrowserr% r% r& ro � s z Zeroconf.removeServiceListenerc C s. | � |� || j|j�� < |j| jv r<| j|j d7 < n d| j|j< t� }|}d}|dk �r*||k r~| �|| � t� }qVtt t B �}|� t |jt t||j�d� |� t|jtt||j|j|j|j�d� |� t|jtt||j�d� |j�r|� t|jtt||j�d� | �|� |d7 }|t 7 }qVdS )a Registers service information to the network with a default TTL of 60 seconds. Zeroconf will then respond to requests for information for that service. The name of the service may be changed if needed to make it unique on the network.r r r N) � checkServicer0 r: r8 r; rg r' r r� r� � _FLAGS_AAr� r� r� r6 r� r� r� r� r� r� r� r� r� r{ rz r� r@ �_REGISTER_TIME)r4 r� rb ro r1 r� rw r% r% r& �registerService� sV �� � ��� zZeroconf.registerServicec C s< zB| j |j�� = | j|j dkr6| j|j d8 < n | j|j= W n tyT Y n0 t� }|}d}|dk �r8||k r�| �|| � t� }qdtt t B �}|� t |jt td|j�d� |� t|jttd|j|j|j|j�d� |� t|jttd|j�d� |j�r |� t|jttd|j�d� | �|� |d7 }|t 7 }qddS )zUnregister a service.r r r N) r0 r: r8 rg r; rJ r' r r� r� rr r� r� r� r6 r� r� r� r� r� r� r� r� r{ rz r� r� r@ �_UNREGISTER_TIME)r4 r� ro r1 r� rw r% r% r& �unregisterService sZ �� � ��� zZeroconf.unregisterServicec C s t | j�dkr�t� }|}d}|dk r�||k rB| �|| � t� }q tttB �}| j�� D ]�}|�t |j t t d|j �d� |�t|j tt d|j|j|j|j�d� |�t|j tt d|j�d� |jrX|�t|jtt d|j�d� qX| �|� |d7 }|t7 }q dS )z#Unregister all registered services.r r r N) r~ r0 r' r r� r� rr r r� r� r; r� r6 r: r� r� r� r� r� r� r� r� r� r{ rz r� r@ ru )r4 ro r1 r� rw r� r% r% r& �unregisterAllServices5 sX ��� � ��� z Zeroconf.unregisterAllServicesc C s t � }|}d}|dk �r| j�|j�D ]b}|jtkr&|�|�s&|j|jkr&|j�d�dk r�d|j|j |j |jf |_| � |� dS t �q&||k r�| � || � t � }qtttB �}|| _|�t|jtt�� |�t|jttt|j�� | �|� |d7 }|t7 }qdS )znChecks the network for a unique service name, modifying the ServiceInfo passed in if it is not unique.r r r� s %s.[%s:%d].%sNr )r' r� r� r; r� rr r� r: rL r{ r� rq r- r r� r� rr �debugr� rU r6 r� r� �_DNS_TTLr@ � _CHECK_TIME)r4 r� ro r1 r� rg rw r% r% r& rq d sD �� �� � zZeroconf.checkServicec C s\ t � }| j�|� |durP| j�|j�D ]&}|�|�r(|�|�s(|�| ||� q(| � � dS )z�Adds a listener for a given question. The listener will have its updateRecord method called when information is available to answer the question.N) r' re r� r� r� r: r\ rr r- r> )r4 r/ r� ro rg r% r% r& r5 � s zZeroconf.addListenerc C s0 z| j �|� | �� W n ty* Y n0 dS )zRemoves a listener.N)re r� r> r2 )r4 r/ r% r% r& rX � s zZeroconf.removeListenerc C s& | j D ]}|�| ||� q| �� dS )zNUsed to notify listeners of new information that has updated a record.N)re r- r> )r4 ro r[ r/ r% r% r& r- � s zZeroconf.updateRecordc C s| t � }|jD ]j}|�|�}|| j�� v r^|r:| j�|� qj| j�|�}|durj|�|� |}n | j�|� | � ||� q dS )zlDeal with incoming response packets. All answers are held in the cache, and listeners are notified.N) r' rd rr r� r r� r� ru r� r- )r4 rf ro rg r< r� r% r% r& r) � s zZeroconf.handleResponsec C s` d}|t kr0tttB d�}|jD ]}|�|� q |jD �]�}|jtkr�|jdkr�| j � � D ]0}|du rrtttB �}|� |t dtt t|�� qZ| j�� D ]@}|j|jkr�|du r�tttB �}|� |t |jtt t|j�� q�q6�z8|du r�tttB �}|jtk�s |jtk�rN| j�� D ]8}|j|j�� k�r|� |t|jtt tB t|j�� �q| j�|j�� d�}|�slW q6|jtk�s�|jtk�r�|� |t|jtt tB t|j|j |j |j�� |jt k�s�|jtk�r�|� |t |jt t tB t|j �� |jtk�r|�!t|jtt tB t|j�� W q6 t"�y0 t#�$� Y q60 q6|du�r\|j%�r\|j&|_&| �'|||� dS )zKDeal with incoming query packets. Provides a response if possible.Nr s _services._dns-sd._udp.local.)(r% r� r� rr r� r� r; r� r: rg r r� r� r6 ry r0 r r� rZ r� r8 rz r> r{ r� r� r� r� r� r� r� r� r� r� r2 r r rd r� r@ )r4 rf r* r� rw r� �stypeZservicer% r% r& r& � s� �� �� �� �� �� �� zZeroconf.handleQueryc C s4 z | j �|�� d||f� W n ty. Y n0 dS )zSends an outgoing packet.r N)r Zsendtor� r2 )r4 rw r* r� r% r% r& r@ # s z Zeroconf.sendc C sd t � d dkr`dt � d< | �� | j�� | �� | j�tjtjt� t �t� d� � | j� � dS )z^Ends the background threads, and prevent this instance from servicing further queries.r r r r_ N) r r> r r rw r rb rc ZIP_DROP_MEMBERSHIPrd r' �closer^ r% r% r& r| - s �zZeroconf.close)N)rm )r* r+ r, rT r3 rk rl r r> rn rp ro ry rt rv rw rq r5 rX r- r) r& r' r% r@ r| r% r% r% r& r\ o s( 3 21/% l r\ �__main__s3 Multicast DNS Service Discovery for Python, versions' 1. Testing registration of a service...s 0.10s test values another value)s versionr" � bs _http._tcp.local.s! My Service Name._http._tcp.local.rh i� s Registering service...s Registration done.s* 2. Testing query of service information...s Getting ZOE service:s ZOE._http._tcp.local.s Query done.s" 3. Testing query of own service...s Getting self:s/ 4. Testing unregister of service information...s Unregister done.)kZ __future__r r � __author__Z __email__� __version__r# r� r r r� r r$ r Z mercurialr �__all__r ru rz rs rW r2 r' r% r( ry Z_MAX_MSG_TYPICALr! r� r� r� rr Z _FLAGS_TCZ _FLAGS_RDZ _FLAGS_RAZ_FLAGS_ZZ _FLAGS_ADZ _FLAGS_CDr6 Z _CLASS_CSZ _CLASS_CHZ _CLASS_HSZ _CLASS_NONEZ _CLASS_ANYr< r> r� Z_TYPE_NSZ_TYPE_MDZ_TYPE_MFr� Z _TYPE_SOAZ_TYPE_MBZ_TYPE_MGZ_TYPE_MRZ _TYPE_NULLZ _TYPE_WKSr� r� Z _TYPE_MINFOZ_TYPE_MXr� r� r� rZ rI rM r' r2 r( r- r. r/ r0 r1 r6 �objectr7 rU r` rz r� r� r� r� r� r� r� r r r r, r. rC r\ r* �print�rZdescrd r� rt rW rn rv r| r% r% r% r& �<module> s< N � �9B! E #4B'U T � ����
cải xoăn