//+==========================================================================
//
// File: CDir.cxx
//
// Purpose: Define the CDirectory class.
//
// This class is used to represent a directory name.
// Along with maintaining the name, it can determine
// the type of FileSystem.
//
//+==========================================================================
// --------
// Includes
// --------
#include <windows.h>
#include <stdlib.h>
#include <stdio.h>
#include <wchar.h>
#include <wtypes.h>
#include "CFMEx.hxx"
#include "CDir.hxx"
//+------------------------------------------------------------------------------
//
// Function: CDirectory::Initialize (no arguments)
//
// Synopsis: Generate a directory name, using the TEMP environment
// variable, and use it to initialize this object.
//
// Inputs: None.
//
// Outputs: TRUE if the function succeeds, FALSE otherwise.
//
//+------------------------------------------------------------------------------
BOOL CDirectory::Initialize()
{
// ---------------
// Local Variables
// ---------------
// Assume failure for now.
BOOL bSuccess = FALSE;
// The TEMP environment variable.
WCHAR wszSystemTempPath[ MAX_PATH + sizeof( L'\0' )];
// Reset the error code.
m_lError = 0L;
// ----------
// Get %TEMP%
// ----------
if( !GetTempPath( MAX_PATH,
wszSystemTempPath )
)
{
m_lError = GetLastError();
EXIT( L"GetTempPath() failed (%d)" );
}
// ----------------------
// Initialize this object
// ----------------------
// Initialize using the temporary path. We must never pass a NULL here,
// or we'll cause an infinite recursion.
if( wszSystemTempPath == NULL )
EXIT( L"Invalid temporary path" );
bSuccess = Initialize( wszSystemTempPath );
// ----
// Exit
// ----
Exit:
return( bSuccess );
}
//+-----------------------------------------------------------------------
//
// Function: CDirectory::Initialize (with an ANSI string)
//
// Synopsis: This function converts the ANSI string to a Unicode
// string, then initializes the object with it.
//
// Inputs: A Unicode string.
//
// Outputs: TRUE if the function succeeds, FALSE otherwise.
//
//+-----------------------------------------------------------------------
BOOL CDirectory::Initialize( LPCSTR szDirectory )
{
// ---------------
// Local Variables
// ---------------
// Assume failure.
BOOL bSuccess = FALSE;
// A buffer for the Unicode path
WCHAR wszDirectory[ MAX_UNICODE_PATH + sizeof( L'\0' )];
// -----
// Begin
// -----
// Initialize the error code.
m_lError = 0L;
// If we were givin a NULL path, use the version of Initialize
// that requires no path.
if( szDirectory == NULL )
{
bSuccess = Initialize();
goto Exit;
}
// Convert the Ansi name to Unicode.
if( m_lError = (long) AnsiToUnicode( szDirectory,
wszDirectory,
strlen( szDirectory )
)
)
{
EXIT( L"Unable to convert directory to Unicode" );
}
// Initialize using the temporary path. We must never pass a NULL here,
// or we'll cause an infinite recursion.
if( wszDirectory == NULL )
EXIT( L"Invalid Directory (internal error)" );
bSuccess = Initialize( wszDirectory );
// ----
// Exit
// ----
Exit:
return( bSuccess );
}
//+-----------------------------------------------------------------------
//
// Function: CDirectory::Initialize (with a Unicode string)
//
// Synopsis: This function is the only form of Initialize
// (there are several variations of the Initialize member)
// which really initializes the object. It stores the
// directory name, and determines the type of filesystem
// on which it resides.
//
// Inputs: A Unicode string.
//
// Outputs: TRUE if the function succeeds, FALSE otherwise.
//
//+-----------------------------------------------------------------------
BOOL CDirectory::Initialize( LPCWSTR wszDirectory )
{
// ---------------
// Local Variables
// ---------------
// Assume failure.
BOOL bSuccess = FALSE;
// Buffers for the root of the path and for the volume name.
WCHAR wszDirectoryRoot[ MAX_PATH + sizeof( L'\0' )];
WCHAR wszVolumeName[ MAX_PATH + sizeof( L'\0' )];
// Parameters to GetVolumeInformation which we won't use.
DWORD dwMaxComponentLength = 0L;
DWORD dwFileSystemFlags = 0L;
// -----
// Begin
// -----
// Initialize the error code.
m_lError = 0L;
// If we were given a NULL path, use the variation of Initialization()
// which does not require one. Note that we will then be called again,
// but this time with a path.
if( wszDirectory == NULL )
{
bSuccess = Initialize();
goto Exit;
}
// Validate the path.
if( wcslen( wszDirectory ) > MAX_PATH )
{
m_lError = wcslen( wszDirectory );
EXIT( L"Input path is too long (%d)\n" );
}
// Save the path to our member buffer
wcscpy( m_wszDirectory, wszDirectory );
// ------------------------
// Get the file system name
// ------------------------
// Get the root path to the directory.
wcscpy( wszDirectoryRoot, wszDirectory );
MakeRoot( wszDirectoryRoot );
// Get the volume information, which will include the filesystem name.
if( !GetVolumeInformation( wszDirectoryRoot, // Root path name.
wszVolumeName, // Buffer for volume name
MAX_PATH, // Length of the above buffer
NULL, // Buffer for serial number
// Longest filename length.
&dwMaxComponentLength,
&dwFileSystemFlags, // Compression, etc.
m_wszFileSystemName,// Buffer for the FS name.
MAX_PATH ) // Length of above buffer
)
{
m_lError = GetLastError();
EXIT( L"GetVolumeInformation() failed" );
}
// Determine the file system type from the name.
if( !wcscmp( m_wszFileSystemName, L"FAT" ))
m_FileSystemType = fstFAT;
else if( !wcscmp( m_wszFileSystemName, L"NTFS" ))
m_FileSystemType = fstNTFS;
else if( !wcscmp( m_wszFileSystemName, L"OFS" ))
m_FileSystemType = fstOFS;
else
m_FileSystemType = fstUnknown;
bSuccess = TRUE;
// ----
// Exit
// ----
Exit:
DisplayErrors( bSuccess, L"CDirectory::Initialize( wszDirectory )" );
return( bSuccess );
}
//
// GetRootLength
//
// This routine was simply copied from private\windows\shell\shelldll\tracker.cxx,
// and should not be modified here.
//
unsigned
CDirectory::GetRootLength(const WCHAR *pwszPath)
{
ULONG cwcRoot = 0;
m_lError = 0L;
if (pwszPath == 0)
pwszPath = L"";
if (*pwszPath == L'\\')
{
// If the first character is a path separator (backslash), this
// must be a UNC drive designator which must be of the form:
// <path-separator><path-separator>(<alnum>+)
// <path-separator>(<alnum>+)<path-separator>
//
// This covers drives like these: \\worf\scratch\ and
// \\savik\win4dev\.
//
pwszPath++;
cwcRoot++;
BOOL fMachine = FALSE;
BOOL fShare = FALSE;
if (*pwszPath == L'\\')
{
cwcRoot++;
pwszPath++;
while (*pwszPath != '\0' && *pwszPath != L'\\')
{
cwcRoot++;
pwszPath++;
fMachine = TRUE;
}
if (*pwszPath == L'\\')
{
cwcRoot++;
pwszPath++;
while (*pwszPath != '\0' && *pwszPath != L'\\')
{
cwcRoot++;
pwszPath++;
fShare = TRUE;
}
// If there weren't any characters in the machine or
// share portions of the UNC name, then the drive
// designator is bogus.
//
if (!fMachine || !fShare)
{
cwcRoot = 0;
}
}
else
{
cwcRoot = 0;
}
}
else
{
cwcRoot = 0;
}
}
else
if (iswalpha(*pwszPath))
{
// If the first character is an alphanumeric, we must have
// a drive designator of this form:
// (<alnum>)+<drive-separator><path-separator>
//
// This covers drives like these: a:\, c:\, etc
//
pwszPath++;
cwcRoot++;
if (*pwszPath == L':')
{
cwcRoot++;
pwszPath++;
}
else
{
cwcRoot = 0;
}
}
// If we have counted one or more characters in the root and these
// are followed by a component separator, we need to add the separator
// to the root length. Otherwise this is not a valid root and we need
// to return a length of zero.
//
if ((cwcRoot > 0) && (*pwszPath == L'\\'))
{
cwcRoot++;
}
else
{
cwcRoot = 0;
}
return (cwcRoot);
}
//
// MakeRoot
//
// This routine was simply copied from private\windows\shell\shelldll\tracker.cxx,
// and should not be modified here.
//
VOID
CDirectory::MakeRoot(WCHAR *pwszPath)
{
unsigned rootlength = GetRootLength(pwszPath);
m_lError = 0L;
if (rootlength)
{
pwszPath[rootlength] = L'\0';
}
}