[dquintana] 66940: [SHELL32] * Implement CLSID_QueryAssociations class. Patch by Ivan Rodionov, with some unrotting by me. CORE-8188 #resolve
Author: dquintana Date: Sat Mar 28 23:27:10 2015 New Revision: 66940
URL: http://svn.reactos.org/svn/reactos?rev=66940&view=rev Log: [SHELL32] * Implement CLSID_QueryAssociations class. Patch by Ivan Rodionov, with some unrotting by me. CORE-8188 #resolve
Added: trunk/reactos/dll/win32/shell32/CQueryAssociations.cpp (with props) trunk/reactos/dll/win32/shell32/CQueryAssociations.h (with props) Modified: trunk/reactos/dll/win32/shell32/CMakeLists.txt trunk/reactos/dll/win32/shell32/precomp.h trunk/reactos/dll/win32/shell32/rgs_res.rc trunk/reactos/dll/win32/shell32/shell32.cpp trunk/reactos/dll/win32/shell32/shresdef.h
Modified: trunk/reactos/dll/win32/shell32/CMakeLists.txt URL: http://svn.reactos.org/svn/reactos/trunk/reactos/dll/win32/shell32/CMakeList... ============================================================================== --- trunk/reactos/dll/win32/shell32/CMakeLists.txt [iso-8859-1] (original) +++ trunk/reactos/dll/win32/shell32/CMakeLists.txt [iso-8859-1] Sat Mar 28 23:27:10 2015 @@ -19,6 +19,7 @@ CMenuBand.cpp CMenuDeskBar.cpp CIDLDataObj.cpp + CQueryAssociations.cpp dde.cpp debughlp.cpp CDesktopBrowser.cpp
Added: trunk/reactos/dll/win32/shell32/CQueryAssociations.cpp URL: http://svn.reactos.org/svn/reactos/trunk/reactos/dll/win32/shell32/CQueryAss... ============================================================================== --- trunk/reactos/dll/win32/shell32/CQueryAssociations.cpp (added) +++ trunk/reactos/dll/win32/shell32/CQueryAssociations.cpp [iso-8859-1] Sat Mar 28 23:27:10 2015 @@ -0,0 +1,771 @@ +/* + * IQueryAssociations object and helper functions + * + * Copyright 2002 Jon Griffiths + * + * This library is free software; you can redistribute it and/or + * modify it under the terms of the GNU Lesser General Public + * License as published by the Free Software Foundation; either + * version 2.1 of the License, or (at your option) any later version. + * + * This library is distributed in the hope that it will be useful, + * but WITHOUT ANY WARRANTY; without even the implied warranty of + * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU + * Lesser General Public License for more details. + * + * You should have received a copy of the GNU Lesser General Public + * License along with this library; if not, write to the Free Software + * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA + */ + +#include "precomp.h" + +WINE_DEFAULT_DEBUG_CHANNEL(shell); + +/************************************************************************** + * IQueryAssociations + * + * DESCRIPTION + * This object provides a layer of abstraction over the system registry in + * order to simplify the process of parsing associations between files. + * Associations in this context means the registry entries that link (for + * example) the extension of a file with its description, list of + * applications to open the file with, and actions that can be performed on it + * (the shell displays such information in the context menu of explorer + * when you right-click on a file). + * + * HELPERS + * You can use this object transparently by calling the helper functions + * AssocQueryKeyA(), AssocQueryStringA() and AssocQueryStringByKeyA(). These + * create an IQueryAssociations object, perform the requested actions + * and then dispose of the object. Alternatively, you can create an instance + * of the object using AssocCreate() and call the following methods on it: + * + * METHODS + */ + +CQueryAssociations::CQueryAssociations() : hkeySource(0), hkeyProgID(0) +{ +} + +CQueryAssociations::~CQueryAssociations() +{ +} + +/************************************************************************** + * IQueryAssociations_Init + * + * Initialise an IQueryAssociations object. + * + * PARAMS + * cfFlags [I] ASSOCF_ flags from "shlwapi.h" + * pszAssoc [I] String for the root key name, or NULL if hkeyProgid is given + * hkeyProgid [I] Handle for the root key, or NULL if pszAssoc is given + * hWnd [I] Reserved, must be NULL. + * + * RETURNS + * Success: S_OK. iface is initialised with the parameters given. + * Failure: An HRESULT error code indicating the error. + */ +HRESULT STDMETHODCALLTYPE CQueryAssociations::Init( + ASSOCF cfFlags, + LPCWSTR pszAssoc, + HKEY hkeyProgid, + HWND hWnd) +{ + static const WCHAR szProgID[] = L"ProgID"; + + TRACE("(%p)->(%d,%s,%p,%p)\n", this, + cfFlags, + debugstr_w(pszAssoc), + hkeyProgid, + hWnd); + + if (hWnd != NULL) + { + FIXME("hwnd != NULL not supported\n"); + } + + if (cfFlags != 0) + { + FIXME("unsupported flags: %x\n", cfFlags); + } + + RegCloseKey(this->hkeySource); + RegCloseKey(this->hkeyProgID); + this->hkeySource = this->hkeyProgID = NULL; + if (pszAssoc != NULL) + { + LONG ret = RegOpenKeyExW(HKEY_CLASSES_ROOT, + pszAssoc, + 0, + KEY_READ, + &this->hkeySource); + if (ret) + { + return S_OK; + } + /* if this is not a prog id */ + if ((*pszAssoc == '.') || (*pszAssoc == '{')) + { + RegOpenKeyExW(this->hkeySource, + szProgID, + 0, + KEY_READ, + &this->hkeyProgID); + } + else + { + this->hkeyProgID = this->hkeySource; + } + + return S_OK; + } + else if (hkeyProgid != NULL) + { + this->hkeyProgID = hkeyProgid; + return S_OK; + } + + return E_INVALIDARG; +} + +/************************************************************************** + * IQueryAssociations_GetString + * + * Get a file association string from the registry. + * + * PARAMS + * cfFlags [I] ASSOCF_ flags from "shlwapi.h" + * str [I] Type of string to get (ASSOCSTR enum from "shlwapi.h") + * pszExtra [I] Extra information about the string location + * pszOut [O] Destination for the association string + * pcchOut [I/O] Length of pszOut + * + * RETURNS + * Success: S_OK. pszOut contains the string, pcchOut contains its length. + * Failure: An HRESULT error code indicating the error. + */ +HRESULT STDMETHODCALLTYPE CQueryAssociations::GetString( + ASSOCF flags, + ASSOCSTR str, + LPCWSTR pszExtra, + LPWSTR pszOut, + DWORD *pcchOut) +{ + const ASSOCF unimplemented_flags = ~ASSOCF_NOTRUNCATE; + DWORD len = 0; + HRESULT hr; + WCHAR path[MAX_PATH]; + + TRACE("(%p)->(0x%08x, %u, %s, %p, %p)\n", this, flags, str, debugstr_w(pszExtra), pszOut, pcchOut); + if (flags & unimplemented_flags) + { + FIXME("%08x: unimplemented flags\n", flags & unimplemented_flags); + } + + if (!pcchOut) + { + return E_UNEXPECTED; + } + + if (!this->hkeySource && !this->hkeyProgID) + { + return HRESULT_FROM_WIN32(ERROR_NO_ASSOCIATION); + } + + switch (str) + { + case ASSOCSTR_COMMAND: + { + WCHAR *command; + hr = this->GetCommand(pszExtra, &command); + if (SUCCEEDED(hr)) + { + hr = this->ReturnString(flags, pszOut, pcchOut, command, strlenW(command) + 1); + HeapFree(GetProcessHeap(), 0, command); + } + return hr; + } + case ASSOCSTR_EXECUTABLE: + { + hr = this->GetExecutable(pszExtra, path, MAX_PATH, &len); + if (FAILED(hr)) + { + return hr; + } + len++; + return this->ReturnString(flags, pszOut, pcchOut, path, len); + } + case ASSOCSTR_FRIENDLYDOCNAME: + { + WCHAR *pszFileType; + + hr = this->GetValue(this->hkeySource, NULL, (void**)&pszFileType, NULL); + if (FAILED(hr)) + { + return hr; + } + DWORD size = 0; + DWORD ret = RegGetValueW(HKEY_CLASSES_ROOT, pszFileType, NULL, RRF_RT_REG_SZ, NULL, NULL, &size); + if (ret == ERROR_SUCCESS) + { + WCHAR *docName = static_cast<WCHAR *>(HeapAlloc(GetProcessHeap(), 0, size)); + if (docName) + { + ret = RegGetValueW(HKEY_CLASSES_ROOT, pszFileType, NULL, RRF_RT_REG_SZ, NULL, docName, &size); + if (ret == ERROR_SUCCESS) + { + hr = this->ReturnString(flags, pszOut, pcchOut, docName, strlenW(docName) + 1); + } + else + { + hr = HRESULT_FROM_WIN32(ret); + } + HeapFree(GetProcessHeap(), 0, docName); + } + else + { + hr = E_OUTOFMEMORY; + } + } + else + { + hr = HRESULT_FROM_WIN32(ret); + } + HeapFree(GetProcessHeap(), 0, pszFileType); + return hr; + } + case ASSOCSTR_FRIENDLYAPPNAME: + { + PVOID verinfoW = NULL; + DWORD size, retval = 0; + UINT flen; + WCHAR *bufW; + static const WCHAR translationW[] = L"\VarFileInfo\Translation"; + static const WCHAR fileDescFmtW[] = L"\StringFileInfo\%04x%04x\FileDescription"; + WCHAR fileDescW[41]; + + hr = this->GetExecutable(pszExtra, path, MAX_PATH, &len); + if (FAILED(hr)) + { + return hr; + } + retval = GetFileVersionInfoSizeW(path, &size); + if (!retval) + { + goto get_friendly_name_fail; + } + verinfoW = HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, retval); + if (!verinfoW) + { + return E_OUTOFMEMORY; + } + if (!GetFileVersionInfoW(path, 0, retval, verinfoW)) + { + goto get_friendly_name_fail; + } + if (VerQueryValueW(verinfoW, translationW, (LPVOID *)&bufW, &flen)) + { + UINT i; + DWORD *langCodeDesc = (DWORD *)bufW; + for (i = 0; i < flen / sizeof(DWORD); i++) + { + sprintfW(fileDescW, fileDescFmtW, LOWORD(langCodeDesc[i]), HIWORD(langCodeDesc[i])); + if (VerQueryValueW(verinfoW, fileDescW, (LPVOID *)&bufW, &flen)) + { + /* Does strlenW(bufW) == 0 mean we use the filename? */ + len = strlenW(bufW) + 1; + TRACE("found FileDescription: %s\n", debugstr_w(bufW)); + hr = this->ReturnString(flags, pszOut, pcchOut, bufW, len); + HeapFree(GetProcessHeap(), 0, verinfoW); + return hr; + } + } + } + get_friendly_name_fail: + PathRemoveExtensionW(path); + PathStripPathW(path); + TRACE("using filename: %s\n", debugstr_w(path)); + hr = this->ReturnString(flags, pszOut, pcchOut, path, strlenW(path) + 1); + HeapFree(GetProcessHeap(), 0, verinfoW); + return hr; + } + case ASSOCSTR_CONTENTTYPE: + { + static const WCHAR Content_TypeW[] = L"Content Type"; + + DWORD size = 0; + DWORD ret = RegGetValueW(this->hkeySource, NULL, Content_TypeW, RRF_RT_REG_SZ, NULL, NULL, &size); + if (ret != ERROR_SUCCESS) + { + return HRESULT_FROM_WIN32(ret); + } + WCHAR *contentType = static_cast<WCHAR *>(HeapAlloc(GetProcessHeap(), 0, size)); + if (contentType != NULL) + { + ret = RegGetValueW(this->hkeySource, NULL, Content_TypeW, RRF_RT_REG_SZ, NULL, contentType, &size); + if (ret == ERROR_SUCCESS) + { + hr = this->ReturnString(flags, pszOut, pcchOut, contentType, strlenW(contentType) + 1); + } + else + { + hr = HRESULT_FROM_WIN32(ret); + } + HeapFree(GetProcessHeap(), 0, contentType); + } + else + { + hr = E_OUTOFMEMORY; + } + return hr; + } + case ASSOCSTR_DEFAULTICON: + { + static const WCHAR DefaultIconW[] = L"DefaultIcon"; + WCHAR *pszFileType; + HKEY hkeyFile; + + hr = this->GetValue(this->hkeySource, NULL, (void**)&pszFileType, NULL); + if (FAILED(hr)) + { + return hr; + } + DWORD ret = RegOpenKeyExW(HKEY_CLASSES_ROOT, pszFileType, 0, KEY_READ, &hkeyFile); + if (ret == ERROR_SUCCESS) + { + DWORD size = 0; + ret = RegGetValueW(hkeyFile, DefaultIconW, NULL, RRF_RT_REG_SZ, NULL, NULL, &size); + if (ret == ERROR_SUCCESS) + { + WCHAR *icon = static_cast<WCHAR *>(HeapAlloc(GetProcessHeap(), 0, size)); + if (icon) + { + ret = RegGetValueW(hkeyFile, DefaultIconW, NULL, RRF_RT_REG_SZ, NULL, icon, &size); + if (ret == ERROR_SUCCESS) + { + hr = this->ReturnString(flags, pszOut, pcchOut, icon, strlenW(icon) + 1); + } + else + { + hr = HRESULT_FROM_WIN32(ret); + } + HeapFree(GetProcessHeap(), 0, icon); + } + else + { + hr = E_OUTOFMEMORY; + } + } + else + { + hr = HRESULT_FROM_WIN32(ret); + } + RegCloseKey(hkeyFile); + } + else + { + hr = HRESULT_FROM_WIN32(ret); + } + HeapFree(GetProcessHeap(), 0, pszFileType); + return hr; + } + case ASSOCSTR_SHELLEXTENSION: + { + static const WCHAR shellexW[] = L"ShellEx\"; + WCHAR keypath[sizeof(shellexW) / sizeof(shellexW[0]) + 39], guid[39]; + CLSID clsid; + HKEY hkey; + + hr = CLSIDFromString(pszExtra, &clsid); + if (FAILED(hr)) + { + return hr; + } + strcpyW(keypath, shellexW); + strcatW(keypath, pszExtra); + LONG ret = RegOpenKeyExW(this->hkeySource, keypath, 0, KEY_READ, &hkey); + if (ret) + { + return HRESULT_FROM_WIN32(ret); + } + DWORD size = sizeof(guid); + ret = RegGetValueW(hkey, NULL, NULL, RRF_RT_REG_SZ, NULL, guid, &size); + RegCloseKey(hkey); + if (ret) + { + return HRESULT_FROM_WIN32(ret); + } + return this->ReturnString(flags, pszOut, pcchOut, guid, size / sizeof(WCHAR)); + } + + default: + { + FIXME("assocstr %d unimplemented!\n", str); + return E_NOTIMPL; + } + } +} + +/************************************************************************** + * IQueryAssociations_GetKey + * + * Get a file association key from the registry. + * + * PARAMS + * cfFlags [I] ASSOCF_ flags from "shlwapi.h" + * assockey [I] Type of key to get (ASSOCKEY enum from "shlwapi.h") + * pszExtra [I] Extra information about the key location + * phkeyOut [O] Destination for the association key + * + * RETURNS + * Success: S_OK. phkeyOut contains a handle to the key. + * Failure: An HRESULT error code indicating the error. + */ +HRESULT STDMETHODCALLTYPE CQueryAssociations::GetKey( + ASSOCF cfFlags, + ASSOCKEY assockey, + LPCWSTR pszExtra, + HKEY *phkeyOut) +{ + FIXME("(%p,0x%8x,0x%8x,%s,%p)-stub!\n", this, cfFlags, assockey, + debugstr_w(pszExtra), phkeyOut); + return E_NOTIMPL; +} + +/************************************************************************** + * IQueryAssociations_GetData + * + * Get the data for a file association key from the registry. + * + * PARAMS + * cfFlags [I] ASSOCF_ flags from "shlwapi.h" + * assocdata [I] Type of data to get (ASSOCDATA enum from "shlwapi.h") + * pszExtra [I] Extra information about the data location + * pvOut [O] Destination for the association key + * pcbOut [I/O] Size of pvOut + * + * RETURNS + * Success: S_OK. pszOut contains the data, pcbOut contains its length. + * Failure: An HRESULT error code indicating the error. + */ +HRESULT STDMETHODCALLTYPE CQueryAssociations::GetData(ASSOCF cfFlags, ASSOCDATA assocdata, LPCWSTR pszExtra, LPVOID pvOut, DWORD *pcbOut) +{ + static const WCHAR edit_flags[] = L"EditFlags"; + + TRACE("(%p,0x%8x,0x%8x,%s,%p,%p)\n", this, cfFlags, assocdata, + debugstr_w(pszExtra), pvOut, pcbOut); + + if(cfFlags) + { + FIXME("Unsupported flags: %x\n", cfFlags); + } + + switch(assocdata) + { + case ASSOCDATA_EDITFLAGS: + { + if(!this->hkeyProgID) + { + return HRESULT_FROM_WIN32(ERROR_NO_ASSOCIATION); + } + + void *data; + DWORD size; + HRESULT hres = this->GetValue(this->hkeyProgID, edit_flags, &data, &size); + if(FAILED(hres) || !pcbOut) + { + return hres; + } + + hres = this->ReturnData(pvOut, pcbOut, data, size); + HeapFree(GetProcessHeap(), 0, data); + return hres; + } + default: + { + FIXME("Unsupported ASSOCDATA value: %d\n", assocdata); + return E_NOTIMPL; + } + } +} + +/************************************************************************** + * IQueryAssociations_GetEnum + * + * Not yet implemented in native Win32. + * + * PARAMS + * cfFlags [I] ASSOCF_ flags from "shlwapi.h" + * assocenum [I] Type of enum to get (ASSOCENUM enum from "shlwapi.h") + * pszExtra [I] Extra information about the enum location + * riid [I] REFIID to look for + * ppvOut [O] Destination for the interface. + * + * RETURNS + * Success: S_OK. + * Failure: An HRESULT error code indicating the error. + * + * NOTES + * Presumably this function returns an enumerator object. + */ +HRESULT STDMETHODCALLTYPE CQueryAssociations::GetEnum( + ASSOCF cfFlags, + ASSOCENUM assocenum, + LPCWSTR pszExtra, + REFIID riid, + LPVOID *ppvOut) +{ + return E_NOTIMPL; +} + +HRESULT CQueryAssociations::GetValue(HKEY hkey, const WCHAR *name, void **data, DWORD *data_size) +{ + DWORD size; + LONG ret; + + ret = RegQueryValueExW(hkey, name, 0, NULL, NULL, &size); + if (ret != ERROR_SUCCESS) + { + return HRESULT_FROM_WIN32(ret); + } + if (!size) + { + return E_FAIL; + } + *data = HeapAlloc(GetProcessHeap(), 0, size); + if (!*data) + { + return E_OUTOFMEMORY; + } + ret = RegQueryValueExW(hkey, name, 0, NULL, (LPBYTE)*data, &size); + if (ret != ERROR_SUCCESS) + { + HeapFree(GetProcessHeap(), 0, *data); + return HRESULT_FROM_WIN32(ret); + } + if(data_size) + { + *data_size = size; + } + return S_OK; +} + +HRESULT CQueryAssociations::GetCommand(const WCHAR *extra, WCHAR **command) +{ + HKEY hkeyCommand; + HKEY hkeyShell; + HKEY hkeyVerb; + HRESULT hr; + LONG ret; + WCHAR *extra_from_reg = NULL; + WCHAR *filetype; + static const WCHAR commandW[] = L"command"; + static const WCHAR shellW[] = L"shell"; + + /* When looking for file extension it's possible to have a default value + that points to another key that contains 'shell/<verb>/command' subtree. */ + hr = this->GetValue(this->hkeySource, NULL, (void**)&filetype, NULL); + if (hr == S_OK) + { + HKEY hkeyFile; + + ret = RegOpenKeyExW(HKEY_CLASSES_ROOT, filetype, 0, KEY_READ, &hkeyFile); + HeapFree(GetProcessHeap(), 0, filetype); + + if (ret == ERROR_SUCCESS) + { + ret = RegOpenKeyExW(hkeyFile, shellW, 0, KEY_READ, &hkeyShell); + RegCloseKey(hkeyFile); + } + else + { + ret = RegOpenKeyExW(this->hkeySource, shellW, 0, KEY_READ, &hkeyShell); + } + } + else + { + ret = RegOpenKeyExW(this->hkeySource, shellW, 0, KEY_READ, &hkeyShell); + } + + if (ret) + { + return HRESULT_FROM_WIN32(ret); + } + + if (!extra) + { + /* check for default verb */ + hr = this->GetValue(hkeyShell, NULL, (void**)&extra_from_reg, NULL); + if (FAILED(hr)) + { + /* no default verb, try first subkey */ + DWORD max_subkey_len; + + ret = RegQueryInfoKeyW(hkeyShell, NULL, NULL, NULL, NULL, &max_subkey_len, NULL, NULL, NULL, NULL, NULL, NULL); + if (ret) + { + RegCloseKey(hkeyShell); + return HRESULT_FROM_WIN32(ret); + } + + max_subkey_len++; + extra_from_reg = static_cast<WCHAR*>(HeapAlloc(GetProcessHeap(), 0, max_subkey_len * sizeof(WCHAR))); + if (!extra_from_reg) + { + RegCloseKey(hkeyShell); + return E_OUTOFMEMORY; + } + + ret = RegEnumKeyExW(hkeyShell, 0, extra_from_reg, &max_subkey_len, NULL, NULL, NULL, NULL); + if (ret) + { + HeapFree(GetProcessHeap(), 0, extra_from_reg); + RegCloseKey(hkeyShell); + return HRESULT_FROM_WIN32(ret); + } + } + extra = extra_from_reg; + } + + /* open verb subkey */ + ret = RegOpenKeyExW(hkeyShell, extra, 0, KEY_READ, &hkeyVerb); + HeapFree(GetProcessHeap(), 0, extra_from_reg); + RegCloseKey(hkeyShell); + if (ret) + { + return HRESULT_FROM_WIN32(ret); + } + /* open command subkey */ + ret = RegOpenKeyExW(hkeyVerb, commandW, 0, KEY_READ, &hkeyCommand); + RegCloseKey(hkeyVerb); + if (ret) + { + return HRESULT_FROM_WIN32(ret); + } + hr = this->GetValue(hkeyCommand, NULL, (void**)command, NULL); + RegCloseKey(hkeyCommand); + return hr; +} + +HRESULT CQueryAssociations::GetExecutable(LPCWSTR pszExtra, LPWSTR path, DWORD pathlen, DWORD *len) +{ + WCHAR *pszCommand; + WCHAR *pszStart; + WCHAR *pszEnd; + + HRESULT hr = this->GetCommand(pszExtra, &pszCommand); + if (FAILED(hr)) + { + return hr; + } + + DWORD expLen = ExpandEnvironmentStringsW(pszCommand, NULL, 0); + if (expLen > 0) + { + expLen++; + WCHAR *buf = static_cast<WCHAR *>(HeapAlloc(GetProcessHeap(), 0, expLen * sizeof(WCHAR))); + ExpandEnvironmentStringsW(pszCommand, buf, expLen); + HeapFree(GetProcessHeap(), 0, pszCommand); + pszCommand = buf; + } + + /* cleanup pszCommand */ + if (pszCommand[0] == '"') + { + pszStart = pszCommand + 1; + pszEnd = strchrW(pszStart, '"'); + if (pszEnd) + { + *pszEnd = 0; + } + *len = SearchPathW(NULL, pszStart, NULL, pathlen, path, NULL); + } + else + { + pszStart = pszCommand; + for (pszEnd = pszStart; (pszEnd = strchrW(pszEnd, ' ')); pszEnd++) + { + WCHAR c = *pszEnd; + *pszEnd = 0; + if ((*len = SearchPathW(NULL, pszStart, NULL, pathlen, path, NULL))) + { + break; + } + *pszEnd = c; + } + if (!pszEnd) + { + *len = SearchPathW(NULL, pszStart, NULL, pathlen, path, NULL); + } + } + + HeapFree(GetProcessHeap(), 0, pszCommand); + if (!*len) + { + return HRESULT_FROM_WIN32(ERROR_FILE_NOT_FOUND); + } + return S_OK; +} + +HRESULT CQueryAssociations::ReturnData(void *out, DWORD *outlen, const void *data, DWORD datalen) +{ + if (out) + { + if (*outlen < datalen) + { + *outlen = datalen; + return E_POINTER; + } + *outlen = datalen; + memcpy(out, data, datalen); + return S_OK; + } + else + { + *outlen = datalen; + return S_FALSE; + } +} + +HRESULT CQueryAssociations::ReturnString(ASSOCF flags, LPWSTR out, DWORD *outlen, LPCWSTR data, DWORD datalen) +{ + HRESULT hr = S_OK; + DWORD len; + + TRACE("flags=0x%08x, data=%s\n", flags, debugstr_w(data)); + + if (!out) + { + *outlen = datalen; + return S_FALSE; + } + + if (*outlen < datalen) + { + if (flags & ASSOCF_NOTRUNCATE) + { + len = 0; + if (*outlen > 0) out[0] = 0; + hr = E_POINTER; + } + else + { + len = min(*outlen, datalen); + hr = HRESULT_FROM_WIN32(ERROR_INSUFFICIENT_BUFFER); + } + *outlen = datalen; + } + else + { + len = datalen; + } + + if (len) + { + memcpy(out, data, len*sizeof(WCHAR)); + } + + return hr; +}
Propchange: trunk/reactos/dll/win32/shell32/CQueryAssociations.cpp ------------------------------------------------------------------------------ svn:eol-style = native
Added: trunk/reactos/dll/win32/shell32/CQueryAssociations.h URL: http://svn.reactos.org/svn/reactos/trunk/reactos/dll/win32/shell32/CQueryAss... ============================================================================== --- trunk/reactos/dll/win32/shell32/CQueryAssociations.h (added) +++ trunk/reactos/dll/win32/shell32/CQueryAssociations.h [iso-8859-1] Sat Mar 28 23:27:10 2015 @@ -0,0 +1,36 @@ +#pragma once + +class CQueryAssociations : + public CComCoClass<CQueryAssociations, &CLSID_QueryAssociations>, + public CComObjectRootEx<CComMultiThreadModelNoCS>, + public IQueryAssociations +{ +private: + HKEY hkeySource; + HKEY hkeyProgID; + + HRESULT GetValue(HKEY hkey, const WCHAR *name, void **data, DWORD *data_size); + HRESULT GetCommand(const WCHAR *extra, WCHAR **command); + HRESULT GetExecutable(LPCWSTR pszExtra, LPWSTR path, DWORD pathlen, DWORD *len); + HRESULT ReturnData(void *out, DWORD *outlen, const void *data, DWORD datalen); + HRESULT ReturnString(ASSOCF flags, LPWSTR out, DWORD *outlen, LPCWSTR data, DWORD datalen); + +public: + CQueryAssociations(); + ~CQueryAssociations(); + + // *** IQueryAssociations methods *** + virtual HRESULT STDMETHODCALLTYPE Init(ASSOCF flags, LPCWSTR pwszAssoc, HKEY hkProgid, HWND hwnd); + virtual HRESULT STDMETHODCALLTYPE GetString(ASSOCF flags, ASSOCSTR str, LPCWSTR pwszExtra, LPWSTR pwszOut, DWORD *pcchOut); + virtual HRESULT STDMETHODCALLTYPE GetKey(ASSOCF flags, ASSOCKEY key, LPCWSTR pwszExtra, HKEY *phkeyOut); + virtual HRESULT STDMETHODCALLTYPE GetData(ASSOCF flags, ASSOCDATA data, LPCWSTR pwszExtra, void *pvOut, DWORD *pcbOut); + virtual HRESULT STDMETHODCALLTYPE GetEnum(ASSOCF cfFlags, ASSOCENUM assocenum, LPCWSTR pszExtra, REFIID riid, LPVOID *ppvOut); + +DECLARE_REGISTRY_RESOURCEID(IDR_QUERYASSOCIATIONS) +DECLARE_NOT_AGGREGATABLE(CQueryAssociations) +DECLARE_PROTECT_FINAL_CONSTRUCT() + +BEGIN_COM_MAP(CQueryAssociations) + COM_INTERFACE_ENTRY_IID(IID_IQueryAssociations, IQueryAssociations) +END_COM_MAP() +};
Propchange: trunk/reactos/dll/win32/shell32/CQueryAssociations.h ------------------------------------------------------------------------------ svn:eol-style = native
Modified: trunk/reactos/dll/win32/shell32/precomp.h URL: http://svn.reactos.org/svn/reactos/trunk/reactos/dll/win32/shell32/precomp.h... ============================================================================== --- trunk/reactos/dll/win32/shell32/precomp.h [iso-8859-1] (original) +++ trunk/reactos/dll/win32/shell32/precomp.h [iso-8859-1] Sat Mar 28 23:27:10 2015 @@ -66,6 +66,7 @@ #include "drvdefext.h" #include "CMenuBand.h" #include "CMenuDeskBar.h" +#include "CQueryAssociations.h"
#include <wine/debug.h> #include <wine/unicode.h>
Modified: trunk/reactos/dll/win32/shell32/rgs_res.rc URL: http://svn.reactos.org/svn/reactos/trunk/reactos/dll/win32/shell32/rgs_res.r... ============================================================================== --- trunk/reactos/dll/win32/shell32/rgs_res.rc [iso-8859-1] (original) +++ trunk/reactos/dll/win32/shell32/rgs_res.rc [iso-8859-1] Sat Mar 28 23:27:10 2015 @@ -14,6 +14,7 @@ IDR_NETWORKPLACES REGISTRY "res/rgs/networkplaces.rgs" IDR_NEWMENU REGISTRY "res/rgs/newmenu.rgs" IDR_PRINTERS REGISTRY "res/rgs/printers.rgs" +IDR_QUERYASSOCIATIONS REGISTRY "res/rgs/queryassociations.rgs" IDR_RECYCLEBIN REGISTRY "res/rgs/recyclebin.rgs" IDR_SHELLDESKTOP REGISTRY "res/rgs/shelldesktop.rgs" IDR_SHELLFSFOLDER REGISTRY "res/rgs/shellfsfolder.rgs"
Modified: trunk/reactos/dll/win32/shell32/shell32.cpp URL: http://svn.reactos.org/svn/reactos/trunk/reactos/dll/win32/shell32/shell32.c... ============================================================================== --- trunk/reactos/dll/win32/shell32/shell32.cpp [iso-8859-1] (original) +++ trunk/reactos/dll/win32/shell32/shell32.cpp [iso-8859-1] Sat Mar 28 23:27:10 2015 @@ -173,6 +173,7 @@ OBJECT_ENTRY(CLSID_MenuBand, CMenuBand) OBJECT_ENTRY(CLSID_MenuDeskBar, CMenuDeskBar) OBJECT_ENTRY(CLSID_ExeDropHandler, CExeDropHandler) + OBJECT_ENTRY(CLSID_QueryAssociations, CQueryAssociations) END_OBJECT_MAP()
CShell32Module gModule;
Modified: trunk/reactos/dll/win32/shell32/shresdef.h URL: http://svn.reactos.org/svn/reactos/trunk/reactos/dll/win32/shell32/shresdef.... ============================================================================== --- trunk/reactos/dll/win32/shell32/shresdef.h [iso-8859-1] (original) +++ trunk/reactos/dll/win32/shell32/shresdef.h [iso-8859-1] Sat Mar 28 23:27:10 2015 @@ -565,3 +565,4 @@ #define IDR_MENUBAND 149 #define IDR_MENUDESKBAR 150 #define IDR_EXEDROPHANDLER 151 +#define IDR_QUERYASSOCIATIONS 152
-
dquintana＠svn.reactos.org