3
\3k �
@ sl d dl Z d dl T d dlZd dlZd dlZd dlZd dlmZmZ y d dlZW n e k
rd dZY nX e
edd�Z e
edd�Z e
edd�Z
d d
d d d
gZejeje �� ejd edd� � ejd
edd� � ejdedd� � ejdedd� � dZdZdd� ZeZejj� jd��r�i Zded< d ed < d ed < d ed!< d"ed#< d$ed%< d&ed'< d(ed)< d*ed+< d,ed-< d.ed/< d0ed1< d2ed3< d4ed5< d6ed7< ej d8� G d9d:� d:e �Z G d;d<� d<e j�ZdLd=d �Z e e jd>��r�d?d@� Z!ej d@� e e dA��rde"d fdBdA�Z#ne$e"d fdCdA�Z#ej dA� dDe#_%e e
hZ&G dEdF� dFej'�Z(dMdHd
�Z)e*� Z+e+dfdId �Z,dNdJdK�Z-dS )O� N)�*)�IntEnum�IntFlag�EBADF� �EAGAIN� � EWOULDBLOCK�fromfd�getfqdn�create_connection�
AddressFamily�
SocketKindc C s | j � o| jd�S )NZAF_)�isupper�
startswith)�C� r � /usr/lib64/python3.6/socket.py�<lambda>L s r c C s | j � o| jd�S )NZSOCK_)r r )r r r r r Q s ZMsgFlagc C s | j � o| jd�S )NZMSG_)r r )r r r r r V s Z AddressInfoc C s | j � o| jd�S )NZAI_)r r )r r r r r [ s z 127.0.0.1z::1c C s" y|| �S t k
r | S X d S )N)�
ValueError)�valueZ
enum_klassr r r �_intenum_convertera s r �winz The operation was interrupted.i' z A bad file handle was passed.i' zPermission denied.i ' z!A fault occurred on the network??i ' z#An invalid operation was attempted.i&' z The socket operation would blocki3' z,A blocking operation is already in progress.i4' z The network address is in use.i@' z The connection has been reset.iF' z The network has been shut down.iJ' zThe operation timed out.iL' zConnection refused.iM' zThe name is too long.iO' zThe host is down.iP' zThe host is unreachable.iQ' �errorTabc @ s e Zd ZdS )�_GiveupOnSendfileN)�__name__�
__module__� __qualname__r r r r r � s r c s0 e Zd ZdddgZeeddfdd�Zdd � Zd
d � Zd d
� Z dd� Z
dd� Z dd� Z d7dddd�dd�Z
eed�r~d8dd�Zn
d9dd�Zd:d d �Zd d � Zd;d d!�Zd"d#� Zejfd$d%�Zd&d'� Z� fd(d)�Ze� fd*d+��Ze� fd,d-��Z ej d.k�r d/d0� Z d1d2� Z nd3d0� Z d4d2� Z d5e _ d6e _ � Z!S )<�socket� __weakref__�_io_refs�_closedr Nc C s$ t jj| ||||� d| _d| _d S )Nr F)�_socketr �__init__r r! )�self�family�type�proto�filenor r r r# � s zsocket.__init__c C s | S )Nr )r$ r r r � __enter__� s zsocket.__enter__c G s | j s| j� d S )N)r! �close)r$ �argsr r r �__exit__� s zsocket.__exit__c C s� t | dd�}d| jj| jj|r"dnd| j� | j| j| jf }|s�y | j� }|r^|dt |� 7 }W n t
k
rt Y nX y | j � }|r�|dt |� 7 }W n t
k
r� Y nX |d7 }|S ) Nr! Fz,<%s.%s%s fd=%i, family=%s, type=%s, proto=%iz [closed]� z
, laddr=%sz
, raddr=%s�>) �getattr� __class__r r r( r% r& r' � getsockname�str�errorZ getpeername)r$ �closed�sZladdrZraddrr r r �__repr__� s0
zsocket.__repr__c C s t d��d S )Nz Cannot serialize socket object)� TypeError)r$ r r r � __getstate__� s zsocket.__getstate__c C s6 t | j� �}| j| j| j| j|d�}|j| j� � |S )N)r( )�dupr( r0 r% r& r' �
settimeout�
gettimeout)r$ �fd�sockr r r r9 � s z
socket.dupc C sZ | j � \}}| jt� jdd� @ }t| j|| j|d�}t� d krR| j� rR|j d� ||fS )NZ
SOCK_NONBLOCKr )r( T)
Z_acceptr& �globals�getr r% r' Zgetdefaulttimeoutr; � setblocking)r$ r< �addrr& r= r r r �accept� s
z
socket.accept�r)�encoding�errors�newlinec
C s� t |�dddhks td|f ��d|k}d|kp4| }d|k}d} |rN| d7 } |rZ| d7 } t| | �}
| jd7 _|d kr~d }|dk r�tj}|dkr�|s�td��|
S |r�|r�tj|
|
|�} n |r�tj|
|�} n tj|
|�} |r�| S tj | |||�} || _
| S )
NrC �w�bz&invalid mode %r (only r, w, b allowed)r- � r z!unbuffered streams must be binary���) �setr �SocketIOr �io�DEFAULT_BUFFER_SIZE�BufferedRWPair�BufferedReader�BufferedWriter�
TextIOWrapper�mode)
r$ rS � bufferingrD rE rF ZwritingZreadingZbinaryZrawmode�raw�buffer�textr r r �makefile� s<
zsocket.makefile�sendfilec 2 C s� | j |||� | j� }y |j� }W n0 ttjfk
rR } zt|��W Y d d }~X nX ytj|�j}W n* t k
r� } zt|��W Y d d }~X nX |s�dS |s�|n|}| j
� } | dkr�t d��t t
d�r�t
j� }
nt
j� }
|
j|t
j� d} |
j} tj}
z�x�| �r| | � �rtjd��|�r0|| }|dk�r0P y|
||||�}W n^ tk
�rf | �s`| � w�Y q� t k
�r� } z | dk�r�t|��|d �W Y d d }~X q�X |dk�r�P ||7 }| |7 } q�W | S | dk�r�t |d��r�|j|� X d S )Nr z&non-blocking sockets are not supported� PollSelectorz timed out�seek)�_check_sendfile_paramsr( �AttributeErrorrM �UnsupportedOperationr �os�fstat�st_size�OSErrorr; r �hasattr� selectorsrZ ZSelectSelector�registerZ EVENT_WRITEZselectrY r" �timeout�BlockingIOErrorr[ )r$ �file�offset�countZsocknor( �errZfsize� blocksizerf Zselector�
total_sentZselector_selectZ os_sendfile�sentr r r �_sendfile_use_sendfile s`
z socket._sendfile_use_sendfilec C s t d��d S )Nz,os.sendfile() not available on this platform)r )r$ rh ri rj r r r ro G s c
C s | j |||� | j� dkr"td��|r0|j|� |r>t|d�nd}d}|j}| j}z�x�|rrt|| |�}|dkrrP t||��}|s�P xLy ||�} W n tk
r� w�Y q�X || 7 }| t |�k r�|| d � }q�P q�W qVW |S |dko�t
|d�� r�|j|| � X d S )Nr z&non-blocking sockets are not supportedi r[ ) r\ r; r r[ �min�read�send�
memoryviewrg �lenrc )
r$ rh ri rj rl rm Z file_readZ sock_send�datarn r r r �_sendfile_use_sendK s<
zsocket._sendfile_use_sendc C sd dt |dd�krtd��| jt@ s*td��|d k r`t|t�sJtdj|���|dkr`tdj|���d S )NrH rS z$file should be opened in binary modez+only SOCK_STREAM type sockets are supportedz+count must be a positive integer (got {!r})r )r/ r r& � SOCK_STREAM�
isinstance�intr7 �format)r$ rh ri rj r r r r\ o s
z socket._check_sendfile_paramsc C s2 y| j |||�S tk
r, | j|||�S X d S )N)ro r rv )r$ rh ri rj r r r rY | s zsocket.sendfilec C s* | j dkr| j d8 _ | jr&| j� d S )Nr rI )r r! r* )r$ r r r �_decref_socketios� s
zsocket._decref_socketiosc C s |j | � d S )N)r* )r$ Z_ssr r r � _real_close� s zsocket._real_closec C s d| _ | jdkr| j� d S )NTr )r! r r| )r$ r r r r* � s
z socket.closec s d| _ t� j� S )NT)r! �super�detach)r$ )r0 r r r~ � s z
socket.detachc s t t� jt�S )N)r r} r% r
)r$ )r0 r r r% � s z
socket.familyc s t t� jt�S )N)r r} r& r )r$ )r0 r r r& � s z socket.type�ntc C s t j| j� �S )N)r_ Zget_handle_inheritabler( )r$ r r r �get_inheritable� s zsocket.get_inheritablec C s t j| j� |� d S )N)r_ Zset_handle_inheritabler( )r$ � inheritabler r r �set_inheritable� s zsocket.set_inheritablec C s t j| j� �S )N)r_ r� r( )r$ r r r r� � s c C s t j| j� |� d S )N)r_ r� r( )r$ r� r r r r� � s z&Get the inheritable flag of the socketz&Set the inheritable flag of the socket)rC N)r N)r N)r N)r N)"r r r � __slots__�AF_INETrw r# r) r, r6 r8 r9 rB rX rc r_ ro rv r\ rY r{ r" r r| r* r~ �propertyr% r& �namer� r� �__doc__�
__classcell__r r )r0 r r � s:
)
A
$
r c C s t | �}t||||�S )N)r9 r )r< r% r& r' Znfdr r r r
� s Zsharec C s t ddd| �S )Nr )r )�infor r r � fromshare� s r� �
socketpairc
C sh | d kr*yt } W n tk
r( t} Y nX tj| ||�\}}t| |||j� �}t| |||j� �}||fS )N)ZAF_UNIX� NameErrorr� r" r� r r~ )r% r& r' �arH r r r r� � s
c
C s | t krt}n| tkr t}ntd��|tkr4td��|dkrDtd��t| ||�}z�|j|df� |j� |j � d d� \}}t| ||�}yP|j
d� y|j ||f� W n t t
fk
r� Y nX |j
d� |j� \}} W n |j� � Y nX W d |j� X ||fS )Nz?Only AF_INET and AF_INET6 socket address families are supportedz)Only SOCK_STREAM socket type is supportedr z Only protocol zero is supported� FT)r� �
_LOCALHOSTZAF_INET6�
_LOCALHOST_V6r rw r �bindZlistenr1 r@ �connectrg �InterruptedErrorrB r* )
r% r& r' �hostZlsockrA �portZcsockZssock�_r r r r� � s8
a8 socketpair([family[, type[, proto]]]) -> (socket object, socket object)
Create a pair of socket objects from the sockets returned by the platform
socketpair() function.
The arguments are the same as for socket() except the default family is AF_UNIX
if defined on the platform; otherwise, the default is AF_INET.
c sl e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Z� fd d �Zd
d� Z e
dd� �Z e
dd� �Z dd� Z
� ZS )rL c C sZ |d
krt d| ��tjj| � || _d|kr6|d7 }|| _d|k| _d|k| _d | _d S ) NrC rG �rw�rb�wb�rwbzinvalid mode: %rrH F)rC rG r� r� r� r� ) r rM � RawIOBaser# �_sock�_mode�_reading�_writing�_timeout_occurred)r$ r= rS r r r r# 0 s
zSocketIO.__init__c C s� | j � | j� | jr td��xby | jj|�S tk
rH d| _� Y q tk
r| } z|jd t krjd S � W Y d d }~X q X q W d S )Nz!cannot read from timed out objectTr )
� _checkClosed�_checkReadabler� rb r� Z recv_intorf r3 r+ �_blocking_errnos)r$ rH �er r r �readinto< s zSocketIO.readintoc C sX | j � | j� y | jj|�S tk
rR } z|jd tkr@d S � W Y d d }~X nX d S )Nr )r� �_checkWritabler� rr r3 r+ r� )r$ rH r� r r r �writeS s zSocketIO.writec C s | j rtd��| jS )Nz I/O operation on closed socket.)r4 r r� )r$ r r r �readablec s zSocketIO.readablec C s | j rtd��| jS )Nz I/O operation on closed socket.)r4 r r� )r$ r r r �writablej s zSocketIO.writablec s | j rtd��t� j� S )Nz I/O operation on closed socket.)r4 r r} �seekable)r$ )r0 r r r� q s zSocketIO.seekablec C s | j � | jj� S )N)r� r� r( )r$ r r r r( x s zSocketIO.filenoc C s | j s| j� S dS d S )NrI rJ )r4 r( )r$ r r r r� ~ s z
SocketIO.namec C s | j S )N)r� )r$ r r r rS � s z
SocketIO.modec C s* | j r
d S tjj| � | jj� d | _d S )N)r4 rM r� r* r� r{ )r$ r r r r* � s
zSocketIO.close)r r r r# r� r� r� r� r� r( r� r� rS r* r� r r )r0 r rL s rL r- c C sp | j � } | s| dkr t� } yt| �\}}}W n tk
rB Y n*X |jd|� x|D ]} d| krVP qVW |} | S )Nz0.0.0.0r �.)�stripZ gethostnameZ
gethostbyaddrr3 �insert)r� Zhostname�aliasesZipaddrsr r r r � s
c C s� | \}}d }x�t ||dt�D ]�}|\}}} }
} d } y>t||| �} |tk rR| j|� |r`| j|� | j| � d }| S tk
r� }
z|
}| d k r�| j� W Y d d }
~
X q X q W |d k r�|�ntd��d S )Nr z!getaddrinfo returns an empty list) � getaddrinforw r �_GLOBAL_DEFAULT_TIMEOUTr: r� r� r3 r* )Zaddressrf Zsource_addressr� r� rk �res�af�socktyper' � canonname�sar= r� r r r r � s*
c C sT g }xJt j| |||||�D ]2}|\}} }}
} |jt|t�t| t�||
| f� qW |S )N)r" r� �appendr r
r ) r� r� r% r& r' �flagsZaddrlistr� r� r� r� r� r r r r� � s r� )r )r- )r r r r ).r" r_ �sysrM rd �enumr r �errno� ImportErrorr/ r r r �__all__�extend�_get_exports_list�_convertr r� r� r r Z _realsocket�platform�lowerr r r� � Exceptionr r
rc r� rw r� r� r� r� r� rL r �objectr� r r� r r r r �<module>1 s�
C
$
u
'