WIP: Handling a new pylint design checker message #17
|
@ -272,7 +272,7 @@ class Manager(threading.Thread):
|
||||||
dest_pub_raw = base64.b64decode(
|
dest_pub_raw = base64.b64decode(
|
||||||
shared.i2p_dest_pub, altchars=b'-~')
|
shared.i2p_dest_pub, altchars=b'-~')
|
||||||
obj = structure.Object(
|
obj = structure.Object(
|
||||||
b'\x00' * 8, int(time.time() + 2 * 3600),
|
int(time.time() + 2 * 3600),
|
||||||
shared.i2p_dest_obj_type, shared.i2p_dest_obj_version,
|
shared.i2p_dest_obj_type, shared.i2p_dest_obj_version,
|
||||||
shared.stream, dest_pub_raw)
|
shared.stream, object_payload=dest_pub_raw)
|
||||||
proofofwork.do_pow_and_publish(obj)
|
proofofwork.do_pow_and_publish(obj)
|
||||||
|
|
|
@ -39,8 +39,8 @@ def _worker(obj):
|
||||||
logging.debug(
|
logging.debug(
|
||||||
'Finished doing POW, nonce: %s, time: %ss', nonce, time.time() - t)
|
'Finished doing POW, nonce: %s, time: %ss', nonce, time.time() - t)
|
||||||
obj = structure.Object(
|
obj = structure.Object(
|
||||||
nonce, obj.expires_time, obj.object_type, obj.version,
|
obj.expires_time, obj.object_type, obj.version, obj.stream_number,
|
||||||
obj.stream_number, obj.object_payload)
|
object_payload=obj.object_payload, nonce=nonce)
|
||||||
logging.debug(
|
logging.debug(
|
||||||
'Object vector is %s', base64.b16encode(obj.vector).decode())
|
'Object vector is %s', base64.b16encode(obj.vector).decode())
|
||||||
|
|
||||||
|
|
|
@ -48,8 +48,8 @@ class VarInt():
|
||||||
class Object():
|
class Object():
|
||||||
"""The 'object' message payload"""
|
"""The 'object' message payload"""
|
||||||
def __init__(
|
def __init__(
|
||||||
self, nonce, expires_time, object_type, version,
|
self, expires_time, object_type, version, stream_number,
|
||||||
stream_number, object_payload
|
*, object_payload, tag=None, nonce=b'\x00' * 8
|
||||||
):
|
):
|
||||||
self.nonce = nonce
|
self.nonce = nonce
|
||||||
self.expires_time = expires_time
|
self.expires_time = expires_time
|
||||||
|
@ -60,7 +60,7 @@ class Object():
|
||||||
self.vector = hashlib.sha512(hashlib.sha512(
|
self.vector = hashlib.sha512(hashlib.sha512(
|
||||||
self.to_bytes()).digest()).digest()[:32]
|
self.to_bytes()).digest()).digest()[:32]
|
||||||
|
|
||||||
self.tag = (
|
self.tag = tag or (
|
||||||
# broadcast from version 5 and pubkey/getpukey from version 4
|
# broadcast from version 5 and pubkey/getpukey from version 4
|
||||||
self.object_payload[:32] if object_type == 3 and version == 5
|
self.object_payload[:32] if object_type == 3 and version == 5
|
||||||
or (object_type in (0, 1) and version == 4)
|
or (object_type in (0, 1) and version == 4)
|
||||||
|
@ -84,7 +84,8 @@ class Object():
|
||||||
payload[:stream_number_varint_length]).n
|
payload[:stream_number_varint_length]).n
|
||||||
payload = payload[stream_number_varint_length:]
|
payload = payload[stream_number_varint_length:]
|
||||||
return cls(
|
return cls(
|
||||||
nonce, expires_time, object_type, version, stream_number, payload)
|
expires_time, object_type, version, stream_number,
|
||||||
|
object_payload=payload, nonce=nonce)
|
||||||
|
|
||||||
def to_bytes(self):
|
def to_bytes(self):
|
||||||
"""Serialize to bytes"""
|
"""Serialize to bytes"""
|
||||||
|
|
|
@ -19,7 +19,7 @@ sample_addr_data = unhexlify(
|
||||||
|
|
||||||
# data for an object with expires_time 1697063939
|
# data for an object with expires_time 1697063939
|
||||||
# structure.Object(
|
# structure.Object(
|
||||||
# b'\x00' * 8, expires_time, 42, 1, 2, b'HELLO').to_bytes()
|
# expires_time, 42, 1, 2, object_payload=b'HELLO').to_bytes()
|
||||||
sample_object_data = unhexlify(
|
sample_object_data = unhexlify(
|
||||||
'000000000000000000000000652724030000002a010248454c4c4f')
|
'000000000000000000000000652724030000002a010248454c4c4f')
|
||||||
|
|
||||||
|
@ -135,13 +135,13 @@ class TestStructure(unittest.TestCase):
|
||||||
self.assertEqual(obj.object_payload, b'HELLO')
|
self.assertEqual(obj.object_payload, b'HELLO')
|
||||||
|
|
||||||
obj = structure.Object(
|
obj = structure.Object(
|
||||||
b'\x00' * 8, int(time.time() + 3000000), 42, 1, 1, b'HELLO')
|
int(time.time() + 3000000), 42, 1, 1, object_payload=b'HELLO')
|
||||||
self.assertFalse(obj.is_valid())
|
self.assertFalse(obj.is_valid())
|
||||||
obj.expires_time = int(time.time() - 11000)
|
obj.expires_time = int(time.time() - 11000)
|
||||||
self.assertFalse(obj.is_valid())
|
self.assertFalse(obj.is_valid())
|
||||||
|
|
||||||
obj = structure.Object(
|
obj = structure.Object(
|
||||||
b'\x00' * 8, int(time.time() + 300), 42, 1, 2, b'HELLO')
|
int(time.time() + 300), 42, 1, 2, object_payload=b'HELLO')
|
||||||
vector = obj.vector
|
vector = obj.vector
|
||||||
proofofwork._worker(obj) # pylint: disable=protected-access
|
proofofwork._worker(obj) # pylint: disable=protected-access
|
||||||
obj = shared.objects.popitem()[1]
|
obj = shared.objects.popitem()[1]
|
||||||
|
@ -159,8 +159,8 @@ class TestStructure(unittest.TestCase):
|
||||||
"""Check the main proofofwork call and worker"""
|
"""Check the main proofofwork call and worker"""
|
||||||
shared.vector_advertise_queue = queue.Queue()
|
shared.vector_advertise_queue = queue.Queue()
|
||||||
obj = structure.Object(
|
obj = structure.Object(
|
||||||
b'\x00' * 8, int(time.time() + 300), 42, 1,
|
int(time.time() + 300), 42, 1,
|
||||||
shared.stream, b'HELLO')
|
shared.stream, object_payload=b'HELLO')
|
||||||
start_time = time.time()
|
start_time = time.time()
|
||||||
proofofwork.do_pow_and_publish(obj)
|
proofofwork.do_pow_and_publish(obj)
|
||||||
try:
|
try:
|
||||||
|
@ -189,6 +189,6 @@ class TestStructure(unittest.TestCase):
|
||||||
self.fail("No nonce found in the queue")
|
self.fail("No nonce found in the queue")
|
||||||
|
|
||||||
obj = structure.Object(
|
obj = structure.Object(
|
||||||
nonce, obj.expires_time, obj.object_type, obj.version,
|
obj.expires_time, obj.object_type, obj.version, obj.stream_number,
|
||||||
obj.stream_number, obj.object_payload)
|
object_payload=obj.object_payload, nonce=nonce)
|
||||||
self.assertTrue(obj.is_valid())
|
self.assertTrue(obj.is_valid())
|
||||||
|
|
Loading…
Reference in New Issue
Block a user