Adam Harrison
0f492b563e
Tests weren't compiling due to change of ListenHttp() signature. Fixed for now by reintroducing domain parameter in addition to server, as this allows the HTTP tests to pass in a nil server parameter. In the long term DNSServer should probably become an interface so we can mock it easily in such a scenario.
97 lines
2.7 KiB
Go
97 lines
2.7 KiB
Go
package nameserver
|
|
|
|
import (
|
|
"fmt"
|
|
"github.com/miekg/dns"
|
|
. "github.com/zettio/weave/common"
|
|
"io"
|
|
"log"
|
|
"net"
|
|
"net/http"
|
|
"github.com/gorilla/mux"
|
|
)
|
|
|
|
func httpErrorAndLog(level *log.Logger, w http.ResponseWriter, msg string,
|
|
status int, logmsg string, logargs ...interface{}) {
|
|
http.Error(w, msg, status)
|
|
level.Printf("[http] "+logmsg, logargs...)
|
|
}
|
|
|
|
func ListenHttp(version string, server *DNSServer, domain string, db Zone, port int) {
|
|
|
|
router := mux.NewRouter()
|
|
|
|
router.Methods("GET").Path("/status").HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
|
io.WriteString(w, fmt.Sprintln("weave DNS", version))
|
|
io.WriteString(w, server.Status())
|
|
})
|
|
|
|
router.Methods("PUT").Path("/name/{identifier:.+}/{ip:.+}").HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
|
reqError := func(msg string, logmsg string, logargs ...interface{}) {
|
|
httpErrorAndLog(Warning, w, msg, http.StatusBadRequest, logmsg, logargs...)
|
|
}
|
|
|
|
vars := mux.Vars(r)
|
|
ident := vars["identifier"]
|
|
ipStr := vars["ip"]
|
|
name := r.FormValue("fqdn")
|
|
|
|
if name == "" {
|
|
reqError("Invalid FQDN", "Invalid FQDN in request: %s, %s", r.URL, r.Form)
|
|
return
|
|
}
|
|
|
|
ip := net.ParseIP(ipStr)
|
|
if ip == nil {
|
|
reqError("Invalid IP", "Invalid IP in request: %s", ipStr)
|
|
return
|
|
}
|
|
|
|
if dns.IsSubDomain(domain, name) {
|
|
Info.Printf("[http] Adding %s -> %s", name, ipStr)
|
|
if err := db.AddRecord(ident, name, ip); err != nil {
|
|
if _, ok := err.(DuplicateError); !ok {
|
|
httpErrorAndLog(
|
|
Error, w, "Internal error", http.StatusInternalServerError,
|
|
"Unexpected error from DB: %s", err)
|
|
return
|
|
} // oh, I already know this. whatever.
|
|
}
|
|
} else {
|
|
Info.Printf("[http] Ignoring name %s, not in %s", name, domain)
|
|
}
|
|
})
|
|
|
|
router.Methods("DELETE").Path("/name/{identifier:.+}/{ip:.+}").HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
|
reqError := func(msg string, logmsg string, logargs ...interface{}) {
|
|
httpErrorAndLog(Warning, w, msg, http.StatusBadRequest, logmsg, logargs...)
|
|
}
|
|
|
|
vars := mux.Vars(r)
|
|
ident := vars["identifier"]
|
|
ipStr := vars["ip"]
|
|
|
|
ip := net.ParseIP(ipStr)
|
|
if ip == nil {
|
|
reqError("Invalid IP in request", "Invalid IP in request: %s", ipStr)
|
|
return
|
|
}
|
|
Info.Printf("[http] Deleting %s (%s)", ident, ipStr)
|
|
if err := db.DeleteRecord(ident, ip); err != nil {
|
|
if _, ok := err.(LookupError); !ok {
|
|
httpErrorAndLog(
|
|
Error, w, "Internal error", http.StatusInternalServerError,
|
|
"Unexpected error from DB: %s", err)
|
|
return
|
|
}
|
|
}
|
|
})
|
|
|
|
http.Handle("/", router)
|
|
|
|
address := fmt.Sprintf(":%d", port)
|
|
if err := http.ListenAndServe(address, nil); err != nil {
|
|
Error.Fatal("[http] Unable to create http listener: ", err)
|
|
}
|
|
}
|