- Switch to use `CryptoRandomBytes` instead of `CryptoRandomString`, OAuth's secrets are copied pasted and don't need to avoid dubious characters etc. - `CryptoRandomBytes` gives  `CryptoRandomString` gives  possible states. - Add a prefix, such that code scanners can easily grep these in source code. - 32 Bytes + prefix
		
			
				
	
	
		
			579 lines
		
	
	
		
			18 KiB
		
	
	
	
		
			Go
		
	
	
	
	
	
			
		
		
	
	
			579 lines
		
	
	
		
			18 KiB
		
	
	
	
		
			Go
		
	
	
	
	
	
// Copyright 2019 The Gitea Authors. All rights reserved.
 | 
						|
// Use of this source code is governed by a MIT-style
 | 
						|
// license that can be found in the LICENSE file.
 | 
						|
 | 
						|
package auth
 | 
						|
 | 
						|
import (
 | 
						|
	"crypto/sha256"
 | 
						|
	"encoding/base32"
 | 
						|
	"encoding/base64"
 | 
						|
	"fmt"
 | 
						|
	"net/url"
 | 
						|
	"strings"
 | 
						|
 | 
						|
	"code.gitea.io/gitea/models/db"
 | 
						|
	"code.gitea.io/gitea/modules/timeutil"
 | 
						|
	"code.gitea.io/gitea/modules/util"
 | 
						|
 | 
						|
	uuid "github.com/google/uuid"
 | 
						|
	"golang.org/x/crypto/bcrypt"
 | 
						|
	"xorm.io/xorm"
 | 
						|
)
 | 
						|
 | 
						|
// OAuth2Application represents an OAuth2 client (RFC 6749)
 | 
						|
type OAuth2Application struct {
 | 
						|
	ID           int64 `xorm:"pk autoincr"`
 | 
						|
	UID          int64 `xorm:"INDEX"`
 | 
						|
	Name         string
 | 
						|
	ClientID     string `xorm:"unique"`
 | 
						|
	ClientSecret string
 | 
						|
	RedirectURIs []string           `xorm:"redirect_uris JSON TEXT"`
 | 
						|
	CreatedUnix  timeutil.TimeStamp `xorm:"INDEX created"`
 | 
						|
	UpdatedUnix  timeutil.TimeStamp `xorm:"INDEX updated"`
 | 
						|
}
 | 
						|
 | 
						|
func init() {
 | 
						|
	db.RegisterModel(new(OAuth2Application))
 | 
						|
	db.RegisterModel(new(OAuth2AuthorizationCode))
 | 
						|
	db.RegisterModel(new(OAuth2Grant))
 | 
						|
}
 | 
						|
 | 
						|
// TableName sets the table name to `oauth2_application`
 | 
						|
func (app *OAuth2Application) TableName() string {
 | 
						|
	return "oauth2_application"
 | 
						|
}
 | 
						|
 | 
						|
// PrimaryRedirectURI returns the first redirect uri or an empty string if empty
 | 
						|
func (app *OAuth2Application) PrimaryRedirectURI() string {
 | 
						|
	if len(app.RedirectURIs) == 0 {
 | 
						|
		return ""
 | 
						|
	}
 | 
						|
	return app.RedirectURIs[0]
 | 
						|
}
 | 
						|
 | 
						|
// ContainsRedirectURI checks if redirectURI is allowed for app
 | 
						|
func (app *OAuth2Application) ContainsRedirectURI(redirectURI string) bool {
 | 
						|
	return util.IsStringInSlice(redirectURI, app.RedirectURIs, true)
 | 
						|
}
 | 
						|
 | 
						|
// Base32 characters, but lowercased.
 | 
						|
const lowerBase32Chars = "abcdefghijklmnopqrstuvwxyz234567"
 | 
						|
 | 
						|
// base32 encoder that uses lowered characters without padding.
 | 
						|
var base32Lower = base32.NewEncoding(lowerBase32Chars).WithPadding(base32.NoPadding)
 | 
						|
 | 
						|
