![[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:
1.11 parser 8: 2001.07.30 using Oracle 8.1.6 [@test tested with Oracle 7.x.x]
1.1 parser 9: */
1.12 ! parser 10: static const char *RCSId="$Id: parser3oracle.C,v 1.11 2001/08/24 07:04:25 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
1.11 parser 34: # define STOP _asm int 3
1.1 parser 35: #endif
36:
37: #ifndef max
38: inline int max(int a, int b) { return a>b?a:b; }
39: inline int min(int a, int b){ return a<b?a:b; }
40: #endif
41:
42: static char *lsplit(char *string, char delim) {
43: if(string) {
44: char *v=strchr(string, delim);
45: if(v) {
46: *v=0;
47: return v+1;
48: }
49: }
50: return 0;
51: }
52:
53: struct OracleSQL_connection_struct {
54: jmp_buf mark; char error[MAX_STRING];
55: OCIEnv *envhp;
56: OCIServer *srvhp;
57: OCIError *errhp;
58: OCISvcCtx *svchp;
59: OCISession *usrhp;
60: };
61:
62: struct OracleSQL_query_lobs {
63: struct return_rows {
64: struct return_row {
65: OCILobLocator *locator; ub4 len;
66: int ind;
67: ub2 rcode;
68: } *row;
69: int count;
70: };
71: struct cbf_context_struct {
72: SQL_Driver_services *services;
73: OracleSQL_connection_struct *cs;
74: return_rows *rows;
75: };
76: struct Item {
77: const char *name_ptr; size_t name_size;
78: char *data_ptr; size_t data_size;
79: OCILobLocator *locator;
80: OCIBind *bind;
81: return_rows rows;
82: } items[MAX_IN_LOBS];
83: int count;
84: };
85:
86:
87: // forwards
88: void check(
89: SQL_Driver_services& services, OracleSQL_connection_struct &cs,
90: const char *step, sword status);
91: static sb4 cbf_no_data(
92: dvoid *ctxp,
93: OCIBind *bindp,
94: ub4 iter, ub4 index,
95: dvoid **bufpp,
96: ub4 *alenpp,
97: ub1 *piecep,
98: dvoid **indpp);
99: static sb4 cbf_get_data(dvoid *ctxp,
100: OCIBind *bindp,
101: ub4 iter, ub4 index,
102: dvoid **bufpp,
103: ub4 **alenp,
104: ub1 *piecep,
105: dvoid **indpp,
106: ub2 **rcodepp);
1.3 parser 107: void tolower(char *out, const char *in, size_t size);
1.1 parser 108:
109: /**
110: OracleSQL server driver
111: */
112: class OracleSQL_Driver : public SQL_Driver {
113: public:
114:
115: OracleSQL_Driver() : SQL_Driver() {
116: }
117:
118: /// get api version
119: int api_version() { return SQL_DRIVER_API_VERSION; }
120: /// initialize driver by loading sql dynamic link library
121: const char *initialize(const char *dlopen_file_spec) {
1.6 parser 122: const char *error=dlopen_file_spec?
1.1 parser 123: dlink(dlopen_file_spec):"client library column is empty";
1.6 parser 124: if(!error) {
1.4 parser 125: OCIInitialize((ub4)OCI_THREADED, (dvoid *)0,
1.5 parser 126: (dvoid * (*)(void *, unsigned int))0,
127: (dvoid * (*)(void*, void*, unsigned int))0,
128: (void (*)(void*, void*))0 );
1.6 parser 129: }
1.1 parser 130:
1.6 parser 131: return error;
1.1 parser 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: ) {
1.11 parser 143: // connections are cross-request, do not use services._alloc [linked with request]
144: OracleSQL_connection_struct &cs=
145: *(OracleSQL_connection_struct *)::calloc(sizeof(OracleSQL_connection_struct), 1);
146:
1.1 parser 147: char *user=used_only_in_connect_url;
148: char *service=lsplit(user, '@');
149: char *pwd=lsplit(user, ':');
150:
1.2 parser 151: if(!(user && pwd && service))
1.3 parser 152: services._throw("mailformed connect part, must be 'user:pass@service'");
1.2 parser 153:
1.1 parser 154: if(setjmp(cs.mark))
155: services._throw(cs.error);
156:
1.3 parser 157: // Allocate and initialize OCIError handle, attempt #1
158: /*
159: grabbed from sample
160: /server.804/a58234/oci_func.htm#446192
161: but doc
162: /server.804/a58234/oci_func.htm#446100
163: doesnt have this param listed as allowed
164: 8.1.6 client library barks as OCI_INVALID_HANDLE
165: and debugging revealed that OCI_HTYPE_ENV param value is invalid
166: later in doc
167: /server.804/a58234/oci_func.htm#446192
168: on OCIEnvInit thay say
169: "No changes are done to an already initialized handle"
170: think, this is some sort of backward compatibility wonder.
171: leaving as it is, and without check()
172: */
1.10 parser 173: OCIHandleAlloc((dvoid *)NULL, (dvoid **) &cs.envhp, (ub4)OCI_HTYPE_ENV, 0, 0);
1.3 parser 174: // Initialize an environment handle, attempt #2
1.4 parser 175: check(services, cs, "EnvInit", OCIEnvInit(
1.3 parser 176: &cs.envhp, (ub4)OCI_DEFAULT, 0, 0));
177: // Allocate and initialize OCIError handle
1.4 parser 178: check(services, cs, "HandleAlloc errhp", OCIHandleAlloc(
1.3 parser 179: (dvoid *)cs.envhp, (dvoid **) &cs.errhp, (ub4)OCI_HTYPE_ERROR, 0, 0));
180: // Allocate and initialize OCIServer handle
1.4 parser 181: check(services, cs, "HandleAlloc srvhp", OCIHandleAlloc(
1.3 parser 182: (dvoid *)cs.envhp, (dvoid **) &cs.srvhp, (ub4)OCI_HTYPE_SERVER, 0, 0));
183: // Attach to a 'service'; initialize server context handle
1.4 parser 184: check(services, cs, "ServerAttach", OCIServerAttach(
1.10 parser 185: cs.srvhp, cs.errhp, (text *)service, (sb4)strlen(service), (ub4)OCI_DEFAULT));
1.3 parser 186: // Allocate and initialize OCISvcCtx handle
1.4 parser 187: check(services, cs, "HandleAlloc svchp", OCIHandleAlloc(
1.3 parser 188: (dvoid *)cs.envhp, (dvoid **) &cs.svchp, (ub4)OCI_HTYPE_SVCCTX, 0, 0));
189: // set attribute server context in the service context
1.4 parser 190: check(services, cs, "AttrSet server-service", OCIAttrSet(
1.3 parser 191: (dvoid *)cs.svchp, (ub4)OCI_HTYPE_SVCCTX,
1.5 parser 192: (dvoid *)cs.srvhp, (ub4)0,
1.3 parser 193: (ub4)OCI_ATTR_SERVER, (OCIError *)cs.errhp));
194: // allocate a user context handle
1.4 parser 195: check(services, cs, "HandleAlloc usrhp", OCIHandleAlloc(
1.3 parser 196: (dvoid *)cs.envhp, (dvoid **)&cs.usrhp, (ub4)OCI_HTYPE_SESSION, 0, 0));
197: // set 'user' name
1.4 parser 198: check(services, cs, "AttrSet user-session", OCIAttrSet(
1.2 parser 199: (dvoid *)cs.usrhp, (ub4)OCI_HTYPE_SESSION,
1.1 parser 200: (dvoid *)user, (ub4)strlen(user),
1.3 parser 201: OCI_ATTR_USERNAME, cs.errhp));
202: // set 'pwd' password
1.4 parser 203: check(services, cs, "AttrSet pwd-session", OCIAttrSet(
1.2 parser 204: (dvoid *)cs.usrhp, (ub4)OCI_HTYPE_SESSION,
1.1 parser 205: (dvoid *)pwd, (ub4)strlen(pwd),
1.2 parser 206: OCI_ATTR_PASSWORD, cs.errhp));
1.3 parser 207: // Authenticate a user
1.4 parser 208: check(services, cs, "SessionBegin", OCISessionBegin(
1.3 parser 209: cs.svchp, cs.errhp, cs.usrhp,
1.1 parser 210: OCI_CRED_RDBMS, OCI_DEFAULT));
1.3 parser 211: // remember connection in session
1.4 parser 212: check(services, cs, "AttrSet service-session", OCIAttrSet(
1.2 parser 213: (dvoid *)cs.svchp, (ub4)OCI_HTYPE_SVCCTX,
1.1 parser 214: (dvoid *)cs.usrhp, (ub4)0,
1.2 parser 215: OCI_ATTR_SESSION, cs.errhp));
1.1 parser 216:
1.3 parser 217: // return created connection
1.1 parser 218: *(OracleSQL_connection_struct **)connection=&cs;
219: }
1.11 parser 220: void disconnect(void *connection) {
1.1 parser 221: OracleSQL_connection_struct &cs=*(OracleSQL_connection_struct *)connection;
1.3 parser 222: // Terminate a user session
1.11 parser 223: OCISessionEnd(
1.12 ! parser 224: cs.svchp, cs.errhp, cs.usrhp, (ub4)OCI_DEFAULT);
1.3 parser 225: // Detach from a server; uninitialize server context handle
1.11 parser 226: OCIServerDetach(
1.12 ! parser 227: cs.srvhp, cs.errhp, (ub4)OCI_DEFAULT);
1.11 parser 228: // Free a previously allocated handles
229: OCIHandleFree(
1.12 ! parser 230: (dvoid *)cs.srvhp, (ub4)OCI_HTYPE_SERVER);
1.11 parser 231: OCIHandleFree(
1.12 ! parser 232: (dvoid *)cs.svchp, (ub4)OCI_HTYPE_SVCCTX);
1.11 parser 233: OCIHandleFree(
1.12 ! parser 234: (dvoid *)cs.errhp, (ub4)OCI_HTYPE_ERROR);
1.11 parser 235: OCIHandleFree(
1.12 ! parser 236: (dvoid *)cs.envhp, (ub4)OCI_HTYPE_ENV);
1.11 parser 237:
238: // connections are cross-request, do not use services._alloc [linked with request]
239: ::free(&cs);
1.1 parser 240: }
241: void commit(SQL_Driver_services& services, void *connection) {
242: OracleSQL_connection_struct &cs=*(OracleSQL_connection_struct *)connection;
1.4 parser 243: if(setjmp(cs.mark))
244: services._throw(cs.error);
1.1 parser 245:
246: check(services, cs, "commit", OCITransCommit(cs.svchp, cs.errhp, 0));
247: }
248: void rollback(SQL_Driver_services& services, void *connection) {
249: OracleSQL_connection_struct &cs=*(OracleSQL_connection_struct *)connection;
1.4 parser 250: if(setjmp(cs.mark))
251: services._throw(cs.error);
1.1 parser 252:
253: check(services, cs, "rollback", OCITransRollback(cs.svchp, cs.errhp, 0));
254: }
255:
256: bool ping(SQL_Driver_services&, void *connection) {
1.4 parser 257: // maybe OCIServerVersion?
258: return true;
1.1 parser 259: }
260:
261: unsigned int quote(
262: SQL_Driver_services&, void *,
263: char *to, const char *from, unsigned int length) {
264: /*
265: it's already UNTAINT_TIMES_BIGGER
266: */
267: unsigned int result=length;
268: while(length--) {
269: switch(*from) {
270: case '\'': // "'" -> "''"
271: *to++='\'';
272: break;
273: case '\\': // "\" -> "\\"
274: *to++='\'';
275: break;
276: }
277: *to++=*from++;
278: }
279: return result;
280: }
281: void query(
282: SQL_Driver_services& services, void *connection,
283: const char *astatement, unsigned long offset, unsigned long limit,
284: SQL_Driver_query_event_handlers& handlers) {
1.10 parser 285:
1.1 parser 286: OracleSQL_connection_struct &cs=*(OracleSQL_connection_struct *)connection;
287: OracleSQL_query_lobs lobs={{0}, 0};
288: OCIStmt *stmthp=0;
289:
290: bool failed=false;
291: if(setjmp(cs.mark)) {
292: failed=true;
293: goto cleanup;
1.7 parser 294: } else {
1.1 parser 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,
1.10 parser 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,
1.5 parser 311: (ub4)1+i,
1.1 parser 312: (dvoid *)&lobs.items[i].locator,
1.5 parser 313: (sword)sizeof (lobs.items[i].locator), SQLT_CLOB, (dvoid *)0,
314: (ub2 *)0, (ub2 *)0, (ub4)0, (ub4 *)0, OCI_DATA_AT_EXEC));
1.1 parser 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.10 parser 334: if(OCILobLocator *locator=lobs.items[i].locator)
335: OCIDescriptorFree((dvoid *)locator, (ub4)OCI_DTYPE_LOB);
1.1 parser 336:
1.10 parser 337: /* free rows descriptors */
338: OracleSQL_query_lobs::return_rows &rows=lobs.items[i].rows;
339: for(int r=0; r<rows.count; r++)
340: OCIDescriptorFree((dvoid *)rows.row[r].locator, (ub4)OCI_DTYPE_LOB);
1.1 parser 341: }
342: }
343: if(stmthp)
1.10 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++=',';
1.9 parser 419: n+=sprintf(n, "%.*s", lobs.items[i].name_size, lobs.items[i].name_ptr);
1.1 parser 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';
1.9 parser 427: n+=sprintf(n, ":%.*s", lobs.items[i].name_size, lobs.items[i].name_ptr);
1.1 parser 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,
1.5 parser 449: (ub4 *)0, OCI_ATTR_STMT_TYPE, cs.errhp));
1.1 parser 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:
1.6 parser 458: sword status=OCIStmtExecute(cs.svchp, stmthp, cs.errhp,
1.10 parser 459: (ub4)stmt_type==OCI_STMT_SELECT?0:1, (ub4)0,
460: (OCISnapshot *)NULL,
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,
1.10 parser 475: (dvoid *)lobs.items[i].data_ptr, (ub4)bytes_to_write, OCI_ONE_PIECE,
1.5 parser 476: (dvoid *)0, 0, (ub2)0,
1.1 parser 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;
1.7 parser 488: default:
489: /*
1.1 parser 490: case OCI_STMT_INSERT:
491: case OCI_STMT_UPDATE:
1.7 parser 492: */
1.1 parser 493: break;
494: }
495: }
496:
497: void fetch_table(SQL_Driver_services& services, OracleSQL_connection_struct &cs,
1.3 parser 498: OCIStmt *stmthp, unsigned long offset, unsigned long limit,
1.1 parser 499: SQL_Driver_query_event_handlers& handlers) {
1.7 parser 500:
1.1 parser 501: OCIParam *mypard;
502: ub2 dtype;
503: text *col_name;
504:
505: struct {
506: ub2 type;
507: char *str;
508: OCILobLocator *var;
509: OCIDefine *def;
510: sb2 indicator;
1.4 parser 511: } cols[MAX_COLS]={0};
512: int column_count=0;
1.1 parser 513:
1.4 parser 514: bool failed=false;
515: jmp_buf saved_mark; memcpy(saved_mark, cs.mark, sizeof(jmp_buf));
516: if(setjmp(cs.mark)) {
517: failed=true;
518: goto cleanup;
1.7 parser 519: } else {
1.4 parser 520: // idea of preincrementing is that at error time all handles would free up
1.5 parser 521: while(++column_count<=MAX_COLS) {
1.4 parser 522: /* get next descriptor, if there is one */
523: if(OCIParamGet(stmthp, OCI_HTYPE_STMT, cs.errhp, (void **)&mypard,
524: (ub4) column_count)!=OCI_SUCCESS)
525: break;
526:
527: /* Retrieve the data type attribute */
528: check(services, cs, "get type", OCIAttrGet(
529: (dvoid*) mypard, (ub4)OCI_DTYPE_PARAM,
1.5 parser 530: (dvoid*) &dtype, (ub4 *)0, (ub4)OCI_ATTR_DATA_TYPE,
1.4 parser 531: (OCIError *)cs.errhp));
532:
533: /* Retrieve the column name attribute */
534: ub4 col_name_len;
535: check(services, cs, "get name", OCIAttrGet(
536: (dvoid*) mypard, (ub4)OCI_DTYPE_PARAM,
537: (dvoid**) &col_name, (ub4 *) &col_name_len, (ub4)OCI_ATTR_NAME,
538: (OCIError *)cs.errhp));
539:
1.1 parser 540: {
1.4 parser 541: size_t size=(size_t)col_name_len;
542: char *ptr=(char *)services.malloc(size);
543: tolower(ptr, (char *)col_name, size);
544: handlers.add_column(ptr, size);
545: }
546:
547: ub2 coerce_type=dtype;
548: sb4 size=0;
549: void *ptr;
550:
551: switch(dtype) {
552: case SQLT_CLOB:
553: {
554: check(services, cs, "alloc output var desc", OCIDescriptorAlloc(
555: (dvoid *)cs.envhp, (dvoid **)(ptr=&cols[column_count-1].var),
556: (ub4)OCI_DTYPE_LOB,
1.5 parser 557: 0, (dvoid **)0));
1.4 parser 558:
559: size=0;
560: break;
561: }
562: default:
563: coerce_type=SQLT_STR;
564: ptr=cols[column_count-1].str=(char *)services.malloc(MAX_OUT_STRING_LENGTH+1);
565: size=MAX_OUT_STRING_LENGTH;
1.1 parser 566: break;
567: }
1.4 parser 568:
569: cols[column_count-1].type=coerce_type;
570:
571: check(services, cs, "DefineByPos", OCIDefineByPos(
572: stmthp, &cols[column_count-1].def, cs.errhp,
573: column_count, (ub1 *) ptr, size,
574: coerce_type, (dvoid *) &cols[column_count-1].indicator,
1.5 parser 575: (ub2 *)0, (ub2 *)0, OCI_DEFAULT));
1.1 parser 576: }
1.4 parser 577:
578: handlers.before_rows();
579:
1.5 parser 580: for(unsigned long row=0; !limit||row<limit+offset; row++) {
581: sword status=OCIStmtFetch(stmthp, cs.errhp, (ub4)1, (ub4)OCI_FETCH_NEXT,
1.4 parser 582: (ub4)OCI_DEFAULT);
583: if(status!=OCI_SUCCESS) {
584: if(status!=OCI_NO_DATA)
585: check(services, cs, "fetch", status);
586: break;
587: }
588: if(row>=offset) {
589: handlers.add_row();
590: for(int i=0; i<column_count; i++) {
591: size_t size;
592: void *ptr;
593: if(cols[i].indicator) { // NULL?
594: size=0;
595: ptr=0;
1.6 parser 596: } else // not NULL
1.4 parser 597: switch(cols[i].type) {
598: case SQLT_CLOB:
599: {
600: ub4 amtp=4096000000UL;
601: ub4 offset=1;
602: ub4 loblen=0;
603: OCILobLocator *var=(OCILobLocator *)cols[i].var;
604: OCILobGetLength(cs.svchp, cs.errhp, var, &loblen);
605: if(loblen) {
606: size=(size_t)loblen;
607: ptr=services.malloc(size);
608: check(services, cs, "lobread", OCILobRead(cs.svchp, cs.errhp,
609: var, &amtp, offset, (dvoid *) ptr,
610: loblen, (dvoid *)0,
611: 0,
1.5 parser 612: (ub2)0, (ub1) SQLCS_IMPLICIT));
1.4 parser 613: }
614: break;
615: }
616: default:
1.6 parser 617: if(const char *str=cols[i].str) {
618: size=strlen(str);
619: ptr=services.malloc(size);
620: memcpy(ptr, str, size);
621: } else {
622: size=0;
623: ptr=0;
624: }
1.4 parser 625: break;
1.1 parser 626: }
1.4 parser 627: handlers.add_row_cell(ptr, size);
1.1 parser 628: }
629: }
630: }
631: }
632:
1.4 parser 633: cleanup: // no check call after this point!
1.3 parser 634: for(int i=0; i<column_count; i++) {
1.1 parser 635: switch(cols[i].type) {
636: case SQLT_CLOB:
637: /* free var locator */
1.4 parser 638: OCIDescriptorFree((dvoid *) cols[i].var, (ub4)OCI_DTYPE_LOB);
1.1 parser 639: break;
640: default:
641: break;
642: }
643: }
1.4 parser 644:
645: if(failed) // need rethrow?
1.5 parser 646: longjmp(saved_mark, 1);
1.1 parser 647: }
648:
649: private: // conn client library funcs
1.6 parser 650:
651: friend void check(
652: SQL_Driver_services& services, OracleSQL_connection_struct &cs,
653: const char *step, sword status);
654: friend sb4 cbf_get_data(dvoid *ctxp,
655: OCIBind *bindp,
656: ub4 iter, ub4 index,
657: dvoid **bufpp,
658: ub4 **alenp,
659: ub1 *piecep,
660: dvoid **indpp,
661: ub2 **rcodepp);
662:
663:
664: #define OCI_DECL(name, params) \
665: typedef sword (*t_OCI##name)params; t_OCI##name OCI##name
666:
667: OCI_DECL(Initialize, (ub4 mode, dvoid *ctxp,
668: dvoid * (*malocfp)(dvoid *ctxp, size_t size),
669: dvoid * (*ralocfp)(dvoid *ctxp, dvoid *memptr, size_t newsize),
670: void (*mfreefp)(dvoid *ctxp, dvoid *memptr) ));
671:
672: OCI_DECL(EnvInit, (OCIEnv **envp, ub4 mode,
673: size_t xtramem_sz, dvoid **usrmempp));
674:
675: OCI_DECL(AttrGet, (CONST dvoid *trgthndlp, ub4 trghndltyp,
676: dvoid *attributep, ub4 *sizep, ub4 attrtype,
677: OCIError *errhp));
678:
679: OCI_DECL(AttrSet, (dvoid *trgthndlp, ub4 trghndltyp, dvoid *attributep,
680: ub4 size, ub4 attrtype, OCIError *errhp));
681:
682: OCI_DECL(BindByPos, (OCIStmt *stmtp, OCIBind **bindp, OCIError *errhp,
683: ub4 position, dvoid *valuep, sb4 value_sz,
684: ub2 dty, dvoid *indp, ub2 *alenp, ub2 *rcodep,
685: ub4 maxarr_len, ub4 *curelep, ub4 mode));
686:
687: OCI_DECL(BindDynamic, (OCIBind *bindp, OCIError *errhp, dvoid *ictxp,
688: OCICallbackInBind icbfp, dvoid *octxp,
689: OCICallbackOutBind ocbfp));
690:
691: OCI_DECL(DefineByPos, (OCIStmt *stmtp, OCIDefine **defnp, OCIError *errhp,
692: ub4 position, dvoid *valuep, sb4 value_sz, ub2 dty,
693: dvoid *indp, ub2 *rlenp, ub2 *rcodep, ub4 mode));
694:
695: OCI_DECL(DescriptorAlloc, (CONST dvoid *parenth, dvoid **descpp,
696: CONST ub4 type, CONST size_t xtramem_sz,
697: dvoid **usrmempp));
698:
699: OCI_DECL(DescriptorFree, (dvoid *descp, CONST ub4 type));
700:
701:
702: OCI_DECL(ErrorGet, (dvoid *hndlp, ub4 recordno, OraText *sqlstate,
703: sb4 *errcodep, OraText *bufp, ub4 bufsiz, ub4 type));
704:
705: OCI_DECL(HandleAlloc, (CONST dvoid *parenth, dvoid **hndlpp, CONST ub4 type,
706: CONST size_t xtramem_sz, dvoid **usrmempp));
1.1 parser 707:
1.6 parser 708: OCI_DECL(HandleFree, (dvoid *hndlp, CONST ub4 type));
709:
710: OCI_DECL(LobGetLength, (OCISvcCtx *svchp, OCIError *errhp,
711: OCILobLocator *locp,
712: ub4 *lenp));
713:
714: OCI_DECL(LobRead, (OCISvcCtx *svchp, OCIError *errhp, OCILobLocator *locp,
715: ub4 *amtp, ub4 offset, dvoid *bufp, ub4 bufl,
716: dvoid *ctxp, sb4 (*cbfp)(dvoid *ctxp,
717: CONST dvoid *bufp,
718: ub4 len,
719: ub1 piece),
720: ub2 csid, ub1 csfrm));
721:
722: OCI_DECL(LobWrite, (OCISvcCtx *svchp, OCIError *errhp, OCILobLocator *locp,
723: ub4 *amtp, ub4 offset, dvoid *bufp, ub4 buflen,
724: ub1 piece, dvoid *ctxp,
725: sb4 (*cbfp)(dvoid *ctxp,
726: dvoid *bufp,
727: ub4 *len,
728: ub1 *piece),
729: ub2 csid, ub1 csfrm));
730:
731: OCI_DECL(ParamGet, (CONST dvoid *hndlp, ub4 htype, OCIError *errhp,
732: dvoid **parmdpp, ub4 pos));
733:
734: OCI_DECL(ServerAttach, (OCIServer *srvhp, OCIError *errhp,
735: CONST OraText *dblink, sb4 dblink_len, ub4 mode));
736:
737: OCI_DECL(ServerDetach, (OCIServer *srvhp, OCIError *errhp, ub4 mode));
738:
739: OCI_DECL(SessionBegin, (OCISvcCtx *svchp, OCIError *errhp, OCISession *usrhp,
740: ub4 credt, ub4 mode));
741:
742: OCI_DECL(SessionEnd, (OCISvcCtx *svchp, OCIError *errhp, OCISession *usrhp,
743: ub4 mode));
744:
745: OCI_DECL(StmtExecute, (OCISvcCtx *svchp, OCIStmt *stmtp, OCIError *errhp,
746: ub4 iters, ub4 rowoff, CONST OCISnapshot *snap_in,
747: OCISnapshot *snap_out, ub4 mode));
748:
749: OCI_DECL(StmtFetch, (OCIStmt *stmtp, OCIError *errhp, ub4 nrows,
750: ub2 orientation, ub4 mode));
751:
752: OCI_DECL(StmtPrepare, (OCIStmt *stmtp, OCIError *errhp, CONST OraText *stmt,
753: ub4 stmt_len, ub4 language, ub4 mode));
754:
755: OCI_DECL(TransCommit, (OCISvcCtx *svchp, OCIError *errhp, ub4 flags));
756:
757: OCI_DECL(TransRollback, (OCISvcCtx *svchp, OCIError *errhp, ub4 flags));
1.1 parser 758:
759: private: // conn client library funcs linking
760:
761: const char *dlink(const char *dlopen_file_spec) {
762: lt_dlhandle handle=lt_dlopen(dlopen_file_spec);
763: if(!handle)
764: return "can not open the dynamic link module";
765:
766: #define DSLINK(name, action) \
767: name=(t_##name)lt_dlsym(handle, #name); \
768: if(!name) \
769: action;
770:
1.6 parser 771: #define OLINK(name) DSLINK(OCI##name, return "function OCI" #name " was not found")
1.1 parser 772:
1.6 parser 773: OLINK(Initialize);
774: OLINK(EnvInit);
775: OLINK(AttrGet); OLINK(AttrSet);
776: OLINK(BindByPos); OLINK(BindDynamic);
777: OLINK(DefineByPos);
778: OLINK(DescriptorAlloc); OLINK(DescriptorFree);
779: OLINK(ErrorGet);
780: OLINK(HandleAlloc); OLINK(HandleFree);
781: OLINK(LobGetLength);
782: OLINK(LobRead); OLINK(LobWrite);
783: OLINK(ParamGet);
784: OLINK(ServerAttach); OLINK(ServerDetach);
785: OLINK(SessionBegin); OLINK(SessionEnd);
786: OLINK(StmtExecute); OLINK(StmtFetch); OLINK(StmtPrepare);
787: OLINK(TransCommit); OLINK(TransRollback);
1.1 parser 788:
789: return 0;
790: }
791:
1.6 parser 792: } *OracleSQL_driver;
1.1 parser 793:
794: void check(
795: SQL_Driver_services& services, OracleSQL_connection_struct &cs,
796: const char *step, sword status) {
797:
1.4 parser 798: const char *msg;
1.2 parser 799: char reason[MAX_STRING/2];
1.1 parser 800:
801: switch (status) {
802: case OCI_SUCCESS:
1.4 parser 803: return; // hurrah
1.1 parser 804: case OCI_ERROR:
805: {
1.4 parser 806: sb4 errcode;
1.10 parser 807: if(OracleSQL_driver->OCIErrorGet((dvoid *)cs.errhp, (ub4)1, (text *)NULL, &errcode,
808: (text *)reason, (ub4)sizeof(reason), OCI_HTYPE_ERROR)==OCI_SUCCESS)
1.4 parser 809: msg=reason;
810: else
811: msg="[can not get error description]";
1.8 parser 812: break;
1.1 parser 813: }
814: case OCI_SUCCESS_WITH_INFO:
1.4 parser 815: msg="SUCCESS_WITH_INFO"; break;
1.1 parser 816: case OCI_NEED_DATA:
1.4 parser 817: msg="NEED_DATA"; break;
1.1 parser 818: case OCI_NO_DATA:
1.4 parser 819: msg="NODATA"; break;
1.1 parser 820: case OCI_INVALID_HANDLE:
1.4 parser 821: msg="INVALID_HANDLE"; break;
1.1 parser 822: case OCI_STILL_EXECUTING:
1.4 parser 823: msg="STILL_EXECUTE"; break;
1.1 parser 824: case OCI_CONTINUE:
1.4 parser 825: msg="CONTINUE"; break;
826: default:
827: msg="unknown"; break;
1.2 parser 828: }
1.1 parser 829:
1.3 parser 830: snprintf(cs.error, sizeof(cs.error), "%s (%s, %d)",
831: msg, step, (int)status);
1.1 parser 832: longjmp(cs.mark, 1);
833: }
834:
835:
836: /* ----------------------------------------------------------------- */
837: /* Intbind callback that does not do any data input. */
838: /* ----------------------------------------------------------------- */
839: static sb4 cbf_no_data(
840: dvoid *ctxp,
841: OCIBind *bindp,
842: ub4 iter, ub4 index,
843: dvoid **bufpp,
844: ub4 *alenpp,
845: ub1 *piecep,
846: dvoid **indpp) {
1.5 parser 847: *bufpp=(dvoid *)0;
1.2 parser 848: *alenpp=0;
849: static sb2 null_ind=-1;
850: *indpp=(dvoid *) &null_ind;
851: *piecep=OCI_ONE_PIECE;
1.1 parser 852:
853: return OCI_CONTINUE;
854: }
855:
856: /* ----------------------------------------------------------------- */
857: /* Outbind callback for returning data. */
858: /* ----------------------------------------------------------------- */
859: static sb4 cbf_get_data(dvoid *ctxp,
860: OCIBind *bindp,
861: ub4 iter, ub4 index,
862: dvoid **bufpp,
863: ub4 **alenp,
864: ub1 *piecep,
865: dvoid **indpp,
866: ub2 **rcodepp) {
867: OracleSQL_query_lobs::cbf_context_struct &context=
868: *(OracleSQL_query_lobs::cbf_context_struct *)ctxp;
869:
1.3 parser 870: if(index==0) {
1.6 parser 871: static ub4 rows;
872: check(*context.services, *context.cs, "AttrGet cbf_get_data ROWS_RETURNED",
873: OracleSQL_driver->OCIAttrGet(
874: (CONST dvoid *) bindp, OCI_HTYPE_BIND, (dvoid *)&rows,
875: (ub4 *)sizeof(ub2), OCI_ATTR_ROWS_RETURNED, context.cs->errhp)) ;
1.2 parser 876: context.rows->count=(ub2)rows;
877: context.rows->row=(OracleSQL_query_lobs::return_rows::return_row *)
1.1 parser 878: context.services->malloc(sizeof(OracleSQL_query_lobs::return_rows::return_row)*rows);
879: }
880:
881: OracleSQL_query_lobs::return_rows::return_row &var=context.rows->row[index];
882:
1.6 parser 883: check(*context.services, *context.cs, "alloc output var desc dynamic", OracleSQL_driver->OCIDescriptorAlloc(
1.1 parser 884: (dvoid *) context.cs->envhp, (dvoid **)&var.locator,
1.3 parser 885: (ub4)OCI_DTYPE_LOB,
1.5 parser 886: 0, (dvoid **)0));
1.1 parser 887:
1.2 parser 888: *bufpp=var.locator;
889: *alenp=&var.len;
890: *indpp=(dvoid *) &var.ind;
891: *piecep=OCI_ONE_PIECE;
892: *rcodepp=&var.rcode;
1.1 parser 893:
894: return OCI_CONTINUE;
895: }
896:
1.3 parser 897: void tolower(char *out, const char *in, size_t size) {
898: while(size--)
899: *out++=tolower(*in++);
900: }
1.1 parser 901:
902: extern "C" SQL_Driver *create() {
1.10 parser 903: //_asm int 3;
1.6 parser 904: return OracleSQL_driver=new OracleSQL_Driver();
1.1 parser 905: }