From 6aa511f2ae3d5c84e7db02627ffd4d7f6796ce75 Mon Sep 17 00:00:00 2001
From: Arda Serdar Pektezol <1669855+pektezol@users.noreply.github.com>
Date: Sun, 17 Sep 2023 18:24:18 +0300
Subject: decide what message types are left

---
 pkg/messages/types/svcClassInfo.go | 10 ++++------
 1 file changed, 4 insertions(+), 6 deletions(-)

(limited to 'pkg/messages/types/svcClassInfo.go')

diff --git a/pkg/messages/types/svcClassInfo.go b/pkg/messages/types/svcClassInfo.go
index 27862bf..d2f6a90 100644
--- a/pkg/messages/types/svcClassInfo.go
+++ b/pkg/messages/types/svcClassInfo.go
@@ -1,14 +1,13 @@
 package messages
 
 import (
-	"fmt"
 	"math"
 
 	"github.com/pektezol/bitreader"
 )
 
 type SvcClassInfo struct {
-	Length         int16
+	ClassCount     uint16
 	CreateOnClient bool
 	ServerClasses  []serverClass
 }
@@ -21,15 +20,14 @@ type serverClass struct {
 
 func ParseSvcClassInfo(reader *bitreader.Reader) SvcClassInfo {
 	svcClassInfo := SvcClassInfo{
-		Length:         int16(reader.TryReadBits(16)),
+		ClassCount:     reader.TryReadUInt16(),
 		CreateOnClient: reader.TryReadBool(),
 	}
 	classes := []serverClass{}
 	if !svcClassInfo.CreateOnClient {
-		for count := 0; count < int(svcClassInfo.Length); count++ {
-			fmt.Println(classes)
+		for count := 0; count < int(svcClassInfo.ClassCount); count++ {
 			classes = append(classes, serverClass{
-				ClassId:       int16(reader.TryReadBits(uint64(math.Log2(float64(svcClassInfo.Length)) + 1))),
+				ClassId:       int16(reader.TryReadBits(uint64(math.Log2(float64(svcClassInfo.ClassCount)) + 1))),
 				ClassName:     reader.TryReadString(),
 				DataTableName: reader.TryReadString(),
 			})
-- 
cgit v1.2.3