î
yË X5J  ã               @   s  d  Z  d d l m Z m Z d d l m Z m Z d d l Z d d l Z d d l	 Z	 d d l
 m Z m Z d d l m Z d# Z d$ Z d d	 „  Z e d
 d d d d g ƒ Z Gd d „  d e ƒ Z Gd d „  d e j e ƒ ƒ Z Gd d „  d e ƒ Z Gd d „  d e ƒ Z e e d ƒ r8Gd d „  d e ƒ Z n  e e d ƒ r`Gd d „  d e ƒ Z n  e e d ƒ rˆGd d „  d e ƒ Z n  e e d  ƒ r°Gd! d" „  d" e ƒ Z n  d" e ƒ  k rÈe Z nN d e ƒ  k ràe Z n6 d e ƒ  k røe Z n d e ƒ  k re Z n e Z d S)%z±Selectors module.

This module allows high-level and efficient I/O multiplexing, built upon the
`select` module primitives.

The following code adapted from trollius.selectors.
é    )ÚABCMetaÚabstractmethod)Ú
namedtupleÚMappingN)Ú
wrap_errorÚInterruptedError)Úsixé   c             C   sŒ   t  |  t j ƒ r |  } nI y t |  j ƒ  ƒ } Wn0 t t t f k
 rc t d j |  ƒ ƒ ‚ Yn X| d k  rˆ t d j | ƒ ƒ ‚ n  | S)zÕReturn a file descriptor from a file object.

    Parameters:
    fileobj -- file object or file descriptor

    Returns:
    corresponding file descriptor

    Raises:
    ValueError if the object is invalid
    zInvalid file object: {0!r}r   zInvalid file descriptor: {0})	Ú
isinstancer   Úinteger_typesÚintÚfilenoÚAttributeErrorÚ	TypeErrorÚ
ValueErrorÚformat)ÚfileobjÚfd© r   úB/home/ubuntu/projects/ifolica/build/gunicorn/gunicorn/selectors.pyÚ_fileobj_to_fd   s    		r   ÚSelectorKeyr   r   ÚeventsÚdatac               @   sF   e  Z d  Z d Z d d „  Z d d „  Z d d „  Z d d	 „  Z d
 S)Ú_SelectorMappingz)Mapping of file objects to selector keys.c             C   s   | |  _  d  S)N)Ú	_selector)ÚselfÚselectorr   r   r   Ú__init__:   s    z_SelectorMapping.__init__c             C   s   t  |  j j ƒ S)N)Úlenr   Ú
_fd_to_key)r   r   r   r   Ú__len__=   s    z_SelectorMapping.__len__c             C   sR   y$ |  j  j | ƒ } |  j  j | SWn' t k
 rM t d j | ƒ ƒ ‚ Yn Xd  S)Nz{0!r} is not registered)r   Ú_fileobj_lookupr    ÚKeyErrorr   )r   r   r   r   r   r   Ú__getitem__@   s
    z_SelectorMapping.__getitem__c             C   s   t  |  j j ƒ S)N)Úiterr   r    )r   r   r   r   Ú__iter__G   s    z_SelectorMapping.__iter__N)Ú__name__Ú
__module__Ú__qualname__Ú__doc__r   r!   r$   r&   r   r   r   r   r   7   s
   r   c               @   s£   e  Z d  Z d Z e d d d „ ƒ Z e d d „  ƒ Z d d d „ Z e d d	 d
 „ ƒ Z d d „  Z	 d d „  Z
 e d d „  ƒ Z d d „  Z d d „  Z d S)ÚBaseSelectora-  Selector abstract base 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 information, a callback, etc.

    A selector can use various implementations (select(), poll(), epoll()...)
    depending on the platform. The default `Selector` class uses the most
    efficient implementation on the current platform.
    Nc             C   s
   t  ‚ d S)a3  Register a file object.

        Parameters:
        fileobj -- file object or file descriptor
        events  -- events to monitor (bitwise mask of EVENT_READ|EVENT_WRITE)
        data    -- attached data

        Returns:
        SelectorKey instance

        Raises:
        ValueError if events is invalid
        KeyError if fileobj is already registered
        OSError if fileobj is closed or otherwise is unacceptable to
                the underlying system call (if a system call is made)

        Note:
        OSError may or may not be raised
        N)ÚNotImplementedError)r   r   r   r   r   r   r   ÚregisterZ   s    zBaseSelector.registerc             C   s
   t  ‚ d S)aj  Unregister a file object.

        Parameters:
        fileobj -- file object or file descriptor

        Returns:
        SelectorKey instance

        Raises:
        KeyError if fileobj is not registered

        Note:
        If fileobj is registered but has since been closed this does
        *not* raise OSError (even if the wrapped syscall does)
        N)r,   )r   r   r   r   r   Ú
unregisterq   s    zBaseSelector.unregisterc             C   s    |  j  | ƒ |  j | | | ƒ S)ay  Change a registered file object monitored events or attached data.

        Parameters:
        fileobj -- file object or file descriptor
        events  -- events to monitor (bitwise mask of EVENT_READ|EVENT_WRITE)
        data    -- attached data

        Returns:
        SelectorKey instance

        Raises:
        Anything that unregister() or register() raises
        )r.   r-   )r   r   r   r   r   r   r   Úmodify„   s    zBaseSelector.modifyc             C   s
   t  ‚ d S)aq  Perform the actual selection, until some monitored file objects are
        ready or a timeout expires.

        Parameters:
        timeout -- if timeout > 0, this specifies the maximum wait time, in
                   seconds
                   if timeout <= 0, the select() call won't block, and will
                   report the currently ready file objects
                   if timeout is None, select() will block until a monitored
                   file object becomes ready

        Returns:
        list of (key, events) for ready file objects
        `events` is a bitwise mask of EVENT_READ|EVENT_WRITE
        N)r,   )r   Útimeoutr   r   r   Úselect•   s    zBaseSelector.selectc             C   s   d S)zmClose the selector.

        This must be called to make sure that any underlying resource is freed.
        Nr   )r   r   r   r   Úclose¨   s    zBaseSelector.closec             C   sF   |  j  ƒ  } y | | SWn' t k
 rA t d j | ƒ ƒ ‚ Yn Xd S)zzReturn the key associated to a registered file object.

        Returns:
        SelectorKey for this file object
        z{0!r} is not registeredN)Úget_mapr#   r   )r   r   Úmappingr   r   r   Úget_key¯   s
    zBaseSelector.get_keyc             C   s
   t  ‚ d S)z2Return a mapping of file objects to selector keys.N)r,   )r   r   r   r   r3   »   s    zBaseSelector.get_mapc             C   s   |  S)Nr   )r   r   r   r   Ú	__enter__À   s    zBaseSelector.__enter__c             G   s   |  j  ƒ  d  S)N)r2   )r   Úargsr   r   r   Ú__exit__Ã   s    zBaseSelector.__exit__)r'   r(   r)   r*   r   r-   r.   r/   r1   r2   r5   r3   r6   r8   r   r   r   r   r+   K   s   r+   c               @   s|   e  Z d  Z d Z d d „  Z d d „  Z d d d „ Z d	 d
 „  Z d d d „ Z d d „  Z	 d d „  Z
 d d „  Z d S)Ú_BaseSelectorImplzBase selector implementation.c             C   s   i  |  _  t |  ƒ |  _ d  S)N)r    r   Ú_map)r   r   r   r   r   Ê   s    	z_BaseSelectorImpl.__init__c             C   sZ   y t  | ƒ SWnE t k
 rU x- |  j j ƒ  D] } | j | k r. | j Sq. W‚  Yn Xd S)al  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.  This
        is used by unregister() so we can unregister an object that
        was previously registered even if it is closed.  It is also
        used by _SelectorMapping.
        N)r   r   r    Úvaluesr   r   )r   r   Úkeyr   r   r   r"   Ð   s    	z!_BaseSelectorImpl._fileobj_lookupNc             C   s   | s | t  t B@r. t d j | ƒ ƒ ‚ n  t | |  j | ƒ | | ƒ } | j |  j k r| t d j | | j ƒ ƒ ‚ n  | |  j | j <| S)NzInvalid events: {0!r}z${0!r} (FD {1}) is already registered)	Ú
EVENT_READÚEVENT_WRITEr   r   r   r"   r   r    r#   )r   r   r   r   r<   r   r   r   r-   ã   s    	z_BaseSelectorImpl.registerc             C   sM   y |  j  j |  j | ƒ ƒ } Wn' t k
 rH t d j | ƒ ƒ ‚ Yn X| S)Nz{0!r} is not registered)r    Úpopr"   r#   r   )r   r   r<   r   r   r   r.   ð   s
    z_BaseSelectorImpl.unregisterc             C   s°   y |  j  |  j | ƒ } Wn' t k
 rC t d j | ƒ ƒ ‚ Yn X| | j k rx |  j | ƒ |  j | | | ƒ } n4 | | j k r¬ | j d | ƒ } | |  j  | j	 <n  | S)Nz{0!r} is not registeredr   )
r    r"   r#   r   r   r.   r-   r   Ú_replacer   )r   r   r   r   r<   r   r   r   r/   ÷   s    z_BaseSelectorImpl.modifyc             C   s   |  j  j ƒ  d  S)N)r    Úclear)r   r   r   r   r2     s    z_BaseSelectorImpl.closec             C   s   |  j  S)N)r:   )r   r   r   r   r3   	  s    z_BaseSelectorImpl.get_mapc             C   s,   y |  j  | SWn t k
 r' d SYn Xd S)z³Return the key associated to a given file descriptor.

        Parameters:
        fd -- file descriptor

        Returns:
        corresponding key, or None if not found
        N)r    r#   )r   r   r   r   r   Ú_key_from_fd  s    	z_BaseSelectorImpl._key_from_fd)r'   r(   r)   r*   r   r"   r-   r.   r/   r2   r3   rB   r   r   r   r   r9   Ç   s   r9   c                   sˆ   e  Z d  Z d Z ‡  f d d †  Z d ‡  f d d † Z ‡  f d d †  Z e j d	 k rl d d
 d „ Z	 n	 e
 j
 Z	 d d d „ Z
 ‡  S)ÚSelectSelectorzSelect-based selector.c                s/   t  t |  ƒ j ƒ  t ƒ  |  _ t ƒ  |  _ d  S)N)ÚsuperrC   r   ÚsetÚ_readersÚ_writers)r   )Ú	__class__r   r   r     s    zSelectSelector.__init__Nc                sb   t  t |  ƒ j | | | ƒ } | t @r> |  j j | j ƒ n  | t @r^ |  j j | j ƒ n  | S)N)	rD   rC   r-   r=   rF   Úaddr   r>   rG   )r   r   r   r   r<   )rH   r   r   r-   #  s    

zSelectSelector.registerc                sB   t  t |  ƒ j | ƒ } |  j j | j ƒ |  j j | j ƒ | S)N)rD   rC   r.   rF   Údiscardr   rG   )r   r   r<   )rH   r   r   r.   +  s    zSelectSelector.unregisterÚwin32c             C   s2   t  j  | | | | ƒ \ } } } | | | g  f S)N)r1   )r   ÚrÚwÚ_r0   Úxr   r   r   Ú_select2  s    !zSelectSelector._selectc       	      C   s	  | d  k r d  n t  | d ƒ } g  } y. t |  j |  j |  j g  | ƒ \ } } } Wn t k
 rm | SYn Xt | ƒ } t | ƒ } x| | | BD]p } d } | | k r¶ | t O} n  | | k rÏ | t O} n  |  j	 | ƒ } | r‘ | j
 | | | j @f ƒ q‘ q‘ W| S)Nr   )Úmaxr   rP   rF   rG   r   rE   r=   r>   rB   Úappendr   )	r   r0   ÚreadyrL   rM   rN   r   r   r<   r   r   r   r1   8  s&    !	%	!zSelectSelector.select)r'   r(   r)   r*   r   r-   r.   ÚsysÚplatformrP   r1   r   r   )rH   r   rC     s   	rC   Úpollc                   s^   e  Z d  Z d Z ‡  f d d †  Z d ‡  f d d † Z ‡  f d d †  Z d d	 d
 „ Z ‡  S)ÚPollSelectorzPoll-based selector.c                s&   t  t |  ƒ j ƒ  t j ƒ  |  _ d  S)N)rD   rW   r   r1   rV   Ú_poll)r   )rH   r   r   r   T  s    zPollSelector.__init__Nc                sr   t  t |  ƒ j | | | ƒ } d } | t @r> | t j O} n  | t @rX | t j O} n  |  j j | j	 | ƒ | S)Nr   )
rD   rW   r-   r=   r1   ÚPOLLINr>   ÚPOLLOUTrX   r   )r   r   r   r   r<   Úpoll_events)rH   r   r   r-   X  s    

zPollSelector.registerc                s/   t  t |  ƒ j | ƒ } |  j j | j ƒ | S)N)rD   rW   r.   rX   r   )r   r   r<   )rH   r   r   r.   b  s    zPollSelector.unregisterc             C   s  | d  k r d  } n. | d k r* d } n t  t j | d ƒ ƒ } g  } y t |  j j | ƒ } Wn t k
 rz | SYn Xx‚ | D]z \ } } d } | t j @r¯ | t	 O} n  | t j
 @rÊ | t O} n  |  j | ƒ } | r‚ | j | | | j @f ƒ q‚ q‚ W| S)Nr   g     @@)r   ÚmathÚceilr   rX   rV   r   r1   rY   r>   rZ   r=   rB   rR   r   )r   r0   rS   Úfd_event_listr   Úeventr   r<   r   r   r   r1   g  s(    			!zPollSelector.select)r'   r(   r)   r*   r   r-   r.   r1   r   r   )rH   r   rW   Q  s
   
rW   Úepollc                   s|   e  Z d  Z d Z ‡  f d d †  Z d d „  Z d ‡  f d d † Z ‡  f d	 d
 †  Z d d d „ Z ‡  f d d †  Z	 ‡  S)ÚEpollSelectorzEpoll-based selector.c                s&   t  t |  ƒ j ƒ  t j ƒ  |  _ d  S)N)rD   ra   r   r1   r`   Ú_epoll)r   )rH   r   r   r   ‡  s    zEpollSelector.__init__c             C   s   |  j  j ƒ  S)N)rb   r   )r   r   r   r   r   ‹  s    zEpollSelector.filenoNc                sr   t  t |  ƒ j | | | ƒ } d } | t @r> | t j O} n  | t @rX | t j O} n  |  j j | j	 | ƒ | S)Nr   )
