util.c File Reference

#include "shimgvw.h"
#include <windowsx.h>
#include <shlobj.h>
#include <shellapi.h>
#include <shellutils.h>
#include <shlwapi_undoc.h>

Include dependency graph for util.c:

Go to the source code of this file.

Classes
struct	_ENABLECOMMANDDATA

Typedefs
typedef struct _ENABLECOMMANDDATA	ENABLECOMMANDDATA

Functions
static int	GetMenuItemIdByPos (HMENU hMenu, UINT Pos)
static BOOL	IsMenuSeparator (HMENU hMenu, UINT Pos)
static BOOL	IsSelfShellVerb (PCWSTR Assoc, PCWSTR Verb)
static void	ModifyShellContextMenu (IContextMenu *pCM, HMENU hMenu, UINT CmdIdFirst, PCWSTR Assoc)
static LRESULT CALLBACK	ShellContextMenuWindowProc (HWND hwnd, UINT uMsg, WPARAM wParam, LPARAM lParam)
static void	DoShellContextMenu (HWND hwnd, IContextMenu *pCM, PCWSTR File, LPARAM lParam)
HRESULT	GetUIObjectOfPath (HWND hwnd, PCWSTR File, REFIID riid, void **ppv)
void	DoShellContextMenuOnFile (HWND hwnd, PCWSTR File, LPARAM lParam)
static DWORD CALLBACK	EnableCommandIfVerbExistsProc (LPVOID ThreadParam)
void	EnableCommandIfVerbExists (UINT ImageId, HWND hwnd, UINT CmdId, PCWSTR Verb, PCWSTR File)
void	ShellExecuteVerb (HWND hwnd, PCWSTR Verb, PCWSTR File, BOOL Quit)
UINT	ErrorBox (HWND hwnd, UINT Error)
void	DisplayHelp (HWND hwnd)

Variables
IContextMenu *	g_pContextMenu = NULL

Typedef Documentation

ENABLECOMMANDDATA

typedef struct _ENABLECOMMANDDATA ENABLECOMMANDDATA

Function Documentation

DisplayHelp()

void DisplayHelp

(

HWND

hwnd

)

Definition at line 253 of file util.c.

{
    ErrorBox(hwnd, ERROR_NOT_SUPPORTED);
}

Referenced by Preview_OnCommand().

DoShellContextMenu()

static void DoShellContextMenu ( HWND  hwnd, IContextMenu *  pCM, PCWSTR  File, LPARAM  lParam  )

static

Definition at line 93 of file util.c.

{
    enum { first = 1, last = 0x7fff };
    HRESULT hr;
    HMENU hMenu = CreatePopupMenu();
    UINT cmf = GetKeyState(VK_SHIFT) < 0 ? CMF_EXTENDEDVERBS : 0;
 
    POINT pt = { GET_X_LPARAM(lParam), GET_Y_LPARAM(lParam) };
    if ((int)lParam == -1)
    {
        RECT rect;
        GetWindowRect(hwnd, &rect);
        pt.x = (rect.left + rect.right) / 2;
        pt.y = rect.top;
    }
 
    g_pContextMenu = pCM;
    hwnd = SHCreateWorkerWindowW(ShellContextMenuWindowProc, hwnd, 0, WS_VISIBLE | WS_CHILD, NULL, 0);
    if (!hwnd)
        goto die;
    hr = IContextMenu_QueryContextMenu(pCM, hMenu, 0, first, last, cmf | CMF_NODEFAULT);
    if (SUCCEEDED(hr))
    {
        UINT id;
        ModifyShellContextMenu(pCM, hMenu, first, PathFindExtensionW(File));
        id = TrackPopupMenuEx(hMenu, TPM_RETURNCMD, pt.x, pt.y, hwnd, NULL);
        if (id)
        {
            UINT flags = (GetKeyState(VK_SHIFT) < 0 ? CMIC_MASK_SHIFT_DOWN : 0) |
                         (GetKeyState(VK_CONTROL) < 0 ? CMIC_MASK_CONTROL_DOWN : 0);
            CMINVOKECOMMANDINFO ici = { sizeof(ici), flags, hwnd, MAKEINTRESOURCEA(id - first) };
            ici.nShow = SW_SHOW;
            hr = IContextMenu_InvokeCommand(pCM, &ici);
        }
    }
    DestroyWindow(hwnd);
die:
    DestroyMenu(hMenu);
    g_pContextMenu = NULL;
}

Referenced by DoShellContextMenuOnFile().

DoShellContextMenuOnFile()

void DoShellContextMenuOnFile	(	HWND	hwnd,
		PCWSTR	File,
		LPARAM	lParam
	)

Definition at line 152 of file util.c.

