]> git.sur5r.net Git - openldap/blob - servers/slapd/connection.c
ITS#3534, 3546, 3571 revert abandon behavior
[openldap] / servers / slapd / connection.c
1 /* $OpenLDAP$ */
2 /* This work is part of OpenLDAP Software <http://www.openldap.org/>.
3  *
4  * Copyright 1998-2005 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;
776
777         LDAP_STAILQ_FOREACH( o, &c->c_ops, o_next ) {
778                 o->o_abandon = 1;
779         }
780
781         /* remove pending operations */
782         while ( (o = LDAP_STAILQ_FIRST( &c->c_pending_ops )) != NULL) {
783                 LDAP_STAILQ_REMOVE_HEAD( &c->c_pending_ops, o_next );
784                 LDAP_STAILQ_NEXT(o, o_next) = NULL;
785                 slap_op_free( o );
786         }
787 }
788
789 void connection_closing( Connection *c )
790 {
791         assert( connections != NULL );
792         assert( c != NULL );
793         assert( c->c_struct_state == SLAP_C_USED );
794         assert( c->c_conn_state != SLAP_C_INVALID );
795
796         /* c_mutex must be locked by caller */
797
798         if( c->c_conn_state != SLAP_C_CLOSING ) {
799                 ber_socket_t    sd;
800
801                 ber_sockbuf_ctrl( c->c_sb, LBER_SB_OPT_GET_FD, &sd );
802 #ifdef NEW_LOGGING
803                 LDAP_LOG( CONNECTION, DETAIL1, 
804                         "connection_closing: conn %lu readying socket %d for close.\n",
805                         c->c_connid, sd, 0 );
806 #else
807                 Debug( LDAP_DEBUG_TRACE,
808                         "connection_closing: readying conn=%lu sd=%d for close\n",
809                         c->c_connid, sd, 0 );
810 #endif
811                 /* update state to closing */
812                 c->c_conn_state = SLAP_C_CLOSING;
813
814                 /* don't listen on this port anymore */
815                 slapd_clr_read( sd, 1 );
816
817                 /* abandon active operations */
818                 connection_abandon( c );
819
820                 /* wake write blocked operations */
821                 slapd_clr_write( sd, 1 );
822                 ldap_pvt_thread_cond_signal( &c->c_write_cv );
823         }
824 }
825
826 static void connection_close( Connection *c )
827 {
828         ber_socket_t    sd;
829
830         assert( connections != NULL );
831         assert( c != NULL );
832         assert( c->c_struct_state == SLAP_C_USED );
833         assert( c->c_conn_state == SLAP_C_CLOSING );
834
835         /* note: connections_mutex and c_mutex should be locked by caller */
836
837         ber_sockbuf_ctrl( c->c_sb, LBER_SB_OPT_GET_FD, &sd );
838         if( !LDAP_STAILQ_EMPTY(&c->c_ops) ) {
839 #ifdef NEW_LOGGING
840                 LDAP_LOG( CONNECTION, DETAIL1, 
841                         "connection_close: conn %lu deferring sd %d\n",
842                                 c->c_connid, sd, 0 );
843 #else
844                 Debug( LDAP_DEBUG_TRACE,
845                         "connection_close: deferring conn=%lu sd=%d\n",
846                         c->c_connid, sd, 0 );
847 #endif
848                 return;
849         }
850
851 #ifdef NEW_LOGGING
852         LDAP_LOG( CONNECTION, RESULTS, 
853                 "connection_close: conn %lu sd %d\n", c->c_connid, sd, 0 );
854 #else
855         Debug( LDAP_DEBUG_TRACE, "connection_close: conn=%lu sd=%d\n",
856                 c->c_connid, sd, 0 );
857 #endif
858         connection_destroy( c );
859 }
860
861 unsigned long connections_nextid(void)
862 {
863         unsigned long id;
864         assert( connections != NULL );
865
866         ldap_pvt_thread_mutex_lock( &connections_mutex );
867
868         id = conn_nextid;
869
870         ldap_pvt_thread_mutex_unlock( &connections_mutex );
871
872         return id;
873 }
874
875 Connection* connection_first( ber_socket_t *index )
876 {
877         assert( connections != NULL );
878         assert( index != NULL );
879
880         ldap_pvt_thread_mutex_lock( &connections_mutex );
881
882         *index = 0;
883
884         return connection_next(NULL, index);
885 }
886
887 Connection* connection_next( Connection *c, ber_socket_t *index )
888 {
889         assert( connections != NULL );
890         assert( index != NULL );
891         assert( *index <= dtblsize );
892
893         if( c != NULL ) {
894                 ldap_pvt_thread_mutex_unlock( &c->c_mutex );
895         }
896
897         c = NULL;
898
899         for(; *index < dtblsize; (*index)++) {
900                 if( connections[*index].c_struct_state == SLAP_C_UNINITIALIZED ) {
901                         assert( connections[*index].c_conn_state == SLAP_C_INVALID );
902 #ifndef HAVE_WINSOCK
903                         continue;
904 #else
905                         break;
906 #endif
907                 }
908
909                 if( connections[*index].c_struct_state == SLAP_C_USED ) {
910                         assert( connections[*index].c_conn_state != SLAP_C_INVALID );
911                         c = &connections[(*index)++];
912                         break;
913                 }
914
915                 assert( connections[*index].c_struct_state == SLAP_C_UNUSED );
916                 assert( connections[*index].c_conn_state == SLAP_C_INVALID );
917         }
918
919         if( c != NULL ) {
920                 ldap_pvt_thread_mutex_lock( &c->c_mutex );
921         }
922
923         return c;
924 }
925
926 void connection_done( Connection *c )
927 {
928         assert( connections != NULL );
929
930         if( c != NULL ) {
931                 ldap_pvt_thread_mutex_unlock( &c->c_mutex );
932         }
933
934         ldap_pvt_thread_mutex_unlock( &connections_mutex );
935 }
936
937 /*
938  * connection_activity - handle the request operation op on connection
939  * conn.  This routine figures out what kind of operation it is and
940  * calls the appropriate stub to handle it.
941  */
942
943 #ifdef SLAPD_MONITOR
944 #define INCR_OP(var,index) \
945         do { \
946                 ldap_pvt_thread_mutex_lock( &num_ops_mutex ); \
947                 (var)[(index)]++; \
948                 ldap_pvt_thread_mutex_unlock( &num_ops_mutex ); \
949         } while (0)
950 #else /* !SLAPD_MONITOR */
951 #define INCR_OP(var,index) 
952 #endif /* !SLAPD_MONITOR */
953
954 static void *
955 connection_operation( void *ctx, void *arg_v )
956 {
957         int rc = LDAP_OTHER;
958         Operation *op = arg_v;
959         SlapReply rs = {REP_RESULT};
960         ber_tag_t tag = op->o_tag;
961 #ifdef SLAPD_MONITOR
962         ber_tag_t oldtag = tag;
963 #endif /* SLAPD_MONITOR */
964         Connection *conn = op->o_conn;
965         void *memctx = NULL;
966         void *memctx_null = NULL;
967         ber_len_t memsiz;
968
969         ldap_pvt_thread_mutex_lock( &num_ops_mutex );
970         num_ops_initiated++;
971         ldap_pvt_thread_mutex_unlock( &num_ops_mutex );
972
973         op->o_threadctx = ctx;
974
975         switch ( tag ) {
976         case LDAP_REQ_BIND:
977         case LDAP_REQ_UNBIND:
978         case LDAP_REQ_ADD:
979         case LDAP_REQ_DELETE:
980         case LDAP_REQ_MODRDN:
981         case LDAP_REQ_MODIFY:
982         case LDAP_REQ_COMPARE:
983         case LDAP_REQ_SEARCH:
984         case LDAP_REQ_ABANDON:
985         case LDAP_REQ_EXTENDED:
986                 break;
987         default:
988 #ifdef NEW_LOGGING
989                 LDAP_LOG( CONNECTION, INFO, "connection_operation: "
990                         "conn %lu unknown LDAP request 0x%lx\n",
991                         conn->c_connid, tag, 0 );
992 #else
993                 Debug( LDAP_DEBUG_ANY, "connection_operation: "
994                         "conn %lu unknown LDAP request 0x%lx\n",
995                         conn->c_connid, tag, 0 );
996 #endif
997                 op->o_tag = LBER_ERROR;
998                 rs.sr_err = LDAP_PROTOCOL_ERROR;
999                 rs.sr_text = "unknown LDAP request";
1000                 send_ldap_disconnect( op, &rs );
1001                 rc = SLAPD_DISCONNECT;
1002                 goto operations_error;
1003         }
1004
1005         if( conn->c_sasl_bind_in_progress && tag != LDAP_REQ_BIND ) {
1006 #ifdef NEW_LOGGING
1007                 LDAP_LOG( CONNECTION, ERR, 
1008                         "connection_operation: conn %lu SASL bind in progress (tag=%ld).\n",
1009                         conn->c_connid, (long)tag, 0 );
1010 #else
1011                 Debug( LDAP_DEBUG_ANY, "connection_operation: "
1012                         "error: SASL bind in progress (tag=%ld).\n",
1013                         (long) tag, 0, 0 );
1014 #endif
1015                 send_ldap_error( op, &rs, LDAP_OPERATIONS_ERROR,
1016                         "SASL bind in progress" );
1017                 rc = LDAP_OPERATIONS_ERROR;
1018                 goto operations_error;
1019         }
1020
1021         /* We can use Thread-Local storage for most mallocs. We can
1022          * also use TL for ber parsing, but not on Add or Modify.
1023          */
1024 #if 0
1025         memsiz = ber_len( op->o_ber ) * 64;
1026         if ( SLMALLOC_SLAB_SIZE > memsiz ) memsiz = SLMALLOC_SLAB_SIZE;
1027 #endif
1028         memsiz = SLMALLOC_SLAB_SIZE;
1029
1030         memctx = sl_mem_create( memsiz, ctx );
1031         op->o_tmpmemctx = memctx;
1032         op->o_tmpmfuncs = &sl_mfuncs;
1033         if ( tag != LDAP_REQ_ADD && tag != LDAP_REQ_MODIFY ) {
1034                 /* Note - the ber and its buffer are already allocated from
1035                  * regular memory; this only affects subsequent mallocs that
1036                  * ber_scanf may invoke.
1037                  */
1038                 ber_set_option( op->o_ber, LBER_OPT_BER_MEMCTX, &memctx );
1039         }
1040
1041         switch ( tag ) {
1042         case LDAP_REQ_BIND:
1043                 INCR_OP(num_ops_initiated_, SLAP_OP_BIND);
1044                 rc = do_bind( op, &rs );
1045                 break;
1046
1047         case LDAP_REQ_UNBIND:
1048                 INCR_OP(num_ops_initiated_, SLAP_OP_UNBIND);
1049                 rc = do_unbind( op, &rs );
1050                 break;
1051
1052         case LDAP_REQ_ADD:
1053                 INCR_OP(num_ops_initiated_, SLAP_OP_ADD);
1054                 rc = do_add( op, &rs );
1055                 break;
1056
1057         case LDAP_REQ_DELETE:
1058                 INCR_OP(num_ops_initiated_, SLAP_OP_DELETE);
1059                 rc = do_delete( op, &rs );
1060                 break;
1061
1062         case LDAP_REQ_MODRDN:
1063                 INCR_OP(num_ops_initiated_, SLAP_OP_MODRDN);
1064                 rc = do_modrdn( op, &rs );
1065                 break;
1066
1067         case LDAP_REQ_MODIFY:
1068                 INCR_OP(num_ops_initiated_, SLAP_OP_MODIFY);
1069                 rc = do_modify( op, &rs );
1070                 break;
1071
1072         case LDAP_REQ_COMPARE:
1073                 INCR_OP(num_ops_initiated_, SLAP_OP_COMPARE);
1074                 rc = do_compare( op, &rs );
1075                 break;
1076
1077         case LDAP_REQ_SEARCH:
1078                 INCR_OP(num_ops_initiated_, SLAP_OP_SEARCH);
1079                 rc = do_search( op, &rs );
1080                 break;
1081
1082         case LDAP_REQ_ABANDON:
1083                 INCR_OP(num_ops_initiated_, SLAP_OP_ABANDON);
1084                 rc = do_abandon( op, &rs );
1085                 break;
1086
1087         case LDAP_REQ_EXTENDED:
1088                 INCR_OP(num_ops_initiated_, SLAP_OP_EXTENDED);
1089                 rc = do_extended( op, &rs );
1090                 break;
1091
1092         default:
1093                 /* not reachable */
1094                 assert( 0 );
1095         }
1096
1097 operations_error:
1098         if( rc == SLAPD_DISCONNECT ) tag = LBER_ERROR;
1099
1100         ldap_pvt_thread_mutex_lock( &num_ops_mutex );
1101
1102         num_ops_completed++;
1103 #ifdef SLAPD_MONITOR
1104         switch (oldtag) {
1105         case LDAP_REQ_BIND:
1106                 num_ops_completed_[SLAP_OP_BIND]++;
1107                 break;
1108         case LDAP_REQ_UNBIND:
1109                 num_ops_completed_[SLAP_OP_UNBIND]++;
1110                 break;
1111         case LDAP_REQ_ADD:
1112                 num_ops_completed_[SLAP_OP_ADD]++;
1113                 break;
1114         case LDAP_REQ_DELETE:
1115                 num_ops_completed_[SLAP_OP_DELETE]++;
1116                 break;
1117         case LDAP_REQ_MODRDN:
1118                 num_ops_completed_[SLAP_OP_MODRDN]++;
1119                 break;
1120         case LDAP_REQ_MODIFY:
1121                 num_ops_completed_[SLAP_OP_MODIFY]++;
1122                 break;
1123         case LDAP_REQ_COMPARE:
1124                 num_ops_completed_[SLAP_OP_COMPARE]++;
1125                 break;
1126         case LDAP_REQ_SEARCH:
1127                 num_ops_completed_[SLAP_OP_SEARCH]++;
1128                 break;
1129         case LDAP_REQ_ABANDON:
1130                 num_ops_completed_[SLAP_OP_ABANDON]++;
1131                 break;
1132         case LDAP_REQ_EXTENDED:
1133                 num_ops_completed_[SLAP_OP_EXTENDED]++;
1134                 break;
1135         default:
1136                 /* this is reachable */
1137                 break;
1138         }
1139 #endif /* SLAPD_MONITOR */
1140         ldap_pvt_thread_mutex_unlock( &num_ops_mutex );
1141
1142         if ( op->o_cancel == SLAP_CANCEL_REQ ) {
1143                 op->o_cancel = LDAP_TOO_LATE;
1144         }
1145         while ( op->o_cancel != SLAP_CANCEL_NONE &&
1146                 op->o_cancel != SLAP_CANCEL_DONE )
1147         {
1148                 ldap_pvt_thread_yield();
1149         }
1150
1151         ldap_pvt_thread_mutex_lock( &conn->c_mutex );
1152
1153         ber_set_option( op->o_ber, LBER_OPT_BER_MEMCTX, &memctx_null );
1154
1155         if ( op->o_cancel != SLAP_CANCEL_ACK &&
1156                 ( op->o_sync_mode & SLAP_SYNC_PERSIST ) )
1157         {
1158                 sl_mem_detach( ctx, memctx );
1159
1160         } else if ( op->o_sync_slog_size != -1 ) {
1161                 sl_mem_detach( ctx, memctx );
1162                 LDAP_STAILQ_REMOVE( &conn->c_ops, op, slap_op, o_next);
1163                 LDAP_STAILQ_NEXT(op, o_next) = NULL;
1164                 conn->c_n_ops_executing--;
1165                 conn->c_n_ops_completed++;
1166
1167         } else {
1168                 LDAP_STAILQ_REMOVE( &conn->c_ops, op, slap_op, o_next);
1169                 LDAP_STAILQ_NEXT(op, o_next) = NULL;
1170                 slap_op_free( op );
1171                 conn->c_n_ops_executing--;
1172                 conn->c_n_ops_completed++;
1173         }
1174
1175         switch( tag ) {
1176         case LBER_ERROR:
1177         case LDAP_REQ_UNBIND:
1178                 /* c_mutex is locked */
1179                 connection_closing( conn );
1180                 break;
1181
1182         case LDAP_REQ_BIND:
1183                 conn->c_sasl_bind_in_progress =
1184                         rc == LDAP_SASL_BIND_IN_PROGRESS ? 1 : 0;
1185
1186                 if( conn->c_conn_state == SLAP_C_BINDING) {
1187                         conn->c_conn_state = SLAP_C_ACTIVE;
1188                 }
1189         }
1190
1191         connection_resched( conn );
1192         ldap_pvt_thread_mutex_unlock( &conn->c_mutex );
1193         return NULL;
1194 }
1195
1196 static const Listener dummy_list = { BER_BVC(""), BER_BVC("") };
1197
1198 int connection_client_setup(
1199         ber_socket_t s,
1200         ldap_pvt_thread_start_t *func,
1201         void *arg )
1202 {
1203         int rc;
1204         Connection *c;
1205
1206         rc = connection_init( s, (Listener *)&dummy_list, "", "",
1207                 CONN_IS_CLIENT, 0, NULL );
1208         if ( rc < 0 ) return -1;
1209
1210         c = connection_get( s );
1211         c->c_clientfunc = func;
1212         c->c_clientarg = arg;
1213         connection_return( c );
1214         slapd_add_internal( s, 0 );
1215         slapd_set_read( s, 1 );
1216         return 0;
1217 }
1218
1219 void connection_client_enable(
1220         ber_socket_t s )
1221 {
1222         slapd_set_read( s, 1 );
1223 }
1224
1225 void connection_client_stop(
1226         ber_socket_t s )
1227 {
1228         Connection *c;
1229
1230         /* get (locked) connection */
1231         c = connection_get( s );
1232         
1233         assert( c->c_conn_state == SLAP_C_CLIENT );
1234
1235         c->c_listener = NULL;
1236         c->c_conn_state = SLAP_C_INVALID;
1237         c->c_struct_state = SLAP_C_UNUSED;
1238         connection_return( c );
1239         slapd_remove( s, 0, 1 );
1240 }
1241
1242 int connection_read(ber_socket_t s)
1243 {
1244         int rc = 0;
1245         Connection *c;
1246
1247         assert( connections != NULL );
1248
1249         ldap_pvt_thread_mutex_lock( &connections_mutex );
1250
1251         /* get (locked) connection */
1252         c = connection_get( s );
1253
1254         if( c == NULL ) {
1255 #ifdef NEW_LOGGING
1256                 LDAP_LOG( CONNECTION, INFO, 
1257                         "connection_read: sock %ld no connection\n", (long)s, 0, 0 );
1258 #else
1259                 Debug( LDAP_DEBUG_ANY,
1260                         "connection_read(%ld): no connection!\n",
1261                         (long) s, 0, 0 );
1262 #endif
1263                 slapd_remove(s, 1, 0);
1264
1265                 ldap_pvt_thread_mutex_unlock( &connections_mutex );
1266                 return -1;
1267         }
1268
1269         c->c_n_read++;
1270
1271         if( c->c_conn_state == SLAP_C_CLOSING ) {
1272 #ifdef NEW_LOGGING
1273                 LDAP_LOG( CONNECTION, INFO, 
1274                         "connection_read: conn %lu connection closing, ignoring input\n",
1275                         c->c_connid, 0, 0 );
1276 #else
1277                 Debug( LDAP_DEBUG_TRACE,
1278                         "connection_read(%d): closing, ignoring input for id=%lu\n",
1279                         s, c->c_connid, 0 );
1280 #endif
1281                 connection_return( c );
1282                 ldap_pvt_thread_mutex_unlock( &connections_mutex );
1283                 return 0;
1284         }
1285
1286         if ( c->c_conn_state == SLAP_C_CLIENT ) {
1287                 slapd_clr_read( s, 0 );
1288                 ldap_pvt_thread_pool_submit( &connection_pool,
1289                         c->c_clientfunc, c->c_clientarg );
1290                 connection_return( c );
1291                 ldap_pvt_thread_mutex_unlock( &connections_mutex );
1292                 return 0;
1293         }
1294
1295 #ifdef NEW_LOGGING
1296         LDAP_LOG( CONNECTION, DETAIL1, 
1297                 "connection_read: conn %lu checking for input.\n", 
1298                         c->c_connid, 0, 0 );
1299 #else
1300         Debug( LDAP_DEBUG_TRACE,
1301                 "connection_read(%d): checking for input on id=%lu\n",
1302                 s, c->c_connid, 0 );
1303 #endif
1304
1305 #ifdef HAVE_TLS
1306         if ( c->c_is_tls && c->c_needs_tls_accept ) {
1307                 rc = ldap_pvt_tls_accept( c->c_sb, slap_tls_ctx );
1308                 if ( rc < 0 ) {
1309 #if 0 /* required by next #if 0 */
1310                         struct timeval tv;
1311                         fd_set rfd;
1312 #endif
1313
1314 #ifdef NEW_LOGGING
1315                         LDAP_LOG( CONNECTION, ERR, 
1316                                 "connection_read: conn %lu TLS accept error, error %d\n",
1317                                 c->c_connid, rc, 0 );
1318 #else
1319                         Debug( LDAP_DEBUG_TRACE,
1320                                 "connection_read(%d): TLS accept error "
1321                                 "error=%d id=%lu, closing\n",
1322                                 s, rc, c->c_connid );
1323 #endif
1324                         c->c_needs_tls_accept = 0;
1325                         /* connections_mutex and c_mutex are locked */
1326                         connection_closing( c );
1327
1328 #if 0
1329                         /* Drain input before close, to allow SSL error codes
1330                          * to propagate to client. */
1331                         FD_ZERO(&rfd);
1332                         FD_SET(s, &rfd);
1333                         for (rc=1; rc>0;) {
1334                                 tv.tv_sec = 1;
1335                                 tv.tv_usec = 0;
1336                                 rc = select(s+1, &rfd, NULL, NULL, &tv);
1337                                 if (rc == 1) {
1338                                         ber_sockbuf_ctrl( c->c_sb, LBER_SB_OPT_DRAIN, NULL);
1339                                 }
1340                         }
1341 #endif
1342                         connection_close( c );
1343
1344                 } else if ( rc == 0 ) {
1345                         void *ssl;
1346                         struct berval authid = BER_BVNULL;
1347
1348                         c->c_needs_tls_accept = 0;
1349
1350                         /* we need to let SASL know */
1351                         ssl = ldap_pvt_tls_sb_ctx( c->c_sb );
1352
1353                         c->c_tls_ssf = (slap_ssf_t) ldap_pvt_tls_get_strength( ssl );
1354                         if( c->c_tls_ssf > c->c_ssf ) {
1355                                 c->c_ssf = c->c_tls_ssf;
1356                         }
1357
1358                         rc = dnX509peerNormalize( ssl, &authid );
1359                         if ( rc != LDAP_SUCCESS ) {
1360 #ifdef NEW_LOGGING
1361                                 LDAP_LOG( CONNECTION, INFO, "connection_read: "
1362                                         "conn %lu unable to get TLS client DN, error %d\n",
1363                                         c->c_connid, rc, 0 );
1364 #else
1365                                 Debug( LDAP_DEBUG_TRACE, "connection_read(%d): "
1366                                         "unable to get TLS client DN, error=%d id=%lu\n",
1367                                         s, rc, c->c_connid );
1368 #endif
1369                         }
1370                         slap_sasl_external( c, c->c_tls_ssf, &authid );
1371                         if ( authid.bv_val ) free( authid.bv_val );
1372                 }
1373
1374                 /* if success and data is ready, fall thru to data input loop */
1375                 if( rc != 0 ||
1376                         !ber_sockbuf_ctrl( c->c_sb, LBER_SB_OPT_DATA_READY, NULL ) )
1377                 {
1378                         connection_return( c );
1379                         ldap_pvt_thread_mutex_unlock( &connections_mutex );
1380                         return 0;
1381                 }
1382         }
1383 #endif
1384
1385 #ifdef HAVE_CYRUS_SASL
1386         if ( c->c_sasl_layers ) {
1387                 /* If previous layer is not removed yet, give up for now */
1388                 if ( !c->c_sasl_sockctx ) {
1389                         connection_return( c );
1390                         ldap_pvt_thread_mutex_unlock( &connections_mutex );
1391                         return 0;
1392                 }
1393
1394                 c->c_sasl_layers = 0;
1395
1396                 rc = ldap_pvt_sasl_install( c->c_sb, c->c_sasl_sockctx );
1397
1398                 if( rc != LDAP_SUCCESS ) {
1399 #ifdef NEW_LOGGING
1400                         LDAP_LOG( CONNECTION, ERR, 
1401                                 "connection_read: conn %lu SASL install error %d, closing\n",
1402                                 c->c_connid, rc, 0 );
1403 #else
1404                         Debug( LDAP_DEBUG_TRACE,
1405                                 "connection_read(%d): SASL install error "
1406                                 "error=%d id=%lu, closing\n",
1407                                 s, rc, c->c_connid );
1408 #endif
1409                         /* connections_mutex and c_mutex are locked */
1410                         connection_closing( c );
1411                         connection_close( c );
1412                         connection_return( c );
1413                         ldap_pvt_thread_mutex_unlock( &connections_mutex );
1414                         return 0;
1415                 }
1416         }
1417 #endif
1418
1419 #define CONNECTION_INPUT_LOOP 1
1420 /* #define      DATA_READY_LOOP 1 */
1421
1422         do {
1423                 /* How do we do this without getting into a busy loop ? */
1424                 rc = connection_input( c );
1425         }
1426 #ifdef DATA_READY_LOOP
1427         while( !rc && ber_sockbuf_ctrl( c->c_sb, LBER_SB_OPT_DATA_READY, NULL ));
1428 #elif CONNECTION_INPUT_LOOP
1429         while(!rc);
1430 #else
1431         while(0);
1432 #endif
1433
1434         if( rc < 0 ) {
1435 #ifdef NEW_LOGGING
1436                 LDAP_LOG( CONNECTION, ERR, 
1437                         "connection_read: conn %lu input error %d, closing.\n",
1438                         c->c_connid, rc, 0 );
1439 #else
1440                 Debug( LDAP_DEBUG_TRACE,
1441                         "connection_read(%d): input error=%d id=%lu, closing.\n",
1442                         s, rc, c->c_connid );
1443 #endif
1444                 /* connections_mutex and c_mutex are locked */
1445                 connection_closing( c );
1446                 connection_close( c );
1447                 connection_return( c );
1448                 ldap_pvt_thread_mutex_unlock( &connections_mutex );
1449                 return 0;
1450         }
1451
1452         if ( ber_sockbuf_ctrl( c->c_sb, LBER_SB_OPT_NEEDS_READ, NULL ) ) {
1453                 slapd_set_read( s, 1 );
1454         }
1455
1456         if ( ber_sockbuf_ctrl( c->c_sb, LBER_SB_OPT_NEEDS_WRITE, NULL ) ) {
1457                 slapd_set_write( s, 1 );
1458         }
1459
1460         connection_return( c );
1461         ldap_pvt_thread_mutex_unlock( &connections_mutex );
1462         return 0;
1463 }
1464
1465 static int
1466 connection_input(
1467         Connection *conn )
1468 {
1469         Operation *op;
1470         ber_tag_t       tag;
1471         ber_len_t       len;
1472         ber_int_t       msgid;
1473         BerElement      *ber;
1474         int             rc;
1475 #ifdef LDAP_CONNECTIONLESS
1476         Sockaddr        peeraddr;
1477         char            *cdn = NULL;
1478 #endif
1479         char *defer = NULL;
1480
1481         if ( conn->c_currentber == NULL &&
1482                 ( conn->c_currentber = ber_alloc()) == NULL )
1483         {
1484 #ifdef NEW_LOGGING
1485                 LDAP_LOG( CONNECTION, ERR, 
1486                         "connection_input: conn %lu ber_alloc failed.\n", 
1487                         conn->c_connid, 0, 0 );
1488 #else
1489                 Debug( LDAP_DEBUG_ANY, "ber_alloc failed\n", 0, 0, 0 );
1490 #endif
1491                 return -1;
1492         }
1493
1494         errno = 0;
1495
1496 #ifdef LDAP_CONNECTIONLESS
1497         if ( conn->c_is_udp ) {
1498                 char    peername[sizeof("IP=255.255.255.255:65336")];
1499                 len = ber_int_sb_read(conn->c_sb, &peeraddr,
1500                         sizeof(struct sockaddr));
1501                 if (len != sizeof(struct sockaddr))
1502                         return 1;
1503                 sprintf( peername, "IP=%s:%d",
1504                         inet_ntoa( peeraddr.sa_in_addr.sin_addr ),
1505                         (unsigned) ntohs( peeraddr.sa_in_addr.sin_port ) );
1506                 Statslog( LDAP_DEBUG_STATS,
1507                         "conn=%lu UDP request from %s (%s) accepted.\n",
1508                         conn->c_connid, peername, conn->c_sock_name.bv_val, 0, 0 );
1509         }
1510 #endif
1511         tag = ber_get_next( conn->c_sb, &len, conn->c_currentber );
1512         if ( tag != LDAP_TAG_MESSAGE ) {
1513                 int err = errno;
1514                 ber_socket_t    sd;
1515
1516                 ber_sockbuf_ctrl( conn->c_sb, LBER_SB_OPT_GET_FD, &sd );
1517
1518 #ifdef NEW_LOGGING
1519                 LDAP_LOG( CONNECTION, ERR, 
1520                         "connection_input: conn %lu ber_get_next failed, errno %d (%s).\n",
1521                         conn->c_connid, err, sock_errstr(err) );
1522 #else
1523                 Debug( LDAP_DEBUG_TRACE,
1524                         "ber_get_next on fd %d failed errno=%d (%s)\n",
1525                         sd, err, sock_errstr(err) );
1526 #endif
1527                 if ( err != EWOULDBLOCK && err != EAGAIN ) {
1528                         /* log, close and send error */
1529                         ber_free( conn->c_currentber, 1 );
1530                         conn->c_currentber = NULL;
1531
1532                         return -2;
1533                 }
1534                 return 1;
1535         }
1536
1537         ber = conn->c_currentber;
1538         conn->c_currentber = NULL;
1539
1540         if ( (tag = ber_get_int( ber, &msgid )) != LDAP_TAG_MSGID ) {
1541                 /* log, close and send error */
1542 #ifdef NEW_LOGGING
1543                 LDAP_LOG( CONNECTION, ERR, 
1544                         "connection_input: conn %lu ber_get_int returns 0x%lx.\n",
1545                         conn->c_connid, tag, 0 );
1546 #else
1547                 Debug( LDAP_DEBUG_ANY, "ber_get_int returns 0x%lx\n",
1548                         tag, 0, 0 );
1549 #endif
1550                 ber_free( ber, 1 );
1551                 return -1;
1552         }
1553
1554         if ( (tag = ber_peek_tag( ber, &len )) == LBER_ERROR ) {
1555                 /* log, close and send error */
1556 #ifdef NEW_LOGGING
1557                 LDAP_LOG( CONNECTION, ERR, 
1558                         "connection_input: conn %lu ber_peek_tag returns 0x%lx.\n",
1559                         conn->c_connid, tag, 0 );
1560 #else
1561                 Debug( LDAP_DEBUG_ANY, "ber_peek_tag returns 0x%lx\n",
1562                         tag, 0, 0 );
1563 #endif
1564                 ber_free( ber, 1 );
1565
1566                 return -1;
1567         }
1568
1569 #ifdef LDAP_CONNECTIONLESS
1570         if( conn->c_is_udp ) {
1571                 if( tag == LBER_OCTETSTRING ) {
1572                         ber_get_stringa( ber, &cdn );
1573                         tag = ber_peek_tag(ber, &len);
1574                 }
1575                 if( tag != LDAP_REQ_ABANDON && tag != LDAP_REQ_SEARCH ) {
1576 #ifdef NEW_LOGGING
1577                         LDAP_LOG( CONNECTION, ERR, 
1578                                 "connection_input: conn %lu invalid req for UDP 0x%lx.\n",
1579                                 conn->c_connid, tag, 0 );
1580 #else
1581                         Debug( LDAP_DEBUG_ANY, "invalid req for UDP 0x%lx\n", tag, 0, 0 );
1582 #endif
1583                         ber_free( ber, 1 );
1584                         return 0;
1585                 }
1586         }
1587 #endif
1588         if(tag == LDAP_REQ_BIND) {
1589                 /* immediately abandon all exiting operations upon BIND */
1590                 connection_abandon( conn );
1591         }
1592
1593         op = slap_op_alloc( ber, msgid, tag, conn->c_n_ops_received++ );
1594
1595         op->o_conn = conn;
1596         op->o_assertion = NULL;
1597         op->o_preread_attrs = NULL;
1598         op->o_postread_attrs = NULL;
1599         op->o_vrFilter = NULL;
1600         /* clear state if the connection is being reused from inactive */
1601         if ( conn->c_conn_state == SLAP_C_INACTIVE ) {
1602                 memset( &conn->c_pagedresults_state, 0, sizeof( conn->c_pagedresults_state ) );
1603         }
1604         op->o_pagedresults_state = conn->c_pagedresults_state;
1605
1606         op->o_res_ber = NULL;
1607
1608 #ifdef LDAP_CONNECTIONLESS
1609         if (conn->c_is_udp) {
1610                 if ( cdn ) {
1611                         ber_str2bv( cdn, 0, 1, &op->o_dn );
1612                         op->o_protocol = LDAP_VERSION2;
1613                 }
1614                 op->o_res_ber = ber_alloc_t( LBER_USE_DER );
1615                 if (op->o_res_ber == NULL) return 1;
1616
1617                 rc = ber_write( op->o_res_ber, (char *)&peeraddr,
1618                         sizeof(struct sockaddr), 0 );
1619
1620                 if (rc != sizeof(struct sockaddr)) {
1621 #ifdef NEW_LOGGING
1622                         LDAP_LOG( CONNECTION, INFO, 
1623                                 "connection_input: conn %lu ber_write failed\n",
1624                                 conn->c_connid, 0, 0 );
1625 #else
1626                         Debug( LDAP_DEBUG_ANY, "ber_write failed\n", 0, 0, 0 );
1627 #endif
1628                         return 1;
1629                 }
1630
1631                 if (op->o_protocol == LDAP_VERSION2) {
1632                         rc = ber_printf(op->o_res_ber, "{is{" /*}}*/, op->o_msgid, "");
1633                         if (rc == -1) {
1634 #ifdef NEW_LOGGING
1635                                 LDAP_LOG( CONNECTION, INFO, 
1636                                         "connection_input: conn %lu put outer sequence failed\n",
1637                                         conn->c_connid, 0, 0 );
1638 #else
1639                                 Debug( LDAP_DEBUG_ANY, "ber_write failed\n", 0, 0, 0 );
1640 #endif
1641                                 return rc;
1642                         }
1643                 }
1644         }
1645 #endif /* LDAP_CONNECTIONLESS */
1646
1647         rc = 0;
1648
1649         /* Don't process requests when the conn is in the middle of a
1650          * Bind, or if it's closing. Also, don't let any single conn
1651          * use up all the available threads, and don't execute if we're
1652          * currently blocked on output. And don't execute if there are
1653          * already pending ops, let them go first.  Abandon operations
1654          * get exceptions to some, but not all, cases.
1655          */
1656         if (tag != LDAP_REQ_ABANDON && conn->c_conn_state == SLAP_C_CLOSING) {
1657                 defer = "closing";
1658         } else if (tag != LDAP_REQ_ABANDON && conn->c_writewaiter) {
1659                 defer = "awaiting write";
1660         } else if (conn->c_n_ops_executing >= connection_pool_max/2) {
1661                 defer = "too many executing";
1662         } else if (conn->c_conn_state == SLAP_C_BINDING) {
1663                 defer = "binding";
1664         } else if (tag != LDAP_REQ_ABANDON && conn->c_n_ops_pending) {
1665                 defer = "pending operations";
1666         }
1667
1668         if( defer ) {
1669                 int max = conn->c_dn.bv_len
1670                         ? slap_conn_max_pending_auth
1671                         : slap_conn_max_pending;
1672
1673 #ifdef NEW_LOGGING
1674                 LDAP_LOG( CONNECTION, INFO, 
1675                         "connection_input: conn %lu deferring operation: %s\n",
1676                         conn->c_connid, defer, 0 );
1677 #else
1678                 Debug( LDAP_DEBUG_ANY,
1679                         "connection_input: conn=%lu deferring operation: %s\n",
1680                         conn->c_connid, defer, 0 );
1681 #endif
1682                 conn->c_n_ops_pending++;
1683                 LDAP_STAILQ_INSERT_TAIL( &conn->c_pending_ops, op, o_next );
1684                 if ( conn->c_n_ops_pending > max ) {
1685                         rc = -1;
1686                 } else {
1687                         rc = 1;
1688                 }
1689         } else {
1690                 conn->c_n_ops_executing++;
1691                 connection_op_activate( op );
1692         }
1693
1694 #ifdef NO_THREADS
1695         if ( conn->c_struct_state != SLAP_C_USED ) {
1696                 /* connection must have got closed underneath us */
1697                 return 1;
1698         }
1699 #endif
1700         assert( conn->c_struct_state == SLAP_C_USED );
1701
1702         return rc;
1703 }
1704
1705 static int
1706 connection_resched( Connection *conn )
1707 {
1708         Operation *op;
1709
1710         if( conn->c_conn_state == SLAP_C_CLOSING ) {
1711                 int rc;
1712                 ber_socket_t    sd;
1713                 ber_sockbuf_ctrl( conn->c_sb, LBER_SB_OPT_GET_FD, &sd );
1714
1715                 /* us trylock to avoid possible deadlock */
1716                 rc = ldap_pvt_thread_mutex_trylock( &connections_mutex );
1717
1718                 if( rc ) {
1719 #ifdef NEW_LOGGING
1720                         LDAP_LOG( CONNECTION, DETAIL1, 
1721                                 "connection_resched: conn %lu reaquiring locks.\n",
1722                                 conn->c_connid, 0, 0 );
1723 #else
1724                         Debug( LDAP_DEBUG_TRACE,
1725                                 "connection_resched: reaquiring locks conn=%lu sd=%d\n",
1726                                 conn->c_connid, sd, 0 );
1727 #endif
1728                         /*
1729                          * reaquire locks in the right order...
1730                          * this may allow another thread to close this connection,
1731                          * so recheck state below.
1732                          */
1733                         ldap_pvt_thread_mutex_unlock( &conn->c_mutex );
1734                         ldap_pvt_thread_mutex_lock( &connections_mutex );
1735                         ldap_pvt_thread_mutex_lock( &conn->c_mutex );
1736                 }
1737
1738                 if( conn->c_conn_state != SLAP_C_CLOSING ) {
1739 #ifdef NEW_LOGGING
1740                         LDAP_LOG( CONNECTION, INFO, 
1741                                 "connection_resched: conn %lu closed by other thread.\n",
1742                                 conn->c_connid, 0, 0 );
1743 #else
1744                         Debug( LDAP_DEBUG_TRACE, "connection_resched: "
1745                                 "closed by other thread conn=%lu sd=%d\n",
1746                                 conn->c_connid, sd, 0 );
1747 #endif
1748                 } else {
1749 #ifdef NEW_LOGGING
1750                         LDAP_LOG( CONNECTION, DETAIL1, 
1751                                 "connection_resched: conn %lu attempting closing.\n",
1752                                 conn->c_connid, 0, 0 );
1753 #else
1754                         Debug( LDAP_DEBUG_TRACE, "connection_resched: "
1755                                 "attempting closing conn=%lu sd=%d\n",
1756                                 conn->c_connid, sd, 0 );
1757 #endif
1758                         connection_close( conn );
1759                 }
1760
1761                 ldap_pvt_thread_mutex_unlock( &connections_mutex );
1762                 return 0;
1763         }
1764
1765         if( conn->c_conn_state != SLAP_C_ACTIVE || conn->c_writewaiter ) {
1766                 /* other states need different handling */
1767                 return 0;
1768         }
1769
1770         while ((op = LDAP_STAILQ_FIRST( &conn->c_pending_ops )) != NULL) {
1771                 if ( conn->c_n_ops_executing > connection_pool_max/2 ) {
1772                         break;
1773                 }
1774                 LDAP_STAILQ_REMOVE_HEAD( &conn->c_pending_ops, o_next );
1775                 LDAP_STAILQ_NEXT(op, o_next) = NULL;
1776                 /* pending operations should not be marked for abandonment */
1777                 assert(!op->o_abandon);
1778
1779                 conn->c_n_ops_pending--;
1780                 conn->c_n_ops_executing++;
1781
1782                 connection_op_activate( op );
1783
1784                 if ( conn->c_conn_state == SLAP_C_BINDING ) {
1785                         break;
1786                 }
1787         }
1788         return 0;
1789 }
1790
1791 static int connection_op_activate( Operation *op )
1792 {
1793         int status;
1794         ber_tag_t tag = op->o_tag;
1795
1796         if(tag == LDAP_REQ_BIND) {
1797                 op->o_conn->c_conn_state = SLAP_C_BINDING;
1798         }
1799
1800         if (!op->o_dn.bv_len) {
1801                 op->o_authz = op->o_conn->c_authz;
1802                 ber_dupbv( &op->o_dn, &op->o_conn->c_dn );
1803                 ber_dupbv( &op->o_ndn, &op->o_conn->c_ndn );
1804         }
1805         op->o_authtype = op->o_conn->c_authtype;
1806         ber_dupbv( &op->o_authmech, &op->o_conn->c_authmech );
1807         
1808         if (!op->o_protocol) {
1809                 op->o_protocol = op->o_conn->c_protocol
1810                         ? op->o_conn->c_protocol : LDAP_VERSION3;
1811         }
1812         if (op->o_conn->c_conn_state == SLAP_C_INACTIVE
1813                 && op->o_protocol > LDAP_VERSION2)
1814         {
1815                 op->o_conn->c_conn_state = SLAP_C_ACTIVE;
1816         }
1817
1818         op->o_connid = op->o_conn->c_connid;
1819
1820         LDAP_STAILQ_INSERT_TAIL( &op->o_conn->c_ops, op, o_next );
1821
1822         status = ldap_pvt_thread_pool_submit( &connection_pool,
1823                 connection_operation, (void *) op );
1824
1825         if ( status != 0 ) {
1826 #ifdef NEW_LOGGING
1827                 LDAP_LOG( CONNECTION, ERR, 
1828                         "connection_op_activate: conn %lu        thread pool submit failed.\n",
1829                         op->o_connid, 0, 0 );
1830 #else
1831                 Debug( LDAP_DEBUG_ANY,
1832                         "ldap_pvt_thread_pool_submit: failed (%d) for conn=%lu\n",
1833                         status, op->o_connid, 0 );
1834 #endif
1835                 /* should move op to pending list */
1836         }
1837
1838         return status;
1839 }
1840
1841 int connection_write(ber_socket_t s)
1842 {
1843         Connection *c;
1844
1845         assert( connections != NULL );
1846
1847         ldap_pvt_thread_mutex_lock( &connections_mutex );
1848
1849         c = connection_get( s );
1850
1851         slapd_clr_write( s, 0);
1852
1853         if( c == NULL ) {
1854 #ifdef NEW_LOGGING
1855                 LDAP_LOG( CONNECTION, ERR, 
1856                         "connection_write: sock %ld no connection!\n", (long)s, 0, 0);
1857 #else
1858                 Debug( LDAP_DEBUG_ANY,
1859                         "connection_write(%ld): no connection!\n",
1860                         (long)s, 0, 0 );
1861 #endif
1862                 slapd_remove(s, 1, 0);
1863                 ldap_pvt_thread_mutex_unlock( &connections_mutex );
1864                 return -1;
1865         }
1866
1867         c->c_n_write++;
1868
1869 #ifdef NEW_LOGGING
1870         LDAP_LOG( CONNECTION, DETAIL1, 
1871                 "connection_write conn %lu waking output.\n", c->c_connid, 0, 0 );
1872 #else
1873         Debug( LDAP_DEBUG_TRACE,
1874                 "connection_write(%d): waking output for id=%lu\n",
1875                 s, c->c_connid, 0 );
1876 #endif
1877         ldap_pvt_thread_cond_signal( &c->c_write_cv );
1878
1879         if ( ber_sockbuf_ctrl( c->c_sb, LBER_SB_OPT_NEEDS_READ, NULL ) ) {
1880                 slapd_set_read( s, 1 );
1881         }
1882         if ( ber_sockbuf_ctrl( c->c_sb, LBER_SB_OPT_NEEDS_WRITE, NULL ) ) {
1883                 slapd_set_write( s, 1 );
1884         }
1885         connection_return( c );
1886         ldap_pvt_thread_mutex_unlock( &connections_mutex );
1887         return 0;
1888 }
1889
1890 void
1891 connection_fake_init(
1892         Connection *conn,
1893         Operation *op,
1894         void *ctx )
1895 {
1896         conn->c_connid = -1;
1897         conn->c_send_ldap_result = slap_send_ldap_result;
1898         conn->c_send_search_entry = slap_send_search_entry;
1899         conn->c_send_search_reference = slap_send_search_reference;
1900         conn->c_listener = (Listener *)&dummy_list;
1901         conn->c_peer_domain = slap_empty_bv;
1902         conn->c_peer_name = slap_empty_bv;
1903
1904         /* set memory context */
1905         op->o_tmpmemctx = sl_mem_create( SLMALLOC_SLAB_SIZE, ctx );
1906         op->o_tmpmfuncs = &sl_mfuncs;
1907         op->o_threadctx = ctx;
1908
1909         op->o_conn = conn;
1910         op->o_connid = op->o_conn->c_connid;
1911
1912         op->o_time = slap_get_time();
1913 }
1914
1915 void
1916 connection_assign_nextid( Connection *conn )
1917 {
1918         ldap_pvt_thread_mutex_lock( &conn_nextid_mutex );
1919         conn->c_connid = conn_nextid++;
1920         ldap_pvt_thread_mutex_unlock( &conn_nextid_mutex );
1921 }
1922