rD   ra   r-   r=   r1   ÚEPOLLINr>   ÚEPOLLOUTrb   r   )r   r   r   r   r<   Úepoll_events)rH   r   r   r-   Ž  s    

zEpollSelector.registerc                sH   t  t |  ƒ j | ƒ } y |  j j | j ƒ Wn t k
 rC Yn X| S)N)rD   ra   r.   rb   r   ÚOSError)r   r   r<   )rH   r   r   r.   ˜  s    zEpollSelector.unregisterc       	      C   s  | d  k r d } n, | d k r* d } n t  j | d ƒ d } t |  j ƒ } g  } y t |  j j | | ƒ } Wn t k
 rŠ | SYn Xx‚ | D]z \ } } d } | t j	 @r¿ | t
 O} n  | t j @rÚ | t O} n  |  j | ƒ } | r’ | j | | | j @f ƒ q’ q’ W| S)Nr	   r   g     @@gü©ñÒMbP?éÿÿÿÿ)r\   r]   r   r    r   rb   rV   r   r1   rc   r>   rd   r=   rB   rR   r   )	r   r0   Úmax_evrS   r^   r   r_   r   r<   r   r   r   r1   ¢  s*    			!zEpollSelector.selectc                s$   |  j  j ƒ  t t |  ƒ j ƒ  d  S)N)rb   r2   rD   ra   )r   )rH   r   r   r2   ½  s    zEpollSelector.close)
