Sindbad~EG File Manager
a
��g�' � @ 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,� Zed-k�rdd.lm Z e � ej d/ Z e � e � dS )0z�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 �,/usr/local/lib/python3.9/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>& � 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� || jv r2| �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 � assertEqual1 s r* c C s | |vrt | �d|����d S )Nz should be in r% ��elZseriesr r r �assertIn6 s r- c C s | |v rt | �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>C r z&assertSequenceEqual.<locals>.<genexpr>)�lenr& �any�zipr' r r r �assertSequenceEqual@ s r3 c
c sp z d V dsJ d| � ���W nN t yj } z6t|t�r8� t|�| u sVJ |� d| � ���W Y d }~n
d }~0 0 d S )NFz expected z should be )�
BaseException�
isinstancer& �type)Zex_type�exr r r � assertRaisesG s
r8 c C sb t � �H} t�dddd� t| jd d d� t| jd d d� W d � n1 sT0 Y d S )N�
test_event� � � r )r: r; r< )r r �auditr* r ��hookr r r �
test_basicR s r@ c C sv t dd��X} t � �2}t�d� td| j� td|j� W d � n1 sJ0 Y W d � n1 sh0 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��b tdtd��6} t� �}W d � n1 s40 Y W d � n1 sR0 Y W d � n1 sp0 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� ��}t|| �| �|��� z~tdd�� }| �||� W d � n1 sb0 Y tdd�� }t|| �|�� W d � n1 s�0 Y W t� d� n t� d� 0 W d � n1 s�0 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 s( g | ] \}}|d kr|d |d f�qS )z
marshal.dumpsr r: r �r �er r r r r ~ r z test_marshal.<locals>.<listcomp>r; c S s g | ]\}}|d kr|d �qS )z
marshal.loadsr r rK r r r r � r c S s g | ]\}}|d kr|�qS )� marshal.loadr rK r r r r � r rM )
�marshal�dumpsr r* �loads�open�dump�load�os�unlinkr r3 �version)rN �o�payloadr? �f�actualr r r � test_marshaln s"
* 06 r[ c C s� dd l } G dd� d�}| �|� �}| �d�}td| �|�� tdd��L}tt�� | �|� W d � n1 sp0 Y | �|� W d � n1 s�0 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 r_ r r r r � PicklePrint� s r` rF r\ zpickle.find_classrB )�picklerO r* rP r r8 r" )ra r` Z payload_1Z payload_2r? r r r � test_pickle� s
(rb c C s� G dd� d�} G dd� d�}G dd� d| �}| � }t � �F}d|_|f|_tjd �||f� |j|_d |_||_W d � n1 s�0 Y 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 rd c @ s e Zd ZdS )ztest_monkeypatch.<locals>.BNrc r r r r �B� s re c @ s e Zd ZdS )ztest_monkeypatch.<locals>.CNrc r r r r �C� s rf �X� __bases__�{ c S s( g | ] \}}|d kr|d |d f�qS )zobject.__setattr__r r: r rK r r r r � r z$test_monkeypatch.<locals>.<listcomp>r � __class__) r r rh r6 �__dict__�__set__r
Znew_attrrj r r3 )rd re rf r r? rZ r r r �test_monkeypatch� s $ �rm c C sb zdd l } | �� j}W n ty, d }Y n0 tdhd���}ttjd dfttjdftddfttjd d d
d d d d d d
� f |tjd ffD ]>^}}|s�q�t t
�� ||� W d � q�1 s�0 Y q�W d � n1 s�0 Y dd� |j D �}dd� |j D �}t dd� tjd dftjdfdtjd d f|�rFtjd dfnd fD �|� t g |� d S )Nr rQ rB r; �rrJ r< rI �w���Fc W s dS )Nr: r �r r r r �<lambda>� r ztest_open.<locals>.<lambda>c S s0 g | ](\}}|d kr|d r|d |d f�qS )rQ r: r r rK r r r r � r z test_open.<locals>.<listcomp>c S s0 g | ](\}}|d kr|d s|d |d f�qS )rQ r: r r; r rK r r r r � r c S s g | ]}|d ur|�qS r r r r r r r � s �)r< ro )
�sslZcreate_default_context�load_dh_params� ImportErrorr rQ r �argv�
executabler8 r" r r3 )rs rt r? �fnr Z actual_modeZ actual_flagr r r � test_open� s<
�
F �� �ry c s� g � � fdd�} t �| �}zlt� �H}td� d|_td� d|_td� d|_td � d
|_W d � n1 sp0 Y W t �|� n t �|� 0 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�3r: �4r �call� )r �settracer �evalZ __cantrace__r3 )r~ �oldr? r r} r �
test_cantrace� s
&r� c C sZ dd l } t� �8}| � dd� t|jd d d d� d� W d � n1 sL0 Y d S )Nr rp � r: r; )rp r� )�mmapr r* r )r� r? r r r � test_mmap s r� c s0 dd� � � fdd�} t �| � � t _td��d S )Nc S s | t urt�| ||� d S r )r" r �__excepthook__)r � exc_value�exc_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.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 sL | dkrH|d � kr,t d|d � d� � ���t| t|d j�|d j� 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 s| ddl m} m}m}m} dd� }t�|� | |d�}||d� z||d� W n ty^ Y n
0 td��|� � }||� 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 sp dd l } dd� }t�|� | �� | � | j| j�}z.z|�d� W n tyT Y n0 W |�� n
|�� 0 d S )Nr c S s | � d�rt| g|�R � d S )Nzsocket.r� r� r r r r? N s
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 s | � d�rt| g|�R � d S )Nzgc.r� r� r r r r? d s
ztest_gc.<locals>.hookr: )Z
generation)�gcr r
� get_objects�objectZ
get_referrersZ
get_referents)r� r? r( r) r r r �test_gca s
r� c C sB dd l } dd� }t�|� | �� D ]}t|t�r"||vs"J �q"d S )Nr c W s d S r r rq r r r rr v r z test_not_in_gc.<locals>.<lambda>)r� r r
r� r5 �list)r� r? rW r r r �test_not_in_gcs s
r� �__main__)�suppress_msvcrt_assertsr: ) r! �
contextlibrT r r r* r- r. r3 �contextmanagerr8 r@ rD rE r[ rb rm ry r� r� r� r� r� r� r� r� r Z test.supportr� rv �test�globalsr r r r �<module> s< $
+!
Sindbad File Manager Version 1.0, Coded By Sindbad EG ~ The Terrorists