Initial Commit

Akari.Prototype.Server/Services/GreeterService.cs

@@ -0,0 +1,26 @@
+using Grpc.Core;
+using Microsoft.Extensions.Logging;
+using System;
+using System.Collections.Generic;
+using System.Linq;
+using System.Threading.Tasks;
+
+namespace Akari.Prototype.Server
+{
+    public class GreeterService : Greeter.GreeterBase
+    {
+        private readonly ILogger<GreeterService> _logger;
+        public GreeterService(ILogger<GreeterService> logger)
+        {
+            _logger = logger;
+        }
+
+        public override Task<HelloReply> SayHello(HelloRequest request, ServerCallContext context)
+        {
+            return Task.FromResult(new HelloReply
+            {
+                Message = "Hello " + request.Name
+            });
+        }
+    }
+}

Akari.Prototype.Server/Services/TcpProviderService.cs

@@ -0,0 +1,124 @@
+using Akari.Prototype.Server.Options;
+using Microsoft.Extensions.Hosting;
+using Microsoft.Extensions.Logging;
+using Microsoft.Extensions.Options;
+using System;
+using System.Buffers.Text;
+using System.Collections.Generic;
+using System.IO;
+using System.Linq;
+using System.Net;
+using System.Net.Sockets;
+using System.Threading;
+using System.Threading.Tasks;
+
+namespace Akari.Prototype.Server.Services
+{
+    public class TcpProviderService : BackgroundService
+    {
+        public const int BufferLength = 4096;
+        public const int MaxTokenLength = 192;
+        public const int ReceiveTimeout = 500_000;
+
+        private readonly ILogger<TcpProviderService> _logger;
+        private readonly TcpProviderOptions _options;
+        private readonly IHostApplicationLifetime _hostApplicationLifetime;
+        private Task _backgroundTask;
+        private TcpListener _listener;
+
+        public TcpProviderService(ILogger<TcpProviderService> logger, IOptions<TcpProviderOptions> options, IHostApplicationLifetime hostApplicationLifetime)
+        {
+            _logger = logger;
+            _options = options.Value;
+            _hostApplicationLifetime = hostApplicationLifetime;
+        }
+
+        protected override async Task ExecuteAsync(CancellationToken stoppingToken)
+        {
+            await TcpLoop(stoppingToken);
+        }
+
+        private async Task TcpLoop(CancellationToken cancellationToken)
+        {
+            if (cancellationToken.IsCancellationRequested)
+            {
+                return;
+            }
+
+            try
+            {
+                _listener = new TcpListener(IPAddress.Parse(_options.ListeningAddress), _options.Port);
+                _listener.Start();
+
+                _logger.LogInformation($"Now listening on: {_listener.LocalEndpoint}");
+
+                while (!cancellationToken.IsCancellationRequested)
+                {
+                    _logger.LogDebug("Waiting for client...");
+                    var task = await Task.WhenAny(_listener.AcceptTcpClientAsync(), Task.Delay(Timeout.Infinite, cancellationToken));
+
+                    if (task is not Task<TcpClient> clientTask)
+                    {
+                        return;
+                    }
+
+                    using var client = clientTask.Result;
+
+                    _logger.LogDebug($"Client connected from: {client.Client.RemoteEndPoint}");
+
+                    try
+                    {
+                        await ReceiveAuth(client, cancellationToken);
+                    }
+                    catch (Exception e)
+                    {
+                        _logger.LogDebug($"Couldn't proccess client request: {e.Message}");
+                    }
+                }
+            }
+            catch (Exception e)
+            {
+                _logger.LogError(e.ToString());
+                _logger.LogError("An error occured in TcpProvider, exiting...");
+
+                _hostApplicationLifetime.StopApplication();
+
+                Environment.Exit(1);
+            }
+            finally
+            {
+                _listener?.Stop();
+            }
+        }
+
+        private async Task ReceiveAuth(TcpClient client, CancellationToken cancellationToken)
+        {
+            using var stream = client.GetStream();
+            var data = new Memory<byte>(new byte[MaxTokenLength]);
+            var buffer = new Memory<byte>(new byte[BufferLength]);
+
+            // Receive token
+            int position = 0;
+            int read = 0;
+
+            var timeoutToken = CancellationTokenSource.CreateLinkedTokenSource(cancellationToken, new CancellationTokenSource(ReceiveTimeout).Token).Token;
+
+            _logger.LogDebug("Waiting for token...");
+            while (position < MaxTokenLength && (read = await stream.ReadAsync(buffer, timeoutToken)) != 0)
+            {
+                if (position + read > data.Length)
+                {
+                    // Invalid token
+                    return;
+                }
+
+                buffer[..read].CopyTo(data[position..]);
+
+                position += read;
+            }
+
+            // TODO: Send token to AuthManager that will compare it to stored hash using Argon2
+            _logger.LogDebug($"Received token: {Convert.ToBase64String(data[..position].Span)}");
+        }
+    }
+}