// GenerateClientSecret will generate the client secret and returns the plaintext and saves the hash at the database
 | 
						|
func (app *OAuth2Application) GenerateClientSecret() (string, error) {
 | 
						|
	rBytes, err := util.CryptoRandomBytes(32)
 | 
						|
	if err != nil {
 | 
						|
		return "", err
 | 
						|
	}
 | 
						|
	// Add a prefix to the base32, this is in order to make it easier
 | 
						|
	// for code scanners to grab sensitive tokens.
 | 
						|
	clientSecret := "gto_" + base32Lower.EncodeToString(rBytes)
 | 
						|
 | 
						|
	hashedSecret, err := bcrypt.GenerateFromPassword([]byte(clientSecret), bcrypt.DefaultCost)
 | 
						|
	if err != nil {
 | 
						|
		return "", err
 | 
						|
	}
 | 
						|
	app.ClientSecret = string(hashedSecret)
 | 
						|
	if _, err := db.GetEngine(db.DefaultContext).ID(app.ID).Cols("client_secret").Update(app); err != nil {
 | 
						|
		return "", err
 | 
						|
	}
 | 
						|
	return clientSecret, nil
 | 
						|
}
 | 
						|
 | 
						|
// ValidateClientSecret validates the given secret by the hash saved in database
 | 
						|
func (app *OAuth2Application) ValidateClientSecret(secret []byte) bool {
 | 
						|
	return bcrypt.CompareHashAndPassword([]byte(app.ClientSecret), secret) == nil
 | 
						|
}
 | 
						|
 | 
						|
// GetGrantByUserID returns a OAuth2Grant by its user and application ID
 | 
						|
func (app *OAuth2Application) GetGrantByUserID(userID int64) (*OAuth2Grant, error) {
 | 
						|
	return app.getGrantByUserID(db.GetEngine(db.DefaultContext), userID)
 | 
						|
}
 | 
						|
 | 
						|
func (app *OAuth2Application) getGrantByUserID(e db.Engine, userID int64) (grant *OAuth2Grant, err error) {
 | 
						|
	grant = new(OAuth2Grant)
 | 
						|
	if has, err := e.Where("user_id = ? AND application_id = ?", userID, app.ID).Get(grant); err != nil {
 | 
						|
		return nil, err
 | 
						|
	} else if !has {
 | 
						|
		return nil, nil
 | 
						|
	}
 | 
						|
	return grant, nil
 | 
						|
}
 | 
						|
 | 
						|
// CreateGrant generates a grant for an user
 | 
						|
func (app *OAuth2Application) CreateGrant(userID int64, scope string) (*OAuth2Grant, error) {
 | 
						|
	return app.createGrant(db.GetEngine(db.DefaultContext), userID, scope)
 | 
						|
}
 | 
						|
 | 
						|
func (app *OAuth2Application) createGrant(e db.Engine, userID int64, scope string) (*OAuth2Grant, error) {
 | 
						|
	grant := &OAuth2Grant{
 | 
						|
		ApplicationID: app.ID,
 | 
						|
		UserID:        userID,
 | 
						|
		Scope:         scope,
 | 
						|
	}
 | 
						|
	_, err := e.Insert(grant)
 | 
						|
	if err != nil {
 | 
						|
		return nil, err
 | 
						|
	}
 | 
						|
	return grant, nil
 | 
						|
}
 | 
						|
 | 
						|
// GetOAuth2ApplicationByClientID returns the oauth2 application with the given client_id. Returns an error if not found.
 | 
						|
func GetOAuth2ApplicationByClientID(clientID string) (app *OAuth2Application, err error) {
 | 
						|
	return getOAuth2ApplicationByClientID(db.GetEngine(db.DefaultContext), clientID)
 | 
						|
}
 | 
						|
 | 
						|
func getOAuth2ApplicationByClientID(e db.Engine, clientID string) (app *OAuth2Application, err error) {
 | 
						|
	app = new(OAuth2Application)
 | 
						|
	has, err := e.Where("client_id = ?", clientID).Get(app)
 | 
						|
	if !has {
 | 
						|
		return nil, ErrOAuthClientIDInvalid{ClientID: clientID}
 | 
						|
	}
 | 
						|
	return
 | 
						|
}
 | 
						|
 | 
						|
