Category

Development Platform
Home > SourceCode/Document > Windows Develop > Windows Kernel > View Code
proppgs.cpp
Package: shell.rar　[view]
Upload User: xhy777
Upload Date: 2007-02-14
Package Size: 24088k
Code Size: 25k
Category:

Windows Kernel

Development Platform:

Visual C++

proppgs.cpp：Code Content
  1. //\//\//\//\//\//\//\//\//\//\//\//\//\//\//\//\//\//\//\//\
  2. //
  3. // proppgs.cpp 
  4. //
  5. //   IShellPropSheetExt for channel shortcuts.
  6. //
  7. //   History:
  8. //
  9. //       6/12/97  edwardp   Created.
  10. //
  11. //   Note: The hotkey stuff is comment out, the shell/windows doesn't make it 
  12. //         possible to persist hotkey settings across sessions and it isn't 
  13. //         worth kicking off another thread at boot to enable this feature.
  14. //
  15. ////////////////////////////////////////////////////////////////////////////////
  17. //
  18. // Includes
  19. //
  21. #include "stdinc.h"
  22. #include "cdfidl.h"
  23. #include "persist.h"
  24. #include "cdfview.h"
  25. #include "proppgs.h"
  26. #include "xmlutil.h"
  27. #include "dll.h"
  28. #include "iconhand.h"
  29. #include "resource.h"
  30. #include "winineti.h"
  31. #include <iehelpid.h>
  33. #include <mluisupp.h>
  35. #pragma warning(disable:4800)
  37. //
  38. // Constructor and destructor.
  39. //
  41. //\//\//\//\//\//\//\//\//\//\//\//\//\//\//\//\//\//\//\//\
  42. //
  43. // *** CPropertyPages::CPropertyPages ***
  44. //
  45. //    Constructor for CPropertyPages.
  46. //
  47. ////////////////////////////////////////////////////////////////////////////////
  48. CPropertyPages::CPropertyPages (
  49.     void
  50. )
  51. : m_cRef(1)
  52. {
  53.     ASSERT(NULL == m_pSubscriptionMgr2);
  54.     ASSERT(NULL == m_pInitDataObject);
  56.     TraceMsg(TF_OBJECTS, "+ IShellPropSheetExt");
  58.     DllAddRef();
  60.     return;
  61. }
  63. //\//\//\//\//\//\//\//\//\//\//\//\//\//\//\//\//\//\//\//\
  64. //
  65. // *** CPropertyPages::~CPropertyPages ***
  66. //
  67. //    Destructor.
  68. //
  69. ////////////////////////////////////////////////////////////////////////////////
  70. CPropertyPages::~CPropertyPages (
  71.     void
  72. )
  73. {
  74.     if (m_pSubscriptionMgr2)
  75.         m_pSubscriptionMgr2->Release();
  77.     if (m_pInitDataObject)
  78.         m_pInitDataObject->Release();
  80.     ASSERT(0 == m_cRef);
  82.     //
  83.     // Matching Release for the constructor Addref.
  84.     //
  86.     TraceMsg(TF_OBJECTS, "- IShellPropSheetExt");
  88.     DllRelease();
  90.     return;
  91. }
  94. //
  95. // IUnknown methods.
  96. //
  98. //\//\//\//\//\//\//\//\//\//\//\//\//\//\//\//\//\//\//\//\
  99. //
  100. // *** CPropertyPages::QueryInterface ***
  101. //
  102. //    CPropertyPages QI.
  103. //
  104. ////////////////////////////////////////////////////////////////////////////////
  105. STDMETHODIMP
  106. CPropertyPages::QueryInterface (
  107.     REFIID riid,
  108.     void **ppv
  109. )
  110. {
  111.     ASSERT(ppv);
  113.     HRESULT hr;
  115.     *ppv = NULL;
  117.     if (IID_IUnknown == riid || IID_IShellPropSheetExt == riid)
  118.     {
  119.         *ppv = (IShellPropSheetExt*)this;
  120.     }
  121.     else if (IID_IShellExtInit == riid)
  122.     {
  123.         *ppv = (IShellExtInit*)this;
  124.     }
  126.     if (*ppv)
  127.     {
  128.         ((IUnknown*)*ppv)->AddRef();
  129.         hr = S_OK;
  130.     }
  131.     else
  132.     {
  133.         hr = E_NOINTERFACE;
  134.     }
  136.     ASSERT((SUCCEEDED(hr) && *ppv) || (FAILED(hr) && NULL == *ppv));
  138.     return hr;
  139. }
  141. //\//\//\//\//\//\//\//\//\//\//\//\//\//\//\//\//\//\//\//\
  142. //
  143. // *** CPropertyPages::AddRef ***
  144. //
  145. //    CPropertyPages AddRef.
  146. //
  147. ////////////////////////////////////////////////////////////////////////////////
  148. STDMETHODIMP_(ULONG)
  149. CPropertyPages::AddRef (
  150.     void
  151. )
  152. {
  153.     ASSERT(m_cRef != 0);
  154.     ASSERT(m_cRef < (ULONG)-1);
  156.     return ++m_cRef;
  157. }
  159. //\//\//\//\//\//\//\//\//\//\//\//\//\//\//\//\//\//\//\//\
  160. //
  161. // *** CPropertyPages::Release ***
  162. //
  163. //    CContextMenu Release.
  164. //
  165. ////////////////////////////////////////////////////////////////////////////////
  166. STDMETHODIMP_(ULONG)
  167. CPropertyPages::Release (
  168.     void
  169. )
  170. {
  171.     ASSERT (m_cRef != 0);
  173.     ULONG cRef = --m_cRef;
  174.     
  175.     if (0 == cRef)
  176.         delete this;
  178.     return cRef;
  179. }
  182. //
  183. //  IShellPropSheetExt methods.
  184. //
  186. //\//\//\//\//\//\//\//\//\//\//\//\//\//\//\//\//\//\//\//\
  187. //
  188. // *** Name ***
  189. //
  190. //
  191. // Description:
  192. //
  193. //
  194. // Parameters:
  195. //
  196. //
  197. // Return:
  198. //
  199. //
  200. // Comments:
  201. //
  202. //
  203. ////////////////////////////////////////////////////////////////////////////////
  204. STDMETHODIMP
  205. CPropertyPages::AddPages(
  206.     LPFNADDPROPSHEETPAGE lpfnAddPage,
  207.     LPARAM lParam
  208. )
  209. {
  210.     HRESULT hr = S_OK;
  211.     PROPSHEETPAGE psp;
  213.     psp.dwSize = sizeof(PROPSHEETPAGE);
  214.     psp.dwFlags = PSP_DEFAULT | PSP_USECALLBACK;
  215.     psp.hInstance = MLGetHinst();
  216.     psp.pszTemplate = MAKEINTRESOURCE(IDD_CHANNEL_PROP);
  217.     psp.hIcon = NULL;
  218.     psp.pszTitle = NULL;
  219.     psp.pfnDlgProc = PropSheetDlgProc;
  220.     psp.lParam = (LPARAM)(CPropertyPages *)this;
  221.     psp.pfnCallback = PropSheetCallback;
  223.     HPROPSHEETPAGE hpage = CreatePropertySheetPage(&psp);
  225.     if (hpage)
  226.     {
  227.         //  Release() happens in PropSheetCallback.
  228.         AddRef();
  230.         //  Assume the mess below doesn't work, we want the default page to be us.
  231.         hr = 1;
  233.         //  HACKHACK: This code attempts to remove the Folder property pages such as
  234.         //  General and Sharing (it will also whack any 3rd party pages which were
  235.         //  unfortunate enough to have been loaded before use :)
  236.         PROPSHEETHEADER *ppsh = (PROPSHEETHEADER *)lParam;
  238.         //  First make sure we can safely access the memory as if it were a
  239.         //  PROPSHEETHEADER structure.
  240.         if (!IsBadReadPtr(ppsh, PROPSHEETHEADER_V1_SIZE) &&
  241.             !IsBadWritePtr(ppsh, PROPSHEETHEADER_V1_SIZE))
  242.         {
  243.             //  Now see if the module matches shell32
  244.             if (ppsh->hInstance == GetModuleHandle(TEXT("shell32.dll")))
  245.             {
  246.                 //  looks good so rip 'em out
  248.                 for (UINT i = 0; i < ppsh->nPages; i++)
  249.                 {
  250.                     //  At least be a good citizen and delete their pages so we
  251.                     //  don't leak
  252.                     DestroyPropertySheetPage(ppsh->phpage[i]);
  253.                 }
  254.                 ppsh->nPages = 0;
  256.                 //  Now we shouldn't need to mess with the default page.  If someone
  257.                 //  loads after us, we may not win.
  258.                 hr = 0;
  259.             }
  260.         }
  262.         if (lpfnAddPage(hpage, lParam))
  263.         {
  264.             WCHAR wszURL[INTERNET_MAX_URL_LENGTH];
  266.             SHTCharToUnicode(m_szURL, wszURL, ARRAYSIZE(wszURL));
  268.             if (SUCCEEDED(InitializeSubsMgr2()))
  269.             {
  270.                 m_pSubscriptionMgr2->IsSubscribed(wszURL, &m_bStartSubscribed);
  272.                 if (m_bStartSubscribed)
  273.                 {
  274.                     IShellPropSheetExt *pspse;
  276.                     if (SUCCEEDED(m_pSubscriptionMgr2->QueryInterface(IID_IShellPropSheetExt,
  277.                                                                       (void **)&pspse)))
  278.                     {
  279.                         pspse->AddPages(lpfnAddPage, lParam);
  280.                         pspse->Release();
  281.                     }
  282.                 }
  283.             }
  284.         }
  285.         else
  286.         {
  287.             DestroyPropertySheetPage(hpage);
  288.             hr = E_FAIL;
  289.         }
  290.     }
  291.     else
  292.     {
  293.         hr = E_OUTOFMEMORY;
  294.     }
  295.     
  296.     return hr;
  297. }
  299. //\//\//\//\//\//\//\//\//\//\//\//\//\//\//\//\//\//\//\//\
  300. //
  301. // *** Name ***
  302. //
  303. //
  304. // Description:
  305. //
  306. //
  307. // Parameters:
  308. //
  309. //
  310. // Return:
  311. //
  312. //
  313. // Comments:
  314. //
  315. //
  316. ////////////////////////////////////////////////////////////////////////////////
  317. STDMETHODIMP
  318. CPropertyPages::ReplacePage(
  319.     UINT uPageID,
  320.     LPFNADDPROPSHEETPAGE lpfnAddPage,
  321.     LPARAM lParam
  322. )
  323. {
  324.     return E_NOTIMPL; 
  325. }
  328. //
  329. // IShellExtInit methods.
  330. //
  332. //\//\//\//\//\//\//\//\//\//\//\//\//\//\//\//\//\//\//\//\
  333. //
  334. // *** Name ***
  335. //
  336. //
  337. // Description:
  338. //
  339. //
  340. // Parameters:
  341. //
  342. //
  343. // Return:
  344. //
  345. //
  346. // Comments:
  347. //
  348. //
  349. ////////////////////////////////////////////////////////////////////////////////
  351. STDMETHODIMP
  352. CPropertyPages::Initialize(
  353.     LPCITEMIDLIST pidl,
  354.     LPDATAOBJECT pIDataObject,
  355.     HKEY hkey
  356. )
  357. {
  358.     HRESULT hr;
  360.     STGMEDIUM stgmed;
  361.     FORMATETC fmtetc = {CF_HDROP, NULL, DVASPECT_CONTENT, -1,
  362.                         TYMED_HGLOBAL};
  364.     if (m_pInitDataObject)
  365.         m_pInitDataObject->Release();
  367.     m_pInitDataObject = pIDataObject;
  368.     m_pInitDataObject->AddRef();
  370.     hr = pIDataObject->GetData(&fmtetc, &stgmed);
  372.     if (SUCCEEDED(hr))
  373.     {
  374.         if (DragQueryFile((HDROP)stgmed.hGlobal, 0, m_szPath, 
  375.                           ARRAYSIZE(m_szPath)))
  376.         {       
  377.             TCHAR szDesktopINI[MAX_PATH];
  379.             PathCombine(szDesktopINI, m_szPath, c_szDesktopINI);
  380.             
  381.             GetPrivateProfileString(c_szChannel, c_szCDFURL, TEXT(""), m_szURL, 
  382.                                     ARRAYSIZE(m_szURL), szDesktopINI);
  384.             //m_wHotkey = GetPrivateProfileInt(c_szChannel, c_szHotkey, 0, szDesktopINI);
  385.         }
  386.         else
  387.         {
  388.             hr = E_FAIL;
  389.         }
  391.         ReleaseStgMedium(&stgmed);
  392.     }
  394.     return hr;
  395. }
  398. //
  399. // Helper functions
  400. //
  402. HRESULT CPropertyPages::InitializeSubsMgr2()
  403. {
  404.     HRESULT hr = E_FAIL;
  406. #ifndef UNIX
  407.     if (NULL != m_pSubscriptionMgr2)
  408.     {
  409.         hr = S_OK;
  410.     }
  411.     else
  412.     {
  413.         hr = CoInitialize(NULL);
  415.         if (SUCCEEDED(hr))
  416.         {
  417.             DLL_ForcePreloadDlls(PRELOAD_WEBCHECK);
  419.             hr = CoCreateInstance(CLSID_SubscriptionMgr, NULL,
  420.                                   CLSCTX_INPROC_SERVER, IID_ISubscriptionMgr2,
  421.                                   (void**)&m_pSubscriptionMgr2);
  423.             if (SUCCEEDED(hr))
  424.             {
  425.                 IShellExtInit* pIShellExtInit;
  427.                 hr = m_pSubscriptionMgr2->QueryInterface(IID_IShellExtInit, 
  428.                                                          (void **)&pIShellExtInit);
  429.                 if (SUCCEEDED(hr))
  430.                 {
  431.                     hr = pIShellExtInit->Initialize(NULL, m_pInitDataObject, NULL);
  432.                     pIShellExtInit->Release();
  433.                 }
  434.             }
  435.         }
  436.         CoUninitialize();
  437.     }
  438. #endif /* !UNIX */
  440.     return hr;
  441. }
  443. void CPropertyPages::ShowOfflineSummary(HWND hdlg, BOOL bShow)
  444. {
  445.     static const int offSumIDs[] =
  446.     {
  447.         IDC_SUMMARY,
  448.         IDC_LAST_SYNC_TEXT,
  449.         IDC_LAST_SYNC,
  450.         IDC_DOWNLOAD_SIZE_TEXT,
  451.         IDC_DOWNLOAD_SIZE,
  452.         IDC_DOWNLOAD_RESULT,
  453.         IDC_DOWNLOAD_RESULT_TEXT,
  454.         IDC_FREE_SPACE_TEXT
  455.     };
  457.     if (bShow)
  458.     {
  459.         TCHAR szLastSync[128];
  460.         TCHAR szDownloadSize[128];
  461.         TCHAR szDownloadResult[128];
  462.         WCHAR wszURL[INTERNET_MAX_URL_LENGTH];
  464.         MLLoadString(IDS_VALUE_UNKNOWN, szLastSync, ARRAYSIZE(szLastSync));
  465.         StrCpyN(szDownloadSize, szLastSync, ARRAYSIZE(szDownloadSize));
  466.         StrCpyN(szDownloadResult, szLastSync, ARRAYSIZE(szDownloadResult));
  468.         SHTCharToUnicode(m_szURL, wszURL, ARRAYSIZE(wszURL));
  470.         ASSERT(NULL != m_pSubscriptionMgr2);
  472.         if (NULL != m_pSubscriptionMgr2)
  473.         {
  474.             ISubscriptionItem *psi;
  475.             
  476.             if (SUCCEEDED(m_pSubscriptionMgr2->GetItemFromURL(wszURL, &psi)))
  477.             {
  478.                 enum { spLastSync, spDownloadSize, spDownloadResult };
  480.                 static const LPCWSTR pProps[] =
  481.                 { 
  482.                     c_szPropCompletionTime,
  483.                     c_szPropCrawlActualSize,
  484.                     c_szPropStatusString
  485.                 };
  486.                 VARIANT vars[ARRAYSIZE(pProps)];
  488.                 if (SUCCEEDED(psi->ReadProperties(ARRAYSIZE(pProps), pProps, vars)))
  489.                 {
  490.                     if (VT_DATE == vars[spLastSync].vt)
  491.                     {
  492.                         FILETIME ft, ft2;
  493.                         DWORD dwFlags = FDTF_DEFAULT;
  494.                         SYSTEMTIME st;
  496.                         VariantTimeToSystemTime(vars[spLastSync].date, &st);
  497.                         SystemTimeToFileTime(&st, &ft);
  498.                         LocalFileTimeToFileTime(&ft, &ft2);
  499.                         SHFormatDateTime(&ft2, &dwFlags, szLastSync, ARRAYSIZE(szLastSync));
  500.                     }
  502.                     if (VT_I4 == vars[spDownloadSize].vt)
  503.                     {
  504.                         StrFormatByteSize(vars[spDownloadSize].lVal * 1024, 
  505.                                           szDownloadSize, ARRAYSIZE(szDownloadSize));
  506.                     }
  508.                     if (VT_BSTR == vars[spDownloadResult].vt)
  509.                     {
  510.                     #ifdef UNICODE
  511.                         wnsprintf(szDownloadResult, ARRAYSIZE(szDownloadResult),
  512.                                   TEXT("%s"), vars[spDownloadResult].bstrVal);
  513.                     #else
  514.                         wnsprintf(szDownloadResult, ARRAYSIZE(szDownloadResult),
  515.                                   TEXT("%S"), vars[spDownloadResult].bstrVal);
  516.                     #endif
  517.                     }
  519.                     for (int i = 0; i < ARRAYSIZE(pProps); i++)
  520.                     {
  521.                         VariantClear(&vars[i]);
  522.                     }
  523.                 }
  524.                 psi->Release();
  525.             }
  526.         }
  528.         SetDlgItemText(hdlg, IDC_LAST_SYNC, szLastSync);
  529.         SetDlgItemText(hdlg, IDC_DOWNLOAD_SIZE, szDownloadSize);
  530.         SetDlgItemText(hdlg, IDC_DOWNLOAD_RESULT, szDownloadResult);
  531.     }
  533.     for (int i = 0; i < ARRAYSIZE(offSumIDs); i++)
  534.     {
  535.         ShowWindow(GetDlgItem(hdlg, offSumIDs[i]), bShow ? SW_SHOW : SW_HIDE);
  536.     }
  537. }
  539. BOOL CPropertyPages::OnInitDialog(HWND hdlg)
  540. {
  541.     TCHAR szName[MAX_PATH];
  542.     HICON hicon = NULL;
  543.     HRESULT hr;
  545.     CIconHandler *pIconHandler = new CIconHandler;
  547.     if (pIconHandler)
  548.     {
  549.         if (SUCCEEDED(pIconHandler->Load(m_szPath, 0)))
  550.         {
  551.             TCHAR szIconFile[MAX_PATH];
  552.             int iIndex;
  553.             UINT wFlags;
  554.             
  555.             if (SUCCEEDED(pIconHandler->GetIconLocation(0, szIconFile, ARRAYSIZE(szIconFile),
  556.                                                         &iIndex, &wFlags)))
  557.             {
  558.                 HICON hiconScrap = NULL;
  559.                 
  560.                 hr = pIconHandler->Extract(szIconFile, iIndex, &hicon, &hiconScrap, 
  561.                                            MAKELONG(GetSystemMetrics(SM_CXICON), 
  562.                                                     GetSystemMetrics(SM_CXSMICON)));
  564.                 if (S_FALSE == hr)
  565.                 {
  566.                     //  Do it ourselves
  567.                     hicon = ExtractIcon(g_hinst, szIconFile, iIndex);
  569.                 }
  570.                 else if ((NULL != hiconScrap) && (hicon != hiconScrap))
  571.                 {
  572.                     //  Otherwise cleanup unwanted little icon
  573.                     DestroyIcon(hiconScrap);
  574.                 }
  575.             }
  577.         }
  578.         pIconHandler->Release();
  579.     }
  581.     if (NULL == hicon)
  582.     {
  583.         hicon = LoadIcon(g_hinst, MAKEINTRESOURCE(IDI_CHANNEL));
  584.     }
  585.     
  586.     BOOL bEnableMakeOffline = TRUE;
  588.     SendDlgItemMessage(hdlg, IDC_ICONEX2, STM_SETICON, (WPARAM)hicon, 0);
  589.     StrCpyN(szName, m_szPath, ARRAYSIZE(szName));
  590.     PathStripPath(szName);
  592.     SetDlgItemText(hdlg, IDC_NAME, szName);
  593.     SetDlgItemText(hdlg, IDC_URL, m_szURL);
  595.     TCHAR szVisits[256];
  597.     szVisits[0] = 0;
  599.     CCdfView* pCCdfView = new CCdfView;
  601.     if (pCCdfView)
  602.     {
  603.         hr = pCCdfView->Load(m_szURL, 0);
  605.         if (SUCCEEDED(hr))
  606.         {
  607.             IXMLDocument* pIXMLDocument;
  609.             hr = pCCdfView->ParseCdf(NULL, &pIXMLDocument, PARSE_LOCAL);
  611.             if (SUCCEEDED(hr))
  612.             {
  613.                 IXMLElement*    pIXMLElement;
  614.                 LONG            nIndex;
  616.                 hr = XML_GetFirstChannelElement(pIXMLDocument,
  617.                                                 &pIXMLElement, &nIndex);
  619.                 if (SUCCEEDED(hr))
  620.                 {
  621.                     BSTR bstrURL = XML_GetAttribute(pIXMLElement, XML_HREF);
  623.                     if (bstrURL && *bstrURL)
  624.                     {
  625.                         BYTE cei[MAX_CACHE_ENTRY_INFO_SIZE];
  626.                         LPINTERNET_CACHE_ENTRY_INFO pcei = (LPINTERNET_CACHE_ENTRY_INFO)cei;
  627.                         DWORD cbcei = MAX_CACHE_ENTRY_INFO_SIZE;
  629.                         if (GetUrlCacheEntryInfoW(bstrURL, pcei, &cbcei))
  630.                         {
  631.                             wnsprintf(szVisits, ARRAYSIZE(szVisits), TEXT("%d"), 
  632.                                       pcei->dwHitRate);
  633.                         }
  634.                     }
  635.                     SysFreeString(bstrURL);
  637.                     pIXMLElement->Release();
  638.                 }
  640.                 pIXMLDocument->Release();
  641.             }
  642.         }
  643.         pCCdfView->Release();
  644.     }
  645.     
  646.     if (0 == szVisits[0])
  648.     {
  649.         MLLoadString(IDS_VALUE_UNKNOWN, szVisits, 
  650.                    ARRAYSIZE(szVisits));
  651.     }
  652.     SetDlgItemText(hdlg, IDC_VISITS, szVisits);
  653. /*
  654.     SendDlgItemMessage(hdlg, IDC_HOTKEY, HKM_SETRULES,
  655.                        (HKCOMB_NONE | HKCOMB_A | HKCOMB_C | HKCOMB_S),
  656.                        (HOTKEYF_CONTROL | HOTKEYF_ALT));
  658.     SendDlgItemMessage(hdlg, IDC_HOTKEY, HKM_SETHOTKEY, m_wHotkey, 0);
  659. */
  660.     WCHAR wszURL[INTERNET_MAX_URL_LENGTH];
  661.     SHTCharToUnicode(m_szURL, wszURL, ARRAYSIZE(wszURL));
  663.     CheckDlgButton(hdlg, IDC_MAKE_OFFLINE, m_bStartSubscribed ? 1 : 0);
  664.     
  665.     if (m_bStartSubscribed)
  666.     {
  667.         if (SHRestricted2(REST_NoRemovingSubscriptions, m_szURL, 0))
  668.         {
  669.             bEnableMakeOffline = FALSE;
  670.         }
  671.     }
  672.     else
  673.     {
  674.         if (SHRestricted2(REST_NoAddingSubscriptions, m_szURL, 0))
  675.         {
  676.             bEnableMakeOffline = FALSE;
  677.         }
  678.     }
  680.     if (!CanSubscribe(wszURL))
  681.     {
  682.         bEnableMakeOffline = FALSE;
  683.     }
  685.     if (!bEnableMakeOffline)
  686.     {
  687.         EnableWindow(GetDlgItem(hdlg, IDC_MAKE_OFFLINE), FALSE);
  688.     }
  690.     ShowOfflineSummary(hdlg, m_bStartSubscribed);
  692.     return TRUE;
  693. }
  695. BOOL AddSubsPropsCallback(HPROPSHEETPAGE hpage, LPARAM lParam)
  696. {
  697.     return (bool) PropSheet_AddPage((HWND)lParam, hpage);
  698. }
  700. void CPropertyPages::AddRemoveSubsPages(HWND hdlg, BOOL bAdd)
  701. {
  702.     ASSERT(NULL != m_pSubscriptionMgr2);
  704.     if (NULL != m_pSubscriptionMgr2)
  705.     {
  706.         if (bAdd)
  707.         {
  708.             IShellPropSheetExt *pspse;
  710.             if (SUCCEEDED(m_pSubscriptionMgr2->QueryInterface(IID_IShellPropSheetExt,
  711.                                                               (void **)&pspse)))
  712.             {
  713.                 pspse->AddPages(AddSubsPropsCallback, (LPARAM)GetParent(hdlg));
  714.                 pspse->Release();
  715.             }
  716.         }
  717.         else
  718.         {
  719.             ISubscriptionMgrPriv *psmp;
  721.             if (SUCCEEDED(m_pSubscriptionMgr2->QueryInterface(IID_ISubscriptionMgrPriv,
  722.                                                               (void **)&psmp)))
  723.             {
  724.                 psmp->RemovePages(GetParent(hdlg));
  725.                 psmp->Release();
  726.             }
  727.         }
  729.         ShowOfflineSummary(hdlg, bAdd);
  730.     }
  731. }
  733. BOOL CPropertyPages::OnCommand(HWND hdlg, WORD wNotifyCode, WORD wID, HWND hwndCtl)
  734. {
  735.     BOOL bHandled = TRUE;
  736.     switch (wID)
  737.     {
  738.         case IDC_MAKE_OFFLINE:
  739.             if (wNotifyCode == BN_CLICKED)
  740.             {
  741.                 AddRemoveSubsPages(hdlg, IsDlgButtonChecked(hdlg, IDC_MAKE_OFFLINE));
  742.                 PropSheet_Changed(GetParent(hdlg), hdlg);
  743.             }
  744.             break;
  745. /*
  746.         case IDC_HOTKEY:
  747.             if (wNotifyCode == EN_CHANGE)
  748.             {
  749.                 PropSheet_Changed(GetParent(hdlg), hdlg);
  750.             }
  751.             break;
  752. */
  753.         default:
  754.             bHandled = FALSE;
  755.             break;
  756.     }
  758.     return bHandled;
  759. }
  761. BOOL CPropertyPages::OnNotify(HWND hdlg, WPARAM idCtrl, LPNMHDR pnmh)
  762. {
  763.     BOOL bHandled = FALSE;
  765.     switch (pnmh->code)
  766.     {
  767.         case PSN_APPLY:
  768.         {
  769.         /*
  770.             TCHAR szHotkey[32];
  771.             TCHAR szDesktopINI[MAX_PATH];
  772.             WORD wOldHotkey = m_wHotkey;
  774.             m_wHotkey = (WORD)SendDlgItemMessage(hdlg, IDC_HOTKEY, HKM_GETHOTKEY, 0, 0);
  775.             wnsprintf(szHotkey, ARRAYSIZE(szHotkey), TEXT("%d"), m_wHotkey);
  777.             PathCombine(szDesktopINI, m_szPath, c_szDesktopINI);
  778.             WritePrivateProfileString(c_szChannel, c_szHotkey, szHotkey, szDesktopINI);
  780.             RegisterGlobalHotkey(wOldHotkey, m_wHotkey, m_szPath);
  781.         */
  782.             BOOL bIsSubscribed = IsDlgButtonChecked(hdlg, IDC_MAKE_OFFLINE);
  784.             if (!bIsSubscribed)
  785.             {
  786.                 WCHAR wszURL[INTERNET_MAX_URL_LENGTH];
  788.                 SHTCharToUnicode(m_szURL, wszURL, ARRAYSIZE(wszURL));
  789.                 
  790.                 if (NULL != m_pSubscriptionMgr2) 
  791.                 {
  792.                     m_pSubscriptionMgr2->DeleteSubscription(wszURL, NULL);
  793.                 }
  794.             }
  795.             else
  796.             {
  797.                 ISubscriptionMgrPriv *psmp;
  799.                 if ((NULL != m_pSubscriptionMgr2) &&
  800.                     SUCCEEDED(m_pSubscriptionMgr2->QueryInterface(IID_ISubscriptionMgrPriv,
  801.                                                                   (void **)&psmp)))
  802.                 {
  803.                     psmp->SaveSubscription();
  804.                     psmp->Release();
  805.                 }                
  806.             }
  808.             bHandled = TRUE;
  809.             break;
  810.         }
  811.     }
  813.     return bHandled;
  814. }
  816. void CPropertyPages::OnDestroy(HWND hdlg)
  817. {
  818.     if (!m_bStartSubscribed && 
  819.         IsDlgButtonChecked(hdlg, IDC_MAKE_OFFLINE) && 
  820.         (NULL != m_pSubscriptionMgr2))
  821.     {
  822.         WCHAR wszURL[INTERNET_MAX_URL_LENGTH];
  824.         SHTCharToUnicode(m_szURL, wszURL, ARRAYSIZE(wszURL));
  826.         m_pSubscriptionMgr2->UpdateSubscription(wszURL);
  827.     }
  829.     //  Ensure sys bit is still set
  830.     SetFileAttributes(m_szPath, FILE_ATTRIBUTE_SYSTEM);
  831. }
  833. UINT CPropertyPages::PropSheetCallback(HWND hwnd, UINT uMsg, LPPROPSHEETPAGE ppsp)
  834. {
  835.     switch (uMsg)
  836.     {
  837.         case PSPCB_RELEASE:
  838.             if (NULL != ppsp->lParam)
  839.             {
  840.                 ((CPropertyPages *)ppsp->lParam)->Release();
  841.             }
  842.             break;
  843.     }
  845.     return 1;
  846. }
  848. TCHAR c_szHelpFile[] = TEXT("iexplore.hlp");
  850. DWORD aHelpIDs[] = {
  851.     IDC_NAME,                   IDH_SUBPROPS_SUBTAB_SUBSCRIBED_NAME,
  852.     IDC_URL_TEXT,               IDH_SUBPROPS_SUBTAB_SUBSCRIBED_URL,
  853.     IDC_URL,                    IDH_SUBPROPS_SUBTAB_SUBSCRIBED_URL,
  854. //    IDC_HOTKEY_TEXT,            IDH_WEBDOC_HOTKEY,
  855. //    IDC_HOTKEY,                 IDH_WEBDOC_HOTKEY,
  856.     IDC_VISITS_TEXT,            IDH_WEBDOC_VISITS,
  857.     IDC_VISITS,                 IDH_WEBDOC_VISITS,
  858.     IDC_MAKE_OFFLINE,           IDH_MAKE_AVAIL_OFFLINE,
  859.     IDC_SUMMARY,                IDH_GROUPBOX,
  860.     IDC_LAST_SYNC_TEXT,         IDH_SUBPROPS_SUBTAB_LAST,
  861.     IDC_LAST_SYNC,              IDH_SUBPROPS_SUBTAB_LAST,
  862.     IDC_DOWNLOAD_SIZE_TEXT,     IDH_SUBPROPS_DLSIZE,
  863.     IDC_DOWNLOAD_SIZE,          IDH_SUBPROPS_DLSIZE,
  864.     IDC_DOWNLOAD_RESULT_TEXT,   IDH_SUBPROPS_SUBTAB_RESULT,
  865.     IDC_DOWNLOAD_RESULT,        IDH_SUBPROPS_SUBTAB_RESULT,
  866.     0, 0
  867. };
  870. INT_PTR CPropertyPages::PropSheetDlgProc(HWND hdlg, UINT msg, WPARAM wParam, LPARAM lParam)
  871. {
  872.     INT_PTR lrHandled = FALSE;
  873.     CPropertyPages *pThis;
  874.     
  875.     switch (msg)
  876.     {
  877.         case WM_INITDIALOG:
  878.         {
  879.             LPPROPSHEETPAGE pPropSheetPage = (LPPROPSHEETPAGE)lParam;
  881.             ASSERT(NULL != pPropSheetPage);
  882.             if (NULL != pPropSheetPage)
  883.             {
  884.                 SetWindowLongPtr(hdlg, DWLP_USER, pPropSheetPage->lParam);
  885.             }
  886.             
  887.             pThis = GetThis(hdlg);
  889.             if (NULL != pThis)
  890.             {               
  891.                 lrHandled = pThis->OnInitDialog(hdlg);
  892.             }
  894.             break;
  895.         }
  897.         case WM_COMMAND:
  898.             pThis = GetThis(hdlg);
  899.             if (NULL != pThis)
  900.             {               
  901.                 lrHandled = pThis->OnCommand(hdlg, HIWORD(wParam), LOWORD(wParam), (HWND)lParam);
  902.             }
  903.             break;
  905.         case WM_NOTIFY:
  906.             pThis = GetThis(hdlg);
  908.             if (NULL != pThis)
  909.             {               
  910.                 lrHandled = pThis->OnNotify(hdlg, wParam, (LPNMHDR)lParam);
  911.             }
  912.             break;
  914.         case WM_DESTROY:
  915.             pThis = GetThis(hdlg);
  917.             if (NULL != pThis)
  918.             {
  919.                 pThis->OnDestroy(hdlg);
  920.             }
  921.             break;
  923.         case WM_HELP:
  924.             SHWinHelpOnDemandWrap((HWND)((LPHELPINFO) lParam)->hItemHandle, c_szHelpFile,
  925.                     HELP_WM_HELP, (DWORD_PTR) aHelpIDs);
  926.             lrHandled = TRUE;
  927.             break;
  929.         case WM_CONTEXTMENU:
  930.             SHWinHelpOnDemandWrap((HWND)wParam, c_szHelpFile, HELP_CONTEXTMENU, (DWORD_PTR)aHelpIDs);
  931.             lrHandled = TRUE;
  932.             break;
  933.     }
  934.     
  935.     return lrHandled;
  936. }