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
,
299 DeviceExt
->MasterFileTable
,
303 &DeviceExt
->MFTContext
,
304 &DeviceExt
->MftDataOffset
);
305 if (!NT_SUCCESS(Status
))
307 DPRINT1("Can't find data attribute for Master File Table.\n");
308 ExFreePool(DeviceExt
->MasterFileTable
);
312 VolumeRecord
= ExAllocatePoolWithTag(NonPagedPool
,
313 NtfsInfo
->BytesPerFileRecord
,
315 if (VolumeRecord
== NULL
)
317 DPRINT1("Allocation failed for volume record\n");
318 ExFreePool(DeviceExt
->MasterFileTable
);
319 return STATUS_INSUFFICIENT_RESOURCES
;
322 /* Read Volume File (MFT index 3) */
323 DeviceExt
->StorageDevice
= DeviceObject
;
324 Status
= ReadFileRecord(DeviceExt
,
327 if (!NT_SUCCESS(Status
))
329 DPRINT1("Failed reading volume file\n");
330 ExFreePool(VolumeRecord
);
331 ExFreePool(DeviceExt
->MasterFileTable
);
335 /* Enumerate attributes */
336 NtfsDumpFileAttributes(DeviceExt
, DeviceExt
->MasterFileTable
);
338 /* Enumerate attributes */
339 NtfsDumpFileAttributes(DeviceExt
, VolumeRecord
);
341 /* Get volume name */
342 Status
= FindAttribute(DeviceExt
, VolumeRecord
, AttributeVolumeName
, L
"", 0, &AttrCtxt
, NULL
);
344 if (NT_SUCCESS(Status
) && AttrCtxt
->Record
.Resident
.ValueLength
!= 0)
346 Attribute
= &AttrCtxt
->Record
;
347 DPRINT("Data length %lu\n", AttributeDataLength(Attribute
));
348 NtfsInfo
->VolumeLabelLength
=
349 min (Attribute
->Resident
.ValueLength
, MAXIMUM_VOLUME_LABEL_LENGTH
);
350 RtlCopyMemory(NtfsInfo
->VolumeLabel
,
351 (PVOID
)((ULONG_PTR
)Attribute
+ Attribute
->Resident
.ValueOffset
),
352 NtfsInfo
->VolumeLabelLength
);
353 VolumeNameU
= NtfsInfo
->VolumeLabel
;
357 NtfsInfo
->VolumeLabelLength
= 0;
361 if (NT_SUCCESS(Status
))
363 ReleaseAttributeContext(AttrCtxt
);
366 VolumeFcb
= NtfsCreateFCB(VolumeNameU
, NULL
, DeviceExt
);
367 if (VolumeFcb
== NULL
)
369 DPRINT1("Failed allocating volume FCB\n");
370 ExFreePool(VolumeRecord
);
371 ExFreePool(DeviceExt
->MasterFileTable
);
372 return STATUS_INSUFFICIENT_RESOURCES
;
375 VolumeFcb
->Flags
= FCB_IS_VOLUME
;
376 VolumeFcb
->RFCB
.FileSize
.QuadPart
= DeviceExt
->NtfsInfo
.SectorCount
* DeviceExt
->NtfsInfo
.BytesPerSector
;
377 VolumeFcb
->RFCB
.ValidDataLength
= VolumeFcb
->RFCB
.FileSize
;
378 VolumeFcb
->RFCB
.AllocationSize
= VolumeFcb
->RFCB
.FileSize
;
379 VolumeFcb
->MFTIndex
= 0;
380 DeviceExt
->VolumeFcb
= VolumeFcb
;
382 /* Get volume information */
383 Status
= FindAttribute(DeviceExt
, VolumeRecord
, AttributeVolumeInformation
, L
"", 0, &AttrCtxt
, NULL
);
385 if (NT_SUCCESS(Status
) && AttrCtxt
->Record
.Resident
.ValueLength
!= 0)
387 Attribute
= &AttrCtxt
->Record
;
388 DPRINT("Data length %lu\n", AttributeDataLength (Attribute
));
389 VolumeInfo
= (PVOID
)((ULONG_PTR
)Attribute
+ Attribute
->Resident
.ValueOffset
);
391 NtfsInfo
->MajorVersion
= VolumeInfo
->MajorVersion
;
392 NtfsInfo
->MinorVersion
= VolumeInfo
->MinorVersion
;
393 NtfsInfo
->Flags
= VolumeInfo
->Flags
;
396 if (NT_SUCCESS(Status
))
398 ReleaseAttributeContext(AttrCtxt
);
401 ExFreePool(VolumeRecord
);
403 NtfsInfo
->MftZoneReservation
= NtfsQueryMftZoneReservation();
411 NtfsMountVolume(PDEVICE_OBJECT DeviceObject
,
414 PDEVICE_OBJECT NewDeviceObject
= NULL
;
415 PDEVICE_OBJECT DeviceToMount
;
416 PIO_STACK_LOCATION Stack
;
417 PNTFS_FCB Fcb
= NULL
;
418 PNTFS_CCB Ccb
= NULL
;
419 PNTFS_VCB Vcb
= NULL
;
422 DPRINT1("NtfsMountVolume() called\n");
424 if (DeviceObject
!= NtfsGlobalData
->DeviceObject
)
426 Status
= STATUS_INVALID_DEVICE_REQUEST
;
430 Stack
= IoGetCurrentIrpStackLocation(Irp
);
431 DeviceToMount
= Stack
->Parameters
.MountVolume
.DeviceObject
;
433 Status
= NtfsHasFileSystem(DeviceToMount
);
434 if (!NT_SUCCESS(Status
))
439 Status
= IoCreateDevice(NtfsGlobalData
->DriverObject
,
440 sizeof(DEVICE_EXTENSION
),
442 FILE_DEVICE_DISK_FILE_SYSTEM
,
446 if (!NT_SUCCESS(Status
))
449 NewDeviceObject
->Flags
|= DO_DIRECT_IO
;
450 Vcb
= (PVOID
)NewDeviceObject
->DeviceExtension
;
451 RtlZeroMemory(Vcb
, sizeof(NTFS_VCB
));
453 Vcb
->Identifier
.Type
= NTFS_TYPE_VCB
;
454 Vcb
->Identifier
.Size
= sizeof(NTFS_TYPE_VCB
);
456 Status
= NtfsGetVolumeData(DeviceToMount
,
458 if (!NT_SUCCESS(Status
))
461 NewDeviceObject
->Vpb
= DeviceToMount
->Vpb
;
463 Vcb
->StorageDevice
= DeviceToMount
;
464 Vcb
->StorageDevice
->Vpb
->DeviceObject
= NewDeviceObject
;
465 Vcb
->StorageDevice
->Vpb
->RealDevice
= Vcb
->StorageDevice
;
466 Vcb
->StorageDevice
->Vpb
->Flags
|= VPB_MOUNTED
;
467 NewDeviceObject
->StackSize
= Vcb
->StorageDevice
->StackSize
+ 1;
468 NewDeviceObject
->Flags
&= ~DO_DEVICE_INITIALIZING
;
470 Vcb
->StreamFileObject
= IoCreateStreamFileObject(NULL
,
473 InitializeListHead(&Vcb
->FcbListHead
);
475 Fcb
= NtfsCreateFCB(NULL
, NULL
, Vcb
);
478 Status
= STATUS_INSUFFICIENT_RESOURCES
;
482 Ccb
= ExAllocatePoolWithTag(NonPagedPool
,
487 Status
= STATUS_INSUFFICIENT_RESOURCES
;
491 RtlZeroMemory(Ccb
, sizeof(NTFS_CCB
));
493 Ccb
->Identifier
.Type
= NTFS_TYPE_CCB
;
494 Ccb
->Identifier
.Size
= sizeof(NTFS_TYPE_CCB
);
496 Vcb
->StreamFileObject
->FsContext
= Fcb
;
497 Vcb
->StreamFileObject
->FsContext2
= Ccb
;
498 Vcb
->StreamFileObject
->SectionObjectPointer
= &Fcb
->SectionObjectPointers
;
499 Vcb
->StreamFileObject
->PrivateCacheMap
= NULL
;
500 Vcb
->StreamFileObject
->Vpb
= Vcb
->Vpb
;
501 Ccb
->PtrFileObject
= Vcb
->StreamFileObject
;
502 Fcb
->FileObject
= Vcb
->StreamFileObject
;
503 Fcb
->Vcb
= (PDEVICE_EXTENSION
)Vcb
->StorageDevice
;
505 Fcb
->Flags
= FCB_IS_VOLUME_STREAM
;
507 Fcb
->RFCB
.FileSize
.QuadPart
= Vcb
->NtfsInfo
.SectorCount
* Vcb
->NtfsInfo
.BytesPerSector
;
508 Fcb
->RFCB
.ValidDataLength
.QuadPart
= Vcb
->NtfsInfo
.SectorCount
* Vcb
->NtfsInfo
.BytesPerSector
;
509 Fcb
->RFCB
.AllocationSize
.QuadPart
= Vcb
->NtfsInfo
.SectorCount
* Vcb
->NtfsInfo
.BytesPerSector
; /* Correct? */
511 // Fcb->Entry.ExtentLocationL = 0;
512 // Fcb->Entry.DataLengthL = DeviceExt->CdInfo.VolumeSpaceSize * BLOCKSIZE;
516 CcInitializeCacheMap(Vcb
->StreamFileObject
,
517 (PCC_FILE_SIZES
)(&Fcb
->RFCB
.AllocationSize
),
519 &(NtfsGlobalData
->CacheMgrCallbacks
),
522 _SEH2_EXCEPT(EXCEPTION_EXECUTE_HANDLER
)
524 Status
= _SEH2_GetExceptionCode();
529 ExInitializeResourceLite(&Vcb
->DirResource
);
531 KeInitializeSpinLock(&Vcb
->FcbListLock
);
533 /* Get serial number */
534 NewDeviceObject
->Vpb
->SerialNumber
= Vcb
->NtfsInfo
.SerialNumber
;
536 /* Get volume label */
537 NewDeviceObject
->Vpb
->VolumeLabelLength
= Vcb
->NtfsInfo
.VolumeLabelLength
;
538 RtlCopyMemory(NewDeviceObject
->Vpb
->VolumeLabel
,
539 Vcb
->NtfsInfo
.VolumeLabel
,
540 Vcb
->NtfsInfo
.VolumeLabelLength
);
542 FsRtlNotifyVolumeEvent(Vcb
->StreamFileObject
, FSRTL_VOLUME_MOUNT
);
544 Status
= STATUS_SUCCESS
;
547 if (!NT_SUCCESS(Status
))
550 if (Vcb
&& Vcb
->StreamFileObject
)
551 ObDereferenceObject(Vcb
->StreamFileObject
);
560 IoDeleteDevice(NewDeviceObject
);
563 DPRINT("NtfsMountVolume() done (Status: %lx)\n", Status
);
571 NtfsVerifyVolume(PDEVICE_OBJECT DeviceObject
,
574 UNREFERENCED_PARAMETER(DeviceObject
);
575 UNREFERENCED_PARAMETER(Irp
);
576 DPRINT1("NtfsVerifyVolume() called\n");
577 return STATUS_WRONG_VOLUME
;
583 GetNfsVolumeData(PDEVICE_EXTENSION DeviceExt
,
586 PIO_STACK_LOCATION Stack
;
587 PNTFS_VOLUME_DATA_BUFFER DataBuffer
;
588 PNTFS_ATTR_RECORD Attribute
;
589 FIND_ATTR_CONTXT Context
;
592 DataBuffer
= (PNTFS_VOLUME_DATA_BUFFER
)Irp
->AssociatedIrp
.SystemBuffer
;
593 Stack
= IoGetCurrentIrpStackLocation(Irp
);
595 if (Stack
->Parameters
.FileSystemControl
.OutputBufferLength
< sizeof(NTFS_VOLUME_DATA_BUFFER
) ||
596 Irp
->UserBuffer
== NULL
)
598 DPRINT1("Invalid output! %d %p\n", Stack
->Parameters
.FileSystemControl
.OutputBufferLength
, Irp
->UserBuffer
);
599 return STATUS_INVALID_PARAMETER
;
602 DataBuffer
->VolumeSerialNumber
.QuadPart
= DeviceExt
->NtfsInfo
.SerialNumber
;
603 DataBuffer
->NumberSectors
.QuadPart
= DeviceExt
->NtfsInfo
.SectorCount
;
604 DataBuffer
->TotalClusters
.QuadPart
= DeviceExt
->NtfsInfo
.ClusterCount
;
605 DataBuffer
->FreeClusters
.QuadPart
= NtfsGetFreeClusters(DeviceExt
);
606 DataBuffer
->TotalReserved
.QuadPart
= 0LL; // FIXME
607 DataBuffer
->BytesPerSector
= DeviceExt
->NtfsInfo
.BytesPerSector
;
608 DataBuffer
->BytesPerCluster
= DeviceExt
->NtfsInfo
.BytesPerCluster
;
609 DataBuffer
->BytesPerFileRecordSegment
= DeviceExt
->NtfsInfo
.BytesPerFileRecord
;
610 DataBuffer
->ClustersPerFileRecordSegment
= DeviceExt
->NtfsInfo
.BytesPerFileRecord
/ DeviceExt
->NtfsInfo
.BytesPerCluster
;
611 DataBuffer
->MftStartLcn
.QuadPart
= DeviceExt
->NtfsInfo
.MftStart
.QuadPart
;
612 DataBuffer
->Mft2StartLcn
.QuadPart
= DeviceExt
->NtfsInfo
.MftMirrStart
.QuadPart
;
613 DataBuffer
->MftZoneStart
.QuadPart
= 0; // FIXME
614 DataBuffer
->MftZoneEnd
.QuadPart
= 0; // FIXME
616 Status
= FindFirstAttribute(&Context
, DeviceExt
, DeviceExt
->MasterFileTable
, FALSE
, &Attribute
);
617 while (NT_SUCCESS(Status
))
619 if (Attribute
->Type
== AttributeData
)
621 ASSERT(Attribute
->IsNonResident
);
622 DataBuffer
->MftValidDataLength
.QuadPart
= Attribute
->NonResident
.DataSize
;
627 Status
= FindNextAttribute(&Context
, &Attribute
);
629 FindCloseAttribute(&Context
);
631 Irp
->IoStatus
.Information
= sizeof(NTFS_VOLUME_DATA_BUFFER
);
633 if (Stack
->Parameters
.FileSystemControl
.OutputBufferLength
>= sizeof(NTFS_EXTENDED_VOLUME_DATA
) + sizeof(NTFS_VOLUME_DATA_BUFFER
))
635 PNTFS_EXTENDED_VOLUME_DATA ExtendedData
= (PNTFS_EXTENDED_VOLUME_DATA
)((ULONG_PTR
)Irp
->UserBuffer
+ sizeof(NTFS_VOLUME_DATA_BUFFER
));
637 ExtendedData
->ByteCount
= sizeof(NTFS_EXTENDED_VOLUME_DATA
);
638 ExtendedData
->MajorVersion
= DeviceExt
->NtfsInfo
.MajorVersion
;
639 ExtendedData
->MinorVersion
= DeviceExt
->NtfsInfo
.MinorVersion
;
640 Irp
->IoStatus
.Information
+= sizeof(NTFS_EXTENDED_VOLUME_DATA
);
643 return STATUS_SUCCESS
;
649 GetNtfsFileRecord(PDEVICE_EXTENSION DeviceExt
,
653 PIO_STACK_LOCATION Stack
;
654 PNTFS_FILE_RECORD_INPUT_BUFFER InputBuffer
;
655 PFILE_RECORD_HEADER FileRecord
;
656 PNTFS_FILE_RECORD_OUTPUT_BUFFER OutputBuffer
;
659 Stack
= IoGetCurrentIrpStackLocation(Irp
);
661 if (Stack
->Parameters
.FileSystemControl
.InputBufferLength
< sizeof(NTFS_FILE_RECORD_INPUT_BUFFER
) ||
662 Irp
->AssociatedIrp
.SystemBuffer
== NULL
)
664 DPRINT1("Invalid input! %d %p\n", Stack
->Parameters
.FileSystemControl
.InputBufferLength
, Irp
->AssociatedIrp
.SystemBuffer
);
665 return STATUS_INVALID_PARAMETER
;
668 if (Stack
->Parameters
.FileSystemControl
.OutputBufferLength
< (FIELD_OFFSET(NTFS_FILE_RECORD_OUTPUT_BUFFER
, FileRecordBuffer
) + DeviceExt
->NtfsInfo
.BytesPerFileRecord
) ||
669 Irp
->AssociatedIrp
.SystemBuffer
== NULL
)
671 DPRINT1("Invalid output! %d %p\n", Stack
->Parameters
.FileSystemControl
.OutputBufferLength
, Irp
->AssociatedIrp
.SystemBuffer
);
672 return STATUS_BUFFER_TOO_SMALL
;
675 FileRecord
= ExAllocatePoolWithTag(NonPagedPool
,
676 DeviceExt
->NtfsInfo
.BytesPerFileRecord
,
678 if (FileRecord
== NULL
)
680 return STATUS_INSUFFICIENT_RESOURCES
;
683 InputBuffer
= (PNTFS_FILE_RECORD_INPUT_BUFFER
)Irp
->AssociatedIrp
.SystemBuffer
;
685 MFTRecord
= InputBuffer
->FileReferenceNumber
.QuadPart
;
686 DPRINT1("Requesting: %I64x\n", MFTRecord
);
690 Status
= ReadFileRecord(DeviceExt
, MFTRecord
, FileRecord
);
691 if (NT_SUCCESS(Status
))
693 if (FileRecord
->Flags
& FRH_IN_USE
)
702 DPRINT1("Returning: %I64x\n", MFTRecord
);
703 OutputBuffer
= (PNTFS_FILE_RECORD_OUTPUT_BUFFER
)Irp
->AssociatedIrp
.SystemBuffer
;
704 OutputBuffer
->FileReferenceNumber
.QuadPart
= MFTRecord
;
705 OutputBuffer
->FileRecordLength
= DeviceExt
->NtfsInfo
.BytesPerFileRecord
;
706 RtlCopyMemory(OutputBuffer
->FileRecordBuffer
, FileRecord
, DeviceExt
->NtfsInfo
.BytesPerFileRecord
);
708 ExFreePoolWithTag(FileRecord
, TAG_NTFS
);
710 Irp
->IoStatus
.Information
= FIELD_OFFSET(NTFS_FILE_RECORD_OUTPUT_BUFFER
, FileRecordBuffer
) + DeviceExt
->NtfsInfo
.BytesPerFileRecord
;
712 return STATUS_SUCCESS
;
718 GetVolumeBitmap(PDEVICE_EXTENSION DeviceExt
,
721 NTSTATUS Status
= STATUS_SUCCESS
;
722 PIO_STACK_LOCATION Stack
;
723 PVOLUME_BITMAP_BUFFER BitmapBuffer
;
724 LONGLONG StartingLcn
;
725 PFILE_RECORD_HEADER BitmapRecord
;
726 PNTFS_ATTR_CONTEXT DataContext
;
727 ULONGLONG TotalClusters
;
729 BOOLEAN Overflow
= FALSE
;
731 DPRINT1("GetVolumeBitmap(%p, %p)\n", DeviceExt
, Irp
);
733 Stack
= IoGetCurrentIrpStackLocation(Irp
);
735 if (Stack
->Parameters
.FileSystemControl
.InputBufferLength
< sizeof(STARTING_LCN_INPUT_BUFFER
))
737 DPRINT1("Invalid input! %d\n", Stack
->Parameters
.FileSystemControl
.InputBufferLength
);
738 return STATUS_INVALID_PARAMETER
;
741 if (Stack
->Parameters
.FileSystemControl
.OutputBufferLength
< sizeof(VOLUME_BITMAP_BUFFER
))
743 DPRINT1("Invalid output! %d\n", Stack
->Parameters
.FileSystemControl
.OutputBufferLength
);
744 return STATUS_BUFFER_TOO_SMALL
;
747 BitmapBuffer
= NtfsGetUserBuffer(Irp
, FALSE
);
748 if (Irp
->RequestorMode
== UserMode
)
752 ProbeForRead(Stack
->Parameters
.FileSystemControl
.Type3InputBuffer
,
753 Stack
->Parameters
.FileSystemControl
.InputBufferLength
,
755 ProbeForWrite(BitmapBuffer
, Stack
->Parameters
.FileSystemControl
.OutputBufferLength
,
758 _SEH2_EXCEPT(EXCEPTION_EXECUTE_HANDLER
)
760 Status
= _SEH2_GetExceptionCode();
766 if (Stack
->Parameters
.FileSystemControl
.Type3InputBuffer
== NULL
||
767 BitmapBuffer
== NULL
)
769 Status
= STATUS_INVALID_PARAMETER
;
773 if (!NT_SUCCESS(Status
))
775 DPRINT1("Invalid buffer! %p %p\n", Stack
->Parameters
.FileSystemControl
.Type3InputBuffer
, BitmapBuffer
);
779 StartingLcn
= ((PSTARTING_LCN_INPUT_BUFFER
)Stack
->Parameters
.FileSystemControl
.Type3InputBuffer
)->StartingLcn
.QuadPart
;
780 if (StartingLcn
> DeviceExt
->NtfsInfo
.ClusterCount
)
782 DPRINT1("Requested bitmap start beyond partition end: %I64x %I64x\n", DeviceExt
->NtfsInfo
.ClusterCount
, StartingLcn
);
783 return STATUS_INVALID_PARAMETER
;
786 /* Round down to a multiple of 8 */
787 StartingLcn
= StartingLcn
& ~7;
788 TotalClusters
= DeviceExt
->NtfsInfo
.ClusterCount
- StartingLcn
;
789 ToCopy
= TotalClusters
/ 8;
790 if ((ToCopy
+ FIELD_OFFSET(VOLUME_BITMAP_BUFFER
, Buffer
)) > Stack
->Parameters
.FileSystemControl
.OutputBufferLength
)
792 DPRINT1("Buffer too small: %x, needed: %x\n", Stack
->Parameters
.FileSystemControl
.OutputBufferLength
, (ToCopy
+ FIELD_OFFSET(VOLUME_BITMAP_BUFFER
, Buffer
)));
794 ToCopy
= Stack
->Parameters
.FileSystemControl
.OutputBufferLength
- FIELD_OFFSET(VOLUME_BITMAP_BUFFER
, Buffer
);
797 BitmapRecord
= ExAllocatePoolWithTag(NonPagedPool
,
798 DeviceExt
->NtfsInfo
.BytesPerFileRecord
,
800 if (BitmapRecord
== NULL
)
802 return STATUS_INSUFFICIENT_RESOURCES
;
805 Status
= ReadFileRecord(DeviceExt
, NTFS_FILE_BITMAP
, BitmapRecord
);
806 if (!NT_SUCCESS(Status
))
808 DPRINT1("Failed reading volume bitmap: %lx\n", Status
);
809 ExFreePoolWithTag(BitmapRecord
, TAG_NTFS
);
813 Status
= FindAttribute(DeviceExt
, BitmapRecord
, AttributeData
, L
"", 0, &DataContext
, NULL
);
814 if (!NT_SUCCESS(Status
))
816 DPRINT1("Failed find $DATA for bitmap: %lx\n", Status
);
817 ExFreePoolWithTag(BitmapRecord
, TAG_NTFS
);
821 BitmapBuffer
->StartingLcn
.QuadPart
= StartingLcn
;
822 BitmapBuffer
->BitmapSize
.QuadPart
= ToCopy
* 8;
824 Irp
->IoStatus
.Information
= FIELD_OFFSET(VOLUME_BITMAP_BUFFER
, Buffer
);
827 Irp
->IoStatus
.Information
+= ReadAttribute(DeviceExt
, DataContext
, StartingLcn
/ 8, (PCHAR
)BitmapBuffer
->Buffer
, ToCopy
);
828 Status
= (Overflow
? STATUS_BUFFER_OVERFLOW
: STATUS_SUCCESS
);
830 _SEH2_EXCEPT(EXCEPTION_EXECUTE_HANDLER
)
832 Status
= _SEH2_GetExceptionCode();
835 ReleaseAttributeContext(DataContext
);
836 ExFreePoolWithTag(BitmapRecord
, TAG_NTFS
);
844 LockOrUnlockVolume(PDEVICE_EXTENSION DeviceExt
,
848 PFILE_OBJECT FileObject
;
850 PIO_STACK_LOCATION Stack
;
852 DPRINT("LockOrUnlockVolume(%p, %p, %d)\n", DeviceExt
, Irp
, Lock
);
854 Stack
= IoGetCurrentIrpStackLocation(Irp
);
855 FileObject
= Stack
->FileObject
;
856 Fcb
= FileObject
->FsContext
;
858 /* Only allow locking with the volume open */
859 if (!(Fcb
->Flags
& FCB_IS_VOLUME
))
861 return STATUS_ACCESS_DENIED
;
864 /* Bail out if it's already in the demanded state */
865 if (((DeviceExt
->Flags
& VCB_VOLUME_LOCKED
) && Lock
) ||
866 (!(DeviceExt
->Flags
& VCB_VOLUME_LOCKED
) && !Lock
))
868 return STATUS_ACCESS_DENIED
;
871 /* Deny locking if we're not alone */
872 if (Lock
&& DeviceExt
->OpenHandleCount
!= 1)
874 return STATUS_ACCESS_DENIED
;
877 /* Finally, proceed */
880 DeviceExt
->Flags
|= VCB_VOLUME_LOCKED
;
884 DeviceExt
->Flags
&= ~VCB_VOLUME_LOCKED
;
887 return STATUS_SUCCESS
;
893 NtfsUserFsRequest(PDEVICE_OBJECT DeviceObject
,
897 PIO_STACK_LOCATION Stack
;
898 PDEVICE_EXTENSION DeviceExt
;
900 DPRINT1("NtfsUserFsRequest(%p, %p)\n", DeviceObject
, Irp
);
902 Stack
= IoGetCurrentIrpStackLocation(Irp
);
903 DeviceExt
= DeviceObject
->DeviceExtension
;
904 switch (Stack
->Parameters
.FileSystemControl
.FsControlCode
)
906 case FSCTL_CREATE_USN_JOURNAL
:
907 case FSCTL_DELETE_USN_JOURNAL
:
908 case FSCTL_ENUM_USN_DATA
:
909 case FSCTL_EXTEND_VOLUME
:
910 //case FSCTL_GET_RETRIEVAL_POINTER_BASE:
911 case FSCTL_GET_RETRIEVAL_POINTERS
:
912 //case FSCTL_LOOKUP_STREAM_FROM_CLUSTER:
913 case FSCTL_MARK_HANDLE
:
914 case FSCTL_MOVE_FILE
:
915 case FSCTL_QUERY_USN_JOURNAL
:
916 case FSCTL_READ_FILE_USN_DATA
:
917 case FSCTL_READ_USN_JOURNAL
:
918 //case FSCTL_SHRINK_VOLUME:
919 case FSCTL_WRITE_USN_CLOSE_RECORD
:
921 DPRINT1("Unimplemented user request: %x\n", Stack
->Parameters
.FileSystemControl
.FsControlCode
);
922 Status
= STATUS_NOT_IMPLEMENTED
;
925 case FSCTL_LOCK_VOLUME
:
926 Status
= LockOrUnlockVolume(DeviceExt
, Irp
, TRUE
);
929 case FSCTL_UNLOCK_VOLUME
:
930 Status
= LockOrUnlockVolume(DeviceExt
, Irp
, FALSE
);
933 case FSCTL_GET_NTFS_VOLUME_DATA
:
934 Status
= GetNfsVolumeData(DeviceExt
, Irp
);
937 case FSCTL_GET_NTFS_FILE_RECORD
:
938 Status
= GetNtfsFileRecord(DeviceExt
, Irp
);
941 case FSCTL_GET_VOLUME_BITMAP
:
942 Status
= GetVolumeBitmap(DeviceExt
, Irp
);
946 DPRINT("Invalid user request: %x\n", Stack
->Parameters
.FileSystemControl
.FsControlCode
);
947 Status
= STATUS_INVALID_DEVICE_REQUEST
;
956 NtfsFileSystemControl(PNTFS_IRP_CONTEXT IrpContext
)
960 PDEVICE_OBJECT DeviceObject
;
962 DPRINT1("NtfsFileSystemControl() called\n");
964 DeviceObject
= IrpContext
->DeviceObject
;
965 Irp
= IrpContext
->Irp
;
966 Irp
->IoStatus
.Information
= 0;
968 switch (IrpContext
->MinorFunction
)
970 case IRP_MN_KERNEL_CALL
:
971 DPRINT1("NTFS: IRP_MN_USER_FS_REQUEST\n");
972 Status
= STATUS_INVALID_DEVICE_REQUEST
;
975 case IRP_MN_USER_FS_REQUEST
:
976 Status
= NtfsUserFsRequest(DeviceObject
, Irp
);
979 case IRP_MN_MOUNT_VOLUME
:
980 DPRINT("NTFS: IRP_MN_MOUNT_VOLUME\n");
981 Status
= NtfsMountVolume(DeviceObject
, Irp
);
984 case IRP_MN_VERIFY_VOLUME
:
985 DPRINT1("NTFS: IRP_MN_VERIFY_VOLUME\n");
986 Status
= NtfsVerifyVolume(DeviceObject
, Irp
);
990 DPRINT1("NTFS FSC: MinorFunction %d\n", IrpContext
->MinorFunction
);
991 Status
= STATUS_INVALID_DEVICE_REQUEST
;