dtlspipe/client/config.go

package client

import (
	"context"
	"net"
	"time"

	"github.com/SenseUnit/dtlspipe/ciphers"
	"github.com/SenseUnit/dtlspipe/util"
)

type Config struct {
	BindAddress    string
	RemoteDialFunc func(ctx context.Context) (net.PacketConn, net.Addr, error)
	Timeout        time.Duration
	IdleTimeout    time.Duration
	BaseContext    context.Context
	PSKCallback    func([]byte) ([]byte, error)
	PSKIdentity    string
	MTU            int
	CipherSuites   ciphers.CipherList
	EllipticCurves ciphers.CurveList
	StaleMode      util.StaleMode
	TimeLimitFunc  func() time.Duration
	AllowFunc      func(net.Addr) bool
	EnableCID      bool
}

func (cfg *Config) populateDefaults() *Config {
	newCfg := new(Config)
	*newCfg = *cfg
	cfg = newCfg
	if cfg.BaseContext == nil {
		cfg.BaseContext = context.Background()
	}
	if cfg.IdleTimeout == 0 {
		cfg.IdleTimeout = 90 * time.Second
	}
	if cfg.CipherSuites == nil {
		cfg.CipherSuites = ciphers.DefaultCipherList
	}
	if cfg.EllipticCurves == nil {
		cfg.EllipticCurves = ciphers.DefaultCurveList
	}
	if cfg.TimeLimitFunc == nil {
		cfg.TimeLimitFunc = util.FixedTimeLimitFunc(0)
	}
	if cfg.AllowFunc == nil {
		cfg.AllowFunc = util.AllowAllFunc
	}
	return cfg
}