feat: reverse listener configuration

shoriwe · May 22, 2023 · fc5e41a · fc5e41a
1 parent adaa1de
commit fc5e41a
Show file tree

Hide file tree

Showing 2 changed files with 86 additions and 0 deletions.
diff --git a/config/reverse.go b/config/reverse.go
@@ -0,0 +1,33 @@
+package config
+
+import (
+	"net"
+
+	"github.com/shoriwe/fullproxy/v3/reverse"
+)
+
+type Reverse struct {
+	Listener   *Listener `yaml:"listener"`
+	Controller Listener  `yaml:"controller"`
+}
+
+func (r *Reverse) Master() (net.Listener, error) {
+	l, lErr := r.Listener.Listen()
+	if lErr != nil {
+		return nil, lErr
+	}
+	cl, clErr := r.Listener.Listen()
+	if clErr != nil {
+		l.Close()
+		return nil, clErr
+	}
+	return reverse.NewMaster(l, cl)
+}
+
+func (r *Reverse) Slave() (*reverse.Slave, error) {
+	cl, clErr := r.Controller.Dial()
+	if clErr != nil {
+		return nil, clErr
+	}
+	return reverse.NewSlave(cl)
+}
diff --git a/config/reverse_test.go b/config/reverse_test.go
@@ -0,0 +1,53 @@
+package config
+
+import (
+	"testing"
+
+	"github.com/stretchr/testify/assert"
+)
+
+func TestReverse_Master(t *testing.T) {
+	t.Run("Valid", func(tt *testing.T) {
+		r := Reverse{
+			Listener: &Listener{
+				Network: "tcp",
+				Address: "localhost:0",
+			},
+			Controller: Listener{
+				Network: "tcp",
+				Address: "localhost:0",
+			},
+		}
+		m, err := r.Master()
+		assert.Nil(tt, err)
+		defer m.Close()
+	})
+	t.Run("Invalid User listener", func(tt *testing.T) {
+		r := Reverse{
+			Listener: &Listener{
+				Network: "tcp",
+				Address: "localhost:99999999",
+			},
+			Controller: Listener{
+				Network: "tcp",
+				Address: "localhost:0",
+			},
+		}
+		_, err := r.Master()
+		assert.NotNil(tt, err)
+	})
+	t.Run("Invalid Controller listener", func(tt *testing.T) {
+		r := Reverse{
+			Listener: &Listener{
+				Network: "tcp",
+				Address: "localhost:0",
+			},
+			Controller: Listener{
+				Network: "tcp",
+				Address: "localhost:9999999999",
+			},
+		}
+		_, err := r.Master()
+		assert.NotNil(tt, err)
+	})
+}