Add opt-in screenshot attachment for crash events captured on Windows/Linux

CHANGELOG.md

@@ -2,6 +2,10 @@
 
 ## Unreleased
 
+### Features
+
+- Add opt-in screenshot attachment for crash events captured on Windows/Linux ([#582](https://github.com/getsentry/sentry-unreal/pull/582))
+
 ### Dependencies
 
 - Bump Cocoa SDK (iOS) from v8.29.0 to v8.29.1 ([#580](https://github.com/getsentry/sentry-unreal/pull/580))

plugin-dev/Source/Sentry/Private/Desktop/SentrySubsystemDesktop.cpp

@@ -21,6 +21,8 @@
 #include "SentryTransactionContext.h"
 #include "SentryUserFeedbackDesktop.h"
 
+#include "Utils/SentryScreenshotUtils.h"
+
 #include "Infrastructure/SentryConvertorsDesktop.h"
 
 #include "CrashReporter/SentryCrashReporter.h"
@@ -77,6 +79,7 @@ sentry_value_t HandleBeforeSend(sentry_value_t event, void *hint, void *closure)
 sentry_value_t HandleBeforeCrash(const sentry_ucontext_t *uctx, sentry_value_t event, void *closure)
 {
 	SentrySubsystemDesktop* SentrySubsystem = static_cast<SentrySubsystemDesktop*>(closure);
+	SentrySubsystem->TryCaptureScreenshot();
 
 	FSentryCrashContext::Get()->Apply(SentrySubsystem->GetCurrentScope());
 
@@ -107,6 +110,7 @@ SentrySubsystemDesktop::SentrySubsystemDesktop()
 	, crashReporter(MakeShareable(new SentryCrashReporter))
 	, isEnabled(false)
 	, isStackTraceEnabled(true)
+	, isScreenshotAttachmentEnabled(false)
 {
 }
 
@@ -129,6 +133,19 @@ void SentrySubsystemDesktop::InitWithSettings(const USentrySettings* settings, U
 #endif
 	}
 
+	if(settings->AttachScreenshot)
+	{
+		isScreenshotAttachmentEnabled = true;
+
+		const FString ScreenshotPath = FGenericPlatformOutputDevices::GetAbsoluteLogFilename();
+
+#if PLATFORM_WINDOWS
+		sentry_options_add_attachmentw(options, *GetScreenshotPath());
+#elif PLATFORM_LINUX
+		sentry_options_add_attachment(options, TCHAR_TO_UTF8(*GetScreenshotPath()));
+#endif
+	}
+
 	if(settings->UseProxy)
 	{
 		sentry_options_set_http_proxy(options, TCHAR_TO_ANSI(*settings->ProxyUrl));
@@ -450,6 +467,17 @@ USentryBeforeSendHandler* SentrySubsystemDesktop::GetBeforeSendHandler()
 	return beforeSend;
 }
 
+void SentrySubsystemDesktop::TryCaptureScreenshot() const
+{
+	if(!isScreenshotAttachmentEnabled)
+	{
+		UE_LOG(LogSentrySdk, Log, TEXT("Screenshot attachment is disabled in plugin settings."));
+		return;
+	}
+
+	SentryScreenshotUtils::CaptureScreenshot(GetScreenshotPath());
+}
+
 TSharedPtr<SentryScopeDesktop> SentrySubsystemDesktop::GetCurrentScope()
 {
 	if(scopeStack.Num() == 0)
@@ -483,4 +511,12 @@ FString SentrySubsystemDesktop::GetDatabasePath() const
 	return DatabaseFullPath;
 }
 
+FString SentrySubsystemDesktop::GetScreenshotPath() const
+{
+	const FString ScreenshotPath = FPaths::Combine(GetDatabasePath(), TEXT("screenshots"), TEXT("crash_screenshot.png"));
+	const FString ScreenshotFullPath = FPaths::ConvertRelativePathToFull(ScreenshotPath);
+
+	return ScreenshotFullPath;
+}
+
 #endif

plugin-dev/Source/Sentry/Private/Desktop/SentrySubsystemDesktop.h

@@ -46,11 +46,14 @@ class SentrySubsystemDesktop : public ISentrySubsystem
 
 	USentryBeforeSendHandler* GetBeforeSendHandler();
 
+	void TryCaptureScreenshot() const;
+
 	TSharedPtr<SentryScopeDesktop> GetCurrentScope();
 
 private:
 	FString GetHandlerPath() const;
 	FString GetDatabasePath() const;
+	FString GetScreenshotPath() const;
 
 	USentryBeforeSendHandler* beforeSend;
 
@@ -61,6 +64,7 @@ class SentrySubsystemDesktop : public ISentrySubsystem
 	bool isEnabled;
 
 	bool isStackTraceEnabled;
+	bool isScreenshotAttachmentEnabled;
 
 	FCriticalSection CriticalSection;
 };

plugin-dev/Source/Sentry/Private/SentrySubsystem.cpp

@@ -206,15 +206,15 @@ USentryId* USentrySubsystem::CaptureMessage(const FString& Message, ESentryLevel
 
 USentryId* USentrySubsystem::CaptureMessageWithScope(const FString& Message, const FConfigureScopeDelegate& OnConfigureScope, ESentryLevel Level)
 {
-    return CaptureMessageWithScope(Message, FConfigureScopeNativeDelegate::CreateUFunction(const_cast<UObject*>(OnConfigureScope.GetUObject()), OnConfigureScope.GetFunctionName()), Level);
+	return CaptureMessageWithScope(Message, FConfigureScopeNativeDelegate::CreateUFunction(const_cast<UObject*>(OnConfigureScope.GetUObject()), OnConfigureScope.GetFunctionName()), Level);
 }
 
 USentryId* USentrySubsystem::CaptureMessageWithScope(const FString& Message, const FConfigureScopeNativeDelegate& OnConfigureScope, ESentryLevel Level)
 {
 	if (!SubsystemNativeImpl || !SubsystemNativeImpl->IsEnabled())
 		return nullptr;
 
-    return SubsystemNativeImpl->CaptureMessageWithScope(Message, OnConfigureScope, Level);
+	return SubsystemNativeImpl->CaptureMessageWithScope(Message, OnConfigureScope, Level);
 }
 
 USentryId* USentrySubsystem::CaptureEvent(USentryEvent* Event)
@@ -227,15 +227,15 @@ USentryId* USentrySubsystem::CaptureEvent(USentryEvent* Event)
 
 USentryId* USentrySubsystem::CaptureEventWithScope(USentryEvent* Event, const FConfigureScopeDelegate& OnConfigureScope)
 {
-    return CaptureEventWithScope(Event, FConfigureScopeNativeDelegate::CreateUFunction(const_cast<UObject*>(OnConfigureScope.GetUObject()), OnConfigureScope.GetFunctionName()));
+	return CaptureEventWithScope(Event, FConfigureScopeNativeDelegate::CreateUFunction(const_cast<UObject*>(OnConfigureScope.GetUObject()), OnConfigureScope.GetFunctionName()));
 }
 
 USentryId* USentrySubsystem::CaptureEventWithScope(USentryEvent* Event, const FConfigureScopeNativeDelegate& OnConfigureScope)
 {
 	if (!SubsystemNativeImpl || !SubsystemNativeImpl->IsEnabled())
 		return nullptr;
 
-    return SubsystemNativeImpl->CaptureEventWithScope(Event, OnConfigureScope);
+	return SubsystemNativeImpl->CaptureEventWithScope(Event, OnConfigureScope);
 }
 
 void USentrySubsystem::CaptureUserFeedback(USentryUserFeedback* UserFeedback)
@@ -275,15 +275,15 @@ void USentrySubsystem::RemoveUser()
 
 void USentrySubsystem::ConfigureScope(const FConfigureScopeDelegate& OnConfigureScope)
 {
-    ConfigureScope(FConfigureScopeNativeDelegate::CreateUFunction(const_cast<UObject*>(OnConfigureScope.GetUObject()), OnConfigureScope.GetFunctionName()));
+	ConfigureScope(FConfigureScopeNativeDelegate::CreateUFunction(const_cast<UObject*>(OnConfigureScope.GetUObject()), OnConfigureScope.GetFunctionName()));
 }
 
 void USentrySubsystem::ConfigureScope(const FConfigureScopeNativeDelegate& OnConfigureScope)
 {
 	if (!SubsystemNativeImpl || !SubsystemNativeImpl->IsEnabled())
 		return;
 
-    SubsystemNativeImpl->ConfigureScope(OnConfigureScope);
+	SubsystemNativeImpl->ConfigureScope(OnConfigureScope);
 }
 
 void USentrySubsystem::SetContext(const FString& Key, const TMap<FString, FString>& Values)

plugin-dev/Source/Sentry/Private/Utils/SentryScreenshotUtils.cpp

@@ -0,0 +1,59 @@
+// Copyright (c) 2024 Sentry. All Rights Reserved.
+
+#include "SentryScreenshotUtils.h"
+
+#include "HighResScreenshot.h"
+#include "SentryDefines.h"
+
+#include "Engine/Engine.h"
+#include "ImageUtils.h"
+#include "UnrealClient.h"
+#include "Misc/FileHelper.h"
+#include "Engine/GameViewportClient.h"
+#include "Framework/Application/SlateApplication.h"
+
+bool SentryScreenshotUtils::CaptureScreenshot(const FString& ScreenshotSavePath)
+{
+	if (!GEngine || !GEngine->GameViewport)
+	{
+		UE_LOG(LogSentrySdk, Error, TEXT("GameViewport required for screenshot capturing is not valid"));
+		return false;
+	}
+
+	UGameViewportClient* GameViewportClient = GEngine->GameViewport;
+	if (!GameViewportClient)
+	{
+		UE_LOG(LogSentrySdk, Error, TEXT("Game Viewport Client required for screenshot capturing is not valid"));
+		return false;
+	}
+
+	FIntVector ViewportSize(GameViewportClient->Viewport->GetSizeXY().X, GameViewportClient->Viewport->GetSizeXY().Y, 0);
+
+	TArray<FColor> Bitmap;
+
+	if (!FSlateApplication::IsInitialized())
+	{
+		UE_LOG(LogSentrySdk, Error, TEXT("Slate application required for screenshot capturing is not initialized"));
+		return false;
+	}
+
+	TSharedPtr<SWindow> WindowPtr = GameViewportClient->GetWindow();
+	TSharedRef<SWidget> WindowRef = WindowPtr.ToSharedRef();
+
+	bool bScreenshotSuccessful = FSlateApplication::Get().TakeScreenshot(WindowRef, Bitmap, ViewportSize);
+	if (!bScreenshotSuccessful)
+	{
+		UE_LOG(LogSentrySdk, Error, TEXT("Failed to capture screenshot"));
+		return false;
+	}
+
+	GetHighResScreenshotConfig().MergeMaskIntoAlpha(Bitmap, FIntRect());
+
+	TArray64<uint8> CompressedBitmap;
+	FImageUtils::PNGCompressImageArray(ViewportSize.X, ViewportSize.Y, Bitmap, CompressedBitmap);
+	FFileHelper::SaveArrayToFile(CompressedBitmap, *ScreenshotSavePath);
+
+	UE_LOG(LogSentrySdk, Log, TEXT("Screenshot saved to: %s"), *ScreenshotSavePath);
+
+	return true;
+}

plugin-dev/Source/Sentry/Private/Utils/SentryScreenshotUtils.h

@@ -0,0 +1,11 @@
+// Copyright (c) 2024 Sentry. All Rights Reserved.
+
+#pragma once
+
+#include "CoreMinimal.h"
+
+class SentryScreenshotUtils
+{
+public:
+	static bool CaptureScreenshot(const FString& ScreenshotSavePath);
+};

scripts/packaging/package-github.snapshot

@@ -169,6 +169,8 @@ Source/Sentry/Private/Tests/SentryUser.spec.cpp
 Source/Sentry/Private/Utils/
 Source/Sentry/Private/Utils/SentryFileUtils.cpp
 Source/Sentry/Private/Utils/SentryFileUtils.h
+Source/Sentry/Private/Utils/SentryScreenshotUtils.cpp
+Source/Sentry/Private/Utils/SentryScreenshotUtils.h
 Source/Sentry/Public/
 Source/Sentry/Public/SentryAttachment.h
 Source/Sentry/Public/SentryBeforeSendHandler.h

scripts/packaging/package-marketplace.snapshot

@@ -167,6 +167,8 @@ Source/Sentry/Private/Tests/SentryUser.spec.cpp
 Source/Sentry/Private/Utils/
 Source/Sentry/Private/Utils/SentryFileUtils.cpp
 Source/Sentry/Private/Utils/SentryFileUtils.h
+Source/Sentry/Private/Utils/SentryScreenshotUtils.cpp
+Source/Sentry/Private/Utils/SentryScreenshotUtils.h
 Source/Sentry/Public/
 Source/Sentry/Public/SentryAttachment.h
 Source/Sentry/Public/SentryBeforeSendHandler.h