← Back
Editing: 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 � �
Save File
Cancel