X-Git-Url: https://git.sur5r.net/?a=blobdiff_plain;f=servers%2Fslapd%2Fback-sql%2Fsql-wrap.c;h=caa9d5896ebb7c204b8aa9a9321fc0d2acbf775f;hb=f51efd4b0345baea1a9c977d0763292891c37436;hp=7e174bacf888c6bad5a3ad0e9e7a2bd98ecb98f4;hpb=fd5ded3afba14e818b912110c5b669a1ab4e5ad1;p=openldap diff --git a/servers/slapd/back-sql/sql-wrap.c b/servers/slapd/back-sql/sql-wrap.c index 7e174bacf8..caa9d5896e 100644 --- a/servers/slapd/back-sql/sql-wrap.c +++ b/servers/slapd/back-sql/sql-wrap.c @@ -1,8 +1,10 @@ /* $OpenLDAP$ */ /* This work is part of OpenLDAP Software . * - * Copyright 1999-2003 The OpenLDAP Foundation. + * Copyright 1999-2012 The OpenLDAP Foundation. * Portions Copyright 1999 Dmitry Kovalev. + * Portions Copyright 2002 Pierangelo Masarati. + * Portions Copyright 2004 Mark Adamson. * All rights reserved. * * Redistribution and use in source and binary forms, with or without @@ -15,30 +17,21 @@ */ /* ACKNOWLEDGEMENTS: * This work was initially developed by Dmitry Kovalev for inclusion - * by OpenLDAP Software. + * by OpenLDAP Software. Additional significant contributors include + * Pierangelo Masarati and Mark Adamson. */ #include "portable.h" -#ifdef SLAPD_SQL - #include #include "ac/string.h" #include + #include "slap.h" -#include "ldap_pvt.h" -#include "back-sql.h" -#include "sql-types.h" -#include "sql-wrap.h" -#include "schema-map.h" +#include "proto-sql.h" #define MAX_ATTR_LEN 16384 -typedef struct backsql_conn { - int ldap_cid; - SQLHDBC dbh; -} backsql_db_conn; - void backsql_PrintErrors( SQLHENV henv, SQLHDBC hdbc, SQLHSTMT sth, int rc ) { @@ -49,26 +42,19 @@ backsql_PrintErrors( SQLHENV henv, SQLHDBC hdbc, SQLHSTMT sth, int rc ) Debug( LDAP_DEBUG_TRACE, "Return code: %d\n", rc, 0, 0 ); - - rc = SQLError( henv, hdbc, sth, state, &iSqlCode, msg, - SQL_MAX_MESSAGE_LENGTH - 1, &len ); - for ( ; BACKSQL_SUCCESS( rc ); ) { - Debug( LDAP_DEBUG_TRACE, "Native error code: %d\n", - (int)iSqlCode, 0, 0 ); - Debug( LDAP_DEBUG_TRACE, "SQL engine state: %s\n", - state, 0, 0 ); - Debug( LDAP_DEBUG_TRACE, "Message: %s\n", msg, 0, 0 ); - rc = SQLError( henv, hdbc, sth, state, &iSqlCode, msg, - SQL_MAX_MESSAGE_LENGTH - 1, &len ); + for ( ; rc = SQLError( henv, hdbc, sth, state, &iSqlCode, msg, + SQL_MAX_MESSAGE_LENGTH - 1, &len ), BACKSQL_SUCCESS( rc ); ) + { + Debug( LDAP_DEBUG_TRACE, + " nativeErrCode=%d SQLengineState=%s msg=\"%s\"\n", + (int)iSqlCode, state, msg ); } } RETCODE -backsql_Prepare( SQLHDBC dbh, SQLHSTMT *sth, char *query, int timeout ) +backsql_Prepare( SQLHDBC dbh, SQLHSTMT *sth, const char *query, int timeout ) { RETCODE rc; - char drv_name[ 30 ]; - SWORD len; rc = SQLAllocStmt( dbh, sth ); if ( rc != SQL_SUCCESS ) { @@ -79,34 +65,43 @@ backsql_Prepare( SQLHDBC dbh, SQLHSTMT *sth, char *query, int timeout ) Debug( LDAP_DEBUG_TRACE, "==>backsql_Prepare()\n", 0, 0, 0 ); #endif /* BACKSQL_TRACE */ - SQLGetInfo( dbh, SQL_DRIVER_NAME, drv_name, sizeof( drv_name ), &len ); +#ifdef BACKSQL_MSSQL_WORKAROUND + { + char drv_name[ 30 ]; + SWORD len; + + SQLGetInfo( dbh, SQL_DRIVER_NAME, drv_name, sizeof( drv_name ), &len ); #ifdef BACKSQL_TRACE - Debug( LDAP_DEBUG_TRACE, "backsql_Prepare(): driver name='%s'\n", - drv_name, 0, 0 ); + Debug( LDAP_DEBUG_TRACE, "backsql_Prepare(): driver name=\"%s\"\n", + drv_name, 0, 0 ); #endif /* BACKSQL_TRACE */ - ldap_pvt_str2upper( drv_name ); - if ( !strncmp( drv_name, "SQLSRV32.DLL", sizeof( drv_name ) ) ) { - /* - * stupid default result set in MS SQL Server - * does not support multiple active statements - * on the same connection -- so we are trying - * to make it not to use default result set... - */ - Debug( LDAP_DEBUG_TRACE, "_SQLprepare(): " - "enabling MS SQL Server default result " - "set workaround\n", 0, 0, 0 ); - rc = SQLSetStmtOption( *sth, SQL_CONCURRENCY, - SQL_CONCUR_ROWVER ); - if ( rc != SQL_SUCCESS && rc != SQL_SUCCESS_WITH_INFO ) { - Debug( LDAP_DEBUG_TRACE, "backsql_Prepare(): " - "SQLSetStmtOption(SQL_CONCURRENCY," - "SQL_CONCUR_ROWVER) failed:\n", - 0, 0, 0 ); - backsql_PrintErrors( SQL_NULL_HENV, dbh, *sth, rc ); + ldap_pvt_str2upper( drv_name ); + if ( !strncmp( drv_name, "SQLSRV32.DLL", STRLENOF( "SQLSRV32.DLL" ) ) ) { + /* + * stupid default result set in MS SQL Server + * does not support multiple active statements + * on the same connection -- so we are trying + * to make it not to use default result set... + */ + Debug( LDAP_DEBUG_TRACE, "_SQLprepare(): " + "enabling MS SQL Server default result " + "set workaround\n", 0, 0, 0 ); + rc = SQLSetStmtOption( *sth, SQL_CONCURRENCY, + SQL_CONCUR_ROWVER ); + if ( rc != SQL_SUCCESS && rc != SQL_SUCCESS_WITH_INFO ) { + Debug( LDAP_DEBUG_TRACE, "backsql_Prepare(): " + "SQLSetStmtOption(SQL_CONCURRENCY," + "SQL_CONCUR_ROWVER) failed:\n", + 0, 0, 0 ); + backsql_PrintErrors( SQL_NULL_HENV, dbh, *sth, rc ); + SQLFreeStmt( *sth, SQL_DROP ); + return rc; + } } } +#endif /* BACKSQL_MSSQL_WORKAROUND */ if ( timeout > 0 ) { Debug( LDAP_DEBUG_TRACE, "_SQLprepare(): " @@ -115,6 +110,8 @@ backsql_Prepare( SQLHDBC dbh, SQLHSTMT *sth, char *query, int timeout ) rc = SQLSetStmtOption( *sth, SQL_QUERY_TIMEOUT, timeout ); if ( rc != SQL_SUCCESS ) { backsql_PrintErrors( SQL_NULL_HENV, dbh, *sth, rc ); + SQLFreeStmt( *sth, SQL_DROP ); + return rc; } } @@ -123,42 +120,13 @@ backsql_Prepare( SQLHDBC dbh, SQLHSTMT *sth, char *query, int timeout ) 0, 0, 0 ); #endif /* BACKSQL_TRACE */ - return SQLPrepare( *sth, query, SQL_NTS ); -} - -#if 0 -/* - * Turned into macros --- see sql-wrap.h - */ -RETCODE -backsql_BindParamStr( SQLHSTMT sth, int par_ind, char *str, int maxlen ) -{ - RETCODE rc; - - rc = SQLBindParameter( sth, (SQLUSMALLINT)par_ind, SQL_PARAM_INPUT, - SQL_C_CHAR, SQL_VARCHAR, - (SQLUINTEGER)maxlen, 0, (SQLPOINTER)str, - (SQLUINTEGER)maxlen, NULL ); - return rc; -} - -RETCODE -backsql_BindParamID( SQLHSTMT sth, int par_ind, unsigned long *id ) -{ - return SQLBindParameter( sth, (SQLUSMALLINT)par_ind, - SQL_PARAM_INPUT, SQL_C_ULONG, SQL_INTEGER, - 0, 0, (SQLPOINTER)id, 0, (SQLINTEGER*)NULL ); + return SQLPrepare( *sth, (SQLCHAR *)query, SQL_NTS ); } -#endif RETCODE -backsql_BindRowAsStrings( SQLHSTMT sth, BACKSQL_ROW_NTS *row ) +backsql_BindRowAsStrings_x( SQLHSTMT sth, BACKSQL_ROW_NTS *row, void *ctx ) { RETCODE rc; - SQLCHAR colname[ 64 ]; - SQLSMALLINT name_len, col_type, col_scale, col_null; - UDWORD col_prec; - int i; if ( row == NULL ) { return SQL_ERROR; @@ -176,69 +144,129 @@ backsql_BindRowAsStrings( SQLHSTMT sth, BACKSQL_ROW_NTS *row ) #endif /* BACKSQL_TRACE */ backsql_PrintErrors( SQL_NULL_HENV, SQL_NULL_HDBC, sth, rc ); + } else { + SQLCHAR colname[ 64 ]; + SQLSMALLINT name_len, col_type, col_scale, col_null; + SQLLEN col_prec; + int i; + #ifdef BACKSQL_TRACE Debug( LDAP_DEBUG_TRACE, "backsql_BindRowAsStrings: " "ncols=%d\n", (int)row->ncols, 0, 0 ); #endif /* BACKSQL_TRACE */ - row->col_names = (BerVarray)ch_calloc( row->ncols + 1, - sizeof( struct berval ) ); - row->cols = (char **)ch_calloc( row->ncols + 1, - sizeof( char * ) ); - row->col_prec = (UDWORD *)ch_calloc( row->ncols, - sizeof( UDWORD ) ); - row->value_len = (SQLINTEGER *)ch_calloc( row->ncols, - sizeof( SQLINTEGER ) ); - for ( i = 1; i <= row->ncols; i++ ) { - rc = SQLDescribeCol( sth, (SQLSMALLINT)i, &colname[ 0 ], + row->col_names = (BerVarray)ber_memcalloc_x( row->ncols + 1, + sizeof( struct berval ), ctx ); + if ( row->col_names == NULL ) { + goto nomem; + } + + row->col_prec = (UDWORD *)ber_memcalloc_x( row->ncols, + sizeof( UDWORD ), ctx ); + if ( row->col_prec == NULL ) { + goto nomem; + } + + row->col_type = (SQLSMALLINT *)ber_memcalloc_x( row->ncols, + sizeof( SQLSMALLINT ), ctx ); + if ( row->col_type == NULL ) { + goto nomem; + } + + row->cols = (char **)ber_memcalloc_x( row->ncols + 1, + sizeof( char * ), ctx ); + if ( row->cols == NULL ) { + goto nomem; + } + + row->value_len = (SQLLEN *)ber_memcalloc_x( row->ncols, + sizeof( SQLLEN ), ctx ); + if ( row->value_len == NULL ) { + goto nomem; + } + + if ( 0 ) { +nomem: + ber_memfree_x( row->col_names, ctx ); + row->col_names = NULL; + ber_memfree_x( row->col_prec, ctx ); + row->col_prec = NULL; + ber_memfree_x( row->col_type, ctx ); + row->col_type = NULL; + ber_memfree_x( row->cols, ctx ); + row->cols = NULL; + ber_memfree_x( row->value_len, ctx ); + row->value_len = NULL; + + Debug( LDAP_DEBUG_ANY, "backsql_BindRowAsStrings: " + "out of memory\n", 0, 0, 0 ); + + return LDAP_NO_MEMORY; + } + + for ( i = 0; i < row->ncols; i++ ) { + SQLSMALLINT TargetType; + + rc = SQLDescribeCol( sth, (SQLSMALLINT)(i + 1), &colname[ 0 ], (SQLUINTEGER)( sizeof( colname ) - 1 ), &name_len, &col_type, &col_prec, &col_scale, &col_null ); - ber_str2bv( colname, 0, 1, &row->col_names[ i - 1 ] ); + /* FIXME: test rc? */ + + ber_str2bv_x( (char *)colname, 0, 1, + &row->col_names[ i ], ctx ); #ifdef BACKSQL_TRACE Debug( LDAP_DEBUG_TRACE, "backsql_BindRowAsStrings: " "col_name=%s, col_prec[%d]=%d\n", - colname, (int)i, (int)col_prec ); + colname, (int)(i + 1), (int)col_prec ); #endif /* BACKSQL_TRACE */ - if ( col_type == SQL_LONGVARCHAR - || col_type == SQL_LONGVARBINARY) { -#if 0 - row->cols[ i - 1 ] = NULL; - row->col_prec[ i - 1 ] = -1; - - /* - * such fields must be handled - * in some other way since they return 2G - * as their precision (at least it does so - * with MS SQL Server w/native driver) - * for now, we just set fixed precision - * for such fields - dirty hack, but... - * no time to deal with SQLGetData() - */ -#endif + if ( col_type != SQL_CHAR && col_type != SQL_VARCHAR ) + { col_prec = MAX_ATTR_LEN; - row->cols[ i - 1 ] = (char *)ch_calloc( col_prec + 1, sizeof( char ) ); - row->col_prec[ i - 1 ] = col_prec; - rc = SQLBindCol( sth, (SQLUSMALLINT)i, - SQL_C_CHAR, - (SQLPOINTER)row->cols[ i - 1 ], - col_prec + 1, - &row->value_len[ i - 1 ] ); + } + + row->cols[ i ] = (char *)ber_memcalloc_x( col_prec + 1, + sizeof( char ), ctx ); + row->col_prec[ i ] = col_prec; + row->col_type[ i ] = col_type; + + /* + * ITS#3386, ITS#3113 - 20070308 + * Note: there are many differences between various DPMS and ODBC + * Systems; some support SQL_C_BLOB, SQL_C_BLOB_LOCATOR. YMMV: + * This has only been tested on Linux/MySQL/UnixODBC + * For BINARY-type Fields (BLOB, etc), read the data as BINARY + */ + if ( BACKSQL_IS_BINARY( col_type ) ) { +#ifdef BACKSQL_TRACE + Debug( LDAP_DEBUG_TRACE, "backsql_BindRowAsStrings: " + "col_name=%s, col_type[%d]=%d: reading binary data\n", + colname, (int)(i + 1), (int)col_type); +#endif /* BACKSQL_TRACE */ + TargetType = SQL_C_BINARY; + } else { - row->cols[ i - 1 ] = (char *)ch_calloc( col_prec + 1, sizeof( char ) ); - row->col_prec[ i - 1 ] = col_prec; - rc = SQLBindCol( sth, (SQLUSMALLINT)i, - SQL_C_CHAR, - (SQLPOINTER)row->cols[ i - 1 ], - col_prec + 1, - &row->value_len[ i - 1 ] ); + /* Otherwise read it as Character data */ +#ifdef BACKSQL_TRACE + Debug( LDAP_DEBUG_TRACE, "backsql_BindRowAsStrings: " + "col_name=%s, col_type[%d]=%d: reading character data\n", + colname, (int)(i + 1), (int)col_type); +#endif /* BACKSQL_TRACE */ + TargetType = SQL_C_CHAR; } + + rc = SQLBindCol( sth, (SQLUSMALLINT)(i + 1), + TargetType, + (SQLPOINTER)row->cols[ i ], + col_prec + 1, + &row->value_len[ i ] ); + + /* FIXME: test rc? */ } - row->col_names[ i - 1 ].bv_val = NULL; - row->col_names[ i - 1 ].bv_len = 0; - row->cols[ i - 1 ] = NULL; + BER_BVZERO( &row->col_names[ i ] ); + row->cols[ i ] = NULL; } #ifdef BACKSQL_TRACE @@ -249,39 +277,43 @@ backsql_BindRowAsStrings( SQLHSTMT sth, BACKSQL_ROW_NTS *row ) } RETCODE -backsql_FreeRow( BACKSQL_ROW_NTS *row ) +backsql_BindRowAsStrings( SQLHSTMT sth, BACKSQL_ROW_NTS *row ) +{ + return backsql_BindRowAsStrings_x( sth, row, NULL ); +} + +RETCODE +backsql_FreeRow_x( BACKSQL_ROW_NTS *row, void *ctx ) { if ( row->cols == NULL ) { return SQL_ERROR; } - ber_bvarray_free( row->col_names ); - ldap_charray_free( row->cols ); - free( row->col_prec ); - free( row->value_len ); + ber_bvarray_free_x( row->col_names, ctx ); + ber_memfree_x( row->col_prec, ctx ); + ber_memfree_x( row->col_type, ctx ); + ber_memvfree_x( (void **)row->cols, ctx ); + ber_memfree_x( row->value_len, ctx ); return SQL_SUCCESS; } -static int -backsql_cmp_connid( const void *v_c1, const void *v_c2 ) + +RETCODE +backsql_FreeRow( BACKSQL_ROW_NTS *row ) { - const backsql_db_conn *c1 = v_c1, *c2 = v_c2; - if ( c1->ldap_cid > c2->ldap_cid ) { - return 1; - } - - if ( c1->ldap_cid < c2->ldap_cid ) { - return -1; - } - - return 0; + return backsql_FreeRow_x( row, NULL ); } -static int -backsql_close_db_conn( backsql_db_conn *conn ) +static void +backsql_close_db_handle( SQLHDBC dbh ) { - Debug( LDAP_DEBUG_TRACE, "==>backsql_close_db_conn()\n", 0, 0, 0 ); + if ( dbh == SQL_NULL_HDBC ) { + return; + } + + Debug( LDAP_DEBUG_TRACE, "==>backsql_close_db_handle(%p)\n", + (void *)dbh, 0, 0 ); /* * Default transact is SQL_ROLLBACK; commit is required only @@ -290,39 +322,50 @@ backsql_close_db_conn( backsql_db_conn *conn ) */ /* TimesTen */ - SQLTransact( SQL_NULL_HENV, conn->dbh, SQL_ROLLBACK ); - SQLDisconnect( conn->dbh ); - SQLFreeConnect( conn->dbh ); - Debug( LDAP_DEBUG_TRACE, "<==backsql_close_db_conn()\n", 0, 0, 0 ); - return 1; + SQLTransact( SQL_NULL_HENV, dbh, SQL_ROLLBACK ); + SQLDisconnect( dbh ); + SQLFreeConnect( dbh ); + + Debug( LDAP_DEBUG_TRACE, "<==backsql_close_db_handle(%p)\n", + (void *)dbh, 0, 0 ); +} + +int +backsql_conn_destroy( + backsql_info *bi ) +{ + return 0; } int -backsql_init_db_env( backsql_info *si ) +backsql_init_db_env( backsql_info *bi ) { RETCODE rc; + int ret = SQL_SUCCESS; Debug( LDAP_DEBUG_TRACE, "==>backsql_init_db_env()\n", 0, 0, 0 ); - rc = SQLAllocEnv( &si->db_env ); + + rc = SQLAllocEnv( &bi->sql_db_env ); if ( rc != SQL_SUCCESS ) { Debug( LDAP_DEBUG_TRACE, "init_db_env: SQLAllocEnv failed:\n", 0, 0, 0 ); backsql_PrintErrors( SQL_NULL_HENV, SQL_NULL_HDBC, SQL_NULL_HENV, rc ); + ret = SQL_ERROR; } - Debug( LDAP_DEBUG_TRACE, "<==backsql_init_db_env()\n", 0, 0, 0 ); - return SQL_SUCCESS; + + Debug( LDAP_DEBUG_TRACE, "<==backsql_init_db_env()=%d\n", ret, 0, 0 ); + + return ret; } int -backsql_free_db_env( backsql_info *si ) +backsql_free_db_env( backsql_info *bi ) { Debug( LDAP_DEBUG_TRACE, "==>backsql_free_db_env()\n", 0, 0, 0 ); -#ifdef BACKSQL_TRACE - Debug( LDAP_DEBUG_TRACE, "free_db_env(): delete AVL tree here!!!\n", - 0, 0, 0 ); -#endif /* BACKSQL_TRACE */ + (void)SQLFreeEnv( bi->sql_db_env ); + bi->sql_db_env = SQL_NULL_HENV; /* * stop, if frontend waits for all threads to shutdown @@ -330,42 +373,49 @@ backsql_free_db_env( backsql_info *si ) * everything is already deleted... */ Debug( LDAP_DEBUG_TRACE, "<==backsql_free_db_env()\n", 0, 0, 0 ); + return SQL_SUCCESS; } static int -backsql_open_db_conn( backsql_info *si, int ldap_cid, backsql_db_conn **pdbc ) +backsql_open_db_handle( + backsql_info *bi, + SQLHDBC *dbhp ) { /* TimesTen */ char DBMSName[ 32 ]; - backsql_db_conn *dbc; int rc; - assert( pdbc ); - *pdbc = NULL; + assert( dbhp != NULL ); + *dbhp = SQL_NULL_HDBC; - Debug( LDAP_DEBUG_TRACE, "==>backsql_open_db_conn()\n", 0, 0, 0 ); - dbc = (backsql_db_conn *)ch_calloc( 1, sizeof( backsql_db_conn ) ); - dbc->ldap_cid = ldap_cid; - rc = SQLAllocConnect( si->db_env, &dbc->dbh ); + Debug( LDAP_DEBUG_TRACE, "==>backsql_open_db_handle()\n", + 0, 0, 0 ); + + rc = SQLAllocConnect( bi->sql_db_env, dbhp ); if ( !BACKSQL_SUCCESS( rc ) ) { - Debug( LDAP_DEBUG_TRACE, "backsql_open_db_conn: " - "SQLAllocConnect() failed:\n", 0, 0, 0 ); - backsql_PrintErrors( si->db_env, SQL_NULL_HDBC, - SQL_NULL_HENV, rc ); + Debug( LDAP_DEBUG_TRACE, "backsql_open_db_handle(): " + "SQLAllocConnect() failed:\n", + 0, 0, 0 ); + backsql_PrintErrors( bi->sql_db_env, SQL_NULL_HDBC, + SQL_NULL_HENV, rc ); return LDAP_UNAVAILABLE; } - rc = SQLConnect( dbc->dbh, si->dbname, SQL_NTS, si->dbuser, - SQL_NTS, si->dbpasswd, SQL_NTS ); + rc = SQLConnect( *dbhp, + (SQLCHAR*)bi->sql_dbname, SQL_NTS, + (SQLCHAR*)bi->sql_dbuser, SQL_NTS, + (SQLCHAR*)bi->sql_dbpasswd, SQL_NTS ); if ( rc != SQL_SUCCESS ) { - Debug( LDAP_DEBUG_TRACE, "backsql_open_db_conn: " - "SQLConnect() to database '%s' as user '%s' " - "%s:\n", si->dbname, si->dbuser, + Debug( LDAP_DEBUG_TRACE, "backsql_open_db_handle(): " + "SQLConnect() to database \"%s\" %s.\n", + bi->sql_dbname, rc == SQL_SUCCESS_WITH_INFO ? - "succeeded with info" : "failed" ); - backsql_PrintErrors( si->db_env, dbc->dbh, SQL_NULL_HENV, rc ); + "succeeded with info" : "failed", + 0 ); + backsql_PrintErrors( bi->sql_db_env, *dbhp, SQL_NULL_HENV, rc ); if ( rc != SQL_SUCCESS_WITH_INFO ) { + SQLFreeConnect( *dbhp ); return LDAP_UNAVAILABLE; } } @@ -374,117 +424,115 @@ backsql_open_db_conn( backsql_info *si, int ldap_cid, backsql_db_conn **pdbc ) * TimesTen : Turn off autocommit. We must explicitly * commit any transactions. */ - SQLSetConnectOption( dbc->dbh, SQL_AUTOCOMMIT, SQL_AUTOCOMMIT_OFF ); + SQLSetConnectOption( *dbhp, SQL_AUTOCOMMIT, + BACKSQL_AUTOCOMMIT_ON( bi ) ? SQL_AUTOCOMMIT_ON : SQL_AUTOCOMMIT_OFF ); /* * See if this connection is to TimesTen. If it is, * remember that fact for later use. */ /* Assume until proven otherwise */ - si->bsql_flags &= ~BSQLF_USE_REVERSE_DN; + bi->sql_flags &= ~BSQLF_USE_REVERSE_DN; DBMSName[ 0 ] = '\0'; - rc = SQLGetInfo( dbc->dbh, SQL_DBMS_NAME, (PTR)&DBMSName, + rc = SQLGetInfo( *dbhp, SQL_DBMS_NAME, (PTR)&DBMSName, sizeof( DBMSName ), NULL ); if ( rc == SQL_SUCCESS ) { if ( strcmp( DBMSName, "TimesTen" ) == 0 || - strcmp( DBMSName, "Front-Tier" ) == 0 ) { - Debug( LDAP_DEBUG_TRACE, "backsql_open_db_conn: " - "TimesTen database!\n", 0, 0, 0 ); - si->bsql_flags |= BSQLF_USE_REVERSE_DN; + strcmp( DBMSName, "Front-Tier" ) == 0 ) + { + Debug( LDAP_DEBUG_TRACE, "backsql_open_db_handle(): " + "TimesTen database!\n", + 0, 0, 0 ); + bi->sql_flags |= BSQLF_USE_REVERSE_DN; } + } else { - Debug( LDAP_DEBUG_TRACE, "backsql_open_db_conn: " - "SQLGetInfo() failed:\n", 0, 0, 0 ); - backsql_PrintErrors( si->db_env, dbc->dbh, SQL_NULL_HENV, rc ); + Debug( LDAP_DEBUG_TRACE, "backsql_open_db_handle(): " + "SQLGetInfo() failed.\n", + 0, 0, 0 ); + backsql_PrintErrors( bi->sql_db_env, *dbhp, SQL_NULL_HENV, rc ); + SQLDisconnect( *dbhp ); + SQLFreeConnect( *dbhp ); + return LDAP_UNAVAILABLE; } /* end TimesTen */ - Debug( LDAP_DEBUG_TRACE, "backsql_open_db_conn(): " - "connected, adding to tree\n", 0, 0, 0 ); - ldap_pvt_thread_mutex_lock( &si->dbconn_mutex ); - avl_insert( &si->db_conns, dbc, backsql_cmp_connid, NULL ); - ldap_pvt_thread_mutex_unlock( &si->dbconn_mutex ); - Debug( LDAP_DEBUG_TRACE, "<==backsql_open_db_conn()\n", 0, 0, 0 ); - - *pdbc = dbc; + Debug( LDAP_DEBUG_TRACE, "<==backsql_open_db_handle()\n", + 0, 0, 0 ); return LDAP_SUCCESS; } -int -backsql_free_db_conn( Operation *op ) +static void *backsql_db_conn_dummy; + +static void +backsql_db_conn_keyfree( + void *key, + void *data ) { - backsql_info *si = (backsql_info *)op->o_bd->be_private; - backsql_db_conn tmp, *conn; + (void)backsql_close_db_handle( (SQLHDBC)data ); +} +int +backsql_free_db_conn( Operation *op, SQLHDBC dbh ) +{ Debug( LDAP_DEBUG_TRACE, "==>backsql_free_db_conn()\n", 0, 0, 0 ); - tmp.ldap_cid = op->o_connid; - ldap_pvt_thread_mutex_lock( &si->dbconn_mutex ); - conn = avl_delete( &si->db_conns, &tmp, backsql_cmp_connid ); - ldap_pvt_thread_mutex_unlock( &si->dbconn_mutex ); - /* - * we have one thread per connection, as I understand -- so we can - * get this out of critical section - */ - if ( conn != NULL ) { - Debug( LDAP_DEBUG_TRACE, "backsql_free_db_conn(): " - "closing db connection\n", 0, 0, 0 ); - backsql_close_db_conn( conn ); - } + (void)backsql_close_db_handle( dbh ); + ldap_pvt_thread_pool_setkey( op->o_threadctx, + &backsql_db_conn_dummy, (void *)SQL_NULL_HDBC, + backsql_db_conn_keyfree, NULL, NULL ); + Debug( LDAP_DEBUG_TRACE, "<==backsql_free_db_conn()\n", 0, 0, 0 ); - return SQL_SUCCESS; + + return LDAP_SUCCESS; } int -backsql_get_db_conn( Operation *op, SQLHDBC *dbh ) +backsql_get_db_conn( Operation *op, SQLHDBC *dbhp ) { - backsql_info *si = (backsql_info *)op->o_bd->be_private; - backsql_db_conn *dbc; - backsql_db_conn tmp; - int rc = LDAP_SUCCESS; + backsql_info *bi = (backsql_info *)op->o_bd->be_private; + int rc = LDAP_SUCCESS; + SQLHDBC dbh = SQL_NULL_HDBC; Debug( LDAP_DEBUG_TRACE, "==>backsql_get_db_conn()\n", 0, 0, 0 ); - assert( dbh ); - *dbh = SQL_NULL_HDBC; + assert( dbhp != NULL ); + *dbhp = SQL_NULL_HDBC; - tmp.ldap_cid = op->o_connid; + if ( op->o_threadctx ) { + void *data = NULL; - /* - * we have one thread per connection, as I understand -- - * so we do not need locking here - */ - dbc = avl_find( si->db_conns, &tmp, backsql_cmp_connid ); - if ( !dbc ) { - rc = backsql_open_db_conn( si, op->o_connid, &dbc ); - if ( rc != LDAP_SUCCESS) { - Debug( LDAP_DEBUG_TRACE, "backsql_get_db_conn(): " - "could not get connection handle " - "-- returning NULL\n", 0, 0, 0 ); - return rc; - } + ldap_pvt_thread_pool_getkey( op->o_threadctx, + &backsql_db_conn_dummy, &data, NULL ); + dbh = (SQLHDBC)data; + + } else { + dbh = bi->sql_dbh; } - ldap_pvt_thread_mutex_lock( &si->schema_mutex ); - if ( !BACKSQL_SCHEMA_LOADED( si ) ) { - Debug( LDAP_DEBUG_TRACE, "backsql_get_db_conn(): " - "first call -- reading schema map\n", 0, 0, 0 ); - rc = backsql_load_schema_map( si, dbc->dbh ); + if ( dbh == SQL_NULL_HDBC ) { + rc = backsql_open_db_handle( bi, &dbh ); if ( rc != LDAP_SUCCESS ) { - ldap_pvt_thread_mutex_unlock( &si->schema_mutex ); - backsql_free_db_conn( op ); return rc; } + + if ( op->o_threadctx ) { + void *data = (void *)dbh; + + ldap_pvt_thread_pool_setkey( op->o_threadctx, + &backsql_db_conn_dummy, data, + backsql_db_conn_keyfree, NULL, NULL ); + + } else { + bi->sql_dbh = dbh; + } } - ldap_pvt_thread_mutex_unlock( &si->schema_mutex ); - *dbh = dbc->dbh; + *dbhp = dbh; Debug( LDAP_DEBUG_TRACE, "<==backsql_get_db_conn()\n", 0, 0, 0 ); return LDAP_SUCCESS; } -#endif /* SLAPD_SQL */ -