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

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	GetExplorerRegValueSet (IN HKEY hKey, IN LPCWSTR lpSubKey, IN LPCWSTR lpValue)
BOOL	GetVersionInfoString (IN LPCWSTR szFileName, IN LPCWSTR szVersionInfo, OUT LPWSTR szBuffer, IN UINT cbBufLen)

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().

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().

GetExplorerRegValueSet()

BOOL GetExplorerRegValueSet	(	IN HKEY	hKey,
		IN LPCWSTR	lpSubKey,
		IN LPCWSTR	lpValue
	)

Definition at line 144 of file util.cpp.

{
    WCHAR szBuffer[MAX_PATH];
    HKEY hkSubKey;
    DWORD dwType, dwSize;
    BOOL Ret = FALSE;

    StringCbCopyW(szBuffer, sizeof(szBuffer),
                  L"Software\\Microsoft\\Windows\\CurrentVersion\\Explorer");
    if (FAILED_UNEXPECTEDLY(StringCbCatW(szBuffer, sizeof(szBuffer), L"\\")))
        return FALSE;
    if (FAILED_UNEXPECTEDLY(StringCbCatW(szBuffer, sizeof(szBuffer), lpSubKey)))
        return FALSE;

    dwSize = sizeof(szBuffer);
    if (RegOpenKeyExW(hKey,
                      szBuffer,
                      0,
                      KEY_QUERY_VALUE,
                      &hkSubKey) == ERROR_SUCCESS)
    {
        ZeroMemory(szBuffer, sizeof(szBuffer));

        if (RegQueryValueExW(hkSubKey,
                             lpValue,
                             0,
                             &dwType,
                             (LPBYTE)szBuffer,
                             &dwSize) == ERROR_SUCCESS)
        {
            if ((dwType == REG_DWORD) && (dwSize == sizeof(DWORD)))
                Ret = *((PDWORD)szBuffer) != 0;
            else if (dwSize > 0)
                Ret = *((PWCHAR)szBuffer) != 0;
        }

        RegCloseKey(hkSubKey);
    }
    return Ret;
}

Referenced by CStartMenuSite::AppendMenu().

GetVersionInfoString()

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

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