IBConnection.cpp

Go to the documentation of this file.

#include <dcl/Config.h>
 
#include <stdlib.h>     // strtol
#include <string.h>     // strncpy
 
#include <ibase.h>
 
#include <dcl/Object.h>
#if __DCL_HAVE_ALLOC_DEBUG
#undef __DCL_ALLOC_LEVEL
#define __DCL_ALLOC_LEVEL   __DCL_ALLOC_INTERNAL
#endif
 
#include <dcl/Numeric.h>
#include <dcl/Regex.h>
#include <dcl/SQLCore.h>
 
#include "IBConnection.h"
#include "IBQuery.h"
 
#define __TRACE_THIS    0
#if __TRACE_THIS
#define __DCL_TRACE0_N  __DCL_TRACE0
#define __DCL_TRACE1_N  __DCL_TRACE1
#define __DCL_TRACE2_N  __DCL_TRACE2
#define __DCL_TRACE3_N  __DCL_TRACE3
#define __DCL_TRACE4_N  __DCL_TRACE4
#else
#define __DCL_TRACE0_N(fmt)
#define __DCL_TRACE1_N(fmt, arg)
#define __DCL_TRACE2_N(fmt, arg1, arg2)
#define __DCL_TRACE3_N(fmt, arg1, arg2, arg3)
#define __DCL_TRACE4_N(fmt, arg1, arg2, arg3, arg4)
#endif
 
#undef __THIS_FILE__
static const char_t __THIS_FILE__[] = __T("dcl/sql/IBConnection.cpp");
 
__DCL_BEGIN_NAMESPACE
 
#define __SET_ERROR(_error) \
    setErrorHandle(_error, __THIS_FILE__, __LINE__)
#define __SET_ERROR_MSG(_msg) \
    setErrorMessage(_msg, __THIS_FILE__, __LINE__)
 
IMPLEMENT_CLASSINFO(IBConnection, SQL::Connection)
 
IBConnection::IBConnection(const wchar_t* _serverTitle)
    : Connection(_serverTitle)
{
    Connection::__canTransact = true;
 
    __dialect = SQL_DIALECT_CURRENT;    // default ibase 6
    __dbHandle = NULL_HANDLE;
    __trHandle = NULL_HANDLE;
//  memset(__status, 0, sizeof(ISC_STATUS) * ISC_STATUS_VECTOR_LENGTH);
}
 
IBConnection::~IBConnection()
{
    if (__dbHandle) {
        __DCL_TRACE0_N(L"Warning!! The connection was not closed\n");
        close();
    }
}
 
void IBConnection::destroy()
{
    delete this;
}
 
/*
_conns에 가능한 property
    USER,
    PASSWORD,
    SERVER, -- host name
    DATABASE
    SQL_DIALECT
 
    or CREATE DATABASE statement
*/
 
static const char* __createDatabasePattern__ = "CREATE[[:space:]]+DATABASE";
 
