forked from PGR/ascnet
1
0
Fork 0
ascnet/AscNet.GameServer/Session.cs

198 lines
7.7 KiB
C#
Raw Normal View History

2023-10-10 09:56:08 +00:00
using System.Buffers.Binary;
using System.Net.Sockets;
using AscNet.Common.Util;
2023-10-10 09:56:08 +00:00
using MessagePack;
using Newtonsoft.Json;
namespace AscNet.GameServer
{
public class Session
{
public readonly string id;
public readonly TcpClient client;
public readonly Logger c;
private long lastPacketTime = DateTimeOffset.UtcNow.ToUnixTimeMilliseconds();
private ushort packetNo = 0;
2023-10-10 09:56:08 +00:00
private readonly MessagePackSerializerOptions lz4Options = MessagePackSerializerOptions.Standard.WithCompression(MessagePackCompression.Lz4Block);
public Session(string id, TcpClient tcpClient)
{
this.id = id;
client = tcpClient;
c = new(id, ConsoleColor.DarkGray);
Task.Run(ClientLoop);
}
public async void ClientLoop()
{
NetworkStream stream = client.GetStream();
byte[] msg = new byte[1 << 16];
while (client.Connected)
{
try
{
Array.Clear(msg, 0, msg.Length);
int len = stream.Read(msg, 0, msg.Length);
if (len > 0)
{
lastPacketTime = DateTimeOffset.UtcNow.ToUnixTimeMilliseconds();
2023-10-10 09:56:08 +00:00
List<Packet> packets = new();
2023-10-10 09:56:08 +00:00
int readbytes = 0;
while (readbytes < len)
{
int packetLen = BinaryPrimitives.ReadInt32LittleEndian(msg.AsSpan()[readbytes..]);
readbytes += 4;
if (packetLen < 4)
break;
else
{
byte[] packet = GC.AllocateUninitializedArray<byte>(packetLen);
Array.Copy(msg, readbytes, packet, 0, packetLen);
readbytes += packetLen;
Crypto.HaruCrypt.Decrypt(packet);
try
{
packets.Add(MessagePackSerializer.Deserialize<Packet>(packet, lz4Options));
}
catch (Exception)
{
c.Error("Failed to deserialize packet: " + BitConverter.ToString(packet).Replace("-", ""));
}
}
}
foreach (var packet in packets)
{
2023-10-11 13:20:26 +00:00
byte[] debugContent = packet.Content;
try
2023-10-10 09:56:08 +00:00
{
2023-10-11 13:20:26 +00:00
switch (packet.Type)
{
case Packet.ContentType.Request:
Packet.Request request = MessagePackSerializer.Deserialize<Packet.Request>(packet.Content);
debugContent = request.Content;
RequestPacketHandlerDelegate? requestPacketHandler = PacketFactory.GetRequestPacketHandler(request.Name);
if (requestPacketHandler is not null)
{
c.Log(request.Name);
requestPacketHandler.Invoke(this, request);
}
else
c.Warn($"{request.Name} handler not found!");
2023-10-11 13:20:26 +00:00
break;
case Packet.ContentType.Push:
Packet.Push push = MessagePackSerializer.Deserialize<Packet.Push>(packet.Content);
debugContent = push.Content;
c.Log(push.Name);
throw new NotImplementedException($"Packet push handlers not implemented ({push.Name})");
2023-10-11 13:20:26 +00:00
break;
case Packet.ContentType.Exception:
Packet.Exception exception = MessagePackSerializer.Deserialize<Packet.Exception>(packet.Content);
c.Error($"Exception packet received: {exception.Code}, {exception.Message}");
break;
default:
c.Error($"Unknown packet received: {packet}");
break;
}
}
catch (Exception ex)
{
c.Error("Failed to invoke handler: " + ex.Message + $", Raw {packet.Type} packet: " + BitConverter.ToString(debugContent).Replace("-", ""));
2023-10-10 09:56:08 +00:00
}
}
}
}
catch (Exception)
{
break;
}
await Task.Delay(10);
// 10 sec timeout
if (DateTimeOffset.UtcNow.ToUnixTimeMilliseconds() - lastPacketTime > 10000)
break;
}
DisconnectProtocol();
}
2023-10-14 01:23:16 +00:00
public void SendPush<T>(string name, T push)
2023-10-10 09:56:08 +00:00
{
Packet.Push packet = new()
2023-10-10 09:56:08 +00:00
{
2023-10-14 01:23:16 +00:00
Name = name,
Content = MessagePackSerializer.Serialize(push)
};
Send(new Packet()
2023-10-10 09:56:08 +00:00
{
No = ++packetNo,
Type = Packet.ContentType.Push,
Content = MessagePackSerializer.Serialize(packet)
});
c.Log(packet.Name);
2023-10-10 09:56:08 +00:00
}
public void SendPush(string name, byte[] push)
{
Packet.Push packet = new()
{
Name = name,
Content = push
};
Send(new Packet()
{
No = ++packetNo,
Type = Packet.ContentType.Push,
Content = MessagePackSerializer.Serialize(packet)
});
c.Log(packet.Name);
}
public void SendResponse<T>(T response, int clientSeq = 0)
2023-10-10 09:56:08 +00:00
{
Packet.Response packet = new()
{
Id = clientSeq,
Name = response!.GetType().Name,
2023-10-10 09:56:08 +00:00
Content = MessagePackSerializer.Serialize(response)
};
Send(new Packet()
{
No = 0,
2023-10-10 09:56:08 +00:00
Type = Packet.ContentType.Response,
Content = MessagePackSerializer.Serialize(packet)
});
c.Log(packet.Name);
2023-10-10 09:56:08 +00:00
}
private void Send(Packet packet)
{
byte[] serializedPacket = MessagePackSerializer.Serialize(packet, lz4Options);
2023-10-10 10:10:29 +00:00
Crypto.HaruCrypt.Encrypt(serializedPacket);
2023-10-10 09:56:08 +00:00
byte[] sendBytes = GC.AllocateUninitializedArray<byte>(serializedPacket.Length + 4);
2023-10-10 10:10:29 +00:00
BinaryPrimitives.WriteInt32LittleEndian(sendBytes.AsSpan()[0..4], serializedPacket.Length);
2023-10-10 09:56:08 +00:00
Array.Copy(serializedPacket, 0, sendBytes, 4, serializedPacket.Length);
client.GetStream().Write(sendBytes);
}
public void DisconnectProtocol()
{
if (Server.Instance.Sessions.GetValueOrDefault(id) is null)
return;
c.Warn($"{id} disconnected");
client.Close();
Server.Instance.Sessions.Remove(id);
}
}
}