ReactOS 0.4.16-dev-2122-g1628f5e
Macros | Functions
mountmgr.c File Reference
#include "diskpart.h"
#include <mountmgr.h>
#include <debug.h>
Include dependency graph for mountmgr.c:

Go to the source code of this file.

Macros

#define NDEBUG
 

Functions

static NTSTATUS OpenMountManager (_Out_ PHANDLE MountMgrHandle, _In_ ACCESS_MASK Access)
 
BOOL GetAutomountState (_Out_ PBOOL State)
 
BOOL SetAutomountState (_In_ BOOL bEnable)
 
BOOL ScrubAutomount (VOID)
 
BOOL AssignDriveLetter (_In_ PWSTR DeviceName, _In_ WCHAR DriveLetter)
 
BOOL AssignNextDriveLetter (_In_ PWSTR DeviceName, _Out_ PWCHAR DriveLetter)
 
BOOL DeleteDriveLetter (_In_ WCHAR DriveLetter)
 

Macro Definition Documentation

◆ NDEBUG

#define NDEBUG

Definition at line 13 of file mountmgr.c.

Function Documentation

◆ AssignDriveLetter()

BOOL AssignDriveLetter ( _In_ PWSTR  DeviceName,
_In_ WCHAR  DriveLetter 
)

Definition at line 181 of file mountmgr.c.

