subsys/system/usetup/drivesup.c

   1 /*
   2  *  ReactOS kernel
   3  *  Copyright (C) 2002 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  * COPYRIGHT:       See COPYING in the top level directory
  21  * PROJECT:         ReactOS text-mode setup
  22  * FILE:            subsys/system/usetup/drivesup.c
  23  * PURPOSE:         Drive support functions
  24  * PROGRAMMER:      Eric Kohl
  25  */
  26
  27 /* INCLUDES *****************************************************************/
  28
  29 #include <ddk/ntddk.h>
  30 #include <ntdll/rtl.h>
  31
  32 #include "usetup.h"
  33 #include "drivesup.h"
  34
  35
  36 /* FUNCTIONS ****************************************************************/
  37
  38 NTSTATUS
  39 GetSourcePaths(PUNICODE_STRING SourcePath,
  40                PUNICODE_STRING SourceRootPath)
  41 {
  42   OBJECT_ATTRIBUTES ObjectAttributes;
  43   UNICODE_STRING LinkName;
  44   UNICODE_STRING SourceName;
  45   WCHAR SourceBuffer[MAX_PATH];
  46   HANDLE Handle;
  47   NTSTATUS Status;
  48   ULONG Length;
  49   PWCHAR Ptr;
  50
  51   RtlInitUnicodeString(&LinkName,
  52                        L"\\SystemRoot");
  53
  54   InitializeObjectAttributes(&ObjectAttributes,
  55                              &LinkName,
  56                              OBJ_OPENLINK,
  57                              NULL,
  58                              NULL);
  59
  60   Status = NtOpenSymbolicLinkObject(&Handle,
  61                                     SYMBOLIC_LINK_ALL_ACCESS,
  62                                     &ObjectAttributes);
  63   if (!NT_SUCCESS(Status))
  64     return(Status);
  65
  66   SourceName.Length = 0;
  67   SourceName.MaximumLength = MAX_PATH * sizeof(WCHAR);
  68   SourceName.Buffer = SourceBuffer;
  69
  70   Status = NtQuerySymbolicLinkObject(Handle,
  71                                      &SourceName,
  72                                      &Length);
  73   NtClose(Handle);
  74
  75   if (NT_SUCCESS(Status))
  76     {
  77       RtlCreateUnicodeString(SourcePath,
  78                              SourceName.Buffer);
  79
  80       /* strip trailing directory */
  81       Ptr = wcsrchr(SourceName.Buffer, L'\\');
  82 //      if ((Ptr != NULL) &&
  83 //        (wcsicmp(Ptr, L"\\reactos") == 0))
  84       if (Ptr != NULL)
  85         *Ptr = 0;
  86
  87       RtlCreateUnicodeString(SourceRootPath,
  88                              SourceName.Buffer);
  89     }
  90
  91   NtClose(Handle);
  92
  93   return(STATUS_SUCCESS);
  94 }
  95
  96
  97 CHAR
  98 GetDriveLetter(ULONG DriveNumber,
  99                ULONG PartitionNumber)
 100 {
 101   OBJECT_ATTRIBUTES ObjectAttributes;
 102   UNICODE_STRING LinkName;
 103   WCHAR LinkBuffer[8];
 104   CHAR Letter;
 105   HANDLE LinkHandle;
 106   UNICODE_STRING TargetName;
 107   PWCHAR TargetBuffer;
 108   PWCHAR DeviceBuffer;
 109   ULONG Length;
 110   NTSTATUS Status;
 111
 112   wcscpy(LinkBuffer, L"\\??\\A:");
 113
 114   RtlInitUnicodeString(&LinkName,
 115                        LinkBuffer);
 116
 117   InitializeObjectAttributes(&ObjectAttributes,
 118                              &LinkName,
 119                              OBJ_OPENLINK,
 120                              NULL,
 121                              NULL);
 122
 123   TargetBuffer = RtlAllocateHeap(ProcessHeap, 0, MAX_PATH * sizeof(WCHAR));
 124   DeviceBuffer = RtlAllocateHeap(ProcessHeap, 0, MAX_PATH * sizeof(WCHAR));
 125
 126   TargetName.Length = 0;
 127   TargetName.MaximumLength = MAX_PATH * sizeof(WCHAR);
 128   TargetName.Buffer = TargetBuffer;
 129
 130   swprintf(DeviceBuffer,
 131            L"\\Device\\Harddisk%lu\\Partition%lu",
 132            DriveNumber,
 133            PartitionNumber);
 134
 135   for (Letter = 'C'; Letter <= 'Z'; Letter++)
 136     {
 137       LinkBuffer[4] = (WCHAR)Letter;
 138       TargetName.Length = 0;
 139
 140       Status = NtOpenSymbolicLinkObject(&LinkHandle,
 141                                         SYMBOLIC_LINK_ALL_ACCESS,
 142                                         &ObjectAttributes);
 143       if (NT_SUCCESS(Status))
 144         {
 145           Status = NtQuerySymbolicLinkObject(LinkHandle,
 146                                              &TargetName,
 147                                              &Length);
 148           NtClose(LinkHandle);
 149           if (NT_SUCCESS(Status))
 150             {
 151               if (_wcsicmp(DeviceBuffer, TargetBuffer) == 0)
 152                 {
 153                   RtlFreeHeap(ProcessHeap, 0, DeviceBuffer);
 154                   RtlFreeHeap(ProcessHeap, 0, TargetBuffer);
 155                   return(Letter);
 156                 }
 157             }
 158         }
 159     }
 160
 161   RtlFreeHeap(ProcessHeap, 0, DeviceBuffer);
 162   RtlFreeHeap(ProcessHeap, 0, TargetBuffer);
 163
 164   return((CHAR)0);
 165 }
 166
 167 /* EOF */