github/steve
1
0
Fork 0
mirror of https://github.com/niusmallnan/steve.git synced 2025-07-01 17:01:54 +00:00
Code Issues Projects Releases Wiki Activity

Adding user authentication failed context value

Browse Source
This commit is contained in:
Michael Bolot 2022-08-04 20:39:40 -05:00
parent dbf9ef88ce
commit 7b5a48f871
1 changed files with 5 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
pkg/auth/filter.go
View File

@ -33,6 +33,8 @@ var ExistingContext = ToMiddleware(AuthenticatorFunc(func(req *http.Request) (us
return user, ok, nil return user, ok, nil
})) }))
const CattleAuthFailed = "X-API-Cattle-Auth-Failed"
type Authenticator interface { type Authenticator interface {
Authenticate(req *http.Request) (user.Info, bool, error) Authenticate(req *http.Request) (user.Info, bool, error)
} }
@ -144,6 +146,7 @@ func ToMiddleware(auth Authenticator) Middleware {
return func(next http.Handler) http.Handler { return func(next http.Handler) http.Handler {
return http.HandlerFunc(func(rw http.ResponseWriter, req *http.Request) { return http.HandlerFunc(func(rw http.ResponseWriter, req *http.Request) {
info, ok, err := auth.Authenticate(req) info, ok, err := auth.Authenticate(req)
ctx := req.Context()
if err != nil { if err != nil {
info = &user.DefaultInfo{ info = &user.DefaultInfo{
Name: "system:cattle:error", Name: "system:cattle:error",
@ -153,6 +156,7 @@ func ToMiddleware(auth Authenticator) Middleware {
"system:cattle:error", "system:cattle:error",
}, },
} }
ctx = request.WithValue(ctx, CattleAuthFailed, "true")
} else if !ok { } else if !ok {
info = &user.DefaultInfo{ info = &user.DefaultInfo{
Name: "system:unauthenticated", Name: "system:unauthenticated",
@ -162,8 +166,8 @@ func ToMiddleware(auth Authenticator) Middleware {
}, },
} }
} }
ctx = request.WithUser(ctx, info)
ctx := request.WithUser(req.Context(), info)
req = req.WithContext(ctx) req = req.WithContext(ctx)
next.ServeHTTP(rw, req) next.ServeHTTP(rw, req)
}) })