184{
185 WCHAR DosDeviceName[30];
186 ULONG DosDeviceNameLength, DeviceNameLength;
187 ULONG InputBufferLength;
188 PMOUNTMGR_CREATE_POINT_INPUT InputBuffer;
189 HANDLE MountMgrHandle = NULL;
190 IO_STATUS_BLOCK Iosb;
191 NTSTATUS Status;
192 BOOL Ret = TRUE;
193
194 DPRINT1("AssignDriveLetter(%S %c)\n", DeviceName, DriveLetter);
195
196 DeviceNameLength = wcslen(DeviceName) * sizeof(WCHAR);
197
198 swprintf(DosDeviceName, L"\\DosDevices\\%c:", DriveLetter);
199 DosDeviceNameLength = wcslen(DosDeviceName) * sizeof(WCHAR);
200
201 /* Allocate the input buffer for the MountMgr */
202 InputBufferLength = DosDeviceNameLength + DeviceNameLength + sizeof(MOUNTMGR_CREATE_POINT_INPUT);
203 InputBuffer = RtlAllocateHeap(RtlGetProcessHeap(), HEAP_ZERO_MEMORY, InputBufferLength);
204 if (InputBuffer == NULL)
205 {
206 DPRINT1("InputBuffer allocation failed!\n");
207 return FALSE;
208 }
209
210 /* Fill the input buffer */
211 InputBuffer->SymbolicLinkNameOffset = sizeof(MOUNTMGR_CREATE_POINT_INPUT);
212 InputBuffer->SymbolicLinkNameLength = DosDeviceNameLength;
213 InputBuffer->DeviceNameOffset = DosDeviceNameLength + sizeof(MOUNTMGR_CREATE_POINT_INPUT);
214 InputBuffer->DeviceNameLength = DeviceNameLength;
215 RtlCopyMemory((PVOID)((ULONG_PTR)InputBuffer + sizeof(MOUNTMGR_CREATE_POINT_INPUT)),
216 DosDeviceName,
217 DosDeviceNameLength);
218 RtlCopyMemory((PVOID)((ULONG_PTR)InputBuffer + InputBuffer->DeviceNameOffset),
219 DeviceName,
220 DeviceNameLength);
221
222DPRINT1("\n");
223 Status = OpenMountManager(&MountMgrHandle, GENERIC_READ | GENERIC_WRITE);
224 if (!NT_SUCCESS(Status))
225 {
226 DPRINT1("OpenMountManager() failed (Status 0x%08lx)\n", Status);
227 Ret = FALSE;
228 goto done;
229 }
230
231DPRINT1("\n");
232 Status = NtDeviceIoControlFile(MountMgrHandle,
233 NULL,
234 NULL,
235 NULL,
236 &Iosb,
237 IOCTL_MOUNTMGR_CREATE_POINT,
238 InputBuffer,
239 InputBufferLength,
240 NULL,
241 0);
242 if (!NT_SUCCESS(Status))
243 {
244 DPRINT1("NtDeviceIoControlFile() failed (Status 0x%08lx)\n", Status);
245 Ret = FALSE;
246 goto done;
247 }
248
249DPRINT1("\n");
250done:
251 if (MountMgrHandle)
252 NtClose(MountMgrHandle);
253
254 RtlFreeHeap(RtlGetProcessHeap(), 0, InputBuffer);
255
256 return Ret;
257}
NTSTATUS
LONG NTSTATUS
Definition: precomp.h:26
DPRINT1
#define DPRINT1
Definition: precomp.h:8
OpenMountManager
static NTSTATUS OpenMountManager(_Out_ PHANDLE MountMgrHandle, _In_ ACCESS_MASK Access)
Definition: mountmgr.c:20
RtlAllocateHeap
PVOID NTAPI RtlAllocateHeap(IN PVOID HeapHandle, IN ULONG Flags, IN SIZE_T Size)
Definition: heap.c:616
RtlFreeHeap
BOOLEAN NTAPI RtlFreeHeap(IN PVOID HeapHandle, IN ULONG Flags, IN PVOID HeapBase)
Definition: heap.c:634
NULL
#define NULL
Definition: types.h:112
TRUE
#define TRUE
Definition: types.h:120
FALSE
#define FALSE
Definition: types.h:117
NT_SUCCESS
#define NT_SUCCESS(StatCode)
Definition: apphelp.c:33
GENERIC_READ
#define GENERIC_READ
Definition: compat.h:135
HEAP_ZERO_MEMORY
#define HEAP_ZERO_MEMORY
Definition: compat.h:134
wcslen
_ACRTIMP size_t __cdecl wcslen(const wchar_t *)
Definition: wcs.c:2983
swprintf
#define swprintf
Definition: precomp.h:40
L
#define L(x)
Definition: resources.c:13
Iosb
return Iosb
Definition: create.c:4403
BOOL
unsigned int BOOL
Definition: ntddk_ex.h:94
Status
Status
Definition: gdiplustypes.h:25
DosDeviceName
static char DosDeviceName[DEVICE_SIZE]
Definition: lsdd.c:26
MOUNTMGR_CREATE_POINT_INPUT
struct _MOUNTMGR_CREATE_POINT_INPUT MOUNTMGR_CREATE_POINT_INPUT
IOCTL_MOUNTMGR_CREATE_POINT
#define IOCTL_MOUNTMGR_CREATE_POINT
Definition: imports.h:116
NtDeviceIoControlFile
NTSYSAPI NTSTATUS NTAPI NtDeviceIoControlFile(IN HANDLE hFile, IN HANDLE hEvent OPTIONAL, IN PIO_APC_ROUTINE IoApcRoutine OPTIONAL, IN PVOID IoApcContext OPTIONAL, OUT PIO_STATUS_BLOCK pIoStatusBlock, IN ULONG DeviceIoControlCode, IN PVOID InBuffer OPTIONAL, IN ULONG InBufferLength, OUT PVOID OutBuffer OPTIONAL, IN ULONG OutBufferLength)
NtClose
NTSTATUS NTAPI NtClose(IN HANDLE Handle)
Definition: obhandle.c:3402
GENERIC_WRITE
#define GENERIC_WRITE
Definition: nt_native.h:90
_IO_STATUS_BLOCK
Definition: env_spec_w32.h:870
_MOUNTMGR_CREATE_POINT_INPUT
Definition: imports.h:156
RtlCopyMemory
#define RtlCopyMemory(Destination, Source, Length)
Definition: typedefs.h:263
ULONG_PTR
uint32_t ULONG_PTR
Definition: typedefs.h:65
ULONG
uint32_t ULONG
Definition: typedefs.h:59
DeviceName
_Must_inspect_result_ _In_ PWDFDEVICE_INIT _In_opt_ PCUNICODE_STRING DeviceName
Definition: wdfdevice.h:3281
InputBufferLength
_In_ WDFREQUEST _In_ size_t _In_ size_t InputBufferLength
Definition: wdfio.h:322
InputBuffer
_Must_inspect_result_ _In_ WDFIOTARGET _In_opt_ WDFREQUEST _In_opt_ PWDF_MEMORY_DESCRIPTOR InputBuffer
Definition: wdfiotarget.h:953
WCHAR
__wchar_t WCHAR
Definition: xmlstorage.h:180

