utility.cpp File Reference

#include "objects.h"
#include <strsafe.h>
#include <wine/debug.h>

Include dependency graph for utility.cpp:

Go to the source code of this file.

Classes
struct	ADDFAVORITEDIALOGDATA

Macros
#define	PATH_VALID_ELEMENT

Functions
	WINE_DEFAULT_DEBUG_CHANNEL (shdocvw)
static INT_PTR	GetMenuItemIdByPos (HMENU hMenu, UINT Pos)
static BOOL	IsMenuSeparator (HMENU hMenu, UINT Pos)
EXTERN_C HRESULT	SHELL_GetIDListFromObject (IUnknown *punk, PIDLIST_ABSOLUTE *ppidl)
static HRESULT	SHELL_CompareAbsoluteIDs (LPARAM lParam, PCIDLIST_ABSOLUTE a, PCIDLIST_ABSOLUTE b)
EXTERN_C BOOL	SHELL_IsEqualAbsoluteID (PCIDLIST_ABSOLUTE a, PCIDLIST_ABSOLUTE b)
EXTERN_C BOOL	SHELL_IsVerb (IContextMenu *pcm, UINT_PTR idCmd, LPCWSTR Verb)
static int	SHELL_FindVerbPos (IContextMenu *pcm, UINT idCmdFirst, HMENU hMenu, LPCWSTR Verb)
EXTERN_C VOID	SHELL_RemoveVerb (IContextMenu *pcm, UINT idCmdFirst, HMENU hMenu, LPCWSTR Verb)
EXTERN_C BOOL	_ILIsDesktop (LPCITEMIDLIST pidl)
EXTERN_C BOOL WINAPI	IEILIsEqual (_In_ LPCITEMIDLIST pidl1, _In_ LPCITEMIDLIST pidl2, _In_ BOOL bUnknown)
static VOID	SHDOCVW_PathDeleteInvalidChars (LPWSTR pszDisplayName)
static HRESULT	SHDOCVW_CreateShortcut (_In_ LPCWSTR pszLnkFileName, _In_ PCIDLIST_ABSOLUTE pidlTarget, _In_opt_ LPCWSTR pszDescription)
static LRESULT	AddFavoriteDialogProc (_In_ HWND hwnd, _In_ UINT uMsg, _In_ WPARAM wParam, _In_ LPARAM lParam)
static int CALLBACK	AddFavoriteDialogCallback (_In_ HWND hwnd, _In_ UINT uMsg, _In_ LPARAM lParam, _In_ LPARAM lpData)
static HRESULT	AddFavoriteDialog (_In_ HWND hwnd, _Inout_ PWSTR pszDir, _Inout_ PWSTR pszTitle, _In_ PCWSTR pszText)
EXTERN_C HRESULT WINAPI	AddUrlToFavorites (_In_ HWND hwnd, _In_ LPCWSTR pszUrlW, _In_ LPCWSTR pszTitleW, _In_ BOOL NoUI)

Macro Definition Documentation

PATH_VALID_ELEMENT

#define PATH_VALID_ELEMENT

Value:

    ( \
    PATH_CHAR_CLASS_DOT | PATH_CHAR_CLASS_SEMICOLON | PATH_CHAR_CLASS_COMMA | \
    PATH_CHAR_CLASS_SPACE | PATH_CHAR_CLASS_OTHER_VALID \
)

Function Documentation

_ILIsDesktop()

EXTERN_C BOOL _ILIsDesktop ( LPCITEMIDLIST  pidl )

inline

Definition at line 123 of file utility.cpp.

{
    return (pidl == NULL || pidl->mkid.cb == 0);
}

AddFavoriteDialog()

static HRESULT AddFavoriteDialog ( _In_ HWND  hwnd, _Inout_ PWSTR  pszDir, _Inout_ PWSTR  pszTitle, _In_ PCWSTR  pszText  )

static

Definition at line 249 of file utility.cpp.

{
    CComHeapPtr<ITEMIDLIST> pidlBaseDir, pidl;
    HRESULT hr = SHParseDisplayName(pszDir, NULL, &pidlBaseDir, 0, NULL);
    if (FAILED(hr))
        return hr;
 
    ADDFAVORITEDIALOGDATA data = { pszTitle, PathFindFileNameW(pszDir) };
    const UINT flags = BIF_RETURNONLYFSDIRS | BIF_EDITBOX | BIF_USENEWUI;
    WCHAR szBuf[MAX_PATH];
    *szBuf = UNICODE_NULL;
    BROWSEINFOW info = { hwnd, pidlBaseDir, szBuf, pszText, flags, AddFavoriteDialogCallback, (LPARAM)&data };
    PIDLIST_ABSOLUTE pidlResult = SHBrowseForFolderW(&info);
    if (!pidlResult)
        return S_FALSE;
    pidl.Attach(pidlResult);
    return SHGetPathFromIDListW(pidl, pszDir) ? S_OK : E_FAIL;
}