// GetOAuth2ApplicationByID returns the oauth2 application with the given id. Returns an error if not found.
 | 
						|
func GetOAuth2ApplicationByID(id int64) (app *OAuth2Application, err error) {
 | 
						|
	return getOAuth2ApplicationByID(db.GetEngine(db.DefaultContext), id)
 | 
						|
}
 | 
						|
 | 
						|
func getOAuth2ApplicationByID(e db.Engine, id int64) (app *OAuth2Application, err error) {
 | 
						|
	app = new(OAuth2Application)
 | 
						|
	has, err := e.ID(id).Get(app)
 | 
						|
	if err != nil {
 | 
						|
		return nil, err
 | 
						|
	}
 | 
						|
	if !has {
 | 
						|
		return nil, ErrOAuthApplicationNotFound{ID: id}
 | 
						|
	}
 | 
						|
	return app, nil
 | 
						|
}
 | 
						|
 | 
						|
// GetOAuth2ApplicationsByUserID returns all oauth2 applications owned by the user
 | 
						|
func GetOAuth2ApplicationsByUserID(userID int64) (apps []*OAuth2Application, err error) {
 | 
						|
	return getOAuth2ApplicationsByUserID(db.GetEngine(db.DefaultContext), userID)
 | 
						|
}
 | 
						|
 | 
						|
func getOAuth2ApplicationsByUserID(e db.Engine, userID int64) (apps []*OAuth2Application, err error) {
 | 
						|
	apps = make([]*OAuth2Application, 0)
 | 
						|
	err = e.Where("uid = ?", userID).Find(&apps)
 | 
						|
	return
 | 
						|
}
 | 
						|
 | 
						|
// CreateOAuth2ApplicationOptions holds options to create an oauth2 application
 | 
						|
type CreateOAuth2ApplicationOptions struct {
 | 
						|
	Name         string
 | 
						|
	UserID       int64
 | 
						|
	RedirectURIs []string
 | 
						|
}
 | 
						|
 | 
						|
// CreateOAuth2Application inserts a new oauth2 application
 | 
						|
func CreateOAuth2Application(opts CreateOAuth2ApplicationOptions) (*OAuth2Application, error) {
 | 
						|
	return createOAuth2Application(db.GetEngine(db.DefaultContext), opts)
 | 
						|
}
 | 
						|
 | 
						|
func createOAuth2Application(e db.Engine, opts CreateOAuth2ApplicationOptions) (*OAuth2Application, error) {
 | 
						|
	clientID := uuid.New().String()
 | 
						|
	app := &OAuth2Application{
 | 
						|
		UID:          opts.UserID,
 | 
						|
		Name:         opts.Name,
 | 
						|
		ClientID:     clientID,
 | 
						|
		RedirectURIs: opts.RedirectURIs,
 | 
						|
	}
 | 
						|
	if _, err := e.Insert(app); err != nil {
 | 
						|
		return nil, err
 | 
						|
	}
 | 
						|
	return app, nil
 | 
						|
}
 | 
						|
 | 
						|
// UpdateOAuth2ApplicationOptions holds options to update an oauth2 application
 | 
						|
type UpdateOAuth2ApplicationOptions struct {
 | 
						|
	ID           int64
 | 
						|
	Name         string
 | 
						|
	UserID       int64
 | 
						|
	RedirectURIs []string
 | 
						|
}
 | 
						|
 | 
						|
// UpdateOAuth2Application updates an oauth2 application
 | 
						|
