git.sur5r.net Git - openldap/blob - tests/progs/slapd-tester.c

   1 /* $OpenLDAP$ */
   2 /* This work is part of OpenLDAP Software <http://www.openldap.org/>.
   3  *
   4  * Copyright 1999-2006 The OpenLDAP Foundation.
   5  * All rights reserved.
   6  *
   7  * Redistribution and use in source and binary forms, with or without
   8  * modification, are permitted only as authorized by the OpenLDAP
   9  * Public License.
  10  *
  11  * A copy of this license is available in file LICENSE in the
  12  * top-level directory of the distribution or, alternatively, at
  13  * <http://www.OpenLDAP.org/license.html>.
  14  */
  15 /* ACKNOWLEDGEMENTS:
  16  * This work was initially developed by Kurt Spanier for inclusion
  17  * in OpenLDAP Software.
  18  */
  19
  20 #include "portable.h"
  21
  22 #include <stdio.h>
  23
  24 #include <ac/stdlib.h>
  25
  26 #include <ac/ctype.h>
  27 #include <ac/dirent.h>
  28 #include <ac/param.h>
  29 #include <ac/socket.h>
  30 #include <ac/string.h>
  31 #include <ac/unistd.h>
  32 #include <ac/wait.h>
  33
  34
  35 #include "ldap_defaults.h"
  36 #include "lutil.h"
  37
  38 #include "ldap.h"
  39 #include "ldap_pvt.h"
  40 #include "lber_pvt.h"
  41 #include "slapd-common.h"
  42
  43 #define SEARCHCMD               "slapd-search"
  44 #define READCMD                 "slapd-read"
  45 #define ADDCMD                  "slapd-addel"
  46 #define MODRDNCMD               "slapd-modrdn"
  47 #define MODIFYCMD               "slapd-modify"
  48 #define BINDCMD                 "slapd-bind"
  49 #define MAXARGS                 100
  50 #define MAXREQS                 5000
  51 #define LOOPS                   100
  52 #define OUTERLOOPS              "1"
  53 #define RETRIES                 "0"
  54
  55 #define TSEARCHFILE             "do_search.0"
  56 #define TREADFILE               "do_read.0"
  57 #define TADDFILE                "do_add."
  58 #define TMODRDNFILE             "do_modrdn.0"
  59 #define TMODIFYFILE             "do_modify.0"
  60 #define TBINDFILE               "do_bind.0"
  61
  62 static char *get_file_name( char *dirname, char *filename );
  63 static int  get_search_filters( char *filename, char *filters[], char *attrs[], char *bases[] );
  64 static int  get_read_entries( char *filename, char *entries[], char *filters[] );
  65 static void fork_child( char *prog, char **args );
  66 static void     wait4kids( int nkidval );
  67
  68 static int      maxkids = 20;
  69 static int      nkids;
  70
  71 #ifdef HAVE_WINSOCK
  72 static HANDLE   *children;
  73 static char argbuf[BUFSIZ];
  74 #define ArgDup(x) strdup(strcat(strcat(strcpy(argbuf,"\""),x),"\""))
  75 #else
  76 #define ArgDup(x) strdup(x)
  77 #endif
  78
  79 static void
  80 usage( char *name, char opt )
  81 {
  82         if ( opt ) {
  83                 fprintf( stderr, "%s: unable to handle option \'%c\'\n\n",
  84                         name, opt );
  85         }
  86
  87         fprintf( stderr,
  88                 "usage: %s "
  89                 "-H <uri> | ([-h <host>] -p <port>) "
  90                 "-D <manager> "
  91                 "-w <passwd> "
  92                 "-d <datadir> "
  93                 "[-i <ignore>] "
  94                 "[-j <maxchild>] "
  95                 "[-l {<loops>|<type>=<loops>[,...]}] "
  96                 "[-L <outerloops>] "
  97                 "-P <progdir> "
  98                 "[-r <maxretries>] "
  99                 "[-t <delay>] "
 100                 "[-C] "
 101                 "[-F] "
 102                 "[-I] "
 103                 "[-N]\n",
 104                 name );
 105         exit( EXIT_FAILURE );
 106 }
 107
 108 int
 109 main( int argc, char **argv )
 110 {
 111         int             i, j;
 112         char            *uri = NULL;
 113         char            *host = "localhost";
 114         char            *port = NULL;
 115         char            *manager = NULL;
 116         char            *passwd = NULL;
 117         char            *dirname = NULL;
 118         char            *progdir = NULL;
 119         int             loops = LOOPS;
 120         char            *outerloops = OUTERLOOPS;
 121         char            *retries = RETRIES;
 122         char            *delay = "0";
 123         DIR             *datadir;
 124         struct dirent   *file;
 125         int             friendly = 0;
 126         int             chaserefs = 0;
 127         int             noattrs = 0;
 128         int             nobind = 0;
 129         int             noinit = 1;
 130         char            *ignore = NULL;
 131         /* search */
 132         char            *sfile = NULL;
 133         char            *sreqs[MAXREQS];
 134         char            *sattrs[MAXREQS];
 135         char            *sbase[MAXREQS];
 136         int             snum = 0;
 137         char            *sargs[MAXARGS];
 138         int             sanum;
 139         char            scmd[MAXPATHLEN];
 140         char            sloops[] = "18446744073709551615UL";
 141         /* read */
 142         char            *rfile = NULL;
 143         char            *rreqs[MAXREQS];
 144         int             rnum = 0;
 145         char            *rargs[MAXARGS];
 146         char            *rflts[MAXREQS];
 147         int             ranum;
 148         char            rcmd[MAXPATHLEN];
 149         char            rloops[] = "18446744073709551615UL";
 150         /* addel */
 151         char            *afiles[MAXREQS];
 152         int             anum = 0;
 153         char            *aargs[MAXARGS];
 154         int             aanum;
 155         char            acmd[MAXPATHLEN];
 156         char            aloops[] = "18446744073709551615UL";
 157         /* modrdn */
 158         char            *nfile = NULL;
 159         char            *nreqs[MAXREQS];
 160         int             nnum = 0;
 161         char            *nargs[MAXARGS];
 162         int             nanum;
 163         char            ncmd[MAXPATHLEN];
 164         char            nloops[] = "18446744073709551615UL";
 165         /* modify */
 166         char            *mfile = NULL;
 167         char            *mreqs[MAXREQS];
 168         char            *mdn[MAXREQS];
 169         int             mnum = 0;
 170         char            *margs[MAXARGS];
 171         int             manum;
 172         char            mcmd[MAXPATHLEN];
 173         char            mloops[] = "18446744073709551615UL";
 174         /* bind */
 175         char            *bfile = NULL;
 176         char            *breqs[MAXREQS];
 177         char            *bcreds[MAXREQS];
 178         char            *battrs[MAXREQS];
 179         int             bnum = 0;
 180         char            *bargs[MAXARGS];
 181         int             banum;
 182         char            bcmd[MAXPATHLEN];
 183         char            bloops[] = "18446744073709551615UL";
 184         char            **bargs_extra = NULL;
 185
 186         char            *friendlyOpt = NULL;
 187         int             pw_ask = 0;
 188         char            *pw_file = NULL;
 189
 190         /* extra action to do after bind... */
 191         typedef struct extra_t {
 192                 char            *action;
 193                 struct extra_t  *next;
 194         }               extra_t;
 195
 196         extra_t         *extra = NULL;
 197         int             nextra = 0;
 198
 199         tester_init( "slapd-tester", TESTER_TESTER );
 200
 201         sloops[0] = '\0';
 202         rloops[0] = '\0';
 203         aloops[0] = '\0';
 204         nloops[0] = '\0';
 205         mloops[0] = '\0';
 206         bloops[0] = '\0';
 207
 208         while ( (i = getopt( argc, argv, "AB:CD:d:FH:h:Ii:j:l:L:NP:p:r:t:Ww:y:" )) != EOF ) {
 209                 switch( i ) {
 210                 case 'A':
 211                         noattrs++;
 212                         break;
 213
 214                 case 'B':
 215                         {
 216                         char    **p,
 217                                 **b = ldap_str2charray( optarg, "," );
 218                         extra_t **epp;
 219
 220                         for ( epp = &extra; *epp; epp = &(*epp)->next )
 221                                 ;
 222
 223                         for ( p = b; p[0]; p++ ) {
 224                                 *epp = calloc( 1, sizeof( extra_t ) );
 225                                 (*epp)->action = p[0];
 226                                 epp = &(*epp)->next;
 227                                 nextra++;
 228                         }
 229
 230                         ldap_memfree( b );
 231                         } break;
 232
 233                 case 'C':
 234                         chaserefs++;
 235                         break;
 236
 237                 case 'D':               /* slapd manager */
 238                         manager = ArgDup( optarg );
 239                         break;
 240
 241                 case 'd':               /* data directory */
 242                         dirname = strdup( optarg );
 243                         break;
 244
 245                 case 'F':
 246                         friendly++;
 247                         break;
 248
 249                 case 'H':               /* slapd uri */
 250                         uri = strdup( optarg );
 251                         break;
 252
 253                 case 'h':               /* slapd host */
 254                         host = strdup( optarg );
 255                         break;
 256
 257                 case 'I':
 258                         noinit = 0;
 259                         break;
 260
 261                 case 'i':
 262                         ignore = optarg;
 263                         break;
 264
 265                 case 'j':               /* the number of parallel clients */
 266                         if ( lutil_atoi( &maxkids, optarg ) != 0 ) {
 267                                 usage( argv[0], 'j' );
 268                         }
 269                         break;
 270
 271                 case 'l':               /* the number of loops per client */
 272                         if ( !isdigit( optarg[0] ) ) {
 273                                 char    **p,
 274                                         **l = ldap_str2charray( optarg, "," );
 275
 276                                 for ( p = l; p[0]; p++) {
 277                                         struct {
 278                                                 struct berval   type;
 279                                                 char            *buf;
 280                                         } types[] = {
 281                                                 { BER_BVC( "add=" ),    aloops },
 282                                                 { BER_BVC( "bind=" ),   bloops },
 283                                                 { BER_BVC( "modify=" ), mloops },
 284                                                 { BER_BVC( "modrdn=" ), nloops },
 285                                                 { BER_BVC( "read=" ),   rloops },
 286                                                 { BER_BVC( "search=" ), sloops },
 287                                                 { BER_BVNULL,           NULL }
 288                                         };
 289                                         int     c, n;
 290
 291                                         for ( c = 0; types[c].type.bv_val; c++ ) {
 292                                                 if ( strncasecmp( p[0], types[c].type.bv_val, types[c].type.bv_len ) == 0 ) {
 293                                                         break;
 294                                                 }
 295                                         }
 296
 297                                         if ( types[c].type.bv_val == NULL ) {
 298                                                 usage( argv[0], 'l' );
 299                                         }
 300
 301                                         if ( lutil_atoi( &n, &p[0][types[c].type.bv_len] ) != 0 ) {
 302                                                 usage( argv[0], 'l' );
 303                                         }
 304
 305                                         snprintf( types[c].buf, sizeof( aloops ), "%d", n );
 306                                 }
 307
 308                                 ldap_charray_free( l );
 309
 310                         } else if ( lutil_atoi( &loops, optarg ) != 0 ) {
 311                                 usage( argv[0], 'l' );
 312                         }
 313                         break;
 314
 315                 case 'L':               /* the number of outerloops per client */
 316                         outerloops = strdup( optarg );
 317                         break;
 318
 319                 case 'N':
 320                         nobind++;
 321                         break;
 322
 323                 case 'P':               /* prog directory */
 324                         progdir = strdup( optarg );
 325                         break;
 326
 327                 case 'p':               /* the servers port number */
 328                         port = strdup( optarg );
 329                         break;
 330
 331                 case 'r':               /* the number of retries in case of error */
 332                         retries = strdup( optarg );
 333                         break;
 334
 335                 case 't':               /* the delay in seconds between each retry */
 336                         delay = strdup( optarg );
 337                         break;
 338
 339                 case 'w':               /* the managers passwd */
 340                         passwd = ArgDup( optarg );
 341                         memset( optarg, '*', strlen( optarg ) );
 342                         break;
 343
 344                 case 'W':
 345                         pw_ask++;
 346                         break;
 347
 348                 case 'y':
 349                         pw_file = optarg;
 350                         break;
 351
 352                 default:
 353                         usage( argv[0], '\0' );
 354                         break;
 355                 }
 356         }
 357
 358         if (( dirname == NULL ) || ( port == NULL && uri == NULL ) ||
 359                         ( manager == NULL ) || ( passwd == NULL ) || ( progdir == NULL ))
 360                 usage( argv[0], '\0' );
 361
 362 #ifdef HAVE_WINSOCK
 363         children = malloc( maxkids * sizeof(HANDLE) );
 364 #endif
 365         /* get the file list */
 366         if ( ( datadir = opendir( dirname )) == NULL ) {
 367                 fprintf( stderr, "%s: couldn't open data directory \"%s\".\n",
 368                                         argv[0], dirname );
 369                 exit( EXIT_FAILURE );
 370         }
 371
 372         /*  look for search, read, modrdn, and add/delete files */
 373         for ( file = readdir( datadir ); file; file = readdir( datadir )) {
 374
 375                 if ( !strcasecmp( file->d_name, TSEARCHFILE )) {
 376                         sfile = get_file_name( dirname, file->d_name );
 377                         continue;
 378                 } else if ( !strcasecmp( file->d_name, TREADFILE )) {
 379                         rfile = get_file_name( dirname, file->d_name );
 380                         continue;
 381                 } else if ( !strcasecmp( file->d_name, TMODRDNFILE )) {
 382                         nfile = get_file_name( dirname, file->d_name );
 383                         continue;
 384                 } else if ( !strcasecmp( file->d_name, TMODIFYFILE )) {
 385                         mfile = get_file_name( dirname, file->d_name );
 386                         continue;
 387                 } else if ( !strncasecmp( file->d_name, TADDFILE, strlen( TADDFILE ))
 388                         && ( anum < MAXREQS )) {
 389                         afiles[anum++] = get_file_name( dirname, file->d_name );
 390                         continue;
 391                 } else if ( !strcasecmp( file->d_name, TBINDFILE )) {
 392                         bfile = get_file_name( dirname, file->d_name );
 393                         continue;
 394                 }
 395         }
 396
 397         closedir( datadir );
 398
 399         if ( pw_ask ) {
 400                 passwd = getpassphrase( _("Enter LDAP Password: ") );
 401
 402         } else if ( pw_file ) {
 403                 struct berval   pw;
 404
 405                 if ( lutil_get_filed_password( pw_file, &pw ) ) {
 406                         exit( EXIT_FAILURE );
 407                 }
 408
 409                 passwd = pw.bv_val;
 410         }
 411
 412         /* look for search requests */
 413         if ( sfile ) {
 414                 snum = get_search_filters( sfile, sreqs, sattrs, sbase );
 415         }
 416
 417         /* look for read requests */
 418         if ( rfile ) {
 419                 rnum = get_read_entries( rfile, rreqs, rflts );
 420         }
 421
 422         /* look for modrdn requests */
 423         if ( nfile ) {
 424                 nnum = get_read_entries( nfile, nreqs, NULL );
 425         }
 426
 427         /* look for modify requests */
 428         if ( mfile ) {
 429                 mnum = get_search_filters( mfile, mreqs, NULL, mdn );
 430         }
 431
 432         /* look for bind requests */
 433         if ( bfile ) {
 434                 bnum = get_search_filters( bfile, bcreds, battrs, breqs );
 435         }
 436
 437         /* setup friendly option */
 438
 439         switch ( friendly ) {
 440         case 0:
 441                 break;
 442
 443         case 1:
 444                 friendlyOpt = "-F";
 445                 break;
 446
 447         default:
 448                 /* NOTE: right now we don't need it more than twice */
 449         case 2:
 450                 friendlyOpt = "-FF";
 451                 break;
 452         }
 453
 454         if ( sloops[0] == '\0' ) snprintf( sloops, sizeof( sloops ), "%d", 10 * loops );
 455         if ( rloops[0] == '\0' ) snprintf( rloops, sizeof( rloops ), "%d", 20 * loops );
 456         if ( aloops[0] == '\0' ) snprintf( aloops, sizeof( aloops ), "%d", loops );
 457         if ( nloops[0] == '\0' ) snprintf( nloops, sizeof( nloops ), "%d", loops );
 458         if ( mloops[0] == '\0' ) snprintf( mloops, sizeof( mloops ), "%d", loops );
 459         if ( bloops[0] == '\0' ) snprintf( bloops, sizeof( bloops ), "%d", 20 * loops );
 460
 461         /*
 462          * generate the search clients
 463          */
 464
 465         sanum = 0;
 466         snprintf( scmd, sizeof scmd, "%s" LDAP_DIRSEP SEARCHCMD,
 467                 progdir );
 468         sargs[sanum++] = scmd;
 469         if ( uri ) {
 470                 sargs[sanum++] = "-H";
 471                 sargs[sanum++] = uri;
 472         } else {
 473                 sargs[sanum++] = "-h";
 474                 sargs[sanum++] = host;
 475                 sargs[sanum++] = "-p";
 476                 sargs[sanum++] = port;
 477         }
 478         sargs[sanum++] = "-D";
 479         sargs[sanum++] = manager;
 480         sargs[sanum++] = "-w";
 481         sargs[sanum++] = passwd;
 482         sargs[sanum++] = "-l";
 483         sargs[sanum++] = sloops;
 484         sargs[sanum++] = "-L";
 485         sargs[sanum++] = outerloops;
 486         sargs[sanum++] = "-r";
 487         sargs[sanum++] = retries;
 488         sargs[sanum++] = "-t";
 489         sargs[sanum++] = delay;
 490         if ( friendly ) {
 491                 sargs[sanum++] = friendlyOpt;
 492         }
 493         if ( chaserefs ) {
 494                 sargs[sanum++] = "-C";
 495         }
 496         if ( noattrs ) {
 497                 sargs[sanum++] = "-A";
 498         }
 499         if ( nobind ) {
 500                 sargs[sanum++] = "-N";
 501         }
 502         if ( ignore ) {
 503                 sargs[sanum++] = "-i";
 504                 sargs[sanum++] = ignore;
 505         }
 506         sargs[sanum++] = "-b";
 507         sargs[sanum++] = NULL;          /* will hold the search base */
 508         sargs[sanum++] = "-f";
 509         sargs[sanum++] = NULL;          /* will hold the search request */
 510
 511         sargs[sanum++] = NULL;
 512         sargs[sanum] = NULL;            /* might hold the "attr" request */
 513
 514         sargs[sanum + 1] = NULL;
 515
 516         /*
 517          * generate the read clients
 518          */
 519
 520         ranum = 0;
 521         snprintf( rcmd, sizeof rcmd, "%s" LDAP_DIRSEP READCMD,
 522                 progdir );
 523         rargs[ranum++] = rcmd;
 524         if ( uri ) {
 525                 rargs[ranum++] = "-H";
 526                 rargs[ranum++] = uri;
 527         } else {
 528                 rargs[ranum++] = "-h";
 529                 rargs[ranum++] = host;
 530                 rargs[ranum++] = "-p";
 531                 rargs[ranum++] = port;
 532         }
 533         rargs[ranum++] = "-D";
 534         rargs[ranum++] = manager;
 535         rargs[ranum++] = "-w";
 536         rargs[ranum++] = passwd;
 537         rargs[ranum++] = "-l";
 538         rargs[ranum++] = rloops;
 539         rargs[ranum++] = "-L";
 540         rargs[ranum++] = outerloops;
 541         rargs[ranum++] = "-r";
 542         rargs[ranum++] = retries;
 543         rargs[ranum++] = "-t";
 544         rargs[ranum++] = delay;
 545         if ( friendly ) {
 546                 rargs[ranum++] = friendlyOpt;
 547         }
 548         if ( chaserefs ) {
 549                 rargs[ranum++] = "-C";
 550         }
 551         if ( noattrs ) {
 552                 rargs[ranum++] = "-A";
 553         }
 554         if ( ignore ) {
 555                 rargs[ranum++] = "-i";
 556                 rargs[ranum++] = ignore;
 557         }
 558         rargs[ranum++] = "-e";
 559         rargs[ranum++] = NULL;          /* will hold the read entry */
 560
 561         rargs[ranum++] = NULL;
 562         rargs[ranum] = NULL;            /* might hold the filter arg */
 563
 564         rargs[ranum + 1] = NULL;
 565
 566         /*
 567          * generate the modrdn clients
 568          */
 569
 570         nanum = 0;
 571         snprintf( ncmd, sizeof ncmd, "%s" LDAP_DIRSEP MODRDNCMD,
 572                 progdir );
 573         nargs[nanum++] = ncmd;
 574         if ( uri ) {
 575                 nargs[nanum++] = "-H";
 576                 nargs[nanum++] = uri;
 577         } else {
 578                 nargs[nanum++] = "-h";
 579                 nargs[nanum++] = host;
 580                 nargs[nanum++] = "-p";
 581                 nargs[nanum++] = port;
 582         }
 583         nargs[nanum++] = "-D";
 584         nargs[nanum++] = manager;
 585         nargs[nanum++] = "-w";
 586         nargs[nanum++] = passwd;
 587         nargs[nanum++] = "-l";
 588         nargs[nanum++] = nloops;
 589         nargs[nanum++] = "-L";
 590         nargs[nanum++] = outerloops;
 591         nargs[nanum++] = "-r";
 592         nargs[nanum++] = retries;
 593         nargs[nanum++] = "-t";
 594         nargs[nanum++] = delay;
 595         if ( friendly ) {
 596                 nargs[nanum++] = friendlyOpt;
 597         }
 598         if ( chaserefs ) {
 599                 nargs[nanum++] = "-C";
 600         }
 601         if ( ignore ) {
 602                 nargs[nanum++] = "-i";
 603                 nargs[nanum++] = ignore;
 604         }
 605         nargs[nanum++] = "-e";
 606         nargs[nanum++] = NULL;          /* will hold the modrdn entry */
 607         nargs[nanum++] = NULL;
 608
 609         /*
 610          * generate the modify clients
 611          */
 612
 613         manum = 0;
 614         snprintf( mcmd, sizeof mcmd, "%s" LDAP_DIRSEP MODIFYCMD,
 615                 progdir );
 616         margs[manum++] = mcmd;
 617         if ( uri ) {
 618                 margs[manum++] = "-H";
 619                 margs[manum++] = uri;
 620         } else {
 621                 margs[manum++] = "-h";
 622                 margs[manum++] = host;
 623                 margs[manum++] = "-p";
 624                 margs[manum++] = port;
 625         }
 626         margs[manum++] = "-D";
 627         margs[manum++] = manager;
 628         margs[manum++] = "-w";
 629         margs[manum++] = passwd;
 630         margs[manum++] = "-l";
 631         margs[manum++] = mloops;
 632         margs[manum++] = "-L";
 633         margs[manum++] = outerloops;
 634         margs[manum++] = "-r";
 635         margs[manum++] = retries;
 636         margs[manum++] = "-t";
 637         margs[manum++] = delay;
 638         if ( friendly ) {
 639                 margs[manum++] = friendlyOpt;
 640         }
 641         if ( chaserefs ) {
 642                 margs[manum++] = "-C";
 643         }
 644         if ( ignore ) {
 645                 margs[manum++] = "-i";
 646                 margs[manum++] = ignore;
 647         }
 648         margs[manum++] = "-e";
 649         margs[manum++] = NULL;          /* will hold the modify entry */
 650         margs[manum++] = "-a";;
 651         margs[manum++] = NULL;          /* will hold the ava */
 652         margs[manum++] = NULL;
 653
 654         /*
 655          * generate the add/delete clients
 656          */
 657
 658         aanum = 0;
 659         snprintf( acmd, sizeof acmd, "%s" LDAP_DIRSEP ADDCMD,
 660                 progdir );
 661         aargs[aanum++] = acmd;
 662         if ( uri ) {
 663                 aargs[aanum++] = "-H";
 664                 aargs[aanum++] = uri;
 665         } else {
 666                 aargs[aanum++] = "-h";
 667                 aargs[aanum++] = host;
 668                 aargs[aanum++] = "-p";
 669                 aargs[aanum++] = port;
 670         }
 671         aargs[aanum++] = "-D";
 672         aargs[aanum++] = manager;
 673         aargs[aanum++] = "-w";
 674         aargs[aanum++] = passwd;
 675         aargs[aanum++] = "-l";
 676         aargs[aanum++] = aloops;
 677         aargs[aanum++] = "-L";
 678         aargs[aanum++] = outerloops;
 679         aargs[aanum++] = "-r";
 680         aargs[aanum++] = retries;
 681         aargs[aanum++] = "-t";
 682         aargs[aanum++] = delay;
 683         if ( friendly ) {
 684                 aargs[aanum++] = friendlyOpt;
 685         }
 686         if ( chaserefs ) {
 687                 aargs[aanum++] = "-C";
 688         }
 689         if ( ignore ) {
 690                 aargs[aanum++] = "-i";
 691                 aargs[aanum++] = ignore;
 692         }
 693         aargs[aanum++] = "-f";
 694         aargs[aanum++] = NULL;          /* will hold the add data file */
 695         aargs[aanum++] = NULL;
 696
 697         /*
 698          * generate the bind clients
 699          */
 700
 701         banum = 0;
 702         snprintf( bcmd, sizeof bcmd, "%s" LDAP_DIRSEP BINDCMD,
 703                 progdir );
 704         bargs[banum++] = bcmd;
 705         if ( !noinit ) {
 706                 bargs[banum++] = "-I";  /* init on each bind */
 707         }
 708         if ( uri ) {
 709                 bargs[banum++] = "-H";
 710                 bargs[banum++] = uri;
 711         } else {
 712                 bargs[banum++] = "-h";
 713                 bargs[banum++] = host;
 714                 bargs[banum++] = "-p";
 715                 bargs[banum++] = port;
 716         }
 717         bargs[banum++] = "-l";
 718         bargs[banum++] = bloops;
 719         bargs[banum++] = "-L";
 720         bargs[banum++] = outerloops;
 721 #if 0
 722         bargs[banum++] = "-r";
 723         bargs[banum++] = retries;
 724         bargs[banum++] = "-t";
 725         bargs[banum++] = delay;
 726 #endif
 727         if ( friendly ) {
 728                 bargs[banum++] = friendlyOpt;
 729         }
 730         if ( chaserefs ) {
 731                 bargs[banum++] = "-C";
 732         }
 733         if ( ignore ) {
 734                 bargs[banum++] = "-i";
 735                 bargs[banum++] = ignore;
 736         }
 737         if ( nextra ) {
 738                 bargs[banum++] = "-B";
 739                 bargs_extra = &bargs[banum++];
 740         }
 741         bargs[banum++] = "-D";
 742         bargs[banum++] = NULL;
 743         bargs[banum++] = "-w";
 744         bargs[banum++] = NULL;
 745         bargs[banum++] = NULL;
 746
 747 #define DOREQ(n,j) ((n) && ((maxkids > (n)) ? ((j) < maxkids ) : ((j) < (n))))
 748
 749         for ( j = 0; j < MAXREQS; j++ ) {
 750                 if ( DOREQ( snum, j ) ) {
 751                         int     jj = j % snum;
 752
 753                         sargs[sanum - 2] = sreqs[jj];
 754                         sargs[sanum - 4] = sbase[jj];
 755                         if ( sattrs[jj] != NULL ) {
 756                                 sargs[sanum - 1] = "-a";
 757                                 sargs[sanum] = sattrs[jj];
 758
 759                         } else {
 760                                 sargs[sanum - 1] = NULL;
 761                         }
 762                         fork_child( scmd, sargs );
 763                 }
 764
 765                 if ( DOREQ( rnum, j ) ) {
 766                         int     jj = j % rnum;
 767
 768                         rargs[ranum - 2] = rreqs[jj];
 769                         if ( rflts[jj] != NULL ) {
 770                                 rargs[ranum - 1] = "-f";
 771                                 rargs[ranum] = rflts[jj];
 772
 773                         } else {
 774                                 rargs[ranum - 1] = NULL;
 775                         }
 776                         fork_child( rcmd, rargs );
 777                 }
 778
 779                 if ( j < nnum ) {
 780                         nargs[nanum - 2] = nreqs[j];
 781                         fork_child( ncmd, nargs );
 782                 }
 783
 784                 if ( j < mnum ) {
 785                         margs[manum - 4] = mdn[j];
 786                         margs[manum - 2] = mreqs[j];
 787                         fork_child( mcmd, margs );
 788                 }
 789
 790                 if ( j < anum ) {
 791                         aargs[aanum - 2] = afiles[j];
 792                         fork_child( acmd, aargs );
 793                 }
 794
 795                 if ( DOREQ( bnum, j ) ) {
 796                         int     jj = j % bnum;
 797
 798                         if ( nextra ) {
 799                                 int     n = ((double)nextra)*rand()/(RAND_MAX + 1.0);
 800                                 extra_t *e;
 801
 802                                 for ( e = extra; n-- > 0; e = e->next )
 803                                         ;
 804                                 *bargs_extra = e->action;
 805                         }
 806
 807                         if ( battrs[jj] != NULL ) {
 808                                 bargs[banum - 4] = manager ? manager : "";
 809                                 bargs[banum - 2] = passwd ? passwd : "";
 810
 811                                 bargs[banum - 1] = "-b";
 812                                 bargs[banum] = breqs[jj];
 813                                 bargs[banum + 1] = "-f";
 814                                 bargs[banum + 2] = bcreds[jj];
 815                                 bargs[banum + 3] = "-a";
 816                                 bargs[banum + 4] = battrs[jj];
 817                         } else {
 818                                 bargs[banum - 4] = breqs[jj];
 819                                 bargs[banum - 2] = bcreds[jj];
 820                                 bargs[banum - 1] = NULL;
 821                         }
 822
 823                         fork_child( bcmd, bargs );
 824                         bargs[banum - 1] = NULL;
 825                 }
 826         }
 827
 828         wait4kids( -1 );
 829
 830         exit( EXIT_SUCCESS );
 831 }
 832
 833 static char *
 834 get_file_name( char *dirname, char *filename )
 835 {
 836         char buf[MAXPATHLEN];
 837
 838         snprintf( buf, sizeof buf, "%s" LDAP_DIRSEP "%s",
 839                 dirname, filename );
 840         return( strdup( buf ));
 841 }
 842
 843
 844 static int
 845 get_search_filters( char *filename, char *filters[], char *attrs[], char *bases[] )
 846 {
 847         FILE    *fp;
 848         int     filter = 0;
 849
 850         if ( (fp = fopen( filename, "r" )) != NULL ) {
 851                 char  line[BUFSIZ];
 852
 853                 while (( filter < MAXREQS ) && ( fgets( line, BUFSIZ, fp ))) {
 854                         char *nl;
 855
 856                         if (( nl = strchr( line, '\r' )) || ( nl = strchr( line, '\n' )))
 857                                 *nl = '\0';
 858                         bases[filter] = ArgDup( line );
 859                         fgets( line, BUFSIZ, fp );
 860                         if (( nl = strchr( line, '\r' )) || ( nl = strchr( line, '\n' )))
 861                                 *nl = '\0';
 862
 863                         filters[filter] = ArgDup( line );
 864                         if ( attrs ) {
 865                                 if ( filters[filter][0] == '+') {
 866                                         char    *sep = strchr( filters[filter], ':' );
 867
 868                                         if ( sep != NULL ) {
 869                                                 attrs[ filter ] = &filters[ filter ][ 1 ];
 870                                                 sep[ 0 ] = '\0';
 871                                                 /* NOTE: don't free this! */
 872                                                 filters[ filter ] = &sep[ 1 ];
 873                                         }
 874
 875                                 } else {
 876                                         attrs[ filter] = NULL;
 877                                 }
 878                         }
 879                         filter++;
 880
 881                 }
 882                 fclose( fp );
 883         }
 884
 885         return( filter );
 886 }
 887
 888
 889 static int
 890 get_read_entries( char *filename, char *entries[], char *filters[] )
 891 {
 892         FILE    *fp;
 893         int     entry = 0;
 894
 895         if ( (fp = fopen( filename, "r" )) != NULL ) {
 896                 char  line[BUFSIZ];
 897
 898                 while (( entry < MAXREQS ) && ( fgets( line, BUFSIZ, fp ))) {
 899                         char *nl;
 900
 901                         if (( nl = strchr( line, '\r' )) || ( nl = strchr( line, '\n' )))
 902                                 *nl = '\0';
 903                         if ( filters != NULL && line[0] == '+' ) {
 904                                 LDAPURLDesc     *lud;
 905
 906                                 if ( ldap_url_parse( &line[1], &lud ) != LDAP_URL_SUCCESS ) {
 907                                         entry = -1;
 908                                         break;
 909                                 }
 910
 911                                 if ( lud->lud_dn == NULL || lud->lud_dn[ 0 ] == '\0' ) {
 912                                         ldap_free_urldesc( lud );
 913                                         entry = -1;
 914                                         break;
 915                                 }
 916
 917                                 entries[entry] = ArgDup( lud->lud_dn );
 918
 919                                 if ( lud->lud_filter ) {
 920                                         filters[entry] = ArgDup( lud->lud_filter );
 921
 922                                 } else {
 923                                         filters[entry] = ArgDup( "(objectClass=*)" );
 924                                 }
 925                                 ldap_free_urldesc( lud );
 926
 927                         } else {
 928                                 entries[entry] = ArgDup( line );
 929                         }
 930
 931                         entry++;
 932
 933                 }
 934                 fclose( fp );
 935         }
 936
 937         return( entry );
 938 }
 939
 940 #ifndef HAVE_WINSOCK
 941 static void
 942 fork_child( char *prog, char **args )
 943 {
 944         /* note: obscures global pid var; intended */
 945         pid_t   pid;
 946
 947         wait4kids( maxkids );
 948
 949         switch ( pid = fork() ) {
 950         case 0:         /* child */
 951 #ifdef HAVE_EBCDIC
 952                 /* The __LIBASCII execvp only handles ASCII "prog",
 953                  * we still need to translate the arg vec ourselves.
 954                  */
 955                 { char *arg2[MAXREQS];
 956                 int i;
 957
 958                 for (i=0; args[i]; i++) {
 959                         arg2[i] = ArgDup(args[i]);
 960                         __atoe(arg2[i]);
 961                 }
 962                 arg2[i] = NULL;
 963                 args = arg2; }
 964 #endif
 965                 execvp( prog, args );
 966                 tester_perror( "execvp", NULL );
 967                 exit( EXIT_FAILURE );
 968                 break;
 969
 970         case -1:        /* trouble */
 971                 tester_perror( "fork", NULL );
 972                 break;
 973
 974         default:        /* parent */
 975                 nkids++;
 976                 break;
 977         }
 978 }
 979
 980 static void
 981 wait4kids( int nkidval )
 982 {
 983         int             status;
 984
 985         while ( nkids >= nkidval ) {
 986                 wait( &status );
 987
 988                 if ( WIFSTOPPED(status) ) {
 989                         fprintf( stderr,
 990                             "stopping: child stopped with signal %d\n",
 991                             (int) WSTOPSIG(status) );
 992
 993                 } else if ( WIFSIGNALED(status) ) {
 994                         fprintf( stderr,
 995                             "stopping: child terminated with signal %d%s\n",
 996                             (int) WTERMSIG(status),
 997 #ifdef WCOREDUMP
 998                                 WCOREDUMP(status) ? ", core dumped" : ""
 999 #else
1000                                 ""
1001 #endif
1002                                 );
1003                         exit( WEXITSTATUS(status)  );
1004
1005                 } else if ( WEXITSTATUS(status) != 0 ) {
1006                         fprintf( stderr,
1007                             "stopping: child exited with status %d\n",
1008                             (int) WEXITSTATUS(status) );
1009                         exit( WEXITSTATUS(status) );
1010
1011                 } else {
1012                         nkids--;
1013                 }
1014         }
1015 }
1016 #else
1017
1018 static void
1019 wait4kids( int nkidval )
1020 {
1021         int rc, i;
1022
1023         while ( nkids >= nkidval ) {
1024                 rc = WaitForMultipleObjects( nkids, children, FALSE, INFINITE );
1025                 for ( i=rc - WAIT_OBJECT_0; i<nkids-1; i++)
1026                         children[i] = children[i+1];
1027                 nkids--;
1028         }
1029 }
1030
1031 static void
1032 fork_child( char *prog, char **args )
1033 {
1034         int rc;
1035
1036         wait4kids( maxkids );
1037
1038         rc = _spawnvp( _P_NOWAIT, prog, args );
1039
1040         if ( rc == -1 ) {
1041                 tester_perror( "_spawnvp", NULL );
1042         } else {
1043                 children[nkids++] = (HANDLE)rc;
1044         }
1045 }
1046 #endif