Fix .pk.ygg resolver after refactoring

Signed-off-by: Vasyl Gello <vasek.gello@gmail.com>
This commit is contained in:
Vasyl Gello 2024-07-23 22:44:14 +03:00
parent 5008a4ea2d
commit 2c09a1d7b2
No known key found for this signature in database
GPG key ID: 8A52BC6C291FB280

View file

@ -292,13 +292,14 @@ func main() {
socksOptions := []socks5.Option{ socksOptions := []socks5.Option{
socks5.WithDial(s.DialContext), socks5.WithDial(s.DialContext),
} }
if nameserver != nil && *nameserver != "" { if nameserver != nil {
resolver := types.NewNameResolver(s, *nameserver) if *nameserver == "" {
socksOptions = append(socksOptions, socks5.WithResolver(resolver))
} else {
logger.Infof("DNS nameserver is not set!") logger.Infof("DNS nameserver is not set!")
logger.Infof("SOCKS server will not be able to resolve hostnames other than .pk.ygg !") logger.Infof("SOCKS server will not be able to resolve hostnames other than .pk.ygg !")
} }
resolver := types.NewNameResolver(s, *nameserver)
socksOptions = append(socksOptions, socks5.WithResolver(resolver))
}
if logger.GetLevel("debug") { if logger.GetLevel("debug") {
socksOptions = append(socksOptions, socks5.WithLogger(logger)) socksOptions = append(socksOptions, socks5.WithLogger(logger))
} }