windows-xp/Source/XPSP1/NT/multimedia/media/deluxe2/cdopt/cddata.cpp

//--------------------------------------------------------------------------;
//
//  File: cddb.cpp
//
//  Copyright (c) 1998 Microsoft Corporation.  All rights reserved 
//
//--------------------------------------------------------------------------;

#include "precomp.h"  
#include "cdopti.h"
#include "cddata.h"

#define SECTION_BUFFER      (512)
#define PLAYLIST_BUFFER     (255)
#define BUFFERINC   (128)

#define MAXNUMPLAY (127)
#define ROWSET  1

#ifndef SQL_SUCCEEDED
#define SQL_SUCCEEDED(rc) (((rc)&(~1))==0)
#endif

#ifndef SQL_W_CHAR
#define SQL_C_WCHAR		 	(-8)
#endif

#ifndef SQL_LONGCHAR_FIELD
#define SQL_LONGCHAR_FIELD (-10)
#endif

const TCHAR     gszIniFile[]     = TEXT("cdplayer.ini");
const TCHAR     gszDefault[]     = TEXT("\0");
const TCHAR     gszArtist[]      = TEXT("artist");
const TCHAR     gszTitle[]       = TEXT("title");
const TCHAR     gszNumTracks[]   = TEXT("numtracks");
const TCHAR     gszOrder[]       = TEXT("order");
const TCHAR     gszNumPlay[]     = TEXT("numplay");

const TCHAR     gszDriverName[]  = TEXT("Microsoft Access Driver (*.mdb)");
const TCHAR     gszDSNAttr[]     = TEXT("DSN=DeluxeCD%cDefaultDir=%s%cDriverID=25%cDBQ=DeluxeCD.mdb%c");
const TCHAR     gszDSNCreate[]   = TEXT("CREATE_DB=%s\\DeluxeCD.mdb%c");

const TCHAR     gszTitleTable[]  = TEXT("Titles");
const TCHAR     gszTrackTable[]  = TEXT("Tracks");
const TCHAR     gszMenuTable[]   = TEXT("Menus");
const TCHAR     gszBatchTable[]  = TEXT("Batch");

TCHAR         gszDSN[]         = TEXT("DeluxeCD");

TCHAR         gszTitlesCreate[] = TEXT("create table Titles ")
                                       TEXT("(")
                                            TEXT("TitleID long, ")
                                            TEXT("Artist longchar, ")
                                            TEXT("Title longchar, ")
                                            TEXT("Copyright longchar, ")
                                            TEXT("Label longchar, ")
                                            TEXT("ReleaseDate longchar, ")
                                            TEXT("NumTracks long, ")
                                            TEXT("NumMenus long, ")
                                            TEXT("PlayList longchar, ")
                                            TEXT("TitleQuery longchar ")
                                       TEXT(")");

TCHAR        gszTracksCreate[] = TEXT("create table Tracks ")
                                       TEXT("(")
                                            TEXT("TitleID long, ")
                                            TEXT("TrackID long, ")
                                            TEXT("TrackName longchar ")
                                       TEXT(")");

TCHAR         gszMenusCreate[] = TEXT("create table Menus ")
                                       TEXT("(")
                                            TEXT("TitleID long, ")
                                            TEXT("MenuID long, ")
                                            TEXT("MenuName longchar, ")
                                            TEXT("MenuQuery longchar ")
                                       TEXT(")");

TCHAR         gszBatchCreate[] = TEXT("create table Batch ")
                                       TEXT("(")
                                            TEXT("TitleID long, ")
                                            TEXT("NumTracks long, ")
                                            TEXT("TitleQuery longchar")
                                       TEXT(")");


SDWORD              gcbTitles[] = { 0, SQL_NTS, SQL_NTS, SQL_NTS, SQL_NTS, SQL_NTS, 0, 0, SQL_NTS, INTERNET_MAX_PATH_LENGTH };
SDWORD              gcbTracks[] = { 0, 0, SQL_NTS };
SDWORD              gcbMenus[]  = { 0, 0, SQL_NTS, INTERNET_MAX_PATH_LENGTH };
SDWORD              gcbBatch[]  = { 0, 0, INTERNET_MAX_PATH_LENGTH };



extern HINSTANCE g_dllInst;

////////////
// Functions
////////////

CCDData::CCDData()
{
    m_pTitleList = NULL;
    m_pBatchList = NULL;
    m_dwLoadCnt = 0;
    m_dwBatchCnt = 0;
    m_henv = NULL;
    m_hdbc = NULL;
    m_fToldUser = FALSE;
    m_pSQL = NULL;
    m_dwRef = 0;

    memset(&m_bd, 0, sizeof(m_bd));

	InitializeCriticalSection(&m_crit);

}


CCDData::~CCDData()
{
    CloseDatabase();

    if (m_pTitleList)
    {
        DestroyTitles(&m_pTitleList);
    }

    if (m_pSQL)
    {
        delete m_pSQL;
    }

	DeleteCriticalSection(&m_crit);
}


STDMETHODIMP CCDData::QueryInterface(REFIID riid, void** ppv)
{
    *ppv = NULL;
    if (IID_IUnknown == riid || IID_ICDData == riid)
    {  
        *ppv = this;
    }

    if (NULL==*ppv)
    {
        return E_NOINTERFACE;
    }

    AddRef();

    return S_OK;
}


STDMETHODIMP CCDData::GetSQLPtr(BOOL fInited)
{
	HRESULT hr = S_OK;
    
	if (m_pSQL == NULL)
	{
        static BOOL fAttempt = FALSE;

        if (!fAttempt)
        {
            m_pSQL = new SQL;

            if (m_pSQL && !m_pSQL->Initialize())
            {
                delete m_pSQL;
                m_pSQL = NULL;
		        hr = E_FAIL;
            }
            else
            {
                if (!fInited)     // we must be running in shell mode, quietly initialize
                {
                    Initialize(NULL);
                    CheckDatabase(NULL);
                }
            }
        }
        else
        {
            hr = E_FAIL;
        }

        fAttempt = TRUE;
    }

	return(hr);
}

STDMETHODIMP_(ULONG) CCDData::AddRef(void)
{
    return ++m_dwRef;
}


STDMETHODIMP_(ULONG) CCDData::Release(void)
{
    if (0!=--m_dwRef)
        return m_dwRef;

    delete this;
    return 0;
}


STDMETHODIMP_(void) CCDData::Enter(void)
{
	EnterCriticalSection(&m_crit);
}

STDMETHODIMP_(void) CCDData::Leave(void)
{
	LeaveCriticalSection(&m_crit);
}





STDMETHODIMP CCDData::GetUnknownString(TCHAR **ppStr, const TCHAR *szSection, const TCHAR *szEntry, DWORD dwInitialAlloc)
{
    TCHAR   *pStr;
    DWORD   dwSize;
    DWORD   dwResult;
    HRESULT hr = S_OK;

    dwSize = dwInitialAlloc - BUFFERINC;

    pStr = NULL;

    do 
    {
        dwSize += BUFFERINC; 

        if (pStr)
        {
            delete pStr;
        }

        pStr = new(TCHAR[dwSize]);

        if (pStr == NULL)
        {
            hr = E_OUTOFMEMORY;
            break;
        }

        dwResult = GetPrivateProfileString(szSection, szEntry, TEXT("\0"), pStr, dwSize, gszIniFile);
    }
    while (dwResult == (dwSize - 2));

    *ppStr = pStr;

    return(hr);
}




STDMETHODIMP_(void) CCDData::ImportTrack(TCHAR *szDiscID, DWORD dwTrack)
{
    TCHAR szTrack[6];

    wsprintf(szTrack,TEXT("%d"),dwTrack);
    GetPrivateProfileString(szDiscID,szTrack,gszDefault,m_bd.CDTrack.szName,CDSTR,gszIniFile);
}


STDMETHODIMP_(RETCODE) CCDData::ImportTracks(HSTMT hstmt, TCHAR *szDiscID)
{
    RETCODE rc = SQL_SUCCESS;

    for (DWORD dwTrack = 0; dwTrack < m_bd.CDTitle.dwNumTracks; dwTrack++)
    {
        ImportTrack(szDiscID, dwTrack);

        rc = m_pSQL->SetPos(hstmt, 0, SQL_ADD, SQL_LOCK_NO_CHANGE);

        if (rc == SQL_SUCCESS)
        {
            m_bd.dwTrackID++;
        }
        else
        {
            ReportError(&m_bd, hstmt);
            break;
        }
        
    }

    return(rc);
}


STDMETHODIMP CCDData::ImportTitle(TCHAR *szDiscID)
{
    HRESULT     hr = S_OK;
    DWORD       dwTitleID;

    _stscanf(szDiscID,TEXT("%xd"),&dwTitleID);

    if (dwTitleID == CDTITLE_NODISC)
    {
        hr = E_INVALIDARG;
    }
    else
    {
        m_bd.CDTitle.dwTitleID      = dwTitleID;
        m_bd.CDTitle.dwNumTracks    = GetPrivateProfileInt(szDiscID,gszNumTracks,0,gszIniFile);
        m_bd.CDTitle.dwNumPlay      = GetPrivateProfileInt(szDiscID,gszNumPlay,0,gszIniFile);
        GetPrivateProfileString(szDiscID, gszArtist, gszDefault, m_bd.CDTitle.szArtist, CDSTR, gszIniFile);
        GetPrivateProfileString(szDiscID, gszTitle,  gszDefault, m_bd.CDTitle.szTitle,  CDSTR, gszIniFile);

        m_bd.szPlayList[0] = TEXT('\0');
        m_bd.szQuery[0] = TEXT('\0');

        if (m_bd.CDTitle.dwNumPlay)
        {
            TCHAR   *pPlayTable = NULL;   
            TCHAR   *pText = NULL;
            DWORD   dwIndex;
            int     iNum;
            TCHAR   *pDst = m_bd.szPlayList;

            m_bd.CDTitle.dwNumPlay = min(m_bd.CDTitle.dwNumPlay, MAXNUMPLAY);

            if (FAILED(GetUnknownString(&pPlayTable, szDiscID, gszOrder, PLAYLIST_BUFFER)))
            {
                m_bd.CDTitle.dwNumPlay = 0;
            }
            else
            {
                pText = pPlayTable;

                for (dwIndex = 0; dwIndex < m_bd.CDTitle.dwNumPlay && *pText; dwIndex++)
                {
                    _stscanf(pText,TEXT("%d"), &iNum);
                    wsprintf(pDst, TEXT("%02x"), iNum);

                    while(*pDst != TEXT('\0'))
                    {
                        pDst++;
                    }

                    while(isdigit(*pText++));
                }

                delete pPlayTable;
            }
        }
    }

    return(hr);
}


