adoublef · October 14, 2024 11:42 · Oct 14, 2024
diff --git a/simple_test.go b/simple_test.go
@@ -0,0 +1,87 @@
+package aesctr256hmacsha256_test
+
+import (
+	"crypto/aes"
+	"crypto/cipher"
+	"embed"
+	"encoding/hex"
+	"io"
+	"os"
+	"testing"
+)
+
+//go:embed all:testdata/input/*
+var embedFS embed.FS
+
+func Benchmark_aesctr256hmacsha256(b *testing.B) {
+
+	b.Run("Copy32", func(b *testing.B) {
+		// create a temp file
+		dst, err := os.CreateTemp(b.TempDir(), "sink-*")
+		if err != nil {
+			b.Errorf(`dst, %q := os.CreateTemp(b.TempDir(), "sink-*")`, err)
+		}
+		// create compression
+		// create a tar file
+		b.ResetTimer()
+		for n := 0; n < b.N; n++ {
+			b.StopTimer()
+			src, err := embedFS.Open("testdata/input/hello.txt")
+			if err != nil {
+				b.Errorf(`_, %q := embedFS.Open("testdata/input/hello.txt")`, err)
+			}
+			var (
+				buf = make([]byte, 32*1024)
+			)
+			b.StartTimer()
+			_, err = io.CopyBuffer(dst, src, buf)
+			if err != nil {
+				b.Errorf(`_, %q := io.Copy(dst, src)`, err)
+			}
+		}
+	})
+
+	b.Run("BlockCopy32", func(b *testing.B) {
+		// create a temp file
+		dst, err := os.CreateTemp(b.TempDir(), "sink-*")
+		if err != nil {
+			b.Errorf(`dst, %q := os.CreateTemp(b.TempDir(), "sink-*")`, err)
+		}
+		var (
+			key, _ = hex.DecodeString("6368616e676520746869732070617373776f726420746f206120736563726574")
+			iv     [aes.BlockSize]byte
+		)
+
+		block, err := aes.NewCipher(key)
+		if err != nil {
+			panic(err)
+		}
+		var (
+			stream = cipher.NewCTR(block, iv[:])
+			writer = &cipher.StreamWriter{S: stream, W: dst}
+		)
+		// create compression
+		// create a tar file
+		b.ResetTimer()
+		for n := 0; n < b.N; n++ {
+			b.StopTimer()
+			src, err := embedFS.Open("testdata/input/hello.txt")
+			if err != nil {
+				b.Errorf(`_, %q := embedFS.Open("testdata/input/hello.txt")`, err)
+			}
+			var (
+				// block+hmac
+				buf = make([]byte, 32*1024)
+			)
+			b.StartTimer()
+			benchCopy(b, writer, src, buf)
+		}
+	})
+}
+
+func benchCopy(b *testing.B, dst io.Writer, src io.Reader, buf []byte) {
+	_, err := io.CopyBuffer(dst, src, buf)
+	if err != nil {
+		b.Errorf(`_, %q := io.Copy(dst, src)`, err)
+	}
+}