Improve flags

This commit is contained in:
Daniel 2021-05-05 00:23:48 +02:00
parent 168ef32c44
commit 48b55a88bc
5 changed files with 18 additions and 34 deletions

View file

@ -4,7 +4,6 @@ import (
"flag"
"github.com/safing/portbase/config"
"github.com/safing/portbase/log"
)
// Configuration Keys.
@ -19,13 +18,12 @@ var (
)
func init() {
flag.BoolVar(&defaultNetworkServiceMode, "network-service", false, "force network service mode")
}
func logFlagOverrides() {
if defaultNetworkServiceMode {
log.Warningf("core: %s config is being forced by the -network-service flag", CfgNetworkServiceKey)
}
flag.BoolVar(
&defaultNetworkServiceMode,
"network-service",
false,
"set default network service mode; configuration is stronger",
)
}
func registerConfig() error {

View file

@ -15,7 +15,7 @@ var (
)
func init() {
flag.BoolVar(&disableInterception, "disable-interception", false, "disable packet interception - this breaks a lot of functionality")
flag.BoolVar(&disableInterception, "disable-interception", false, "disable packet interception; this breaks a lot of functionality")
}
// Start starts the interception.

View file

@ -18,7 +18,7 @@ var (
)
func init() {
flag.StringVar(&packetMetricsDestination, "write-packet-metrics", "", "Write packet metrics to the specified file")
flag.StringVar(&packetMetricsDestination, "write-packet-metrics", "", "write packet metrics to the specified file")
}
type (

View file

@ -5,7 +5,6 @@ import (
"runtime"
"github.com/safing/portbase/config"
"github.com/safing/portbase/log"
"github.com/safing/portmaster/core"
)
@ -15,10 +14,9 @@ const (
)
var (
nameserverAddressFlag string
nameserverAddressConfig config.StringOption
defaultNameserverAddress = "localhost:53"
nameserverAddress string
nameserverAddressConfig config.StringOption
networkServiceMode config.BoolOption
)
@ -29,22 +27,12 @@ func init() {
defaultNameserverAddress = "0.0.0.0:53"
}
flag.StringVar(&nameserverAddressFlag, "nameserver-address", "", "override nameserver listen address")
}
func logFlagOverrides() {
if nameserverAddressFlag != "" {
log.Warning("nameserver: dns/listenAddress default config is being overridden by the -nameserver-address flag")
}
}
func getDefaultNameserverAddress() string {
// check if overridden
if nameserverAddressFlag != "" {
return nameserverAddressFlag
}
// return internal default
return defaultNameserverAddress
flag.StringVar(
&nameserverAddress,
"nameserver-address",
defaultNameserverAddress,
"set default nameserver address; configuration is stronger",
)
}
func registerConfig() error {
@ -55,7 +43,7 @@ func registerConfig() error {
OptType: config.OptTypeString,
ExpertiseLevel: config.ExpertiseLevelDeveloper,
ReleaseLevel: config.ReleaseLevelStable,
DefaultValue: getDefaultNameserverAddress(),
DefaultValue: nameserverAddress,
ValidationRegex: "^(localhost|[0-9]{1,3}.[0-9]{1,3}.[0-9]{1,3}.[0-9]{1,3}|\\[[:0-9A-Fa-f]+\\]):[0-9]{1,5}$",
RequiresRestart: true,
Annotations: config.Annotations{
@ -66,7 +54,7 @@ func registerConfig() error {
if err != nil {
return err
}
nameserverAddressConfig = config.GetAsString(CfgDefaultNameserverAddressKey, getDefaultNameserverAddress())
nameserverAddressConfig = config.GetAsString(CfgDefaultNameserverAddressKey, nameserverAddress)
networkServiceMode = config.Concurrent.GetAsBool(core.CfgNetworkServiceKey, false)

View file

@ -37,8 +37,6 @@ func prep() error {
}
func start() error {
logFlagOverrides()
if err := registerMetrics(); err != nil {
return err
}