[PATCH 2/7] winewayland.drv: Handle dynamic Wayland output events.

[Alexandros Frantzis]

From: Alexandros Frantzis

Handle Wayland output events sent after process initialization,
and update the win32u display devices when handling these
events in the desktop window process.

Signed-off-by: Alexandros Frantzis 
---
 dlls/winewayland.drv/display.c        | 14 +++++++++++---
 dlls/winewayland.drv/wayland.c        |  5 ++++-
 dlls/winewayland.drv/wayland_output.c | 24 ++++++++++++++++++++++++
 dlls/winewayland.drv/waylanddrv.h     |  3 ++-
 4 files changed, 41 insertions(+), 5 deletions(-)

diff --git a/dlls/winewayland.drv/display.c b/dlls/winewayland.drv/display.c
index 5234a1f03aa..0e1c7f17d1e 100644
--- a/dlls/winewayland.drv/display.c
+++ b/dlls/winewayland.drv/display.c
@@ -34,9 +34,15 @@

WINE_DEFAULT_DEBUG_CHANNEL(waylanddrv);

-void wayland_init_display_devices(void)
+BOOL force_display_devices_refresh = FALSE;
+
+void wayland_init_display_devices(BOOL force)
 {
     UINT32 num_path, num_mode;
+
+    TRACE("force=%d\n", force);
+
+    if (force) force_display_devices_refresh = TRUE;
     /* Trigger refresh in win32u */
     NtUserGetDisplayConfigBufferSizes(QDC_ONLY_ACTIVE_PATHS, &num_path, &num_mode);
 }
@@ -284,9 +290,11 @@ BOOL WAYLAND_UpdateDisplayDevices(const struct gdi_device_manager *device_manage
     struct wl_array output_info_array;
     struct output_info *output_info;

-    if (!force) return TRUE;
+    if (!force && !force_display_devices_refresh) return TRUE;

-    TRACE("force=%d\n", force);
+    TRACE("force=%d force_refresh=%d\n", force, force_display_devices_refresh);
+
+    force_display_devices_refresh = FALSE;

wl_array_init(&output_info_array);

diff --git a/dlls/winewayland.drv/wayland.c b/dlls/winewayland.drv/wayland.c
index 22a7a8cc63f..c6ebbe9aeb9 100644
--- a/dlls/winewayland.drv/wayland.c
+++ b/dlls/winewayland.drv/wayland.c
@@ -34,6 +34,7 @@ WINE_DEFAULT_DEBUG_CHANNEL(waylanddrv);

struct wl_display *process_wl_display = NULL;
 struct wayland *process_wayland = NULL;
+BOOL process_wayland_initialized = FALSE;

/**********************************************************************
  *          Registry handling
@@ -140,7 +141,9 @@ BOOL wayland_process_init(void)
     wl_display_roundtrip_queue(process_wl_display, process_wayland->wl_event_queue);
     wl_display_roundtrip_queue(process_wl_display, process_wayland->wl_event_queue);

-    wayland_init_display_devices();
+    wayland_init_display_devices(FALSE);
+
+    process_wayland_initialized = TRUE;

return TRUE;
 }
diff --git a/dlls/winewayland.drv/wayland_output.c b/dlls/winewayland.drv/wayland_output.c
index 5cd6e74f24a..7d49c6d81fe 100644
--- a/dlls/winewayland.drv/wayland_output.c
+++ b/dlls/winewayland.drv/wayland_output.c
@@ -97,6 +97,26 @@ static void wayland_output_add_mode(struct wayland_output *output,
     if (current) output->current_mode = mode;
 }

+static void maybe_init_display_devices(void)
+{
+    DWORD desktop_pid = 0;
+    HWND desktop_hwnd;
+
+    /* Right after process init we initialize all the display devices, so there
+     * is no need to react to each individual event at that time. This check
+     * also helps us avoid calling NtUserGetDesktopWindow() (see below) at
+     * process init time, since it may not be safe. */
+    if (!process_wayland_initialized) return;
+
+    desktop_hwnd = NtUserGetDesktopWindow();
+    NtUserGetWindowThread(desktop_hwnd, &desktop_pid);
+
+    /* We only update the display devices from the desktop process. */
+    if (GetCurrentProcessId() != desktop_pid) return;
+
+    wayland_init_display_devices(TRUE);
+}
+
 static void wayland_output_done(struct wayland_output *output)
 {
     struct wayland_output_mode *mode;
@@ -111,6 +131,8 @@ static void wayland_output_done(struct wayland_output *output)
               mode->width, mode->height, mode->refresh,
               output->current_mode == mode ? "*" : "");
     }
+
+    maybe_init_display_devices();
 }

static void output_handle_geometry(void *data, struct wl_output *wl_output,
@@ -283,6 +305,8 @@ void wayland_output_destroy(struct wayland_output *output)
     wl_output_destroy(output->wl_output);
     free(output->name);
     free(output);
+
+    maybe_init_display_devices();
 }

/**********************************************************************
diff --git a/dlls/winewayland.drv/waylanddrv.h b/dlls/winewayland.drv/waylanddrv.h
index e1ea66f3b33..6316079019a 100644
--- a/dlls/winewayland.drv/waylanddrv.h
+++ b/dlls/winewayland.drv/waylanddrv.h
@@ -41,6 +41,7 @@

extern struct wl_display *process_wl_display DECLSPEC_HIDDEN;
 extern struct wayland *process_wayland DECLSPEC_HIDDEN;
+extern BOOL process_wayland_initialized DECLSPEC_HIDDEN;

/**********************************************************************
  *          Definitions for wayland types
@@ -80,7 +81,7 @@ struct wayland_output
  */

BOOL wayland_process_init(void) DECLSPEC_HIDDEN;
-void wayland_init_display_devices(void) DECLSPEC_HIDDEN;
+void wayland_init_display_devices(BOOL force) DECLSPEC_HIDDEN;

/**********************************************************************
  *          Wayland output