Category

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

Windows Kernel

Development Platform:

Visual C++

cdfsubs.cpp：Code Content
  1. #include <windows.h>
  2. #include <commctrl.h>
  3. #include <stdio.h>
  4. #include <memory.h>
  5. #include <subsmgr.h>
  6. #include <mstask.h>
  7. #include <wininet.h>
  8. #include "cdfsubs.hpp"
  10. // CDFAutoSubscribe
  11. //
  12. // Create the cache entry and subscribe to the channel.
  13. //
  14. // BUGBUG: Right now, if you create the cache entry for the CDF, subscribing
  15. // to the channel doesn't work properly. Updating the channel will fail.
  16. //
  18. BOOL WINAPI CDFAutoSubscribe(HWND hwnd, LPSTR szFriendlyName, LPSTR szCDFUrl,
  19.                              LPSTR szCDFFilePath, BOOL bSilent)
  20. {
  21.     BOOL                   bRet = FALSE;
  23.     if (CreateCDFCacheEntry(szCDFUrl, szCDFFilePath)) {
  24.         if (SubscribeChannel(hwnd, szFriendlyName, szCDFUrl, bSilent)) {
  25.             bRet = TRUE;
  26.         }
  27.     }
  29.     return bRet;
  30. }
  32. //
  33. // CreateCDFCacheEntry
  34. //
  36. BOOL CreateCDFCacheEntry(LPSTR szCDFUrl, LPSTR szLocalFileName)
  37. {
  38.     BOOL                           bRet;
  39.     FILETIME                       ftExpireTime;
  40.     FILETIME                       ftLastMod;
  41.     TCHAR                          achFileName[MAX_PATH];
  43.     bRet = FALSE;
  44.     GetSystemTimeAsFileTime(&ftLastMod);
  45.     ftExpireTime.dwLowDateTime = (DWORD)0;
  46.     ftExpireTime.dwHighDateTime = (DWORD)0;
  48.     if (CreateUrlCacheEntry(szCDFUrl, 0, NULL, achFileName, 0)) {
  49.         CopyFile(szLocalFileName, achFileName, FALSE);
  50.         if (CommitUrlCacheEntry(szCDFUrl, achFileName, ftExpireTime,
  51.                                 ftLastMod, 0, NULL, 0, NULL, 0)) {
  52.             bRet = TRUE;
  53.         }
  54.     }
  56.     return bRet;
  57. }
  59. //
  60. // SubscribeChannel
  61. //
  63. BOOL SubscribeChannel(HWND hwnd, LPSTR szName, LPSTR szURL, BOOL bSilent)
  64. {
  65.     ISubscriptionMgr              *pISubscriptionMgr = NULL;
  66.     DWORD                          dwFlags = 0;
  67.     BOOL                           bRet = FALSE;
  68.     HRESULT                        hr = S_OK;
  69.     SUBSCRIPTIONTYPE               subsType = SUBSTYPE_CHANNEL;
  70.     SUBSCRIPTIONINFO               subsInfo;
  71.     LPWSTR                         wzName = NULL;
  72.     LPWSTR                         wzURL = NULL;
  74.     if (FAILED(Ansi2Unicode(szName, &wzName))) {
  75.         goto Exit;
  76.     }
  78.     if (FAILED(Ansi2Unicode(szURL, &wzURL))) {
  79.         goto Exit;
  80.     }
  82.     hr = CoCreateInstance(CLSID_SubscriptionMgr, NULL, CLSCTX_INPROC_SERVER,
  83.                           IID_ISubscriptionMgr, (void **)&pISubscriptionMgr);
  84.     if (FAILED(hr)) {
  85.         goto Exit;
  86.     }
  87.                           
  88.     ZeroMemory(&subsInfo, sizeof(SUBSCRIPTIONINFO));
  89.     subsInfo.cbSize = sizeof(SUBSCRIPTIONINFO);
  90.     subsInfo.fUpdateFlags = SUBSINFO_CHANNELFLAGS | SUBSINFO_TASKFLAGS | SUBSINFO_GLEAM;
  91.     subsInfo.bGleam = TRUE;
  93.     subsInfo.fChannelFlags |= CHANNEL_AGENT_PRECACHE_ALL |
  94.                               CHANNEL_AGENT_DYNAMIC_SCHEDULE;
  96.     subsInfo.fTaskFlags = TASK_FLAG_START_ONLY_IF_IDLE | 
  97.                           TASK_FLAG_RUN_IF_CONNECTED_TO_INTERNET;
  98.     
  99.     dwFlags = CREATESUBS_ADDTOFAVORITES | CREATESUBS_SOFTWAREUPDATE;
  100.     if (bSilent) {
  101.         dwFlags |= CREATESUBS_NOUI;
  102.     }
  104.     hr = pISubscriptionMgr->CreateSubscription(hwnd, wzURL, wzName,
  105.                                                dwFlags, subsType, &subsInfo);
  107.     if (hr == S_OK) {
  108.         bRet = TRUE;
  109.     }
  111.     pISubscriptionMgr->Release();
  113. Exit:
  115.     if (wzName) {
  116.         delete wzName;
  117.     }
  119.     if (wzURL) {
  120.         delete wzURL;
  121.     }
  123.     return bRet;
  124. }
  126. // Ansi2Unicode: Convert ANSI strings to Unicode
  127. // Ripped from urlmondownloadhelpers.cxx
  129. HRESULT Ansi2Unicode(const char * src, wchar_t **dest)
  130. {
  131.     if ((src == NULL) || (dest == NULL))
  132.         return E_INVALIDARG;
  134.     // find out required buffer size and allocate it
  135.     int len = MultiByteToWideChar(CP_ACP, MB_PRECOMPOSED, src, -1, NULL, 0);
  136.     *dest = new WCHAR [len*sizeof(WCHAR)];
  137.     if (!*dest)
  138.         return E_OUTOFMEMORY;
  140.     // Do the actual conversion.
  141.     if ((MultiByteToWideChar(CP_ACP, MB_PRECOMPOSED, src, -1, *dest, 
  142.                                                     len*sizeof(wchar_t))) != 0)
  143.         return S_OK; 
  144.     else
  145.         return E_FAIL;
  146. }