#ifndef TC_TO_STRING

# define TC_TO_STRING2(n) #n

# define TC_TO_STRING(n) TC_TO_STRING2(n)

#endif

#ifdef DEVICE_DRIVER

# define Dump DbgPrint

# else

# define Dump(...) DbgPrintEx (DPFLTR_IHVDRIVER_ID, DPFLTR_ERROR_LEVEL, __VA_ARGS__)

# endif

# define DumpMem(...) DumpMemory (__VA_ARGS__)

# else

# define Dump(...)

# define DumpMem(...)

# endif

#endif

#if !defined (trace_msg) && !defined (TC_WINDOWS_BOOT)

# ifdef DEVICE_DRIVER

# define trace_msg Dump

# elif defined (_WIN32)

# define trace_msg(...) do { char msg[2048]; StringCbPrintfA (msg, sizeof (msg), __VA_ARGS__); OutputDebugString (msg); } while (0)

# endif

# define trace_point trace_msg (__FUNCTION__ ":" TC_TO_STRING(__LINE__) "\n")

Dump ("DriverEntry " TC_APP_NAME " " VERSION_STRING "\n");

DetectX86Features ();

PsGetVersion (&OsMajorVersion, &OsMinorVersion, NULL, NULL);

// Load dump filter if the main driver is already loaded

if (NT_SUCCESS (TCDeviceIoControl (NT_ROOT_PREFIX, TC_IOCTL_GET_DRIVER_VERSION, NULL, 0, &version, sizeof (version))))

return DumpFilterEntry ((PFILTER_EXTENSION) DriverObject, (PFILTER_INITIALIZATION_DATA) RegistryPath);

TCDriverObject = DriverObject;

memset (VirtualVolumeDeviceObjects, 0, sizeof (VirtualVolumeDeviceObjects));

return TCCreateRootDeviceObject (DriverObject);

}

NTSTATUS DriverAddDevice (PDRIVER_OBJECT driverObject, PDEVICE_OBJECT pdo)

