reactos/dll/win32/shell32/CShellDispatch.cpp
Whindmar Saksit 7a810c17e7
[SHELL32] Implement several ShellDispatch methods (#6145)
* Implement most of the methods

* Some suggested changes and better error handling (that I then mask like Windows)
2024-04-10 14:29:26 -07:00

589 lines
16 KiB
C++

/*
* PROJECT: shell32
* LICENSE: LGPL-2.1+ (https://spdx.org/licenses/LGPL-2.1+)
* PURPOSE: IShellDispatch implementation
* COPYRIGHT: Copyright 2015-2018 Mark Jansen (mark.jansen@reactos.org)
* Copyright 2018 Katayama Hirofumi MZ (katayama.hirofumi.mz@gmail.com)
* Copyright 2023 Whindmar Saksit (whindsaks@proton.me)
*/
#include "precomp.h"
#include "winsvc.h"
WINE_DEFAULT_DEBUG_CHANNEL(shell);
EXTERN_C DWORD WINAPI SHGetRestriction(LPCWSTR lpSubKey, LPCWSTR lpSubName, LPCWSTR lpValue);
static HRESULT PostTrayCommand(UINT cmd)
{
HWND hTrayWnd = FindWindowW(L"Shell_TrayWnd", NULL);
return hTrayWnd && PostMessageW(hTrayWnd, WM_COMMAND, cmd, 0) ? S_OK : S_FALSE;
}
CShellDispatch::CShellDispatch()
{
}
CShellDispatch::~CShellDispatch()
{
}
HRESULT CShellDispatch::Initialize()
{
return S_OK;
}
// *** IShellDispatch methods ***
HRESULT STDMETHODCALLTYPE CShellDispatch::get_Application(IDispatch **ppid)
{
TRACE("(%p, %p)\n", this, ppid);
if (!ppid)
return E_INVALIDARG;
*ppid = this;
AddRef();
return S_OK;
}
HRESULT STDMETHODCALLTYPE CShellDispatch::get_Parent(IDispatch **ppid)
{
TRACE("(%p, %p)\n", this, ppid);
if (ppid)
{
*ppid = static_cast<IDispatch*>(this);
AddRef();
}
return S_OK;
}
HRESULT VariantToIdlist(VARIANT* var, LPITEMIDLIST* idlist)
{
HRESULT hr = E_FAIL;
if(V_VT(var) == VT_I4)
{
hr = SHGetSpecialFolderLocation(NULL, V_I4(var), idlist);
}
else if(V_VT(var) == VT_BSTR)
{
hr = SHILCreateFromPathW(V_BSTR(var), idlist, NULL);
}
return hr;
}
HRESULT STDMETHODCALLTYPE CShellDispatch::NameSpace(VARIANT vDir, Folder **ppsdf)
{
TRACE("(%p, %s, %p)\n", this, debugstr_variant(&vDir), ppsdf);
if (!ppsdf)
return E_POINTER;
*ppsdf = NULL;
HRESULT hr;
if (V_VT(&vDir) == VT_I2)
{
hr = VariantChangeType(&vDir, &vDir, 0, VT_I4);
if (FAILED_UNEXPECTEDLY(hr))
return hr;
}
CComHeapPtr<ITEMIDLIST> idlist;
hr = VariantToIdlist(&vDir, &idlist);
if (!SUCCEEDED(hr))
return S_FALSE;
return ShellObjectCreatorInit<CFolder>(static_cast<LPITEMIDLIST>(idlist), IID_PPV_ARG(Folder, ppsdf));
}
static BOOL is_optional_argument(const VARIANT *arg)
{
return V_VT(arg) == VT_ERROR && V_ERROR(arg) == DISP_E_PARAMNOTFOUND;
}
HRESULT STDMETHODCALLTYPE CShellDispatch::BrowseForFolder(LONG Hwnd, BSTR Title, LONG Options, VARIANT RootFolder, Folder **ppsdf)
{
TRACE("(%p, %lu, %ls, %lu, %s, %p)\n", this, Hwnd, Title, Options, debugstr_variant(&RootFolder), ppsdf);
*ppsdf = NULL;
BROWSEINFOW bi = { 0 };
bi.hwndOwner = reinterpret_cast<HWND>(LongToHandle(Hwnd));
bi.lpszTitle = Title;
bi.ulFlags = Options | BIF_NEWDIALOGSTYLE;
CComHeapPtr<ITEMIDLIST> idlist;
if (!is_optional_argument(&RootFolder) && VariantToIdlist(&RootFolder, &idlist) == S_OK)
bi.pidlRoot = idlist;
CComHeapPtr<ITEMIDLIST> selection;
selection.Attach(SHBrowseForFolderW(&bi));
if (!selection)
return S_FALSE;
return ShellObjectCreatorInit<CFolder>(static_cast<LPITEMIDLIST>(selection), IID_PPV_ARG(Folder, ppsdf));
}
HRESULT STDMETHODCALLTYPE CShellDispatch::Windows(IDispatch **ppid)
{
TRACE("(%p, %p)\n", this, ppid);
return CoCreateInstance(CLSID_ShellWindows, NULL, CLSCTX_LOCAL_SERVER, IID_PPV_ARG(IDispatch, ppid));
}
static HRESULT SHELL_OpenFolder(LPCITEMIDLIST pidl, LPCWSTR verb = NULL)
{
SHELLEXECUTEINFOW sei;
sei.cbSize = sizeof(sei);
sei.fMask = SEE_MASK_IDLIST | SEE_MASK_FLAG_DDEWAIT;
sei.hwnd = NULL;
sei.lpVerb = verb;
sei.lpFile = sei.lpParameters = sei.lpDirectory = NULL;
sei.nShow = SW_SHOW;
sei.lpIDList = const_cast<LPITEMIDLIST>(pidl);
if (ShellExecuteExW(&sei))
return S_OK;
DWORD error = GetLastError();
return HRESULT_FROM_WIN32(error);
}
static HRESULT OpenFolder(VARIANT vDir, LPCWSTR verb = NULL)
{
CComHeapPtr<ITEMIDLIST> idlist;
HRESULT hr = VariantToIdlist(&vDir, &idlist);
if (hr == S_OK && SHELL_OpenFolder(idlist, verb) == S_OK)
{
return S_OK;
}
return S_FALSE;
}
HRESULT STDMETHODCALLTYPE CShellDispatch::Open(VARIANT vDir)
{
TRACE("(%p, %s)\n", this, debugstr_variant(&vDir));
return OpenFolder(vDir);
}
HRESULT STDMETHODCALLTYPE CShellDispatch::Explore(VARIANT vDir)
{
TRACE("(%p, %s)\n", this, debugstr_variant(&vDir));
return OpenFolder(vDir, L"explore");
}
HRESULT STDMETHODCALLTYPE CShellDispatch::MinimizeAll()
{
TRACE("(%p)\n", this);
return PostTrayCommand(TRAYCMD_MINIMIZE_ALL);
}
HRESULT STDMETHODCALLTYPE CShellDispatch::UndoMinimizeALL()
{
TRACE("(%p)\n", this);
return PostTrayCommand(TRAYCMD_RESTORE_ALL);
}
HRESULT STDMETHODCALLTYPE CShellDispatch::FileRun()
{
TRACE("(%p)\n", this);
return PostTrayCommand(TRAYCMD_RUN_DIALOG);
}
HRESULT STDMETHODCALLTYPE CShellDispatch::CascadeWindows()
{
TRACE("(%p)\n", this);
return PostTrayCommand(TRAYCMD_CASCADE);
}
HRESULT STDMETHODCALLTYPE CShellDispatch::TileVertically()
{
TRACE("(%p)\n", this);
return PostTrayCommand(TRAYCMD_TILE_V);
}
HRESULT STDMETHODCALLTYPE CShellDispatch::TileHorizontally()
{
TRACE("(%p)\n", this);
return PostTrayCommand(TRAYCMD_TILE_H);
}
HRESULT STDMETHODCALLTYPE CShellDispatch::ShutdownWindows()
{
ExitWindowsDialog(NULL);
return S_OK;
}
HRESULT STDMETHODCALLTYPE CShellDispatch::Suspend()
{
TRACE("(%p)\n", this);
return E_NOTIMPL;
}
HRESULT STDMETHODCALLTYPE CShellDispatch::EjectPC()
{
TRACE("(%p)\n", this);
return E_NOTIMPL;
}
HRESULT STDMETHODCALLTYPE CShellDispatch::SetTime()
{
TRACE("(%p)\n", this);
return PostTrayCommand(TRAYCMD_DATE_AND_TIME);
}
HRESULT STDMETHODCALLTYPE CShellDispatch::TrayProperties()
{
TRACE("(%p)\n", this);
return PostTrayCommand(TRAYCMD_TASKBAR_PROPERTIES);
}
HRESULT STDMETHODCALLTYPE CShellDispatch::Help()
{
TRACE("(%p)\n", this);
return PostTrayCommand(TRAYCMD_HELP_AND_SUPPORT);
}
HRESULT STDMETHODCALLTYPE CShellDispatch::FindFiles()
{
TRACE("(%p)\n", this);
return PostTrayCommand(TRAYCMD_SEARCH_FILES);
}
HRESULT STDMETHODCALLTYPE CShellDispatch::FindComputer()
{
TRACE("(%p)\n", this);
return PostTrayCommand(TRAYCMD_SEARCH_COMPUTERS);
}
HRESULT STDMETHODCALLTYPE CShellDispatch::RefreshMenu()
{
TRACE("(%p)\n", this);
return E_NOTIMPL;
}
HRESULT STDMETHODCALLTYPE CShellDispatch::ControlPanelItem(BSTR szDir)
{
TRACE("(%p, %ls)\n", this, szDir);
return SHRunControlPanel(szDir, NULL) ? S_OK : S_FALSE;
}
// *** IShellDispatch2 methods ***
HRESULT STDMETHODCALLTYPE CShellDispatch::IsRestricted(BSTR group, BSTR restriction, LONG *value)
{
TRACE("(%p, %ls, %ls, %p)\n", this, group, restriction, value);
if (!value)
return E_INVALIDARG;
*value = SHGetRestriction(NULL, group, restriction);
return S_OK;
}
HRESULT STDMETHODCALLTYPE CShellDispatch::ShellExecute(BSTR file, VARIANT v_args, VARIANT v_dir, VARIANT v_op, VARIANT v_show)
{
CComVariant args_str, dir_str, op_str, show_int;
WCHAR *args = NULL, *dir = NULL, *op = NULL;
INT show = SW_SHOW;
HINSTANCE ret;
TRACE("(%s, %s, %s, %s, %s)\n", debugstr_w(file), debugstr_variant(&v_args),
debugstr_variant(&v_dir), debugstr_variant(&v_op), debugstr_variant(&v_show));
args_str.ChangeType(VT_BSTR, &v_args);
if (V_VT(&args_str) == VT_BSTR)
args = V_BSTR(&args_str);
dir_str.ChangeType(VT_BSTR, &v_dir);
if (V_VT(&dir_str) == VT_BSTR)
dir = V_BSTR(&dir_str);
op_str.ChangeType(VT_BSTR, &v_op);
if (V_VT(&op_str) == VT_BSTR)
op = V_BSTR(&op_str);
show_int.ChangeType(VT_I4, &v_show);
if (V_VT(&show_int) == VT_I4)
show = V_I4(&show_int);
ret = ShellExecuteW(NULL, op, file, args, dir, show);
return (ULONG_PTR)ret > 32 ? S_OK : S_FALSE;
}
HRESULT STDMETHODCALLTYPE CShellDispatch::FindPrinter(BSTR name, BSTR location, BSTR model)
{
TRACE("(%p, %ls, %ls, %ls)\n", this, name, location, model);
return E_NOTIMPL;
}
HRESULT STDMETHODCALLTYPE CShellDispatch::GetSystemInformation(BSTR name, VARIANT *ret)
{
TRACE("(%p, %ls, %p)\n", this, name, ret);
if (!lstrcmpiW(name, L"ProcessorArchitecture"))
{
SYSTEM_INFO si;
GetSystemInfo(&si);
V_VT(ret) = VT_I4;
V_UI4(ret) = si.wProcessorArchitecture;
return S_OK;
}
UINT os = 0;
if (!lstrcmpiW(name, L"IsOS_Professional"))
os = OS_PROFESSIONAL;
else if (!lstrcmpiW(name, L"IsOS_Personal"))
os = OS_HOME;
else if (!lstrcmpiW(name, L"IsOS_DomainMember"))
os = OS_DOMAINMEMBER;
if (os)
{
V_VT(ret) = VT_BOOL;
V_BOOL(ret) = IsOS(os) ? VARIANT_TRUE : VARIANT_FALSE;
return S_OK;
}
return E_NOTIMPL;
}
static HRESULT OpenServiceHelper(LPCWSTR name, DWORD access, SC_HANDLE &hSvc)
{
hSvc = NULL;
SC_HANDLE hScm = OpenSCManagerW(NULL, NULL, SC_MANAGER_CONNECT);
if (!hScm)
return HResultFromWin32(GetLastError());
HRESULT hr = S_OK;
hSvc = OpenServiceW(hScm, name, access);
if (!hSvc)
hr = HResultFromWin32(GetLastError());
CloseServiceHandle(hScm);
return hr;
}
static HRESULT SHELL32_ControlService(BSTR name, DWORD control, VARIANT &persistent)
{
BOOL persist = V_VT(&persistent) == VT_BOOL && V_BOOL(&persistent);
DWORD access = persist ? SERVICE_CHANGE_CONFIG : 0;
switch (control)
{
case 0:
access |= SERVICE_START;
break;
case SERVICE_CONTROL_STOP:
access |= SERVICE_STOP;
break;
}
SC_HANDLE hSvc;
HRESULT hr = OpenServiceHelper(name, access, hSvc);
if (SUCCEEDED(hr))
{
BOOL success;
DWORD error, already;
if (control)
{
SERVICE_STATUS ss;
success = ControlService(hSvc, control, &ss);
error = GetLastError();
already = ERROR_SERVICE_NOT_ACTIVE;
}
else
{
success = StartService(hSvc, 0, NULL);
error = GetLastError();
already = ERROR_SERVICE_ALREADY_RUNNING;
}
hr = success ? S_OK : error == already ? S_FALSE : HRESULT_FROM_WIN32(error);
if (SUCCEEDED(hr) && persist)
{
ChangeServiceConfigW(hSvc, SERVICE_NO_CHANGE,
control ? SERVICE_DEMAND_START : SERVICE_AUTO_START,
SERVICE_NO_CHANGE, NULL, NULL, NULL, NULL, NULL, NULL, NULL);
}
CloseServiceHandle(hSvc);
}
return hr;
}
HRESULT STDMETHODCALLTYPE CShellDispatch::ServiceStart(BSTR service, VARIANT persistent, VARIANT *ret)
{
TRACE("(%p, %ls, %s, %p)\n", this, service, wine_dbgstr_variant(&persistent), ret);
HRESULT hr = SHELL32_ControlService(service, 0, persistent);
V_VT(ret) = VT_BOOL;
V_BOOL(ret) = (hr == S_OK ? VARIANT_TRUE : VARIANT_FALSE);
return hr == S_OK ? S_OK : S_FALSE;
}
HRESULT STDMETHODCALLTYPE CShellDispatch::ServiceStop(BSTR service, VARIANT persistent, VARIANT *ret)
{
TRACE("(%p, %ls, %s, %p)\n", this, service, wine_dbgstr_variant(&persistent), ret);
HRESULT hr = SHELL32_ControlService(service, SERVICE_CONTROL_STOP, persistent);
V_VT(ret) = VT_BOOL;
V_BOOL(ret) = (hr == S_OK ? VARIANT_TRUE : VARIANT_FALSE);
return hr == S_OK ? S_OK : S_FALSE;
}
HRESULT STDMETHODCALLTYPE CShellDispatch::IsServiceRunning(BSTR name, VARIANT *running)
{
SERVICE_STATUS_PROCESS status;
SC_HANDLE scm, service;
DWORD dummy;
TRACE("(%s, %p)\n", debugstr_w(name), running);
V_VT(running) = VT_BOOL;
V_BOOL(running) = VARIANT_FALSE;
scm = OpenSCManagerW(NULL, NULL, SC_MANAGER_CONNECT);
if (!scm)
{
ERR("failed to connect to service manager\n");
return S_OK;
}
service = OpenServiceW(scm, name, SERVICE_QUERY_STATUS);
if (!service)
{
ERR("Failed to open service %s (%u)\n", debugstr_w(name), GetLastError());
CloseServiceHandle(scm);
return S_OK;
}
if (!QueryServiceStatusEx(service, SC_STATUS_PROCESS_INFO, (BYTE *)&status,
sizeof(SERVICE_STATUS_PROCESS), &dummy))
{
TRACE("failed to query service status (%u)\n", GetLastError());
CloseServiceHandle(service);
CloseServiceHandle(scm);
return S_OK;
}
if (status.dwCurrentState == SERVICE_RUNNING)
V_BOOL(running) = VARIANT_TRUE;
CloseServiceHandle(service);
CloseServiceHandle(scm);
return S_OK;
}
HRESULT STDMETHODCALLTYPE CShellDispatch::CanStartStopService(BSTR service, VARIANT *ret)
{
TRACE("(%p, %ls, %p)\n", this, service, ret);
SC_HANDLE hSvc;
HRESULT hr = OpenServiceHelper(service, SERVICE_START | SERVICE_STOP, hSvc);
if (SUCCEEDED(hr))
CloseServiceHandle(hSvc);
V_VT(ret) = VT_BOOL;
V_BOOL(ret) = (hr == S_OK ? VARIANT_TRUE : VARIANT_FALSE);
return S_OK;
}
HRESULT STDMETHODCALLTYPE CShellDispatch::ShowBrowserBar(BSTR clsid, VARIANT show, VARIANT *ret)
{
TRACE("(%p, %ls, %s, %p)\n", this, clsid, wine_dbgstr_variant(&show), ret);
return E_NOTIMPL;
}
// *** IShellDispatch3 methods ***
HRESULT STDMETHODCALLTYPE CShellDispatch::AddToRecent(VARIANT file, BSTR category)
{
TRACE("(%p, %s, %ls)\n", this, wine_dbgstr_variant(&file), category);
CComHeapPtr<ITEMIDLIST> idlist;
HRESULT hr = VariantToIdlist(&file, &idlist);
if (hr == S_OK)
SHAddToRecentDocs(SHARD_PIDL, (LPCITEMIDLIST)idlist);
else
hr = S_FALSE;
return hr;
}
// *** IShellDispatch4 methods ***
#define IDM_SECURITY 5001 // From base/shell/explorer/resource.h
HRESULT STDMETHODCALLTYPE CShellDispatch::WindowsSecurity()
{
TRACE("(%p)\n", this);
return PostTrayCommand(IDM_SECURITY);
}
HRESULT STDMETHODCALLTYPE CShellDispatch::ToggleDesktop()
{
TRACE("(%p)\n", this);
return PostTrayCommand(TRAYCMD_TOGGLE_DESKTOP);
}
HRESULT STDMETHODCALLTYPE CShellDispatch::ExplorerPolicy(BSTR policy, VARIANT *value)
{
TRACE("(%p, %ls, %p)\n", this, policy, value);
return E_NOTIMPL;
}
#ifndef SSF_SERVERADMINUI
#define SSF_SERVERADMINUI 4
#endif
HRESULT STDMETHODCALLTYPE CShellDispatch::GetSetting(LONG setting, VARIANT_BOOL *result)
{
TRACE("(%p, %lu, %p)\n", this, setting, result);
int flag = -1;
SHELLSTATE ss = { };
SHGetSetSettings(&ss, setting, FALSE);
switch (setting)
{
case SSF_SHOWALLOBJECTS: flag = ss.fShowAllObjects; break;
case SSF_SHOWEXTENSIONS: flag = ss.fShowExtensions; break;
case SSF_SHOWSYSFILES: flag = ss.fShowSysFiles; break;
case SSF_DONTPRETTYPATH: flag = ss.fDontPrettyPath; break;
case SSF_NOCONFIRMRECYCLE: flag = ss.fNoConfirmRecycle; break;
case SSF_SHOWSUPERHIDDEN: flag = ss.fShowSuperHidden; break;
case SSF_SEPPROCESS: flag = ss.fSepProcess; break;
case SSF_STARTPANELON: flag = ss.fStartPanelOn; break;
case SSF_SERVERADMINUI: flag = IsOS(OS_SERVERADMINUI); break;
}
if (flag >= 0)
{
*result = flag ? VARIANT_TRUE : VARIANT_FALSE;
return S_OK;
}
return S_FALSE;
}
// *** IObjectSafety methods ***
HRESULT STDMETHODCALLTYPE CShellDispatch::GetInterfaceSafetyOptions(REFIID riid, DWORD *pdwSupportedOptions, DWORD *pdwEnabledOptions)
{
TRACE("(%p, %s, %p, %p)\n", this, wine_dbgstr_guid(&riid), pdwSupportedOptions, pdwEnabledOptions);
return E_NOTIMPL;
}
HRESULT STDMETHODCALLTYPE CShellDispatch::SetInterfaceSafetyOptions(REFIID riid, DWORD dwOptionSetMask, DWORD dwEnabledOptions)
{
TRACE("(%p, %s, %lu, %lu)\n", this, wine_dbgstr_guid(&riid), dwOptionSetMask, dwEnabledOptions);
return E_NOTIMPL;
}
// *** IObjectWithSite methods ***
HRESULT STDMETHODCALLTYPE CShellDispatch::SetSite(IUnknown *pUnkSite)
{
TRACE("(%p, %p)\n", this, pUnkSite);
return E_NOTIMPL;
}
HRESULT STDMETHODCALLTYPE CShellDispatch::GetSite(REFIID riid, PVOID *ppvSite)
{
TRACE("(%p, %s, %p)\n", this, wine_dbgstr_guid(&riid), ppvSite);
return E_NOTIMPL;
}
HRESULT WINAPI CShellDispatch_Constructor(REFIID riid, LPVOID * ppvOut)
{
return ShellObjectCreatorInit<CShellDispatch>(riid, ppvOut);
}