diff options
Diffstat (limited to 'dbaccess/source/ui/dlg/odbcconfig.cxx')
| -rw-r--r-- | dbaccess/source/ui/dlg/odbcconfig.cxx | 395 |
1 files changed, 395 insertions, 0 deletions
diff --git a/dbaccess/source/ui/dlg/odbcconfig.cxx b/dbaccess/source/ui/dlg/odbcconfig.cxx new file mode 100644 index 000000000000..3c04790c2f73 --- /dev/null +++ b/dbaccess/source/ui/dlg/odbcconfig.cxx @@ -0,0 +1,395 @@ +/* -*- Mode: C++; tab-width: 4; indent-tabs-mode: nil; c-basic-offset: 4 -*- */ +/************************************************************************* + * + * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER. + * + * Copyright 2000, 2010 Oracle and/or its affiliates. + * + * OpenOffice.org - a multi-platform office productivity suite + * + * This file is part of OpenOffice.org. + * + * OpenOffice.org is free software: you can redistribute it and/or modify + * it under the terms of the GNU Lesser General Public License version 3 + * only, as published by the Free Software Foundation. + * + * OpenOffice.org is distributed in the hope that it will be useful, + * but WITHOUT ANY WARRANTY; without even the implied warranty of + * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the + * GNU Lesser General Public License version 3 for more details + * (a copy is included in the LICENSE file that accompanied this code). + * + * You should have received a copy of the GNU Lesser General Public License + * version 3 along with OpenOffice.org. If not, see + * <http://www.openoffice.org/license.html> + * for a copy of the LGPLv3 License. + * + ************************************************************************/ + +// MARKER(update_precomp.py): autogen include statement, do not remove +#include "precompiled_dbaccess.hxx" + +#include "odbcconfig.hxx" + +#ifdef SYSTEM_ODBC_HEADERS +#include <sqltypes.h> +#else +#include <odbc/sqltypes.h> +#endif + +#include <rtl/bootstrap.hxx> +#include <rtl/ustring.hxx> +#include <rtl/ustrbuf.hxx> +#include <osl/diagnose.h> +#include <osl/process.h> +#include <osl/thread.hxx> +#include <tools/debug.hxx> +#include <vcl/svapp.hxx> + +#ifdef HAVE_ODBC_SUPPORT + +#if defined(OS2) +#define ODBC_LIBRARY "ODBC.DLL" +#define ODBC_UI_LIBRARY "ODBCINST.DLL" +#endif +#if defined WNT +#define ODBC_LIBRARY "ODBC32.DLL" +#define ODBC_UI_LIBRARY "ODBCCP32.DLL" +#endif +#ifdef UNX +#ifdef MACOSX +#define ODBC_LIBRARY "libiodbc.dylib" +#define ODBC_UI_LIBRARY "libiodbcinst.dylib" +#else +#define ODBC_LIBRARY_1 "libodbc.so.1" +#define ODBC_UI_LIBRARY_1 "libodbcinst.so.1" +#define ODBC_LIBRARY "libodbc.so" +#define ODBC_UI_LIBRARY "libodbcinst.so" +#endif +#endif + +// just to go with calling convention of windows +// so don't touch this +#if defined(WNT) +#define SQL_API __stdcall +// At least under some circumstances, the below #include <odbc/sqlext.h> re- +// defines SQL_API to an empty string, leading to a compiler warning on MSC; to +// not break the current behavior, this is worked around by locally disabling +// that warning: +#if defined _MSC_VER +#pragma warning(push) +#pragma warning(disable: 4005) +#endif +#endif // defined(WNT) + +#if defined(OS2) +#define ALLREADY_HAVE_OS2_TYPES +#define DONT_TD_VOID +#endif + +#ifdef SYSTEM_ODBC_HEADERS +#include <sqlext.h> +#else +#include <odbc/sqlext.h> +#endif + +#if defined(WNT) +#if defined _MSC_VER +#pragma warning(pop) +#endif +#undef SQL_API +#define SQL_API __stdcall +#endif // defined(WNT) +// from here on you can do what you want to + +#if defined(OS2) +#define SQL_API _System +#endif // defined(OS2) + +#else + +#define ODBC_LIBRARY "" +#define ODBC_UI_LIBRARY "" + +#endif // HAVE_ODBC_SUPPORT + +//......................................................................... +namespace dbaui +{ +//......................................................................... + + +#ifdef HAVE_ODBC_SUPPORT +typedef SQLRETURN (SQL_API* TSQLManageDataSource) (SQLHWND hwndParent); +typedef SQLRETURN (SQL_API* TSQLAllocHandle) (SQLSMALLINT HandleType, SQLHANDLE InputHandle, SQLHANDLE* OutputHandlePtr); +typedef SQLRETURN (SQL_API* TSQLFreeHandle) (SQLSMALLINT HandleType, SQLHANDLE Handle); +typedef SQLRETURN (SQL_API* TSQLSetEnvAttr) (SQLHENV EnvironmentHandle, SQLINTEGER Attribute, SQLPOINTER ValuePtr, SQLINTEGER StringLength); +typedef SQLRETURN (SQL_API* TSQLDataSources) (SQLHENV EnvironmentHandle, SQLUSMALLINT Direction, SQLCHAR* ServerName, + SQLSMALLINT BufferLength1, SQLSMALLINT* NameLength1Ptr, SQLCHAR* Description, SQLSMALLINT BufferLength2, SQLSMALLINT* NameLength2Ptr); + +#define NSQLManageDataSource(a) (*(TSQLManageDataSource)(m_pSQLManageDataSource))(a) +#define NSQLAllocHandle(a,b,c) (*(TSQLAllocHandle)(m_pAllocHandle))(a,b,c) +#define NSQLFreeHandle(a,b) (*(TSQLFreeHandle)(m_pFreeHandle))(a,b) +#define NSQLSetEnvAttr(a,b,c,d) (*(TSQLSetEnvAttr)(m_pSetEnvAttr))(a,b,c,d) +#define NSQLDataSources(a,b,c,d,e,f,g,h) (*(TSQLDataSources)(m_pDataSources))(a,b,c,d,e,f,g,h) +#endif + +//========================================================================= +//= OOdbcLibWrapper +//========================================================================= +DBG_NAME(OOdbcLibWrapper) +//------------------------------------------------------------------------- +#ifdef HAVE_ODBC_SUPPORT +OOdbcLibWrapper::OOdbcLibWrapper() + :m_pOdbcLib(NULL) +{ + DBG_CTOR(OOdbcLibWrapper,NULL); + +} +#endif + +//------------------------------------------------------------------------- +sal_Bool OOdbcLibWrapper::load(const sal_Char* _pLibPath) +{ + m_sLibPath = ::rtl::OUString::createFromAscii(_pLibPath); +#ifdef HAVE_ODBC_SUPPORT + // load the module + m_pOdbcLib = osl_loadModule(m_sLibPath.pData, SAL_LOADMODULE_NOW); + return (NULL != m_pOdbcLib); +#endif +} + +//------------------------------------------------------------------------- +void OOdbcLibWrapper::unload() +{ +#ifdef HAVE_ODBC_SUPPORT + if (isLoaded()) + { + osl_unloadModule(m_pOdbcLib); + m_pOdbcLib = NULL; + } +#endif +} + +//------------------------------------------------------------------------- +oslGenericFunction OOdbcLibWrapper::loadSymbol(const sal_Char* _pFunctionName) +{ + return osl_getFunctionSymbol(m_pOdbcLib, ::rtl::OUString::createFromAscii(_pFunctionName).pData); +} + +//------------------------------------------------------------------------- +OOdbcLibWrapper::~OOdbcLibWrapper() +{ + unload(); + + DBG_DTOR(OOdbcLibWrapper,NULL); +} + +//========================================================================= +//= OOdbcEnumeration +//========================================================================= +struct OdbcTypesImpl +{ +#ifdef HAVE_ODBC_SUPPORT + SQLHANDLE hEnvironment; + OdbcTypesImpl() : hEnvironment(0) { } +#else + void* pDummy; +#endif +}; +DBG_NAME(OOdbcEnumeration) +//------------------------------------------------------------------------- +OOdbcEnumeration::OOdbcEnumeration() +#ifdef HAVE_ODBC_SUPPORT + :m_pAllocHandle(NULL) + ,m_pSetEnvAttr(NULL) + ,m_pDataSources(NULL) + ,m_pImpl(new OdbcTypesImpl) +#endif +{ + DBG_CTOR(OOdbcEnumeration,NULL); + + sal_Bool bLoaded = load(ODBC_LIBRARY); +#ifdef ODBC_LIBRARY_1 + if ( !bLoaded ) + bLoaded = load(ODBC_LIBRARY_1); +#endif + + if ( bLoaded ) + { +#ifdef HAVE_ODBC_SUPPORT + // load the generic functions + m_pAllocHandle = loadSymbol("SQLAllocHandle"); + m_pFreeHandle = loadSymbol("SQLFreeHandle"); + m_pSetEnvAttr = loadSymbol("SQLSetEnvAttr"); + m_pDataSources = loadSymbol("SQLDataSources"); + + // all or nothing + if (!m_pAllocHandle || !m_pSetEnvAttr || !m_pDataSources || !m_pFreeHandle) + { + unload(); + m_pAllocHandle = m_pFreeHandle = m_pSetEnvAttr = m_pDataSources = NULL; + } +#endif + } +} + +//------------------------------------------------------------------------- +OOdbcEnumeration::~OOdbcEnumeration() +{ + freeEnv(); + delete m_pImpl; + + DBG_DTOR(OOdbcEnumeration,NULL); +} + +//------------------------------------------------------------------------- +sal_Bool OOdbcEnumeration::allocEnv() +{ + OSL_ENSURE(isLoaded(), "OOdbcEnumeration::allocEnv: not loaded!"); + if (!isLoaded()) + return sal_False; + +#ifdef HAVE_ODBC_SUPPORT + if (m_pImpl->hEnvironment) + // nothing to do + return sal_True; + SQLRETURN nResult = NSQLAllocHandle(SQL_HANDLE_ENV, SQL_NULL_HANDLE, &m_pImpl->hEnvironment); + if (SQL_SUCCESS != nResult) + // can't do anything without environment + return sal_False; + + NSQLSetEnvAttr(m_pImpl->hEnvironment, SQL_ATTR_ODBC_VERSION, (SQLPOINTER)SQL_OV_ODBC3, SQL_IS_INTEGER); + return sal_True; +#else + return sal_False; +#endif +} + +//------------------------------------------------------------------------- +void OOdbcEnumeration::freeEnv() +{ +#ifdef HAVE_ODBC_SUPPORT + if (m_pImpl->hEnvironment) + NSQLFreeHandle(SQL_HANDLE_ENV, m_pImpl->hEnvironment); + m_pImpl->hEnvironment = 0; +#endif +} + +//------------------------------------------------------------------------- +void OOdbcEnumeration::getDatasourceNames(StringBag& _rNames) +{ + OSL_ENSURE(isLoaded(), "OOdbcEnumeration::getDatasourceNames: not loaded!"); + if (!isLoaded()) + return; + + if (!allocEnv()) + { + OSL_FAIL("OOdbcEnumeration::getDatasourceNames: could not allocate an ODBC environment!"); + return; + } + +#ifdef HAVE_ODBC_SUPPORT + // now that we have an environment collect the data source names + UCHAR szDSN[SQL_MAX_DSN_LENGTH+1]; + SWORD pcbDSN; + UCHAR szDescription[1024+1]; + SWORD pcbDescription; + SQLRETURN nResult = SQL_SUCCESS; + rtl_TextEncoding nTextEncoding = osl_getThreadTextEncoding(); + + for ( nResult = NSQLDataSources(m_pImpl->hEnvironment, SQL_FETCH_FIRST, szDSN, sizeof(szDSN), &pcbDSN, szDescription, sizeof(szDescription)-1, &pcbDescription); + ; + nResult = NSQLDataSources(m_pImpl->hEnvironment, SQL_FETCH_NEXT, szDSN, sizeof(szDSN), &pcbDSN, szDescription, sizeof(szDescription)-1, &pcbDescription) + ) + { + if (nResult != SQL_SUCCESS) + // no further error handling + break; + else + { + ::rtl::OUString aCurrentDsn(reinterpret_cast<const char*>(szDSN),pcbDSN, nTextEncoding); + _rNames.insert(aCurrentDsn); + } + } +#endif +} + +#ifdef HAVE_ODBC_ADMINISTRATION + +//========================================================================= +//= ProcessTerminationWait +//========================================================================= +class ProcessTerminationWait : public ::osl::Thread +{ + oslProcess m_hProcessHandle; + Link m_aFinishHdl; + +public: + ProcessTerminationWait( oslProcess _hProcessHandle, const Link& _rFinishHdl ) + :m_hProcessHandle( _hProcessHandle ) + ,m_aFinishHdl( _rFinishHdl ) + { + } + +protected: + virtual void SAL_CALL run() + { + osl_joinProcess( m_hProcessHandle ); + osl_freeProcessHandle( m_hProcessHandle ); + Application::PostUserEvent( m_aFinishHdl ); + } +}; + +//========================================================================= +//= OOdbcManagement +//========================================================================= +//------------------------------------------------------------------------- +OOdbcManagement::OOdbcManagement( const Link& _rAsyncFinishCallback ) + :m_pProcessWait( NULL ) + ,m_aAsyncFinishCallback( _rAsyncFinishCallback ) +{ +} + +//------------------------------------------------------------------------- +OOdbcManagement::~OOdbcManagement() +{ + // wait for our thread to be finished + if ( m_pProcessWait.get() ) + m_pProcessWait->join(); +} + +//------------------------------------------------------------------------- +bool OOdbcManagement::manageDataSources_async() +{ + OSL_PRECOND( !isRunning(), "OOdbcManagement::manageDataSources_async: still running from the previous call!" ); + if ( isRunning() ) + return false; + + // this is done in an external process, due to #i78733# + // (and note this whole functionality is supported on Windows only, ATM) + ::rtl::OUString sExecutableName( RTL_CONSTASCII_USTRINGPARAM( "$OOO_BASE_DIR/program/odbcconfig.exe" ) ); + ::rtl::Bootstrap::expandMacros( sExecutableName ); //TODO: detect failure + oslProcess hProcessHandle(0); + oslProcessError eError = osl_executeProcess( sExecutableName.pData, NULL, 0, 0, NULL, NULL, NULL, 0, &hProcessHandle ); + if ( eError != osl_Process_E_None ) + return false; + + m_pProcessWait.reset( new ProcessTerminationWait( hProcessHandle, m_aAsyncFinishCallback ) ); + m_pProcessWait->create(); + return true; +} + +//------------------------------------------------------------------------- +bool OOdbcManagement::isRunning() const +{ + return ( m_pProcessWait.get() && m_pProcessWait->isRunning() ); +} + +#endif // HAVE_ODBC_ADMINISTRATION + +//......................................................................... +} // namespace dbaui +//......................................................................... + +/* vim:set shiftwidth=4 softtabstop=4 expandtab: */ |