mirror of
https://github.com/gnh1201/welsonjs.git
synced 2025-03-11 16:35:13 +00:00
Sometimes we have to destroy everything to solve a problem. #132
This commit is contained in:
parent
4aace2a3ae
commit
a8ad7746ee
|
|
@ -1,135 +0,0 @@
|
|||
// MessageReceiver.cs
|
||||
// https://github.com/gnh1201/welsonjs
|
||||
using Grpc.Core;
|
||||
using Grpc.Net.Client;
|
||||
using System;
|
||||
using System.Management;
|
||||
using System.Net.Http;
|
||||
using System.ServiceProcess;
|
||||
using System.Threading.Tasks;
|
||||
using WelsonJS.GrpcService;
|
||||
|
||||
namespace WelsonJS.Service
|
||||
{
|
||||
public class MessageReceiver
|
||||
{
|
||||
private GrpcChannel channel;
|
||||
private ServiceMain parent;
|
||||
private string deviceId;
|
||||
private string serverAddress;
|
||||
|
||||
public MessageReceiver(ServiceBase parent, string workingDirectory)
|
||||
{
|
||||
this.parent = (ServiceMain)parent;
|
||||
|
||||
// Read the device ID on this computer
|
||||
deviceId = GetSystemUUID();
|
||||
|
||||
// Read configuration from settings.ini
|
||||
if (!String.IsNullOrEmpty(deviceId))
|
||||
{
|
||||
this.parent.Log($"Resolved the device ID: {deviceId}");
|
||||
|
||||
try
|
||||
{
|
||||
serverAddress = this.parent.GetSettingsFileHandler().Read("GRPC_HOST", "Service");
|
||||
if (String.IsNullOrEmpty(serverAddress))
|
||||
{
|
||||
throw new Exception("The server address could not be empty.");
|
||||
}
|
||||
}
|
||||
catch (Exception ex)
|
||||
{
|
||||
serverAddress = "http://localhost:50051";
|
||||
this.parent.Log($"Failed to read the server address. {ex.Message} Use default value: {serverAddress}");
|
||||
}
|
||||
}
|
||||
else
|
||||
{
|
||||
serverAddress = null;
|
||||
this.parent.Log($"Failed to resolve the device ID");
|
||||
}
|
||||
|
||||
// Set the GRPC channel
|
||||
if (serverAddress != null)
|
||||
{
|
||||
try
|
||||
{
|
||||
HttpClientHandler httpHandler = new HttpClientHandler();
|
||||
|
||||
this.parent.Log($"Use the remote address: {serverAddress}");
|
||||
channel = GrpcChannel.ForAddress(serverAddress, new GrpcChannelOptions
|
||||
{
|
||||
HttpHandler = httpHandler // use HTTP/1.1
|
||||
});
|
||||
}
|
||||
catch (Exception ex)
|
||||
{
|
||||
channel = null;
|
||||
this.parent.Log($"Failed to initialize the GRPC channel: {ex.Message}");
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
public void Start()
|
||||
{
|
||||
if (channel != null)
|
||||
{
|
||||
Task.Run(() => GetTask());
|
||||
parent.Log("GRPC Message Receiver Started");
|
||||
}
|
||||
else
|
||||
{
|
||||
parent.Log("Failed to initalize the GRPC channel");
|
||||
}
|
||||
}
|
||||
|
||||
private async Task GetTask()
|
||||
{
|
||||
parent.Log("Use the device ID: " + deviceId);
|
||||
|
||||
try
|
||||
{
|
||||
var client = new MessageController.MessageControllerClient(channel);
|
||||
|
||||
var request = new MessageRequest {
|
||||
ClientId = deviceId
|
||||
};
|
||||
var call = client.SendMessageStream(request);
|
||||
|
||||
while (await call.ResponseStream.MoveNext())
|
||||
{
|
||||
var response = call.ResponseStream.Current;
|
||||
parent.Log($"< {response.Message}");
|
||||
|
||||
// dispatch to the script runtime
|
||||
parent.DispatchServiceEvent("messageReceived", new string[] { response.Message });
|
||||
}
|
||||
}
|
||||
finally
|
||||
{
|
||||
channel?.Dispose();
|
||||
}
|
||||
}
|
||||
|
||||
private string GetSystemUUID()
|
||||
{
|
||||
try
|
||||
{
|
||||
using (var searcher = new ManagementObjectSearcher("SELECT UUID FROM Win32_ComputerSystemProduct"))
|
||||
{
|
||||
foreach (var mo in searcher.Get())
|
||||
{
|
||||
return mo["UUID"].ToString();
|
||||
}
|
||||
}
|
||||
}
|
||||
catch (Exception ex)
|
||||
{
|
||||
parent.Log($"An error occurred while retrieving the system UUID: {ex.Message}");
|
||||
}
|
||||
|
||||
return null;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
|
@ -1,15 +0,0 @@
|
|||
syntax = "proto3";
|
||||
|
||||
option csharp_namespace = "WelsonJS.GrpcService";
|
||||
|
||||
service MessageController {
|
||||
rpc SendMessageStream (MessageRequest) returns (stream MessageReply);
|
||||
}
|
||||
|
||||
message MessageRequest {
|
||||
string clientId = 1;
|
||||
}
|
||||
|
||||
message MessageReply {
|
||||
string message = 1;
|
||||
}
|
||||
|
|
@ -261,16 +261,6 @@ namespace WelsonJS.Service
|
|||
Log("File Event Monitor is Disabled");
|
||||
}
|
||||
|
||||
// Start GRPC Message Receiver
|
||||
if (!disabledMessageReceiver) {
|
||||
MessageReceiver receiver = new MessageReceiver(this, workingDirectory);
|
||||
receiver.Start();
|
||||
}
|
||||
else
|
||||
{
|
||||
Log("GRPC Message Receiver is Disabled");
|
||||
}
|
||||
|
||||
// Start all the registered timers
|
||||
timers.ForEach(timer => timer?.Start());
|
||||
|
||||
|
|
|
|||
|
|
@ -89,9 +89,6 @@
|
|||
<Reference Include="Grpc.Core, Version=2.0.0.0, Culture=neutral, PublicKeyToken=d754f35622e28bad, processorArchitecture=MSIL">
|
||||
<HintPath>..\packages\Grpc.Core.2.46.6\lib\net45\Grpc.Core.dll</HintPath>
|
||||
</Reference>
|
||||
<Reference Include="Grpc.Core.Api, Version=2.0.0.0, Culture=neutral, PublicKeyToken=d754f35622e28bad, processorArchitecture=MSIL">
|
||||
<HintPath>..\packages\Grpc.Core.Api.2.65.0\lib\net462\Grpc.Core.Api.dll</HintPath>
|
||||
</Reference>
|
||||
<Reference Include="Grpc.Net.Client, Version=2.0.0.0, Culture=neutral, PublicKeyToken=d754f35622e28bad, processorArchitecture=MSIL">
|
||||
<HintPath>..\packages\Grpc.Net.Client.2.65.0\lib\net462\Grpc.Net.Client.dll</HintPath>
|
||||
</Reference>
|
||||
|
|
@ -112,9 +109,6 @@
|
|||
<HintPath>..\packages\RestSharp.111.4.1\lib\net48\RestSharp.dll</HintPath>
|
||||
</Reference>
|
||||
<Reference Include="System" />
|
||||
<Reference Include="System.Buffers, Version=4.0.3.0, Culture=neutral, PublicKeyToken=cc7b13ffcd2ddd51, processorArchitecture=MSIL">
|
||||
<HintPath>..\packages\System.Buffers.4.5.1\lib\net461\System.Buffers.dll</HintPath>
|
||||
</Reference>
|
||||
<Reference Include="System.Configuration.Install" />
|
||||
<Reference Include="System.Data" />
|
||||
<Reference Include="System.Diagnostics.DiagnosticSource, Version=8.0.0.1, Culture=neutral, PublicKeyToken=cc7b13ffcd2ddd51, processorArchitecture=MSIL">
|
||||
|
|
@ -122,20 +116,11 @@
|
|||
</Reference>
|
||||
<Reference Include="System.Drawing" />
|
||||
<Reference Include="System.Management" />
|
||||
<Reference Include="System.Memory, Version=4.0.1.2, Culture=neutral, PublicKeyToken=cc7b13ffcd2ddd51, processorArchitecture=MSIL">
|
||||
<HintPath>..\packages\System.Memory.4.5.5\lib\net461\System.Memory.dll</HintPath>
|
||||
</Reference>
|
||||
<Reference Include="System.Net.Http" />
|
||||
<Reference Include="System.Net.Http.WinHttpHandler, Version=8.0.0.2, Culture=neutral, PublicKeyToken=b03f5f7f11d50a3a, processorArchitecture=MSIL">
|
||||
<HintPath>..\packages\System.Net.Http.WinHttpHandler.8.0.2\lib\net462\System.Net.Http.WinHttpHandler.dll</HintPath>
|
||||
</Reference>
|
||||
<Reference Include="System.Numerics" />
|
||||
<Reference Include="System.Numerics.Vectors, Version=4.1.4.0, Culture=neutral, PublicKeyToken=b03f5f7f11d50a3a, processorArchitecture=MSIL">
|
||||
<HintPath>..\packages\System.Numerics.Vectors.4.5.0\lib\net46\System.Numerics.Vectors.dll</HintPath>
|
||||
</Reference>
|
||||
<Reference Include="System.Runtime.CompilerServices.Unsafe, Version=6.0.0.0, Culture=neutral, PublicKeyToken=b03f5f7f11d50a3a, processorArchitecture=MSIL">
|
||||
<HintPath>..\packages\System.Runtime.CompilerServices.Unsafe.6.0.0\lib\net461\System.Runtime.CompilerServices.Unsafe.dll</HintPath>
|
||||
</Reference>
|
||||
<Reference Include="System.ServiceProcess" />
|
||||
<Reference Include="System.Text.Encodings.Web, Version=8.0.0.0, Culture=neutral, PublicKeyToken=cc7b13ffcd2ddd51, processorArchitecture=MSIL">
|
||||
<HintPath>..\packages\System.Text.Encodings.Web.8.0.0\lib\net462\System.Text.Encodings.Web.dll</HintPath>
|
||||
|
|
@ -155,7 +140,6 @@
|
|||
</ItemGroup>
|
||||
<ItemGroup>
|
||||
<Compile Include="FileEventMonitor.cs" />
|
||||
<Compile Include="MessageReceiver.cs" />
|
||||
<Compile Include="Model\FileMatchResult.cs" />
|
||||
<Compile Include="ServiceMain.cs">
|
||||
<SubType>Component</SubType>
|
||||
|
|
@ -202,9 +186,7 @@
|
|||
</ProjectReference>
|
||||
</ItemGroup>
|
||||
<ItemGroup>
|
||||
<Protobuf Include="Protos\WelsonJS.GrpcService.proto" GrpcServices="Client">
|
||||
<CopyToOutputDirectory>Always</CopyToOutputDirectory>
|
||||
</Protobuf>
|
||||
<Folder Include="Protos\" />
|
||||
</ItemGroup>
|
||||
<Import Project="$(MSBuildToolsPath)\Microsoft.CSharp.targets" />
|
||||
<Target Name="EnsureNuGetPackageBuildImports" BeforeTargets="PrepareForBuild">
|
||||
|
|
|
|||
|
|
@ -3,7 +3,6 @@
|
|||
<package id="Google.Protobuf" version="3.27.3" targetFramework="net48" />
|
||||
<package id="Grpc" version="2.46.6" targetFramework="net48" />
|
||||
<package id="Grpc.Core" version="2.46.6" targetFramework="net48" />
|
||||
<package id="Grpc.Core.Api" version="2.65.0" targetFramework="net48" />
|
||||
<package id="Grpc.Net.Client" version="2.65.0" targetFramework="net48" />
|
||||
<package id="Grpc.Net.Common" version="2.65.0" targetFramework="net48" />
|
||||
<package id="Grpc.Tools" version="2.65.0" targetFramework="net48" developmentDependency="true" />
|
||||
|
|
@ -12,12 +11,8 @@
|
|||
<package id="Microsoft.Extensions.Logging.Abstractions" version="8.0.1" targetFramework="net48" />
|
||||
<package id="Microsoft.O365.Security.Native.libyara.NET" version="4.5.1" targetFramework="net48" />
|
||||
<package id="RestSharp" version="111.4.1" targetFramework="net48" />
|
||||
<package id="System.Buffers" version="4.5.1" targetFramework="net48" />
|
||||
<package id="System.Diagnostics.DiagnosticSource" version="8.0.1" targetFramework="net48" />
|
||||
<package id="System.Memory" version="4.5.5" targetFramework="net48" />
|
||||
<package id="System.Net.Http.WinHttpHandler" version="8.0.2" targetFramework="net48" />
|
||||
<package id="System.Numerics.Vectors" version="4.5.0" targetFramework="net48" />
|
||||
<package id="System.Runtime.CompilerServices.Unsafe" version="6.0.0" targetFramework="net48" />
|
||||
<package id="System.Text.Encodings.Web" version="8.0.0" targetFramework="net48" />
|
||||
<package id="System.Text.Json" version="8.0.4" targetFramework="net48" />
|
||||
<package id="System.Threading.Tasks.Extensions" version="4.5.4" targetFramework="net48" />
|
||||
|
|
|
|||
Loading…
Reference in New Issue
Block a user