Code cleanup

This commit is contained in:
Stone_Red
2023-02-06 15:15:47 +01:00
parent 7d662dc113
commit fb21a9f492
10 changed files with 235 additions and 237 deletions
+14 -14
View File
@@ -20,31 +20,31 @@ internal class InputHandler
DownloadCommands downloadCommands = new DownloadCommands(hostsManager, filesManager); DownloadCommands downloadCommands = new DownloadCommands(hostsManager, filesManager);
ClientCommands clientCommands = new ClientCommands(); ClientCommands clientCommands = new ClientCommands();
commander.Register(input => commander.PrintHelp(input), "help"); _ = commander.Register(input => commander.PrintHelp(input), "help");
commander.Register(_ => Console.Clear(), (HelpText)"Clears the console.", "clear", "cls"); _ = commander.Register(_ => Console.Clear(), (HelpText)"Clears the console.", "clear", "cls");
commander.Register(_ => exit = true, (HelpText)"Exits the application.", "exit", "quit"); _ = commander.Register(_ => exit = true, (HelpText)"Exits the application.", "exit", "quit");
commander.Register(clientCommands.ShowInfo, (HelpText)"Displays the private and public IP address.", "info", "inf"); _ = commander.Register(clientCommands.ShowInfo, (HelpText)"Displays the private and public IP address.", "info", "inf");
commander.Register(hostCommands.Discover, (HelpText)"Tries to find other active hosts on the local network.", "discover", "disc"); _ = commander.Register(hostCommands.Discover, (HelpText)"Tries to find other active hosts on the local network.", "discover", "disc");
Command getCommand = commander.Register(downloadCommands.GetFile, (HelpText)"Attempts to retrieve a file from another host using a hash.", "get"); Command getCommand = commander.Register(downloadCommands.GetFile, (HelpText)"Attempts to retrieve a file from another host using a hash.", "get");
getCommand.Register(downloadCommands.GetFileFrom, (HelpText)"Attempts to retrieve a file from another host using a .hyper file.", "from"); _ = getCommand.Register(downloadCommands.GetFileFrom, (HelpText)"Attempts to retrieve a file from another host using a .hyper file.", "from");
Command generateCommad = commander.Register(fileCommands.GenerateFileFull, (HelpText)"Generates a .hyper file from a file hash.", "generate", "gen"); Command generateCommad = commander.Register(fileCommands.GenerateFileFull, (HelpText)"Generates a .hyper file from a file hash.", "generate", "gen");
generateCommad.Register(fileCommands.GenerateFileSingle, (HelpText)"Generates a .hyper file from a file hash without checking the known hosts. This adds only the local host to the file.", "noscan"); _ = generateCommad.Register(fileCommands.GenerateFileSingle, (HelpText)"Generates a .hyper file from a file hash without checking the known hosts. This adds only the local host to the file.", "noscan");
Command addCommand = commander.Register(fileCommands.AddFile, (HelpText)"Adds a file to the tracking list.", "add"); Command addCommand = commander.Register(fileCommands.AddFile, (HelpText)"Adds a file to the tracking list.", "add");
addCommand.Register(fileCommands.AddFile, (HelpText)"Adds a file to the tracking list.", "file"); _ = addCommand.Register(fileCommands.AddFile, (HelpText)"Adds a file to the tracking list.", "file");
addCommand.Register(hostCommands.AddHost, (HelpText)"Adds a host to the list of known hosts.", "host"); _ = addCommand.Register(hostCommands.AddHost, (HelpText)"Adds a host to the list of known hosts.", "host");
Command removeCommand = commander.Register(fileCommands.RemoveFile, (HelpText)"Removes a file from the tracking list.", "remove", "rm"); Command removeCommand = commander.Register(fileCommands.RemoveFile, (HelpText)"Removes a file from the tracking list.", "remove", "rm");
removeCommand.Register(fileCommands.RemoveFile, (HelpText)"Removes a file from the tracking list.", "file"); _ = removeCommand.Register(fileCommands.RemoveFile, (HelpText)"Removes a file from the tracking list.", "file");
removeCommand.Register(hostCommands.RemoveHost, (HelpText)"Removes a host from the list of known hosts.", "host"); _ = removeCommand.Register(hostCommands.RemoveHost, (HelpText)"Removes a host from the list of known hosts.", "host");
Command listCommand = commander.Register(fileCommands.ListFiles, (HelpText)"Lists all files.", "list", "ls"); Command listCommand = commander.Register(fileCommands.ListFiles, (HelpText)"Lists all files.", "list", "ls");
listCommand.Register(fileCommands.ListFiles, (HelpText)"Lists all files.", "files"); _ = listCommand.Register(fileCommands.ListFiles, (HelpText)"Lists all files.", "files");
listCommand.Register(hostCommands.ListHosts, (HelpText)"lists all hosts.", "hosts"); _ = listCommand.Register(hostCommands.ListHosts, (HelpText)"lists all hosts.", "hosts");
commander.Register(hostCommands.CheckActiveHosts, (HelpText)"Checks the status of known hosts.", "status", "check"); _ = commander.Register(hostCommands.CheckActiveHosts, (HelpText)"Checks the status of known hosts.", "status", "check");
} }
public void ReadInput() public void ReadInput()
+9 -4
View File
@@ -45,6 +45,14 @@ internal static class Program
Console.WriteLine(); Console.WriteLine();
} }
await Initialize();
inputHandler.ReadInput();
Close();
}
private static async Task Initialize()
{
Console.WriteLine("Searching for a UPnP/NAT-PMP device..."); Console.WriteLine("Searching for a UPnP/NAT-PMP device...");
_ = await ApiManager.OpenPorts(); _ = await ApiManager.OpenPorts();
@@ -54,7 +62,7 @@ internal static class Program
Console.WriteLine("Starting TCP listener..."); Console.WriteLine("Starting TCP listener...");
if (!apiManager.StartTcpListener()) if (!apiManager.StartTcpListener())
{ {
Console.ReadLine(); _ = Console.ReadLine();
} }
Console.WriteLine("Starting broadcast listener..."); Console.WriteLine("Starting broadcast listener...");
@@ -77,9 +85,6 @@ internal static class Program
Console.WriteLine($"{activeHostsCount} active host(s)."); Console.WriteLine($"{activeHostsCount} active host(s).");
ConsoleExt.WriteLine("Ready", ConsoleColor.Green); ConsoleExt.WriteLine("Ready", ConsoleColor.Green);
inputHandler.ReadInput();
Close();
} }
private static void ApiManager_OnNotificationMessageRecived(object? sender, NotificationMessageEventArgs e) private static void ApiManager_OnNotificationMessageRecived(object? sender, NotificationMessageEventArgs e)
@@ -152,7 +152,7 @@ public class DownloadCommands
if (!Directory.Exists(directoryPath)) if (!Directory.Exists(directoryPath))
{ {
Directory.CreateDirectory(directoryPath); _ = Directory.CreateDirectory(directoryPath);
} }
using FileStream? fileStream = new FileStream(filePath, FileMode.Create); using FileStream? fileStream = new FileStream(filePath, FileMode.Create);
@@ -72,7 +72,7 @@ public class FileCommands
string directoryPath = Path.Combine(ApiConfiguration.BasePath, "GeneratedFiles"); string directoryPath = Path.Combine(ApiConfiguration.BasePath, "GeneratedFiles");
if (!Directory.Exists(directoryPath)) if (!Directory.Exists(directoryPath))
{ {
Directory.CreateDirectory(directoryPath); _ = Directory.CreateDirectory(directoryPath);
} }
hash = hash.Trim().ToLower(); hash = hash.Trim().ToLower();
@@ -111,7 +111,7 @@ public class FileCommands
string directoryPath = Path.Combine(ApiConfiguration.BasePath, "GeneratedFiles"); string directoryPath = Path.Combine(ApiConfiguration.BasePath, "GeneratedFiles");
if (!Directory.Exists(directoryPath)) if (!Directory.Exists(directoryPath))
{ {
Directory.CreateDirectory(directoryPath); _ = Directory.CreateDirectory(directoryPath);
} }
hash = hash.Trim().ToLower(); hash = hash.Trim().ToLower();
@@ -72,7 +72,7 @@ public class HostCommands
string portInput = parts[1]; string portInput = parts[1];
_ = int.TryParse(portInput, out int port); _ = int.TryParse(portInput, out int port);
if (port < 1000 || port >= 6000) if (port is < 1000 or >= 6000)
{ {
ApiManager.SendNotificationMessageNewLine("Invalid port number!", NotificationMessageType.Error); ApiManager.SendNotificationMessageNewLine("Invalid port number!", NotificationMessageType.Error);
return; return;
@@ -112,7 +112,7 @@ public class HostCommands
_ = int.TryParse(portInput, out int port); _ = int.TryParse(portInput, out int port);
if (port < 1000 || port >= 6000) if (port is < 1000 or >= 6000)
{ {
ApiManager.SendNotificationMessageNewLine("Invalid port number!", NotificationMessageType.Error); ApiManager.SendNotificationMessageNewLine("Invalid port number!", NotificationMessageType.Error);
return; return;
@@ -1,6 +1,7 @@
using System.Reflection; using System.Reflection;
namespace HyperbolicDownloaderApi.Managment; namespace HyperbolicDownloaderApi.Managment;
public static class ApiConfiguration public static class ApiConfiguration
{ {
public const int BroadcastPort = 2155; public const int BroadcastPort = 2155;
@@ -9,5 +10,4 @@ public static class ApiConfiguration
public static string BasePath { get; } = Path.GetDirectoryName(Assembly.GetEntryAssembly()!.Location) ?? string.Empty; 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 HostsFilePath { get; } = Path.Combine(BasePath, "Hosts.json");
public static string FilesInfoPath { get; } = Path.Combine(BasePath, "Files.json"); public static string FilesInfoPath { get; } = Path.Combine(BasePath, "Files.json");
}
}
@@ -43,7 +43,7 @@ internal class BroadcastClient
byte[] sendbuf = Encoding.ASCII.GetBytes(message); byte[] sendbuf = Encoding.ASCII.GetBytes(message);
IPEndPoint ep = new IPEndPoint(broadcast, port); IPEndPoint ep = new IPEndPoint(broadcast, port);
socket.SendTo(sendbuf, ep); _ = socket.SendTo(sendbuf, ep);
} }
public void StartListening(int port) public void StartListening(int port)
@@ -58,7 +58,7 @@ internal class BroadcastClient
udpListener = new UdpClient(port); udpListener = new UdpClient(port);
IPEndPoint groupEP = new IPEndPoint(IPAddress.Any, port); IPEndPoint groupEP = new IPEndPoint(IPAddress.Any, port);
Task.Run(() => _ = Task.Run(() =>
{ {
while (IsListening) while (IsListening)
{ {
@@ -1,14 +1,13 @@
namespace HyperbolicDownloaderApi namespace HyperbolicDownloaderApi.Networking;
{
internal class DataContainer
{
public string EventName { get; set; }
public string JsonData { get; set; }
public DataContainer(string eventName, string jsonData) internal class DataContainer
{ {
JsonData = jsonData; public string EventName { get; set; }
EventName = eventName; public string JsonData { get; set; }
}
public DataContainer(string eventName, string jsonData)
{
JsonData = jsonData;
EventName = eventName;
} }
} }
@@ -3,32 +3,31 @@ using System.Net.Sockets;
using System.Text; using System.Text;
using System.Text.Json; using System.Text.Json;
namespace HyperbolicDownloaderApi namespace HyperbolicDownloaderApi.Networking;
internal class MessageRecivedEventArgs<T> : EventArgs
{ {
internal class MessageRecivedEventArgs<T> : EventArgs private readonly NetworkStream networkStream;
public MessageRecivedEventArgs(NetworkStream networkStream, IPAddress ipAddress, T data)
{ {
private readonly NetworkStream networkStream; this.networkStream = networkStream;
Data = data;
IpAddress = ipAddress;
}
public MessageRecivedEventArgs(NetworkStream networkStream, IPAddress ipAddress, T data) public T Data { get; set; }
{
this.networkStream = networkStream;
Data = data;
IpAddress = ipAddress;
}
public T Data { get; set; } public IPAddress IpAddress { get; set; }
public IPAddress IpAddress { get; set; } public async Task SendResponseAsync(object response)
{
byte[] bytesToSend = Encoding.ASCII.GetBytes(JsonSerializer.Serialize(response));
await networkStream.WriteAsync(bytesToSend);
}
public async Task SendResponseAsync(object response) public void SendResponse(object response)
{ {
byte[] bytesToSend = Encoding.ASCII.GetBytes(JsonSerializer.Serialize(response)); SendResponseAsync(response).GetAwaiter().GetResult();
await networkStream.WriteAsync(bytesToSend);
}
public void SendResponse(object response)
{
SendResponseAsync(response).GetAwaiter().GetResult();
}
} }
} }
@@ -6,205 +6,200 @@ using System.Net.Sockets;
using System.Text; using System.Text;
using System.Text.Json; using System.Text.Json;
namespace HyperbolicDownloaderApi.Networking namespace HyperbolicDownloaderApi.Networking;
internal class NetworkClient
{ {
internal class NetworkClient public bool IsListening { get; private set; } = false;
private TcpListener? tcpListener;
private readonly FilesManager filesManager;
private readonly Dictionary<string, (Type type, Delegate method)> events = new();
public NetworkClient(FilesManager filesManager)
{ {
public bool IsListening { get; private set; } = false; this.filesManager = filesManager;
}
private TcpListener? tcpListener; public static async Task<T?> SendAsync<T>(IPAddress remoteIp, int remotePort, string eventName, object data)
private readonly FilesManager filesManager; {
private readonly Dictionary<string, (Type type, Delegate method)> events = new(); if (remoteIp is null)
public NetworkClient(FilesManager filesManager)
{ {
this.filesManager = filesManager; throw new ArgumentNullException(nameof(remoteIp));
} }
public static async Task<T?> SendAsync<T>(IPAddress remoteIp, int remotePort, string eventName, object data) TcpClient client = new TcpClient();
await client.ConnectAsync(remoteIp, remotePort);
NetworkStream nwStream = client.GetStream();
string stringData = JsonSerializer.Serialize(new DataContainer(eventName, JsonSerializer.Serialize(data)));
byte[] bytesToSend = Encoding.ASCII.GetBytes(stringData);
await nwStream.WriteAsync(bytesToSend);
byte[] bytesToRead = new byte[client.ReceiveBufferSize];
int bytesRead = await nwStream.ReadAsync(bytesToRead.AsMemory(0, client.ReceiveBufferSize));
string response = Encoding.ASCII.GetString(bytesToRead, 0, bytesRead);
client.Close();
if (string.IsNullOrWhiteSpace(response))
{ {
if (remoteIp is null) return default;
}
else
{
return JsonSerializer.Deserialize<T>(response);
}
}
public static async Task SendAsync(IPAddress remoteIp, int remotePort, string eventName, object data)
{
if (remoteIp is null)
{
throw new ArgumentNullException(nameof(remoteIp));
}
TcpClient client = new TcpClient();
await client.ConnectAsync(remoteIp, remotePort);
NetworkStream nwStream = client.GetStream();
string stringData = JsonSerializer.Serialize(new DataContainer(eventName, JsonSerializer.Serialize(data)));
byte[] bytesToSend = Encoding.ASCII.GetBytes(stringData);
await nwStream.WriteAsync(bytesToSend);
client.Close();
}
public static T? Send<T>(IPAddress remoteIp, int remotePort, string eventName, object data)
{
return SendAsync<T>(remoteIp, remotePort, eventName, data).GetAwaiter().GetResult();
}
public static void Send(IPAddress remoteIp, int remotePort, string eventName, object data)
{
SendAsync(remoteIp, remotePort, eventName, data).GetAwaiter().GetResult();
}
public void StartListening(int port)
{
if (IsListening)
{
throw new InvalidOperationException("Already listening!");
}
tcpListener = new TcpListener(IPAddress.Any, port);
tcpListener.Start();
IsListening = true;
_ = Task.Run(async () =>
{
while (IsListening)
{ {
throw new ArgumentNullException(nameof(remoteIp)); try
{
TcpClient client = tcpListener.AcceptTcpClient();
NetworkStream nwStream = client.GetStream();
byte[] buffer = new byte[client.ReceiveBufferSize];
int bytesRead = await nwStream.ReadAsync(buffer.AsMemory(0, client.ReceiveBufferSize));
string dataReceived = Encoding.ASCII.GetString(buffer, 0, bytesRead);
if (dataReceived.StartsWith("Download"))
{
_ = Upload(client, dataReceived[8..]);
continue;
}
if (string.IsNullOrWhiteSpace(dataReceived))
{
client.Close();
continue;
}
DataContainer? dataContainer = JsonSerializer.Deserialize<DataContainer>(dataReceived);
if (dataContainer is not null && events.ContainsKey(dataContainer.EventName))
{
(Type type, Delegate method) = events[dataContainer.EventName];
Type eventArgsType = typeof(MessageRecivedEventArgs<>).MakeGenericType(type);
object? eventArgs = Activator.CreateInstance(
eventArgsType,
nwStream,
(client.Client.RemoteEndPoint as IPEndPoint)?.Address,
JsonSerializer.Deserialize(dataContainer.JsonData, type));
_ = (method?.DynamicInvoke(this, eventArgs));
}
client.Close();
}
catch (SocketException ex)
{
if (ex.SocketErrorCode != SocketError.Interrupted)
{
throw;
}
}
} }
tcpListener.Stop();
});
}
TcpClient client = new TcpClient(); private async Task Upload(TcpClient client, string hash)
{
await client.ConnectAsync(remoteIp, remotePort); try
{
byte[] bytesToSend;
hash = hash.Trim();
NetworkStream nwStream = client.GetStream(); NetworkStream nwStream = client.GetStream();
client.SendBufferSize = 64000;
string stringData = JsonSerializer.Serialize(new DataContainer(eventName, JsonSerializer.Serialize(data))); if (filesManager.TryGet(hash, out PrivateHyperFileInfo? hyperFileInfo) && File.Exists(hyperFileInfo?.FilePath))
byte[] bytesToSend = Encoding.ASCII.GetBytes(stringData);
await nwStream.WriteAsync(bytesToSend);
byte[] bytesToRead = new byte[client.ReceiveBufferSize];
int bytesRead = await nwStream.ReadAsync(bytesToRead.AsMemory(0, client.ReceiveBufferSize));
string response = Encoding.ASCII.GetString(bytesToRead, 0, bytesRead);
client.Close();
if (string.IsNullOrWhiteSpace(response))
{ {
return default; FileInfo fileInfo = new FileInfo(hyperFileInfo.FilePath);
bytesToSend = Encoding.ASCII.GetBytes($"{fileInfo.Length}/{Path.GetFileName(hyperFileInfo.FilePath)}");
Array.Resize(ref bytesToSend, 1000);
await nwStream.WriteAsync(bytesToSend);
foreach (byte[]? chunk in FileCompressor.ReadChunks(hyperFileInfo.FilePath, 64000).Where(chunk => chunk is not null))
{
await nwStream.WriteAsync(chunk);
}
} }
else else
{ {
return JsonSerializer.Deserialize<T>(response); bytesToSend = Encoding.ASCII.GetBytes("File not found!");
await nwStream.WriteAsync(bytesToSend);
} }
} }
catch (Exception ex)
public static async Task SendAsync(IPAddress remoteIp, int remotePort, string eventName, object data) {
Debug.WriteLine(ex);
}
finally
{ {
if (remoteIp is null)
{
throw new ArgumentNullException(nameof(remoteIp));
}
TcpClient client = new TcpClient();
await client.ConnectAsync(remoteIp, remotePort);
NetworkStream nwStream = client.GetStream();
string stringData = JsonSerializer.Serialize(new DataContainer(eventName, JsonSerializer.Serialize(data)));
byte[] bytesToSend = Encoding.ASCII.GetBytes(stringData);
await nwStream.WriteAsync(bytesToSend);
client.Close(); client.Close();
} }
}
public static T? Send<T>(IPAddress remoteIp, int remotePort, string eventName, object data) public void StopListening()
{ {
return SendAsync<T>(remoteIp, remotePort, eventName, data).GetAwaiter().GetResult(); tcpListener?.Stop();
} IsListening = false;
}
public static void Send(IPAddress remoteIp, int remotePort, string eventName, object data) public void ListenTo<T>(string eventName, EventHandler<MessageRecivedEventArgs<T>> eventHandler)
{ {
SendAsync(remoteIp, remotePort, eventName, data).GetAwaiter().GetResult(); events.Add(eventName, (typeof(T), eventHandler));
}
public void StartListening(int port)
{
if (IsListening)
{
throw new InvalidOperationException("Already listening!");
}
tcpListener = new TcpListener(IPAddress.Any, port);
tcpListener.Start();
IsListening = true;
Task.Run(async () =>
{
while (IsListening)
{
try
{
TcpClient client = tcpListener.AcceptTcpClient();
NetworkStream nwStream = client.GetStream();
byte[] buffer = new byte[client.ReceiveBufferSize];
int bytesRead = await nwStream.ReadAsync(buffer.AsMemory(0, client.ReceiveBufferSize));
string dataReceived = Encoding.ASCII.GetString(buffer, 0, bytesRead);
if (dataReceived.StartsWith("Download"))
{
_ = Upload(client, dataReceived[8..]);
continue;
}
if (string.IsNullOrWhiteSpace(dataReceived))
{
client.Close();
continue;
}
DataContainer? dataContainer = JsonSerializer.Deserialize<DataContainer>(dataReceived);
if (dataContainer is not null && events.ContainsKey(dataContainer.EventName))
{
(Type type, Delegate method) = events[dataContainer.EventName];
Type eventArgsType = typeof(MessageRecivedEventArgs<>).MakeGenericType(type);
object? eventArgs = Activator.CreateInstance(
eventArgsType,
nwStream,
(client.Client.RemoteEndPoint as IPEndPoint)?.Address,
JsonSerializer.Deserialize(dataContainer.JsonData, type));
method?.DynamicInvoke(this, eventArgs);
}
client.Close();
}
catch (SocketException ex)
{
if (ex.SocketErrorCode != SocketError.Interrupted)
{
throw;
}
}
}
tcpListener.Stop();
});
}
private async Task Upload(TcpClient client, string hash)
{
try
{
byte[] bytesToSend;
hash = hash.Trim();
NetworkStream nwStream = client.GetStream();
client.SendBufferSize = 64000;
if (filesManager.TryGet(hash, out PrivateHyperFileInfo? hyperFileInfo) && File.Exists(hyperFileInfo?.FilePath))
{
FileInfo fileInfo = new FileInfo(hyperFileInfo.FilePath);
bytesToSend = Encoding.ASCII.GetBytes($"{fileInfo.Length}/{Path.GetFileName(hyperFileInfo.FilePath)}");
Array.Resize(ref bytesToSend, 1000);
await nwStream.WriteAsync(bytesToSend);
foreach (byte[]? chunk in FileCompressor.ReadChunks(hyperFileInfo.FilePath, 64000))
{
if (chunk is not null)
{
await nwStream.WriteAsync(chunk);
}
}
}
else
{
bytesToSend = Encoding.ASCII.GetBytes("File not found!");
await nwStream.WriteAsync(bytesToSend);
}
}
catch (Exception ex)
{
Debug.WriteLine(ex);
}
finally
{
client.Close();
}
}
public void StopListening()
{
tcpListener?.Stop();
IsListening = false;
}
public void ListenTo<T>(string eventName, EventHandler<MessageRecivedEventArgs<T>> eventHandler)
{
events.Add(eventName, (typeof(T), eventHandler));
}
} }
} }