STDMETHODIMP_(void) CCDData::ImportDatabase(LPTIMEDMETER ptm, HSTMT *hstmt, TCHAR *szDiscID)
{
    DWORD   dwCount = 0;
    RETCODE rc;

    while(*szDiscID)
    {
        m_bd.dwTrackID = 0;
        m_bd.dwMenuID = 0;
       
        if (SUCCEEDED(ImportTitle(szDiscID)))    
        {
            rc = ImportTracks(hstmt[1], szDiscID);

            if (rc == SQL_SUCCESS)
            {
                rc = m_pSQL->SetPos(hstmt[0], 0, SQL_ADD, SQL_LOCK_NO_CHANGE);
                ReportError(&m_bd, hstmt);
            }
            
            if (!SQL_SUCCEEDED(rc))
            {
                break;
            }  
        }

        UpdateMeter(ptm);

        while(*szDiscID++);
    }   
}



STDMETHODIMP_(DWORD) CCDData::ImportCount(TCHAR *pEntries)
{
    TCHAR *szDiscID = pEntries;
    DWORD dwCount = 0;

    while(*szDiscID)
    {
        dwCount++;
        while(*szDiscID++);
    }

    return(dwCount);
}




STDMETHODIMP_(void) CCDData::CreateMeter(LPTIMEDMETER ptm, HWND hWnd, DWORD dwCount, DWORD dwJump, UINT uStringID)
{
    if (hWnd && ptm)
    {
        ptm->hMeter = CreateDialog(g_dllInst,MAKEINTRESOURCE(IDD_LOADSTATUS),hWnd,(DLGPROC) MeterHandler);

        if (ptm->hMeter)
        {
            ptm->hParent = hWnd;
            ptm->dwStartTime = timeGetTime();
            ptm->dwRange = dwCount;
            ptm->fShowing = FALSE;
            ptm->dwCount = 0;
            ptm->dwShowCount = 0;
            ptm->dwJump = dwJump;

            if (uStringID != 0)
            {
                TCHAR szTitle[255];
                LoadString(g_dllInst, uStringID, szTitle, sizeof(szTitle)/sizeof(TCHAR));
                SetWindowText(ptm->hMeter, szTitle);
            }
        }
    }
    else
    {
        if (ptm)
        {
            memset(ptm, 0, sizeof(TIMEDMETER));
        }
    }
}


STDMETHODIMP_(void) CCDData::UpdateMeter(LPTIMEDMETER ptm)
{
    if (ptm && ptm->hMeter)
    {
        ptm->dwCount++;

        if (ptm->fShowing)
        {
            if (ptm->hMeter && ((ptm->dwCount % ptm->dwJump) == 0))
            {
                SendDlgItemMessage( ptm->hMeter, IDC_PROGRESSMETER, PBM_SETPOS, (WPARAM) ptm->dwCount - ptm->dwShowCount, 0);
            }           
        }
        else
        {
            if ((ptm->dwCount % ptm->dwJump) == 0)
            {
                DWORD dwUsedTime = timeGetTime() - ptm->dwStartTime;               // Compute time used
                DWORD dwProjected = (ptm->dwRange / ptm->dwCount) * dwUsedTime;    // Project Time to complete
                DWORD dwProjRemain = dwProjected - dwUsedTime;                     // Compute projected remaining time

                if (dwProjRemain >= 1500)        // If it looks like it's going to take a while, put up the meter
                {
                    DWORD dwNumJumps = ptm->dwCount / ptm->dwJump;
                    DWORD dwJumpTime = dwUsedTime / dwNumJumps;

                    if (dwJumpTime > 200)       // To make sure the meter moves smoothly, re-compute jump count
                    {
                        ptm->dwJump = (ptm->dwJump / ((dwJumpTime / 200) + 1)) + 1; 
                    }

                    ptm->dwShowCount = ptm->dwCount;
                    SendDlgItemMessage( ptm->hMeter, IDC_PROGRESSMETER, PBM_SETRANGE, 0, MAKELPARAM(0, ptm->dwRange - ptm->dwShowCount));
                    SendDlgItemMessage( ptm->hMeter, IDC_PROGRESSMETER, PBM_SETPOS, (WPARAM) 0, 0);
			        ShowWindow(ptm->hMeter,SW_SHOWNORMAL);
			        UpdateWindow(ptm->hMeter);
                    ptm->fShowing = TRUE;
                }
            }
        }
    }
}



STDMETHODIMP_(void) CCDData::DestroyMeter(LPTIMEDMETER ptm)
{
    if (ptm && ptm->hMeter)
    {
        DestroyWindow(ptm->hMeter);
        SetForegroundWindow(ptm->hParent);
        memset(ptm, 0, sizeof(LPTIMEDMETER));
    }
}



STDMETHODIMP_(void) CCDData::InitDatabase(HSTMT *hstmt)
{
    m_pSQL->ExecDirect(hstmt[0], (UCHAR *) TEXT("drop table Titles"), SQL_NTS);
    m_pSQL->ExecDirect(hstmt[1], (UCHAR *) TEXT("drop table Tracks"), SQL_NTS);
    m_pSQL->ExecDirect(hstmt[2], (UCHAR *) TEXT("drop table Menus"), SQL_NTS);
    m_pSQL->ExecDirect(hstmt[3], (UCHAR *) TEXT("drop table Batch"), SQL_NTS);

    m_pSQL->ExecDirect(hstmt[0], (UCHAR *) gszTitlesCreate, SQL_NTS);
    m_pSQL->ExecDirect(hstmt[1], (UCHAR *) gszTracksCreate, SQL_NTS);
    m_pSQL->ExecDirect(hstmt[2], (UCHAR *) gszMenusCreate, SQL_NTS);
    m_pSQL->ExecDirect(hstmt[3], (UCHAR *) gszBatchCreate, SQL_NTS);

    m_pSQL->FreeStmt(hstmt[0], SQL_CLOSE); 
    m_pSQL->FreeStmt(hstmt[1], SQL_CLOSE); 
    m_pSQL->FreeStmt(hstmt[2], SQL_CLOSE); 
    m_pSQL->FreeStmt(hstmt[3], SQL_CLOSE); 
}



STDMETHODIMP_(void) CCDData::SetCursors(HSTMT *hstmt)
{
    for (DWORD dwIndex = 0; dwIndex < NUMTABLES; dwIndex++)
    {
        m_pSQL->SetStmtOption(hstmt[dwIndex], SQL_CONCURRENCY, SQL_CONCUR_VALUES); 
        m_pSQL->SetStmtOption(hstmt[dwIndex], SQL_CURSOR_TYPE, SQL_CURSOR_KEYSET_DRIVEN);
        m_pSQL->SetStmtOption(hstmt[dwIndex], SQL_ROWSET_SIZE, ROWSET);
    }
}


STDMETHODIMP_(void) CCDData::InitCBTable(LPBOUND pbd)
{
    memcpy(pbd->cbt.cbTitles, gcbTitles, sizeof(gcbTitles));
    memcpy(pbd->cbt.cbTracks, gcbTracks, sizeof(gcbTracks));
    memcpy(pbd->cbt.cbMenus,  gcbMenus,  sizeof(gcbMenus));
    memcpy(pbd->cbt.cbBatch,  gcbBatch,  sizeof(gcbBatch));
}

STDMETHODIMP_(void) CCDData::BindTitles(HSTMT *hstmt, LPBOUND pbd)
{
    m_pSQL->BindCol(hstmt[0],  1, SQL_C_ULONG,  &(pbd->CDTitle.dwTitleID),    sizeof(pbd->CDTitle.dwTitleID),     &pbd->cbt.cbTitles[0]);
    m_pSQL->BindCol(hstmt[0],  2, SQL_C_WCHAR,   pbd->CDTitle.szArtist,        sizeof(pbd->CDTitle.szArtist),      &pbd->cbt.cbTitles[1]);
    m_pSQL->BindCol(hstmt[0],  3, SQL_C_WCHAR,   pbd->CDTitle.szTitle,         sizeof(pbd->CDTitle.szTitle),       &pbd->cbt.cbTitles[2]);
    m_pSQL->BindCol(hstmt[0],  4, SQL_C_WCHAR,   pbd->CDTitle.szCopyright,     sizeof(pbd->CDTitle.szCopyright),   &pbd->cbt.cbTitles[3]);
    m_pSQL->BindCol(hstmt[0],  5, SQL_C_WCHAR,   pbd->CDTitle.szLabel,         sizeof(pbd->CDTitle.szLabel),       &pbd->cbt.cbTitles[4]);
    m_pSQL->BindCol(hstmt[0],  6, SQL_C_WCHAR,   pbd->CDTitle.szDate,          sizeof(pbd->CDTitle.szDate),        &pbd->cbt.cbTitles[5]);
    m_pSQL->BindCol(hstmt[0],  7, SQL_C_ULONG,  &(pbd->CDTitle.dwNumTracks),  sizeof(pbd->CDTitle.dwNumTracks),   &pbd->cbt.cbTitles[6]);
    m_pSQL->BindCol(hstmt[0],  8, SQL_C_ULONG,  &(pbd->CDTitle.dwNumMenus),   sizeof(pbd->CDTitle.dwNumMenus),    &pbd->cbt.cbTitles[7]);
    m_pSQL->BindCol(hstmt[0],  9, SQL_C_WCHAR,   pbd->szPlayList,              sizeof(pbd->szPlayList),            &pbd->cbt.cbTitles[8]);
    m_pSQL->BindCol(hstmt[0], 10, SQL_C_WCHAR,   pbd->szQuery,                 sizeof(pbd->szQuery),               &pbd->cbt.cbTitles[9]);
}


STDMETHODIMP_(void) CCDData::BindTracks(HSTMT *hstmt, LPBOUND pbd)
{
    m_pSQL->BindCol(hstmt[1],  1, SQL_C_ULONG,  &(pbd->CDTitle.dwTitleID),    sizeof(pbd->CDTitle.dwTitleID),     &pbd->cbt.cbTracks[0]);
    m_pSQL->BindCol(hstmt[1],  2, SQL_C_ULONG,  &(pbd->dwTrackID),            sizeof(pbd->dwTrackID),             &pbd->cbt.cbTracks[1]);
    m_pSQL->BindCol(hstmt[1],  3, SQL_C_WCHAR,   pbd->CDTrack.szName,          sizeof(pbd->CDTrack.szName),        &pbd->cbt.cbTracks[2]);
}

STDMETHODIMP_(void) CCDData::BindMenus(HSTMT *hstmt, LPBOUND pbd)
{
    m_pSQL->BindCol(hstmt[2],  1, SQL_C_ULONG,  &(pbd->CDTitle.dwTitleID),    sizeof(pbd->CDTitle.dwTitleID),     &pbd->cbt.cbMenus[0]);
    m_pSQL->BindCol(hstmt[2],  2, SQL_C_ULONG,  &(pbd->dwMenuID),             sizeof(pbd->dwMenuID),              &pbd->cbt.cbMenus[1]);
    m_pSQL->BindCol(hstmt[2],  3, SQL_C_WCHAR,   pbd->CDMenu.szMenuText,       sizeof(pbd->CDMenu.szMenuText),     &pbd->cbt.cbMenus[2]);
    m_pSQL->BindCol(hstmt[2],  4, SQL_C_WCHAR,   pbd->szQuery,                 sizeof(pbd->szQuery),               &pbd->cbt.cbMenus[3]);
}

