1
0
Fork 0
mirror of https://github.com/Luzifer/go-openssl.git synced 2024-12-21 03:11:18 +00:00
go-openssl/stream_test.go
Asutorufa 8d84455575
Fix panic when reading incomplete blocks from underlying reader (#27)
Co-authored-by: Knut Ahlers <knut@ahlers.me>
2023-09-11 13:59:01 +02:00

68 lines
1.5 KiB
Go

package openssl
import (
"bytes"
"crypto/aes"
"fmt"
"io"
"testing"
"github.com/stretchr/testify/assert"
"github.com/stretchr/testify/require"
)
type limitedSizeReader struct {
size int
r io.Reader
}
func (o *limitedSizeReader) Read(b []byte) (int, error) {
if len(b) == 0 {
return 0, nil
}
return o.r.Read(b[:o.size]) //nolint:wrapcheck
}
func TestReader(t *testing.T) {
o := New()
pass := "abcd"
plaintext := []byte("123abc,./vvvczcekdewfeojdosndsdlsndlncnepcnodcnviorf409eofnvkdfvjfvdsoijvo")
data, err := o.EncryptBinaryBytes(pass, plaintext, BytesToKeyMD5)
require.NoError(t, err)
for i := 1; i <= aes.BlockSize+1; i++ {
t.Run(fmt.Sprintf("read_size_%d", i), func(t *testing.T) {
var (
buf = new(bytes.Buffer)
bytesBuf = make([]byte, aes.BlockSize+1)
r = &limitedSizeReader{i, bytes.NewReader(data)}
)
_, err = io.CopyBuffer(buf, NewReader(r, pass, BytesToKeyMD5), bytesBuf)
require.NoError(t, err)
assert.Equal(t, plaintext, buf.Bytes())
})
}
}
func TestWriter(t *testing.T) {
o := New()
pass := "abcd"
plaintext := []byte("123abc,./vvvczcekdewfeojzaasdsddsdosnd432pdneonkefnoescndisbcisfheosfbdk vsdovsdn]sdlsndlncnepcnodcnviorf409eofnvkdfvjfvdsoijvo")
buf := bytes.NewBuffer(nil)
es := NewWriter(buf, pass, BytesToKeyMD5)
_, err := es.Write(plaintext)
require.NoError(t, err)
require.NoError(t, es.Close())
da, err := o.DecryptBinaryBytes(pass, buf.Bytes(), BytesToKeyMD5)
require.NoError(t, err)
require.Equal(t, da, plaintext)
}