mirror of
https://gitlab.torproject.org/tpo/core/tor.git
synced 2024-09-22 14:04:58 +02:00
f3899acdbf
This also adds unit test and a small python script generating a deterministic test vector that a unit test tries to match. Signed-off-by: David Goulet <dgoulet@torproject.org>
38 lines
1.4 KiB
Python
38 lines
1.4 KiB
Python
import sys
|
|
import hashlib
|
|
import struct
|
|
import base64
|
|
|
|
# Python 3.6+, the SHA3 is available in hashlib natively. Else this requires
|
|
# the pysha3 package (pip install pysha3).
|
|
if sys.version_info < (3, 6):
|
|
import sha3
|
|
|
|
# Test vector to make sure the right sha3 version will be used. pysha3 < 1.0
|
|
# used the old Keccak implementation. During the finalization of SHA3, NIST
|
|
# changed the delimiter suffix from 0x01 to 0x06. The Keccak sponge function
|
|
# stayed the same. pysha3 1.0 provides the previous Keccak hash, too.
|
|
TEST_VALUE = "e167f68d6563d75bb25f3aa49c29ef612d41352dc00606de7cbd630bb2665f51"
|
|
if TEST_VALUE != sha3.sha3_256(b"Hello World").hexdigest():
|
|
print("pysha3 version is < 1.0. Please install from:")
|
|
print("https://github.com/tiran/pysha3https://github.com/tiran/pysha3")
|
|
sys.exit(1)
|
|
|
|
# Checksum is built like so:
|
|
# CHECKSUM = SHA3(".onion checksum" || PUBKEY || VERSION)
|
|
PREFIX = ".onion checksum".encode()
|
|
# 32 bytes ed25519 pubkey.
|
|
PUBKEY = ("\x42" * 32).encode()
|
|
# Version 3 is proposal224
|
|
VERSION = 3
|
|
|
|
data = struct.pack('15s32sb', PREFIX, PUBKEY, VERSION)
|
|
checksum = hashlib.sha3_256(data).digest()
|
|
|
|
# Onion address is built like so:
|
|
# onion_address = base32(PUBKEY || CHECKSUM || VERSION) + ".onion"
|
|
address = struct.pack('!32s2sb', PUBKEY, checksum, VERSION)
|
|
onion_addr = base64.b32encode(address).decode().lower()
|
|
|
|
print("%s" % (onion_addr))
|