C / C++ / MFC

void Cvep::stoprecThread()
{
	KillTimer(4);
	KillTimer(5);
	CSingleLock lock1(&c_s0);
	lock1.Lock();
	// Unlock
	recflag=0;
	lock1.Unlock();
	PatternsFinish();

	ResumeThread(hand[1]);
	ResumeThread(hand[0]);
	
//////////////////////////////
//  Wait for the main thread 
	
	DWORD lpExitCode0;
	DWORD lpExitCode1;
	::GetExitCodeThread(hand[0], &lpExitCode0);
	::GetExitCodeThread(hand[1], &lpExitCode1);

 	if(STILL_ACTIVE == lpExitCode1 )
	{
	//	TerminateThread(hand[1],GetExitCodeThread(hand[1], &lpExitCode1));
			WaitForSingleObject(hand[1], INFINITE);
	}
	if(STILL_ACTIVE == lpExitCode0 )
	{
	//	TerminateThread(hand[0],GetExitCodeThread(hand[0], &lpExitCode0));
		WaitForSingleObject(hand[0], INFINITE);
	}
	
	CloseHandle(hand[1]);
	CloseHandle(hand[0]);
}

BOOL CFreeCluster::GetBitmap()
{
	STARTING_LCN_INPUT_BUFFER StartingLCN;
	VOLUME_BITMAP_BUFFER *Bitmap = NULL;
	ULONGLONG BitmapSize;
	DWORD BytesReturned;
	BOOL Result;

	StartingLCN.StartingLcn.QuadPart = 0;

	// Allocate buffer
	// Call FSCTL_GET_VOLUME_BITMAP once with a very small buffer
	// This will leave the total number of clusters in Bitmap->BitmapSize and we can
	// then correctly allocate based off that
	// I suppose this won't work if your drive has only 40 clusters on it or so :)
	BitmapSize = sizeof (VOLUME_BITMAP_BUFFER) + 4;
	Bitmap = (VOLUME_BITMAP_BUFFER *) malloc (BitmapSize);
	HANDLE hHandle=OpenDrive();

	Result = DeviceIoControl
		(
		hHandle,
		FSCTL_GET_VOLUME_BITMAP,
		&StartingLCN,
		sizeof (StartingLCN),
		Bitmap,
		BitmapSize,
		&BytesReturned,
		NULL
		);

	// Bad result?
	if (Result == FALSE  &&  GetLastError () != ERROR_MORE_DATA)
	{
		//wprintf ("\nDeviceIoControl returned false, GetLastError() was not ERROR_MORE_DATA\n");
		free (Bitmap);
		return (false);
	}

	// Otherwise, we're good
	//BitmapSize = sizeof (VOLUME_BITMAP_BUFFER) + (Bitmap->BitmapSize.QuadPart / 8) + 1;
	BitmapSize = (Bitmap->BitmapSize.QuadPart / 8) + 1;
	int nBufferSize=8*512*3;//3 cluster at a time
	ULONGLONG ullItr=BitmapSize/nBufferSize;
	
	if(BitmapSize%nBufferSize!=0)
		ullItr++;

	Bitmap = (VOLUME_BITMAP_BUFFER *) realloc (Bitmap, nBufferSize+1);

	for (ULONGLONG ullTmp=0; ullTmp < ullItr; ullTmp++)
	{
		if(BitmapSize < nBufferSize)
			nBufferSize=BitmapSize;

		StartingLCN.StartingLcn.QuadPart=ullTmp*nBufferSize;
		Result = DeviceIoControl
			(
			hHandle,
			FSCTL_GET_VOLUME_BITMAP,
			&StartingLCN,
			nBufferSize,//sizeof (StartingLCN),
			Bitmap,
			nBufferSize,
			&BytesReturned,
			NULL
			);

		BitmapSize=BitmapSize-nBufferSize;

		
		}
	}

	return (true);
}