mirror of
https://github.com/reactos/reactos.git
synced 2024-11-10 08:43:28 +00:00
573 lines
14 KiB
C
Executable file
573 lines
14 KiB
C
Executable file
// Copyright (c) 1998 Mark Russinovich
|
|
// Systems Internals
|
|
// http://www.sysinternals.com
|
|
#include <stdio.h>
|
|
#include <stdlib.h>
|
|
#include <string.h>
|
|
#include <windows.h>
|
|
#include <winternl.h>
|
|
#include <fmifs/fmifs.h>
|
|
#include <tchar.h>
|
|
#include "resource.h"
|
|
|
|
// Globals
|
|
BOOL Error = FALSE;
|
|
|
|
// switches
|
|
BOOL QuickFormat = FALSE;
|
|
DWORD ClusterSize = 0;
|
|
BOOL CompressDrive = FALSE;
|
|
BOOL GotALabel = FALSE;
|
|
LPTSTR Label = _T("");
|
|
LPTSTR Drive = NULL;
|
|
LPTSTR Format = _T("FAT");
|
|
|
|
TCHAR RootDirectory[MAX_PATH];
|
|
TCHAR LabelString[12];
|
|
|
|
//
|
|
// Size array
|
|
//
|
|
typedef struct {
|
|
TCHAR SizeString[16];
|
|
DWORD ClusterSize;
|
|
} SIZEDEFINITION, *PSIZEDEFINITION;
|
|
|
|
SIZEDEFINITION LegalSizes[] = {
|
|
{ _T("512"), 512 },
|
|
{ _T("1024"), 1024 },
|
|
{ _T("2048"), 2048 },
|
|
{ _T("4096"), 4096 },
|
|
{ _T("8192"), 8192 },
|
|
{ _T("16K"), 16384 },
|
|
{ _T("32K"), 32768 },
|
|
{ _T("64K"), 65536 },
|
|
{ _T("128K"), 65536 * 2 },
|
|
{ _T("256K"), 65536 * 4 },
|
|
{ _T(""), 0 },
|
|
};
|
|
|
|
|
|
//----------------------------------------------------------------------
|
|
//
|
|
// PrintWin32Error
|
|
//
|
|
// Takes the win32 error code and prints the text version.
|
|
//
|
|
//----------------------------------------------------------------------
|
|
static VOID PrintWin32Error( LPTSTR Message, DWORD ErrorCode )
|
|
{
|
|
LPTSTR lpMsgBuf;
|
|
|
|
FormatMessage( FORMAT_MESSAGE_ALLOCATE_BUFFER | FORMAT_MESSAGE_FROM_SYSTEM,
|
|
NULL, ErrorCode,
|
|
MAKELANGID(LANG_NEUTRAL, SUBLANG_DEFAULT),
|
|
(LPTSTR)&lpMsgBuf, 0, NULL );
|
|
|
|
_tprintf(_T("%s: %s\n"), Message, lpMsgBuf );
|
|
LocalFree( lpMsgBuf );
|
|
}
|
|
|
|
|
|
//----------------------------------------------------------------------
|
|
//
|
|
// ParseCommandLine
|
|
//
|
|
// Get the switches.
|
|
//
|
|
//----------------------------------------------------------------------
|
|
static int ParseCommandLine( int argc, TCHAR *argv[] )
|
|
{
|
|
int i, j;
|
|
BOOLEAN gotFormat = FALSE;
|
|
BOOLEAN gotQuick = FALSE;
|
|
BOOLEAN gotSize = FALSE;
|
|
BOOLEAN gotLabel = FALSE;
|
|
BOOLEAN gotCompressed = FALSE;
|
|
|
|
|
|
for( i = 1; i < argc; i++ ) {
|
|
|
|
switch( argv[i][0] ) {
|
|
|
|
case '-':
|
|
case '/':
|
|
|
|
if( !_tcsnicmp( &argv[i][1], _T("FS:"), 3 )) {
|
|
|
|
if( gotFormat) return -1;
|
|
Format = &argv[i][4];
|
|
gotFormat = TRUE;
|
|
|
|
|
|
} else if( !_tcsnicmp( &argv[i][1], _T("A:"), 2 )) {
|
|
|
|
if( gotSize ) return -1;
|
|
j = 0;
|
|
while( LegalSizes[j].ClusterSize &&
|
|
_tcsicmp( LegalSizes[j].SizeString, &argv[i][3] )) j++;
|
|
|
|
if( !LegalSizes[j].ClusterSize ) return i;
|
|
ClusterSize = LegalSizes[j].ClusterSize;
|
|
gotSize = TRUE;
|
|
|
|
} else if( ! _tcsnicmp( &argv[i][1], _T("V:"), 2 )) {
|
|
|
|
if( gotLabel ) return -1;
|
|
Label = &argv[i][3];
|
|
gotLabel = TRUE;
|
|
GotALabel = TRUE;
|
|
|
|
} else if( !_tcsicmp( &argv[i][1], _T("Q") )) {
|
|
|
|
if( gotQuick ) return -1;
|
|
QuickFormat = TRUE;
|
|
gotQuick = TRUE;
|
|
|
|
} else if( !_tcsicmp( &argv[i][1], _T("C") )) {
|
|
|
|
if( gotCompressed ) return -1;
|
|
CompressDrive = TRUE;
|
|
gotCompressed = TRUE;
|
|
|
|
} else return i;
|
|
break;
|
|
|
|
default:
|
|
|
|
if( Drive ) return i;
|
|
if( argv[i][1] != _T(':') ) return i;
|
|
|
|
Drive = argv[i];
|
|
break;
|
|
}
|
|
}
|
|
return 0;
|
|
}
|
|
|
|
//----------------------------------------------------------------------
|
|
//
|
|
// FormatExCallback
|
|
//
|
|
// The file system library will call us back with commands that we
|
|
// can interpret. If we wanted to halt the chkdsk we could return FALSE.
|
|
//
|
|
//----------------------------------------------------------------------
|
|
BOOLEAN WINAPI
|
|
FormatExCallback (
|
|
CALLBACKCOMMAND Command,
|
|
ULONG Modifier,
|
|
PVOID Argument)
|
|
{
|
|
PDWORD percent;
|
|
PTEXTOUTPUT output;
|
|
PBOOLEAN status;
|
|
TCHAR szMsg[RC_STRING_MAX_SIZE];
|
|
|
|
//
|
|
// We get other types of commands, but we don't have to pay attention to them
|
|
//
|
|
switch( Command ) {
|
|
|
|
case PROGRESS:
|
|
percent = (PDWORD) Argument;
|
|
LoadString( GetModuleHandle(NULL), STRING_COMPLETE, (LPTSTR) szMsg,RC_STRING_MAX_SIZE);
|
|
_tprintf(szMsg, *percent);
|
|
break;
|
|
|
|
case OUTPUT:
|
|
output = (PTEXTOUTPUT) Argument;
|
|
fprintf(stdout, "%s", output->Output);
|
|
break;
|
|
|
|
case DONE:
|
|
status = (PBOOLEAN) Argument;
|
|
if( *status == FALSE ) {
|
|
|
|
LoadString( GetModuleHandle(NULL), STRING_FORMAT_FAIL, (LPTSTR) szMsg,RC_STRING_MAX_SIZE);
|
|
_tprintf("%s", szMsg);
|
|
Error = TRUE;
|
|
}
|
|
break;
|
|
case DONEWITHSTRUCTURE:
|
|
case UNKNOWN2:
|
|
case UNKNOWN3:
|
|
case UNKNOWN4:
|
|
case UNKNOWN5:
|
|
case INSUFFICIENTRIGHTS:
|
|
case FSNOTSUPPORTED:
|
|
case VOLUMEINUSE:
|
|
case UNKNOWN9:
|
|
case UNKNOWNA:
|
|
case UNKNOWNC:
|
|
case UNKNOWND:
|
|
case STRUCTUREPROGRESS:
|
|
case CLUSTERSIZETOOSMALL:
|
|
LoadString( GetModuleHandle(NULL), STRING_NO_SUPPORT, (LPTSTR) szMsg,RC_STRING_MAX_SIZE);
|
|
_tprintf("%s", szMsg);
|
|
return FALSE;
|
|
}
|
|
return TRUE;
|
|
}
|
|
|
|
|
|
//----------------------------------------------------------------------
|
|
//
|
|
// LoadFMIFSEntryPoints
|
|
//
|
|
// Loads FMIFS.DLL and locates the entry point(s) we are going to use
|
|
//
|
|
//----------------------------------------------------------------------
|
|
BOOLEAN LoadFMIFSEntryPoints()
|
|
{
|
|
HMODULE hFmifs = LoadLibrary( _T("fmifs.dll") );
|
|
if( !(void*) GetProcAddress( hFmifs, "FormatEx" ) ) {
|
|
|
|
return FALSE;
|
|
}
|
|
|
|
if( !((void *) GetProcAddress( hFmifs,
|
|
"EnableVolumeCompression" )) ) {
|
|
|
|
return FALSE;
|
|
}
|
|
|
|
if( !((void *) GetProcAddress( hFmifs,
|
|
"QueryAvailableFileSystemFormat" )) ) {
|
|
|
|
return FALSE;
|
|
}
|
|
|
|
return TRUE;
|
|
}
|
|
|
|
|
|
//----------------------------------------------------------------------
|
|
//
|
|
// Usage
|
|
//
|
|
// Tell the user how to use the program
|
|
//
|
|
//----------------------------------------------------------------------
|
|
static VOID Usage( LPTSTR ProgramName )
|
|
{
|
|
TCHAR szMsg[RC_STRING_MAX_SIZE];
|
|
TCHAR szFormats[MAX_PATH];
|
|
#ifndef UNICODE
|
|
TCHAR szFormatA[MAX_PATH];
|
|
#endif
|
|
WCHAR szFormatW[MAX_PATH];
|
|
DWORD Index = 0;
|
|
BYTE dummy;
|
|
BOOLEAN lastestVersion;
|
|
|
|
LoadString( GetModuleHandle(NULL), STRING_HELP, (LPTSTR) szMsg,RC_STRING_MAX_SIZE);
|
|
if (!LoadFMIFSEntryPoints())
|
|
{
|
|
_tprintf(szMsg, ProgramName, _T(""));
|
|
return;
|
|
}
|
|
|
|
szFormats[0] = 0;
|
|
while (QueryAvailableFileSystemFormat(Index++, szFormatW, &dummy, &dummy, &lastestVersion))
|
|
{
|
|
if (!lastestVersion)
|
|
continue;
|
|
if (szFormats[0])
|
|
_tcscat(szFormats, _T(", "));
|
|
#ifdef UNICODE
|
|
_tcscat(szFormats, szFormatW);
|
|
#else
|
|
if (0 != WideCharToMultiByte(CP_ACP, 0, szFormatW, -1, szFormatA, sizeof(szFormatA), NULL, NULL))
|
|
_tcscat(szFormats, szFormatA);
|
|
#endif
|
|
}
|
|
_tprintf(szMsg, ProgramName, szFormats);
|
|
}
|
|
|
|
|
|
//----------------------------------------------------------------------
|
|
//
|
|
// WMain
|
|
//
|
|
// Engine. Just get command line switches and fire off a format. This
|
|
// could also be done in a GUI like Explorer does when you select a
|
|
// drive and run a check on it.
|
|
//
|
|
// We do this in UNICODE because the chkdsk command expects PWCHAR
|
|
// arguments.
|
|
//
|
|
//----------------------------------------------------------------------
|
|
int
|
|
_tmain(int argc, TCHAR *argv[])
|
|
{
|
|
int badArg;
|
|
DWORD media = FMIFS_HARDDISK;
|
|
DWORD driveType;
|
|
TCHAR fileSystem[1024];
|
|
TCHAR volumeName[1024];
|
|
TCHAR input[1024];
|
|
DWORD serialNumber;
|
|
DWORD flags, maxComponent;
|
|
ULARGE_INTEGER freeBytesAvailableToCaller, totalNumberOfBytes, totalNumberOfFreeBytes;
|
|
#ifndef UNICODE
|
|
WCHAR RootDirectoryW[MAX_PATH], FormatW[MAX_PATH], LabelW[MAX_PATH];
|
|
#endif
|
|
TCHAR szMsg[RC_STRING_MAX_SIZE];
|
|
|
|
//
|
|
// Get function pointers
|
|
//
|
|
if( !LoadFMIFSEntryPoints()) {
|
|
LoadString( GetModuleHandle(NULL), STRING_FMIFS_FAIL, (LPTSTR) szMsg,RC_STRING_MAX_SIZE);
|
|
_tprintf("%s", szMsg);
|
|
return -1;
|
|
}
|
|
|
|
//
|
|
// Parse command line
|
|
//
|
|
if( (badArg = ParseCommandLine( argc, argv ))) {
|
|
|
|
LoadString( GetModuleHandle(NULL), STRING_UNKNOW_ARG, (LPTSTR) szMsg,RC_STRING_MAX_SIZE);
|
|
_tprintf(szMsg, argv[badArg] );
|
|
|
|
Usage(argv[0]);
|
|
return -1;
|
|
}
|
|
|
|
//
|
|
// Get the drive's format
|
|
//
|
|
if( !Drive ) {
|
|
|
|
LoadString( GetModuleHandle(NULL), STRING_DRIVE_PARM, (LPTSTR) szMsg,RC_STRING_MAX_SIZE);
|
|
_tprintf(szMsg);
|
|
Usage( argv[0] );
|
|
return -1;
|
|
|
|
} else {
|
|
|
|
_tcscpy( RootDirectory, Drive );
|
|
}
|
|
RootDirectory[2] = _T('\\');
|
|
RootDirectory[3] = _T('\0');
|
|
|
|
//
|
|
// See if the drive is removable or not
|
|
//
|
|
driveType = GetDriveType( RootDirectory );
|
|
|
|
if( driveType == 0 ) {
|
|
LoadString( GetModuleHandle(NULL), STRING_ERROR_DRIVE_TYPE, (LPTSTR) szMsg,RC_STRING_MAX_SIZE);
|
|
PrintWin32Error( szMsg, GetLastError());
|
|
return -1;
|
|
}
|
|
else if ( driveType == 1 )
|
|
{
|
|
LoadString( GetModuleHandle(NULL), STRING_NO_VOLUME, (LPTSTR) szMsg,RC_STRING_MAX_SIZE);
|
|
PrintWin32Error( szMsg, GetLastError());
|
|
return -1;
|
|
}
|
|
|
|
if( driveType != DRIVE_FIXED ) {
|
|
LoadString( GetModuleHandle(NULL), STRING_INSERT_DISK, (LPTSTR) szMsg,RC_STRING_MAX_SIZE);
|
|
_tprintf(szMsg, RootDirectory[0] );
|
|
_fgetts( input, sizeof(input)/2, stdin );
|
|
|
|
media = FMIFS_FLOPPY;
|
|
}
|
|
|
|
//
|
|
// Determine the drive's file system format
|
|
//
|
|
if( !GetVolumeInformation( RootDirectory,
|
|
volumeName, sizeof(volumeName)/2,
|
|
&serialNumber, &maxComponent, &flags,
|
|
fileSystem, sizeof(fileSystem)/2)) {
|
|
|
|
LoadString( GetModuleHandle(NULL), STRING_NO_VOLUME, (LPTSTR) szMsg,RC_STRING_MAX_SIZE);
|
|
PrintWin32Error( szMsg, GetLastError());
|
|
return -1;
|
|
}
|
|
|
|
if( !GetDiskFreeSpaceEx( RootDirectory,
|
|
&freeBytesAvailableToCaller,
|
|
&totalNumberOfBytes,
|
|
&totalNumberOfFreeBytes )) {
|
|
|
|
LoadString( GetModuleHandle(NULL), STRING_NO_VOLUME_SIZE, (LPTSTR) szMsg,RC_STRING_MAX_SIZE);
|
|
PrintWin32Error( szMsg, GetLastError());
|
|
return -1;
|
|
}
|
|
LoadString( GetModuleHandle(NULL), STRING_FILESYSTEM, (LPTSTR) szMsg,RC_STRING_MAX_SIZE);
|
|
_tprintf(szMsg, fileSystem );
|
|
|
|
//
|
|
// Make sure they want to do this
|
|
//
|
|
if( driveType == DRIVE_FIXED ) {
|
|
|
|
if( volumeName[0] ) {
|
|
|
|
while(1 ) {
|
|
|
|
LoadString( GetModuleHandle(NULL), STRING_LABEL_NAME_EDIT, (LPTSTR) szMsg,RC_STRING_MAX_SIZE);
|
|
_tprintf(szMsg, RootDirectory[0] );
|
|
_fgetts( input, sizeof(input)/2, stdin );
|
|
input[ _tcslen( input ) - 1] = 0;
|
|
|
|
if( !_tcsicmp( input, volumeName )) {
|
|
|
|
break;
|
|
}
|
|
LoadString( GetModuleHandle(NULL), STRING_ERROR_LABEL, (LPTSTR) szMsg,RC_STRING_MAX_SIZE);
|
|
_tprintf("%s", szMsg);
|
|
}
|
|
}
|
|
|
|
LoadString( GetModuleHandle(NULL), STRING_YN_FORMAT, (LPTSTR) szMsg,RC_STRING_MAX_SIZE);
|
|
_tprintf(szMsg, RootDirectory[0] );
|
|
|
|
LoadString( GetModuleHandle(NULL), STRING_YES_NO_FAQ, (LPTSTR) szMsg,RC_STRING_MAX_SIZE);
|
|
|
|
while( 1 ) {
|
|
_fgetts( input, sizeof(input)/2, stdin );
|
|
if(_strnicmp(&input[0],&szMsg[0],1) == 0) break;
|
|
if(_strnicmp(&input[0],&szMsg[1],1) == 0) {
|
|
_tprintf(_T("\n"));
|
|
return 0;
|
|
}
|
|
}
|
|
media = FMIFS_HARDDISK;
|
|
}
|
|
|
|
//
|
|
// Tell the user we're doing a long format if appropriate
|
|
//
|
|
if( !QuickFormat ) {
|
|
|
|
LoadString( GetModuleHandle(NULL), STRING_VERIFYING, (LPTSTR) szMsg,RC_STRING_MAX_SIZE);
|
|
|
|
if( totalNumberOfBytes.QuadPart > 1024*1024*10 ) {
|
|
|
|
_tprintf(_T("%s %luM\n"),szMsg, (DWORD) (totalNumberOfBytes.QuadPart/(1024*1024)));
|
|
|
|
} else {
|
|
|
|
_tprintf(_T("%s %.1fM\n"),szMsg,
|
|
((float)(LONGLONG)totalNumberOfBytes.QuadPart)/(float)(1024.0*1024.0));
|
|
}
|
|
} else {
|
|
|
|
LoadString( GetModuleHandle(NULL), STRING_FAST_FMT, (LPTSTR) szMsg,RC_STRING_MAX_SIZE);
|
|
if( totalNumberOfBytes.QuadPart > 1024*1024*10 ) {
|
|
|
|
_tprintf(_T("%s %luM\n"),szMsg, (DWORD) (totalNumberOfBytes.QuadPart/(1024*1024)));
|
|
|
|
} else {
|
|
|
|
_tprintf(_T("%s %.2fM\n"),szMsg,
|
|
((float)(LONGLONG)totalNumberOfBytes.QuadPart)/(float)(1024.0*1024.0));
|
|
}
|
|
LoadString( GetModuleHandle(NULL), STRING_CREATE_FSYS, (LPTSTR) szMsg,RC_STRING_MAX_SIZE);
|
|
_tprintf("%s", szMsg);
|
|
}
|
|
|
|
//
|
|
// Format away!
|
|
//
|
|
#ifndef UNICODE
|
|
MultiByteToWideChar(CP_ACP, 0, RootDirectory, -1, RootDirectoryW, MAX_PATH);
|
|
MultiByteToWideChar(CP_ACP, 0, Format, -1, FormatW, MAX_PATH);
|
|
MultiByteToWideChar(CP_ACP, 0, Label, -1, LabelW, MAX_PATH);
|
|
FormatEx( RootDirectoryW, media, FormatW, LabelW, QuickFormat,
|
|
ClusterSize, FormatExCallback );
|
|
#else
|
|
FormatEx( RootDirectory, media, Format, Label, QuickFormat,
|
|
ClusterSize, FormatExCallback );
|
|
#endif
|
|
if( Error ) return -1;
|
|
LoadString( GetModuleHandle(NULL), STRING_FMT_COMPLETE, (LPTSTR) szMsg,RC_STRING_MAX_SIZE);
|
|
_tprintf("%s", szMsg);
|
|
|
|
//
|
|
// Enable compression if desired
|
|
//
|
|
if( CompressDrive ) {
|
|
|
|
#ifndef UNICODE
|
|
MultiByteToWideChar(CP_ACP, 0, RootDirectory, -1, RootDirectoryW, MAX_PATH);
|
|
if( !EnableVolumeCompression( RootDirectoryW, TRUE )) {
|
|
#else
|
|
if( !EnableVolumeCompression( RootDirectory, TRUE )) {
|
|
#endif
|
|
|
|
LoadString( GetModuleHandle(NULL), STRING_VOL_COMPRESS, (LPTSTR) szMsg,RC_STRING_MAX_SIZE);
|
|
_tprintf("%s", szMsg);
|
|
}
|
|
}
|
|
|
|
//
|
|
// Get the label if we don't have it
|
|
//
|
|
if( !GotALabel ) {
|
|
|
|
LoadString( GetModuleHandle(NULL), STRING_ENTER_LABEL, (LPTSTR) szMsg,RC_STRING_MAX_SIZE);
|
|
_tprintf("%s", szMsg);
|
|
_fgetts( input, sizeof(LabelString)/2, stdin );
|
|
|
|
input[ _tcslen(input)-1] = 0;
|
|
if( !SetVolumeLabel( RootDirectory, input )) {
|
|
|
|
LoadString( GetModuleHandle(NULL), STRING_NO_LABEL, (LPTSTR) szMsg,RC_STRING_MAX_SIZE);
|
|
PrintWin32Error(szMsg, GetLastError());
|
|
return -1;
|
|
}
|
|
}
|
|
|
|
if( !GetVolumeInformation( RootDirectory,
|
|
volumeName, sizeof(volumeName)/2,
|
|
&serialNumber, &maxComponent, &flags,
|
|
fileSystem, sizeof(fileSystem)/2)) {
|
|
|
|
LoadString( GetModuleHandle(NULL), STRING_NO_VOLUME, (LPTSTR) szMsg,RC_STRING_MAX_SIZE);
|
|
PrintWin32Error( szMsg, GetLastError());
|
|
return -1;
|
|
}
|
|
|
|
//
|
|
// Print out some stuff including the formatted size
|
|
//
|
|
if( !GetDiskFreeSpaceEx( RootDirectory,
|
|
&freeBytesAvailableToCaller,
|
|
&totalNumberOfBytes,
|
|
&totalNumberOfFreeBytes )) {
|
|
|
|
LoadString( GetModuleHandle(NULL), STRING_NO_VOLUME_SIZE, (LPTSTR) szMsg,RC_STRING_MAX_SIZE);
|
|
PrintWin32Error(szMsg, GetLastError());
|
|
return -1;
|
|
}
|
|
|
|
LoadString( GetModuleHandle(NULL), STRING_FREE_SPACE, (LPTSTR) szMsg,RC_STRING_MAX_SIZE);
|
|
_tprintf(szMsg, totalNumberOfBytes.QuadPart, totalNumberOfFreeBytes.QuadPart );
|
|
|
|
//
|
|
// Get the drive's serial number
|
|
//
|
|
if( !GetVolumeInformation( RootDirectory,
|
|
volumeName, sizeof(volumeName)/2,
|
|
&serialNumber, &maxComponent, &flags,
|
|
fileSystem, sizeof(fileSystem)/2)) {
|
|
|
|
LoadString( GetModuleHandle(NULL), STRING_NO_VOLUME, (LPTSTR) szMsg,RC_STRING_MAX_SIZE);
|
|
PrintWin32Error( szMsg, GetLastError());
|
|
return -1;
|
|
}
|
|
LoadString( GetModuleHandle(NULL), STRING_SERIAL_NUMBER, (LPTSTR) szMsg,RC_STRING_MAX_SIZE);
|
|
_tprintf(szMsg, (unsigned int)(serialNumber >> 16),
|
|
(unsigned int)(serialNumber & 0xFFFF) );
|
|
|
|
return 0;
|
|
}
|
|
|