Referenced by AddUrlToFavorites().

AddFavoriteDialogCallback()

static int CALLBACK AddFavoriteDialogCallback ( _In_ HWND  hwnd, _In_ UINT  uMsg, _In_ LPARAM  lParam, _In_ LPARAM  lpData  )

static

Definition at line 222 of file utility.cpp.

{
    ADDFAVORITEDIALOGDATA &data = *(ADDFAVORITEDIALOGDATA*)lpData;
    switch (uMsg)
    {
    case BFFM_INITIALIZED:
        SetWindowLongPtrW(hwnd, GWLP_USERDATA, (LONG_PTR)&data);
        data.pOrgProc = (WNDPROC)SetWindowLongPtrW(hwnd, GWLP_WNDPROC, (LONG_PTR)AddFavoriteDialogProc);
        data.hEdit = FindWindowExW(hwnd, NULL, L"EDIT", NULL);
        SendMessageW(hwnd, WM_SETTEXT, 0, (LPARAM)data.pszCaption);
        SendMessageW(hwnd, WM_NEXTDLGCTL, (WPARAM)data.hEdit, TRUE);
        SendMessageW(data.hEdit, WM_SETTEXT, 0, (LPARAM)data.pszName); // Initialize
        break;
    case BFFM_SELCHANGED:
        SendMessageW(data.hEdit, WM_SETTEXT, 0, (LPARAM)data.pszName); // Restore
        break;
    case BFFM_VALIDATEFAILED:
        return TRUE;
    }
    return 0;
}

Referenced by AddFavoriteDialog().

AddFavoriteDialogProc()

static LRESULT AddFavoriteDialogProc ( _In_ HWND  hwnd, _In_ UINT  uMsg, _In_ WPARAM  wParam, _In_ LPARAM  lParam  )

static

Definition at line 198 of file utility.cpp.

{
    ADDFAVORITEDIALOGDATA &data = *(ADDFAVORITEDIALOGDATA*)GetWindowLongPtrW(hwnd, GWLP_USERDATA);
    NMHDR *pHdr = (NMHDR*)lParam;
    if (uMsg == WM_NOTIFY)
        data.IgnoreChanges |= (pHdr->code == NM_CLICK || pHdr->code == NM_SETFOCUS); // Ignore changes from the tree
    if (uMsg == WM_COMMAND && HIWORD(wParam) == EN_SETFOCUS)
        data.IgnoreChanges = (HWND)lParam != data.hEdit; // Ignore changes in the tree rename edit
    if (uMsg == WM_COMMAND && HIWORD(wParam) == EN_UPDATE && (HWND)lParam == data.hEdit && !data.IgnoreChanges)
    {
        SendMessageW(data.hEdit, WM_GETTEXT, MAX_PATH, (LPARAM)data.pszName); // Save user input
        BOOL bad = FALSE;
        for (PWSTR p = data.pszName; *p; ++p)
            bad |= !PathIsValidCharW(*p, PATH_VALID_ELEMENT);
        SendMessageW(hwnd, BFFM_ENABLEOK, 0, !bad);
    }
    return CallWindowProcW(data.pOrgProc, hwnd, uMsg, wParam, lParam);
}

Referenced by AddFavoriteDialogCallback().

AddUrlToFavorites()

EXTERN_C HRESULT WINAPI AddUrlToFavorites	(	_In_ HWND	hwnd,
		_In_ LPCWSTR	pszUrlW,
		_In_ LPCWSTR	pszTitleW,
		_In_ BOOL	NoUI
	)

Definition at line 276 of file utility.cpp.

