git.sur5r.net Git - openldap/blob - servers/slapd/back-sql/sql-wrap.c

   1 /*
   2  *       Copyright 1999, Dmitry Kovalev <mit@openldap.org>, All rights reserved.
   3  *
   4  *       Redistribution and use in source and binary forms are permitted only
   5  *       as authorized by the OpenLDAP Public License.  A copy of this
   6  *       license is available at http://www.OpenLDAP.org/license.html or
   7  *       in file LICENSE in the top-level directory of the distribution.
   8  */
   9
  10 #include "portable.h"
  11
  12 #ifdef SLAPD_SQL
  13
  14 #include <stdio.h>
  15 #include <string.h>
  16 #include <sys/types.h>
  17 #include "slap.h"
  18 #include "back-sql.h"
  19 #include "sql-types.h"
  20 #include "sql-wrap.h"
  21 #include "schema-map.h"
  22
  23 #define MAX_ATTR_LEN 16384
  24
  25 typedef struct backsql_conn
  26 {
  27  int ldap_cid;
  28  SQLHDBC dbh;
  29 }backsql_db_conn;
  30
  31 int backsql_dummy(void *,void *);
  32
  33 void backsql_PrintErrors(SQLHENV henv, SQLHDBC hdbc, SQLHSTMT sth,int rc)
  34 {
  35  SQLCHAR           msg[SQL_MAX_MESSAGE_LENGTH];    /* msg. buffer    */
  36  SQLCHAR           state[SQL_SQLSTATE_SIZE];       /* statement buf. */
  37  SDWORD        iSqlCode;                              /* return code    */
  38  SWORD       len=SQL_MAX_MESSAGE_LENGTH-1;                               /* return length  */
  39
  40
  41  Debug(LDAP_DEBUG_TRACE,"Return code: %d\n", rc,0,0);
  42
  43  while((rc=SQLError(henv,hdbc,sth,state,&iSqlCode,msg,
  44              SQL_MAX_MESSAGE_LENGTH - 1, &len)) == SQL_SUCCESS
  45         || rc == SQL_SUCCESS_WITH_INFO
  46        )
  47  {
  48   Debug(LDAP_DEBUG_TRACE,"Native error code: %d\n",(int) iSqlCode,0,0);
  49   Debug(LDAP_DEBUG_TRACE,"SQL engine state: %s\n", state,0,0);
  50   Debug(LDAP_DEBUG_TRACE,"Message: %s\n",msg,0,0);
  51  }
  52 }
  53
  54 RETCODE backsql_Prepare(SQLHDBC dbh,SQLHSTMT *sth,char* query,int timeout)
  55 {
  56  RETCODE rc;
  57  char drv_name[30];
  58  SWORD len;
  59  int i;
  60
  61  rc=SQLAllocStmt(dbh,sth);
  62  if (rc != SQL_SUCCESS)
  63   return rc;
  64
  65  //Debug(LDAP_DEBUG_TRACE,"==>_SQLPrepare()\n", 0,0,0);
  66  SQLGetInfo(dbh,SQL_DRIVER_NAME,drv_name,30,&len);
  67  //Debug(LDAP_DEBUG_TRACE,"_SQLPrepare(): driver name='%s'\n", drv_name,0,0);
  68  if (!strncmp(ldap_pvt_str2upper(drv_name),"SQLSRV32.DLL",30))
  69   {
  70    //stupid default result set in MS SQL Server does not support multiple active statements
  71    //on the same connection -- so we are trying to make it not to use default result set...
  72    Debug(LDAP_DEBUG_TRACE,"_SQLprepare(): enabling MS SQL Server default result set workaround\n", 0,0,0);
  73    rc=SQLSetStmtOption(*sth,SQL_CONCURRENCY,SQL_CONCUR_ROWVER);
  74    if (rc != SQL_SUCCESS && rc != SQL_SUCCESS_WITH_INFO)
  75    {
  76     Debug(LDAP_DEBUG_TRACE,"_SQLPrepare(): SQLSetStmtOption(SQL_CONCURRENCY,SQL_CONCUR_ROWVER) failed:\n", 0,0,0);
  77     backsql_PrintErrors(SQL_NULL_HENV,dbh,sth,rc);
  78    }
  79   }
  80  if (timeout>0)
  81   {
  82    Debug(LDAP_DEBUG_TRACE,"_SQLprepare(): setting query timeout to %d sec.\n", timeout,0,0);
  83    if ((rc=SQLSetStmtOption(*sth,SQL_QUERY_TIMEOUT,timeout)) != SQL_SUCCESS)
  84     {
  85      backsql_PrintErrors(SQL_NULL_HENV,dbh,sth,rc);
  86     }
  87   }
  88
  89  //Debug(LDAP_DEBUG_TRACE,"<==_SQLPrepare() calling SQLPrepare()\n", 0,0,0);
  90  return SQLPrepare(*sth,query,SQL_NTS);
  91 }
  92
  93 RETCODE backsql_BindParamStr(SQLHSTMT sth,int par_ind,char *str,int maxlen)
  94 {
  95  RETCODE rc;
  96  SQLINTEGER len=SQL_NTS;
  97
  98  rc=SQLBindParameter(sth,(SQLUSMALLINT)par_ind,SQL_PARAM_INPUT,SQL_C_CHAR,SQL_VARCHAR,
  99          (SQLUINTEGER)maxlen,0,(SQLPOINTER)str,(SQLUINTEGER)maxlen,NULL);
 100  return rc;
 101 }
 102
 103 RETCODE backsql_BindParamID(SQLHSTMT sth,int par_ind,unsigned long *id)
 104 {
 105
 106  return SQLBindParameter(sth,(SQLUSMALLINT)par_ind,SQL_PARAM_INPUT,SQL_C_ULONG,SQL_INTEGER,
 107          0,0,(SQLPOINTER)id,0,(SQLINTEGER*)NULL);
 108 }
 109
 110 RETCODE backsql_BindRowAsStrings(SQLHSTMT sth,BACKSQL_ROW_NTS *row)
 111 {
 112  RETCODE rc;
 113  SQLCHAR colname[64];
 114  SQLSMALLINT name_len,col_type,col_scale,col_null;
 115  UDWORD col_prec;
 116  int i;
 117
 118  if (row == NULL)
 119   return SQL_ERROR;
 120
 121  //Debug(LDAP_DEBUG_TRACE,"==> backsql_BindRowAsStrings()\n",0,0,0);
 122  rc=SQLNumResultCols(sth,&row->ncols);
 123  if (rc != SQL_SUCCESS)
 124  {
 125   //Debug(LDAP_DEBUG_TRACE,"_SQLBindRowAsStrings(): SQLNumResultCols() failed:\n",0,0,0);
 126   backsql_PrintErrors(SQL_NULL_HENV,SQL_NULL_HDBC,sth,rc);
 127  }
 128  else
 129  {
 130   //Debug(LDAP_DEBUG_TRACE,"backsql_BindRowAsStrings: ncols=%d\n",(int)row->ncols,0,0);
 131   row->col_names=(char**)ch_calloc(row->ncols,sizeof(char*));
 132   row->cols=(char**)ch_calloc(row->ncols,sizeof(char*));
 133   row->col_prec=(UDWORD*)ch_calloc(row->ncols,sizeof(UDWORD));
 134   row->is_null=(SQLINTEGER*)ch_calloc(row->ncols,sizeof(SQLINTEGER));
 135   for (i=1;i<=row->ncols;i++)
 136   {
 137    rc=SQLDescribeCol(sth,(SQLSMALLINT)i,&colname[0],(SQLUINTEGER)sizeof(colname)-1,&name_len,&col_type,
 138           (UDWORD*) &col_prec,&col_scale,&col_null);
 139    row->col_names[i-1]=ch_strdup(colname);
 140    //Debug(LDAP_DEBUG_TRACE,"backsql_BindRowAsStrings: col_name=%s, col_prec[%d]=%d\n",colname,(int)i,(int)col_prec);
 141    if (col_type == SQL_LONGVARCHAR || col_type== SQL_LONGVARBINARY)
 142    {
 143         //row->cols[i-1]=NULL;
 144     //row->col_prec[i-1]=-1;
 145         //such fields must be handled in some other way since they return 2G
 146         //as their precision (at least it does so with MS SQL Server w/native driver)
 147         //for now, we just set fixed precision for such fields - dirty hack, but...
 148         //no time to deal with SQLGetData()
 149         col_prec=MAX_ATTR_LEN;
 150         row->cols[i-1]=(char*)ch_calloc((col_prec+1),sizeof(char));
 151     row->col_prec[i-1]=col_prec;
 152         rc=SQLBindCol(sth,(SQLUSMALLINT)i,SQL_C_CHAR,(SQLPOINTER)row->cols[i-1],col_prec+1,
 153                                                        &row->is_null[i-1]);
 154    }
 155    else
 156    {
 157     row->cols[i-1]=(char*)ch_calloc((col_prec+1),sizeof(char));
 158     row->col_prec[i-1]=col_prec;
 159         rc=SQLBindCol(sth,(SQLUSMALLINT)i,SQL_C_CHAR,(SQLPOINTER)row->cols[i-1],col_prec+1,
 160                                                        &row->is_null[i-1]);
 161    }
 162   }
 163  }
 164  //Debug(LDAP_DEBUG_TRACE,"<== backsql_BindRowAsStrings()\n",0,0,0);
 165  return rc;
 166 }
 167
 168 RETCODE backsql_FreeRow(BACKSQL_ROW_NTS *row)
 169 {
 170  int i;
 171
 172  if (row->cols == NULL)
 173   return SQL_ERROR;
 174  for(i=0;i<row->ncols;i++)
 175   {
 176    free(row->cols[i]);
 177   }
 178  free(row->col_names);
 179  free(row->col_prec);
 180  free(row->cols);
 181  free(row->is_null);
 182  return SQL_SUCCESS;
 183 }
 184
 185 int backsql_cmp_connid(backsql_db_conn *c1,backsql_db_conn *c2)
 186 {
 187  if (c1->ldap_cid > c2->ldap_cid)
 188   return 1;
 189  if (c1->ldap_cid < c2->ldap_cid)
 190   return -1;
 191  return 0;
 192 }
 193
 194
 195 int backsql_close_db_conn(backsql_db_conn *conn)
 196 {
 197  Debug(LDAP_DEBUG_TRACE,"==>backsql_close_db_conn()\n",0,0,0);
 198  SQLTransact(NULL, conn->dbh, SQL_COMMIT);  // TimesTen
 199  SQLDisconnect(conn->dbh);
 200  SQLFreeConnect(conn->dbh);
 201  Debug(LDAP_DEBUG_TRACE,"<==backsql_close_db_conn()\n",0,0,0);
 202  return 1;
 203 }
 204
 205 int backsql_init_db_env(backsql_info *si)
 206 {
 207  RETCODE rc;
 208  Debug(LDAP_DEBUG_TRACE,"==>backsql_init_db_env()\n",0,0,0);
 209  if ((rc=SQLAllocEnv(&si->db_env)) != SQL_SUCCESS)
 210   {
 211    Debug(LDAP_DEBUG_TRACE,"init_db_env: SQLAllocEnv failed:\n",0,0,0);
 212    backsql_PrintErrors(SQL_NULL_HENV,SQL_NULL_HDBC,SQL_NULL_HENV,rc);
 213   }
 214  Debug(LDAP_DEBUG_TRACE,"<==backsql_init_db_env()\n",0,0,0);
 215  return SQL_SUCCESS;
 216 }
 217
 218 int backsql_free_db_env(backsql_info *si)
 219 {
 220  Debug(LDAP_DEBUG_TRACE,"==>backsql_free_db_env()\n",0,0,0);
 221  //Debug(LDAP_DEBUG_TRACE,"free_db_env(): delete AVL tree here!!!\n",0,0,0);
 222
 223  //stop, if frontend waits for all threads to shutdown before calling this --
 224  //then what we are going to delete?? everything is deleted already...
 225  Debug(LDAP_DEBUG_TRACE,"<==backsql_free_db_env()\n",0,0,0);
 226  return SQL_SUCCESS;
 227 }
 228
 229 backsql_db_conn* backsql_open_db_conn(backsql_info *si,int ldap_cid)
 230 {
 231  char DBMSName[32]; // TimesTen
 232
 233  backsql_db_conn *dbc=(backsql_db_conn*)ch_calloc(1,sizeof(backsql_db_conn));
 234  int rc;
 235
 236  Debug(LDAP_DEBUG_TRACE,"==>backsql_open_db_conn()\n",0,0,0);
 237  dbc->ldap_cid=ldap_cid;
 238  if ((rc=SQLAllocConnect(si->db_env,&dbc->dbh)) != SQL_SUCCESS && rc != SQL_SUCCESS_WITH_INFO)
 239   {
 240    Debug(LDAP_DEBUG_TRACE,"backsql_open_db_conn: SQLAllocConnect() failed:\n",0,0,0);
 241    backsql_PrintErrors(si->db_env,SQL_NULL_HDBC,SQL_NULL_HENV,rc);
 242    return NULL;
 243   }
 244  if ((rc=SQLConnect(dbc->dbh,si->dbname,SQL_NTS,si->dbuser,SQL_NTS,
 245                       si->dbpasswd,SQL_NTS) != SQL_SUCCESS))
 246   {
 247    if (rc != SQL_SUCCESS_WITH_INFO)
 248     Debug(LDAP_DEBUG_TRACE,"backsql_open_db_conn: SQLConnect() failed:\n",0,0,0);
 249    else
 250         Debug(LDAP_DEBUG_TRACE,"backsql_open_db_conn: SQLConnect() succeeded with info:\n",0,0,0);
 251    backsql_PrintErrors(si->db_env,dbc->dbh,SQL_NULL_HENV,rc);
 252    if (rc != SQL_SUCCESS_WITH_INFO)
 253     return NULL;
 254   }
 255
 256  /* TimesTen : Turn off autocommit.  We must explicitly commit any transactions. */
 257
 258  SQLSetConnectOption(dbc->dbh, SQL_AUTOCOMMIT, SQL_AUTOCOMMIT_OFF);
 259
 260  /* See if this connection is to TimesTen.  If it is,
 261     remember that fact for later use. */
 262
 263  si->isTimesTen = 0;        /* Assume until proven otherwise */
 264
 265  DBMSName[0] = '\0';
 266  rc = SQLGetInfo(dbc->dbh, SQL_DBMS_NAME, (PTR) &DBMSName,
 267          sizeof(DBMSName), NULL);
 268  if (rc == SQL_SUCCESS) {
 269    if (strcmp(DBMSName, "TimesTen") == 0 ||
 270        strcmp(DBMSName, "Front-Tier") == 0) {
 271      Debug(LDAP_DEBUG_TRACE,"backsql_open_db_conn: TimesTen database!\n",0,0,0);
 272      si->isTimesTen = 1;
 273    }
 274  }
 275  else {
 276    Debug(LDAP_DEBUG_TRACE,"backsql_open_db_conn: SQLGetInfo() failed:\n",0,0,0);
 277    backsql_PrintErrors(si->db_env,dbc->dbh,SQL_NULL_HENV,rc);
 278  }
 279  // end TimesTen
 280
 281  Debug(LDAP_DEBUG_TRACE,"backsql_open_db_conn(): connected, adding to tree\n",0,0,0);
 282  ldap_pvt_thread_mutex_lock(&si->dbconn_mutex);
 283  avl_insert(&si->db_conns,dbc,(AVL_CMP)backsql_cmp_connid,backsql_dummy);
 284  ldap_pvt_thread_mutex_unlock(&si->dbconn_mutex);
 285  Debug(LDAP_DEBUG_TRACE,"<==backsql_open_db_conn()\n",0,0,0);
 286  return dbc;
 287 }
 288
 289 int backsql_free_db_conn(Backend *be,Connection *ldapc)
 290 {
 291  backsql_info *si=(backsql_info*)be->be_private;
 292  backsql_db_conn tmp,*conn;
 293
 294  Debug(LDAP_DEBUG_TRACE,"==>backsql_free_db_conn()\n",0,0,0);
 295  tmp.ldap_cid=ldapc->c_connid;
 296  ldap_pvt_thread_mutex_lock(&si->dbconn_mutex);
 297  conn=(backsql_db_conn*)avl_delete(&si->db_conns,&tmp,(AVL_CMP)backsql_cmp_connid);
 298  ldap_pvt_thread_mutex_unlock(&si->dbconn_mutex);
 299  //we have one thread per connection, as I understand -- so we can
 300  //get this out of critical section
 301  if (conn!=NULL)
 302   {
 303    Debug(LDAP_DEBUG_TRACE,"backsql_free_db_conn(): closing db connection\n",0,0,0);
 304    backsql_close_db_conn(conn);
 305   }
 306  Debug(LDAP_DEBUG_TRACE,"<==backsql_free_db_conn()\n",0,0,0);
 307  return SQL_SUCCESS;
 308 }
 309
 310 SQLHDBC backsql_get_db_conn(Backend *be,Connection *ldapc)
 311 {
 312  backsql_info *si=(backsql_info*)be->be_private;
 313  backsql_db_conn *dbc;
 314  backsql_db_conn tmp;
 315
 316  Debug(LDAP_DEBUG_TRACE,"==>backsql_get_db_conn()\n",0,0,0);
 317
 318  tmp.ldap_cid=ldapc->c_connid;
 319  //we have one thread per connection, as I understand -- so we do not need
 320  // locking here
 321  dbc=(backsql_db_conn*)avl_find(si->db_conns,&tmp,(AVL_CMP)backsql_cmp_connid);
 322  if (!dbc)
 323   dbc=backsql_open_db_conn(si,ldapc->c_connid);
 324
 325  if (!dbc)
 326  {
 327    Debug(LDAP_DEBUG_TRACE,"backsql_get_db_conn(): could not get connection handle -- returning NULL\n",0,0,0);
 328    return NULL;
 329  }
 330  ldap_pvt_thread_mutex_lock(&si->schema_mutex);
 331  if (!si->schema_loaded)
 332   {
 333    Debug(LDAP_DEBUG_TRACE,"backsql_get_db_conn(): first call -- reading schema map\n",0,0,0);
 334    backsql_load_schema_map(si,dbc->dbh);
 335   }
 336  ldap_pvt_thread_mutex_unlock(&si->schema_mutex);
 337
 338  Debug(LDAP_DEBUG_TRACE,"<==backsql_get_db_conn()\n",0,0,0);
 339  return dbc->dbh;
 340 }
 341
 342 #endif /* SLAPD_SQL */