STDMETHODIMP_(void) CCDData::BindBatch(HSTMT *hstmt, LPBOUND pbd)
{
    m_pSQL->BindCol(hstmt[3],  1, SQL_C_ULONG,  &(pbd->CDTitle.dwTitleID),    sizeof(pbd->CDTitle.dwTitleID),     &pbd->cbt.cbBatch[0]);
    m_pSQL->BindCol(hstmt[3],  2, SQL_C_ULONG,  &(pbd->CDTitle.dwNumTracks),  sizeof(pbd->CDTitle.dwNumTracks),   &pbd->cbt.cbBatch[1]);
    m_pSQL->BindCol(hstmt[3],  3, SQL_C_WCHAR, pbd->szQuery,                 sizeof(pbd->szQuery),               &pbd->cbt.cbBatch[2]);
}


STDMETHODIMP_(void) CCDData::SetBindings(HSTMT *hstmt, LPBOUND pbd)
{
    BindTitles(hstmt,pbd);
    BindTracks(hstmt,pbd);
    BindMenus(hstmt,pbd);
    BindBatch(hstmt,pbd);
}


STDMETHODIMP_(void) CCDData::ReportError(LPBOUND pbd, HSTMT hstmt)
{
#ifdef DEBUG

    UCHAR state[255];
    SDWORD dwErr;
    UCHAR szErr[SQL_MAX_MESSAGE_LENGTH];
    SWORD cbErr;

    m_pSQL->Error(pbd->henv, pbd->hdbc, hstmt, state, &dwErr, szErr, SQL_MAX_MESSAGE_LENGTH, &cbErr);

    OutputDebugString((TCHAR *) szErr);

    if (szErr[0] != TEXT('\0'))
    {
        OutputDebugString(TEXT("\n"));
    }
#endif 
}





STDMETHODIMP_(RETCODE) CCDData::AllocStmt(HDBC hdbc, HSTMT *hstmt)
{
    RETCODE rc = SQL_SUCCESS;
    DWORD dwIndex;

    for (dwIndex = 0; dwIndex < NUMTABLES; dwIndex++)
    {
        hstmt[dwIndex] = NULL;
    }

    for (dwIndex = 0; dwIndex < NUMTABLES; dwIndex++)
    {
        rc = m_pSQL->AllocStmt(hdbc, &hstmt[dwIndex]);

        if (rc != SQL_SUCCESS)
        {
            break;
        }
    }

    return(rc);
}

STDMETHODIMP_(void) CCDData::FreeStmt(HSTMT *hstmt)
{
    RETCODE rc = SQL_SUCCESS;
    DWORD dwIndex;

    for (dwIndex = 0; dwIndex < NUMTABLES; dwIndex++)
    {
        if (hstmt[dwIndex] != NULL)
        {
            m_pSQL->FreeStmt(hstmt[dwIndex], SQL_DROP);
            hstmt[dwIndex] = NULL;
        }
    }
}




STDMETHODIMP_(void) CCDData::CreateDatabase(void)
{
    HENV        henv;
    HDBC        hdbc;
    HSTMT       hstmt[NUMTABLES]; 
    TCHAR       szWinDir[MAX_PATH];
    TCHAR       szCreateAttr[MAX_PATH];
    TCHAR       szDSNAttr[MAX_PATH];
    
    if (GetAppDataDir(szWinDir, sizeof(szWinDir)/sizeof(TCHAR)) != 0)
    {
        wsprintf(szCreateAttr, gszDSNCreate, szWinDir, TEXT('\0'));
        wsprintf(szDSNAttr, gszDSNAttr, TEXT('\0'), szWinDir, TEXT('\0'), TEXT('\0'), TEXT('\0'));
        
        if (m_pSQL->ConfigDataSource(NULL, ODBC_ADD_DSN, gszDriverName, szCreateAttr))
        {
            if (m_pSQL->ConfigDataSource(NULL, ODBC_ADD_DSN, gszDriverName, szDSNAttr))
            {
                if (m_pSQL->AllocEnv(&henv) == SQL_SUCCESS)
                {            
                    if (m_pSQL->AllocConnect(henv, &hdbc) == SQL_SUCCESS)
                    { 
                        if (m_pSQL->Connect(hdbc, (UCHAR *) gszDSN, SQL_NTS, NULL, 0, NULL, 0) == SQL_SUCCESS)
                        {
                            if (AllocStmt(hdbc, hstmt) == SQL_SUCCESS)
                            {
                                InitDatabase(hstmt);
                            }
                            FreeStmt(hstmt);
                        }
                        m_pSQL->Disconnect(hdbc);
                    }
                    m_pSQL->FreeConnect(hdbc);
                }
                m_pSQL->FreeEnv(henv);
            }
        }
    }
}


STDMETHODIMP CCDData::CheckDatabase(HWND hWnd)
{
    HRESULT hr = S_OK;

    GetSQLPtr(FALSE);

    if (!m_pSQL)
    {
        if (!m_fToldUser)
        {
            TCHAR szTitle[255];
            TCHAR szCaption[255];
            LoadString(g_dllInst, IDS_ERROR_NO_ODBC, szTitle, sizeof(szTitle)/sizeof(TCHAR));
            LoadString(g_dllInst, IDS_ERROR_SETUP, szCaption, sizeof(szTitle)/sizeof(TCHAR));
            MessageBox(hWnd, szTitle, szCaption, MB_OK | MB_ICONEXCLAMATION);
            m_fToldUser = TRUE;
        }
        
        hr = E_FAIL;
    }

    return(hr);
}


//finds the dir for allowing roaming profiles
STDMETHODIMP_(DWORD) CCDData::GetAppDataDir(TCHAR* pstrDir, DWORD cchSize)
{
    DWORD dwRet = 0;
/*
    TCHAR szDir[MAX_PATH] = TEXT("%userprofile%\\Application Data\\Microsoft\\DeluxeCD");

    dwRet = ExpandEnvironmentStrings(szDir,pstrDir,cchSize);
    if (dwRet > 0)
    {
        CreateDirectory(pstrDir,NULL);

        //shorten the pathname, because ODBC can't handle spaces
        _tcscpy(szDir,pstrDir);
        dwRet = GetShortPathName(szDir,pstrDir,cchSize);
    }
*/

    dwRet = GetWindowsDirectory(pstrDir,cchSize);

    return (dwRet);
}

STDMETHODIMP CCDData::Initialize(HWND hWnd)
{
    TCHAR       *pEntries = NULL;
    HRESULT     hr = S_OK;
    DWORD       dwCount;
    UDWORD      row[NUMTABLES];
    UWORD       rgfRowStat[NUMTABLES][ROWSET];
    TCHAR       szWinDir[MAX_PATH];
    TCHAR       szCreateAttr[MAX_PATH];
    TIMEDMETER  tm;

    if (SUCCEEDED(GetSQLPtr(TRUE)))
    {
        if (GetAppDataDir(szWinDir, sizeof(szWinDir)/sizeof(TCHAR)) != 0)
        {
            wsprintf(szCreateAttr, gszDSNCreate, szWinDir, TEXT('\0'));

            if (m_pSQL->ConfigDataSource(NULL, ODBC_ADD_DSN, gszDriverName, szCreateAttr))  // If this fails, we assume we've already imported the file
            {
                hr = GetUnknownString(&pEntries, NULL, NULL, SECTION_BUFFER);

                if (SUCCEEDED(hr))
                {
                    dwCount = ImportCount(pEntries);

                    CreateMeter(&tm, hWnd, dwCount, 5,  IDS_IMPORTING);

                    OpenDatabase(TRUE,hWnd);

                    if (m_henv)
                    {
                        TCHAR    *szDiscID = pEntries;
                        DWORD   dwIndex;

                        memset(&m_bd, 0, sizeof(m_bd));
                        m_bd.henv = m_henv;
                        m_bd.hdbc = m_hdbc;

                        InitDatabase(m_hstmt);

                        InitCBTable(&m_bd);
                        SetCursors( m_hstmt );
                        SetBindings( m_hstmt, &m_bd);

                        m_pSQL->SetConnectOption(m_hdbc, SQL_AUTOCOMMIT, SQL_AUTOCOMMIT_OFF);

                        m_pSQL->ExecDirect(m_hstmt[0], (UCHAR *) TEXT("select * from Titles"), SQL_NTS);
                        m_pSQL->ExecDirect(m_hstmt[1], (UCHAR *) TEXT("select * from Tracks"), SQL_NTS);
                        m_pSQL->ExecDirect(m_hstmt[2], (UCHAR *) TEXT("select * from Menus"), SQL_NTS);

                        for (dwIndex = 0; dwIndex < NUMTABLES; dwIndex++)
                        {
                            m_pSQL->ExtendedFetch(m_hstmt[dwIndex], SQL_FETCH_FIRST, 1, &row[dwIndex], rgfRowStat[dwIndex]);
                        }

                        ImportDatabase(&tm, m_hstmt, szDiscID);

                        m_pSQL->Transact(SQL_NULL_HENV, m_hdbc, SQL_COMMIT);

                        for (dwIndex = 0; dwIndex < NUMTABLES; dwIndex++)
                        {
                            m_pSQL->FreeStmt(m_hstmt[dwIndex], SQL_CLOSE);
                        }
               
                        m_pSQL->ExecDirect(m_hstmt[0], (UCHAR *) TEXT("create index TitleIDx on Titles(TitleID)"), SQL_NTS);
                        m_pSQL->ExecDirect(m_hstmt[1], (UCHAR *) TEXT("create index TitleIDx on Tracks(TitleID)"), SQL_NTS);
                        m_pSQL->ExecDirect(m_hstmt[2], (UCHAR *) TEXT("create index TitleIDx on Menus(TitleID)"), SQL_NTS);
                                          
                        m_pSQL->Transact(SQL_NULL_HENV, m_hdbc, SQL_COMMIT);
                        m_pSQL->SetConnectOption(m_hdbc, SQL_AUTOCOMMIT, SQL_AUTOCOMMIT_ON);

                        CloseDatabase();
                    }

                    DestroyMeter(&tm);
                }

                if (pEntries)
                {
                    delete pEntries;
                }
            }
        
            hr = OpenDatabase(FALSE,hWnd);       // We either already have a database or we just created one, open it. (If we didn't import OpenDatabase will create new one
        }
    }

    return(hr);
}

