mirror of https://github.com/tongzx/nt5src
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
1567 lines
44 KiB
1567 lines
44 KiB
/******************************Module*Header*******************************\
|
|
* Module Name: getinfo.cpp
|
|
*
|
|
* Author: David Stewart [dstewart]
|
|
*
|
|
* Copyright (c) 1998 Microsoft Corporation. All rights reserved.
|
|
\**************************************************************************/
|
|
|
|
#include <TCHAR.H>
|
|
#include <objbase.h>
|
|
#include <mmsystem.h> //for mci commands
|
|
#include <urlmon.h>
|
|
#include <hlguids.h> //for IID_IBindStatusCallback
|
|
#include "getinfo.h"
|
|
#include "netres.h"
|
|
#include "wininet.h"
|
|
#include "condlg.h"
|
|
#include "..\main\mmfw.h"
|
|
#include "..\cdopt\cdopt.h"
|
|
#include "mapi.h"
|
|
#include <stdio.h>
|
|
|
|
extern HINSTANCE g_dllInst;
|
|
|
|
#define MODE_OK 0
|
|
#define MODE_MULTIPLE 1
|
|
#define MODE_NOT_FOUND 2
|
|
|
|
#define FRAMES_PER_SECOND 75
|
|
#define FRAMES_PER_MINUTE (60*FRAMES_PER_SECOND)
|
|
#define MAX_UPLOAD_URL_LENGTH 1500
|
|
|
|
#ifdef UNICODE
|
|
#define URLFUNCTION "URLOpenStreamW"
|
|
#define CANONFUNCTION "InternetCanonicalizeUrlW"
|
|
#else
|
|
#define URLFUNCTION "URLOpenStreamA"
|
|
#define CANONFUNCTION "InternetCanonicalizeUrlA"
|
|
#endif
|
|
|
|
HWND g_hwndParent = NULL;
|
|
extern HINSTANCE g_hURLMon;
|
|
LPCDOPT g_pNetOpt = NULL;
|
|
LPCDDATA g_pNetData = NULL;
|
|
BOOL g_fCancelDownload = FALSE; //ANY ACCESS MUST BE SURROUNDED by Enter/Leave g_Critical
|
|
IBinding* g_pBind = NULL; //ANY ACCESS MUST BE SURROUNDED by Enter/Leave g_Critical
|
|
long g_lNumDownloadingThreads = 0; //MUST USE InterlockedIncrement/Decrement
|
|
BOOL g_fDownloadDone = FALSE;
|
|
BOOL g_fDBWriteFailure = FALSE;
|
|
extern CRITICAL_SECTION g_Critical;
|
|
extern CRITICAL_SECTION g_BatchCrit;
|
|
|
|
DWORD WINAPI SpawnSingleDownload(LPVOID pParam);
|
|
DWORD WINAPI SpawnBatchDownload(LPVOID pParam);
|
|
DWORD WINAPI DoBatchDownload(LPCDBATCH pBatchList, HWND hwndParent);
|
|
BOOL DoDownload(TCHAR* url, TCHAR* szFilename, HWND hwndParent);
|
|
|
|
CCDNet::CCDNet()
|
|
{
|
|
m_dwRef = 0;
|
|
}
|
|
|
|
CCDNet::~CCDNet()
|
|
{
|
|
}
|
|
|
|
STDMETHODIMP CCDNet::QueryInterface(REFIID riid, void** ppv)
|
|
{
|
|
*ppv = NULL;
|
|
if (IID_IUnknown == riid || IID_ICDNet == riid)
|
|
{
|
|
*ppv = this;
|
|
}
|
|
|
|
if (NULL==*ppv)
|
|
{
|
|
return E_NOINTERFACE;
|
|
}
|
|
|
|
AddRef();
|
|
|
|
return S_OK;
|
|
}
|
|
|
|
STDMETHODIMP_(ULONG) CCDNet::AddRef(void)
|
|
{
|
|
return ++m_dwRef;
|
|
}
|
|
|
|
STDMETHODIMP_(ULONG) CCDNet::Release(void)
|
|
{
|
|
if (0!=--m_dwRef)
|
|
return m_dwRef;
|
|
|
|
delete this;
|
|
return 0;
|
|
}
|
|
|
|
STDMETHODIMP CCDNet::SetOptionsAndData(void* pOpts, void* pData)
|
|
{
|
|
g_pNetOpt = (LPCDOPT)pOpts;
|
|
g_pNetData = (LPCDDATA)pData;
|
|
|
|
return S_OK;
|
|
}
|
|
|
|
//this is a start to implementing the "upload via http" case rather than the
|
|
//upload via mail case
|
|
BOOL UploadToProvider(LPCDPROVIDER pProvider, LPCDTITLE pTitle, HWND hwndParent)
|
|
{
|
|
TCHAR szURL[INTERNET_MAX_URL_LENGTH];
|
|
TCHAR szMainURL[INTERNET_MAX_URL_LENGTH];
|
|
TCHAR szFilename[MAX_PATH];
|
|
|
|
//get the InternetCanonicalizeURL function
|
|
typedef BOOL (PASCAL *CANPROC)(LPCTSTR, LPTSTR, LPDWORD, DWORD);
|
|
CANPROC canProc = NULL;
|
|
|
|
HMODULE hNet = LoadLibrary(TEXT("WININET.DLL"));
|
|
if (hNet!=NULL)
|
|
{
|
|
canProc = (CANPROC)GetProcAddress(hNet,CANONFUNCTION);
|
|
}
|
|
else
|
|
{
|
|
return FALSE;
|
|
}
|
|
|
|
if (pProvider && pTitle && canProc)
|
|
{
|
|
//check for provider URL
|
|
if (_tcslen(pProvider->szProviderUpload)>0)
|
|
{
|
|
TCHAR szTempCan[MAX_PATH*2];
|
|
|
|
//create the URL to send
|
|
wsprintf(szMainURL,TEXT("%s%s"),pProvider->szProviderUpload,pTitle->szTitleQuery);
|
|
_tcscpy(szURL,szMainURL);
|
|
|
|
//add title
|
|
_tcscat(szURL,TEXT("&t="));
|
|
DWORD dwSize = sizeof(szTempCan);
|
|
canProc(pTitle->szTitle,szTempCan,&dwSize,0);
|
|
_tcscat(szURL,szTempCan);
|
|
|
|
//add artist
|
|
_tcscat(szURL,TEXT("&a="));
|
|
dwSize = sizeof(szTempCan);
|
|
canProc(pTitle->szArtist,szTempCan,&dwSize,0);
|
|
_tcscat(szURL,szTempCan);
|
|
|
|
//add tracks
|
|
TCHAR szTrack[MAX_PATH];
|
|
for (DWORD i = 0; i < pTitle->dwNumTracks; i++)
|
|
{
|
|
wsprintf(szTrack,TEXT("&%u="),i+1);
|
|
|
|
dwSize = sizeof(szTempCan);
|
|
canProc(pTitle->pTrackTable[i].szName,szTempCan,&dwSize,0);
|
|
|
|
if ((_tcslen(szURL) + _tcslen(szTrack) + _tcslen(szTempCan)) >
|
|
MAX_UPLOAD_URL_LENGTH-sizeof(TCHAR))
|
|
{
|
|
//we're coming close to the limit. Send what we have and start rebuilding
|
|
if (!g_fCancelDownload)
|
|
{
|
|
if (DoDownload(szURL,szFilename, hwndParent))
|
|
{
|
|
DeleteFile(szFilename);
|
|
} //end if "upload" successful
|
|
else
|
|
{
|
|
//bad upload, don't bother sending the rest
|
|
//probably a timeout
|
|
return FALSE;
|
|
}
|
|
}
|
|
|
|
//reset the URL to just the provider + toc
|
|
_tcscpy(szURL,szMainURL);
|
|
} //end if length
|
|
|
|
_tcscat(szURL,szTrack);
|
|
_tcscat(szURL,szTempCan);
|
|
} //end for track
|
|
|
|
//send it
|
|
if (!g_fCancelDownload)
|
|
{
|
|
if (DoDownload(szURL,szFilename, hwndParent))
|
|
{
|
|
DeleteFile(szFilename);
|
|
} //end if "upload" successful
|
|
else
|
|
{
|
|
return FALSE;
|
|
}
|
|
}
|
|
} //end if url exists
|
|
} //end if state OK
|
|
|
|
if (hNet)
|
|
{
|
|
FreeLibrary(hNet);
|
|
}
|
|
|
|
return TRUE;
|
|
}
|
|
|
|
DWORD WINAPI UploadThread(LPVOID pParam)
|
|
{
|
|
InterlockedIncrement((LONG*)&g_lNumDownloadingThreads);
|
|
|
|
//this will block us against the batch download happening, too
|
|
EnterCriticalSection(&g_BatchCrit);
|
|
|
|
LPCDTITLE pTitle = (LPCDTITLE)pParam;
|
|
HWND hwndParent = g_hwndParent;
|
|
|
|
int nTries = 0;
|
|
int nSuccessful = 0;
|
|
|
|
if (pTitle)
|
|
{
|
|
LPCDOPTIONS pOptions = g_pNetOpt->GetCDOpts(); // Get the options, needed for provider list
|
|
|
|
if (pOptions && pOptions->pCurrentProvider) // Make sure we have providers
|
|
{
|
|
LPCDPROVIDER pProviderList = NULL;
|
|
LPCDPROVIDER pProvider = NULL;
|
|
|
|
g_pNetOpt->CreateProviderList(&pProviderList); // Get the sorted provider list
|
|
pProvider = pProviderList; // Get the head of the list
|
|
|
|
while ((pProvider) && (!g_fCancelDownload))
|
|
{
|
|
nTries++;
|
|
if (UploadToProvider(pProvider,pTitle,hwndParent))
|
|
{
|
|
nSuccessful++;
|
|
}
|
|
pProvider = pProvider->pNext;
|
|
}
|
|
|
|
g_pNetOpt->DestroyProviderList(&pProviderList);
|
|
} //end if providers
|
|
}
|
|
|
|
//addref'ed before thread was created
|
|
g_pNetOpt->Release();
|
|
g_pNetData->Release();
|
|
|
|
long status = UPLOAD_STATUS_NO_PROVIDERS;
|
|
|
|
if ((nSuccessful != nTries) && (nSuccessful > 0))
|
|
{
|
|
status = UPLOAD_STATUS_SOME_PROVIDERS;
|
|
}
|
|
|
|
if ((nSuccessful == nTries) && (nSuccessful > 0))
|
|
{
|
|
status = UPLOAD_STATUS_ALL_PROVIDERS;
|
|
}
|
|
|
|
if (g_fCancelDownload)
|
|
{
|
|
status = UPLOAD_STATUS_CANCELED;
|
|
}
|
|
|
|
LeaveCriticalSection(&g_BatchCrit);
|
|
InterlockedDecrement((LONG*)&g_lNumDownloadingThreads);
|
|
|
|
//post message saying we're done
|
|
PostMessage(hwndParent,WM_NET_DONE,(WPARAM)g_dllInst,status);
|
|
|
|
return 0;
|
|
}
|
|
|
|
STDMETHODIMP CCDNet::Upload(LPCDTITLE pTitle, HWND hwndParent)
|
|
{
|
|
HRESULT hr = E_FAIL;
|
|
DWORD dwHow;
|
|
BOOL fConnected;
|
|
|
|
if (g_pNetOpt && g_pNetData && pTitle) // Make sure we are in a valid state
|
|
{
|
|
fConnected = _InternetGetConnectedState(&dwHow,0,TRUE); // Make sure we are connected to net
|
|
|
|
if (fConnected) // Make sure we are in a valid state
|
|
{
|
|
EnterCriticalSection(&g_Critical);
|
|
g_fCancelDownload = FALSE;
|
|
LeaveCriticalSection(&g_Critical);
|
|
|
|
DWORD dwThreadID;
|
|
HANDLE hNetThread = NULL;
|
|
|
|
g_hwndParent = hwndParent;
|
|
g_pNetOpt->AddRef();
|
|
g_pNetData->AddRef();
|
|
hNetThread = CreateThread(NULL,0,UploadThread,(void*)pTitle,0,&dwThreadID);
|
|
if (hNetThread)
|
|
{
|
|
CloseHandle(hNetThread);
|
|
hr = S_OK;
|
|
}
|
|
} //end if connected
|
|
} //end if options and data ok
|
|
|
|
return (hr);
|
|
}
|
|
|
|
STDMETHODIMP_(BOOL) CCDNet::CanUpload()
|
|
{
|
|
BOOL retcode = FALSE;
|
|
|
|
if (g_pNetOpt && g_pNetData) // Make sure we are in a valid state
|
|
{
|
|
//check all providers to be sure at least one has upload capability
|
|
LPCDOPTIONS pOptions = g_pNetOpt->GetCDOpts(); // Get the options, needed for provider list
|
|
|
|
if (pOptions && pOptions->pCurrentProvider) // Make sure we have providers
|
|
{
|
|
LPCDPROVIDER pProviderList = NULL;
|
|
LPCDPROVIDER pProvider = NULL;
|
|
|
|
g_pNetOpt->CreateProviderList(&pProviderList); // Get the sorted provider list
|
|
pProvider = pProviderList; // Get the head of the list
|
|
|
|
while (pProvider)
|
|
{
|
|
if (_tcslen(pProvider->szProviderUpload) > 0)
|
|
{
|
|
retcode = TRUE;
|
|
}
|
|
pProvider = pProvider->pNext;
|
|
} //end while
|
|
|
|
g_pNetOpt->DestroyProviderList(&pProviderList);
|
|
} //end if providers
|
|
} //end if set up properly
|
|
|
|
return (retcode);
|
|
}
|
|
|
|
STDMETHODIMP CCDNet::Download(DWORD dwDeviceHandle, TCHAR chDrive, DWORD dwMSID, LPCDTITLE pTitle, BOOL fManual, HWND hwndParent)
|
|
{
|
|
if (g_pNetOpt==NULL)
|
|
{
|
|
return E_FAIL;
|
|
}
|
|
|
|
if (g_pNetData==NULL)
|
|
{
|
|
return E_FAIL;
|
|
}
|
|
|
|
if (FAILED(g_pNetData->CheckDatabase(hwndParent)))
|
|
{
|
|
return E_FAIL;
|
|
}
|
|
|
|
CGetInfoFromNet netinfo(dwDeviceHandle,
|
|
dwMSID,
|
|
hwndParent);
|
|
|
|
EnterCriticalSection(&g_Critical);
|
|
g_fCancelDownload = FALSE;
|
|
LeaveCriticalSection(&g_Critical);
|
|
|
|
BOOL fResult = netinfo.DoIt(fManual, pTitle, chDrive);
|
|
|
|
return fResult ? S_OK : E_FAIL;
|
|
}
|
|
|
|
STDMETHODIMP_(BOOL) CCDNet::IsDownloading()
|
|
{
|
|
BOOL retcode = FALSE;
|
|
|
|
if (g_lNumDownloadingThreads > 0)
|
|
{
|
|
retcode = TRUE;
|
|
}
|
|
|
|
return (retcode);
|
|
}
|
|
|
|
STDMETHODIMP CCDNet::CancelDownload()
|
|
{
|
|
EnterCriticalSection(&g_Critical);
|
|
if (g_pBind)
|
|
{
|
|
g_pBind->Abort();
|
|
}
|
|
g_fCancelDownload = TRUE;
|
|
LeaveCriticalSection(&g_Critical);
|
|
while (IsDownloading())
|
|
{
|
|
Sleep(10);
|
|
}
|
|
|
|
return S_OK;
|
|
}
|
|
|
|
struct CBindStatusCallback : IBindStatusCallback
|
|
{
|
|
///// object state
|
|
ULONG m_cRef; // object reference count
|
|
BOOL m_fAbort; // set to true if we want this to abort
|
|
HWND m_hMessage; // callback window
|
|
IStream* m_pStream; // holds downloaded data
|
|
|
|
///// construction and destruction
|
|
CBindStatusCallback(IStream* pStream, HWND hwndParent);
|
|
~CBindStatusCallback();
|
|
|
|
///// IUnknown methods
|
|
STDMETHODIMP QueryInterface(REFIID riid, LPVOID *ppvObj);
|
|
STDMETHODIMP_(ULONG) AddRef();
|
|
STDMETHODIMP_(ULONG) Release();
|
|
|
|
///// IBindStatusCallback methods
|
|
STDMETHODIMP OnStartBinding(DWORD dwReserved, IBinding *pib);
|
|
STDMETHODIMP GetPriority(LONG *pnPriority);
|
|
STDMETHODIMP OnLowResource(DWORD reserved);
|
|
STDMETHODIMP OnProgress(ULONG ulProgress, ULONG ulProgressMax,
|
|
ULONG ulStatusCode, LPCWSTR szStatusText);
|
|
STDMETHODIMP OnStopBinding(HRESULT hresult, LPCWSTR szError);
|
|
STDMETHODIMP GetBindInfo(DWORD *grfBINDF, BINDINFO *pbindinfo);
|
|
STDMETHODIMP OnDataAvailable(DWORD grfBSCF, DWORD dwSize,
|
|
FORMATETC *pformatetc, STGMEDIUM *pstgmed);
|
|
STDMETHODIMP OnObjectAvailable(REFIID riid, IUnknown *punk);
|
|
};
|
|
|
|
/////////////////////////////////////////////////////////////////////////////
|
|
// CBindStatusCallback Creation & Destruction
|
|
//
|
|
CBindStatusCallback::CBindStatusCallback(IStream* pStream, HWND hwndParent)
|
|
{
|
|
HRESULT hr = S_OK;
|
|
m_cRef = 0;
|
|
m_fAbort = FALSE;
|
|
m_pStream = pStream;
|
|
m_pStream->AddRef();
|
|
m_hMessage = hwndParent;
|
|
|
|
PostMessage(m_hMessage,WM_NET_STATUS,(WPARAM)g_dllInst,IDS_STRING_CONNECTING);
|
|
}
|
|
|
|
|
|
CBindStatusCallback::~CBindStatusCallback()
|
|
{
|
|
EnterCriticalSection(&g_Critical);
|
|
if (g_pBind)
|
|
{
|
|
g_pBind->Release();
|
|
g_pBind = NULL;
|
|
}
|
|
LeaveCriticalSection(&g_Critical);
|
|
|
|
if( m_pStream )
|
|
{
|
|
m_pStream->Release();
|
|
m_pStream = NULL;
|
|
}
|
|
}
|
|
|
|
/////////////////////////////////////////////////////////////////////////////
|
|
// CBindStatusCallback IUnknown Methods
|
|
//
|
|
|
|
STDMETHODIMP CBindStatusCallback::QueryInterface(REFIID riid, LPVOID *ppvObj)
|
|
{
|
|
if (IsEqualIID(riid, IID_IUnknown) ||
|
|
IsEqualIID(riid, IID_IBindStatusCallback))
|
|
{
|
|
*ppvObj = (IBindStatusCallback *) this;
|
|
AddRef();
|
|
return NOERROR;
|
|
}
|
|
else
|
|
{
|
|
*ppvObj = NULL;
|
|
return E_NOINTERFACE;
|
|
}
|
|
}
|
|
|
|
STDMETHODIMP_(ULONG) CBindStatusCallback::AddRef()
|
|
{
|
|
InterlockedIncrement((LONG*)&m_cRef);
|
|
return m_cRef;
|
|
}
|
|
|
|
STDMETHODIMP_(ULONG) CBindStatusCallback::Release()
|
|
{
|
|
ULONG cRef = m_cRef;
|
|
if (InterlockedDecrement((LONG*)&m_cRef) == 0)
|
|
{
|
|
delete this;
|
|
return 0;
|
|
}
|
|
else
|
|
return cRef-1;
|
|
}
|
|
|
|
|
|
/////////////////////////////////////////////////////////////////////////////
|
|
// CBindStatusCallback IBindStatusCallback Methods
|
|
//
|
|
|
|
STDMETHODIMP CBindStatusCallback::OnStartBinding(DWORD dwReserved, IBinding *pib)
|
|
{
|
|
EnterCriticalSection(&g_Critical);
|
|
g_pBind = pib;
|
|
g_pBind->AddRef();
|
|
LeaveCriticalSection(&g_Critical);
|
|
return S_OK;
|
|
}
|
|
|
|
STDMETHODIMP CBindStatusCallback::GetPriority(LONG *pnPriority)
|
|
{
|
|
return S_OK;
|
|
}
|
|
|
|
STDMETHODIMP CBindStatusCallback::OnLowResource(DWORD reserved)
|
|
{
|
|
return S_OK;
|
|
}
|
|
|
|
STDMETHODIMP CBindStatusCallback::OnProgress(ULONG ulProgress, ULONG ulProgressMax,
|
|
ULONG ulStatusCode, LPCWSTR szStatusText)
|
|
{
|
|
int nResID = 0;
|
|
|
|
switch (ulStatusCode)
|
|
{
|
|
case (BINDSTATUS_FINDINGRESOURCE) : nResID = IDS_STRING_FINDINGRESOURCE; break;
|
|
case (BINDSTATUS_CONNECTING) : nResID = IDS_STRING_CONNECTING; break;
|
|
case (BINDSTATUS_REDIRECTING) : nResID = IDS_STRING_REDIRECTING; break;
|
|
case (BINDSTATUS_BEGINDOWNLOADDATA) : nResID = IDS_STRING_BEGINDOWNLOAD; break;
|
|
case (BINDSTATUS_DOWNLOADINGDATA) : nResID = IDS_STRING_DOWNLOAD; break;
|
|
case (BINDSTATUS_ENDDOWNLOADDATA) : nResID = IDS_STRING_ENDDOWNLOAD; break;
|
|
case (BINDSTATUS_SENDINGREQUEST) : nResID = IDS_STRING_SENDINGREQUEST; break;
|
|
} //end switch
|
|
|
|
if (nResID > 0)
|
|
{
|
|
PostMessage(m_hMessage,WM_NET_STATUS,(WPARAM)g_dllInst,nResID);
|
|
}
|
|
|
|
if (( m_fAbort ) || (g_fCancelDownload))
|
|
{
|
|
EnterCriticalSection(&g_Critical);
|
|
g_fCancelDownload = TRUE;
|
|
g_fDownloadDone = TRUE;
|
|
LeaveCriticalSection(&g_Critical);
|
|
return E_ABORT;
|
|
}
|
|
return S_OK;
|
|
}
|
|
|
|
STDMETHODIMP CBindStatusCallback::OnStopBinding(HRESULT hresult, LPCWSTR szError)
|
|
{
|
|
EnterCriticalSection(&g_Critical);
|
|
if (g_pBind)
|
|
{
|
|
g_pBind->Release();
|
|
g_pBind = NULL;
|
|
}
|
|
LeaveCriticalSection(&g_Critical);
|
|
return S_OK;
|
|
}
|
|
|
|
|
|
STDMETHODIMP CBindStatusCallback::GetBindInfo(DWORD *pgrfBINDF, BINDINFO *pbindinfo)
|
|
{
|
|
*pgrfBINDF = BINDF_ASYNCHRONOUS | BINDF_ASYNCSTORAGE;
|
|
pbindinfo->cbSize = sizeof(BINDINFO);
|
|
pbindinfo->szExtraInfo = NULL;
|
|
ZeroMemory(&pbindinfo->stgmedData, sizeof(STGMEDIUM));
|
|
pbindinfo->grfBindInfoF = 0;
|
|
pbindinfo->dwBindVerb = BINDVERB_GET;
|
|
pbindinfo->szCustomVerb = NULL;
|
|
|
|
return S_OK;
|
|
}
|
|
|
|
STDMETHODIMP CBindStatusCallback::OnDataAvailable(DWORD grfBSCF, DWORD dwSize,
|
|
FORMATETC *pformatetc, STGMEDIUM *pstgmed)
|
|
{
|
|
// fill our stream with the data from the stream passed to us
|
|
if( m_pStream )
|
|
{
|
|
ULARGE_INTEGER cb;
|
|
|
|
cb.LowPart = dwSize;
|
|
cb.HighPart = 0;
|
|
if( pstgmed && pstgmed->pstm )
|
|
{
|
|
pstgmed->pstm->CopyTo( m_pStream, cb, NULL, NULL );
|
|
}
|
|
}
|
|
|
|
// Notify owner when download is complete
|
|
if( grfBSCF & BSCF_LASTDATANOTIFICATION )
|
|
{
|
|
g_fDownloadDone = TRUE;
|
|
|
|
if( m_pStream )
|
|
{
|
|
m_pStream->Release();
|
|
m_pStream = NULL;
|
|
}
|
|
}
|
|
return S_OK;
|
|
}
|
|
|
|
STDMETHODIMP CBindStatusCallback::OnObjectAvailable(REFIID riid, IUnknown *punk)
|
|
{
|
|
return E_NOTIMPL;
|
|
}
|
|
|
|
/////////////////////////////////////////////////////////////////////////////
|
|
// CGetInfoFromNet
|
|
|
|
CGetInfoFromNet::CGetInfoFromNet(DWORD cdrom, DWORD dwMSID, HWND hwndParent)
|
|
{
|
|
DevHandle = cdrom;
|
|
m_MS = dwMSID;
|
|
g_hwndParent = hwndParent;
|
|
}
|
|
|
|
CGetInfoFromNet::~CGetInfoFromNet()
|
|
{
|
|
}
|
|
|
|
BOOL CGetInfoFromNet::DoIt(BOOL fManual, LPCDTITLE pTitle, TCHAR chDrive)
|
|
{
|
|
BOOL fRet = FALSE;
|
|
|
|
int nMode = CONNECTION_GETITNOW;
|
|
|
|
if (!fManual)
|
|
{
|
|
if (g_lNumDownloadingThreads == 0)
|
|
{
|
|
//if no threads are running already,
|
|
//check the connection, possibly prompting the user
|
|
nMode = ConnectionCheck(g_hwndParent,g_pNetOpt, chDrive);
|
|
}
|
|
}
|
|
|
|
if (nMode == CONNECTION_DONOTHING)
|
|
{
|
|
return FALSE;
|
|
}
|
|
|
|
//if passed-in ID is not > 0, then we don't want to scan current disc
|
|
if ((m_MS > 0) && (pTitle == NULL))
|
|
{
|
|
m_Tracks = readtoc();
|
|
|
|
if (m_Tracks > 0)
|
|
{
|
|
BuildQuery();
|
|
}
|
|
} //if msid is greater than 0
|
|
|
|
if (nMode == CONNECTION_BATCH)
|
|
{
|
|
if (m_MS > 0)
|
|
{
|
|
AddToBatch(m_Tracks,m_Query);
|
|
}
|
|
return FALSE;
|
|
}
|
|
|
|
//we need to determine now whether we spawn a batching thread or a single-item downloader
|
|
g_fDBWriteFailure = FALSE;
|
|
DWORD dwThreadID;
|
|
HANDLE hNetThread = NULL;
|
|
|
|
//addref the global pointers before entering the thread
|
|
g_pNetOpt->AddRef();
|
|
g_pNetData->AddRef();
|
|
|
|
if (m_MS > 0)
|
|
{
|
|
//need to create a batch item for this thread to use
|
|
LPCDBATCH pBatch = new CDBATCH;
|
|
pBatch->fRemove = FALSE;
|
|
pBatch->fFresh = TRUE;
|
|
pBatch->pNext = NULL;
|
|
|
|
if (!pTitle)
|
|
{
|
|
pBatch->dwTitleID = m_MS;
|
|
pBatch->dwNumTracks = m_Tracks;
|
|
pBatch->szTitleQuery = new TCHAR[_tcslen(m_Query)+1];
|
|
_tcscpy(pBatch->szTitleQuery,m_Query);
|
|
}
|
|
else
|
|
{
|
|
pBatch->dwTitleID = pTitle->dwTitleID;
|
|
pBatch->dwNumTracks = pTitle->dwNumTracks;
|
|
if (pTitle->szTitleQuery)
|
|
{
|
|
pBatch->szTitleQuery = new TCHAR[_tcslen(pTitle->szTitleQuery)+1];
|
|
_tcscpy(pBatch->szTitleQuery,pTitle->szTitleQuery);
|
|
}
|
|
else
|
|
{
|
|
pBatch->szTitleQuery = new TCHAR[_tcslen(m_Query)+1];
|
|
_tcscpy(pBatch->szTitleQuery,m_Query);
|
|
}
|
|
}
|
|
|
|
hNetThread = CreateThread(NULL,0,SpawnSingleDownload,(void*)pBatch,0,&dwThreadID);
|
|
}
|
|
else
|
|
{
|
|
hNetThread = CreateThread(NULL,0,SpawnBatchDownload,(void*)NULL,0,&dwThreadID);
|
|
}
|
|
|
|
if (hNetThread)
|
|
{
|
|
CloseHandle(hNetThread);
|
|
fRet = TRUE;
|
|
}
|
|
|
|
return (fRet);
|
|
}
|
|
|
|
int CGetInfoFromNet::readtoc()
|
|
{
|
|
DWORD dwRet;
|
|
MCI_SET_PARMS mciSet;
|
|
|
|
ZeroMemory( &mciSet, sizeof(mciSet) );
|
|
|
|
mciSet.dwTimeFormat = MCI_FORMAT_MSF;
|
|
mciSendCommand( DevHandle, MCI_SET, MCI_SET_TIME_FORMAT, (DWORD_PTR)(LPVOID)&mciSet );
|
|
|
|
MCI_STATUS_PARMS mciStatus;
|
|
long lAddress, lStartPos, lDiskLen;
|
|
int i;
|
|
|
|
ZeroMemory( &mciStatus, sizeof(mciStatus) );
|
|
mciStatus.dwItem = MCI_STATUS_NUMBER_OF_TRACKS;
|
|
|
|
//
|
|
// NOTE: none of the mciSendCommand calls below bother to check the
|
|
// return code. This is asking for trouble... but if the
|
|
// commands fail we cannot do much about it.
|
|
//
|
|
dwRet = mciSendCommand( DevHandle, MCI_STATUS,
|
|
MCI_STATUS_ITEM, (DWORD_PTR)(LPVOID)&mciStatus);
|
|
|
|
int tracks = -1;
|
|
tracks = (UCHAR)mciStatus.dwReturn;
|
|
|
|
mciStatus.dwItem = MCI_STATUS_POSITION;
|
|
for ( i = 0; i < tracks; i++ )
|
|
{
|
|
|
|
mciStatus.dwTrack = i + 1;
|
|
dwRet = mciSendCommand( DevHandle, MCI_STATUS,
|
|
MCI_STATUS_ITEM | MCI_TRACK,
|
|
(DWORD_PTR)(LPVOID)&mciStatus);
|
|
|
|
lAddress = (long)mciStatus.dwReturn;
|
|
|
|
//converts "packed" time into pure frames
|
|
lAddress = (MCI_MSF_MINUTE(lAddress) * FRAMES_PER_MINUTE) +
|
|
(MCI_MSF_SECOND(lAddress) * FRAMES_PER_SECOND) +
|
|
(MCI_MSF_FRAME( lAddress));
|
|
|
|
m_toc[i] = lAddress;
|
|
|
|
if (i==0)
|
|
{
|
|
lStartPos = lAddress;
|
|
}
|
|
}
|
|
|
|
mciStatus.dwItem = MCI_STATUS_LENGTH;
|
|
dwRet = mciSendCommand( DevHandle, MCI_STATUS,
|
|
MCI_STATUS_ITEM, (DWORD_PTR)(LPVOID)&mciStatus);
|
|
|
|
/*
|
|
** Convert the total disk length into frames
|
|
*/
|
|
lAddress = (long)mciStatus.dwReturn;
|
|
lDiskLen = (MCI_MSF_MINUTE(lAddress) * FRAMES_PER_MINUTE) +
|
|
(MCI_MSF_SECOND(lAddress) * FRAMES_PER_SECOND) +
|
|
(MCI_MSF_FRAME( lAddress));
|
|
|
|
/*
|
|
** Now, determine the absolute start position of the sentinel
|
|
** track. That is, the special track that marks the end of the
|
|
** disk.
|
|
*/
|
|
lAddress = lStartPos + lDiskLen + 1; //dstewart: add one for true time
|
|
|
|
m_toc[i] = lAddress;
|
|
|
|
return (tracks);
|
|
}
|
|
|
|
void CGetInfoFromNet::BuildQuery()
|
|
{
|
|
wsprintf(m_Query,TEXT("cd=%X"),m_Tracks);
|
|
|
|
//add each frame stattime to query, include end time of disc
|
|
TCHAR tempstr[MAX_PATH];
|
|
for (int i = 0; i < m_Tracks+1; i++)
|
|
{
|
|
wsprintf(tempstr,TEXT("+%X"),m_toc[i]);
|
|
_tcscat(m_Query,tempstr);
|
|
}
|
|
}
|
|
|
|
void CGetInfoFromNet::AddToBatch(int nNumTracks, TCHAR* szQuery)
|
|
{
|
|
if ((g_pNetData) && (g_pNetOpt))
|
|
{
|
|
g_pNetData->AddToBatch(m_MS, szQuery, nNumTracks);
|
|
LPCDOPTIONS pOptions = g_pNetOpt->GetCDOpts();
|
|
if (pOptions)
|
|
{
|
|
pOptions->dwBatchedTitles = g_pNetData->GetNumBatched();
|
|
g_pNetOpt->DownLoadCompletion(0,NULL);
|
|
}
|
|
}
|
|
}
|
|
|
|
void CopyStreamToFile( IStream* pStream, HANDLE hFile )
|
|
{
|
|
TCHAR achBuf[512];
|
|
ULONG cb = 1;
|
|
DWORD dwWritten;
|
|
LARGE_INTEGER dlib;
|
|
|
|
dlib.LowPart = 0;
|
|
dlib.HighPart = 0;
|
|
pStream->Seek( dlib, STREAM_SEEK_SET, NULL );
|
|
pStream->Read( achBuf, 512, &cb );
|
|
while( cb )
|
|
{
|
|
if( FALSE == WriteFile( hFile, achBuf, cb, &dwWritten, NULL ))
|
|
{
|
|
break;
|
|
}
|
|
pStream->Read( achBuf, 512, &cb );
|
|
}
|
|
}
|
|
|
|
BOOL DoDownload(TCHAR* url, TCHAR* szFilename, HWND hwndParent)
|
|
{
|
|
TCHAR szPath[_MAX_PATH];
|
|
TCHAR sz[_MAX_PATH];
|
|
BOOL fGotFileName = FALSE;
|
|
|
|
// Get a file name
|
|
if(GetTempPath(_MAX_PATH, szPath))
|
|
{
|
|
if(GetTempFileName(szPath, TEXT("cdd"), 0, sz))
|
|
{
|
|
fGotFileName = TRUE;
|
|
}
|
|
}
|
|
|
|
if (!fGotFileName)
|
|
{
|
|
return FALSE;
|
|
}
|
|
|
|
IStream* pStream = NULL;
|
|
|
|
g_fDownloadDone = FALSE;
|
|
if (FAILED(CreateStreamOnHGlobal( NULL, TRUE, &pStream )))
|
|
{
|
|
return FALSE;
|
|
}
|
|
//pStream was addref'ed by createstreamonhgobal
|
|
|
|
CBindStatusCallback* pCDC = new CBindStatusCallback(pStream, hwndParent);
|
|
if(!pCDC)
|
|
{
|
|
pStream->Release();
|
|
return FALSE;
|
|
}
|
|
pCDC->AddRef();
|
|
|
|
HRESULT hr = E_NOTIMPL;
|
|
|
|
if (g_hURLMon == NULL)
|
|
{
|
|
g_hURLMon = LoadLibrary(TEXT("URLMON.DLL"));
|
|
}
|
|
|
|
if (g_hURLMon!=NULL)
|
|
{
|
|
typedef BOOL (PASCAL *URLDOWNLOADPROC)(LPUNKNOWN, LPCTSTR, DWORD, LPBINDSTATUSCALLBACK);
|
|
URLDOWNLOADPROC URLDownload = (URLDOWNLOADPROC)GetProcAddress(g_hURLMon,URLFUNCTION);
|
|
|
|
if (URLDownload!=NULL)
|
|
{
|
|
#ifdef DBG
|
|
OutputDebugString(url);
|
|
OutputDebugString(TEXT("\n"));
|
|
#endif
|
|
hr = URLDownload(NULL, url, 0, pCDC);
|
|
}
|
|
}
|
|
|
|
if(FAILED(hr))
|
|
{
|
|
pCDC->Release();
|
|
pStream->Release();
|
|
return FALSE;
|
|
}
|
|
|
|
pCDC->Release();
|
|
|
|
if (g_fCancelDownload)
|
|
{
|
|
return FALSE;
|
|
}
|
|
|
|
// Create the file for writing
|
|
HANDLE hFileWrite = CreateFile(sz, GENERIC_READ | GENERIC_WRITE,
|
|
FILE_SHARE_READ | FILE_SHARE_WRITE,
|
|
NULL, CREATE_ALWAYS, FILE_ATTRIBUTE_TEMPORARY, NULL);
|
|
if( hFileWrite != INVALID_HANDLE_VALUE )
|
|
{
|
|
CopyStreamToFile( pStream, hFileWrite );
|
|
CloseHandle( hFileWrite );
|
|
}
|
|
|
|
pStream->Release();
|
|
|
|
_tcscpy(szFilename,sz);
|
|
|
|
return TRUE;
|
|
}
|
|
|
|
//dialog box handler for multiple hits
|
|
INT_PTR CALLBACK MultiHitDlgProc(HWND hwnd, UINT message, WPARAM wParam, LPARAM lParam)
|
|
{
|
|
switch (message)
|
|
{
|
|
case WM_INITDIALOG :
|
|
{
|
|
TCHAR* szFilename = (TCHAR*)lParam;
|
|
TCHAR szTemp[MAX_PATH];
|
|
TCHAR szArtist[MAX_PATH];
|
|
TCHAR szTitle[MAX_PATH];
|
|
int i = 1;
|
|
|
|
_tcscpy(szTitle,TEXT("."));
|
|
|
|
while (_tcslen(szTitle)>0)
|
|
{
|
|
wsprintf(szTemp,TEXT("Title%i"),i);
|
|
GetPrivateProfileString(TEXT("CD"),szTemp,TEXT(""),szTitle,sizeof(szTitle)/sizeof(TCHAR),szFilename);
|
|
wsprintf(szTemp,TEXT("Artist%i"),i);
|
|
GetPrivateProfileString(TEXT("CD"),szTemp,TEXT(""),szArtist,sizeof(szArtist)/sizeof(TCHAR),szFilename);
|
|
i++;
|
|
|
|
if (_tcslen(szTitle)>0)
|
|
{
|
|
wsprintf(szTemp,TEXT("%s (%s)"),szTitle,szArtist);
|
|
SendDlgItemMessage(hwnd,IDC_LIST_DISCS,LB_ADDSTRING,0,(LPARAM)szTemp);
|
|
}
|
|
}
|
|
|
|
SendDlgItemMessage(hwnd,IDC_LIST_DISCS,LB_SETCURSEL,0,0);
|
|
}
|
|
break;
|
|
|
|
case WM_COMMAND :
|
|
{
|
|
if (LOWORD(wParam)==IDCANCEL)
|
|
{
|
|
EndDialog(hwnd,-1);
|
|
}
|
|
|
|
if (LOWORD(wParam)==IDOK)
|
|
{
|
|
LRESULT nSel = SendDlgItemMessage(hwnd,IDC_LIST_DISCS,LB_GETCURSEL,0,0);
|
|
EndDialog(hwnd,nSel+1);
|
|
}
|
|
}
|
|
break;
|
|
}
|
|
|
|
return FALSE;
|
|
}
|
|
|
|
BOOL ResolveMultiples(TCHAR* szFilename, BOOL fCurrent, HWND hwndParent)
|
|
{
|
|
//special case ... sometimes, this comes back with <2 hits!!!
|
|
//in this case, go ahead and ask for URL1
|
|
|
|
TCHAR sznewurl[INTERNET_MAX_URL_LENGTH];
|
|
GetPrivateProfileString(TEXT("CD"),TEXT("URL2"),TEXT(""),sznewurl,sizeof(sznewurl)/sizeof(TCHAR),szFilename);
|
|
|
|
INT_PTR nSelection = 0;
|
|
|
|
if (_tcslen(sznewurl)==0)
|
|
{
|
|
nSelection = 1;
|
|
}
|
|
else
|
|
{
|
|
if (fCurrent)
|
|
{
|
|
nSelection = DialogBoxParam(g_dllInst, MAKEINTRESOURCE(IDD_MULTIPLE_HITS),
|
|
hwndParent, MultiHitDlgProc, (LPARAM)szFilename );
|
|
}
|
|
}
|
|
|
|
if (nSelection > 0)
|
|
{
|
|
TCHAR szSelected[MAX_PATH];
|
|
wsprintf(szSelected,TEXT("URL%i"),nSelection);
|
|
|
|
GetPrivateProfileString(TEXT("CD"),szSelected,TEXT(""),sznewurl,sizeof(sznewurl)/sizeof(TCHAR),szFilename);
|
|
|
|
DeleteFile(szFilename);
|
|
|
|
if (DoDownload(sznewurl,szFilename, hwndParent))
|
|
{
|
|
return TRUE;
|
|
}
|
|
}
|
|
|
|
return FALSE;
|
|
}
|
|
|
|
//no more cover art in first version
|
|
#if 0
|
|
void TranslateTempCoverToFinal(TCHAR* szCurrent, TCHAR* szFinal, long discid, TCHAR* extension)
|
|
{
|
|
//we want to put the cover art in a "coverart" subdir relative to whereever CD player is
|
|
TCHAR szPath[MAX_PATH];
|
|
GetModuleFileName(NULL,szPath,sizeof(szPath));
|
|
|
|
TCHAR* szPathEnd;
|
|
szPathEnd = _tcsrchr(szPath, TEXT('\\'))+sizeof(TCHAR);
|
|
_tcscpy(szPathEnd,TEXT("coverart\\"));
|
|
|
|
CreateDirectory(szPath,NULL); //create the coverart subdir
|
|
|
|
wsprintf(szFinal,TEXT("%s%08X%s"),szPath,discid,extension);
|
|
}
|
|
#endif
|
|
|
|
DWORD GetNextDisc(long lOriginal, LPCDBATCH* ppBatch)
|
|
{
|
|
DWORD discid = (DWORD)-1;
|
|
|
|
//only do the batch if no discid was passed in originally to thread
|
|
if (lOriginal < 1)
|
|
{
|
|
if (*ppBatch!=NULL)
|
|
{
|
|
*ppBatch = (*ppBatch)->pNext;
|
|
if (*ppBatch != NULL)
|
|
{
|
|
discid = (*ppBatch)->dwTitleID;
|
|
}
|
|
}
|
|
}
|
|
|
|
return (discid);
|
|
}
|
|
|
|
LPCDPROVIDER GetNewProvider(LPCDPROVIDER pList, LPCDPROVIDER pCurrent, LPCDPROVIDER pDefault)
|
|
{
|
|
//find the next provider that isn't the current
|
|
if (pCurrent == pDefault)
|
|
{
|
|
//we've just done the current provider, so go to the head of the list next
|
|
pCurrent = pList;
|
|
if (pCurrent == pDefault)
|
|
{
|
|
//if the default was also the head of the list, go to the next and return
|
|
pCurrent = pCurrent->pNext;
|
|
}
|
|
return (pCurrent);
|
|
}
|
|
|
|
//get the next entry on the list
|
|
pCurrent = pCurrent->pNext;
|
|
|
|
//is the next entry the same as the default entry? if so, move on one more
|
|
if (pCurrent == pDefault)
|
|
{
|
|
pCurrent = pCurrent->pNext;
|
|
}
|
|
|
|
return (pCurrent);
|
|
}
|
|
|
|
//if szProvider is NULL, szURL is filled in with "just the query" ...
|
|
//if szProvider is not NULL, it is prepended to the query in szURL
|
|
int GetTracksAndQuery(LPCDBATCH pBatch, TCHAR* szURL, TCHAR* szProvider)
|
|
{
|
|
if (pBatch == NULL)
|
|
{
|
|
return 0;
|
|
}
|
|
|
|
int nReturn = pBatch->dwNumTracks;
|
|
|
|
if (szProvider != NULL)
|
|
{
|
|
wsprintf(szURL,TEXT("%s%s"),szProvider,pBatch->szTitleQuery);
|
|
}
|
|
else
|
|
{
|
|
_tcscpy(szURL,pBatch->szTitleQuery);
|
|
}
|
|
|
|
return nReturn;
|
|
}
|
|
|
|
void WINAPI AddTitleToDatabase(DWORD dwDiscID, DWORD dwTracks, TCHAR *szURL, TCHAR *szTempFile)
|
|
{
|
|
LPCDTITLE pCDTitle = NULL;
|
|
TCHAR tempstr[CDSTR];
|
|
BOOL fContinue = TRUE;
|
|
DWORD dwMenus = 0;
|
|
|
|
while (fContinue)
|
|
{
|
|
TCHAR szMenuIndex[10];
|
|
TCHAR szMenuEntry[INTERNET_MAX_URL_LENGTH];
|
|
wsprintf(szMenuIndex,TEXT("MENU%i"),dwMenus+1);
|
|
|
|
GetPrivateProfileString( TEXT("CD"), szMenuIndex, TEXT(""),
|
|
szMenuEntry, sizeof(szMenuEntry)/sizeof(TCHAR), szTempFile );
|
|
|
|
if (_tcslen(szMenuEntry)>0)
|
|
{
|
|
dwMenus++;
|
|
}
|
|
else
|
|
{
|
|
fContinue = FALSE;
|
|
}
|
|
}
|
|
|
|
if (SUCCEEDED(g_pNetData->CreateTitle(&pCDTitle, dwDiscID, dwTracks, dwMenus)))
|
|
{
|
|
GetPrivateProfileString(TEXT("CD"),TEXT("TITLE"),TEXT(""),tempstr,sizeof(tempstr)/sizeof(TCHAR),szTempFile);
|
|
_tcscpy(pCDTitle->szTitle,tempstr);
|
|
|
|
GetPrivateProfileString(TEXT("CD"),TEXT("ARTIST"),TEXT(""),tempstr,sizeof(tempstr)/sizeof(TCHAR),szTempFile);
|
|
_tcscpy(pCDTitle->szArtist,tempstr);
|
|
|
|
GetPrivateProfileString(TEXT("CD"),TEXT("LABEL"),TEXT(""),tempstr,sizeof(tempstr)/sizeof(TCHAR),szTempFile);
|
|
_tcscpy(pCDTitle->szLabel,tempstr);
|
|
|
|
GetPrivateProfileString(TEXT("CD"),TEXT("COPYRIGHT"),TEXT(""),tempstr,sizeof(tempstr)/sizeof(TCHAR),szTempFile);
|
|
_tcscpy(pCDTitle->szCopyright,tempstr);
|
|
|
|
GetPrivateProfileString(TEXT("CD"),TEXT("RELEASEDATE"),TEXT(""),tempstr,sizeof(tempstr)/sizeof(TCHAR),szTempFile);
|
|
_tcscpy(pCDTitle->szDate,tempstr);
|
|
|
|
g_pNetData->SetTitleQuery(pCDTitle, szURL);
|
|
|
|
for (int i = 1; i < (int) dwTracks + 1; i++)
|
|
{
|
|
TCHAR tempstrtrack[10];
|
|
TCHAR tempstrtitle[CDSTR];
|
|
wsprintf(tempstrtrack,TEXT("TRACK%i"),i);
|
|
GetPrivateProfileString(TEXT("CD"),tempstrtrack,TEXT(""),tempstrtitle,sizeof(tempstrtitle)/sizeof(TCHAR),szTempFile);
|
|
|
|
if (_tcslen(tempstrtitle) == 0)
|
|
{
|
|
TCHAR strFormat[CDSTR];
|
|
LoadString(g_dllInst,IDS_STRING_DEFAULTTRACK,strFormat,sizeof(strFormat)/sizeof(TCHAR));
|
|
wsprintf(tempstrtitle,strFormat,i);
|
|
}
|
|
|
|
_tcscpy(pCDTitle->pTrackTable[i-1].szName,tempstrtitle);
|
|
}
|
|
|
|
for (i = 1; i < (int) (dwMenus + 1); i++)
|
|
{
|
|
TCHAR tempstrmenu[10];
|
|
TCHAR tempstrmenuvalue[CDSTR+INTERNET_MAX_URL_LENGTH+(3*sizeof(TCHAR))]; //3 = two colons and a terminating null
|
|
wsprintf(tempstrmenu,TEXT("MENU%i"),i);
|
|
GetPrivateProfileString(TEXT("CD"),tempstrmenu,TEXT(""),tempstrmenuvalue,sizeof(tempstrmenuvalue)/sizeof(TCHAR),szTempFile);
|
|
|
|
//need to split menu into its component parts
|
|
if (_tcslen(tempstrmenuvalue)!=0)
|
|
{
|
|
TCHAR* szNamePart;
|
|
szNamePart = _tcsstr(tempstrmenuvalue,URL_SEPARATOR);
|
|
|
|
TCHAR* szURLPart;
|
|
szURLPart = _tcsstr(tempstrmenuvalue,URL_SEPARATOR);
|
|
if (szURLPart!=NULL)
|
|
{
|
|
//need to move past two colons
|
|
szURLPart = _tcsinc(szURLPart);
|
|
szURLPart = _tcsinc(szURLPart);
|
|
}
|
|
|
|
if (szNamePart!=NULL)
|
|
{
|
|
*szNamePart = '\0';
|
|
}
|
|
|
|
if (tempstrmenuvalue)
|
|
{
|
|
if (_tcslen(tempstrmenuvalue) >= sizeof(pCDTitle->pMenuTable[i-1].szMenuText)/sizeof(TCHAR))
|
|
{
|
|
tempstrmenuvalue[sizeof(pCDTitle->pMenuTable[i-1].szMenuText)/sizeof(TCHAR) - 1] = TEXT('\0'); // Trunc string to max len
|
|
}
|
|
_tcscpy(pCDTitle->pMenuTable[i-1].szMenuText,tempstrmenuvalue);
|
|
}
|
|
|
|
if (szURLPart)
|
|
{
|
|
g_pNetData->SetMenuQuery(&(pCDTitle->pMenuTable[i-1]), szURLPart);
|
|
}
|
|
}
|
|
}
|
|
|
|
g_pNetData->UnlockTitle(pCDTitle,TRUE);
|
|
|
|
//at this point, if the title is not in the database, we have a major problem
|
|
if (!g_pNetData->QueryTitle(dwDiscID))
|
|
{
|
|
g_fDBWriteFailure = TRUE;
|
|
}
|
|
else
|
|
{
|
|
g_fDBWriteFailure = FALSE;
|
|
}
|
|
}
|
|
}
|
|
|
|
|
|
BOOL IsCertifiedProvider(LPCDPROVIDER pProvider, TCHAR *szTempFile)
|
|
{
|
|
BOOL fCertified = TRUE;
|
|
TCHAR szCert[MAX_PATH];
|
|
|
|
GetPrivateProfileString(TEXT("CD"),TEXT("CERTIFICATE"),TEXT(""),szCert,sizeof(szCert)/sizeof(TCHAR),szTempFile);
|
|
|
|
fCertified = g_pNetOpt->VerifyProvider(pProvider,szCert);
|
|
|
|
return(fCertified);
|
|
}
|
|
|
|
void UpdatePropertyPage(DWORD dwDiscID, BOOL fDownloading, HWND hwndParent)
|
|
{
|
|
if (g_pNetOpt)
|
|
{
|
|
LPCDUNIT pUnit = g_pNetOpt->GetCDOpts()->pCDUnitList;
|
|
|
|
while (pUnit!=NULL)
|
|
{
|
|
if (pUnit->dwTitleID == dwDiscID)
|
|
{
|
|
pUnit->fDownLoading = fDownloading;
|
|
PostMessage(hwndParent,WM_NET_DB_UPDATE_DISC,0,(LPARAM)pUnit); //Tell the UI we changed status of disc
|
|
break;
|
|
}
|
|
pUnit = pUnit->pNext;
|
|
}
|
|
}
|
|
}
|
|
|
|
BOOL WINAPI DownloadBatch(LPCDBATCH pBatch, LPCDPROVIDER pProvider, LPDWORD pdwMultiHit, LPBOOL pfTimeout, HWND hwndParent)
|
|
{
|
|
BOOL fSuccess = FALSE;
|
|
DWORD dwTracks;
|
|
TCHAR szURL[INTERNET_MAX_URL_LENGTH];
|
|
TCHAR szTempFile[MAX_PATH];
|
|
DWORD dwDiscID;
|
|
|
|
dwTracks = GetTracksAndQuery(pBatch, szURL, pProvider->szProviderURL);
|
|
dwDiscID = pBatch->dwTitleID;
|
|
*pfTimeout = FALSE;
|
|
|
|
UpdatePropertyPage(pBatch->dwTitleID, TRUE, hwndParent); //tell prop page ui that disc is downloading
|
|
|
|
if (dwTracks > 0 && dwDiscID != 0)
|
|
{
|
|
if (DoDownload(szURL,szTempFile,hwndParent))
|
|
{
|
|
if (IsCertifiedProvider(pProvider, szTempFile))
|
|
{
|
|
int nMode = GetPrivateProfileInt(TEXT("CD"),TEXT("MODE"),MODE_NOT_FOUND,szTempFile);
|
|
|
|
if (nMode == MODE_NOT_FOUND)
|
|
{
|
|
DeleteFile(szTempFile);
|
|
}
|
|
else if (nMode == MODE_MULTIPLE)
|
|
{
|
|
if (pdwMultiHit)
|
|
{
|
|
(*pdwMultiHit)++;
|
|
}
|
|
|
|
if (!ResolveMultiples(szTempFile,TRUE,hwndParent))
|
|
{
|
|
DeleteFile(szTempFile);
|
|
}
|
|
else
|
|
{
|
|
nMode = MODE_OK;
|
|
}
|
|
}
|
|
|
|
if (nMode == MODE_OK)
|
|
{
|
|
GetTracksAndQuery(pBatch,szURL,NULL); //reset szURL to lose the provider
|
|
AddTitleToDatabase(dwDiscID, dwTracks, szURL, szTempFile);
|
|
DeleteFile(szTempFile);
|
|
fSuccess = TRUE;
|
|
} //end if mode ok
|
|
} //end if certified provider
|
|
} //end if download ok
|
|
else
|
|
{
|
|
*pfTimeout = TRUE;
|
|
}
|
|
} //end if valid query
|
|
|
|
UpdatePropertyPage(pBatch->dwTitleID, FALSE, hwndParent); //tell prop page ui that disc is no longer downloading
|
|
|
|
return(fSuccess);
|
|
}
|
|
|
|
DWORD WINAPI SpawnSingleDownload(LPVOID pParam)
|
|
{
|
|
InterlockedIncrement((LONG*)&g_lNumDownloadingThreads);
|
|
|
|
LPCDBATCH pBatch = (LPCDBATCH)pParam;
|
|
HWND hwndParent = g_hwndParent;
|
|
|
|
if (pBatch)
|
|
{
|
|
DoBatchDownload(pBatch,hwndParent);
|
|
|
|
//if download failed, add to batch if not already in db
|
|
//but only do this if batching is turned on
|
|
LPCDOPTIONS pOptions = g_pNetOpt->GetCDOpts();
|
|
if (pOptions)
|
|
{
|
|
LPCDOPTDATA pOptionData = pOptions->pCDData;
|
|
if (pOptionData)
|
|
{
|
|
if (pOptionData->fBatchEnabled)
|
|
{
|
|
if (!g_pNetData->QueryTitle(pBatch->dwTitleID))
|
|
{
|
|
g_pNetData->AddToBatch(pBatch->dwTitleID, pBatch->szTitleQuery, pBatch->dwNumTracks);
|
|
pOptions->dwBatchedTitles = g_pNetData->GetNumBatched();
|
|
PostMessage(hwndParent,WM_NET_DB_UPDATE_BATCH,0,0); //Tell the UI we changed number in batch
|
|
} //end if not in db
|
|
} //if batching is on
|
|
} //end if option data
|
|
} //end if poptions
|
|
|
|
delete [] pBatch->szTitleQuery;
|
|
delete pBatch;
|
|
}
|
|
|
|
//addref'ed before thread was created
|
|
g_pNetOpt->Release();
|
|
g_pNetData->Release();
|
|
|
|
InterlockedDecrement((LONG*)&g_lNumDownloadingThreads);
|
|
return 0;
|
|
}
|
|
|
|
DWORD WINAPI SpawnBatchDownload(LPVOID pParam)
|
|
{
|
|
InterlockedIncrement((LONG*)&g_lNumDownloadingThreads);
|
|
|
|
LPCDBATCH pBatchList = NULL;
|
|
HWND hwndParent = g_hwndParent;
|
|
|
|
if (g_pNetData)
|
|
{
|
|
if (SUCCEEDED(g_pNetData->LoadBatch(NULL,&pBatchList)))
|
|
{
|
|
DoBatchDownload(pBatchList,hwndParent);
|
|
g_pNetData->UnloadBatch(pBatchList);
|
|
}
|
|
}
|
|
|
|
//addref'ed before thread was created
|
|
g_pNetOpt->Release();
|
|
g_pNetData->Release();
|
|
|
|
InterlockedDecrement((LONG*)&g_lNumDownloadingThreads);
|
|
return 0;
|
|
}
|
|
|
|
DWORD WINAPI DoBatchDownload(LPCDBATCH pBatchList, HWND hwndParent)
|
|
{
|
|
EnterCriticalSection(&g_BatchCrit);
|
|
|
|
BOOL retcode = FALSE;
|
|
DWORD dwHow;
|
|
BOOL fConnected;
|
|
DWORD dwCurrent = 0;
|
|
DWORD dwOther = 0;
|
|
DWORD dwMultiHit = 0;
|
|
DWORD dwTimedOut = 0;
|
|
|
|
fConnected = _InternetGetConnectedState(&dwHow,0,TRUE); // Make sure we are connected to net
|
|
|
|
if (fConnected && g_pNetOpt && g_pNetData) // Make sure we are in a valid state
|
|
{
|
|
LPCDOPTIONS pOptions = g_pNetOpt->GetCDOpts(); // Get the options, needed for provider list
|
|
|
|
if (pOptions && pOptions->pCurrentProvider) // Make sure we have providers
|
|
{
|
|
LPCDPROVIDER pProviderList = NULL;
|
|
LPCDPROVIDER pProvider = NULL;
|
|
|
|
g_pNetOpt->CreateProviderList(&pProviderList); // Get the sorted provider list
|
|
pProvider = pProviderList; // Get the head of the list
|
|
|
|
LPCDBATCH pBatch;
|
|
|
|
if (pBatchList)
|
|
{
|
|
while (pProvider && !g_fCancelDownload) // loop thru providers, but check current first and only once.
|
|
{
|
|
BOOL fNotifiedUIProvider = FALSE;
|
|
pBatch = pBatchList;
|
|
|
|
while (pBatch && !g_fCancelDownload && !pProvider->fTimedOut) // We will loop thru each batched title
|
|
{
|
|
BOOL fAttemptDownload = TRUE; // Assume we are going to try to download all in batch
|
|
if (pBatch->fRemove)
|
|
{
|
|
fAttemptDownload = FALSE; //we've already tried this disc on one provider and got it
|
|
}
|
|
|
|
if (fAttemptDownload)
|
|
{
|
|
if (!fNotifiedUIProvider)
|
|
{
|
|
PostMessage(hwndParent,WM_NET_CHANGEPROVIDER,0,(LPARAM)pProvider); //Tell the UI who the provider is
|
|
fNotifiedUIProvider = TRUE;
|
|
}
|
|
|
|
BOOL fTimeout = FALSE;
|
|
|
|
if (DownloadBatch(pBatch, pProvider, &dwMultiHit, &fTimeout, hwndParent)) // attempt to download this batch
|
|
{
|
|
pBatch->fRemove = TRUE; // This batch download succeeded, mark for termination from batch
|
|
|
|
if (pProvider == pOptions->pCurrentProvider)
|
|
{
|
|
dwCurrent++;
|
|
}
|
|
else
|
|
{
|
|
dwOther++;
|
|
}
|
|
}
|
|
else
|
|
{
|
|
pProvider->fTimedOut = fTimeout;
|
|
}
|
|
|
|
//check to see if db write failed
|
|
if (g_fDBWriteFailure)
|
|
{
|
|
//let the UI know
|
|
PostMessage(hwndParent,WM_NET_DB_FAILURE,0,0);
|
|
|
|
//get out of the batch loop
|
|
break;
|
|
}
|
|
|
|
//let ui know that something happened with this disc
|
|
PostMessage(hwndParent,WM_NET_DONE,(WPARAM)g_dllInst,pBatch->dwTitleID);
|
|
|
|
//increment the meter if we know this is the last time we're
|
|
//visiting this particular disc ... either it was found, or
|
|
//we are out of possible places to look
|
|
if ((pBatch->fRemove) || (pProvider->pNext == NULL))
|
|
{
|
|
PostMessage(hwndParent,WM_NET_INCMETER,(WPARAM)g_dllInst,pBatch->dwTitleID);
|
|
}
|
|
|
|
} //end attempt on disc
|
|
|
|
pBatch = pBatch->pNext;
|
|
} //end batch
|
|
|
|
if (g_fDBWriteFailure)
|
|
{
|
|
//get out of the provider loop
|
|
break;
|
|
}
|
|
|
|
pProvider = pProvider->pNext; //providers are "in order"
|
|
} //end while cycling providers
|
|
|
|
} //end if load batch OK
|
|
|
|
//check to see if ALL providers timed out ... possible net problem
|
|
BOOL fAllFailed = TRUE;
|
|
pProvider = pProviderList;
|
|
while (pProvider!=NULL)
|
|
{
|
|
if (!pProvider->fTimedOut)
|
|
{
|
|
fAllFailed = FALSE;
|
|
break;
|
|
}
|
|
pProvider = pProvider->pNext;
|
|
}
|
|
|
|
if (fAllFailed)
|
|
{
|
|
//let the UI know
|
|
PostMessage(hwndParent,WM_NET_NET_FAILURE,0,0);
|
|
}
|
|
|
|
g_pNetOpt->DestroyProviderList(&pProviderList);
|
|
} //end if pointers ok
|
|
|
|
#ifdef DBG
|
|
// Ok, output some interesting stat's about what happened.
|
|
{
|
|
TCHAR str[255];
|
|
wsprintf(str, TEXT("current = %d, other = %d, multihits = %d\n"), dwCurrent, dwOther, dwMultiHit);
|
|
OutputDebugString(str);
|
|
}
|
|
#endif
|
|
} //end if connected to net and pointers ok
|
|
|
|
if (!fConnected)
|
|
{
|
|
//may be a net problem
|
|
if ((dwHow & (INTERNET_CONNECTION_MODEM|INTERNET_CONNECTION_LAN)) == 0)
|
|
{
|
|
PostMessage(hwndParent,WM_NET_NET_FAILURE,0,0);
|
|
}
|
|
}
|
|
|
|
PostMessage(hwndParent,WM_NET_DONE,(WPARAM)g_dllInst,(LPARAM) 0); //fBadGuy ? -1 : 0);
|
|
|
|
LeaveCriticalSection(&g_BatchCrit);
|
|
|
|
return (retcode);
|
|
}
|
|
|
|
|