Referenced by assign_main().

◆ AssignNextDriveLetter()

BOOL AssignNextDriveLetter ( _In_ PWSTR  DeviceName,
_Out_ PWCHAR  DriveLetter 
)

Definition at line 261 of file mountmgr.c.

264{
265 MOUNTMGR_DRIVE_LETTER_INFORMATION LetterInfo;
266 PMOUNTMGR_DRIVE_LETTER_TARGET InputBuffer;
267 ULONG DeviceNameLength, InputBufferLength;
268 HANDLE MountMgrHandle = NULL;
269 IO_STATUS_BLOCK Iosb;
270 NTSTATUS Status;
271 BOOL Ret = TRUE;
272
273 DPRINT1("AssignNextDriveLetter(%S %p)\n", DeviceName, DriveLetter);
274
275 DeviceNameLength = wcslen(DeviceName) * sizeof(WCHAR);
276
277 InputBufferLength = DeviceNameLength + FIELD_OFFSET(MOUNTMGR_DRIVE_LETTER_TARGET, DeviceName);
278 InputBuffer = RtlAllocateHeap(RtlGetProcessHeap(), HEAP_ZERO_MEMORY, InputBufferLength);
279 if (InputBuffer == NULL)
280 {
281 DPRINT1("InputBuffer allocation failed!\n");
282 return FALSE;
283 }
284
285 /* And fill it with the device hat needs a drive letter */
286 InputBuffer->DeviceNameLength = DeviceNameLength;
287 RtlCopyMemory(&InputBuffer->DeviceName[0],
288 DeviceName,
289 DeviceNameLength);
290
291DPRINT1("\n");
292 Status = OpenMountManager(&MountMgrHandle, GENERIC_READ | GENERIC_WRITE);
293 if (!NT_SUCCESS(Status))
294 {
295 DPRINT1("OpenMountManager() failed (Status 0x%08lx)\n", Status);
296 Ret = FALSE;
297 goto done;
298 }
299
300DPRINT1("\n");
301 Status = NtDeviceIoControlFile(MountMgrHandle,
302 NULL,
303 NULL,
304 NULL,
305 &Iosb,
306 IOCTL_MOUNTMGR_CREATE_POINT,
307 InputBuffer,
308 InputBufferLength,
309 &LetterInfo,
310 sizeof(LetterInfo));
311 if (!NT_SUCCESS(Status))
312 {
313 DPRINT1("NtDeviceIoControlFile() failed (Status 0x%08lx)\n", Status);
314 Ret = FALSE;
315 goto done;
316 }
317
318DPRINT1("\n");
319done:
320 if (MountMgrHandle)
321 NtClose(MountMgrHandle);
322
323 RtlFreeHeap(RtlGetProcessHeap(), 0, InputBuffer);
324
325 if (Ret)
326 {
327 if (LetterInfo.DriveLetterWasAssigned)
328 *DriveLetter = LetterInfo.CurrentDriveLetter;
329 else
330 *DriveLetter = UNICODE_NULL;
331 }
332
333 return Ret;
334}
UNICODE_NULL
#define UNICODE_NULL
_MOUNTMGR_DRIVE_LETTER_INFORMATION
Definition: mountmgr.h:117
_MOUNTMGR_DRIVE_LETTER_INFORMATION::DriveLetterWasAssigned
BOOLEAN DriveLetterWasAssigned
Definition: mountmgr.h:118
_MOUNTMGR_DRIVE_LETTER_INFORMATION::CurrentDriveLetter
UCHAR CurrentDriveLetter
Definition: mountmgr.h:119
_MOUNTMGR_DRIVE_LETTER_TARGET
Definition: mountmgr.h:112
FIELD_OFFSET
#define FIELD_OFFSET(t, f)
Definition: typedefs.h:255

Referenced by assign_main().

◆ DeleteDriveLetter()

BOOL DeleteDriveLetter ( _In_ WCHAR  DriveLetter)

Definition at line 338 of file mountmgr.c.

