File: //usr/local/CyberCP/lib/python3.10/site-packages/asyncssh/__pycache__/tuntap.cpython-310.pyc
o
�h�/ �
@ s d Z ddlZddlZddlZddlZddlZddlZddlZddlm Z m
Z
m Z m Z ej
dkr3ddlZdZdZdZdZdZd Zd
Zd Zd Zd
ZdZdZdZdZ dZ G dd� dej �Z G dd� de �Z G dd� de �Z!G dd� de �Z"de g ej#f de$de
e$ de e ej#f fd d �Z%dS ) z SSH TUN/TAP forwarding support� N)�Callable�Optional�Tuple�cast�win32� � i���� l N� �I96sl i@ l i@ i�T@i �16sHc s e Zd ZdZdejdef� fdd� Zdejfdd�Z d ejdd
fd d �Z
d(d
d�Z de fdd�Z
d(dd�Zd(dd�Zde fdd�Zdefdd�Zdeeef fdd�Z
d)dee d ee dd
fd d �Zd(d d �Zd!edd
fd"d#�Zde fd$d%�Zd(d&d'�Z� Z S )*�SSHTunTapTransportzLayer 2/3 tunnel transport�loop� interfacec s" t � jd|id� || _d | _d S )Nr )�extra)�super�__init__�_loop� _protocol)�selfr
r �� __class__� �B/usr/local/CyberCP/lib/python3.10/site-packages/asyncssh/tuntap.pyr ? s
zSSHTunTapTransport.__init__�returnc C s | j dusJ �| j S )z-Get protocol object associated with transportN)r �r r r r � get_protocolE s z SSHTunTapTransport.get_protocol�protocolNc C s t tj|�| _dS )z&Set protocol associated with transportN)r �asyncio�Protocolr )r r r r r � set_protocolL s z SSHTunTapTransport.set_protocolc C s | � � dS )zAbort this transportN)�closer r r r �abortQ s zSSHTunTapTransport.abortc C � t ���'Return if the transport is reading data��NotImplementedErrorr r r r �
is_readingV � z SSHTunTapTransport.is_readingc C r" )�
Pause readingr% r r r r �
pause_reading[ r( z SSHTunTapTransport.pause_readingc C r" )�Resume readingr% r r r r �resume_reading` r( z!SSHTunTapTransport.resume_readingc C � dS )z*This transport doesn't support writing EOFFr r r r r �
can_write_eofe r( z SSHTunTapTransport.can_write_eofc C r- )z#This transport has no output bufferr r r r r r �get_write_buffer_sizej r( z(SSHTunTapTransport.get_write_buffer_sizec C r- )�2This transport doesn't support write buffer limits)r r r r r r r �get_write_buffer_limitso r( z*SSHTunTapTransport.get_write_buffer_limits�high�lowc C r- )r0 Nr )r r2 r3 r r r �set_write_buffer_limitst � z*SSHTunTapTransport.set_write_buffer_limitsc C r- )z$Ignore writing EOF on this transportNr r r r r � write_eofx r5 z SSHTunTapTransport.write_eof�datac C r" )�Write a packetr% �r r7 r r r �write{ r( zSSHTunTapTransport.writec C r- )z"Return if the transport is closingFr r r r r �
is_closing� r( z SSHTunTapTransport.is_closingc C r" )�Close this transportr% r r r r r � r( zSSHTunTapTransport.close�r N)NN) �__name__�
__module__� __qualname__�__doc__r �AbstractEventLoop�strr � BaseProtocolr r r! �boolr' r* r, r. �intr/ r r1 r r4 r6 �bytesr: r; r �
__classcell__r r r r r < s.
��
�
r c �| e Zd ZdZdejdedee f� fdd� Zde fdd �Z
dd d �Z dd
d�Z ddd�Z
dedd
fdd�Zddd�Z� ZS )�SSHTunTapOSXTransportzTunTapOSX transportr
�mode�unitc
sD |t krdnd}|d u r9td�D ] }ztd|� |� �ddd�}W n ty) Y qw |} n
ttjd|� d ���n td|� |� �ddd�}|� |� �}|�� }t�tjtj �} z.t
� t |d�}
t
�| t|
�}
t
�t |
�\} } | tO } t
� t || �}
t
�| t|
� W | �� n| �� w t� �||� || _d | _t�|�� d
� d S ) N�tun�tap� z/dev/�rb+r �� bufferingzNo z devices availableT)�SSH_TUN_MODE_POINTTOPOINT�range�open�OSError�errno�EBUSY�encode�socket�AF_INET�
SOCK_DGRAM�struct�pack�IFF_FMT�fcntl�ioctl�DARWIN_SIOCGIFFLAGS�unpack�IFF_UP�DARWIN_SIOCSIFFLAGSr r r �_file� _read_thread�os� set_blocking�fileno)
r r
rK rL �prefix�i�filer �name�sock�ifr�_�flagsr r r r � s8 �� z SSHTunTapOSXTransport.__init__r c C s
| j duS )r$ N)rg r r r r r' � s
z SSHTunTapOSXTransport.is_readingNc C s | j r
| j �� d| _ dS dS )r) N)rg �joinr r r r r* � s
�z#SSHTunTapOSXTransport.pause_readingc C s0 | j stj| jd�| _ d| j _| j �� dS dS )r+ )�targetTN)rg � threading�Thread�
_read_loop�daemon�startr r r r r, � s
�z$SSHTunTapOSXTransport.resume_readingc
C sz | j dusJ � z| j�d�}W n" ty2 } z|jtjkr'| j�| j j|� W Y d}~dS d}~ww | j�| j j |� q)z1Loop reading packets until read is paused or doneNT� )
r rf �readrV rW �EBADFr �call_soon_threadsafe�connection_lost�
data_received�r r7 �excr r r rw � s �����z SSHTunTapOSXTransport._read_loopr7 c C � | j �|� dS �r8 N�rf r: r9 r r r r: � � zSSHTunTapOSXTransport.writec C � | j �� | �� dS �r< N�rf r r* r r r r r � �
zSSHTunTapOSXTransport.closer= )r> r? r@ rA r rB rF r r rE r' r* r, rw rG r: r rH r r r r rJ � s �)
rJ c sx e Zd ZdZdejdee f� fdd� Zde fdd�Z
dd
d �Z dd d
�Z ddd�Z
dedd fdd�Zddd�Z� ZS )�SSHDarwinUTunTransportzDarwin UTun transportr
rL c s� t � t jt jt j�}z4t�tdd�}t�|t |�}t�
t|�\}}|d u r'dn|d }|� d� |� ||f� |�
� \}}W n tyJ |�� � w t� �|d|d � �� || _d| _d S )Nr s com.apple.net.utun_control� F�utunr )rZ � PF_SYSTEMr\ �SYSPROTO_CONTROLr] r^ �DARWIN_CTLIOCGINFO_FMTr` ra �DARWIN_CTLIOCGINFOrc � setblocking�connect� getpeernamerV r r r �_sock�_reading)r r
rL ro �arg�ctl_info�ctl_idrq r r r r � s( ��
�
z SSHDarwinUTunTransport.__init__r c C � | j S r# �r� r r r r r' � z!SSHDarwinUTunTransport.is_readingNc C s d| _ | j�| j� dS �r) FN)r� r �
remove_readerr� r r r r r*
s z$SSHDarwinUTunTransport.pause_readingc C � d| _ | j�| j| j� dS �r+ TN)r� r �
add_readerr� � _read_readyr r r r r, � z%SSHDarwinUTunTransport.resume_readingc
C s� | j dusJ � z | j�d�dd� }W n$ ttfy Y dS ty8 } z
| j �|� W Y d}~dS d}~ww | j �|� q)�)Read available packets from the transportNTi � )r r� �recv�BlockingIOError�InterruptedErrorrV r~ r r� r r r r� s �� �z"SSHDarwinUTunTransport._read_readyr7 c C s@ |d d? }|dkr
t jnt j}|�dd�| }| j�|� dS )r8 r r� �bigN)rZ r[ �AF_INET6�to_bytesr� �send)r r7 �version�familyr r r r: ) s z SSHDarwinUTunTransport.writec C r� r� )r� r r* r r r r r 2 r� z SSHDarwinUTunTransport.closer= )r> r? r@ rA r rB r rF r rE r' r* r, r� rG r: r rH r r r r r� � s
r� c rI )�SSHLinuxTunTapTransportzLinux TUN/TAP transportr
rK rL c s� t dddd�}|tkrttB }d}nttB }d}|d u r dn|� |� ��� }t�t||�}z t �
|t |�}W n t yB |�
� � w t�t|�\}} |�d��� }
t� �||
� || _d | _t�|�� d � d S )
Nz /dev/net/tunrP r rQ rM rN � � F)rU rS �
LINUX_IFF_TUN�LINUX_IFF_NO_PI�
LINUX_IFF_TAPrY r] r^ r_ r` ra �LINUX_TUNSETIFFrV r rc �strip�decoder r rf r� rh ri rj ) r r
rK rL rm rr rk rn rp rq r r r r r < s( �z SSHLinuxTunTapTransport.__init__r c C r� r# r� r r r r r' Z r� z"SSHLinuxTunTapTransport.is_readingNc C s0 d| _ z
| j�| j� W dS ty Y dS w r� )r� r r� rf rV r r r r r* _ s �z%SSHLinuxTunTapTransport.pause_readingc C r� r� )r� r r� rf r� r r r r r, i r� z&SSHLinuxTunTapTransport.resume_readingc
C sn | j dusJ � z| j�d�}W n ty) } z
| j �|� W Y d}~dS d}~ww |du r0dS | j �|� q)r� NTrz )r rf r{ rV r~ r r� r r r r� o s �� �z#SSHLinuxTunTapTransport._read_readyr7 c C r� r� r� r9 r r r r: � r� z SSHLinuxTunTapTransport.writec C r� r� r� r r r r r � r� z SSHLinuxTunTapTransport.closer= )r> r? r@ rA r rB rF r r rE r' r* r, r� rG r: r rH r r r r r� 9 s �
r� �protocol_factoryrK rL r c C s� t �� }d}tjdkrA|du r,zt|||�}W n? ty+ |tkr(t||�}n� Y n,w |tkr:|dkr:t||�}n t|||�}ntjdkrMt|||�}n
tt j
dtj� ���|dus]J �| � }|� |� |� |� |�
� ||fS )z+Create a local TUN or TAP network interfaceN�darwinrO �linuxzTunTap not supported on )r �get_event_loop�sys�platformrJ rV rS r� r� rW �EPROTONOSUPPORT�connection_mader r, )r� rK rL r
� transportr r r r �
create_tuntap� s4
��
�
r� )&rA r rW rh rZ r] r� ru �typingr r r r r� r` rS �SSH_TUN_MODE_ETHERNET�SSH_TUN_UNIT_ANY�SSH_TUN_AF_INET�SSH_TUN_AF_INET6r� r� rb re r� r� r� r� r_ rd � Transportr rJ r� r� rD rF r� r r r r �<module> sJ
O^PS�� �