File: //lib64/python3.9/site-packages/slapdtest/__pycache__/_slapdtest.cpython-39.opt-1.pyc
a
�.dlL � @ s6 d Z ddlZddlZddlZddlZddlZddlZddlZddlm Z ddl
Z
ddl m Z ddl
mZ dejd< ddlZej�ej�e��ZdZd Zeej�d
d ��d ��Zd
ev r�dZ n
e ed�Z dd� Z d&dd�Z dd� Z dd� Z!dd� Z"dd� Z#d d � Z$ej%d d fd d!�Z&G d"d#� d#�Z'G d$d%� d%e
j(�Z)dS )'zz
slapdtest - module for spawning test instances of OpenLDAP's slapd server
See https://www.python-ldap.org/ for details.
� N)�
SysLogHandler)�which)�
quote_plus�1Z
LDAPNOINITa dn: cn=config
objectClass: olcGlobal
cn: config
olcServerID: %(serverid)s
olcLogLevel: %(loglevel)s
olcAllows: bind_v2
olcAuthzRegexp: {0}"gidnumber=%(root_gid)s\+uidnumber=%(root_uid)s,cn=peercred,cn=external,cn=auth" "%(rootdn)s"
olcAuthzRegexp: {1}"C=DE, O=python-ldap, OU=slapd-test, CN=([A-Za-z]+)" "ldap://ou=people,dc=local???($1)"
olcTLSCACertificateFile: %(cafile)s
olcTLSCertificateFile: %(servercert)s
olcTLSCertificateKeyFile: %(serverkey)s
olcTLSVerifyClient: try
dn: cn=module,cn=config
objectClass: olcModuleList
cn: module
olcModuleLoad: back_%(database)s
dn: olcDatabase=%(database)s,cn=config
objectClass: olcDatabaseConfig
objectClass: olcMdbConfig
olcDatabase: %(database)s
olcSuffix: %(suffix)s
olcRootDN: %(rootdn)s
olcRootPW: %(rootpw)s
olcDbDirectory: %(directory)s
z 127.0.0.1� CI_DISABLED� �:�LDAPIFZAF_UNIXc C s | S )zIdentity decorator
� )Z test_itemr
r
�:/usr/lib64/python3.9/site-packages/slapdtest/_slapdtest.py�identity? s r c C s2 t j�dd�st�| �S |tv r*t�| �S tS dS )z@Skip test unless test case is executed on CI like Travis CI
ZCIFN)�os�environ�get�unittest�skipr r )�reason�featurer
r
r �skip_unless_ciF s
r c C s t jstddd�S tS dS )zNDecorator for TLS tests
Tests are not skipped on CI (e.g. Travis CI)
ztest needs ldap.TLS_AVAILZTLS�r N)�ldapZ TLS_AVAILr r r
r
r
r � requires_tlsR s r c C s t jstddd�S tS d S )Nztest needs ldap.SASL_AVAILZSASLr )r �
SASL_AVAILr r r
r
r
r �
requires_sasl] s
�r c C s t stddd�S tS d S )Nz"test needs ldapi support (AF_UNIX)r r )�
HAVE_LDAPIr r r
r
r
r �requires_ldapie s
�r c C s t jstddd�S tS d S )Nztest needs ldap.INIT_FDZINIT_FDr )r Z
INIT_FD_AVAILr r r
r
r
r �requires_init_fdl s
�r c C s> | � tj�}tjdkr2dD ]}||vr|�|� qtj�|�S )z:Add /sbin and related directories to a command search path�win32)z/usr/local/sbinz/sbinz /usr/sbin)�splitr
�pathsep�sys�platform�append�join)�pathZ directoriesZsbinr
r
r � _add_sbint s
r% z%(levelname)s %(message)sz%%(asctime)s %(levelname)s %(message)sc C s� dt jv r4t jd }z t|�}W n ty2 Y n0 t�| �}|r�t j�d�r�tjd� | |f�d�}tj
j dt j d�}|�
|� |�|� |r�tj|d�}t�� }|�
|� |�|� |�|� |S )z]
Returns a combined SysLogHandler/StreamHandler logging instance
with formatters
ZLOGLEVELz/dev/log� )Zfmt)ZaddressZfacility)r
r �int�
ValueError�loggingZ getLoggerr$ �existsZ Formatterr# �handlersr Z
LOG_DAEMONZ setFormatterZ
addHandlerZ
StreamHandlerZsetLevel) Zlog_nameZ log_levelZsys_log_formatZconsole_log_formatZ
new_loggerZmy_syslog_formatterZmy_syslog_handlerZmy_stream_formatterZmy_stream_handlerr
r
r �combined_logger} s0
��
r, c @ s� e Zd ZdZeZdZdZdZdZ dZ
e Z dZ
dZej�d e�� �Zd
ejv rXejd
Zn(ej�d �rjd Znej�d �r|d Znd
Zej�dej�dej��Zej�dee��Zed�Z dd� Z e dd� �Z e dd� �Z e dd� �Z!dd� Z"dKd d �Z#d d � Z$d!d"� Z%d#d$� Z&d%d&� Z'd'd(� Z(d)d*� Z)d+d,� Z*d-d.� Z+d/d0� Z,d1d2� Z-d3d4� Z.d5d6� Z/d7d8� Z0d9d:� Z1dLd;d<�Z2dMd=d>�Z3dNd?d@�Z4dOdAdB�Z5dPdCdD�Z6dQdEdF�Z7dGdH� Z8dIdJ� Z9d
S )R� SlapdObjecta�
Controller class for a slapd instance, OpenLDAP's server.
This class creates a temporary data store for slapd, runs it
listening on a private Unix domain socket and TCP port,
and initializes it with a top-level entry and the root user.
When a reference to an instance of this class is lost, the slapd
server is shut down.
An instance can be used as a context manager. When exiting the context
manager, the slapd server is shut down and the temporary data store is
removed.
:param openldap_schema_files: A list of schema names or schema paths to
load at startup. By default this only contains `core`.
.. versionchanged:: 3.1
Added context manager functionality
Zmdbz#dc=slapd-test,dc=python-ldap,dc=orgZManager�passwordz stats stats2)�slapd.d)z core.ldifZTMPZSCHEMAz/etc/openldap/schemaz/etc/ldap/schemaN�BIN�PATH�SBINzpython-ldap-testc C s d | _ | �� | _| jd | _tj�| jd| j �| _tj�| jd�| _ tj�| jd�| _
d| j | jf | _ t
r�tj�| jd�}dt|� | _| j| _tj| _nd | _| j | _d| _| �� | jd u r�td ��tj�td
�| _tj�td �| _tj�td �| _tj�td
�| _tj�td�| _ d S )Ni zpython-ldap-test-%dr/ z
openldap-dataz
ldap://%s:%d/Zldapiz
ldapi://%sFz,SCHEMADIR is None, ldap schemas are missing.z certs/ca.pemzcerts/server.pemzcerts/server.keyzcerts/client.pemzcerts/client.key) �_proc�_avail_tcp_port�_port� server_idr
r$ r# �TMPDIR�
testrundir� _slapd_conf�
_db_directory�
local_host�ldap_urir r � ldapi_uri�default_ldap_urir r �cli_sasl_external�_find_commands� SCHEMADIRr( �HERE�cafile�
servercert� serverkeyZ
clientcertZ clientkey)�selfZ
ldapi_pathr
r
r �__init__� s.
zSlapdObject.__init__c C s dj | d�S )Nz cn={self.root_cn},{self.suffix}�rF )�formatrH r
r
r �root_dn� s zSlapdObject.root_dnc C s | j S �N)r; rH r
r
r �hostname� s zSlapdObject.hostnamec C s | j S rK )r5 rH r
r
r �port� s zSlapdObject.portc C sf | � d�| _| � d�| _| � d�| _| � d�| _| � d�| _tj�dd �| _ | j sb| j ddd �| _ d S )
N�ldapadd�
ldapdelete�
ldapmodify�
ldapwhoami�slapaddZSLAPDZslapdT)�in_sbin)
�
_find_command� PATH_LDAPADD�PATH_LDAPDELETE�PATH_LDAPMODIFY�PATH_LDAPWHOAMI� PATH_SLAPADDr
r r �
PATH_SLAPDrH r
r
r r@ s zSlapdObject._find_commandsFc C sB |r| j }d}n
| j}d}t||d�}|d u r>td�||���|S )Nr2 r0 )r$ z\Command '{}' not found. Set the {} environment variable to override slapdtest's search path.)� SBIN_PATH�BIN_PATHr r( rI )rF �cmdrS r$ Zvar_nameZcommandr
r
r rT s ��zSlapdObject._find_commandc C s( t �| j� t �| j� | �| j� dS )z�
creates rundir structure
for setting up a custom directory structure you have to override
this method
N)r
�mkdirr8 r: �_create_sub_dirs�testrunsubdirsrH r
r
r � setup_rundir s zSlapdObject.setup_rundirc C s� ddl }ddl}|j�| j�s"dS | j�d| j� |j| jdd�D ]v\}}}|D ]0}| j�d|j�||�� |� |j�||�� qP|D ]0}| j�d|j�||�� |�
|j�||�� q�qB|�
| j� | j� d| j� dS ) zH
Recursively delete whole directory specified by `path'
r Nz clean-up %sF)�topdownz remove %szrmdir %sz
cleaned-up %s) r
Zos.pathr$ r* r8 �_log�debug�walkr# �remove�rmdir�info)rF r
�dirpath�dirnames� filenames�filename�dirnamer
r
r �_cleanup_rundir& s � zSlapdObject._cleanup_rundirc C sN t � � }z(|�| jdf� |�� d }W |�� n
|�� 0 | j�d|� |S )z;
find an available port for TCP connection
r � zFound available port %d)�socketZbindr; Z getsockname�closerc rh )rF ZsockrM r
r
r r4 = s zSlapdObject._avail_tcp_portc
C sH t | j�| j| j| j| j| j| jt� � t�
� | j | j | j
d� }| j| S )z�
generates a slapd.conf and returns it as one string
for generating specific static configuration files you have to
override this method
) ZserveridZloglevel�databaseZ directory�suffixZrootdnZrootpwZroot_uidZroot_gidrC rD rE )�hexr6 �slapd_loglevelrr r: rs rJ �root_pwr
�getuid�getgidrC rD rE �slapd_conf_template)rF Z config_dictr
r
r �
gen_configJ s �zSlapdObject.gen_configc C s6 |D ],}t j�| j|�}| j�d|� t �|� qdS )z@
create sub-directories beneath self.testrundir
zCreate directory %sN)r
r$ r# r8 rc rd r^ )rF Z dir_namesZdnameZdir_namer
r
r r_ a s z SlapdObject._create_sub_dirsc sf � j �d� j� � �� �� dg� � fdd�� jD �}|D ]}� �ddd|g� q:� j �d� j� dS )z Loads the slapd.d configuration.zimporting configuration: %sz-n0c s, g | ]$}t j�|�r|nt j�� j|��qS r
)r
r$ r* r# rA )�.0ZschemarH r
r �
<listcomp>o s �
�z-SlapdObject._write_config.<locals>.<listcomp>Nz-lz
import ok: %s)rc rd r9 rR rz �openldap_schema_files)rF Z
ldif_pathsZ ldif_pathr
rH r �
_write_configj s
�zSlapdObject._write_configc C sv | j �d| j� | jdd| jddddg}tj|tjtjd�}|jd krb| j � |j
� d
�� t d ��| j �
d | j� d S )
Nztesting config %sz-Ttest�-Fz-uz-v�-dZconfig)�stdout�stderrr �utf-8zconfiguration test failedz
config ok: %s)rc rd r9 rZ �
subprocess�run�PIPEZSTDOUT�
returncode�errorr� �decode� RuntimeErrorrh )rF Z
popen_list�pr
r
r � _test_configz s" ��
zSlapdObject._test_configc C s | j g}| jr|�| j� | jd| jdd�|�g}| j�tj �rP|�
ddg� n|�
ddg� | j� dd�|�� t �
|�| _t�� d }| j�� d ur�| �� td
��z | j�d | j� | �� W n, ty� t�� |kr�Y q�t�d � Y q�0 d S q�td
��d S )z0
Spawns/forks the slapd process
r z-hr& r� z-1�0zstarting slapd: %r�
Nz slapd exited before opening portz slapd connection check to %sg�������?z slapd did not start properly)r< r= r" rZ r9 r# rc � isEnabledForr) �DEBUG�extendrh r� �Popenr3 �time� monotonicZpoll�_stoppedr� rd r> rQ �sleep)rF ZurlsZ
slapd_argsZdeadliner
r
r � _start_slapd� s6
� � zSlapdObject._start_slapdc C s\ | j du rXt�| j� | �� | �� | �� | �� | �� | j �
d| j j | j | j
� dS )zW
Starts the slapd server process running, and waits for it to come up.
Nz(slapd with pid=%d listening on %s and %s)r3 �atexit�register�stoprn ra r~ r� r� rc rd �pidr< r= rH r
r
r �start� s
�zSlapdObject.startc C sF | j dur.| j�d| j j� | j �� | �� | �� t�| j � dS )zU
Stops the slapd server, and waits for it to terminate and cleans up
Nzstopping slapd with pid %d)
r3 rc rd r� � terminate�waitrn r� Z
unregisterr� rH r
r
r r� � s
zSlapdObject.stopc C s | j �� | �� | �� dS )z:
Restarts the slapd server with same data
N)r3 r� r� r� rH r
r
r �restart� s
zSlapdObject.restartc C s | j r| j �� | �� dS )z3Waits for the slapd process to terminate by itself.N)r3 r� r� rH r
r
r r� � s
zSlapdObject.waitc C s& | j dur"| j�d| j j� d| _ dS )z8Called when the slapd server is known to have terminatedNzslapd[%d] terminated)r3 rc rh r� rH r
r
r r� � s
zSlapdObject._stoppedc C s> | j r(ddg}| j�tj�s:|�d� ndd| jd| jg}|S )Nz-YZEXTERNALz-Qz-xz-Dz-w)r? rc r� r) r� r" rJ rv )rF Z
authc_argsr
r
r �_cli_auth_args� s � �zSlapdObject._cli_auth_argsc C s� |d u r| j }|�d�d �d�r6|d|g| �� }n |d| jg}||pJg 7 }| j�dd�|�� tj |tj
tj
tj
d�}| j�d |� |� |�\}}|d ur�| j�d
|� |d ur�| j�d |� |� � d kr�t
d
�|||���||fS )N�/���r z-Hr zRun command: %rr& )�stdinr� r� z
stdin_data=%rzstdout_data=%rzstderr_data=%rr z {!r} process failed:
{!r}
{!r})r> r �
startswithr� r9 rc rd r# r� r� r� Z communicater� r� rI ) rF Z ldapcommand�
extra_argsr< �
stdin_data�args�procZ stdout_dataZ stderr_datar
r
r �
_cli_popen� s0
� ��zSlapdObject._cli_popenc C s | j | j|d� dS )z8
Runs ldapwhoami on this slapd instance
�r� N)r� rX )rF r� r
r
r rQ s zSlapdObject.ldapwhoamic C s | j | j||�d�d� dS �zR
Runs ldapadd on this slapd instance, passing it the ldif content
r� )r� r� N)r� rU �encode�rF Zldifr� r
r
r rN s
�zSlapdObject.ldapaddc C s | j | j||�d�d� dS r� )r� rW r� r� r
r
r rP " s
�zSlapdObject.ldapmodifyc C s8 |du r g }|r|� d� |� |� | j| j|d� dS )zG
Runs ldapdelete on this slapd instance, deleting 'dn'
Nz-rr� )r" r� rV )rF Zdn� recursiver� r
r
r rO ) s
zSlapdObject.ldapdeletec C s$ | j | j|r|�d�nd|d� dS )zR
Runs slapadd on this slapd instance, passing it the ldif content
r� N)r� r� )r� rY r� r� r
r
r rR 4 s
�zSlapdObject.slapaddc C s | � � | S rK )r� rH r
r
r � __enter__> s zSlapdObject.__enter__c C s | � � d S rK )r� )rF �exc_type� exc_value� tracebackr
r
r �__exit__B s zSlapdObject.__exit__)F)NNN)N)N)N)FN)N):�__name__�
__module__� __qualname__�__doc__�SLAPD_CONF_TEMPLATEry rr rs Zroot_cnrv ru � LOCALHOSTr; r` r} r
r r �getcwdr7 rA r$ �isdir�defpathr\ r% r[ r, rc rG �propertyrJ rL rM r@ rT ra rn r4 rz r_ r~ r� r� r� r� r� r� r� r� r� rQ rN rP rO rR r� r� r
r
r
r r- � sl
' �
r- c @ s> e Zd ZdZeZdZdZd dd�Ze dd� �Z
e dd� �Z dS )
�
SlapdTestCasezE
test class which also clones or initializes a running slapd
Nc K s< | j | jjfi |��}d|_|�|p*| jj|p4| jj� |S )z@
return a LDAPObject instance after simple bind
� )�ldap_object_class�serverr< Zprotocol_versionZ
simple_bind_srJ rv )rF ZwhoZcred�kwargsZ ldap_connr
r
r �_open_ldap_connO s z SlapdTestCase._open_ldap_connc C s | � � | _| j�� d S rK )� server_classr� r� ��clsr
r
r �
setUpClassY s
zSlapdTestCase.setUpClassc C s | j �� d S rK )r� r� r� r
r
r �
tearDownClass^ s zSlapdTestCase.tearDownClass)NN) r� r� r� r� r- r� r� r� r� � classmethodr� r� r
r
r
r r� F s
r� )N)*r� r
rp r r� r� r) r� Zlogging.handlersr r �shutilr � urllib.parser r r r$ �abspathrm �__file__rB r� r� �setr r r r �hasattrr r r r r r r% ZWARNr, r- ZTestCaser� r
r
r
r �<module> sH
�
$ (