340{
341 PMOUNTMGR_MOUNT_POINT InputBuffer = NULL;
342 PMOUNTMGR_MOUNT_POINTS OutputBuffer = NULL;
343 WCHAR DosDeviceName[30];
344 ULONG InputBufferLength, DosDeviceNameLength;
345 HANDLE MountMgrHandle = NULL;
346 IO_STATUS_BLOCK Iosb;
347 NTSTATUS Status;
348 BOOL Ret = TRUE;
349
350 DPRINT1("DeleteDriveLetter(%c)\n", DriveLetter);
351
352 /* Setup the device name of the letter to delete */
353 swprintf(DosDeviceName, L"\\DosDevices\\%c:", DriveLetter);
354 DosDeviceNameLength = wcslen(DosDeviceName) * sizeof(WCHAR);
355
356 /* Allocate the input buffer for MountMgr */
357 InputBufferLength = DosDeviceNameLength + sizeof(MOUNTMGR_MOUNT_POINT);
358 InputBuffer = RtlAllocateHeap(RtlGetProcessHeap(), HEAP_ZERO_MEMORY, InputBufferLength);
359 if (InputBuffer == NULL)
360 {
361 DPRINT1("InputBuffer allocation failed!\n");
362 return FALSE;
363 }
364
365 /* Fill it in */
366// RtlZeroMemory(InputBuffer, InputBufferLength);
367 InputBuffer->SymbolicLinkNameOffset = sizeof(MOUNTMGR_MOUNT_POINT);
368 InputBuffer->SymbolicLinkNameLength = DosDeviceNameLength;
369 RtlCopyMemory(&InputBuffer[1], DosDeviceName, DosDeviceNameLength);
370
371 /* Allocate big enough output buffer (we don't care about the output) */
372 OutputBuffer = RtlAllocateHeap(GetProcessHeap(), HEAP_ZERO_MEMORY, 0x1000);
373 if (OutputBuffer == NULL)
374 {
375 DPRINT1("OutputBuffer allocation failed!\n");
376 Ret = FALSE;
377 goto done;
378 }
379
380DPRINT1("\n");
381 Status = OpenMountManager(&MountMgrHandle, GENERIC_READ | GENERIC_WRITE);
382 if (!NT_SUCCESS(Status))
383 {
384 DPRINT1("OpenMountManager() failed (Status 0x%08lx)\n", Status);
385 Ret = FALSE;
386 goto done;
387 }
388
389DPRINT1("\n");
390 Status = NtDeviceIoControlFile(MountMgrHandle,
391 NULL,
392 NULL,
393 NULL,
394 &Iosb,
395 IOCTL_MOUNTMGR_DELETE_POINTS,
396 InputBuffer,
397 InputBufferLength,
398 OutputBuffer,
399 0x1000);
400 if (!NT_SUCCESS(Status))
401 {
402 DPRINT1("NtDeviceIoControlFile() failed (Status 0x%08lx)\n", Status);
403 Ret = FALSE;
404 goto done;
405 }
406
407DPRINT1("\n");
408done:
409 if (MountMgrHandle)
410 NtClose(MountMgrHandle);
411
412 if (InputBuffer)
413 RtlFreeHeap(RtlGetProcessHeap(), 0, InputBuffer);
414
415 if (OutputBuffer)
416 RtlFreeHeap(RtlGetProcessHeap(), 0, OutputBuffer);
417
418 return Ret;
419}
GetProcessHeap
#define GetProcessHeap()
Definition: compat.h:736
IOCTL_MOUNTMGR_DELETE_POINTS
#define IOCTL_MOUNTMGR_DELETE_POINTS
Definition: imports.h:122
MOUNTMGR_MOUNT_POINT
struct _MOUNTMGR_MOUNT_POINT MOUNTMGR_MOUNT_POINT
_MOUNTMGR_MOUNT_POINTS
Definition: imports.h:172
_MOUNTMGR_MOUNT_POINT
Definition: imports.h:163
OutputBuffer
_Must_inspect_result_ _In_ WDFIOTARGET _In_opt_ WDFREQUEST _In_opt_ PWDF_MEMORY_DESCRIPTOR OutputBuffer
Definition: wdfiotarget.h:863

Referenced by assign_main(), and remove_main().

◆ GetAutomountState()

BOOL GetAutomountState ( _Out_ PBOOL  State)

Definition at line 46 of file mountmgr.c.

