Restore SEBPatch
This commit is contained in:
171
SafeExamBrowser.WindowsApi/Processes/Process.cs
Normal file
171
SafeExamBrowser.WindowsApi/Processes/Process.cs
Normal file
@@ -0,0 +1,171 @@
|
||||
/*
|
||||
* 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 System;
|
||||
using System.Text;
|
||||
using SafeExamBrowser.Logging.Contracts;
|
||||
using SafeExamBrowser.WindowsApi.Contracts;
|
||||
using SafeExamBrowser.WindowsApi.Contracts.Events;
|
||||
|
||||
namespace SafeExamBrowser.WindowsApi.Processes
|
||||
{
|
||||
internal class Process : IProcess
|
||||
{
|
||||
private readonly ILogger logger;
|
||||
private readonly System.Diagnostics.Process process;
|
||||
|
||||
private bool eventInitialized;
|
||||
|
||||
public bool HasTerminated
|
||||
{
|
||||
get { return IsTerminated(); }
|
||||
}
|
||||
|
||||
public int Id
|
||||
{
|
||||
get { return process.Id; }
|
||||
}
|
||||
|
||||
public string Name { get; }
|
||||
public string OriginalName { get; }
|
||||
public string Path { get; }
|
||||
public string Signature { get; }
|
||||
|
||||
private event ProcessTerminatedEventHandler TerminatedEvent;
|
||||
|
||||
public event ProcessTerminatedEventHandler Terminated
|
||||
{
|
||||
add { TerminatedEvent += value; InitializeEvent(); }
|
||||
remove { TerminatedEvent -= value; }
|
||||
}
|
||||
|
||||
internal Process(System.Diagnostics.Process process, string name, string originalName, ILogger logger, string path, string signature)
|
||||
{
|
||||
this.logger = logger;
|
||||
this.process = process;
|
||||
this.Name = name;
|
||||
this.OriginalName = originalName;
|
||||
this.Path = path;
|
||||
this.Signature = signature?.ToLower();
|
||||
}
|
||||
|
||||
public string GetAdditionalInfo()
|
||||
{
|
||||
var info = new StringBuilder();
|
||||
|
||||
info.Append($"Original Name: {(string.IsNullOrWhiteSpace(OriginalName) ? "n/a" : $"'{OriginalName}'")}, ");
|
||||
info.Append($"Path: {(string.IsNullOrWhiteSpace(Path) ? "n/a" : $"'{Path}'")}, ");
|
||||
info.Append($"Signature: {(string.IsNullOrWhiteSpace(Signature) ? "n/a" : Signature)}");
|
||||
|
||||
return info.ToString();
|
||||
}
|
||||
|
||||
public bool TryClose(int timeout_ms = 0)
|
||||
{
|
||||
try
|
||||
{
|
||||
logger.Debug("Attempting to close process...");
|
||||
process.Refresh();
|
||||
|
||||
var success = process.CloseMainWindow();
|
||||
|
||||
if (success)
|
||||
{
|
||||
logger.Debug("Successfully sent close message to main window.");
|
||||
}
|
||||
else
|
||||
{
|
||||
logger.Warn("Failed to send close message to main window!");
|
||||
}
|
||||
|
||||
return success && WaitForTermination(timeout_ms);
|
||||
}
|
||||
catch (Exception e)
|
||||
{
|
||||
logger.Error("Failed to close main window!", e);
|
||||
}
|
||||
|
||||
return false;
|
||||
}
|
||||
|
||||
public bool TryKill(int timeout_ms = 0)
|
||||
{
|
||||
try
|
||||
{
|
||||
logger.Debug("Attempting to kill process...");
|
||||
|
||||
process.Refresh();
|
||||
process.Kill();
|
||||
|
||||
return WaitForTermination(timeout_ms);
|
||||
}
|
||||
catch (Exception e)
|
||||
{
|
||||
logger.Error("Failed to kill process!", e);
|
||||
}
|
||||
|
||||
return false;
|
||||
}
|
||||
|
||||
public override string ToString()
|
||||
{
|
||||
return $"'{Name}' ({Id})";
|
||||
}
|
||||
|
||||
private bool IsTerminated()
|
||||
{
|
||||
try
|
||||
{
|
||||
process.Refresh();
|
||||
|
||||
return process.HasExited;
|
||||
}
|
||||
catch (Exception e)
|
||||
{
|
||||
logger.Error("Failed to check whether process is terminated!", e);
|
||||
}
|
||||
|
||||
return false;
|
||||
}
|
||||
|
||||
private void InitializeEvent()
|
||||
{
|
||||
if (!eventInitialized)
|
||||
{
|
||||
eventInitialized = true;
|
||||
|
||||
process.Exited += Process_Exited;
|
||||
process.EnableRaisingEvents = true;
|
||||
|
||||
logger.Debug("Initialized termination event.");
|
||||
}
|
||||
}
|
||||
|
||||
private bool WaitForTermination(int timeout_ms)
|
||||
{
|
||||
var terminated = process.WaitForExit(timeout_ms);
|
||||
|
||||
if (terminated)
|
||||
{
|
||||
logger.Debug($"Process has terminated within {timeout_ms}ms.");
|
||||
}
|
||||
else
|
||||
{
|
||||
logger.Warn($"Process failed to terminate within {timeout_ms}ms!");
|
||||
}
|
||||
|
||||
return terminated;
|
||||
}
|
||||
|
||||
private void Process_Exited(object sender, EventArgs e)
|
||||
{
|
||||
TerminatedEvent?.Invoke(process.ExitCode);
|
||||
logger.Debug("Process has terminated.");
|
||||
}
|
||||
}
|
||||
}
|
||||
243
SafeExamBrowser.WindowsApi/Processes/ProcessFactory.cs
Normal file
243
SafeExamBrowser.WindowsApi/Processes/ProcessFactory.cs
Normal file
@@ -0,0 +1,243 @@
|
||||
/*
|
||||
* 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 System;
|
||||
using System.Collections.Generic;
|
||||
using System.ComponentModel;
|
||||
using System.Diagnostics;
|
||||
using System.IO;
|
||||
using System.Linq;
|
||||
using System.Management;
|
||||
using System.Runtime.InteropServices;
|
||||
using System.Security.Cryptography.X509Certificates;
|
||||
using SafeExamBrowser.Logging.Contracts;
|
||||
using SafeExamBrowser.WindowsApi.Constants;
|
||||
using SafeExamBrowser.WindowsApi.Contracts;
|
||||
using SafeExamBrowser.WindowsApi.Types;
|
||||
|
||||
namespace SafeExamBrowser.WindowsApi.Processes
|
||||
{
|
||||
public class ProcessFactory : IProcessFactory
|
||||
{
|
||||
private readonly IModuleLogger logger;
|
||||
|
||||
public IDesktop StartupDesktop { private get; set; }
|
||||
|
||||
public ProcessFactory(IModuleLogger logger)
|
||||
{
|
||||
this.logger = logger;
|
||||
}
|
||||
|
||||
public IList<IProcess> GetAllRunning()
|
||||
{
|
||||
var processes = new List<IProcess>();
|
||||
var running = System.Diagnostics.Process.GetProcesses();
|
||||
var names = LoadAllProcessNames();
|
||||
|
||||
foreach (var process in running)
|
||||
{
|
||||
if (names.Any(n => n.processId == process.Id))
|
||||
{
|
||||
var (_, name, originalName, path, signature) = names.First(n => n.processId == process.Id);
|
||||
|
||||
processes.Add(new Process(process, name, originalName, LoggerFor(process, name), path, signature));
|
||||
}
|
||||
}
|
||||
|
||||
return processes;
|
||||
}
|
||||
|
||||
public IProcess StartNew(string path, params string[] args)
|
||||
{
|
||||
var raw = default(System.Diagnostics.Process);
|
||||
|
||||
logger.Info($"Attempting to start process '{path}'...");
|
||||
|
||||
if (StartupDesktop != default(IDesktop))
|
||||
{
|
||||
raw = StartOnDesktop(path, args);
|
||||
}
|
||||
else
|
||||
{
|
||||
raw = StartNormal(path, args);
|
||||
}
|
||||
|
||||
var (name, originalName, _, signature) = LoadProcessNamesFor(raw);
|
||||
var process = new Process(raw, name, originalName, LoggerFor(raw, name), path, signature);
|
||||
|
||||
logger.Info($"Successfully started process '{path}' with ID = {process.Id}.");
|
||||
|
||||
return process;
|
||||
}
|
||||
|
||||
public bool TryGetById(int id, out IProcess process)
|
||||
{
|
||||
process = default;
|
||||
|
||||
try
|
||||
{
|
||||
var raw = System.Diagnostics.Process.GetProcessById(id);
|
||||
var (name, originalName, path, signature) = LoadProcessNamesFor(raw);
|
||||
|
||||
process = new Process(raw, name, originalName, LoggerFor(raw, name), path, signature);
|
||||
}
|
||||
catch (Exception e)
|
||||
{
|
||||
logger.Error($"Failed to get process with ID = {id}!", e);
|
||||
}
|
||||
|
||||
return process != default(IProcess);
|
||||
}
|
||||
|
||||
private IEnumerable<(int processId, string name, string originalName, string path, string signature)> LoadAllProcessNames()
|
||||
{
|
||||
var names = new List<(int, string, string, string, string)>();
|
||||
|
||||
try
|
||||
{
|
||||
using (var searcher = new ManagementObjectSearcher($"SELECT ProcessId, Name, ExecutablePath FROM Win32_Process"))
|
||||
using (var results = searcher.Get())
|
||||
{
|
||||
var processData = results.Cast<ManagementObject>().ToList();
|
||||
|
||||
foreach (var process in processData)
|
||||
{
|
||||
using (process)
|
||||
{
|
||||
var name = Convert.ToString(process["Name"]);
|
||||
var originalName = default(string);
|
||||
var path = Convert.ToString(process["ExecutablePath"]);
|
||||
var processId = Convert.ToInt32(process["ProcessId"]);
|
||||
var signature = default(string);
|
||||
|
||||
if (File.Exists(path))
|
||||
{
|
||||
TryLoadOriginalName(path, out originalName);
|
||||
TryLoadSignature(path, out signature);
|
||||
}
|
||||
|
||||
names.Add((processId, name, originalName, path, signature));
|
||||
}
|
||||
|
||||
}
|
||||
}
|
||||
}
|
||||
catch (Exception e)
|
||||
{
|
||||
logger.Error("Failed to load process names!", e);
|
||||
}
|
||||
|
||||
return names;
|
||||
}
|
||||
|
||||
private (string name, string originalName, string path, string signature) LoadProcessNamesFor(System.Diagnostics.Process process)
|
||||
{
|
||||
var name = process.ProcessName;
|
||||
var originalName = default(string);
|
||||
var path = default(string);
|
||||
var signature = default(string);
|
||||
|
||||
try
|
||||
{
|
||||
using (var searcher = new ManagementObjectSearcher($"SELECT Name, ExecutablePath FROM Win32_Process WHERE ProcessId = {process.Id}"))
|
||||
using (var results = searcher.Get())
|
||||
using (var processData = results.Cast<ManagementObject>().First())
|
||||
{
|
||||
name = Convert.ToString(processData["Name"]);
|
||||
path = Convert.ToString(processData["ExecutablePath"]);
|
||||
|
||||
if (File.Exists(path))
|
||||
{
|
||||
TryLoadOriginalName(path, out originalName);
|
||||
TryLoadSignature(path, out signature);
|
||||
}
|
||||
}
|
||||
}
|
||||
catch (Exception e)
|
||||
{
|
||||
logger.Error($"Failed to load process names for {process.ProcessName}!", e);
|
||||
}
|
||||
|
||||
return (name, originalName, path, signature);
|
||||
}
|
||||
|
||||
private ILogger LoggerFor(System.Diagnostics.Process process, string name)
|
||||
{
|
||||
return logger.CloneFor($"{nameof(Process)} '{name}' ({process.Id})");
|
||||
}
|
||||
|
||||
private System.Diagnostics.Process StartNormal(string path, params string[] args)
|
||||
{
|
||||
var process = new System.Diagnostics.Process();
|
||||
|
||||
process.StartInfo.Arguments = string.Join(" ", args);
|
||||
process.StartInfo.FileName = path;
|
||||
process.StartInfo.UseShellExecute = false;
|
||||
process.StartInfo.WorkingDirectory = Environment.GetFolderPath(Environment.SpecialFolder.Windows);
|
||||
process.Start();
|
||||
|
||||
return process;
|
||||
}
|
||||
|
||||
private System.Diagnostics.Process StartOnDesktop(string path, params string[] args)
|
||||
{
|
||||
var commandLine = $"{'"' + path + '"'} {string.Join(" ", args)}";
|
||||
var processInfo = new PROCESS_INFORMATION();
|
||||
var startupInfo = new STARTUPINFO();
|
||||
|
||||
startupInfo.cb = Marshal.SizeOf(startupInfo);
|
||||
startupInfo.lpDesktop = StartupDesktop?.Name;
|
||||
|
||||
var success = Kernel32.CreateProcess(null, commandLine, IntPtr.Zero, IntPtr.Zero, true, Constant.NORMAL_PRIORITY_CLASS, IntPtr.Zero, null, ref startupInfo, ref processInfo);
|
||||
|
||||
if (success)
|
||||
{
|
||||
return System.Diagnostics.Process.GetProcessById(processInfo.dwProcessId);
|
||||
}
|
||||
|
||||
var errorCode = Marshal.GetLastWin32Error();
|
||||
|
||||
logger.Error($"Failed to start process '{path}' on desktop '{StartupDesktop}'! Error code: {errorCode}.");
|
||||
|
||||
throw new Win32Exception(errorCode);
|
||||
}
|
||||
|
||||
private bool TryLoadOriginalName(string path, out string originalName)
|
||||
{
|
||||
originalName = default;
|
||||
|
||||
try
|
||||
{
|
||||
originalName = FileVersionInfo.GetVersionInfo(path).OriginalFilename;
|
||||
}
|
||||
catch
|
||||
{
|
||||
}
|
||||
|
||||
return originalName != default;
|
||||
}
|
||||
|
||||
private bool TryLoadSignature(string path, out string signature)
|
||||
{
|
||||
signature = default;
|
||||
|
||||
try
|
||||
{
|
||||
using (var certificate = X509Certificate.CreateFromSignedFile(path))
|
||||
{
|
||||
signature = certificate.GetCertHashString();
|
||||
}
|
||||
}
|
||||
catch
|
||||
{
|
||||
}
|
||||
|
||||
return signature != default;
|
||||
}
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user