Fixed potential division by zero

markqvist · Dec 25, 2023 · d67c8eb · d67c8eb
1 parent f4de5d5
commit d67c8eb
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/RNS/Interfaces/Interface.py b/RNS/Interfaces/Interface.py
@@ -68,6 +68,7 @@ def __init__(self):
         self.txb = 0
         self.created = time.time()
         self.online = False
+        self.bitrate = 1e6
 
         self.ingress_control = True
         self.ic_max_held_announces = Interface.MAX_HELD_ANNOUNCES

diff --git a/RNS/Transport.py b/RNS/Transport.py
@@ -876,7 +876,7 @@ def packet_sent(packet):
                                             interface.announce_queue = []
 
                                     queued_announces = True if len(interface.announce_queue) > 0 else False
-                                    if not queued_announces and outbound_time > interface.announce_allowed_at:
+                                    if not queued_announces and outbound_time > interface.announce_allowed_at and interface.bitrate != None and interface.bitrate != 0:
                                         tx_time   = (len(packet.raw)*8) / interface.bitrate
                                         wait_time = (tx_time / interface.announce_cap)
                                         interface.announce_allowed_at = outbound_time + wait_time