git.sur5r.net Git - openldap/blob - 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_pvt_thread_mutex_unlock( &op->o_conn->c_mutex );
  72                         /* TODO: We could instead remove the cancelled operation
  73                          * from c_pending_ops like Abandon does, and send its
  74                          * response here.  Not if it is pending because of a
  75                          * congested connection though.
  76                          */
  77                         rs->sr_text = "too busy for Cancel, try Abandon instead";
  78                         return LDAP_CANNOT_CANCEL;
  79                 }
  80         }
  81
  82         LDAP_STAILQ_FOREACH( o, &op->o_conn->c_ops, o_next ) {
  83                 if ( o->o_msgid == opid ) {
  84                         break;
  85                 }
  86         }
  87
  88         if ( o == NULL ) {
  89                 rc = LDAP_NO_SUCH_OPERATION;
  90                 rs->sr_text = "message ID not found";
  91         } else if ( o->o_cancel != SLAP_CANCEL_NONE ) {
  92                 rc = LDAP_PROTOCOL_ERROR;
  93                 rs->sr_text = "message ID already being cancelled";
  94         } else {
  95                 rc = LDAP_SUCCESS;
  96                 o->o_cancel = SLAP_CANCEL_REQ;
  97                 o->o_abandon = 1;
  98         }
  99
 100         ldap_pvt_thread_mutex_unlock( &op->o_conn->c_mutex );
 101
 102         if ( rc == LDAP_SUCCESS ) {
 103                 LDAP_STAILQ_FOREACH( op->o_bd, &backendDB, be_next ) {
 104                         if( !op->o_bd->be_cancel ) continue;
 105
 106                         op->oq_cancel.rs_msgid = opid;
 107                         if ( op->o_bd->be_cancel( op, rs ) == LDAP_SUCCESS ) {
 108                                 return LDAP_SUCCESS;
 109                         }
 110                 }
 111
 112                 while ( o->o_cancel == SLAP_CANCEL_REQ ) {
 113                         ldap_pvt_thread_yield();
 114                 }
 115
 116                 if ( o->o_cancel == SLAP_CANCEL_ACK ) {
 117                         rc = LDAP_SUCCESS;
 118                 } else {
 119                         rc = o->o_cancel;
 120                 }
 121
 122                 o->o_cancel = SLAP_CANCEL_DONE;
 123         }
 124
 125         return rc;
 126 }