You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
84 lines
1.8 KiB
Go
84 lines
1.8 KiB
Go
package main
|
|
|
|
import (
|
|
"crypto/tls"
|
|
"io/ioutil"
|
|
"net"
|
|
"net/http"
|
|
|
|
"github.com/fangdingjun/go-log"
|
|
"github.com/fangdingjun/nghttp2-go"
|
|
"github.com/miekg/dns"
|
|
)
|
|
|
|
func (srv *server) handleHTTPSConn(c net.Conn) {
|
|
defer c.Close()
|
|
tlsconn := c.(*tls.Conn)
|
|
if err := tlsconn.Handshake(); err != nil {
|
|
log.Errorln("handshake", err)
|
|
return
|
|
}
|
|
state := tlsconn.ConnectionState()
|
|
if state.NegotiatedProtocol != "h2" {
|
|
log.Errorln("http2 is needed")
|
|
return
|
|
}
|
|
h2conn, err := nghttp2.Server(tlsconn, srv)
|
|
if err != nil {
|
|
log.Errorf("create http2 error: %s", err)
|
|
return
|
|
}
|
|
h2conn.Run()
|
|
}
|
|
|
|
func (srv *server) handleHTTP2Req(w http.ResponseWriter, r *http.Request) {
|
|
ctype := r.Header.Get("content-type")
|
|
if ctype != "application/dns-message" {
|
|
http.Error(w, "dns message is required", http.StatusBadRequest)
|
|
return
|
|
}
|
|
|
|
data, err := ioutil.ReadAll(r.Body)
|
|
if err != nil {
|
|
log.Errorln("read request body", err)
|
|
w.WriteHeader(http.StatusBadRequest)
|
|
return
|
|
}
|
|
|
|
msg := new(dns.Msg)
|
|
if err := msg.Unpack(data); err != nil {
|
|
log.Errorln("parse dns message", err)
|
|
return
|
|
}
|
|
reply := false
|
|
for _, up := range srv.upstreams {
|
|
log.Debugf("from %s query upstream %s", r.RemoteAddr, up.String())
|
|
log.Debugln("query", msg.Question[0].String())
|
|
m, err := queryUpstream(msg, up)
|
|
if err == nil {
|
|
w.Header().Set("content-type", "application/dns-message")
|
|
w.WriteHeader(http.StatusOK)
|
|
for _, a := range m.Answer {
|
|
log.Debugln("result", a.String())
|
|
}
|
|
d, _ := m.Pack()
|
|
w.Write(d)
|
|
reply = true
|
|
break
|
|
} else {
|
|
log.Errorf("https query upstream %s", err)
|
|
}
|
|
}
|
|
if !reply {
|
|
w.WriteHeader(http.StatusServiceUnavailable)
|
|
}
|
|
}
|
|
|
|
func (srv *server) ServeHTTP(w http.ResponseWriter, r *http.Request) {
|
|
if r.RequestURI != srv.addr.Path {
|
|
w.WriteHeader(http.StatusNotFound)
|
|
return
|
|
}
|
|
srv.handleHTTP2Req(w, r)
|
|
}
|