e635b4e773
Add port 53 in the proxy host if not specified. Check if the host is actually an IP address (v4 or v6) Remove the http headers and other TODOs
36 lines
641 B
Go
36 lines
641 B
Go
// Package proxy is middleware that proxies requests.
|
|
package proxy
|
|
|
|
import (
|
|
"github.com/miekg/coredns/middleware"
|
|
|
|
"github.com/miekg/dns"
|
|
)
|
|
|
|
type ReverseProxy struct {
|
|
Host string
|
|
Client Client
|
|
Options Options
|
|
}
|
|
|
|
func (p ReverseProxy) ServeDNS(w dns.ResponseWriter, r *dns.Msg, extra []dns.RR) error {
|
|
var (
|
|
reply *dns.Msg
|
|
err error
|
|
)
|
|
|
|
switch {
|
|
case middleware.Proto(w) == "tcp":
|
|
reply, err = middleware.Exchange(p.Client.TCP, r, p.Host)
|
|
default:
|
|
reply, err = middleware.Exchange(p.Client.UDP, r, p.Host)
|
|
}
|
|
|
|
if err != nil {
|
|
return err
|
|
}
|
|
reply.Compress = true
|
|
reply.Id = r.Id
|
|
w.WriteMsg(reply)
|
|
return nil
|
|
}
|