bool IBConnection::__open(const char* _conns, size_t _connlen)
{
    try  {
        if (Regex::test(__createDatabasePattern__, _conns, true)) {
            // CREATE DATABASE
            if(isc_dsql_execute_immediate(
                __status,
                &__dbHandle,
                &__trHandle,
                0,
                (_CONST char*)_conns,
                __dialect,
                NULL
            )) {
                __SET_ERROR(SQL::eServerError);
                return false;
            }
            return true;
        }
    }
    catch(RegexException* e) {
        e->destroy();
    }
 
    ListedByteStringToByteStringMap map;
    Connection::splitConnStr(_conns, _connlen, map);
 
    ByteString _SERVER = map["SERVER"];
    ByteString _DATABASE = map["DATABASE"];
    ByteString _USER = map["USER"];
    ByteString _PASSWORD = map["PASSWORD"];
    ByteString _SQL_DIALECT = map["SQL_DIALECT"];
 
    if (!_SQL_DIALECT.isEmpty()) {
        int nDialect = SQL_DIALECT_CURRENT;
        char* endptr;
        long n = strtol(_SQL_DIALECT, &endptr, 10);
        if (n == LONG_MAX || n == LONG_MIN
            || n < INT_MIN || INT_MAX < n
        ) {
            __SET_ERROR_MSG("연결 문자열이 잘못되었습니다. SQL_DIALECT="
                + _SQL_DIALECT);
            return false;
        }
 
        nDialect = (int)n;
 
        if (!(nDialect == SQL_DIALECT_V5
            || nDialect == SQL_DIALECT_V6_TRANSITION
            || nDialect == SQL_DIALECT_V6)
        ) {
            __SET_ERROR_MSG("\"SQL_DIALECT\"- invalid value");
            return false;
        }
 
        __dialect = nDialect;
    }
 
    if ((_USER.length() > 127) || (_PASSWORD.length() > 127)) {
        // USER NAME 과 PASSWORD 의 길이는 127을 넘을 수 없다.
        __SET_ERROR(SQL::eInvalidConnectionString);
        return false;
    }
 
    ByteString dbEnv;
    if (!_SERVER.isEmpty())
        dbEnv = _SERVER + ":";
 
    if (!_DATABASE.isEmpty())
        dbEnv = dbEnv + _DATABASE;
 
    if (dbEnv.isEmpty()) {
        __SET_ERROR(SQL::eInvalidConnectionString);
        return false;
    }
 
    // database parameter block
    ByteStringBuilder dpb;
    dpb = (char)isc_dpb_version1;
    dpb += (char)isc_dpb_user_name;
    dpb += (char)_USER.length();
    dpb += _USER;
    dpb += (char)isc_dpb_password;
    dpb += (char)_PASSWORD.length();
    dpb += _PASSWORD;
 
    static const char __UTF8__[] = "UTF8";
    dpb += (char)isc_dpb_lc_ctype;
    dpb += (char)(sizeof(__UTF8__) - 1);
    dpb += __UTF8__;
 
#if defined(FIREBIRD_IBASE_H) && FB_API_VER >= 40
    static const char __SET_BIND__[] =
        "DECFLOAT TO VARCHAR;"      // 20
        "INT128 TO VARCHAR;"        // 18
        "TIME ZONE TO EXTENDED";    // 21
    dpb += (char)isc_dpb_set_bind;
    dpb += (char)(sizeof(__SET_BIND__) - 1);
    dpb += __SET_BIND__;
#endif
 
#ifdef FIREBIRD_IBASE_H
    #undef _CONST
    #define _CONST const
#endif
 
    if (isc_attach_database(
        __status,
        (short) dbEnv.length(),
        (_CONST char*) dbEnv.data(),
        &__dbHandle,
        (short) dpb.length(),
        (_CONST char*) dpb.data())
    ) {
//      __DCL_ASSERT(STATUS_FAILED(__status));
        __SET_ERROR(SQL::eServerError);
        __dbHandle = NULL_HANDLE;
        return false;
    }
    return true;
}
 
bool IBConnection::__close()
{
    if (!__dbHandle) {
        __SET_ERROR(SQL::eNotConnected);
        return false;
    }
    __DCL_ASSERT(__trHandle == NULL_HANDLE);
 
    if (isc_detach_database(
        __status,
        &__dbHandle
    )) {
//      __DCL_ASSERT(STATUS_FAILED(__status));
        __SET_ERROR(SQL::eServerError);
        return false;
    }
    __dbHandle = NULL_HANDLE;
    return true;
}
 
enum StmtType {
    StmtOther,
    StmtTransBegin,
    StmtTransEnd,
    StmtCreateDatabase
};
 
typedef struct {
    StmtType type;
    const char* pattern;
} STMT_PATTERN;
 
static STMT_PATTERN sp[] = {
    { StmtTransBegin,     "SET[[:space:]]+TRANSACTION"    },
    { StmtTransEnd,         "COMMIT|ROLLBACK"               },
    { StmtCreateDatabase,  __createDatabasePattern__      },
    { StmtOther,           NULL                            }
};
 
static StmtType __GetStmtType(const char* _sql)
{
    for(size_t i = 0; sp[i].type != StmtOther; i++) {
        if (Regex::test(sp[i].pattern, _sql, true))
            return sp[i].type;
    }
    return StmtOther;
}
 
bool IBConnection::__execute(const char* _sql, size_t n)
{
    __DCL_ASSERT(n <= UINT16_MAX);
    __DCL_ASSERT(__dbHandle != NULL_HANDLE);
    if(isc_dsql_execute_immediate(
        __status,
        &__dbHandle,
        &__trHandle,
        (unsigned short int)n,
        (_CONST char*)_sql,
        __dialect,
        NULL
    )) {
//      __DCL_ASSERT(STATUS_FAILED(__status));
        __SET_ERROR(SQL::eServerError);
        return false;
    }
 
    switch (__GetStmtType(_sql)) {
        case StmtTransBegin :
            __SET_STATE(stInTransaction);
            __DCL_ASSERT(__trHandle != NULL_HANDLE);
            break;
        case StmtTransEnd :
            __UNSET_STATE(stInTransaction);
            __trHandle = NULL_HANDLE;
            break;
        default :
            ;
    }
 
    return true;
}
 
