File: //usr/lib/python3/dist-packages/samba/tests/__pycache__/segfault.cpython-310.pyc
o
�Kya � @ s� d Z ddlZddlZddlZddlmZmZ ddlm Z ddlm
Z
m Z ddl m
Z
mZmZmZmZ ddlmZmZ ddlmZ dd lmZ dd
lmZ dd lmZ dd lmZ ddlZd
d� Zdd� Z G dd� dej j �Z dS )zHTest whether various python calls segfault when given unexpected input.
� N)�Net�LIBNET_JOIN_AUTOMATIC)�DONT_USE_KERBEROS)�
NTSTATUSError�ntstatus)�misc�drsuapi�samr�unixinfo� dnsserver)�auth�gensec)�SamDB)�netbios)�registry)�ldb)� messagingc s � fdd�}|S )Nc
s� t �� }|dkr7z � | i |�� W n ty' } z
t�� W Y d }~nd }~ww tj�� tj�� t � d� t �
|d�\}}t � |�rOt � |�}t
d| ��d S )Nr zFailed with signal %d)�os�fork� Exception� traceback� print_exc�sys�stderr�flush�stdout�_exit�waitpid� WIFSIGNALED�WTERMSIG�AssertionError)�args�kwargs�pid�e�pid2�status�signal��f� �6/usr/lib/python3/dist-packages/samba/tests/segfault.py�wrapper( s ��
�z"segfault_detector.<locals>.wrapperr* )r) r, r* r( r+ �segfault_detector' s r- c s ddl m� � �fdd�}|S )Nr )�environc s d� d< �| i |�� � d= d S )N�1�PLEASE_NO_GDB_BACKTRACEr* )r! r" �r. r) r* r+ �w= s
zno_gdb_backtrace.<locals>.w)r r. )r) r2 r* r1 r+ �no_gdb_backtrace; s r3 c @ sH e Zd Zdd� Zdd� Zedd� �Zeedd� ��Zed d
� �Z ed d � �Z
ed
d� �Z edd� �Z edd� �Z
edd� �Zedd� �Zedd� �Zedd� �Zedd � �Zed d � �Zed d � �Zed!d"� �Zed#d$� �Zed%d&� �Zed'd(� �Zeed)d*� ��Zeed+d,� ��Zeed-d.� ��Zeed/d0� ��Z eed1d2� ��Z d3S )4�
SegfaultTestsc C s. t jd }| �� }| j| �� td�}|||fS )N�SERVER)�template�kerberos_state)r r. � get_loadparm� insta_creds�get_credentialsr )�self�server�lp�credsr* r* r+ � get_lp_et_alF s
�
zSegfaultTests.get_lp_et_alc C s( | � � \}}}d| }t|||d�}|S )Nzldap://)� credentialsr= )r? r )r; r= r> r< �urlr r* r* r+ � get_samdbN s zSegfaultTests.get_samdbc C s4 | � � \}}}t|||d�}|�d|d t�� � d S �N�r< �* )r? r �replicate_initr �GUID)r; r= r> r< �netr* r* r+ �test_net_replicate_init__1T s z(SegfaultTests.test_net_replicate_init__1c C s< | � � }| �� \}}}t|||d�}|�||dt�� � d S rC )rB r? r rF r rG )r; �samdbr= r> r< rH r* r* r+ �test_net_replicate_init__3Z s z(SegfaultTests.test_net_replicate_init__3c C s6 | � � \}}}t�� }t|||d�}|�dd|� d S )NrD rE � )r? r �DsGetNCChangesCtr6r �replicate_chunk)r; r= r> r< �ctrrH r* r* r+ �test_net_replicate_chunk_1c s z(SegfaultTests.test_net_replicate_chunk_1c C s" t jddgd�}tjj|d�}d S )NrE �sam)r �methods�� auth_context)r � AuthContextr
�Security� start_server)r; �a�gr* r* r+ �%test_auth_context_gensec_start_serverj s z3SegfaultTests.test_auth_context_gensec_start_serverc C s t jddd�}d S )NrE �foo)r � principal)r � user_session)r; �sr* r* r+ �test_auth_user_sessionr � z$SegfaultTests.test_auth_user_sessionc C s t jjdd� d S )NrE rS )r
rV rW �r; r* r* r+ �test_gensec_start_serverv r` z&SegfaultTests.test_gensec_start_serverc C s t �� }|�dd�}d S )N)rE r[ � localhost)r �Node�
query_name)r; �n�tr* r* r+ �test_netbios_query_namez s z%SegfaultTests.test_netbios_query_namec C s | � � \}}}|�d� d S )NrE )r? �encrypt_netr_crypt_password)r; r= r> r< r* r* r+ � test_encrypt_netr_crypt_password s z.SegfaultTests.test_encrypt_netr_crypt_passwordc
C sJ z
t jddd� W d S tjy$ } z
td| � W Y d }~d S d }~ww )N� rE )r@ �failed with %s)r �open_ldbr �LdbError�print�r; r$ r* r* r+ �test_hive_open_ldb� s ��z SegfaultTests.test_hive_open_ldbc
C sL z t �dddd� W d S tjy% } z
td| � W Y d }~d S d }~ww )Nr^ rl )r � open_hiver rn ro rp r* r* r+ �test_hive_open_hive� s ��z!SegfaultTests.test_hive_open_hivec C sB t �� }t �d�}z|�|� W n
t jy Y nw t|� d S )N�q)r �Message�MessageElement�addrn �str)r; �mr$ r* r* r+ � test_ldb_add_nameless_element� s
� z+SegfaultTests.test_ldb_add_nameless_elementc C s t �d� d S �Nrk )r �register_modulera r* r* r+ �test_ldb_register_module� s z&SegfaultTests.test_ldb_register_modulec C s t �dddd� d S )Nr^ F)r �
deregisterra r* r* r+ �test_messaging_deregister� s z'SegfaultTests.test_messaging_deregisterc C s ddl m} |�d� d S )Nr )�echork )� samba.dcerpcr� �rpcecho)r; r� r* r* r+ � test_rpcecho� s zSegfaultTests.test_rpcechoc C s dt �� _dS )z�There are many pidl generated functions that crashed on this
pattern, where a NULL pointer was created rather than an empty
structure.rL N)r �Connect5� out_info_outra r* r* r+ � test_dcerpc_idl_ref_elements� s z*SegfaultTests.test_dcerpc_idl_ref_elementsc C � t �� j dS )z$Dereferencing is sufficient to crashN)r
�GetPWUid� out_infosra r* r* r+ �!test_dcerpc_idl_unixinfo_elements� � z/SegfaultTests.test_dcerpc_idl_unixinfo_elementsc C r� )z'Inline arrays were incorrectly handled.N)r �DNS_RPC_SERVER_INFO_DOTNET� pExtensionsra r* r* r+ � test_dcerpc_idl_inline_arrays� r� z+SegfaultTests.test_dcerpc_idl_inline_arraysc C s t �� }t �� }|||g|_dS )z0Setting an inline array was incorrectly handled.N)r �
DNS_EXTENSION�DNS_RPC_DP_INFO� pwszReserved)r; rX �xr* r* r+ �!test_dcerpc_idl_set_inline_arrays� s z/SegfaultTests.test_dcerpc_idl_set_inline_arraysc C s$ ddl m} |�� }d|_|j d S )Nr )�dnsp� )r� r� � string_list�countrx )r; r� r^ r* r* r+ �test_dnsp_string_list� s
z#SegfaultTests.test_dnsp_string_listc C s: ddl m} ddlm}m} |g d��}|j|_|j d S )Nr )� TXTRecord)r� r )rX �b�c)�samba.dnsserverr� r� r� r �
DNS_TYPE_A�wType�data)r; r� r� r �recr* r* r+ �test_dns_record� s
z SegfaultTests.test_dns_recordc C s6 | � � }t�� }t�|d�|_|�||�}~|j d S r{ )rB r ru �Dn�dn�msg_diff)r; rJ �msg�diffr* r* r+ �test_ldb_msg_diff� s
z SegfaultTests.test_ldb_msg_diffc C s t �� }|`d S )N)r ru r� )r; r� r* r* r+ �test_ldb_msg_del_dn� s z!SegfaultTests.test_ldb_msg_del_dnc C s | � � }t�|d�}|`d S )Nzrelax:1)rB r �Control�critical)r; rJ r� r* r* r+ � test_ldb_control_del_critical� s z+SegfaultTests.test_ldb_control_del_criticalN) �__name__�
__module__� __qualname__r? rB r- rI r3 rK rP rZ r_ rb rh rj rq rs rz r} r r� r� r� r� r� r� r� r� r� r� r* r* r* r+ r4 E sn
r4 ) �__doc__� samba.tests�sambar r � samba.netr r �samba.credentialsr r r r� r r r r
r r r
� samba.samdbr r r r r r r- r3 �tests�TestCaser4 r* r* r* r+ �<module> s$