amorfo77/restic
1
0
Fork 0
mirror of https://github.com/restic/restic.git synced 2026-09-21 03:28:30 +00:00
Code Issues Projects Releases Wiki Activity

crypto: Clarify slice usage for Encrypt/Decrypt

Browse Source
This commit is contained in:
Alexander Neumann
2015-04-19 14:57:35 +02:00
parent 63e79539e9
commit 8e867817a5
2 changed files with 20 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files
crypto/crypto_test.go
+10 -4
View File
@@ -31,9 +31,15 @@ func TestEncryptDecrypt(t *testing.T) {
ciphertext, err := crypto.Encrypt(k, restic.GetChunkBuf("TestEncryptDecrypt"), data)
OK(t, err)
Assert(t, len(ciphertext) == len(data)+crypto.Extension,
"ciphertext length does not match: want %d, got %d",
len(data)+crypto.Extension, len(ciphertext))
plaintext, err := crypto.Decrypt(k, nil, ciphertext)
OK(t, err)
Assert(t, len(plaintext) == len(data),
"plaintext length does not match: want %d, got %d",
len(data), len(plaintext))
restic.FreeChunkBuf("TestEncryptDecrypt", ciphertext)
@@ -54,7 +60,7 @@ func TestSmallBuffer(t *testing.T) {
ciphertext := make([]byte, size/2)
ciphertext, err = crypto.Encrypt(k, ciphertext, data)
// this must throw an error, since the target slice is too small
// this must extend the slice
Assert(t, cap(ciphertext) > size/2,
"expected extended slice, but capacity is only %d bytes",
cap(ciphertext))
@@ -77,12 +83,12 @@ func TestSameBuffer(t *testing.T) {
_, err = io.ReadFull(f, data)
OK(t, err)
ciphertext := make([]byte, size)
copy(ciphertext, data)
ciphertext := make([]byte, 0, size+crypto.Extension)
ciphertext, err = crypto.Encrypt(k, ciphertext, ciphertext)
ciphertext, err = crypto.Encrypt(k, ciphertext, data)
OK(t, err)
// use the same buffer for decryption
ciphertext, err = crypto.Decrypt(k, ciphertext, ciphertext)
OK(t, err)
Assert(t, bytes.Equal(ciphertext, data),