r'   r(   r)   r*   r   r   r-   r.   r1   r2   r   r   )rH   r   ra   „  s   

ra   Údevpollc                   s|   e  Z d  Z d Z ‡  f d d †  Z d d „  Z d ‡  f d d † Z ‡  f d	 d
 †  Z d d d „ Z ‡  f d d †  Z	 ‡  S)ÚDevpollSelectorzSolaris /dev/poll selector.c                s&   t  t |  ƒ j ƒ  t j ƒ  |  _ d  S)N)rD   rj   r   r1   ri   Ú_devpoll)r   )rH   r   r   r   Ç  s    zDevpollSelector.__init__c             C   s   |  j  j ƒ  S)N)rk   r   )r   r   r   r   r   Ë  s    zDevpollSelector.filenoNc                sr   t  t |  ƒ j | | | ƒ } d } | t @r> | t j O} n  | t @rX | t j O} n  |  j j | j	 | ƒ | S)Nr   )
rD   rj   r-   r=   r1   rY   r>   rZ   rk   r   )r   r   r   r   r<   r[   )rH   r   r   r-   Î  s    

zDevpollSelector.registerc                s/   t  t |  ƒ j | ƒ } |  j j | j ƒ | S)N)rD   rj   r.   rk   r   )r   r   r<   )rH   r   r   r.   Ø  s    zDevpollSelector.unregisterc             C   sû   | d  k r d  } n( | d k r* d } n t  j | d ƒ } g  } y |  j j | ƒ } Wn t k
 rq | SYn Xx‚ | D]z \ } } d } | t j @r¦ | t O} n  | t j @rÁ | t	 O} n  |  j
 | ƒ } | ry | j | | | j @f ƒ qy qy W| S)Nr   g     @@)r\   r]   rk   rV   r   r1   rY   r>   rZ   r=   rB   rR   r   )r   r0   rS   r^   r   r_   r   r<   r   r   r   r1   Ý  s(    			!zDevpollSelector.selectc                s$   |  j  j ƒ  t t |  ƒ j ƒ  d  S)N)rk   r2   rD   rj   )r   )rH   r   r   r2   ÷  s    zDevpollSelector.close)
