Bryan Boreham
6bff651e87
Implementation matches design.md better: - http action changes from POST /add to PUT /name - each record now has the container identifier
54 lines
1.4 KiB
Go
54 lines
1.4 KiB
Go
package weavedns
|
|
|
|
import (
|
|
"errors"
|
|
"fmt"
|
|
"log"
|
|
"net"
|
|
"net/http"
|
|
"strings"
|
|
)
|
|
|
|
// Parse a URL of the form PUT /name/<identifier>/<ip-address>
|
|
func parseUrl(url string) (identifier string, ipaddr string, err error) {
|
|
parts := strings.Split(url, "/")
|
|
if len(parts) != 4 {
|
|
return "", "", errors.New("Unable to parse url: " + url)
|
|
}
|
|
return parts[2], parts[3], nil
|
|
}
|
|
|
|
func ListenHttp(db Zone, port int) {
|
|
http.HandleFunc("/name/", func(w http.ResponseWriter, r *http.Request) {
|
|
switch r.Method {
|
|
case "PUT":
|
|
identifier, weave_ipstr, err := parseUrl(r.URL.Path)
|
|
name := r.FormValue("fqdn")
|
|
prefix := r.FormValue("routing_prefix")
|
|
local_ip := r.FormValue("local_ip")
|
|
if identifier == "" || weave_ipstr == "" || name == "" || prefix == "" || local_ip == "" {
|
|
log.Printf("Invalid request: %s, %s", r.URL, r.Form)
|
|
return
|
|
}
|
|
ip := net.ParseIP(local_ip)
|
|
if ip == nil {
|
|
log.Printf("Invalid IP in request: %s", local_ip)
|
|
return
|
|
}
|
|
weave_cidr := weave_ipstr + "/" + prefix
|
|
weave_ip, subnet, err := net.ParseCIDR(weave_cidr)
|
|
if err == nil {
|
|
log.Printf("Adding %s (%s) -> %s", name, local_ip, weave_cidr)
|
|
db.AddRecord(identifier, name, ip, weave_ip, subnet)
|
|
} else if err != nil {
|
|
log.Printf("Invalid CIDR in request: %s", weave_cidr)
|
|
}
|
|
}
|
|
})
|
|
address := fmt.Sprintf(":%d", port)
|
|
err := http.ListenAndServe(address, nil)
|
|
if err != nil {
|
|
log.Fatal("Unable to create http listener: ", err)
|
|
}
|
|
}
|