c87f74e51e411b2f53993b0225beca65bdd7961e
[reactos.git] / 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 /*
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 #if 0
98 CHAR
99 GetDriveLetter(IN ULONG DriveNumber,
100                IN ULONG PartitionNumber)
101 {
102   OBJECT_ATTRIBUTES ObjectAttributes;
103   UNICODE_STRING LinkName;
104   WCHAR LinkBuffer[8];
105   WCHAR Letter;
106   HANDLE LinkHandle;
107   UNICODE_STRING TargetName;
108   WCHAR TargetBuffer[MAX_PATH];
109 //  WCHAR DeviceBuffer[MAX_PATH];
110   ULONG Length;
111
112   wcscpy(LinkBuffer,
113          L"\\??\\A:");
114
115   RtlInitUnicodeString(&LinkName,
116                        LinkBuffer);
117
118   InitializeObjectAttributes(&ObjectAttributes,
119                              &LinkName,
120                              OBJ_OPENLINK,
121                              NULL,
122                              NULL);
123
124   TargetName.Length = 0;
125   TargetName.MaximumLength = MAX_PATH * sizeof(WCHAR);
126   TargetName.Buffer = TargetBuffer;
127
128   for (Letter = L'C'; Letter <= L'Z'; Letter++)
129     {
130       LinkBuffer[4] = Letter;
131       TargetName.Length = 0;
132
133       Status = NtOpenSymbolicLinkObject(&LinkHandle,
134                                         SYMBOLIC_LINK_ALL_ACCESS,
135                                         &ObjectAttributes);
136       if (NT_SUCCESS(Status))
137         {
138           Status = NtQuerySymbolicLinkObject(LinkHandle,
139                                              &TargetName,
140                                              &Length);
141           if (NT_SUCCESS(Status))
142             {
143
144
145
146             }
147           NtClose(LinkHandle);
148         }
149     }
150
151   return((CHAR)0);
152 }
153 #endif
154
155 #if 0
156 STATUS
157 GetFileSystem()
158 {
159
160 }
161 #endif
162
163 /* EOF */