Category

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

Windows Kernel

Development Platform:

Visual C++

delagent.h：Code Content
  1. //
  2. // Delivery Agents base class
  4. #ifndef _DELAGENT_H
  5. #define _DELAGENT_H
  7. #include "offline.h"
  9. #define INET_S_AGENT_BASIC_SUCCESS           _HRESULT_TYPEDEF_(0x000C0FFEL)
  11. class CDeliveryAgent :  public ISubscriptionAgentControl,
  12.                         public IShellPropSheetExt,
  13. #ifdef UNICODE
  14.                         public IExtractIconA,
  15. #endif
  16.                         public IExtractIcon,
  17.                         public ISubscriptionAgentShellExt
  18. {
  19. private:
  20. // Data for our OLE support
  21.     ULONG           m_cRef;
  23. #ifdef AGENT_AUTODIAL
  24.     enum DIALER_STATUS { DIALER_OFFLINE, DIALER_CONNECTING, DIALER_ONLINE };
  25.     DIALER_STATUS   m_iDialerStatus;
  26. #endif
  28.     enum {
  29.         FLAG_BUSY          =0x00010000,   // addrefed ourselves; between begin & end reports
  30.         FLAG_PAUSED        =0x00020000,   // We are paused
  31.         FLAG_OPSTARTED     =0x00040000,   // We've entered StartOperation
  32.     };
  34.     // Derived agents can use high 8 bits of this field
  35.     DWORD       m_dwAgentFlags;
  37.     void        SendUpdateBegin();
  38.     void        SendUpdateEnd();
  39.     HRESULT     ProcessEndItem(ISubscriptionItem *pEndItem);
  42. protected:
  43.     // Upper 16 bits allowable here
  44.     enum    {
  45.         FLAG_HOSTED        =0x00100000,     // hosted by another delivery agent
  46.         FLAG_CHANGESONLY   =0x00200000,     // We're in "Changes Only" mode
  47.         FLAG_WAITING_FOR_INCREASED_CACHE = 0x00400000, // Special paused state
  48.     };
  50.     POOEBuf         m_pBuf;
  51.     HPROPSHEETPAGE  m_hPage[MAX_WC_AGENT_PAGES];
  53.     ISubscriptionAgentEvents *m_pAgentEvents;
  54.     ISubscriptionItem        *m_pSubscriptionItem;
  55.     
  56.     SUBSCRIPTIONCOOKIE      m_SubscriptionCookie;
  58.     long        m_lSizeDownloadedKB;    // Size downloaded in KB
  60.     SCODE       m_scEndStatus;
  62.     void        SendUpdateNone();   // Call from StartOperation if we won't be doing anything
  64.     void        SendUpdateProgress(LPCWSTR pwszURL, long lProgress, long lMax, long lCurSizeKB=-1);
  66.     BOOL        IsAgentFlagSet(int iFlag) { return (m_dwAgentFlags & iFlag); }
  67.     void        ClearAgentFlag(int iFlag) { m_dwAgentFlags &= ~iFlag; }
  68.     void        SetAgentFlag(int iFlag) { m_dwAgentFlags |= iFlag; }
  69.     
  70.     HRESULT     CheckResponseCode(DWORD dwHttpResponseCode);    // Also sets EndStatus. E_ABORT on error.
  72. //  DIALER_STATUS GetDialerStatus() { return m_iDialerStatus; }
  73.     void          SetEndStatus(SCODE sc) { m_scEndStatus = sc; }
  75.     virtual ~CDeliveryAgent();
  77. public:
  78.     CDeliveryAgent();
  80.     BOOL        GetBusy() { return IsAgentFlagSet(FLAG_BUSY); }
  81.     BOOL        IsPaused() { return IsAgentFlagSet(FLAG_PAUSED); }
  83.     SCODE       GetEndStatus() { return m_scEndStatus; }
  85.     // IUnknown members
  86.     STDMETHODIMP         QueryInterface(REFIID riid, void **punk);
  87.     STDMETHODIMP_(ULONG) AddRef(void);
  88.     STDMETHODIMP_(ULONG) Release(void);
  90.     // ISubscriptionAgentControl members
  91.     STDMETHODIMP    StartUpdate(IUnknown *pItem, IUnknown *punkAdvise);
  92.     STDMETHODIMP    PauseUpdate(DWORD dwFlags);
  93.     STDMETHODIMP    ResumeUpdate(DWORD dwFlags);
  94.     STDMETHODIMP    AbortUpdate(DWORD dwFlags);
  95.     STDMETHODIMP    SubscriptionControl(IUnknown *pItem, DWORD dwControl);   // Called on delete
  97.     // IShellPropSheetExt members
  98.     STDMETHODIMP    AddPages(LPFNADDPROPSHEETPAGE, LPARAM);
  99.     STDMETHODIMP    ReplacePage(UINT, LPFNADDPROPSHEETPAGE, LPARAM);
  101.     // ISubscriptionAgentShellExt
  102.     STDMETHODIMP    Initialize(SUBSCRIPTIONCOOKIE *pSubscriptionCookie, 
  103.                                LPCWSTR pwszURL, LPCWSTR pwszName, 
  104.                                SUBSCRIPTIONTYPE subsType);
  105.     STDMETHODIMP    RemovePages(HWND hdlg);
  106.     STDMETHODIMP    SaveSubscription();
  107.     STDMETHODIMP    URLChange(LPCWSTR pwszNewURL);
  109. #ifdef UNICODE
  110.     //  IExtractIconA
  111.     STDMETHODIMP    GetIconLocation(UINT uFlags, LPSTR szIconFile, UINT cchMax, int * piIndex, UINT * pwFlags);
  112.     STDMETHODIMP    Extract(LPCSTR pszFile, UINT nIconIndex, HICON * phiconLarge, HICON * phiconSmall, UINT nIconSize);
  113. #endif
  115.     //  IExtractIconT
  116.     STDMETHODIMP    GetIconLocation(UINT uFlags, LPTSTR szIconFile, UINT cchMax, int * piIndex, UINT * pwFlags);
  117.     STDMETHODIMP    Extract(LPCTSTR pszFile, UINT nIconIndex, HICON * phiconLarge, HICON * phiconSmall, UINT nIconSize);
  119. private:
  120.     // Functions we provide common implementations for
  121.     HRESULT DoStartDownload();
  123. #ifdef AGENT_AUTODIAL
  124.     HRESULT NotifyAutoDialer();
  126.     HRESULT OnInetOnline();
  127.     HRESULT OnInetOffline();
  128. #endif
  130. protected:
  131.     // Virtual functions for our derived classes to override as necessary
  132.     // We provide implementations which should be called after processing
  133.     virtual HRESULT     AgentPause(DWORD dwFlags);
  134.     virtual HRESULT     AgentResume(DWORD dwFlags);
  135.     virtual HRESULT     AgentAbort(DWORD dwFlags);
  137.     virtual HRESULT     ModifyUpdateEnd(ISubscriptionItem *pEndItem, UINT *puiRes);
  139.     virtual HRESULT     StartOperation();       // connects to internet
  140.     virtual HRESULT     StartDownload() = 0;    // we just got connected
  141.     virtual void        CleanUp();
  142. };
  144. #endif // _DELAGENT_H