![[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] / parser3 / src / sql / oracle|
Return to parser3oracle.C CVS log | Up to [parser3project] / parser3 / src / 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, tested with Oracle 7.x.x
9: */
1.4 ! parser 10: static const char *RCSId="$Id: parser3oracle.C,v 1.3 2001/08/23 10:37:46 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.4 ! parser 124: OCIInitialize((ub4)OCI_THREADED, (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
1.4 ! parser 174: check(services, cs, "EnvInit", OCIEnvInit(
1.3 parser 175: &cs.envhp, (ub4)OCI_DEFAULT, 0, 0));
176: // Allocate and initialize OCIError handle
1.4 ! parser 177: check(services, cs, "HandleAlloc 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.4 ! parser 180: check(services, cs, "HandleAlloc 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.4 ! parser 183: check(services, cs, "ServerAttach", OCIServerAttach(
1.3 parser 184: cs.srvhp, cs.errhp, (text *) service, (sb4) strlen(service), (ub4)OCI_DEFAULT));
185: // Allocate and initialize OCISvcCtx handle
1.4 ! parser 186: check(services, cs, "HandleAlloc 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.4 ! parser 189: check(services, cs, "AttrSet 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.4 ! parser 194: check(services, cs, "HandleAlloc usrhp", OCIHandleAlloc(
1.3 parser 195: (dvoid *)cs.envhp, (dvoid **)&cs.usrhp, (ub4)OCI_HTYPE_SESSION, 0, 0));
196: // set 'user' name
1.4 ! parser 197: check(services, cs, "AttrSet user-session", OCIAttrSet(
1.2 parser 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.4 ! parser 202: check(services, cs, "AttrSet pwd-session", OCIAttrSet(
1.2 parser 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
1.4 ! parser 207: check(services, cs, "SessionBegin", OCISessionBegin(
1.3 parser 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.4 ! parser 211: check(services, cs, "AttrSet service-session", OCIAttrSet(
1.2 parser 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;
1.4 ! parser 221: if(setjmp(cs.mark))
! 222: services._throw(cs.error);
1.1 parser 223:
1.3 parser 224: // Terminate a user session
1.4 ! parser 225: check(services, cs, "SessionEnd", OCISessionEnd(
1.3 parser 226: cs.svchp, cs.errhp, cs.usrhp, (ub4)OCI_DEFAULT));
227: // Detach from a server; uninitialize server context handle
1.4 ! parser 228: check(services, cs, "ServerDetach", OCIServerDetach(
1.3 parser 229: cs.srvhp, cs.errhp, (ub4)OCI_DEFAULT));
230: // Free a previously allocated handle
1.4 ! parser 231: check(services, cs, "HandleFree srvhp", OCIHandleFree(
1.3 parser 232: (dvoid *)cs.srvhp, (ub4)OCI_HTYPE_SERVER));
1.4 ! parser 233: check(services, cs, "HandleFree svchp", OCIHandleFree(
1.3 parser 234: (dvoid *)cs.svchp, (ub4)OCI_HTYPE_SVCCTX));
1.4 ! parser 235: check(services, cs, "HandleFree errhp", OCIHandleFree(
1.3 parser 236: (dvoid *)cs.errhp, (ub4)OCI_HTYPE_ERROR)); cs.errhp=0;
1.4 ! parser 237: check(services, cs, "HandleFree envhp", OCIHandleFree(
1.3 parser 238: (dvoid *)cs.envhp, (ub4)OCI_HTYPE_ENV));
1.1 parser 239: }
240: void commit(SQL_Driver_services& services, void *connection) {
241: OracleSQL_connection_struct &cs=*(OracleSQL_connection_struct *)connection;
1.4 ! parser 242: if(setjmp(cs.mark))
! 243: services._throw(cs.error);
1.1 parser 244:
245: check(services, cs, "commit", OCITransCommit(cs.svchp, cs.errhp, 0));
246: }
247: void rollback(SQL_Driver_services& services, void *connection) {
248: OracleSQL_connection_struct &cs=*(OracleSQL_connection_struct *)connection;
1.4 ! parser 249: if(setjmp(cs.mark))
! 250: services._throw(cs.error);
1.1 parser 251:
252: check(services, cs, "rollback", OCITransRollback(cs.svchp, cs.errhp, 0));
253: }
254:
255: bool ping(SQL_Driver_services&, void *connection) {
1.4 ! parser 256: // maybe OCIServerVersion?
! 257: return true;
1.1 parser 258: }
259:
260: unsigned int quote(
261: SQL_Driver_services&, void *,
262: char *to, const char *from, unsigned int length) {
263: /*
264: it's already UNTAINT_TIMES_BIGGER
265: */
266: unsigned int result=length;
267: while(length--) {
268: switch(*from) {
269: case '\'': // "'" -> "''"
270: *to++='\'';
271: break;
272: case '\\': // "\" -> "\\"
273: *to++='\'';
274: break;
275: }
276: *to++=*from++;
277: }
278: return result;
279: }
280: void query(
281: SQL_Driver_services& services, void *connection,
282: const char *astatement, unsigned long offset, unsigned long limit,
283: SQL_Driver_query_event_handlers& handlers) {
284: // _asm int 3;
285: OracleSQL_connection_struct &cs=*(OracleSQL_connection_struct *)connection;
286: OracleSQL_query_lobs lobs={{0}, 0};
287: OCIStmt *stmthp=0;
288:
289: bool failed=false;
290: if(setjmp(cs.mark)) {
291: failed=true;
292: goto cleanup;
293: }
294: {
295: const char *statement=preprocess_statement(services, cs,
296: astatement, lobs);
297:
1.4 ! parser 298: check(services, cs, "HandleAlloc STMT", OCIHandleAlloc(
1.3 parser 299: (dvoid *)cs.envhp, (dvoid **) &stmthp, (ub4)OCI_HTYPE_STMT, 0, 0));
1.1 parser 300: check(services, cs, "syntax",
301: OCIStmtPrepare(stmthp, cs.errhp, (unsigned char *)statement,
302: (ub4) strlen((char *) statement),
1.3 parser 303: (ub4)OCI_NTV_SYNTAX, (ub4)OCI_DEFAULT));
1.1 parser 304: {
305: for(int i=0; i<lobs.count; i++) {
306: check(services, cs, "alloc output var desc", OCIDescriptorAlloc(
1.3 parser 307: (dvoid *)cs.envhp, (dvoid **)&lobs.items[i].locator, (ub4)OCI_DTYPE_LOB, 0, 0));
1.1 parser 308:
309: check(services, cs, "bind output", OCIBindByPos(stmthp,
310: &lobs.items[i].bind, cs.errhp,
311: (ub4) 1+i,
312: (dvoid *)&lobs.items[i].locator,
313: (sword)sizeof (lobs.items[i].locator), SQLT_CLOB, (dvoid *) 0,
314: (ub2 *) 0, (ub2 *) 0, (ub4) 0, (ub4 *) 0, OCI_DATA_AT_EXEC));
315:
316: lobs.items[i].rows.count=0;
317: OracleSQL_query_lobs::cbf_context_struct cbf_context={
318: &services, &cs, &lobs.items[i].rows};
319: check(services, cs, "bind dynamic", OCIBindDynamic(
320: lobs.items[i].bind, cs.errhp,
321: (dvoid *) &cbf_context, cbf_no_data,
322: (dvoid *) &cbf_context, cbf_get_data));
323: }
324: }
325:
326: execute_prepared(services, cs,
327: statement, stmthp, lobs,
328: offset, limit, handlers);
329: }
330: cleanup: // no check call after this point!
331: {
332: for(int i=0; i<lobs.count; i++) {
333: /* free var locator */
1.4 ! parser 334: OCIDescriptorFree((dvoid *) lobs.items[i].locator, (ub4)OCI_DTYPE_LOB);
1.1 parser 335:
336: OracleSQL_query_lobs::return_rows *rows=&lobs.items[i].rows;
337: for(int r=0; r<rows->count; r++) {
338: /* free var locator */
1.4 ! parser 339: OCIDescriptorFree((dvoid *) rows->row[r].locator, (ub4)OCI_DTYPE_LOB);
1.1 parser 340: }
341: }
342: }
343: if(stmthp)
1.4 ! parser 344: OCIHandleFree((dvoid *) stmthp, (ub4)OCI_HTYPE_STMT);
1.1 parser 345:
346: if(failed)
347: services._throw(cs.error);
348: }
349:
350: private: // private funcs
351:
352: const char *preprocess_statement(SQL_Driver_services& services, OracleSQL_connection_struct &cs,
353: const char *astatement, OracleSQL_query_lobs &lobs) {
354: size_t statement_size=strlen(astatement);
355:
356: char *result=(char *)services.malloc(statement_size
357: +MAX_STRING // in case of short 'strings'
358: +11/* returning */+6/* into */+(MAX_LOB_NAME_LENGTH+2/*:, */)*2/*ret into*/*MAX_IN_LOBS
359: +1);
360: const char *o=astatement;
361:
362: // /**xxx**/'literal' -> EMPTY_CLOB_FUNC_CALL
363: char *n=result;
364: while(*o) {
365: if(
366: o[0]=='/' &&
367: o[1]=='*' &&
368: o[2]=='*') { // name start
369: o+=3;
370: const char *name_begin=o;
371: while(*o)
372: if(
373: o[0]=='*' &&
374: o[1]=='*' &&
375: o[2]=='/' &&
376: o[3]=='\'') { // name end
377: const char *name_end=o;
378: o+=4;
379: OracleSQL_query_lobs::Item &item=lobs.items[lobs.count++];
380: item.name_ptr=name_begin; item.name_size=name_end-name_begin;
381: item.data_ptr=(char *)services.malloc(statement_size/*max*/); item.data_size=0;
382:
383: const char *start=o;
384: bool escaped=false;
385: while(*o && !(o[0]=='\'' && o[1]!='\'' && !escaped)) {
386: escaped=*o=='\\' || (o[0]=='\'' && o[1]=='\'');
387: if(escaped) {
388: // write pending, skip "\" or "'"
389: if(size_t size=o-start) {
390: memcpy(item.data_ptr+item.data_size, start, size);
391: item.data_size+=size;
392: }
393: start=++o;
394: } else
395: o++;
396: }
397: if(size_t size=o-start) {
398: memcpy(item.data_ptr+item.data_size, start, size);
399: item.data_size+=size;
400: }
401: if(*o)
402: o++; // skip "'"
403:
404: n+=sprintf(n, EMPTY_CLOB_FUNC_CALL);
405: break;
406: } else
407: o++; // /**skip**/'xxx'
408: } else
409: *n++=*o++;
410: }
411: *n=0;
412:
413: if(lobs.count) {
414: int i;
415: n+=sprintf(n, " returning ");
416: for(i=0; i<lobs.count; i++) {
417: if(i)
418: *n++=',';
419: n+=sprintf(n, "%*s", lobs.items[i].name_size, lobs.items[i].name_ptr);
420: /*memcpy(n, lobs.items[i].name_ptr, lobs.items[i].name_size);
421: n+=lobs.items[i].name_size;*/
422: }
423: n+=sprintf(n, " into ");
424: for(i=0; i<lobs.count; i++) {
425: if(i)
426: *n++='x';
427: n+=sprintf(n, ":%*s", lobs.items[i].name_size, lobs.items[i].name_ptr);
428: /**n++=':';
429: memcpy(n, lobs.items[i].name_ptr, lobs.items[i].name_size);
430: n+=lobs.items[i].name_size;*/
431: }
432: }
433:
434: return result;
435: }
436:
437: void execute_prepared(
438: SQL_Driver_services& services, OracleSQL_connection_struct &cs,
439: const char *statement, OCIStmt *stmthp, OracleSQL_query_lobs &lobs,
440: unsigned long offset, unsigned long limit,
441: SQL_Driver_query_event_handlers& handlers) {
442:
443: ub2 stmt_type=-1; // UNKNOWN // OCI_STMT_SELECT;//
444: /*
445: //gpfs on sun. paf 000818
446: //Zanyway, this is needed before.
447: check(services, cs, "get stmt type", OCIAttrGet(
448: (dvoid *)stmthp, (ub4)OCI_HTYPE_STMT, (ub1 *)&stmt_type,
449: (ub4 *) 0, OCI_ATTR_STMT_TYPE, cs.errhp));
450: */
451: if(strncasecmp(statement, "select", 6)==0)
452: stmt_type=OCI_STMT_SELECT;
1.3 parser 453: else if(strncasecmp(statement, "insert", 6)==0)
1.1 parser 454: stmt_type=OCI_STMT_INSERT;
1.3 parser 455: else if(strncasecmp(statement, "update", 6)==0)
1.1 parser 456: stmt_type=OCI_STMT_UPDATE;
457:
458: int status=OCIStmtExecute(cs.svchp, stmthp, cs.errhp,
459: (ub4) stmt_type==OCI_STMT_SELECT?0:1, (ub4) 0,
460: (OCISnapshot *) NULL,
1.3 parser 461: (OCISnapshot *) NULL, (ub4)OCI_DEFAULT);
1.1 parser 462:
1.3 parser 463: if(status!=OCI_NO_DATA)
1.1 parser 464: check(services, cs, "execute", status);
465:
466: {
467: for(int i=0; i<lobs.count; i++)
468: if(ub4 bytes_to_write=lobs.items[i].data_size) {
469: OracleSQL_query_lobs::return_rows *rows=&lobs.items[i].rows;
470: for(int r=0; r<rows->count; r++) {
471: OCILobLocator *locator=rows->row[r].locator;
472: check(services, cs, "lobwrite", OCILobWrite (
473: cs.svchp, cs.errhp,
474: locator, &bytes_to_write, 1,
475: (dvoid *) lobs.items[i].data_ptr, (ub4)bytes_to_write, OCI_ONE_PIECE,
476: (dvoid *)0, 0, (ub2) 0,
477: (ub1) SQLCS_IMPLICIT));
478: }
479: }
480: }
481:
482: switch(stmt_type) {
483: case OCI_STMT_SELECT:
484: fetch_table(services, cs,
1.3 parser 485: stmthp, offset, limit,
1.1 parser 486: handlers);
487: break;
488: case OCI_STMT_INSERT:
489: break;
490: default:/*
491: case OCI_STMT_UPDATE:
492: case OCI_STMT_DELETE:
493: case OCI_STMT_CREATE:
494: case OCI_STMT_DROP:
495: case OCI_STMT_ALTER:
496: case OCI_STMT_BEGIN:
497: case OCI_STMT_DECLARE:*/
498: break;
499: }
500: }
501:
502: void fetch_table(SQL_Driver_services& services, OracleSQL_connection_struct &cs,
1.3 parser 503: OCIStmt *stmthp, unsigned long offset, unsigned long limit,
1.1 parser 504: SQL_Driver_query_event_handlers& handlers) {
1.3 parser 505: // todo another set/longjump
1.1 parser 506: OCIParam *mypard;
507: ub2 dtype;
508: text *col_name;
509:
510: struct {
511: ub2 type;
512: char *str;
513: OCILobLocator *var;
514: OCIDefine *def;
515: sb2 indicator;
1.4 ! parser 516: } cols[MAX_COLS]={0};
! 517: int column_count=0;
1.1 parser 518:
1.4 ! parser 519: bool failed=false;
! 520: jmp_buf saved_mark; memcpy(saved_mark, cs.mark, sizeof(jmp_buf));
! 521: if(setjmp(cs.mark)) {
! 522: failed=true;
! 523: memcpy(cs.mark, saved_mark, sizeof(jmp_buf));
! 524: goto cleanup;
! 525: }
! 526: {
! 527: // idea of preincrementing is that at error time all handles would free up
! 528: for(; ++column_count<=MAX_COLS; ) {
! 529: /* get next descriptor, if there is one */
! 530: if(OCIParamGet(stmthp, OCI_HTYPE_STMT, cs.errhp, (void **)&mypard,
! 531: (ub4) column_count)!=OCI_SUCCESS)
! 532: break;
! 533:
! 534: /* Retrieve the data type attribute */
! 535: check(services, cs, "get type", OCIAttrGet(
! 536: (dvoid*) mypard, (ub4)OCI_DTYPE_PARAM,
! 537: (dvoid*) &dtype, (ub4 *) 0, (ub4)OCI_ATTR_DATA_TYPE,
! 538: (OCIError *)cs.errhp));
! 539:
! 540: /* Retrieve the column name attribute */
! 541: ub4 col_name_len;
! 542: check(services, cs, "get name", OCIAttrGet(
! 543: (dvoid*) mypard, (ub4)OCI_DTYPE_PARAM,
! 544: (dvoid**) &col_name, (ub4 *) &col_name_len, (ub4)OCI_ATTR_NAME,
! 545: (OCIError *)cs.errhp));
! 546:
1.1 parser 547: {
1.4 ! parser 548: size_t size=(size_t)col_name_len;
! 549: char *ptr=(char *)services.malloc(size);
! 550: tolower(ptr, (char *)col_name, size);
! 551: handlers.add_column(ptr, size);
! 552: }
! 553:
! 554: ub2 coerce_type=dtype;
! 555: sb4 size=0;
! 556: void *ptr;
! 557:
! 558: switch(dtype) {
! 559: case SQLT_CLOB:
! 560: {
! 561: check(services, cs, "alloc output var desc", OCIDescriptorAlloc(
! 562: (dvoid *)cs.envhp, (dvoid **)(ptr=&cols[column_count-1].var),
! 563: (ub4)OCI_DTYPE_LOB,
! 564: 0, (dvoid **) 0));
! 565:
! 566: size=0;
! 567: break;
! 568: }
! 569: default:
! 570: coerce_type=SQLT_STR;
! 571: ptr=cols[column_count-1].str=(char *)services.malloc(MAX_OUT_STRING_LENGTH+1);
! 572: size=MAX_OUT_STRING_LENGTH;
1.1 parser 573: break;
574: }
1.4 ! parser 575:
! 576: cols[column_count-1].type=coerce_type;
! 577:
! 578: check(services, cs, "DefineByPos", OCIDefineByPos(
! 579: stmthp, &cols[column_count-1].def, cs.errhp,
! 580: column_count, (ub1 *) ptr, size,
! 581: coerce_type, (dvoid *) &cols[column_count-1].indicator,
! 582: (ub2 *) 0, (ub2 *) 0, OCI_DEFAULT));
1.1 parser 583: }
1.4 ! parser 584:
! 585: handlers.before_rows();
! 586:
! 587: for(unsigned long row=0;!limit||row<limit+offset;row++) {
! 588: sword status=OCIStmtFetch(stmthp, cs.errhp, (ub4) 1, (ub4)OCI_FETCH_NEXT,
! 589: (ub4)OCI_DEFAULT);
! 590: if(status!=OCI_SUCCESS) {
! 591: if(status!=OCI_NO_DATA)
! 592: check(services, cs, "fetch", status);
! 593: break;
! 594: }
! 595: if(row>=offset) {
! 596: handlers.add_row();
! 597: for(int i=0; i<column_count; i++) {
! 598: size_t size;
! 599: void *ptr;
! 600: if(cols[i].indicator) { // NULL?
! 601: size=0;
! 602: ptr=0;
! 603: } else
! 604: switch(cols[i].type) {
! 605: case SQLT_CLOB:
! 606: {
! 607: ub4 amtp=4096000000UL;
! 608: ub4 offset=1;
! 609: ub4 loblen=0;
! 610: OCILobLocator *var=(OCILobLocator *)cols[i].var;
! 611: OCILobGetLength(cs.svchp, cs.errhp, var, &loblen);
! 612: if(loblen) {
! 613: size=(size_t)loblen;
! 614: ptr=services.malloc(size);
! 615: check(services, cs, "lobread", OCILobRead(cs.svchp, cs.errhp,
! 616: var, &amtp, offset, (dvoid *) ptr,
! 617: loblen, (dvoid *)0,
! 618: 0,
! 619: (ub2) 0, (ub1) SQLCS_IMPLICIT));
! 620: }
! 621: break;
! 622: }
! 623: default:
! 624: const char *str=cols[i].str;
! 625: size=strlen(str);
1.1 parser 626: ptr=services.malloc(size);
1.4 ! parser 627: memcpy(ptr, str, size);
! 628: break;
1.1 parser 629: }
1.4 ! parser 630: handlers.add_row_cell(ptr, size);
1.1 parser 631: }
632: }
633: }
634: }
635:
1.4 ! parser 636: cleanup: // no check call after this point!
1.3 parser 637: for(int i=0; i<column_count; i++) {
1.1 parser 638: switch(cols[i].type) {
639: case SQLT_CLOB:
640: /* free var locator */
1.4 ! parser 641: OCIDescriptorFree((dvoid *) cols[i].var, (ub4)OCI_DTYPE_LOB);
1.1 parser 642: break;
643: default:
644: break;
645: }
646: }
1.4 ! parser 647:
! 648: if(failed) // need rethrow?
! 649: longjmp(cs.mark, 1);
1.1 parser 650: }
651:
652: private: // conn client library funcs
653:
654: // typedef void (*t_PQfinish)(PGconn *conn); t_PQfinish PQfinish;
655:
656: private: // conn client library funcs linking
657:
658: const char *dlink(const char *dlopen_file_spec) {
659: lt_dlhandle handle=lt_dlopen(dlopen_file_spec);
660: if(!handle)
661: return "can not open the dynamic link module";
662:
663: #define DSLINK(name, action) \
664: name=(t_##name)lt_dlsym(handle, #name); \
665: if(!name) \
666: action;
667:
668: #define DLINK(name) DSLINK(name, return "function " #name " was not found")
669:
670: // DLINK(PQfinish);
671:
672: return 0;
673: }
674:
675: };
676:
677: void check(
678: SQL_Driver_services& services, OracleSQL_connection_struct &cs,
679: const char *step, sword status) {
680:
1.4 ! parser 681: const char *msg;
1.2 parser 682: char reason[MAX_STRING/2];
1.1 parser 683:
684: switch (status) {
685: case OCI_SUCCESS:
1.4 ! parser 686: return; // hurrah
1.1 parser 687: case OCI_ERROR:
688: {
1.4 ! parser 689: sb4 errcode;
! 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
! 694: msg="[can not get error description]";
1.1 parser 695: }
696: case OCI_SUCCESS_WITH_INFO:
1.4 ! parser 697: msg="SUCCESS_WITH_INFO"; break;
1.1 parser 698: case OCI_NEED_DATA:
1.4 ! parser 699: msg="NEED_DATA"; break;
1.1 parser 700: case OCI_NO_DATA:
1.4 ! parser 701: msg="NODATA"; break;
1.1 parser 702: case OCI_INVALID_HANDLE:
1.4 ! parser 703: msg="INVALID_HANDLE"; break;
1.1 parser 704: case OCI_STILL_EXECUTING:
1.4 ! parser 705: msg="STILL_EXECUTE"; break;
1.1 parser 706: case OCI_CONTINUE:
1.4 ! parser 707: msg="CONTINUE"; break;
! 708: default:
! 709: msg="unknown"; break;
1.2 parser 710: }
1.1 parser 711:
1.3 parser 712: snprintf(cs.error, sizeof(cs.error), "%s (%s, %d)",
713: msg, step, (int)status);
1.1 parser 714: longjmp(cs.mark, 1);
715: }
716:
717:
718: /* ----------------------------------------------------------------- */
719: /* Intbind callback that does not do any data input. */
720: /* ----------------------------------------------------------------- */
721: static sb4 cbf_no_data(
722: dvoid *ctxp,
723: OCIBind *bindp,
724: ub4 iter, ub4 index,
725: dvoid **bufpp,
726: ub4 *alenpp,
727: ub1 *piecep,
728: dvoid **indpp) {
1.2 parser 729: *bufpp=(dvoid *) 0;
730: *alenpp=0;
731: static sb2 null_ind=-1;
732: *indpp=(dvoid *) &null_ind;
733: *piecep=OCI_ONE_PIECE;
1.1 parser 734:
735: return OCI_CONTINUE;
736: }
737:
738: /* ----------------------------------------------------------------- */
739: /* Outbind callback for returning data. */
740: /* ----------------------------------------------------------------- */
741: static sb4 cbf_get_data(dvoid *ctxp,
742: OCIBind *bindp,
743: ub4 iter, ub4 index,
744: dvoid **bufpp,
745: ub4 **alenp,
746: ub1 *piecep,
747: dvoid **indpp,
748: ub2 **rcodepp) {
749: OracleSQL_query_lobs::cbf_context_struct &context=
750: *(OracleSQL_query_lobs::cbf_context_struct *)ctxp;
751:
1.3 parser 752: if(index==0) {
1.2 parser 753: static ub4 rows=0;
1.4 ! parser 754: OCIAttrGet((CONST dvoid *) bindp, OCI_HTYPE_BIND, (dvoid *)&rows,
1.1 parser 755: (ub4 *)sizeof(ub2), OCI_ATTR_ROWS_RETURNED, context.cs->errhp);
1.2 parser 756: context.rows->count=(ub2)rows;
757: context.rows->row=(OracleSQL_query_lobs::return_rows::return_row *)
1.1 parser 758: context.services->malloc(sizeof(OracleSQL_query_lobs::return_rows::return_row)*rows);
759: }
760:
761: OracleSQL_query_lobs::return_rows::return_row &var=context.rows->row[index];
762:
763: check(*context.services, *context.cs, "alloc output var desc dynamic", OCIDescriptorAlloc(
764: (dvoid *) context.cs->envhp, (dvoid **)&var.locator,
1.3 parser 765: (ub4)OCI_DTYPE_LOB,
766: 0, (dvoid **) 0));
1.1 parser 767:
1.2 parser 768: *bufpp=var.locator;
769: *alenp=&var.len;
770: *indpp=(dvoid *) &var.ind;
771: *piecep=OCI_ONE_PIECE;
772: *rcodepp=&var.rcode;
1.1 parser 773:
774: return OCI_CONTINUE;
775: }
776:
1.3 parser 777: void tolower(char *out, const char *in, size_t size) {
778: while(size--)
779: *out++=tolower(*in++);
780: }
1.1 parser 781:
782: extern "C" SQL_Driver *create() {
783: return new OracleSQL_Driver();
784: }