File: //usr/lib/python3/dist-packages/hgext/zeroconf/__pycache__/Zeroconf.cpython-310.pyc
o
^�Ub� � @ sF 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,d4iZBi e/d5�e0d6�e1d7�e2d8�e3d9�e4d:�e5d;�e6d<�e7d=�e8d>�e9d?�e:d@�e;dA�e<dB�e=dC�e>dD�e?dE�e@dFeAd4i�ZCdGdH� 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�r!eadze� 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 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% �9/usr/lib/python3/dist-packages/hgext/zeroconf/Zeroconf.py�currentTimeMillis� � r' c @ � e Zd ZdS )�NonLocalNameExceptionN��__name__�
__module__� __qualname__r% r% r% r& r* � � r* c @ r) )�NonUniqueNameExceptionNr+ r% r% r% r& r0 � r/ r0 c @ r) )�NamePartTooLongExceptionNr+ r% r% r% r& r1 � r/ r1 c @ r) )�AbstractMethodExceptionNr+ r% r% r% r& r2 � r/ r2 c @ r) )�BadTypeInNameExceptionNr+ r% r% r% r& r3 � r/ r3 c @ s e Zd Zdd� ZdS )�
BadDomainNamec C s t �| d| � d S )Ns at position %s)� Exception�__init__)�self�posr% r% r& r6 � s zBadDomainName.__init__N)r, r- r. r6 r% r% r% r& r4 � s r4 c @ r) )�BadDomainNameCircularNr+ r% r% r% r& r9 � r/ r9 c @ �@ 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�r7 r> r? rA r% r% r& r6 � s
zDNSEntry.__init__c C s2 t |t�r| j|jko| j|jko| j|jkS dS )z&Equality test on name, type, and classr )�
isinstancer; r> r? rA �r7 �otherr% r% r& �__eq__� s
�
�zDNSEntry.__eq__c C � | � |� S �zNon-equality test�rH rF r% r% r& �__ne__ r( zDNSEntry.__ne__c C �& zt | W S ty d| Y S w )zClass accessor� ?(%s))�_CLASSES�KeyError)r7 rA r% r% r& �getClazz �
�zDNSEntry.getClazzc C rM )�
Type accessorrN )�_TYPESrP )r7 r? r% r% r& �getType rR zDNSEntry.getTypec C sd d|| � | j�| �| j�f }| jr|d7 }n|d7 }|| j7 }|dur,|d| 7 }|S |d7 }|S )�1String representation with additional informations %s[%s,%ss -unique,� ,Ns ,%s]� ])rU r? rQ rA rC r> )r7 �hdrrG �resultr% r% r& �toString s
�
�zDNSEntry.toStringN)
r, r- r. �__doc__r6 rH rL rQ rU r[ r% r% r% r& r; � s
r; c @ s( e Zd ZdZdd� Zdd� Zdd� ZdS ) � DNSQuestionzA DNS question entryc C s@ t jr
t|t�r
|�d�}|�d�st|��t�| |||� d S )N�asciis .local.) r �ispy3rE �str�encode�endswithr* r; r6 rD r% r% r& r6 ( s
zDNSQuestion.__init__c C s. | j |j ko| j|jkp| jtko| j|jkS )z6Returns true if the question is answered by the record)rA r? � _TYPE_ANYr> )r7 �recr% r% r& �
answeredBy/ s
�
�zDNSQuestion.answeredByc C s t �| dd�S )�String representations questionN)r; r[ �r7 r% r% r& �__repr__7 � zDNSQuestion.__repr__N)r, r- r. r\ r6 re rh r% r% r% r& r] % s
r] 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)r; r6 �ttlr' �created)r7 r> r? rA rl r% r% r& r6 ? s zDNSRecord.__init__c C s t |t�r t�| |�S dS )z Tests equality as per DNSRecordr )rE rj r; rH rF r% r% r& rH D �
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)r7 �msg�recordr% r% r& � suppressedByJ s
�zDNSRecord.suppressedByc C s | |kr|j | j d krdS 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 )rl rF r% r% r& rp 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 �rm rl )r7 �percentr% r% r& �getExpirationTimeY s zDNSRecord.getExpirationTimec C s t d| �d�| d �S )z%Returns the remaining TTL in seconds.r �d r# )�maxrv �r7 �nowr% r% r& �getRemainingTTL^ � zDNSRecord.getRemainingTTLc C � | � d�|kS )z(Returns true if this record has expired.rw �rv ry r% r% r& � isExpiredb ri zDNSRecord.isExpiredc C r} )z9Returns true if this record is at least half way expired.�2 r~ ry r% r% r& �isStalef ri zDNSRecord.isStalec C s |j | _ |j| _dS )zJSets this record's TTL and created time to that of
another record.Nrt rF r% r% r& �resetTTLj s zDNSRecord.resetTTLc C s t �)zAbstract method)r2 �r7 �outr% r% r& �writep s zDNSRecord.writec C s& d| j | �t� �|f }t�| d|�S )rV s %s/%s,%ss record)rl r{ r' r; r[ )r7 rG �argr% r% r& r[ t s
�zDNSRecord.toStringN)r, r- r. r\ r6 rH rs rp rv r{ r r� r� r� r[ r% r% r% r& rj < s rj c @ �0 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
S ) �
DNSAddresszA DNS address recordc C � t �| ||||� || _d S rk )rj r6 �address)r7 r> r? rA rl r� r% r% r& r6 � �
zDNSAddress.__init__c C � |� | jt| j�� dS �z'Used in constructing an outgoing packetN)� writeStringr� �lenr� r% r% r& r� � r| zDNSAddress.writec C � t |t�r | j|jkS dS )zTests equality on addressr )rE r� r� rF r% r% r& rH � rn zDNSAddress.__eq__c C s( zt �| j�W S ty | j Y S w �rf )�socket� inet_ntoar� r5 rg r% r% r& rh � s
�zDNSAddress.__repr__N�r, r- r. r\ r6 r� rH rh r% r% r% r& r� ~ � r� c @ r� ) �DNSHinfoz A DNS host information recordc C s" t �| ||||� || _|| _d S rk )rj r6 �cpu�os)r7 r> r? rA rl r� r� r% r% r& r6 � s
zDNSHinfo.__init__c C s, |� | jt| j�� |� | jt| j�� dS r� )r� r� r� r� r� r% r% r& r� � s zDNSHinfo.writec C s& t |t�r| j|jko| j|jkS dS )z Tests equality on cpu and osr )rE r� r� r� rF r% r% r& rH � s
zDNSHinfo.__eq__c C s | j d | j S )rf � )r� r� rg r% r% r& rh � � zDNSHinfo.__repr__Nr� r% r% r% r& r� � s r� c @ r� ) �
DNSPointerzA DNS pointer recordc C r� rk )rj r6 �alias)r7 r> r? rA rl r� r% r% r& r6 � r� zDNSPointer.__init__c C s |� | j� dS r� )� writeNamer� r� r% r% r& r� � r� zDNSPointer.writec C r� )zTests equality on aliasr )rE r� r� rF r% r% r& rH � rn zDNSPointer.__eq__c C s | � | j�S r� )r[ r� rg r% r% r& rh � r( zDNSPointer.__repr__Nr� r% r% r% r& r� � r� r� c @ r� ) �DNSTextzA DNS text recordc C r� rk )rj r6 �text)r7 r> r? rA rl r� r% r% r& r6 � r� zDNSText.__init__c C r� r� )r� r� r� r� r% r% r& r� � r| z
DNSText.writec C r� )zTests equality on textr )rE r� r� rF r% r% r& rH � rn zDNSText.__eq__c C s2 t | j�dkr| �| jdd� d �S | �| j�S )rf r Nr � ...)r� r� r[ rg r% r% r& rh � s zDNSText.__repr__Nr� r% r% r% r& r� � r� r� c @ r� ) �
DNSServicezA DNS service recordc C s. t �| ||||� || _|| _|| _|| _d S rk )rj r6 �priority�weight�port�server) r7 r> r? rA rl r� r� r� r� r% r% r& r6 � s
zDNSService.__init__c C s4 |� | j� |� | j� |� | j� |�| j� dS r� )�
writeShortr� r� r� r� r� r� r% r% r& r� � s zDNSService.writec C s> t |t�r | j|jko | j|jko | j|jko | j|jkS dS )z3Tests equality on priority, weight, port and serverr )rE r� r� r� r� r� rF r% r% r& rH � s
�
�
�zDNSService.__eq__c C s | � d| j| jf �S )rf s %s:%s)r[ r� r� rg r% r% r& rh � 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� questionsro � numquestions�
numanswers�numauthorities�numadditionals�
readHeader�
readQuestions�
readOthers)r7 r� r% r% r& r6 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� �r7 �format�length�infor% r% r& r� s
�
zDNSIncoming.readHeaderc C s� d}t �|�}td| j�D ]7}| �� }t �|| j| j| j| � �}| j|7 _zt||d |d �}| j �
|� W q
t yD Y q
w dS )z!Reads questions section of packets !HHr r N) r� r� �ranger� �readNamer� r� r� r] r� �appendr* )r7 r� r� �ir> r� �questionr% r% r& r� ! s
� ��zDNSIncoming.readQuestionsc C �B 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 �
�zDNSIncoming.readIntc C s( t | j| j �}| jd7 _| �|�S )z(Reads a character string from the packetr )�ordr� r� �
readString)r7 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� )r7 r� r� r� r� r% r% r& r� B s
�zDNSIncoming.readStringc C r� )z'Reads an unsigned short from the packet� !Hr r� r� r% r% r& �readUnsignedShortL r� z DNSIncoming.readUnsignedShortc C s� d}t �|�}| j| j | j }td|�D ]�}| �� }t �|| j| j | j | � �}| j |7 _ d}|d t
krKt ||d |d |d | � d��}n�|d t
ksW|d tkrht||d |d |d | �� �}nu|d tkr�t||d |d |d | � |d ��}n[|d tkr�t||d |d |d | �� | �� | �� | �� �}n;|d tkr�t||d |d |d | �� | �� �}n!|d tkr�t ||d |d |d | � d��}n | j |d 7 _ |dur�| j�|� qdS ) 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_Ar� r� � _TYPE_CNAME� _TYPE_PTRr� � _TYPE_TXTr� � _TYPE_SRVr� r� � _TYPE_HINFOr� r� �
_TYPE_AAAAro r� )r7 r� r� �nr� �domainr� rd r% r% r& r� V sf
� �� � �
� � ��zDNSIncoming.readOthersc C � | j t@ tkS )z Returns true if this is a query)r� �_FLAGS_QR_MASK�_FLAGS_QR_QUERYrg r% r% r& �isQuery� ri zDNSIncoming.isQueryc C r� )z"Returns true if this is a response)r� r� �_FLAGS_QR_RESPONSErg r% r% r& �
isResponse� ri zDNSIncoming.isResponsec C s | j ||| � �d�S )z6Reads a UTF-8 string of a given length from the packet�utf-8)r� �decode)r7 r� r� r% r% r& �readUTF� r| zDNSIncoming.readUTFc C s� d}| j }d}|} t| j||d � �}|d7 }|dkr nG|d@ }|dkr8d�|| �||�d f�}||7 }n,|dkr`|dk rD|d }|d@ d > t| j||d � �B }||kr]t|��|}nt|��q
|dkrn|| _ |S || _ |S )
z#Reads a domain name from the packet� ���Tr r � �.�? r )r� r� r� �joinr� r9 r4 )r7 rZ �off�next�firstr� �tr% r% r& r� � s6
"��zDNSIncoming.readNameN)r, r- r. r\ r6 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� ro � authorities� additionals)r7 r� r� r% r% r& r6 � s
zDNSOutgoing.__init__c C � | j �|� dS )zAdds a questionN)r� r� �r7 rr r% r% r& � addQuestion� r� zDNSOutgoing.addQuestionc C s |� |�s
| �|d� dS dS )zAdds an answerr N)rs �addAnswerAtTime)r7 �inprr r% r% r& � addAnswer� s
�zDNSOutgoing.addAnswerc C s6 |dur|dks
|� |�s| j�||f� dS dS dS )z6Adds an answer if if does not expire by a certain timeNr )r ro r� )r7 rr rz r% r% r& r� � s
�zDNSOutgoing.addAnswerAtTimec C r� )z Adds an authoritative answerN)r� r� r� r% r% r& �addAuthoritativeAnswer� r� z"DNSOutgoing.addAuthoritativeAnswerc C r� )zAdds an additional answerN)r� r� r� r% r% r& �addAdditionalAnswer� r� z DNSOutgoing.addAdditionalAnswerc C �. d}| j �t�|t|��� | jd7 _dS )z"Writes a single byte to the packet� !cr N)r� r� r� �pack�chrr� �r7 �valuer� r% r% r& � writeByte� � 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� )r7 �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 r� )z(Writes an unsigned integer to the packetr� r N)r� r� r� r� �intr� r� r% r% r& �writeInt� r� zDNSOutgoing.writeIntc C s6 dt |� d }| j�t�||�� | j|7 _dS )z Writes a string to the packet�!�sN)r` r� r� r� r� r� )r7 r� r� r� r% r% r& r� s zDNSOutgoing.writeStringc C s8 |� d�}t|�}|dkrt�| �|� | �||� dS )z5Writes a UTF-8 string of a given length to the packetr� r N)ra r� r1 r� r� )r7 r �utfstrr� r% r% r& �writeUTF s
zDNSOutgoing.writeUTFc C s� z| j | }W n0 ty7 | j| j |< |�d�}|d dkr%|dd� }|D ]}| �|� q'| �d� Y dS w | �|d? dB � | �|� dS )z"Writes a domain name to the packet� .r� � Nr r r� )r� rP r� �splitr r� )r7 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? rA )r7 r� r% r% r& �
writeQuestion, s zDNSOutgoing.writeQuestionc C s� | � |j� | �|j� |jr| jr| �|jtB � n| �|j� |dkr,| �|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? rC r� rA rB r rl r{ r� r� r� r� r� r� )r7 rr rz r� r� r% r% r& � writeRecord2 s
zDNSOutgoing.writeRecordc C s� | j ssd| _ | jD ]}| �|� q | jD ]
\}}| �||� q| jD ]}| �|d� q"| jD ]}| �|d� q.| �dt| j�� | �dt| j�� | �dt| j�� | �dt| j�� | �d| j � | j
rl| �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 ro r r� r� r� r� r� r� r� r� r� )r7 r� �answer�time_� authority�
additionalr% r% r& �packetI s&
zDNSOutgoing.packetN)r )r, r- r. r\ r6 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 rk )�cacherg r% r% r& r6 h s
zDNSCache.__init__c C sB z| j |j }W n ty g }| j |j< Y nw |�|� dS )z
Adds an entryN)r r= rP r� �r7 �entry�listr% r% r& �addk s �z DNSCache.addc C s2 z| j |j }|�|� W dS ty Y dS w )zRemoves an entryN)r r= �removerP r r% r% r& r s s �zDNSCache.removec C s6 z| j |j }||�|� W S ttfy Y dS w )zNGets an entry by key. Will return None if there is no
matching entry.N)r r= r� rP �
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.)r; r )r7 r> r? rA r r% r% r& � getByDetails� s
zDNSCache.getByDetailsc C s$ z| j | W S ty g Y S w )z5Returns a list of entries whose key matches the name.)r rP )r7 r> r% r% r& �entriesWithName� s
�zDNSCache.entriesWithNamec C s0 z t tj�| j�� ��W S ty g Y S w )z Returns a list of all entries)r � itertools�chain�
from_iterabler �valuesr5 rg r% r% r& �entries� s
�zDNSCache.entriesN) r, r- r. r\ r6 r r r r r r! r% r% r% r& r e s r c @ r: )�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�Threadr6 �zeroconf�readers�timeout� Condition� condition�start�r7 r% r% r% r& r6 � s
zEngine.__init__c C s� t � d s`| �� }t|�dkr!| j�� | j�| j� | j�� n8z.t�|g g | j�\}}}|D ] }z | j | �
� W q0 t yM t � d sKt �
� Y q0w W n t yX Y nw t � d rd S d S )Nr r )�globals�
getReadersr� r) �acquire�waitr' �release�selectr& � handle_readr5 � traceback� print_exc)r7 �rs�rr�wr�er�sockr% r% r& �run� s*
��� ��z
Engine.runc C s" | j �� | j�� }| j �� |S rk )r) r. r& �keysr0 �r7 rZ r% r% r& r- � s
zEngine.getReadersc C s, | j �� || j|< | j �� | j �� d S rk �r) r. r& �notifyr0 )r7 �readerr� r% r% r& � addReader� s
zEngine.addReaderc C s* | j �� | j|= | j �� | j �� d S rk r= )r7 r� r% r% r& � delReader� s
zEngine.delReaderc C s" | j �� | j �� | j �� d S rk )r) r. r> r0 rg r% r% r& r> � s
z
Engine.notifyN)
r, r- r. r\ r6 r: r- r@ rA r> r% r% r% r& r" � s r" c @ � 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 rk )r% �enginer@ r� r+ r% r% r& r6 � s zListener.__init__c
C s� | j j}z |�t�\}\}}W n tjy* } z|jtjkr%W Y d }~d S � d }~ww || _t|�}|� � r\|t
krD| j � |t t
� d S |t
krZ| j � |||� | j � |t t
� d S d S | j �|� d S rk )r% r� �recvfrom�_MAX_MSG_ABSOLUTE�error�errno�EBADFr� r� r� �
_MDNS_PORT� handleQuery�
_MDNS_ADDR� _DNS_PORT�handleResponse)r7 r9 r� �addrr� �erq r% r% r& r2 � s& ���zListener.handle_readN)r, r- r. r\ r6 r2 r% r% r% r& rC � s rC c @ rB )�ReaperzNA Reaper is used by this module to remove cache entries that
have expired.c C s t j�| � || _| �� d S rk )r# r$ r6 r% r* r+ r% r% r& r6 s zReaper.__init__c C s\ | j �d� t� d rd S t� }| j j�� D ]}|�|�r,| j �||� | j j�|� qq)NTi' r ) r% r/ r, r' r r! r � updateRecordr )r7 rz rr r% r% r& r: s
��z
Reaper.runN)r, r- r. r\ r6 r: r% r% r% r& rQ s rQ c @ r� ) �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$ r6 r% r? �listener�servicesr' �nexttime�
_BROWSER_TIME�delayr �done� addListenerr] r� � _CLASS_INr* )r7 r% r? rT r% r% r& r6 s � zServiceBrowser.__init__c s� � j tkri� j�j krk� �|�}z)�j� j�� }|s!|�� � n�j� j�� = � �fdd�}�j� |� W dS W n t
yY |sW� �j� j�� < � �fdd�}�j� |� Y nw � � d�}|�j k rm|�_ dS dS dS dS )z�Callback invoked by Zeroconf when new information arrives.
Updates information required by browser in the Zeroconf cache.c � �j �| �j� j�S rk )rT �
removeServicer? r� ��x�rr r7 r% r& �<lambda>< �
�z-ServiceBrowser.updateRecord.<locals>.<lambda>Nc r\ rk )rT �
addServicer? r� r^ r` r% r& ra D rb �K )
r? r� r> r rU r� r< r� r r� r5 rv rV )r7 r% rz rr �expired� oldrecord�callback�expiresr% r` r& rR 0 s.
� ��
�zServiceBrowser.updateRecordc C s d| _ | j�� d S )Nr )rY r% � notifyAllrg r% r% r& �cancelM s zServiceBrowser.cancelc C s� d }t � }t| j�dkr| j|kr| j�| j| � t� d s#| jr%d S t � }| j|krctt �}|�
t | j t
t�� | j�� D ]
}|�|�sM|�||� q@| j�|� || j | _td| jd �| _t| j�dkrp| j�d�}|d ury|| j� q)NTr r i N r )r' r� r rV r% r/ r, rY r� r� r� r] r? r� r[ rU r r r� �sendrX �min�pop)r7 �eventrz r� rr r% r% r& r: Q s.
�
�zServiceBrowser.runN)r, r- r. r\ r6 rR rj r: r% r% r% r& rS s rS c @ s� e Zd ZdZ d&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)
rb r3 r? r> r� r� r� r� r� �
setProperties) r7 r? r> r� r� r� r� �
propertiesr� r% r% r& r6 o s
zServiceInfo.__init__c C s� t |t�rX|| _g }d}|D ].}|| }|du rd}nt |t�r#|}nt |t�r0|r-d}nd}nd}|�d�||f�� q|D ]}d�|t�dt �
t |���|f�}q?|| _ dS || _ dS )z7Sets properties and text of this info from a dictionaryr N� true� false� =r� )
rE �dictrq r` r r� r� r� r� r �bytechrr� r� )r7 rq r rZ r= r� �suffix�itemr% r% r& rp � s2
��
zServiceInfo.setPropertiesc C s || _ zni }t|�}d}g }||k r/t|| �}|d7 }|�|||| � � ||7 }||k s|D ]:}|�d�}|dkrA|} d}
n |d|� } ||d d� }
|
dkrVd}
n|
dks\|
s^d}
| rk|�| �du rk|
|| < q1|| _W dS ty� t� � d| _Y dS w )z+Sets properties and text given a text fieldr r rt r� Nrr rs )
r� r� r� r� �findr rq r5 r3 r4 ) r7 r� rZ �endr� �strsr� r �eindexr= r� r% r% r& �setText� s> �
� �zServiceInfo.setTextc C � | j S )rS )r? rg r% r% r& rU � � zServiceInfo.getTypec C sD | j dur | j�d| j �r | jdt| j�t| j � d � S | jS )z
Name accessorNr r )r? r> rb r� rg r% r% r& �getName� s "zServiceInfo.getNamec C r~ )zAddress accessor)r� rg r% r% r& �
getAddress� r zServiceInfo.getAddressc C r~ )z
Port accessor)r� rg r% r% r& �getPort� r zServiceInfo.getPortc C r~ )zPriority accessor)r� rg r% r% r& � getPriority� r zServiceInfo.getPriorityc C r~ )zWeight accessor)r� rg r% r% r& � getWeight� r zServiceInfo.getWeightc C r~ )zProperties accessor)rq rg r% r% r& �
getProperties� r zServiceInfo.getPropertiesc C r~ )z
Text accessor)r� rg r% r% r& �getText� r zServiceInfo.getTextc C r~ )zServer accessor)r� rg r% r% r& � getServer� r zServiceInfo.getServerc C s� |dur\|� |�s^|jtkr |j| jkr|j| _dS dS |jtkrI|j| jkrG|j| _|j| _|j| _|j | _ | �
|||j � | jtt
�� dS dS |jtkr`|j| jkrb| �|j� dS dS dS dS dS )z-Updates service information from a DNS recordN)r r? r� r> r� r� r� r� r� r� rR r r r[ r� r} r� )r7 r% rz rr r% r% r& rR � s2
�
���
�zServiceInfo.updateRecordc C s~ t � }t}|| }|| }z�|�| t| jtt�� | jdu s(| jdu s(| j du r�||kr4W |�
| � dS ||kr�t t �}|�
t| jtt�� |�|j�| jtt�|� |�
t| jtt�� |�|j�| jtt�|� | jdur�|�
t| jtt�� |�|j�| jtt�|� |�|� || }|d }|�t||�| � t � }| jdu s(| jdu s(| j du s(d}W |�
| � |S |�
| � w )z�Returns true if the service could be discovered on the
network, and updates this object with details discovered.
Nr r r )r' �_LISTENER_TIMErZ r] r> rc r[ r� r� r� �removeListenerr� r� r� r� r� r r r� r� rk r/ rl ) r7 r% r' rz rX r� �lastr� rZ r% r% r& �request sh � '� ��� ���
���
�(
�zServiceInfo.requestc C s t |t�r |j| jkS dS )z Tests equality of service namer )rE ro r> rF r% r% r& rH S rn zServiceInfo.__eq__c C rI rJ rK rF r% r% r& rL Y r( zServiceInfo.__ne__c C sl d| j t�| �� �| jf }| jdu r|d7 }nt| j�dk r%|| j7 }n || jdd� d 7 }|d7 }|S )rf s service[%s,%s:%s,Ns None� � r� rX )r> r� r� r� r� r� r� r< r% r% r& rh ] s �
zServiceInfo.__repr__)NNr r NN)r, r- r. r\ r6 rp r} rU r� r� r� r� r� r� r� r� rR r� rH rL rh r% r% r% r& ro l s2
�% %; ro 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 rt�t�� �| _n|| _dtf| _t�tjtj�| _z| j� tj
tj d� | j� tj
tj d� W n t
yB Y nw | j� tjtjd� | j� tjtjd� z | j�| j� W n t
yi Y nw | 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� �
gethostbyname� gethostname�intfrJ �group�AF_INET�
SOCK_DGRAM�
setsockopt�
SOL_SOCKET� SO_REUSEADDR� SO_REUSEPORTr5 �SOL_IP�IP_MULTICAST_TTL�IP_MULTICAST_LOOP�bind�IP_ADD_MEMBERSHIP� inet_atonrL � listeners�browsersrU � servicetypesr r r# r( r) r" rD rC rT rQ �reaper)r7 � bindaddressr% r% r& r6 u sB
� ��
zZeroconf.__init__c C � | j �d�S )N� 127.0.0.1�r� �
startswithrg r% r% r& �
isLoopback� � zZeroconf.isLoopbackc C r� )Ns 169.254.r� rg r% r% r& � isLinklocal� r� 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/ r0 )r7 r' r% r% r& r/ � s
z
Zeroconf.waitc C s" | j �� | j �� | j �� dS )z Notifies all waiting threadsN)r) r. �
notify_allr0 rg r% r% r& ri � 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)ro r� )r7 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)�removeServiceListenerr� r� rS )r7 r? rT 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)r� rT rj )r7 rT �browserr% r% r& r� � s
��z Zeroconf.removeServiceListenerc C s4 | � |� || j|j�� < |j| jv r | j|j d7 < nd| j|j< t� }|}d}|dk r�||k r>| �|| � t� }q+tt t
B �}|� t |jt
t||j�d� |� t|jtt||j|j|j|j�d� |� t|jtt||j�d� |jr�|� t|jtt||j�d� | �|� |d7 }|t 7 }|dk s/dS dS )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) � checkServicerU r> r< r? r� r' r/ r� r� � _FLAGS_AAr� r� r� r[ r� r� r� r� r� r� r� r� r� r� r� r� rk �_REGISTER_TIME)r7 r� rl rz rV r� r� r% r% r& �registerService� sX
��
�
���
�zZeroconf.registerServicec C sB z!| j |j�� = | j|j dkr| j|j d8 < n| j|j= W n ty* Y nw t� }|}d}|dk r�||k rE| �|| � t� }q2tt t
B �}|� t |jt
td|j�d� |� t|jttd|j|j|j|j�d� |� t|jttd|j�d� |jr�|� t|jttd|j�d� | �|� |d7 }|t 7 }|dk s6dS dS )zUnregister a service.r r r N) rU r> r< r� r? rP r' r/ r� r� r� r� r� r� r[ r� r� r� r� r� r� r� r� r� r� r� r� rk �_UNREGISTER_TIME)r7 r� rz rV r� r� r% r% r& �unregisterService s`
� � ��
�
���
�zZeroconf.unregisterServicec C s t | j�dkr�t� }|}d}|dk r�||k r!| �|| � t� }qtttB �}| j�� D ]C}|�t |j
t t d|j
�d� |�t|j
tt d|j|j|j|j�d� |�t|j
tt d|j�d� |jro|�t|jtt d|j�d� q,| �|� |d7 }|t7 }|dk sdS dS dS )z#Unregister all registered services.r r r N) r� rU r' r/ r� r� r� r r� r� r? r� r[ r> r� r� r� r� r� r� r� r� r� r� r� r� rk r� )r7 rz rV r� r� r� r% r% r& �unregisterAllServices5 s` ���
�
����
��z Zeroconf.unregisterAllServicesc C s t � }|}d}|dk r�| j�|j�D ]1}|jtkrC|�|�sC|j|jkrC|j�d�dk rAd|j|j |j
|jf |_| � |� dS t �q||k rS| �
|| � t � }qtttB �}|| _|�t|jtt�� |�t|jttt|j�� | �|� |d7 }|t7 }|dk s dS dS )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� r r� r> ry r� r� r� r0 r/ r� r� r� �debugr� r] r[ r� r� �_DNS_TTLrk � _CHECK_TIME)r7 r� rz rV r� rr r� r% r% r& r� d sD
� �
� �
�zZeroconf.checkServicec C s\ t � }| j�|� |dur(| 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' r� r� r r r> re r rR ri )r7 rT r� rz rr r% r% r& rZ � s � zZeroconf.addListenerc C s0 z
| j �|� | �� W dS ty Y dS w )zRemoves a listener.N)r� r ri r5 )r7 rT r% r% r& r� � s �zZeroconf.removeListenerc C s& | j D ] }|�| ||� q| �� dS )zNUsed to notify listeners of new information that has updated
a record.N)r� rR ri )r7 rz rd rT r% r% r& rR � s
zZeroconf.updateRecordc C s| t � }|jD ]5}|�|�}|| j�� v r/|r | j�|� n| j�|�}|dur.|�|� |}n| j�|� | � ||� qdS )zlDeal with incoming response packets. All answers
are held in the cache, and listeners are notified.N)
r' ro r r r! r r r� r rR )r7 rq rz rr re r r% r% r& rN � s
� �zZeroconf.handleResponsec
C sN d}|t krtttB d�}|jD ]}|�|� q|jD ]�}|jtkrl|jdkrE| j �
� D ]}|du r8tttB �}|� |t dtt
t|�� q,| j�� D ] }|j|jkrj|du r\tttB �}|� |t |jtt
t|j�� qJqz�|du rwtttB �}|jtks�|jtkr�| j�� D ]}|j|j�� kr�|� |t|jtt
tB t|j�� q�| j�|j�� d�}|s�W q|jtks�|jtkr�|� |t|jtt
tB t|j|j |j |j�� |jt ks�|jtkr�|� |t |jt t
tB t|j �� |jtkr�|�!t|jtt
tB t|j�� W q t"�y t#�$� Y qw |du�r#|j%�r%|j&|_&| �'|||� dS dS dS )zKDeal with incoming query packets. Provides a response if
possible.Nr s _services._dns-sd._udp.local.)(rJ r� r� r� r� r� r? r� r> r� r; r� r� r[ r� rU r r� rc r� r< r� rB r� r r� r� r� r� r� r� r� r� r� r5 r3 r4 ro r� rk )r7 rq rO r� r� r� �stype�servicer% r% r& rK � s�
��
���� ��� ��
��
��� ��zZeroconf.handleQueryc C s4 z| j �|�� d||f� W dS ty Y dS w )zSends an outgoing packet.r N)r� �sendtor r5 )r7 r� rO r� r% r% r& rk # s
�z
Zeroconf.sendc C sh t � d dkr2dt � d< | �� | j�� | �� | j�tjtjt� t
�t� d� � | j� � dS dS )z^Ends the background threads, and prevent this instance from
servicing further queries.r r r r� N) r, ri rD r> r� r� r� r� �IP_DROP_MEMBERSHIPr� rL �closerg r% r% r& r� - s
��zZeroconf.closerk )r� )r, r- r. r\ r6 r� r� r/ ri r� r� r� r� r� r� r� r� rZ r� rR rN rK rL rJ rk 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.r� 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.)k�
__future__r r �
__author__� __email__� __version__rH r r1 r� r� r# r$ r3 � mercurialr �__all__r, r� r� r� r� rW rL rJ rM r� �_MAX_MSG_TYPICALrF r� r� r� r� � _FLAGS_TC� _FLAGS_RD� _FLAGS_RA�_FLAGS_Z� _FLAGS_AD� _FLAGS_CDr[ � _CLASS_CS� _CLASS_CH� _CLASS_HS� _CLASS_NONE�
_CLASS_ANYr@ rB r� �_TYPE_NS�_TYPE_MD�_TYPE_MFr� � _TYPE_SOA�_TYPE_MB�_TYPE_MG�_TYPE_MR�
_TYPE_NULL� _TYPE_WKSr� r� � _TYPE_MINFO�_TYPE_MXr� r� r� rc rO rT r' r5 r* r0 r1 r2 r3 r4 r9 �objectr; r] rj r� r� r� r� r� r� r� r r$ r" rC rQ rS ro r� r, �print�r�descr� r� r� r` r� r� r� r% r% r% r& �<module> sx
� �������� �
� � �
������9B! E #4B'U
T
�
����
