2 /* This work is part of OpenLDAP Software <http://www.openldap.org/>.
4 * Copyright 1999-2007 The OpenLDAP Foundation.
7 * Redistribution and use in source and binary forms, with or without
8 * modification, are permitted only as authorized by the OpenLDAP
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>.
16 * This work was initially developed by Kurt Spanier for inclusion
17 * in OpenLDAP Software.
24 #include <ac/stdlib.h>
27 #include <ac/dirent.h>
29 #include <ac/socket.h>
30 #include <ac/string.h>
31 #include <ac/unistd.h>
35 #include "ldap_defaults.h"
41 #include "slapd-common.h"
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"
52 #define OUTERLOOPS "1"
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"
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 );
68 static int maxkids = 20;
72 static HANDLE *children;
73 static char argbuf[BUFSIZ];
74 #define ArgDup(x) strdup(strcat(strcat(strcpy(argbuf,"\""),x),"\""))
76 #define ArgDup(x) strdup(x)
80 usage( char *name, char opt )
83 fprintf( stderr, "%s: unable to handle option \'%c\'\n\n",
89 "-H <uri> | ([-h <host>] -p <port>) "
95 "[-l {<loops>|<type>=<loops>[,...]}] "
105 exit( EXIT_FAILURE );
109 main( int argc, char **argv )
113 char *host = "localhost";
115 char *manager = NULL;
117 char *dirname = NULL;
118 char *progdir = NULL;
120 char *outerloops = OUTERLOOPS;
121 char *retries = RETRIES;
133 char *sreqs[MAXREQS];
134 char *sattrs[MAXREQS];
135 char *sbase[MAXREQS];
137 char *sargs[MAXARGS];
139 char scmd[MAXPATHLEN];
140 char sloops[] = "18446744073709551615UL";
143 char *rreqs[MAXREQS];
145 char *rargs[MAXARGS];
146 char *rflts[MAXREQS];
148 char rcmd[MAXPATHLEN];
149 char rloops[] = "18446744073709551615UL";
151 char *afiles[MAXREQS];
153 char *aargs[MAXARGS];
155 char acmd[MAXPATHLEN];
156 char aloops[] = "18446744073709551615UL";
159 char *nreqs[MAXREQS];
161 char *nargs[MAXARGS];
163 char ncmd[MAXPATHLEN];
164 char nloops[] = "18446744073709551615UL";
167 char *mreqs[MAXREQS];
170 char *margs[MAXARGS];
172 char mcmd[MAXPATHLEN];
173 char mloops[] = "18446744073709551615UL";
176 char *breqs[MAXREQS];
177 char *bcreds[MAXREQS];
178 char *battrs[MAXREQS];
180 char *bargs[MAXARGS];
182 char bcmd[MAXPATHLEN];
183 char bloops[] = "18446744073709551615UL";
184 char **bargs_extra = NULL;
186 char *friendlyOpt = NULL;
188 char *pw_file = NULL;
190 /* extra action to do after bind... */
191 typedef struct extra_t {
193 struct extra_t *next;
196 extra_t *extra = NULL;
199 tester_init( "slapd-tester", TESTER_TESTER );
208 while ( ( i = getopt( argc, argv, "AB:CD:d:FH:h:Ii:j:L:l:NP:p:r:t:Ww:y:" ) ) != EOF )
217 **b = ldap_str2charray( optarg, "," );
220 for ( epp = &extra; *epp; epp = &(*epp)->next )
223 for ( p = b; p[0]; p++ ) {
224 *epp = calloc( 1, sizeof( extra_t ) );
225 (*epp)->action = p[0];
237 case 'D': /* slapd manager */
238 manager = ArgDup( optarg );
241 case 'd': /* data directory */
242 dirname = strdup( optarg );
249 case 'H': /* slapd uri */
250 uri = strdup( optarg );
253 case 'h': /* slapd host */
254 host = strdup( optarg );
265 case 'j': /* the number of parallel clients */
266 if ( lutil_atoi( &maxkids, optarg ) != 0 ) {
267 usage( argv[0], 'j' );
271 case 'l': /* the number of loops per client */
272 if ( !isdigit( optarg[0] ) ) {
274 **l = ldap_str2charray( optarg, "," );
276 for ( p = l; p[0]; p++) {
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 },
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 ) {
297 if ( types[c].type.bv_val == NULL ) {
298 usage( argv[0], 'l' );
301 if ( lutil_atoi( &n, &p[0][types[c].type.bv_len] ) != 0 ) {
302 usage( argv[0], 'l' );
305 snprintf( types[c].buf, sizeof( aloops ), "%d", n );
308 ldap_charray_free( l );
310 } else if ( lutil_atoi( &loops, optarg ) != 0 ) {
311 usage( argv[0], 'l' );
315 case 'L': /* the number of outerloops per client */
316 outerloops = strdup( optarg );
323 case 'P': /* prog directory */
324 progdir = strdup( optarg );
327 case 'p': /* the servers port number */
328 port = strdup( optarg );
331 case 'r': /* the number of retries in case of error */
332 retries = strdup( optarg );
335 case 't': /* the delay in seconds between each retry */
336 delay = strdup( optarg );
339 case 'w': /* the managers passwd */
340 passwd = ArgDup( optarg );
341 memset( optarg, '*', strlen( optarg ) );
353 usage( argv[0], '\0' );
358 if (( dirname == NULL ) || ( port == NULL && uri == NULL ) ||
359 ( manager == NULL ) || ( passwd == NULL ) || ( progdir == NULL ))
361 usage( argv[0], '\0' );
365 children = malloc( maxkids * sizeof(HANDLE) );
367 /* get the file list */
368 if ( ( datadir = opendir( dirname )) == NULL ) {
369 fprintf( stderr, "%s: couldn't open data directory \"%s\".\n",
371 exit( EXIT_FAILURE );
374 /* look for search, read, modrdn, and add/delete files */
375 for ( file = readdir( datadir ); file; file = readdir( datadir )) {
377 if ( !strcasecmp( file->d_name, TSEARCHFILE )) {
378 sfile = get_file_name( dirname, file->d_name );
380 } else if ( !strcasecmp( file->d_name, TREADFILE )) {
381 rfile = get_file_name( dirname, file->d_name );
383 } else if ( !strcasecmp( file->d_name, TMODRDNFILE )) {
384 nfile = get_file_name( dirname, file->d_name );
386 } else if ( !strcasecmp( file->d_name, TMODIFYFILE )) {
387 mfile = get_file_name( dirname, file->d_name );
389 } else if ( !strncasecmp( file->d_name, TADDFILE, strlen( TADDFILE ))
390 && ( anum < MAXREQS )) {
391 afiles[anum++] = get_file_name( dirname, file->d_name );
393 } else if ( !strcasecmp( file->d_name, TBINDFILE )) {
394 bfile = get_file_name( dirname, file->d_name );
402 passwd = getpassphrase( _("Enter LDAP Password: ") );
404 } else if ( pw_file ) {
407 if ( lutil_get_filed_password( pw_file, &pw ) ) {
408 exit( EXIT_FAILURE );
414 /* look for search requests */
416 snum = get_search_filters( sfile, sreqs, sattrs, sbase );
419 /* look for read requests */
421 rnum = get_read_entries( rfile, rreqs, rflts );
424 /* look for modrdn requests */
426 nnum = get_read_entries( nfile, nreqs, NULL );
429 /* look for modify requests */
431 mnum = get_search_filters( mfile, mreqs, NULL, mdn );
434 /* look for bind requests */
436 bnum = get_search_filters( bfile, bcreds, battrs, breqs );
439 /* setup friendly option */
441 switch ( friendly ) {
450 /* NOTE: right now we don't need it more than twice */
456 if ( sloops[0] == '\0' ) snprintf( sloops, sizeof( sloops ), "%d", 10 * loops );
457 if ( rloops[0] == '\0' ) snprintf( rloops, sizeof( rloops ), "%d", 20 * loops );
458 if ( aloops[0] == '\0' ) snprintf( aloops, sizeof( aloops ), "%d", loops );
459 if ( nloops[0] == '\0' ) snprintf( nloops, sizeof( nloops ), "%d", loops );
460 if ( mloops[0] == '\0' ) snprintf( mloops, sizeof( mloops ), "%d", loops );
461 if ( bloops[0] == '\0' ) snprintf( bloops, sizeof( bloops ), "%d", 20 * loops );
464 * generate the search clients
468 snprintf( scmd, sizeof scmd, "%s" LDAP_DIRSEP SEARCHCMD,
470 sargs[sanum++] = scmd;
472 sargs[sanum++] = "-H";
473 sargs[sanum++] = uri;
475 sargs[sanum++] = "-h";
476 sargs[sanum++] = host;
477 sargs[sanum++] = "-p";
478 sargs[sanum++] = port;
480 sargs[sanum++] = "-D";
481 sargs[sanum++] = manager;
482 sargs[sanum++] = "-w";
483 sargs[sanum++] = passwd;
484 sargs[sanum++] = "-l";
485 sargs[sanum++] = sloops;
486 sargs[sanum++] = "-L";
487 sargs[sanum++] = outerloops;
488 sargs[sanum++] = "-r";
489 sargs[sanum++] = retries;
490 sargs[sanum++] = "-t";
491 sargs[sanum++] = delay;
493 sargs[sanum++] = friendlyOpt;
496 sargs[sanum++] = "-C";
499 sargs[sanum++] = "-A";
502 sargs[sanum++] = "-N";
505 sargs[sanum++] = "-i";
506 sargs[sanum++] = ignore;
508 sargs[sanum++] = "-b";
509 sargs[sanum++] = NULL; /* will hold the search base */
510 sargs[sanum++] = "-f";
511 sargs[sanum++] = NULL; /* will hold the search request */
513 sargs[sanum++] = NULL;
514 sargs[sanum] = NULL; /* might hold the "attr" request */
516 sargs[sanum + 1] = NULL;
519 * generate the read clients
523 snprintf( rcmd, sizeof rcmd, "%s" LDAP_DIRSEP READCMD,
525 rargs[ranum++] = rcmd;
527 rargs[ranum++] = "-H";
528 rargs[ranum++] = uri;
530 rargs[ranum++] = "-h";
531 rargs[ranum++] = host;
532 rargs[ranum++] = "-p";
533 rargs[ranum++] = port;
535 rargs[ranum++] = "-D";
536 rargs[ranum++] = manager;
537 rargs[ranum++] = "-w";
538 rargs[ranum++] = passwd;
539 rargs[ranum++] = "-l";
540 rargs[ranum++] = rloops;
541 rargs[ranum++] = "-L";
542 rargs[ranum++] = outerloops;
543 rargs[ranum++] = "-r";
544 rargs[ranum++] = retries;
545 rargs[ranum++] = "-t";
546 rargs[ranum++] = delay;
548 rargs[ranum++] = friendlyOpt;
551 rargs[ranum++] = "-C";
554 rargs[ranum++] = "-A";
557 rargs[ranum++] = "-i";
558 rargs[ranum++] = ignore;
560 rargs[ranum++] = "-e";
561 rargs[ranum++] = NULL; /* will hold the read entry */
563 rargs[ranum++] = NULL;
564 rargs[ranum] = NULL; /* might hold the filter arg */
566 rargs[ranum + 1] = NULL;
569 * generate the modrdn clients
573 snprintf( ncmd, sizeof ncmd, "%s" LDAP_DIRSEP MODRDNCMD,
575 nargs[nanum++] = ncmd;
577 nargs[nanum++] = "-H";
578 nargs[nanum++] = uri;
580 nargs[nanum++] = "-h";
581 nargs[nanum++] = host;
582 nargs[nanum++] = "-p";
583 nargs[nanum++] = port;
585 nargs[nanum++] = "-D";
586 nargs[nanum++] = manager;
587 nargs[nanum++] = "-w";
588 nargs[nanum++] = passwd;
589 nargs[nanum++] = "-l";
590 nargs[nanum++] = nloops;
591 nargs[nanum++] = "-L";
592 nargs[nanum++] = outerloops;
593 nargs[nanum++] = "-r";
594 nargs[nanum++] = retries;
595 nargs[nanum++] = "-t";
596 nargs[nanum++] = delay;
598 nargs[nanum++] = friendlyOpt;
601 nargs[nanum++] = "-C";
604 nargs[nanum++] = "-i";
605 nargs[nanum++] = ignore;
607 nargs[nanum++] = "-e";
608 nargs[nanum++] = NULL; /* will hold the modrdn entry */
609 nargs[nanum++] = NULL;
612 * generate the modify clients
616 snprintf( mcmd, sizeof mcmd, "%s" LDAP_DIRSEP MODIFYCMD,
618 margs[manum++] = mcmd;
620 margs[manum++] = "-H";
621 margs[manum++] = uri;
623 margs[manum++] = "-h";
624 margs[manum++] = host;
625 margs[manum++] = "-p";
626 margs[manum++] = port;
628 margs[manum++] = "-D";
629 margs[manum++] = manager;
630 margs[manum++] = "-w";
631 margs[manum++] = passwd;
632 margs[manum++] = "-l";
633 margs[manum++] = mloops;
634 margs[manum++] = "-L";
635 margs[manum++] = outerloops;
636 margs[manum++] = "-r";
637 margs[manum++] = retries;
638 margs[manum++] = "-t";
639 margs[manum++] = delay;
641 margs[manum++] = friendlyOpt;
644 margs[manum++] = "-C";
647 margs[manum++] = "-i";
648 margs[manum++] = ignore;
650 margs[manum++] = "-e";
651 margs[manum++] = NULL; /* will hold the modify entry */
652 margs[manum++] = "-a";;
653 margs[manum++] = NULL; /* will hold the ava */
654 margs[manum++] = NULL;
657 * generate the add/delete clients
661 snprintf( acmd, sizeof acmd, "%s" LDAP_DIRSEP ADDCMD,
663 aargs[aanum++] = acmd;
665 aargs[aanum++] = "-H";
666 aargs[aanum++] = uri;
668 aargs[aanum++] = "-h";
669 aargs[aanum++] = host;
670 aargs[aanum++] = "-p";
671 aargs[aanum++] = port;
673 aargs[aanum++] = "-D";
674 aargs[aanum++] = manager;
675 aargs[aanum++] = "-w";
676 aargs[aanum++] = passwd;
677 aargs[aanum++] = "-l";
678 aargs[aanum++] = aloops;
679 aargs[aanum++] = "-L";
680 aargs[aanum++] = outerloops;
681 aargs[aanum++] = "-r";
682 aargs[aanum++] = retries;
683 aargs[aanum++] = "-t";
684 aargs[aanum++] = delay;
686 aargs[aanum++] = friendlyOpt;
689 aargs[aanum++] = "-C";
692 aargs[aanum++] = "-i";
693 aargs[aanum++] = ignore;
695 aargs[aanum++] = "-f";
696 aargs[aanum++] = NULL; /* will hold the add data file */
697 aargs[aanum++] = NULL;
700 * generate the bind clients
704 snprintf( bcmd, sizeof bcmd, "%s" LDAP_DIRSEP BINDCMD,
706 bargs[banum++] = bcmd;
708 bargs[banum++] = "-I"; /* init on each bind */
711 bargs[banum++] = "-H";
712 bargs[banum++] = uri;
714 bargs[banum++] = "-h";
715 bargs[banum++] = host;
716 bargs[banum++] = "-p";
717 bargs[banum++] = port;
719 bargs[banum++] = "-l";
720 bargs[banum++] = bloops;
721 bargs[banum++] = "-L";
722 bargs[banum++] = outerloops;
724 bargs[banum++] = "-r";
725 bargs[banum++] = retries;
726 bargs[banum++] = "-t";
727 bargs[banum++] = delay;
730 bargs[banum++] = friendlyOpt;
733 bargs[banum++] = "-C";
736 bargs[banum++] = "-i";
737 bargs[banum++] = ignore;
740 bargs[banum++] = "-B";
741 bargs_extra = &bargs[banum++];
743 bargs[banum++] = "-D";
744 bargs[banum++] = NULL;
745 bargs[banum++] = "-w";
746 bargs[banum++] = NULL;
747 bargs[banum++] = NULL;
749 #define DOREQ(n,j) ((n) && ((maxkids > (n)) ? ((j) < maxkids ) : ((j) < (n))))
751 for ( j = 0; j < MAXREQS; j++ ) {
752 if ( DOREQ( snum, j ) ) {
755 sargs[sanum - 2] = sreqs[jj];
756 sargs[sanum - 4] = sbase[jj];
757 if ( sattrs[jj] != NULL ) {
758 sargs[sanum - 1] = "-a";
759 sargs[sanum] = sattrs[jj];
762 sargs[sanum - 1] = NULL;
764 fork_child( scmd, sargs );
767 if ( DOREQ( rnum, j ) ) {
770 rargs[ranum - 2] = rreqs[jj];
771 if ( rflts[jj] != NULL ) {
772 rargs[ranum - 1] = "-f";
773 rargs[ranum] = rflts[jj];
776 rargs[ranum - 1] = NULL;
778 fork_child( rcmd, rargs );
782 nargs[nanum - 2] = nreqs[j];
783 fork_child( ncmd, nargs );
787 margs[manum - 4] = mdn[j];
788 margs[manum - 2] = mreqs[j];
789 fork_child( mcmd, margs );
793 aargs[aanum - 2] = afiles[j];
794 fork_child( acmd, aargs );
797 if ( DOREQ( bnum, j ) ) {
801 int n = ((double)nextra)*rand()/(RAND_MAX + 1.0);
804 for ( e = extra; n-- > 0; e = e->next )
806 *bargs_extra = e->action;
809 if ( battrs[jj] != NULL ) {
810 bargs[banum - 4] = manager ? manager : "";
811 bargs[banum - 2] = passwd ? passwd : "";
813 bargs[banum - 1] = "-b";
814 bargs[banum] = breqs[jj];
815 bargs[banum + 1] = "-f";
816 bargs[banum + 2] = bcreds[jj];
817 bargs[banum + 3] = "-a";
818 bargs[banum + 4] = battrs[jj];
820 bargs[banum - 4] = breqs[jj];
821 bargs[banum - 2] = bcreds[jj];
822 bargs[banum - 1] = NULL;
825 fork_child( bcmd, bargs );
826 bargs[banum - 1] = NULL;
832 exit( EXIT_SUCCESS );
836 get_file_name( char *dirname, char *filename )
838 char buf[MAXPATHLEN];
840 snprintf( buf, sizeof buf, "%s" LDAP_DIRSEP "%s",
842 return( strdup( buf ));
847 get_search_filters( char *filename, char *filters[], char *attrs[], char *bases[] )
852 if ( (fp = fopen( filename, "r" )) != NULL ) {
855 while (( filter < MAXREQS ) && ( fgets( line, BUFSIZ, fp ))) {
858 if (( nl = strchr( line, '\r' )) || ( nl = strchr( line, '\n' )))
860 bases[filter] = ArgDup( line );
861 fgets( line, BUFSIZ, fp );
862 if (( nl = strchr( line, '\r' )) || ( nl = strchr( line, '\n' )))
865 filters[filter] = ArgDup( line );
867 if ( filters[filter][0] == '+') {
868 char *sep = strchr( filters[filter], ':' );
871 attrs[ filter ] = &filters[ filter ][ 1 ];
873 /* NOTE: don't free this! */
874 filters[ filter ] = &sep[ 1 ];
878 attrs[ filter] = NULL;
892 get_read_entries( char *filename, char *entries[], char *filters[] )
897 if ( (fp = fopen( filename, "r" )) != NULL ) {
900 while (( entry < MAXREQS ) && ( fgets( line, BUFSIZ, fp ))) {
903 if (( nl = strchr( line, '\r' )) || ( nl = strchr( line, '\n' )))
905 if ( filters != NULL && line[0] == '+' ) {
908 if ( ldap_url_parse( &line[1], &lud ) != LDAP_URL_SUCCESS ) {
913 if ( lud->lud_dn == NULL || lud->lud_dn[ 0 ] == '\0' ) {
914 ldap_free_urldesc( lud );
919 entries[entry] = ArgDup( lud->lud_dn );
921 if ( lud->lud_filter ) {
922 filters[entry] = ArgDup( lud->lud_filter );
925 filters[entry] = ArgDup( "(objectClass=*)" );
927 ldap_free_urldesc( lud );
930 entries[entry] = ArgDup( line );
944 fork_child( char *prog, char **args )
946 /* note: obscures global pid var; intended */
949 wait4kids( maxkids );
951 switch ( pid = fork() ) {
954 /* The __LIBASCII execvp only handles ASCII "prog",
955 * we still need to translate the arg vec ourselves.
957 { char *arg2[MAXREQS];
960 for (i=0; args[i]; i++) {
961 arg2[i] = ArgDup(args[i]);
967 execvp( prog, args );
968 tester_perror( "execvp", NULL );
969 exit( EXIT_FAILURE );
972 case -1: /* trouble */
973 tester_perror( "fork", NULL );
976 default: /* parent */
983 wait4kids( int nkidval )
987 while ( nkids >= nkidval ) {
990 if ( WIFSTOPPED(status) ) {
992 "stopping: child stopped with signal %d\n",
993 (int) WSTOPSIG(status) );
995 } else if ( WIFSIGNALED(status) ) {
997 "stopping: child terminated with signal %d%s\n",
998 (int) WTERMSIG(status),
1000 WCOREDUMP(status) ? ", core dumped" : ""
1005 exit( WEXITSTATUS(status) );
1007 } else if ( WEXITSTATUS(status) != 0 ) {
1009 "stopping: child exited with status %d\n",
1010 (int) WEXITSTATUS(status) );
1011 exit( WEXITSTATUS(status) );
1021 wait4kids( int nkidval )
1025 while ( nkids >= nkidval ) {
1026 rc = WaitForMultipleObjects( nkids, children, FALSE, INFINITE );
1027 for ( i=rc - WAIT_OBJECT_0; i<nkids-1; i++)
1028 children[i] = children[i+1];
1034 fork_child( char *prog, char **args )
1038 wait4kids( maxkids );
1040 rc = _spawnvp( _P_NOWAIT, prog, args );
1043 tester_perror( "_spawnvp", NULL );
1045 children[nkids++] = (HANDLE)rc;