STDMETHODIMP CHashCheck::AddPages( LPFNADDPROPSHEETPAGE pfnAddPage, LPARAM lParam ) { PROPSHEETPAGE psp; psp.dwSize = sizeof(psp); psp.dwFlags = PSP_USECALLBACK | PSP_USEREFPARENT | PSP_USETITLE; psp.hInstance = g_hModThisDll; psp.pszTemplate = MAKEINTRESOURCE(IDD_HASHPROP); psp.pszTitle = MAKEINTRESOURCE(IDS_HP_TITLE); psp.pfnDlgProc = HashPropDlgProc; psp.lParam = (LPARAM)m_hList; psp.pfnCallback = HashPropCallback; psp.pcRefParent = (PUINT)&g_cRefThisDll; if (ActivateManifest(FALSE)) { psp.dwFlags |= PSP_USEFUSIONCONTEXT; psp.hActCtx = g_hActCtx; } HPROPSHEETPAGE hPage = CreatePropertySheetPage(&psp); if (hPage && !pfnAddPage(hPage, lParam)) DestroyPropertySheetPage(hPage); // HashProp has AddRef'ed and now owns our list SLRelease(m_hList); m_hList = NULL; return(S_OK); }
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); }
STDMETHODIMP CHashCheck::Drop( LPDATAOBJECT pdtobj, DWORD grfKeyState, POINTL pt, PDWORD pdwEffect ) { FORMATETC format = { CF_HDROP, NULL, DVASPECT_CONTENT, -1, TYMED_HGLOBAL }; STGMEDIUM medium; UINT uThreads = 0; if (pdtobj && pdtobj->GetData(&format, &medium) == S_OK) { if (HDROP hDrop = (HDROP)GlobalLock(medium.hGlobal)) { UINT uDrops = DragQueryFile(hDrop, -1, NULL, 0); UINT cchPath; LPTSTR lpszPath; // Reduce the likelihood of a race condition when trying to create // an activation context by creating it before creating threads ActivateManifest(FALSE); for (UINT uDrop = 0; uDrop < uDrops; ++uDrop) { if ( (cchPath = DragQueryFile(hDrop, uDrop, NULL, 0)) && (lpszPath = (LPTSTR)malloc((cchPath + 1) * sizeof(TCHAR))) ) { InterlockedIncrement(&g_cRefThisDll); HANDLE hThread; if ( (DragQueryFile(hDrop, uDrop, lpszPath, cchPath + 1) == cchPath) && (!(GetFileAttributes(lpszPath) & FILE_ATTRIBUTE_DIRECTORY)) && (hThread = CreateThreadCRT(HashVerifyThread, lpszPath)) ) { // The thread should free lpszPath, not us CloseHandle(hThread); ++uThreads; } else { free(lpszPath); InterlockedDecrement(&g_cRefThisDll); } } } GlobalUnlock(medium.hGlobal); } ReleaseStgMedium(&medium); } if (uThreads) { // DROPEFFECT_LINK would work here as well; it really doesn't matter *pdwEffect = DROPEFFECT_COPY; return(S_OK); } else { // We shouldn't ever be hitting this case *pdwEffect = DROPEFFECT_NONE; return(E_INVALIDARG); } }