STDMETHODIMP CCDData::ConnectToDatabase(WORD fRequest)
{
    HRESULT     hr = E_FAIL;
    TCHAR       szWinDir[MAX_PATH];
    TCHAR       szDSNAttr[MAX_PATH];
    
    if (GetAppDataDir(szWinDir, sizeof(szWinDir)/sizeof(TCHAR)) != 0)
    {
        wsprintf(szDSNAttr, gszDSNAttr, TEXT('\0'), szWinDir, TEXT('\0'), TEXT('\0'), TEXT('\0'));

        if (m_pSQL->ConfigDataSource(NULL, fRequest, gszDriverName, szDSNAttr))
        {
            if (m_pSQL->AllocEnv(&m_henv) == SQL_SUCCESS)
            {            
                if (m_pSQL->AllocConnect(m_henv, &m_hdbc) == SQL_SUCCESS)
                { 
                    if (m_pSQL->Connect(m_hdbc, (UCHAR *) gszDSN, SQL_NTS, NULL, 0, NULL, 0) == SQL_SUCCESS)
                    {
                        if (AllocStmt(m_hdbc, m_hstmt) == SQL_SUCCESS)
                        {
                            hr = S_OK;
                        }
                    }
                }
            }
        }
    }
    
    return(hr);
}


STDMETHODIMP CCDData::OpenDatabase(BOOL fCreate, HWND hWnd)
{
    HRESULT hr = E_FAIL;

    m_henv = NULL;
    m_hdbc = NULL;
    
    for (DWORD dwIndex = 0; dwIndex < NUMTABLES; dwIndex++)
    {
        m_hstmt[dwIndex] = NULL;
    }
    
    hr = ConnectToDatabase(ODBC_ADD_DSN);

    if (FAILED(hr))                 // Lets try creating the database
    {
        CreateDatabase();
        hr = ConnectToDatabase(ODBC_CONFIG_DSN);
    }
    else
    {
        //only do this if connection to database succeeded right off the bat,
        //because if we just created a new one, it is in the right format already.
        if (IsOldFormat())
        {
            hr = UpgradeDatabase(hWnd);
        }
    }

    if (FAILED(hr))
    {
        CloseDatabase();
    }

    return (hr);
}



STDMETHODIMP_(void) CCDData::CloseDatabase(void)
{
    if (m_pSQL)
    {
        FreeStmt(m_hstmt);

        if (m_hdbc)
        {
            m_pSQL->Disconnect(m_hdbc);
            m_pSQL->FreeConnect(m_hdbc);
        }

        if (m_henv)
        {
            m_pSQL->FreeEnv(m_henv);
        }
    }
}




STDMETHODIMP_(DWORD) CCDData::GetNumRows(UCHAR *szDSN)
{
    HSTMT hstmt;
    DWORD dwCount = 0;
    SWORD Type,Cardinality;
    SDWORD cbCardinality = 0, cbType = 0;
    RETCODE rc;

    if (m_pSQL->AllocStmt(m_hdbc, &hstmt) == SQL_SUCCESS)
    {
        if (m_pSQL->Statistics(hstmt, NULL, 0, NULL, 0, szDSN, SQL_NTS, SQL_INDEX_ALL, SQL_ENSURE) == SQL_SUCCESS)
        {
            rc = m_pSQL->BindCol(hstmt,  7, SQL_C_SSHORT,  &Type,        sizeof(Type),         &cbType);
            rc = m_pSQL->BindCol(hstmt, 11, SQL_C_SSHORT,  &Cardinality, sizeof(Cardinality),  &cbCardinality);

            if (rc != SQL_SUCCESS)
            {
                ReportError(&m_bd, hstmt);
            }

            while (rc == SQL_SUCCESS)
            {
                rc = m_pSQL->Fetch(hstmt);
            
                if (rc == SQL_SUCCESS)
                {
                    if (Type == SQL_TABLE_STAT)
                    {
                        dwCount = (DWORD) Cardinality;
                        break;
                    }
                }
            }
        }

        m_pSQL->FreeStmt(hstmt, SQL_DROP);
    }
    return(dwCount);
}






STDMETHODIMP CCDData::ExtractTitle(LPCDTITLE *ppCDTitle)
{
    HRESULT     hr;
    LPCDTITLE   pCDTitle = NULL;

    m_bd.CDTitle.fLoaded = FALSE;

    hr = NewTitle(&pCDTitle, m_bd.CDTitle.dwTitleID, m_bd.CDTitle.dwNumTracks, m_bd.CDTitle.dwNumMenus);

    if (SUCCEEDED(hr))
    {    
        DWORD dwTrack = 0;
        DWORD dwMenu = 0;

        m_bd.CDTitle.pTrackTable = pCDTitle->pTrackTable;
        m_bd.CDTitle.pMenuTable = pCDTitle->pMenuTable;
        m_bd.CDTitle.pPlayList = pCDTitle->pPlayList;

        memcpy(pCDTitle,&m_bd.CDTitle,sizeof(m_bd.CDTitle));
        memset(&m_bd.CDTitle, 0, sizeof(m_bd.CDTitle));

        DWORD dwSize = lstrlen(m_bd.szQuery) + 1;

        if (dwSize)
        {
            pCDTitle->szTitleQuery = new(TCHAR[dwSize]);

            if (pCDTitle->szTitleQuery)
            {
                lstrcpy(pCDTitle->szTitleQuery, m_bd.szQuery);
            }
        }

        pCDTitle->dwNumPlay = lstrlen(m_bd.szPlayList) >> 1; // Null terminated, two digits per entry (two chars) thus divide by 2

        if (pCDTitle->dwNumPlay)
        {
            pCDTitle->dwNumPlay = min(pCDTitle->dwNumPlay, MAXNUMPLAY);    

            pCDTitle->pPlayList = new(WORD[pCDTitle->dwNumPlay]);

            if (pCDTitle->pPlayList == NULL)
            {
                pCDTitle->dwNumPlay = 0;
            }
            else
            {
                TCHAR   *pText = m_bd.szPlayList;
                LPWORD  pNum = pCDTitle->pPlayList;
                DWORD   dwIndex = 0;
                TCHAR   szNum[3];
                int     iNum;

                szNum[2] = TEXT('\0');
                for (dwIndex = 0; dwIndex < pCDTitle->dwNumPlay; dwIndex++, pNum++)
                {
                    szNum[0] = *pText++;    
                    szNum[1] = *pText++;    
                    _stscanf(szNum, TEXT("%02x"), &iNum);
                    *pNum = (WORD) iNum;   
                }
            }
        }
    }

    if (SUCCEEDED(hr) && ppCDTitle)
    {
        *ppCDTitle = pCDTitle;
    }
    else if (pCDTitle)
    {
        DestroyTitle(pCDTitle);
    }

    return(hr);
}


STDMETHODIMP_(LPCDTITLE) CCDData::FindTitle(LPCDTITLE pCDTitle, DWORD dwTitleID)
{
    while (pCDTitle)
    {
        if (pCDTitle->dwTitleID == dwTitleID)
        {
            break;
        }
        
        pCDTitle = pCDTitle->pNext;
    }

    return(pCDTitle);
}



STDMETHODIMP CCDData::ExtractTitles(LPCDTITLE *ppCDTitleList, HWND hWnd)
{
    HRESULT     hr = S_OK;
    LPCDTITLE   pCDTitle = NULL;;
    TIMEDMETER  tm;

    InitCBTable(&m_bd);
    SetCursors(m_hstmt);
    SetBindings(m_hstmt, &m_bd);

    if (ppCDTitleList == NULL || m_henv == NULL)
    {
        hr = E_INVALIDARG;
    }
    else
    {
        RETCODE rc;
        DWORD dwCount = GetNumRows((UCHAR *) gszTitleTable) + GetNumRows((UCHAR *) gszTrackTable) + GetNumRows((UCHAR *) gszMenuTable);

        CreateMeter(&tm, hWnd, dwCount, 100, 0);

        dwCount = 0;

        rc = m_pSQL->ExecDirect(m_hstmt[0], (UCHAR *) TEXT("select * from Titles"), SQL_NTS);

        if (rc == SQL_SUCCESS)
        {
            m_bd.szQuery[0] = TEXT('\0');

            while (m_pSQL->Fetch(m_hstmt[0]) == SQL_SUCCESS)
            {
                hr = ExtractTitle(&pCDTitle);

                if (SUCCEEDED(hr))
                {
                    AddTitle(ppCDTitleList, pCDTitle);
                    UpdateMeter(&tm);
                }

                m_bd.szQuery[0] = TEXT('\0');
            }

            if (*ppCDTitleList)
            {
                hr = S_OK;
            }

            m_pSQL->FreeStmt(m_hstmt[0], SQL_CLOSE);

            m_pSQL->ExecDirect(m_hstmt[1], (UCHAR *) TEXT("select * from Tracks"), SQL_NTS);
            pCDTitle = NULL;

            while (m_pSQL->Fetch(m_hstmt[1]) == SQL_SUCCESS)
            {
                if (pCDTitle == NULL || pCDTitle->dwTitleID != m_bd.CDTitle.dwTitleID)
                {
                    pCDTitle = FindTitle(*ppCDTitleList, m_bd.CDTitle.dwTitleID);
                }

                if (pCDTitle)
                {
                    if (m_bd.dwTrackID < pCDTitle->dwNumTracks)
                    {
                        memcpy(&(pCDTitle->pTrackTable[m_bd.dwTrackID]), &(m_bd.CDTrack), sizeof(m_bd.CDTrack));
                        UpdateMeter(&tm);
                    }
                }
            }

            m_pSQL->FreeStmt(m_hstmt[1], SQL_CLOSE);



            m_pSQL->ExecDirect(m_hstmt[2], (UCHAR *) TEXT("select * from Menus"), SQL_NTS);
            pCDTitle = NULL;
            m_bd.szQuery[0] = TEXT('\0');

            while (m_pSQL->Fetch(m_hstmt[2]) == SQL_SUCCESS)
            {
                if (pCDTitle == NULL || pCDTitle->dwTitleID != m_bd.CDTitle.dwTitleID)
                {
                    pCDTitle = FindTitle(*ppCDTitleList, m_bd.CDTitle.dwTitleID);
                }

                if (pCDTitle)
                {
                    if (m_bd.dwMenuID < pCDTitle->dwNumMenus)
                    {
                        LPCDMENU pCDMenu = &(pCDTitle->pMenuTable[m_bd.dwMenuID]);
                        TCHAR *szMenuQuery;

                        SetMenuQuery(pCDMenu, m_bd.szQuery);
                        szMenuQuery = pCDMenu->szMenuQuery;
                        memcpy(pCDMenu, &(m_bd.CDMenu), sizeof(m_bd.CDMenu));
                        pCDMenu->szMenuQuery = szMenuQuery;
                        m_bd.szQuery[0] = TEXT('\0');

                        UpdateMeter(&tm);
                    }
                }
            }

            m_pSQL->FreeStmt(m_hstmt[2], SQL_CLOSE);
        }

        DestroyMeter(&tm);
    }

    return(hr);
}



