//+--------------------------------------------------------------------------
//
// Microsoft Windows
// Copyright (C) Microsoft Corporation, 1996-1998
//
// File:
//
// Contents:    
//
// History:     
//---------------------------------------------------------------------------
#include <windows.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <tchar.h>
#include <odbcinst.h>

#define STRSAFE_NO_DEPRECATE
#include <strsafe.h>

#define szSQLSetConfigMode              "SQLSetConfigMode"

//---------------------------------------------------------------------------
// various installation routine we use, can't link with library as it will
// cause 
#ifdef  UNICODE
#define  szSQLInstallDriver             "SQLInstallDriverW"
#define  szSQLGetInstalledDrivers       "SQLGetInstalledDriversW"
#define  szSQLConfigDataSource          "SQLConfigDataSourceW"
#define  szSQLGetPrivateProfileString   "SQLGetPrivateProfileStringW"
#define  szSQLConfigDriver              "SQLConfigDriverW"
#define  szSQLInstallerError            "SQLInstallerErrorW"
#else
#define  szSQLInstallDriver             "SQLInstallDriver"
#define  szSQLGetInstalledDrivers       "SQLGetInstalledDrivers"
#define  szSQLConfigDataSource          "SQLConfigDataSource"
#define  szSQLGetPrivateProfileString   "SQLGetPrivateProfileString"
#define  szSQLConfigDriver              "SQLConfigDriver"
#define  szSQLInstallerError            "SQLInstallerError"
#endif

//
// ODBC install functions - To prevent system to 'force' odbccp32.dll
// odbccp32 may not present on the system nor it is available at the 
// time setup got kick off.
//
typedef BOOL (* SQLCONFIGDATASOURCE)(HWND, WORD, LPCTSTR, LPCTSTR);

typedef SQLRETURN (* SQLINSTALLERERROR)(WORD , DWORD*, LPTSTR, WORD, WORD*);

typedef int (* SQLGETPRIVATEPROFILESTRING)( LPCTSTR lpszSection,
                                            LPCTSTR lpszEntry,
                                            LPCTSTR lpszDefault,
                                            LPTSTR  lpszRetBuffer,
                                            int    cbRetBuffer,
                                            LPCTSTR lpszFilename);

typedef BOOL (* SQLINSTALLDRIVER)(LPCTSTR  lpszInfFile,
                                  LPCTSTR  lpszDriver,
                                  LPTSTR   lpszPath,
                                  WORD     cbPathMax,
                                  WORD*    pcbPathOut);

typedef BOOL (* SQLGETINSTALLEDDRIVER)(LPTSTR  lpszBuf,
                                       WORD    cbBufMax,
                                       WORD*   pcbBufOut);

typedef BOOL (* SQLSETCONFIGMODE)(UWORD wConfigMode);

static SQLCONFIGDATASOURCE          fpSQLConfigDataSource=NULL;
static SQLINSTALLDRIVER             fpSQLInstallDriver=NULL;
static SQLINSTALLERERROR            fpSQLInstallerError=NULL;
static SQLGETPRIVATEPROFILESTRING   fpSQLGetPrivateProfileString=NULL;
static SQLSETCONFIGMODE             fpSQLSetConfigMode=NULL;
static SQLGETINSTALLEDDRIVER        fpSQLGetInstalledDriver=NULL;
static HINSTANCE                    hODBCCP32=NULL;

//------------------------------------------------------------

void
ReportError(
    IN HWND hWnd, 
    IN LPTSTR pszDefaultMsg
    )
/*++

Abstract:

    Popup a error message.

Parameters:

    hWnd - Parent window handle.
    pszDefaultMsg - ignore.

Returns:

    None.
    
++*/
{
#if DBG
    DWORD dwErrCode;
    TCHAR szErrMsg[2048];
    WORD  szErrMsgSize=sizeof(szErrMsg)/sizeof(szErrMsg[0]);
    WORD  cbErrMsg;

    fpSQLInstallerError(
                    1, 
                    &dwErrCode, 
                    szErrMsg, 
                    szErrMsgSize, 
                    &cbErrMsg
                );

    MessageBox(hWnd, szErrMsg, _TEXT("Setup Error"), MB_OK);
#else
    return;
#endif
}


//------------------------------------------------------------

