reactos/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 <usetup.h>
  30
  31 #define NDEBUG
  32 #include <debug.h>
  33
  34 /* FUNCTIONS ****************************************************************/
  35
  36 NTSTATUS
  37 GetSourcePaths(PUNICODE_STRING SourcePath,
  38                PUNICODE_STRING SourceRootPath)
  39 {
  40   OBJECT_ATTRIBUTES ObjectAttributes;
  41   UNICODE_STRING LinkName;
  42   UNICODE_STRING SourceName;
  43   WCHAR SourceBuffer[MAX_PATH];
  44   HANDLE Handle;
  45   NTSTATUS Status;
  46   ULONG Length;
  47   PWCHAR Ptr;
  48
  49   RtlInitUnicodeString(&LinkName,
  50                        L"\\SystemRoot");
  51
  52   InitializeObjectAttributes(&ObjectAttributes,
  53                              &LinkName,
  54                              OBJ_OPENLINK,
  55                              NULL,
  56                              NULL);
  57
  58   Status = NtOpenSymbolicLinkObject(&Handle,
  59                                     SYMBOLIC_LINK_ALL_ACCESS,
  60                                     &ObjectAttributes);
  61   if (!NT_SUCCESS(Status))
  62     return(Status);
  63
  64   SourceName.Length = 0;
  65   SourceName.MaximumLength = MAX_PATH * sizeof(WCHAR);
  66   SourceName.Buffer = SourceBuffer;
  67
  68   Status = NtQuerySymbolicLinkObject(Handle,
  69                                      &SourceName,
  70                                      &Length);
  71   NtClose(Handle);
  72
  73   if (NT_SUCCESS(Status))
  74     {
  75       RtlCreateUnicodeString(SourcePath,
  76                              SourceName.Buffer);
  77
  78       /* strip trailing directory */
  79       Ptr = wcsrchr(SourceName.Buffer, L'\\');
  80 //      if ((Ptr != NULL) &&
  81 //        (wcsicmp(Ptr, L"\\reactos") == 0))
  82       if (Ptr != NULL)
  83         *Ptr = 0;
  84
  85       RtlCreateUnicodeString(SourceRootPath,
  86                              SourceName.Buffer);
  87     }
  88
  89   NtClose(Handle);
  90
  91   return(STATUS_SUCCESS);
  92 }
  93
  94
  95 /* EOF */