ca20611d11
util/def: add def.Bool and def.Duration default parse helpers Replace multiple instances of def.Bool and def.Duration with a new util/def package. Updates #20018 Co-authored-by: Bobby <boby@codelabs.co.id> Co-authored-by: Simon Law <sfllaw@tailscale.com> Signed-off-by: Bobby <boby@codelabs.co.id> Signed-off-by: Simon Law <sfllaw@tailscale.com>
64 lines
1.7 KiB
Go
64 lines
1.7 KiB
Go
// Copyright (c) Tailscale Inc & contributors
|
||
// SPDX-License-Identifier: BSD-3-Clause
|
||
|
||
package routecheck
|
||
|
||
import (
|
||
"net/http"
|
||
"time"
|
||
|
||
jsonv2 "github.com/go-json-experiment/json"
|
||
jsonv1 "github.com/go-json-experiment/json/v1"
|
||
|
||
"tailscale.com/ipn/localapi"
|
||
"tailscale.com/net/routecheck"
|
||
"tailscale.com/util/def"
|
||
"tailscale.com/util/httpm"
|
||
)
|
||
|
||
func init() {
|
||
localapi.Register("routecheck", serveRouteCheck)
|
||
}
|
||
|
||
// ServeRouteCheck handles the API endpoint that serves the routecheck Report.
|
||
// If the probe form field is true, then this handler will refresh the Report
|
||
// before serving it.
|
||
// If the timeout form field is a valid duration, the probe will consider a node
|
||
// to be unreachable if it doesn’t respond before the timeout expires.
|
||
func serveRouteCheck(h *localapi.Handler, w http.ResponseWriter, r *http.Request) {
|
||
rc := ClientFor(h.LocalBackend())
|
||
if rc == nil {
|
||
http.Error(w, "routecheck is not enabled", http.StatusServiceUnavailable)
|
||
return
|
||
}
|
||
|
||
if r.Method != httpm.POST {
|
||
http.Error(w, "want POST", http.StatusMethodNotAllowed)
|
||
return
|
||
}
|
||
|
||
var err error
|
||
var report *routecheck.Report
|
||
if def.Bool(r.FormValue("probe"), false) {
|
||
timeout := def.Duration(r.FormValue("timeout"), routecheck.DefaultTimeout)
|
||
timeout = min(max(0, timeout), 60*time.Second) // clamp to [0s, 60s]
|
||
report, err = rc.Refresh(r.Context(), timeout)
|
||
} else {
|
||
report = rc.Report()
|
||
}
|
||
if err != nil {
|
||
localapi.WriteErrorJSON(w, err)
|
||
return
|
||
}
|
||
|
||
w.Header().Set("Content-Type", "application/json")
|
||
if report == nil {
|
||
w.WriteHeader(http.StatusNoContent)
|
||
return
|
||
}
|
||
|
||
// TODO(sfllaw): Since ipn/localapi is still using encoding/json
|
||
// with its default options, marshal with DefaultOptionsV1.
|
||
jsonv2.MarshalWrite(w, report, jsonv1.DefaultOptionsV1())
|
||
}
|