bool IBConnection::__startTrans()
{
    __DCL_ASSERT(__trHandle == NULL_HANDLE);
    if (isc_start_transaction(
        __status, 
        &__trHandle, 
        1, 
        &__dbHandle, 
        0, 
        NULL
    )) {
//      __DCL_ASSERT(STATUS_FAILED(__status));
        __SET_ERROR(SQL::eServerError);
        return false;
    }
 
    return true;
}
 
bool IBConnection::__commitTrans()
{
    __DCL_ASSERT(__trHandle != NULL_HANDLE);
    if(isc_commit_transaction(
        __status, 
        &__trHandle
    )) {
//      __DCL_ASSERT(STATUS_FAILED(__status));
        __SET_ERROR(SQL::eServerError);
        return false;
    }
    __trHandle = NULL_HANDLE;
 
    return true;
}
 
bool IBConnection::__rollbackTrans()
{
    __DCL_ASSERT(__trHandle != NULL_HANDLE);
    if(isc_rollback_transaction(
        __status, 
        &__trHandle
    )) {
//      __DCL_ASSERT(STATUS_FAILED(__status));
        __SET_ERROR(SQL::eServerError);
        return false;
    }
    __trHandle = NULL_HANDLE;
 
    return true;
}
 
bool IBConnection::__createQueryInstance(SQL::Query** _queryHandleOut)
{
    __DCL_ASSERT(_queryHandleOut != NULL);
 
    SQL::Query* newQuery = new IBQuery(this);
    if (!newQuery) {
        __SET_ERROR(SQL::eOutOfMemory);
        return false;
    }
 
    *_queryHandleOut = newQuery;
    return true;
}
 
bool IBConnection::__getErrorMessage(char* _buf, size_t* _buflen)
{
    ByteStringBuilder sb;
    if (!__lastErrorMessage.isEmpty())
        sb = __lastErrorMessage;
    else {
        char buf[512];
        ISC_STATUS* status = __status;
        size_t len = 0;
#ifdef FIREBIRD_IBASE_H
#define isc_interprete(buf, pp) fb_interpret(buf, sizeof(buf), (const ISC_STATUS**) pp)
#endif
        for(size_t i = 0; (len = isc_interprete(buf, &status))
                && (i < ISC_STATUS_VECTOR_LENGTH); i++) {
            if (i > 0)
                sb += ", ";
            sb.append(buf, 0, len);
        }
    }
 
    if (sb.length() < *_buflen) {
        *_buflen = sb.length();
        *(_buf + *_buflen) = '\0';
    }
    strncpy(_buf, sb.data(), *_buflen);
    return true;
}
 
static char database_info_items[] = {
    isc_info_version,
    //isc_info_db_id,
    isc_info_end
};
 
bool IBConnection::__getServerInfo(char* _buf, size_t* _buflen)
{
    char buffer[512];
    if (isc_database_info (
        __status,
        &__dbHandle,
        sizeof(database_info_items),
        database_info_items,
        sizeof(buffer),
        buffer
    )) {
        __SET_ERROR(SQL::eServerError);
        return false;
    }
 
    ByteString version;
    ByteString dbId;
    ByteString site;
 
    char* p = buffer;
    char item;
    unsigned short length;
    __DCL_ASSERT(sizeof(unsigned short) == 2);
    while(*p != isc_info_end) {
        item = *p++;
        length = (unsigned short)isc_vax_integer(p, 2);
        p += 2;
        if (item == isc_info_version) {
        //  __DCL_ASSERT(p[0] == 1);
            //__DCL_TRACE1(L"[%d]\n", p[0]);
            version.assign(p, 2, p[1]);
        }
        else if (item == isc_info_db_id) {
            char* t = p;
            //__DCL_TRACE1(L"[%d]\n", *t);
            //__DCL_ASSERT(*t == 2);
            t++;
            unsigned short strlen = *t++;
            dbId.assign(t, strlen);
            t += strlen;
            strlen = *t++;
            site.assign(t, strlen);
        }
        else {
            __SET_ERROR(SQL::eNotAvailable);
            return false;
        }
        p += length;
    }
 
    ByteString info =
#ifndef FIREBIRD_IBASE_H
        "InterBase " + version
#else
        version
#endif
    // +" \"" + site + ':' + dbId + "\""
    ;
 
    if (info.length() < *_buflen)
        *_buflen = info.length();
 
    strncpy(_buf, info.data(), *_buflen);
    return true;
}
 
__DCL_END_NAMESPACE