STDMETHODIMP CCDData::ExtractSingleTitle(LPCDTITLE *ppCDTitle, DWORD dwTitleID)
{
    HRESULT     hr = S_OK;
    LPCDTITLE   pCDTitle = NULL;;

    InitCBTable(&m_bd);
    SetCursors(m_hstmt);
    SetBindings( m_hstmt, &m_bd);

    if (ppCDTitle == NULL || m_henv == NULL)
    {
        hr = E_INVALIDARG;
    }
    else
    {
        RETCODE     rc;
        UCHAR       szQuery[MAX_PATH];
        
        wsprintf((TCHAR *) szQuery, TEXT("select * from Titles where TitleID = %d"), dwTitleID);
        rc = m_pSQL->ExecDirect(m_hstmt[0], szQuery, SQL_NTS);

        if (rc != SQL_SUCCESS)
        {
            hr = E_FAIL;
        }
        else
        {
            if (m_pSQL->Fetch(m_hstmt[0]) == SQL_SUCCESS)
            {
                hr = ExtractTitle(&pCDTitle);
            }
            else
            {
                hr = E_FAIL;
            }
        
            m_pSQL->FreeStmt(m_hstmt[0], SQL_CLOSE);

            if (SUCCEEDED(hr))
            {
                wsprintf((TCHAR *) szQuery, TEXT("select * from Tracks where TitleID = %d"), dwTitleID);
                m_pSQL->ExecDirect(m_hstmt[1], szQuery, SQL_NTS);

                while (m_pSQL->Fetch(m_hstmt[1]) == SQL_SUCCESS)
                {
                    if (m_bd.dwTrackID < pCDTitle->dwNumTracks)
                    {
                        memcpy(&(pCDTitle->pTrackTable[m_bd.dwTrackID]), &(m_bd.CDTrack), sizeof(m_bd.CDTrack));
                    }
                }

                m_pSQL->FreeStmt(m_hstmt[1], SQL_CLOSE);


                wsprintf((TCHAR *) szQuery, TEXT("select * from Menus where TitleID = %d"), dwTitleID);
                m_pSQL->ExecDirect(m_hstmt[2], szQuery, SQL_NTS);

                while (m_pSQL->Fetch(m_hstmt[2]) == SQL_SUCCESS)
                {
                    if (m_bd.dwMenuID < pCDTitle->dwNumMenus)
                    {
                        LPCDMENU pCDMenu = &(pCDTitle->pMenuTable[m_bd.dwMenuID]);
                        TCHAR *szMenuQuery;

                        SetMenuQuery(pCDMenu, m_bd.szQuery);
                        szMenuQuery = pCDMenu->szMenuQuery;
                        memcpy(pCDMenu, &(m_bd.CDMenu), sizeof(m_bd.CDMenu));
                        pCDMenu->szMenuQuery = szMenuQuery;
                        m_bd.szQuery[0] = TEXT('\0');
                    }
                }

                m_pSQL->FreeStmt(m_hstmt[2], SQL_CLOSE);
            }
        } //end else main query successful
    } //end else args ok

    if (SUCCEEDED(hr))
    {
        *ppCDTitle = pCDTitle;
    }

    return(hr);
}


STDMETHODIMP_(BOOL) CCDData::QueryDatabase(DWORD dwTitleID, const TCHAR *szTable)
{
    BOOL        fResult = FALSE;
    RETCODE     rc;
    UCHAR       szQuery[MAX_PATH];
    HSTMT       hstmt;

    if (m_pSQL->AllocStmt(m_hdbc, &hstmt) == SQL_SUCCESS)
    {
        wsprintf((TCHAR *) szQuery, TEXT("select * from %s where TitleID = %d"), szTable, dwTitleID);
        rc = m_pSQL->ExecDirect(hstmt, szQuery, SQL_NTS);

        if (rc == SQL_SUCCESS)
        {
            if (m_pSQL->Fetch(hstmt) == SQL_SUCCESS)
            {
                fResult = TRUE;
            }
        }

        m_pSQL->FreeStmt(hstmt, SQL_DROP);
    }

    return (fResult);
}




STDMETHODIMP_(BOOL) CCDData::QueryTitle(DWORD dwTitleID)
{
    BOOL fResult = FALSE;
    LPCDTITLE pCDTitle;

    Enter();

    if (SUCCEEDED(GetSQLPtr(FALSE)))
    {
        if (m_pTitleList == NULL)       // Database Not in memory, just query database directly
        {
            fResult = QueryDatabase(dwTitleID, gszTitleTable);
        }
        else                            // Database is in memory, just look here.
        {
            pCDTitle = m_pTitleList;

            while (pCDTitle)
            {
                if (pCDTitle->dwTitleID == dwTitleID)
                {
                    if (!pCDTitle->fRemove)
                    {
                        fResult = TRUE;
                    }
                    break;
                }
            
                pCDTitle = pCDTitle->pNext;
            }
        }
    }

    Leave();

    return(fResult);
}


STDMETHODIMP CCDData::LockTitle(LPCDTITLE *ppCDTitle, DWORD dwTitleID)
{
    HRESULT hr = E_INVALIDARG;

    Enter();

    if (SUCCEEDED(GetSQLPtr(FALSE)))
    {
        if (ppCDTitle)
        {
            if (m_pTitleList == NULL)       // Database Not in memory, load explicitly
            {
                if (SUCCEEDED(ExtractSingleTitle(ppCDTitle, dwTitleID)))
                {
                    (*ppCDTitle)->fLoaded = TRUE;
                    (*ppCDTitle)->dwLockCnt = 1;
                    hr = S_OK;
                }
            }
            else                            // Database is in memory, just look here.
            {
                LPCDTITLE pCDTitle;
                pCDTitle = m_pTitleList;

                while (pCDTitle)
                {
                    if (pCDTitle->dwTitleID == dwTitleID)
                    {
                        *ppCDTitle = pCDTitle;
                        pCDTitle->fLoaded = FALSE;
                        pCDTitle->dwLockCnt++;
                        hr = S_OK;
                        break;
                    }
            
                    pCDTitle = pCDTitle->pNext;
                }
            }
        }
    }
    
    Leave();

    return(hr);
}



STDMETHODIMP_(void) CCDData::UnlockTitle(LPCDTITLE pCDTitle, BOOL fPresist)
{
    Enter();

    if (SUCCEEDED(GetSQLPtr(FALSE)))
    {
        if (pCDTitle)
        {
            if (fPresist)               // They either made a new one, or made changes
            {
                DBSaveTitle(pCDTitle);  // write it to database
            }

            if (pCDTitle->fLoaded)      // This was NOT pulled from our database (Either locked while we didn't have the database, or was created.
            {
                pCDTitle->dwLockCnt--;          // they are unlocking

                if (fPresist && m_dwLoadCnt)    // This is a new item, lets add it to load database
                {
                    AddTitle(&m_pTitleList, pCDTitle);    // Insert into the loaded databased
                }
                else if (pCDTitle->dwLockCnt == 0)  // Not being saved and not in database, so nuke it
                {
                    DestroyTitle(pCDTitle);   
                }         
            }
            else                         // This title was pulled from our database, so just dec the ref
            {
                pCDTitle->dwLockCnt--;
            }
        }
    }

    Leave();
}



BOOL CALLBACK CCDData::MeterHandler(HWND hDlg, UINT msg, WPARAM wParam, LPARAM lParam)
{
	BOOL fReturnVal = TRUE;
	    	
    switch (msg) 
    { 
    	default:
			fReturnVal = FALSE;
		break;
		
        case WM_INITDIALOG:
        {
			fReturnVal = TRUE; 
        }		
		break;
    }

    return fReturnVal;
}




STDMETHODIMP CCDData::LoadTitles(HWND hWnd)
{
    HRESULT hr = E_FAIL;

    Enter();

    if (SUCCEEDED(GetSQLPtr(FALSE)))
    {
        hr = S_OK;

        if (m_pTitleList == NULL)
        {
            hr = ExtractTitles(&m_pTitleList, hWnd);
        }

        if (SUCCEEDED(hr))
        {
            m_dwLoadCnt++;
        }
    }

    Leave();

    return(hr);
}


STDMETHODIMP CCDData::PersistTitles()
{
    Enter();

    if (SUCCEEDED(GetSQLPtr(FALSE)))
    {
        SaveTitles(&m_pTitleList);
    }

    Leave();

    return S_OK;
}


STDMETHODIMP CCDData::UnloadTitles()
{
    HRESULT hr = E_FAIL;
    
    Enter();

    if (SUCCEEDED(GetSQLPtr(FALSE)))
    {
        if (m_pTitleList)
        {
            m_dwLoadCnt--;

            if (m_dwLoadCnt == 0)
            {
                DestroyTitles(&m_pTitleList);
            }

            hr = S_OK;
        }
    }

    Leave();

    return(hr);
}

STDMETHODIMP CCDData::CreateTitle(LPCDTITLE *ppCDTitle, DWORD dwTitleID, DWORD dwNumTracks, DWORD dwNumMenus)
{
    HRESULT hr = E_FAIL;
    LPCDTITLE pCDTitle;

    Enter();

    if (SUCCEEDED(GetSQLPtr(FALSE)))
    {
        hr = S_OK;

        if (!QueryTitle(dwTitleID))      // Only create a blank if it's not already in database
        {
			RemoveFromBatch(dwTitleID);

            hr = NewTitle(&pCDTitle, dwTitleID, dwNumTracks, dwNumMenus);
        }
        else
        {
            hr = LockTitle(&pCDTitle, dwTitleID);           // Let's get return the one we got

            if (SUCCEEDED(hr))                              // but first, re-fit it according to params
            {
                if (pCDTitle->dwNumTracks != dwNumTracks)
                {
                    if (pCDTitle->pTrackTable)
                    {
                        delete pCDTitle->pTrackTable;
                    }

                    pCDTitle->dwNumTracks = dwNumTracks;
                    pCDTitle->pTrackTable = new(CDTRACK[dwNumTracks]);

                    if (pCDTitle->pTrackTable == NULL)
                    {
                        hr = E_OUTOFMEMORY;
                    }
                }
        
                if (SUCCEEDED(hr) && pCDTitle->dwNumMenus != dwNumMenus)
                {
                    if (pCDTitle->pMenuTable)
                    {
                        delete pCDTitle->pMenuTable;
                    }

                    pCDTitle->dwNumMenus = dwNumMenus;
                    pCDTitle->pMenuTable = new(CDMENU[dwNumMenus]);
                    ZeroMemory(pCDTitle->pMenuTable,sizeof(CDMENU)*dwNumMenus);

                    if (pCDTitle->pMenuTable == NULL)
                    {
                        hr = E_OUTOFMEMORY;
                    }
                }
                
                if (FAILED(hr))
                {
                    DestroyTitle(pCDTitle);
                }
            }
        }

        if (SUCCEEDED(hr))
        {
            *ppCDTitle = pCDTitle;
        }
    }
    
    Leave();

    return(hr);
}




