mirror of
https://github.com/Stone-Red-Code/HyperbolicDownloader.git
synced 2026-09-04 00:56:09 +02:00
Code cleanup
This commit is contained in:
@@ -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()
|
||||||
|
|||||||
@@ -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));
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
Reference in New Issue
Block a user