public PacketLogWriter(MessageTables messages, ServerInfo[] servers,
string directory, string fileNameFormat, bool compress)
{
if (servers == null)
{
throw new ArgumentNullException(nameof(servers));
}
if (servers.Any(x => x == null))
{
throw new ArgumentException("A null server was given.",
nameof(servers));
}
if (fileNameFormat == null)
{
throw new ArgumentNullException(nameof(fileNameFormat));
}
IsCompressed = compress;
Messages = messages ?? throw new ArgumentNullException(nameof(messages));
Servers = servers.ToDictionary(x => x.Id);
Directory.CreateDirectory(directory);
Stream stream = File.Open(
Path.Combine(directory, DateTime.Now.ToString(fileNameFormat) + ".pkt"),
FileMode.Create, FileAccess.Write);
var magic = PacketLogEntry.Magic.ToArray();
stream.Write(magic, 0, magic.Length);
stream.WriteByte((byte)(compress ? 6 : 0));
if (compress)
{
stream = new DeflateStream(stream, CompressionLevel.Optimal);
}
_writer = new TeraBinaryWriter(stream);
_writer.WriteUInt32(Version);
_writer.WriteByte((byte)messages.Region);
_writer.WriteUInt32(messages.Game.Version);
_writer.WriteUInt32((uint)servers.Length);
foreach (var server in servers)
{
_writer.WriteInt32(server.Id);
_writer.WriteString(server.Name);
_writer.WriteBoolean(server.RealEndPoint.AddressFamily ==
AddressFamily.InterNetworkV6);
_writer.WriteBytes(server.RealEndPoint.Address.GetAddressBytes());
_writer.WriteUInt16((ushort)server.RealEndPoint.Port);
_writer.WriteBytes(server.ProxyEndPoint.Address.GetAddressBytes());
_writer.WriteUInt16((ushort)server.ProxyEndPoint.Port);
}
}
public void Write(PacketLogEntry entry)
{
if (entry == null)
{
throw new ArgumentNullException(nameof(entry));
}
if (!Servers.ContainsKey(entry.ServerId))
{
throw new ArgumentException("Invalid server ID.", nameof(entry));
}
if (!Messages.Game.OpCodeToName.ContainsKey(entry.OpCode))
{
throw new ArgumentException("Invalid opcode.", nameof(entry));
}
if (_disposed)
{
throw new ObjectDisposedException(GetType().FullName);
}
_writer.WriteInt64(new DateTimeOffset(entry.Timestamp).ToUnixTimeMilliseconds());
_writer.WriteInt32(entry.ServerId);
_writer.WriteByte((byte)entry.Direction);
_writer.WriteUInt16(entry.OpCode);
_writer.WriteUInt16((ushort)entry.Payload.Count);
_writer.WriteBytes(entry.Payload.ToArray());
}
void SerializeObject(TeraBinaryWriter writer, object source)
{
var fields = GetPacketFields <ReflectionPacketFieldInfo>(source.GetType());
var offsets = new List <(ReflectionPacketFieldInfo, int)>();
foreach (var info in fields)
{
if (info.IsByteArray)
{
offsets.Add((info, writer.Position));
writer.WriteUInt16(0);
writer.WriteUInt16((ushort)((List <byte>)info.Property
.GetValue(source)).Count);
}
else if (info.IsArray)
{
writer.WriteUInt16(
(ushort)((IList)info.Property.GetValue(source)).Count);
offsets.Add((info, writer.Position));
writer.WriteUInt16(0);
}
else if (info.IsString)
{
offsets.Add((info, writer.Position));
writer.WriteUInt16(0);
}
else
{
info.PrimitiveSerializer(writer, info.Property.GetValue(source));
}
}
foreach (var(info, offset) in offsets)
{
if (info.IsByteArray)
{
var list = (List <byte>)info.Property.GetValue(source);
if (list.Count == 0)
{
continue;
}
writer.Seek(offset, (w, op) => w.WriteOffset(op));
writer.WriteBytes(list.ToArray());
}
else if (info.IsArray)
{
var list = (IList)info.Property.GetValue(source);
if (list.Count == 0)
{
continue;
}
writer.Seek(offset, (w, op) => w.WriteOffset(op));
for (var i = 0; i < list.Count; i++)
{
var pos = writer.Position;
writer.WriteOffset(pos);
writer.WriteUInt16(0);
SerializeObject(writer, list[i]);
if (i != list.Count - 1)
{
writer.Seek(pos + sizeof(ushort), (w, op) => w.WriteOffset(op));
}
}
}
else
{
writer.Seek(offset, (w, op) => w.WriteOffset(op));
writer.WriteString((string)info.Property.GetValue(source) ?? string.Empty);
}
}
}
