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