implement session

Author: uoosef

config/config.go

@@ -28,6 +28,7 @@ type Config struct {
 	DelayBetweenChunks     [2]int          `mapstructure:"DelayBetweenChunks"`
 	Hosts                  []resolve.Hosts `mapstructure:"Hosts"`
 	ResolveSystem          string          `mapstructure:"-"`
+	UserSession            string          `mapstructure:"-"`
 }
 
 var G *Config

server/server.go

@@ -12,6 +12,7 @@ import (
 	"context"
 	"fmt"
 	"io"
+	"math/rand"
 	"os"
 	"os/signal"
 	"strings"
@@ -22,6 +23,7 @@ import (
 var s5 *socks5.Server
 
 func Run(captureCTRLC bool) error {
+	config.G.UserSession = fmt.Sprintf("%08d", rand.Intn(1000))
 	appCache := utils.NewCache(time.Duration(config.G.DnsCacheTTL) * time.Second)
 
 	var resolveSystem string

utils/helper.go

@@ -2,6 +2,7 @@
 package utils
 
 import (
+	"bepass/config"
 	"fmt"
 	"net"
 	"net/url"
@@ -22,6 +23,6 @@ func WSEndpointHelper(workerAddress, rawDestAddress, network string) (string, er
 	if err != nil {
 		return "", err
 	}
-	endpoint := fmt.Sprintf("wss://%s/connect?host=%s&port=%s&net=%s", u.Host, dh, dp, network)
+	endpoint := fmt.Sprintf("wss://%s/connect?host=%s&port=%s&net=%s&session=%s", u.Host, dh, dp, network, config.G.UserSession)
 	return endpoint, nil
 }