173 lines
3.3 KiB
C++
173 lines
3.3 KiB
C++
/*++
|
||
|
||
Copyright (c) 1998 Microsoft Corporation
|
||
|
||
Module Name:
|
||
|
||
FTMan
|
||
|
||
File Name:
|
||
|
||
RootVol.cpp
|
||
|
||
Abstract:
|
||
|
||
Implementation of the CRootVolumesData class. The class who stores the properties of the "fake" root of the
|
||
volumes tree
|
||
|
||
Author:
|
||
|
||
Cristian Teodorescu October 22, 1998
|
||
|
||
Notes:
|
||
|
||
Revision History:
|
||
|
||
--*/
|
||
|
||
#include "stdafx.h"
|
||
|
||
#include "DiskMap.h"
|
||
#include "Global.h"
|
||
#include "LogVol.h"
|
||
#include "Resource.h"
|
||
#include "RootVol.h"
|
||
|
||
extern "C"
|
||
{
|
||
#include <FTAPI.h>
|
||
}
|
||
|
||
#include <winioctl.h>
|
||
#include <basetyps.h>
|
||
#include <mountmgr.h>
|
||
|
||
#ifdef _DEBUG
|
||
#define new DEBUG_NEW
|
||
#undef THIS_FILE
|
||
static char THIS_FILE[] = __FILE__;
|
||
#endif
|
||
|
||
///////////////////////////////////////////////////////////////////////////////////////////////////
|
||
// CRootVolumesData
|
||
|
||
// Constructor
|
||
CRootVolumesData::CRootVolumesData() : CItemData( IT_RootVolumes, NULL, FALSE )
|
||
{
|
||
}
|
||
|
||
////////////////////////////////////////////////////////////////////////////////////////////////////
|
||
// Public methods
|
||
|
||
BOOL CRootVolumesData::ReadItemInfo( CString& strErrors )
|
||
{
|
||
MY_TRY
|
||
|
||
m_bValid = TRUE;
|
||
strErrors = _T("");
|
||
|
||
m_ulNumMembers = 1; // Just to notify the tree that this item has children
|
||
|
||
m_iImage = ComputeImageIndex();
|
||
|
||
return m_bValid;
|
||
|
||
MY_CATCH_AND_THROW
|
||
}
|
||
|
||
BOOL CRootVolumesData::ReadMembers( CObArray& arrMembersData, CString& strErrors )
|
||
{
|
||
MY_TRY
|
||
|
||
BOOL bReturn = TRUE;
|
||
|
||
arrMembersData.RemoveAll();
|
||
strErrors = _T("");
|
||
|
||
// 1. Read all root logical volumes
|
||
|
||
FT_LOGICAL_DISK_ID diskId[100];
|
||
|
||
if( FtEnumerateLogicalDisks( 100,
|
||
diskId,
|
||
&m_ulNumMembers ) )
|
||
{
|
||
for( ULONG i = 0; i < m_ulNumMembers; i++ )
|
||
{
|
||
// Create the logical volume item data
|
||
CLogicalVolumeData* pData = new CLogicalVolumeData( diskId[i], this, TRUE );
|
||
// Read logical volume info and collect errors ( if any )
|
||
CString strMemberErrors;
|
||
pData->ReadItemInfo( strMemberErrors );
|
||
strErrors += strMemberErrors;
|
||
// Add the structure to the members' data array
|
||
arrMembersData.Add(pData);
|
||
}
|
||
}
|
||
else
|
||
{
|
||
AddError( strErrors, IDS_ERR_ENUMERATE_ROOT_VOLS, TRUE );
|
||
bReturn = FALSE;
|
||
m_ulNumMembers = 0;
|
||
}
|
||
|
||
// 2. Read all physical partitions not promoted as logical volumes
|
||
|
||
CDiskMap diskMap;
|
||
for( DWORD dwDiskNumber = 0; ; dwDiskNumber++ )
|
||
{
|
||
CObArray arrPartitions;
|
||
CString strDiskMapErrors;
|
||
BOOL bMissingDisk;
|
||
|
||
diskMap.SetDiskNumber(dwDiskNumber);
|
||
BOOL bResult = diskMap.ReadPartitions( arrPartitions, strDiskMapErrors, bMissingDisk, this );
|
||
strErrors += strDiskMapErrors;
|
||
|
||
if( !bResult )
|
||
{
|
||
if( bMissingDisk ) // It's over. There are no more disks in the system
|
||
break;
|
||
else // Continue with the following disk
|
||
{
|
||
bReturn = FALSE;
|
||
continue;
|
||
}
|
||
|
||
}
|
||
|
||
arrMembersData.Append(arrPartitions);
|
||
m_ulNumMembers += (ULONG)arrPartitions.GetSize();
|
||
}
|
||
|
||
// 3. Read all mount paths for every member
|
||
QueryMountList( arrMembersData );
|
||
|
||
return bReturn;
|
||
|
||
MY_CATCH_AND_THROW
|
||
}
|
||
|
||
int CRootVolumesData::ComputeImageIndex() const
|
||
{
|
||
return II_Root;
|
||
}
|
||
|
||
BOOL CRootVolumesData::operator==(CItemData& rData) const
|
||
{
|
||
return(rData.GetItemType() == IT_RootVolumes );
|
||
}
|
||
|
||
void CRootVolumesData::GetDisplayName( CString& strDisplay ) const
|
||
{
|
||
MY_TRY
|
||
|
||
strDisplay.LoadString(IDS_ROOT_VOLUMES_DISPLAY_NAME);
|
||
|
||
MY_CATCH_AND_THROW
|
||
}
|
||
|
||
////////////////////////////////////////////////////////////////////////////////////////////////////
|
||
// Protected methods
|
||
|
||
|