BOOL 
InitODBCSetup()
/*++


++*/
{
    hODBCCP32 = LoadLibrary(_TEXT("odbccp32"));
    if(hODBCCP32)
    {
        fpSQLConfigDataSource=(SQLCONFIGDATASOURCE)GetProcAddress(
                                                            hODBCCP32, 
                                                            szSQLConfigDataSource
                                                        );

        fpSQLInstallDriver=(SQLINSTALLDRIVER)GetProcAddress(
                                                            hODBCCP32, 
                                                            szSQLInstallDriver
                                                        );

        fpSQLInstallerError=(SQLINSTALLERERROR)GetProcAddress(
                                                            hODBCCP32, 
                                                            szSQLInstallerError
                                                        );

        fpSQLGetPrivateProfileString=(SQLGETPRIVATEPROFILESTRING)GetProcAddress(
                                                                        hODBCCP32, 
                                                                        szSQLGetPrivateProfileString
                                                                    );

        fpSQLSetConfigMode=(SQLSETCONFIGMODE)GetProcAddress(
                                                        hODBCCP32, 
                                                        szSQLSetConfigMode
                                                    );

        fpSQLGetInstalledDriver=(SQLGETINSTALLEDDRIVER)GetProcAddress(
                                                                hODBCCP32, 
                                                                szSQLGetInstalledDrivers
                                                            );
    }

    if( hODBCCP32 == NULL || fpSQLConfigDataSource == NULL || 
        fpSQLInstallDriver == NULL || fpSQLInstallerError == NULL || 
        fpSQLGetPrivateProfileString == NULL || fpSQLSetConfigMode == NULL ||
        fpSQLGetInstalledDriver == NULL)
    {
        ReportError(NULL, _TEXT("Can't load odbccp32.dll "));
        return FALSE;
    } 

    return TRUE;
}

//------------------------------------------------------------

void 
CleanupODBCSetup()
/*++

++*/
{
    if(hODBCCP32)
        FreeLibrary(hODBCCP32);

    fpSQLConfigDataSource=NULL;
    fpSQLInstallDriver=NULL;
    fpSQLInstallerError=NULL;
    fpSQLGetPrivateProfileString=NULL;
    fpSQLSetConfigMode=NULL;
    fpSQLGetInstalledDriver=NULL;
    hODBCCP32=NULL;

    return;
}

//---------------------------------------------------------------------------
//
// Access Driver installation
//
LPTSTR szAccessDriver=_TEXT("Microsoft Access Driver (*.mdb)\0")
                      _TEXT("Driver=odbcjt32.dll\0")
                      _TEXT("Setup=odbcjt32.dll\0")
                      _TEXT("Name=Microsoft Access Driver (*.mdb)\0")
                      _TEXT("APILevel=1\0")
                      _TEXT("ConnectFunctions=YYN\0")
                      _TEXT("DriverODBCVer=02.50\0")
                      //_TEXT("FileUsage=2\0")
                      _TEXT("FileExtns=*.mdb\0")
                      _TEXT("SQLLevel=0\0");


//---------------------------------------------------------------------------

BOOL 
IsDriverInstalled( 
    IN LPTSTR szDriverName 
    )
/*++

Abstract:

    Check if a ODBC driver installed on system.

Parameters:

    szDriveName - Name of the drive.

Returns:

    TRUE if driver installed, FALSE otherwise.

++*/
{
    TCHAR szBuf[8096];  // this got to be enough
    WORD cbBufMax=sizeof(szBuf)/sizeof(szBuf[0]);
    WORD cbBufOut;
    LPTSTR pszBuf=szBuf;

    if(hODBCCP32 == NULL && !InitODBCSetup())
        return FALSE;

    if(fpSQLGetInstalledDriver(szBuf, cbBufMax, &cbBufOut))
    {
        ReportError(NULL, _TEXT("SQLGetInstalledDrivers"));
    }
    else
    {
        do {
            if(_tcsnicmp(szDriverName, pszBuf, min(lstrlen(szDriverName), lstrlen(pszBuf))) == 0)
                break;

            pszBuf += lstrlen(pszBuf) + 1;
        } while(pszBuf[1] != _TEXT('\0'));
    }

    return (pszBuf[1] != _TEXT('\0'));
}
    
//---------------------------------------------------------------------------

