Merge pull request #155 from phil-zxx/FixArrayIndexBounds

Fixed Array Bounds for Fragment
iotaledger · Feb 24, 2018 · a64e99c · a64e99c
2 parents 46d2d96 + a6622f7
commit a64e99c
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/iota/crypto/types.py b/iota/crypto/types.py
@@ -189,10 +189,10 @@ def get_digest(self):
       sponge.absorb(key_fragment)
       sponge.squeeze(hash_trits)
 
-      fragment_start  = i * FRAGMENT_LENGTH
-      fragment_end    = fragment_start + FRAGMENT_LENGTH
+      fragment_hash_start  = i * HASH_LENGTH
+      fragment_hash_end    = fragment_hash_start + HASH_LENGTH
 
-      digest[fragment_start:fragment_end] = hash_trits
+      digest[fragment_hash_start:fragment_hash_end] = hash_trits
 
     return Digest(TryteString.from_trits(digest), self.key_index)