--- sql/oracle/parser3oracle.C 2003/09/29 06:15:27 1.35 +++ sql/oracle/parser3oracle.C 2008/06/30 15:22:28 1.69 @@ -7,7 +7,8 @@ 2001.07.30 using Oracle 8.1.6 [@test tested with Oracle 7.x.x] */ -static const char *RCSId="$Id: parser3oracle.C,v 1.35 2003/09/29 06:15:27 paf Exp $"; + +static const char *RCSId="$Id: parser3oracle.C,v 1.69 2008/06/30 15:22:28 misha 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,14 @@ 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; + } + */ + // 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); - check(cs, "commit", OCITransCommit(cs.svchp, cs.errhp, 0)); + void commit(void *aconnection) { + Connection& connection=*static_cast(aconnection); + if(setjmp(connection.mark)) + connection.services->_throw(connection.error); + + 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(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; } - const char* quote( - SQL_Driver_services& services, void *connection, - const char *from, unsigned int length) { - char *result=(char*)services.malloc_atomic(length*2+1); + 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) { @@ -376,54 +445,170 @@ public: *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); + Query_lobs lobs={{0}, 0}; OCIStmt *stmthp=0; + SQL_Driver_services& services=*connection.services; + + bool transcode_needed=_transcode_required(connection); + + if(transcode_needed){ + // transcode query from $request:charset to ?ClientCharset + size_t transcoded_xxx_size; + services.transcode(astatement, strlen(astatement), + astatement, transcoded_xxx_size, + services.request_charset(), + connection.options.client_charset); + } + 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(cs, "HandleAlloc STMT", OCIHandleAlloc( - (dvoid *)cs.envhp, (dvoid **) &stmthp, (ub4)OCI_HTYPE_STMT, 0, 0)); - check(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(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, @@ -591,7 +772,7 @@ private: // private funcs switch(stmt_type) { case OCI_STMT_SELECT: - fetch_table(services, cs, + fetch_table(connection, stmthp, offset, limit, handlers); break; @@ -604,27 +785,31 @@ 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) + { + bool transcode_needed=_transcode_required(connection); + + SQL_Driver_services& services=*connection.services; ub4 prefetch_rows=100; - check(cs, "AttrSet prefetch-rows", OCIAttrSet( + check(connection, "AttrSet prefetch-rows", OCIAttrSet( (dvoid *)stmthp, (ub4)OCI_HTYPE_STMT, (dvoid *)&prefetch_rows, (ub4)0, - (ub4)OCI_ATTR_PREFETCH_ROWS, (OCIError *)cs.errhp)); + (ub4)OCI_ATTR_PREFETCH_ROWS, (OCIError *)connection.errhp)); ub4 prefetch_mem_size=100*0x400; - check(cs, "AttrSet prefetch-memory", OCIAttrSet( + check(connection, "AttrSet prefetch-memory", OCIAttrSet( (dvoid *)stmthp, (ub4)OCI_HTYPE_STMT, (dvoid *)&prefetch_mem_size, (ub4)0, - (ub4)OCI_ATTR_PREFETCH_MEMORY, (OCIError *)cs.errhp)); + (ub4)OCI_ATTR_PREFETCH_MEMORY, (OCIError *)connection.errhp)); - OCIParam *mypard; - ub2 dtype; - text *col_name; + OCIParam *mypard; + ub2 dtype; + const char* col_name; - struct { + struct Col { ub2 type; char *str; OCILobLocator *var; @@ -634,38 +819,51 @@ 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, + if(OCIParamGet(stmthp, OCI_HTYPE_STMT, connection.errhp, (void **)&mypard, (ub4) column_count)!=OCI_SUCCESS) { --column_count; break; } /* Retrieve the data type attribute */ - check(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(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)); + + if(transcode_needed){ + // transcode column name from ?ClientCharset to $request:charset + services.transcode(col_name, col_name_len, + col_name, col_name_len, + connection.options.client_charset, + services.request_charset()); + } + + Col& col=cols[column_count-1]; { - size_t size=(size_t)col_name_len; - char *ptr=(char *)services.malloc_atomic(size); - tolower(ptr, (char *)col_name, size); - check(cs, handlers.add_column(cs.sql_error, 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; @@ -673,79 +871,130 @@ private: // private funcs void *ptr; switch(dtype) { - case SQLT_CLOB: - { - check(cs, "alloc output var desc", OCIDescriptorAlloc( - (dvoid *)cs.envhp, (dvoid **)(ptr=&cols[column_count-1].var), - (ub4)OCI_DTYPE_LOB, - 0, (dvoid **)0)); - - size=0; + case SQLT_CLOB: + { + check(connection, "alloc output var desc", OCIDescriptorAlloc( + (dvoid *)connection.envhp, (dvoid **)(ptr=&col.var), + (ub4)OCI_DTYPE_LOB, + 0, (dvoid **)0)); + + size=0; + break; + } + default: + coerce_type=SQLT_STR; + 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; - } - default: - coerce_type=SQLT_STR; - ptr=cols[column_count-1].str=(char *)services.malloc_atomic(MAX_OUT_STRING_LENGTH+1); - size=MAX_OUT_STRING_LENGTH; - break; } - cols[column_count-1].type=coerce_type; + col.type=coerce_type; - check(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)); } - check(cs, handlers.before_rows(cs.sql_error)); + check(connection, handlers.before_rows(connection.sql_error)); - for(unsigned long row=0; !limit||row=offset) { - check(cs, handlers.add_row(cs.sql_error)); + check(connection, handlers.add_row(connection.sql_error)); for(int i=0; irequest_charset())!=0); + } + private: // conn client library funcs - friend void check(OracleSQL_connection_struct &cs, const char *step, sword status); + friend void fail(Connection& connection, const char *msg); + friend void check(Connection& connection, const char *step, sword status); friend sb4 cbf_get_data(dvoid *ctxp, OCIBind *bindp, ub4 iter, ub4 index, @@ -803,6 +1057,11 @@ private: // conn client library funcs ub2 dty, dvoid *indp, ub2 *alenp, ub2 *rcodep, ub4 maxarr_len, ub4 *curelep, ub4 mode)); + OCI_DECL(BindByName, (OCIStmt *stmtp, OCIBind **bindp, OCIError *errhp, + text* placeholder, sb4 placeh_len, dvoid *valuep, sb4 value_sz, + ub2 dty, dvoid *indp, ub2 *alenp, ub2 *rcodep, + ub4 maxarr_len, ub4 *curelep, ub4 mode)); + OCI_DECL(BindDynamic, (OCIBind *bindp, OCIError *errhp, dvoid *ictxp, OCICallbackInBind icbfp, dvoid *octxp, OCICallbackOutBind ocbfp)); @@ -894,7 +1153,7 @@ private: // conn client library funcs li OCI_LINK(Initialize); OCI_LINK(EnvInit); OCI_LINK(AttrGet); OCI_LINK(AttrSet); - OCI_LINK(BindByPos); OCI_LINK(BindDynamic); + OCI_LINK(BindByPos); OCI_LINK(BindByName); OCI_LINK(BindDynamic); OCI_LINK(DefineByPos); OCI_LINK(DescriptorAlloc); OCI_LINK(DescriptorFree); OCI_LINK(ErrorGet); @@ -912,12 +1171,11 @@ private: // conn client library funcs li } *OracleSQL_driver; -void check(OracleSQL_connection_struct &cs, const char *step, sword status) { +void check(Connection& connection, const char *step, sword status) { const char *msg; char reason[MAX_STRING/2]; - const char *prefix="ERROR"; switch (status) { case OCI_SUCCESS: // hurrah case OCI_SUCCESS_WITH_INFO: // ignoring. example: count(column) when column contains NULLs, @@ -925,13 +1183,23 @@ void check(OracleSQL_connection_struct & return; case OCI_ERROR: { - sb4 errcode; - if(OracleSQL_driver->OCIErrorGet((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 server error message from ?ClientCharset to $request:charset + if(msg && connection.options.client_charset && strcmp(connection.options.client_charset, connection.services->request_charset())!=0){ + if(size_t msg_length=strlen(msg)){ + connection.services->transcode(msg, msg_length, + msg, msg_length, + connection.options.client_charset, + connection.services->request_charset()); + } + } + } else + msg="[can not get error description]"; + break; } case OCI_NEED_DATA: msg="NEED_DATA"; break; @@ -947,23 +1215,28 @@ void check(OracleSQL_connection_struct & 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(OracleSQL_connection_struct &cs, bool error) { +void check(Connection& connection, bool error) { if(error) - longjmp(cs.mark, 1); + 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, @@ -982,30 +1255,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.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_atomic(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.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)); @@ -1018,12 +1290,6 @@ static sb4 cbf_get_data(dvoid *ctxp, return OCI_CONTINUE; } -void tolower(char *out, const char *in, size_t size) { - while(size--) - *out++=tolower(*in++); -} - extern "C" SQL_Driver *SQL_DRIVER_CREATE() { - //_asm int 3; return OracleSQL_driver=new OracleSQL_Driver(); }