mirror/gmsm
1
0
Fork 0
mirror of https://github.com/emmansun/gmsm.git synced 2025-06-29 00:37:51 +08:00
Code Issues Packages Projects Releases Wiki Activity

sm4: fix test case issue

Browse Source
This commit is contained in:
Sun Yimin 2022-07-12 10:08:23 +08:00 committed by GitHub
parent daa6853722
commit 87f6f6a736
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 50 additions and 42 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

50
sm4/cipher_asm_test.go Normal file
View File

@ -0,0 +1,50 @@
//go:build (amd64 && !generic) || (arm64 && !generic)
// +build amd64,!generic arm64,!generic
package sm4
import (
"bytes"
"crypto/cipher"
"testing"
)
func TestWithoutGFMUL(t *testing.T) {
key := make([]byte, 16)
src := make([]byte, 16)
var dst []byte
var nonce [12]byte
var c cipher.Block
var err error
if supportSM4 {
c, err = newCipherNI(key)
} else if !supportsAES {
c, err = newCipherGeneric(key)
} else {
blocks := 4
if useAVX2 {
blocks = 8
}
c1 := &sm4CipherAsm{sm4Cipher{make([]uint32, rounds), make([]uint32, rounds)}, blocks, blocks * BlockSize}
expandKeyAsm(&key[0], &ck[0], &c1.enc[0], &c1.dec[0], INST_AES)
c = c1
}
if err != nil {
t.Fatal(err)
}
var sm4gcm cipher.AEAD
sm4gcm, err = cipher.NewGCM(c)
if err != nil {
t.Fatal(err)
}
dst = sm4gcm.Seal(nil, nonce[:], src, nil)
src, err = sm4gcm.Open(nil, nonce[:], dst, nil)
if err != nil {
t.Fatal(err)
}
if !bytes.Equal(key, src) {
t.Errorf("bad encryption")
}
}

42
sm4/cipher_test.go
View File

@ -1,8 +1,6 @@
package sm4 package sm4
import ( import (
"bytes"
"crypto/cipher"
"reflect" "reflect"
"testing" "testing"
) )
@ -82,46 +80,6 @@ func TestEncryptDecryptPanic(t *testing.T) {
shouldPanic(t, func() { c.Decrypt(src[1:], src) }) shouldPanic(t, func() { c.Decrypt(src[1:], src) })
} }
func TestWithoutGFMUL(t *testing.T) {
key := make([]byte, 16)
src := make([]byte, 16)
var dst []byte
var nonce [12]byte
var c cipher.Block
var err error
if supportSM4 {
c, err = newCipherNI(key)
} else if !supportsAES {
c, err = newCipherGeneric(key)
} else {
blocks := 4
if useAVX2 {
blocks = 8
}
c1 := &sm4CipherAsm{sm4Cipher{make([]uint32, rounds), make([]uint32, rounds)}, blocks, blocks * BlockSize}
expandKeyAsm(&key[0], &ck[0], &c1.enc[0], &c1.dec[0], INST_AES)
c = c1
}
if err != nil {
t.Fatal(err)
}
var sm4gcm cipher.AEAD
sm4gcm, err = cipher.NewGCM(c)
if err != nil {
t.Fatal(err)
}
dst = sm4gcm.Seal(nil, nonce[:], src, nil)
src, err = sm4gcm.Open(nil, nonce[:], dst, nil)
if err != nil {
t.Fatal(err)
}
if !bytes.Equal(key, src) {
t.Errorf("bad encryption")
}
}
func shouldPanic(t *testing.T, f func()) { func shouldPanic(t *testing.T, f func()) {
t.Helper() t.Helper()
defer func() { _ = recover() }() defer func() { _ = recover() }()