BOOL 
IsDataSourceInstalled( 
    IN LPTSTR pszDataSource, 
    IN UWORD wConfigMode, 
    IN OUT LPTSTR pszDbFile, 
    IN DWORD cbBufSize 
    )

/*++

Abstract:

    Check if a ODBC datasource are installed.

Parameters:

    pszDataSource - name of data source.
    wConfigMode - configuration mode, refer to ODBC for detail, 
                  license server uses ODBC_SYSTEM_DSN.
    pszDbFile - Pointer to buffer for receving full path to database file 
                if data source is installed.
    cbBufSize - size of buffer in characters.

    
Returns:

    TRUE if datasource is installed, FALSE otherwise.

++*/

{
    BOOL bSuccess = TRUE;

    if(hODBCCP32 == NULL && !InitODBCSetup())
    {
        bSuccess = FALSE;
        goto cleanup;
    }

    if(fpSQLSetConfigMode(wConfigMode) == FALSE)
    {
        ReportError(NULL, _TEXT("SQLSetConfigMode failed"));
        bSuccess = FALSE;
        goto cleanup;
    }

    if(fpSQLGetPrivateProfileString(
                    pszDataSource, 
                    _TEXT("DBQ"), 
                    _TEXT(""), 
                    pszDbFile, 
                    cbBufSize, 
                    _TEXT("ODBC.INI")
                ) == 0)
    {
        bSuccess = FALSE;
    }

cleanup:

    return bSuccess;
}

//---------------------------------------------------------------------------

BOOL
ConfigDataSource( 
    HWND     hWnd, 
    BOOL     bInstall,       // TRUE to install FALSE to remove
    LPTSTR   pszDriver,       // driver
    LPTSTR   pszDsn,          // DSN
    LPTSTR   pszUser,         // User
    LPTSTR   pszPwd,          // Password
    LPTSTR   pszMdbFile       // MDB file
    )

/*++

Abstract:

    Routine to add/remove ODBC data source.

Parameters:

    hWnd - Parent window handle.
    bInstall - TRUE if installing ODBC data source, FALSE otherwise.
    pszDrive - Name of the ODBC drive to be used on data source.
    pszDsn - ODBC Data Source Name.
    pszUser - Login use name.
    pszPwd - Login password.
    pszMdbFile - Name of the Database file.
    
Returns:

    TRUE if successfule, FALSE otherwise.

++*/

{
    TCHAR   szAttributes[MAX_PATH*6+1];
    BOOL    bConfig=TRUE;
    TCHAR   *pszAttributesEnd;
    HRESULT hr;
    size_t  cbRemaining;

    if(hODBCCP32 == NULL && !InitODBCSetup())
        return FALSE;

    //
    // for attribute string
    //
    memset(szAttributes, 0, sizeof(szAttributes));

    hr = StringCbCopyEx(szAttributes,sizeof(szAttributes),_TEXT("DSN="),&pszAttributesEnd, &cbRemaining,0);

    if (FAILED(hr))
    {
        return FALSE;
    }

    hr = StringCbCopyEx(pszAttributesEnd,cbRemaining,pszDsn,&pszAttributesEnd, &cbRemaining,0);

    if (FAILED(hr))
    {
        return FALSE;
    }

    hr = StringCbCopyEx(pszAttributesEnd,cbRemaining,_TEXT("UID="),&pszAttributesEnd, &cbRemaining,0);

    if (FAILED(hr))
    {
        return FALSE;
    }

    hr = StringCbCopyEx(pszAttributesEnd,cbRemaining,pszUser,&pszAttributesEnd, &cbRemaining,0);

    if (FAILED(hr))
    {
        return FALSE;
    }

    if(pszPwd)
    {
        hr = StringCbCopyEx(pszAttributesEnd,cbRemaining,_TEXT("PASSWORD="),&pszAttributesEnd, &cbRemaining,0);

        if (FAILED(hr))
        {
            return FALSE;
        }

        hr = StringCbCopyEx(pszAttributesEnd,cbRemaining,pszPwd,&pszAttributesEnd, &cbRemaining,0);

        if (FAILED(hr))
        {
            return FALSE;
        }
    }

    hr = StringCbCopyEx(pszAttributesEnd,cbRemaining,_TEXT("DBQ="),&pszAttributesEnd, &cbRemaining,0);

    if (FAILED(hr))
    {
        return FALSE;
    }

    hr = StringCbCopyEx(pszAttributesEnd,cbRemaining,pszMdbFile,&pszAttributesEnd, &cbRemaining,0);

    if (FAILED(hr))
    {
        return FALSE;
    }

    bConfig=fpSQLConfigDataSource(NULL,
                                  (WORD)((bInstall) ? ODBC_ADD_SYS_DSN : ODBC_REMOVE_SYS_DSN),
                                  pszDriver,
                                  szAttributes);
    // ignore error on uninstall
    if(!bConfig && bInstall)
    {
        ReportError(hWnd, _TEXT("Can't config data source"));
    }

    return bConfig;
}