{
    TRACE("%p, %s, %s, %d\n", hwnd, wine_dbgstr_w(pszUrlW), wine_dbgstr_w(pszTitleW), NoUI);
 
    if (!pszUrlW || !pszTitleW)
        return E_INVALIDARG;
 
    CCoInit CoInit;
    HRESULT hr = S_OK;
    BOOL IsUrl = PathIsURLW(pszUrlW);
    PCWSTR pszExt = IsUrl ? L".url" : L".lnk";
    WCHAR szTitle[MAX_PATH];
    StringCchCopyW(szTitle, _countof(szTitle), pszTitleW);
 
    WCHAR szDir[MAX_PATH];
    if (!SHGetSpecialFolderPathW(hwnd, szDir, CSIDL_FAVORITES, TRUE))
        return E_FAIL;
 
    if (!NoUI)
    {
        PCWSTR pszText = IsUrl ? pszUrlW : PathFindFileNameW(pszUrlW);
        if ((hr = AddFavoriteDialog(hwnd, szDir, szTitle, pszText)) == S_FALSE)
            return hr;
    }
    SHDOCVW_PathDeleteInvalidChars(szTitle);
    
    WCHAR szLnk[MAX_PATH * 2];
    if (SUCCEEDED(hr))
        hr = StringCchPrintfW(szLnk, _countof(szLnk), L"%s\\%s%s", szDir, szTitle, pszExt);
 
    if (IsUrl && SUCCEEDED(hr))
    {
        // TODO: For the !NoUI case, we should ask the user if they want to overwrite if the file already exists
        hr = WritePrivateProfileStringW(L"InternetShortcut", L"URL", pszUrlW, szLnk) ? S_OK : E_FAIL;
    }
    else if (SUCCEEDED(hr))
    {
        BOOL bMustCopy;
        if (NoUI)
            hr = SHGetNewLinkInfoW(pszUrlW, szDir, szLnk, &bMustCopy, 0) ? S_OK : E_FAIL;
 
        CComHeapPtr<ITEMIDLIST> pidl;
        if (!FAILED_UNEXPECTEDLY(hr))
            hr = SHParseDisplayName(pszUrlW, NULL, &pidl, 0, NULL);
        if (!FAILED_UNEXPECTEDLY(hr))
            hr = SHDOCVW_CreateShortcut(szLnk, pidl, NULL);
    }
 
    if (FAILED(hr) && !NoUI)
        SHELL_ErrorBox(hwnd, hr);
    return hr;
}

Referenced by AddItemToFavorites().

GetMenuItemIdByPos()

static INT_PTR GetMenuItemIdByPos ( HMENU  hMenu, UINT  Pos  )

inlinestatic

Definition at line 16 of file utility.cpp.

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

Referenced by SHELL_FindVerbPos().

IEILIsEqual()

EXTERN_C BOOL WINAPI IEILIsEqual	(	_In_ LPCITEMIDLIST	pidl1,
		_In_ LPCITEMIDLIST	pidl2,
		_In_ BOOL	bUnknown
	)

Definition at line 132 of file utility.cpp.

{
    UINT cb1 = ILGetSize(pidl1), cb2 = ILGetSize(pidl2);
    if (cb1 == cb2 && memcmp(pidl1, pidl2, cb1) == 0)
        return TRUE;
 
    FIXME("%p, %p\n", pidl1, pidl2);
    return FALSE;
}

Referenced by CMruBase::_IsEqual().

IsMenuSeparator()

static BOOL IsMenuSeparator ( HMENU  hMenu, UINT  Pos  )

inlinestatic

Definition at line 26 of file utility.cpp.

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

SHDOCVW_CreateShortcut()

static HRESULT SHDOCVW_CreateShortcut ( _In_ LPCWSTR  pszLnkFileName, _In_ PCIDLIST_ABSOLUTE  pidlTarget, _In_opt_ LPCWSTR  pszDescription  )

static

Definition at line 162 of file utility.cpp.

{
    HRESULT hr;
 
    CComPtr<IShellLink> psl;
    hr = CoCreateInstance(CLSID_ShellLink, NULL,  CLSCTX_INPROC_SERVER,
                          IID_PPV_ARG(IShellLink, &psl));
    if (FAILED_UNEXPECTEDLY(hr))
        return hr;
 
    psl->SetIDList(pidlTarget);
 
    if (pszDescription)
        psl->SetDescription(pszDescription);
 
    CComPtr<IPersistFile> ppf;
    hr = psl->QueryInterface(IID_PPV_ARG(IPersistFile, &ppf));
    if (FAILED_UNEXPECTEDLY(hr))
        return hr;
 
    return ppf->Save(pszLnkFileName, TRUE);
}

Referenced by AddUrlToFavorites().

SHDOCVW_PathDeleteInvalidChars()

static VOID SHDOCVW_PathDeleteInvalidChars ( LPWSTR  pszDisplayName )

static

Definition at line 146 of file utility.cpp.

{
#define PATH_VALID_ELEMENT ( \
    PATH_CHAR_CLASS_DOT | PATH_CHAR_CLASS_SEMICOLON | PATH_CHAR_CLASS_COMMA | \
    PATH_CHAR_CLASS_SPACE | PATH_CHAR_CLASS_OTHER_VALID \
)
    PWCHAR pch, pchSrc;
    for (pch = pchSrc = pszDisplayName; *pchSrc; ++pchSrc)
    {
        if (PathIsValidCharW(*pchSrc, PATH_VALID_ELEMENT))
            *pch++ = *pchSrc;
    }
    *pch = UNICODE_NULL;
}

Referenced by AddUrlToFavorites().

SHELL_CompareAbsoluteIDs()

static HRESULT SHELL_CompareAbsoluteIDs ( LPARAM  lParam, PCIDLIST_ABSOLUTE  a, PCIDLIST_ABSOLUTE  b  )