r'   r(   r)   r*   r   r   r-   r.   r1   r2   r   r   )rH   r   rj   Ä  s   
rj   Úkqueuec                   s|   e  Z d  Z d Z ‡  f d d †  Z d d „  Z d ‡  f d d † Z ‡  f d	 d
 †  Z d d d „ Z ‡  f d d †  Z	 ‡  S)ÚKqueueSelectorzKqueue-based selector.c                s&   t  t |  ƒ j ƒ  t j ƒ  |  _ d  S)N)rD   rm   r   r1   rl   Ú_kqueue)r   )rH   r   r   r     s    zKqueueSelector.__init__c             C   s   |  j  j ƒ  S)N)rn   r   )r   r   r   r   r     s    zKqueueSelector.filenoNc                sª   t  t |  ƒ j | | | ƒ } | t @rb t j | j t j t j ƒ } |  j	 j
 | g d d ƒ n  | t @r¦ t j | j t j t j ƒ } |  j	 j
 | g d d ƒ n  | S)Nr   )rD   rm   r-   r=   r1   Úkeventr   ÚKQ_FILTER_READZ	KQ_EV_ADDrn   Úcontrolr>   ÚKQ_FILTER_WRITE)r   r   r   r   r<   Úkev)rH   r   r   r-     s    

