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