From 1a924895559306fabea5a11548da49e7f2b387a5 Mon Sep 17 00:00:00 2001
From: Florent Kermarrec <florent@enjoy-digital.fr>
Date: Tue, 17 Nov 2015 11:38:19 +0100
Subject: [PATCH] soc/tools/remote: add comm_pcie and comm_udp (to be tested)

---
 litex/soc/tools/remote/comm_pcie.py | 44 ++++++++++++++++++++
 litex/soc/tools/remote/comm_udp.py  | 62 +++++++++++++++++++++++++++++
 2 files changed, 106 insertions(+)
 create mode 100644 litex/soc/tools/remote/comm_pcie.py
 create mode 100644 litex/soc/tools/remote/comm_udp.py

diff --git a/litex/soc/tools/remote/comm_pcie.py b/litex/soc/tools/remote/comm_pcie.py
new file mode 100644
index 00000000..0fd3db18
--- /dev/null
+++ b/litex/soc/tools/remote/comm_pcie.py
@@ -0,0 +1,44 @@
+import mmap
+
+
+class CommPCIeLinux:
+    def __init__(self, bar, bar_size, debug=False):
+        self.bar = bar
+        self.bar_size = bar_size
+        self.debug = debug
+
+    def open(self):
+        if hasattr(self, "sysfs"):
+            return
+        self.sysfs = open(self.bar, "r+b")
+        self.sysfs.flush()
+        self.mmap = mmap.mmap(self.sysfs.fileno(), self.bar_size)
+
+    def close(self):
+        if not hasattr(self, "sysfs"):
+            return
+        self.mmap.close()
+        del self.mmap
+        self.sysfs.close()
+        del self.sysfs
+
+    def read(self, addr, length=None):
+        data = []
+        length_int = 1 if length is None else length
+         for i in range(length_int):
+            self.mmap.seek(addr + 4*i)
+            value = int.from_bytes(self.mmap.read(4), "big")
+            if self.debug:
+                print("RD {:08X} @ {:08X}".format(data, addr + 4*i))
+             if length is None:
+                return value
+            data.append(value)
+        return data
+
+    def write(self, addr, data):
+        data = data if isinstance(data, list) else [data]
+        length = len(data)
+        for i, value in enumerate(data):
+            self.mmap[addr + 4*i:addr + 4*(i + 1)] = value.to_bytes(4, byteorder="big")
+            if self.debug:
+                print("WR {:08X} @ {:08X}".format(value, 4*(addr + i)))
diff --git a/litex/soc/tools/remote/comm_udp.py b/litex/soc/tools/remote/comm_udp.py
new file mode 100644
index 00000000..9de91f9e
--- /dev/null
+++ b/litex/soc/tools/remote/comm_udp.py
@@ -0,0 +1,62 @@
+import socket
+
+from litex.soc.tools.remote.etherbone import EtherbonePacket, EtherboneRecord
+from litex.soc.tools.remote.etherbone import EtherboneReads, EtherboneWrites
+
+
+class CommUDP:
+    def __init__(self, host="localhost", port=1234, debug=False):
+        self.host = host
+        self.port = port
+        self.debug = debug
+
+    def open(self):
+        if hasattr(self, "tx_socket"):
+            return
+        self.tx_socket = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
+        self.rx_socket = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
+        self.rx_socket.bind(("", self.port))
+
+    def close(self):
+        if not hasattr(self, "tx_socket"):
+            return
+        self.tx_socket.close()
+        del self.tx_socket
+        self.rx_socket.close()
+        del self.rx_socket
+
+    def read(self, addr, length=None):
+        length_int = 1 if length is None else length
+        record = EtherboneRecord()
+        record.reads = EtherboneReads(addrs=[addr+4*j for j in range(length_int)])
+        record.rcount = len(record.reads)
+
+        packet = EtherbonePacket()
+        packet.records = [record]
+        packet.encode()
+        self.tx_socket.sendto(bytes(packet), (self.server, self.port))
+
+        data, dummy = self.rx_socket.recvfrom(8192)
+        packet = EtherbonePacket(datas)
+        packet.decode()
+        data = packet.records.pop().writes.get_datas()
+        if self.debug:
+            for i, value in enumerate(data):
+                print("RD {:08X} @ {:08X}".format(data, addr + 4*i))
+        return data[0] if length is None else data
+
+    def write(self, addr, data):
+        data = data if isinstance(data, list) else [data]
+        length = len(data)
+        record = EtherboneRecord()
+        record.writes = EtherboneWrites(base_addr=addr, datas=iter(data))
+        record.wcount = len(record.write)
+
+        packet = EtherbonePacket()
+        packet.records = [record]
+        packet.encode()
+        self.tx_socket.sendto(bytes(packet), (self.server, self.port))
+
+        if self.debug:
+            for i, value in enumerate(data):
+                print("WR {:08X} @ {:08X}".format(data, addr + 4*i))
-- 
2.30.2