zKqueueSelector.registerc                sÜ   t  t |  ƒ j | ƒ } | j t @rx t j | j t j t j	 ƒ } y |  j
 j | g d d ƒ Wqx t k
 rt Yqx Xn  | j t @rØ t j | j t j t j	 ƒ } y |  j
 j | g d d ƒ WqØ t k
 rÔ YqØ Xn  | S)Nr   )rD   rm   r.   r   r=   r1   ro   r   rp   ZKQ_EV_DELETErn   rq   rf   r>   rr   )r   r   r<   rs   )rH   r   r   r.     s     zKqueueSelector.unregisterc       
      C   s  | d  k r d  n t  | d ƒ } t |  j ƒ } g  } y t |  j j d  | | ƒ } Wn t k
 rm | SYn Xx | D]ˆ } | j } | j } d } | t	 j
 k r¯ | t O} n  | t	 j k rË | t O} n  |  j | ƒ }	 |	 ru | j |	 | |	 j @f ƒ qu qu W| S)Nr   )rQ   r   r    r   rn   rq   r   ÚidentÚfilterr1   rp   r=   rr   r>   rB   rR   r   )
r   r0   rh   rS   Zkev_listrs   r   Úflagr   r<   r   r   r   r1   )  s(    !			!zKqueueSelector.selectc                s$   |  j  j ƒ  t t |  ƒ j ƒ  d  S)N)rn   r2   rD   rm   )r   )rH   r   r   r2   @  s    zKqueueSelector.close)
r'   r(   r)   r*   r   r   r-   r.   r1   r2   r   r   )rH   r   rm   þ  s   rm   r	   é   )r*   Úabcr   r   Úcollectionsr   r   r\   r1   rT   Zgunicorn._compatr   r   Zgunicornr   r=   r>   r   r   r   Úwith_metaclassr+   r9   rC   ÚhasattrrW   ra   rj   rm   ÚglobalsÚDefaultSelectorr   r   r   r   Ú<module>   s@   |T41>8I				