Public/winfsp
1
0
Fork 0
mirror of https://github.com/winfsp/winfsp.git synced 2025-04-22 08:23:05 -05:00
Code Issues Packages Projects Releases Wiki Activity

dll: FspMountBroadcastDriveChange

Browse Source 
Broadcast WM_DEVICECHANGE asynchronously.
This commit is contained in:
Bill Zissimopoulos 2022-06-07 13:11:23 +01:00
parent 8e45f7d795
commit 0a919d317a
1 changed files with 64 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

67
src/dll/mount.c
View File

@ -385,7 +385,14 @@ static NTSTATUS FspLauncherDefineDosDevice(
return !NT_SUCCESS(Result) ? Result : FspNtStatusFromWin32(ErrorCode); return !NT_SUCCESS(Result) ? Result : FspNtStatusFromWin32(ErrorCode);
} }
static VOID FspMountBroadcastDriveChange(PWSTR MountPoint, WPARAM WParam) struct FspMountBroadcastDriveChangeData
{
HMODULE Module;
WPARAM WParam;
WCHAR MountPoint[];
};
static DWORD WINAPI FspMountBroadcastDriveChangeThread(PVOID Data0)
{ {
/* /*
* DefineDosDeviceW (either directly or via the CSRSS) broadcasts a WM_DEVICECHANGE message * DefineDosDeviceW (either directly or via the CSRSS) broadcasts a WM_DEVICECHANGE message
@ -405,6 +412,7 @@ static VOID FspMountBroadcastDriveChange(PWSTR MountPoint, WPARAM WParam)
* faster). * faster).
*/ */
struct FspMountBroadcastDriveChangeData *Data = Data0;
BOOLEAN IsLocalSystem; BOOLEAN IsLocalSystem;
DEV_BROADCAST_VOLUME DriveChange; DEV_BROADCAST_VOLUME DriveChange;
DWORD Recipients; DWORD Recipients;
@ -415,15 +423,68 @@ static VOID FspMountBroadcastDriveChange(PWSTR MountPoint, WPARAM WParam)
DriveChange.dbcv_size = sizeof DriveChange; DriveChange.dbcv_size = sizeof DriveChange;
DriveChange.dbcv_devicetype = DBT_DEVTYP_VOLUME; DriveChange.dbcv_devicetype = DBT_DEVTYP_VOLUME;
DriveChange.dbcv_flags = DBTF_NET; DriveChange.dbcv_flags = DBTF_NET;
DriveChange.dbcv_unitmask = 1 << ((MountPoint[0] | 0x20) - 'a'); DriveChange.dbcv_unitmask = 1 << ((Data->MountPoint[0] | 0x20) - 'a');
Recipients = BSM_APPLICATIONS | (IsLocalSystem ? BSM_ALLDESKTOPS : 0); Recipients = BSM_APPLICATIONS | (IsLocalSystem ? BSM_ALLDESKTOPS : 0);
BroadcastSystemMessageW( BroadcastSystemMessageW(
BSF_POSTMESSAGE, BSF_POSTMESSAGE,
&Recipients, &Recipients,
WM_DEVICECHANGE, WM_DEVICECHANGE,
WParam, Data->WParam,
(LPARAM)&DriveChange); (LPARAM)&DriveChange);
FreeLibraryAndExitThread(Data->Module, 0);
return 0;
}
static NTSTATUS FspMountBroadcastDriveChange(PWSTR MountPoint, WPARAM WParam)
{
NTSTATUS Result;
HMODULE Module;
HANDLE Thread;
struct FspMountBroadcastDriveChangeData *Data = 0;
int Size;
if (!GetModuleHandleExW(
GET_MODULE_HANDLE_EX_FLAG_FROM_ADDRESS,
(PVOID)FspMountBroadcastDriveChangeThread,
&Module))
{
Result = FspNtStatusFromWin32(GetLastError());
goto exit;
}
Size = (lstrlenW(MountPoint) + 1) * sizeof(WCHAR);
Data = MemAlloc(sizeof *Data + Size);
if (0 == Data)
{
Result = STATUS_INSUFFICIENT_RESOURCES;
goto exit;
}
Data->Module = Module;
Data->WParam = WParam;
memcpy(Data->MountPoint, MountPoint, Size);
Thread = CreateThread(0, 0, FspMountBroadcastDriveChangeThread, Data, 0, 0);
if (0 == Thread)
{
Result = FspNtStatusFromWin32(GetLastError());
goto exit;
}
CloseHandle(Thread);
Result = STATUS_SUCCESS;
exit:
if (!NT_SUCCESS(Result))
{
MemFree(Data);
FreeLibrary(Module);
}
return Result;
} }
static NTSTATUS FspMountSet_Drive(PWSTR VolumeName, PWSTR MountPoint, PHANDLE PMountHandle) static NTSTATUS FspMountSet_Drive(PWSTR VolumeName, PWSTR MountPoint, PHANDLE PMountHandle)