![[BACK]](/cvsweb/icons/back.gif){kind=icon}
Return to parser3oracle.C CVS log ![[TXT]](/cvsweb/icons/text.gif){kind=icon}
![[DIR]](/cvsweb/icons/dir.gif){kind=icon}
Up to [parser3project] / sql / oracle|
Return to parser3oracle.C CVS log | Up to [parser3project] / sql / oracle |
1.1 parser 1: /** @file
2: Parser Oracle driver.
3:
1.29 paf 4: Copyright(c) 2001, 2003 ArtLebedev Group (http://www.artlebedev.com)
1.1 parser 5:
1.19 paf 6: Author: Alexandr Petrosian <paf@design.ru> (http://paf.design.ru)
1.1 parser 7:
8: 2001.07.30 using Oracle 8.1.6 [@test tested with Oracle 7.x.x]
9: */
1.42 ! paf 10: static const char *RCSId="$Id: parser3oracle.C,v 1.41 2003/12/15 08:57:22 paf Exp $";
1.1 parser 11:
12: #include "config_includes.h"
13:
14: #include "pa_sql_driver.h"
15:
16: #include <oci.h>
17:
18: #define MAX_COLS 500
19: #define MAX_IN_LOBS 5
20: #define MAX_LOB_NAME_LENGTH 100
21: #define MAX_OUT_STRING_LENGTH 4000
22:
23: #define EMPTY_CLOB_FUNC_CALL "empty_clob()"
24:
25: #include "ltdl.h"
26:
27: #define MAX_STRING 0x400
28: #define MAX_NUMBER 20
29:
30: #if _MSC_VER
31: # define snprintf _snprintf
32: # define strcasecmp _stricmp
33: # define strncasecmp _strnicmp
34: #endif
35:
36: #ifndef max
37: inline int max(int a, int b) { return a>b?a:b; }
38: inline int min(int a, int b){ return a<b?a:b; }
39: #endif
40:
1.33 paf 41: /// @todo small memory leaks here
1.10 paf 42: static int pa_setenv(const char *name, const char *value, bool do_append) {
43: const char *prev_value=0;
44: if(do_append)
45: prev_value=getenv(name);
1.4 paf 46: #ifdef HAVE_PUTENV
47: // MEM_LEAK_HERE. refer to EOF man putenv
1.10 paf 48: char *buf=(char *)::malloc(strlen(name)
49: +1
50: +(prev_value?strlen(prev_value):0)
51: +strlen(value)
52: +1);
1.4 paf 53: strcpy(buf, name);
54: strcat(buf, "=");
1.10 paf 55: if(prev_value)
56: strcat(buf, prev_value);
1.4 paf 57: strcat(buf, value);
1.5 paf 58: /*
59: if(FILE *f=fopen("f", "at")) {
60: fprintf(f, "****************************%s\n", buf);
61: // for (char **env = environ; env != NULL && *env != NULL; env++)
62: // fputs(*env, f);
63:
64: fclose(f);
65: }
66: */
1.4 paf 67: return putenv(buf);
68: #else
69: //#ifdef HAVE_SETENV
1.10 paf 70: if(value) {
71: if(prev_value) {
72: // MEM_LEAK_HERE
1.33 paf 73: char *buf=(char *)::malloc(strlen(prev_value)
1.10 paf 74: +strlen(value)
75: +1);
76: strcpy(buf, prev_value);
77: strcat(buf, value);
1.24 paf 78: value=buf;
79: }
1.10 paf 80:
1.24 paf 81: return setenv(name, value, 1/*overwrite*/);
1.10 paf 82: } else {
1.4 paf 83: unsetenv(name);
84: return 0;
85: }
86: #endif
87: }
88:
1.1 parser 89: static char *lsplit(char *string, char delim) {
90: if(string) {
91: char *v=strchr(string, delim);
92: if(v) {
93: *v=0;
94: return v+1;
95: }
96: }
97: return 0;
98: }
99:
1.4 paf 100: static char *lsplit(char **string_ref, char delim) {
101: char *result=*string_ref;
102: char *next=lsplit(*string_ref, delim);
103: *string_ref=next;
104: return result;
105: }
106:
1.1 parser 107: #ifndef DOXYGEN
108: struct OracleSQL_connection_struct {
1.42 ! paf 109: SQL_Driver_services *services;
! 110:
1.1 parser 111: jmp_buf mark; char error[MAX_STRING];
1.27 paf 112: SQL_Error sql_error;
1.1 parser 113: OCIEnv *envhp;
114: OCIServer *srvhp;
115: OCIError *errhp;
116: OCISvcCtx *svchp;
117: OCISession *usrhp;
1.39 paf 118:
1.42 ! paf 119: struct Options {
! 120: bool bLowerCaseColumnNames;
! 121: const char* cstrClientCharset;
! 122: } options;
1.1 parser 123: };
124:
125: struct OracleSQL_query_lobs {
126: struct return_rows {
127: struct return_row {
128: OCILobLocator *locator; ub4 len;
129: int ind;
130: ub2 rcode;
131: } *row;
132: int count;
133: };
134: struct cbf_context_struct {
135: OracleSQL_connection_struct *cs;
136: return_rows *rows;
137: };
138: struct Item {
139: const char *name_ptr; size_t name_size;
140: char *data_ptr; size_t data_size;
141: OCILobLocator *locator;
142: OCIBind *bind;
143: return_rows rows;
144: } items[MAX_IN_LOBS];
145: int count;
146: };
147: #endif
148:
149: // forwards
1.27 paf 150: void check(OracleSQL_connection_struct &cs, const char *step, sword status);
151: void check(OracleSQL_connection_struct &cs, bool error);
1.1 parser 152: static sb4 cbf_no_data(
153: dvoid *ctxp,
154: OCIBind *bindp,
155: ub4 iter, ub4 index,
156: dvoid **bufpp,
157: ub4 *alenpp,
158: ub1 *piecep,
159: dvoid **indpp);
160: static sb4 cbf_get_data(dvoid *ctxp,
161: OCIBind *bindp,
162: ub4 iter, ub4 index,
163: dvoid **bufpp,
164: ub4 **alenp,
165: ub1 *piecep,
166: dvoid **indpp,
167: ub2 **rcodepp);
1.38 paf 168: void tolower(char *out, const char *in, size_t size);
1.1 parser 169:
1.42 ! paf 170: static const char *options2env(char *s, OracleSQL_connection_struct::Options* options) {
! 171: while(s) {
! 172: if(char *key=lsplit(&s, '&')) {
! 173: if(*key) {
! 174: if(char *value=lsplit(key, '=')) {
! 175: if( strcmp( key, "ClientCharset" ) == 0 ) {
! 176: if(options)
! 177: options->cstrClientCharset = value;
! 178: continue;
! 179: }
! 180:
! 181: if( strcmp( key, "LowerCaseColumnNames" ) == 0 ) {
! 182: if(options)
! 183: options->bLowerCaseColumnNames = atoi(value)!=0;
! 184: continue;
! 185: }
! 186:
! 187: bool do_append=key[strlen(key)-1]=='+'; // PATH+=
! 188: if(do_append)
! 189: key[strlen(key)-1]=0; // remove trailing +
! 190: if(strncmp(key, "ORACLE_", 7)==0 // ORACLE_HOME & co
! 191: || strncmp(key, "ORA_", 4)==0 // ORA_ENCRYPT_LOGIN & co
! 192: || strncmp(key, "NLS_", 4)==0 // NLS_LANG & co
! 193: || do_append
! 194: ) {
! 195: if(pa_setenv(key, value, do_append)!=0)
! 196: return "problem changing process environment" /*key*/;
! 197: } else
! 198: return "unknown option" /*key*/;
! 199: } else
! 200: return "option without =value" /*key*/;
! 201: }
! 202: }
! 203: }
! 204: return 0;
! 205: }
! 206:
1.1 parser 207: /**
208: OracleSQL server driver
209: */
210: class OracleSQL_Driver : public SQL_Driver {
211: public:
212:
213: OracleSQL_Driver() : SQL_Driver() {
214: }
215:
216: /// get api version
217: int api_version() { return SQL_DRIVER_API_VERSION; }
1.6 paf 218: /** initialize driver by loading sql dynamic link library
219: @todo ?objects=1 which would turn on OCI_OBJECT init flag
220: */
1.5 paf 221: const char *initialize(char *dlopen_file_spec) {
222: char *options=lsplit(dlopen_file_spec, '?');
1.1 parser 223:
224: const char *error=dlopen_file_spec?
225: dlink(dlopen_file_spec):"client library column is empty";
226: if(!error) {
1.39 paf 227: error=options2env(options, 0);
1.1 parser 228:
1.5 paf 229: if(!error)
230: OCIInitialize((ub4)OCI_THREADED/*| OCI_OBJECT*/, (dvoid *)0,
231: (dvoid * (*)(void *, unsigned int))0,
232: (dvoid * (*)(void*, void*, unsigned int))0,
233: (void (*)(void*, void*))0
234: );
1.1 parser 235: }
236:
237: return error;
238: }
239:
240: /** connect
241: @param used_only_in_connect_url
1.4 paf 242: format: @b user:pass@service?
243: ORACLE_HOME=/u01/app/oracle/product/8.1.5&
1.5 paf 244: ORA_NLS33=/u01/app/oracle/product/8.1.5/ocommon/nls/admin/data&
1.4 paf 245: NLS_LANG=RUSSIAN_AMERICA.CL8MSWIN1251&
246: ORA_ENCRYPT_LOGIN=TRUE
247:
248: @todo environment manupulation doesnt look thread safe
1.1 parser 249: */
250: void connect(
251: char *used_only_in_connect_url,
252: SQL_Driver_services& services,
253: void **connection ///< output: OracleSQL_connection_struct *
254: ) {
255: // connections are cross-request, do not use services._alloc [linked with request]
256: OracleSQL_connection_struct &cs=
257: *(OracleSQL_connection_struct *)::calloc(sizeof(OracleSQL_connection_struct), 1);
1.42 ! paf 258: cs.services=&services;
! 259: cs.options.bLowerCaseColumnNames = true;
1.1 parser 260:
261: char *user=used_only_in_connect_url;
262: char *service=lsplit(user, '@');
263: char *pwd=lsplit(user, ':');
1.4 paf 264: char *options=lsplit(service, '?');
1.1 parser 265:
266: if(!(user && pwd && service))
267: services._throw("mailformed connect part, must be 'user:pass@service'");
268:
1.42 ! paf 269: if(const char *error=options2env(options, &cs.options))
1.5 paf 270: services._throw(error);
1.4 paf 271:
1.1 parser 272: if(setjmp(cs.mark))
273: services._throw(cs.error);
274:
275: // Allocate and initialize OCIError handle, attempt #1
276: /*
277: grabbed from sample
278: /server.804/a58234/oci_func.htm#446192
279: but doc
280: /server.804/a58234/oci_func.htm#446100
281: doesnt have this param listed as allowed
282: 8.1.6 client library barks as OCI_INVALID_HANDLE
283: and debugging revealed that OCI_HTYPE_ENV param value is invalid
284: later in doc
285: /server.804/a58234/oci_func.htm#446192
286: on OCIEnvInit thay say
287: "No changes are done to an already initialized handle"
288: think, this is some sort of backward compatibility wonder.
289: leaving as it is, and without check()
290: */
291: OCIHandleAlloc((dvoid *)NULL, (dvoid **) &cs.envhp, (ub4)OCI_HTYPE_ENV, 0, 0);
292: // Initialize an environment handle, attempt #2
1.27 paf 293: check(cs, "EnvInit", OCIEnvInit(
1.1 parser 294: &cs.envhp, (ub4)OCI_DEFAULT, 0, 0));
295: // Allocate and initialize OCIError handle
1.27 paf 296: check(cs, "HandleAlloc errhp", OCIHandleAlloc(
1.1 parser 297: (dvoid *)cs.envhp, (dvoid **) &cs.errhp, (ub4)OCI_HTYPE_ERROR, 0, 0));
298: // Allocate and initialize OCIServer handle
1.27 paf 299: check(cs, "HandleAlloc srvhp", OCIHandleAlloc(
1.1 parser 300: (dvoid *)cs.envhp, (dvoid **) &cs.srvhp, (ub4)OCI_HTYPE_SERVER, 0, 0));
301: // Attach to a 'service'; initialize server context handle
1.27 paf 302: check(cs, "ServerAttach", OCIServerAttach(
1.1 parser 303: cs.srvhp, cs.errhp, (text *)service, (sb4)strlen(service), (ub4)OCI_DEFAULT));
304: // Allocate and initialize OCISvcCtx handle
1.27 paf 305: check(cs, "HandleAlloc svchp", OCIHandleAlloc(
1.1 parser 306: (dvoid *)cs.envhp, (dvoid **) &cs.svchp, (ub4)OCI_HTYPE_SVCCTX, 0, 0));
307: // set attribute server context in the service context
1.27 paf 308: check(cs, "AttrSet server-service", OCIAttrSet(
1.1 parser 309: (dvoid *)cs.svchp, (ub4)OCI_HTYPE_SVCCTX,
310: (dvoid *)cs.srvhp, (ub4)0,
311: (ub4)OCI_ATTR_SERVER, (OCIError *)cs.errhp));
312: // allocate a user context handle
1.27 paf 313: check(cs, "HandleAlloc usrhp", OCIHandleAlloc(
1.1 parser 314: (dvoid *)cs.envhp, (dvoid **)&cs.usrhp, (ub4)OCI_HTYPE_SESSION, 0, 0));
315: // set 'user' name
1.27 paf 316: check(cs, "AttrSet user-session", OCIAttrSet(
1.1 parser 317: (dvoid *)cs.usrhp, (ub4)OCI_HTYPE_SESSION,
318: (dvoid *)user, (ub4)strlen(user),
319: OCI_ATTR_USERNAME, cs.errhp));
320: // set 'pwd' password
1.27 paf 321: check(cs, "AttrSet pwd-session", OCIAttrSet(
1.1 parser 322: (dvoid *)cs.usrhp, (ub4)OCI_HTYPE_SESSION,
323: (dvoid *)pwd, (ub4)strlen(pwd),
324: OCI_ATTR_PASSWORD, cs.errhp));
325: // Authenticate a user
1.27 paf 326: check(cs, "SessionBegin", OCISessionBegin(
1.1 parser 327: cs.svchp, cs.errhp, cs.usrhp,
328: OCI_CRED_RDBMS, OCI_DEFAULT));
329: // remember connection in session
1.27 paf 330: check(cs, "AttrSet service-session", OCIAttrSet(
1.1 parser 331: (dvoid *)cs.svchp, (ub4)OCI_HTYPE_SVCCTX,
332: (dvoid *)cs.usrhp, (ub4)0,
333: OCI_ATTR_SESSION, cs.errhp));
334:
335: // return created connection
336: *(OracleSQL_connection_struct **)connection=&cs;
337: }
338: void disconnect(void *connection) {
339: OracleSQL_connection_struct &cs=*(OracleSQL_connection_struct *)connection;
340: // Terminate a user session
341: OCISessionEnd(
342: cs.svchp, cs.errhp, cs.usrhp, (ub4)OCI_DEFAULT);
343: // Detach from a server; uninitialize server context handle
344: OCIServerDetach(
345: cs.srvhp, cs.errhp, (ub4)OCI_DEFAULT);
346: // Free a previously allocated handles
347: /*
348: oci will free them up as belonging to env
349: OCIHandleFree(
350: (dvoid *)cs.srvhp, (ub4)OCI_HTYPE_SERVER);
351: OCIHandleFree(
352: (dvoid *)cs.svchp, (ub4)OCI_HTYPE_SVCCTX);
353: OCIHandleFree(
354: (dvoid *)cs.errhp, (ub4)OCI_HTYPE_ERROR);
355: */
356: OCIHandleFree(
357: (dvoid *)cs.envhp, (ub4)OCI_HTYPE_ENV);
358:
359: // connections are cross-request, do not use services._alloc [linked with request]
360: ::free(&cs);
361: }
1.42 ! paf 362: void commit(void *connection) {
1.1 parser 363: OracleSQL_connection_struct &cs=*(OracleSQL_connection_struct *)connection;
364: if(setjmp(cs.mark))
1.42 ! paf 365: cs.services->_throw(cs.error);
1.1 parser 366:
1.27 paf 367: check(cs, "commit", OCITransCommit(cs.svchp, cs.errhp, 0));
1.1 parser 368: }
1.42 ! paf 369: void rollback(void *connection) {
1.1 parser 370: OracleSQL_connection_struct &cs=*(OracleSQL_connection_struct *)connection;
371: if(setjmp(cs.mark))
1.42 ! paf 372: cs.services->_throw(cs.error);
1.1 parser 373:
1.42 ! paf 374: // sometimes rollback is done in context when this yields error which masks previous error
! 375: // consider consequent errors not very important to report, reporting first one
! 376: /*check(cs, "rollback", */OCITransRollback(cs.svchp, cs.errhp, 0)/*)*/;
1.1 parser 377: }
378:
1.42 ! paf 379: bool ping(void *connection) {
1.1 parser 380: // maybe OCIServerVersion?
1.4 paf 381: // select 0 from dual
1.1 parser 382: return true;
383: }
384:
1.42 ! paf 385: const char* quote(void *connection,
! 386: const char *from, unsigned int length)
! 387: {
! 388: OracleSQL_connection_struct &cs=*(OracleSQL_connection_struct *)connection;
! 389: char *result=(char*)cs.services->malloc_atomic(length*2+1);
1.32 paf 390: char *to=result;
391: while(length--) {
392: switch(*from) {
393: case '\'': // "'" -> "''"
1.35 paf 394: *to++='\'';
1.32 paf 395: break;
1.1 parser 396: }
1.32 paf 397: *to++=*from++;
398: }
399: *to=0;
400: return result;
1.1 parser 401: }
1.42 ! paf 402: void query(void *connection,
1.1 parser 403: const char *astatement, unsigned long offset, unsigned long limit,
1.42 ! paf 404: SQL_Driver_query_event_handlers& handlers)
! 405: {
1.1 parser 406: OracleSQL_connection_struct &cs=*(OracleSQL_connection_struct *)connection;
407: OracleSQL_query_lobs lobs={{0}, 0};
408: OCIStmt *stmthp=0;
409:
1.42 ! paf 410: SQL_Driver_services& services=*cs.services;
! 411:
! 412: // transcode from $request:charset to connect-string?client_charset
! 413: size_t transcoded_statement_size;
! 414: if(const char* cstrClientCharset=cs.options.cstrClientCharset)
! 415: services.transcode(astatement, strlen(astatement),
! 416: astatement, transcoded_statement_size,
! 417: services.request_charset(),
! 418: cstrClientCharset);
! 419:
1.1 parser 420: bool failed=false;
421: if(setjmp(cs.mark)) {
422: failed=true;
423: goto cleanup;
424: } else {
1.42 ! paf 425: const char *statement=preprocess_statement(cs, astatement, lobs);
1.1 parser 426:
1.27 paf 427: check(cs, "HandleAlloc STMT", OCIHandleAlloc(
1.1 parser 428: (dvoid *)cs.envhp, (dvoid **) &stmthp, (ub4)OCI_HTYPE_STMT, 0, 0));
1.27 paf 429: check(cs, "syntax",
1.1 parser 430: OCIStmtPrepare(stmthp, cs.errhp, (unsigned char *)statement,
431: (ub4)strlen((char *)statement),
432: (ub4)OCI_NTV_SYNTAX, (ub4)OCI_DEFAULT));
433: {
434: for(int i=0; i<lobs.count; i++) {
1.27 paf 435: check(cs, "alloc output var desc", OCIDescriptorAlloc(
1.1 parser 436: (dvoid *)cs.envhp, (dvoid **)&lobs.items[i].locator, (ub4)OCI_DTYPE_LOB, 0, 0));
437:
1.27 paf 438: check(cs, "bind output", OCIBindByPos(stmthp,
1.1 parser 439: &lobs.items[i].bind, cs.errhp,
440: (ub4)1+i,
441: (dvoid *)&lobs.items[i].locator,
442: (sword)sizeof (lobs.items[i].locator), SQLT_CLOB, (dvoid *)0,
443: (ub2 *)0, (ub2 *)0, (ub4)0, (ub4 *)0, OCI_DATA_AT_EXEC));
444:
445: lobs.items[i].rows.count=0;
1.42 ! paf 446: OracleSQL_query_lobs::cbf_context_struct cbf_context={&cs, &lobs.items[i].rows};
1.27 paf 447: check(cs, "bind dynamic", OCIBindDynamic(
1.1 parser 448: lobs.items[i].bind, cs.errhp,
449: (dvoid *) &cbf_context, cbf_no_data,
450: (dvoid *) &cbf_context, cbf_get_data));
451: }
452: }
453:
1.26 paf 454: execute_prepared(services, cs,
455: statement, stmthp, lobs,
456: offset, limit, handlers);
1.1 parser 457: }
458: cleanup: // no check call after this point!
459: {
460: for(int i=0; i<lobs.count; i++) {
461: /* free var locator */
462: if(OCILobLocator *locator=lobs.items[i].locator)
463: OCIDescriptorFree((dvoid *)locator, (ub4)OCI_DTYPE_LOB);
464:
465: /* free rows descriptors */
466: OracleSQL_query_lobs::return_rows &rows=lobs.items[i].rows;
467: for(int r=0; r<rows.count; r++)
468: OCIDescriptorFree((dvoid *)rows.row[r].locator, (ub4)OCI_DTYPE_LOB);
469: }
470: }
471: if(stmthp)
472: OCIHandleFree((dvoid *)stmthp, (ub4)OCI_HTYPE_STMT);
473:
1.26 paf 474: if(failed) {
1.27 paf 475: if(cs.sql_error.defined())
476: services._throw(cs.sql_error);
1.1 parser 477: services._throw(cs.error);
1.26 paf 478: }
1.1 parser 479: }
480:
481: private: // private funcs
482:
1.42 ! paf 483: const char *preprocess_statement(OracleSQL_connection_struct &cs,
1.1 parser 484: const char *astatement, OracleSQL_query_lobs &lobs) {
485: size_t statement_size=strlen(astatement);
1.42 ! paf 486: SQL_Driver_services& services=*cs.services;
1.1 parser 487:
1.32 paf 488: char *result=(char *)services.malloc_atomic(statement_size
1.1 parser 489: +MAX_STRING // in case of short 'strings'
490: +11/* returning */+6/* into */+(MAX_LOB_NAME_LENGTH+2/*:, */)*2/*ret into*/*MAX_IN_LOBS
491: +1);
492: const char *o=astatement;
493:
494: // /**xxx**/'literal' -> EMPTY_CLOB_FUNC_CALL
495: char *n=result;
496: while(*o) {
497: if(
498: o[0]=='/' &&
499: o[1]=='*' &&
500: o[2]=='*') { // name start
1.34 paf 501: const char* saved_o=o;
1.1 parser 502: o+=3;
503: const char *name_begin=o;
504: while(*o)
505: if(
506: o[0]=='*' &&
507: o[1]=='*' &&
508: o[2]=='/' &&
509: o[3]=='\'') { // name end
1.34 paf 510: saved_o=0; // found, marking that
1.1 parser 511: const char *name_end=o;
512: o+=4;
513: OracleSQL_query_lobs::Item &item=lobs.items[lobs.count++];
514: item.name_ptr=name_begin; item.name_size=name_end-name_begin;
1.32 paf 515: item.data_ptr=(char *)services.malloc_atomic(statement_size/*max*/); item.data_size=0;
1.1 parser 516:
517: const char *start=o;
518: bool escaped=false;
519: while(*o && !(o[0]=='\'' && o[1]!='\'' && !escaped)) {
1.14 paf 520: escaped=o[0]=='\'' && o[1]=='\'';
1.1 parser 521: if(escaped) {
522: // write pending, skip "\" or "'"
523: if(size_t size=o-start) {
524: memcpy(item.data_ptr+item.data_size, start, size);
525: item.data_size+=size;
526: }
527: start=++o;
528: } else
529: o++;
530: }
531: if(size_t size=o-start) {
532: memcpy(item.data_ptr+item.data_size, start, size);
533: item.data_size+=size;
534: }
535: if(*o)
536: o++; // skip "'"
537:
538: n+=sprintf(n, EMPTY_CLOB_FUNC_CALL);
539: break;
540: } else
541: o++; // /**skip**/'xxx'
1.34 paf 542: if(saved_o) {
543: o=saved_o;
544: *n++=*o++;
545: }
1.1 parser 546: } else
547: *n++=*o++;
548: }
549: *n=0;
550:
551: if(lobs.count) {
552: int i;
553: n+=sprintf(n, " returning ");
554: for(i=0; i<lobs.count; i++) {
555: if(i)
556: *n++=',';
557: n+=sprintf(n, "%.*s", lobs.items[i].name_size, lobs.items[i].name_ptr);
558: }
559: n+=sprintf(n, " into ");
560: for(i=0; i<lobs.count; i++) {
561: if(i)
1.41 paf 562: *n++=',';
1.1 parser 563: n+=sprintf(n, ":%.*s", lobs.items[i].name_size, lobs.items[i].name_ptr);
564: }
565: }
566:
567: return result;
568: }
569:
570: void execute_prepared(
571: SQL_Driver_services& services, OracleSQL_connection_struct &cs,
572: const char *statement, OCIStmt *stmthp, OracleSQL_query_lobs &lobs,
573: unsigned long offset, unsigned long limit,
574: SQL_Driver_query_event_handlers& handlers) {
575:
576: ub2 stmt_type=0; // UNKNOWN
577: /*
578: //gpfs on sun. paf 000818
579: //Zanyway, this is needed before.
1.27 paf 580: check(cs, "get stmt type", OCIAttrGet(
1.1 parser 581: (dvoid *)stmthp, (ub4)OCI_HTYPE_STMT, (ub1 *)&stmt_type,
582: (ub4 *)0, OCI_ATTR_STMT_TYPE, cs.errhp));
583: */
1.16 paf 584:
1.17 paf 585: while(isspace(*statement))
1.16 paf 586: statement++;
1.1 parser 587: if(strncasecmp(statement, "select", 6)==0)
588: stmt_type=OCI_STMT_SELECT;
589: else if(strncasecmp(statement, "insert", 6)==0)
590: stmt_type=OCI_STMT_INSERT;
591: else if(strncasecmp(statement, "update", 6)==0)
592: stmt_type=OCI_STMT_UPDATE;
593:
594: sword status=OCIStmtExecute(cs.svchp, stmthp, cs.errhp,
595: (ub4)stmt_type==OCI_STMT_SELECT?0:1, (ub4)0,
596: (OCISnapshot *)NULL,
597: (OCISnapshot *)NULL, (ub4)OCI_DEFAULT);
598:
599: if(status!=OCI_NO_DATA)
1.27 paf 600: check(cs, "execute", status);
1.1 parser 601:
602: {
603: for(int i=0; i<lobs.count; i++)
604: if(ub4 bytes_to_write=lobs.items[i].data_size) {
605: OracleSQL_query_lobs::return_rows *rows=&lobs.items[i].rows;
606: for(int r=0; r<rows->count; r++) {
607: OCILobLocator *locator=rows->row[r].locator;
1.27 paf 608: check(cs, "lobwrite", OCILobWrite (
1.1 parser 609: cs.svchp, cs.errhp,
610: locator, &bytes_to_write, 1,
611: (dvoid *)lobs.items[i].data_ptr, (ub4)bytes_to_write, OCI_ONE_PIECE,
612: (dvoid *)0, 0, (ub2)0,
613: (ub1) SQLCS_IMPLICIT));
614: }
615: }
616: }
617:
618: switch(stmt_type) {
619: case OCI_STMT_SELECT:
1.42 ! paf 620: fetch_table(cs,
1.1 parser 621: stmthp, offset, limit,
622: handlers);
623: break;
624: default:
625: /*
626: case OCI_STMT_INSERT:
627: case OCI_STMT_UPDATE:
628: */
629: break;
630: }
631: }
632:
1.42 ! paf 633: void fetch_table(OracleSQL_connection_struct &cs,
1.1 parser 634: OCIStmt *stmthp, unsigned long offset, unsigned long limit,
1.42 ! paf 635: SQL_Driver_query_event_handlers& handlers)
! 636: {
! 637: SQL_Driver_services& services=*cs.services;
1.12 paf 638:
1.10 paf 639: ub4 prefetch_rows=100;
1.27 paf 640: check(cs, "AttrSet prefetch-rows", OCIAttrSet(
1.9 paf 641: (dvoid *)stmthp, (ub4)OCI_HTYPE_STMT,
642: (dvoid *)&prefetch_rows, (ub4)0,
643: (ub4)OCI_ATTR_PREFETCH_ROWS, (OCIError *)cs.errhp));
644:
1.20 paf 645: ub4 prefetch_mem_size=100*0x400;
1.27 paf 646: check(cs, "AttrSet prefetch-memory", OCIAttrSet(
1.9 paf 647: (dvoid *)stmthp, (ub4)OCI_HTYPE_STMT,
648: (dvoid *)&prefetch_mem_size, (ub4)0,
649: (ub4)OCI_ATTR_PREFETCH_MEMORY, (OCIError *)cs.errhp));
1.1 parser 650:
651: OCIParam *mypard;
652: ub2 dtype;
653: text *col_name;
654:
1.40 paf 655: struct Col {
1.1 parser 656: ub2 type;
657: char *str;
658: OCILobLocator *var;
659: OCIDefine *def;
660: sb2 indicator;
661: } cols[MAX_COLS]={0};
662: int column_count=0;
663:
664: bool failed=false;
665: jmp_buf saved_mark; memcpy(saved_mark, cs.mark, sizeof(jmp_buf));
666: if(setjmp(cs.mark)) {
667: failed=true;
668: goto cleanup;
669: } else {
1.27 paf 670: // idea of preincrementing is that at error time all handles would free up
671: while(++column_count<=MAX_COLS) {
672: /* get next descriptor, if there is one */
673: if(OCIParamGet(stmthp, OCI_HTYPE_STMT, cs.errhp, (void **)&mypard,
674: (ub4) column_count)!=OCI_SUCCESS) {
675: --column_count;
676: break;
677: }
678:
679: /* Retrieve the data type attribute */
680: check(cs, "get type", OCIAttrGet(
681: (dvoid*) mypard, (ub4)OCI_DTYPE_PARAM,
682: (dvoid*) &dtype, (ub4 *)0, (ub4)OCI_ATTR_DATA_TYPE,
683: (OCIError *)cs.errhp));
684:
685: /* Retrieve the column name attribute */
686: ub4 col_name_len;
687: check(cs, "get name", OCIAttrGet(
688: (dvoid*) mypard, (ub4)OCI_DTYPE_PARAM,
689: (dvoid**) &col_name, (ub4 *) &col_name_len, (ub4)OCI_ATTR_NAME,
690: (OCIError *)cs.errhp));
691:
1.40 paf 692: Col& col=cols[column_count-1];
1.27 paf 693: {
1.38 paf 694: size_t length=(size_t)col_name_len;
695: char *ptr=(char *)services.malloc_atomic(length+1);
1.42 ! paf 696: if( cs.options.bLowerCaseColumnNames )
1.39 paf 697: tolower(ptr, (char *)col_name, length);
698: else
699: memcpy(ptr, col_name, length);
1.38 paf 700: ptr[length]=0;
701: check(cs, handlers.add_column(cs.sql_error, ptr, length));
1.27 paf 702: }
703:
704: ub2 coerce_type=dtype;
705: sb4 size=0;
706: void *ptr;
707:
708: switch(dtype) {
709: case SQLT_CLOB:
1.1 parser 710: {
1.27 paf 711: check(cs, "alloc output var desc", OCIDescriptorAlloc(
1.40 paf 712: (dvoid *)cs.envhp, (dvoid **)(ptr=&col.var),
1.27 paf 713: (ub4)OCI_DTYPE_LOB,
714: 0, (dvoid **)0));
715:
716: size=0;
1.1 parser 717: break;
718: }
1.27 paf 719: default:
720: coerce_type=SQLT_STR;
1.40 paf 721: ptr=col.str=(char *)services.malloc_atomic(MAX_OUT_STRING_LENGTH+1);
1.27 paf 722: size=MAX_OUT_STRING_LENGTH;
723: break;
1.1 parser 724: }
725:
1.40 paf 726: col.type=coerce_type;
1.1 parser 727:
1.40 paf 728: col.def=0; check(cs, "DefineByPos", OCIDefineByPos(
729: stmthp, &col.def, cs.errhp,
1.27 paf 730: column_count, (ub1 *) ptr, size,
1.40 paf 731: coerce_type, (dvoid *) &col.indicator,
1.27 paf 732: (ub2 *)0, (ub2 *)0, OCI_DEFAULT));
733: }
734:
735: check(cs, handlers.before_rows(cs.sql_error));
736:
737: for(unsigned long row=0; !limit||row<offset+limit; row++) {
738: sword status=OCIStmtFetch(stmthp, cs.errhp, (ub4)1, (ub4)OCI_FETCH_NEXT,
739: (ub4)OCI_DEFAULT);
740: if(status==OCI_NO_DATA)
741: break;
742: check(cs, "fetch", status);
1.3 paf 743:
1.27 paf 744: if(row>=offset) {
745: check(cs, handlers.add_row(cs.sql_error));
746: for(int i=0; i<column_count; i++) {
1.37 paf 747: size_t length=0;
1.42 ! paf 748: char* strm=0;
1.27 paf 749: if(!cols[i].indicator) // not NULL
750: switch(cols[i].type) {
751: case SQLT_CLOB:
752: {
753: ub4 offset=1;
754: ub4 loblen=0;
755: OCILobLocator *var=(OCILobLocator *)cols[i].var;
1.42 ! paf 756: size_t read_size=0;
! 757: strm=(char*)services.malloc_atomic(1); // set type of memory block
! 758: do {
! 759: char buf[MAX_STRING*10];
! 760: ub4 amtp=sizeof(buf)-1;
! 761: status=OCILobRead(cs.svchp, cs.errhp,
! 762: var, &amtp, offset, (dvoid *)buf,
! 763: sizeof(buf),
! 764: (dvoid *)0, 0,
! 765: (ub2)0, (ub1)SQLCS_IMPLICIT);
! 766: if(status!=OCI_SUCCESS && status!=OCI_NEED_DATA)
! 767: check(cs, "lobread", status);
! 768:
! 769: strm=(char*)services.realloc(strm, read_size+amtp+1/*for zero termintator*/);
! 770: memcpy(strm+read_size, buf, amtp);
! 771: read_size+=amtp;
! 772: offset+=amtp;
! 773: } while(status==OCI_NEED_DATA);
! 774:
! 775: length=(size_t)read_size;
! 776: strm[length]=0;
1.1 parser 777: break;
778: }
1.27 paf 779: default:
1.32 paf 780: if(const char *value=cols[i].str) {
1.37 paf 781: length=strlen(value);
1.42 ! paf 782: strm=(char*)services.malloc_atomic(length+1);
! 783: memcpy(strm, value, length+1);
1.27 paf 784: } else {
1.37 paf 785: length=0;
1.42 ! paf 786: strm=0;
1.27 paf 787: }
788: break;
789: }
1.42 ! paf 790:
! 791: const char* str=strm;
! 792: if(str && length)
! 793: {
! 794: // transcode to $request:charset from connect-string?client_charset
! 795: const char* dest;
! 796: size_t dest_length;
! 797: if(const char* cstrClientCharset=cs.options.cstrClientCharset)
! 798: services.transcode(str, length,
! 799: dest, dest_length,
! 800: cstrClientCharset,
! 801: services.request_charset());
! 802: str=dest;
! 803: length=dest_length;
! 804: }
! 805:
1.37 paf 806: check(cs, handlers.add_row_cell(cs.sql_error, str, length));
1.1 parser 807: }
808: }
809: }
810: }
811:
812: cleanup: // no check call after this point!
813: for(int i=0; i<column_count; i++) {
814: switch(cols[i].type) {
815: case SQLT_CLOB:
816: /* free var locator */
817: OCIDescriptorFree((dvoid *) cols[i].var, (ub4)OCI_DTYPE_LOB);
818: break;
819: default:
820: break;
821: }
822: }
823:
824: if(failed) // need rethrow?
825: longjmp(saved_mark, 1);
826: }
827:
828: private: // conn client library funcs
829:
1.27 paf 830: friend void check(OracleSQL_connection_struct &cs, const char *step, sword status);
1.1 parser 831: friend sb4 cbf_get_data(dvoid *ctxp,
832: OCIBind *bindp,
833: ub4 iter, ub4 index,
834: dvoid **bufpp,
835: ub4 **alenp,
836: ub1 *piecep,
837: dvoid **indpp,
838: ub2 **rcodepp);
839:
840:
841: #define OCI_DECL(name, params) \
842: typedef sword (*t_OCI##name)params; t_OCI##name OCI##name
843:
844: OCI_DECL(Initialize, (ub4 mode, dvoid *ctxp,
845: dvoid * (*malocfp)(dvoid *ctxp, size_t size),
846: dvoid * (*ralocfp)(dvoid *ctxp, dvoid *memptr, size_t newsize),
847: void (*mfreefp)(dvoid *ctxp, dvoid *memptr) ));
848:
849: OCI_DECL(EnvInit, (OCIEnv **envp, ub4 mode,
850: size_t xtramem_sz, dvoid **usrmempp));
851:
852: OCI_DECL(AttrGet, (CONST dvoid *trgthndlp, ub4 trghndltyp,
853: dvoid *attributep, ub4 *sizep, ub4 attrtype,
854: OCIError *errhp));
855:
856: OCI_DECL(AttrSet, (dvoid *trgthndlp, ub4 trghndltyp, dvoid *attributep,
857: ub4 size, ub4 attrtype, OCIError *errhp));
858:
859: OCI_DECL(BindByPos, (OCIStmt *stmtp, OCIBind **bindp, OCIError *errhp,
860: ub4 position, dvoid *valuep, sb4 value_sz,
861: ub2 dty, dvoid *indp, ub2 *alenp, ub2 *rcodep,
862: ub4 maxarr_len, ub4 *curelep, ub4 mode));
863:
864: OCI_DECL(BindDynamic, (OCIBind *bindp, OCIError *errhp, dvoid *ictxp,
865: OCICallbackInBind icbfp, dvoid *octxp,
866: OCICallbackOutBind ocbfp));
867:
868: OCI_DECL(DefineByPos, (OCIStmt *stmtp, OCIDefine **defnp, OCIError *errhp,
869: ub4 position, dvoid *valuep, sb4 value_sz, ub2 dty,
870: dvoid *indp, ub2 *rlenp, ub2 *rcodep, ub4 mode));
871:
872: OCI_DECL(DescriptorAlloc, (CONST dvoid *parenth, dvoid **descpp,
873: CONST ub4 type, CONST size_t xtramem_sz,
874: dvoid **usrmempp));
875:
876: OCI_DECL(DescriptorFree, (dvoid *descp, CONST ub4 type));
877:
878:
879: OCI_DECL(ErrorGet, (dvoid *hndlp, ub4 recordno, OraText *sqlstate,
880: sb4 *errcodep, OraText *bufp, ub4 bufsiz, ub4 type));
881:
882: OCI_DECL(HandleAlloc, (CONST dvoid *parenth, dvoid **hndlpp, CONST ub4 type,
883: CONST size_t xtramem_sz, dvoid **usrmempp));
884:
885: OCI_DECL(HandleFree, (dvoid *hndlp, CONST ub4 type));
886:
887: OCI_DECL(LobGetLength, (OCISvcCtx *svchp, OCIError *errhp,
888: OCILobLocator *locp,
889: ub4 *lenp));
890:
891: OCI_DECL(LobRead, (OCISvcCtx *svchp, OCIError *errhp, OCILobLocator *locp,
892: ub4 *amtp, ub4 offset, dvoid *bufp, ub4 bufl,
893: dvoid *ctxp, sb4 (*cbfp)(dvoid *ctxp,
894: CONST dvoid *bufp,
895: ub4 len,
896: ub1 piece),
897: ub2 csid, ub1 csfrm));
898:
899: OCI_DECL(LobWrite, (OCISvcCtx *svchp, OCIError *errhp, OCILobLocator *locp,
900: ub4 *amtp, ub4 offset, dvoid *bufp, ub4 buflen,
901: ub1 piece, dvoid *ctxp,
902: sb4 (*cbfp)(dvoid *ctxp,
903: dvoid *bufp,
904: ub4 *len,
905: ub1 *piece),
906: ub2 csid, ub1 csfrm));
907:
908: OCI_DECL(ParamGet, (CONST dvoid *hndlp, ub4 htype, OCIError *errhp,
909: dvoid **parmdpp, ub4 pos));
910:
911: OCI_DECL(ServerAttach, (OCIServer *srvhp, OCIError *errhp,
912: CONST OraText *dblink, sb4 dblink_len, ub4 mode));
913:
914: OCI_DECL(ServerDetach, (OCIServer *srvhp, OCIError *errhp, ub4 mode));
915:
916: OCI_DECL(SessionBegin, (OCISvcCtx *svchp, OCIError *errhp, OCISession *usrhp,
917: ub4 credt, ub4 mode));
918:
919: OCI_DECL(SessionEnd, (OCISvcCtx *svchp, OCIError *errhp, OCISession *usrhp,
920: ub4 mode));
921:
922: OCI_DECL(StmtExecute, (OCISvcCtx *svchp, OCIStmt *stmtp, OCIError *errhp,
923: ub4 iters, ub4 rowoff, CONST OCISnapshot *snap_in,
924: OCISnapshot *snap_out, ub4 mode));
925:
926: OCI_DECL(StmtFetch, (OCIStmt *stmtp, OCIError *errhp, ub4 nrows,
927: ub2 orientation, ub4 mode));
928:
929: OCI_DECL(StmtPrepare, (OCIStmt *stmtp, OCIError *errhp, CONST OraText *stmt,
930: ub4 stmt_len, ub4 language, ub4 mode));
931:
932: OCI_DECL(TransCommit, (OCISvcCtx *svchp, OCIError *errhp, ub4 flags));
933:
934: OCI_DECL(TransRollback, (OCISvcCtx *svchp, OCIError *errhp, ub4 flags));
935:
936: private: // conn client library funcs linking
937:
938: const char *dlink(const char *dlopen_file_spec) {
939: if(lt_dlinit())
940: return lt_dlerror();
941: lt_dlhandle handle=lt_dlopen(dlopen_file_spec);
942: if(!handle)
943: return lt_dlerror(); //"can not open the dynamic link module";
944:
945: #define DSLINK(name, action) \
946: name=(t_##name)lt_dlsym(handle, #name); \
947: if(!name) \
948: action;
949:
950: #define OCI_LINK(name) DSLINK(OCI##name, return "function OCI" #name " was not found")
951:
952: OCI_LINK(Initialize);
953: OCI_LINK(EnvInit);
954: OCI_LINK(AttrGet); OCI_LINK(AttrSet);
955: OCI_LINK(BindByPos); OCI_LINK(BindDynamic);
956: OCI_LINK(DefineByPos);
957: OCI_LINK(DescriptorAlloc); OCI_LINK(DescriptorFree);
958: OCI_LINK(ErrorGet);
959: OCI_LINK(HandleAlloc); OCI_LINK(HandleFree);
960: OCI_LINK(LobGetLength);
961: OCI_LINK(LobRead); OCI_LINK(LobWrite);
962: OCI_LINK(ParamGet);
963: OCI_LINK(ServerAttach); OCI_LINK(ServerDetach);
964: OCI_LINK(SessionBegin); OCI_LINK(SessionEnd);
965: OCI_LINK(StmtExecute); OCI_LINK(StmtFetch); OCI_LINK(StmtPrepare);
966: OCI_LINK(TransCommit); OCI_LINK(TransRollback);
967:
968: return 0;
969: }
970:
971: } *OracleSQL_driver;
972:
1.27 paf 973: void check(OracleSQL_connection_struct &cs, const char *step, sword status) {
1.1 parser 974:
975: const char *msg;
976: char reason[MAX_STRING/2];
977:
1.21 paf 978: const char *prefix="ERROR";
1.1 parser 979: switch (status) {
1.22 paf 980: case OCI_SUCCESS: // hurrah
981: case OCI_SUCCESS_WITH_INFO: // ignoring. example: count(column) when column contains NULLs,
982: // count() not counting them and gives that status
983: return;
1.1 parser 984: case OCI_ERROR:
985: {
986: sb4 errcode;
987: if(OracleSQL_driver->OCIErrorGet((dvoid *)cs.errhp, (ub4)1, (text *)NULL, &errcode,
988: (text *)reason, (ub4)sizeof(reason), OCI_HTYPE_ERROR)==OCI_SUCCESS)
989: msg=reason;
1.42 ! paf 990:
! 991: // transcode to $request:charset from connect-string?client_charset
! 992: if(const char* cstrClientCharset=cs.options.cstrClientCharset)
! 993: if(msg) {
! 994: if(size_t msg_length=strlen(msg)) {
! 995: cs.services->transcode(msg, msg_length,
! 996: msg, msg_length,
! 997: cstrClientCharset,
! 998: cs.services->request_charset());
! 999: }
! 1000: }
1.1 parser 1001: else
1002: msg="[can not get error description]";
1003: break;
1004: }
1005: case OCI_NEED_DATA:
1006: msg="NEED_DATA"; break;
1007: case OCI_NO_DATA:
1008: msg="NODATA"; break;
1009: case OCI_INVALID_HANDLE:
1010: msg="INVALID_HANDLE"; break;
1011: case OCI_STILL_EXECUTING:
1012: msg="STILL_EXECUTE"; break;
1013: case OCI_CONTINUE:
1014: msg="CONTINUE"; break;
1015: default:
1016: msg="unknown"; break;
1017: }
1018:
1.22 paf 1019: snprintf(cs.error, sizeof(cs.error), "%s (%s, %d)",
1020: msg, step, (int)status);
1.1 parser 1021: longjmp(cs.mark, 1);
1022: }
1023:
1.27 paf 1024: void check(OracleSQL_connection_struct &cs, bool error) {
1025: if(error)
1026: longjmp(cs.mark, 1);
1027: }
1.1 parser 1028:
1029: /* ----------------------------------------------------------------- */
1030: /* Intbind callback that does not do any data input. */
1031: /* ----------------------------------------------------------------- */
1032: static sb4 cbf_no_data(
1033: dvoid *ctxp,
1034: OCIBind *bindp,
1035: ub4 iter, ub4 index,
1036: dvoid **bufpp,
1037: ub4 *alenpp,
1038: ub1 *piecep,
1039: dvoid **indpp) {
1040: *bufpp=(dvoid *)0;
1041: *alenpp=0;
1042: static sb2 null_ind=-1;
1043: *indpp=(dvoid *) &null_ind;
1044: *piecep=OCI_ONE_PIECE;
1045:
1046: return OCI_CONTINUE;
1047: }
1048:
1049: /* ----------------------------------------------------------------- */
1050: /* Outbind callback for returning data. */
1051: /* ----------------------------------------------------------------- */
1052: static sb4 cbf_get_data(dvoid *ctxp,
1053: OCIBind *bindp,
1054: ub4 iter, ub4 index,
1055: dvoid **bufpp,
1056: ub4 **alenp,
1057: ub1 *piecep,
1058: dvoid **indpp,
1059: ub2 **rcodepp) {
1060: OracleSQL_query_lobs::cbf_context_struct &context=
1061: *(OracleSQL_query_lobs::cbf_context_struct *)ctxp;
1062:
1063: if(index==0) {
1064: static ub4 rows;
1.27 paf 1065: check(*context.cs, "AttrGet cbf_get_data ROWS_RETURNED",
1.1 parser 1066: OracleSQL_driver->OCIAttrGet(
1067: (CONST dvoid *) bindp, OCI_HTYPE_BIND, (dvoid *)&rows,
1068: (ub4 *)sizeof(ub2), OCI_ATTR_ROWS_RETURNED, context.cs->errhp)) ;
1069: context.rows->count=(ub2)rows;
1070: context.rows->row=(OracleSQL_query_lobs::return_rows::return_row *)
1.42 ! paf 1071: context.cs->services->malloc_atomic(sizeof(OracleSQL_query_lobs::return_rows::return_row)*rows);
1.1 parser 1072: }
1073:
1074: OracleSQL_query_lobs::return_rows::return_row &var=context.rows->row[index];
1075:
1.27 paf 1076: check(*context.cs, "alloc output var desc dynamic", OracleSQL_driver->OCIDescriptorAlloc(
1.1 parser 1077: (dvoid *) context.cs->envhp, (dvoid **)&var.locator,
1078: (ub4)OCI_DTYPE_LOB,
1079: 0, (dvoid **)0));
1080:
1081: *bufpp=var.locator;
1082: *alenp=&var.len;
1083: *indpp=(dvoid *) &var.ind;
1084: *piecep=OCI_ONE_PIECE;
1085: *rcodepp=&var.rcode;
1086:
1087: return OCI_CONTINUE;
1088: }
1089:
1.38 paf 1090: void tolower(char *out, const char *in, size_t size) {
1091: while(size--)
1092: *out++=tolower(*in++);
1.1 parser 1093: }
1094:
1095: extern "C" SQL_Driver *SQL_DRIVER_CREATE() {
1096: return OracleSQL_driver=new OracleSQL_Driver();
1097: }