D7net
Home
Console
Upload
information
Create File
Create Folder
About
Tools
:
/
lib64
/
python3.9
/
site-packages
/
slapdtest
/
__pycache__
/
Filename :
_slapdtest.cpython-39.pyc
back
Copy
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 � $ (