Intravision/reactos
1
0
Fork 0
mirror of https://github.com/reactos/reactos.git synced 2025-05-06 18:31:26 +00:00
Code Issues Projects Releases Packages Wiki Activity

[VCDCONTROLTOOL] Implement support for persistent mounts

Browse source
This commit is contained in:
Pierre Schweitzer 2018-12-11 14:55:09 +01:00
parent 10ce523a2c
commit b518fc065d
No known key found for this signature in database
GPG key ID: 7545556C3D585B0B
1 changed files with 28 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

35
modules/rosapps/applications/vcdcontroltool/vcdcontroltool.c
View file

@ -14,6 +14,7 @@
#include <winuser.h> #include <winuser.h>
#include <wingdi.h> #include <wingdi.h>
#include <winsvc.h> #include <winsvc.h>
#include <winreg.h>
#include <commctrl.h> #include <commctrl.h>
#include <commdlg.h> #include <commdlg.h>
#include <wchar.h> #include <wchar.h>
@ -371,12 +372,6 @@ SetMountFileName(HWND hDlg,
hEditText = GetDlgItem(hMountWnd, IDC_MOUNTIMAGE); hEditText = GetDlgItem(hMountWnd, IDC_MOUNTIMAGE);
SendMessage(hEditText, WM_SETTEXT, 0, lParam); SendMessage(hEditText, WM_SETTEXT, 0, lParam);
/* FIXME: we don't support persistent mounts yet*/
{
hEditText = GetDlgItem(hMountWnd, IDC_MOUNTPERSIST);
EnableWindow(hEditText, FALSE);
}
/* Show our window */ /* Show our window */
ShowWindow(hDlg, SW_SHOW); ShowWindow(hDlg, SW_SHOW);
@ -400,6 +395,9 @@ PerformMount(VOID)
UNICODE_STRING NtPathName; UNICODE_STRING NtPathName;
HANDLE hLet; HANDLE hLet;
DWORD BytesRead; DWORD BytesRead;
BOOLEAN bPersist, Res;
WCHAR szKeyName[256];
HKEY hKey;
/* Zero our input structure */ /* Zero our input structure */
ZeroMemory(&MountParams, sizeof(MOUNT_PARAMETERS)); ZeroMemory(&MountParams, sizeof(MOUNT_PARAMETERS));
@ -418,6 +416,10 @@ PerformMount(VOID)
MountParams.Flags |= MOUNT_FLAG_SUPP_JOLIET; MountParams.Flags |= MOUNT_FLAG_SUPP_JOLIET;
} }
/* Should the mount be persistent? */
hControl = GetDlgItem(hMountWnd, IDC_MOUNTPERSIST);
bPersist = (SendMessage(hControl, BM_GETCHECK, 0, 0) == BST_CHECKED);
/* Get the file name */ /* Get the file name */
hControl = GetDlgItem(hMountWnd, IDC_MOUNTIMAGE); hControl = GetDlgItem(hMountWnd, IDC_MOUNTIMAGE);
GetWindowText(hControl, szFileName, sizeof(szFileName) / sizeof(WCHAR)); GetWindowText(hControl, szFileName, sizeof(szFileName) / sizeof(WCHAR));
@ -435,12 +437,31 @@ PerformMount(VOID)
if (hLet != INVALID_HANDLE_VALUE) if (hLet != INVALID_HANDLE_VALUE)
{ {
/* And issue the mount IOCTL */ /* And issue the mount IOCTL */
DeviceIoControl(hLet, IOCTL_VCDROM_MOUNT_IMAGE, &MountParams, sizeof(MountParams), NULL, 0, &BytesRead, NULL); Res = DeviceIoControl(hLet, IOCTL_VCDROM_MOUNT_IMAGE, &MountParams, sizeof(MountParams), NULL, 0, &BytesRead, NULL);
CloseHandle(hLet); CloseHandle(hLet);
/* Refresh the list so that our mount appears */ /* Refresh the list so that our mount appears */
RefreshDevicesList(0); RefreshDevicesList(0);
/* If mount succeed and has to persistent, write it to registry */
if (Res && bPersist)
{
wsprintf(szKeyName, L"SYSTEM\\CurrentControlSet\\Services\\Vcdrom\\Parameters\\Device%c", wMountLetter);
if (RegCreateKeyEx(HKEY_LOCAL_MACHINE, szKeyName, 0, NULL, REG_OPTION_NON_VOLATILE, KEY_CREATE_SUB_KEY | KEY_SET_VALUE, NULL, &hKey, NULL) == ERROR_SUCCESS)
{
wcsncpy(szKeyName, MountParams.Path, MountParams.Length);
szKeyName[MountParams.Length / sizeof(WCHAR)] = 0;
RegSetValueExW(hKey, L"IMAGE", 0, REG_SZ, (BYTE *)szKeyName, MountParams.Length);
szKeyName[0] = wMountLetter;
szKeyName[1] = L':';
szKeyName[2] = 0;
RegSetValueExW(hKey, L"DRIVE", 0, REG_SZ, (BYTE *)szKeyName, 3 * sizeof(WCHAR));
RegCloseKey(hKey);
}
}
} }
} }