DWORD WINAPI HashVerifyThread( PTSTR pszPath )
{
// We will need to free the memory allocated for the data when done
PBYTE pbRawData;
// First, activate our manifest and AddRef our host
ULONG_PTR uActCtxCookie = ActivateManifest(TRUE);
ULONG_PTR uHostCookie = HostAddRef();
// Allocate the context data that will persist across this session
HASHVERIFYCONTEXT hvctx;
// It's important that we zero the memory since an initial value of zero is
// assumed for many of the elements
ZeroMemory(&hvctx, sizeof(hvctx));
// Prep the path
HashNormalizeString(pszPath);
StrTrim(pszPath, TEXT(" "));
hvctx.pszPath = pszPath;
// Load the raw data
pbRawData = HashVerifyLoadData(&hvctx);
if (hvctx.pszFileData && (hvctx.hList = SLCreateEx(TRUE)))
{
HashVerifyParseData(&hvctx);
DialogBoxParam(
g_hModThisDll,
MAKEINTRESOURCE(IDD_HASHVERF),
NULL,
HashVerifyDlgProc,
(LPARAM)&hvctx
);
SLRelease(hvctx.hList);
}
else if (*pszPath)
{
// Technically, we could reach this point by either having a file read
// error or a memory allocation error, but I really don't feel like
// doing separate messages for what are supposed to be rare edge cases.
TCHAR szFormat[MAX_STRINGRES], szMessage[0x100];
LoadString(g_hModThisDll, IDS_HV_LOADERROR_FMT, szFormat, countof(szFormat));
wnsprintf(szMessage, countof(szMessage), szFormat, pszPath);
MessageBox(NULL, szMessage, NULL, MB_OK | MB_ICONERROR);
}
free(pbRawData);
free(pszPath);
// Clean up the manifest activation and release our host
DeactivateManifest(uActCtxCookie);
HostRelease(uHostCookie);
InterlockedDecrement(&g_cRefThisDll);
return(0);
}
DWORD WINAPI HashSaveThread( PHASHSAVECONTEXT phsctx )
{
// First, activate our manifest and AddRef our host
ULONG_PTR uActCtxCookie = ActivateManifest(TRUE);
ULONG_PTR uHostCookie = HostAddRef();
// Calling HashCalcPrepare with a NULL hList will cause it to calculate
// and set cchPrefix, but it will not copy the data or walk the directories
// (we will leave that for the worker thread to do); the reason we do a
// limited scan now is so that we can show the file dialog (which requires
// cchPrefix for the automatic name generation) as soon as possible
phsctx->status = INACTIVE;
phsctx->hList = NULL;
HashCalcPrepare(phsctx);
// Get a file name from the user
ZeroMemory(&phsctx->ofn, sizeof(phsctx->ofn));
HashCalcInitSave(phsctx);
if (phsctx->hFileOut != INVALID_HANDLE_VALUE)
{
BOOL bDeletionFailed = TRUE;
if (phsctx->hList = SLCreateEx(TRUE))
{
bDeletionFailed = ! DialogBoxParam(
g_hModThisDll,
MAKEINTRESOURCE(IDD_HASHSAVE),
NULL,
HashSaveDlgProc,
(LPARAM)phsctx
);
SLRelease(phsctx->hList);
}
CloseHandle(phsctx->hFileOut);
// Should only happen on Windows XP
if (bDeletionFailed)
DeleteFile(phsctx->ofn.lpstrFile);
}
// This must be the last thing that we free, since this is what supports
// our context!
SLRelease(phsctx->hListRaw);
// Clean up the manifest activation and release our host
DeactivateManifest(uActCtxCookie);
HostRelease(uHostCookie);
InterlockedDecrement(&g_cRefThisDll);
return(0);
}