289 lines
13 KiB
C
289 lines
13 KiB
C
|
#ifndef _DESKSTAT_H_
|
||
|
|
#define _DESKSTAT_H_
|
||
|
|
|
||
|
|
#include "mshtml.h"
|
||
|
|
|
||
|
|
|
||
|
|
|
||
|
|
// NOTE: This is NOT a IE4COMPONENTA struct, it is a IE4COMPONENTT struct.
|
||
|
|
// Note: This is the old COMPONENTA structure used in IE4. It is kept here for compatibility.
|
||
|
|
typedef struct _tagIE4COMPONENTA
|
||
|
|
{
|
||
|
|
DWORD dwSize;
|
||
|
|
DWORD dwID; //Don't make this DWORD_PTR. The public version of this struct uses DWORD here.
|
||
|
|
int iComponentType;
|
||
|
|
BOOL fChecked;
|
||
|
|
BOOL fDirty;
|
||
|
|
BOOL fNoScroll;
|
||
|
|
COMPPOS cpPos;
|
||
|
|
TCHAR szFriendlyName[MAX_PATH];
|
||
|
|
TCHAR szSource[INTERNET_MAX_URL_LENGTH];
|
||
|
|
TCHAR szSubscribedURL[INTERNET_MAX_URL_LENGTH];
|
||
|
|
} IE4COMPONENTA;
|
||
|
|
typedef IE4COMPONENTA *LPIE4COMPONENTA;
|
||
|
|
typedef const IE4COMPONENTA *LPCIE4COMPONENTA;
|
||
|
|
|
||
|
|
// NOTE: This is NOT a COMPONENTA struct, it is a COMPONENTT struct.
|
||
|
|
|
||
|
|
// Note: This is the new NT5 COMPONENT structure. The old component structure is kept at the
|
||
|
|
// begining of this struct and the new fields are added at the end. The dwSize field is used to
|
||
|
|
// distinguish between the old and new structures.
|
||
|
|
//
|
||
|
|
typedef struct _tagCOMPONENTA
|
||
|
|
{
|
||
|
|
DWORD dwSize;
|
||
|
|
DWORD dwID; //Don't make this DWORD_PTR. The public version of this struct uses DWORD here.
|
||
|
|
int iComponentType;
|
||
|
|
BOOL fChecked;
|
||
|
|
BOOL fDirty;
|
||
|
|
BOOL fNoScroll;
|
||
|
|
COMPPOS cpPos;
|
||
|
|
TCHAR szFriendlyName[MAX_PATH];
|
||
|
|
TCHAR szSource[INTERNET_MAX_URL_LENGTH];
|
||
|
|
TCHAR szSubscribedURL[INTERNET_MAX_URL_LENGTH];
|
||
|
|
// Add the new fields below this point. Everything above must exactly match the
|
||
|
|
// old IE4COMPONENTA structure for compatibility.
|
||
|
|
DWORD dwCurItemState;
|
||
|
|
COMPSTATEINFO csiOriginal;
|
||
|
|
COMPSTATEINFO csiRestored;
|
||
|
|
} COMPONENTA;
|
||
|
|
typedef COMPONENTA *LPCOMPONENTA;
|
||
|
|
typedef const COMPONENTA *LPCCOMPONENTA;
|
||
|
|
|
||
|
|
typedef struct _tagTAGENTRYA
|
||
|
|
{
|
||
|
|
LPCSTR pszTag;
|
||
|
|
BOOL fSkipPast;
|
||
|
|
} TAGENTRYA;
|
||
|
|
|
||
|
|
typedef struct _tagTAGENTRYW
|
||
|
|
{
|
||
|
|
LPCWSTR pwszTag;
|
||
|
|
BOOL fSkipPast;
|
||
|
|
} TAGENTRYW;
|
||
|
|
|
||
|
|
#define c_wszNULL (L"")
|
||
|
|
|
||
|
|
//The following are the possible values for _iCharset field of CReadFileObj
|
||
|
|
#define ANSI_HTML_CHARSET 1
|
||
|
|
#define UNICODE_HTML_CHARSET 2
|
||
|
|
|
||
|
|
//
|
||
|
|
// The following class implements a uniform way to read chars from an ANSI or UNICODE file.
|
||
|
|
//
|
||
|
|
|
||
|
|
class CReadFileObj
|
||
|
|
{
|
||
|
|
friend class CActiveDesktop;
|
||
|
|
|
||
|
|
private:
|
||
|
|
HANDLE _hFile; //File handle
|
||
|
|
int _iCharset; //ANSI_HTML_CHARSET or UNICODE_HTML_CHARSET
|
||
|
|
|
||
|
|
public:
|
||
|
|
//Constructor
|
||
|
|
CReadFileObj(LPCTSTR lpszFileName);
|
||
|
|
//Destructor
|
||
|
|
~CReadFileObj();
|
||
|
|
//Reads the given number of ansi characters, if an ansi file.
|
||
|
|
HRESULT FileReadCharsA(LPSTR lpszBuff, UINT uiCharsToRead, UINT *puiCharsActuallyRead);
|
||
|
|
//Reads the given number of UNICODE characters, if a UNICODE file.
|
||
|
|
HRESULT FileReadCharsW(LPWSTR lpwszBuff, UINT uiCharsToRead, UINT *puiCharsActuallyRead);
|
||
|
|
//Reads the given number of characters, and convert them if necessary to ANSI or UNICODE based on destCharset.
|
||
|
|
HRESULT FileReadAndConvertChars(int iDestCharset, LPWSTR lpwszBuff, UINT uiCharsToRead, UINT *puiCharsActuallyRead, UINT *puiCharsConverted);
|
||
|
|
//Seeks to a given character offset (not byte offset).
|
||
|
|
HRESULT FileSeekChars(LONG lCharOffset, DWORD dwOrigin);
|
||
|
|
//Gets the offset of the current file pointer from the begining of the file.
|
||
|
|
HRESULT FileGetCurCharOffset(LONG *plCharOffset);
|
||
|
|
};
|
||
|
|
|
||
|
|
interface IShellImageDataFactory;
|
||
|
|
|
||
|
|
class CActiveDesktop : public IActiveDesktop, IActiveDesktopP, IADesktopP2, IPropertyBag
|
||
|
|
{
|
||
|
|
public:
|
||
|
|
// *** IUnknown ***
|
||
|
|
virtual STDMETHODIMP QueryInterface(REFIID riid, LPVOID *ppvObj);
|
||
|
|
virtual STDMETHODIMP_(ULONG) AddRef(void);
|
||
|
|
virtual STDMETHODIMP_(ULONG) Release(void);
|
||
|
|
|
||
|
|
// *** IActiveDesktop ***
|
||
|
|
virtual STDMETHODIMP ApplyChanges(DWORD dwFlags);
|
||
|
|
virtual STDMETHODIMP GetWallpaper(LPWSTR pwszWallpaper, UINT cchWallpaper, DWORD dwReserved);
|
||
|
|
virtual STDMETHODIMP SetWallpaper(LPCWSTR pwszWallpaper, DWORD dwReserved);
|
||
|
|
virtual STDMETHODIMP GetWallpaperOptions(LPWALLPAPEROPT pwpo, DWORD dwReserved);
|
||
|
|
virtual STDMETHODIMP SetWallpaperOptions(LPCWALLPAPEROPT pwpo, DWORD dwReserved);
|
||
|
|
virtual STDMETHODIMP GetPattern(LPWSTR pwszPattern, UINT cchPattern, DWORD dwReserved);
|
||
|
|
virtual STDMETHODIMP SetPattern(LPCWSTR pszPattern, DWORD dwReserved);
|
||
|
|
virtual STDMETHODIMP GetDesktopItemOptions(LPCOMPONENTSOPT pco, DWORD dwReserved);
|
||
|
|
virtual STDMETHODIMP SetDesktopItemOptions(LPCCOMPONENTSOPT pco, DWORD dwReserved);
|
||
|
|
virtual STDMETHODIMP AddDesktopItem(LPCCOMPONENT pcomp, DWORD dwReserved);
|
||
|
|
virtual STDMETHODIMP AddDesktopItemWithUI(HWND hwnd, LPCOMPONENT pcomp, DWORD dwReserved);
|
||
|
|
virtual STDMETHODIMP ModifyDesktopItem(LPCCOMPONENT pcomp, DWORD dwFlags);
|
||
|
|
virtual STDMETHODIMP RemoveDesktopItem(LPCCOMPONENT pcomp, DWORD dwReserved);
|
||
|
|
virtual STDMETHODIMP GetDesktopItemCount(LPINT lpiCount, DWORD dwReserved);
|
||
|
|
virtual STDMETHODIMP GetDesktopItem(int nComponent, LPCOMPONENT pcomp, DWORD dwReserved);
|
||
|
|
virtual STDMETHODIMP GetDesktopItemByID(ULONG_PTR dwID, LPCOMPONENT pcomp, DWORD dwReserved);
|
||
|
|
virtual STDMETHODIMP GenerateDesktopItemHtml(LPCWSTR pwszFileName, LPCOMPONENT pcomp, DWORD dwReserved);
|
||
|
|
virtual STDMETHODIMP AddUrl(HWND hwnd, LPCWSTR pszSource, LPCOMPONENT pcomp, DWORD dwFlags);
|
||
|
|
virtual STDMETHODIMP GetDesktopItemBySource(LPCWSTR pszSource, LPCOMPONENT pcomp, DWORD dwReserved);
|
||
|
|
// *** IActiveDesktopP ***
|
||
|
|
virtual STDMETHODIMP SetSafeMode(DWORD dwFlags);
|
||
|
|
virtual STDMETHODIMP EnsureUpdateHTML(void);
|
||
|
|
virtual STDMETHODIMP SetScheme(LPCWSTR pwszSchemeName, DWORD dwFlags);
|
||
|
|
virtual STDMETHODIMP GetScheme(LPWSTR pwszSchemeName, LPDWORD lpdwcchBuffer, DWORD dwFlags);
|
||
|
|
// *** IADesktopP2 ***
|
||
|
|
virtual STDMETHODIMP ReReadWallpaper(void);
|
||
|
|
virtual STDMETHODIMP GetADObjectFlags(LPDWORD lpdwFlags, DWORD dwMask);
|
||
|
|
virtual STDMETHODIMP UpdateAllDesktopSubscriptions();
|
||
|
|
virtual STDMETHODIMP MakeDynamicChanges(IOleObject *pOleObj);
|
||
|
|
// *** IPropertyBag ***
|
||
|
|
virtual STDMETHODIMP Read(LPCOLESTR pszPropName, VARIANT *pVar, IErrorLog *pErrorLog);
|
||
|
|
virtual STDMETHODIMP Write(LPCOLESTR pszPropName, VARIANT *pVar);
|
||
|
|
|
||
|
|
//Some Ansi versions of the methods for Internal Use
|
||
|
|
BOOL AddComponentPrivate(COMPONENTA *pcomp, DWORD dwID);
|
||
|
|
BOOL UpdateComponentPrivate(int iIndex, COMPONENTA *pcomp);
|
||
|
|
BOOL RemoveComponentPrivate(int iIndex, COMPONENTA *pcomp);
|
||
|
|
BOOL GetComponentPrivate(int nComponent, COMPONENTA *pcomp);
|
||
|
|
|
||
|
|
CActiveDesktop();
|
||
|
|
|
||
|
|
protected:
|
||
|
|
DWORD _cRef;
|
||
|
|
DWORD _dwNextID;
|
||
|
|
HDSA _hdsaComponent;
|
||
|
|
TCHAR _szSelectedWallpaper[MAX_PATH];
|
||
|
|
TCHAR _szSelectedWallpaperConverted[MAX_PATH];
|
||
|
|
TCHAR _szBackupWallpaper[MAX_PATH];
|
||
|
|
TCHAR _szSelectedPattern[MAX_PATH];
|
||
|
|
LPTSTR _pszScheme;
|
||
|
|
WALLPAPEROPT _wpo;
|
||
|
|
COMPONENTSOPT _co;
|
||
|
|
HANDLE _hFileHtml; //File handle of desktop.htt
|
||
|
|
CReadFileObj* _pReadFileObjHtmlBkgd; //Object used to read the background html wallpaper
|
||
|
|
int _iDestFileCharset; //Whether the dest file desktop.htt is ANSI or UNICODE.
|
||
|
|
|
||
|
|
// used for the temp wallpaper
|
||
|
|
LPWSTR _pszWallpaperInUse; // If using a temp file, keep the name in use so we don't stomp it while the user is previewing other files.
|
||
|
|
FILETIME _ftLastWrite; // The date that the original file was last written to.
|
||
|
|
LPTSTR _pszOrigLastApplied; // Same as _pszOriginalFile except it doesn't change until apply is pressed.
|
||
|
|
|
||
|
|
// put the BOOLs together
|
||
|
|
BOOL _fDirty;
|
||
|
|
BOOL _fWallpaperDirty;
|
||
|
|
BOOL _fWallpaperChangedDuringInit;
|
||
|
|
BOOL _fPatternDirty;
|
||
|
|
BOOL _fSingleItem;
|
||
|
|
BOOL _fInitialized;
|
||
|
|
BOOL _fNeedBodyEnd;
|
||
|
|
BOOL _fNoDeskMovr;
|
||
|
|
BOOL _fBackgroundHtml;
|
||
|
|
IStream *_pStream; //To generate individual components.
|
||
|
|
BOOL _fUseDynamicHtml; //Indicates if Dynamic Html can be used to make changes.
|
||
|
|
BOOL _fPolicyForWPName; // A policy exists for the wallpaper to be used.
|
||
|
|
BOOL _fPolicyForWPStyle; // A policy exists for the wallpaper style to be used.
|
||
|
|
BOOL _fIgnoreAddRemovePolicies; //If this is set, ignore the policy to add remove items.
|
||
|
|
|
||
|
|
~CActiveDesktop();
|
||
|
|
|
||
|
|
int _FindComponentIndexByID(DWORD dwID);
|
||
|
|
int _FindComponentBySource(LPTSTR lpszSource, COMPONENTA *pComp);
|
||
|
|
int _FindComponentBySrcUrl(LPTSTR lpszSrcUrl, COMPONENTA *pComp);
|
||
|
|
|
||
|
|
void _ReadComponent(HKEY hkey, LPCTSTR pszComp);
|
||
|
|
void _SortAndRationalize(void);
|
||
|
|
void _ReadComponents(BOOL fActiveDesktop);
|
||
|
|
void _ReadWallpaper(BOOL fActiveDesktop);
|
||
|
|
void _ReadPattern(void);
|
||
|
|
void _Initialize(void);
|
||
|
|
|
||
|
|
void _SaveComponent(HKEY hkey, int iIndex, COMPONENTA *pcomp);
|
||
|
|
void _SaveComponents(void);
|
||
|
|
void _SaveWallpaper(void);
|
||
|
|
void _SavePattern(DWORD dwFlags);
|
||
|
|
void _SaveSettings(DWORD dwFlags);
|
||
|
|
|
||
|
|
void _GenerateHtmlHeader(void);
|
||
|
|
void _GenerateHtmlPicture(COMPONENTA *pcomp);
|
||
|
|
void _GenerateHtmlDoc(COMPONENTA *pcomp);
|
||
|
|
void _GenerateHtmlSite(COMPONENTA *pcomp);
|
||
|
|
void _GenerateHtmlControl(COMPONENTA *pcomp);
|
||
|
|
void _GenerateHtmlComponent(COMPONENTA *pcomp);
|
||
|
|
void _GenerateHtmlFooter(void);
|
||
|
|
void _GenerateHtml(void);
|
||
|
|
|
||
|
|
void _WriteHtmlFromString(LPCTSTR psz);
|
||
|
|
void _WriteHtmlFromId(UINT uid);
|
||
|
|
void _WriteHtmlFromIdF(UINT uid, ...);
|
||
|
|
void _WriteHtmlFromFile(LPCTSTR pszContents);
|
||
|
|
void _WriteHtmlFromReadFileObj(CReadFileObj *pReadFileObj, int iOffsetStart, int iOffsetEnd);
|
||
|
|
void _WriteResizeable(COMPONENTA *pcomp);
|
||
|
|
void _WriteHtmlW(LPCWSTR pwstr, UINT cchToWrite, UINT *pcchWritten);
|
||
|
|
|
||
|
|
int _ScanTagEntriesA(CReadFileObj *pReadFileObj, int iOffsetStart, TAGENTRYA *pte, int cte);
|
||
|
|
int _ScanTagEntriesW(CReadFileObj *pReadFileObj, int iOffsetStart, TAGENTRYW *pte, int cte);
|
||
|
|
int _ScanForTagA(CReadFileObj *pReadFileObj, int iOffsetStart, LPCSTR pwszTag);
|
||
|
|
int _ScanForTagW(CReadFileObj *pReadFileObj, int iOffsetStart, LPCWSTR pwszTag);
|
||
|
|
void _ParseAnsiInputHtmlFile(LPTSTR szSelectedWallpaper, int *piOffsetBase, int *piOffsetComp);
|
||
|
|
HRESULT _CopyComponent(COMPONENTA *pCompDest, COMPONENTA *pCompSrc, DWORD dwFlags);
|
||
|
|
HRESULT _ConvertFileToTempBitmap(IN LPWSTR pszFile, IN int cchSize);
|
||
|
|
HRESULT _SaveTempWallpaperSettings(void);
|
||
|
|
|
||
|
|
private:
|
||
|
|
HRESULT _AddDTIWithUIPrivateA(HWND hwnd, LPCCOMPONENT pComp, DWORD dwFlags);
|
||
|
|
BOOL _IsDisplayInSafeMode(void);
|
||
|
|
|
||
|
|
// The following are used for manipulating ActiveDesktop using Dynamic HTML.
|
||
|
|
HRESULT _InsertNewDesktopItems(IHTMLDocument2 *pDoc);
|
||
|
|
HRESULT _UpdateDesktopItemHtmlElements(IHTMLDocument2 *pDoc);
|
||
|
|
HRESULT _UpdateHtmlElement(IHTMLElement *pElem);
|
||
|
|
BOOL _UpdateIdOfElement(IHTMLElement *pElem, LPCOMPONENTA lpCompA);
|
||
|
|
void _UpdateStyleOfElement(IHTMLElement *pElem, LPCOMPONENTA lpCompA);
|
||
|
|
void _GenerateHtmlBStrForComp(COMPONENTA *pComp, BSTR *pbstr);
|
||
|
|
};
|
||
|
|
|
||
|
|
extern IActiveDesktop *g_pActiveDesk;
|
||
|
|
int GetIntFromSubkey(HKEY hKey, LPCTSTR lpszValueName, int iDefault);
|
||
|
|
int GetIntFromReg(HKEY hKey, LPCTSTR lpszSubkey, LPCTSTR lpszNameValue, int iDefault);
|
||
|
|
BOOL GetStringFromReg(HKEY hkey, LPCTSTR lpszSubkey, LPCTSTR lpszValueName, LPCTSTR lpszDefault, LPTSTR lpszValue, DWORD cchSizeofValueBuff);
|
||
|
|
STDAPI CActiveDesktop_InternalCreateInstance(LPUNKNOWN * ppunk, REFIID riid);
|
||
|
|
HRESULT GetPerUserFileName(LPTSTR pszOutputFileName, DWORD dwSize, LPTSTR pszPartialFileName);
|
||
|
|
STDAPI CDeskHtmlProp_RegUnReg(BOOL bReg);
|
||
|
|
|
||
|
|
//Function to convert components in either direction.
|
||
|
|
void ConvertCompStruct(COMPONENTA *pCompDest, COMPONENTA *pCompSrc, BOOL fPubToPriv);
|
||
|
|
void SetStateInfo(COMPSTATEINFO *pCompStateInfo, COMPPOS *pCompPos, DWORD dwItemState);
|
||
|
|
|
||
|
|
#define MultiCompToWideComp(MultiComp, WideComp) ConvertCompStruct((COMPONENTA *)WideComp, MultiComp, FALSE)
|
||
|
|
#define WideCompToMultiComp(WideComp, MultiComp) ConvertCompStruct(MultiComp, (COMPONENTA *)WideComp, TRUE)
|
||
|
|
|
||
|
|
#define COMPONENT_TOP_WINDOWLESS (COMPONENT_TOP / 2)
|
||
|
|
#define IsWindowLessComponent(pcomp) (((pcomp)->iComponentType == COMP_TYPE_PICTURE) || ((pcomp)->iComponentType == COMP_TYPE_HTMLDOC))
|
||
|
|
|
||
|
|
#define COMPONENT_DEFAULT_WIDTH ((DWORD)-1)
|
||
|
|
#define COMPONENT_DEFAULT_HEIGHT ((DWORD)-1)
|
||
|
|
|
||
|
|
#define DESKMOVR_FILENAME TEXT("\\Web\\DeskMovr.htt")
|
||
|
|
#define DESKTOPHTML_FILENAME TEXT("\\Microsoft\\Internet Explorer\\Desktop.htt")
|
||
|
|
#define PATTERN_FILENAME TEXT("\\Microsoft\\Internet Explorer\\Pattern.bmp")
|
||
|
|
|
||
|
|
#define SAVE_PATTERN_NAME 0x00000001
|
||
|
|
#define GENERATE_PATTERN_FILE 0x00000002
|
||
|
|
|
||
|
|
//Property names
|
||
|
|
#define c_wszPropName_TSPerfBGPolicy L"TSConnectEvent"
|
||
|
|
#define c_wszPropName_IgnorePolicies L"IgnorePolicies"
|
||
|
|
|
||
|
|
#define SZ_REGKEY_ACTDESKTOP_TSPERFPOLICY TEXT("Software\\Microsoft\\Windows\\CurrentVersion\\Explorer\\Remote")
|
||
|
|
#define SZ_REGVALUE_ACTDESKTOP_TSPERFPOLICY TEXT("ActiveDesktop")
|
||
|
|
#define SZ_REGVDATE_ACTDESKTOP_TSPERFPOLICY TEXT("Force Blank")
|
||
|
|
|
||
|
|
|
||
|
|
#define REGSTR_PATH_WP_POLICY REGSTR_PATH_POLICIES TEXT("\\") REGSTR_KEY_SYSTEM
|
||
|
|
|
||
|
|
BOOL ReadPolicyForWallpaper(LPTSTR lpszPolicyForWallpaper, DWORD dwSizeofBuff);
|
||
|
|
BOOL ReadPolicyForWPStyle(LPDWORD lpdwStyle);
|
||
|
|
|
||
|
|
#endif // _DESKSTAT_H_
|