From 81727ae239f69555a836cd15207600543e5c051b Mon Sep 17 00:00:00 2001
From: Vadym Vikulin <vadym.vikulin@rivchain.org>
Date: Fri, 22 Dec 2023 23:40:43 +0200
Subject: [PATCH] Flow label test

---
 src/core/core.go | 12 ++++--------
 1 file changed, 4 insertions(+), 8 deletions(-)

diff --git a/src/core/core.go b/src/core/core.go
index 9bdfae844..4d8106057 100644
--- a/src/core/core.go
+++ b/src/core/core.go
@@ -212,18 +212,17 @@ func (c *Core) ReadFrom(p []byte) (n int, from net.Addr, err error) {
 		if n == 0 {
 			continue
 		}
-		switch bs[0] {
+		switch bs[3] {
 		case typeSessionTraffic:
 			// This is what we want to handle here
 		case typeSessionProto:
-			data := append([]byte(nil), bs[1:n]...)
+			data := append([]byte(nil), bs[:n]...)
 			c.proto.handleProto(nil, from.(iwt.Addr), data)
 			continue
 		default:
 			continue
 		}
-		bs = bs[1:n]
-		copy(p, bs)
+		copy(p, bs[:n])
 		if len(p) < len(bs) {
 			n = len(p)
 		} else {
@@ -236,12 +235,9 @@ func (c *Core) ReadFrom(p []byte) (n int, from net.Addr, err error) {
 func (c *Core) WriteTo(p []byte, addr net.Addr) (n int, err error) {
 	buf := allocBytes(0)
 	defer freeBytes(buf)
-	buf = append(buf, typeSessionTraffic)
 	buf = append(buf, p...)
+	buf[3] = typeSessionTraffic
 	n, err = c.PacketConn.WriteTo(buf, addr)
-	if n > 0 {
-		n -= 1
-	}
 	return
 }