From 5fde176a9a3cc746f8aab24fb850e5b90aa51710 Mon Sep 17 00:00:00 2001
From: pg9182 <96569817+pg9182@users.noreply.github.com>
Date: Sun, 11 Dec 2022 18:47:55 -0500
Subject: Add platform information to User-Agent (#367)

* Add platform information to User-Agent.
* Identify dedicated servers in the User-Agent.
---
 NorthstarDLL/version.cpp | 76 ++++++++++++++++++++++++++++++++++++++++++++++--
 1 file changed, 73 insertions(+), 3 deletions(-)

(limited to 'NorthstarDLL')

diff --git a/NorthstarDLL/version.cpp b/NorthstarDLL/version.cpp
index 3755e4c4..088fb491 100644
--- a/NorthstarDLL/version.cpp
+++ b/NorthstarDLL/version.cpp
@@ -1,13 +1,15 @@
 #include "version.h"
 #include "pch.h"
 #include "ns_version.h"
+#include "dedicated.h"
 
 char version[16];
-char NSUserAgent[32];
+char NSUserAgent[256];
 
 void InitialiseVersion()
 {
 	constexpr int northstar_version[4] {NORTHSTAR_VERSION};
+	int ua_len = 0;
 
 	// We actually use the rightmost integer do determine whether or not we're a debug/dev build
 	// If it is set to 1, we are a dev build
@@ -15,12 +17,80 @@ void InitialiseVersion()
 	if (northstar_version[3] == 1)
 	{
 		sprintf(version, "%d.%d.%d.%d+dev", northstar_version[0], northstar_version[1], northstar_version[2], northstar_version[3]);
-		sprintf(NSUserAgent, "R2Northstar/%d.%d.%d+dev", northstar_version[0], northstar_version[1], northstar_version[2]);
+		ua_len += snprintf(
+			NSUserAgent + ua_len,
+			sizeof(NSUserAgent) - ua_len,
+			"R2Northstar/%d.%d.%d+dev",
+			northstar_version[0],
+			northstar_version[1],
+			northstar_version[2]);
 	}
 	else
 	{
 		sprintf(version, "%d.%d.%d.%d", northstar_version[0], northstar_version[1], northstar_version[2], northstar_version[3]);
-		sprintf(NSUserAgent, "R2Northstar/%d.%d.%d", northstar_version[0], northstar_version[1], northstar_version[2]);
+		ua_len += snprintf(
+			NSUserAgent + ua_len,
+			sizeof(NSUserAgent) - ua_len,
+			"R2Northstar/%d.%d.%d",
+			northstar_version[0],
+			northstar_version[1],
+			northstar_version[2]);
 	}
+
+	if (IsDedicatedServer())
+		ua_len += snprintf(NSUserAgent + ua_len, sizeof(NSUserAgent) - ua_len, " (Dedicated)");
+
+	// Add the host platform info to the user agent.
+	//
+	// note: ntdll will always be loaded
+	HMODULE ntdll = GetModuleHandleA("ntdll");
+	if (ntdll)
+	{
+		// real win32 version info (i.e., ignore manifest)
+		DWORD(WINAPI * RtlGetVersion)(LPOSVERSIONINFOEXW);
+		*(FARPROC*)(&RtlGetVersion) = GetProcAddress(ntdll, "RtlGetVersion");
+
+		// wine version (7.0-rc1, 7.0, 7.11, etc)
+		const char*(CDECL * wine_get_version)(void);
+		*(FARPROC*)(&wine_get_version) = GetProcAddress(ntdll, "wine_get_version");
+
+		// human-readable build string (e.g., "wine-7.22 (Staging)")
+		const char*(CDECL * wine_get_build_id)(void);
+		*(FARPROC*)(&wine_get_build_id) = GetProcAddress(ntdll, "wine_get_build_id");
+
+		// uname sysname (Darwin, Linux, etc) and release (kernel version string)
+		void(CDECL * wine_get_host_version)(const char** sysname, const char** release);
+		*(FARPROC*)(&wine_get_host_version) = GetProcAddress(ntdll, "wine_get_host_version");
+
+		OSVERSIONINFOEXW osvi = {};
+		const char *wine_version = NULL, *wine_build_id = NULL, *wine_sysname = NULL, *wine_release = NULL;
+		if (RtlGetVersion)
+			RtlGetVersion(&osvi);
+		if (wine_get_version)
+			wine_version = wine_get_version();
+		if (wine_get_build_id)
+			wine_build_id = wine_get_build_id();
+		if (wine_get_host_version)
+			wine_get_host_version(&wine_sysname, &wine_release);
+
+		// windows version
+		if (osvi.dwMajorVersion)
+			ua_len += snprintf(
+				NSUserAgent + ua_len,
+				sizeof(NSUserAgent) - ua_len,
+				" Windows/%d.%d.%d",
+				osvi.dwMajorVersion,
+				osvi.dwMinorVersion,
+				osvi.dwBuildNumber);
+
+		// wine version
+		if (wine_version && wine_build_id)
+			ua_len += snprintf(NSUserAgent + ua_len, sizeof(NSUserAgent) - ua_len, " Wine/%s (%s)", wine_version, wine_build_id);
+
+		// wine host system version
+		if (wine_sysname && wine_release)
+			ua_len += snprintf(NSUserAgent + ua_len, sizeof(NSUserAgent) - ua_len, " %s/%s", wine_sysname, wine_release);
+	}
+
 	return;
 }
-- 
cgit v1.2.3