Back to index

openldap  2.4.31
Defines | Functions | Variables
slapd-search.c File Reference
#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"
#include "ldap.h"
#include "lutil.h"
#include "ldap_pvt.h"
#include "slapd-common.h"

Go to the source code of this file.

Defines

#define LOOPS   100
#define RETRIES   0

Functions

static void do_search (char *uri, char *manager, struct berval *passwd, char *sbase, int scope, char *filter, LDAP **ldp, char **attrs, int noattrs, int nobind, int innerloop, int maxretries, int delay, int force, int chaserefs)
static void do_random (char *uri, char *manager, struct berval *passwd, char *sbase, int scope, char *filter, char *attr, char **attrs, int noattrs, int nobind, int innerloop, int maxretries, int delay, int force, int chaserefs)
static void usage (char *name, char o)
int main (int argc, char **argv)

Variables

static int swamp

Define Documentation

#define LOOPS   100

Definition at line 39 of file slapd-search.c.

#define RETRIES   0

Definition at line 40 of file slapd-search.c.


Function Documentation

static void do_random ( char *  uri,
char *  manager,
struct berval passwd,
char *  sbase,
int  scope,
char *  filter,
char *  attr,
char **  attrs,
int  noattrs,
int  nobind,
int  innerloop,
int  maxretries,
int  delay,
int  force,
int  chaserefs 
) [static]

Definition at line 265 of file slapd-search.c.

{
       LDAP   *ld = NULL;
       int    i = 0, do_retry = maxretries;
       char   *attrs[ 2 ];
       int     rc = LDAP_SUCCESS;
       int    version = LDAP_VERSION3;
       int    nvalues = 0;
       char   **values = NULL;
       LDAPMessage *res = NULL, *e = NULL;

       attrs[ 0 ] = attr;
       attrs[ 1 ] = NULL;

       ldap_initialize( &ld, uri );
       if ( ld == NULL ) {
              tester_perror( "ldap_initialize", NULL );
              exit( EXIT_FAILURE );
       }

       (void) ldap_set_option( ld, LDAP_OPT_PROTOCOL_VERSION, &version ); 
       (void) ldap_set_option( ld, LDAP_OPT_REFERRALS,
              chaserefs ? LDAP_OPT_ON : LDAP_OPT_OFF );

       if ( do_retry == maxretries ) {
              fprintf( stderr, "PID=%ld - Search(%d): base=\"%s\", filter=\"%s\" attr=\"%s\".\n",
                            (long) pid, innerloop, sbase, filter, attr );
       }

       if ( nobind == 0 ) {
              rc = ldap_sasl_bind_s( ld, manager, LDAP_SASL_SIMPLE, passwd, NULL, NULL, NULL );
              if ( rc != LDAP_SUCCESS ) {
                     tester_ldap_error( ld, "ldap_sasl_bind_s", NULL );
                     switch ( rc ) {
                     case LDAP_BUSY:
                     case LDAP_UNAVAILABLE:
                     /* fallthru */
                     default:
                            break;
                     }
                     exit( EXIT_FAILURE );
              }
       }

       rc = ldap_search_ext_s( ld, sbase, LDAP_SCOPE_SUBTREE,
              filter, attrs, 0, NULL, NULL, NULL, LDAP_NO_LIMIT, &res );
       switch ( rc ) {
       case LDAP_SIZELIMIT_EXCEEDED:
       case LDAP_TIMELIMIT_EXCEEDED:
       case LDAP_SUCCESS:
              if ( ldap_count_entries( ld, res ) == 0 ) {
                     if ( rc ) {
                            tester_ldap_error( ld, "ldap_search_ext_s", NULL );
                     }
                     break;
              }

              for ( e = ldap_first_entry( ld, res ); e != NULL; e = ldap_next_entry( ld, e ) )
              {
                     struct berval **v = ldap_get_values_len( ld, e, attr );

                     if ( v != NULL ) {
                            int n = ldap_count_values_len( v );
                            int j;

                            values = realloc( values, ( nvalues + n + 1 )*sizeof( char * ) );
                            for ( j = 0; j < n; j++ ) {
                                   values[ nvalues + j ] = strdup( v[ j ]->bv_val );
                            }
                            values[ nvalues + j ] = NULL;
                            nvalues += n;
                            ldap_value_free_len( v );
                     }
              }

              ldap_msgfree( res );

              if ( !values ) {
                     fprintf( stderr, "  PID=%ld - Search base=\"%s\" filter=\"%s\" got %d values.\n",
                            (long) pid, sbase, filter, nvalues );
                     exit(EXIT_FAILURE);
              }

              if ( do_retry == maxretries ) {
                     fprintf( stderr, "  PID=%ld - Search base=\"%s\" filter=\"%s\" got %d values.\n",
                            (long) pid, sbase, filter, nvalues );
              }

              for ( i = 0; i < innerloop; i++ ) {
                     char   buf[ BUFSIZ ];
#if 0  /* use high-order bits for better randomness (Numerical Recipes in "C") */
                     int    r = rand() % nvalues;
#endif
                     int    r = ((double)nvalues)*rand()/(RAND_MAX + 1.0);

                     snprintf( buf, sizeof( buf ), "(%s=%s)", attr, values[ r ] );

                     do_search( uri, manager, passwd,
                            sbase, scope, buf, &ld,
                            srchattrs, noattrs, nobind,
                            1, maxretries, delay, force, chaserefs );
              }
              break;

       default:
              tester_ldap_error( ld, "ldap_search_ext_s", NULL );
              break;
       }

       fprintf( stderr, "  PID=%ld - Search done (%d).\n", (long) pid, rc );

       if ( ld != NULL ) {
              ldap_unbind_ext( ld, NULL, NULL );
       }
}

