|
|
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, tested with Oracle 7.x.x
9: */
1.3 ! parser 10: static const char *RCSId="$Id: parser3oracle.C,v 1.2 2001/08/22 14:45:16 parser 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:
41: static char *lsplit(char *string, char delim) {
42: if(string) {
43: char *v=strchr(string, delim);
44: if(v) {
45: *v=0;
46: return v+1;
47: }
48: }
49: return 0;
50: }
51:
52: struct OracleSQL_connection_struct {
53: jmp_buf mark; char error[MAX_STRING];
54: OCIEnv *envhp;
55: OCIServer *srvhp;
56: OCIError *errhp;
57: OCISvcCtx *svchp;
58: OCISession *usrhp;
59: };
60:
61: struct OracleSQL_query_lobs {
62: struct return_rows {
63: struct return_row {
64: OCILobLocator *locator; ub4 len;
65: int ind;
66: ub2 rcode;
67: } *row;
68: int count;
69: };
70: struct cbf_context_struct {
71: SQL_Driver_services *services;
72: OracleSQL_connection_struct *cs;
73: return_rows *rows;
74: };
75: struct Item {
76: const char *name_ptr; size_t name_size;
77: char *data_ptr; size_t data_size;
78: OCILobLocator *locator;
79: OCIBind *bind;
80: return_rows rows;
81: } items[MAX_IN_LOBS];
82: int count;
83: };
84:
85:
86: // forwards
87: void check(
88: SQL_Driver_services& services, OracleSQL_connection_struct &cs,
89: const char *step, sword status);
90: static sb4 cbf_no_data(
91: dvoid *ctxp,
92: OCIBind *bindp,
93: ub4 iter, ub4 index,
94: dvoid **bufpp,
95: ub4 *alenpp,
96: ub1 *piecep,
97: dvoid **indpp);
98: static sb4 cbf_get_data(dvoid *ctxp,
99: OCIBind *bindp,
100: ub4 iter, ub4 index,
101: dvoid **bufpp,
102: ub4 **alenp,
103: ub1 *piecep,
104: dvoid **indpp,
105: ub2 **rcodepp);
1.3 ! parser 106: void tolower(char *out, const char *in, size_t size);
1.1 parser 107:
108: /**
109: OracleSQL server driver
110: */
111: class OracleSQL_Driver : public SQL_Driver {
112: public:
113:
114: OracleSQL_Driver() : SQL_Driver() {
115: }
116:
117: /// get api version
118: int api_version() { return SQL_DRIVER_API_VERSION; }
119: /// initialize driver by loading sql dynamic link library
120: const char *initialize(const char *dlopen_file_spec) {
121: /* const char *error=dlopen_file_spec?
122: dlink(dlopen_file_spec):"client library column is empty";
123: if(!error) {*/
1.3 ! parser 124: OCIInitialize((ub4)OCI_THREADED | OCI_OBJECT, (dvoid *)0,
1.1 parser 125: (dvoid * (*)(void *, unsigned int)) 0,
126: (dvoid * (*)(void*, void*, unsigned int)) 0,
127: (void (*)(void*, void*)) 0 );
128: /*}
129:
130: return error;*/
131: return 0;
132: }
133:
134: /** connect
135: @param used_only_in_connect_url
1.2 parser 136: format: @b user:pass@service
1.1 parser 137: */
138: void connect(
139: char *used_only_in_connect_url,
140: SQL_Driver_services& services,
141: void **connection ///< output: OracleSQL_connection_struct *
142: ) {
143: char *user=used_only_in_connect_url;
144: char *service=lsplit(user, '@');
145: char *pwd=lsplit(user, ':');
146:
147: OracleSQL_connection_struct &cs=
148: *(OracleSQL_connection_struct *)services.calloc(sizeof(OracleSQL_connection_struct));
149:
1.2 parser 150: if(!(user && pwd && service))
1.3 ! parser 151: services._throw("mailformed connect part, must be 'user:pass@service'");
1.2 parser 152:
1.1 parser 153: if(setjmp(cs.mark))
154: services._throw(cs.error);
155:
1.3 ! parser 156: // Allocate and initialize OCIError handle, attempt #1
! 157: /*
! 158: grabbed from sample
! 159: /server.804/a58234/oci_func.htm#446192
! 160: but doc
! 161: /server.804/a58234/oci_func.htm#446100
! 162: doesnt have this param listed as allowed
! 163: 8.1.6 client library barks as OCI_INVALID_HANDLE
! 164: and debugging revealed that OCI_HTYPE_ENV param value is invalid
! 165: later in doc
! 166: /server.804/a58234/oci_func.htm#446192
! 167: on OCIEnvInit thay say
! 168: "No changes are done to an already initialized handle"
! 169: think, this is some sort of backward compatibility wonder.
! 170: leaving as it is, and without check()
! 171: */
! 172: OCIHandleAlloc((dvoid *) NULL, (dvoid **) &cs.envhp, (ub4)OCI_HTYPE_ENV, 0, 0);
! 173: // Initialize an environment handle, attempt #2
! 174: check(services, cs, "OCIEnvInit", OCIEnvInit(
! 175: &cs.envhp, (ub4)OCI_DEFAULT, 0, 0));
! 176: // Allocate and initialize OCIError handle
1.2 parser 177: check(services, cs, "OCIHandleAlloc errhp", OCIHandleAlloc(
1.3 ! parser 178: (dvoid *)cs.envhp, (dvoid **) &cs.errhp, (ub4)OCI_HTYPE_ERROR, 0, 0));
! 179: // Allocate and initialize OCIServer handle
1.2 parser 180: check(services, cs, "OCIHandleAlloc srvhp", OCIHandleAlloc(
1.3 ! parser 181: (dvoid *)cs.envhp, (dvoid **) &cs.srvhp, (ub4)OCI_HTYPE_SERVER, 0, 0));
! 182: // Attach to a 'service'; initialize server context handle
1.2 parser 183: check(services, cs, "OCIServerAttach", OCIServerAttach(
1.3 ! parser 184: cs.srvhp, cs.errhp, (text *) service, (sb4) strlen(service), (ub4)OCI_DEFAULT));
! 185: // Allocate and initialize OCISvcCtx handle
1.2 parser 186: check(services, cs, "OCIHandleAlloc svchp", OCIHandleAlloc(
1.3 ! parser 187: (dvoid *)cs.envhp, (dvoid **) &cs.svchp, (ub4)OCI_HTYPE_SVCCTX, 0, 0));
! 188: // set attribute server context in the service context
1.2 parser 189: check(services, cs, "OCIAttrSet server-service", OCIAttrSet(
1.3 ! parser 190: (dvoid *)cs.svchp, (ub4)OCI_HTYPE_SVCCTX,
! 191: (dvoid *)cs.srvhp, (ub4) 0,
! 192: (ub4)OCI_ATTR_SERVER, (OCIError *)cs.errhp));
! 193: // allocate a user context handle
1.2 parser 194: check(services, cs, "OCIHandleAlloc usrhp", OCIHandleAlloc(
1.3 ! parser 195: (dvoid *)cs.envhp, (dvoid **)&cs.usrhp, (ub4)OCI_HTYPE_SESSION, 0, 0));
! 196: // set 'user' name
1.2 parser 197: check(services, cs, "OCIAttrSet user-session", OCIAttrSet(
198: (dvoid *)cs.usrhp, (ub4)OCI_HTYPE_SESSION,
1.1 parser 199: (dvoid *)user, (ub4)strlen(user),
1.3 ! parser 200: OCI_ATTR_USERNAME, cs.errhp));
! 201: // set 'pwd' password
1.2 parser 202: check(services, cs, "OCIAttrSet pwd-session", OCIAttrSet(
203: (dvoid *)cs.usrhp, (ub4)OCI_HTYPE_SESSION,
1.1 parser 204: (dvoid *)pwd, (ub4)strlen(pwd),
1.2 parser 205: OCI_ATTR_PASSWORD, cs.errhp));
1.3 ! parser 206: // Authenticate a user
! 207: check(services, cs, "OCISessionBegin", OCISessionBegin(
! 208: cs.svchp, cs.errhp, cs.usrhp,
1.1 parser 209: OCI_CRED_RDBMS, OCI_DEFAULT));
1.3 ! parser 210: // remember connection in session
1.2 parser 211: check(services, cs, "OCIAttrSet service-session", OCIAttrSet(
212: (dvoid *)cs.svchp, (ub4)OCI_HTYPE_SVCCTX,
1.1 parser 213: (dvoid *)cs.usrhp, (ub4)0,
1.2 parser 214: OCI_ATTR_SESSION, cs.errhp));
1.1 parser 215:
1.3 ! parser 216: // return created connection
1.1 parser 217: *(OracleSQL_connection_struct **)connection=&cs;
218: }
219: void disconnect(SQL_Driver_services& services, void *connection) {
220: OracleSQL_connection_struct &cs=*(OracleSQL_connection_struct *)connection;
221:
1.3 ! parser 222: // Terminate a user session
! 223: check(services, cs, "OCISessionEnd", OCISessionEnd(
! 224: cs.svchp, cs.errhp, cs.usrhp, (ub4)OCI_DEFAULT));
! 225: // Detach from a server; uninitialize server context handle
! 226: check(services, cs, "OCIServerDetach", OCIServerDetach(
! 227: cs.srvhp, cs.errhp, (ub4)OCI_DEFAULT));
! 228: // Free a previously allocated handle
! 229: check(services, cs, "OCIHandleFree srvhp", OCIHandleFree(
! 230: (dvoid *)cs.srvhp, (ub4)OCI_HTYPE_SERVER));
! 231: check(services, cs, "OCIHandleFree svchp", OCIHandleFree(
! 232: (dvoid *)cs.svchp, (ub4)OCI_HTYPE_SVCCTX));
! 233: check(services, cs, "OCIHandleFree errhp", OCIHandleFree(
! 234: (dvoid *)cs.errhp, (ub4)OCI_HTYPE_ERROR)); cs.errhp=0;
! 235: check(services, cs, "OCIHandleFree envhp", OCIHandleFree(
! 236: (dvoid *)cs.envhp, (ub4)OCI_HTYPE_ENV));
1.1 parser 237: }
238: void commit(SQL_Driver_services& services, void *connection) {
239: OracleSQL_connection_struct &cs=*(OracleSQL_connection_struct *)connection;
240:
241: check(services, cs, "commit", OCITransCommit(cs.svchp, cs.errhp, 0));
242: }
243: void rollback(SQL_Driver_services& services, void *connection) {
244: OracleSQL_connection_struct &cs=*(OracleSQL_connection_struct *)connection;
245:
246: check(services, cs, "rollback", OCITransRollback(cs.svchp, cs.errhp, 0));
247: }
248:
249: bool ping(SQL_Driver_services&, void *connection) {
250: return false;
251: }
252:
253: unsigned int quote(
254: SQL_Driver_services&, void *,
255: char *to, const char *from, unsigned int length) {
256: /*
257: it's already UNTAINT_TIMES_BIGGER
258: */
259: unsigned int result=length;
260: while(length--) {
261: switch(*from) {
262: case '\'': // "'" -> "''"
263: *to++='\'';
264: break;
265: case '\\': // "\" -> "\\"
266: *to++='\'';
267: break;
268: }
269: *to++=*from++;
270: }
271: return result;
272: }
273: void query(
274: SQL_Driver_services& services, void *connection,
275: const char *astatement, unsigned long offset, unsigned long limit,
276: SQL_Driver_query_event_handlers& handlers) {
277: // _asm int 3;
278: OracleSQL_connection_struct &cs=*(OracleSQL_connection_struct *)connection;
279: OracleSQL_query_lobs lobs={{0}, 0};
280: OCIStmt *stmthp=0;
281:
282: bool failed=false;
283: if(setjmp(cs.mark)) {
284: failed=true;
285: goto cleanup;
286: }
287: {
288: const char *statement=preprocess_statement(services, cs,
289: astatement, lobs);
290:
291: check(services, cs, "OCIHandleAlloc STMT", OCIHandleAlloc(
1.3 ! parser 292: (dvoid *)cs.envhp, (dvoid **) &stmthp, (ub4)OCI_HTYPE_STMT, 0, 0));
1.1 parser 293: check(services, cs, "syntax",
294: OCIStmtPrepare(stmthp, cs.errhp, (unsigned char *)statement,
295: (ub4) strlen((char *) statement),
1.3 ! parser 296: (ub4)OCI_NTV_SYNTAX, (ub4)OCI_DEFAULT));
1.1 parser 297: {
298: for(int i=0; i<lobs.count; i++) {
299: check(services, cs, "alloc output var desc", OCIDescriptorAlloc(
1.3 ! parser 300: (dvoid *)cs.envhp, (dvoid **)&lobs.items[i].locator, (ub4)OCI_DTYPE_LOB, 0, 0));
1.1 parser 301:
302: check(services, cs, "bind output", OCIBindByPos(stmthp,
303: &lobs.items[i].bind, cs.errhp,
304: (ub4) 1+i,
305: (dvoid *)&lobs.items[i].locator,
306: (sword)sizeof (lobs.items[i].locator), SQLT_CLOB, (dvoid *) 0,
307: (ub2 *) 0, (ub2 *) 0, (ub4) 0, (ub4 *) 0, OCI_DATA_AT_EXEC));
308:
309: lobs.items[i].rows.count=0;
310: OracleSQL_query_lobs::cbf_context_struct cbf_context={
311: &services, &cs, &lobs.items[i].rows};
312: check(services, cs, "bind dynamic", OCIBindDynamic(
313: lobs.items[i].bind, cs.errhp,
314: (dvoid *) &cbf_context, cbf_no_data,
315: (dvoid *) &cbf_context, cbf_get_data));
316: }
317: }
318:
319: execute_prepared(services, cs,
320: statement, stmthp, lobs,
321: offset, limit, handlers);
322: }
323: cleanup: // no check call after this point!
324: {
325: for(int i=0; i<lobs.count; i++) {
326: /* free var locator */
1.3 ! parser 327: (void)OCIDescriptorFree((dvoid *) lobs.items[i].locator, (ub4)OCI_DTYPE_LOB);
1.1 parser 328:
329: OracleSQL_query_lobs::return_rows *rows=&lobs.items[i].rows;
330: for(int r=0; r<rows->count; r++) {
331: /* free var locator */
1.3 ! parser 332: (void)OCIDescriptorFree((dvoid *) rows->row[r].locator, (ub4)OCI_DTYPE_LOB);
1.1 parser 333: }
334: }
335: }
336: if(stmthp)
1.3 ! parser 337: (void)OCIHandleFree((dvoid *) stmthp, (ub4)OCI_HTYPE_STMT);
1.1 parser 338:
339: if(failed)
340: services._throw(cs.error);
341: }
342:
343: private: // private funcs
344:
345: const char *preprocess_statement(SQL_Driver_services& services, OracleSQL_connection_struct &cs,
346: const char *astatement, OracleSQL_query_lobs &lobs) {
347: size_t statement_size=strlen(astatement);
348:
349: char *result=(char *)services.malloc(statement_size
350: +MAX_STRING // in case of short 'strings'
351: +11/* returning */+6/* into */+(MAX_LOB_NAME_LENGTH+2/*:, */)*2/*ret into*/*MAX_IN_LOBS
352: +1);
353: const char *o=astatement;
354:
355: // /**xxx**/'literal' -> EMPTY_CLOB_FUNC_CALL
356: char *n=result;
357: while(*o) {
358: if(
359: o[0]=='/' &&
360: o[1]=='*' &&
361: o[2]=='*') { // name start
362: o+=3;
363: const char *name_begin=o;
364: while(*o)
365: if(
366: o[0]=='*' &&
367: o[1]=='*' &&
368: o[2]=='/' &&
369: o[3]=='\'') { // name end
370: const char *name_end=o;
371: o+=4;
372: OracleSQL_query_lobs::Item &item=lobs.items[lobs.count++];
373: item.name_ptr=name_begin; item.name_size=name_end-name_begin;
374: item.data_ptr=(char *)services.malloc(statement_size/*max*/); item.data_size=0;
375:
376: const char *start=o;
377: bool escaped=false;
378: while(*o && !(o[0]=='\'' && o[1]!='\'' && !escaped)) {
379: escaped=*o=='\\' || (o[0]=='\'' && o[1]=='\'');
380: if(escaped) {
381: // write pending, skip "\" or "'"
382: if(size_t size=o-start) {
383: memcpy(item.data_ptr+item.data_size, start, size);
384: item.data_size+=size;
385: }
386: start=++o;
387: } else
388: o++;
389: }
390: if(size_t size=o-start) {
391: memcpy(item.data_ptr+item.data_size, start, size);
392: item.data_size+=size;
393: }
394: if(*o)
395: o++; // skip "'"
396:
397: n+=sprintf(n, EMPTY_CLOB_FUNC_CALL);
398: break;
399: } else
400: o++; // /**skip**/'xxx'
401: } else
402: *n++=*o++;
403: }
404: *n=0;
405:
406: if(lobs.count) {
407: int i;
408: n+=sprintf(n, " returning ");
409: for(i=0; i<lobs.count; i++) {
410: if(i)
411: *n++=',';
412: n+=sprintf(n, "%*s", lobs.items[i].name_size, lobs.items[i].name_ptr);
413: /*memcpy(n, lobs.items[i].name_ptr, lobs.items[i].name_size);
414: n+=lobs.items[i].name_size;*/
415: }
416: n+=sprintf(n, " into ");
417: for(i=0; i<lobs.count; i++) {
418: if(i)
419: *n++='x';
420: n+=sprintf(n, ":%*s", lobs.items[i].name_size, lobs.items[i].name_ptr);
421: /**n++=':';
422: memcpy(n, lobs.items[i].name_ptr, lobs.items[i].name_size);
423: n+=lobs.items[i].name_size;*/
424: }
425: }
426:
427: return result;
428: }
429:
430: void execute_prepared(
431: SQL_Driver_services& services, OracleSQL_connection_struct &cs,
432: const char *statement, OCIStmt *stmthp, OracleSQL_query_lobs &lobs,
433: unsigned long offset, unsigned long limit,
434: SQL_Driver_query_event_handlers& handlers) {
435:
436: ub2 stmt_type=-1; // UNKNOWN // OCI_STMT_SELECT;//
437: /*
438: //gpfs on sun. paf 000818
439: //Zanyway, this is needed before.
440: check(services, cs, "get stmt type", OCIAttrGet(
441: (dvoid *)stmthp, (ub4)OCI_HTYPE_STMT, (ub1 *)&stmt_type,
442: (ub4 *) 0, OCI_ATTR_STMT_TYPE, cs.errhp));
443: */
444: if(strncasecmp(statement, "select", 6)==0)
445: stmt_type=OCI_STMT_SELECT;
1.3 ! parser 446: else if(strncasecmp(statement, "insert", 6)==0)
1.1 parser 447: stmt_type=OCI_STMT_INSERT;
1.3 ! parser 448: else if(strncasecmp(statement, "update", 6)==0)
1.1 parser 449: stmt_type=OCI_STMT_UPDATE;
450:
451: int status=OCIStmtExecute(cs.svchp, stmthp, cs.errhp,
452: (ub4) stmt_type==OCI_STMT_SELECT?0:1, (ub4) 0,
453: (OCISnapshot *) NULL,
1.3 ! parser 454: (OCISnapshot *) NULL, (ub4)OCI_DEFAULT);
1.1 parser 455:
1.3 ! parser 456: if(status!=OCI_NO_DATA)
1.1 parser 457: check(services, cs, "execute", status);
458:
459: {
460: for(int i=0; i<lobs.count; i++)
461: if(ub4 bytes_to_write=lobs.items[i].data_size) {
462: OracleSQL_query_lobs::return_rows *rows=&lobs.items[i].rows;
463: for(int r=0; r<rows->count; r++) {
464: OCILobLocator *locator=rows->row[r].locator;
465: check(services, cs, "lobwrite", OCILobWrite (
466: cs.svchp, cs.errhp,
467: locator, &bytes_to_write, 1,
468: (dvoid *) lobs.items[i].data_ptr, (ub4)bytes_to_write, OCI_ONE_PIECE,
469: (dvoid *)0, 0, (ub2) 0,
470: (ub1) SQLCS_IMPLICIT));
471: }
472: }
473: }
474:
475: switch(stmt_type) {
476: case OCI_STMT_SELECT:
477: fetch_table(services, cs,
1.3 ! parser 478: stmthp, offset, limit,
1.1 parser 479: handlers);
480: break;
481: case OCI_STMT_INSERT:
482: break;
483: default:/*
484: case OCI_STMT_UPDATE:
485: case OCI_STMT_DELETE:
486: case OCI_STMT_CREATE:
487: case OCI_STMT_DROP:
488: case OCI_STMT_ALTER:
489: case OCI_STMT_BEGIN:
490: case OCI_STMT_DECLARE:*/
491: break;
492: }
493: }
494:
495: void fetch_table(SQL_Driver_services& services, OracleSQL_connection_struct &cs,
1.3 ! parser 496: OCIStmt *stmthp, unsigned long offset, unsigned long limit,
1.1 parser 497: SQL_Driver_query_event_handlers& handlers) {
1.3 ! parser 498: // todo another set/longjump
1.1 parser 499: OCIParam *mypard;
500: ub2 dtype;
501: text *col_name;
502:
503: struct {
504: ub2 type;
505: char *str;
506: OCILobLocator *var;
507: OCIDefine *def;
508: sb2 indicator;
509: } cols[MAX_COLS];
510: int column_count;
511:
512: for(column_count=0; column_count<MAX_COLS; column_count++) {
513: /* get next descriptor, if there is one */
1.3 ! parser 514: if(OCIParamGet(stmthp, OCI_HTYPE_STMT, cs.errhp, (void **)&mypard,
! 515: (ub4) 1+column_count)!=OCI_SUCCESS)
1.1 parser 516: break;
517:
518: /* Retrieve the data type attribute */
519: check(services, cs, "get type", OCIAttrGet(
1.3 ! parser 520: (dvoid*) mypard, (ub4)OCI_DTYPE_PARAM,
! 521: (dvoid*) &dtype, (ub4 *) 0, (ub4)OCI_ATTR_DATA_TYPE,
! 522: (OCIError *)cs.errhp));
1.1 parser 523:
524: /* Retrieve the column name attribute */
525: ub4 col_name_len;
526: check(services, cs, "get name", OCIAttrGet(
1.3 ! parser 527: (dvoid*) mypard, (ub4)OCI_DTYPE_PARAM,
! 528: (dvoid**) &col_name, (ub4 *) &col_name_len, (ub4)OCI_ATTR_NAME,
! 529: (OCIError *)cs.errhp));
1.1 parser 530:
531: {
532: size_t size=(size_t)col_name_len;
1.3 ! parser 533: char *ptr=(char *)services.malloc(size);
! 534: tolower(ptr, (char *)col_name, size);
1.1 parser 535: handlers.add_column(ptr, size);
536: }
537:
538: ub2 coerce_type=dtype;
539: sb4 size=0;
540: void *ptr;
541:
542: switch(dtype) {
543: case SQLT_CLOB:
544: {
545: check(services, cs, "alloc output var desc", OCIDescriptorAlloc(
1.3 ! parser 546: (dvoid *)cs.envhp, (dvoid **)(ptr=&cols[column_count].var),
! 547: (ub4)OCI_DTYPE_LOB,
! 548: 0, (dvoid **) 0));
1.1 parser 549:
550: size=0;
551: break;
552: }
553: default:
554: coerce_type=SQLT_STR;
555: ptr=cols[column_count].str=(char *)services.malloc(MAX_OUT_STRING_LENGTH+1);
556: size=MAX_OUT_STRING_LENGTH;
557: break;
558: }
559:
560: cols[column_count].type=coerce_type;
561:
1.3 ! parser 562: check(services, cs, "OCIDefineByPos", OCIDefineByPos(
! 563: stmthp, &cols[column_count].def, cs.errhp,
! 564: 1+column_count, (ub1 *) ptr, size,
! 565: coerce_type, (dvoid *) &cols[column_count].indicator,
! 566: (ub2 *) 0, (ub2 *) 0, OCI_DEFAULT));
1.1 parser 567: }
568:
569: handlers.before_rows();
570:
1.3 ! parser 571: for (unsigned long row=0;!limit||row<limit+offset;row++) {
! 572: sword status=OCIStmtFetch(stmthp, cs.errhp, (ub4) 1, (ub4)OCI_FETCH_NEXT,
! 573: (ub4)OCI_DEFAULT);
! 574: if(status!=OCI_SUCCESS) {
! 575: if(status!=OCI_NO_DATA)
1.1 parser 576: check(services, cs, "fetch", status);
577: break;
578: }
579: if(row>=offset) {
580: handlers.add_row();
1.3 ! parser 581: for(int i=0; i<column_count; i++) {
1.1 parser 582: size_t size=0;
583: void *ptr=0;
584: if(cols[i].indicator==0) switch(cols[i].type) {
585: case SQLT_CLOB:
586: {
1.2 parser 587: ub4 amtp=4096000000UL;
588: ub4 offset=1;
589: ub4 loblen=0;
1.1 parser 590: OCILobLocator *var=(OCILobLocator *)cols[i].var;
1.3 ! parser 591: (void)OCILobGetLength(cs.svchp, cs.errhp, var, &loblen);
1.1 parser 592: if(loblen) {
593: size=(size_t)loblen;
594: ptr=services.malloc(size);
595: check(services, cs, "lobread", OCILobRead(cs.svchp, cs.errhp,
596: var, &amtp, offset, (dvoid *) ptr,
597: loblen, (dvoid *)0,
598: 0,
599: (ub2) 0, (ub1) SQLCS_IMPLICIT));
600: }
601: break;
602: }
603: default:
604: const char *str=cols[i].str;
605: size=strlen(str);
606: ptr=services.malloc(size);
607: memcpy(ptr, str, size);
608: break;
609: }
610: handlers.add_row_cell(ptr, size);
611: }
612: }
613: }
614:
1.3 ! parser 615: for(int i=0; i<column_count; i++) {
1.1 parser 616: switch(cols[i].type) {
617: case SQLT_CLOB:
618: /* free var locator */
1.3 ! parser 619: check(services, cs, "freel lob descriptor", OCIDescriptorFree(
! 620: (dvoid *) cols[i].var, (ub4)OCI_DTYPE_LOB));
1.1 parser 621: break;
622: default:
623: break;
624: }
625: }
626: }
627:
628: private: // conn client library funcs
629:
630: // typedef void (*t_PQfinish)(PGconn *conn); t_PQfinish PQfinish;
631:
632: private: // conn client library funcs linking
633:
634: const char *dlink(const char *dlopen_file_spec) {
635: lt_dlhandle handle=lt_dlopen(dlopen_file_spec);
636: if(!handle)
637: return "can not open the dynamic link module";
638:
639: #define DSLINK(name, action) \
640: name=(t_##name)lt_dlsym(handle, #name); \
641: if(!name) \
642: action;
643:
644: #define DLINK(name) DSLINK(name, return "function " #name " was not found")
645:
646: // DLINK(PQfinish);
647:
648: return 0;
649: }
650:
651: };
652:
653: void check(
654: SQL_Driver_services& services, OracleSQL_connection_struct &cs,
655: const char *step, sword status) {
656:
1.2 parser 657: char reason[MAX_STRING/2];
1.1 parser 658: const char *msg=0;
659:
660: switch (status) {
661: case OCI_SUCCESS:
662: // hurrah
663: return;
664: /*
665: case OCI_ERROR:
666: {
1.2 parser 667: sb4 errcode=0;
1.3 ! parser 668: (void)OCIErrorGet((dvoid *)cs.errhp, (ub4) 1, (text *) NULL, &errcode,
1.1 parser 669: reason, (ub4) sizeof(reason), OCI_HTYPE_ERROR);
670: reason[sizeof(reason)-1]=0;
671: msg=reason;
672: break;
673: }
674: case OCI_SUCCESS_WITH_INFO:
675: msg="OCI_SUCCESS_WITH_INFO"; break;
676: case OCI_NEED_DATA:
677: msg="OCI_NEED_DATA"; break;
678: case OCI_NO_DATA:
679: msg="OCI_NODATA"; break;
680: case OCI_INVALID_HANDLE:
681: msg="OCI_INVALID_HANDLE"; break;
682: case OCI_STILL_EXECUTING:
683: msg="OCI_STILL_EXECUTE"; break;
684: case OCI_CONTINUE:
685: msg="OCI_CONTINUE"; break;
686: */
687: }
1.2 parser 688: if(!msg) {
689: sb4 errcode=0;
690: if(OCIErrorGet((dvoid *)cs.errhp, (ub4) 1, (text *) NULL, &errcode,
691: (text *) reason, (ub4) sizeof(reason), OCI_HTYPE_ERROR)==OCI_SUCCESS)
692: msg=reason;
693: else
1.3 ! parser 694: msg="unknown error";
1.2 parser 695: }
1.1 parser 696:
1.3 ! parser 697: snprintf(cs.error, sizeof(cs.error), "%s (%s, %d)",
! 698: msg, step, (int)status);
1.1 parser 699: longjmp(cs.mark, 1);
700: }
701:
702:
703: /* ----------------------------------------------------------------- */
704: /* Intbind callback that does not do any data input. */
705: /* ----------------------------------------------------------------- */
706: static sb4 cbf_no_data(
707: dvoid *ctxp,
708: OCIBind *bindp,
709: ub4 iter, ub4 index,
710: dvoid **bufpp,
711: ub4 *alenpp,
712: ub1 *piecep,
713: dvoid **indpp) {
1.2 parser 714: *bufpp=(dvoid *) 0;
715: *alenpp=0;
716: static sb2 null_ind=-1;
717: *indpp=(dvoid *) &null_ind;
718: *piecep=OCI_ONE_PIECE;
1.1 parser 719:
720: return OCI_CONTINUE;
721: }
722:
723: /* ----------------------------------------------------------------- */
724: /* Outbind callback for returning data. */
725: /* ----------------------------------------------------------------- */
726: static sb4 cbf_get_data(dvoid *ctxp,
727: OCIBind *bindp,
728: ub4 iter, ub4 index,
729: dvoid **bufpp,
730: ub4 **alenp,
731: ub1 *piecep,
732: dvoid **indpp,
733: ub2 **rcodepp) {
734: OracleSQL_query_lobs::cbf_context_struct &context=
735: *(OracleSQL_query_lobs::cbf_context_struct *)ctxp;
736:
1.3 ! parser 737: if(index==0) {
1.2 parser 738: static ub4 rows=0;
1.3 ! parser 739: (void)OCIAttrGet((CONST dvoid *) bindp, OCI_HTYPE_BIND, (dvoid *)&rows,
1.1 parser 740: (ub4 *)sizeof(ub2), OCI_ATTR_ROWS_RETURNED, context.cs->errhp);
1.2 parser 741: context.rows->count=(ub2)rows;
742: context.rows->row=(OracleSQL_query_lobs::return_rows::return_row *)
1.1 parser 743: context.services->malloc(sizeof(OracleSQL_query_lobs::return_rows::return_row)*rows);
744: }
745:
746: OracleSQL_query_lobs::return_rows::return_row &var=context.rows->row[index];
747:
748: check(*context.services, *context.cs, "alloc output var desc dynamic", OCIDescriptorAlloc(
749: (dvoid *) context.cs->envhp, (dvoid **)&var.locator,
1.3 ! parser 750: (ub4)OCI_DTYPE_LOB,
! 751: 0, (dvoid **) 0));
1.1 parser 752:
1.2 parser 753: *bufpp=var.locator;
754: *alenp=&var.len;
755: *indpp=(dvoid *) &var.ind;
756: *piecep=OCI_ONE_PIECE;
757: *rcodepp=&var.rcode;
1.1 parser 758:
759: return OCI_CONTINUE;
760: }
761:
1.3 ! parser 762: void tolower(char *out, const char *in, size_t size) {
! 763: while(size--)
! 764: *out++=tolower(*in++);
! 765: }
1.1 parser 766:
767: extern "C" SQL_Driver *create() {
768: return new OracleSQL_Driver();
769: }