mirror of
https://github.com/projectdiscovery/nuclei.git
synced 2025-12-17 18:45:28 +00:00
37 lines
836 B
Go
37 lines
836 B
Go
package protocolstate
|
|
|
|
import (
|
|
"github.com/pkg/errors"
|
|
|
|
"github.com/projectdiscovery/fastdialer/fastdialer"
|
|
"github.com/projectdiscovery/nuclei/v2/pkg/types"
|
|
)
|
|
|
|
// Dialer is a shared fastdialer instance for host DNS resolution
|
|
var Dialer *fastdialer.Dialer
|
|
|
|
// Init creates the Dialer instance based on user configuration
|
|
func Init(options *types.Options) error {
|
|
opts := fastdialer.DefaultOptions
|
|
if options.SystemResolvers {
|
|
opts.EnableFallback = true
|
|
}
|
|
if options.ResolversFile != "" {
|
|
opts.BaseResolvers = options.InternalResolversList
|
|
}
|
|
opts.WithDialerHistory = true
|
|
dialer, err := fastdialer.NewDialer(opts)
|
|
if err != nil {
|
|
return errors.Wrap(err, "could not create dialer")
|
|
}
|
|
Dialer = dialer
|
|
return nil
|
|
}
|
|
|
|
// Close closes the global shared fastdialer
|
|
func Close() {
|
|
if Dialer != nil {
|
|
Dialer.Close()
|
|
}
|
|
}
|