Here is the call graph for this function:

Here is the caller graph for this function:

static void do_search ( char *  uri,
char *  manager,
struct berval passwd,
char *  sbase,
int  scope,
char *  filter,
LDAP **  ldp,
char **  attrs,
int  noattrs,
int  nobind,
int  innerloop,
int  maxretries,
int  delay,
int  force,
int  chaserefs 
) [static]

Definition at line 385 of file slapd-search.c.

{
       LDAP   *ld = ldp ? *ldp : NULL;
       int    i = 0, do_retry = maxretries;
       int     rc = LDAP_SUCCESS;
       int    version = LDAP_VERSION3;
       char   buf[ BUFSIZ ];
       int           *msgids = NULL, active = 0;

       /* make room for msgid */
       if ( swamp > 1 ) {
              msgids = (int *)calloc( sizeof(int), innerloop );
       }

retry:;
       if ( ld == NULL ) {
              ldap_initialize( &ld, uri );
              if ( ld == NULL ) {
                     tester_perror( "ldap_initialize", NULL );
                     exit( EXIT_FAILURE );
              }

              (void) ldap_set_option( ld, LDAP_OPT_PROTOCOL_VERSION, &version ); 
              (void) ldap_set_option( ld, LDAP_OPT_REFERRALS,
                     chaserefs ? LDAP_OPT_ON : LDAP_OPT_OFF );

              if ( do_retry == maxretries ) {
                     fprintf( stderr,
                            "PID=%ld - Search(%d): "
                            "base=\"%s\" scope=%s filter=\"%s\" "
                            "attrs=%s%s.\n",
                            (long) pid, innerloop,
                            sbase, ldap_pvt_scope2str( scope ), filter,
                            attrs[0], attrs[1] ? " (more...)" : "" );
              }

              if ( nobind == 0 ) {
                     rc = ldap_sasl_bind_s( ld, manager, LDAP_SASL_SIMPLE, passwd, NULL, NULL, NULL );
                     if ( rc != LDAP_SUCCESS ) {
                            snprintf( buf, sizeof( buf ),
                                   "bindDN=\"%s\"", manager );
                            tester_ldap_error( ld, "ldap_sasl_bind_s", buf );
                            switch ( rc ) {
                            case LDAP_BUSY:
                            case LDAP_UNAVAILABLE:
                                   if ( do_retry > 0 ) {
                                          ldap_unbind_ext( ld, NULL, NULL );
                                          ld = NULL;
                                          do_retry--;
                                          if ( delay != 0 ) {
                                              sleep( delay );
                                          }
                                          goto retry;
                                   }
                            /* fallthru */
                            default:
                                   break;
                            }
                            exit( EXIT_FAILURE );
                     }
              }
       }

       if ( swamp > 1 ) {
              do {
                     LDAPMessage *res = NULL;
                     int j, msgid;

                     if ( i < innerloop ) {
                            rc = ldap_search_ext( ld, sbase, scope,
                                          filter, NULL, noattrs, NULL, NULL,
                                          NULL, LDAP_NO_LIMIT, &msgids[i] );

                            active++;
#if 0
                            fprintf( stderr,
                                   ">>> PID=%ld - Search maxloop=%d cnt=%d active=%d msgid=%d: "
                                   "base=\"%s\" scope=%s filter=\"%s\"\n",
                                   (long) pid, innerloop, i, active, msgids[i],
                                   sbase, ldap_pvt_scope2str( scope ), filter );
#endif
                            i++;

                            if ( rc ) {
                                   int first = tester_ignore_err( rc );
                                   /* if ignore.. */
                                   if ( first ) {
                                          /* only log if first occurrence */
                                          if ( ( force < 2 && first > 0 ) || abs(first) == 1 ) {
                                                 tester_ldap_error( ld, "ldap_search_ext", NULL );
                                          }
                                          continue;
                                   }
              
                                   /* busy needs special handling */
                                   snprintf( buf, sizeof( buf ),
                                          "base=\"%s\" filter=\"%s\"\n",
                                          sbase, filter );
                                   tester_ldap_error( ld, "ldap_search_ext", buf );
                                   if ( rc == LDAP_BUSY && do_retry > 0 ) {
                                          ldap_unbind_ext( ld, NULL, NULL );
                                          ld = NULL;
                                          do_retry--;
                                          goto retry;
                                   }
                                   break;
                            }

                            if ( swamp > 2 ) {
                                   continue;
                            }
                     }

                     rc = ldap_result( ld, LDAP_RES_ANY, 0, NULL, &res );
                     switch ( rc ) {
                     case -1:
                            /* gone really bad */
                            goto cleanup;
       
                     case 0:
                            /* timeout (impossible) */
                            break;
       
                     case LDAP_RES_SEARCH_ENTRY:
                     case LDAP_RES_SEARCH_REFERENCE:
                            /* ignore */
                            break;
       
                     case LDAP_RES_SEARCH_RESULT:
                            /* just remove, no error checking (TODO?) */
                            msgid = ldap_msgid( res );
                            ldap_parse_result( ld, res, &rc, NULL, NULL, NULL, NULL, 1 );
                            res = NULL;

                            /* linear search, bah */
                            for ( j = 0; j < i; j++ ) {
                                   if ( msgids[ j ] == msgid ) {
                                          msgids[ j ] = -1;
                                          active--;
#if 0
                                          fprintf( stderr,
                                                 "<<< PID=%ld - SearchDone maxloop=%d cnt=%d active=%d msgid=%d: "
                                                 "base=\"%s\" scope=%s filter=\"%s\"\n",
                                                 (long) pid, innerloop, j, active, msgid,
                                                 sbase, ldap_pvt_scope2str( scope ), filter );
#endif
                                          break;
                                   }
                            }
                            break;

                     default:
                            /* other messages unexpected */
                            fprintf( stderr,
                                   "### PID=%ld - Search(%d): "
                                   "base=\"%s\" scope=%s filter=\"%s\" "
                                   "attrs=%s%s. unexpected response tag=%d\n",
                                   (long) pid, innerloop,
                                   sbase, ldap_pvt_scope2str( scope ), filter,
                                   attrs[0], attrs[1] ? " (more...)" : "", rc );
                            break;
                     }

                     if ( res != NULL ) {
                            ldap_msgfree( res );
                     }
              } while ( i < innerloop || active > 0 );

       } else {
              for ( ; i < innerloop; i++ ) {
                     LDAPMessage *res = NULL;

                     if (swamp) {
                            int msgid;
                            rc = ldap_search_ext( ld, sbase, scope,
                                          filter, NULL, noattrs, NULL, NULL,
                                          NULL, LDAP_NO_LIMIT, &msgid );
                            if ( rc == LDAP_SUCCESS ) continue;
                            else break;
                     }
       
                     rc = ldap_search_ext_s( ld, sbase, scope,
                                   filter, attrs, noattrs, NULL, NULL,
                                   NULL, LDAP_NO_LIMIT, &res );
                     if ( res != NULL ) {
                            ldap_msgfree( res );
                     }
       
                     if ( rc ) {
                            int first = tester_ignore_err( rc );
                            /* if ignore.. */
                            if ( first ) {
                                   /* only log if first occurrence */
                                   if ( ( force < 2 && first > 0 ) || abs(first) == 1 ) {
                                          tester_ldap_error( ld, "ldap_search_ext_s", NULL );
                                   }
                                   continue;
                            }
       
                            /* busy needs special handling */
                            snprintf( buf, sizeof( buf ),
                                   "base=\"%s\" filter=\"%s\"\n",
                                   sbase, filter );
                            tester_ldap_error( ld, "ldap_search_ext_s", buf );
                            if ( rc == LDAP_BUSY && do_retry > 0 ) {
                                   ldap_unbind_ext( ld, NULL, NULL );
                                   ld = NULL;
                                   do_retry--;
                                   goto retry;
                            }
                            break;
                     }
              }
       }

cleanup:;
       if ( msgids != NULL ) {
              free( msgids );
       }

       if ( ldp != NULL ) {
              *ldp = ld;

       } else {
              fprintf( stderr, "  PID=%ld - Search done (%d).\n", (long) pid, rc );

              if ( ld != NULL ) {
                     ldap_unbind_ext( ld, NULL, NULL );
              }
       }
}

