Removing unnecessary ProxyLogger nil check

By ensuring it is initialized before codes that access it.
This commit is contained in:
Cuong Manh Le
2025-02-21 20:44:34 +07:00
committed by Cuong Manh Le
parent 54cb455522
commit 9fe36ae984
3 changed files with 23 additions and 39 deletions
-5
View File
@@ -9,11 +9,6 @@ import (
"github.com/rs/zerolog" "github.com/rs/zerolog"
) )
func init() {
l := zerolog.New(io.Discard)
ProxyLogger.Store(&l)
}
// ProxyLog emits the log record for proxy operations. // ProxyLog emits the log record for proxy operations.
// The caller should set it only once. // The caller should set it only once.
// DEPRECATED: use ProxyLogger instead. // DEPRECATED: use ProxyLogger instead.
+6 -17
View File
@@ -17,7 +17,6 @@ import (
"github.com/microsoft/wmi/pkg/base/query" "github.com/microsoft/wmi/pkg/base/query"
"github.com/microsoft/wmi/pkg/constant" "github.com/microsoft/wmi/pkg/constant"
"github.com/microsoft/wmi/pkg/hardware/network/netadapter" "github.com/microsoft/wmi/pkg/hardware/network/netadapter"
"github.com/rs/zerolog"
"golang.org/x/sys/windows" "golang.org/x/sys/windows"
"golang.zx2c4.com/wireguard/windows/tunnel/winipcfg" "golang.zx2c4.com/wireguard/windows/tunnel/winipcfg"
"tailscale.com/net/netmon" "tailscale.com/net/netmon"
@@ -63,10 +62,7 @@ func dnsFromAdapter() []string {
var ns []string var ns []string
var err error var err error
logger := zerolog.New(io.Discard) logger := *ProxyLogger.Load()
if ProxyLogger.Load() != nil {
logger = *ProxyLogger.Load()
}
for i := 0; i < maxDNSAdapterRetries; i++ { for i := 0; i < maxDNSAdapterRetries; i++ {
if ctx.Err() != nil { if ctx.Err() != nil {
@@ -112,10 +108,8 @@ func dnsFromAdapter() []string {
} }
func getDNSServers(ctx context.Context) ([]string, error) { func getDNSServers(ctx context.Context) ([]string, error) {
logger := zerolog.New(io.Discard) logger := *ProxyLogger.Load()
if ProxyLogger.Load() != nil {
logger = *ProxyLogger.Load()
}
// Check context before making the call // Check context before making the call
if ctx.Err() != nil { if ctx.Err() != nil {
return nil, ctx.Err() return nil, ctx.Err()
@@ -343,10 +337,8 @@ func nameserversFromResolvconf() []string {
// checkDomainJoined checks if the machine is joined to an Active Directory domain // checkDomainJoined checks if the machine is joined to an Active Directory domain
// Returns whether it's domain joined and the domain name if available // Returns whether it's domain joined and the domain name if available
func checkDomainJoined() bool { func checkDomainJoined() bool {
logger := zerolog.New(io.Discard) logger := *ProxyLogger.Load()
if ProxyLogger.Load() != nil {
logger = *ProxyLogger.Load()
}
var domain *uint16 var domain *uint16
var status uint32 var status uint32
@@ -423,10 +415,7 @@ func validInterfaces() map[string]struct{} {
defer log.SetOutput(os.Stderr) defer log.SetOutput(os.Stderr)
//load the logger //load the logger
logger := zerolog.New(io.Discard) logger := *ProxyLogger.Load()
if ProxyLogger.Load() != nil {
logger = *ProxyLogger.Load()
}
whost := host.NewWmiLocalHost() whost := host.NewWmiLocalHost()
q := query.NewWmiQuery("MSFT_NetAdapter") q := query.NewWmiQuery("MSFT_NetAdapter")
+17 -17
View File
@@ -48,7 +48,15 @@ const (
var controldPublicDnsWithPort = net.JoinHostPort(controldPublicDns, "53") var controldPublicDnsWithPort = net.JoinHostPort(controldPublicDns, "53")
var localResolver = newLocalResolver() var localResolver Resolver
func init() {
// Initializing ProxyLogger here, so other places don't have to do nil check.
l := zerolog.New(io.Discard)
ProxyLogger.Store(&l)
localResolver = newLocalResolver()
}
var ( var (
resolverMutex sync.Mutex resolverMutex sync.Mutex
@@ -91,10 +99,8 @@ func availableNameservers() []string {
machineIPsMap := make(map[string]struct{}, len(regularIPs)) machineIPsMap := make(map[string]struct{}, len(regularIPs))
//load the logger //load the logger
logger := zerolog.New(io.Discard) logger := *ProxyLogger.Load()
if ProxyLogger.Load() != nil {
logger = *ProxyLogger.Load()
}
Log(context.Background(), logger.Debug(), Log(context.Background(), logger.Debug(),
"Got local addresses - regular IPs: %v, loopback IPs: %v", regularIPs, loopbackIPs) "Got local addresses - regular IPs: %v, loopback IPs: %v", regularIPs, loopbackIPs)
@@ -549,10 +555,8 @@ func lookupIP(domain string, timeout int, withBootstrapDNS bool) (ips []string)
// - Gateway IP address (depends on OS). // - Gateway IP address (depends on OS).
// - Input servers. // - Input servers.
func NewBootstrapResolver(servers ...string) Resolver { func NewBootstrapResolver(servers ...string) Resolver {
logger := zerolog.New(io.Discard) logger := *ProxyLogger.Load()
if ProxyLogger.Load() != nil {
logger = *ProxyLogger.Load()
}
Log(context.Background(), logger.Debug(), "NewBootstrapResolver called with servers: %v", servers) Log(context.Background(), logger.Debug(), "NewBootstrapResolver called with servers: %v", servers)
nss := defaultNameservers() nss := defaultNameservers()
nss = append([]string{controldPublicDnsWithPort}, nss...) nss = append([]string{controldPublicDnsWithPort}, nss...)
@@ -571,10 +575,8 @@ func NewBootstrapResolver(servers ...string) Resolver {
// This is useful for doing PTR lookup in LAN network. // This is useful for doing PTR lookup in LAN network.
func NewPrivateResolver() Resolver { func NewPrivateResolver() Resolver {
logger := zerolog.New(io.Discard) logger := *ProxyLogger.Load()
if ProxyLogger.Load() != nil {
logger = *ProxyLogger.Load()
}
Log(context.Background(), logger.Debug(), "NewPrivateResolver called") Log(context.Background(), logger.Debug(), "NewPrivateResolver called")
nss := defaultNameservers() nss := defaultNameservers()
@@ -621,10 +623,8 @@ func NewResolverWithNameserver(nameservers []string) Resolver {
// newResolverWithNameserver returns an OS resolver from given nameservers list. // newResolverWithNameserver returns an OS resolver from given nameservers list.
// The caller must ensure each server in list is formed "ip:53". // The caller must ensure each server in list is formed "ip:53".
func newResolverWithNameserver(nameservers []string) *osResolver { func newResolverWithNameserver(nameservers []string) *osResolver {
logger := zerolog.New(io.Discard) logger := *ProxyLogger.Load()
if ProxyLogger.Load() != nil {
logger = *ProxyLogger.Load()
}
Log(context.Background(), logger.Debug(), "newResolverWithNameserver called with nameservers: %v", nameservers) Log(context.Background(), logger.Debug(), "newResolverWithNameserver called with nameservers: %v", nameservers)
r := &osResolver{} r := &osResolver{}
var publicNss []string var publicNss []string