HRESULT UrlCombineUnescape(LPCWSTR pszBase, LPCWSTR pszRelative, LPWSTR pszCombined, LPDWORD pcchCombined, DWORD dwFlags)
{
HRESULT hr = S_OK;
DWORD dwSize;
LPWSTR pwzCombined = NULL;
LPWSTR pwzFileCombined = NULL;
pwzCombined = NEW(WCHAR[MAX_URL_LENGTH]);
if (!pwzCombined) {
hr = E_OUTOFMEMORY;
goto Exit;
}
// If we're just combining an absolute file path to an relative file
// path, do this by concatenating the strings, and canonicalizing it.
// This avoids UrlCombine randomness where you could end up with
// a partially escaped (and partially unescaped) resulting URL!
if (!PathIsURLW(pszBase) && PathIsRelativeWrap(pszRelative)) {
pwzFileCombined = NEW(WCHAR[MAX_URL_LENGTH]);
if (!pwzFileCombined) {
hr = E_OUTOFMEMORY;
goto Exit;
}
wnsprintfW(pwzFileCombined, MAX_URL_LENGTH, L"%ws%ws", pszBase, pszRelative);
hr = UrlCanonicalizeUnescape(pwzFileCombined, pszCombined, pcchCombined, 0);
goto Exit;
}
else {
dwSize = MAX_URL_LENGTH;
hr = UrlCombineW(pszBase, pszRelative, pwzCombined, &dwSize, dwFlags);
if (FAILED(hr)) {
goto Exit;
}
}
// Don't unescape if the relative part was already an URL because
// URLs wouldn't have been escaped during the UrlCombined.
if (UrlIsW(pwzCombined, URLIS_FILEURL)) {
hr = UrlUnescapeW(pwzCombined, pszCombined, pcchCombined, 0);
if (FAILED(hr)) {
goto Exit;
}
}
else {
if (*pcchCombined >= dwSize) {
lstrcpyW(pszCombined, pwzCombined);
}
*pcchCombined = dwSize;
}
Exit:
SAFEDELETEARRAY(pwzCombined);
SAFEDELETEARRAY(pwzFileCombined);
return hr;
}
HRESULT CDebugLog::SetAsmName(IAssemblyName *pName, LPCWSTR szCodebase)
{
HRESULT hr = S_OK;
DWORD dwSize = 0;
BOOL bWhereRefBind = FALSE;
LPWSTR wzBuf = NULL;
if(!pName) {
bWhereRefBind = TRUE;
}
else {
dwSize = 0;
hr = pName->GetProperty(ASM_NAME_NAME, NULL, &dwSize);
if(FAILED(hr) && hr != HRESULT_FROM_WIN32(ERROR_INSUFFICIENT_BUFFER)) {
goto Exit;
}
if (!dwSize) {
bWhereRefBind = TRUE;
}
hr = S_OK;
}
if(bWhereRefBind) {
if(!szCodebase) {
hr = E_INVALIDARG;
goto Exit;
}
wzBuf = NEW(WCHAR[MAX_URL_LENGTH+1]);
if (!wzBuf) {
hr = E_OUTOFMEMORY;
goto Exit;
}
wzBuf[0] = L'\0';
dwSize = MAX_URL_LENGTH;
hr = UrlCanonicalizeUnescape(szCodebase, wzBuf, &dwSize, 0);
if (SUCCEEDED(hr)) {
_pwzAsmName = WSTRDupDynamic(wzBuf);
if (!_pwzAsmName) {
hr = E_OUTOFMEMORY;
goto Exit;
}
}
}
else {
dwSize = 0;
hr = pName->GetDisplayName(NULL, &dwSize, 0);
if (FAILED(hr) && hr != HRESULT_FROM_WIN32(ERROR_INSUFFICIENT_BUFFER)) {
goto Exit;
}
if (dwSize) {
_pwzAsmName = NEW(WCHAR[dwSize]);
if (!_pwzAsmName) {
hr = E_OUTOFMEMORY;
goto Exit;
}
hr = pName->GetDisplayName(_pwzAsmName, &dwSize, 0);
if (FAILED(hr)) {
goto Exit;
}
}
}
Exit:
SAFEDELETEARRAY(wzBuf);
return hr;
}
// ---------------------------------------------------------------------------
// CCache::InsertTransCacheEntry
//---------------------------------------------------------------------------
HRESULT CCache::InsertTransCacheEntry(IAssemblyName *pName,
LPTSTR szPath, DWORD dwKBSize, DWORD dwFlags,
DWORD dwCommitFlags, DWORD dwPinBits, CTransCache **ppTransCache)
{
HRESULT hr;
DWORD cb, dwCacheId;
TRANSCACHEINFO *pTCInfo = NULL;
CTransCache *pTransCache = NULL;
LPWSTR pwzCodebaseUrl = NULL;
WCHAR pwzCanonicalized[MAX_URL_LENGTH];
// Determine which cache index to insert to.
if (FAILED(hr = ResolveCacheIndex(pName, dwFlags, &dwCacheId)))
goto exit;
// Construct new CTransCache object.
if(FAILED(hr = CreateTransCacheEntry(dwCacheId, &pTransCache)))
goto exit;
// Cast pTransCache base info ptr to TRANSCACHEINFO ptr
pTCInfo = (TRANSCACHEINFO*) pTransCache->_pInfo;
// Downcased text name from target
if (FAILED(hr = NameObjGetWrapper(pName, ASM_NAME_NAME,
(LPBYTE*) &pTCInfo->pwzName, &(cb = 0)))
// Version
|| FAILED(hr = pName->GetVersion(&pTCInfo->dwVerHigh, &pTCInfo->dwVerLow))
// Culture (downcased)
|| FAILED(hr = NameObjGetWrapper(pName, ASM_NAME_CULTURE,
(LPBYTE*) &pTCInfo->pwzCulture, &cb))
|| (pTCInfo->pwzCulture && !_wcslwr(pTCInfo->pwzCulture))
// PublicKeyToken
|| FAILED(hr = NameObjGetWrapper(pName, ASM_NAME_PUBLIC_KEY_TOKEN,
&pTCInfo->blobPKT.pBlobData, &pTCInfo->blobPKT.cbSize))
// Custom
|| FAILED(hr = NameObjGetWrapper(pName, ASM_NAME_CUSTOM,
&pTCInfo->blobCustom.pBlobData, &pTCInfo->blobCustom.cbSize))
// MVID
|| FAILED(hr = NameObjGetWrapper(pName, ASM_NAME_MVID,
&pTCInfo->blobMVID.pBlobData, &pTCInfo->blobMVID.cbSize))
// Codebase url if any from target
|| FAILED(hr = NameObjGetWrapper(pName, ASM_NAME_CODEBASE_URL,
(LPBYTE*)&pwzCodebaseUrl, &(cb = 0)))
// Codebase last modified time if any from target.
|| FAILED(hr = pName->GetProperty(ASM_NAME_CODEBASE_LASTMOD,
&pTCInfo->ftLastModified, &(cb = sizeof(FILETIME))))
// PK if any from source.
|| FAILED(hr = NameObjGetWrapper(pName, ASM_NAME_PUBLIC_KEY,
&pTCInfo->blobPK.pBlobData, &pTCInfo->blobPK.cbSize))
)
{
goto exit;
}
if (pwzCodebaseUrl)
{
cb = MAX_URL_LENGTH;
hr = UrlCanonicalizeUnescape(pwzCodebaseUrl, pwzCanonicalized, &cb, 0);
if (FAILED(hr)) {
goto exit;
}
pTCInfo->pwzCodebaseURL = WSTRDupDynamic(pwzCanonicalized);
if (!pTCInfo->pwzCodebaseURL) {
hr = E_OUTOFMEMORY;
goto exit;
}
}
else
{
pTCInfo->pwzCodebaseURL = NULL;
}
// Copy in path.
if (!(pTCInfo->pwzPath = TSTRDupDynamic(szPath)))
{
hr = E_OUTOFMEMORY;
goto exit;
}
// File size
pTCInfo->dwKBSize = dwKBSize;
// Set pin Bits
pTCInfo->dwPinBits = dwPinBits;
exit:
SAFEDELETEARRAY(pwzCodebaseUrl);
// Base destructor takes care
// of everything.
if (!ppTransCache
|| (FAILED(hr) && (hr != DB_E_DUPLICATE)))
{
SAFERELEASE(pTransCache);
}
else
{
*ppTransCache = pTransCache;
}
return hr;
}