func UpdateOAuth2Application(opts UpdateOAuth2ApplicationOptions) (*OAuth2Application, error) {
 | 
						|
	ctx, committer, err := db.TxContext()
 | 
						|
	if err != nil {
 | 
						|
		return nil, err
 | 
						|
	}
 | 
						|
	defer committer.Close()
 | 
						|
	sess := db.GetEngine(ctx)
 | 
						|
 | 
						|
	app, err := getOAuth2ApplicationByID(sess, opts.ID)
 | 
						|
	if err != nil {
 | 
						|
		return nil, err
 | 
						|
	}
 | 
						|
	if app.UID != opts.UserID {
 | 
						|
		return nil, fmt.Errorf("UID mismatch")
 | 
						|
	}
 | 
						|
 | 
						|
	app.Name = opts.Name
 | 
						|
	app.RedirectURIs = opts.RedirectURIs
 | 
						|
 | 
						|
	if err = updateOAuth2Application(sess, app); err != nil {
 | 
						|
		return nil, err
 | 
						|
	}
 | 
						|
	app.ClientSecret = ""
 | 
						|
 | 
						|
	return app, committer.Commit()
 | 
						|
}
 | 
						|
 | 
						|
func updateOAuth2Application(e db.Engine, app *OAuth2Application) error {
 | 
						|
	if _, err := e.ID(app.ID).Update(app); err != nil {
 | 
						|
		return err
 | 
						|
	}
 | 
						|
	return nil
 | 
						|
}
 | 
						|
 | 
						|
func deleteOAuth2Application(sess db.Engine, id, userid int64) error {
 | 
						|
	if deleted, err := sess.Delete(&OAuth2Application{ID: id, UID: userid}); err != nil {
 | 
						|
		return err
 | 
						|
	} else if deleted == 0 {
 | 
						|
		return ErrOAuthApplicationNotFound{ID: id}
 | 
						|
	}
 | 
						|
	codes := make([]*OAuth2AuthorizationCode, 0)
 | 
						|
	// delete correlating auth codes
 | 
						|
	if err := sess.Join("INNER", "oauth2_grant",
 | 
						|
		"oauth2_authorization_code.grant_id = oauth2_grant.id AND oauth2_grant.application_id = ?", id).Find(&codes); err != nil {
 | 
						|
		return err
 | 
						|
	}
 | 
						|
	codeIDs := make([]int64, 0)
 | 
						|
	for _, grant := range codes {
 | 
						|
		codeIDs = append(codeIDs, grant.ID)
 | 
						|
	}
 | 
						|
 | 
						|
	if _, err := sess.In("id", codeIDs).Delete(new(OAuth2AuthorizationCode)); err != nil {
 | 
						|
		return err
 | 
						|
	}
 | 
						|
 | 
						|
	if _, err := sess.Where("application_id = ?", id).Delete(new(OAuth2Grant)); err != nil {
 | 
						|
		return err
 | 
						|
	}
 | 
						|
	return nil
 | 
						|
}
 | 
						|
 | 
						|
// DeleteOAuth2Application deletes the application with the given id and the grants and auth codes related to it. It checks if the userid was the creator of the app.
 | 
						|
func DeleteOAuth2Application(id, userid int64) error {
 | 
						|
	ctx, committer, err := db.TxContext()
 | 
						|
	if err != nil {
 | 
						|
		return err
 | 
						|
	}
 | 
						|
	defer committer.Close()
 | 
						|
	if err := deleteOAuth2Application(db.GetEngine(ctx), id, userid); err != nil {
 | 
						|
		return err
 | 
						|
	}
 | 
						|
	return committer.Commit()
 | 
						|
}
 | 
						|
 | 
						|
// ListOAuth2Applications returns a list of oauth2 applications belongs to given user.
 | 
						|
func ListOAuth2Applications(uid int64, listOptions db.ListOptions) ([]*OAuth2Application, int64, error) {
 | 
						|
	sess := db.GetEngine(db.DefaultContext).
 | 
						|
		Where("uid=?", uid).
 | 
						|
		Desc("id")
 | 
						|
 | 
						|
	if listOptions.Page != 0 {
 | 
						|
		sess = db.SetSessionPagination(sess, &listOptions)
 | 
						|
 | 
						|
		apps := make([]*OAuth2Application, 0, listOptions.PageSize)
 | 
						|
		total, err := sess.FindAndCount(&apps)
 | 
						|
		return apps, total, err
 | 
						|
	}
 | 
						|
 | 
						|
	apps := make([]*OAuth2Application, 0, 5)
 | 
						|
	total, err := sess.FindAndCount(&apps)
 | 
						|
	return apps, total, err
 | 
						|
}
 | 
						|
 | 
						|
