← Back
Editing: unix.cpython-38.pyc
U ��W[�U � @ s d Z ddlmZmZ ddlZddlZddlZddlZddlm Z m Z m Z m Z m Z mZ ddlmZmZmZ eed�szed��ddlmZmZmZmZmZmZ dd lm Z m Z dd l m Z m Z m!Z!m"Z" dd l#m$Z$ dd l%m&Z& dd l'm(Z( zddl m)Z) W n ek �r dZ)Y nX dd� Z*eej+�G dd� de,��Z-G dd� de,�Z.e)�rPe-Z-ne.Z-G dd� de-ej/�Z/dd� Z0G dd� de,�Z1G dd � d e1ej2�Z2G d d � d e-ej3�Z4G d d � d ej5�Z6eej7�G d!d"� d"e1ej2��Z8eej9feej:��� G d#d$� d$e8��Z;dS )%z� UNIX socket support for Twisted. End users shouldn't use this module directly - use the reactor APIs instead. Maintainer: Itamar Shtull-Trauring � )�division�absolute_importN)�EINTR�EMSGSIZE�EAGAIN� EWOULDBLOCK� ECONNREFUSED�ENOBUFS)� implementer�implementer_only� implementedBy�AF_UNIXz+UNIX sockets not supported on this platform)�main�base�tcp�udp�error� interfaces)�protocol�address)�lockfile�log�reflect�failure)�_coerceToFilesystemEncoding)�untilConcludes)� lazyByteSlice)�sendmsgc C s t �d| �}tjtj|fgS )zh Pack an integer into an ancillary data structure suitable for use with L{sendmsg.sendmsg}. �i)�structZpack�socket� SOL_SOCKETr � SCM_RIGHTS)�fdZpacked� r$ �7/usr/lib/python3/dist-packages/twisted/internet/unix.py�_ancillaryDescriptor) s r& c @ sH e Zd ZdZdZdZdd� Zdd� Zdd � Zd d � Z d d � Z dd� Z dS )� _SendmsgMixina Mixin for stream-oriented UNIX transports which uses sendmsg and recvmsg to offer additional functionality, such as copying file descriptors into other processes. @ivar _writeSomeDataBase: The class which provides the basic implementation of C{writeSomeData}. Ultimately this should be a subclass of L{twisted.internet.abstract.FileDescriptor}. Subclasses which mix in L{_SendmsgMixin} must define this. @ivar _sendmsgQueue: A C{list} of C{int} holding file descriptors which are currently buffered before being sent. @ivar _fileDescriptorBufferSize: An C{int} giving the maximum number of file descriptors to accept and queue for sending before pausing the registered producer, if there is one. N�@ c C s g | _ d S �N)� _sendmsgQueue��selfr$ r$ r% �__init__J s z_SendmsgMixin.__init__c C s t | j�| jkp| j�| �S )a� Determine whether the user-space send buffer for this transport is full or not. This extends the base determination by adding consideration of how many file descriptors need to be sent using L{sendmsg.sendmsg}. When there are more than C{self._fileDescriptorBufferSize}, the buffer is considered full. @return: C{True} if it is full, C{False} otherwise. )�lenr* �_fileDescriptorBufferSize�_writeSomeDataBase�_isSendBufferFullr+ r$ r$ r% r1 N s �z _SendmsgMixin._isSendBufferFullc C s | j �|� | �� | �� dS )zY Queue the given file descriptor to be sent and start trying to send it. N)r* �appendZ_maybePauseProducerZ startWriting)r, �filenor$ r$ r% �sendFileDescriptorf s z _SendmsgMixin.sendFileDescriptorc C s t | j�t |�krt�� S d}z�|t | j�k r�| j| }z&ttj| j|||d � t|�� W nX tjk r� } z8|jd t t fkr�| W Y � W �4S t j W Y � W �"S W 5 d}~X Y q X |d7 }q W 5 | jd|�= X t ||�}| j�| |�}z || W S tk �r | Y S X dS )zg Send as much of C{data} as possible. Also send any pending file descriptors. r N� )r. r* r ZFileDescriptorOverrunr r r r& �argsr r r �CONNECTION_LOSTr r0 � writeSomeData� TypeError)r, �data�indexr# �seZ limitedData�resultr$ r$ r% r8 o s2 �$ z_SendmsgMixin.writeSomeDatac C s� z t tj| j| j�\}}}W nJ tjk rf } z*|jd tkrHW Y �dS tj W Y �S W 5 d}~X Y nX |D ]P\}}}|tj kr�|tj kr�| � |� qlt jd| �� | �� | �| j�||d� ql| �|�S )a� Calls {IProtocol.dataReceived} with all available data and L{IFileDescriptorReceiver.fileDescriptorReceived} once for each received file descriptor in ancillary data. This reads up to C{self.bufferSize} bytes of data from its socket, then dispatches the data to protocol callbacks to be handled. If the connection is not lost through an error in the underlying recvmsg(), this function will return the result of the dataReceived call. r Nz�%(protocolName)s (on %(hostAddress)r) received unsupported ancillary data (level=%(cmsgLevel)r, type=%(cmsgType)r) from %(peerAddress)r.)�format� hostAddress� peerAddress� protocolName� cmsgLevel�cmsgType)r r Zrecvmsgr Z bufferSizer r6 r r r7 r! r" �%_ancillaryLevelSOLSOCKETTypeSCMRIGHTSr �msg�getHost�getPeer� _getLogPrefixr Z _dataReceived)r, r: Z ancillary�flagsr< rB rC �cmsgDatar$ r$ r% �doRead� s0 � � � z_SendmsgMixin.doReadc C s| t |�d }t�d| |�}tj�| j�rB|D ]}| j�|� q.n6tj d| � � | � � | � | j�d� |D ]}t �|� qhdS )a� Processes ancillary data with level SOL_SOCKET and type SCM_RIGHTS, indicating that the ancillary data payload holds file descriptors. Calls L{IFileDescriptorReceiver.fileDescriptorReceived} once for each received file descriptor or logs a message if the protocol does not implement L{IFileDescriptorReceiver}. @param cmsgData: Ancillary data payload. @type cmsgData: L{bytes} � r z�%(protocolName)s (on %(hostAddress)r) does not provide IFileDescriptorReceiver; closing file descriptor received (from %(peerAddress)r).)r> r? r@ rA N)r. r Zunpackr ZIFileDescriptorReceiverZ providedByr ZfileDescriptorReceivedr rE rF rG rH �os�close)r, rJ ZfdCountZfdsr# r$ r$ r% rD � s �z3_SendmsgMixin._ancillaryLevelSOLSOCKETTypeSCMRIGHTS) �__name__� __module__� __qualname__�__doc__r0 r/ r- r1 r4 r8 rK rD r$ r$ r$ r% r' 3 s ,'r' c @ s e Zd ZdZdS )�_UnsupportedSendmsgMixinz� Behaviorless placeholder used when C{twisted.python.sendmsg} is not available, preventing L{IUNIXTransport} from being supported. N)rO rP rQ rR r$ r$ r$ r% rS � s rS c @ s6 e Zd ZejZdd� Zedd� �Zdd� Z dd� Z d S ) �Serverc C s* t �| � tj�| |||d f|||� d S r) )r' r- r rT )r, Zsockr ZclientZserver� sessionno�reactorr$ r$ r% r- � s zServer.__init__c C s� t �|t jt j�}t�|�� �}|�|�}|dkr>|�� dS d}| |||� � d||�}d|j j j |j |�� f |_d|j j j |j |�� f |_|�|� |S )a Create a new L{Server} based on an existing connected I{SOCK_STREAM} socket. Arguments are the same as to L{Server.__init__}, except where noted. @param fileDescriptor: An integer file descriptor associated with a connected socket. The socket must be in non-blocking mode. Any additional attributes desired, such as I{FD_CLOEXEC}, must also be set already. @return: A new instance of C{cls} wrapping the socket given by C{fileDescriptor}. Nr z<%s #%s on %s>z%s,%s,%s)r �fromfdr � SOCK_STREAMr � UNIXAddress� getsocknameZ buildProtocolrN Z getpeernamer � __class__rO rU Zrepstr�logstrZmakeConnection) �clsZfileDescriptor�factoryrV �sktZ protocolAddr�protorU r, r$ r$ r% �_fromConnectedSocket� s( � � zServer._fromConnectedSocketc C s t �| j�� �S r) �r rY r rZ r+ r$ r$ r% rF s zServer.getHostc C s t �| jp d �S r) )r rY Zhostnamer+ r$ r$ r% rG s zServer.getPeerN) rO rP rQ r rT r0 r- � classmethodra rF rG r$ r$ r$ r% rT � s !rT c C s | dd� dkS )a� Determine whether the given unix socket path is in a filesystem namespace. While most PF_UNIX sockets are entries in the filesystem, Linux 2.2 and above support PF_UNIX sockets in an "abstract namespace" that does not correspond to any path. This function returns C{True} if the given socket path is stored in the filesystem and C{False} if the path is in this abstract namespace. Nr5 )� � r$ )�pathr$ r$ r% �_inFilesystemNamespace" s rg c @ s e Zd Zdd� ZdS )� _UNIXPortc C s t �| j�� �S )zV Returns a UNIXAddress. This indicates the server's address. rb r+ r$ r$ r% rF 0 s z_UNIXPort.getHostN)rO rP rQ rF r$ r$ r$ r% rh / s rh c @ s^ e Zd ZejZejZeZ dZ ddd�Z e dd� �Z d d � Zd d � Zd d� Zdd� Zdd� ZdS )�PortN�2 � r c C s4 t jj| | �|�j|||d� || _|| _d | _d S )N�rV )r ri r- � _buildAddr�name�mode�wantPID�_preexistingSocket)r, ZfileNamer^ �backlogro rV rp r$ r$ r% r- A s �z Port.__init__c C s. t �|| j| j�}| |�� ||d�}||_|S )a� Create a new L{Port} based on an existing listening I{SOCK_STREAM} socket. Arguments are the same as to L{Port.__init__}, except where noted. @param fd: An integer file descriptor associated with a listening socket. The socket must be in non-blocking mode. Any additional attributes desired, such as I{FD_CLOEXEC}, must also be set already. @return: A new instance of C{cls} wrapping the socket given by C{fd}. rl )r rW � addressFamily� socketTyperZ rq )r] rV r# r^ �portr, r$ r$ r% �_fromListeningDescriptorI s z Port._fromListeningDescriptorc C s: t �| jj�}t| d�r,d|td| j�f S d|f S d S )Nr � <%s on %r>� �<%s (not listening)>)r �qualr^ r[ �hasattrr ru )r, Z factoryNamer$ r$ r% �__repr__\ s �z Port.__repr__c C s t �|�S r) �r rY )r, rn r$ r$ r% rm d s zPort._buildAddrc C s\ t j�� t�d| �| j�td| j�f � | j r�t � | jd �| _ | j � � s`t�d| jd��n:| j js�z$t�t�| j�j�r�t�| j� W n Y nX | j�� z0| jdk r�| j}d| _n| �� }|�| j� W n6 tjk �r } zt�d| j|��W 5 d}~X Y nNX t| j��r(t� | j| j � |� | j � d| _ || _| jj!| _!d| _"| �#� 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. z%s starting on %rrx � .lockNzCannot acquire lockT�d )$r Z _reservedFDZreserver rE rH r^ r ru rp r ZFilesystemLock�lockFile�lockr �CannotListenErrorZclean�stat�S_ISSOCKrM �st_mode�removeZdoStartrq �createInternetSocket�bindr rg �chmodro Zlistenrr � connectedr3 Z numberAcceptsZ startReading�r, r_ �ler$ r$ r% �startListeningg sD � � " zPort.startListeningc C s t �dtd| j� � dS )z0 Log message for closing socket z(UNIX Port %s Closed)rx N)r rE r ru r+ r$ r$ r% �_logConnectionLostMsg� s �zPort._logConnectionLostMsgc C s< t | j�rt�| j� | jd k r*| j�� tj�| |� d S r) ) rg ru rM �unlinkr� Zunlockr ri �connectionLost�r, �reasonr$ r$ r% r� � s zPort.connectionLost)rj rk Nr )rO rP rQ r r rs rX rt rT Z transportr� r- rc rv r| rm r� r� r� r$ r$ r$ r% ri : s � 4ri c @ s< e Zd ZdZejZejZe j Z d dd�Z dd� Z dd � ZdS ) �ClientzA client for Unix sockets.Nr c C sj t �| � t�|�j}|| _| | _| _|rPt� |d �sP| � d d t � |�|� | � | j | �� d |� d S )Nr~ )r' r- r rY rn � connectorZ realAddress�addrr ZisLockedZ _finishInitr Z BadFileErrorZ doConnectr� )r, �filenamer� rV �checkPIDr$ r$ r% r- � s �zClient.__init__c C s t �| j�S r) )r rY r� r+ r$ r$ r% rG � s zClient.getPeerc C s t �d �S r) r} r+ r$ r$ r% rF � s zClient.getHost)Nr )rO rP rQ rR r r rs rX rt r � BaseClientr0 r- rG rF r$ r$ r$ r% r� � s r� c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )� Connectorc C s" t j�| |||� || _|| _d S r) )r � BaseConnectorr- r r� )r, r r^ ZtimeoutrV r� r$ r$ r% r- � s zConnector.__init__c C s t | j| | j| j�S r) )r� r rV r� r+ r$ r$ r% �_makeTransport� s zConnector._makeTransportc C s t �| j �S r) r} r+ r$ r$ r% �getDestination� s zConnector.getDestinationN)rO rP rQ r- r� r� r$ r$ r$ r% r� � s r� c @ sJ e Zd ZdZejZddd�Zdd� Zd d � Z d d � Z dd d�Z dd� Z dS )� DatagramPortz4 Datagram UNIX port, listening for packets. � rk Nc C s t jj| ||||d� || _dS )z.Initialize with address to listen on. )� maxPacketSizerV N)r ri r- ro )r, r� r` r� ro rV r$ r$ r% r- � s zDatagramPort.__init__c C s4 t �| jj�}t| d�r&d|| jf S d|f S d S )Nr rw ry )r rz r r[ r{ ru )r, rA r$ r$ r% r| � s zDatagramPort.__repr__c C s� t �d| jjt| j�f � z | �� }| jr8|�| j� W n4 tj k rn } zt � d | j|��W 5 d }~X Y nX | jr�t | j�r�t � | j| j� d| _|| _| jj| _d S )Nz%s starting on %sr5 )r rE r r[ �reprru r� r� r r r� rg rM r� ro r� r3 r� r$ r$ r% � _bindSocket� s "zDatagramPort._bindSocketc C s� z| j �||�W S t jk rz } zJ|jd }|tkrJ| �||� W Y �&S |tkr^t�d��n |tkrhn� W 5 d}~X Y nX dS )zWrite a datagram.r �message too longN) r Zsendtor r6 r �writer �MessageLengthErrorr )r, Zdatagramr r< �nor$ r$ r% r� � s zDatagramPort.writec C sl t �dt| j� � tj�| |� t| d�r6| j� � d| _ | j � � | ` | ` t| d�rh| j�d� | `dS )z Cleans up my socket. z(Port %s Closed)r r �dN)r rE r� ru r �BasePortr� r{ r ZdoStopr� r rN r3 r� �callbackr� r$ r$ r% r� s zDatagramPort.connectionLostc C s t �| jj�d | _d S )Nz (UDP))r rz r r[ r\ r+ r$ r$ r% � setLogStr s zDatagramPort.setLogStr)r� rk N)N) rO rP rQ rR r r rs r- r| r� r� r� r� r$ r$ r$ r% r� � s r� c @ sB e Zd ZdZddd�Zdd� Zd d � Zd d � Zd d� Zdd� Z dS )�ConnectedDatagramPortz+ A connected datagram UNIX socket. r� rk Nc C s. t |tj�st�t�| |||||� || _d S r) )� isinstancer ZConnectedDatagramProtocol�AssertionErrorr� r- � remoteaddr)r, r� r` r� ro Z bindAddressrV r$ r$ r% r- % s �z ConnectedDatagramPort.__init__c C sB z"| � � | j�| j� | �� W n | �t�� � Y nX d S r) )r� r Zconnectr� Z_connectToProtocol�connectionFailedr ZFailurer+ r$ r$ r% r� - s z$ConnectedDatagramPort.startListeningc C s | � � | j�|� | `dS )z� Called when a connection fails. Stop listening on the socket. @type reason: L{Failure} @param reason: Why the connection failed. N)Z stopListeningr r� r� r$ r$ r% r� 6 s z&ConnectedDatagramPort.connectionFailedc C s� d}|| j k r�z.| j�| j�\}}|t|�7 }| j�|� W q tjk r� } z<|jd }|t t t fkrrW Y � dS |t kr�| j� � n� W 5 d}~X Y q t�� Y qX qdS )z= Called when my socket is ready for reading. r N)Z maxThroughputr Zrecvfromr� r. r ZdatagramReceivedr r6 r r r r �connectionRefusedr Zdeferr)r, �readr: r� r< r� r$ r$ r% rK B s z ConnectedDatagramPort.doReadc C s� z| j �|�W S t jk r� } z\|jd }|tkrF| �|� W Y �:S |tkrZt�d��n |tkrn| j � � n |t krxn� W 5 d}~X Y nX dS )z# Write a datagram. r r� N) r �sendr r6 r r� r r� r r r� r )r, r: r< r� r$ r$ r% r� X s zConnectedDatagramPort.writec C s t �| j�S r) )r rY r� r+ r$ r$ r% rG o s z ConnectedDatagramPort.getPeer)r� rk NN) rO rP rQ rR r- r� r� rK r� rG r$ r$ r$ r% r� s � r� )<rR Z __future__r r rM r� r r �errnor r r r r r Zzope.interfacer r r r{ � ImportErrorZtwisted.internetr r r r r r r r Ztwisted.pythonr r r r Ztwisted.python.filepathr Ztwisted.python.utilr Ztwisted.python.compatr r r& ZIUNIXTransport�objectr' rS rT rg rh ri r� r� r� r� ZIUNIXDatagramTransportr� Z IUNIXDatagramConnectedTransportr� r� r$ r$ r$ r% �<module> sN .2 r M�
Save File
Cancel