]> git.sur5r.net Git - openldap/blob - servers/slapd/connection.c
import fix to ITS#3474 from HEAD
[openldap] / servers / slapd / connection.c
1 /* $OpenLDAP$ */
2 /* This work is part of OpenLDAP Software <http://www.openldap.org/>.
3  *
4  * Copyright 1998-2004 The OpenLDAP Foundation.
5  * All rights reserved.
6  *
7  * Redistribution and use in source and binary forms, with or without
8  * modification, are permitted only as authorized by the OpenLDAP
9  * Public License.
10  *
11  * A copy of this license is available in the file LICENSE in the
12  * top-level directory of the distribution or, alternatively, at
13  * <http://www.OpenLDAP.org/license.html>.
14  */
15 /* Portions Copyright (c) 1995 Regents of the University of Michigan.
16  * All rights reserved.
17  *
18  * Redistribution and use in source and binary forms are permitted
19  * provided that this notice is preserved and that due credit is given
20  * to the University of Michigan at Ann Arbor. The name of the University
21  * may not be used to endorse or promote products derived from this
22  * software without specific prior written permission. This software
23  * is provided ``as is'' without express or implied warranty.
24  */
25
26 #include "portable.h"
27
28 #include <stdio.h>
29 #include <limits.h>
30
31 #include <ac/socket.h>
32 #include <ac/errno.h>
33 #include <ac/string.h>
34 #include <ac/time.h>
35 #include <ac/unistd.h>
36
37 #include "ldap_pvt.h"
38 #include "lutil.h"
39 #include "slap.h"
40
41 #ifdef LDAP_SLAPI
42 #include "slapi/slapi.h"
43 #endif
44
45 /* protected by connections_mutex */
46 static ldap_pvt_thread_mutex_t connections_mutex;
47 static Connection *connections = NULL;
48
49 static ldap_pvt_thread_mutex_t conn_nextid_mutex;
50 static unsigned long conn_nextid = 0;
51
52 /* structure state (protected by connections_mutex) */
53 #define SLAP_C_UNINITIALIZED    0x00    /* MUST BE ZERO (0) */
54 #define SLAP_C_UNUSED                   0x01
55 #define SLAP_C_USED                             0x02
56
57 /* connection state (protected by c_mutex ) */
58 #define SLAP_C_INVALID                  0x00    /* MUST BE ZERO (0) */
59 #define SLAP_C_INACTIVE                 0x01    /* zero threads */
60 #define SLAP_C_ACTIVE                   0x02    /* one or more threads */
61 #define SLAP_C_BINDING                  0x03    /* binding */
62 #define SLAP_C_CLOSING                  0x04    /* closing */
63 #define SLAP_C_CLIENT                   0x05    /* outbound client conn */
64
65 const char *
66 connection_state2str( int state )
67 {
68         switch( state ) {
69         case SLAP_C_INVALID:    return "!";
70         case SLAP_C_INACTIVE:   return "|";
71         case SLAP_C_ACTIVE:             return "";
72         case SLAP_C_BINDING:    return "B";
73         case SLAP_C_CLOSING:    return "C";
74         case SLAP_C_CLIENT:             return "L";
75         }
76
77         return "?";
78 }
79
80 static Connection* connection_get( ber_socket_t s );
81
82 static int connection_input( Connection *c );
83 static void connection_close( Connection *c );
84
85 static int connection_op_activate( Operation *op );
86 static int connection_resched( Connection *conn );
87 static void connection_abandon( Connection *conn );
88 static void connection_destroy( Connection *c );
89
90 static ldap_pvt_thread_start_t connection_operation;
91
92 /*
93  * Initialize connection management infrastructure.
94  */
95 int connections_init(void)
96 {
97         int i;
98
99         assert( connections == NULL );
100
101         if( connections != NULL) {
102 #ifdef NEW_LOGGING
103                 LDAP_LOG( CONNECTION, INFO,
104                         "connections_init: already initialized.\n", 0, 0, 0 );
105 #else
106                 Debug( LDAP_DEBUG_ANY, "connections_init: already initialized.\n",
107                         0, 0, 0 );
108 #endif
109                 return -1;
110         }
111
112         /* should check return of every call */
113         ldap_pvt_thread_mutex_init( &connections_mutex );
114         ldap_pvt_thread_mutex_init( &conn_nextid_mutex );
115
116         connections = (Connection *) ch_calloc( dtblsize, sizeof(Connection) );
117
118         if( connections == NULL ) {
119 #ifdef NEW_LOGGING
120                 LDAP_LOG( CONNECTION, ERR,
121                         "connections_init: allocation (%d * %ld) of connection "
122                         "array failed\n", dtblsize, (long) sizeof(Connection), 0 );
123 #else
124                 Debug( LDAP_DEBUG_ANY,
125                         "connections_init: allocation (%d*%ld) of connection array failed\n",
126                         dtblsize, (long) sizeof(Connection), 0 );
127 #endif
128                 return -1;
129         }
130
131         assert( connections[0].c_struct_state == SLAP_C_UNINITIALIZED );
132         assert( connections[dtblsize-1].c_struct_state == SLAP_C_UNINITIALIZED );
133
134         for (i=0; i<dtblsize; i++) connections[i].c_conn_idx = i;
135
136         /*
137          * per entry initialization of the Connection array initialization
138          * will be done by connection_init()
139          */ 
140
141         return 0;
142 }
143
144 /*
145  * Destroy connection management infrastructure.
146  */
147 int connections_destroy(void)
148 {
149         ber_socket_t i;
150
151         /* should check return of every call */
152
153         if( connections == NULL) {
154 #ifdef NEW_LOGGING
155                 LDAP_LOG( CONNECTION, INFO,
156                         "connections_destroy: nothing to destroy.\n", 0, 0, 0 );
157 #else
158                 Debug( LDAP_DEBUG_ANY, "connections_destroy: nothing to destroy.\n",
159                         0, 0, 0 );
160 #endif
161                 return -1;
162         }
163
164         for ( i = 0; i < dtblsize; i++ ) {
165                 if( connections[i].c_struct_state != SLAP_C_UNINITIALIZED ) {
166                         ber_sockbuf_free( connections[i].c_sb );
167                         ldap_pvt_thread_mutex_destroy( &connections[i].c_mutex );
168                         ldap_pvt_thread_mutex_destroy( &connections[i].c_write_mutex );
169                         ldap_pvt_thread_cond_destroy( &connections[i].c_write_cv );
170 #ifdef LDAP_SLAPI
171                         if ( slapi_plugins_used ) {
172                                 slapi_int_free_object_extensions( SLAPI_X_EXT_CONNECTION, &connections[i] );
173                         }
174 #endif
175                 }
176         }
177
178         free( connections );
179         connections = NULL;
180
181         ldap_pvt_thread_mutex_destroy( &connections_mutex );
182         ldap_pvt_thread_mutex_destroy( &conn_nextid_mutex );
183         return 0;
184 }
185
186 /*
187  * shutdown all connections
188  */
189 int connections_shutdown(void)
190 {
191         ber_socket_t i;
192
193         ldap_pvt_thread_mutex_lock( &connections_mutex );
194
195         for ( i = 0; i < dtblsize; i++ ) {
196                 if( connections[i].c_struct_state != SLAP_C_USED ) {
197                         continue;
198                 }
199                 /* give persistent clients a chance to cleanup */
200                 if( connections[i].c_conn_state == SLAP_C_CLIENT ) {
201                         ldap_pvt_thread_pool_submit( &connection_pool,
202                         connections[i].c_clientfunc, connections[i].c_clientarg );
203                         continue;
204                 }
205
206                 ldap_pvt_thread_mutex_lock( &connections[i].c_mutex );
207
208                 /* connections_mutex and c_mutex are locked */
209                 connection_closing( &connections[i] );
210                 connection_close( &connections[i] );
211
212                 ldap_pvt_thread_mutex_unlock( &connections[i].c_mutex );
213         }
214
215         ldap_pvt_thread_mutex_unlock( &connections_mutex );
216
217         return 0;
218 }
219
220 /*
221  * Timeout idle connections.
222  */
223 int connections_timeout_idle(time_t now)
224 {
225         int i = 0;
226         int connindex;
227         Connection* c;
228
229         for( c = connection_first( &connindex );
230                 c != NULL;
231                 c = connection_next( c, &connindex ) )
232         {
233                 /* Don't timeout a slow-running request or a persistent
234                  * outbound connection */
235                 if( c->c_n_ops_executing ||
236                         c->c_conn_state == SLAP_C_CLIENT ) continue;
237
238                 if( difftime( c->c_activitytime+global_idletimeout, now) < 0 ) {
239                         /* close it */
240                         connection_closing( c );
241                         connection_close( c );
242                         i++;
243                 }
244         }
245         connection_done( c );
246
247         return i;
248 }
249
250 static Connection* connection_get( ber_socket_t s )
251 {
252         /* connections_mutex should be locked by caller */
253
254         Connection *c;
255
256 #ifdef NEW_LOGGING
257         LDAP_LOG( CONNECTION, ENTRY, "connection_get: socket %ld\n", (long)s, 0, 0 );
258 #else
259         Debug( LDAP_DEBUG_ARGS,
260                 "connection_get(%ld)\n",
261                 (long) s, 0, 0 );
262 #endif
263
264         assert( connections != NULL );
265
266         if(s == AC_SOCKET_INVALID) {
267                 return NULL;
268         }
269
270 #ifndef HAVE_WINSOCK
271         c = &connections[s];
272
273         assert( c->c_struct_state != SLAP_C_UNINITIALIZED );
274
275 #else
276         c = NULL;
277         {
278                 ber_socket_t i, sd;
279
280                 for(i=0; i<dtblsize; i++) {
281                         if( connections[i].c_struct_state == SLAP_C_UNINITIALIZED ) {
282                                 assert( connections[i].c_conn_state == SLAP_C_INVALID );
283                                 assert( connections[i].c_sb == 0 );
284                                 break;
285                         }
286
287                         ber_sockbuf_ctrl( connections[i].c_sb,
288                                 LBER_SB_OPT_GET_FD, &sd );
289
290                         if( connections[i].c_struct_state == SLAP_C_UNUSED ) {
291                                 assert( connections[i].c_conn_state == SLAP_C_INVALID );
292                                 assert( sd == AC_SOCKET_INVALID );
293                                 continue;
294                         }
295
296                         /* state can actually change from used -> unused by resched,
297                          * so don't assert details here.
298                          */
299
300                         if( sd == s ) {
301                                 c = &connections[i];
302                                 break;
303                         }
304                 }
305         }
306 #endif
307
308         if( c != NULL ) {
309                 ber_socket_t    sd;
310
311                 ldap_pvt_thread_mutex_lock( &c->c_mutex );
312
313                 ber_sockbuf_ctrl( c->c_sb, LBER_SB_OPT_GET_FD, &sd );
314                 if( c->c_struct_state != SLAP_C_USED ) {
315                         /* connection must have been closed due to resched */
316
317                         assert( c->c_conn_state == SLAP_C_INVALID );
318                         assert( sd == AC_SOCKET_INVALID );
319
320 #ifdef NEW_LOGGING
321                         LDAP_LOG( CONNECTION, ARGS, 
322                                 "connection_get: connection %d not used\n", s, 0, 0 );
323 #else
324                         Debug( LDAP_DEBUG_TRACE,
325                                 "connection_get(%d): connection not used\n",
326                                 s, 0, 0 );
327 #endif
328
329                         ldap_pvt_thread_mutex_unlock( &c->c_mutex );
330                         return NULL;
331                 }
332
333 #ifdef NEW_LOGGING
334                 LDAP_LOG( CONNECTION, RESULTS, 
335                         "connection_get: get for %d got connid %lu\n", s, c->c_connid, 0 );
336 #else
337                 Debug( LDAP_DEBUG_TRACE,
338                         "connection_get(%d): got connid=%lu\n",
339                         s, c->c_connid, 0 );
340 #endif
341
342                 c->c_n_get++;
343
344                 assert( c->c_struct_state == SLAP_C_USED );
345                 assert( c->c_conn_state != SLAP_C_INVALID );
346                 assert( sd != AC_SOCKET_INVALID );
347
348 #ifdef SLAPD_MONITOR
349                 c->c_activitytime = slap_get_time();
350 #else
351                 if( global_idletimeout > 0 ) {
352                         c->c_activitytime = slap_get_time();
353                 }
354 #endif
355         }
356
357         return c;
358 }
359
360 static void connection_return( Connection *c )
361 {
362         ldap_pvt_thread_mutex_unlock( &c->c_mutex );
363 }
364
365 long connection_init(
366         ber_socket_t s,
367         Listener *listener,
368         const char* dnsname,
369         const char* peername,
370         int flags,
371         slap_ssf_t ssf,
372         struct berval *authid )
373 {
374         unsigned long id;
375         Connection *c;
376
377         assert( connections != NULL );
378
379         assert( listener != NULL );
380         assert( dnsname != NULL );
381         assert( peername != NULL );
382
383 #ifndef HAVE_TLS
384         assert( flags != CONN_IS_TLS );
385 #endif
386
387         if( s == AC_SOCKET_INVALID ) {
388 #ifdef NEW_LOGGING
389                 LDAP_LOG( CONNECTION, INFO, 
390                         "connection_init: init of socket %ld invalid.\n", (long)s, 0, 0 );
391 #else
392                 Debug( LDAP_DEBUG_ANY,
393                         "connection_init: init of socket %ld invalid.\n", (long)s, 0, 0 );
394 #endif
395                 return -1;
396         }
397
398         assert( s >= 0 );
399 #ifndef HAVE_WINSOCK
400         assert( s < dtblsize );
401 #endif
402
403         ldap_pvt_thread_mutex_lock( &connections_mutex );
404
405 #ifndef HAVE_WINSOCK
406         c = &connections[s];
407
408 #else
409         {
410                 ber_socket_t i;
411                 c = NULL;
412
413                 for( i=0; i < dtblsize; i++) {
414                         ber_socket_t    sd;
415
416                         if( connections[i].c_struct_state == SLAP_C_UNINITIALIZED ) {
417                                 assert( connections[i].c_sb == 0 );
418                                 c = &connections[i];
419                                 break;
420                         }
421
422                         sd = AC_SOCKET_INVALID;
423                         if (connections[i].c_sb != NULL) {
424                                 ber_sockbuf_ctrl( connections[i].c_sb,
425                                         LBER_SB_OPT_GET_FD, &sd );
426                         }
427
428                         if( connections[i].c_struct_state == SLAP_C_UNUSED ) {
429                                 assert( sd == AC_SOCKET_INVALID );
430                                 c = &connections[i];
431                                 break;
432                         }
433
434                         if( connections[i].c_conn_state == SLAP_C_CLIENT ) {
435                                 continue;
436                         }
437
438                         assert( connections[i].c_struct_state == SLAP_C_USED );
439                         assert( connections[i].c_conn_state != SLAP_C_INVALID );
440                         assert( sd != AC_SOCKET_INVALID );
441                 }
442
443                 if( c == NULL ) {
444 #ifdef NEW_LOGGING
445                         LDAP_LOG( CONNECTION, INFO, 
446                                 "connection_init(%d): connection table full "
447                                 "(%d/%d)\n", s, i, dtblsize );
448 #else
449                         Debug( LDAP_DEBUG_ANY,
450                                 "connection_init(%d): connection table full "
451                                 "(%d/%d)\n", s, i, dtblsize);
452 #endif
453                         ldap_pvt_thread_mutex_unlock( &connections_mutex );
454                         return -1;
455                 }
456         }
457 #endif
458
459         assert( c != NULL );
460
461         if( c->c_struct_state == SLAP_C_UNINITIALIZED ) {
462                 c->c_send_ldap_result = slap_send_ldap_result;
463                 c->c_send_search_entry = slap_send_search_entry;
464                 c->c_send_search_reference = slap_send_search_reference;
465                 c->c_send_ldap_extended = slap_send_ldap_extended;
466 #ifdef LDAP_RES_INTERMEDIATE
467                 c->c_send_ldap_intermediate = slap_send_ldap_intermediate;
468 #endif
469
470                 c->c_authmech.bv_val = NULL;
471                 c->c_authmech.bv_len = 0;
472                 c->c_dn.bv_val = NULL;
473                 c->c_dn.bv_len = 0;
474                 c->c_ndn.bv_val = NULL;
475                 c->c_ndn.bv_len = 0;
476
477                 c->c_listener = NULL;
478                 c->c_peer_domain.bv_val = NULL;
479                 c->c_peer_domain.bv_len = 0;
480                 c->c_peer_name.bv_val = NULL;
481                 c->c_peer_name.bv_len = 0;
482
483                 LDAP_STAILQ_INIT(&c->c_ops);
484                 LDAP_STAILQ_INIT(&c->c_pending_ops);
485
486                 c->c_sasl_bind_mech.bv_val = NULL;
487                 c->c_sasl_bind_mech.bv_len = 0;
488                 c->c_sasl_done = 0;
489                 c->c_sasl_authctx = NULL;
490                 c->c_sasl_sockctx = NULL;
491                 c->c_sasl_extra = NULL;
492                 c->c_sasl_bindop = NULL;
493
494                 c->c_sb = ber_sockbuf_alloc( );
495
496                 {
497                         ber_len_t max = sockbuf_max_incoming;
498                         ber_sockbuf_ctrl( c->c_sb, LBER_SB_OPT_SET_MAX_INCOMING, &max );
499                 }
500
501                 c->c_currentber = NULL;
502
503                 /* should check status of thread calls */
504                 ldap_pvt_thread_mutex_init( &c->c_mutex );
505                 ldap_pvt_thread_mutex_init( &c->c_write_mutex );
506                 ldap_pvt_thread_cond_init( &c->c_write_cv );
507
508 #ifdef LDAP_SLAPI
509                 if ( slapi_plugins_used ) {
510                         slapi_int_create_object_extensions( SLAPI_X_EXT_CONNECTION, c );
511                 }
512 #endif
513
514                 c->c_struct_state = SLAP_C_UNUSED;
515         }
516
517         ldap_pvt_thread_mutex_lock( &c->c_mutex );
518
519         assert( c->c_struct_state == SLAP_C_UNUSED );
520         assert( c->c_authmech.bv_val == NULL );
521         assert( c->c_dn.bv_val == NULL );
522         assert( c->c_ndn.bv_val == NULL );
523         assert( c->c_listener == NULL );
524         assert( c->c_peer_domain.bv_val == NULL );
525         assert( c->c_peer_name.bv_val == NULL );
526         assert( LDAP_STAILQ_EMPTY(&c->c_ops) );
527         assert( LDAP_STAILQ_EMPTY(&c->c_pending_ops) );
528         assert( c->c_sasl_bind_mech.bv_val == NULL );
529         assert( c->c_sasl_done == 0 );
530         assert( c->c_sasl_authctx == NULL );
531         assert( c->c_sasl_sockctx == NULL );
532         assert( c->c_sasl_extra == NULL );
533         assert( c->c_sasl_bindop == NULL );
534         assert( c->c_currentber == NULL );
535         assert( c->c_writewaiter == 0);
536
537         c->c_listener = listener;
538
539         if ( flags == CONN_IS_CLIENT ) {
540                 c->c_conn_state = SLAP_C_CLIENT;
541                 c->c_struct_state = SLAP_C_USED;
542                 ber_sockbuf_ctrl( c->c_sb, LBER_SB_OPT_SET_FD, &s );
543                 ldap_pvt_thread_mutex_unlock( &c->c_mutex );
544                 ldap_pvt_thread_mutex_unlock( &connections_mutex );
545
546                 return 0;
547         }
548
549         ber_str2bv( dnsname, 0, 1, &c->c_peer_domain );
550         ber_str2bv( peername, 0, 1, &c->c_peer_name );
551
552         c->c_n_ops_received = 0;
553         c->c_n_ops_executing = 0;
554         c->c_n_ops_pending = 0;
555         c->c_n_ops_completed = 0;
556
557         c->c_n_get = 0;
558         c->c_n_read = 0;
559         c->c_n_write = 0;
560
561         /* set to zero until bind, implies LDAP_VERSION3 */
562         c->c_protocol = 0;
563
564 #ifdef SLAPD_MONITOR
565         c->c_activitytime = c->c_starttime = slap_get_time();
566 #else
567         if( global_idletimeout > 0 ) {
568                 c->c_activitytime = c->c_starttime = slap_get_time();
569         }
570 #endif
571
572 #ifdef LDAP_CONNECTIONLESS
573         c->c_is_udp = 0;
574         if( flags == CONN_IS_UDP ) {
575                 c->c_is_udp = 1;
576 #ifdef LDAP_DEBUG
577                 ber_sockbuf_add_io( c->c_sb, &ber_sockbuf_io_debug,
578                         LBER_SBIOD_LEVEL_PROVIDER, (void*)"udp_" );
579 #endif
580                 ber_sockbuf_add_io( c->c_sb, &ber_sockbuf_io_udp,
581                         LBER_SBIOD_LEVEL_PROVIDER, (void *)&s );
582                 ber_sockbuf_add_io( c->c_sb, &ber_sockbuf_io_readahead,
583                         LBER_SBIOD_LEVEL_PROVIDER, NULL );
584         } else
585 #endif
586         {
587 #ifdef LDAP_DEBUG
588                 ber_sockbuf_add_io( c->c_sb, &ber_sockbuf_io_debug,
589                         LBER_SBIOD_LEVEL_PROVIDER, (void*)"tcp_" );
590 #endif
591                 ber_sockbuf_add_io( c->c_sb, &ber_sockbuf_io_tcp,
592                         LBER_SBIOD_LEVEL_PROVIDER, (void *)&s );
593         }
594
595 #ifdef LDAP_DEBUG
596         ber_sockbuf_add_io( c->c_sb, &ber_sockbuf_io_debug,
597                 INT_MAX, (void*)"ldap_" );
598 #endif
599
600         if( ber_sockbuf_ctrl( c->c_sb, LBER_SB_OPT_SET_NONBLOCK,
601                 c /* non-NULL */ ) < 0 )
602         {
603 #ifdef NEW_LOGGING
604                 LDAP_LOG( CONNECTION, INFO, 
605                         "connection_init: conn %lu set nonblocking failed\n",
606                         c->c_connid, 0, 0 );
607 #else
608                 Debug( LDAP_DEBUG_ANY,
609                         "connection_init(%d, %s): set nonblocking failed\n",
610                         s, c->c_peer_name.bv_val, 0 );
611 #endif
612         }
613
614         ldap_pvt_thread_mutex_lock( &conn_nextid_mutex );
615         id = c->c_connid = conn_nextid++;
616         ldap_pvt_thread_mutex_unlock( &conn_nextid_mutex );
617
618         c->c_conn_state = SLAP_C_INACTIVE;
619         c->c_struct_state = SLAP_C_USED;
620
621         c->c_ssf = c->c_transport_ssf = ssf;
622         c->c_tls_ssf = 0;
623
624 #ifdef HAVE_TLS
625         if ( flags == CONN_IS_TLS ) {
626                 c->c_is_tls = 1;
627                 c->c_needs_tls_accept = 1;
628         } else {
629                 c->c_is_tls = 0;
630                 c->c_needs_tls_accept = 0;
631         }
632 #endif
633
634         slap_sasl_open( c, 0 );
635         slap_sasl_external( c, ssf, authid );
636
637         ldap_pvt_thread_mutex_unlock( &c->c_mutex );
638         ldap_pvt_thread_mutex_unlock( &connections_mutex );
639
640         backend_connection_init(c);
641
642         return id;
643 }
644
645 void connection2anonymous( Connection *c )
646 {
647         assert( connections != NULL );
648         assert( c != NULL );
649
650         {
651                 ber_len_t max = sockbuf_max_incoming;
652                 ber_sockbuf_ctrl( c->c_sb, LBER_SB_OPT_SET_MAX_INCOMING, &max );
653         }
654
655         if(c->c_authmech.bv_val != NULL ) {
656                 free(c->c_authmech.bv_val);
657                 c->c_authmech.bv_val = NULL;
658         }
659         c->c_authmech.bv_len = 0;
660
661         if(c->c_dn.bv_val != NULL) {
662                 free(c->c_dn.bv_val);
663                 c->c_dn.bv_val = NULL;
664         }
665         c->c_dn.bv_len = 0;
666         if(c->c_ndn.bv_val != NULL) {
667                 free(c->c_ndn.bv_val);
668                 c->c_ndn.bv_val = NULL;
669         }
670         c->c_ndn.bv_len = 0;
671
672         c->c_authz_backend = NULL;
673 }
674
675 static void
676 connection_destroy( Connection *c )
677 {
678         /* note: connections_mutex should be locked by caller */
679         ber_socket_t    sd;
680         unsigned long   connid;
681
682         assert( connections != NULL );
683         assert( c != NULL );
684         assert( c->c_struct_state != SLAP_C_UNUSED );
685         assert( c->c_conn_state != SLAP_C_INVALID );
686         assert( LDAP_STAILQ_EMPTY(&c->c_ops) );
687         assert( c->c_writewaiter == 0);
688
689         /* only for stats (print -1 as "%lu" may give unexpected results ;) */
690         connid = c->c_connid;
691
692         backend_connection_destroy(c);
693
694         c->c_protocol = 0;
695         c->c_connid = -1;
696
697         c->c_activitytime = c->c_starttime = 0;
698
699         connection2anonymous( c );
700         c->c_listener = NULL;
701
702         if(c->c_peer_domain.bv_val != NULL) {
703                 free(c->c_peer_domain.bv_val);
704                 c->c_peer_domain.bv_val = NULL;
705         }
706         c->c_peer_domain.bv_len = 0;
707         if(c->c_peer_name.bv_val != NULL) {
708                 free(c->c_peer_name.bv_val);
709                 c->c_peer_name.bv_val = NULL;
710         }
711         c->c_peer_name.bv_len = 0;
712
713         c->c_sasl_bind_in_progress = 0;
714         if(c->c_sasl_bind_mech.bv_val != NULL) {
715                 free(c->c_sasl_bind_mech.bv_val);
716                 c->c_sasl_bind_mech.bv_val = NULL;
717         }
718         c->c_sasl_bind_mech.bv_len = 0;
719
720         slap_sasl_close( c );
721
722         if ( c->c_currentber != NULL ) {
723                 ber_free( c->c_currentber, 1 );
724                 c->c_currentber = NULL;
725         }
726
727         ber_sockbuf_ctrl( c->c_sb, LBER_SB_OPT_GET_FD, &sd );
728         if ( sd != AC_SOCKET_INVALID ) {
729                 slapd_remove( sd, 1, 0 );
730
731                 Statslog( LDAP_DEBUG_STATS,
732                         "conn=%lu fd=%ld closed\n",
733                         connid, (long) sd, 0, 0, 0 );
734         }
735
736         ber_sockbuf_free( c->c_sb );
737
738         c->c_sb = ber_sockbuf_alloc( );
739
740         {
741                 ber_len_t max = sockbuf_max_incoming;
742                 ber_sockbuf_ctrl( c->c_sb, LBER_SB_OPT_SET_MAX_INCOMING, &max );
743         }
744
745         c->c_conn_state = SLAP_C_INVALID;
746         c->c_struct_state = SLAP_C_UNUSED;
747
748 #ifdef LDAP_SLAPI
749         /* call destructors, then constructors; avoids unnecessary allocation */
750         if ( slapi_plugins_used ) {
751                 slapi_int_clear_object_extensions( SLAPI_X_EXT_CONNECTION, c );
752         }
753 #endif
754 }
755
756 int connection_state_closing( Connection *c )
757 {
758         /* c_mutex must be locked by caller */
759
760         int state;
761         assert( c != NULL );
762         assert( c->c_struct_state == SLAP_C_USED );
763
764         state = c->c_conn_state;
765
766         assert( state != SLAP_C_INVALID );
767
768         return state == SLAP_C_CLOSING;
769 }
770
771 static void connection_abandon( Connection *c )
772 {
773         /* c_mutex must be locked by caller */
774
775         Operation *o, *next, op = {0};
776         SlapReply rs = {0};
777
778         op.o_conn = c;
779         op.o_connid = c->c_connid;
780         op.o_tag = LDAP_REQ_ABANDON;
781         for ( o = LDAP_STAILQ_FIRST( &c->c_ops ); o; o=next ) {
782                 next = LDAP_STAILQ_NEXT( o, o_next );
783                 op.orn_msgid = o->o_msgid;
784                 o->o_abandon = 1;
785                 fe_op_abandon( &op, &rs );
786         }
787
788         /* remove pending operations */
789         while ( (o = LDAP_STAILQ_FIRST( &c->c_pending_ops )) != NULL) {
790                 LDAP_STAILQ_REMOVE_HEAD( &c->c_pending_ops, o_next );
791                 LDAP_STAILQ_NEXT(o, o_next) = NULL;
792                 slap_op_free( o );
793         }
794 }
795
796 void connection_closing( Connection *c )
797 {
798         assert( connections != NULL );
799         assert( c != NULL );
800         assert( c->c_struct_state == SLAP_C_USED );
801         assert( c->c_conn_state != SLAP_C_INVALID );
802
803         /* c_mutex must be locked by caller */
804
805         if( c->c_conn_state != SLAP_C_CLOSING ) {
806                 ber_socket_t    sd;
807
808                 ber_sockbuf_ctrl( c->c_sb, LBER_SB_OPT_GET_FD, &sd );
809 #ifdef NEW_LOGGING
810                 LDAP_LOG( CONNECTION, DETAIL1, 
811                         "connection_closing: conn %lu readying socket %d for close.\n",
812                         c->c_connid, sd, 0 );
813 #else
814                 Debug( LDAP_DEBUG_TRACE,
815                         "connection_closing: readying conn=%lu sd=%d for close\n",
816                         c->c_connid, sd, 0 );
817 #endif
818                 /* update state to closing */
819                 c->c_conn_state = SLAP_C_CLOSING;
820
821                 /* don't listen on this port anymore */
822                 slapd_clr_read( sd, 1 );
823
824                 /* abandon active operations */
825                 connection_abandon( c );
826
827                 /* wake write blocked operations */
828                 slapd_clr_write( sd, 1 );
829                 ldap_pvt_thread_cond_signal( &c->c_write_cv );
830         }
831 }
832
833 static void connection_close( Connection *c )
834 {
835         ber_socket_t    sd;
836
837         assert( connections != NULL );
838         assert( c != NULL );
839         assert( c->c_struct_state == SLAP_C_USED );
840         assert( c->c_conn_state == SLAP_C_CLOSING );
841
842         /* note: connections_mutex and c_mutex should be locked by caller */
843
844         ber_sockbuf_ctrl( c->c_sb, LBER_SB_OPT_GET_FD, &sd );
845         if( !LDAP_STAILQ_EMPTY(&c->c_ops) ) {
846 #ifdef NEW_LOGGING
847                 LDAP_LOG( CONNECTION, DETAIL1, 
848                         "connection_close: conn %lu deferring sd %d\n",
849                                 c->c_connid, sd, 0 );
850 #else
851                 Debug( LDAP_DEBUG_TRACE,
852                         "connection_close: deferring conn=%lu sd=%d\n",
853                         c->c_connid, sd, 0 );
854 #endif
855                 return;
856         }
857
858 #ifdef NEW_LOGGING
859         LDAP_LOG( CONNECTION, RESULTS, 
860                 "connection_close: conn %lu sd %d\n", c->c_connid, sd, 0 );
861 #else
862         Debug( LDAP_DEBUG_TRACE, "connection_close: conn=%lu sd=%d\n",
863                 c->c_connid, sd, 0 );
864 #endif
865         connection_destroy( c );
866 }
867
868 unsigned long connections_nextid(void)
869 {
870         unsigned long id;
871         assert( connections != NULL );
872
873         ldap_pvt_thread_mutex_lock( &connections_mutex );
874
875         id = conn_nextid;
876
877         ldap_pvt_thread_mutex_unlock( &connections_mutex );
878
879         return id;
880 }
881
882 Connection* connection_first( ber_socket_t *index )
883 {
884         assert( connections != NULL );
885         assert( index != NULL );
886
887         ldap_pvt_thread_mutex_lock( &connections_mutex );
888
889         *index = 0;
890
891         return connection_next(NULL, index);
892 }
893
894 Connection* connection_next( Connection *c, ber_socket_t *index )
895 {
896         assert( connections != NULL );
897         assert( index != NULL );
898         assert( *index <= dtblsize );
899
900         if( c != NULL ) {
901                 ldap_pvt_thread_mutex_unlock( &c->c_mutex );
902         }
903
904         c = NULL;
905
906         for(; *index < dtblsize; (*index)++) {
907                 if( connections[*index].c_struct_state == SLAP_C_UNINITIALIZED ) {
908                         assert( connections[*index].c_conn_state == SLAP_C_INVALID );
909 #ifndef HAVE_WINSOCK
910                         continue;
911 #else
912                         break;
913 #endif
914                 }
915
916                 if( connections[*index].c_struct_state == SLAP_C_USED ) {
917                         assert( connections[*index].c_conn_state != SLAP_C_INVALID );
918                         c = &connections[(*index)++];
919                         break;
920                 }
921
922                 assert( connections[*index].c_struct_state == SLAP_C_UNUSED );
923                 assert( connections[*index].c_conn_state == SLAP_C_INVALID );
924         }
925
926         if( c != NULL ) {
927                 ldap_pvt_thread_mutex_lock( &c->c_mutex );
928         }
929
930         return c;
931 }
932
933 void connection_done( Connection *c )
934 {
935         assert( connections != NULL );
936
937         if( c != NULL ) {
938                 ldap_pvt_thread_mutex_unlock( &c->c_mutex );
939         }
940
941         ldap_pvt_thread_mutex_unlock( &connections_mutex );
942 }
943
944 /*
945  * connection_activity - handle the request operation op on connection
946  * conn.  This routine figures out what kind of operation it is and
947  * calls the appropriate stub to handle it.
948  */
949
950 #ifdef SLAPD_MONITOR
951 #define INCR_OP(var,index) \
952         do { \
953                 ldap_pvt_thread_mutex_lock( &num_ops_mutex ); \
954                 (var)[(index)]++; \
955                 ldap_pvt_thread_mutex_unlock( &num_ops_mutex ); \
956         } while (0)
957 #else /* !SLAPD_MONITOR */
958 #define INCR_OP(var,index) 
959 #endif /* !SLAPD_MONITOR */
960
961 static void *
962 connection_operation( void *ctx, void *arg_v )
963 {
964         int rc = LDAP_OTHER;
965         Operation *op = arg_v;
966         SlapReply rs = {REP_RESULT};
967         ber_tag_t tag = op->o_tag;
968 #ifdef SLAPD_MONITOR
969         ber_tag_t oldtag = tag;
970 #endif /* SLAPD_MONITOR */
971         Connection *conn = op->o_conn;
972         void *memctx = NULL;
973         void *memctx_null = NULL;
974         ber_len_t memsiz;
975
976         ldap_pvt_thread_mutex_lock( &num_ops_mutex );
977         num_ops_initiated++;
978         ldap_pvt_thread_mutex_unlock( &num_ops_mutex );
979
980         op->o_threadctx = ctx;
981
982         switch ( tag ) {
983         case LDAP_REQ_BIND:
984         case LDAP_REQ_UNBIND:
985         case LDAP_REQ_ADD:
986         case LDAP_REQ_DELETE:
987         case LDAP_REQ_MODRDN:
988         case LDAP_REQ_MODIFY:
989         case LDAP_REQ_COMPARE:
990         case LDAP_REQ_SEARCH:
991         case LDAP_REQ_ABANDON:
992         case LDAP_REQ_EXTENDED:
993                 break;
994         default:
995 #ifdef NEW_LOGGING
996                 LDAP_LOG( CONNECTION, INFO, "connection_operation: "
997                         "conn %lu unknown LDAP request 0x%lx\n",
998                         conn->c_connid, tag, 0 );
999 #else
1000                 Debug( LDAP_DEBUG_ANY, "connection_operation: "
1001                         "conn %lu unknown LDAP request 0x%lx\n",
1002                         conn->c_connid, tag, 0 );
1003 #endif
1004                 op->o_tag = LBER_ERROR;
1005                 rs.sr_err = LDAP_PROTOCOL_ERROR;
1006                 rs.sr_text = "unknown LDAP request";
1007                 send_ldap_disconnect( op, &rs );
1008                 rc = SLAPD_DISCONNECT;
1009                 goto operations_error;
1010         }
1011
1012         if( conn->c_sasl_bind_in_progress && tag != LDAP_REQ_BIND ) {
1013 #ifdef NEW_LOGGING
1014                 LDAP_LOG( CONNECTION, ERR, 
1015                         "connection_operation: conn %lu SASL bind in progress (tag=%ld).\n",
1016                         conn->c_connid, (long)tag, 0 );
1017 #else
1018                 Debug( LDAP_DEBUG_ANY, "connection_operation: "
1019                         "error: SASL bind in progress (tag=%ld).\n",
1020                         (long) tag, 0, 0 );
1021 #endif
1022                 send_ldap_error( op, &rs, LDAP_OPERATIONS_ERROR,
1023                         "SASL bind in progress" );
1024                 rc = LDAP_OPERATIONS_ERROR;
1025                 goto operations_error;
1026         }
1027
1028         /* We can use Thread-Local storage for most mallocs. We can
1029          * also use TL for ber parsing, but not on Add or Modify.
1030          */
1031 #if 0
1032         memsiz = ber_len( op->o_ber ) * 64;
1033         if ( SLMALLOC_SLAB_SIZE > memsiz ) memsiz = SLMALLOC_SLAB_SIZE;
1034 #endif
1035         memsiz = SLMALLOC_SLAB_SIZE;
1036
1037         memctx = sl_mem_create( memsiz, ctx );
1038         op->o_tmpmemctx = memctx;
1039         op->o_tmpmfuncs = &sl_mfuncs;
1040         if ( tag != LDAP_REQ_ADD && tag != LDAP_REQ_MODIFY ) {
1041                 /* Note - the ber and its buffer are already allocated from
1042                  * regular memory; this only affects subsequent mallocs that
1043                  * ber_scanf may invoke.
1044                  */
1045                 ber_set_option( op->o_ber, LBER_OPT_BER_MEMCTX, &memctx );
1046         }
1047
1048         switch ( tag ) {
1049         case LDAP_REQ_BIND:
1050                 INCR_OP(num_ops_initiated_, SLAP_OP_BIND);
1051                 rc = do_bind( op, &rs );
1052                 break;
1053
1054         case LDAP_REQ_UNBIND:
1055                 INCR_OP(num_ops_initiated_, SLAP_OP_UNBIND);
1056                 rc = do_unbind( op, &rs );
1057                 break;
1058
1059         case LDAP_REQ_ADD:
1060                 INCR_OP(num_ops_initiated_, SLAP_OP_ADD);
1061                 rc = do_add( op, &rs );
1062                 break;
1063
1064         case LDAP_REQ_DELETE:
1065                 INCR_OP(num_ops_initiated_, SLAP_OP_DELETE);
1066                 rc = do_delete( op, &rs );
1067                 break;
1068
1069         case LDAP_REQ_MODRDN:
1070                 INCR_OP(num_ops_initiated_, SLAP_OP_MODRDN);
1071                 rc = do_modrdn( op, &rs );
1072                 break;
1073
1074         case LDAP_REQ_MODIFY:
1075                 INCR_OP(num_ops_initiated_, SLAP_OP_MODIFY);
1076                 rc = do_modify( op, &rs );
1077                 break;
1078
1079         case LDAP_REQ_COMPARE:
1080                 INCR_OP(num_ops_initiated_, SLAP_OP_COMPARE);
1081                 rc = do_compare( op, &rs );
1082                 break;
1083
1084         case LDAP_REQ_SEARCH:
1085                 INCR_OP(num_ops_initiated_, SLAP_OP_SEARCH);
1086                 rc = do_search( op, &rs );
1087                 break;
1088
1089         case LDAP_REQ_ABANDON:
1090                 INCR_OP(num_ops_initiated_, SLAP_OP_ABANDON);
1091                 rc = do_abandon( op, &rs );
1092                 break;
1093
1094         case LDAP_REQ_EXTENDED:
1095                 INCR_OP(num_ops_initiated_, SLAP_OP_EXTENDED);
1096                 rc = do_extended( op, &rs );
1097                 break;
1098
1099         default:
1100                 /* not reachable */
1101                 assert( 0 );
1102         }
1103
1104 operations_error:
1105         if( rc == SLAPD_DISCONNECT ) tag = LBER_ERROR;
1106
1107         ldap_pvt_thread_mutex_lock( &num_ops_mutex );
1108
1109         num_ops_completed++;
1110 #ifdef SLAPD_MONITOR
1111         switch (oldtag) {
1112         case LDAP_REQ_BIND:
1113                 num_ops_completed_[SLAP_OP_BIND]++;
1114                 break;
1115         case LDAP_REQ_UNBIND:
1116                 num_ops_completed_[SLAP_OP_UNBIND]++;
1117                 break;
1118         case LDAP_REQ_ADD:
1119                 num_ops_completed_[SLAP_OP_ADD]++;
1120                 break;
1121         case LDAP_REQ_DELETE:
1122                 num_ops_completed_[SLAP_OP_DELETE]++;
1123                 break;
1124         case LDAP_REQ_MODRDN:
1125                 num_ops_completed_[SLAP_OP_MODRDN]++;
1126                 break;
1127         case LDAP_REQ_MODIFY:
1128                 num_ops_completed_[SLAP_OP_MODIFY]++;
1129                 break;
1130         case LDAP_REQ_COMPARE:
1131                 num_ops_completed_[SLAP_OP_COMPARE]++;
1132                 break;
1133         case LDAP_REQ_SEARCH:
1134                 num_ops_completed_[SLAP_OP_SEARCH]++;
1135                 break;
1136         case LDAP_REQ_ABANDON:
1137                 num_ops_completed_[SLAP_OP_ABANDON]++;
1138                 break;
1139         case LDAP_REQ_EXTENDED:
1140                 num_ops_completed_[SLAP_OP_EXTENDED]++;
1141                 break;
1142         default:
1143                 /* this is reachable */
1144                 break;
1145         }
1146 #endif /* SLAPD_MONITOR */
1147         ldap_pvt_thread_mutex_unlock( &num_ops_mutex );
1148
1149         if ( op->o_cancel == SLAP_CANCEL_REQ ) {
1150                 op->o_cancel = LDAP_TOO_LATE;
1151         }
1152         while ( op->o_cancel != SLAP_CANCEL_NONE &&
1153                 op->o_cancel != SLAP_CANCEL_DONE )
1154         {
1155                 ldap_pvt_thread_yield();
1156         }
1157
1158         ldap_pvt_thread_mutex_lock( &conn->c_mutex );
1159
1160         ber_set_option( op->o_ber, LBER_OPT_BER_MEMCTX, &memctx_null );
1161
1162         if ( op->o_cancel != SLAP_CANCEL_ACK &&
1163                 ( op->o_sync_mode & SLAP_SYNC_PERSIST ) )
1164         {
1165                 sl_mem_detach( ctx, memctx );
1166
1167         } else if ( op->o_sync_slog_size != -1 ) {
1168                 sl_mem_detach( ctx, memctx );
1169                 LDAP_STAILQ_REMOVE( &conn->c_ops, op, slap_op, o_next);
1170                 LDAP_STAILQ_NEXT(op, o_next) = NULL;
1171                 conn->c_n_ops_executing--;
1172                 conn->c_n_ops_completed++;
1173
1174         } else {
1175                 LDAP_STAILQ_REMOVE( &conn->c_ops, op, slap_op, o_next);
1176                 LDAP_STAILQ_NEXT(op, o_next) = NULL;
1177                 slap_op_free( op );
1178                 conn->c_n_ops_executing--;
1179                 conn->c_n_ops_completed++;
1180         }
1181
1182         switch( tag ) {
1183         case LBER_ERROR:
1184         case LDAP_REQ_UNBIND:
1185                 /* c_mutex is locked */
1186                 connection_closing( conn );
1187                 break;
1188
1189         case LDAP_REQ_BIND:
1190                 conn->c_sasl_bind_in_progress =
1191                         rc == LDAP_SASL_BIND_IN_PROGRESS ? 1 : 0;
1192
1193                 if( conn->c_conn_state == SLAP_C_BINDING) {
1194                         conn->c_conn_state = SLAP_C_ACTIVE;
1195                 }
1196         }
1197
1198         connection_resched( conn );
1199         ldap_pvt_thread_mutex_unlock( &conn->c_mutex );
1200         return NULL;
1201 }
1202
1203 static const Listener dummy_list = { BER_BVC(""), BER_BVC("") };
1204
1205 int connection_client_setup(
1206         ber_socket_t s,
1207         ldap_pvt_thread_start_t *func,
1208         void *arg )
1209 {
1210         int rc;
1211         Connection *c;
1212
1213         rc = connection_init( s, (Listener *)&dummy_list, "", "",
1214                 CONN_IS_CLIENT, 0, NULL );
1215         if ( rc < 0 ) return -1;
1216
1217         c = connection_get( s );
1218         c->c_clientfunc = func;
1219         c->c_clientarg = arg;
1220         connection_return( c );
1221         slapd_add_internal( s, 0 );
1222         slapd_set_read( s, 1 );
1223         return 0;
1224 }
1225
1226 void connection_client_enable(
1227         ber_socket_t s )
1228 {
1229         slapd_set_read( s, 1 );
1230 }
1231
1232 void connection_client_stop(
1233         ber_socket_t s )
1234 {
1235         Connection *c;
1236
1237         /* get (locked) connection */
1238         c = connection_get( s );
1239         
1240         assert( c->c_conn_state == SLAP_C_CLIENT );
1241
1242         c->c_listener = NULL;
1243         c->c_conn_state = SLAP_C_INVALID;
1244         c->c_struct_state = SLAP_C_UNUSED;
1245         connection_return( c );
1246         slapd_remove( s, 0, 1 );
1247 }
1248
1249 int connection_read(ber_socket_t s)
1250 {
1251         int rc = 0;
1252         Connection *c;
1253
1254         assert( connections != NULL );
1255
1256         ldap_pvt_thread_mutex_lock( &connections_mutex );
1257
1258         /* get (locked) connection */
1259         c = connection_get( s );
1260
1261         if( c == NULL ) {
1262 #ifdef NEW_LOGGING
1263                 LDAP_LOG( CONNECTION, INFO, 
1264                         "connection_read: sock %ld no connection\n", (long)s, 0, 0 );
1265 #else
1266                 Debug( LDAP_DEBUG_ANY,
1267                         "connection_read(%ld): no connection!\n",
1268                         (long) s, 0, 0 );
1269 #endif
1270                 slapd_remove(s, 1, 0);
1271
1272                 ldap_pvt_thread_mutex_unlock( &connections_mutex );
1273                 return -1;
1274         }
1275
1276         c->c_n_read++;
1277
1278         if( c->c_conn_state == SLAP_C_CLOSING ) {
1279 #ifdef NEW_LOGGING
1280                 LDAP_LOG( CONNECTION, INFO, 
1281                         "connection_read: conn %lu connection closing, ignoring input\n",
1282                         c->c_connid, 0, 0 );
1283 #else
1284                 Debug( LDAP_DEBUG_TRACE,
1285                         "connection_read(%d): closing, ignoring input for id=%lu\n",
1286                         s, c->c_connid, 0 );
1287 #endif
1288                 connection_return( c );
1289                 ldap_pvt_thread_mutex_unlock( &connections_mutex );
1290                 return 0;
1291         }
1292
1293         if ( c->c_conn_state == SLAP_C_CLIENT ) {
1294                 slapd_clr_read( s, 0 );
1295                 ldap_pvt_thread_pool_submit( &connection_pool,
1296                         c->c_clientfunc, c->c_clientarg );
1297                 connection_return( c );
1298                 ldap_pvt_thread_mutex_unlock( &connections_mutex );
1299                 return 0;
1300         }
1301
1302 #ifdef NEW_LOGGING
1303         LDAP_LOG( CONNECTION, DETAIL1, 
1304                 "connection_read: conn %lu checking for input.\n", 
1305                         c->c_connid, 0, 0 );
1306 #else
1307         Debug( LDAP_DEBUG_TRACE,
1308                 "connection_read(%d): checking for input on id=%lu\n",
1309                 s, c->c_connid, 0 );
1310 #endif
1311
1312 #ifdef HAVE_TLS
1313         if ( c->c_is_tls && c->c_needs_tls_accept ) {
1314                 rc = ldap_pvt_tls_accept( c->c_sb, slap_tls_ctx );
1315                 if ( rc < 0 ) {
1316 #if 0 /* required by next #if 0 */
1317                         struct timeval tv;
1318                         fd_set rfd;
1319 #endif
1320
1321 #ifdef NEW_LOGGING
1322                         LDAP_LOG( CONNECTION, ERR, 
1323                                 "connection_read: conn %lu TLS accept error, error %d\n",
1324                                 c->c_connid, rc, 0 );
1325 #else
1326                         Debug( LDAP_DEBUG_TRACE,
1327                                 "connection_read(%d): TLS accept error "
1328                                 "error=%d id=%lu, closing\n",
1329                                 s, rc, c->c_connid );
1330 #endif
1331                         c->c_needs_tls_accept = 0;
1332                         /* connections_mutex and c_mutex are locked */
1333                         connection_closing( c );
1334
1335 #if 0
1336                         /* Drain input before close, to allow SSL error codes
1337                          * to propagate to client. */
1338                         FD_ZERO(&rfd);
1339                         FD_SET(s, &rfd);
1340                         for (rc=1; rc>0;) {
1341                                 tv.tv_sec = 1;
1342                                 tv.tv_usec = 0;
1343                                 rc = select(s+1, &rfd, NULL, NULL, &tv);
1344                                 if (rc == 1) {
1345                                         ber_sockbuf_ctrl( c->c_sb, LBER_SB_OPT_DRAIN, NULL);
1346                                 }
1347                         }
1348 #endif
1349                         connection_close( c );
1350
1351                 } else if ( rc == 0 ) {
1352                         void *ssl;
1353                         struct berval authid = BER_BVNULL;
1354
1355                         c->c_needs_tls_accept = 0;
1356
1357                         /* we need to let SASL know */
1358                         ssl = ldap_pvt_tls_sb_ctx( c->c_sb );
1359
1360                         c->c_tls_ssf = (slap_ssf_t) ldap_pvt_tls_get_strength( ssl );
1361                         if( c->c_tls_ssf > c->c_ssf ) {
1362                                 c->c_ssf = c->c_tls_ssf;
1363                         }
1364
1365                         rc = dnX509peerNormalize( ssl, &authid );
1366                         if ( rc != LDAP_SUCCESS ) {
1367 #ifdef NEW_LOGGING
1368                                 LDAP_LOG( CONNECTION, INFO, "connection_read: "
1369                                         "conn %lu unable to get TLS client DN, error %d\n",
1370                                         c->c_connid, rc, 0 );
1371 #else
1372                                 Debug( LDAP_DEBUG_TRACE, "connection_read(%d): "
1373                                         "unable to get TLS client DN, error=%d id=%lu\n",
1374                                         s, rc, c->c_connid );
1375 #endif
1376                         }
1377                         slap_sasl_external( c, c->c_tls_ssf, &authid );
1378                         if ( authid.bv_val ) free( authid.bv_val );
1379                 }
1380
1381                 /* if success and data is ready, fall thru to data input loop */
1382                 if( rc != 0 ||
1383                         !ber_sockbuf_ctrl( c->c_sb, LBER_SB_OPT_DATA_READY, NULL ) )
1384                 {
1385                         connection_return( c );
1386                         ldap_pvt_thread_mutex_unlock( &connections_mutex );
1387                         return 0;
1388                 }
1389         }
1390 #endif
1391
1392 #ifdef HAVE_CYRUS_SASL
1393         if ( c->c_sasl_layers ) {
1394                 /* If previous layer is not removed yet, give up for now */
1395                 if ( !c->c_sasl_sockctx ) {
1396                         connection_return( c );
1397                         ldap_pvt_thread_mutex_unlock( &connections_mutex );
1398                         return 0;
1399                 }
1400
1401                 c->c_sasl_layers = 0;
1402
1403                 rc = ldap_pvt_sasl_install( c->c_sb, c->c_sasl_sockctx );
1404
1405                 if( rc != LDAP_SUCCESS ) {
1406 #ifdef NEW_LOGGING
1407                         LDAP_LOG( CONNECTION, ERR, 
1408                                 "connection_read: conn %lu SASL install error %d, closing\n",
1409                                 c->c_connid, rc, 0 );
1410 #else
1411                         Debug( LDAP_DEBUG_TRACE,
1412                                 "connection_read(%d): SASL install error "
1413                                 "error=%d id=%lu, closing\n",
1414                                 s, rc, c->c_connid );
1415 #endif
1416                         /* connections_mutex and c_mutex are locked */
1417                         connection_closing( c );
1418                         connection_close( c );
1419                         connection_return( c );
1420                         ldap_pvt_thread_mutex_unlock( &connections_mutex );
1421                         return 0;
1422                 }
1423         }
1424 #endif
1425
1426 #define CONNECTION_INPUT_LOOP 1
1427 /* #define      DATA_READY_LOOP 1 */
1428
1429         do {
1430                 /* How do we do this without getting into a busy loop ? */
1431                 rc = connection_input( c );
1432         }
1433 #ifdef DATA_READY_LOOP
1434         while( !rc && ber_sockbuf_ctrl( c->c_sb, LBER_SB_OPT_DATA_READY, NULL ));
1435 #elif CONNECTION_INPUT_LOOP
1436         while(!rc);
1437 #else
1438         while(0);
1439 #endif
1440
1441         if( rc < 0 ) {
1442 #ifdef NEW_LOGGING
1443                 LDAP_LOG( CONNECTION, ERR, 
1444                         "connection_read: conn %lu input error %d, closing.\n",
1445                         c->c_connid, rc, 0 );
1446 #else
1447                 Debug( LDAP_DEBUG_TRACE,
1448                         "connection_read(%d): input error=%d id=%lu, closing.\n",
1449                         s, rc, c->c_connid );
1450 #endif
1451                 /* connections_mutex and c_mutex are locked */
1452                 connection_closing( c );
1453                 connection_close( c );
1454                 connection_return( c );
1455                 ldap_pvt_thread_mutex_unlock( &connections_mutex );
1456                 return 0;
1457         }
1458
1459         if ( ber_sockbuf_ctrl( c->c_sb, LBER_SB_OPT_NEEDS_READ, NULL ) ) {
1460                 slapd_set_read( s, 1 );
1461         }
1462
1463         if ( ber_sockbuf_ctrl( c->c_sb, LBER_SB_OPT_NEEDS_WRITE, NULL ) ) {
1464                 slapd_set_write( s, 1 );
1465         }
1466
1467         connection_return( c );
1468         ldap_pvt_thread_mutex_unlock( &connections_mutex );
1469         return 0;
1470 }
1471
1472 static int
1473 connection_input(
1474         Connection *conn )
1475 {
1476         Operation *op;
1477         ber_tag_t       tag;
1478         ber_len_t       len;
1479         ber_int_t       msgid;
1480         BerElement      *ber;
1481         int             rc;
1482 #ifdef LDAP_CONNECTIONLESS
1483         Sockaddr        peeraddr;
1484         char            *cdn = NULL;
1485 #endif
1486         char *defer = NULL;
1487
1488         if ( conn->c_currentber == NULL &&
1489                 ( conn->c_currentber = ber_alloc()) == NULL )
1490         {
1491 #ifdef NEW_LOGGING
1492                 LDAP_LOG( CONNECTION, ERR, 
1493                         "connection_input: conn %lu ber_alloc failed.\n", 
1494                         conn->c_connid, 0, 0 );
1495 #else
1496                 Debug( LDAP_DEBUG_ANY, "ber_alloc failed\n", 0, 0, 0 );
1497 #endif
1498                 return -1;
1499         }
1500
1501         errno = 0;
1502
1503 #ifdef LDAP_CONNECTIONLESS
1504         if ( conn->c_is_udp ) {
1505                 char    peername[sizeof("IP=255.255.255.255:65336")];
1506                 len = ber_int_sb_read(conn->c_sb, &peeraddr,
1507                         sizeof(struct sockaddr));
1508                 if (len != sizeof(struct sockaddr))
1509                         return 1;
1510                 sprintf( peername, "IP=%s:%d",
1511                         inet_ntoa( peeraddr.sa_in_addr.sin_addr ),
1512                         (unsigned) ntohs( peeraddr.sa_in_addr.sin_port ) );
1513                 Statslog( LDAP_DEBUG_STATS,
1514                         "conn=%lu UDP request from %s (%s) accepted.\n",
1515                         conn->c_connid, peername, conn->c_sock_name.bv_val, 0, 0 );
1516         }
1517 #endif
1518         tag = ber_get_next( conn->c_sb, &len, conn->c_currentber );
1519         if ( tag != LDAP_TAG_MESSAGE ) {
1520                 int err = errno;
1521                 ber_socket_t    sd;
1522
1523                 ber_sockbuf_ctrl( conn->c_sb, LBER_SB_OPT_GET_FD, &sd );
1524
1525 #ifdef NEW_LOGGING
1526                 LDAP_LOG( CONNECTION, ERR, 
1527                         "connection_input: conn %lu ber_get_next failed, errno %d (%s).\n",
1528                         conn->c_connid, err, sock_errstr(err) );
1529 #else
1530                 Debug( LDAP_DEBUG_TRACE,
1531                         "ber_get_next on fd %d failed errno=%d (%s)\n",
1532                         sd, err, sock_errstr(err) );
1533 #endif
1534                 if ( err != EWOULDBLOCK && err != EAGAIN ) {
1535                         /* log, close and send error */
1536                         ber_free( conn->c_currentber, 1 );
1537                         conn->c_currentber = NULL;
1538
1539                         return -2;
1540                 }
1541                 return 1;
1542         }
1543
1544         ber = conn->c_currentber;
1545         conn->c_currentber = NULL;
1546
1547         if ( (tag = ber_get_int( ber, &msgid )) != LDAP_TAG_MSGID ) {
1548                 /* log, close and send error */
1549 #ifdef NEW_LOGGING
1550                 LDAP_LOG( CONNECTION, ERR, 
1551                         "connection_input: conn %lu ber_get_int returns 0x%lx.\n",
1552                         conn->c_connid, tag, 0 );
1553 #else
1554                 Debug( LDAP_DEBUG_ANY, "ber_get_int returns 0x%lx\n",
1555                         tag, 0, 0 );
1556 #endif
1557                 ber_free( ber, 1 );
1558                 return -1;
1559         }
1560
1561         if ( (tag = ber_peek_tag( ber, &len )) == LBER_ERROR ) {
1562                 /* log, close and send error */
1563 #ifdef NEW_LOGGING
1564                 LDAP_LOG( CONNECTION, ERR, 
1565                         "connection_input: conn %lu ber_peek_tag returns 0x%lx.\n",
1566                         conn->c_connid, tag, 0 );
1567 #else
1568                 Debug( LDAP_DEBUG_ANY, "ber_peek_tag returns 0x%lx\n",
1569                         tag, 0, 0 );
1570 #endif
1571                 ber_free( ber, 1 );
1572
1573                 return -1;
1574         }
1575
1576 #ifdef LDAP_CONNECTIONLESS
1577         if( conn->c_is_udp ) {
1578                 if( tag == LBER_OCTETSTRING ) {
1579                         ber_get_stringa( ber, &cdn );
1580                         tag = ber_peek_tag(ber, &len);
1581                 }
1582                 if( tag != LDAP_REQ_ABANDON && tag != LDAP_REQ_SEARCH ) {
1583 #ifdef NEW_LOGGING
1584                         LDAP_LOG( CONNECTION, ERR, 
1585                                 "connection_input: conn %lu invalid req for UDP 0x%lx.\n",
1586                                 conn->c_connid, tag, 0 );
1587 #else
1588                         Debug( LDAP_DEBUG_ANY, "invalid req for UDP 0x%lx\n", tag, 0, 0 );
1589 #endif
1590                         ber_free( ber, 1 );
1591                         return 0;
1592                 }
1593         }
1594 #endif
1595         if(tag == LDAP_REQ_BIND) {
1596                 /* immediately abandon all exiting operations upon BIND */
1597                 connection_abandon( conn );
1598         }
1599
1600         op = slap_op_alloc( ber, msgid, tag, conn->c_n_ops_received++ );
1601
1602         op->o_conn = conn;
1603         op->o_assertion = NULL;
1604         op->o_preread_attrs = NULL;
1605         op->o_postread_attrs = NULL;
1606         op->o_vrFilter = NULL;
1607         /* clear state if the connection is being reused from inactive */
1608         if ( conn->c_conn_state == SLAP_C_INACTIVE ) {
1609                 memset( &conn->c_pagedresults_state, 0, sizeof( conn->c_pagedresults_state ) );
1610         }
1611         op->o_pagedresults_state = conn->c_pagedresults_state;
1612
1613         op->o_res_ber = NULL;
1614
1615 #ifdef LDAP_CONNECTIONLESS
1616         if (conn->c_is_udp) {
1617                 if ( cdn ) {
1618                         ber_str2bv( cdn, 0, 1, &op->o_dn );
1619                         op->o_protocol = LDAP_VERSION2;
1620                 }
1621                 op->o_res_ber = ber_alloc_t( LBER_USE_DER );
1622                 if (op->o_res_ber == NULL) return 1;
1623
1624                 rc = ber_write( op->o_res_ber, (char *)&peeraddr,
1625                         sizeof(struct sockaddr), 0 );
1626
1627                 if (rc != sizeof(struct sockaddr)) {
1628 #ifdef NEW_LOGGING
1629                         LDAP_LOG( CONNECTION, INFO, 
1630                                 "connection_input: conn %lu ber_write failed\n",
1631                                 conn->c_connid, 0, 0 );
1632 #else
1633                         Debug( LDAP_DEBUG_ANY, "ber_write failed\n", 0, 0, 0 );
1634 #endif
1635                         return 1;
1636                 }
1637
1638                 if (op->o_protocol == LDAP_VERSION2) {
1639                         rc = ber_printf(op->o_res_ber, "{is{" /*}}*/, op->o_msgid, "");
1640                         if (rc == -1) {
1641 #ifdef NEW_LOGGING
1642                                 LDAP_LOG( CONNECTION, INFO, 
1643                                         "connection_input: conn %lu put outer sequence failed\n",
1644                                         conn->c_connid, 0, 0 );
1645 #else
1646                                 Debug( LDAP_DEBUG_ANY, "ber_write failed\n", 0, 0, 0 );
1647 #endif
1648                                 return rc;
1649                         }
1650                 }
1651         }
1652 #endif /* LDAP_CONNECTIONLESS */
1653
1654         rc = 0;
1655
1656         /* Don't process requests when the conn is in the middle of a
1657          * Bind, or if it's closing. Also, don't let any single conn
1658          * use up all the available threads, and don't execute if we're
1659          * currently blocked on output. And don't execute if there are
1660          * already pending ops, let them go first.  Abandon operations
1661          * get exceptions to some, but not all, cases.
1662          */
1663         if (tag != LDAP_REQ_ABANDON && conn->c_conn_state == SLAP_C_CLOSING) {
1664                 defer = "closing";
1665         } else if (tag != LDAP_REQ_ABANDON && conn->c_writewaiter) {
1666                 defer = "awaiting write";
1667         } else if (conn->c_n_ops_executing >= connection_pool_max/2) {
1668                 defer = "too many executing";
1669         } else if (conn->c_conn_state == SLAP_C_BINDING) {
1670                 defer = "binding";
1671         } else if (tag != LDAP_REQ_ABANDON && conn->c_n_ops_pending) {
1672                 defer = "pending operations";
1673         }
1674
1675         if( defer ) {
1676                 int max = conn->c_dn.bv_len
1677                         ? slap_conn_max_pending_auth
1678                         : slap_conn_max_pending;
1679
1680 #ifdef NEW_LOGGING
1681                 LDAP_LOG( CONNECTION, INFO, 
1682                         "connection_input: conn %lu deferring operation: %s\n",
1683                         conn->c_connid, defer, 0 );
1684 #else
1685                 Debug( LDAP_DEBUG_ANY,
1686                         "connection_input: conn=%lu deferring operation: %s\n",
1687                         conn->c_connid, defer, 0 );
1688 #endif
1689                 conn->c_n_ops_pending++;
1690                 LDAP_STAILQ_INSERT_TAIL( &conn->c_pending_ops, op, o_next );
1691                 if ( conn->c_n_ops_pending > max ) {
1692                         rc = -1;
1693                 } else {
1694                         rc = 1;
1695                 }
1696         } else {
1697                 conn->c_n_ops_executing++;
1698                 connection_op_activate( op );
1699         }
1700
1701 #ifdef NO_THREADS
1702         if ( conn->c_struct_state != SLAP_C_USED ) {
1703                 /* connection must have got closed underneath us */
1704                 return 1;
1705         }
1706 #endif
1707         assert( conn->c_struct_state == SLAP_C_USED );
1708
1709         return rc;
1710 }
1711
1712 static int
1713 connection_resched( Connection *conn )
1714 {
1715         Operation *op;
1716
1717         if( conn->c_conn_state == SLAP_C_CLOSING ) {
1718                 int rc;
1719                 ber_socket_t    sd;
1720                 ber_sockbuf_ctrl( conn->c_sb, LBER_SB_OPT_GET_FD, &sd );
1721
1722                 /* us trylock to avoid possible deadlock */
1723                 rc = ldap_pvt_thread_mutex_trylock( &connections_mutex );
1724
1725                 if( rc ) {
1726 #ifdef NEW_LOGGING
1727                         LDAP_LOG( CONNECTION, DETAIL1, 
1728                                 "connection_resched: conn %lu reaquiring locks.\n",
1729                                 conn->c_connid, 0, 0 );
1730 #else
1731                         Debug( LDAP_DEBUG_TRACE,
1732                                 "connection_resched: reaquiring locks conn=%lu sd=%d\n",
1733                                 conn->c_connid, sd, 0 );
1734 #endif
1735                         /*
1736                          * reaquire locks in the right order...
1737                          * this may allow another thread to close this connection,
1738                          * so recheck state below.
1739                          */
1740                         ldap_pvt_thread_mutex_unlock( &conn->c_mutex );
1741                         ldap_pvt_thread_mutex_lock( &connections_mutex );
1742                         ldap_pvt_thread_mutex_lock( &conn->c_mutex );
1743                 }
1744
1745                 if( conn->c_conn_state != SLAP_C_CLOSING ) {
1746 #ifdef NEW_LOGGING
1747                         LDAP_LOG( CONNECTION, INFO, 
1748                                 "connection_resched: conn %lu closed by other thread.\n",
1749                                 conn->c_connid, 0, 0 );
1750 #else
1751                         Debug( LDAP_DEBUG_TRACE, "connection_resched: "
1752                                 "closed by other thread conn=%lu sd=%d\n",
1753                                 conn->c_connid, sd, 0 );
1754 #endif
1755                 } else {
1756 #ifdef NEW_LOGGING
1757                         LDAP_LOG( CONNECTION, DETAIL1, 
1758                                 "connection_resched: conn %lu attempting closing.\n",
1759                                 conn->c_connid, 0, 0 );
1760 #else
1761                         Debug( LDAP_DEBUG_TRACE, "connection_resched: "
1762                                 "attempting closing conn=%lu sd=%d\n",
1763                                 conn->c_connid, sd, 0 );
1764 #endif
1765                         connection_close( conn );
1766                 }
1767
1768                 ldap_pvt_thread_mutex_unlock( &connections_mutex );
1769                 return 0;
1770         }
1771
1772         if( conn->c_conn_state != SLAP_C_ACTIVE || conn->c_writewaiter ) {
1773                 /* other states need different handling */
1774                 return 0;
1775         }
1776
1777         while ((op = LDAP_STAILQ_FIRST( &conn->c_pending_ops )) != NULL) {
1778                 if ( conn->c_n_ops_executing > connection_pool_max/2 ) {
1779                         break;
1780                 }
1781                 LDAP_STAILQ_REMOVE_HEAD( &conn->c_pending_ops, o_next );
1782                 LDAP_STAILQ_NEXT(op, o_next) = NULL;
1783                 /* pending operations should not be marked for abandonment */
1784                 assert(!op->o_abandon);
1785
1786                 conn->c_n_ops_pending--;
1787                 conn->c_n_ops_executing++;
1788
1789                 connection_op_activate( op );
1790
1791                 if ( conn->c_conn_state == SLAP_C_BINDING ) {
1792                         break;
1793                 }
1794         }
1795         return 0;
1796 }
1797
1798 static int connection_op_activate( Operation *op )
1799 {
1800         int status;
1801         ber_tag_t tag = op->o_tag;
1802
1803         if(tag == LDAP_REQ_BIND) {
1804                 op->o_conn->c_conn_state = SLAP_C_BINDING;
1805         }
1806
1807         if (!op->o_dn.bv_len) {
1808                 op->o_authz = op->o_conn->c_authz;
1809                 ber_dupbv( &op->o_dn, &op->o_conn->c_dn );
1810                 ber_dupbv( &op->o_ndn, &op->o_conn->c_ndn );
1811         }
1812         op->o_authtype = op->o_conn->c_authtype;
1813         ber_dupbv( &op->o_authmech, &op->o_conn->c_authmech );
1814         
1815         if (!op->o_protocol) {
1816                 op->o_protocol = op->o_conn->c_protocol
1817                         ? op->o_conn->c_protocol : LDAP_VERSION3;
1818         }
1819         if (op->o_conn->c_conn_state == SLAP_C_INACTIVE
1820                 && op->o_protocol > LDAP_VERSION2)
1821         {
1822                 op->o_conn->c_conn_state = SLAP_C_ACTIVE;
1823         }
1824
1825         op->o_connid = op->o_conn->c_connid;
1826
1827         LDAP_STAILQ_INSERT_TAIL( &op->o_conn->c_ops, op, o_next );
1828
1829         status = ldap_pvt_thread_pool_submit( &connection_pool,
1830                 connection_operation, (void *) op );
1831
1832         if ( status != 0 ) {
1833 #ifdef NEW_LOGGING
1834                 LDAP_LOG( CONNECTION, ERR, 
1835                         "connection_op_activate: conn %lu        thread pool submit failed.\n",
1836                         op->o_connid, 0, 0 );
1837 #else
1838                 Debug( LDAP_DEBUG_ANY,
1839                         "ldap_pvt_thread_pool_submit: failed (%d) for conn=%lu\n",
1840                         status, op->o_connid, 0 );
1841 #endif
1842                 /* should move op to pending list */
1843         }
1844
1845         return status;
1846 }
1847
1848 int connection_write(ber_socket_t s)
1849 {
1850         Connection *c;
1851
1852         assert( connections != NULL );
1853
1854         ldap_pvt_thread_mutex_lock( &connections_mutex );
1855
1856         c = connection_get( s );
1857
1858         slapd_clr_write( s, 0);
1859
1860         if( c == NULL ) {
1861 #ifdef NEW_LOGGING
1862                 LDAP_LOG( CONNECTION, ERR, 
1863                         "connection_write: sock %ld no connection!\n", (long)s, 0, 0);
1864 #else
1865                 Debug( LDAP_DEBUG_ANY,
1866                         "connection_write(%ld): no connection!\n",
1867                         (long)s, 0, 0 );
1868 #endif
1869                 slapd_remove(s, 1, 0);
1870                 ldap_pvt_thread_mutex_unlock( &connections_mutex );
1871                 return -1;
1872         }
1873
1874         c->c_n_write++;
1875
1876 #ifdef NEW_LOGGING
1877         LDAP_LOG( CONNECTION, DETAIL1, 
1878                 "connection_write conn %lu waking output.\n", c->c_connid, 0, 0 );
1879 #else
1880         Debug( LDAP_DEBUG_TRACE,
1881                 "connection_write(%d): waking output for id=%lu\n",
1882                 s, c->c_connid, 0 );
1883 #endif
1884         ldap_pvt_thread_cond_signal( &c->c_write_cv );
1885
1886         if ( ber_sockbuf_ctrl( c->c_sb, LBER_SB_OPT_NEEDS_READ, NULL ) ) {
1887                 slapd_set_read( s, 1 );
1888         }
1889         if ( ber_sockbuf_ctrl( c->c_sb, LBER_SB_OPT_NEEDS_WRITE, NULL ) ) {
1890                 slapd_set_write( s, 1 );
1891         }
1892         connection_return( c );
1893         ldap_pvt_thread_mutex_unlock( &connections_mutex );
1894         return 0;
1895 }
1896
1897 void
1898 connection_fake_init(
1899         Connection *conn,
1900         Operation *op,
1901         void *ctx )
1902 {
1903         conn->c_connid = -1;
1904         conn->c_send_ldap_result = slap_send_ldap_result;
1905         conn->c_send_search_entry = slap_send_search_entry;
1906         conn->c_send_search_reference = slap_send_search_reference;
1907         conn->c_listener = (Listener *)&dummy_list;
1908         conn->c_peer_domain = slap_empty_bv;
1909         conn->c_peer_name = slap_empty_bv;
1910
1911         /* set memory context */
1912         op->o_tmpmemctx = sl_mem_create( SLMALLOC_SLAB_SIZE, ctx );
1913         op->o_tmpmfuncs = &sl_mfuncs;
1914         op->o_threadctx = ctx;
1915
1916         op->o_conn = conn;
1917         op->o_connid = op->o_conn->c_connid;
1918
1919         op->o_time = slap_get_time();
1920 }
1921
1922 void
1923 connection_assign_nextid( Connection *conn )
1924 {
1925         ldap_pvt_thread_mutex_lock( &conn_nextid_mutex );
1926         conn->c_connid = conn_nextid++;
1927         ldap_pvt_thread_mutex_unlock( &conn_nextid_mutex );
1928 }
1929