mirror-ac/service/Worker.cs

94 lines
3 KiB
C#
Raw Normal View History

2023-08-18 07:33:13 +02:00
using System.IO.Pipes;
2023-08-20 16:12:04 +02:00
using System.Runtime.CompilerServices;
2023-08-18 09:18:00 +02:00
using System.Runtime.InteropServices;
2023-08-24 09:29:16 +02:00
using System;
2023-09-02 15:47:15 +02:00
using System.Reflection.PortableExecutable;
2023-09-02 17:56:46 +02:00
using System.Net.Sockets;
using System.Net;
using System.Net.Http;
2023-08-18 10:39:21 +02:00
#pragma warning disable CS1998 // Async method lacks 'await' operators and will run synchronously
#pragma warning disable CS8600
#pragma warning disable CS8603
2023-08-18 07:33:13 +02:00
namespace service
{
public class Worker : BackgroundService
{
private readonly ILogger<Worker> _logger;
private NamedPipeServerStream _pipeServer;
2023-09-02 17:56:46 +02:00
private byte[] _buffer;
private int _bufferSize;
private static int MAX_BUFFER_SIZE = 8192;
2023-08-18 15:22:53 +02:00
2023-08-23 17:16:13 +02:00
private enum MESSAGE_TYPE
{
2023-08-24 09:29:16 +02:00
MESSAGE_TYPE_REPORT = 1,
MESSAGE_TYPE_SEND = 2,
MESSAGE_TYPE_RECEIVE = 3
2023-08-23 17:16:13 +02:00
}
2023-08-18 16:34:15 +02:00
struct PIPE_PACKET_HEADER
{
int message_type;
};
2023-08-18 07:33:13 +02:00
public Worker(ILogger<Worker> logger)
{
_logger = logger;
_pipeServer = new NamedPipeServerStream("DonnaACPipe", PipeDirection.InOut, 1);
2023-09-02 17:56:46 +02:00
_bufferSize = MAX_BUFFER_SIZE;
_buffer = new byte[_bufferSize];
2023-08-18 07:33:13 +02:00
}
protected override async Task ExecuteAsync(CancellationToken stoppingToken)
{
_logger.LogInformation("Windows service starting, waiting for client to connect");
2023-08-18 16:34:15 +02:00
// to do: verify whos connecting
2023-08-18 09:18:00 +02:00
_pipeServer.WaitForConnection();
2023-08-18 07:33:13 +02:00
2023-08-18 09:18:00 +02:00
_logger.LogInformation("Client connected to the pipe server");
2023-08-18 07:33:13 +02:00
2023-08-18 09:18:00 +02:00
while (!stoppingToken.IsCancellationRequested)
2023-09-02 17:56:46 +02:00
{
2023-08-18 07:33:13 +02:00
try
{
2023-09-02 17:56:46 +02:00
int numBytesRead = _pipeServer.Read(_buffer, 0, _bufferSize);
if (numBytesRead > 0)
2023-08-18 07:33:13 +02:00
{
2023-09-02 17:56:46 +02:00
_logger.LogInformation("Message received at pipe server with size: {0}", numBytesRead);
2023-09-02 15:47:15 +02:00
2023-09-02 17:56:46 +02:00
Message message = new Message(_buffer, numBytesRead);
2023-09-02 15:47:15 +02:00
message.SendMessageToServer();
2023-08-18 07:33:13 +02:00
}
}
catch (Exception ex)
{
2023-08-18 09:18:00 +02:00
_logger.LogError("Reading buffer from pipe failed with message: {0}", ex.Message);
2023-08-18 07:33:13 +02:00
}
2023-08-18 15:22:53 +02:00
2023-09-02 17:56:46 +02:00
Array.Clear(_buffer, 0, _bufferSize);
2023-08-18 09:18:00 +02:00
}
}
2023-08-20 16:12:04 +02:00
[DllImport("kernel32.dll", SetLastError = true)]
internal static extern bool GetNamedPipeClientProcessId(IntPtr Pipe, out uint ClientProcessId);
public static uint GetNamedPipeClientProcId(NamedPipeServerStream PipeServer)
{
UInt32 procId;
IntPtr pipeHandle = PipeServer.SafePipeHandle.DangerousGetHandle();
if (GetNamedPipeClientProcessId(pipeHandle, out procId))
return procId;
return 0;
}
2023-08-18 07:33:13 +02:00
}
}
2023-08-18 10:39:21 +02:00
#pragma warning restore CS1998 // Async method lacks 'await' operators and will run synchronously
#pragma warning restore CS8600
#pragma warning restore CS8603