mirror of
https://github.com/octoleo/restic.git
synced 2024-12-23 03:18:55 +00:00
140 lines
2.9 KiB
Go
140 lines
2.9 KiB
Go
package khepri_test
|
|
|
|
import (
|
|
"flag"
|
|
"io"
|
|
"io/ioutil"
|
|
"os"
|
|
"testing"
|
|
|
|
"github.com/fd0/khepri"
|
|
"github.com/fd0/khepri/backend"
|
|
"github.com/fd0/khepri/chunker"
|
|
)
|
|
|
|
var testPassword = "foobar"
|
|
var testCleanup = flag.Bool("test.cleanup", true, "clean up after running tests (remove local backend directory with all content)")
|
|
|
|
func setupBackend(t testing.TB) *backend.Local {
|
|
tempdir, err := ioutil.TempDir("", "khepri-test-")
|
|
ok(t, err)
|
|
|
|
b, err := backend.CreateLocal(tempdir)
|
|
ok(t, err)
|
|
|
|
return b
|
|
}
|
|
|
|
func teardownBackend(t testing.TB, b *backend.Local) {
|
|
if !*testCleanup {
|
|
t.Logf("leaving local backend at %s\n", b.Location())
|
|
return
|
|
}
|
|
|
|
ok(t, os.RemoveAll(b.Location()))
|
|
}
|
|
|
|
func setupKey(t testing.TB, be backend.Server, password string) *khepri.Key {
|
|
k, err := khepri.CreateKey(be, password)
|
|
ok(t, err)
|
|
|
|
return k
|
|
}
|
|
|
|
func TestRepo(t *testing.T) {
|
|
be := setupBackend(t)
|
|
defer teardownBackend(t, be)
|
|
_ = setupKey(t, be, testPassword)
|
|
}
|
|
|
|
func TestEncryptDecrypt(t *testing.T) {
|
|
be := setupBackend(t)
|
|
defer teardownBackend(t, be)
|
|
k := setupKey(t, be, testPassword)
|
|
|
|
for _, size := range []int{5, 23, 1 << 20, 7<<20 + 123} {
|
|
data := make([]byte, size)
|
|
f, err := os.Open("/dev/urandom")
|
|
ok(t, err)
|
|
|
|
_, err = io.ReadFull(f, data)
|
|
ok(t, err)
|
|
|
|
ciphertext := khepri.GetChunkBuf("TestEncryptDecrypt")
|
|
n, err := k.Encrypt(ciphertext, data)
|
|
ok(t, err)
|
|
|
|
plaintext, err := k.Decrypt(ciphertext[:n])
|
|
ok(t, err)
|
|
|
|
khepri.FreeChunkBuf("TestEncryptDecrypt", ciphertext)
|
|
|
|
equals(t, plaintext, data)
|
|
}
|
|
}
|
|
|
|
func TestLargeEncrypt(t *testing.T) {
|
|
be := setupBackend(t)
|
|
defer teardownBackend(t, be)
|
|
k := setupKey(t, be, testPassword)
|
|
|
|
for _, size := range []int{chunker.MaxSize, chunker.MaxSize + 1} {
|
|
data := make([]byte, size)
|
|
f, err := os.Open("/dev/urandom")
|
|
ok(t, err)
|
|
|
|
_, err = io.ReadFull(f, data)
|
|
ok(t, err)
|
|
|
|
ciphertext := make([]byte, size+khepri.CiphertextExtension)
|
|
n, err := k.Encrypt(ciphertext, data)
|
|
ok(t, err)
|
|
|
|
plaintext, err := k.Decrypt(ciphertext[:n])
|
|
ok(t, err)
|
|
|
|
equals(t, plaintext, data)
|
|
}
|
|
}
|
|
|
|
func BenchmarkEncrypt(b *testing.B) {
|
|
size := 8 << 20 // 8MiB
|
|
data := make([]byte, size)
|
|
|
|
be := setupBackend(b)
|
|
defer teardownBackend(b, be)
|
|
k := setupKey(b, be, testPassword)
|
|
|
|
b.ResetTimer()
|
|
b.SetBytes(int64(size))
|
|
|
|
buf := khepri.GetChunkBuf("BenchmarkEncrypt")
|
|
for i := 0; i < b.N; i++ {
|
|
_, err := k.Encrypt(buf, data)
|
|
ok(b, err)
|
|
}
|
|
khepri.FreeChunkBuf("BenchmarkEncrypt", buf)
|
|
}
|
|
|
|
func BenchmarkDecrypt(b *testing.B) {
|
|
size := 8 << 20 // 8MiB
|
|
data := make([]byte, size)
|
|
|
|
be := setupBackend(b)
|
|
defer teardownBackend(b, be)
|
|
k := setupKey(b, be, testPassword)
|
|
|
|
ciphertext := khepri.GetChunkBuf("BenchmarkDecrypt")
|
|
n, err := k.Encrypt(ciphertext, data)
|
|
ok(b, err)
|
|
|
|
b.ResetTimer()
|
|
b.SetBytes(int64(size))
|
|
|
|
for i := 0; i < b.N; i++ {
|
|
_, err := k.Decrypt(ciphertext[:n])
|
|
ok(b, err)
|
|
}
|
|
khepri.FreeChunkBuf("BenchmarkDecrypt", ciphertext)
|
|
}
|