]> git.sur5r.net Git - openldap/blob - libraries/libldap/cyrus.c
c03d5464adb3ead007ca7ef54534b993f4dbc854
[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 SASL_VERSION_MAJOR >= 2
603                 /* XXX the application should free interact results. */
604                 if ( prompts != NULL && prompts->result != NULL ) {
605                         LDAP_FREE( (void *)prompts->result );
606                         prompts->result = NULL;
607                 }
608 #endif
609
610                 if( saslrc == SASL_INTERACT ) {
611                         int res;
612                         if( !interact ) break;
613                         res = (interact)( ld, flags, defaults, prompts );
614                         if( res != LDAP_SUCCESS ) {
615                                 break;
616                         }
617                 }
618         } while ( saslrc == SASL_INTERACT );
619
620         ccred.bv_len = credlen;
621
622         if ( (saslrc != SASL_OK) && (saslrc != SASL_CONTINUE) ) {
623                 ld->ld_errno = sasl_err2ldap( saslrc );
624 #if SASL_VERSION_MAJOR >= 2
625                 ld->ld_error = (char *)sasl_errdetail( ctx );
626 #endif
627                 return ld->ld_errno;
628         }
629
630         do {
631                 struct berval *scred;
632                 unsigned credlen;
633
634                 scred = NULL;
635
636                 rc = ldap_sasl_bind_s( ld, dn, mech, &ccred, sctrls, cctrls, &scred );
637
638                 if ( ccred.bv_val != NULL ) {
639 #if SASL_VERSION_MAJOR < 2
640                         LDAP_FREE( ccred.bv_val );
641 #endif
642                         ccred.bv_val = NULL;
643                 }
644
645                 if ( rc != LDAP_SUCCESS && rc != LDAP_SASL_BIND_IN_PROGRESS ) {
646                         if( scred && scred->bv_len ) {
647                                 /* and server provided us with data? */
648 #ifdef NEW_LOGGING
649                                 LDAP_LOG ( TRANSPORT, DETAIL1, 
650                                         "ldap_int_sasl_bind: rc=%d sasl=%d len=%ld\n", 
651                                         rc, saslrc, scred->bv_len );
652 #else
653                                 Debug( LDAP_DEBUG_TRACE,
654                                         "ldap_int_sasl_bind: rc=%d sasl=%d len=%ld\n",
655                                         rc, saslrc, scred->bv_len );
656 #endif
657                                 ber_bvfree( scred );
658                         }
659                         return ld->ld_errno;
660                 }
661
662                 if( rc == LDAP_SUCCESS && saslrc == SASL_OK ) {
663                         /* we're done, no need to step */
664                         if( scred && scred->bv_len ) {
665                                 /* but server provided us with data! */
666 #ifdef NEW_LOGGING
667                                 LDAP_LOG ( TRANSPORT, DETAIL1, 
668                                         "ldap_int_sasl_bind: rc=%d sasl=%d len=%ld\n", 
669                                         rc, saslrc, scred->bv_len );
670 #else
671                                 Debug( LDAP_DEBUG_TRACE,
672                                         "ldap_int_sasl_bind: rc=%d sasl=%d len=%ld\n",
673                                         rc, saslrc, scred->bv_len );
674 #endif
675                                 ber_bvfree( scred );
676                                 return ld->ld_errno = LDAP_LOCAL_ERROR;
677                         }
678                         break;
679                 }
680
681                 do {
682                         saslrc = sasl_client_step( ctx,
683                                 (scred == NULL) ? NULL : scred->bv_val,
684                                 (scred == NULL) ? 0 : scred->bv_len,
685                                 &prompts,
686                                 (SASL_CONST char **)&ccred.bv_val,
687                                 &credlen );
688
689 #ifdef NEW_LOGGING
690                                 LDAP_LOG ( TRANSPORT, DETAIL1, 
691                                         "ldap_int_sasl_bind: sasl_client_step: %d\n", saslrc,0,0 );
692 #else
693                         Debug( LDAP_DEBUG_TRACE, "sasl_client_step: %d\n",
694                                 saslrc, 0, 0 );
695 #endif
696
697 #if SASL_VERSION_MAJOR >= 2
698                         /* XXX the application should free interact results. */
699                         if ( prompts != NULL && prompts->result != NULL ) {
700                                 LDAP_FREE( (void *)prompts->result );
701                                 prompts->result = NULL;
702                         }
703 #endif
704
705                         if( saslrc == SASL_INTERACT ) {
706                                 int res;
707                                 if( !interact ) break;
708                                 res = (interact)( ld, flags, defaults, prompts );
709                                 if( res != LDAP_SUCCESS ) {
710                                         break;
711                                 }
712                         }
713                 } while ( saslrc == SASL_INTERACT );
714
715                 ccred.bv_len = credlen;
716                 ber_bvfree( scred );
717
718                 if ( (saslrc != SASL_OK) && (saslrc != SASL_CONTINUE) ) {
719                         ld->ld_errno = sasl_err2ldap( saslrc );
720 #if SASL_VERSION_MAJOR >= 2
721                         ld->ld_error = (char *)sasl_errdetail( ctx );
722 #endif
723                         return ld->ld_errno;
724                 }
725         } while ( rc == LDAP_SASL_BIND_IN_PROGRESS );
726
727         if ( rc != LDAP_SUCCESS ) {
728                 return rc;
729         }
730
731         if ( saslrc != SASL_OK ) {
732 #if SASL_VERSION_MAJOR >= 2
733                 ld->ld_error = (char *)sasl_errdetail( ctx );
734 #endif
735                 return ld->ld_errno = sasl_err2ldap( saslrc );
736         }
737
738         if( flags != LDAP_SASL_QUIET ) {
739                 saslrc = sasl_getprop( ctx, SASL_USERNAME, (SASL_CONST void **) &data );
740                 if( saslrc == SASL_OK && data && *data ) {
741                         fprintf( stderr, "SASL username: %s\n", data );
742                 }
743
744 #if SASL_VERSION_MAJOR >= 2
745                 saslrc = sasl_getprop( ctx, SASL_DEFUSERREALM, (SASL_CONST void **) &data );
746 #else
747                 saslrc = sasl_getprop( ctx, SASL_REALM, (SASL_CONST void **) &data );
748 #endif
749                 if( saslrc == SASL_OK && data && *data ) {
750                         fprintf( stderr, "SASL realm: %s\n", data );
751                 }
752         }
753
754         saslrc = sasl_getprop( ctx, SASL_SSF, (SASL_CONST void **) &ssf );
755         if( saslrc == SASL_OK ) {
756                 if( flags != LDAP_SASL_QUIET ) {
757                         fprintf( stderr, "SASL SSF: %lu\n",
758                                 (unsigned long) *ssf );
759                 }
760
761                 if( ssf && *ssf ) {
762                         if( flags != LDAP_SASL_QUIET ) {
763                                 fprintf( stderr, "SASL installing layers\n" );
764                         }
765                         ldap_pvt_sasl_install( ld->ld_conns->lconn_sb, ctx );
766                 }
767         }
768
769         return rc;
770 }
771
772 int
773 ldap_int_sasl_external(
774         LDAP *ld,
775         LDAPConn *conn,
776         const char * authid,
777         ber_len_t ssf )
778 {
779         int sc;
780         sasl_conn_t *ctx;
781 #if SASL_VERSION_MAJOR < 2
782         sasl_external_properties_t extprops;
783 #endif
784
785         ctx = conn->lconn_sasl_ctx;
786
787         if ( ctx == NULL ) {
788                 return LDAP_LOCAL_ERROR;
789         }
790    
791 #if SASL_VERSION_MAJOR >= 2
792         sc = sasl_setprop( ctx, SASL_SSF_EXTERNAL, &ssf );
793         if ( sc == SASL_OK )
794                 sc = sasl_setprop( ctx, SASL_AUTH_EXTERNAL, authid );
795 #else
796         memset( &extprops, '\0', sizeof(extprops) );
797         extprops.ssf = ssf;
798         extprops.auth_id = (char *) authid;
799
800         sc = sasl_setprop( ctx, SASL_SSF_EXTERNAL,
801                 (void *) &extprops );
802 #endif
803
804         if ( sc != SASL_OK ) {
805                 return LDAP_LOCAL_ERROR;
806         }
807
808         return LDAP_SUCCESS;
809 }
810
811
812 int ldap_pvt_sasl_secprops(
813         const char *in,
814         sasl_security_properties_t *secprops )
815 {
816         int i;
817         char **props = ldap_str2charray( in, "," );
818         unsigned sflags = 0;
819         int got_sflags = 0;
820         sasl_ssf_t max_ssf = 0;
821         int got_max_ssf = 0;
822         sasl_ssf_t min_ssf = 0;
823         int got_min_ssf = 0;
824         unsigned maxbufsize = 0;
825         int got_maxbufsize = 0;
826
827         if( props == NULL || secprops == NULL ) {
828                 return LDAP_PARAM_ERROR;
829         }
830
831         for( i=0; props[i]; i++ ) {
832                 if( !strcasecmp(props[i], "none") ) {
833                         got_sflags++;
834
835                 } else if( !strcasecmp(props[i], "noplain") ) {
836                         got_sflags++;
837                         sflags |= SASL_SEC_NOPLAINTEXT;
838
839                 } else if( !strcasecmp(props[i], "noactive") ) {
840                         got_sflags++;
841                         sflags |= SASL_SEC_NOACTIVE;
842
843                 } else if( !strcasecmp(props[i], "nodict") ) {
844                         got_sflags++;
845                         sflags |= SASL_SEC_NODICTIONARY;
846
847                 } else if( !strcasecmp(props[i], "forwardsec") ) {
848                         got_sflags++;
849                         sflags |= SASL_SEC_FORWARD_SECRECY;
850
851                 } else if( !strcasecmp(props[i], "noanonymous")) {
852                         got_sflags++;
853                         sflags |= SASL_SEC_NOANONYMOUS;
854
855                 } else if( !strcasecmp(props[i], "passcred") ) {
856                         got_sflags++;
857                         sflags |= SASL_SEC_PASS_CREDENTIALS;
858
859                 } else if( !strncasecmp(props[i],
860                         "minssf=", sizeof("minssf")) )
861                 {
862                         if( isdigit( (unsigned char) props[i][sizeof("minssf")] ) ) {
863                                 got_min_ssf++;
864                                 min_ssf = atoi( &props[i][sizeof("minssf")] );
865                         } else {
866                                 return LDAP_NOT_SUPPORTED;
867                         }
868
869                 } else if( !strncasecmp(props[i],
870                         "maxssf=", sizeof("maxssf")) )
871                 {
872                         if( isdigit( (unsigned char) props[i][sizeof("maxssf")] ) ) {
873                                 got_max_ssf++;
874                                 max_ssf = atoi( &props[i][sizeof("maxssf")] );
875                         } else {
876                                 return LDAP_NOT_SUPPORTED;
877                         }
878
879                 } else if( !strncasecmp(props[i],
880                         "maxbufsize=", sizeof("maxbufsize")) )
881                 {
882                         if( isdigit( (unsigned char) props[i][sizeof("maxbufsize")] ) ) {
883                                 got_maxbufsize++;
884                                 maxbufsize = atoi( &props[i][sizeof("maxbufsize")] );
885                         } else {
886                                 return LDAP_NOT_SUPPORTED;
887                         }
888
889                         if( maxbufsize && (( maxbufsize < SASL_MIN_BUFF_SIZE )
890                                 || (maxbufsize > SASL_MAX_BUFF_SIZE )))
891                         {
892                                 /* bad maxbufsize */
893                                 return LDAP_PARAM_ERROR;
894                         }
895
896                 } else {
897                         return LDAP_NOT_SUPPORTED;
898                 }
899         }
900
901         if(got_sflags) {
902                 secprops->security_flags = sflags;
903         }
904         if(got_min_ssf) {
905                 secprops->min_ssf = min_ssf;
906         }
907         if(got_max_ssf) {
908                 secprops->max_ssf = max_ssf;
909         }
910         if(got_maxbufsize) {
911                 secprops->maxbufsize = maxbufsize;
912         }
913
914         ldap_charray_free( props );
915         return LDAP_SUCCESS;
916 }
917
918 int
919 ldap_int_sasl_config( struct ldapoptions *lo, int option, const char *arg )
920 {
921         int rc;
922
923         switch( option ) {
924         case LDAP_OPT_X_SASL_SECPROPS:
925                 rc = ldap_pvt_sasl_secprops( arg, &lo->ldo_sasl_secprops );
926                 if( rc == LDAP_SUCCESS ) return 0;
927         }
928
929         return -1;
930 }
931
932 int
933 ldap_int_sasl_get_option( LDAP *ld, int option, void *arg )
934 {
935         if ( ld == NULL )
936                 return -1;
937
938         switch ( option ) {
939                 case LDAP_OPT_X_SASL_MECH: {
940                         *(char **)arg = ld->ld_options.ldo_def_sasl_mech
941                                 ? LDAP_STRDUP( ld->ld_options.ldo_def_sasl_mech ) : NULL;
942                 } break;
943                 case LDAP_OPT_X_SASL_REALM: {
944                         *(char **)arg = ld->ld_options.ldo_def_sasl_realm
945                                 ? LDAP_STRDUP( ld->ld_options.ldo_def_sasl_realm ) : NULL;
946                 } break;
947                 case LDAP_OPT_X_SASL_AUTHCID: {
948                         *(char **)arg = ld->ld_options.ldo_def_sasl_authcid
949                                 ? LDAP_STRDUP( ld->ld_options.ldo_def_sasl_authcid ) : NULL;
950                 } break;
951                 case LDAP_OPT_X_SASL_AUTHZID: {
952                         *(char **)arg = ld->ld_options.ldo_def_sasl_authzid
953                                 ? LDAP_STRDUP( ld->ld_options.ldo_def_sasl_authzid ) : NULL;
954                 } break;
955
956                 case LDAP_OPT_X_SASL_SSF: {
957                         int sc;
958                         sasl_ssf_t      *ssf;
959                         sasl_conn_t *ctx;
960
961                         if( ld->ld_defconn == NULL ) {
962                                 return -1;
963                         }
964
965                         ctx = ld->ld_defconn->lconn_sasl_ctx;
966
967                         if ( ctx == NULL ) {
968                                 return -1;
969                         }
970
971                         sc = sasl_getprop( ctx, SASL_SSF,
972                                 (SASL_CONST void **) &ssf );
973
974                         if ( sc != SASL_OK ) {
975                                 return -1;
976                         }
977
978                         *(ber_len_t *)arg = *ssf;
979                 } break;
980
981                 case LDAP_OPT_X_SASL_SSF_EXTERNAL:
982                         /* this option is write only */
983                         return -1;
984
985                 case LDAP_OPT_X_SASL_SSF_MIN:
986                         *(ber_len_t *)arg = ld->ld_options.ldo_sasl_secprops.min_ssf;
987                         break;
988                 case LDAP_OPT_X_SASL_SSF_MAX:
989                         *(ber_len_t *)arg = ld->ld_options.ldo_sasl_secprops.max_ssf;
990                         break;
991                 case LDAP_OPT_X_SASL_MAXBUFSIZE:
992                         *(ber_len_t *)arg = ld->ld_options.ldo_sasl_secprops.maxbufsize;
993                         break;
994
995                 case LDAP_OPT_X_SASL_SECPROPS:
996                         /* this option is write only */
997                         return -1;
998
999                 default:
1000                         return -1;
1001         }
1002         return 0;
1003 }
1004
1005 int
1006 ldap_int_sasl_set_option( LDAP *ld, int option, void *arg )
1007 {
1008         if ( ld == NULL )
1009                 return -1;
1010
1011         switch ( option ) {
1012         case LDAP_OPT_X_SASL_SSF:
1013                 /* This option is read-only */
1014                 return -1;
1015
1016         case LDAP_OPT_X_SASL_SSF_EXTERNAL: {
1017                 int sc;
1018 #if SASL_VERSION_MAJOR < 2
1019                 sasl_external_properties_t extprops;
1020 #endif
1021                 sasl_conn_t *ctx;
1022
1023                 if( ld->ld_defconn == NULL ) {
1024                         return -1;
1025                 }
1026
1027                 ctx = ld->ld_defconn->lconn_sasl_ctx;
1028
1029                 if ( ctx == NULL ) {
1030                         return -1;
1031                 }
1032
1033 #if SASL_VERSION_MAJOR >= 2
1034                 sc = sasl_setprop( ctx, SASL_SSF_EXTERNAL, arg);
1035 #else
1036                 memset(&extprops, 0L, sizeof(extprops));
1037
1038                 extprops.ssf = * (ber_len_t *) arg;
1039
1040                 sc = sasl_setprop( ctx, SASL_SSF_EXTERNAL,
1041                         (void *) &extprops );
1042 #endif
1043
1044                 if ( sc != SASL_OK ) {
1045                         return -1;
1046                 }
1047                 } break;
1048
1049         case LDAP_OPT_X_SASL_SSF_MIN:
1050                 ld->ld_options.ldo_sasl_secprops.min_ssf = *(ber_len_t *)arg;
1051                 break;
1052         case LDAP_OPT_X_SASL_SSF_MAX:
1053                 ld->ld_options.ldo_sasl_secprops.max_ssf = *(ber_len_t *)arg;
1054                 break;
1055         case LDAP_OPT_X_SASL_MAXBUFSIZE:
1056                 ld->ld_options.ldo_sasl_secprops.maxbufsize = *(ber_len_t *)arg;
1057                 break;
1058
1059         case LDAP_OPT_X_SASL_SECPROPS: {
1060                 int sc;
1061                 sc = ldap_pvt_sasl_secprops( (char *) arg,
1062                         &ld->ld_options.ldo_sasl_secprops );
1063
1064                 return sc == LDAP_SUCCESS ? 0 : -1;
1065                 }
1066
1067         default:
1068                 return -1;
1069         }
1070         return 0;
1071 }
1072
1073 #ifdef LDAP_R_COMPILE
1074 #define LDAP_DEBUG_R_SASL
1075 void *ldap_pvt_sasl_mutex_new(void)
1076 {
1077         ldap_pvt_thread_mutex_t *mutex;
1078
1079         mutex = (ldap_pvt_thread_mutex_t *) LDAP_MALLOC(
1080                 sizeof(ldap_pvt_thread_mutex_t) );
1081
1082         if ( ldap_pvt_thread_mutex_init( mutex ) == 0 ) {
1083                 return mutex;
1084         }
1085 #ifndef LDAP_DEBUG_R_SASL
1086         assert( 0 );
1087 #endif /* !LDAP_DEBUG_R_SASL */
1088         return NULL;
1089 }
1090
1091 int ldap_pvt_sasl_mutex_lock(void *mutex)
1092 {
1093 #ifdef LDAP_DEBUG_R_SASL
1094         if ( mutex == NULL ) {
1095                 return SASL_OK;
1096         }
1097 #else /* !LDAP_DEBUG_R_SASL */
1098         assert( mutex );
1099 #endif /* !LDAP_DEBUG_R_SASL */
1100         return ldap_pvt_thread_mutex_lock( (ldap_pvt_thread_mutex_t *)mutex )
1101                 ? SASL_FAIL : SASL_OK;
1102 }
1103
1104 int ldap_pvt_sasl_mutex_unlock(void *mutex)
1105 {
1106 #ifdef LDAP_DEBUG_R_SASL
1107         if ( mutex == NULL ) {
1108                 return SASL_OK;
1109         }
1110 #else /* !LDAP_DEBUG_R_SASL */
1111         assert( mutex );
1112 #endif /* !LDAP_DEBUG_R_SASL */
1113         return ldap_pvt_thread_mutex_unlock( (ldap_pvt_thread_mutex_t *)mutex )
1114                 ? SASL_FAIL : SASL_OK;
1115 }
1116
1117 void ldap_pvt_sasl_mutex_dispose(void *mutex)
1118 {
1119 #ifdef LDAP_DEBUG_R_SASL
1120         if ( mutex == NULL ) {
1121                 return;
1122         }
1123 #else /* !LDAP_DEBUG_R_SASL */
1124         assert( mutex );
1125 #endif /* !LDAP_DEBUG_R_SASL */
1126         (void) ldap_pvt_thread_mutex_destroy( (ldap_pvt_thread_mutex_t *)mutex );
1127         LDAP_FREE( mutex );
1128 }
1129 #endif
1130
1131 #else
1132 int ldap_int_sasl_init( void )
1133 { return LDAP_SUCCESS; }
1134
1135 int ldap_int_sasl_close( LDAP *ld, LDAPConn *lc )
1136 { return LDAP_SUCCESS; }
1137
1138 int
1139 ldap_int_sasl_bind(
1140         LDAP                    *ld,
1141         const char              *dn,
1142         const char              *mechs,
1143         LDAPControl             **sctrls,
1144         LDAPControl             **cctrls,
1145         unsigned                flags,
1146         LDAP_SASL_INTERACT_PROC *interact,
1147         void * defaults )
1148 { return LDAP_NOT_SUPPORTED; }
1149
1150 int
1151 ldap_int_sasl_external(
1152         LDAP *ld,
1153         LDAPConn *conn,
1154         const char * authid,
1155         ber_len_t ssf )
1156 { return LDAP_SUCCESS; }
1157
1158 #endif /* HAVE_CYRUS_SASL */