util.cpp File Reference

#include "precomp.h"
#include <winver.h>

Include dependency graph for util.cpp:

Go to the source code of this file.

Classes
struct	_LANGCODEPAGE

Macros
#define	REGKEY_ADVANCED   L"Software\\Microsoft\\Windows\\CurrentVersion\\Explorer\\Advanced"

Typedefs
typedef struct _LANGCODEPAGE	LANGCODEPAGE
typedef struct _LANGCODEPAGE *	PLANGCODEPAGE

Functions
HRESULT	IsSameObject (IN IUnknown *punk1, IN IUnknown *punk2)
HMENU	LoadPopupMenu (IN HINSTANCE hInstance, IN LPCWSTR lpMenuName)
HMENU	FindSubMenu (IN HMENU hMenu, IN UINT uItem, IN BOOL fByPosition)
BOOL	GetCurrentLoggedOnUserName (OUT LPWSTR szBuffer, IN DWORD dwBufferSize)
BOOL	FormatMenuString (IN HMENU hMenu, IN UINT uPosition, IN UINT uFlags,...)
BOOL	GetRegBool (IN LPCWSTR pszSubKey, IN LPCWSTR pszValueName, IN BOOL bDefaultValue)
BOOL	SetRegDword (IN LPCWSTR pszSubKey, IN LPCWSTR pszValueName, IN DWORD dwValue)
BOOL	GetAdvancedBool (IN LPCWSTR pszValueName, IN BOOL bDefaultValue)
BOOL	SetAdvancedDword (IN LPCWSTR pszValueName, IN DWORD dwValue)
BOOL	GetVersionInfoString (IN LPCWSTR szFileName, IN LPCWSTR szVersionInfo, OUT LPWSTR szBuffer, IN UINT cbBufLen)

Macro Definition Documentation

REGKEY_ADVANCED

#define REGKEY_ADVANCED   L"Software\\Microsoft\\Windows\\CurrentVersion\\Explorer\\Advanced"

Definition at line 154 of file util.cpp.

Typedef Documentation

LANGCODEPAGE

typedef struct _LANGCODEPAGE LANGCODEPAGE

PLANGCODEPAGE

typedef struct _LANGCODEPAGE * PLANGCODEPAGE

Function Documentation

FindSubMenu()

HMENU FindSubMenu	(	IN HMENU	hMenu,
		IN UINT	uItem,
		IN BOOL	fByPosition
	)

Definition at line 55 of file util.cpp.

{
    MENUITEMINFOW mii;
 
    mii.cbSize = sizeof(mii);
    mii.fMask = MIIM_SUBMENU;
 
    if (GetMenuItemInfoW(hMenu, uItem, fByPosition, &mii))
    {
        return mii.hSubMenu;
    }
 
    return NULL;
}

Referenced by CStartMenuSite::AppendMenu().

FormatMenuString()

BOOL FormatMenuString	(	IN HMENU	hMenu,
		IN UINT	uPosition,
		IN UINT	uFlags,
			...
	)

Definition at line 106 of file util.cpp.

{
    va_list vl;
    MENUITEMINFOW mii;
    WCHAR szBuf[128];
    WCHAR szBufFmt[128];
 
    /* Find the menu item and read the formatting string */
    mii.cbSize = sizeof(mii);
    mii.fMask = MIIM_STRING;
    mii.dwTypeData = szBufFmt;
    mii.cch = _countof(szBufFmt);
    if (GetMenuItemInfoW(hMenu, uPosition, uFlags, &mii))
    {
        /* Format the string */
        va_start(vl, uFlags);
        _vsntprintf(szBuf,
                    _countof(szBuf) - 1,
                    szBufFmt,
                    vl);
        va_end(vl);
        szBuf[_countof(szBuf) - 1] = L'\0';
 
        /* Update the menu item */
        mii.dwTypeData = szBuf;
        if (SetMenuItemInfo(hMenu, uPosition, uFlags, &mii))
        {
            return TRUE;
        }
    }
 
    return FALSE;
}

Referenced by CStartMenuSite::AppendMenu().

GetAdvancedBool()

BOOL GetAdvancedBool	(	IN LPCWSTR	pszValueName,
		IN BOOL	bDefaultValue
	)

Definition at line 156 of file util.cpp.

{
    return GetRegBool(REGKEY_ADVANCED, pszValueName, bDefaultValue);
}

Referenced by CStartMenuSite::AppendMenu(), and CustomGetAdvanced().

GetCurrentLoggedOnUserName()

BOOL GetCurrentLoggedOnUserName	(	OUT LPWSTR	szBuffer,
		IN DWORD	dwBufferSize
	)

Definition at line 73 of file util.cpp.

{
    DWORD dwType;
    DWORD dwSize;
 
    /* Query the user name from the registry */
    dwSize = (dwBufferSize * sizeof(WCHAR)) - 1;
    if (RegQueryValueExW(hkExplorer,
                         L"Logon User Name",
                         0,
                         &dwType,
                         (LPBYTE)szBuffer,
                         &dwSize) == ERROR_SUCCESS &&
        (dwSize / sizeof(WCHAR)) > 1 &&
        szBuffer[0] != L'\0')
    {
        szBuffer[dwSize / sizeof(WCHAR)] = L'\0';
        return TRUE;
    }
 
    /* Fall back to GetUserName() */
    dwSize = dwBufferSize;
    if (!GetUserNameW(szBuffer, &dwSize))
    {
        szBuffer[0] = L'\0';
        return FALSE;
    }
 
    return TRUE;
}

