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 = NULL;
426 c->c_peer_domain = NULL;
427 c->c_peer_name = NULL;
428 c->c_sock_name = NULL;
430 LDAP_STAILQ_INIT(&c->c_ops);
431 LDAP_STAILQ_INIT(&c->c_pending_ops);
433 c->c_sasl_bind_mech.bv_val = NULL;
434 c->c_sasl_bind_mech.bv_len = 0;
435 c->c_sasl_context = NULL;
436 c->c_sasl_extra = NULL;
438 c->c_sb = ber_sockbuf_alloc( );
441 ber_len_t max = sockbuf_max_incoming;
442 ber_sockbuf_ctrl( c->c_sb, LBER_SB_OPT_SET_MAX_INCOMING, &max );
445 c->c_currentber = NULL;
447 /* should check status of thread calls */
448 ldap_pvt_thread_mutex_init( &c->c_mutex );
449 ldap_pvt_thread_mutex_init( &c->c_write_mutex );
450 ldap_pvt_thread_cond_init( &c->c_write_cv );
452 c->c_struct_state = SLAP_C_UNUSED;
455 ldap_pvt_thread_mutex_lock( &c->c_mutex );
457 assert( c->c_struct_state == SLAP_C_UNUSED );
458 assert( c->c_authmech.bv_val == NULL );
459 assert( c->c_dn.bv_val == NULL );
460 assert( c->c_ndn.bv_val == NULL );
461 assert( c->c_cdn.bv_val == NULL );
462 assert( c->c_groups == NULL );
463 assert( c->c_listener_url == NULL );
464 assert( c->c_peer_domain == NULL );
465 assert( c->c_peer_name == NULL );
466 assert( c->c_sock_name == NULL );
467 assert( LDAP_STAILQ_EMPTY(&c->c_ops) );
468 assert( LDAP_STAILQ_EMPTY(&c->c_pending_ops) );
469 assert( c->c_sasl_bind_mech.bv_val == NULL );
470 assert( c->c_sasl_context == NULL );
471 assert( c->c_sasl_extra == NULL );
472 assert( c->c_currentber == NULL );
474 c->c_listener_url = ch_strdup( url );
475 c->c_peer_domain = ch_strdup( dnsname );
476 c->c_peer_name = ch_strdup( peername );
477 c->c_sock_name = ch_strdup( sockname );
479 c->c_n_ops_received = 0;
480 c->c_n_ops_executing = 0;
481 c->c_n_ops_pending = 0;
482 c->c_n_ops_completed = 0;
488 /* set to zero until bind, implies LDAP_VERSION3 */
492 c->c_activitytime = c->c_starttime = slap_get_time();
494 if( global_idletimeout > 0 ) {
495 c->c_activitytime = c->c_starttime = slap_get_time();
499 #ifdef LDAP_CONNECTIONLESS
501 if (tls_udp_option == 2)
505 ber_sockbuf_add_io( c->c_sb, &ber_sockbuf_io_debug,
506 LBER_SBIOD_LEVEL_PROVIDER, (void*)"udp_" );
508 ber_sockbuf_add_io( c->c_sb, &ber_sockbuf_io_udp,
509 LBER_SBIOD_LEVEL_PROVIDER, (void *)&s );
510 ber_sockbuf_add_io( c->c_sb, &ber_sockbuf_io_readahead,
511 LBER_SBIOD_LEVEL_PROVIDER, NULL );
516 ber_sockbuf_add_io( c->c_sb, &ber_sockbuf_io_debug,
517 LBER_SBIOD_LEVEL_PROVIDER, (void*)"tcp_" );
519 ber_sockbuf_add_io( c->c_sb, &ber_sockbuf_io_tcp,
520 LBER_SBIOD_LEVEL_PROVIDER, (void *)&s );
524 ber_sockbuf_add_io( c->c_sb, &ber_sockbuf_io_debug,
525 INT_MAX, (void*)"ldap_" );
528 if( ber_sockbuf_ctrl( c->c_sb, LBER_SB_OPT_SET_NONBLOCK,
529 c /* non-NULL */ ) < 0 )
532 LDAP_LOG(( "connection", LDAP_LEVEL_INFO,
533 "connection_init: conn %d set nonblocking failed\n",
536 Debug( LDAP_DEBUG_ANY,
537 "connection_init(%d, %s): set nonblocking failed\n",
538 s, c->c_peer_name, 0 );
542 id = c->c_connid = conn_nextid++;
544 c->c_conn_state = SLAP_C_INACTIVE;
545 c->c_struct_state = SLAP_C_USED;
547 c->c_ssf = c->c_transport_ssf = ssf;
551 if ( tls_udp_option == 1 ) {
553 c->c_needs_tls_accept = 1;
556 c->c_needs_tls_accept = 0;
561 slap_sasl_external( c, ssf, authid );
563 ldap_pvt_thread_mutex_unlock( &c->c_mutex );
564 ldap_pvt_thread_mutex_unlock( &connections_mutex );
566 backend_connection_init(c);
571 void connection2anonymous( Connection *c )
573 assert( connections != NULL );
577 ber_len_t max = sockbuf_max_incoming;
578 ber_sockbuf_ctrl( c->c_sb, LBER_SB_OPT_SET_MAX_INCOMING, &max );
581 if(c->c_authmech.bv_val != NULL ) {
582 free(c->c_authmech.bv_val);
583 c->c_authmech.bv_val = NULL;
585 c->c_authmech.bv_len = 0;
587 if(c->c_dn.bv_val != NULL) {
588 free(c->c_dn.bv_val);
589 c->c_dn.bv_val = NULL;
592 if(c->c_ndn.bv_val != NULL) {
593 free(c->c_ndn.bv_val);
594 c->c_ndn.bv_val = NULL;
598 if(c->c_cdn.bv_val != NULL) {
599 free(c->c_cdn.bv_val);
600 c->c_cdn.bv_val = NULL;
604 c->c_authc_backend = NULL;
605 c->c_authz_backend = NULL;
608 GroupAssertion *g, *n;
609 for (g = c->c_groups; g; g=n)
620 connection_destroy( Connection *c )
622 /* note: connections_mutex should be locked by caller */
625 assert( connections != NULL );
627 assert( c->c_struct_state != SLAP_C_UNUSED );
628 assert( c->c_conn_state != SLAP_C_INVALID );
629 assert( LDAP_STAILQ_EMPTY(&c->c_ops) );
631 backend_connection_destroy(c);
636 c->c_activitytime = c->c_starttime = 0;
638 connection2anonymous( c );
640 if(c->c_listener_url != NULL) {
641 free(c->c_listener_url);
642 c->c_listener_url = NULL;
645 if(c->c_peer_domain != NULL) {
646 free(c->c_peer_domain);
647 c->c_peer_domain = NULL;
649 if(c->c_peer_name != NULL) {
652 * If peer was a domain socket, unlink. Mind you,
653 * they may be un-named. Should we leave this to
656 if (strncmp(c->c_peer_name, "PATH=", 5) == 0) {
657 char *path = c->c_peer_name + 5;
662 #endif /* LDAP_PF_LOCAL */
664 free(c->c_peer_name);
665 c->c_peer_name = NULL;
667 if(c->c_sock_name != NULL) {
668 free(c->c_sock_name);
669 c->c_sock_name = NULL;
672 c->c_sasl_bind_in_progress = 0;
673 if(c->c_sasl_bind_mech.bv_val != NULL) {
674 free(c->c_sasl_bind_mech.bv_val);
675 c->c_sasl_bind_mech.bv_val = NULL;
677 c->c_sasl_bind_mech.bv_len = 0;
679 slap_sasl_close( c );
681 if ( c->c_currentber != NULL ) {
682 ber_free( c->c_currentber, 1 );
683 c->c_currentber = NULL;
686 ber_sockbuf_ctrl( c->c_sb, LBER_SB_OPT_GET_FD, &sd );
687 if ( sd != AC_SOCKET_INVALID ) {
688 slapd_remove( sd, 0 );
690 Statslog( LDAP_DEBUG_STATS,
691 "conn=%ld fd=%d closed\n",
692 c->c_connid, sd, 0, 0, 0 );
695 ber_sockbuf_free( c->c_sb );
697 c->c_sb = ber_sockbuf_alloc( );
700 ber_len_t max = sockbuf_max_incoming;
701 ber_sockbuf_ctrl( c->c_sb, LBER_SB_OPT_SET_MAX_INCOMING, &max );
704 c->c_conn_state = SLAP_C_INVALID;
705 c->c_struct_state = SLAP_C_UNUSED;
708 int connection_state_closing( Connection *c )
710 /* c_mutex must be locked by caller */
714 assert( c->c_struct_state == SLAP_C_USED );
716 state = c->c_conn_state;
718 assert( state != SLAP_C_INVALID );
720 return state == SLAP_C_CLOSING;
723 static void connection_abandon( Connection *c )
725 /* c_mutex must be locked by caller */
729 LDAP_STAILQ_FOREACH(o, &c->c_ops, o_next) {
730 ldap_pvt_thread_mutex_lock( &o->o_abandonmutex );
732 ldap_pvt_thread_mutex_unlock( &o->o_abandonmutex );
735 /* remove pending operations */
736 while ( (o = LDAP_STAILQ_FIRST( &c->c_pending_ops )) != NULL) {
737 LDAP_STAILQ_REMOVE_HEAD( &c->c_pending_ops, o_next );
738 LDAP_STAILQ_NEXT(o, o_next) = NULL;
743 void connection_closing( Connection *c )
745 assert( connections != NULL );
747 assert( c->c_struct_state == SLAP_C_USED );
748 assert( c->c_conn_state != SLAP_C_INVALID );
750 /* c_mutex must be locked by caller */
752 if( c->c_conn_state != SLAP_C_CLOSING ) {
755 ber_sockbuf_ctrl( c->c_sb, LBER_SB_OPT_GET_FD, &sd );
757 LDAP_LOG(( "connection", LDAP_LEVEL_DETAIL1,
758 "connection_closing: conn %d readying socket %d for close.\n",
761 Debug( LDAP_DEBUG_TRACE,
762 "connection_closing: readying conn=%ld sd=%d for close\n",
763 c->c_connid, sd, 0 );
765 /* update state to closing */
766 c->c_conn_state = SLAP_C_CLOSING;
768 /* don't listen on this port anymore */
769 slapd_clr_read( sd, 1 );
771 /* abandon active operations */
772 connection_abandon( c );
774 /* wake write blocked operations */
775 slapd_clr_write( sd, 1 );
776 ldap_pvt_thread_cond_signal( &c->c_write_cv );
780 static void connection_close( Connection *c )
784 assert( connections != NULL );
786 assert( c->c_struct_state == SLAP_C_USED );
787 assert( c->c_conn_state == SLAP_C_CLOSING );
789 /* note: connections_mutex and c_mutex should be locked by caller */
791 ber_sockbuf_ctrl( c->c_sb, LBER_SB_OPT_GET_FD, &sd );
792 if( !LDAP_STAILQ_EMPTY(&c->c_ops) ) {
794 LDAP_LOG(( "connection", LDAP_LEVEL_DETAIL1,
795 "connection_close: conn %d deferring sd %d\n",
798 Debug( LDAP_DEBUG_TRACE,
799 "connection_close: deferring conn=%ld sd=%d\n",
800 c->c_connid, sd, 0 );
806 LDAP_LOG(( "connection", LDAP_LEVEL_RESULTS,
807 "connection_close: conn %d sd %d\n",
810 Debug( LDAP_DEBUG_TRACE, "connection_close: conn=%ld sd=%d\n",
811 c->c_connid, sd, 0 );
813 connection_destroy( c );
816 unsigned long connections_nextid(void)
819 assert( connections != NULL );
821 ldap_pvt_thread_mutex_lock( &connections_mutex );
825 ldap_pvt_thread_mutex_unlock( &connections_mutex );
830 Connection* connection_first( ber_socket_t *index )
832 assert( connections != NULL );
833 assert( index != NULL );
835 ldap_pvt_thread_mutex_lock( &connections_mutex );
839 return connection_next(NULL, index);
842 Connection* connection_next( Connection *c, ber_socket_t *index )
844 assert( connections != NULL );
845 assert( index != NULL );
846 assert( *index <= dtblsize );
849 ldap_pvt_thread_mutex_unlock( &c->c_mutex );
854 for(; *index < dtblsize; (*index)++) {
855 if( connections[*index].c_struct_state == SLAP_C_UNINITIALIZED ) {
856 assert( connections[*index].c_conn_state == SLAP_C_INVALID );
864 if( connections[*index].c_struct_state == SLAP_C_USED ) {
865 assert( connections[*index].c_conn_state != SLAP_C_INVALID );
866 c = &connections[(*index)++];
870 assert( connections[*index].c_struct_state == SLAP_C_UNUSED );
871 assert( connections[*index].c_conn_state == SLAP_C_INVALID );
875 ldap_pvt_thread_mutex_lock( &c->c_mutex );
881 void connection_done( Connection *c )
883 assert( connections != NULL );
886 ldap_pvt_thread_mutex_unlock( &c->c_mutex );
889 ldap_pvt_thread_mutex_unlock( &connections_mutex );
893 * connection_activity - handle the request operation op on connection
894 * conn. This routine figures out what kind of operation it is and
895 * calls the appropriate stub to handle it.
899 connection_operation( void *arg_v )
902 struct co_arg *arg = arg_v;
903 ber_tag_t tag = arg->co_op->o_tag;
904 Connection *conn = arg->co_conn;
906 ldap_pvt_thread_mutex_lock( &num_ops_mutex );
908 ldap_pvt_thread_mutex_unlock( &num_ops_mutex );
910 if( conn->c_sasl_bind_in_progress && tag != LDAP_REQ_BIND ) {
912 LDAP_LOG(( "connection", LDAP_LEVEL_ERR,
913 "connection_operation: conn %d SASL bind in progress (tag=%ld).\n",
914 conn->c_connid, (long)tag ));
916 Debug( LDAP_DEBUG_ANY, "connection_operation: "
917 "error: SASL bind in progress (tag=%ld).\n",
920 send_ldap_result( conn, arg->co_op,
921 rc = LDAP_OPERATIONS_ERROR,
922 NULL, "SASL bind in progress", NULL, NULL );
923 goto operations_error;
928 rc = do_bind( conn, arg->co_op );
931 case LDAP_REQ_UNBIND:
932 rc = do_unbind( conn, arg->co_op );
936 rc = do_add( conn, arg->co_op );
939 case LDAP_REQ_DELETE:
940 rc = do_delete( conn, arg->co_op );
943 case LDAP_REQ_MODRDN:
944 rc = do_modrdn( conn, arg->co_op );
947 case LDAP_REQ_MODIFY:
948 rc = do_modify( conn, arg->co_op );
951 case LDAP_REQ_COMPARE:
952 rc = do_compare( conn, arg->co_op );
955 case LDAP_REQ_SEARCH:
956 rc = do_search( conn, arg->co_op );
959 case LDAP_REQ_ABANDON:
960 rc = do_abandon( conn, arg->co_op );
963 case LDAP_REQ_EXTENDED:
964 rc = do_extended( conn, arg->co_op );
969 LDAP_LOG(( "connection", LDAP_LEVEL_INFO,
970 "connection_operation: conn %d unknown LDAP request 0x%lx\n",
971 conn->c_connid, tag ));
973 Debug( LDAP_DEBUG_ANY, "unknown LDAP request 0x%lx\n",
976 arg->co_op->o_tag = LBER_ERROR;
977 send_ldap_disconnect( conn, arg->co_op,
978 LDAP_PROTOCOL_ERROR, "unknown LDAP request" );
983 if( rc == SLAPD_DISCONNECT ) tag = LBER_ERROR;
986 ldap_pvt_thread_mutex_lock( &num_ops_mutex );
988 ldap_pvt_thread_mutex_unlock( &num_ops_mutex );
990 ldap_pvt_thread_mutex_lock( &conn->c_mutex );
992 conn->c_n_ops_executing--;
993 conn->c_n_ops_completed++;
995 LDAP_STAILQ_REMOVE( &conn->c_ops, arg->co_op, slap_op, o_next);
996 LDAP_STAILQ_NEXT(arg->co_op, o_next) = NULL;
997 slap_op_free( arg->co_op );
1000 free( (char *) arg );
1005 case LDAP_REQ_UNBIND:
1006 /* c_mutex is locked */
1007 connection_closing( conn );
1011 conn->c_sasl_bind_in_progress =
1012 rc == LDAP_SASL_BIND_IN_PROGRESS ? 1 : 0;
1014 if( conn->c_conn_state == SLAP_C_BINDING) {
1015 conn->c_conn_state = SLAP_C_ACTIVE;
1019 connection_resched( conn );
1021 ldap_pvt_thread_mutex_unlock( &conn->c_mutex );
1026 int connection_read(ber_socket_t s)
1030 assert( connections != NULL );
1032 ldap_pvt_thread_mutex_lock( &connections_mutex );
1034 /* get (locked) connection */
1035 c = connection_get( s );
1039 LDAP_LOG(( "connection", LDAP_LEVEL_INFO,
1040 "connection_read: sock %ld no connection\n",
1043 Debug( LDAP_DEBUG_ANY,
1044 "connection_read(%ld): no connection!\n",
1049 ldap_pvt_thread_mutex_unlock( &connections_mutex );
1055 if( c->c_conn_state == SLAP_C_CLOSING ) {
1057 LDAP_LOG(( "connection", LDAP_LEVEL_INFO,
1058 "connection_read: conn %d connection closing, ignoring input\n",
1061 Debug( LDAP_DEBUG_TRACE,
1062 "connection_read(%d): closing, ignoring input for id=%ld\n",
1063 s, c->c_connid, 0 );
1065 connection_return( c );
1066 ldap_pvt_thread_mutex_unlock( &connections_mutex );
1071 LDAP_LOG(( "connection", LDAP_LEVEL_DETAIL1,
1072 "connection_read: conn %d checking for input.\n", c->c_connid ));
1074 Debug( LDAP_DEBUG_TRACE,
1075 "connection_read(%d): checking for input on id=%ld\n",
1076 s, c->c_connid, 0 );
1080 if ( c->c_is_tls && c->c_needs_tls_accept ) {
1081 rc = ldap_pvt_tls_accept( c->c_sb, NULL );
1083 #if 0 /* required by next #if 0 */
1089 LDAP_LOG(( "connection", LDAP_LEVEL_ERR,
1090 "connection_read: conn %d TLS accept error, error %d\n",
1093 Debug( LDAP_DEBUG_TRACE,
1094 "connection_read(%d): TLS accept error "
1095 "error=%d id=%ld, closing\n",
1096 s, rc, c->c_connid );
1098 c->c_needs_tls_accept = 0;
1099 /* connections_mutex and c_mutex are locked */
1100 connection_closing( c );
1103 /* Drain input before close, to allow SSL error codes
1104 * to propagate to client. */
1111 rc = select(s+1, &rfd, NULL, NULL, &tv);
1113 ber_sockbuf_ctrl( c->c_sb, LBER_SB_OPT_DRAIN,
1117 connection_close( c );
1119 } else if ( rc == 0 ) {
1123 c->c_needs_tls_accept = 0;
1125 /* we need to let SASL know */
1126 ssl = (void *)ldap_pvt_tls_sb_ctx( c->c_sb );
1128 c->c_tls_ssf = (slap_ssf_t) ldap_pvt_tls_get_strength( ssl );
1129 if( c->c_tls_ssf > c->c_ssf ) {
1130 c->c_ssf = c->c_tls_ssf;
1133 authid = (char *)ldap_pvt_tls_get_peer( ssl );
1134 slap_sasl_external( c, c->c_tls_ssf, authid );
1136 connection_return( c );
1137 ldap_pvt_thread_mutex_unlock( &connections_mutex );
1142 #ifdef HAVE_CYRUS_SASL
1143 if ( c->c_sasl_layers ) {
1144 c->c_sasl_layers = 0;
1146 rc = ldap_pvt_sasl_install( c->c_sb, c->c_sasl_context );
1148 if( rc != LDAP_SUCCESS ) {
1150 LDAP_LOG(( "connection", LDAP_LEVEL_ERR,
1151 "connection_read: conn %d SASL install error %d, closing\n",
1154 Debug( LDAP_DEBUG_TRACE,
1155 "connection_read(%d): SASL install error "
1156 "error=%d id=%ld, closing\n",
1157 s, rc, c->c_connid );
1159 /* connections_mutex and c_mutex are locked */
1160 connection_closing( c );
1161 connection_close( c );
1162 connection_return( c );
1163 ldap_pvt_thread_mutex_unlock( &connections_mutex );
1169 #define CONNECTION_INPUT_LOOP 1
1171 #ifdef DATA_READY_LOOP
1172 while( !rc && ber_sockbuf_ctrl( c->c_sb, LBER_SB_DATA_READY, NULL ) )
1173 #elif CONNECTION_INPUT_LOOP
1177 /* How do we do this without getting into a busy loop ? */
1178 rc = connection_input( c );
1183 LDAP_LOG(( "connection", LDAP_LEVEL_ERR,
1184 "connection_read: conn %d input error %d, closing.\n",
1187 Debug( LDAP_DEBUG_TRACE,
1188 "connection_read(%d): input error=%d id=%ld, closing.\n",
1189 s, rc, c->c_connid );
1191 /* connections_mutex and c_mutex are locked */
1192 connection_closing( c );
1193 connection_close( c );
1194 connection_return( c );
1195 ldap_pvt_thread_mutex_unlock( &connections_mutex );
1199 if ( ber_sockbuf_ctrl( c->c_sb, LBER_SB_OPT_NEEDS_READ, NULL ) ) {
1200 slapd_set_read( s, 1 );
1203 if ( ber_sockbuf_ctrl( c->c_sb, LBER_SB_OPT_NEEDS_WRITE, NULL ) ) {
1204 slapd_set_write( s, 1 );
1207 connection_return( c );
1208 ldap_pvt_thread_mutex_unlock( &connections_mutex );
1222 #ifdef LDAP_CONNECTIONLESS
1227 if ( conn->c_currentber == NULL && (conn->c_currentber = ber_alloc())
1230 LDAP_LOG(( "connection", LDAP_LEVEL_ERR,
1231 "connection_input: conn %d ber_alloc failed.\n", conn->c_connid ));
1233 Debug( LDAP_DEBUG_ANY, "ber_alloc failed\n", 0, 0, 0 );
1240 #ifdef LDAP_CONNECTIONLESS
1243 char peername[sizeof("IP=255.255.255.255:65336")];
1244 len = ber_int_sb_read(conn->c_sb, &peeraddr,
1245 sizeof(struct sockaddr));
1246 if (len != sizeof(struct sockaddr))
1248 sprintf( peername, "IP=%s:%d",
1249 inet_ntoa( peeraddr.sa_in_addr.sin_addr ),
1250 (unsigned) ntohs( peeraddr.sa_in_addr.sin_port ) );
1251 Statslog( LDAP_DEBUG_STATS,
1252 "conn=%ld UDP request from %s (%s) accepted.\n",
1253 conn->c_connid, peername,
1254 conn->c_sock_name, 0, 0 );
1257 tag = ber_get_next( conn->c_sb, &len, conn->c_currentber );
1258 if ( tag != LDAP_TAG_MESSAGE ) {
1262 ber_sockbuf_ctrl( conn->c_sb, LBER_SB_OPT_GET_FD, &sd );
1265 LDAP_LOG(( "connection", LDAP_LEVEL_ERR,
1266 "connection_input: conn %d ber_get_next failed, errno %d (%s).\n",
1267 conn->c_connid, err, sock_errstr(err) ));
1269 Debug( LDAP_DEBUG_TRACE,
1270 "ber_get_next on fd %d failed errno=%d (%s)\n",
1271 sd, err, sock_errstr(err) );
1273 if ( err != EWOULDBLOCK && err != EAGAIN ) {
1274 /* log, close and send error */
1275 ber_free( conn->c_currentber, 1 );
1276 conn->c_currentber = NULL;
1283 ber = conn->c_currentber;
1284 conn->c_currentber = NULL;
1286 if ( (tag = ber_get_int( ber, &msgid )) != LDAP_TAG_MSGID ) {
1287 /* log, close and send error */
1289 LDAP_LOG(( "connection", LDAP_LEVEL_ERR,
1290 "connection_input: conn %d ber_get_int returns 0x%lx.\n",
1291 conn->c_connid, tag ));
1293 Debug( LDAP_DEBUG_ANY, "ber_get_int returns 0x%lx\n", tag, 0,
1300 if ( (tag = ber_peek_tag( ber, &len )) == LBER_ERROR ) {
1301 /* log, close and send error */
1303 LDAP_LOG(( "connection", LDAP_LEVEL_ERR,
1304 "connection_input: conn %d ber_peek_tag returns 0x%lx.\n",
1305 conn->c_connid, tag ));
1307 Debug( LDAP_DEBUG_ANY, "ber_peek_tag returns 0x%lx\n", tag, 0,
1315 #ifdef LDAP_CONNECTIONLESS
1316 if (conn->c_is_udp) {
1317 if (tag == LBER_OCTETSTRING) {
1318 ber_get_stringa( ber, &cdn );
1319 tag = ber_peek_tag(ber, &len);
1321 if (tag != LDAP_REQ_ABANDON && tag != LDAP_REQ_SEARCH) {
1323 LDAP_LOG(( "connection", LDAP_LEVEL_ERR,
1324 "connection_input: conn %d invalid req for UDP 0x%lx.\n",
1325 conn->c_connid, tag ));
1327 Debug( LDAP_DEBUG_ANY, "invalid req for UDP 0x%lx\n", tag, 0,
1335 if(tag == LDAP_REQ_BIND) {
1336 /* immediately abandon all exiting operations upon BIND */
1337 connection_abandon( conn );
1340 op = slap_op_alloc( ber, msgid, tag, conn->c_n_ops_received++ );
1342 #ifdef LDAP_CONNECTIONLESS
1343 op->o_peeraddr = peeraddr;
1346 op->o_protocol = LDAP_VERSION2;
1349 if ( conn->c_conn_state == SLAP_C_BINDING
1350 || conn->c_conn_state == SLAP_C_CLOSING )
1353 LDAP_LOG(( "connection", LDAP_LEVEL_INFO,
1354 "connection_input: conn %d deferring operation\n",
1357 Debug( LDAP_DEBUG_ANY, "deferring operation\n", 0, 0, 0 );
1359 conn->c_n_ops_pending++;
1360 LDAP_STAILQ_INSERT_TAIL( &conn->c_pending_ops, op, o_next );
1363 conn->c_n_ops_executing++;
1364 connection_op_activate( conn, op );
1368 if ( conn->c_struct_state != SLAP_C_USED ) {
1369 /* connection must have got closed underneath us */
1373 assert( conn->c_struct_state == SLAP_C_USED );
1379 connection_resched( Connection *conn )
1383 if( conn->c_conn_state == SLAP_C_CLOSING ) {
1386 ber_sockbuf_ctrl( conn->c_sb, LBER_SB_OPT_GET_FD, &sd );
1388 /* us trylock to avoid possible deadlock */
1389 rc = ldap_pvt_thread_mutex_trylock( &connections_mutex );
1393 LDAP_LOG(( "connection", LDAP_LEVEL_DETAIL1,
1394 "connection_resched: conn %d reaquiring locks.\n",
1397 Debug( LDAP_DEBUG_TRACE,
1398 "connection_resched: reaquiring locks conn=%ld sd=%d\n",
1399 conn->c_connid, sd, 0 );
1402 * reaquire locks in the right order...
1403 * this may allow another thread to close this connection,
1404 * so recheck state below.
1406 ldap_pvt_thread_mutex_unlock( &conn->c_mutex );
1407 ldap_pvt_thread_mutex_lock( &connections_mutex );
1408 ldap_pvt_thread_mutex_lock( &conn->c_mutex );
1411 if( conn->c_conn_state != SLAP_C_CLOSING ) {
1413 LDAP_LOG(( "connection", LDAP_LEVEL_INFO,
1414 "connection_resched: conn %d closed by other thread.\n",
1417 Debug( LDAP_DEBUG_TRACE, "connection_resched: "
1418 "closed by other thread conn=%ld sd=%d\n",
1419 conn->c_connid, sd, 0 );
1423 LDAP_LOG(( "connection", LDAP_LEVEL_DETAIL1,
1424 "connection_resched: conn %d attempting closing.\n",
1427 Debug( LDAP_DEBUG_TRACE, "connection_resched: "
1428 "attempting closing conn=%ld sd=%d\n",
1429 conn->c_connid, sd, 0 );
1431 connection_close( conn );
1434 ldap_pvt_thread_mutex_unlock( &connections_mutex );
1438 if( conn->c_conn_state != SLAP_C_ACTIVE ) {
1439 /* other states need different handling */
1443 while ((op = LDAP_STAILQ_FIRST( &conn->c_pending_ops )) != NULL) {
1444 LDAP_STAILQ_REMOVE_HEAD( &conn->c_pending_ops, o_next );
1445 LDAP_STAILQ_NEXT(op, o_next) = NULL;
1446 /* pending operations should not be marked for abandonment */
1447 assert(!op->o_abandon);
1449 conn->c_n_ops_pending--;
1450 conn->c_n_ops_executing++;
1452 connection_op_activate( conn, op );
1454 if ( conn->c_conn_state == SLAP_C_BINDING ) {
1461 static int connection_op_activate( Connection *conn, Operation *op )
1465 ber_tag_t tag = op->o_tag;
1467 if(tag == LDAP_REQ_BIND) {
1468 conn->c_conn_state = SLAP_C_BINDING;
1471 arg = (struct co_arg *) ch_malloc( sizeof(struct co_arg) );
1472 arg->co_conn = conn;
1475 if (!arg->co_op->o_dn.bv_len) {
1476 arg->co_op->o_authz = conn->c_authz;
1477 arg->co_op->o_dn.bv_val = ch_strdup( conn->c_dn.bv_val ?
1478 conn->c_dn.bv_val : "" );
1479 arg->co_op->o_ndn.bv_val = ch_strdup( conn->c_ndn.bv_val ?
1480 conn->c_ndn.bv_val : "" );
1482 arg->co_op->o_authtype = conn->c_authtype;
1483 ber_dupbv( &arg->co_op->o_authmech, &conn->c_authmech );
1485 if (!arg->co_op->o_protocol) {
1486 arg->co_op->o_protocol = conn->c_protocol
1487 ? conn->c_protocol : LDAP_VERSION3;
1489 arg->co_op->o_connid = conn->c_connid;
1491 LDAP_STAILQ_INSERT_TAIL( &conn->c_ops, arg->co_op, o_next );
1493 status = ldap_pvt_thread_pool_submit( &connection_pool,
1494 connection_operation, (void *) arg );
1496 if ( status != 0 ) {
1498 LDAP_LOG(( "connection", LDAP_LEVEL_ERR,
1499 "connection_op_activate: conn %d thread pool submit failed.\n",
1502 Debug( LDAP_DEBUG_ANY,
1503 "ldap_pvt_thread_pool_submit failed (%d)\n", status, 0, 0 );
1505 /* should move op to pending list */
1511 int connection_write(ber_socket_t s)
1514 assert( connections != NULL );
1516 ldap_pvt_thread_mutex_lock( &connections_mutex );
1518 c = connection_get( s );
1520 slapd_clr_write( s, 0);
1524 LDAP_LOG(( "connection", LDAP_LEVEL_ERR,
1525 "connection_write: sock %ld no connection!\n",(long)s));
1527 Debug( LDAP_DEBUG_ANY,
1528 "connection_write(%ld): no connection!\n",
1532 ldap_pvt_thread_mutex_unlock( &connections_mutex );
1539 LDAP_LOG(( "connection", LDAP_LEVEL_DETAIL1,
1540 "connection_write conn %d waking output.\n",
1543 Debug( LDAP_DEBUG_TRACE,
1544 "connection_write(%d): waking output for id=%ld\n",
1545 s, c->c_connid, 0 );
1547 ldap_pvt_thread_cond_signal( &c->c_write_cv );
1549 if ( ber_sockbuf_ctrl( c->c_sb, LBER_SB_OPT_NEEDS_READ, NULL ) )
1550 slapd_set_read( s, 1 );
1551 if ( ber_sockbuf_ctrl( c->c_sb, LBER_SB_OPT_NEEDS_WRITE, NULL ) )
1552 slapd_set_write( s, 1 );
1553 connection_return( c );
1554 ldap_pvt_thread_mutex_unlock( &connections_mutex );