//////////////////////////////////////////////////////
 | 
						|
 | 
						|
// OAuth2AuthorizationCode is a code to obtain an access token in combination with the client secret once. It has a limited lifetime.
 | 
						|
type OAuth2AuthorizationCode struct {
 | 
						|
	ID                  int64        `xorm:"pk autoincr"`
 | 
						|
	Grant               *OAuth2Grant `xorm:"-"`
 | 
						|
	GrantID             int64
 | 
						|
	Code                string `xorm:"INDEX unique"`
 | 
						|
	CodeChallenge       string
 | 
						|
	CodeChallengeMethod string
 | 
						|
	RedirectURI         string
 | 
						|
	ValidUntil          timeutil.TimeStamp `xorm:"index"`
 | 
						|
}
 | 
						|
 | 
						|
// TableName sets the table name to `oauth2_authorization_code`
 | 
						|
func (code *OAuth2AuthorizationCode) TableName() string {
 | 
						|
	return "oauth2_authorization_code"
 | 
						|
}
 | 
						|
 | 
						|
// GenerateRedirectURI generates a redirect URI for a successful authorization request. State will be used if not empty.
 | 
						|
func (code *OAuth2AuthorizationCode) GenerateRedirectURI(state string) (redirect *url.URL, err error) {
 | 
						|
	if redirect, err = url.Parse(code.RedirectURI); err != nil {
 | 
						|
		return
 | 
						|
	}
 | 
						|
	q := redirect.Query()
 | 
						|
	if state != "" {
 | 
						|
		q.Set("state", state)
 | 
						|
	}
 | 
						|
	q.Set("code", code.Code)
 | 
						|
	redirect.RawQuery = q.Encode()
 | 
						|
	return
 | 
						|
}
 | 
						|
 | 
						|
// Invalidate deletes the auth code from the database to invalidate this code
 | 
						|
func (code *OAuth2AuthorizationCode) Invalidate() error {
 | 
						|
	return code.invalidate(db.GetEngine(db.DefaultContext))
 | 
						|
}
 | 
						|
 | 
						|
func (code *OAuth2AuthorizationCode) invalidate(e db.Engine) error {
 | 
						|
	_, err := e.Delete(code)
 | 
						|
	return err
 | 
						|
}
 | 
						|
 | 
						|
// ValidateCodeChallenge validates the given verifier against the saved code challenge. This is part of the PKCE implementation.
 | 
						|
func (code *OAuth2AuthorizationCode) ValidateCodeChallenge(verifier string) bool {
 | 
						|
	return code.validateCodeChallenge(verifier)
 | 
						|
}
 | 
						|
 | 
						|
func (code *OAuth2AuthorizationCode) validateCodeChallenge(verifier string) bool {
 | 
						|
	switch code.CodeChallengeMethod {
 | 
						|
	case "S256":
 | 
						|
		// base64url(SHA256(verifier)) see https://tools.ietf.org/html/rfc7636#section-4.6
 | 
						|
		h := sha256.Sum256([]byte(verifier))
 | 
						|
		hashedVerifier := base64.RawURLEncoding.EncodeToString(h[:])
 | 
						|
		return hashedVerifier == code.CodeChallenge
 | 
						|
	case "plain":
 | 
						|
		return verifier == code.CodeChallenge
 | 
						|
	case "":
 | 
						|
		return true
 | 
						|
	default:
 | 
						|
		// unsupported method -> return false
 | 
						|
		return false
 | 
						|
	}
 | 
						|
}
 | 
						|
 | 
						|
// GetOAuth2AuthorizationByCode returns an authorization by its code
 | 
						|
