Arav
/
justcaptcha
1
0
Fork 0
Code Issues Releases 6 Activity

ID generating moved off to a separate function NewID().

This commit is contained in:
Alexander Andreev 2022-06-27 01:02:24 +04:00
parent 6307433ea0
commit b65f3be236
Signed by: Arav
GPG Key ID: 0388CC8FAA51063F
1 changed files with 14 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
pkg/captcha/db.go
View File

@ -9,9 +9,21 @@ import (
"time" "time"
) )
var expiredScanInterval = 60 * time.Second
type ID string type ID string
var expiredScanInterval = 60 * time.Second // NewID generates an ID as a sha256 hash of additionalData, current time
// and answer encoded with base64 in raw URL variant.
func NewID(additionalData string, answer Answer) ID {
idHash := sha256.New()
idHash.Write([]byte(additionalData))
idHash.Write([]byte(strconv.FormatInt(time.Now().UnixMicro(), 16)))
idHash.Write([]byte(answer))
return ID(base64.RawURLEncoding.EncodeToString(idHash.Sum(nil)))
}
type ICaptchaDB interface { type ICaptchaDB interface {
New(data string) (ICaptcha, ID) New(data string) (ICaptcha, ID)
@ -54,13 +66,7 @@ func (cdb *CaptchaDB) SetExpiry(expire time.Duration) {
// `data` string is an additional random data used to generate an ID, // `data` string is an additional random data used to generate an ID,
// e.g. IP-address. // e.g. IP-address.
func (cdb *CaptchaDB) New(data string, captcha ICaptcha) (ICaptcha, ID) { func (cdb *CaptchaDB) New(data string, captcha ICaptcha) (ICaptcha, ID) {
idHash := sha256.New() id := NewID(data, captcha.GetAnswer())
idHash.Write([]byte(data))
idHash.Write([]byte(strconv.FormatInt(time.Now().UnixMicro(), 16)))
idHash.Write([]byte(captcha.GetAnswer()))
id := ID(base64.RawURLEncoding.EncodeToString(idHash.Sum(nil)))
cdb.mut.Lock() cdb.mut.Lock()
defer cdb.mut.Unlock() defer cdb.mut.Unlock()