[PATCH v2 2/3] wineusd.sys: Add new service to update user shared data continuously.
Rémi Bernon
rbernon at codeweavers.com
Mon Dec 2 07:32:17 CST 2019
This service is started at boot, and hopefully before any process that
might rely on the timestamp update. If the service is not yet started,
then the opening the device will fail and the process should continue
and use its private user shared data instead.
Signed-off-by: Rémi Bernon <rbernon at codeweavers.com>
---
configure.ac | 1 +
dlls/wineusd.sys/Makefile.in | 6 +
dlls/wineusd.sys/wineusd.sys.spec | 1 +
dlls/wineusd.sys/wineusd_main.c | 274 ++++++++++++++++++++++++++++++
include/wine/usd.h | 27 +++
loader/wine.inf.in | 12 ++
6 files changed, 321 insertions(+)
create mode 100644 dlls/wineusd.sys/Makefile.in
create mode 100644 dlls/wineusd.sys/wineusd.sys.spec
create mode 100644 dlls/wineusd.sys/wineusd_main.c
create mode 100644 include/wine/usd.h
diff --git a/configure.ac b/configure.ac
index 64721e96449..b1463d69823 100644
--- a/configure.ac
+++ b/configure.ac
@@ -3793,6 +3793,7 @@ WINE_CONFIG_MAKEFILE(dlls/wineps.drv)
WINE_CONFIG_MAKEFILE(dlls/wineps16.drv16,enable_win16)
WINE_CONFIG_MAKEFILE(dlls/winepulse.drv)
WINE_CONFIG_MAKEFILE(dlls/wineqtdecoder)
+WINE_CONFIG_MAKEFILE(dlls/wineusd.sys)
WINE_CONFIG_MAKEFILE(dlls/winevulkan)
WINE_CONFIG_MAKEFILE(dlls/winex11.drv)
WINE_CONFIG_MAKEFILE(dlls/wing.dll16,enable_win16)
diff --git a/dlls/wineusd.sys/Makefile.in b/dlls/wineusd.sys/Makefile.in
new file mode 100644
index 00000000000..0792860c09b
--- /dev/null
+++ b/dlls/wineusd.sys/Makefile.in
@@ -0,0 +1,6 @@
+MODULE = wineusd.sys
+IMPORTS = ntoskrnl
+EXTRADLLFLAGS = -Wl,--subsystem,native -mno-cygwin
+
+C_SRCS = \
+ wineusd_main.c
diff --git a/dlls/wineusd.sys/wineusd.sys.spec b/dlls/wineusd.sys/wineusd.sys.spec
new file mode 100644
index 00000000000..76421d7e35b
--- /dev/null
+++ b/dlls/wineusd.sys/wineusd.sys.spec
@@ -0,0 +1 @@
+# nothing to export
diff --git a/dlls/wineusd.sys/wineusd_main.c b/dlls/wineusd.sys/wineusd_main.c
new file mode 100644
index 00000000000..f4f1132e24c
--- /dev/null
+++ b/dlls/wineusd.sys/wineusd_main.c
@@ -0,0 +1,274 @@
+/*
+ * User shared data update service
+ *
+ * Copyright 2019 Rémi Bernon for CodeWeavers
+ *
+ * This library is free software; you can redistribute it and/or
+ * modify it under the terms of the GNU Lesser General Public
+ * License as published by the Free Software Foundation; either
+ * version 2.1 of the License, or (at your option) any later version.
+ *
+ * This library is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
+ * Lesser General Public License for more details.
+ *
+ * You should have received a copy of the GNU Lesser General Public
+ * License along with this library; if not, write to the Free Software
+ * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
+ */
+
+#include <assert.h>
+#include <stdarg.h>
+#include "ntstatus.h"
+#define WIN32_NO_STATUS
+#include "winternl.h"
+#include "winioctl.h"
+#include "ddk/wdm.h"
+#include "wine/debug.h"
+#include "wine/heap.h"
+#include "wine/usd.h"
+#include "wine/list.h"
+
+static DEVICE_OBJECT *device_obj;
+
+WINE_DEFAULT_DEBUG_CHANNEL(wineusd);
+
+#define DECLARE_CRITICAL_SECTION(cs) \
+ static CRITICAL_SECTION cs; \
+ static CRITICAL_SECTION_DEBUG cs##_debug = \
+ { 0, 0, &cs, { &cs##_debug.ProcessLocksList, &cs##_debug.ProcessLocksList }, \
+ 0, 0, { (DWORD_PTR)(__FILE__ ": " # cs) }}; \
+ static CRITICAL_SECTION cs = { &cs##_debug, -1, 0, 0, 0, 0 };
+
+DECLARE_CRITICAL_SECTION(wineusd_cs);
+
+static struct list wineusd_entries = LIST_INIT(wineusd_entries);
+static HANDLE wineusd_thread, wineusd_thread_stop;
+
+struct wineusd_entry
+{
+ struct list link;
+ CLIENT_ID cid;
+ HANDLE process;
+ HANDLE section;
+ void *page;
+};
+
+static DWORD WINAPI wineusd_thread_proc(void *arg)
+{
+ struct wineusd_entry *entry;
+ ULARGE_INTEGER interrupt;
+ ULARGE_INTEGER tick;
+ LARGE_INTEGER now;
+
+ TRACE("Started user shared data thread.\n");
+
+ while (WaitForSingleObject(wineusd_thread_stop, 16) == WAIT_TIMEOUT)
+ {
+ EnterCriticalSection(&wineusd_cs);
+
+ NtQuerySystemTime(&now);
+ RtlQueryUnbiasedInterruptTime(&interrupt.QuadPart);
+
+ tick = interrupt;
+ tick.QuadPart /= 10000;
+
+ LIST_FOR_EACH_ENTRY(entry, &wineusd_entries, struct wineusd_entry, link)
+ {
+ KSHARED_USER_DATA *usd = entry->page;
+
+ usd->SystemTime.High2Time = now.u.HighPart;
+ usd->SystemTime.LowPart = now.u.LowPart;
+ usd->SystemTime.High1Time = now.u.HighPart;
+
+ usd->InterruptTime.High2Time = interrupt.HighPart;
+ usd->InterruptTime.LowPart = interrupt.LowPart;
+ usd->InterruptTime.High1Time = interrupt.HighPart;
+
+ usd->TickCount.High2Time = tick.HighPart;
+ usd->TickCount.LowPart = tick.LowPart;
+ usd->TickCount.High1Time = tick.HighPart;
+ usd->TickCountLowDeprecated = tick.LowPart;
+ usd->TickCountMultiplier = 1 << 24;
+ }
+
+ LeaveCriticalSection(&wineusd_cs);
+ }
+
+ TRACE("Stopped user shared data thread.\n");
+
+ return 0;
+}
+
+static NTSTATUS wineusd_initialize(struct wineusd_entry *entry, IRP *irp)
+{
+ const KSHARED_USER_DATA *init = irp->AssociatedIrp.SystemBuffer;
+ void *user_shared_data_address = (void *)0x7ffe0000;
+ NTSTATUS status = STATUS_SUCCESS;
+ LARGE_INTEGER section_size;
+ SIZE_T view_size;
+
+ entry->cid.UniqueProcess = PsGetCurrentProcessId();
+ entry->cid.UniqueThread = (HANDLE)0;
+ if ((status = NtOpenProcess(&entry->process, PROCESS_VM_OPERATION, NULL, &entry->cid)))
+ {
+ ERR("Failed to open target process, status: %x.\n", status);
+ return status;
+ }
+
+ section_size.HighPart = 0;
+ section_size.LowPart = 0x10000;
+ if ((status = NtCreateSection(&entry->section, SECTION_ALL_ACCESS, NULL, §ion_size,
+ PAGE_READWRITE, SEC_COMMIT, NULL)))
+ {
+ ERR("Failed to create section, status: %x.\n", status);
+ return status;
+ }
+
+ view_size = 0;
+ if ((status = NtMapViewOfSection(entry->section, NtCurrentProcess(), &entry->page, 0, 0, 0,
+ &view_size, ViewShare, 0, PAGE_READWRITE)))
+ {
+ ERR("Failed to map section to driver memory, status: %x.\n", status);
+ return status;
+ }
+ memcpy(entry->page, init, sizeof(*init));
+
+ view_size = 0;
+ if ((status = NtMapViewOfSection(entry->section, entry->process, &user_shared_data_address, 0, 0, 0,
+ &view_size, ViewShare, 0, PAGE_READONLY)))
+ {
+ ERR("Failed to map user shared data to target process, status: %x.\n", status);
+ return status;
+ }
+
+ EnterCriticalSection(&wineusd_cs);
+ list_add_head(&wineusd_entries, &entry->link);
+ LeaveCriticalSection(&wineusd_cs);
+
+ TRACE("Initialized user shared data for process %04x.\n", HandleToULong(PsGetCurrentProcessId()));
+
+ return STATUS_SUCCESS;
+}
+
+static void wineusd_close(struct wineusd_entry *entry)
+{
+ void *user_shared_data_address = (void *)0x7ffe0000;
+
+ TRACE("Closing user shared data for process %04x.\n", HandleToULong(entry->cid.UniqueProcess));
+
+ EnterCriticalSection(&wineusd_cs);
+ list_remove(&entry->link);
+ LeaveCriticalSection(&wineusd_cs);
+
+ NtUnmapViewOfSection(entry->process, user_shared_data_address);
+ NtUnmapViewOfSection(NtCurrentProcess(), entry->page);
+
+ NtClose(entry->section);
+ NtClose(entry->process);
+ heap_free(entry);
+}
+
+static NTSTATUS WINAPI wineusd_dispatch_create(DEVICE_OBJECT *device, IRP *irp)
+{
+ IO_STACK_LOCATION *stack = IoGetCurrentIrpStackLocation(irp);
+ struct wineusd_entry *entry;
+ NTSTATUS status;
+
+ status = STATUS_NO_MEMORY;
+ if (!(entry = heap_alloc_zero(sizeof(*entry))))
+ {
+ ERR("Failed to allocate memory.\n");
+ goto done;
+ }
+ stack->FileObject->FsContext = entry;
+
+ list_init(&entry->link);
+ entry->process = INVALID_HANDLE_VALUE;
+ entry->section = INVALID_HANDLE_VALUE;
+ entry->page = NULL;
+
+ TRACE("Created user shared data for process %04x.\n", HandleToULong(PsGetCurrentProcessId()));
+
+done:
+ irp->IoStatus.Status = status;
+ IoCompleteRequest(irp, IO_NO_INCREMENT);
+ return status;
+}
+
+static NTSTATUS WINAPI wineusd_dispatch_close(DEVICE_OBJECT *device, IRP *irp)
+{
+ IO_STACK_LOCATION *stack = IoGetCurrentIrpStackLocation(irp);
+ struct wineusd_entry *entry = stack->FileObject->FsContext;
+
+ if (entry) wineusd_close(entry);
+
+ irp->IoStatus.Status = STATUS_SUCCESS;
+ IoCompleteRequest(irp, IO_NO_INCREMENT);
+ return STATUS_SUCCESS;
+}
+
+static NTSTATUS WINAPI wineusd_dispatch_ioctl(DEVICE_OBJECT *device, IRP *irp)
+{
+ IO_STACK_LOCATION *stack = IoGetCurrentIrpStackLocation(irp);
+ struct wineusd_entry *entry = stack->FileObject->FsContext;
+ NTSTATUS ret;
+
+ switch (stack->Parameters.DeviceIoControl.IoControlCode)
+ {
+ case IOCTL_WINEUSD_INITIALIZE:
+ ret = wineusd_initialize(entry, irp);
+ break;
+ default:
+ FIXME("Unhandled ioctl %#x.\n", stack->Parameters.DeviceIoControl.IoControlCode);
+ ret = STATUS_NOT_IMPLEMENTED;
+ break;
+ }
+
+ irp->IoStatus.Status = ret;
+ IoCompleteRequest(irp, IO_NO_INCREMENT);
+ return ret;
+}
+
+static void WINAPI wineusd_unload(DRIVER_OBJECT *driver)
+{
+ struct wineusd_entry *entry, *cursor;
+
+ SetEvent(wineusd_thread_stop);
+ WaitForSingleObject(wineusd_thread, INFINITE);
+ CloseHandle(wineusd_thread);
+ CloseHandle(wineusd_thread_stop);
+
+ LIST_FOR_EACH_ENTRY_SAFE(entry, cursor, &wineusd_entries, struct wineusd_entry, link)
+ wineusd_close(entry);
+
+ IoDeleteDevice(device_obj);
+}
+
+NTSTATUS WINAPI DriverEntry(DRIVER_OBJECT *driver, UNICODE_STRING *path)
+{
+ static const WCHAR device_nameW[] = {'\\','D','e','v','i','c','e','\\','W','i','n','e','U','s','d',0};
+ OBJECT_ATTRIBUTES attr = {sizeof(attr)};
+ UNICODE_STRING string;
+ NTSTATUS ret;
+
+ TRACE("Driver %p, path %s.\n", driver, debugstr_w(path->Buffer));
+
+ RtlInitUnicodeString(&string, device_nameW);
+ if ((ret = IoCreateDevice(driver, 0, &string, FILE_DEVICE_UNKNOWN, 0, FALSE, &device_obj)))
+ {
+ ERR("Failed to create user shared data device, status %#x.\n", ret);
+ return ret;
+ }
+
+ driver->MajorFunction[IRP_MJ_CREATE] = wineusd_dispatch_create;
+ driver->MajorFunction[IRP_MJ_CLOSE] = wineusd_dispatch_close;
+ driver->MajorFunction[IRP_MJ_DEVICE_CONTROL] = wineusd_dispatch_ioctl;
+ driver->DriverUnload = wineusd_unload;
+
+ wineusd_thread_stop = CreateEventW(NULL, FALSE, FALSE, NULL);
+ wineusd_thread = CreateThread(NULL, 0, wineusd_thread_proc, NULL, 0, NULL);
+
+ return STATUS_SUCCESS;
+}
diff --git a/include/wine/usd.h b/include/wine/usd.h
new file mode 100644
index 00000000000..e92efa5e29e
--- /dev/null
+++ b/include/wine/usd.h
@@ -0,0 +1,27 @@
+/*
+ * Copyright 2019 Rémi Bernon for CodeWeavers
+ *
+ * This library is free software; you can redistribute it and/or
+ * modify it under the terms of the GNU Lesser General Public
+ * License as published by the Free Software Foundation; either
+ * version 2.1 of the License, or (at your option) any later version.
+ *
+ * This library is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
+ * Lesser General Public License for more details.
+ *
+ * You should have received a copy of the GNU Lesser General Public
+ * License along with this library; if not, write to the Free Software
+ * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
+ */
+
+#ifndef __WINE_WINE_USD_H
+#define __WINE_WINE_USD_H
+
+#include <windef.h>
+#include <winioctl.h>
+
+#define IOCTL_WINEUSD_INITIALIZE CTL_CODE(FILE_DEVICE_UNKNOWN, 0x800, METHOD_BUFFERED, FILE_ANY_ACCESS)
+
+#endif /* __WINE_WINE_USD_H */
diff --git a/loader/wine.inf.in b/loader/wine.inf.in
index c0f6e0a3917..4c95c4e7fe2 100644
--- a/loader/wine.inf.in
+++ b/loader/wine.inf.in
@@ -161,6 +161,7 @@ AddService=FontCache3.0.0.0,0,WPFFontCacheService
AddService=LanmanServer,0,LanmanServerService
AddService=FontCache,0,FontCacheService
AddService=Schedule,0,TaskSchedulerService
+AddService=WineUsd,0,WineUsdService
AddService=Winmgmt,0,WinmgmtService
AddService=wuauserv,0,wuauService
@@ -178,6 +179,7 @@ AddService=FontCache3.0.0.0,0,WPFFontCacheService
AddService=LanmanServer,0,LanmanServerService
AddService=FontCache,0,FontCacheService
AddService=Schedule,0,TaskSchedulerService
+AddService=WineUsd,0,WineUsdService
AddService=Winmgmt,0,WinmgmtService
AddService=wuauserv,0,wuauService
@@ -195,6 +197,7 @@ AddService=FontCache3.0.0.0,0,WPFFontCacheService
AddService=LanmanServer,0,LanmanServerService
AddService=FontCache,0,FontCacheService
AddService=Schedule,0,TaskSchedulerService
+AddService=WineUsd,0,WineUsdService
AddService=Winmgmt,0,WinmgmtService
AddService=wuauserv,0,wuauService
@@ -213,6 +216,7 @@ AddService=LanmanServer,0,LanmanServerService
AddService=FontCache,0,FontCacheService
AddService=Schedule,0,TaskSchedulerService
AddService=WineBus,0,WineBusService
+AddService=WineUsd,0,WineUsdService
AddService=Winmgmt,0,WinmgmtService
AddService=wuauserv,0,wuauService
@@ -3641,6 +3645,14 @@ ServiceType=32
StartType=3
ErrorControl=1
+[WineUsdService]
+Description="User shared data update service"
+DisplayName="Wine User Shared Data"
+ServiceBinary="%12%\WineUsd.sys"
+ServiceType=1
+StartType=0
+ErrorControl=1
+
[WinmgmtService]
Description="Provides access to Windows Management Instrumentation"
DisplayName="Windows Management Instrumentation Service"
--
2.24.0
More information about the wine-devel
mailing list