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