mirror of
https://github.com/netbirdio/netbird.git
synced 2024-12-15 19:31:06 +01:00
78 lines
2.4 KiB
Go
78 lines
2.4 KiB
Go
package peer
|
|
|
|
import (
|
|
"os"
|
|
"strconv"
|
|
"strings"
|
|
"time"
|
|
|
|
log "github.com/sirupsen/logrus"
|
|
)
|
|
|
|
const (
|
|
envICEKeepAliveIntervalSec = "NB_ICE_KEEP_ALIVE_INTERVAL_SEC"
|
|
envICEDisconnectedTimeoutSec = "NB_ICE_DISCONNECTED_TIMEOUT_SEC"
|
|
envICERelayAcceptanceMinWaitSec = "NB_ICE_RELAY_ACCEPTANCE_MIN_WAIT_SEC"
|
|
envICEForceRelayConn = "NB_ICE_FORCE_RELAY_CONN"
|
|
)
|
|
|
|
func ForcedRelayAddress() string {
|
|
if envRelay := os.Getenv("NB_RELAY_ADDRESS"); envRelay != "" {
|
|
return envRelay
|
|
}
|
|
return ""
|
|
}
|
|
|
|
func iceKeepAlive() time.Duration {
|
|
keepAliveEnv := os.Getenv(envICEKeepAliveIntervalSec)
|
|
if keepAliveEnv == "" {
|
|
return iceKeepAliveDefault
|
|
}
|
|
|
|
log.Infof("setting ICE keep alive interval to %s seconds", keepAliveEnv)
|
|
keepAliveEnvSec, err := strconv.Atoi(keepAliveEnv)
|
|
if err != nil {
|
|
log.Warnf("invalid value %s set for %s, using default %v", keepAliveEnv, envICEKeepAliveIntervalSec, iceKeepAliveDefault)
|
|
return iceKeepAliveDefault
|
|
}
|
|
|
|
return time.Duration(keepAliveEnvSec) * time.Second
|
|
}
|
|
|
|
func iceDisconnectedTimeout() time.Duration {
|
|
disconnectedTimeoutEnv := os.Getenv(envICEDisconnectedTimeoutSec)
|
|
if disconnectedTimeoutEnv == "" {
|
|
return iceDisconnectedTimeoutDefault
|
|
}
|
|
|
|
log.Infof("setting ICE disconnected timeout to %s seconds", disconnectedTimeoutEnv)
|
|
disconnectedTimeoutSec, err := strconv.Atoi(disconnectedTimeoutEnv)
|
|
if err != nil {
|
|
log.Warnf("invalid value %s set for %s, using default %v", disconnectedTimeoutEnv, envICEDisconnectedTimeoutSec, iceDisconnectedTimeoutDefault)
|
|
return iceDisconnectedTimeoutDefault
|
|
}
|
|
|
|
return time.Duration(disconnectedTimeoutSec) * time.Second
|
|
}
|
|
|
|
func iceRelayAcceptanceMinWait() time.Duration {
|
|
iceRelayAcceptanceMinWaitEnv := os.Getenv(envICERelayAcceptanceMinWaitSec)
|
|
if iceRelayAcceptanceMinWaitEnv == "" {
|
|
return iceRelayAcceptanceMinWaitDefault
|
|
}
|
|
|
|
log.Infof("setting ICE relay acceptance min wait to %s seconds", iceRelayAcceptanceMinWaitEnv)
|
|
disconnectedTimeoutSec, err := strconv.Atoi(iceRelayAcceptanceMinWaitEnv)
|
|
if err != nil {
|
|
log.Warnf("invalid value %s set for %s, using default %v", iceRelayAcceptanceMinWaitEnv, envICERelayAcceptanceMinWaitSec, iceRelayAcceptanceMinWaitDefault)
|
|
return iceRelayAcceptanceMinWaitDefault
|
|
}
|
|
|
|
return time.Duration(disconnectedTimeoutSec) * time.Second
|
|
}
|
|
|
|
func hasICEForceRelayConn() bool {
|
|
disconnectedTimeoutEnv := os.Getenv(envICEForceRelayConn)
|
|
return strings.ToLower(disconnectedTimeoutEnv) == "true"
|
|
}
|