func GetOAuth2AuthorizationByCode(code string) (*OAuth2AuthorizationCode, error) {
 | 
						|
	return getOAuth2AuthorizationByCode(db.GetEngine(db.DefaultContext), code)
 | 
						|
}
 | 
						|
 | 
						|
func getOAuth2AuthorizationByCode(e db.Engine, code string) (auth *OAuth2AuthorizationCode, err error) {
 | 
						|
	auth = new(OAuth2AuthorizationCode)
 | 
						|
	if has, err := e.Where("code = ?", code).Get(auth); err != nil {
 | 
						|
		return nil, err
 | 
						|
	} else if !has {
 | 
						|
		return nil, nil
 | 
						|
	}
 | 
						|
	auth.Grant = new(OAuth2Grant)
 | 
						|
	if has, err := e.ID(auth.GrantID).Get(auth.Grant); err != nil {
 | 
						|
		return nil, err
 | 
						|
	} else if !has {
 | 
						|
		return nil, nil
 | 
						|
	}
 | 
						|
	return auth, nil
 | 
						|
}
 | 
						|
 | 
						|
//////////////////////////////////////////////////////
 | 
						|
 | 
						|
// OAuth2Grant represents the permission of an user for a specific application to access resources
 | 
						|
type OAuth2Grant struct {
 | 
						|
	ID            int64              `xorm:"pk autoincr"`
 | 
						|
	UserID        int64              `xorm:"INDEX unique(user_application)"`
 | 
						|
	Application   *OAuth2Application `xorm:"-"`
 | 
						|
	ApplicationID int64              `xorm:"INDEX unique(user_application)"`
 | 
						|
	Counter       int64              `xorm:"NOT NULL DEFAULT 1"`
 | 
						|
	Scope         string             `xorm:"TEXT"`
 | 
						|
	Nonce         string             `xorm:"TEXT"`
 | 
						|
	CreatedUnix   timeutil.TimeStamp `xorm:"created"`
 | 
						|
	UpdatedUnix   timeutil.TimeStamp `xorm:"updated"`
 | 
						|
}
 | 
						|
 | 
						|
// TableName sets the table name to `oauth2_grant`
 | 
						|
func (grant *OAuth2Grant) TableName() string {
 | 
						|
	return "oauth2_grant"
 | 
						|
}
 | 
						|
 | 
						|
// GenerateNewAuthorizationCode generates a new authorization code for a grant and saves it to the database
 | 
						|
func (grant *OAuth2Grant) GenerateNewAuthorizationCode(redirectURI, codeChallenge, codeChallengeMethod string) (*OAuth2AuthorizationCode, error) {
 | 
						|
	return grant.generateNewAuthorizationCode(db.GetEngine(db.DefaultContext), redirectURI, codeChallenge, codeChallengeMethod)
 | 
						|
}
 | 
						|
 | 
						|
func (grant *OAuth2Grant) generateNewAuthorizationCode(e db.Engine, redirectURI, codeChallenge, codeChallengeMethod string) (code *OAuth2AuthorizationCode, err error) {
 | 
						|
	rBytes, err := util.CryptoRandomBytes(32)
 | 
						|
	if err != nil {
 | 
						|
		return &OAuth2AuthorizationCode{}, err
 | 
						|
	}
 | 
						|
	// Add a prefix to the base32, this is in order to make it easier
 | 
						|
	// for code scanners to grab sensitive tokens.
 | 
						|
	codeSecret := "gta_" + base32Lower.EncodeToString(rBytes)
 | 
						|
 | 
						|
	code = &OAuth2AuthorizationCode{
 | 
						|
		Grant:               grant,
 | 
						|
		GrantID:             grant.ID,
 | 
						|
		RedirectURI:         redirectURI,
 | 
						|
		Code:                codeSecret,
 | 
						|
		CodeChallenge:       codeChallenge,
 | 
						|
		CodeChallengeMethod: codeChallengeMethod,
 | 
						|
	}
 | 
						|
	if _, err := e.Insert(code); err != nil {
 | 
						|
		return nil, err
 | 
						|
	}
 | 
						|
	return code, nil
 | 
						|
}
 | 
						|
 | 
						|
