Something went wrong. Try again.
Reactos
Something went wrong. Try again.
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835/* * 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);
EXTERN_C HRESULT SHELL32_AssocGetFSDirectoryDescription(PWSTR Buf, UINT cchBuf){ static WCHAR cache[33] = {}; if (!*cache) LoadStringW(shell32_hInstance, IDS_DIRECTORY, cache, _countof(cache)); return StringCchCopyW(Buf, cchBuf, cache);}
static HRESULT GetExtensionDefaultDescription(PCWSTR DotExt, PWSTR Buf, UINT cchBuf){ static WCHAR fmt[33] = {}; if (!*fmt) LoadStringW(shell32_hInstance, IDS_ANY_FILE, fmt, _countof(fmt)); return StringCchPrintfW(Buf, cchBuf, fmt, DotExt);}
static HRESULT SHELL32_AssocGetExtensionDescription(PCWSTR DotExt, PWSTR Buf, UINT cchBuf){ HRESULT hr; if (!DotExt[0] || (!DotExt[1] && DotExt[0] == '.')) { if (SUCCEEDED(hr = GetExtensionDefaultDescription(L"", Buf, cchBuf))) StrTrimW(Buf, L" -"); // Remove the empty %s so we are left with "File" return hr; } HKEY hKey; if (SUCCEEDED(hr = HCR_GetProgIdKeyOfExtension(DotExt, &hKey, TRUE))) { DWORD err = RegLoadMUIStringW(hKey, L"FriendlyTypeName", Buf, cchBuf, NULL, 0, NULL); if (err && hr == S_OK) // ProgId default value fallback (but not if we only have a .ext key) { DWORD cb = cchBuf * sizeof(*Buf); err = RegGetValueW(hKey, NULL, NULL, RRF_RT_REG_SZ, NULL, Buf, &cb); } RegCloseKey(hKey); if (!err) return err; } // No information in the registry, default to "UPPERCASEEXT File" WCHAR ext[MAX_PATH + 33]; if (LCMapStringW(LOCALE_USER_DEFAULT, LCMAP_UPPERCASE, ++DotExt, -1, ext, _countof(ext))) DotExt = ext; return GetExtensionDefaultDescription(DotExt, Buf, cchBuf);}
EXTERN_C HRESULT SHELL32_AssocGetFileDescription(PCWSTR Name, PWSTR Buf, UINT cchBuf){ return SHELL32_AssocGetExtensionDescription(PathFindExtensionW(Name), Buf, cchBuf);}
/************************************************************************** * 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){ 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) { WCHAR *progId; HRESULT hr; LPCWSTR pchDotExt;
if (StrChrW(pszAssoc, L'\\')) { pchDotExt = PathFindExtensionW(pszAssoc); if (pchDotExt && *pchDotExt) pszAssoc = pchDotExt; }
LONG ret = RegOpenKeyExW(HKEY_CLASSES_ROOT, pszAssoc, 0, KEY_READ, &this->hkeySource); if (ret) { return S_OK; }
/* if this is a progid */ if (*pszAssoc != '.' && *pszAssoc != '{') { this->hkeyProgID = this->hkeySource; return S_OK; }
/* if it's not a progid, it's a file extension or clsid */ if (*pszAssoc == '.') { /* for a file extension, the progid is the default value */ hr = this->GetValue(this->hkeySource, NULL, (void**)&progId, NULL); if (FAILED(hr)) return S_OK; } else /* if (*pszAssoc == '{') */ { HKEY progIdKey; /* for a clsid, the progid is the default value of the ProgID subkey */ ret = RegOpenKeyExW(this->hkeySource, L"ProgID", 0, KEY_READ, &progIdKey); if (ret != ERROR_SUCCESS) return S_OK; hr = this->GetValue(progIdKey, NULL, (void**)&progId, NULL); if (FAILED(hr)) return S_OK; RegCloseKey(progIdKey); }
/* open the actual progid key, the one with the shell subkey */ ret = RegOpenKeyExW(HKEY_CLASSES_ROOT, progId, 0, KEY_READ, &this->hkeyProgID); HeapFree(GetProcessHeap(), 0, progId);
return S_OK; } else if (hkeyProgid != NULL) { /* reopen the key so we don't end up closing a key owned by the caller */ RegOpenKeyExW(hkeyProgid, NULL, 0, KEY_READ, &this->hkeyProgID); this->hkeySource = this->hkeyProgID; return S_OK; } else 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_UNEXPECTEDLY(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; 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, L"\\VarFileInfo\\Translation", (LPVOID *)&bufW, &flen)) { UINT i; DWORD *langCodeDesc = (DWORD *)bufW; for (i = 0; i < flen / sizeof(DWORD); i++) { sprintfW(fileDescW, L"\\StringFileInfo\\%04x%04x\\FileDescription", 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: { DWORD size = 0; DWORD ret = RegGetValueW(this->hkeySource, NULL, L"Content Type", 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, L"Content Type", 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: { DWORD ret; DWORD size = 0; ret = RegGetValueW(this->hkeyProgID, L"DefaultIcon", 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(this->hkeyProgID, L"DefaultIcon", 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 = HRESULT_FROM_WIN32(ret); } } else { hr = HRESULT_FROM_WIN32(ret); } return hr; } case ASSOCSTR_SHELLEXTENSION: { WCHAR keypath[ARRAY_SIZE(L"ShellEx\\") + 39], guid[39]; CLSID clsid; HKEY hkey;
hr = CLSIDFromString(pszExtra, &clsid); if (FAILED(hr)) { return hr; } strcpyW(keypath, L"ShellEx\\"); 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){ 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, L"EditFlags", &data, &size); if(FAILED(hres)) { return hres; }
if (!pcbOut) { HeapFree(GetProcessHeap(), 0, data); 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 = SHQueryValueExW(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 = SHQueryValueExW(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;
/* 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, L"shell", 0, KEY_READ, &hkeyShell); RegCloseKey(hkeyFile); } else { ret = RegOpenKeyExW(this->hkeySource, L"shell", 0, KEY_READ, &hkeyShell); } } else { ret = RegOpenKeyExW(this->hkeySource, L"shell", 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)) hr = this->GetValue(hkeyShell, L"open", (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, L"command", 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_UNEXPECTEDLY(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 { *outlen = len = datalen; }
if (len) { memcpy(out, data, len*sizeof(WCHAR)); }
return hr;}