]> git.sur5r.net Git - openldap/blob - libraries/libldap/cyrus.c
I suggest the coder(s) of the cyrus stuff to check out the prompt stuff
[openldap] / libraries / libldap / cyrus.c
1 /* $OpenLDAP$ */
2 /*
3  * Copyright 1999-2003 The OpenLDAP Foundation, All Rights Reserved.
4  * COPYING RESTRICTIONS APPLY, see COPYRIGHT file
5  */
6
7 #include "portable.h"
8
9 #include <stdio.h>
10
11 #include <ac/socket.h>
12 #include <ac/stdlib.h>
13 #include <ac/string.h>
14 #include <ac/time.h>
15 #include <ac/errno.h>
16 #include <ac/ctype.h>
17
18 #include "ldap-int.h"
19
20 #ifdef HAVE_CYRUS_SASL
21
22 #ifdef LDAP_R_COMPILE
23 ldap_pvt_thread_mutex_t ldap_int_sasl_mutex;
24 #endif
25
26 #ifdef HAVE_SASL_SASL_H
27 #include <sasl/sasl.h>
28 #else
29 #include <sasl.h>
30 #endif
31
32 #if SASL_VERSION_MAJOR >= 2
33 #define SASL_CONST const
34 #else
35 #define SASL_CONST
36 #endif
37
38 /*
39 * Various Cyrus SASL related stuff.
40 */
41
42 int ldap_int_sasl_init( void )
43 {
44         /* XXX not threadsafe */
45         static int sasl_initialized = 0;
46
47         static sasl_callback_t client_callbacks[] = {
48 #ifdef SASL_CB_GETREALM
49                 { SASL_CB_GETREALM, NULL, NULL },
50 #endif
51                 { SASL_CB_USER, NULL, NULL },
52                 { SASL_CB_AUTHNAME, NULL, NULL },
53                 { SASL_CB_PASS, NULL, NULL },
54                 { SASL_CB_ECHOPROMPT, NULL, NULL },
55                 { SASL_CB_NOECHOPROMPT, NULL, NULL },
56                 { SASL_CB_LIST_END, NULL, NULL }
57         };
58
59 #ifdef HAVE_SASL_VERSION
60 #define SASL_BUILD_VERSION ((SASL_VERSION_MAJOR << 24) |\
61         (SASL_VERSION_MINOR << 16) | SASL_VERSION_STEP)
62
63         { int rc;
64         sasl_version( NULL, &rc );
65         if ( ((rc >> 16) != ((SASL_VERSION_MAJOR << 8)|SASL_VERSION_MINOR)) ||
66                 (rc & 0xffff) < SASL_VERSION_STEP) {
67
68 #ifdef NEW_LOGGING
69                 LDAP_LOG( TRANSPORT, INFO,
70                 "ldap_int_sasl_init: SASL version mismatch, got %x, wanted %x.\n",
71                         rc, SASL_BUILD_VERSION, 0 );
72 #else
73                 Debug( LDAP_DEBUG_ANY,
74                 "ldap_int_sasl_init: SASL version mismatch, got %x, wanted %x.\n",
75                         rc, SASL_BUILD_VERSION, 0 );
76 #endif
77                 return -1;
78         }
79         }
80 #endif
81         if ( sasl_initialized ) {
82                 return 0;
83         }
84
85 /* SASL 2 takes care of its own memory completely internally */
86 #if SASL_VERSION_MAJOR < 2 && !defined(CSRIMALLOC)
87         sasl_set_alloc(
88                 ber_memalloc,
89                 ber_memcalloc,
90                 ber_memrealloc,
91                 ber_memfree );
92 #endif /* CSRIMALLOC */
93
94 #ifdef LDAP_R_COMPILE
95         sasl_set_mutex(
96                 ldap_pvt_sasl_mutex_new,
97                 ldap_pvt_sasl_mutex_lock,
98                 ldap_pvt_sasl_mutex_unlock,    
99                 ldap_pvt_sasl_mutex_dispose );    
100
101         ldap_pvt_thread_mutex_init( &ldap_int_sasl_mutex );
102 #endif
103
104         if ( sasl_client_init( client_callbacks ) == SASL_OK ) {
105                 sasl_initialized = 1;
106                 return 0;
107         }
108
109 #if SASL_VERSION_MAJOR < 2
110         /* A no-op to make sure we link with Cyrus 1.5 */
111         sasl_client_auth( NULL, NULL, NULL, 0, NULL, NULL );
112 #endif
113         return -1;
114 }
115
116 /*
117  * SASL encryption support for LBER Sockbufs
118  */
119
120 struct sb_sasl_data {
121         sasl_conn_t             *sasl_context;
122         unsigned                *sasl_maxbuf;
123         Sockbuf_Buf             sec_buf_in;
124         Sockbuf_Buf             buf_in;
125         Sockbuf_Buf             buf_out;
126 };
127
128 static int
129 sb_sasl_setup( Sockbuf_IO_Desc *sbiod, void *arg )
130 {
131         struct sb_sasl_data     *p;
132
133         assert( sbiod != NULL );
134
135         p = LBER_MALLOC( sizeof( *p ) );
136         if ( p == NULL )
137                 return -1;
138         p->sasl_context = (sasl_conn_t *)arg;
139         ber_pvt_sb_buf_init( &p->sec_buf_in );
140         ber_pvt_sb_buf_init( &p->buf_in );
141         ber_pvt_sb_buf_init( &p->buf_out );
142         if ( ber_pvt_sb_grow_buffer( &p->sec_buf_in, SASL_MIN_BUFF_SIZE ) < 0 ) {
143                 LBER_FREE( p );
144                 errno = ENOMEM;
145                 return -1;
146         }
147         sasl_getprop( p->sasl_context, SASL_MAXOUTBUF,
148                 (SASL_CONST void **) &p->sasl_maxbuf );
149
150         sbiod->sbiod_pvt = p;
151
152         return 0;
153 }
154
155 static int
156 sb_sasl_remove( Sockbuf_IO_Desc *sbiod )
157 {
158         struct sb_sasl_data     *p;
159
160         assert( sbiod != NULL );
161         
162         p = (struct sb_sasl_data *)sbiod->sbiod_pvt;
163 #if SASL_VERSION_MAJOR >= 2
164         /*
165          * SASLv2 encode/decode buffers are managed by
166          * libsasl2. Ensure they are not freed by liblber.
167          */
168         p->buf_in.buf_base = NULL;
169         p->buf_out.buf_base = NULL;
170 #endif
171         ber_pvt_sb_buf_destroy( &p->sec_buf_in );
172         ber_pvt_sb_buf_destroy( &p->buf_in );
173         ber_pvt_sb_buf_destroy( &p->buf_out );
174         LBER_FREE( p );
175         sbiod->sbiod_pvt = NULL;
176         return 0;
177 }
178
179 static ber_len_t
180 sb_sasl_pkt_length( const unsigned char *buf, unsigned max, int debuglevel )
181 {
182         ber_len_t               size;
183
184         assert( buf != NULL );
185
186         size = buf[0] << 24
187                 | buf[1] << 16
188                 | buf[2] << 8
189                 | buf[3];
190    
191         if ( size > SASL_MAX_BUFF_SIZE ) {
192                 /* somebody is trying to mess me up. */
193                 ber_log_printf( LDAP_DEBUG_ANY, debuglevel,
194                         "sb_sasl_pkt_length: received illegal packet length "
195                         "of %lu bytes\n", (unsigned long)size );      
196                 size = 16; /* this should lead to an error. */
197         }
198
199         return size + 4; /* include the size !!! */
200 }
201
202 /* Drop a processed packet from the input buffer */
203 static void
204 sb_sasl_drop_packet ( Sockbuf_Buf *sec_buf_in, unsigned max, int debuglevel )
205 {
206         ber_slen_t                      len;
207
208         len = sec_buf_in->buf_ptr - sec_buf_in->buf_end;
209         if ( len > 0 )
210                 AC_MEMCPY( sec_buf_in->buf_base, sec_buf_in->buf_base +
211                         sec_buf_in->buf_end, len );
212    
213         if ( len >= 4 ) {
214                 sec_buf_in->buf_end = sb_sasl_pkt_length( sec_buf_in->buf_base,
215                         max, debuglevel);
216         }
217         else {
218                 sec_buf_in->buf_end = 0;
219         }
220         sec_buf_in->buf_ptr = len;
221 }
222
223 static ber_slen_t
224 sb_sasl_read( Sockbuf_IO_Desc *sbiod, void *buf, ber_len_t len)
225 {
226         struct sb_sasl_data     *p;
227         ber_slen_t              ret, bufptr;
228    
229         assert( sbiod != NULL );
230         assert( SOCKBUF_VALID( sbiod->sbiod_sb ) );
231
232         p = (struct sb_sasl_data *)sbiod->sbiod_pvt;
233
234         /* Are there anything left in the buffer? */
235         ret = ber_pvt_sb_copy_out( &p->buf_in, buf, len );
236         bufptr = ret;
237         len -= ret;
238
239         if ( len == 0 )
240                 return bufptr;
241
242 #if SASL_VERSION_MAJOR >= 2
243         ber_pvt_sb_buf_init( &p->buf_in );
244 #else
245         ber_pvt_sb_buf_destroy( &p->buf_in );
246 #endif
247
248         /* Read the length of the packet */
249         while ( p->sec_buf_in.buf_ptr < 4 ) {
250                 ret = LBER_SBIOD_READ_NEXT( sbiod, p->sec_buf_in.buf_base,
251                         4 - p->sec_buf_in.buf_ptr );
252 #ifdef EINTR
253                 if ( ( ret < 0 ) && ( errno == EINTR ) )
254                         continue;
255 #endif
256                 if ( ret <= 0 )
257                         return bufptr ? bufptr : ret;
258
259                 p->sec_buf_in.buf_ptr += ret;
260         }
261
262         /* The new packet always starts at p->sec_buf_in.buf_base */
263         ret = sb_sasl_pkt_length( p->sec_buf_in.buf_base,
264                 *p->sasl_maxbuf, sbiod->sbiod_sb->sb_debug );
265
266         /* Grow the packet buffer if neccessary */
267         if ( ( p->sec_buf_in.buf_size < (ber_len_t) ret ) && 
268                 ber_pvt_sb_grow_buffer( &p->sec_buf_in, ret ) < 0 )
269         {
270                 errno = ENOMEM;
271                 return -1;
272         }
273         p->sec_buf_in.buf_end = ret;
274
275         /* Did we read the whole encrypted packet? */
276         while ( p->sec_buf_in.buf_ptr < p->sec_buf_in.buf_end ) {
277                 /* No, we have got only a part of it */
278                 ret = p->sec_buf_in.buf_end - p->sec_buf_in.buf_ptr;
279
280                 ret = LBER_SBIOD_READ_NEXT( sbiod, p->sec_buf_in.buf_base +
281                         p->sec_buf_in.buf_ptr, ret );
282 #ifdef EINTR
283                 if ( ( ret < 0 ) && ( errno == EINTR ) )
284                         continue;
285 #endif
286                 if ( ret <= 0 )
287                         return bufptr ? bufptr : ret;
288
289                 p->sec_buf_in.buf_ptr += ret;
290         }
291
292         /* Decode the packet */
293         ret = sasl_decode( p->sasl_context, p->sec_buf_in.buf_base,
294                 p->sec_buf_in.buf_end,
295                 (SASL_CONST char **)&p->buf_in.buf_base,
296                 (unsigned *)&p->buf_in.buf_end );
297
298         /* Drop the packet from the input buffer */
299         sb_sasl_drop_packet( &p->sec_buf_in,
300                         *p->sasl_maxbuf, sbiod->sbiod_sb->sb_debug );
301
302         if ( ret != SASL_OK ) {
303                 ber_log_printf( LDAP_DEBUG_ANY, sbiod->sbiod_sb->sb_debug,
304                         "sb_sasl_read: failed to decode packet: %s\n",
305                         sasl_errstring( ret, NULL, NULL ) );
306                 errno = EIO;
307                 return -1;
308         }
309         
310         p->buf_in.buf_size = p->buf_in.buf_end;
311
312         bufptr += ber_pvt_sb_copy_out( &p->buf_in, (char*) buf + bufptr, len );
313
314         return bufptr;
315 }
316
317 static ber_slen_t
318 sb_sasl_write( Sockbuf_IO_Desc *sbiod, void *buf, ber_len_t len)
319 {
320         struct sb_sasl_data     *p;
321         int                     ret;
322
323         assert( sbiod != NULL );
324         assert( SOCKBUF_VALID( sbiod->sbiod_sb ) );
325
326         p = (struct sb_sasl_data *)sbiod->sbiod_pvt;
327
328         /* Are there anything left in the buffer? */
329         if ( p->buf_out.buf_ptr != p->buf_out.buf_end ) {
330                 ret = ber_pvt_sb_do_write( sbiod, &p->buf_out );
331                 if ( ret < 0 )
332                         return ret;
333                 /* Still have something left?? */
334                 if ( p->buf_out.buf_ptr != p->buf_out.buf_end ) {
335                         errno = EAGAIN;
336                         return 0;
337                 }
338         }
339
340         /* now encode the next packet. */
341 #if SASL_VERSION_MAJOR >= 2
342         ber_pvt_sb_buf_init( &p->buf_out );
343         /* sasl v2 makes sure this number is correct */
344         if ( len > *p->sasl_maxbuf )
345                 len = *p->sasl_maxbuf;
346 #else
347         ber_pvt_sb_buf_destroy( &p->buf_out );
348         if ( len > *p->sasl_maxbuf - 100 )
349                 len = *p->sasl_maxbuf - 100;    /* For safety margin */
350 #endif
351         ret = sasl_encode( p->sasl_context, buf, len,
352                 (SASL_CONST char **)&p->buf_out.buf_base,
353                 (unsigned *)&p->buf_out.buf_size );
354         if ( ret != SASL_OK ) {
355                 ber_log_printf( LDAP_DEBUG_ANY, sbiod->sbiod_sb->sb_debug,
356                         "sb_sasl_write: failed to encode packet: %s\n",
357                         sasl_errstring( ret, NULL, NULL ) );
358                 return -1;
359         }
360         p->buf_out.buf_end = p->buf_out.buf_size;
361
362         ret = ber_pvt_sb_do_write( sbiod, &p->buf_out );
363         if ( ret <= 0 )
364                 return ret;
365         return len;
366 }
367
368 static int
369 sb_sasl_ctrl( Sockbuf_IO_Desc *sbiod, int opt, void *arg )
370 {
371         struct sb_sasl_data     *p;
372
373         p = (struct sb_sasl_data *)sbiod->sbiod_pvt;
374
375         if ( opt == LBER_SB_OPT_DATA_READY ) {
376                 if ( p->buf_in.buf_ptr != p->buf_in.buf_end )
377                         return 1;
378         }
379         
380         return LBER_SBIOD_CTRL_NEXT( sbiod, opt, arg );
381 }
382
383 Sockbuf_IO ldap_pvt_sockbuf_io_sasl = {
384         sb_sasl_setup,          /* sbi_setup */
385         sb_sasl_remove,         /* sbi_remove */
386         sb_sasl_ctrl,           /* sbi_ctrl */
387         sb_sasl_read,           /* sbi_read */
388         sb_sasl_write,          /* sbi_write */
389         NULL                    /* sbi_close */
390 };
391
392 int ldap_pvt_sasl_install( Sockbuf *sb, void *ctx_arg )
393 {
394 #ifdef NEW_LOGGING
395         LDAP_LOG ( TRANSPORT, ENTRY, "ldap_pvt_sasl_install\n", 0, 0, 0 );
396 #else
397         Debug( LDAP_DEBUG_TRACE, "ldap_pvt_sasl_install\n",
398                 0, 0, 0 );
399 #endif
400
401         /* don't install the stuff unless security has been negotiated */
402
403         if ( !ber_sockbuf_ctrl( sb, LBER_SB_OPT_HAS_IO,
404                         &ldap_pvt_sockbuf_io_sasl ) )
405         {
406 #ifdef LDAP_DEBUG
407                 ber_sockbuf_add_io( sb, &ber_sockbuf_io_debug,
408                         LBER_SBIOD_LEVEL_APPLICATION, (void *)"sasl_" );
409 #endif
410                 ber_sockbuf_add_io( sb, &ldap_pvt_sockbuf_io_sasl,
411                         LBER_SBIOD_LEVEL_APPLICATION, ctx_arg );
412         }
413
414         return LDAP_SUCCESS;
415 }
416
417 static int
418 sasl_err2ldap( int saslerr )
419 {
420         int rc;
421
422         switch (saslerr) {
423                 case SASL_CONTINUE:
424                         rc = LDAP_MORE_RESULTS_TO_RETURN;
425                         break;
426                 case SASL_INTERACT:
427                         rc = LDAP_LOCAL_ERROR;
428                         break;
429                 case SASL_OK:
430                         rc = LDAP_SUCCESS;
431                         break;
432                 case SASL_FAIL:
433                         rc = LDAP_LOCAL_ERROR;
434                         break;
435                 case SASL_NOMEM:
436                         rc = LDAP_NO_MEMORY;
437                         break;
438                 case SASL_NOMECH:
439                         rc = LDAP_AUTH_UNKNOWN;
440                         break;
441                 case SASL_BADAUTH:
442                         rc = LDAP_AUTH_UNKNOWN;
443                         break;
444                 case SASL_NOAUTHZ:
445                         rc = LDAP_PARAM_ERROR;
446                         break;
447                 case SASL_TOOWEAK:
448                 case SASL_ENCRYPT:
449                         rc = LDAP_AUTH_UNKNOWN;
450                         break;
451                 default:
452                         rc = LDAP_LOCAL_ERROR;
453                         break;
454         }
455
456         assert( rc == LDAP_SUCCESS || LDAP_API_ERROR( rc ) );
457         return rc;
458 }
459
460 int
461 ldap_int_sasl_open(
462         LDAP *ld, 
463         LDAPConn *lc,
464         const char * host )
465 {
466         int rc;
467         sasl_conn_t *ctx;
468
469         assert( lc->lconn_sasl_ctx == NULL );
470
471         if ( host == NULL ) {
472                 ld->ld_errno = LDAP_LOCAL_ERROR;
473                 return ld->ld_errno;
474         }
475
476 #if SASL_VERSION_MAJOR >= 2
477         rc = sasl_client_new( "ldap", host, NULL, NULL,
478                 NULL, 0, &ctx );
479 #else
480         rc = sasl_client_new( "ldap", host, NULL,
481                 SASL_SECURITY_LAYER, &ctx );
482 #endif
483
484         if ( rc != SASL_OK ) {
485                 ld->ld_errno = sasl_err2ldap( rc );
486                 return ld->ld_errno;
487         }
488
489 #ifdef NEW_LOGGING
490         LDAP_LOG ( TRANSPORT, DETAIL1, "ldap_int_sasl_open: host=%s\n", 
491                 host, 0, 0 );
492 #else
493         Debug( LDAP_DEBUG_TRACE, "ldap_int_sasl_open: host=%s\n",
494                 host, 0, 0 );
495 #endif
496
497         lc->lconn_sasl_ctx = ctx;
498
499         return LDAP_SUCCESS;
500 }
501
502 int ldap_int_sasl_close( LDAP *ld, LDAPConn *lc )
503 {
504         sasl_conn_t *ctx = lc->lconn_sasl_ctx;
505
506         if( ctx != NULL ) {
507                 sasl_dispose( &ctx );
508                 lc->lconn_sasl_ctx = NULL;
509         }
510
511         return LDAP_SUCCESS;
512 }
513
514 int
515 ldap_int_sasl_bind(
516         LDAP                    *ld,
517         const char              *dn,
518         const char              *mechs,
519         LDAPControl             **sctrls,
520         LDAPControl             **cctrls,
521         unsigned                flags,
522         LDAP_SASL_INTERACT_PROC *interact,
523         void * defaults )
524 {
525         char *data;
526         const char *mech = NULL;
527         const char *pmech = NULL;
528         int                     saslrc, rc;
529         sasl_ssf_t              *ssf = NULL;
530         sasl_conn_t     *ctx;
531         sasl_interact_t *prompts = NULL;
532         unsigned credlen;
533         struct berval ccred;
534         ber_socket_t            sd;
535
536 #ifdef NEW_LOGGING
537         LDAP_LOG ( TRANSPORT, ARGS, "ldap_int_sasl_bind: %s\n", 
538                 mechs ? mechs : "<null>", 0, 0 );
539 #else
540         Debug( LDAP_DEBUG_TRACE, "ldap_int_sasl_bind: %s\n",
541                 mechs ? mechs : "<null>", 0, 0 );
542 #endif
543
544         /* do a quick !LDAPv3 check... ldap_sasl_bind will do the rest. */
545         if (ld->ld_version < LDAP_VERSION3) {
546                 ld->ld_errno = LDAP_NOT_SUPPORTED;
547                 return ld->ld_errno;
548         }
549
550         ber_sockbuf_ctrl( ld->ld_sb, LBER_SB_OPT_GET_FD, &sd );
551
552         if ( sd == AC_SOCKET_INVALID ) {
553                 /* not connected yet */
554                 int rc;
555
556                 rc = ldap_open_defconn( ld );
557                 if( rc < 0 ) return ld->ld_errno;
558
559                 ber_sockbuf_ctrl( ld->ld_sb, LBER_SB_OPT_GET_FD, &sd );
560
561                 if( sd == AC_SOCKET_INVALID ) {
562                         ld->ld_errno = LDAP_LOCAL_ERROR;
563                         return ld->ld_errno;
564                 }
565         }   
566
567         ctx = ld->ld_defconn->lconn_sasl_ctx;
568
569         if( ctx == NULL ) {
570                 ld->ld_errno = LDAP_LOCAL_ERROR;
571                 return ld->ld_errno;
572         }
573
574         /* (re)set security properties */
575         sasl_setprop( ctx, SASL_SEC_PROPS,
576                 &ld->ld_options.ldo_sasl_secprops );
577
578         ccred.bv_val = NULL;
579         ccred.bv_len = 0;
580
581         do {
582                 saslrc = sasl_client_start( ctx,
583                         mechs,
584 #if SASL_VERSION_MAJOR < 2
585                         NULL,
586 #endif
587                         &prompts,
588                         (SASL_CONST char **)&ccred.bv_val,
589                         &credlen,
590                         &mech );
591
592                 if( pmech == NULL && mech != NULL ) {
593                         pmech = mech;
594
595                         if( flags != LDAP_SASL_QUIET ) {
596                                 fprintf(stderr,
597                                         "SASL/%s authentication started\n",
598                                         pmech );
599                         }
600                 }
601
602                 if( saslrc == SASL_INTERACT ) {
603                         int res;
604
605 #if SASL_VERSION_MAJOR >= 2
606                         /* XXX the application should free interact results.
607                          * FIXME: this should happen only 
608                          * if saslrc == SASL_INTERACT
609                          *
610                          * I assume that prompts->result is not needed
611                          * by the subsequent call to (interact)() */
612                         if ( prompts != NULL && prompts->result != NULL ) {
613                                 LDAP_FREE( (void *)prompts->result );
614                                 prompts->result = NULL;
615                         }
616 #endif
617
618                         if( !interact ) break;
619                         res = (interact)( ld, flags, defaults, prompts );
620                         if( res != LDAP_SUCCESS ) {
621                                 break;
622                         }
623                 }
624         } while ( saslrc == SASL_INTERACT );
625
626         ccred.bv_len = credlen;
627
628         if ( (saslrc != SASL_OK) && (saslrc != SASL_CONTINUE) ) {
629                 ld->ld_errno = sasl_err2ldap( saslrc );
630 #if SASL_VERSION_MAJOR >= 2
631                 ld->ld_error = (char *)sasl_errdetail( ctx );
632 #endif
633                 return ld->ld_errno;
634         }
635
636         do {
637                 struct berval *scred;
638                 unsigned credlen;
639
640                 scred = NULL;
641
642                 rc = ldap_sasl_bind_s( ld, dn, mech, &ccred, sctrls, cctrls, &scred );
643
644                 if ( ccred.bv_val != NULL ) {
645 #if SASL_VERSION_MAJOR < 2
646                         LDAP_FREE( ccred.bv_val );
647 #endif
648                         ccred.bv_val = NULL;
649                 }
650
651                 if ( rc != LDAP_SUCCESS && rc != LDAP_SASL_BIND_IN_PROGRESS ) {
652                         if( scred && scred->bv_len ) {
653                                 /* and server provided us with data? */
654 #ifdef NEW_LOGGING
655                                 LDAP_LOG ( TRANSPORT, DETAIL1, 
656                                         "ldap_int_sasl_bind: rc=%d sasl=%d len=%ld\n", 
657                                         rc, saslrc, scred->bv_len );
658 #else
659                                 Debug( LDAP_DEBUG_TRACE,
660                                         "ldap_int_sasl_bind: rc=%d sasl=%d len=%ld\n",
661                                         rc, saslrc, scred->bv_len );
662 #endif
663                                 ber_bvfree( scred );
664                         }
665                         return ld->ld_errno;
666                 }
667
668                 if( rc == LDAP_SUCCESS && saslrc == SASL_OK ) {
669                         /* we're done, no need to step */
670                         if( scred && scred->bv_len ) {
671                                 /* but server provided us with data! */
672 #ifdef NEW_LOGGING
673                                 LDAP_LOG ( TRANSPORT, DETAIL1, 
674                                         "ldap_int_sasl_bind: rc=%d sasl=%d len=%ld\n", 
675                                         rc, saslrc, scred->bv_len );
676 #else
677                                 Debug( LDAP_DEBUG_TRACE,
678                                         "ldap_int_sasl_bind: rc=%d sasl=%d len=%ld\n",
679                                         rc, saslrc, scred->bv_len );
680 #endif
681                                 ber_bvfree( scred );
682                                 return ld->ld_errno = LDAP_LOCAL_ERROR;
683                         }
684                         break;
685                 }
686
687                 do {
688                         saslrc = sasl_client_step( ctx,
689                                 (scred == NULL) ? NULL : scred->bv_val,
690                                 (scred == NULL) ? 0 : scred->bv_len,
691                                 &prompts,
692                                 (SASL_CONST char **)&ccred.bv_val,
693                                 &credlen );
694
695 #ifdef NEW_LOGGING
696                                 LDAP_LOG ( TRANSPORT, DETAIL1, 
697                                         "ldap_int_sasl_bind: sasl_client_step: %d\n", saslrc,0,0 );
698 #else
699                         Debug( LDAP_DEBUG_TRACE, "sasl_client_step: %d\n",
700                                 saslrc, 0, 0 );
701 #endif
702
703 #if SASL_VERSION_MAJOR >= 2
704                         /* XXX the application should free interact results. */
705                         if ( prompts != NULL && prompts->result != NULL ) {
706                                 LDAP_FREE( (void *)prompts->result );
707                                 prompts->result = NULL;
708                         }
709 #endif
710
711                         if( saslrc == SASL_INTERACT ) {
712                                 int res;
713                                 if( !interact ) break;
714                                 res = (interact)( ld, flags, defaults, prompts );
715                                 if( res != LDAP_SUCCESS ) {
716                                         break;
717                                 }
718                         }
719                 } while ( saslrc == SASL_INTERACT );
720
721                 ccred.bv_len = credlen;
722                 ber_bvfree( scred );
723
724                 if ( (saslrc != SASL_OK) && (saslrc != SASL_CONTINUE) ) {
725                         ld->ld_errno = sasl_err2ldap( saslrc );
726 #if SASL_VERSION_MAJOR >= 2
727                         ld->ld_error = (char *)sasl_errdetail( ctx );
728 #endif
729                         return ld->ld_errno;
730                 }
731         } while ( rc == LDAP_SASL_BIND_IN_PROGRESS );
732
733         if ( rc != LDAP_SUCCESS ) {
734                 return rc;
735         }
736
737         if ( saslrc != SASL_OK ) {
738 #if SASL_VERSION_MAJOR >= 2
739                 ld->ld_error = (char *)sasl_errdetail( ctx );
740 #endif
741                 return ld->ld_errno = sasl_err2ldap( saslrc );
742         }
743
744         if( flags != LDAP_SASL_QUIET ) {
745                 saslrc = sasl_getprop( ctx, SASL_USERNAME, (SASL_CONST void **) &data );
746                 if( saslrc == SASL_OK && data && *data ) {
747                         fprintf( stderr, "SASL username: %s\n", data );
748                 }
749
750 #if SASL_VERSION_MAJOR >= 2
751                 saslrc = sasl_getprop( ctx, SASL_DEFUSERREALM, (SASL_CONST void **) &data );
752 #else
753                 saslrc = sasl_getprop( ctx, SASL_REALM, (SASL_CONST void **) &data );
754 #endif
755                 if( saslrc == SASL_OK && data && *data ) {
756                         fprintf( stderr, "SASL realm: %s\n", data );
757                 }
758         }
759
760         saslrc = sasl_getprop( ctx, SASL_SSF, (SASL_CONST void **) &ssf );
761         if( saslrc == SASL_OK ) {
762                 if( flags != LDAP_SASL_QUIET ) {
763                         fprintf( stderr, "SASL SSF: %lu\n",
764                                 (unsigned long) *ssf );
765                 }
766
767                 if( ssf && *ssf ) {
768                         if( flags != LDAP_SASL_QUIET ) {
769                                 fprintf( stderr, "SASL installing layers\n" );
770                         }
771                         ldap_pvt_sasl_install( ld->ld_conns->lconn_sb, ctx );
772                 }
773         }
774
775         return rc;
776 }
777
778 int
779 ldap_int_sasl_external(
780         LDAP *ld,
781         LDAPConn *conn,
782         const char * authid,
783         ber_len_t ssf )
784 {
785         int sc;
786         sasl_conn_t *ctx;
787 #if SASL_VERSION_MAJOR < 2
788         sasl_external_properties_t extprops;
789 #endif
790
791         ctx = conn->lconn_sasl_ctx;
792
793         if ( ctx == NULL ) {
794                 return LDAP_LOCAL_ERROR;
795         }
796    
797 #if SASL_VERSION_MAJOR >= 2
798         sc = sasl_setprop( ctx, SASL_SSF_EXTERNAL, &ssf );
799         if ( sc == SASL_OK )
800                 sc = sasl_setprop( ctx, SASL_AUTH_EXTERNAL, authid );
801 #else
802         memset( &extprops, '\0', sizeof(extprops) );
803         extprops.ssf = ssf;
804         extprops.auth_id = (char *) authid;
805
806         sc = sasl_setprop( ctx, SASL_SSF_EXTERNAL,
807                 (void *) &extprops );
808 #endif
809
810         if ( sc != SASL_OK ) {
811                 return LDAP_LOCAL_ERROR;
812         }
813
814         return LDAP_SUCCESS;
815 }
816
817
818 int ldap_pvt_sasl_secprops(
819         const char *in,
820         sasl_security_properties_t *secprops )
821 {
822         int i;
823         char **props = ldap_str2charray( in, "," );
824         unsigned sflags = 0;
825         int got_sflags = 0;
826         sasl_ssf_t max_ssf = 0;
827         int got_max_ssf = 0;
828         sasl_ssf_t min_ssf = 0;
829         int got_min_ssf = 0;
830         unsigned maxbufsize = 0;
831         int got_maxbufsize = 0;
832
833         if( props == NULL || secprops == NULL ) {
834                 return LDAP_PARAM_ERROR;
835         }
836
837         for( i=0; props[i]; i++ ) {
838                 if( !strcasecmp(props[i], "none") ) {
839                         got_sflags++;
840
841                 } else if( !strcasecmp(props[i], "noplain") ) {
842                         got_sflags++;
843                         sflags |= SASL_SEC_NOPLAINTEXT;
844
845                 } else if( !strcasecmp(props[i], "noactive") ) {
846                         got_sflags++;
847                         sflags |= SASL_SEC_NOACTIVE;
848
849                 } else if( !strcasecmp(props[i], "nodict") ) {
850                         got_sflags++;
851                         sflags |= SASL_SEC_NODICTIONARY;
852
853                 } else if( !strcasecmp(props[i], "forwardsec") ) {
854                         got_sflags++;
855                         sflags |= SASL_SEC_FORWARD_SECRECY;
856
857                 } else if( !strcasecmp(props[i], "noanonymous")) {
858                         got_sflags++;
859                         sflags |= SASL_SEC_NOANONYMOUS;
860
861                 } else if( !strcasecmp(props[i], "passcred") ) {
862                         got_sflags++;
863                         sflags |= SASL_SEC_PASS_CREDENTIALS;
864
865                 } else if( !strncasecmp(props[i],
866                         "minssf=", sizeof("minssf")) )
867                 {
868                         if( isdigit( (unsigned char) props[i][sizeof("minssf")] ) ) {
869                                 got_min_ssf++;
870                                 min_ssf = atoi( &props[i][sizeof("minssf")] );
871                         } else {
872                                 return LDAP_NOT_SUPPORTED;
873                         }
874
875                 } else if( !strncasecmp(props[i],
876                         "maxssf=", sizeof("maxssf")) )
877                 {
878                         if( isdigit( (unsigned char) props[i][sizeof("maxssf")] ) ) {
879                                 got_max_ssf++;
880                                 max_ssf = atoi( &props[i][sizeof("maxssf")] );
881                         } else {
882                                 return LDAP_NOT_SUPPORTED;
883                         }
884
885                 } else if( !strncasecmp(props[i],
886                         "maxbufsize=", sizeof("maxbufsize")) )
887                 {
888                         if( isdigit( (unsigned char) props[i][sizeof("maxbufsize")] ) ) {
889                                 got_maxbufsize++;
890                                 maxbufsize = atoi( &props[i][sizeof("maxbufsize")] );
891                         } else {
892                                 return LDAP_NOT_SUPPORTED;
893                         }
894
895                         if( maxbufsize && (( maxbufsize < SASL_MIN_BUFF_SIZE )
896                                 || (maxbufsize > SASL_MAX_BUFF_SIZE )))
897                         {
898                                 /* bad maxbufsize */
899                                 return LDAP_PARAM_ERROR;
900                         }
901
902                 } else {
903                         return LDAP_NOT_SUPPORTED;
904                 }
905         }
906
907         if(got_sflags) {
908                 secprops->security_flags = sflags;
909         }
910         if(got_min_ssf) {
911                 secprops->min_ssf = min_ssf;
912         }
913         if(got_max_ssf) {
914                 secprops->max_ssf = max_ssf;
915         }
916         if(got_maxbufsize) {
917                 secprops->maxbufsize = maxbufsize;
918         }
919
920         ldap_charray_free( props );
921         return LDAP_SUCCESS;
922 }
923
924 int
925 ldap_int_sasl_config( struct ldapoptions *lo, int option, const char *arg )
926 {
927         int rc;
928
929         switch( option ) {
930         case LDAP_OPT_X_SASL_SECPROPS:
931                 rc = ldap_pvt_sasl_secprops( arg, &lo->ldo_sasl_secprops );
932                 if( rc == LDAP_SUCCESS ) return 0;
933         }
934
935         return -1;
936 }
937
938 int
939 ldap_int_sasl_get_option( LDAP *ld, int option, void *arg )
940 {
941         if ( ld == NULL )
942                 return -1;
943
944         switch ( option ) {
945                 case LDAP_OPT_X_SASL_MECH: {
946                         *(char **)arg = ld->ld_options.ldo_def_sasl_mech
947                                 ? LDAP_STRDUP( ld->ld_options.ldo_def_sasl_mech ) : NULL;
948                 } break;
949                 case LDAP_OPT_X_SASL_REALM: {
950                         *(char **)arg = ld->ld_options.ldo_def_sasl_realm
951                                 ? LDAP_STRDUP( ld->ld_options.ldo_def_sasl_realm ) : NULL;
952                 } break;
953                 case LDAP_OPT_X_SASL_AUTHCID: {
954                         *(char **)arg = ld->ld_options.ldo_def_sasl_authcid
955                                 ? LDAP_STRDUP( ld->ld_options.ldo_def_sasl_authcid ) : NULL;
956                 } break;
957                 case LDAP_OPT_X_SASL_AUTHZID: {
958                         *(char **)arg = ld->ld_options.ldo_def_sasl_authzid
959                                 ? LDAP_STRDUP( ld->ld_options.ldo_def_sasl_authzid ) : NULL;
960                 } break;
961
962                 case LDAP_OPT_X_SASL_SSF: {
963                         int sc;
964                         sasl_ssf_t      *ssf;
965                         sasl_conn_t *ctx;
966
967                         if( ld->ld_defconn == NULL ) {
968                                 return -1;
969                         }
970
971                         ctx = ld->ld_defconn->lconn_sasl_ctx;
972
973                         if ( ctx == NULL ) {
974                                 return -1;
975                         }
976
977                         sc = sasl_getprop( ctx, SASL_SSF,
978                                 (SASL_CONST void **) &ssf );
979
980                         if ( sc != SASL_OK ) {
981                                 return -1;
982                         }
983
984                         *(ber_len_t *)arg = *ssf;
985                 } break;
986
987                 case LDAP_OPT_X_SASL_SSF_EXTERNAL:
988                         /* this option is write only */
989                         return -1;
990
991                 case LDAP_OPT_X_SASL_SSF_MIN:
992                         *(ber_len_t *)arg = ld->ld_options.ldo_sasl_secprops.min_ssf;
993                         break;
994                 case LDAP_OPT_X_SASL_SSF_MAX:
995                         *(ber_len_t *)arg = ld->ld_options.ldo_sasl_secprops.max_ssf;
996                         break;
997                 case LDAP_OPT_X_SASL_MAXBUFSIZE:
998                         *(ber_len_t *)arg = ld->ld_options.ldo_sasl_secprops.maxbufsize;
999                         break;
1000
1001                 case LDAP_OPT_X_SASL_SECPROPS:
1002                         /* this option is write only */
1003                         return -1;
1004
1005                 default:
1006                         return -1;
1007         }
1008         return 0;
1009 }
1010
1011 int
1012 ldap_int_sasl_set_option( LDAP *ld, int option, void *arg )
1013 {
1014         if ( ld == NULL )
1015                 return -1;
1016
1017         switch ( option ) {
1018         case LDAP_OPT_X_SASL_SSF:
1019                 /* This option is read-only */
1020                 return -1;
1021
1022         case LDAP_OPT_X_SASL_SSF_EXTERNAL: {
1023                 int sc;
1024 #if SASL_VERSION_MAJOR < 2
1025                 sasl_external_properties_t extprops;
1026 #endif
1027                 sasl_conn_t *ctx;
1028
1029                 if( ld->ld_defconn == NULL ) {
1030                         return -1;
1031                 }
1032
1033                 ctx = ld->ld_defconn->lconn_sasl_ctx;
1034
1035                 if ( ctx == NULL ) {
1036                         return -1;
1037                 }
1038
1039 #if SASL_VERSION_MAJOR >= 2
1040                 sc = sasl_setprop( ctx, SASL_SSF_EXTERNAL, arg);
1041 #else
1042                 memset(&extprops, 0L, sizeof(extprops));
1043
1044                 extprops.ssf = * (ber_len_t *) arg;
1045
1046                 sc = sasl_setprop( ctx, SASL_SSF_EXTERNAL,
1047                         (void *) &extprops );
1048 #endif
1049
1050                 if ( sc != SASL_OK ) {
1051                         return -1;
1052                 }
1053                 } break;
1054
1055         case LDAP_OPT_X_SASL_SSF_MIN:
1056                 ld->ld_options.ldo_sasl_secprops.min_ssf = *(ber_len_t *)arg;
1057                 break;
1058         case LDAP_OPT_X_SASL_SSF_MAX:
1059                 ld->ld_options.ldo_sasl_secprops.max_ssf = *(ber_len_t *)arg;
1060                 break;
1061         case LDAP_OPT_X_SASL_MAXBUFSIZE:
1062                 ld->ld_options.ldo_sasl_secprops.maxbufsize = *(ber_len_t *)arg;
1063                 break;
1064
1065         case LDAP_OPT_X_SASL_SECPROPS: {
1066                 int sc;
1067                 sc = ldap_pvt_sasl_secprops( (char *) arg,
1068                         &ld->ld_options.ldo_sasl_secprops );
1069
1070                 return sc == LDAP_SUCCESS ? 0 : -1;
1071                 }
1072
1073         default:
1074                 return -1;
1075         }
1076         return 0;
1077 }
1078
1079 #ifdef LDAP_R_COMPILE
1080 #define LDAP_DEBUG_R_SASL
1081 void *ldap_pvt_sasl_mutex_new(void)
1082 {
1083         ldap_pvt_thread_mutex_t *mutex;
1084
1085         mutex = (ldap_pvt_thread_mutex_t *) LDAP_MALLOC(
1086                 sizeof(ldap_pvt_thread_mutex_t) );
1087
1088         if ( ldap_pvt_thread_mutex_init( mutex ) == 0 ) {
1089                 return mutex;
1090         }
1091 #ifndef LDAP_DEBUG_R_SASL
1092         assert( 0 );
1093 #endif /* !LDAP_DEBUG_R_SASL */
1094         return NULL;
1095 }
1096
1097 int ldap_pvt_sasl_mutex_lock(void *mutex)
1098 {
1099 #ifdef LDAP_DEBUG_R_SASL
1100         if ( mutex == NULL ) {
1101                 return SASL_OK;
1102         }
1103 #else /* !LDAP_DEBUG_R_SASL */
1104         assert( mutex );
1105 #endif /* !LDAP_DEBUG_R_SASL */
1106         return ldap_pvt_thread_mutex_lock( (ldap_pvt_thread_mutex_t *)mutex )
1107                 ? SASL_FAIL : SASL_OK;
1108 }
1109
1110 int ldap_pvt_sasl_mutex_unlock(void *mutex)
1111 {
1112 #ifdef LDAP_DEBUG_R_SASL
1113         if ( mutex == NULL ) {
1114                 return SASL_OK;
1115         }
1116 #else /* !LDAP_DEBUG_R_SASL */
1117         assert( mutex );
1118 #endif /* !LDAP_DEBUG_R_SASL */
1119         return ldap_pvt_thread_mutex_unlock( (ldap_pvt_thread_mutex_t *)mutex )
1120                 ? SASL_FAIL : SASL_OK;
1121 }
1122
1123 void ldap_pvt_sasl_mutex_dispose(void *mutex)
1124 {
1125 #ifdef LDAP_DEBUG_R_SASL
1126         if ( mutex == NULL ) {
1127                 return;
1128         }
1129 #else /* !LDAP_DEBUG_R_SASL */
1130         assert( mutex );
1131 #endif /* !LDAP_DEBUG_R_SASL */
1132         (void) ldap_pvt_thread_mutex_destroy( (ldap_pvt_thread_mutex_t *)mutex );
1133         LDAP_FREE( mutex );
1134 }
1135 #endif
1136
1137 #else
1138 int ldap_int_sasl_init( void )
1139 { return LDAP_SUCCESS; }
1140
1141 int ldap_int_sasl_close( LDAP *ld, LDAPConn *lc )
1142 { return LDAP_SUCCESS; }
1143
1144 int
1145 ldap_int_sasl_bind(
1146         LDAP                    *ld,
1147         const char              *dn,
1148         const char              *mechs,
1149         LDAPControl             **sctrls,
1150         LDAPControl             **cctrls,
1151         unsigned                flags,
1152         LDAP_SASL_INTERACT_PROC *interact,
1153         void * defaults )
1154 { return LDAP_NOT_SUPPORTED; }
1155
1156 int
1157 ldap_int_sasl_external(
1158         LDAP *ld,
1159         LDAPConn *conn,
1160         const char * authid,
1161         ber_len_t ssf )
1162 { return LDAP_SUCCESS; }
1163
1164 #endif /* HAVE_CYRUS_SASL */