public byte[] Encode(object message, ref short serial)
{
var factory = _factories.FirstOrDefault(f => f.ContainsType(message.GetType()));
byte[] result = null;
if (factory == null)
{
throw new InvalidProgramException($"Invalid message type {message.GetType()}");
}
ushort opCode = factory.GetOpCode(message.GetType());
var att = message
.GetType()
.GetCustomAttributes(false)
.Where(x => x.GetType() == typeof(WZContractAttribute))
.FirstOrDefault() as WZContractAttribute;
if (att == null)
{
throw new InvalidOperationException("Invalid message format");
}
using (var h = new MemoryStream())
using (var b = new MemoryStream())
{
Serializer.Serialize(b, message);
var body = b.ToArray();
var length = body.Length;
var sizeFix = (((opCode & 0xFF00) == 0xFF00) ? 1 : 0);
if (att.LongMessage)
{
var header = new WZWPacket
{
Type = (byte)0xC2,
Size = (ushort)(length + 5 - sizeFix),
Operation = opCode
};
if (att.Serialized)
{
var temp = new byte[header.Size - 2];
temp[0] = (byte)serial;
Array.Copy(body, 0, temp, 1, body.Length);
body = SimpleModulus.Encoder(temp);
header.Type += 2;
serial++;
}
Serializer.Serialize(h, header);
}
else
{
var header = new WZBPacket
{
Type = (byte)0xC1,
Size = (byte)(length + 4 - sizeFix),
Operation = opCode
};
if (att.Serialized)
{
var temp = new byte[header.Size - 1];
temp[0] = (byte)serial;
Array.Copy(body, 0, temp, 1, body.Length);
body = SimpleModulus.Encoder(temp);
header.Type += 2;
serial++;
}
Serializer.Serialize(h, header);
}
var head = h.ToArray();
var headLen = head.Length - sizeFix;
result = new byte[headLen + body.Length];
Array.Copy(head, result, headLen);
Array.Copy(body, 0, result, headLen, body.Length);
using (var spe = new StreamPacketEngine(false))
{
spe.AddData(result);
result = spe.ExtractData();
}
}
return(result);
}
public int Decode(MemoryStream rawPacket, out short serial, List <object> messages)
{
using (var decPacket = new MemoryStream())
using (var posPacket = new MemoryStream())
{
//var type = rawPacket[0];
var type = (byte)rawPacket.ReadByte();
ushort size = 0;
serial = -1;
if (type == 0xC1 || type == 0xC3)
{
size = (ushort)rawPacket.ReadByte();
}
else if (type == 0xC2 || type == 0xC4)
{
size = (ushort)((rawPacket.ReadByte() << 8) | rawPacket.ReadByte());
}
else
{
throw new InvalidProgramException($"Invalid packet type {type:X2}");
}
// Packet Encrypted
byte[] tmp;
byte[] dec;
switch (type)
{
case 0xC1:
decPacket.WriteByte(type);
decPacket.WriteByte((byte)size);
tmp = new byte[size - 2];
rawPacket.Read(tmp, 0, tmp.Length);
decPacket.Write(tmp, 0, tmp.Length);
break;
case 0xC2:
decPacket.WriteByte(type);
decPacket.WriteByte((byte)((size >> 8) & 0xff));
decPacket.WriteByte((byte)(size & 0xff));
tmp = new byte[size - 3];
rawPacket.Read(tmp, 0, tmp.Length);
decPacket.Write(tmp, 0, tmp.Length);
break;
case 0xC3:
tmp = new byte[size - 2];
rawPacket.Read(tmp, 0, tmp.Length);
if (_packetRijndael)
{
dec = PacketEncrypt.Decrypt(tmp);
if (dec.Length == 0)
{
return(size);
}
decPacket.WriteByte(0xC1);
decPacket.WriteByte((byte)(dec.Length + 2));
decPacket.Write(dec, 0, dec.Length);
}
else
{
dec = SimpleModulus.Decoder(tmp);
serial = dec[0];
decPacket.WriteByte(0xC1);
decPacket.WriteByte((byte)(dec.Length + 1));
decPacket.Write(dec, 1, dec.Length - 1);
}
break;
case 0xC4:
tmp = new byte[size - 3];
rawPacket.Read(tmp, 0, tmp.Length);
if (_packetRijndael)
{
dec = PacketEncrypt.Decrypt(tmp);
if (dec.Length == 0)
{
return(size);
}
decPacket.WriteByte(0xC2);
decPacket.WriteByte((byte)((dec.Length + 3) >> 8));
decPacket.WriteByte((byte)((dec.Length + 3) & 255));
decPacket.Write(dec, 0, dec.Length);
}
else
{
dec = SimpleModulus.Decoder(tmp);
serial = dec[0];
decPacket.WriteByte(0xC2);
decPacket.WriteByte((byte)((dec.Length + 2) >> 8));
decPacket.WriteByte((byte)((dec.Length + 2) & 255));
decPacket.Write(dec, 1, dec.Length - 1);
}
break;
}
using (var spe = new StreamPacketEngine(_packetRijndael))
{
spe.AddData(decPacket.ToArray());
var posProcess = spe.ExtractPacket();
posPacket.Write(posProcess, 0, posProcess.Length);
}
posPacket.Seek(0, SeekOrigin.Begin);
ushort opCode;
ushort pkSize;
if (type == 0xC1 || type == 0xC3)
{
if (posPacket.Length == 3)
{
posPacket.Seek(0, SeekOrigin.End);
posPacket.WriteByte(0);
posPacket.Seek(0, SeekOrigin.Begin);
}
else if (posPacket.Length < 3)
{
throw new Exception("Invalid Packet " + type.ToString("X2") + " size " + posPacket.Length + " -");
}
var tmph = Serializer.Deserialize <WZBPacket>(posPacket);
opCode = tmph.Operation;
pkSize = tmph.Size;
}
else
{
if (posPacket.Length == 4)
{
posPacket.Seek(0, SeekOrigin.End);
posPacket.WriteByte(0);
posPacket.Seek(0, SeekOrigin.Begin);
}
else if (posPacket.Length < 4)
{
throw new Exception("Invalid Packet " + type.ToString("X2") + " size " + posPacket.Length + " -");
}
var tmph = Serializer.Deserialize <WZWPacket>(posPacket);
opCode = tmph.Operation;
pkSize = tmph.Size;
}
//posPacket.Seek(0, SeekOrigin.Begin);
var factory = _factories.FirstOrDefault(f => f.ContainsOpCode(opCode));
try
{
if (factory != null)
{
messages.Add(factory.GetMessage(opCode, posPacket));
}
else
{
var orgOpCode = opCode;
opCode |= 0xFF00;
factory = _factories.FirstOrDefault(f => f.ContainsOpCode(opCode));
if (factory != null)
{
posPacket.Position--;
messages.Add(factory.GetMessage(opCode, posPacket));
}
else
{
Logger.Error("Invalid OpCoder {opCodea:X4}|{opCode:X2}", orgOpCode, opCode & 0xff);
}
}
}catch (Exception ex)
{
throw new Exception("Factory: " + factory, ex);
}
return(size);
}
}
