net/svc messages

15 files changed, 306 insertions, 6 deletions

diff --git a/packets/messages/message.go b/packets/messages/message.go
new file mode 100644
index 0000000..c508bfa
--- /dev/null
+++ b/packets/messages/message.go
@@ -0,0 +1,50 @@
+package messages
+
+import (
+        "bytes"
+
+        "github.com/pektezol/bitreader"
+        "github.com/pektezol/demoparser/packets/messages/types"
+)
+
+func ParseMessage(data []byte) []Message {
+        reader := bitreader.Reader(bytes.NewReader(data), true)
+        var messages []Message
+        for {
+                messageType, err := reader.ReadBits(6)
+                if err != nil {
+                        break
+                }
+                switch messageType {
+                case 0x00:
+                        messages = append(messages, Message{Data: types.NetNop{}})
+                case 0x01:
+                        messages = append(messages, Message{Data: types.ParseNetDisconnect(reader)})
+                case 0x02:
+                        messages = append(messages, Message{Data: types.ParseNetFile(reader)})
+                case 0x03:
+                        messages = append(messages, Message{Data: types.ParseNetSplitScreenUser(reader)})
+                case 0x04:
+                        messages = append(messages, Message{Data: types.ParseNetTick(reader)})
+                case 0x05:
+                        messages = append(messages, Message{Data: types.ParseNetStringCmd(reader)})
+                case 0x06:
+                        messages = append(messages, Message{Data: types.ParseNetSetConVar(reader)})
+                case 0x07:
+                        messages = append(messages, Message{Data: types.ParseNetSignOnState(reader)})
+                case 0x08:
+                        messages = append(messages, Message{Data: types.ParseSvcServerInfo(reader)})
+                case 0x09:
+                        messages = append(messages, Message{Data: types.ParseSvcSendTable(reader)})
+                case 0x10:
+                        messages = append(messages, Message{Data: types.ParseSvcClassInfo(reader)})
+                case 0x11:
+                        messages = append(messages, Message{Data: types.ParseSvcSetPause(reader)})
+                }
+        }
+        return messages
+}
+
+type Message struct {
+        Data any
+}
diff --git a/packets/messages/types/NetDisconnect.go b/packets/messages/types/NetDisconnect.go
new file mode 100644
index 0000000..8b65568
--- /dev/null
+++ b/packets/messages/types/NetDisconnect.go
@@ -0,0 +1,11 @@
+package types
+
+import "github.com/pektezol/bitreader"
+
+type NetDisconnect struct {
+        Text string
+}
+
+func ParseNetDisconnect(reader *bitreader.ReaderType) NetDisconnect {
+        return NetDisconnect{Text: reader.TryReadString()}
+}
diff --git a/packets/messages/types/NetFile.go b/packets/messages/types/NetFile.go
new file mode 100644
index 0000000..c65873d
--- /dev/null
+++ b/packets/messages/types/NetFile.go
@@ -0,0 +1,17 @@
+package types
+
+import "github.com/pektezol/bitreader"
+
+type NetFile struct {
+        TransferId    uint32
+        FileName      string
+        FileRequested bool
+}
+
+func ParseNetFile(reader *bitreader.ReaderType) NetFile {
+        return NetFile{
+                TransferId:    reader.TryReadInt32(),
+                FileName:      reader.TryReadString(),
+                FileRequested: reader.TryReadBool(),
+        }
+}
diff --git a/packets/messages/types/NetNop.go b/packets/messages/types/NetNop.go
new file mode 100644
index 0000000..d0a1f7a
--- /dev/null
+++ b/packets/messages/types/NetNop.go
@@ -0,0 +1,3 @@
+package types
+
+type NetNop struct{}
diff --git a/packets/messages/types/NetSetConVar.go b/packets/messages/types/NetSetConVar.go
new file mode 100644
index 0000000..b502953
--- /dev/null
+++ b/packets/messages/types/NetSetConVar.go
@@ -0,0 +1,28 @@
+package types
+
+import "github.com/pektezol/bitreader"
+
+type NetSetConVar struct {
+        Length  uint8
+        ConVars []ConVar
+}
+
+type ConVar struct {
+        Name  string
+        Value string
+}
+
+func ParseNetSetConVar(reader *bitreader.ReaderType) NetSetConVar {
+        var convars []ConVar
+        netsetconvar := NetSetConVar{
+                Length: reader.TryReadInt8(),
+        }
+        for i := 0; i < int(netsetconvar.Length); i++ {
+                convars = append(convars, ConVar{
+                        Name:  reader.TryReadString(),
+                        Value: reader.TryReadString(),
+                })
+        }
+        netsetconvar.ConVars = convars
+        return netsetconvar
+}
diff --git a/packets/messages/types/NetSignOnState.go b/packets/messages/types/NetSignOnState.go
new file mode 100644
index 0000000..3d02e91
--- /dev/null
+++ b/packets/messages/types/NetSignOnState.go
@@ -0,0 +1,31 @@
+package types
+
+import (
+        "fmt"
+
+        "github.com/pektezol/bitreader"
+)
+
+type NetSignOnState struct {
+        SignonState       int8
+        SpawnCount        uint32
+        NumServerPlayers  uint32
+        IdsLength         uint32
+        PlayersNetworkIds []byte
+        MapNameLength     uint32
+        MapName           string
+}
+
+func ParseNetSignOnState(reader *bitreader.ReaderType) NetSignOnState {
+        netsignonstate := NetSignOnState{
+                SignonState:      int8(reader.TryReadInt8()),
+                SpawnCount:       reader.TryReadInt32(),
+                NumServerPlayers: reader.TryReadInt32(),
+                IdsLength:        reader.TryReadInt32(),
+        }
+        fmt.Println(netsignonstate.IdsLength)
+        netsignonstate.PlayersNetworkIds = reader.TryReadBytesToSlice(int(netsignonstate.IdsLength))
+        netsignonstate.MapNameLength = reader.TryReadInt32()
+        netsignonstate.MapName = reader.TryReadStringLen(int(netsignonstate.MapNameLength))
+        return netsignonstate
+}
diff --git a/packets/messages/types/NetSplitScreenUser.go b/packets/messages/types/NetSplitScreenUser.go
new file mode 100644
index 0000000..66a7c85
--- /dev/null
+++ b/packets/messages/types/NetSplitScreenUser.go
@@ -0,0 +1,13 @@
+package types
+
+import "github.com/pektezol/bitreader"
+
+type NetSplitScreenUser struct {
+        Unknown bool
+}
+
+func ParseNetSplitScreenUser(reader *bitreader.ReaderType) NetSplitScreenUser {
+        return NetSplitScreenUser{
+                Unknown: reader.TryReadBool(),
+        }
+}
diff --git a/packets/messages/types/NetStringCmd.go b/packets/messages/types/NetStringCmd.go
new file mode 100644
index 0000000..1ee9f67
--- /dev/null
+++ b/packets/messages/types/NetStringCmd.go
@@ -0,0 +1,13 @@
+package types
+
+import "github.com/pektezol/bitreader"
+
+type NetStringCmd struct {
+        Command string
+}
+
+func ParseNetStringCmd(reader *bitreader.ReaderType) NetStringCmd {
+        return NetStringCmd{
+                Command: reader.TryReadString(),
+        }
+}
diff --git a/packets/messages/types/NetTick.go b/packets/messages/types/NetTick.go
new file mode 100644
index 0000000..9dae14b
--- /dev/null
+++ b/packets/messages/types/NetTick.go
@@ -0,0 +1,17 @@
+package types
+
+import "github.com/pektezol/bitreader"
+
+type NetTick struct {
+        Tick                      uint32
+        HostFrameTime             float32
+        HostFrameTimeStdDeviation float32
+}
+
+func ParseNetTick(reader *bitreader.ReaderType) NetTick {
+        return NetTick{
+                Tick:                      reader.TryReadInt32(),
+                HostFrameTime:             float32(reader.TryReadInt16()) / 1e5,
+                HostFrameTimeStdDeviation: float32(reader.TryReadInt16()) / 1e5,
+        }
+}
diff --git a/packets/messages/types/SvcClassInfo.go b/packets/messages/types/SvcClassInfo.go
new file mode 100644
index 0000000..58e43b7
--- /dev/null
+++ b/packets/messages/types/SvcClassInfo.go
@@ -0,0 +1,42 @@
+package types
+
+import (
+        "math"
+
+        "github.com/pektezol/bitreader"
+)
+
+type SvcClassInfo struct {
+        Length         uint16
+        CreateOnClient bool
+        ServerClasses  []ServerClass
+}
+
+type ServerClass struct {
+        ClassId       int32
+        ClassName     string
+        DataTableName string
+}
+
+func ParseSvcClassInfo(reader *bitreader.ReaderType) SvcClassInfo {
+        var serverclasses []ServerClass
+        svcclassinfo := SvcClassInfo{
+                Length:         reader.TryReadInt16(),
+                CreateOnClient: reader.TryReadBool(),
+        }
+        if svcclassinfo.CreateOnClient {
+                for i := 0; i < int(svcclassinfo.Length); i++ {
+                        id, err := reader.ReadBits(int(math.Log2(float64(svcclassinfo.Length))) + 1)
+                        if err != nil {
+                                panic(err)
+                        }
+                        serverclasses = append(serverclasses, ServerClass{
+                                ClassId:       int32(id),
+                                ClassName:     reader.TryReadString(),
+                                DataTableName: reader.TryReadString(),
+                        })
+                }
+        }
+        svcclassinfo.ServerClasses = serverclasses
+        return svcclassinfo
+}
diff --git a/packets/messages/types/SvcSendTable.go b/packets/messages/types/SvcSendTable.go
new file mode 100644
index 0000000..f76aadb
--- /dev/null
+++ b/packets/messages/types/SvcSendTable.go
@@ -0,0 +1,17 @@
+package types
+
+import "github.com/pektezol/bitreader"
+
+type SvcSendTable struct {
+        NeedsDecoder bool
+        Length       uint8
+        Props        int32
+}
+
+func ParseSvcSendTable(reader *bitreader.ReaderType) SvcSendTable {
+        return SvcSendTable{
+                NeedsDecoder: reader.TryReadBool(),
+                Length:       reader.TryReadInt8(),
+        }
+        // No one cares about SvcSendTable
+}
diff --git a/packets/messages/types/SvcServerInfo.go b/packets/messages/types/SvcServerInfo.go
new file mode 100644
index 0000000..9d2648b
--- /dev/null
+++ b/packets/messages/types/SvcServerInfo.go
@@ -0,0 +1,43 @@
+package types
+
+import "github.com/pektezol/bitreader"
+
+type SvcServerInfo struct {
+        Protocol     uint16
+        ServerCount  uint32
+        IsHltv       bool
+        IsDedicated  bool
+        ClientCrc    int32
+        MaxClasses   uint16
+        MapCrc       uint32
+        PlayerSlot   uint8
+        MaxClients   uint8
+        Unk          uint32
+        TickInterval float32
+        COs          byte
+        GameDir      string
+        MapName      string
+        SkyName      string
+        HostName     string
+}
+
+func ParseSvcServerInfo(reader *bitreader.ReaderType) SvcServerInfo {
+        return SvcServerInfo{
+                Protocol:     reader.TryReadInt16(),
+                ServerCount:  reader.TryReadInt32(),
+                IsHltv:       reader.TryReadBool(),
+                IsDedicated:  reader.TryReadBool(),
+                ClientCrc:    int32(reader.TryReadInt32()),
+                MaxClasses:   reader.TryReadInt16(),
+                MapCrc:       reader.TryReadInt32(),
+                PlayerSlot:   reader.TryReadInt8(),
+                MaxClients:   reader.TryReadInt8(),
+                Unk:          reader.TryReadInt32(),
+                TickInterval: reader.TryReadFloat32(),
+                COs:          reader.TryReadInt8(),
+                GameDir:      reader.TryReadString(),
+                MapName:      reader.TryReadString(),
+                SkyName:      reader.TryReadString(),
+                HostName:     reader.TryReadString(),
+        }
+}
diff --git a/packets/messages/types/SvcSetPause.go b/packets/messages/types/SvcSetPause.go
new file mode 100644
index 0000000..040fb6a
--- /dev/null
+++ b/packets/messages/types/SvcSetPause.go
@@ -0,0 +1,11 @@
+package types
+
+import "github.com/pektezol/bitreader"
+
+type SvcSetPause struct {
+        Paused bool
+}
+
+func ParseSvcSetPause(reader *bitreader.ReaderType) SvcSetPause {
+        return SvcSetPause{Paused: reader.TryReadBool()}
+}
diff --git a/packets/message.go b/packets/packet.go
index 922f918..23a21e6 100644
--- a/packets/message.go
+++ b/packets/packet.go
@@ -5,11 +5,12 @@ import (
 
         "github.com/pektezol/bitreader"
         "github.com/pektezol/demoparser/packets/classes"
+        "github.com/pektezol/demoparser/packets/messages"
 )
 
 const MSSC = 2
 
