File: //lib/python3/dist-packages/samba/tests/dcerpc/__pycache__/raw_testcase.cpython-310.pyc
o
��a@� � @ s� d dl Z d dlZd dlm mZ d dlZd dlZd dlm mZ d dl Zd dlm
Z
d dlmZ d dl m
Z
d dlmZmZmZ d dlmZmZmZ d dlmZ d dlmZ d d lmZ d d
lmZ G dd� de�ZG d
d� de
�ZdS )� N)�gensec)�Credentials)�TestCase)�ndr_pack�
ndr_unpack�ndr_unpack_out)�!NT_STATUS_CONNECTION_DISCONNECTED�NT_STATUS_PIPE_DISCONNECTED�NT_STATUS_IO_TIMEOUT)�
NTSTATUSError)�param)�libsmb_samba_internal)�SMB_SIGNING_REQUIREDc @ s8 e Zd Zdd� Zdd� Zdd� Zd
dd �Zd
d
d�ZdS )�smb_pipe_socketc C sd t �� }|�|j� |�� }|�t� tj|d||dd�| _ |�|� | j j
|dddd|d�| _d S ) NzIPC$T)�creds�ipci� � � i@ @ )�
DesiredAccess�ShareAccess�CreateDisposition�
CreateOptions�ImpersonationLevel)�s3param�get_context�load�
configfile�get_smb_ipc_signing�set_smb_ipc_signingr �libsmb�Conn�smbconn�create�smbfid)�self�target_hostname�pipenamer �impersonation_level�lp�lp3�saved_signing_state� r+ �A/usr/lib/python3/dist-packages/samba/tests/dcerpc/raw_testcase.py�__init__* s
�
�zsmb_pipe_socket.__init__c C s | j �| j� | ` d S �N)r! �closer# �r$ r+ r+ r, r/ : s zsmb_pipe_socket.closec C s( t |d �}|dksJ �| j�|� d S )Ni� r )�intr! �
settimeout)r$ �timeo�msecsr+ r+ r, r2 >