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