C / C++ / MFC

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);
}

#ifdef _DEBUG
    POPUP "Debug"
    BEGIN
        MENUITEM "Debug Command",               IDM_DEMO
    END
#endif