init: v1.0.0

sm/sm4/block_test.go

@@ -0,0 +1,96 @@
+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])
+	}
+
+}