2018-06-03 20:04:11 +02:00
|
|
|
// Binary diagd provides automated network diagnostics.
|
|
|
|
|
package main
|
|
|
|
|
|
|
|
|
|
import (
|
2018-06-26 21:52:29 +02:00
|
|
|
"encoding/json"
|
2018-06-03 20:04:11 +02:00
|
|
|
"flag"
|
|
|
|
|
"fmt"
|
|
|
|
|
"html"
|
|
|
|
|
"io"
|
|
|
|
|
"log"
|
2018-06-26 09:47:55 +02:00
|
|
|
"net"
|
2018-06-03 20:04:11 +02:00
|
|
|
"net/http"
|
2018-06-26 09:47:55 +02:00
|
|
|
"os"
|
|
|
|
|
"os/signal"
|
2018-06-03 20:04:11 +02:00
|
|
|
"sync"
|
2018-06-26 09:47:55 +02:00
|
|
|
"syscall"
|
|
|
|
|
|
|
|
|
|
"github.com/gokrazy/gokrazy"
|
2018-06-03 20:04:11 +02:00
|
|
|
|
|
|
|
|
"router7/internal/diag"
|
2018-06-26 09:47:55 +02:00
|
|
|
"router7/internal/multilisten"
|
2018-06-03 20:04:11 +02:00
|
|
|
)
|
|
|
|
|
|
2018-06-26 09:47:55 +02:00
|
|
|
var httpListeners = multilisten.NewPool()
|
|
|
|
|
|
|
|
|
|
func updateListeners() error {
|
|
|
|
|
hosts, err := gokrazy.PrivateInterfaceAddrs()
|
|
|
|
|
if err != nil {
|
|
|
|
|
return err
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
httpListeners.ListenAndServe(hosts, func(host string) multilisten.Listener {
|
|
|
|
|
return &http.Server{Addr: net.JoinHostPort(host, "7733")}
|
|
|
|
|
})
|
|
|
|
|
return nil
|
|
|
|
|
}
|
|
|
|
|
|
2018-06-03 20:04:11 +02:00
|
|
|
func dump(w io.Writer, re *diag.EvalResult) {
|
|
|
|
|
symbol := "✔"
|
|
|
|
|
if re.Error {
|
|
|
|
|
symbol = "✘"
|
|
|
|
|
}
|
|
|
|
|
fmt.Fprintf(w, "<li>%s %s: %s<ul>", symbol, html.EscapeString(re.Name), html.EscapeString(re.Status))
|
|
|
|
|
for _, ch := range re.Children {
|
|
|
|
|
dump(w, ch)
|
|
|
|
|
}
|
|
|
|
|
fmt.Fprintf(w, "</ul></li>")
|
|
|
|
|
}
|
|
|
|
|
|
2018-06-26 21:52:29 +02:00
|
|
|
func firstError(re *diag.EvalResult) string {
|
|
|
|
|
if re.Error {
|
|
|
|
|
return fmt.Sprintf("%s: %s", re.Name, re.Status)
|
|
|
|
|
}
|
|
|
|
|
for _, ch := range re.Children {
|
|
|
|
|
if msg := firstError(ch); msg != "" {
|
|
|
|
|
return msg
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
return ""
|
|
|
|
|
}
|
|
|
|
|
|
2018-06-03 20:04:11 +02:00
|
|
|
func logic() error {
|
|
|
|
|
const (
|
|
|
|
|
uplink = "uplink0" /* enp0s31f6 */
|
|
|
|
|
ip6allrouters = "ff02::2" // no /etc/hosts on gokrazy
|
|
|
|
|
)
|
|
|
|
|
m := diag.NewMonitor(diag.Link(uplink).
|
|
|
|
|
Then(diag.DHCPv4().
|
|
|
|
|
Then(diag.Ping4Gateway().
|
|
|
|
|
Then(diag.Ping4("google.ch").
|
|
|
|
|
Then(diag.TCP4("www.google.ch:80"))))).
|
|
|
|
|
Then(diag.DHCPv6().
|
|
|
|
|
Then(diag.Ping6("lan0", "google.ch"))).
|
|
|
|
|
Then(diag.RouterAdvertisments(uplink).
|
|
|
|
|
Then(diag.Ping6Gateway().
|
|
|
|
|
Then(diag.Ping6(uplink, "google.ch").
|
|
|
|
|
Then(diag.TCP6("www.google.ch:80"))))).
|
|
|
|
|
Then(diag.Ping6("", ip6allrouters+"%"+uplink)))
|
|
|
|
|
var mu sync.Mutex
|
|
|
|
|
http.HandleFunc("/", func(w http.ResponseWriter, r *http.Request) {
|
|
|
|
|
mu.Lock()
|
|
|
|
|
re := m.Evaluate()
|
|
|
|
|
mu.Unlock()
|
|
|
|
|
fmt.Fprintf(w, `<!DOCTYPE html><style type="text/css">ul { list-style-type: none; }</style><ul>`)
|
|
|
|
|
dump(w, re)
|
|
|
|
|
})
|
2018-06-26 21:52:29 +02:00
|
|
|
http.HandleFunc("/health.json", func(w http.ResponseWriter, r *http.Request) {
|
|
|
|
|
mu.Lock()
|
|
|
|
|
re := m.Evaluate()
|
|
|
|
|
mu.Unlock()
|
|
|
|
|
reply := struct {
|
|
|
|
|
FirstError string `json:"first_error"`
|
|
|
|
|
}{
|
|
|
|
|
FirstError: firstError(re),
|
|
|
|
|
}
|
|
|
|
|
b, err := json.Marshal(&reply)
|
|
|
|
|
if err != nil {
|
|
|
|
|
http.Error(w, err.Error(), http.StatusInternalServerError)
|
|
|
|
|
return
|
|
|
|
|
}
|
|
|
|
|
w.Write(b)
|
|
|
|
|
})
|
2018-06-26 09:47:55 +02:00
|
|
|
updateListeners()
|
|
|
|
|
ch := make(chan os.Signal, 1)
|
|
|
|
|
signal.Notify(ch, syscall.SIGUSR1)
|
|
|
|
|
for range ch {
|
|
|
|
|
if err := updateListeners(); err != nil {
|
|
|
|
|
log.Printf("updateListeners: %v", err)
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
return nil
|
2018-06-03 20:04:11 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
|
|
func main() {
|
|
|
|
|
flag.Parse()
|
|
|
|
|
|
|
|
|
|
if err := logic(); err != nil {
|
|
|
|
|
log.Fatal(err)
|
|
|
|
|
}
|
|
|
|
|
}
|
