111 lines
3.0 KiB
Go
111 lines
3.0 KiB
Go
//go:build plus
|
|
|
|
package nodeconfigs
|
|
|
|
import (
|
|
"fmt"
|
|
"github.com/iwind/TeaGo/rands"
|
|
"net"
|
|
"runtime"
|
|
)
|
|
|
|
// NewTOAConfig 默认的TOA配置
|
|
func NewTOAConfig() *TOAConfig {
|
|
return &TOAConfig{
|
|
IsOn: false,
|
|
Debug: false,
|
|
OptionTypeV4: 0xfe,
|
|
OptionTypeV6: 0xfe,
|
|
MinQueueId: 100,
|
|
AutoSetup: true,
|
|
}
|
|
}
|
|
|
|
// TOAConfig TOA相关配置
|
|
type TOAConfig struct {
|
|
IsOn bool `yaml:"isOn" json:"isOn"` // 是否启用
|
|
Debug bool `yaml:"debug" json:"debug"`
|
|
OldOptionType uint8 `yaml:"optionType" json:"optionType"` // Deprecated 已过期，使用 OptionTypeV4 和 OptionTypeV6 代替
|
|
OptionTypeV4 uint8 `yaml:"optionTypeV4" json:"optionTypeV4"` // IPv4的类型
|
|
OptionTypeV6 uint8 `yaml:"OptionTypeV6" json:"optionTypeV6"`
|
|
MinQueueId uint8 `yaml:"minQueueId" json:"minQueueId"`
|
|
AutoSetup bool `yaml:"autoSetup" json:"autoSetup"`
|
|
MinLocalPort uint16 `yaml:"minLocalPort" json:"minLocalPort"` // 本地可使用的最小端口 TODO
|
|
MaxLocalPort uint16 `yaml:"maxLocalPort" json:"maxLocalPort"` // 本地可使用的最大端口 TODO
|
|
SockPath string `yaml:"sockPath" json:"sockPath"` // Sock文件路径 TODO
|
|
ByPassPorts []uint16 `yaml:"byPassPorts" json:"byPassPorts"` // 忽略的端口 TODO
|
|
|
|
minLocalPort int
|
|
maxLocalPort int
|
|
}
|
|
|
|
func (this *TOAConfig) Init() error {
|
|
// LocalPort
|
|
minPort := this.MinLocalPort
|
|
maxPort := this.MaxLocalPort
|
|
if minPort == 0 {
|
|
minPort = 1025
|
|
}
|
|
if maxPort == 0 {
|
|
maxPort = 65534
|
|
}
|
|
if minPort > maxPort {
|
|
minPort, maxPort = maxPort, minPort
|
|
}
|
|
this.minLocalPort = int(minPort)
|
|
this.maxLocalPort = int(maxPort)
|
|
|
|
// 兼容
|
|
if this.OptionTypeV4 == 0 && this.OldOptionType > 0 {
|
|
this.OptionTypeV4 = this.OldOptionType
|
|
}
|
|
if this.OptionTypeV6 == 0 && this.OldOptionType > 0 {
|
|
this.OptionTypeV6 = this.OldOptionType
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
// SockFile Sock路径
|
|
func (this *TOAConfig) SockFile() string {
|
|
if len(this.SockPath) == 0 {
|
|
return "/tmp/edge-toa.sock"
|
|
}
|
|
return this.SockPath
|
|
}
|
|
|
|
// RandLocalPort 获取随机端口
|
|
func (this *TOAConfig) RandLocalPort() uint16 {
|
|
listener, err := net.Listen("tcp", ":0")
|
|
if err != nil {
|
|
return uint16(rands.Int(this.minLocalPort, this.maxLocalPort))
|
|
}
|
|
_ = listener.Close()
|
|
return uint16(listener.Addr().(*net.TCPAddr).Port)
|
|
}
|
|
|
|
// AsArgs 转换为参数的形式
|
|
func (this *TOAConfig) AsArgs() (args []string) {
|
|
args = append(args, "run")
|
|
args = append(args, "-option-type-v4="+fmt.Sprintf("%d", this.OptionTypeV4))
|
|
args = append(args, "-option-type-v6="+fmt.Sprintf("%d", this.OptionTypeV6))
|
|
args = append(args, "-min-queue-id="+fmt.Sprintf("%d", this.MinQueueId))
|
|
args = append(args, "-max-queue-id="+fmt.Sprintf("%d", this.MaxQueueId()))
|
|
if this.AutoSetup {
|
|
args = append(args, "-auto-setup")
|
|
}
|
|
if this.Debug {
|
|
args = append(args, "-debug")
|
|
}
|
|
return
|
|
}
|
|
|
|
// MaxQueueId 获取队列ID最大值
|
|
func (this *TOAConfig) MaxQueueId() uint8 {
|
|
var maxQueueId = int(this.MinQueueId) + runtime.NumCPU() - 1
|
|
if maxQueueId > 255 {
|
|
maxQueueId = 255
|
|
}
|
|
return uint8(maxQueueId)
|
|
}
|