File: //lib/python3/dist-packages/twisted/conch/__pycache__/unix.cpython-38.pyc
U
��W[�? � @ s� d Z ddlZddlZddlZddlZddlZddlZddlZddlZddl Z ddl
m Z ddl m
Z
ddlmZ ddlmZ ddlmZ ddlmZmZmZ dd lmZmZmZm Z m Z m Z dd
l m Z m!Z!m"Z" dd l#m$Z$ dd l%m&Z& dd
l'm(Z(m)Z) ddl*m+Z,m-Z- z ddl.Z.W n e/k
�r. dZ.Y nX e e$j0�G dd� d��Z1G dd� de�Z2e e �G dd� d��Z3e e!�G dd� d��Z4e e"�G dd� d��Z5G dd� d�Z6e(�7e4e2ej!� e(�7e3e2ej � dS )z
A UNIX SSH server.
� N)� implementer)�ttymodes)� ConchUser)�
ConchError)�lsLine)�session�
forwarding� filetransfer)�FXF_READ� FXF_WRITE�
FXF_APPEND� FXF_CREAT� FXF_TRUNC�FXF_EXCL)�ISession� ISFTPServer� ISFTPFile)�portal)�ProcessExitedAlready)�
components�log)� _bytesChr� nativeStringc @ s e Zd Zdd� ZdS )� UnixSSHRealmc G s t |�}|d ||jfS �Nr )�
UnixConchUser�logout)�self�usernameZmindZ
interfaces�user� r �4/usr/lib/python3/dist-packages/twisted/conch/unix.py�
requestAvatar+ s zUnixSSHRealm.requestAvatarN)�__name__�
__module__� __qualname__r" r r r r! r ) s r c @ sT e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dd� Z dS )r c C s� t �| � || _t�| j�| _| jd g}t�� D ] \}}}}||kr2|�|� q2|| _ i | _
| j � t
jtjd�� | j� dtji� d S )N� )s sessions direct-tcpips sftp)r �__init__r �pwd�getpwnam�pwdData�grpZgetgrall�append� otherGroups� listenersZ
channelLookup�updater Z
SSHSessionr ZopenConnectForwardingClientZsubsystemLookupr ZFileTransferServer)r r �lZ groupnameZpassword�gidZuserlistr r r! r' 3 s"
���zUnixConchUser.__init__c C s | j dd� S )N� � �r* �r r r r! �getUserGroupIdE s z UnixConchUser.getUserGroupIdc C s | j S �N)r- r5 r r r! �getOtherGroupsI s z UnixConchUser.getOtherGroupsc C s
| j d S )N� r4 r5 r r r! �
getHomeDirM s zUnixConchUser.getHomeDirc C s
| j d S )N� r4 r5 r r r! �getShellQ s zUnixConchUser.getShellc C s� t �|�\}}ddlm} z*| j|j|t �| j||ft j�|d�}W n Y dS X || j ||f< |dkr�|�
� d }dt � d|�fS dS d S )Nr ��reactor)Z interfacer2 � z>L)
r �unpackGlobal_tcpip_forward�twisted.internetr> �
_runAsUserZ listenTCPZSSHListenForwardingFactory�connZ SSHListenServerForwardingChannelr. �getHost�struct�pack)r �data�
hostToBind�
portToBindr> �listenerr r r! �global_tcpip_forwardU s( ��
z"UnixConchUser.global_tcpip_forwardc C sD t �|�\}}| j�||fd �}|s(dS | j||f= | �|j� dS )Nr r? )r r@ r. �getrB �
stopListening)r rG rH rI rJ r r r! �global_cancel_tcpip_forwardk s z)UnixConchUser.global_cancel_tcpip_forwardc C s: | j �� D ]}| �|j� q
t�d| jt| j �f � d S )Nzavatar %s logging out (%i))r. �valuesrB rM r �msgr �len)r rJ r r r! r u s ��zUnixConchUser.logoutc O s t �� }t �� }t �� }| �� \}}t �d� t �d� t �| �� � t �|� t �|� z t |�}W n t
k
r� |||fg}Y nX zP|D ]F} | d }
t | �dkr�| d p�d}t | �dkr�| d p�i }|
||�} q�W 5 t �d� t �d� t �|� t �|� t �|� X | S )Nr r? r r2 ) �os�geteuid�getegid� getgroupsr6 �setegid�seteuid� setgroupsr8 �iter� TypeErrorrQ ) r �f�args�kw�euid�egid�groups�uidr1 �i�func�rr r r! rB ~ s2
zUnixConchUser._runAsUserN) r# r$ r% r' r6 r8 r: r< rK rN r rB r r r r! r 1 s
r c @ sh e Zd Zddd�Zddd�Zdd� Zd d
� Zd d � Zd
d� Zdd� Z dd� Z
dd� Z dd� Z dd� Z
dS )�SSHSessionForUnixConchUserNc C s: |dkrddl m} || _|| _ddi| _d| _d| _dS )a
Construct an C{SSHSessionForUnixConchUser}.
@param avatar: The L{UnixConchUser} for whom this is an SSH session.
@param reactor: An L{IReactorProcess} used to handle shell and exec
requests. Uses the default reactor if None.
Nr r= �PATHz /bin:/usr/bin:/usr/local/bin)rA r> �_reactor�avatar�environ�pty�ptyTuple)r rh r> r r r! r' � s
z#SSHSessionForUnixConchUser.__init__r? c C s t sd S | jjjj�� j}t�dt� |��\}| j
d dd � }t � � }t |�}t || d �}t �
� }|rrt jpvt j|_| jj|_||_|dd � |_||f|_|r�| jj|_t�|�d |_|dddf|_t � t j �} | � |� | � � t � t j �}
|
� |� |
� � d S )N�Lr2 r9 g ��.A���r )!�utmprh rC � transport�getPeer�hostrE Zunpack�socketZ inet_atonrk �time�intZ UtmpEntryZ USER_PROCESSZ DEAD_PROCESSZut_typerj �pidZut_pidZut_lineZut_idZut_tvr Zut_userZ
gethostbyaddrZut_hostZ
ut_addr_v6Z
UtmpRecordZ UTMP_FILEZ pututlineZendutentZ WTMP_FILE) r ZloggedInZ ipAddressZpackedIpZttyName�t�t1�t2�entry�a�br r r! � addUTMPEntry� s0
z'SSHSessionForUnixConchUser.addUTMPEntryc C sF || j d< || _|| _t�� \}}t�|�}|| j d< |||f| _d S )NZTERMZSSH_TTY)ri �winSize�modesrj �openptyrR �ttynamerk )r ZtermZ
windowSizer~ ZmasterZslaver� r r r! �getPty� s
z!SSHSessionForUnixConchUser.getPtyc
C s8 | j st�d� td��| j�� \}}| j�� }| j�� }| jj| j d< || j d< || j d< t
j � |�}| jj
jj�� }| jj
jj�� }d|j|j|jf | j d< | �� | jj||d|f g| j |||| j d �| _| �� t�| j�� tj t j d | j �� � | j �r| �!� |jj"| _#| j$|j_"| jj
jj�%d � d S )
Nz'tried to get shell without pty, failingzno ptyZUSER�HOMEZSHELL�%s %s %s�
SSH_CLIENTz-%s�ZusePTY�4Hr? )r� )&rk r rP r rh r6 r: r< r ri rR �path�basenamerC ro rp rD rq �port�getPtyOwnershiprg � spawnProcessrj r| �fcntl�ioctl�fileno�tty�
TIOCSWINSZrE rF r} r~ �setModes�write�oldWrite�
_writeHack�
setTcpNoDelay) r �protora r1 �homeDir�shellZ shellExec�peerrq r r r! � openShell� sJ
�
��
z$SSHSessionForUnixConchUser.openShellc
C s� | j �� \}}| j �� }| j �� p$d}|| jd< |d|f}| j jjj�� }| j jjj�� } d|j |j
| j
f | jd< | j r�| � � | j
j|||| j|||| j p�dd�| _| j r�| �� | jr�| �� | j jjj�d� d S ) Nz/bin/shr� z-cr� r� r r� r? )rh r6 r: r< ri rC ro rp rD rq r� rk r� rg r� rj r| r~ r� r� )
r r� �cmdra r1 r� r� Zcommandr� rq r r r! � execCommand� s:
� �z&SSHSessionForUnixConchUser.execCommandc C s| t �| jd �d }| j�� \}}t �� t �� }}t �d� t �d� zt � | jd ||� W 5 t �|� t �|� X d S )Nr2 r9 r )
rR �statrk rh r6 rS rT rV rW �chown)r ZttyGidra r1 r^ r_ r r r! r� s
z*SSHSessionForUnixConchUser.getPtyOwnershipc C s | j }t�|�� �}| jD ]�\}}|tjkr.qtj| }t|�dkr�|\}}tt|�sXqt t|�}|rx|| |B ||< q�|| | @ ||< q|dkr�t td|f �|tj
< q|dkr�t td|f �|tj < qtt|�s�qt t|�}t |�|tj
|< qt�|�� tj|� d S )Nr2 �OSPEEDzB%s�ISPEED)rj r� � tcgetattrr� r~ r ZTTYMODESrQ �hasattr�getattrr� r� �chrZCCZ tcsetattrZTCSANOW) r rj �attr�modeZ modeValueZttyMode�flagZttyAttrZttyvalr r r! r� s.
z#SSHSessionForUnixConchUser.setModesc C s | j r| j �� d S r7 )rj Z
closeStdinr5 r r r! � eofReceived0 s z&SSHSessionForUnixConchUser.eofReceivedc C s� | j r@tj�| j d �r@t�| j d �d }t�| j d d|� | jr�z| j�d� W n tt fk
rn Y nX | j�
� | � d� t �
d� d S )Nr2 r9 r ZHUPz shell closed)rk rR r� �existsr� r� rj Z
signalProcess�OSErrorr ZloseConnectionr| r rP )r ZttyGIDr r r! �closed5 s
z!SSHSessionForUnixConchUser.closedc C s, || _ t�| j�� tjtjd| j �� � d S )Nr� )r� ) r} r� r� rj r� r� r� rE rF )r r} r r r! �
windowChangedC s �z(SSHSessionForUnixConchUser.windowChangedc C s\ | j dk rNt�| j �� �d }|tj@ sN|tj@ rN| jjj� ddt
|� � | � |� dS )zF
Hack to send ignore messages when we aren't echoing.
Nr& � � ) rj r� r� r� ZECHOZICANONrh rC ro Z
sendIgnorerQ r� )r rG r� r r r! r� J s
z%SSHSessionForUnixConchUser._writeHack)N)r? )r# r$ r% r' r| r� r� r� r� r� r� r� r� r� r r r r! re � s
re c @ s� e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dd� Z dd� Z dd� Z
dd� Zdd� Zdd � Zd d � Zd d � Zd!d"� Zd#S )$�SFTPServerForUnixConchUserc C s
|| _ d S r7 )rh )r rh r r r! r' X s z#SFTPServerForUnixConchUser.__init__c C sj d|kr&d|kr&t �||d |d � d|kr>t �||d � d|krfd|krft �||d |d f� dS )zl
NOTE: this function assumes it runs as the logged-in user:
i.e. under _runAsUser()
ra r1 � permissions�atime�mtimeN)rR r� �chmod�utime�r r� �attrsr r r! � _setAttrs\ s z$SFTPServerForUnixConchUser._setAttrsc C s& |j |j|j|jt|j�t|j�d�S )N)�sizera r1 r� r� r� )�st_size�st_uid�st_gid�st_modert �st_atime�st_mtime�r �sr r r! � _getAttrsi s �z$SFTPServerForUnixConchUser._getAttrsc C s" | j �� }tj�t|j�t|��S r7 )rh r: rR r� �joinr )r r� �homer r r! �_absPatht s
z#SFTPServerForUnixConchUser._absPathc C s i S r7 r )r Z otherVersion�extDatar r r! �
gotVersiony s z%SFTPServerForUnixConchUser.gotVersionc C s t | | �|�||�S r7 )� UnixSFTPFiler� )r �filename�flagsr� r r r! �openFile} s z#SFTPServerForUnixConchUser.openFilec C s | � |�}| j�tj|�S r7 )r� rh rB rR �remove)r r� r r r! �
removeFile� s
z%SFTPServerForUnixConchUser.removeFilec C s& | � |�}| � |�}| j�tj||�S r7 )r� rh rB rR �rename)r Zoldpath�newpathr r r! �
renameFile� s
z%SFTPServerForUnixConchUser.renameFilec C s, | � |�}| j�tj|ff| j||ffg�S r7 )r� rh rB rR �mkdirr� r� r r r! �
makeDirectory� s
�z(SFTPServerForUnixConchUser.makeDirectoryc C s | � |�}| j�tj|� d S r7 )r� rh rB rR �rmdir�r r� r r r! �removeDirectory� s
z*SFTPServerForUnixConchUser.removeDirectoryc C s t | | �|��S r7 )�UnixSFTPDirectoryr� r� r r r! �
openDirectory� s z(SFTPServerForUnixConchUser.openDirectoryc C s: | � |�}|r | j�tj|�}n| j�tj|�}| �|�S r7 )r� rh rB rR r� �lstatr� )r r� Z followLinksr� r r r! �getAttrs� s
z#SFTPServerForUnixConchUser.getAttrsc C s | � |�}| j�| j||� d S r7 )r� rh rB r� r� r r r! �setAttrs� s
z#SFTPServerForUnixConchUser.setAttrsc C s | � |�}| j�tj|�S r7 )r� rh rB rR �readlinkr� r r r! �readLink� s
z#SFTPServerForUnixConchUser.readLinkc C s& | � |�}| � |�}| j�tj||�S r7 )r� rh rB rR �symlink)r ZlinkPathZ
targetPathr r r! �makeLink� s
z#SFTPServerForUnixConchUser.makeLinkc C s t j�| �|��S r7 )rR r� �realpathr� r� r r r! �realPath� s z#SFTPServerForUnixConchUser.realPathc C s t �d S r7 ��NotImplementedError)r ZextNamer� r r r! �extendedRequest� s z*SFTPServerForUnixConchUser.extendedRequestN)r# r$ r% r' r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r r r r! r� V s"
r� c @ s< e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
S )r� c C s || _ d}|t@ tkr(|t@ dkr(tj}|t@ tkrF|t@ dkrFtj}|t@ tkrd|t@ tkrdtj}|t@ tkrz|tjO }|t @ t kr�|tj
O }|t @ t kr�|tj O }|t
@ t
kr�|tjO }d|kr�|d }|d= nd}|j�tj|||�}|�r|j�|j||� || _d S )Nr r� i� )�serverr
r rR �O_RDONLY�O_WRONLY�O_RDWRr �O_APPENDr
�O_CREATr �O_TRUNCr �O_EXCLrh rB �openr� �fd)r r� r� r� r� Z openFlagsr� r� r r r! r' � s0
zUnixSFTPFile.__init__c C s | j j�tj| j�S r7 )r� rh rB rR �closer� r5 r r r! r� � s zUnixSFTPFile.closec C s, | j j�tj| j|dfftj| j|ffg�S r )r� rh rB rR �lseekr� �read)r �offsetZlengthr r r! � readChunk� s
��zUnixSFTPFile.readChunkc C s, | j j�tj| j|dfftj| j|ffg�S r )r� rh rB rR r� r� r� )r r� rG r r r! �
writeChunk� s
��zUnixSFTPFile.writeChunkc C s | j j�tj| j�}| j �|�S r7 )r� rh rB rR �fstatr� r� r� r r r! r� � s zUnixSFTPFile.getAttrsc C s t �d S r7 r� )r r� r r r! r� � s zUnixSFTPFile.setAttrsN) r# r$ r% r' r� r� r� r� r� r r r r! r� � s r� c @ s0 e Zd Zdd� Zdd� Zdd� ZeZdd� Zd S )
r� c C s" || _ |j�tj|�| _|| _d S r7 )r� rh rB rR �listdir�files�dir)r r� Z directoryr r r! r' � s zUnixSFTPDirectory.__init__c C s | S r7 r r5 r r r! �__iter__� s zUnixSFTPDirectory.__iter__c C sl z| j �d�}W n tk
r( t�Y n@X | jj�tjtj �
| j |��}t ||�}| j�
|�}|||fS d S r )r� �pop�
IndexError�
StopIterationr� rh rB rR r� r� r� r� r r� )r r[ r� Zlongnamer� r r r! �__next__ s
�
zUnixSFTPDirectory.__next__c C s
g | _ d S r7 )r� r5 r r r! r� s zUnixSFTPDirectory.closeN)r# r$ r% r' r� r� �nextr� r r r r! r� � s
r� )8�__doc__r� r+ rR rj r( rr rE rs r� Zzope.interfacer Z
twisted.conchr Ztwisted.conch.avatarr Ztwisted.conch.errorr Ztwisted.conch.lsr Ztwisted.conch.sshr r r Z twisted.conch.ssh.filetransferr
r r r
r r Ztwisted.conch.interfacesr r r Z twisted.credr Ztwisted.internet.errorr Ztwisted.pythonr r Ztwisted.python.compatr r� r rn � ImportErrorZIRealmr r re r� r� r� ZregisterAdapterr r r r! �<module> s\
k :f7 � �