dprevot/php-sqlsrv
1
0
Fork 0
Code Issues Pull requests Packages Projects Releases Wiki Activity

search for installed drivers in win

Browse source
This commit is contained in:
Jenny Tam 2017-09-22 16:37:05 -07:00
parent ca257fd2f4
commit 674d7c818b
1 changed files with 40 additions and 32 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

72
source/shared/core_conn.cpp
View file

@ -156,7 +156,7 @@ sqlsrv_conn* core_sqlsrv_connect( _In_ sqlsrv_context& henv_cp, _In_ sqlsrv_cont
r = core_odbc_connect( conn, conn_str, is_missing_driver, is_pooled ); r = core_odbc_connect( conn, conn_str, is_missing_driver, is_pooled );
} }
else if ( conn->ce_option.enabled ) { else if ( conn->ce_option.enabled ) {
#ifndef _WIN32 // #ifndef _WIN32
if( core_search_odbc_driver( DRIVER_VERSION::ODBC_DRIVER_17 ) ) { if( core_search_odbc_driver( DRIVER_VERSION::ODBC_DRIVER_17 ) ) {
conn_str = conn_str + CONNECTION_STRING_DRIVER_NAME[ DRIVER_VERSION::ODBC_DRIVER_17 ]; conn_str = conn_str + CONNECTION_STRING_DRIVER_NAME[ DRIVER_VERSION::ODBC_DRIVER_17 ];
r = core_odbc_connect( conn, conn_str, is_missing_driver, is_pooled ); r = core_odbc_connect( conn, conn_str, is_missing_driver, is_pooled );
@ -167,21 +167,23 @@ sqlsrv_conn* core_sqlsrv_connect( _In_ sqlsrv_context& henv_cp, _In_ sqlsrv_cont
throw core::CoreException(); throw core::CoreException();
} }
} }
#else // #else
conn_str = conn_str + CONNECTION_STRING_DRIVER_NAME[ DRIVER_VERSION::ODBC_DRIVER_17 ]; // conn_str = conn_str + CONNECTION_STRING_DRIVER_NAME[ DRIVER_VERSION::ODBC_DRIVER_17 ];
r = core_odbc_connect( conn, conn_str, is_missing_driver, is_pooled ); // r = core_odbc_connect( conn, conn_str, is_missing_driver, is_pooled );
CHECK_CUSTOM_ERROR( is_missing_driver, conn, SQLSRV_ERROR_AE_DRIVER_NOT_INSTALLED, get_processor_arch()) { // CHECK_CUSTOM_ERROR( is_missing_driver, conn, SQLSRV_ERROR_AE_DRIVER_NOT_INSTALLED, get_processor_arch()) {
throw core::CoreException(); // throw core::CoreException();
} // }
#endif // !_WIN32 // #endif // !_WIN32
} }
else { else {
#ifndef _WIN32 // #ifndef _WIN32
is_missing_driver = true; is_missing_driver = true;
for ( std::size_t i = DRIVER_VERSION::FIRST; i <= DRIVER_VERSION::LAST; ++i ) { for ( std::size_t i = DRIVER_VERSION::FIRST; i <= DRIVER_VERSION::LAST; ++i ) {
#ifndef _WIN32
if (i == DRIVER_VERSION::ODBC_DRIVER_11) // skip ODBC 11 in non-Windows environment if (i == DRIVER_VERSION::ODBC_DRIVER_11) // skip ODBC 11 in non-Windows environment
continue; continue;
#endif // !_WIN32
if( core_search_odbc_driver( i ) ) { if( core_search_odbc_driver( i ) ) {
std::string conn_str_driver = conn_str + CONNECTION_STRING_DRIVER_NAME[ DRIVER_VERSION( i ) ]; std::string conn_str_driver = conn_str + CONNECTION_STRING_DRIVER_NAME[ DRIVER_VERSION( i ) ];
@ -193,19 +195,19 @@ sqlsrv_conn* core_sqlsrv_connect( _In_ sqlsrv_context& henv_cp, _In_ sqlsrv_cont
CHECK_CUSTOM_ERROR( is_missing_driver, conn, SQLSRV_ERROR_DRIVER_NOT_INSTALLED, get_processor_arch()) { CHECK_CUSTOM_ERROR( is_missing_driver, conn, SQLSRV_ERROR_DRIVER_NOT_INSTALLED, get_processor_arch()) {
throw core::CoreException(); throw core::CoreException();
} }
#else // #else
for ( std::size_t i = DRIVER_VERSION::FIRST; i <= DRIVER_VERSION::LAST; ++i ) { // for ( std::size_t i = DRIVER_VERSION::FIRST; i <= DRIVER_VERSION::LAST; ++i ) {
is_missing_driver = false; // is_missing_driver = false;
std::string conn_str_driver = conn_str + CONNECTION_STRING_DRIVER_NAME[ DRIVER_VERSION(i) ]; // std::string conn_str_driver = conn_str + CONNECTION_STRING_DRIVER_NAME[ DRIVER_VERSION(i) ];
r = core_odbc_connect( conn, conn_str_driver, is_missing_driver, is_pooled ); // r = core_odbc_connect( conn, conn_str_driver, is_missing_driver, is_pooled );
CHECK_CUSTOM_ERROR( is_missing_driver && ( i == DRIVER_VERSION::LAST ), conn, SQLSRV_ERROR_DRIVER_NOT_INSTALLED, get_processor_arch()) { // CHECK_CUSTOM_ERROR( is_missing_driver && ( i == DRIVER_VERSION::LAST ), conn, SQLSRV_ERROR_DRIVER_NOT_INSTALLED, get_processor_arch()) {
throw core::CoreException(); // throw core::CoreException();
} // }
if ( !is_missing_driver) { // if ( !is_missing_driver) {
break; // break;
} // }
} // for // } // for
#endif // !_WIN32 // #endif // !_WIN32
} // else ce_option enabled } // else ce_option enabled
CHECK_SQL_ERROR( r, conn ) { CHECK_SQL_ERROR( r, conn ) {
@ -263,16 +265,23 @@ sqlsrv_conn* core_sqlsrv_connect( _In_ sqlsrv_context& henv_cp, _In_ sqlsrv_cont
return return_conn; return return_conn;
} }
// core_search_odbc_driver
// This method is meant to be used in a non-Windows environment,
// searching for a particular ODBC driver name in the odbcinst.ini file
// Parameters:
// driver_version - a valid value in enum DRIVER_VERSION
// Return - a boolean flag that indicates if the specified driver version is found or not
bool core_search_odbc_driver( _In_ std::size_t driver_version ) bool core_search_odbc_driver( _In_ std::size_t driver_version )
{ {
char szBuf[2001]; char szBuf[DEFAULT_CONN_STR_LEN+1]; // use an arbitrary large enough buffer size
WORD cbBufMax = 2000; WORD cbBufMax = DEFAULT_CONN_STR_LEN;
WORD cbBufOut; WORD cbBufOut;
char *pszBuf = szBuf; char *pszBuf = szBuf;
bool found_driver = false; bool found_driver = false;
#ifndef _WIN32 //#ifndef _WIN32
// Get the names of the installed drivers // get all the names of the installed drivers delimited by null characters
if(! SQLGetInstalledDrivers( szBuf, cbBufMax, &cbBufOut ) ) if(! SQLGetInstalledDrivers( szBuf, cbBufMax, &cbBufOut ) )
{ {
return false; return false;
@ -284,25 +293,24 @@ bool core_search_odbc_driver( _In_ std::size_t driver_version )
std::size_t pos2 = driver.find_first_of("}"); std::size_t pos2 = driver.find_first_of("}");
std::string driver_str = driver.substr( pos1 + 1, pos2 - pos1 - 1); std::string driver_str = driver.substr( pos1 + 1, pos2 - pos1 - 1);
// search for the ODBC driver...
const char* driver_name = driver_str.c_str(); const char* driver_name = driver_str.c_str();
//const char* driver_str = CONNECTION_STRING_DRIVER_NAME[ DRIVER_VERSION(driver_version) ].c_str();
// Search for the ODBC driver...
do do
{ {
if( strstr( pszBuf, driver_name ) != 0 ) if( strstr( pszBuf, driver_name ) != 0 )
{ {
// Found !
found_driver = true; found_driver = true;
break; break;
} }
// get the next driver
pszBuf = strchr( pszBuf, '\0' ) + 1; pszBuf = strchr( pszBuf, '\0' ) + 1;
} }
while( pszBuf[1] != '\0' ); while( pszBuf[1] != '\0' ); // end when there are two consecutive null characters
#endif // !_WIN32 //#endif // !_WIN32
return found_driver; return found_driver;
} }
// core_odbc_connect // core_odbc_connect
// calls odbc connect API to establish the connection to server // calls odbc connect API to establish the connection to server
// Parameters: // Parameters: