161 lines
3.7 KiB
Go
161 lines
3.7 KiB
Go
// Copyright 2018 Google Inc.
|
|
//
|
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
|
// you may not use this file except in compliance with the License.
|
|
// You may obtain a copy of the License at
|
|
//
|
|
// http://www.apache.org/licenses/LICENSE-2.0
|
|
//
|
|
// Unless required by applicable law or agreed to in writing, software
|
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
// See the License for the specific language governing permissions and
|
|
// limitations under the License.
|
|
|
|
// Binary diagd provides automated network diagnostics.
|
|
package main
|
|
|
|
import (
|
|
"encoding/json"
|
|
"flag"
|
|
"fmt"
|
|
"html"
|
|
"io"
|
|
"log"
|
|
"net"
|
|
"net/http"
|
|
"os"
|
|
"os/signal"
|
|
"strings"
|
|
"sync"
|
|
"syscall"
|
|
|
|
"github.com/gokrazy/gokrazy"
|
|
|
|
"github.com/rtr7/router7/internal/diag"
|
|
"github.com/rtr7/router7/internal/multilisten"
|
|
|
|
_ "net/http/pprof"
|
|
)
|
|
|
|
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
|
|
}
|
|
|
|
func dump(indent int, w io.Writer, re *diag.EvalResult) {
|
|
symbol := "✔"
|
|
if re.Error {
|
|
symbol = "✘"
|
|
}
|
|
fmt.Fprintf(w, "<li>\n%s%s %s: %s<ul>",
|
|
strings.Repeat(" ", indent),
|
|
symbol,
|
|
html.EscapeString(re.Name),
|
|
html.EscapeString(re.Status))
|
|
for _, ch := range re.Children {
|
|
dump(indent+1, w, ch)
|
|
}
|
|
fmt.Fprintf(w, "</ul></li>")
|
|
}
|
|
|
|
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 ""
|
|
}
|
|
|
|
func graph(uplink string, ipv6 bool) *diag.Monitor {
|
|
const ip6allrouters = "ff02::2" // no /etc/hosts on gokrazy
|
|
graph := diag.Link(uplink).
|
|
Then(diag.DHCPv4().
|
|
Then(diag.Ping4Gateway().
|
|
Then(diag.TCP4("www.google.com:80"))))
|
|
|
|
if ipv6 {
|
|
graph = graph.
|
|
Then(diag.DHCPv6().
|
|
Then(diag.TCP6("lan0", "www.google.com:80"))).
|
|
Then(diag.RouterAdvertisments(uplink).
|
|
Then(diag.Ping6Gateway().
|
|
Then(diag.TCP6(uplink, "www.google.com:80")))).
|
|
Then(diag.Ping6("", ip6allrouters+"%"+uplink))
|
|
}
|
|
return diag.NewMonitor(graph)
|
|
}
|
|
|
|
func logic() error {
|
|
var (
|
|
ifname = flag.String("interface",
|
|
"uplink0",
|
|
"interface name to query")
|
|
ipv6 = flag.Bool("ipv6",
|
|
true,
|
|
"whether to expect IPv6 connectivity in health.json")
|
|
perm = flag.String("perm",
|
|
"/perm",
|
|
"path to replace /perm")
|
|
)
|
|
flag.Parse()
|
|
uplink := *ifname
|
|
diag.Perm = *perm
|
|
mHumanReadable := graph(uplink, true) // for display only
|
|
mJSON := graph(uplink, *ipv6) // for updates
|
|
var mu sync.Mutex
|
|
http.HandleFunc("/", func(w http.ResponseWriter, r *http.Request) {
|
|
mu.Lock()
|
|
re := mHumanReadable.Evaluate()
|
|
mu.Unlock()
|
|
fmt.Fprintf(w, `<!DOCTYPE html><style type="text/css">ul { list-style-type: none; }</style><ul>`)
|
|
dump(0, w, re)
|
|
})
|
|
http.HandleFunc("/health.json", func(w http.ResponseWriter, r *http.Request) {
|
|
mu.Lock()
|
|
re := mJSON.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)
|
|
})
|
|
if err := updateListeners(); err != nil {
|
|
return err
|
|
}
|
|
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
|
|
}
|
|
|
|
func main() {
|
|
if err := logic(); err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
}
|