?? filedisk.c
字號:
#include <windows.h>
#include <winioctl.h>
#include <stdio.h>
#include <stdlib.h>
#ifndef _FILE_DISK_
#define _FILE_DISK_
#ifndef __T
#ifdef _NTDDK_
#define __T(x) L ## x
#else
#define __T(x) x
#endif
#endif
#ifndef _T
#define _T(x) __T(x)
#endif
#define DEVICE_BASE_NAME _T("\\FileDisk")
#define DEVICE_DIR_NAME _T("\\Device") DEVICE_BASE_NAME
#define DEVICE_NAME_PREFIX DEVICE_DIR_NAME DEVICE_BASE_NAME
#define FILE_DEVICE_FILE_DISK 0x8000
#define IOCTL_FILE_DISK_OPEN_FILE CTL_CODE(FILE_DEVICE_FILE_DISK, 0x800, METHOD_BUFFERED, FILE_READ_ACCESS | FILE_WRITE_ACCESS)
#define IOCTL_FILE_DISK_CLOSE_FILE CTL_CODE(FILE_DEVICE_FILE_DISK, 0x801, METHOD_BUFFERED, FILE_READ_ACCESS | FILE_WRITE_ACCESS)
#define IOCTL_FILE_DISK_QUERY_FILE CTL_CODE(FILE_DEVICE_FILE_DISK, 0x802, METHOD_BUFFERED, FILE_READ_ACCESS)
#define FSCTL_LOCK_VOLUME CTL_CODE(FILE_DEVICE_FILE_SYSTEM, 6, METHOD_BUFFERED, FILE_ANY_ACCESS)
#define FSCTL_DISMOUNT_VOLUME CTL_CODE(FILE_DEVICE_FILE_SYSTEM, 8, METHOD_BUFFERED, FILE_ANY_ACCESS)
#define FSCTL_UNLOCK_VOLUME CTL_CODE(FILE_DEVICE_FILE_SYSTEM, 7, METHOD_BUFFERED, FILE_ANY_ACCESS)
typedef struct _OPEN_FILE_INFORMATION {
LARGE_INTEGER FileSize;
BOOLEAN ReadOnly;
USHORT FileNameLength;
UCHAR FileName[1];
} OPEN_FILE_INFORMATION, *POPEN_FILE_INFORMATION;
#endif
int FileDiskSyntax(void)
{
fprintf(stderr, "syntax:\n");
fprintf(stderr, "filedisk /mount <devicenumber> <filename> [size[k|M|G] | /ro | /cd] <drive:>\n");
fprintf(stderr, "filedisk /umount <drive:>\n");
fprintf(stderr, "filedisk /status <drive:>\n");
fprintf(stderr, "\n");
fprintf(stderr, "filename formats:\n");
fprintf(stderr, " c:\\path\\filedisk.img\n");
fprintf(stderr, " \\Device\\Harddisk0\\Partition1\\path\\filedisk.img\n");
fprintf(stderr, " \\\\server\\share\\path\\filedisk.img\n");
fprintf(stderr, "\n");
fprintf(stderr, "example:\n");
fprintf(stderr, "filedisk /mount 0 c:\\temp\\filedisk.img 8M f:\n");
fprintf(stderr, "filedisk /mount 1 c:\\temp\\cdimage.iso /cd i:\n");
fprintf(stderr, "filedisk /umount f:\n");
fprintf(stderr, "filedisk /umount i:\n");
return -1;
}
void PrintLastError(char* Prefix)
{
LPVOID lpMsgBuf;
FormatMessage(
FORMAT_MESSAGE_ALLOCATE_BUFFER |
FORMAT_MESSAGE_FROM_SYSTEM |
FORMAT_MESSAGE_IGNORE_INSERTS,
NULL,
GetLastError(),
0,
(LPTSTR) &lpMsgBuf,
0,
NULL
);
//FORMAT_MESSAGE_ALLOCATE_BUFFER在windows.h中定義為0x00000100
//FORMAT_MESSAGE_FROM_SYSTEM在windows.h中定義為0x00001000
//FORMAT_MESSAGE_IGNORE_INSERTS在windows.h中定義為0x00000200
fprintf(stderr, "%s %s", Prefix, (LPTSTR) lpMsgBuf);
LocalFree(lpMsgBuf);
}
int
FileDiskMount(
int DeviceNumber,
POPEN_FILE_INFORMATION OpenFileInformation,
char DriveLetter,
BOOLEAN CdImage
)
{
char VolumeName[] = "\\\\.\\ :";
char DeviceName[255];
HANDLE Device;
DWORD BytesReturned;
VolumeName[4] = DriveLetter;
Device = CreateFile(
VolumeName,
GENERIC_READ | GENERIC_WRITE,
FILE_SHARE_READ | FILE_SHARE_WRITE,
NULL,
OPEN_EXISTING,
FILE_FLAG_NO_BUFFERING,
NULL
);
//FILE_SHARE_READ在ntddk.h中定義為0x00000001
//FILE_SHARE_WRITE在ntddk.h中定義為0x00000002
//OPEN_EXISTING在winbase.h中定義為3
//FILE_FLAG_NO_BUFFERING在winbase.h中定義為0x20000000
if (Device != INVALID_HANDLE_VALUE)
{
SetLastError(ERROR_BUSY);
PrintLastError(&VolumeName[4]);
return -1;
}
if (CdImage)
{
sprintf(DeviceName, DEVICE_NAME_PREFIX "Cd" "%u", DeviceNumber);
}
else
{
sprintf(DeviceName, DEVICE_NAME_PREFIX "%u", DeviceNumber);
}
if (!DefineDosDevice(
DDD_RAW_TARGET_PATH,
&VolumeName[4],
DeviceName
))
//DDD_RAW_TARGET_PATH在winbase.h中定義為0x00000001
{
PrintLastError(&VolumeName[4]);
return -1;
}
Device = CreateFile(
VolumeName,
GENERIC_READ | GENERIC_WRITE,
FILE_SHARE_READ | FILE_SHARE_WRITE,
NULL,
OPEN_EXISTING,
FILE_FLAG_NO_BUFFERING,
NULL
);
if (Device == INVALID_HANDLE_VALUE)
//INVALID_HANDLE_VALUE在winbase.h中定義為((HANDLE)-1)
{
PrintLastError(&VolumeName[4]);
DefineDosDevice(DDD_REMOVE_DEFINITION, &VolumeName[4], NULL);
//DDD_REMOVE_DEFINITION在winbase.h中定義為0x00000002
return -1;
}
if (!DeviceIoControl(
Device,
IOCTL_FILE_DISK_OPEN_FILE,
OpenFileInformation,
sizeof(OPEN_FILE_INFORMATION) + OpenFileInformation->FileNameLength - 1,
NULL,
0,
&BytesReturned,
NULL
))
{
PrintLastError("FileDisk:");
DefineDosDevice(DDD_REMOVE_DEFINITION, &VolumeName[4], NULL);
return -1;
}
return 0;
}
int FileDiskUmount(char DriveLetter)
{
char VolumeName[] = "\\\\.\\ :";
HANDLE Device;
DWORD BytesReturned;
VolumeName[4] = DriveLetter;
Device = CreateFile(
VolumeName,
GENERIC_READ | GENERIC_WRITE,
FILE_SHARE_READ | FILE_SHARE_WRITE,
NULL,
OPEN_EXISTING,
FILE_FLAG_NO_BUFFERING,
NULL
);
if (Device == INVALID_HANDLE_VALUE)
{
PrintLastError(&VolumeName[4]);
return -1;
}
if (!DeviceIoControl(
Device,
FSCTL_LOCK_VOLUME,
NULL,
0,
NULL,
0,
&BytesReturned,
NULL
))
{
PrintLastError(&VolumeName[4]);
return -1;
}
if (!DeviceIoControl(
Device,
IOCTL_FILE_DISK_CLOSE_FILE,
NULL,
0,
NULL,
0,
&BytesReturned,
NULL
))
{
PrintLastError("FileDisk:");
return -1;
}
if (!DeviceIoControl(
Device,
FSCTL_DISMOUNT_VOLUME,
NULL,
0,
NULL,
0,
&BytesReturned,
NULL
))
{
PrintLastError(&VolumeName[4]);
return -1;
}
if (!DeviceIoControl(
Device,
FSCTL_UNLOCK_VOLUME,
NULL,
0,
NULL,
0,
&BytesReturned,
NULL
))
{
PrintLastError(&VolumeName[4]);
return -1;
}
CloseHandle(Device);
if (!DefineDosDevice(
DDD_REMOVE_DEFINITION,
&VolumeName[4],
NULL
))
{
PrintLastError(&VolumeName[4]);
return -1;
}
return 0;
}
int FileDiskStatus(char DriveLetter)
{
char VolumeName[] = "\\\\.\\ :";
HANDLE Device;
POPEN_FILE_INFORMATION OpenFileInformation;
DWORD BytesReturned;
VolumeName[4] = DriveLetter;
Device = CreateFile(
VolumeName,
GENERIC_READ,
FILE_SHARE_READ | FILE_SHARE_WRITE,
NULL,
OPEN_EXISTING,
FILE_FLAG_NO_BUFFERING,
NULL
);
if (Device == INVALID_HANDLE_VALUE)
{
PrintLastError(&VolumeName[4]);
return -1;
}
OpenFileInformation = malloc(sizeof(OPEN_FILE_INFORMATION) + MAX_PATH);
//MAX_PATH在windows.h中定義為260
if (!DeviceIoControl(
Device,
IOCTL_FILE_DISK_QUERY_FILE,
NULL,
0,
OpenFileInformation,
sizeof(OPEN_FILE_INFORMATION) + MAX_PATH,
&BytesReturned,
NULL
))
{
PrintLastError(&VolumeName[4]);
return -1;
}
if (BytesReturned < sizeof(OPEN_FILE_INFORMATION))
{
SetLastError(ERROR_INSUFFICIENT_BUFFER);
//ERROR_INSUFFICIENT_BUFFER在winerror.h中定義為122L
PrintLastError(&VolumeName[4]);
return -1;
}
printf("%c: %.*s Size: %I64u bytes%s\n",
DriveLetter,
OpenFileInformation->FileNameLength,
OpenFileInformation->FileName,
OpenFileInformation->FileSize,
OpenFileInformation->ReadOnly ? ", ReadOnly" : ""
);
return 0;
}
int __cdecl main(int argc, char* argv[])
{
char* Command;
int DeviceNumber;
char* FileName;
char* Option;
char DriveLetter;
BOOLEAN CdImage = FALSE;
POPEN_FILE_INFORMATION OpenFileInformation;
Command = argv[1];
if ((argc == 5 || argc == 6) && !strcmp(Command, "/mount"))
{
DeviceNumber = atoi(argv[2]);
FileName = argv[3];
if (strlen(FileName) < 2)
{
return FileDiskSyntax();
}
OpenFileInformation =
malloc(sizeof(OPEN_FILE_INFORMATION) + strlen(FileName) + 7);
memset(
OpenFileInformation,
0,
sizeof(OPEN_FILE_INFORMATION) + strlen(FileName) + 7
);
if (FileName[0] == '\\')
{
if (FileName[1] == '\\')
// \\server\share\path\filedisk.img
{
strcpy(OpenFileInformation->FileName, "\\??\\UNC");
strcat(OpenFileInformation->FileName, FileName + 1);
}
else
// \Device\Harddisk0\Partition1\path\filedisk.img
{
strcpy(OpenFileInformation->FileName, FileName);
}
}
else
// c:\path\filedisk.img
{
strcpy(OpenFileInformation->FileName, "\\??\\");
strcat(OpenFileInformation->FileName, FileName);
}
OpenFileInformation->FileNameLength =
(USHORT) strlen(OpenFileInformation->FileName);
if (argc > 5)
{
Option = argv[4];
DriveLetter = argv[5][0];
if (!strcmp(Option, "/ro"))
{
OpenFileInformation->ReadOnly = TRUE;
}
else if (!strcmp(Option, "/cd"))
{
CdImage = TRUE;
}
else
{
if (Option[strlen(Option) - 1] == 'G')
{
OpenFileInformation->FileSize.QuadPart =
_atoi64(Option) * 1024 * 1024 * 1024;
}
else if (Option[strlen(Option) - 1] == 'M')
{
OpenFileInformation->FileSize.QuadPart =
_atoi64(Option) * 1024 * 1024;
}
else if (Option[strlen(Option) - 1] == 'k')
{
OpenFileInformation->FileSize.QuadPart =
_atoi64(Option) * 1024;
}
else
{
OpenFileInformation->FileSize.QuadPart =
_atoi64(Option);
}
}
}
else
{
DriveLetter = argv[4][0];
}
return FileDiskMount(DeviceNumber, OpenFileInformation, DriveLetter, CdImage);
}
else if (argc == 3 && !strcmp(Command, "/umount"))
{
DriveLetter = argv[2][0];
return FileDiskUmount(DriveLetter);
}
else if (argc == 3 && !strcmp(Command, "/status"))
{
DriveLetter = argv[2][0];
return FileDiskStatus(DriveLetter);
}
else
{
return FileDiskSyntax();
}
}
?? 快捷鍵說明
復制代碼
Ctrl + C
搜索代碼
Ctrl + F
全屏模式
F11
切換主題
Ctrl + Shift + D
顯示快捷鍵
?
增大字號
Ctrl + =
減小字號
Ctrl + -