wayland/screencopy: avoid crash when capturing null screen

changelog/next.md

@@ -15,3 +15,4 @@
 - Fixed a crash when quitting or reloading while a `Loader` was still loading asynchronously.
 - Fixed future `Socket` connection attempts after a failed connection attempt.
 - Fixed Open wifi networks showing as Unknown security.
+- Fixed a crash when capturing screencopy with a null screen.

src/wayland/screencopy/image_copy_capture/image_copy_capture.cpp

@@ -220,6 +220,8 @@ IccOutputSourceManager* IccOutputSourceManager::instance() {
 }
 
 ScreencopyContext* IccOutputSourceManager::captureOutput(QScreen* screen, bool paintCursors) {
+	if (screen == nullptr) return nullptr;
+
 	auto* waylandScreen = dynamic_cast<QtWaylandClient::QWaylandScreen*>(screen->handle());
 	if (!waylandScreen) return nullptr;
 

src/wayland/screencopy/manager.cpp

@@ -25,6 +25,8 @@ namespace qs::wayland::screencopy {
 
 ScreencopyContext* ScreencopyManager::createContext(QObject* object, bool paintCursors) {
 	if (auto* screen = qobject_cast<QuickshellScreenInfo*>(object)) {
+		if (screen->screen == nullptr) return nullptr;
+
 #if SCREENCOPY_ICC
 		{
 			auto* manager = icc::IccOutputSourceManager::instance();

src/wayland/screencopy/wlr_screencopy/wlr_screencopy.cpp

@@ -34,6 +34,7 @@ WlrScreencopyManager* WlrScreencopyManager::instance() {
 
 ScreencopyContext*
 WlrScreencopyManager::captureOutput(QScreen* screen, bool paintCursors, QRect region) {
+	if (screen == nullptr) return nullptr;
 	if (!dynamic_cast<QtWaylandClient::QWaylandScreen*>(screen->handle())) return nullptr;
 	return new WlrScreencopyContext(this, screen, paintCursors, region);
 }