]> git.sur5r.net Git - openldap/blob - tests/progs/slapd-bind.c
Merge remote-tracking branch 'origin/mdb.RE/0.9'
[openldap] / tests / progs / slapd-bind.c
1 /* $OpenLDAP$ */
2 /* This work is part of OpenLDAP Software <http://www.openldap.org/>.
3  *
4  * Copyright 1999-2015 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 Howard Chu for inclusion
17  * in OpenLDAP Software.
18  */
19
20 #include "portable.h"
21
22 #include <stdio.h>
23
24 #include "ac/stdlib.h"
25 #include "ac/time.h"
26
27 #include "ac/ctype.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 #include "ac/time.h"
34
35 #include "ldap.h"
36 #include "lutil.h"
37 #include "lber_pvt.h"
38 #include "ldap_pvt.h"
39
40 #include "slapd-common.h"
41
42 #define LOOPS   100
43
44 static int
45 do_bind( char *uri, char *dn, struct berval *pass, int maxloop,
46         int force, int chaserefs, int noinit, LDAP **ldp,
47         int action_type, void *action );
48
49 static int
50 do_base( char *uri, char *dn, struct berval *pass, char *base, char *filter, char *pwattr,
51         int maxloop, int force, int chaserefs, int noinit, int delay,
52         int action_type, void *action );
53
54 /* This program can be invoked two ways: if -D is used to specify a Bind DN,
55  * that DN will be used repeatedly for all of the Binds. If instead -b is used
56  * to specify a base DN, a search will be done for all "person" objects under
57  * that base DN. Then DNs from this list will be randomly selected for each
58  * Bind request. All of the users must have identical passwords. Also it is
59  * assumed that the users are all onelevel children of the base.
60  */
61 static void
62 usage( char *name, char opt )
63 {
64         if ( opt ) {
65                 fprintf( stderr, "%s: unable to handle option \'%c\'\n\n",
66                         name, opt );
67         }
68
69         fprintf( stderr, "usage: %s "
70                 "[-H uri | -h <host> [-p port]] "
71                 "[-D <dn> [-w <passwd>]] "
72                 "[-b <baseDN> [-f <searchfilter>] [-a pwattr]] "
73                 "[-l <loops>] "
74                 "[-L <outerloops>] "
75                 "[-B <extra>[,...]] "
76                 "[-F] "
77                 "[-C] "
78                 "[-I] "
79                 "[-i <ignore>] "
80                 "[-t delay]\n",
81                 name );
82         exit( EXIT_FAILURE );
83 }
84
85 int
86 main( int argc, char **argv )
87 {
88         int             i;
89         char            *uri = NULL;
90         char            *host = "localhost";
91         char            *dn = NULL;
92         char            *base = NULL;
93         char            *filter = "(objectClass=person)";
94         struct berval   pass = { 0, NULL };
95         char            *pwattr = NULL;
96         int             port = -1;
97         int             loops = LOOPS;
98         int             outerloops = 1;
99         int             force = 0;
100         int             chaserefs = 0;
101         int             noinit = 1;
102         int             delay = 0;
103
104         /* extra action to do after bind... */
105         struct berval   type[] = {
106                 BER_BVC( "tester=" ),
107                 BER_BVC( "add=" ),
108                 BER_BVC( "bind=" ),
109                 BER_BVC( "modify=" ),
110                 BER_BVC( "modrdn=" ),
111                 BER_BVC( "read=" ),
112                 BER_BVC( "search=" ),
113                 BER_BVNULL
114         };
115
116         LDAPURLDesc     *extra_ludp = NULL;
117
118         tester_init( "slapd-bind", TESTER_BIND );
119
120         /* by default, tolerate invalid credentials */
121         tester_ignore_str2errlist( "INVALID_CREDENTIALS" );
122
123         while ( ( i = getopt( argc, argv, "a:B:b:D:Ff:H:h:Ii:L:l:p:t:w:" ) ) != EOF )
124         {
125                 switch ( i ) {
126                 case 'a':
127                         pwattr = optarg;
128                         break;
129
130                 case 'b':               /* base DN of a tree of user DNs */
131                         base = optarg;
132                         break;
133
134                 case 'B':
135                         {
136                         int     c;
137
138                         for ( c = 0; type[c].bv_val; c++ ) {
139                                 if ( strncasecmp( optarg, type[c].bv_val, type[c].bv_len ) == 0 )
140                                 {
141                                         break;
142                                 }
143                         }
144
145                         if ( type[c].bv_val == NULL ) {
146                                 usage( argv[0], 'B' );
147                         }
148
149                         switch ( c ) {
150                         case TESTER_TESTER:
151                         case TESTER_BIND:
152                                 /* invalid */
153                                 usage( argv[0], 'B' );
154
155                         case TESTER_SEARCH:
156                                 {
157                                 if ( ldap_url_parse( &optarg[type[c].bv_len], &extra_ludp ) != LDAP_URL_SUCCESS )
158                                 {
159                                         usage( argv[0], 'B' );
160                                 }
161                                 } break;
162
163                         case TESTER_ADDEL:
164                         case TESTER_MODIFY:
165                         case TESTER_MODRDN:
166                         case TESTER_READ:
167                                 /* nothing to do */
168                                 break;
169
170                         default:
171                                 assert( 0 );
172                         }
173
174                         } break;
175
176                 case 'C':
177                         chaserefs++;
178                         break;
179
180                 case 'H':               /* the server uri */
181                         uri = optarg;
182                         break;
183
184                 case 'h':               /* the servers host */
185                         host = optarg;
186                         break;
187
188                 case 'i':
189                         tester_ignore_str2errlist( optarg );
190                         break;
191
192                 case 'p':               /* the servers port */
193                         if ( lutil_atoi( &port, optarg ) != 0 ) {
194                                 usage( argv[0], 'p' );
195                         }
196                         break;
197
198                 case 'D':
199                         dn = optarg;
200                         break;
201
202                 case 'w':
203                         ber_str2bv( optarg, 0, 1, &pass );
204                         memset( optarg, '*', pass.bv_len );
205                         break;
206
207                 case 'l':               /* the number of loops */
208                         if ( lutil_atoi( &loops, optarg ) != 0 ) {
209                                 usage( argv[0], 'l' );
210                         }
211                         break;
212
213                 case 'L':               /* the number of outerloops */
214                         if ( lutil_atoi( &outerloops, optarg ) != 0 ) {
215                                 usage( argv[0], 'L' );
216                         }
217                         break;
218
219                 case 'f':
220                         filter = optarg;
221                         break;
222
223                 case 'F':
224                         force++;
225                         break;
226
227                 case 'I':
228                         /* reuse connection */
229                         noinit = 0;
230                         break;
231
232                 case 't':
233                         /* sleep between binds */
234                         if ( lutil_atoi( &delay, optarg ) != 0 ) {
235                                 usage( argv[0], 't' );
236                         }
237                         break;
238
239                 default:
240                         usage( argv[0], i );
241                         break;
242                 }
243         }
244
245         if ( port == -1 && uri == NULL ) {
246                 usage( argv[0], '\0' );
247         }
248
249         uri = tester_uri( uri, host, port );
250
251         for ( i = 0; i < outerloops; i++ ) {
252                 int rc;
253
254                 if ( base != NULL ) {
255                         rc = do_base( uri, dn, &pass, base, filter, pwattr, loops,
256                                 force, chaserefs, noinit, delay, -1, NULL );
257                 } else {
258                         rc = do_bind( uri, dn, &pass, loops,
259                                 force, chaserefs, noinit, NULL, -1, NULL );
260                 }
261                 if ( rc == LDAP_SERVER_DOWN )
262                         break;
263         }
264
265         exit( EXIT_SUCCESS );
266 }
267
268
269 static int
270 do_bind( char *uri, char *dn, struct berval *pass, int maxloop,
271         int force, int chaserefs, int noinit, LDAP **ldp,
272         int action_type, void *action )
273 {
274         LDAP    *ld = ldp ? *ldp : NULL;
275         int     i, rc = -1;
276
277         /* for internal search */
278         int     timelimit = 0;
279         int     sizelimit = 0;
280
281         switch ( action_type ) {
282         case -1:
283                 break;
284
285         case TESTER_SEARCH:
286                 {
287                 LDAPURLDesc     *ludp = (LDAPURLDesc *)action;
288
289                 assert( action != NULL );
290
291                 if ( ludp->lud_exts != NULL ) {
292                         for ( i = 0; ludp->lud_exts[ i ] != NULL; i++ ) {
293                                 char    *ext = ludp->lud_exts[ i ];
294                                 int     crit = 0;
295
296                                 if (ext[0] == '!') {
297                                         crit++;
298                                         ext++;
299                                 }
300
301                                 if ( strncasecmp( ext, "x-timelimit=", STRLENOF( "x-timelimit=" ) ) == 0 ) {
302                                         if ( lutil_atoi( &timelimit, &ext[ STRLENOF( "x-timelimit=" ) ] ) && crit ) {
303                                                 tester_error( "unable to parse critical extension x-timelimit" );
304                                         }
305
306                                 } else if ( strncasecmp( ext, "x-sizelimit=", STRLENOF( "x-sizelimit=" ) ) == 0 ) {
307                                         if ( lutil_atoi( &sizelimit, &ext[ STRLENOF( "x-sizelimit=" ) ] ) && crit ) {
308                                                 tester_error( "unable to parse critical extension x-sizelimit" );
309                                         }
310
311                                 } else if ( crit ) {
312                                         tester_error( "unknown critical extension" );
313                                 }
314                         }
315                 }
316                 } break;
317
318         default:
319                 /* nothing to do yet */
320                 break;
321         }
322                         
323         if ( maxloop > 1 ) {
324                 fprintf( stderr, "PID=%ld - Bind(%d): dn=\"%s\".\n",
325                          (long) pid, maxloop, dn );
326         }
327
328         for ( i = 0; i < maxloop; i++ ) {
329                 if ( !noinit || ld == NULL ) {
330                         int version = LDAP_VERSION3;
331                         ldap_initialize( &ld, uri );
332                         if ( ld == NULL ) {
333                                 tester_perror( "ldap_initialize", NULL );
334                                 rc = -1;
335                                 break;
336                         }
337
338                         (void) ldap_set_option( ld, LDAP_OPT_PROTOCOL_VERSION,
339                                 &version ); 
340                         (void) ldap_set_option( ld, LDAP_OPT_REFERRALS,
341                                 chaserefs ? LDAP_OPT_ON: LDAP_OPT_OFF );
342                 }
343
344                 rc = ldap_sasl_bind_s( ld, dn, LDAP_SASL_SIMPLE, pass, NULL, NULL, NULL );
345                 if ( rc ) {
346                         int first = tester_ignore_err( rc );
347
348                         /* if ignore.. */
349                         if ( first ) {
350                                 /* only log if first occurrence */
351                                 if ( ( force < 2 && first > 0 ) || abs(first) == 1 ) {
352                                         tester_ldap_error( ld, "ldap_sasl_bind_s", NULL );
353                                 }
354                                 rc = LDAP_SUCCESS;
355
356                         } else {
357                                 tester_ldap_error( ld, "ldap_sasl_bind_s", NULL );
358                         }
359                 }
360
361                 switch ( action_type ) {
362                 case -1:
363                         break;
364
365                 case TESTER_SEARCH:
366                         {
367                         LDAPURLDesc     *ludp = (LDAPURLDesc *)action;
368                         LDAPMessage     *res = NULL;
369                         struct timeval  tv = { 0 }, *tvp = NULL;
370
371                         if ( timelimit ) {
372                                 tv.tv_sec = timelimit;
373                                 tvp = &tv;
374                         }
375
376                         assert( action != NULL );
377
378                         rc = ldap_search_ext_s( ld,
379                                 ludp->lud_dn, ludp->lud_scope,
380                                 ludp->lud_filter, ludp->lud_attrs, 0,
381                                 NULL, NULL, tvp, sizelimit, &res );
382                         ldap_msgfree( res );
383                         } break;
384
385                 default:
386                         /* nothing to do yet */
387                         break;
388                 }
389                         
390                 if ( !noinit ) {
391                         ldap_unbind_ext( ld, NULL, NULL );
392                         ld = NULL;
393                 }
394
395                 if ( rc != LDAP_SUCCESS ) {
396                         break;
397                 }
398         }
399
400         if ( maxloop > 1 ) {
401                 fprintf( stderr, "  PID=%ld - Bind done (%d).\n", (long) pid, rc );
402         }
403
404         if ( ldp && noinit ) {
405                 *ldp = ld;
406
407         } else if ( ld != NULL ) {
408                 ldap_unbind_ext( ld, NULL, NULL );
409         }
410
411         return rc;
412 }
413
414
415 static int
416 do_base( char *uri, char *dn, struct berval *pass, char *base, char *filter, char *pwattr,
417         int maxloop, int force, int chaserefs, int noinit, int delay,
418         int action_type, void *action )
419 {
420         LDAP    *ld = NULL;
421         int     i = 0;
422         int     rc = LDAP_SUCCESS;
423         ber_int_t msgid;
424         LDAPMessage *res, *msg;
425         char **dns = NULL;
426         struct berval *creds = NULL;
427         char *attrs[] = { LDAP_NO_ATTRS, NULL };
428         int ndns = 0;
429 #ifdef _WIN32
430         DWORD beg, end;
431 #else
432         struct timeval beg, end;
433 #endif
434         int version = LDAP_VERSION3;
435         char *nullstr = "";
436
437         ldap_initialize( &ld, uri );
438         if ( ld == NULL ) {
439                 tester_perror( "ldap_initialize", NULL );
440                 exit( EXIT_FAILURE );
441         }
442
443         (void) ldap_set_option( ld, LDAP_OPT_PROTOCOL_VERSION, &version );
444         (void) ldap_set_option( ld, LDAP_OPT_REFERRALS,
445                 chaserefs ? LDAP_OPT_ON: LDAP_OPT_OFF );
446
447         rc = ldap_sasl_bind_s( ld, dn, LDAP_SASL_SIMPLE, pass, NULL, NULL, NULL );
448         if ( rc != LDAP_SUCCESS ) {
449                 tester_ldap_error( ld, "ldap_sasl_bind_s", NULL );
450                 exit( EXIT_FAILURE );
451         }
452
453         fprintf( stderr, "PID=%ld - Bind(%d): base=\"%s\", filter=\"%s\" attr=\"%s\".\n",
454                         (long) pid, maxloop, base, filter, pwattr );
455
456         if ( pwattr != NULL ) {
457                 attrs[ 0 ] = pwattr;
458         }
459         rc = ldap_search_ext( ld, base, LDAP_SCOPE_SUBTREE,
460                         filter, attrs, 0, NULL, NULL, 0, 0, &msgid );
461         if ( rc != LDAP_SUCCESS ) {
462                 tester_ldap_error( ld, "ldap_search_ext", NULL );
463                 exit( EXIT_FAILURE );
464         }
465
466         while ( ( rc = ldap_result( ld, LDAP_RES_ANY, LDAP_MSG_ONE, NULL, &res ) ) > 0 )
467         {
468                 BerElement *ber;
469                 struct berval bv;
470                 int done = 0;
471
472                 for ( msg = ldap_first_message( ld, res ); msg;
473                         msg = ldap_next_message( ld, msg ) )
474                 {
475                         switch ( ldap_msgtype( msg ) ) {
476                         case LDAP_RES_SEARCH_ENTRY:
477                                 rc = ldap_get_dn_ber( ld, msg, &ber, &bv );
478                                 dns = realloc( dns, (ndns + 1)*sizeof(char *) );
479                                 dns[ndns] = ber_strdup( bv.bv_val );
480                                 if ( pwattr != NULL ) {
481                                         struct berval   **values = ldap_get_values_len( ld, msg, pwattr );
482
483                                         creds = realloc( creds, (ndns + 1)*sizeof(struct berval) );
484                                         if ( values == NULL ) {
485 novals:;
486                                                 creds[ndns].bv_len = 0;
487                                                 creds[ndns].bv_val = nullstr;
488
489                                         } else {
490                                                 static struct berval    cleartext = BER_BVC( "{CLEARTEXT} " );
491                                                 struct berval           value = *values[ 0 ];
492
493                                                 if ( value.bv_val[ 0 ] == '{' ) {
494                                                         char *end = ber_bvchr( &value, '}' );
495
496                                                         if ( end ) {
497                                                                 if ( ber_bvcmp( &value, &cleartext ) == 0 ) {
498                                                                         value.bv_val += cleartext.bv_len;
499                                                                         value.bv_len -= cleartext.bv_len;
500
501                                                                 } else {
502                                                                         ldap_value_free_len( values );
503                                                                         goto novals;
504                                                                 }
505                                                         }
506
507                                                 }
508
509                                                 ber_dupbv( &creds[ndns], &value );
510                                                 ldap_value_free_len( values );
511                                         }
512                                 }
513                                 ndns++;
514                                 ber_free( ber, 0 );
515                                 break;
516
517                         case LDAP_RES_SEARCH_RESULT:
518                                 done = 1;
519                                 break;
520                         }
521                         if ( done )
522                                 break;
523                 }
524                 ldap_msgfree( res );
525                 if ( done ) break;
526         }
527
528 #ifdef _WIN32
529         beg = GetTickCount();
530 #else
531         gettimeofday( &beg, NULL );
532 #endif
533
534         if ( ndns == 0 ) {
535                 tester_error( "No DNs" );
536                 return 1;
537         }
538
539         fprintf( stderr, "  PID=%ld - Bind base=\"%s\" filter=\"%s\" got %d values.\n",
540                 (long) pid, base, filter, ndns );
541
542         /* Ok, got list of DNs, now start binding to each */
543         for ( i = 0; i < maxloop; i++ ) {
544                 int             j;
545                 struct berval   cred = { 0, NULL };
546
547
548 #if 0   /* use high-order bits for better randomness (Numerical Recipes in "C") */
549                 j = rand() % ndns;
550 #endif
551                 j = ((double)ndns)*rand()/(RAND_MAX + 1.0);
552
553                 if ( creds && !BER_BVISEMPTY( &creds[j] ) ) {
554                         cred = creds[j];
555                 }
556
557                 if ( do_bind( uri, dns[j], &cred, 1, force, chaserefs, noinit, &ld,
558                         action_type, action ) && !force )
559                 {
560                         break;
561                 }
562
563                 if ( delay ) {
564                         sleep( delay );
565                 }
566         }
567
568         if ( ld != NULL ) {
569                 ldap_unbind_ext( ld, NULL, NULL );
570                 ld = NULL;
571         }
572
573 #ifdef _WIN32
574         end = GetTickCount();
575         end -= beg;
576
577         fprintf( stderr, "  PID=%ld - Bind done %d in %d.%03d seconds.\n",
578                 (long) pid, i, end / 1000, end % 1000 );
579 #else
580         gettimeofday( &end, NULL );
581         end.tv_usec -= beg.tv_usec;
582         if (end.tv_usec < 0 ) {
583                 end.tv_usec += 1000000;
584                 end.tv_sec -= 1;
585         }
586         end.tv_sec -= beg.tv_sec;
587
588         fprintf( stderr, "  PID=%ld - Bind done %d in %ld.%06ld seconds.\n",
589                 (long) pid, i, (long) end.tv_sec, (long) end.tv_usec );
590 #endif
591
592         if ( dns ) {
593                 for ( i = 0; i < ndns; i++ ) {
594                         ber_memfree( dns[i] );
595                 }
596                 free( dns );
597         }
598
599         if ( creds ) {
600                 for ( i = 0; i < ndns; i++ ) {
601                         if ( creds[i].bv_val != nullstr ) {
602                                 ber_memfree( creds[i].bv_val );
603                         }
604                 }
605                 free( creds );
606         }
607
608         return 0;
609 }