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)
DWORD WINAPI	WinList_Init (void)
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)

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 26 of file misc.cpp.

COBJMACROS

#define COBJMACROS

Definition at line 29 of file misc.cpp.

COM_NO_WINDOWS_H

#define COM_NO_WINDOWS_H

Definition at line 27 of file misc.cpp.

WIN32_NO_STATUS

#define WIN32_NO_STATUS

Definition at line 25 of file misc.cpp.

Function Documentation

_RegSetStringValueW()

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

inlinestatic

Definition at line 84 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 173 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 265 of file misc.cpp.

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

DllRegisterServer()

STDAPI DllRegisterServer

(

void

)

Definition at line 180 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 196 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 91 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 1313 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 852 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 360 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 490 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().

STDAPI_()

STDAPI_

(

BOOL

)

UnregisterComponent()

static HRESULT UnregisterComponent ( REFGUID  clsid )

static

Definition at line 130 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().

WinList_Init()

DWORD WINAPI WinList_Init

(

void

)

Definition at line 68 of file misc.cpp.

{
    /* do something here (perhaps we may want to add our own implementation fo win8) */
    return 0;
}

Variable Documentation

dwReason

static IN DWORD dwReason

Definition at line 154 of file misc.cpp.

Referenced by __dyn_tls_dtor(), BaseInitiateSystemShutdownEx(), DllMain(), GdiDllInitialize(), ImmDllInitialize(), InitializeFmIfs(), InitiateShutdownA(), InitiateSystemShutdownExA(), InitiateSystemShutdownExW(), LpkDllInitialize(), 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 154 of file misc.cpp.

Referenced by DllMain().

gModule

CRShellModule gModule

Definition at line 79 of file misc.cpp.

Referenced by DllCanUnloadNow().

gWinModule

CAtlWinModule gWinModule

Definition at line 80 of file misc.cpp.

shell32_hInstance

HINSTANCE shell32_hInstance = NULL

Definition at line 82 of file misc.cpp.

Referenced by RegisterComponent().