3 * Copyright 1998-2002 The OpenLDAP Foundation, All Rights Reserved.
4 * COPYING RESTRICTIONS APPLY, see COPYRIGHT file
12 #include <ac/socket.h>
14 #include <ac/signal.h>
15 #include <ac/string.h>
17 #include <ac/unistd.h>
23 /* protected by connections_mutex */
24 static ldap_pvt_thread_mutex_t connections_mutex;
25 static Connection *connections = NULL;
26 static unsigned long conn_nextid = 0;
28 /* structure state (protected by connections_mutex) */
29 #define SLAP_C_UNINITIALIZED 0x00 /* MUST BE ZERO (0) */
30 #define SLAP_C_UNUSED 0x01
31 #define SLAP_C_USED 0x02
33 /* connection state (protected by c_mutex ) */
34 #define SLAP_C_INVALID 0x00 /* MUST BE ZERO (0) */
35 #define SLAP_C_INACTIVE 0x01 /* zero threads */
36 #define SLAP_C_ACTIVE 0x02 /* one or more threads */
37 #define SLAP_C_BINDING 0x03 /* binding */
38 #define SLAP_C_CLOSING 0x04 /* closing */
41 connection_state2str( int state )
44 case SLAP_C_INVALID: return "!";
45 case SLAP_C_INACTIVE: return "|";
46 case SLAP_C_ACTIVE: return "";
47 case SLAP_C_BINDING: return "B";
48 case SLAP_C_CLOSING: return "C";
54 static Connection* connection_get( ber_socket_t s );
56 static int connection_input( Connection *c );
57 static void connection_close( Connection *c );
59 static int connection_op_activate( Connection *conn, Operation *op );
60 static int connection_resched( Connection *conn );
61 static void connection_abandon( Connection *conn );
62 static void connection_destroy( Connection *c );
70 * Initialize connection management infrastructure.
72 int connections_init(void)
74 assert( connections == NULL );
76 if( connections != NULL) {
78 LDAP_LOG(( "connection", LDAP_LEVEL_INFO,
79 "connections_init: already initialized.\n" ));
81 Debug( LDAP_DEBUG_ANY, "connections_init: already initialized.\n",
87 /* should check return of every call */
88 ldap_pvt_thread_mutex_init( &connections_mutex );
90 connections = (Connection *) calloc( dtblsize, sizeof(Connection) );
92 if( connections == NULL ) {
94 LDAP_LOG(( "connection", LDAP_LEVEL_ERR,
95 "connections_init: allocation (%d * %ld) of connection array failed\n",
96 dtblsize, (long) sizeof(Connection) ));
98 Debug( LDAP_DEBUG_ANY,
99 "connections_init: allocation (%d*%ld) of connection array failed\n",
100 dtblsize, (long) sizeof(Connection), 0 );
105 assert( connections[0].c_struct_state == SLAP_C_UNINITIALIZED );
106 assert( connections[dtblsize-1].c_struct_state == SLAP_C_UNINITIALIZED );
109 * per entry initialization of the Connection array initialization
110 * will be done by connection_init()
117 * Destroy connection management infrastructure.
119 int connections_destroy(void)
123 /* should check return of every call */
125 if( connections == NULL) {
127 LDAP_LOG(( "connection", LDAP_LEVEL_INFO,
128 "connections_destroy: nothing to destroy.\n"));
130 Debug( LDAP_DEBUG_ANY, "connections_destroy: nothing to destroy.\n",
136 for ( i = 0; i < dtblsize; i++ ) {
137 if( connections[i].c_struct_state != SLAP_C_UNINITIALIZED ) {
138 ber_sockbuf_free( connections[i].c_sb );
139 ldap_pvt_thread_mutex_destroy( &connections[i].c_mutex );
140 ldap_pvt_thread_mutex_destroy( &connections[i].c_write_mutex );
141 ldap_pvt_thread_cond_destroy( &connections[i].c_write_cv );
148 ldap_pvt_thread_mutex_destroy( &connections_mutex );
153 * shutdown all connections
155 int connections_shutdown(void)
159 ldap_pvt_thread_mutex_lock( &connections_mutex );
161 for ( i = 0; i < dtblsize; i++ ) {
162 if( connections[i].c_struct_state != SLAP_C_USED ) {
166 ldap_pvt_thread_mutex_lock( &connections[i].c_mutex );
168 /* connections_mutex and c_mutex are locked */
169 connection_closing( &connections[i] );
170 connection_close( &connections[i] );
172 ldap_pvt_thread_mutex_unlock( &connections[i].c_mutex );
175 ldap_pvt_thread_mutex_unlock( &connections_mutex );
181 * Timeout idle connections.
183 int connections_timeout_idle(time_t now)
189 for( c = connection_first( &connindex );
191 c = connection_next( c, &connindex ) )
193 if( difftime( c->c_activitytime+global_idletimeout, now) < 0 ) {
195 connection_closing( c );
196 connection_close( c );
200 connection_done( c );
205 static Connection* connection_get( ber_socket_t s )
207 /* connections_mutex should be locked by caller */
212 LDAP_LOG(( "connection", LDAP_LEVEL_ENTRY,
213 "connection_get: socket %ld\n", (long)s ));
215 Debug( LDAP_DEBUG_ARGS,
216 "connection_get(%ld)\n",
220 assert( connections != NULL );
222 if(s == AC_SOCKET_INVALID) {
229 assert( c->c_struct_state != SLAP_C_UNINITIALIZED );
236 for(i=0; i<dtblsize; i++) {
237 ber_sockbuf_ctrl( connections[i].c_sb,
238 LBER_SB_OPT_GET_FD, &sd );
240 if( connections[i].c_struct_state == SLAP_C_UNINITIALIZED ) {
241 assert( connections[i].c_conn_state == SLAP_C_INVALID );
242 assert( connections[i].c_sb == 0 );
246 if( connections[i].c_struct_state == SLAP_C_UNUSED ) {
247 assert( connections[i].c_conn_state == SLAP_C_INVALID );
248 assert( sd == AC_SOCKET_INVALID );
252 /* state can actually change from used -> unused by resched,
253 * so don't assert details here.
267 ldap_pvt_thread_mutex_lock( &c->c_mutex );
269 ber_sockbuf_ctrl( c->c_sb, LBER_SB_OPT_GET_FD, &sd );
270 if( c->c_struct_state != SLAP_C_USED ) {
271 /* connection must have been closed due to resched */
273 assert( c->c_conn_state == SLAP_C_INVALID );
274 assert( sd == AC_SOCKET_INVALID );
277 LDAP_LOG(( "connection", LDAP_LEVEL_ARGS,
278 "connection_get: connection %d not used\n", s ));
280 Debug( LDAP_DEBUG_TRACE,
281 "connection_get(%d): connection not used\n",
285 ldap_pvt_thread_mutex_unlock( &c->c_mutex );
290 LDAP_LOG(( "connection", LDAP_LEVEL_RESULTS,
291 "connection_get: get for %d got connid %ld\n",s, c->c_connid ));
293 Debug( LDAP_DEBUG_TRACE,
294 "connection_get(%d): got connid=%ld\n",
300 assert( c->c_struct_state == SLAP_C_USED );
301 assert( c->c_conn_state != SLAP_C_INVALID );
302 assert( sd != AC_SOCKET_INVALID );
305 c->c_activitytime = slap_get_time();
307 if( global_idletimeout > 0 ) {
308 c->c_activitytime = slap_get_time();
316 static void connection_return( Connection *c )
318 ldap_pvt_thread_mutex_unlock( &c->c_mutex );
321 long connection_init(
325 const char* peername,
326 const char* sockname,
334 assert( connections != NULL );
336 assert( dnsname != NULL );
337 assert( peername != NULL );
338 assert( sockname != NULL );
341 assert( tls_udp_option != 1 );
344 if( s == AC_SOCKET_INVALID ) {
346 LDAP_LOG(( "connection", LDAP_LEVEL_INFO,
347 "connection_init: init of socket %ld invalid.\n", (long)s ));
349 Debug( LDAP_DEBUG_ANY,
350 "connection_init(%ld): invalid.\n",
358 assert( s < dtblsize );
361 ldap_pvt_thread_mutex_lock( &connections_mutex );
372 for( i=0; i < dtblsize; i++) {
375 if( connections[i].c_struct_state == SLAP_C_UNINITIALIZED ) {
376 assert( connections[i].c_sb == 0 );
381 sd = AC_SOCKET_INVALID;
382 if (connections[i].c_sb != NULL)
383 ber_sockbuf_ctrl( connections[i].c_sb, LBER_SB_OPT_GET_FD, &sd );
385 if( connections[i].c_struct_state == SLAP_C_UNUSED ) {
386 assert( sd == AC_SOCKET_INVALID );
391 assert( connections[i].c_struct_state == SLAP_C_USED );
392 assert( connections[i].c_conn_state != SLAP_C_INVALID );
393 assert( sd != AC_SOCKET_INVALID );
398 LDAP_LOG(( "connection", LDAP_LEVEL_INFO,
399 "connection_init: skt %d connection table full (%d/%d)\n",
402 Debug( LDAP_DEBUG_ANY,
403 "connection_init(%d): connection table full (%d/%d)\n",
406 ldap_pvt_thread_mutex_unlock( &connections_mutex );
414 if( c->c_struct_state == SLAP_C_UNINITIALIZED ) {
415 c->c_authmech.bv_val = NULL;
416 c->c_authmech.bv_len = 0;
417 c->c_dn.bv_val = NULL;
419 c->c_ndn.bv_val = NULL;
421 c->c_cdn.bv_val = NULL;
425 c->c_listener_url.bv_val = NULL;
426 c->c_listener_url.bv_len = 0;
427 c->c_peer_domain.bv_val = NULL;
428 c->c_peer_domain.bv_len = 0;
429 c->c_peer_name.bv_val = NULL;
430 c->c_peer_name.bv_len = 0;
431 c->c_sock_name.bv_val = NULL;
432 c->c_sock_name.bv_len = 0;
434 LDAP_STAILQ_INIT(&c->c_ops);
435 LDAP_STAILQ_INIT(&c->c_pending_ops);
437 c->c_sasl_bind_mech.bv_val = NULL;
438 c->c_sasl_bind_mech.bv_len = 0;
439 c->c_sasl_context = NULL;
440 c->c_sasl_extra = NULL;
442 c->c_sb = ber_sockbuf_alloc( );
445 ber_len_t max = sockbuf_max_incoming;
446 ber_sockbuf_ctrl( c->c_sb, LBER_SB_OPT_SET_MAX_INCOMING, &max );
449 c->c_currentber = NULL;
451 /* should check status of thread calls */
452 ldap_pvt_thread_mutex_init( &c->c_mutex );
453 ldap_pvt_thread_mutex_init( &c->c_write_mutex );
454 ldap_pvt_thread_cond_init( &c->c_write_cv );
456 c->c_struct_state = SLAP_C_UNUSED;
459 ldap_pvt_thread_mutex_lock( &c->c_mutex );
461 assert( c->c_struct_state == SLAP_C_UNUSED );
462 assert( c->c_authmech.bv_val == NULL );
463 assert( c->c_dn.bv_val == NULL );
464 assert( c->c_ndn.bv_val == NULL );
465 assert( c->c_cdn.bv_val == NULL );
466 assert( c->c_groups == NULL );
467 assert( c->c_listener_url.bv_val == NULL );
468 assert( c->c_peer_domain.bv_val == NULL );
469 assert( c->c_peer_name.bv_val == NULL );
470 assert( c->c_sock_name.bv_val == NULL );
471 assert( LDAP_STAILQ_EMPTY(&c->c_ops) );
472 assert( LDAP_STAILQ_EMPTY(&c->c_pending_ops) );
473 assert( c->c_sasl_bind_mech.bv_val == NULL );
474 assert( c->c_sasl_context == NULL );
475 assert( c->c_sasl_extra == NULL );
476 assert( c->c_currentber == NULL );
478 ber_str2bv( url, 0, 1, &c->c_listener_url );
479 ber_str2bv( dnsname, 0, 1, &c->c_peer_domain );
480 ber_str2bv( peername, 0, 1, &c->c_peer_name );
481 ber_str2bv( sockname, 0, 1, &c->c_sock_name );
483 c->c_n_ops_received = 0;
484 c->c_n_ops_executing = 0;
485 c->c_n_ops_pending = 0;
486 c->c_n_ops_completed = 0;
492 /* set to zero until bind, implies LDAP_VERSION3 */
496 c->c_activitytime = c->c_starttime = slap_get_time();
498 if( global_idletimeout > 0 ) {
499 c->c_activitytime = c->c_starttime = slap_get_time();
503 #ifdef LDAP_CONNECTIONLESS
505 if (tls_udp_option == 2)
509 ber_sockbuf_add_io( c->c_sb, &ber_sockbuf_io_debug,
510 LBER_SBIOD_LEVEL_PROVIDER, (void*)"udp_" );
512 ber_sockbuf_add_io( c->c_sb, &ber_sockbuf_io_udp,
513 LBER_SBIOD_LEVEL_PROVIDER, (void *)&s );
514 ber_sockbuf_add_io( c->c_sb, &ber_sockbuf_io_readahead,
515 LBER_SBIOD_LEVEL_PROVIDER, NULL );
520 ber_sockbuf_add_io( c->c_sb, &ber_sockbuf_io_debug,
521 LBER_SBIOD_LEVEL_PROVIDER, (void*)"tcp_" );
523 ber_sockbuf_add_io( c->c_sb, &ber_sockbuf_io_tcp,
524 LBER_SBIOD_LEVEL_PROVIDER, (void *)&s );
528 ber_sockbuf_add_io( c->c_sb, &ber_sockbuf_io_debug,
529 INT_MAX, (void*)"ldap_" );
532 if( ber_sockbuf_ctrl( c->c_sb, LBER_SB_OPT_SET_NONBLOCK,
533 c /* non-NULL */ ) < 0 )
536 LDAP_LOG(( "connection", LDAP_LEVEL_INFO,
537 "connection_init: conn %d set nonblocking failed\n",
540 Debug( LDAP_DEBUG_ANY,
541 "connection_init(%d, %s): set nonblocking failed\n",
542 s, c->c_peer_name, 0 );
546 id = c->c_connid = conn_nextid++;
548 c->c_conn_state = SLAP_C_INACTIVE;
549 c->c_struct_state = SLAP_C_USED;
551 c->c_ssf = c->c_transport_ssf = ssf;
555 if ( tls_udp_option == 1 ) {
557 c->c_needs_tls_accept = 1;
560 c->c_needs_tls_accept = 0;
565 slap_sasl_external( c, ssf, authid );
567 ldap_pvt_thread_mutex_unlock( &c->c_mutex );
568 ldap_pvt_thread_mutex_unlock( &connections_mutex );
570 backend_connection_init(c);
575 void connection2anonymous( Connection *c )
577 assert( connections != NULL );
581 ber_len_t max = sockbuf_max_incoming;
582 ber_sockbuf_ctrl( c->c_sb, LBER_SB_OPT_SET_MAX_INCOMING, &max );
585 if(c->c_authmech.bv_val != NULL ) {
586 free(c->c_authmech.bv_val);
587 c->c_authmech.bv_val = NULL;
589 c->c_authmech.bv_len = 0;
591 if(c->c_dn.bv_val != NULL) {
592 free(c->c_dn.bv_val);
593 c->c_dn.bv_val = NULL;
596 if(c->c_ndn.bv_val != NULL) {
597 free(c->c_ndn.bv_val);
598 c->c_ndn.bv_val = NULL;
602 if(c->c_cdn.bv_val != NULL) {
603 free(c->c_cdn.bv_val);
604 c->c_cdn.bv_val = NULL;
608 c->c_authc_backend = NULL;
609 c->c_authz_backend = NULL;
612 GroupAssertion *g, *n;
613 for (g = c->c_groups; g; g=n)
624 connection_destroy( Connection *c )
626 /* note: connections_mutex should be locked by caller */
629 assert( connections != NULL );
631 assert( c->c_struct_state != SLAP_C_UNUSED );
632 assert( c->c_conn_state != SLAP_C_INVALID );
633 assert( LDAP_STAILQ_EMPTY(&c->c_ops) );
635 backend_connection_destroy(c);
640 c->c_activitytime = c->c_starttime = 0;
642 connection2anonymous( c );
644 if(c->c_listener_url.bv_val != NULL) {
645 free(c->c_listener_url.bv_val);
646 c->c_listener_url.bv_val = NULL;
648 c->c_listener_url.bv_len = 0;
650 if(c->c_peer_domain.bv_val != NULL) {
651 free(c->c_peer_domain.bv_val);
652 c->c_peer_domain.bv_val = NULL;
654 c->c_peer_domain.bv_len = 0;
655 if(c->c_peer_name.bv_val != NULL) {
658 * If peer was a domain socket, unlink. Mind you,
659 * they may be un-named. Should we leave this to
662 if (strncmp(c->c_peer_name.bv_val, "PATH=", 5) == 0) {
663 char *path = c->c_peer_name.bv_val + 5;
668 #endif /* LDAP_PF_LOCAL */
670 free(c->c_peer_name.bv_val);
671 c->c_peer_name.bv_val = NULL;
673 c->c_peer_name.bv_len = 0;
674 if(c->c_sock_name.bv_val != NULL) {
675 free(c->c_sock_name.bv_val);
676 c->c_sock_name.bv_val = NULL;
678 c->c_sock_name.bv_len = 0;
680 c->c_sasl_bind_in_progress = 0;
681 if(c->c_sasl_bind_mech.bv_val != NULL) {
682 free(c->c_sasl_bind_mech.bv_val);
683 c->c_sasl_bind_mech.bv_val = NULL;
685 c->c_sasl_bind_mech.bv_len = 0;
687 slap_sasl_close( c );
689 if ( c->c_currentber != NULL ) {
690 ber_free( c->c_currentber, 1 );
691 c->c_currentber = NULL;
694 ber_sockbuf_ctrl( c->c_sb, LBER_SB_OPT_GET_FD, &sd );
695 if ( sd != AC_SOCKET_INVALID ) {
696 slapd_remove( sd, 0 );
698 Statslog( LDAP_DEBUG_STATS,
699 "conn=%ld fd=%d closed\n",
700 c->c_connid, sd, 0, 0, 0 );
703 ber_sockbuf_free( c->c_sb );
705 c->c_sb = ber_sockbuf_alloc( );
708 ber_len_t max = sockbuf_max_incoming;
709 ber_sockbuf_ctrl( c->c_sb, LBER_SB_OPT_SET_MAX_INCOMING, &max );
712 c->c_conn_state = SLAP_C_INVALID;
713 c->c_struct_state = SLAP_C_UNUSED;
716 int connection_state_closing( Connection *c )
718 /* c_mutex must be locked by caller */
722 assert( c->c_struct_state == SLAP_C_USED );
724 state = c->c_conn_state;
726 assert( state != SLAP_C_INVALID );
728 return state == SLAP_C_CLOSING;
731 static void connection_abandon( Connection *c )
733 /* c_mutex must be locked by caller */
737 LDAP_STAILQ_FOREACH(o, &c->c_ops, o_next) {
738 ldap_pvt_thread_mutex_lock( &o->o_abandonmutex );
740 ldap_pvt_thread_mutex_unlock( &o->o_abandonmutex );
743 /* remove pending operations */
744 while ( (o = LDAP_STAILQ_FIRST( &c->c_pending_ops )) != NULL) {
745 LDAP_STAILQ_REMOVE_HEAD( &c->c_pending_ops, o_next );
746 LDAP_STAILQ_NEXT(o, o_next) = NULL;
751 void connection_closing( Connection *c )
753 assert( connections != NULL );
755 assert( c->c_struct_state == SLAP_C_USED );
756 assert( c->c_conn_state != SLAP_C_INVALID );
758 /* c_mutex must be locked by caller */
760 if( c->c_conn_state != SLAP_C_CLOSING ) {
763 ber_sockbuf_ctrl( c->c_sb, LBER_SB_OPT_GET_FD, &sd );
765 LDAP_LOG(( "connection", LDAP_LEVEL_DETAIL1,
766 "connection_closing: conn %d readying socket %d for close.\n",
769 Debug( LDAP_DEBUG_TRACE,
770 "connection_closing: readying conn=%ld sd=%d for close\n",
771 c->c_connid, sd, 0 );
773 /* update state to closing */
774 c->c_conn_state = SLAP_C_CLOSING;
776 /* don't listen on this port anymore */
777 slapd_clr_read( sd, 1 );
779 /* abandon active operations */
780 connection_abandon( c );
782 /* wake write blocked operations */
783 slapd_clr_write( sd, 1 );
784 ldap_pvt_thread_cond_signal( &c->c_write_cv );
788 static void connection_close( Connection *c )
792 assert( connections != NULL );
794 assert( c->c_struct_state == SLAP_C_USED );
795 assert( c->c_conn_state == SLAP_C_CLOSING );
797 /* note: connections_mutex and c_mutex should be locked by caller */
799 ber_sockbuf_ctrl( c->c_sb, LBER_SB_OPT_GET_FD, &sd );
800 if( !LDAP_STAILQ_EMPTY(&c->c_ops) ) {
802 LDAP_LOG(( "connection", LDAP_LEVEL_DETAIL1,
803 "connection_close: conn %d deferring sd %d\n",
806 Debug( LDAP_DEBUG_TRACE,
807 "connection_close: deferring conn=%ld sd=%d\n",
808 c->c_connid, sd, 0 );
814 LDAP_LOG(( "connection", LDAP_LEVEL_RESULTS,
815 "connection_close: conn %d sd %d\n",
818 Debug( LDAP_DEBUG_TRACE, "connection_close: conn=%ld sd=%d\n",
819 c->c_connid, sd, 0 );
821 connection_destroy( c );
824 unsigned long connections_nextid(void)
827 assert( connections != NULL );
829 ldap_pvt_thread_mutex_lock( &connections_mutex );
833 ldap_pvt_thread_mutex_unlock( &connections_mutex );
838 Connection* connection_first( ber_socket_t *index )
840 assert( connections != NULL );
841 assert( index != NULL );
843 ldap_pvt_thread_mutex_lock( &connections_mutex );
847 return connection_next(NULL, index);
850 Connection* connection_next( Connection *c, ber_socket_t *index )
852 assert( connections != NULL );
853 assert( index != NULL );
854 assert( *index <= dtblsize );
857 ldap_pvt_thread_mutex_unlock( &c->c_mutex );
862 for(; *index < dtblsize; (*index)++) {
863 if( connections[*index].c_struct_state == SLAP_C_UNINITIALIZED ) {
864 assert( connections[*index].c_conn_state == SLAP_C_INVALID );
872 if( connections[*index].c_struct_state == SLAP_C_USED ) {
873 assert( connections[*index].c_conn_state != SLAP_C_INVALID );
874 c = &connections[(*index)++];
878 assert( connections[*index].c_struct_state == SLAP_C_UNUSED );
879 assert( connections[*index].c_conn_state == SLAP_C_INVALID );
883 ldap_pvt_thread_mutex_lock( &c->c_mutex );
889 void connection_done( Connection *c )
891 assert( connections != NULL );
894 ldap_pvt_thread_mutex_unlock( &c->c_mutex );
897 ldap_pvt_thread_mutex_unlock( &connections_mutex );
901 * connection_activity - handle the request operation op on connection
902 * conn. This routine figures out what kind of operation it is and
903 * calls the appropriate stub to handle it.
907 connection_operation( void *arg_v )
910 struct co_arg *arg = arg_v;
911 ber_tag_t tag = arg->co_op->o_tag;
912 Connection *conn = arg->co_conn;
914 ldap_pvt_thread_mutex_lock( &num_ops_mutex );
916 ldap_pvt_thread_mutex_unlock( &num_ops_mutex );
918 if( conn->c_sasl_bind_in_progress && tag != LDAP_REQ_BIND ) {
920 LDAP_LOG(( "connection", LDAP_LEVEL_ERR,
921 "connection_operation: conn %d SASL bind in progress (tag=%ld).\n",
922 conn->c_connid, (long)tag ));
924 Debug( LDAP_DEBUG_ANY, "connection_operation: "
925 "error: SASL bind in progress (tag=%ld).\n",
928 send_ldap_result( conn, arg->co_op,
929 rc = LDAP_OPERATIONS_ERROR,
930 NULL, "SASL bind in progress", NULL, NULL );
931 goto operations_error;
936 rc = do_bind( conn, arg->co_op );
939 case LDAP_REQ_UNBIND:
940 rc = do_unbind( conn, arg->co_op );
944 rc = do_add( conn, arg->co_op );
947 case LDAP_REQ_DELETE:
948 rc = do_delete( conn, arg->co_op );
951 case LDAP_REQ_MODRDN:
952 rc = do_modrdn( conn, arg->co_op );
955 case LDAP_REQ_MODIFY:
956 rc = do_modify( conn, arg->co_op );
959 case LDAP_REQ_COMPARE:
960 rc = do_compare( conn, arg->co_op );
963 case LDAP_REQ_SEARCH:
964 rc = do_search( conn, arg->co_op );
967 case LDAP_REQ_ABANDON:
968 rc = do_abandon( conn, arg->co_op );
971 case LDAP_REQ_EXTENDED:
972 rc = do_extended( conn, arg->co_op );
977 LDAP_LOG(( "connection", LDAP_LEVEL_INFO,
978 "connection_operation: conn %d unknown LDAP request 0x%lx\n",
979 conn->c_connid, tag ));
981 Debug( LDAP_DEBUG_ANY, "unknown LDAP request 0x%lx\n",
984 arg->co_op->o_tag = LBER_ERROR;
985 send_ldap_disconnect( conn, arg->co_op,
986 LDAP_PROTOCOL_ERROR, "unknown LDAP request" );
991 if( rc == SLAPD_DISCONNECT ) tag = LBER_ERROR;
994 ldap_pvt_thread_mutex_lock( &num_ops_mutex );
996 ldap_pvt_thread_mutex_unlock( &num_ops_mutex );
998 ldap_pvt_thread_mutex_lock( &conn->c_mutex );
1000 conn->c_n_ops_executing--;
1001 conn->c_n_ops_completed++;
1003 LDAP_STAILQ_REMOVE( &conn->c_ops, arg->co_op, slap_op, o_next);
1004 LDAP_STAILQ_NEXT(arg->co_op, o_next) = NULL;
1005 slap_op_free( arg->co_op );
1007 arg->co_conn = NULL;
1008 free( (char *) arg );
1013 case LDAP_REQ_UNBIND:
1014 /* c_mutex is locked */
1015 connection_closing( conn );
1019 conn->c_sasl_bind_in_progress =
1020 rc == LDAP_SASL_BIND_IN_PROGRESS ? 1 : 0;
1022 if( conn->c_conn_state == SLAP_C_BINDING) {
1023 conn->c_conn_state = SLAP_C_ACTIVE;
1027 connection_resched( conn );
1029 ldap_pvt_thread_mutex_unlock( &conn->c_mutex );
1034 int connection_read(ber_socket_t s)
1038 assert( connections != NULL );
1040 ldap_pvt_thread_mutex_lock( &connections_mutex );
1042 /* get (locked) connection */
1043 c = connection_get( s );
1047 LDAP_LOG(( "connection", LDAP_LEVEL_INFO,
1048 "connection_read: sock %ld no connection\n",
1051 Debug( LDAP_DEBUG_ANY,
1052 "connection_read(%ld): no connection!\n",
1057 ldap_pvt_thread_mutex_unlock( &connections_mutex );
1063 if( c->c_conn_state == SLAP_C_CLOSING ) {
1065 LDAP_LOG(( "connection", LDAP_LEVEL_INFO,
1066 "connection_read: conn %d connection closing, ignoring input\n",
1069 Debug( LDAP_DEBUG_TRACE,
1070 "connection_read(%d): closing, ignoring input for id=%ld\n",
1071 s, c->c_connid, 0 );
1073 connection_return( c );
1074 ldap_pvt_thread_mutex_unlock( &connections_mutex );
1079 LDAP_LOG(( "connection", LDAP_LEVEL_DETAIL1,
1080 "connection_read: conn %d checking for input.\n", c->c_connid ));
1082 Debug( LDAP_DEBUG_TRACE,
1083 "connection_read(%d): checking for input on id=%ld\n",
1084 s, c->c_connid, 0 );
1088 if ( c->c_is_tls && c->c_needs_tls_accept ) {
1089 rc = ldap_pvt_tls_accept( c->c_sb, NULL );
1091 #if 0 /* required by next #if 0 */
1097 LDAP_LOG(( "connection", LDAP_LEVEL_ERR,
1098 "connection_read: conn %d TLS accept error, error %d\n",
1101 Debug( LDAP_DEBUG_TRACE,
1102 "connection_read(%d): TLS accept error "
1103 "error=%d id=%ld, closing\n",
1104 s, rc, c->c_connid );
1106 c->c_needs_tls_accept = 0;
1107 /* connections_mutex and c_mutex are locked */
1108 connection_closing( c );
1111 /* Drain input before close, to allow SSL error codes
1112 * to propagate to client. */
1119 rc = select(s+1, &rfd, NULL, NULL, &tv);
1121 ber_sockbuf_ctrl( c->c_sb, LBER_SB_OPT_DRAIN,
1125 connection_close( c );
1127 } else if ( rc == 0 ) {
1131 c->c_needs_tls_accept = 0;
1133 /* we need to let SASL know */
1134 ssl = (void *)ldap_pvt_tls_sb_ctx( c->c_sb );
1136 c->c_tls_ssf = (slap_ssf_t) ldap_pvt_tls_get_strength( ssl );
1137 if( c->c_tls_ssf > c->c_ssf ) {
1138 c->c_ssf = c->c_tls_ssf;
1141 authid = (char *)ldap_pvt_tls_get_peer( ssl );
1142 slap_sasl_external( c, c->c_tls_ssf, authid );
1144 connection_return( c );
1145 ldap_pvt_thread_mutex_unlock( &connections_mutex );
1150 #ifdef HAVE_CYRUS_SASL
1151 if ( c->c_sasl_layers ) {
1152 c->c_sasl_layers = 0;
1154 rc = ldap_pvt_sasl_install( c->c_sb, c->c_sasl_context );
1156 if( rc != LDAP_SUCCESS ) {
1158 LDAP_LOG(( "connection", LDAP_LEVEL_ERR,
1159 "connection_read: conn %d SASL install error %d, closing\n",
1162 Debug( LDAP_DEBUG_TRACE,
1163 "connection_read(%d): SASL install error "
1164 "error=%d id=%ld, closing\n",
1165 s, rc, c->c_connid );
1167 /* connections_mutex and c_mutex are locked */
1168 connection_closing( c );
1169 connection_close( c );
1170 connection_return( c );
1171 ldap_pvt_thread_mutex_unlock( &connections_mutex );
1177 #define CONNECTION_INPUT_LOOP 1
1179 #ifdef DATA_READY_LOOP
1180 while( !rc && ber_sockbuf_ctrl( c->c_sb, LBER_SB_DATA_READY, NULL ) )
1181 #elif CONNECTION_INPUT_LOOP
1185 /* How do we do this without getting into a busy loop ? */
1186 rc = connection_input( c );
1191 LDAP_LOG(( "connection", LDAP_LEVEL_ERR,
1192 "connection_read: conn %d input error %d, closing.\n",
1195 Debug( LDAP_DEBUG_TRACE,
1196 "connection_read(%d): input error=%d id=%ld, closing.\n",
1197 s, rc, c->c_connid );
1199 /* connections_mutex and c_mutex are locked */
1200 connection_closing( c );
1201 connection_close( c );
1202 connection_return( c );
1203 ldap_pvt_thread_mutex_unlock( &connections_mutex );
1207 if ( ber_sockbuf_ctrl( c->c_sb, LBER_SB_OPT_NEEDS_READ, NULL ) ) {
1208 slapd_set_read( s, 1 );
1211 if ( ber_sockbuf_ctrl( c->c_sb, LBER_SB_OPT_NEEDS_WRITE, NULL ) ) {
1212 slapd_set_write( s, 1 );
1215 connection_return( c );
1216 ldap_pvt_thread_mutex_unlock( &connections_mutex );
1230 #ifdef LDAP_CONNECTIONLESS
1235 if ( conn->c_currentber == NULL && (conn->c_currentber = ber_alloc())
1238 LDAP_LOG(( "connection", LDAP_LEVEL_ERR,
1239 "connection_input: conn %d ber_alloc failed.\n", conn->c_connid ));
1241 Debug( LDAP_DEBUG_ANY, "ber_alloc failed\n", 0, 0, 0 );
1248 #ifdef LDAP_CONNECTIONLESS
1251 char peername[sizeof("IP=255.255.255.255:65336")];
1252 len = ber_int_sb_read(conn->c_sb, &peeraddr,
1253 sizeof(struct sockaddr));
1254 if (len != sizeof(struct sockaddr))
1256 sprintf( peername, "IP=%s:%d",
1257 inet_ntoa( peeraddr.sa_in_addr.sin_addr ),
1258 (unsigned) ntohs( peeraddr.sa_in_addr.sin_port ) );
1259 Statslog( LDAP_DEBUG_STATS,
1260 "conn=%ld UDP request from %s (%s) accepted.\n",
1261 conn->c_connid, peername,
1262 conn->c_sock_name, 0, 0 );
1265 tag = ber_get_next( conn->c_sb, &len, conn->c_currentber );
1266 if ( tag != LDAP_TAG_MESSAGE ) {
1270 ber_sockbuf_ctrl( conn->c_sb, LBER_SB_OPT_GET_FD, &sd );
1273 LDAP_LOG(( "connection", LDAP_LEVEL_ERR,
1274 "connection_input: conn %d ber_get_next failed, errno %d (%s).\n",
1275 conn->c_connid, err, sock_errstr(err) ));
1277 Debug( LDAP_DEBUG_TRACE,
1278 "ber_get_next on fd %d failed errno=%d (%s)\n",
1279 sd, err, sock_errstr(err) );
1281 if ( err != EWOULDBLOCK && err != EAGAIN ) {
1282 /* log, close and send error */
1283 ber_free( conn->c_currentber, 1 );
1284 conn->c_currentber = NULL;
1291 ber = conn->c_currentber;
1292 conn->c_currentber = NULL;
1294 if ( (tag = ber_get_int( ber, &msgid )) != LDAP_TAG_MSGID ) {
1295 /* log, close and send error */
1297 LDAP_LOG(( "connection", LDAP_LEVEL_ERR,
1298 "connection_input: conn %d ber_get_int returns 0x%lx.\n",
1299 conn->c_connid, tag ));
1301 Debug( LDAP_DEBUG_ANY, "ber_get_int returns 0x%lx\n", tag, 0,
1308 if ( (tag = ber_peek_tag( ber, &len )) == LBER_ERROR ) {
1309 /* log, close and send error */
1311 LDAP_LOG(( "connection", LDAP_LEVEL_ERR,
1312 "connection_input: conn %d ber_peek_tag returns 0x%lx.\n",
1313 conn->c_connid, tag ));
1315 Debug( LDAP_DEBUG_ANY, "ber_peek_tag returns 0x%lx\n", tag, 0,
1323 #ifdef LDAP_CONNECTIONLESS
1324 if (conn->c_is_udp) {
1325 if (tag == LBER_OCTETSTRING) {
1326 ber_get_stringa( ber, &cdn );
1327 tag = ber_peek_tag(ber, &len);
1329 if (tag != LDAP_REQ_ABANDON && tag != LDAP_REQ_SEARCH) {
1331 LDAP_LOG(( "connection", LDAP_LEVEL_ERR,
1332 "connection_input: conn %d invalid req for UDP 0x%lx.\n",
1333 conn->c_connid, tag ));
1335 Debug( LDAP_DEBUG_ANY, "invalid req for UDP 0x%lx\n", tag, 0,
1343 if(tag == LDAP_REQ_BIND) {
1344 /* immediately abandon all exiting operations upon BIND */
1345 connection_abandon( conn );
1348 op = slap_op_alloc( ber, msgid, tag, conn->c_n_ops_received++ );
1350 #ifdef LDAP_CONNECTIONLESS
1351 op->o_peeraddr = peeraddr;
1354 op->o_protocol = LDAP_VERSION2;
1357 if ( conn->c_conn_state == SLAP_C_BINDING
1358 || conn->c_conn_state == SLAP_C_CLOSING )
1361 LDAP_LOG(( "connection", LDAP_LEVEL_INFO,
1362 "connection_input: conn %d deferring operation\n",
1365 Debug( LDAP_DEBUG_ANY, "deferring operation\n", 0, 0, 0 );
1367 conn->c_n_ops_pending++;
1368 LDAP_STAILQ_INSERT_TAIL( &conn->c_pending_ops, op, o_next );
1371 conn->c_n_ops_executing++;
1372 connection_op_activate( conn, op );
1376 if ( conn->c_struct_state != SLAP_C_USED ) {
1377 /* connection must have got closed underneath us */
1381 assert( conn->c_struct_state == SLAP_C_USED );
1387 connection_resched( Connection *conn )
1391 if( conn->c_conn_state == SLAP_C_CLOSING ) {
1394 ber_sockbuf_ctrl( conn->c_sb, LBER_SB_OPT_GET_FD, &sd );
1396 /* us trylock to avoid possible deadlock */
1397 rc = ldap_pvt_thread_mutex_trylock( &connections_mutex );
1401 LDAP_LOG(( "connection", LDAP_LEVEL_DETAIL1,
1402 "connection_resched: conn %d reaquiring locks.\n",
1405 Debug( LDAP_DEBUG_TRACE,
1406 "connection_resched: reaquiring locks conn=%ld sd=%d\n",
1407 conn->c_connid, sd, 0 );
1410 * reaquire locks in the right order...
1411 * this may allow another thread to close this connection,
1412 * so recheck state below.
1414 ldap_pvt_thread_mutex_unlock( &conn->c_mutex );
1415 ldap_pvt_thread_mutex_lock( &connections_mutex );
1416 ldap_pvt_thread_mutex_lock( &conn->c_mutex );
1419 if( conn->c_conn_state != SLAP_C_CLOSING ) {
1421 LDAP_LOG(( "connection", LDAP_LEVEL_INFO,
1422 "connection_resched: conn %d closed by other thread.\n",
1425 Debug( LDAP_DEBUG_TRACE, "connection_resched: "
1426 "closed by other thread conn=%ld sd=%d\n",
1427 conn->c_connid, sd, 0 );
1431 LDAP_LOG(( "connection", LDAP_LEVEL_DETAIL1,
1432 "connection_resched: conn %d attempting closing.\n",
1435 Debug( LDAP_DEBUG_TRACE, "connection_resched: "
1436 "attempting closing conn=%ld sd=%d\n",
1437 conn->c_connid, sd, 0 );
1439 connection_close( conn );
1442 ldap_pvt_thread_mutex_unlock( &connections_mutex );
1446 if( conn->c_conn_state != SLAP_C_ACTIVE ) {
1447 /* other states need different handling */
1451 while ((op = LDAP_STAILQ_FIRST( &conn->c_pending_ops )) != NULL) {
1452 LDAP_STAILQ_REMOVE_HEAD( &conn->c_pending_ops, o_next );
1453 LDAP_STAILQ_NEXT(op, o_next) = NULL;
1454 /* pending operations should not be marked for abandonment */
1455 assert(!op->o_abandon);
1457 conn->c_n_ops_pending--;
1458 conn->c_n_ops_executing++;
1460 connection_op_activate( conn, op );
1462 if ( conn->c_conn_state == SLAP_C_BINDING ) {
1469 static int connection_op_activate( Connection *conn, Operation *op )
1473 ber_tag_t tag = op->o_tag;
1475 if(tag == LDAP_REQ_BIND) {
1476 conn->c_conn_state = SLAP_C_BINDING;
1479 arg = (struct co_arg *) ch_malloc( sizeof(struct co_arg) );
1480 arg->co_conn = conn;
1483 if (!arg->co_op->o_dn.bv_len) {
1484 arg->co_op->o_authz = conn->c_authz;
1485 arg->co_op->o_dn.bv_val = ch_strdup( conn->c_dn.bv_val ?
1486 conn->c_dn.bv_val : "" );
1487 arg->co_op->o_ndn.bv_val = ch_strdup( conn->c_ndn.bv_val ?
1488 conn->c_ndn.bv_val : "" );
1490 arg->co_op->o_authtype = conn->c_authtype;
1491 ber_dupbv( &arg->co_op->o_authmech, &conn->c_authmech );
1493 if (!arg->co_op->o_protocol) {
1494 arg->co_op->o_protocol = conn->c_protocol
1495 ? conn->c_protocol : LDAP_VERSION3;
1497 arg->co_op->o_connid = conn->c_connid;
1499 LDAP_STAILQ_INSERT_TAIL( &conn->c_ops, arg->co_op, o_next );
1501 status = ldap_pvt_thread_pool_submit( &connection_pool,
1502 connection_operation, (void *) arg );
1504 if ( status != 0 ) {
1506 LDAP_LOG(( "connection", LDAP_LEVEL_ERR,
1507 "connection_op_activate: conn %d thread pool submit failed.\n",
1510 Debug( LDAP_DEBUG_ANY,
1511 "ldap_pvt_thread_pool_submit failed (%d)\n", status, 0, 0 );
1513 /* should move op to pending list */
1519 int connection_write(ber_socket_t s)
1522 assert( connections != NULL );
1524 ldap_pvt_thread_mutex_lock( &connections_mutex );
1526 c = connection_get( s );
1528 slapd_clr_write( s, 0);
1532 LDAP_LOG(( "connection", LDAP_LEVEL_ERR,
1533 "connection_write: sock %ld no connection!\n",(long)s));
1535 Debug( LDAP_DEBUG_ANY,
1536 "connection_write(%ld): no connection!\n",
1540 ldap_pvt_thread_mutex_unlock( &connections_mutex );
1547 LDAP_LOG(( "connection", LDAP_LEVEL_DETAIL1,
1548 "connection_write conn %d waking output.\n",
1551 Debug( LDAP_DEBUG_TRACE,
1552 "connection_write(%d): waking output for id=%ld\n",
1553 s, c->c_connid, 0 );
1555 ldap_pvt_thread_cond_signal( &c->c_write_cv );
1557 if ( ber_sockbuf_ctrl( c->c_sb, LBER_SB_OPT_NEEDS_READ, NULL ) )
1558 slapd_set_read( s, 1 );
1559 if ( ber_sockbuf_ctrl( c->c_sb, LBER_SB_OPT_NEEDS_WRITE, NULL ) )
1560 slapd_set_write( s, 1 );
1561 connection_return( c );
1562 ldap_pvt_thread_mutex_unlock( &connections_mutex );