.github/workflows/test.yaml

@@ -0,0 +1,15 @@
+name: Test
+on:
+  pull_request:
+
+jobs:
+  Test:
+    runs-on: ubuntu-latest
+    steps:
+      - uses: actions/checkout@v4
+      - uses: actions/setup-go@v5
+        with:
+          go-version-file: go.mod
+          check-latest: true
+      - name: Go test
+        run: go test ./...

client/client.go

@@ -84,12 +84,11 @@ type toWr struct {
 func (t toWr) Write(w []byte) (int, error) {
 	err := structcode.NewEncode(t.tun.Conn, proto.Request{
 		DataTX: &proto.ClientData{
+			Data: w,
 			Client: proto.Client{
 				Client: t.To,
 				Proto:  t.Proto,
 			},
-			Size: uint64(len(w)),
-			Data: w[:],
 		},
 	})
 	if err == nil {

internal/structcode/decode.go

@@ -5,19 +5,55 @@ import (
 	"encoding/binary"
 	"io"
 	"reflect"
-	"time"
 )
 
+func readBuff(r io.Reader) ([]byte, error) {
+	size := uint32(0)
+	if err := binary.Read(r, binary.BigEndian, &size); err != nil {
+		return nil, err
+	}
+	buff := make([]byte, size)
+	_, err := r.Read(buff)
+	return buff, err
+}
+
+func decodeTypeof(r io.Reader, reflectValue reflect.Value) (bool, error) {
+	var data any
+	npoint := reflect.New(reflectValue.Type())
+	typeof := npoint.Type()
+	switch {
+	default:
+		return false, nil
+	case typeof.Implements(typeofBinUnmarshal), typeof.ConvertibleTo(typeofBinUnmarshal):
+		buff, err := readBuff(r)
+		if err != nil {
+			return true, err
+		}
+		t := npoint.Interface()
+		if err := t.(encoding.BinaryUnmarshaler).UnmarshalBinary(buff); err != nil {
+			return true, err
+		}
+		data = t
+	case typeof.Implements(typeofTextUnmarshal), typeof.ConvertibleTo(typeofTextUnmarshal):
+		buff, err := readBuff(r)
+		if err != nil {
+			return true, err
+		}
+		t := npoint.Interface()
+		if err := t.(encoding.TextUnmarshaler).UnmarshalText(buff); err != nil {
+			return true, err
+		}
+		data = t
+	}
+	reflectValue.Set(reflect.ValueOf(data).Elem())
+	return true, nil
+}
+
 func decodeRecursive(r io.Reader, reflectValue reflect.Value) error {
 	switch reflectValue.Type().Kind() {
-	case reflect.Interface:
 	case reflect.String:
-		size := int64(0)
+		buff, err := readBuff(r)
-		if err := binary.Read(r, binary.BigEndian, &size); err != nil {
+		if err != nil {
-			return err
-		}
-		buff := make([]byte, size)
-		if _, err := r.Read(buff); err != nil {
 			return err
 		}
 		reflectValue.SetString(string(buff))
@@ -27,55 +63,28 @@ func decodeRecursive(r io.Reader, reflectValue reflect.Value) error {
 			return err
 		}
 		reflectValue.Set(reflect.ValueOf(data).Elem())
+	case reflect.Interface:
 	case reflect.Struct:
-		if reflectValue.Type().ConvertibleTo(typeofTimer) || reflectValue.Type().Implements(typeofBinUnmarshal) || reflectValue.Type().Implements(typeofTextUnmarshal) {
+		if ok, err := decodeTypeof(r, reflectValue); ok {
-			size := int64(0)
-			if err := binary.Read(r, binary.BigEndian, &size); err != nil {
 			return err
 		}
-			buff := make([]byte, size)
-			if _, err := r.Read(buff); err != nil {
-				return err
-			} else if reflectValue.Type().ConvertibleTo(typeofTimer) {
-				ttime := reflectValue.Interface().(time.Time)
-				if err := ttime.UnmarshalBinary(buff); err != nil {
-					return err
-				}
-				reflectValue.Set(reflect.ValueOf(ttime))
-				return nil
-			} else if reflectValue.Type().Implements(typeofBinUnmarshal) {
-				data := reflectValue.Interface().(encoding.BinaryUnmarshaler)
-				if err := data.UnmarshalBinary(buff); err != nil {
-					return err
-				}
-				reflectValue.Set(reflect.ValueOf(data))
-				return nil
-			}
-			data := reflectValue.Interface().(encoding.TextUnmarshaler)
-			if err := data.UnmarshalText(buff); err != nil {
-				return err
-			}
-			reflectValue.Set(reflect.ValueOf(data))
-			return nil
-		}
-
 		typeof := reflectValue.Type()
 		for fieldIndex := range typeof.NumField() {
-			if typeof.Field(fieldIndex).Tag.Get(selectorTagName) == "-" || !typeof.Field(fieldIndex).IsExported() {
+			fieldType := typeof.Field(fieldIndex)
+			if fieldType.Tag.Get(selectorTagName) == "-" || !fieldType.IsExported() {
 				continue
 			} else if err := decodeRecursive(r, reflectValue.Field(fieldIndex)); err != nil {
 				return err
 			}
 		}
 	case reflect.Pointer:
-		read := int8(0)
+		var read bool
 		if err := binary.Read(r, binary.BigEndian, &read); err != nil {
 			return err
-		} else if read == 0 {
+		} else if read {
-			return nil
-		}
 			reflectValue.Set(reflect.New(reflectValue.Type().Elem()))
 			return decodeRecursive(r, reflectValue.Elem())
+		}
 	case reflect.Array:
 		for arrIndex := range reflectValue.Len() {
 			if err := decodeRecursive(r, reflectValue.Index(arrIndex)); err != nil {
@@ -83,16 +92,18 @@ func decodeRecursive(r io.Reader, reflectValue reflect.Value) error {
 			}
 		}
 	case reflect.Slice:
-		size := int64(0)
+		if reflectValue.Type().ConvertibleTo(typeofBytes) {
-		if err := binary.Read(r, binary.BigEndian, &size); err != nil {
+			buff, err := readBuff(r)
-			return err
+			if err != nil {
-		} else if reflectValue.Type().Elem().Kind() == typeofByte.Kind() {
-			buff := make([]byte, size)
-			if _, err = r.Read(buff); err != nil {
 				return err
 			}
 			reflectValue.SetBytes(buff)
-		} else {
+			return nil
+		}
+		size := int64(0)
+		if err := binary.Read(r, binary.BigEndian, &size); err != nil {
+			return err
+		}
 		typeof := reflectValue.Type().Elem()
 		for range size {
 			newData := reflect.New(typeof)
@@ -102,6 +113,5 @@ func decodeRecursive(r io.Reader, reflectValue reflect.Value) error {
 			reflectValue.Set(reflect.AppendSlice(reflectValue, newData.Elem()))
 		}
 	}
-	}
 	return nil
 }

internal/structcode/encode.go

@@ -7,37 +7,46 @@ import (
 	"reflect"
 )
 
-func encodeRecursive(w io.Writer, reflectValue reflect.Value) error {
+func writeBuff(w io.Writer, buff []byte) error {
-	switch reflectValue.Type().Kind() {
+	if err := binary.Write(w, binary.BigEndian, uint32(len(buff))); err != nil {
-	case reflect.Interface:
-	case reflect.String:
-		str := reflectValue.String()
-		if err := binary.Write(w, binary.BigEndian, int64(len(str))); err != nil {
-			return err
-		} else if _, err := w.Write([]byte(str)); err != nil {
 		return err
 	}
-	case reflect.Bool, reflect.Float32, reflect.Float64, reflect.Int8, reflect.Int16, reflect.Int32, reflect.Int64, reflect.Uint8, reflect.Uint16, reflect.Uint32, reflect.Uint64:
+	_, err := w.Write(buff)
-		return binary.Write(w, binary.BigEndian, reflectValue.Interface())
+	return err
-	case reflect.Struct:
+}
-		if reflectValue.Type().Implements(typeofBinMarshal) || reflectValue.Type().Implements(typeofTextMarshal) {
+
-			var err error
+func encodeTypeof(w io.Writer, reflectValue reflect.Value) (bool, error) {
+	var err error = nil
 	var data []byte
-			if reflectValue.Type().Implements(typeofBinMarshal) {
+	switch {
+	default:
+		return false, nil
+	case reflectValue.Type().Implements(typeofBinMarshal), reflectValue.Type().ConvertibleTo(typeofBinMarshal):
 		data, err = reflectValue.Interface().(encoding.BinaryMarshaler).MarshalBinary()
-			} else {
+	case reflectValue.Type().Implements(typeofTextMarshal), reflectValue.Type().ConvertibleTo(typeofTextMarshal):
 		data, err = reflectValue.Interface().(encoding.TextMarshaler).MarshalText()
 	}
 	if err == nil {
-				if err = binary.Write(w, binary.BigEndian, int64(len(data))); err == nil {
+		err = writeBuff(w, data)
-					_, err = w.Write(data)
-				}
 	}
+	return true, err
+}
+
+func encodeRecursive(w io.Writer, reflectValue reflect.Value) error {
+	switch reflectValue.Type().Kind() {
+	case reflect.String:
+		return writeBuff(w, []byte(reflectValue.String()))
+	case reflect.Bool, reflect.Float32, reflect.Float64, reflect.Int8, reflect.Int16, reflect.Int32, reflect.Int64, reflect.Uint8, reflect.Uint16, reflect.Uint32, reflect.Uint64:
+		return binary.Write(w, binary.BigEndian, reflectValue.Interface())
+	case reflect.Interface:
+	case reflect.Struct:
+		if ok, err := encodeTypeof(w, reflectValue); ok {
 			return err
 		}
 		typeof := reflectValue.Type()
 		for fieldIndex := range typeof.NumField() {
-			if typeof.Field(fieldIndex).Tag.Get(selectorTagName) == "-" || !typeof.Field(fieldIndex).IsExported() {
+			fieldType := typeof.Field(fieldIndex)
+			if fieldType.Tag.Get(selectorTagName) == "-" || !fieldType.IsExported() {
 				continue
 			} else if err := encodeRecursive(w, reflectValue.Field(fieldIndex)); err != nil {
 				return err
@@ -45,8 +54,8 @@ func encodeRecursive(w io.Writer, reflectValue reflect.Value) error {
 		}
 	case reflect.Pointer:
 		if reflectValue.IsNil() || reflectValue.IsZero() {
-			return binary.Write(w, binary.BigEndian, int8(0))
+			return binary.Write(w, binary.BigEndian, false)
-		} else if err := binary.Write(w, binary.BigEndian, int8(1)); err != nil {
+		} else if err := binary.Write(w, binary.BigEndian, true); err != nil {
 			return err
 		}
 		return encodeRecursive(w, reflectValue.Elem())
@@ -57,9 +66,11 @@ func encodeRecursive(w io.Writer, reflectValue reflect.Value) error {
 			}
 		}
 	case reflect.Slice:
-		if err := binary.Write(w, binary.BigEndian, int64(reflectValue.Len())); err != nil {
+		if reflectValue.Type().ConvertibleTo(typeofBytes) {
+			return writeBuff(w, reflectValue.Bytes())
+		} else if err := binary.Write(w, binary.BigEndian, int64(reflectValue.Len())); err != nil {
 			return err
-		} else if reflectValue.Type().Elem().Kind() == typeofByte.Kind() {
+		} else if reflectValue.Type().Elem().Kind() == typeofBytes.Elem().Kind() {
 			_, err = w.Write(reflectValue.Bytes())
 			return err
 		}

internal/structcode/structcode.go

@@ -21,20 +21,17 @@ import (
 	"fmt"
 	"io"
 	"reflect"
-	"time"
 )
 
 const selectorTagName = "ser"
 
 var (
-	typeofTimer = reflect.TypeFor[time.Time]()
+	typeofBytes = reflect.TypeFor[[]byte]()
-	typeofBytes = reflect.TypeOf([]byte{})
-	typeofByte  = typeofBytes.Elem()
 
-	typeofBinMarshal    = reflect.TypeFor[encoding.BinaryMarshaler]()
-	typeofTextMarshal   = reflect.TypeFor[encoding.TextMarshaler]()
-	typeofBinUnmarshal  = reflect.TypeFor[encoding.BinaryUnmarshaler]()
 	typeofTextUnmarshal = reflect.TypeFor[encoding.TextUnmarshaler]()
+	typeofBinUnmarshal  = reflect.TypeFor[encoding.BinaryUnmarshaler]()
+	typeofTextMarshal   = reflect.TypeFor[encoding.TextMarshaler]()
+	typeofBinMarshal    = reflect.TypeFor[encoding.BinaryMarshaler]()
 )
 
 func NewEncode(w io.Writer, target any) error {

internal/structcode/structcode_test.go

@@ -2,51 +2,109 @@ package structcode
 
 import (
 	"bytes"
-	"encoding/json"
+	"encoding/hex"
+	"io"
+	"net/netip"
+	"sync"
 	"testing"
 	"time"
+
+	"sirherobrine23.org/Minecraft-Server/go-pproxit/proto"
 )
 
-type structTest struct {
+func TestSerelelize(t *testing.T) {
-	Text  string
+	t.Run("Response", func(t *testing.T) {
-	Int8  int8
+		var err error
-	Int16 int16
+		var encodeRes, decodeRes proto.Response
-	Int32 int32
+		encodeRes.BadRequest = true
-	Int64 int64
+		encodeRes.NotListened = true
-	Date  time.Time
+		encodeRes.SendAuth = true
 
-	Bytes      []byte
+		encodeRes.AgentInfo = &proto.AgentInfo{
-	ArrayBytes [4]byte
+			Protocol: 1,
-
+			UDPPort:  2555,
-	Pointer  *structTest
+			TCPPort:  3000,
-	Pointer2 any
+			AddrPort: netip.MustParseAddrPort("[::]:10000"),
-}
-
-func TestDeSerelelize(t *testing.T) {
-	var decodeTest structTest
-	var testData structTest
-	testData.Text = "Golang is best"
-	testData.Int8 = 2
-	testData.Int16 = 200
-	testData.Int32 = 1_000_000
-	testData.Int64 = 1024 * 12 * 12
-	testData.Bytes = []byte("google maintener go")
-	testData.ArrayBytes = [4]byte{0, 1, 1, 1}
-	testData.Date = time.Now()
-	testData.Pointer = &structTest{
-		Text: "Golang",
 		}
 
-	buff := new(bytes.Buffer)
+		var waiter sync.WaitGroup
-	if err := NewEncode(buff, testData); err != nil {
+		waiter.Add(2)
+		r, w := io.Pipe()
+		go func() {
+			defer waiter.Done()
+			if err = NewDecode(r, &decodeRes); err != nil {
 				t.Error(err)
 				return
 			}
-	t.Log(buff.Bytes())
+		}()
-	if err := NewDecode(buff, &decodeTest); err != nil {
+		go func() {
+			defer waiter.Done()
+			if err = NewEncode(w, encodeRes); err != nil {
 				t.Error(err)
 				return
 			}
-	d, _ := json.MarshalIndent(decodeTest, "", "  ")
+		}()
-	t.Log(string(d))
+		waiter.Wait()
+		if err != nil {
+			return
+		} else if decodeRes.BadRequest != encodeRes.BadRequest {
+			t.Errorf("invalid decode/encode, Current values to BadRequest, Decode %v, Encode %v", decodeRes.BadRequest, encodeRes.BadRequest)
+			return
+		} else if decodeRes.NotListened != encodeRes.NotListened {
+			t.Errorf("invalid decode/encode, Current values to NotListened, Decode %v, Encode %v", decodeRes.NotListened, encodeRes.NotListened)
+			return
+		} else if decodeRes.SendAuth != encodeRes.SendAuth {
+			t.Errorf("invalid decode/encode, Current values to SendAuth, Decode %v, Encode %v", decodeRes.SendAuth, encodeRes.SendAuth)
+			return
+		} else if decodeRes.AgentInfo == nil {
+			t.Errorf("invalid decode, Current values to AgentInfo, Decode %+v, Encode %+v", decodeRes.AgentInfo, encodeRes.AgentInfo)
+			return
+		} else if decodeRes.AgentInfo.Protocol != encodeRes.AgentInfo.Protocol {
+			t.Errorf("invalid decode/encode, Current values to AgentInfo.Protocol, Decode %d, Encode %d", decodeRes.AgentInfo.Protocol, encodeRes.AgentInfo.Protocol)
+			return
+		} else if decodeRes.AgentInfo.TCPPort != encodeRes.AgentInfo.TCPPort {
+			t.Errorf("invalid decode/encode, Current values to AgentInfo.TCPPort, Decode %d, Encode %d", decodeRes.AgentInfo.TCPPort, encodeRes.AgentInfo.TCPPort)
+		} else if decodeRes.AgentInfo.UDPPort != encodeRes.AgentInfo.UDPPort {
+			t.Errorf("invalid decode/encode, Current values to AgentInfo.UDPPort, Decode %d, Encode %d", decodeRes.AgentInfo.UDPPort, encodeRes.AgentInfo.UDPPort)
+			return
+		} else if decodeRes.AgentInfo.AddrPort.Compare(encodeRes.AgentInfo.AddrPort) != 0 {
+			t.Errorf("invalid decode/encode, Current values to AgentInfo.AddrPort, Decode %s, Encode %s", decodeRes.AgentInfo.AddrPort, encodeRes.AgentInfo.AddrPort)
+			return
+		}
+	})
+	t.Run("Request", func(t *testing.T) {
+		var err error
+		var encodeRequest, decodeRequest proto.Request
+		encodeRequest.AgentAuth = &[]byte{0, 0, 1, 1, 1, 1, 1, 0, 255}
+		encodeRequest.Ping = new(time.Time)
+		*encodeRequest.Ping = time.Now()
+
+		var waiter sync.WaitGroup
+		waiter.Add(2)
+		r, w := io.Pipe()
+		go func() {
+			defer waiter.Done()
+			if err = NewEncode(w, encodeRequest); err != nil {
+				t.Error(err)
+				return
+			}
+		}()
+		go func() {
+			defer waiter.Done()
+			if err = NewDecode(r, &decodeRequest); err != nil {
+				t.Error(err)
+				return
+			}
+		}()
+		waiter.Wait()
+		if err != nil {
+			return
+		} else if decodeRequest.Ping.Unix() != encodeRequest.Ping.Unix() {
+			t.Errorf("cannot decode/encode Ping date, Decode %d, Encode: %d", decodeRequest.Ping.Unix(), encodeRequest.Ping.Unix())
+			return
+		} else if !bytes.Equal(*decodeRequest.AgentAuth, *encodeRequest.AgentAuth) {
+			t.Errorf("cannot decode/encode auth data, Decode %q, Encode: %q", hex.EncodeToString(*decodeRequest.AgentAuth), hex.EncodeToString(*encodeRequest.AgentAuth))
+			return
+		}
+	})
 }

proto/proto.go

@@ -26,6 +26,5 @@ type Client struct {
 
 type ClientData struct {
 	Client Client // Client Destination
-	Size   uint64 // Data size
 	Data   []byte `json:"-"` // Bytes to send
 }

server/server.go

@@ -12,9 +12,7 @@ import (
 	"sirherobrine23.org/Minecraft-Server/go-pproxit/proto"
 )
 
-var (
+var ErrAuthAgentFail error = errors.New("cannot authenticate agent") // Send unathorized client and close new accepts from current port
-	ErrAuthAgentFail error = errors.New("cannot authenticate agent") // Send unathorized client and close new accepts from current port
-)
 
 type ServerCall interface {
 	// Authenticate agents

server/tunnel.go

@@ -1,9 +1,9 @@
 package server
 
 import (
+	"encoding/json"
 	"fmt"
 	"io"
-	"log"
 	"net"
 	"net/netip"
 	"os"
@@ -74,12 +74,11 @@ func (t toWr) Write(w []byte) (int, error) {
 	go t.tun.TunInfo.Callbacks.RegisterRX(t.To, len(w), t.Proto)
 	err := t.tun.send(proto.Response{
 		DataRX: &proto.ClientData{
+			Data: w,
 			Client: proto.Client{
 				Proto:  t.Proto,
 				Client: t.To,
 			},
-			Size: uint64(len(w)),
-			Data: w[:],
 		},
 	})
 	if err == nil {
@@ -120,12 +119,14 @@ func (tun *Tunnel) Setup() {
 	})
 
 	for {
-		log.Printf("waiting request from %s", tun.RootConn.RemoteAddr().String())
 		var req proto.Request
 		if err := structcode.NewDecode(tun.RootConn, &req); err != nil {
 			fmt.Fprintln(os.Stderr, err.Error())
 			return
 		}
+		d, _ := json.MarshalIndent(req, "", "  ")
+		fmt.Println(string(d))
+
 		if req.AgentAuth != nil {
 			go tun.send(proto.Response{
 				AgentInfo: &proto.AgentInfo{
@@ -151,7 +152,7 @@ func (tun *Tunnel) Setup() {
 				}
 			}
 		} else if data := req.DataTX; req.DataTX != nil {
-			go tun.TunInfo.Callbacks.RegisterTX(data.Client.Client, int(data.Size), data.Client.Proto)
+			go tun.TunInfo.Callbacks.RegisterTX(data.Client.Client, len(data.Data), data.Client.Proto)
 			if data.Client.Proto == proto.ProtoTCP {
 				if cl, ok := tun.TCPClients[data.Client.Client.String()]; ok {
 					go cl.Write(data.Data) // Process in backgroud