Your IP : 216.73.216.196
o
0�ii- � @ s d dl Z d dlZd dlZG dd� d�Zdd� Zdd� Zdd � Zd
d � Ze jd 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)� Zd*d+� Zd,d-� Zd.d/� Zd0d1� Ze d2kr�d d3l m Z e � ej d4 Z e!� e � dS dS )5� Nc @ sH e Zd Z defdd�Zdd� Zdd� Zdd � Zed
d � �Z d d
� Z
dS )�TestHookNc C s |pd| _ || _g | _d| _d S )N� F)�raise_on_events�exc_type�seen�closed)�selfr r r r �7/opt/alt/python310/lib64/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__� 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# �ZelZseriesr 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
�r1 c
c sB � zd V W d S t y } z
t|t�r� W Y d }~d S d }~ww r )�
BaseException�
isinstancer$ )Zex_typeZexr r r � assertRaisesG s �
���r4 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 )r6 r7 r8 )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 r5 )r r r9 r+ r r, �Zhook1Zhook2r r r �test_block_add_hookY s
��"�r@ 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 )Nr= )r r )r4 r2 r r? r r r �!test_block_add_hook_baseexceptionc s
� �� ��"�rA 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�cr6 r7 r8 ztest-marshal.bin�wb�rbc S �( g | ]\}}|d kr|d |d f�qS )z
marshal.dumpsr r6 r �r �er r r r r ~ � ( z test_marshal.<locals>.<listcomp>r7 c S s g | ] \}}|d kr|d �qS )z
marshal.loadsr r rH r r r r � s c S s g | ]
\}}|d kr|�qS )� marshal.loadr rH r r r r � s rK )
�marshal�dumpsr r( �loads�open�dump�load�os�unlinkr r1 �version)rL �oZpayloadr; �f�actualr r r � test_marshaln s*
� �� � rX 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 r\ r r r r � PicklePrint� s r] rB rY zpickle.find_classr> )�picklerM r( rN r r4 r )r^ r] Z payload_1Z payload_2r; r r r � test_pickle� s
� "�r_ 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� � rb c @ r` )ztest_monkeypatch.<locals>.BNra r r r r �B� rc rd c @ r` )ztest_monkeypatch.<locals>.CNra r r r r �C� rc re �X� __bases__�{ c S rG )zobject.__setattr__r r6 r rH r r r r � rJ z$test_monkeypatch.<locals>.<listcomp>r � __class__) r r rg �type�__dict__�__set__r
Znew_attrri r r1 )rb rd re r r; rW r r r �test_monkeypatch� s � �rm 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 rO r> r7 �rrF r8 rE �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 )rO r6 r r rH r r r r � � 0 z test_open.<locals>.<listcomp>c S s0 g | ]\}}|d kr|d s|d |d f�qS )rO r6 r r7 r rH r r r r � rt c S s g | ]}|d ur|�qS r r r r r r r � s
�)r8 ro )
�sslZcreate_default_context�load_dh_params� ImportErrorr rO r �argv�
executabler4 r r r1 )ru rv 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�3r6 �4r Zcall� )r �settracer �evalZ __cantrace__r1 )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 rp � r6 r7 )rp 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_valueZexc_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.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 � 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� 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 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 ZDetach)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
Z gethostnameZAF_INETZ SOCK_STREAMZbind� Exceptionr )r� r; Zsockr 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>.hookr6 )Z
generation)�gcr r
� 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.r6 r� r� r r r r; v s
�z test_http_client.<locals>.hookzwww.python.orgZGET�/zhttp.client.sendz
[cannot send]) Z http.clientr r
ZclientZHTTPConnectionZrequestr� r� r )Zhttpr; Zconnr 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
ZconnectZ
Connection�hasattrr� Zload_extensionZOperationalErrorr )r� r; Zcx1Zcx2r r r � test_sqlite3� s
���r� c C sx dd l } dd� }t�|� | �d� | � d� | �| j� | �� | � d� | j| j| jd� d t_ | �� | �� d S )Nr c S r� )Nzsyslog.r� r� r r r r; � r� ztest_syslog.<locals>.hookZpythonr� Ztest2)Z logoptionZfacility)
�syslogr r
ZopenlogZ
setlogmaskZ LOG_DEBUGZcloselogZ
LOG_NDELAYZ
LOG_LOCAL0rx )r� r; r r r � test_syslog� s
r� c C s6 dd l } dd� }t�|� | �� D ]}t|t�rqd S )Nr c W r� r r rq r r r rr � rs z test_not_in_gc.<locals>.<lambda>)r� r r
r� r3 �list)r� r; rU r r r �test_not_in_gc� s
��r� �__main__)�suppress_msvcrt_assertsr6 )"�
contextlibrR r r r( r+ r, r1 �contextmanagerr4 r< r@ rA rX r_ rm r{ r� r� r� r� r� r� r� r� r� r� r� r Z test.supportr� rx r� �globalsr r r r �<module> sF $
+!
