aboutsummaryrefslogtreecommitdiff
path: root/pkg/classes/packet.go
blob: c4f9676c718974aa244de924e539c3f1e4622522 (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
package classes

import (
	"fmt"

	"github.com/pektezol/bitreader"
	"github.com/pektezol/sdp.go/pkg/messages"
	"github.com/pektezol/sdp.go/pkg/types"
)

const MSSC int = 2

type Packet struct {
	PacketInfo  []CmdInfo `json:"packet_info"`
	InSequence  uint32    `json:"in_sequence"`
	OutSequence uint32    `json:"out_sequence"`
	Size        uint32    `json:"size"`
	Data        []any     `json:"data"`
}

type CmdInfo struct {
	Flags            uint32    `json:"flags"`
	ViewOrigin       []float32 `json:"view_origin"`
	ViewAngles       []float32 `json:"view_angles"`
	LocalViewAngles  []float32 `json:"local_view_angles"`
	ViewOrigin2      []float32 `json:"view_origin_2"`
	ViewAngles2      []float32 `json:"view_angles_2"`
	LocalViewAngles2 []float32 `json:"local_view_angles_2"`
}

func (packet *Packet) ParsePacket(reader *bitreader.Reader, demo *types.Demo) {
	for count := 0; count < MSSC; count++ {
		packet.ParseCmdInfo(reader, demo)
	}
	packet.InSequence = reader.TryReadUInt32()
	packet.OutSequence = reader.TryReadUInt32()
	packet.Size = reader.TryReadUInt32()
	packetReader := bitreader.NewReaderFromBytes(reader.TryReadBytesToSlice(uint64(packet.Size)), true)
	for {
		messageType, err := packetReader.ReadBits(6)
		if err != nil {
			break
		}
		packet.Data = append(packet.Data, messages.ParseMessages(messageType, packetReader, demo))
	}
}

func (packet *Packet) ParseCmdInfo(reader *bitreader.Reader, demo *types.Demo) {
	packet.PacketInfo = append(packet.PacketInfo, CmdInfo{
		Flags:            reader.TryReadUInt32(),
		ViewOrigin:       []float32{reader.TryReadFloat32(), reader.TryReadFloat32(), reader.TryReadFloat32()},
		ViewAngles:       []float32{reader.TryReadFloat32(), reader.TryReadFloat32(), reader.TryReadFloat32()},
		LocalViewAngles:  []float32{reader.TryReadFloat32(), reader.TryReadFloat32(), reader.TryReadFloat32()},
		ViewOrigin2:      []float32{reader.TryReadFloat32(), reader.TryReadFloat32(), reader.TryReadFloat32()},
		ViewAngles2:      []float32{reader.TryReadFloat32(), reader.TryReadFloat32(), reader.TryReadFloat32()},
		LocalViewAngles2: []float32{reader.TryReadFloat32(), reader.TryReadFloat32(), reader.TryReadFloat32()},
	})
	demo.Writer.AppendLine("\tFlags: %s", CmdInfoFlags(packet.PacketInfo[len(packet.PacketInfo)-1].Flags).String())
	demo.Writer.AppendLine("\tView Origin: %v", packet.PacketInfo[len(packet.PacketInfo)-1].ViewOrigin)
	demo.Writer.AppendLine("\tView Angles: %v", packet.PacketInfo[len(packet.PacketInfo)-1].ViewAngles)
	demo.Writer.AppendLine("\tLocal View Angles: %v", packet.PacketInfo[len(packet.PacketInfo)-1].LocalViewAngles)
	demo.Writer.AppendLine("\tView Origin 2: %v", packet.PacketInfo[len(packet.PacketInfo)-1].ViewOrigin2)
	demo.Writer.AppendLine("\tView Angles 2: %v", packet.PacketInfo[len(packet.PacketInfo)-1].ViewAngles2)
	demo.Writer.AppendLine("\tLocal View Angles 2: %v", packet.PacketInfo[len(packet.PacketInfo)-1].LocalViewAngles2)
	demo.Writer.AppendLine("")
}

type CmdInfoFlags int

const (
	ECmdInfoFlagsNone        = 0
	ECmdInfoFlagsUseOrigin2  = 1
	ECmdInfoFlagsUserAngles2 = 1 << 1
	ECmdInfoFlagsNoInterp    = 1 << 2
)

func (cmdInfoFlags CmdInfoFlags) String() string {
	switch cmdInfoFlags {
	case ECmdInfoFlagsNone:
		return "None"
	case ECmdInfoFlagsUseOrigin2:
		return "UseOrigin2"
	case ECmdInfoFlagsUserAngles2:
		return "UserAngles2"
	case ECmdInfoFlagsNoInterp:
		return "NoInterp"
	default:
		return fmt.Sprintf("%d", int(cmdInfoFlags))
	}
}