STDMETHODIMP CCDData::SetTitleQuery(LPCDTITLE pCDTitle, TCHAR *szTitleQuery)
{
    HRESULT hr = E_FAIL;
 
    if (SUCCEEDED(GetSQLPtr(FALSE)))
    {
        if (pCDTitle && szTitleQuery && lstrlen(szTitleQuery) < INTERNET_MAX_PATH_LENGTH)
        {
            if (pCDTitle->szTitleQuery)
            {
                delete pCDTitle->szTitleQuery;
            }

            DWORD dwSize = lstrlen(szTitleQuery) + 1;

            pCDTitle->szTitleQuery = new(TCHAR[dwSize]);

            if (pCDTitle->szTitleQuery == NULL)
            {
                hr = E_OUTOFMEMORY;
            }
            else
            {
                lstrcpy(pCDTitle->szTitleQuery, szTitleQuery);
                hr = S_OK;
            }
        }
    }

    return(hr);
}


STDMETHODIMP CCDData::SetMenuQuery(LPCDMENU pCDMenu, TCHAR *szMenuQuery)
{
    HRESULT hr = E_FAIL;
    
    if (SUCCEEDED(GetSQLPtr(FALSE)))
    {
        if (pCDMenu && szMenuQuery && lstrlen(szMenuQuery) < INTERNET_MAX_PATH_LENGTH)
        {
            if (pCDMenu->szMenuQuery)
            {
                delete pCDMenu->szMenuQuery;
            }

            DWORD dwSize = lstrlen(szMenuQuery) + 1;

            pCDMenu->szMenuQuery = new(TCHAR[dwSize]);

            if (pCDMenu->szMenuQuery== NULL)
            {
                hr = E_OUTOFMEMORY;
            }
            else
            {
                lstrcpy(pCDMenu->szMenuQuery, szMenuQuery);
                hr = S_OK;
            }
        }
    }

    return(hr);
}



STDMETHODIMP CCDData::NewTitle(LPCDTITLE *ppCDTitle, DWORD dwTitleID, DWORD dwNumTracks, DWORD dwNumMenus)
{
    HRESULT hr = S_OK;

    if (ppCDTitle == NULL || dwTitleID == CDTITLE_NODISC || dwNumTracks == 0)
    {
        hr = E_INVALIDARG;
    }
    else
    {
        LPCDTITLE pCDTitle = new(CDTITLE);

        if (pCDTitle == NULL)
        {
            hr = E_OUTOFMEMORY;
        }
        else
        {
            memset(pCDTitle, 0, sizeof(CDTITLE));

            pCDTitle->dwTitleID = dwTitleID;
            pCDTitle->dwNumTracks = dwNumTracks;
            pCDTitle->dwNumMenus = dwNumMenus;
            pCDTitle->fLoaded = TRUE;
            pCDTitle->dwLockCnt = 1;

            pCDTitle->pTrackTable = new(CDTRACK[dwNumTracks]);

            if (pCDTitle->pTrackTable == NULL)
            {
                hr = E_OUTOFMEMORY;
            }
            else
            {
                memset(pCDTitle->pTrackTable, 0, sizeof(CDTRACK) * dwNumTracks);

                if (dwNumMenus)
                {
                    pCDTitle->pMenuTable = new(CDMENU[dwNumMenus]);

                    if (pCDTitle->pMenuTable == NULL)
                    {
                        hr = E_OUTOFMEMORY;
                    }
                    else
                    {
                        memset(pCDTitle->pMenuTable, 0, sizeof(CDMENU) * dwNumMenus);
                    }
                }
            }

            if (FAILED(hr))
            {
                if (pCDTitle->pMenuTable)
                {
                    delete pCDTitle->pMenuTable;
                }

                if (pCDTitle->pTrackTable)
                {
                    delete pCDTitle->pTrackTable;
                }

                delete(pCDTitle);
            }
            else
            {
                *ppCDTitle = pCDTitle;
            }
        }
    }

    return(hr);
}




STDMETHODIMP_(void) CCDData::DestroyTitle(LPCDTITLE pCDTitle)
{
    if (pCDTitle)
    {
        if (pCDTitle->szTitleQuery)
        {
            delete pCDTitle->szTitleQuery;
        }

        if (pCDTitle->pTrackTable)
        {
            delete pCDTitle->pTrackTable;
        }

        if (pCDTitle->pPlayList)
        {
            delete pCDTitle->pPlayList;
        }

        if (pCDTitle->pMenuTable)
        {
            for (DWORD dwMenu = 0; dwMenu < pCDTitle->dwNumMenus; dwMenu++)
            {
                TCHAR *szMenuQuery = pCDTitle->pMenuTable[dwMenu].szMenuQuery;

                if (szMenuQuery)
                {
                    delete szMenuQuery;
                }
            }
            delete pCDTitle->pMenuTable;
        }
    }

    delete pCDTitle;
}



STDMETHODIMP_(void) CCDData::SaveTitle(LPCDTITLE pCDTitle, BOOL fExist)
{
    UCHAR       szQuery[MAX_PATH];
    UDWORD      row[NUMTABLES];
    UWORD       rgfRowStat[NUMTABLES][ROWSET];
    RETCODE     rc;

    if (fExist)
    {
        wsprintf((TCHAR *) szQuery, TEXT("select * from Titles where TitleID = %d"), pCDTitle->dwTitleID);
        rc = m_pSQL->ExecDirect(m_hstmt[0], szQuery, SQL_NTS);
        ReportError(&m_bd, m_hstmt[0]);
    }
    else
    {
        rc = m_pSQL->ExecDirect(m_hstmt[0], (UCHAR *) TEXT("select * from Titles"), SQL_NTS);
        ReportError(&m_bd, m_hstmt[0]);
    }

    rc = m_pSQL->ExtendedFetch(m_hstmt[0], SQL_FETCH_LAST, 1, &row[0], rgfRowStat[0]);
    ReportError(&m_bd, m_hstmt[0]);

    InitCBTable(&m_bd);
    memcpy(&m_bd.CDTitle, pCDTitle, sizeof(CDTITLE));

    if (pCDTitle->szTitleQuery)
    {
        lstrcpy(m_bd.szQuery, pCDTitle->szTitleQuery);
    }
    else
    {
        m_bd.szQuery[0] = TEXT('\0');
    }

    pCDTitle->dwNumPlay = min(pCDTitle->dwNumPlay, MAXNUMPLAY);

    m_bd.szPlayList[0] = TEXT('\0');

    if (pCDTitle->dwNumPlay && pCDTitle->pPlayList)
    {
        DWORD  dwIndex;
        TCHAR  *pDst = m_bd.szPlayList;
        LPWORD pNum = pCDTitle->pPlayList;

        for (dwIndex = 0; dwIndex < pCDTitle->dwNumPlay; dwIndex++, pNum++)
        {
            wsprintf(pDst, TEXT("%02x"), *pNum);

            while(*pDst != TEXT('\0'))
            {
                pDst++;
            }
        }
    }

    if (fExist)
    {
        rc = m_pSQL->SetPos(m_hstmt[0], 0, SQL_UPDATE, SQL_LOCK_NO_CHANGE);
       // ReportError(&m_bd, m_hstmt[0]);
    }
    else
    {
        rc = m_pSQL->SetPos(m_hstmt[0], 0, SQL_ADD, SQL_LOCK_NO_CHANGE);
       // ReportError(&m_bd, m_hstmt[0]);
    }
}



STDMETHODIMP_(void) CCDData::SaveTracks(LPCDTITLE pCDTitle, BOOL fExist)
{
    UCHAR       szQuery[MAX_PATH];
    UDWORD      row[NUMTABLES];
    UWORD       rgfRowStat[NUMTABLES][ROWSET];
    RETCODE     rc;
    DWORD       dwTrack;
    DWORD       dwTitleID = pCDTitle->dwTitleID;

    if (fExist)
    {
        wsprintf((TCHAR *) szQuery, TEXT("select * from Tracks where TitleID = %d"), pCDTitle->dwTitleID);
        rc = m_pSQL->ExecDirect(m_hstmt[1], szQuery, SQL_NTS);
        ReportError(&m_bd, m_hstmt[1]);
    }
    else
    {
        rc = m_pSQL->ExecDirect(m_hstmt[1], (UCHAR *) TEXT("select * from Tracks"), SQL_NTS);
        ReportError(&m_bd, m_hstmt[1]);
    }

    if (fExist)
    {
        rc = m_pSQL->ExtendedFetch(m_hstmt[1], SQL_FETCH_FIRST, 1, &row[0], rgfRowStat[0]);
        ReportError(&m_bd, m_hstmt[1]);

        for (dwTrack = 0; dwTrack < pCDTitle->dwNumTracks; dwTrack++)
        {
            InitCBTable(&m_bd);
            m_bd.CDTitle.dwTitleID = dwTitleID;

            if (m_bd.dwTrackID < pCDTitle->dwNumTracks)
            {
                memcpy(&m_bd.CDTrack, &(pCDTitle->pTrackTable[m_bd.dwTrackID]), sizeof(CDTRACK));
                rc = m_pSQL->SetPos(m_hstmt[1], 0, SQL_UPDATE, SQL_LOCK_NO_CHANGE);
                ReportError(&m_bd, m_hstmt[1]);
            }

            rc = m_pSQL->ExtendedFetch(m_hstmt[1], SQL_FETCH_NEXT, 1, &row[0], rgfRowStat[0]);
            ReportError(&m_bd, m_hstmt[1]);

        }
    }
    else
    {
        rc = m_pSQL->ExtendedFetch(m_hstmt[1], SQL_FETCH_LAST, 1, &row[0], rgfRowStat[0]);
        ReportError(&m_bd, m_hstmt[1]);

        InitCBTable(&m_bd);
        m_bd.CDTitle.dwTitleID = dwTitleID;

        for (dwTrack = 0; dwTrack < pCDTitle->dwNumTracks; dwTrack++)
        {
            m_bd.dwTrackID = dwTrack;
            memcpy(&m_bd.CDTrack, &(pCDTitle->pTrackTable[dwTrack]), sizeof(CDTRACK));
            rc = m_pSQL->SetPos(m_hstmt[1], 0, SQL_ADD, SQL_LOCK_NO_CHANGE);
            ReportError(&m_bd, m_hstmt[1]);
        }
    }
}


