reactos/drivers/fs/fs_rec/ntfs.c

   1 /*
   2  *  ReactOS kernel
   3  *  Copyright (C) 2002,2003 ReactOS Team
   4  *
   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.
   9  *
  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.
  14  *
  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., 675 Mass Ave, Cambridge, MA 02139, USA.
  18  */
  19 /*
  20  * COPYRIGHT:        See COPYING in the top level directory
  21  * PROJECT:          ReactOS kernel
  22  * FILE:             drivers/fs/fs_rec/ntfs.c
  23  * PURPOSE:          Filesystem recognizer driver
  24  * PROGRAMMER:       Eric Kohl
  25  */
  26
  27 /* INCLUDES *****************************************************************/
  28
  29 #define NDEBUG
  30 #include <debug.h>
  31
  32 #include "fs_rec.h"
  33
  34
  35 /* FUNCTIONS ****************************************************************/
  36
  37 static NTSTATUS
  38 FsRecIsNtfsVolume(IN PDEVICE_OBJECT DeviceObject)
  39 {
  40   DISK_GEOMETRY DiskGeometry;
  41   PUCHAR Buffer;
  42   ULONG Size;
  43   NTSTATUS Status;
  44
  45   Size = sizeof(DISK_GEOMETRY);
  46   Status = FsRecDeviceIoControl(DeviceObject,
  47                                 IOCTL_DISK_GET_DRIVE_GEOMETRY,
  48                                 NULL,
  49                                 0,
  50                                 &DiskGeometry,
  51                                 &Size);
  52   if (!NT_SUCCESS(Status))
  53     {
  54       DPRINT("FsRecDeviceIoControl() failed (Status %lx)\n", Status);
  55       return(Status);
  56     }
  57
  58   DPRINT("BytesPerSector: %lu\n", DiskGeometry.BytesPerSector);
  59   Buffer = ExAllocatePool(NonPagedPool,
  60                           DiskGeometry.BytesPerSector);
  61   if (Buffer == NULL)
  62     {
  63       return(STATUS_INSUFFICIENT_RESOURCES);
  64     }
  65
  66   Status = FsRecReadSectors(DeviceObject,
  67                             0, /* Partition boot sector */
  68                             1,
  69                             DiskGeometry.BytesPerSector,
  70                             Buffer);
  71   if (!NT_SUCCESS(Status))
  72     {
  73       DPRINT("FsRecReadSectors() failed (Status %lx)\n", Status);
  74       ExFreePool(Buffer);
  75       return(Status);
  76     }
  77
  78   DPRINT("NTFS-identifier: [%.8s]\n", &Buffer[3]);
  79   if (RtlCompareMemory(&Buffer[3], "NTFS    ", 8) == 8)
  80     {
  81       Status = STATUS_SUCCESS;
  82     }
  83   else
  84     {
  85       Status = STATUS_UNRECOGNIZED_VOLUME;
  86     }
  87
  88   ExFreePool(Buffer);
  89
  90   return(Status);
  91 }
  92
  93
  94 NTSTATUS
  95 FsRecNtfsFsControl(IN PDEVICE_OBJECT DeviceObject,
  96                    IN PIRP Irp)
  97 {
  98   PIO_STACK_LOCATION Stack;
  99   static UNICODE_STRING RegistryPath =
 100     RTL_CONSTANT_STRING(L"\\Registry\\Machine\\System\\CurrentControlSet\\Services\\Ntfs");
 101   NTSTATUS Status;
 102
 103   Stack = IoGetCurrentIrpStackLocation(Irp);
 104
 105   switch (Stack->MinorFunction)
 106     {
 107       case IRP_MN_MOUNT_VOLUME:
 108         DPRINT("NTFS: IRP_MN_MOUNT_VOLUME\n");
 109
 110         Status = FsRecIsNtfsVolume(Stack->Parameters.MountVolume.DeviceObject);
 111         if (NT_SUCCESS(Status))
 112           {
 113             DPRINT("Identified NTFS volume\n");
 114             Status = STATUS_FS_DRIVER_REQUIRED;
 115           }
 116         break;
 117
 118       case IRP_MN_LOAD_FILE_SYSTEM:
 119         DPRINT("NTFS: IRP_MN_LOAD_FILE_SYSTEM\n");
 120         Status = ZwLoadDriver(&RegistryPath);
 121         if (!NT_SUCCESS(Status))
 122           {
 123             DPRINT("ZwLoadDriver failed (Status %x)\n", Status);
 124           }
 125         else
 126           {
 127             IoUnregisterFileSystem(DeviceObject);
 128           }
 129         break;
 130
 131       default:
 132         DPRINT("NTFS: Unknown minor function %lx\n", Stack->MinorFunction);
 133         Status = STATUS_INVALID_DEVICE_REQUEST;
 134         break;
 135     }
 136   return(Status);
 137 }
 138
 139 /* EOF */