forked from TrueCloudLab/restic
119 lines
2.3 KiB
Go
119 lines
2.3 KiB
Go
package restic_test
|
|
|
|
import (
|
|
"bytes"
|
|
"io"
|
|
"math/rand"
|
|
"testing"
|
|
|
|
"github.com/restic/restic"
|
|
"github.com/restic/restic/chunker"
|
|
)
|
|
|
|
func get_random(seed, count int) []byte {
|
|
buf := make([]byte, count)
|
|
|
|
rnd := rand.New(rand.NewSource(23))
|
|
for i := 0; i < count; i += 4 {
|
|
r := rnd.Uint32()
|
|
buf[i] = byte(r)
|
|
buf[i+1] = byte(r >> 8)
|
|
buf[i+2] = byte(r >> 16)
|
|
buf[i+3] = byte(r >> 24)
|
|
}
|
|
|
|
return buf
|
|
}
|
|
|
|
const bufSize = chunker.MiB
|
|
|
|
func benchmarkChunkEncrypt(b testing.TB, rd interface {
|
|
io.ReadSeeker
|
|
io.ReaderAt
|
|
}, key *restic.Key) {
|
|
chunkBuf := make([]byte, restic.CiphertextExtension+chunker.MaxSize)
|
|
|
|
ch := restic.GetChunker("BenchmarkChunkEncrypt")
|
|
rd.Seek(0, 0)
|
|
ch.Reset(rd)
|
|
|
|
for {
|
|
chunk, err := ch.Next()
|
|
|
|
if err == io.EOF {
|
|
break
|
|
}
|
|
|
|
ok(b, err)
|
|
|
|
// reduce length of chunkBuf
|
|
chunkBuf = chunkBuf[:chunk.Length]
|
|
n, err := io.ReadFull(chunk.Reader(rd), chunkBuf)
|
|
ok(b, err)
|
|
assert(b, uint(n) == chunk.Length, "invalid length: got %d, expected %d", n, chunk.Length)
|
|
|
|
_, err = key.Encrypt(chunkBuf, chunkBuf)
|
|
ok(b, err)
|
|
}
|
|
|
|
restic.FreeChunker("BenchmarkChunkEncrypt", ch)
|
|
}
|
|
|
|
func BenchmarkChunkEncrypt(b *testing.B) {
|
|
data := get_random(23, 10<<20) // 10MiB
|
|
rd := bytes.NewReader(data)
|
|
|
|
be := setupBackend(b)
|
|
defer teardownBackend(b, be)
|
|
key := setupKey(b, be, "geheim")
|
|
|
|
b.ResetTimer()
|
|
b.SetBytes(int64(len(data)))
|
|
|
|
for i := 0; i < b.N; i++ {
|
|
benchmarkChunkEncrypt(b, rd, key)
|
|
}
|
|
}
|
|
|
|
func benchmarkChunkEncryptP(b *testing.PB, rd interface {
|
|
io.ReadSeeker
|
|
io.ReaderAt
|
|
}, key *restic.Key) {
|
|
chunkBuf := make([]byte, restic.CiphertextExtension+chunker.MaxSize)
|
|
|
|
ch := restic.GetChunker("BenchmarkChunkEncryptP")
|
|
rd.Seek(0, 0)
|
|
ch.Reset(rd)
|
|
|
|
for {
|
|
chunk, err := ch.Next()
|
|
if err == io.EOF {
|
|
break
|
|
}
|
|
|
|
// reduce length of chunkBuf
|
|
chunkBuf = chunkBuf[:chunk.Length]
|
|
io.ReadFull(chunk.Reader(rd), chunkBuf)
|
|
key.Encrypt(chunkBuf, chunkBuf)
|
|
}
|
|
|
|
restic.FreeChunker("BenchmarkChunkEncryptP", ch)
|
|
}
|
|
|
|
func BenchmarkChunkEncryptParallel(b *testing.B) {
|
|
be := setupBackend(b)
|
|
defer teardownBackend(b, be)
|
|
key := setupKey(b, be, "geheim")
|
|
|
|
data := get_random(23, 10<<20) // 10MiB
|
|
|
|
b.ResetTimer()
|
|
b.SetBytes(int64(len(data)))
|
|
|
|
b.RunParallel(func(pb *testing.PB) {
|
|
for pb.Next() {
|
|
rd := bytes.NewReader(data)
|
|
benchmarkChunkEncryptP(pb, rd, key)
|
|
}
|
|
})
|
|
}
|