fix padding

AntonKueltz · Nov 16, 2018 · e7f924b · e7f924b
1 parent 10e0284
commit e7f924b
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 3 deletions.
diff --git a/rfc7539/aead.py b/rfc7539/aead.py
@@ -27,9 +27,16 @@ def _len_bytes(n):
     return slen
 
 
+def _pad16(x):
+    if len(x) % 16 == 0:
+        return b''
+    else:
+        return b'\x00' * (16 - (len(x) % 16))
+
+
 def _tag_data(aad, ciphertext):
-    tag_data = aad + b'\x00' * (16 - (len(aad) % 16))
-    tag_data += ciphertext + b'\x00' * (16 - (len(ciphertext) % 16))
+    tag_data = aad + _pad16(aad)
+    tag_data += ciphertext + _pad16(ciphertext)
     tag_data += _len_bytes(len(aad))
     tag_data += _len_bytes(len(ciphertext))
     return tag_data

diff --git a/setup.py b/setup.py
@@ -32,7 +32,7 @@ def run(self):
 
 setup(
     name='rfc7539',
-    version='1.1.0',
+    version='1.2.1',
     author='Anton Kueltz',
     author_email='[email protected]',
     license='GNU General Public License v3 (GPLv3)',