shell bypass 403
o
` b�* � @ s d Z ddlZddlZddlZG dd� d�Zdd� Zdd� Zd d
� Zd d � Zej d
d� �Z
dd� Z dd� Z dd� Z
dd� Zdd� Zdd� Zdd � Zd d � Zd d � Zd!d"� Zd#d$� Zd%d&� Zd'd(� Zd)d*� Zd+d,� Zd-d.� Zed/kr�dd0l m Z e � ej d1 Z e � e � dS dS )2z�This script contains the actual auditing tests.
It should not be imported directly, but should be run by the test_audit
module with arguments identifying each test.
� Nc @ sJ e Zd ZdZdefdd�Zdd� Zdd� Zd d
� Ze d d � �Z
d
d� Z dS )�TestHookz�Used in standard hook tests to collect any logged events.
Should be used in a with block to ensure that it has no impact
after the test completes.
Nc C s |pd| _ || _g | _d| _d S )N� F)�raise_on_events�exc_type�seen�closed)�selfr r r r �9/usr/local/python-3.10/lib/python3.10/test/audit-tests.py�__init__ s
zTestHook.__init__c G s t �| � | S �N)�sys� addaudithook�r �ar r r � __enter__ s
zTestHook.__enter__c G s | � � d S r )�closer r r r �__exit__ s zTestHook.__exit__c C s
d| _ d S )NT)r �r r r r r ! s
zTestHook.closec C s dd� | j D �S )Nc S s g | ]}|d �qS )r r ��.0�ir r r �
<listcomp>& s z(TestHook.seen_events.<locals>.<listcomp>)r r r r r � seen_events$ s zTestHook.seen_eventsc C s6 | j rd S | j�||f� || jv r| �d| ��d S )Nz
saw event )r r �appendr r )r �event�argsr r r �__call__( s
�zTestHook.__call__) �__name__�
__module__� __qualname__�__doc__� RuntimeErrorr
r r r �propertyr r r r r r r
s
r c C s | |kr
t | �d|����d S )N� should equal ��AssertionError��x�yr r r � assertEqual1 � �r) c C s | |vr
t | �d|����d S )Nz should be in r$ ��elZseriesr r r �assertIn6 r* r- c C s | |v r
t | �d|����d S )Nz should not be in r$ r+ r r r � assertNotIn; r* r. c C sP t | �t |�krt| �d|����tdd� t| |�D ��r&t| �d|����d S )Nr# c s s � | ] \}}||kV qd S r r )r ZixZiyr r r � <genexpr>C s � z&assertSequenceEqual.<locals>.<genexpr>)�lenr% �any�zipr& r r r �assertSequenceEqual@ s
�r3 c
c sh � z
d V J d| � ��� t y3 } z t|t�r� t|�| u s(J |� d| � ���W Y d }~d S d }~ww )NFz expected z should be )�
BaseException�
isinstancer% �type)Zex_type�exr r r � assertRaisesG s �
,��r8 c C sd t � �%} t�dddd� t| jd d d� t| jd d d� W d � d S 1 s+w Y d S )N�
test_event� � � r )r: r; r< )r r �auditr) r ��hookr r r �
test_basicR s
"�r@ c C s� t dd��5} t � �}t�d� td| j� td|j� W d � n1 s%w Y W d � d S W d � d S 1 s=w Y d S )N�sys.addaudithook�r r9 )r r r= r- r r. �Zhook1Zhook2r r r �test_block_add_hookY s
��"�rD c
C s� t t��: tdtd��} t� �}W d � n1 sw Y W d � n1 s)w Y W d � d S W d � d S 1 sAw Y d S )NrA )r r )r8 r4 r rC r r r �!test_block_add_hook_baseexceptionc s
� �� ��"�rE c
C s@ dd l } d}| �|�}t� �Y}t|| �| �|��� z?tdd��}| �||� W d � n1 s1w Y tdd��}t|| �|�� W d � n1 sNw Y W t� d� nt� d� w W d � n1 sjw Y dd� |j
D �}t ||| j fgd � d d� |j
D �}t ||g� d
d� |j
D �}t |d g� d S ) Nr �r �b�cr: r; r< ztest-marshal.bin�wb�rbc S �( g | ]\}}|d kr|d |d f�qS )z
marshal.dumpsr r: r �r �er r r r r ~ � ( z test_marshal.<locals>.<listcomp>r; c S s g | ] \}}|d kr|d �qS )z
marshal.loadsr r rL r r r r � s c S s g | ]
\}}|d kr|�qS )� marshal.loadr rL r r r r � s rO )
�marshal�dumpsr r) �loads�open�dump�load�os�unlinkr r3 �version)rP �o�payloadr? �f�actualr r r � test_marshaln s*
� �� � r] c C s� dd l } G dd� d�}| �|� �}| �d�}td| �|�� tdd��'}tt��
| �|� W d � n1 s8w Y | �|� W d � d S 1 sMw Y d S )Nr c @ s e Zd Zdd� ZdS )z test_pickle.<locals>.PicklePrintc S s t dfS )N)�Pwned!)�str)r �pr r r �
__reduce_ex__� s z.test_pickle.<locals>.PicklePrint.__reduce_ex__N)r r r ra r r r r � PicklePrint� s rb rF r^ zpickle.find_classrB )�picklerQ r) rR r r8 r! )rc rb Z payload_1Z payload_2r? r r r � test_pickle� s
� "�rd c C s� G dd� d�} G dd� d�}G dd� d| �}| � }t � �#}d|_|f|_tjd �||f� |j|_d |_||_W d � n1 sBw Y d
d � |j D �}t
|d f|df|df|d
fg|� d S )Nc @ � e Zd ZdS )ztest_monkeypatch.<locals>.AN�r r r r r r r �A� � rg c @ re )ztest_monkeypatch.<locals>.BNrf r r r r �B� rh ri c @ re )ztest_monkeypatch.<locals>.CNrf r r r r �C� rh rj �X� __bases__�{ c S rK )zobject.__setattr__r r: r rL r r r r � rN z$test_monkeypatch.<locals>.<listcomp>r � __class__) r r rl r6 �__dict__�__set__r
Znew_attrrn r r3 )rg ri rj r r? r\ r r r �test_monkeypatch� s � �rq c C s` z dd l } | �� j}W n ty d }Y nw tdhd��O}ttjd dfttjdftddfttjd d d
d d d d d d
� f |tjd ffD ] ^}}|sLqEt t
�� ||� W d � n1 s_w Y qEW d � n1 sow Y dd� |j D �}dd� |j D �}t dd� tjd dftjdfdtjd d f|r�tjd dfnd fD �|� t g |� d S )Nr rS rB r; �rrJ r< rI �w���Fc W s dS )Nr: r )r r r r �<lambda>� s ztest_open.<locals>.<lambda>c S s0 g | ]\}}|d kr|d r|d |d f�qS )rS r: r r rL r r r r � � 0 z test_open.<locals>.<listcomp>c S s0 g | ]\}}|d kr|d s|d |d f�qS )rS r: r r; r rL r r r r � rv c S s g | ]}|d ur|�qS r r r r r r r � s ��)r< rs )
�sslZcreate_default_context�load_dh_params� ImportErrorr rS r �argv�
executabler8 r! r r3 )rw rx r? �fnr Z actual_modeZ actual_flagr r r � test_open� sF �
�
��� �
�� �r} c s� g � � fdd�} t �| �}z6t� �$}td� d|_td� d|_td� d|_td � d
|_W d � n1 s8w Y W t �|� nt �|� w td gd � � d S )
Nc s | j tjjkr� �|� d S d S r )�f_coder r �__code__r )�framer r �Ztracedr r �trace� s �z test_cantrace.<locals>.trace�1F�2T�3r: �4r �call� )r �settracer �evalZ __cantrace__r3 )r� �oldr? r r� r �
test_cantrace� s"
��r� c C s\ dd l } t� � }| � dd� t|jd d d d� d� W d � d S 1 s'w Y d S )Nr rt � r: r; )rt r� )�mmapr r) r )r� r? r r r � test_mmap s
"�r� c s, dd� � � fdd�} t �| � � t _td��)Nc S s | t ur
t�| ||� d S d S r )r! r �__excepthook__)r � exc_value�exc_tbr r r �
excepthook r* z#test_excepthook.<locals>.excepthookc sv | dkr9t |d |d �s td|d �d|d �d���|d � kr.td|d � d � � ���t| t|d �� d S d S )
Nzsys.excepthookr; r: zExpected isinstance(z, �)r � Expected � == )r5 � TypeError�
ValueError�print�repr�r r �r� r r r? s �z test_excepthook.<locals>.hookz fatal-error)r r
r� r! r> r r� r �test_excepthook s
r� c sD ddl m} dd� � � fdd�}t�|� � t_| td�dd � d S ) Nr )�write_unraisable_excc S s d S r r )Zhookargsr r r �unraisablehook& s z+test_unraisablehook.<locals>.unraisablehookc sP | dkr&|d � krt d|d � d� � ���t| t|d j�|d j� d S d S )Nzsys.unraisablehookr r� r� r: )r� r� r� r� �err_msgr� �r� r r r? ) s
�z!test_unraisablehook.<locals>.hookznonfatal-errorzfor audit hook test)Z _testcapir� r r
r� r! )r� r? r r� r �test_unraisablehook# s
r� c C sz ddl m} m}m}m} dd� }t�|� | |d�}||d� z
||d� W td�� ty2 Y nw |� � }||� d S )Nr )�OpenKey�EnumKey�CloseKey�HKEY_LOCAL_MACHINEc S s" | � d�sd S t| g|�R � d S )Nzwinreg.��
startswithr� r� r r r r? 7 s
ztest_winreg.<locals>.hookZSoftwarei' z%Expected EnumKey(HKLM, 10000) to fail)
�winregr� r� r� r� r r
�OSErrorr! �Detach)r� r� r� r� r? �kZkvr r r � test_winreg4 s
�� r� c C s| dd l } dd� }t�|� | �� | � | j| j�}z z|�d� W n ty* Y nw W |�� d S W |�� d S |�� w )Nr c S �" | � d�rt| g|�R � d S d S )Nzsocket.r� r� r r r r? N �
�ztest_socket.<locals>.hook)z 127.0.0.1i� ) �socketr r
� gethostname�AF_INET� SOCK_STREAM�bind� Exceptionr )r� r? �sockr r r � test_socketK s
�� �r� c C sJ dd l } dd� }t�|� | jdd� t� }|g}| �|� | �|� d S )Nr c S r� )Nzgc.r� r� r r r r? d r� ztest_gc.<locals>.hookr: )Z
generation)�gcr r
Z get_objects�objectZ
get_referrersZ
get_referents)r� r? r' r( r r r �test_gca s
r� c C s| dd l } dd� }t�|� | j�d�}z%z|�dd� W n ty* tdd� Y nw W |�� d S W |�� d S |�� w ) Nr c S s* | � d�rt| g|dd � �R � d S d S )Nz http.client.r: r� r� r r r r? v s
�z test_http_client.<locals>.hookzwww.python.org�GET�/zhttp.client.sendz
[cannot send]) Z http.clientr r
�clientZHTTPConnectionZrequestr� r� r )�httpr? �connr r r �test_http_clients s
�� �r� c C st dd l } dd� }t�|� | �d�}| �d�}t| jd�r8|�d� z
|�d� W t d�� | jy7 Y d S w d S ) Nr c W r� )Nzsqlite3.r� r� r r r r? � r� ztest_sqlite3.<locals>.hookz:memory:�enable_load_extensionF�testz'Expected sqlite3.load_extension to fail)
�sqlite3r r
�connect�
Connection�hasattrr� Zload_extensionZOperationalErrorr! )r� r? Zcx1Zcx2r r r � test_sqlite3� s
���r� �__main__)�suppress_msvcrt_assertsr: )!r �
contextlibrV r r r) r- r. r3 �contextmanagerr8 r@ rD rE r] rd rq r} r� r� r� r� r� r� r� r� r� r Z test.supportr� rz r� �globalsr r r r �<module> sB $
+!
