//--------------------------------------------------------------------------; // // File: cddb.cpp // // Copyright (c) 1998 Microsoft Corporation. All rights reserved // //--------------------------------------------------------------------------; #include "precomp.h" #include "cdopti.h" #include "cddata.h" #include "shlobj.h" #include "shlwapi.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 #undef SQL_C_WCHAR #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, 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 that is correct for storing cd information STDMETHODIMP_(DWORD) CCDData::GetAppDataDir(TCHAR* pstrDir, DWORD cchSize) { DWORD dwRet = 0; TCHAR szDataDir[MAX_PATH]; //get the allusers application folder if (SUCCEEDED(SHGetFolderPath(NULL,CSIDL_APPDATA,NULL,SHGFP_TYPE_CURRENT,pstrDir))) { if (PathAppend(pstrDir,TEXT("Microsoft\\CD Player"))) { SHCreateDirectoryExW(NULL, pstrDir, NULL); //shorten the pathname, because ODBC can't handle spaces TCHAR szDir[MAX_PATH]; _tcscpy(szDir,pstrDir); dwRet = GetShortPathName(szDir,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(); } INT_PTR 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); } //helper function to convert ascii hex to unicode chars void TranslateToUnicode(TCHAR* pszTrans) { TCHAR* pszTrans2 = pszTrans; int nLen = _tcslen(pszTrans); int x = 0; while (!((pszTrans2[0] == TEXT('0')) && (pszTrans2[1] == TEXT('0')))) { TCHAR CH; CH = ( pszTrans2 [0] >= TEXT('A') ? ( ( pszTrans2 [0] & 0xDF ) - TEXT('A') ) + 10 : ( pszTrans2 [0] - TEXT('0') ) ); CH *= 16; CH += ( pszTrans2 [1] >= TEXT('A') ? ( ( pszTrans2 [1] & 0xDF ) - TEXT('A') ) + 10 : ( pszTrans2 [1] - TEXT('0') ) ); pszTrans[x++] = CH; pszTrans2 += 2; //get out of the loop if we're at the end of the buffer's length if (x >= nLen) { break; } } pszTrans[x] = 0; } 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); //need to copy the batch list to ensure that it isn't already in cached in memory, //and therefore rejected by the "AddToBatch" call LPCDBATCH pBatchHead = NULL; LPCDBATCH pBatchTail = NULL; LPCDBATCH pBatch = pBatchList; while (pBatch) { LPCDBATCH pNewBatch = new CDBATCH; if (!pNewBatch) return E_OUTOFMEMORY; memcpy(pNewBatch,pBatch,sizeof(CDBATCH)); pNewBatch->szTitleQuery = new TCHAR[lstrlen(pBatch->szTitleQuery)+1]; if (!(pNewBatch->szTitleQuery)) return E_OUTOFMEMORY; _tcscpy(pNewBatch->szTitleQuery,pBatch->szTitleQuery); if (!pBatchHead) { pBatchHead = pNewBatch; pBatchTail = pNewBatch; } else { pBatchTail->pNext = pNewBatch; pBatchTail = pNewBatch; } pNewBatch->pNext = NULL; pBatch = pBatch->pNext; } UnloadBatch(pBatchList); //Step 2: Delete the original database CloseDatabase(); TCHAR szFileName[MAX_PATH]; GetAppDataDir(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) { TranslateToUnicode(pCDTitle->szTitleQuery); for (DWORD x = 0; x < pCDTitle->dwNumMenus; x++) { TranslateToUnicode(pCDTitle->pMenuTable[x].szMenuQuery); } DBSaveTitle(pCDTitle); UpdateMeter(&tm); pCDTitle = pCDTitle->pNext; } //put all the batches into the new db pBatch = pBatchHead; while (pBatch) { TranslateToUnicode(pBatch->szTitleQuery); AddToBatch(pBatch->dwTitleID,pBatch->szTitleQuery,pBatch->dwNumTracks); pBatch->fRemove = FALSE; UpdateMeter(&tm); LPCDBATCH pBatchToDelete = pBatch; pBatch = pBatch->pNext; delete pBatchToDelete->szTitleQuery; delete pBatchToDelete; } //Step 5: Clean up DestroyMeter(&tm); return (hr); }