Bryan Boreham
b0274e11be
instead of assuming it's on the Docker bridge Also move DNS address validation into NewDNSServer(), so we only do it once.
98 lines
2.6 KiB
Go
98 lines
2.6 KiB
Go
package nameserver
|
|
|
|
import (
|
|
"encoding/json"
|
|
"fmt"
|
|
"net"
|
|
"net/http"
|
|
|
|
"github.com/gorilla/mux"
|
|
"github.com/miekg/dns"
|
|
|
|
"github.com/weaveworks/weave/common/docker"
|
|
"github.com/weaveworks/weave/net/address"
|
|
)
|
|
|
|
func (n *Nameserver) badRequest(w http.ResponseWriter, err error) {
|
|
http.Error(w, err.Error(), http.StatusBadRequest)
|
|
n.infof("%v", err)
|
|
}
|
|
|
|
func (n *Nameserver) HandleHTTP(router *mux.Router, dockerCli *docker.Client) {
|
|
router.Methods("GET").Path("/domain").HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
|
fmt.Fprint(w, n.domain)
|
|
})
|
|
|
|
router.Methods("PUT").Path("/name/{container}/{ip}").HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
|
var (
|
|
vars = mux.Vars(r)
|
|
container = vars["container"]
|
|
ipStr = vars["ip"]
|
|
fqdn = r.FormValue("fqdn")
|
|
ip, err = address.ParseIP(ipStr)
|
|
)
|
|
if err != nil {
|
|
n.badRequest(w, err)
|
|
return
|
|
}
|
|
|
|
n.AddEntryFQDN(fqdn, container, n.ourName, ip)
|
|
|
|
if r.FormValue("check-alive") == "true" && dockerCli != nil && dockerCli.IsContainerNotRunning(container) {
|
|
n.infof("container '%s' is not running: removing", container)
|
|
n.Delete(dns.Fqdn(fqdn), container, ipStr, ip)
|
|
}
|
|
|
|
w.WriteHeader(204)
|
|
})
|
|
|
|
deleteHandler := func(w http.ResponseWriter, r *http.Request) {
|
|
vars := mux.Vars(r)
|
|
|
|
hostname := r.FormValue("fqdn")
|
|
if hostname == "" {
|
|
hostname = "*"
|
|
} else {
|
|
hostname = dns.Fqdn(hostname)
|
|
}
|
|
|
|
container, ok := vars["container"]
|
|
if !ok {
|
|
container = "*"
|
|
}
|
|
|
|
ipStr, ok := vars["ip"]
|
|
ip, err := address.ParseIP(ipStr)
|
|
if ok && err != nil {
|
|
n.badRequest(w, err)
|
|
return
|
|
} else if !ok {
|
|
ipStr = "*"
|
|
}
|
|
|
|
n.Delete(hostname, container, ipStr, ip)
|
|
w.WriteHeader(204)
|
|
}
|
|
router.Methods("DELETE").Path("/name/{container}/{ip}").HandlerFunc(deleteHandler)
|
|
router.Methods("DELETE").Path("/name/{container}").HandlerFunc(deleteHandler)
|
|
router.Methods("DELETE").Path("/name").HandlerFunc(deleteHandler)
|
|
|
|
router.Methods("GET").Path("/name").Headers("Accept", "application/json").HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
|
n.RLock()
|
|
defer n.RUnlock()
|
|
if err := json.NewEncoder(w).Encode(n.entries); err != nil {
|
|
n.badRequest(w, fmt.Errorf("Error marshalling response: %v", err))
|
|
}
|
|
})
|
|
}
|
|
|
|
func (d *DNSServer) HandleHTTP(router *mux.Router) {
|
|
router.Methods("GET").Path("/dns-address-port").HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
|
fmt.Fprint(w, d.address)
|
|
})
|
|
router.Methods("GET").Path("/dns-address").HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
|
host, _, _ := net.SplitHostPort(d.address) // address is validated in NewDNSServer()
|
|
fmt.Fprint(w, host)
|
|
})
|
|
}
|