/* Sog 游戏基础库 2016 by zouwei */ using System; using System.Net; using System.Text; using System.Collections.Generic; using Google.Protobuf.WellKnownTypes; using Google.Protobuf; namespace SogClient { public class ProtoBufPacker : IProtoPacker { // todo 效率不高,换成数组 private Dictionary m_typeMessageParser = new Dictionary(); public ProtoBufPacker() { } //最小4 public int GetHeaderLength() { return 8; } public int GetFullHeaderLength() { return 16; } //打包头 public int PackHeader(MessageHeader header, byte[] headerBytes) { //小字节序 if(header.ObjectID == 0) { headerBytes[0] = (byte)(header.Length % 256); headerBytes[1] = (byte)(header.Length / 256); headerBytes[2] = (byte)(header.Type % 256); headerBytes[3] = (byte)(header.Type / 256); Span pbServerID = new Span(headerBytes, 4, 4); BitConverter.TryWriteBytes(pbServerID, header.ServerID); return 8; } else { headerBytes[0] = (byte)(header.Length % 256); headerBytes[1] = (byte)(header.Length / 256); //最高位置1 int newType = header.Type | 0x8000; headerBytes[2] = (byte)(newType % 256); headerBytes[3] = (byte)(newType / 256); Span pbServerID = new Span(headerBytes, 4, 4); BitConverter.TryWriteBytes(pbServerID, header.ServerID); Span pbObjectID = new Span(headerBytes, 8, 8); BitConverter.TryWriteBytes(pbObjectID, header.ObjectID); return 16; } } //解包头 public int UnPackHeader(byte[] data, int offset, int length, ref MessageHeader header) { int byte0 = data[offset + 0]; int byte1 = data[offset + 1]; int byte2 = data[offset + 2]; int byte3 = data[offset + 3]; header.Length = byte1 * 256 + byte0; header.Type = byte3 * 256 + byte2; header.ServerID = BitConverter.ToUInt32(data, offset + 4); bool haveObjectID = (header.Type & 0x00008000) > 0; if (!haveObjectID) { return 8; } //头有16字节 if (length < 16) { return 0; } //去掉最高位 header.Type &= 0x00007fff; header.ObjectID = BitConverter.ToInt64(data, offset + 8); return 16; } public virtual bool UnpackMessage(MessageData message, out RequestPacket packet) { try { MessageParser parser; m_typeMessageParser.TryGetValue(message.Header.Type,out parser); IMessage pbmessage = null; packet = new RequestPacket(message.Header.Type, 0, 0); if (parser != null) { pbmessage = parser.ParseFrom(message.Data); } else { TraceLog.Trace("protobuf decode package type {0} error, can not find MessageParser,please register it first!", message.Header.Type); return false; } packet.Packet = pbmessage; return true; } catch(Exception error) { TraceLog.Error("protobuf decode package type {0} error, {1}", message.Header.Type, error.Message); packet = null; } return false; } public virtual bool PackMessage(RequestPacket packet, ref MessageData message) { try { message.Data = MessageExtensions.ToByteArray((IMessage)packet.Packet); message.Header.Length = message.Data.Length; message.Header.Type = packet.MsgID; return true; } catch (Exception error) { TraceLog.Error("protobuf encode package type {0} error, {1}", message.Header.Type, error.Message); packet = null; } return false; } //需要注册,提高效率 public void RegisterProtoType(int iMsgID, object obj) { MessageParser parser = (MessageParser)obj; m_typeMessageParser.Add(iMsgID, parser); } } }