From fd9e6f5edea4620d5c4f867b57536bbad3dba37f Mon Sep 17 00:00:00 2001
From: Kevin F <kevin@voidnet>
Date: Sun, 20 Mar 2022 13:00:04 -0500
Subject: [PATCH] Dandelion phases now have IDs so it can be known more easily
 if an epoch elapsed

---
 src/gossip/dandelion/phase.py | 13 +++++++------
 1 file changed, 7 insertions(+), 6 deletions(-)

diff --git a/src/gossip/dandelion/phase.py b/src/gossip/dandelion/phase.py
index 81344152..998ff36c 100644
--- a/src/gossip/dandelion/phase.py
+++ b/src/gossip/dandelion/phase.py
@@ -9,18 +9,19 @@ class DandelionPhase:
         self.epoch = int(time())
         self.epoch_interval = epoch_interval_secs
         self._is_stem = True
+        self.phase_id = b''
 
 
     def _update_stem_phase(self, cur_time):
         self.epoch = cur_time
-        # Hash the seed with the time stamp to produce 1 pseudorandom byte
-        # We produce an len(8) byte string for year 2038 problem
-        choice = shake_128(
+        # Hash the seed with the time stamp to produce 8 pseudorandom bytes
+        # Produce an len(8) byte string for time as well for year 2038 problem
+        self.phase_id = shake_128(
             self.seed +
-            int.to_bytes(cur_time, 8, 'big')).digest(1)
+            int.to_bytes(cur_time, 8, 'big')).digest(8)
 
-        # If the byte is even
-        if int.from_bytes(choice, 'big') % 2:
+        # Use first byte of phase id as random source for stem phase picking
+        if int.from_bytes(self.phase_id[0], 'big') % 2:
             self._is_stem = True
         else:
             self._is_stem = False