1
0
mirror of https://github.com/henrydcase/nobs.git synced 2024-11-22 15:18:57 +00:00

CTR-DRBG: Use hardware acceleration on X86 (#18)

benchmark              old ns/op     new ns/op     delta
BenchmarkInit-4        3403          397           -88.33%
BenchmarkRead-4        14535         1560          -89.27%
This commit is contained in:
Henry Case 2019-04-09 23:50:21 +01:00 committed by GitHub
parent 71624cdc4c
commit 6f9706df01
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
3 changed files with 15 additions and 4 deletions

View File

@ -14,6 +14,7 @@ package drbg
import ( import (
"github.com/henrydcase/nobs/drbg/internal/aes" "github.com/henrydcase/nobs/drbg/internal/aes"
"github.com/henrydcase/nobs/utils"
) )
// Constants below correspond to AES-256, which is currently // Constants below correspond to AES-256, which is currently
@ -30,12 +31,15 @@ type CtrDrbg struct {
counter uint counter uint
strength uint strength uint
resistance bool resistance bool
blockEnc aes.AES blockEnc aes.IAES
tmpBlk [3 * BlockLen]byte tmpBlk [3 * BlockLen]byte
} }
func NewCtrDrbg() *CtrDrbg { func NewCtrDrbg() *CtrDrbg {
return new(CtrDrbg) if utils.X86.HasAES {
return &CtrDrbg{blockEnc: &aes.AESAsm{}}
}
return &CtrDrbg{blockEnc: &aes.AES{}}
} }
func (c *CtrDrbg) inc() { func (c *CtrDrbg) inc() {

View File

@ -18,6 +18,13 @@ type AES struct {
keyLen int keyLen int
} }
// AES interface
type IAES interface {
SetKey(key []byte) error
Encrypt(dst, src []byte)
Decrypt(dst, src []byte)
}
type KeySizeError int type KeySizeError int
func (k KeySizeError) Error() string { func (k KeySizeError) Error() string {

View File

@ -21,8 +21,8 @@ func decryptBlockAsm(nr int, xk *uint32, dst, src *byte)
func expandKeyAsm(nr int, key *byte, enc *uint32, dec *uint32) func expandKeyAsm(nr int, key *byte, enc *uint32, dec *uint32)
type AESAsm struct { type AESAsm struct {
enc []uint32 enc [32 + 28]uint32
dec []uint32 dec [32 + 28]uint32
} }
func (c *AESAsm) SetKey(key []byte) error { func (c *AESAsm) SetKey(key []byte) error {