apps/pg
1
0
Fork 0
mirror of https://github.com/sigcn/pg.git synced 2025-09-27 01:05:51 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

secure/chacha20poly1305: add func SetDefaultTimeWindow

Browse Source
This commit is contained in:
rkonfj
2024-09-27 11:01:47 +08:00
parent a5f6b68079
commit f16f1319ec
1 changed files with 10 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
secure/chacha20poly1305/chacha20poly1305.go
View File

@@ -15,6 +15,12 @@ import (
var _ secure.SymmAlgo = (*Chacha20Poly1305)(nil) var _ secure.SymmAlgo = (*Chacha20Poly1305)(nil)
var timeWindow int64 = 10
func SetDefaultTimeWindow(seconds int64) {
timeWindow = seconds
}
type Chacha20Poly1305 struct { type Chacha20Poly1305 struct {
mut sync.RWMutex mut sync.RWMutex
cipher *lru.Cache[string, cipher.AEAD] cipher *lru.Cache[string, cipher.AEAD]
@@ -30,7 +36,7 @@ func (s *Chacha20Poly1305) Encrypt(data []byte, pubKey string) ([]byte, error) {
return nil, err return nil, err
} }
nonce := make([]byte, aead.NonceSize()) nonce := make([]byte, aead.NonceSize())
binary.LittleEndian.PutUint64(nonce[aead.NonceSize()-8:], uint64(time.Now().Unix()/5)) binary.LittleEndian.PutUint64(nonce[aead.NonceSize()-8:], uint64(time.Now().Unix()/timeWindow))
return aead.Seal(nil, nonce, data, nil), nil return aead.Seal(nil, nonce, data, nil), nil
} }
@@ -46,13 +52,13 @@ func (s *Chacha20Poly1305) Decrypt(data []byte, pubKey string) ([]byte, error) {
startIndex := aead.NonceSize() - 8 startIndex := aead.NonceSize() - 8
nowUnix := time.Now().Unix() nowUnix := time.Now().Unix()
binary.LittleEndian.PutUint64(nonce[startIndex:], uint64(nowUnix/5)) binary.LittleEndian.PutUint64(nonce[startIndex:], uint64(nowUnix/timeWindow))
plain, err := aead.Open(nil, nonce, data, nil) plain, err := aead.Open(nil, nonce, data, nil)
if err != nil { if err != nil {
binary.LittleEndian.PutUint64(nonce[startIndex:], uint64(nowUnix/5+1)) binary.LittleEndian.PutUint64(nonce[startIndex:], uint64(nowUnix/timeWindow+1))
plain, err = aead.Open(nil, nonce, data, nil) plain, err = aead.Open(nil, nonce, data, nil)
if err != nil { if err != nil {
binary.LittleEndian.PutUint64(nonce[startIndex:], uint64(nowUnix/5-1)) binary.LittleEndian.PutUint64(nonce[startIndex:], uint64(nowUnix/timeWindow-1))
plain, err = aead.Open(nil, nonce, data, nil) plain, err = aead.Open(nil, nonce, data, nil)
if err != nil { if err != nil {
return nil, errors.New("invalid data") return nil, errors.New("invalid data")