Restore SEBPatch

2025-06-01 11:44:20 +02:00
commit 8c656e3137
1297 changed files with 142172 additions and 0 deletions
--- a/SafeExamBrowser.Runtime/Operations/KioskModeOperation.cs
+++ b/SafeExamBrowser.Runtime/Operations/KioskModeOperation.cs
@@ -0,0 +1,183 @@
+/*
+ * Copyright (c) 2024 ETH Zürich, IT Services
+ *
+ * This Source Code Form is subject to the terms of the Mozilla Public
+ * License, v. 2.0. If a copy of the MPL was not distributed with this
+ * file, You can obtain one at http://mozilla.org/MPL/2.0/.
+ */
+
+using SafeExamBrowser.Core.Contracts.OperationModel;
+using SafeExamBrowser.Core.Contracts.OperationModel.Events;
+using SafeExamBrowser.I18n.Contracts;
+using SafeExamBrowser.Logging.Contracts;
+using SafeExamBrowser.Settings.Security;
+using SafeExamBrowser.WindowsApi.Contracts;
+
+namespace SafeExamBrowser.Runtime.Operations
+{
+	internal class KioskModeOperation : SessionOperation
+	{
+		private readonly IDesktopFactory desktopFactory;
+		private readonly IDesktopMonitor desktopMonitor;
+		private readonly IExplorerShell explorerShell;
+		private readonly ILogger logger;
+		private readonly IProcessFactory processFactory;
+
+		private KioskMode? activeMode;
+		private IDesktop customDesktop;
+		private IDesktop originalDesktop;
+
+		public override event ActionRequiredEventHandler ActionRequired { add { } remove { } }
+		public override event StatusChangedEventHandler StatusChanged;
+
+		public KioskModeOperation(
+			IDesktopFactory desktopFactory,
+			IDesktopMonitor desktopMonitor,
+			IExplorerShell explorerShell,
+			ILogger logger,
+			IProcessFactory processFactory,
+			SessionContext sessionContext) : base(sessionContext)
+		{
+			this.desktopFactory = desktopFactory;
+			this.desktopMonitor = desktopMonitor;
+			this.explorerShell = explorerShell;
+			this.logger = logger;
+			this.processFactory = processFactory;
+		}
+
+		public override OperationResult Perform()
+		{
+			logger.Info($"Initializing kiosk mode '{Context.Next.Settings.Security.KioskMode}'...");
+			/*
+			StatusChanged?.Invoke(TextKey.OperationStatus_InitializeKioskMode);
+
+			activeMode = Context.Next.Settings.Security.KioskMode;
+
+			switch (Context.Next.Settings.Security.KioskMode)
+			{
+				case KioskMode.CreateNewDesktop:
+					CreateCustomDesktop();
+					break;
+				case KioskMode.DisableExplorerShell:
+					TerminateExplorerShell();
+					break;
+			}
+			*/
+			return OperationResult.Success;
+		}
+
+		public override OperationResult Repeat()
+		{
+			var newMode = Context.Next.Settings.Security.KioskMode;
+
+			if (activeMode == newMode)
+			{
+				logger.Info($"New kiosk mode '{newMode}' is the same as the currently active mode, skipping re-initialization...");
+			}
+			else
+			{
+				logger.Info($"Switching from kiosk mode '{activeMode}' to '{newMode}'...");
+				StatusChanged?.Invoke(TextKey.OperationStatus_InitializeKioskMode);
+
+				switch (activeMode)
+				{
+					case KioskMode.CreateNewDesktop:
+						CloseCustomDesktop();
+						break;
+					case KioskMode.DisableExplorerShell:
+						RestartExplorerShell();
+						break;
+				}
+
+				activeMode = newMode;
+
+				switch (newMode)
+				{
+					case KioskMode.CreateNewDesktop:
+						CreateCustomDesktop();
+						break;
+					case KioskMode.DisableExplorerShell:
+						TerminateExplorerShell();
+						break;
+				}
+			}
+
+			return OperationResult.Success;
+		}
+
+		public override OperationResult Revert()
+		{
+			logger.Info($"Reverting kiosk mode '{activeMode}'...");
+			StatusChanged?.Invoke(TextKey.OperationStatus_RevertKioskMode);
+
+			switch (activeMode)
+			{
+				case KioskMode.CreateNewDesktop:
+					CloseCustomDesktop();
+					break;
+				case KioskMode.DisableExplorerShell:
+					RestartExplorerShell();
+					break;
+			}
+
+			return OperationResult.Success;
+		}
+
+		private void CreateCustomDesktop()
+		{
+			originalDesktop = desktopFactory.GetCurrent();
+			logger.Info($"Current desktop is {originalDesktop}.");
+
+			customDesktop = desktopFactory.CreateRandom();
+			logger.Info($"Created custom desktop {customDesktop}.");
+
+			customDesktop.Activate();
+			processFactory.StartupDesktop = customDesktop;
+			logger.Info("Successfully activated custom desktop.");
+
+			desktopMonitor.Start(customDesktop);
+		}
+
+		private void CloseCustomDesktop()
+		{
+			desktopMonitor.Stop();
+
+			if (originalDesktop != default)
+			{
+				originalDesktop.Activate();
+				processFactory.StartupDesktop = originalDesktop;
+				logger.Info($"Switched back to original desktop {originalDesktop}.");
+			}
+			else
+			{
+				logger.Warn($"No original desktop found to activate!");
+			}
+
+			if (customDesktop != default)
+			{
+				customDesktop.Close();
+				logger.Info($"Closed custom desktop {customDesktop}.");
+			}
+			else
+			{
+				logger.Warn($"No custom desktop found to close!");
+			}
+		}
+
+		private void TerminateExplorerShell()
+		{
+			StatusChanged?.Invoke(TextKey.OperationStatus_WaitExplorerTermination);
+
+			explorerShell.HideAllWindows();
+			explorerShell.Terminate();
+		}
+
+		private void RestartExplorerShell()
+		{
+			StatusChanged?.Invoke(TextKey.OperationStatus_WaitExplorerStartup);
+
+			explorerShell.Start();
+			explorerShell.RestoreAllWindows();
+		}
+	}
+}