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

209 lines
8.6 KiB
C#
Raw Normal View History

2023-10-10 09:56:08 +00:00
using System.Buffers.Binary;
using System.Net.Sockets;
2023-10-14 15:01:49 +00:00
using AscNet.Common;
using AscNet.Common.Util;
using AscNet.Logging;
2023-10-10 09:56:08 +00:00
using MessagePack;
using Newtonsoft.Json;
using Logger = AscNet.Logging.Logger;
namespace AscNet.GameServer
{
public class Session
{
public readonly string id;
public readonly TcpClient client;
public readonly Logger log;
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;
// TODO: add session based configuration? maybe from database?
log = new(typeof(Session), LogLevel.DEBUG, LogLevel.DEBUG);
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)
{
log.Error("Failed to deserialize packet: " + BitConverter.ToString(packet).Replace("-", ""));
2023-10-10 09:56:08 +00:00
}
}
}
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)
{
// TODO: with new logger this will be unnecessary
2023-10-14 15:01:49 +00:00
if (Common.Common.config.VerboseLevel > VerboseLevel.Silent)
log.Info($"{request.Name}{(Common.Common.config.VerboseLevel >= VerboseLevel.Debug ? (", " + JsonConvert.SerializeObject(MessagePackSerializer.Typeless.Deserialize(request.Content))) : "")}");
requestPacketHandler.Invoke(this, request);
}
else
2023-10-14 15:01:49 +00:00
{
if (Common.Common.config.VerboseLevel > VerboseLevel.Silent)
log.Warn($"{request.Name} handler not found!{(Common.Common.config.VerboseLevel >= VerboseLevel.Debug ? (", " + JsonConvert.SerializeObject(MessagePackSerializer.Typeless.Deserialize(request.Content))) : "")}");
2023-10-14 15:01:49 +00:00
}
2023-10-11 13:20:26 +00:00
break;
2023-10-11 13:20:26 +00:00
case Packet.ContentType.Push:
Packet.Push push = MessagePackSerializer.Deserialize<Packet.Push>(packet.Content);
debugContent = push.Content;
log.Info(push.Name);
throw new NotImplementedException($"Packet push handlers not implemented ({push.Name})");
2023-10-11 13:20:26 +00:00
case Packet.ContentType.Exception:
Packet.Exception exception = MessagePackSerializer.Deserialize<Packet.Exception>(packet.Content);
log.Error($"Exception packet received: {exception.Code}, {exception.Message}");
2023-10-11 13:20:26 +00:00
break;
2023-10-11 13:20:26 +00:00
default:
log.Error($"Unknown packet received: {packet}");
2023-10-11 13:20:26 +00:00
break;
}
}
catch (Exception ex)
{
log.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 13:07:07 +00:00
public void SendPush<T>(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 13:07:07 +00:00
Name = typeof(T).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)
});
log.Info(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)
});
log.Info(packet.Name);
}
2023-10-14 17:34:14 +00:00
public void SendResponse<T>(T response, int clientSeq = 0) where T : new()
2023-10-10 09:56:08 +00:00
{
Packet.Response packet = new()
{
Id = clientSeq,
2023-10-14 13:07:07 +00:00
Name = typeof(T).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)
});
log.Info(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;
log.Warn($"{id} disconnected");
client.Close();
Server.Instance.Sessions.Remove(id);
}
}
}