File: //usr/lib/python3/dist-packages/twisted/conch/__pycache__/unix.cpython-310.pyc
o
�b�@ � @ s� d Z ddlZddlZddlZddlZddlZddlZddlZddlZddl Z ddl
m Z ddl m
Z
ddlmZ ddlmZ ddlmZmZmZ ddlmZ dd lmZmZmZ dd
l m Z m Z m Z m Z m!Z!m"Z" dd l#m$Z$ dd l%m&Z& dd
l'm(Z( ddl)m*Z* ddl+m,Z, zddl-Z-W n e.y� dZ-Y nw e e$j/�G dd� d��Z0G dd� de�Z1e e�G dd� d��Z2e e�G dd� d��Z3e e�G dd� d��Z4G dd� d�Z5e*�6e3e1ej� e*�6e2e1ej� dS ) z
A UNIX SSH server.
� N)� implementer)�ttymodes)� ConchUser)�
ConchError)�ISession� ISFTPFile� ISFTPServer)�lsLine)� filetransfer�
forwarding�session)�
FXF_APPEND� FXF_CREAT�FXF_EXCL�FXF_READ� FXF_TRUNC� FXF_WRITE)�portal)�ProcessExitedAlready)�Logger)�
components)� nativeStringc @ s e Zd Zdd� ZdS )� UnixSSHRealmc G s t |�}|d ||jfS �Nr )�
UnixConchUser�logout)�self�username�mind�
interfaces�user� r! �4/usr/lib/python3/dist-packages/twisted/conch/unix.py�
requestAvatar0 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 ]\}}}}||v r(|�|� q|| _ i | _
| j � t
jtjd�� | j� dtji� d S )N� )s sessions direct-tcpips sftp)r �__init__r �pwd�getpwnam�pwdData�grp�getgrall�append� otherGroups� listeners�
channelLookup�updater �
SSHSessionr �openConnectForwardingClient�subsystemLookupr
�FileTransferServer)r r �l� groupname�password�gid�userlistr! r! r" r( 6 s
���zUnixConchUser.__init__c C s | j dd� S )N� � �r+ �r r! r! r" �getUserGroupIdI s z UnixConchUser.getUserGroupIdc C s | j S �N)r/ r? r! r! r" �getOtherGroupsL s z UnixConchUser.getOtherGroupsc C �
| j d S )N� r> r? r! r! r" �
getHomeDirO �
zUnixConchUser.getHomeDirc C rC )N� r> r? r! r! r" �getShellR rF zUnixConchUser.getShellc C s� t �|�\}}ddlm} z| j|j|t �| j||ft j�|d�}W n
t y, Y dS w || j
||f< |dkrF|� � d }dt �
d|�fS dS )Nr ��reactor)� interfacer< � z>L)r �unpackGlobal_tcpip_forward�twisted.internetrJ �
_runAsUser� listenTCP�SSHListenForwardingFactory�conn� SSHListenServerForwardingChannel�
BaseExceptionr0 �getHost�struct�pack)r �data�
hostToBind�
portToBindrJ �listenerr! r! r" �global_tcpip_forwardU s* �
�
� z"UnixConchUser.global_tcpip_forwardc C sD t �|�\}}| j�||fd �}|sdS | j||f= | �|j� dS )Nr rL )r rM r0 �getrO �
stopListening)r rX rY rZ r[ r! r! r" �global_cancel_tcpip_forwardn s z)UnixConchUser.global_cancel_tcpip_forwardc C s: | j �� D ]}| �|j� q| jjd| jt| j �d� d S )Nz,avatar {username} logging out ({nlisteners}))r �
nlisteners)r0 �valuesrO r^ �_log�infor �len)r r[ r! r! r" r w s
�zUnixConchUser.logoutc O sD t �� }t �� }t �� }| �� \}}t �d� t �d� t �| �� � t �|� t �|� zt |�}W n t
yB |||fg}Y nw zD|D ]%} | d }
t | �dkrV| d pWd}t | �dkrb| d pci }|
|i |��} qFW t �d� t �d� t �|� t �|� t �|� | S t �d� t �d� t �|� t �|� t �|� w )Nr rL r! r< ) �os�geteuid�getegid� getgroupsr@ �setegid�seteuid� setgroupsrB �iter� TypeErrorrd ) r �f�args�kw�euid�egid�groups�uidr: �i�func�rr! r! r" rO � s@
��
�
zUnixConchUser._runAsUserN) r$ r% r& r( r@ rB rE rH r\ r_ r rO r! r! r! r" r 5 s
r c @ sn e Zd Ze� Zddd�Zddd�Zdd� Zd d
� Zd d � Z d
d� Z
dd� Z dd� Z dd� Z
dd� Zdd� ZdS )�SSHSessionForUnixConchUserNc C s: |du r
ddl 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 rI �PATHz /bin:/usr/bin:/usr/local/bin)rN rJ �_reactor�avatar�environ�pty�ptyTuple)r r{ rJ r! r! r" r( � s
z#SSHSessionForUnixConchUser.__init__rL c C s t sd S | jjjj�� j}t�dt� |��\}| j
d dd � }t � � }t |�}t || d �}t �
� }|r9t jp;t j|_| jj|_||_|dd � |_||f|_|rg| jj|_t�|�d |_|dddf|_t � t j �} | � |� | � � t � t j �}
|
� |� |
� � d S )N�Lr< rD g ��.A���r )!�utmpr{ rR � transport�getPeer�hostrV �unpack�socket� inet_atonr~ �time�int� UtmpEntry� USER_PROCESS� DEAD_PROCESS�ut_typer} �pid�ut_pid�ut_line�ut_id�ut_tvr �ut_user�
gethostbyaddr�ut_host�
ut_addr_v6�
UtmpRecord� UTMP_FILE� pututline�endutent� WTMP_FILE) r �loggedIn� ipAddress�packedIp�ttyName�t�t1�t2�entry�a�br! r! r" � addUTMPEntry� s0
z'SSHSessionForUnixConchUser.addUTMPEntryc C sF || j d< || _|| _t�� \}}t�|�}|| j d< |||f| _d S )N�TERM�SSH_TTY)r| �winSize�modesr} �openptyre �ttynamer~ )r �term�
windowSizer� �master�slaver� r! r! r" �getPty� s
z!SSHSessionForUnixConchUser.getPtyc
C sB | j s
| j�d� td��| j�� \}}| j�� }| j�� }| jj| j d< || j d< || j d< t
j � |�}| jj
jj�� }| jj
jj�� }|j� d|j� d|j� �| j d< | �� | jj||d|� �g| j |||| j d �| _| �� t�| j�� tj t j d
g| j �R � � | j r�| �!� |jj"| _#| j$|j_"| jj
jj�%d � d S ) Nz'tried to get shell without pty, failingzno pty�USER�HOME�SHELL� �
SSH_CLIENT�-��usePTY�4HrL )&r~ rb �errorr r{ r@ rE rH r r| re �path�basenamerR r� r� rU r� �port�getPtyOwnershiprz � spawnProcessr} r� �fcntl�ioctl�fileno�tty�
TIOCSWINSZrV rW r� r� �setModes�write�oldWrite�
_writeHack�
setTcpNoDelay) r �protort r: �homeDir�shell� shellExec�peerr� r! r! r" � openShell� s>
�
&
z$SSHSessionForUnixConchUser.openShellc
C s� | j �� \}}| j �� }| j �� pd}|| jd< |d|f}| j jjj�� }| j jjj�� } |j � d|j
� d| j
� �| jd< | j rD| � � | j
j|||| j|||| j pSdd�| _| j re| �� | jre| �� | j jjj�d� d S ) Nz/bin/shr� z-cr� r� r r� rL )r{ r@ rE rH r| rR r� r� rU r� r� r~ r� rz r� r} r� r� r� r� )
r r� �cmdrt r: r� r� �commandr� r� r! r! r" � execCommand� s2
�
z&SSHSessionForUnixConchUser.execCommandc C s� t �| jd �d }| j�� \}}t �� t �� }}t �d� t �d� zt � | jd ||� W t �|� t �|� d S t �|� t �|� w )Nr< rD r )
re �statr~ r{ r@ rf rg ri rj �chown)r �ttyGidrt r: rq rr r! r! r" r� s
� z*SSHSessionForUnixConchUser.getPtyOwnershipc C s | j }t�|�� �}| jD ]n\}}|tjvrq
tj| }t|�dkrF|\}}tt|�s,q
t t|�}|r<|| |B ||< q
|| | @ ||< q
|dkrVt td|� ��|tj
< q
|dkrft td|� ��|tj < q
tt|�slq
t t|�}t |f�|tj
|< q
t�|�� tj|� d S )Nr< �OSPEED�B�ISPEED)r} r� � tcgetattrr� r� r �TTYMODESrd �hasattr�getattrr� r� �bytes�CC� tcsetattr�TCSANOW) r r} �attr�mode� modeValue�ttyMode�flag�ttyAttr�ttyvalr! r! r" r� s.
z#SSHSessionForUnixConchUser.setModesc C s | j r
| j �� d S d S rA )r} �
closeStdinr? r! r! r" � eofReceived8 s �z&SSHSessionForUnixConchUser.eofReceivedc C s� | j r tj�| j d �r t�| j d �d }t�| j d d|� | jrAz| j�d� W n tt fy6 Y nw | j�
� | � d� | j �
d� d S )Nr< rD r �HUPz shell closed)r~ re r� �existsr� r� r} �
signalProcess�OSErrorr �loseConnectionr� rb rc )r �ttyGIDr! r! r" �closed< s �
z!SSHSessionForUnixConchUser.closedc C s0 || _ t�| j�� tjtjdg| j �R � � d S )Nr� ) r� r� r� r} r� r� r� rV rW )r r� r! r! r" �
windowChangedI s *z(SSHSessionForUnixConchUser.windowChangedc C s\ | j dur't�| j �� �d }|tj@ s'|tj@ r'| jjj� ddt
|� � | � |� dS )zF
Hack to send ignore messages when we aren't echoing.
Nr' � � ) r} r� r� r� �ECHO�ICANONr{ rR r� �
sendIgnorerd r� )r rX r� r! r! r" r� M s
z%SSHSessionForUnixConchUser._writeHackrA )rL )r$ r% r& r rb r( r� r� r� r� r� r� r� r� r� r� r! r! r! r" rx � s
!
rx 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 rA )r{ )r r{ r! r! r" r( Z rF z#SFTPServerForUnixConchUser.__init__c C sr d|v rd|v rt �||d |d � d|v r t �||d � d|v r5d|v r7t �||d |d f� dS dS dS )zl
NOTE: this function assumes it runs as the logged-in user:
i.e. under _runAsUser()
rt r: � permissions�atime�mtimeN)re 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)�sizert r: r� r� r� )�st_size�st_uid�st_gid�st_moder� �st_atime�st_mtime�r �sr! r! r" � _getAttrsi s �z$SFTPServerForUnixConchUser._getAttrsc C s" | j �� }tj�t|j�t|��S rA )r{ rE re r� �joinr )r r� �homer! r! r" �_absPaths s
z#SFTPServerForUnixConchUser._absPathc C s i S rA r! )r � otherVersion�extDatar! r! r" �
gotVersionw � z%SFTPServerForUnixConchUser.gotVersionc C s t | | �|�||�S rA )� UnixSFTPFiler
)r �filename�flagsr� r! r! r" �openFilez � z#SFTPServerForUnixConchUser.openFilec C � | � |�}| j�tj|�S rA )r
r{ rO re �remove)r r r! r! r" �
removeFile} �
z%SFTPServerForUnixConchUser.removeFilec C s& | � |�}| � |�}| j�tj||�S rA )r
r{ rO re �rename)r �oldpath�newpathr! r! r" �
renameFile� �
z%SFTPServerForUnixConchUser.renameFilec C s, | � |�}| j�tj|ff| j||ffg�S rA )r
r{ rO re �mkdirr r� r! r! r" �
makeDirectory� s
�z(SFTPServerForUnixConchUser.makeDirectoryc C s | � |�}| j�tj|� d S rA )r
r{ rO re �rmdir�r r� r! r! r" �removeDirectory� s
z*SFTPServerForUnixConchUser.removeDirectoryc C s t | | �|��S rA )�UnixSFTPDirectoryr
r# r! r! r" �
openDirectory� s z(SFTPServerForUnixConchUser.openDirectoryc C s: | � |�}|r| j�tj|�}n| j�tj|�}| �|�S rA )r
r{ rO re r� �lstatr
)r r� � followLinksr r! r! r" �getAttrs� s
z#SFTPServerForUnixConchUser.getAttrsc C s | � |�}| j�| j||� d S rA )r
r{ rO r r� r! r! r" �setAttrs� s
z#SFTPServerForUnixConchUser.setAttrsc C r rA )r
r{ rO re �readlinkr# r! r! r" �readLink� r z#SFTPServerForUnixConchUser.readLinkc C s& | � |�}| � |�}| j�tj||�S rA )r
r{ rO re �symlink)r �linkPath�
targetPathr! r! r" �makeLink� r z#SFTPServerForUnixConchUser.makeLinkc C s t j�| �|��S rA )re r� �realpathr
r# r! r! r" �realPath� s z#SFTPServerForUnixConchUser.realPathc C � t �rA ��NotImplementedError)r �extNamer r! r! r" �extendedRequest� r z*SFTPServerForUnixConchUser.extendedRequestN)r$ r% r& r( r r
r
r r r r r! r$ r& r) r* r, r0 r2 r7 r! r! r! r" r� X s$
r� c @ s< e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
S )r c C s || _ d}|t@ tkr|t@ dkrtj}|t@ tkr#|t@ dkr#tj}|t@ tkr2|t@ tkr2tj}|t@ tkr=|tjO }|t @ t krH|tj
O }|t @ t krS|tj O }|t
@ t
kr^|tjO }d|v rj|d }|d= nd}|j�tj|||�}|r�|j�|j||� || _d S )Nr r� i� )�serverr r re �O_RDONLY�O_WRONLY�O_RDWRr
�O_APPENDr �O_CREATr �O_TRUNCr �O_EXCLr{ rO �openr �fd)r r8 r r r� � openFlagsr� rA r! r! r" r( � s0
zUnixSFTPFile.__init__c C s | j j�tj| j�S rA )r8 r{ rO re �closerA r? r! r! r" rC � r zUnixSFTPFile.closec C �, | j j�tj| j|dfftj| j|ffg�S r )r8 r{ rO re �lseekrA �read)r �offset�lengthr! r! r" � readChunk� � �zUnixSFTPFile.readChunkc C rD r )r8 r{ rO re rE rA r� )r rG rX r! r! r" �
writeChunk� rJ zUnixSFTPFile.writeChunkc C s | j j�tj| j�}| j �|�S rA )r8 r{ rO re �fstatrA r
r r! r! r" r) � s zUnixSFTPFile.getAttrsc C r3 rA r4 )r r� r! r! r" r* � r zUnixSFTPFile.setAttrsN) r$ r% r& r( rC rI rK 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 rA )r8 r{ rO re �listdir�files�dir)r r8 � directoryr! r! r" r( � s
zUnixSFTPDirectory.__init__c C s | S rA r! r? r! r! r" �__iter__� r zUnixSFTPDirectory.__iter__c C sb z| j �d�}W n ty t�w | jj�tjtj �
| j |��}t ||�}| j�
|�}|||fS r )rN �pop�
IndexError�
StopIterationr8 r{ rO re r' r� r rO r r
)r rn r �longnamer� r! r! r" �__next__� s �
zUnixSFTPDirectory.__next__c C s
g | _ d S rA )rN r? r! r! r" rC � rF zUnixSFTPDirectory.closeN)r$ r% r& r( rQ rV �nextrC r! r! r! r" r% � s r% )7�__doc__r� r, re r} r) r� rV r� r� �zope.interfacer �
twisted.conchr �twisted.conch.avatarr �twisted.conch.errorr �twisted.conch.interfacesr r r �twisted.conch.lsr �twisted.conch.sshr
r r � twisted.conch.ssh.filetransferr
r r r r r � twisted.credr �twisted.internet.errorr �twisted.loggerr �twisted.pythonr �twisted.python.compatr r� � ImportError�IRealmr r rx r� r r% �registerAdapterr! r! r! r" �<module> sV �i :V1