mirror of
https://git.openldap.org/openldap/openldap.git
synced 2024-12-21 03:10:25 +08:00
156 lines
3.8 KiB
C
156 lines
3.8 KiB
C
/* $OpenLDAP$ */
|
|
/* This work is part of OpenLDAP Software <http://www.openldap.org/>.
|
|
*
|
|
* Copyright 1998-2020 The OpenLDAP Foundation.
|
|
* All rights reserved.
|
|
*
|
|
* Redistribution and use in source and binary forms, with or without
|
|
* modification, are permitted only as authorized by the OpenLDAP
|
|
* Public License.
|
|
*
|
|
* A copy of this license is available in the file LICENSE in the
|
|
* top-level directory of the distribution or, alternatively, at
|
|
* <http://www.OpenLDAP.org/license.html>.
|
|
*/
|
|
|
|
#include "portable.h"
|
|
|
|
#include "lutil.h"
|
|
#include "slap.h"
|
|
|
|
int
|
|
operation_client_cmp( const void *left, const void *right )
|
|
{
|
|
const Operation *l = left, *r = right;
|
|
|
|
assert( l->o_client == r->o_client );
|
|
return ( l->o_client_msgid < r->o_client_msgid ) ?
|
|
-1 :
|
|
( l->o_client_msgid > r->o_client_msgid );
|
|
}
|
|
|
|
int
|
|
operation_upstream_cmp( const void *left, const void *right )
|
|
{
|
|
const Operation *l = left, *r = right;
|
|
|
|
assert( l->o_upstream == r->o_upstream );
|
|
return ( l->o_upstream_msgid < r->o_upstream_msgid ) ?
|
|
-1 :
|
|
( l->o_upstream_msgid > r->o_upstream_msgid );
|
|
}
|
|
|
|
void
|
|
operation_destroy( Operation *op )
|
|
{
|
|
Connection *c;
|
|
|
|
/* TODO: this is a stopgap and there are many races here, just get
|
|
* something in to test with until we implement the freelist */
|
|
if ( op->o_client ) {
|
|
c = op->o_client;
|
|
ldap_pvt_thread_mutex_lock( &c->c_mutex );
|
|
tavl_delete( &c->c_ops, op, operation_client_cmp );
|
|
ldap_pvt_thread_mutex_unlock( &c->c_mutex );
|
|
}
|
|
|
|
if ( op->o_upstream ) {
|
|
c = op->o_upstream;
|
|
ldap_pvt_thread_mutex_lock( &c->c_mutex );
|
|
tavl_delete( &c->c_ops, op, operation_upstream_cmp );
|
|
ldap_pvt_thread_mutex_unlock( &c->c_mutex );
|
|
}
|
|
|
|
ch_free( op );
|
|
}
|
|
|
|
Operation *
|
|
operation_init( Connection *c, BerElement *ber )
|
|
{
|
|
Operation *op;
|
|
ber_tag_t tag;
|
|
ber_len_t len;
|
|
int rc;
|
|
|
|
op = ch_calloc( 1, sizeof(Operation) );
|
|
op->o_client = c;
|
|
|
|
tag = ber_get_int( ber, &op->o_client_msgid );
|
|
if ( tag != LDAP_TAG_MSGID ) {
|
|
goto fail;
|
|
}
|
|
|
|
rc = tavl_insert( &c->c_ops, op, operation_client_cmp, avl_dup_error );
|
|
if ( rc ) {
|
|
Debug( LDAP_DEBUG_PACKETS, "operation_init: "
|
|
"several operations with same msgid=%d in-flight "
|
|
"from the client\n",
|
|
op->o_client_msgid );
|
|
goto fail;
|
|
}
|
|
|
|
tag = op->o_tag = ber_skip_element( ber, &op->o_request );
|
|
switch ( tag ) {
|
|
case LBER_ERROR:
|
|
rc = -1;
|
|
break;
|
|
}
|
|
if ( rc ) {
|
|
tavl_delete( &c->c_ops, op, operation_client_cmp );
|
|
goto fail;
|
|
}
|
|
|
|
tag = ber_peek_tag( ber, &len );
|
|
if ( tag == LDAP_TAG_CONTROLS ) {
|
|
ber_skip_element( ber, &op->o_ctrls );
|
|
}
|
|
|
|
return op;
|
|
|
|
fail:
|
|
ch_free( op );
|
|
return NULL;
|
|
}
|
|
|
|
void *
|
|
operation_process( void *ctx, void *arg )
|
|
{
|
|
Operation *op = arg;
|
|
BerElement *output;
|
|
Connection *c;
|
|
ber_int_t msgid;
|
|
int rc;
|
|
|
|
c = backend_select( op );
|
|
if ( !c ) {
|
|
Debug( LDAP_DEBUG_STATS, "operation_process: "
|
|
"no available connection found\n" );
|
|
goto fail;
|
|
}
|
|
op->o_upstream = c;
|
|
|
|
c->c_pendingber = output = ber_alloc();
|
|
if ( !output ) {
|
|
goto fail;
|
|
}
|
|
|
|
op->o_upstream_msgid = msgid = c->c_next_msgid++;
|
|
rc = tavl_insert( &c->c_ops, op, operation_upstream_cmp, avl_dup_error );
|
|
assert( rc == LDAP_SUCCESS );
|
|
|
|
ber_start_seq( output, LDAP_TAG_MESSAGE );
|
|
ber_put_int( output, msgid, LDAP_TAG_MSGID );
|
|
ber_put_berval( output, &op->o_request, op->o_tag );
|
|
if ( !BER_BVISNULL( &op->o_ctrls ) ) {
|
|
ber_put_berval( output, &op->o_ctrls, LDAP_TAG_CONTROLS );
|
|
}
|
|
ber_put_seq( output );
|
|
|
|
ldap_pvt_thread_mutex_unlock( &c->c_mutex );
|
|
upstream_write_cb( -1, 0, c );
|
|
|
|
return NULL;
|
|
fail:
|
|
return NULL;
|
|
}
|