[DISKPART] Fix the drive letter deletion code

The remove and assign commands work properly now.

base/system/diskpart/mountmgr.c

···
                   DeviceName,
                   DeviceNameLength);
 
-DPRINT1("\n");
     Status = OpenMountManager(&MountMgrHandle, GENERIC_READ | GENERIC_WRITE);
     if (!NT_SUCCESS(Status))
     {
···
         goto done;
     }
 
-DPRINT1("\n");
     Status = NtDeviceIoControlFile(MountMgrHandle,
                                    NULL,
                                    NULL,
···
         goto done;
     }
 
-DPRINT1("\n");
 done:
     if (MountMgrHandle)
         NtClose(MountMgrHandle);
···
     NTSTATUS Status;
     BOOL Ret = TRUE;
 
-    DPRINT1("AssignNextDriveLetter(%S %p)\n", DeviceName, DriveLetter);
+    DPRINT("AssignNextDriveLetter(%S %p)\n", DeviceName, DriveLetter);
 
     DeviceNameLength = wcslen(DeviceName) * sizeof(WCHAR);
 
···
                   DeviceName,
                   DeviceNameLength);
 
-DPRINT1("\n");
     Status = OpenMountManager(&MountMgrHandle, GENERIC_READ | GENERIC_WRITE);
     if (!NT_SUCCESS(Status))
     {
···
         goto done;
     }
 
-DPRINT1("\n");
     Status = NtDeviceIoControlFile(MountMgrHandle,
                                    NULL,
                                    NULL,
···
         goto done;
     }
 
-DPRINT1("\n");
 done:
     if (MountMgrHandle)
         NtClose(MountMgrHandle);
···
     PMOUNTMGR_MOUNT_POINTS OutputBuffer = NULL;
     WCHAR DosDeviceName[30];
     ULONG InputBufferLength, DosDeviceNameLength;
+    ULONG OutputBufferLength = 0x1000;
     HANDLE MountMgrHandle = NULL;
     IO_STATUS_BLOCK Iosb;
     NTSTATUS Status;
     BOOL Ret = TRUE;
 
-    DPRINT1("DeleteDriveLetter(%c)\n", DriveLetter);
+    DPRINT("DeleteDriveLetter(%c)\n", DriveLetter);
 
     /* Setup the device name of the letter to delete */
     swprintf(DosDeviceName, L"\\DosDevices\\%c:", DriveLetter);
···
     }
 
     /* Fill it in */
-//    RtlZeroMemory(InputBuffer, InputBufferLength);
     InputBuffer->SymbolicLinkNameOffset = sizeof(MOUNTMGR_MOUNT_POINT);
     InputBuffer->SymbolicLinkNameLength = DosDeviceNameLength;
     RtlCopyMemory(&InputBuffer[1], DosDeviceName, DosDeviceNameLength);
 
     /* Allocate big enough output buffer (we don't care about the output) */
-    OutputBuffer = RtlAllocateHeap(GetProcessHeap(), HEAP_ZERO_MEMORY, 0x1000);
+    OutputBuffer = RtlAllocateHeap(GetProcessHeap(), HEAP_ZERO_MEMORY, OutputBufferLength);
     if (OutputBuffer == NULL)
     {
         DPRINT1("OutputBuffer allocation failed!\n");
···
         goto done;
     }
 
-DPRINT1("\n");
+    OutputBuffer->Size = OutputBufferLength;
+
     Status = OpenMountManager(&MountMgrHandle, GENERIC_READ | GENERIC_WRITE);
     if (!NT_SUCCESS(Status))
     {
···
         goto done;
     }
 
-DPRINT1("\n");
     Status = NtDeviceIoControlFile(MountMgrHandle,
                                    NULL,
                                    NULL,
···
                                    InputBuffer,
                                    InputBufferLength,
                                    OutputBuffer,
-                                   0x1000);
+                                   OutputBufferLength);
     if (!NT_SUCCESS(Status))
     {
         DPRINT1("NtDeviceIoControlFile() failed (Status 0x%08lx)\n", Status);
···
         goto done;
     }
 
-DPRINT1("\n");
 done:
     if (MountMgrHandle)
         NtClose(MountMgrHandle);