frontend/stream: Add with_csr parameter to LiteEthUDP2StreamRX, similar to LiteEthStream2UDPTX.

liteeth/frontend/stream.py

@@ -103,22 +103,30 @@ class LiteEthStream2UDPTX(LiteXModule):
 # UDP to Stream RX ---------------------------------------------------------------------------------
 
 class LiteEthUDP2StreamRX(LiteXModule):
-    def __init__(self, ip_address=None, udp_port=None, data_width=8, fifo_depth=None, with_broadcast=True):
+    def __init__(self, ip_address=0, udp_port=0, data_width=8, fifo_depth=None, with_broadcast=True, with_csr=False):
         self.sink   = sink   = stream.Endpoint(eth_udp_user_description(data_width))
         self.source = source = stream.Endpoint(eth_tty_rx_description(data_width))
 
         # # #
 
+        self.ip_address = Signal(32, reset=convert_ip(ip_address))
+        self.udp_port   = Signal(16, reset=udp_port)
+        self.enable     = Signal(reset=1)
+
+        if with_csr:
+            self.add_csr()
+
         valid = Signal(reset=1)
 
+        # Disable RX when enable=0.
+        self.comb += If(~self.enable, valid.eq(0))
+
         # Check UDP Port.
-        assert udp_port is not None
-        self.comb += If(sink.dst_port != udp_port, valid.eq(0))
+        self.comb += If(sink.dst_port != self.udp_port, valid.eq(0))
 
         # Check IP Address (Optional).
-        if (ip_address is not None) and (not with_broadcast):
-            ip_address = convert_ip(ip_address)
-            self.comb += If(sink.ip_address != ip_address, valid.eq(0))
+        if not with_broadcast:
+            self.comb += If(sink.ip_address != self.ip_address, valid.eq(0))
 
         # Data-Path / Buffering (Optional).
         if fifo_depth is None:
@@ -140,6 +148,18 @@ class LiteEthUDP2StreamRX(LiteXModule):
                 fifo.source.connect(source)
             ]
 
+    def add_csr(self):
+        self._enable     = CSRStorage(1,  description="Enable Module", reset=1)
+        self._ip_address = CSRStorage(32, description="IP Address",    reset=self.ip_address.reset.value)
+        self._udp_port   = CSRStorage(16, description="UDP Port",      reset=self.udp_port.reset.value)
+
+        self.comb += [
+            self.enable.eq(self._enable.storage),
+            self.ip_address.eq(self._ip_address.storage),
+            self.udp_port.eq(self._udp_port.storage),
+        ]
+
+
 # UDP Streamer -------------------------------------------------------------------------------------
 
 class LiteEthUDPStreamer(LiteXModule):