{

char nameInfoBuffer[128];

POBJECT_NAME_INFORMATION nameInfo = (POBJECT_NAME_INFORMATION) nameInfoBuffer;

ULONG nameInfoSize;

Dump ("AddDevice pdo=%p type=%x name=%ws\n", pdo, pdo->DeviceType, NT_SUCCESS (ObQueryNameString (pdo, nameInfo, sizeof (nameInfoBuffer), &nameInfoSize)) ? nameInfo->Name.Buffer : L"?");

#endif

{

UNICODE_STRING ntUnicodeString;

WCHAR ntname[32];

PEXTENSION Extension;

NTSTATUS ntStatus;

ULONG devChars = 0;

WCHAR dosname[32];

#endif

Dump ("TCCreateDeviceObject BEGIN\n");

ASSERT (KeGetCurrentIrql() == PASSIVE_LEVEL);

TCGetNTNameFromNumber (ntname, sizeof(ntname),mount->nDosDriveNo);

RtlInitUnicodeString (&ntUnicodeString, ntname);

TCGetDosNameFromNumber (dosname, sizeof(dosname),mount->nDosDriveNo, DeviceNamespaceDefault);

#endif

devChars = FILE_DEVICE_SECURE_OPEN;

devChars |= mount->bMountReadOnly ? FILE_READ_ONLY_DEVICE : 0;

devChars |= mount->bMountRemovable ? FILE_REMOVABLE_MEDIA : 0;

Dump ("Creating device nt=%ls dos=%ls\n", ntname, dosname);

ntStatus = IoCreateDevice (

DriverObject, /* Our Driver Object */

sizeof (EXTENSION), /* Size of state information */

&ntUnicodeString, /* Device name "\Device\Name" */

FILE_DEVICE_DISK, /* Device type */

void RootDeviceControlMutexRelease ()

{

KeReleaseMutex (&RootDeviceControlMutex, FALSE);

}

NTSTATUS ProcessVolumeDeviceControlIrp (PDEVICE_OBJECT DeviceObject, PEXTENSION Extension, PIRP Irp)

{

PIO_STACK_LOCATION irpSp = IoGetCurrentIrpStackLocation (Irp);

switch (irpSp->Parameters.DeviceIoControl.IoControlCode)

{

case IOCTL_MOUNTDEV_QUERY_DEVICE_NAME:

if (!ValidateIOBufferSize (Irp, sizeof (MOUNTDEV_NAME), ValidateOutput))

{

Irp->IoStatus.Information = sizeof (MOUNTDEV_NAME);

Irp->IoStatus.Status = STATUS_BUFFER_OVERFLOW;

}

else

Dump ("name = %ls\n",ntName);

}

break;

case IOCTL_MOUNTDEV_QUERY_UNIQUE_ID:

if (!ValidateIOBufferSize (Irp, sizeof (MOUNTDEV_UNIQUE_ID), ValidateOutput))

{

Irp->IoStatus.Information = sizeof (MOUNTDEV_UNIQUE_ID);

Irp->IoStatus.Status = STATUS_BUFFER_OVERFLOW;

}

else

Dump ("id = %s\n",volId);

}

break;

case IOCTL_MOUNTDEV_QUERY_SUGGESTED_LINK_NAME:

{

ULONG outLength;

UNICODE_STRING ntUnicodeString;

WCHAR ntName[256];

PMOUNTDEV_SUGGESTED_LINK_NAME outputBuffer = (PMOUNTDEV_SUGGESTED_LINK_NAME) Irp->AssociatedIrp.SystemBuffer;

Dump ("link = %ls\n",ntName);

}

break;

case IOCTL_DISK_GET_MEDIA_TYPES:

case IOCTL_DISK_GET_DRIVE_GEOMETRY:

/* Return the drive geometry for the disk. Note that we

return values which were made up to suit the disk size. */

if (ValidateIOBufferSize (Irp, sizeof (DISK_GEOMETRY), ValidateOutput))

{

PDISK_GEOMETRY outputBuffer = (PDISK_GEOMETRY)

Irp->AssociatedIrp.SystemBuffer;

Irp->IoStatus.Status = STATUS_SUCCESS;

Irp->IoStatus.Information = sizeof (DISK_GEOMETRY);

}

break;

case IOCTL_STORAGE_QUERY_PROPERTY:

if (EnableExtendedIoctlSupport)

{

if (ValidateIOBufferSize (Irp, sizeof (STORAGE_PROPERTY_QUERY), ValidateInput))

{

PSTORAGE_PROPERTY_QUERY pStoragePropQuery = (PSTORAGE_PROPERTY_QUERY) Irp->AssociatedIrp.SystemBuffer;

STORAGE_QUERY_TYPE type = pStoragePropQuery->QueryType;

else

return TCCompleteIrp (Irp, STATUS_INVALID_DEVICE_REQUEST, 0);

break;

case IOCTL_DISK_GET_PARTITION_INFO:

if (ValidateIOBufferSize (Irp, sizeof (PARTITION_INFORMATION), ValidateOutput))

{

PPARTITION_INFORMATION outputBuffer = (PPARTITION_INFORMATION)

Irp->AssociatedIrp.SystemBuffer;

outputBuffer->PartitionType = Extension->PartitionType;

Irp->IoStatus.Status = STATUS_SUCCESS;

Irp->IoStatus.Information = sizeof (PARTITION_INFORMATION);

}

break;

case IOCTL_DISK_GET_PARTITION_INFO_EX:

if (ValidateIOBufferSize (Irp, sizeof (PARTITION_INFORMATION_EX), ValidateOutput))

{

PPARTITION_INFORMATION_EX outputBuffer = (PPARTITION_INFORMATION_EX) Irp->AssociatedIrp.SystemBuffer;

outputBuffer->PartitionStyle = PARTITION_STYLE_MBR;

outputBuffer->RewritePartition = FALSE;

Irp->IoStatus.Status = STATUS_SUCCESS;

Irp->IoStatus.Information = sizeof (PARTITION_INFORMATION_EX);

}

break;

case IOCTL_DISK_GET_DRIVE_LAYOUT:

if (ValidateIOBufferSize (Irp, sizeof (DRIVE_LAYOUT_INFORMATION), ValidateOutput))

{

PDRIVE_LAYOUT_INFORMATION outputBuffer = (PDRIVE_LAYOUT_INFORMATION)

Irp->AssociatedIrp.SystemBuffer;

outputBuffer->PartitionCount = 1;

Irp->IoStatus.Status = STATUS_SUCCESS;

Irp->IoStatus.Information = sizeof (PARTITION_INFORMATION);

}

break;

case IOCTL_DISK_GET_LENGTH_INFO:

if (!ValidateIOBufferSize (Irp, sizeof (GET_LENGTH_INFORMATION), ValidateOutput))

{

Irp->IoStatus.Status = STATUS_BUFFER_OVERFLOW;

Irp->IoStatus.Information = sizeof (GET_LENGTH_INFORMATION);

}

else

Irp->IoStatus.Status = STATUS_SUCCESS;

Irp->IoStatus.Information = sizeof (GET_LENGTH_INFORMATION);

}

break;

case IOCTL_DISK_VERIFY:

if (ValidateIOBufferSize (Irp, sizeof (VERIFY_INFORMATION), ValidateInput))

{

HRESULT hResult;

ULONGLONG ullStartingOffset, ullNewOffset, ullEndOffset;

PVERIFY_INFORMATION pVerifyInformation;

pVerifyInformation = (PVERIFY_INFORMATION) Irp->AssociatedIrp.SystemBuffer;

Irp->IoStatus.Information = 0;

}

break;

case IOCTL_DISK_CHECK_VERIFY:

case IOCTL_STORAGE_CHECK_VERIFY:

{

Irp->IoStatus.Status = STATUS_SUCCESS;

Irp->IoStatus.Information = 0;

if (irpSp->Parameters.DeviceIoControl.OutputBufferLength >= sizeof (ULONG))

{

Irp->IoStatus.Information = sizeof (ULONG);

}

}

break;

case IOCTL_DISK_IS_WRITABLE:

{

if (Extension->bReadOnly)

Irp->IoStatus.Status = STATUS_MEDIA_WRITE_PROTECTED;

else

Irp->IoStatus.Status = STATUS_SUCCESS;

Irp->IoStatus.Information = 0;

}

break;

case IOCTL_VOLUME_ONLINE:

Irp->IoStatus.Status = STATUS_SUCCESS;

Irp->IoStatus.Information = 0;

break;

case IOCTL_VOLUME_GET_VOLUME_DISK_EXTENTS:

// Vista's filesystem defragmenter fails if IOCTL_VOLUME_GET_VOLUME_DISK_EXTENTS does not succeed.

if (!(OsMajorVersion == 6 && OsMinorVersion == 0))

{

Irp->IoStatus.Status = STATUS_INVALID_DEVICE_REQUEST;

Irp->IoStatus.Information = 0;

}

Irp->IoStatus.Status = STATUS_SUCCESS;

Irp->IoStatus.Information = sizeof (*extents);

}

break;

default:

return TCCompleteIrp (Irp, STATUS_INVALID_DEVICE_REQUEST, 0);

}

if (!NT_SUCCESS (Irp->IoStatus.Status))

{

Dump ("IOCTL error 0x%08x (0x%x %d)\n",

Irp->IoStatus.Status,

(int) (irpSp->Parameters.DeviceIoControl.IoControlCode >> 16),

(int) ((irpSp->Parameters.DeviceIoControl.IoControlCode & 0x1FFF) >> 2));

default:

return TCCompleteIrp (Irp, STATUS_INVALID_DEVICE_REQUEST, 0);

}

if (!NT_SUCCESS (Irp->IoStatus.Status))

{

switch (irpSp->Parameters.DeviceIoControl.IoControlCode)

{

case TC_IOCTL_GET_MOUNTED_VOLUMES:

case TC_IOCTL_GET_PASSWORD_CACHE_STATUS: