avoid panic while encrypting empty data (#109)

c2dcc5cc · Ramesh V Rayaprolu · Jeff Wendling · a66df3e4 · c2dcc5cc
Commit c2dcc5cc authored Oct 17, 2018 by Ramesh V Rayaprolu Committed by Jeff Wendling Oct 17, 2018
Hide whitespace changes
Inline Side-by-side

Showing with 6 additions and 0 deletions

ciphers.go ciphers.go +6 -0

No files found.
--- a/ciphers.go
+++ b/ciphers.go
@@ -287,6 +287,9 @@ func NewDecryptionCipherCtx(c *Cipher, e *Engine, key, iv []byte) (
 }

 func (ctx *encryptionCipherCtx) EncryptUpdate(input []byte) ([]byte, error) {
+	if len(input) == 0 {
+		return nil, nil
+	}
 	outbuf := make([]byte, len(input)+ctx.BlockSize())
 	outlen := C.int(len(outbuf))
 	res := C.EVP_EncryptUpdate(ctx.ctx, (*C.uchar)(&outbuf[0]), &outlen,
@@ -298,6 +301,9 @@ func (ctx *encryptionCipherCtx) EncryptUpdate(input []byte) ([]byte, error) {
 }

 func (ctx *decryptionCipherCtx) DecryptUpdate(input []byte) ([]byte, error) {
+	if len(input) == 0 {
+		return nil, nil
+	}
 	outbuf := make([]byte, len(input)+ctx.BlockSize())
 	outlen := C.int(len(outbuf))
 	res := C.EVP_DecryptUpdate(ctx.ctx, (*C.uchar)(&outbuf[0]), &outlen,