--- sql/odbc/parser3odbc.C 2002/03/05 15:04:59 1.6 +++ sql/odbc/parser3odbc.C 2003/11/10 08:44:53 1.16 @@ -1,11 +1,11 @@ /** @file Parser ODBC driver. - Copyright(c) 2001, 2002 ArtLebedev Group (http://www.artlebedev.com) + Copyright(c) 2001, 2003 ArtLebedev Group (http://www.artlebedev.com) Author: Alexandr Petrosian (http://paf.design.ru) */ -static const char *RCSId="$Id: parser3odbc.C,v 1.6 2002/03/05 15:04:59 paf Exp $"; +static const char *RCSId="$Id: parser3odbc.C,v 1.16 2003/11/10 08:44:53 paf Exp $"; #ifndef _MSC_VER # error compile ISAPI module with MSVC [no urge for now to make it autoconf-ed (PAF)] @@ -18,9 +18,27 @@ static const char *RCSId="$Id: parser3od #include "pa_sql_driver.h" +#define WINVER 0x0400 #include +// defines + +#define MAX_COLS 500 + #define MAX_STRING 0x400 +#define MAX_NUMBER 40 + +// new in MSSQL2000, no MFC constants +#ifndef SQL_NVARCHAR +#define SQL_NVARCHAR (-9) +#endif +#ifndef SQL_NTEXT +#define SQL_NTEXT (-10) +#endif +#ifndef SQL_SMALLDATETIME +#define SQL_SMALLDATETIME 11 +#endif +// create table test (id int, a smalldatetime, b ntext, c nvarchar(100)) #define snprintf _snprintf #ifndef strncasecmp @@ -109,20 +127,19 @@ public: return true; } - unsigned int quote( - SQL_Driver_services&, void *connection, - char *to, const char *from, unsigned int length) { - if(to) { // store mode - unsigned int result=length; - while(length--) { - if(*from=='\'') { // ' -> '' - *to++='\''; result++; - } - *to++=*from++; + const char* quote( + SQL_Driver_services& services, void *connection, + const char *from, unsigned int length) { + char *result=(char*)services.malloc_atomic(length*2+1); + char *to=result; + while(length--) { + if(*from=='\'') { // ' -> '' + *to++='\''; } - return result; - } else // estimate mode - return length*2; + *to++=*from++; + } + *to=0; + return result; } void query( SQL_Driver_services& services, void *connection, @@ -135,52 +152,91 @@ public: statement++; TRY { - if(strncasecmp(statement, "select", 6)==0) { + // mk:@MSITStore:C:\Program%20Files\Microsoft%20SQL%20Server\80\Tools\Books\adosql.chm::/adoprg02_4g33.htm + // Server cursors are created only for statements that begin with: + // SELECT + // EXEC[ute] procedure_name + // call procedure_name + // mk:@MSITStore:C:\Program%20Files\Microsoft%20SQL%20Server\80\Tools\Books\odbcsql.chm::/od_6_035_5dnp.htm + // The ODBC CALL escape sequence for calling a procedure is: + // {[?=]call procedure_name[([parameter][,[parameter]]...)]} + if(strncasecmp(statement, "select", 6)==0 + || strncasecmp(statement, "EXEC", 4)==0 + || strncasecmp(statement, "call", 4)==0 + || strncasecmp(statement, "{", 1)==0) { CRecordset rs(db); - rs.Open( - CRecordset::forwardOnly, - statement, - CRecordset::executeDirect - ); + TRY { + rs.Open( + CRecordset::forwardOnly, + statement, + CRecordset::executeDirect + ); + } CATCH_ALL (e) { + // could not fetch a table + TRY { + // then try resultless query + db->ExecuteSQL(statement); + // OK then + return; + } CATCH_ALL (e2) { + // still nothing good + _throw(services, e); // throw ORIGINAL exception + } END_CATCH_ALL + } END_CATCH_ALL int column_count=rs.GetODBCFieldCount(); if(!column_count) services._throw("result contains no columns"); + SWORD column_types[MAX_COLS]; + if(column_count>MAX_COLS) + column_count=MAX_COLS; + + SQL_Error sql_error; +#define CHECK(afailed) if(afailed) services._throw(sql_error) + for(int i=0; i=offset) { - handlers.add_row(); - for(int i=0; iExecuteSQL(statement); } - } - CATCH_ALL (e) { + } CATCH_ALL (e) { _throw(services, e); + } END_CATCH_ALL + } + + void getFromDBVariant(SQL_Driver_services& services, CDBVariant& v, char*& str, size_t& size) { + switch(v.m_dwType) { + case DBVT_BINARY: /* << would cause problems with current String implementation + now falling into NULL case, effectively ignoring such columns [not failing] + { + if(size=v.m_pbinary->m_dwDataLength) { + str=services.malloc_atomic(size+1); + memcpy(ptr, ::GlobalLock(v.m_pbinary->m_hData), size); + ::GlobalUnlock(v.m_pbinary->m_hData); + } else + str=0; + break; + }*/ + case DBVT_NULL: // No union member is valid for access. + str=0; + size=0; + break; +/* case DBVT_BOOL: + ptr=v.m_boolVal?"1":"0"; + size=1; + break;*/ +/* case DBVT_UCHAR: + size=strlen(ptr=v.m_chVal); + break; + case DBVT_SHORT: + char buf[MAX_NUMBER]; + size=snprintf(HEAPIZE buf, "%d", v.m_iVal); + break;*/ +/* case DBVT_LONG: + { + char local_buf[MAX_NUMBER]; + size=snprintf(local_buf, MAX_NUMBER, "%ld", v.m_lVal); + ptr=services.malloc_atomic(size); + memcpy(ptr, local_buf, size); + break; + }*/ + /*case DBVT_SINGLE: + m_fltVal + break; + case DBVT_DOUBLE m_dblVal + case DBVT_STRING m_pstring */ + case DBVT_DATE: + { + char local_buf[MAX_STRING]; + size=snprintf(local_buf, MAX_STRING, + "%04d-%02d-%02d %02d:%02d:%02d.%03d", + v.m_pdate->year, + v.m_pdate->month, + v.m_pdate->day, + v.m_pdate->hour, + v.m_pdate->minute, + v.m_pdate->second, + v.m_pdate->fraction); + str=(char*)services.malloc_atomic(size+1); + memcpy(str, local_buf, size+1); + break; + } + default: + char msg[MAX_STRING]; + snprintf(msg, MAX_STRING, "unknown column return variant type (%d)", + v.m_dwType); + services._throw(msg); + } + } + + void getFromString(SQL_Driver_services& services, CString& s, char*& astr, size_t& size) { + if(s.IsEmpty()) { + astr=0; + size=0; + } else { + const char *cstr=LPCTSTR(s); + size=strlen(cstr); //string.GetLength() works wrong with non-string types: + astr=(char*)services.malloc_atomic(size+1); + memcpy(astr, cstr, size+1); } - END_CATCH_ALL } void _throw(SQL_Driver_services& services, CException *e) {