{
    IContextMenu *pCM;
    HRESULT hr = GetUIObjectOfPath(hwnd, File, IID_PPV_ARG(IContextMenu, &pCM));
    if (SUCCEEDED(hr))
    {
        DoShellContextMenu(hwnd, pCM, File, lParam);
        IContextMenu_Release(pCM);
    }
}

Referenced by ZoomWndProc().

EnableCommandIfVerbExists()

void EnableCommandIfVerbExists	(	UINT	ImageId,
		HWND	hwnd,
		UINT	CmdId,
		PCWSTR	Verb,
		PCWSTR	File
	)

Definition at line 209 of file util.c.

{
    const SIZE_T cch = lstrlenW(File) + 1;
    ENABLECOMMANDDATA *pData = SHAlloc(FIELD_OFFSET(ENABLECOMMANDDATA, File[cch]));
    if (pData)
    {
        pData->hwnd = hwnd;
        pData->Verb = Verb; // Note: This assumes the string is valid for the lifetime of the thread.
        pData->CmdId = CmdId;
        pData->ImageId = ImageId;
        CopyMemory(pData->File, File, cch * sizeof(*File));
        SHCreateThread(EnableCommandIfVerbExistsProc, pData, CTF_COINIT | CTF_INSIST, NULL);
    }
}

Referenced by Preview_pLoadImage().

EnableCommandIfVerbExistsProc()

static DWORD CALLBACK EnableCommandIfVerbExistsProc ( LPVOID  ThreadParam )

static

Definition at line 173 of file util.c.

{
    enum { first = 1, last = 0x7fff };
    ENABLECOMMANDDATA *pData = ThreadParam;
    IContextMenu *pCM;
    HRESULT hr = GetUIObjectOfPath(pData->hwnd, pData->File, IID_PPV_ARG(IContextMenu, &pCM));
    if (SUCCEEDED(hr))
    {
        HMENU hMenu = CreatePopupMenu();
        hr = IContextMenu_QueryContextMenu(pCM, hMenu, 0, first, last, CMF_NORMAL);
        if (SUCCEEDED(hr))
        {
            for (UINT i = 0, c = GetMenuItemCount(hMenu); i < c; ++i)
            {
                WCHAR buf[200];
                UINT id = GetMenuItemIdByPos(hMenu, i);
                if (id == (UINT)-1)
                    continue;
 
                *buf = UNICODE_NULL;
                hr = IContextMenu_GetCommandString(pCM, id - first, GCS_VERBW, NULL, (char*)buf, _countof(buf));
                if (SUCCEEDED(hr) && !lstrcmpiW(buf, pData->Verb))
                {
                    PostMessageW(pData->hwnd, WM_UPDATECOMMANDSTATE, MAKELONG(pData->CmdId, TRUE), pData->ImageId);
                    break;
                }
            }
        }
        DestroyMenu(hMenu);
        IContextMenu_Release(pCM);
    }
    SHFree(pData);
    return 0;
}

Referenced by EnableCommandIfVerbExists().

ErrorBox()

UINT ErrorBox	(	HWND	hwnd,
		UINT	Error
	)

Definition at line 247 of file util.c.

{
    return SHELL_ErrorBox(hwnd, Error);
}

Referenced by DisplayHelp().

GetMenuItemIdByPos()

static int GetMenuItemIdByPos ( HMENU  hMenu, UINT  Pos  )

static

Definition at line 18 of file util.c.

{
    MENUITEMINFOW mii;
    mii.cbSize = FIELD_OFFSET(MENUITEMINFOW, hbmpItem); /* USER32 version agnostic */
    mii.fMask = MIIM_ID;
    mii.cch = 0;
    return GetMenuItemInfoW(hMenu, Pos, TRUE, &mii) ? mii.wID : -1;
}

Referenced by EnableCommandIfVerbExistsProc(), and ModifyShellContextMenu().

GetUIObjectOfPath()

HRESULT GetUIObjectOfPath	(	HWND	hwnd,
		PCWSTR	File,
		REFIID	riid,
		void **	ppv
	)

Definition at line 135 of file util.c.

{
    HRESULT hr = HRESULT_FROM_WIN32(ERROR_FILE_NOT_FOUND);
    IShellFolder *pSF;
    PCUITEMID_CHILD pidlItem;
    PIDLIST_ABSOLUTE pidl = ILCreateFromPath(File);
    *ppv = NULL;
    if (pidl && SUCCEEDED(SHBindToParent(pidl, IID_PPV_ARG(IShellFolder, &pSF), &pidlItem)))
    {
        hr = IShellFolder_GetUIObjectOf(pSF, hwnd, 1, &pidlItem, riid, NULL, ppv);
        IShellFolder_Release(pSF);
    }
    SHFree(pidl);
    return hr;
}

