]> git.sur5r.net Git - openldap/blob - servers/slapd/connection.c
Import memory leak fix from devel.
[openldap] / servers / slapd / connection.c
1 #include "portable.h"
2
3 #include <stdio.h>
4
5 #include <ac/errno.h>
6 #include <ac/signal.h>
7 #include <ac/socket.h>
8 #include <ac/string.h>
9 #include <ac/time.h>
10
11 #include "slap.h"
12
13 struct co_arg {
14         Connection      *co_conn;
15         Operation       *co_op;
16 };
17
18 /*
19  * connection_activity - handle the request operation op on connection
20  * conn.  This routine figures out what kind of operation it is and
21  * calls the appropriate stub to handle it.
22  */
23
24 static void *
25 connection_operation( void *arg_v )
26 {
27         struct co_arg   *arg = (struct co_arg *) arg_v;
28         unsigned long   len;
29
30         ldap_pvt_thread_mutex_lock( &arg->co_conn->c_opsmutex );
31         arg->co_conn->c_opsinitiated++;
32         ldap_pvt_thread_mutex_unlock( &arg->co_conn->c_opsmutex );
33
34         ldap_pvt_thread_mutex_lock( &ops_mutex );
35         ops_initiated++;
36         ldap_pvt_thread_mutex_unlock( &ops_mutex );
37
38         switch ( arg->co_op->o_tag ) {
39         case LDAP_REQ_BIND:
40                 do_bind( arg->co_conn, arg->co_op );
41                 break;
42
43 #ifdef LDAP_COMPAT30
44         case LDAP_REQ_UNBIND_30:
45 #endif
46         case LDAP_REQ_UNBIND:
47                 do_unbind( arg->co_conn, arg->co_op );
48                 break;
49
50         case LDAP_REQ_ADD:
51                 do_add( arg->co_conn, arg->co_op );
52                 break;
53
54 #ifdef LDAP_COMPAT30
55         case LDAP_REQ_DELETE_30:
56 #endif
57         case LDAP_REQ_DELETE:
58                 do_delete( arg->co_conn, arg->co_op );
59                 break;
60
61         case LDAP_REQ_MODRDN:
62                 do_modrdn( arg->co_conn, arg->co_op );
63                 break;
64
65         case LDAP_REQ_MODIFY:
66                 do_modify( arg->co_conn, arg->co_op );
67                 break;
68
69         case LDAP_REQ_COMPARE:
70                 do_compare( arg->co_conn, arg->co_op );
71                 break;
72
73         case LDAP_REQ_SEARCH:
74                 do_search( arg->co_conn, arg->co_op );
75                 break;
76
77 #ifdef LDAP_COMPAT30
78         case LDAP_REQ_ABANDON_30:
79 #endif
80         case LDAP_REQ_ABANDON:
81                 do_abandon( arg->co_conn, arg->co_op );
82                 break;
83
84         default:
85                 Debug( LDAP_DEBUG_ANY, "unknown request 0x%lx\n",
86                     arg->co_op->o_tag, 0, 0 );
87                 break;
88         }
89
90         ldap_pvt_thread_mutex_lock( &arg->co_conn->c_opsmutex );
91         arg->co_conn->c_opscompleted++;
92
93         slap_op_delete( &arg->co_conn->c_ops, arg->co_op );
94         arg->co_op = NULL;
95
96         ldap_pvt_thread_mutex_unlock( &arg->co_conn->c_opsmutex );
97
98         arg->co_conn = NULL;
99         free( (char *) arg );
100
101         ldap_pvt_thread_mutex_lock( &ops_mutex );
102         ops_completed++;
103         ldap_pvt_thread_mutex_unlock( &ops_mutex );
104
105         ldap_pvt_thread_mutex_lock( &active_threads_mutex );
106         active_threads--;
107         if( active_threads < 1 ) {
108                 ldap_pvt_thread_cond_signal(&active_threads_cond);
109         }
110         ldap_pvt_thread_mutex_unlock( &active_threads_mutex );
111
112         return NULL;
113 }
114
115 void
116 connection_activity(
117     Connection *conn
118 )
119 {
120         int status;
121         struct co_arg   *arg;
122         unsigned long   tag, len;
123         long            msgid;
124         BerElement      *ber;
125         char            *tmpdn;
126
127         if ( conn->c_currentber == NULL && (conn->c_currentber = ber_alloc())
128             == NULL ) {
129                 Debug( LDAP_DEBUG_ANY, "ber_alloc failed\n", 0, 0, 0 );
130                 return;
131         }
132
133         errno = 0;
134         if ( (tag = ber_get_next( &conn->c_sb, &len, conn->c_currentber ))
135             != LDAP_TAG_MESSAGE ) {
136                 Debug( LDAP_DEBUG_TRACE,
137                     "ber_get_next on fd %d failed errno %d (%s)\n",
138                     conn->c_sb.sb_sd, errno, errno > -1 && errno < sys_nerr ?
139                     sys_errlist[errno] : "unknown" );
140                 Debug( LDAP_DEBUG_TRACE, "*** got %ld of %lu so far\n",
141                     (long)(conn->c_currentber->ber_rwptr - conn->c_currentber->ber_buf),
142                     conn->c_currentber->ber_len, 0 );
143
144                 if ( errno != EWOULDBLOCK && errno != EAGAIN ) {
145                         /* log, close and send error */
146                         ber_free( conn->c_currentber, 1 );
147                         conn->c_currentber = NULL;
148
149                         close_connection( conn, conn->c_connid, -1 );
150                 }
151
152                 return;
153         }
154         ber = conn->c_currentber;
155         conn->c_currentber = NULL;
156
157         if ( (tag = ber_get_int( ber, &msgid )) != LDAP_TAG_MSGID ) {
158                 /* log, close and send error */
159                 Debug( LDAP_DEBUG_ANY, "ber_get_int returns 0x%lx\n", tag, 0,
160                     0 );
161                 ber_free( ber, 1 );
162
163                 close_connection( conn, conn->c_connid, -1 );
164                 return;
165         }
166
167         if ( (tag = ber_peek_tag( ber, &len )) == LBER_ERROR ) {
168                 /* log, close and send error */
169                 Debug( LDAP_DEBUG_ANY, "ber_peek_tag returns 0x%lx\n", tag, 0,
170                     0 );
171                 ber_free( ber, 1 );
172
173                 close_connection( conn, conn->c_connid, -1 );
174                 return;
175         }
176
177 #ifdef LDAP_COMPAT30
178         if ( conn->c_version == 30 ) {
179                 (void) ber_skip_tag( ber, &len );
180         }
181 #endif
182
183         arg = (struct co_arg *) ch_malloc( sizeof(struct co_arg) );
184         arg->co_conn = conn;
185
186         ldap_pvt_thread_mutex_lock( &conn->c_dnmutex );
187         if ( conn->c_dn != NULL ) {
188                 tmpdn = ch_strdup( conn->c_dn );
189         } else {
190                 tmpdn = NULL;
191         }
192         ldap_pvt_thread_mutex_unlock( &conn->c_dnmutex );
193
194         ldap_pvt_thread_mutex_lock( &conn->c_opsmutex );
195         arg->co_op = slap_op_add( &conn->c_ops, ber, msgid, tag, tmpdn,
196             conn->c_opsinitiated, conn->c_connid );
197         ldap_pvt_thread_mutex_unlock( &conn->c_opsmutex );
198
199         if ( tmpdn != NULL ) {
200                 free( tmpdn );
201         }
202
203         ldap_pvt_thread_mutex_lock( &active_threads_mutex );
204         active_threads++;
205         ldap_pvt_thread_mutex_unlock( &active_threads_mutex );
206
207         if ( status = ldap_pvt_thread_create( &arg->co_op->o_tid, 1,
208             connection_operation, (void *) arg ) != 0 ) {
209                 Debug( LDAP_DEBUG_ANY, "ldap_pvt_thread_create failed (%d)\n", status, 0, 0 );
210         }
211 }