STDMETHODIMP_(void) CCDData::SaveMenus(LPCDTITLE pCDTitle)
{
    UCHAR       szQuery[MAX_PATH];
    UDWORD      row[NUMTABLES];
    UWORD       rgfRowStat[NUMTABLES][ROWSET];
    RETCODE     rc;
    DWORD       dwMenu;
    DWORD       dwTitleID = pCDTitle->dwTitleID;


    wsprintf((TCHAR *) szQuery, TEXT("delete from Menus where TitleID = %d"), pCDTitle->dwTitleID);
    rc = m_pSQL->ExecDirect(m_hstmt[2], szQuery, SQL_NTS);
    
    if (pCDTitle->dwNumMenus)
    {
        rc = m_pSQL->ExecDirect(m_hstmt[2], (UCHAR *) TEXT("select * from Menus"), SQL_NTS);
        ReportError(&m_bd, m_hstmt[2]);

        rc = m_pSQL->ExtendedFetch(m_hstmt[2], SQL_FETCH_LAST, 1, &row[0], rgfRowStat[0]);
        ReportError(&m_bd, m_hstmt[2]);

        InitCBTable(&m_bd);
        m_bd.CDTitle.dwTitleID = dwTitleID;

        for (dwMenu = 0; dwMenu < pCDTitle->dwNumMenus; dwMenu++)
        {
            LPCDMENU pCDMenu = &(pCDTitle->pMenuTable[dwMenu]);

            m_bd.dwMenuID = dwMenu;
            memcpy(&m_bd.CDMenu, pCDMenu, sizeof(CDMENU));
            m_bd.szQuery[0] = TEXT('\0');

            if (pCDMenu->szMenuQuery)
            {
                lstrcpy(m_bd.szQuery, pCDMenu->szMenuQuery);
            }
            
            rc = m_pSQL->SetPos(m_hstmt[2], 0, SQL_ADD, SQL_LOCK_NO_CHANGE);
            //ReportError(&m_bd, m_hstmt[2]);
        }
    }
}



STDMETHODIMP_(void) CCDData::DBSaveTitle(LPCDTITLE pCDTitle)
{
    if (pCDTitle)
    {
        RETCODE     rc;
        DWORD       dwIndex;

        BOOL fExist = QueryDatabase(pCDTitle->dwTitleID, gszTitleTable);

        InitCBTable(&m_bd);
        SetCursors( m_hstmt );
        SetBindings( m_hstmt, &m_bd);

        rc = m_pSQL->SetConnectOption(m_hdbc, SQL_AUTOCOMMIT, SQL_AUTOCOMMIT_OFF);
        ReportError(&m_bd, m_hstmt[0]);

        SaveTitle(pCDTitle, fExist);
        SaveTracks(pCDTitle, fExist);
        SaveMenus(pCDTitle);

        rc = m_pSQL->Transact(SQL_NULL_HENV, m_hdbc, SQL_COMMIT);
//        ReportError(&m_bd, m_hstmt[0]);

        rc = m_pSQL->SetConnectOption(m_hdbc, SQL_AUTOCOMMIT, SQL_AUTOCOMMIT_ON);
//        ReportError(&m_bd, m_hstmt[0]);

        for (dwIndex = 0; dwIndex < NUMTABLES; dwIndex++)
        {
            m_pSQL->FreeStmt(m_hstmt[dwIndex], SQL_CLOSE);
        }
    }
}



STDMETHODIMP_(void) CCDData::DBRemoveTitle(LPCDTITLE pCDTitle)
{
    if (pCDTitle && pCDTitle->dwTitleID != CDTITLE_NODISC)
    {
        RETCODE     rc;
        UCHAR       szQuery[MAX_PATH];
        HSTMT       hstmt;

        if (m_pSQL->AllocStmt(m_hdbc, &hstmt) == SQL_SUCCESS)
        {
            wsprintf((TCHAR *) szQuery, TEXT("delete from Titles where TitleID = %d"), pCDTitle->dwTitleID);
            rc = m_pSQL->ExecDirect(hstmt, szQuery, SQL_NTS);

            wsprintf((TCHAR *) szQuery, TEXT("delete from Tracks where TitleID = %d"), pCDTitle->dwTitleID);
            rc = m_pSQL->ExecDirect(hstmt, szQuery, SQL_NTS);
  
            wsprintf((TCHAR *) szQuery, TEXT("delete from Menus where TitleID = %d"), pCDTitle->dwTitleID);
            rc = m_pSQL->ExecDirect(hstmt, szQuery, SQL_NTS);

            m_pSQL->FreeStmt(hstmt, SQL_DROP);
        }
    } 
}





STDMETHODIMP_(void) CCDData::DestroyTitles(LPCDTITLE *ppCDTitles)
{
    if (ppCDTitles && *ppCDTitles)
    {
        LPCDTITLE pNext;
        LPCDTITLE pCDTitle = *ppCDTitles;

        *ppCDTitles = NULL;

        pNext = pCDTitle;

        while (pCDTitle)
        {
            pNext = pCDTitle->pNext;

            if (pCDTitle->dwLockCnt)        // Someone locked a node that was in memory and have unloaded before unlocking
            {                               // We will abandon the node, it will be destroyed when they unlock it.
                pCDTitle->fLoaded = TRUE;
                pCDTitle->pNext = NULL;
            }
            else
            {
                DestroyTitle(pCDTitle);
            }

            pCDTitle = pNext;
        }
    }
}



STDMETHODIMP_(void) CCDData::SaveTitles(LPCDTITLE *ppCDTitles)
{
    if (ppCDTitles && *ppCDTitles)
    {
        LPCDTITLE pLast = NULL;
        LPCDTITLE pNext;
        LPCDTITLE pCDTitle = *ppCDTitles;

        pNext = pCDTitle;

        while (pCDTitle)
        {
            pNext = pCDTitle->pNext;

            if (pCDTitle->fRemove)
            {
                if (pCDTitle->dwTitleID != DWORD(-1))
                {
                    DBRemoveTitle(pCDTitle);
                }

                if (pLast == NULL)
                {
                    *ppCDTitles = pNext;
                }
                else
                {
                    pLast->pNext = pNext;
                }

                DestroyTitle(pCDTitle);
            }
            else 
            {
                if (pCDTitle->fChanged)
                {
                    DBSaveTitle(pCDTitle);
                    pCDTitle->fChanged = FALSE;
                }

                pLast = pCDTitle;
            }

            pCDTitle = pNext;
        }
    }
}





STDMETHODIMP_(void) CCDData::AddTitle(LPCDTITLE *ppCDTitles, LPCDTITLE pCDNewTitle)
{
    if (ppCDTitles && pCDNewTitle)
    {
        if (*ppCDTitles == NULL)        // Inserting into empty list
        {
            *ppCDTitles = pCDNewTitle;
            pCDNewTitle->pNext = NULL;
        }
        else                            // Insertion sort based on artist name (for view by artist feature)
        {
            LPCDTITLE pCDTitle = *ppCDTitles;
            LPCDTITLE pLast = NULL;
            TCHAR *szArtist = pCDNewTitle->szArtist;

            while (pCDTitle)
            {
                if (lstrcmp(pCDTitle->szArtist, szArtist) >= 0) // we'll only sort by artist, the tree control will sort titles for us
                {
                    if (pLast == NULL)              // insert at head of list
                    {
                        pCDNewTitle->pNext = *ppCDTitles;
                        *ppCDTitles = pCDNewTitle;
                    }
                    else
                    {
                        pCDNewTitle->pNext = pCDTitle;
                        pLast->pNext = pCDNewTitle;
                    }
                    
                    break;             
                }  
              
                if (pCDTitle->pNext == NULL)        // Insert on end of list
                {
                    pCDTitle->pNext = pCDNewTitle;
                    pCDNewTitle->pNext = NULL;
                    break;
                }

                pLast = pCDTitle;
                pCDTitle = pCDTitle->pNext;
            }
        }
    }
}


STDMETHODIMP_(LPCDTITLE) CCDData::GetTitleList(void)
{
    LPCDTITLE pCDTitle;

    Enter();

    if (SUCCEEDED(GetSQLPtr(FALSE)))
    {
        pCDTitle = m_pTitleList;
    }

    Leave();

    return(pCDTitle);
}




STDMETHODIMP_(DWORD) CCDData::GetNumBatched(void)
{
    DWORD dwCount = 0;

    Enter();

    if (SUCCEEDED(GetSQLPtr(FALSE)))
    {
        if (m_pBatchList == NULL)       // Batch Not in memory, just query Batch database directly
        {
            dwCount = GetNumRows((UCHAR *) gszBatchTable);
        }
        else                            
        {
            LPCDBATCH pCDBatch = m_pBatchList;

            while (pCDBatch)
            {
                if (!pCDBatch->fRemove)
                {
                    dwCount++;
                }
            
                pCDBatch = pCDBatch->pNext;
            }
        }
    }

    Leave();

    return(dwCount);
}




STDMETHODIMP_(BOOL) CCDData::FindBatchTitle(LPCDBATCH pCDBatchList, DWORD dwTitleID)
{
    BOOL        fFound = FALSE;
    LPCDBATCH   pCDBatch = pCDBatchList;

    while (pCDBatch)
    {
        if (pCDBatch->dwTitleID == dwTitleID)
        {
            if (!pCDBatch->fRemove)
            {
                fFound = TRUE;
            }

            break;
        }
        
        pCDBatch = pCDBatch->pNext;
    }

    return(fFound);
}



STDMETHODIMP_(BOOL) CCDData::QueryBatch(DWORD dwTitleID)
{
    BOOL        fFound = FALSE;

    Enter();

    if (SUCCEEDED(GetSQLPtr(FALSE)))
    {
        if (m_pBatchList == NULL)       // Batch Not in memory, just query Batch database directly
        {
            fFound = QueryDatabase(dwTitleID, gszBatchTable);
        }
        else                            // Batch is in memory, just look here.
        {
            fFound = FindBatchTitle(m_pBatchList, dwTitleID);
        }
    }

    Leave();

    return(fFound);
}


STDMETHODIMP CCDData::ExtractBatch(LPCDBATCH *ppCDBatchList, HWND hWnd)
{
    HRESULT     hr = E_FAIL;
    LPCDBATCH   pCDBatch= NULL;
    TIMEDMETER  tm;

    InitCBTable(&m_bd);
    SetCursors(m_hstmt);
    SetBindings(m_hstmt, &m_bd);

    if (ppCDBatchList == NULL || m_henv == NULL)
    {
        hr = E_INVALIDARG;
    }
    else
    {
        RETCODE rc;
        DWORD dwCount = GetNumRows((UCHAR *) gszBatchTable);

        CreateMeter(&tm, hWnd, dwCount, 100, 0);

        dwCount = 0;

        rc = m_pSQL->ExecDirect(m_hstmt[3], (UCHAR *) TEXT("select * from Batch"), SQL_NTS);

        if (rc == SQL_SUCCESS)
        {
            m_bd.szQuery[0] = TEXT('\0');

            while (m_pSQL->Fetch(m_hstmt[3]) == SQL_SUCCESS)
            {
                pCDBatch = new CDBATCH;

                if (pCDBatch == NULL)
                {
                    hr = E_OUTOFMEMORY;
                    break;
                }
                else
                {
                    DWORD dwSize = lstrlen(m_bd.szQuery) + 1;
                    
                    pCDBatch->szTitleQuery = new(TCHAR[dwSize]);
                    
                    if (pCDBatch->szTitleQuery == NULL)
                    {
                        delete pCDBatch;
                        hr = E_OUTOFMEMORY;
                        break;
                    }  
                    else
                    {
                        lstrcpy(pCDBatch->szTitleQuery, m_bd.szQuery);
                        pCDBatch->dwTitleID = m_bd.CDTitle.dwTitleID;
                        pCDBatch->dwNumTracks = m_bd.CDTitle.dwNumTracks;
                        pCDBatch->fRemove = FALSE;
                        pCDBatch->fFresh = TRUE;
                        pCDBatch->pNext = *ppCDBatchList;
                        *ppCDBatchList = pCDBatch;
                        
                        m_bd.szQuery[0] = TEXT('\0');
                        UpdateMeter(&tm);
                        hr = S_OK;
                    }
                }
            }

            m_pSQL->FreeStmt(m_hstmt[3], SQL_CLOSE);
        }

        DestroyMeter(&tm);
    }

    return(hr);
}








