103 lines
3.5 KiB
C#
103 lines
3.5 KiB
C#
// File: Services/ProcessMonitor.cs
|
|
|
|
using Microsoft.Extensions.Options;
|
|
using System.Runtime.Versioning;
|
|
using WebmrAPI.Configuration;
|
|
using WebmrAPI.Exceptions;
|
|
using WebmrAPI.Models;
|
|
using WebmrAPI.Services.Scanners;
|
|
using WebmrAPI.Utils;
|
|
|
|
namespace WebmrAPI.Services
|
|
{
|
|
[SupportedOSPlatform("windows")]
|
|
public class ProcessMonitor : BackgroundService
|
|
{
|
|
private readonly ILogger<ProcessMonitor> _logger;
|
|
private readonly MonitoringSettings _config;
|
|
private LazyConcurrentContainer<ProcessInfo> _processes = new();
|
|
private ScanProvider _provider;
|
|
|
|
public ILogger<ProcessMonitor> Logger { get => _logger; }
|
|
public MonitoringSettings Config { get => _config; }
|
|
public LazyConcurrentContainer<ProcessInfo> Processes { get => _processes; }
|
|
|
|
public IEnumerable<ProcessBaseInfo>? GetBufferedProcesses()
|
|
{
|
|
return _processes.Values;
|
|
}
|
|
|
|
async public Task<ProcessInfo?> GetProcessDetails(int pid, ScanTarget target = ScanTarget.ProcessDetails)
|
|
{
|
|
if (_processes.Container != null && _processes.Container.TryGetValue(pid, out var info))
|
|
{
|
|
try
|
|
{
|
|
await _provider.CreateScanTask(info, target).ScanAsync();
|
|
_logger.LogInformation($"Scan details of the process {info.Name} (PID: {info.PID}) was completed.");
|
|
return info;
|
|
}
|
|
catch (ProcessMonitorException ex)
|
|
{
|
|
_logger.LogWarning(ex.Message);
|
|
}
|
|
catch (Exception ex)
|
|
{
|
|
_logger.LogError($"Unhandled error during scanning of the process {info.Name} (PID: {info.PID}): {ex.Message}");
|
|
}
|
|
}
|
|
|
|
return null;
|
|
}
|
|
|
|
public ProcessMonitor(ILogger<ProcessMonitor> logger, IOptions<AppSettings> settings)
|
|
{
|
|
_logger = logger;
|
|
_config = settings.Value.Monitoring;
|
|
_provider = new(this);
|
|
}
|
|
|
|
protected override async Task ExecuteAsync(CancellationToken stoppingToken)
|
|
{
|
|
_logger.LogInformation($"ProcessMonitor started. Scan interval: {_config.ProcessScanInterval} seconds.");
|
|
|
|
await ScanAsync();
|
|
|
|
using var timer = new PeriodicTimer(TimeSpan.FromSeconds(_config.ProcessScanInterval));
|
|
try
|
|
{
|
|
while (await timer.WaitForNextTickAsync(stoppingToken))
|
|
{
|
|
await ScanAsync();
|
|
}
|
|
}
|
|
catch (OperationCanceledException)
|
|
{
|
|
_logger.LogInformation("ProcessMonitor is stopping due to cancellation.");
|
|
}
|
|
catch (Exception ex)
|
|
{
|
|
_logger.LogError(ex, "ProcessMonitor encountered an unhandled exception and is stopping.");
|
|
}
|
|
}
|
|
|
|
private async Task ScanAsync()
|
|
{
|
|
_logger.LogDebug("Initiating process scan...");
|
|
try
|
|
{
|
|
await _provider.CreateScanTask().ScanAsync();
|
|
_logger.LogInformation($"Process buffer updated, contains {Processes.Container?.Count} processes.");
|
|
}
|
|
catch (ProcessMonitorException ex)
|
|
{
|
|
_logger.LogWarning(ex.Message);
|
|
}
|
|
catch (Exception ex)
|
|
{
|
|
_logger.LogError($"Unhandled error during process monitoring cycle: {ex.Message}");
|
|
}
|
|
}
|
|
}
|
|
}
|