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

9
tailscaleAuth.go
View file

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