Intravision/reactos
1
0
Fork 0
mirror of https://github.com/reactos/reactos.git synced 2025-08-05 04:33:32 +00:00
Code Issues Projects Releases Packages Wiki Activity

[SHELL32][SHELL32_APITEST] Add SHGetAttributesFromDataObject

Browse source
This commit is contained in:
Mark Jansen 2021-07-23 20:34:02 +02:00
parent 1f44552d36
commit e5ea6041c9
No known key found for this signature in database
GPG key ID: B39240EE84BEAE8B
8 changed files with 503 additions and 14 deletions
Download patch file Download diff file Expand all files Collapse all files

103
dll/win32/shell32/shldataobject.cpp Normal file
View file

@ -0,0 +1,103 @@
/*
* PROJECT: shell32
* LICENSE: LGPL-2.1-or-later (https://spdx.org/licenses/LGPL-2.1-or-later)
* PURPOSE: SHGetAttributesFromDataObject implementation
* COPYRIGHT: Copyright 2021 Mark Jansen <mark.jansen@reactos.org>
*/
#include "precomp.h"
WINE_DEFAULT_DEBUG_CHANNEL(shell);
static CLIPFORMAT g_DataObjectAttributes = 0;
static const DWORD dwDefaultAttributeMask = SFGAO_CANCOPY | SFGAO_CANMOVE | SFGAO_STORAGE | SFGAO_CANRENAME |
SFGAO_CANDELETE | SFGAO_READONLY | SFGAO_STREAM | SFGAO_FOLDER;
struct DataObjectAttributes
{
DWORD dwMask;
DWORD dwAttributes;
UINT cItems;
};
static_assert(sizeof(DataObjectAttributes) == 0xc, "Unexpected struct size!");
static
HRESULT _BindToObject(PCUIDLIST_ABSOLUTE pidl, CComPtr<IShellFolder>& spFolder)
{
CComPtr<IShellFolder> spDesktop;
HRESULT hr = SHGetDesktopFolder(&spDesktop);
if (FAILED(hr))
return hr;
return spDesktop->BindToObject(pidl, NULL, IID_PPV_ARG(IShellFolder, &spFolder));
}
EXTERN_C
HRESULT WINAPI SHGetAttributesFromDataObject(IDataObject* pDataObject, DWORD dwAttributeMask, DWORD* pdwAttributes, UINT* pcItems)
{
DWORD dwAttributes = 0;
DWORD cItems = 0;
HRESULT hr = S_OK;
TRACE("(%p, 0x%x, %p, %p)\n", pDataObject, dwAttributeMask, pdwAttributes, pcItems);
if (!g_DataObjectAttributes)
g_DataObjectAttributes = (CLIPFORMAT)RegisterClipboardFormatW(L"DataObjectAttributes");
if (pDataObject)
{
DataObjectAttributes data = {};
if (FAILED(DataObject_GetData(pDataObject, g_DataObjectAttributes, &data, sizeof(data))))
{
TRACE("No attributes yet, creating new\n");
memset(&data, 0, sizeof(data));
}
DWORD dwQueryAttributes = dwAttributeMask | dwDefaultAttributeMask;
if ((data.dwMask & dwQueryAttributes) != dwQueryAttributes)
{
CDataObjectHIDA hida(pDataObject);
CComPtr<IShellFolder> spFolder;
if (!FAILED_UNEXPECTEDLY(hr = hida.hr()) &&
!FAILED_UNEXPECTEDLY(hr = _BindToObject(HIDA_GetPIDLFolder(hida), spFolder)))
{
CSimpleArray<PCUIDLIST_RELATIVE> apidl;
for (UINT n = 0; n < hida->cidl; ++n)
{
apidl.Add(HIDA_GetPIDLItem(hida, n));
}
SFGAOF rgfInOut = dwQueryAttributes;
hr = spFolder->GetAttributesOf(apidl.GetSize(), apidl.GetData(), &rgfInOut);
if (!FAILED_UNEXPECTEDLY(hr))
{
data.dwMask = dwQueryAttributes;
// Only store what we asked for
data.dwAttributes = rgfInOut & dwQueryAttributes;
data.cItems = apidl.GetSize();
hr = DataObject_SetData(pDataObject, g_DataObjectAttributes, &data, sizeof(data));
FAILED_UNEXPECTEDLY(hr);
}
}
}
// Only give the user what they asked for, not everything else we have!
dwAttributes = data.dwAttributes & dwAttributeMask;
cItems = data.cItems;
}
if (pdwAttributes)
*pdwAttributes = dwAttributes;
if (pcItems)
*pcItems = cItems;
return hr;
}