48{
49 HANDLE MountMgrHandle;
50 MOUNTMGR_QUERY_AUTO_MOUNT AutoMount;
51 IO_STATUS_BLOCK Iosb;
52 NTSTATUS Status;
53
54 DPRINT("ShowAutomountState()\n");
55
56 Status = OpenMountManager(&MountMgrHandle, GENERIC_READ);
57 if (!NT_SUCCESS(Status))
58 {
59 DPRINT1("OpenMountManager() Status 0x%08lx\n", Status);
60 return FALSE;
61 }
62
63 Status = NtDeviceIoControlFile(MountMgrHandle,
64 NULL,
65 NULL,
66 NULL,
67 &Iosb,
68 IOCTL_MOUNTMGR_QUERY_AUTO_MOUNT,
69 NULL,
70 0,
71 &AutoMount,
72 sizeof(AutoMount));
73
74 NtClose(MountMgrHandle);
75
76 if (!NT_SUCCESS(Status))
77 {
78 DPRINT1("NtDeviceIoControlFile() Status 0x%08lx\n", Status);
79 return FALSE;
80 }
81
82 if (State)
83 *State = (AutoMount.CurrentState == Enabled);
84
85 return TRUE;
86}
Enabled
@ Enabled
Definition: mountmgr.h:179
IOCTL_MOUNTMGR_QUERY_AUTO_MOUNT
#define IOCTL_MOUNTMGR_QUERY_AUTO_MOUNT
Definition: mountmgr.h:172
DPRINT
#define DPRINT
Definition: sndvol32.h:73
_MOUNTMGR_QUERY_AUTO_MOUNT
Definition: mountmgr.h:182
_MOUNTMGR_QUERY_AUTO_MOUNT::CurrentState
MOUNTMGR_AUTO_MOUNT_STATE CurrentState
Definition: mountmgr.h:183

Referenced by automount_main().

◆ OpenMountManager()

static NTSTATUS OpenMountManager ( _Out_ PHANDLE  MountMgrHandle,
_In_ ACCESS_MASK  Access 
)
static

Definition at line 20 of file mountmgr.c.

23{
24 OBJECT_ATTRIBUTES ObjectAttributes;
25 UNICODE_STRING DeviceName;
26 IO_STATUS_BLOCK Iosb;
27
28 RtlInitUnicodeString(&DeviceName, MOUNTMGR_DEVICE_NAME);
29
30 InitializeObjectAttributes(&ObjectAttributes,
31 &DeviceName,
32 0,
33 NULL,
34 NULL);
35
36 return NtOpenFile(MountMgrHandle,
37 Access | SYNCHRONIZE,
38 &ObjectAttributes,
39 &Iosb,
40 0,
41 FILE_SYNCHRONOUS_IO_NONALERT);
42}
ObjectAttributes
IN PUNICODE_STRING IN POBJECT_ATTRIBUTES ObjectAttributes
Definition: conport.c:36
FILE_SYNCHRONOUS_IO_NONALERT
#define FILE_SYNCHRONOUS_IO_NONALERT
Definition: from_kernel.h:31
MOUNTMGR_DEVICE_NAME
#define MOUNTMGR_DEVICE_NAME
Definition: imports.h:74
InitializeObjectAttributes
#define InitializeObjectAttributes(p, n, a, r, s)
Definition: reg.c:106
NtOpenFile
NTSYSAPI NTSTATUS NTAPI NtOpenFile(OUT PHANDLE phFile, IN ACCESS_MASK DesiredAccess, IN POBJECT_ATTRIBUTES ObjectAttributes, OUT PIO_STATUS_BLOCK pIoStatusBlock, IN ULONG ShareMode, IN ULONG OpenMode)
Definition: file.c:3953
SYNCHRONIZE
#define SYNCHRONIZE
Definition: nt_native.h:61
RtlInitUnicodeString
NTSYSAPI VOID NTAPI RtlInitUnicodeString(PUNICODE_STRING DestinationString, PCWSTR SourceString)
_OBJECT_ATTRIBUTES
Definition: umtypes.h:184
_UNICODE_STRING
Definition: env_spec_w32.h:368

Referenced by AssignDriveLetter(), AssignNextDriveLetter(), DeleteDriveLetter(), GetAutomountState(), ScrubAutomount(), and SetAutomountState().

