File: //usr/local/lib/python3.10/test/__pycache__/audit-tests.cpython-310.opt-2.pyc
o
�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 �-/usr/local/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__� 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
�r2 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_type�exr r r � assertRaisesG s �
���r6 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 )r8 r9 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 r7 )r r r; r, r r- �Zhook1Zhook2r r r �test_block_add_hookY s
��"�rB 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 )r6 r3 r rA r r r �!test_block_add_hook_baseexceptionc s
� �� ��"�rC 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�cr8 r9 r: ztest-marshal.bin�wb�rbc S �( g | ]\}}|d kr|d |d f�qS )z
marshal.dumpsr r8 r �r �er r r r r ~ � ( z test_marshal.<locals>.<listcomp>r9 c S s g | ] \}}|d kr|d �qS )z
marshal.loadsr r rJ r r r r � s c S s g | ]
\}}|d kr|�qS )� marshal.loadr rJ r r r r � s rM )
�marshal�dumpsr r( �loads�open�dump�load�os�unlinkr r2 �version)rN �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 r_ r r r r � PicklePrint� s r` rD r\ zpickle.find_classr@ )�picklerO r( rP r r6 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 � �#}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� � re c @ rc )ztest_monkeypatch.<locals>.BNrd r r r r �B� rf rg c @ rc )ztest_monkeypatch.<locals>.CNrd r r r r �C� rf rh �X� __bases__�{ c S rI )zobject.__setattr__r r8 r rJ r r r r � rL z$test_monkeypatch.<locals>.<listcomp>r � __class__) r r rj �type�__dict__�__set__r
Znew_attrrl r r2 )re rg rh r r= rZ r r r �test_monkeypatch� s � �rp 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 rQ r@ r9 �rrH r: rG �w���Fc W s dS )Nr8 r �r r r r �<lambda>� � ztest_open.<locals>.<lambda>c S s0 g | ]\}}|d kr|d r|d |d f�qS )rQ r8 r r rJ r r r r � � 0 z test_open.<locals>.<listcomp>c S s0 g | ]\}}|d kr|d s|d |d f�qS )rQ r8 r r9 r rJ r r r r � rw c S s g | ]}|d ur|�qS r r r r r r r � s
�)r: rr )
�sslZcreate_default_context�load_dh_params� ImportErrorr rQ r �argv�
executabler6 r r r2 )rx ry 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�3r8 �4r �call� )r �settracer �evalZ __cantrace__r2 )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 rs � r8 r9 )rs 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.excepthookr9 r8 zExpected isinstance(z, �)r � Expected � == )r4 � 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� r8 )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>.hookr8 )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.r8 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� 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_LOCAL0r{ )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 rt r r r ru � rv z test_not_in_gc.<locals>.<lambda>)r� r r
r� r4 �list)r� r= rW r r r �test_not_in_gc� s
��r� �__main__)�suppress_msvcrt_assertsr8 )"�
contextlibrT r r r( r, r- r2 �contextmanagerr6 r> rB rC r[ rb rp r~ r� r� r� r� r� r� r� r� r� r� r� r Z test.supportr� r{ r� �globalsr r r r �<module> sF $
+!
