Added Support for VR in Launcher Setup

Launcher/WiiCompiled.Setup.Windows/CommandLine.cs

@@ -14,6 +14,7 @@ internal enum AppMode
     CheckProducts,
     RepairProducts,
     Version,
+    InfoJson,
     EmitPayloadIdentities
 }
 
@@ -58,6 +59,7 @@ internal sealed class CommandLine
             switch (args[i].ToLowerInvariant())
             {
                 case "--version": result.Mode = AppMode.Version; break;
+                case "--info-json": result.Mode = AppMode.InfoJson; break;
                 case "--silent": result.Mode = AppMode.SilentInstall; break;
                 case "--verify-inputs": result.Mode = AppMode.VerifyInputs; break;
                 case "--uninstall": result.Mode = AppMode.Uninstall; break;

Launcher/WiiCompiled.Setup.Windows/ConsoleCommands.cs

@@ -19,6 +19,7 @@ internal static class ConsoleCommands
         Console.Out.WriteLine("  --launch-retro | --launch-base");
         Console.Out.WriteLine("  --uninstall --install-dir <dir>");
         Console.Out.WriteLine("  --version");
+        Console.Out.WriteLine("  --info-json");
         return 0;
     }
 
@@ -28,6 +29,15 @@ internal static class ConsoleCommands
         return 0;
     }
 
