merge ss and ssu

pkg/handler/ss/metadata.go

@@ -3,19 +3,53 @@ package ss
 import (
 	"time"
 
+	"github.com/go-gost/gost/pkg/internal/utils/ss"
+	md "github.com/go-gost/gost/pkg/metadata"
 	"github.com/shadowsocks/go-shadowsocks2/core"
 )
 
-const (
-	method      = "method"
-	password    = "password"
-	key         = "key"
-	readTimeout = "readTimeout"
-	retryCount  = "retry"
-)
-
 type metadata struct {
 	cipher      core.Cipher
 	readTimeout time.Duration
 	retryCount  int
+	bufferSize  int
+	enableUDP   bool
+}
+
+func (h *ssHandler) parseMetadata(md md.Metadata) (err error) {
+	const (
+		method      = "method"
+		password    = "password"
+		key         = "key"
+		readTimeout = "readTimeout"
+		retryCount  = "retry"
+		enableUDP   = "udp"
+		bufferSize  = "bufferSize"
+	)
+
+	h.md.cipher, err = ss.ShadowCipher(
+		md.GetString(method),
+		md.GetString(password),
+		md.GetString(key),
+	)
+	if err != nil {
+		return
+	}
+
+	h.md.readTimeout = md.GetDuration(readTimeout)
+	h.md.retryCount = md.GetInt(retryCount)
+	h.md.enableUDP = md.GetBool(enableUDP)
+
+	h.md.bufferSize = md.GetInt(bufferSize)
+	if h.md.bufferSize > 0 {
+		if h.md.bufferSize < 512 {
+			h.md.bufferSize = 512 // min buffer size
+		}
+		if h.md.bufferSize > 65*1024 {
+			h.md.bufferSize = 65 * 1024 // max buffer size
+		}
+	} else {
+		h.md.bufferSize = 4096 // default buffer size
+	}
+	return
 }