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[], LDAPURLDesc *luds[] );
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];
136 LDAPURLDesc *slud[MAXREQS];
138 char *sargs[MAXARGS];
141 char scmd[MAXPATHLEN];
142 /* static so that its address can be used in initializer below. */
143 static char sloops[] = "18446744073709551615UL";
146 char *rreqs[MAXREQS];
148 char *rargs[MAXARGS];
149 char *rflts[MAXREQS];
152 char rcmd[MAXPATHLEN];
153 static char rloops[] = "18446744073709551615UL";
155 char *afiles[MAXREQS];
157 char *aargs[MAXARGS];
159 char acmd[MAXPATHLEN];
160 static char aloops[] = "18446744073709551615UL";
163 char *nreqs[MAXREQS];
165 char *nargs[MAXARGS];
167 char ncmd[MAXPATHLEN];
168 static char nloops[] = "18446744073709551615UL";
171 char *mreqs[MAXREQS];
174 char *margs[MAXARGS];
176 char mcmd[MAXPATHLEN];
177 static char mloops[] = "18446744073709551615UL";
180 char *breqs[MAXREQS];
181 char *bcreds[MAXREQS];
182 char *battrs[MAXREQS];
184 char *bargs[MAXARGS];
186 char bcmd[MAXPATHLEN];
187 static char bloops[] = "18446744073709551615UL";
188 char **bargs_extra = NULL;
190 char *friendlyOpt = NULL;
192 char *pw_file = NULL;
194 /* extra action to do after bind... */
195 typedef struct extra_t {
197 struct extra_t *next;
200 extra_t *extra = NULL;
203 tester_init( "slapd-tester", TESTER_TESTER );
212 while ( ( i = getopt( argc, argv, "AB:CD:d:FH:h:Ii:j:L:l:NP:p:r:t:Ww:y:" ) ) != EOF )
221 **b = ldap_str2charray( optarg, "," );
224 for ( epp = &extra; *epp; epp = &(*epp)->next )
227 for ( p = b; p[0]; p++ ) {
228 *epp = calloc( 1, sizeof( extra_t ) );
229 (*epp)->action = p[0];
241 case 'D': /* slapd manager */
242 manager = ArgDup( optarg );
245 case 'd': /* data directory */
246 dirname = strdup( optarg );
253 case 'H': /* slapd uri */
254 uri = strdup( optarg );
257 case 'h': /* slapd host */
258 host = strdup( optarg );
269 case 'j': /* the number of parallel clients */
270 if ( lutil_atoi( &maxkids, optarg ) != 0 ) {
271 usage( argv[0], 'j' );
275 case 'l': /* the number of loops per client */
276 if ( !isdigit( (unsigned char) optarg[0] ) ) {
278 **l = ldap_str2charray( optarg, "," );
280 for ( p = l; p[0]; p++) {
285 { BER_BVC( "add=" ), aloops },
286 { BER_BVC( "bind=" ), bloops },
287 { BER_BVC( "modify=" ), mloops },
288 { BER_BVC( "modrdn=" ), nloops },
289 { BER_BVC( "read=" ), rloops },
290 { BER_BVC( "search=" ), sloops },
295 for ( c = 0; types[c].type.bv_val; c++ ) {
296 if ( strncasecmp( p[0], types[c].type.bv_val, types[c].type.bv_len ) == 0 ) {
301 if ( types[c].type.bv_val == NULL ) {
302 usage( argv[0], 'l' );
305 if ( lutil_atoi( &n, &p[0][types[c].type.bv_len] ) != 0 ) {
306 usage( argv[0], 'l' );
309 snprintf( types[c].buf, sizeof( aloops ), "%d", n );
312 ldap_charray_free( l );
314 } else if ( lutil_atoi( &loops, optarg ) != 0 ) {
315 usage( argv[0], 'l' );
319 case 'L': /* the number of outerloops per client */
320 outerloops = strdup( optarg );
327 case 'P': /* prog directory */
328 progdir = strdup( optarg );
331 case 'p': /* the servers port number */
332 port = strdup( optarg );
335 case 'r': /* the number of retries in case of error */
336 retries = strdup( optarg );
339 case 't': /* the delay in seconds between each retry */
340 delay = strdup( optarg );
343 case 'w': /* the managers passwd */
344 passwd = ArgDup( optarg );
345 memset( optarg, '*', strlen( optarg ) );
357 usage( argv[0], '\0' );
362 if (( dirname == NULL ) || ( port == NULL && uri == NULL ) ||
363 ( manager == NULL ) || ( passwd == NULL ) || ( progdir == NULL ))
365 usage( argv[0], '\0' );
369 children = malloc( maxkids * sizeof(HANDLE) );
371 /* get the file list */
372 if ( ( datadir = opendir( dirname )) == NULL ) {
373 fprintf( stderr, "%s: couldn't open data directory \"%s\".\n",
375 exit( EXIT_FAILURE );
378 /* look for search, read, modrdn, and add/delete files */
379 for ( file = readdir( datadir ); file; file = readdir( datadir )) {
381 if ( !strcasecmp( file->d_name, TSEARCHFILE )) {
382 sfile = get_file_name( dirname, file->d_name );
384 } else if ( !strcasecmp( file->d_name, TREADFILE )) {
385 rfile = get_file_name( dirname, file->d_name );
387 } else if ( !strcasecmp( file->d_name, TMODRDNFILE )) {
388 nfile = get_file_name( dirname, file->d_name );
390 } else if ( !strcasecmp( file->d_name, TMODIFYFILE )) {
391 mfile = get_file_name( dirname, file->d_name );
393 } else if ( !strncasecmp( file->d_name, TADDFILE, strlen( TADDFILE ))
394 && ( anum < MAXREQS )) {
395 afiles[anum++] = get_file_name( dirname, file->d_name );
397 } else if ( !strcasecmp( file->d_name, TBINDFILE )) {
398 bfile = get_file_name( dirname, file->d_name );
406 passwd = getpassphrase( _("Enter LDAP Password: ") );
408 } else if ( pw_file ) {
411 if ( lutil_get_filed_password( pw_file, &pw ) ) {
412 exit( EXIT_FAILURE );
418 /* look for search requests */
420 snum = get_search_filters( sfile, sreqs, sattrs, sbase, slud );
423 /* look for read requests */
425 rnum = get_read_entries( rfile, rreqs, rflts );
428 /* look for modrdn requests */
430 nnum = get_read_entries( nfile, nreqs, NULL );
433 /* look for modify requests */
435 mnum = get_search_filters( mfile, mreqs, NULL, mdn, NULL );
438 /* look for bind requests */
440 bnum = get_search_filters( bfile, bcreds, battrs, breqs, NULL );
443 /* setup friendly option */
445 switch ( friendly ) {
454 /* NOTE: right now we don't need it more than twice */
460 if ( sloops[0] == '\0' ) snprintf( sloops, sizeof( sloops ), "%d", 10 * loops );
461 if ( rloops[0] == '\0' ) snprintf( rloops, sizeof( rloops ), "%d", 20 * loops );
462 if ( aloops[0] == '\0' ) snprintf( aloops, sizeof( aloops ), "%d", loops );
463 if ( nloops[0] == '\0' ) snprintf( nloops, sizeof( nloops ), "%d", loops );
464 if ( mloops[0] == '\0' ) snprintf( mloops, sizeof( mloops ), "%d", loops );
465 if ( bloops[0] == '\0' ) snprintf( bloops, sizeof( bloops ), "%d", 20 * loops );
468 * generate the search clients
472 snprintf( scmd, sizeof scmd, "%s" LDAP_DIRSEP SEARCHCMD,
474 sargs[sanum++] = scmd;
476 sargs[sanum++] = "-H";
477 sargs[sanum++] = uri;
479 sargs[sanum++] = "-h";
480 sargs[sanum++] = host;
481 sargs[sanum++] = "-p";
482 sargs[sanum++] = port;
484 sargs[sanum++] = "-D";
485 sargs[sanum++] = manager;
486 sargs[sanum++] = "-w";
487 sargs[sanum++] = passwd;
488 sargs[sanum++] = "-l";
489 sargs[sanum++] = sloops;
490 sargs[sanum++] = "-L";
491 sargs[sanum++] = outerloops;
492 sargs[sanum++] = "-r";
493 sargs[sanum++] = retries;
494 sargs[sanum++] = "-t";
495 sargs[sanum++] = delay;
497 sargs[sanum++] = friendlyOpt;
500 sargs[sanum++] = "-C";
503 sargs[sanum++] = "-A";
506 sargs[sanum++] = "-N";
509 sargs[sanum++] = "-i";
510 sargs[sanum++] = ignore;
512 sargs[sanum++] = "-b";
513 sargs[sanum++] = NULL; /* will hold the search base */
514 sargs[sanum++] = "-s";
515 sargs[sanum++] = NULL; /* will hold the search scope */
516 sargs[sanum++] = "-f";
517 sargs[sanum++] = NULL; /* will hold the search request */
519 sargs[sanum++] = NULL;
520 sargs[sanum++] = NULL; /* might hold the "attr" request */
526 * generate the read clients
530 snprintf( rcmd, sizeof rcmd, "%s" LDAP_DIRSEP READCMD,
532 rargs[ranum++] = rcmd;
534 rargs[ranum++] = "-H";
535 rargs[ranum++] = uri;
537 rargs[ranum++] = "-h";
538 rargs[ranum++] = host;
539 rargs[ranum++] = "-p";
540 rargs[ranum++] = port;
542 rargs[ranum++] = "-D";
543 rargs[ranum++] = manager;
544 rargs[ranum++] = "-w";
545 rargs[ranum++] = passwd;
546 rargs[ranum++] = "-l";
547 rargs[ranum++] = rloops;
548 rargs[ranum++] = "-L";
549 rargs[ranum++] = outerloops;
550 rargs[ranum++] = "-r";
551 rargs[ranum++] = retries;
552 rargs[ranum++] = "-t";
553 rargs[ranum++] = delay;
555 rargs[ranum++] = friendlyOpt;
558 rargs[ranum++] = "-C";
561 rargs[ranum++] = "-A";
564 rargs[ranum++] = "-i";
565 rargs[ranum++] = ignore;
567 rargs[ranum++] = "-e";
568 rargs[ranum++] = NULL; /* will hold the read entry */
570 rargs[ranum++] = NULL;
571 rargs[ranum++] = NULL; /* might hold the filter arg */
577 * generate the modrdn clients
581 snprintf( ncmd, sizeof ncmd, "%s" LDAP_DIRSEP MODRDNCMD,
583 nargs[nanum++] = ncmd;
585 nargs[nanum++] = "-H";
586 nargs[nanum++] = uri;
588 nargs[nanum++] = "-h";
589 nargs[nanum++] = host;
590 nargs[nanum++] = "-p";
591 nargs[nanum++] = port;
593 nargs[nanum++] = "-D";
594 nargs[nanum++] = manager;
595 nargs[nanum++] = "-w";
596 nargs[nanum++] = passwd;
597 nargs[nanum++] = "-l";
598 nargs[nanum++] = nloops;
599 nargs[nanum++] = "-L";
600 nargs[nanum++] = outerloops;
601 nargs[nanum++] = "-r";
602 nargs[nanum++] = retries;
603 nargs[nanum++] = "-t";
604 nargs[nanum++] = delay;
606 nargs[nanum++] = friendlyOpt;
609 nargs[nanum++] = "-C";
612 nargs[nanum++] = "-i";
613 nargs[nanum++] = ignore;
615 nargs[nanum++] = "-e";
616 nargs[nanum++] = NULL; /* will hold the modrdn entry */
620 * generate the modify clients
624 snprintf( mcmd, sizeof mcmd, "%s" LDAP_DIRSEP MODIFYCMD,
626 margs[manum++] = mcmd;
628 margs[manum++] = "-H";
629 margs[manum++] = uri;
631 margs[manum++] = "-h";
632 margs[manum++] = host;
633 margs[manum++] = "-p";
634 margs[manum++] = port;
636 margs[manum++] = "-D";
637 margs[manum++] = manager;
638 margs[manum++] = "-w";
639 margs[manum++] = passwd;
640 margs[manum++] = "-l";
641 margs[manum++] = mloops;
642 margs[manum++] = "-L";
643 margs[manum++] = outerloops;
644 margs[manum++] = "-r";
645 margs[manum++] = retries;
646 margs[manum++] = "-t";
647 margs[manum++] = delay;
649 margs[manum++] = friendlyOpt;
652 margs[manum++] = "-C";
655 margs[manum++] = "-i";
656 margs[manum++] = ignore;
658 margs[manum++] = "-e";
659 margs[manum++] = NULL; /* will hold the modify entry */
660 margs[manum++] = "-a";;
661 margs[manum++] = NULL; /* will hold the ava */
665 * generate the add/delete clients
669 snprintf( acmd, sizeof acmd, "%s" LDAP_DIRSEP ADDCMD,
671 aargs[aanum++] = acmd;
673 aargs[aanum++] = "-H";
674 aargs[aanum++] = uri;
676 aargs[aanum++] = "-h";
677 aargs[aanum++] = host;
678 aargs[aanum++] = "-p";
679 aargs[aanum++] = port;
681 aargs[aanum++] = "-D";
682 aargs[aanum++] = manager;
683 aargs[aanum++] = "-w";
684 aargs[aanum++] = passwd;
685 aargs[aanum++] = "-l";
686 aargs[aanum++] = aloops;
687 aargs[aanum++] = "-L";
688 aargs[aanum++] = outerloops;
689 aargs[aanum++] = "-r";
690 aargs[aanum++] = retries;
691 aargs[aanum++] = "-t";
692 aargs[aanum++] = delay;
694 aargs[aanum++] = friendlyOpt;
697 aargs[aanum++] = "-C";
700 aargs[aanum++] = "-i";
701 aargs[aanum++] = ignore;
703 aargs[aanum++] = "-f";
704 aargs[aanum++] = NULL; /* will hold the add data file */
708 * generate the bind clients
712 snprintf( bcmd, sizeof bcmd, "%s" LDAP_DIRSEP BINDCMD,
714 bargs[banum++] = bcmd;
716 bargs[banum++] = "-I"; /* init on each bind */
719 bargs[banum++] = "-H";
720 bargs[banum++] = uri;
722 bargs[banum++] = "-h";
723 bargs[banum++] = host;
724 bargs[banum++] = "-p";
725 bargs[banum++] = port;
727 bargs[banum++] = "-l";
728 bargs[banum++] = bloops;
729 bargs[banum++] = "-L";
730 bargs[banum++] = outerloops;
732 bargs[banum++] = "-r";
733 bargs[banum++] = retries;
734 bargs[banum++] = "-t";
735 bargs[banum++] = delay;
738 bargs[banum++] = friendlyOpt;
741 bargs[banum++] = "-C";
744 bargs[banum++] = "-i";
745 bargs[banum++] = ignore;
748 bargs[banum++] = "-B";
749 bargs_extra = &bargs[banum++];
751 bargs[banum++] = "-D";
752 bargs[banum++] = NULL;
753 bargs[banum++] = "-w";
754 bargs[banum++] = NULL;
757 #define DOREQ(n,j) ((n) && ((maxkids > (n)) ? ((j) < maxkids ) : ((j) < (n))))
759 for ( j = 0; j < MAXREQS; j++ ) {
761 if ( DOREQ( snum, j ) ) {
763 int x = sanum - sextra_args;
766 if ( sbase[jj] != NULL ) {
767 sargs[sanum - 7] = sbase[jj];
770 sargs[sanum - 7] = slud[jj]->lud_dn;
774 if ( slud[jj] != NULL ) {
775 sargs[sanum - 5] = (char *)ldap_pvt_scope2str( slud[jj]->lud_scope );
778 sargs[sanum - 5] = "sub";
782 if ( sreqs[jj] != NULL ) {
783 sargs[sanum - 3] = sreqs[jj];
785 } else if ( slud[jj]->lud_filter != NULL ) {
786 sargs[sanum - 3] = slud[jj]->lud_filter;
789 sargs[sanum - 3] = "(objectClass=*)";
796 if ( sattrs[jj] != NULL ) {
798 sargs[x++] = sattrs[jj];
802 if ( slud[jj] != NULL && slud[jj]->lud_attrs != NULL ) {
805 for ( i = 0; slud[jj]->lud_attrs[ i ] != NULL && x + i < MAXARGS - 1; i++ ) {
806 sargs[x + i] = slud[jj]->lud_attrs[ i ];
811 fork_child( scmd, sargs );
815 if ( DOREQ( rnum, j ) ) {
817 int x = ranum - rextra_args;
819 rargs[ranum - 3] = rreqs[jj];
820 if ( rflts[jj] != NULL ) {
822 rargs[x++] = rflts[jj];
825 fork_child( rcmd, rargs );
830 nargs[nanum - 1] = nreqs[j];
831 fork_child( ncmd, nargs );
836 margs[manum - 3] = mdn[j];
837 margs[manum - 1] = mreqs[j];
838 fork_child( mcmd, margs );
843 aargs[aanum - 1] = afiles[j];
844 fork_child( acmd, aargs );
848 if ( DOREQ( bnum, j ) ) {
852 int n = ((double)nextra)*rand()/(RAND_MAX + 1.0);
855 for ( e = extra; n-- > 0; e = e->next )
857 *bargs_extra = e->action;
860 if ( battrs[jj] != NULL ) {
861 bargs[banum - 3] = manager ? manager : "";
862 bargs[banum - 1] = passwd ? passwd : "";
864 bargs[banum - 2] = "-b";
865 bargs[banum - 1] = breqs[jj];
866 bargs[banum + 0] = "-f";
867 bargs[banum + 1] = bcreds[jj];
868 bargs[banum + 2] = "-a";
869 bargs[banum + 3] = battrs[jj];
872 bargs[banum - 3] = breqs[jj];
873 bargs[banum - 1] = bcreds[jj];
877 fork_child( bcmd, bargs );
884 exit( EXIT_SUCCESS );
888 get_file_name( char *dirname, char *filename )
890 char buf[MAXPATHLEN];
892 snprintf( buf, sizeof buf, "%s" LDAP_DIRSEP "%s",
894 return( strdup( buf ));
899 get_search_filters( char *filename, char *filters[], char *attrs[], char *bases[], LDAPURLDesc *luds[] )
904 if ( (fp = fopen( filename, "r" )) != NULL ) {
907 while (( filter < MAXREQS ) && ( fgets( line, BUFSIZ, fp ))) {
911 if (( nl = strchr( line, '\r' )) || ( nl = strchr( line, '\n' )))
914 if ( luds ) luds[filter] = NULL;
916 if ( luds && strncmp( line, "ldap:///", STRLENOF( "ldap:///" ) ) == 0 ) {
920 bases[filter] = NULL;
921 if ( ldap_url_parse( line, &lud ) != LDAP_URL_SUCCESS ) {
925 if ( lud->lud_dn == NULL || lud->lud_exts != NULL ) {
932 bases[filter] = ArgDup( line );
934 fgets( line, BUFSIZ, fp );
935 if (( nl = strchr( line, '\r' )) || ( nl = strchr( line, '\n' )))
938 filters[filter] = ArgDup( line );
940 if ( filters[filter][0] == '+') {
941 char *sep = strchr( filters[filter], ':' );
943 attrs[ filter ] = &filters[ filter ][ 1 ];
946 /* NOTE: don't free this! */
947 filters[ filter ] = &sep[ 1 ];
951 attrs[ filter ] = NULL;
965 get_read_entries( char *filename, char *entries[], char *filters[] )
970 if ( (fp = fopen( filename, "r" )) != NULL ) {
973 while (( entry < MAXREQS ) && ( fgets( line, BUFSIZ, fp ))) {
976 if (( nl = strchr( line, '\r' )) || ( nl = strchr( line, '\n' )))
978 if ( filters != NULL && line[0] == '+' ) {
981 if ( ldap_url_parse( &line[1], &lud ) != LDAP_URL_SUCCESS ) {
986 if ( lud->lud_dn == NULL || lud->lud_dn[ 0 ] == '\0' ) {
987 ldap_free_urldesc( lud );
992 entries[entry] = ArgDup( lud->lud_dn );
994 if ( lud->lud_filter ) {
995 filters[entry] = ArgDup( lud->lud_filter );
998 filters[entry] = ArgDup( "(objectClass=*)" );
1000 ldap_free_urldesc( lud );
1003 entries[entry] = ArgDup( line );
1015 #ifndef HAVE_WINSOCK
1017 fork_child( char *prog, char **args )
1019 /* note: obscures global pid var; intended */
1022 wait4kids( maxkids );
1024 switch ( pid = fork() ) {
1027 /* The __LIBASCII execvp only handles ASCII "prog",
1028 * we still need to translate the arg vec ourselves.
1030 { char *arg2[MAXREQS];
1033 for (i=0; args[i]; i++) {
1034 arg2[i] = ArgDup(args[i]);
1040 execvp( prog, args );
1041 tester_perror( "execvp", NULL );
1042 exit( EXIT_FAILURE );
1045 case -1: /* trouble */
1046 tester_perror( "fork", NULL );
1049 default: /* parent */
1056 wait4kids( int nkidval )
1060 while ( nkids >= nkidval ) {
1063 if ( WIFSTOPPED(status) ) {
1065 "stopping: child stopped with signal %d\n",
1066 (int) WSTOPSIG(status) );
1068 } else if ( WIFSIGNALED(status) ) {
1070 "stopping: child terminated with signal %d%s\n",
1071 (int) WTERMSIG(status),
1073 WCOREDUMP(status) ? ", core dumped" : ""
1078 exit( WEXITSTATUS(status) );
1080 } else if ( WEXITSTATUS(status) != 0 ) {
1082 "stopping: child exited with status %d\n",
1083 (int) WEXITSTATUS(status) );
1084 exit( WEXITSTATUS(status) );
1094 wait4kids( int nkidval )
1098 while ( nkids >= nkidval ) {
1099 rc = WaitForMultipleObjects( nkids, children, FALSE, INFINITE );
1100 for ( i=rc - WAIT_OBJECT_0; i<nkids-1; i++)
1101 children[i] = children[i+1];
1107 fork_child( char *prog, char **args )
1111 wait4kids( maxkids );
1113 rc = _spawnvp( _P_NOWAIT, prog, args );
1116 tester_perror( "_spawnvp", NULL );
1118 children[nkids++] = (HANDLE)rc;