Referenced by CStartMenuSite::AppendMenu().

GetRegBool()

BOOL GetRegBool	(	IN LPCWSTR	pszSubKey,
		IN LPCWSTR	pszValueName,
		IN BOOL	bDefaultValue
	)

Definition at line 143 of file util.cpp.

{
    return SHRegGetBoolUSValueW(pszSubKey, pszValueName, FALSE, bDefaultValue);
}

Referenced by GetAdvancedBool().

GetVersionInfoString()

BOOL GetVersionInfoString	(	IN LPCWSTR	szFileName,
		IN LPCWSTR	szVersionInfo,
		OUT LPWSTR	szBuffer,
		IN UINT	cbBufLen
	)

Definition at line 167 of file util.cpp.

{
    LPVOID lpData = NULL;
    WCHAR szSubBlock[128];
    WCHAR *lpszLocalBuf = NULL;
    LANGID UserLangId;
    PLANGCODEPAGE lpTranslate = NULL;
    DWORD dwLen;
    DWORD dwHandle;
    UINT cbTranslate;
    UINT cbLen;
    BOOL bRet = FALSE;
    UINT i;
 
    dwLen = GetFileVersionInfoSizeW(szFileName, &dwHandle);
 
    if (dwLen > 0)
    {
        lpData = HeapAlloc(hProcessHeap, 0, dwLen);
 
        if (lpData != NULL)
        {
            if (GetFileVersionInfoW(szFileName,
                                    0,
                                    dwLen,
                                    lpData) != 0)
            {
                UserLangId = GetUserDefaultLangID();
 
                VerQueryValueW(lpData,
                               L"\\VarFileInfo\\Translation",
                               (LPVOID*)&lpTranslate,
                               &cbTranslate);
 
                for (i = 0; i < cbTranslate / sizeof(LANGCODEPAGE); i++)
                {
                    /* If the bottom eight bits of the language id's
                    match, use this version information (since this
                    means that the version information and the users
                    default language are the same). */
                    if (LOBYTE(lpTranslate[i].wLanguage) == LOBYTE(UserLangId))
                    {
                        wnsprintf(szSubBlock,
                                  _countof(szSubBlock),
                                  L"\\StringFileInfo\\%04X%04X\\%s",
                                  lpTranslate[i].wLanguage,
                                  lpTranslate[i].wCodePage,
                                  szVersionInfo);
 
                        if (VerQueryValueW(lpData,
                                           szSubBlock,
                                           (LPVOID*)&lpszLocalBuf,
                                           &cbLen) != 0)
                        {
                            wcsncpy(szBuffer, lpszLocalBuf, cbBufLen / sizeof(*szBuffer));
 
                            bRet = TRUE;
                            break;
                        }
                    }
                }
            }
 
            HeapFree(hProcessHeap, 0, lpData);
            lpData = NULL;
        }
    }
 
    return bRet;
}

IsSameObject()

HRESULT IsSameObject	(	IN IUnknown *	punk1,
		IN IUnknown *	punk2
	)

Definition at line 11 of file util.cpp.

{
    HRESULT hRet;
 
    hRet = punk1->QueryInterface(IID_PPV_ARG(IUnknown, &punk1));
    if (!SUCCEEDED(hRet))
        return hRet;
 
    hRet = punk2->QueryInterface(IID_PPV_ARG(IUnknown, &punk2));
 
    punk1->Release();
 
    if (!SUCCEEDED(hRet))
        return hRet;
 
    punk2->Release();
 
    /* We're dealing with the same object if the IUnknown pointers are equal */
    return (punk1 == punk2) ? S_OK : S_FALSE;
}

Referenced by CTrayBandSite::IsTaskBand(), and CTrayBandSite::QueryBand().

LoadPopupMenu()

HMENU LoadPopupMenu	(	IN HINSTANCE	hInstance,
		IN LPCWSTR	lpMenuName
	)

Definition at line 33 of file util.cpp.

{
    HMENU hMenu, hSubMenu = NULL;
 
    hMenu = LoadMenuW(hInstance, lpMenuName);
    if (hMenu != NULL)
    {
        hSubMenu = GetSubMenu(hMenu, 0);
        if ((hSubMenu != NULL) &&
            !RemoveMenu(hMenu, 0, MF_BYPOSITION))
        {
            hSubMenu = NULL;
        }
 
        DestroyMenu(hMenu);
    }
 
    return hSubMenu;
}

Referenced by CStartMenuSite::AppendMenu(), CTrayWindowCtxMenu::QueryContextMenu(), and SettingsPageProc().

SetAdvancedDword()

BOOL SetAdvancedDword	(	IN LPCWSTR	pszValueName,
		IN DWORD	dwValue
	)

Definition at line 161 of file util.cpp.

{
    return SetRegDword(REGKEY_ADVANCED, pszValueName, dwValue);
}

Referenced by CustomSetAdvanced().

SetRegDword()

BOOL SetRegDword	(	IN LPCWSTR	pszSubKey,
		IN LPCWSTR	pszValueName,
		IN DWORD	dwValue
	)

Definition at line 148 of file util.cpp.

{
    return (SHRegSetUSValueW(pszSubKey, pszValueName, REG_DWORD, &dwValue,
                             sizeof(dwValue), SHREGSET_FORCE_HKCU) == ERROR_SUCCESS);
}

Referenced by SetAdvancedDword().