From 09ecd9abc91b90c1e7d2a92958dd5998be90964e Mon Sep 17 00:00:00 2001
From: rprinz08 <richard.prinz@min.at>
Date: Sat, 7 Nov 2020 11:32:50 +0100
Subject: [PATCH] Make commUDP more reliable in case of bad Ethernet connection

---
 litex/tools/remote/comm_udp.py | 41 +++++++++++++++++++++++-----------
 1 file changed, 28 insertions(+), 13 deletions(-)

diff --git a/litex/tools/remote/comm_udp.py b/litex/tools/remote/comm_udp.py
index cd2f98871..a276ddb36 100644
--- a/litex/tools/remote/comm_udp.py
+++ b/litex/tools/remote/comm_udp.py
@@ -22,6 +22,7 @@ class CommUDP:
             return
         self.socket = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
         self.socket.bind(("", self.port))
+        self.socket.settimeout(1)
 
     def close(self):
         if not hasattr(self, "socket"):
@@ -39,16 +40,23 @@ class CommUDP:
         packet = EtherbonePacket()
         packet.records = [record]
         packet.encode()
-        self.socket.sendto(bytes(packet), (self.server, self.port))
 
-        datas, dummy = self.socket.recvfrom(8192)
-        packet = EtherbonePacket(datas)
-        packet.decode()
-        datas = packet.records.pop().writes.get_datas()
-        if self.debug:
-            for i, value in enumerate(datas):
-                print("read {:08x} @ {:08x}".format(value, addr + 4*i))
-        return datas[0] if length is None else datas
+        try:
+            self.socket.sendto(bytes(packet), (self.server, self.port))
+
+            datas, dummy = self.socket.recvfrom(8192)
+            packet = EtherbonePacket(datas)
+            packet.decode()
+            datas = packet.records.pop().writes.get_datas()
+            if self.debug:
+                for i, value in enumerate(datas):
+                    print("read {:08x} @ {:08x}".format(value, addr + 4*i))
+            return datas[0] if length is None else datas
+        except Exception as ex:
+            print("commUDP read: exception ({})".format(ex))
+            self.close()
+            self.open()
+            return []
 
     def write(self, addr, datas):
         datas = datas if isinstance(datas, list) else [datas]
@@ -60,8 +68,15 @@ class CommUDP:
         packet = EtherbonePacket()
         packet.records = [record]
         packet.encode()
-        self.socket.sendto(bytes(packet), (self.server, self.port))
 
-        if self.debug:
-            for i, value in enumerate(datas):
-                print("write {:08x} @ {:08x}".format(value, addr + 4*i))
+        try:
+            self.socket.sendto(bytes(packet), (self.server, self.port))
+
+            if self.debug:
+                for i, value in enumerate(datas):
+                    print("write {:08x} @ {:08x}".format(value, addr + 4*i))
+        except Exception as ex:
+            print("commUDP write: exception ({})".format(ex))
+            self.close()
+            self.open()
+