--- sql/oracle/parser3oracle.C 2001/10/28 14:51:30 1.2 +++ sql/oracle/parser3oracle.C 2004/10/07 09:27:02 1.65 @@ -1,13 +1,14 @@ /** @file Parser Oracle driver. - Copyright(c) 2001 ArtLebedev Group(http://www.artlebedev.com) + Copyright(c) 2001, 2003 ArtLebedev Group (http://www.artlebedev.com) - Author: Alexander Petrosyan (http://design.ru/paf) + Author: Alexandr Petrosian (http://paf.design.ru) 2001.07.30 using Oracle 8.1.6 [@test tested with Oracle 7.x.x] */ -static const char *RCSId="$Id: parser3oracle.C,v 1.2 2001/10/28 14:51:30 paf Exp $"; + +static const char *RCSId="$Id: parser3oracle.C,v 1.65 2004/10/07 09:27:02 paf Exp $"; #include "config_includes.h" @@ -19,6 +20,7 @@ static const char *RCSId="$Id: parser3or #define MAX_IN_LOBS 5 #define MAX_LOB_NAME_LENGTH 100 #define MAX_OUT_STRING_LENGTH 4000 +#define MAX_BINDS 100 #define EMPTY_CLOB_FUNC_CALL "empty_clob()" @@ -38,6 +40,61 @@ inline int max(int a, int b) { return a> inline int min(int a, int b){ return a(aconnection); + // free fetch buffers. leave that to GC [no such services func. yet?] + /* + for(int i=0; ifree(fetch_buffer); + else + break; + } + */ - OracleSQL_connection_struct &cs=*(OracleSQL_connection_struct *)connection; // Terminate a user session OCISessionEnd( - cs.svchp, cs.errhp, cs.usrhp, (ub4)OCI_DEFAULT); + connection.svchp, connection.errhp, connection.usrhp, (ub4)OCI_DEFAULT); // Detach from a server; uninitialize server context handle OCIServerDetach( - cs.srvhp, cs.errhp, (ub4)OCI_DEFAULT); + connection.srvhp, connection.errhp, (ub4)OCI_DEFAULT); // Free a previously allocated handles /* oci will free them up as belonging to env OCIHandleFree( - (dvoid *)cs.srvhp, (ub4)OCI_HTYPE_SERVER); + (dvoid *)connection.srvhp, (ub4)OCI_HTYPE_SERVER); OCIHandleFree( - (dvoid *)cs.svchp, (ub4)OCI_HTYPE_SVCCTX); + (dvoid *)connection.svchp, (ub4)OCI_HTYPE_SVCCTX); OCIHandleFree( - (dvoid *)cs.errhp, (ub4)OCI_HTYPE_ERROR); + (dvoid *)connection.errhp, (ub4)OCI_HTYPE_ERROR); */ OCIHandleFree( - (dvoid *)cs.envhp, (ub4)OCI_HTYPE_ENV); + (dvoid *)connection.envhp, (ub4)OCI_HTYPE_ENV); - // connections are cross-request, do not use services._alloc [linked with request] - ::free(&cs); + // free connection. leave that to GC [no such services func. yet?] + // connection.services->free(&connection); } - void commit(SQL_Driver_services& services, void *connection) { - OracleSQL_connection_struct &cs=*(OracleSQL_connection_struct *)connection; - if(setjmp(cs.mark)) - services._throw(cs.error); + void commit(void *aconnection) { + Connection& connection=*static_cast(aconnection); + if(setjmp(connection.mark)) + connection.services->_throw(connection.error); - check(services, cs, "commit", OCITransCommit(cs.svchp, cs.errhp, 0)); + check(connection, "commit", OCITransCommit(connection.svchp, connection.errhp, 0)); } - void rollback(SQL_Driver_services& services, void *connection) { - OracleSQL_connection_struct &cs=*(OracleSQL_connection_struct *)connection; - if(setjmp(cs.mark)) - services._throw(cs.error); - - check(services, cs, "rollback", OCITransRollback(cs.svchp, cs.errhp, 0)); + void rollback(void *aconnection) { + Connection& connection=*static_cast(aconnection); + if(setjmp(connection.mark)) + connection.services->_throw(connection.error); + + // sometimes rollback is done in context when this yields error which masks previous error + // consider consequent errors not very important to report, reporting first one + /*check(connection, "rollback", */OCITransRollback(connection.svchp, connection.errhp, 0)/*)*/; } - bool ping(SQL_Driver_services&, void *connection) { + bool ping(void* /*connection*/) { // maybe OCIServerVersion? + // select 0 from dual return true; } - unsigned int quote( - SQL_Driver_services&, void *, - char *to, const char *from, unsigned int length) { - /* - it's already UNTAINT_TIMES_BIGGER - */ - unsigned int result=length; + const char* quote(void *aconnection, + const char *from, unsigned int length) + { + Connection& connection=*static_cast(aconnection); + char *result=(char*)connection.services->malloc_atomic(length*2+1); + char *to=result; while(length--) { switch(*from) { case '\'': // "'" -> "''" *to++='\''; break; - case '\\': // "\" -> "\\" - *to++='\''; - break; } *to++=*from++; } + *to=0; return result; } - void query( - SQL_Driver_services& services, void *connection, - const char *astatement, unsigned long offset, unsigned long limit, - SQL_Driver_query_event_handlers& handlers) { - - OracleSQL_connection_struct &cs=*(OracleSQL_connection_struct *)connection; - OracleSQL_query_lobs lobs={{0}, 0}; + void query(void* aconnection, + const char* astatement, + size_t placeholders_count, Placeholder* placeholders, + unsigned long offset, unsigned long limit, + SQL_Driver_query_event_handlers& handlers) + { + + Connection& connection=*static_cast(aconnection); + const char* cstrClientCharset=connection.options.cstrClientCharset; + Query_lobs lobs={{0}, 0}; OCIStmt *stmthp=0; + SQL_Driver_services& services=*connection.services; + + // transcode from $request:charset to connect-string?client_charset + if(cstrClientCharset) { + size_t transcoded_xxx_size; + services.transcode(astatement, strlen(astatement), + astatement, transcoded_xxx_size, + services.request_charset(), + cstrClientCharset); + } + bool failed=false; - if(setjmp(cs.mark)) { + if(setjmp(connection.mark)) { failed=true; goto cleanup; } else { - const char *statement=preprocess_statement(services, cs, - astatement, lobs); + if(placeholders_count>MAX_BINDS) + fail(connection, "too many bind variables"); - check(services, cs, "HandleAlloc STMT", OCIHandleAlloc( - (dvoid *)cs.envhp, (dvoid **) &stmthp, (ub4)OCI_HTYPE_STMT, 0, 0)); - check(services, cs, "syntax", - OCIStmtPrepare(stmthp, cs.errhp, (unsigned char *)statement, + const char *statement=preprocess_statement(connection, astatement, lobs); + + check(connection, "HandleAlloc STMT", OCIHandleAlloc( + (dvoid *)connection.envhp, (dvoid **) &stmthp, (ub4)OCI_HTYPE_STMT, 0, 0)); + check(connection, "syntax", + OCIStmtPrepare(stmthp, connection.errhp, (unsigned char *)statement, (ub4)strlen((char *)statement), (ub4)OCI_NTV_SYNTAX, (ub4)OCI_DEFAULT)); + + struct Bind_info { + OCIBind *bind; + sb2 indicator; + }; + + int binds_size=sizeof(Bind_info) * placeholders_count; + // we DO store OCIBind* into ATOMIC gc memory, + // but we do not allocate/free it, that's done automatically from oracle [using environment handles] + // so we don't have to bother with that + Bind_info* binds=static_cast(services.malloc_atomic(binds_size)); { - for(int i=0; icount; r++) { OCILobLocator *locator=rows->row[r].locator; - check(services, cs, "lobwrite", OCILobWrite ( - cs.svchp, cs.errhp, + check(connection, "lobwrite", OCILobWrite ( + connection.svchp, connection.errhp, locator, &bytes_to_write, 1, (dvoid *)lobs.items[i].data_ptr, (ub4)bytes_to_write, OCI_ONE_PIECE, (dvoid *)0, 0, (ub2)0, @@ -495,7 +744,7 @@ private: // private funcs switch(stmt_type) { case OCI_STMT_SELECT: - fetch_table(services, cs, + fetch_table(connection, stmthp, offset, limit, handlers); break; @@ -508,15 +757,30 @@ private: // private funcs } } - void fetch_table(SQL_Driver_services& services, OracleSQL_connection_struct &cs, + void fetch_table(Connection& connection, OCIStmt *stmthp, unsigned long offset, unsigned long limit, - SQL_Driver_query_event_handlers& handlers) { + SQL_Driver_query_event_handlers& handlers) + { + const char* cstrClientCharset=connection.options.cstrClientCharset; + SQL_Driver_services& services=*connection.services; + + ub4 prefetch_rows=100; + check(connection, "AttrSet prefetch-rows", OCIAttrSet( + (dvoid *)stmthp, (ub4)OCI_HTYPE_STMT, + (dvoid *)&prefetch_rows, (ub4)0, + (ub4)OCI_ATTR_PREFETCH_ROWS, (OCIError *)connection.errhp)); + + ub4 prefetch_mem_size=100*0x400; + check(connection, "AttrSet prefetch-memory", OCIAttrSet( + (dvoid *)stmthp, (ub4)OCI_HTYPE_STMT, + (dvoid *)&prefetch_mem_size, (ub4)0, + (ub4)OCI_ATTR_PREFETCH_MEMORY, (OCIError *)connection.errhp)); OCIParam *mypard; ub2 dtype; - text *col_name; + const char* col_name; - struct { + struct Col { ub2 type; char *str; OCILobLocator *var; @@ -526,36 +790,50 @@ private: // private funcs int column_count=0; bool failed=false; - jmp_buf saved_mark; memcpy(saved_mark, cs.mark, sizeof(jmp_buf)); - if(setjmp(cs.mark)) { + jmp_buf saved_mark; memcpy(saved_mark, connection.mark, sizeof(jmp_buf)); + if(setjmp(connection.mark)) { failed=true; goto cleanup; } else { // idea of preincrementing is that at error time all handles would free up while(++column_count<=MAX_COLS) { /* get next descriptor, if there is one */ - if(OCIParamGet(stmthp, OCI_HTYPE_STMT, cs.errhp, (void **)&mypard, - (ub4) column_count)!=OCI_SUCCESS) + if(OCIParamGet(stmthp, OCI_HTYPE_STMT, connection.errhp, (void **)&mypard, + (ub4) column_count)!=OCI_SUCCESS) { + --column_count; break; + } /* Retrieve the data type attribute */ - check(services, cs, "get type", OCIAttrGet( + check(connection, "get type", OCIAttrGet( (dvoid*) mypard, (ub4)OCI_DTYPE_PARAM, (dvoid*) &dtype, (ub4 *)0, (ub4)OCI_ATTR_DATA_TYPE, - (OCIError *)cs.errhp)); + (OCIError *)connection.errhp)); /* Retrieve the column name attribute */ ub4 col_name_len; - check(services, cs, "get name", OCIAttrGet( + check(connection, "get name", OCIAttrGet( (dvoid*) mypard, (ub4)OCI_DTYPE_PARAM, (dvoid**) &col_name, (ub4 *) &col_name_len, (ub4)OCI_ATTR_NAME, - (OCIError *)cs.errhp)); - + (OCIError *)connection.errhp)); + // transcode to $request:charset from connect-string?client_charset + if(cstrClientCharset) { + services.transcode(col_name, col_name_len, + col_name, col_name_len, + cstrClientCharset, + services.request_charset()); + } + + Col& col=cols[column_count-1]; { - size_t size=(size_t)col_name_len; - char *ptr=(char *)services.malloc(size); - tolower(ptr, (char *)col_name, size); - handlers.add_column(ptr, size); + size_t length=(size_t)col_name_len; + char *ptr=(char *)services.malloc_atomic(length+1); + if( connection.options.bLowerCaseColumnNames ) + tolower_str(ptr, col_name, length); + else + memcpy(ptr, col_name, length); + ptr[length]=0; + check(connection, handlers.add_column(connection.sql_error, ptr, length)); } ub2 coerce_type=dtype; @@ -565,8 +843,8 @@ private: // private funcs switch(dtype) { case SQLT_CLOB: { - check(services, cs, "alloc output var desc", OCIDescriptorAlloc( - (dvoid *)cs.envhp, (dvoid **)(ptr=&cols[column_count-1].var), + check(connection, "alloc output var desc", OCIDescriptorAlloc( + (dvoid *)connection.envhp, (dvoid **)(ptr=&col.var), (ub4)OCI_DTYPE_LOB, 0, (dvoid **)0)); @@ -575,70 +853,105 @@ private: // private funcs } default: coerce_type=SQLT_STR; - ptr=cols[column_count-1].str=(char *)services.malloc(MAX_OUT_STRING_LENGTH+1); + char*& buf=connection.fetch_buffers[column_count-1]; + ptr=buf; // get cached buffer + if(!ptr) // allocate if needed, caching it + ptr=buf=(char *)services.malloc_atomic(MAX_OUT_STRING_LENGTH+1/*terminator*/); + col.str=(char*)ptr; size=MAX_OUT_STRING_LENGTH; break; } - cols[column_count-1].type=coerce_type; + col.type=coerce_type; - check(services, cs, "DefineByPos", OCIDefineByPos( - stmthp, &cols[column_count-1].def, cs.errhp, + // http://i/docs/oracle/server.804/a58234/oci_func.htm#449680 + // this call implicitly allocates the define handle + // http://sunsite.eunnet.net/documentation/oracle.8.0.4/server.804/a58234/basics.htm + // when a statement handle is freed, any bind and define handles associated with it + // are also freed + col.def=0; check(connection, "DefineByPos", OCIDefineByPos( + stmthp, &col.def, connection.errhp, column_count, (ub1 *) ptr, size, - coerce_type, (dvoid *) &cols[column_count-1].indicator, + coerce_type, (dvoid *) &col.indicator, (ub2 *)0, (ub2 *)0, OCI_DEFAULT)); } - handlers.before_rows(); + check(connection, handlers.before_rows(connection.sql_error)); - for(unsigned long row=0; !limit||row=offset) { - handlers.add_row(); + check(connection, handlers.add_row(connection.sql_error)); for(int i=0; iOCIErrorGet((dvoid *)cs.errhp, (ub4)1, (text *)NULL, &errcode, - (text *)reason, (ub4)sizeof(reason), OCI_HTYPE_ERROR)==OCI_SUCCESS) - msg=reason; - else - msg="[can not get error description]"; - break; + sb4 errcode; + if(OracleSQL_driver->OCIErrorGet((dvoid *)connection.errhp, (ub4)1, (text *)NULL, &errcode, + (text *)reason, (ub4)sizeof(reason), OCI_HTYPE_ERROR)==OCI_SUCCESS) { + msg=reason; + + // transcode to $request:charset from connect-string?client_charset + if(const char* cstrClientCharset=connection.options.cstrClientCharset) { + if(msg) { + if(size_t msg_length=strlen(msg)) { + connection.services->transcode(msg, msg_length, + msg, msg_length, + cstrClientCharset, + connection.services->request_charset()); + } + } + } + } else + msg="[can not get error description]"; + break; } - case OCI_SUCCESS_WITH_INFO: - msg="SUCCESS_WITH_INFO"; break; case OCI_NEED_DATA: msg="NEED_DATA"; break; case OCI_NO_DATA: @@ -850,19 +1170,28 @@ void check( msg="unknown"; break; } - snprintf(cs.error, sizeof(cs.error), "%s (%s, %d)", + snprintf(connection.error, sizeof(connection.error), "%s (%s, %d)", msg, step, (int)status); - longjmp(cs.mark, 1); + longjmp(connection.mark, 1); } +void fail(Connection& connection, const char* msg) { + snprintf(connection.error, sizeof(connection.error), "%s", msg); + longjmp(connection.mark, 1); +} + +void check(Connection& connection, bool error) { + if(error) + longjmp(connection.mark, 1); +} /* ----------------------------------------------------------------- */ /* Intbind callback that does not do any data input. */ /* ----------------------------------------------------------------- */ -static sb4 cbf_no_data( - dvoid *ctxp, - OCIBind *bindp, - ub4 iter, ub4 index, +sb4 cbf_no_data( + dvoid* /*ctxp*/, + OCIBind* /*bindp*/, + ub4 /*iter*/, ub4 /*index*/, dvoid **bufpp, ub4 *alenpp, ub1 *piecep, @@ -881,30 +1210,29 @@ static sb4 cbf_no_data( /* ----------------------------------------------------------------- */ static sb4 cbf_get_data(dvoid *ctxp, OCIBind *bindp, - ub4 iter, ub4 index, + ub4 /*iter*/, ub4 index, dvoid **bufpp, ub4 **alenp, ub1 *piecep, dvoid **indpp, ub2 **rcodepp) { - OracleSQL_query_lobs::cbf_context_struct &context= - *(OracleSQL_query_lobs::cbf_context_struct *)ctxp; + Query_lobs::Item& context=*static_cast(ctxp); if(index==0) { static ub4 rows; - check(*context.services, *context.cs, "AttrGet cbf_get_data ROWS_RETURNED", + check(*context.connection, "AttrGet cbf_get_data ROWS_RETURNED", OracleSQL_driver->OCIAttrGet( (CONST dvoid *) bindp, OCI_HTYPE_BIND, (dvoid *)&rows, - (ub4 *)sizeof(ub2), OCI_ATTR_ROWS_RETURNED, context.cs->errhp)) ; - context.rows->count=(ub2)rows; - context.rows->row=(OracleSQL_query_lobs::return_rows::return_row *) - context.services->malloc(sizeof(OracleSQL_query_lobs::return_rows::return_row)*rows); + (ub4 *)sizeof(ub2), OCI_ATTR_ROWS_RETURNED, context.connection->errhp)) ; + context.rows.count=(ub2)rows; + context.rows.row=(Query_lobs::return_rows::return_row *) + context.connection->services->malloc_atomic(sizeof(Query_lobs::return_rows::return_row)*rows); } - OracleSQL_query_lobs::return_rows::return_row &var=context.rows->row[index]; + Query_lobs::return_rows::return_row &var=context.rows.row[index]; - check(*context.services, *context.cs, "alloc output var desc dynamic", OracleSQL_driver->OCIDescriptorAlloc( - (dvoid *) context.cs->envhp, (dvoid **)&var.locator, + check(*context.connection, "alloc output var desc dynamic", OracleSQL_driver->OCIDescriptorAlloc( + (dvoid *) context.connection->envhp, (dvoid **)&var.locator, (ub4)OCI_DTYPE_LOB, 0, (dvoid **)0)); @@ -917,12 +1245,16 @@ static sb4 cbf_get_data(dvoid *ctxp, return OCI_CONTINUE; } -void tolower(char *out, const char *in, size_t size) { +static void tolower_str(char *out, const char *in, size_t size) { while(size--) - *out++=tolower(*in++); + *out++=(char)tolower(*in++); } +static void toupper_str(char *out, const char *in, size_t size) { + while(size--) + *out++=(char)toupper(*in++); +} + extern "C" SQL_Driver *SQL_DRIVER_CREATE() { - //_asm int 3; return OracleSQL_driver=new OracleSQL_Driver(); }