+    public static int InfoJson()
+    {
+        Console.Out.WriteLine(JsonSerializer.Serialize(new
+        {
+            productId = ProductInfo.Id, version = ProductInfo.Version, openxrD3D12 = true
+        }));
+        return 0;
+    }
+
 
     public static int EmitPayloadIdentities(CommandLine command)
     {

Launcher/WiiCompiled.Setup.Windows/GameLaunchService.cs

@@ -7,6 +7,7 @@ internal static class GameLaunchService
     public static Task<int> LaunchAsync(BuildProfile profile)
     {
         var installation = new Installation(AppContext.BaseDirectory);
+        ProductOwnership.Ensure(installation.Root);
         using var operation = InstallOperationLock.Acquire(installation.Root);
         // A portable installation the user moved is reconciled before anything is read from its
         // recorded location. Launch never needs the discarded native build tree.

Launcher/WiiCompiled.Setup.Windows/InstallerEngine.cs

@@ -12,6 +12,7 @@ internal sealed class InstallerEngine
     {
         var installDirectory = Path.GetFullPath(options.InstallDirectory);
         ValidateInstallDirectory(installDirectory);
+        ProductOwnership.Ensure(installDirectory, allowEmpty: true);
 
         InputValidation.ValidateExtension(options.GamePath);
         var parent = Directory.GetParent(installDirectory)?.FullName
@@ -37,6 +38,8 @@ internal sealed class InstallerEngine
 
         using var payload = PayloadArchive.OpenCurrent();
         var manifest = payload.ReadManifest();
+        if (manifest.ProductId != ProductInfo.Id || manifest.ProductVersion != ProductInfo.Version)
+            throw new InvalidDataException("This setup payload does not match WiiCompiled OpenXR VR and its version.");
         var toolkit = InstalledLayout.Toolkit(staging);
         var workspace = InstalledLayout.Workspace(staging);
 
@@ -366,6 +369,7 @@ internal sealed class InstallerEngine
             ProductVersion = manifest.ProductVersion,
             InstallDir = installDirectory,
             InstalledUtc = previousState?.InstalledUtc ?? DateTime.UtcNow,
+            ProductId = ProductInfo.Id,
             RetroRewindInstalled = previousState?.RetroRewindInstalled ?? false,
             ToolkitReleaseTag = manifest.ToolkitReleaseTag,
             DolSha256 = manifest.ExpectedDolSha256,

Launcher/WiiCompiled.Setup.Windows/Models.cs

@@ -35,6 +35,7 @@ internal sealed class InstallOptions
 
 internal sealed class PayloadManifest
 {
+    public string ProductId { get; set; } = "";
     public int SchemaVersion { get; set; }
     public string ProductVersion { get; set; } = "";
     public string ExpectedGameId { get; set; } = "RMCP01";
@@ -72,6 +73,7 @@ internal sealed class DiscHeader
 
 internal sealed class InstallState
 {
+    public string ProductId { get; set; } = "";
     [JsonRequired]
     public int SchemaVersion { get; set; } = 1;
 

Launcher/WiiCompiled.Setup.Windows/ProductOwnership.cs

@@ -0,0 +1,29 @@
+using System.Text.Json;
+
+namespace WiiCompiled.Setup.Windows;
+
+internal static class ProductOwnership
+{
+    public static void Ensure(string directory, bool allowEmpty = false)
+    {
+        for (var current = new DirectoryInfo(Path.GetFullPath(directory)); current is not null; current = current.Parent)
+            if (current.Exists && (current.Attributes & FileAttributes.ReparsePoint) != 0)
+                throw new InvalidDataException("Use a VR installation directory without junctions or symbolic links.");
+        if (!Directory.Exists(directory))
+        {
+            if (allowEmpty) return;
+            throw new InvalidDataException("WiiCompiled OpenXR VR is not installed in this directory.");
+        }
+        // A junction must not let an apparently separate VR destination overwrite another install.
+        if (allowEmpty && !Directory.EnumerateFileSystemEntries(directory).Any()) return;
+        var statePath = Path.Combine(directory, InstalledLayout.InstallStateFileName);
+        if (File.Exists(statePath))
+        {
+            using var state = JsonDocument.Parse(File.ReadAllText(statePath));
+            if (state.RootElement.TryGetProperty("ProductId", out var id) && id.GetString() == ProductInfo.Id)
+                return;
+        }
+        throw new InvalidDataException("This folder belongs to a normal or unidentified WiiCompiled installation. " +
+            "Choose a separate empty folder for OpenXR VR; existing files have not been changed.");
+    }
+}

Launcher/WiiCompiled.Setup.Windows/ProductRepairService.cs

@@ -77,6 +77,7 @@ internal sealed class ProductRepairService
         RetroWfcPayloadMode requestedPayloadMode, ReconcileOptions options,
         CancellationToken cancellationToken)
     {
+        ProductOwnership.Ensure(_installation.Root);
         if (!_installation.IsPresent)
             throw new InvalidOperationException("WiiCompiled is not installed here.");
         if (!_installation.HasToolkit)

Launcher/WiiCompiled.Setup.Windows/Program.cs

@@ -16,7 +16,7 @@ internal static class Program
             {
                 Console.Out.WriteLine(
                     "Mario Kart WiiCompiled is installed through Wheel Wizard - download it from " +
-                    "https://github.com/TeamWheelWizard/WheelWizard");
+                    "https://github.com/iChris4/WheelWizard_VR");
                 Console.ReadKey(intercept: true);
                 return 0;
             }
@@ -37,6 +37,8 @@ internal static class Program
 
             if (command.Mode == AppMode.Version)
                 return ConsoleCommands.Version();
+            if (command.Mode == AppMode.InfoJson)
+                return ConsoleCommands.InfoJson();
 
             if (command.Mode == AppMode.Help)
                 return ConsoleCommands.Help();
@@ -120,8 +122,9 @@ internal static class PlatformChecks
 
 internal static class ProductInfo
 {
-    public const string Name = "WiiCompiled";
-    public const string Version = "0.2.31";
+    public const string Name = "WiiCompiled OpenXR VR";
+    public const string Id = "wiicompiled-openxr-vr";
+    public static string Version => typeof(ProductInfo).Assembly.GetName().Version!.ToString(3);
 
     /// <summary>
     /// The setup executable is copied into the installation under this name. It is the launcher and
@@ -129,8 +132,8 @@ internal static class ProductInfo
     /// </summary>
     public const string SetupCopyName = "WiiCompiled-Setup.exe";
 
-    public const string UninstallKey = @"Software\Microsoft\Windows\CurrentVersion\Uninstall\WiiCompiled";
+    public const string UninstallKey = @"Software\Microsoft\Windows\CurrentVersion\Uninstall\WiiCompiledOpenXRVR";
     public static string DefaultInstallDirectory =>
         Path.Combine(Environment.GetFolderPath(Environment.SpecialFolder.LocalApplicationData),
-            "Programs", "WiiCompiled");
+            "Programs", "WiiCompiledOpenXRVR");
 }

Launcher/WiiCompiled.Setup.Windows/SelfTests.cs

@@ -6,9 +6,36 @@ namespace WiiCompiled.Setup.Windows;
 
 internal static class SelfTests
 {
+    private static void TestVrOwnership()
+    {
+        var root = Path.Combine(Path.GetTempPath(), "WiiCompiled-VR-ownership-" + Guid.NewGuid().ToString("N"));
+        Directory.CreateDirectory(root);
+        try
+        {
+            ProductOwnership.Ensure(root, allowEmpty: true);
+            var statePath = Path.Combine(root, InstalledLayout.InstallStateFileName);
+            File.WriteAllText(statePath, "{\"SetupVersion\":\"0.2.32\"}");
+            var original = File.ReadAllText(statePath);
+            try { ProductOwnership.Ensure(root, allowEmpty: true); throw new Exception("Accepted an unidentified install."); }
+            catch (InvalidDataException) { }
+            if (File.ReadAllText(statePath) != original) throw new Exception("Normal state was changed by rejection.");
+            File.WriteAllText(statePath, "{\"ProductId\":\"wiicompiled\"}");
+            try { ProductOwnership.Ensure(root); throw new Exception("Accepted a normal install."); }
+            catch (InvalidDataException) { }
+            JsonState.Write(statePath, new InstallState { ProductId = ProductInfo.Id, InstallDir = root });
+            ProductOwnership.Ensure(root);
+            if (RuntimeConfiguration.ApplicationDataConfigPath.Contains("WiiCompiled\\Config.toml", StringComparison.Ordinal))
+                throw new Exception("The VR setup still uses normal application data.");
+            if (!ProductInfo.UninstallKey.EndsWith("WiiCompiledOpenXRVR", StringComparison.Ordinal))
+                throw new Exception("The VR uninstall identity is not isolated.");
+        }
+        finally { Directory.Delete(root, recursive: true); }
+    }
+
     public static int Run()
     {
         var failures = new List<string>();
+        Test("VR installation ownership", TestVrOwnership, failures);
         Test("ISO extension", () => TestAcceptedExtension(".iso"), failures);
         Test("GCM extension", () => TestAcceptedExtension(".gcm"), failures);
         Test("GCZ extension", () => TestAcceptedExtension(".gcz"), failures);
@@ -1291,7 +1318,7 @@ internal static class SelfTests
                     RetroWfcPayloadMode = "skipped" });
             JsonState.Write(installation.InstallStatePath, new InstallState
             {
-                InstallDir = root, RetroRewindInstalled = true,
+                ProductId = ProductInfo.Id, InstallDir = root, RetroRewindInstalled = true,
                 DolSha256 = dolSha, RelSha256 = relSha,
                 RetroRewindCodePulSha256 = inputsA.CodePulSha256,
                 RetroRewindCompileInputsSha256 = inputsA.CompileInputsSha256,

Launcher/WiiCompiled.Setup.Windows/ShellIntegration.cs

@@ -4,7 +4,7 @@ namespace WiiCompiled.Setup.Windows;
 
 internal static class ShellIntegration
 {
-    private const string ShortcutFileName = "wiicompiled (base) (beta).lnk";
+    private const string ShortcutFileName = "WiiCompiled OpenXR VR (base) (beta).lnk";
 
     public static void RegisterUninstaller(string installDirectory, bool retroInstalled)
     {

Launcher/WiiCompiled.Setup.Windows/UninstallService.cs

@@ -28,6 +28,9 @@ internal static class UninstallService
     public static int RunWorker(string installDirectory, bool quiet)
     {
         installDirectory = Path.GetFullPath(installDirectory);
+        ProductOwnership.Ensure(installDirectory);
+        using var operation = InstallOperationLock.Acquire(installDirectory);
+        RunningProductGuard.EnsureProductsNotRunning(installDirectory);
         Thread.Sleep(750);
         TryCleanup("remove shortcuts", quiet, ShellIntegration.RemoveShortcuts);
 

Launcher/WiiCompiled.Setup.Windows/WiiCompiled.Setup.Windows.csproj

@@ -7,9 +7,8 @@
     <AssemblyName>WiiCompiled.Setup</AssemblyName>
     <RootNamespace>WiiCompiled.Setup.Windows</RootNamespace>
     <ApplicationManifest>app.manifest</ApplicationManifest>
-    <Version>0.2.31</Version>
     <Authors>patchzy</Authors>
-    <Product>WiiCompiled</Product>
+    <Product>WiiCompiled OpenXR VR</Product>
     <Description>Command-line installer and launcher for WiiCompiled</Description>
     <DebugType>embedded</DebugType>
     <SatelliteResourceLanguages>en</SatelliteResourceLanguages>