mirror of
https://github.com/xvzc/SpoofDPI.git
synced 2024-12-23 06:42:51 +00:00
50 lines
756 B
Go
50 lines
756 B
Go
package doh
|
|
|
|
import (
|
|
"errors"
|
|
"sync"
|
|
|
|
"regexp"
|
|
|
|
"github.com/babolivier/go-doh-client"
|
|
)
|
|
|
|
var resolver *doh.Resolver
|
|
var once sync.Once
|
|
|
|
func Init(dns string) {
|
|
getInstance().Host = dns
|
|
}
|
|
|
|
func Lookup(domain string) (string, error) {
|
|
ipRegex := "^(?:(?:25[0-5]|2[0-4][0-9]|[01]?[0-9][0-9]?)\\.){3}(?:25[0-5]|2[0-4][0-9]|[01]?[0-9][0-9]?)$"
|
|
|
|
if r, _ := regexp.MatchString(ipRegex, domain); r {
|
|
return domain, nil
|
|
}
|
|
|
|
a, _, err := resolver.LookupA(domain)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
|
|
if len(a) < 1 {
|
|
return "", errors.New(" couldn't resolve the domain")
|
|
}
|
|
|
|
ip := a[0].IP4
|
|
|
|
return ip, nil
|
|
}
|
|
|
|
func getInstance() *doh.Resolver {
|
|
once.Do(func() {
|
|
resolver = &doh.Resolver{
|
|
Host: "",
|
|
Class: doh.IN,
|
|
}
|
|
})
|
|
|
|
return resolver
|
|
}
|