Source code of Windows XP (NT5)
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.
 
 
 
 
 
 

473 lines
8.8 KiB

/*++
Copyright (c) 1997 Microsoft Corporation
Module Name:
delrp.c
This module contains a simple program to blatantly delete a reparse point
of any kind.
Author:
Felipe Cabrera (Cabrera) 11-Jul-1997
Revision History:
--*/
#define UNICODE
#define _UNICODE
#include <stdio.h>
#include <stdlib.h> // exit
#include <io.h> // _get_osfhandle
#include <nt.h>
#include <ntrtl.h>
#include <nturtl.h>
#include <ntioapi.h>
#include <windows.h>
#include <locale.h> // setlocale
//
//
// Functions forward referenced.
//
void
SzToWsz (
OUT WCHAR *Unicode,
IN char *Ansi
);
void
ScanArgs (
int argc,
char **argv
);
void
__cdecl
printmessage (
DWORD messageID,
...
);
void
__cdecl
DisplayMsg (
DWORD MsgNum,
...
);
int
FileIsConsole (
int fh
);
//
// I/O stream handles and variables.
//
HANDLE hInput;
HANDLE hOutput;
HANDLE hError;
#define STDIN 0
#define STDOUT 1
#define STDERR 2
BOOL ConsoleInput;
BOOL ConsoleOutput;
BOOL ConsoleError;
//
// Core control state vars
//
BOOLEAN NeedHelp;
#include "delrpmsg.h"
TCHAR Buf[1024]; // for displaying stuff
//
// Main
//
void
__cdecl
main(
int argc,
char **argv
)
/*++
Routine Description:
Main procedure for pentnt.
First, we call a series of routines that build a state vector
in some booleans.
We'll then act on these control variables:
NeedHelp - User has asked for help, or made a command error
Arguments:
argc - count of arguments, including the name of our proggram
argv - argument list - see command line syntax above
Return Value:
Exit(0) - the file was deleted
Exit(1) - a problem ocurred.
--*/
{
CHAR lBuf[16];
DWORD dwCodePage;
LANGID LangId;
NTSTATUS Status = STATUS_SUCCESS;
OBJECT_ATTRIBUTES ObjectAttributes;
HANDLE Handle;
UNICODE_STRING UnicodeName;
IO_STATUS_BLOCK IoStatusBlock;
BOOLEAN TranslationStatus;
PVOID FreeBuffer;
FILE_DISPOSITION_INFORMATION Disposition = {TRUE};
WCHAR WFileName[MAX_PATH];
//
// Build up state vector in global booleans.
//
ScanArgs(argc, argv);
//
// printf( "argc = %d NeedHelp = %d\n", argc, NeedHelp );
//
//
// Since FormatMessage checks the current TEB's locale, and the Locale for
// CHCP is initialized when the message class is initialized, the TEB has to
// be updated after the code page is changed successfully.
// Why are we doing this, you ask. Well, the FE guys have plans to add
// more than one set of language resources to this module, but not all
// the possible resources. So this limited set is what they plan for.
// If FormatMessage can't find the right language, it falls back to
// something hopefully useful.
//
dwCodePage = GetConsoleOutputCP();
sprintf(lBuf, ".%d", dwCodePage);
switch( dwCodePage )
{
case 437:
LangId = MAKELANGID( LANG_ENGLISH, SUBLANG_ENGLISH_US );
break;
case 932:
LangId = MAKELANGID( LANG_JAPANESE, SUBLANG_DEFAULT );
break;
case 949:
LangId = MAKELANGID( LANG_KOREAN, SUBLANG_KOREAN );
break;
case 936:
LangId = MAKELANGID( LANG_CHINESE, SUBLANG_CHINESE_SIMPLIFIED );
break;
case 950:
LangId = MAKELANGID( LANG_CHINESE, SUBLANG_CHINESE_TRADITIONAL );
break;
default:
LangId = MAKELANGID( LANG_NEUTRAL, SUBLANG_DEFAULT );
lBuf[0] = '\0';
break;
}
SetThreadLocale( MAKELCID(LangId, SORT_DEFAULT) );
setlocale(LC_ALL, lBuf);
//
// Set the appropriate handles.
//
hInput = GetStdHandle(STD_INPUT_HANDLE);
ConsoleInput = FileIsConsole(STDIN);
hOutput = GetStdHandle(STD_OUTPUT_HANDLE);
ConsoleOutput = FileIsConsole(STDOUT);
hError = GetStdHandle(STD_ERROR_HANDLE);
ConsoleError = FileIsConsole(STDERR);
//
// OK, we know the state of the command, do work
//
//
// If they asked for help, or did something that indicates they don't
// understand how the program works, print help and exit.
//
if (NeedHelp) {
printmessage( MSG_DELRP_HELP );
exit(1);
}
//
// Change the string to Unicode and pass down to open the file.
//
SzToWsz( WFileName, argv[1] );
TranslationStatus = RtlDosPathNameToNtPathName_U(
WFileName,
&UnicodeName,
NULL,
NULL
);
if (!TranslationStatus) {
printmessage( MSG_DELRP_WRONG_NAME );
exit(1);
}
FreeBuffer = UnicodeName.Buffer;
InitializeObjectAttributes(
&ObjectAttributes,
&UnicodeName,
OBJ_CASE_INSENSITIVE,
NULL,
NULL
);
//
// printf( "Transformed unicode str is %Z\n", &UnicodeName );
//
//
// Open the file for delete access.
// Inhibit the reparse behavior using FILE_OPEN_REPARSE_POINT.
// This will get a handle to the entity whether the appropriate filter is or not in place.
//
Status = NtOpenFile(
&Handle,
(ACCESS_MASK)DELETE,
&ObjectAttributes,
&IoStatusBlock,
FILE_SHARE_DELETE | FILE_SHARE_READ | FILE_SHARE_WRITE,
FILE_OPEN_FOR_BACKUP_INTENT | FILE_OPEN_REPARSE_POINT
);
if (!NT_SUCCESS(Status)) {
SzToWsz( WFileName, argv[1] );
swprintf(&Buf[0], TEXT("%s"), WFileName);
DisplayMsg(MSG_DELRP_OPEN_FAILED_NL, Buf);
// printmessage( MSG_DELRP_OPEN_FAILED );
RtlFreeHeap(RtlProcessHeap(), 0, FreeBuffer);
exit(1);
}
RtlFreeHeap(RtlProcessHeap(), 0, FreeBuffer);
//
// Delete the file
//
Status = NtSetInformationFile(
Handle,
&IoStatusBlock,
&Disposition,
sizeof(Disposition),
FileDispositionInformation
);
NtClose(Handle);
if (!NT_SUCCESS(Status)) {
printmessage( MSG_DELRP_DELETE_FAILED );
exit(1);
}
exit(0);
} // main
VOID
ScanArgs(
int argc,
char **argv
)
/*++
Routine Description:
ScanArgs - parse command line arguments, and set control flags
to reflect what we find.
Sets NeedHelp.
Arguments:
argc - count of command line args
argv - argument vector
Return Value:
--*/
{
int i;
NeedHelp = FALSE;
if ((argc == 1) ||
(argc > 3)) {
NeedHelp = TRUE;
goto done;
}
//
// At this point argc == 2
//
if ((argv[1][0] == '/') &&
(argv[1][1] == '?') &&
(strlen(argv[1]) == 2)) {
NeedHelp = TRUE;
goto done;
}
done:
return;
} // ScanArgs
//
// Changing a file name to wide characters.
//
void
SzToWsz (
OUT WCHAR *Unicode,
IN char *Ansi
)
{
while (*Unicode++ = *Ansi++)
;
return;
} // SzToWsz
//
// Call FormatMessage and dump the result. All messages to Stdout
//
void
__cdecl
printmessage (
DWORD messageID,
...
)
{
unsigned short messagebuffer[4096];
va_list ap;
va_start(ap, messageID);
FormatMessage(FORMAT_MESSAGE_FROM_HMODULE, NULL, messageID, 0,
messagebuffer, 4095, &ap);
wprintf(messagebuffer);
va_end(ap);
} // printmessage
TCHAR DisplayBuffer[4096];
CHAR DisplayBuffer2[4096];
void
__cdecl
DisplayMsg (
DWORD MsgNum,
...
)
{
DWORD len, bytes_written;
BOOL success;
DWORD status;
va_list ap;
va_start(ap, MsgNum);
len = FormatMessage(FORMAT_MESSAGE_FROM_HMODULE, NULL, MsgNum, 0,
DisplayBuffer, 4096, &ap);
if (ConsoleOutput) {
success = WriteConsole(hOutput, (LPVOID)DisplayBuffer, len,
&bytes_written, NULL);
} else {
CharToOem(DisplayBuffer, DisplayBuffer2);
success = WriteFile(hOutput, (LPVOID)DisplayBuffer2, len,
&bytes_written, NULL);
}
if (!success || bytes_written != len) {
status = GetLastError();
}
va_end(ap);
} // DisplayMsg
int
FileIsConsole(int fh)
{
unsigned htype;
DWORD dwMode;
HANDLE hFile;
hFile = (HANDLE)_get_osfhandle(fh);
htype = GetFileType(hFile);
htype &= ~FILE_TYPE_REMOTE;
if (FILE_TYPE_CHAR == htype) {
switch (fh) {
case STDIN:
hFile = GetStdHandle(STD_INPUT_HANDLE);
break;
case STDOUT:
hFile = GetStdHandle(STD_OUTPUT_HANDLE);
break;
case STDERR:
hFile = GetStdHandle(STD_ERROR_HANDLE);
break;
}
if (GetConsoleMode(hFile, &dwMode)) {
return TRUE;
}
}
return FALSE;
} // FileIsConsole