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