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