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