diff --git a/tsnet/example/tshello/tshello.go b/tsnet/example/tshello/tshello.go index 57b8fa2a4..b5ed4c711 100644 --- a/tsnet/example/tshello/tshello.go +++ b/tsnet/example/tshello/tshello.go @@ -29,13 +29,20 @@ func main() { if err != nil { log.Fatal(err) } + defer ln.Close() + + lc, err := s.LocalClient() + if err != nil { + log.Fatal(err) + } + if *addr == ":443" { ln = tls.NewListener(ln, &tls.Config{ GetCertificate: tailscale.GetCertificate, }) } log.Fatal(http.Serve(ln, http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) { - who, err := tailscale.WhoIs(r.Context(), r.RemoteAddr) + who, err := lc.WhoIs(r.Context(), r.RemoteAddr) if err != nil { http.Error(w, err.Error(), 500) return