3 * Copyright (C) 2002 ReactOS Team
5 * This program is free software; you can redistribute it and/or modify
6 * it under the terms of the GNU General Public License as published by
7 * the Free Software Foundation; either version 2 of the License, or
8 * (at your option) any later version.
10 * This program is distributed in the hope that it will be useful,
11 * but WITHOUT ANY WARRANTY; without even the implied warranty of
12 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
13 * GNU General Public License for more details.
15 * You should have received a copy of the GNU General Public License
16 * along with this program; if not, write to the Free Software
17 * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA.
19 * COPYRIGHT: See COPYING in the top level directory
20 * PROJECT: ReactOS kernel
21 * FILE: drivers/filesystems/ntfs/fsctl.c
22 * PURPOSE: NTFS filesystem driver
23 * PROGRAMMER: Eric Kohl
28 /* INCLUDES *****************************************************************/
37 /* FUNCTIONS ****************************************************************/
40 * FUNCTION: Tests if the device contains a filesystem that can be mounted
45 NtfsHasFileSystem(PDEVICE_OBJECT DeviceToMount
)
47 PARTITION_INFORMATION PartitionInfo
;
48 DISK_GEOMETRY DiskGeometry
;
49 ULONG ClusterSize
, Size
, k
;
50 PBOOT_SECTOR BootSector
;
53 DPRINT1("NtfsHasFileSystem() called\n");
55 Size
= sizeof(DISK_GEOMETRY
);
56 Status
= NtfsDeviceIoControl(DeviceToMount
,
57 IOCTL_DISK_GET_DRIVE_GEOMETRY
,
63 if (!NT_SUCCESS(Status
))
65 DPRINT1("NtfsDeviceIoControl() failed (Status %lx)\n", Status
);
69 if (DiskGeometry
.MediaType
== FixedMedia
)
71 /* We have found a hard disk */
72 Size
= sizeof(PARTITION_INFORMATION
);
73 Status
= NtfsDeviceIoControl(DeviceToMount
,
74 IOCTL_DISK_GET_PARTITION_INFO
,
80 if (!NT_SUCCESS(Status
))
82 DPRINT1("NtfsDeviceIoControl() failed (Status %lx)\n", Status
);
86 if (PartitionInfo
.PartitionType
!= PARTITION_IFS
)
88 DPRINT1("Invalid partition type\n");
89 return STATUS_UNRECOGNIZED_VOLUME
;
93 DPRINT1("BytesPerSector: %lu\n", DiskGeometry
.BytesPerSector
);
94 BootSector
= ExAllocatePoolWithTag(NonPagedPool
,
95 DiskGeometry
.BytesPerSector
,
97 if (BootSector
== NULL
)
99 return STATUS_INSUFFICIENT_RESOURCES
;
102 Status
= NtfsReadSectors(DeviceToMount
,
105 DiskGeometry
.BytesPerSector
,
108 if (!NT_SUCCESS(Status
))
114 * Check values of different fields. If those fields have not expected
115 * values, we fail, to avoid mounting partitions that Windows won't mount.
118 /* OEMID: this field must be NTFS */
119 if (RtlCompareMemory(BootSector
->OEMID
, "NTFS ", 8) != 8)
121 DPRINT1("Failed with NTFS-identifier: [%.8s]\n", BootSector
->OEMID
);
122 Status
= STATUS_UNRECOGNIZED_VOLUME
;
126 /* Unused0: this field must be COMPLETELY null */
127 for (k
= 0; k
< 7; k
++)
129 if (BootSector
->BPB
.Unused0
[k
] != 0)
131 DPRINT1("Failed in field Unused0: [%.7s]\n", BootSector
->BPB
.Unused0
);
132 Status
= STATUS_UNRECOGNIZED_VOLUME
;
137 /* Unused3: this field must be COMPLETELY null */
138 for (k
= 0; k
< 4; k
++)
140 if (BootSector
->BPB
.Unused3
[k
] != 0)
142 DPRINT1("Failed in field Unused3: [%.4s]\n", BootSector
->BPB
.Unused3
);
143 Status
= STATUS_UNRECOGNIZED_VOLUME
;
148 /* Check cluster size */
149 ClusterSize
= BootSector
->BPB
.BytesPerSector
* BootSector
->BPB
.SectorsPerCluster
;
150 if (ClusterSize
!= 512 && ClusterSize
!= 1024 &&
151 ClusterSize
!= 2048 && ClusterSize
!= 4096 &&
152 ClusterSize
!= 8192 && ClusterSize
!= 16384 &&
153 ClusterSize
!= 32768 && ClusterSize
!= 65536)
155 DPRINT1("Cluster size failed: %hu, %hu, %hu\n",
156 BootSector
->BPB
.BytesPerSector
,
157 BootSector
->BPB
.SectorsPerCluster
,
159 Status
= STATUS_UNRECOGNIZED_VOLUME
;
164 ExFreePool(BootSector
);
172 NtfsQueryMftZoneReservation(VOID
)
174 ULONG ZoneReservation
= 1;
175 RTL_QUERY_REGISTRY_TABLE QueryTable
[2];
177 RtlZeroMemory(QueryTable
, sizeof(QueryTable
));
178 QueryTable
[0].Flags
= RTL_QUERY_REGISTRY_DIRECT
;
179 QueryTable
[0].Name
= L
"NtfsMftZoneReservation";
180 QueryTable
[0].EntryContext
= &ZoneReservation
;
182 RtlQueryRegistryValues(RTL_REGISTRY_CONTROL
,
188 return ZoneReservation
;
194 NtfsGetVolumeData(PDEVICE_OBJECT DeviceObject
,
195 PDEVICE_EXTENSION DeviceExt
)
197 DISK_GEOMETRY DiskGeometry
;
198 PFILE_RECORD_HEADER VolumeRecord
;
199 PVOLINFO_ATTRIBUTE VolumeInfo
;
200 PBOOT_SECTOR BootSector
;
202 PNTFS_INFO NtfsInfo
= &DeviceExt
->NtfsInfo
;
204 PNTFS_ATTR_CONTEXT AttrCtxt
;
205 PNTFS_ATTR_RECORD Attribute
;
209 DPRINT("NtfsGetVolumeData() called\n");
211 Size
= sizeof(DISK_GEOMETRY
);
212 Status
= NtfsDeviceIoControl(DeviceObject
,
213 IOCTL_DISK_GET_DRIVE_GEOMETRY
,
219 if (!NT_SUCCESS(Status
))
221 DPRINT("NtfsDeviceIoControl() failed (Status %lx)\n", Status
);
225 DPRINT("BytesPerSector: %lu\n", DiskGeometry
.BytesPerSector
);
226 BootSector
= ExAllocatePoolWithTag(NonPagedPool
,
227 DiskGeometry
.BytesPerSector
,
229 if (BootSector
== NULL
)
231 return STATUS_INSUFFICIENT_RESOURCES
;
234 Status
= NtfsReadSectors(DeviceObject
,
235 0, /* Partition boot sector */
237 DiskGeometry
.BytesPerSector
,
240 if (!NT_SUCCESS(Status
))
242 ExFreePool(BootSector
);
246 /* Read data from the bootsector */
247 NtfsInfo
->BytesPerSector
= BootSector
->BPB
.BytesPerSector
;
248 NtfsInfo
->SectorsPerCluster
= BootSector
->BPB
.SectorsPerCluster
;
249 NtfsInfo
->BytesPerCluster
= BootSector
->BPB
.BytesPerSector
* BootSector
->BPB
.SectorsPerCluster
;
250 NtfsInfo
->SectorCount
= BootSector
->EBPB
.SectorCount
;
251 NtfsInfo
->ClusterCount
= DeviceExt
->NtfsInfo
.SectorCount
/ (ULONGLONG
)DeviceExt
->NtfsInfo
.SectorsPerCluster
;
253 NtfsInfo
->MftStart
.QuadPart
= BootSector
->EBPB
.MftLocation
;
254 NtfsInfo
->MftMirrStart
.QuadPart
= BootSector
->EBPB
.MftMirrLocation
;
255 NtfsInfo
->SerialNumber
= BootSector
->EBPB
.SerialNumber
;
256 if (BootSector
->EBPB
.ClustersPerMftRecord
> 0)
257 NtfsInfo
->BytesPerFileRecord
= BootSector
->EBPB
.ClustersPerMftRecord
* NtfsInfo
->BytesPerCluster
;
259 NtfsInfo
->BytesPerFileRecord
= 1 << (-BootSector
->EBPB
.ClustersPerMftRecord
);
260 if (BootSector
->EBPB
.ClustersPerIndexRecord
> 0)
261 NtfsInfo
->BytesPerIndexRecord
= BootSector
->EBPB
.ClustersPerIndexRecord
* NtfsInfo
->BytesPerCluster
;
263 NtfsInfo
->BytesPerIndexRecord
= 1 << (-BootSector
->EBPB
.ClustersPerIndexRecord
);
265 DPRINT("Boot sector information:\n");
266 DPRINT(" BytesPerSector: %hu\n", BootSector
->BPB
.BytesPerSector
);
267 DPRINT(" SectorsPerCluster: %hu\n", BootSector
->BPB
.SectorsPerCluster
);
268 DPRINT(" SectorCount: %I64u\n", BootSector
->EBPB
.SectorCount
);
269 DPRINT(" MftStart: %I64u\n", BootSector
->EBPB
.MftLocation
);
270 DPRINT(" MftMirrStart: %I64u\n", BootSector
->EBPB
.MftMirrLocation
);
271 DPRINT(" ClustersPerMftRecord: %lx\n", BootSector
->EBPB
.ClustersPerMftRecord
);
272 DPRINT(" ClustersPerIndexRecord: %lx\n", BootSector
->EBPB
.ClustersPerIndexRecord
);
273 DPRINT(" SerialNumber: %I64x\n", BootSector
->EBPB
.SerialNumber
);
275 ExFreePool(BootSector
);
277 DeviceExt
->MasterFileTable
= ExAllocatePoolWithTag(NonPagedPool
,
278 NtfsInfo
->BytesPerFileRecord
,
280 if (DeviceExt
->MasterFileTable
== NULL
)
282 return STATUS_INSUFFICIENT_RESOURCES
;
285 Status
= NtfsReadSectors(DeviceObject
,
286 NtfsInfo
->MftStart
.u
.LowPart
* NtfsInfo
->SectorsPerCluster
,
287 NtfsInfo
->BytesPerFileRecord
/ NtfsInfo
->BytesPerSector
,
288 NtfsInfo
->BytesPerSector
,
289 (PVOID
)DeviceExt
->MasterFileTable
,
291 if (!NT_SUCCESS(Status
))
293 DPRINT1("Failed reading MFT.\n");
294 ExFreePool(DeviceExt
->MasterFileTable
);
298 Status
= FindAttribute(DeviceExt
, DeviceExt
->MasterFileTable
, AttributeData
, L
"", 0, &DeviceExt
->MFTContext
);
299 if (!NT_SUCCESS(Status
))
301 DPRINT1("Can't find data attribute for Master File Table.\n");
302 ExFreePool(DeviceExt
->MasterFileTable
);
306 VolumeRecord
= ExAllocatePoolWithTag(NonPagedPool
,
307 NtfsInfo
->BytesPerFileRecord
,
309 if (VolumeRecord
== NULL
)
311 DPRINT1("Allocation failed for volume record\n");
312 ExFreePool(DeviceExt
->MasterFileTable
);
313 return STATUS_INSUFFICIENT_RESOURCES
;
316 /* Read Volume File (MFT index 3) */
317 DeviceExt
->StorageDevice
= DeviceObject
;
318 Status
= ReadFileRecord(DeviceExt
,
321 if (!NT_SUCCESS(Status
))
323 DPRINT1("Failed reading volume file\n");
324 ExFreePool(VolumeRecord
);
325 ExFreePool(DeviceExt
->MasterFileTable
);
329 /* Enumerate attributes */
330 NtfsDumpFileAttributes(DeviceExt
->MasterFileTable
);
332 /* Enumerate attributes */
333 NtfsDumpFileAttributes(VolumeRecord
);
335 /* Get volume name */
336 Status
= FindAttribute(DeviceExt
, VolumeRecord
, AttributeVolumeName
, L
"", 0, &AttrCtxt
);
338 if (NT_SUCCESS(Status
) && AttrCtxt
->Record
.Resident
.ValueLength
!= 0)
340 Attribute
= &AttrCtxt
->Record
;
341 DPRINT("Data length %lu\n", AttributeDataLength(Attribute
));
342 NtfsInfo
->VolumeLabelLength
=
343 min (Attribute
->Resident
.ValueLength
, MAXIMUM_VOLUME_LABEL_LENGTH
);
344 RtlCopyMemory(NtfsInfo
->VolumeLabel
,
345 (PVOID
)((ULONG_PTR
)Attribute
+ Attribute
->Resident
.ValueOffset
),
346 NtfsInfo
->VolumeLabelLength
);
347 VolumeNameU
= NtfsInfo
->VolumeLabel
;
351 NtfsInfo
->VolumeLabelLength
= 0;
355 VolumeFcb
= NtfsCreateFCB(VolumeNameU
, DeviceExt
);
356 if (VolumeFcb
== NULL
)
358 DPRINT1("Failed allocating volume FCB\n");
359 ExFreePool(VolumeRecord
);
360 ExFreePool(DeviceExt
->MasterFileTable
);
361 return STATUS_INSUFFICIENT_RESOURCES
;
364 VolumeFcb
->Flags
= FCB_IS_VOLUME
;
365 VolumeFcb
->RFCB
.FileSize
.QuadPart
= DeviceExt
->NtfsInfo
.SectorCount
* DeviceExt
->NtfsInfo
.BytesPerSector
;
366 VolumeFcb
->RFCB
.ValidDataLength
= VolumeFcb
->RFCB
.FileSize
;
367 VolumeFcb
->RFCB
.AllocationSize
= VolumeFcb
->RFCB
.FileSize
;
368 VolumeFcb
->MFTIndex
= 0;
369 DeviceExt
->VolumeFcb
= VolumeFcb
;
371 /* Get volume information */
372 Status
= FindAttribute(DeviceExt
, VolumeRecord
, AttributeVolumeInformation
, L
"", 0, &AttrCtxt
);
374 if (NT_SUCCESS(Status
) && AttrCtxt
->Record
.Resident
.ValueLength
!= 0)
376 Attribute
= &AttrCtxt
->Record
;
377 DPRINT("Data length %lu\n", AttributeDataLength (Attribute
));
378 VolumeInfo
= (PVOID
)((ULONG_PTR
)Attribute
+ Attribute
->Resident
.ValueOffset
);
380 NtfsInfo
->MajorVersion
= VolumeInfo
->MajorVersion
;
381 NtfsInfo
->MinorVersion
= VolumeInfo
->MinorVersion
;
382 NtfsInfo
->Flags
= VolumeInfo
->Flags
;
385 ExFreePool(VolumeRecord
);
387 NtfsInfo
->MftZoneReservation
= NtfsQueryMftZoneReservation();
395 NtfsMountVolume(PDEVICE_OBJECT DeviceObject
,
398 PDEVICE_OBJECT NewDeviceObject
= NULL
;
399 PDEVICE_OBJECT DeviceToMount
;
400 PIO_STACK_LOCATION Stack
;
401 PNTFS_FCB Fcb
= NULL
;
402 PNTFS_CCB Ccb
= NULL
;
403 PNTFS_VCB Vcb
= NULL
;
406 DPRINT1("NtfsMountVolume() called\n");
408 if (DeviceObject
!= NtfsGlobalData
->DeviceObject
)
410 Status
= STATUS_INVALID_DEVICE_REQUEST
;
414 Stack
= IoGetCurrentIrpStackLocation(Irp
);
415 DeviceToMount
= Stack
->Parameters
.MountVolume
.DeviceObject
;
417 Status
= NtfsHasFileSystem(DeviceToMount
);
418 if (!NT_SUCCESS(Status
))
423 Status
= IoCreateDevice(NtfsGlobalData
->DriverObject
,
424 sizeof(DEVICE_EXTENSION
),
426 FILE_DEVICE_DISK_FILE_SYSTEM
,
430 if (!NT_SUCCESS(Status
))
433 NewDeviceObject
->Flags
|= DO_DIRECT_IO
;
434 Vcb
= (PVOID
)NewDeviceObject
->DeviceExtension
;
435 RtlZeroMemory(Vcb
, sizeof(NTFS_VCB
));
437 Vcb
->Identifier
.Type
= NTFS_TYPE_VCB
;
438 Vcb
->Identifier
.Size
= sizeof(NTFS_TYPE_VCB
);
440 Status
= NtfsGetVolumeData(DeviceToMount
,
442 if (!NT_SUCCESS(Status
))
445 NewDeviceObject
->Vpb
= DeviceToMount
->Vpb
;
447 Vcb
->StorageDevice
= DeviceToMount
;
448 Vcb
->StorageDevice
->Vpb
->DeviceObject
= NewDeviceObject
;
449 Vcb
->StorageDevice
->Vpb
->RealDevice
= Vcb
->StorageDevice
;
450 Vcb
->StorageDevice
->Vpb
->Flags
|= VPB_MOUNTED
;
451 NewDeviceObject
->StackSize
= Vcb
->StorageDevice
->StackSize
+ 1;
452 NewDeviceObject
->Flags
&= ~DO_DEVICE_INITIALIZING
;
454 Vcb
->StreamFileObject
= IoCreateStreamFileObject(NULL
,
457 InitializeListHead(&Vcb
->FcbListHead
);
459 Fcb
= NtfsCreateFCB(NULL
, Vcb
);
462 Status
= STATUS_INSUFFICIENT_RESOURCES
;
466 Ccb
= ExAllocatePoolWithTag(NonPagedPool
,
471 Status
= STATUS_INSUFFICIENT_RESOURCES
;
475 RtlZeroMemory(Ccb
, sizeof(NTFS_CCB
));
477 Ccb
->Identifier
.Type
= NTFS_TYPE_CCB
;
478 Ccb
->Identifier
.Size
= sizeof(NTFS_TYPE_CCB
);
480 Vcb
->StreamFileObject
->FsContext
= Fcb
;
481 Vcb
->StreamFileObject
->FsContext2
= Ccb
;
482 Vcb
->StreamFileObject
->SectionObjectPointer
= &Fcb
->SectionObjectPointers
;
483 Vcb
->StreamFileObject
->PrivateCacheMap
= NULL
;
484 Vcb
->StreamFileObject
->Vpb
= Vcb
->Vpb
;
485 Ccb
->PtrFileObject
= Vcb
->StreamFileObject
;
486 Fcb
->FileObject
= Vcb
->StreamFileObject
;
487 Fcb
->Vcb
= (PDEVICE_EXTENSION
)Vcb
->StorageDevice
;
489 Fcb
->Flags
= FCB_IS_VOLUME_STREAM
;
491 Fcb
->RFCB
.FileSize
.QuadPart
= Vcb
->NtfsInfo
.SectorCount
* Vcb
->NtfsInfo
.BytesPerSector
;
492 Fcb
->RFCB
.ValidDataLength
.QuadPart
= Vcb
->NtfsInfo
.SectorCount
* Vcb
->NtfsInfo
.BytesPerSector
;
493 Fcb
->RFCB
.AllocationSize
.QuadPart
= Vcb
->NtfsInfo
.SectorCount
* Vcb
->NtfsInfo
.BytesPerSector
; /* Correct? */
495 // Fcb->Entry.ExtentLocationL = 0;
496 // Fcb->Entry.DataLengthL = DeviceExt->CdInfo.VolumeSpaceSize * BLOCKSIZE;
498 CcInitializeCacheMap(Vcb
->StreamFileObject
,
499 (PCC_FILE_SIZES
)(&Fcb
->RFCB
.AllocationSize
),
501 &(NtfsGlobalData
->CacheMgrCallbacks
),
504 ExInitializeResourceLite(&Vcb
->DirResource
);
506 KeInitializeSpinLock(&Vcb
->FcbListLock
);
508 /* Get serial number */
509 NewDeviceObject
->Vpb
->SerialNumber
= Vcb
->NtfsInfo
.SerialNumber
;
511 /* Get volume label */
512 NewDeviceObject
->Vpb
->VolumeLabelLength
= Vcb
->NtfsInfo
.VolumeLabelLength
;
513 RtlCopyMemory(NewDeviceObject
->Vpb
->VolumeLabel
,
514 Vcb
->NtfsInfo
.VolumeLabel
,
515 Vcb
->NtfsInfo
.VolumeLabelLength
);
517 Status
= STATUS_SUCCESS
;
520 if (!NT_SUCCESS(Status
))
523 if (Vcb
&& Vcb
->StreamFileObject
)
524 ObDereferenceObject(Vcb
->StreamFileObject
);
533 IoDeleteDevice(NewDeviceObject
);
536 DPRINT("NtfsMountVolume() done (Status: %lx)\n", Status
);
544 NtfsVerifyVolume(PDEVICE_OBJECT DeviceObject
,
547 UNREFERENCED_PARAMETER(DeviceObject
);
548 UNREFERENCED_PARAMETER(Irp
);
549 DPRINT1("NtfsVerifyVolume() called\n");
550 return STATUS_WRONG_VOLUME
;
556 GetNfsVolumeData(PDEVICE_EXTENSION DeviceExt
,
559 PIO_STACK_LOCATION Stack
;
560 PNTFS_VOLUME_DATA_BUFFER DataBuffer
;
561 PNTFS_ATTR_RECORD Attribute
;
563 DataBuffer
= (PNTFS_VOLUME_DATA_BUFFER
)Irp
->UserBuffer
;
564 Stack
= IoGetCurrentIrpStackLocation(Irp
);
566 if (Stack
->Parameters
.FileSystemControl
.OutputBufferLength
< sizeof(NTFS_VOLUME_DATA_BUFFER
) ||
567 Irp
->UserBuffer
== NULL
)
569 DPRINT1("Invalid output! %d %p\n", Stack
->Parameters
.FileSystemControl
.OutputBufferLength
, Irp
->UserBuffer
);
570 return STATUS_INVALID_PARAMETER
;
573 DataBuffer
->VolumeSerialNumber
.QuadPart
= DeviceExt
->NtfsInfo
.SerialNumber
;
574 DataBuffer
->NumberSectors
.QuadPart
= DeviceExt
->NtfsInfo
.SectorCount
;
575 DataBuffer
->TotalClusters
.QuadPart
= DeviceExt
->NtfsInfo
.ClusterCount
;
576 DataBuffer
->FreeClusters
.QuadPart
= NtfsGetFreeClusters(DeviceExt
);
577 DataBuffer
->TotalReserved
.QuadPart
= 0LL; // FIXME
578 DataBuffer
->BytesPerSector
= DeviceExt
->NtfsInfo
.BytesPerSector
;
579 DataBuffer
->BytesPerCluster
= DeviceExt
->NtfsInfo
.BytesPerCluster
;
580 DataBuffer
->BytesPerFileRecordSegment
= DeviceExt
->NtfsInfo
.BytesPerFileRecord
;
581 DataBuffer
->ClustersPerFileRecordSegment
= DeviceExt
->NtfsInfo
.BytesPerFileRecord
/ DeviceExt
->NtfsInfo
.BytesPerCluster
;
582 DataBuffer
->MftStartLcn
.QuadPart
= DeviceExt
->NtfsInfo
.MftStart
.QuadPart
;
583 DataBuffer
->Mft2StartLcn
.QuadPart
= DeviceExt
->NtfsInfo
.MftMirrStart
.QuadPart
;
584 DataBuffer
->MftZoneStart
.QuadPart
= 0; // FIXME
585 DataBuffer
->MftZoneEnd
.QuadPart
= 0; // FIXME
587 Attribute
= (PNTFS_ATTR_RECORD
)((ULONG_PTR
)DeviceExt
->MasterFileTable
+ DeviceExt
->MasterFileTable
->AttributeOffset
);
588 while (Attribute
< (PNTFS_ATTR_RECORD
)((ULONG_PTR
)DeviceExt
->MasterFileTable
+ DeviceExt
->MasterFileTable
->BytesInUse
) &&
589 Attribute
->Type
!= AttributeEnd
)
591 if (Attribute
->Type
== AttributeData
)
593 ASSERT(Attribute
->IsNonResident
);
594 DataBuffer
->MftValidDataLength
.QuadPart
= Attribute
->NonResident
.DataSize
;
599 Attribute
= (PNTFS_ATTR_RECORD
)((ULONG_PTR
)Attribute
+ Attribute
->Length
);
602 if (Stack
->Parameters
.FileSystemControl
.OutputBufferLength
>= sizeof(NTFS_EXTENDED_VOLUME_DATA
) + sizeof(NTFS_VOLUME_DATA_BUFFER
))
604 PNTFS_EXTENDED_VOLUME_DATA ExtendedData
= (PNTFS_EXTENDED_VOLUME_DATA
)((ULONG_PTR
)Irp
->UserBuffer
+ sizeof(NTFS_VOLUME_DATA_BUFFER
));
606 ExtendedData
->ByteCount
= sizeof(NTFS_EXTENDED_VOLUME_DATA
);
607 ExtendedData
->MajorVersion
= DeviceExt
->NtfsInfo
.MajorVersion
;
608 ExtendedData
->MinorVersion
= DeviceExt
->NtfsInfo
.MinorVersion
;
611 return STATUS_SUCCESS
;
617 GetNtfsFileRecord(PDEVICE_EXTENSION DeviceExt
,
621 PIO_STACK_LOCATION Stack
;
622 PNTFS_FILE_RECORD_INPUT_BUFFER InputBuffer
;
623 PFILE_RECORD_HEADER FileRecord
;
624 PNTFS_FILE_RECORD_OUTPUT_BUFFER OutputBuffer
;
627 Stack
= IoGetCurrentIrpStackLocation(Irp
);
629 if (Stack
->Parameters
.FileSystemControl
.InputBufferLength
< sizeof(NTFS_FILE_RECORD_INPUT_BUFFER
) ||
630 Irp
->AssociatedIrp
.SystemBuffer
== NULL
)
632 DPRINT1("Invalid input! %d %p\n", Stack
->Parameters
.FileSystemControl
.InputBufferLength
, Irp
->AssociatedIrp
.SystemBuffer
);
633 return STATUS_INVALID_PARAMETER
;
636 if (Stack
->Parameters
.FileSystemControl
.OutputBufferLength
< (FIELD_OFFSET(NTFS_FILE_RECORD_OUTPUT_BUFFER
, FileRecordBuffer
) + DeviceExt
->NtfsInfo
.BytesPerFileRecord
) ||
637 Irp
->AssociatedIrp
.SystemBuffer
== NULL
)
639 DPRINT1("Invalid output! %d %p\n", Stack
->Parameters
.FileSystemControl
.OutputBufferLength
, Irp
->AssociatedIrp
.SystemBuffer
);
640 return STATUS_BUFFER_TOO_SMALL
;
643 FileRecord
= ExAllocatePoolWithTag(NonPagedPool
,
644 DeviceExt
->NtfsInfo
.BytesPerFileRecord
,
646 if (FileRecord
== NULL
)
648 return STATUS_INSUFFICIENT_RESOURCES
;
651 InputBuffer
= (PNTFS_FILE_RECORD_INPUT_BUFFER
)Irp
->AssociatedIrp
.SystemBuffer
;
653 MFTRecord
= InputBuffer
->FileReferenceNumber
.QuadPart
;
654 DPRINT1("Requesting: %I64x\n", MFTRecord
);
658 Status
= ReadFileRecord(DeviceExt
, MFTRecord
, FileRecord
);
659 if (NT_SUCCESS(Status
))
661 if (FileRecord
->Flags
& FRH_IN_USE
)
670 DPRINT1("Returning: %I64x\n", MFTRecord
);
671 OutputBuffer
= (PNTFS_FILE_RECORD_OUTPUT_BUFFER
)Irp
->AssociatedIrp
.SystemBuffer
;
672 OutputBuffer
->FileReferenceNumber
.QuadPart
= MFTRecord
;
673 OutputBuffer
->FileRecordLength
= DeviceExt
->NtfsInfo
.BytesPerFileRecord
;
674 RtlCopyMemory(OutputBuffer
->FileRecordBuffer
, FileRecord
, DeviceExt
->NtfsInfo
.BytesPerFileRecord
);
676 ExFreePoolWithTag(FileRecord
, TAG_NTFS
);
678 Irp
->IoStatus
.Information
= FIELD_OFFSET(NTFS_FILE_RECORD_OUTPUT_BUFFER
, FileRecordBuffer
) + DeviceExt
->NtfsInfo
.BytesPerFileRecord
;
680 return STATUS_SUCCESS
;
686 GetVolumeBitmap(PDEVICE_EXTENSION DeviceExt
,
689 NTSTATUS Status
= STATUS_SUCCESS
;
690 PIO_STACK_LOCATION Stack
;
691 PVOLUME_BITMAP_BUFFER BitmapBuffer
;
692 LONGLONG StartingLcn
;
693 PFILE_RECORD_HEADER BitmapRecord
;
694 PNTFS_ATTR_CONTEXT DataContext
;
695 ULONGLONG TotalClusters
;
697 BOOLEAN Overflow
= FALSE
;
699 DPRINT1("GetVolumeBitmap(%p, %p)\n", DeviceExt
, Irp
);
701 Stack
= IoGetCurrentIrpStackLocation(Irp
);
703 if (Stack
->Parameters
.FileSystemControl
.InputBufferLength
< sizeof(STARTING_LCN_INPUT_BUFFER
))
705 DPRINT1("Invalid input! %d\n", Stack
->Parameters
.FileSystemControl
.InputBufferLength
);
706 return STATUS_INVALID_PARAMETER
;
709 if (Stack
->Parameters
.FileSystemControl
.OutputBufferLength
< sizeof(VOLUME_BITMAP_BUFFER
))
711 DPRINT1("Invalid output! %d\n", Stack
->Parameters
.FileSystemControl
.OutputBufferLength
);
712 return STATUS_BUFFER_TOO_SMALL
;
715 BitmapBuffer
= NtfsGetUserBuffer(Irp
);
716 if (Irp
->RequestorMode
== UserMode
)
720 ProbeForRead(Stack
->Parameters
.FileSystemControl
.Type3InputBuffer
,
721 Stack
->Parameters
.FileSystemControl
.InputBufferLength
,
723 ProbeForWrite(BitmapBuffer
, Stack
->Parameters
.FileSystemControl
.OutputBufferLength
,
726 _SEH2_EXCEPT(EXCEPTION_EXECUTE_HANDLER
)
728 Status
= _SEH2_GetExceptionCode();
734 if (Stack
->Parameters
.FileSystemControl
.Type3InputBuffer
== NULL
||
735 BitmapBuffer
== NULL
)
737 Status
= STATUS_INVALID_PARAMETER
;
741 if (!NT_SUCCESS(Status
))
743 DPRINT1("Invalid buffer! %p %p\n", Stack
->Parameters
.FileSystemControl
.Type3InputBuffer
, BitmapBuffer
);
747 StartingLcn
= ((PSTARTING_LCN_INPUT_BUFFER
)Stack
->Parameters
.FileSystemControl
.Type3InputBuffer
)->StartingLcn
.QuadPart
;
748 if (StartingLcn
> DeviceExt
->NtfsInfo
.ClusterCount
)
750 DPRINT1("Requested bitmap start beyond partition end: %I64x %I64x\n", DeviceExt
->NtfsInfo
.ClusterCount
, StartingLcn
);
751 return STATUS_INVALID_PARAMETER
;
754 /* Round down to a multiple of 8 */
755 StartingLcn
= StartingLcn
& ~7;
756 TotalClusters
= DeviceExt
->NtfsInfo
.ClusterCount
- StartingLcn
;
757 ToCopy
= TotalClusters
/ 8;
758 if ((ToCopy
+ FIELD_OFFSET(VOLUME_BITMAP_BUFFER
, Buffer
)) > Stack
->Parameters
.FileSystemControl
.OutputBufferLength
)
760 DPRINT1("Buffer too small: %x, needed: %x\n", Stack
->Parameters
.FileSystemControl
.OutputBufferLength
, (ToCopy
+ FIELD_OFFSET(VOLUME_BITMAP_BUFFER
, Buffer
)));
762 ToCopy
= Stack
->Parameters
.FileSystemControl
.OutputBufferLength
- FIELD_OFFSET(VOLUME_BITMAP_BUFFER
, Buffer
);
765 BitmapRecord
= ExAllocatePoolWithTag(NonPagedPool
,
766 DeviceExt
->NtfsInfo
.BytesPerFileRecord
,
768 if (BitmapRecord
== NULL
)
770 return STATUS_INSUFFICIENT_RESOURCES
;
773 Status
= ReadFileRecord(DeviceExt
, NTFS_FILE_BITMAP
, BitmapRecord
);
774 if (!NT_SUCCESS(Status
))
776 DPRINT1("Failed reading volume bitmap: %lx\n", Status
);
777 ExFreePoolWithTag(BitmapRecord
, TAG_NTFS
);
781 Status
= FindAttribute(DeviceExt
, BitmapRecord
, AttributeData
, L
"", 0, &DataContext
);
782 if (!NT_SUCCESS(Status
))
784 DPRINT1("Failed find $DATA for bitmap: %lx\n", Status
);
785 ExFreePoolWithTag(BitmapRecord
, TAG_NTFS
);
789 BitmapBuffer
->StartingLcn
.QuadPart
= StartingLcn
;
790 BitmapBuffer
->BitmapSize
.QuadPart
= ToCopy
* 8;
792 Irp
->IoStatus
.Information
= FIELD_OFFSET(VOLUME_BITMAP_BUFFER
, Buffer
);
795 Irp
->IoStatus
.Information
+= ReadAttribute(DeviceExt
, DataContext
, StartingLcn
/ 8, (PCHAR
)BitmapBuffer
->Buffer
, ToCopy
);
796 Status
= (Overflow
? STATUS_BUFFER_OVERFLOW
: STATUS_SUCCESS
);
798 _SEH2_EXCEPT(EXCEPTION_EXECUTE_HANDLER
)
800 Status
= _SEH2_GetExceptionCode();
803 ReleaseAttributeContext(DataContext
);
804 ExFreePoolWithTag(BitmapRecord
, TAG_NTFS
);
806 return STATUS_SUCCESS
;
812 NtfsUserFsRequest(PDEVICE_OBJECT DeviceObject
,
816 PIO_STACK_LOCATION Stack
;
817 PDEVICE_EXTENSION DeviceExt
;
819 DPRINT1("NtfsUserFsRequest(%p, %p)\n", DeviceObject
, Irp
);
821 Stack
= IoGetCurrentIrpStackLocation(Irp
);
822 DeviceExt
= DeviceObject
->DeviceExtension
;
823 switch (Stack
->Parameters
.FileSystemControl
.FsControlCode
)
825 case FSCTL_CREATE_USN_JOURNAL
:
826 case FSCTL_DELETE_USN_JOURNAL
:
827 case FSCTL_ENUM_USN_DATA
:
828 case FSCTL_EXTEND_VOLUME
:
829 //case FSCTL_GET_RETRIEVAL_POINTER_BASE:
830 case FSCTL_GET_RETRIEVAL_POINTERS
:
831 case FSCTL_LOCK_VOLUME
:
832 //case FSCTL_LOOKUP_STREAM_FROM_CLUSTER:
833 case FSCTL_MARK_HANDLE
:
834 case FSCTL_MOVE_FILE
:
835 case FSCTL_QUERY_USN_JOURNAL
:
836 case FSCTL_READ_FILE_USN_DATA
:
837 case FSCTL_READ_USN_JOURNAL
:
838 //case FSCTL_SHRINK_VOLUME:
839 case FSCTL_UNLOCK_VOLUME
:
840 case FSCTL_WRITE_USN_CLOSE_RECORD
:
842 DPRINT1("Unimplemented user request: %x\n", Stack
->Parameters
.FileSystemControl
.FsControlCode
);
843 Status
= STATUS_NOT_IMPLEMENTED
;
846 case FSCTL_GET_NTFS_VOLUME_DATA
:
847 Status
= GetNfsVolumeData(DeviceExt
, Irp
);
850 case FSCTL_GET_NTFS_FILE_RECORD
:
851 Status
= GetNtfsFileRecord(DeviceExt
, Irp
);
854 case FSCTL_GET_VOLUME_BITMAP
:
855 Status
= GetVolumeBitmap(DeviceExt
, Irp
);
859 DPRINT("Invalid user request: %x\n", Stack
->Parameters
.FileSystemControl
.FsControlCode
);
860 Status
= STATUS_INVALID_DEVICE_REQUEST
;
869 NtfsFileSystemControl(PNTFS_IRP_CONTEXT IrpContext
)
873 PDEVICE_OBJECT DeviceObject
;
875 DPRINT1("NtfsFileSystemControl() called\n");
877 DeviceObject
= IrpContext
->DeviceObject
;
878 Irp
= IrpContext
->Irp
;
879 Irp
->IoStatus
.Information
= 0;
881 switch (IrpContext
->MinorFunction
)
883 case IRP_MN_KERNEL_CALL
:
884 DPRINT1("NTFS: IRP_MN_USER_FS_REQUEST\n");
885 Status
= STATUS_INVALID_DEVICE_REQUEST
;
888 case IRP_MN_USER_FS_REQUEST
:
889 Status
= NtfsUserFsRequest(DeviceObject
, Irp
);
892 case IRP_MN_MOUNT_VOLUME
:
893 DPRINT("NTFS: IRP_MN_MOUNT_VOLUME\n");
894 Status
= NtfsMountVolume(DeviceObject
, Irp
);
897 case IRP_MN_VERIFY_VOLUME
:
898 DPRINT1("NTFS: IRP_MN_VERIFY_VOLUME\n");
899 Status
= NtfsVerifyVolume(DeviceObject
, Irp
);
903 DPRINT1("NTFS FSC: MinorFunction %d\n", IrpContext
->MinorFunction
);
904 Status
= STATUS_INVALID_DEVICE_REQUEST
;