misc.cpp File Reference

#include <stdio.h>
#include <tchar.h>
#include <windef.h>
#include <winbase.h>
#include <winreg.h>
#include <wingdi.h>
#include <winnls.h>
#include <wincon.h>
#include <shellapi.h>
#include <shlobj.h>
#include <shlobj_undoc.h>
#include <shlwapi.h>
#include <shlguid_undoc.h>
#include <uxtheme.h>
#include <strsafe.h>
#include <atlbase.h>
#include <atlcom.h>
#include <atlwin.h>
#include <undocshell.h>
#include <undocuser.h>
#include <shellutils.h>
#include <CQuickLaunchBand.h>
#include <wine/debug.h>

Include dependency graph for misc.cpp:

Go to the source code of this file.

Classes
class	CRShellModule
class	CRShellClassFactory

Macros
#define	WIN32_NO_STATUS
#define	_INC_WINDOWS
#define	COM_NO_WINDOWS_H
#define	COBJMACROS

Functions
HRESULT WINAPI	RSHELL_CStartMenu_CreateInstance (REFIID riid, void **ppv)
HRESULT WINAPI	RSHELL_CMenuDeskBar_CreateInstance (REFIID riid, LPVOID *ppv)
HRESULT WINAPI	RSHELL_CMenuSite_CreateInstance (REFIID riid, LPVOID *ppv)
HRESULT WINAPI	RSHELL_CMenuBand_CreateInstance (REFIID riid, LPVOID *ppv)
HRESULT WINAPI	RSHELL_CMergedFolder_CreateInstance (REFIID riid, LPVOID *ppv)
static LSTATUS	_RegSetStringValueW (HKEY hKey, LPCWSTR lpValueName, LPCWSTR lpStringData)
static HRESULT	RegisterComponent (REFGUID clsid, LPCWSTR szDisplayName)
static HRESULT	UnregisterComponent (REFGUID clsid)
	STDAPI_ (BOOL) DllMain(HINSTANCE hInstance
HRESULT WINAPI	DllCanUnloadNow (void)
STDAPI	DllRegisterServer (void)
STDAPI	DllUnregisterServer (void)
STDAPI	DllGetClassObject (REFCLSID rclsid, REFIID riid, LPVOID *ppv)
VOID WINAPI	ShowFolderOptionsDialog (UINT Page, BOOL Async)

Variables
HINSTANCE	shell32_hInstance = NULL
CRShellModule	gModule
CAtlWinModule	gWinModule
DWORD	dwReason
DWORD LPVOID	fImpLoad

Macro Definition Documentation

_INC_WINDOWS

#define _INC_WINDOWS

Definition at line 13 of file misc.cpp.

COBJMACROS

#define COBJMACROS

Definition at line 16 of file misc.cpp.

COM_NO_WINDOWS_H

#define COM_NO_WINDOWS_H

Definition at line 14 of file misc.cpp.

WIN32_NO_STATUS

#define WIN32_NO_STATUS

Definition at line 12 of file misc.cpp.

Function Documentation

_RegSetStringValueW()

static LSTATUS _RegSetStringValueW ( HKEY  hKey, LPCWSTR  lpValueName, LPCWSTR  lpStringData  )

inlinestatic

Definition at line 65 of file misc.cpp.

{
    DWORD dwStringDataLen = lstrlenW(lpStringData);
 
    return RegSetValueExW(hKey, lpValueName, 0, REG_SZ, (BYTE *) lpStringData, 2 * (dwStringDataLen + 1));
}

Referenced by RegisterComponent().

DllCanUnloadNow()

HRESULT WINAPI DllCanUnloadNow

(

void

)

Definition at line 154 of file misc.cpp.

{
    gModule.DllCanUnloadNow();
    return S_FALSE;
}

Referenced by COMPOBJ_DllList_Add().

DllGetClassObject()

STDAPI DllGetClassObject	(	REFCLSID	rclsid,
		REFIID	riid,
		LPVOID *	ppv
	)

Definition at line 246 of file misc.cpp.

{
    if (!ppv)
        return E_INVALIDARG;
 
    *ppv = NULL;
 
    return ShellObjectCreatorInit<CRShellClassFactory>(rclsid, riid, ppv);
}

DllRegisterServer()

STDAPI DllRegisterServer

(

void

)

Definition at line 161 of file misc.cpp.

{
#if 0
    RegisterComponent(CLSID_StartMenu, L"Shell Start Menu");
    RegisterComponent(CLSID_MenuDeskBar, L"Shell Menu Desk Bar");
    RegisterComponent(CLSID_MenuBand, L"Shell Menu Band");
    RegisterComponent(CLSID_MenuBandSite, L"Shell Menu Band Site");
    RegisterComponent(CLSID_MergedFolder, L"Merged Shell Folder");
    RegisterComponent(CLSID_RebarBandSite, L"Shell Rebar Band Site");
#endif
    RegisterComponent(CLSID_QuickLaunchBand, L"Quick Launch");
    RegisterComCat();
    return S_OK;
}

DllUnregisterServer()

STDAPI DllUnregisterServer

(

void

)

Definition at line 177 of file misc.cpp.

{
#if 0
    UnregisterComponent(CLSID_StartMenu);
    UnregisterComponent(CLSID_MenuDeskBar);
    UnregisterComponent(CLSID_MenuBand);
    UnregisterComponent(CLSID_MenuBandSite);
    UnregisterComponent(CLSID_MergedFolder);
    UnregisterComponent(CLSID_RebarBandSite);
#endif
    UnregisterComponent(CLSID_QuickLaunchBand);
    UnregisterComCat();
    return S_OK;
}

RegisterComponent()

static HRESULT RegisterComponent ( REFGUID  clsid, LPCWSTR  szDisplayName  )

static

Definition at line 72 of file misc.cpp.

{
    WCHAR szFilename[MAX_PATH];
    WCHAR szClsid[MAX_PATH];
    WCHAR szRoot[MAX_PATH];
 
    if (!StringFromGUID2(clsid, szClsid, _countof(szClsid)))
        return E_FAIL;
 
    if (!GetModuleFileNameW(shell32_hInstance, szFilename, _countof(szFilename)))
        return E_FAIL;
 
    HRESULT hr = StringCchPrintfW(szRoot, 0x104u, L"CLSID\\%s", szClsid);
    if (FAILED(hr))
        return hr;
 
    DWORD dwDisposition;
    HKEY hkRoot;
    if (RegCreateKeyExW(HKEY_CLASSES_ROOT, szRoot, 0, NULL, 0, KEY_WRITE, 0, &hkRoot, &dwDisposition) != 0)
        return E_FAIL;
 
    HKEY hkServer;
 
    _RegSetStringValueW(hkRoot, NULL, szDisplayName);
 
    if (RegCreateKeyExW(hkRoot, L"InprocServer32", 0, NULL, 0, KEY_SET_VALUE, 0, &hkServer, &dwDisposition) != 0)
    {
        RegCloseKey(hkRoot);
        return E_FAIL;
    }
 
    _RegSetStringValueW(hkServer, NULL, szFilename);
    _RegSetStringValueW(hkServer, L"ThreadingModel", L"Both");
 
    RegCloseKey(hkServer);
    RegCloseKey(hkRoot);
    return S_OK;
}

Referenced by DllRegisterServer().

RSHELL_CMenuBand_CreateInstance()

HRESULT WINAPI RSHELL_CMenuBand_CreateInstance	(	REFIID	riid,
		LPVOID *	ppv
	)

Definition at line 1321 of file CMenuBand.cpp.

{
    return ShellObjectCreator<CMenuBand>(riid, ppv);
}

Referenced by CRShellClassFactory::CreateInstance().

RSHELL_CMenuDeskBar_CreateInstance()

HRESULT WINAPI RSHELL_CMenuDeskBar_CreateInstance	(	REFIID	riid,
		LPVOID *	ppv
	)

Definition at line 855 of file CMenuDeskBar.cpp.

{
    return ShellObjectCreator<CMenuDeskBar>(riid, ppv);
}

Referenced by CRShellClassFactory::CreateInstance().

RSHELL_CMenuSite_CreateInstance()

HRESULT WINAPI RSHELL_CMenuSite_CreateInstance	(	REFIID	riid,
		LPVOID *	ppv
	)

Definition at line 363 of file CMenuSite.cpp.

{
    return ShellObjectCreator<CMenuSite>(riid, ppv);
}

Referenced by CRShellClassFactory::CreateInstance().

RSHELL_CMergedFolder_CreateInstance()

HRESULT WINAPI RSHELL_CMergedFolder_CreateInstance	(	REFIID	riid,
		LPVOID *	ppv
	)

Definition at line 813 of file CMergedFolder.cpp.

{
    return ShellObjectCreator<CMergedFolder>(riid, ppv);
}

Referenced by CRShellClassFactory::CreateInstance().

RSHELL_CStartMenu_CreateInstance()

HRESULT WINAPI RSHELL_CStartMenu_CreateInstance	(	REFIID	riid,
		void **	ppv
	)

Definition at line 592 of file CStartMenu.cpp.

{
    CComPtr<IShellMenu> pShellMenu;
    CComPtr<IBandSite> pBandSite;
    CComPtr<IDeskBar> pDeskBar;
 
    HRESULT hr;
    IShellFolder * psf;
 
    LPITEMIDLIST pidlProgramsAbsolute;
    LPITEMIDLIST pidlPrograms;
    CComPtr<IShellFolder> psfPrograms;
 
    hr = CMenuBand_CreateInstance(IID_PPV_ARG(IShellMenu, &pShellMenu));
    if (FAILED_UNEXPECTEDLY(hr))
        return hr;
 
    hr = CMenuSite_CreateInstance(IID_PPV_ARG(IBandSite, &pBandSite));
    if (FAILED_UNEXPECTEDLY(hr))
        return hr;
 
    hr = CMenuDeskBar_CreateInstance(IID_PPV_ARG(IDeskBar, &pDeskBar));
    if (FAILED_UNEXPECTEDLY(hr))
        return hr;
 
    CComObject<CShellMenuCallback> *pCallback;
    hr = CComObject<CShellMenuCallback>::CreateInstance(&pCallback);
    if (FAILED_UNEXPECTEDLY(hr))
        return hr;
 
    pCallback->AddRef(); // CreateInstance returns object with 0 ref count */
    pCallback->Initialize(pShellMenu, pBandSite, pDeskBar);
 
    hr = pShellMenu->Initialize(pCallback, (UINT) -1, 0, SMINIT_TOPLEVEL | SMINIT_VERTICAL);
    if (FAILED_UNEXPECTEDLY(hr))
        return hr;
 
    hr = GetStartMenuFolder(&psf);
    if (FAILED_UNEXPECTEDLY(hr))
        return hr;
 
    /* psf is a merged folder, so now we want to get the pidl of the programs item from the merged folder */
    {
        hr = SHGetSpecialFolderLocation(NULL, CSIDL_PROGRAMS, &pidlProgramsAbsolute);
        if (FAILED_UNEXPECTEDLY(hr))
        {
            WARN("USER Programs folder not found\n");
            hr = SHGetSpecialFolderLocation(NULL, CSIDL_COMMON_PROGRAMS, &pidlProgramsAbsolute);
            if (FAILED_UNEXPECTEDLY(hr))
                return hr;
        }
 
        LPCITEMIDLIST pcidlPrograms;
        CComPtr<IShellFolder> psfParent;
        STRRET str;
        TCHAR szDisplayName[MAX_PATH];
 
        hr = SHBindToParent(pidlProgramsAbsolute, IID_PPV_ARG(IShellFolder, &psfParent), &pcidlPrograms);
        if (FAILED_UNEXPECTEDLY(hr))
            return hr;
 
        hr = psfParent->GetDisplayNameOf(pcidlPrograms, SHGDN_FORPARSING | SHGDN_INFOLDER, &str);
        if (FAILED_UNEXPECTEDLY(hr))
            return hr;
 
        StrRetToBuf(&str, pcidlPrograms, szDisplayName, _countof(szDisplayName));
        ILFree(pidlProgramsAbsolute);
 
        /* We got the display name from the fs folder and we parse it with the merged folder here */
        hr = psf->ParseDisplayName(NULL, NULL, szDisplayName, NULL, &pidlPrograms, NULL);
        if (FAILED_UNEXPECTEDLY(hr))
            return hr;
    }
 
    hr = GetProgramsFolder(&psfPrograms);
    if (FAILED_UNEXPECTEDLY(hr))
        return hr;
 
    hr = pCallback->_SetProgramsFolder(psfPrograms, pidlPrograms);
    if (FAILED_UNEXPECTEDLY(hr))
        return hr;
 
    hr = pShellMenu->SetShellFolder(psf, NULL, NULL, SMSET_TOP);
    if (FAILED_UNEXPECTEDLY(hr))
        return hr;
 
    hr = pDeskBar->SetClient(pBandSite);
    if (FAILED_UNEXPECTEDLY(hr))
        return hr;
 
    hr = pBandSite->AddBand(pShellMenu);
    if (FAILED_UNEXPECTEDLY(hr))
        return hr;
 
    return pDeskBar->QueryInterface(riid, ppv);
}

Referenced by CRShellClassFactory::CreateInstance(), and CStartMenuDummy::_CreatorClass::CreateInstance().

ShowFolderOptionsDialog()

VOID WINAPI ShowFolderOptionsDialog	(	UINT	Page,
		BOOL	Async
	)

Definition at line 259 of file misc.cpp.

{
    char buf[MAX_PATH];
    wsprintfA(buf, "rundll32.exe shell32.dll,Options_RunDLL %u", Page);
    WinExec(buf, SW_SHOW);
}

STDAPI_()

STDAPI_

(

BOOL

)

UnregisterComponent()

static HRESULT UnregisterComponent ( REFGUID  clsid )

static

Definition at line 111 of file misc.cpp.

{
    WCHAR szClsid[MAX_PATH];
    WCHAR szRoot[MAX_PATH];
    HKEY hkRoot;
 
    if (!StringFromGUID2(clsid, szClsid, _countof(szClsid)))
        return E_FAIL;
 
    HRESULT hr = StringCchPrintfW(szRoot, 0x104u, L"CLSID\\%s", szClsid);
    if (FAILED(hr))
        return hr;
 
    if (RegOpenKeyExW(HKEY_CLASSES_ROOT, szRoot, 0, KEY_WRITE, &hkRoot) != 0)
        return E_FAIL;
 
    RegDeleteKeyW(hkRoot, L"InprocServer32");
    RegCloseKey(hkRoot);
 
    RegDeleteKeyW(HKEY_CLASSES_ROOT, szRoot);
 
    return S_OK;
}

Referenced by DllUnregisterServer().

Variable Documentation

dwReason

static IN DWORD dwReason

Definition at line 135 of file misc.cpp.

Referenced by __dyn_tls_dtor(), BaseCheckAppcompatCache(), BaseInitiateSystemShutdownEx(), DllMain(), GdiDllInitialize(), Imm32CheckAndApplyAppCompat(), ImmDllInitialize(), InitializeFmIfs(), InitiateShutdownA(), InitiateSystemShutdownExA(), InitiateSystemShutdownExW(), LpkDllInitialize(), msvcrt_init_exception(), notify_DllMain(), notify_TlsCallback(), NotifyShims(), StartSystemShutdown(), test_one(), TlsCallback(), TOOLBAR_SetHotItemEx(), TOOLBAR_SetRelativeHotItem(), and UpdateNTPStatus().

fImpLoad

DWORD LPVOID fImpLoad

Initial value:

{
    if (dwReason == DLL_PROCESS_ATTACH)
    {
        shell32_hInstance = hInstance;
 
        gModule.Init(NULL, hInstance, NULL);
        DisableThreadLibraryCalls(hInstance);
    }
    else if (dwReason == DLL_PROCESS_DETACH)
    {
        shell32_hInstance = NULL;
        gModule.Term();
    }
    return TRUE

Definition at line 135 of file misc.cpp.

Referenced by DllMain().

gModule

CRShellModule gModule

Definition at line 60 of file misc.cpp.

Referenced by DllCanUnloadNow().

gWinModule

CAtlWinModule gWinModule

Definition at line 61 of file misc.cpp.

shell32_hInstance

HINSTANCE shell32_hInstance = NULL

Definition at line 63 of file misc.cpp.

Referenced by RegisterComponent().