astra/ntfy
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

add skeleton code to server.go

Browse source
This commit is contained in:
ngerstle 2023-02-05 00:11:13 +01:00
parent a69f9c5bf2
commit 524f9c3082
1 changed files with 14 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

12
server/server.go
View file

@ -1411,9 +1411,14 @@ func (s *Server) withAuth(next handleFunc, perm auth.Permission) handleFunc {
return err
}
var user *auth.User // may stay nil if no auth header!
if len(s.config.UserHeader) > 0 { //Lookup user from header if user-header is configured
username := extractUserHeader(r, s.config.UserHeader)
if user, err = s.auth.PreAuthenticatedUser(username); err != nil {
log.Info("unable to associate %s to user account: %s", username, err.Error()) //unknown users are assumed anonymous, not unauthorized/401
}
} else { //Fall back to native Basic Auth if no user-header is configured
username, password, ok := extractUserPass(r)
if ok {
if user, err = s.auth.Authenticate(username, password); err != nil {
log.Info("authentication failed: %s", err.Error())
return errHTTPUnauthorized
}
@ -1449,6 +1454,11 @@ func extractUserPass(r *http.Request) (username string, password string, ok bool
return
}
// extractUserHader pulls the username of an already authenticated user from the configured header
func extractUserHeader(r *http.Request, h string) (username string) {
return readParam(r, h)
}
// visitor creates or retrieves a rate.Limiter for the given visitor.
// This function was taken from https://www.alexedwards.net/blog/how-to-rate-limit-http-requests (MIT).
func (s *Server) visitor(r *http.Request) *visitor {