Improve code structure

- Add separate API project
- Add notification system
This commit is contained in:
Stone_Red
2022-02-24 22:56:45 +01:00
parent 6ffdc720a4
commit cf7f952012
28 changed files with 590 additions and 488 deletions
+4
View File
@@ -0,0 +1,4 @@
[*.cs]
# S4457: Parameter validation in "async"/"await" methods should be wrapped
dotnet_diagnostic.S4457.severity = silent
+13 -2
View File
@@ -3,9 +3,16 @@ Microsoft Visual Studio Solution File, Format Version 12.00
# Visual Studio Version 17 # Visual Studio Version 17
VisualStudioVersion = 17.0.32112.339 VisualStudioVersion = 17.0.32112.339
MinimumVisualStudioVersion = 10.0.40219.1 MinimumVisualStudioVersion = 10.0.40219.1
Project("{9A19103F-16F7-4668-BE54-9A1E7A4F7556}") = "HyperbolicDownloader", "HyperbolicDownloader\HyperbolicDownloader.csproj", "{7BA90CAF-36A1-4D55-9CE2-E498ECC1B62D}" Project("{9A19103F-16F7-4668-BE54-9A1E7A4F7556}") = "HyperbolicDownloaderCli", "HyperbolicDownloader\HyperbolicDownloaderCli.csproj", "{7BA90CAF-36A1-4D55-9CE2-E498ECC1B62D}"
EndProject EndProject
Project("{FAE04EC0-301F-11D3-BF4B-00C04F79EFBC}") = "HyperbolicDownloaderGUI", "HyperbolicDownloaderGUI\HyperbolicDownloaderGUI.csproj", "{E1CE6E07-F8A6-447F-837D-9D1943A42E96}" Project("{9A19103F-16F7-4668-BE54-9A1E7A4F7556}") = "HyperbolicDownloaderGUI", "HyperbolicDownloaderGUI\HyperbolicDownloaderGUI.csproj", "{E1CE6E07-F8A6-447F-837D-9D1943A42E96}"
EndProject
Project("{FAE04EC0-301F-11D3-BF4B-00C04F79EFBC}") = "HyperbolicDownloaderApi", "HyperbolicDownloaderApi\HyperbolicDownloaderApi.csproj", "{45ACEEC6-9AE4-4DA5-9A08-E22F1EBA7E22}"
EndProject
Project("{2150E333-8FDC-42A3-9474-1A3956D46DE8}") = "Solution Items", "Solution Items", "{CA4F21C6-5B63-46C1-8576-91BB3FAE4FA7}"
ProjectSection(SolutionItems) = preProject
.editorconfig = .editorconfig
EndProjectSection
EndProject EndProject
Global Global
GlobalSection(SolutionConfigurationPlatforms) = preSolution GlobalSection(SolutionConfigurationPlatforms) = preSolution
@@ -21,6 +28,10 @@ Global
{E1CE6E07-F8A6-447F-837D-9D1943A42E96}.Debug|Any CPU.Build.0 = Debug|Any CPU {E1CE6E07-F8A6-447F-837D-9D1943A42E96}.Debug|Any CPU.Build.0 = Debug|Any CPU
{E1CE6E07-F8A6-447F-837D-9D1943A42E96}.Release|Any CPU.ActiveCfg = Release|Any CPU {E1CE6E07-F8A6-447F-837D-9D1943A42E96}.Release|Any CPU.ActiveCfg = Release|Any CPU
{E1CE6E07-F8A6-447F-837D-9D1943A42E96}.Release|Any CPU.Build.0 = Release|Any CPU {E1CE6E07-F8A6-447F-837D-9D1943A42E96}.Release|Any CPU.Build.0 = Release|Any CPU
{45ACEEC6-9AE4-4DA5-9A08-E22F1EBA7E22}.Debug|Any CPU.ActiveCfg = Debug|Any CPU
{45ACEEC6-9AE4-4DA5-9A08-E22F1EBA7E22}.Debug|Any CPU.Build.0 = Debug|Any CPU
{45ACEEC6-9AE4-4DA5-9A08-E22F1EBA7E22}.Release|Any CPU.ActiveCfg = Release|Any CPU
{45ACEEC6-9AE4-4DA5-9A08-E22F1EBA7E22}.Release|Any CPU.Build.0 = Release|Any CPU
EndGlobalSection EndGlobalSection
GlobalSection(SolutionProperties) = preSolution GlobalSection(SolutionProperties) = preSolution
HideSolutionNode = FALSE HideSolutionNode = FALSE
@@ -7,10 +7,17 @@
<Nullable>enable</Nullable> <Nullable>enable</Nullable>
</PropertyGroup> </PropertyGroup>
<ItemGroup>
<None Include="..\.editorconfig" Link=".editorconfig" />
</ItemGroup>
<ItemGroup> <ItemGroup>
<PackageReference Include="Console.Commander.NET" Version="1.0.0" /> <PackageReference Include="Console.Commander.NET" Version="1.0.0" />
<PackageReference Include="Open.NAT" Version="2.1.0" />
<PackageReference Include="Stone_Red-C-Sharp-Utilities" Version="1.0.3.1" /> <PackageReference Include="Stone_Red-C-Sharp-Utilities" Version="1.0.3.1" />
</ItemGroup> </ItemGroup>
<ItemGroup>
<ProjectReference Include="..\HyperbolicDownloaderApi\HyperbolicDownloaderApi.csproj" />
</ItemGroup>
</Project> </Project>
+36 -213
View File
@@ -1,61 +1,42 @@
using HyperbolicDownloader.FileProcessing; using HyperbolicDownloaderApi.FileProcessing;
using HyperbolicDownloader.Networking; using HyperbolicDownloaderApi.Managment;
using HyperbolicDownloader.UserInterface; using HyperbolicDownloaderApi.Networking;
using HyperbolicDownloaderApi.UserInterface;
using Open.Nat;
using Stone_Red_Utilities.ConsoleExtentions; using Stone_Red_Utilities.ConsoleExtentions;
using System.Diagnostics;
using System.Net;
using System.Net.Sockets;
using System.Reflection;
using System.Text.Json; using System.Text.Json;
namespace HyperbolicDownloader; namespace HyperbolicDownloaderApi;
internal static class Program internal static class Program
{ {
public const int BroadcastPort = 2155; private static readonly ApiManager apiManager = new ApiManager();
public static string BasePath { get; } = Path.GetDirectoryName(Assembly.GetEntryAssembly()!.Location) ?? string.Empty;
public static string HostsFilePath { get; } = Path.Combine(BasePath, "Hosts.json");
public static string FilesInfoPath { get; } = Path.Combine(BasePath, "Files.json");
public static int PublicPort { get; private set; }
public static int PrivatePort { get; } = 3055;
public static IPAddress? PublicIpAddress => device?.GetExternalIPAsync().GetAwaiter().GetResult();
private static NatDevice? device;
private static Mapping? portMapping;
private static readonly HostsManager hostsManager = new();
private static readonly FilesManager filesManager = new();
private static readonly NetworkClient networkClient = new(filesManager);
private static readonly Random random = new();
private static async Task Main(string[] args) private static async Task Main(string[] args)
{ {
Console.CancelKeyPress += Console_CancelKeyPress; Console.CancelKeyPress += Console_CancelKeyPress;
Console.CursorVisible = false; Console.CursorVisible = false;
if (File.Exists(HostsFilePath)) ApiManager.OnNotificationMessageRecived += ApiManager_OnNotificationMessageRecived;
if (File.Exists(ApiConfiguration.HostsFilePath))
{ {
string hostsJson = await File.ReadAllTextAsync(HostsFilePath); string hostsJson = await File.ReadAllTextAsync(ApiConfiguration.HostsFilePath);
hostsManager.AddRange(JsonSerializer.Deserialize<List<NetworkSocket>>(hostsJson) ?? new()); apiManager.HostsManager.AddRange(JsonSerializer.Deserialize<List<NetworkSocket>>(hostsJson) ?? new());
} }
if (File.Exists(FilesInfoPath)) if (File.Exists(ApiConfiguration.FilesInfoPath))
{ {
string filesJson = await File.ReadAllTextAsync(FilesInfoPath); string filesJson = await File.ReadAllTextAsync(ApiConfiguration.FilesInfoPath);
filesManager.AddRange(JsonSerializer.Deserialize<List<PrivateHyperFileInfo>>(filesJson) ?? new()); apiManager.FilesManager.AddRange(JsonSerializer.Deserialize<List<PrivateHyperFileInfo>>(filesJson) ?? new());
} }
InputHandler inputHandler = new InputHandler(hostsManager, filesManager); InputHandler inputHandler = new InputHandler(apiManager.HostsManager, apiManager.FilesManager);
if (args.Length > 0 && File.Exists(args[0])) if (args.Length > 0 && File.Exists(args[0]))
{ {
UserInterface.Commands.DownloadCommands downloadCommands = new UserInterface.Commands.DownloadCommands(hostsManager, filesManager); Commands.DownloadCommands downloadCommands = new Commands.DownloadCommands(apiManager.HostsManager, apiManager.FilesManager);
downloadCommands.GetFileFrom(args[0]); downloadCommands.GetFileFrom(args[0]);
Console.WriteLine("Do you want to continue using this instance? [y/N]"); Console.WriteLine("Do you want to continue using this instance? [y/N]");
if (char.ToLower(Console.ReadKey().KeyChar) != 'y') if (char.ToLower(Console.ReadKey().KeyChar) != 'y')
@@ -66,211 +47,53 @@ internal static class Program
} }
Console.WriteLine("Searching for a UPnP/NAT-PMP device..."); Console.WriteLine("Searching for a UPnP/NAT-PMP device...");
_ = await OpenPorts(); _ = await ApiManager.OpenPorts();
ConsoleExt.WriteLine($"The private IP address is: {NetworkUtilities.GetIP4Adress()} ", ConsoleColor.Green); ConsoleExt.WriteLine($"The private IP address is: {NetworkUtilities.GetIP4Adress()} ", ConsoleColor.Green);
ConsoleExt.WriteLine($"The private port is: {PrivatePort}", ConsoleColor.Green); ConsoleExt.WriteLine($"The private port is: {ApiConfiguration.PrivatePort}", ConsoleColor.Green);
Console.WriteLine("Starting TCP listener..."); Console.WriteLine("Starting TCP listener...");
apiManager.StartTcpListener();
try Console.WriteLine("Starting broadcast listener...");
{ apiManager.StartBroadcastListener();
networkClient.ListenTo<NetworkSocket>("GetHostsList", GetHostList);
networkClient.ListenTo<List<NetworkSocket>>("DiscoverAnswer", DiscoverAnswer);
networkClient.ListenTo<string>("Message", ReciveMessage);
networkClient.ListenTo<string>("HasFile", HasFile);
networkClient.StartListening(PrivatePort);
}
catch (SocketException ex)
{
ConsoleExt.WriteLine($"An error occurred while starting the TCP listener! Error message: {ex.Message}", ConsoleColor.Red); // net stop hns && net start hns
Console.ReadKey();
return;
}
BroadcastClient broadcastClient = new BroadcastClient();
Console.WriteLine("Running local discovery routine...");
BroadcastClient.Send(BroadcastPort, PrivatePort.ToString());
await Task.Delay(3000);
int activeHostsCount = 0; int activeHostsCount = 0;
if (hostsManager.Count > 0) if (apiManager.HostsManager.Count > 0)
{ {
Console.WriteLine("Checking if hosts are active..."); Console.WriteLine("Checking if hosts are active...");
activeHostsCount = hostsManager.CheckHostsActivity(); activeHostsCount = apiManager.HostsManager.CheckHostsActivity();
} }
if (activeHostsCount == 0) if (activeHostsCount == 0)
{ {
ConsoleExt.WriteLine("No active hosts found!", ConsoleColor.Red); ConsoleExt.WriteLine("No active hosts found!", ConsoleColor.Red);
ConsoleExt.WriteLine("Use 'add host xxx.xxx.xxx.xxx:yyyy' to add a new host.", ConsoleColor.Red); ConsoleExt.WriteLine("Use 'add host xxx.xxx.xxx.xxx:yyyy' to add a new host or use 'discover' to find hosts in the local network.", ConsoleColor.Red);
} }
Console.WriteLine($"{hostsManager.Count} known host(s)."); Console.WriteLine($"{apiManager.HostsManager.Count} known host(s).");
Console.WriteLine($"{activeHostsCount} active host(s)."); Console.WriteLine($"{activeHostsCount} active host(s).");
Console.WriteLine("Starting broadcast listener...");
broadcastClient.StartListening(BroadcastPort); ConsoleExt.WriteLine("Ready", ConsoleColor.Green);
broadcastClient.OnBroadcastRecived += BroadcastClient_OnBroadcastRecived;
ConsoleExt.WriteLine("Done", ConsoleColor.Green);
inputHandler.ReadInput(); inputHandler.ReadInput();
ClosePorts(); ApiManager.ClosePorts();
} }
private static async void BroadcastClient_OnBroadcastRecived(object? sender, BroadcastRecivedEventArgs recivedEventArgs) private static void ApiManager_OnNotificationMessageRecived(object? sender, NotificationMessageEventArgs e)
{ {
Debug.WriteLine($"Received broadcast \"{recivedEventArgs.Message}\" from {recivedEventArgs.IPEndPoint.Address}"); switch (e.NotificationMessageType)
List<NetworkSocket> hostsToSend = hostsManager.ToList();
NetworkSocket? localSocket = GetLocalSocket();
if (localSocket is null)
{ {
return; case NotificationMessageType.Raw: Console.Write(e.Message); break;
case NotificationMessageType.Success: ConsoleExt.Write(e.Message, ConsoleColor.Green); break;
case NotificationMessageType.Warning: ConsoleExt.Write(e.Message, ConsoleColor.DarkYellow); break;
case NotificationMessageType.Error: ConsoleExt.Write(e.Message, ConsoleColor.Red); break;
} }
hostsToSend.RemoveAll(x => x.IPAddress == recivedEventArgs.IPEndPoint.Address.ToString());
hostsToSend.Add(localSocket);
bool success = int.TryParse(recivedEventArgs.Message, out int remotePort);
if (success)
{
hostsManager.Add(new NetworkSocket(recivedEventArgs.IPEndPoint.Address.ToString(), remotePort, DateTime.Now));
try
{
await NetworkClient.SendAsync(recivedEventArgs.IPEndPoint.Address, remotePort, "DiscoverAnswer", hostsToSend);
}
catch (SocketException ex)
{
Debug.WriteLine(ex);
}
}
}
private static void DiscoverAnswer(object? sender, MessageRecivedEventArgs<List<NetworkSocket>> recivedEventArgs)
{
Console.WriteLine($"Received answer from {recivedEventArgs.IpAddress}. Returned {recivedEventArgs.Data.Count} host(s).");
hostsManager.AddRange(recivedEventArgs.Data);
}
private static void ReciveMessage(object? sender, MessageRecivedEventArgs<string> recivedEventArgs)
{
Console.WriteLine($"Received \"{recivedEventArgs.Data}\" from {recivedEventArgs.IpAddress}.");
}
private static async void GetHostList(object? sender, MessageRecivedEventArgs<NetworkSocket> recivedEventArgs)
{
List<NetworkSocket> hostsToSend = hostsManager.ToList();
NetworkSocket? localSocket = GetLocalSocket();
if (localSocket is null)
{
return;
}
hostsToSend.RemoveAll(x => x.IPAddress == recivedEventArgs.IpAddress.ToString());
hostsToSend.Add(localSocket);
if (recivedEventArgs.Data.Port != 0)
{
hostsManager.Add(recivedEventArgs.Data);
}
await recivedEventArgs.SendResponseAsync(hostsToSend);
}
private static async void HasFile(object? sender, MessageRecivedEventArgs<string> recivedEventArgs)
{
await recivedEventArgs.SendResponseAsync(filesManager.Contains(recivedEventArgs.Data));
}
public static async Task<bool> OpenPorts()
{
try
{
PublicPort = random.Next(1000, 6000);
NatDiscoverer? discoverer = new NatDiscoverer();
device = await discoverer.DiscoverDeviceAsync();
IPAddress? ip = await device.GetExternalIPAsync();
ConsoleExt.WriteLine($"The public IP address is: {ip} ", ConsoleColor.Green);
portMapping = new Mapping(Protocol.Tcp, PrivatePort, PublicPort, "HyperbolicDowloader");
await device.CreatePortMapAsync(portMapping);
ConsoleExt.WriteLine($"The public port is: {PublicPort}", ConsoleColor.Green);
return true;
}
catch (NatDeviceNotFoundException)
{
ConsoleExt.WriteLine($"Could not find a UPnP or NAT-PMP device!", ConsoleColor.Red);
return false;
}
catch (MappingException ex)
{
ConsoleExt.WriteLine($"An error occurred while mapping the private port ({PrivatePort}) to the public port ({PublicPort})! Error message: {ex.Message}", ConsoleColor.Red);
return false;
}
}
private static void ClosePorts()
{
Console.WriteLine("Closing ports...");
if (device is not null)
{
try
{
IEnumerable<Mapping>? mappings = device.GetAllMappingsAsync().GetAwaiter().GetResult();
foreach (Mapping? mapping in mappings)
{
if (mapping.Description.Contains("HyperbolicDowloader") && mapping.PrivateIP.ToString() == portMapping?.PrivateIP.ToString())
{
device.DeletePortMapAsync(mapping).GetAwaiter().GetResult();
}
}
}
catch (Exception ex)
{
Console.WriteLine(ex);
}
}
ConsoleExt.WriteLine("Ports closed!", ConsoleColor.DarkYellow);
Environment.Exit(0);
} }
private static void Console_CancelKeyPress(object? sender, ConsoleCancelEventArgs e) private static void Console_CancelKeyPress(object? sender, ConsoleCancelEventArgs e)
{ {
ClosePorts(); ApiManager.ClosePorts();
hostsManager.SaveHosts(); apiManager.HostsManager.SaveHosts();
}
public static NetworkSocket? GetLocalSocket()
{
int port = PublicPort;
string? ipAddress = null;
if (device is not null)
{
ipAddress = (device.GetExternalIPAsync()).GetAwaiter().GetResult()?.ToString();
}
if (ipAddress is null || ipAddress == "0.0.0.0")
{
ipAddress = NetworkUtilities.GetIP4Adress()?.ToString();
port = PrivatePort;
}
if (ipAddress is null)
{
return null;
}
return new NetworkSocket(ipAddress, port, DateTime.Now);
} }
} }
@@ -1,21 +0,0 @@
using HyperbolicDownloader.Networking;
using Stone_Red_Utilities.ConsoleExtentions;
namespace HyperbolicDownloader.UserInterface.Commands;
internal class ClientCommands
{
public void ShowInfo(string _)
{
if (Program.PublicIpAddress is not null)
{
ConsoleExt.WriteLine($"The public IP address is: {Program.PublicIpAddress}", ConsoleColor.Green);
ConsoleExt.WriteLine($"The public port is: {Program.PublicPort}", ConsoleColor.Green);
Console.WriteLine();
}
ConsoleExt.WriteLine($"The private IP address is: {NetworkUtilities.GetIP4Adress()}", ConsoleColor.Green);
ConsoleExt.WriteLine($"The private port is: {Program.PrivatePort}", ConsoleColor.Green);
}
}
@@ -1,149 +0,0 @@
using HyperbolicDownloader.Networking;
using Stone_Red_Utilities.ConsoleExtentions;
using System.Net;
using System.Net.Sockets;
namespace HyperbolicDownloader.UserInterface.Commands;
internal class HostCommands
{
private readonly HostsManager hostsManager;
public HostCommands(HostsManager hostsManager)
{
this.hostsManager = hostsManager;
}
public void Discover(string _)
{
Console.WriteLine("Running local discovery routine...");
BroadcastClient.Send(Program.BroadcastPort, Program.PrivatePort.ToString());
Thread.Sleep(3000);
}
public void CheckActiveHosts(string _)
{
int activeHostsCount = hostsManager.CheckHostsActivity();
if (activeHostsCount == 0)
{
ConsoleExt.WriteLine("No active hosts found!", ConsoleColor.Red);
ConsoleExt.WriteLine("Use 'add host xxx.xxx.xxx.xxx:yyyy' to add a new host.", ConsoleColor.Red);
}
Console.WriteLine();
Console.WriteLine($"{hostsManager.Count} known host(s).");
Console.WriteLine($"{activeHostsCount} active host(s).");
}
public void ListHosts(string _)
{
int index = 0;
List<NetworkSocket> hosts = hostsManager.ToList();
if (hosts.Count == 0)
{
ConsoleExt.WriteLine("No known hosts", ConsoleColor.DarkYellow);
return;
}
foreach (NetworkSocket host in hosts)
{
index++;
Console.WriteLine($"{index}) {host.IPAddress}:{host.Port}");
Console.WriteLine($"Last active: {host.LastActive}");
Console.WriteLine();
}
Console.CursorTop--;
}
public void RemoveHost(string args)
{
string[] parts = args.Split(":");
if (parts.Length != 2)
{
ConsoleExt.WriteLine("Invalid format! Use this format: (xxx.xxx.xxx.xxx:yyyy)", ConsoleColor.Red);
return;
}
string ipAddressInput = parts[0];
string portInput = parts[1];
_ = int.TryParse(portInput, out int port);
if (port < 1000 || port >= 6000)
{
ConsoleExt.WriteLine("Invalid port number!", ConsoleColor.Red);
return;
}
if (!IPAddress.TryParse(ipAddressInput, out IPAddress? ipAddress))
{
ConsoleExt.WriteLine("Invalid IP address!", ConsoleColor.Red);
return;
}
NetworkSocket hostToRemove = new NetworkSocket(ipAddress.ToString(), port, DateTime.MinValue);
if (!hostsManager.Contains(hostToRemove))
{
ConsoleExt.WriteLine("Host not in list", ConsoleColor.Red);
return;
}
hostsManager.Remove(new NetworkSocket(ipAddress.ToString(), port, DateTime.MinValue), true);
ConsoleExt.WriteLine($"Successfully Removed host!", ConsoleColor.Green);
}
public void AddHost(string args)
{
string[] parts = args.Split(":");
if (parts.Length != 2)
{
ConsoleExt.WriteLine("Invalid format! Use this format: (xxx.xxx.xxx.xxx:yyyy)", ConsoleColor.Red);
return;
}
string ipAddressInput = parts[0];
string portInput = parts[1];
_ = int.TryParse(portInput, out int port);
if (port < 1000 || port >= 6000)
{
ConsoleExt.WriteLine("Invalid port number!", ConsoleColor.Red);
}
else if (IPAddress.TryParse(ipAddressInput, out IPAddress? ipAddress))
{
try
{
Console.WriteLine("Waiting for response...");
NetworkSocket? localSocket = Program.GetLocalSocket() ?? new NetworkSocket("0.0.0.0", 0, DateTime.MinValue);
List<NetworkSocket>? recivedHosts = NetworkClient.Send<List<NetworkSocket>>(ipAddress, port, "GetHostsList", localSocket);
if (recivedHosts is not null)
{
ConsoleExt.WriteLine($"Success! Added {recivedHosts.Count} new host(s).", ConsoleColor.Green);
hostsManager.AddRange(recivedHosts);
}
else
{
ConsoleExt.WriteLine($"Invalid response!", ConsoleColor.Red);
}
}
catch (SocketException ex)
{
ConsoleExt.WriteLine($"Invalid host! Error message: {ex.Message}", ConsoleColor.Red);
}
catch (IOException ex)
{
ConsoleExt.WriteLine($"Invalid host! Error message: {ex.Message}", ConsoleColor.Red);
}
}
else
{
ConsoleExt.WriteLine("Invalid IP address!", ConsoleColor.Red);
}
}
}
@@ -1,11 +1,11 @@
using Commander_Net; using Commander_Net;
using HyperbolicDownloader.FileProcessing; using HyperbolicDownloaderApi.Commands;
using HyperbolicDownloader.UserInterface.Commands; using HyperbolicDownloaderApi.FileProcessing;
using Stone_Red_Utilities.ConsoleExtentions; using Stone_Red_Utilities.ConsoleExtentions;
namespace HyperbolicDownloader.UserInterface; namespace HyperbolicDownloaderApi.UserInterface;
internal class InputHandler internal class InputHandler
{ {
@@ -0,0 +1,20 @@
using HyperbolicDownloaderApi.Managment;
using HyperbolicDownloaderApi.Networking;
namespace HyperbolicDownloaderApi.Commands;
public class ClientCommands
{
public void ShowInfo(string _)
{
if (ApiManager.PublicIpAddress is not null)
{
ApiManager.SendMessageNewLine($"The public IP address is: {ApiManager.PublicIpAddress}", NotificationMessageType.Success);
ApiManager.SendMessageNewLine($"The public port is: {ApiConfiguration.PublicPort}", NotificationMessageType.Success);
ApiManager.SendMessageNewLine(string.Empty, NotificationMessageType.Raw);
}
ApiManager.SendMessageNewLine($"The private IP address is: {NetworkUtilities.GetIP4Adress()}", NotificationMessageType.Success);
ApiManager.SendMessageNewLine($"The private port is: {ApiConfiguration.PrivatePort}", NotificationMessageType.Success);
}
}
@@ -1,7 +1,7 @@
using HyperbolicDownloader.FileProcessing; using HyperbolicDownloaderApi.FileProcessing;
using HyperbolicDownloader.Networking; using HyperbolicDownloaderApi.Managment;
using HyperbolicDownloaderApi.Networking;
using Stone_Red_Utilities.ConsoleExtentions;
using Stone_Red_Utilities.StringExtentions; using Stone_Red_Utilities.StringExtentions;
using System.Diagnostics; using System.Diagnostics;
@@ -10,9 +10,9 @@ using System.Net.Sockets;
using System.Text; using System.Text;
using System.Text.Json; using System.Text.Json;
namespace HyperbolicDownloader.UserInterface.Commands; namespace HyperbolicDownloaderApi.Commands;
internal class DownloadCommands public class DownloadCommands
{ {
private readonly HostsManager hostsManager; private readonly HostsManager hostsManager;
private readonly FilesManager filesManager; private readonly FilesManager filesManager;
@@ -27,7 +27,7 @@ internal class DownloadCommands
{ {
if (string.IsNullOrWhiteSpace(path)) if (string.IsNullOrWhiteSpace(path))
{ {
ConsoleExt.WriteLine("Path is empty!", ConsoleColor.Red); ApiManager.SendMessageNewLine("Path is empty!", NotificationMessageType.Error);
return; return;
} }
@@ -35,7 +35,7 @@ internal class DownloadCommands
if (!File.Exists(fullPath)) if (!File.Exists(fullPath))
{ {
ConsoleExt.WriteLine("Invalid file path!", ConsoleColor.Red); ApiManager.SendMessageNewLine("Invalid file path!", NotificationMessageType.Error);
} }
string json = File.ReadAllText(fullPath); string json = File.ReadAllText(fullPath);
@@ -43,7 +43,7 @@ internal class DownloadCommands
PublicHyperFileInfo? publicHyperFileInfo = JsonSerializer.Deserialize<PublicHyperFileInfo>(json); PublicHyperFileInfo? publicHyperFileInfo = JsonSerializer.Deserialize<PublicHyperFileInfo>(json);
if (publicHyperFileInfo == null) if (publicHyperFileInfo == null)
{ {
ConsoleExt.WriteLine("Parsing file failed!", ConsoleColor.Red); ApiManager.SendMessageNewLine("Parsing file failed!", NotificationMessageType.Error);
return; return;
} }
@@ -55,7 +55,7 @@ internal class DownloadCommands
{ {
if (string.IsNullOrEmpty(hash)) if (string.IsNullOrEmpty(hash))
{ {
ConsoleExt.WriteLine("No hash value specified!", ConsoleColor.Red); ApiManager.SendMessageNewLine("No hash value specified!", NotificationMessageType.Error);
return; return;
} }
@@ -71,7 +71,7 @@ internal class DownloadCommands
continue; continue;
} }
ConsoleExt.Write($"{host.IPAddress}:{host.Port} > ???", ConsoleColor.DarkYellow); ApiManager.SendMessage($"{host.IPAddress}:{host.Port} > ???", NotificationMessageType.Warning);
Console.CursorLeft = 0; Console.CursorLeft = 0;
@@ -82,7 +82,7 @@ internal class DownloadCommands
if (!sendTask.IsCompletedSuccessfully) if (!sendTask.IsCompletedSuccessfully)
{ {
Console.CursorLeft = 0; Console.CursorLeft = 0;
ConsoleExt.WriteLine($"{host.IPAddress}:{host.Port} > Inactive", ConsoleColor.Red); ApiManager.SendMessageNewLine($"{host.IPAddress}:{host.Port} > Inactive", NotificationMessageType.Error);
hostsManager.Remove(host); hostsManager.Remove(host);
continue; continue;
@@ -90,14 +90,14 @@ internal class DownloadCommands
else if (!sendTask.Result) else if (!sendTask.Result)
{ {
host.LastActive = DateTime.Now; host.LastActive = DateTime.Now;
ConsoleExt.WriteLine($"{host.IPAddress}:{host.Port} > Does not have the requested file", ConsoleColor.Red); ApiManager.SendMessageNewLine($"{host.IPAddress}:{host.Port} > Does not have the requested file", NotificationMessageType.Error);
continue; continue;
} }
host.LastActive = DateTime.Now; host.LastActive = DateTime.Now;
ConsoleExt.WriteLine($"{host.IPAddress}:{host.Port} > Has the requested file", ConsoleColor.Green); ApiManager.SendMessageNewLine($"{host.IPAddress}:{host.Port} > Has the requested file", NotificationMessageType.Success);
Console.WriteLine("Requesting file..."); ApiManager.SendMessageNewLine("Requesting file...");
using TcpClient tcpClient = new TcpClient(); using TcpClient tcpClient = new TcpClient();
tcpClient.Connect(ipAddress!, host.Port); tcpClient.Connect(ipAddress!, host.Port);
@@ -118,8 +118,8 @@ internal class DownloadCommands
} }
catch (IOException) catch (IOException)
{ {
Console.WriteLine(); ApiManager.SendMessageNewLine(string.Empty);
ConsoleExt.WriteLine("Lost connection to other host!", ConsoleColor.Red); ApiManager.SendMessageNewLine("Lost connection to other host!", NotificationMessageType.Error);
continue; continue;
} }
@@ -133,16 +133,16 @@ internal class DownloadCommands
if (!validFileSize || fileSize <= 0) if (!validFileSize || fileSize <= 0)
{ {
ConsoleExt.WriteLine("Invalid file size!", ConsoleColor.Red); ApiManager.SendMessageNewLine("Invalid file size!", NotificationMessageType.Error);
continue; continue;
} }
string fileName = parts[1].ToFileName(); string fileName = parts[1].ToFileName();
string directoryPath = Path.Combine(Program.BasePath, "Downloads"); string directoryPath = Path.Combine(ApiConfiguration.BasePath, "Downloads");
string filePath = Path.Combine(directoryPath, fileName); string filePath = Path.Combine(directoryPath, fileName);
Console.WriteLine($"File name: {fileName}"); ApiManager.SendMessageNewLine($"File name: {fileName}");
Console.WriteLine($"Starting download..."); ApiManager.SendMessageNewLine($"Starting download...");
int totalBytesRead = 0; int totalBytesRead = 0;
@@ -167,8 +167,8 @@ internal class DownloadCommands
} }
catch (IOException) catch (IOException)
{ {
Console.WriteLine(); ApiManager.SendMessageNewLine(string.Empty);
ConsoleExt.WriteLine("Lost connection to other host!", ConsoleColor.Red); ApiManager.SendMessageNewLine("Lost connection to other host!", NotificationMessageType.Error);
break; break;
} }
@@ -196,9 +196,7 @@ internal class DownloadCommands
stopWatch.Restart(); stopWatch.Restart();
} }
Console.CursorLeft = 0; ApiManager.SendMessage($"\rDownloading: {Math.Clamp(Math.Ceiling(100d / fileSize * totalBytesRead), 0, 100)}% {totalBytesRead / 1000}/{fileSize / 1000}KB [{unitsPerSecond}{unit}/s] ");
Console.Out.WriteAsync($"Downloading: {Math.Clamp(Math.Ceiling(100d / fileSize * totalBytesRead), 0, 100)}% {totalBytesRead / 1000}/{fileSize / 1000}KB [{unitsPerSecond}{unit}/s] ");
} }
fileStream.Close(); fileStream.Close();
@@ -208,30 +206,30 @@ internal class DownloadCommands
continue; continue;
} }
Console.WriteLine(); ApiManager.SendMessageNewLine(string.Empty);
Console.WriteLine("Validating file..."); ApiManager.SendMessageNewLine("Validating file...");
if (FileValidator.ValidateHash(filePath, hash)) if (FileValidator.ValidateHash(filePath, hash))
{ {
_ = filesManager.TryAdd(filePath, out _, out _); _ = filesManager.TryAdd(filePath, out _, out _);
} }
else else
{ {
ConsoleExt.WriteLine("Warning: File hash does not match! File might me corrupted or manipulated!", ConsoleColor.DarkYellow); ApiManager.SendMessageNewLine("Warning: File hash does not match! File might me corrupted or manipulated!", NotificationMessageType.Warning);
} }
Console.WriteLine($"File saved at: {Path.GetFullPath(filePath)}"); ApiManager.SendMessageNewLine($"File saved at: {Path.GetFullPath(filePath)}");
ConsoleExt.WriteLine("Done", ConsoleColor.Green); ApiManager.SendMessageNewLine("Done", NotificationMessageType.Success);
stopWatch.Stop(); stopWatch.Stop();
hostsManager.SaveHosts(); hostsManager.SaveHosts();
return; return;
} }
else else
{ {
ConsoleExt.WriteLine(dataReceived, ConsoleColor.Red); ApiManager.SendMessageNewLine(dataReceived, NotificationMessageType.Error);
} }
} }
ConsoleExt.WriteLine("None of the available hosts have the requested file!", ConsoleColor.Red); ApiManager.SendMessageNewLine("None of the available hosts have the requested file!", NotificationMessageType.Error);
hostsManager.SaveHosts(); hostsManager.SaveHosts();
} }
} }
@@ -1,14 +1,13 @@
using HyperbolicDownloader.FileProcessing; using HyperbolicDownloaderApi.FileProcessing;
using HyperbolicDownloader.Networking; using HyperbolicDownloaderApi.Managment;
using HyperbolicDownloaderApi.Networking;
using Stone_Red_Utilities.ConsoleExtentions;
using System.Net; using System.Net;
using System.Text.Json; using System.Text.Json;
namespace HyperbolicDownloader.UserInterface.Commands; namespace HyperbolicDownloaderApi.Commands;
internal class FileCommands public class FileCommands
{ {
private readonly HostsManager hostsManager; private readonly HostsManager hostsManager;
@@ -24,12 +23,12 @@ internal class FileCommands
{ {
if (filesManager.TryAdd(path, out PrivateHyperFileInfo? fileInfo, out string? message)) if (filesManager.TryAdd(path, out PrivateHyperFileInfo? fileInfo, out string? message))
{ {
ConsoleExt.WriteLine($"Added file: {fileInfo!.FilePath}", ConsoleColor.Green); ApiManager.SendMessageNewLine($"Added file: {fileInfo!.FilePath}", NotificationMessageType.Success);
Console.WriteLine($"Hash: {fileInfo.Hash}"); ApiManager.SendMessageNewLine($"Hash: {fileInfo.Hash}");
} }
else else
{ {
ConsoleExt.WriteLine(message, ConsoleColor.Red); ApiManager.SendMessageNewLine(message, NotificationMessageType.Error);
} }
} }
@@ -39,11 +38,11 @@ internal class FileCommands
if (filesManager.TryRemove(hash)) if (filesManager.TryRemove(hash))
{ {
ConsoleExt.WriteLine($"Successfully removed file!", ConsoleColor.Green); ApiManager.SendMessageNewLine($"Successfully removed file!", NotificationMessageType.Success);
} }
else else
{ {
ConsoleExt.WriteLine("The file is not being tracked!", ConsoleColor.Red); ApiManager.SendMessageNewLine("The file is not being tracked!", NotificationMessageType.Error);
} }
} }
@@ -54,23 +53,23 @@ internal class FileCommands
if (fileInfos.Count == 0) if (fileInfos.Count == 0)
{ {
ConsoleExt.WriteLine("No tracked files!", ConsoleColor.DarkYellow); ApiManager.SendMessageNewLine("No tracked files!", NotificationMessageType.Warning);
return; return;
} }
foreach (PrivateHyperFileInfo fileInfo in fileInfos) foreach (PrivateHyperFileInfo fileInfo in fileInfos)
{ {
index++; index++;
Console.WriteLine($"{index}) {fileInfo.FilePath}"); ApiManager.SendMessageNewLine($"{index}) {fileInfo.FilePath}");
Console.WriteLine($"Hash: {fileInfo.Hash}"); ApiManager.SendMessageNewLine($"Hash: {fileInfo.Hash}");
Console.WriteLine(); ApiManager.SendMessageNewLine(string.Empty);
} }
Console.CursorTop--; Console.CursorTop--;
} }
public void GenerateFileSingle(string hash) public void GenerateFileSingle(string hash)
{ {
string directoryPath = Path.Combine(Program.BasePath, "GeneratedFiles"); string directoryPath = Path.Combine(ApiConfiguration.BasePath, "GeneratedFiles");
if (!Directory.Exists(directoryPath)) if (!Directory.Exists(directoryPath))
{ {
Directory.CreateDirectory(directoryPath); Directory.CreateDirectory(directoryPath);
@@ -80,7 +79,7 @@ internal class FileCommands
if (!filesManager.TryGet(hash, out PrivateHyperFileInfo? localHyperFileInfo)) if (!filesManager.TryGet(hash, out PrivateHyperFileInfo? localHyperFileInfo))
{ {
ConsoleExt.WriteLine("The file is not being tracked!", ConsoleColor.Red); ApiManager.SendMessageNewLine("The file is not being tracked!", NotificationMessageType.Error);
return; return;
} }
@@ -89,10 +88,10 @@ internal class FileCommands
PublicHyperFileInfo publicHyperFileInfo = new PublicHyperFileInfo(hash); PublicHyperFileInfo publicHyperFileInfo = new PublicHyperFileInfo(hash);
NetworkSocket? localHost = Program.GetLocalSocket(); NetworkSocket? localHost = ApiManager.GetLocalSocket();
if (localHost is null) if (localHost is null)
{ {
ConsoleExt.WriteLine("Network error!", ConsoleColor.Red); ApiManager.SendMessageNewLine("Network error!", NotificationMessageType.Error);
return; return;
} }
@@ -103,13 +102,13 @@ internal class FileCommands
File.WriteAllText(filePath, json); File.WriteAllText(filePath, json);
ConsoleExt.WriteLine("Done", ConsoleColor.Green); ApiManager.SendMessageNewLine("Done", NotificationMessageType.Success);
Console.WriteLine($"File saved at: {Path.GetFullPath(filePath)}"); ApiManager.SendMessageNewLine($"File saved at: {Path.GetFullPath(filePath)}");
} }
public void GenerateFileFull(string hash) public void GenerateFileFull(string hash)
{ {
string directoryPath = Path.Combine(Program.BasePath, "GeneratedFiles"); string directoryPath = Path.Combine(ApiConfiguration.BasePath, "GeneratedFiles");
if (!Directory.Exists(directoryPath)) if (!Directory.Exists(directoryPath))
{ {
Directory.CreateDirectory(directoryPath); Directory.CreateDirectory(directoryPath);
@@ -119,7 +118,7 @@ internal class FileCommands
if (!filesManager.TryGet(hash, out PrivateHyperFileInfo? localHyperFileInfo)) if (!filesManager.TryGet(hash, out PrivateHyperFileInfo? localHyperFileInfo))
{ {
ConsoleExt.WriteLine("The file is not being tracked!", ConsoleColor.Red); ApiManager.SendMessageNewLine("The file is not being tracked!", NotificationMessageType.Error);
return; return;
} }
@@ -128,10 +127,10 @@ internal class FileCommands
PublicHyperFileInfo publicHyperFileInfo = new PublicHyperFileInfo(hash); PublicHyperFileInfo publicHyperFileInfo = new PublicHyperFileInfo(hash);
NetworkSocket? localHost = Program.GetLocalSocket(); NetworkSocket? localHost = ApiManager.GetLocalSocket();
if (localHost is null) if (localHost is null)
{ {
ConsoleExt.WriteLine("Network error!", ConsoleColor.Red); ApiManager.SendMessageNewLine("Network error!", NotificationMessageType.Error);
return; return;
} }
@@ -145,7 +144,7 @@ internal class FileCommands
continue; continue;
} }
ConsoleExt.Write($"{host.IPAddress}:{host.Port} > ???", ConsoleColor.DarkYellow); ApiManager.SendMessage($"{host.IPAddress}:{host.Port} > ???", NotificationMessageType.Warning);
Console.CursorLeft = 0; Console.CursorLeft = 0;
@@ -156,7 +155,7 @@ internal class FileCommands
if (!sendTask.IsCompletedSuccessfully) if (!sendTask.IsCompletedSuccessfully)
{ {
Console.CursorLeft = 0; Console.CursorLeft = 0;
ConsoleExt.WriteLine($"{host.IPAddress}:{host.Port} > Inactive", ConsoleColor.Red); ApiManager.SendMessageNewLine($"{host.IPAddress}:{host.Port} > Inactive", NotificationMessageType.Error);
hostsManager.Remove(host); hostsManager.Remove(host);
continue; continue;
@@ -164,13 +163,13 @@ internal class FileCommands
else if (!sendTask.Result) else if (!sendTask.Result)
{ {
host.LastActive = DateTime.Now; host.LastActive = DateTime.Now;
ConsoleExt.WriteLine($"{host.IPAddress}:{host.Port} > Does not have the requested file", ConsoleColor.Red); ApiManager.SendMessageNewLine($"{host.IPAddress}:{host.Port} > Does not have the requested file", NotificationMessageType.Error);
continue; continue;
} }
host.LastActive = DateTime.Now; host.LastActive = DateTime.Now;
ConsoleExt.WriteLine($"{host.IPAddress}:{host.Port} > Has the requested file", ConsoleColor.Green); ApiManager.SendMessageNewLine($"{host.IPAddress}:{host.Port} > Has the requested file", NotificationMessageType.Success);
publicHyperFileInfo.Hosts.Add(host); publicHyperFileInfo.Hosts.Add(host);
} }
@@ -181,7 +180,7 @@ internal class FileCommands
File.WriteAllText(filePath, json); File.WriteAllText(filePath, json);
ConsoleExt.WriteLine("Done", ConsoleColor.Green); ApiManager.SendMessageNewLine("Done", NotificationMessageType.Success);
Console.WriteLine($"File saved at: {Path.GetFullPath(filePath)}"); ApiManager.SendMessageNewLine($"File saved at: {Path.GetFullPath(filePath)}");
} }
} }
@@ -0,0 +1,148 @@
using HyperbolicDownloaderApi.Managment;
using HyperbolicDownloaderApi.Networking;
using System.Net;
using System.Net.Sockets;
namespace HyperbolicDownloaderApi.Commands;
public class HostCommands
{
private readonly HostsManager hostsManager;
public HostCommands(HostsManager hostsManager)
{
this.hostsManager = hostsManager;
}
public void Discover(string _)
{
ApiManager.SendMessageNewLine("Running local discovery routine...", NotificationMessageType.Raw);
BroadcastClient.Send(ApiConfiguration.BroadcastPort, ApiConfiguration.PrivatePort.ToString());
Thread.Sleep(3000);
}
public void CheckActiveHosts(string _)
{
int activeHostsCount = hostsManager.CheckHostsActivity();
if (activeHostsCount == 0)
{
ApiManager.SendMessageNewLine("No active hosts found!", NotificationMessageType.Error);
ApiManager.SendMessageNewLine("Use 'add host xxx.xxx.xxx.xxx:yyyy' to add a new host.", NotificationMessageType.Error);
}
ApiManager.SendMessageNewLine(string.Empty, NotificationMessageType.Raw);
ApiManager.SendMessageNewLine($"{hostsManager.Count} known host(s).", NotificationMessageType.Raw);
ApiManager.SendMessageNewLine($"{activeHostsCount} active host(s).", NotificationMessageType.Raw);
}
public void ListHosts(string _)
{
int index = 0;
List<NetworkSocket> hosts = hostsManager.ToList();
if (hosts.Count == 0)
{
ApiManager.SendMessageNewLine("No known hosts", NotificationMessageType.Warning);
return;
}
foreach (NetworkSocket host in hosts)
{
index++;
ApiManager.SendMessageNewLine($"{index}) {host.IPAddress}:{host.Port}", NotificationMessageType.Raw);
ApiManager.SendMessageNewLine($"Last active: {host.LastActive}", NotificationMessageType.Raw);
ApiManager.SendMessageNewLine(string.Empty, NotificationMessageType.Raw);
}
Console.CursorTop--;
}
public void RemoveHost(string args)
{
string[] parts = args.Split(":");
if (parts.Length != 2)
{
ApiManager.SendMessageNewLine("Invalid format! Use this format: (xxx.xxx.xxx.xxx:yyyy)", NotificationMessageType.Error);
return;
}
string ipAddressInput = parts[0];
string portInput = parts[1];
_ = int.TryParse(portInput, out int port);
if (port < 1000 || port >= 6000)
{
ApiManager.SendMessageNewLine("Invalid port number!", NotificationMessageType.Error);
return;
}
if (!IPAddress.TryParse(ipAddressInput, out IPAddress? ipAddress))
{
ApiManager.SendMessageNewLine("Invalid IP address!", NotificationMessageType.Error);
return;
}
NetworkSocket hostToRemove = new NetworkSocket(ipAddress.ToString(), port, DateTime.MinValue);
if (!hostsManager.Contains(hostToRemove))
{
ApiManager.SendMessageNewLine("Host not in list", NotificationMessageType.Error);
return;
}
hostsManager.Remove(new NetworkSocket(ipAddress.ToString(), port, DateTime.MinValue), true);
ApiManager.SendMessageNewLine($"Successfully Removed host!", NotificationMessageType.Success);
}
public void AddHost(string args)
{
string[] parts = args.Split(":");
if (parts.Length != 2)
{
ApiManager.SendMessageNewLine("Invalid format! Use this format: (xxx.xxx.xxx.xxx:yyyy)", NotificationMessageType.Error);
return;
}
string ipAddressInput = parts[0];
string portInput = parts[1];
_ = int.TryParse(portInput, out int port);
if (port < 1000 || port >= 6000)
{
ApiManager.SendMessageNewLine("Invalid port number!", NotificationMessageType.Error);
}
else if (IPAddress.TryParse(ipAddressInput, out IPAddress? ipAddress))
{
try
{
ApiManager.SendMessageNewLine("Waiting for response...", NotificationMessageType.Raw);
NetworkSocket? localSocket = ApiManager.GetLocalSocket() ?? new NetworkSocket("0.0.0.0", 0, DateTime.MinValue);
List<NetworkSocket>? recivedHosts = NetworkClient.Send<List<NetworkSocket>>(ipAddress, port, "GetHostsList", localSocket);
if (recivedHosts is not null)
{
ApiManager.SendMessageNewLine($"Success! Added {recivedHosts.Count} new host(s).", NotificationMessageType.Success);
hostsManager.AddRange(recivedHosts);
}
else
{
ApiManager.SendMessageNewLine($"Invalid response!", NotificationMessageType.Error);
}
}
catch (SocketException ex)
{
ApiManager.SendMessageNewLine($"Invalid host! Error message: {ex.Message}", NotificationMessageType.Error);
}
catch (IOException ex)
{
ApiManager.SendMessageNewLine($"Invalid host! Error message: {ex.Message}", NotificationMessageType.Error);
}
}
else
{
ApiManager.SendMessageNewLine("Invalid IP address!", NotificationMessageType.Error);
}
}
}
@@ -1,6 +1,6 @@
using System.IO.Compression; using System.IO.Compression;
namespace HyperbolicDownloader.FileProcessing; namespace HyperbolicDownloaderApi.FileProcessing;
internal static class FileCompressor internal static class FileCompressor
{ {
@@ -1,8 +1,8 @@
using System.Security.Cryptography; using System.Security.Cryptography;
namespace HyperbolicDownloader.FileProcessing; namespace HyperbolicDownloaderApi.FileProcessing;
internal class FileValidator internal static class FileValidator
{ {
public static async Task<string> CalculateHashAsync(string filePath) public static async Task<string> CalculateHashAsync(string filePath)
{ {
@@ -1,8 +1,10 @@
using System.Text.Json; using HyperbolicDownloaderApi.Managment;
namespace HyperbolicDownloader.FileProcessing; using System.Text.Json;
internal class FilesManager namespace HyperbolicDownloaderApi.FileProcessing;
public class FilesManager
{ {
private readonly List<PrivateHyperFileInfo> files = new List<PrivateHyperFileInfo>(); private readonly List<PrivateHyperFileInfo> files = new List<PrivateHyperFileInfo>();
@@ -91,6 +93,6 @@ internal class FilesManager
private void SaveFiles() private void SaveFiles()
{ {
File.WriteAllText(Program.FilesInfoPath, JsonSerializer.Serialize(files)); File.WriteAllText(ApiConfiguration.FilesInfoPath, JsonSerializer.Serialize(files));
} }
} }
@@ -1,6 +1,6 @@
namespace HyperbolicDownloader.FileProcessing; namespace HyperbolicDownloaderApi.FileProcessing;
internal class PrivateHyperFileInfo public class PrivateHyperFileInfo
{ {
public string Hash { get; set; } public string Hash { get; set; }
public string FilePath { get; set; } public string FilePath { get; set; }
@@ -1,8 +1,8 @@
using HyperbolicDownloader.Networking; using HyperbolicDownloaderApi.Networking;
namespace HyperbolicDownloader.FileProcessing; namespace HyperbolicDownloaderApi.FileProcessing;
internal class PublicHyperFileInfo public class PublicHyperFileInfo
{ {
public string Hash { get; set; } = string.Empty; public string Hash { get; set; } = string.Empty;
public List<NetworkSocket> Hosts { get; set; } = new(); public List<NetworkSocket> Hosts { get; set; } = new();
@@ -0,0 +1,14 @@
<Project Sdk="Microsoft.NET.Sdk">
<PropertyGroup>
<TargetFramework>net6.0</TargetFramework>
<ImplicitUsings>enable</ImplicitUsings>
<Nullable>enable</Nullable>
</PropertyGroup>
<ItemGroup>
<PackageReference Include="Open.Nat" Version="2.1.0" />
<PackageReference Include="Stone_Red-C-Sharp-Utilities" Version="1.0.3.1" />
</ItemGroup>
</Project>
@@ -0,0 +1,13 @@
using System.Reflection;
namespace HyperbolicDownloaderApi.Managment;
public static class ApiConfiguration
{
public const int BroadcastPort = 2155;
public const int PrivatePort = 3055;
public static int PublicPort { get; set; }
public static string BasePath { get; } = Path.GetDirectoryName(Assembly.GetEntryAssembly()!.Location) ?? string.Empty;
public static string HostsFilePath { get; } = Path.Combine(BasePath, "Hosts.json");
public static string FilesInfoPath { get; } = Path.Combine(BasePath, "Files.json");
}
@@ -0,0 +1,213 @@
using HyperbolicDownloaderApi.FileProcessing;
using HyperbolicDownloaderApi.Networking;
using Open.Nat;
using System.Diagnostics;
using System.Net;
using System.Net.Sockets;
namespace HyperbolicDownloaderApi.Managment;
public class ApiManager
{
public static event EventHandler<NotificationMessageEventArgs>? OnNotificationMessageRecived;
public static IPAddress? PublicIpAddress => device?.GetExternalIPAsync().GetAwaiter().GetResult();
public FilesManager FilesManager { get; } = new();
public HostsManager HostsManager { get; } = new();
private static readonly Random random = new();
private static NatDevice? device;
private static Mapping? portMapping;
private readonly BroadcastClient broadcastClient = new BroadcastClient();
private readonly NetworkClient networkClient;
public ApiManager()
{
networkClient = new(FilesManager);
}
public static void ClosePorts()
{
ApiManager.SendMessageNewLine("Closing ports...", NotificationMessageType.Success);
if (device is not null)
{
try
{
IEnumerable<Mapping>? mappings = device.GetAllMappingsAsync().GetAwaiter().GetResult();
foreach (Mapping? mapping in mappings)
{
if (mapping.Description.Contains("HyperbolicDowloader") && mapping.PrivateIP.ToString() == portMapping?.PrivateIP.ToString())
{
device.DeletePortMapAsync(mapping).GetAwaiter().GetResult();
}
}
}
catch (Exception ex)
{
ApiManager.SendMessageNewLine(ex.ToString(), NotificationMessageType.Error);
}
}
ApiManager.SendMessageNewLine("Ports closed!", NotificationMessageType.Warning);
Environment.Exit(0);
}
public static NetworkSocket? GetLocalSocket()
{
int port = ApiConfiguration.PublicPort;
string? ipAddress = null;
if (device is not null)
{
ipAddress = (device.GetExternalIPAsync()).GetAwaiter().GetResult()?.ToString();
}
if (ipAddress is null || ipAddress == "0.0.0.0")
{
ipAddress = NetworkUtilities.GetIP4Adress()?.ToString();
port = ApiConfiguration.PrivatePort;
}
if (ipAddress is null)
{
return null;
}
return new NetworkSocket(ipAddress, port, DateTime.Now);
}
public static async Task<bool> OpenPorts()
{
try
{
ApiConfiguration.PublicPort = random.Next(1000, 6000);
NatDiscoverer? discoverer = new NatDiscoverer();
device = await discoverer.DiscoverDeviceAsync();
IPAddress? ip = await device.GetExternalIPAsync();
ApiManager.SendMessageNewLine($"The public IP address is: {ip} ", NotificationMessageType.Success);
portMapping = new Mapping(Protocol.Tcp, ApiConfiguration.PrivatePort, ApiConfiguration.PublicPort, "HyperbolicDowloader");
await device.CreatePortMapAsync(portMapping);
ApiManager.SendMessageNewLine($"The public port is: {ApiConfiguration.PublicPort}", NotificationMessageType.Success);
return true;
}
catch (NatDeviceNotFoundException)
{
ApiManager.SendMessageNewLine($"Could not find a UPnP or NAT-PMP device!", NotificationMessageType.Error);
return false;
}
catch (MappingException ex)
{
ApiManager.SendMessageNewLine($"An error occurred while mapping the private port ({ApiConfiguration.PrivatePort}) to the public port ({ApiConfiguration.PublicPort})! Error message: {ex.Message}", NotificationMessageType.Error);
return false;
}
}
public void StartBroadcastListener()
{
broadcastClient.StartListening(ApiConfiguration.BroadcastPort);
broadcastClient.OnBroadcastRecived += BroadcastClient_OnBroadcastRecived;
}
public void StartTcpListener()
{
try
{
networkClient.ListenTo<NetworkSocket>("GetHostsList", GetHostList);
networkClient.ListenTo<List<NetworkSocket>>("DiscoverAnswer", DiscoverAnswer);
networkClient.ListenTo<string>("Message", ReciveMessage);
networkClient.ListenTo<string>("HasFile", HasFile);
networkClient.StartListening(ApiConfiguration.PrivatePort);
}
catch (SocketException ex)
{
ApiManager.SendMessageNewLine($"An error occurred while starting the TCP listener! Error message: {ex.Message}", NotificationMessageType.Error); // net stop hns && net start hns
Console.ReadKey();
}
}
private async void BroadcastClient_OnBroadcastRecived(object? sender, BroadcastRecivedEventArgs recivedEventArgs)
{
Debug.WriteLine($"Received broadcast \"{recivedEventArgs.Message}\" from {recivedEventArgs.IPEndPoint.Address}");
List<NetworkSocket> hostsToSend = HostsManager.ToList();
NetworkSocket? localSocket = GetLocalSocket();
if (localSocket is null)
{
return;
}
hostsToSend.RemoveAll(x => x.IPAddress == recivedEventArgs.IPEndPoint.Address.ToString());
hostsToSend.Add(localSocket);
bool success = int.TryParse(recivedEventArgs.Message, out int remotePort);
if (success)
{
HostsManager.Add(new NetworkSocket(recivedEventArgs.IPEndPoint.Address.ToString(), remotePort, DateTime.Now));
try
{
await NetworkClient.SendAsync(recivedEventArgs.IPEndPoint.Address, remotePort, "DiscoverAnswer", hostsToSend);
}
catch (SocketException ex)
{
Debug.WriteLine(ex);
}
}
}
private void DiscoverAnswer(object? sender, MessageRecivedEventArgs<List<NetworkSocket>> recivedEventArgs)
{
ApiManager.SendMessageNewLine($"Received answer from {recivedEventArgs.IpAddress}. Returned {recivedEventArgs.Data.Count} host(s).", NotificationMessageType.Raw);
HostsManager.AddRange(recivedEventArgs.Data);
}
private async void GetHostList(object? sender, MessageRecivedEventArgs<NetworkSocket> recivedEventArgs)
{
List<NetworkSocket> hostsToSend = HostsManager.ToList();
NetworkSocket? localSocket = GetLocalSocket();
if (localSocket is null)
{
return;
}
hostsToSend.RemoveAll(x => x.IPAddress == recivedEventArgs.IpAddress.ToString());
hostsToSend.Add(localSocket);
if (recivedEventArgs.Data.Port != 0)
{
HostsManager.Add(recivedEventArgs.Data);
}
await recivedEventArgs.SendResponseAsync(hostsToSend);
}
private async void HasFile(object? sender, MessageRecivedEventArgs<string> recivedEventArgs)
{
await recivedEventArgs.SendResponseAsync(FilesManager.Contains(recivedEventArgs.Data));
}
private void ReciveMessage(object? sender, MessageRecivedEventArgs<string> recivedEventArgs)
{
ApiManager.SendMessageNewLine($"Received \"{recivedEventArgs.Data}\" from {recivedEventArgs.IpAddress}.", NotificationMessageType.Raw);
}
internal static void SendMessage(string message, NotificationMessageType messageType = NotificationMessageType.Raw)
{
OnNotificationMessageRecived?.Invoke(null, new NotificationMessageEventArgs(messageType, message));
}
internal static void SendMessageNewLine(string message, NotificationMessageType messageType = NotificationMessageType.Raw)
{
OnNotificationMessageRecived?.Invoke(null, new NotificationMessageEventArgs(messageType, message + Environment.NewLine));
}
}
@@ -0,0 +1,21 @@
namespace HyperbolicDownloaderApi.Managment;
public class NotificationMessageEventArgs : EventArgs
{
public NotificationMessageType NotificationMessageType { get; }
public string? Message { get; }
public NotificationMessageEventArgs(NotificationMessageType notificationMessageType, string? message)
{
NotificationMessageType = notificationMessageType;
Message = message;
}
}
public enum NotificationMessageType
{
Raw,
Success,
Warning,
Error
}
@@ -1,11 +1,11 @@
using Stone_Red_Utilities.ConsoleExtentions; using HyperbolicDownloaderApi.Managment;
using System.Net; using System.Net;
using System.Net.NetworkInformation; using System.Net.NetworkInformation;
using System.Net.Sockets; using System.Net.Sockets;
using System.Text; using System.Text;
namespace HyperbolicDownloader.Networking; namespace HyperbolicDownloaderApi.Networking;
internal class BroadcastClient internal class BroadcastClient
{ {
@@ -26,7 +26,7 @@ internal class BroadcastClient
if (ip4Address is null) if (ip4Address is null)
{ {
ConsoleExt.WriteLine("Could not find suitable network adapter!", ConsoleColor.Red); ApiManager.SendMessageNewLine("Could not find suitable network adapter!", NotificationMessageType.Error);
return; return;
} }
@@ -34,7 +34,7 @@ internal class BroadcastClient
if (addressInformation is null) if (addressInformation is null)
{ {
ConsoleExt.WriteLine("Could not find suitable network adapter!", ConsoleColor.Red); ApiManager.SendMessageNewLine("Could not find suitable network adapter!", NotificationMessageType.Error);
return; return;
} }
@@ -48,7 +48,7 @@ internal class BroadcastClient
public void StartListening(int port) public void StartListening(int port)
{ {
if (IsListening == true) if (IsListening)
{ {
throw new InvalidOperationException("Already listening!"); throw new InvalidOperationException("Already listening!");
} }
@@ -1,6 +1,6 @@
using System.Net; using System.Net;
namespace HyperbolicDownloader.Networking; namespace HyperbolicDownloaderApi.Networking;
internal class BroadcastRecivedEventArgs : EventArgs internal class BroadcastRecivedEventArgs : EventArgs
{ {
@@ -1,4 +1,4 @@
namespace HyperbolicDownloader namespace HyperbolicDownloaderApi
{ {
internal class DataContainer internal class DataContainer
{ {
@@ -1,13 +1,12 @@
using HyperbolicDownloader.Networking; using HyperbolicDownloaderApi.Managment;
using HyperbolicDownloaderApi.Networking;
using Stone_Red_Utilities.ConsoleExtentions;
using System.Net.Sockets; using System.Net.Sockets;
using System.Text.Json; using System.Text.Json;
namespace HyperbolicDownloader; namespace HyperbolicDownloaderApi;
internal class HostsManager public class HostsManager
{ {
private List<NetworkSocket> hosts = new List<NetworkSocket>(); private List<NetworkSocket> hosts = new List<NetworkSocket>();
public int Count => hosts.Count; public int Count => hosts.Count;
@@ -54,7 +53,7 @@ internal class HostsManager
int activeHostsCount = 0; int activeHostsCount = 0;
foreach (NetworkSocket host in hosts) foreach (NetworkSocket host in hosts)
{ {
ConsoleExt.Write($"{host.IPAddress}:{host.Port} > ???", ConsoleColor.DarkYellow); ApiManager.SendMessage($"{host.IPAddress}:{host.Port} > ???", NotificationMessageType.Warning);
using TcpClient tcpClient = new TcpClient(); using TcpClient tcpClient = new TcpClient();
try try
@@ -63,7 +62,7 @@ internal class HostsManager
Console.CursorLeft = 0; Console.CursorLeft = 0;
if (tcpClient.Connected) if (tcpClient.Connected)
{ {
ConsoleExt.WriteLine($"{host.IPAddress}:{host.Port} > Active", ConsoleColor.Green); ApiManager.SendMessageNewLine($"{host.IPAddress}:{host.Port} > Active", NotificationMessageType.Success);
host.LastActive = DateTime.Now; host.LastActive = DateTime.Now;
activeHostsCount++; activeHostsCount++;
} }
@@ -73,7 +72,7 @@ internal class HostsManager
{ {
hostsToRemove.Add(host); hostsToRemove.Add(host);
} }
ConsoleExt.WriteLine($"{host.IPAddress}:{host.Port} > Inactive", ConsoleColor.Red); ApiManager.SendMessageNewLine($"{host.IPAddress}:{host.Port} > Inactive", NotificationMessageType.Error);
} }
} }
catch catch
@@ -83,7 +82,7 @@ internal class HostsManager
hostsToRemove.Add(host); hostsToRemove.Add(host);
} }
Console.CursorLeft = 0; Console.CursorLeft = 0;
ConsoleExt.WriteLine($"{host.IPAddress}:{host.Port} > Inactive", ConsoleColor.Red); ApiManager.SendMessageNewLine($"{host.IPAddress}:{host.Port} > Inactive", NotificationMessageType.Error);
} }
} }
@@ -104,6 +103,6 @@ internal class HostsManager
public void SaveHosts() public void SaveHosts()
{ {
hosts = hosts.OrderByDescending(h => h.LastActive).ToList(); hosts = hosts.OrderByDescending(h => h.LastActive).ToList();
File.WriteAllText(Program.HostsFilePath, JsonSerializer.Serialize(hosts)); File.WriteAllText(ApiConfiguration.HostsFilePath, JsonSerializer.Serialize(hosts));
} }
} }
@@ -3,7 +3,7 @@ using System.Net.Sockets;
using System.Text; using System.Text;
using System.Text.Json; using System.Text.Json;
namespace HyperbolicDownloader namespace HyperbolicDownloaderApi
{ {
internal class MessageRecivedEventArgs<T> : EventArgs internal class MessageRecivedEventArgs<T> : EventArgs
{ {
@@ -1,4 +1,4 @@
using HyperbolicDownloader.FileProcessing; using HyperbolicDownloaderApi.FileProcessing;
using System.Diagnostics; using System.Diagnostics;
using System.Net; using System.Net;
@@ -6,7 +6,7 @@ using System.Net.Sockets;
using System.Text; using System.Text;
using System.Text.Json; using System.Text.Json;
namespace HyperbolicDownloader.Networking namespace HyperbolicDownloaderApi.Networking
{ {
internal class NetworkClient internal class NetworkClient
{ {
@@ -1,6 +1,6 @@
namespace HyperbolicDownloader.Networking; namespace HyperbolicDownloaderApi.Networking;
internal class NetworkSocket public class NetworkSocket
{ {
public string IPAddress { get; set; } public string IPAddress { get; set; }
public int Port { get; set; } public int Port { get; set; }
@@ -2,9 +2,9 @@
using System.Net.NetworkInformation; using System.Net.NetworkInformation;
using System.Net.Sockets; using System.Net.Sockets;
namespace HyperbolicDownloader.Networking; namespace HyperbolicDownloaderApi.Networking;
internal static class NetworkUtilities public static class NetworkUtilities
{ {
public static UnicastIPAddressInformation? GetUnicastIPAddressInformation(IPAddress address) public static UnicastIPAddressInformation? GetUnicastIPAddressInformation(IPAddress address)
{ {