123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154 |
- // Copyright 2018 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 login
-
- import (
- "fmt"
-
- "code.gitea.io/gitea/models/db"
- "code.gitea.io/gitea/modules/log"
- "code.gitea.io/gitea/modules/timeutil"
-
- "github.com/tstranex/u2f"
- )
-
- // ____ ________________________________ .__ __ __ .__
- // | | \_____ \_ _____/\______ \ ____ ____ |__| _______/ |_____________ _/ |_|__| ____ ____
- // | | // ____/| __) | _// __ \ / ___\| |/ ___/\ __\_ __ \__ \\ __\ |/ _ \ / \
- // | | // \| \ | | \ ___// /_/ > |\___ \ | | | | \// __ \| | | ( <_> ) | \
- // |______/ \_______ \___ / |____|_ /\___ >___ /|__/____ > |__| |__| (____ /__| |__|\____/|___| /
- // \/ \/ \/ \/_____/ \/ \/ \/
-
- // ErrU2FRegistrationNotExist represents a "ErrU2FRegistrationNotExist" kind of error.
- type ErrU2FRegistrationNotExist struct {
- ID int64
- }
-
- func (err ErrU2FRegistrationNotExist) Error() string {
- return fmt.Sprintf("U2F registration does not exist [id: %d]", err.ID)
- }
-
- // IsErrU2FRegistrationNotExist checks if an error is a ErrU2FRegistrationNotExist.
- func IsErrU2FRegistrationNotExist(err error) bool {
- _, ok := err.(ErrU2FRegistrationNotExist)
- return ok
- }
-
- // U2FRegistration represents the registration data and counter of a security key
- type U2FRegistration struct {
- ID int64 `xorm:"pk autoincr"`
- Name string
- UserID int64 `xorm:"INDEX"`
- Raw []byte
- Counter uint32 `xorm:"BIGINT"`
- CreatedUnix timeutil.TimeStamp `xorm:"INDEX created"`
- UpdatedUnix timeutil.TimeStamp `xorm:"INDEX updated"`
- }
-
- func init() {
- db.RegisterModel(new(U2FRegistration))
- }
-
- // TableName returns a better table name for U2FRegistration
- func (reg U2FRegistration) TableName() string {
- return "u2f_registration"
- }
-
- // Parse will convert the db entry U2FRegistration to an u2f.Registration struct
- func (reg *U2FRegistration) Parse() (*u2f.Registration, error) {
- r := new(u2f.Registration)
- return r, r.UnmarshalBinary(reg.Raw)
- }
-
- func (reg *U2FRegistration) updateCounter(e db.Engine) error {
- _, err := e.ID(reg.ID).Cols("counter").Update(reg)
- return err
- }
-
- // UpdateCounter will update the database value of counter
- func (reg *U2FRegistration) UpdateCounter() error {
- return reg.updateCounter(db.GetEngine(db.DefaultContext))
- }
-
- // U2FRegistrationList is a list of *U2FRegistration
- type U2FRegistrationList []*U2FRegistration
-
- // ToRegistrations will convert all U2FRegistrations to u2f.Registrations
- func (list U2FRegistrationList) ToRegistrations() []u2f.Registration {
- regs := make([]u2f.Registration, 0, len(list))
- for _, reg := range list {
- r, err := reg.Parse()
- if err != nil {
- log.Error("parsing u2f registration: %v", err)
- continue
- }
- regs = append(regs, *r)
- }
-
- return regs
- }
-
- func getU2FRegistrationsByUID(e db.Engine, uid int64) (U2FRegistrationList, error) {
- regs := make(U2FRegistrationList, 0)
- return regs, e.Where("user_id = ?", uid).Find(®s)
- }
-
- // GetU2FRegistrationByID returns U2F registration by id
- func GetU2FRegistrationByID(id int64) (*U2FRegistration, error) {
- return getU2FRegistrationByID(db.GetEngine(db.DefaultContext), id)
- }
-
- func getU2FRegistrationByID(e db.Engine, id int64) (*U2FRegistration, error) {
- reg := new(U2FRegistration)
- if found, err := e.ID(id).Get(reg); err != nil {
- return nil, err
- } else if !found {
- return nil, ErrU2FRegistrationNotExist{ID: id}
- }
- return reg, nil
- }
-
- // GetU2FRegistrationsByUID returns all U2F registrations of the given user
- func GetU2FRegistrationsByUID(uid int64) (U2FRegistrationList, error) {
- return getU2FRegistrationsByUID(db.GetEngine(db.DefaultContext), uid)
- }
-
- // HasU2FRegistrationsByUID returns whether a given user has U2F registrations
- func HasU2FRegistrationsByUID(uid int64) (bool, error) {
- return db.GetEngine(db.DefaultContext).Where("user_id = ?", uid).Exist(&U2FRegistration{})
- }
-
- func createRegistration(e db.Engine, userID int64, name string, reg *u2f.Registration) (*U2FRegistration, error) {
- raw, err := reg.MarshalBinary()
- if err != nil {
- return nil, err
- }
- r := &U2FRegistration{
- UserID: userID,
- Name: name,
- Counter: 0,
- Raw: raw,
- }
- _, err = e.InsertOne(r)
- if err != nil {
- return nil, err
- }
- return r, nil
- }
-
- // CreateRegistration will create a new U2FRegistration from the given Registration
- func CreateRegistration(userID int64, name string, reg *u2f.Registration) (*U2FRegistration, error) {
- return createRegistration(db.GetEngine(db.DefaultContext), userID, name, reg)
- }
-
- // DeleteRegistration will delete U2FRegistration
- func DeleteRegistration(reg *U2FRegistration) error {
- return deleteRegistration(db.GetEngine(db.DefaultContext), reg)
- }
-
- func deleteRegistration(e db.Engine, reg *U2FRegistration) error {
- _, err := e.Delete(reg)
- return err
- }
|