| Server IP : 80.87.202.40 / Your IP : 216.73.216.169 Web Server : Apache System : Linux rospirotorg.ru 5.14.0-539.el9.x86_64 #1 SMP PREEMPT_DYNAMIC Thu Dec 5 22:26:13 UTC 2024 x86_64 User : bitrix ( 600) PHP Version : 8.2.27 Disable Function : NONE MySQL : OFF | cURL : ON | WGET : ON | Perl : ON | Python : OFF | Sudo : ON | Pkexec : ON Directory : /lib64/python3.9/site-packages/hgext/zeroconf/__pycache__/ |
Upload File : |
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
�
����