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