pkg/caddy-tailscale
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Add timeout to Tailscale WhoIs call

Browse source
This commit is contained in:
akp 2024-10-10 13:15:14 +01:00
parent 7a1a654b73
commit 972ff18012
No known key found for this signature in database
GPG key ID: CF8D58F3DEB20755
1 changed files with 6 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

7
tailscaleAuth.go
View file

@ -7,6 +7,7 @@ import (
"net/http" "net/http"
"strconv" "strconv"
"tailscale.com/client/tailscale" "tailscale.com/client/tailscale"
"time"
) )
func init() { func init() {
@ -30,8 +31,10 @@ func (ta *TailscaleAuth) Provision(caddy.Context) error {
return nil return nil
} }
func (ta *TailscaleAuth) Authenticate(wr http.ResponseWriter, req *http.Request) (caddyauth.User, bool, error) { func (ta *TailscaleAuth) Authenticate(_ http.ResponseWriter, req *http.Request) (caddyauth.User, bool, error) {
whois, err := ta.lc.WhoIs(context.Background(), req.RemoteAddr) ctx, cancel := context.WithTimeout(context.Background(), time.Second*30)
defer cancel()
whois, err := ta.lc.WhoIs(ctx, req.RemoteAddr)
if err != nil { if err != nil {
return caddyauth.User{}, false, err return caddyauth.User{}, false, err
} }