yaole
97 lines
1.9 KiB
Go
97 lines
1.9 KiB
Go
package sm4
|
|
|
|
import (
|
|
"crypto/rand"
|
|
"fmt"
|
|
"testing"
|
|
"time"
|
|
)
|
|
|
|
func TestSBox8(t *testing.T) {
|
|
// Generate the sbox8
|
|
sbox8 := make([]uint32, 256)
|
|
for i := 0; i < 256; i++ {
|
|
sbox8[i] = l(uint32(sbox[i]))
|
|
}
|
|
|
|
for i := 0; i < 256; i++ {
|
|
fmt.Printf("0x%08x ", sbox8[i])
|
|
if i > 0 && (i+1)%8 == 0 {
|
|
fmt.Println()
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestSBox16(t *testing.T) {
|
|
for i := uint32(0); i < 0xffffffff; i++ {
|
|
if lt(i) != l(tau(i)) {
|
|
panic(fmt.Sprintf("lt(%x) != l(tau(%x))", i, i))
|
|
}
|
|
}
|
|
}
|
|
func BenchmarkBlock(b *testing.B) {
|
|
var data, dst [16]byte
|
|
var key [16]byte
|
|
rand.Read(key[:])
|
|
rand.Read(data[:])
|
|
block, _ := NewCipher(key[:])
|
|
|
|
b.ResetTimer()
|
|
for i := 0; i < b.N; i++ {
|
|
block.Encrypt(dst[:], data[:])
|
|
}
|
|
b.StopTimer()
|
|
}
|
|
|
|
func TestBlockSpeed(t *testing.T) {
|
|
var dst, data [16 * 1000]byte
|
|
var key [16]byte
|
|
rand.Read(key[:])
|
|
rand.Read(data[:])
|
|
b, _ := NewCipher(key[:])
|
|
|
|
times := 10000
|
|
start := time.Now()
|
|
for i := 0; i < times; i++ {
|
|
for j := 0; j < len(data); j += 16 {
|
|
b.Encrypt(data[j:], data[j:])
|
|
// b.Encrypt(dst[j:], data[j:])
|
|
}
|
|
}
|
|
end := time.Now()
|
|
elapsed := end.Sub(start)
|
|
t.Log("SM4 Encrypt Block: ", int(float64(times*len(data))/float64(1024*1024)/elapsed.Seconds()), "MBps")
|
|
|
|
start = time.Now()
|
|
for i := 0; i < times; i++ {
|
|
for j := 0; j < 16*1000; j += 16 {
|
|
// b.Decrypt(data[j:], data[j:])
|
|
b.Decrypt(data[j:], dst[j:])
|
|
}
|
|
}
|
|
end = time.Now()
|
|
elapsed = end.Sub(start)
|
|
t.Log("SM4 Encrypt Block: ", int(float64(times*len(data))/float64(1024*1024)/elapsed.Seconds()), "MBps")
|
|
}
|
|
|
|
func TestBlock(t *testing.T) {
|
|
data := make([]byte, 16)
|
|
|
|
encKey := make([]uint32, 32)
|
|
decKey := make([]uint32, 32)
|
|
expandKeyGo([]byte("1234567812345678"), encKey, decKey)
|
|
for i := 0; i < 1000; i++ {
|
|
encryptBlockGo(encKey, data, data)
|
|
}
|
|
|
|
fmt.Printf("Key:")
|
|
for i := 0; i < 32; i++ {
|
|
fmt.Printf("0x%08x, ", encKey[i])
|
|
}
|
|
fmt.Printf("cipher:")
|
|
for i := 0; i < 16; i++ {
|
|
fmt.Printf("%02x", data[i])
|
|
}
|
|
|
|
}
|