mirror of
https://github.com/coredns/coredns.git
synced 2025-11-02 02:03:13 -05:00
First commit
This commit is contained in:
36
middleware/proxy/reverseproxy.go
Normal file
36
middleware/proxy/reverseproxy.go
Normal file
@@ -0,0 +1,36 @@
|
||||
// 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
|
||||
}
|
||||
|
||||
func (p ReverseProxy) ServeDNS(w dns.ResponseWriter, r *dns.Msg, extra []dns.RR) error {
|
||||
// TODO(miek): use extra!
|
||||
var (
|
||||
reply *dns.Msg
|
||||
err error
|
||||
)
|
||||
context := middleware.Context{W: w, Req: r}
|
||||
|
||||
// tls+tcp ?
|
||||
if context.Proto() == "tcp" {
|
||||
reply, err = middleware.Exchange(p.Client.TCP, r, p.Host)
|
||||
} else {
|
||||
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
|
||||
}
|
||||
Reference in New Issue
Block a user