STDMETHODIMP CCDData::LoadBatch(HWND hWnd, LPCDBATCH *ppCDBatchList)
{
    HRESULT hr = E_FAIL;

    Enter();

    if (SUCCEEDED(GetSQLPtr(FALSE)))
    {
        if (ppCDBatchList)
        {
            if (m_pBatchList == NULL)
            {
                hr = ExtractBatch(&m_pBatchList, hWnd);

                if (SUCCEEDED(hr))
                {
                    m_dwBatchCnt++;
                    *ppCDBatchList = m_pBatchList;
                }
            }
        }
    }

    Leave();

    return(hr);
}





STDMETHODIMP_(void) CCDData::DeleteBatch(LPCDBATCH pCDBatch)
{
    RETCODE     rc;
    UCHAR       szQuery[MAX_PATH];
    HSTMT       hstmt;

    if (pCDBatch->fRemove)
    {
        if (m_pSQL->AllocStmt(m_hdbc, &hstmt) == SQL_SUCCESS)
        {
            wsprintf((TCHAR *) szQuery, TEXT("delete from Batch where TitleID = %d"), pCDBatch->dwTitleID);
            rc = m_pSQL->ExecDirect(hstmt, szQuery, SQL_NTS);

            m_pSQL->FreeStmt(hstmt, SQL_DROP);
        }
    }

    if (pCDBatch->szTitleQuery)
    {
        delete pCDBatch->szTitleQuery;
    }

    delete pCDBatch;
}






STDMETHODIMP_(void) CCDData::DestroyBatch(LPCDBATCH *ppCDBatchList)
{
    if (ppCDBatchList && *ppCDBatchList)
    {
        LPCDBATCH pNext;
        LPCDBATCH pCDBatch = *ppCDBatchList;

        *ppCDBatchList = NULL;

        while (pCDBatch)
        {
            pNext = pCDBatch->pNext;           
            DeleteBatch(pCDBatch);
            pCDBatch = pNext;
        }
    }
}




STDMETHODIMP CCDData::UnloadBatch(LPCDBATCH pCDBatchList)
{    
    HRESULT hr = E_FAIL;

    Enter();

    if (SUCCEEDED(GetSQLPtr(FALSE)))
    {
        if (pCDBatchList == m_pBatchList && m_dwBatchCnt)
        {
            m_dwBatchCnt--;

            if (m_dwBatchCnt == 0)
            {
                DestroyBatch(&m_pBatchList);
            }

            hr = S_OK;
        }
    }
    
    Leave();
    
    return(hr);
}







STDMETHODIMP CCDData::DumpBatch(void)
{
    HRESULT hr = E_FAIL;

    Enter();

    if (SUCCEEDED(GetSQLPtr(FALSE)))
    {
        if (m_dwBatchCnt == 0)
        {
            HSTMT       hstmt;

            if (m_pSQL->AllocStmt(m_hdbc, &hstmt) == SQL_SUCCESS)
            {
                m_pSQL->ExecDirect(hstmt, (UCHAR *) TEXT("drop table Batch"), SQL_NTS);
                m_pSQL->ExecDirect(hstmt, (UCHAR *) gszBatchCreate, SQL_NTS);
                m_pSQL->FreeStmt(hstmt, SQL_CLOSE); 

                m_pSQL->FreeStmt(hstmt, SQL_DROP);

                hr = S_OK;
            }
        }
    }

    Leave();

    return (hr);
}


STDMETHODIMP_(void) CCDData::RemoveFromBatch(DWORD dwTitleID)
{
    if (QueryBatch(dwTitleID))
    {
        if (m_pBatchList)
        {
            LPCDBATCH pBatch = m_pBatchList;

            while(pBatch)
            {
                if (pBatch->dwTitleID == dwTitleID)
                {
                    pBatch->fRemove = TRUE;
                    break;
                }

                pBatch = pBatch->pNext;
            }
        }
        else
        {
            RETCODE     rc;
            UCHAR       szQuery[MAX_PATH];
            HSTMT       hstmt;

            if (m_pSQL->AllocStmt(m_hdbc, &hstmt) == SQL_SUCCESS)
            {
                wsprintf((TCHAR *) szQuery, TEXT("delete from Batch where TitleID = %d"), dwTitleID);
                rc = m_pSQL->ExecDirect(hstmt, szQuery, SQL_NTS);

                m_pSQL->FreeStmt(hstmt, SQL_DROP);
            }
        }
    }
}



STDMETHODIMP CCDData::AddToBatch(DWORD dwTitleID, TCHAR *szTitleQuery, DWORD dwNumTracks)
{
    HRESULT hr = E_FAIL;

    Enter();

    if (SUCCEEDED(GetSQLPtr(FALSE)))
    {
        if (!QueryBatch(dwTitleID) && szTitleQuery && (UINT)lstrlen(szTitleQuery) < (sizeof(m_bd.szQuery)/sizeof(TCHAR))) 
        {
            RETCODE     rc;
            DWORD       dwIndex;
            UDWORD      row;
            UWORD       rgfRowStat[ROWSET];

            InitCBTable(&m_bd);
            SetCursors( m_hstmt );
            SetBindings( m_hstmt, &m_bd);

            m_pSQL->SetConnectOption(m_hdbc, SQL_AUTOCOMMIT, SQL_AUTOCOMMIT_OFF);

            rc = m_pSQL->ExecDirect(m_hstmt[3], (UCHAR *) TEXT("select * from Batch"), SQL_NTS);
            ReportError(&m_bd, m_hstmt[3]);

            rc = m_pSQL->ExtendedFetch(m_hstmt[3], SQL_FETCH_LAST, 1, &row, rgfRowStat);
            ReportError(&m_bd, m_hstmt[3]);

            InitCBTable(&m_bd);
            m_bd.CDTitle.dwTitleID = dwTitleID;
            m_bd.CDTitle.dwNumTracks = dwNumTracks;
        
            lstrcpy(m_bd.szQuery, szTitleQuery);

            rc = m_pSQL->SetPos(m_hstmt[3], 0, SQL_ADD, SQL_LOCK_NO_CHANGE);
 //           ReportError(&m_bd, m_hstmt[3]);

            rc = m_pSQL->Transact(SQL_NULL_HENV, m_hdbc, SQL_COMMIT);
//            ReportError(&m_bd, m_hstmt[3]);

            m_pSQL->SetConnectOption(m_hdbc, SQL_AUTOCOMMIT, SQL_AUTOCOMMIT_ON);

            for (dwIndex = 0; dwIndex < NUMTABLES; dwIndex++)
            {
                m_pSQL->FreeStmt(m_hstmt[dwIndex], SQL_CLOSE);
            }

            hr = S_OK;
        }
    }

    Leave();

    return(hr);
}

STDMETHODIMP_(BOOL) CCDData::IsOldFormat()
{
    BOOL fResult = FALSE;
    TCHAR szName[MAX_PATH];
    SQLSMALLINT nName;
    SQLSMALLINT sqlType;
    SQLUINTEGER dwColDef;
    SQLSMALLINT ibScale;
    SQLSMALLINT fNullable;

    RETCODE rc = SQL_SUCCESS;

    rc = m_pSQL->ExecDirect(m_hstmt[0], (UCHAR *) TEXT("select * from Titles"), SQL_NTS);

    if (SQL_SUCCEEDED(rc))
    {
        //check "artist" and "titlequery" columns to find out if this is an old-format db.
        //if either is the old type, then this is an old-format db
        rc = m_pSQL->DescribeCol(m_hstmt[0], 2, (UCHAR*)(&szName), sizeof(szName), &nName, &sqlType, &dwColDef, &ibScale, &fNullable);
        if (SQL_SUCCEEDED(rc))
        {
            if (sqlType != SQL_LONGCHAR_FIELD)
            {
                fResult = TRUE;
            }
        }

        rc = m_pSQL->DescribeCol(m_hstmt[0], 10, (UCHAR*)(&szName), sizeof(szName), &nName, &sqlType, &dwColDef, &ibScale, &fNullable);
        if (SQL_SUCCEEDED(rc))
        {
            if (sqlType == SQL_C_BINARY)
            {
                fResult = TRUE;
            }
        }
    }

    m_pSQL->FreeStmt(m_hstmt[0],SQL_CLOSE);

    return (fResult);
}

STDMETHODIMP CCDData::UpgradeDatabase(HWND hWnd)
{
    HRESULT hr = S_OK;
    HSTMT hstmt = NULL;
    TIMEDMETER tm;

    DWORD dwCount =   GetNumRows((UCHAR *) gszTitleTable)
                    + GetNumRows((UCHAR *) gszBatchTable);

    //steps:
    //1. Read values from old version of database
    //2. Delete the original database
    //3. Create new database with correct data types
    //4. Put data into new database
    //5. Clean up

    //Step 1: Read values from old version of database
    LPCDBATCH pBatchList = NULL;
    LoadTitles(hWnd); //(also gets tracks and menus)
    LoadBatch(hWnd,&pBatchList);

    //Step 2: Delete the original database
    CloseDatabase();

    TCHAR szFileName[MAX_PATH];
    GetWindowsDirectory(szFileName,sizeof(szFileName)/sizeof(TCHAR));
    _tcscat(szFileName,TEXT("\\DeluxeCD.MDB"));

    DeleteFile(szFileName);

    //Step 3: Create new database with correct data types
    CreateDatabase();
    ConnectToDatabase(ODBC_CONFIG_DSN);

    //Step 4: Put data into new database
    CreateMeter(&tm, hWnd, dwCount, 5,  IDS_IMPORTING);

    //put all the titles into the new db
    LPCDTITLE pCDTitle = m_pTitleList;
    while (pCDTitle)
    {
        DBSaveTitle(pCDTitle);
        UpdateMeter(&tm);
        pCDTitle = pCDTitle->pNext;
    }

    //put all the batches into the new db
    LPCDBATCH pBatch = pBatchList;
    while (pBatch)
    {
        AddToBatch(pBatch->dwTitleID,pBatch->szTitleQuery,pBatch->dwNumTracks);
        pBatch->fRemove = FALSE;
        UpdateMeter(&tm);
        pBatch = pBatch->pNext;
    }
   
    //Step 5: Clean up
    UnloadBatch(pBatchList);
    DestroyMeter(&tm);

    return (hr);
}