--- /dev/null Thu Jan 01 00:00:00 1970 +0000
+++ b/buildframework/helium/external/filedisk/exe/filedisk.c Wed Oct 28 14:39:48 2009 +0000
@@ -0,0 +1,423 @@
+/*
+ Control program for a virtual disk driver for Windows NT/2000/XP.
+ Copyright (C) 1999, 2000, 2001, 2002 Bo Brantén.
+ This program is free software; you can redistribute it and/or modify
+ it under the terms of the GNU General Public License as published by
+ the Free Software Foundation; either version 2 of the License, or
+ (at your option) any later version.
+ This program is distributed in the hope that it will be useful,
+ but WITHOUT ANY WARRANTY; without even the implied warranty of
+ MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
+ GNU General Public License for more details.
+ You should have received a copy of the GNU General Public License
+ along with this program; if not, write to the Free Software
+ Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
+*/
+
+#include <windows.h>
+#include <winioctl.h>
+#include <stdio.h>
+#include <stdlib.h>
+#include "filedisk.h"
+
+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
+ );
+
+ 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
+ );
+
+ 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
+ ))
+ {
+ 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)
+ {
+ PrintLastError(&VolumeName[4]);
+ DefineDosDevice(DDD_REMOVE_DEFINITION, &VolumeName[4], NULL);
+ 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);
+
+ 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);
+ 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();
+ }
+}