-func ParseMessage(reader *bitreader.ReaderType) (status int) {
+func ParsePacket(reader *bitreader.ReaderType) (status int) {
         messageType := reader.TryReadInt8()
         messageTick := reader.TryReadInt32()
         messageSlot := reader.TryReadInt8()
@@ -31,8 +32,8 @@ func ParseMessage(reader *bitreader.ReaderType) (status int) {
                         OutSequence: int32(reader.TryReadInt32()),
                         Size:        int32(reader.TryReadInt32()),
                 }
-                reader.SkipBytes(int(packet.Size))
-                //fmt.Printf("[%d] (%d) Packet: %v\n", messageTick, messageType, packet)
+                packet.Data = messages.ParseMessage(reader.TryReadBytesToSlice(int(packet.Size)))
+                fmt.Printf("[%d] (%d) Packet: %v\n", messageTick, messageType, packet)
                 return 2
         case 0x03:
                 syncTick := SyncTick{}
diff --git a/packets/types.go b/packets/types.go
index 2a4cf4b..9a1f667 100644
--- a/packets/types.go
+++ b/packets/types.go
@@ -1,6 +1,9 @@
 package packets
 
-import "github.com/pektezol/demoparser/packets/classes"
+import (
+        "github.com/pektezol/demoparser/packets/classes"
+        "github.com/pektezol/demoparser/packets/messages"
+)
 
 type Header struct {
         DemoFileStamp   string
@@ -21,7 +24,7 @@ type SignOn struct {
         InSequence  int32
         OutSequence int32
         Size        int32
-        Data        []byte
+        Data        []messages.Message
 }
 
 type Packet struct {
@@ -29,7 +32,7 @@ type Packet struct {
         InSequence  int32
         OutSequence int32
         Size        int32
-        Data        []byte
+        Data        []messages.Message
 }
 
 type SyncTick struct{}