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 /* static so that its address can be used in initializer below. */
141 static char sloops[] = "18446744073709551615UL";
144 char *rreqs[MAXREQS];
146 char *rargs[MAXARGS];
147 char *rflts[MAXREQS];
149 char rcmd[MAXPATHLEN];
150 static char rloops[] = "18446744073709551615UL";
152 char *afiles[MAXREQS];
154 char *aargs[MAXARGS];
156 char acmd[MAXPATHLEN];
157 static char aloops[] = "18446744073709551615UL";
160 char *nreqs[MAXREQS];
162 char *nargs[MAXARGS];
164 char ncmd[MAXPATHLEN];
165 static char nloops[] = "18446744073709551615UL";
168 char *mreqs[MAXREQS];
171 char *margs[MAXARGS];
173 char mcmd[MAXPATHLEN];
174 static char mloops[] = "18446744073709551615UL";
177 char *breqs[MAXREQS];
178 char *bcreds[MAXREQS];
179 char *battrs[MAXREQS];
181 char *bargs[MAXARGS];
183 char bcmd[MAXPATHLEN];
184 static char bloops[] = "18446744073709551615UL";
185 char **bargs_extra = NULL;
187 char *friendlyOpt = NULL;
189 char *pw_file = NULL;
191 /* extra action to do after bind... */
192 typedef struct extra_t {
194 struct extra_t *next;
197 extra_t *extra = NULL;
200 tester_init( "slapd-tester", TESTER_TESTER );
209 while ( ( i = getopt( argc, argv, "AB:CD:d:FH:h:Ii:j:L:l:NP:p:r:t:Ww:y:" ) ) != EOF )
218 **b = ldap_str2charray( optarg, "," );
221 for ( epp = &extra; *epp; epp = &(*epp)->next )
224 for ( p = b; p[0]; p++ ) {
225 *epp = calloc( 1, sizeof( extra_t ) );
226 (*epp)->action = p[0];
238 case 'D': /* slapd manager */
239 manager = ArgDup( optarg );
242 case 'd': /* data directory */
243 dirname = strdup( optarg );
250 case 'H': /* slapd uri */
251 uri = strdup( optarg );
254 case 'h': /* slapd host */
255 host = strdup( optarg );
266 case 'j': /* the number of parallel clients */
267 if ( lutil_atoi( &maxkids, optarg ) != 0 ) {
268 usage( argv[0], 'j' );
272 case 'l': /* the number of loops per client */
273 if ( !isdigit( (unsigned char) optarg[0] ) ) {
275 **l = ldap_str2charray( optarg, "," );
277 for ( p = l; p[0]; p++) {
282 { BER_BVC( "add=" ), aloops },
283 { BER_BVC( "bind=" ), bloops },
284 { BER_BVC( "modify=" ), mloops },
285 { BER_BVC( "modrdn=" ), nloops },
286 { BER_BVC( "read=" ), rloops },
287 { BER_BVC( "search=" ), sloops },
292 for ( c = 0; types[c].type.bv_val; c++ ) {
293 if ( strncasecmp( p[0], types[c].type.bv_val, types[c].type.bv_len ) == 0 ) {
298 if ( types[c].type.bv_val == NULL ) {
299 usage( argv[0], 'l' );
302 if ( lutil_atoi( &n, &p[0][types[c].type.bv_len] ) != 0 ) {
303 usage( argv[0], 'l' );
306 snprintf( types[c].buf, sizeof( aloops ), "%d", n );
309 ldap_charray_free( l );
311 } else if ( lutil_atoi( &loops, optarg ) != 0 ) {
312 usage( argv[0], 'l' );
316 case 'L': /* the number of outerloops per client */
317 outerloops = strdup( optarg );
324 case 'P': /* prog directory */
325 progdir = strdup( optarg );
328 case 'p': /* the servers port number */
329 port = strdup( optarg );
332 case 'r': /* the number of retries in case of error */
333 retries = strdup( optarg );
336 case 't': /* the delay in seconds between each retry */
337 delay = strdup( optarg );
340 case 'w': /* the managers passwd */
341 passwd = ArgDup( optarg );
342 memset( optarg, '*', strlen( optarg ) );
354 usage( argv[0], '\0' );
359 if (( dirname == NULL ) || ( port == NULL && uri == NULL ) ||
360 ( manager == NULL ) || ( passwd == NULL ) || ( progdir == NULL ))
362 usage( argv[0], '\0' );
366 children = malloc( maxkids * sizeof(HANDLE) );
368 /* get the file list */
369 if ( ( datadir = opendir( dirname )) == NULL ) {
370 fprintf( stderr, "%s: couldn't open data directory \"%s\".\n",
372 exit( EXIT_FAILURE );
375 /* look for search, read, modrdn, and add/delete files */
376 for ( file = readdir( datadir ); file; file = readdir( datadir )) {
378 if ( !strcasecmp( file->d_name, TSEARCHFILE )) {
379 sfile = get_file_name( dirname, file->d_name );
381 } else if ( !strcasecmp( file->d_name, TREADFILE )) {
382 rfile = get_file_name( dirname, file->d_name );
384 } else if ( !strcasecmp( file->d_name, TMODRDNFILE )) {
385 nfile = get_file_name( dirname, file->d_name );
387 } else if ( !strcasecmp( file->d_name, TMODIFYFILE )) {
388 mfile = get_file_name( dirname, file->d_name );
390 } else if ( !strncasecmp( file->d_name, TADDFILE, strlen( TADDFILE ))
391 && ( anum < MAXREQS )) {
392 afiles[anum++] = get_file_name( dirname, file->d_name );
394 } else if ( !strcasecmp( file->d_name, TBINDFILE )) {
395 bfile = get_file_name( dirname, file->d_name );
403 passwd = getpassphrase( _("Enter LDAP Password: ") );
405 } else if ( pw_file ) {
408 if ( lutil_get_filed_password( pw_file, &pw ) ) {
409 exit( EXIT_FAILURE );
415 /* look for search requests */
417 snum = get_search_filters( sfile, sreqs, sattrs, sbase );
420 /* look for read requests */
422 rnum = get_read_entries( rfile, rreqs, rflts );
425 /* look for modrdn requests */
427 nnum = get_read_entries( nfile, nreqs, NULL );
430 /* look for modify requests */
432 mnum = get_search_filters( mfile, mreqs, NULL, mdn );
435 /* look for bind requests */
437 bnum = get_search_filters( bfile, bcreds, battrs, breqs );
440 /* setup friendly option */
442 switch ( friendly ) {
451 /* NOTE: right now we don't need it more than twice */
457 if ( sloops[0] == '\0' ) snprintf( sloops, sizeof( sloops ), "%d", 10 * loops );
458 if ( rloops[0] == '\0' ) snprintf( rloops, sizeof( rloops ), "%d", 20 * loops );
459 if ( aloops[0] == '\0' ) snprintf( aloops, sizeof( aloops ), "%d", loops );
460 if ( nloops[0] == '\0' ) snprintf( nloops, sizeof( nloops ), "%d", loops );
461 if ( mloops[0] == '\0' ) snprintf( mloops, sizeof( mloops ), "%d", loops );
462 if ( bloops[0] == '\0' ) snprintf( bloops, sizeof( bloops ), "%d", 20 * loops );
465 * generate the search clients
469 snprintf( scmd, sizeof scmd, "%s" LDAP_DIRSEP SEARCHCMD,
471 sargs[sanum++] = scmd;
473 sargs[sanum++] = "-H";
474 sargs[sanum++] = uri;
476 sargs[sanum++] = "-h";
477 sargs[sanum++] = host;
478 sargs[sanum++] = "-p";
479 sargs[sanum++] = port;
481 sargs[sanum++] = "-D";
482 sargs[sanum++] = manager;
483 sargs[sanum++] = "-w";
484 sargs[sanum++] = passwd;
485 sargs[sanum++] = "-l";
486 sargs[sanum++] = sloops;
487 sargs[sanum++] = "-L";
488 sargs[sanum++] = outerloops;
489 sargs[sanum++] = "-r";
490 sargs[sanum++] = retries;
491 sargs[sanum++] = "-t";
492 sargs[sanum++] = delay;
494 sargs[sanum++] = friendlyOpt;
497 sargs[sanum++] = "-C";
500 sargs[sanum++] = "-A";
503 sargs[sanum++] = "-N";
506 sargs[sanum++] = "-i";
507 sargs[sanum++] = ignore;
509 sargs[sanum++] = "-b";
510 sargs[sanum++] = NULL; /* will hold the search base */
511 sargs[sanum++] = "-f";
512 sargs[sanum++] = NULL; /* will hold the search request */
514 sargs[sanum++] = NULL;
515 sargs[sanum] = NULL; /* might hold the "attr" request */
517 sargs[sanum + 1] = NULL;
520 * generate the read clients
524 snprintf( rcmd, sizeof rcmd, "%s" LDAP_DIRSEP READCMD,
526 rargs[ranum++] = rcmd;
528 rargs[ranum++] = "-H";
529 rargs[ranum++] = uri;
531 rargs[ranum++] = "-h";
532 rargs[ranum++] = host;
533 rargs[ranum++] = "-p";
534 rargs[ranum++] = port;
536 rargs[ranum++] = "-D";
537 rargs[ranum++] = manager;
538 rargs[ranum++] = "-w";
539 rargs[ranum++] = passwd;
540 rargs[ranum++] = "-l";
541 rargs[ranum++] = rloops;
542 rargs[ranum++] = "-L";
543 rargs[ranum++] = outerloops;
544 rargs[ranum++] = "-r";
545 rargs[ranum++] = retries;
546 rargs[ranum++] = "-t";
547 rargs[ranum++] = delay;
549 rargs[ranum++] = friendlyOpt;
552 rargs[ranum++] = "-C";
555 rargs[ranum++] = "-A";
558 rargs[ranum++] = "-i";
559 rargs[ranum++] = ignore;
561 rargs[ranum++] = "-e";
562 rargs[ranum++] = NULL; /* will hold the read entry */
564 rargs[ranum++] = NULL;
565 rargs[ranum] = NULL; /* might hold the filter arg */
567 rargs[ranum + 1] = NULL;
570 * generate the modrdn clients
574 snprintf( ncmd, sizeof ncmd, "%s" LDAP_DIRSEP MODRDNCMD,
576 nargs[nanum++] = ncmd;
578 nargs[nanum++] = "-H";
579 nargs[nanum++] = uri;
581 nargs[nanum++] = "-h";
582 nargs[nanum++] = host;
583 nargs[nanum++] = "-p";
584 nargs[nanum++] = port;
586 nargs[nanum++] = "-D";
587 nargs[nanum++] = manager;
588 nargs[nanum++] = "-w";
589 nargs[nanum++] = passwd;
590 nargs[nanum++] = "-l";
591 nargs[nanum++] = nloops;
592 nargs[nanum++] = "-L";
593 nargs[nanum++] = outerloops;
594 nargs[nanum++] = "-r";
595 nargs[nanum++] = retries;
596 nargs[nanum++] = "-t";
597 nargs[nanum++] = delay;
599 nargs[nanum++] = friendlyOpt;
602 nargs[nanum++] = "-C";
605 nargs[nanum++] = "-i";
606 nargs[nanum++] = ignore;
608 nargs[nanum++] = "-e";
609 nargs[nanum++] = NULL; /* will hold the modrdn entry */
610 nargs[nanum++] = NULL;
613 * generate the modify clients
617 snprintf( mcmd, sizeof mcmd, "%s" LDAP_DIRSEP MODIFYCMD,
619 margs[manum++] = mcmd;
621 margs[manum++] = "-H";
622 margs[manum++] = uri;
624 margs[manum++] = "-h";
625 margs[manum++] = host;
626 margs[manum++] = "-p";
627 margs[manum++] = port;
629 margs[manum++] = "-D";
630 margs[manum++] = manager;
631 margs[manum++] = "-w";
632 margs[manum++] = passwd;
633 margs[manum++] = "-l";
634 margs[manum++] = mloops;
635 margs[manum++] = "-L";
636 margs[manum++] = outerloops;
637 margs[manum++] = "-r";
638 margs[manum++] = retries;
639 margs[manum++] = "-t";
640 margs[manum++] = delay;
642 margs[manum++] = friendlyOpt;
645 margs[manum++] = "-C";
648 margs[manum++] = "-i";
649 margs[manum++] = ignore;
651 margs[manum++] = "-e";
652 margs[manum++] = NULL; /* will hold the modify entry */
653 margs[manum++] = "-a";;
654 margs[manum++] = NULL; /* will hold the ava */
655 margs[manum++] = NULL;
658 * generate the add/delete clients
662 snprintf( acmd, sizeof acmd, "%s" LDAP_DIRSEP ADDCMD,
664 aargs[aanum++] = acmd;
666 aargs[aanum++] = "-H";
667 aargs[aanum++] = uri;
669 aargs[aanum++] = "-h";
670 aargs[aanum++] = host;
671 aargs[aanum++] = "-p";
672 aargs[aanum++] = port;
674 aargs[aanum++] = "-D";
675 aargs[aanum++] = manager;
676 aargs[aanum++] = "-w";
677 aargs[aanum++] = passwd;
678 aargs[aanum++] = "-l";
679 aargs[aanum++] = aloops;
680 aargs[aanum++] = "-L";
681 aargs[aanum++] = outerloops;
682 aargs[aanum++] = "-r";
683 aargs[aanum++] = retries;
684 aargs[aanum++] = "-t";
685 aargs[aanum++] = delay;
687 aargs[aanum++] = friendlyOpt;
690 aargs[aanum++] = "-C";
693 aargs[aanum++] = "-i";
694 aargs[aanum++] = ignore;
696 aargs[aanum++] = "-f";
697 aargs[aanum++] = NULL; /* will hold the add data file */
698 aargs[aanum++] = NULL;
701 * generate the bind clients
705 snprintf( bcmd, sizeof bcmd, "%s" LDAP_DIRSEP BINDCMD,
707 bargs[banum++] = bcmd;
709 bargs[banum++] = "-I"; /* init on each bind */
712 bargs[banum++] = "-H";
713 bargs[banum++] = uri;
715 bargs[banum++] = "-h";
716 bargs[banum++] = host;
717 bargs[banum++] = "-p";
718 bargs[banum++] = port;
720 bargs[banum++] = "-l";
721 bargs[banum++] = bloops;
722 bargs[banum++] = "-L";
723 bargs[banum++] = outerloops;
725 bargs[banum++] = "-r";
726 bargs[banum++] = retries;
727 bargs[banum++] = "-t";
728 bargs[banum++] = delay;
731 bargs[banum++] = friendlyOpt;
734 bargs[banum++] = "-C";
737 bargs[banum++] = "-i";
738 bargs[banum++] = ignore;
741 bargs[banum++] = "-B";
742 bargs_extra = &bargs[banum++];
744 bargs[banum++] = "-D";
745 bargs[banum++] = NULL;
746 bargs[banum++] = "-w";
747 bargs[banum++] = NULL;
748 bargs[banum++] = NULL;
750 #define DOREQ(n,j) ((n) && ((maxkids > (n)) ? ((j) < maxkids ) : ((j) < (n))))
752 for ( j = 0; j < MAXREQS; j++ ) {
753 if ( DOREQ( snum, j ) ) {
756 sargs[sanum - 2] = sreqs[jj];
757 sargs[sanum - 4] = sbase[jj];
758 if ( sattrs[jj] != NULL ) {
759 sargs[sanum - 1] = "-a";
760 sargs[sanum] = sattrs[jj];
763 sargs[sanum - 1] = NULL;
765 fork_child( scmd, sargs );
768 if ( DOREQ( rnum, j ) ) {
771 rargs[ranum - 2] = rreqs[jj];
772 if ( rflts[jj] != NULL ) {
773 rargs[ranum - 1] = "-f";
774 rargs[ranum] = rflts[jj];
777 rargs[ranum - 1] = NULL;
779 fork_child( rcmd, rargs );
783 nargs[nanum - 2] = nreqs[j];
784 fork_child( ncmd, nargs );
788 margs[manum - 4] = mdn[j];
789 margs[manum - 2] = mreqs[j];
790 fork_child( mcmd, margs );
794 aargs[aanum - 2] = afiles[j];
795 fork_child( acmd, aargs );
798 if ( DOREQ( bnum, j ) ) {
802 int n = ((double)nextra)*rand()/(RAND_MAX + 1.0);
805 for ( e = extra; n-- > 0; e = e->next )
807 *bargs_extra = e->action;
810 if ( battrs[jj] != NULL ) {
811 bargs[banum - 4] = manager ? manager : "";
812 bargs[banum - 2] = passwd ? passwd : "";
814 bargs[banum - 1] = "-b";
815 bargs[banum] = breqs[jj];
816 bargs[banum + 1] = "-f";
817 bargs[banum + 2] = bcreds[jj];
818 bargs[banum + 3] = "-a";
819 bargs[banum + 4] = battrs[jj];
821 bargs[banum - 4] = breqs[jj];
822 bargs[banum - 2] = bcreds[jj];
823 bargs[banum - 1] = NULL;
826 fork_child( bcmd, bargs );
827 bargs[banum - 1] = NULL;
833 exit( EXIT_SUCCESS );
837 get_file_name( char *dirname, char *filename )
839 char buf[MAXPATHLEN];
841 snprintf( buf, sizeof buf, "%s" LDAP_DIRSEP "%s",
843 return( strdup( buf ));
848 get_search_filters( char *filename, char *filters[], char *attrs[], char *bases[] )
853 if ( (fp = fopen( filename, "r" )) != NULL ) {
856 while (( filter < MAXREQS ) && ( fgets( line, BUFSIZ, fp ))) {
859 if (( nl = strchr( line, '\r' )) || ( nl = strchr( line, '\n' )))
861 bases[filter] = ArgDup( line );
862 fgets( line, BUFSIZ, fp );
863 if (( nl = strchr( line, '\r' )) || ( nl = strchr( line, '\n' )))
866 filters[filter] = ArgDup( line );
868 if ( filters[filter][0] == '+') {
869 char *sep = strchr( filters[filter], ':' );
872 attrs[ filter ] = &filters[ filter ][ 1 ];
874 /* NOTE: don't free this! */
875 filters[ filter ] = &sep[ 1 ];
879 attrs[ filter] = NULL;
893 get_read_entries( char *filename, char *entries[], char *filters[] )
898 if ( (fp = fopen( filename, "r" )) != NULL ) {
901 while (( entry < MAXREQS ) && ( fgets( line, BUFSIZ, fp ))) {
904 if (( nl = strchr( line, '\r' )) || ( nl = strchr( line, '\n' )))
906 if ( filters != NULL && line[0] == '+' ) {
909 if ( ldap_url_parse( &line[1], &lud ) != LDAP_URL_SUCCESS ) {
914 if ( lud->lud_dn == NULL || lud->lud_dn[ 0 ] == '\0' ) {
915 ldap_free_urldesc( lud );
920 entries[entry] = ArgDup( lud->lud_dn );
922 if ( lud->lud_filter ) {
923 filters[entry] = ArgDup( lud->lud_filter );
926 filters[entry] = ArgDup( "(objectClass=*)" );
928 ldap_free_urldesc( lud );
931 entries[entry] = ArgDup( line );
945 fork_child( char *prog, char **args )
947 /* note: obscures global pid var; intended */
950 wait4kids( maxkids );
952 switch ( pid = fork() ) {
955 /* The __LIBASCII execvp only handles ASCII "prog",
956 * we still need to translate the arg vec ourselves.
958 { char *arg2[MAXREQS];
961 for (i=0; args[i]; i++) {
962 arg2[i] = ArgDup(args[i]);
968 execvp( prog, args );
969 tester_perror( "execvp", NULL );
970 exit( EXIT_FAILURE );
973 case -1: /* trouble */
974 tester_perror( "fork", NULL );
977 default: /* parent */
984 wait4kids( int nkidval )
988 while ( nkids >= nkidval ) {
991 if ( WIFSTOPPED(status) ) {
993 "stopping: child stopped with signal %d\n",
994 (int) WSTOPSIG(status) );
996 } else if ( WIFSIGNALED(status) ) {
998 "stopping: child terminated with signal %d%s\n",
999 (int) WTERMSIG(status),
1001 WCOREDUMP(status) ? ", core dumped" : ""
1006 exit( WEXITSTATUS(status) );
1008 } else if ( WEXITSTATUS(status) != 0 ) {
1010 "stopping: child exited with status %d\n",
1011 (int) WEXITSTATUS(status) );
1012 exit( WEXITSTATUS(status) );
1022 wait4kids( int nkidval )
1026 while ( nkids >= nkidval ) {
1027 rc = WaitForMultipleObjects( nkids, children, FALSE, INFINITE );
1028 for ( i=rc - WAIT_OBJECT_0; i<nkids-1; i++)
1029 children[i] = children[i+1];
1035 fork_child( char *prog, char **args )
1039 wait4kids( maxkids );
1041 rc = _spawnvp( _P_NOWAIT, prog, args );
1044 tester_perror( "_spawnvp", NULL );
1046 children[nkids++] = (HANDLE)rc;