13 #include <sys/param.h>
15 #include "ldapconfig.h"
18 #define SEARCHCMD "slapd-search"
19 #define READCMD "slapd-read"
20 #define ADDCMD "slapd-addel"
25 #define TSEARCHFILE "do_search.0"
26 #define TREADFILE "do_read.0"
27 #define TADDFILE "do_add."
29 static char *get_file_name( char *dirname, char *filename );
30 static int get_search_filters( char *filename, char *filters[] );
31 static int get_read_entries( char *filename, char *entries[] );
32 static void fork_child( char *prog, char *args[] );
33 static void wait4kids( int nkidval );
35 static int maxkids = 20;
41 fprintf( stderr, "usage: %s [-h <host>] -p <port> -D <manager> -w <passwd> -d <datadir> -b <baseDN> [-j <maxchild>] [-l <loops>] -P <progdir>\n", name );
46 main( int argc, char **argv )
49 char *host = "localhost";
65 char *afiles[MAXREQS];
69 char scmd[MAXPATHLEN];
72 char rcmd[MAXPATHLEN];
75 char acmd[MAXPATHLEN];
77 while ( (i = getopt( argc, argv, "h:p:D:w:b:d:j:l:P:" )) != EOF ) {
79 case 'h': /* slapd host */
80 host = strdup( optarg );
83 case 'p': /* the servers port number */
84 port = strdup( optarg );
87 case 'D': /* slapd manager */
88 manager = strdup( optarg );
91 case 'w': /* the managers passwd */
92 passwd = strdup( optarg );
95 case 'b': /* the base DN */
96 sbase = strdup( optarg );
99 case 'd': /* data directory */
100 dirname = strdup( optarg );
103 case 'P': /* prog directory */
104 progdir = strdup( optarg );
107 case 'j': /* the number of parallel clients */
108 maxkids = atoi( optarg );
111 case 'l': /* the number of loops per client */
112 loops = strdup( optarg );
121 if (( dirname == NULL ) || ( sbase == NULL ) || ( port == NULL ) ||
122 ( manager == NULL ) || ( passwd == NULL ) || ( progdir == NULL ))
125 /* get the file list */
126 if ( ( datadir = opendir( dirname )) == NULL ) {
128 fprintf( stderr, "%s: couldn't open data directory \"%s\".\n",
134 /* look for search, read, and add/delete files */
135 for ( file = readdir( datadir ); file; file = readdir( datadir )) {
137 if ( !strcasecmp( file->d_name, TSEARCHFILE )) {
138 sfile = get_file_name( dirname, file->d_name );
140 } else if ( !strcasecmp( file->d_name, TREADFILE )) {
141 rfile = get_file_name( dirname, file->d_name );
143 } else if ( !strncasecmp( file->d_name, TADDFILE, strlen( TADDFILE ))
144 && ( anum < MAXREQS )) {
145 afiles[anum++] = get_file_name( dirname, file->d_name );
150 /* look for search requests */
152 snum = get_search_filters( sfile, sreqs );
155 /* look for read requests */
157 rnum = get_read_entries( rfile, rreqs );
161 * generate the search clients
165 sprintf( scmd, "%s%s%s", progdir, DEFAULT_DIRSEP, SEARCHCMD );
166 sargs[sanum++] = scmd;
167 sargs[sanum++] = "-h";
168 sargs[sanum++] = host;
169 sargs[sanum++] = "-p";
170 sargs[sanum++] = port;
171 sargs[sanum++] = "-b";
172 sargs[sanum++] = sbase;
173 sargs[sanum++] = "-l";
174 sargs[sanum++] = loops;
175 sargs[sanum++] = "-f";
176 sargs[sanum++] = NULL; /* will hold the search request */
177 sargs[sanum++] = NULL;
180 * generate the read clients
184 sprintf( rcmd, "%s%s%s", progdir, DEFAULT_DIRSEP, READCMD );
185 rargs[ranum++] = rcmd;
186 rargs[ranum++] = "-h";
187 rargs[ranum++] = host;
188 rargs[ranum++] = "-p";
189 rargs[ranum++] = port;
190 rargs[ranum++] = "-l";
191 rargs[ranum++] = loops;
192 rargs[ranum++] = "-e";
193 rargs[ranum++] = NULL; /* will hold the read entry */
194 rargs[ranum++] = NULL;
197 * generate the add/delete clients
201 sprintf( acmd, "%s%s%s", progdir, DEFAULT_DIRSEP, ADDCMD );
202 aargs[aanum++] = acmd;
203 aargs[aanum++] = "-h";
204 aargs[aanum++] = host;
205 aargs[aanum++] = "-p";
206 aargs[aanum++] = port;
207 aargs[aanum++] = "-D";
208 aargs[aanum++] = manager;
209 aargs[aanum++] = "-w";
210 aargs[aanum++] = passwd;
211 aargs[aanum++] = "-l";
212 aargs[aanum++] = loops;
213 aargs[aanum++] = "-f";
214 aargs[aanum++] = NULL; /* will hold the add data file */
215 aargs[aanum++] = NULL;
217 for ( j = 0; j < MAXREQS; j++ ) {
221 sargs[sanum - 2] = sreqs[j];
222 fork_child( scmd, sargs );
228 rargs[ranum - 2] = rreqs[j];
229 fork_child( rcmd, rargs );
235 aargs[aanum - 2] = afiles[j];
236 fork_child( acmd, aargs );
248 get_file_name( char *dirname, char *filename )
250 char buf[MAXPATHLEN];
252 sprintf( buf, "%s%s%s", dirname, DEFAULT_DIRSEP, filename );
253 return( strdup( buf ));
258 get_search_filters( char *filename, char *filters[] )
263 if ( (fp = fopen( filename, "r" )) != NULL ) {
266 while (( filter < MAXREQS ) && ( fgets( line, BUFSIZ, fp ))) {
269 if (( nl = strchr( line, '\r' )) || ( nl = strchr( line, '\n' )))
271 filters[filter++] = strdup( line );
282 get_read_entries( char *filename, char *entries[] )
287 if ( (fp = fopen( filename, "r" )) != NULL ) {
290 while (( entry < MAXREQS ) && ( fgets( line, BUFSIZ, fp ))) {
293 if (( nl = strchr( line, '\r' )) || ( nl = strchr( line, '\n' )))
295 entries[entry++] = strdup( line );
306 fork_child( char *prog, char *args[] )
310 wait4kids( maxkids );
312 switch ( pid = fork() ) {
314 execvp( prog, args );
315 fprintf( stderr, "%s: ", prog );
320 case -1: /* trouble */
321 fprintf( stderr, "Could not fork to run %s\n", prog );
325 default: /* parent */
332 wait4kids( int nkidval )
337 while ( nkids >= nkidval ) {
339 p = (unsigned char *) &status;
340 if ( p[sizeof(int) - 1] == 0177 ) {
342 "stopping: child stopped with signal %d\n",
343 p[sizeof(int) - 2] );
344 } else if ( p[sizeof(int) - 1] != 0 ) {
346 "stopping: child terminated with signal %d\n",
347 p[sizeof(int) - 1] );
348 exit( p[sizeof(int) - 1] );
349 } else if ( p[sizeof(int) - 2] != 0 ) {
351 "stopping: child exited with status %d\n",
352 p[sizeof(int) - 2] );
353 exit( p[sizeof(int) - 2] );