static

Definition at line 58 of file utility.cpp.

{
    IShellFolder *psf;
    HRESULT hr = SHGetDesktopFolder(&psf);
    if (FAILED(hr))
        return hr;
    hr = psf->CompareIDs(lParam, a, b);
    psf->Release();
    return hr;
}

Referenced by SHELL_IsEqualAbsoluteID().

SHELL_FindVerbPos()

static int SHELL_FindVerbPos ( IContextMenu *  pcm, UINT  idCmdFirst, HMENU  hMenu, LPCWSTR  Verb  )

static

Definition at line 93 of file utility.cpp.

{
    for (UINT i = 0, c = GetMenuItemCount(hMenu); i < c; ++i)
    {
        INT_PTR id = GetMenuItemIdByPos(hMenu, i);
        if (id != -1 && SHELL_IsVerb(pcm, id - idCmdFirst, Verb))
            return i;
    }
    return -1;
}

Referenced by SHELL_RemoveVerb().

SHELL_GetIDListFromObject()

EXTERN_C HRESULT SHELL_GetIDListFromObject	(	IUnknown *	punk,
		PIDLIST_ABSOLUTE *	ppidl
	)

Definition at line 36 of file utility.cpp.

{
#if DLL_EXPORT_VERSION >= _WIN32_WINNT_VISTA && 0 // FIXME: SHELL32 not ready yet
    return SHGetIDListFromObject(punk, ppidl);
#else
    HRESULT hr;
    IPersistFolder2 *pf2;
    if (SUCCEEDED(hr = punk->QueryInterface(IID_PPV_ARG(IPersistFolder2, &pf2))))
    {
        hr = pf2->GetCurFolder(ppidl);
        pf2->Release();
    }
    IPersistIDList *pil;
    if (FAILED(hr) && SUCCEEDED(hr = punk->QueryInterface(IID_PPV_ARG(IPersistIDList, &pil))))
    {
        hr = pil->GetIDList(ppidl);
        pil->Release();
    }
    return hr;
#endif
}

Referenced by SHDOCVW_GetCurrentLocationFromView().

SHELL_IsEqualAbsoluteID()

EXTERN_C BOOL SHELL_IsEqualAbsoluteID	(	PCIDLIST_ABSOLUTE	a,
		PCIDLIST_ABSOLUTE	b
	)

Definition at line 70 of file utility.cpp.

{
    return !SHELL_CompareAbsoluteIDs(SHCIDS_CANONICALONLY, a, b);
}

SHELL_IsVerb()

EXTERN_C BOOL SHELL_IsVerb	(	IContextMenu *	pcm,
		UINT_PTR	idCmd,
		LPCWSTR	Verb
	)

Definition at line 76 of file utility.cpp.

{
    HRESULT hr;
    WCHAR wide[MAX_PATH];
    if (SUCCEEDED(hr = pcm->GetCommandString(idCmd, GCS_VERBW, NULL, (LPSTR)wide, _countof(wide))))
        return !lstrcmpiW(wide, Verb);
 
    CHAR ansi[_countof(wide)], buf[MAX_PATH];
    if (SHUnicodeToAnsi(Verb, buf, _countof(buf)))
    {
        if (SUCCEEDED(hr = pcm->GetCommandString(idCmd, GCS_VERBA, NULL, ansi, _countof(ansi))))
            return !lstrcmpiA(ansi, buf);
    }
    return FALSE;
}

Referenced by SHELL_FindVerbPos().

SHELL_RemoveVerb()

EXTERN_C VOID SHELL_RemoveVerb	(	IContextMenu *	pcm,
		UINT	idCmdFirst,
		HMENU	hMenu,
		LPCWSTR	Verb
	)

Definition at line 105 of file utility.cpp.

{
    int nPos = SHELL_FindVerbPos(pcm, idCmdFirst, hMenu, Verb);
    if (nPos < 0)
        return;
    int nCount = GetMenuItemCount(hMenu);
    BOOL bSepBefore = nPos && IsMenuSeparator(hMenu, nPos - 1);
    BOOL bSepAfter = IsMenuSeparator(hMenu, nPos + 1);
    if (DeleteMenu(hMenu, nPos, MF_BYPOSITION))
    {
        if ((bSepBefore && bSepAfter) || (bSepAfter && nPos == 0))
            DeleteMenu(hMenu, nPos, MF_BYPOSITION);
        else if (bSepBefore && nPos + 1 == nCount)
            DeleteMenu(hMenu, nPos - 1, MF_BYPOSITION);
    }
}

WINE_DEFAULT_DEBUG_CHANNEL()

WINE_DEFAULT_DEBUG_CHANNEL

(

shdocvw

)