![[BACK]](/cvsweb/icons/back.gif){kind=icon}
Return to parser3odbc.C CVS log ![[TXT]](/cvsweb/icons/text.gif){kind=icon}
![[DIR]](/cvsweb/icons/dir.gif){kind=icon}
Up to [parser3project] / sql / odbc|
Return to parser3odbc.C CVS log | Up to [parser3project] / sql / odbc |
1.1 parser 1: /** @file
2: Parser ODBC driver.
3:
1.13 paf 4: Copyright(c) 2001, 2003 ArtLebedev Group (http://www.artlebedev.com)
1.1 parser 5:
1.5 paf 6: Author: Alexandr Petrosian <paf@design.ru> (http://paf.design.ru)
1.1 parser 7: */
1.13.6.1! paf 8: static const char *RCSId="$Id: parser3odbc.C,v 1.14 2003/03/25 14:07:29 paf Exp $";
1.1 parser 9:
10: #ifndef _MSC_VER
11: # error compile ISAPI module with MSVC [no urge for now to make it autoconf-ed (PAF)]
12: #endif
13:
14: #include <string.h>
15: #include <stdio.h>
16: #include <stdlib.h>
1.2 paf 17: #include <setjmp.h>
1.1 parser 18:
19: #include "pa_sql_driver.h"
20:
21: #include <AFXDB.H>
22:
1.10 paf 23: // defines
24:
25: #define MAX_COLS 500
26:
1.1 parser 27: #define MAX_STRING 0x400
1.7 paf 28: #define MAX_NUMBER 40
29:
1.10 paf 30: // new in MSSQL2000, no MFC constants
1.7 paf 31: #ifndef SQL_NVARCHAR
32: #define SQL_NVARCHAR (-9)
33: #endif
34: #ifndef SQL_NTEXT
35: #define SQL_NTEXT (-10)
36: #endif
37: #ifndef SQL_SMALLDATETIME
38: #define SQL_SMALLDATETIME 11
39: #endif
40: // create table test (id int, a smalldatetime, b ntext, c nvarchar(100))
1.1 parser 41:
42: #define snprintf _snprintf
43: #ifndef strncasecmp
44: # define strncasecmp _strnicmp
45: #endif
46:
47: static char *lsplit(char *string, char delim) {
48: if(string) {
49: char *v=strchr(string, delim);
50: if(v) {
51: *v=0;
52: return v+1;
53: }
54: }
55: return 0;
56: }
57:
58: /**
59: ODBC server driver
60: */
61: class ODBC_Driver : public SQL_Driver {
62: public:
63:
64: ODBC_Driver() : SQL_Driver() {
65: }
66:
67: /// get api version
68: int api_version() { return SQL_DRIVER_API_VERSION; }
1.3 paf 69: const char *initialize(char *dlopen_file_spec) { return 0; }
1.1 parser 70: /** connect
71: @param used_only_in_connect_url
72: format: @b DSN=dsn;UID=user;PWD=password (ODBC connect string)
73: WARNING: must be used only to connect, for buffer doesn't live long
74: */
75: void connect(
76: char *used_only_in_connect_url,
77: SQL_Driver_services& services,
78: void **connection ///< output: CDatabase *
79: ) {
80: // _asm int 3;
81: CDatabase *db;
82: TRY {
83: db=new CDatabase();
84: db->OpenEx(used_only_in_connect_url, CDatabase::noOdbcDialog);
85: db->BeginTrans();
86: }
87: CATCH_ALL (e) {
88: _throw(services, e);
89: db=0; // calm, compiler
90: }
91: END_CATCH_ALL
92:
93: *(CDatabase **)connection=db;
94: }
95: void disconnect(void *connection) {
96: CDatabase *db=static_cast<CDatabase *>(connection);
97: TRY
98: delete db;
99: CATCH_ALL (e) {
100: // nothing
101: }
102: END_CATCH_ALL
103: }
104: void commit(SQL_Driver_services& services, void *connection) {
105: CDatabase *db=static_cast<CDatabase *>(connection);
106: TRY
107: db->CommitTrans();
108: db->BeginTrans();
109: CATCH_ALL (e) {
110: _throw(services, e);
111: }
112: END_CATCH_ALL
113: }
114: void rollback(SQL_Driver_services& services, void *connection) {
115: CDatabase *db=static_cast<CDatabase *>(connection);
116: TRY
117: db->Rollback();
118: db->BeginTrans();
119: CATCH_ALL (e) {
120: _throw(services, e);
121: }
122: END_CATCH_ALL
123: }
124:
125: bool ping(SQL_Driver_services&, void *connection) {
126: return true;
127: }
128:
1.13.6.1! paf 129: const char* quote(
! 130: SQL_Driver_services& services, void *connection,
! 131: const char *from, unsigned int length) {
! 132: char *result=(char*)services.malloc_atomic(length*2+1);
! 133: char *to=result;
! 134: while(length--) {
! 135: if(*from=='\'') { // ' -> ''
! 136: *to++='\''; result++;
1.3 paf 137: }
1.13.6.1! paf 138: *to++=*from++;
! 139: }
! 140: *to=0;
! 141: return result;
1.1 parser 142: }
143: void query(
144: SQL_Driver_services& services, void *connection,
145: const char *statement, unsigned long offset, unsigned long limit,
146: SQL_Driver_query_event_handlers& handlers) {
147:
148: CDatabase *db=static_cast<CDatabase *>(connection);
149:
150: while(isspace(*statement))
151: statement++;
152:
153: TRY {
1.8 paf 154: // mk:@MSITStore:C:\Program%20Files\Microsoft%20SQL%20Server\80\Tools\Books\adosql.chm::/adoprg02_4g33.htm
155: // Server cursors are created only for statements that begin with:
156: // SELECT
157: // EXEC[ute] procedure_name
158: // call procedure_name
159: // mk:@MSITStore:C:\Program%20Files\Microsoft%20SQL%20Server\80\Tools\Books\odbcsql.chm::/od_6_035_5dnp.htm
160: // The ODBC CALL escape sequence for calling a procedure is:
161: // {[?=]call procedure_name[([parameter][,[parameter]]...)]}
162: if(strncasecmp(statement, "select", 6)==0
163: || strncasecmp(statement, "EXEC", 4)==0
164: || strncasecmp(statement, "call", 4)==0
165: || strncasecmp(statement, "{", 1)==0) {
1.1 parser 166: CRecordset rs(db);
1.9 paf 167: TRY {
168: rs.Open(
169: CRecordset::forwardOnly,
170: statement,
171: CRecordset::executeDirect
172: );
173: } CATCH_ALL (e) {
174: // could not fetch a table
175: TRY {
176: // then try resultless query
177: db->ExecuteSQL(statement);
178: // OK then
179: return;
180: } CATCH_ALL (e2) {
181: // still nothing good
182: _throw(services, e); // throw ORIGINAL exception
183: } END_CATCH_ALL
184: } END_CATCH_ALL
1.1 parser 185:
186: int column_count=rs.GetODBCFieldCount();
187: if(!column_count)
188: services._throw("result contains no columns");
189:
1.10 paf 190: SWORD column_types[MAX_COLS];
191: if(column_count>MAX_COLS)
192: column_count=MAX_COLS;
193:
1.12 paf 194: SQL_Error sql_error;
195: #define CHECK(afailed) if(afailed) services._throw(sql_error)
196:
1.1 parser 197: for(int i=0; i<column_count; i++){
198: CString string;
199: CODBCFieldInfo fieldinfo;
200: rs.GetODBCFieldInfo(i, fieldinfo);
1.10 paf 201: column_types[i]=fieldinfo.m_nSQLType;
1.1 parser 202: size_t size=fieldinfo.m_strName.GetLength();
203: void *ptr=0;
204: if(size) {
1.13.6.1! paf 205: ptr=services.malloc_atomic(size);
1.1 parser 206: memcpy(ptr, (char *)LPCTSTR(fieldinfo.m_strName), size);
207: }
1.12 paf 208: CHECK(handlers.add_column(sql_error, ptr, size));
1.1 parser 209: }
210:
1.12 paf 211: CHECK(handlers.before_rows(sql_error));
1.1 parser 212:
213: unsigned long row=0;
1.10 paf 214: CDBVariant v;
215: CString s;
1.1 parser 216: while(!rs.IsEOF() && (!limit||(row<offset+limit))) {
217: if(row>=offset) {
1.12 paf 218: CHECK(handlers.add_row(sql_error));
1.10 paf 219: for(int i=0; i<column_count; i++) {
1.6 paf 220: size_t size;
221: void *ptr;
1.10 paf 222: switch(column_types[i]) {
1.7 paf 223: //case xBOOL:
224: // case SQL_INTEGER: // serg@design.ru did that in parser2. test first!
225: //case SQL_DATETIME:
1.11 paf 226: case SQL_BINARY:
227: case SQL_VARBINARY:
228: case SQL_LONGVARBINARY:
1.7 paf 229: case SQL_SMALLDATETIME:
1.10 paf 230: rs.GetFieldValue(i, v);
231: getFromDBVariant(services, v, ptr, size);
232: break;
1.7 paf 233: default:
1.10 paf 234: rs.GetFieldValue(i, s);
235: getFromString(services, s, ptr, size);
236: break;
1.1 parser 237: }
1.12 paf 238: CHECK(handlers.add_row_cell(sql_error, ptr, size));
1.1 parser 239: }
240: }
241: rs.MoveNext(); row++;
242: }
243:
244: rs.Close();
245: } else {
246: db->ExecuteSQL(statement);
247: }
1.9 paf 248: } CATCH_ALL (e) {
1.1 parser 249: _throw(services, e);
1.9 paf 250: } END_CATCH_ALL
1.7 paf 251: }
252:
253: void getFromDBVariant(SQL_Driver_services& services, CDBVariant& v, void *& ptr, size_t& size) {
254: switch(v.m_dwType) {
255: case DBVT_NULL: // No union member is valid for access.
256: ptr=0;
257: size=0;
258: break;
259: /* case DBVT_BOOL:
260: ptr=v.m_boolVal?"1":"0";
261: size=1;
262: break;*/
263: /* case DBVT_UCHAR:
264: size=strlen(ptr=v.m_chVal);
265: break;
266: case DBVT_SHORT:
267: char buf[MAX_NUMBER];
268: size=snprintf(HEAPIZE buf, "%d", v.m_iVal);
269: break;*/
270: /* case DBVT_LONG:
271: {
272: char local_buf[MAX_NUMBER];
273: size=snprintf(local_buf, MAX_NUMBER, "%ld", v.m_lVal);
1.13.6.1! paf 274: ptr=services.malloc_atomic(size);
1.7 paf 275: memcpy(ptr, local_buf, size);
276: break;
277: }*/
278: /*case DBVT_SINGLE:
279: m_fltVal
280: break;
281: case DBVT_DOUBLE m_dblVal
1.11 paf 282: case DBVT_STRING m_pstring */
283: case DBVT_BINARY:
284: {
285: if(size=v.m_pbinary->m_dwDataLength) {
1.13.6.1! paf 286: ptr=services.malloc_atomic(size);
1.11 paf 287: memcpy(ptr, ::GlobalLock(v.m_pbinary->m_hData), size);
288: ::GlobalUnlock(v.m_pbinary->m_hData);
289: } else
290: ptr=0;
291: break;
292: }
1.7 paf 293: case DBVT_DATE:
294: {
295: char local_buf[MAX_STRING];
296: size=snprintf(local_buf, MAX_STRING,
297: "%04d-%02d-%02d %02d:%02d:%02d.%03d",
298: v.m_pdate->year,
299: v.m_pdate->month,
300: v.m_pdate->day,
301: v.m_pdate->hour,
302: v.m_pdate->minute,
303: v.m_pdate->second,
304: v.m_pdate->fraction);
1.13.6.1! paf 305: ptr=services.malloc_atomic(size);
1.7 paf 306: memcpy(ptr, local_buf, size);
307: break;
308: }
309: default:
310: char msg[MAX_STRING];
311: snprintf(msg, MAX_STRING, "unknown column return variant type (%d)",
312: v.m_dwType);
313: services._throw(msg);
314: }
315: }
316:
317: void getFromString(SQL_Driver_services& services, CString& s, void *& ptr, size_t& size) {
318: if(s.IsEmpty()) {
319: ptr=0;
320: size=0;
321: } else {
322: const char *cstr=LPCTSTR(s);
323: size=strlen(cstr); //string.GetLength() works wrong with non-string types:
1.13.6.1! paf 324: ptr=services.malloc_atomic(size);
1.7 paf 325: memcpy(ptr, cstr, size);
326: }
1.1 parser 327: }
328:
329: void _throw(SQL_Driver_services& services, CException *e) {
330: char szCause[MAX_STRING]; szCause[0]=0;
331: e->GetErrorMessage(szCause, MAX_STRING);
332: char msg[MAX_STRING];
333: snprintf(msg, MAX_STRING, "%s: %s",
334: e->GetRuntimeClass()->m_lpszClassName,
335: *szCause?szCause:"unknown");
336: services._throw(msg);
337: }
338:
339: };
340:
341: extern "C" SQL_Driver *SQL_DRIVER_CREATE() {
342: return new ODBC_Driver();
343: }