464 lines
11 KiB
C++
464 lines
11 KiB
C++
|
//\\//\\//\\//\\//\\//\\//\\//\\//\\//\\//\\//\\//\\//\\//\\//\\//\\//\\//\\//\\
|
||
|
|
//
|
||
|
|
// utils.cpp
|
||
|
|
//
|
||
|
|
// Misc routines.
|
||
|
|
//
|
||
|
|
// History:
|
||
|
|
//
|
||
|
|
// 6/25/97 tnoonan Created.
|
||
|
|
//
|
||
|
|
////////////////////////////////////////////////////////////////////////////////
|
||
|
|
|
||
|
|
//
|
||
|
|
// Includes
|
||
|
|
//
|
||
|
|
|
||
|
|
#include "stdinc.h"
|
||
|
|
#include "cdfidl.h"
|
||
|
|
#include "persist.h"
|
||
|
|
#include "cdfview.h"
|
||
|
|
#include "xmlutil.h"
|
||
|
|
#include "bindstcb.h"
|
||
|
|
#include "dll.h"
|
||
|
|
#include "resource.h"
|
||
|
|
#include "chanapi.h"
|
||
|
|
|
||
|
|
#include <mluisupp.h>
|
||
|
|
|
||
|
|
typedef struct _tagDialogData {
|
||
|
|
LPCWSTR pszwURL;
|
||
|
|
IXMLDocument* pIXMLDocument;
|
||
|
|
int nProgress;
|
||
|
|
} DIALOGDATA;
|
||
|
|
|
||
|
|
|
||
|
|
HRESULT
|
||
|
|
GetURLFromIni(
|
||
|
|
LPCTSTR pszPath,
|
||
|
|
BSTR* pbstrURL
|
||
|
|
)
|
||
|
|
{
|
||
|
|
ASSERT(pszPath);
|
||
|
|
ASSERT(pbstrURL);
|
||
|
|
|
||
|
|
HRESULT hr = E_FAIL;
|
||
|
|
|
||
|
|
LPTSTR szFile = TSTR_INI_FILE;
|
||
|
|
LPTSTR szSection = TSTR_INI_SECTION;
|
||
|
|
LPTSTR szKey = TSTR_INI_URL;
|
||
|
|
|
||
|
|
TCHAR szURL[INTERNET_MAX_URL_LENGTH];
|
||
|
|
TCHAR szPath[MAX_PATH];
|
||
|
|
|
||
|
|
StrCpyN(szPath, pszPath, ARRAYSIZE(szPath) - ARRAYSIZE(TSTR_INI_FILE));
|
||
|
|
StrCat(szPath, szFile);
|
||
|
|
|
||
|
|
if (GetPrivateProfileString(szSection, szKey, TEXT(""), szURL,
|
||
|
|
INTERNET_MAX_URL_LENGTH, szPath))
|
||
|
|
{
|
||
|
|
WCHAR wszURL[INTERNET_MAX_URL_LENGTH];
|
||
|
|
if (SHTCharToUnicode(szURL, wszURL, ARRAYSIZE(wszURL)))
|
||
|
|
{
|
||
|
|
*pbstrURL = SysAllocString(wszURL);
|
||
|
|
hr = S_OK;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
return hr;
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
//\\//\\//\\//\\//\\//\\//\\//\\//\\//\\//\\//\\//\\//\\//\\//\\//\\//\\//\\//\\
|
||
|
|
//
|
||
|
|
// *** Name ***
|
||
|
|
//
|
||
|
|
//
|
||
|
|
// Description:
|
||
|
|
//
|
||
|
|
//
|
||
|
|
// Parameters:
|
||
|
|
//
|
||
|
|
//
|
||
|
|
// Return:
|
||
|
|
//
|
||
|
|
//
|
||
|
|
// Comments:
|
||
|
|
//
|
||
|
|
//
|
||
|
|
////////////////////////////////////////////////////////////////////////////////
|
||
|
|
HRESULT
|
||
|
|
GetNameAndURLAndSubscriptionInfo(
|
||
|
|
LPCTSTR pszPath,
|
||
|
|
BSTR* pbstrName,
|
||
|
|
BSTR* pbstrURL,
|
||
|
|
SUBSCRIPTIONINFO* psi
|
||
|
|
)
|
||
|
|
{
|
||
|
|
ASSERT(pszPath);
|
||
|
|
ASSERT(pbstrName);
|
||
|
|
ASSERT(pbstrURL);
|
||
|
|
|
||
|
|
HRESULT hr;
|
||
|
|
|
||
|
|
*pbstrName = NULL;
|
||
|
|
*pbstrURL = NULL;
|
||
|
|
|
||
|
|
CCdfView* pCCdfView = new CCdfView;
|
||
|
|
|
||
|
|
if (pCCdfView)
|
||
|
|
{
|
||
|
|
WCHAR wszPath[MAX_PATH];
|
||
|
|
|
||
|
|
if (SHTCharToUnicode(pszPath, wszPath, ARRAYSIZE(wszPath)))
|
||
|
|
{
|
||
|
|
hr = pCCdfView->Load(wszPath, 0);
|
||
|
|
|
||
|
|
if (SUCCEEDED(hr))
|
||
|
|
{
|
||
|
|
IXMLDocument* pIXMLDocument;
|
||
|
|
|
||
|
|
hr = pCCdfView->ParseCdf(NULL, &pIXMLDocument, PARSE_LOCAL);
|
||
|
|
|
||
|
|
if (SUCCEEDED(hr))
|
||
|
|
{
|
||
|
|
ASSERT(pIXMLDocument);
|
||
|
|
|
||
|
|
IXMLElement* pIXMLElement;
|
||
|
|
LONG nIndex;
|
||
|
|
|
||
|
|
hr = XML_GetFirstChannelElement(pIXMLDocument,
|
||
|
|
&pIXMLElement, &nIndex);
|
||
|
|
|
||
|
|
if (SUCCEEDED(hr))
|
||
|
|
{
|
||
|
|
ASSERT(pIXMLElement);
|
||
|
|
|
||
|
|
*pbstrName = XML_GetAttribute(pIXMLElement, XML_TITLE);
|
||
|
|
|
||
|
|
if (*pbstrName && 0 == **pbstrName)
|
||
|
|
{
|
||
|
|
SysFreeString(*pbstrName);
|
||
|
|
*pbstrName = NULL;
|
||
|
|
}
|
||
|
|
|
||
|
|
*pbstrURL = XML_GetAttribute(pIXMLElement, XML_SELF);
|
||
|
|
|
||
|
|
if (*pbstrURL && 0 == **pbstrURL)
|
||
|
|
{
|
||
|
|
SysFreeString(*pbstrName);
|
||
|
|
*pbstrURL = NULL;
|
||
|
|
}
|
||
|
|
|
||
|
|
if (psi)
|
||
|
|
XML_GetSubscriptionInfo(pIXMLElement, psi);
|
||
|
|
|
||
|
|
pIXMLElement->Release();
|
||
|
|
}
|
||
|
|
|
||
|
|
pIXMLDocument->Release();
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
if (NULL == *pbstrName)
|
||
|
|
{
|
||
|
|
TCHAR szName[MAX_PATH];
|
||
|
|
WCHAR wszName[MAX_PATH];
|
||
|
|
|
||
|
|
StrCpyN(szName, pszPath, ARRAYSIZE(szName));
|
||
|
|
PathStripPath(szName);
|
||
|
|
|
||
|
|
if (SHTCharToUnicode(szName, wszName, ARRAYSIZE(wszName)))
|
||
|
|
*pbstrName = SysAllocString(wszName);
|
||
|
|
}
|
||
|
|
|
||
|
|
if (NULL == *pbstrURL)
|
||
|
|
{
|
||
|
|
GetURLFromIni(pszPath, pbstrURL);
|
||
|
|
}
|
||
|
|
|
||
|
|
hr = (NULL != *pbstrName) && (NULL != **pbstrName) &&
|
||
|
|
(NULL != *pbstrURL) && (NULL != **pbstrURL) ? S_OK : E_FAIL;
|
||
|
|
|
||
|
|
if (FAILED(hr))
|
||
|
|
{
|
||
|
|
if (NULL != *pbstrName)
|
||
|
|
{
|
||
|
|
SysFreeString(*pbstrName);
|
||
|
|
*pbstrName = NULL;
|
||
|
|
}
|
||
|
|
if (NULL != *pbstrURL)
|
||
|
|
{
|
||
|
|
SysFreeString(*pbstrURL);
|
||
|
|
*pbstrURL = NULL;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
pCCdfView->Release();
|
||
|
|
}
|
||
|
|
else
|
||
|
|
{
|
||
|
|
hr = E_OUTOFMEMORY;
|
||
|
|
}
|
||
|
|
|
||
|
|
return hr;
|
||
|
|
}
|
||
|
|
|
||
|
|
int CDFMessageBox(HWND hwnd, UINT idTextFmt, UINT idCaption, UINT uType, ...)
|
||
|
|
{
|
||
|
|
TCHAR szCaption[80];
|
||
|
|
TCHAR szTextFmt[256];
|
||
|
|
LPTSTR pszText;
|
||
|
|
int result;
|
||
|
|
va_list va;
|
||
|
|
|
||
|
|
va_start(va, uType);
|
||
|
|
|
||
|
|
MLLoadString(idTextFmt, szTextFmt, ARRAYSIZE(szTextFmt));
|
||
|
|
MLLoadString(idCaption, szCaption, ARRAYSIZE(szCaption));
|
||
|
|
|
||
|
|
FormatMessage(FORMAT_MESSAGE_ALLOCATE_BUFFER | FORMAT_MESSAGE_FROM_STRING,
|
||
|
|
szTextFmt, 0, 0, (LPTSTR)&pszText, 0, &va);
|
||
|
|
|
||
|
|
result = MessageBox(hwnd, pszText, szCaption, uType);
|
||
|
|
|
||
|
|
LocalFree((HLOCAL)pszText);
|
||
|
|
|
||
|
|
return result;
|
||
|
|
}
|
||
|
|
|
||
|
|
INT_PTR RefreshDlgProc(HWND hDlg, UINT msg, WPARAM wParam, LPARAM lParam)
|
||
|
|
{
|
||
|
|
BOOL fRet = TRUE;
|
||
|
|
DIALOGDATA* pdd = (DIALOGDATA*) GetWindowLongPtr(hDlg, DWLP_USER);
|
||
|
|
|
||
|
|
switch (msg)
|
||
|
|
{
|
||
|
|
case WM_INITDIALOG:
|
||
|
|
SetWindowLongPtr(hDlg, DWLP_USER, lParam);
|
||
|
|
pdd = (DIALOGDATA*)lParam;
|
||
|
|
|
||
|
|
IMoniker* pIMoniker;
|
||
|
|
|
||
|
|
if (SUCCEEDED(CreateURLMoniker(NULL, pdd->pszwURL, &pIMoniker)))
|
||
|
|
{
|
||
|
|
ASSERT(pIMoniker);
|
||
|
|
|
||
|
|
IBindCtx* pIBindCtx;
|
||
|
|
|
||
|
|
IBindStatusCallback* pIBindStatusCallback =
|
||
|
|
(IBindStatusCallback*) new CBindStatusCallback2(hDlg);
|
||
|
|
|
||
|
|
if (pIBindStatusCallback)
|
||
|
|
{
|
||
|
|
if (SUCCEEDED(CreateBindCtx(0, &pIBindCtx)) )
|
||
|
|
{
|
||
|
|
ASSERT(pIBindCtx);
|
||
|
|
|
||
|
|
if (SUCCEEDED(RegisterBindStatusCallback(pIBindCtx,
|
||
|
|
pIBindStatusCallback,
|
||
|
|
NULL, 0)))
|
||
|
|
{
|
||
|
|
IPersistMoniker* pIPersistMoniker;
|
||
|
|
|
||
|
|
if (SUCCEEDED(pdd->pIXMLDocument->QueryInterface(
|
||
|
|
IID_IPersistMoniker,
|
||
|
|
(void**)&pIPersistMoniker)))
|
||
|
|
{
|
||
|
|
ASSERT(pIPersistMoniker);
|
||
|
|
|
||
|
|
pIPersistMoniker->Load(FALSE, pIMoniker, pIBindCtx,
|
||
|
|
0);
|
||
|
|
|
||
|
|
pIPersistMoniker->Release();
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
pIBindCtx->Release();
|
||
|
|
}
|
||
|
|
|
||
|
|
pIBindStatusCallback->Release();
|
||
|
|
}
|
||
|
|
|
||
|
|
pIMoniker->Release();
|
||
|
|
}
|
||
|
|
|
||
|
|
Animate_Open(GetDlgItem(hDlg, IDC_DOWNLOADANIMATE), IDA_DOWNLOAD);
|
||
|
|
Animate_Play(GetDlgItem(hDlg, IDC_DOWNLOADANIMATE), 0, -1, -1);
|
||
|
|
|
||
|
|
//SendMessage(GetDlgItem(hDlg, IDC_DOWNLOADPROGRESS), PBM_SETRANGE32, 0, 100);
|
||
|
|
pdd->nProgress = 0;
|
||
|
|
|
||
|
|
break;
|
||
|
|
|
||
|
|
case WM_COMMAND:
|
||
|
|
switch (LOWORD(wParam))
|
||
|
|
{
|
||
|
|
case IDCANCEL:
|
||
|
|
EndDialog(hDlg, FALSE);
|
||
|
|
break;
|
||
|
|
|
||
|
|
case DOWNLOAD_PROGRESS:
|
||
|
|
SendMessage(GetDlgItem(hDlg, IDC_DOWNLOADPROGRESS), PBM_DELTAPOS,
|
||
|
|
pdd->nProgress += 2, 0);
|
||
|
|
break;
|
||
|
|
|
||
|
|
case DOWNLOAD_COMPLETE:
|
||
|
|
if (lParam)
|
||
|
|
{
|
||
|
|
SendMessage(GetDlgItem(hDlg, IDC_DOWNLOADPROGRESS), PBM_SETPOS,
|
||
|
|
100, 0);
|
||
|
|
|
||
|
|
XML_DownloadImages(pdd->pIXMLDocument);
|
||
|
|
}
|
||
|
|
EndDialog(hDlg, lParam);
|
||
|
|
break;
|
||
|
|
}
|
||
|
|
break;
|
||
|
|
|
||
|
|
case WM_CLOSE:
|
||
|
|
EndDialog(hDlg, FALSE);
|
||
|
|
break;
|
||
|
|
|
||
|
|
case WM_DESTROY:
|
||
|
|
break;
|
||
|
|
|
||
|
|
default:
|
||
|
|
fRet = FALSE;
|
||
|
|
}
|
||
|
|
|
||
|
|
return fRet;
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
BOOL DownloadCdfUI(HWND hwnd, LPCWSTR pszwURL, IXMLDocument* pIXMLDocument)
|
||
|
|
{
|
||
|
|
BOOL fRet = FALSE;
|
||
|
|
|
||
|
|
DIALOGDATA dd;
|
||
|
|
|
||
|
|
dd.pszwURL = pszwURL;
|
||
|
|
dd.pIXMLDocument = pIXMLDocument;
|
||
|
|
|
||
|
|
if (hwnd)
|
||
|
|
{
|
||
|
|
DWORD dwCacheCount = g_dwCacheCount;
|
||
|
|
|
||
|
|
INT_PTR nRes = DialogBoxParam(MLGetHinst(),
|
||
|
|
(LPWSTR)MAKEINTRESOURCE(IDD_CHANNELREFRESH),
|
||
|
|
hwnd,
|
||
|
|
RefreshDlgProc,
|
||
|
|
(LPARAM)&dd);
|
||
|
|
|
||
|
|
if (-1 == nRes)
|
||
|
|
{
|
||
|
|
int err = GetLastError();
|
||
|
|
}
|
||
|
|
else if (TRUE == nRes)
|
||
|
|
{
|
||
|
|
TCHAR szURL[INTERNET_MAX_URL_LENGTH];
|
||
|
|
|
||
|
|
if (SHUnicodeToTChar(pszwURL, szURL, ARRAYSIZE(szURL)))
|
||
|
|
{
|
||
|
|
FILETIME ftLastMod;
|
||
|
|
|
||
|
|
URLGetLastModTime(szURL, &ftLastMod);
|
||
|
|
Cache_EnterWriteLock();
|
||
|
|
|
||
|
|
Cache_RemoveItem(szURL);
|
||
|
|
|
||
|
|
if (SUCCEEDED(Cache_AddItem(szURL, pIXMLDocument, PARSE_NET,
|
||
|
|
ftLastMod, dwCacheCount)))
|
||
|
|
fRet = TRUE;
|
||
|
|
|
||
|
|
Cache_LeaveWriteLock();
|
||
|
|
}
|
||
|
|
|
||
|
|
BSTR bstrSSUrl;
|
||
|
|
|
||
|
|
if (SUCCEEDED(XML_GetScreenSaverURL(pIXMLDocument, &bstrSSUrl)))
|
||
|
|
{
|
||
|
|
Channel_WriteScreenSaverURL(pszwURL, bstrSSUrl);
|
||
|
|
SysFreeString(bstrSSUrl);
|
||
|
|
}
|
||
|
|
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
return fRet;
|
||
|
|
}
|
||
|
|
|
||
|
|
// Checks if global state is offline
|
||
|
|
// Stolen from webcheck utils.cpp
|
||
|
|
|
||
|
|
BOOL IsGlobalOffline(void)
|
||
|
|
{
|
||
|
|
DWORD dwState = 0,
|
||
|
|
dwSize = sizeof(DWORD);
|
||
|
|
BOOL fRet = FALSE;
|
||
|
|
|
||
|
|
if (InternetQueryOption(NULL, INTERNET_OPTION_CONNECTED_STATE, &dwState, &dwSize))
|
||
|
|
{
|
||
|
|
if (dwState & INTERNET_STATE_DISCONNECTED_BY_USER)
|
||
|
|
fRet = TRUE;
|
||
|
|
}
|
||
|
|
|
||
|
|
return fRet;
|
||
|
|
}
|
||
|
|
|
||
|
|
void SetGlobalOffline(BOOL fOffline)
|
||
|
|
{
|
||
|
|
INTERNET_CONNECTED_INFO ci;
|
||
|
|
|
||
|
|
memset(&ci, 0, sizeof(ci));
|
||
|
|
|
||
|
|
if (fOffline)
|
||
|
|
{
|
||
|
|
ci.dwConnectedState = INTERNET_STATE_DISCONNECTED_BY_USER;
|
||
|
|
ci.dwFlags = ISO_FORCE_DISCONNECTED;
|
||
|
|
}
|
||
|
|
else
|
||
|
|
{
|
||
|
|
ci.dwConnectedState = INTERNET_STATE_CONNECTED;
|
||
|
|
}
|
||
|
|
|
||
|
|
InternetSetOption(NULL, INTERNET_OPTION_CONNECTED_STATE, &ci, sizeof(ci));
|
||
|
|
}
|
||
|
|
|
||
|
|
//
|
||
|
|
// Can the given url be subscribed?
|
||
|
|
//
|
||
|
|
|
||
|
|
BOOL
|
||
|
|
CanSubscribe(
|
||
|
|
LPCWSTR pwszURL
|
||
|
|
)
|
||
|
|
{
|
||
|
|
ASSERT(pwszURL);
|
||
|
|
|
||
|
|
BOOL fRet = FALSE;
|
||
|
|
|
||
|
|
TCHAR szURL[INTERNET_MAX_URL_LENGTH];
|
||
|
|
|
||
|
|
if (SHUnicodeToTChar(pwszURL, szURL, ARRAYSIZE(szURL)))
|
||
|
|
{
|
||
|
|
TCHAR szCanonicalURL[INTERNET_MAX_URL_LENGTH];
|
||
|
|
DWORD dwSize = ARRAYSIZE(szCanonicalURL);
|
||
|
|
|
||
|
|
URL_COMPONENTS uc = {0};
|
||
|
|
uc.dwStructSize = sizeof(URL_COMPONENTS);
|
||
|
|
|
||
|
|
if (InternetCanonicalizeUrl(szURL, szCanonicalURL, &dwSize, ICU_DECODE)
|
||
|
|
&&
|
||
|
|
InternetCrackUrl(szCanonicalURL, 0, 0, &uc)
|
||
|
|
&&
|
||
|
|
((INTERNET_SCHEME_HTTP == uc.nScheme) ||
|
||
|
|
(INTERNET_SCHEME_HTTPS == uc.nScheme)))
|
||
|
|
{
|
||
|
|
fRet = TRUE;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
return fRet;
|
||
|
|
}
|