File: //usr/lib/python3/dist-packages/mercurial/thirdparty/__pycache__/selectors2.cpython-310.pyc
o
�]LbVk � @ s d Z ddlmZ ddlZddlZddlZddlZddlZddlZddl Z ddl
m Z ej Z ej
Z
ze jZW n eyA e j ZY nw dZdZdZd Zd
Zg d �Zd ZdZdae� Zee ej fZ e d
g d��Z G dd� de
�Z dd� Z!G dd� de�Z"e#ed�r�G dd� de"�Z$e�%d� e j&r�G dd� de�Z'G dd� de$�Z(e�%d� e(Z$e#ed �r�G d d � d e"�Z)e�%d � e#ed �r�G d d!� d!e"�Z*e�%d!� e#ed"�r�G d#d$� d$e"�Z+e�%d$� e#ed%�r�G d&d'� d'e"�Z,e�%d'� d(d)� Z-ej.d*kr�d+d,� Z/nd-d,� Z/d.d/� Z0dS )0z. Back-ported, durable, and portable selectors � )�absolute_importN� )�pycompatzSeth Michael Larsonz sethmichaellarson@protonmail.comz2.0.0�MITz3https://www.github.com/SethMichaelLarson/selectors2)�
EVENT_READ� EVENT_WRITE� SelectorKey�DefaultSelector� BaseSelector� r )�fileobj�fd�events�datac @ s0 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
S ) �_SelectorMappingz* Mapping of file objects to selector keys c C s
|| _ d S �N)� _selector��self�selector� r �A/usr/lib/python3/dist-packages/mercurial/thirdparty/selectors2.py�__init__F �
z_SelectorMapping.__init__c C � t | jj�S r )�lenr �
_fd_to_key�r r r r �__len__I � z_SelectorMapping.__len__c C s8 z
| j �|�}| j j| W S ty td�|���w �Nz{0!r} is not registered.)r �_fileobj_lookupr �KeyError�format)r r r
r r r � __getitem__L s �z _SelectorMapping.__getitem__c C r r )�iterr r r r r r �__iter__S r z_SelectorMapping.__iter__N)�__name__�
__module__� __qualname__�__doc__r r r$ r& r r r r r C s r c
C s^ t | t�r| }nzt| �� �}W n tttfy! td�| ���w |dk r-td�|���|S )zl Return a file descriptor from a file object. If
given an integer will simply return that integer back. zInvalid file object: {0!r}r z Invalid file descriptor: {0})�
isinstance�int�fileno�AttributeError� TypeError�
ValueErrorr# )r r
r r r �_fileobj_to_fdW s
�r1 c @ sv e Zd ZdZdd� Zdd� Zddd�Zd d
� Zdd d �Zdd
d�Z dd� Z
dd� Z dd� Z dd� Z
dd� Zdd� ZdS ) r
a/ Abstract Selector class
A selector supports registering file objects to be monitored
for specific I/O events.
A file object is a file descriptor or any object with a
`fileno()` method. An arbitrary object can be attached to the
file object which can be used for example to store context info,
a callback, etc.
A selector can use various implementations (select(), poll(), epoll(),
and kqueue()) depending on the platform. The 'DefaultSelector' class uses
the most efficient implementation for the current platform.
c C s i | _ t| �| _d S r )r r �_mapr r r r r u s zBaseSelector.__init__c C sD zt |�W S ty! | j�� D ]}|j|u r |j Y S q� w )aa Return a file descriptor from a file object.
This wraps _fileobj_to_fd() to do an exhaustive
search in case the object is invalid but we still
have it in our map. Used by unregister() so we can
unregister an object that was previously registered
even if it is closed. It is also used by _SelectorMapping
)r1 r0 r �valuesr r
�r r �keyr r r r! | s
��z BaseSelector._fileobj_lookupNc C sb |r |t tB @ rtd�|���t|| �|�||�}|j| jv r)td�||j���|| j|j< |S )z8 Register a file object for a set of events to monitor. zInvalid events: {0!r}z${0!r} (FD {1}) is already registered) r r r0 r# r r! r
r r" �r r r r r5 r r r �register� s
� zBaseSelector.registerc
C s� z | j �| �|��}W |S ty td�|��� tjyQ } z+|jtjkr)� | j � � D ]}|j
|u r>| j �|j � nq.td�|���W Y d}~|S d}~ww )z0 Unregister a file object from being monitored. �{0!r} is not registeredN) r �popr! r" r# �socket�error�errno�EBADFr3 r r
)r r r5 �errr r r �
unregister� s$ �
�
���zBaseSelector.unregisterc C s~ z
| j | �|� }W n ty td�|���w ||jkr,| �|� | �|||�}|S ||jkr=|j|d�}|| j |j < |S )z< Change a registered file object monitored events and data. r8 )r )
r r! r" r# r r? r7 r �_replacer
r6 r r r �modify� s �
� zBaseSelector.modifyc C s t � �)zj Perform the actual selection until some monitored file objects
are ready or the timeout expires. )�NotImplementedError)r �timeoutr r r �select� s zBaseSelector.selectc C s | j �� d| _dS )zd Close the selector. This must be called to ensure that all
underlying resources are freed. N)r �clearr2 r r r r �close� s
zBaseSelector.closec C s@ | � � }|du r td��z|| W S ty td�|���w )z: Return the key associated with a registered file object. NzSelector is closedr8 )�get_map� RuntimeErrorr" r# )r r �mappingr r r �get_key� s
�zBaseSelector.get_keyc C s | j S )z3 Return a mapping of file objects to selector keys )r2 r r r r rG � s zBaseSelector.get_mapc C s" z| j | W S ty Y dS w )z_ Return the key associated to a given file descriptor
Return None if it is not found. N)r r" )r r
r r r � _key_from_fd� s
�zBaseSelector._key_from_fdc C s | S r r r r r r � __enter__� s zBaseSelector.__enter__c G s | � � d S r )rF )r �_r r r �__exit__� r zBaseSelector.__exit__r )r' r( r) r* r r! r7 r? rA rD rF rJ rG rK rL rN r r r r r
f s
r
rD c �N e Zd ZdZ� fdd�Zd
� fdd� Z� fdd�Zd
d d
�Zd
d d �Z� Z S )�SelectSelectorz Select-based selector. c s" t t| ��� t� | _t� | _d S r )�superrP r �set�_readers�_writersr �� __class__r r r � s zSelectSelector.__init__Nc sD t t| ��|||�}|t@ r| j�|j� |t@ r | j�|j� |S r ) rQ rP r7 r rS �addr
r rT r6 rU r r r7 � s zSelectSelector.registerc s0 t t| ��|�}| j�|j� | j�|j� |S r )rQ rP r? rS �discardr
rT r4 rU r r r? � s zSelectSelector.unregisterc C s� t | j�s t | j�s g S |d u rd nt|d�}g }t| jd| j| j|�\}}}t|�}t|�}||B D ]%}d}||v r?|tO }||v rG|tO }| � |�}|rX|�
|||j @ f� q3|S )N� Tr ) r rS rT �max�_syscall_wrapper� _wrap_selectrR r r rK �appendr ) r rC �ready�r�wrM r
r r5 r r r rD s(
�
�zSelectSelector.selectc C s t � ||g |�S �z? Wrapper for select.select because timeout is a positional arg )rD �r r_ r` rC r r r r\ � zSelectSelector._wrap_selectr )
r' r( r) r* r r7 r? rD r\ �
__classcell__r r rU r rP � s
rP c @ s( e Zd ZdZdd� Zdd� Zdd� ZdS ) �_JythonSelectorMappingz� This is an implementation of _SelectorMapping that is built
for use specifically with Jython, which does not provide a hashable
value from socket.socket.fileno(). c C s t |t�sJ �|| _d S r )r+ �JythonSelectSelectorr r r r r r ( s
z _JythonSelectorMapping.__init__c C r r )r r �_socketsr r r r r , r z _JythonSelectorMapping.__len__c C s0 | j jD ] \}}||u r| S qtd�|���r )r rg r" r# )r r �sockr5 r r r r$ / s
�z"_JythonSelectorMapping.__getitem__N)r' r( r) r* r r r$ r r r r re # s
re c s< e Zd ZdZ� fdd�Zd dd�Zdd� Zd d d
�Z� ZS ) rf az This is an implementation of SelectSelector that is for Jython
which works around that Jython's socket.socket.fileno() does not
return an integer fd value. All SelectorKey.fd will be equal to -1
and should not be used. This instead uses object id to compare fileobj
and will only use select.select as it's the only selector that allows
directly passing in socket objects rather than registering fds.
See: http://bugs.jython.org/issue1678
https://wiki.python.org/jython/NewSocketModule#socket.fileno.28.29_does_not_return_an_integer
c s> t t| ��� g | _t| �| _g | _g | _tt dt j �| _
d S )N�cpython_compatible_select) rQ rf r rg re r2 rS rT �getattrrD � _select_funcr rU r r r A s
z JythonSelectSelector.__init__Nc C sr | j D ]\}}||u rtd�||���qt|d||�}| j �||f� |t@ r-| j�|� |t@ r7| j�|� |S )Nz{0!r} is already registered���) rg r" r# r r] r rS r rT )r r r r rh rM r5 r r r r7 L s �� z JythonSelectSelector.registerc C sj t | j�D ] \}\}}||u r nqtd�|���|jt@ r$| j�|� |jt@ r/| j �|� | j|= |S r )
� enumeraterg r" r# r r rS �remover rT )r r �irh r5 r r r r? [ s �
z JythonSelectSelector.unregisterc C s | � ||g |�S ra )rk rb r r r r\ j rc z!JythonSelectSelector._wrap_selectr ) r' r( r) r* r r7 r? r\ rd r r rU r rf 6 s
rf �pollc rO )� PollSelectorz Poll-based selector c � t t| ��� t�� | _d S r )rQ rq r rD rp �_pollr rU r r r u � zPollSelector.__init__Nc �P t t| ��|||�}d}|t@ r|tjO }|t@ r |tjO }| j�|j |� |S �Nr )
rQ rq r7 r rD �POLLINr �POLLOUTrs r
)r r r r r5 �
event_maskrU r r r7 y �
zPollSelector.registerc �" t t| ��|�}| j�|j� |S r )rQ rq r? rs r
r4 rU r r r? � � zPollSelector.unregisterc C �4 |dur|dkr d}nt �|d �}| j�|�}|S �zj Wrapper function for select.poll.poll() so that
_syscall_wrapper can work with only seconds. Nr �� )�math�ceilrs rp �r rC �resultr r r �
_wrap_poll� � zPollSelector._wrap_pollc C �t g }t | jd|d�}|D ]+\}}d}|tj @ r |tO }|tj @ r&|tO }| �|�}|r7|�|||j @ f� q |S �NT)rC r �
r[ r� rD rw r rx r rK r] r �r rC r^ � fd_eventsr
ry r r5 r r r rD � �
�zPollSelector.selectr )
r' r( r) r* r r7 r? r� rD rd r r rU r rq s s
rq �epollc �X e Zd ZdZ� fdd�Zdd� Zd� fdd� Z� fd d
�Zdd d �Z� fd
d�Z � Z
S )�
EpollSelectorz Epoll-based selector c rr r )rQ r� r rD r� �_epollr rU r r r � rt zEpollSelector.__init__c C �
| j �� S r )r� r- r r r r r- � r zEpollSelector.filenoNc sT t t| ��|||�}d}|t@ r|tjO }|t@ r |tjO }t| j jd|j
|� |S )Nr F) rQ r� r7 r rD �EPOLLINr �EPOLLOUTr[ r� r
)r r r r r5 � events_maskrU r r r7 � s
zEpollSelector.registerc s> t t| ��|�}z t| jjd|j� W |S ty Y |S w )NF)rQ r� r? r[ r� r
� _ERROR_TYPESr4 rU r r r? � s ��zEpollSelector.unregisterc C s� |d ur|dkr d}n t �|d �d }t|�}nd}tt| j�d�}g }t| jjd||d�}|D ]+\}}d}|t j
@ rA|t O }|t j @ rK|t
O }| �|�}|r\|�|||j@ f� q1|S ) Nr rY r g����MbP?g �r T)rC � maxevents)r� r� �floatrZ r r r[ r� rp rD r� r r� r rK r] r ) r rC �
max_eventsr^ r� r
ry r r5 r r r rD � s.
�
�zEpollSelector.selectc � | j �� tt| ��� d S r )r� rF rQ r� r rU r r rF � �
zEpollSelector.closer � r' r( r) r* r r- r7 r? rD rF rd r r rU r r� � s
!r� �devpollc sb e Zd ZdZ� fdd�Zdd� Zd� fdd� Z� fd d
�Zdd d �Zdd
d�Z � fdd�Z
� Z S )�DevpollSelectorzSolaris /dev/poll selector.c rr r )rQ r� r rD r� �_devpollr rU r r r � rt zDevpollSelector.__init__c C r� r )r� r- r r r r r- � r zDevpollSelector.filenoNc ru rv )
rQ r� r7 r rD rw r rx r� r
)r r r r r5 � poll_eventsrU r r r7 � rz zDevpollSelector.registerc r{ r )rQ r� r? r� r
r4 rU r r r? r| zDevpollSelector.unregisterc C r} r~ )r� r� r� rp r� r r r r� r� zDevpollSelector._wrap_pollc C r� r� r� r� r r r rD r� zDevpollSelector.selectc r� r )r� rF rQ r� r rU r r rF % r� zDevpollSelector.closer ) r' r( r) r* r r- r7 r? r� rD rF rd r r rU r r� � s
r� �kqueuec r� )�KqueueSelectorz Kqueue / Kevent-based selector c rr r )rQ r� r rD r� �_kqueuer rU r r r / rt zKqueueSelector.__init__c C r� r )r� r- r r r r r- 3 r zKqueueSelector.filenoNc s| t t| ��|||�}|t@ r#t�|jtjtj�}t | j
j d|gdd� |t @ r<t�|jtj
tj�}t | j
j d|gdd� |S �NFr )rQ r� r7 r rD �keventr
�KQ_FILTER_READ� KQ_EV_ADDr[ r� �controlr �KQ_FILTER_WRITE)r r r r r5 r� rU r r r7 6 s ��zKqueueSelector.registerc s� t t| ��|�}|jt@ r.t�|jtjtj �}z
t
| j j d|gdd� W n t
y- Y nw |jt@ rVt�|jtjtj �}zt
| j j d|gdd� W |S t
yU Y |S w |S r� )rQ r� r? r r rD r� r
r� � KQ_EV_DELETEr[ r� r� r� r r� )r r r5 r� rU r r r? H s0
� �
� ��zKqueueSelector.unregisterc C s� |d ur t |d�}t| j�d }i }t| jjdd ||�}|D ]F}|j}|j}d}|tj kr1|t
O }|tj kr:|t O }| �
|�} | rd| j|vrQ| || j@ f|| j< q || j d }
| ||
B | j@ f|| j< q t|�� �S )Nr r Tr )rZ r r r[ r� r� �ident�filterrD r� r r� r rK r
r �listr3 ) r rC r� � ready_fds� kevent_listr� r
ry r r5 �
old_eventsr r r rD ] s.
�
� zKqueueSelector.selectc r� r )r� rF rQ r� r rU r r rF z r� zKqueueSelector.closer r� r r rU r r� - s
r� c C sP z| dkrt �� }|�d� W dS tt | �� �� W dS ttfy' Y dS w )a
Checks that select structs can be allocated by the underlying
operating system, not just advertised by the select module. We don't
check select() because we'll be hopeful that most platforms that
don't have it available will not advertise it. (ie: GAE) rp r TF)rD rp rj rF �OSErrorr. )�struct�pr r r �
_can_allocate� s
��r� )� � c O s | |i |��S )zy This is the short-circuit version of the below logic
because in Python 3.5+ all selectors restart system calls. r )�funcrM �args�kwargsr r r r[ � s r[ c
O s: |� dd�}|du rd}d}nt|�}|dk rd}nt� | }t|�}|r-d|vr-td��t}|tu r�z | |i |��}W nZ tttj fy� } zJd}t
|d�rS|j }n
t
|d�r]|j d }|t j
kpkt
t d �ok|t jk} | r�|dur�t� }
|
|krtt jd
��|r�d|v r�||
|d< W Y d}~q/� d}~ww |tu s3|S ) z� Wrapper function for syscalls that could fail due to EINTR.
All functions should be retried if there is time left in the timeout
in accordance with PEP 475. rC NFrY z4Timeout must be in args or kwargs to be recalculatedr< r� r �WSAEINTRzConnection timed out)�getr� � monotonicr� r0 �_SYSCALL_SENTINELr� �IOErrorrD r; �hasattrr< r� �EINTRr� � ETIMEDOUT) r� �recalc_timeoutr� r� rC �expiresr� �e�errcode� is_interrupt� current_timer r r r[ � sP
�
� ��� c C s� t du r>tjr ta t � S td�rta t � S td�r ta t � S td�r'ta t � S td�r0ta t � S t t
d�r:t a t � S t d��t � S )z� This function serves as a first call for DefaultSelector to
detect if the select module is being monkey-patched incorrectly
by eventlet, greenlet, and preserve proper behavior. Nr� r� r� rp rD z"Platform does not have a selector.)
�_DEFAULT_SELECTORr �isjythonrf r� r� r� r� rq r� rD rP rH r r r r r � s*
� � ��
��r )1r* �
__future__r � collectionsr< r� rD r: �sys�time� r �
namedtuple�Mappingr� r. �
__author__� __email__� __version__� __license__�__url__�__all__r r r� �objectr� r� r� r; r� r r r1 r
r� rP r] r� re rf rq r� r� r� r� � version_infor[ r r r r r �<module> st
�
0
8
3
A
;
Q
8