mirror of https://github.com/v2ray/v2ray-core
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
40 lines
724 B
40 lines
724 B
package dns
|
|
|
|
import (
|
|
"net"
|
|
|
|
"github.com/v2ray/v2ray-core/app"
|
|
)
|
|
|
|
const (
|
|
APP_ID = app.ID(2)
|
|
)
|
|
|
|
// A DnsCache is an internal cache of DNS resolutions.
|
|
type Server interface {
|
|
Get(domain string) []net.IP
|
|
}
|
|
|
|
type dnsServerWithContext interface {
|
|
Get(context app.Context, domain string) []net.IP
|
|
}
|
|
|
|
type contextedDnsServer struct {
|
|
context app.Context
|
|
dnsCache dnsServerWithContext
|
|
}
|
|
|
|
func (this *contextedDnsServer) Get(domain string) []net.IP {
|
|
return this.dnsCache.Get(this.context, domain)
|
|
}
|
|
|
|
func init() {
|
|
app.Register(APP_ID, func(context app.Context, obj interface{}) interface{} {
|
|
dcContext := obj.(dnsServerWithContext)
|
|
return &contextedDnsServer{
|
|
context: context,
|
|
dnsCache: dcContext,
|
|
}
|
|
})
|
|
}
|