From c2571480c1a35338e7b1b7d4b5ffeab018ef179e Mon Sep 17 00:00:00 2001
From: rsiddharth <s@ricketyspace.net>
Date: Wed, 9 Dec 2020 20:35:31 -0500
Subject: lib/aes: update AESEncryptCBC

Add padding
---
 lib/aes.go | 7 +++++++
 1 file changed, 7 insertions(+)

(limited to 'lib/aes.go')

diff --git a/lib/aes.go b/lib/aes.go
index 4e835e5..396ee5c 100644
--- a/lib/aes.go
+++ b/lib/aes.go
@@ -4,6 +4,9 @@
 package lib
 
 func AESEncryptCBC(plain, key, iv []byte) []byte {
+	// Pad input
+	plain = Pkcs7Padding(plain, 16)
+
 	iter := len(plain) / 16
 
 	lc := iv
@@ -33,6 +36,10 @@ func AESDecryptCBC(cipher, key, iv []byte) []byte {
 
 		lc = c
 	}
+
+	// Undo padding
+	output = Pkcs7PaddingUndo(output)
+
 	return output
 }
 
-- 
cgit v1.2.3