3
\`Z � @ sF d dl Z d dlZd dlZd dlmZ dgZd ZdZe dg�Z
e dg�Z e dg�Z e dg�Z
e d g�Ze d g�Ze d
g�Ze d g�Ze d g�Ze d
g�Ze dg�Ze dg�Ze dg�Ze dg�Ze dg�Ze dg�Ze dg�Ze d g�Ze dg�Z e dg�Z e dg�Z e dg�Z e dg�Z e dg�Z!e dg�Z"e d g�Z#e d g�Z$e d g�Z%e d g�Z&e d g�Z'e d!g�Z(e d"g�Z)e d#g�Z*e d$g�Z+e d%g�Z,e d&g�Z-e d'g�Z.e d(g�Z/e d)g�Z0e d*g�Z1e dg�Z2e d+g�Z3e d,g�Z4e d-g�Z5e d.g�Z6e d/g�Z7e d0g�Z8e d1g�Z9e d2g�Z:e d3g�Z;e d4g�Ze d7g�Z?e d8g�Z@e d9g�ZAe d:g�ZBe d;g�ZCe dg�ZFe d?g�ZGe d@g�ZHe dAg�ZIe dBg�ZJe dCg�ZKe dDg�ZLe dEg�ZMe dFg�ZNe dGg�ZOe dg�ZPe d g�ZQeRedH��rejSZTnejUZTG dId� d�ZVdJdK� ZWeXdLk�rBeW� dS )M� N)� monotonic�Telnet� �� �� �� �� �� �� �� �� �� �� �� �� �� �� �� �� � � � � � � � � � �
� � �
� � � � � � � � � � � � � � � � � � �! �" �# �$ �% �& �' �( �) �* �+ �, �- �. �/ �0 �1 � � � � PollSelectorc @ s e Zd Zddejfdd�Zdejfdd�Zdd� Zd d
� Zd d � Z d
d� Z
dd� Z dd� Z dd� Z
d;dd�Zdd� Zdd� Zdd � Zd d � Zd d � Zd!d"� Zd#d$� Zd%d&� Zd'd(� Zd)d*� Zd+d,� Zd-d.� Zd/d0� Zd1d2� Z d3d4� Z d | j � x$| j r,| j r,| j� | j � q
W | j}d| _|S )NrI )ro rR rS rx )rY r} rZ rZ r[ � read_someS s zTelnet.read_somec C s6 | j � x$| j r,| j� r,| j� | j � q
W | j� S )N)ro rS �
sock_availrx ry )rY rZ rZ r[ �read_very_eagerb s
zTelnet.read_very_eagerc C s> | j � x,| j r4| j r4| j� r4| j� | j � q
W | j� S )N)ro rR rS r� rx ry )rY rZ rZ r[ �
read_eagerp s
zTelnet.read_eagerc C s | j � | j� S )N)ro ry )rY rZ rZ r[ � read_lazy~ s zTelnet.read_lazyc C s, | j }d| _ | r(| jr(| j r(td��|S )NrI ztelnet connection closed)rR rS rP �EOFError)rY r} rZ rZ r[ ry � s
zTelnet.read_very_lazyc C s | j }d| _ |S )NrI )rV )rY r} rZ rZ r[ � read_sb_data� s zTelnet.read_sb_datac C s
|| _ d S )N)rW )rY �callbackrZ rZ r[ � set_option_negotiation_callback� s z&Telnet.set_option_negotiation_callbackc C sX ddg}�y�x�| j �r | j� }| jsj|tkr0q|dkr:q|tkrZ|| j | || j< qn| j|7 _qt| j�dk�r(|ttt t
fkr�| j|7 _qd| _|tkr�|| j | || j< nh|t kr�d| _d| _ n&|t
kr�d| _| j |d | _ d|d< | j�r| j| j|t� n| jdt|� � qt| j�dkr| jdd� }d| _|}|ttfk�r�| jd|tk�rrd�ptd t|�� | j�r�| j| j||� n| jjtt
| � q|t t
fkr| jd|t k�r�d
�p�d t|�� | j�r�| j| j||� q| jjtt | � qW W n" tk
�r2 d| _d| _Y nX | j|d | _| j |d | _ d S ) NrI � r r zIAC %d not recognizedr z IAC %s %d�DO�DONT�WILL�WONT)rP � rawq_getcharrT �theNULLri rU rn r� r� r� r� �SBrV �SErW rO �NOOPTrd �ordrk r� rR )rY r} �c�cmd�optrZ rZ r[ ro � sh zTelnet.process_rawqc C sZ | j s| j� | jrt�| j | j| jd � }| jd | _| jt| j �krVd| _ d| _|S )Nr rI r )rP rx rS r� rQ rn )rY r� rZ rZ r[ r� � s zTelnet.rawq_getcharc C sL | j t| j�kr d| _d| _ | jjd�}| jd|� | | _| j| | _d S )NrI r �2 zrecv %r)rQ rn rP rO Zrecvrd rS )rY r} rZ rZ r[ rx s zTelnet.fill_rawqc
C s. t � � }|j| tj� t|jd��S Q R X d S )Nr )rr rs rt ru �boolrv )rY r rZ rZ r[ r� s zTelnet.sock_availc C s� t jdkr| j� d S t� ��}|j| tj� |jt jtj� x�x�|j� D ]�\}}|j | kr�y | j
� }W n t k
r� t d� d S X |r�t j
j|jd�� t j
j� qH|j t jkrHt jj� jd�}|s�d S | j|� qHW q>W W d Q R X d S )NZwin32z(*** Connection closed by remote host ***�ascii)�sys�platform� mt_interactrr rs rt ru �stdinrv Zfileobjr� r� rc �stdoutrm �decode�flush�readline�encode)rY r �keyZevents�text�linerZ rZ r[ �interact s,
zTelnet.interactc C s@ dd l }|j| jf � x$tjj� }|s(P | j|jd�� qW d S )Nr r� )�_thread�start_new_thread�listenerr� r� r� rm r� )rY r� r� rZ rZ r[ r� 1 s
zTelnet.mt_interactc C sV xPy | j � }W n tk
r* td� d S X |rDtjj|jd�� qtjj� qW d S )Nz(*** Connection closed by remote host ***r� )r� r� rc r� r� rm r� r� )rY �datarZ rZ r[ r� ; s zTelnet.listenerc C sP d }|d d � }t t|��}x8|D ]0}t|| d�s"|s@dd l}|j|| �||< q"W |d k rht� | }t� ��}|j| tj � x�| j
�s | j � xP|D ]H}|| j | j
�}|r�|j� } | j
d | � }
| j
| d � | _
|||
fS q�W |d k �r|j|�} |t� }| �s|dk r�P nq�| j� q�W W d Q R X | j� }
|
�rF| j
�rFt�dd |
fS )N�searchr r �����)�rangern �hasattr�re�compilerq rr rs rt ru rS ro r� rR rb rv rx ry r� ) rY �listrN r� �indicesr| r~ r �m�er� ZreadyrZ rZ r[ �expectH sB
z
Telnet.expectc C s | S )NrZ )rY rZ rZ r[ � __enter__� s zTelnet.__enter__c C s | j � d S )N)r_ )rY �type�value� tracebackrZ rZ r[ �__exit__� s zTelnet.__exit__)N)N)!�__name__�
__module__� __qualname__r^ Z_GLOBAL_DEFAULT_TIMEOUTr\ rX r` rd rf r_ rg rh rm r� r� r� r� r� r� ry r� r� ro r� rx r� r� r� r� r� r� r� rZ rZ rZ r[ r � s: 7
%
H
8c C s� d} x0t jdd � r4t jd dkr4| d } t jd= qW d}t jdd � rRt jd }d}t jdd � r�t jd }y t|�}W n tk
r� tj|d�}Y nX t� �(}|j| � |j||dd� |j � W d Q R X d S ) Nr r z-dZ localhostr Ztcpg �?)rN )
r� �argv�int�
ValueErrorr^ Z
getservbynamer rf rX r� )rK rL rM ZportstrZtnrZ rZ r[ �test� s$
r� �__main__)Yr� r^ rt Ztimer rq �__all__rJ r] �bytesri r� r� r� r� r� r� ZNOPZDMZBRKZIPZAOZAYTZECZELZGAr� ZBINARYZECHOZRCPZSGAZNAMSZSTATUSZTMZRCTEZNAOLZNAOPZNAOCRDZNAOHTSZNAOHTDZNAOFFDZNAOVTSZNAOVTDZNAOLFDZXASCIIZLOGOUTZBMZDETZSUPDUPZ SUPDUPOUTPUTZSNDLOCZTTYPEZEORZTUIDZOUTMRKZTTYLOCZ VT3270REGIMEZX3PADZNAWSZTSPEEDZLFLOWZLINEMODEZXDISPLOCZ OLD_ENVIRONZAUTHENTICATIONZENCRYPTZ NEW_ENVIRONZTN3270EZXAUTH�CHARSETZRSPZCOM_PORT_OPTIONZSUPPRESS_LOCAL_ECHOZTLSZKERMITZSEND_URLZ FORWARD_XZ PRAGMA_LOGONZ
SSPI_LOGONZPRAGMA_HEARTBEATZEXOPLr� r� rH rr ZSelectSelectorr r� r� rZ rZ rZ r[ �$ s�
}