BOOL
RepairDataSource( 
    HWND     hWnd, 
    LPTSTR   pszDriver,
    LPTSTR   pszDsn,          // DSN
    LPTSTR   pszUser,         // User
    LPTSTR   pszPwd,          // Password
    LPTSTR   pszMdbFile       // MDB file
    )

/*++

Abstract:

    Routine to Compact/Repair a database file

Parameters:

    hWnd - Parent window handle.
    pszDsn - ODBC Data Source Name.
    pszUser - Login use name.
    pszPwd - Login password.
    pszMdbFile - Name of the Database file.
    
Returns:

    TRUE if successfule, FALSE otherwise.

++*/

{
    TCHAR   szAttributes[MAX_PATH*6+1];
    BOOL    bConfig=TRUE;
    TCHAR*  pszAttributesEnd;
    size_t  cbRemaining;
    HRESULT hr;

    if(hODBCCP32 == NULL && !InitODBCSetup())
        return FALSE;

    //
    // for attribute string
    //
    memset(szAttributes, 0, sizeof(szAttributes));

    hr = StringCbCopyEx(szAttributes,sizeof(szAttributes),_TEXT("DSN="),&pszAttributesEnd, &cbRemaining,0);

    if (FAILED(hr))
    {
        return FALSE;
    }

    hr = StringCbCopyEx(pszAttributesEnd,cbRemaining,pszDsn,&pszAttributesEnd, &cbRemaining,0);

    if (FAILED(hr))
    {
        return FALSE;
    }

    hr = StringCbCopyEx(pszAttributesEnd,cbRemaining,_TEXT("UID="),&pszAttributesEnd, &cbRemaining,0);

    if (FAILED(hr))
    {
        return FALSE;
    }

    hr = StringCbCopyEx(pszAttributesEnd,cbRemaining,pszUser,&pszAttributesEnd, &cbRemaining,0);

    if (FAILED(hr))
    {
        return FALSE;
    }

    if(pszPwd)
    {
        hr = StringCbCopyEx(pszAttributesEnd,cbRemaining,_TEXT("PASSWORD="),&pszAttributesEnd, &cbRemaining,0);

        if (FAILED(hr))
        {
            return FALSE;
        }

        hr = StringCbCopyEx(pszAttributesEnd,cbRemaining,pszPwd,&pszAttributesEnd, &cbRemaining,0);

        if (FAILED(hr))
        {
            return FALSE;
        }
    }

    hr = StringCbCopyEx(pszAttributesEnd,cbRemaining,_TEXT("DBQ="),&pszAttributesEnd, &cbRemaining,0);

    if (FAILED(hr))
    {
        return FALSE;
    }

    hr = StringCbCopyEx(pszAttributesEnd,cbRemaining,pszMdbFile,&pszAttributesEnd, &cbRemaining,0);

    if (FAILED(hr))
    {
        return FALSE;
    }

    hr = StringCbCopyEx(pszAttributesEnd,cbRemaining,_TEXT("REPAIR_DB="),&pszAttributesEnd, &cbRemaining,0);

    if (FAILED(hr))
    {
        return FALSE;
    }

    hr = StringCbCopyEx(pszAttributesEnd,cbRemaining,pszMdbFile,&pszAttributesEnd, &cbRemaining,0);

    if (FAILED(hr))
    {
        return FALSE;
    }

    bConfig=fpSQLConfigDataSource(
                            NULL,
                            (WORD)ODBC_CONFIG_SYS_DSN,
                            pszDriver,
                            szAttributes
                        );

    // ignore error on uninstall
    if(bConfig == FALSE)
    {
        ReportError(hWnd, _TEXT("Can't repair data source"));
    }

    return bConfig;
}