Here is the call graph for this function:

int main ( int  argc,
char **  argv 
)

Definition at line 93 of file slapd-search.c.

{
       int           i;
       char          *uri = NULL;
       char          *host = "localhost";
       int           port = -1;
       char          *manager = NULL;
       struct berval passwd = { 0, NULL };
       char          *sbase = NULL;
       int           scope = LDAP_SCOPE_SUBTREE;
       char          *filter  = NULL;
       char          *attr = NULL;
       char          *srchattrs[] = { "cn", "sn", NULL };
       char          **attrs = srchattrs;
       int           loops = LOOPS;
       int           outerloops = 1;
       int           retries = RETRIES;
       int           delay = 0;
       int           force = 0;
       int           chaserefs = 0;
       int           noattrs = 0;
       int           nobind = 0;

       tester_init( "slapd-search", TESTER_SEARCH );

       /* by default, tolerate referrals and no such object */
       tester_ignore_str2errlist( "REFERRAL,NO_SUCH_OBJECT" );

       while ( ( i = getopt( argc, argv, "Aa:b:CD:f:FH:h:i:l:L:Np:r:Ss:t:T:w:" ) ) != EOF )
       {
              switch ( i ) {
              case 'A':
                     noattrs++;
                     break;

              case 'C':
                     chaserefs++;
                     break;

              case 'H':            /* the server uri */
                     uri = strdup( optarg );
                     break;

              case 'h':            /* the servers host */
                     host = strdup( optarg );
                     break;

              case 'i':
                     tester_ignore_str2errlist( optarg );
                     break;

              case 'N':
                     nobind++;
                     break;

              case 'p':            /* the servers port */
                     if ( lutil_atoi( &port, optarg ) != 0 ) {
                            usage( argv[0], i );
                     }
                     break;

              case 'D':            /* the servers manager */
                     manager = strdup( optarg );
                     break;

              case 'w':            /* the server managers password */
                     passwd.bv_val = strdup( optarg );
                     passwd.bv_len = strlen( optarg );
                     memset( optarg, '*', passwd.bv_len );
                     break;

              case 'a':
                     attr = strdup( optarg );
                     break;

              case 'b':            /* file with search base */
                     sbase = strdup( optarg );
                     break;

              case 'f':            /* the search request */
                     filter = strdup( optarg );
                     break;

              case 'F':
                     force++;
                     break;

              case 'l':            /* number of loops */
                     if ( lutil_atoi( &loops, optarg ) != 0 ) {
                            usage( argv[0], i );
                     }
                     break;

              case 'L':            /* number of loops */
                     if ( lutil_atoi( &outerloops, optarg ) != 0 ) {
                            usage( argv[0], i );
                     }
                     break;

              case 'r':            /* number of retries */
                     if ( lutil_atoi( &retries, optarg ) != 0 ) {
                            usage( argv[0], i );
                     }
                     break;

              case 't':            /* delay in seconds */
                     if ( lutil_atoi( &delay, optarg ) != 0 ) {
                            usage( argv[0], i );
                     }
                     break;

              case 'T':
                     attrs = ldap_str2charray( optarg, "," );
                     if ( attrs == NULL ) {
                            usage( argv[0], i );
                     }
                     break;

              case 'S':
                     swamp++;
                     break;

              case 's':
                     scope = ldap_pvt_str2scope( optarg );
                     if ( scope == -1 ) {
                            usage( argv[0], i );
                     }
                     break;

              default:
                     usage( argv[0], i );
                     break;
              }
       }

       if (( sbase == NULL ) || ( filter == NULL ) || ( port == -1 && uri == NULL ))
              usage( argv[0], '\0' );

       if ( *filter == '\0' ) {

              fprintf( stderr, "%s: invalid EMPTY search filter.\n",
                            argv[0] );
              exit( EXIT_FAILURE );

       }

       if ( argv[optind] != NULL ) {
              attrs = &argv[optind];
       }

       uri = tester_uri( uri, host, port );

       for ( i = 0; i < outerloops; i++ ) {
              if ( attr != NULL ) {
                     do_random( uri, manager, &passwd,
                            sbase, scope, filter, attr,
                            attrs, noattrs, nobind,
                            loops, retries, delay, force, chaserefs );

              } else {
                     do_search( uri, manager, &passwd,
                            sbase, scope, filter, NULL,
                            attrs, noattrs, nobind,
                            loops, retries, delay, force, chaserefs );
              }
       }

       exit( EXIT_SUCCESS );
}

Here is the call graph for this function:

static void usage ( char *  name,
char  o 
) [static]

Definition at line 55 of file slapd-search.c.

{
       if ( o != '\0' ) {
              fprintf( stderr, "unknown/incorrect option \"%c\"\n", o );
       }

        fprintf( stderr,
              "usage: %s "
              "-H <uri> | ([-h <host>] -p <port>) "
              "-D <manager> "
              "-w <passwd> "
              "-b <searchbase> "
              "-s <scope> "
              "-f <searchfilter> "
              "[-a <attr>] "
              "[-A] "
              "[-C] "
              "[-F] "
              "[-N] "
              "[-S[S[S]]] "
              "[-i <ignore>] "
              "[-l <loops>] "
              "[-L <outerloops>] "
              "[-r <maxretries>] "
              "[-t <delay>] "
              "[<attrs>] "
              "\n",
                     name );
       exit( EXIT_FAILURE );
}

Variable Documentation

int swamp [static]

Definition at line 90 of file slapd-search.c.