From 25fdad62e84a4d1c61a52d5b96dc5325982a30de Mon Sep 17 00:00:00 2001
From: Mattias Andrée <maandree@operamail.com>
Date: Fri, 3 May 2013 20:06:03 +0200
Subject: fix partial byte output
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

Signed-off-by: Mattias Andrée <maandree@operamail.com>
---
 python3/sha3sum.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

(limited to 'python3/sha3sum.py')

diff --git a/python3/sha3sum.py b/python3/sha3sum.py
index 43bdf2f..aa0cced 100755
--- a/python3/sha3sum.py
+++ b/python3/sha3sum.py
@@ -465,7 +465,7 @@ class SHA3:
         ptr = 0
         
         rr = SHA3.r >> 3
-        nn = (SHA3.n + 3) >> 3
+        nn = (SHA3.n + 7) >> 3
         ww = SHA3.w >> 3
         
         # Absorbing phase
@@ -546,6 +546,8 @@ class SHA3:
             olen -= SHA3.r
             if olen > 0:
                 SHA3.keccakF(SHA3.S)
+        if (SHA3.n & 7) != 0:
+            rc[len(rc) - 1] &= (1 << (SHA3.n & 7)) - 1
         
         return bytes(rc)
 
-- 
cgit v1.2.3-70-g09d2