WireGold/config/cfg.go

package config

import (
	"bytes"
	"log"
	"os"

	"gopkg.in/yaml.v3"
)

// EndPoint 一个终结点的信息
type EndPoint struct {
	Host             string `yaml:"Host"`
	Port             int64  `yaml:"Port"`
	Poly             uint64 `yaml:"Poly"`             // Poly 是 port 随机切换算法的生成多项式, 0 为禁用
	Protocol         string `yaml:"Protocol"`         // Protocol is udp/tcp
	ReconnectSeconds int64  `yaml:"ReconnectSeconds"` // ReconnectSeconds 断开重连间隔, 每次到时即向对端通报并切换到新的端口, 0 为禁用
	FECMethod        string `yaml:"FECMethod"`        // FECMethod 可选 1/2 2/3
}

// Config WireGold 配置文件
type Config struct {
	IP         string `yaml:"IP"`
	SubNet     string `yaml:"SubNet"`
	PrivateKey string `yaml:"PrivateKey"`
	EndPoint   string `yaml:"EndPoint"`
	MTU        int64  `yaml:"MTU"`
	Mask       uint64 `yaml:"Mask"` // Mask 是异或报文所用掩码, 必须保证各端统一
	Peers      []Peer `yaml:"Peers"`
}

// Peer 对端信息
type Peer struct {
	IP               string   `yaml:"IP"`
	SubNet           string   `yaml:"SubNet"`
	PublicKey        string   `yaml:"PublicKey"`
	PresharedKey     string   `yaml:"PresharedKey"`
	EndPoint         string   `yaml:"EndPoint"`
	AllowedIPs       []string `yaml:"AllowedIPs"`
	KeepAliveSeconds int64    `yaml:"KeepAliveSeconds"`
	QueryList        []string `yaml:"QueryList"`
	QuerySeconds     int64    `yaml:"QuerySeconds"`
	AllowTrans       bool     `yaml:"AllowTrans"`
	UseZstd          bool     `yaml:"UseZstd"`
	MTU              int64    `yaml:"MTU"`
	MTURandomRange   int64    `yaml:"MTURandomRange"`
}

// Parse 解析配置文件
func Parse(path string) (c Config) {
	file, err := os.ReadFile(path)
	if err != nil {
		log.Fatal("open config file failed:", err)
	}
	err = yaml.NewDecoder(bytes.NewReader(file)).Decode(&c)
	if err != nil {
		log.Fatal("invalid config file:", err)
	}
	return
}