From 1f914618538920db4bfec7b106ee97038b157c9b Mon Sep 17 00:00:00 2001 From: Wim Date: Thu, 22 Jun 2017 01:00:27 +0200 Subject: Add vendor (steam) --- .../Philipp15b/go-steam/protocol/packet.go | 116 +++++++++++++++++++++ 1 file changed, 116 insertions(+) create mode 100644 vendor/github.com/Philipp15b/go-steam/protocol/packet.go (limited to 'vendor/github.com/Philipp15b/go-steam/protocol/packet.go') diff --git a/vendor/github.com/Philipp15b/go-steam/protocol/packet.go b/vendor/github.com/Philipp15b/go-steam/protocol/packet.go new file mode 100644 index 00000000..d6b2e425 --- /dev/null +++ b/vendor/github.com/Philipp15b/go-steam/protocol/packet.go @@ -0,0 +1,116 @@ +package protocol + +import ( + "bytes" + "github.com/golang/protobuf/proto" + "encoding/binary" + "fmt" + . "github.com/Philipp15b/go-steam/protocol/steamlang" +) + +// TODO: Headers are always deserialized twice. + +// Represents an incoming, partially unread message. +type Packet struct { + EMsg EMsg + IsProto bool + TargetJobId JobId + SourceJobId JobId + Data []byte +} + +func NewPacket(data []byte) (*Packet, error) { + var rawEMsg uint32 + err := binary.Read(bytes.NewReader(data), binary.LittleEndian, &rawEMsg) + if err != nil { + return nil, err + } + eMsg := NewEMsg(rawEMsg) + buf := bytes.NewReader(data) + if eMsg == EMsg_ChannelEncryptRequest || eMsg == EMsg_ChannelEncryptResult { + header := NewMsgHdr() + header.Msg = eMsg + err = header.Deserialize(buf) + if err != nil { + return nil, err + } + return &Packet{ + EMsg: eMsg, + IsProto: false, + TargetJobId: JobId(header.TargetJobID), + SourceJobId: JobId(header.SourceJobID), + Data: data, + }, nil + } else if IsProto(rawEMsg) { + header := NewMsgHdrProtoBuf() + header.Msg = eMsg + err = header.Deserialize(buf) + if err != nil { + return nil, err + } + return &Packet{ + EMsg: eMsg, + IsProto: true, + TargetJobId: JobId(header.Proto.GetJobidTarget()), + SourceJobId: JobId(header.Proto.GetJobidSource()), + Data: data, + }, nil + } else { + header := NewExtendedClientMsgHdr() + header.Msg = eMsg + err = header.Deserialize(buf) + if err != nil { + return nil, err + } + return &Packet{ + EMsg: eMsg, + IsProto: false, + TargetJobId: JobId(header.TargetJobID), + SourceJobId: JobId(header.SourceJobID), + Data: data, + }, nil + } +} + +func (p *Packet) String() string { + return fmt.Sprintf("Packet{EMsg = %v, Proto = %v, Len = %v, TargetJobId = %v, SourceJobId = %v}", p.EMsg, p.IsProto, len(p.Data), p.TargetJobId, p.SourceJobId) +} + +func (p *Packet) ReadProtoMsg(body proto.Message) *ClientMsgProtobuf { + header := NewMsgHdrProtoBuf() + buf := bytes.NewBuffer(p.Data) + header.Deserialize(buf) + proto.Unmarshal(buf.Bytes(), body) + return &ClientMsgProtobuf{ // protobuf messages have no payload + Header: header, + Body: body, + } +} + +func (p *Packet) ReadClientMsg(body MessageBody) *ClientMsg { + header := NewExtendedClientMsgHdr() + buf := bytes.NewReader(p.Data) + header.Deserialize(buf) + body.Deserialize(buf) + payload := make([]byte, buf.Len()) + buf.Read(payload) + return &ClientMsg{ + Header: header, + Body: body, + Payload: payload, + } +} + +func (p *Packet) ReadMsg(body MessageBody) *Msg { + header := NewMsgHdr() + buf := bytes.NewReader(p.Data) + header.Deserialize(buf) + body.Deserialize(buf) + payload := make([]byte, buf.Len()) + buf.Read(payload) + return &Msg{ + Header: header, + Body: body, + Payload: payload, + } +} -- cgit v1.2.3