initial commit of a worker skeleton

2025-10-11 04:26:16 +00:00 · 2020-10-20 16:20:38 +03:00
commit 7e63c77419
28 changed files with 665 additions and 0 deletions
--- a/MalwareMultiScan.Backends/Backends/Abstracts/AbstractLocalProcessScanBackend.cs
+++ b/MalwareMultiScan.Backends/Backends/Abstracts/AbstractLocalProcessScanBackend.cs
@@ -0,0 +1,72 @@
+using System;
+using System.Diagnostics;
+using System.IO;
+using System.Linq;
+using System.Text.RegularExpressions;
+using System.Threading;
+using System.Threading.Tasks;
+using Microsoft.Extensions.Logging;
+
+namespace MalwareMultiScan.Backends.Backends.Abstracts
+{
+    public abstract class AbstractLocalProcessScanBackend : AbstractScanBackend
+    {
+        private readonly ILogger _logger;
+        
+        protected virtual Regex MatchRegex { get; }
+        protected virtual string BackendPath { get; }
+        protected virtual bool ParseStdErr { get; }
+
+        protected virtual string GetBackendArguments(string path)
+        {
+            throw new NotImplementedException();
+        }
+
+        protected AbstractLocalProcessScanBackend(ILogger logger)
+        {
+            _logger = logger;
+        }
+
+        public override async Task<string[]> ScanAsync(string path, CancellationToken cancellationToken)
+        {
+            var process = new Process
+            {
+                StartInfo = new ProcessStartInfo(BackendPath, GetBackendArguments(path))
+                {
+                    RedirectStandardError = true,
+                    RedirectStandardOutput = true,
+                    WorkingDirectory = Path.GetDirectoryName(BackendPath) ?? Directory.GetCurrentDirectory()
+                }
+            };
+            
+            _logger.LogInformation(
+                $"Starting process {process.StartInfo.FileName} " +
+                $"with arguments {process.StartInfo.Arguments} " +
+                $"in working directory {process.StartInfo.WorkingDirectory}");
+            
+            process.Start();
+
+            cancellationToken.Register(() =>
+            {
+                if (!process.HasExited)
+                    process.Kill(true);
+            });
+            
+            process.WaitForExit();
+
+            _logger.LogInformation($"Process has exited with code {process.ExitCode}");
+
+            var standardOutput = await process.StandardOutput.ReadToEndAsync();
+            var standardError = await process.StandardError.ReadToEndAsync();
+            
+            _logger.LogDebug($"Process standard output: {standardOutput}");
+            _logger.LogDebug($"Process standard error: {standardError}");
+
+            return MatchRegex
+                .Matches(ParseStdErr ? standardError : standardOutput)
+                .Where(x => x.Success)
+                .Select(x => x.Groups["threat"].Value)
+                .ToArray();
+        }
+    }
+}
--- a/MalwareMultiScan.Backends/Backends/Abstracts/AbstractScanBackend.cs
+++ b/MalwareMultiScan.Backends/Backends/Abstracts/AbstractScanBackend.cs
@@ -0,0 +1,54 @@
+using System;
+using System.IO;
+using System.Net.Http;
+using System.Threading;
+using System.Threading.Tasks;
+using MalwareMultiScan.Shared.Interfaces;
+using Microsoft.AspNetCore.Http;
+
+namespace MalwareMultiScan.Backends.Backends.Abstracts
+{
+    public abstract class AbstractScanBackend : IScanBackend
+    {
+        public virtual string Name => throw new NotImplementedException();
+        
+        public virtual DateTime DatabaseLastUpdate => throw new NotImplementedException();
+        
+        public virtual Task<string[]> ScanAsync(string path, CancellationToken cancellationToken)
+        {
+            throw new NotImplementedException();
+        }
+
+        public async Task<string[]> ScanAsync(Uri uri, CancellationToken cancellationToken)
+        {
+            using var httpClient = new HttpClient();
+            await using var uriStream = await httpClient.GetStreamAsync(uri);
+            
+            return await ScanAsync(uriStream, cancellationToken);
+        }
+
+        public async Task<string[]> ScanAsync(IFormFile file, CancellationToken cancellationToken)
+        {
+            await using var fileStream = file.OpenReadStream();
+            
+            return await ScanAsync(fileStream, cancellationToken);
+        }
+
+        public async Task<string[]> ScanAsync(Stream stream, CancellationToken cancellationToken)
+        {
+            var tempFile = Path.GetTempFileName();
+
+            try
+            {
+                await using (var tempFileStream = File.OpenWrite(tempFile))
+                    await stream.CopyToAsync(tempFileStream, cancellationToken);
+
+                return await ScanAsync(tempFile, cancellationToken);
+            }
+            finally
+            {
+                File.Delete(tempFile);
+            }
+        }
+    }
+}