◆ ScrubAutomount()

BOOL ScrubAutomount ( VOID  )

Definition at line 141 of file mountmgr.c.

142{
143 HANDLE MountMgrHandle;
144 IO_STATUS_BLOCK Iosb;
145 NTSTATUS Status;
146
147 DPRINT("ScrubAutomount()\n");
148
149 Status = OpenMountManager(&MountMgrHandle, GENERIC_READ | GENERIC_WRITE);
150 if (!NT_SUCCESS(Status))
151 {
152 DPRINT1("OpenMountManager() Status 0x%08lx\n", Status);
153 return FALSE;
154 }
155
156 Status = NtDeviceIoControlFile(MountMgrHandle,
157 NULL,
158 NULL,
159 NULL,
160 &Iosb,
161 IOCTL_MOUNTMGR_SCRUB_REGISTRY,
162 NULL,
163 0,
164 NULL,
165 0);
166
167 NtClose(MountMgrHandle);
168
169 if (!NT_SUCCESS(Status))
170 {
171 DPRINT1("NtDeviceIoControlFile() Status 0x%08lx\n", Status);
172 return FALSE;
173 }
174
175 return TRUE;
176}
IOCTL_MOUNTMGR_SCRUB_REGISTRY
#define IOCTL_MOUNTMGR_SCRUB_REGISTRY
Definition: mountmgr.h:170

Referenced by automount_main().

◆ SetAutomountState()

BOOL SetAutomountState ( _In_ BOOL  bEnable)

Definition at line 90 of file mountmgr.c.

92{
93 HANDLE MountMgrHandle;
94 MOUNTMGR_SET_AUTO_MOUNT AutoMount;
95 IO_STATUS_BLOCK Iosb;
96 NTSTATUS Status;
97
98 DPRINT("SetAutomountState()\n");
99
100 Status = OpenMountManager(&MountMgrHandle, GENERIC_READ | GENERIC_WRITE);
101 if (!NT_SUCCESS(Status))
102 {
103 DPRINT1("OpenMountManager() Status 0x%08lx\n", Status);
104 return TRUE;
105 }
106
107 if (bEnable)
108 AutoMount.NewState = Enabled;
109 else
110 AutoMount.NewState = Disabled;
111
112 Status = NtDeviceIoControlFile(MountMgrHandle,
113 NULL,
114 NULL,
115 NULL,
116 &Iosb,
117 IOCTL_MOUNTMGR_SET_AUTO_MOUNT,
118 &AutoMount,
119 sizeof(AutoMount),
120 NULL,
121 0);
122
123 NtClose(MountMgrHandle);
124
125 if (!NT_SUCCESS(Status))
126 {
127 DPRINT1("NtDeviceIoControlFile() Status 0x%08lx\n", Status);
128 return TRUE;
129 }
130
131 if (AutoMount.NewState == Enabled)
132 ConResPuts(StdOut, IDS_AUTOMOUNT_ENABLED);
133 else
134 ConResPuts(StdOut, IDS_AUTOMOUNT_DISABLED);
135 ConPuts(StdOut, L"\n");
136
137 return TRUE;
138}
ConPuts
void ConPuts(FILE *fp, LPCWSTR psz)
Definition: fc.c:16
StdOut
#define StdOut
Definition: fc.c:14
ConResPuts
void ConResPuts(FILE *fp, UINT nID)
Definition: fc.c:27
IDS_AUTOMOUNT_DISABLED
#define IDS_AUTOMOUNT_DISABLED
Definition: resource.h:27
IDS_AUTOMOUNT_ENABLED
#define IDS_AUTOMOUNT_ENABLED
Definition: resource.h:26
Disabled
@ Disabled
Definition: mountmgr.h:178
IOCTL_MOUNTMGR_SET_AUTO_MOUNT
#define IOCTL_MOUNTMGR_SET_AUTO_MOUNT
Definition: mountmgr.h:174
_MOUNTMGR_SET_AUTO_MOUNT
Definition: mountmgr.h:186
_MOUNTMGR_SET_AUTO_MOUNT::NewState
MOUNTMGR_AUTO_MOUNT_STATE NewState
Definition: mountmgr.h:187
bEnable
_In_ BOOL bEnable
Definition: winddi.h:3426

Referenced by automount_main().