Referenced by DoShellContextMenuOnFile(), and EnableCommandIfVerbExistsProc().

IsMenuSeparator()

static BOOL IsMenuSeparator ( HMENU  hMenu, UINT  Pos  )

static

Definition at line 28 of file util.c.

{
    MENUITEMINFOW mii;
    mii.cbSize = FIELD_OFFSET(MENUITEMINFOW, hbmpItem); /* USER32 version agnostic */
    mii.fMask = MIIM_FTYPE;
    mii.cch = 0;
    return GetMenuItemInfoW(hMenu, Pos, TRUE, &mii) && (mii.fType & MFT_SEPARATOR);
}

Referenced by ModifyShellContextMenu().

IsSelfShellVerb()

static BOOL IsSelfShellVerb ( PCWSTR  Assoc, PCWSTR  Verb  )

static

Definition at line 38 of file util.c.

{
    WCHAR buf[MAX_PATH * 3];
    DWORD cch = _countof(buf);
    HRESULT hr = AssocQueryStringW(ASSOCF_NOTRUNCATE, ASSOCSTR_COMMAND, Assoc, Verb, buf, &cch);
    return hr == S_OK && *Assoc == L'.' && StrStrW(buf, L",ImageView_Fullscreen");
}

Referenced by ModifyShellContextMenu().

ModifyShellContextMenu()

static void ModifyShellContextMenu ( IContextMenu *  pCM, HMENU  hMenu, UINT  CmdIdFirst, PCWSTR  Assoc  )

static

Definition at line 47 of file util.c.

{
    HRESULT hr;
    for (UINT i = 0, c = GetMenuItemCount(hMenu); i < c; ++i)
    {
        WCHAR buf[200];
        UINT id = GetMenuItemIdByPos(hMenu, i);
        if (id == (UINT)-1)
            continue;
 
        *buf = UNICODE_NULL;
        /* Note: We just ask for the wide string because all the items we care about come from shell32 and it handles both */
        hr = IContextMenu_GetCommandString(pCM, id - CmdIdFirst, GCS_VERBW, NULL, (char*)buf, _countof(buf));
        if (SUCCEEDED(hr))
        {
            UINT remove = FALSE;
            if (IsSelfShellVerb(Assoc, buf))
                ++remove;
            else if (!lstrcmpiW(L"cut", buf) || !lstrcmpiW(L"paste", buf) || !lstrcmpiW(L"pastelink", buf) ||
                     !lstrcmpiW(L"delete", buf) || !lstrcmpiW(L"link", buf))
                ++remove;
 
            if (remove && DeleteMenu(hMenu, i, MF_BYPOSITION))
            {
                if (i-- > 0)
                {
                    if (IsMenuSeparator(hMenu, i) && IsMenuSeparator(hMenu, i + 1))
                        DeleteMenu(hMenu, i, MF_BYPOSITION);
                }
            }
        }
    }
 
    while (IsMenuSeparator(hMenu, 0) && DeleteMenu(hMenu, 0, MF_BYPOSITION)) {}
}

Referenced by DoShellContextMenu().

ShellContextMenuWindowProc()

static LRESULT CALLBACK ShellContextMenuWindowProc ( HWND  hwnd, UINT  uMsg, WPARAM  wParam, LPARAM  lParam  )

static

Definition at line 84 of file util.c.

{
    LRESULT lRes = 0;
    if (FAILED(SHForwardContextMenuMsg((IUnknown*)g_pContextMenu, uMsg, wParam, lParam, &lRes, TRUE)))
        lRes = DefWindowProc(hwnd, uMsg, wParam, lParam);
    return lRes;
}

Referenced by DoShellContextMenu().

ShellExecuteVerb()

void ShellExecuteVerb	(	HWND	hwnd,
		PCWSTR	Verb,
		PCWSTR	File,
		BOOL	Quit
	)

Definition at line 225 of file util.c.

{
    SHELLEXECUTEINFOW sei = { sizeof(sei), SEE_MASK_INVOKEIDLIST | SEE_MASK_ASYNCOK };
    if (!*File)
        return;
 
    sei.hwnd = hwnd;
    sei.lpVerb = Verb;
    sei.lpFile = File;
    sei.nShow = SW_SHOW;
    if (!ShellExecuteExW(&sei))
    {
        DPRINT1("ShellExecuteExW(%ls, %ls) failed with code %ld\n", Verb, File, GetLastError());
    }
    else if (Quit)
    {
        // Destroy the window to quit the application
        DestroyWindow(hwnd);
    }
}

Referenced by Preview_Edit(), and Preview_pPrintImage().

Variable Documentation

g_pContextMenu

IContextMenu* g_pContextMenu = NULL

Definition at line 15 of file util.c.

Referenced by DoShellContextMenu(), and ShellContextMenuWindowProc().