windows-nt/Source/XPSP1/NT/shell/shell32/openwith.cpp
2020-09-26 16:20:57 +08:00

774 lines
21 KiB
C++

#include "shellprv.h"
#include "ids.h"
#include <shlwapi.h>
#include "openwith.h"
#include "uemapp.h"
#include "mtpt.h"
#include "fassoc.h"
#include "filetbl.h"
#include "datautil.h"
#include <dpa.h>
#include "defcm.h"
#define TF_OPENWITHMENU 0x00000000
#define SZOPENWITHLIST TEXT("OpenWithList")
#define REGSTR_PATH_EXPLORER_FILEEXTS TEXT("Software\\Microsoft\\Windows\\CurrentVersion\\Explorer\\FileExts")
#define OPEN_WITH_LIST_MAX_ITEMS 10
//
// OpenWithListOpen()
// allocates and initializes the state for the openwithlist
//
HRESULT OpenWithListOpen(IN LPCTSTR pszExt, HANDLE *phmru)
{
*phmru = 0;
if (pszExt && *pszExt)
{
TCHAR szSubKey[MAX_PATH];
// Build up the subkey string.
wnsprintf(szSubKey, SIZECHARS(szSubKey), TEXT("%s\\%s\\%s"), REGSTR_PATH_EXPLORER_FILEEXTS, pszExt, SZOPENWITHLIST);
MRUINFO mi = {sizeof(mi), OPEN_WITH_LIST_MAX_ITEMS, 0, HKEY_CURRENT_USER, szSubKey, NULL};
*phmru = CreateMRUList(&mi);
}
return *phmru ? S_OK : E_OUTOFMEMORY;
}
HRESULT _AddItem(HANDLE hmru, LPCTSTR pszName)
{
HRESULT hr = S_OK;
if (hmru)
{
int cItems = EnumMRUList(hmru, -1, NULL, 0);
// just trim us down to make room...
while (cItems >= OPEN_WITH_LIST_MAX_ITEMS)
DelMRUString(hmru, --cItems);
if (0 > AddMRUString(hmru, pszName))
hr = E_UNEXPECTED;
}
return hr;
}
void _DeleteItem(HANDLE hmru, LPCTSTR pszName)
{
int iItem = FindMRUString(hmru, pszName, NULL);
if (0 <= iItem)
{
DelMRUString(hmru, iItem);
}
}
void _AddProgidForExt(LPCWSTR pszExt);
STDAPI OpenWithListRegister(DWORD dwFlags, LPCTSTR pszExt, LPCTSTR pszVerb, HKEY hkProgid)
{
//
// ----> Peruser entries are stored here
// HKCU\Software\Microsoft\Windows\CurrentVersion\Explorer\FileExts
// \.Ext
// Application = "foo.exe"
// \OpenWithList
// MRUList = "ab"
// a = "App.exe"
// b = "foo.exe"
//
// ----> for permanent entries are stored un HKCR
// HKCR
// \.Ext
// \OpenWithList
// \app.exe
//
// ----> and applications or the system can write app association here
// \Applications
// \APP.EXE
// \shell...
// \foo.exe
// \shell...
//
//
HANDLE hmru;
HRESULT hr = OpenWithListOpen(pszExt, &hmru);
if (SUCCEEDED(hr))
{
TCHAR szPath[MAX_PATH];
hr = AssocQueryStringByKey(ASSOCF_VERIFY, ASSOCSTR_EXECUTABLE, hkProgid, pszVerb, szPath, (LPDWORD)MAKEINTRESOURCE(SIZECHARS(szPath)));
if (SUCCEEDED(hr))
{
LPCTSTR pszExe = PathFindFileName(szPath);
if (IsPathInOpenWithKillList(pszExe))
hr = E_ACCESSDENIED;
else
hr = AssocMakeApplicationByKey(ASSOCMAKEF_VERIFY, hkProgid, pszVerb);
if (SUCCEEDED(hr))
{
TraceMsg(TF_OPENWITHMENU, "[%X] OpenWithListRegister() adding %s",hmru, pszExe);
hr = _AddItem(hmru, pszExe);
}
if (FAILED(hr))
_DeleteItem(hmru, pszExe);
}
FreeMRUList(hmru);
}
_AddProgidForExt(pszExt);
return hr;
}
STDAPI_(void) OpenWithListSoftRegisterProcess(DWORD dwFlags, LPCTSTR pszExt, LPCTSTR pszProcess)
{
HANDLE hmru;
if (SUCCEEDED(OpenWithListOpen(pszExt, &hmru)))
{
TCHAR szApp[MAX_PATH];
if (!pszProcess)
{
if (GetModuleFileName(NULL, szApp, SIZECHARS(szApp)))
pszProcess = szApp;
}
if (pszProcess && !IsPathInOpenWithKillList(pszProcess))
_AddItem(hmru, PathFindFileName(pszProcess));
FreeMRUList(hmru);
}
}
class COpenWithArray : public CDPA<CAppInfo>
{
public:
~COpenWithArray();
HRESULT FillArray(PCWSTR pszExt);
private:
static int CALLBACK _DeleteAppInfo(CAppInfo *pai, void *pv)
{ if (pai) delete pai; return 1; }
};
COpenWithArray::~COpenWithArray()
{
if ((HDPA)this)
DestroyCallback(_DeleteAppInfo, NULL);
}
HRESULT COpenWithArray::FillArray(PCWSTR pszExt)
{
IEnumAssocHandlers *penum;
HRESULT hr = SHAssocEnumHandlers(pszExt, &penum);
if (SUCCEEDED(hr))
{
IAssocHandler *pah;
while (S_OK == penum->Next(1, &pah, NULL))
{
// we only want the best
if (S_OK == pah->IsRecommended())
{
CAppInfo *pai = new CAppInfo(pah);
if (pai)
{
if (pai->Init())
{
// Trim duplicate items before we add them for other programs
int i = 0;
for (; i < GetPtrCount(); i++)
{
if (0 == lstrcmpi(pai->Name(), GetPtr(i)->Name()))
{
// its a match
break;
}
}
// if we dont add this to the dpa
// then we need to clean it up
if (i == GetPtrCount() && -1 != AppendPtr(pai))
pai = NULL;
}
if (pai)
delete pai;
}
}
pah->Release();
}
penum->Release();
}
return hr;
}
class COpenWithMenu : public IContextMenu3, IShellExtInit
{
// IUnknown
STDMETHOD(QueryInterface)(REFIID riid, void **ppvObj);
STDMETHOD_(ULONG,AddRef)(void);
STDMETHOD_(ULONG,Release)(void);
// IContextMenu
STDMETHOD(QueryContextMenu)(HMENU hmenu, UINT indexMenu, UINT idCmdFirst, UINT idCmdLast, UINT uFlags);
STDMETHOD(InvokeCommand)(LPCMINVOKECOMMANDINFO lpici);
STDMETHOD(GetCommandString)(UINT_PTR idCmd, UINT uType, UINT *pRes, LPSTR pszName, UINT cchMax);
// IContextMenu2
STDMETHOD(HandleMenuMsg)(UINT uMsg, WPARAM wParam, LPARAM lParam);
// IContextMenu3
STDMETHOD(HandleMenuMsg2)(UINT uMsg, WPARAM wParam, LPARAM lParam,LRESULT *lResult);
// IShellExtInit
STDMETHOD(Initialize)(LPCITEMIDLIST pidlFolder, IDataObject *pdtobj, HKEY hkeyProgID);
friend HRESULT COpenWithMenu_CreateInstance(IUnknown* pUnkOuter, REFIID riid, void **ppvOut);
protected: // methods
COpenWithMenu();
~COpenWithMenu();
//Handle Menu messages submitted to HandleMenuMsg
void DrawItem(DRAWITEMSTRUCT *lpdi);
LRESULT MeasureItem(MEASUREITEMSTRUCT *lpmi);
BOOL InitMenuPopup(HMENU hMenu);
//Internal Helpers
HRESULT _GetHelpText(UINT_PTR idCmd, LPSTR pszName, UINT cchMax, BOOL fUnicode);
HRESULT _MatchMenuItem(TCHAR ch, LRESULT* plRes);
protected: // members
LONG _cRef;
HMENU _hMenu;
BOOL _fMenuNeedsInit;
UINT _idCmdFirst;
COpenWithArray _owa;
int _nItems;
UINT _uFlags;
IDataObject *_pdtobj;
TCHAR _szPath[MAX_PATH];
};
COpenWithMenu::COpenWithMenu() : _cRef(1)
{
TraceMsg(TF_OPENWITHMENU, "ctor COpenWithMenu %x", this);
}
COpenWithMenu::~COpenWithMenu()
{
TraceMsg(TF_OPENWITHMENU, "dtor COpenWithMenu %x", this);
if (_pdtobj)
_pdtobj->Release();
if (_hMenu)
{
ASSERT(_nItems);
DestroyMenu(_hMenu);
}
}
STDAPI COpenWithMenu_CreateInstance(IUnknown* pUnkOuter, REFIID riid, void **ppvOut)
{
*ppvOut = NULL;
if (pUnkOuter)
return CLASS_E_NOAGGREGATION;
COpenWithMenu * powm = new COpenWithMenu();
if (!powm)
return E_OUTOFMEMORY;
HRESULT hr = powm->QueryInterface(riid, ppvOut);
powm->Release();
return hr;
}
HRESULT COpenWithMenu::QueryInterface(REFIID riid, void **ppvObj)
{
static const QITAB qit[] = {
QITABENTMULTI(COpenWithMenu, IContextMenu, IContextMenu3),
QITABENTMULTI(COpenWithMenu, IContextMenu2, IContextMenu3),
QITABENT(COpenWithMenu, IContextMenu3),
QITABENT(COpenWithMenu, IShellExtInit),
{ 0 },
};
return QISearch(this, qit, riid, ppvObj);
}
ULONG COpenWithMenu::AddRef()
{
return InterlockedIncrement(&_cRef);
}
ULONG COpenWithMenu::Release()
{
if (InterlockedDecrement(&_cRef))
return _cRef;
delete this;
return 0;
}
/*
Purpose:
Add Open/Edit/Default verb to extension app list
*/
HRESULT AddVerbItems(LPCTSTR pszExt)
{
IQueryAssociations *pqa;
HRESULT hr = AssocCreate(CLSID_QueryAssociations, IID_PPV_ARG(IQueryAssociations, &pqa));
if (SUCCEEDED(hr))
{
hr = pqa->Init(0, pszExt, NULL, NULL);
if (SUCCEEDED(hr))
{
HKEY hkeyClass;
hr = pqa->GetKey(0, ASSOCKEY_SHELLEXECCLASS, NULL, &hkeyClass);
if (SUCCEEDED(hr))
{
OpenWithListRegister(0, pszExt, NULL, hkeyClass);
RegCloseKey(hkeyClass);
}
// we add in the editor too
if (SUCCEEDED(pqa->GetKey(0, ASSOCKEY_SHELLEXECCLASS, L"Edit", &hkeyClass)))
{
OpenWithListRegister(0, pszExt, NULL, hkeyClass);
RegCloseKey(hkeyClass);
}
hr = S_OK;
}
pqa->Release();
}
return hr;
}
//
// Our context menu IDs are assigned like this
//
// idCmdFirst = Open With Custom Program (either on main menu or on popup)
// idCmdFirst+1 through idCmdFirst+_nItems = Open With program in OpenWithList
#define OWMENU_BROWSE 0
#define OWMENU_APPFIRST 1
HRESULT COpenWithMenu::QueryContextMenu(HMENU hmenu, UINT indexMenu, UINT idCmdFirst, UINT idCmdLast, UINT uFlags)
{
MENUITEMINFO mii;
LPTSTR pszExt;
TCHAR szOpenWithMenu[80];
_idCmdFirst = idCmdFirst;
_uFlags = uFlags;
if (SUCCEEDED(PathFromDataObject(_pdtobj, _szPath, ARRAYSIZE(_szPath))))
{
// No openwith context menu for executables.
if (PathIsExe(_szPath))
return S_OK;
pszExt = PathFindExtension(_szPath);
if (pszExt && *pszExt)
{
// Add Open/Edit/Default verb to extension app list
if (SUCCEEDED(AddVerbItems(pszExt)))
{
// Do this only if AddVerbItems succeeded; otherwise,
// we would create an empty MRU for a nonexisting class,
// causing the class to spring into existence and cause
// the "Open With" dialog to think we are overriding
// rather than creating new.
// get extension app list
if (_owa.Create(4) && SUCCEEDED(_owa.FillArray(pszExt)))
{
_nItems = _owa.GetPtrCount();
if (1 == _nItems)
{
// For known file type(there is at least one verb under its progid),
// if there is only one item in its openwithlist, don't show open with sub menu
_nItems = 0;
}
}
}
}
}
LoadString(g_hinst, (_nItems ? IDS_OPENWITH : IDS_OPENWITHNEW), szOpenWithMenu, ARRAYSIZE(szOpenWithMenu));
if (_nItems)
{
// we need to create a submenu
// with all of our goodies
_hMenu = CreatePopupMenu();
if (_hMenu)
{
_fMenuNeedsInit = TRUE;
mii.cbSize = sizeof(MENUITEMINFO);
mii.fMask = MIIM_ID|MIIM_TYPE|MIIM_DATA;
mii.wID = idCmdFirst+OWMENU_APPFIRST;
mii.fType = MFT_STRING;
mii.dwTypeData = szOpenWithMenu;
mii.dwItemData = 0;
InsertMenuItem(_hMenu,0,TRUE,&mii);
mii.fMask = MIIM_ID|MIIM_SUBMENU|MIIM_TYPE;
mii.fType = MFT_STRING;
mii.wID = idCmdFirst+OWMENU_BROWSE;
mii.hSubMenu = _hMenu;
mii.dwTypeData = szOpenWithMenu;
InsertMenuItem(hmenu,indexMenu,TRUE,&mii);
}
}
else
{
mii.cbSize = sizeof(MENUITEMINFO);
mii.fMask = MIIM_ID|MIIM_TYPE|MIIM_DATA;
mii.fType = MFT_STRING;
mii.wID = idCmdFirst+OWMENU_BROWSE;
mii.dwTypeData = szOpenWithMenu;
mii.dwItemData = 0;
InsertMenuItem(hmenu,indexMenu,TRUE,&mii);
}
return ResultFromShort(_nItems + 1);
}
HRESULT COpenWithMenu::InvokeCommand(LPCMINVOKECOMMANDINFO pici)
{
HRESULT hr = E_OUTOFMEMORY;
CMINVOKECOMMANDINFOEX ici;
void * pvFree;
// maybe these two routines should be collapsed into one?
if ((IS_INTRESOURCE(pici->lpVerb) || 0 == lstrcmpiA(pici->lpVerb, "openas"))
&& SUCCEEDED(ICI2ICIX(pici, &ici, &pvFree)))
{
BOOL fOpenAs = TRUE;
if (pici->lpVerb && IS_INTRESOURCE(pici->lpVerb))
{
int i = LOWORD(pici->lpVerb) - OWMENU_APPFIRST;
if (i < _owa.GetPtrCount())
{
hr = _owa.GetPtr(i)->Handler()->Invoke(&ici, _szPath);
fOpenAs = FALSE;
}
}
if (fOpenAs)
{
SHELLEXECUTEINFO ei = {0};
hr = ICIX2SEI(&ici, &ei);
if (SUCCEEDED(hr))
{
// use the "Unknown" key so we get the openwith prompt
ei.lpFile = _szPath;
// dont do the zone check before the user picks the app.
// wait until they actually try to invoke the file.
ei.fMask |= SEE_MASK_NOZONECHECKS;
RegOpenKeyEx(HKEY_CLASSES_ROOT, TEXT("Unknown"), 0, MAXIMUM_ALLOWED, &ei.hkeyClass);
if (!(_uFlags & CMF_DEFAULTONLY))
{
// defview sets CFM_DEFAULTONLY when the user is double-clicking. We check it
// here since we want do NOT want to query the class store if the user explicitly
// right-clicked on the menu and choo se openwith.
// pop up open with dialog without querying class store
ei.fMask |= SEE_MASK_NOQUERYCLASSSTORE;
}
if (ei.hkeyClass)
{
ei.fMask |= SEE_MASK_CLASSKEY;
if (ShellExecuteEx(&ei))
{
hr = S_OK;
if (UEMIsLoaded())
{
// note that we already got a UIBL_DOTASSOC (from
// OpenAs_RunDLL or whatever it is that 'Unknown'
// runs). so the Uassist analysis app will have to
// subtract it off
UEMFireEvent(&UEMIID_SHELL, UEME_INSTRBROWSER, UEMF_INSTRUMENT, UIBW_RUNASSOC, UIBL_DOTNOASSOC);
}
}
else
{
hr = HRESULT_FROM_WIN32(GetLastError());
}
RegCloseKey(ei.hkeyClass);
}
else
hr = E_FAIL;
}
}
LocalFree(pvFree); // accepts NULL
}
return hr;
}
HRESULT COpenWithMenu::_GetHelpText(UINT_PTR idCmd, LPSTR pszName, UINT cchMax, BOOL fUnicode)
{
UINT ids;
LPCTSTR pszFriendly = NULL;
if (idCmd == OWMENU_BROWSE)
{
ids = IDS_OPENWITHHELP;
pszFriendly = TEXT("");
}
else if ((idCmd-OWMENU_APPFIRST) < (UINT_PTR)_owa.GetPtrCount())
{
ids = IDS_OPENWITHAPPHELP;
pszFriendly = _owa.GetPtr(idCmd-OWMENU_APPFIRST)->UIName();
}
if (!pszFriendly)
return E_FAIL;
if (fUnicode)
{
WCHAR wszFormat[80];
LoadStringW(HINST_THISDLL, ids, wszFormat, ARRAYSIZE(wszFormat));
wnsprintfW((LPWSTR)pszName, cchMax, wszFormat, pszFriendly);
}
else
{
CHAR szFormat[80];
LoadStringA(HINST_THISDLL, ids, szFormat, ARRAYSIZE(szFormat));
wnsprintfA(pszName, cchMax, szFormat, pszFriendly);
}
return S_OK;
}
const ICIVERBTOIDMAP c_sIDVerbMap[] =
{
{ L"openas", "openas", OWMENU_BROWSE, OWMENU_BROWSE, },
};
HRESULT COpenWithMenu::GetCommandString(UINT_PTR idCmd, UINT uType, UINT *pRes, LPSTR pszName, UINT cchMax)
{
switch (uType)
{
case GCS_VERBA:
case GCS_VERBW:
return SHMapCmdIDToVerb(idCmd, c_sIDVerbMap, ARRAYSIZE(c_sIDVerbMap), pszName, cchMax, GCS_VERBW == uType);
case GCS_HELPTEXTA:
case GCS_HELPTEXTW:
return _GetHelpText(idCmd, pszName, cchMax, uType == GCS_HELPTEXTW);
}
return E_NOTIMPL;
}
HRESULT COpenWithMenu::HandleMenuMsg(UINT uMsg, WPARAM wParam, LPARAM lParam)
{
return HandleMenuMsg2(uMsg,wParam,lParam,NULL);
}
// Defined in fsmenu.cpp
BOOL _MenuCharMatch(LPCTSTR lpsz, TCHAR ch, BOOL fIgnoreAmpersand);
HRESULT COpenWithMenu::_MatchMenuItem(TCHAR ch, LRESULT* plRes)
{
if (plRes == NULL)
return S_FALSE;
int iLastSelectedItem = -1;
int iNextMatch = -1;
BOOL fMoreThanOneMatch = FALSE;
int c = GetMenuItemCount(_hMenu);
// Pass 1: Locate the Selected Item
for (int i = 0; i < c; i++)
{
MENUITEMINFO mii = {0};
mii.cbSize = sizeof(mii);
mii.fMask = MIIM_STATE;
if (GetMenuItemInfo(_hMenu, i, MF_BYPOSITION, &mii))
{
if (mii.fState & MFS_HILITE)
{
iLastSelectedItem = i;
break;
}
}
}
// Pass 2: Starting from the selected item, locate the first item with the matching name.
for (int i = iLastSelectedItem + 1; i < c; i++)
{
if (i < _owa.GetPtrCount()
&& _MenuCharMatch(_owa.GetPtr(i)->UIName(), ch, FALSE))
{
if (iNextMatch != -1)
{
fMoreThanOneMatch = TRUE;
break; // We found all the info we need
}
else
{
iNextMatch = i;
}
}
}
// Pass 3: If we did not find a match, or if there was only one match
// Search from the first item, to the Selected Item
if (iNextMatch == -1 || fMoreThanOneMatch == FALSE)
{
for (int i = 0; i <= iLastSelectedItem; i++)
{
if (i < _owa.GetPtrCount()
&& _MenuCharMatch(_owa.GetPtr(i)->UIName(), ch, FALSE))
{
if (iNextMatch != -1)
{
fMoreThanOneMatch = TRUE;
break;
}
else
{
iNextMatch = i;
}
}
}
}
if (iNextMatch != -1)
{
*plRes = MAKELONG(iNextMatch, fMoreThanOneMatch? MNC_SELECT : MNC_EXECUTE);
}
else
{
*plRes = MAKELONG(0, MNC_IGNORE);
}
return S_OK;
}
HRESULT COpenWithMenu::HandleMenuMsg2(UINT uMsg, WPARAM wParam, LPARAM lParam,LRESULT *plResult)
{
LRESULT lResult = 0;
HRESULT hr = S_OK;
switch (uMsg)
{
case WM_INITMENUPOPUP:
InitMenuPopup(_hMenu);
break;
case WM_DRAWITEM:
DrawItem((DRAWITEMSTRUCT *)lParam);
break;
case WM_MEASUREITEM:
lResult = MeasureItem((MEASUREITEMSTRUCT *)lParam);
break;
case WM_MENUCHAR:
hr = _MatchMenuItem((TCHAR)LOWORD(wParam), &lResult);
break;
default:
hr = E_NOTIMPL;
break;
}
if (plResult)
*plResult = lResult;
return hr;
}
HRESULT COpenWithMenu::Initialize(LPCITEMIDLIST pidlFolder, IDataObject *pdtobj, HKEY hkeyProgID)
{
IUnknown_Set((IUnknown **)&_pdtobj, pdtobj);
return S_OK;
}
#define CXIMAGEGAP 6
#define SRCSTENCIL 0x00B8074AL
void COpenWithMenu::DrawItem(DRAWITEMSTRUCT *lpdi)
{
CAppInfo *pai = _owa.GetPtr(lpdi->itemData);
DrawMenuItem(lpdi, pai->UIName(), pai->IconIndex());
}
LRESULT COpenWithMenu::MeasureItem(MEASUREITEMSTRUCT *pmi)
{
CAppInfo *pai = _owa.GetPtr(pmi->itemData);
return MeasureMenuItem(pmi, pai->UIName());
}
BOOL COpenWithMenu::InitMenuPopup(HMENU hmenu)
{
TraceMsg(TF_OPENWITHMENU, "COpenWithMenu::InitMenuPopup");
if (_fMenuNeedsInit)
{
TCHAR szMenuText[80];
MENUITEMINFO mii;
// remove the place holder.
DeleteMenu(hmenu,0,MF_BYPOSITION);
// add app's in mru list to context menu
for (int i = 0; i < _owa.GetPtrCount(); i++)
{
mii.cbSize = sizeof(MENUITEMINFO);
mii.fMask = MIIM_ID|MIIM_TYPE|MIIM_DATA;
mii.wID = _idCmdFirst + OWMENU_APPFIRST + i;
mii.fType = MFT_OWNERDRAW;
mii.dwItemData = i;
InsertMenuItem(hmenu,GetMenuItemCount(hmenu),TRUE,&mii);
}
// add seperator
AppendMenu(hmenu,MF_SEPARATOR,0,NULL);
// add "Choose Program..."
LoadString(g_hinst, IDS_OPENWITHBROWSE, szMenuText, ARRAYSIZE(szMenuText));
mii.cbSize = sizeof(MENUITEMINFO);
mii.fMask = MIIM_ID|MIIM_TYPE|MIIM_DATA;
mii.wID = _idCmdFirst + OWMENU_BROWSE;
mii.fType = MFT_STRING;
mii.dwTypeData = szMenuText;
mii.dwItemData = 0;
InsertMenuItem(hmenu,GetMenuItemCount(hmenu),TRUE,&mii);
_fMenuNeedsInit = FALSE;
return TRUE;
}
return FALSE;
}