24904585a2
Fix user tests Add swagger docs Fix lint Add totp check when logging in Make totp enrollment work Add migration for totp table go mod vendor Add routes for totp routes Add route handler for totp routes Add basic implementation to enroll a user in totp Co-authored-by: kolaente <k@knt.li> Reviewed-on: https://kolaente.dev/vikunja/api/pulls/383
44 lines
1.1 KiB
Go
44 lines
1.1 KiB
Go
package utils
|
|
|
|
import (
|
|
"sync"
|
|
)
|
|
|
|
type ReedSolomonEncoder struct {
|
|
gf *GaloisField
|
|
polynomes []*GFPoly
|
|
m *sync.Mutex
|
|
}
|
|
|
|
func NewReedSolomonEncoder(gf *GaloisField) *ReedSolomonEncoder {
|
|
return &ReedSolomonEncoder{
|
|
gf, []*GFPoly{NewGFPoly(gf, []int{1})}, new(sync.Mutex),
|
|
}
|
|
}
|
|
|
|
func (rs *ReedSolomonEncoder) getPolynomial(degree int) *GFPoly {
|
|
rs.m.Lock()
|
|
defer rs.m.Unlock()
|
|
|
|
if degree >= len(rs.polynomes) {
|
|
last := rs.polynomes[len(rs.polynomes)-1]
|
|
for d := len(rs.polynomes); d <= degree; d++ {
|
|
next := last.Multiply(NewGFPoly(rs.gf, []int{1, rs.gf.ALogTbl[d-1+rs.gf.Base]}))
|
|
rs.polynomes = append(rs.polynomes, next)
|
|
last = next
|
|
}
|
|
}
|
|
return rs.polynomes[degree]
|
|
}
|
|
|
|
func (rs *ReedSolomonEncoder) Encode(data []int, eccCount int) []int {
|
|
generator := rs.getPolynomial(eccCount)
|
|
info := NewGFPoly(rs.gf, data)
|
|
info = info.MultByMonominal(eccCount, 1)
|
|
_, remainder := info.Divide(generator)
|
|
|
|
result := make([]int, eccCount)
|
|
numZero := int(eccCount) - len(remainder.Coefficients)
|
|
copy(result[numZero:], remainder.Coefficients)
|
|
return result
|
|
}
|