File: //lib/python3/dist-packages/twisted/internet/iocpreactor/__pycache__/udp.cpython-38.pyc
U
��W[�6 � @ s d Z ddlZddlZddlZddlZddlZddlmZ ddlm Z m
Z
m Z m Z ddl
mZmZ ddlmZmZ ddlmZ ddlmZ dd lmZ dd
lmZ dd lmZm Z eee j e j e j �G d d
� d
e j ��Z!G dd� d�Z"ee j#�G dd� de"e!��Z$dS )z
UDP support for IOCP reactor
� N)� implementer)�defer�address�error�
interfaces)� isIPAddress�
isIPv6Address)�log�failure)�ERROR_IO_PENDING)�ERROR_CONNECTION_REFUSED)�ERROR_PORT_UNREACHABLE)�IReadWriteHandle)� iocpsupport�abstractc @ s� e Zd ZdZejZejZdZ dZ
d4dd�Z dd � Z d
d � Z
d d
� Zdd� Zdd� Zdd� Zdd� Zdd� Zdd� Zdd� Zd5d d �Zd d � Zd d!� Zd"d#� Zd$d%� Zd&d'� Zd6d(d)�Z d*d+� Z d,d-� Z d.d/� Z d0d1� Z d2d3� Z!dS )7�Portz�
UDP port, listening for packets.
@ivar addressFamily: L{socket.AF_INET} or L{socket.AF_INET6}, depending on
whether this port is listening on an IPv4 address or an IPv6 address.
FN� � c C sx || _ || _|| _|| _| �� d| _| �� tj� | |� t
�
| j | j �}t
�|�� �}t|�| _tt�d��| _dS )z>
Initialize with a numeric port to listen on.
N�i)�port�protocolZreadBufferSize� interface� setLogStr�_connectedAddr�_setAddressFamilyr �
FileHandle�__init__�socket�
addressFamily�
socketType�_iocpZ
maxAddrLen�fileno� bytearray�
addressBuffer�structZcalcsize�addressLengthBuffer)�selfr �protor �
maxPacketSize�reactor�sktZaddrLen� r+ �B/usr/lib/python3/dist-packages/twisted/internet/iocpreactor/udp.pyr * s
z
Port.__init__c C s@ t | j�rtj| _n(t| j�r(tj| _n| jr<t�| jd��dS )z8
Resolve address family for the socket.
znot an IPv4 or IPv6 addressN) r r r �AF_INET6r r �AF_INETr �InvalidAddressError�r& r+ r+ r, r @ s
�zPort._setAddressFamilyc C s. | j d k r d| jj| j f S d| jjf S d S )Nz
<%s on %s>z<%s not connected>)�_realPortNumberr � __class__r0 r+ r+ r, �__repr__M s
�z
Port.__repr__c C s | j S )z)
Return a socket object.
)r r0 r+ r+ r, � getHandleU s zPort.getHandlec C s | � � | �� dS )z�
Create and bind my socket, and begin listening on it.
This is called on unserialization, and must be called after creating a
server to begin listening on the specified port.
N)� _bindSocket�_connectToProtocolr0 r+ r+ r, �startListening\ s zPort.startListeningc C s | j �| j| j�S �N)r) � createSocketr r r0 r+ r+ r, r9 g s zPort.createSocketc
C s� z | � � }|�| j| jf� W n6 tjk
rT } zt�| j| j|��W 5 d }~X Y nX |�� d | _t �
d| � | j �| jf � d| _
|| _| jj| _d S )N� z%s starting on %sT)r9 Zbindr r r r ZCannotListenError� getsocknamer1 r �msg�
_getLogPrefixr � connectedr! �
getFileHandle)r& r* �ler+ r+ r, r5 k s $
�zPort._bindSocketc C s$ | j �| � | �� | j�| � d S r8 )r ZmakeConnectionZ startReadingr) ZaddActiveHandler0 r+ r+ r, r6 ~ s zPort._connectToProtocolc C s | j r | �|||� | �� d S r8 )Zreading�
handleRead�doRead�r& �rc�data�evtr+ r+ r, �cbRead� s z Port.cbReadc C s� |t jt jttfkr&| jr�| j�� n`|rHt� dt j
� |d�|f � n>z(| j� t
|jd |� �t�|j�� W n t�� Y nX d S )Nz error in recvfrom -- %s (%s)z
unknown error)�errno�WSAECONNREFUSED�
WSAECONNRESETr r
r r �connectionRefusedr r< Z errorcode�getZdatagramReceived�bytes�buffr Z makesockaddr� addr_buff�errrC r+ r+ r, rA � s
� �
�zPort.handleReadc C sn t �| j| �}| jd |_}| j |_}| j |_}t � | �
� ||||�\}}|rj|t krj| � |||� d S �Nr )
r ZEventrG Z _readBuffersrN r# rO r% �
addr_len_buffZrecvfromr? r rA )r& rF rN rO rR rD rE r+ r+ r, rB � s � z Port.doReadc
C s� | j r�|d| j fkst�z| j�|�W S tjk
r� } zb|jd }|tjkr`| �|� W Y �>S |tj krvt�
d��n"|tj tj t
tfkr�| j�� n� W 5 d}~X Y nX �n4|dks�t�t|d �s�t|d �s�|d dkr�t�|d d��t|d ��r| jtjk�rt�|d d��t|d ��rF| jtjk�rFt�|d d��z| j�||�W S tjk
�r� } zh|jd }|tjk�r�| �||� W Y �@S |tj k�r�t�
d��n"|tj tj t
tfk�r�W Y �
dS � W 5 d}~X Y nX dS )z~
Write a datagram.
@param addr: should be a tuple (ip, port), can be None in connected
mode.
Nr zmessage too longz <broadcast>z0write() only accepts IP addresses, not hostnamesz*IPv6 port write() called with IPv4 addressz*IPv4 port write() called with IPv6 address)r �AssertionErrorr �sendr �argsrH ZWSAEINTR�writeZ WSAEMSGSIZEZMessageLengthErrorrI rJ r r
r rK r r r/ r r- r. Zsendto)r& Zdatagram�addrZseZnor+ r+ r, rV � s`
�
�� � �
�
z
Port.writec C s | � d�|�|� d S )N� )rV �join)r& �seqrW r+ r+ r, �
writeSequence� s zPort.writeSequencec C sH | j rtd��t|�s*t|�s*t�|d��||f| _ | j�||f� dS )z-
'Connect' to remote server.
z\already connected, reconnecting is not currently supported (talk to itamar if you want this)z not an IPv4 or IPv6 address.N)r � RuntimeErrorr r r r/ r �connect)r& Zhostr r+ r+ r, r] � s � �
z Port.connectc C s. | � � | j�| � | jr*| j�d| j� d S rQ )Z stopReadingr) ZremoveActiveHandler> Z callLater�connectionLostr0 r+ r+ r, �_loseConnection� s zPort._loseConnectionc C s& | j rt�� }| _nd }| �� |S r8 )r> r ZDeferred�dr_ )r& �resultr+ r+ r, �
stopListening� s
zPort.stopListeningc C s t jdtdd� | �� d S )Nz-Please use stopListening() to disconnect port� )�
stacklevel)�warnings�warn�DeprecationWarningrb r0 r+ r+ r, �loseConnection� s
�zPort.loseConnectionc C s^ t �d| j � d| _tj�| |� | j�� | j� � | `| `
t | d�rZ| j �
d� | ` dS )z&
Cleans up my socket.
z(UDP Port %s Closed)Nr` )r r< r1 r r r^ r ZdoStopr �closer? �hasattrr` �callback)r& �reasonr+ r+ r, r^ s
zPort.connectionLostc C s | � | j�}d| | _dS )zP
Initialize the C{logstr} attribute to be used by C{logPrefix}.
z%s (UDP)N)r= r �logstr)r& � logPrefixr+ r+ r, r s zPort.setLogStrc C s | j S )zK
Returns the name of my class, to prefix log entries with.
)rm r0 r+ r+ r, rn s zPort.logPrefixc C sJ | j �� }| jt jkr$tjd|�� S | jt jkrFtjd|dd� �� S dS )z�
Return the local address of the UDP connection
@returns: the local address of the UDP connection
@rtype: L{IPv4Address} or L{IPv6Address}
�UDPNrc )ro )ro )r r; r r. r Z IPv4Addressr- Z IPv6Address�r& rW r+ r+ r, �getHost$ s
z Port.getHostc C s | j �t jt j|� dS )z�
Set whether this port may broadcast. This is disabled by default.
@param enabled: Whether the port may broadcast.
@type enabled: L{bool}
N)r �
setsockopt�
SOL_SOCKET� SO_BROADCAST)r& �enabledr+ r+ r, �setBroadcastAllowed2 s
�zPort.setBroadcastAllowedc C s t �| j�tjtj��S )z�
Checks if broadcast is currently allowed on this port.
@return: Whether this port may broadcast.
@rtype: L{bool}
)�operator�truthr �
getsockoptrs rt r0 r+ r+ r, �getBroadcastAllowed= s �zPort.getBroadcastAllowed)r r N)N)N)"�__name__�
__module__� __qualname__�__doc__r r. r Z
SOCK_DGRAMr ZdynamicReadBuffersr1 r r r3 r4 r7 r9 r5 r6 rG rA rB rV r[ r] r_ rb rh r^ r rn rq rv rz r+ r+ r+ r, r s< �
5
r c @ sl e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dd� Z
ddd�Z dd� Z dd� Z
ddd�ZdS ) �MulticastMixinz,
Implement multicast functionality.
c C s$ | j �t jt j�}t �t�d|��S )Nz@i)r ry �
IPPROTO_IP�IP_MULTICAST_IFZ inet_ntoar$ �pack)r& r r+ r+ r, �getOutgoingInterfaceO s z#MulticastMixin.getOutgoingInterfacec C s | j �|��| j�S )z.
Returns Deferred of success.
)r) �resolve� addCallback�
_setInterfacerp r+ r+ r, �setOutgoingInterfaceT s z#MulticastMixin.setOutgoingInterfacec C s" t �|�}| j �t jt j|� dS )Nr: )r � inet_atonrr r� r� )r& rW r r+ r+ r, r� [ s
z MulticastMixin._setInterfacec C s | j �t jt j�S r8 )r ry r� �IP_MULTICAST_LOOPr0 r+ r+ r, �getLoopbackModea s
�z MulticastMixin.getLoopbackModec C s* t �dt�|��}| j�tjtj|� d S )N�b)r$ r� rw rx r rr r� r� )r& �moder+ r+ r, �setLoopbackModef s �z MulticastMixin.setLoopbackModec C s | j �t jt j�S r8 )r ry r� �IP_MULTICAST_TTLr0 r+ r+ r, �getTTLl s
�zMulticastMixin.getTTLc C s$ t �d|�}| j�tjtj|� d S )N�B)r$ r� r rr r� r� )r& Zttlr+ r+ r, �setTTLq s zMulticastMixin.setTTLr c C s | j �|��| j|d�S )zF
Join a multicast group. Returns Deferred of success.
r: �r) r� r� �
_joinAddr1�r& rW r r+ r+ r, � joinGroupv s �zMulticastMixin.joinGroupc C s | j �|��| j||�S r8 )r) r� r� �
_joinAddr2)r& rW r rY r+ r+ r, r� ~ s �zMulticastMixin._joinAddr1c
C s� t �|�}t �|�}|r t j}nt j}z| j �t j||| � W nB t jk
r� } z"t�tj ||f|j
�� � W Y �S d }~X Y nX d S r8 ) r r� ZIP_ADD_MEMBERSHIPZIP_DROP_MEMBERSHIPrr r� r r
ZFailureZMulticastJoinErrorrU )r& r rW rY �cmd�er+ r+ r, r� � s
�zMulticastMixin._joinAddr2c C s | j �|��| j|d�S )zD
Leave multicast group, return Deferred of success.
r r� r� r+ r+ r, �
leaveGroup� s �zMulticastMixin.leaveGroupN)r )r )r{ r| r} r~ r� r� r� r� r� r� r� r� r� r� r� r+ r+ r+ r, r I s
r c @ s" e Zd ZdZd
dd�Zdd � ZdS ) �
MulticastPortz.
UDP Port that supports multicasting.
r r NFc C s t �| |||||� || _d S r8 )r r �listenMultiple)r& r r' r r( r) r� r+ r+ r, r � s zMulticastPort.__init__c C sB t �| �}| jr>|�tjtjd� ttd�r>|�tjtjd� |S )Nr: � SO_REUSEPORT) r r9 r� rr r rs Z SO_REUSEADDRrj r� )r& r* r+ r+ r, r9 � s
zMulticastPort.createSocket)r r NF)r{ r| r} r~ r r9 r+ r+ r+ r, r� � s �
r� )%r~ r rw r$ re rH Zzope.interfacer Ztwisted.internetr r r r Ztwisted.internet.abstractr r Ztwisted.pythonr r
Z"twisted.internet.iocpreactor.constr r r
Z'twisted.internet.iocpreactor.interfacesr Z twisted.internet.iocpreactorr r r ZIListeningPortZ
IUDPTransportZ
ISystemHandler r r ZIMulticastTransportr� r+ r+ r+ r, �<module> s( ( � 1Q