Fixed network positioning issues

2 files changed, 8 insertions, 3 deletions

diff --git a/internal/server/api/api.go b/internal/server/api/api.go
index 4dfa172..df29448 100644
--- a/internal/server/api/api.go
+++ b/internal/server/api/api.go
@@ -669,7 +669,7 @@ func SetUserData(ctx context.Context, sess *session.Session, request *packet.Set
 		return &ErrInternalError
 	}
 
-	return &packet.GetUserData{
+	return &packet.SetUserData{
 		Data: request.Data,
 	}
 }
@@ -683,7 +683,7 @@ func GetUserData(ctx context.Context, sess *session.Session, request *packet.Get
 		return &ErrInternalError
 	}
 
-	return &packet.GetUserData{
+	return &packet.SetUserData{
 		Data: data,
 	}
 }
diff --git a/internal/server/server.go b/internal/server/server.go
index 9abf51f..327eaa8 100644
--- a/internal/server/server.go
+++ b/internal/server/server.go
@@ -241,13 +241,18 @@ func (server *server) handleConnection(conn net.Conn) {
 	}()
 
 	// Send initial packets
-	payload, err := api.GetNetworksInfo(ctx, sess)
+	payload := api.GetUserData(ctx, sess, &packet.GetUserData{})
+	dataPacket := packet.NewPacket(packet.NewMsgPackEncoder(payload))
+	sess.Write(ctx, dataPacket)
+
+	payload, err = api.GetNetworksInfo(ctx, sess)
 	if err != nil {
 		return // closes the connection
 	}
 	infoPacket := packet.NewPacket(packet.NewMsgPackEncoder(payload))
 	sess.Write(ctx, infoPacket)
 
+	// Infinite read loop
 	buffer := make([]byte, 512)
 	for {
 		n, err := conn.Read(buffer)