Merge pull request #634 from paulhauner/patch-10

Add explicit assert to get_permutated_index
This commit is contained in:
Danny Ryan 2019-02-18 10:14:07 -07:00 committed by GitHub
commit 75e649cc15
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 0 deletions

View File

@ -747,6 +747,7 @@ def get_permuted_index(index: int, list_size: int, seed: Bytes32) -> int:
See the 'generalized domain' algorithm on page 3. See the 'generalized domain' algorithm on page 3.
""" """
assert index < list_size assert index < list_size
assert list_size <= 2**40
for round in range(SHUFFLE_ROUND_COUNT): for round in range(SHUFFLE_ROUND_COUNT):
pivot = bytes_to_int(hash(seed + int_to_bytes1(round))[0:8]) % list_size pivot = bytes_to_int(hash(seed + int_to_bytes1(round))[0:8]) % list_size