brutal

Author: null

infra/conf/transport_internet.go

@@ -3,6 +3,7 @@ package conf
 import (
 	"context"
 	"encoding/base64"
+	"encoding/binary"
 	"encoding/hex"
 	"encoding/json"
 	"math"
@@ -22,6 +23,7 @@ import (
 	"github.com/xtls/xray-core/common/platform/filesystem"
 	"github.com/xtls/xray-core/common/serial"
 	"github.com/xtls/xray-core/transport/internet"
+	"github.com/xtls/xray-core/transport/internet/finalmask/brutal"
 	"github.com/xtls/xray-core/transport/internet/finalmask/fragment"
 	"github.com/xtls/xray-core/transport/internet/finalmask/header/custom"
 	"github.com/xtls/xray-core/transport/internet/finalmask/mkcp/aes128gcm"
@@ -1230,6 +1232,7 @@ var (
 	customVarNamePattern = regexp.MustCompile(`^[A-Za-z_][A-Za-z0-9_]*$`)
 
 	tcpmaskLoader = NewJSONConfigLoader(ConfigCreatorCache{
+		"brutal":        func() interface{} { return new(Brutal) },
 		"header-custom": func() interface{} { return new(HeaderCustomTCP) },
 		"fragment":      func() interface{} { return new(FragmentMask) },
 		"sudoku":        func() interface{} { return new(Sudoku) },
@@ -1247,6 +1250,34 @@ var (
 	}, "type", "settings")
 )
 
+type Brutal struct {
+	Rate uint64 `json:"rate"`
+	Cwnd uint64 `json:"cwnd"`
+}
+
+func (c *Brutal) Build() (proto.Message, error) {
+	rate := c.Rate
+	cwnd := c.Cwnd
+	if rate == 0 {
+		rate = 125000
+	}
+	if cwnd == 0 {
+		cwnd = 20
+	}
+	if rate < 62500 {
+		return nil, errors.New("invalid rate ", rate)
+	}
+	if cwnd < 5 || cwnd > 80 {
+		return nil, errors.New("invalid cwnd ", cwnd)
+	}
+	params := make([]byte, 16)
+	binary.NativeEndian.PutUint64(params, rate)
+	binary.NativeEndian.PutUint64(params[8:], cwnd)
+	return &brutal.Config{
+		Params: params,
+	}, nil
+}
+
 type TCPItem struct {
 	Delay     Int32Range       `json:"delay"`
 	Rand      int32            `json:"rand"`

transport/internet/finalmask/brutal/config.go

@@ -0,0 +1,13 @@
+package brutal
+
+import "net"
+
+func (c *Config) TCP() {}
+
+func (c *Config) WrapConnClient(raw net.Conn) (net.Conn, error) {
+	return NewConn(c, raw)
+}
+
+func (c *Config) WrapConnServer(raw net.Conn) (net.Conn, error) {
+	return NewConn(c, raw)
+}

transport/internet/finalmask/brutal/config.pb.go

@@ -0,0 +1,11 @@
+syntax = "proto3";
+
+package xray.transport.internet.finalmask.brutal;
+option csharp_namespace = "Xray.Transport.Internet.Finalmask.Brutal";
+option go_package = "github.com/xtls/xray-core/transport/internet/finalmask/brutal";
+option java_package = "com.xray.transport.internet.finalmask.brutal";
+option java_multiple_files = true;
+
+message Config {
+  bytes params = 1;
+}

transport/internet/finalmask/brutal/config.proto

@@ -0,0 +1,15 @@
+//go:build !linux
+
+package brutal
+
+import (
+	"context"
+	"net"
+
+	"github.com/xtls/xray-core/common/errors"
+)
+
+func NewConn(c *Config, raw net.Conn) (net.Conn, error) {
+	errors.LogError(context.Background(), "unsupported system")
+	return raw, nil
+}

transport/internet/finalmask/brutal/conn.go

@@ -0,0 +1,40 @@
+//go:build linux
+
+package brutal
+
+import (
+	"context"
+	"net"
+	"reflect"
+	"syscall"
+
+	"github.com/pires/go-proxyproto"
+	"github.com/xtls/xray-core/common"
+	"github.com/xtls/xray-core/common/errors"
+	"golang.org/x/sys/unix"
+)
+
+func NewConn(c *Config, raw net.Conn) (net.Conn, error) {
+	conn := raw
+	if pc, ok := conn.(*proxyproto.Conn); ok {
+		conn = pc.Raw()
+	}
+	if _, ok := conn.(*net.TCPConn); !ok {
+		errors.LogError(context.Background(), "unsupported conn ", reflect.TypeOf(conn))
+	}
+	sysConn := common.Must2(conn.(*net.TCPConn).SyscallConn())
+	err := sysConn.Control(func(fd uintptr) {
+		if err := unix.SetsockoptString(int(fd), unix.IPPROTO_TCP, unix.TCP_CONGESTION, "brutal"); err != nil {
+			errors.LogErrorInner(context.Background(), err, "failed to set congestion")
+			return
+		}
+		if err := syscall.SetsockoptString(int(fd), unix.IPPROTO_TCP, 23301, string(c.Params)); err != nil {
+			errors.LogErrorInner(context.Background(), err, "failed to set params")
+			return
+		}
+	})
+	if err != nil {
+		errors.LogErrorInner(context.Background(), err, "failed to control connection")
+	}
+	return raw, nil
+}

transport/internet/finalmask/brutal/conn_linux.go

@@ -256,9 +256,8 @@ func (l *tcpListener) Accept() (net.Conn, error) {
 
 	newConn, err := l.m.WrapConnServer(conn)
 	if err != nil {
-		errors.LogDebugInner(context.Background(), err, "mask err")
-		_ = conn.Close()
-		return nil, err
+		errors.LogErrorInner(context.Background(), err, "[mask] failed to wrap connection, falling back to the original connection")
+		return conn, nil
 	}
 
 	return newConn, nil