shell bypass 403
o
a biC � @ sJ d dl mZmZ d dlmZ ed� d dlZd dlZd dlZd dlZd dl Z d dl
Z
d dl Z d dl Z d dl
Z
d dlZd dlZd dlZd dlZd dlZdZdZz
ejZejZdZW n eyc dZY nw erkd d
� Znd d
� Zd d
� Zdd� Z dd� Z dd� Z dd� Z G dd� dej �Z!G dd� dej �Z"dd� Z#e$d kr�e�%� dS dS ) � )�verbose�
reap_children)�
import_moduleZtermiosN� I wish to buy a fish license.
� For my pet fish, Eric.
TFc C s t | � d S �N)�print��msg� r �6/usr/local/python-3.10/lib/python3.10/test/test_pty.py�debug" s r
c C � d S r r r r r r r
% s c C s0 | � d�r | �dd�S | � d�r| �dd�S | S )Ns
�
�
)�endswith�replace)�datar r r �normalize_output3 s
r c C s t j| ddd�}|�� S )z8Read one line. May block forever if no newline is read.�rbF)�mode�closefd)�io�FileIO�readline)�fd�readerr r r � _readlineG s r c C s0 z t �tj� t�| �W S t jy Y | S w r )�tty� tcgetattr�pty� STDIN_FILENO�unittestZexpectedFailure�error)Zfunr r r �expectedFailureIfStdinIsTTYL s �r$ c C s t �ddddd�}t�| t|�S )N�HHHHr )�struct�pack�fcntl�ioctl� _TIOCGWINSZ)r �sr r r �_get_term_winszU s r, c C s t �| t|� d S r )r( r) � _TIOCSWINSZ)r Zwinszr r r �_set_term_winszY s r. c @ sL e Zd Zdd� Zdd� Zedd� �Zedd� �Zd d
� Z d d � Z
d
d� Z dS )�PtyTestc C s� t � t j| j�}| �t j t j|� t � t j| j�}| �t j t j|� | �t jd� t �d� d | _d | _t rez$t
� t j
�}|j| _|j| _t�d| j| jdd�}| �tt j
|� W d S tyd Y d S w d S )Nr �
r% )�signal�SIGALRM�
handle_sig�
addCleanup�SIGHUP�
handle_sighup�alarm�
stdin_rows�
stdin_cols� _HAVE_WINSZ�os�get_terminal_sizer r! �lines�columnsr& r' r. �OSError)�selfZ old_alarmZ
old_sighupZ stdin_dimZold_stdin_winszr r r �setUp` s*
� ��z
PtyTest.setUpc C s | � d� d S )Nz isatty hung)�fail)r@ �sig�framer r r r3 z s zPtyTest.handle_sigc C r r r )�signumrD r r r r6 } s zPtyTest.handle_sighupc
C s� zt �tj�}W n t jy td� d }Y nw d }| jd urw| jd urwzCtd� td| j� d| j� d�� | jd }| jd }td|� d|� d�� t� d||d d �}t
tj|� t tj�}| � ||d
� W n t
yv t�d � Y nw z+td � z
t�||�\}}W n ty� t�� \}}Y nw td
|� d|� d�� W n t
y� t�d��w | �tj|� | �tj|� | �t�|�d� |r�| � t �|�|d� |r�| � t |�|d� t�|�}z8t�|d� zt�|d�} | � d| � W n t
�y }
z|
j t j k�r� W Y d }
~
nd }
~
ww W t�||� nt�||� w td� t� |t � t |�} | � dt!| �� td� t� |t"d d� � t� |t"dd � � t |�} | � dt!| �� d S ) Nz&tty.tcgetattr(pty.STDIN_FILENO) failedz$Setting pty.STDIN_FILENO window sizezoriginal size: (rows=z, cols=�)� ztarget size: (rows=r% r z&pty.STDIN_FILENO window size unchangedz*Failed to set pty.STDIN_FILENO window size�Calling pty.openpty()�Got master_fd '�
', slave_fd '�'z,Pseudo-terminals (seemingly) not functional.zslave_fd is not a ttyz%openpty() failed to set slave termiosz)openpty() failed to set slave window sizeFi � zWriting to slave_fdr zWriting chunked output� r )#r r r r! r# r
r8 r9 r& r' r. r, � assertEqualr? �warnings�warn�openpty� TypeErrorr" ZSkipTestr4 r; �closeZ
assertTrue�isatty� get_blocking� set_blocking�read�errno�EAGAIN�write�
TEST_STRING_1r r �
TEST_STRING_2) r@ r Znew_stdin_winszZtarget_stdin_rowsZtarget_stdin_colsZtarget_stdin_winsz� master_fd�slave_fd�blocking�s1�e�s2r r r � test_openpty� s� �
�
�
� �
�� �
���� �zPtyTest.test_openptyc C s� t d� t�� \}}| �tj|� |tjkret�d�s$t d� t�d� t d� zt� � W n& t
y; t d� Y n# t yG t d� Y n t d� t�d� Y n t d � t�d
� t�d � d S t d | � zt� |d�}W n t
y} Y q�w |s�q�t
j�t|�dd�dd�� qlt�|d�\}}t�|�}t d|||f � |dkr�| �d� d S |d
kr�| �d� d S |dkr�| �d� d S |d kr�| �d� d S d S )Nzcalling pty.fork()rG zChild's fd 1 is not a tty?!� z In child, calling os.setsid()zGood: OSError was raised.zNo setsid() available?z An unexpected error was raised.z os.setsid() succeeded! (bad!)� � z!Waiting for child (%d) to finish.T�P r r �ascii)�encodingr z+Child (%d) exited with code %d (status %d).z3Child raised an unexpected exception in os.setsid()z1pty.fork() failed to make child a session leader.z8Child spawned by pty.fork() did not have a tty as stdoutz&pty.fork() failed for unknown reasons.)r
r �forkr4 r; rS ZCHILDrT �_exit�setsidr? �AttributeErrorrW �sys�stdoutrZ �strr �waitpid�waitstatus_to_exitcoderB )r@ �pidr] r �status�resr r r � test_fork� s^
���
�zPtyTest.test_forkc C s� t d� t�� \}}t d|� d|� d�� | �tj|� t d� t�|� t d� zt�|d�}W n ty< d}Y nw | �|d� d S ) NrH rI rJ rK zClosing slave_fdzReading from master_fdrG rL ) r
r rQ r4 r; rS rW r? rN )r@ r] r^ r r r r �test_master_read( s
�zPtyTest.test_master_readc C s t �tjddg� d S )Nz-czprint("hi there"))r �spawnrn �
executable�r@ r r r �test_spawn_doesnt_hang; s z PtyTest.test_spawn_doesnt_hangN) �__name__�
__module__� __qualname__rA r3 � staticmethodr6 r$ rc rv rw r{ r r r r r/ _ s
TR r/ c @ sb e Zd ZdZdd� Zdd� Zdd� Zdd � Zdd d �Zd
d� Z dd� Z
dd� Z dd� Z dd� Z
dS )�
SmallPtyTestsz)These tests don't spawn children or hang.c C sr t j| _t j| _t j| _t j| _t j | _
t j | _ t j
| _t j| _t j| _t j| _g | _g | _g | _g | _d | _d S r )r r! �orig_stdin_fileno�
STDOUT_FILENO�orig_stdout_filenorS �orig_pty_close�_copy�orig_pty__copyrj �
orig_pty_fork�select�orig_pty_select�setraw�orig_pty_setrawr �orig_pty_tcgetattr� tcsetattr�orig_pty_tcsetattrrq �orig_pty_waitpid�fds�files�select_rfds_lengths�select_rfds_results�tcsetattr_mode_settingrz r r r rA A s
zSmallPtyTests.setUpc C s� | j t_| jt_| jt_| jt_| j t_
| j t_ | j
t_| jt_| jt_| jt_| jD ]}z|�� W q+ ty< Y q+w | jD ]}zt�|� W q@ tyR Y q@w d S r )r� r r! r� r� r� rS r� r� r� rj r� r� r� r� r� r r� r� r� rq r� r? r� r; )r@ �filer r r r �tearDownR s.
�
��zSmallPtyTests.tearDownc C � t �� }| j�|� |S r )r; �piper� �extend)r@ Zpipe_fdsr r r �_pipeh � zSmallPtyTests._pipec C r� r )�socket�
socketpairr� r� )r@ r� r r r � _socketpairm r� zSmallPtyTests._socketpairr c C s* | � | j�d�t|�� | j�d�g g fS )Nr )rN r� �pop�lenr� )r@ �rfdsZwfdsZxfds�timeoutr r r � _mock_selectr s zSmallPtyTests._mock_selectc s � fdd�}|S )Nc s � dfS )N� r r �rs r r � mock_forky s z0SmallPtyTests._make_mock_fork.<locals>.mock_forkr )r@ rs r� r r� r �_make_mock_forkx s z SmallPtyTests._make_mock_forkc C s
|| _ d S r )r� )r@ �fileno�optr r r r �_mock_tcsetattr} �
z SmallPtyTests._mock_tcsetattrc C s" | � � \}}|t_| � � \}}|t_| �� }dd� |D �}t�|d d� t�|d� | jt_| j �
d� | j �
||d g� | j �
d� | � t
�� t�|d � W d � n1 s\w Y t�||d gg g d�d }| �||d g|� | �t�|d �d� | �t�|d d �d� dS )
z6Test the normal data case on both master_fd and stdin.c S � g | ]}|� � �qS r �r� ��.0r+ r r r �
<listcomp>� � z4SmallPtyTests.test__copy_to_each.<locals>.<listcomp>rG s from masters
from stdinre r N� )r� r r� r! r� r; rZ r� r� r� �appendr� Z assertRaises�
IndexErrorr� rN rW )r@ �read_from_stdout_fd�mock_stdout_fd�
mock_stdin_fd�write_to_stdin_fdr� �mastersr� r r r �test__copy_to_each� s& � z SmallPtyTests.test__copy_to_eachc C s� | � � \}}|t_| � � \}}|t_| �� }dd� |D �}|d �� t�|� | jt_| j �
d� | j �
||d g� | j �
d� | � t�
|d �d� dS )z9Test the empty read EOF case on both master_fd and stdin.c S r� r r� r� r r r r� � r� z7SmallPtyTests.test__copy_eof_on_all.<locals>.<listcomp>rG re r N)r� r r� r! r� rS r; r� r� r� r� r� rN r� )r@ r� r� r� r� r� r� r r r �test__copy_eof_on_all� s
z#SmallPtyTests.test__copy_eof_on_allc s� | � d�t_t� ��fdd�t_dd� t_dd� t_t� � � fdd�t_| jt_ dd� t_
| � t� g ��d� | � | j
� d � d
S ) zCTest that spawn resets the tty mode no when _copy returns normally.rG c s d � gS r r )�_1�_2)�status_sentinelr r �<lambda>� s zDSmallPtyTests.test__restore_tty_mode_normal_return.<locals>.<lambda>c S r r r ��_r r r r� � � c S r r r )r� r� Z_3r r r r� � r� c s � S r r )r )�
mode_sentinelr r r� � r� c S r r r r� r r r r� � r� z4pty.waitpid process status not returned by pty.spawnz1pty.tcsetattr not called with original mode valueN)r� r rj �objectrq rS r� r r� r� r� rN rx r� rz r )r� r� r �$test__restore_tty_mode_normal_return� s
z2SmallPtyTests.test__restore_tty_mode_normal_returnN)r )r| r} r~ �__doc__rA r� r� r� r� r� r� r� r� r� r r r r r� > s
r� c C s
t � d S r )r r r r r �tearDownModule� r� r� �__main__)&Z test.supportr r Ztest.support.import_helperr rX r; r r rn r� r1 r� r r" r& r( rO r[ r\ Z
TIOCGWINSZr* Z
TIOCSWINSZr- r: rm r
r r r$ r, r. ZTestCaser/ r� r� r| �mainr r r r �<module> sT �
` �