File: //usr/local/CyberCP/lib/python3.10/site-packages/asyncssh/__pycache__/packet.cpython-310.pyc
o
�hk � @ sP d Z ddlmZmZmZmZmZmZ ddlmZm Z ddl
m Z ddl m
Z
mZ e edf Zeeeedge
d f ZG d d
� d
e�Zd ed efd
d�Zd ed efdd�Zd ed efdd�Zd ed efdd�Zd ed efdd�Zd e eef d efdd�Z d ed efdd�Z d ee d efdd �Z G d d� d�Z G d d � d �Z G d d!� d!e �Z!dS )"z*SSH packet encoding and decoding functions� )�Any� Awaitable�Callable�Iterable�Mapping�Optional)�Sequence�Union� )� SSHLogger)�
MaybeAwait�plural� SSHPacketNc @ s e Zd ZdZdS )�PacketDecodeErrorzPacket decoding errorN)�__name__�
__module__� __qualname__�__doc__� r r �B/usr/local/CyberCP/lib/python3.10/site-packages/asyncssh/packet.pyr " s r �value�returnc C s
t | f�S )zEncode a single byte)�bytes�r r r r �Byte& s
r c C s t t| ��S )zEncode a boolean value)r �boolr r r r �Boolean, � r c C � | � dd�S )z Encode a 16-bit integer value� �big��to_bytesr r r r �UInt162 r r# c C r )z Encode a 32-bit integer value� r r! r r r r �UInt328 r r% c C r )z Encode a 64-bit integer value� r r! r r r r �UInt64> r r' c C s, t | t�r | jddd�} t| ��dd�| S )z*Encode a byte string or UTF-8 string valuezutf-8�strict)�errorsr$ r )�
isinstance�str�encode�lenr" r r r r �StringD s
r. c C sX | � � }||d dko| dko| d|d > k7 }|d d }|�dd�| j|ddd � S )
z)Encode a multiple precision integer valuer& r ���r
� r$ r T��signed)�
bit_lengthr" )r �lr r r �MPIntM s ( r5 c C s t d�| ��S )z-Encode a comma-separated list of byte strings� ,)r. �joinr r r r �NameListW � r8 c @ s� e Zd ZdZdefdd�Zdefdd�Zd$d d
�Zdefd d �Z defd
d�Z
defdd�Z de defdd�Z
de fdd�Zdefdd�Zde fdd�Zde fdd�Zde fd d �Zdefd d �Zde fd d!�Zdee fd"d#�ZdS )%r z Decoder class for SSH packets�packetc C s || _ d| _t|�| _d S )Nr )�_packet�_idxr- �_len)�selfr: r r r �__init__` s zSSHPacket.__init__r c C s | j | jkS )N)r< r= �r> r r r �__bool__e s zSSHPacket.__bool__Nc C s | rt d��dS )z<Confirm that all of the data in the packet has been consumedz Unexpected data at end of packetN)r r@ r r r � check_endh s �zSSHPacket.check_endc C s | j d| j� S )z0Return the portion of the packet consumed so farN�r; r< r@ r r r �get_consumed_payloadn � z SSHPacket.get_consumed_payloadc C s | j | jd� S )z1Return the portion of the packet not yet consumedNrC r@ r r r �get_remaining_payloads rE z SSHPacket.get_remaining_payloadc C s | j S )zReturn the full packet)r; r@ r r r �get_full_payloadx s zSSHPacket.get_full_payload�sizec C s@ | j | | jkr td��| j| j | j | � }| j |7 _ |S )z5Extract the requested number of bytes from the packetzIncomplete packet)r< r= r r; )r> rH r r r r � get_bytes} s
zSSHPacket.get_bytesc C s | � d�d S )z%Extract a single byte from the packetr
r )rI r@ r r r �get_byte� r9 zSSHPacket.get_bytec C s t | �� �S )z!Extract a boolean from the packet)r rJ r@ r r r � get_boolean� r zSSHPacket.get_booleanc C � t �| �d�d�S )z(Extract a 16-bit integer from the packetr r ��int�
from_bytesrI r@ r r r �
get_uint16� � zSSHPacket.get_uint16c C rL )z(Extract a 32-bit integer from the packetr$ r rM r@ r r r �
get_uint32� rQ zSSHPacket.get_uint32c C rL )z(Extract a 64-bit integer from the packetr& r rM r@ r r r �
get_uint64� rQ zSSHPacket.get_uint64c C s | � | �� �S )z&Extract a UTF-8 string from the packet)rI rR r@ r r r �
get_string� r9 zSSHPacket.get_stringc C s t j| �� ddd�S )z4Extract a multiple precision integer from the packetr Tr1 )rN rO rT r@ r r r � get_mpint� s zSSHPacket.get_mpintc C s | � � }|r |�d�S g S )z>Extract a comma-separated list of byte strings from the packetr6 )rT �split)r> �namelistr r r � get_namelist� s zSSHPacket.get_namelist)r N)r r r r r r? r rA rB rD rF rG rN rI rJ rK rP rR rS rT rU r rX r r r r r ] s"
c @ s� e Zd ZU dZi Zeeef ed< e de
fdd��Z dedede e d e
d
edd f d d
�Z ddede e d e
d
edd f
dd�Z ddede e d e
d
edd f
dd�Zd S )�SSHPacketLoggerz#Parent class for SSH packet loggers�_handler_namesr c C � t �)z$The logger to use for packet logging��NotImplementedErrorr@ r r r �logger� � zSSHPacketLogger.logger�msg�pkttype�pktidr: �noteNc C s� t |t�r |�� }z
| j| � d|� d�}W n ty$ d|� �}Y nw tt|�d�}|r4d|� d�}| j�||d||||� dS )zLog a sent/received packetz (�)z packet type �bytez %s %s, %s%sN) r* r rG rZ �KeyErrorr
r- r^ r: )r> r` ra rb r: rc �name�countr r r � _log_packet� s
� �zSSHPacketLogger._log_packet� c C � | � d||||� dS )zLog a sent packet�SentN�ri �r> ra rb r: rc r r r �log_sent_packet� � z SSHPacketLogger.log_sent_packetc C rk )zLog a received packet�ReceivedNrm rn r r r �log_received_packet� rp z#SSHPacketLogger.log_received_packet)rj )r r r r rZ r rN r+ �__annotations__�propertyr r^ r �
_LoggedPacketri ro rr r r r r rY � s8
��
����
�����rY c @ s^ e Zd ZU dZi Zeeef ed< e de
fdd��Z dedede de
eed f fd
d �Zd S ) �SSHPacketHandlerz$Parent class for SSH packet handlers�_packet_handlersr c C r[ )z.The logger associated with this packet handlerr\ r@ r r r r^ � r_ zSSHPacketHandler.loggerra rb r: Nc C s& || j v r| j | | |||�pdS dS )z!Log and process a received packetTF)rw )r> ra rb r: r r r �process_packet� s
��z SSHPacketHandler.process_packet)r r r r rw r rN �_PacketHandlerrs rt r r^ r r r r rx r r r r rv � s
��rv )"r �typingr r r r r r r r �loggingr �miscr r
r ru rN ry �
ValueErrorr r r r r# r% r' r+ r. r5 r8 r rY rv r r r r �<module> s&
T-