]> git.sur5r.net Git - openldap/blob - servers/slapd/cancel.c
For ITS#6104: Protect initial o_abandon/o_cancel with op->o_conn->c_mutex.
[openldap] / servers / slapd / cancel.c
1 /* cancel.c - LDAP cancel extended operation */
2 /* $OpenLDAP$ */
3 /* This work is part of OpenLDAP Software <http://www.openldap.org/>.
4  *
5  * Copyright 1998-2009 The OpenLDAP Foundation.
6  * All rights reserved.
7  *
8  * Redistribution and use in source and binary forms, with or without
9  * modification, are permitted only as authorized by the OpenLDAP
10  * Public License.
11  *
12  * A copy of this license is available in the file LICENSE in the
13  * top-level directory of the distribution or, alternatively, at
14  * <http://www.OpenLDAP.org/license.html>.
15  */
16
17 #include "portable.h"
18
19 #include <stdio.h>
20
21 #include <ac/socket.h>
22 #include <ac/string.h>
23 #include <ac/unistd.h>
24
25 #include "slap.h"
26
27 #include <lber_pvt.h>
28 #include <lutil.h>
29
30 const struct berval slap_EXOP_CANCEL = BER_BVC(LDAP_EXOP_CANCEL);
31
32 int cancel_extop( Operation *op, SlapReply *rs )
33 {
34         Operation *o;
35         int rc;
36         int opid;
37         BerElement *ber;
38
39         assert( ber_bvcmp( &slap_EXOP_CANCEL, &op->ore_reqoid ) == 0 );
40
41         if ( op->ore_reqdata == NULL ) {
42                 rs->sr_text = "no message ID supplied";
43                 return LDAP_PROTOCOL_ERROR;
44         }
45
46         ber = ber_init( op->ore_reqdata );
47         if ( ber == NULL ) {
48                 rs->sr_text = "internal error";
49                 return LDAP_OTHER;
50         }
51
52         if ( ber_scanf( ber, "{i}", &opid ) == LBER_ERROR ) {
53                 rs->sr_text = "message ID parse failed";
54                 return LDAP_PROTOCOL_ERROR;
55         }
56
57         (void) ber_free( ber, 1 );
58
59         Statslog( LDAP_DEBUG_STATS, "%s CANCEL msg=%d\n",
60                 op->o_log_prefix, opid, 0, 0, 0 );
61
62         if ( opid < 0 ) {
63                 rs->sr_text = "message ID invalid";
64                 return LDAP_PROTOCOL_ERROR;
65         }
66
67         ldap_pvt_thread_mutex_lock( &op->o_conn->c_mutex );
68
69         LDAP_STAILQ_FOREACH( o, &op->o_conn->c_pending_ops, o_next ) {
70                 if ( o->o_msgid == opid ) {
71                         LDAP_STAILQ_REMOVE( &op->o_conn->c_pending_ops, o, Operation, o_next );
72                         LDAP_STAILQ_NEXT(o, o_next) = NULL;
73                         op->o_conn->c_n_ops_pending--;
74                         slap_op_free( o, NULL );
75                         ldap_pvt_thread_mutex_unlock( &op->o_conn->c_mutex );
76                         return LDAP_SUCCESS;
77                 }
78         }
79
80         LDAP_STAILQ_FOREACH( o, &op->o_conn->c_ops, o_next ) {
81                 if ( o->o_msgid == opid ) {
82                         break;
83                 }
84         }
85
86         if ( o == NULL ) {
87                 rc = LDAP_NO_SUCH_OPERATION;
88                 rs->sr_text = "message ID not found";
89         } else if ( o->o_cancel != SLAP_CANCEL_NONE ) {
90                 rc = LDAP_PROTOCOL_ERROR;
91                 rs->sr_text = "message ID already being cancelled";
92         } else {
93                 rc = LDAP_SUCCESS;
94                 o->o_cancel = SLAP_CANCEL_REQ;
95                 o->o_abandon = 1;
96         }
97
98         ldap_pvt_thread_mutex_unlock( &op->o_conn->c_mutex );
99
100         if ( rc == LDAP_SUCCESS ) {
101                 LDAP_STAILQ_FOREACH( op->o_bd, &backendDB, be_next ) {
102                         if( !op->o_bd->be_cancel ) continue;
103
104                         op->oq_cancel.rs_msgid = opid;
105                         if ( op->o_bd->be_cancel( op, rs ) == LDAP_SUCCESS ) {
106                                 return LDAP_SUCCESS;
107                         }
108                 }
109
110                 while ( o->o_cancel == SLAP_CANCEL_REQ ) {
111                         ldap_pvt_thread_yield();
112                 }
113
114                 if ( o->o_cancel == SLAP_CANCEL_ACK ) {
115                         rc = LDAP_SUCCESS;
116                 } else {
117                         rc = o->o_cancel;
118                 }
119
120                 o->o_cancel = SLAP_CANCEL_DONE;
121         }
122
123         return rc;
124 }