mirror of https://github.com/tongzx/nt5src
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
943 lines
26 KiB
943 lines
26 KiB
/**********************************************************************************
|
|
|
|
FILENAME: DiskView.cpp
|
|
|
|
CLASS: DISKVIEW
|
|
|
|
CLASS DESCRIPTION:
|
|
DiskView takes data from the engines about what each cluster of the disk is used for.
|
|
It creates an array in memory (ClusterArray) to hold this data and then creates a
|
|
similar array to hold the data that DiskDisplay will use (LineArray) to display a
|
|
graphical image on the screen. This means that DiskView must look through many
|
|
clusters in it's array an determine that the bulk of them are fragmented or not,
|
|
(or some special type of file), and set a byte to a specific color in the LineArray
|
|
(memory array for the display data -- 1 line = 1 block of clusters).
|
|
|
|
COPYRIGHT© 2001 Microsoft Corporation and Executive Software International, Inc.
|
|
|
|
*/
|
|
|
|
|
|
#include "stdafx.h"
|
|
#ifndef SNAPIN
|
|
#include <windows.h>
|
|
#endif
|
|
|
|
extern "C" {
|
|
#include "SysStruc.h"
|
|
}
|
|
#include "ErrMacro.h"
|
|
|
|
#include "DfrgCmn.h"
|
|
#include "DfrgEngn.h"
|
|
#include "DfrgRes.h"
|
|
|
|
#include "DiskView.h"
|
|
#include "Graphix.h"
|
|
#include "IntFuncs.h"
|
|
|
|
#include "GetReg.h"
|
|
|
|
|
|
#ifdef _DEBUG
|
|
#define CLASSINVARIANT() EH_ASSERT(Invariant())
|
|
#else
|
|
#define CLASSINVARIANT()
|
|
#endif
|
|
|
|
// default: set line to frag color if 3% of clusters are fragged
|
|
#define FRAGTHRESHDEFAULT 3
|
|
|
|
// TEMP TEMP
|
|
#define ANALYZE_MEM_FAIL 0
|
|
#define DEFRAG_MEM_FAIL 0
|
|
|
|
|
|
/**********************************************************************************
|
|
|
|
Construction/Destruction
|
|
|
|
**********************************************************************************/
|
|
|
|
DiskView::DiskView()
|
|
{
|
|
::InitializeCriticalSection(&m_CriticalSection);
|
|
|
|
// data we need from UI
|
|
NumLines = 0;
|
|
ClusterFactor = 0;
|
|
pLineArray = NULL;
|
|
|
|
// control variables
|
|
m_bIsDataSent = FALSE;
|
|
m_IsActive = FALSE;
|
|
m_FragColorThreshold = FRAGTHRESHDEFAULT;
|
|
m_bMemAllocFailed = FALSE;
|
|
|
|
m_nFreeDelta = 0;
|
|
m_nFragDelta = 0;
|
|
m_nNonMovableDelta = 0;
|
|
m_nUsedDelta = 0;
|
|
|
|
// init basic variables.
|
|
MftZoneStart = 0;
|
|
MftZoneEnd = 0;
|
|
Clusters = 0;
|
|
pClusterArray = NULL;
|
|
|
|
InitFragColorThreshold();
|
|
|
|
CLASSINVARIANT();
|
|
}
|
|
|
|
DiskView::DiskView(const int InClusters,
|
|
const int InMftZoneStart = 0,
|
|
const int InMftZoneEnd = 0)
|
|
{
|
|
require(InClusters > 0);
|
|
|
|
::InitializeCriticalSection(&m_CriticalSection);
|
|
|
|
// data we need from UI
|
|
NumLines = 0;
|
|
ClusterFactor = 0;
|
|
pLineArray = NULL;
|
|
|
|
// control variables
|
|
m_bIsDataSent = FALSE;
|
|
m_IsActive = FALSE;
|
|
m_FragColorThreshold = FRAGTHRESHDEFAULT;
|
|
m_bMemAllocFailed = FALSE;
|
|
|
|
// init basic variables.
|
|
MftZoneStart = InMftZoneStart;
|
|
MftZoneEnd = InMftZoneEnd;
|
|
Clusters = InClusters;
|
|
|
|
// cluster map
|
|
ClusterArraySize = Clusters / 2 + Clusters % 2; // one cluster per nibble
|
|
#if ANALYZE_MEM_FAIL // TEMP TEMP
|
|
pClusterArray = new char [ClusterArraySize * 10000];
|
|
#else
|
|
pClusterArray = new char [ClusterArraySize];
|
|
#endif
|
|
EH(pClusterArray);
|
|
|
|
// fill the array initially with free space.
|
|
if (pClusterArray) {
|
|
FillMemory(pClusterArray, ClusterArraySize, (FreeSpaceColor << 4) | FreeSpaceColor);
|
|
}
|
|
else {
|
|
// new failed, keep class consistent
|
|
m_bMemAllocFailed = TRUE;
|
|
Clusters = 0;
|
|
ClusterArraySize = 0;
|
|
}
|
|
|
|
InitFragColorThreshold();
|
|
|
|
CLASSINVARIANT();
|
|
}
|
|
|
|
/*
|
|
|
|
METHOD: DiskView::InitFragColorThreshold
|
|
|
|
DESCRIPTION:
|
|
Get the threshold percentage of fragmented clusters that it takes to cause
|
|
a line to display in the fragmented color. If it is not in the registry at
|
|
all then we don't set it. Otherwise it will be defaulted by the constructor.
|
|
|
|
*/
|
|
|
|
void DiskView::InitFragColorThreshold()
|
|
{
|
|
// get frag color threshold percent from registry
|
|
TCHAR cRegValue[100];
|
|
HKEY hValue = NULL;
|
|
DWORD dwRegValueSize = sizeof(cRegValue);
|
|
|
|
// get the free space error threshold from the registry
|
|
long ret = GetRegValue(
|
|
&hValue,
|
|
TEXT("SOFTWARE\\Microsoft\\Dfrg"),
|
|
TEXT("FragColorThreshold"),
|
|
cRegValue,
|
|
&dwRegValueSize);
|
|
|
|
RegCloseKey(hValue);
|
|
|
|
// convert it and apply range limits
|
|
if (ret == ERROR_SUCCESS) {
|
|
m_FragColorThreshold = (UINT) _ttoi(cRegValue);
|
|
|
|
// > 50 does not make sense!
|
|
if (m_FragColorThreshold > 50)
|
|
m_FragColorThreshold = 50;
|
|
|
|
// < 0 does not either!
|
|
if (m_FragColorThreshold < 1)
|
|
m_FragColorThreshold = 0;
|
|
}
|
|
}
|
|
|
|
/*
|
|
|
|
METHOD: DiskView::DiskView (Copy Contructor)
|
|
|
|
DESCRIPTION:
|
|
Copies a DiskView class.
|
|
|
|
RETURN:
|
|
New Class reference.
|
|
*/
|
|
|
|
DiskView::DiskView(const DiskView& InDiskView)
|
|
{
|
|
EH_ASSERT(InDiskView.Invariant());
|
|
require(InDiskView.HasMapMemory());
|
|
|
|
::InitializeCriticalSection(&m_CriticalSection);
|
|
|
|
// copy over some basic variables from the input DiskView.
|
|
Clusters = InDiskView.Clusters;
|
|
ClusterArraySize = Clusters / 2 + Clusters % 2; // one cluster per nibble
|
|
NumLines = InDiskView.NumLines;
|
|
if (NumLines > 0)
|
|
ClusterFactor = (Clusters / NumLines) + ((Clusters % NumLines) ? 1 : 0);
|
|
else
|
|
ClusterFactor = 0;
|
|
EH_ASSERT(ClusterFactor == InDiskView.ClusterFactor);
|
|
|
|
MftZoneEnd = InDiskView.MftZoneEnd;
|
|
MftZoneStart = InDiskView.MftZoneStart;
|
|
m_bIsDataSent = FALSE;
|
|
m_bMemAllocFailed = FALSE;
|
|
m_IsActive = InDiskView.m_IsActive;
|
|
m_FragColorThreshold = InDiskView.m_FragColorThreshold;
|
|
|
|
// cluster map
|
|
#if DEFRAG_MEM_FAIL // TEMP TEMP
|
|
pClusterArray = new char [ClusterArraySize * 10000];
|
|
#else
|
|
pClusterArray = new char [ClusterArraySize];
|
|
#endif
|
|
EH(pClusterArray);
|
|
|
|
// copy over the cluster array.
|
|
if (pClusterArray) {
|
|
CopyMemory(pClusterArray, InDiskView.pClusterArray, ClusterArraySize);
|
|
}
|
|
else {
|
|
// new failed, keep class consistent
|
|
m_bMemAllocFailed = TRUE;
|
|
Clusters = 0;
|
|
ClusterArraySize = 0;
|
|
}
|
|
|
|
// line array
|
|
if (NumLines > 0) {
|
|
pLineArray = new char [NumLines];
|
|
EH(pLineArray);
|
|
if (pLineArray) {
|
|
FillMemory(pLineArray, NumLines, DirtyColor);
|
|
}
|
|
else {
|
|
// new failed, keep class consistent
|
|
m_bMemAllocFailed = TRUE;
|
|
NumLines = 0;
|
|
}
|
|
}
|
|
else {
|
|
pLineArray = NULL;
|
|
}
|
|
|
|
CLASSINVARIANT();
|
|
}
|
|
|
|
/*
|
|
|
|
METHOD: DiskView::operator= (Assignment Operator)
|
|
|
|
DESCRIPTION:
|
|
Makes the Lvalue a duplicate of the Rvalue.
|
|
|
|
*/
|
|
|
|
DiskView& DiskView::operator=(const DiskView& InDiskView)
|
|
{
|
|
// check for assignment to self
|
|
if (this == &InDiskView) {
|
|
return *this;
|
|
}
|
|
|
|
EH_ASSERT(InDiskView.Invariant());
|
|
require(InDiskView.HasMapMemory());
|
|
|
|
__try
|
|
{
|
|
|
|
::EnterCriticalSection(&m_CriticalSection);
|
|
|
|
// copy over some basic variables from the input DiskView.
|
|
MftZoneEnd = InDiskView.MftZoneEnd;
|
|
MftZoneStart = InDiskView.MftZoneStart;
|
|
m_bIsDataSent = FALSE;
|
|
m_bMemAllocFailed = FALSE;
|
|
m_IsActive = InDiskView.m_IsActive;
|
|
m_FragColorThreshold = InDiskView.m_FragColorThreshold;
|
|
|
|
// if the number of clusters in the cluster array are not identical, realloc the array.
|
|
if (Clusters != InDiskView.Clusters) {
|
|
|
|
// get the new size for the cluster array.
|
|
Clusters = InDiskView.Clusters;
|
|
ClusterArraySize = Clusters / 2 + Clusters % 2; // one cluster per nibble
|
|
|
|
// redimension the cluster array.
|
|
if (pClusterArray)
|
|
delete [] pClusterArray;
|
|
#if DEFRAG_MEM_FAIL // TEMP TEMP
|
|
pClusterArray = new char [ClusterArraySize * 10000];
|
|
#else
|
|
pClusterArray = new char [ClusterArraySize];
|
|
#endif
|
|
EH(pClusterArray);
|
|
}
|
|
|
|
// copy over the cluster array.
|
|
if (pClusterArray) {
|
|
CopyMemory(pClusterArray, InDiskView.pClusterArray, ClusterArraySize);
|
|
}
|
|
else {
|
|
// new failed, keep class consistent
|
|
m_bMemAllocFailed = TRUE;
|
|
Clusters = 0;
|
|
ClusterArraySize = 0;
|
|
}
|
|
|
|
// line array
|
|
NumLines = InDiskView.NumLines;
|
|
if (NumLines > 0)
|
|
ClusterFactor = (Clusters / NumLines) + ((Clusters % NumLines) ? 1 : 0);
|
|
else
|
|
ClusterFactor = 0;
|
|
|
|
if (NumLines > 0) {
|
|
pLineArray = new char [NumLines];
|
|
EH(pLineArray);
|
|
if (pLineArray) {
|
|
FillMemory(pLineArray, NumLines, DirtyColor);
|
|
}
|
|
else {
|
|
// new failed, keep class consistent
|
|
m_bMemAllocFailed = TRUE;
|
|
NumLines = 0;
|
|
}
|
|
}
|
|
else {
|
|
pLineArray = NULL;
|
|
}
|
|
|
|
CLASSINVARIANT();
|
|
}
|
|
__finally
|
|
{
|
|
::LeaveCriticalSection(&m_CriticalSection);
|
|
}
|
|
|
|
return *this;
|
|
}
|
|
|
|
/*
|
|
|
|
METHOD: DiskView::~DiskView (Destructor)
|
|
|
|
*/
|
|
|
|
DiskView::~DiskView()
|
|
{
|
|
CLASSINVARIANT();
|
|
|
|
::DeleteCriticalSection(&m_CriticalSection);
|
|
|
|
if (pClusterArray)
|
|
delete [] pClusterArray;
|
|
|
|
if (pLineArray)
|
|
delete [] pLineArray;
|
|
}
|
|
|
|
/**********************************************************************************
|
|
|
|
Implementation
|
|
|
|
**********************************************************************************/
|
|
|
|
void DiskView::SetClusterCount(const int InClusters)
|
|
{
|
|
require(InClusters > 0);
|
|
|
|
__try
|
|
{
|
|
|
|
::EnterCriticalSection(&m_CriticalSection);
|
|
|
|
m_IsActive = TRUE;
|
|
|
|
Clusters = InClusters;
|
|
|
|
// cluster map
|
|
ClusterArraySize = Clusters / 2 + Clusters % 2; // one cluster per nibble
|
|
#if ANALYZE_MEM_FAIL // TEMP TEMP
|
|
pClusterArray = new char [ClusterArraySize * 10000];
|
|
#else
|
|
pClusterArray = new char [ClusterArraySize];
|
|
#endif
|
|
EH(pClusterArray);
|
|
|
|
// fill the array initially with free space.
|
|
if (pClusterArray) {
|
|
FillMemory(pClusterArray, ClusterArraySize, (FreeSpaceColor << 4) | FreeSpaceColor);
|
|
}
|
|
else {
|
|
// new failed, keep class consistent
|
|
m_bMemAllocFailed = TRUE;
|
|
Clusters = 0;
|
|
ClusterArraySize = 0;
|
|
}
|
|
|
|
CLASSINVARIANT();
|
|
}
|
|
__finally
|
|
{
|
|
::LeaveCriticalSection(&m_CriticalSection);
|
|
}
|
|
}
|
|
|
|
void DiskView::SetMftZone(const int InZoneStart, const int InZoneEnd)
|
|
{
|
|
require(InZoneEnd >= InZoneStart);
|
|
|
|
__try
|
|
{
|
|
::EnterCriticalSection(&m_CriticalSection);
|
|
|
|
MftZoneStart = InZoneStart;
|
|
MftZoneEnd = InZoneEnd;
|
|
|
|
CLASSINVARIANT();
|
|
}
|
|
__finally
|
|
{
|
|
::LeaveCriticalSection(&m_CriticalSection);
|
|
}
|
|
}
|
|
|
|
/*
|
|
|
|
METHOD: DiskView::SetNumLines
|
|
|
|
DESCRIPTION:
|
|
Called to update the number of lines (pixels) that are displayed on the UI
|
|
|
|
*/
|
|
|
|
void DiskView::SetNumLines(const int inNumLines)
|
|
{
|
|
if (NumLines != inNumLines && inNumLines > 0 && pClusterArray != NULL) {
|
|
|
|
__try
|
|
{
|
|
|
|
::EnterCriticalSection(&m_CriticalSection);
|
|
|
|
CLASSINVARIANT();
|
|
|
|
NumLines = inNumLines;
|
|
|
|
ClusterFactor = (Clusters / NumLines) + ((Clusters % NumLines) ? 1 : 0);
|
|
|
|
EH(ClusterFactor > 0);
|
|
|
|
if (pLineArray)
|
|
delete [] pLineArray;
|
|
|
|
pLineArray = new char [NumLines];
|
|
EH(pLineArray);
|
|
|
|
if (pLineArray) {
|
|
FillMemory(pLineArray, NumLines, DirtyColor);
|
|
}
|
|
else {
|
|
// new failed, keep class consistent
|
|
m_bMemAllocFailed = TRUE;
|
|
NumLines = 0;
|
|
}
|
|
|
|
CLASSINVARIANT();
|
|
}
|
|
|
|
__finally
|
|
{
|
|
::LeaveCriticalSection(&m_CriticalSection);
|
|
}
|
|
|
|
}
|
|
}
|
|
|
|
/*
|
|
|
|
METHOD: DiskView::GetLineArray
|
|
|
|
DESCRIPTION:
|
|
Makes LineArray available outside of the class so it can be passed to DiskDisplay.
|
|
|
|
*/
|
|
|
|
BOOL DiskView::GetLineArray(char ** ppOutLineArray, DWORD *pNumLines)
|
|
{
|
|
BOOL bReturn = TRUE;
|
|
// initialize outputs
|
|
*ppOutLineArray = NULL;
|
|
*pNumLines = 0;
|
|
|
|
if (!m_IsActive || pClusterArray == NULL){
|
|
bReturn = FALSE;
|
|
|
|
} else
|
|
{
|
|
// see if we have a line array
|
|
if (pLineArray != NULL && NumLines > 0) {
|
|
|
|
__try
|
|
{
|
|
::EnterCriticalSection(&m_CriticalSection);
|
|
|
|
CLASSINVARIANT();
|
|
|
|
// update the line array from the cluster array
|
|
TransferToLineArray();
|
|
|
|
// allocate space for a copy of the line array
|
|
*ppOutLineArray = new char[NumLines];
|
|
|
|
// if we got memory, set the outputs
|
|
if (*ppOutLineArray) {
|
|
CopyMemory(*ppOutLineArray, pLineArray, NumLines);
|
|
*pNumLines = (DWORD) NumLines;
|
|
}
|
|
|
|
m_bIsDataSent = TRUE;
|
|
|
|
}
|
|
__finally
|
|
{
|
|
::LeaveCriticalSection(&m_CriticalSection);
|
|
bReturn = TRUE;
|
|
}
|
|
}
|
|
else {
|
|
bReturn = FALSE;
|
|
}
|
|
}
|
|
return bReturn;
|
|
}
|
|
|
|
/*
|
|
|
|
METHOD: DiskView::UpdateClusterArray
|
|
|
|
DESCRIPTION:
|
|
Updates portions of the cluster array from extent data about the disk. (An extent
|
|
is a series of adjacent clusters that are part of the same file -- or sometimes in
|
|
our program, part of no file.) The extents are passed in as a series of EXTENT_LIST
|
|
structures with a LONGLONG header. The first 7 bytes of the LONGLONG contain the
|
|
number of EXTENT_LIST structures following, the last byte of the LONGLONG contains
|
|
the color code of the extents (meaning a series of extents all represent the same
|
|
type of data: contiguous files, fragmented files, etc.) After the last EXTENT_LIST
|
|
structure, there may be another LONGLONG identifying that there is another series of
|
|
EXTENT_LIST's. Eventually there will be a zeroed out LONGLONG indicating there is
|
|
no more data. UpdateClusterArray goes through each of these extents and updates the
|
|
appropriate clusters in ClusterArray. It then re-evaluates those specific lines in
|
|
the LineArray that would be affected by the changes in ClusterArray.
|
|
|
|
INPUT + OUTPUT:
|
|
pBuffer - The buffer holding the EXTENT_LIST's and their LONGLONG headers.
|
|
|
|
*/
|
|
|
|
void DiskView::UpdateClusterArray(PBYTE pBuffer)
|
|
{
|
|
LONGLONG StartingLcn;
|
|
BYTE Color;
|
|
int ClusterCount;
|
|
EXTENT_LIST* pUpdateArray;
|
|
LONGLONG Extents = 0;
|
|
LONGLONG Extent = 0;
|
|
int TotalRunLength = 0;
|
|
BYTE * pTempBuffer = pBuffer;
|
|
|
|
if (!m_IsActive || pClusterArray == NULL){
|
|
return;
|
|
}
|
|
|
|
__try
|
|
{
|
|
|
|
::EnterCriticalSection(&m_CriticalSection);
|
|
|
|
CLASSINVARIANT();
|
|
|
|
//Indefinite length loop that goes through all the series of EXTENT_LIST's
|
|
//until there are no more.
|
|
while(TRUE){
|
|
|
|
//Get the number of extents.
|
|
Extents = *(LONGLONG*)pBuffer & 0x0FFFFFFFFFFF;
|
|
if(Extents == 0){
|
|
break;
|
|
}
|
|
//Point to the EXTENT_LIST's.
|
|
pBuffer += sizeof(LONGLONG);
|
|
|
|
//Get the color code.
|
|
Color = pBuffer[-1];
|
|
|
|
//Get a movable pointer to the EXTENT_LIST's
|
|
pUpdateArray = (EXTENT_LIST*)pBuffer;
|
|
|
|
//Set the high nibble to have the same color code as the low nibble of color.
|
|
//Color = (Low nibble) | (High nibble which is low nibble shifted)
|
|
Color = (Color & 0x0F) | (Color << 4);
|
|
|
|
//Go through each extent and
|
|
//update the ClusterArray and LineArray for that extent.
|
|
for(Extent = 0; Extent < Extents; pUpdateArray ++, Extent ++) {
|
|
|
|
//Get the data for this extent and validate it is correct
|
|
//(doesn't run of the end of the disk)
|
|
StartingLcn = pUpdateArray->StartingLcn;
|
|
ClusterCount = (UINT)pUpdateArray->ClusterCount;
|
|
if ((StartingLcn < 0) || (ClusterCount <= 0) ||
|
|
(StartingLcn + ClusterCount > Clusters)) {
|
|
__leave;
|
|
}
|
|
|
|
if (pLineArray != NULL && NumLines > 0) {
|
|
// mark the affected lines in the line array dirty
|
|
UINT nStartLine = (UINT) (StartingLcn / ClusterFactor);
|
|
UINT nEndLine = (UINT) ((StartingLcn + ClusterCount - 1) / ClusterFactor);
|
|
|
|
for (UINT ii = nStartLine; ii <= nEndLine; ii++) {
|
|
//sks bug#206244
|
|
if(ii >= (UINT) NumLines)
|
|
{
|
|
__leave;
|
|
}
|
|
pLineArray[ii] = DirtyColor;
|
|
}
|
|
}
|
|
|
|
//Set the ClusterArray nibbles to the appropriate color for this extent.
|
|
//If the extent starts on an odd number,
|
|
//then the first cluster is a high nibble of one byte.
|
|
if(StartingLcn % 2){
|
|
//Clear out that high nibble.
|
|
pClusterArray[StartingLcn / 2] &= 0x0F;
|
|
//Put our color in it.
|
|
pClusterArray[StartingLcn / 2] |= (Color & 0xF0);
|
|
//Change the counters to reflect that we wrote this cluster.
|
|
ClusterCount --;
|
|
StartingLcn ++;
|
|
}
|
|
//Fill in the middle range that consists of whole bytes
|
|
//(high and low nibbles) needing to be set.
|
|
if(ClusterCount != 0){
|
|
FillMemory(pClusterArray + (StartingLcn / 2), ClusterCount / 2, Color);
|
|
}
|
|
//If the extent ends on an odd number,
|
|
//then the last cluster is a low nibble of one byte.
|
|
if ((ClusterCount % 2) && (ClusterCount != 0)) {
|
|
//Update counter to point to it after previous writes.
|
|
StartingLcn += ClusterCount - 1;
|
|
//Clear out that low nibble.
|
|
pClusterArray[StartingLcn / 2] &= 0xF0;
|
|
//Put our color in it.
|
|
pClusterArray[StartingLcn / 2] |= (Color & 0x0F);
|
|
}
|
|
|
|
}
|
|
//Point to the next series of extents (since the for loop we just exited exits
|
|
//when all the extents in that series have been done).
|
|
pBuffer = (PBYTE)pUpdateArray;
|
|
}
|
|
}
|
|
|
|
__finally
|
|
{
|
|
::LeaveCriticalSection(&m_CriticalSection);
|
|
}
|
|
}
|
|
|
|
/*
|
|
|
|
METHOD: DiskView::TransferToLineArray
|
|
|
|
DESCRIPTION:
|
|
Create the compressed line array from the current data in the cluster array.
|
|
|
|
RETURN:
|
|
TRUE=ok, FALSE=error.
|
|
*/
|
|
|
|
BOOL DiskView::TransferToLineArray()
|
|
{
|
|
// make sure we have the arrays
|
|
if (!m_IsActive || pClusterArray == NULL || pLineArray == NULL) {
|
|
return FALSE;
|
|
}
|
|
|
|
EF_ASSERT(ClusterFactor > 0);
|
|
EF_ASSERT(NumLines > 0);
|
|
|
|
__try
|
|
{
|
|
|
|
::EnterCriticalSection(&m_CriticalSection);
|
|
|
|
CLASSINVARIANT();
|
|
|
|
// loop variables
|
|
int nClust; // nClust = cluster number
|
|
int nClustByte; // nClustByte = byte in cluster array
|
|
// (nClust / 2, bytes to nibbles)
|
|
BOOL bHighNibble; // bHighNibble = TRUE for high nibble and FALSE for low nibble
|
|
// (stored one cluster per nibble)
|
|
int nLine; // nLine = line number
|
|
char cClustColor; // current cluster's color
|
|
UINT nUsed; // number of used clusters in this line
|
|
UINT nFree; // number of free clusters in this line
|
|
UINT nFrag; // fragmented cluster(s) in this line
|
|
UINT nNonMovable; // number of non movable clusters in this line
|
|
BOOL bNonMovable; // system file cluster(s) in this line
|
|
BOOL bMftFragmented; // MFT in this line sks added MFT defrag
|
|
BOOL bMftUnFragmented; // MFT in this line sks added MFT defrag
|
|
|
|
// loop through line array
|
|
for (nLine = 0; nLine < NumLines; nLine++) {
|
|
|
|
// if this item is not dirty, move on
|
|
if (pLineArray[nLine] != DirtyColor) {
|
|
continue;
|
|
}
|
|
|
|
// initialize line variables
|
|
nUsed = 0;
|
|
nFree = 0;
|
|
nFrag = 0;
|
|
nNonMovable = 0; //bug#184739 sks 9/18/2000
|
|
bNonMovable = FALSE;
|
|
bMftFragmented = FALSE;
|
|
bMftUnFragmented = FALSE;
|
|
|
|
// reset to the starting point for this line
|
|
nClust = nLine * ClusterFactor;
|
|
|
|
// loop through the clusters in this line
|
|
for (int ii = 0; ii < ClusterFactor && nClust < Clusters; ii++, nClust++) {
|
|
|
|
// set loop variables
|
|
nClustByte = nClust / 2; // cluster array byte index
|
|
bHighNibble = (nClust % 2) ? TRUE : FALSE; // which nibble in byte?
|
|
|
|
// get the current cluster's color
|
|
// (the >> 4 extracts the high nibble, the & 0x0F extracts the low nibble)
|
|
if(bHighNibble) {
|
|
cClustColor = pClusterArray[nClustByte] >> 4;
|
|
}
|
|
else {
|
|
cClustColor = pClusterArray[nClustByte] & 0x0F;
|
|
}
|
|
|
|
// evaluate the color
|
|
switch (cClustColor) {
|
|
|
|
|
|
case SystemFileColor:
|
|
case PageFileColor:
|
|
nNonMovable++; //bug#184739 sks 9/18/2000
|
|
break;
|
|
|
|
case FragmentColor:
|
|
nFrag++;
|
|
break;
|
|
|
|
case UsedSpaceColor:
|
|
case DirectoryColor:
|
|
nUsed++;
|
|
break;
|
|
|
|
case FreeSpaceColor:
|
|
nFree++;
|
|
break;
|
|
|
|
default:
|
|
__leave;
|
|
break;
|
|
}
|
|
|
|
//bug#184739 sks 9/18/2000
|
|
// system file overrides all, break out of inner loop
|
|
|
|
} // end inner loop: clusters in current line
|
|
|
|
|
|
//
|
|
// #243245 changing the color scheme as follows
|
|
//
|
|
// Add the current file counts for each type to our running
|
|
// total. Think of this running total as "the file-counts of each
|
|
// type that were ignored in the previous lines displayed."
|
|
//
|
|
// For each line, then test which running total is more:
|
|
// contiguous, unmovable, fragmented files or free space.
|
|
//
|
|
// Whoever is greater wins, (if all are equal, fragmented wins)
|
|
// Subtract the ClusterFactor from the running count for the
|
|
// winner (since this line is his)
|
|
//
|
|
m_nFreeDelta += nFree;
|
|
m_nFragDelta += nFrag;
|
|
m_nNonMovableDelta += nNonMovable;
|
|
m_nUsedDelta += nUsed;
|
|
|
|
if((m_nFragDelta > m_nNonMovableDelta) &&
|
|
(m_nFragDelta > m_nFreeDelta) &&
|
|
(m_nFragDelta > m_nUsedDelta)
|
|
) {
|
|
//
|
|
// Fragmented is greatest
|
|
//
|
|
m_nFragDelta -= ClusterFactor;
|
|
pLineArray[nLine] = FragmentColor;
|
|
}
|
|
else if ((m_nUsedDelta > m_nFreeDelta) &&
|
|
(m_nUsedDelta > m_nNonMovableDelta) &&
|
|
(m_nUsedDelta > m_nFragDelta)
|
|
) {
|
|
//
|
|
// Used is greatest
|
|
//
|
|
m_nUsedDelta -= ClusterFactor;
|
|
pLineArray[nLine] = UsedSpaceColor;
|
|
}
|
|
else if ((m_nFreeDelta > m_nUsedDelta) &&
|
|
(m_nFreeDelta > m_nNonMovableDelta) &&
|
|
(m_nFreeDelta > m_nFragDelta)
|
|
) {
|
|
//
|
|
// Free is greatest
|
|
//
|
|
m_nFreeDelta -= ClusterFactor;
|
|
pLineArray[nLine] = FreeSpaceColor;
|
|
}
|
|
else if ((m_nNonMovableDelta > m_nUsedDelta) &&
|
|
(m_nNonMovableDelta > m_nFreeDelta) &&
|
|
(m_nNonMovableDelta > m_nFragDelta)
|
|
) {
|
|
//
|
|
// Nonmovable is greatest
|
|
//
|
|
m_nNonMovableDelta -= ClusterFactor;
|
|
pLineArray[nLine] = SystemFileColor;
|
|
}
|
|
else {
|
|
//
|
|
// They're all equal--use Fragmented colour
|
|
//
|
|
m_nFragDelta -= ClusterFactor;
|
|
pLineArray[nLine] = FragmentColor;
|
|
}
|
|
} // end outer loop: lines
|
|
}
|
|
|
|
__finally
|
|
{
|
|
::LeaveCriticalSection(&m_CriticalSection);
|
|
}
|
|
|
|
return TRUE;
|
|
}
|
|
|
|
/*
|
|
|
|
METHOD: DiskView::Invariant
|
|
|
|
DESCRIPTION:
|
|
Check the internal state of a DiskView instance.
|
|
|
|
*/
|
|
|
|
BOOL DiskView::Invariant() const
|
|
{
|
|
// clusters
|
|
EF((pClusterArray == NULL && Clusters == 0) ||
|
|
(pClusterArray != NULL && Clusters > 0));
|
|
EF(ClusterArraySize == Clusters / 2 + Clusters % 2); // one cluster per nibble
|
|
|
|
// lines
|
|
EF((pLineArray == NULL && NumLines == 0) ||
|
|
(pLineArray != NULL && NumLines > 0));
|
|
EF((NumLines == 0 && ClusterFactor == 0) ||
|
|
(NumLines > 0 && ClusterFactor > 0));
|
|
|
|
// MFT zone
|
|
EF(MftZoneEnd >= MftZoneStart);
|
|
|
|
return TRUE;
|
|
}
|
|
|
|
/*
|
|
|
|
METHOD: DiskView::HasMapMemory
|
|
|
|
DESCRIPTION:
|
|
Check if the DiskView has memory for the cluster array
|
|
(and line array if applicable).
|
|
|
|
*/
|
|
|
|
BOOL DiskView::HasMapMemory() const
|
|
{
|
|
return !m_bMemAllocFailed;
|
|
}
|
|
|
|
/*
|
|
|
|
METHOD: DiskView::EnterMyCriticalSection()
|
|
|
|
DESCRIPTION:
|
|
Enter the critical section, so that when we terminate thread was have the critical section
|
|
bug #26213 sks .
|
|
|
|
*/
|
|
void DiskView::EnterMyCriticalSection()
|
|
{
|
|
::EnterCriticalSection(&m_CriticalSection);
|
|
}
|
|
|
|
/*
|
|
|
|
METHOD: DiskView::LeaveMyCriticalSection()
|
|
|
|
DESCRIPTION:
|
|
Leave the critical section, so that when we terminate thread was have the critical section
|
|
bug #26213 sks .
|
|
|
|
*/
|
|
void DiskView::LeaveMyCriticalSection()
|
|
{
|
|
::LeaveCriticalSection(&m_CriticalSection);
|
|
}
|
|
|