// IncreaseCounter increases the counter and updates the grant
 | 
						|
func (grant *OAuth2Grant) IncreaseCounter() error {
 | 
						|
	return grant.increaseCount(db.GetEngine(db.DefaultContext))
 | 
						|
}
 | 
						|
 | 
						|
func (grant *OAuth2Grant) increaseCount(e db.Engine) error {
 | 
						|
	_, err := e.ID(grant.ID).Incr("counter").Update(new(OAuth2Grant))
 | 
						|
	if err != nil {
 | 
						|
		return err
 | 
						|
	}
 | 
						|
	updatedGrant, err := getOAuth2GrantByID(e, grant.ID)
 | 
						|
	if err != nil {
 | 
						|
		return err
 | 
						|
	}
 | 
						|
	grant.Counter = updatedGrant.Counter
 | 
						|
	return nil
 | 
						|
}
 | 
						|
 | 
						|
// ScopeContains returns true if the grant scope contains the specified scope
 | 
						|
func (grant *OAuth2Grant) ScopeContains(scope string) bool {
 | 
						|
	for _, currentScope := range strings.Split(grant.Scope, " ") {
 | 
						|
		if scope == currentScope {
 | 
						|
			return true
 | 
						|
		}
 | 
						|
	}
 | 
						|
	return false
 | 
						|
}
 | 
						|
 | 
						|
// SetNonce updates the current nonce value of a grant
 | 
						|
func (grant *OAuth2Grant) SetNonce(nonce string) error {
 | 
						|
	return grant.setNonce(db.GetEngine(db.DefaultContext), nonce)
 | 
						|
}
 | 
						|
 | 
						|
func (grant *OAuth2Grant) setNonce(e db.Engine, nonce string) error {
 | 
						|
	grant.Nonce = nonce
 | 
						|
	_, err := e.ID(grant.ID).Cols("nonce").Update(grant)
 | 
						|
	if err != nil {
 | 
						|
		return err
 | 
						|
	}
 | 
						|
	return nil
 | 
						|
}
 | 
						|
 | 
						|
// GetOAuth2GrantByID returns the grant with the given ID
 | 
						|
func GetOAuth2GrantByID(id int64) (*OAuth2Grant, error) {
 | 
						|
	return getOAuth2GrantByID(db.GetEngine(db.DefaultContext), id)
 | 
						|
}
 | 
						|
 | 
						|
func getOAuth2GrantByID(e db.Engine, id int64) (grant *OAuth2Grant, err error) {
 | 
						|
	grant = new(OAuth2Grant)
 | 
						|
	if has, err := e.ID(id).Get(grant); err != nil {
 | 
						|
		return nil, err
 | 
						|
	} else if !has {
 | 
						|
		return nil, nil
 | 
						|
	}
 | 
						|
	return
 | 
						|
}
 | 
						|
 | 
						|
// GetOAuth2GrantsByUserID lists all grants of a certain user
 | 
						|
func GetOAuth2GrantsByUserID(uid int64) ([]*OAuth2Grant, error) {
 | 
						|
	return getOAuth2GrantsByUserID(db.GetEngine(db.DefaultContext), uid)
 | 
						|
}
 | 
						|
 | 
						|
func getOAuth2GrantsByUserID(e db.Engine, uid int64) ([]*OAuth2Grant, error) {
 | 
						|
	type joinedOAuth2Grant struct {
 | 
						|
		Grant       *OAuth2Grant       `xorm:"extends"`
 | 
						|
		Application *OAuth2Application `xorm:"extends"`
 | 
						|
	}
 | 
						|
	var results *xorm.Rows
 | 
						|
	var err error
 | 
						|
	if results, err = e.
 | 
						|
		Table("oauth2_grant").
 | 
						|
		Where("user_id = ?", uid).
 | 
						|
		Join("INNER", "oauth2_application", "application_id = oauth2_application.id").
 | 
						|
		Rows(new(joinedOAuth2Grant)); err != nil {
 | 
						|
		return nil, err
 | 
						|
	}
 | 
						|
	defer results.Close()
 | 
						|
	grants := make([]*OAuth2Grant, 0)
 | 
						|
	for results.Next() {
 | 
						|
		joinedGrant := new(joinedOAuth2Grant)
 | 
						|
		if err := results.Scan(joinedGrant); err != nil {
 | 
						|
			return nil, err
 | 
						|
		}
 | 
						|
		joinedGrant.Grant.Application = joinedGrant.Application
 | 
						|
		grants = append(grants, joinedGrant.Grant)
 | 
						|
	}
 | 
						|
	return grants, nil
 | 
						|
}
 | 
						|
 | 
						|
