Onionr/static-data/default-plugins/torgossip/runtest.py

115 lines
3.7 KiB
Python
Raw Normal View History

2021-01-31 04:40:51 +00:00
import socket
import os
2021-02-04 01:52:36 +00:00
import secrets
2021-02-05 02:19:23 +00:00
from base64 import b32decode, b32encode
2021-02-05 04:38:57 +00:00
from time import sleep
2021-01-31 04:40:51 +00:00
from utils import identifyhome
from onionrblocks import blockcreator
from blockio import subprocgenerate
from onionrutils import localcommand
import blockio
2021-02-04 01:52:36 +00:00
def _fake_onion():
return b32encode(os.urandom(34)).decode('utf-8') + ".onion"
2021-02-05 04:38:57 +00:00
def _shrink_peer_address(peer):
# strip .onion and b32decode peer address for lower database mem usage
if len(peer) == 62: # constant time
return b32decode(peer[:-6]) # O(N)
return peer
2021-02-04 01:52:36 +00:00
2021-01-31 04:40:51 +00:00
def torgossip_runtest(test_manager):
s_file = identifyhome.identify_home() + "/torgossip.sock"
bl_test = blockcreator.create_anonvdf_block(b"test", "txt", 10)
2021-02-02 23:24:35 +00:00
shared_state = test_manager._too_many
2021-01-31 04:40:51 +00:00
2021-02-02 23:24:35 +00:00
#shared_state.get_by_string("PassToSafeDB").queue_then_store(b"test", "txt", 10)
2021-01-31 04:40:51 +00:00
bl = subprocgenerate.vdf_block(b"test", "txt", 100)
2021-02-02 23:24:35 +00:00
blockio.store_block(bl, shared_state.get_by_string("SafeDB"))
2021-01-31 04:40:51 +00:00
2021-02-01 06:59:53 +00:00
tsts = b''
2021-02-04 01:52:36 +00:00
2021-02-01 06:59:53 +00:00
for i in range(3):
2021-02-04 01:52:36 +00:00
bl2 = subprocgenerate.vdf_block(b"what" + os.urandom(4), "tbt", 100)
tsts += bl2.id
blockio.store_block(bl2, shared_state.get_by_string("SafeDB"))
2021-02-01 06:59:53 +00:00
2021-02-04 01:52:36 +00:00
bl_new = blockcreator.create_anonvdf_block(b"what", "txt", 10)
2021-01-31 04:40:51 +00:00
with socket.socket(socket.AF_UNIX, socket.SOCK_STREAM) as s:
s.connect(s_file)
s.sendall(b'1')
resp = s.recv(5)
assert resp == b"PONG"
s.sendall(b'3txx')
assert s.recv(64) == b"0"
s.sendall(b'3txt')
assert bl.id in s.recv(10000)
# test getting a block that doesn't exist
2021-02-04 01:52:36 +00:00
s.sendall(b'5' + int(secrets.randbits(64)).to_bytes(64, 'little'))
#print(len(s.recv(64)))
assert s.recv(64)[0] == ord(b'0')
2021-01-31 04:40:51 +00:00
# test getting a block that does exist
2021-02-01 06:59:53 +00:00
s.sendall(b'5' + bl.id)
2021-01-31 04:40:51 +00:00
assert s.recv(64) == bl.get_packed()
2021-02-01 06:59:53 +00:00
# test putting block
s.sendall(b'6' + bl_new.id + bl_new.get_packed())
2021-01-31 04:40:51 +00:00
assert s.recv(2) == b"1"
# test block was uploaded by getting it
2021-02-01 06:59:53 +00:00
s.sendall(b'5' + bl_new.id)
2021-01-31 04:40:51 +00:00
assert s.recv(64) == bl_new.get_packed()
2021-02-04 05:21:37 +00:00
s.sendall(b'40,tbt')
2021-02-05 04:57:35 +00:00
assert len(s.recv(100000)) == len(
shared_state.get_by_string("SafeDB").get('bl-tbt'))
2021-02-01 06:59:53 +00:00
2021-02-04 05:21:37 +00:00
s.sendall(b'41,tbt')
assert len(s.recv(100000)) == len(
shared_state.get_by_string("SafeDB").get('bl-tbt')) - 64
2021-02-01 06:59:53 +00:00
2021-02-02 23:24:35 +00:00
# test peer list
2021-02-05 02:19:23 +00:00
fake_peer = _fake_onion()
2021-02-05 04:38:57 +00:00
fakes = []
for i in range(3):
fake = _fake_onion()
fakes.append(fake)
shared_state.get_by_string('TorGossipPeers').add_peer(fake)
2021-02-05 02:19:23 +00:00
shared_state.get_by_string('TorGossipPeers').add_peer(fake_peer)
2021-02-05 04:57:35 +00:00
shared_state.get_by_string(
'TorGossipPeers').add_score(fake_peer, 100000)
2021-02-05 02:19:23 +00:00
s.sendall(b'71')
stored = s.recv(1000)
2021-02-05 04:57:35 +00:00
expected = _shrink_peer_address(fake_peer)
2021-02-05 04:38:57 +00:00
try:
assert stored == expected
except AssertionError:
print(stored, '!=', expected)
2021-02-05 02:19:23 +00:00
shared_state.get_by_string('TorGossipPeers').remove_peer(fake_peer)
2021-02-05 04:38:57 +00:00
for i in fakes:
shared_state.get_by_string('TorGossipPeers').remove_peer(i)
announce_peer = _fake_onion()
announce_raw = _shrink_peer_address(announce_peer)
s.sendall(b'8' + announce_raw)
assert s.recv(1) == b'1'
2021-02-05 04:57:35 +00:00
print(type(announce_raw), type(shared_state.get_by_string(
'TorGossipPeers').get_highest_score_peers(100)[0][0]))
assert announce_raw == shared_state.get_by_string(
'TorGossipPeers').get_highest_score_peers(100)[0][0]
2021-02-05 04:38:57 +00:00
shared_state.get_by_string('TorGossipPeers').remove_peer(announce_raw)
2021-02-02 23:24:35 +00:00
s.sendall(b'9')
assert s.recv(64) == b"BYE"