Changeset View
Changeset View
Standalone View
Standalone View
contrib/testgen/gen_base58_test_vectors.py
| Show First 20 Lines • Show All 66 Lines • ▼ Show 20 Lines | while True: | ||||
| metadata = {x: y for x, y in zip( | metadata = {x: y for x, y in zip( | ||||
| metadata_keys, template[3]) if y is not None} | metadata_keys, template[3]) if y is not None} | ||||
| hexrepr = b2a_hex(payload) | hexrepr = b2a_hex(payload) | ||||
| if isinstance(hexrepr, bytes): | if isinstance(hexrepr, bytes): | ||||
| hexrepr = hexrepr.decode('utf8') | hexrepr = hexrepr.decode('utf8') | ||||
| yield (rv, hexrepr, metadata) | yield (rv, hexrepr, metadata) | ||||
| def gen_invalid_vector(template, corrupt_prefix, randomize_payload_size, corrupt_suffix): | def gen_invalid_vector(template, corrupt_prefix, | ||||
| randomize_payload_size, corrupt_suffix): | |||||
| '''Generate possibly invalid vector''' | '''Generate possibly invalid vector''' | ||||
| if corrupt_prefix: | if corrupt_prefix: | ||||
| prefix = os.urandom(1) | prefix = os.urandom(1) | ||||
| else: | else: | ||||
| prefix = bytearray(template[0]) | prefix = bytearray(template[0]) | ||||
| if randomize_payload_size: | if randomize_payload_size: | ||||
| payload = os.urandom(max(int(random.expovariate(0.5)), 50)) | payload = os.urandom(max(int(random.expovariate(0.5)), 50)) | ||||
| ▲ Show 20 Lines • Show All 56 Lines • Show Last 20 Lines | |||||