mirror of
https://github.com/awnumar/memguard.git
synced 2026-02-06 17:59:49 +00:00
8750c16425
Memcall is not touched yet. Otherwise this replaces sentinel error values with IsX exported predicate functions. This enables more effective error handling by the users and leaves much more freedom for changing the implementation in the future without breaking API. Updates #111
88 lines
1.7 KiB
Go
88 lines
1.7 KiB
Go
package memguard
|
|
|
|
import (
|
|
"bytes"
|
|
"testing"
|
|
|
|
"github.com/awnumar/memguard/core"
|
|
)
|
|
|
|
func TestNewEnclave(t *testing.T) {
|
|
e := NewEnclave([]byte("yellow submarine"))
|
|
if e == nil {
|
|
t.Error("got nil enclave")
|
|
}
|
|
data, err := e.Open()
|
|
if err != nil {
|
|
t.Error("unexpected error:", err)
|
|
}
|
|
if !bytes.Equal(data.Bytes(), []byte("yellow submarine")) {
|
|
t.Error("data doesn't match input")
|
|
}
|
|
data.Destroy()
|
|
e = NewEnclave([]byte{})
|
|
if e != nil {
|
|
t.Error("enclave should be nil")
|
|
}
|
|
}
|
|
|
|
func TestNewEnclaveRandom(t *testing.T) {
|
|
e := NewEnclaveRandom(32)
|
|
if e == nil {
|
|
t.Error("got nil enclave")
|
|
}
|
|
data, err := e.Open()
|
|
if err != nil {
|
|
t.Error("unexpected error:", err)
|
|
}
|
|
if len(data.Bytes()) != 32 || cap(data.Bytes()) != 32 {
|
|
t.Error("buffer sizes incorrect")
|
|
}
|
|
if bytes.Equal(data.Bytes(), make([]byte, 32)) {
|
|
t.Error("buffer not randomised")
|
|
}
|
|
data.Destroy()
|
|
e = NewEnclaveRandom(0)
|
|
if e != nil {
|
|
t.Error("should be nil")
|
|
}
|
|
}
|
|
|
|
func TestOpen(t *testing.T) {
|
|
e := NewEnclave([]byte("yellow submarine"))
|
|
if e == nil {
|
|
t.Error("got nil enclave")
|
|
}
|
|
b, err := e.Open()
|
|
if err != nil {
|
|
t.Error("unexpected error;", err)
|
|
}
|
|
if b == nil {
|
|
t.Error("buffer should not be nil")
|
|
}
|
|
if !bytes.Equal(b.Bytes(), []byte("yellow submarine")) {
|
|
t.Error("data does not match")
|
|
}
|
|
Purge() // reset the session
|
|
b, err = e.Open()
|
|
if !core.IsDecryptionFailed(err) {
|
|
t.Error("expected decryption error; got", err)
|
|
}
|
|
if b != nil {
|
|
t.Error("buffer should be nil")
|
|
}
|
|
e = NewEnclaveRandom(0)
|
|
if !panics(func() {
|
|
e.Open()
|
|
}) {
|
|
t.Error("func should panic on nil enclave")
|
|
}
|
|
}
|
|
|
|
func panics(fn func()) (panicked bool) {
|
|
defer func() {
|
|
panicked = (recover() != nil)
|
|
}()
|
|
fn()
|
|
return
|
|
}
|