experiments: add support for netmem and memnic (WIP)

This should probably never make it's way into main. It's a horrible mess in
terms of the class hierarchy.

experiments/simbricks/orchestration/experiments.py

@@ -25,7 +25,8 @@ import typing as tp
 
 from simbricks.orchestration.proxy import NetProxyConnecter, NetProxyListener
 from simbricks.orchestration.simulators import (
-    HostSim, I40eMultiNIC, NetSim, NICSim, PCIDevSim, MemDevSim, Simulator
+    HostSim, I40eMultiNIC, NetSim, NICSim, PCIDevSim, MemDevSim, NetMemSim,
+    Simulator
 )
 
 
@@ -86,6 +87,9 @@ class Experiment(object):
                 raise Exception('Duplicate memdev name')
         self.memdevs.append(sim)
 
+    def add_netmem(self, sim: NetMemSim):
+        self.memdevs.append(sim)
+
     def add_network(self, sim: NetSim):
         for n in self.networks:
             if n.name == sim.name:

experiments/simbricks/orchestration/simulators.py

@@ -198,7 +198,8 @@ class NetSim(Simulator):
         return [s for (_, s) in self.listen_sockets(env)]
 
 
-class MemDevSim(Simulator):
+# FIXME: Class hierarchy is broken here as an ugly hack
+class MemDevSim(NICSim):
     """Base class for memory device simulators."""
     def __init__(self):
         super().__init__()
@@ -221,6 +222,27 @@ class MemDevSim(Simulator):
     def sockets_wait(self, env):
         return [env.dev_mem_path(self)]
 
+class NetMemSim(NICSim):
+    """Base class for netork memory simulators."""
+    def __init__(self):
+        super().__init__()
+
+        self.name = ''
+        self.sync_mode = 0
+        self.start_tick = 0
+        self.sync_period = 500
+        self.eth_latency = 500
+
+    def full_name(self):
+        return 'netmem.' + self.name
+
+    def sockets_cleanup(self, env):
+        return [env.nic_eth_path(self), env.dev_shm_path(self)]
+
+    def sockets_wait(self, env):
+        return [env.nic_eth_path(self)]
+
+
 
 class HostSim(Simulator):
     """Base class for host simulators."""
@@ -676,3 +698,32 @@ class BasicMemDev(MemDevSim):
             f' {self.mem_latency}'
         )
         return cmd
+
+class MemNIC(MemDevSim):
+
+    def run_cmd(self, env):
+        cmd = (
+            f'{env.repodir}/sims/mem/memnic/memnic'
+            f' {env.dev_mem_path(self)} {env.nic_eth_path(self)}'
+            f' {env.dev_shm_path(self)}'
+            f' {self.sync_mode} {self.start_tick} {self.sync_period}'
+            f' {self.mem_latency} {self.eth_latency}'
+        )
+        return cmd
+
+    def sockets_cleanup(self, env):
+        return super().sockets_cleanup(env) + [env.nic_eth_path(self)]
+
+    def sockets_wait(self, env):
+        return super().sockets_wait(env) + [env.nic_eth_path(self)]
+
+class NetMem(NetMemSim):
+
+    def run_cmd(self, env):
+        cmd = (
+            f'{env.repodir}/sims/mem/netmem/netmem'
+            f' {env.nic_eth_path(self)} {env.dev_shm_path(self)}'
+            f' {self.sync_mode} {self.start_tick} {self.sync_period}'
+            f'  {self.eth_latency}'
+        )
+        return cmd