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