123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159 |
- // Copyright 2014 The Gogs Authors. All rights reserved.
- // Copyright 2019 The Gitea Authors. All rights reserved.
- // SPDX-License-Identifier: MIT
-
- package auth
-
- import (
- "net/http"
- "strings"
- "time"
-
- actions_model "code.gitea.io/gitea/models/actions"
- auth_model "code.gitea.io/gitea/models/auth"
- "code.gitea.io/gitea/models/db"
- user_model "code.gitea.io/gitea/models/user"
- "code.gitea.io/gitea/modules/log"
- "code.gitea.io/gitea/modules/timeutil"
- "code.gitea.io/gitea/modules/web/middleware"
- "code.gitea.io/gitea/services/auth/source/oauth2"
- )
-
- // Ensure the struct implements the interface.
- var (
- _ Method = &OAuth2{}
- _ Named = &OAuth2{}
- )
-
- // CheckOAuthAccessToken returns uid of user from oauth token
- func CheckOAuthAccessToken(accessToken string) int64 {
- // JWT tokens require a "."
- if !strings.Contains(accessToken, ".") {
- return 0
- }
- token, err := oauth2.ParseToken(accessToken, oauth2.DefaultSigningKey)
- if err != nil {
- log.Trace("oauth2.ParseToken: %v", err)
- return 0
- }
- var grant *auth_model.OAuth2Grant
- if grant, err = auth_model.GetOAuth2GrantByID(db.DefaultContext, token.GrantID); err != nil || grant == nil {
- return 0
- }
- if token.Type != oauth2.TypeAccessToken {
- return 0
- }
- if token.ExpiresAt.Before(time.Now()) || token.IssuedAt.After(time.Now()) {
- return 0
- }
- return grant.UserID
- }
-
- // OAuth2 implements the Auth interface and authenticates requests
- // (API requests only) by looking for an OAuth token in query parameters or the
- // "Authorization" header.
- type OAuth2 struct{}
-
- // Name represents the name of auth method
- func (o *OAuth2) Name() string {
- return "oauth2"
- }
-
- // userIDFromToken returns the user id corresponding to the OAuth token.
- // It will set 'IsApiToken' to true if the token is an API token and
- // set 'ApiTokenScope' to the scope of the access token
- func (o *OAuth2) userIDFromToken(req *http.Request, store DataStore) int64 {
- _ = req.ParseForm()
-
- // Check access token.
- tokenSHA := req.Form.Get("token")
- if len(tokenSHA) == 0 {
- tokenSHA = req.Form.Get("access_token")
- }
- if len(tokenSHA) == 0 {
- // Well, check with header again.
- auHead := req.Header.Get("Authorization")
- if len(auHead) > 0 {
- auths := strings.Fields(auHead)
- if len(auths) == 2 && (auths[0] == "token" || strings.ToLower(auths[0]) == "bearer") {
- tokenSHA = auths[1]
- }
- }
- }
- if len(tokenSHA) == 0 {
- return 0
- }
-
- // Let's see if token is valid.
- if strings.Contains(tokenSHA, ".") {
- uid := CheckOAuthAccessToken(tokenSHA)
- if uid != 0 {
- store.GetData()["IsApiToken"] = true
- store.GetData()["ApiTokenScope"] = auth_model.AccessTokenScopeAll // fallback to all
- }
- return uid
- }
- t, err := auth_model.GetAccessTokenBySHA(tokenSHA)
- if err != nil {
- if auth_model.IsErrAccessTokenNotExist(err) {
- // check task token
- task, err := actions_model.GetRunningTaskByToken(db.DefaultContext, tokenSHA)
- if err == nil && task != nil {
- log.Trace("Basic Authorization: Valid AccessToken for task[%d]", task.ID)
-
- store.GetData()["IsActionsToken"] = true
- store.GetData()["ActionsTaskID"] = task.ID
-
- return user_model.ActionsUserID
- }
- } else if !auth_model.IsErrAccessTokenNotExist(err) && !auth_model.IsErrAccessTokenEmpty(err) {
- log.Error("GetAccessTokenBySHA: %v", err)
- }
- return 0
- }
- t.UpdatedUnix = timeutil.TimeStampNow()
- if err = auth_model.UpdateAccessToken(t); err != nil {
- log.Error("UpdateAccessToken: %v", err)
- }
- store.GetData()["IsApiToken"] = true
- store.GetData()["ApiTokenScope"] = t.Scope
- return t.UID
- }
-
- // Verify extracts the user ID from the OAuth token in the query parameters
- // or the "Authorization" header and returns the corresponding user object for that ID.
- // If verification is successful returns an existing user object.
- // Returns nil if verification fails.
- func (o *OAuth2) Verify(req *http.Request, w http.ResponseWriter, store DataStore, sess SessionStore) (*user_model.User, error) {
- if !middleware.IsAPIPath(req) && !isAttachmentDownload(req) && !isAuthenticatedTokenRequest(req) {
- return nil, nil
- }
-
- id := o.userIDFromToken(req, store)
-
- if id <= 0 && id != -2 { // -2 means actions, so we need to allow it.
- return nil, nil
- }
- log.Trace("OAuth2 Authorization: Found token for user[%d]", id)
-
- user, err := user_model.GetPossibleUserByID(req.Context(), id)
- if err != nil {
- if !user_model.IsErrUserNotExist(err) {
- log.Error("GetUserByName: %v", err)
- }
- return nil, err
- }
-
- log.Trace("OAuth2 Authorization: Logged in user %-v", user)
- return user, nil
- }
-
- func isAuthenticatedTokenRequest(req *http.Request) bool {
- switch req.URL.Path {
- case "/login/oauth/userinfo":
- fallthrough
- case "/login/oauth/introspect":
- return true
- }
- return false
- }
|