// RevokeOAuth2Grant deletes the grant with grantID and userID
 | 
						|
func RevokeOAuth2Grant(grantID, userID int64) error {
 | 
						|
	return revokeOAuth2Grant(db.GetEngine(db.DefaultContext), grantID, userID)
 | 
						|
}
 | 
						|
 | 
						|
func revokeOAuth2Grant(e db.Engine, grantID, userID int64) error {
 | 
						|
	_, err := e.Delete(&OAuth2Grant{ID: grantID, UserID: userID})
 | 
						|
	return err
 | 
						|
}
 | 
						|
 | 
						|
// ErrOAuthClientIDInvalid will be thrown if client id cannot be found
 | 
						|
type ErrOAuthClientIDInvalid struct {
 | 
						|
	ClientID string
 | 
						|
}
 | 
						|
 | 
						|
// IsErrOauthClientIDInvalid checks if an error is a ErrReviewNotExist.
 | 
						|
func IsErrOauthClientIDInvalid(err error) bool {
 | 
						|
	_, ok := err.(ErrOAuthClientIDInvalid)
 | 
						|
	return ok
 | 
						|
}
 | 
						|
 | 
						|
// Error returns the error message
 | 
						|
func (err ErrOAuthClientIDInvalid) Error() string {
 | 
						|
	return fmt.Sprintf("Client ID invalid [Client ID: %s]", err.ClientID)
 | 
						|
}
 | 
						|
 | 
						|
// ErrOAuthApplicationNotFound will be thrown if id cannot be found
 | 
						|
type ErrOAuthApplicationNotFound struct {
 | 
						|
	ID int64
 | 
						|
}
 | 
						|
 | 
						|
// IsErrOAuthApplicationNotFound checks if an error is a ErrReviewNotExist.
 | 
						|
func IsErrOAuthApplicationNotFound(err error) bool {
 | 
						|
	_, ok := err.(ErrOAuthApplicationNotFound)
 | 
						|
	return ok
 | 
						|
}
 | 
						|
 | 
						|
// Error returns the error message
 | 
						|
func (err ErrOAuthApplicationNotFound) Error() string {
 | 
						|
	return fmt.Sprintf("OAuth application not found [ID: %d]", err.ID)
 | 
						|
}
 | 
						|
 | 
						|
// GetActiveOAuth2ProviderSources returns all actived LoginOAuth2 sources
 | 
						|
func GetActiveOAuth2ProviderSources() ([]*Source, error) {
 | 
						|
	sources := make([]*Source, 0, 1)
 | 
						|
	if err := db.GetEngine(db.DefaultContext).Where("is_active = ? and type = ?", true, OAuth2).Find(&sources); err != nil {
 | 
						|
		return nil, err
 | 
						|
	}
 | 
						|
	return sources, nil
 | 
						|
}
 | 
						|
 | 
						|
// GetActiveOAuth2SourceByName returns a OAuth2 AuthSource based on the given name
 | 
						|
func GetActiveOAuth2SourceByName(name string) (*Source, error) {
 | 
						|
	authSource := new(Source)
 | 
						|
	has, err := db.GetEngine(db.DefaultContext).Where("name = ? and type = ? and is_active = ?", name, OAuth2, true).Get(authSource)
 | 
						|
	if !has || err != nil {
 | 
						|
		return nil, err
 | 
						|
	}
 | 
						|
 | 
						|
	return authSource, nil
 | 
						|
}
 |