Perform binds in a loop, using either a single DN or many DNs

tests/progs/slapd-bind.c

+/* $OpenLDAP$ */
+/* This work is part of OpenLDAP Software <http://www.openldap.org/>.
+ *
+ * Copyright 1999-2005 The OpenLDAP Foundation.
+ * All rights reserved.
+ *
+ * Redistribution and use in source and binary forms, with or without
+ * modification, are permitted only as authorized by the OpenLDAP
+ * Public License.
+ *
+ * A copy of this license is available in file LICENSE in the
+ * top-level directory of the distribution or, alternatively, at
+ * <http://www.OpenLDAP.org/license.html>.
+ */
+/* ACKNOWLEDGEMENTS:
+ * This work was initially developed by Kurt Spanier for inclusion
+ * in OpenLDAP Software.
+ */
+
+#include "portable.h"
+
+#include <stdio.h>
+
+#include <ac/stdlib.h>
+
+#include <ac/ctype.h>
+#include <ac/param.h>
+#include <ac/socket.h>
+#include <ac/string.h>
+#include <ac/unistd.h>
+#include <ac/wait.h>
+
+#define LDAP_DEPRECATED 1
+#include <ldap.h>
+#include <lutil.h>
+
+#define LOOPS	100
+
+static void
+do_bind( char *uri, char *host, int port, char *dn, char *pass, int maxloop );
+
+static void
+do_base( char *uri, char *host, int port, char *base, char *pass, int maxloop );
+
+static void
+usage( char *name )
+{
+	fprintf( stderr, "usage: %s [-h <host>] -p port (-D <dn>|-b <baseDN>) -w <passwd> [-l <loops>]\n",
+			name );
+	exit( EXIT_FAILURE );
+}
+
+static char *filter = "(objectClass=person)";
+
+int
+main( int argc, char **argv )
+{
+	int		i;
+	char		*uri = NULL;
+	char        *host = "localhost";
+	char		*dn = NULL;
+	char		*base = NULL;
+	char		*pass = NULL;
+	int			port = -1;
+	int			loops = LOOPS;
+
+	while ( (i = getopt( argc, argv, "b:H:h:p:D:w:l:" )) != EOF ) {
+		switch( i ) {
+			case 'b':		/* base DN of a tree of user DNs */
+				base = strdup( optarg );
+				break;
+
+			case 'H':		/* the server uri */
+				uri = strdup( optarg );
+			break;
+			case 'h':		/* the servers host */
+				host = strdup( optarg );
+			break;
+
+			case 'p':		/* the servers port */
+				port = atoi( optarg );
+				break;
+
+			case 'D':
+				dn = strdup( optarg );
+				break;
+
+			case 'w':
+				pass = strdup( optarg );
+				break;
+
+			case 'l':		/* the number of loops */
+				loops = atoi( optarg );
+				break;
+
+			default:
+				usage( argv[0] );
+				break;
+		}
+	}
+
+	if ( port == -1 && uri == NULL )
+		usage( argv[0] );
+
+	if ( base )
+		do_base( uri, host, port, base, pass, ( 20 * loops ));
+	else
+		do_bind( uri, host, port, dn, pass, ( 20 * loops ));
+	exit( EXIT_SUCCESS );
+}
+
+
+static void
+do_bind( char *uri, char *host, int port, char *dn, char *pass, int maxloop )
+{
+	LDAP	*ld = NULL;
+	int  	i;
+	char	*attrs[] = { "1.1", NULL };
+	pid_t	pid = getpid();
+
+	if ( maxloop > 1 )
+		fprintf( stderr, "PID=%ld - Bind(%d): dn=\"%s\".\n",
+			 (long) pid, maxloop, dn );
+
+	for ( i = 0; i < maxloop; i++ ) {
+		LDAPMessage *res;
+		int         rc;
+
+		if ( uri ) {
+			ldap_initialize( &ld, uri );
+		} else {
+			ld = ldap_init( host, port );
+		}
+		if ( ld == NULL ) {
+			perror( "ldap_init" );
+			break;
+		}
+
+		{
+			int version = LDAP_VERSION3;
+			(void) ldap_set_option( ld, LDAP_OPT_PROTOCOL_VERSION,
+				&version ); 
+		}
+
+		rc = ldap_bind_s( ld, dn, pass, LDAP_AUTH_SIMPLE );
+		if ( rc != LDAP_SUCCESS ) {
+			ldap_perror( ld, "ldap_bind" );
+		}
+		ldap_unbind( ld );
+		if ( rc != LDAP_SUCCESS ) {
+			break;
+		}
+	}
+
+	if ( maxloop > 1 )
+		fprintf( stderr, " PID=%ld - Bind done.\n", (long) pid );
+
+}
+
+
+static void
+do_base( char *uri, char *host, int port, char *base, char *pass, int maxloop )
+{
+	LDAP	*ld = NULL;
+	int  	i = 0;
+	pid_t	pid = getpid();
+	int     rc = LDAP_SUCCESS;
+	ber_int_t msgid;
+	LDAPMessage *res, *msg;
+	char **rdns = NULL;
+	char *attrs[] = { "dn", NULL };
+	int nrdns = 0;
+
+	srand(pid);
+
+	if ( uri ) {
+		ldap_initialize( &ld, uri );
+	} else {
+		ld = ldap_init( host, port );
+	}
+	if ( ld == NULL ) {
+		perror( "ldap_init" );
+		exit( EXIT_FAILURE );
+	}
+
+	{
+		int version = LDAP_VERSION3;
+		(void) ldap_set_option( ld, LDAP_OPT_PROTOCOL_VERSION,
+			&version ); 
+	}
+	(void) ldap_set_option( ld, LDAP_OPT_REFERRALS, LDAP_OPT_OFF );
+
+	rc = ldap_bind_s( ld, NULL, NULL, LDAP_AUTH_SIMPLE );
+	if ( rc != LDAP_SUCCESS ) {
+		ldap_perror( ld, "ldap_bind" );
+		exit( EXIT_FAILURE );
+	}
+
+	rc = ldap_search_ext( ld, base, LDAP_SCOPE_SUBTREE,
+			filter, attrs, 0, NULL, NULL, 0, 0, &msgid );
+	if ( rc != LDAP_SUCCESS ) {
+		ldap_perror( ld, "ldap_search_ex" );
+		exit( EXIT_FAILURE );
+	}
+
+	while (( rc=ldap_result( ld, LDAP_RES_ANY, LDAP_MSG_ONE, NULL, &res )) >0){
+		BerElement *ber;
+		struct berval bv;
+		char *ptr;
+		int done = 0;
+
+		for (msg = ldap_first_message( ld, res ); msg;
+			msg = ldap_next_message( ld, msg )) {
+			switch ( ldap_msgtype( msg )) {
+			case LDAP_RES_SEARCH_ENTRY:
+				rc = ldap_get_dn_ber( ld, msg, &ber, &bv );
+				ptr = strchr( bv.bv_val, ',');
+				i = ptr-bv.bv_val;
+				rdns = realloc( rdns, (nrdns+1)*sizeof(char *));
+				rdns[nrdns] = malloc( i+1 );
+				strncpy(rdns[nrdns], bv.bv_val, i );
+				rdns[nrdns][i] = '\0';
+				nrdns++;
+				ber_free( ber, 0 );
+				break;
+			case LDAP_RES_SEARCH_RESULT:
+				done = 1;
+				break;
+			}
+			if ( done )
+				break;
+		}
+		ldap_msgfree( res );
+		if ( done ) break;
+	}
+	ldap_unbind( ld );
+
+	/* Ok, got list of RDNs, now start binding to each */
+	for (i=0; i<maxloop; i++) {
+		char dn[BUFSIZ], *ptr;
+		int j = rand() % nrdns;
+		ptr = lutil_strcopy(dn, rdns[j]);
+		*ptr++ = ',';
+		strcpy(ptr, base);
+		do_bind( uri, host, port, dn, pass, 1 );
+	}
+}