Your IP : 216.73.216.85
U
i�fT$ � @ s� d Z ddlZddlZG dd� d�Zdd� Zdd� Zd d
� Zd d � Zejd
d� �Z dd� Z
dd� Z dd� Z dd� Z
dd� Zdd� Zdd � Zd d � Zd d � Zd!d"� Zd#d$� Zd%d&� Zd'd(� Zd)d*� Zed+kr�dd,lmZ e� ejd- Z e � e � dS ).z�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 �5/opt/alt/python38/lib64/python3.8/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 r
d S | j�||f� || jkr2| �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 | |krt | �d|����d S )N� should equal ��AssertionError��x�yr r r � assertEqual0 s r) c C s | |krt | �d|����d S )Nz should be in r$ �ZelZseriesr r r �assertIn5 s r+ c C s | |krt | �d|����d S )Nz should not be in r$ r* r r r � assertNotIn: s r, c C sP t | �t |�kr"t| �d|����tdd� t| |�D ��rLt| �d|����d S )Nr# c s s | ]\}}||kV qd S r r )r ZixZiyr r r � <genexpr>B s z&assertSequenceEqual.<locals>.<genexpr>)�lenr% �any�zipr& r r r �assertSequenceEqual? s r1 c
c s> z
d V W n. t k
r8 } zt|t�r(� W 5 d }~X Y nX d S r )�
BaseException�
isinstancer% )Zex_typeZexr r r � assertRaisesF s
r4 c C sN t � �>} t�dddd� t| jd d d� t| jd d d� W 5 Q R X d S )N�
test_event� � � r )r6 r7 r8 )r r �auditr) r ��hookr r r �
test_basicQ s r< c
C sN t dd��:} t � �(}t�d� td| j� td|j� W 5 Q R X W 5 Q R X d S )N�sys.addaudithook�r r5 )r r r9 r+ r r, �Zhook1Zhook2r r r �test_block_add_hookX s
r@ c C sB t t��0 tdtd��} t� �}W 5 Q R X W 5 Q R X W 5 Q R X d S )Nr= )r r )r4 r2 r r? r r r �!test_block_add_hook_baseexceptionb s
�rA c
C s~ dd l } G dd� d�}| �|� �}| �d�}td| �|�� tdd��.}tt�� | �|� W 5 Q R X | �|� W 5 Q R X 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__q s z.test_pickle.<locals>.PicklePrint.__reduce_ex__N)r r r rE r r r r � PicklePrintp s rF )r �b�cr6 r7 r8 rB zpickle.find_classr> )�pickle�dumpsr) �loadsr r4 r! )rI rF Z payload_1Z payload_2r; r r r � test_picklem s
rL c C s� G dd� d�} G dd� d�}G dd� d| �}| � }t � �<}d|_|f|_tjd �||f� |j|_d |_||_W 5 Q R X d
d � |j D �}t
|d f|df|df|d
fg|� d S )Nc @ s e Zd ZdS )ztest_monkeypatch.<locals>.AN�r r r r r r r �A� s rN c @ s e Zd ZdS )ztest_monkeypatch.<locals>.BNrM r r r r �B� s rO c @ s e Zd ZdS )ztest_monkeypatch.<locals>.CNrM r r r r �C� s rP �X� __bases__�{ c S s( g | ] \}}|d kr|d |d f�qS )zobject.__setattr__r r6 r �r �er r r r r � s z$test_monkeypatch.<locals>.<listcomp>r � __class__) r r rR �type�__dict__�__set__r
Znew_attrrV r r1 )rN rO rP r r; Zactualr r r �test_monkeypatch� s �rZ c C s< zdd l } | �� j}W n tk
r. d }Y nX tdhd���}ttjd dfttjdftddfttjd d d
d d d d d d
� f |tjd ffD ]*^}}|s�q�t t
�� ||� W 5 Q R X q�W 5 Q R X dd� |j D �}dd� |j D �}t dd� tjd dftjdfdtjd d f|�r tjd dfnd fD �|� t g |� d S )Nr �openr> r7 �r�rbr8 �wb�w���Fc W s dS )Nr6 r �r r r r �<lambda>� � ztest_open.<locals>.<lambda>c S s0 g | ](\}}|d kr|d r|d |d f�qS )r[ r6 r r rT r r r r � s z test_open.<locals>.<listcomp>c S s0 g | ](\}}|d kr|d s|d |d f�qS )r[ r6 r r7 r rT r r r r � s c S s g | ]}|d k r|�qS r r r r r r r � s �)r8 r_ )
�sslZcreate_default_context�load_dh_params� ImportErrorr r[ r �argv�
executabler4 r! r r1 )rd re r; �fnr Z actual_modeZ actual_flagr r r � test_open� s<
�
�� �rj c s� g � � fdd�} t �| �}zNt� �>}td� d|_td� d|_td� d|_td � d
|_W 5 Q R X W 5 t �|� X td gd � � d S )
Nc s | j tjjkr� �|� d S r )�f_coder r �__code__r )�framer r �Ztracedr r �trace� s z test_cantrace.<locals>.trace�1F�2T�3r6 �4r Zcall� )r �settracer �evalZ __cantrace__r1 )ro �oldr; r rn r �
test_cantrace� s
rx c C sF dd l } t� �.}| � dd� t|jd d d d� d� W 5 Q R X d S )Nr r` � r6 r7 )r` ry )�mmapr r) r )rz r; r r r � test_mmap� s r{ c s0 dd� � � fdd�} t �| � � t _td��d S )Nc S s | t k rt�| ||� d S r )r! r �__excepthook__)r � exc_valueZexc_tbr r r �
excepthook� s z#test_excepthook.<locals>.excepthookc sr | dkrnt |d |d �s8td|d �d|d �d���|d � kr\td|d � d � � ���t| t|d �� d S )
Nzsys.excepthookr7 r6 zExpected isinstance(z, �)r � Expected � == )r3 � 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 sL | dkrH|d � kr,t d|d � d� � ���t| t|d j�|d j� d S )Nzsys.unraisablehookr r� r� r6 )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 s~ ddl m} m}m}m} dd� }t�|� | |d�}||d� z||d� W n tk
r` Y n
X td��|� � }||� d S )Nr )�OpenKey�EnumKey�CloseKey�HKEY_LOCAL_MACHINEc S s | � d�sd S t| f|�� d S )Nzwinreg.��
startswithr� r� r r r r; s
ztest_winreg.<locals>.hookZSoftwarei' z%Expected EnumKey(HKLM, 10000) to fail)
�winregr� r� r� r� r r
�OSErrorr! ZDetach)r� r� r� r� r; �kZkvr r r � test_winreg s
r� c C sj dd l } dd� }t�|� | �� | � | j| j�}z(z|�d� W n tk
rV Y nX W 5 |�� X d S )Nr c S s | � d�rt| f|�� d S )Nzsocket.r� r� r r r r; 3 s
ztest_socket.<locals>.hook)z 127.0.0.1i� ) �socketr r
Z gethostnameZAF_INETZ SOCK_STREAMr Zbind� Exception)r� r; Zsockr r r � test_socket0 s
r� c C sJ dd l } dd� }t�|� | jdd� t� }|g}| �|� | �|� d S )Nr c S s | � d�rt| f|�� d S )Nzgc.r� r� r r r r; I s
ztest_gc.<locals>.hookr6 )Z
generation)�gcr r
� get_objects�objectZ
get_referrersZ
get_referents)r� r; r' r( r r r �test_gcF s
r� c C s6 dd l } dd� }t�|� | �� D ]}t|t�r"q"d S )Nr c W s d S r r ra r r r rb [ rc z test_not_in_gc.<locals>.<lambda>)r� r r
r� r3 �list)r� r; �or r r �test_not_in_gcX s
r� �__main__)�suppress_msvcrt_assertsr6 ) r �
contextlibr r r) r+ r, r1 �contextmanagerr4 r< r@ rA rL rZ rj rx r{ r� r� r� r� r� r� r Z test.supportr� rg Ztest�globalsr r r r �<module> s8 $
+!